src/mesh/test_mesh_2dtorus.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2011 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 mesh/test_mesh_2dtorus.c
  22  *
  23  * @brief Test for creating a 2dtorus.
  24  */
  25 #include "platform.h"
  26 #include "gnunet_testing_lib.h"
  27
  28 #define VERBOSE GNUNET_YES
  29 #define REMOVE_DIR GNUNET_YES
  30
  31 /**
  32  * How long until we give up on connecting the peers?
  33  */
  34 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 1500)
  35
  36 /**
  37  * Time to wait for stuff that should be rather fast
  38  */
  39 #define SHORT_TIME GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30)
  40
  41
  42 /**
  43  * How many events have happened
  44  */
  45 static int ok;
  46
  47 /**
  48  * Be verbose
  49  */
  50 static int verbose;
  51
  52 /**
  53  * Total number of peers in the test.
  54  */
  55 static unsigned long long num_peers;
  56
  57 /**
  58  * Global configuration file
  59  */
  60 static struct GNUNET_CONFIGURATION_Handle *testing_cfg;
  61
  62 /**
  63  * Total number of currently running peers.
  64  */
  65 static unsigned long long peers_running;
  66
  67 /**
  68  * Total number of successful connections in the whole network.
  69  */
  70 static unsigned int total_connections;
  71
  72 /**
  73  * Total number of counted topo connections
  74  */
  75 static unsigned int topo_connections;
  76
  77 /**
  78  * Total number of failed connections in the whole network.
  79  */
  80 static unsigned int failed_connections;
  81
  82 /**
  83  * The currently running peer group.
  84  */
  85 static struct GNUNET_TESTING_PeerGroup *pg;
  86
  87 /**
  88  * Task called to disconnect peers
  89  */
  90 static GNUNET_SCHEDULER_TaskIdentifier disconnect_task;
  91
  92 /**
  93  * Task called to shutdown test.
  94  */
  95 static GNUNET_SCHEDULER_TaskIdentifier shutdown_handle;
  96
  97
  98 /**
  99  * Check whether peers successfully shut down.
 100  */
 101 static void
 102 shutdown_callback (void *cls, const char *emsg)
 103 {
 104   if (emsg != NULL)
 105   {
 106     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 107                 "test: Shutdown of peers failed! (%s)\n", emsg);
 108     ok--;
 109   }
 110 #if VERBOSE
 111   else
 112   {
 113     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 114                 "test: All peers successfully shut down!\n");
 115   }
 116 #endif
 117   GNUNET_CONFIGURATION_destroy (testing_cfg);
 118 }
 119
 120
 121 static void
 122 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 123 {
 124 #if VERBOSE
 125   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test: Ending test.\n");
 126 #endif
 127
 128   GNUNET_TESTING_daemons_stop (pg, TIMEOUT, &shutdown_callback, NULL);
 129 }
 130
 131
 132 static void
 133 disconnect_peers (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 134 {
 135   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test: disconnecting peers\n");
 136
 137   if (GNUNET_SCHEDULER_NO_TASK != shutdown_handle)
 138   {
 139     GNUNET_SCHEDULER_cancel (shutdown_handle);
 140     shutdown_handle = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL);
 141   }
 142 }
 143
 144
 145 /**
 146  * Prototype of a callback function indicating that two peers
 147  * are currently connected.
 148  *
 149  * @param cls closure
 150  * @param first peer id for first daemon
 151  * @param second peer id for the second daemon
 152  * @param distance distance between the connected peers
 153  * @param emsg error message (NULL on success)
 154  */
 155 void
 156 topo_cb (void *cls, const struct GNUNET_PeerIdentity *first,
 157          const struct GNUNET_PeerIdentity *second, const char *emsg)
 158 {
 159   topo_connections++;
 160   if (NULL != emsg)
 161   {
 162     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "test: Error by topo %u: %s\n",
 163                 topo_connections, emsg);
 164   }
 165   else
 166   {
 167     if (first == NULL || second == NULL)
 168     {
 169       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test: Connection %u NULL\n",
 170                   topo_connections);
 171       if (disconnect_task != GNUNET_SCHEDULER_NO_TASK)
 172       {
 173         GNUNET_SCHEDULER_cancel (disconnect_task);
 174         disconnect_task = GNUNET_SCHEDULER_add_now (&disconnect_peers, NULL);
 175       }
 176       return;
 177     }
 178     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test: Connection %u ok\n",
 179                 topo_connections);
 180     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test:   %s\n", GNUNET_i2s (first));
 181     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test:   %s\n", GNUNET_i2s (second));
 182   }
 183 }
 184
 185
 186 /**
 187  * peergroup_ready: start test when all peers are connected
 188  * @param cls closure
 189  * @param emsg error message
 190  */
 191 static void
 192 peergroup_ready (void *cls, const char *emsg)
 193 {
 194   if (emsg != NULL)
 195   {
 196     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 197                 "test: Peergroup callback called with error, aborting test!\n");
 198     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test: Error from testing: `%s'\n",
 199                 emsg);
 200     ok--;
 201     GNUNET_TESTING_daemons_stop (pg, TIMEOUT, &shutdown_callback, NULL);
 202     return;
 203   }
 204 #if VERBOSE
 205   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 206               "************************************************************\n");
 207   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 208               "test: Peer Group started successfully!\n");
 209   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test: Have %u connections\n",
 210               total_connections);
 211 #endif
 212
 213   peers_running = GNUNET_TESTING_daemons_running (pg);
 214   if (0 < failed_connections)
 215   {
 216     ok = GNUNET_SYSERR;
 217     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "test: %u connections have FAILED!\n",
 218                 failed_connections);
 219     disconnect_task = GNUNET_SCHEDULER_add_now (&disconnect_peers, NULL);
 220
 221   }
 222   else
 223   {
 224     GNUNET_TESTING_get_topology (pg, &topo_cb, NULL);
 225     disconnect_task =
 226         GNUNET_SCHEDULER_add_delayed (SHORT_TIME, &disconnect_peers, NULL);
 227     ok = GNUNET_OK;
 228   }
 229
 230 }
 231
 232
 233 /**
 234  * Function that will be called whenever two daemons are connected by
 235  * the testing library.
 236  *
 237  * @param cls closure
 238  * @param first peer id for first daemon
 239  * @param second peer id for the second daemon
 240  * @param distance distance between the connected peers
 241  * @param first_cfg config for the first daemon
 242  * @param second_cfg config for the second daemon
 243  * @param first_daemon handle for the first daemon
 244  * @param second_daemon handle for the second daemon
 245  * @param emsg error message (NULL on success)
 246  */
 247 static void
 248 connect_cb (void *cls, const struct GNUNET_PeerIdentity *first,
 249             const struct GNUNET_PeerIdentity *second, uint32_t distance,
 250             const struct GNUNET_CONFIGURATION_Handle *first_cfg,
 251             const struct GNUNET_CONFIGURATION_Handle *second_cfg,
 252             struct GNUNET_TESTING_Daemon *first_daemon,
 253             struct GNUNET_TESTING_Daemon *second_daemon, const char *emsg)
 254 {
 255   if (emsg == NULL)
 256   {
 257     total_connections++;
 258   }
 259   else
 260   {
 261     failed_connections++;
 262     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 263                 "test: Problem with new connection (%s)\n", emsg);
 264     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test:   (%s)\n", GNUNET_i2s (first));
 265     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test:   (%s)\n", GNUNET_i2s (second));
 266   }
 267
 268 }
 269
 270
 271 /**
 272  * run: load configuration options and schedule test to run (start peergroup)
 273  * @param cls closure
 274  * @param args argv
 275  * @param cfgfile configuration file name (can be NULL)
 276  * @param cfg configuration handle
 277  */
 278 static void
 279 run (void *cls, char *const *args, const char *cfgfile,
 280      const struct GNUNET_CONFIGURATION_Handle *cfg)
 281 {
 282   struct GNUNET_TESTING_Host *hosts;
 283
 284   ok = GNUNET_NO;
 285   total_connections = 0;
 286   failed_connections = 0;
 287   testing_cfg = GNUNET_CONFIGURATION_dup (cfg);
 288
 289   GNUNET_log_setup ("test_mesh_2dtorus",
 290 #if VERBOSE
 291                     "DEBUG",
 292 #else
 293                     "WARNING",
 294 #endif
 295                     NULL);
 296
 297 #if VERBOSE
 298   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test: Starting daemons.\n");
 299   GNUNET_CONFIGURATION_set_value_string (testing_cfg, "testing_old",
 300                                          "use_progressbars", "YES");
 301 #endif
 302
 303   if (GNUNET_OK !=
 304       GNUNET_CONFIGURATION_get_value_number (testing_cfg, "testing_old",
 305                                              "num_peers", &num_peers))
 306   {
 307     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 308                 "Option TESTING:NUM_PEERS is required!\n");
 309     return;
 310   }
 311
 312   hosts = GNUNET_TESTING_hosts_load (testing_cfg);
 313
 314   pg = GNUNET_TESTING_peergroup_start (testing_cfg, num_peers, TIMEOUT,
 315                                        &connect_cb, &peergroup_ready, NULL,
 316                                        hosts);
 317   GNUNET_assert (pg != NULL);
 318   shutdown_handle =
 319     GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
 320                                     &shutdown_task, NULL);
 321 }
 322
 323
 324 /**
 325  * test_mesh_2dtorus command line options
 326  */
 327 static struct GNUNET_GETOPT_CommandLineOption options[] = {
 328   {'V', "verbose", NULL,
 329    gettext_noop ("be verbose (print progress information)"),
 330    0, &GNUNET_GETOPT_set_one, &verbose},
 331   GNUNET_GETOPT_OPTION_END
 332 };
 333
 334
 335 /**
 336  * Main: start test
 337  */
 338 int
 339 main (int argc, char *argv[])
 340 {
 341   char *const argv2[] = {
 342     argv[0],
 343     "-c",
 344     "test_mesh_2dtorus.conf",
 345 #if VERBOSE
 346     "-L",
 347     "DEBUG",
 348 #endif
 349     NULL
 350   };
 351
 352   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test: Start\n");
 353
 354
 355   GNUNET_PROGRAM_run ((sizeof (argv2) / sizeof (char *)) - 1, argv2,
 356                       "test_mesh_2dtorus", gettext_noop ("Test mesh 2d torus."),
 357                       options, &run, NULL);
 358 #if REMOVE_DIR
 359   GNUNET_DISK_directory_remove ("/tmp/test_mesh_2dtorus");
 360 #endif
 361   if (GNUNET_OK != ok)
 362   {
 363     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "test: FAILED!\n");
 364     return 1;
 365   }
 366   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test: success\n");
 367   return 0;
 368 }
 369
 370 /* end of test_mesh_2dtorus.c */