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