2 * libratbox: a library used by ircd-ratbox and other things
3 * gnutls.c: gnutls related code
5 * Copyright (C) 2007-2008 ircd-ratbox development team
6 * Copyright (C) 2007-2008 Aaron Sethman <androsyn@ratbox.org>
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301
23 * $Id: gnutls.c 25568 2008-06-20 18:46:08Z androsyn $
26 #include <libratbox_config.h>
27 #include <ratbox_lib.h>
28 #include <commio-int.h>
29 #include <commio-ssl.h>
32 #include <gnutls/gnutls.h>
35 static gnutls_certificate_credentials x509
;
36 static gnutls_dh_params dh_params
;
40 #define SSL_P(x) *((gnutls_session_t *)F->ssl)
43 rb_ssl_shutdown(rb_fde_t
* F
)
46 if(F
== NULL
|| F
->ssl
== NULL
)
48 for (i
= 0; i
< 4; i
++)
50 if(gnutls_bye(SSL_P(F
), GNUTLS_SHUT_RDWR
) == GNUTLS_E_SUCCESS
)
53 gnutls_deinit(SSL_P(F
));
58 rb_ssl_handshake_count(rb_fde_t
*F
)
60 return F
->handshake_count
;
64 rb_ssl_clear_handshake_count(rb_fde_t
*F
)
66 F
->handshake_count
= 0;
70 rb_ssl_timeout(rb_fde_t
* F
, void *notused
)
72 lrb_assert(F
->accept
!= NULL
);
73 F
->accept
->callback(F
, RB_ERR_TIMEOUT
, NULL
, 0, F
->accept
->data
);
77 static int do_ssl_handshake(rb_fde_t
*F
, PF
*callback
)
82 ret
= gnutls_handshake(SSL_P(F
));
85 if((ret
== GNUTLS_E_INTERRUPTED
&& rb_ignore_errno(errno
)) || ret
== GNUTLS_E_AGAIN
)
87 if(gnutls_record_get_direction(SSL_P(F
)) == 0)
88 flags
= RB_SELECT_READ
;
90 flags
= RB_SELECT_WRITE
;
91 rb_setselect(F
, flags
, callback
, NULL
);
97 return 1; /* handshake is finished..go about life */
101 rb_ssl_tryaccept(rb_fde_t
* F
, void *data
)
104 struct acceptdata
*ad
;
106 lrb_assert(F
->accept
!= NULL
);
108 ret
= do_ssl_handshake(F
, rb_ssl_tryaccept
);
113 F
->accept
->callback(F
, RB_ERROR_SSL
, NULL
, 0, F
->accept
->data
);
116 /* do_ssl_handshake does the rb_setselect stuff */
123 rb_settimeout(F
, 0, NULL
, NULL
);
124 rb_setselect(F
, RB_SELECT_READ
| RB_SELECT_WRITE
, NULL
, NULL
);
128 ad
->callback(F
, RB_OK
, (struct sockaddr
*) &ad
->S
, ad
->addrlen
,
135 rb_ssl_start_accepted(rb_fde_t
* new_F
, ACCB
* cb
, void *data
, int timeout
)
137 gnutls_session_t
*ssl
;
138 new_F
->type
|= RB_FD_SSL
;
139 ssl
= new_F
->ssl
= rb_malloc(sizeof(gnutls_session_t
));
140 new_F
->accept
= rb_malloc(sizeof(struct acceptdata
));
142 new_F
->accept
->callback
= cb
;
143 new_F
->accept
->data
= data
;
144 rb_settimeout(new_F
, timeout
, rb_ssl_timeout
, NULL
);
146 new_F
->accept
->addrlen
= 0;
148 gnutls_init(ssl
, GNUTLS_SERVER
);
149 gnutls_set_default_priority(*ssl
);
150 gnutls_credentials_set(*ssl
, GNUTLS_CRD_CERTIFICATE
, x509
);
151 gnutls_dh_set_prime_bits(*ssl
, 1024);
152 gnutls_transport_set_ptr(*ssl
, (gnutls_transport_ptr_t
) (long int)new_F
->fd
);
153 if(do_ssl_handshake(new_F
, rb_ssl_tryaccept
))
155 struct acceptdata
*ad
= new_F
->accept
;
156 new_F
->accept
= NULL
;
157 ad
->callback(new_F
, RB_OK
, (struct sockaddr
*) &ad
->S
, ad
->addrlen
, ad
->data
);
167 rb_ssl_accept_setup(rb_fde_t
* F
, int new_fd
, struct sockaddr
*st
, int addrlen
)
171 new_F
= rb_find_fd(new_fd
);
172 new_F
->type
|= RB_FD_SSL
;
173 new_F
->ssl
= rb_malloc(sizeof(gnutls_session_t
));
174 new_F
->accept
= rb_malloc(sizeof(struct acceptdata
));
176 new_F
->accept
->callback
= F
->accept
->callback
;
177 new_F
->accept
->data
= F
->accept
->data
;
178 rb_settimeout(new_F
, 10, rb_ssl_timeout
, NULL
);
179 memcpy(&new_F
->accept
->S
, st
, addrlen
);
180 new_F
->accept
->addrlen
= addrlen
;
182 gnutls_init((gnutls_session_t
*)new_F
->ssl
, GNUTLS_SERVER
);
183 gnutls_set_default_priority(SSL_P(new_F
));
184 gnutls_credentials_set(SSL_P(new_F
), GNUTLS_CRD_CERTIFICATE
, x509
);
185 gnutls_dh_set_prime_bits(SSL_P(new_F
), 1024);
186 gnutls_transport_set_ptr(SSL_P(new_F
), (gnutls_transport_ptr_t
) (long int)new_fd
);
187 if(do_ssl_handshake(F
, rb_ssl_tryaccept
))
189 struct acceptdata
*ad
= F
->accept
;
191 ad
->callback(F
, RB_OK
, (struct sockaddr
*) &ad
->S
, ad
->addrlen
, ad
->data
);
200 rb_ssl_read_or_write(int r_or_w
, rb_fde_t
* F
, void *rbuf
, const void *wbuf
, size_t count
)
203 gnutls_session_t
*ssl
= F
->ssl
;
206 ret
= gnutls_record_recv(*ssl
, rbuf
, count
);
208 ret
= gnutls_record_send(*ssl
, wbuf
, count
);
215 case GNUTLS_E_INTERRUPTED
:
216 if(rb_ignore_errno(errno
))
218 if(gnutls_record_get_direction(*ssl
) == 0)
219 return RB_RW_SSL_NEED_READ
;
221 return RB_RW_SSL_NEED_WRITE
;
227 return RB_RW_IO_ERROR
;
234 rb_ssl_read(rb_fde_t
* F
, void *buf
, size_t count
)
236 return rb_ssl_read_or_write(0, F
, buf
, NULL
, count
);
240 rb_ssl_write(rb_fde_t
* F
, const void *buf
, size_t count
)
242 return rb_ssl_read_or_write(1, F
, NULL
, buf
, count
);
246 rb_gcry_random_seed(void *unused
)
248 gcry_fast_random_poll();
254 gnutls_global_init();
256 if(gnutls_certificate_allocate_credentials(&x509
) != GNUTLS_E_SUCCESS
)
258 rb_lib_log("rb_init_ssl: Unable to allocate SSL/TLS certificate credentials");
261 rb_event_addish("rb_gcry_random_seed", rb_gcry_random_seed
, NULL
, 300);
266 rb_free_datum_t(gnutls_datum_t
*d
)
272 static gnutls_datum_t
*
273 rb_load_file_into_datum_t(const char *file
)
276 gnutls_datum_t
*datum
;
277 struct stat fileinfo
;
278 if((f
= fopen(file
, "r")) == NULL
)
280 if(fstat(fileno(f
), &fileinfo
))
283 datum
= rb_malloc(sizeof(gnutls_datum_t
));
285 if(fileinfo
.st_size
> 131072) /* deal with retards */
286 datum
->size
= 131072;
288 datum
->size
= fileinfo
.st_size
;
290 datum
->data
= rb_malloc(datum
->size
+ 1);
291 fread(datum
->data
, datum
->size
, 1, f
);
297 rb_setup_ssl_server(const char *cert
, const char *keyfile
, const char *dhfile
)
300 gnutls_datum_t
*d_cert
, *d_key
;
303 rb_lib_log("rb_setup_ssl_server: No certificate file");
307 if((d_cert
= rb_load_file_into_datum_t(cert
)) == NULL
)
309 rb_lib_log("rb_setup_ssl_server: Error loading certificate: %s", strerror(errno
));
313 if((d_key
= rb_load_file_into_datum_t(keyfile
)) == NULL
)
315 rb_lib_log("rb_setup_ssl_server: Error loading key: %s", strerror(errno
));
320 if((ret
= gnutls_certificate_set_x509_key_mem(x509
, d_cert
, d_key
, GNUTLS_X509_FMT_PEM
)) != GNUTLS_E_SUCCESS
)
322 rb_lib_log("rb_setup_ssl_server: Error loading certificate or key file: %s", gnutls_strerror(ret
));
325 rb_free_datum_t(d_cert
);
326 rb_free_datum_t(d_key
);
330 if(gnutls_dh_params_init(&dh_params
) == GNUTLS_E_SUCCESS
)
332 gnutls_datum_t
*data
;
334 data
= rb_load_file_into_datum_t(dhfile
);
337 xret
= gnutls_dh_params_import_pkcs3(dh_params
, data
, GNUTLS_X509_FMT_PEM
);
339 rb_lib_log("rb_setup_ssl_server: Error parsing DH file: %s\n", gnutls_strerror(xret
));
340 rb_free_datum_t(data
);
342 gnutls_certificate_set_dh_params(x509
, dh_params
);
344 rb_lib_log("rb_setup_ssl_server: Unable to setup DH parameters");
350 rb_ssl_listen(rb_fde_t
* F
, int backlog
)
352 F
->type
= RB_FD_SOCKET
| RB_FD_LISTEN
| RB_FD_SSL
;
353 return listen(F
->fd
, backlog
);
364 rb_ssl_connect_realcb(rb_fde_t
* F
, int status
, struct ssl_connect
*sconn
)
366 F
->connect
->callback
= sconn
->callback
;
367 F
->connect
->data
= sconn
->data
;
369 rb_connect_callback(F
, status
);
373 rb_ssl_tryconn_timeout_cb(rb_fde_t
* F
, void *data
)
375 rb_ssl_connect_realcb(F
, RB_ERR_TIMEOUT
, data
);
379 rb_ssl_tryconn_cb(rb_fde_t
* F
, void *data
)
381 struct ssl_connect
*sconn
= data
;
384 ret
= do_ssl_handshake(F
, rb_ssl_tryconn_cb
);
389 rb_ssl_connect_realcb(F
, RB_ERROR_SSL
, sconn
);
392 /* do_ssl_handshake does the rb_setselect stuff */
399 rb_ssl_connect_realcb(F
, RB_OK
, sconn
);
403 rb_ssl_tryconn(rb_fde_t
* F
, int status
, void *data
)
405 struct ssl_connect
*sconn
= data
;
408 rb_ssl_connect_realcb(F
, status
, sconn
);
412 F
->type
|= RB_FD_SSL
;
415 rb_settimeout(F
, sconn
->timeout
, rb_ssl_tryconn_timeout_cb
, sconn
);
416 F
->ssl
= rb_malloc(sizeof(gnutls_session_t
));
417 gnutls_init(F
->ssl
, GNUTLS_CLIENT
);
418 gnutls_set_default_priority(SSL_P(F
));
419 gnutls_dh_set_prime_bits(SSL_P(F
), 1024);
420 gnutls_transport_set_ptr(SSL_P(F
), (gnutls_transport_ptr_t
) (long int)F
->fd
);
422 if(do_ssl_handshake(F
, rb_ssl_tryconn_cb
))
424 rb_ssl_connect_realcb(F
, RB_OK
, sconn
);
429 rb_connect_tcp_ssl(rb_fde_t
* F
, struct sockaddr
*dest
,
430 struct sockaddr
*clocal
, int socklen
, CNCB
* callback
, void *data
, int timeout
)
432 struct ssl_connect
*sconn
;
436 sconn
= rb_malloc(sizeof(struct ssl_connect
));
438 sconn
->callback
= callback
;
439 sconn
->timeout
= timeout
;
440 rb_connect_tcp(F
, dest
, clocal
, socklen
, rb_ssl_tryconn
, sconn
, timeout
);
445 rb_ssl_start_connected(rb_fde_t
* F
, CNCB
* callback
, void *data
, int timeout
)
447 struct ssl_connect
*sconn
;
451 sconn
= rb_malloc(sizeof(struct ssl_connect
));
453 sconn
->callback
= callback
;
454 sconn
->timeout
= timeout
;
455 F
->connect
= rb_malloc(sizeof(struct conndata
));
456 F
->connect
->callback
= callback
;
457 F
->connect
->data
= data
;
458 F
->type
|= RB_FD_SSL
;
459 F
->ssl
= rb_malloc(sizeof(gnutls_session_t
));
461 gnutls_init(F
->ssl
, GNUTLS_CLIENT
);
462 gnutls_set_default_priority(SSL_P(F
));
463 gnutls_dh_set_prime_bits(SSL_P(F
), 1024);
464 gnutls_transport_set_ptr(SSL_P(F
), (gnutls_transport_ptr_t
) (long int)F
->fd
);
466 rb_settimeout(F
, sconn
->timeout
, rb_ssl_tryconn_timeout_cb
, sconn
);
468 if(do_ssl_handshake(F
, rb_ssl_tryconn_cb
))
470 rb_ssl_connect_realcb(F
, RB_OK
, sconn
);
475 rb_init_prng(const char *path
, prng_seed_t seed_type
)
477 gcry_fast_random_poll();
482 rb_get_random(void *buf
, size_t length
)
484 gcry_randomize(buf
, length
, GCRY_STRONG_RANDOM
);
490 rb_get_ssl_strerror(rb_fde_t
* F
)
492 return gnutls_strerror(F
->ssl_errno
);
496 rb_supports_ssl(void)
501 #endif /* HAVE_GNUTLS */