src/transport/test_plugin_transport_http.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2010 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 2, 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 transport/test_plugin_transport_http.c
  22  * @brief testcase for plugin_transport_http.c
  23  * @author Matthias Wachs
  24  */
  25
  26 #include "platform.h"
  27 #include "gnunet_constants.h"
  28 #include "gnunet_getopt_lib.h"
  29 #include "gnunet_hello_lib.h"
  30 #include "gnunet_os_lib.h"
  31 #include "gnunet_peerinfo_service.h"
  32 #include "gnunet_plugin_lib.h"
  33 #include "gnunet_protocols.h"
  34 #include "gnunet_program_lib.h"
  35 #include "gnunet_signatures.h"
  36 #include "gnunet_service_lib.h"
  37 #include "plugin_transport.h"
  38 #include "gnunet_statistics_service.h"
  39 #include "transport.h"
  40 #include <curl/curl.h>
  41
  42 #define VERBOSE GNUNET_YES
  43 #define DEBUG GNUNET_YES
  44
  45 #define PLUGIN libgnunet_plugin_transport_template
  46
  47 /**
  48  * How long until we give up on transmitting the message?
  49  */
  50 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30)
  51
  52 /**
  53  * How long until we give up on transmitting the message?
  54  */
  55 #define STAT_INTERVALL GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 3)
  56
  57 /**
  58  * Our public key.
  59  */
  60 /* static struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded my_public_key; */
  61
  62 /**
  63  * Our identity.
  64  */
  65 static struct GNUNET_PeerIdentity my_identity;
  66
  67 #if 0
  68 /**
  69  * Our private key.
  70  */
  71 static struct GNUNET_CRYPTO_RsaPrivateKey *my_private_key;
  72 #endif
  73
  74 /**
  75  * Our scheduler.
  76  */
  77 struct GNUNET_SCHEDULER_Handle *sched;
  78
  79 /**
  80  * Our statistics handle.
  81  */
  82 struct GNUNET_STATISTICS_Handle *stats;
  83
  84
  85 /**
  86  * Our configuration.
  87  */
  88 const struct GNUNET_CONFIGURATION_Handle *cfg;
  89
  90 /**
  91  * Number of neighbours we'd like to have.
  92  */
  93 static uint32_t max_connect_per_transport;
  94
  95 /**
  96  * Environment for this plugin.
  97  */
  98 static struct GNUNET_TRANSPORT_PluginEnvironment env;
  99
 100 /**
 101  *handle for the api provided by this plugin
 102  */
 103 static struct GNUNET_TRANSPORT_PluginFunctions *api;
 104
 105 /**
 106  * ID of the task controlling the locking between two hostlist tests
 107  */
 108 static GNUNET_SCHEDULER_TaskIdentifier ti_check_stat;
 109
 110 static struct GNUNET_STATISTICS_GetHandle * stat_get_handle;
 111
 112 static unsigned int timeout_count;
 113
 114 /**
 115  * Did the test pass or fail?
 116  */
 117 static int fail;
 118
 119 pid_t pid;
 120
 121 /**
 122  * Initialize Environment for this plugin
 123  */
 124 static struct GNUNET_TIME_Relative
 125 receive (void *cls,
 126          const struct GNUNET_PeerIdentity * peer,
 127          const struct GNUNET_MessageHeader * message,
 128          uint32_t distance,
 129          struct Session *session,
 130          const char *sender_address,
 131          uint16_t sender_address_len)
 132 {
 133   /* do nothing */
 134   return GNUNET_TIME_UNIT_ZERO;
 135 }
 136
 137 void
 138 notify_address (void *cls,
 139                 const char *name,
 140                 const void *addr,
 141                 uint16_t addrlen,
 142                 struct GNUNET_TIME_Relative expires)
 143 {
 144
 145 }
 146
 147 /**
 148  * Simple example test that invokes
 149  * the check_address function of the plugin.
 150  */
 151 /* FIXME: won't work on IPv6 enabled systems where IPv4 mapping
 152  * isn't enabled (eg. FreeBSD > 4)
 153  */
 154 static void
 155 shutdown_clean ()
 156 {
 157   if (stat_get_handle != NULL)
 158   {
 159     GNUNET_STATISTICS_get_cancel(stat_get_handle);
 160   }
 161
 162   GNUNET_STATISTICS_destroy (stats, GNUNET_YES);
 163   if (ti_check_stat != GNUNET_SCHEDULER_NO_TASK)
 164     GNUNET_SCHEDULER_cancel(sched,ti_check_stat);
 165   ti_check_stat = GNUNET_SCHEDULER_NO_TASK;
 166
 167   GNUNET_assert (NULL == GNUNET_PLUGIN_unload ("libgnunet_plugin_transport_template", api));
 168   GNUNET_SCHEDULER_shutdown(sched);
 169   /* FIXME: */ fail = GNUNET_NO;
 170   return;
 171 }
 172
 173 static void
 174 setup_plugin_environment ()
 175 {
 176   env.cfg = cfg;
 177   env.sched = sched;
 178   env.stats = stats;
 179   env.my_identity = &my_identity;
 180   env.cls = &env;
 181   env.receive = &receive;
 182   env.notify_address = &notify_address;
 183   env.max_connections = max_connect_per_transport;
 184 }
 185
 186 static int
 187 process_stat (void *cls,
 188               const char *subsystem,
 189               const char *name,
 190               uint64_t value,
 191               int is_persistent)
 192 {
 193   stat_get_handle = NULL;
 194   if (value==1)
 195     {
 196     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Shutdown, plugin failed \n");
 197     fail = GNUNET_YES;
 198     shutdown_clean();
 199     return GNUNET_YES;
 200     }
 201   if (value==2)
 202     {
 203     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Shutdown, plugin not failed \n");
 204     shutdown_clean();
 205     return GNUNET_YES;
 206     }
 207   return GNUNET_YES;
 208 }
 209
 210 static void
 211 cont_func (void *cls, int success)
 212 {
 213   stat_get_handle = NULL;
 214 }
 215
 216 /**
 217  * Task that checks if we should try to download a hostlist.
 218  * If so, we initiate the download, otherwise we schedule
 219  * this task again for a later time.
 220  */
 221 static void
 222 task_check_stat (void *cls,
 223             const struct GNUNET_SCHEDULER_TaskContext *tc)
 224 {
 225   ti_check_stat = GNUNET_SCHEDULER_NO_TASK;
 226   if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
 227     return;
 228
 229   if ( timeout_count > 2 )
 230   {
 231     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Testcase timeout\n",  timeout_count);
 232     fail = GNUNET_YES;
 233     shutdown_clean();
 234     return;
 235   }
 236   timeout_count++;
 237
 238   stat_get_handle = GNUNET_STATISTICS_get (stats,
 239                                            "http-transport",
 240                                            gettext_noop("shutdown"),
 241                                            GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 2),
 242                                            &cont_func,
 243                                            &process_stat,
 244                                            NULL);
 245
 246   ti_check_stat = GNUNET_SCHEDULER_add_delayed (sched, STAT_INTERVALL, &task_check_stat, NULL);
 247   return;
 248 }
 249
 250 /**
 251  * Runs the test.
 252  *
 253  * @param cls closure
 254  * @param s scheduler to use
 255  * @param c configuration to use
 256  */
 257 static void
 258 run (void *cls,
 259      struct GNUNET_SCHEDULER_Handle *s,
 260      char *const *args,
 261      const char *cfgfile, const struct GNUNET_CONFIGURATION_Handle *c)
 262 {
 263   char * libname;
 264
 265   sched = s;
 266   cfg = c;
 267
 268   /* settings up statistics */
 269   stats = GNUNET_STATISTICS_create (sched, "http-transport", cfg);
 270   if (NULL == stats)
 271   {
 272     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 273                 _("Failed to retrieve statistics handle\n"));
 274     fail = GNUNET_YES;
 275     shutdown_clean();
 276     return ;
 277   }
 278
 279   /* load plugins... */
 280   setup_plugin_environment ();
 281   GNUNET_log (GNUNET_ERROR_TYPE_INFO, _("Loading HTTP transport plugin `%s'\n"),"libgnunet_plugin_transport_template");
 282   GNUNET_asprintf (&libname, "libgnunet_plugin_transport_template");
 283   api = GNUNET_PLUGIN_load (libname, &env);
 284   GNUNET_free (libname);
 285   if (api == NULL)
 286   {
 287     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 288                 _("Failed to load transport plugin for udp\n"));
 289     fail = GNUNET_YES;
 290     return;
 291   }
 292
 293   ti_check_stat = GNUNET_SCHEDULER_add_now (sched, &task_check_stat, NULL);
 294 }
 295
 296
 297 /**
 298  * The main function for the transport service.
 299  *
 300  * @param argc number of arguments from the command line
 301  * @param argv command line arguments
 302  * @return 0 ok, 1 on error
 303  */
 304 int
 305 main (int argc, char *const *argv)
 306 {
 307   static struct GNUNET_GETOPT_CommandLineOption options[] = {
 308     GNUNET_GETOPT_OPTION_END
 309   };
 310   int ret;
 311   char *const argv_prog[] = {
 312     "test_plugin_transport_http",
 313     "-c",
 314     "test_plugin_transport_data_http.conf",
 315     "-L",
 316 #if VERBOSE
 317     "DEBUG",
 318 #else
 319     "WARNING",
 320 #endif
 321     NULL
 322   };
 323   GNUNET_log_setup ("test_plugin_transport_http",
 324 #if VERBOSE
 325                     "DEBUG",
 326 #else
 327                     "WARNING",
 328 #endif
 329                     NULL);
 330   GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Starting statistics service\n");
 331   pid = GNUNET_OS_start_process (NULL, NULL, "gnunet-service-statistics",
 332                                  "gnunet-service-statistics",
 333                                  "-L", "DEBUG",
 334                                  "-c", "test_plugin_transport_data_http.conf", NULL);
 335
 336   fail = GNUNET_NO;
 337   if (pid==-1 )
 338   {
 339     GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Failed to start statistics service\n");
 340     fail = GNUNET_YES;
 341     return fail;
 342   }
 343
 344   ret = (GNUNET_OK ==
 345          GNUNET_PROGRAM_run (5,
 346                              argv_prog,
 347                              "test_plugin_transport_http",
 348                              "testcase", options, &run, NULL)) ? GNUNET_NO : GNUNET_YES;
 349   GNUNET_DISK_directory_remove ("/tmp/test_plugin_transport_http");
 350
 351   if (0 != PLIBC_KILL (pid, SIGTERM))
 352   {
 353     GNUNET_log_strerror (GNUNET_ERROR_TYPE_DEBUG, "Failed to kill statistics service");
 354     fail = GNUNET_YES;
 355   }
 356   if (GNUNET_OS_process_wait(pid) != GNUNET_OK)
 357     GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "waitpid");
 358   GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Killed statistics service\n");
 359   return fail;
 360 }
 361
 362 /* end of test_plugin_transport_http.c */