Small fix, improved comment
[oweals/gnunet.git] / src / mesh / test_mesh_small.c
index 0fd0b976630e591b83d57b7baafbdb102ab09f1c..42390e1a6e7ce1efdae6a6fab3ae4c34c97fd69a 100644 (file)
 /**
  * @file mesh/test_mesh_small.c
  *
- * @brief Test for the mesh service.
+ * @brief Test for the mesh service: retransmission of traffic.
  */
 #include "platform.h"
 #include "gnunet_testing_lib.h"
-#include "gnunet_mesh_service_new.h"
+#include "gnunet_mesh_service.h"
 
 #define VERBOSE GNUNET_YES
 #define REMOVE_DIR GNUNET_YES
@@ -47,17 +47,37 @@ struct StatsContext
 };
 
 
-// static struct MeshPeer *peer_head;
-//
-// static struct MeshPeer *peer_tail;
-
 /**
  * How long until we give up on connecting the peers?
  */
 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 1500)
 
+/**
+ * Time to wait for stuff that should be rather fast
+ */
+#define SHORT_TIME GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30)
+
+/**
+ * DIFFERENT TESTS TO RUN
+ */
+#define SETUP 0
+#define UNICAST 1
+#define MULTICAST 2
+
+/**
+ * Which test are we running?
+ */
+static int test;
+
+/**
+ * How many events have happened
+ */
 static int ok;
 
+static int peers_in_tunnel;
+
+static int peers_responded;
+
 /**
  * Be verbose
  */
@@ -98,11 +118,21 @@ static struct GNUNET_DISK_FileHandle *output_file;
  */
 static struct GNUNET_DISK_FileHandle *data_file;
 
+/**
+ * How many data points to capture before triggering next round?
+ */
+static struct GNUNET_TIME_Relative wait_time;
+
 /**
  * Task called to disconnect peers.
  */
 static GNUNET_SCHEDULER_TaskIdentifier disconnect_task;
 
+/**
+ * Task To perform tests
+ */
+static GNUNET_SCHEDULER_TaskIdentifier test_task;
+
 /**
  * Task called to shutdown test.
  */
@@ -110,9 +140,27 @@ static GNUNET_SCHEDULER_TaskIdentifier shutdown_handle;
 
 static char *topology_file;
 
-static char *data_filename;
+static struct GNUNET_TESTING_Daemon *d1;
+
+static GNUNET_PEER_Id pid1;
+
+static struct GNUNET_TESTING_Daemon *d2;
+
+static struct GNUNET_TESTING_Daemon *d3;
+
+static struct GNUNET_MESH_Handle *h1;
+
+static struct GNUNET_MESH_Handle *h2;
+
+static struct GNUNET_MESH_Handle *h3;
+
+static struct GNUNET_MESH_Tunnel *t;
+
+static struct GNUNET_MESH_Tunnel *incoming_t;
 
-static struct GNUNET_TIME_Relative time_out;
+static struct GNUNET_MESH_Tunnel *incoming_t2;
+
+static uint16_t *mesh_peers;
 
 /**
  * Check whether peers successfully shut down.
@@ -123,17 +171,17 @@ shutdown_callback (void *cls, const char *emsg)
   if (emsg != NULL)
   {
 #if VERBOSE
-    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Shutdown of peers failed!\n");
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "***************** test: Shutdown of peers failed!\n");
 #endif
-    if (ok == 0)
-      ok = 666;
+    ok--;
   }
   else
   {
 #if VERBOSE
-    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "All peers successfully shut down!\n");
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "***************** test: All peers successfully shut down!\n");
 #endif
-    ok = 0;
   }
 }
 
@@ -142,7 +190,8 @@ static void
 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
 #if VERBOSE
-  fprintf (stderr, "Ending test.\n");
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "***************** test: Ending test.\n");
 #endif
 
   if (disconnect_task != GNUNET_SCHEDULER_NO_TASK)
@@ -150,6 +199,7 @@ shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
     GNUNET_SCHEDULER_cancel (disconnect_task);
     disconnect_task = GNUNET_SCHEDULER_NO_TASK;
   }
+
   if (data_file != NULL)
     GNUNET_DISK_file_close (data_file);
   GNUNET_TESTING_daemons_stop (pg, TIMEOUT, &shutdown_callback, NULL);
@@ -157,26 +207,147 @@ shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 }
 
 
+static void
+disconnect_mesh_peers (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+{
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "***************** test: disconnecting mesh service of peers\n");
+  disconnect_task = GNUNET_SCHEDULER_NO_TASK;
+  GNUNET_MESH_disconnect (h1);
+  GNUNET_MESH_disconnect (h2);
+  if (test == MULTICAST)
+    GNUNET_MESH_disconnect (h3);
+  if (GNUNET_SCHEDULER_NO_TASK != shutdown_handle)
+  {
+    GNUNET_SCHEDULER_cancel (shutdown_handle);
+    shutdown_handle = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL);
+  }
+}
+
+
+/**
+ * Transmit ready callback
+ */
+size_t
+tmt_rdy (void *cls, size_t size, void *buf)
+{
+  struct GNUNET_MessageHeader *msg = buf;
+
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "***************** test:  tmt_rdy called\n");
+  if (size < sizeof (struct GNUNET_MessageHeader) || NULL == buf)
+    return 0;
+  msg->size = htons (sizeof (struct GNUNET_MessageHeader));
+  msg->type = htons ((long) cls);
+  return sizeof (struct GNUNET_MessageHeader);
+}
+
+
+/**
+ * Function is called whenever a message is received.
+ *
+ * @param cls closure (set from GNUNET_MESH_connect)
+ * @param tunnel connection to the other end
+ * @param tunnel_ctx place to store local state associated with the tunnel
+ * @param sender who sent the message
+ * @param message the actual message
+ * @param atsi performance data for the connection
+ * @return GNUNET_OK to keep the connection open,
+ *         GNUNET_SYSERR to close it (signal serious error)
+ */
+int
+data_callback (void *cls, struct GNUNET_MESH_Tunnel *tunnel, void **tunnel_ctx,
+               const struct GNUNET_PeerIdentity *sender,
+               const struct GNUNET_MessageHeader *message,
+               const struct GNUNET_ATS_Information *atsi)
+{
+  long client = (long) cls;
+
+  switch (client)
+  {
+  case 1L:
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "***************** test: Origin client got a response!\n");
+    ok++;
+    peers_responded++;
+    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;
+    GNUNET_MESH_tunnel_destroy (tunnel);
+    break;
+  case 2L:
+  case 3L:
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "***************** test: Destination client %u got a message.\n",
+                client);
+    ok++;
+    GNUNET_MESH_notify_transmit_ready (tunnel, GNUNET_NO, 0,
+                                       GNUNET_TIME_UNIT_FOREVER_REL, sender,
+                                       sizeof (struct GNUNET_MessageHeader),
+                                       &tmt_rdy, (void *) 1L);
+    if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
+    {
+      GNUNET_SCHEDULER_cancel (disconnect_task);
+      disconnect_task =
+          GNUNET_SCHEDULER_add_delayed (SHORT_TIME, &disconnect_mesh_peers,
+                                        NULL);
+    }
+    break;
+  default:
+    break;
+  }
+  return GNUNET_OK;
+}
+
+
 /**
  * Handlers, for diverse services
  */
 static struct GNUNET_MESH_MessageHandler handlers[] = {
-//    {&callback, 1, 0},
+  {&data_callback, 1, sizeof (struct GNUNET_MessageHeader)},
   {NULL, 0, 0}
 };
 
 
-static void
-disconnect_mesh (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+/**
+ * Method called whenever another peer has added us to a tunnel
+ * the other peer initiated.
+ *
+ * @param cls closure
+ * @param tunnel new handle to the tunnel
+ * @param initiator peer that started the tunnel
+ * @param atsi performance information for the tunnel
+ * @return initial tunnel context for the tunnel
+ *         (can be NULL -- that's not an error)
+ */
+static void *
+incoming_tunnel (void *cls, struct GNUNET_MESH_Tunnel *tunnel,
+                 const struct GNUNET_PeerIdentity *initiator,
+                 const struct GNUNET_ATS_Information *atsi)
 {
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-              "test: disconnecting mesh service of peers\n");
-  disconnect_task = GNUNET_SCHEDULER_NO_TASK;
-  GNUNET_SCHEDULER_cancel (shutdown_handle);
-  shutdown_handle = GNUNET_SCHEDULER_add_now(&shutdown_task, NULL);
+              "***************** test: Incoming tunnel from %s to peer %d\n",
+              GNUNET_i2s (initiator), (long) cls);
+  ok++;
+  if ((long) cls == 1L)
+    incoming_t = tunnel;
+  else
+    incoming_t2 = tunnel;
+  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 NULL;
 }
 
-
 /**
  * Function called whenever an inbound tunnel is destroyed.  Should clean up
  * any associated state.
@@ -190,12 +361,32 @@ static void
 tunnel_cleaner (void *cls, const struct GNUNET_MESH_Tunnel *tunnel,
                 void *tunnel_ctx)
 {
-#if VERBOSE
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "tunnel disconnected\n");
-#endif
+  long i = (long) cls;
+
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "***************** test: Incoming tunnel disconnected at peer %d\n",
+              i);
+  if (2L == i)
+    ok++;
+  else if (3L == i)
+    ok++;
+  else
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "***************** test: Unknown peer! %d\n", i);
+  peers_in_tunnel--;
+  if (peers_in_tunnel > 0)
+    return;
+
+  if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
+  {
+    GNUNET_SCHEDULER_cancel (disconnect_task);
+    disconnect_task = GNUNET_SCHEDULER_add_now (&disconnect_mesh_peers, NULL);
+  }
+
   return;
 }
 
+
 /**
  * Method called whenever a tunnel falls apart.
  *
@@ -205,13 +396,15 @@ tunnel_cleaner (void *cls, const struct GNUNET_MESH_Tunnel *tunnel,
 static void
 dh (void *cls, const struct GNUNET_PeerIdentity *peer)
 {
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "peer disconnected\n");
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "***************** test: peer %s disconnected\n",
+              GNUNET_i2s (peer));
   return;
 }
 
 
 /**
- * Method called whenever a tunnel is established.
+ * Method called whenever a peer connects to a tunnel.
  *
  * @param cls closure
  * @param peer peer identity the tunnel was created to, NULL on timeout
@@ -219,46 +412,210 @@ dh (void *cls, const struct GNUNET_PeerIdentity *peer)
  */
 static void
 ch (void *cls, const struct GNUNET_PeerIdentity *peer,
-    const struct GNUNET_TRANSPORT_ATS_Information *atsi)
+    const struct GNUNET_ATS_Information *atsi)
 {
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "peer connected\n");
+  struct GNUNET_PeerIdentity *dest;
+
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "***************** test: peer %s connected\n", GNUNET_i2s (peer));
+
+  if (0 == memcmp (&d2->id, peer, sizeof (d2->id)) && (long) cls == 1L)
+    ok++;
+  if (test == MULTICAST && 0 == memcmp (&d3->id, peer, sizeof (d3->id)) &&
+      (long) cls == 1L)
+    ok++;
+  switch (test)
+  {
+  case UNICAST:
+    dest = &d2->id;
+    break;
+  case MULTICAST:
+    peers_in_tunnel++;
+    if (peers_in_tunnel < 2)
+      return;
+    dest = NULL;
+    break;
+  default:
+    return;
+  }
+  if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
+  {
+    GNUNET_SCHEDULER_cancel (disconnect_task);
+    disconnect_task =
+        GNUNET_SCHEDULER_add_delayed (SHORT_TIME, &disconnect_mesh_peers, NULL);
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "***************** test: Sending data...\n");
+    peers_responded = 0;
+    GNUNET_MESH_notify_transmit_ready (t, GNUNET_NO, 0,
+                                       GNUNET_TIME_UNIT_FOREVER_REL, dest,
+                                       sizeof (struct GNUNET_MessageHeader),
+                                       &tmt_rdy, (void *) 1L);
+  }
+  else
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "***************** test: Disconnect already run?\n");
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "***************** test: Aborting...\n");
+  }
   return;
 }
 
 
+static void
+do_test (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+{
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "***************** test: test_task\n");
+  if (test == MULTICAST)
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "***************** test: add peer 3\n");
+    GNUNET_MESH_peer_request_connect_add (t, &d3->id);
+  }
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "***************** test: add peer 2\n");
+  GNUNET_MESH_peer_request_connect_add (t, &d2->id);
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "***************** test: schedule timeout in 30s\n");
+  if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
+  {
+    GNUNET_SCHEDULER_cancel (disconnect_task);
+    disconnect_task =
+        GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
+                                      (GNUNET_TIME_UNIT_SECONDS, 30),
+                                      &disconnect_mesh_peers, NULL);
+  }
+}
+
+
+/**
+ * Prototype of a callback function indicating that two peers
+ * are currently connected.
+ *
+ * @param cls closure
+ * @param first peer id for first daemon
+ * @param second peer id for the second daemon
+ * @param distance distance between the connected peers
+ * @param emsg error message (NULL on success)
+ */
+void
+topo_cb (void *cls, const struct GNUNET_PeerIdentity *first,
+         const struct GNUNET_PeerIdentity *second, const char *emsg)
+{
+  GNUNET_PEER_Id p1;
+  GNUNET_PEER_Id p2;
+  struct GNUNET_PeerIdentity id;
+
+  GNUNET_PEER_resolve (1, &id);
+  p1 = GNUNET_PEER_search (first);
+  if (p1 == pid1)
+  {
+    p2 = GNUNET_PEER_search (second);
+    if (p2 == 0 || p2 > num_peers)
+    {
+      GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
+                  "***************** test: %s is UNKNOWN!? (%u)\n",
+                  GNUNET_i2s (second), p2);
+      return;
+    }
+    mesh_peers[p2]++;
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "***************** test: %s IS a neighbor\n",
+                GNUNET_i2s (second));
+    return;
+  }
+  p1 = GNUNET_PEER_search (second);
+  if (p1 == pid1)
+  {
+    p2 = GNUNET_PEER_search (first);
+    if (p2 == 0 || p2 > num_peers)
+    {
+      GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
+                  "***************** test: %s is UNKNOWN!? (%u)\n",
+                  GNUNET_i2s (first), p2);
+      return;
+    }
+    mesh_peers[p2]++;
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "***************** test: %s IS a neighbor\n",
+                GNUNET_i2s (first));
+    return;
+  }
+}
+
 /**
  * connect_mesh_service: connect to the mesh service of one of the peers
  *
  */
 static void
-connect_mesh (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+connect_mesh_service (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
-  struct GNUNET_TESTING_Daemon *d;
-  struct GNUNET_MESH_Handle *h;
-  struct GNUNET_MESH_Tunnel *t;
   GNUNET_MESH_ApplicationType app;
+  unsigned int i;
+  struct GNUNET_PeerIdentity id;
 
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "connect_mesh_service\n");
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "***************** test: connect_mesh_service\n");
 
-  d = GNUNET_TESTING_daemon_get (pg, 1);
+  for (i = 1; i <= num_peers; i++)
+  {
+    GNUNET_PEER_resolve (i, &id);
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "***************** test:   peer %s has %u conns to d1\n",
+                GNUNET_i2s (&id), mesh_peers[i]);
+    if (mesh_peers[i] == 0)
+      break;
+  }
+  GNUNET_assert (i < num_peers);
+  d2 = GNUNET_TESTING_daemon_get_by_id (pg, &id);
+  if (test == MULTICAST)
+  {
+    for (i++; i <= num_peers; i++)
+    {
+      GNUNET_PEER_resolve (i, &id);
+      GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                  "***************** test:   peer %s has %u conns to d1\n",
+                  GNUNET_i2s (&id), mesh_peers[i]);
+      if (mesh_peers[i] == 0)
+        break;
+    }
+    GNUNET_assert (i < num_peers);
+    d3 = GNUNET_TESTING_daemon_get_by_id (pg, &id);
+  }
   app = (GNUNET_MESH_ApplicationType) 0;
 
 #if VERBOSE
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-              "connecting to mesh service of peer %s\n", GNUNET_i2s (&d->id));
-#endif
-  h = GNUNET_MESH_connect (d->cfg, 10, NULL, NULL, &tunnel_cleaner, handlers,
-                           &app);
-#if VERBOSE
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "connected to mesh service of peer %s\n",
-              GNUNET_i2s (&d->id));
+              "***************** test: connecting to mesh service of peer %s (%u)\n",
+              GNUNET_i2s (&d1->id), mesh_peers[0]);
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "***************** test: connecting to mesh service of peer %s (%u)\n",
+              GNUNET_i2s (&d2->id), i);
+  if (test == MULTICAST)
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "***************** test: connecting to mesh service of peer %s (%u)\n",
+                GNUNET_i2s (&d3->id), i);
+  }
 #endif
-  t = GNUNET_MESH_tunnel_create (h, NULL, &ch, &dh, NULL);
-  GNUNET_MESH_tunnel_destroy (t);
-  GNUNET_MESH_disconnect (h);
+  h1 = GNUNET_MESH_connect (d1->cfg, 10, (void *) 1L, NULL, &tunnel_cleaner,
+                            handlers, &app);
+  h2 = GNUNET_MESH_connect (d2->cfg, 10, (void *) 2L, &incoming_tunnel,
+                            &tunnel_cleaner, handlers, &app);
+  if (test == MULTICAST)
+  {
+    h3 = GNUNET_MESH_connect (d3->cfg, 10, (void *) 3L, &incoming_tunnel,
+                              &tunnel_cleaner, handlers, &app);
+  }
+  t = GNUNET_MESH_tunnel_create (h1, NULL, &ch, &dh, (void *) 1L);
+  peers_in_tunnel = 0;
+  test_task =
+      GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
+                                    (GNUNET_TIME_UNIT_SECONDS, 6), &do_test,
+                                    NULL);
 }
 
 
+
 /**
  * peergroup_ready: start test when all peers are connected
  * @param cls closure
@@ -269,22 +626,28 @@ peergroup_ready (void *cls, const char *emsg)
 {
   char *buf;
   int buf_len;
+  unsigned int i;
 
   if (emsg != NULL)
   {
     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-                "Peergroup callback called with error, aborting test!\n");
-    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Error from testing: `%s'\n");
-    ok = 1;
+                "***************** test: Peergroup callback called with error, aborting test!\n");
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "***************** test: Error from testing: `%s'\n", emsg);
+    ok--;
     GNUNET_TESTING_daemons_stop (pg, TIMEOUT, &shutdown_callback, NULL);
     return;
   }
 #if VERBOSE
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Peer Group started successfully!\n");
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "************************************************************\n");
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "***************** test: Peer Group started successfully!\n");
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "***************** test: Have %u connections\n",
+              total_connections);
 #endif
 
-  GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Have %u connections\n",
-              total_connections);
   if (data_file != NULL)
   {
     buf = NULL;
@@ -294,10 +657,28 @@ peergroup_ready (void *cls, const char *emsg)
     GNUNET_free (buf);
   }
   peers_running = GNUNET_TESTING_daemons_running (pg);
+  for (i = 0; i < num_peers; i++)
+  {
+    GNUNET_PEER_Id peer_id;
 
-  GNUNET_SCHEDULER_add_now (&connect_mesh, NULL);
+    d1 = GNUNET_TESTING_daemon_get (pg, i);
+    peer_id = GNUNET_PEER_intern (&d1->id);
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "***************** test:   %u: %s\n",
+                peer_id, GNUNET_i2s (&d1->id));
+  }
+  d1 = GNUNET_TESTING_daemon_get (pg, 0);
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "***************** test: Peer looking: %s\n",
+              GNUNET_i2s (&d1->id));
+  pid1 = GNUNET_PEER_intern (&d1->id);
+  mesh_peers[pid1] = 100;
+  GNUNET_TESTING_get_topology (pg, &topo_cb, NULL);
+
+  GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
+                                (GNUNET_TIME_UNIT_SECONDS, 4),
+                                &connect_mesh_service, NULL);
   disconnect_task =
-      GNUNET_SCHEDULER_add_delayed (time_out, &disconnect_mesh, NULL);
+      GNUNET_SCHEDULER_add_delayed (wait_time, &disconnect_mesh_peers, NULL);
 
 }
 
@@ -325,7 +706,20 @@ connect_cb (void *cls, const struct GNUNET_PeerIdentity *first,
             struct GNUNET_TESTING_Daemon *second_daemon, const char *emsg)
 {
   if (emsg == NULL)
+  {
     total_connections++;
+  }
+  else
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "***************** test: Problem with new connection (%s)\n",
+                emsg);
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "***************** test:   (%s)\n",
+                GNUNET_i2s (first));
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "***************** test:   (%s)\n",
+                GNUNET_i2s (second));
+  }
+
 }
 
 
@@ -341,9 +735,11 @@ run (void *cls, char *const *args, const char *cfgfile,
      const struct GNUNET_CONFIGURATION_Handle *cfg)
 {
   char *temp_str;
+  unsigned long long temp_wait;
   struct GNUNET_TESTING_Host *hosts;
+  char *data_filename;
 
-  ok = 1;
+  ok = 0;
   testing_cfg = GNUNET_CONFIGURATION_dup (cfg);
 
   GNUNET_log_setup ("test_mesh_small",
@@ -355,28 +751,30 @@ run (void *cls, char *const *args, const char *cfgfile,
                     NULL);
 
 #if VERBOSE
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Starting daemons.\n");
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "***************** test: Starting daemons.\n");
   GNUNET_CONFIGURATION_set_value_string (testing_cfg, "testing",
                                          "use_progressbars", "YES");
 #endif
 
-  time_out = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30);
-
   if (GNUNET_OK !=
       GNUNET_CONFIGURATION_get_value_number (testing_cfg, "testing",
                                              "num_peers", &num_peers))
   {
-    GNUNET_assert (GNUNET_OK ==
-                   GNUNET_CONFIGURATION_load (testing_cfg,
-                                              "test_mesh_small.conf"));
-    if (GNUNET_OK !=
-        GNUNET_CONFIGURATION_get_value_number (testing_cfg, "testing",
-                                               "num_peers", &num_peers))
-    {
-      GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
-                  "Option TESTING:NUM_PEERS is required!\n");
-      return;
-    }
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+                "Option TESTING:NUM_PEERS is required!\n");
+    return;
+  }
+
+  mesh_peers = GNUNET_malloc (sizeof (uint16_t) * (num_peers + 1));
+
+  if (GNUNET_OK !=
+      GNUNET_CONFIGURATION_get_value_number (testing_cfg, "test_mesh_small",
+                                             "wait_time", &temp_wait))
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+                "Option test_mesh_small:wait_time is required!\n");
+    return;
   }
 
   if (GNUNET_OK !=
@@ -389,28 +787,27 @@ run (void *cls, char *const *args, const char *cfgfile,
     return;
   }
 
-  if (GNUNET_OK !=
+  if (GNUNET_OK ==
       GNUNET_CONFIGURATION_get_value_string (testing_cfg, "test_mesh_small",
                                              "data_output_file",
                                              &data_filename))
   {
-    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
-                "Option test_mesh_small:data_output_file is required!\n");
-    return;
+    data_file =
+        GNUNET_DISK_file_open (data_filename,
+                               GNUNET_DISK_OPEN_READWRITE |
+                               GNUNET_DISK_OPEN_CREATE,
+                               GNUNET_DISK_PERM_USER_READ |
+                               GNUNET_DISK_PERM_USER_WRITE);
+    if (data_file == NULL)
+    {
+      GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Failed to open %s for output!\n",
+                  data_filename);
+      GNUNET_free (data_filename);
+    }
   }
 
-  data_file =
-      GNUNET_DISK_file_open (data_filename,
-                             GNUNET_DISK_OPEN_READWRITE |
-                             GNUNET_DISK_OPEN_CREATE,
-                             GNUNET_DISK_PERM_USER_READ |
-                             GNUNET_DISK_PERM_USER_WRITE);
-  if (data_file == NULL)
-  {
-    GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Failed to open %s for output!\n",
-                data_filename);
-    GNUNET_free (data_filename);
-  }
+  wait_time =
+      GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, temp_wait);
 
   if (GNUNET_YES ==
       GNUNET_CONFIGURATION_get_value_string (cfg, "test_mesh_small",
@@ -458,13 +855,63 @@ static struct GNUNET_GETOPT_CommandLineOption options[] = {
 int
 main (int argc, char *argv[])
 {
-  GNUNET_PROGRAM_run (argc, argv, "test_mesh_small",
+  char *const argv2[] = {
+    argv[0],
+    "-c",
+    "test_mesh_small.conf",
+#if VERBOSE
+    "-L",
+    "DEBUG",
+#endif
+    NULL
+  };
+
+  /* Each peer is supposed to generate the following callbacks:
+   * 1 incoming tunnel (@dest)
+   * 1 connected peer (@orig)
+   * 1 received data packet (@dest)
+   * 1 received data packet (@orig)
+   * 1 received tunnel destroy (@dest)
+   * _________________________________
+   * 5 x ok expected per peer
+   */
+  int ok_goal;
+
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "***************** test: Start\n");
+  if (strstr (argv[0], "test_mesh_small_unicast") != NULL)
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "***************** test: UNICAST\n");
+    test = UNICAST;
+    ok_goal = 5;
+  }
+  else if (strstr (argv[0], "test_mesh_small_multicast") != NULL)
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "***************** test: MULTICAST\n");
+    test = MULTICAST;
+    ok_goal = 10;
+  }
+  else
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "***************** test: UNKNOWN\n");
+    test = SETUP;
+    ok_goal = 0;
+  }
+
+  GNUNET_PROGRAM_run ((sizeof (argv2) / sizeof (char *)) - 1, argv2,
+                      "test_mesh_small",
                       gettext_noop ("Test mesh in a small network."), options,
                       &run, NULL);
 #if REMOVE_DIR
   GNUNET_DISK_directory_remove ("/tmp/test_mesh_small");
 #endif
-  return ok;
+  if (ok_goal != ok)
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
+                "***************** test: FAILED! (%d/%d)\n", ok, ok_goal);
+    return 1;
+  }
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "***************** test: success\n");
+  return 0;
 }
 
 /* end of test_mesh_small.c */