crypto/bio/bio_meth.c

   1 /*
   2  * Copyright 2016-2018 The OpenSSL Project Authors. All Rights Reserved.
   3  *
   4  * Licensed under the Apache License 2.0 (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 "bio_local.h"
  11 #include "internal/thread_once.h"
  12
  13 CRYPTO_RWLOCK *bio_type_lock = NULL;
  14 static CRYPTO_ONCE bio_type_init = CRYPTO_ONCE_STATIC_INIT;
  15
  16 DEFINE_RUN_ONCE_STATIC(do_bio_type_init)
  17 {
  18     bio_type_lock = CRYPTO_THREAD_lock_new();
  19     return bio_type_lock != NULL;
  20 }
  21
  22 int BIO_get_new_index(void)
  23 {
  24     static CRYPTO_REF_COUNT bio_count = BIO_TYPE_START;
  25     int newval;
  26
  27     if (!RUN_ONCE(&bio_type_init, do_bio_type_init)) {
  28         BIOerr(BIO_F_BIO_GET_NEW_INDEX, ERR_R_MALLOC_FAILURE);
  29         return -1;
  30     }
  31     if (!CRYPTO_UP_REF(&bio_count, &newval, bio_type_lock))
  32         return -1;
  33     return newval;
  34 }
  35
  36 BIO_METHOD *BIO_meth_new(int type, const char *name)
  37 {
  38     BIO_METHOD *biom = OPENSSL_zalloc(sizeof(BIO_METHOD));
  39
  40     if (biom == NULL
  41             || (biom->name = OPENSSL_strdup(name)) == NULL) {
  42         OPENSSL_free(biom);
  43         BIOerr(BIO_F_BIO_METH_NEW, ERR_R_MALLOC_FAILURE);
  44         return NULL;
  45     }
  46     biom->type = type;
  47     return biom;
  48 }
  49
  50 void BIO_meth_free(BIO_METHOD *biom)
  51 {
  52     if (biom != NULL) {
  53         OPENSSL_free(biom->name);
  54         OPENSSL_free(biom);
  55     }
  56 }
  57
  58 int (*BIO_meth_get_write(const BIO_METHOD *biom)) (BIO *, const char *, int)
  59 {
  60     return biom->bwrite_old;
  61 }
  62
  63 int (*BIO_meth_get_write_ex(const BIO_METHOD *biom)) (BIO *, const char *, size_t,
  64                                                 size_t *)
  65 {
  66     return biom->bwrite;
  67 }
  68
  69 /* Conversion for old style bwrite to new style */
  70 int bwrite_conv(BIO *bio, const char *data, size_t datal, size_t *written)
  71 {
  72     int ret;
  73
  74     if (datal > INT_MAX)
  75         datal = INT_MAX;
  76
  77     ret = bio->method->bwrite_old(bio, data, (int)datal);
  78
  79     if (ret <= 0) {
  80         *written = 0;
  81         return ret;
  82     }
  83
  84     *written = (size_t)ret;
  85
  86     return 1;
  87 }
  88
  89 int BIO_meth_set_write(BIO_METHOD *biom,
  90                        int (*bwrite) (BIO *, const char *, int))
  91 {
  92     biom->bwrite_old = bwrite;
  93     biom->bwrite = bwrite_conv;
  94     return 1;
  95 }
  96
  97 int BIO_meth_set_write_ex(BIO_METHOD *biom,
  98                        int (*bwrite) (BIO *, const char *, size_t, size_t *))
  99 {
 100     biom->bwrite_old = NULL;
 101     biom->bwrite = bwrite;
 102     return 1;
 103 }
 104
 105 int (*BIO_meth_get_read(const BIO_METHOD *biom)) (BIO *, char *, int)
 106 {
 107     return biom->bread_old;
 108 }
 109
 110 int (*BIO_meth_get_read_ex(const BIO_METHOD *biom)) (BIO *, char *, size_t, size_t *)
 111 {
 112     return biom->bread;
 113 }
 114
 115 /* Conversion for old style bread to new style */
 116 int bread_conv(BIO *bio, char *data, size_t datal, size_t *readbytes)
 117 {
 118     int ret;
 119
 120     if (datal > INT_MAX)
 121         datal = INT_MAX;
 122
 123     ret = bio->method->bread_old(bio, data, (int)datal);
 124
 125     if (ret <= 0) {
 126         *readbytes = 0;
 127         return ret;
 128     }
 129
 130     *readbytes = (size_t)ret;
 131
 132     return 1;
 133 }
 134
 135 int BIO_meth_set_read(BIO_METHOD *biom,
 136                       int (*bread) (BIO *, char *, int))
 137 {
 138     biom->bread_old = bread;
 139     biom->bread = bread_conv;
 140     return 1;
 141 }
 142
 143 int BIO_meth_set_read_ex(BIO_METHOD *biom,
 144                          int (*bread) (BIO *, char *, size_t, size_t *))
 145 {
 146     biom->bread_old = NULL;
 147     biom->bread = bread;
 148     return 1;
 149 }
 150
 151 int (*BIO_meth_get_puts(const BIO_METHOD *biom)) (BIO *, const char *)
 152 {
 153     return biom->bputs;
 154 }
 155
 156 int BIO_meth_set_puts(BIO_METHOD *biom,
 157                       int (*bputs) (BIO *, const char *))
 158 {
 159     biom->bputs = bputs;
 160     return 1;
 161 }
 162
 163 int (*BIO_meth_get_gets(const BIO_METHOD *biom)) (BIO *, char *, int)
 164 {
 165     return biom->bgets;
 166 }
 167
 168 int BIO_meth_set_gets(BIO_METHOD *biom,
 169                       int (*bgets) (BIO *, char *, int))
 170 {
 171     biom->bgets = bgets;
 172     return 1;
 173 }
 174
 175 long (*BIO_meth_get_ctrl(const BIO_METHOD *biom)) (BIO *, int, long, void *)
 176 {
 177     return biom->ctrl;
 178 }
 179
 180 int BIO_meth_set_ctrl(BIO_METHOD *biom,
 181                       long (*ctrl) (BIO *, int, long, void *))
 182 {
 183     biom->ctrl = ctrl;
 184     return 1;
 185 }
 186
 187 int (*BIO_meth_get_create(const BIO_METHOD *biom)) (BIO *)
 188 {
 189     return biom->create;
 190 }
 191
 192 int BIO_meth_set_create(BIO_METHOD *biom, int (*create) (BIO *))
 193 {
 194     biom->create = create;
 195     return 1;
 196 }
 197
 198 int (*BIO_meth_get_destroy(const BIO_METHOD *biom)) (BIO *)
 199 {
 200     return biom->destroy;
 201 }
 202
 203 int BIO_meth_set_destroy(BIO_METHOD *biom, int (*destroy) (BIO *))
 204 {
 205     biom->destroy = destroy;
 206     return 1;
 207 }
 208
 209 long (*BIO_meth_get_callback_ctrl(const BIO_METHOD *biom)) (BIO *, int, BIO_info_cb *)
 210 {
 211     return biom->callback_ctrl;
 212 }
 213
 214 int BIO_meth_set_callback_ctrl(BIO_METHOD *biom,
 215                                long (*callback_ctrl) (BIO *, int,
 216                                                       BIO_info_cb *))
 217 {
 218     biom->callback_ctrl = callback_ctrl;
 219     return 1;
 220 }