Move address caching logic from AsyncSSLSocket to AsyncSocket.
[folly.git] / folly / portability / OpenSSL.cpp
1 /*
2  * Copyright 2017 Facebook, Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *   http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 #include <folly/portability/OpenSSL.h>
17
18 #include <stdexcept>
19
20 namespace folly {
21 namespace portability {
22 namespace ssl {
23
24 #if OPENSSL_IS_BORINGSSL
25 int SSL_CTX_set1_sigalgs_list(SSL_CTX*, const char*) {
26   return 1; // 0 implies error
27 }
28
29 int TLS1_get_client_version(SSL* s) {
30   // Note that this isn't the client version, and the API to
31   // get this has been hidden. It may be found by parsing the
32   // ClientHello (there is a callback via the SSL_HANDSHAKE struct)
33   return s->version;
34 }
35 #endif
36
37 #if FOLLY_OPENSSL_IS_100
38 uint32_t SSL_CIPHER_get_id(const SSL_CIPHER* c) {
39   return c->id;
40 }
41
42 int TLS1_get_client_version(const SSL* s) {
43   return (s->client_version >> 8) == TLS1_VERSION_MAJOR ? s->client_version : 0;
44 }
45 #endif
46
47 #if FOLLY_OPENSSL_IS_100 || FOLLY_OPENSSL_IS_101
48 int X509_get_signature_nid(X509* cert) {
49   return OBJ_obj2nid(cert->sig_alg->algorithm);
50 }
51 #endif
52
53 #if FOLLY_OPENSSL_IS_100 || FOLLY_OPENSSL_IS_101 || FOLLY_OPENSSL_IS_102
54 int SSL_CTX_up_ref(SSL_CTX* ctx) {
55   return CRYPTO_add(&ctx->references, 1, CRYPTO_LOCK_SSL_CTX);
56 }
57
58 int SSL_SESSION_up_ref(SSL_SESSION* session) {
59   return CRYPTO_add(&session->references, 1, CRYPTO_LOCK_SSL_SESSION);
60 }
61
62 int X509_up_ref(X509* x) {
63   return CRYPTO_add(&x->references, 1, CRYPTO_LOCK_X509);
64 }
65
66 int EVP_PKEY_up_ref(EVP_PKEY* evp) {
67   return CRYPTO_add(&evp->references, 1, CRYPTO_LOCK_EVP_PKEY);
68 }
69
70 void RSA_get0_key(
71     const RSA* r,
72     const BIGNUM** n,
73     const BIGNUM** e,
74     const BIGNUM** d) {
75   if (n != nullptr) {
76     *n = r->n;
77   }
78   if (e != nullptr) {
79     *e = r->e;
80   }
81   if (d != nullptr) {
82     *d = r->d;
83   }
84 }
85
86 RSA* EVP_PKEY_get0_RSA(EVP_PKEY* pkey) {
87   if (pkey->type != EVP_PKEY_RSA) {
88     return nullptr;
89   }
90   return pkey->pkey.rsa;
91 }
92
93 EC_KEY* EVP_PKEY_get0_EC_KEY(EVP_PKEY* pkey) {
94   if (pkey->type != EVP_PKEY_EC) {
95     return nullptr;
96   }
97   return pkey->pkey.ec;
98 }
99 #endif
100
101 #if !FOLLY_OPENSSL_IS_110
102 void BIO_meth_free(BIO_METHOD* biom) {
103   OPENSSL_free((void*)biom);
104 }
105
106 int BIO_meth_set_read(BIO_METHOD* biom, int (*read)(BIO*, char*, int)) {
107   biom->bread = read;
108   return 1;
109 }
110
111 int BIO_meth_set_write(BIO_METHOD* biom, int (*write)(BIO*, const char*, int)) {
112   biom->bwrite = write;
113   return 1;
114 }
115
116 const char* SSL_SESSION_get0_hostname(const SSL_SESSION* s) {
117   return s->tlsext_hostname;
118 }
119
120 unsigned char* ASN1_STRING_get0_data(const ASN1_STRING* x) {
121   return ASN1_STRING_data((ASN1_STRING*)x);
122 }
123
124 int SSL_SESSION_has_ticket(const SSL_SESSION* s) {
125   return (s->tlsext_ticklen > 0) ? 1 : 0;
126 }
127
128 unsigned long SSL_SESSION_get_ticket_lifetime_hint(const SSL_SESSION* s) {
129   return s->tlsext_tick_lifetime_hint;
130 }
131
132 // This is taken from OpenSSL 1.1.0
133 int DH_set0_pqg(DH* dh, BIGNUM* p, BIGNUM* q, BIGNUM* g) {
134   /* If the fields p and g in d are nullptr, the corresponding input
135    * parameters MUST not be nullptr.  q may remain nullptr.
136    */
137   if (dh == nullptr || (dh->p == nullptr && p == nullptr) ||
138       (dh->g == nullptr && g == nullptr)) {
139     return 0;
140   }
141
142   if (p != nullptr) {
143     BN_free(dh->p);
144     dh->p = p;
145   }
146   if (q != nullptr) {
147     BN_free(dh->q);
148     dh->q = q;
149   }
150   if (g != nullptr) {
151     BN_free(dh->g);
152     dh->g = g;
153   }
154
155   // In OpenSSL 1.1.0, DH_set0_pqg also sets
156   //   dh->length = BN_num_bits(q)
157   // With OpenSSL 1.0.2, the output of openssl dhparam -C 2048 doesn't set
158   // the length field. So as far as the compat lib is concerned, this wrapper
159   // mimics the functionality of OpenSSL 1.0.2
160   // Note: BoringSSL doesn't even have a length field anymore, just something
161   // called 'priv_length'. Let's not mess with that for now.
162
163   return 1;
164 }
165
166 X509* X509_STORE_CTX_get0_cert(X509_STORE_CTX* ctx) {
167   return ctx->cert;
168 }
169
170 STACK_OF(X509) * X509_STORE_CTX_get0_chain(X509_STORE_CTX* ctx) {
171   return X509_STORE_CTX_get_chain(ctx);
172 }
173
174 STACK_OF(X509) * X509_STORE_CTX_get0_untrusted(X509_STORE_CTX* ctx) {
175   return ctx->untrusted;
176 }
177
178 EVP_MD_CTX* EVP_MD_CTX_new() {
179   EVP_MD_CTX* ctx = (EVP_MD_CTX*)OPENSSL_malloc(sizeof(EVP_MD_CTX));
180   if (!ctx) {
181     throw std::runtime_error("Cannot allocate EVP_MD_CTX");
182   }
183   EVP_MD_CTX_init(ctx);
184   return ctx;
185 }
186
187 void EVP_MD_CTX_free(EVP_MD_CTX* ctx) {
188   if (ctx) {
189     EVP_MD_CTX_cleanup(ctx);
190     OPENSSL_free(ctx);
191   }
192 }
193
194 HMAC_CTX* HMAC_CTX_new() {
195   HMAC_CTX* ctx = (HMAC_CTX*)OPENSSL_malloc(sizeof(HMAC_CTX));
196   if (!ctx) {
197     throw std::runtime_error("Cannot allocate HMAC_CTX");
198   }
199   HMAC_CTX_init(ctx);
200   return ctx;
201 }
202
203 void HMAC_CTX_free(HMAC_CTX* ctx) {
204   if (ctx) {
205     HMAC_CTX_cleanup(ctx);
206     OPENSSL_free(ctx);
207   }
208 }
209
210 #endif
211 }
212 }
213 }