michael@0: /*
michael@0:  * Copyright (c) 2007-2012 Niels Provos and Nick Mathewson
michael@0:  * Copyright (c) 2002-2006 Niels Provos <provos@citi.umich.edu>
michael@0:  * All rights reserved.
michael@0:  *
michael@0:  * Redistribution and use in source and binary forms, with or without
michael@0:  * modification, are permitted provided that the following conditions
michael@0:  * are met:
michael@0:  * 1. Redistributions of source code must retain the above copyright
michael@0:  *    notice, this list of conditions and the following disclaimer.
michael@0:  * 2. Redistributions in binary form must reproduce the above copyright
michael@0:  *    notice, this list of conditions and the following disclaimer in the
michael@0:  *    documentation and/or other materials provided with the distribution.
michael@0:  * 3. The name of the author may not be used to endorse or promote products
michael@0:  *    derived from this software without specific prior written permission.
michael@0:  *
michael@0:  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
michael@0:  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
michael@0:  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
michael@0:  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
michael@0:  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
michael@0:  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
michael@0:  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
michael@0:  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
michael@0:  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
michael@0:  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
michael@0:  */
michael@0: 
michael@0: #include <sys/types.h>
michael@0: 
michael@0: #include "event2/event-config.h"
michael@0: 
michael@0: #ifdef _EVENT_HAVE_SYS_TIME_H
michael@0: #include <sys/time.h>
michael@0: #endif
michael@0: 
michael@0: #include <errno.h>
michael@0: #include <stdio.h>
michael@0: #include <stdlib.h>
michael@0: #include <string.h>
michael@0: #ifdef _EVENT_HAVE_STDARG_H
michael@0: #include <stdarg.h>
michael@0: #endif
michael@0: #ifdef _EVENT_HAVE_UNISTD_H
michael@0: #include <unistd.h>
michael@0: #endif
michael@0: 
michael@0: #ifdef WIN32
michael@0: #include <winsock2.h>
michael@0: #include <ws2tcpip.h>
michael@0: #endif
michael@0: 
michael@0: #ifdef _EVENT_HAVE_SYS_SOCKET_H
michael@0: #include <sys/socket.h>
michael@0: #endif
michael@0: #ifdef _EVENT_HAVE_NETINET_IN_H
michael@0: #include <netinet/in.h>
michael@0: #endif
michael@0: #ifdef _EVENT_HAVE_NETINET_IN6_H
michael@0: #include <netinet/in6.h>
michael@0: #endif
michael@0: 
michael@0: #include "event2/util.h"
michael@0: #include "event2/bufferevent.h"
michael@0: #include "event2/buffer.h"
michael@0: #include "event2/bufferevent_struct.h"
michael@0: #include "event2/bufferevent_compat.h"
michael@0: #include "event2/event.h"
michael@0: #include "log-internal.h"
michael@0: #include "mm-internal.h"
michael@0: #include "bufferevent-internal.h"
michael@0: #include "util-internal.h"
michael@0: #ifdef WIN32
michael@0: #include "iocp-internal.h"
michael@0: #endif
michael@0: 
michael@0: /* prototypes */
michael@0: static int be_socket_enable(struct bufferevent *, short);
michael@0: static int be_socket_disable(struct bufferevent *, short);
michael@0: static void be_socket_destruct(struct bufferevent *);
michael@0: static int be_socket_adj_timeouts(struct bufferevent *);
michael@0: static int be_socket_flush(struct bufferevent *, short, enum bufferevent_flush_mode);
michael@0: static int be_socket_ctrl(struct bufferevent *, enum bufferevent_ctrl_op, union bufferevent_ctrl_data *);
michael@0: 
michael@0: static void be_socket_setfd(struct bufferevent *, evutil_socket_t);
michael@0: 
michael@0: const struct bufferevent_ops bufferevent_ops_socket = {
michael@0: 	"socket",
michael@0: 	evutil_offsetof(struct bufferevent_private, bev),
michael@0: 	be_socket_enable,
michael@0: 	be_socket_disable,
michael@0: 	be_socket_destruct,
michael@0: 	be_socket_adj_timeouts,
michael@0: 	be_socket_flush,
michael@0: 	be_socket_ctrl,
michael@0: };
michael@0: 
michael@0: #define be_socket_add(ev, t)			\
michael@0: 	_bufferevent_add_event((ev), (t))
michael@0: 
michael@0: static void
michael@0: bufferevent_socket_outbuf_cb(struct evbuffer *buf,
michael@0:     const struct evbuffer_cb_info *cbinfo,
michael@0:     void *arg)
michael@0: {
michael@0: 	struct bufferevent *bufev = arg;
michael@0: 	struct bufferevent_private *bufev_p =
michael@0: 	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
michael@0: 
michael@0: 	if (cbinfo->n_added &&
michael@0: 	    (bufev->enabled & EV_WRITE) &&
michael@0: 	    !event_pending(&bufev->ev_write, EV_WRITE, NULL) &&
michael@0: 	    !bufev_p->write_suspended) {
michael@0: 		/* Somebody added data to the buffer, and we would like to
michael@0: 		 * write, and we were not writing.  So, start writing. */
michael@0: 		if (be_socket_add(&bufev->ev_write, &bufev->timeout_write) == -1) {
michael@0: 		    /* Should we log this? */
michael@0: 		}
michael@0: 	}
michael@0: }
michael@0: 
michael@0: static void
michael@0: bufferevent_readcb(evutil_socket_t fd, short event, void *arg)
michael@0: {
michael@0: 	struct bufferevent *bufev = arg;
michael@0: 	struct bufferevent_private *bufev_p =
michael@0: 	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
michael@0: 	struct evbuffer *input;
michael@0: 	int res = 0;
michael@0: 	short what = BEV_EVENT_READING;
michael@0: 	ev_ssize_t howmuch = -1, readmax=-1;
michael@0: 
michael@0: 	_bufferevent_incref_and_lock(bufev);
michael@0: 
michael@0: 	if (event == EV_TIMEOUT) {
michael@0: 		/* Note that we only check for event==EV_TIMEOUT. If
michael@0: 		 * event==EV_TIMEOUT|EV_READ, we can safely ignore the
michael@0: 		 * timeout, since a read has occurred */
michael@0: 		what |= BEV_EVENT_TIMEOUT;
michael@0: 		goto error;
michael@0: 	}
michael@0: 
michael@0: 	input = bufev->input;
michael@0: 
michael@0: 	/*
michael@0: 	 * If we have a high watermark configured then we don't want to
michael@0: 	 * read more data than would make us reach the watermark.
michael@0: 	 */
michael@0: 	if (bufev->wm_read.high != 0) {
michael@0: 		howmuch = bufev->wm_read.high - evbuffer_get_length(input);
michael@0: 		/* we somehow lowered the watermark, stop reading */
michael@0: 		if (howmuch <= 0) {
michael@0: 			bufferevent_wm_suspend_read(bufev);
michael@0: 			goto done;
michael@0: 		}
michael@0: 	}
michael@0: 	readmax = _bufferevent_get_read_max(bufev_p);
michael@0: 	if (howmuch < 0 || howmuch > readmax) /* The use of -1 for "unlimited"
michael@0: 					       * uglifies this code. XXXX */
michael@0: 		howmuch = readmax;
michael@0: 	if (bufev_p->read_suspended)
michael@0: 		goto done;
michael@0: 
michael@0: 	evbuffer_unfreeze(input, 0);
michael@0: 	res = evbuffer_read(input, fd, (int)howmuch); /* XXXX evbuffer_read would do better to take and return ev_ssize_t */
michael@0: 	evbuffer_freeze(input, 0);
michael@0: 
michael@0: 	if (res == -1) {
michael@0: 		int err = evutil_socket_geterror(fd);
michael@0: 		if (EVUTIL_ERR_RW_RETRIABLE(err))
michael@0: 			goto reschedule;
michael@0: 		/* error case */
michael@0: 		what |= BEV_EVENT_ERROR;
michael@0: 	} else if (res == 0) {
michael@0: 		/* eof case */
michael@0: 		what |= BEV_EVENT_EOF;
michael@0: 	}
michael@0: 
michael@0: 	if (res <= 0)
michael@0: 		goto error;
michael@0: 
michael@0: 	_bufferevent_decrement_read_buckets(bufev_p, res);
michael@0: 
michael@0: 	/* Invoke the user callback - must always be called last */
michael@0: 	if (evbuffer_get_length(input) >= bufev->wm_read.low)
michael@0: 		_bufferevent_run_readcb(bufev);
michael@0: 
michael@0: 	goto done;
michael@0: 
michael@0:  reschedule:
michael@0: 	goto done;
michael@0: 
michael@0:  error:
michael@0: 	bufferevent_disable(bufev, EV_READ);
michael@0: 	_bufferevent_run_eventcb(bufev, what);
michael@0: 
michael@0:  done:
michael@0: 	_bufferevent_decref_and_unlock(bufev);
michael@0: }
michael@0: 
michael@0: static void
michael@0: bufferevent_writecb(evutil_socket_t fd, short event, void *arg)
michael@0: {
michael@0: 	struct bufferevent *bufev = arg;
michael@0: 	struct bufferevent_private *bufev_p =
michael@0: 	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
michael@0: 	int res = 0;
michael@0: 	short what = BEV_EVENT_WRITING;
michael@0: 	int connected = 0;
michael@0: 	ev_ssize_t atmost = -1;
michael@0: 
michael@0: 	_bufferevent_incref_and_lock(bufev);
michael@0: 
michael@0: 	if (event == EV_TIMEOUT) {
michael@0: 		/* Note that we only check for event==EV_TIMEOUT. If
michael@0: 		 * event==EV_TIMEOUT|EV_WRITE, we can safely ignore the
michael@0: 		 * timeout, since a read has occurred */
michael@0: 		what |= BEV_EVENT_TIMEOUT;
michael@0: 		goto error;
michael@0: 	}
michael@0: 	if (bufev_p->connecting) {
michael@0: 		int c = evutil_socket_finished_connecting(fd);
michael@0: 		/* we need to fake the error if the connection was refused
michael@0: 		 * immediately - usually connection to localhost on BSD */
michael@0: 		if (bufev_p->connection_refused) {
michael@0: 		  bufev_p->connection_refused = 0;
michael@0: 		  c = -1;
michael@0: 		}
michael@0: 
michael@0: 		if (c == 0)
michael@0: 			goto done;
michael@0: 
michael@0: 		bufev_p->connecting = 0;
michael@0: 		if (c < 0) {
michael@0: 			event_del(&bufev->ev_write);
michael@0: 			event_del(&bufev->ev_read);
michael@0: 			_bufferevent_run_eventcb(bufev, BEV_EVENT_ERROR);
michael@0: 			goto done;
michael@0: 		} else {
michael@0: 			connected = 1;
michael@0: #ifdef WIN32
michael@0: 			if (BEV_IS_ASYNC(bufev)) {
michael@0: 				event_del(&bufev->ev_write);
michael@0: 				bufferevent_async_set_connected(bufev);
michael@0: 				_bufferevent_run_eventcb(bufev,
michael@0: 						BEV_EVENT_CONNECTED);
michael@0: 				goto done;
michael@0: 			}
michael@0: #endif
michael@0: 			_bufferevent_run_eventcb(bufev,
michael@0: 					BEV_EVENT_CONNECTED);
michael@0: 			if (!(bufev->enabled & EV_WRITE) ||
michael@0: 			    bufev_p->write_suspended) {
michael@0: 				event_del(&bufev->ev_write);
michael@0: 				goto done;
michael@0: 			}
michael@0: 		}
michael@0: 	}
michael@0: 
michael@0: 	atmost = _bufferevent_get_write_max(bufev_p);
michael@0: 
michael@0: 	if (bufev_p->write_suspended)
michael@0: 		goto done;
michael@0: 
michael@0: 	if (evbuffer_get_length(bufev->output)) {
michael@0: 		evbuffer_unfreeze(bufev->output, 1);
michael@0: 		res = evbuffer_write_atmost(bufev->output, fd, atmost);
michael@0: 		evbuffer_freeze(bufev->output, 1);
michael@0: 		if (res == -1) {
michael@0: 			int err = evutil_socket_geterror(fd);
michael@0: 			if (EVUTIL_ERR_RW_RETRIABLE(err))
michael@0: 				goto reschedule;
michael@0: 			what |= BEV_EVENT_ERROR;
michael@0: 		} else if (res == 0) {
michael@0: 			/* eof case
michael@0: 			   XXXX Actually, a 0 on write doesn't indicate
michael@0: 			   an EOF. An ECONNRESET might be more typical.
michael@0: 			 */
michael@0: 			what |= BEV_EVENT_EOF;
michael@0: 		}
michael@0: 		if (res <= 0)
michael@0: 			goto error;
michael@0: 
michael@0: 		_bufferevent_decrement_write_buckets(bufev_p, res);
michael@0: 	}
michael@0: 
michael@0: 	if (evbuffer_get_length(bufev->output) == 0) {
michael@0: 		event_del(&bufev->ev_write);
michael@0: 	}
michael@0: 
michael@0: 	/*
michael@0: 	 * Invoke the user callback if our buffer is drained or below the
michael@0: 	 * low watermark.
michael@0: 	 */
michael@0: 	if ((res || !connected) &&
michael@0: 	    evbuffer_get_length(bufev->output) <= bufev->wm_write.low) {
michael@0: 		_bufferevent_run_writecb(bufev);
michael@0: 	}
michael@0: 
michael@0: 	goto done;
michael@0: 
michael@0:  reschedule:
michael@0: 	if (evbuffer_get_length(bufev->output) == 0) {
michael@0: 		event_del(&bufev->ev_write);
michael@0: 	}
michael@0: 	goto done;
michael@0: 
michael@0:  error:
michael@0: 	bufferevent_disable(bufev, EV_WRITE);
michael@0: 	_bufferevent_run_eventcb(bufev, what);
michael@0: 
michael@0:  done:
michael@0: 	_bufferevent_decref_and_unlock(bufev);
michael@0: }
michael@0: 
michael@0: struct bufferevent *
michael@0: bufferevent_socket_new(struct event_base *base, evutil_socket_t fd,
michael@0:     int options)
michael@0: {
michael@0: 	struct bufferevent_private *bufev_p;
michael@0: 	struct bufferevent *bufev;
michael@0: 
michael@0: #ifdef WIN32
michael@0: 	if (base && event_base_get_iocp(base))
michael@0: 		return bufferevent_async_new(base, fd, options);
michael@0: #endif
michael@0: 
michael@0: 	if ((bufev_p = mm_calloc(1, sizeof(struct bufferevent_private)))== NULL)
michael@0: 		return NULL;
michael@0: 
michael@0: 	if (bufferevent_init_common(bufev_p, base, &bufferevent_ops_socket,
michael@0: 				    options) < 0) {
michael@0: 		mm_free(bufev_p);
michael@0: 		return NULL;
michael@0: 	}
michael@0: 	bufev = &bufev_p->bev;
michael@0: 	evbuffer_set_flags(bufev->output, EVBUFFER_FLAG_DRAINS_TO_FD);
michael@0: 
michael@0: 	event_assign(&bufev->ev_read, bufev->ev_base, fd,
michael@0: 	    EV_READ|EV_PERSIST, bufferevent_readcb, bufev);
michael@0: 	event_assign(&bufev->ev_write, bufev->ev_base, fd,
michael@0: 	    EV_WRITE|EV_PERSIST, bufferevent_writecb, bufev);
michael@0: 
michael@0: 	evbuffer_add_cb(bufev->output, bufferevent_socket_outbuf_cb, bufev);
michael@0: 
michael@0: 	evbuffer_freeze(bufev->input, 0);
michael@0: 	evbuffer_freeze(bufev->output, 1);
michael@0: 
michael@0: 	return bufev;
michael@0: }
michael@0: 
michael@0: int
michael@0: bufferevent_socket_connect(struct bufferevent *bev,
michael@0:     struct sockaddr *sa, int socklen)
michael@0: {
michael@0: 	struct bufferevent_private *bufev_p =
michael@0: 	    EVUTIL_UPCAST(bev, struct bufferevent_private, bev);
michael@0: 
michael@0: 	evutil_socket_t fd;
michael@0: 	int r = 0;
michael@0: 	int result=-1;
michael@0: 	int ownfd = 0;
michael@0: 
michael@0: 	_bufferevent_incref_and_lock(bev);
michael@0: 
michael@0: 	if (!bufev_p)
michael@0: 		goto done;
michael@0: 
michael@0: 	fd = bufferevent_getfd(bev);
michael@0: 	if (fd < 0) {
michael@0: 		if (!sa)
michael@0: 			goto done;
michael@0: 		fd = socket(sa->sa_family, SOCK_STREAM, 0);
michael@0: 		if (fd < 0)
michael@0: 			goto done;
michael@0: 		if (evutil_make_socket_nonblocking(fd)<0)
michael@0: 			goto done;
michael@0: 		ownfd = 1;
michael@0: 	}
michael@0: 	if (sa) {
michael@0: #ifdef WIN32
michael@0: 		if (bufferevent_async_can_connect(bev)) {
michael@0: 			bufferevent_setfd(bev, fd);
michael@0: 			r = bufferevent_async_connect(bev, fd, sa, socklen);
michael@0: 			if (r < 0)
michael@0: 				goto freesock;
michael@0: 			bufev_p->connecting = 1;
michael@0: 			result = 0;
michael@0: 			goto done;
michael@0: 		} else
michael@0: #endif
michael@0: 		r = evutil_socket_connect(&fd, sa, socklen);
michael@0: 		if (r < 0)
michael@0: 			goto freesock;
michael@0: 	}
michael@0: #ifdef WIN32
michael@0: 	/* ConnectEx() isn't always around, even when IOCP is enabled.
michael@0: 	 * Here, we borrow the socket object's write handler to fall back
michael@0: 	 * on a non-blocking connect() when ConnectEx() is unavailable. */
michael@0: 	if (BEV_IS_ASYNC(bev)) {
michael@0: 		event_assign(&bev->ev_write, bev->ev_base, fd,
michael@0: 		    EV_WRITE|EV_PERSIST, bufferevent_writecb, bev);
michael@0: 	}
michael@0: #endif
michael@0: 	bufferevent_setfd(bev, fd);
michael@0: 	if (r == 0) {
michael@0: 		if (! be_socket_enable(bev, EV_WRITE)) {
michael@0: 			bufev_p->connecting = 1;
michael@0: 			result = 0;
michael@0: 			goto done;
michael@0: 		}
michael@0: 	} else if (r == 1) {
michael@0: 		/* The connect succeeded already. How very BSD of it. */
michael@0: 		result = 0;
michael@0: 		bufev_p->connecting = 1;
michael@0: 		event_active(&bev->ev_write, EV_WRITE, 1);
michael@0: 	} else {
michael@0: 		/* The connect failed already.  How very BSD of it. */
michael@0: 		bufev_p->connection_refused = 1;
michael@0: 		bufev_p->connecting = 1;
michael@0: 		result = 0;
michael@0: 		event_active(&bev->ev_write, EV_WRITE, 1);
michael@0: 	}
michael@0: 
michael@0: 	goto done;
michael@0: 
michael@0: freesock:
michael@0: 	_bufferevent_run_eventcb(bev, BEV_EVENT_ERROR);
michael@0: 	if (ownfd)
michael@0: 		evutil_closesocket(fd);
michael@0: 	/* do something about the error? */
michael@0: done:
michael@0: 	_bufferevent_decref_and_unlock(bev);
michael@0: 	return result;
michael@0: }
michael@0: 
michael@0: static void
michael@0: bufferevent_connect_getaddrinfo_cb(int result, struct evutil_addrinfo *ai,
michael@0:     void *arg)
michael@0: {
michael@0: 	struct bufferevent *bev = arg;
michael@0: 	struct bufferevent_private *bev_p =
michael@0: 	    EVUTIL_UPCAST(bev, struct bufferevent_private, bev);
michael@0: 	int r;
michael@0: 	BEV_LOCK(bev);
michael@0: 
michael@0: 	bufferevent_unsuspend_write(bev, BEV_SUSPEND_LOOKUP);
michael@0: 	bufferevent_unsuspend_read(bev, BEV_SUSPEND_LOOKUP);
michael@0: 
michael@0: 	if (result != 0) {
michael@0: 		bev_p->dns_error = result;
michael@0: 		_bufferevent_run_eventcb(bev, BEV_EVENT_ERROR);
michael@0: 		_bufferevent_decref_and_unlock(bev);
michael@0: 		if (ai)
michael@0: 			evutil_freeaddrinfo(ai);
michael@0: 		return;
michael@0: 	}
michael@0: 
michael@0: 	/* XXX use the other addrinfos? */
michael@0: 	/* XXX use this return value */
michael@0: 	r = bufferevent_socket_connect(bev, ai->ai_addr, (int)ai->ai_addrlen);
michael@0: 	(void)r;
michael@0: 	_bufferevent_decref_and_unlock(bev);
michael@0: 	evutil_freeaddrinfo(ai);
michael@0: }
michael@0: 
michael@0: int
michael@0: bufferevent_socket_connect_hostname(struct bufferevent *bev,
michael@0:     struct evdns_base *evdns_base, int family, const char *hostname, int port)
michael@0: {
michael@0: 	char portbuf[10];
michael@0: 	struct evutil_addrinfo hint;
michael@0: 	int err;
michael@0: 	struct bufferevent_private *bev_p =
michael@0: 	    EVUTIL_UPCAST(bev, struct bufferevent_private, bev);
michael@0: 
michael@0: 	if (family != AF_INET && family != AF_INET6 && family != AF_UNSPEC)
michael@0: 		return -1;
michael@0: 	if (port < 1 || port > 65535)
michael@0: 		return -1;
michael@0: 
michael@0: 	BEV_LOCK(bev);
michael@0: 	bev_p->dns_error = 0;
michael@0: 	BEV_UNLOCK(bev);
michael@0: 
michael@0: 	evutil_snprintf(portbuf, sizeof(portbuf), "%d", port);
michael@0: 
michael@0: 	memset(&hint, 0, sizeof(hint));
michael@0: 	hint.ai_family = family;
michael@0: 	hint.ai_protocol = IPPROTO_TCP;
michael@0: 	hint.ai_socktype = SOCK_STREAM;
michael@0: 
michael@0: 	bufferevent_suspend_write(bev, BEV_SUSPEND_LOOKUP);
michael@0: 	bufferevent_suspend_read(bev, BEV_SUSPEND_LOOKUP);
michael@0: 
michael@0: 	bufferevent_incref(bev);
michael@0: 	err = evutil_getaddrinfo_async(evdns_base, hostname, portbuf,
michael@0: 	    &hint, bufferevent_connect_getaddrinfo_cb, bev);
michael@0: 
michael@0: 	if (err == 0) {
michael@0: 		return 0;
michael@0: 	} else {
michael@0: 		bufferevent_unsuspend_write(bev, BEV_SUSPEND_LOOKUP);
michael@0: 		bufferevent_unsuspend_read(bev, BEV_SUSPEND_LOOKUP);
michael@0: 		return -1;
michael@0: 	}
michael@0: }
michael@0: 
michael@0: int
michael@0: bufferevent_socket_get_dns_error(struct bufferevent *bev)
michael@0: {
michael@0: 	int rv;
michael@0: 	struct bufferevent_private *bev_p =
michael@0: 	    EVUTIL_UPCAST(bev, struct bufferevent_private, bev);
michael@0: 
michael@0: 	BEV_LOCK(bev);
michael@0: 	rv = bev_p->dns_error;
michael@0: 	BEV_LOCK(bev);
michael@0: 
michael@0: 	return rv;
michael@0: }
michael@0: 
michael@0: /*
michael@0:  * Create a new buffered event object.
michael@0:  *
michael@0:  * The read callback is invoked whenever we read new data.
michael@0:  * The write callback is invoked whenever the output buffer is drained.
michael@0:  * The error callback is invoked on a write/read error or on EOF.
michael@0:  *
michael@0:  * Both read and write callbacks maybe NULL.  The error callback is not
michael@0:  * allowed to be NULL and have to be provided always.
michael@0:  */
michael@0: 
michael@0: struct bufferevent *
michael@0: bufferevent_new(evutil_socket_t fd,
michael@0:     bufferevent_data_cb readcb, bufferevent_data_cb writecb,
michael@0:     bufferevent_event_cb eventcb, void *cbarg)
michael@0: {
michael@0: 	struct bufferevent *bufev;
michael@0: 
michael@0: 	if (!(bufev = bufferevent_socket_new(NULL, fd, 0)))
michael@0: 		return NULL;
michael@0: 
michael@0: 	bufferevent_setcb(bufev, readcb, writecb, eventcb, cbarg);
michael@0: 
michael@0: 	return bufev;
michael@0: }
michael@0: 
michael@0: 
michael@0: static int
michael@0: be_socket_enable(struct bufferevent *bufev, short event)
michael@0: {
michael@0: 	if (event & EV_READ) {
michael@0: 		if (be_socket_add(&bufev->ev_read,&bufev->timeout_read) == -1)
michael@0: 			return -1;
michael@0: 	}
michael@0: 	if (event & EV_WRITE) {
michael@0: 		if (be_socket_add(&bufev->ev_write,&bufev->timeout_write) == -1)
michael@0: 			return -1;
michael@0: 	}
michael@0: 	return 0;
michael@0: }
michael@0: 
michael@0: static int
michael@0: be_socket_disable(struct bufferevent *bufev, short event)
michael@0: {
michael@0: 	struct bufferevent_private *bufev_p =
michael@0: 	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
michael@0: 	if (event & EV_READ) {
michael@0: 		if (event_del(&bufev->ev_read) == -1)
michael@0: 			return -1;
michael@0: 	}
michael@0: 	/* Don't actually disable the write if we are trying to connect. */
michael@0: 	if ((event & EV_WRITE) && ! bufev_p->connecting) {
michael@0: 		if (event_del(&bufev->ev_write) == -1)
michael@0: 			return -1;
michael@0: 	}
michael@0: 	return 0;
michael@0: }
michael@0: 
michael@0: static void
michael@0: be_socket_destruct(struct bufferevent *bufev)
michael@0: {
michael@0: 	struct bufferevent_private *bufev_p =
michael@0: 	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
michael@0: 	evutil_socket_t fd;
michael@0: 	EVUTIL_ASSERT(bufev->be_ops == &bufferevent_ops_socket);
michael@0: 
michael@0: 	fd = event_get_fd(&bufev->ev_read);
michael@0: 
michael@0: 	event_del(&bufev->ev_read);
michael@0: 	event_del(&bufev->ev_write);
michael@0: 
michael@0: 	if ((bufev_p->options & BEV_OPT_CLOSE_ON_FREE) && fd >= 0)
michael@0: 		EVUTIL_CLOSESOCKET(fd);
michael@0: }
michael@0: 
michael@0: static int
michael@0: be_socket_adj_timeouts(struct bufferevent *bufev)
michael@0: {
michael@0: 	int r = 0;
michael@0: 	if (event_pending(&bufev->ev_read, EV_READ, NULL))
michael@0: 		if (be_socket_add(&bufev->ev_read, &bufev->timeout_read) < 0)
michael@0: 			r = -1;
michael@0: 	if (event_pending(&bufev->ev_write, EV_WRITE, NULL)) {
michael@0: 		if (be_socket_add(&bufev->ev_write, &bufev->timeout_write) < 0)
michael@0: 			r = -1;
michael@0: 	}
michael@0: 	return r;
michael@0: }
michael@0: 
michael@0: static int
michael@0: be_socket_flush(struct bufferevent *bev, short iotype,
michael@0:     enum bufferevent_flush_mode mode)
michael@0: {
michael@0: 	return 0;
michael@0: }
michael@0: 
michael@0: 
michael@0: static void
michael@0: be_socket_setfd(struct bufferevent *bufev, evutil_socket_t fd)
michael@0: {
michael@0: 	BEV_LOCK(bufev);
michael@0: 	EVUTIL_ASSERT(bufev->be_ops == &bufferevent_ops_socket);
michael@0: 
michael@0: 	event_del(&bufev->ev_read);
michael@0: 	event_del(&bufev->ev_write);
michael@0: 
michael@0: 	event_assign(&bufev->ev_read, bufev->ev_base, fd,
michael@0: 	    EV_READ|EV_PERSIST, bufferevent_readcb, bufev);
michael@0: 	event_assign(&bufev->ev_write, bufev->ev_base, fd,
michael@0: 	    EV_WRITE|EV_PERSIST, bufferevent_writecb, bufev);
michael@0: 
michael@0: 	if (fd >= 0)
michael@0: 		bufferevent_enable(bufev, bufev->enabled);
michael@0: 
michael@0: 	BEV_UNLOCK(bufev);
michael@0: }
michael@0: 
michael@0: /* XXXX Should non-socket bufferevents support this? */
michael@0: int
michael@0: bufferevent_priority_set(struct bufferevent *bufev, int priority)
michael@0: {
michael@0: 	int r = -1;
michael@0: 
michael@0: 	BEV_LOCK(bufev);
michael@0: 	if (bufev->be_ops != &bufferevent_ops_socket)
michael@0: 		goto done;
michael@0: 
michael@0: 	if (event_priority_set(&bufev->ev_read, priority) == -1)
michael@0: 		goto done;
michael@0: 	if (event_priority_set(&bufev->ev_write, priority) == -1)
michael@0: 		goto done;
michael@0: 
michael@0: 	r = 0;
michael@0: done:
michael@0: 	BEV_UNLOCK(bufev);
michael@0: 	return r;
michael@0: }
michael@0: 
michael@0: /* XXXX Should non-socket bufferevents support this? */
michael@0: int
michael@0: bufferevent_base_set(struct event_base *base, struct bufferevent *bufev)
michael@0: {
michael@0: 	int res = -1;
michael@0: 
michael@0: 	BEV_LOCK(bufev);
michael@0: 	if (bufev->be_ops != &bufferevent_ops_socket)
michael@0: 		goto done;
michael@0: 
michael@0: 	bufev->ev_base = base;
michael@0: 
michael@0: 	res = event_base_set(base, &bufev->ev_read);
michael@0: 	if (res == -1)
michael@0: 		goto done;
michael@0: 
michael@0: 	res = event_base_set(base, &bufev->ev_write);
michael@0: done:
michael@0: 	BEV_UNLOCK(bufev);
michael@0: 	return res;
michael@0: }
michael@0: 
michael@0: static int
michael@0: be_socket_ctrl(struct bufferevent *bev, enum bufferevent_ctrl_op op,
michael@0:     union bufferevent_ctrl_data *data)
michael@0: {
michael@0: 	switch (op) {
michael@0: 	case BEV_CTRL_SET_FD:
michael@0: 		be_socket_setfd(bev, data->fd);
michael@0: 		return 0;
michael@0: 	case BEV_CTRL_GET_FD:
michael@0: 		data->fd = event_get_fd(&bev->ev_read);
michael@0: 		return 0;
michael@0: 	case BEV_CTRL_GET_UNDERLYING:
michael@0: 	case BEV_CTRL_CANCEL_ALL:
michael@0: 	default:
michael@0: 		return -1;
michael@0: 	}
michael@0: }