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>
28 static struct ssl_data *
29 ssl_data_new (char *server)
32 int kx_prio[] = {GNUTLS_KX_RSA, 0};
33 gnutls_certificate_credentials cred;
34 gnutls_certificate_allocate_credentials (&cred);
35 ssl = g_slice_new (struct ssl_data);
36 ssl->server = g_strdup (server);
37 gnutls_init (&ssl->session, GNUTLS_CLIENT);
38 gnutls_set_default_priority (ssl->session);
39 gnutls_set_default_priority (ssl->session);
40 gnutls_kx_set_priority (ssl->session, kx_prio);
41 gnutls_credentials_set (ssl->session, GNUTLS_CRD_CERTIFICATE, cred);
42 ssl->buffer = g_string_sized_new (4096);
43 ssl->handshaking = FALSE;
48 ssl_data_destroy (struct ssl_data *ssl)
50 gnutls_deinit (ssl->session);
52 g_string_free (ssl->buffer, TRUE);
53 g_slice_free (struct ssl_data, ssl);
57 ssl_push (gnutls_transport_ptr_t ptr, const void *buffer, size_t len)
59 net_hook_t *hook = ptr;
60 struct ssl_data *ssl = hook->data;
62 if (ssl->handshaking == TRUE)
64 g_io_channel_write_chars (hook->conn->iochannel, buffer, len,
68 gnet_conn_write (hook->conn, (void *) buffer, len);
73 ssl_pull (gnutls_transport_ptr_t ptr, void *buffer, size_t len)
75 net_hook_t *hook = ptr;
76 struct ssl_data *ssl = hook->data;
78 if (ssl->handshaking == TRUE)
80 g_io_channel_read_chars (hook->conn->iochannel, buffer, len,
84 if (len > ssl->buffer->len)
87 memcpy (buffer, ssl->buffer->str, r);
88 g_string_truncate (ssl->buffer, 0);
93 memcpy (buffer, ssl->buffer->str, r);
94 g_string_erase (ssl->buffer, 0, r);
98 gnutls_transport_set_errno (ssl->session, EAGAIN);
105 ssl_server_connect (net_hook_t *hook)
107 struct ssl_data *ssl = hook->data;
109 gnutls_transport_set_ptr (ssl->session, (gnutls_transport_ptr_t) hook);
110 gnutls_transport_set_push_function (ssl->session, ssl_push);
111 gnutls_transport_set_pull_function (ssl->session, ssl_pull);
112 ssl->handshaking = TRUE;
113 if ((error = gnutls_handshake (ssl->session)) < 0)
115 g_message ("%satal error while doing TLS handshaking.\n",
116 gnutls_error_is_fatal (error) ? "F" : "Nonf");
117 g_message ("%s\n", gnutls_strerror (error));
119 ssl->handshaking = FALSE;
123 ssl_server_close (net_hook_t *hook)
125 struct ssl_data *ssl = hook->data;
128 hook->peer->peer = NULL;
129 gnet_conn_disconnect (hook->peer->conn);
131 gnet_conn_delete (hook->conn);
134 gnutls_bye (ssl->session, GNUTLS_SHUT_RDWR);
135 ssl_data_destroy (ssl);
137 g_slice_free (net_hook_t, hook);
141 ssl_server_write (net_hook_t *hook)
146 ssl_server_read (net_hook_t *hook, gchar *buffer, size_t len)
148 struct ssl_data *ssl = hook->data;
150 g_string_append_len (ssl->buffer, buffer, len);
153 r = gnutls_record_recv (ssl->session, buffer, len);
155 gnet_conn_write (hook->peer->conn, buffer, r);
160 ssl_server_error (net_hook_t *hook)
162 g_message ("Error in POP3 client connection.");
166 ssl_server_hook_new (net_hook_t *client_hook, char *server, int port)
169 hook = g_slice_new (net_hook_t);
170 hook->conn = gnet_conn_new (server, port, nethook_event, hook);
171 hook->peer = client_hook;
173 hook->connect = ssl_server_connect;
174 hook->close = ssl_server_close;
175 hook->write = ssl_server_write;
176 hook->read = ssl_server_read;
177 hook->data = ssl_data_new (server);
178 gnet_conn_connect (hook->conn);
179 gnet_conn_read (hook->conn);