4efa24beb586caccb17a01e2a43086f7e835bc29
[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_GNUTLS)
92
93 #include <gnutls/gnutls.h>
94
95 #define SSLI_ERROR_WANT_READ GNUTLS_E_AGAIN
96 #define SSLI_ERROR_WANT_WRITE GNUTLS_E_AGAIN
97 #define SSLI_ERROR_ZERO_RETURN GNUTLS_E_PREMATURE_TERMINATION
98 #define SSLI_ERROR_CONNRESET GNUTLS_E_PREMATURE_TERMINATION
99 #define SSLI_ERROR_SYSCALL GNUTLS_E_PREMATURE_TERMINATION
100
101 typedef gnutls_session_t SSL_handle_t;
102
103 #else /* OpenSSL */
104 #include <openssl/x509v3.h>
105 #include <openssl/ssl.h>
106
107 #define SSLI_ERROR_WANT_READ SSL_ERROR_WANT_READ
108 #define SSLI_ERROR_WANT_WRITE SSL_ERROR_WANT_WRITE
109 #define SSLI_ERROR_ZERO_RETURN SSL_ERROR_ZERO_RETURN
110 #define SSLI_ERROR_CONNRESET SSL_ERROR_ZERO_RETURN
111 #define SSLI_ERROR_SYSCALL SSL_ERROR_SYSCALL
112
113 typedef SSL SSL_handle_t;
114
115 #endif
116
117 void SSLi_init(void);
118 void SSLi_deinit(void);
119 SSL_handle_t *SSLi_newconnection(int *fd, bool_t *SSLready);
120 bool_t SSLi_getSHA1Hash(SSL_handle_t *ssl, uint8_t *hash);
121 void SSLi_closeconnection(SSL_handle_t *ssl);
122 int SSLi_nonblockaccept(SSL_handle_t *ssl, bool_t *SSLready);
123 int SSLi_read(SSL_handle_t *ssl, uint8_t *buf, int len);
124 int SSLi_write(SSL_handle_t *ssl, uint8_t *buf, int len);
125 int SSLi_get_error(SSL_handle_t *ssl, int code);
126 bool_t SSLi_data_pending(SSL_handle_t *ssl);
127 void SSLi_shutdown(SSL_handle_t *ssl);
128 void SSLi_free(SSL_handle_t *ssl);
129
130 static inline void SSLi_hash2hex(uint8_t *hash, char *out)
131 {
132         int i, offset = 0;
133         for (i = 0; i < 20; i++)
134                 offset += sprintf(out + offset, "%02x", hash[i]);
135 }
136
137 static inline void SSLi_hex2hash(char *in, uint8_t *hash)
138 {
139         int i;
140         char byte[3];
141         int scanned;
142
143         byte[2] = '\0';
144         for (i = 0; i < 20; i++) {
145                 memcpy(byte, &in[i * 2], 2);
146                 sscanf(byte, "%02x", &scanned);
147                 hash[i] = scanned;
148         }
149 }
150 #endif
151