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 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 static ssize_t
  52 client_push_300(gnutls_transport_ptr_t tr, const void *data, size_t len)
  53 {
  54         size_t newlen;
  55
  56         if (len > 300) {
  57                 gnutls_transport_set_errno((gnutls_session_t) tr,
  58                                            EMSGSIZE);
  59                 return -1;
  60         }
  61
  62         len = min(len, sizeof(to_server) - to_server_len);
  63
  64         newlen = to_server_len + len;
  65         memcpy(to_server + to_server_len, data, len);
  66         to_server_len = newlen;
  67 #ifdef EAGAIN_DEBUG
  68         fprintf(stderr, "eagain: pushed %d bytes to server (avail: %d)\n",
  69                 (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                 gnutls_transport_set_errno((gnutls_session_t) tr,
  81                                            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",
  92                 (int) len, (int) to_client_len);
  93 #endif
  94
  95         return len;
  96 }
  97
  98 void 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         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,
 126                     GNUTLS_SERVER | GNUTLS_DATAGRAM | GNUTLS_NONBLOCK);
 127         ret =
 128             gnutls_priority_set_direct(server,
 129                                        "NONE:+VERS-DTLS1.2:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-DH",
 130                                        NULL);
 131         if (ret < 0)
 132                 exit(1);
 133         gnutls_credentials_set(server, GNUTLS_CRD_ANON, s_anoncred);
 134         gnutls_transport_set_push_function(server, server_push_300);
 135         gnutls_transport_set_pull_function(server, server_pull);
 136         gnutls_transport_set_pull_timeout_function(server,
 137                                                    server_pull_timeout_func);
 138         gnutls_transport_set_ptr(server, server);
 139
 140         /* Init client */
 141         gnutls_anon_allocate_client_credentials(&c_anoncred);
 142         gnutls_init(&client,
 143                     GNUTLS_CLIENT | GNUTLS_DATAGRAM | GNUTLS_NONBLOCK);
 144         cret =
 145             gnutls_priority_set_direct(client,
 146                                        "NONE:+VERS-DTLS1.2:+CIPHER-ALL:+MAC-ALL:+SIGN-ALL:+COMP-ALL:+ANON-DH",
 147                                        NULL);
 148         if (cret < 0)
 149                 exit(1);
 150         gnutls_credentials_set(client, GNUTLS_CRD_ANON, c_anoncred);
 151         gnutls_transport_set_push_function(client, client_push_300);
 152         gnutls_transport_set_pull_function(client, client_pull);
 153         gnutls_transport_set_pull_timeout_function(client,
 154                                                    client_pull_timeout_func);
 155         gnutls_transport_set_ptr(client, client);
 156
 157         handshake = 1;
 158         HANDSHAKE_DTLS(client, server);
 159
 160         if (gnutls_protocol_get_version(client) != GNUTLS_DTLS1_2) {
 161                 fail("Error in negotiated version\n");
 162                 exit(1);
 163         }
 164
 165         handshake = 0;
 166         if (debug)
 167                 success("Handshake established\n");
 168
 169         do {
 170                 ret = gnutls_record_send(client, MSG, strlen(MSG));
 171         }
 172         while (ret == GNUTLS_E_AGAIN);
 173         //success ("client: sent %d\n", ns);
 174
 175         msglen = strlen(MSG);
 176         TRANSFER(client, server, MSG, msglen, buffer, MAX_BUF);
 177
 178         if (debug)
 179                 fputs("\n", stdout);
 180
 181         gnutls_bye(client, GNUTLS_SHUT_WR);
 182         gnutls_bye(server, GNUTLS_SHUT_WR);
 183
 184         gnutls_deinit(client);
 185         gnutls_deinit(server);
 186
 187         gnutls_anon_free_client_credentials(c_anoncred);
 188         gnutls_anon_free_server_credentials(s_anoncred);
 189
 190         gnutls_dh_params_deinit(dh_params);
 191
 192         gnutls_global_deinit();
 193 }