- arm doesn't have time to finish in 30s
[oweals/gnunet.git] / src / mesh / test_mesh_small.c
index 96f448461d3293de4de92da3a4e0b633b6290b9d..c5afd576adc993ac96f1d0c69caa69d38c2977c5 100644 (file)
@@ -56,7 +56,7 @@ struct MeshPeer
 /**
  * Time to wait for stuff that should be rather fast
  */
-#define SHORT_TIME GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30)
+#define SHORT_TIME GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 60)
 
 /**
  * DIFFERENT TESTS TO RUN
@@ -74,6 +74,11 @@ struct MeshPeer
  */
 static int test;
 
+/**
+ * String with test name
+ */
+char *test_name;
+
 /**
  * Flag to send traffic leaf->root in speed tests to test BCK_ACK logic.
  */
@@ -96,6 +101,12 @@ static int ok;
   */
 int ok_goal;
 
+
+/**
+ * Is the setup initialized?
+ */
+static int initialized;
+
 /**
  * Peers that have been connected
  */
@@ -181,6 +192,9 @@ static GNUNET_SCHEDULER_TaskIdentifier test_task;
  */
 static GNUNET_SCHEDULER_TaskIdentifier shutdown_handle;
 
+/**
+ * Filename of the file containing the topology.
+ */
 static char *topology_file;
 
 /**
@@ -228,17 +242,42 @@ static struct GNUNET_MESH_Tunnel *incoming_t;
  */
 static struct GNUNET_MESH_Tunnel *incoming_t2;
 
-static GNUNET_PEER_Id pid1;
-
+/**
+ * Time we started the data transmission (after tunnel has been established
+ * and initilized).
+ */
 static struct GNUNET_TIME_Absolute start_time;
 
-static struct GNUNET_TIME_Absolute end_time;
 
-static struct GNUNET_TIME_Relative total_time;
+/**
+ * Show the results of the test (banwidth acheived) and log them to GAUGER
+ */
+static void
+show_end_data (void)
+{
+  static struct GNUNET_TIME_Absolute end_time;
+  static struct GNUNET_TIME_Relative total_time;
+
+  end_time = GNUNET_TIME_absolute_get();
+  total_time = GNUNET_TIME_absolute_get_difference(start_time, end_time);
+  FPRINTF (stderr, "\nResults of test \"%s\"\n", test_name);
+  FPRINTF (stderr, "Test time %llu ms\n",
+            (unsigned long long) total_time.rel_value);
+  FPRINTF (stderr, "Test bandwidth: %f kb/s\n",
+            4 * TOTAL_PACKETS * 1.0 / total_time.rel_value); // 4bytes * ms
+  FPRINTF (stderr, "Test throughput: %f packets/s\n\n",
+            TOTAL_PACKETS * 1000.0 / total_time.rel_value); // packets * ms
+  GAUGER ("MESH", test_name,
+          TOTAL_PACKETS * 1000.0 / total_time.rel_value,
+          "packets/s");
+}
 
 
 /**
  * Check whether peers successfully shut down.
+ * 
+ * @param cls Closure (unused).
+ * @param emsg Error message.
  */
 static void
 shutdown_callback (void *cls, const char *emsg)
@@ -264,6 +303,9 @@ shutdown_callback (void *cls, const char *emsg)
 
 /**
  * Shut down peergroup, clean up.
+ * 
+ * @param cls Closure (unused).
+ * @param tc Task Context.
  */
 static void
 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
@@ -303,6 +345,9 @@ shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 
 /**
  * Disconnect from mesh services af all peers, call shutdown.
+ * 
+ * @param cls Closure (unused).
+ * @param tc Task Context.
  */
 static void
 disconnect_mesh_peers (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
@@ -340,9 +385,26 @@ disconnect_mesh_peers (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
   }
 }
 
+
+/**
+ * Transmit ready callback.
+ * 
+ * @param cls Closure (peer #).
+ * @param size Size of the tranmist buffer.
+ * @param buf Pointer to the beginning of the buffer.
+ * 
+ * @return Number of bytes written to buf.
+ */
 static size_t
 tmt_rdy (void *cls, size_t size, void *buf);
 
+
+/**
+ * Task to schedule a new data transmission.
+ * 
+ * @param cls Closure (peer #).
+ * @param tc Task Context.
+ */
 static void
 data_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
@@ -352,6 +414,8 @@ data_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 
   if ((GNUNET_SCHEDULER_REASON_SHUTDOWN & tc->reason) != 0)
     return;
+
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Data task\n");
   if (GNUNET_YES == test_backwards)
   {
     tunnel = incoming_t;
@@ -390,6 +454,7 @@ data_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
   }
 }
 
+
 /**
  * Transmit ready callback
  *
@@ -408,7 +473,7 @@ tmt_rdy (void *cls, size_t size, void *buf)
     return 0;
   msg->size = htons (sizeof (struct GNUNET_MessageHeader));
   msg->type = htons ((long) cls);
-  if (test == SPEED)
+  if (SPEED == test && GNUNET_YES == initialized)
   {
     data_sent++;
     if (data_sent < TOTAL_PACKETS)
@@ -441,73 +506,57 @@ data_callback (void *cls, struct GNUNET_MESH_Tunnel *tunnel, void **tunnel_ctx,
                const struct GNUNET_ATS_Information *atsi)
 {
   long client = (long) cls;
-  long expected_client;
-  struct GNUNET_MESH_Tunnel *tunnel_to_use;
-  struct GNUNET_PeerIdentity *dest_to_use;
-
-  if (GNUNET_YES == test_backwards)
-  {
-    expected_client = 1L;
-    dest_to_use = &d1->id;
-    tunnel_to_use = incoming_t;
-  }
-  else
-  {
-    expected_client = 0L;
-    dest_to_use = &d2->id;
-    tunnel_to_use = t;
-  }
+  long expected_target_client;
 
+  ok++;
   switch (client)
   {
   case 1L:
-    GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Root client got a response!\n");
-    ok++;
+    GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Root client got a message!\n");
     GNUNET_log (GNUNET_ERROR_TYPE_INFO, " ok: %d\n", ok);
     peers_responded++;
-    data_ack++;
-    if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
-    {
-      GNUNET_SCHEDULER_cancel (disconnect_task);
-      disconnect_task =
-          GNUNET_SCHEDULER_add_delayed (SHORT_TIME, &disconnect_mesh_peers,
-                                        NULL);
-    }
     if (test == MULTICAST && peers_responded < 2)
       return GNUNET_OK;
-    if (test == SPEED_ACK || test == SPEED)
-    {
-      GNUNET_log (GNUNET_ERROR_TYPE_INFO,
-              " received ack %u\n", data_ack);
-      GNUNET_MESH_notify_transmit_ready (tunnel, GNUNET_NO,
-                                        GNUNET_TIME_UNIT_FOREVER_REL, sender,
-                                        sizeof (struct GNUNET_MessageHeader),
-                                        &tmt_rdy, (void *) 1L);
-      if (data_ack < TOTAL_PACKETS && test != SPEED)
-        return GNUNET_OK;
-      end_time = GNUNET_TIME_absolute_get();
-      total_time = GNUNET_TIME_absolute_get_difference(start_time, end_time);
-      FPRINTF (stderr, "\nTest time %llu ms\n",
-               (unsigned long long) total_time.rel_value);
-      FPRINTF (stderr, "Test bandwidth: %f kb/s\n",
-               4 * TOTAL_PACKETS * 1.0 / total_time.rel_value); // 4bytes * ms
-      FPRINTF (stderr, "Test throughput: %f packets/s\n\n",
-               TOTAL_PACKETS * 1000.0 / total_time.rel_value); // packets * ms
-      GAUGER ("MESH", "Tunnel 5 peers",
-              TOTAL_PACKETS * 1000.0 / total_time.rel_value,
-              "packets/s");
-    }
-    GNUNET_assert (tunnel == t);
-    GNUNET_MESH_tunnel_destroy (t);
-    t = NULL;
     break;
   case 2L:
   case 3L:
     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
-                "Leaf client %u got a message.\n",
+                "Leaf client %li got a message.\n",
                 client);
-    ok++;
     GNUNET_log (GNUNET_ERROR_TYPE_INFO, " ok: %d\n", ok);
+    client = 2L;
+    break;
+  default:
+    GNUNET_assert (0);
+    break;
+  }
+
+  if (SPEED == test && GNUNET_YES == test_backwards)
+  {
+    expected_target_client = 1L;
+  }
+  else
+  {
+    expected_target_client = 2L;
+  }
+
+  if (GNUNET_NO == initialized)
+  {
+    initialized = GNUNET_YES;
+    start_time = GNUNET_TIME_absolute_get ();
+    if (SPEED == test)
+    {
+      GNUNET_assert (2L == client);
+      GNUNET_SCHEDULER_add_now (&data_task, NULL);
+      return GNUNET_OK;
+    }
+  }
+
+  if (client == expected_target_client) // Normally 2 or 3
+  {
+    data_received++;
+    GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+                " received data %u\n", data_received);
     if (SPEED != test || (ok_goal - 2) == ok)
     {
       GNUNET_MESH_notify_transmit_ready (tunnel, GNUNET_NO,
@@ -517,27 +566,39 @@ data_callback (void *cls, struct GNUNET_MESH_Tunnel *tunnel, void **tunnel_ctx,
     }
     else
     {
-      data_received++;
-      GNUNET_log (GNUNET_ERROR_TYPE_INFO,
-              " received data %u\n", data_received);
       if (data_received < TOTAL_PACKETS)
         return GNUNET_OK;
     }
-    if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
+  }
+  else // Normally 1
+  {
+    if (test == SPEED_ACK || test == SPEED)
     {
-      GNUNET_SCHEDULER_cancel (disconnect_task);
-      disconnect_task =
-          GNUNET_SCHEDULER_add_delayed (SHORT_TIME, &disconnect_mesh_peers,
-                                        NULL);
+      data_ack++;
+      GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+              " received ack %u\n", data_ack);
+      GNUNET_MESH_notify_transmit_ready (tunnel, GNUNET_NO,
+                                        GNUNET_TIME_UNIT_FOREVER_REL, sender,
+                                        sizeof (struct GNUNET_MessageHeader),
+                                        &tmt_rdy, (void *) 1L);
+      if (data_ack < TOTAL_PACKETS && SPEED != test)
+        return GNUNET_OK;
+      if (ok == 2 && SPEED == test)
+        return GNUNET_OK;
+      show_end_data();
     }
-    break;
-  default:
-    break;
+    GNUNET_MESH_tunnel_destroy (t);
+    t = NULL;
   }
-  
-  
-  
-  
+
+  if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
+  {
+    GNUNET_SCHEDULER_cancel (disconnect_task);
+    disconnect_task =
+        GNUNET_SCHEDULER_add_delayed (SHORT_TIME, &disconnect_mesh_peers,
+                                      NULL);
+  }
+
   return GNUNET_OK;
 }
 
@@ -580,6 +641,7 @@ incoming_tunnel (void *cls, struct GNUNET_MESH_Tunnel *tunnel,
   {
     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
                 "Incoming tunnel for unknown client %lu\n", (long) cls);
+    GNUNET_break(0);
   }
   if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
   {
@@ -587,6 +649,7 @@ incoming_tunnel (void *cls, struct GNUNET_MESH_Tunnel *tunnel,
     disconnect_task =
         GNUNET_SCHEDULER_add_delayed (SHORT_TIME, &disconnect_mesh_peers, NULL);
   }
+
   return NULL;
 }
 
@@ -664,7 +727,6 @@ ch (void *cls, const struct GNUNET_PeerIdentity *peer,
     const struct GNUNET_ATS_Information *atsi)
 {
   struct GNUNET_PeerIdentity *dest;
-  struct GNUNET_MESH_Tunnel *tunnel;
 
   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
               "peer %s connected\n", GNUNET_i2s (peer));
@@ -684,26 +746,16 @@ ch (void *cls, const struct GNUNET_PeerIdentity *peer,
     case UNICAST:
     case SPEED:
     case SPEED_ACK:
-      if (GNUNET_YES == test_backwards)
-      {
-        dest = &d1->id;
-        tunnel = incoming_t;
-      }
-      else
-      {
-        dest = &d2->id;
-        tunnel = t;
-      }
+      // incoming_t is NULL unless we send a relevant data packet
+      dest = &d2->id;
       break;
     case MULTICAST:
       peers_in_tunnel++;
       if (peers_in_tunnel < 2)
         return;
       dest = NULL;
-      tunnel = t;
       break;
     default:
-      tunnel = t;
       return;
   }
   if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
@@ -712,13 +764,12 @@ ch (void *cls, const struct GNUNET_PeerIdentity *peer,
     disconnect_task =
         GNUNET_SCHEDULER_add_delayed (SHORT_TIME, &disconnect_mesh_peers, NULL);
     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-                "Sending data...\n");
+                "Sending data initializer...\n");
     peers_responded = 0;
     data_ack = 0;
     data_received = 0;
     data_sent = 0;
-    start_time = GNUNET_TIME_absolute_get();
-    GNUNET_MESH_notify_transmit_ready (tunnel, GNUNET_NO,
+    GNUNET_MESH_notify_transmit_ready (t, GNUNET_NO,
                                        GNUNET_TIME_UNIT_FOREVER_REL, dest,
                                        sizeof (struct GNUNET_MessageHeader),
                                        &tmt_rdy, (void *) 1L);
@@ -734,18 +785,29 @@ ch (void *cls, const struct GNUNET_PeerIdentity *peer,
 }
 
 
+/**
+ * START THE TESTCASE ITSELF, AS WE ARE CONNECTED TO THE MESH SERVICES.
+ * 
+ * Testcase continues when the root receives confirmation of connected peers,
+ * on callback funtion ch.
+ * 
+ * @param cls Closure (unsued).
+ * @param tc Task Context.
+ */
 static void
 do_test (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test_task\n");
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "add peer 2\n");
+  GNUNET_MESH_peer_request_connect_add (t, &d2->id);
+
   if (test == MULTICAST)
   {
     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
                 "add peer 3\n");
     GNUNET_MESH_peer_request_connect_add (t, &d3->id);
   }
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "add peer 2\n");
-  GNUNET_MESH_peer_request_connect_add (t, &d2->id);
+
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
               "schedule timeout in SHORT_TIME\n");
   if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
@@ -760,12 +822,17 @@ do_test (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 /**
  * connect_mesh_service: connect to the mesh service of one of the peers
  *
+ * @param cls Closure (unsued).
+ * @param tc Task Context.
  */
 static void
 connect_mesh_service (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
   GNUNET_MESH_ApplicationType app;
 
+  if ((GNUNET_SCHEDULER_REASON_SHUTDOWN & tc->reason) != 0)
+    return;
+
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
               "connect_mesh_service\n");
 
@@ -821,6 +888,7 @@ connect_mesh_service (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 
 /**
  * peergroup_ready: start test when all peers are connected
+ * 
  * @param cls closure
  * @param emsg error message
  */
@@ -873,7 +941,6 @@ peergroup_ready (void *cls, const char *emsg)
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
               "Peer looking: %s\n",
               GNUNET_i2s (&d1->id));
-  pid1 = GNUNET_PEER_intern (&d1->id);
 
   GNUNET_SCHEDULER_add_now (&connect_mesh_service, NULL);
   disconnect_task =
@@ -922,6 +989,7 @@ connect_cb (void *cls, const struct GNUNET_PeerIdentity *first,
 
 /**
  * run: load configuration options and schedule test to run (start peergroup)
+ * 
  * @param cls closure
  * @param args argv
  * @param cfgfile configuration file name (can be NULL)
@@ -1058,17 +1126,21 @@ main (int argc, char *argv[])
   };
   int argc2 = (sizeof (argv2) / sizeof (char *)) - 1;
 
+  initialized = GNUNET_NO;
+
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Start\n");
   if (strstr (argv[0], "test_mesh_small_unicast") != NULL)
   {
     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "UNICAST\n");
     test = UNICAST;
+    test_name = "unicast";
     ok_goal = 5;
   }
   else if (strstr (argv[0], "test_mesh_small_multicast") != NULL)
   {
     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MULTICAST\n");
     test = MULTICAST;
+    test_name = "multicast";
     ok_goal = 10;
   }
   else if (strstr (argv[0], "test_mesh_small_speed_ack") != NULL)
@@ -1084,6 +1156,7 @@ main (int argc, char *argv[])
     */
     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "SPEED_ACK\n");
     test = SPEED_ACK;
+    test_name = "speed ack";
     ok_goal = TOTAL_PACKETS * 2 + 3;
     argv2 [3] = NULL; // remove -L DEBUG
 #if VERBOSE
@@ -1103,11 +1176,20 @@ main (int argc, char *argv[])
     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "SPEED\n");
     ok_goal = TOTAL_PACKETS + 4;
     if (strstr (argv[0], "_min") != NULL)
+    {
       test = SPEED_MIN;
+      test_name = "speed min";
+    }
     else if (strstr (argv[0], "_nobuf") != NULL)
+    {
       test = SPEED_NOBUF;
+      test_name = "speed nobuf";
+    }
     else
+    {
       test = SPEED;
+      test_name = "speed";
+    }
   }
   else
   {
@@ -1118,8 +1200,14 @@ main (int argc, char *argv[])
 
   if (strstr (argv[0], "backwards") != NULL)
   {
+    char *aux;
+
     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "BACKWARDS (LEAF TO ROOT)\n");
     test_backwards = GNUNET_YES;
+    ok_goal++; // need one root->leaf packet to initialize tunnel
+    aux = malloc (32); // "leaked"
+    sprintf (aux, "backwards %s", test_name);
+    test_name = aux;
   }
 
   GNUNET_PROGRAM_run (argc2, argv2,