path: root/contrib/libs/openssl/ssl/bio_ssl.c

/* 
 * Copyright 1995-2021 The OpenSSL Project Authors. All Rights Reserved.
 * 
 * Licensed under the OpenSSL license (the "License").  You may not use 
 * this file except in compliance with the License.  You can obtain a copy 
 * in the file LICENSE in the source distribution or at 
 * https://www.openssl.org/source/license.html 
 */ 
 
#include <stdio.h> 
#include <stdlib.h> 
#include <string.h> 
#include <errno.h> 
#include <openssl/crypto.h> 
#include "internal/bio.h" 
#include <openssl/err.h> 
#include "ssl_local.h"
 
static int ssl_write(BIO *h, const char *buf, size_t size, size_t *written); 
static int ssl_read(BIO *b, char *buf, size_t size, size_t *readbytes); 
static int ssl_puts(BIO *h, const char *str); 
static long ssl_ctrl(BIO *h, int cmd, long arg1, void *arg2); 
static int ssl_new(BIO *h); 
static int ssl_free(BIO *data); 
static long ssl_callback_ctrl(BIO *h, int cmd, BIO_info_cb *fp); 
typedef struct bio_ssl_st { 
    SSL *ssl;                   /* The ssl handle :-) */ 
    /* re-negotiate every time the total number of bytes is this size */ 
    int num_renegotiates; 
    unsigned long renegotiate_count; 
    size_t byte_count; 
    unsigned long renegotiate_timeout; 
    unsigned long last_time; 
} BIO_SSL; 
 
static const BIO_METHOD methods_sslp = { 
    BIO_TYPE_SSL, 
    "ssl", 
    ssl_write, 
    NULL,                       /* ssl_write_old, */ 
    ssl_read, 
    NULL,                       /* ssl_read_old,  */ 
    ssl_puts, 
    NULL,                       /* ssl_gets,      */ 
    ssl_ctrl, 
    ssl_new, 
    ssl_free, 
    ssl_callback_ctrl, 
}; 
 
const BIO_METHOD *BIO_f_ssl(void) 
{ 
    return &methods_sslp; 
} 
 
static int ssl_new(BIO *bi) 
{ 
    BIO_SSL *bs = OPENSSL_zalloc(sizeof(*bs)); 
 
    if (bs == NULL) { 
        BIOerr(BIO_F_SSL_NEW, ERR_R_MALLOC_FAILURE); 
        return 0; 
    } 
    BIO_set_init(bi, 0); 
    BIO_set_data(bi, bs); 
    /* Clear all flags */ 
    BIO_clear_flags(bi, ~0); 
 
    return 1; 
} 
 
static int ssl_free(BIO *a) 
{ 
    BIO_SSL *bs; 
 
    if (a == NULL) 
        return 0; 
    bs = BIO_get_data(a); 
    if (bs->ssl != NULL) 
        SSL_shutdown(bs->ssl); 
    if (BIO_get_shutdown(a)) { 
        if (BIO_get_init(a)) 
            SSL_free(bs->ssl); 
        /* Clear all flags */ 
        BIO_clear_flags(a, ~0); 
        BIO_set_init(a, 0); 
    } 
    OPENSSL_free(bs); 
    return 1; 
} 
 
static int ssl_read(BIO *b, char *buf, size_t size, size_t *readbytes) 
{ 
    int ret = 1; 
    BIO_SSL *sb; 
    SSL *ssl; 
    int retry_reason = 0; 
    int r = 0; 
 
    if (buf == NULL) 
        return 0; 
    sb = BIO_get_data(b); 
    ssl = sb->ssl; 
 
    BIO_clear_retry_flags(b); 
 
    ret = ssl_read_internal(ssl, buf, size, readbytes); 
 
    switch (SSL_get_error(ssl, ret)) { 
    case SSL_ERROR_NONE: 
        if (sb->renegotiate_count > 0) { 
            sb->byte_count += *readbytes; 
            if (sb->byte_count > sb->renegotiate_count) { 
                sb->byte_count = 0; 
                sb->num_renegotiates++; 
                SSL_renegotiate(ssl); 
                r = 1; 
            } 
        } 
        if ((sb->renegotiate_timeout > 0) && (!r)) { 
            unsigned long tm; 
 
            tm = (unsigned long)time(NULL); 
            if (tm > sb->last_time + sb->renegotiate_timeout) { 
                sb->last_time = tm; 
                sb->num_renegotiates++; 
                SSL_renegotiate(ssl); 
            } 
        } 
 
        break; 
    case SSL_ERROR_WANT_READ: 
        BIO_set_retry_read(b); 
        break; 
    case SSL_ERROR_WANT_WRITE: 
        BIO_set_retry_write(b); 
        break; 
    case SSL_ERROR_WANT_X509_LOOKUP: 
        BIO_set_retry_special(b); 
        retry_reason = BIO_RR_SSL_X509_LOOKUP; 
        break; 
    case SSL_ERROR_WANT_ACCEPT: 
        BIO_set_retry_special(b); 
        retry_reason = BIO_RR_ACCEPT; 
        break; 
    case SSL_ERROR_WANT_CONNECT: 
        BIO_set_retry_special(b); 
        retry_reason = BIO_RR_CONNECT; 
        break; 
    case SSL_ERROR_SYSCALL: 
    case SSL_ERROR_SSL: 
    case SSL_ERROR_ZERO_RETURN: 
    default: 
        break; 
    } 
 
    BIO_set_retry_reason(b, retry_reason); 
 
    return ret; 
} 
 
static int ssl_write(BIO *b, const char *buf, size_t size, size_t *written) 
{ 
    int ret, r = 0; 
    int retry_reason = 0; 
    SSL *ssl; 
    BIO_SSL *bs; 
 
    if (buf == NULL) 
        return 0; 
    bs = BIO_get_data(b); 
    ssl = bs->ssl; 
 
    BIO_clear_retry_flags(b); 
 
    ret = ssl_write_internal(ssl, buf, size, written); 
 
    switch (SSL_get_error(ssl, ret)) { 
    case SSL_ERROR_NONE: 
        if (bs->renegotiate_count > 0) { 
            bs->byte_count += *written; 
            if (bs->byte_count > bs->renegotiate_count) { 
                bs->byte_count = 0; 
                bs->num_renegotiates++; 
                SSL_renegotiate(ssl); 
                r = 1; 
            } 
        } 
        if ((bs->renegotiate_timeout > 0) && (!r)) { 
            unsigned long tm; 
 
            tm = (unsigned long)time(NULL); 
            if (tm > bs->last_time + bs->renegotiate_timeout) { 
                bs->last_time = tm; 
                bs->num_renegotiates++; 
                SSL_renegotiate(ssl); 
            } 
        } 
        break; 
    case SSL_ERROR_WANT_WRITE: 
        BIO_set_retry_write(b); 
        break; 
    case SSL_ERROR_WANT_READ: 
        BIO_set_retry_read(b); 
        break; 
    case SSL_ERROR_WANT_X509_LOOKUP: 
        BIO_set_retry_special(b); 
        retry_reason = BIO_RR_SSL_X509_LOOKUP; 
        break; 
    case SSL_ERROR_WANT_CONNECT: 
        BIO_set_retry_special(b); 
        retry_reason = BIO_RR_CONNECT; 
    case SSL_ERROR_SYSCALL: 
    case SSL_ERROR_SSL: 
    default: 
        break; 
    } 
 
    BIO_set_retry_reason(b, retry_reason); 
 
    return ret; 
} 
 
static long ssl_ctrl(BIO *b, int cmd, long num, void *ptr) 
{ 
    SSL **sslp, *ssl; 
    BIO_SSL *bs, *dbs; 
    BIO *dbio, *bio; 
    long ret = 1; 
    BIO *next; 
 
    bs = BIO_get_data(b); 
    next = BIO_next(b); 
    ssl = bs->ssl; 
    if ((ssl == NULL) && (cmd != BIO_C_SET_SSL)) 
        return 0; 
    switch (cmd) { 
    case BIO_CTRL_RESET: 
        SSL_shutdown(ssl); 
 
        if (ssl->handshake_func == ssl->method->ssl_connect) 
            SSL_set_connect_state(ssl); 
        else if (ssl->handshake_func == ssl->method->ssl_accept) 
            SSL_set_accept_state(ssl); 
 
        if (!SSL_clear(ssl)) { 
            ret = 0; 
            break; 
        } 
 
        if (next != NULL) 
            ret = BIO_ctrl(next, cmd, num, ptr); 
        else if (ssl->rbio != NULL) 
            ret = BIO_ctrl(ssl->rbio, cmd, num, ptr); 
        else 
            ret = 1; 
        break; 
    case BIO_CTRL_INFO: 
        ret = 0; 
        break; 
    case BIO_C_SSL_MODE: 
        if (num)                /* client mode */ 
            SSL_set_connect_state(ssl); 
        else 
            SSL_set_accept_state(ssl); 
        break; 
    case BIO_C_SET_SSL_RENEGOTIATE_TIMEOUT: 
        ret = bs->renegotiate_timeout; 
        if (num < 60) 
            num = 5; 
        bs->renegotiate_timeout = (unsigned long)num; 
        bs->last_time = (unsigned long)time(NULL); 
        break; 
    case BIO_C_SET_SSL_RENEGOTIATE_BYTES: 
        ret = bs->renegotiate_count; 
        if ((long)num >= 512) 
            bs->renegotiate_count = (unsigned long)num; 
        break; 
    case BIO_C_GET_SSL_NUM_RENEGOTIATES: 
        ret = bs->num_renegotiates; 
        break; 
    case BIO_C_SET_SSL: 
        if (ssl != NULL) { 
            ssl_free(b); 
            if (!ssl_new(b)) 
                return 0; 
            bs = BIO_get_data(b);
        } 
        BIO_set_shutdown(b, num); 
        ssl = (SSL *)ptr; 
        bs->ssl = ssl; 
        bio = SSL_get_rbio(ssl); 
        if (bio != NULL) { 
            if (next != NULL) 
                BIO_push(bio, next); 
            BIO_set_next(b, bio); 
            BIO_up_ref(bio); 
        } 
        BIO_set_init(b, 1); 
        break; 
    case BIO_C_GET_SSL: 
        if (ptr != NULL) { 
            sslp = (SSL **)ptr; 
            *sslp = ssl; 
        } else 
            ret = 0; 
        break; 
    case BIO_CTRL_GET_CLOSE: 
        ret = BIO_get_shutdown(b); 
        break; 
    case BIO_CTRL_SET_CLOSE: 
        BIO_set_shutdown(b, (int)num); 
        break; 
    case BIO_CTRL_WPENDING: 
        ret = BIO_ctrl(ssl->wbio, cmd, num, ptr); 
        break; 
    case BIO_CTRL_PENDING: 
        ret = SSL_pending(ssl); 
        if (ret == 0) 
            ret = BIO_pending(ssl->rbio); 
        break; 
    case BIO_CTRL_FLUSH: 
        BIO_clear_retry_flags(b); 
        ret = BIO_ctrl(ssl->wbio, cmd, num, ptr); 
        BIO_copy_next_retry(b); 
        break; 
    case BIO_CTRL_PUSH: 
        if ((next != NULL) && (next != ssl->rbio)) { 
            /* 
             * We are going to pass ownership of next to the SSL object...but 
             * we don't own a reference to pass yet - so up ref 
             */ 
            BIO_up_ref(next); 
            SSL_set_bio(ssl, next, next); 
        } 
        break; 
    case BIO_CTRL_POP: 
        /* Only detach if we are the BIO explicitly being popped */ 
        if (b == ptr) { 
            /* This will clear the reference we obtained during push */ 
            SSL_set_bio(ssl, NULL, NULL); 
        } 
        break; 
    case BIO_C_DO_STATE_MACHINE: 
        BIO_clear_retry_flags(b); 
 
        BIO_set_retry_reason(b, 0); 
        ret = (int)SSL_do_handshake(ssl); 
 
        switch (SSL_get_error(ssl, (int)ret)) { 
        case SSL_ERROR_WANT_READ: 
            BIO_set_flags(b, BIO_FLAGS_READ | BIO_FLAGS_SHOULD_RETRY); 
            break; 
        case SSL_ERROR_WANT_WRITE: 
            BIO_set_flags(b, BIO_FLAGS_WRITE | BIO_FLAGS_SHOULD_RETRY); 
            break; 
        case SSL_ERROR_WANT_CONNECT: 
            BIO_set_flags(b, BIO_FLAGS_IO_SPECIAL | BIO_FLAGS_SHOULD_RETRY); 
            BIO_set_retry_reason(b, BIO_get_retry_reason(next)); 
            break; 
        case SSL_ERROR_WANT_X509_LOOKUP: 
            BIO_set_retry_special(b); 
            BIO_set_retry_reason(b, BIO_RR_SSL_X509_LOOKUP); 
            break; 
        default: 
            break; 
        } 
        break; 
    case BIO_CTRL_DUP: 
        dbio = (BIO *)ptr; 
        dbs = BIO_get_data(dbio); 
        SSL_free(dbs->ssl); 
        dbs->ssl = SSL_dup(ssl); 
        dbs->num_renegotiates = bs->num_renegotiates; 
        dbs->renegotiate_count = bs->renegotiate_count; 
        dbs->byte_count = bs->byte_count; 
        dbs->renegotiate_timeout = bs->renegotiate_timeout; 
        dbs->last_time = bs->last_time; 
        ret = (dbs->ssl != NULL); 
        break; 
    case BIO_C_GET_FD: 
        ret = BIO_ctrl(ssl->rbio, cmd, num, ptr); 
        break; 
    case BIO_CTRL_SET_CALLBACK: 
        ret = 0; /* use callback ctrl */ 
        break; 
    default: 
        ret = BIO_ctrl(ssl->rbio, cmd, num, ptr); 
        break; 
    } 
    return ret; 
} 
 
static long ssl_callback_ctrl(BIO *b, int cmd, BIO_info_cb *fp) 
{ 
    SSL *ssl; 
    BIO_SSL *bs; 
    long ret = 1; 
 
    bs = BIO_get_data(b); 
    ssl = bs->ssl; 
    switch (cmd) { 
    case BIO_CTRL_SET_CALLBACK: 
        ret = BIO_callback_ctrl(ssl->rbio, cmd, fp); 
        break; 
    default: 
        ret = 0; 
        break; 
    } 
    return ret; 
} 
 
static int ssl_puts(BIO *bp, const char *str) 
{ 
    int n, ret; 
 
    n = strlen(str); 
    ret = BIO_write(bp, str, n); 
    return ret; 
} 
 
BIO *BIO_new_buffer_ssl_connect(SSL_CTX *ctx) 
{ 
#ifndef OPENSSL_NO_SOCK 
    BIO *ret = NULL, *buf = NULL, *ssl = NULL; 
 
    if ((buf = BIO_new(BIO_f_buffer())) == NULL) 
        return NULL; 
    if ((ssl = BIO_new_ssl_connect(ctx)) == NULL) 
        goto err; 
    if ((ret = BIO_push(buf, ssl)) == NULL) 
        goto err; 
    return ret; 
 err: 
    BIO_free(buf); 
    BIO_free(ssl); 
#endif 
    return NULL; 
} 
 
BIO *BIO_new_ssl_connect(SSL_CTX *ctx) 
{ 
#ifndef OPENSSL_NO_SOCK 
    BIO *ret = NULL, *con = NULL, *ssl = NULL; 
 
    if ((con = BIO_new(BIO_s_connect())) == NULL) 
        return NULL; 
    if ((ssl = BIO_new_ssl(ctx, 1)) == NULL) 
        goto err; 
    if ((ret = BIO_push(ssl, con)) == NULL) 
        goto err; 
    return ret; 
 err: 
    BIO_free(ssl);
    BIO_free(con); 
#endif 
    return NULL; 
} 
 
BIO *BIO_new_ssl(SSL_CTX *ctx, int client) 
{ 
    BIO *ret; 
    SSL *ssl; 
 
    if ((ret = BIO_new(BIO_f_ssl())) == NULL) 
        return NULL; 
    if ((ssl = SSL_new(ctx)) == NULL) { 
        BIO_free(ret); 
        return NULL; 
    } 
    if (client) 
        SSL_set_connect_state(ssl); 
    else 
        SSL_set_accept_state(ssl); 
 
    BIO_set_ssl(ret, ssl, BIO_CLOSE); 
    return ret; 
} 
 
int BIO_ssl_copy_session_id(BIO *t, BIO *f) 
{ 
    BIO_SSL *tdata, *fdata; 
    t = BIO_find_type(t, BIO_TYPE_SSL); 
    f = BIO_find_type(f, BIO_TYPE_SSL); 
    if ((t == NULL) || (f == NULL)) 
        return 0; 
    tdata = BIO_get_data(t); 
    fdata = BIO_get_data(f); 
    if ((tdata->ssl == NULL) || (fdata->ssl == NULL)) 
        return 0; 
    if (!SSL_copy_session_id(tdata->ssl, (fdata->ssl))) 
        return 0; 
    return 1; 
} 
 
void BIO_ssl_shutdown(BIO *b) 
{ 
    BIO_SSL *bdata; 
 
    for (; b != NULL; b = BIO_next(b)) { 
        if (BIO_method_type(b) != BIO_TYPE_SSL) 
            continue; 
        bdata = BIO_get_data(b); 
        if (bdata != NULL && bdata->ssl != NULL) 
            SSL_shutdown(bdata->ssl); 
    } 
}