src/nse/test_nse_multipeer.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2009 Christian Grothoff (and other contributing authors)
   4
   5      GNUnet is free software; you can redistribute it and/or modify
   6      it under the terms of the GNU General Public License as published
   7      by the Free Software Foundation; either version 3, or (at your
   8      option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      General Public License for more details.
  14
  15      You should have received a copy of the GNU General Public License
  16      along with GNUnet; see the file COPYING.  If not, write to the
  17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18      Boston, MA 02111-1307, USA.
  19 */
  20 /**
  21  * @file nse/test_nse_multipeer.c
  22  *
  23  * @brief Testcase for the network size estimation service.  Starts
  24  *        a peergroup with a given number of peers, then waits to
  25  *        receive size estimates from each peer.  Expects to wait
  26  *        for one message from each peer.
  27  */
  28 #include "platform.h"
  29 #include "gnunet_testing_lib.h"
  30 #include "gnunet_nse_service.h"
  31
  32 #define VERBOSE GNUNET_EXTRA_LOGGING
  33
  34 #define NUM_PEERS 4
  35
  36 struct NSEPeer
  37 {
  38   struct NSEPeer *prev;
  39
  40   struct NSEPeer *next;
  41
  42   struct GNUNET_TESTING_Daemon *daemon;
  43
  44   struct GNUNET_NSE_Handle *nse_handle;
  45 };
  46
  47 struct NSEPeer *peer_head;
  48
  49 struct NSEPeer *peer_tail;
  50
  51 /**
  52  * How long do we run the test?
  53  */
  54 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 300)
  55
  56 static int ok;
  57
  58 static int peers_left;
  59
  60 static unsigned int num_peers;
  61
  62 static unsigned int total_connections;
  63
  64 static struct GNUNET_TESTING_PeerGroup *pg;
  65
  66 /**
  67  * Check whether peers successfully shut down.
  68  */
  69 static void
  70 shutdown_callback (void *cls, const char *emsg)
  71 {
  72   if (emsg != NULL)
  73   {
  74 #if VERBOSE
  75     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Shutdown of peers failed: %s!\n",
  76                 emsg);
  77 #endif
  78     if (ok == 0)
  79       ok = 666;
  80   }
  81   else
  82   {
  83 #if VERBOSE
  84     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "All peers successfully shut down!\n");
  85 #endif
  86     ok = 0;
  87   }
  88 }
  89
  90 static void
  91 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
  92 {
  93   struct NSEPeer *pos;
  94
  95 #if VERBOSE
  96   fprintf (stderr, "Ending test.\n");
  97 #endif
  98
  99   while (NULL != (pos = peer_head))
 100   {
 101     GNUNET_NSE_disconnect (pos->nse_handle);
 102     GNUNET_CONTAINER_DLL_remove (peer_head, peer_tail, pos);
 103     GNUNET_free (pos);
 104   }
 105
 106   GNUNET_TESTING_daemons_stop (pg, TIMEOUT, &shutdown_callback, NULL);
 107 }
 108
 109 /**
 110  * Callback to call when network size estimate is updated.
 111  *
 112  * @param cls closure
 113  * @param timestamp server timestamp
 114  * @param estimate the value of the current network size estimate
 115  * @param std_dev standard deviation (rounded down to nearest integer)
 116  *                of the size estimation values seen
 117  *
 118  */
 119 static void
 120 handle_estimate (void *cls, struct GNUNET_TIME_Absolute timestamp,
 121                  double estimate, double std_dev)
 122 {
 123   struct NSEPeer *peer = cls;
 124
 125   fprintf (stderr,
 126            "Received network size estimate from peer %s. logSize: %f std.dev. %f (%f/%u)\n",
 127            GNUNET_i2s (&peer->daemon->id), estimate, std_dev,
 128            GNUNET_NSE_log_estimate_to_n (estimate), num_peers);
 129 }
 130
 131
 132 static void
 133 connect_nse_service (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 134 {
 135   struct NSEPeer *current_peer;
 136   unsigned int i;
 137
 138 #if VERBOSE
 139   fprintf (stderr, "TEST_NSE_MULTIPEER: connecting to nse service of peers\n");
 140 #endif
 141   for (i = 0; i < num_peers; i++)
 142   {
 143     current_peer = GNUNET_malloc (sizeof (struct NSEPeer));
 144     current_peer->daemon = GNUNET_TESTING_daemon_get (pg, i);
 145     current_peer->nse_handle =
 146         GNUNET_NSE_connect (current_peer->daemon->cfg, &handle_estimate,
 147                             current_peer);
 148     GNUNET_assert (current_peer->nse_handle != NULL);
 149     GNUNET_CONTAINER_DLL_insert (peer_head, peer_tail, current_peer);
 150   }
 151 }
 152
 153
 154 static void
 155 my_cb (void *cls, const char *emsg)
 156 {
 157   if (emsg != NULL)
 158   {
 159     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 160                 "Peergroup callback called with error, aborting test!\n");
 161     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Error from testing: `%s'\n");
 162     ok = 1;
 163     GNUNET_TESTING_daemons_stop (pg, TIMEOUT, &shutdown_callback, NULL);
 164     return;
 165   }
 166 #if VERBOSE
 167   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 168               "Peer Group started successfully, connecting to NSE service for each peer!\n");
 169 #endif
 170   GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Have %u connections\n",
 171               total_connections);
 172
 173   GNUNET_SCHEDULER_add_now (&connect_nse_service, NULL);
 174 }
 175
 176
 177 /**
 178  * Function that will be called whenever
 179  * two daemons are connected by the testing library.
 180  *
 181  * @param cls closure
 182  * @param first peer id for first daemon
 183  * @param second peer id for the second daemon
 184  * @param distance distance between the connected peers
 185  * @param first_cfg config for the first daemon
 186  * @param second_cfg config for the second daemon
 187  * @param first_daemon handle for the first daemon
 188  * @param second_daemon handle for the second daemon
 189  * @param emsg error message (NULL on success)
 190  */
 191 static void
 192 connect_cb (void *cls, const struct GNUNET_PeerIdentity *first,
 193             const struct GNUNET_PeerIdentity *second, uint32_t distance,
 194             const struct GNUNET_CONFIGURATION_Handle *first_cfg,
 195             const struct GNUNET_CONFIGURATION_Handle *second_cfg,
 196             struct GNUNET_TESTING_Daemon *first_daemon,
 197             struct GNUNET_TESTING_Daemon *second_daemon, const char *emsg)
 198 {
 199   if (emsg == NULL)
 200   {
 201     //fprintf(stderr, "Connected %s -> %s\n", GNUNET_i2s(first), second_id);
 202     total_connections++;
 203   }
 204 }
 205
 206
 207
 208 static void
 209 run (void *cls, char *const *args, const char *cfgfile,
 210      const struct GNUNET_CONFIGURATION_Handle *cfg)
 211 {
 212   struct GNUNET_CONFIGURATION_Handle *testing_cfg;
 213   unsigned long long total_peers;
 214
 215   ok = 1;
 216   testing_cfg = GNUNET_CONFIGURATION_create ();
 217   GNUNET_assert (GNUNET_OK == GNUNET_CONFIGURATION_load (testing_cfg, cfgfile));
 218
 219 #if VERBOSE
 220   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Starting daemons.\n");
 221   GNUNET_CONFIGURATION_set_value_string (testing_cfg, "testing",
 222                                          "use_progressbars", "YES");
 223 #endif
 224   if (GNUNET_OK !=
 225       GNUNET_CONFIGURATION_get_value_number (testing_cfg, "testing",
 226                                              "num_peers", &total_peers))
 227     total_peers = NUM_PEERS;
 228
 229   peers_left = total_peers;
 230   num_peers = peers_left;
 231   pg = GNUNET_TESTING_peergroup_start (testing_cfg, peers_left, TIMEOUT,
 232                                        &connect_cb, &my_cb, NULL, NULL);
 233   GNUNET_assert (pg != NULL);
 234   GNUNET_SCHEDULER_add_delayed (TIMEOUT, &shutdown_task, NULL);
 235 }
 236
 237
 238 static int
 239 check ()
 240 {
 241   char *const argv[] = { "test-nse-multipeer",
 242     "-c",
 243     "test_nse.conf",
 244 #if VERBOSE
 245     "-L", "DEBUG",
 246 #endif
 247     NULL
 248   };
 249   struct GNUNET_GETOPT_CommandLineOption options[] = {
 250     GNUNET_GETOPT_OPTION_END
 251   };
 252   GNUNET_PROGRAM_run ((sizeof (argv) / sizeof (char *)) - 1, argv,
 253                       "test-nse-multipeer", "nohelp", options, &run, &ok);
 254   return ok;
 255 }
 256
 257
 258 int
 259 main (int argc, char *argv[])
 260 {
 261   int ret;
 262
 263   GNUNET_log_setup ("test-nse-multipeer",
 264 #if VERBOSE
 265                     "DEBUG",
 266 #else
 267                     "WARNING",
 268 #endif
 269                     NULL);
 270   ret = check ();
 271   // GNUNET_DISK_directory_remove ("/tmp/test-nse-multipeer");
 272   return ret;
 273 }
 274
 275 /* end of test_nse_multipeer.c */