/*

* Copyright (c) 2002-2007 Niels Provos <provos@citi.umich.edu>

* Copyright (c) 2007-2012 Niels Provos, Nick Mathewson

*

* Redistribution and use in source and binary forms, with or without

* modification, are permitted provided that the following conditions

* are met:

* 1. Redistributions of source code must retain the above copyright

* notice, this list of conditions and the following disclaimer.

* 2. Redistributions in binary form must reproduce the above copyright

* notice, this list of conditions and the following disclaimer in the

* documentation and/or other materials provided with the distribution.

* 3. The name of the author may not be used to endorse or promote products

* derived from this software without specific prior written permission.

*

* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR

* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES

* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.

* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,

* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT

* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF

* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

*/

#include "event2/event-config.h"

#include "evconfig-private.h"

#include <sys/types.h>

#ifdef EVENT__HAVE_SYS_TIME_H

#include <sys/time.h>

#endif

#include <errno.h>

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#ifdef EVENT__HAVE_STDARG_H

#include <stdarg.h>

#endif

#ifdef _WIN32

#include <winsock2.h>

#endif

#include "event2/util.h"

#include "event2/buffer.h"

#include "event2/buffer_compat.h"

#include "event2/bufferevent.h"

#include "event2/bufferevent_struct.h"

#include "event2/bufferevent_compat.h"

#include "event2/event.h"

#include "event-internal.h"

#include "log-internal.h"

#include "mm-internal.h"

#include "bufferevent-internal.h"

#include "evbuffer-internal.h"

#include "util-internal.h"

static void bufferevent_cancel_all_(struct bufferevent *bev);

static void bufferevent_finalize_cb_(struct event_callback *evcb, void *arg_);

void

bufferevent_suspend_read_(struct bufferevent *bufev, bufferevent_suspend_flags what)

{

struct bufferevent_private *bufev_private = BEV_UPCAST(bufev);

BEV_LOCK(bufev);

if (!bufev_private->read_suspended)

bufev->be_ops->disable(bufev, EV_READ);

bufev_private->read_suspended |= what;

BEV_UNLOCK(bufev);

}

void

bufferevent_unsuspend_read_(struct bufferevent *bufev, bufferevent_suspend_flags what)

{

struct bufferevent_private *bufev_private = BEV_UPCAST(bufev);

BEV_LOCK(bufev);

bufev_private->read_suspended &= ~what;

if (!bufev_private->read_suspended && (bufev->enabled & EV_READ))

bufev->be_ops->enable(bufev, EV_READ);

BEV_UNLOCK(bufev);

}

void

bufferevent_suspend_write_(struct bufferevent *bufev, bufferevent_suspend_flags what)

{

struct bufferevent_private *bufev_private = BEV_UPCAST(bufev);

BEV_LOCK(bufev);

if (!bufev_private->write_suspended)

bufev->be_ops->disable(bufev, EV_WRITE);

bufev_private->write_suspended |= what;

BEV_UNLOCK(bufev);

}

void

bufferevent_unsuspend_write_(struct bufferevent *bufev, bufferevent_suspend_flags what)

{

struct bufferevent_private *bufev_private = BEV_UPCAST(bufev);

BEV_LOCK(bufev);

bufev_private->write_suspended &= ~what;

if (!bufev_private->write_suspended && (bufev->enabled & EV_WRITE))

bufev->be_ops->enable(bufev, EV_WRITE);

BEV_UNLOCK(bufev);

}

/**

* Sometimes bufferevent's implementation can overrun high watermarks

* (one of examples is openssl) and in this case if the read callback

* will not handle enough data do over condition above the read

* callback will never be called again (due to suspend above).

*

* To avoid this we are scheduling read callback again here, but only

* from the user callback to avoid multiple scheduling:

* - when the data had been added to it

* - when the data had been drained from it (user specified read callback)

*/

static void bufferevent_inbuf_wm_check(struct bufferevent *bev)

{

if (!bev->wm_read.high)

return;

if (!(bev->enabled & EV_READ))

return;

if (evbuffer_get_length(bev->input) < bev->wm_read.high)

return;

bufferevent_trigger(bev, EV_READ, BEV_OPT_DEFER_CALLBACKS);

}

/* Callback to implement watermarks on the input buffer. Only enabled

* if the watermark is set. */

static void

bufferevent_inbuf_wm_cb(struct evbuffer *buf,

const struct evbuffer_cb_info *cbinfo,

void *arg)

{

struct bufferevent *bufev = arg;

size_t size;

size = evbuffer_get_length(buf);

if (size >= bufev->wm_read.high)

bufferevent_wm_suspend_read(bufev);

else

bufferevent_wm_unsuspend_read(bufev);

}

static void

bufferevent_run_deferred_callbacks_locked(struct event_callback *cb, void *arg)

{

struct bufferevent_private *bufev_private = arg;

struct bufferevent *bufev = &bufev_private->bev;

BEV_LOCK(bufev);

if ((bufev_private->eventcb_pending & BEV_EVENT_CONNECTED) &&

bufev->errorcb) {

/* The "connected" happened before any reads or writes, so

send it first. */

bufev_private->eventcb_pending &= ~BEV_EVENT_CONNECTED;

bufev->errorcb(bufev, BEV_EVENT_CONNECTED, bufev->cbarg);

}

if (bufev_private->readcb_pending && bufev->readcb) {

bufev_private->readcb_pending = 0;

bufev->readcb(bufev, bufev->cbarg);

bufferevent_inbuf_wm_check(bufev);

}

if (bufev_private->writecb_pending && bufev->writecb) {

bufev_private->writecb_pending = 0;

bufev->writecb(bufev, bufev->cbarg);

}

if (bufev_private->eventcb_pending && bufev->errorcb) {

short what = bufev_private->eventcb_pending;

int err = bufev_private->errno_pending;

bufev_private->eventcb_pending = 0;

bufev_private->errno_pending = 0;

EVUTIL_SET_SOCKET_ERROR(err);

bufev->errorcb(bufev, what, bufev->cbarg);

}

bufferevent_decref_and_unlock_(bufev);

}

static void

bufferevent_run_deferred_callbacks_unlocked(struct event_callback *cb, void *arg)

{

struct bufferevent_private *bufev_private = arg;

struct bufferevent *bufev = &bufev_private->bev;

BEV_LOCK(bufev);

#define UNLOCKED(stmt) \

do { BEV_UNLOCK(bufev); stmt; BEV_LOCK(bufev); } while(0)

if ((bufev_private->eventcb_pending & BEV_EVENT_CONNECTED) &&

bufev->errorcb) {

/* The "connected" happened before any reads or writes, so

send it first. */

bufferevent_event_cb errorcb = bufev->errorcb;

void *cbarg = bufev->cbarg;

bufev_private->eventcb_pending &= ~BEV_EVENT_CONNECTED;

UNLOCKED(errorcb(bufev, BEV_EVENT_CONNECTED, cbarg));

}

if (bufev_private->readcb_pending && bufev->readcb) {

bufferevent_data_cb readcb = bufev->readcb;

void *cbarg = bufev->cbarg;

bufev_private->readcb_pending = 0;

UNLOCKED(readcb(bufev, cbarg));

bufferevent_inbuf_wm_check(bufev);

}

if (bufev_private->writecb_pending && bufev->writecb) {

bufferevent_data_cb writecb = bufev->writecb;

void *cbarg = bufev->cbarg;

bufev_private->writecb_pending = 0;

UNLOCKED(writecb(bufev, cbarg));

}

if (bufev_private->eventcb_pending && bufev->errorcb) {

bufferevent_event_cb errorcb = bufev->errorcb;

void *cbarg = bufev->cbarg;

short what = bufev_private->eventcb_pending;

int err = bufev_private->errno_pending;

bufev_private->eventcb_pending = 0;

bufev_private->errno_pending = 0;

EVUTIL_SET_SOCKET_ERROR(err);

UNLOCKED(errorcb(bufev,what,cbarg));

}

bufferevent_decref_and_unlock_(bufev);

#undef UNLOCKED

}

#define SCHEDULE_DEFERRED(bevp) \

do { \

if (event_deferred_cb_schedule_( \

(bevp)->bev.ev_base, \

&(bevp)->deferred)) \

bufferevent_incref_(&(bevp)->bev); \

} while (0)

void

bufferevent_run_readcb_(struct bufferevent *bufev, int options)

{

/* Requires that we hold the lock and a reference */

struct bufferevent_private *p = BEV_UPCAST(bufev);

if (bufev->readcb == NULL)

return;

if ((p->options|options) & BEV_OPT_DEFER_CALLBACKS) {

p->readcb_pending = 1;

SCHEDULE_DEFERRED(p);

} else {

bufev->readcb(bufev, bufev->cbarg);

bufferevent_inbuf_wm_check(bufev);

}

}

void

bufferevent_run_writecb_(struct bufferevent *bufev, int options)

{

/* Requires that we hold the lock and a reference */

struct bufferevent_private *p = BEV_UPCAST(bufev);

if (bufev->writecb == NULL)

return;

if ((p->options|options) & BEV_OPT_DEFER_CALLBACKS) {

p->writecb_pending = 1;

SCHEDULE_DEFERRED(p);

} else {

bufev->writecb(bufev, bufev->cbarg);

}

}

#define BEV_TRIG_ALL_OPTS ( \

void

bufferevent_trigger(struct bufferevent *bufev, short iotype, int options)

{

bufferevent_incref_and_lock_(bufev);

bufferevent_trigger_nolock_(bufev, iotype, options&BEV_TRIG_ALL_OPTS);

bufferevent_decref_and_unlock_(bufev);

}

void

bufferevent_run_eventcb_(struct bufferevent *bufev, short what, int options)

{

/* Requires that we hold the lock and a reference */

struct bufferevent_private *p = BEV_UPCAST(bufev);

if (bufev->errorcb == NULL)

return;

if ((p->options|options) & BEV_OPT_DEFER_CALLBACKS) {

p->eventcb_pending |= what;

p->errno_pending = EVUTIL_SOCKET_ERROR();

SCHEDULE_DEFERRED(p);

} else {

bufev->errorcb(bufev, what, bufev->cbarg);

}

}

void

bufferevent_trigger_event(struct bufferevent *bufev, short what, int options)

{

bufferevent_incref_and_lock_(bufev);

bufferevent_run_eventcb_(bufev, what, options&BEV_TRIG_ALL_OPTS);

bufferevent_decref_and_unlock_(bufev);

}

int

bufferevent_init_common_(struct bufferevent_private *bufev_private,

struct event_base *base,

const struct bufferevent_ops *ops,

enum bufferevent_options options)

{

struct bufferevent *bufev = &bufev_private->bev;

if (!bufev->input) {

if ((bufev->input = evbuffer_new()) == NULL)

goto err;

}

if (!bufev->output) {

if ((bufev->output = evbuffer_new()) == NULL)

goto err;

}

bufev_private->refcnt = 1;

bufev->ev_base = base;

/* Disable timeouts. */

evutil_timerclear(&bufev->timeout_read);

evutil_timerclear(&bufev->timeout_write);

bufev->be_ops = ops;

if (bufferevent_ratelim_init_(bufev_private))

goto err;

/*

* Set to EV_WRITE so that using bufferevent_write is going to

* trigger a callback. Reading needs to be explicitly enabled

* because otherwise no data will be available.

*/

bufev->enabled = EV_WRITE;

#ifndef EVENT__DISABLE_THREAD_SUPPORT

if (options & BEV_OPT_THREADSAFE) {

if (bufferevent_enable_locking_(bufev, NULL) < 0)

goto err;

}

#endif

if ((options & (BEV_OPT_DEFER_CALLBACKS|BEV_OPT_UNLOCK_CALLBACKS))

== BEV_OPT_UNLOCK_CALLBACKS) {

event_warnx("UNLOCK_CALLBACKS requires DEFER_CALLBACKS");

goto err;

}

if (options & BEV_OPT_UNLOCK_CALLBACKS)

event_deferred_cb_init_(

&bufev_private->deferred,

event_base_get_npriorities(base) / 2,

bufferevent_run_deferred_callbacks_unlocked,

bufev_private);

else

event_deferred_cb_init_(

&bufev_private->deferred,

event_base_get_npriorities(base) / 2,

bufferevent_run_deferred_callbacks_locked,

bufev_private);

bufev_private->options = options;

evbuffer_set_parent_(bufev->input, bufev);

evbuffer_set_parent_(bufev->output, bufev);

return 0;

err:

if (bufev->input) {

evbuffer_free(bufev->input);

bufev->input = NULL;

}

if (bufev->output) {

evbuffer_free(bufev->output);

bufev->output = NULL;

}

return -1;

}

void

bufferevent_setcb(struct bufferevent *bufev,

bufferevent_data_cb readcb, bufferevent_data_cb writecb,

bufferevent_event_cb eventcb, void *cbarg)

{

BEV_LOCK(bufev);

bufev->readcb = readcb;

bufev->writecb = writecb;

bufev->errorcb = eventcb;

bufev->cbarg = cbarg;

BEV_UNLOCK(bufev);

}

void

bufferevent_getcb(struct bufferevent *bufev,

void **cbarg_ptr)

{

BEV_LOCK(bufev);

if (readcb_ptr)

*readcb_ptr = bufev->readcb;

if (writecb_ptr)

*writecb_ptr = bufev->writecb;

if (eventcb_ptr)

*eventcb_ptr = bufev->errorcb;

if (cbarg_ptr)

*cbarg_ptr = bufev->cbarg;

BEV_UNLOCK(bufev);

}

struct evbuffer *

bufferevent_get_input(struct bufferevent *bufev)

{

return bufev->input;

}

struct evbuffer *

bufferevent_get_output(struct bufferevent *bufev)

{

return bufev->output;

}

struct event_base *

bufferevent_get_base(struct bufferevent *bufev)

{

return bufev->ev_base;

}

int

bufferevent_get_priority(const struct bufferevent *bufev)

{

if (event_initialized(&bufev->ev_read)) {

return event_get_priority(&bufev->ev_read);

} else {

return event_base_get_npriorities(bufev->ev_base) / 2;

}

}

int

bufferevent_write(struct bufferevent *bufev, const void *data, size_t size)

{

if (evbuffer_add(bufev->output, data, size) == -1)

return (-1);

return 0;

}

int

bufferevent_write_buffer(struct bufferevent *bufev, struct evbuffer *buf)

{

if (evbuffer_add_buffer(bufev->output, buf) == -1)

return (-1);

return 0;

}

size_t

bufferevent_read(struct bufferevent *bufev, void *data, size_t size)

{

int r = evbuffer_remove(bufev->input, data, size);

if (r == -1)

return 0;

return r;

}

int

bufferevent_read_buffer(struct bufferevent *bufev, struct evbuffer *buf)

{

return (evbuffer_add_buffer(buf, bufev->input));

}

int

bufferevent_enable(struct bufferevent *bufev, short event)

{

struct bufferevent_private *bufev_private = BEV_UPCAST(bufev);

short impl_events = event;

int r = 0;

bufferevent_incref_and_lock_(bufev);

if (bufev_private->read_suspended)

impl_events &= ~EV_READ;

if (bufev_private->write_suspended)

impl_events &= ~EV_WRITE;

bufev->enabled |= event;

if (impl_events && bufev->be_ops->enable(bufev, impl_events) < 0)

r = -1;

if (r)

event_debug(("%s: cannot enable 0x%hx on %p", __func__, event, (void *)bufev));

bufferevent_decref_and_unlock_(bufev);

return r;

}

int

bufferevent_set_timeouts(struct bufferevent *bufev,

const struct timeval *tv_read,

const struct timeval *tv_write)

{

int r = 0;

BEV_LOCK(bufev);

if (tv_read) {

bufev->timeout_read = *tv_read;

} else {

evutil_timerclear(&bufev->timeout_read);

}

if (tv_write) {

bufev->timeout_write = *tv_write;

} else {

evutil_timerclear(&bufev->timeout_write);

}

if (bufev->be_ops->adj_timeouts)

r = bufev->be_ops->adj_timeouts(bufev);

BEV_UNLOCK(bufev);

return r;

}

/* Obsolete; use bufferevent_set_timeouts */

void

bufferevent_settimeout(struct bufferevent *bufev,

int timeout_read, int timeout_write)

{

struct timeval tv_read, tv_write;

struct timeval *ptv_read = NULL, *ptv_write = NULL;

memset(&tv_read, 0, sizeof(tv_read));

memset(&tv_write, 0, sizeof(tv_write));

if (timeout_read) {

tv_read.tv_sec = timeout_read;

ptv_read = &tv_read;

}

if (timeout_write) {

tv_write.tv_sec = timeout_write;

ptv_write = &tv_write;

}

bufferevent_set_timeouts(bufev, ptv_read, ptv_write);

}

int

bufferevent_disable_hard_(struct bufferevent *bufev, short event)

{

int r = 0;

struct bufferevent_private *bufev_private = BEV_UPCAST(bufev);

BEV_LOCK(bufev);

bufev->enabled &= ~event;

bufev_private->connecting = 0;

if (bufev->be_ops->disable(bufev, event) < 0)

r = -1;

BEV_UNLOCK(bufev);

return r;

}

int

bufferevent_disable(struct bufferevent *bufev, short event)

{

int r = 0;

BEV_LOCK(bufev);

bufev->enabled &= ~event;

if (bufev->be_ops->disable(bufev, event) < 0)

r = -1;

if (r)

event_debug(("%s: cannot disable 0x%hx on %p", __func__, event, (void *)bufev));

BEV_UNLOCK(bufev);

return r;

}

/*

* Sets the water marks

*/

void

bufferevent_setwatermark(struct bufferevent *bufev, short events,

size_t lowmark, size_t highmark)

{

struct bufferevent_private *bufev_private = BEV_UPCAST(bufev);

BEV_LOCK(bufev);

if (events & EV_WRITE) {

bufev->wm_write.low = lowmark;

bufev->wm_write.high = highmark;

}

if (events & EV_READ) {

bufev->wm_read.low = lowmark;

bufev->wm_read.high = highmark;

if (highmark) {

/* There is now a new high-water mark for read.

enable the callback if needed, and see if we should

suspend/bufferevent_wm_unsuspend. */

if (bufev_private->read_watermarks_cb == NULL) {

bufev_private->read_watermarks_cb =

evbuffer_add_cb(bufev->input,

bufferevent_inbuf_wm_cb,

bufev);

}

evbuffer_cb_set_flags(bufev->input,

bufev_private->read_watermarks_cb,

EVBUFFER_CB_ENABLED|EVBUFFER_CB_NODEFER);

if (evbuffer_get_length(bufev->input) >= highmark)

bufferevent_wm_suspend_read(bufev);

else if (evbuffer_get_length(bufev->input) < highmark)

bufferevent_wm_unsuspend_read(bufev);

} else {

/* There is now no high-water mark for read. */

if (bufev_private->read_watermarks_cb)

evbuffer_cb_clear_flags(bufev->input,

bufev_private->read_watermarks_cb,

EVBUFFER_CB_ENABLED);

bufferevent_wm_unsuspend_read(bufev);

}

}

BEV_UNLOCK(bufev);

}

int

bufferevent_getwatermark(struct bufferevent *bufev, short events,

size_t *lowmark, size_t *highmark)

{

if (events == EV_WRITE) {

BEV_LOCK(bufev);

if (lowmark)

*lowmark = bufev->wm_write.low;

if (highmark)

*highmark = bufev->wm_write.high;

BEV_UNLOCK(bufev);

return 0;

}

if (events == EV_READ) {

BEV_LOCK(bufev);

if (lowmark)

*lowmark = bufev->wm_read.low;

if (highmark)

*highmark = bufev->wm_read.high;

BEV_UNLOCK(bufev);

return 0;

}

return -1;

}

int

bufferevent_flush(struct bufferevent *bufev,

short iotype,

enum bufferevent_flush_mode mode)

{

int r = -1;

BEV_LOCK(bufev);

if (bufev->be_ops->flush)

r = bufev->be_ops->flush(bufev, iotype, mode);

BEV_UNLOCK(bufev);

return r;

}

void

bufferevent_incref_and_lock_(struct bufferevent *bufev)

{

struct bufferevent_private *bufev_private = BEV_UPCAST(bufev);

BEV_LOCK(bufev);

++bufev_private->refcnt;

}

#if 0

static void

bufferevent_transfer_lock_ownership_(struct bufferevent *donor,

struct bufferevent *recipient)

{

struct bufferevent_private *d = BEV_UPCAST(donor);

struct bufferevent_private *r = BEV_UPCAST(recipient);

if (d->lock != r->lock)

return;

if (r->own_lock)

return;

if (d->own_lock) {

d->own_lock = 0;

r->own_lock = 1;

}

}

#endif

int

bufferevent_decref_and_unlock_(struct bufferevent *bufev)

{

struct bufferevent_private *bufev_private = BEV_UPCAST(bufev);

int n_cbs = 0;

#define MAX_CBS 16

struct event_callback *cbs[MAX_CBS];

EVUTIL_ASSERT(bufev_private->refcnt > 0);

if (--bufev_private->refcnt) {

BEV_UNLOCK(bufev);

return 0;

}

if (bufev->be_ops->unlink)

bufev->be_ops->unlink(bufev);

/* Okay, we're out of references. Let's finalize this once all the

* callbacks are done running. */

cbs[0] = &bufev->ev_read.ev_evcallback;

cbs[1] = &bufev->ev_write.ev_evcallback;

cbs[2] = &bufev_private->deferred;

n_cbs = 3;

if (bufev_private->rate_limiting) {

struct event *e = &bufev_private->rate_limiting->refill_bucket_event;

if (event_initialized(e))

cbs[n_cbs++] = &e->ev_evcallback;

}

n_cbs += evbuffer_get_callbacks_(bufev->input, cbs+n_cbs, MAX_CBS-n_cbs);

n_cbs += evbuffer_get_callbacks_(bufev->output, cbs+n_cbs, MAX_CBS-n_cbs);

event_callback_finalize_many_(bufev->ev_base, n_cbs, cbs,

bufferevent_finalize_cb_);

#undef MAX_CBS

BEV_UNLOCK(bufev);

return 1;

}

static void

bufferevent_finalize_cb_(struct event_callback *evcb, void *arg_)

{

struct bufferevent *bufev = arg_;

struct bufferevent *underlying;

struct bufferevent_private *bufev_private = BEV_UPCAST(bufev);

BEV_LOCK(bufev);

underlying = bufferevent_get_underlying(bufev);

/* Clean up the shared info */

if (bufev->be_ops->destruct)

bufev->be_ops->destruct(bufev);

/* XXX what happens if refcnt for these buffers is > 1?

* The buffers can share a lock with this bufferevent object,

* but the lock might be destroyed below. */

/* evbuffer will free the callbacks */

evbuffer_free(bufev->input);

evbuffer_free(bufev->output);

if (bufev_private->rate_limiting) {

if (bufev_private->rate_limiting->group)

bufferevent_remove_from_rate_limit_group_internal_(bufev,0);

mm_free(bufev_private->rate_limiting);

bufev_private->rate_limiting = NULL;

}

BEV_UNLOCK(bufev);

if (bufev_private->own_lock)

EVTHREAD_FREE_LOCK(bufev_private->lock,

EVTHREAD_LOCKTYPE_RECURSIVE);

/* Free the actual allocated memory. */

mm_free(((char*)bufev) - bufev->be_ops->mem_offset);

/* Release the reference to underlying now that we no longer need the

* reference to it. We wait this long mainly in case our lock is

* shared with underlying.

*

* The 'destruct' function will also drop a reference to underlying

* if BEV_OPT_CLOSE_ON_FREE is set.

*

* XXX Should we/can we just refcount evbuffer/bufferevent locks?

* It would probably save us some headaches.

*/

if (underlying)

bufferevent_decref_(underlying);

}

int

bufferevent_decref(struct bufferevent *bufev)

{

BEV_LOCK(bufev);

return bufferevent_decref_and_unlock_(bufev);

}

void

bufferevent_free(struct bufferevent *bufev)

{

BEV_LOCK(bufev);

bufferevent_setcb(bufev, NULL, NULL, NULL, NULL);

bufferevent_cancel_all_(bufev);

bufferevent_decref_and_unlock_(bufev);

}

void

bufferevent_incref(struct bufferevent *bufev)

{

struct bufferevent_private *bufev_private = BEV_UPCAST(bufev);

/* XXX: now that this function is public, we might want to

* - return the count from this function

* - create a new function to atomically grab the current refcount

*/

BEV_LOCK(bufev);

++bufev_private->refcnt;

BEV_UNLOCK(bufev);

}

int

bufferevent_enable_locking_(struct bufferevent *bufev, void *lock)

{

#ifdef EVENT__DISABLE_THREAD_SUPPORT

return -1;

#else

struct bufferevent *underlying;

if (BEV_UPCAST(bufev)->lock)

return -1;

underlying = bufferevent_get_underlying(bufev);

if (!lock && underlying && BEV_UPCAST(underlying)->lock) {

lock = BEV_UPCAST(underlying)->lock;

BEV_UPCAST(bufev)->lock = lock;

BEV_UPCAST(bufev)->own_lock = 0;

} else if (!lock) {

EVTHREAD_ALLOC_LOCK(lock, EVTHREAD_LOCKTYPE_RECURSIVE);

if (!lock)

return -1;

BEV_UPCAST(bufev)->lock = lock;

BEV_UPCAST(bufev)->own_lock = 1;

} else {

BEV_UPCAST(bufev)->lock = lock;

BEV_UPCAST(bufev)->own_lock = 0;

}

evbuffer_enable_locking(bufev->input, lock);

evbuffer_enable_locking(bufev->output, lock);

if (underlying && !BEV_UPCAST(underlying)->lock)

bufferevent_enable_locking_(underlying, lock);

return 0;

#endif

}

int

bufferevent_setfd(struct bufferevent *bev, evutil_socket_t fd)

{

union bufferevent_ctrl_data d;

int res = -1;

d.fd = fd;

BEV_LOCK(bev);

if (bev->be_ops->ctrl)

res = bev->be_ops->ctrl(bev, BEV_CTRL_SET_FD, &d);

if (res)

event_debug(("%s: cannot set fd for %p to "EV_SOCK_FMT, __func__, (void *)bev, fd));

BEV_UNLOCK(bev);

return res;

}

int

bufferevent_replacefd(struct bufferevent *bev, evutil_socket_t fd)

{

union bufferevent_ctrl_data d;

int err = -1;

evutil_socket_t old_fd = EVUTIL_INVALID_SOCKET;

BEV_LOCK(bev);

if (bev->be_ops->ctrl) {

err = bev->be_ops->ctrl(bev, BEV_CTRL_GET_FD, &d);

if (!err) {

old_fd = d.fd;

if (old_fd != EVUTIL_INVALID_SOCKET) {

err = evutil_closesocket(old_fd);

}

}

if (!err) {

d.fd = fd;

err = bev->be_ops->ctrl(bev, BEV_CTRL_SET_FD, &d);

}

}

if (err)

event_debug(("%s: cannot replace fd for %p from "EV_SOCK_FMT" to "EV_SOCK_FMT, __func__, (void *)bev, old_fd, fd));

BEV_UNLOCK(bev);

return err;

}

evutil_socket_t

bufferevent_getfd(struct bufferevent *bev)

{

union bufferevent_ctrl_data d;

int res = -1;

d.fd = -1;

BEV_LOCK(bev);

if (bev->be_ops->ctrl)

res = bev->be_ops->ctrl(bev, BEV_CTRL_GET_FD, &d);

if (res)

event_debug(("%s: cannot get fd for %p", __func__, (void *)bev));

BEV_UNLOCK(bev);

return (res<0) ? -1 : d.fd;

}

enum bufferevent_options

bufferevent_get_options_(struct bufferevent *bev)

{

struct bufferevent_private *bev_p = BEV_UPCAST(bev);

enum bufferevent_options options;

BEV_LOCK(bev);

options = bev_p->options;

BEV_UNLOCK(bev);

return options;

}

static void

bufferevent_cancel_all_(struct bufferevent *bev)

{

union bufferevent_ctrl_data d;

memset(&d, 0, sizeof(d));

BEV_LOCK(bev);

if (bev->be_ops->ctrl)

bev->be_ops->ctrl(bev, BEV_CTRL_CANCEL_ALL, &d);

BEV_UNLOCK(bev);

}

short

bufferevent_get_enabled(struct bufferevent *bufev)

{

short r;

BEV_LOCK(bufev);

r = bufev->enabled;

BEV_UNLOCK(bufev);

return r;

}

struct bufferevent *

bufferevent_get_underlying(struct bufferevent *bev)

{

union bufferevent_ctrl_data d;

int res = -1;

d.ptr = NULL;

BEV_LOCK(bev);

if (bev->be_ops->ctrl)

res = bev->be_ops->ctrl(bev, BEV_CTRL_GET_UNDERLYING, &d);

BEV_UNLOCK(bev);

return (res<0) ? NULL : d.ptr;

}

static void

bufferevent_generic_read_timeout_cb(evutil_socket_t fd, short event, void *ctx)

{

struct bufferevent *bev = ctx;

bufferevent_incref_and_lock_(bev);

bufferevent_disable(bev, EV_READ);

bufferevent_run_eventcb_(bev, BEV_EVENT_TIMEOUT|BEV_EVENT_READING, 0);

bufferevent_decref_and_unlock_(bev);

}

static void

bufferevent_generic_write_timeout_cb(evutil_socket_t fd, short event, void *ctx)

{

struct bufferevent *bev = ctx;

bufferevent_incref_and_lock_(bev);

bufferevent_disable(bev, EV_WRITE);

bufferevent_run_eventcb_(bev, BEV_EVENT_TIMEOUT|BEV_EVENT_WRITING, 0);

bufferevent_decref_and_unlock_(bev);

}

void

bufferevent_init_generic_timeout_cbs_(struct bufferevent *bev)

{

event_assign(&bev->ev_read, bev->ev_base, -1, EV_FINALIZE,

bufferevent_generic_read_timeout_cb, bev);

event_assign(&bev->ev_write, bev->ev_base, -1, EV_FINALIZE,

bufferevent_generic_write_timeout_cb, bev);

}