crypto/bio/bio_meth.c

   1 /*
   2  * Copyright 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 "bio_lcl.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()
  23 {
  24     static int 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_atomic_add(&bio_count, 1, &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->type = type;
  42         biom->name = name;
  43     }
  44     return biom;
  45 }
  46
  47 void BIO_meth_free(BIO_METHOD *biom)
  48 {
  49     OPENSSL_free(biom);
  50 }
  51
  52 int (*BIO_meth_get_write(BIO_METHOD *biom)) (BIO *, const char *, int)
  53 {
  54     return biom->bwrite_old;
  55 }
  56
  57 int (*BIO_meth_get_write_ex(BIO_METHOD *biom)) (BIO *, const char *, size_t,
  58                                                 size_t *)
  59 {
  60     return biom->bwrite;
  61 }
  62
  63 /* Conversion for old style bwrite to new style */
  64 int bwrite_conv(BIO *bio, const char *data, size_t datal, size_t *written)
  65 {
  66     int ret;
  67
  68     if (datal > INT_MAX)
  69         datal = INT_MAX;
  70
  71     ret = bio->method->bwrite_old(bio, data, (int)datal);
  72
  73     if (ret <= 0) {
  74         *written = 0;
  75         return ret;
  76     }
  77
  78     *written = (size_t)ret;
  79
  80     return 1;
  81 }
  82
  83 int BIO_meth_set_write(BIO_METHOD *biom,
  84                        int (*bwrite) (BIO *, const char *, int))
  85 {
  86     biom->bwrite_old = bwrite;
  87     biom->bwrite = bwrite_conv;
  88     return 1;
  89 }
  90
  91 int BIO_meth_set_write_ex(BIO_METHOD *biom,
  92                        int (*bwrite) (BIO *, const char *, size_t, size_t *))
  93 {
  94     biom->bwrite_old = NULL;
  95     biom->bwrite = bwrite;
  96     return 1;
  97 }
  98
  99 int (*BIO_meth_get_read(BIO_METHOD *biom)) (BIO *, char *, int)
 100 {
 101     return biom->bread_old;
 102 }
 103
 104 int (*BIO_meth_get_read_ex(BIO_METHOD *biom)) (BIO *, char *, size_t, size_t *)
 105 {
 106     return biom->bread;
 107 }
 108
 109 /* Conversion for old style bread to new style */
 110 int bread_conv(BIO *bio, char *data, size_t datal, size_t *readbytes)
 111 {
 112     int ret;
 113
 114     if (datal > INT_MAX)
 115         datal = INT_MAX;
 116
 117     ret = bio->method->bread_old(bio, data, (int)datal);
 118
 119     if (ret <= 0) {
 120         *readbytes = 0;
 121         return ret;
 122     }
 123
 124     *readbytes = (size_t)ret;
 125
 126     return 1;
 127 }
 128
 129 int BIO_meth_set_read(BIO_METHOD *biom,
 130                       int (*bread) (BIO *, char *, int))
 131 {
 132     biom->bread_old = bread;
 133     biom->bread = bread_conv;
 134     return 1;
 135 }
 136
 137 int BIO_meth_set_read_ex(BIO_METHOD *biom,
 138                          int (*bread) (BIO *, char *, size_t, size_t *))
 139 {
 140     biom->bread_old = NULL;
 141     biom->bread = bread;
 142     return 1;
 143 }
 144
 145 int (*BIO_meth_get_puts(BIO_METHOD *biom)) (BIO *, const char *)
 146 {
 147     return biom->bputs;
 148 }
 149
 150 int BIO_meth_set_puts(BIO_METHOD *biom,
 151                       int (*bputs) (BIO *, const char *))
 152 {
 153     biom->bputs = bputs;
 154     return 1;
 155 }
 156
 157 int (*BIO_meth_get_gets(BIO_METHOD *biom)) (BIO *, char *, int)
 158 {
 159     return biom->bgets;
 160 }
 161
 162 int BIO_meth_set_gets(BIO_METHOD *biom,
 163                       int (*bgets) (BIO *, char *, int))
 164 {
 165     biom->bgets = bgets;
 166     return 1;
 167 }
 168
 169 long (*BIO_meth_get_ctrl(BIO_METHOD *biom)) (BIO *, int, long, void *)
 170 {
 171     return biom->ctrl;
 172 }
 173
 174 int BIO_meth_set_ctrl(BIO_METHOD *biom,
 175                       long (*ctrl) (BIO *, int, long, void *))
 176 {
 177     biom->ctrl = ctrl;
 178     return 1;
 179 }
 180
 181 int (*BIO_meth_get_create(BIO_METHOD *biom)) (BIO *)
 182 {
 183     return biom->create;
 184 }
 185
 186 int BIO_meth_set_create(BIO_METHOD *biom, int (*create) (BIO *))
 187 {
 188     biom->create = create;
 189     return 1;
 190 }
 191
 192 int (*BIO_meth_get_destroy(BIO_METHOD *biom)) (BIO *)
 193 {
 194     return biom->destroy;
 195 }
 196
 197 int BIO_meth_set_destroy(BIO_METHOD *biom, int (*destroy) (BIO *))
 198 {
 199     biom->destroy = destroy;
 200     return 1;
 201 }
 202
 203 long (*BIO_meth_get_callback_ctrl(BIO_METHOD *biom)) (BIO *, int, bio_info_cb *)
 204 {
 205     return biom->callback_ctrl;
 206 }
 207
 208 int BIO_meth_set_callback_ctrl(BIO_METHOD *biom,
 209                                long (*callback_ctrl) (BIO *, int,
 210                                                       bio_info_cb *))
 211 {
 212     biom->callback_ctrl = callback_ctrl;
 213     return 1;
 214 }