1 /*
2 * Copyright 2021 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 <openssl/core.h>
11 #include "bio_local.h"
12
13 /*-
14 * Core BIO structure
15 * This is distinct from a BIO to prevent casting between the two which could
16 * lead to versioning problems.
17 */
18 struct ossl_core_bio_st {
19 CRYPTO_REF_COUNT ref_cnt;
20 CRYPTO_RWLOCK *ref_lock;
21 BIO *bio;
22 };
23
core_bio_new(void)24 static OSSL_CORE_BIO *core_bio_new(void)
25 {
26 OSSL_CORE_BIO *cb = OPENSSL_malloc(sizeof(*cb));
27
28 if (cb == NULL || (cb->ref_lock = CRYPTO_THREAD_lock_new()) == NULL) {
29 OPENSSL_free(cb);
30 return NULL;
31 }
32 cb->ref_cnt = 1;
33 return cb;
34 }
35
ossl_core_bio_up_ref(OSSL_CORE_BIO * cb)36 int ossl_core_bio_up_ref(OSSL_CORE_BIO *cb)
37 {
38 int ref = 0;
39
40 return CRYPTO_UP_REF(&cb->ref_cnt, &ref, cb->ref_lock);
41 }
42
ossl_core_bio_free(OSSL_CORE_BIO * cb)43 int ossl_core_bio_free(OSSL_CORE_BIO *cb)
44 {
45 int ref = 0, res = 1;
46
47 if (cb != NULL) {
48 CRYPTO_DOWN_REF(&cb->ref_cnt, &ref, cb->ref_lock);
49 if (ref <= 0) {
50 res = BIO_free(cb->bio);
51 CRYPTO_THREAD_lock_free(cb->ref_lock);
52 OPENSSL_free(cb);
53 }
54 }
55 return res;
56 }
57
ossl_core_bio_new_from_bio(BIO * bio)58 OSSL_CORE_BIO *ossl_core_bio_new_from_bio(BIO *bio)
59 {
60 OSSL_CORE_BIO *cb = core_bio_new();
61
62 if (cb == NULL || !BIO_up_ref(bio)) {
63 ossl_core_bio_free(cb);
64 return NULL;
65 }
66 cb->bio = bio;
67 return cb;
68 }
69
core_bio_new_from_new_bio(BIO * bio)70 static OSSL_CORE_BIO *core_bio_new_from_new_bio(BIO *bio)
71 {
72 OSSL_CORE_BIO *cb = NULL;
73
74 if (bio == NULL)
75 return NULL;
76 if ((cb = core_bio_new()) == NULL) {
77 BIO_free(bio);
78 return NULL;
79 }
80 cb->bio = bio;
81 return cb;
82 }
83
ossl_core_bio_new_file(const char * filename,const char * mode)84 OSSL_CORE_BIO *ossl_core_bio_new_file(const char *filename, const char *mode)
85 {
86 return core_bio_new_from_new_bio(BIO_new_file(filename, mode));
87 }
88
ossl_core_bio_new_mem_buf(const void * buf,int len)89 OSSL_CORE_BIO *ossl_core_bio_new_mem_buf(const void *buf, int len)
90 {
91 return core_bio_new_from_new_bio(BIO_new_mem_buf(buf, len));
92 }
93
ossl_core_bio_read_ex(OSSL_CORE_BIO * cb,void * data,size_t dlen,size_t * readbytes)94 int ossl_core_bio_read_ex(OSSL_CORE_BIO *cb, void *data, size_t dlen,
95 size_t *readbytes)
96 {
97 return BIO_read_ex(cb->bio, data, dlen, readbytes);
98 }
99
ossl_core_bio_write_ex(OSSL_CORE_BIO * cb,const void * data,size_t dlen,size_t * written)100 int ossl_core_bio_write_ex(OSSL_CORE_BIO *cb, const void *data, size_t dlen,
101 size_t *written)
102 {
103 return BIO_write_ex(cb->bio, data, dlen, written);
104 }
105
ossl_core_bio_gets(OSSL_CORE_BIO * cb,char * buf,int size)106 int ossl_core_bio_gets(OSSL_CORE_BIO *cb, char *buf, int size)
107 {
108 return BIO_gets(cb->bio, buf, size);
109 }
110
ossl_core_bio_puts(OSSL_CORE_BIO * cb,const char * buf)111 int ossl_core_bio_puts(OSSL_CORE_BIO *cb, const char *buf)
112 {
113 return BIO_puts(cb->bio, buf);
114 }
115
ossl_core_bio_ctrl(OSSL_CORE_BIO * cb,int cmd,long larg,void * parg)116 long ossl_core_bio_ctrl(OSSL_CORE_BIO *cb, int cmd, long larg, void *parg)
117 {
118 return BIO_ctrl(cb->bio, cmd, larg, parg);
119 }
120
ossl_core_bio_vprintf(OSSL_CORE_BIO * cb,const char * format,va_list args)121 int ossl_core_bio_vprintf(OSSL_CORE_BIO *cb, const char *format, va_list args)
122 {
123 return BIO_vprintf(cb->bio, format, args);
124 }
125