use testbed methods to alter underlaylinkmodel

3 files changed, 64 insertions, 23 deletions

diff --git a/src/cadet/test_cadeT.c b/src/cadet/test_cadeT.c
index 97f315167..71170aa80 100644
--- a/src/cadet/test_cadeT.c
+++ b/src/cadet/test_cadeT.c
@@ -63,6 +63,23 @@
 
 static int kx_initiator;
 static struct GNUNET_TESTBED_UnderlayLinkModel *model;
+static int msg_count;
+static struct GNUNET_SCHEDULER_Task *task;
+
+enum RES {
+  RECEIVED_MESSAGE = 1
+};
+
+enum RES check;
+
+static void
+set_data_loss_rate (int rate)
+{
+  GNUNET_TESTBED_underlaylinkmodel_set_link (model,
+                                             test_peers[0].testbed_peer,
+                                             0, rate, 100);
+  GNUNET_log (GNUNET_ERROR_TYPE_INFO, "%s: %i loss.\n", __func__, rate);
+}
 
 static void
 send_message ()
@@ -73,13 +90,26 @@ send_message ()
 
   GNUNET_log (GNUNET_ERROR_TYPE_INFO, "%s\n", __func__);
 
-  envelope = GNUNET_MQ_msg_extra (msg, 10000,
+  envelope = GNUNET_MQ_msg_extra (msg, 1000,
                                   GNUNET_MESSAGE_TYPE_DUMMY);
   data = (int *) &msg[1];
   *data = 1000;
 
   GNUNET_MQ_send (GNUNET_CADET_get_mq (test_peers[0].channel), 
                   envelope);
+
+  msg_count++;
+
+  switch (msg_count) 
+  {
+    case 2: set_data_loss_rate (100); break;
+    case 4: set_data_loss_rate (0); break;
+  }
+
+  if (msg_count < 5)
+    task = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 1),
+                                         &send_message,
+                                         NULL);
 }
 
 int
@@ -94,16 +124,9 @@ handle_message (void *cls,
                 const struct GNUNET_MessageHeader *msg)
 {
   GNUNET_log (GNUNET_ERROR_TYPE_INFO, "%s\n", __func__);
+  GNUNET_CADET_receive_done (test_peers[1].channel);
 
-/*
-  model = GNUNET_TESTBED_underlaylinkmodel_create (test_peers[1].testbed_peer,
-                                                   GNUNET_TESTBED_UNDERLAYLINKMODELTYPE_BLACKLIST);
-  GNUNET_TESTBED_underlaylinkmodel_set_link (model,
-                                             test_peers[0].testbed_peer,
-                                             0, 100, 0);
-  GNUNET_log (GNUNET_ERROR_TYPE_INFO, "%s: Modified link model.\n", __func__);
-*/
-  send_message();
+  check = RECEIVED_MESSAGE;
 }
 
 /**
@@ -116,6 +139,10 @@ run_test ()
 {
   GNUNET_log (GNUNET_ERROR_TYPE_INFO, "%s\n", __func__);
 
+  // Init underlay link model to manipulate links
+  model = GNUNET_TESTBED_underlaylinkmodel_create (test_peers[1].testbed_peer,
+                                                   GNUNET_TESTBED_UNDERLAYLINKMODELTYPE_BLACKLIST);
+
   kx_initiator = (0 < GNUNET_memcmp (&test_peers[0].id, &test_peers[1].id)) ? 1 : 0;
 
   GNUNET_log (GNUNET_ERROR_TYPE_INFO, 
@@ -123,8 +150,7 @@ run_test ()
               GNUNET_i2s (&test_peers[kx_initiator].id),
               kx_initiator);
 
-  for (int i=0; i<10; i++)
-    send_message();
+  send_message();
 }
 
 
diff --git a/src/cadet/test_cadeT_util.c b/src/cadet/test_cadeT_util.c
index f2082a006..bf65aca32 100644
--- a/src/cadet/test_cadeT_util.c
+++ b/src/cadet/test_cadeT_util.c
@@ -41,6 +41,11 @@ static struct GNUNET_TESTBED_Operation *testbed_info_req[REQUESTED_PEERS];
 static struct GNUNET_HashCode hashed_portname;
 
 /**
+ * Port handle.
+ */
+struct GNUNET_CADET_Port *port;
+
+/**
  * Result of the test.
  */
 int test_result = 0;
@@ -78,6 +83,9 @@ disconnect_from_peer (void *cls,
 
   GNUNET_log (GNUNET_ERROR_TYPE_INFO, "%s\n", __func__);
 
+  if (0 == strncmp ("listening_peer", cls, sizeof (cls)))
+    GNUNET_CADET_close_port (port);
+
   GNUNET_CADET_disconnect (cadet);
 }
 
@@ -120,6 +128,7 @@ handle_port_connects (void *cls,
                       struct GNUNET_CADET_Channel *channel,
                       const struct GNUNET_PeerIdentity *source)
 {
+  test_peers[1].channel = channel;
   return NULL;
 }
 
@@ -134,12 +143,11 @@ setup_listening_peer (void *cls,
                       const struct GNUNET_CONFIGURATION_Handle *cfg)
 {
   struct GNUNET_CADET_Handle *cadet;
-  struct GNUNET_CADET_Port *port;
   struct GNUNET_MQ_MessageHandler msg_handlers[] = {
-    GNUNET_MQ_hd_fixed_size (message,
-                             GNUNET_MESSAGE_TYPE_DUMMY,
-                             struct GNUNET_MessageHeader,
-                             NULL),
+    GNUNET_MQ_hd_var_size (message,
+                           GNUNET_MESSAGE_TYPE_DUMMY,
+                           struct GNUNET_MessageHeader,
+                           NULL),
     GNUNET_MQ_handler_end ()
   };
 
@@ -198,13 +206,13 @@ connect_to_service (void *cb_cls,
                                       "cadet", 
                                       &check_test_readyness, NULL,
                                       &setup_listening_peer,
-                                      &disconnect_from_peer, NULL);
+                                      &disconnect_from_peer, "listening_peer");
     testbed_to_svc[0] = 
       GNUNET_TESTBED_service_connect (NULL, test_peers[0].testbed_peer,
                                       "cadet",
                                       &check_test_readyness, NULL,
                                       &setup_initiating_peer,
-                                      &disconnect_from_peer, NULL);
+                                      &disconnect_from_peer, "initiating_peer");
   }
 }
 
diff --git a/src/cadet/test_cadeT_util.h b/src/cadet/test_cadeT_util.h
index 1c85abab6..b93804613 100644
--- a/src/cadet/test_cadeT_util.h
+++ b/src/cadet/test_cadeT_util.h
@@ -33,12 +33,19 @@
 
 int test_result;
 
-void prepare_test ();
+void 
+prepare_test ();
 
-void run_test ();
+void 
+run_test ();
 
-void handle_message (void *cls,
-                     const struct GNUNET_MessageHeader *msg);
+void 
+handle_message (void *cls,
+                const struct GNUNET_MessageHeader *msg);
+
+int
+check_message (void *cls,
+               const struct GNUNET_MessageHeader *message);
 
 /**
  * Structure for storing information of testbed peers.