2 * ustream-ssl - library for SSL over ustream
4 * Copyright (C) 2012 Felix Fietkau <nbd@openwrt.org>
6 * Permission to use, copy, modify, and/or distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
19 #include <sys/types.h>
25 #include "ustream-ssl.h"
26 #include "ustream-internal.h"
28 static int urandom_fd = -1;
30 static int s_ustream_read(void *ctx, unsigned char *buf, size_t len)
32 struct ustream *s = ctx;
39 sbuf = ustream_get_read_buf(s, &slen);
44 return POLARSSL_ERR_NET_WANT_READ;
46 memcpy(buf, sbuf, slen);
47 ustream_consume(s, slen);
52 static int s_ustream_write(void *ctx, const unsigned char *buf, size_t len)
54 struct ustream *s = ctx;
57 ret = ustream_write(s, (const char *) buf, len, false);
58 if (ret < 0 || s->write_error)
59 return POLARSSL_ERR_NET_SEND_FAILED;
64 __hidden void ustream_set_io(struct ustream_ssl_ctx *ctx, void *ssl, struct ustream *conn)
66 ssl_set_bio(ssl, s_ustream_read, conn, s_ustream_write, conn);
69 static bool urandom_init(void)
74 urandom_fd = open("/dev/urandom", O_RDONLY);
81 static int _urandom(void *ctx, unsigned char *out, size_t len)
83 if (read(urandom_fd, out, len) < 0)
84 return POLARSSL_ERR_ENTROPY_SOURCE_FAILED;
89 __hidden struct ustream_ssl_ctx *
90 __ustream_ssl_context_new(bool server)
92 struct ustream_ssl_ctx *ctx;
97 ctx = calloc(1, sizeof(*ctx));
101 ctx->server = server;
107 __hidden int __ustream_ssl_add_ca_crt_file(struct ustream_ssl_ctx *ctx, const char *file)
111 ret = x509_crt_parse_file(&ctx->ca_cert, file);
118 __hidden int __ustream_ssl_set_crt_file(struct ustream_ssl_ctx *ctx, const char *file)
122 ret = x509_crt_parse_file(&ctx->cert, file);
129 __hidden int __ustream_ssl_set_key_file(struct ustream_ssl_ctx *ctx, const char *file)
133 ret = pk_parse_keyfile(&ctx->key, file, NULL);
140 __hidden void __ustream_ssl_context_free(struct ustream_ssl_ctx *ctx)
143 x509_crt_free(&ctx->cert);
147 static void ustream_ssl_error(struct ustream_ssl *us, int ret)
150 uloop_timeout_set(&us->error_timer, 0);
153 static bool ssl_do_wait(int ret)
156 case POLARSSL_ERR_NET_WANT_READ:
157 case POLARSSL_ERR_NET_WANT_WRITE:
164 static void ustream_ssl_verify_cert(struct ustream_ssl *us)
167 const char *msg = NULL;
171 r = ssl_get_verify_result(ssl);
172 cn_mismatch = r & BADCERT_CN_MISMATCH;
173 r &= ~BADCERT_CN_MISMATCH;
175 if (r & BADCERT_EXPIRED)
176 msg = "certificate has expired";
177 else if (r & BADCERT_REVOKED)
178 msg = "certificate has been revoked";
179 else if (r & BADCERT_NOT_TRUSTED)
180 msg = "certificate is self-signed or not signed by a trusted CA";
182 msg = "unknown error";
185 if (us->notify_verify_error)
186 us->notify_verify_error(us, r, msg);
194 __hidden enum ssl_conn_status __ustream_ssl_connect(struct ustream_ssl *us)
199 r = ssl_handshake(ssl);
201 ustream_ssl_verify_cert(us);
206 return U_SSL_PENDING;
208 ustream_ssl_error(us, r);
212 __hidden int __ustream_ssl_write(struct ustream_ssl *us, const char *buf, int len)
215 int done = 0, ret = 0;
217 while (done != len) {
218 ret = ssl_write(ssl, (const unsigned char *) buf + done, len - done);
221 if (ssl_do_wait(ret))
224 ustream_ssl_error(us, ret);
234 __hidden int __ustream_ssl_read(struct ustream_ssl *us, char *buf, int len)
236 int ret = ssl_read(us->ssl, (unsigned char *) buf, len);
239 if (ssl_do_wait(ret))
240 return U_SSL_PENDING;
242 ustream_ssl_error(us, ret);
249 static const int default_ciphersuites[] =
251 #if defined(POLARSSL_AES_C)
252 #if defined(POLARSSL_SHA2_C)
253 TLS_RSA_WITH_AES_256_CBC_SHA256,
254 #endif /* POLARSSL_SHA2_C */
255 #if defined(POLARSSL_GCM_C) && defined(POLARSSL_SHA4_C)
256 TLS_RSA_WITH_AES_256_GCM_SHA384,
257 #endif /* POLARSSL_SHA2_C */
258 TLS_RSA_WITH_AES_256_CBC_SHA,
260 #if defined(POLARSSL_CAMELLIA_C)
261 #if defined(POLARSSL_SHA2_C)
262 TLS_RSA_WITH_CAMELLIA_256_CBC_SHA256,
263 #endif /* POLARSSL_SHA2_C */
264 TLS_RSA_WITH_CAMELLIA_256_CBC_SHA,
266 #if defined(POLARSSL_AES_C)
267 #if defined(POLARSSL_SHA2_C)
268 TLS_RSA_WITH_AES_128_CBC_SHA256,
269 #endif /* POLARSSL_SHA2_C */
270 #if defined(POLARSSL_GCM_C) && defined(POLARSSL_SHA2_C)
271 TLS_RSA_WITH_AES_128_GCM_SHA256,
272 #endif /* POLARSSL_SHA2_C */
273 TLS_RSA_WITH_AES_128_CBC_SHA,
275 #if defined(POLARSSL_CAMELLIA_C)
276 #if defined(POLARSSL_SHA2_C)
277 TLS_RSA_WITH_CAMELLIA_128_CBC_SHA256,
278 #endif /* POLARSSL_SHA2_C */
279 TLS_RSA_WITH_CAMELLIA_128_CBC_SHA,
281 #if defined(POLARSSL_DES_C)
282 TLS_RSA_WITH_3DES_EDE_CBC_SHA,
287 __hidden void *__ustream_ssl_session_new(struct ustream_ssl_ctx *ctx)
293 ssl = calloc(1, sizeof(ssl_context));
304 auth = SSL_VERIFY_NONE;
307 auth = SSL_VERIFY_OPTIONAL;
310 ssl_set_ciphersuites(ssl, default_ciphersuites);
311 ssl_set_endpoint(ssl, ep);
312 ssl_set_authmode(ssl, auth);
313 ssl_set_rng(ssl, _urandom, NULL);
317 ssl_set_ca_chain(ssl, ctx->cert.next, NULL, NULL);
318 ssl_set_own_cert(ssl, &ctx->cert, &ctx->key);
320 ssl_set_ca_chain(ssl, &ctx->cert, NULL, NULL);
323 ssl_session_reset(ssl);
328 __hidden void __ustream_ssl_session_free(void *ssl)
334 __hidden void __ustream_ssl_update_peer_cn(struct ustream_ssl *us)
336 struct ustream_ssl_ctx *ctx = us->ctx;
338 ssl_set_ca_chain(us->ssl, &ctx->ca_cert, NULL, us->peer_cn);