]> git.ipfire.org Git - thirdparty/gnutls.git/commitdiff
Added mini-dtls-rehandshake.c to test rehandshake in DTLS.
authorNikos Mavrogiannopoulos <nmav@gnutls.org>
Sun, 26 Feb 2012 10:43:01 +0000 (11:43 +0100)
committerNikos Mavrogiannopoulos <nmav@gnutls.org>
Sun, 26 Feb 2012 10:43:01 +0000 (11:43 +0100)
.gitignore
tests/Makefile.am
tests/mini-dtls-rehandshake.c [new file with mode: 0644]
tests/mini-rehandshake.c [moved from tests/mini-x509-rehandshake.c with 99% similarity]
tests/resume-dtls.c [new file with mode: 0644]

index 9b7090c20cbb34ac533c82ecbec7b5498c24ba19..9d9be4b62dd3e88eb56facac2e7595e381f41025 100644 (file)
@@ -520,7 +520,8 @@ tests/mini-deflate
 tests/mini-eagain
 tests/mini-eagain-dtls
 tests/mini-x509
-tests/mini-x509-rehandshake
+tests/mini-dtls-rehandshake
+tests/mini-rehandshake
 tests/moredn
 tests/mpi
 tests/netconf-psk
@@ -588,3 +589,4 @@ tests/dtls/dtls-stress
 doc/gnutls.epub
 doc/gnutls.xml
 tests/mini-tdb
+tests/resume-dtls
index 085d018955cc226f1dd6fbeb3c57be8cbb21f5c1..40ce2c6178fe73be36b1cdcb3f079e0d4f3235ac 100644 (file)
@@ -65,9 +65,9 @@ ctests = mini-deflate simple gc set_pkcs12_cred certder certuniqueid  \
         crq_key_id x509sign-verify cve-2009-1415 cve-2009-1416         \
         crq_apis init_roundtrip pkcs12_s2k_pem dn2 mini-eagain         \
         nul-in-x509-names x509_altname pkcs12_encode mini-x509         \
-        mini-x509-rehandshake rng-fork mini-eagain-dtls \
+        mini-rehandshake rng-fork mini-eagain-dtls resume-dtls \
         x509cert x509cert-tl infoaccess rsa-encrypt-decrypt \
-        mini-loss-time mini-tdb
+        mini-loss-time mini-tdb mini-dtls-rehandshake
 
 if ENABLE_OCSP
 ctests += ocsp
diff --git a/tests/mini-dtls-rehandshake.c b/tests/mini-dtls-rehandshake.c
new file mode 100644 (file)
index 0000000..50fe6e4
--- /dev/null
@@ -0,0 +1,409 @@
+/*
+ * Copyright (C) 2012 Free Software Foundation, Inc.
+ *
+ * Author: Nikos Mavrogiannopoulos
+ *
+ * This file is part of GnuTLS.
+ *
+ * GnuTLS is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * GnuTLS is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with GnuTLS; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
+ */
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <stdio.h>
+#include <stdlib.h>
+
+#if defined(_WIN32)
+
+int main()
+{
+  exit(77);
+}
+
+#else
+
+#include <string.h>
+#include <sys/types.h>
+#include <netinet/in.h>
+#include <sys/socket.h>
+#include <sys/wait.h>
+#include <arpa/inet.h>
+#include <unistd.h>
+#include <gnutls/gnutls.h>
+#include <gnutls/dtls.h>
+
+#include "utils.h"
+
+static void terminate(void);
+
+/* This program tests the rehandshake in DTLS
+ */
+
+static void
+server_log_func (int level, const char *str)
+{
+  fprintf (stderr, "server|<%d>| %s", level, str);
+}
+
+static void
+client_log_func (int level, const char *str)
+{
+  fprintf (stderr, "client|<%d>| %s", level, str);
+}
+
+/* A very basic TLS client, with anonymous authentication.
+ */
+
+#define MAX_BUF 1024
+#define MSG "Hello TLS"
+
+gnutls_session_t session;
+
+static ssize_t
+push (gnutls_transport_ptr_t tr, const void *data, size_t len)
+{
+int fd = (long int)tr;
+
+  return send(fd, data, len, 0);
+}
+
+static void
+client (int fd, int server_init)
+{
+  int ret;
+  char buffer[MAX_BUF + 1];
+  gnutls_anon_client_credentials_t anoncred;
+  /* Need to enable anonymous KX specifically. */
+
+  gnutls_global_init ();
+
+  if (debug)
+    {
+      gnutls_global_set_log_function (client_log_func);
+      gnutls_global_set_log_level (4711);
+    }
+
+  gnutls_anon_allocate_client_credentials (&anoncred);
+
+  /* Initialize TLS session
+   */
+  gnutls_init (&session, GNUTLS_CLIENT|GNUTLS_DATAGRAM);
+  gnutls_dtls_set_mtu( session, 1500);
+
+  /* Use default priorities */
+  gnutls_priority_set_direct (session, "NONE:+VERS-DTLS1.0:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-ECDH:+CURVE-ALL", NULL);
+
+  /* put the anonymous credentials to the current session
+   */
+  gnutls_credentials_set (session, GNUTLS_CRD_ANON, anoncred);
+
+  gnutls_transport_set_ptr (session, (gnutls_transport_ptr_t) fd);
+  gnutls_transport_set_push_function (session, push);
+
+  /* Perform the TLS handshake
+   */
+  do 
+    {
+      ret = gnutls_handshake (session);
+    }
+  while (ret < 0 && gnutls_error_is_fatal(ret) == 0);
+
+  if (ret < 0)
+    {
+      fail ("client: Handshake failed\n");
+      gnutls_perror (ret);
+      exit(1);
+    }
+  else
+    {
+      if (debug)
+        success ("client: Handshake was completed\n");
+    }
+
+  if (debug)
+    success ("client: TLS version is: %s\n",
+             gnutls_protocol_get_name (gnutls_protocol_get_version
+                                       (session)));
+
+  if (!server_init)
+    {
+      do {
+        ret = gnutls_record_send (session, MSG, strlen (MSG));
+      } while (ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED);
+
+      if (debug) success("Initiating client rehandshake\n");
+      do 
+        {
+          ret = gnutls_handshake (session);
+        }
+      while (ret < 0 && gnutls_error_is_fatal(ret) == 0);
+
+      if (ret < 0)
+        {
+          fail ("2nd client gnutls_handshake: %s\n", gnutls_strerror(ret));
+          terminate();
+        }
+    }
+  else
+    {
+      do {
+        ret = gnutls_record_recv (session, buffer, MAX_BUF);
+      } while (ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED);
+    }
+
+  if (ret == 0)
+    {
+      if (debug)
+        success ("client: Peer has closed the TLS connection\n");
+      goto end;
+    }
+  else if (ret < 0)
+    {
+      if (server_init && ret == GNUTLS_E_REHANDSHAKE)
+        {
+          if (debug) success("Initiating rehandshake due to server request\n");
+          do 
+            {
+              ret = gnutls_handshake (session);
+            }
+          while (ret < 0 && gnutls_error_is_fatal(ret) == 0);
+        }
+
+      if (ret != 0)
+        {
+          fail ("client: Error: %s\n", gnutls_strerror (ret));
+          exit(1);
+        }
+    }
+
+  do {
+    ret = gnutls_record_send (session, MSG, strlen (MSG));
+  } while (ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED);
+  gnutls_bye (session, GNUTLS_SHUT_WR);
+
+end:
+
+  close (fd);
+
+  gnutls_deinit (session);
+
+  gnutls_anon_free_client_credentials (anoncred);
+
+  gnutls_global_deinit ();
+}
+
+
+/* These are global */
+gnutls_anon_server_credentials_t anoncred;
+pid_t child;
+
+static gnutls_session_t
+initialize_tls_session (void)
+{
+  gnutls_session_t session;
+
+  gnutls_init (&session, GNUTLS_SERVER|GNUTLS_DATAGRAM);
+  gnutls_dtls_set_mtu( session, 1500);
+
+  /* avoid calling all the priority functions, since the defaults
+   * are adequate.
+   */
+  gnutls_priority_set_direct (session, "NONE:+VERS-DTLS1.0:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-ECDH:+CURVE-ALL", NULL);
+
+  gnutls_credentials_set (session, GNUTLS_CRD_ANON, anoncred);
+
+  return session;
+}
+
+static void terminate(void)
+{
+int status;
+
+  kill(child, SIGTERM);
+  wait(&status);
+  exit(1);
+}
+
+static void
+server (int fd, int server_init)
+{
+int ret;
+char buffer[MAX_BUF + 1];
+  /* this must be called once in the program
+   */
+  gnutls_global_init ();
+
+  if (debug)
+    {
+      gnutls_global_set_log_function (server_log_func);
+      gnutls_global_set_log_level (4711);
+    }
+
+  gnutls_anon_allocate_server_credentials (&anoncred);
+
+  session = initialize_tls_session ();
+
+  gnutls_transport_set_ptr (session, (gnutls_transport_ptr_t) fd);
+  gnutls_transport_set_push_function (session, push);
+
+  do 
+    {
+      ret = gnutls_handshake (session);
+    }
+  while (ret < 0 && gnutls_error_is_fatal(ret) == 0);
+  if (ret < 0)
+    {
+      close (fd);
+      gnutls_deinit (session);
+      fail ("server: Handshake has failed (%s)\n\n", gnutls_strerror (ret));
+      terminate();
+    }
+  if (debug)
+    success ("server: Handshake was completed\n");
+
+  if (debug)
+    success ("server: TLS version is: %s\n",
+             gnutls_protocol_get_name (gnutls_protocol_get_version
+                                       (session)));
+
+  /* see the Getting peer's information example */
+  /* print_info(session); */
+
+  if (server_init)
+    {
+      if (debug) success("server: Sending dummy packet\n");
+      ret = gnutls_rehandshake(session);
+      if (ret < 0)
+        {
+          fail ("gnutls_rehandshake: %s\n", gnutls_strerror(ret));
+          terminate();
+        }
+
+      if (debug) success("server: Initiating rehandshake\n");
+      do 
+        {
+          ret = gnutls_handshake (session);
+        }
+      while (ret < 0 && gnutls_error_is_fatal(ret) == 0);
+
+      if (ret < 0)
+        {
+          fail ("server: 2nd gnutls_handshake: %s\n", gnutls_strerror(ret));
+          terminate();
+        }
+    }
+
+  for (;;)
+    {
+      memset (buffer, 0, MAX_BUF + 1);
+
+      do {
+        ret = gnutls_record_recv (session, buffer, MAX_BUF);
+      } while (ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED);
+
+      if (ret == 0)
+        {
+          if (debug)
+            success ("server: Peer has closed the GnuTLS connection\n");
+          break;
+        }
+      else if (ret < 0)
+        {
+          if (!server_init && ret == GNUTLS_E_REHANDSHAKE)
+            {
+              if (debug) success("Initiating rehandshake due to client request\n");
+              do 
+                {
+                  ret = gnutls_handshake (session);
+                }
+              while (ret < 0 && gnutls_error_is_fatal(ret) == 0);
+              if (ret == 0) break;
+            }
+
+          fail ("server: Received corrupted data(%s). Closing...\n", gnutls_strerror(ret));
+          terminate();
+        }
+      else if (ret > 0)
+        {
+          /* echo data back to the client
+           */
+          do {
+            ret = gnutls_record_send (session, buffer, strlen (buffer));
+          } while (ret == GNUTLS_E_AGAIN || ret == GNUTLS_E_INTERRUPTED);
+        }
+    }
+  
+
+  /* do not wait for the peer to close the connection.
+   */
+  gnutls_bye (session, GNUTLS_SHUT_WR);
+
+  close (fd);
+  gnutls_deinit (session);
+
+  gnutls_anon_free_server_credentials (anoncred);
+
+  gnutls_global_deinit ();
+
+  if (debug)
+    success ("server: finished\n");
+}
+
+static void start (int server_initiated)
+{
+  int fd[2];
+  int ret;
+  
+  ret = socketpair(AF_LOCAL, SOCK_DGRAM, 0, fd);
+  if (ret < 0)
+    {
+      perror("socketpair");
+      exit(1);
+    }
+
+  child = fork ();
+  if (child < 0)
+    {
+      perror ("fork");
+      fail ("fork");
+      exit(1);
+    }
+
+  if (child)
+    {
+      int status;
+      /* parent */
+      server (fd[0], server_initiated);
+      wait (&status);
+      if (WEXITSTATUS(status) != 0)
+        fail("Child died with status %d\n", WEXITSTATUS(status));
+    }
+  else 
+    {
+      client (fd[1], server_initiated);
+      exit(0);
+    }
+}
+
+void
+doit (void)
+{
+  start(0);
+  start(1);
+}
+
+#endif /* _WIN32 */
similarity index 99%
rename from tests/mini-x509-rehandshake.c
rename to tests/mini-rehandshake.c
index afad1bf38a7b3cc4a691f01bef5bb7d56b77143f..a5dc9013aef714814425462414ae06bd64370b0a 100644 (file)
@@ -32,7 +32,7 @@
 #include "utils.h"
 #include "eagain-common.h"
 
-const char* side;
+const char* side = "";
 
 static void
 tls_log_func (int level, const char *str)
diff --git a/tests/resume-dtls.c b/tests/resume-dtls.c
new file mode 100644 (file)
index 0000000..acdb8b5
--- /dev/null
@@ -0,0 +1,667 @@
+/*
+ * Copyright (C) 2004-2012 Free Software Foundation, Inc.
+ *
+ * Author: Simon Josefsson
+ *
+ * This file is part of GnuTLS.
+ *
+ * GnuTLS is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * GnuTLS is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with GnuTLS; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
+ */
+
+/* Parts copied from GnuTLS example programs. */
+
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#if !defined(_WIN32)
+#include <sys/wait.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#endif
+#include <unistd.h>
+#include <gnutls/gnutls.h>
+
+#include "tcp.c"
+
+#include "utils.h"
+
+static void wrap_db_init (void);
+static void wrap_db_deinit (void);
+static int wrap_db_store (void *dbf, gnutls_datum_t key, gnutls_datum_t data);
+static gnutls_datum_t wrap_db_fetch (void *dbf, gnutls_datum_t key);
+static int wrap_db_delete (void *dbf, gnutls_datum_t key);
+
+#define TLS_SESSION_CACHE 50
+
+struct params_res
+{
+  const char *desc;
+  int enable_db;
+  int enable_session_ticket_server;
+  int enable_session_ticket_client;
+  int expect_resume;
+};
+
+pid_t child;
+
+struct params_res resume_tests[] = {
+  {"try to resume from db", 50, 0, 0, 1},
+  {"try to resume from session ticket", 0, 1, 1, 1},
+  {"try to resume from session ticket (server only)", 0, 1, 0, 0},
+  {"try to resume from session ticket (client only)", 0, 0, 1, 0},
+  {NULL, -1}
+};
+
+/* A very basic TLS client, with anonymous authentication.
+ */
+
+#define MAX_BUF 5*1024
+#define MSG "Hello TLS"
+
+static void
+tls_log_func (int level, const char *str)
+{
+  fprintf (stderr, "%s |<%d>| %s", child ? "server" : "client", level, str);
+}
+
+static void
+client (struct params_res *params)
+{
+  int ret, sd, ii;
+  gnutls_session_t session;
+  char buffer[MAX_BUF + 1];
+  gnutls_anon_client_credentials_t anoncred;
+  /* Need to enable anonymous KX specifically. */
+
+  /* variables used in session resuming
+   */
+  int t;
+  gnutls_datum_t session_data;
+
+  if (debug)
+    {
+      gnutls_global_set_log_function (tls_log_func);
+      gnutls_global_set_log_level (2);
+    }
+  gnutls_global_init ();
+
+  gnutls_anon_allocate_client_credentials (&anoncred);
+
+  for (t = 0; t < 2; t++)
+    {                           /* connect 2 times to the server */
+      /* connect to the peer
+       */
+      sd = tcp_connect ();
+
+      /* Initialize TLS session
+       */
+      gnutls_init (&session, GNUTLS_CLIENT|GNUTLS_DATAGRAM);
+
+      /* Use default priorities */
+      gnutls_priority_set_direct (session, "NONE:+VERS-DTLS1.0:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-DH", NULL);
+
+      /* put the anonymous credentials to the current session
+       */
+      gnutls_credentials_set (session, GNUTLS_CRD_ANON, anoncred);
+
+      if (params->enable_session_ticket_client)
+        gnutls_session_ticket_enable_client (session);
+
+      if (t > 0)
+        {
+          /* if this is not the first time we connect */
+          gnutls_session_set_data (session, session_data.data,
+                                   session_data.size);
+          gnutls_free (session_data.data);
+        }
+
+      gnutls_transport_set_ptr (session, (gnutls_transport_ptr_t) sd);
+
+      /* Perform the TLS handshake
+       */
+      ret = gnutls_handshake (session);
+
+      if (ret < 0)
+        {
+          fail ("client: Handshake failed\n");
+          gnutls_perror (ret);
+          goto end;
+        }
+      else
+        {
+          if (debug)
+            success ("client: Handshake was completed\n");
+        }
+
+      if (t == 0)
+        {                       /* the first time we connect */
+          /* get the session data size */
+          ret = gnutls_session_get_data2 (session, &session_data);
+          if (ret < 0)
+            fail ("Getting resume data failed\n");
+        }
+      else
+        {                       /* the second time we connect */
+
+          /* check if we actually resumed the previous session */
+          if (gnutls_session_is_resumed (session) != 0)
+            {
+              if (params->expect_resume)
+                {
+                  if (debug)
+                    success ("- Previous session was resumed\n");
+                }
+              else
+                fail ("- Previous session was resumed\n");
+            }
+          else
+            {
+              if (params->expect_resume)
+                {
+                  fail ("*** Previous session was NOT resumed\n");
+                }
+              else
+                {
+                  if (debug)
+                    success
+                      ("*** Previous session was NOT resumed (expected)\n");
+                }
+            }
+        }
+
+      gnutls_record_send (session, MSG, strlen (MSG));
+
+      ret = gnutls_record_recv (session, buffer, MAX_BUF);
+      if (ret == 0)
+        {
+          if (debug)
+            success ("client: Peer has closed the TLS connection\n");
+          goto end;
+        }
+      else if (ret < 0)
+        {
+          fail ("client: Error: %s\n", gnutls_strerror (ret));
+          goto end;
+        }
+
+      if (debug )
+        {
+          printf ("- Received %d bytes: ", ret);
+          for (ii = 0; ii < ret; ii++)
+            {
+              fputc (buffer[ii], stdout);
+            }
+          fputs ("\n", stdout);
+        }
+
+      gnutls_bye (session, GNUTLS_SHUT_RDWR);
+
+
+      tcp_close (sd);
+
+      gnutls_deinit (session);
+    }
+
+end:
+  gnutls_anon_free_client_credentials (anoncred);
+}
+
+/* This is a sample TLS 1.0 echo server, for anonymous authentication only.
+ */
+
+#define SA struct sockaddr
+#define PORT 5556               /* listen to 5556 port */
+#define DH_BITS 1024
+
+/* These are global */
+gnutls_anon_server_credentials_t anoncred;
+static gnutls_datum_t session_ticket_key = { NULL, 0 };
+
+static gnutls_session_t
+initialize_tls_session (struct params_res *params)
+{
+  gnutls_session_t session;
+
+  gnutls_init (&session, GNUTLS_SERVER|GNUTLS_DATAGRAM);
+
+  /* avoid calling all the priority functions, since the defaults
+   * are adequate.
+   */
+  gnutls_priority_set_direct (session, "NONE:+VERS-DTLS1.0:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-DH", NULL);
+
+  gnutls_credentials_set (session, GNUTLS_CRD_ANON, anoncred);
+
+  gnutls_dh_set_prime_bits (session, DH_BITS);
+
+  if (params->enable_db)
+    {
+      gnutls_db_set_retrieve_function (session, wrap_db_fetch);
+      gnutls_db_set_remove_function (session, wrap_db_delete);
+      gnutls_db_set_store_function (session, wrap_db_store);
+      gnutls_db_set_ptr (session, NULL);
+    }
+
+  if (params->enable_session_ticket_server)
+    gnutls_session_ticket_enable_server (session, &session_ticket_key);
+
+  return session;
+}
+
+static gnutls_dh_params_t dh_params;
+
+static int
+generate_dh_params (void)
+{
+  const gnutls_datum_t p3 = { (void *) pkcs3, strlen (pkcs3) };
+  /* Generate Diffie-Hellman parameters - for use with DHE
+   * kx algorithms. These should be discarded and regenerated
+   * once a day, once a week or once a month. Depending on the
+   * security requirements.
+   */
+  gnutls_dh_params_init (&dh_params);
+  return gnutls_dh_params_import_pkcs3 (dh_params, &p3, GNUTLS_X509_FMT_PEM);
+}
+
+int err, listen_sd, i;
+int sd, ret;
+struct sockaddr_in sa_serv;
+struct sockaddr_in sa_cli;
+socklen_t client_len;
+char topbuf[512];
+gnutls_session_t session;
+char buffer[MAX_BUF + 1];
+int optval = 1;
+
+static void
+global_start (void)
+{
+  /* Socket operations
+   */
+  listen_sd = socket (AF_INET, SOCK_STREAM, 0);
+  if (err == -1)
+    {
+      perror ("socket");
+      fail ("server: socket failed\n");
+      return;
+    }
+
+  memset (&sa_serv, '\0', sizeof (sa_serv));
+  sa_serv.sin_family = AF_INET;
+  sa_serv.sin_addr.s_addr = INADDR_ANY;
+  sa_serv.sin_port = htons (PORT);      /* Server Port number */
+
+  setsockopt (listen_sd, SOL_SOCKET, SO_REUSEADDR, (void *) &optval,
+              sizeof (int));
+
+  err = bind (listen_sd, (SA *) & sa_serv, sizeof (sa_serv));
+  if (err == -1)
+    {
+      perror ("bind");
+      fail ("server: bind failed\n");
+      return;
+    }
+
+  err = listen (listen_sd, 1024);
+  if (err == -1)
+    {
+      perror ("listen");
+      fail ("server: listen failed\n");
+      return;
+    }
+
+  if (debug)
+    success ("server: ready. Listening to port '%d'.\n", PORT);
+}
+
+static void
+global_stop (void)
+{
+  if (debug)
+    success ("global stop\n");
+
+  gnutls_anon_free_server_credentials (anoncred);
+
+  gnutls_dh_params_deinit (dh_params);
+
+  gnutls_global_deinit ();
+
+  shutdown (listen_sd, SHUT_RDWR);
+}
+
+static void
+server (struct params_res *params)
+{
+  size_t t;
+
+  /* this must be called once in the program, it is mostly for the server.
+   */
+  if (debug)
+    {
+      gnutls_global_set_log_function (tls_log_func);
+      gnutls_global_set_log_level (2);
+    }
+
+  gnutls_global_init ();
+  gnutls_anon_allocate_server_credentials (&anoncred);
+
+  if (debug)
+    success ("Launched, generating DH parameters...\n");
+
+  generate_dh_params ();
+
+  gnutls_anon_set_server_dh_params (anoncred, dh_params);
+
+  if (params->enable_db)
+    {
+      wrap_db_init ();
+    }
+
+  if (params->enable_session_ticket_server)
+    gnutls_session_ticket_key_generate (&session_ticket_key);
+
+  for (t = 0; t < 2; t++)
+    {
+      client_len = sizeof (sa_cli);
+
+      session = initialize_tls_session (params);
+
+      sd = accept (listen_sd, (SA *) & sa_cli, &client_len);
+
+      if (debug)
+        success ("server: connection from %s, port %d\n",
+                 inet_ntop (AF_INET, &sa_cli.sin_addr, topbuf,
+                            sizeof (topbuf)), ntohs (sa_cli.sin_port));
+
+      gnutls_transport_set_ptr (session, (gnutls_transport_ptr_t) sd);
+      ret = gnutls_handshake (session);
+      if (ret < 0)
+        {
+          close (sd);
+          gnutls_deinit (session);
+          fail ("server: Handshake has failed (%s)\n\n",
+                gnutls_strerror (ret));
+          return;
+        }
+      if (debug)
+        success ("server: Handshake was completed\n");
+
+      /* see the Getting peer's information example */
+      /* print_info(session); */
+
+      i = 0;
+      for (;;)
+        {
+          memset (buffer, 0, MAX_BUF + 1);
+          ret = gnutls_record_recv (session, buffer, MAX_BUF);
+
+          if (ret == 0)
+            {
+              if (debug)
+                success ("server: Peer has closed the GnuTLS connection\n");
+              break;
+            }
+          else if (ret < 0)
+            {
+              fail ("server: Received corrupted data(%d). Closing...\n", ret);
+              break;
+            }
+          else if (ret > 0)
+            {
+              /* echo data back to the client
+               */
+              gnutls_record_send (session, buffer, strlen (buffer));
+            }
+        }
+      /* do not wait for the peer to close the connection.
+       */
+      gnutls_bye (session, GNUTLS_SHUT_WR);
+
+      close (sd);
+
+      gnutls_deinit (session);
+    }
+
+  close (listen_sd);
+
+  if (params->enable_db)
+    {
+      wrap_db_deinit ();
+    }
+
+  gnutls_free (session_ticket_key.data);
+  session_ticket_key.data = NULL;
+
+  if (debug)
+    success ("server: finished\n");
+}
+
+void
+doit (void)
+{
+  int i;
+
+  for (i = 0; resume_tests[i].desc; i++)
+    {
+      printf ("%s\n", resume_tests[i].desc);
+
+      global_start ();
+      if (error_count)
+        return;
+
+      child = fork ();
+      if (child < 0)
+        {
+          perror ("fork");
+          fail ("fork");
+          return;
+        }
+
+      if (child)
+        {
+          int status;
+          /* parent */
+          server (&resume_tests[i]);
+          wait (&status);
+          if (WEXITSTATUS(status) > 0)
+            error_count++;
+          global_stop ();
+        }
+      else
+        {
+          client (&resume_tests[i]);
+          gnutls_global_deinit ();
+          if (error_count)
+            exit(1);
+          exit (0);
+        }
+    }
+}
+
+/* Functions and other stuff needed for session resuming.
+ * This is done using a very simple list which holds session ids
+ * and session data.
+ */
+
+#define MAX_SESSION_ID_SIZE 32
+#define MAX_SESSION_DATA_SIZE 1024
+
+typedef struct
+{
+  unsigned char session_id[MAX_SESSION_ID_SIZE];
+  unsigned int session_id_size;
+
+  char session_data[MAX_SESSION_DATA_SIZE];
+  int session_data_size;
+} CACHE;
+
+static CACHE *cache_db;
+static int cache_db_ptr = 0;
+
+static void
+wrap_db_init (void)
+{
+
+  /* allocate cache_db */
+  cache_db = calloc (1, TLS_SESSION_CACHE * sizeof (CACHE));
+}
+
+static void
+wrap_db_deinit (void)
+{
+  free (cache_db);
+  cache_db = NULL;
+  return;
+}
+
+static int
+wrap_db_store (void *dbf, gnutls_datum_t key, gnutls_datum_t data)
+{
+  if (debug)
+    success ("resume db storing... (%d-%d)\n", key.size, data.size);
+
+  if (debug)
+    {
+      unsigned int i;
+      printf ("key:\n");
+      for (i = 0; i < key.size; i++)
+        {
+          printf ("%02x ", key.data[i] & 0xFF);
+          if ((i + 1) % 16 == 0)
+            printf ("\n");
+        }
+      printf ("\n");
+      printf ("data:\n");
+      for (i = 0; i < data.size; i++)
+        {
+          printf ("%02x ", data.data[i] & 0xFF);
+          if ((i + 1) % 16 == 0)
+            printf ("\n");
+        }
+      printf ("\n");
+    }
+
+  if (cache_db == NULL)
+    return -1;
+
+  if (key.size > MAX_SESSION_ID_SIZE)
+    return -1;
+
+  if (data.size > MAX_SESSION_DATA_SIZE)
+    return -1;
+
+  memcpy (cache_db[cache_db_ptr].session_id, key.data, key.size);
+  cache_db[cache_db_ptr].session_id_size = key.size;
+
+  memcpy (cache_db[cache_db_ptr].session_data, data.data, data.size);
+  cache_db[cache_db_ptr].session_data_size = data.size;
+
+  cache_db_ptr++;
+  cache_db_ptr %= TLS_SESSION_CACHE;
+
+  return 0;
+}
+
+static gnutls_datum_t
+wrap_db_fetch (void *dbf, gnutls_datum_t key)
+{
+  gnutls_datum_t res = { NULL, 0 };
+  int i;
+
+  if (debug)
+    success ("resume db fetch... (%d)\n", key.size);
+  if (debug)
+    {
+      unsigned int i;
+      printf ("key:\n");
+      for (i = 0; i < key.size; i++)
+        {
+          printf ("%02x ", key.data[i] & 0xFF);
+          if ((i + 1) % 16 == 0)
+            printf ("\n");
+        }
+      printf ("\n");
+    }
+
+  if (cache_db == NULL)
+    return res;
+
+  for (i = 0; i < TLS_SESSION_CACHE; i++)
+    {
+      if (key.size == cache_db[i].session_id_size &&
+          memcmp (key.data, cache_db[i].session_id, key.size) == 0)
+        {
+          if (debug)
+            success ("resume db fetch... return info\n");
+
+          res.size = cache_db[i].session_data_size;
+
+          res.data = gnutls_malloc (res.size);
+          if (res.data == NULL)
+            return res;
+
+          memcpy (res.data, cache_db[i].session_data, res.size);
+
+          if (debug)
+            {
+              unsigned int i;
+              printf ("data:\n");
+              for (i = 0; i < res.size; i++)
+                {
+                  printf ("%02x ", res.data[i] & 0xFF);
+                  if ((i + 1) % 16 == 0)
+                    printf ("\n");
+                }
+              printf ("\n");
+            }
+
+          return res;
+        }
+    }
+
+  if (debug)
+    success ("resume db fetch... NOT FOUND\n");
+  return res;
+}
+
+static int
+wrap_db_delete (void *dbf, gnutls_datum_t key)
+{
+  int i;
+
+  if (cache_db == NULL)
+    return -1;
+
+  for (i = 0; i < TLS_SESSION_CACHE; i++)
+    {
+      if (key.size == cache_db[i].session_id_size &&
+          memcmp (key.data, cache_db[i].session_id, key.size) == 0)
+        {
+
+          cache_db[i].session_id_size = 0;
+          cache_db[i].session_data_size = 0;
+
+          return 0;
+        }
+    }
+
+  return -1;
+
+}