kc3-lang/libevent/bufferevent.c

Tag - branch -mastergithub/HEADgithub/dependabot/github_actions/github-actions-c04319c41egithub/mastergithub/patches-1.4github/patches-2.0github/patches-2.1github/pr/529-reproducerorigin/masterupstream/HEADupstream/dependabot/github_actions/github-actions-c04319c41eupstream/masterupstream/patches-1.4upstream/patches-2.0upstream/patches-2.1upstream/pr/529-reproducer- tag -release-2.2.1-alpharelease-2.1.9-betarelease-2.1.8-stablerelease-2.1.7-rcrelease-2.1.6-betarelease-2.1.5-betarelease-2.1.4-alpharelease-2.1.3-alpharelease-2.1.2-alpharelease-2.1.12-stablerelease-2.1.11-stablerelease-2.1.10-stablerelease-2.1.1-alpharelease-2.0.9-rcrelease-2.0.8-rcrelease-2.0.7-rcrelease-2.0.6-rcrelease-2.0.5-betarelease-2.0.4-alpharelease-2.0.3-alpharelease-2.0.23-betarelease-2.0.22-stablerelease-2.0.21-stablerelease-2.0.20-stablerelease-2.0.19-stablerelease-2.0.18-stablerelease-2.0.17-stablerelease-2.0.16-stablerelease-2.0.15-stablerelease-2.0.14-stablerelease-2.0.13-stablerelease-2.0.12-stablerelease-2.0.11-stablerelease-2.0.10-stablerelease-2.0.1-alpharelease-1.4.9-stablerelease-1.4.8-stablerelease-1.4.7-stablerelease-1.4.6release-1.4.5-stablerelease-1.4.4-stablerelease-1.4.3-stablerelease-1.4.2-rcrelease-1.4.15-stablerelease-1.4.14b-stablerelease-1.4.14-stablerelease-1.4.13-stablerelease-1.4.12-stablerelease-1.4.11-stablerelease-1.4.10-stablerelease-1.4.1-betarelease-1.4.0-betarelease-1.1b

• Show log

  Commit

• Hash : 5dc56628
  Author :
  Date : 2011-02-24T12:30:40
  

  Workaround libevent bug
  https://sourceforge.net/tracker/index.php?func=detail&aid=3078187&group_id=50884&atid=461324
  
  The problem is that bufferevent_disable() doesn't disable EV_WRITE
  when 'connecting' flag is set.  However from evhttp_connection_reset()
  we want to disable EV_WRITE for sure (we are closing the socket next).
  So we add bufferevent_disable_hard(), which acts like
  bufferevent_disable(), but resets 'connecting' flag before the call to
  the actual handler.
  
  TODO: bufferevent_disable_hard() shouldn't be public, remove it from
  event2/bufferevent.h.
  

Download

• bufferevent.c

• /*
   * Copyright (c) 2002-2007 Niels Provos <provos@citi.umich.edu>
   * Copyright (c) 2007-2010 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 <sys/types.h>
  
  #include "event2/event-config.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 <errno.h>
  
  #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 "log-internal.h"
  #include "mm-internal.h"
  #include "bufferevent-internal.h"
  #include "evbuffer-internal.h"
  #include "util-internal.h"
  
  void
  bufferevent_suspend_read(struct bufferevent *bufev, bufferevent_suspend_flags what)
  {
  	struct bufferevent_private *bufev_private =
  	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
  	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 =
  	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
  	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 =
  	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
  	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 =
  	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
  	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);
  }
  
  
  /* 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 deferred_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);
  	}
  	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 deferred_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));
  	}
  	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 {								\
  		bufferevent_incref(&(bevp)->bev);			\
  		event_deferred_cb_schedule(				\
  			event_base_get_deferred_cb_queue((bevp)->bev.ev_base), \
  			&(bevp)->deferred);				\
  	} while (0)
  
  
  void
  _bufferevent_run_readcb(struct bufferevent *bufev)
  {
  	/* Requires that we hold the lock and a reference */
  	struct bufferevent_private *p =
  	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
  	if (bufev->readcb == NULL)
  		return;
  	if (p->options & BEV_OPT_DEFER_CALLBACKS) {
  		p->readcb_pending = 1;
  		if (!p->deferred.queued)
  			SCHEDULE_DEFERRED(p);
  	} else {
  		bufev->readcb(bufev, bufev->cbarg);
  	}
  }
  
  void
  _bufferevent_run_writecb(struct bufferevent *bufev)
  {
  	/* Requires that we hold the lock and a reference */
  	struct bufferevent_private *p =
  	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
  	if (bufev->writecb == NULL)
  		return;
  	if (p->options & BEV_OPT_DEFER_CALLBACKS) {
  		p->writecb_pending = 1;
  		if (!p->deferred.queued)
  			SCHEDULE_DEFERRED(p);
  	} else {
  		bufev->writecb(bufev, bufev->cbarg);
  	}
  }
  
  void
  _bufferevent_run_eventcb(struct bufferevent *bufev, short what)
  {
  	/* Requires that we hold the lock and a reference */
  	struct bufferevent_private *p =
  	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
  	if (bufev->errorcb == NULL)
  		return;
  	if (p->options & BEV_OPT_DEFER_CALLBACKS) {
  		p->eventcb_pending |= what;
  		p->errno_pending = EVUTIL_SOCKET_ERROR();
  		if (!p->deferred.queued)
  			SCHEDULE_DEFERRED(p);
  	} else {
  		bufev->errorcb(bufev, what, bufev->cbarg);
  	}
  }
  
  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)
  			return -1;
  	}
  
  	if (!bufev->output) {
  		if ((bufev->output = evbuffer_new()) == NULL) {
  			evbuffer_free(bufev->input);
  			return -1;
  		}
  	}
  
  	bufev_private->refcnt = 1;
  	bufev->ev_base = base;
  
  	/* Disable timeouts. */
  	evutil_timerclear(&bufev->timeout_read);
  	evutil_timerclear(&bufev->timeout_write);
  
  	bufev->be_ops = ops;
  
  	/*
  	 * 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) {
  			/* cleanup */
  			evbuffer_free(bufev->input);
  			evbuffer_free(bufev->output);
  			bufev->input = NULL;
  			bufev->output = NULL;
  			return -1;
  		}
  	}
  #endif
  	if ((options & (BEV_OPT_DEFER_CALLBACKS|BEV_OPT_UNLOCK_CALLBACKS))
  	    == BEV_OPT_UNLOCK_CALLBACKS) {
  		event_warnx("UNLOCK_CALLBACKS requires DEFER_CALLBACKS");
  		return -1;
  	}
  	if (options & BEV_OPT_DEFER_CALLBACKS) {
  		if (options & BEV_OPT_UNLOCK_CALLBACKS)
  			event_deferred_cb_init(&bufev_private->deferred,
  			    bufferevent_run_deferred_callbacks_unlocked,
  			    bufev_private);
  		else
  			event_deferred_cb_init(&bufev_private->deferred,
  			    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;
  }
  
  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);
  }
  
  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_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)
  {
  	return (evbuffer_remove(bufev->input, data, size));
  }
  
  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 =
  	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
  	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;
  
  	_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 =
  	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
  
  	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;
  
  	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 =
  	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
  
  	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_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 =
  	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
  	struct bufferevent *underlying;
  
  	EVUTIL_ASSERT(bufev_private->refcnt > 0);
  
  	if (--bufev_private->refcnt) {
  		BEV_UNLOCK(bufev);
  		return 0;
  	}
  
  	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);
  		if (event_initialized(&bufev_private->rate_limiting->refill_bucket_event))
  			event_del(&bufev_private->rate_limiting->refill_bucket_event);
  		event_debug_unassign(&bufev_private->rate_limiting->refill_bucket_event);
  		mm_free(bufev_private->rate_limiting);
  		bufev_private->rate_limiting = NULL;
  	}
  
  	event_debug_unassign(&bufev->ev_read);
  	event_debug_unassign(&bufev->ev_write);
  
  	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);
  
  	return 1;
  }
  
  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_decref_and_unlock(bufev);
  }
  
  void
  bufferevent_incref(struct bufferevent *bufev)
  {
  	struct bufferevent_private *bufev_private =
  	    EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
  
  	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);
  	BEV_UNLOCK(bev);
  	return res;
  }
  
  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);
  	BEV_UNLOCK(bev);
  	return (res<0) ? -1 : d.fd;
  }
  
  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);
  	_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);
  	_bufferevent_decref_and_unlock(bev);
  }
  
  void
  _bufferevent_init_generic_timeout_cbs(struct bufferevent *bev)
  {
  	evtimer_assign(&bev->ev_read, bev->ev_base,
  	    bufferevent_generic_read_timeout_cb, bev);
  	evtimer_assign(&bev->ev_write, bev->ev_base,
  	    bufferevent_generic_write_timeout_cb, bev);
  }
  
  int
  _bufferevent_del_generic_timeout_cbs(struct bufferevent *bev)
  {
  	int r1,r2;
  	r1 = event_del(&bev->ev_read);
  	r2 = event_del(&bev->ev_write);
  	if (r1<0 || r2<0)
  		return -1;
  	return 0;
  }
  
  int
  _bufferevent_generic_adj_timeouts(struct bufferevent *bev)
  {
  	const short enabled = bev->enabled;
  	struct bufferevent_private *bev_p =
  	    EVUTIL_UPCAST(bev, struct bufferevent_private, bev);
  	int r1=0, r2=0;
  	if ((enabled & EV_READ) && !bev_p->read_suspended &&
  	    evutil_timerisset(&bev->timeout_read))
  		r1 = event_add(&bev->ev_read, &bev->timeout_read);
  	else
  		r1 = event_del(&bev->ev_read);
  
  	if ((enabled & EV_WRITE) && !bev_p->write_suspended &&
  	    evutil_timerisset(&bev->timeout_write) &&
  	    evbuffer_get_length(bev->output))
  		r2 = event_add(&bev->ev_write, &bev->timeout_write);
  	else
  		r2 = event_del(&bev->ev_write);
  	if (r1 < 0 || r2 < 0)
  		return -1;
  	return 0;
  }
  
  int
  _bufferevent_add_event(struct event *ev, const struct timeval *tv)
  {
  	if (tv->tv_sec == 0 && tv->tv_usec == 0)
  		return event_add(ev, NULL);
  	else
  		return event_add(ev, tv);
  }
  
  /* For use by user programs only; internally, we should be calling
     either _bufferevent_incref_and_lock(), or BEV_LOCK. */
  void
  bufferevent_lock(struct bufferevent *bev)
  {
  	_bufferevent_incref_and_lock(bev);
  }
  
  void
  bufferevent_unlock(struct bufferevent *bev)
  {
  	_bufferevent_decref_and_unlock(bev);
  }
  

Download