indentation
[oweals/gnunet.git] / src / dht / test_dht_api.c
index 29606db605b6632b121f05f26b87e08221e13b75..407e2279b292ea4d53034f3bb80c9f99d7aa3c57 100644 (file)
@@ -34,7 +34,7 @@
 #include "gnunet_dht_service.h"
 #include "gnunet_hello_lib.h"
 
-#define VERBOSE GNUNET_YES
+#define VERBOSE GNUNET_NO
 
 #define VERBOSE_ARM GNUNET_NO
 
@@ -43,7 +43,7 @@
 /**
  * How long until we really give up on a particular testcase portion?
  */
-#define TOTAL_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 50)
+#define TOTAL_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 600)
 
 /**
  * How long until we give up on any particular operation (and retry)?
@@ -85,7 +85,7 @@ struct PeerContext
   struct GNUNET_DHT_FindPeerHandle *find_peer_handle;
 
 #if START_ARM
-  pid_t arm_pid;
+  struct GNUNET_OS_Process *arm_proc;
 #endif
 };
 
@@ -93,7 +93,6 @@ static struct PeerContext p1;
 
 struct RetryContext retry_context;
 
-static struct GNUNET_SCHEDULER_Handle *sched;
 
 static int ok;
 
@@ -109,34 +108,23 @@ GNUNET_SCHEDULER_TaskIdentifier die_task;
 static void
 end (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
-  /* do work here */
-  sleep(2);
-  GNUNET_SCHEDULER_cancel (sched, die_task);
-
-  GNUNET_DHT_disconnect (p1.dht_handle);
+  GNUNET_SCHEDULER_cancel (die_task);
   die_task = GNUNET_SCHEDULER_NO_TASK;
-
-  if (tc->reason == GNUNET_SCHEDULER_REASON_TIMEOUT)
-    {
-      GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-                  "DHT disconnected, returning FAIL!\n");
-      ok = 365;
-    }
-  else
-    {
-      GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-                  "DHT disconnected, returning success!\n");
-      ok = 0;
-    }
+  GNUNET_DHT_disconnect (p1.dht_handle);
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "DHT disconnected, returning success!\n");
+  ok = 0;
 }
 
 static void
 stop_arm (struct PeerContext *p)
 {
 #if START_ARM
-  if (0 != PLIBC_KILL (p->arm_pid, SIGTERM))
+  if (0 != GNUNET_OS_process_kill (p->arm_proc, SIGTERM))
     GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "kill");
-  GNUNET_OS_process_wait (p->arm_pid);
+  GNUNET_OS_process_wait (p->arm_proc);
+  GNUNET_OS_process_close (p->arm_proc);
+  p->arm_proc = NULL;
 #endif
   GNUNET_CONFIGURATION_destroy (p->cfg);
 }
@@ -150,72 +138,24 @@ end_badly ()
   fprintf (stderr, "Ending on an unhappy note.\n");
 #endif
 
-  if ((retry_context.peer_ctx != NULL) && (retry_context.peer_ctx->find_peer_handle != NULL))
-    GNUNET_DHT_find_peer_stop(retry_context.peer_ctx->find_peer_handle, NULL, NULL);
+  if ((retry_context.peer_ctx != NULL) &&
+      (retry_context.peer_ctx->find_peer_handle != NULL))
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Stopping find peer request!\n");
+    GNUNET_DHT_find_peer_stop (retry_context.peer_ctx->find_peer_handle);
+  }
+  if ((retry_context.peer_ctx != NULL) &&
+      (retry_context.peer_ctx->get_handle != NULL))
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Stopping get request!\n");
+    GNUNET_DHT_get_stop (retry_context.peer_ctx->get_handle);
+  }
   if (retry_context.retry_task != GNUNET_SCHEDULER_NO_TASK)
-    GNUNET_SCHEDULER_cancel(sched, retry_context.retry_task);
+    GNUNET_SCHEDULER_cancel (retry_context.retry_task);
   GNUNET_DHT_disconnect (p1.dht_handle);
-
   ok = 1;
-  return;
-}
-
-/**
- * Signature of the main function of a task.
- *
- * @param cls closure
- * @param tc context information (why was this task triggered now)
- */
-void test_set_peer_malicious_drop (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
-{
-  struct PeerContext *peer = cls;
-
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Called test_set_peer_malicious_drop!\n");
-  if (tc->reason == GNUNET_SCHEDULER_REASON_TIMEOUT)
-    GNUNET_SCHEDULER_add_now (sched, &end_badly, NULL);
-
-  GNUNET_assert (peer->dht_handle != NULL);
-
-  GNUNET_DHT_set_malicious_dropper (peer->dht_handle, &end, &p1);
 }
 
-/**
- * Signature of the main function of a task.
- *
- * @param cls closure
- * @param tc context information (why was this task triggered now)
- */
-void test_set_peer_malicious_put (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
-{
-  struct PeerContext *peer = cls;
-
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Called test_set_peer_malicious_put!\n");
-  if (tc->reason == GNUNET_SCHEDULER_REASON_TIMEOUT)
-    GNUNET_SCHEDULER_add_now (sched, &end_badly, NULL);
-
-  GNUNET_assert (peer->dht_handle != NULL);
-
-  GNUNET_DHT_set_malicious_putter (peer->dht_handle, 750, &test_set_peer_malicious_drop, &p1);
-}
-
-/**
- * Signature of the main function of a task.
- *
- * @param cls closure
- * @param tc context information (why was this task triggered now)
- */
-void test_set_peer_malicious_get (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
-{
-  struct PeerContext *peer = cls;
-
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Called test_set_peer_malicious_get!\n");
-  if (tc->reason == GNUNET_SCHEDULER_REASON_TIMEOUT)
-    GNUNET_SCHEDULER_add_now (sched, &end_badly, NULL);
-
-  GNUNET_assert (peer->dht_handle != NULL);
-
-  GNUNET_DHT_set_malicious_getter (peer->dht_handle, 1500, &test_set_peer_malicious_put, &p1);
-}
 
 /**
  * Signature of the main function of a task.
@@ -229,15 +169,28 @@ test_find_peer_stop (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
   struct PeerContext *peer = cls;
 
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Called test_find_peer_stop!\n");
-  if (tc->reason == GNUNET_SCHEDULER_REASON_TIMEOUT)
-    GNUNET_SCHEDULER_add_now (sched, &end_badly, NULL);
+  if ((tc->reason & GNUNET_SCHEDULER_REASON_TIMEOUT) != 0)
+  {
+    GNUNET_break (0);
+    GNUNET_SCHEDULER_cancel (die_task);
+    GNUNET_SCHEDULER_add_now (&end_badly, NULL);
+    return;
+  }
 
   GNUNET_assert (peer->dht_handle != NULL);
 
-  GNUNET_DHT_find_peer_stop (peer->find_peer_handle, &test_set_peer_malicious_get, &p1);
-
-  //GNUNET_SCHEDULER_add_delayed(sched, GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_SECONDS, 1), &end, &p1);
+  GNUNET_DHT_find_peer_stop (peer->find_peer_handle);
+  peer->find_peer_handle = NULL;
 
+#if HAVE_MALICIOUS
+  GNUNET_DHT_set_malicious_getter (peer->dht_handle, GNUNET_TIME_UNIT_SECONDS,
+                                   NULL, NULL);
+  GNUNET_DHT_set_malicious_putter (peer->dht_handle, GNUNET_TIME_UNIT_SECONDS,
+                                   NULL, NULL);
+  GNUNET_DHT_set_malicious_dropper (peer->dht_handle, NULL, NULL);
+#endif
+  GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
+                                (GNUNET_TIME_UNIT_SECONDS, 1), &end, &p1);
 }
 
 
@@ -249,32 +202,33 @@ test_find_peer_stop (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
  * @param peer the peer we learned about
  * @param reply response
  */
-void test_find_peer_processor (void *cls,
-                               const struct GNUNET_HELLO_Message *hello)
+void
+test_find_peer_processor (void *cls, const struct GNUNET_HELLO_Message *hello)
 {
   struct RetryContext *retry_ctx = cls;
   struct GNUNET_PeerIdentity peer;
 
-  if (GNUNET_OK == GNUNET_HELLO_get_id(hello, &peer))
-    {
-      GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-                  "test_find_peer_processor called (peer `%s'), stopping find peer request!\n", GNUNET_i2s(&peer));
-
-      if (retry_ctx->retry_task != GNUNET_SCHEDULER_NO_TASK)
-        {
-          GNUNET_SCHEDULER_cancel(sched, retry_ctx->retry_task);
-          retry_ctx->retry_task = GNUNET_SCHEDULER_NO_TASK;
-        }
+  if (GNUNET_OK == GNUNET_HELLO_get_id (hello, &peer))
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "test_find_peer_processor called (peer `%s'), stopping find peer request!\n",
+                GNUNET_i2s (&peer));
 
-      GNUNET_SCHEDULER_add_continuation (sched, &test_find_peer_stop, &p1,
-                                         GNUNET_SCHEDULER_REASON_PREREQ_DONE);
-    }
-  else
+    if (retry_ctx->retry_task != GNUNET_SCHEDULER_NO_TASK)
     {
-      GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-                  "received find peer request, but hello_get_id failed!\n");
+      GNUNET_SCHEDULER_cancel (retry_ctx->retry_task);
+      retry_ctx->retry_task = GNUNET_SCHEDULER_NO_TASK;
     }
 
+    GNUNET_SCHEDULER_add_continuation (&test_find_peer_stop, &p1,
+                                       GNUNET_SCHEDULER_REASON_PREREQ_DONE);
+  }
+  else
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "received find peer request, but hello_get_id failed!\n");
+  }
+
 }
 
 /**
@@ -297,29 +251,40 @@ retry_find_peer (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
   struct RetryContext *retry_ctx = cls;
   GNUNET_HashCode hash;
+
   memset (&hash, 42, sizeof (GNUNET_HashCode));
 
-  if (GNUNET_TIME_absolute_get_remaining(retry_ctx->real_timeout).value > 0)
-    {
-      GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-                  "test_find_peer timed out, retrying!\n");
-      retry_ctx->next_timeout = GNUNET_TIME_relative_multiply(retry_ctx->next_timeout, 2);
-      retry_ctx->peer_ctx->find_peer_handle =
-          GNUNET_DHT_find_peer_start (retry_ctx->peer_ctx->dht_handle, retry_ctx->next_timeout, 0, &hash,
-                                      &test_find_peer_processor, retry_ctx, NULL, NULL);
-    }
+  if (GNUNET_TIME_absolute_get_remaining (retry_ctx->real_timeout).rel_value >
+      0)
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "test_find_peer timed out, retrying!\n");
+    retry_ctx->next_timeout =
+        GNUNET_TIME_relative_multiply (retry_ctx->next_timeout, 2);
+    retry_ctx->peer_ctx->find_peer_handle =
+        GNUNET_DHT_find_peer_start (retry_ctx->peer_ctx->dht_handle,
+                                    retry_ctx->next_timeout, &hash,
+                                    GNUNET_DHT_RO_NONE,
+                                    &test_find_peer_processor, retry_ctx);
+  }
   else
-    {
-      GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
-                  "test_find_peer timed out for good, failing!\n");
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                "test_find_peer timed out for good, failing!\n");
 
-      retry_ctx->peer_ctx->find_peer_handle = NULL;
-    }
+    retry_ctx->peer_ctx->find_peer_handle = NULL;
+  }
 
   if (retry_ctx->peer_ctx->find_peer_handle == NULL)
-    GNUNET_SCHEDULER_add_now (sched, &end_badly, &p1);
-  else
-    retry_ctx->retry_task = GNUNET_SCHEDULER_add_delayed(sched, retry_ctx->next_timeout, &retry_find_peer_stop, retry_ctx);
+  {
+    GNUNET_break (0);
+    GNUNET_SCHEDULER_cancel (die_task);
+    GNUNET_SCHEDULER_add_now (&end_badly, &p1);
+    return;
+  }
+  retry_ctx->retry_task =
+      GNUNET_SCHEDULER_add_delayed (retry_ctx->next_timeout,
+                                    &retry_find_peer_stop, retry_ctx);
 }
 
 /**
@@ -333,13 +298,15 @@ retry_find_peer_stop (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
   struct RetryContext *retry_ctx = cls;
   GNUNET_HashCode hash;
+
   memset (&hash, 42, sizeof (GNUNET_HashCode));
 
   if (retry_ctx->peer_ctx->find_peer_handle != NULL)
-    GNUNET_DHT_find_peer_stop(retry_ctx->peer_ctx->find_peer_handle, &retry_find_peer, retry_ctx);
-  else
-    GNUNET_SCHEDULER_add_now (sched, &retry_find_peer, retry_ctx);
-
+  {
+    GNUNET_DHT_find_peer_stop (retry_ctx->peer_ctx->find_peer_handle);
+    retry_ctx->peer_ctx->find_peer_handle = NULL;
+  }
+  GNUNET_SCHEDULER_add_now (&retry_find_peer, retry_ctx);
 }
 
 /**
@@ -353,23 +320,32 @@ test_find_peer (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
   struct PeerContext *peer = cls;
   GNUNET_HashCode hash;
+
   memset (&hash, 42, sizeof (GNUNET_HashCode));
 
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Called test_find_peer!\n");
   GNUNET_assert (peer->dht_handle != NULL);
 
-  retry_context.real_timeout = GNUNET_TIME_relative_to_absolute(TOTAL_TIMEOUT);
+  retry_context.real_timeout = GNUNET_TIME_relative_to_absolute (TOTAL_TIMEOUT);
   retry_context.next_timeout = BASE_TIMEOUT;
   retry_context.peer_ctx = peer;
 
-  peer->find_peer_handle =
-    GNUNET_DHT_find_peer_start (peer->dht_handle, retry_context.next_timeout, 1, &hash,
-                                &test_find_peer_processor, &retry_context, NULL, NULL);
+  peer->find_peer_handle
+      =
+      GNUNET_DHT_find_peer_start (peer->dht_handle, retry_context.next_timeout,
+                                  &hash, GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE,
+                                  &test_find_peer_processor, &retry_context);
 
   if (peer->find_peer_handle == NULL)
-    GNUNET_SCHEDULER_add_now (sched, &end_badly, &p1);
-  else
-    retry_context.retry_task = GNUNET_SCHEDULER_add_delayed(sched, retry_context.next_timeout, &retry_find_peer_stop, &retry_context);
+  {
+    GNUNET_break (0);
+    GNUNET_SCHEDULER_cancel (die_task);
+    GNUNET_SCHEDULER_add_now (&end_badly, &p1);
+    return;
+  }
+  retry_context.retry_task =
+      GNUNET_SCHEDULER_add_delayed (retry_context.next_timeout,
+                                    &retry_find_peer_stop, &retry_context);
 }
 
 /**
@@ -384,27 +360,31 @@ test_get_stop (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
   struct PeerContext *peer = cls;
 
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Called test_get_stop!\n");
-  if (tc->reason == GNUNET_SCHEDULER_REASON_TIMEOUT)
-    GNUNET_SCHEDULER_add_now (sched, &end_badly, NULL);
-
+  if ((tc->reason & GNUNET_SCHEDULER_REASON_TIMEOUT) != 0)
+  {
+    GNUNET_break (0);
+    GNUNET_SCHEDULER_cancel (die_task);
+    GNUNET_SCHEDULER_add_now (&end_badly, NULL);
+    return;
+  }
   GNUNET_assert (peer->dht_handle != NULL);
-
-  GNUNET_DHT_get_stop (peer->get_handle, &test_find_peer, &p1);
-
-  //GNUNET_SCHEDULER_add_delayed(sched, GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_SECONDS, 1), &test_put, &p1);
-
+  GNUNET_DHT_get_stop (peer->get_handle);
+  peer->get_handle = NULL;
+  GNUNET_SCHEDULER_add_now (&test_find_peer, &p1);
 }
 
 void
 test_get_iterator (void *cls,
                    struct GNUNET_TIME_Absolute exp,
                    const GNUNET_HashCode * key,
-                   uint32_t type, uint32_t size, const void *data)
+                   const struct GNUNET_PeerIdentity *const *get_path,
+                   const struct GNUNET_PeerIdentity *const *put_path,
+                   enum GNUNET_BLOCK_Type type, size_t size, const void *data)
 {
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
               "test_get_iterator called (we got a result), stopping get request!\n");
 
-  GNUNET_SCHEDULER_add_continuation (sched, &test_get_stop, &p1,
+  GNUNET_SCHEDULER_add_continuation (&test_get_stop, &p1,
                                      GNUNET_SCHEDULER_REASON_PREREQ_DONE);
 }
 
@@ -419,18 +399,33 @@ test_get (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
   struct PeerContext *peer = cls;
   GNUNET_HashCode hash;
+
   memset (&hash, 42, sizeof (GNUNET_HashCode));
 
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Called test_get!\n");
 
   GNUNET_assert (peer->dht_handle != NULL);
+  retry_context.real_timeout = GNUNET_TIME_relative_to_absolute (TOTAL_TIMEOUT);
+  retry_context.next_timeout = BASE_TIMEOUT;
 
   peer->get_handle =
-    GNUNET_DHT_get_start (peer->dht_handle, TOTAL_TIMEOUT, 42, &hash,
-                          &test_get_iterator, NULL, NULL, NULL);
+      GNUNET_DHT_get_start (peer->dht_handle,
+                            TOTAL_TIMEOUT,
+                            GNUNET_BLOCK_TYPE_TEST,
+                            &hash,
+                            DEFAULT_GET_REPLICATION,
+                            GNUNET_DHT_RO_NONE,
+                            NULL, 0, NULL, 0, &test_get_iterator, NULL);
 
   if (peer->get_handle == NULL)
-    GNUNET_SCHEDULER_add_now (sched, &end_badly, &p1);
+  {
+    GNUNET_break (0);
+    GNUNET_SCHEDULER_cancel (die_task);
+    GNUNET_SCHEDULER_add_now (&end_badly, &p1);
+    return;
+  }
+
+  retry_context.peer_ctx = peer;
 }
 
 /**
@@ -446,18 +441,23 @@ test_put (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
   GNUNET_HashCode hash;
   char *data;
   size_t data_size = 42;
+
   memset (&hash, 42, sizeof (GNUNET_HashCode));
   data = GNUNET_malloc (data_size);
   memset (data, 43, data_size);
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Called test_put!\n");
-  peer->dht_handle = GNUNET_DHT_connect (sched, peer->cfg, 100);
+  peer->dht_handle = GNUNET_DHT_connect (peer->cfg, 100);
 
   GNUNET_assert (peer->dht_handle != NULL);
 
-  GNUNET_DHT_put (peer->dht_handle, &hash, 42, data_size, data,
-                  GNUNET_TIME_relative_to_absolute (TOTAL_TIMEOUT), TOTAL_TIMEOUT,
-                  &test_get, &p1);
-  GNUNET_free(data);
+  GNUNET_DHT_put (peer->dht_handle, &hash,
+                  DEFAULT_PUT_REPLICATION,
+                  GNUNET_DHT_RO_NONE,
+                  GNUNET_BLOCK_TYPE_TEST,
+                  data_size, data,
+                  GNUNET_TIME_relative_to_absolute (TOTAL_TIMEOUT),
+                  TOTAL_TIMEOUT, &test_get, &p1);
+  GNUNET_free (data);
 }
 
 static void
@@ -465,12 +465,12 @@ setup_peer (struct PeerContext *p, const char *cfgname)
 {
   p->cfg = GNUNET_CONFIGURATION_create ();
 #if START_ARM
-  p->arm_pid = GNUNET_OS_start_process (NULL, NULL, "gnunet-service-arm",
-                                        "gnunet-service-arm",
+  p->arm_proc = GNUNET_OS_start_process (NULL, NULL, "gnunet-service-arm",
+                                         "gnunet-service-arm",
 #if VERBOSE_ARM
-                                        "-L", "DEBUG",
+                                         "-L", "DEBUG",
 #endif
-                                        "-c", cfgname, NULL);
+                                         "-c", cfgname, NULL);
 #endif
   GNUNET_assert (GNUNET_OK == GNUNET_CONFIGURATION_load (p->cfg, cfgname));
 
@@ -478,25 +478,20 @@ setup_peer (struct PeerContext *p, const char *cfgname)
 
 static void
 run (void *cls,
-     struct GNUNET_SCHEDULER_Handle *s,
      char *const *args,
      const char *cfgfile, const struct GNUNET_CONFIGURATION_Handle *cfg)
 {
   GNUNET_assert (ok == 1);
   OKPP;
-  sched = s;
 
-  die_task = GNUNET_SCHEDULER_add_delayed (sched,
-                                           GNUNET_TIME_relative_multiply
+  die_task = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
                                            (GNUNET_TIME_UNIT_MINUTES, 1),
                                            &end_badly, NULL);
 
   setup_peer (&p1, "test_dht_api_peer1.conf");
 
-  GNUNET_SCHEDULER_add_delayed (sched,
-                                GNUNET_TIME_relative_multiply
-                                (GNUNET_TIME_UNIT_SECONDS, 1), &test_put,
-                                &p1);
+  GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
+                                (GNUNET_TIME_UNIT_SECONDS, 1), &test_put, &p1);
 }
 
 static int
@@ -528,9 +523,6 @@ int
 main (int argc, char *argv[])
 {
   int ret;
-#ifdef MINGW
-  return GNUNET_SYSERR;
-#endif
 
   GNUNET_log_setup ("test-dht-api",
 #if VERBOSE