Do user switch after SSL keys are initialized
[umurmur.git] / src / ssl.h
1 /* Copyright (C) 2009-2014, Martin Johansson <martin@fatbob.nu>
2
3    Copyright (C) 2005-2014, Thorvald Natvig <thorvald@natvig.com>
4    All rights reserved.
5
6    Redistribution and use in source and binary forms, with or without
7    modification, are permitted provided that the following conditions
8    are met:
9
10    - Redistributions of source code must retain the above copyright notice,
11      this list of conditions and the following disclaimer.
12    - Redistributions in binary form must reproduce the above copyright notice,
13      this list of conditions and the following disclaimer in the documentation
14      and/or other materials provided with the distribution.
15    - Neither the name of the Developers nor the names of its contributors may
16      be used to endorse or promote products derived from this software without
17      specific prior written permission.
18
19    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20    ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21    LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22    A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
23    CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
24    EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
25    PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
26    PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
27    LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
28    NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
29    SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30 */
31
32 #ifndef SSL_H_987698
33 #define SSL_H_987698
34
35 #include "config.h"
36 #include "types.h"
37
38 #include <inttypes.h>
39 #include <stdio.h>
40 #include <string.h>
41
42 #if defined(USE_POLARSSL)
43 #include <polarssl/ssl.h>
44 #include <polarssl/version.h>
45
46 #if defined(POLARSSL_VERSION_MAJOR)
47 #if (POLARSSL_VERSION_MAJOR < 1)
48 #error PolarSSL version 1.0.0 or greater is required!
49 #endif
50 #else
51 #error PolarSSL version 1.0.0 or greater is required!
52 #endif
53
54 #if defined(USE_POLARSSL_HAVEGE)
55 #include <polarssl/havege.h>
56     #if (POLARSSL_VERSION_MINOR >= 1)
57         #define HAVEGE_RAND (havege_random)
58         #define RAND_bytes(_dst_, _size_) do { \
59                 havege_random(&hs, _dst_, _size_); \
60         } while (0)
61     #else
62         #define HAVEGE_RAND (havege_rand)
63         #define RAND_bytes(_dst_, _size_) do { \
64             int i; \
65                 for (i = 0; i < _size_; i++) { \
66                     _dst_[i] = havege_rand(&hs); \
67                 } \
68         } while (0)
69     #endif
70 #else
71 #define RAND_bytes(_dst_, _size_) do { urandom_bytes(NULL, _dst_, _size_); } while (0)
72 int urandom_bytes(void *ctx, unsigned char *dest, size_t len);
73 #endif
74
75 #if (POLARSSL_VERSION_MINOR >= 2)
76     #define POLARSSL_API_V1_2_ABOVE
77 #endif
78 #if (POLARSSL_VERSION_MINOR == 3)
79     #define POLARSSL_API_V1_3_ABOVE
80 #endif
81
82 #define SSLI_ERROR_WANT_READ -0x0F300 /* PolarSSL v0.x.x uses -0x0f00 -> --0x0f90, v1.x.x uses -0x7080 -> -0x7e80 */
83 #define SSLI_ERROR_WANT_WRITE -0x0F310
84
85 #define SSLI_ERROR_ZERO_RETURN 0
86 #define SSLI_ERROR_CONNRESET POLARSSL_ERR_NET_CONN_RESET
87 #define SSLI_ERROR_SYSCALL POLARSSL_ERR_NET_RECV_FAILED
88
89 typedef ssl_context SSL_handle_t;
90
91 #elif defined(USE_MBEDTLS)
92 #include <mbedtls/ssl.h>
93 #include <mbedtls/net.h>
94 #include <mbedtls/version.h>
95
96 #if defined(MBEDTLS_VERSION_MAJOR)
97 #if (MBEDTLS_VERSION_MAJOR < 2)
98 #error mbedTLS version 2.0.0 or greater is required!
99 #endif
100 #else
101 #error mbedTLS version 2.0.0 or greater is required!
102 #endif
103
104 #if defined(USE_MBEDTLS_HAVEGE)
105 #include <mbedtls/havege.h>
106     #define HAVEGE_RAND (havege_random)
107     #define RAND_bytes(_dst_, _size_) do { \
108         mbedtls_havege_random(&hs, _dst_, _size_); \
109     } while (0)
110 #else
111 #define RAND_bytes(_dst_, _size_) do { urandom_bytes(NULL, _dst_, _size_); } while (0)
112 int urandom_bytes(void *ctx, unsigned char *dest, size_t len);
113 #endif
114
115 #define SSLI_ERROR_WANT_READ -0x0F300 /* mbedTLS v0.x.x uses -0x0f00 -> --0x0f90, v1.x.x uses -0x7080 -> -0x7e80 */
116 #define SSLI_ERROR_WANT_WRITE -0x0F310
117
118 #define SSLI_ERROR_ZERO_RETURN 0
119 #define SSLI_ERROR_CONNRESET MBEDTLS_ERR_NET_CONN_RESET
120 #define SSLI_ERROR_SYSCALL MBEDTLS_ERR_NET_RECV_FAILED
121
122 typedef mbedtls_ssl_context SSL_handle_t;
123
124 #elif defined(USE_GNUTLS)
125
126 #include <gnutls/gnutls.h>
127
128 #define SSLI_ERROR_WANT_READ GNUTLS_E_AGAIN
129 #define SSLI_ERROR_WANT_WRITE GNUTLS_E_AGAIN
130 #define SSLI_ERROR_ZERO_RETURN GNUTLS_E_PREMATURE_TERMINATION
131 #define SSLI_ERROR_CONNRESET GNUTLS_E_PREMATURE_TERMINATION
132 #define SSLI_ERROR_SYSCALL GNUTLS_E_PREMATURE_TERMINATION
133
134 typedef gnutls_session_t SSL_handle_t;
135
136 #else /* OpenSSL */
137 #include <openssl/x509v3.h>
138 #include <openssl/ssl.h>
139
140 #define SSLI_ERROR_WANT_READ SSL_ERROR_WANT_READ
141 #define SSLI_ERROR_WANT_WRITE SSL_ERROR_WANT_WRITE
142 #define SSLI_ERROR_ZERO_RETURN SSL_ERROR_ZERO_RETURN
143 #define SSLI_ERROR_CONNRESET SSL_ERROR_ZERO_RETURN
144 #define SSLI_ERROR_SYSCALL SSL_ERROR_SYSCALL
145
146 typedef SSL SSL_handle_t;
147
148 #endif
149
150 void SSLi_init(void);
151 void SSLi_deinit(void);
152 SSL_handle_t *SSLi_newconnection(int *fd, bool_t *SSLready);
153 bool_t SSLi_getSHA1Hash(SSL_handle_t *ssl, uint8_t *hash);
154 void SSLi_closeconnection(SSL_handle_t *ssl);
155 int SSLi_nonblockaccept(SSL_handle_t *ssl, bool_t *SSLready);
156 int SSLi_read(SSL_handle_t *ssl, uint8_t *buf, int len);
157 int SSLi_write(SSL_handle_t *ssl, uint8_t *buf, int len);
158 int SSLi_get_error(SSL_handle_t *ssl, int code);
159 bool_t SSLi_data_pending(SSL_handle_t *ssl);
160 void SSLi_shutdown(SSL_handle_t *ssl);
161 void SSLi_free(SSL_handle_t *ssl);
162
163 static inline void SSLi_hash2hex(uint8_t *hash, char *out)
164 {
165         int i, offset = 0;
166         for (i = 0; i < 20; i++)
167                 offset += sprintf(out + offset, "%02x", hash[i]);
168 }
169
170 static inline void SSLi_hex2hash(char *in, uint8_t *hash)
171 {
172         int i;
173         char byte[3];
174         int scanned;
175
176         byte[2] = '\0';
177         for (i = 0; i < 20; i++) {
178                 memcpy(byte, &in[i * 2], 2);
179                 sscanf(byte, "%02x", &scanned);
180                 hash[i] = scanned;
181         }
182 }
183 #endif
184