ssl/bio_ssl.c

   1 /*
   2  * Copyright 1995-2016 The OpenSSL Project Authors. All Rights Reserved.
   3  *
   4  * Licensed under the OpenSSL license (the "License").  You may not use
   5  * this file except in compliance with the License.  You can obtain a copy
   6  * in the file LICENSE in the source distribution or at
   7  * https://www.openssl.org/source/license.html
   8  */
   9
  10 #include <stdio.h>
  11 #include <stdlib.h>
  12 #include <string.h>
  13 #include <errno.h>
  14 #include <openssl/crypto.h>
  15 #include "internal/bio.h"
  16 #include <openssl/err.h>
  17 #include "ssl_locl.h"
  18
  19 static int ssl_write(BIO *h, const char *buf, size_t size, size_t *written);
  20 static int ssl_read(BIO *b, char *buf, size_t size, size_t *readbytes);
  21 static int ssl_puts(BIO *h, const char *str);
  22 static long ssl_ctrl(BIO *h, int cmd, long arg1, void *arg2);
  23 static int ssl_new(BIO *h);
  24 static int ssl_free(BIO *data);
  25 static long ssl_callback_ctrl(BIO *h, int cmd, bio_info_cb *fp);
  26 typedef struct bio_ssl_st {
  27     SSL *ssl;                   /* The ssl handle :-) */
  28     /* re-negotiate every time the total number of bytes is this size */
  29     int num_renegotiates;
  30     unsigned long renegotiate_count;
  31     size_t byte_count;
  32     unsigned long renegotiate_timeout;
  33     unsigned long last_time;
  34 } BIO_SSL;
  35
  36 static const BIO_METHOD methods_sslp = {
  37     BIO_TYPE_SSL, "ssl",
  38     ssl_write,
  39     NULL,
  40     ssl_read,
  41     NULL,
  42     ssl_puts,
  43     NULL,                       /* ssl_gets, */
  44     ssl_ctrl,
  45     ssl_new,
  46     ssl_free,
  47     ssl_callback_ctrl,
  48 };
  49
  50 const BIO_METHOD *BIO_f_ssl(void)
  51 {
  52     return (&methods_sslp);
  53 }
  54
  55 static int ssl_new(BIO *bi)
  56 {
  57     BIO_SSL *bs = OPENSSL_zalloc(sizeof(*bs));
  58
  59     if (bs == NULL) {
  60         BIOerr(BIO_F_SSL_NEW, ERR_R_MALLOC_FAILURE);
  61         return (0);
  62     }
  63     BIO_set_init(bi, 0);
  64     BIO_set_data(bi, bs);
  65     /* Clear all flags */
  66     BIO_clear_flags(bi, ~0);
  67
  68     return 1;
  69 }
  70
  71 static int ssl_free(BIO *a)
  72 {
  73     BIO_SSL *bs;
  74
  75     if (a == NULL)
  76         return (0);
  77     bs = BIO_get_data(a);
  78     if (bs->ssl != NULL)
  79         SSL_shutdown(bs->ssl);
  80     if (BIO_get_shutdown(a)) {
  81         if (BIO_get_init(a))
  82             SSL_free(bs->ssl);
  83         /* Clear all flags */
  84         BIO_clear_flags(a, ~0);
  85         BIO_set_init(a, 0);
  86     }
  87     OPENSSL_free(bs);
  88     return 1;
  89 }
  90
  91 static int ssl_read(BIO *b, char *buf, size_t size, size_t *readbytes)
  92 {
  93     int ret = 1;
  94     BIO_SSL *sb;
  95     SSL *ssl;
  96     int retry_reason = 0;
  97     int r = 0;
  98
  99     if (buf == NULL)
 100         return 0;
 101     sb = BIO_get_data(b);
 102     ssl = sb->ssl;
 103
 104     BIO_clear_retry_flags(b);
 105
 106     ret = ssl_read_internal(ssl, buf, size, readbytes);
 107
 108     switch (SSL_get_error(ssl, ret)) {
 109     case SSL_ERROR_NONE:
 110         if (sb->renegotiate_count > 0) {
 111             sb->byte_count += *readbytes;
 112             if (sb->byte_count > sb->renegotiate_count) {
 113                 sb->byte_count = 0;
 114                 sb->num_renegotiates++;
 115                 SSL_renegotiate(ssl);
 116                 r = 1;
 117             }
 118         }
 119         if ((sb->renegotiate_timeout > 0) && (!r)) {
 120             unsigned long tm;
 121
 122             tm = (unsigned long)time(NULL);
 123             if (tm > sb->last_time + sb->renegotiate_timeout) {
 124                 sb->last_time = tm;
 125                 sb->num_renegotiates++;
 126                 SSL_renegotiate(ssl);
 127             }
 128         }
 129
 130         break;
 131     case SSL_ERROR_WANT_READ:
 132         BIO_set_retry_read(b);
 133         break;
 134     case SSL_ERROR_WANT_WRITE:
 135         BIO_set_retry_write(b);
 136         break;
 137     case SSL_ERROR_WANT_X509_LOOKUP:
 138         BIO_set_retry_special(b);
 139         retry_reason = BIO_RR_SSL_X509_LOOKUP;
 140         break;
 141     case SSL_ERROR_WANT_ACCEPT:
 142         BIO_set_retry_special(b);
 143         retry_reason = BIO_RR_ACCEPT;
 144         break;
 145     case SSL_ERROR_WANT_CONNECT:
 146         BIO_set_retry_special(b);
 147         retry_reason = BIO_RR_CONNECT;
 148         break;
 149     case SSL_ERROR_SYSCALL:
 150     case SSL_ERROR_SSL:
 151     case SSL_ERROR_ZERO_RETURN:
 152     default:
 153         break;
 154     }
 155
 156     BIO_set_retry_reason(b, retry_reason);
 157
 158     return ret;
 159 }
 160
 161 static int ssl_write(BIO *b, const char *buf, size_t size, size_t *written)
 162 {
 163     int ret, r = 0;
 164     int retry_reason = 0;
 165     SSL *ssl;
 166     BIO_SSL *bs;
 167
 168     if (buf == NULL)
 169         return 0;
 170     bs = BIO_get_data(b);
 171     ssl = bs->ssl;
 172
 173     BIO_clear_retry_flags(b);
 174
 175     ret = ssl_write_internal(ssl, buf, size, written);
 176
 177     switch (SSL_get_error(ssl, ret)) {
 178     case SSL_ERROR_NONE:
 179         if (bs->renegotiate_count > 0) {
 180             bs->byte_count += *written;
 181             if (bs->byte_count > bs->renegotiate_count) {
 182                 bs->byte_count = 0;
 183                 bs->num_renegotiates++;
 184                 SSL_renegotiate(ssl);
 185                 r = 1;
 186             }
 187         }
 188         if ((bs->renegotiate_timeout > 0) && (!r)) {
 189             unsigned long tm;
 190
 191             tm = (unsigned long)time(NULL);
 192             if (tm > bs->last_time + bs->renegotiate_timeout) {
 193                 bs->last_time = tm;
 194                 bs->num_renegotiates++;
 195                 SSL_renegotiate(ssl);
 196             }
 197         }
 198         break;
 199     case SSL_ERROR_WANT_WRITE:
 200         BIO_set_retry_write(b);
 201         break;
 202     case SSL_ERROR_WANT_READ:
 203         BIO_set_retry_read(b);
 204         break;
 205     case SSL_ERROR_WANT_X509_LOOKUP:
 206         BIO_set_retry_special(b);
 207         retry_reason = BIO_RR_SSL_X509_LOOKUP;
 208         break;
 209     case SSL_ERROR_WANT_CONNECT:
 210         BIO_set_retry_special(b);
 211         retry_reason = BIO_RR_CONNECT;
 212     case SSL_ERROR_SYSCALL:
 213     case SSL_ERROR_SSL:
 214     default:
 215         break;
 216     }
 217
 218     BIO_set_retry_reason(b, retry_reason);
 219
 220     return ret;
 221 }
 222
 223 static long ssl_ctrl(BIO *b, int cmd, long num, void *ptr)
 224 {
 225     SSL **sslp, *ssl;
 226     BIO_SSL *bs, *dbs;
 227     BIO *dbio, *bio;
 228     long ret = 1;
 229     BIO *next;
 230
 231     bs = BIO_get_data(b);
 232     next = BIO_next(b);
 233     ssl = bs->ssl;
 234     if ((ssl == NULL) && (cmd != BIO_C_SET_SSL))
 235         return (0);
 236     switch (cmd) {
 237     case BIO_CTRL_RESET:
 238         SSL_shutdown(ssl);
 239
 240         if (ssl->handshake_func == ssl->method->ssl_connect)
 241             SSL_set_connect_state(ssl);
 242         else if (ssl->handshake_func == ssl->method->ssl_accept)
 243             SSL_set_accept_state(ssl);
 244
 245         if (!SSL_clear(ssl)) {
 246             ret = 0;
 247             break;
 248         }
 249
 250         if (next != NULL)
 251             ret = BIO_ctrl(next, cmd, num, ptr);
 252         else if (ssl->rbio != NULL)
 253             ret = BIO_ctrl(ssl->rbio, cmd, num, ptr);
 254         else
 255             ret = 1;
 256         break;
 257     case BIO_CTRL_INFO:
 258         ret = 0;
 259         break;
 260     case BIO_C_SSL_MODE:
 261         if (num)                /* client mode */
 262             SSL_set_connect_state(ssl);
 263         else
 264             SSL_set_accept_state(ssl);
 265         break;
 266     case BIO_C_SET_SSL_RENEGOTIATE_TIMEOUT:
 267         ret = bs->renegotiate_timeout;
 268         if (num < 60)
 269             num = 5;
 270         bs->renegotiate_timeout = (unsigned long)num;
 271         bs->last_time = (unsigned long)time(NULL);
 272         break;
 273     case BIO_C_SET_SSL_RENEGOTIATE_BYTES:
 274         ret = bs->renegotiate_count;
 275         if ((long)num >= 512)
 276             bs->renegotiate_count = (unsigned long)num;
 277         break;
 278     case BIO_C_GET_SSL_NUM_RENEGOTIATES:
 279         ret = bs->num_renegotiates;
 280         break;
 281     case BIO_C_SET_SSL:
 282         if (ssl != NULL) {
 283             ssl_free(b);
 284             if (!ssl_new(b))
 285                 return 0;
 286         }
 287         BIO_set_shutdown(b, num);
 288         ssl = (SSL *)ptr;
 289         bs->ssl = ssl;
 290         bio = SSL_get_rbio(ssl);
 291         if (bio != NULL) {
 292             if (next != NULL)
 293                 BIO_push(bio, next);
 294             BIO_set_next(b, bio);
 295             BIO_up_ref(bio);
 296         }
 297         BIO_set_init(b, 1);
 298         break;
 299     case BIO_C_GET_SSL:
 300         if (ptr != NULL) {
 301             sslp = (SSL **)ptr;
 302             *sslp = ssl;
 303         } else
 304             ret = 0;
 305         break;
 306     case BIO_CTRL_GET_CLOSE:
 307         ret = BIO_get_shutdown(b);
 308         break;
 309     case BIO_CTRL_SET_CLOSE:
 310         BIO_set_shutdown(b, (int)num);
 311         break;
 312     case BIO_CTRL_WPENDING:
 313         ret = BIO_ctrl(ssl->wbio, cmd, num, ptr);
 314         break;
 315     case BIO_CTRL_PENDING:
 316         ret = SSL_pending(ssl);
 317         if (ret == 0)
 318             ret = BIO_pending(ssl->rbio);
 319         break;
 320     case BIO_CTRL_FLUSH:
 321         BIO_clear_retry_flags(b);
 322         ret = BIO_ctrl(ssl->wbio, cmd, num, ptr);
 323         BIO_copy_next_retry(b);
 324         break;
 325     case BIO_CTRL_PUSH:
 326         if ((next != NULL) && (next != ssl->rbio)) {
 327             /*
 328              * We are going to pass ownership of next to the SSL object...but
 329              * we don't own a reference to pass yet - so up ref
 330              */
 331             BIO_up_ref(next);
 332             SSL_set_bio(ssl, next, next);
 333         }
 334         break;
 335     case BIO_CTRL_POP:
 336         /* Only detach if we are the BIO explicitly being popped */
 337         if (b == ptr) {
 338             /* This will clear the reference we obtained during push */
 339             SSL_set_bio(ssl, NULL, NULL);
 340         }
 341         break;
 342     case BIO_C_DO_STATE_MACHINE:
 343         BIO_clear_retry_flags(b);
 344
 345         BIO_set_retry_reason(b, 0);
 346         ret = (int)SSL_do_handshake(ssl);
 347
 348         switch (SSL_get_error(ssl, (int)ret)) {
 349         case SSL_ERROR_WANT_READ:
 350             BIO_set_flags(b, BIO_FLAGS_READ | BIO_FLAGS_SHOULD_RETRY);
 351             break;
 352         case SSL_ERROR_WANT_WRITE:
 353             BIO_set_flags(b, BIO_FLAGS_WRITE | BIO_FLAGS_SHOULD_RETRY);
 354             break;
 355         case SSL_ERROR_WANT_CONNECT:
 356             BIO_set_flags(b, BIO_FLAGS_IO_SPECIAL | BIO_FLAGS_SHOULD_RETRY);
 357             BIO_set_retry_reason(b, BIO_get_retry_reason(next));
 358             break;
 359         case SSL_ERROR_WANT_X509_LOOKUP:
 360             BIO_set_retry_special(b);
 361             BIO_set_retry_reason(b, BIO_RR_SSL_X509_LOOKUP);
 362             break;
 363         default:
 364             break;
 365         }
 366         break;
 367     case BIO_CTRL_DUP:
 368         dbio = (BIO *)ptr;
 369         dbs = BIO_get_data(dbio);
 370         SSL_free(dbs->ssl);
 371         dbs->ssl = SSL_dup(ssl);
 372         dbs->num_renegotiates = bs->num_renegotiates;
 373         dbs->renegotiate_count = bs->renegotiate_count;
 374         dbs->byte_count = bs->byte_count;
 375         dbs->renegotiate_timeout = bs->renegotiate_timeout;
 376         dbs->last_time = bs->last_time;
 377         ret = (dbs->ssl != NULL);
 378         break;
 379     case BIO_C_GET_FD:
 380         ret = BIO_ctrl(ssl->rbio, cmd, num, ptr);
 381         break;
 382     case BIO_CTRL_SET_CALLBACK:
 383         ret = 0; /* use callback ctrl */
 384         break;
 385     case BIO_CTRL_GET_CALLBACK:
 386         {
 387             void (**fptr) (const SSL *xssl, int type, int val);
 388
 389             fptr = (void (**)(const SSL *xssl, int type, int val))ptr;
 390             *fptr = SSL_get_info_callback(ssl);
 391         }
 392         break;
 393     default:
 394         ret = BIO_ctrl(ssl->rbio, cmd, num, ptr);
 395         break;
 396     }
 397     return (ret);
 398 }
 399
 400 static long ssl_callback_ctrl(BIO *b, int cmd, bio_info_cb *fp)
 401 {
 402     SSL *ssl;
 403     BIO_SSL *bs;
 404     long ret = 1;
 405
 406     bs = BIO_get_data(b);
 407     ssl = bs->ssl;
 408     switch (cmd) {
 409     case BIO_CTRL_SET_CALLBACK:
 410         {
 411             /*
 412              * FIXME: setting this via a completely different prototype seems
 413              * like a crap idea
 414              */
 415             SSL_set_info_callback(ssl, (void (*)(const SSL *, int, int))fp);
 416         }
 417         break;
 418     default:
 419         ret = BIO_callback_ctrl(ssl->rbio, cmd, fp);
 420         break;
 421     }
 422     return (ret);
 423 }
 424
 425 static int ssl_puts(BIO *bp, const char *str)
 426 {
 427     int n, ret;
 428
 429     n = strlen(str);
 430     ret = BIO_write(bp, str, n);
 431     return (ret);
 432 }
 433
 434 BIO *BIO_new_buffer_ssl_connect(SSL_CTX *ctx)
 435 {
 436 #ifndef OPENSSL_NO_SOCK
 437     BIO *ret = NULL, *buf = NULL, *ssl = NULL;
 438
 439     if ((buf = BIO_new(BIO_f_buffer())) == NULL)
 440         return (NULL);
 441     if ((ssl = BIO_new_ssl_connect(ctx)) == NULL)
 442         goto err;
 443     if ((ret = BIO_push(buf, ssl)) == NULL)
 444         goto err;
 445     return (ret);
 446  err:
 447     BIO_free(buf);
 448     BIO_free(ssl);
 449 #endif
 450     return (NULL);
 451 }
 452
 453 BIO *BIO_new_ssl_connect(SSL_CTX *ctx)
 454 {
 455 #ifndef OPENSSL_NO_SOCK
 456     BIO *ret = NULL, *con = NULL, *ssl = NULL;
 457
 458     if ((con = BIO_new(BIO_s_connect())) == NULL)
 459         return (NULL);
 460     if ((ssl = BIO_new_ssl(ctx, 1)) == NULL)
 461         goto err;
 462     if ((ret = BIO_push(ssl, con)) == NULL)
 463         goto err;
 464     return (ret);
 465  err:
 466     BIO_free(con);
 467 #endif
 468     return (NULL);
 469 }
 470
 471 BIO *BIO_new_ssl(SSL_CTX *ctx, int client)
 472 {
 473     BIO *ret;
 474     SSL *ssl;
 475
 476     if ((ret = BIO_new(BIO_f_ssl())) == NULL)
 477         return (NULL);
 478     if ((ssl = SSL_new(ctx)) == NULL) {
 479         BIO_free(ret);
 480         return (NULL);
 481     }
 482     if (client)
 483         SSL_set_connect_state(ssl);
 484     else
 485         SSL_set_accept_state(ssl);
 486
 487     BIO_set_ssl(ret, ssl, BIO_CLOSE);
 488     return (ret);
 489 }
 490
 491 int BIO_ssl_copy_session_id(BIO *t, BIO *f)
 492 {
 493     BIO_SSL *tdata, *fdata;
 494     t = BIO_find_type(t, BIO_TYPE_SSL);
 495     f = BIO_find_type(f, BIO_TYPE_SSL);
 496     if ((t == NULL) || (f == NULL))
 497         return 0;
 498     tdata = BIO_get_data(t);
 499     fdata = BIO_get_data(f);
 500     if ((tdata->ssl == NULL) || (fdata->ssl == NULL))
 501         return (0);
 502     if (!SSL_copy_session_id(tdata->ssl, (fdata->ssl)))
 503         return 0;
 504     return (1);
 505 }
 506
 507 void BIO_ssl_shutdown(BIO *b)
 508 {
 509     BIO_SSL *bdata;
 510
 511     for (; b != NULL; b = BIO_next(b)) {
 512         if (BIO_method_type(b) != BIO_TYPE_SSL)
 513             continue;
 514         bdata = BIO_get_data(b);
 515         if (bdata != NULL && bdata->ssl != NULL)
 516             SSL_shutdown(bdata->ssl);
 517     }
 518 }