src/testbed/test_testbed_underlay.c

   1 /*
   2       This file is part of GNUnet
   3       Copyright (C) 2008--2013 GNUnet e.V.
   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., 51 Franklin Street, Fifth Floor,
  18       Boston, MA 02110-1301, USA.
  19  */
  20
  21 /**
  22  * @file testbed/test_testbed_underlay.c
  23  * @brief testcase binary for testing testbed underlay restrictions
  24  * @author Sree Harsha Totakura <sreeharsha@totakura.in>
  25  */
  26
  27 #include "platform.h"
  28 #include "gnunet_util_lib.h"
  29 #include "gnunet_testbed_service.h"
  30
  31
  32 /**
  33  * Number of peers we start in this test case
  34  */
  35 #define NUM_PEERS 3
  36
  37 /**
  38  * Result of this test case
  39  */
  40 static int result;
  41
  42 static struct GNUNET_TESTBED_Operation *op;
  43
  44
  45 /**
  46  * Shutdown testcase
  47  *
  48  * @param cls NULL
  49  * @param tc scheduler task context
  50  */
  51 static void
  52 do_shutdown (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
  53 {
  54   if (NULL != op)
  55     GNUNET_TESTBED_operation_done (op);
  56   op = NULL;
  57 }
  58
  59
  60 /**
  61  * Callback to be called when an operation is completed
  62  *
  63  * @param cls the callback closure from functions generating an operation
  64  * @param op the operation that has been finished
  65  * @param emsg error message in case the operation has failed; will be NULL if
  66  *          operation has executed successfully.
  67  */
  68 static void
  69 overlay_connect_status (void *cls,
  70                         struct GNUNET_TESTBED_Operation *op_,
  71                         const char *emsg)
  72 {
  73   GNUNET_assert (op_ == op);
  74   GNUNET_TESTBED_operation_done (op);
  75   op = NULL;
  76   if (NULL == emsg)
  77     GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Peers 0 and 2 should not get connected\n");
  78   else
  79   {
  80     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Peers 0 and 2 not connected: %s.  Success!\n", emsg);
  81     result = GNUNET_OK;
  82   }
  83   GNUNET_SCHEDULER_shutdown ();
  84 }
  85
  86
  87
  88 /**
  89  * Signature of a main function for a testcase.
  90  *
  91  * @param cls closure
  92  * @param h the run handle
  93  * @param num_peers number of peers in 'peers'
  94  * @param peers_ handle to peers run in the testbed
  95  * @param links_succeeded the number of overlay link connection attempts that
  96  *          succeeded
  97  * @param links_failed the number of overlay link connection attempts that
  98  *          failed
  99  */
 100 static void
 101 test_master (void *cls,
 102              struct GNUNET_TESTBED_RunHandle *h,
 103              unsigned int num_peers,
 104              struct GNUNET_TESTBED_Peer **peers_,
 105              unsigned int links_succeeded,
 106              unsigned int links_failed)
 107 {
 108   GNUNET_assert (NULL == cls);
 109   if (NULL == peers_)
 110   {
 111     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Failing test due to timeout\n");
 112     GNUNET_SCHEDULER_shutdown ();
 113     return;
 114   }
 115   GNUNET_assert (NUM_PEERS == num_peers);
 116   op = GNUNET_TESTBED_overlay_connect (NULL,
 117                                        &overlay_connect_status,
 118                                        NULL,
 119                                        peers_[0],
 120                                        peers_[2]);
 121   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS,
 122                                                                60),
 123                                 &do_shutdown, NULL);
 124 }
 125
 126
 127 #ifndef PATH_MAX
 128 /**
 129  * Assumed maximum path length (for the log file name).
 130  */
 131 #define PATH_MAX 4096
 132 #endif
 133
 134
 135 /**
 136  * Main function
 137  */
 138 int
 139 main (int argc, char **argv)
 140 {
 141   struct GNUNET_CONFIGURATION_Handle *cfg;
 142   char pwd[PATH_MAX];
 143   char *dbfile;
 144   uint64_t event_mask;
 145
 146   result = GNUNET_SYSERR;
 147   event_mask = 0;
 148   cfg = GNUNET_CONFIGURATION_create ();
 149   GNUNET_assert (GNUNET_YES ==
 150                  GNUNET_CONFIGURATION_parse (cfg,
 151                                              "test_testbed_underlay.conf.in"));
 152   if (NULL == getcwd (pwd, PATH_MAX))
 153     return 1;
 154   GNUNET_assert (0 < GNUNET_asprintf (&dbfile, "%s/%s", pwd,
 155                                       "test-underlay.sqlite"));
 156   GNUNET_CONFIGURATION_set_value_string (cfg, "TESTBED-UNDERLAY","DBFILE", dbfile);
 157   GNUNET_assert (GNUNET_OK == GNUNET_CONFIGURATION_write
 158                  (cfg, "test_testbed_underlay.conf"));
 159   GNUNET_CONFIGURATION_destroy (cfg);
 160   cfg = NULL;
 161   GNUNET_free (dbfile);
 162   dbfile = NULL;
 163   (void) GNUNET_TESTBED_test_run ("test_testbed_underlay",
 164                                   "test_testbed_underlay.conf", NUM_PEERS,
 165                                   event_mask, NULL, NULL,
 166                                   &test_master, NULL);
 167   (void) unlink ("test_testbed_underlay.conf");
 168   if (GNUNET_OK != result)
 169     return 1;
 170   return 0;
 171 }