Home | History | Annotate | Line # | Download | only in libevent
bufferevent_openssl.c revision 1.1.1.1.6.1
      1  1.1.1.1.6.1       riz /*	$NetBSD: bufferevent_openssl.c,v 1.1.1.1.6.1 2014/12/24 00:05:25 riz Exp $	*/
      2          1.1  christos 
      3          1.1  christos /*
      4          1.1  christos  * Copyright (c) 2009-2012 Niels Provos and Nick Mathewson
      5          1.1  christos  *
      6          1.1  christos  * Redistribution and use in source and binary forms, with or without
      7          1.1  christos  * modification, are permitted provided that the following conditions
      8          1.1  christos  * are met:
      9          1.1  christos  * 1. Redistributions of source code must retain the above copyright
     10          1.1  christos  *    notice, this list of conditions and the following disclaimer.
     11          1.1  christos  * 2. Redistributions in binary form must reproduce the above copyright
     12          1.1  christos  *    notice, this list of conditions and the following disclaimer in the
     13          1.1  christos  *    documentation and/or other materials provided with the distribution.
     14          1.1  christos  * 3. The name of the author may not be used to endorse or promote products
     15          1.1  christos  *    derived from this software without specific prior written permission.
     16          1.1  christos  *
     17          1.1  christos  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     18          1.1  christos  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     19          1.1  christos  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     20          1.1  christos  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     21          1.1  christos  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     22          1.1  christos  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     23          1.1  christos  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     24          1.1  christos  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     25          1.1  christos  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     26          1.1  christos  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     27          1.1  christos  */
     28          1.1  christos 
     29  1.1.1.1.6.1       riz // Get rid of OSX 10.7 and greater deprecation warnings.
     30  1.1.1.1.6.1       riz #if defined(__APPLE__) && defined(__clang__)
     31  1.1.1.1.6.1       riz #pragma clang diagnostic ignored "-Wdeprecated-declarations"
     32  1.1.1.1.6.1       riz #endif
     33  1.1.1.1.6.1       riz 
     34          1.1  christos #include "event2/event-config.h"
     35          1.1  christos #include "evconfig-private.h"
     36          1.1  christos 
     37          1.1  christos #include <sys/types.h>
     38          1.1  christos 
     39          1.1  christos #ifdef EVENT__HAVE_SYS_TIME_H
     40          1.1  christos #include <sys/time.h>
     41          1.1  christos #endif
     42          1.1  christos 
     43          1.1  christos #include <errno.h>
     44          1.1  christos #include <stdio.h>
     45          1.1  christos #include <stdlib.h>
     46          1.1  christos #include <string.h>
     47          1.1  christos #ifdef EVENT__HAVE_STDARG_H
     48          1.1  christos #include <stdarg.h>
     49          1.1  christos #endif
     50          1.1  christos #ifdef EVENT__HAVE_UNISTD_H
     51          1.1  christos #include <unistd.h>
     52          1.1  christos #endif
     53          1.1  christos 
     54          1.1  christos #ifdef _WIN32
     55          1.1  christos #include <winsock2.h>
     56          1.1  christos #endif
     57          1.1  christos 
     58          1.1  christos #include "event2/bufferevent.h"
     59          1.1  christos #include "event2/bufferevent_struct.h"
     60          1.1  christos #include "event2/bufferevent_ssl.h"
     61          1.1  christos #include "event2/buffer.h"
     62          1.1  christos #include "event2/event.h"
     63          1.1  christos 
     64          1.1  christos #include "mm-internal.h"
     65          1.1  christos #include "bufferevent-internal.h"
     66          1.1  christos #include "log-internal.h"
     67          1.1  christos 
     68          1.1  christos #include <openssl/bio.h>
     69          1.1  christos #include <openssl/ssl.h>
     70          1.1  christos #include <openssl/err.h>
     71          1.1  christos 
     72          1.1  christos /*
     73          1.1  christos  * Define an OpenSSL bio that targets a bufferevent.
     74          1.1  christos  */
     75          1.1  christos 
     76          1.1  christos /* --------------------
     77          1.1  christos    A BIO is an OpenSSL abstraction that handles reading and writing data.  The
     78          1.1  christos    library will happily speak SSL over anything that implements a BIO
     79          1.1  christos    interface.
     80          1.1  christos 
     81          1.1  christos    Here we define a BIO implementation that directs its output to a
     82          1.1  christos    bufferevent.  We'll want to use this only when none of OpenSSL's built-in
     83          1.1  christos    IO mechanisms work for us.
     84          1.1  christos    -------------------- */
     85          1.1  christos 
     86          1.1  christos /* every BIO type needs its own integer type value. */
     87          1.1  christos #define BIO_TYPE_LIBEVENT 57
     88          1.1  christos /* ???? Arguably, we should set BIO_TYPE_FILTER or BIO_TYPE_SOURCE_SINK on
     89          1.1  christos  * this. */
     90          1.1  christos 
     91          1.1  christos #if 0
     92          1.1  christos static void
     93          1.1  christos print_err(int val)
     94          1.1  christos {
     95          1.1  christos 	int err;
     96          1.1  christos 	printf("Error was %d\n", val);
     97          1.1  christos 
     98          1.1  christos 	while ((err = ERR_get_error())) {
     99          1.1  christos 		const char *msg = (const char*)ERR_reason_error_string(err);
    100          1.1  christos 		const char *lib = (const char*)ERR_lib_error_string(err);
    101          1.1  christos 		const char *func = (const char*)ERR_func_error_string(err);
    102          1.1  christos 
    103          1.1  christos 		printf("%s in %s %s\n", msg, lib, func);
    104          1.1  christos 	}
    105          1.1  christos }
    106          1.1  christos #else
    107          1.1  christos #define print_err(v) ((void)0)
    108          1.1  christos #endif
    109          1.1  christos 
    110          1.1  christos /* Called to initialize a new BIO */
    111          1.1  christos static int
    112          1.1  christos bio_bufferevent_new(BIO *b)
    113          1.1  christos {
    114          1.1  christos 	b->init = 0;
    115          1.1  christos 	b->num = -1;
    116          1.1  christos 	b->ptr = NULL; /* We'll be putting the bufferevent in this field.*/
    117          1.1  christos 	b->flags = 0;
    118          1.1  christos 	return 1;
    119          1.1  christos }
    120          1.1  christos 
    121          1.1  christos /* Called to uninitialize the BIO. */
    122          1.1  christos static int
    123          1.1  christos bio_bufferevent_free(BIO *b)
    124          1.1  christos {
    125          1.1  christos 	if (!b)
    126          1.1  christos 		return 0;
    127          1.1  christos 	if (b->shutdown) {
    128          1.1  christos 		if (b->init && b->ptr)
    129          1.1  christos 			bufferevent_free(b->ptr);
    130          1.1  christos 		b->init = 0;
    131          1.1  christos 		b->flags = 0;
    132          1.1  christos 		b->ptr = NULL;
    133          1.1  christos 	}
    134          1.1  christos 	return 1;
    135          1.1  christos }
    136          1.1  christos 
    137          1.1  christos /* Called to extract data from the BIO. */
    138          1.1  christos static int
    139          1.1  christos bio_bufferevent_read(BIO *b, char *out, int outlen)
    140          1.1  christos {
    141          1.1  christos 	int r = 0;
    142          1.1  christos 	struct evbuffer *input;
    143          1.1  christos 
    144          1.1  christos 	BIO_clear_retry_flags(b);
    145          1.1  christos 
    146          1.1  christos 	if (!out)
    147          1.1  christos 		return 0;
    148          1.1  christos 	if (!b->ptr)
    149          1.1  christos 		return -1;
    150          1.1  christos 
    151          1.1  christos 	input = bufferevent_get_input(b->ptr);
    152          1.1  christos 	if (evbuffer_get_length(input) == 0) {
    153          1.1  christos 		/* If there's no data to read, say so. */
    154          1.1  christos 		BIO_set_retry_read(b);
    155          1.1  christos 		return -1;
    156          1.1  christos 	} else {
    157          1.1  christos 		r = evbuffer_remove(input, out, outlen);
    158          1.1  christos 	}
    159          1.1  christos 
    160          1.1  christos 	return r;
    161          1.1  christos }
    162          1.1  christos 
    163          1.1  christos /* Called to write data info the BIO */
    164          1.1  christos static int
    165          1.1  christos bio_bufferevent_write(BIO *b, const char *in, int inlen)
    166          1.1  christos {
    167          1.1  christos 	struct bufferevent *bufev = b->ptr;
    168          1.1  christos 	struct evbuffer *output;
    169          1.1  christos 	size_t outlen;
    170          1.1  christos 
    171          1.1  christos 	BIO_clear_retry_flags(b);
    172          1.1  christos 
    173          1.1  christos 	if (!b->ptr)
    174          1.1  christos 		return -1;
    175          1.1  christos 
    176          1.1  christos 	output = bufferevent_get_output(bufev);
    177          1.1  christos 	outlen = evbuffer_get_length(output);
    178          1.1  christos 
    179          1.1  christos 	/* Copy only as much data onto the output buffer as can fit under the
    180          1.1  christos 	 * high-water mark. */
    181          1.1  christos 	if (bufev->wm_write.high && bufev->wm_write.high <= (outlen+inlen)) {
    182          1.1  christos 		if (bufev->wm_write.high <= outlen) {
    183          1.1  christos 			/* If no data can fit, we'll need to retry later. */
    184          1.1  christos 			BIO_set_retry_write(b);
    185          1.1  christos 			return -1;
    186          1.1  christos 		}
    187          1.1  christos 		inlen = bufev->wm_write.high - outlen;
    188          1.1  christos 	}
    189          1.1  christos 
    190          1.1  christos 	EVUTIL_ASSERT(inlen > 0);
    191          1.1  christos 	evbuffer_add(output, in, inlen);
    192          1.1  christos 	return inlen;
    193          1.1  christos }
    194          1.1  christos 
    195          1.1  christos /* Called to handle various requests */
    196          1.1  christos static long
    197          1.1  christos bio_bufferevent_ctrl(BIO *b, int cmd, long num, void *ptr)
    198          1.1  christos {
    199          1.1  christos 	struct bufferevent *bufev = b->ptr;
    200          1.1  christos 	long ret = 1;
    201          1.1  christos 
    202          1.1  christos 	switch (cmd) {
    203          1.1  christos 	case BIO_CTRL_GET_CLOSE:
    204          1.1  christos 		ret = b->shutdown;
    205          1.1  christos 		break;
    206          1.1  christos 	case BIO_CTRL_SET_CLOSE:
    207          1.1  christos 		b->shutdown = (int)num;
    208          1.1  christos 		break;
    209          1.1  christos 	case BIO_CTRL_PENDING:
    210          1.1  christos 		ret = evbuffer_get_length(bufferevent_get_input(bufev)) != 0;
    211          1.1  christos 		break;
    212          1.1  christos 	case BIO_CTRL_WPENDING:
    213          1.1  christos 		ret = evbuffer_get_length(bufferevent_get_output(bufev)) != 0;
    214          1.1  christos 		break;
    215          1.1  christos 	/* XXXX These two are given a special-case treatment because
    216          1.1  christos 	 * of cargo-cultism.  I should come up with a better reason. */
    217          1.1  christos 	case BIO_CTRL_DUP:
    218          1.1  christos 	case BIO_CTRL_FLUSH:
    219          1.1  christos 		ret = 1;
    220          1.1  christos 		break;
    221          1.1  christos 	default:
    222          1.1  christos 		ret = 0;
    223          1.1  christos 		break;
    224          1.1  christos 	}
    225          1.1  christos 	return ret;
    226          1.1  christos }
    227          1.1  christos 
    228          1.1  christos /* Called to write a string to the BIO */
    229          1.1  christos static int
    230          1.1  christos bio_bufferevent_puts(BIO *b, const char *s)
    231          1.1  christos {
    232          1.1  christos 	return bio_bufferevent_write(b, s, strlen(s));
    233          1.1  christos }
    234          1.1  christos 
    235          1.1  christos /* Method table for the bufferevent BIO */
    236          1.1  christos static BIO_METHOD methods_bufferevent = {
    237          1.1  christos 	BIO_TYPE_LIBEVENT, "bufferevent",
    238          1.1  christos 	bio_bufferevent_write,
    239          1.1  christos 	bio_bufferevent_read,
    240          1.1  christos 	bio_bufferevent_puts,
    241          1.1  christos 	NULL /* bio_bufferevent_gets */,
    242          1.1  christos 	bio_bufferevent_ctrl,
    243          1.1  christos 	bio_bufferevent_new,
    244          1.1  christos 	bio_bufferevent_free,
    245          1.1  christos 	NULL /* callback_ctrl */,
    246          1.1  christos };
    247          1.1  christos 
    248          1.1  christos /* Return the method table for the bufferevents BIO */
    249          1.1  christos static BIO_METHOD *
    250          1.1  christos BIO_s_bufferevent(void)
    251          1.1  christos {
    252          1.1  christos 	return &methods_bufferevent;
    253          1.1  christos }
    254          1.1  christos 
    255          1.1  christos /* Create a new BIO to wrap communication around a bufferevent.  If close_flag
    256          1.1  christos  * is true, the bufferevent will be freed when the BIO is closed. */
    257          1.1  christos static BIO *
    258          1.1  christos BIO_new_bufferevent(struct bufferevent *bufferevent, int close_flag)
    259          1.1  christos {
    260          1.1  christos 	BIO *result;
    261          1.1  christos 	if (!bufferevent)
    262          1.1  christos 		return NULL;
    263          1.1  christos 	if (!(result = BIO_new(BIO_s_bufferevent())))
    264          1.1  christos 		return NULL;
    265          1.1  christos 	result->init = 1;
    266          1.1  christos 	result->ptr = bufferevent;
    267          1.1  christos 	result->shutdown = close_flag ? 1 : 0;
    268          1.1  christos 	return result;
    269          1.1  christos }
    270          1.1  christos 
    271          1.1  christos /* --------------------
    272          1.1  christos    Now, here's the OpenSSL-based implementation of bufferevent.
    273          1.1  christos 
    274          1.1  christos    The implementation comes in two flavors: one that connects its SSL object
    275          1.1  christos    to an underlying bufferevent using a BIO_bufferevent, and one that has the
    276          1.1  christos    SSL object connect to a socket directly.  The latter should generally be
    277          1.1  christos    faster, except on Windows, where your best bet is using a
    278          1.1  christos    bufferevent_async.
    279          1.1  christos 
    280          1.1  christos    (OpenSSL supports many other BIO types, too.  But we can't use any unless
    281          1.1  christos    we have a good way to get notified when they become readable/writable.)
    282          1.1  christos    -------------------- */
    283          1.1  christos 
    284          1.1  christos struct bio_data_counts {
    285          1.1  christos 	unsigned long n_written;
    286          1.1  christos 	unsigned long n_read;
    287          1.1  christos };
    288          1.1  christos 
    289          1.1  christos struct bufferevent_openssl {
    290          1.1  christos 	/* Shared fields with common bufferevent implementation code.
    291          1.1  christos 	   If we were set up with an underlying bufferevent, we use the
    292          1.1  christos 	   events here as timers only.  If we have an SSL, then we use
    293          1.1  christos 	   the events as socket events.
    294          1.1  christos 	 */
    295          1.1  christos 	struct bufferevent_private bev;
    296          1.1  christos 	/* An underlying bufferevent that we're directing our output to.
    297          1.1  christos 	   If it's NULL, then we're connected to an fd, not an evbuffer. */
    298          1.1  christos 	struct bufferevent *underlying;
    299          1.1  christos 	/* The SSL object doing our encryption. */
    300          1.1  christos 	SSL *ssl;
    301          1.1  christos 
    302          1.1  christos 	/* A callback that's invoked when data arrives on our outbuf so we
    303          1.1  christos 	   know to write data to the SSL. */
    304          1.1  christos 	struct evbuffer_cb_entry *outbuf_cb;
    305          1.1  christos 
    306          1.1  christos 	/* A count of how much data the bios have read/written total.  Used
    307          1.1  christos 	   for rate-limiting. */
    308          1.1  christos 	struct bio_data_counts counts;
    309          1.1  christos 
    310          1.1  christos 	/* If this value is greater than 0, then the last SSL_write blocked,
    311          1.1  christos 	 * and we need to try it again with this many bytes. */
    312          1.1  christos 	ev_ssize_t last_write;
    313          1.1  christos 
    314          1.1  christos #define NUM_ERRORS 3
    315          1.1  christos 	ev_uint32_t errors[NUM_ERRORS];
    316          1.1  christos 
    317          1.1  christos 	/* When we next get available space, we should say "read" instead of
    318          1.1  christos 	   "write". This can happen if there's a renegotiation during a read
    319          1.1  christos 	   operation. */
    320          1.1  christos 	unsigned read_blocked_on_write : 1;
    321          1.1  christos 	/* When we next get data, we should say "write" instead of "read". */
    322          1.1  christos 	unsigned write_blocked_on_read : 1;
    323          1.1  christos 	/* Treat TCP close before SSL close on SSL >= v3 as clean EOF. */
    324          1.1  christos 	unsigned allow_dirty_shutdown : 1;
    325          1.1  christos 	/* XXXX */
    326          1.1  christos 	unsigned fd_is_set : 1;
    327          1.1  christos 	/* XXX */
    328          1.1  christos 	unsigned n_errors : 2;
    329          1.1  christos 
    330          1.1  christos 	/* Are we currently connecting, accepting, or doing IO? */
    331          1.1  christos 	unsigned state : 2;
    332          1.1  christos };
    333          1.1  christos 
    334          1.1  christos static int be_openssl_enable(struct bufferevent *, short);
    335          1.1  christos static int be_openssl_disable(struct bufferevent *, short);
    336  1.1.1.1.6.1       riz static void be_openssl_unlink(struct bufferevent *);
    337          1.1  christos static void be_openssl_destruct(struct bufferevent *);
    338          1.1  christos static int be_openssl_adj_timeouts(struct bufferevent *);
    339          1.1  christos static int be_openssl_flush(struct bufferevent *bufev,
    340          1.1  christos     short iotype, enum bufferevent_flush_mode mode);
    341          1.1  christos static int be_openssl_ctrl(struct bufferevent *, enum bufferevent_ctrl_op, union bufferevent_ctrl_data *);
    342          1.1  christos 
    343          1.1  christos const struct bufferevent_ops bufferevent_ops_openssl = {
    344          1.1  christos 	"ssl",
    345          1.1  christos 	evutil_offsetof(struct bufferevent_openssl, bev.bev),
    346          1.1  christos 	be_openssl_enable,
    347          1.1  christos 	be_openssl_disable,
    348  1.1.1.1.6.1       riz 	be_openssl_unlink,
    349          1.1  christos 	be_openssl_destruct,
    350          1.1  christos 	be_openssl_adj_timeouts,
    351          1.1  christos 	be_openssl_flush,
    352          1.1  christos 	be_openssl_ctrl,
    353          1.1  christos };
    354          1.1  christos 
    355          1.1  christos /* Given a bufferevent, return a pointer to the bufferevent_openssl that
    356          1.1  christos  * contains it, if any. */
    357          1.1  christos static inline struct bufferevent_openssl *
    358          1.1  christos upcast(struct bufferevent *bev)
    359          1.1  christos {
    360          1.1  christos 	struct bufferevent_openssl *bev_o;
    361          1.1  christos 	if (bev->be_ops != &bufferevent_ops_openssl)
    362          1.1  christos 		return NULL;
    363          1.1  christos 	bev_o = (void*)( ((char*)bev) -
    364          1.1  christos 			 evutil_offsetof(struct bufferevent_openssl, bev.bev));
    365          1.1  christos 	EVUTIL_ASSERT(bev_o->bev.bev.be_ops == &bufferevent_ops_openssl);
    366          1.1  christos 	return bev_o;
    367          1.1  christos }
    368          1.1  christos 
    369          1.1  christos static inline void
    370          1.1  christos put_error(struct bufferevent_openssl *bev_ssl, unsigned long err)
    371          1.1  christos {
    372          1.1  christos 	if (bev_ssl->n_errors == NUM_ERRORS)
    373          1.1  christos 		return;
    374          1.1  christos 	/* The error type according to openssl is "unsigned long", but
    375          1.1  christos 	   openssl never uses more than 32 bits of it.  It _can't_ use more
    376          1.1  christos 	   than 32 bits of it, since it needs to report errors on systems
    377          1.1  christos 	   where long is only 32 bits.
    378          1.1  christos 	 */
    379          1.1  christos 	bev_ssl->errors[bev_ssl->n_errors++] = (ev_uint32_t) err;
    380          1.1  christos }
    381          1.1  christos 
    382          1.1  christos /* Have the base communications channel (either the underlying bufferevent or
    383          1.1  christos  * ev_read and ev_write) start reading.  Take the read-blocked-on-write flag
    384          1.1  christos  * into account. */
    385          1.1  christos static int
    386          1.1  christos start_reading(struct bufferevent_openssl *bev_ssl)
    387          1.1  christos {
    388          1.1  christos 	if (bev_ssl->underlying) {
    389          1.1  christos 		bufferevent_unsuspend_read_(bev_ssl->underlying,
    390          1.1  christos 		    BEV_SUSPEND_FILT_READ);
    391          1.1  christos 		return 0;
    392          1.1  christos 	} else {
    393          1.1  christos 		struct bufferevent *bev = &bev_ssl->bev.bev;
    394          1.1  christos 		int r;
    395          1.1  christos 		r = bufferevent_add_event_(&bev->ev_read, &bev->timeout_read);
    396          1.1  christos 		if (r == 0 && bev_ssl->read_blocked_on_write)
    397          1.1  christos 			r = bufferevent_add_event_(&bev->ev_write,
    398          1.1  christos 			    &bev->timeout_write);
    399          1.1  christos 		return r;
    400          1.1  christos 	}
    401          1.1  christos }
    402          1.1  christos 
    403          1.1  christos /* Have the base communications channel (either the underlying bufferevent or
    404          1.1  christos  * ev_read and ev_write) start writing.  Take the write-blocked-on-read flag
    405          1.1  christos  * into account. */
    406          1.1  christos static int
    407          1.1  christos start_writing(struct bufferevent_openssl *bev_ssl)
    408          1.1  christos {
    409          1.1  christos 	int r = 0;
    410          1.1  christos 	if (bev_ssl->underlying) {
    411          1.1  christos 		;
    412          1.1  christos 	} else {
    413          1.1  christos 		struct bufferevent *bev = &bev_ssl->bev.bev;
    414          1.1  christos 		r = bufferevent_add_event_(&bev->ev_write, &bev->timeout_write);
    415          1.1  christos 		if (!r && bev_ssl->write_blocked_on_read)
    416          1.1  christos 			r = bufferevent_add_event_(&bev->ev_read,
    417          1.1  christos 			    &bev->timeout_read);
    418          1.1  christos 	}
    419          1.1  christos 	return r;
    420          1.1  christos }
    421          1.1  christos 
    422          1.1  christos static void
    423          1.1  christos stop_reading(struct bufferevent_openssl *bev_ssl)
    424          1.1  christos {
    425          1.1  christos 	if (bev_ssl->write_blocked_on_read)
    426          1.1  christos 		return;
    427          1.1  christos 	if (bev_ssl->underlying) {
    428          1.1  christos 		bufferevent_suspend_read_(bev_ssl->underlying,
    429          1.1  christos 		    BEV_SUSPEND_FILT_READ);
    430          1.1  christos 	} else {
    431          1.1  christos 		struct bufferevent *bev = &bev_ssl->bev.bev;
    432          1.1  christos 		event_del(&bev->ev_read);
    433          1.1  christos 	}
    434          1.1  christos }
    435          1.1  christos 
    436          1.1  christos static void
    437          1.1  christos stop_writing(struct bufferevent_openssl *bev_ssl)
    438          1.1  christos {
    439          1.1  christos 	if (bev_ssl->read_blocked_on_write)
    440          1.1  christos 		return;
    441          1.1  christos 	if (bev_ssl->underlying) {
    442          1.1  christos 		;
    443          1.1  christos 	} else {
    444          1.1  christos 		struct bufferevent *bev = &bev_ssl->bev.bev;
    445          1.1  christos 		event_del(&bev->ev_write);
    446          1.1  christos 	}
    447          1.1  christos }
    448          1.1  christos 
    449          1.1  christos static int
    450          1.1  christos set_rbow(struct bufferevent_openssl *bev_ssl)
    451          1.1  christos {
    452          1.1  christos 	if (!bev_ssl->underlying)
    453          1.1  christos 		stop_reading(bev_ssl);
    454          1.1  christos 	bev_ssl->read_blocked_on_write = 1;
    455          1.1  christos 	return start_writing(bev_ssl);
    456          1.1  christos }
    457          1.1  christos 
    458          1.1  christos static int
    459          1.1  christos set_wbor(struct bufferevent_openssl *bev_ssl)
    460          1.1  christos {
    461          1.1  christos 	if (!bev_ssl->underlying)
    462          1.1  christos 		stop_writing(bev_ssl);
    463          1.1  christos 	bev_ssl->write_blocked_on_read = 1;
    464          1.1  christos 	return start_reading(bev_ssl);
    465          1.1  christos }
    466          1.1  christos 
    467          1.1  christos static int
    468          1.1  christos clear_rbow(struct bufferevent_openssl *bev_ssl)
    469          1.1  christos {
    470          1.1  christos 	struct bufferevent *bev = &bev_ssl->bev.bev;
    471          1.1  christos 	int r = 0;
    472          1.1  christos 	bev_ssl->read_blocked_on_write = 0;
    473          1.1  christos 	if (!(bev->enabled & EV_WRITE))
    474          1.1  christos 		stop_writing(bev_ssl);
    475          1.1  christos 	if (bev->enabled & EV_READ)
    476          1.1  christos 		r = start_reading(bev_ssl);
    477          1.1  christos 	return r;
    478          1.1  christos }
    479          1.1  christos 
    480          1.1  christos 
    481          1.1  christos static int
    482          1.1  christos clear_wbor(struct bufferevent_openssl *bev_ssl)
    483          1.1  christos {
    484          1.1  christos 	struct bufferevent *bev = &bev_ssl->bev.bev;
    485          1.1  christos 	int r = 0;
    486          1.1  christos 	bev_ssl->write_blocked_on_read = 0;
    487          1.1  christos 	if (!(bev->enabled & EV_READ))
    488          1.1  christos 		stop_reading(bev_ssl);
    489          1.1  christos 	if (bev->enabled & EV_WRITE)
    490          1.1  christos 		r = start_writing(bev_ssl);
    491          1.1  christos 	return r;
    492          1.1  christos }
    493          1.1  christos 
    494          1.1  christos static void
    495          1.1  christos conn_closed(struct bufferevent_openssl *bev_ssl, int when, int errcode, int ret)
    496          1.1  christos {
    497          1.1  christos 	int event = BEV_EVENT_ERROR;
    498          1.1  christos 	int dirty_shutdown = 0;
    499          1.1  christos 	unsigned long err;
    500          1.1  christos 
    501          1.1  christos 	switch (errcode) {
    502          1.1  christos 	case SSL_ERROR_ZERO_RETURN:
    503          1.1  christos 		/* Possibly a clean shutdown. */
    504          1.1  christos 		if (SSL_get_shutdown(bev_ssl->ssl) & SSL_RECEIVED_SHUTDOWN)
    505          1.1  christos 			event = BEV_EVENT_EOF;
    506          1.1  christos 		else
    507          1.1  christos 			dirty_shutdown = 1;
    508          1.1  christos 		break;
    509          1.1  christos 	case SSL_ERROR_SYSCALL:
    510          1.1  christos 		/* IO error; possibly a dirty shutdown. */
    511          1.1  christos 		if (ret == 0 && ERR_peek_error() == 0)
    512          1.1  christos 			dirty_shutdown = 1;
    513          1.1  christos 		break;
    514          1.1  christos 	case SSL_ERROR_SSL:
    515          1.1  christos 		/* Protocol error. */
    516          1.1  christos 		break;
    517          1.1  christos 	case SSL_ERROR_WANT_X509_LOOKUP:
    518          1.1  christos 		/* XXXX handle this. */
    519          1.1  christos 		break;
    520          1.1  christos 	case SSL_ERROR_NONE:
    521          1.1  christos 	case SSL_ERROR_WANT_READ:
    522          1.1  christos 	case SSL_ERROR_WANT_WRITE:
    523          1.1  christos 	case SSL_ERROR_WANT_CONNECT:
    524          1.1  christos 	case SSL_ERROR_WANT_ACCEPT:
    525          1.1  christos 	default:
    526          1.1  christos 		/* should be impossible; treat as normal error. */
    527          1.1  christos 		event_warnx("BUG: Unexpected OpenSSL error code %d", errcode);
    528          1.1  christos 		break;
    529          1.1  christos 	}
    530          1.1  christos 
    531          1.1  christos 	while ((err = ERR_get_error())) {
    532          1.1  christos 		put_error(bev_ssl, err);
    533          1.1  christos 	}
    534          1.1  christos 
    535          1.1  christos 	if (dirty_shutdown && bev_ssl->allow_dirty_shutdown)
    536          1.1  christos 		event = BEV_EVENT_EOF;
    537          1.1  christos 
    538          1.1  christos 	stop_reading(bev_ssl);
    539          1.1  christos 	stop_writing(bev_ssl);
    540          1.1  christos 
    541          1.1  christos 	/* when is BEV_EVENT_{READING|WRITING} */
    542          1.1  christos 	event = when | event;
    543  1.1.1.1.6.1       riz 	bufferevent_run_eventcb_(&bev_ssl->bev.bev, event, 0);
    544          1.1  christos }
    545          1.1  christos 
    546          1.1  christos static void
    547          1.1  christos init_bio_counts(struct bufferevent_openssl *bev_ssl)
    548          1.1  christos {
    549          1.1  christos 	bev_ssl->counts.n_written =
    550          1.1  christos 	    BIO_number_written(SSL_get_wbio(bev_ssl->ssl));
    551          1.1  christos 	bev_ssl->counts.n_read =
    552          1.1  christos 	    BIO_number_read(SSL_get_rbio(bev_ssl->ssl));
    553          1.1  christos }
    554          1.1  christos 
    555          1.1  christos static inline void
    556          1.1  christos decrement_buckets(struct bufferevent_openssl *bev_ssl)
    557          1.1  christos {
    558          1.1  christos 	unsigned long num_w = BIO_number_written(SSL_get_wbio(bev_ssl->ssl));
    559          1.1  christos 	unsigned long num_r = BIO_number_read(SSL_get_rbio(bev_ssl->ssl));
    560          1.1  christos 	/* These next two subtractions can wrap around. That's okay. */
    561          1.1  christos 	unsigned long w = num_w - bev_ssl->counts.n_written;
    562          1.1  christos 	unsigned long r = num_r - bev_ssl->counts.n_read;
    563          1.1  christos 	if (w)
    564          1.1  christos 		bufferevent_decrement_write_buckets_(&bev_ssl->bev, w);
    565          1.1  christos 	if (r)
    566          1.1  christos 		bufferevent_decrement_read_buckets_(&bev_ssl->bev, r);
    567          1.1  christos 	bev_ssl->counts.n_written = num_w;
    568          1.1  christos 	bev_ssl->counts.n_read = num_r;
    569          1.1  christos }
    570          1.1  christos 
    571          1.1  christos #define OP_MADE_PROGRESS 1
    572          1.1  christos #define OP_BLOCKED 2
    573          1.1  christos #define OP_ERR 4
    574          1.1  christos 
    575          1.1  christos /* Return a bitmask of OP_MADE_PROGRESS (if we read anything); OP_BLOCKED (if
    576          1.1  christos    we're now blocked); and OP_ERR (if an error occurred). */
    577          1.1  christos static int
    578          1.1  christos do_read(struct bufferevent_openssl *bev_ssl, int n_to_read) {
    579          1.1  christos 	/* Requires lock */
    580          1.1  christos 	struct bufferevent *bev = &bev_ssl->bev.bev;
    581          1.1  christos 	struct evbuffer *input = bev->input;
    582          1.1  christos 	int r, n, i, n_used = 0, atmost;
    583          1.1  christos 	struct evbuffer_iovec space[2];
    584          1.1  christos 	int result = 0;
    585          1.1  christos 
    586          1.1  christos 	if (bev_ssl->bev.read_suspended)
    587          1.1  christos 		return 0;
    588          1.1  christos 
    589          1.1  christos 	atmost = bufferevent_get_read_max_(&bev_ssl->bev);
    590          1.1  christos 	if (n_to_read > atmost)
    591          1.1  christos 		n_to_read = atmost;
    592          1.1  christos 
    593          1.1  christos 	n = evbuffer_reserve_space(input, n_to_read, space, 2);
    594          1.1  christos 	if (n < 0)
    595          1.1  christos 		return OP_ERR;
    596          1.1  christos 
    597          1.1  christos 	for (i=0; i<n; ++i) {
    598          1.1  christos 		if (bev_ssl->bev.read_suspended)
    599          1.1  christos 			break;
    600          1.1  christos 		r = SSL_read(bev_ssl->ssl, space[i].iov_base, space[i].iov_len);
    601          1.1  christos 		if (r>0) {
    602          1.1  christos 			result |= OP_MADE_PROGRESS;
    603          1.1  christos 			if (bev_ssl->read_blocked_on_write)
    604          1.1  christos 				if (clear_rbow(bev_ssl) < 0)
    605          1.1  christos 					return OP_ERR | result;
    606          1.1  christos 			++n_used;
    607          1.1  christos 			space[i].iov_len = r;
    608          1.1  christos 			decrement_buckets(bev_ssl);
    609          1.1  christos 		} else {
    610          1.1  christos 			int err = SSL_get_error(bev_ssl->ssl, r);
    611          1.1  christos 			print_err(err);
    612          1.1  christos 			switch (err) {
    613          1.1  christos 			case SSL_ERROR_WANT_READ:
    614          1.1  christos 				/* Can't read until underlying has more data. */
    615          1.1  christos 				if (bev_ssl->read_blocked_on_write)
    616          1.1  christos 					if (clear_rbow(bev_ssl) < 0)
    617          1.1  christos 						return OP_ERR | result;
    618          1.1  christos 				break;
    619          1.1  christos 			case SSL_ERROR_WANT_WRITE:
    620          1.1  christos 				/* This read operation requires a write, and the
    621          1.1  christos 				 * underlying is full */
    622          1.1  christos 				if (!bev_ssl->read_blocked_on_write)
    623          1.1  christos 					if (set_rbow(bev_ssl) < 0)
    624          1.1  christos 						return OP_ERR | result;
    625          1.1  christos 				break;
    626          1.1  christos 			default:
    627          1.1  christos 				conn_closed(bev_ssl, BEV_EVENT_READING, err, r);
    628          1.1  christos 				break;
    629          1.1  christos 			}
    630          1.1  christos 			result |= OP_BLOCKED;
    631          1.1  christos 			break; /* out of the loop */
    632          1.1  christos 		}
    633          1.1  christos 	}
    634          1.1  christos 
    635          1.1  christos 	if (n_used) {
    636          1.1  christos 		evbuffer_commit_space(input, space, n_used);
    637          1.1  christos 		if (bev_ssl->underlying)
    638          1.1  christos 			BEV_RESET_GENERIC_READ_TIMEOUT(bev);
    639          1.1  christos 	}
    640          1.1  christos 
    641          1.1  christos 	return result;
    642          1.1  christos }
    643          1.1  christos 
    644          1.1  christos /* Return a bitmask of OP_MADE_PROGRESS (if we wrote anything); OP_BLOCKED (if
    645          1.1  christos    we're now blocked); and OP_ERR (if an error occurred). */
    646          1.1  christos static int
    647          1.1  christos do_write(struct bufferevent_openssl *bev_ssl, int atmost)
    648          1.1  christos {
    649          1.1  christos 	int i, r, n, n_written = 0;
    650          1.1  christos 	struct bufferevent *bev = &bev_ssl->bev.bev;
    651          1.1  christos 	struct evbuffer *output = bev->output;
    652          1.1  christos 	struct evbuffer_iovec space[8];
    653          1.1  christos 	int result = 0;
    654          1.1  christos 
    655          1.1  christos 	if (bev_ssl->last_write > 0)
    656          1.1  christos 		atmost = bev_ssl->last_write;
    657          1.1  christos 	else
    658          1.1  christos 		atmost = bufferevent_get_write_max_(&bev_ssl->bev);
    659          1.1  christos 
    660          1.1  christos 	n = evbuffer_peek(output, atmost, NULL, space, 8);
    661          1.1  christos 	if (n < 0)
    662          1.1  christos 		return OP_ERR | result;
    663          1.1  christos 
    664          1.1  christos 	if (n > 8)
    665          1.1  christos 		n = 8;
    666          1.1  christos 	for (i=0; i < n; ++i) {
    667          1.1  christos 		if (bev_ssl->bev.write_suspended)
    668          1.1  christos 			break;
    669          1.1  christos 
    670          1.1  christos 		/* SSL_write will (reasonably) return 0 if we tell it to
    671          1.1  christos 		   send 0 data.  Skip this case so we don't interpret the
    672          1.1  christos 		   result as an error */
    673          1.1  christos 		if (space[i].iov_len == 0)
    674          1.1  christos 			continue;
    675          1.1  christos 
    676          1.1  christos 		r = SSL_write(bev_ssl->ssl, space[i].iov_base,
    677          1.1  christos 		    space[i].iov_len);
    678          1.1  christos 		if (r > 0) {
    679          1.1  christos 			result |= OP_MADE_PROGRESS;
    680          1.1  christos 			if (bev_ssl->write_blocked_on_read)
    681          1.1  christos 				if (clear_wbor(bev_ssl) < 0)
    682          1.1  christos 					return OP_ERR | result;
    683          1.1  christos 			n_written += r;
    684          1.1  christos 			bev_ssl->last_write = -1;
    685          1.1  christos 			decrement_buckets(bev_ssl);
    686          1.1  christos 		} else {
    687          1.1  christos 			int err = SSL_get_error(bev_ssl->ssl, r);
    688          1.1  christos 			print_err(err);
    689          1.1  christos 			switch (err) {
    690          1.1  christos 			case SSL_ERROR_WANT_WRITE:
    691          1.1  christos 				/* Can't read until underlying has more data. */
    692          1.1  christos 				if (bev_ssl->write_blocked_on_read)
    693          1.1  christos 					if (clear_wbor(bev_ssl) < 0)
    694          1.1  christos 						return OP_ERR | result;
    695          1.1  christos 				bev_ssl->last_write = space[i].iov_len;
    696          1.1  christos 				break;
    697          1.1  christos 			case SSL_ERROR_WANT_READ:
    698          1.1  christos 				/* This read operation requires a write, and the
    699          1.1  christos 				 * underlying is full */
    700          1.1  christos 				if (!bev_ssl->write_blocked_on_read)
    701          1.1  christos 					if (set_wbor(bev_ssl) < 0)
    702          1.1  christos 						return OP_ERR | result;
    703          1.1  christos 				bev_ssl->last_write = space[i].iov_len;
    704          1.1  christos 				break;
    705          1.1  christos 			default:
    706          1.1  christos 				conn_closed(bev_ssl, BEV_EVENT_WRITING, err, r);
    707          1.1  christos 				bev_ssl->last_write = -1;
    708          1.1  christos 				break;
    709          1.1  christos 			}
    710          1.1  christos 			result |= OP_BLOCKED;
    711          1.1  christos 			break;
    712          1.1  christos 		}
    713          1.1  christos 	}
    714          1.1  christos 	if (n_written) {
    715          1.1  christos 		evbuffer_drain(output, n_written);
    716          1.1  christos 		if (bev_ssl->underlying)
    717          1.1  christos 			BEV_RESET_GENERIC_WRITE_TIMEOUT(bev);
    718          1.1  christos 
    719  1.1.1.1.6.1       riz 		bufferevent_trigger_nolock_(bev, EV_WRITE, 0);
    720          1.1  christos 	}
    721          1.1  christos 	return result;
    722          1.1  christos }
    723          1.1  christos 
    724          1.1  christos #define WRITE_FRAME 15000
    725          1.1  christos 
    726          1.1  christos #define READ_DEFAULT 4096
    727          1.1  christos 
    728          1.1  christos /* Try to figure out how many bytes to read; return 0 if we shouldn't be
    729          1.1  christos  * reading. */
    730          1.1  christos static int
    731          1.1  christos bytes_to_read(struct bufferevent_openssl *bev)
    732          1.1  christos {
    733          1.1  christos 	struct evbuffer *input = bev->bev.bev.input;
    734          1.1  christos 	struct event_watermark *wm = &bev->bev.bev.wm_read;
    735          1.1  christos 	int result = READ_DEFAULT;
    736          1.1  christos 	ev_ssize_t limit;
    737          1.1  christos 	/* XXX 99% of this is generic code that nearly all bufferevents will
    738          1.1  christos 	 * want. */
    739          1.1  christos 
    740          1.1  christos 	if (bev->write_blocked_on_read) {
    741          1.1  christos 		return 0;
    742          1.1  christos 	}
    743          1.1  christos 
    744          1.1  christos 	if (! (bev->bev.bev.enabled & EV_READ)) {
    745          1.1  christos 		return 0;
    746          1.1  christos 	}
    747          1.1  christos 
    748          1.1  christos 	if (bev->bev.read_suspended) {
    749          1.1  christos 		return 0;
    750          1.1  christos 	}
    751          1.1  christos 
    752          1.1  christos 	if (wm->high) {
    753          1.1  christos 		if (evbuffer_get_length(input) >= wm->high) {
    754          1.1  christos 			return 0;
    755          1.1  christos 		}
    756          1.1  christos 
    757          1.1  christos 		result = wm->high - evbuffer_get_length(input);
    758          1.1  christos 	} else {
    759          1.1  christos 		result = READ_DEFAULT;
    760          1.1  christos 	}
    761          1.1  christos 
    762          1.1  christos 	/* Respect the rate limit */
    763          1.1  christos 	limit = bufferevent_get_read_max_(&bev->bev);
    764          1.1  christos 	if (result > limit) {
    765          1.1  christos 		result = limit;
    766          1.1  christos 	}
    767          1.1  christos 
    768          1.1  christos 	return result;
    769          1.1  christos }
    770          1.1  christos 
    771          1.1  christos 
    772          1.1  christos /* Things look readable.  If write is blocked on read, write till it isn't.
    773          1.1  christos  * Read from the underlying buffer until we block or we hit our high-water
    774          1.1  christos  * mark.
    775          1.1  christos  */
    776          1.1  christos static void
    777          1.1  christos consider_reading(struct bufferevent_openssl *bev_ssl)
    778          1.1  christos {
    779          1.1  christos 	int r;
    780          1.1  christos 	int n_to_read;
    781          1.1  christos 	int all_result_flags = 0;
    782          1.1  christos 
    783          1.1  christos 	while (bev_ssl->write_blocked_on_read) {
    784          1.1  christos 		r = do_write(bev_ssl, WRITE_FRAME);
    785          1.1  christos 		if (r & (OP_BLOCKED|OP_ERR))
    786          1.1  christos 			break;
    787          1.1  christos 	}
    788          1.1  christos 	if (bev_ssl->write_blocked_on_read)
    789          1.1  christos 		return;
    790          1.1  christos 
    791          1.1  christos 	n_to_read = bytes_to_read(bev_ssl);
    792          1.1  christos 
    793          1.1  christos 	while (n_to_read) {
    794          1.1  christos 		r = do_read(bev_ssl, n_to_read);
    795          1.1  christos 		all_result_flags |= r;
    796          1.1  christos 
    797          1.1  christos 		if (r & (OP_BLOCKED|OP_ERR))
    798          1.1  christos 			break;
    799          1.1  christos 
    800          1.1  christos 		if (bev_ssl->bev.read_suspended)
    801          1.1  christos 			break;
    802          1.1  christos 
    803          1.1  christos 		/* Read all pending data.  This won't hit the network
    804          1.1  christos 		 * again, and will (most importantly) put us in a state
    805          1.1  christos 		 * where we don't need to read anything else until the
    806          1.1  christos 		 * socket is readable again.  It'll potentially make us
    807          1.1  christos 		 * overrun our read high-watermark (somewhat
    808          1.1  christos 		 * regrettable).  The damage to the rate-limit has
    809          1.1  christos 		 * already been done, since OpenSSL went and read a
    810          1.1  christos 		 * whole SSL record anyway. */
    811          1.1  christos 		n_to_read = SSL_pending(bev_ssl->ssl);
    812          1.1  christos 
    813          1.1  christos 		/* XXX This if statement is actually a bad bug, added to avoid
    814          1.1  christos 		 * XXX a worse bug.
    815          1.1  christos 		 *
    816          1.1  christos 		 * The bad bug: It can potentially cause resource unfairness
    817          1.1  christos 		 * by reading too much data from the underlying bufferevent;
    818          1.1  christos 		 * it can potentially cause read looping if the underlying
    819          1.1  christos 		 * bufferevent is a bufferevent_pair and deferred callbacks
    820          1.1  christos 		 * aren't used.
    821          1.1  christos 		 *
    822          1.1  christos 		 * The worse bug: If we didn't do this, then we would
    823          1.1  christos 		 * potentially not read any more from bev_ssl->underlying
    824          1.1  christos 		 * until more data arrived there, which could lead to us
    825          1.1  christos 		 * waiting forever.
    826          1.1  christos 		 */
    827          1.1  christos 		if (!n_to_read && bev_ssl->underlying)
    828          1.1  christos 			n_to_read = bytes_to_read(bev_ssl);
    829          1.1  christos 	}
    830          1.1  christos 
    831          1.1  christos 	if (all_result_flags & OP_MADE_PROGRESS) {
    832          1.1  christos 		struct bufferevent *bev = &bev_ssl->bev.bev;
    833          1.1  christos 
    834  1.1.1.1.6.1       riz 		bufferevent_trigger_nolock_(bev, EV_READ, 0);
    835          1.1  christos 	}
    836          1.1  christos 
    837          1.1  christos 	if (!bev_ssl->underlying) {
    838          1.1  christos 		/* Should be redundant, but let's avoid busy-looping */
    839          1.1  christos 		if (bev_ssl->bev.read_suspended ||
    840          1.1  christos 		    !(bev_ssl->bev.bev.enabled & EV_READ)) {
    841          1.1  christos 			event_del(&bev_ssl->bev.bev.ev_read);
    842          1.1  christos 		}
    843          1.1  christos 	}
    844          1.1  christos }
    845          1.1  christos 
    846          1.1  christos static void
    847          1.1  christos consider_writing(struct bufferevent_openssl *bev_ssl)
    848          1.1  christos {
    849          1.1  christos 	int r;
    850          1.1  christos 	struct evbuffer *output = bev_ssl->bev.bev.output;
    851          1.1  christos 	struct evbuffer *target = NULL;
    852          1.1  christos 	struct event_watermark *wm = NULL;
    853          1.1  christos 
    854          1.1  christos 	while (bev_ssl->read_blocked_on_write) {
    855          1.1  christos 		r = do_read(bev_ssl, 1024); /* XXXX 1024 is a hack */
    856          1.1  christos 		if (r & OP_MADE_PROGRESS) {
    857          1.1  christos 			struct bufferevent *bev = &bev_ssl->bev.bev;
    858          1.1  christos 
    859  1.1.1.1.6.1       riz 			bufferevent_trigger_nolock_(bev, EV_READ, 0);
    860          1.1  christos 		}
    861          1.1  christos 		if (r & (OP_ERR|OP_BLOCKED))
    862          1.1  christos 			break;
    863          1.1  christos 	}
    864          1.1  christos 	if (bev_ssl->read_blocked_on_write)
    865          1.1  christos 		return;
    866          1.1  christos 	if (bev_ssl->underlying) {
    867          1.1  christos 		target = bev_ssl->underlying->output;
    868          1.1  christos 		wm = &bev_ssl->underlying->wm_write;
    869          1.1  christos 	}
    870          1.1  christos 	while ((bev_ssl->bev.bev.enabled & EV_WRITE) &&
    871          1.1  christos 	    (! bev_ssl->bev.write_suspended) &&
    872          1.1  christos 	    evbuffer_get_length(output) &&
    873          1.1  christos 	    (!target || (! wm->high || evbuffer_get_length(target) < wm->high))) {
    874          1.1  christos 		int n_to_write;
    875          1.1  christos 		if (wm && wm->high)
    876          1.1  christos 			n_to_write = wm->high - evbuffer_get_length(target);
    877          1.1  christos 		else
    878          1.1  christos 			n_to_write = WRITE_FRAME;
    879          1.1  christos 		r = do_write(bev_ssl, n_to_write);
    880          1.1  christos 		if (r & (OP_BLOCKED|OP_ERR))
    881          1.1  christos 			break;
    882          1.1  christos 	}
    883          1.1  christos 
    884          1.1  christos 	if (!bev_ssl->underlying) {
    885          1.1  christos 		if (evbuffer_get_length(output) == 0) {
    886          1.1  christos 			event_del(&bev_ssl->bev.bev.ev_write);
    887          1.1  christos 		} else if (bev_ssl->bev.write_suspended ||
    888          1.1  christos 		    !(bev_ssl->bev.bev.enabled & EV_WRITE)) {
    889          1.1  christos 			/* Should be redundant, but let's avoid busy-looping */
    890          1.1  christos 			event_del(&bev_ssl->bev.bev.ev_write);
    891          1.1  christos 		}
    892          1.1  christos 	}
    893          1.1  christos }
    894          1.1  christos 
    895          1.1  christos static void
    896          1.1  christos be_openssl_readcb(struct bufferevent *bev_base, void *ctx)
    897          1.1  christos {
    898          1.1  christos 	struct bufferevent_openssl *bev_ssl = ctx;
    899          1.1  christos 	consider_reading(bev_ssl);
    900          1.1  christos }
    901          1.1  christos 
    902          1.1  christos static void
    903          1.1  christos be_openssl_writecb(struct bufferevent *bev_base, void *ctx)
    904          1.1  christos {
    905          1.1  christos 	struct bufferevent_openssl *bev_ssl = ctx;
    906          1.1  christos 	consider_writing(bev_ssl);
    907          1.1  christos }
    908          1.1  christos 
    909          1.1  christos static void
    910          1.1  christos be_openssl_eventcb(struct bufferevent *bev_base, short what, void *ctx)
    911          1.1  christos {
    912          1.1  christos 	struct bufferevent_openssl *bev_ssl = ctx;
    913          1.1  christos 	int event = 0;
    914          1.1  christos 
    915          1.1  christos 	if (what & BEV_EVENT_EOF) {
    916          1.1  christos 		if (bev_ssl->allow_dirty_shutdown)
    917          1.1  christos 			event = BEV_EVENT_EOF;
    918          1.1  christos 		else
    919          1.1  christos 			event = BEV_EVENT_ERROR;
    920          1.1  christos 	} else if (what & BEV_EVENT_TIMEOUT) {
    921          1.1  christos 		/* We sure didn't set this.  Propagate it to the user. */
    922          1.1  christos 		event = what;
    923          1.1  christos 	} else if (what & BEV_EVENT_ERROR) {
    924          1.1  christos 		/* An error occurred on the connection.  Propagate it to the user. */
    925          1.1  christos 		event = what;
    926          1.1  christos 	} else if (what & BEV_EVENT_CONNECTED) {
    927          1.1  christos 		/* Ignore it.  We're saying SSL_connect() already, which will
    928          1.1  christos 		   eat it. */
    929          1.1  christos 	}
    930          1.1  christos 	if (event)
    931  1.1.1.1.6.1       riz 		bufferevent_run_eventcb_(&bev_ssl->bev.bev, event, 0);
    932          1.1  christos }
    933          1.1  christos 
    934          1.1  christos static void
    935          1.1  christos be_openssl_readeventcb(evutil_socket_t fd, short what, void *ptr)
    936          1.1  christos {
    937          1.1  christos 	struct bufferevent_openssl *bev_ssl = ptr;
    938          1.1  christos 	bufferevent_incref_and_lock_(&bev_ssl->bev.bev);
    939          1.1  christos 	if (what == EV_TIMEOUT) {
    940          1.1  christos 		bufferevent_run_eventcb_(&bev_ssl->bev.bev,
    941  1.1.1.1.6.1       riz 		    BEV_EVENT_TIMEOUT|BEV_EVENT_READING, 0);
    942          1.1  christos 	} else {
    943          1.1  christos 		consider_reading(bev_ssl);
    944          1.1  christos 	}
    945          1.1  christos 	bufferevent_decref_and_unlock_(&bev_ssl->bev.bev);
    946          1.1  christos }
    947          1.1  christos 
    948          1.1  christos static void
    949          1.1  christos be_openssl_writeeventcb(evutil_socket_t fd, short what, void *ptr)
    950          1.1  christos {
    951          1.1  christos 	struct bufferevent_openssl *bev_ssl = ptr;
    952          1.1  christos 	bufferevent_incref_and_lock_(&bev_ssl->bev.bev);
    953          1.1  christos 	if (what == EV_TIMEOUT) {
    954          1.1  christos 		bufferevent_run_eventcb_(&bev_ssl->bev.bev,
    955  1.1.1.1.6.1       riz 		    BEV_EVENT_TIMEOUT|BEV_EVENT_WRITING, 0);
    956          1.1  christos 	} else {
    957          1.1  christos 		consider_writing(bev_ssl);
    958          1.1  christos 	}
    959          1.1  christos 	bufferevent_decref_and_unlock_(&bev_ssl->bev.bev);
    960          1.1  christos }
    961          1.1  christos 
    962          1.1  christos static int
    963          1.1  christos set_open_callbacks(struct bufferevent_openssl *bev_ssl, evutil_socket_t fd)
    964          1.1  christos {
    965          1.1  christos 	if (bev_ssl->underlying) {
    966          1.1  christos 		bufferevent_setcb(bev_ssl->underlying,
    967          1.1  christos 		    be_openssl_readcb, be_openssl_writecb, be_openssl_eventcb,
    968          1.1  christos 		    bev_ssl);
    969          1.1  christos 		return 0;
    970          1.1  christos 	} else {
    971          1.1  christos 		struct bufferevent *bev = &bev_ssl->bev.bev;
    972          1.1  christos 		int rpending=0, wpending=0, r1=0, r2=0;
    973          1.1  christos 		if (fd < 0 && bev_ssl->fd_is_set)
    974          1.1  christos 			fd = event_get_fd(&bev->ev_read);
    975          1.1  christos 		if (bev_ssl->fd_is_set) {
    976          1.1  christos 			rpending = event_pending(&bev->ev_read, EV_READ, NULL);
    977          1.1  christos 			wpending = event_pending(&bev->ev_write, EV_WRITE, NULL);
    978          1.1  christos 			event_del(&bev->ev_read);
    979          1.1  christos 			event_del(&bev->ev_write);
    980          1.1  christos 		}
    981          1.1  christos 		event_assign(&bev->ev_read, bev->ev_base, fd,
    982  1.1.1.1.6.1       riz 		    EV_READ|EV_PERSIST|EV_FINALIZE,
    983  1.1.1.1.6.1       riz 		    be_openssl_readeventcb, bev_ssl);
    984          1.1  christos 		event_assign(&bev->ev_write, bev->ev_base, fd,
    985  1.1.1.1.6.1       riz 		    EV_WRITE|EV_PERSIST|EV_FINALIZE,
    986  1.1.1.1.6.1       riz 		    be_openssl_writeeventcb, bev_ssl);
    987          1.1  christos 		if (rpending)
    988          1.1  christos 			r1 = bufferevent_add_event_(&bev->ev_read, &bev->timeout_read);
    989          1.1  christos 		if (wpending)
    990          1.1  christos 			r2 = bufferevent_add_event_(&bev->ev_write, &bev->timeout_write);
    991          1.1  christos 		if (fd >= 0) {
    992          1.1  christos 			bev_ssl->fd_is_set = 1;
    993          1.1  christos 		}
    994          1.1  christos 		return (r1<0 || r2<0) ? -1 : 0;
    995          1.1  christos 	}
    996          1.1  christos }
    997          1.1  christos 
    998          1.1  christos static int
    999          1.1  christos do_handshake(struct bufferevent_openssl *bev_ssl)
   1000          1.1  christos {
   1001          1.1  christos 	int r;
   1002          1.1  christos 
   1003          1.1  christos 	switch (bev_ssl->state) {
   1004          1.1  christos 	default:
   1005          1.1  christos 	case BUFFEREVENT_SSL_OPEN:
   1006          1.1  christos 		EVUTIL_ASSERT(0);
   1007          1.1  christos 		return -1;
   1008          1.1  christos 	case BUFFEREVENT_SSL_CONNECTING:
   1009          1.1  christos 	case BUFFEREVENT_SSL_ACCEPTING:
   1010          1.1  christos 		r = SSL_do_handshake(bev_ssl->ssl);
   1011          1.1  christos 		break;
   1012          1.1  christos 	}
   1013          1.1  christos 	decrement_buckets(bev_ssl);
   1014          1.1  christos 
   1015          1.1  christos 	if (r==1) {
   1016          1.1  christos 		/* We're done! */
   1017          1.1  christos 		bev_ssl->state = BUFFEREVENT_SSL_OPEN;
   1018          1.1  christos 		set_open_callbacks(bev_ssl, -1); /* XXXX handle failure */
   1019          1.1  christos 		/* Call do_read and do_write as needed */
   1020          1.1  christos 		bufferevent_enable(&bev_ssl->bev.bev, bev_ssl->bev.bev.enabled);
   1021          1.1  christos 		bufferevent_run_eventcb_(&bev_ssl->bev.bev,
   1022  1.1.1.1.6.1       riz 		    BEV_EVENT_CONNECTED, 0);
   1023          1.1  christos 		return 1;
   1024          1.1  christos 	} else {
   1025          1.1  christos 		int err = SSL_get_error(bev_ssl->ssl, r);
   1026          1.1  christos 		print_err(err);
   1027          1.1  christos 		switch (err) {
   1028          1.1  christos 		case SSL_ERROR_WANT_WRITE:
   1029          1.1  christos 			if (!bev_ssl->underlying) {
   1030          1.1  christos 				stop_reading(bev_ssl);
   1031          1.1  christos 				return start_writing(bev_ssl);
   1032          1.1  christos 			}
   1033          1.1  christos 			return 0;
   1034          1.1  christos 		case SSL_ERROR_WANT_READ:
   1035          1.1  christos 			if (!bev_ssl->underlying) {
   1036          1.1  christos 				stop_writing(bev_ssl);
   1037          1.1  christos 				return start_reading(bev_ssl);
   1038          1.1  christos 			}
   1039          1.1  christos 			return 0;
   1040          1.1  christos 		default:
   1041          1.1  christos 			conn_closed(bev_ssl, BEV_EVENT_READING, err, r);
   1042          1.1  christos 			return -1;
   1043          1.1  christos 		}
   1044          1.1  christos 	}
   1045          1.1  christos }
   1046          1.1  christos 
   1047          1.1  christos static void
   1048          1.1  christos be_openssl_handshakecb(struct bufferevent *bev_base, void *ctx)
   1049          1.1  christos {
   1050          1.1  christos 	struct bufferevent_openssl *bev_ssl = ctx;
   1051          1.1  christos 	do_handshake(bev_ssl);/* XXX handle failure */
   1052          1.1  christos }
   1053          1.1  christos 
   1054          1.1  christos static void
   1055          1.1  christos be_openssl_handshakeeventcb(evutil_socket_t fd, short what, void *ptr)
   1056          1.1  christos {
   1057          1.1  christos 	struct bufferevent_openssl *bev_ssl = ptr;
   1058          1.1  christos 
   1059          1.1  christos 	bufferevent_incref_and_lock_(&bev_ssl->bev.bev);
   1060          1.1  christos 	if (what & EV_TIMEOUT) {
   1061  1.1.1.1.6.1       riz 		bufferevent_run_eventcb_(&bev_ssl->bev.bev, BEV_EVENT_TIMEOUT, 0);
   1062          1.1  christos 	} else
   1063          1.1  christos 		do_handshake(bev_ssl);/* XXX handle failure */
   1064          1.1  christos 	bufferevent_decref_and_unlock_(&bev_ssl->bev.bev);
   1065          1.1  christos }
   1066          1.1  christos 
   1067          1.1  christos static int
   1068          1.1  christos set_handshake_callbacks(struct bufferevent_openssl *bev_ssl, evutil_socket_t fd)
   1069          1.1  christos {
   1070          1.1  christos 	if (bev_ssl->underlying) {
   1071          1.1  christos 		bufferevent_setcb(bev_ssl->underlying,
   1072          1.1  christos 		    be_openssl_handshakecb, be_openssl_handshakecb,
   1073          1.1  christos 		    be_openssl_eventcb,
   1074          1.1  christos 		    bev_ssl);
   1075          1.1  christos 		return do_handshake(bev_ssl);
   1076          1.1  christos 	} else {
   1077          1.1  christos 		struct bufferevent *bev = &bev_ssl->bev.bev;
   1078          1.1  christos 		int r1=0, r2=0;
   1079          1.1  christos 		if (fd < 0 && bev_ssl->fd_is_set)
   1080          1.1  christos 			fd = event_get_fd(&bev->ev_read);
   1081          1.1  christos 		if (bev_ssl->fd_is_set) {
   1082          1.1  christos 			event_del(&bev->ev_read);
   1083          1.1  christos 			event_del(&bev->ev_write);
   1084          1.1  christos 		}
   1085          1.1  christos 		event_assign(&bev->ev_read, bev->ev_base, fd,
   1086  1.1.1.1.6.1       riz 		    EV_READ|EV_PERSIST|EV_FINALIZE,
   1087  1.1.1.1.6.1       riz 		    be_openssl_handshakeeventcb, bev_ssl);
   1088          1.1  christos 		event_assign(&bev->ev_write, bev->ev_base, fd,
   1089  1.1.1.1.6.1       riz 		    EV_WRITE|EV_PERSIST|EV_FINALIZE,
   1090  1.1.1.1.6.1       riz 		    be_openssl_handshakeeventcb, bev_ssl);
   1091          1.1  christos 		if (fd >= 0) {
   1092          1.1  christos 			r1 = bufferevent_add_event_(&bev->ev_read, &bev->timeout_read);
   1093          1.1  christos 			r2 = bufferevent_add_event_(&bev->ev_write, &bev->timeout_write);
   1094          1.1  christos 			bev_ssl->fd_is_set = 1;
   1095          1.1  christos 		}
   1096          1.1  christos 		return (r1<0 || r2<0) ? -1 : 0;
   1097          1.1  christos 	}
   1098          1.1  christos }
   1099          1.1  christos 
   1100          1.1  christos int
   1101          1.1  christos bufferevent_ssl_renegotiate(struct bufferevent *bev)
   1102          1.1  christos {
   1103          1.1  christos 	struct bufferevent_openssl *bev_ssl = upcast(bev);
   1104          1.1  christos 	if (!bev_ssl)
   1105          1.1  christos 		return -1;
   1106          1.1  christos 	if (SSL_renegotiate(bev_ssl->ssl) < 0)
   1107          1.1  christos 		return -1;
   1108          1.1  christos 	bev_ssl->state = BUFFEREVENT_SSL_CONNECTING;
   1109          1.1  christos 	if (set_handshake_callbacks(bev_ssl, -1) < 0)
   1110          1.1  christos 		return -1;
   1111          1.1  christos 	if (!bev_ssl->underlying)
   1112          1.1  christos 		return do_handshake(bev_ssl);
   1113          1.1  christos 	return 0;
   1114          1.1  christos }
   1115          1.1  christos 
   1116          1.1  christos static void
   1117          1.1  christos be_openssl_outbuf_cb(struct evbuffer *buf,
   1118          1.1  christos     const struct evbuffer_cb_info *cbinfo, void *arg)
   1119          1.1  christos {
   1120          1.1  christos 	struct bufferevent_openssl *bev_ssl = arg;
   1121          1.1  christos 	int r = 0;
   1122          1.1  christos 	/* XXX need to hold a reference here. */
   1123          1.1  christos 
   1124          1.1  christos 	if (cbinfo->n_added && bev_ssl->state == BUFFEREVENT_SSL_OPEN) {
   1125          1.1  christos 		if (cbinfo->orig_size == 0)
   1126          1.1  christos 			r = bufferevent_add_event_(&bev_ssl->bev.bev.ev_write,
   1127          1.1  christos 			    &bev_ssl->bev.bev.timeout_write);
   1128          1.1  christos 		consider_writing(bev_ssl);
   1129          1.1  christos 	}
   1130          1.1  christos 	/* XXX Handle r < 0 */
   1131          1.1  christos         (void)r;
   1132          1.1  christos }
   1133          1.1  christos 
   1134          1.1  christos 
   1135          1.1  christos static int
   1136          1.1  christos be_openssl_enable(struct bufferevent *bev, short events)
   1137          1.1  christos {
   1138          1.1  christos 	struct bufferevent_openssl *bev_ssl = upcast(bev);
   1139          1.1  christos 	int r1 = 0, r2 = 0;
   1140          1.1  christos 
   1141          1.1  christos 	if (bev_ssl->state != BUFFEREVENT_SSL_OPEN)
   1142          1.1  christos 		return 0;
   1143          1.1  christos 
   1144          1.1  christos 	if (events & EV_READ)
   1145          1.1  christos 		r1 = start_reading(bev_ssl);
   1146          1.1  christos 	if (events & EV_WRITE)
   1147          1.1  christos 		r2 = start_writing(bev_ssl);
   1148          1.1  christos 
   1149          1.1  christos 	if (bev_ssl->underlying) {
   1150          1.1  christos 		if (events & EV_READ)
   1151          1.1  christos 			BEV_RESET_GENERIC_READ_TIMEOUT(bev);
   1152          1.1  christos 		if (events & EV_WRITE)
   1153          1.1  christos 			BEV_RESET_GENERIC_WRITE_TIMEOUT(bev);
   1154          1.1  christos 
   1155          1.1  christos 		if (events & EV_READ)
   1156          1.1  christos 			consider_reading(bev_ssl);
   1157          1.1  christos 		if (events & EV_WRITE)
   1158          1.1  christos 			consider_writing(bev_ssl);
   1159          1.1  christos 	}
   1160          1.1  christos 	return (r1 < 0 || r2 < 0) ? -1 : 0;
   1161          1.1  christos }
   1162          1.1  christos 
   1163          1.1  christos static int
   1164          1.1  christos be_openssl_disable(struct bufferevent *bev, short events)
   1165          1.1  christos {
   1166          1.1  christos 	struct bufferevent_openssl *bev_ssl = upcast(bev);
   1167          1.1  christos 	if (bev_ssl->state != BUFFEREVENT_SSL_OPEN)
   1168          1.1  christos 		return 0;
   1169          1.1  christos 
   1170          1.1  christos 	if (events & EV_READ)
   1171          1.1  christos 		stop_reading(bev_ssl);
   1172          1.1  christos 	if (events & EV_WRITE)
   1173          1.1  christos 		stop_writing(bev_ssl);
   1174          1.1  christos 
   1175          1.1  christos 	if (bev_ssl->underlying) {
   1176          1.1  christos 		if (events & EV_READ)
   1177          1.1  christos 			BEV_DEL_GENERIC_READ_TIMEOUT(bev);
   1178          1.1  christos 		if (events & EV_WRITE)
   1179          1.1  christos 			BEV_DEL_GENERIC_WRITE_TIMEOUT(bev);
   1180          1.1  christos 	}
   1181          1.1  christos 	return 0;
   1182          1.1  christos }
   1183          1.1  christos 
   1184          1.1  christos static void
   1185  1.1.1.1.6.1       riz be_openssl_unlink(struct bufferevent *bev)
   1186          1.1  christos {
   1187          1.1  christos 	struct bufferevent_openssl *bev_ssl = upcast(bev);
   1188          1.1  christos 
   1189          1.1  christos 	if (bev_ssl->bev.options & BEV_OPT_CLOSE_ON_FREE) {
   1190          1.1  christos 		if (bev_ssl->underlying) {
   1191          1.1  christos 			if (BEV_UPCAST(bev_ssl->underlying)->refcnt < 2) {
   1192          1.1  christos 				event_warnx("BEV_OPT_CLOSE_ON_FREE set on an "
   1193          1.1  christos 				    "bufferevent with too few references");
   1194          1.1  christos 			} else {
   1195          1.1  christos 				bufferevent_free(bev_ssl->underlying);
   1196  1.1.1.1.6.1       riz 				/* We still have a reference to it, via our
   1197  1.1.1.1.6.1       riz 				 * BIO. So we don't drop this. */
   1198  1.1.1.1.6.1       riz 				// bev_ssl->underlying = NULL;
   1199          1.1  christos 			}
   1200          1.1  christos 		}
   1201          1.1  christos 	} else {
   1202          1.1  christos 		if (bev_ssl->underlying) {
   1203          1.1  christos 			if (bev_ssl->underlying->errorcb == be_openssl_eventcb)
   1204          1.1  christos 				bufferevent_setcb(bev_ssl->underlying,
   1205          1.1  christos 				    NULL,NULL,NULL,NULL);
   1206          1.1  christos 			bufferevent_unsuspend_read_(bev_ssl->underlying,
   1207          1.1  christos 			    BEV_SUSPEND_FILT_READ);
   1208          1.1  christos 		}
   1209          1.1  christos 	}
   1210          1.1  christos }
   1211          1.1  christos 
   1212  1.1.1.1.6.1       riz static void
   1213  1.1.1.1.6.1       riz be_openssl_destruct(struct bufferevent *bev)
   1214  1.1.1.1.6.1       riz {
   1215  1.1.1.1.6.1       riz 	struct bufferevent_openssl *bev_ssl = upcast(bev);
   1216  1.1.1.1.6.1       riz 
   1217  1.1.1.1.6.1       riz 	if (bev_ssl->bev.options & BEV_OPT_CLOSE_ON_FREE) {
   1218  1.1.1.1.6.1       riz 		if (! bev_ssl->underlying) {
   1219  1.1.1.1.6.1       riz 			evutil_socket_t fd = -1;
   1220  1.1.1.1.6.1       riz 			BIO *bio = SSL_get_wbio(bev_ssl->ssl);
   1221  1.1.1.1.6.1       riz 			if (bio)
   1222  1.1.1.1.6.1       riz 				fd = BIO_get_fd(bio, NULL);
   1223  1.1.1.1.6.1       riz 			if (fd >= 0)
   1224  1.1.1.1.6.1       riz 				evutil_closesocket(fd);
   1225  1.1.1.1.6.1       riz 		}
   1226  1.1.1.1.6.1       riz 		SSL_free(bev_ssl->ssl);
   1227  1.1.1.1.6.1       riz 	}
   1228  1.1.1.1.6.1       riz }
   1229  1.1.1.1.6.1       riz 
   1230          1.1  christos static int
   1231          1.1  christos be_openssl_adj_timeouts(struct bufferevent *bev)
   1232          1.1  christos {
   1233          1.1  christos 	struct bufferevent_openssl *bev_ssl = upcast(bev);
   1234          1.1  christos 
   1235          1.1  christos 	if (bev_ssl->underlying) {
   1236          1.1  christos 		return bufferevent_generic_adj_timeouts_(bev);
   1237          1.1  christos 	} else {
   1238          1.1  christos 		int r1=0, r2=0;
   1239          1.1  christos 		if (event_pending(&bev->ev_read, EV_READ, NULL)) {
   1240          1.1  christos 			if (evutil_timerisset(&bev->timeout_read)) {
   1241          1.1  christos 				r1 = bufferevent_add_event_(&bev->ev_read, &bev->timeout_read);
   1242          1.1  christos 			} else {
   1243          1.1  christos 				event_remove_timer(&bev->ev_read);
   1244          1.1  christos 			}
   1245          1.1  christos 		}
   1246          1.1  christos 		if (event_pending(&bev->ev_write, EV_WRITE, NULL)) {
   1247          1.1  christos 			if (evutil_timerisset(&bev->timeout_write)) {
   1248          1.1  christos 				r2 = bufferevent_add_event_(&bev->ev_write, &bev->timeout_write);
   1249          1.1  christos 			} else {
   1250          1.1  christos 				event_remove_timer(&bev->ev_write);
   1251          1.1  christos 			}
   1252          1.1  christos 		}
   1253          1.1  christos 
   1254          1.1  christos 		return (r1<0 || r2<0) ? -1 : 0;
   1255          1.1  christos 	}
   1256          1.1  christos }
   1257          1.1  christos 
   1258          1.1  christos static int
   1259          1.1  christos be_openssl_flush(struct bufferevent *bufev,
   1260          1.1  christos     short iotype, enum bufferevent_flush_mode mode)
   1261          1.1  christos {
   1262          1.1  christos 	/* XXXX Implement this. */
   1263          1.1  christos 	return 0;
   1264          1.1  christos }
   1265          1.1  christos 
   1266          1.1  christos static int
   1267          1.1  christos be_openssl_ctrl(struct bufferevent *bev,
   1268          1.1  christos     enum bufferevent_ctrl_op op, union bufferevent_ctrl_data *data)
   1269          1.1  christos {
   1270          1.1  christos 	struct bufferevent_openssl *bev_ssl = upcast(bev);
   1271          1.1  christos 	switch (op) {
   1272          1.1  christos 	case BEV_CTRL_SET_FD:
   1273          1.1  christos 		if (bev_ssl->underlying)
   1274          1.1  christos 			return -1;
   1275          1.1  christos 		{
   1276          1.1  christos 			BIO *bio;
   1277          1.1  christos 			bio = BIO_new_socket(data->fd, 0);
   1278          1.1  christos 			SSL_set_bio(bev_ssl->ssl, bio, bio);
   1279          1.1  christos 			bev_ssl->fd_is_set = 1;
   1280          1.1  christos 		}
   1281          1.1  christos 		if (bev_ssl->state == BUFFEREVENT_SSL_OPEN)
   1282          1.1  christos 			return set_open_callbacks(bev_ssl, data->fd);
   1283          1.1  christos 		else {
   1284          1.1  christos 			return set_handshake_callbacks(bev_ssl, data->fd);
   1285          1.1  christos 		}
   1286          1.1  christos 	case BEV_CTRL_GET_FD:
   1287          1.1  christos 		if (bev_ssl->underlying)
   1288          1.1  christos 			return -1;
   1289          1.1  christos 		if (!bev_ssl->fd_is_set)
   1290          1.1  christos 			return -1;
   1291          1.1  christos 		data->fd = event_get_fd(&bev->ev_read);
   1292          1.1  christos 		return 0;
   1293          1.1  christos 	case BEV_CTRL_GET_UNDERLYING:
   1294          1.1  christos 		if (!bev_ssl->underlying)
   1295          1.1  christos 			return -1;
   1296          1.1  christos 		data->ptr = bev_ssl->underlying;
   1297          1.1  christos 		return 0;
   1298          1.1  christos 	case BEV_CTRL_CANCEL_ALL:
   1299          1.1  christos 	default:
   1300          1.1  christos 		return -1;
   1301          1.1  christos 	}
   1302          1.1  christos }
   1303          1.1  christos 
   1304          1.1  christos SSL *
   1305          1.1  christos bufferevent_openssl_get_ssl(struct bufferevent *bufev)
   1306          1.1  christos {
   1307          1.1  christos 	struct bufferevent_openssl *bev_ssl = upcast(bufev);
   1308          1.1  christos 	if (!bev_ssl)
   1309          1.1  christos 		return NULL;
   1310          1.1  christos 	return bev_ssl->ssl;
   1311          1.1  christos }
   1312          1.1  christos 
   1313          1.1  christos static struct bufferevent *
   1314          1.1  christos bufferevent_openssl_new_impl(struct event_base *base,
   1315          1.1  christos     struct bufferevent *underlying,
   1316          1.1  christos     evutil_socket_t fd,
   1317          1.1  christos     SSL *ssl,
   1318          1.1  christos     enum bufferevent_ssl_state state,
   1319          1.1  christos     int options)
   1320          1.1  christos {
   1321          1.1  christos 	struct bufferevent_openssl *bev_ssl = NULL;
   1322          1.1  christos 	struct bufferevent_private *bev_p = NULL;
   1323          1.1  christos 	int tmp_options = options & ~BEV_OPT_THREADSAFE;
   1324          1.1  christos 
   1325          1.1  christos 	if (underlying != NULL && fd >= 0)
   1326          1.1  christos 		return NULL; /* Only one can be set. */
   1327          1.1  christos 
   1328          1.1  christos 	if (!(bev_ssl = mm_calloc(1, sizeof(struct bufferevent_openssl))))
   1329          1.1  christos 		goto err;
   1330          1.1  christos 
   1331          1.1  christos 	bev_p = &bev_ssl->bev;
   1332          1.1  christos 
   1333          1.1  christos 	if (bufferevent_init_common_(bev_p, base,
   1334          1.1  christos 		&bufferevent_ops_openssl, tmp_options) < 0)
   1335          1.1  christos 		goto err;
   1336          1.1  christos 
   1337          1.1  christos 	/* Don't explode if we decide to realloc a chunk we're writing from in
   1338          1.1  christos 	 * the output buffer. */
   1339          1.1  christos 	SSL_set_mode(ssl, SSL_MODE_ACCEPT_MOVING_WRITE_BUFFER);
   1340          1.1  christos 
   1341          1.1  christos 	bev_ssl->underlying = underlying;
   1342          1.1  christos 	bev_ssl->ssl = ssl;
   1343          1.1  christos 
   1344          1.1  christos 	bev_ssl->outbuf_cb = evbuffer_add_cb(bev_p->bev.output,
   1345          1.1  christos 	    be_openssl_outbuf_cb, bev_ssl);
   1346          1.1  christos 
   1347          1.1  christos 	if (options & BEV_OPT_THREADSAFE)
   1348          1.1  christos 		bufferevent_enable_locking_(&bev_ssl->bev.bev, NULL);
   1349          1.1  christos 
   1350          1.1  christos 	if (underlying) {
   1351          1.1  christos 		bufferevent_init_generic_timeout_cbs_(&bev_ssl->bev.bev);
   1352          1.1  christos 		bufferevent_incref_(underlying);
   1353          1.1  christos 	}
   1354          1.1  christos 
   1355          1.1  christos 	bev_ssl->state = state;
   1356          1.1  christos 	bev_ssl->last_write = -1;
   1357          1.1  christos 
   1358          1.1  christos 	init_bio_counts(bev_ssl);
   1359          1.1  christos 
   1360          1.1  christos 	switch (state) {
   1361          1.1  christos 	case BUFFEREVENT_SSL_ACCEPTING:
   1362          1.1  christos 		SSL_set_accept_state(bev_ssl->ssl);
   1363          1.1  christos 		if (set_handshake_callbacks(bev_ssl, fd) < 0)
   1364          1.1  christos 			goto err;
   1365          1.1  christos 		break;
   1366          1.1  christos 	case BUFFEREVENT_SSL_CONNECTING:
   1367          1.1  christos 		SSL_set_connect_state(bev_ssl->ssl);
   1368          1.1  christos 		if (set_handshake_callbacks(bev_ssl, fd) < 0)
   1369          1.1  christos 			goto err;
   1370          1.1  christos 		break;
   1371          1.1  christos 	case BUFFEREVENT_SSL_OPEN:
   1372          1.1  christos 		if (set_open_callbacks(bev_ssl, fd) < 0)
   1373          1.1  christos 			goto err;
   1374          1.1  christos 		break;
   1375          1.1  christos 	default:
   1376          1.1  christos 		goto err;
   1377          1.1  christos 	}
   1378          1.1  christos 
   1379          1.1  christos 	if (underlying) {
   1380          1.1  christos 		bufferevent_setwatermark(underlying, EV_READ, 0, 0);
   1381          1.1  christos 		bufferevent_enable(underlying, EV_READ|EV_WRITE);
   1382          1.1  christos 		if (state == BUFFEREVENT_SSL_OPEN)
   1383          1.1  christos 			bufferevent_suspend_read_(underlying,
   1384          1.1  christos 			    BEV_SUSPEND_FILT_READ);
   1385          1.1  christos 	} else {
   1386          1.1  christos 		bev_ssl->bev.bev.enabled = EV_READ|EV_WRITE;
   1387          1.1  christos 		if (bev_ssl->fd_is_set) {
   1388          1.1  christos 			if (state != BUFFEREVENT_SSL_OPEN)
   1389          1.1  christos 				if (event_add(&bev_ssl->bev.bev.ev_read, NULL) < 0)
   1390          1.1  christos 					goto err;
   1391          1.1  christos 			if (event_add(&bev_ssl->bev.bev.ev_write, NULL) < 0)
   1392          1.1  christos 				goto err;
   1393          1.1  christos 		}
   1394          1.1  christos 	}
   1395          1.1  christos 
   1396          1.1  christos 	return &bev_ssl->bev.bev;
   1397          1.1  christos err:
   1398          1.1  christos 	if (bev_ssl)
   1399          1.1  christos 		bufferevent_free(&bev_ssl->bev.bev);
   1400          1.1  christos 	return NULL;
   1401          1.1  christos }
   1402          1.1  christos 
   1403          1.1  christos struct bufferevent *
   1404          1.1  christos bufferevent_openssl_filter_new(struct event_base *base,
   1405          1.1  christos     struct bufferevent *underlying,
   1406          1.1  christos     SSL *ssl,
   1407          1.1  christos     enum bufferevent_ssl_state state,
   1408          1.1  christos     int options)
   1409          1.1  christos {
   1410          1.1  christos 	/* We don't tell the BIO to close the bufferevent; we do it ourselves
   1411          1.1  christos 	 * on be_openssl_destruct */
   1412          1.1  christos 	int close_flag = 0; /* options & BEV_OPT_CLOSE_ON_FREE; */
   1413          1.1  christos 	BIO *bio;
   1414          1.1  christos 	if (!underlying)
   1415          1.1  christos 		return NULL;
   1416          1.1  christos 	if (!(bio = BIO_new_bufferevent(underlying, close_flag)))
   1417          1.1  christos 		return NULL;
   1418          1.1  christos 
   1419          1.1  christos 	SSL_set_bio(ssl, bio, bio);
   1420          1.1  christos 
   1421          1.1  christos 	return bufferevent_openssl_new_impl(
   1422          1.1  christos 		base, underlying, -1, ssl, state, options);
   1423          1.1  christos }
   1424          1.1  christos 
   1425          1.1  christos struct bufferevent *
   1426          1.1  christos bufferevent_openssl_socket_new(struct event_base *base,
   1427          1.1  christos     evutil_socket_t fd,
   1428          1.1  christos     SSL *ssl,
   1429          1.1  christos     enum bufferevent_ssl_state state,
   1430          1.1  christos     int options)
   1431          1.1  christos {
   1432          1.1  christos 	/* Does the SSL already have an fd? */
   1433          1.1  christos 	BIO *bio = SSL_get_wbio(ssl);
   1434          1.1  christos 	long have_fd = -1;
   1435          1.1  christos 
   1436          1.1  christos 	if (bio)
   1437          1.1  christos 		have_fd = BIO_get_fd(bio, NULL);
   1438          1.1  christos 
   1439          1.1  christos 	if (have_fd >= 0) {
   1440          1.1  christos 		/* The SSL is already configured with an fd. */
   1441          1.1  christos 		if (fd < 0) {
   1442          1.1  christos 			/* We should learn the fd from the SSL. */
   1443          1.1  christos 			fd = (evutil_socket_t) have_fd;
   1444          1.1  christos 		} else if (have_fd == (long)fd) {
   1445          1.1  christos 			/* We already know the fd from the SSL; do nothing */
   1446          1.1  christos 		} else {
   1447          1.1  christos 			/* We specified an fd different from that of the SSL.
   1448          1.1  christos 			   This is probably an error on our part.  Fail. */
   1449          1.1  christos 			return NULL;
   1450          1.1  christos 		}
   1451          1.1  christos 		(void) BIO_set_close(bio, 0);
   1452          1.1  christos 	} else {
   1453          1.1  christos 		/* The SSL isn't configured with a BIO with an fd. */
   1454          1.1  christos 		if (fd >= 0) {
   1455          1.1  christos 			/* ... and we have an fd we want to use. */
   1456          1.1  christos 			bio = BIO_new_socket(fd, 0);
   1457          1.1  christos 			SSL_set_bio(ssl, bio, bio);
   1458          1.1  christos 		} else {
   1459          1.1  christos 			/* Leave the fd unset. */
   1460          1.1  christos 		}
   1461          1.1  christos 	}
   1462          1.1  christos 
   1463          1.1  christos 	return bufferevent_openssl_new_impl(
   1464          1.1  christos 		base, NULL, fd, ssl, state, options);
   1465          1.1  christos }
   1466          1.1  christos 
   1467          1.1  christos int
   1468          1.1  christos bufferevent_openssl_get_allow_dirty_shutdown(struct bufferevent *bev)
   1469          1.1  christos {
   1470          1.1  christos 	int allow_dirty_shutdown = -1;
   1471          1.1  christos 	struct bufferevent_openssl *bev_ssl;
   1472          1.1  christos 	BEV_LOCK(bev);
   1473          1.1  christos 	bev_ssl = upcast(bev);
   1474          1.1  christos 	if (bev_ssl)
   1475          1.1  christos 		allow_dirty_shutdown = bev_ssl->allow_dirty_shutdown;
   1476          1.1  christos 	BEV_UNLOCK(bev);
   1477          1.1  christos 	return allow_dirty_shutdown;
   1478          1.1  christos }
   1479          1.1  christos 
   1480          1.1  christos void
   1481          1.1  christos bufferevent_openssl_set_allow_dirty_shutdown(struct bufferevent *bev,
   1482          1.1  christos     int allow_dirty_shutdown)
   1483          1.1  christos {
   1484          1.1  christos 	struct bufferevent_openssl *bev_ssl;
   1485          1.1  christos 	BEV_LOCK(bev);
   1486          1.1  christos 	bev_ssl = upcast(bev);
   1487          1.1  christos 	if (bev_ssl)
   1488          1.1  christos 		bev_ssl->allow_dirty_shutdown = !!allow_dirty_shutdown;
   1489          1.1  christos 	BEV_UNLOCK(bev);
   1490          1.1  christos }
   1491          1.1  christos 
   1492          1.1  christos unsigned long
   1493          1.1  christos bufferevent_get_openssl_error(struct bufferevent *bev)
   1494          1.1  christos {
   1495          1.1  christos 	unsigned long err = 0;
   1496          1.1  christos 	struct bufferevent_openssl *bev_ssl;
   1497          1.1  christos 	BEV_LOCK(bev);
   1498          1.1  christos 	bev_ssl = upcast(bev);
   1499          1.1  christos 	if (bev_ssl && bev_ssl->n_errors) {
   1500          1.1  christos 		err = bev_ssl->errors[--bev_ssl->n_errors];
   1501          1.1  christos 	}
   1502          1.1  christos 	BEV_UNLOCK(bev);
   1503          1.1  christos 	return err;
   1504          1.1  christos }
   1505