migrate more tests to new transmission API

1 files changed, 61 insertions, 155 deletions

diff --git a/src/transport/test_transport_api_manipulation_recv_tcp.c b/src/transport/test_transport_api_manipulation_recv_tcp.c
index 023d5a714..e902aed62 100644
--- a/src/transport/test_transport_api_manipulation_recv_tcp.c
+++ b/src/transport/test_transport_api_manipulation_recv_tcp.c
@@ -36,22 +36,9 @@
  */
 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 120)
 
-/**
- * How long until we give up on transmitting the message?
- */
-#define TIMEOUT_TRANSMIT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30)
-
-#define TEST_MESSAGE_SIZE 2600
-
-#define TEST_MESSAGE_TYPE 12345
-
 
 static struct GNUNET_TRANSPORT_TESTING_ConnectCheckContext *ccc;
 
-static struct GNUNET_TRANSPORT_TransmitHandle *th;
-
-static struct GNUNET_SCHEDULER_Task *send_task;
-
 static int messages_recv;
 
 static struct GNUNET_TIME_Absolute start_normal;
@@ -64,26 +51,46 @@ static struct GNUNET_TIME_Relative dur_delayed;
 
 
 static void
-custom_shutdown (void *cls)
+do_free (void *cls)
 {
-  if (NULL != th)
+  struct GNUNET_TRANSPORT_TESTING_SendClosure *sc = cls;
+  
+  GNUNET_free (sc);
+}
+
+
+static void
+sendtask (void *cls)
+{
+  struct GNUNET_TRANSPORT_TESTING_SendClosure *sc;
+  struct GNUNET_ATS_Properties prop;
+  struct GNUNET_TIME_Relative delay;
+
+  sc = GNUNET_new (struct GNUNET_TRANSPORT_TESTING_SendClosure);
+  sc->num_messages = 1;
+  sc->ccc = ccc;
+  sc->cont = &do_free;
+  sc->cont_cls = sc;
+  if (0 == messages_recv)
   {
-    GNUNET_TRANSPORT_notify_transmit_ready_cancel (th);
-    th = NULL;
+    start_normal = GNUNET_TIME_absolute_get();
   }
-  if (NULL != send_task)
+  if (0 < messages_recv)
   {
-    GNUNET_SCHEDULER_cancel (send_task);
-    send_task = NULL;
+    memset (&prop, 0, sizeof (prop));
+    delay = GNUNET_TIME_UNIT_SECONDS;
+    GNUNET_TRANSPORT_set_traffic_metric (ccc->p[1]->th,
+                                         &ccc->p[0]->id,
+                                         &prop,
+                                         delay,
+                                         GNUNET_TIME_UNIT_ZERO);
+    start_delayed = GNUNET_TIME_absolute_get();
   }
+  GNUNET_TRANSPORT_TESTING_large_send (sc);
 }
 
 
 static void
-sendtask (void *cls);
-
-
-static void
 notify_receive (void *cls,
                 struct GNUNET_TRANSPORT_TESTING_PeerContext *receiver,
                 const struct GNUNET_PeerIdentity *sender,
@@ -101,13 +108,8 @@ notify_receive (void *cls,
                 GNUNET_i2s (sender));
     GNUNET_free (ps);
   }
-  if ((TEST_MESSAGE_TYPE == ntohs (message->type)) &&
-      (TEST_MESSAGE_SIZE == ntohs (message->size)))
-  {
-    if (GNUNET_NO == ccc->global_ret)
-      ccc->global_ret = GNUNET_OK;
-  }
-  else
+  if ( (GNUNET_TRANSPORT_TESTING_SIMPLE_MTYPE != ntohs (message->type)) ||
+       (GNUNET_TRANSPORT_TESTING_LARGE_MESSAGE_SIZE != ntohs (message->size)) )
   {
     GNUNET_break (0);
     ccc->global_ret = GNUNET_SYSERR;
@@ -118,150 +120,54 @@ notify_receive (void *cls,
   if (messages_recv <= 1)
   {
     /* Received non-delayed message */
-    dur_normal = GNUNET_TIME_absolute_get_duration(start_normal);
+    dur_normal = GNUNET_TIME_absolute_get_duration (start_normal);
     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
                 "Received non-delayed message %u after %s\n",
                 messages_recv,
                 GNUNET_STRINGS_relative_time_to_string (dur_normal,
                                                         GNUNET_YES));
-
-    struct GNUNET_ATS_Properties prop;
-    struct GNUNET_TIME_Relative delay;
-
-    delay.rel_value_us = 1000 * 1000LL;
-    memset (&prop, 0, sizeof (prop));
-    prop.distance = 10;
-    GNUNET_TRANSPORT_set_traffic_metric (ccc->p[0]->th,
-                                         &ccc->p[1]->id,
-                                         &prop,
-                                         delay,
-                                         GNUNET_TIME_UNIT_ZERO);
-    send_task = GNUNET_SCHEDULER_add_now (&sendtask, NULL);
-  }
-  if (2 == messages_recv)
-  {
-        /* Received manipulated message */
-    dur_delayed = GNUNET_TIME_absolute_get_duration(start_delayed);
-    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-                "Received delayed message %u after %s\n",
-                messages_recv,
-                GNUNET_STRINGS_relative_time_to_string (dur_delayed,
-                                                          GNUNET_YES));
-    if (dur_delayed.rel_value_us < GNUNET_TIME_UNIT_SECONDS.rel_value_us)
-    {
-      GNUNET_break (0);
-      ccc->global_ret = GNUNET_SYSERR;
-      GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
-                  "Delayed message was not delayed correctly: took only %s\n",
-                  GNUNET_STRINGS_relative_time_to_string (dur_delayed,
-                                                          GNUNET_YES));
-    }
-    /* shutdown */
-    GNUNET_SCHEDULER_shutdown ();
+    GNUNET_SCHEDULER_add_now (&sendtask,
+                              NULL);
+    messages_recv++;
+    return;
   }
-  messages_recv++;
-}
-
-
-static size_t
-notify_ready (void *cls, size_t size, void *buf)
-{
-  struct GNUNET_TRANSPORT_TESTING_PeerContext *p = cls;
-  struct GNUNET_MessageHeader *hdr;
-
-  th = NULL;
-  if (NULL == buf)
+  /* Received manipulated message */
+  dur_delayed = GNUNET_TIME_absolute_get_duration(start_delayed);
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "Received delayed message %u after %s\n",
+              messages_recv,
+              GNUNET_STRINGS_relative_time_to_string (dur_delayed,
+                                                      GNUNET_YES));
+  if (dur_delayed.rel_value_us < GNUNET_TIME_UNIT_SECONDS.rel_value_us)
   {
+    GNUNET_break (0);
+    ccc->global_ret = GNUNET_SYSERR;
     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
-                "Timeout occurred while waiting for transmit_ready\n");
-    GNUNET_SCHEDULER_shutdown ();
-    ccc->global_ret = 42;
-    return 0;
-  }
-
-  GNUNET_assert (size >= TEST_MESSAGE_SIZE);
-  memset (buf, '\0', TEST_MESSAGE_SIZE);
-  hdr = buf;
-  hdr->size = htons (TEST_MESSAGE_SIZE);
-  hdr->type = htons (TEST_MESSAGE_TYPE);
-
-  {
-    char *ps = GNUNET_strdup (GNUNET_i2s (&ccc->p[1]->id));
-
-    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-                "Peer %u (`%s') sending message with type %u and size %u bytes to peer %u (`%s')\n",
-                ccc->p[1]->no,
-                ps,
-                ntohs (hdr->type),
-                ntohs (hdr->size),
-                p->no,
-                GNUNET_i2s (&p->id));
-    GNUNET_free (ps);
-  }
-
-  return TEST_MESSAGE_SIZE;
-}
-
-
-static void
-sendtask (void *cls)
-{
-  if (NULL == send_task)
-    ccc->global_ret = GNUNET_NO; /* first call! */
-  send_task = NULL;
-  {
-    char *receiver_s = GNUNET_strdup (GNUNET_i2s (&ccc->p[0]->id));
-
-    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-                "Sending message from peer %u (`%4s') -> peer %u (`%s') !\n",
-                ccc->p[1]->no, GNUNET_i2s (&ccc->p[1]->id), ccc->p[0]->no, receiver_s);
-    GNUNET_free (receiver_s);
-  }
-  if (0 == messages_recv)
-  {
-    start_normal = GNUNET_TIME_absolute_get();
-  }
-  if (1 == messages_recv)
-  {
-    start_delayed = GNUNET_TIME_absolute_get();
+                "Delayed message was not delayed correctly: took only %s\n",
+                GNUNET_STRINGS_relative_time_to_string (dur_delayed,
+                                                        GNUNET_YES));
   }
-
-  th = GNUNET_TRANSPORT_notify_transmit_ready (ccc->p[1]->th,
-                                               &ccc->p[0]->id,
-                                               TEST_MESSAGE_SIZE,
-                                               TIMEOUT_TRANSMIT,
-                                               &notify_ready,
-                                               ccc->p[0]);
-}
-
-
-static void
-notify_disconnect (void *cls,
-                   struct GNUNET_TRANSPORT_TESTING_PeerContext *me,
-                   const struct GNUNET_PeerIdentity *other)
-{
-  GNUNET_TRANSPORT_TESTING_log_disconnect (cls,
-                                           me,
-                                           other);
-  if (NULL != th)
-  {
-    GNUNET_TRANSPORT_notify_transmit_ready_cancel (th);
-    th = NULL;
+  else
+  { 
+    ccc->global_ret = GNUNET_OK;
   }
+  /* shutdown */
+  GNUNET_SCHEDULER_shutdown ();
 }
 
 
 int
-main (int argc, char *argv[])
+main (int argc,
+      char *argv[])
 {
   struct GNUNET_TRANSPORT_TESTING_ConnectCheckContext my_ccc = {
     .connect_continuation = &sendtask,
     .config_file = "test_transport_api_data.conf",
     .rec = &notify_receive,
     .nc = &GNUNET_TRANSPORT_TESTING_log_connect,
-    .nd = &notify_disconnect,
-    .shutdown_task = &custom_shutdown,
-    .timeout = TIMEOUT
+    .nd = &GNUNET_TRANSPORT_TESTING_log_disconnect,
+    .timeout = TIMEOUT,
+    .global_ret = GNUNET_NO
   };
 
   ccc = &my_ccc;