doc/documentation/testbed_test.c

   1 #include <unistd.h>
   2 #include <gnunet/platform.h>
   3 #include <gnunet/gnunet_util_lib.h>
   4 #include <gnunet/gnunet_testbed_service.h>
   5 #include <gnunet/gnunet_dht_service.h>
   6
   7 #define NUM_PEERS 20
   8
   9 static struct GNUNET_TESTBED_Operation *dht_op;
  10
  11 static struct GNUNET_DHT_Handle *dht_handle;
  12
  13
  14 struct MyContext
  15 {
  16   int ht_len;
  17 } ctxt;
  18
  19
  20 static int result;
  21
  22
  23 static void
  24 shutdown_task (void *cls)
  25 {
  26   if (NULL != dht_op)
  27   {
  28     GNUNET_TESTBED_operation_done (dht_op);
  29     dht_op = NULL;
  30     dht_handle = NULL;
  31   }
  32   result = GNUNET_OK;
  33 }
  34
  35
  36 static void
  37 service_connect_comp (void *cls,
  38                       struct GNUNET_TESTBED_Operation *op,
  39                       void *ca_result,
  40                       const char *emsg)
  41 {
  42   GNUNET_assert (op == dht_op);
  43   dht_handle = ca_result;
  44   // Do work here...
  45   GNUNET_SCHEDULER_shutdown ();
  46 }
  47
  48
  49 static void *
  50 dht_ca (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg)
  51 {
  52   struct MyContext *ctxt = cls;
  53
  54   dht_handle = GNUNET_DHT_connect (cfg, ctxt->ht_len);
  55   return dht_handle;
  56 }
  57
  58
  59 static void
  60 dht_da (void *cls, void *op_result)
  61 {
  62   struct MyContext *ctxt = cls;
  63
  64   GNUNET_DHT_disconnect ((struct GNUNET_DHT_Handle *) op_result);
  65   dht_handle = NULL;
  66 }
  67
  68
  69 static void
  70 test_master (void *cls,
  71              struct GNUNET_TESTBED_RunHandle *h,
  72              unsigned int num_peers,
  73              struct GNUNET_TESTBED_Peer **peers,
  74              unsigned int links_succeeded,
  75              unsigned int links_failed)
  76 {
  77   ctxt.ht_len = 10;
  78   dht_op = GNUNET_TESTBED_service_connect
  79       (NULL, peers[0], "dht",
  80        &service_connect_comp, NULL,
  81        &dht_ca, &dht_da, &ctxt);
  82   GNUNET_SCHEDULER_add_shutdown (&shutdown_task, NULL);
  83 }
  84
  85
  86 int
  87 main (int argc, char **argv)
  88 {
  89   int ret;
  90
  91   result = GNUNET_SYSERR;
  92   ret = GNUNET_TESTBED_test_run
  93       ("awesome-test", "template.conf",
  94        NUM_PEERS, 0LL,
  95        NULL, NULL, &test_master, NULL);
  96   if ( (GNUNET_OK != ret) || (GNUNET_OK != result) )
  97     return 1;
  98   return 0;
  99 }