tests/mini-emsgsize-dtls.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/dtls.h>
  33 #include <gnutls/crypto.h>
  34 #include "utils.h"
  35 #define RANDOMIZE
  36 #define IGNORE_PUSH
  37 #include "eagain-common.h"
  38
  39 const char* side = "";
  40
  41 static void
  42 tls_log_func (int level, const char *str)
  43 {
  44   fprintf (stderr, "%s|<%d>| %s", side, level, str);
  45 }
  46
  47 static int handshake = 0;
  48
  49 #define MAX_BUF 1024
  50 #define MSG "Hello TLS, and hi and how are you and more data here... and more... and even more and even more more data..."
  51
  52 static ssize_t
  53 client_push_300 (gnutls_transport_ptr_t tr, const void *data, size_t len)
  54 {
  55   size_t newlen;
  56
  57   if (len > 300)
  58     {
  59       gnutls_transport_set_errno ((gnutls_session_t)tr, EMSGSIZE);
  60       return -1;
  61     }
  62
  63   len = min(len, sizeof(to_server)-to_server_len);
  64
  65   newlen = to_server_len + len;
  66   memcpy (to_server + to_server_len, data, len);
  67   to_server_len = newlen;
  68 #ifdef EAGAIN_DEBUG
  69   fprintf(stderr, "eagain: pushed %d bytes to server (avail: %d)\n", (int)len, (int)to_server_len);
  70 #endif
  71   return len;
  72 }
  73
  74 static ssize_t
  75 server_push_300 (gnutls_transport_ptr_t tr, const void *data, size_t len)
  76 {
  77   size_t newlen;
  78
  79   if (len > 300)
  80     {
  81       gnutls_transport_set_errno ((gnutls_session_t)tr, EMSGSIZE);
  82       return -1;
  83     }
  84
  85   len = min(len, sizeof(to_client)-to_client_len);
  86
  87   newlen = to_client_len + len;
  88   memcpy (to_client + to_client_len, data, len);
  89   to_client_len = newlen;
  90 #ifdef EAGAIN_DEBUG
  91   fprintf(stderr, "eagain: pushed %d bytes to client (avail: %d)\n", (int)len, (int)to_client_len);
  92 #endif
  93
  94   return len;
  95 }
  96
  97 void
  98 doit (void)
  99 {
 100   /* Server stuff. */
 101   gnutls_anon_server_credentials_t s_anoncred;
 102   const gnutls_datum_t p3 = { (void *) pkcs3, strlen (pkcs3) };
 103   static gnutls_dh_params_t dh_params;
 104   gnutls_session_t server;
 105   int sret, cret;
 106   /* Client stuff. */
 107   gnutls_anon_client_credentials_t c_anoncred;
 108   gnutls_session_t client;
 109   /* Need to enable anonymous KX specifically. */
 110   char buffer[MAX_BUF + 1];
 111   ssize_t ns;
 112   int ret, transferred = 0, msglen;
 113
 114   /* General init. */
 115   gnutls_global_init ();
 116   gnutls_global_set_log_function (tls_log_func);
 117   if (debug)
 118     gnutls_global_set_log_level (99);
 119
 120   /* Init server */
 121   gnutls_anon_allocate_server_credentials (&s_anoncred);
 122   gnutls_dh_params_init (&dh_params);
 123   gnutls_dh_params_import_pkcs3 (dh_params, &p3, GNUTLS_X509_FMT_PEM);
 124   gnutls_anon_set_server_dh_params (s_anoncred, dh_params);
 125   gnutls_init (&server, GNUTLS_SERVER|GNUTLS_DATAGRAM|GNUTLS_NONBLOCK);
 126   ret = gnutls_priority_set_direct (server, "NONE:+VERS-DTLS1.0:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-DH", NULL);
 127   if (ret < 0)
 128     exit(1);
 129   gnutls_credentials_set (server, GNUTLS_CRD_ANON, s_anoncred);
 130   gnutls_dh_set_prime_bits (server, 1024);
 131   gnutls_transport_set_push_function (server, server_push_300);
 132   gnutls_transport_set_pull_function (server, server_pull);
 133   gnutls_transport_set_pull_timeout_function (server, server_pull_timeout_func);
 134   gnutls_transport_set_ptr (server, (gnutls_transport_ptr_t)server);
 135
 136   /* Init client */
 137   gnutls_anon_allocate_client_credentials (&c_anoncred);
 138   gnutls_init (&client, GNUTLS_CLIENT|GNUTLS_DATAGRAM|GNUTLS_NONBLOCK);
 139   cret = gnutls_priority_set_direct (client, "NONE:+VERS-DTLS1.0:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-DH", NULL);
 140   if (cret < 0)
 141     exit(1);
 142   gnutls_credentials_set (client, GNUTLS_CRD_ANON, c_anoncred);
 143   gnutls_transport_set_push_function (client, client_push_300);
 144   gnutls_transport_set_pull_function (client, client_pull);
 145   gnutls_transport_set_pull_timeout_function (client, client_pull_timeout_func);
 146   gnutls_transport_set_ptr (client, (gnutls_transport_ptr_t)client);
 147
 148   handshake = 1;
 149   HANDSHAKE_DTLS(client, server);
 150
 151   handshake = 0;
 152   if (debug)
 153     success ("Handshake established\n");
 154
 155   do
 156     {
 157       ret = gnutls_record_send (client, MSG, strlen (MSG));
 158     }
 159   while(ret == GNUTLS_E_AGAIN);
 160   //success ("client: sent %d\n", ns);
 161
 162   msglen = strlen(MSG);
 163   TRANSFER(client, server, MSG, msglen, buffer, MAX_BUF);
 164
 165   if (debug)
 166     fputs ("\n", stdout);
 167
 168   gnutls_bye (client, GNUTLS_SHUT_WR);
 169   gnutls_bye (server, GNUTLS_SHUT_WR);
 170
 171   gnutls_deinit (client);
 172   gnutls_deinit (server);
 173
 174   gnutls_anon_free_client_credentials (c_anoncred);
 175   gnutls_anon_free_server_credentials (s_anoncred);
 176
 177   gnutls_dh_params_deinit (dh_params);
 178
 179   gnutls_global_deinit ();
 180 }