src/transport/test_transport_api_manipulation_recv_tcp.c

   1 /*
   2      This file is part of GNUnet.
   3      Copyright (C) 2009, 2010, 2016 GNUnet e.V.
   4
   5      GNUnet is free software: you can redistribute it and/or modify it
   6      under the terms of the GNU Affero General Public License as published
   7      by the Free Software Foundation, either version 3 of the License,
   8      or (at your option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      Affero General Public License for more details.
  14 */
  15 /**
  16  * @file transport/test_transport_api_manipulation_recv_tcp.c
  17  * @brief base test case for transport traffic manipulation implementation
  18  *
  19  * This test case will setup 2 peers and connect them, the first message
  20  * will be sent without manipulation, then a receive delay of 1 second will
  21  * be configured and 2 more message will be sent. Time will be measured
  22  *
  23  * In addition the distance on receiver side will be manipulated to be 10
  24  */
  25 #include "platform.h"
  26 #include "gnunet_transport_service.h"
  27 #include "transport-testing.h"
  28
  29 /**
  30  * How long until we give up on transmitting the message?
  31  */
  32 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 120)
  33
  34
  35 static struct GNUNET_TRANSPORT_TESTING_ConnectCheckContext *ccc;
  36
  37 static int messages_recv;
  38
  39 static struct GNUNET_TIME_Absolute start_normal;
  40
  41 static struct GNUNET_TIME_Relative dur_normal;
  42
  43 static struct GNUNET_TIME_Absolute start_delayed;
  44
  45 static struct GNUNET_TIME_Relative dur_delayed;
  46
  47
  48 static void
  49 do_free (void *cls)
  50 {
  51   struct GNUNET_TRANSPORT_TESTING_SendClosure *sc = cls;
  52
  53   GNUNET_free (sc);
  54 }
  55
  56
  57 static void
  58 delayed_transmit (void *cls)
  59 {
  60   struct GNUNET_TRANSPORT_TESTING_SendClosure *sc = cls;
  61
  62   start_delayed = GNUNET_TIME_absolute_get ();
  63   GNUNET_TRANSPORT_TESTING_large_send (sc);
  64 }
  65
  66
  67 static void
  68 sendtask (void *cls)
  69 {
  70   struct GNUNET_TRANSPORT_TESTING_SendClosure *sc;
  71   struct GNUNET_ATS_Properties prop;
  72   struct GNUNET_TIME_Relative delay;
  73
  74   sc = GNUNET_new (struct GNUNET_TRANSPORT_TESTING_SendClosure);
  75   sc->num_messages = 1;
  76   sc->ccc = ccc;
  77   sc->cont = &do_free;
  78   sc->cont_cls = sc;
  79   if (0 == messages_recv)
  80   {
  81     start_normal = GNUNET_TIME_absolute_get();
  82   }
  83   if (0 < messages_recv)
  84   {
  85     memset (&prop,
  86             0,
  87             sizeof (prop));
  88     delay = GNUNET_TIME_UNIT_SECONDS;
  89     GNUNET_TRANSPORT_manipulation_set (ccc->p[1]->tmh,
  90                                        &ccc->p[0]->id,
  91                                        &prop,
  92                                        delay,
  93                                        GNUNET_TIME_UNIT_ZERO);
  94     /* wait 1s to allow manipulation to go into effect */
  95     if (1 == messages_recv)
  96     {
  97       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS,
  98                                     &delayed_transmit,
  99                                     sc);
 100       return;
 101     }
 102   }
 103   GNUNET_TRANSPORT_TESTING_large_send (sc);
 104 }
 105
 106
 107 static void
 108 notify_receive (void *cls,
 109                 struct GNUNET_TRANSPORT_TESTING_PeerContext *receiver,
 110                 const struct GNUNET_PeerIdentity *sender,
 111                 const struct GNUNET_TRANSPORT_TESTING_TestMessage *message)
 112 {
 113   {
 114     char *ps = GNUNET_strdup (GNUNET_i2s (&receiver->id));
 115
 116     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 117                 "Peer %u (`%s') received message of type %d and size %u size from peer %s)!\n",
 118                 receiver->no,
 119                 ps,
 120                 ntohs (message->header.type),
 121                 ntohs (message->header.size),
 122                 GNUNET_i2s (sender));
 123     GNUNET_free (ps);
 124   }
 125   if ( (GNUNET_TRANSPORT_TESTING_SIMPLE_MTYPE != ntohs (message->header.type)) ||
 126        (GNUNET_TRANSPORT_TESTING_LARGE_MESSAGE_SIZE != ntohs (message->header.size)) )
 127   {
 128     GNUNET_break (0);
 129     ccc->global_ret = GNUNET_SYSERR;
 130     GNUNET_SCHEDULER_shutdown ();
 131     return;
 132   }
 133
 134   if (messages_recv <= 2)
 135   {
 136     /* Received non-delayed message */
 137     dur_normal = GNUNET_TIME_absolute_get_duration (start_normal);
 138     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 139                 "Received non-delayed message %u after %s\n",
 140                 messages_recv,
 141                 GNUNET_STRINGS_relative_time_to_string (dur_normal,
 142                                                         GNUNET_YES));
 143     GNUNET_SCHEDULER_add_now (&sendtask,
 144                               NULL);
 145     messages_recv++;
 146     return;
 147   }
 148   /* Received manipulated message */
 149   dur_delayed = GNUNET_TIME_absolute_get_duration(start_delayed);
 150   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 151               "Received delayed message %u after %s\n",
 152               messages_recv,
 153               GNUNET_STRINGS_relative_time_to_string (dur_delayed,
 154                                                       GNUNET_YES));
 155   if (dur_delayed.rel_value_us < GNUNET_TIME_UNIT_SECONDS.rel_value_us)
 156   {
 157     GNUNET_break (0);
 158     ccc->global_ret = GNUNET_SYSERR;
 159     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 160                 "Delayed message was not delayed correctly: took only %s\n",
 161                 GNUNET_STRINGS_relative_time_to_string (dur_delayed,
 162                                                         GNUNET_YES));
 163   }
 164   else
 165   {
 166     ccc->global_ret = GNUNET_OK;
 167   }
 168   /* shutdown */
 169   GNUNET_SCHEDULER_shutdown ();
 170 }
 171
 172
 173 int
 174 main (int argc,
 175       char *argv[])
 176 {
 177   struct GNUNET_TRANSPORT_TESTING_ConnectCheckContext my_ccc = {
 178     .connect_continuation = &sendtask,
 179     .config_file = "test_transport_api_data.conf",
 180     .rec = &notify_receive,
 181     .nc = &GNUNET_TRANSPORT_TESTING_log_connect,
 182     .nd = &GNUNET_TRANSPORT_TESTING_log_disconnect,
 183     .timeout = TIMEOUT,
 184     .global_ret = GNUNET_NO
 185   };
 186
 187   ccc = &my_ccc;
 188   if (GNUNET_OK !=
 189       GNUNET_TRANSPORT_TESTING_main (2,
 190                                      &GNUNET_TRANSPORT_TESTING_connect_check,
 191                                      ccc))
 192     return 1;
 193   return 0;
 194 }
 195
 196 /* end of test_transport_api_manipulation_recv_tcp.c */