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 tls_log_func(int level, const char *str)
  41 {
  42         fprintf(stderr, "%s|<%d>| %s", side, level, str);
  43 }
  44
  45 static int handshake = 0;
  46
  47 #define MAX_BUF 1024
  48 #define MSG "Hello TLS, and hi and how are you and more data here... and more... and even more and even more more data..."
  49
  50 void doit(void)
  51 {
  52         /* Server stuff. */
  53         gnutls_anon_server_credentials_t s_anoncred;
  54         const gnutls_datum_t p3 =
  55             { (unsigned char *) pkcs3, strlen(pkcs3) };
  56         static gnutls_dh_params_t dh_params;
  57         gnutls_session_t server;
  58         int sret, cret;
  59         /* Client stuff. */
  60         gnutls_anon_client_credentials_t c_anoncred;
  61         gnutls_session_t client;
  62         /* Need to enable anonymous KX specifically. */
  63         char buffer[MAX_BUF + 1];
  64         ssize_t ns;
  65         int ret, transferred = 0, msglen;
  66
  67         /* General init. */
  68         global_init();
  69         gnutls_global_set_log_function(tls_log_func);
  70         if (debug)
  71                 gnutls_global_set_log_level(2);
  72
  73         /* Init server */
  74         gnutls_anon_allocate_server_credentials(&s_anoncred);
  75         gnutls_dh_params_init(&dh_params);
  76         gnutls_dh_params_import_pkcs3(dh_params, &p3, GNUTLS_X509_FMT_PEM);
  77         gnutls_anon_set_server_dh_params(s_anoncred, dh_params);
  78         gnutls_init(&server, GNUTLS_SERVER);
  79         ret =
  80             gnutls_priority_set_direct(server,
  81                                        "NONE:+VERS-TLS-ALL:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-DH",
  82                                        NULL);
  83         if (ret < 0)
  84                 exit(1);
  85         gnutls_credentials_set(server, GNUTLS_CRD_ANON, s_anoncred);
  86         gnutls_transport_set_push_function(server, server_push);
  87         gnutls_transport_set_pull_function(server, server_pull);
  88         gnutls_transport_set_ptr(server, server);
  89
  90         /* Init client */
  91         gnutls_anon_allocate_client_credentials(&c_anoncred);
  92         gnutls_init(&client, GNUTLS_CLIENT);
  93         ret =
  94             gnutls_priority_set_direct(client,
  95                                        "NONE:+VERS-TLS-ALL:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-DH",
  96                                        NULL);
  97         if (ret < 0)
  98                 exit(1);
  99         gnutls_credentials_set(client, GNUTLS_CRD_ANON, c_anoncred);
 100         gnutls_transport_set_push_function(client, client_push);
 101         gnutls_transport_set_pull_function(client, client_pull);
 102         gnutls_transport_set_ptr(client, client);
 103
 104         handshake = 1;
 105         HANDSHAKE(client, server);
 106
 107         handshake = 0;
 108         if (debug)
 109                 success("Handshake established\n");
 110
 111         msglen = strlen(MSG);
 112         TRANSFER(client, server, MSG, msglen, buffer, MAX_BUF);
 113         if (debug)
 114                 fputs("\n", stdout);
 115
 116         gnutls_bye(client, GNUTLS_SHUT_WR);
 117         gnutls_bye(server, GNUTLS_SHUT_WR);
 118
 119         gnutls_deinit(client);
 120         gnutls_deinit(server);
 121
 122         gnutls_anon_free_client_credentials(c_anoncred);
 123         gnutls_anon_free_server_credentials(s_anoncred);
 124
 125         gnutls_dh_params_deinit(dh_params);
 126
 127         gnutls_global_deinit();
 128 }