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