Home | History | Annotate | Line # | Download | only in libevent
buffer.c revision 1.1.1.7
      1  1.1.1.7  christos /*	$NetBSD: buffer.c,v 1.1.1.7 2016/01/08 21:21:30 christos Exp $	*/
      2  1.1.1.7  christos 
      3      1.1  christos /*
      4      1.1  christos  * Copyright (c) 2002-2007 Niels Provos <provos (at) citi.umich.edu>
      5      1.1  christos  * Copyright (c) 2007-2012 Niels Provos and Nick Mathewson
      6      1.1  christos  *
      7      1.1  christos  * Redistribution and use in source and binary forms, with or without
      8      1.1  christos  * modification, are permitted provided that the following conditions
      9      1.1  christos  * are met:
     10      1.1  christos  * 1. Redistributions of source code must retain the above copyright
     11      1.1  christos  *    notice, this list of conditions and the following disclaimer.
     12      1.1  christos  * 2. Redistributions in binary form must reproduce the above copyright
     13      1.1  christos  *    notice, this list of conditions and the following disclaimer in the
     14      1.1  christos  *    documentation and/or other materials provided with the distribution.
     15      1.1  christos  * 3. The name of the author may not be used to endorse or promote products
     16      1.1  christos  *    derived from this software without specific prior written permission.
     17      1.1  christos  *
     18      1.1  christos  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     19      1.1  christos  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     20      1.1  christos  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     21      1.1  christos  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     22      1.1  christos  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     23      1.1  christos  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     24      1.1  christos  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     25      1.1  christos  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     26      1.1  christos  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     27      1.1  christos  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     28      1.1  christos  */
     29      1.1  christos 
     30      1.1  christos #include "event2/event-config.h"
     31      1.1  christos #include "evconfig-private.h"
     32      1.1  christos 
     33      1.1  christos #ifdef _WIN32
     34      1.1  christos #include <winsock2.h>
     35      1.1  christos #include <windows.h>
     36      1.1  christos #include <io.h>
     37      1.1  christos #endif
     38      1.1  christos 
     39      1.1  christos #ifdef EVENT__HAVE_VASPRINTF
     40      1.1  christos /* If we have vasprintf, we need to define _GNU_SOURCE before we include
     41      1.1  christos  * stdio.h.  This comes from evconfig-private.h.
     42      1.1  christos  */
     43      1.1  christos #endif
     44      1.1  christos 
     45      1.1  christos #include <sys/types.h>
     46      1.1  christos 
     47      1.1  christos #ifdef EVENT__HAVE_SYS_TIME_H
     48      1.1  christos #include <sys/time.h>
     49      1.1  christos #endif
     50      1.1  christos 
     51      1.1  christos #ifdef EVENT__HAVE_SYS_SOCKET_H
     52      1.1  christos #include <sys/socket.h>
     53      1.1  christos #endif
     54      1.1  christos 
     55      1.1  christos #ifdef EVENT__HAVE_SYS_UIO_H
     56      1.1  christos #include <sys/uio.h>
     57      1.1  christos #endif
     58      1.1  christos 
     59      1.1  christos #ifdef EVENT__HAVE_SYS_IOCTL_H
     60      1.1  christos #include <sys/ioctl.h>
     61      1.1  christos #endif
     62      1.1  christos 
     63      1.1  christos #ifdef EVENT__HAVE_SYS_MMAN_H
     64      1.1  christos #include <sys/mman.h>
     65      1.1  christos #endif
     66      1.1  christos 
     67      1.1  christos #ifdef EVENT__HAVE_SYS_SENDFILE_H
     68      1.1  christos #include <sys/sendfile.h>
     69      1.1  christos #endif
     70      1.1  christos #ifdef EVENT__HAVE_SYS_STAT_H
     71      1.1  christos #include <sys/stat.h>
     72      1.1  christos #endif
     73      1.1  christos 
     74      1.1  christos 
     75      1.1  christos #include <errno.h>
     76      1.1  christos #include <stdio.h>
     77      1.1  christos #include <stdlib.h>
     78      1.1  christos #include <string.h>
     79      1.1  christos #ifdef EVENT__HAVE_STDARG_H
     80      1.1  christos #include <stdarg.h>
     81      1.1  christos #endif
     82      1.1  christos #ifdef EVENT__HAVE_UNISTD_H
     83      1.1  christos #include <unistd.h>
     84      1.1  christos #endif
     85      1.1  christos #include <limits.h>
     86      1.1  christos 
     87      1.1  christos #include "event2/event.h"
     88      1.1  christos #include "event2/buffer.h"
     89      1.1  christos #include "event2/buffer_compat.h"
     90      1.1  christos #include "event2/bufferevent.h"
     91      1.1  christos #include "event2/bufferevent_compat.h"
     92      1.1  christos #include "event2/bufferevent_struct.h"
     93      1.1  christos #include "event2/thread.h"
     94      1.1  christos #include "log-internal.h"
     95      1.1  christos #include "mm-internal.h"
     96      1.1  christos #include "util-internal.h"
     97      1.1  christos #include "evthread-internal.h"
     98      1.1  christos #include "evbuffer-internal.h"
     99      1.1  christos #include "bufferevent-internal.h"
    100      1.1  christos 
    101      1.1  christos /* some systems do not have MAP_FAILED */
    102      1.1  christos #ifndef MAP_FAILED
    103      1.1  christos #define MAP_FAILED	((void *)-1)
    104      1.1  christos #endif
    105      1.1  christos 
    106      1.1  christos /* send file support */
    107      1.1  christos #if defined(EVENT__HAVE_SYS_SENDFILE_H) && defined(EVENT__HAVE_SENDFILE) && defined(__linux__)
    108      1.1  christos #define USE_SENDFILE		1
    109      1.1  christos #define SENDFILE_IS_LINUX	1
    110      1.1  christos #elif defined(EVENT__HAVE_SENDFILE) && defined(__FreeBSD__)
    111      1.1  christos #define USE_SENDFILE		1
    112      1.1  christos #define SENDFILE_IS_FREEBSD	1
    113      1.1  christos #elif defined(EVENT__HAVE_SENDFILE) && defined(__APPLE__)
    114      1.1  christos #define USE_SENDFILE		1
    115      1.1  christos #define SENDFILE_IS_MACOSX	1
    116      1.1  christos #elif defined(EVENT__HAVE_SENDFILE) && defined(__sun__) && defined(__svr4__)
    117      1.1  christos #define USE_SENDFILE		1
    118      1.1  christos #define SENDFILE_IS_SOLARIS	1
    119      1.1  christos #endif
    120      1.1  christos 
    121      1.1  christos /* Mask of user-selectable callback flags. */
    122      1.1  christos #define EVBUFFER_CB_USER_FLAGS	    0xffff
    123      1.1  christos /* Mask of all internal-use-only flags. */
    124      1.1  christos #define EVBUFFER_CB_INTERNAL_FLAGS  0xffff0000
    125      1.1  christos 
    126      1.1  christos /* Flag set if the callback is using the cb_obsolete function pointer  */
    127      1.1  christos #define EVBUFFER_CB_OBSOLETE	       0x00040000
    128      1.1  christos 
    129      1.1  christos /* evbuffer_chain support */
    130      1.1  christos #define CHAIN_SPACE_PTR(ch) ((ch)->buffer + (ch)->misalign + (ch)->off)
    131      1.1  christos #define CHAIN_SPACE_LEN(ch) ((ch)->flags & EVBUFFER_IMMUTABLE ? \
    132      1.1  christos 	    0 : (ch)->buffer_len - ((ch)->misalign + (ch)->off))
    133      1.1  christos 
    134      1.1  christos #define CHAIN_PINNED(ch)  (((ch)->flags & EVBUFFER_MEM_PINNED_ANY) != 0)
    135      1.1  christos #define CHAIN_PINNED_R(ch)  (((ch)->flags & EVBUFFER_MEM_PINNED_R) != 0)
    136      1.1  christos 
    137      1.1  christos /* evbuffer_ptr support */
    138      1.1  christos #define PTR_NOT_FOUND(ptr) do {			\
    139      1.1  christos 	(ptr)->pos = -1;					\
    140      1.1  christos 	(ptr)->internal_.chain = NULL;		\
    141      1.1  christos 	(ptr)->internal_.pos_in_chain = 0;	\
    142      1.1  christos } while (0)
    143      1.1  christos 
    144      1.1  christos static void evbuffer_chain_align(struct evbuffer_chain *chain);
    145      1.1  christos static int evbuffer_chain_should_realign(struct evbuffer_chain *chain,
    146      1.1  christos     size_t datalen);
    147      1.1  christos static void evbuffer_deferred_callback(struct event_callback *cb, void *arg);
    148      1.1  christos static int evbuffer_ptr_memcmp(const struct evbuffer *buf,
    149      1.1  christos     const struct evbuffer_ptr *pos, const char *mem, size_t len);
    150      1.1  christos static struct evbuffer_chain *evbuffer_expand_singlechain(struct evbuffer *buf,
    151      1.1  christos     size_t datlen);
    152      1.1  christos static int evbuffer_ptr_subtract(struct evbuffer *buf, struct evbuffer_ptr *pos,
    153      1.1  christos     size_t howfar);
    154      1.1  christos static int evbuffer_file_segment_materialize(struct evbuffer_file_segment *seg);
    155      1.1  christos static inline void evbuffer_chain_incref(struct evbuffer_chain *chain);
    156      1.1  christos 
    157      1.1  christos static struct evbuffer_chain *
    158      1.1  christos evbuffer_chain_new(size_t size)
    159      1.1  christos {
    160      1.1  christos 	struct evbuffer_chain *chain;
    161      1.1  christos 	size_t to_alloc;
    162      1.1  christos 
    163  1.1.1.3  christos 	if (size > EVBUFFER_CHAIN_MAX - EVBUFFER_CHAIN_SIZE)
    164  1.1.1.3  christos 		return (NULL);
    165  1.1.1.3  christos 
    166      1.1  christos 	size += EVBUFFER_CHAIN_SIZE;
    167      1.1  christos 
    168      1.1  christos 	/* get the next largest memory that can hold the buffer */
    169  1.1.1.3  christos 	if (size < EVBUFFER_CHAIN_MAX / 2) {
    170  1.1.1.3  christos 		to_alloc = MIN_BUFFER_SIZE;
    171  1.1.1.3  christos 		while (to_alloc < size) {
    172  1.1.1.3  christos 			to_alloc <<= 1;
    173  1.1.1.3  christos 		}
    174  1.1.1.3  christos 	} else {
    175  1.1.1.3  christos 		to_alloc = size;
    176  1.1.1.3  christos 	}
    177      1.1  christos 
    178      1.1  christos 	/* we get everything in one chunk */
    179      1.1  christos 	if ((chain = mm_malloc(to_alloc)) == NULL)
    180      1.1  christos 		return (NULL);
    181      1.1  christos 
    182      1.1  christos 	memset(chain, 0, EVBUFFER_CHAIN_SIZE);
    183      1.1  christos 
    184      1.1  christos 	chain->buffer_len = to_alloc - EVBUFFER_CHAIN_SIZE;
    185      1.1  christos 
    186      1.1  christos 	/* this way we can manipulate the buffer to different addresses,
    187      1.1  christos 	 * which is required for mmap for example.
    188      1.1  christos 	 */
    189      1.1  christos 	chain->buffer = EVBUFFER_CHAIN_EXTRA(u_char, chain);
    190      1.1  christos 
    191      1.1  christos 	chain->refcnt = 1;
    192      1.1  christos 
    193      1.1  christos 	return (chain);
    194      1.1  christos }
    195      1.1  christos 
    196      1.1  christos static inline void
    197      1.1  christos evbuffer_chain_free(struct evbuffer_chain *chain)
    198      1.1  christos {
    199      1.1  christos 	EVUTIL_ASSERT(chain->refcnt > 0);
    200      1.1  christos 	if (--chain->refcnt > 0) {
    201      1.1  christos 		/* chain is still referenced by other chains */
    202      1.1  christos 		return;
    203      1.1  christos 	}
    204      1.1  christos 
    205      1.1  christos 	if (CHAIN_PINNED(chain)) {
    206      1.1  christos 		/* will get freed once no longer dangling */
    207      1.1  christos 		chain->refcnt++;
    208      1.1  christos 		chain->flags |= EVBUFFER_DANGLING;
    209      1.1  christos 		return;
    210      1.1  christos 	}
    211      1.1  christos 
    212      1.1  christos 	/* safe to release chain, it's either a referencing
    213      1.1  christos 	 * chain or all references to it have been freed */
    214      1.1  christos 	if (chain->flags & EVBUFFER_REFERENCE) {
    215      1.1  christos 		struct evbuffer_chain_reference *info =
    216      1.1  christos 		    EVBUFFER_CHAIN_EXTRA(
    217      1.1  christos 			    struct evbuffer_chain_reference,
    218      1.1  christos 			    chain);
    219      1.1  christos 		if (info->cleanupfn)
    220      1.1  christos 			(*info->cleanupfn)(chain->buffer,
    221      1.1  christos 			    chain->buffer_len,
    222      1.1  christos 			    info->extra);
    223      1.1  christos 	}
    224      1.1  christos 	if (chain->flags & EVBUFFER_FILESEGMENT) {
    225      1.1  christos 		struct evbuffer_chain_file_segment *info =
    226      1.1  christos 		    EVBUFFER_CHAIN_EXTRA(
    227      1.1  christos 			    struct evbuffer_chain_file_segment,
    228      1.1  christos 			    chain);
    229      1.1  christos 		if (info->segment) {
    230      1.1  christos #ifdef _WIN32
    231      1.1  christos 			if (info->segment->is_mapping)
    232      1.1  christos 				UnmapViewOfFile(chain->buffer);
    233      1.1  christos #endif
    234      1.1  christos 			evbuffer_file_segment_free(info->segment);
    235      1.1  christos 		}
    236      1.1  christos 	}
    237      1.1  christos 	if (chain->flags & EVBUFFER_MULTICAST) {
    238      1.1  christos 		struct evbuffer_multicast_parent *info =
    239      1.1  christos 		    EVBUFFER_CHAIN_EXTRA(
    240      1.1  christos 			    struct evbuffer_multicast_parent,
    241      1.1  christos 			    chain);
    242      1.1  christos 		/* referencing chain is being freed, decrease
    243      1.1  christos 		 * refcounts of source chain and associated
    244      1.1  christos 		 * evbuffer (which get freed once both reach
    245      1.1  christos 		 * zero) */
    246      1.1  christos 		EVUTIL_ASSERT(info->source != NULL);
    247      1.1  christos 		EVUTIL_ASSERT(info->parent != NULL);
    248      1.1  christos 		EVBUFFER_LOCK(info->source);
    249      1.1  christos 		evbuffer_chain_free(info->parent);
    250      1.1  christos 		evbuffer_decref_and_unlock_(info->source);
    251      1.1  christos 	}
    252      1.1  christos 
    253      1.1  christos 	mm_free(chain);
    254      1.1  christos }
    255      1.1  christos 
    256      1.1  christos static void
    257      1.1  christos evbuffer_free_all_chains(struct evbuffer_chain *chain)
    258      1.1  christos {
    259      1.1  christos 	struct evbuffer_chain *next;
    260      1.1  christos 	for (; chain; chain = next) {
    261      1.1  christos 		next = chain->next;
    262      1.1  christos 		evbuffer_chain_free(chain);
    263      1.1  christos 	}
    264      1.1  christos }
    265      1.1  christos 
    266      1.1  christos #ifndef NDEBUG
    267      1.1  christos static int
    268      1.1  christos evbuffer_chains_all_empty(struct evbuffer_chain *chain)
    269      1.1  christos {
    270      1.1  christos 	for (; chain; chain = chain->next) {
    271      1.1  christos 		if (chain->off)
    272      1.1  christos 			return 0;
    273      1.1  christos 	}
    274      1.1  christos 	return 1;
    275      1.1  christos }
    276      1.1  christos #else
    277      1.1  christos /* The definition is needed for EVUTIL_ASSERT, which uses sizeof to avoid
    278      1.1  christos "unused variable" warnings. */
    279      1.1  christos static inline int evbuffer_chains_all_empty(struct evbuffer_chain *chain) {
    280      1.1  christos 	return 1;
    281      1.1  christos }
    282      1.1  christos #endif
    283      1.1  christos 
    284      1.1  christos /* Free all trailing chains in 'buf' that are neither pinned nor empty, prior
    285      1.1  christos  * to replacing them all with a new chain.  Return a pointer to the place
    286      1.1  christos  * where the new chain will go.
    287      1.1  christos  *
    288      1.1  christos  * Internal; requires lock.  The caller must fix up buf->last and buf->first
    289      1.1  christos  * as needed; they might have been freed.
    290      1.1  christos  */
    291      1.1  christos static struct evbuffer_chain **
    292      1.1  christos evbuffer_free_trailing_empty_chains(struct evbuffer *buf)
    293      1.1  christos {
    294      1.1  christos 	struct evbuffer_chain **ch = buf->last_with_datap;
    295      1.1  christos 	/* Find the first victim chain.  It might be *last_with_datap */
    296      1.1  christos 	while ((*ch) && ((*ch)->off != 0 || CHAIN_PINNED(*ch)))
    297      1.1  christos 		ch = &(*ch)->next;
    298      1.1  christos 	if (*ch) {
    299      1.1  christos 		EVUTIL_ASSERT(evbuffer_chains_all_empty(*ch));
    300      1.1  christos 		evbuffer_free_all_chains(*ch);
    301      1.1  christos 		*ch = NULL;
    302      1.1  christos 	}
    303      1.1  christos 	return ch;
    304      1.1  christos }
    305      1.1  christos 
    306      1.1  christos /* Add a single chain 'chain' to the end of 'buf', freeing trailing empty
    307      1.1  christos  * chains as necessary.  Requires lock.  Does not schedule callbacks.
    308      1.1  christos  */
    309      1.1  christos static void
    310      1.1  christos evbuffer_chain_insert(struct evbuffer *buf,
    311      1.1  christos     struct evbuffer_chain *chain)
    312      1.1  christos {
    313      1.1  christos 	ASSERT_EVBUFFER_LOCKED(buf);
    314      1.1  christos 	if (*buf->last_with_datap == NULL) {
    315      1.1  christos 		/* There are no chains data on the buffer at all. */
    316      1.1  christos 		EVUTIL_ASSERT(buf->last_with_datap == &buf->first);
    317      1.1  christos 		EVUTIL_ASSERT(buf->first == NULL);
    318      1.1  christos 		buf->first = buf->last = chain;
    319      1.1  christos 	} else {
    320      1.1  christos 		struct evbuffer_chain **chp;
    321      1.1  christos 		chp = evbuffer_free_trailing_empty_chains(buf);
    322      1.1  christos 		*chp = chain;
    323      1.1  christos 		if (chain->off)
    324      1.1  christos 			buf->last_with_datap = chp;
    325      1.1  christos 		buf->last = chain;
    326      1.1  christos 	}
    327      1.1  christos 	buf->total_len += chain->off;
    328      1.1  christos }
    329      1.1  christos 
    330      1.1  christos static inline struct evbuffer_chain *
    331      1.1  christos evbuffer_chain_insert_new(struct evbuffer *buf, size_t datlen)
    332      1.1  christos {
    333      1.1  christos 	struct evbuffer_chain *chain;
    334      1.1  christos 	if ((chain = evbuffer_chain_new(datlen)) == NULL)
    335      1.1  christos 		return NULL;
    336      1.1  christos 	evbuffer_chain_insert(buf, chain);
    337      1.1  christos 	return chain;
    338      1.1  christos }
    339      1.1  christos 
    340      1.1  christos void
    341      1.1  christos evbuffer_chain_pin_(struct evbuffer_chain *chain, unsigned flag)
    342      1.1  christos {
    343      1.1  christos 	EVUTIL_ASSERT((chain->flags & flag) == 0);
    344      1.1  christos 	chain->flags |= flag;
    345      1.1  christos }
    346      1.1  christos 
    347      1.1  christos void
    348      1.1  christos evbuffer_chain_unpin_(struct evbuffer_chain *chain, unsigned flag)
    349      1.1  christos {
    350      1.1  christos 	EVUTIL_ASSERT((chain->flags & flag) != 0);
    351      1.1  christos 	chain->flags &= ~flag;
    352      1.1  christos 	if (chain->flags & EVBUFFER_DANGLING)
    353      1.1  christos 		evbuffer_chain_free(chain);
    354      1.1  christos }
    355      1.1  christos 
    356      1.1  christos static inline void
    357      1.1  christos evbuffer_chain_incref(struct evbuffer_chain *chain)
    358      1.1  christos {
    359      1.1  christos     ++chain->refcnt;
    360      1.1  christos }
    361      1.1  christos 
    362      1.1  christos struct evbuffer *
    363      1.1  christos evbuffer_new(void)
    364      1.1  christos {
    365      1.1  christos 	struct evbuffer *buffer;
    366      1.1  christos 
    367      1.1  christos 	buffer = mm_calloc(1, sizeof(struct evbuffer));
    368      1.1  christos 	if (buffer == NULL)
    369      1.1  christos 		return (NULL);
    370      1.1  christos 
    371      1.1  christos 	LIST_INIT(&buffer->callbacks);
    372      1.1  christos 	buffer->refcnt = 1;
    373      1.1  christos 	buffer->last_with_datap = &buffer->first;
    374      1.1  christos 
    375      1.1  christos 	return (buffer);
    376      1.1  christos }
    377      1.1  christos 
    378      1.1  christos int
    379      1.1  christos evbuffer_set_flags(struct evbuffer *buf, ev_uint64_t flags)
    380      1.1  christos {
    381      1.1  christos 	EVBUFFER_LOCK(buf);
    382      1.1  christos 	buf->flags |= (ev_uint32_t)flags;
    383      1.1  christos 	EVBUFFER_UNLOCK(buf);
    384      1.1  christos 	return 0;
    385      1.1  christos }
    386      1.1  christos 
    387      1.1  christos int
    388      1.1  christos evbuffer_clear_flags(struct evbuffer *buf, ev_uint64_t flags)
    389      1.1  christos {
    390      1.1  christos 	EVBUFFER_LOCK(buf);
    391      1.1  christos 	buf->flags &= ~(ev_uint32_t)flags;
    392      1.1  christos 	EVBUFFER_UNLOCK(buf);
    393      1.1  christos 	return 0;
    394      1.1  christos }
    395      1.1  christos 
    396      1.1  christos void
    397      1.1  christos evbuffer_incref_(struct evbuffer *buf)
    398      1.1  christos {
    399      1.1  christos 	EVBUFFER_LOCK(buf);
    400      1.1  christos 	++buf->refcnt;
    401      1.1  christos 	EVBUFFER_UNLOCK(buf);
    402      1.1  christos }
    403      1.1  christos 
    404      1.1  christos void
    405      1.1  christos evbuffer_incref_and_lock_(struct evbuffer *buf)
    406      1.1  christos {
    407      1.1  christos 	EVBUFFER_LOCK(buf);
    408      1.1  christos 	++buf->refcnt;
    409      1.1  christos }
    410      1.1  christos 
    411      1.1  christos int
    412      1.1  christos evbuffer_defer_callbacks(struct evbuffer *buffer, struct event_base *base)
    413      1.1  christos {
    414      1.1  christos 	EVBUFFER_LOCK(buffer);
    415      1.1  christos 	buffer->cb_queue = base;
    416      1.1  christos 	buffer->deferred_cbs = 1;
    417      1.1  christos 	event_deferred_cb_init_(&buffer->deferred,
    418      1.1  christos 	    event_base_get_npriorities(base) / 2,
    419      1.1  christos 	    evbuffer_deferred_callback, buffer);
    420      1.1  christos 	EVBUFFER_UNLOCK(buffer);
    421      1.1  christos 	return 0;
    422      1.1  christos }
    423      1.1  christos 
    424      1.1  christos int
    425      1.1  christos evbuffer_enable_locking(struct evbuffer *buf, void *lock)
    426      1.1  christos {
    427      1.1  christos #ifdef EVENT__DISABLE_THREAD_SUPPORT
    428      1.1  christos 	return -1;
    429      1.1  christos #else
    430      1.1  christos 	if (buf->lock)
    431      1.1  christos 		return -1;
    432      1.1  christos 
    433      1.1  christos 	if (!lock) {
    434      1.1  christos 		EVTHREAD_ALLOC_LOCK(lock, EVTHREAD_LOCKTYPE_RECURSIVE);
    435      1.1  christos 		if (!lock)
    436      1.1  christos 			return -1;
    437      1.1  christos 		buf->lock = lock;
    438      1.1  christos 		buf->own_lock = 1;
    439      1.1  christos 	} else {
    440      1.1  christos 		buf->lock = lock;
    441      1.1  christos 		buf->own_lock = 0;
    442      1.1  christos 	}
    443      1.1  christos 
    444      1.1  christos 	return 0;
    445      1.1  christos #endif
    446      1.1  christos }
    447      1.1  christos 
    448      1.1  christos void
    449      1.1  christos evbuffer_set_parent_(struct evbuffer *buf, struct bufferevent *bev)
    450      1.1  christos {
    451      1.1  christos 	EVBUFFER_LOCK(buf);
    452      1.1  christos 	buf->parent = bev;
    453      1.1  christos 	EVBUFFER_UNLOCK(buf);
    454      1.1  christos }
    455      1.1  christos 
    456      1.1  christos static void
    457      1.1  christos evbuffer_run_callbacks(struct evbuffer *buffer, int running_deferred)
    458      1.1  christos {
    459      1.1  christos 	struct evbuffer_cb_entry *cbent, *next;
    460      1.1  christos 	struct evbuffer_cb_info info;
    461      1.1  christos 	size_t new_size;
    462      1.1  christos 	ev_uint32_t mask, masked_val;
    463      1.1  christos 	int clear = 1;
    464      1.1  christos 
    465      1.1  christos 	if (running_deferred) {
    466      1.1  christos 		mask = EVBUFFER_CB_NODEFER|EVBUFFER_CB_ENABLED;
    467      1.1  christos 		masked_val = EVBUFFER_CB_ENABLED;
    468      1.1  christos 	} else if (buffer->deferred_cbs) {
    469      1.1  christos 		mask = EVBUFFER_CB_NODEFER|EVBUFFER_CB_ENABLED;
    470      1.1  christos 		masked_val = EVBUFFER_CB_NODEFER|EVBUFFER_CB_ENABLED;
    471      1.1  christos 		/* Don't zero-out n_add/n_del, since the deferred callbacks
    472      1.1  christos 		   will want to see them. */
    473      1.1  christos 		clear = 0;
    474      1.1  christos 	} else {
    475      1.1  christos 		mask = EVBUFFER_CB_ENABLED;
    476      1.1  christos 		masked_val = EVBUFFER_CB_ENABLED;
    477      1.1  christos 	}
    478      1.1  christos 
    479      1.1  christos 	ASSERT_EVBUFFER_LOCKED(buffer);
    480      1.1  christos 
    481      1.1  christos 	if (LIST_EMPTY(&buffer->callbacks)) {
    482      1.1  christos 		buffer->n_add_for_cb = buffer->n_del_for_cb = 0;
    483      1.1  christos 		return;
    484      1.1  christos 	}
    485      1.1  christos 	if (buffer->n_add_for_cb == 0 && buffer->n_del_for_cb == 0)
    486      1.1  christos 		return;
    487      1.1  christos 
    488      1.1  christos 	new_size = buffer->total_len;
    489      1.1  christos 	info.orig_size = new_size + buffer->n_del_for_cb - buffer->n_add_for_cb;
    490      1.1  christos 	info.n_added = buffer->n_add_for_cb;
    491      1.1  christos 	info.n_deleted = buffer->n_del_for_cb;
    492      1.1  christos 	if (clear) {
    493      1.1  christos 		buffer->n_add_for_cb = 0;
    494      1.1  christos 		buffer->n_del_for_cb = 0;
    495      1.1  christos 	}
    496      1.1  christos 	for (cbent = LIST_FIRST(&buffer->callbacks);
    497      1.1  christos 	     cbent != LIST_END(&buffer->callbacks);
    498      1.1  christos 	     cbent = next) {
    499      1.1  christos 		/* Get the 'next' pointer now in case this callback decides
    500      1.1  christos 		 * to remove itself or something. */
    501      1.1  christos 		next = LIST_NEXT(cbent, next);
    502      1.1  christos 
    503      1.1  christos 		if ((cbent->flags & mask) != masked_val)
    504      1.1  christos 			continue;
    505      1.1  christos 
    506      1.1  christos 		if ((cbent->flags & EVBUFFER_CB_OBSOLETE))
    507      1.1  christos 			cbent->cb.cb_obsolete(buffer,
    508      1.1  christos 			    info.orig_size, new_size, cbent->cbarg);
    509      1.1  christos 		else
    510      1.1  christos 			cbent->cb.cb_func(buffer, &info, cbent->cbarg);
    511      1.1  christos 	}
    512      1.1  christos }
    513      1.1  christos 
    514      1.1  christos void
    515      1.1  christos evbuffer_invoke_callbacks_(struct evbuffer *buffer)
    516      1.1  christos {
    517      1.1  christos 	if (LIST_EMPTY(&buffer->callbacks)) {
    518      1.1  christos 		buffer->n_add_for_cb = buffer->n_del_for_cb = 0;
    519      1.1  christos 		return;
    520      1.1  christos 	}
    521      1.1  christos 
    522      1.1  christos 	if (buffer->deferred_cbs) {
    523      1.1  christos 		if (event_deferred_cb_schedule_(buffer->cb_queue, &buffer->deferred)) {
    524      1.1  christos 			evbuffer_incref_and_lock_(buffer);
    525      1.1  christos 			if (buffer->parent)
    526      1.1  christos 				bufferevent_incref_(buffer->parent);
    527      1.1  christos 		}
    528      1.1  christos 		EVBUFFER_UNLOCK(buffer);
    529      1.1  christos 	}
    530      1.1  christos 
    531      1.1  christos 	evbuffer_run_callbacks(buffer, 0);
    532      1.1  christos }
    533      1.1  christos 
    534      1.1  christos static void
    535      1.1  christos evbuffer_deferred_callback(struct event_callback *cb, void *arg)
    536      1.1  christos {
    537      1.1  christos 	struct bufferevent *parent = NULL;
    538      1.1  christos 	struct evbuffer *buffer = arg;
    539      1.1  christos 
    540      1.1  christos 	/* XXXX It would be better to run these callbacks without holding the
    541      1.1  christos 	 * lock */
    542      1.1  christos 	EVBUFFER_LOCK(buffer);
    543      1.1  christos 	parent = buffer->parent;
    544      1.1  christos 	evbuffer_run_callbacks(buffer, 1);
    545      1.1  christos 	evbuffer_decref_and_unlock_(buffer);
    546      1.1  christos 	if (parent)
    547      1.1  christos 		bufferevent_decref_(parent);
    548      1.1  christos }
    549      1.1  christos 
    550      1.1  christos static void
    551      1.1  christos evbuffer_remove_all_callbacks(struct evbuffer *buffer)
    552      1.1  christos {
    553      1.1  christos 	struct evbuffer_cb_entry *cbent;
    554      1.1  christos 
    555      1.1  christos 	while ((cbent = LIST_FIRST(&buffer->callbacks))) {
    556      1.1  christos 		LIST_REMOVE(cbent, next);
    557      1.1  christos 		mm_free(cbent);
    558      1.1  christos 	}
    559      1.1  christos }
    560      1.1  christos 
    561      1.1  christos void
    562      1.1  christos evbuffer_decref_and_unlock_(struct evbuffer *buffer)
    563      1.1  christos {
    564      1.1  christos 	struct evbuffer_chain *chain, *next;
    565      1.1  christos 	ASSERT_EVBUFFER_LOCKED(buffer);
    566      1.1  christos 
    567      1.1  christos 	EVUTIL_ASSERT(buffer->refcnt > 0);
    568      1.1  christos 
    569      1.1  christos 	if (--buffer->refcnt > 0) {
    570      1.1  christos 		EVBUFFER_UNLOCK(buffer);
    571      1.1  christos 		return;
    572      1.1  christos 	}
    573      1.1  christos 
    574      1.1  christos 	for (chain = buffer->first; chain != NULL; chain = next) {
    575      1.1  christos 		next = chain->next;
    576      1.1  christos 		evbuffer_chain_free(chain);
    577      1.1  christos 	}
    578      1.1  christos 	evbuffer_remove_all_callbacks(buffer);
    579      1.1  christos 	if (buffer->deferred_cbs)
    580      1.1  christos 		event_deferred_cb_cancel_(buffer->cb_queue, &buffer->deferred);
    581      1.1  christos 
    582      1.1  christos 	EVBUFFER_UNLOCK(buffer);
    583      1.1  christos 	if (buffer->own_lock)
    584      1.1  christos 		EVTHREAD_FREE_LOCK(buffer->lock, EVTHREAD_LOCKTYPE_RECURSIVE);
    585      1.1  christos 	mm_free(buffer);
    586      1.1  christos }
    587      1.1  christos 
    588      1.1  christos void
    589      1.1  christos evbuffer_free(struct evbuffer *buffer)
    590      1.1  christos {
    591      1.1  christos 	EVBUFFER_LOCK(buffer);
    592      1.1  christos 	evbuffer_decref_and_unlock_(buffer);
    593      1.1  christos }
    594      1.1  christos 
    595      1.1  christos void
    596      1.1  christos evbuffer_lock(struct evbuffer *buf)
    597      1.1  christos {
    598      1.1  christos 	EVBUFFER_LOCK(buf);
    599      1.1  christos }
    600      1.1  christos 
    601      1.1  christos void
    602      1.1  christos evbuffer_unlock(struct evbuffer *buf)
    603      1.1  christos {
    604      1.1  christos 	EVBUFFER_UNLOCK(buf);
    605      1.1  christos }
    606      1.1  christos 
    607      1.1  christos size_t
    608      1.1  christos evbuffer_get_length(const struct evbuffer *buffer)
    609      1.1  christos {
    610      1.1  christos 	size_t result;
    611      1.1  christos 
    612      1.1  christos 	EVBUFFER_LOCK(buffer);
    613      1.1  christos 
    614      1.1  christos 	result = (buffer->total_len);
    615      1.1  christos 
    616      1.1  christos 	EVBUFFER_UNLOCK(buffer);
    617      1.1  christos 
    618      1.1  christos 	return result;
    619      1.1  christos }
    620      1.1  christos 
    621      1.1  christos size_t
    622      1.1  christos evbuffer_get_contiguous_space(const struct evbuffer *buf)
    623      1.1  christos {
    624      1.1  christos 	struct evbuffer_chain *chain;
    625      1.1  christos 	size_t result;
    626      1.1  christos 
    627      1.1  christos 	EVBUFFER_LOCK(buf);
    628      1.1  christos 	chain = buf->first;
    629      1.1  christos 	result = (chain != NULL ? chain->off : 0);
    630      1.1  christos 	EVBUFFER_UNLOCK(buf);
    631      1.1  christos 
    632      1.1  christos 	return result;
    633      1.1  christos }
    634      1.1  christos 
    635      1.1  christos size_t
    636      1.1  christos evbuffer_add_iovec(struct evbuffer * buf, struct evbuffer_iovec * vec, int n_vec) {
    637      1.1  christos 	int n;
    638      1.1  christos 	size_t res;
    639      1.1  christos 	size_t to_alloc;
    640      1.1  christos 
    641      1.1  christos 	EVBUFFER_LOCK(buf);
    642      1.1  christos 
    643      1.1  christos 	res = to_alloc = 0;
    644      1.1  christos 
    645      1.1  christos 	for (n = 0; n < n_vec; n++) {
    646      1.1  christos 		to_alloc += vec[n].iov_len;
    647      1.1  christos 	}
    648      1.1  christos 
    649      1.1  christos 	if (evbuffer_expand_fast_(buf, to_alloc, 2) < 0) {
    650      1.1  christos 		goto done;
    651      1.1  christos 	}
    652      1.1  christos 
    653      1.1  christos 	for (n = 0; n < n_vec; n++) {
    654      1.1  christos 		/* XXX each 'add' call here does a bunch of setup that's
    655      1.1  christos 		 * obviated by evbuffer_expand_fast_, and some cleanup that we
    656      1.1  christos 		 * would like to do only once.  Instead we should just extract
    657      1.1  christos 		 * the part of the code that's needed. */
    658      1.1  christos 
    659      1.1  christos 		if (evbuffer_add(buf, vec[n].iov_base, vec[n].iov_len) < 0) {
    660      1.1  christos 			goto done;
    661      1.1  christos 		}
    662      1.1  christos 
    663      1.1  christos 		res += vec[n].iov_len;
    664      1.1  christos 	}
    665      1.1  christos 
    666      1.1  christos done:
    667      1.1  christos     EVBUFFER_UNLOCK(buf);
    668      1.1  christos     return res;
    669      1.1  christos }
    670      1.1  christos 
    671      1.1  christos int
    672      1.1  christos evbuffer_reserve_space(struct evbuffer *buf, ev_ssize_t size,
    673      1.1  christos     struct evbuffer_iovec *vec, int n_vecs)
    674      1.1  christos {
    675      1.1  christos 	struct evbuffer_chain *chain, **chainp;
    676      1.1  christos 	int n = -1;
    677      1.1  christos 
    678      1.1  christos 	EVBUFFER_LOCK(buf);
    679      1.1  christos 	if (buf->freeze_end)
    680      1.1  christos 		goto done;
    681      1.1  christos 	if (n_vecs < 1)
    682      1.1  christos 		goto done;
    683      1.1  christos 	if (n_vecs == 1) {
    684      1.1  christos 		if ((chain = evbuffer_expand_singlechain(buf, size)) == NULL)
    685      1.1  christos 			goto done;
    686      1.1  christos 
    687      1.1  christos 		vec[0].iov_base = CHAIN_SPACE_PTR(chain);
    688      1.1  christos 		vec[0].iov_len = (size_t) CHAIN_SPACE_LEN(chain);
    689      1.1  christos 		EVUTIL_ASSERT(size<0 || (size_t)vec[0].iov_len >= (size_t)size);
    690      1.1  christos 		n = 1;
    691      1.1  christos 	} else {
    692      1.1  christos 		if (evbuffer_expand_fast_(buf, size, n_vecs)<0)
    693      1.1  christos 			goto done;
    694      1.1  christos 		n = evbuffer_read_setup_vecs_(buf, size, vec, n_vecs,
    695      1.1  christos 				&chainp, 0);
    696      1.1  christos 	}
    697      1.1  christos 
    698      1.1  christos done:
    699      1.1  christos 	EVBUFFER_UNLOCK(buf);
    700      1.1  christos 	return n;
    701      1.1  christos 
    702      1.1  christos }
    703      1.1  christos 
    704      1.1  christos static int
    705      1.1  christos advance_last_with_data(struct evbuffer *buf)
    706      1.1  christos {
    707      1.1  christos 	int n = 0;
    708      1.1  christos 	ASSERT_EVBUFFER_LOCKED(buf);
    709      1.1  christos 
    710      1.1  christos 	if (!*buf->last_with_datap)
    711      1.1  christos 		return 0;
    712      1.1  christos 
    713      1.1  christos 	while ((*buf->last_with_datap)->next && (*buf->last_with_datap)->next->off) {
    714      1.1  christos 		buf->last_with_datap = &(*buf->last_with_datap)->next;
    715      1.1  christos 		++n;
    716      1.1  christos 	}
    717      1.1  christos 	return n;
    718      1.1  christos }
    719      1.1  christos 
    720      1.1  christos int
    721      1.1  christos evbuffer_commit_space(struct evbuffer *buf,
    722      1.1  christos     struct evbuffer_iovec *vec, int n_vecs)
    723      1.1  christos {
    724      1.1  christos 	struct evbuffer_chain *chain, **firstchainp, **chainp;
    725      1.1  christos 	int result = -1;
    726      1.1  christos 	size_t added = 0;
    727      1.1  christos 	int i;
    728      1.1  christos 
    729      1.1  christos 	EVBUFFER_LOCK(buf);
    730      1.1  christos 
    731      1.1  christos 	if (buf->freeze_end)
    732      1.1  christos 		goto done;
    733      1.1  christos 	if (n_vecs == 0) {
    734      1.1  christos 		result = 0;
    735      1.1  christos 		goto done;
    736      1.1  christos 	} else if (n_vecs == 1 &&
    737      1.1  christos 	    (buf->last && vec[0].iov_base == (void*)CHAIN_SPACE_PTR(buf->last))) {
    738      1.1  christos 		/* The user only got or used one chain; it might not
    739      1.1  christos 		 * be the first one with space in it. */
    740      1.1  christos 		if ((size_t)vec[0].iov_len > (size_t)CHAIN_SPACE_LEN(buf->last))
    741      1.1  christos 			goto done;
    742      1.1  christos 		buf->last->off += vec[0].iov_len;
    743      1.1  christos 		added = vec[0].iov_len;
    744      1.1  christos 		if (added)
    745      1.1  christos 			advance_last_with_data(buf);
    746      1.1  christos 		goto okay;
    747      1.1  christos 	}
    748      1.1  christos 
    749      1.1  christos 	/* Advance 'firstchain' to the first chain with space in it. */
    750      1.1  christos 	firstchainp = buf->last_with_datap;
    751      1.1  christos 	if (!*firstchainp)
    752      1.1  christos 		goto done;
    753      1.1  christos 	if (CHAIN_SPACE_LEN(*firstchainp) == 0) {
    754      1.1  christos 		firstchainp = &(*firstchainp)->next;
    755      1.1  christos 	}
    756      1.1  christos 
    757      1.1  christos 	chain = *firstchainp;
    758      1.1  christos 	/* pass 1: make sure that the pointers and lengths of vecs[] are in
    759      1.1  christos 	 * bounds before we try to commit anything. */
    760      1.1  christos 	for (i=0; i<n_vecs; ++i) {
    761      1.1  christos 		if (!chain)
    762      1.1  christos 			goto done;
    763      1.1  christos 		if (vec[i].iov_base != (void*)CHAIN_SPACE_PTR(chain) ||
    764      1.1  christos 		    (size_t)vec[i].iov_len > CHAIN_SPACE_LEN(chain))
    765      1.1  christos 			goto done;
    766      1.1  christos 		chain = chain->next;
    767      1.1  christos 	}
    768      1.1  christos 	/* pass 2: actually adjust all the chains. */
    769      1.1  christos 	chainp = firstchainp;
    770      1.1  christos 	for (i=0; i<n_vecs; ++i) {
    771      1.1  christos 		(*chainp)->off += vec[i].iov_len;
    772      1.1  christos 		added += vec[i].iov_len;
    773      1.1  christos 		if (vec[i].iov_len) {
    774      1.1  christos 			buf->last_with_datap = chainp;
    775      1.1  christos 		}
    776      1.1  christos 		chainp = &(*chainp)->next;
    777      1.1  christos 	}
    778      1.1  christos 
    779      1.1  christos okay:
    780      1.1  christos 	buf->total_len += added;
    781      1.1  christos 	buf->n_add_for_cb += added;
    782      1.1  christos 	result = 0;
    783      1.1  christos 	evbuffer_invoke_callbacks_(buf);
    784      1.1  christos 
    785      1.1  christos done:
    786      1.1  christos 	EVBUFFER_UNLOCK(buf);
    787      1.1  christos 	return result;
    788      1.1  christos }
    789      1.1  christos 
    790      1.1  christos static inline int
    791      1.1  christos HAS_PINNED_R(struct evbuffer *buf)
    792      1.1  christos {
    793      1.1  christos 	return (buf->last && CHAIN_PINNED_R(buf->last));
    794      1.1  christos }
    795      1.1  christos 
    796      1.1  christos static inline void
    797      1.1  christos ZERO_CHAIN(struct evbuffer *dst)
    798      1.1  christos {
    799      1.1  christos 	ASSERT_EVBUFFER_LOCKED(dst);
    800      1.1  christos 	dst->first = NULL;
    801      1.1  christos 	dst->last = NULL;
    802      1.1  christos 	dst->last_with_datap = &(dst)->first;
    803      1.1  christos 	dst->total_len = 0;
    804      1.1  christos }
    805      1.1  christos 
    806      1.1  christos /* Prepares the contents of src to be moved to another buffer by removing
    807      1.1  christos  * read-pinned chains. The first pinned chain is saved in first, and the
    808      1.1  christos  * last in last. If src has no read-pinned chains, first and last are set
    809      1.1  christos  * to NULL. */
    810      1.1  christos static int
    811      1.1  christos PRESERVE_PINNED(struct evbuffer *src, struct evbuffer_chain **first,
    812      1.1  christos 		struct evbuffer_chain **last)
    813      1.1  christos {
    814      1.1  christos 	struct evbuffer_chain *chain, **pinned;
    815      1.1  christos 
    816      1.1  christos 	ASSERT_EVBUFFER_LOCKED(src);
    817      1.1  christos 
    818      1.1  christos 	if (!HAS_PINNED_R(src)) {
    819      1.1  christos 		*first = *last = NULL;
    820      1.1  christos 		return 0;
    821      1.1  christos 	}
    822      1.1  christos 
    823      1.1  christos 	pinned = src->last_with_datap;
    824      1.1  christos 	if (!CHAIN_PINNED_R(*pinned))
    825      1.1  christos 		pinned = &(*pinned)->next;
    826      1.1  christos 	EVUTIL_ASSERT(CHAIN_PINNED_R(*pinned));
    827      1.1  christos 	chain = *first = *pinned;
    828      1.1  christos 	*last = src->last;
    829      1.1  christos 
    830      1.1  christos 	/* If there's data in the first pinned chain, we need to allocate
    831      1.1  christos 	 * a new chain and copy the data over. */
    832      1.1  christos 	if (chain->off) {
    833      1.1  christos 		struct evbuffer_chain *tmp;
    834      1.1  christos 
    835      1.1  christos 		EVUTIL_ASSERT(pinned == src->last_with_datap);
    836      1.1  christos 		tmp = evbuffer_chain_new(chain->off);
    837      1.1  christos 		if (!tmp)
    838      1.1  christos 			return -1;
    839      1.1  christos 		memcpy(tmp->buffer, chain->buffer + chain->misalign,
    840      1.1  christos 			chain->off);
    841      1.1  christos 		tmp->off = chain->off;
    842      1.1  christos 		*src->last_with_datap = tmp;
    843      1.1  christos 		src->last = tmp;
    844      1.1  christos 		chain->misalign += chain->off;
    845      1.1  christos 		chain->off = 0;
    846      1.1  christos 	} else {
    847      1.1  christos 		src->last = *src->last_with_datap;
    848      1.1  christos 		*pinned = NULL;
    849      1.1  christos 	}
    850      1.1  christos 
    851      1.1  christos 	return 0;
    852      1.1  christos }
    853      1.1  christos 
    854      1.1  christos static inline void
    855      1.1  christos RESTORE_PINNED(struct evbuffer *src, struct evbuffer_chain *pinned,
    856      1.1  christos 		struct evbuffer_chain *last)
    857      1.1  christos {
    858      1.1  christos 	ASSERT_EVBUFFER_LOCKED(src);
    859      1.1  christos 
    860      1.1  christos 	if (!pinned) {
    861      1.1  christos 		ZERO_CHAIN(src);
    862      1.1  christos 		return;
    863      1.1  christos 	}
    864      1.1  christos 
    865      1.1  christos 	src->first = pinned;
    866      1.1  christos 	src->last = last;
    867      1.1  christos 	src->last_with_datap = &src->first;
    868      1.1  christos 	src->total_len = 0;
    869      1.1  christos }
    870      1.1  christos 
    871      1.1  christos static inline void
    872      1.1  christos COPY_CHAIN(struct evbuffer *dst, struct evbuffer *src)
    873      1.1  christos {
    874      1.1  christos 	ASSERT_EVBUFFER_LOCKED(dst);
    875      1.1  christos 	ASSERT_EVBUFFER_LOCKED(src);
    876      1.1  christos 	dst->first = src->first;
    877      1.1  christos 	if (src->last_with_datap == &src->first)
    878      1.1  christos 		dst->last_with_datap = &dst->first;
    879      1.1  christos 	else
    880      1.1  christos 		dst->last_with_datap = src->last_with_datap;
    881      1.1  christos 	dst->last = src->last;
    882      1.1  christos 	dst->total_len = src->total_len;
    883      1.1  christos }
    884      1.1  christos 
    885      1.1  christos static void
    886      1.1  christos APPEND_CHAIN(struct evbuffer *dst, struct evbuffer *src)
    887      1.1  christos {
    888      1.1  christos 	ASSERT_EVBUFFER_LOCKED(dst);
    889      1.1  christos 	ASSERT_EVBUFFER_LOCKED(src);
    890      1.1  christos 	dst->last->next = src->first;
    891      1.1  christos 	if (src->last_with_datap == &src->first)
    892      1.1  christos 		dst->last_with_datap = &dst->last->next;
    893      1.1  christos 	else
    894      1.1  christos 		dst->last_with_datap = src->last_with_datap;
    895      1.1  christos 	dst->last = src->last;
    896      1.1  christos 	dst->total_len += src->total_len;
    897      1.1  christos }
    898      1.1  christos 
    899      1.1  christos static inline void
    900      1.1  christos APPEND_CHAIN_MULTICAST(struct evbuffer *dst, struct evbuffer *src)
    901      1.1  christos {
    902      1.1  christos 	struct evbuffer_chain *tmp;
    903      1.1  christos 	struct evbuffer_chain *chain = src->first;
    904      1.1  christos 	struct evbuffer_multicast_parent *extra;
    905      1.1  christos 
    906      1.1  christos 	ASSERT_EVBUFFER_LOCKED(dst);
    907      1.1  christos 	ASSERT_EVBUFFER_LOCKED(src);
    908      1.1  christos 
    909      1.1  christos 	for (; chain; chain = chain->next) {
    910      1.1  christos 		if (!chain->off || chain->flags & EVBUFFER_DANGLING) {
    911      1.1  christos 			/* skip empty chains */
    912      1.1  christos 			continue;
    913      1.1  christos 		}
    914      1.1  christos 
    915      1.1  christos 		tmp = evbuffer_chain_new(sizeof(struct evbuffer_multicast_parent));
    916      1.1  christos 		if (!tmp) {
    917      1.1  christos 			event_warn("%s: out of memory", __func__);
    918      1.1  christos 			return;
    919      1.1  christos 		}
    920      1.1  christos 		extra = EVBUFFER_CHAIN_EXTRA(struct evbuffer_multicast_parent, tmp);
    921      1.1  christos 		/* reference evbuffer containing source chain so it
    922      1.1  christos 		 * doesn't get released while the chain is still
    923      1.1  christos 		 * being referenced to */
    924      1.1  christos 		evbuffer_incref_(src);
    925      1.1  christos 		extra->source = src;
    926      1.1  christos 		/* reference source chain which now becomes immutable */
    927      1.1  christos 		evbuffer_chain_incref(chain);
    928      1.1  christos 		extra->parent = chain;
    929      1.1  christos 		chain->flags |= EVBUFFER_IMMUTABLE;
    930      1.1  christos 		tmp->buffer_len = chain->buffer_len;
    931      1.1  christos 		tmp->misalign = chain->misalign;
    932      1.1  christos 		tmp->off = chain->off;
    933      1.1  christos 		tmp->flags |= EVBUFFER_MULTICAST|EVBUFFER_IMMUTABLE;
    934      1.1  christos 		tmp->buffer = chain->buffer;
    935      1.1  christos 		evbuffer_chain_insert(dst, tmp);
    936      1.1  christos 	}
    937      1.1  christos }
    938      1.1  christos 
    939      1.1  christos static void
    940      1.1  christos PREPEND_CHAIN(struct evbuffer *dst, struct evbuffer *src)
    941      1.1  christos {
    942      1.1  christos 	ASSERT_EVBUFFER_LOCKED(dst);
    943      1.1  christos 	ASSERT_EVBUFFER_LOCKED(src);
    944      1.1  christos 	src->last->next = dst->first;
    945      1.1  christos 	dst->first = src->first;
    946      1.1  christos 	dst->total_len += src->total_len;
    947      1.1  christos 	if (*dst->last_with_datap == NULL) {
    948      1.1  christos 		if (src->last_with_datap == &(src)->first)
    949      1.1  christos 			dst->last_with_datap = &dst->first;
    950      1.1  christos 		else
    951      1.1  christos 			dst->last_with_datap = src->last_with_datap;
    952      1.1  christos 	} else if (dst->last_with_datap == &dst->first) {
    953      1.1  christos 		dst->last_with_datap = &src->last->next;
    954      1.1  christos 	}
    955      1.1  christos }
    956      1.1  christos 
    957      1.1  christos int
    958      1.1  christos evbuffer_add_buffer(struct evbuffer *outbuf, struct evbuffer *inbuf)
    959      1.1  christos {
    960      1.1  christos 	struct evbuffer_chain *pinned, *last;
    961      1.1  christos 	size_t in_total_len, out_total_len;
    962      1.1  christos 	int result = 0;
    963      1.1  christos 
    964      1.1  christos 	EVBUFFER_LOCK2(inbuf, outbuf);
    965      1.1  christos 	in_total_len = inbuf->total_len;
    966      1.1  christos 	out_total_len = outbuf->total_len;
    967      1.1  christos 
    968      1.1  christos 	if (in_total_len == 0 || outbuf == inbuf)
    969      1.1  christos 		goto done;
    970      1.1  christos 
    971      1.1  christos 	if (outbuf->freeze_end || inbuf->freeze_start) {
    972      1.1  christos 		result = -1;
    973      1.1  christos 		goto done;
    974      1.1  christos 	}
    975      1.1  christos 
    976      1.1  christos 	if (PRESERVE_PINNED(inbuf, &pinned, &last) < 0) {
    977      1.1  christos 		result = -1;
    978      1.1  christos 		goto done;
    979      1.1  christos 	}
    980      1.1  christos 
    981      1.1  christos 	if (out_total_len == 0) {
    982      1.1  christos 		/* There might be an empty chain at the start of outbuf; free
    983      1.1  christos 		 * it. */
    984      1.1  christos 		evbuffer_free_all_chains(outbuf->first);
    985      1.1  christos 		COPY_CHAIN(outbuf, inbuf);
    986      1.1  christos 	} else {
    987      1.1  christos 		APPEND_CHAIN(outbuf, inbuf);
    988      1.1  christos 	}
    989      1.1  christos 
    990      1.1  christos 	RESTORE_PINNED(inbuf, pinned, last);
    991      1.1  christos 
    992      1.1  christos 	inbuf->n_del_for_cb += in_total_len;
    993      1.1  christos 	outbuf->n_add_for_cb += in_total_len;
    994      1.1  christos 
    995      1.1  christos 	evbuffer_invoke_callbacks_(inbuf);
    996      1.1  christos 	evbuffer_invoke_callbacks_(outbuf);
    997      1.1  christos 
    998      1.1  christos done:
    999      1.1  christos 	EVBUFFER_UNLOCK2(inbuf, outbuf);
   1000      1.1  christos 	return result;
   1001      1.1  christos }
   1002      1.1  christos 
   1003      1.1  christos int
   1004      1.1  christos evbuffer_add_buffer_reference(struct evbuffer *outbuf, struct evbuffer *inbuf)
   1005      1.1  christos {
   1006      1.1  christos 	size_t in_total_len, out_total_len;
   1007      1.1  christos 	struct evbuffer_chain *chain;
   1008      1.1  christos 	int result = 0;
   1009      1.1  christos 
   1010      1.1  christos 	EVBUFFER_LOCK2(inbuf, outbuf);
   1011      1.1  christos 	in_total_len = inbuf->total_len;
   1012      1.1  christos 	out_total_len = outbuf->total_len;
   1013      1.1  christos 	chain = inbuf->first;
   1014      1.1  christos 
   1015      1.1  christos 	if (in_total_len == 0)
   1016      1.1  christos 		goto done;
   1017      1.1  christos 
   1018      1.1  christos 	if (outbuf->freeze_end || outbuf == inbuf) {
   1019      1.1  christos 		result = -1;
   1020      1.1  christos 		goto done;
   1021      1.1  christos 	}
   1022      1.1  christos 
   1023      1.1  christos 	for (; chain; chain = chain->next) {
   1024      1.1  christos 		if ((chain->flags & (EVBUFFER_FILESEGMENT|EVBUFFER_SENDFILE|EVBUFFER_MULTICAST)) != 0) {
   1025      1.1  christos 			/* chain type can not be referenced */
   1026      1.1  christos 			result = -1;
   1027      1.1  christos 			goto done;
   1028      1.1  christos 		}
   1029      1.1  christos 	}
   1030      1.1  christos 
   1031      1.1  christos 	if (out_total_len == 0) {
   1032      1.1  christos 		/* There might be an empty chain at the start of outbuf; free
   1033      1.1  christos 		 * it. */
   1034      1.1  christos 		evbuffer_free_all_chains(outbuf->first);
   1035      1.1  christos 	}
   1036      1.1  christos 	APPEND_CHAIN_MULTICAST(outbuf, inbuf);
   1037      1.1  christos 
   1038      1.1  christos 	outbuf->n_add_for_cb += in_total_len;
   1039      1.1  christos 	evbuffer_invoke_callbacks_(outbuf);
   1040      1.1  christos 
   1041      1.1  christos done:
   1042      1.1  christos 	EVBUFFER_UNLOCK2(inbuf, outbuf);
   1043      1.1  christos 	return result;
   1044      1.1  christos }
   1045      1.1  christos 
   1046      1.1  christos int
   1047      1.1  christos evbuffer_prepend_buffer(struct evbuffer *outbuf, struct evbuffer *inbuf)
   1048      1.1  christos {
   1049      1.1  christos 	struct evbuffer_chain *pinned, *last;
   1050      1.1  christos 	size_t in_total_len, out_total_len;
   1051      1.1  christos 	int result = 0;
   1052      1.1  christos 
   1053      1.1  christos 	EVBUFFER_LOCK2(inbuf, outbuf);
   1054      1.1  christos 
   1055      1.1  christos 	in_total_len = inbuf->total_len;
   1056      1.1  christos 	out_total_len = outbuf->total_len;
   1057      1.1  christos 
   1058      1.1  christos 	if (!in_total_len || inbuf == outbuf)
   1059      1.1  christos 		goto done;
   1060      1.1  christos 
   1061      1.1  christos 	if (outbuf->freeze_start || inbuf->freeze_start) {
   1062      1.1  christos 		result = -1;
   1063      1.1  christos 		goto done;
   1064      1.1  christos 	}
   1065      1.1  christos 
   1066      1.1  christos 	if (PRESERVE_PINNED(inbuf, &pinned, &last) < 0) {
   1067      1.1  christos 		result = -1;
   1068      1.1  christos 		goto done;
   1069      1.1  christos 	}
   1070      1.1  christos 
   1071      1.1  christos 	if (out_total_len == 0) {
   1072      1.1  christos 		/* There might be an empty chain at the start of outbuf; free
   1073      1.1  christos 		 * it. */
   1074      1.1  christos 		evbuffer_free_all_chains(outbuf->first);
   1075      1.1  christos 		COPY_CHAIN(outbuf, inbuf);
   1076      1.1  christos 	} else {
   1077      1.1  christos 		PREPEND_CHAIN(outbuf, inbuf);
   1078      1.1  christos 	}
   1079      1.1  christos 
   1080      1.1  christos 	RESTORE_PINNED(inbuf, pinned, last);
   1081      1.1  christos 
   1082      1.1  christos 	inbuf->n_del_for_cb += in_total_len;
   1083      1.1  christos 	outbuf->n_add_for_cb += in_total_len;
   1084      1.1  christos 
   1085      1.1  christos 	evbuffer_invoke_callbacks_(inbuf);
   1086      1.1  christos 	evbuffer_invoke_callbacks_(outbuf);
   1087      1.1  christos done:
   1088      1.1  christos 	EVBUFFER_UNLOCK2(inbuf, outbuf);
   1089      1.1  christos 	return result;
   1090      1.1  christos }
   1091      1.1  christos 
   1092      1.1  christos int
   1093      1.1  christos evbuffer_drain(struct evbuffer *buf, size_t len)
   1094      1.1  christos {
   1095      1.1  christos 	struct evbuffer_chain *chain, *next;
   1096      1.1  christos 	size_t remaining, old_len;
   1097      1.1  christos 	int result = 0;
   1098      1.1  christos 
   1099      1.1  christos 	EVBUFFER_LOCK(buf);
   1100      1.1  christos 	old_len = buf->total_len;
   1101      1.1  christos 
   1102      1.1  christos 	if (old_len == 0)
   1103      1.1  christos 		goto done;
   1104      1.1  christos 
   1105      1.1  christos 	if (buf->freeze_start) {
   1106      1.1  christos 		result = -1;
   1107      1.1  christos 		goto done;
   1108      1.1  christos 	}
   1109      1.1  christos 
   1110      1.1  christos 	if (len >= old_len && !HAS_PINNED_R(buf)) {
   1111      1.1  christos 		len = old_len;
   1112      1.1  christos 		for (chain = buf->first; chain != NULL; chain = next) {
   1113      1.1  christos 			next = chain->next;
   1114      1.1  christos 			evbuffer_chain_free(chain);
   1115      1.1  christos 		}
   1116      1.1  christos 
   1117      1.1  christos 		ZERO_CHAIN(buf);
   1118      1.1  christos 	} else {
   1119      1.1  christos 		if (len >= old_len)
   1120      1.1  christos 			len = old_len;
   1121      1.1  christos 
   1122      1.1  christos 		buf->total_len -= len;
   1123      1.1  christos 		remaining = len;
   1124      1.1  christos 		for (chain = buf->first;
   1125      1.1  christos 		     remaining >= chain->off;
   1126      1.1  christos 		     chain = next) {
   1127      1.1  christos 			next = chain->next;
   1128      1.1  christos 			remaining -= chain->off;
   1129      1.1  christos 
   1130      1.1  christos 			if (chain == *buf->last_with_datap) {
   1131      1.1  christos 				buf->last_with_datap = &buf->first;
   1132      1.1  christos 			}
   1133      1.1  christos 			if (&chain->next == buf->last_with_datap)
   1134      1.1  christos 				buf->last_with_datap = &buf->first;
   1135      1.1  christos 
   1136      1.1  christos 			if (CHAIN_PINNED_R(chain)) {
   1137      1.1  christos 				EVUTIL_ASSERT(remaining == 0);
   1138      1.1  christos 				chain->misalign += chain->off;
   1139      1.1  christos 				chain->off = 0;
   1140      1.1  christos 				break;
   1141      1.1  christos 			} else
   1142      1.1  christos 				evbuffer_chain_free(chain);
   1143      1.1  christos 		}
   1144      1.1  christos 
   1145      1.1  christos 		buf->first = chain;
   1146  1.1.1.3  christos 		EVUTIL_ASSERT(chain && remaining <= chain->off);
   1147      1.1  christos 		chain->misalign += remaining;
   1148      1.1  christos 		chain->off -= remaining;
   1149      1.1  christos 	}
   1150      1.1  christos 
   1151      1.1  christos 	buf->n_del_for_cb += len;
   1152      1.1  christos 	/* Tell someone about changes in this buffer */
   1153      1.1  christos 	evbuffer_invoke_callbacks_(buf);
   1154      1.1  christos 
   1155      1.1  christos done:
   1156      1.1  christos 	EVBUFFER_UNLOCK(buf);
   1157      1.1  christos 	return result;
   1158      1.1  christos }
   1159      1.1  christos 
   1160      1.1  christos /* Reads data from an event buffer and drains the bytes read */
   1161      1.1  christos int
   1162      1.1  christos evbuffer_remove(struct evbuffer *buf, void *data_out, size_t datlen)
   1163      1.1  christos {
   1164      1.1  christos 	ev_ssize_t n;
   1165      1.1  christos 	EVBUFFER_LOCK(buf);
   1166      1.1  christos 	n = evbuffer_copyout_from(buf, NULL, data_out, datlen);
   1167      1.1  christos 	if (n > 0) {
   1168      1.1  christos 		if (evbuffer_drain(buf, n)<0)
   1169      1.1  christos 			n = -1;
   1170      1.1  christos 	}
   1171      1.1  christos 	EVBUFFER_UNLOCK(buf);
   1172      1.1  christos 	return (int)n;
   1173      1.1  christos }
   1174      1.1  christos 
   1175      1.1  christos ev_ssize_t
   1176      1.1  christos evbuffer_copyout(struct evbuffer *buf, void *data_out, size_t datlen)
   1177      1.1  christos {
   1178      1.1  christos 	return evbuffer_copyout_from(buf, NULL, data_out, datlen);
   1179      1.1  christos }
   1180      1.1  christos 
   1181      1.1  christos ev_ssize_t
   1182      1.1  christos evbuffer_copyout_from(struct evbuffer *buf, const struct evbuffer_ptr *pos,
   1183      1.1  christos     void *data_out, size_t datlen)
   1184      1.1  christos {
   1185      1.1  christos 	/*XXX fails badly on sendfile case. */
   1186      1.1  christos 	struct evbuffer_chain *chain;
   1187      1.1  christos 	char *data = data_out;
   1188      1.1  christos 	size_t nread;
   1189      1.1  christos 	ev_ssize_t result = 0;
   1190      1.1  christos 	size_t pos_in_chain;
   1191      1.1  christos 
   1192      1.1  christos 	EVBUFFER_LOCK(buf);
   1193      1.1  christos 
   1194      1.1  christos 	if (pos) {
   1195  1.1.1.3  christos 		if (datlen > (size_t)(EV_SSIZE_MAX - pos->pos)) {
   1196  1.1.1.3  christos 			result = -1;
   1197  1.1.1.3  christos 			goto done;
   1198  1.1.1.3  christos 		}
   1199      1.1  christos 		chain = pos->internal_.chain;
   1200      1.1  christos 		pos_in_chain = pos->internal_.pos_in_chain;
   1201      1.1  christos 		if (datlen + pos->pos > buf->total_len)
   1202      1.1  christos 			datlen = buf->total_len - pos->pos;
   1203      1.1  christos 	} else {
   1204      1.1  christos 		chain = buf->first;
   1205      1.1  christos 		pos_in_chain = 0;
   1206      1.1  christos 		if (datlen > buf->total_len)
   1207      1.1  christos 			datlen = buf->total_len;
   1208      1.1  christos 	}
   1209      1.1  christos 
   1210      1.1  christos 
   1211      1.1  christos 	if (datlen == 0)
   1212      1.1  christos 		goto done;
   1213      1.1  christos 
   1214      1.1  christos 	if (buf->freeze_start) {
   1215      1.1  christos 		result = -1;
   1216      1.1  christos 		goto done;
   1217      1.1  christos 	}
   1218      1.1  christos 
   1219      1.1  christos 	nread = datlen;
   1220      1.1  christos 
   1221      1.1  christos 	while (datlen && datlen >= chain->off - pos_in_chain) {
   1222      1.1  christos 		size_t copylen = chain->off - pos_in_chain;
   1223      1.1  christos 		memcpy(data,
   1224      1.1  christos 		    chain->buffer + chain->misalign + pos_in_chain,
   1225      1.1  christos 		    copylen);
   1226      1.1  christos 		data += copylen;
   1227      1.1  christos 		datlen -= copylen;
   1228      1.1  christos 
   1229      1.1  christos 		chain = chain->next;
   1230      1.1  christos 		pos_in_chain = 0;
   1231      1.1  christos 		EVUTIL_ASSERT(chain || datlen==0);
   1232      1.1  christos 	}
   1233      1.1  christos 
   1234      1.1  christos 	if (datlen) {
   1235      1.1  christos 		EVUTIL_ASSERT(chain);
   1236  1.1.1.3  christos 		EVUTIL_ASSERT(datlen+pos_in_chain <= chain->off);
   1237  1.1.1.3  christos 
   1238      1.1  christos 		memcpy(data, chain->buffer + chain->misalign + pos_in_chain,
   1239      1.1  christos 		    datlen);
   1240      1.1  christos 	}
   1241      1.1  christos 
   1242      1.1  christos 	result = nread;
   1243      1.1  christos done:
   1244      1.1  christos 	EVBUFFER_UNLOCK(buf);
   1245      1.1  christos 	return result;
   1246      1.1  christos }
   1247      1.1  christos 
   1248      1.1  christos /* reads data from the src buffer to the dst buffer, avoids memcpy as
   1249      1.1  christos  * possible. */
   1250      1.1  christos /*  XXXX should return ev_ssize_t */
   1251      1.1  christos int
   1252      1.1  christos evbuffer_remove_buffer(struct evbuffer *src, struct evbuffer *dst,
   1253      1.1  christos     size_t datlen)
   1254      1.1  christos {
   1255      1.1  christos 	/*XXX We should have an option to force this to be zero-copy.*/
   1256      1.1  christos 
   1257      1.1  christos 	/*XXX can fail badly on sendfile case. */
   1258      1.1  christos 	struct evbuffer_chain *chain, *previous;
   1259      1.1  christos 	size_t nread = 0;
   1260      1.1  christos 	int result;
   1261      1.1  christos 
   1262      1.1  christos 	EVBUFFER_LOCK2(src, dst);
   1263      1.1  christos 
   1264      1.1  christos 	chain = previous = src->first;
   1265      1.1  christos 
   1266      1.1  christos 	if (datlen == 0 || dst == src) {
   1267      1.1  christos 		result = 0;
   1268      1.1  christos 		goto done;
   1269      1.1  christos 	}
   1270      1.1  christos 
   1271      1.1  christos 	if (dst->freeze_end || src->freeze_start) {
   1272      1.1  christos 		result = -1;
   1273      1.1  christos 		goto done;
   1274      1.1  christos 	}
   1275      1.1  christos 
   1276      1.1  christos 	/* short-cut if there is no more data buffered */
   1277      1.1  christos 	if (datlen >= src->total_len) {
   1278      1.1  christos 		datlen = src->total_len;
   1279      1.1  christos 		evbuffer_add_buffer(dst, src);
   1280      1.1  christos 		result = (int)datlen; /*XXXX should return ev_ssize_t*/
   1281      1.1  christos 		goto done;
   1282      1.1  christos 	}
   1283      1.1  christos 
   1284      1.1  christos 	/* removes chains if possible */
   1285      1.1  christos 	while (chain->off <= datlen) {
   1286      1.1  christos 		/* We can't remove the last with data from src unless we
   1287      1.1  christos 		 * remove all chains, in which case we would have done the if
   1288      1.1  christos 		 * block above */
   1289      1.1  christos 		EVUTIL_ASSERT(chain != *src->last_with_datap);
   1290      1.1  christos 		nread += chain->off;
   1291      1.1  christos 		datlen -= chain->off;
   1292      1.1  christos 		previous = chain;
   1293      1.1  christos 		if (src->last_with_datap == &chain->next)
   1294      1.1  christos 			src->last_with_datap = &src->first;
   1295      1.1  christos 		chain = chain->next;
   1296      1.1  christos 	}
   1297      1.1  christos 
   1298      1.1  christos 	if (nread) {
   1299      1.1  christos 		/* we can remove the chain */
   1300      1.1  christos 		struct evbuffer_chain **chp;
   1301      1.1  christos 		chp = evbuffer_free_trailing_empty_chains(dst);
   1302      1.1  christos 
   1303      1.1  christos 		if (dst->first == NULL) {
   1304      1.1  christos 			dst->first = src->first;
   1305      1.1  christos 		} else {
   1306      1.1  christos 			*chp = src->first;
   1307      1.1  christos 		}
   1308      1.1  christos 		dst->last = previous;
   1309      1.1  christos 		previous->next = NULL;
   1310      1.1  christos 		src->first = chain;
   1311      1.1  christos 		advance_last_with_data(dst);
   1312      1.1  christos 
   1313      1.1  christos 		dst->total_len += nread;
   1314      1.1  christos 		dst->n_add_for_cb += nread;
   1315      1.1  christos 	}
   1316      1.1  christos 
   1317      1.1  christos 	/* we know that there is more data in the src buffer than
   1318      1.1  christos 	 * we want to read, so we manually drain the chain */
   1319      1.1  christos 	evbuffer_add(dst, chain->buffer + chain->misalign, datlen);
   1320      1.1  christos 	chain->misalign += datlen;
   1321      1.1  christos 	chain->off -= datlen;
   1322      1.1  christos 	nread += datlen;
   1323      1.1  christos 
   1324      1.1  christos 	/* You might think we would want to increment dst->n_add_for_cb
   1325      1.1  christos 	 * here too.  But evbuffer_add above already took care of that.
   1326      1.1  christos 	 */
   1327      1.1  christos 	src->total_len -= nread;
   1328      1.1  christos 	src->n_del_for_cb += nread;
   1329      1.1  christos 
   1330      1.1  christos 	if (nread) {
   1331      1.1  christos 		evbuffer_invoke_callbacks_(dst);
   1332      1.1  christos 		evbuffer_invoke_callbacks_(src);
   1333      1.1  christos 	}
   1334      1.1  christos 	result = (int)nread;/*XXXX should change return type */
   1335      1.1  christos 
   1336      1.1  christos done:
   1337      1.1  christos 	EVBUFFER_UNLOCK2(src, dst);
   1338      1.1  christos 	return result;
   1339      1.1  christos }
   1340      1.1  christos 
   1341      1.1  christos unsigned char *
   1342      1.1  christos evbuffer_pullup(struct evbuffer *buf, ev_ssize_t size)
   1343      1.1  christos {
   1344      1.1  christos 	struct evbuffer_chain *chain, *next, *tmp, *last_with_data;
   1345      1.1  christos 	unsigned char *buffer, *result = NULL;
   1346      1.1  christos 	ev_ssize_t remaining;
   1347      1.1  christos 	int removed_last_with_data = 0;
   1348      1.1  christos 	int removed_last_with_datap = 0;
   1349      1.1  christos 
   1350      1.1  christos 	EVBUFFER_LOCK(buf);
   1351      1.1  christos 
   1352      1.1  christos 	chain = buf->first;
   1353      1.1  christos 
   1354      1.1  christos 	if (size < 0)
   1355      1.1  christos 		size = buf->total_len;
   1356      1.1  christos 	/* if size > buf->total_len, we cannot guarantee to the user that she
   1357      1.1  christos 	 * is going to have a long enough buffer afterwards; so we return
   1358      1.1  christos 	 * NULL */
   1359      1.1  christos 	if (size == 0 || (size_t)size > buf->total_len)
   1360      1.1  christos 		goto done;
   1361      1.1  christos 
   1362      1.1  christos 	/* No need to pull up anything; the first size bytes are
   1363      1.1  christos 	 * already here. */
   1364      1.1  christos 	if (chain->off >= (size_t)size) {
   1365      1.1  christos 		result = chain->buffer + chain->misalign;
   1366      1.1  christos 		goto done;
   1367      1.1  christos 	}
   1368      1.1  christos 
   1369      1.1  christos 	/* Make sure that none of the chains we need to copy from is pinned. */
   1370      1.1  christos 	remaining = size - chain->off;
   1371      1.1  christos 	EVUTIL_ASSERT(remaining >= 0);
   1372      1.1  christos 	for (tmp=chain->next; tmp; tmp=tmp->next) {
   1373      1.1  christos 		if (CHAIN_PINNED(tmp))
   1374      1.1  christos 			goto done;
   1375      1.1  christos 		if (tmp->off >= (size_t)remaining)
   1376      1.1  christos 			break;
   1377      1.1  christos 		remaining -= tmp->off;
   1378      1.1  christos 	}
   1379      1.1  christos 
   1380      1.1  christos 	if (CHAIN_PINNED(chain)) {
   1381      1.1  christos 		size_t old_off = chain->off;
   1382      1.1  christos 		if (CHAIN_SPACE_LEN(chain) < size - chain->off) {
   1383      1.1  christos 			/* not enough room at end of chunk. */
   1384      1.1  christos 			goto done;
   1385      1.1  christos 		}
   1386      1.1  christos 		buffer = CHAIN_SPACE_PTR(chain);
   1387      1.1  christos 		tmp = chain;
   1388      1.1  christos 		tmp->off = size;
   1389      1.1  christos 		size -= old_off;
   1390      1.1  christos 		chain = chain->next;
   1391      1.1  christos 	} else if (chain->buffer_len - chain->misalign >= (size_t)size) {
   1392      1.1  christos 		/* already have enough space in the first chain */
   1393      1.1  christos 		size_t old_off = chain->off;
   1394      1.1  christos 		buffer = chain->buffer + chain->misalign + chain->off;
   1395      1.1  christos 		tmp = chain;
   1396      1.1  christos 		tmp->off = size;
   1397      1.1  christos 		size -= old_off;
   1398      1.1  christos 		chain = chain->next;
   1399      1.1  christos 	} else {
   1400      1.1  christos 		if ((tmp = evbuffer_chain_new(size)) == NULL) {
   1401      1.1  christos 			event_warn("%s: out of memory", __func__);
   1402      1.1  christos 			goto done;
   1403      1.1  christos 		}
   1404      1.1  christos 		buffer = tmp->buffer;
   1405      1.1  christos 		tmp->off = size;
   1406      1.1  christos 		buf->first = tmp;
   1407      1.1  christos 	}
   1408      1.1  christos 
   1409      1.1  christos 	/* TODO(niels): deal with buffers that point to NULL like sendfile */
   1410      1.1  christos 
   1411      1.1  christos 	/* Copy and free every chunk that will be entirely pulled into tmp */
   1412      1.1  christos 	last_with_data = *buf->last_with_datap;
   1413      1.1  christos 	for (; chain != NULL && (size_t)size >= chain->off; chain = next) {
   1414      1.1  christos 		next = chain->next;
   1415      1.1  christos 
   1416      1.1  christos 		memcpy(buffer, chain->buffer + chain->misalign, chain->off);
   1417      1.1  christos 		size -= chain->off;
   1418      1.1  christos 		buffer += chain->off;
   1419      1.1  christos 		if (chain == last_with_data)
   1420      1.1  christos 			removed_last_with_data = 1;
   1421      1.1  christos 		if (&chain->next == buf->last_with_datap)
   1422      1.1  christos 			removed_last_with_datap = 1;
   1423      1.1  christos 
   1424      1.1  christos 		evbuffer_chain_free(chain);
   1425      1.1  christos 	}
   1426      1.1  christos 
   1427      1.1  christos 	if (chain != NULL) {
   1428      1.1  christos 		memcpy(buffer, chain->buffer + chain->misalign, size);
   1429      1.1  christos 		chain->misalign += size;
   1430      1.1  christos 		chain->off -= size;
   1431      1.1  christos 	} else {
   1432      1.1  christos 		buf->last = tmp;
   1433      1.1  christos 	}
   1434      1.1  christos 
   1435      1.1  christos 	tmp->next = chain;
   1436      1.1  christos 
   1437      1.1  christos 	if (removed_last_with_data) {
   1438      1.1  christos 		buf->last_with_datap = &buf->first;
   1439      1.1  christos 	} else if (removed_last_with_datap) {
   1440      1.1  christos 		if (buf->first->next && buf->first->next->off)
   1441      1.1  christos 			buf->last_with_datap = &buf->first->next;
   1442      1.1  christos 		else
   1443      1.1  christos 			buf->last_with_datap = &buf->first;
   1444      1.1  christos 	}
   1445      1.1  christos 
   1446      1.1  christos 	result = (tmp->buffer + tmp->misalign);
   1447      1.1  christos 
   1448      1.1  christos done:
   1449      1.1  christos 	EVBUFFER_UNLOCK(buf);
   1450      1.1  christos 	return result;
   1451      1.1  christos }
   1452      1.1  christos 
   1453      1.1  christos /*
   1454      1.1  christos  * Reads a line terminated by either '\r\n', '\n\r' or '\r' or '\n'.
   1455      1.1  christos  * The returned buffer needs to be freed by the called.
   1456      1.1  christos  */
   1457      1.1  christos char *
   1458      1.1  christos evbuffer_readline(struct evbuffer *buffer)
   1459      1.1  christos {
   1460      1.1  christos 	return evbuffer_readln(buffer, NULL, EVBUFFER_EOL_ANY);
   1461      1.1  christos }
   1462      1.1  christos 
   1463      1.1  christos static inline ev_ssize_t
   1464      1.1  christos evbuffer_strchr(struct evbuffer_ptr *it, const char chr)
   1465      1.1  christos {
   1466      1.1  christos 	struct evbuffer_chain *chain = it->internal_.chain;
   1467      1.1  christos 	size_t i = it->internal_.pos_in_chain;
   1468      1.1  christos 	while (chain != NULL) {
   1469      1.1  christos 		char *buffer = (char *)chain->buffer + chain->misalign;
   1470      1.1  christos 		char *cp = memchr(buffer+i, chr, chain->off-i);
   1471      1.1  christos 		if (cp) {
   1472      1.1  christos 			it->internal_.chain = chain;
   1473      1.1  christos 			it->internal_.pos_in_chain = cp - buffer;
   1474      1.1  christos 			it->pos += (cp - buffer - i);
   1475      1.1  christos 			return it->pos;
   1476      1.1  christos 		}
   1477      1.1  christos 		it->pos += chain->off - i;
   1478      1.1  christos 		i = 0;
   1479      1.1  christos 		chain = chain->next;
   1480      1.1  christos 	}
   1481      1.1  christos 
   1482      1.1  christos 	return (-1);
   1483      1.1  christos }
   1484      1.1  christos 
   1485      1.1  christos static inline char *
   1486      1.1  christos find_eol_char(char *s, size_t len)
   1487      1.1  christos {
   1488      1.1  christos #define CHUNK_SZ 128
   1489      1.1  christos 	/* Lots of benchmarking found this approach to be faster in practice
   1490      1.1  christos 	 * than doing two memchrs over the whole buffer, doin a memchr on each
   1491      1.1  christos 	 * char of the buffer, or trying to emulate memchr by hand. */
   1492      1.1  christos 	char *s_end, *cr, *lf;
   1493      1.1  christos 	s_end = s+len;
   1494      1.1  christos 	while (s < s_end) {
   1495      1.1  christos 		size_t chunk = (s + CHUNK_SZ < s_end) ? CHUNK_SZ : (s_end - s);
   1496      1.1  christos 		cr = memchr(s, '\r', chunk);
   1497      1.1  christos 		lf = memchr(s, '\n', chunk);
   1498      1.1  christos 		if (cr) {
   1499      1.1  christos 			if (lf && lf < cr)
   1500      1.1  christos 				return lf;
   1501      1.1  christos 			return cr;
   1502      1.1  christos 		} else if (lf) {
   1503      1.1  christos 			return lf;
   1504      1.1  christos 		}
   1505      1.1  christos 		s += CHUNK_SZ;
   1506      1.1  christos 	}
   1507      1.1  christos 
   1508      1.1  christos 	return NULL;
   1509      1.1  christos #undef CHUNK_SZ
   1510      1.1  christos }
   1511      1.1  christos 
   1512      1.1  christos static ev_ssize_t
   1513      1.1  christos evbuffer_find_eol_char(struct evbuffer_ptr *it)
   1514      1.1  christos {
   1515      1.1  christos 	struct evbuffer_chain *chain = it->internal_.chain;
   1516      1.1  christos 	size_t i = it->internal_.pos_in_chain;
   1517      1.1  christos 	while (chain != NULL) {
   1518      1.1  christos 		char *buffer = (char *)chain->buffer + chain->misalign;
   1519      1.1  christos 		char *cp = find_eol_char(buffer+i, chain->off-i);
   1520      1.1  christos 		if (cp) {
   1521      1.1  christos 			it->internal_.chain = chain;
   1522      1.1  christos 			it->internal_.pos_in_chain = cp - buffer;
   1523      1.1  christos 			it->pos += (cp - buffer) - i;
   1524      1.1  christos 			return it->pos;
   1525      1.1  christos 		}
   1526      1.1  christos 		it->pos += chain->off - i;
   1527      1.1  christos 		i = 0;
   1528      1.1  christos 		chain = chain->next;
   1529      1.1  christos 	}
   1530      1.1  christos 
   1531      1.1  christos 	return (-1);
   1532      1.1  christos }
   1533      1.1  christos 
   1534      1.1  christos static inline int
   1535      1.1  christos evbuffer_strspn(
   1536      1.1  christos 	struct evbuffer_ptr *ptr, const char *chrset)
   1537      1.1  christos {
   1538      1.1  christos 	int count = 0;
   1539      1.1  christos 	struct evbuffer_chain *chain = ptr->internal_.chain;
   1540      1.1  christos 	size_t i = ptr->internal_.pos_in_chain;
   1541      1.1  christos 
   1542      1.1  christos 	if (!chain)
   1543      1.1  christos 		return 0;
   1544      1.1  christos 
   1545      1.1  christos 	while (1) {
   1546      1.1  christos 		char *buffer = (char *)chain->buffer + chain->misalign;
   1547      1.1  christos 		for (; i < chain->off; ++i) {
   1548      1.1  christos 			const char *p = chrset;
   1549      1.1  christos 			while (*p) {
   1550      1.1  christos 				if (buffer[i] == *p++)
   1551      1.1  christos 					goto next;
   1552      1.1  christos 			}
   1553      1.1  christos 			ptr->internal_.chain = chain;
   1554      1.1  christos 			ptr->internal_.pos_in_chain = i;
   1555      1.1  christos 			ptr->pos += count;
   1556      1.1  christos 			return count;
   1557      1.1  christos 		next:
   1558      1.1  christos 			++count;
   1559      1.1  christos 		}
   1560      1.1  christos 		i = 0;
   1561      1.1  christos 
   1562      1.1  christos 		if (! chain->next) {
   1563      1.1  christos 			ptr->internal_.chain = chain;
   1564      1.1  christos 			ptr->internal_.pos_in_chain = i;
   1565      1.1  christos 			ptr->pos += count;
   1566      1.1  christos 			return count;
   1567      1.1  christos 		}
   1568      1.1  christos 
   1569      1.1  christos 		chain = chain->next;
   1570      1.1  christos 	}
   1571      1.1  christos }
   1572      1.1  christos 
   1573      1.1  christos 
   1574      1.1  christos static inline int
   1575      1.1  christos evbuffer_getchr(struct evbuffer_ptr *it)
   1576      1.1  christos {
   1577      1.1  christos 	struct evbuffer_chain *chain = it->internal_.chain;
   1578      1.1  christos 	size_t off = it->internal_.pos_in_chain;
   1579      1.1  christos 
   1580      1.1  christos 	if (chain == NULL)
   1581      1.1  christos 		return -1;
   1582      1.1  christos 
   1583      1.1  christos 	return (unsigned char)chain->buffer[chain->misalign + off];
   1584      1.1  christos }
   1585      1.1  christos 
   1586      1.1  christos struct evbuffer_ptr
   1587      1.1  christos evbuffer_search_eol(struct evbuffer *buffer,
   1588      1.1  christos     struct evbuffer_ptr *start, size_t *eol_len_out,
   1589      1.1  christos     enum evbuffer_eol_style eol_style)
   1590      1.1  christos {
   1591      1.1  christos 	struct evbuffer_ptr it, it2;
   1592      1.1  christos 	size_t extra_drain = 0;
   1593      1.1  christos 	int ok = 0;
   1594      1.1  christos 
   1595      1.1  christos 	/* Avoid locking in trivial edge cases */
   1596      1.1  christos 	if (start && start->internal_.chain == NULL) {
   1597      1.1  christos 		PTR_NOT_FOUND(&it);
   1598      1.1  christos 		if (eol_len_out)
   1599      1.1  christos 			*eol_len_out = extra_drain;
   1600      1.1  christos 		return it;
   1601      1.1  christos 	}
   1602      1.1  christos 
   1603      1.1  christos 	EVBUFFER_LOCK(buffer);
   1604      1.1  christos 
   1605      1.1  christos 	if (start) {
   1606      1.1  christos 		memcpy(&it, start, sizeof(it));
   1607      1.1  christos 	} else {
   1608      1.1  christos 		it.pos = 0;
   1609      1.1  christos 		it.internal_.chain = buffer->first;
   1610      1.1  christos 		it.internal_.pos_in_chain = 0;
   1611      1.1  christos 	}
   1612      1.1  christos 
   1613      1.1  christos 	/* the eol_style determines our first stop character and how many
   1614      1.1  christos 	 * characters we are going to drain afterwards. */
   1615      1.1  christos 	switch (eol_style) {
   1616      1.1  christos 	case EVBUFFER_EOL_ANY:
   1617      1.1  christos 		if (evbuffer_find_eol_char(&it) < 0)
   1618      1.1  christos 			goto done;
   1619      1.1  christos 		memcpy(&it2, &it, sizeof(it));
   1620      1.1  christos 		extra_drain = evbuffer_strspn(&it2, "\r\n");
   1621      1.1  christos 		break;
   1622      1.1  christos 	case EVBUFFER_EOL_CRLF_STRICT: {
   1623      1.1  christos 		it = evbuffer_search(buffer, "\r\n", 2, &it);
   1624      1.1  christos 		if (it.pos < 0)
   1625      1.1  christos 			goto done;
   1626      1.1  christos 		extra_drain = 2;
   1627      1.1  christos 		break;
   1628      1.1  christos 	}
   1629      1.1  christos 	case EVBUFFER_EOL_CRLF: {
   1630      1.1  christos 		ev_ssize_t start_pos = it.pos;
   1631      1.1  christos 		/* Look for a LF ... */
   1632      1.1  christos 		if (evbuffer_strchr(&it, '\n') < 0)
   1633      1.1  christos 			goto done;
   1634      1.1  christos 		extra_drain = 1;
   1635      1.1  christos 		/* ... optionally preceeded by a CR. */
   1636      1.1  christos 		if (it.pos == start_pos)
   1637      1.1  christos 			break; /* If the first character is \n, don't back up */
   1638      1.1  christos 		/* This potentially does an extra linear walk over the first
   1639      1.1  christos 		 * few chains.  Probably, that's not too expensive unless you
   1640      1.1  christos 		 * have a really pathological setup. */
   1641      1.1  christos 		memcpy(&it2, &it, sizeof(it));
   1642      1.1  christos 		if (evbuffer_ptr_subtract(buffer, &it2, 1)<0)
   1643      1.1  christos 			break;
   1644      1.1  christos 		if (evbuffer_getchr(&it2) == '\r') {
   1645      1.1  christos 			memcpy(&it, &it2, sizeof(it));
   1646      1.1  christos 			extra_drain = 2;
   1647      1.1  christos 		}
   1648      1.1  christos 		break;
   1649      1.1  christos 	}
   1650      1.1  christos 	case EVBUFFER_EOL_LF:
   1651      1.1  christos 		if (evbuffer_strchr(&it, '\n') < 0)
   1652      1.1  christos 			goto done;
   1653      1.1  christos 		extra_drain = 1;
   1654      1.1  christos 		break;
   1655      1.1  christos 	case EVBUFFER_EOL_NUL:
   1656      1.1  christos 		if (evbuffer_strchr(&it, '\0') < 0)
   1657      1.1  christos 			goto done;
   1658      1.1  christos 		extra_drain = 1;
   1659      1.1  christos 		break;
   1660      1.1  christos 	default:
   1661      1.1  christos 		goto done;
   1662      1.1  christos 	}
   1663      1.1  christos 
   1664      1.1  christos 	ok = 1;
   1665      1.1  christos done:
   1666      1.1  christos 	EVBUFFER_UNLOCK(buffer);
   1667      1.1  christos 
   1668      1.1  christos 	if (!ok)
   1669      1.1  christos 		PTR_NOT_FOUND(&it);
   1670      1.1  christos 	if (eol_len_out)
   1671      1.1  christos 		*eol_len_out = extra_drain;
   1672      1.1  christos 
   1673      1.1  christos 	return it;
   1674      1.1  christos }
   1675      1.1  christos 
   1676      1.1  christos char *
   1677      1.1  christos evbuffer_readln(struct evbuffer *buffer, size_t *n_read_out,
   1678      1.1  christos 		enum evbuffer_eol_style eol_style)
   1679      1.1  christos {
   1680      1.1  christos 	struct evbuffer_ptr it;
   1681      1.1  christos 	char *line;
   1682      1.1  christos 	size_t n_to_copy=0, extra_drain=0;
   1683      1.1  christos 	char *result = NULL;
   1684      1.1  christos 
   1685      1.1  christos 	EVBUFFER_LOCK(buffer);
   1686      1.1  christos 
   1687      1.1  christos 	if (buffer->freeze_start) {
   1688      1.1  christos 		goto done;
   1689      1.1  christos 	}
   1690      1.1  christos 
   1691      1.1  christos 	it = evbuffer_search_eol(buffer, NULL, &extra_drain, eol_style);
   1692      1.1  christos 	if (it.pos < 0)
   1693      1.1  christos 		goto done;
   1694      1.1  christos 	n_to_copy = it.pos;
   1695      1.1  christos 
   1696      1.1  christos 	if ((line = mm_malloc(n_to_copy+1)) == NULL) {
   1697      1.1  christos 		event_warn("%s: out of memory", __func__);
   1698      1.1  christos 		goto done;
   1699      1.1  christos 	}
   1700      1.1  christos 
   1701      1.1  christos 	evbuffer_remove(buffer, line, n_to_copy);
   1702      1.1  christos 	line[n_to_copy] = '\0';
   1703      1.1  christos 
   1704      1.1  christos 	evbuffer_drain(buffer, extra_drain);
   1705      1.1  christos 	result = line;
   1706      1.1  christos done:
   1707      1.1  christos 	EVBUFFER_UNLOCK(buffer);
   1708      1.1  christos 
   1709      1.1  christos 	if (n_read_out)
   1710      1.1  christos 		*n_read_out = result ? n_to_copy : 0;
   1711      1.1  christos 
   1712      1.1  christos 	return result;
   1713      1.1  christos }
   1714      1.1  christos 
   1715      1.1  christos #define EVBUFFER_CHAIN_MAX_AUTO_SIZE 4096
   1716      1.1  christos 
   1717      1.1  christos /* Adds data to an event buffer */
   1718      1.1  christos 
   1719      1.1  christos int
   1720      1.1  christos evbuffer_add(struct evbuffer *buf, const void *data_in, size_t datlen)
   1721      1.1  christos {
   1722      1.1  christos 	struct evbuffer_chain *chain, *tmp;
   1723      1.1  christos 	const unsigned char *data = data_in;
   1724      1.1  christos 	size_t remain, to_alloc;
   1725      1.1  christos 	int result = -1;
   1726      1.1  christos 
   1727      1.1  christos 	EVBUFFER_LOCK(buf);
   1728      1.1  christos 
   1729      1.1  christos 	if (buf->freeze_end) {
   1730      1.1  christos 		goto done;
   1731      1.1  christos 	}
   1732  1.1.1.3  christos 	/* Prevent buf->total_len overflow */
   1733  1.1.1.3  christos 	if (datlen > EV_SIZE_MAX - buf->total_len) {
   1734  1.1.1.3  christos 		goto done;
   1735  1.1.1.3  christos 	}
   1736      1.1  christos 
   1737      1.1  christos 	chain = buf->last;
   1738      1.1  christos 
   1739      1.1  christos 	/* If there are no chains allocated for this buffer, allocate one
   1740      1.1  christos 	 * big enough to hold all the data. */
   1741      1.1  christos 	if (chain == NULL) {
   1742      1.1  christos 		chain = evbuffer_chain_new(datlen);
   1743      1.1  christos 		if (!chain)
   1744      1.1  christos 			goto done;
   1745      1.1  christos 		evbuffer_chain_insert(buf, chain);
   1746      1.1  christos 	}
   1747      1.1  christos 
   1748      1.1  christos 	if ((chain->flags & EVBUFFER_IMMUTABLE) == 0) {
   1749  1.1.1.3  christos 		/* Always true for mutable buffers */
   1750  1.1.1.3  christos 		EVUTIL_ASSERT(chain->misalign >= 0 &&
   1751  1.1.1.3  christos 		    (ev_uint64_t)chain->misalign <= EVBUFFER_CHAIN_MAX);
   1752  1.1.1.3  christos 		remain = chain->buffer_len - (size_t)chain->misalign - chain->off;
   1753      1.1  christos 		if (remain >= datlen) {
   1754      1.1  christos 			/* there's enough space to hold all the data in the
   1755      1.1  christos 			 * current last chain */
   1756      1.1  christos 			memcpy(chain->buffer + chain->misalign + chain->off,
   1757      1.1  christos 			    data, datlen);
   1758      1.1  christos 			chain->off += datlen;
   1759      1.1  christos 			buf->total_len += datlen;
   1760      1.1  christos 			buf->n_add_for_cb += datlen;
   1761      1.1  christos 			goto out;
   1762      1.1  christos 		} else if (!CHAIN_PINNED(chain) &&
   1763      1.1  christos 		    evbuffer_chain_should_realign(chain, datlen)) {
   1764      1.1  christos 			/* we can fit the data into the misalignment */
   1765      1.1  christos 			evbuffer_chain_align(chain);
   1766      1.1  christos 
   1767      1.1  christos 			memcpy(chain->buffer + chain->off, data, datlen);
   1768      1.1  christos 			chain->off += datlen;
   1769      1.1  christos 			buf->total_len += datlen;
   1770      1.1  christos 			buf->n_add_for_cb += datlen;
   1771      1.1  christos 			goto out;
   1772      1.1  christos 		}
   1773      1.1  christos 	} else {
   1774      1.1  christos 		/* we cannot write any data to the last chain */
   1775      1.1  christos 		remain = 0;
   1776      1.1  christos 	}
   1777      1.1  christos 
   1778      1.1  christos 	/* we need to add another chain */
   1779      1.1  christos 	to_alloc = chain->buffer_len;
   1780      1.1  christos 	if (to_alloc <= EVBUFFER_CHAIN_MAX_AUTO_SIZE/2)
   1781      1.1  christos 		to_alloc <<= 1;
   1782      1.1  christos 	if (datlen > to_alloc)
   1783      1.1  christos 		to_alloc = datlen;
   1784      1.1  christos 	tmp = evbuffer_chain_new(to_alloc);
   1785      1.1  christos 	if (tmp == NULL)
   1786      1.1  christos 		goto done;
   1787      1.1  christos 
   1788      1.1  christos 	if (remain) {
   1789      1.1  christos 		memcpy(chain->buffer + chain->misalign + chain->off,
   1790      1.1  christos 		    data, remain);
   1791      1.1  christos 		chain->off += remain;
   1792      1.1  christos 		buf->total_len += remain;
   1793      1.1  christos 		buf->n_add_for_cb += remain;
   1794      1.1  christos 	}
   1795      1.1  christos 
   1796      1.1  christos 	data += remain;
   1797      1.1  christos 	datlen -= remain;
   1798      1.1  christos 
   1799      1.1  christos 	memcpy(tmp->buffer, data, datlen);
   1800      1.1  christos 	tmp->off = datlen;
   1801      1.1  christos 	evbuffer_chain_insert(buf, tmp);
   1802      1.1  christos 	buf->n_add_for_cb += datlen;
   1803      1.1  christos 
   1804      1.1  christos out:
   1805      1.1  christos 	evbuffer_invoke_callbacks_(buf);
   1806      1.1  christos 	result = 0;
   1807      1.1  christos done:
   1808      1.1  christos 	EVBUFFER_UNLOCK(buf);
   1809      1.1  christos 	return result;
   1810      1.1  christos }
   1811      1.1  christos 
   1812      1.1  christos int
   1813      1.1  christos evbuffer_prepend(struct evbuffer *buf, const void *data, size_t datlen)
   1814      1.1  christos {
   1815      1.1  christos 	struct evbuffer_chain *chain, *tmp;
   1816      1.1  christos 	int result = -1;
   1817      1.1  christos 
   1818      1.1  christos 	EVBUFFER_LOCK(buf);
   1819      1.1  christos 
   1820      1.1  christos 	if (buf->freeze_start) {
   1821      1.1  christos 		goto done;
   1822      1.1  christos 	}
   1823  1.1.1.3  christos 	if (datlen > EV_SIZE_MAX - buf->total_len) {
   1824  1.1.1.3  christos 		goto done;
   1825  1.1.1.3  christos 	}
   1826      1.1  christos 
   1827      1.1  christos 	chain = buf->first;
   1828      1.1  christos 
   1829      1.1  christos 	if (chain == NULL) {
   1830      1.1  christos 		chain = evbuffer_chain_new(datlen);
   1831      1.1  christos 		if (!chain)
   1832      1.1  christos 			goto done;
   1833      1.1  christos 		evbuffer_chain_insert(buf, chain);
   1834      1.1  christos 	}
   1835      1.1  christos 
   1836      1.1  christos 	/* we cannot touch immutable buffers */
   1837      1.1  christos 	if ((chain->flags & EVBUFFER_IMMUTABLE) == 0) {
   1838  1.1.1.3  christos 		/* Always true for mutable buffers */
   1839  1.1.1.3  christos 		EVUTIL_ASSERT(chain->misalign >= 0 &&
   1840  1.1.1.3  christos 		    (ev_uint64_t)chain->misalign <= EVBUFFER_CHAIN_MAX);
   1841  1.1.1.3  christos 
   1842      1.1  christos 		/* If this chain is empty, we can treat it as
   1843      1.1  christos 		 * 'empty at the beginning' rather than 'empty at the end' */
   1844      1.1  christos 		if (chain->off == 0)
   1845      1.1  christos 			chain->misalign = chain->buffer_len;
   1846      1.1  christos 
   1847      1.1  christos 		if ((size_t)chain->misalign >= datlen) {
   1848      1.1  christos 			/* we have enough space to fit everything */
   1849      1.1  christos 			memcpy(chain->buffer + chain->misalign - datlen,
   1850      1.1  christos 			    data, datlen);
   1851      1.1  christos 			chain->off += datlen;
   1852      1.1  christos 			chain->misalign -= datlen;
   1853      1.1  christos 			buf->total_len += datlen;
   1854      1.1  christos 			buf->n_add_for_cb += datlen;
   1855      1.1  christos 			goto out;
   1856      1.1  christos 		} else if (chain->misalign) {
   1857      1.1  christos 			/* we can only fit some of the data. */
   1858      1.1  christos 			memcpy(chain->buffer,
   1859      1.1  christos 			    (char*)data + datlen - chain->misalign,
   1860      1.1  christos 			    (size_t)chain->misalign);
   1861      1.1  christos 			chain->off += (size_t)chain->misalign;
   1862      1.1  christos 			buf->total_len += (size_t)chain->misalign;
   1863      1.1  christos 			buf->n_add_for_cb += (size_t)chain->misalign;
   1864      1.1  christos 			datlen -= (size_t)chain->misalign;
   1865      1.1  christos 			chain->misalign = 0;
   1866      1.1  christos 		}
   1867      1.1  christos 	}
   1868      1.1  christos 
   1869      1.1  christos 	/* we need to add another chain */
   1870      1.1  christos 	if ((tmp = evbuffer_chain_new(datlen)) == NULL)
   1871      1.1  christos 		goto done;
   1872      1.1  christos 	buf->first = tmp;
   1873      1.1  christos 	if (buf->last_with_datap == &buf->first)
   1874      1.1  christos 		buf->last_with_datap = &tmp->next;
   1875      1.1  christos 
   1876      1.1  christos 	tmp->next = chain;
   1877      1.1  christos 
   1878      1.1  christos 	tmp->off = datlen;
   1879  1.1.1.3  christos 	EVUTIL_ASSERT(datlen <= tmp->buffer_len);
   1880      1.1  christos 	tmp->misalign = tmp->buffer_len - datlen;
   1881      1.1  christos 
   1882      1.1  christos 	memcpy(tmp->buffer + tmp->misalign, data, datlen);
   1883      1.1  christos 	buf->total_len += datlen;
   1884      1.1  christos 	buf->n_add_for_cb += (size_t)chain->misalign;
   1885      1.1  christos 
   1886      1.1  christos out:
   1887      1.1  christos 	evbuffer_invoke_callbacks_(buf);
   1888      1.1  christos 	result = 0;
   1889      1.1  christos done:
   1890      1.1  christos 	EVBUFFER_UNLOCK(buf);
   1891      1.1  christos 	return result;
   1892      1.1  christos }
   1893      1.1  christos 
   1894      1.1  christos /** Helper: realigns the memory in chain->buffer so that misalign is 0. */
   1895      1.1  christos static void
   1896      1.1  christos evbuffer_chain_align(struct evbuffer_chain *chain)
   1897      1.1  christos {
   1898      1.1  christos 	EVUTIL_ASSERT(!(chain->flags & EVBUFFER_IMMUTABLE));
   1899      1.1  christos 	EVUTIL_ASSERT(!(chain->flags & EVBUFFER_MEM_PINNED_ANY));
   1900      1.1  christos 	memmove(chain->buffer, chain->buffer + chain->misalign, chain->off);
   1901      1.1  christos 	chain->misalign = 0;
   1902      1.1  christos }
   1903      1.1  christos 
   1904      1.1  christos #define MAX_TO_COPY_IN_EXPAND 4096
   1905      1.1  christos #define MAX_TO_REALIGN_IN_EXPAND 2048
   1906      1.1  christos 
   1907      1.1  christos /** Helper: return true iff we should realign chain to fit datalen bytes of
   1908      1.1  christos     data in it. */
   1909      1.1  christos static int
   1910      1.1  christos evbuffer_chain_should_realign(struct evbuffer_chain *chain,
   1911      1.1  christos     size_t datlen)
   1912      1.1  christos {
   1913      1.1  christos 	return chain->buffer_len - chain->off >= datlen &&
   1914      1.1  christos 	    (chain->off < chain->buffer_len / 2) &&
   1915      1.1  christos 	    (chain->off <= MAX_TO_REALIGN_IN_EXPAND);
   1916      1.1  christos }
   1917      1.1  christos 
   1918      1.1  christos /* Expands the available space in the event buffer to at least datlen, all in
   1919      1.1  christos  * a single chunk.  Return that chunk. */
   1920      1.1  christos static struct evbuffer_chain *
   1921      1.1  christos evbuffer_expand_singlechain(struct evbuffer *buf, size_t datlen)
   1922      1.1  christos {
   1923      1.1  christos 	struct evbuffer_chain *chain, **chainp;
   1924      1.1  christos 	struct evbuffer_chain *result = NULL;
   1925      1.1  christos 	ASSERT_EVBUFFER_LOCKED(buf);
   1926      1.1  christos 
   1927      1.1  christos 	chainp = buf->last_with_datap;
   1928      1.1  christos 
   1929      1.1  christos 	/* XXX If *chainp is no longer writeable, but has enough space in its
   1930      1.1  christos 	 * misalign, this might be a bad idea: we could still use *chainp, not
   1931      1.1  christos 	 * (*chainp)->next. */
   1932      1.1  christos 	if (*chainp && CHAIN_SPACE_LEN(*chainp) == 0)
   1933      1.1  christos 		chainp = &(*chainp)->next;
   1934      1.1  christos 
   1935      1.1  christos 	/* 'chain' now points to the first chain with writable space (if any)
   1936      1.1  christos 	 * We will either use it, realign it, replace it, or resize it. */
   1937      1.1  christos 	chain = *chainp;
   1938      1.1  christos 
   1939      1.1  christos 	if (chain == NULL ||
   1940      1.1  christos 	    (chain->flags & (EVBUFFER_IMMUTABLE|EVBUFFER_MEM_PINNED_ANY))) {
   1941      1.1  christos 		/* We can't use the last_with_data chain at all.  Just add a
   1942      1.1  christos 		 * new one that's big enough. */
   1943      1.1  christos 		goto insert_new;
   1944      1.1  christos 	}
   1945      1.1  christos 
   1946      1.1  christos 	/* If we can fit all the data, then we don't have to do anything */
   1947      1.1  christos 	if (CHAIN_SPACE_LEN(chain) >= datlen) {
   1948      1.1  christos 		result = chain;
   1949      1.1  christos 		goto ok;
   1950      1.1  christos 	}
   1951      1.1  christos 
   1952      1.1  christos 	/* If the chain is completely empty, just replace it by adding a new
   1953      1.1  christos 	 * empty chain. */
   1954      1.1  christos 	if (chain->off == 0) {
   1955      1.1  christos 		goto insert_new;
   1956      1.1  christos 	}
   1957      1.1  christos 
   1958      1.1  christos 	/* If the misalignment plus the remaining space fulfills our data
   1959      1.1  christos 	 * needs, we could just force an alignment to happen.  Afterwards, we
   1960      1.1  christos 	 * have enough space.  But only do this if we're saving a lot of space
   1961      1.1  christos 	 * and not moving too much data.  Otherwise the space savings are
   1962      1.1  christos 	 * probably offset by the time lost in copying.
   1963      1.1  christos 	 */
   1964      1.1  christos 	if (evbuffer_chain_should_realign(chain, datlen)) {
   1965      1.1  christos 		evbuffer_chain_align(chain);
   1966      1.1  christos 		result = chain;
   1967      1.1  christos 		goto ok;
   1968      1.1  christos 	}
   1969      1.1  christos 
   1970      1.1  christos 	/* At this point, we can either resize the last chunk with space in
   1971      1.1  christos 	 * it, use the next chunk after it, or   If we add a new chunk, we waste
   1972      1.1  christos 	 * CHAIN_SPACE_LEN(chain) bytes in the former last chunk.  If we
   1973      1.1  christos 	 * resize, we have to copy chain->off bytes.
   1974      1.1  christos 	 */
   1975      1.1  christos 
   1976      1.1  christos 	/* Would expanding this chunk be affordable and worthwhile? */
   1977      1.1  christos 	if (CHAIN_SPACE_LEN(chain) < chain->buffer_len / 8 ||
   1978  1.1.1.3  christos 	    chain->off > MAX_TO_COPY_IN_EXPAND ||
   1979  1.1.1.3  christos 	    (datlen < EVBUFFER_CHAIN_MAX &&
   1980  1.1.1.3  christos 		EVBUFFER_CHAIN_MAX - datlen >= chain->off)) {
   1981      1.1  christos 		/* It's not worth resizing this chain. Can the next one be
   1982      1.1  christos 		 * used? */
   1983      1.1  christos 		if (chain->next && CHAIN_SPACE_LEN(chain->next) >= datlen) {
   1984      1.1  christos 			/* Yes, we can just use the next chain (which should
   1985      1.1  christos 			 * be empty. */
   1986      1.1  christos 			result = chain->next;
   1987      1.1  christos 			goto ok;
   1988      1.1  christos 		} else {
   1989      1.1  christos 			/* No; append a new chain (which will free all
   1990      1.1  christos 			 * terminal empty chains.) */
   1991      1.1  christos 			goto insert_new;
   1992      1.1  christos 		}
   1993      1.1  christos 	} else {
   1994      1.1  christos 		/* Okay, we're going to try to resize this chain: Not doing so
   1995      1.1  christos 		 * would waste at least 1/8 of its current allocation, and we
   1996      1.1  christos 		 * can do so without having to copy more than
   1997      1.1  christos 		 * MAX_TO_COPY_IN_EXPAND bytes. */
   1998      1.1  christos 		/* figure out how much space we need */
   1999      1.1  christos 		size_t length = chain->off + datlen;
   2000      1.1  christos 		struct evbuffer_chain *tmp = evbuffer_chain_new(length);
   2001      1.1  christos 		if (tmp == NULL)
   2002      1.1  christos 			goto err;
   2003      1.1  christos 
   2004      1.1  christos 		/* copy the data over that we had so far */
   2005      1.1  christos 		tmp->off = chain->off;
   2006      1.1  christos 		memcpy(tmp->buffer, chain->buffer + chain->misalign,
   2007      1.1  christos 		    chain->off);
   2008      1.1  christos 		/* fix up the list */
   2009      1.1  christos 		EVUTIL_ASSERT(*chainp == chain);
   2010      1.1  christos 		result = *chainp = tmp;
   2011      1.1  christos 
   2012      1.1  christos 		if (buf->last == chain)
   2013      1.1  christos 			buf->last = tmp;
   2014      1.1  christos 
   2015      1.1  christos 		tmp->next = chain->next;
   2016      1.1  christos 		evbuffer_chain_free(chain);
   2017      1.1  christos 		goto ok;
   2018      1.1  christos 	}
   2019      1.1  christos 
   2020      1.1  christos insert_new:
   2021      1.1  christos 	result = evbuffer_chain_insert_new(buf, datlen);
   2022      1.1  christos 	if (!result)
   2023      1.1  christos 		goto err;
   2024      1.1  christos ok:
   2025      1.1  christos 	EVUTIL_ASSERT(result);
   2026      1.1  christos 	EVUTIL_ASSERT(CHAIN_SPACE_LEN(result) >= datlen);
   2027      1.1  christos err:
   2028      1.1  christos 	return result;
   2029      1.1  christos }
   2030      1.1  christos 
   2031      1.1  christos /* Make sure that datlen bytes are available for writing in the last n
   2032      1.1  christos  * chains.  Never copies or moves data. */
   2033      1.1  christos int
   2034      1.1  christos evbuffer_expand_fast_(struct evbuffer *buf, size_t datlen, int n)
   2035      1.1  christos {
   2036      1.1  christos 	struct evbuffer_chain *chain = buf->last, *tmp, *next;
   2037      1.1  christos 	size_t avail;
   2038      1.1  christos 	int used;
   2039      1.1  christos 
   2040      1.1  christos 	ASSERT_EVBUFFER_LOCKED(buf);
   2041      1.1  christos 	EVUTIL_ASSERT(n >= 2);
   2042      1.1  christos 
   2043      1.1  christos 	if (chain == NULL || (chain->flags & EVBUFFER_IMMUTABLE)) {
   2044      1.1  christos 		/* There is no last chunk, or we can't touch the last chunk.
   2045      1.1  christos 		 * Just add a new chunk. */
   2046      1.1  christos 		chain = evbuffer_chain_new(datlen);
   2047      1.1  christos 		if (chain == NULL)
   2048      1.1  christos 			return (-1);
   2049      1.1  christos 
   2050      1.1  christos 		evbuffer_chain_insert(buf, chain);
   2051      1.1  christos 		return (0);
   2052      1.1  christos 	}
   2053      1.1  christos 
   2054      1.1  christos 	used = 0; /* number of chains we're using space in. */
   2055      1.1  christos 	avail = 0; /* how much space they have. */
   2056      1.1  christos 	/* How many bytes can we stick at the end of buffer as it is?  Iterate
   2057      1.1  christos 	 * over the chains at the end of the buffer, tring to see how much
   2058      1.1  christos 	 * space we have in the first n. */
   2059      1.1  christos 	for (chain = *buf->last_with_datap; chain; chain = chain->next) {
   2060      1.1  christos 		if (chain->off) {
   2061      1.1  christos 			size_t space = (size_t) CHAIN_SPACE_LEN(chain);
   2062      1.1  christos 			EVUTIL_ASSERT(chain == *buf->last_with_datap);
   2063      1.1  christos 			if (space) {
   2064      1.1  christos 				avail += space;
   2065      1.1  christos 				++used;
   2066      1.1  christos 			}
   2067      1.1  christos 		} else {
   2068      1.1  christos 			/* No data in chain; realign it. */
   2069      1.1  christos 			chain->misalign = 0;
   2070      1.1  christos 			avail += chain->buffer_len;
   2071      1.1  christos 			++used;
   2072      1.1  christos 		}
   2073      1.1  christos 		if (avail >= datlen) {
   2074      1.1  christos 			/* There is already enough space.  Just return */
   2075      1.1  christos 			return (0);
   2076      1.1  christos 		}
   2077      1.1  christos 		if (used == n)
   2078      1.1  christos 			break;
   2079      1.1  christos 	}
   2080      1.1  christos 
   2081      1.1  christos 	/* There wasn't enough space in the first n chains with space in
   2082      1.1  christos 	 * them. Either add a new chain with enough space, or replace all
   2083      1.1  christos 	 * empty chains with one that has enough space, depending on n. */
   2084      1.1  christos 	if (used < n) {
   2085      1.1  christos 		/* The loop ran off the end of the chains before it hit n
   2086      1.1  christos 		 * chains; we can add another. */
   2087      1.1  christos 		EVUTIL_ASSERT(chain == NULL);
   2088      1.1  christos 
   2089      1.1  christos 		tmp = evbuffer_chain_new(datlen - avail);
   2090      1.1  christos 		if (tmp == NULL)
   2091      1.1  christos 			return (-1);
   2092      1.1  christos 
   2093      1.1  christos 		buf->last->next = tmp;
   2094      1.1  christos 		buf->last = tmp;
   2095      1.1  christos 		/* (we would only set last_with_data if we added the first
   2096      1.1  christos 		 * chain. But if the buffer had no chains, we would have
   2097      1.1  christos 		 * just allocated a new chain earlier) */
   2098      1.1  christos 		return (0);
   2099      1.1  christos 	} else {
   2100      1.1  christos 		/* Nuke _all_ the empty chains. */
   2101      1.1  christos 		int rmv_all = 0; /* True iff we removed last_with_data. */
   2102      1.1  christos 		chain = *buf->last_with_datap;
   2103      1.1  christos 		if (!chain->off) {
   2104      1.1  christos 			EVUTIL_ASSERT(chain == buf->first);
   2105      1.1  christos 			rmv_all = 1;
   2106      1.1  christos 			avail = 0;
   2107      1.1  christos 		} else {
   2108  1.1.1.3  christos 			/* can't overflow, since only mutable chains have
   2109  1.1.1.3  christos 			 * huge misaligns. */
   2110      1.1  christos 			avail = (size_t) CHAIN_SPACE_LEN(chain);
   2111      1.1  christos 			chain = chain->next;
   2112      1.1  christos 		}
   2113      1.1  christos 
   2114      1.1  christos 
   2115      1.1  christos 		for (; chain; chain = next) {
   2116      1.1  christos 			next = chain->next;
   2117      1.1  christos 			EVUTIL_ASSERT(chain->off == 0);
   2118      1.1  christos 			evbuffer_chain_free(chain);
   2119      1.1  christos 		}
   2120  1.1.1.3  christos 		EVUTIL_ASSERT(datlen >= avail);
   2121      1.1  christos 		tmp = evbuffer_chain_new(datlen - avail);
   2122      1.1  christos 		if (tmp == NULL) {
   2123      1.1  christos 			if (rmv_all) {
   2124      1.1  christos 				ZERO_CHAIN(buf);
   2125      1.1  christos 			} else {
   2126      1.1  christos 				buf->last = *buf->last_with_datap;
   2127      1.1  christos 				(*buf->last_with_datap)->next = NULL;
   2128      1.1  christos 			}
   2129      1.1  christos 			return (-1);
   2130      1.1  christos 		}
   2131      1.1  christos 
   2132      1.1  christos 		if (rmv_all) {
   2133      1.1  christos 			buf->first = buf->last = tmp;
   2134      1.1  christos 			buf->last_with_datap = &buf->first;
   2135      1.1  christos 		} else {
   2136      1.1  christos 			(*buf->last_with_datap)->next = tmp;
   2137      1.1  christos 			buf->last = tmp;
   2138      1.1  christos 		}
   2139      1.1  christos 		return (0);
   2140      1.1  christos 	}
   2141      1.1  christos }
   2142      1.1  christos 
   2143      1.1  christos int
   2144      1.1  christos evbuffer_expand(struct evbuffer *buf, size_t datlen)
   2145      1.1  christos {
   2146      1.1  christos 	struct evbuffer_chain *chain;
   2147      1.1  christos 
   2148      1.1  christos 	EVBUFFER_LOCK(buf);
   2149      1.1  christos 	chain = evbuffer_expand_singlechain(buf, datlen);
   2150      1.1  christos 	EVBUFFER_UNLOCK(buf);
   2151      1.1  christos 	return chain ? 0 : -1;
   2152      1.1  christos }
   2153      1.1  christos 
   2154      1.1  christos /*
   2155      1.1  christos  * Reads data from a file descriptor into a buffer.
   2156      1.1  christos  */
   2157      1.1  christos 
   2158      1.1  christos #if defined(EVENT__HAVE_SYS_UIO_H) || defined(_WIN32)
   2159      1.1  christos #define USE_IOVEC_IMPL
   2160      1.1  christos #endif
   2161      1.1  christos 
   2162      1.1  christos #ifdef USE_IOVEC_IMPL
   2163      1.1  christos 
   2164      1.1  christos #ifdef EVENT__HAVE_SYS_UIO_H
   2165      1.1  christos /* number of iovec we use for writev, fragmentation is going to determine
   2166      1.1  christos  * how much we end up writing */
   2167      1.1  christos 
   2168      1.1  christos #define DEFAULT_WRITE_IOVEC 128
   2169      1.1  christos 
   2170      1.1  christos #if defined(UIO_MAXIOV) && UIO_MAXIOV < DEFAULT_WRITE_IOVEC
   2171      1.1  christos #define NUM_WRITE_IOVEC UIO_MAXIOV
   2172      1.1  christos #elif defined(IOV_MAX) && IOV_MAX < DEFAULT_WRITE_IOVEC
   2173      1.1  christos #define NUM_WRITE_IOVEC IOV_MAX
   2174      1.1  christos #else
   2175      1.1  christos #define NUM_WRITE_IOVEC DEFAULT_WRITE_IOVEC
   2176      1.1  christos #endif
   2177      1.1  christos 
   2178      1.1  christos #define IOV_TYPE struct iovec
   2179      1.1  christos #define IOV_PTR_FIELD iov_base
   2180      1.1  christos #define IOV_LEN_FIELD iov_len
   2181      1.1  christos #define IOV_LEN_TYPE size_t
   2182      1.1  christos #else
   2183      1.1  christos #define NUM_WRITE_IOVEC 16
   2184      1.1  christos #define IOV_TYPE WSABUF
   2185      1.1  christos #define IOV_PTR_FIELD buf
   2186      1.1  christos #define IOV_LEN_FIELD len
   2187      1.1  christos #define IOV_LEN_TYPE unsigned long
   2188      1.1  christos #endif
   2189      1.1  christos #endif
   2190      1.1  christos #define NUM_READ_IOVEC 4
   2191      1.1  christos 
   2192      1.1  christos #define EVBUFFER_MAX_READ	4096
   2193      1.1  christos 
   2194      1.1  christos /** Helper function to figure out which space to use for reading data into
   2195      1.1  christos     an evbuffer.  Internal use only.
   2196      1.1  christos 
   2197      1.1  christos     @param buf The buffer to read into
   2198      1.1  christos     @param howmuch How much we want to read.
   2199      1.1  christos     @param vecs An array of two or more iovecs or WSABUFs.
   2200      1.1  christos     @param n_vecs_avail The length of vecs
   2201      1.1  christos     @param chainp A pointer to a variable to hold the first chain we're
   2202      1.1  christos       reading into.
   2203      1.1  christos     @param exact Boolean: if true, we do not provide more than 'howmuch'
   2204      1.1  christos       space in the vectors, even if more space is available.
   2205      1.1  christos     @return The number of buffers we're using.
   2206      1.1  christos  */
   2207      1.1  christos int
   2208      1.1  christos evbuffer_read_setup_vecs_(struct evbuffer *buf, ev_ssize_t howmuch,
   2209      1.1  christos     struct evbuffer_iovec *vecs, int n_vecs_avail,
   2210      1.1  christos     struct evbuffer_chain ***chainp, int exact)
   2211      1.1  christos {
   2212      1.1  christos 	struct evbuffer_chain *chain;
   2213      1.1  christos 	struct evbuffer_chain **firstchainp;
   2214      1.1  christos 	size_t so_far;
   2215      1.1  christos 	int i;
   2216      1.1  christos 	ASSERT_EVBUFFER_LOCKED(buf);
   2217      1.1  christos 
   2218      1.1  christos 	if (howmuch < 0)
   2219      1.1  christos 		return -1;
   2220      1.1  christos 
   2221      1.1  christos 	so_far = 0;
   2222      1.1  christos 	/* Let firstchain be the first chain with any space on it */
   2223      1.1  christos 	firstchainp = buf->last_with_datap;
   2224      1.1  christos 	if (CHAIN_SPACE_LEN(*firstchainp) == 0) {
   2225      1.1  christos 		firstchainp = &(*firstchainp)->next;
   2226      1.1  christos 	}
   2227      1.1  christos 
   2228      1.1  christos 	chain = *firstchainp;
   2229      1.1  christos 	for (i = 0; i < n_vecs_avail && so_far < (size_t)howmuch; ++i) {
   2230      1.1  christos 		size_t avail = (size_t) CHAIN_SPACE_LEN(chain);
   2231      1.1  christos 		if (avail > (howmuch - so_far) && exact)
   2232      1.1  christos 			avail = howmuch - so_far;
   2233      1.1  christos 		vecs[i].iov_base = CHAIN_SPACE_PTR(chain);
   2234      1.1  christos 		vecs[i].iov_len = avail;
   2235      1.1  christos 		so_far += avail;
   2236      1.1  christos 		chain = chain->next;
   2237      1.1  christos 	}
   2238      1.1  christos 
   2239      1.1  christos 	*chainp = firstchainp;
   2240      1.1  christos 	return i;
   2241      1.1  christos }
   2242      1.1  christos 
   2243      1.1  christos static int
   2244      1.1  christos get_n_bytes_readable_on_socket(evutil_socket_t fd)
   2245      1.1  christos {
   2246      1.1  christos #if defined(FIONREAD) && defined(_WIN32)
   2247      1.1  christos 	unsigned long lng = EVBUFFER_MAX_READ;
   2248      1.1  christos 	if (ioctlsocket(fd, FIONREAD, &lng) < 0)
   2249      1.1  christos 		return -1;
   2250  1.1.1.3  christos 	/* Can overflow, but mostly harmlessly. XXXX */
   2251      1.1  christos 	return (int)lng;
   2252      1.1  christos #elif defined(FIONREAD)
   2253      1.1  christos 	int n = EVBUFFER_MAX_READ;
   2254      1.1  christos 	if (ioctl(fd, FIONREAD, &n) < 0)
   2255      1.1  christos 		return -1;
   2256      1.1  christos 	return n;
   2257      1.1  christos #else
   2258      1.1  christos 	return EVBUFFER_MAX_READ;
   2259      1.1  christos #endif
   2260      1.1  christos }
   2261      1.1  christos 
   2262      1.1  christos /* TODO(niels): should this function return ev_ssize_t and take ev_ssize_t
   2263      1.1  christos  * as howmuch? */
   2264      1.1  christos int
   2265      1.1  christos evbuffer_read(struct evbuffer *buf, evutil_socket_t fd, int howmuch)
   2266      1.1  christos {
   2267      1.1  christos 	struct evbuffer_chain **chainp;
   2268      1.1  christos 	int n;
   2269      1.1  christos 	int result;
   2270      1.1  christos 
   2271      1.1  christos #ifdef USE_IOVEC_IMPL
   2272      1.1  christos 	int nvecs, i, remaining;
   2273      1.1  christos #else
   2274      1.1  christos 	struct evbuffer_chain *chain;
   2275      1.1  christos 	unsigned char *p;
   2276      1.1  christos #endif
   2277      1.1  christos 
   2278      1.1  christos 	EVBUFFER_LOCK(buf);
   2279      1.1  christos 
   2280      1.1  christos 	if (buf->freeze_end) {
   2281      1.1  christos 		result = -1;
   2282      1.1  christos 		goto done;
   2283      1.1  christos 	}
   2284      1.1  christos 
   2285      1.1  christos 	n = get_n_bytes_readable_on_socket(fd);
   2286      1.1  christos 	if (n <= 0 || n > EVBUFFER_MAX_READ)
   2287      1.1  christos 		n = EVBUFFER_MAX_READ;
   2288      1.1  christos 	if (howmuch < 0 || howmuch > n)
   2289      1.1  christos 		howmuch = n;
   2290      1.1  christos 
   2291      1.1  christos #ifdef USE_IOVEC_IMPL
   2292      1.1  christos 	/* Since we can use iovecs, we're willing to use the last
   2293      1.1  christos 	 * NUM_READ_IOVEC chains. */
   2294      1.1  christos 	if (evbuffer_expand_fast_(buf, howmuch, NUM_READ_IOVEC) == -1) {
   2295      1.1  christos 		result = -1;
   2296      1.1  christos 		goto done;
   2297      1.1  christos 	} else {
   2298      1.1  christos 		IOV_TYPE vecs[NUM_READ_IOVEC];
   2299      1.1  christos #ifdef EVBUFFER_IOVEC_IS_NATIVE_
   2300      1.1  christos 		nvecs = evbuffer_read_setup_vecs_(buf, howmuch, vecs,
   2301      1.1  christos 		    NUM_READ_IOVEC, &chainp, 1);
   2302      1.1  christos #else
   2303      1.1  christos 		/* We aren't using the native struct iovec.  Therefore,
   2304      1.1  christos 		   we are on win32. */
   2305      1.1  christos 		struct evbuffer_iovec ev_vecs[NUM_READ_IOVEC];
   2306      1.1  christos 		nvecs = evbuffer_read_setup_vecs_(buf, howmuch, ev_vecs, 2,
   2307      1.1  christos 		    &chainp, 1);
   2308      1.1  christos 
   2309      1.1  christos 		for (i=0; i < nvecs; ++i)
   2310      1.1  christos 			WSABUF_FROM_EVBUFFER_IOV(&vecs[i], &ev_vecs[i]);
   2311      1.1  christos #endif
   2312      1.1  christos 
   2313      1.1  christos #ifdef _WIN32
   2314      1.1  christos 		{
   2315      1.1  christos 			DWORD bytesRead;
   2316      1.1  christos 			DWORD flags=0;
   2317      1.1  christos 			if (WSARecv(fd, vecs, nvecs, &bytesRead, &flags, NULL, NULL)) {
   2318      1.1  christos 				/* The read failed. It might be a close,
   2319      1.1  christos 				 * or it might be an error. */
   2320      1.1  christos 				if (WSAGetLastError() == WSAECONNABORTED)
   2321      1.1  christos 					n = 0;
   2322      1.1  christos 				else
   2323      1.1  christos 					n = -1;
   2324      1.1  christos 			} else
   2325      1.1  christos 				n = bytesRead;
   2326      1.1  christos 		}
   2327      1.1  christos #else
   2328      1.1  christos 		n = readv(fd, vecs, nvecs);
   2329      1.1  christos #endif
   2330      1.1  christos 	}
   2331      1.1  christos 
   2332      1.1  christos #else /*!USE_IOVEC_IMPL*/
   2333      1.1  christos 	/* If we don't have FIONREAD, we might waste some space here */
   2334      1.1  christos 	/* XXX we _will_ waste some space here if there is any space left
   2335      1.1  christos 	 * over on buf->last. */
   2336      1.1  christos 	if ((chain = evbuffer_expand_singlechain(buf, howmuch)) == NULL) {
   2337      1.1  christos 		result = -1;
   2338      1.1  christos 		goto done;
   2339      1.1  christos 	}
   2340      1.1  christos 
   2341      1.1  christos 	/* We can append new data at this point */
   2342      1.1  christos 	p = chain->buffer + chain->misalign + chain->off;
   2343      1.1  christos 
   2344      1.1  christos #ifndef _WIN32
   2345      1.1  christos 	n = read(fd, p, howmuch);
   2346      1.1  christos #else
   2347      1.1  christos 	n = recv(fd, p, howmuch, 0);
   2348      1.1  christos #endif
   2349      1.1  christos #endif /* USE_IOVEC_IMPL */
   2350      1.1  christos 
   2351      1.1  christos 	if (n == -1) {
   2352      1.1  christos 		result = -1;
   2353      1.1  christos 		goto done;
   2354      1.1  christos 	}
   2355      1.1  christos 	if (n == 0) {
   2356      1.1  christos 		result = 0;
   2357      1.1  christos 		goto done;
   2358      1.1  christos 	}
   2359      1.1  christos 
   2360      1.1  christos #ifdef USE_IOVEC_IMPL
   2361      1.1  christos 	remaining = n;
   2362      1.1  christos 	for (i=0; i < nvecs; ++i) {
   2363  1.1.1.3  christos 		/* can't overflow, since only mutable chains have
   2364  1.1.1.3  christos 		 * huge misaligns. */
   2365  1.1.1.3  christos 		size_t space = (size_t) CHAIN_SPACE_LEN(*chainp);
   2366  1.1.1.3  christos 		/* XXXX This is a kludge that can waste space in perverse
   2367  1.1.1.3  christos 		 * situations. */
   2368  1.1.1.3  christos 		if (space > EVBUFFER_CHAIN_MAX)
   2369  1.1.1.3  christos 			space = EVBUFFER_CHAIN_MAX;
   2370  1.1.1.3  christos 		if ((ev_ssize_t)space < remaining) {
   2371      1.1  christos 			(*chainp)->off += space;
   2372      1.1  christos 			remaining -= (int)space;
   2373      1.1  christos 		} else {
   2374      1.1  christos 			(*chainp)->off += remaining;
   2375      1.1  christos 			buf->last_with_datap = chainp;
   2376      1.1  christos 			break;
   2377      1.1  christos 		}
   2378      1.1  christos 		chainp = &(*chainp)->next;
   2379      1.1  christos 	}
   2380      1.1  christos #else
   2381      1.1  christos 	chain->off += n;
   2382      1.1  christos 	advance_last_with_data(buf);
   2383      1.1  christos #endif
   2384      1.1  christos 	buf->total_len += n;
   2385      1.1  christos 	buf->n_add_for_cb += n;
   2386      1.1  christos 
   2387      1.1  christos 	/* Tell someone about changes in this buffer */
   2388      1.1  christos 	evbuffer_invoke_callbacks_(buf);
   2389      1.1  christos 	result = n;
   2390      1.1  christos done:
   2391      1.1  christos 	EVBUFFER_UNLOCK(buf);
   2392      1.1  christos 	return result;
   2393      1.1  christos }
   2394      1.1  christos 
   2395      1.1  christos #ifdef USE_IOVEC_IMPL
   2396      1.1  christos static inline int
   2397      1.1  christos evbuffer_write_iovec(struct evbuffer *buffer, evutil_socket_t fd,
   2398      1.1  christos     ev_ssize_t howmuch)
   2399      1.1  christos {
   2400      1.1  christos 	IOV_TYPE iov[NUM_WRITE_IOVEC];
   2401      1.1  christos 	struct evbuffer_chain *chain = buffer->first;
   2402      1.1  christos 	int n, i = 0;
   2403      1.1  christos 
   2404      1.1  christos 	if (howmuch < 0)
   2405      1.1  christos 		return -1;
   2406      1.1  christos 
   2407      1.1  christos 	ASSERT_EVBUFFER_LOCKED(buffer);
   2408      1.1  christos 	/* XXX make this top out at some maximal data length?  if the
   2409      1.1  christos 	 * buffer has (say) 1MB in it, split over 128 chains, there's
   2410      1.1  christos 	 * no way it all gets written in one go. */
   2411      1.1  christos 	while (chain != NULL && i < NUM_WRITE_IOVEC && howmuch) {
   2412      1.1  christos #ifdef USE_SENDFILE
   2413      1.1  christos 		/* we cannot write the file info via writev */
   2414      1.1  christos 		if (chain->flags & EVBUFFER_SENDFILE)
   2415      1.1  christos 			break;
   2416      1.1  christos #endif
   2417      1.1  christos 		iov[i].IOV_PTR_FIELD = (void *) (chain->buffer + chain->misalign);
   2418      1.1  christos 		if ((size_t)howmuch >= chain->off) {
   2419      1.1  christos 			/* XXXcould be problematic when windows supports mmap*/
   2420      1.1  christos 			iov[i++].IOV_LEN_FIELD = (IOV_LEN_TYPE)chain->off;
   2421      1.1  christos 			howmuch -= chain->off;
   2422      1.1  christos 		} else {
   2423      1.1  christos 			/* XXXcould be problematic when windows supports mmap*/
   2424      1.1  christos 			iov[i++].IOV_LEN_FIELD = (IOV_LEN_TYPE)howmuch;
   2425      1.1  christos 			break;
   2426      1.1  christos 		}
   2427      1.1  christos 		chain = chain->next;
   2428      1.1  christos 	}
   2429      1.1  christos 	if (! i)
   2430      1.1  christos 		return 0;
   2431      1.1  christos 
   2432      1.1  christos #ifdef _WIN32
   2433      1.1  christos 	{
   2434      1.1  christos 		DWORD bytesSent;
   2435      1.1  christos 		if (WSASend(fd, iov, i, &bytesSent, 0, NULL, NULL))
   2436      1.1  christos 			n = -1;
   2437      1.1  christos 		else
   2438      1.1  christos 			n = bytesSent;
   2439      1.1  christos 	}
   2440      1.1  christos #else
   2441      1.1  christos 	n = writev(fd, iov, i);
   2442      1.1  christos #endif
   2443      1.1  christos 	return (n);
   2444      1.1  christos }
   2445      1.1  christos #endif
   2446      1.1  christos 
   2447      1.1  christos #ifdef USE_SENDFILE
   2448      1.1  christos static inline int
   2449      1.1  christos evbuffer_write_sendfile(struct evbuffer *buffer, evutil_socket_t dest_fd,
   2450      1.1  christos     ev_ssize_t howmuch)
   2451      1.1  christos {
   2452      1.1  christos 	struct evbuffer_chain *chain = buffer->first;
   2453      1.1  christos 	struct evbuffer_chain_file_segment *info =
   2454      1.1  christos 	    EVBUFFER_CHAIN_EXTRA(struct evbuffer_chain_file_segment,
   2455      1.1  christos 		chain);
   2456      1.1  christos 	const int source_fd = info->segment->fd;
   2457      1.1  christos #if defined(SENDFILE_IS_MACOSX) || defined(SENDFILE_IS_FREEBSD)
   2458      1.1  christos 	int res;
   2459      1.1  christos 	ev_off_t len = chain->off;
   2460      1.1  christos #elif defined(SENDFILE_IS_LINUX) || defined(SENDFILE_IS_SOLARIS)
   2461      1.1  christos 	ev_ssize_t res;
   2462      1.1  christos 	ev_off_t offset = chain->misalign;
   2463      1.1  christos #endif
   2464      1.1  christos 
   2465      1.1  christos 	ASSERT_EVBUFFER_LOCKED(buffer);
   2466      1.1  christos 
   2467      1.1  christos #if defined(SENDFILE_IS_MACOSX)
   2468      1.1  christos 	res = sendfile(source_fd, dest_fd, chain->misalign, &len, NULL, 0);
   2469      1.1  christos 	if (res == -1 && !EVUTIL_ERR_RW_RETRIABLE(errno))
   2470      1.1  christos 		return (-1);
   2471      1.1  christos 
   2472      1.1  christos 	return (len);
   2473      1.1  christos #elif defined(SENDFILE_IS_FREEBSD)
   2474      1.1  christos 	res = sendfile(source_fd, dest_fd, chain->misalign, chain->off, NULL, &len, 0);
   2475      1.1  christos 	if (res == -1 && !EVUTIL_ERR_RW_RETRIABLE(errno))
   2476      1.1  christos 		return (-1);
   2477      1.1  christos 
   2478      1.1  christos 	return (len);
   2479      1.1  christos #elif defined(SENDFILE_IS_LINUX)
   2480      1.1  christos 	/* TODO(niels): implement splice */
   2481      1.1  christos 	res = sendfile(dest_fd, source_fd, &offset, chain->off);
   2482      1.1  christos 	if (res == -1 && EVUTIL_ERR_RW_RETRIABLE(errno)) {
   2483      1.1  christos 		/* if this is EAGAIN or EINTR return 0; otherwise, -1 */
   2484      1.1  christos 		return (0);
   2485      1.1  christos 	}
   2486      1.1  christos 	return (res);
   2487      1.1  christos #elif defined(SENDFILE_IS_SOLARIS)
   2488      1.1  christos 	{
   2489      1.1  christos 		const off_t offset_orig = offset;
   2490      1.1  christos 		res = sendfile(dest_fd, source_fd, &offset, chain->off);
   2491      1.1  christos 		if (res == -1 && EVUTIL_ERR_RW_RETRIABLE(errno)) {
   2492      1.1  christos 			if (offset - offset_orig)
   2493      1.1  christos 				return offset - offset_orig;
   2494      1.1  christos 			/* if this is EAGAIN or EINTR and no bytes were
   2495      1.1  christos 			 * written, return 0 */
   2496      1.1  christos 			return (0);
   2497      1.1  christos 		}
   2498      1.1  christos 		return (res);
   2499      1.1  christos 	}
   2500      1.1  christos #endif
   2501      1.1  christos }
   2502      1.1  christos #endif
   2503      1.1  christos 
   2504      1.1  christos int
   2505      1.1  christos evbuffer_write_atmost(struct evbuffer *buffer, evutil_socket_t fd,
   2506      1.1  christos     ev_ssize_t howmuch)
   2507      1.1  christos {
   2508      1.1  christos 	int n = -1;
   2509      1.1  christos 
   2510      1.1  christos 	EVBUFFER_LOCK(buffer);
   2511      1.1  christos 
   2512      1.1  christos 	if (buffer->freeze_start) {
   2513      1.1  christos 		goto done;
   2514      1.1  christos 	}
   2515      1.1  christos 
   2516      1.1  christos 	if (howmuch < 0 || (size_t)howmuch > buffer->total_len)
   2517      1.1  christos 		howmuch = buffer->total_len;
   2518      1.1  christos 
   2519      1.1  christos 	if (howmuch > 0) {
   2520      1.1  christos #ifdef USE_SENDFILE
   2521      1.1  christos 		struct evbuffer_chain *chain = buffer->first;
   2522      1.1  christos 		if (chain != NULL && (chain->flags & EVBUFFER_SENDFILE))
   2523      1.1  christos 			n = evbuffer_write_sendfile(buffer, fd, howmuch);
   2524      1.1  christos 		else {
   2525      1.1  christos #endif
   2526      1.1  christos #ifdef USE_IOVEC_IMPL
   2527      1.1  christos 		n = evbuffer_write_iovec(buffer, fd, howmuch);
   2528      1.1  christos #elif defined(_WIN32)
   2529      1.1  christos 		/* XXX(nickm) Don't disable this code until we know if
   2530      1.1  christos 		 * the WSARecv code above works. */
   2531      1.1  christos 		void *p = evbuffer_pullup(buffer, howmuch);
   2532  1.1.1.3  christos 		EVUTIL_ASSERT(p || !howmuch);
   2533      1.1  christos 		n = send(fd, p, howmuch, 0);
   2534      1.1  christos #else
   2535      1.1  christos 		void *p = evbuffer_pullup(buffer, howmuch);
   2536  1.1.1.3  christos 		EVUTIL_ASSERT(p || !howmuch);
   2537      1.1  christos 		n = write(fd, p, howmuch);
   2538      1.1  christos #endif
   2539      1.1  christos #ifdef USE_SENDFILE
   2540      1.1  christos 		}
   2541      1.1  christos #endif
   2542      1.1  christos 	}
   2543      1.1  christos 
   2544      1.1  christos 	if (n > 0)
   2545      1.1  christos 		evbuffer_drain(buffer, n);
   2546      1.1  christos 
   2547      1.1  christos done:
   2548      1.1  christos 	EVBUFFER_UNLOCK(buffer);
   2549      1.1  christos 	return (n);
   2550      1.1  christos }
   2551      1.1  christos 
   2552      1.1  christos int
   2553      1.1  christos evbuffer_write(struct evbuffer *buffer, evutil_socket_t fd)
   2554      1.1  christos {
   2555      1.1  christos 	return evbuffer_write_atmost(buffer, fd, -1);
   2556      1.1  christos }
   2557      1.1  christos 
   2558      1.1  christos unsigned char *
   2559      1.1  christos evbuffer_find(struct evbuffer *buffer, const unsigned char *what, size_t len)
   2560      1.1  christos {
   2561      1.1  christos 	unsigned char *search;
   2562      1.1  christos 	struct evbuffer_ptr ptr;
   2563      1.1  christos 
   2564      1.1  christos 	EVBUFFER_LOCK(buffer);
   2565      1.1  christos 
   2566      1.1  christos 	ptr = evbuffer_search(buffer, (const char *)what, len, NULL);
   2567      1.1  christos 	if (ptr.pos < 0) {
   2568      1.1  christos 		search = NULL;
   2569      1.1  christos 	} else {
   2570      1.1  christos 		search = evbuffer_pullup(buffer, ptr.pos + len);
   2571      1.1  christos 		if (search)
   2572      1.1  christos 			search += ptr.pos;
   2573      1.1  christos 	}
   2574      1.1  christos 	EVBUFFER_UNLOCK(buffer);
   2575      1.1  christos 	return search;
   2576      1.1  christos }
   2577      1.1  christos 
   2578      1.1  christos /* Subract <b>howfar</b> from the position of <b>pos</b> within
   2579      1.1  christos  * <b>buf</b>. Returns 0 on success, -1 on failure.
   2580      1.1  christos  *
   2581      1.1  christos  * This isn't exposed yet, because of potential inefficiency issues.
   2582      1.1  christos  * Maybe it should be. */
   2583      1.1  christos static int
   2584      1.1  christos evbuffer_ptr_subtract(struct evbuffer *buf, struct evbuffer_ptr *pos,
   2585      1.1  christos     size_t howfar)
   2586      1.1  christos {
   2587  1.1.1.3  christos 	if (pos->pos < 0)
   2588  1.1.1.3  christos 		return -1;
   2589      1.1  christos 	if (howfar > (size_t)pos->pos)
   2590      1.1  christos 		return -1;
   2591      1.1  christos 	if (pos->internal_.chain && howfar <= pos->internal_.pos_in_chain) {
   2592      1.1  christos 		pos->internal_.pos_in_chain -= howfar;
   2593      1.1  christos 		pos->pos -= howfar;
   2594      1.1  christos 		return 0;
   2595      1.1  christos 	} else {
   2596      1.1  christos 		const size_t newpos = pos->pos - howfar;
   2597      1.1  christos 		/* Here's the inefficient part: it walks over the
   2598      1.1  christos 		 * chains until we hit newpos. */
   2599      1.1  christos 		return evbuffer_ptr_set(buf, pos, newpos, EVBUFFER_PTR_SET);
   2600      1.1  christos 	}
   2601      1.1  christos }
   2602      1.1  christos 
   2603      1.1  christos int
   2604      1.1  christos evbuffer_ptr_set(struct evbuffer *buf, struct evbuffer_ptr *pos,
   2605      1.1  christos     size_t position, enum evbuffer_ptr_how how)
   2606      1.1  christos {
   2607      1.1  christos 	size_t left = position;
   2608      1.1  christos 	struct evbuffer_chain *chain = NULL;
   2609      1.1  christos 	int result = 0;
   2610      1.1  christos 
   2611      1.1  christos 	EVBUFFER_LOCK(buf);
   2612      1.1  christos 
   2613      1.1  christos 	switch (how) {
   2614      1.1  christos 	case EVBUFFER_PTR_SET:
   2615      1.1  christos 		chain = buf->first;
   2616      1.1  christos 		pos->pos = position;
   2617      1.1  christos 		position = 0;
   2618      1.1  christos 		break;
   2619      1.1  christos 	case EVBUFFER_PTR_ADD:
   2620      1.1  christos 		/* this avoids iterating over all previous chains if
   2621      1.1  christos 		   we just want to advance the position */
   2622  1.1.1.3  christos 		if (pos->pos < 0 || EV_SIZE_MAX - position < (size_t)pos->pos) {
   2623  1.1.1.3  christos 			EVBUFFER_UNLOCK(buf);
   2624  1.1.1.3  christos 			return -1;
   2625  1.1.1.3  christos 		}
   2626      1.1  christos 		chain = pos->internal_.chain;
   2627      1.1  christos 		pos->pos += position;
   2628      1.1  christos 		position = pos->internal_.pos_in_chain;
   2629      1.1  christos 		break;
   2630      1.1  christos 	}
   2631      1.1  christos 
   2632  1.1.1.3  christos 	EVUTIL_ASSERT(EV_SIZE_MAX - left >= position);
   2633      1.1  christos 	while (chain && position + left >= chain->off) {
   2634      1.1  christos 		left -= chain->off - position;
   2635      1.1  christos 		chain = chain->next;
   2636      1.1  christos 		position = 0;
   2637      1.1  christos 	}
   2638      1.1  christos 	if (chain) {
   2639      1.1  christos 		pos->internal_.chain = chain;
   2640      1.1  christos 		pos->internal_.pos_in_chain = position + left;
   2641      1.1  christos 	} else if (left == 0) {
   2642      1.1  christos 		/* The first byte in the (nonexistent) chain after the last chain */
   2643      1.1  christos 		pos->internal_.chain = NULL;
   2644      1.1  christos 		pos->internal_.pos_in_chain = 0;
   2645      1.1  christos 	} else {
   2646      1.1  christos 		PTR_NOT_FOUND(pos);
   2647      1.1  christos 		result = -1;
   2648      1.1  christos 	}
   2649      1.1  christos 
   2650      1.1  christos 	EVBUFFER_UNLOCK(buf);
   2651      1.1  christos 
   2652      1.1  christos 	return result;
   2653      1.1  christos }
   2654      1.1  christos 
   2655      1.1  christos /**
   2656      1.1  christos    Compare the bytes in buf at position pos to the len bytes in mem.  Return
   2657      1.1  christos    less than 0, 0, or greater than 0 as memcmp.
   2658      1.1  christos  */
   2659      1.1  christos static int
   2660      1.1  christos evbuffer_ptr_memcmp(const struct evbuffer *buf, const struct evbuffer_ptr *pos,
   2661      1.1  christos     const char *mem, size_t len)
   2662      1.1  christos {
   2663      1.1  christos 	struct evbuffer_chain *chain;
   2664      1.1  christos 	size_t position;
   2665      1.1  christos 	int r;
   2666      1.1  christos 
   2667      1.1  christos 	ASSERT_EVBUFFER_LOCKED(buf);
   2668      1.1  christos 
   2669  1.1.1.3  christos 	if (pos->pos < 0 ||
   2670  1.1.1.3  christos 	    EV_SIZE_MAX - len < (size_t)pos->pos ||
   2671  1.1.1.3  christos 	    pos->pos + len > buf->total_len)
   2672      1.1  christos 		return -1;
   2673      1.1  christos 
   2674      1.1  christos 	chain = pos->internal_.chain;
   2675      1.1  christos 	position = pos->internal_.pos_in_chain;
   2676      1.1  christos 	while (len && chain) {
   2677      1.1  christos 		size_t n_comparable;
   2678      1.1  christos 		if (len + position > chain->off)
   2679      1.1  christos 			n_comparable = chain->off - position;
   2680      1.1  christos 		else
   2681      1.1  christos 			n_comparable = len;
   2682      1.1  christos 		r = memcmp(chain->buffer + chain->misalign + position, mem,
   2683      1.1  christos 		    n_comparable);
   2684      1.1  christos 		if (r)
   2685      1.1  christos 			return r;
   2686      1.1  christos 		mem += n_comparable;
   2687      1.1  christos 		len -= n_comparable;
   2688      1.1  christos 		position = 0;
   2689      1.1  christos 		chain = chain->next;
   2690      1.1  christos 	}
   2691      1.1  christos 
   2692      1.1  christos 	return 0;
   2693      1.1  christos }
   2694      1.1  christos 
   2695      1.1  christos struct evbuffer_ptr
   2696      1.1  christos evbuffer_search(struct evbuffer *buffer, const char *what, size_t len, const struct evbuffer_ptr *start)
   2697      1.1  christos {
   2698      1.1  christos 	return evbuffer_search_range(buffer, what, len, start, NULL);
   2699      1.1  christos }
   2700      1.1  christos 
   2701      1.1  christos struct evbuffer_ptr
   2702      1.1  christos evbuffer_search_range(struct evbuffer *buffer, const char *what, size_t len, const struct evbuffer_ptr *start, const struct evbuffer_ptr *end)
   2703      1.1  christos {
   2704      1.1  christos 	struct evbuffer_ptr pos;
   2705      1.1  christos 	struct evbuffer_chain *chain, *last_chain = NULL;
   2706      1.1  christos 	const unsigned char *p;
   2707      1.1  christos 	char first;
   2708      1.1  christos 
   2709      1.1  christos 	EVBUFFER_LOCK(buffer);
   2710      1.1  christos 
   2711      1.1  christos 	if (start) {
   2712      1.1  christos 		memcpy(&pos, start, sizeof(pos));
   2713      1.1  christos 		chain = pos.internal_.chain;
   2714      1.1  christos 	} else {
   2715      1.1  christos 		pos.pos = 0;
   2716      1.1  christos 		chain = pos.internal_.chain = buffer->first;
   2717      1.1  christos 		pos.internal_.pos_in_chain = 0;
   2718      1.1  christos 	}
   2719      1.1  christos 
   2720      1.1  christos 	if (end)
   2721      1.1  christos 		last_chain = end->internal_.chain;
   2722      1.1  christos 
   2723      1.1  christos 	if (!len || len > EV_SSIZE_MAX)
   2724      1.1  christos 		goto done;
   2725      1.1  christos 
   2726      1.1  christos 	first = what[0];
   2727      1.1  christos 
   2728      1.1  christos 	while (chain) {
   2729      1.1  christos 		const unsigned char *start_at =
   2730      1.1  christos 		    chain->buffer + chain->misalign +
   2731      1.1  christos 		    pos.internal_.pos_in_chain;
   2732      1.1  christos 		p = memchr(start_at, first,
   2733      1.1  christos 		    chain->off - pos.internal_.pos_in_chain);
   2734      1.1  christos 		if (p) {
   2735      1.1  christos 			pos.pos += p - start_at;
   2736      1.1  christos 			pos.internal_.pos_in_chain += p - start_at;
   2737      1.1  christos 			if (!evbuffer_ptr_memcmp(buffer, &pos, what, len)) {
   2738      1.1  christos 				if (end && pos.pos + (ev_ssize_t)len > end->pos)
   2739      1.1  christos 					goto not_found;
   2740      1.1  christos 				else
   2741      1.1  christos 					goto done;
   2742      1.1  christos 			}
   2743      1.1  christos 			++pos.pos;
   2744      1.1  christos 			++pos.internal_.pos_in_chain;
   2745      1.1  christos 			if (pos.internal_.pos_in_chain == chain->off) {
   2746      1.1  christos 				chain = pos.internal_.chain = chain->next;
   2747      1.1  christos 				pos.internal_.pos_in_chain = 0;
   2748      1.1  christos 			}
   2749      1.1  christos 		} else {
   2750      1.1  christos 			if (chain == last_chain)
   2751      1.1  christos 				goto not_found;
   2752      1.1  christos 			pos.pos += chain->off - pos.internal_.pos_in_chain;
   2753      1.1  christos 			chain = pos.internal_.chain = chain->next;
   2754      1.1  christos 			pos.internal_.pos_in_chain = 0;
   2755      1.1  christos 		}
   2756      1.1  christos 	}
   2757      1.1  christos 
   2758      1.1  christos not_found:
   2759      1.1  christos 	PTR_NOT_FOUND(&pos);
   2760      1.1  christos done:
   2761      1.1  christos 	EVBUFFER_UNLOCK(buffer);
   2762      1.1  christos 	return pos;
   2763      1.1  christos }
   2764      1.1  christos 
   2765      1.1  christos int
   2766      1.1  christos evbuffer_peek(struct evbuffer *buffer, ev_ssize_t len,
   2767      1.1  christos     struct evbuffer_ptr *start_at,
   2768      1.1  christos     struct evbuffer_iovec *vec, int n_vec)
   2769      1.1  christos {
   2770      1.1  christos 	struct evbuffer_chain *chain;
   2771      1.1  christos 	int idx = 0;
   2772      1.1  christos 	ev_ssize_t len_so_far = 0;
   2773      1.1  christos 
   2774      1.1  christos 	/* Avoid locking in trivial edge cases */
   2775      1.1  christos 	if (start_at && start_at->internal_.chain == NULL)
   2776      1.1  christos 		return 0;
   2777      1.1  christos 
   2778      1.1  christos 	EVBUFFER_LOCK(buffer);
   2779      1.1  christos 
   2780      1.1  christos 	if (start_at) {
   2781      1.1  christos 		chain = start_at->internal_.chain;
   2782      1.1  christos 		len_so_far = chain->off
   2783      1.1  christos 		    - start_at->internal_.pos_in_chain;
   2784      1.1  christos 		idx = 1;
   2785      1.1  christos 		if (n_vec > 0) {
   2786      1.1  christos 			vec[0].iov_base = chain->buffer + chain->misalign
   2787      1.1  christos 			    + start_at->internal_.pos_in_chain;
   2788      1.1  christos 			vec[0].iov_len = len_so_far;
   2789      1.1  christos 		}
   2790      1.1  christos 		chain = chain->next;
   2791      1.1  christos 	} else {
   2792      1.1  christos 		chain = buffer->first;
   2793      1.1  christos 	}
   2794      1.1  christos 
   2795      1.1  christos 	if (n_vec == 0 && len < 0) {
   2796      1.1  christos 		/* If no vectors are provided and they asked for "everything",
   2797      1.1  christos 		 * pretend they asked for the actual available amount. */
   2798  1.1.1.3  christos 		len = buffer->total_len;
   2799  1.1.1.3  christos 		if (start_at) {
   2800  1.1.1.3  christos 			len -= start_at->pos;
   2801  1.1.1.3  christos 		}
   2802      1.1  christos 	}
   2803      1.1  christos 
   2804      1.1  christos 	while (chain) {
   2805      1.1  christos 		if (len >= 0 && len_so_far >= len)
   2806      1.1  christos 			break;
   2807      1.1  christos 		if (idx<n_vec) {
   2808      1.1  christos 			vec[idx].iov_base = chain->buffer + chain->misalign;
   2809      1.1  christos 			vec[idx].iov_len = chain->off;
   2810      1.1  christos 		} else if (len<0) {
   2811      1.1  christos 			break;
   2812      1.1  christos 		}
   2813      1.1  christos 		++idx;
   2814      1.1  christos 		len_so_far += chain->off;
   2815      1.1  christos 		chain = chain->next;
   2816      1.1  christos 	}
   2817      1.1  christos 
   2818      1.1  christos 	EVBUFFER_UNLOCK(buffer);
   2819      1.1  christos 
   2820      1.1  christos 	return idx;
   2821      1.1  christos }
   2822      1.1  christos 
   2823      1.1  christos 
   2824      1.1  christos int
   2825      1.1  christos evbuffer_add_vprintf(struct evbuffer *buf, const char *fmt, va_list ap)
   2826      1.1  christos {
   2827      1.1  christos 	char *buffer;
   2828      1.1  christos 	size_t space;
   2829      1.1  christos 	int sz, result = -1;
   2830      1.1  christos 	va_list aq;
   2831      1.1  christos 	struct evbuffer_chain *chain;
   2832      1.1  christos 
   2833      1.1  christos 
   2834      1.1  christos 	EVBUFFER_LOCK(buf);
   2835      1.1  christos 
   2836      1.1  christos 	if (buf->freeze_end) {
   2837      1.1  christos 		goto done;
   2838      1.1  christos 	}
   2839      1.1  christos 
   2840      1.1  christos 	/* make sure that at least some space is available */
   2841      1.1  christos 	if ((chain = evbuffer_expand_singlechain(buf, 64)) == NULL)
   2842      1.1  christos 		goto done;
   2843      1.1  christos 
   2844      1.1  christos 	for (;;) {
   2845      1.1  christos #if 0
   2846      1.1  christos 		size_t used = chain->misalign + chain->off;
   2847      1.1  christos 		buffer = (char *)chain->buffer + chain->misalign + chain->off;
   2848      1.1  christos 		EVUTIL_ASSERT(chain->buffer_len >= used);
   2849      1.1  christos 		space = chain->buffer_len - used;
   2850      1.1  christos #endif
   2851      1.1  christos 		buffer = (char*) CHAIN_SPACE_PTR(chain);
   2852      1.1  christos 		space = (size_t) CHAIN_SPACE_LEN(chain);
   2853      1.1  christos 
   2854      1.1  christos #ifndef va_copy
   2855      1.1  christos #define	va_copy(dst, src)	memcpy(&(dst), &(src), sizeof(va_list))
   2856      1.1  christos #endif
   2857      1.1  christos 		va_copy(aq, ap);
   2858      1.1  christos 
   2859      1.1  christos 		sz = evutil_vsnprintf(buffer, space, fmt, aq);
   2860      1.1  christos 
   2861      1.1  christos 		va_end(aq);
   2862      1.1  christos 
   2863      1.1  christos 		if (sz < 0)
   2864      1.1  christos 			goto done;
   2865  1.1.1.3  christos 		if (INT_MAX >= EVBUFFER_CHAIN_MAX &&
   2866  1.1.1.3  christos 		    (size_t)sz >= EVBUFFER_CHAIN_MAX)
   2867  1.1.1.3  christos 			goto done;
   2868      1.1  christos 		if ((size_t)sz < space) {
   2869      1.1  christos 			chain->off += sz;
   2870      1.1  christos 			buf->total_len += sz;
   2871      1.1  christos 			buf->n_add_for_cb += sz;
   2872      1.1  christos 
   2873      1.1  christos 			advance_last_with_data(buf);
   2874      1.1  christos 			evbuffer_invoke_callbacks_(buf);
   2875      1.1  christos 			result = sz;
   2876      1.1  christos 			goto done;
   2877      1.1  christos 		}
   2878      1.1  christos 		if ((chain = evbuffer_expand_singlechain(buf, sz + 1)) == NULL)
   2879      1.1  christos 			goto done;
   2880      1.1  christos 	}
   2881      1.1  christos 	/* NOTREACHED */
   2882      1.1  christos 
   2883      1.1  christos done:
   2884      1.1  christos 	EVBUFFER_UNLOCK(buf);
   2885      1.1  christos 	return result;
   2886      1.1  christos }
   2887      1.1  christos 
   2888      1.1  christos int
   2889      1.1  christos evbuffer_add_printf(struct evbuffer *buf, const char *fmt, ...)
   2890      1.1  christos {
   2891      1.1  christos 	int res = -1;
   2892      1.1  christos 	va_list ap;
   2893      1.1  christos 
   2894      1.1  christos 	va_start(ap, fmt);
   2895      1.1  christos 	res = evbuffer_add_vprintf(buf, fmt, ap);
   2896      1.1  christos 	va_end(ap);
   2897      1.1  christos 
   2898      1.1  christos 	return (res);
   2899      1.1  christos }
   2900      1.1  christos 
   2901      1.1  christos int
   2902      1.1  christos evbuffer_add_reference(struct evbuffer *outbuf,
   2903      1.1  christos     const void *data, size_t datlen,
   2904      1.1  christos     evbuffer_ref_cleanup_cb cleanupfn, void *extra)
   2905      1.1  christos {
   2906      1.1  christos 	struct evbuffer_chain *chain;
   2907      1.1  christos 	struct evbuffer_chain_reference *info;
   2908      1.1  christos 	int result = -1;
   2909      1.1  christos 
   2910      1.1  christos 	chain = evbuffer_chain_new(sizeof(struct evbuffer_chain_reference));
   2911      1.1  christos 	if (!chain)
   2912      1.1  christos 		return (-1);
   2913      1.1  christos 	chain->flags |= EVBUFFER_REFERENCE | EVBUFFER_IMMUTABLE;
   2914      1.1  christos 	chain->buffer = (u_char *)data;
   2915      1.1  christos 	chain->buffer_len = datlen;
   2916      1.1  christos 	chain->off = datlen;
   2917      1.1  christos 
   2918      1.1  christos 	info = EVBUFFER_CHAIN_EXTRA(struct evbuffer_chain_reference, chain);
   2919      1.1  christos 	info->cleanupfn = cleanupfn;
   2920      1.1  christos 	info->extra = extra;
   2921      1.1  christos 
   2922      1.1  christos 	EVBUFFER_LOCK(outbuf);
   2923      1.1  christos 	if (outbuf->freeze_end) {
   2924      1.1  christos 		/* don't call chain_free; we do not want to actually invoke
   2925      1.1  christos 		 * the cleanup function */
   2926      1.1  christos 		mm_free(chain);
   2927      1.1  christos 		goto done;
   2928      1.1  christos 	}
   2929      1.1  christos 	evbuffer_chain_insert(outbuf, chain);
   2930      1.1  christos 	outbuf->n_add_for_cb += datlen;
   2931      1.1  christos 
   2932      1.1  christos 	evbuffer_invoke_callbacks_(outbuf);
   2933      1.1  christos 
   2934      1.1  christos 	result = 0;
   2935      1.1  christos done:
   2936      1.1  christos 	EVBUFFER_UNLOCK(outbuf);
   2937      1.1  christos 
   2938      1.1  christos 	return result;
   2939      1.1  christos }
   2940      1.1  christos 
   2941      1.1  christos /* TODO(niels): we may want to add to automagically convert to mmap, in
   2942      1.1  christos  * case evbuffer_remove() or evbuffer_pullup() are being used.
   2943      1.1  christos  */
   2944      1.1  christos struct evbuffer_file_segment *
   2945      1.1  christos evbuffer_file_segment_new(
   2946      1.1  christos 	int fd, ev_off_t offset, ev_off_t length, unsigned flags)
   2947      1.1  christos {
   2948      1.1  christos 	struct evbuffer_file_segment *seg =
   2949      1.1  christos 	    mm_calloc(sizeof(struct evbuffer_file_segment), 1);
   2950      1.1  christos 	if (!seg)
   2951      1.1  christos 		return NULL;
   2952      1.1  christos 	seg->refcnt = 1;
   2953      1.1  christos 	seg->fd = fd;
   2954      1.1  christos 	seg->flags = flags;
   2955      1.1  christos 	seg->file_offset = offset;
   2956      1.1  christos 	seg->cleanup_cb = NULL;
   2957      1.1  christos 	seg->cleanup_cb_arg = NULL;
   2958      1.1  christos #ifdef _WIN32
   2959      1.1  christos #ifndef lseek
   2960      1.1  christos #define lseek _lseeki64
   2961      1.1  christos #endif
   2962      1.1  christos #ifndef fstat
   2963      1.1  christos #define fstat _fstat
   2964      1.1  christos #endif
   2965      1.1  christos #ifndef stat
   2966      1.1  christos #define stat _stat
   2967      1.1  christos #endif
   2968      1.1  christos #endif
   2969      1.1  christos 	if (length == -1) {
   2970      1.1  christos 		struct stat st;
   2971      1.1  christos 		if (fstat(fd, &st) < 0)
   2972      1.1  christos 			goto err;
   2973      1.1  christos 		length = st.st_size;
   2974      1.1  christos 	}
   2975      1.1  christos 	seg->length = length;
   2976      1.1  christos 
   2977  1.1.1.3  christos 	if (offset < 0 || length < 0 ||
   2978  1.1.1.3  christos 	    ((ev_uint64_t)length > EVBUFFER_CHAIN_MAX) ||
   2979  1.1.1.3  christos 	    (ev_uint64_t)offset > (ev_uint64_t)(EVBUFFER_CHAIN_MAX - length))
   2980  1.1.1.3  christos 		goto err;
   2981  1.1.1.3  christos 
   2982      1.1  christos #if defined(USE_SENDFILE)
   2983      1.1  christos 	if (!(flags & EVBUF_FS_DISABLE_SENDFILE)) {
   2984      1.1  christos 		seg->can_sendfile = 1;
   2985      1.1  christos 		goto done;
   2986      1.1  christos 	}
   2987      1.1  christos #endif
   2988      1.1  christos 
   2989      1.1  christos 	if (evbuffer_file_segment_materialize(seg)<0)
   2990      1.1  christos 		goto err;
   2991      1.1  christos 
   2992      1.1  christos #if defined(USE_SENDFILE)
   2993      1.1  christos done:
   2994      1.1  christos #endif
   2995      1.1  christos 	if (!(flags & EVBUF_FS_DISABLE_LOCKING)) {
   2996      1.1  christos 		EVTHREAD_ALLOC_LOCK(seg->lock, 0);
   2997      1.1  christos 	}
   2998      1.1  christos 	return seg;
   2999      1.1  christos err:
   3000      1.1  christos 	mm_free(seg);
   3001      1.1  christos 	return NULL;
   3002      1.1  christos }
   3003      1.1  christos 
   3004  1.1.1.2  christos #ifdef EVENT__HAVE_MMAP
   3005  1.1.1.2  christos static long
   3006  1.1.1.2  christos get_page_size(void)
   3007  1.1.1.2  christos {
   3008  1.1.1.2  christos #ifdef SC_PAGE_SIZE
   3009  1.1.1.2  christos 	return sysconf(SC_PAGE_SIZE);
   3010  1.1.1.2  christos #elif defined(_SC_PAGE_SIZE)
   3011  1.1.1.2  christos 	return sysconf(_SC_PAGE_SIZE);
   3012  1.1.1.2  christos #else
   3013  1.1.1.2  christos 	return 1;
   3014  1.1.1.2  christos #endif
   3015  1.1.1.2  christos }
   3016  1.1.1.2  christos #endif
   3017  1.1.1.2  christos 
   3018      1.1  christos /* DOCDOC */
   3019      1.1  christos /* Requires lock */
   3020      1.1  christos static int
   3021      1.1  christos evbuffer_file_segment_materialize(struct evbuffer_file_segment *seg)
   3022      1.1  christos {
   3023      1.1  christos 	const unsigned flags = seg->flags;
   3024      1.1  christos 	const int fd = seg->fd;
   3025      1.1  christos 	const ev_off_t length = seg->length;
   3026      1.1  christos 	const ev_off_t offset = seg->file_offset;
   3027      1.1  christos 
   3028      1.1  christos 	if (seg->contents)
   3029      1.1  christos 		return 0; /* already materialized */
   3030      1.1  christos 
   3031      1.1  christos #if defined(EVENT__HAVE_MMAP)
   3032      1.1  christos 	if (!(flags & EVBUF_FS_DISABLE_MMAP)) {
   3033      1.1  christos 		off_t offset_rounded = 0, offset_leftover = 0;
   3034      1.1  christos 		void *mapped;
   3035      1.1  christos 		if (offset) {
   3036      1.1  christos 			/* mmap implementations don't generally like us
   3037      1.1  christos 			 * to have an offset that isn't a round  */
   3038  1.1.1.2  christos 			long page_size = get_page_size();
   3039      1.1  christos 			if (page_size == -1)
   3040      1.1  christos 				goto err;
   3041      1.1  christos 			offset_leftover = offset % page_size;
   3042      1.1  christos 			offset_rounded = offset - offset_leftover;
   3043      1.1  christos 		}
   3044      1.1  christos 		mapped = mmap(NULL, length + offset_leftover,
   3045      1.1  christos 		    PROT_READ,
   3046      1.1  christos #ifdef MAP_NOCACHE
   3047      1.1  christos 		    MAP_NOCACHE | /* ??? */
   3048      1.1  christos #endif
   3049      1.1  christos #ifdef MAP_FILE
   3050      1.1  christos 		    MAP_FILE |
   3051      1.1  christos #endif
   3052      1.1  christos 		    MAP_PRIVATE,
   3053      1.1  christos 		    fd, offset_rounded);
   3054      1.1  christos 		if (mapped == MAP_FAILED) {
   3055      1.1  christos 			event_warn("%s: mmap(%d, %d, %zu) failed",
   3056      1.1  christos 			    __func__, fd, 0, (size_t)(offset + length));
   3057      1.1  christos 		} else {
   3058      1.1  christos 			seg->mapping = mapped;
   3059      1.1  christos 			seg->contents = (char*)mapped+offset_leftover;
   3060      1.1  christos 			seg->mmap_offset = 0;
   3061      1.1  christos 			seg->is_mapping = 1;
   3062      1.1  christos 			goto done;
   3063      1.1  christos 		}
   3064      1.1  christos 	}
   3065      1.1  christos #endif
   3066      1.1  christos #ifdef _WIN32
   3067      1.1  christos 	if (!(flags & EVBUF_FS_DISABLE_MMAP)) {
   3068      1.1  christos 		intptr_t h = _get_osfhandle(fd);
   3069      1.1  christos 		HANDLE m;
   3070      1.1  christos 		ev_uint64_t total_size = length+offset;
   3071      1.1  christos 		if ((HANDLE)h == INVALID_HANDLE_VALUE)
   3072      1.1  christos 			goto err;
   3073      1.1  christos 		m = CreateFileMapping((HANDLE)h, NULL, PAGE_READONLY,
   3074      1.1  christos 		    (total_size >> 32), total_size & 0xfffffffful,
   3075      1.1  christos 		    NULL);
   3076      1.1  christos 		if (m != INVALID_HANDLE_VALUE) { /* Does h leak? */
   3077      1.1  christos 			seg->mapping_handle = m;
   3078      1.1  christos 			seg->mmap_offset = offset;
   3079      1.1  christos 			seg->is_mapping = 1;
   3080      1.1  christos 			goto done;
   3081      1.1  christos 		}
   3082      1.1  christos 	}
   3083      1.1  christos #endif
   3084      1.1  christos 	{
   3085      1.1  christos 		ev_off_t start_pos = lseek(fd, 0, SEEK_CUR), pos;
   3086      1.1  christos 		ev_off_t read_so_far = 0;
   3087      1.1  christos 		char *mem;
   3088      1.1  christos 		int e;
   3089      1.1  christos 		ev_ssize_t n = 0;
   3090      1.1  christos 		if (!(mem = mm_malloc(length)))
   3091      1.1  christos 			goto err;
   3092      1.1  christos 		if (start_pos < 0) {
   3093      1.1  christos 			mm_free(mem);
   3094      1.1  christos 			goto err;
   3095      1.1  christos 		}
   3096      1.1  christos 		if (lseek(fd, offset, SEEK_SET) < 0) {
   3097      1.1  christos 			mm_free(mem);
   3098      1.1  christos 			goto err;
   3099      1.1  christos 		}
   3100      1.1  christos 		while (read_so_far < length) {
   3101      1.1  christos 			n = read(fd, mem+read_so_far, length-read_so_far);
   3102      1.1  christos 			if (n <= 0)
   3103      1.1  christos 				break;
   3104      1.1  christos 			read_so_far += n;
   3105      1.1  christos 		}
   3106      1.1  christos 
   3107      1.1  christos 		e = errno;
   3108      1.1  christos 		pos = lseek(fd, start_pos, SEEK_SET);
   3109      1.1  christos 		if (n < 0 || (n == 0 && length > read_so_far)) {
   3110      1.1  christos 			mm_free(mem);
   3111      1.1  christos 			errno = e;
   3112      1.1  christos 			goto err;
   3113      1.1  christos 		} else if (pos < 0) {
   3114      1.1  christos 			mm_free(mem);
   3115      1.1  christos 			goto err;
   3116      1.1  christos 		}
   3117      1.1  christos 
   3118      1.1  christos 		seg->contents = mem;
   3119      1.1  christos 	}
   3120      1.1  christos 
   3121      1.1  christos done:
   3122      1.1  christos 	return 0;
   3123      1.1  christos err:
   3124      1.1  christos 	return -1;
   3125      1.1  christos }
   3126      1.1  christos 
   3127      1.1  christos void evbuffer_file_segment_add_cleanup_cb(struct evbuffer_file_segment *seg,
   3128      1.1  christos 	evbuffer_file_segment_cleanup_cb cb, void* arg)
   3129      1.1  christos {
   3130      1.1  christos 	EVUTIL_ASSERT(seg->refcnt > 0);
   3131      1.1  christos 	seg->cleanup_cb = cb;
   3132      1.1  christos 	seg->cleanup_cb_arg = arg;
   3133      1.1  christos }
   3134      1.1  christos 
   3135      1.1  christos void
   3136      1.1  christos evbuffer_file_segment_free(struct evbuffer_file_segment *seg)
   3137      1.1  christos {
   3138      1.1  christos 	int refcnt;
   3139      1.1  christos 	EVLOCK_LOCK(seg->lock, 0);
   3140      1.1  christos 	refcnt = --seg->refcnt;
   3141      1.1  christos 	EVLOCK_UNLOCK(seg->lock, 0);
   3142      1.1  christos 	if (refcnt > 0)
   3143      1.1  christos 		return;
   3144      1.1  christos 	EVUTIL_ASSERT(refcnt == 0);
   3145      1.1  christos 
   3146      1.1  christos 	if (seg->is_mapping) {
   3147      1.1  christos #ifdef _WIN32
   3148      1.1  christos 		CloseHandle(seg->mapping_handle);
   3149      1.1  christos #elif defined (EVENT__HAVE_MMAP)
   3150  1.1.1.2  christos 		off_t offset_leftover;
   3151  1.1.1.2  christos 		offset_leftover = seg->file_offset % get_page_size();
   3152  1.1.1.2  christos 		if (munmap(seg->mapping, seg->length + offset_leftover) == -1)
   3153      1.1  christos 			event_warn("%s: munmap failed", __func__);
   3154      1.1  christos #endif
   3155      1.1  christos 	} else if (seg->contents) {
   3156      1.1  christos 		mm_free(seg->contents);
   3157      1.1  christos 	}
   3158      1.1  christos 
   3159      1.1  christos 	if ((seg->flags & EVBUF_FS_CLOSE_ON_FREE) && seg->fd >= 0) {
   3160      1.1  christos 		close(seg->fd);
   3161      1.1  christos 	}
   3162      1.1  christos 
   3163      1.1  christos 	if (seg->cleanup_cb) {
   3164      1.1  christos 		(*seg->cleanup_cb)((struct evbuffer_file_segment const*)seg,
   3165      1.1  christos 		    seg->flags, seg->cleanup_cb_arg);
   3166      1.1  christos 		seg->cleanup_cb = NULL;
   3167      1.1  christos 		seg->cleanup_cb_arg = NULL;
   3168      1.1  christos 	}
   3169      1.1  christos 
   3170      1.1  christos 	EVTHREAD_FREE_LOCK(seg->lock, 0);
   3171      1.1  christos 	mm_free(seg);
   3172      1.1  christos }
   3173      1.1  christos 
   3174      1.1  christos int
   3175      1.1  christos evbuffer_add_file_segment(struct evbuffer *buf,
   3176      1.1  christos     struct evbuffer_file_segment *seg, ev_off_t offset, ev_off_t length)
   3177      1.1  christos {
   3178      1.1  christos 	struct evbuffer_chain *chain;
   3179      1.1  christos 	struct evbuffer_chain_file_segment *extra;
   3180      1.1  christos 	int can_use_sendfile = 0;
   3181      1.1  christos 
   3182      1.1  christos 	EVBUFFER_LOCK(buf);
   3183      1.1  christos 	EVLOCK_LOCK(seg->lock, 0);
   3184      1.1  christos 	if (buf->flags & EVBUFFER_FLAG_DRAINS_TO_FD) {
   3185      1.1  christos 		can_use_sendfile = 1;
   3186      1.1  christos 	} else {
   3187      1.1  christos 		if (!seg->contents) {
   3188      1.1  christos 			if (evbuffer_file_segment_materialize(seg)<0) {
   3189      1.1  christos 				EVLOCK_UNLOCK(seg->lock, 0);
   3190      1.1  christos 				EVBUFFER_UNLOCK(buf);
   3191      1.1  christos 				return -1;
   3192      1.1  christos 			}
   3193      1.1  christos 		}
   3194      1.1  christos 	}
   3195      1.1  christos 	++seg->refcnt;
   3196      1.1  christos 	EVLOCK_UNLOCK(seg->lock, 0);
   3197      1.1  christos 
   3198      1.1  christos 	if (buf->freeze_end)
   3199      1.1  christos 		goto err;
   3200      1.1  christos 
   3201      1.1  christos 	if (length < 0) {
   3202      1.1  christos 		if (offset > seg->length)
   3203      1.1  christos 			goto err;
   3204      1.1  christos 		length = seg->length - offset;
   3205      1.1  christos 	}
   3206      1.1  christos 
   3207      1.1  christos 	/* Can we actually add this? */
   3208      1.1  christos 	if (offset+length > seg->length)
   3209      1.1  christos 		goto err;
   3210      1.1  christos 
   3211      1.1  christos 	chain = evbuffer_chain_new(sizeof(struct evbuffer_chain_file_segment));
   3212      1.1  christos 	if (!chain)
   3213      1.1  christos 		goto err;
   3214      1.1  christos 	extra = EVBUFFER_CHAIN_EXTRA(struct evbuffer_chain_file_segment, chain);
   3215      1.1  christos 
   3216      1.1  christos 	chain->flags |= EVBUFFER_IMMUTABLE|EVBUFFER_FILESEGMENT;
   3217      1.1  christos 	if (can_use_sendfile && seg->can_sendfile) {
   3218      1.1  christos 		chain->flags |= EVBUFFER_SENDFILE;
   3219      1.1  christos 		chain->misalign = seg->file_offset + offset;
   3220      1.1  christos 		chain->off = length;
   3221      1.1  christos 		chain->buffer_len = chain->misalign + length;
   3222      1.1  christos 	} else if (seg->is_mapping) {
   3223      1.1  christos #ifdef _WIN32
   3224      1.1  christos 		ev_uint64_t total_offset = seg->mmap_offset+offset;
   3225      1.1  christos 		ev_uint64_t offset_rounded=0, offset_remaining=0;
   3226      1.1  christos 		LPVOID data;
   3227      1.1  christos 		if (total_offset) {
   3228      1.1  christos 			SYSTEM_INFO si;
   3229      1.1  christos 			memset(&si, 0, sizeof(si)); /* cargo cult */
   3230      1.1  christos 			GetSystemInfo(&si);
   3231      1.1  christos 			offset_remaining = total_offset % si.dwAllocationGranularity;
   3232      1.1  christos 			offset_rounded = total_offset - offset_remaining;
   3233      1.1  christos 		}
   3234      1.1  christos 		data = MapViewOfFile(
   3235      1.1  christos 			seg->mapping_handle,
   3236      1.1  christos 			FILE_MAP_READ,
   3237      1.1  christos 			offset_rounded >> 32,
   3238      1.1  christos 			offset_rounded & 0xfffffffful,
   3239      1.1  christos 			length + offset_remaining);
   3240      1.1  christos 		if (data == NULL) {
   3241      1.1  christos 			mm_free(chain);
   3242      1.1  christos 			goto err;
   3243      1.1  christos 		}
   3244      1.1  christos 		chain->buffer = (unsigned char*) data;
   3245      1.1  christos 		chain->buffer_len = length+offset_remaining;
   3246      1.1  christos 		chain->misalign = offset_remaining;
   3247      1.1  christos 		chain->off = length;
   3248      1.1  christos #else
   3249      1.1  christos 		chain->buffer = (unsigned char*)(seg->contents + offset);
   3250      1.1  christos 		chain->buffer_len = length;
   3251      1.1  christos 		chain->off = length;
   3252      1.1  christos #endif
   3253      1.1  christos 	} else {
   3254      1.1  christos 		chain->buffer = (unsigned char*)(seg->contents + offset);
   3255      1.1  christos 		chain->buffer_len = length;
   3256      1.1  christos 		chain->off = length;
   3257      1.1  christos 	}
   3258      1.1  christos 
   3259      1.1  christos 	extra->segment = seg;
   3260      1.1  christos 	buf->n_add_for_cb += length;
   3261      1.1  christos 	evbuffer_chain_insert(buf, chain);
   3262      1.1  christos 
   3263      1.1  christos 	evbuffer_invoke_callbacks_(buf);
   3264      1.1  christos 
   3265      1.1  christos 	EVBUFFER_UNLOCK(buf);
   3266      1.1  christos 
   3267      1.1  christos 	return 0;
   3268      1.1  christos err:
   3269      1.1  christos 	EVBUFFER_UNLOCK(buf);
   3270  1.1.1.3  christos 	evbuffer_file_segment_free(seg); /* Lowers the refcount */
   3271      1.1  christos 	return -1;
   3272      1.1  christos }
   3273      1.1  christos 
   3274      1.1  christos int
   3275      1.1  christos evbuffer_add_file(struct evbuffer *buf, int fd, ev_off_t offset, ev_off_t length)
   3276      1.1  christos {
   3277      1.1  christos 	struct evbuffer_file_segment *seg;
   3278      1.1  christos 	unsigned flags = EVBUF_FS_CLOSE_ON_FREE;
   3279      1.1  christos 	int r;
   3280      1.1  christos 
   3281      1.1  christos 	seg = evbuffer_file_segment_new(fd, offset, length, flags);
   3282      1.1  christos 	if (!seg)
   3283      1.1  christos 		return -1;
   3284      1.1  christos 	r = evbuffer_add_file_segment(buf, seg, 0, length);
   3285      1.1  christos 	if (r == 0)
   3286      1.1  christos 		evbuffer_file_segment_free(seg);
   3287      1.1  christos 	return r;
   3288      1.1  christos }
   3289      1.1  christos 
   3290      1.1  christos void
   3291      1.1  christos evbuffer_setcb(struct evbuffer *buffer, evbuffer_cb cb, void *cbarg)
   3292      1.1  christos {
   3293      1.1  christos 	EVBUFFER_LOCK(buffer);
   3294      1.1  christos 
   3295      1.1  christos 	if (!LIST_EMPTY(&buffer->callbacks))
   3296      1.1  christos 		evbuffer_remove_all_callbacks(buffer);
   3297      1.1  christos 
   3298      1.1  christos 	if (cb) {
   3299      1.1  christos 		struct evbuffer_cb_entry *ent =
   3300      1.1  christos 		    evbuffer_add_cb(buffer, NULL, cbarg);
   3301      1.1  christos 		ent->cb.cb_obsolete = cb;
   3302      1.1  christos 		ent->flags |= EVBUFFER_CB_OBSOLETE;
   3303      1.1  christos 	}
   3304      1.1  christos 	EVBUFFER_UNLOCK(buffer);
   3305      1.1  christos }
   3306      1.1  christos 
   3307      1.1  christos struct evbuffer_cb_entry *
   3308      1.1  christos evbuffer_add_cb(struct evbuffer *buffer, evbuffer_cb_func cb, void *cbarg)
   3309      1.1  christos {
   3310      1.1  christos 	struct evbuffer_cb_entry *e;
   3311      1.1  christos 	if (! (e = mm_calloc(1, sizeof(struct evbuffer_cb_entry))))
   3312      1.1  christos 		return NULL;
   3313      1.1  christos 	EVBUFFER_LOCK(buffer);
   3314      1.1  christos 	e->cb.cb_func = cb;
   3315      1.1  christos 	e->cbarg = cbarg;
   3316      1.1  christos 	e->flags = EVBUFFER_CB_ENABLED;
   3317      1.1  christos 	LIST_INSERT_HEAD(&buffer->callbacks, e, next);
   3318      1.1  christos 	EVBUFFER_UNLOCK(buffer);
   3319      1.1  christos 	return e;
   3320      1.1  christos }
   3321      1.1  christos 
   3322      1.1  christos int
   3323      1.1  christos evbuffer_remove_cb_entry(struct evbuffer *buffer,
   3324      1.1  christos 			 struct evbuffer_cb_entry *ent)
   3325      1.1  christos {
   3326      1.1  christos 	EVBUFFER_LOCK(buffer);
   3327      1.1  christos 	LIST_REMOVE(ent, next);
   3328      1.1  christos 	EVBUFFER_UNLOCK(buffer);
   3329      1.1  christos 	mm_free(ent);
   3330      1.1  christos 	return 0;
   3331      1.1  christos }
   3332      1.1  christos 
   3333      1.1  christos int
   3334      1.1  christos evbuffer_remove_cb(struct evbuffer *buffer, evbuffer_cb_func cb, void *cbarg)
   3335      1.1  christos {
   3336      1.1  christos 	struct evbuffer_cb_entry *cbent;
   3337      1.1  christos 	int result = -1;
   3338      1.1  christos 	EVBUFFER_LOCK(buffer);
   3339      1.1  christos 	LIST_FOREACH(cbent, &buffer->callbacks, next) {
   3340      1.1  christos 		if (cb == cbent->cb.cb_func && cbarg == cbent->cbarg) {
   3341      1.1  christos 			result = evbuffer_remove_cb_entry(buffer, cbent);
   3342      1.1  christos 			goto done;
   3343      1.1  christos 		}
   3344      1.1  christos 	}
   3345      1.1  christos done:
   3346      1.1  christos 	EVBUFFER_UNLOCK(buffer);
   3347      1.1  christos 	return result;
   3348      1.1  christos }
   3349      1.1  christos 
   3350      1.1  christos int
   3351      1.1  christos evbuffer_cb_set_flags(struct evbuffer *buffer,
   3352      1.1  christos 		      struct evbuffer_cb_entry *cb, ev_uint32_t flags)
   3353      1.1  christos {
   3354      1.1  christos 	/* the user isn't allowed to mess with these. */
   3355      1.1  christos 	flags &= ~EVBUFFER_CB_INTERNAL_FLAGS;
   3356      1.1  christos 	EVBUFFER_LOCK(buffer);
   3357      1.1  christos 	cb->flags |= flags;
   3358      1.1  christos 	EVBUFFER_UNLOCK(buffer);
   3359      1.1  christos 	return 0;
   3360      1.1  christos }
   3361      1.1  christos 
   3362      1.1  christos int
   3363      1.1  christos evbuffer_cb_clear_flags(struct evbuffer *buffer,
   3364      1.1  christos 		      struct evbuffer_cb_entry *cb, ev_uint32_t flags)
   3365      1.1  christos {
   3366      1.1  christos 	/* the user isn't allowed to mess with these. */
   3367      1.1  christos 	flags &= ~EVBUFFER_CB_INTERNAL_FLAGS;
   3368      1.1  christos 	EVBUFFER_LOCK(buffer);
   3369      1.1  christos 	cb->flags &= ~flags;
   3370      1.1  christos 	EVBUFFER_UNLOCK(buffer);
   3371      1.1  christos 	return 0;
   3372      1.1  christos }
   3373      1.1  christos 
   3374      1.1  christos int
   3375      1.1  christos evbuffer_freeze(struct evbuffer *buffer, int start)
   3376      1.1  christos {
   3377      1.1  christos 	EVBUFFER_LOCK(buffer);
   3378      1.1  christos 	if (start)
   3379      1.1  christos 		buffer->freeze_start = 1;
   3380      1.1  christos 	else
   3381      1.1  christos 		buffer->freeze_end = 1;
   3382      1.1  christos 	EVBUFFER_UNLOCK(buffer);
   3383      1.1  christos 	return 0;
   3384      1.1  christos }
   3385      1.1  christos 
   3386      1.1  christos int
   3387      1.1  christos evbuffer_unfreeze(struct evbuffer *buffer, int start)
   3388      1.1  christos {
   3389      1.1  christos 	EVBUFFER_LOCK(buffer);
   3390      1.1  christos 	if (start)
   3391      1.1  christos 		buffer->freeze_start = 0;
   3392      1.1  christos 	else
   3393      1.1  christos 		buffer->freeze_end = 0;
   3394      1.1  christos 	EVBUFFER_UNLOCK(buffer);
   3395      1.1  christos 	return 0;
   3396      1.1  christos }
   3397      1.1  christos 
   3398      1.1  christos #if 0
   3399      1.1  christos void
   3400      1.1  christos evbuffer_cb_suspend(struct evbuffer *buffer, struct evbuffer_cb_entry *cb)
   3401      1.1  christos {
   3402      1.1  christos 	if (!(cb->flags & EVBUFFER_CB_SUSPENDED)) {
   3403      1.1  christos 		cb->size_before_suspend = evbuffer_get_length(buffer);
   3404      1.1  christos 		cb->flags |= EVBUFFER_CB_SUSPENDED;
   3405      1.1  christos 	}
   3406      1.1  christos }
   3407      1.1  christos 
   3408      1.1  christos void
   3409      1.1  christos evbuffer_cb_unsuspend(struct evbuffer *buffer, struct evbuffer_cb_entry *cb)
   3410      1.1  christos {
   3411      1.1  christos 	if ((cb->flags & EVBUFFER_CB_SUSPENDED)) {
   3412      1.1  christos 		unsigned call = (cb->flags & EVBUFFER_CB_CALL_ON_UNSUSPEND);
   3413      1.1  christos 		size_t sz = cb->size_before_suspend;
   3414      1.1  christos 		cb->flags &= ~(EVBUFFER_CB_SUSPENDED|
   3415      1.1  christos 			       EVBUFFER_CB_CALL_ON_UNSUSPEND);
   3416      1.1  christos 		cb->size_before_suspend = 0;
   3417      1.1  christos 		if (call && (cb->flags & EVBUFFER_CB_ENABLED)) {
   3418      1.1  christos 			cb->cb(buffer, sz, evbuffer_get_length(buffer), cb->cbarg);
   3419      1.1  christos 		}
   3420      1.1  christos 	}
   3421      1.1  christos }
   3422      1.1  christos #endif
   3423      1.1  christos 
   3424  1.1.1.2  christos int
   3425  1.1.1.2  christos evbuffer_get_callbacks_(struct evbuffer *buffer, struct event_callback **cbs,
   3426  1.1.1.2  christos     int max_cbs)
   3427  1.1.1.2  christos {
   3428  1.1.1.2  christos 	int r = 0;
   3429  1.1.1.2  christos 	EVBUFFER_LOCK(buffer);
   3430  1.1.1.2  christos 	if (buffer->deferred_cbs) {
   3431  1.1.1.2  christos 		if (max_cbs < 1) {
   3432  1.1.1.2  christos 			r = -1;
   3433  1.1.1.2  christos 			goto done;
   3434  1.1.1.2  christos 		}
   3435  1.1.1.2  christos 		cbs[0] = &buffer->deferred;
   3436  1.1.1.2  christos 		r = 1;
   3437  1.1.1.2  christos 	}
   3438  1.1.1.2  christos done:
   3439  1.1.1.2  christos 	EVBUFFER_UNLOCK(buffer);
   3440  1.1.1.2  christos 	return r;
   3441  1.1.1.2  christos }
   3442