forked from memcached/memcached
-
Notifications
You must be signed in to change notification settings - Fork 0
/
tls.c
190 lines (170 loc) · 5.98 KB
/
tls.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
#include "memcached.h"
#ifdef TLS
#include "tls.h"
#include <string.h>
#include <sysexits.h>
#include <sys/param.h>
#ifndef MAXPATHLEN
#define MAXPATHLEN 4096
#endif
static pthread_mutex_t ssl_ctx_lock = PTHREAD_MUTEX_INITIALIZER;
const unsigned MAX_ERROR_MSG_SIZE = 128;
void SSL_LOCK() {
pthread_mutex_lock(&(ssl_ctx_lock));
}
void SSL_UNLOCK(void) {
pthread_mutex_unlock(&(ssl_ctx_lock));
}
/*
* Reads decrypted data from the underlying BIO read buffers,
* which reads from the socket.
*/
ssize_t ssl_read(conn *c, void *buf, size_t count) {
assert (c != NULL);
/* TODO : document the state machine interactions for SSL_read with
non-blocking sockets/ SSL re-negotiations
*/
return SSL_read(c->ssl, buf, count);
}
/*
* SSL sendmsg implementation. Perform a SSL_write.
*/
ssize_t ssl_sendmsg(conn *c, struct msghdr *msg, int flags) {
assert (c != NULL);
size_t buf_remain = settings.ssl_wbuf_size;
size_t bytes = 0;
size_t to_copy;
int i;
// ssl_wbuf is pointing to the buffer allocated in the worker thread.
assert(c->ssl_wbuf);
// TODO: allocate a fix buffer in crawler/logger if they start using
// the sendmsg method. Also, set c->ssl_wbuf when the side thread
// start owning the connection and reset the pointer in
// conn_worker_readd.
// Currntly this connection would not be served by a different thread
// than the one it's assigned.
assert(pthread_equal(c->thread->thread_id, pthread_self()) != 0);
char *bp = c->ssl_wbuf;
for (i = 0; i < msg->msg_iovlen; i++) {
size_t len = msg->msg_iov[i].iov_len;
to_copy = len < buf_remain ? len : buf_remain;
memcpy(bp + bytes, (void*)msg->msg_iov[i].iov_base, to_copy);
buf_remain -= to_copy;
bytes += to_copy;
if (buf_remain == 0)
break;
}
/* TODO : document the state machine interactions for SSL_write with
non-blocking sockets/ SSL re-negotiations
*/
return SSL_write(c->ssl, c->ssl_wbuf, bytes);
}
/*
* Writes data to the underlying BIO write buffers,
* which encrypt and write them to the socket.
*/
ssize_t ssl_write(conn *c, void *buf, size_t count) {
assert (c != NULL);
return SSL_write(c->ssl, buf, count);
}
/*
* Loads server certificates to the SSL context and validate them.
* @return whether certificates are successfully loaded and verified or not.
* @param error_msg contains the error when unsuccessful.
*/
static bool load_server_certificates(char **errmsg) {
bool success = true;
char *error_msg = malloc(MAXPATHLEN + MAX_ERROR_MSG_SIZE);
size_t errmax = MAXPATHLEN + MAX_ERROR_MSG_SIZE - 1;
if (error_msg == NULL) {
*errmsg = NULL;
return false;
}
SSL_LOCK();
if (!SSL_CTX_use_certificate_chain_file(settings.ssl_ctx,
settings.ssl_chain_cert)) {
snprintf(error_msg, errmax, "Error loading the certificate chain: %s\r\n",
settings.ssl_chain_cert);
success = false;
} else if (!SSL_CTX_use_PrivateKey_file(settings.ssl_ctx, settings.ssl_key,
settings.ssl_keyformat)) {
snprintf(error_msg, errmax, "Error loading the key: %s\r\n", settings.ssl_key);
success = false;
} else if (!SSL_CTX_check_private_key(settings.ssl_ctx)) {
snprintf(error_msg, errmax, "Error validating the certificate\r\n");
success = false;
} else if (settings.ssl_ca_cert) {
if (!SSL_CTX_load_verify_locations(settings.ssl_ctx,
settings.ssl_ca_cert, NULL)) {
snprintf(error_msg, errmax,
"Error loading the CA certificate: %s\r\n", settings.ssl_ca_cert);
success = false;
} else {
SSL_CTX_set_client_CA_list(settings.ssl_ctx,
SSL_load_client_CA_file(settings.ssl_ca_cert));
}
}
SSL_UNLOCK();
if (success) {
settings.ssl_last_cert_refresh_time = current_time;
free(error_msg);
} else {
*errmsg = error_msg;
}
return success;
}
/*
* Verify SSL settings and initiates the SSL context.
*/
int ssl_init(void) {
assert(settings.ssl_enabled);
// SSL context for the process. All connections will share one
// process level context.
settings.ssl_ctx = SSL_CTX_new(TLS_server_method());
// Clients should use at least TLSv1.2
int flags = SSL_OP_NO_SSLv2 | SSL_OP_NO_SSLv3 |
SSL_OP_NO_TLSv1 |SSL_OP_NO_TLSv1_1;
SSL_CTX_set_options(settings.ssl_ctx, flags);
// The server certificate, private key and validations.
char *error_msg;
if (!load_server_certificates(&error_msg)) {
if (settings.verbose) {
fprintf(stderr, "%s", error_msg);
}
free(error_msg);
exit(EX_USAGE);
}
// The verification mode of client certificate, default is SSL_VERIFY_PEER.
SSL_CTX_set_verify(settings.ssl_ctx, settings.ssl_verify_mode, NULL);
if (settings.ssl_ciphers && !SSL_CTX_set_cipher_list(settings.ssl_ctx,
settings.ssl_ciphers)) {
if (settings.verbose) {
fprintf(stderr, "Error setting the provided cipher(s): %s\n",
settings.ssl_ciphers);
}
exit(EX_USAGE);
}
return 0;
}
/*
* This method is registered with each SSL connection and abort the SSL session
* if a client initiates a renegotiation.
* TODO : Proper way to do this is to set SSL_OP_NO_RENEGOTIATION
* using the SSL_CTX_set_options but that option only available in
* openssl 1.1.0h or above.
*/
void ssl_callback(const SSL *s, int where, int ret) {
SSL* ssl = (SSL*)s;
if (SSL_in_before(ssl)) {
if (settings.verbose) {
fprintf(stderr, "%d: SSL renegotiation is not supported, "
"closing the connection\n", SSL_get_fd(ssl));
}
SSL_set_shutdown(ssl, SSL_SENT_SHUTDOWN | SSL_RECEIVED_SHUTDOWN);
return;
}
}
bool refresh_certs(char **errmsg) {
return load_server_certificates(errmsg);
}
#endif