ssl_server.c

   1 /*
   2 ** Copyright (C) 2006 Thadeu Lima de Souza Cascardo <cascardo@minaslivre.org>
   3 ** Copyright (C) 2009 Thadeu Lima de Souza Cascardo <cascardo@holoscopio.com>
   4 **
   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.
   9 **
  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.
  14 **
  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.
  18 **
  19 */
  20
  21 #include <gnutls/gnutls.h>
  22 #include <gnet.h>
  23 #include <glib.h>
  24 #include <string.h>
  25 #include <errno.h>
  26 #include "ssl.h"
  27
  28 static struct ssl_data *
  29 ssl_data_new (char *server)
  30 {
  31   struct ssl_data *ssl;
  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;
  44   return ssl;
  45 }
  46
  47 static void
  48 ssl_data_destroy (struct ssl_data *ssl)
  49 {
  50   gnutls_deinit (ssl->session);
  51   g_free (ssl->server);
  52   g_string_free (ssl->buffer, TRUE);
  53   g_slice_free (struct ssl_data, ssl);
  54 }
  55
  56 static ssize_t
  57 ssl_push (gnutls_transport_ptr_t ptr, const void *buffer, size_t len)
  58 {
  59   net_hook_t *hook = ptr;
  60   struct ssl_data *ssl = hook->data;
  61   int r;
  62   if (ssl->handshaking == TRUE)
  63     {
  64       g_io_channel_write_chars (hook->conn->iochannel, buffer, len,
  65                                 &r, NULL);
  66       return r;
  67     }
  68   gnet_conn_write (hook->conn, (void *) buffer, len);
  69   return len;
  70 }
  71
  72 static ssize_t
  73 ssl_pull (gnutls_transport_ptr_t ptr, void *buffer, size_t len)
  74 {
  75   net_hook_t *hook = ptr;
  76   struct ssl_data *ssl = hook->data;
  77   int r;
  78   if (ssl->handshaking == TRUE)
  79     {
  80       g_io_channel_read_chars (hook->conn->iochannel, buffer, len,
  81                                &r, NULL);
  82       return r;
  83     }
  84   if (len > ssl->buffer->len)
  85     {
  86       r = ssl->buffer->len;
  87       memcpy (buffer, ssl->buffer->str, r);
  88       g_string_truncate (ssl->buffer, 0);
  89     }
  90   else
  91     {
  92       r = len;
  93       memcpy (buffer, ssl->buffer->str, r);
  94       g_string_erase (ssl->buffer, 0, r);
  95     }
  96   if (r == 0)
  97     {
  98       gnutls_transport_set_errno (ssl->session, EAGAIN);
  99       return -1;
 100     }
 101   return r;
 102 }
 103
 104 static void
 105 ssl_server_connect (net_hook_t *hook)
 106 {
 107   struct ssl_data *ssl = hook->data;
 108   int error;
 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)
 114     {
 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));
 118     }
 119   ssl->handshaking = FALSE;
 120 }
 121
 122 static void
 123 ssl_server_close (net_hook_t *hook)
 124 {
 125   struct ssl_data *ssl = hook->data;
 126   if (hook->peer)
 127     {
 128       hook->peer->peer = NULL;
 129       gnet_conn_disconnect (hook->peer->conn);
 130     }
 131   gnet_conn_delete (hook->conn);
 132   if (ssl != NULL)
 133     {
 134       gnutls_bye (ssl->session, GNUTLS_SHUT_RDWR);
 135       ssl_data_destroy (ssl);
 136     }
 137   g_slice_free (net_hook_t, hook);
 138 }
 139
 140 static void
 141 ssl_server_write (net_hook_t *hook)
 142 {
 143 }
 144
 145 static void
 146 ssl_server_read (net_hook_t *hook, gchar *buffer, size_t len)
 147 {
 148   struct ssl_data *ssl = hook->data;
 149   int r;
 150   g_string_append_len (ssl->buffer, buffer, len);
 151   do
 152     {
 153       r = gnutls_record_recv (ssl->session, buffer, len);
 154       if (r > 0)
 155         gnet_conn_write (hook->peer->conn, buffer, r);
 156     } while (r > 0);
 157 }
 158
 159 static void
 160 ssl_server_error (net_hook_t *hook)
 161 {
 162   g_message ("Error in POP3 client connection.");
 163 }
 164
 165 net_hook_t *
 166 ssl_server_hook_new (net_hook_t *client_hook, char *server, int port)
 167 {
 168   net_hook_t *hook;
 169   hook = g_slice_new (net_hook_t);
 170   hook->conn = gnet_conn_new (server, port, nethook_event, hook);
 171   hook->peer = client_hook;
 172   hook->server = TRUE;
 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);
 180   return hook;
 181 }