2 ** Copyright (C) 2006 Thadeu Lima de Souza Cascardo <cascardo@minaslivre.org>
3 ** Copyright (C) 2009 Thadeu Lima de Souza Cascardo <cascardo@holoscopio.com>
5 ** This program is free software; you can redistribute it and/or modify
6 ** it under the terms of the GNU General Public License as published by
7 ** the Free Software Foundation; either version 2 of the License, or
8 ** (at your option) any later version.
10 ** This program is distributed in the hope that it will be useful,
11 ** but WITHOUT ANY WARRANTY; without even the implied warranty of
12 ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 ** GNU General Public License for more details.
15 ** You should have received a copy of the GNU General Public License
16 ** along with this program; if not, write to the Free Software
17 ** Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
21 #include <gnutls/gnutls.h>
26 #include "hcconn_internal.h"
30 gnutls_session_t session;
39 hc_conn_ssl_server_init_credentials (char *certfile, char *keyfile)
41 static int initialized = 0;
42 static gnutls_certificate_credentials_t cred;
43 gnutls_dh_params_t dh_params;
46 gnutls_dh_params_init (&dh_params);
47 gnutls_dh_params_generate2 (dh_params, DH_BITS);
48 gnutls_certificate_allocate_credentials (&cred);
49 gnutls_certificate_set_x509_key_file (cred, certfile, keyfile,
51 gnutls_certificate_set_dh_params (cred, dh_params);
57 ssl_server_get_credentials(void)
59 return hc_conn_ssl_server_init_credentials (NULL, NULL);
63 ssl_server_session_new (gnutls_session_t *session)
66 cred = ssl_server_get_credentials ();
67 gnutls_init (session, GNUTLS_SERVER);
68 gnutls_set_default_priority (*session);
69 gnutls_credentials_set (*session, GNUTLS_CRD_CERTIFICATE, cred);
70 gnutls_dh_set_prime_bits (*session, DH_BITS);
75 ssl_client_session_new (gnutls_session_t *session)
77 int kx_prio[] = {GNUTLS_KX_RSA, 0};
78 gnutls_certificate_credentials cred;
79 gnutls_certificate_allocate_credentials (&cred);
80 gnutls_init (session, GNUTLS_CLIENT);
81 gnutls_set_default_priority (*session);
82 gnutls_kx_set_priority (*session, kx_prio);
83 gnutls_credentials_set (*session, GNUTLS_CRD_CERTIFICATE, cred);
86 static struct ssl_data *
87 ssl_data_new (int server)
90 ssl = g_slice_new (struct ssl_data);
92 ssl_server_session_new (&ssl->session);
94 ssl_client_session_new (&ssl->session);
95 ssl->buffer = g_string_sized_new (4096);
96 ssl->handshaking = FALSE;
102 ssl_data_destroy (struct ssl_data *ssl)
104 gnutls_deinit (ssl->session);
105 g_string_free (ssl->buffer, TRUE);
106 g_slice_free (struct ssl_data, ssl);
110 ssl_push (gnutls_transport_ptr_t ptr, const void *buffer, size_t len)
113 struct ssl_data *ssl = conn->layer;
114 hc_conn_write (ssl->lowconn, (void *) buffer, len);
119 ssl_pull (gnutls_transport_ptr_t ptr, void *buffer, size_t len)
122 struct ssl_data *ssl = conn->layer;
124 if (ssl->handshaking == TRUE)
126 r = hc_conn_read (ssl->lowconn, buffer, len);
129 if (len > ssl->buffer->len)
131 r = ssl->buffer->len;
132 memcpy (buffer, ssl->buffer->str, r);
133 g_string_truncate (ssl->buffer, 0);
138 memcpy (buffer, ssl->buffer->str, r);
139 g_string_erase (ssl->buffer, 0, r);
150 ssl_server_handshake (struct ssl_data *ssl)
153 if ((error = gnutls_handshake (ssl->session)) < 0)
155 if (gnutls_error_is_fatal (error))
157 g_critical ("Fatal error while doing TLS handshaking: %s\n",
158 gnutls_strerror (error));
164 ssl->handshaking = FALSE;
169 ssl_server_connect (HCConn *conn)
171 struct ssl_data *ssl = conn->layer;
172 gnutls_transport_set_ptr (ssl->session, (gnutls_transport_ptr_t) conn);
173 gnutls_transport_set_push_function (ssl->session, ssl_push);
174 gnutls_transport_set_pull_function (ssl->session, ssl_pull);
175 ssl->handshaking = TRUE;
176 ssl_server_handshake (ssl);
180 hc_conn_ssl_close (gpointer data)
182 struct ssl_data *ssl = data;
185 gnutls_bye (ssl->session, GNUTLS_SHUT_RDWR);
186 hc_conn_close (ssl->lowconn);
187 ssl_data_destroy (ssl);
192 hc_conn_ssl_read (gpointer data, gchar *buffer, size_t len)
194 struct ssl_data *ssl = data;
195 return gnutls_record_recv (ssl->session, buffer, len);
199 hc_conn_ssl_write (gpointer data, gchar *buffer, size_t len)
201 struct ssl_data *ssl = data;
202 return gnutls_record_send (ssl->session, buffer, len);
206 hc_conn_ssl_watch (HCConn *conn, HCEvent event, gpointer data)
209 HCConn *ssl_conn = data;
210 struct ssl_data *ssl = ssl_conn->layer;
215 if (ssl->handshaking)
217 ssl_server_handshake (ssl);
218 /* FIXME: create HC_CONN_ERROR */
219 if (ssl->failed && ssl_conn->func)
220 ssl_conn->func (ssl_conn, HC_EVENT_CLOSE, ssl_conn->data);
223 while ((r = hc_conn_read (ssl->lowconn, buffer, sizeof (buffer))) > 0)
224 g_string_append_len (ssl->buffer, buffer, r);
225 if (ssl_conn->func && !ssl->handshaking)
226 ssl_conn->func (ssl_conn, event, ssl_conn->data);
230 ssl_conn->func (ssl_conn, event, ssl_conn->data);
235 hc_conn_set_driver_ssl (HCConn *conn, HCConn *lowconn, int server)
237 struct ssl_data *ssl;
238 ssl = ssl_data_new (server);
241 ssl->lowconn = lowconn;
243 conn->read = hc_conn_ssl_read;
244 conn->write = hc_conn_ssl_write;
245 conn->close = hc_conn_ssl_close;
246 hc_conn_set_callback (lowconn, hc_conn_ssl_watch, conn);
247 ssl_server_connect (conn);
252 hc_conn_set_driver_ssl_client (HCConn *conn, HCConn *lowconn)
254 return hc_conn_set_driver_ssl (conn, lowconn, 0);
258 hc_conn_set_driver_ssl_server (HCConn *conn, HCConn *lowconn)
260 return hc_conn_set_driver_ssl (conn, lowconn, 1);