- begin work on enhanced multipart receiving
[oweals/gnunet.git] / src / ats-tests / perf_ats.c
index c5fb665ef29d537fc3b4a6b07c6630882157e70a..f52bcfb25c578013a791ae753bb15f4ea8e478a0 100644 (file)
@@ -1,22 +1,22 @@
 /*
    This file is part of GNUnet.
    (C) 2010-2013 Christian Grothoff (and other contributing authors)
-
    GNUnet is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published
    by the Free Software Foundation; either version 3, or (at your
    option) any later version.
-
    GNUnet is distributed in the hope that it will be useful, but
    WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
    General Public License for more details.
-
    You should have received a copy of the GNU General Public License
    along with GNUnet; see the file COPYING.  If not, write to the
    Free Software Foundation, Inc., 59 Temple Place - Suite 330,
    Boston, MA 02111-1307, USA.
-*/
+ This file is part of GNUnet.
+ (C) 2010-2013 Christian Grothoff (and other contributing authors)
+
+ GNUnet is free software; you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published
+ by the Free Software Foundation; either version 3, or (at your
+ option) any later version.
+
+ GNUnet is distributed in the hope that it will be useful, but
+ WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with GNUnet; see the file COPYING.  If not, write to the
+ Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ Boston, MA 02111-1307, USA.
+ */
 /**
  * @file ats/perf_ats.c
  * @brief ats benchmark: start peers and modify preferences, monitor change over time
 #include "gnunet_testbed_service.h"
 #include "gnunet_ats_service.h"
 #include "gnunet_core_service.h"
+#include "perf_ats.h"
+
 
-#define TEST_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 10)
-#define BENCHMARK_DURATION GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 15)
-#define TESTNAME_PREFIX "perf_ats_"
-#define DEFAULT_SLAVES_NUM 3
-#define DEFAULT_MASTERS_NUM 1
+#define TEST_ATS_PREFRENCE_FREQUENCY GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 1)
+#define TEST_ATS_PREFRENCE_START 1.0
+#define TEST_ATS_PREFRENCE_DELTA 1.0
 
 #define TEST_MESSAGE_TYPE_PING 12345
 #define TEST_MESSAGE_TYPE_PONG 12346
 #define TEST_MESSAGE_SIZE 1000
 #define TEST_MESSAGE_FREQUENCY GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 1)
 
-
-/**
- * Information about a benchmarking partner
- */
-struct BenchmarkPartner
-{
-  /**
-   * The peer itself this partner belongs to
-   */
-  struct BenchmarkPeer *me;
-
-  /**
-   * The partner peer
-   */
-  struct BenchmarkPeer *dest;
-
-  /**
-   * Core transmit handles
-   */
-  void *cth;
-
-  /**
-   * Number of messages sent to this partner
-   */
-  unsigned int messages_sent;
-
-  /**
-   * Number of bytes sent to this partner
-   */
-  unsigned int bytes_sent;
-
-  /**
-     * Number of messages received from this partner
-     */
-  unsigned int messages_received;
-
-  /**
-   * Number of bytes received from this partner
-   */
-  unsigned int bytes_received;
-};
-
-
-struct MasterInformation
-{
-  int core_slave_connections;
-
-  /**
-   * Testbed connect operation
-   */
-  struct TestbedConnectOperation *core_connect_ops;
-};
-
-
 /**
  * Connect peers with testbed
  */
@@ -115,108 +61,34 @@ struct TestbedConnectOperation
   struct GNUNET_TESTBED_Operation *connect_op;
 };
 
-
-
 /**
- * Information we track for a peer in the testbed.
+ * Overall state of the performance benchmark
  */
-struct BenchmarkPeer
+struct BenchmarkState
 {
   /**
-   * Handle with testbed.
+   * Are we connected to ATS service of all peers: GNUNET_YES/NO
    */
-  struct GNUNET_TESTBED_Peer *peer;
-
-  /**
-   * Unique identifier
-   */
-  int no;
-
-  /**
-   * Is this peer a measter: GNUNET_YES/GNUNET_NO
-   */
-  int master;
-
-  /**
-   *  Peer ID
-   */
-  struct GNUNET_PeerIdentity id;
-
-  /**
-   * Testbed operation to get peer information
-   */
-  struct GNUNET_TESTBED_Operation *peer_id_op;
-
-  /**
-   * Testbed operation to connect to ATS performance service
-   */
-  struct GNUNET_TESTBED_Operation *ats_perf_op;
-
-  /**
-   * Testbed operation to connect to core
-   */
-  struct GNUNET_TESTBED_Operation *core_op;
-
-  /**
-   * ATS performance handle
-   */
-  struct GNUNET_ATS_PerformanceHandle *ats_perf_handle;
-
-  /**
-   *  Core handle
-   */
-  struct GNUNET_CORE_Handle *ch;
-
-  /**
-   * Array of partners with num_slaves entries (if master) or
-   * num_master entries (if slave)
-   */
-  struct BenchmarkPartner *partners;
-
-  int core_connections;
-
-  struct MasterInformation mi;
+  int connected_ATS_service;
 
   /**
-   * Total number of messages this peer has sent
+   * Are we connected to CORE service of all peers: GNUNET_YES/NO
    */
-  unsigned int total_messages_sent;
+  int connected_COMM_service;
 
   /**
-   * Total number of bytes this peer has sent
+   * Are we connected to all peers: GNUNET_YES/NO
    */
-  unsigned int total_bytes_sent;
+  int connected_PEERS;
 
   /**
-   * Total number of messages this peer has received
+   * Are we connected to all slave peers on CORE level: GNUNET_YES/NO
    */
-  unsigned int total_messages_received;
+  int connected_CORE;
 
   /**
-   * Total number of bytes this peer has received
+   * Are we connected to CORE service of all peers: GNUNET_YES/NO
    */
-  unsigned int total_bytes_received;
-};
-
-
-/**
- * Overall state of the performance benchmark
- */
-struct BenchmarkState
-{
-  /* Are we connected to ATS service of all peers: GNUNET_YES/NO */
-  int connected_ATS_service;
-
-  /* Are we connected to CORE service of all peers: GNUNET_YES/NO */
-  int connected_CORE_service;
-
-  /* Are we connected to all peers: GNUNET_YES/NO */
-  int connected_PEERS;
-
-  /* Are we connected to all slave peers on CORE level: GNUNET_YES/NO */
-  int connected_CORE;
-
-  /* Are we connected to CORE service of all peers: GNUNET_YES/NO */
   int benchmarking;
 };
 
@@ -226,16 +98,35 @@ struct BenchmarkState
  */
 static GNUNET_SCHEDULER_TaskIdentifier shutdown_task;
 
+/**
+ * Progress task
+ */
+static GNUNET_SCHEDULER_TaskIdentifier progress_task;
+
 /**
  * Test result
  */
 static int result;
 
+/**
+ * Test result logging
+ */
+static int logging;
+
+/**Test core (GNUNET_YES) or transport (GNUNET_NO)
+ */
+static int test_core;
+
 /**
  * Solver string
  */
 static char *solver;
 
+/**
+ * Preference string
+ */
+static char *testname;
+
 /**
  * Preference string
  */
@@ -249,28 +140,38 @@ static int pref_val;
 /**
  * Number master peers
  */
-static int num_masters;
+static unsigned int num_masters;
 
 /**
  * Array of master peers
  */
-struct BenchmarkPeer *mps;
+static struct BenchmarkPeer *mps;
 
 /**
  * Number slave peers
  */
-static int num_slaves;
+static unsigned int num_slaves;
+
 /**
  * Array of slave peers
  */
-struct BenchmarkPeer *sps;
+static struct BenchmarkPeer *sps;
+
+/**
+ * Benchmark duration
+ */
+static struct GNUNET_TIME_Relative perf_duration;
+
+/**
+ * Logging frequency
+ */
+static struct GNUNET_TIME_Relative log_frequency;
 
 /**
  * Benchmark state
  */
 static struct BenchmarkState state;
 
-
 static void
 evaluate ()
 {
@@ -278,32 +179,39 @@ evaluate ()
   int c_s;
   unsigned int duration;
   struct BenchmarkPeer *mp;
+  struct BenchmarkPartner *p;
 
-  duration = (BENCHMARK_DURATION.rel_value_us / (1000 * 1000));
+  duration = (perf_duration.rel_value_us / (1000 * 1000));
   for (c_m = 0; c_m < num_masters; c_m++)
   {
     mp = &mps[c_m];
-    fprintf (stderr, _("Master [%u]: sent: %u KiB in %u sec. = %u KiB/s, received: %u KiB in %u sec. = %u KiB/s\n"),
-        mp->no,
-        mp->total_bytes_sent / 1024,
-        duration,
-        (mp->total_bytes_sent / 1024) / duration ,
-        mp->total_bytes_received / 1024,
-        duration,
+    fprintf (stderr,
+        _("Master [%u]: sent: %u KiB in %u sec. = %u KiB/s, received: %u KiB in %u sec. = %u KiB/s\n"),
+        mp->no, mp->total_bytes_sent / 1024, duration,
+        (mp->total_bytes_sent / 1024) / duration,
+        mp->total_bytes_received / 1024, duration,
         (mp->total_bytes_received / 1024) / duration);
 
-    for (c_s = 0; c_s < num_slaves; c_s ++)
+    for (c_s = 0; c_s < num_slaves; c_s++)
     {
-      fprintf (stderr, "Master [%u] -> Slave [%u]: sent %u KiB/s, received %u KiB/s \n",
-          mp->no,
-          mp->partners[c_s].dest->no,
-          (mp->partners[c_s].bytes_sent / 1024) / duration,
-          (mp->partners[c_s].bytes_received / 1024) / duration);
+      p = &mp->partners[c_s];
+      fprintf (stderr,
+          "%c Master [%u] -> Slave [%u]: sent %u KiB/s (%.2f %%), received %u KiB/s (%.2f %%)\n",
+          (mp->pref_partner == p->dest) ? '*' : ' ',
+          mp->no, p->dest->no,
+          (p->bytes_sent / 1024) / duration,
+          ((double) p->bytes_sent * 100) / mp->total_bytes_sent,
+          (p->bytes_received / 1024) / duration,
+          ((double) p->bytes_received * 100) / mp->total_bytes_received );
+      fprintf (stderr,
+          "%c Master [%u] -> Slave [%u]: Average application layer RTT: %u ms\n",
+          (mp->pref_partner == p->dest) ? '*' : ' ',
+          mp->no, p->dest->no,
+          p->total_app_rtt / (1000 * p->messages_sent));
     }
   }
 }
 
-
 /**
  * Shutdown nicely
  *
@@ -316,88 +224,114 @@ do_shutdown (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
   int c_m;
   int c_s;
   int c_op;
+  struct BenchmarkPeer *p;
+
+  if (GNUNET_YES == logging)
+    perf_logging_stop();
 
   shutdown_task = GNUNET_SCHEDULER_NO_TASK;
-  evaluate();
+  if (GNUNET_SCHEDULER_NO_TASK != progress_task)
+  {
+    fprintf (stderr, "0\n");
+    GNUNET_SCHEDULER_cancel (progress_task);
+  }
+  progress_task = GNUNET_SCHEDULER_NO_TASK;
+
+  evaluate ();
   state.benchmarking = GNUNET_NO;
   GNUNET_log(GNUNET_ERROR_TYPE_INFO, _("Benchmarking done\n"));
 
   for (c_m = 0; c_m < num_masters; c_m++)
   {
+    p = &mps[c_m];
     if (NULL != mps[c_m].peer_id_op)
     {
-      GNUNET_TESTBED_operation_done (mps[c_m].peer_id_op);
-      mps[c_m].peer_id_op = NULL;
+      GNUNET_TESTBED_operation_done (p->peer_id_op);
+      p->peer_id_op = NULL;
     }
 
-    for (c_op = 0; c_op < num_slaves; c_op++)
-    {
+    if (GNUNET_SCHEDULER_NO_TASK != p->ats_task)
+      GNUNET_SCHEDULER_cancel (p->ats_task);
+    p->ats_task = GNUNET_SCHEDULER_NO_TASK;
 
-      if (NULL != mps[c_m].partners[c_op].cth)
+    for (c_op = 0; c_op < p->num_partners; c_op++)
+    {
+      if (NULL != p->partners[c_op].cth)
       {
-        GNUNET_CORE_notify_transmit_ready_cancel (mps[c_m].partners[c_op].cth);
-        mps[c_m].partners[c_op].cth = NULL;
+        GNUNET_CORE_notify_transmit_ready_cancel (p->partners[c_op].cth);
+        p->partners[c_op].cth = NULL;
       }
-
-      if (NULL != mps[c_m].mi.core_connect_ops[c_op].connect_op)
+      if (NULL != p->partners[c_op].tth)
+      {
+        GNUNET_TRANSPORT_notify_transmit_ready_cancel (p->partners[c_op].tth);
+        p->partners[c_op].tth = NULL;
+      }
+      if (NULL != p->core_connect_ops[c_op].connect_op)
       {
         GNUNET_log(GNUNET_ERROR_TYPE_INFO,
             _("Failed to connect peer 0 and %u\n"), c_op);
         GNUNET_TESTBED_operation_done (
-            mps[c_m].mi.core_connect_ops[c_op].connect_op);
-        mps[c_m].mi.core_connect_ops[c_op].connect_op = NULL;
+            p->core_connect_ops[c_op].connect_op);
+        p->core_connect_ops[c_op].connect_op = NULL;
         result = 1;
       }
     }
 
-    if (NULL != mps[c_m].ats_perf_op)
+    if (NULL != p->ats_perf_op)
     {
-      GNUNET_TESTBED_operation_done (mps[c_m].ats_perf_op);
-      mps[c_m].ats_perf_op = NULL;
+      GNUNET_TESTBED_operation_done (p->ats_perf_op);
+      p->ats_perf_op = NULL;
     }
 
-    if (NULL != mps[c_m].core_op)
+    if (NULL != p->comm_op)
     {
-      GNUNET_TESTBED_operation_done (mps[c_m].core_op);
-      mps[c_m].core_op = NULL;
+      GNUNET_TESTBED_operation_done (p->comm_op);
+      p->comm_op = NULL;
     }
+    GNUNET_free(p->core_connect_ops);
+    GNUNET_free(p->partners);
+    p->partners = NULL;
   }
 
-
   for (c_s = 0; c_s < num_slaves; c_s++)
   {
-    if (NULL != sps[c_s].peer_id_op)
+    p = &sps[c_s];
+    if (NULL != p->peer_id_op)
     {
-      GNUNET_TESTBED_operation_done (sps[c_s].peer_id_op);
-      sps[c_s].peer_id_op = NULL;
+      GNUNET_TESTBED_operation_done (p->peer_id_op);
+      p->peer_id_op = NULL;
     }
 
-    for (c_op = 0; c_op < num_slaves; c_op++)
+    for (c_op = 0; c_op < p->num_partners; c_op++)
     {
-      if (NULL != sps[c_s].partners[c_op].cth)
+      if (NULL != p->partners[c_op].cth)
+      {
+        GNUNET_CORE_notify_transmit_ready_cancel (p->partners[c_op].cth);
+        p->partners[c_op].cth = NULL;
+      }
+      if (NULL != p->partners[c_op].tth)
       {
-        GNUNET_CORE_notify_transmit_ready_cancel (sps[c_s].partners[c_op].cth);
-        sps[c_s].partners[c_op].cth = NULL;
+        GNUNET_TRANSPORT_notify_transmit_ready_cancel (p->partners[c_op].tth);
+        p->partners[c_op].tth = NULL;
       }
     }
-
-
-    if (NULL != sps[c_s].ats_perf_op)
+    if (NULL != p->ats_perf_op)
     {
-      GNUNET_TESTBED_operation_done (sps[c_s].ats_perf_op);
-      sps[c_s].ats_perf_op = NULL;
+      GNUNET_TESTBED_operation_done (p->ats_perf_op);
+      p->ats_perf_op = NULL;
     }
-    if (NULL != sps[c_s].core_op)
+    if (NULL != p->comm_op)
     {
-      GNUNET_TESTBED_operation_done (sps[c_s].core_op);
-      sps[c_s].core_op = NULL;
+      GNUNET_TESTBED_operation_done (p->comm_op);
+      p->comm_op = NULL;
     }
+    GNUNET_free(p->partners);
+    p->partners = NULL;
   }
 
   GNUNET_SCHEDULER_shutdown ();
 }
 
-
 static struct BenchmarkPeer *
 find_peer (const struct GNUNET_PeerIdentity * peer)
 {
@@ -405,22 +339,18 @@ find_peer (const struct GNUNET_PeerIdentity * peer)
 
   for (c_p = 0; c_p < num_masters; c_p++)
   {
-    if (0
-        == memcmp (&mps[c_p].id, peer,
-            sizeof(struct GNUNET_PeerIdentity)))
+    if (0 == memcmp (&mps[c_p].id, peer, sizeof(struct GNUNET_PeerIdentity)))
       return &mps[c_p];
   }
 
   for (c_p = 0; c_p < num_slaves; c_p++)
   {
-    if (0 == memcmp (&sps[c_p].id, peer,
-            sizeof(struct GNUNET_PeerIdentity)))
+    if (0 == memcmp (&sps[c_p].id, peer, sizeof(struct GNUNET_PeerIdentity)))
       return &sps[c_p];
   }
-  return NULL;
+  return NULL ;
 }
 
-
 /**
  * Controller event callback
  *
@@ -446,22 +376,36 @@ controller_event_cb (void *cls,
   }
 }
 
-
 static size_t
-core_send_ready (void *cls, size_t size, void *buf)
+comm_send_ready (void *cls, size_t size, void *buf)
 {
   static char msgbuf[TEST_MESSAGE_SIZE];
-  struct BenchmarkPartner *partner = cls;
+  struct BenchmarkPartner *p = cls;
   struct GNUNET_MessageHeader *msg;
 
-  partner->cth = NULL;
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Master [%u]: Sending PING to [%u]\n",
-        partner->me->no, partner->dest->no);
+  if (GNUNET_YES == test_core)
+    p->cth = NULL;
+  else
+    p->tth = NULL;
+
+  if (NULL == buf)
+  {
+    GNUNET_break (0);
+    return 0;
+  }
+  if (size < TEST_MESSAGE_SIZE)
+  {
+    GNUNET_break (0);
+    return 0;
+  }
 
-  partner->messages_sent ++;
-  partner->bytes_sent += TEST_MESSAGE_SIZE;
-  partner->me->total_messages_sent ++;
-  partner->me->total_bytes_sent += TEST_MESSAGE_SIZE;
+  GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "Master [%u]: Sending PING to [%u]\n",
+      p->me->no, p->dest->no);
+
+  p->messages_sent++;
+  p->bytes_sent += TEST_MESSAGE_SIZE;
+  p->me->total_messages_sent++;
+  p->me->total_bytes_sent += TEST_MESSAGE_SIZE;
 
   msg = (struct GNUNET_MessageHeader *) &msgbuf;
   memset (&msgbuf, 'a', TEST_MESSAGE_SIZE);
@@ -471,6 +415,55 @@ core_send_ready (void *cls, size_t size, void *buf)
   return TEST_MESSAGE_SIZE;
 }
 
+static void
+comm_schedule_send (struct BenchmarkPartner *p)
+{
+  p->last_message_sent = GNUNET_TIME_absolute_get();
+  if (GNUNET_YES == test_core)
+  {
+    p->cth = GNUNET_CORE_notify_transmit_ready (
+      p->me->ch, GNUNET_NO, 0, GNUNET_TIME_UNIT_MINUTES, &p->dest->id,
+      TEST_MESSAGE_SIZE, &comm_send_ready, p);
+  }
+  else
+  {
+    p->tth = GNUNET_TRANSPORT_notify_transmit_ready (
+      p->me->th, &p->dest->id, TEST_MESSAGE_SIZE, 0,GNUNET_TIME_UNIT_MINUTES,
+      &comm_send_ready, p);
+  }
+
+}
+
+static void
+print_progress ()
+{
+  static int calls;
+  progress_task = GNUNET_SCHEDULER_NO_TASK;
+
+  fprintf (stderr, "%llu..",
+      (long long unsigned) perf_duration.rel_value_us / (1000 * 1000) - calls);
+  calls++;
+
+  progress_task = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS,
+      &print_progress, NULL );
+}
+
+static void
+ats_pref_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+{
+  struct BenchmarkPeer *me = cls;
+
+  me->ats_task = GNUNET_SCHEDULER_NO_TASK;
+
+  GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, " Master [%u] set preference for slave [%u] to %f\n",
+      me->no, me->pref_partner->no, me->pref_value);
+  GNUNET_ATS_performance_change_preference (me->ats_perf_handle,
+      &me->pref_partner->id,
+      pref_val, me->pref_value, GNUNET_ATS_PREFERENCE_END);
+  me->pref_value += TEST_ATS_PREFRENCE_DELTA;
+  me->ats_task = GNUNET_SCHEDULER_add_delayed (TEST_ATS_PREFRENCE_FREQUENCY,
+      &ats_pref_task, cls);
+}
 
 static void
 do_benchmark ()
@@ -479,7 +472,7 @@ do_benchmark ()
   int c_s;
 
   if ((state.connected_ATS_service == GNUNET_NO)
-      || (state.connected_CORE_service == GNUNET_NO)
+      || (state.connected_COMM_service == GNUNET_NO)
       || (state.connected_PEERS == GNUNET_NO)
       || (state.connected_CORE == GNUNET_NO))
     return;
@@ -489,33 +482,23 @@ do_benchmark ()
 
   if (GNUNET_SCHEDULER_NO_TASK != shutdown_task)
     GNUNET_SCHEDULER_cancel (shutdown_task);
-  shutdown_task = GNUNET_SCHEDULER_add_delayed (BENCHMARK_DURATION,
+  shutdown_task = GNUNET_SCHEDULER_add_delayed (perf_duration,
       &do_shutdown, NULL );
 
-  /* Start sending test messages */
-  for (c_s = 0; c_s < num_slaves; c_s++)
-  {
-    for (c_m = 0; c_m < num_masters; c_m++)
-    {
-
-      sps[c_s].partners[c_m].me = &sps[c_s];
-      sps[c_s].partners[c_m].dest = &mps[c_m];
-    }
-  }
+  progress_task = GNUNET_SCHEDULER_add_now (&print_progress, NULL );
 
+  /* Start sending test messages */
   for (c_m = 0; c_m < num_masters; c_m++)
   {
     for (c_s = 0; c_s < num_slaves; c_s++)
-    {
-      mps[c_m].partners[c_s].me = &mps[c_m];
-      mps[c_m].partners[c_s].dest = &sps[c_s];
-      mps[c_m].partners[c_s].cth = GNUNET_CORE_notify_transmit_ready (mps[c_m].ch,
-          GNUNET_NO, 0, GNUNET_TIME_UNIT_MINUTES, &sps[c_s].id,
-          TEST_MESSAGE_SIZE, &core_send_ready, &mps[c_m].partners[c_s]);
-    }
+      comm_schedule_send (&mps[c_m].partners[c_s]);
+    if (pref_val != GNUNET_ATS_PREFERENCE_END)
+      mps[c_m].ats_task = GNUNET_SCHEDULER_add_now (&ats_pref_task, &mps[c_m]);
   }
-}
+  if (GNUNET_YES == logging)
+    perf_logging_start (log_frequency, testname, mps, num_masters);
 
+}
 
 static void
 connect_completion_callback (void *cls, struct GNUNET_TESTBED_Operation *op,
@@ -527,14 +510,14 @@ connect_completion_callback (void *cls, struct GNUNET_TESTBED_Operation *op,
   if (NULL == emsg)
   {
     GNUNET_log(GNUNET_ERROR_TYPE_INFO,
-        _("Connected master peer %u with peer %u\n"), cop->master->no,
+        _("Connected master [%u] with slave [%u]\n"), cop->master->no,
         cop->slave->no);
   }
   else
   {
     GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
-        _("Failed to connect master peer%u with peer %u\n"), cop->master->no,
-        cop->slave->no);
+        _("Failed to connect master peer [%u] with slave [%u]\n"),
+        cop->master->no, cop->slave->no);
     GNUNET_break(0);
     if (GNUNET_SCHEDULER_NO_TASK != shutdown_task)
       GNUNET_SCHEDULER_cancel (shutdown_task);
@@ -544,8 +527,8 @@ connect_completion_callback (void *cls, struct GNUNET_TESTBED_Operation *op,
   ops++;
   for (c = 0; c < num_slaves; c++)
   {
-    if (cop == &cop->master->mi.core_connect_ops[c])
-      cop->master->mi.core_connect_ops[c].connect_op = NULL;
+    if (cop == &cop->master->core_connect_ops[c])
+      cop->master->core_connect_ops[c].connect_op = NULL;
   }
   if (ops == num_masters * num_slaves)
   {
@@ -554,7 +537,6 @@ connect_completion_callback (void *cls, struct GNUNET_TESTBED_Operation *op,
   }
 }
 
-
 static void
 do_connect_peers (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
@@ -563,33 +545,30 @@ do_connect_peers (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
   struct BenchmarkPeer *p;
 
   if ((state.connected_ATS_service == GNUNET_NO)
-      || (state.connected_CORE_service == GNUNET_NO))
-  {
+      || (state.connected_COMM_service == GNUNET_NO))
     return;
-  }
 
   GNUNET_log(GNUNET_ERROR_TYPE_INFO, _("Connecting peers on CORE level\n"));
 
   for (c_m = 0; c_m < num_masters; c_m++)
   {
     p = &mps[c_m];
-    p->mi.core_connect_ops = GNUNET_malloc (num_slaves *
+    p->core_connect_ops = GNUNET_malloc (num_slaves *
         sizeof (struct TestbedConnectOperation));
 
     for (c_s = 0; c_s < num_slaves; c_s++)
     {
       GNUNET_log(GNUNET_ERROR_TYPE_INFO,
-          _("Connecting master peer %u with slave peer %u\n"), p->no,
-          sps[c_s].no);
-      p->mi.core_connect_ops[c_s].master = p;
-      p->mi.core_connect_ops[c_s].slave = &sps[c_s];
-      p->mi.core_connect_ops[c_s].connect_op = GNUNET_TESTBED_overlay_connect (NULL,
-          &connect_completion_callback, &p->mi.core_connect_ops[c_s],
+          _("Connecting master [%u] with slave [%u]\n"), p->no, sps[c_s].no);
+      p->core_connect_ops[c_s].master = p;
+      p->core_connect_ops[c_s].slave = &sps[c_s];
+      p->core_connect_ops[c_s].connect_op = GNUNET_TESTBED_overlay_connect (
+          NULL, &connect_completion_callback, &p->core_connect_ops[c_s],
           sps[c_s].peer, p->peer);
-      if (NULL == p->mi.core_connect_ops[c_s].connect_op)
+      if (NULL == p->core_connect_ops[c_s].connect_op)
       {
         GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
-            _("Could not connect master peer %u and slave peer %u\n"), p->no,
+            _("Could not connect master [%u] and slave [%u]\n"), p->no,
             sps[c_s].no);
         GNUNET_break(0);
         if (GNUNET_SCHEDULER_NO_TASK != shutdown_task)
@@ -601,7 +580,6 @@ do_connect_peers (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
   }
 }
 
-
 /**
  * Method called whenever a given peer connects.
  *
@@ -609,41 +587,42 @@ do_connect_peers (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
  * @param peer peer identity this notification is about
  */
 static void
-core_connect_cb (void *cls, const struct GNUNET_PeerIdentity * peer)
+comm_connect_cb (void *cls, const struct GNUNET_PeerIdentity * peer)
 {
-  struct BenchmarkPeer *p = cls;
-  struct BenchmarkPeer *t;
+  struct BenchmarkPeer *me = cls;
+  struct BenchmarkPeer *remote;
   char *id;
   int c;
   int completed;
 
-  t = find_peer (peer);
-  if (NULL == t)
+  remote = find_peer (peer);
+  if (NULL == remote)
   {
     GNUNET_break(0);
     return;
   }
 
-  id = GNUNET_strdup (GNUNET_i2s (&p->id));
-  GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "%s %s connected to %s %s\n",
-      (p->master == GNUNET_YES) ? "Master": "Slave", id,
-      (t->master == GNUNET_YES) ? "Master": "Slave", GNUNET_i2s (peer));
+  id = GNUNET_strdup (GNUNET_i2s (&me->id));
+  GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "%s [%u] `%s' connected to %s [%u] %s\n",
+      (me->master == GNUNET_YES) ? "Master": "Slave", me->no, id,
+      (remote->master == GNUNET_YES) ? "Master": "Slave", remote->no,
+      GNUNET_i2s (peer));
 
-  p->core_connections++;
-  if ((GNUNET_YES == p->master) && (GNUNET_NO == t->master)
+  me->core_connections++;
+  if ((GNUNET_YES == me->master) && (GNUNET_NO == remote->master)
       && (GNUNET_NO == state.connected_CORE))
   {
-    p->mi.core_slave_connections++;
+    me->core_slave_connections++;
 
-    if (p->mi.core_slave_connections == num_slaves)
+    if (me->core_slave_connections == num_slaves)
     {
-      GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Master %u connected all slaves\n",
-          p->no);
+      GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Master [%u] connected all slaves\n",
+          me->no);
     }
     completed = GNUNET_YES;
     for (c = 0; c < num_masters; c++)
     {
-      if (mps[c].mi.core_slave_connections != num_slaves)
+      if (mps[c].core_slave_connections != num_slaves)
         completed = GNUNET_NO;
     }
     if (GNUNET_YES == completed)
@@ -658,48 +637,76 @@ core_connect_cb (void *cls, const struct GNUNET_PeerIdentity * peer)
   GNUNET_free(id);
 }
 
+static struct BenchmarkPartner *
+find_partner (struct BenchmarkPeer *me, const struct GNUNET_PeerIdentity * peer)
+{
+  int c_m;
+  GNUNET_assert (NULL != me);
+  GNUNET_assert (NULL != peer);
+
+  for (c_m = 0; c_m < me->num_partners; c_m++)
+  {
+    /* Find a partner with other as destination */
+    if (0 == memcmp (peer, &me->partners[c_m].dest->id,
+            sizeof(struct GNUNET_PeerIdentity)))
+    {
+      return &me->partners[c_m];
+    }
+  }
+
+  return NULL;
+}
 
 static void
-core_disconnect_cb (void *cls, const struct GNUNET_PeerIdentity * peer)
+comm_disconnect_cb (void *cls, const struct GNUNET_PeerIdentity * peer)
 {
-  struct BenchmarkPeer *p = cls;
-  struct BenchmarkPeer *t;
+  struct BenchmarkPeer *me = cls;
+  struct BenchmarkPartner *p;
   char *id;
 
-  t = find_peer (peer);
-  if (NULL == t)
-  {
-    GNUNET_break(0);
+  if (NULL == (p = find_partner (me, peer)))
     return;
-  }
 
-  id = GNUNET_strdup (GNUNET_i2s (&p->id));
+  id = GNUNET_strdup (GNUNET_i2s (&me->id));
   GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "%s disconnected from %s \n", id,
       GNUNET_i2s (peer));
-  GNUNET_assert(p->core_connections > 0);
-  p->core_connections--;
+  GNUNET_assert(me->core_connections > 0);
+  me->core_connections--;
 
   if ((GNUNET_YES == state.benchmarking)
-      && ((GNUNET_YES == p->master) || (GNUNET_YES == t->master)))
+      && ((GNUNET_YES == me->master) || (GNUNET_YES == p->dest->master)))
   {
     GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
         "%s disconnected from %s while benchmarking \n", id, GNUNET_i2s (peer));
+    if (NULL != p->tth)
+    {
+      GNUNET_TRANSPORT_notify_transmit_ready_cancel (p->tth);
+      p->tth = NULL;
+    }
+    if (NULL != p->cth)
+    {
+      GNUNET_CORE_notify_transmit_ready_cancel (p->cth);
+      p->cth = NULL;
+    }
   }
   GNUNET_free(id);
 }
 
 static size_t
-core_send_echo_ready (void *cls, size_t size, void *buf)
+comm_send_pong_ready (void *cls, size_t size, void *buf)
 {
   static char msgbuf[TEST_MESSAGE_SIZE];
   struct BenchmarkPartner *p = cls;
   struct GNUNET_MessageHeader *msg;
 
-  p->cth = NULL;
+  if (GNUNET_YES == test_core)
+    p->cth = NULL;
+  else
+    p->tth = NULL;
 
-  p->messages_sent ++;
+  p->messages_sent++;
   p->bytes_sent += TEST_MESSAGE_SIZE;
-  p->me->total_messages_sent ++;
+  p->me->total_messages_sent++;
   p->me->total_bytes_sent += TEST_MESSAGE_SIZE;
 
   msg = (struct GNUNET_MessageHeader *) &msgbuf;
@@ -711,116 +718,104 @@ core_send_echo_ready (void *cls, size_t size, void *buf)
   return TEST_MESSAGE_SIZE;
 }
 
-
 static int
-core_handle_ping (void *cls, const struct GNUNET_PeerIdentity *other,
+comm_handle_ping (void *cls, const struct GNUNET_PeerIdentity *other,
     const struct GNUNET_MessageHeader *message)
 {
-  int c_m;
+
   struct BenchmarkPeer *me = cls;
   struct BenchmarkPartner *p = NULL;
-  for (c_m = 0; c_m < num_masters; c_m++)
-  {
-    if (0 == memcmp (other, &me->partners[c_m].dest->id, sizeof (struct GNUNET_PeerIdentity)))
-    {
-      p = &me->partners[c_m];
-      break;
-    }
-  }
-  if (NULL == p)
+
+  if (NULL == (p = find_partner(me, other)))
   {
-    GNUNET_break (0);
+    GNUNET_break(0);
     return GNUNET_SYSERR;
   }
-  GNUNET_assert (NULL == p->cth);
 
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-      "Slave [%u]: Received PING from [%u], sending PONG\n",
-      me->no, p->dest->no);
+  GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
+      "Slave [%u]: Received PING from [%u], sending PONG\n", me->no,
+      p->dest->no);
 
-  p->messages_received ++;
+  p->messages_received++;
   p->bytes_received += TEST_MESSAGE_SIZE;
-  p->me->total_messages_received ++;
+  p->me->total_messages_received++;
   p->me->total_bytes_received += TEST_MESSAGE_SIZE;
 
-  p->cth = GNUNET_CORE_notify_transmit_ready (me->ch, GNUNET_NO, 0,
-      GNUNET_TIME_UNIT_MINUTES, &p->dest->id, TEST_MESSAGE_SIZE,
-      &core_send_echo_ready, p);
+  if (GNUNET_YES == test_core)
+  {
+    GNUNET_assert (NULL == p->cth);
+    p->cth = GNUNET_CORE_notify_transmit_ready (me->ch, GNUNET_NO, 0,
+        GNUNET_TIME_UNIT_MINUTES, &p->dest->id, TEST_MESSAGE_SIZE,
+        &comm_send_pong_ready, p);
+  }
+  else
+  {
+    GNUNET_assert (NULL == p->tth);
+    p->tth = GNUNET_TRANSPORT_notify_transmit_ready (me->th, &p->dest->id,
+        TEST_MESSAGE_SIZE, 0, GNUNET_TIME_UNIT_MINUTES, &comm_send_pong_ready,
+        p);
+  }
   return GNUNET_OK;
 }
 
 static int
-core_handle_pong (void *cls, const struct GNUNET_PeerIdentity *other,
+comm_handle_pong (void *cls, const struct GNUNET_PeerIdentity *other,
     const struct GNUNET_MessageHeader *message)
 {
-  int c_s;
   struct BenchmarkPeer *me = cls;
   struct BenchmarkPartner *p = NULL;
 
-  for (c_s = 0; c_s < num_slaves; c_s++)
-  {
-    if (0 == memcmp (other, &me->partners[c_s].dest->id, sizeof (struct GNUNET_PeerIdentity)))
-    {
-      p = &me->partners[c_s];
-      break;
-    }
-  }
-  if (NULL == p)
+  if (NULL == (p = find_partner (me, other)))
   {
-    GNUNET_break (0);
+    GNUNET_break(0);
     return GNUNET_SYSERR;
   }
-  GNUNET_assert (NULL == p->cth);
 
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-      "Master [%u]: Received PONG from [%u], next message\n",
-      me->no, p->dest->no);
+  GNUNET_log(GNUNET_ERROR_TYPE_DEBUG,
+      "Master [%u]: Received PONG from [%u], next message\n", me->no,
+      p->dest->no);
 
-  p->messages_received ++;
+  p->messages_received++;
   p->bytes_received += TEST_MESSAGE_SIZE;
-  p->me->total_messages_received ++;
+  p->me->total_messages_received++;
   p->me->total_bytes_received += TEST_MESSAGE_SIZE;
+  p->total_app_rtt += GNUNET_TIME_absolute_get_difference(p->last_message_sent,
+      GNUNET_TIME_absolute_get()).rel_value_us;
 
-  p->cth = GNUNET_CORE_notify_transmit_ready (me->ch,
-            GNUNET_NO, 0, GNUNET_TIME_UNIT_MINUTES, &p->dest->id,
-            TEST_MESSAGE_SIZE, &core_send_ready, p);
-
+  comm_schedule_send (p);
   return GNUNET_OK;
 }
 
 static void *
 core_connect_adapter (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg)
 {
-  struct BenchmarkPeer *peer = cls;
+  struct BenchmarkPeer *me = cls;
 
-  static const struct GNUNET_CORE_MessageHandler handlers[] = {
-      {&core_handle_ping, TEST_MESSAGE_TYPE_PING, 0 },
-      {&core_handle_pong, TEST_MESSAGE_TYPE_PONG, 0 },
-      { NULL, 0, 0 } };
+  static const struct GNUNET_CORE_MessageHandler handlers[] = { {
+      &comm_handle_ping, TEST_MESSAGE_TYPE_PING, 0 }, { &comm_handle_pong,
+      TEST_MESSAGE_TYPE_PONG, 0 }, { NULL, 0, 0 } };
 
-  peer->ch = GNUNET_CORE_connect (cfg, peer, NULL, core_connect_cb,
-      core_disconnect_cb, NULL, GNUNET_NO, NULL, GNUNET_NO, handlers);
-  if (NULL == peer->ch)
+  me->ch = GNUNET_CORE_connect (cfg, me, NULL, comm_connect_cb,
+      comm_disconnect_cb, NULL, GNUNET_NO, NULL, GNUNET_NO, handlers);
+  if (NULL == me->ch)
     GNUNET_log(GNUNET_ERROR_TYPE_ERROR, "Failed to create core connection \n");
-  return peer->ch;
+  return me->ch;
 }
 
-
 static void
 core_disconnect_adapter (void *cls, void *op_result)
 {
-  struct BenchmarkPeer *peer = cls;
+  struct BenchmarkPeer *me = cls;
 
-  GNUNET_CORE_disconnect (peer->ch);
-  peer->ch = NULL;
+  GNUNET_CORE_disconnect (me->ch);
+  me->ch = NULL;
 }
 
-
 static void
-core_connect_completion_cb (void *cls, struct GNUNET_TESTBED_Operation *op,
+comm_connect_completion_cb (void *cls, struct GNUNET_TESTBED_Operation *op,
     void *ca_result, const char *emsg)
 {
-  static int core_done = 0;
+  static int comm_done = 0;
   if ((NULL != emsg) || (NULL == ca_result))
   {
     GNUNET_log(GNUNET_ERROR_TYPE_INFO, _("Initialization failed, shutdown\n"));
@@ -830,35 +825,90 @@ core_connect_completion_cb (void *cls, struct GNUNET_TESTBED_Operation *op,
     shutdown_task = GNUNET_SCHEDULER_add_now (do_shutdown, NULL );
     return;
   }
-  core_done++;
+  comm_done++;
 
-  if (core_done == num_slaves + num_masters)
+  if (comm_done == num_slaves + num_masters)
   {
-    GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Connected to all CORE services\n");
-    state.connected_CORE_service = GNUNET_YES;
+    GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Connected to all %s services\n",
+        (GNUNET_YES == test_core) ? "CORE" : "TRANSPORT");
+    state.connected_COMM_service = GNUNET_YES;
     GNUNET_SCHEDULER_add_now (&do_connect_peers, NULL );
   }
 }
 
+static void
+transport_recv_cb (void *cls,
+                   const struct GNUNET_PeerIdentity * peer,
+                   const struct GNUNET_MessageHeader * message)
+{
+  if (TEST_MESSAGE_SIZE != ntohs (message->size) ||
+      (TEST_MESSAGE_TYPE_PING != ntohs (message->type) &&
+      TEST_MESSAGE_TYPE_PONG != ntohs (message->type)))
+  {
+    return;
+  }
+  if (TEST_MESSAGE_TYPE_PING == ntohs (message->type))
+    comm_handle_ping (cls, peer, message);
+
+  if (TEST_MESSAGE_TYPE_PONG == ntohs (message->type))
+    comm_handle_pong (cls, peer, message);
+}
+
+
+static void *
+transport_connect_adapter (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg)
+{
+  struct BenchmarkPeer *me = cls;
+
+  me->th = GNUNET_TRANSPORT_connect (cfg, &me->id, me,  &transport_recv_cb,
+      &comm_connect_cb, &comm_disconnect_cb);
+  if (NULL == me->th)
+    GNUNET_log(GNUNET_ERROR_TYPE_ERROR, "Failed to create transport connection \n");
+  return me->th;
+}
+
+static void
+transport_disconnect_adapter (void *cls, void *op_result)
+{
+  struct BenchmarkPeer *me = cls;
+
+  GNUNET_TRANSPORT_disconnect (me->th);
+  me->th = NULL;
+}
 
 static void
-do_connect_core (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+do_comm_connect (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
   int c_s;
   int c_m;
-  GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Connecting to all CORE services\n");
+  GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Connecting to all %s services\n",
+      (GNUNET_YES == test_core) ? "CORE" : "TRANSPORT");
   for (c_m = 0; c_m < num_masters; c_m++)
   {
-    mps[c_m].core_op = GNUNET_TESTBED_service_connect (NULL,
-        mps[c_m].peer, "core", core_connect_completion_cb, NULL,
-        &core_connect_adapter, &core_disconnect_adapter, &mps[c_m]);
+    if (GNUNET_YES == test_core)
+      mps[c_m].comm_op = GNUNET_TESTBED_service_connect (NULL, mps[c_m].peer,
+        "core", &comm_connect_completion_cb, NULL, &core_connect_adapter,
+        &core_disconnect_adapter, &mps[c_m]);
+    else
+    {
+      mps[c_m].comm_op = GNUNET_TESTBED_service_connect (NULL, mps[c_m].peer,
+        "transport", &comm_connect_completion_cb, NULL, &transport_connect_adapter,
+        &transport_disconnect_adapter, &mps[c_m]);
+    }
   }
 
   for (c_s = 0; c_s < num_slaves; c_s++)
   {
-    sps[c_s].core_op = GNUNET_TESTBED_service_connect (NULL,
-        sps[c_s].peer, "core", core_connect_completion_cb, NULL,
-        &core_connect_adapter, &core_disconnect_adapter, &sps[c_s]);
+    if (GNUNET_YES == test_core)
+      sps[c_s].comm_op = GNUNET_TESTBED_service_connect (NULL, sps[c_s].peer,
+        "core", &comm_connect_completion_cb, NULL, &core_connect_adapter,
+        &core_disconnect_adapter, &sps[c_s]);
+    else
+    {
+      sps[c_s].comm_op = GNUNET_TESTBED_service_connect (NULL, sps[c_s].peer,
+        "transport", &comm_connect_completion_cb, NULL, &transport_connect_adapter,
+        &transport_disconnect_adapter, &sps[c_s]);
+    }
   }
 }
 
@@ -868,59 +918,113 @@ ats_performance_info_cb (void *cls, const struct GNUNET_HELLO_Address *address,
     struct GNUNET_BANDWIDTH_Value32NBO bandwidth_in,
     const struct GNUNET_ATS_Information *ats, uint32_t ats_count)
 {
-  struct BenchmarkPeer *p = cls;
+  struct BenchmarkPeer *me = cls;
+  struct BenchmarkPartner *p;
   int c_a;
+  int log;
   char *peer_id;
 
-  peer_id = GNUNET_strdup (GNUNET_i2s (&p->id));
-  for (c_a = 0; c_a < ats_count; c_a++)
+  p = find_partner (me, &address->peer);
+  if (NULL == p)
   {
-    /*GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, _("%c %03u: %s %s %u\n"),
-     (GNUNET_YES == p->master) ? 'M' : 'S',
-     p->no,
-     GNUNET_i2s (&address->peer),
-     GNUNET_ATS_print_property_type(ntohl(ats[c_a].type)),
-     ntohl(ats[c_a].value));*/
+    /* This is not one of my partners
+     * Will happen since the peers will connect to each other due to gossiping
+     */
+    return;
   }
-#if 0
-  if ((GNUNET_YES == p->master)
-      && (0 == memcmp (&address->peer, &p->destination->id,
-                       sizeof(struct GNUNET_PeerIdentity))))
+  peer_id = GNUNET_strdup (GNUNET_i2s (&me->id));
+
+  log = GNUNET_NO;
+  if ((p->bandwidth_in != ntohl (bandwidth_in.value__)) ||
+      (p->bandwidth_out != ntohl (bandwidth_out.value__)))
+      log = GNUNET_YES;
+  p->bandwidth_in = ntohl (bandwidth_in.value__);
+  p->bandwidth_out = ntohl (bandwidth_out.value__);
+
+  for (c_a = 0; c_a < ats_count; c_a++)
   {
-    GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Bandwidth for master %u: %lu %lu\n",
-        p->no, (long unsigned int ) ntohl (bandwidth_in.value__),
-        (long unsigned int ) ntohl (bandwidth_in.value__));
-  }
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "%s [%u] received ATS information: %s %s %u\n",
+        (GNUNET_YES == p->me->master) ? "Master" : "Slave",
+        p->me->no,
+        GNUNET_i2s (&p->dest->id),
+        GNUNET_ATS_print_property_type(ntohl(ats[c_a].type)),
+        ntohl(ats[c_a].value));
+    switch (ntohl (ats[c_a].type ))
+    {
+      case GNUNET_ATS_ARRAY_TERMINATOR:
+        break;
+      case GNUNET_ATS_UTILIZATION_OUT:
+        if (p->ats_utilization_up != ntohl (ats[c_a].value))
+            log = GNUNET_YES;
+        p->ats_utilization_up = ntohl (ats[c_a].value);
 
-  store_information (&bp->id, address, address_active, bandwidth_in,
-      bandwidth_out, ats, ats_count);
-#endif
+        break;
+      case GNUNET_ATS_UTILIZATION_IN:
+        if (p->ats_utilization_down != ntohl (ats[c_a].value))
+            log = GNUNET_YES;
+        p->ats_utilization_down = ntohl (ats[c_a].value);
+        break;
+      case GNUNET_ATS_NETWORK_TYPE:
+        if (p->ats_network_type != ntohl (ats[c_a].value))
+            log = GNUNET_YES;
+        p->ats_network_type = ntohl (ats[c_a].value);
+        break;
+      case GNUNET_ATS_QUALITY_NET_DELAY:
+        if (p->ats_delay != ntohl (ats[c_a].value))
+            log = GNUNET_YES;
+        p->ats_delay = ntohl (ats[c_a].value);
+        break;
+      case GNUNET_ATS_QUALITY_NET_DISTANCE:
+        if (p->ats_distance != ntohl (ats[c_a].value))
+            log = GNUNET_YES;
+        p->ats_distance = ntohl (ats[c_a].value);
+        GNUNET_break (0);
+        break;
+      case GNUNET_ATS_COST_WAN:
+        if (p->ats_cost_wan != ntohl (ats[c_a].value))
+            log = GNUNET_YES;
+        p->ats_cost_wan = ntohl (ats[c_a].value);
+        break;
+      case GNUNET_ATS_COST_LAN:
+        if (p->ats_cost_lan != ntohl (ats[c_a].value))
+            log = GNUNET_YES;
+        p->ats_cost_lan = ntohl (ats[c_a].value);
+        break;
+      case GNUNET_ATS_COST_WLAN:
+        if (p->ats_cost_wlan != ntohl (ats[c_a].value))
+            log = GNUNET_YES;
+        p->ats_cost_wlan = ntohl (ats[c_a].value);
+        break;
+      default:
+        break;
+    }
+  }
+  if ((GNUNET_YES == logging) && (GNUNET_YES == log))
+    collect_log_now();
   GNUNET_free(peer_id);
 }
 
-
 static void *
 ats_perf_connect_adapter (void *cls,
     const struct GNUNET_CONFIGURATION_Handle *cfg)
 {
-  struct BenchmarkPeer *peer = cls;
+  struct BenchmarkPeer *me = cls;
 
-  peer->ats_perf_handle = GNUNET_ATS_performance_init (cfg, &ats_performance_info_cb,
-      peer);
-  if (NULL == peer->ats_perf_handle)
+  me->ats_perf_handle = GNUNET_ATS_performance_init (cfg,
+      &ats_performance_info_cb, me);
+  if (NULL == me->ats_perf_handle)
     GNUNET_log(GNUNET_ERROR_TYPE_ERROR,
         "Failed to create ATS performance handle \n");
-  return peer->ats_perf_handle;
+  return me->ats_perf_handle;
 }
 
-
 static void
 ats_perf_disconnect_adapter (void *cls, void *op_result)
 {
-  struct BenchmarkPeer *peer = cls;
+  struct BenchmarkPeer *me = cls;
 
-  GNUNET_ATS_performance_done (peer->ats_perf_handle);
-  peer->ats_perf_handle = NULL;
+  GNUNET_ATS_performance_done (me->ats_perf_handle);
+  me->ats_perf_handle = NULL;
 }
 
 static void
@@ -939,15 +1043,14 @@ ats_connect_completion_cb (void *cls, struct GNUNET_TESTBED_Operation *op,
     return;
   }
   op_done++;
-  if (op_done == (num_masters+ num_slaves))
+  if (op_done == (num_masters + num_slaves))
   {
     GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Connected to all ATS services\n");
     state.connected_ATS_service = GNUNET_YES;
-    GNUNET_SCHEDULER_add_now (&do_connect_core, NULL );
+    GNUNET_SCHEDULER_add_now (&do_comm_connect, NULL );
   }
 }
 
-
 static void
 do_connect_ats (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
@@ -957,19 +1060,17 @@ do_connect_ats (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
   GNUNET_log(GNUNET_ERROR_TYPE_INFO, "Connecting to all ATS services\n");
   for (c_m = 0; c_m < num_masters; c_m++)
   {
-    mps[c_m].ats_perf_op = GNUNET_TESTBED_service_connect (NULL,
-        mps[c_m].peer, "ats", ats_connect_completion_cb, NULL,
-        &ats_perf_connect_adapter, &ats_perf_disconnect_adapter,
-        &mps[c_m]);
+    mps[c_m].ats_perf_op = GNUNET_TESTBED_service_connect (NULL, mps[c_m].peer,
+        "ats", ats_connect_completion_cb, NULL, &ats_perf_connect_adapter,
+        &ats_perf_disconnect_adapter, &mps[c_m]);
 
   }
 
   for (c_s = 0; c_s < num_slaves; c_s++)
   {
-    sps[c_s].ats_perf_op = GNUNET_TESTBED_service_connect (NULL,
-        sps[c_s].peer, "ats", ats_connect_completion_cb, NULL,
-        &ats_perf_connect_adapter, &ats_perf_disconnect_adapter,
-        &sps[c_s]);
+    sps[c_s].ats_perf_op = GNUNET_TESTBED_service_connect (NULL, sps[c_s].peer,
+        "ats", ats_connect_completion_cb, NULL, &ats_perf_connect_adapter,
+        &ats_perf_disconnect_adapter, &sps[c_s]);
   }
 
 }
@@ -981,11 +1082,12 @@ peerinformation_cb (void *cb_cls, struct GNUNET_TESTBED_Operation *op,
   struct BenchmarkPeer *p = cb_cls;
   static int done = 0;
 
-  GNUNET_assert (pinfo->pit == GNUNET_TESTBED_PIT_IDENTITY);
+  GNUNET_assert(pinfo->pit == GNUNET_TESTBED_PIT_IDENTITY);
 
   p->id = *pinfo->result.id;
-  GNUNET_log(GNUNET_ERROR_TYPE_INFO, "[%c %03u] Peers %s\n",
-      (p->master == GNUNET_YES) ? 'M' : 'S', p->no, GNUNET_i2s (&p->id));
+  GNUNET_log(GNUNET_ERROR_TYPE_INFO, "%s [%u] has peer id `%s'\n",
+      (p->master == GNUNET_YES) ? "Master" : "Slave", p->no,
+      GNUNET_i2s (&p->id));
 
   GNUNET_TESTBED_operation_done (op);
   p->peer_id_op = NULL;
@@ -1012,8 +1114,10 @@ peerinformation_cb (void *cb_cls, struct GNUNET_TESTBED_Operation *op,
  */
 static void
 main_run (void *cls, struct GNUNET_TESTBED_RunHandle *h,
-    unsigned int num_peers, struct GNUNET_TESTBED_Peer **peers_,
-    unsigned int links_succeeded, unsigned int links_failed)
+          unsigned int num_peers,
+          struct GNUNET_TESTBED_Peer **peers_,
+          unsigned int links_succeeded,
+          unsigned int links_failed)
 {
   int c_m;
   int c_s;
@@ -1026,8 +1130,8 @@ main_run (void *cls, struct GNUNET_TESTBED_RunHandle *h,
       solver, pref_str, num_masters, num_slaves);
 
   shutdown_task = GNUNET_SCHEDULER_add_delayed (
-      GNUNET_TIME_relative_multiply (TEST_TIMEOUT,
-          num_masters + num_slaves), &do_shutdown, NULL );
+      GNUNET_TIME_relative_multiply (TEST_TIMEOUT, num_masters + num_slaves),
+      &do_shutdown, NULL );
 
   /* Setup master peers */
   for (c_m = 0; c_m < num_masters; c_m++)
@@ -1036,11 +1140,19 @@ main_run (void *cls, struct GNUNET_TESTBED_RunHandle *h,
     mps[c_m].peer = peers_[c_m];
     mps[c_m].no = c_m;
     mps[c_m].master = GNUNET_YES;
-    mps[c_m].partners = GNUNET_malloc (num_slaves * sizeof (struct BenchmarkPeer));
-    mps[c_m].peer_id_op = GNUNET_TESTBED_peer_get_information (
-            mps[c_m].peer, GNUNET_TESTBED_PIT_IDENTITY,
-            &peerinformation_cb,
-            &mps[c_m]);
+    mps[c_m].pref_partner = &sps[c_m];
+    mps[c_m].pref_value = TEST_ATS_PREFRENCE_START;
+    mps[c_m].partners =
+        GNUNET_malloc (num_slaves * sizeof (struct BenchmarkPartner));
+    mps[c_m].num_partners = num_slaves;
+    /* Initialize partners */
+    for (c_s = 0; c_s < num_slaves; c_s++)
+    {
+      mps[c_m].partners[c_s].me = &mps[c_m];
+      mps[c_m].partners[c_s].dest = &sps[c_s];
+    }
+    mps[c_m].peer_id_op = GNUNET_TESTBED_peer_get_information (mps[c_m].peer,
+        GNUNET_TESTBED_PIT_IDENTITY, &peerinformation_cb, &mps[c_m]);
   }
 
   /* Setup slave peers */
@@ -1050,11 +1162,17 @@ main_run (void *cls, struct GNUNET_TESTBED_RunHandle *h,
     sps[c_s].peer = peers_[c_s + num_masters];
     sps[c_s].no = c_s + num_masters;
     sps[c_s].master = GNUNET_NO;
-    sps[c_s].partners = GNUNET_malloc (num_masters * sizeof (struct BenchmarkPeer));
-    sps[c_s].peer_id_op = GNUNET_TESTBED_peer_get_information (
-            sps[c_s].peer, GNUNET_TESTBED_PIT_IDENTITY,
-            &peerinformation_cb,
-            &sps[c_s]);
+    sps[c_s].partners =
+        GNUNET_malloc (num_masters * sizeof (struct BenchmarkPartner));
+    sps[c_s].num_partners = num_masters;
+    /* Initialize partners */
+    for (c_m = 0; c_m < num_masters; c_m++)
+    {
+      sps[c_s].partners[c_m].me = &sps[c_s];
+      sps[c_s].partners[c_m].dest = &mps[c_m];
+    }
+    sps[c_s].peer_id_op = GNUNET_TESTBED_peer_get_information (sps[c_s].peer,
+        GNUNET_TESTBED_PIT_IDENTITY, &peerinformation_cb, &sps[c_s]);
   }
 }
 
@@ -1065,6 +1183,7 @@ main (int argc, char *argv[])
   char *tmp_sep;
   char *test_name;
   char *conf_name;
+  char *comm_name;
   char *dotexe;
   char *prefs[GNUNET_ATS_PreferenceCount] = GNUNET_ATS_PreferenceTypeString;
   int c;
@@ -1090,6 +1209,28 @@ main (int argc, char *argv[])
     return GNUNET_SYSERR;
   }
   tmp_sep[0] = '\0';
+  comm_name = GNUNET_strdup (&tmp_sep[1]);
+  tmp_sep = strchr (comm_name, '_');
+  if (NULL == tmp_sep)
+  {
+    fprintf (stderr, "Unable to parse test name `%s'\n", argv[0]);
+    GNUNET_free(solver);
+    return GNUNET_SYSERR;
+  }
+  tmp_sep[0] = '\0';
+  for (c = 0; c <= strlen (comm_name); c++)
+    comm_name[c] = toupper (comm_name[c]);
+  if (0 == strcmp (comm_name, "CORE"))
+    test_core = GNUNET_YES;
+  else if (0 == strcmp (comm_name, "TRANSPORT"))
+    test_core = GNUNET_NO;
+  else
+  {
+    GNUNET_free (comm_name);
+    GNUNET_free (solver);
+    return GNUNET_SYSERR;
+  }
+
   pref_str = GNUNET_strdup(tmp_sep + 1);
 
   GNUNET_asprintf (&conf_name, "%s%s_%s.conf", TESTNAME_PREFIX, solver,
@@ -1119,43 +1260,103 @@ main (int argc, char *argv[])
   if (-1 == pref_val)
   {
     fprintf (stderr, "Unknown preference: `%s'\n", pref_str);
-    GNUNET_free (solver);
-    GNUNET_free (pref_str);
+    GNUNET_free(solver);
+    GNUNET_free(pref_str);
+    GNUNET_free (comm_name);
     return -1;
   }
 
-  for (c = 0; c < (argc -1); c++)
+  for (c = 0; c < (argc - 1); c++)
   {
-       if (0 == strcmp(argv[c], "-s"))
-               break;
+    if (0 == strcmp (argv[c], "-d"))
+      break;
+  }
+  if (c < argc - 1)
+  {
+    if (GNUNET_OK != GNUNET_STRINGS_fancy_time_to_relative (argv[c + 1], &perf_duration))
+        fprintf (stderr, "Failed to parse duration `%s'\n", argv[c + 1]);
   }
-  if (c < argc-1)
+  else
   {
-    if ((0L != (num_slaves = strtol (argv[c + 1], NULL, 10))) && (num_slaves >= 1))
+    perf_duration = BENCHMARK_DURATION;
+  }
+  fprintf (stderr, "Running benchmark for %llu secs\n", (unsigned long long) (perf_duration.rel_value_us) / (1000 * 1000));
+
+  for (c = 0; c < (argc - 1); c++)
+  {
+    if (0 == strcmp (argv[c], "-s"))
+      break;
+  }
+  if (c < argc - 1)
+  {
+    if ((0L != (num_slaves = strtol (argv[c + 1], NULL, 10)))
+        && (num_slaves >= 1))
       fprintf (stderr, "Starting %u slave peers\n", num_slaves);
     else
-       num_slaves = DEFAULT_SLAVES_NUM;
+      num_slaves = DEFAULT_SLAVES_NUM;
   }
   else
-       num_slaves = DEFAULT_SLAVES_NUM;
+    num_slaves = DEFAULT_SLAVES_NUM;
 
-  for (c = 0; c < (argc -1); c++)
+  for (c = 0; c < (argc - 1); c++)
   {
-       if (0 == strcmp(argv[c], "-m"))
-               break;
+    if (0 == strcmp (argv[c], "-m"))
+      break;
   }
-  if (c < argc-1)
+  if (c < argc - 1)
   {
-    if ((0L != (num_masters = strtol (argv[c + 1], NULL, 10))) && (num_masters >= 2))
+    if ((0L != (num_masters = strtol (argv[c + 1], NULL, 10)))
+        && (num_masters >= 2))
       fprintf (stderr, "Starting %u master peers\n", num_masters);
     else
-       num_masters = DEFAULT_MASTERS_NUM;
+      num_masters = DEFAULT_MASTERS_NUM;
   }
   else
-       num_masters = DEFAULT_MASTERS_NUM;
+    num_masters = DEFAULT_MASTERS_NUM;
+
+  logging = GNUNET_NO;
+  for (c = 0; c < argc; c++)
+  {
+    if (0 == strcmp (argv[c], "-l"))
+      logging = GNUNET_YES;
+  }
+
+  if (GNUNET_YES == logging)
+  {
+    for (c = 0; c < (argc - 1); c++)
+    {
+      if (0 == strcmp (argv[c], "-f"))
+        break;
+    }
+    if (c < argc - 1)
+    {
+      if (GNUNET_OK != GNUNET_STRINGS_fancy_time_to_relative (argv[c + 1], &log_frequency))
+          fprintf (stderr, "Failed to parse duration `%s'\n", argv[c + 1]);
+    }
+    else
+    {
+      log_frequency = LOGGING_FREQUENCY;
+    }
+    fprintf (stderr, "Using log frequency %llu ms\n",
+        (unsigned long long) (log_frequency.rel_value_us) / (1000));
+  }
+
+  GNUNET_asprintf (&testname, "%s_%s_%s",solver, comm_name, pref_str);
+
+  if (num_slaves < num_masters)
+  {
+    fprintf (stderr, "Number of master peers is lower than slaves! exit...\n");
+    GNUNET_free(test_name);
+    GNUNET_free(solver);
+    GNUNET_free(pref_str);
+    GNUNET_free (comm_name);
+    return GNUNET_SYSERR;
+  }
 
   state.connected_ATS_service = GNUNET_NO;
-  state.connected_CORE_service = GNUNET_NO;
+  state.connected_COMM_service = GNUNET_NO;
+  state.connected_PEERS = GNUNET_NO;
+  state.benchmarking = GNUNET_NO;
   state.connected_PEERS = GNUNET_NO;
 
   mps = GNUNET_malloc (num_masters * sizeof (struct BenchmarkPeer));
@@ -1166,17 +1367,18 @@ main (int argc, char *argv[])
   event_mask = 0;
   event_mask |= (1LL << GNUNET_TESTBED_ET_CONNECT);
   event_mask |= (1LL << GNUNET_TESTBED_ET_OPERATION_FINISHED);
-  (void) GNUNET_TESTBED_test_run ("perf_ats",
-                                  conf_name, num_slaves + num_masters,
-                                  event_mask, &controller_event_cb, NULL,
-                                  &main_run, NULL);
-
-  GNUNET_free (solver);
-  GNUNET_free (pref_str);
-  GNUNET_free (conf_name);
-  GNUNET_free (test_name);
-  GNUNET_free (mps);
-  GNUNET_free (sps);
+  (void) GNUNET_TESTBED_test_run ("perf-ats", conf_name,
+      num_slaves + num_masters, event_mask, &controller_event_cb, NULL,
+      &main_run, NULL );
+
+  GNUNET_free(solver);
+  GNUNET_free(pref_str);
+  GNUNET_free(conf_name);
+  GNUNET_free(test_name);
+  GNUNET_free(testname);
+  GNUNET_free (comm_name);
+  GNUNET_free(mps);
+  GNUNET_free(sps);
 
   return result;
 }