tests/mini-eagain.c

   1 /*
   2  * Copyright (C) 2008-2012 Free Software Foundation, Inc.
   3  *
   4  * Author: Simon Josefsson, Nikos Mavrogiannopoulos
   5  *
   6  * This file is part of GnuTLS.
   7  *
   8  * GnuTLS is free software; you can redistribute it and/or modify it
   9  * under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 3 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * GnuTLS is distributed in the hope that it will be useful, but
  14  * WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16  * General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with GnuTLS; if not, write to the Free Software Foundation,
  20  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
  21  */
  22
  23 #ifdef HAVE_CONFIG_H
  24 #include <config.h>
  25 #endif
  26
  27 #include <stdio.h>
  28 #include <stdlib.h>
  29 #include <string.h>
  30 #include <errno.h>
  31 #include <gnutls/gnutls.h>
  32 #include <gnutls/crypto.h>
  33
  34 #include "utils.h"
  35 #define RANDOMIZE
  36 #include "eagain-common.h"
  37
  38 const char* side = "";
  39
  40 static void
  41 tls_log_func (int level, const char *str)
  42 {
  43   fprintf (stderr, "%s|<%d>| %s", side, level, str);
  44 }
  45
  46 static int handshake = 0;
  47
  48 #define MAX_BUF 1024
  49 #define MSG "Hello TLS, and hi and how are you and more data here... and more... and even more and even more more data..."
  50
  51 void
  52 doit (void)
  53 {
  54   /* Server stuff. */
  55   gnutls_anon_server_credentials_t s_anoncred;
  56   const gnutls_datum_t p3 = { (unsigned char *) pkcs3, strlen (pkcs3) };
  57   static gnutls_dh_params_t dh_params;
  58   gnutls_session_t server;
  59   int sret, cret;
  60   /* Client stuff. */
  61   gnutls_anon_client_credentials_t c_anoncred;
  62   gnutls_session_t client;
  63   /* Need to enable anonymous KX specifically. */
  64   char buffer[MAX_BUF + 1];
  65   ssize_t ns;
  66   int ret, transferred = 0, msglen;
  67
  68   /* General init. */
  69   gnutls_global_init ();
  70   gnutls_global_set_log_function (tls_log_func);
  71   if (debug)
  72     gnutls_global_set_log_level (2);
  73
  74   /* Init server */
  75   gnutls_anon_allocate_server_credentials (&s_anoncred);
  76   gnutls_dh_params_init (&dh_params);
  77   gnutls_dh_params_import_pkcs3 (dh_params, &p3, GNUTLS_X509_FMT_PEM);
  78   gnutls_anon_set_server_dh_params (s_anoncred, dh_params);
  79   gnutls_init (&server, GNUTLS_SERVER);
  80   ret = gnutls_priority_set_direct (server, "NONE:+VERS-TLS-ALL:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-DH", NULL);
  81   if (ret < 0)
  82     exit(1);
  83   gnutls_credentials_set (server, GNUTLS_CRD_ANON, s_anoncred);
  84   gnutls_dh_set_prime_bits (server, 1024);
  85   gnutls_transport_set_push_function (server, server_push);
  86   gnutls_transport_set_pull_function (server, server_pull);
  87   gnutls_transport_set_ptr (server, (gnutls_transport_ptr_t)server);
  88
  89   /* Init client */
  90   gnutls_anon_allocate_client_credentials (&c_anoncred);
  91   gnutls_init (&client, GNUTLS_CLIENT);
  92   ret = gnutls_priority_set_direct (client, "NONE:+VERS-TLS-ALL:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-DH", NULL);
  93   if (ret < 0)
  94     exit(1);
  95   gnutls_credentials_set (client, GNUTLS_CRD_ANON, c_anoncred);
  96   gnutls_transport_set_push_function (client, client_push);
  97   gnutls_transport_set_pull_function (client, client_pull);
  98   gnutls_transport_set_ptr (client, (gnutls_transport_ptr_t)client);
  99
 100   handshake = 1;
 101   HANDSHAKE(client, server);
 102
 103   handshake = 0;
 104   if (debug)
 105     success ("Handshake established\n");
 106
 107   msglen = strlen(MSG);
 108   TRANSFER(client, server, MSG, msglen, buffer, MAX_BUF);
 109   if (debug)
 110     fputs ("\n", stdout);
 111
 112   gnutls_bye (client, GNUTLS_SHUT_WR);
 113   gnutls_bye (server, GNUTLS_SHUT_WR);
 114
 115   gnutls_deinit (client);
 116   gnutls_deinit (server);
 117
 118   gnutls_anon_free_client_credentials (c_anoncred);
 119   gnutls_anon_free_server_credentials (s_anoncred);
 120
 121   gnutls_dh_params_deinit (dh_params);
 122
 123   gnutls_global_deinit ();
 124 }