src/transport/test_plugin_transport.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 transport/test_plugin_transport.c
  22  * @brief testcase for transport_api.c
  23  * @author Sailor Siraj
  24  * @author Christian Grothoff
  25  */
  26
  27 #include "platform.h"
  28 #include "gnunet_constants.h"
  29 #include "gnunet_util_lib.h"
  30 #include "gnunet_hello_lib.h"
  31 #include "gnunet_peerinfo_service.h"
  32 #include "gnunet_protocols.h"
  33 #include "gnunet_signatures.h"
  34 #include "gnunet_transport_plugin.h"
  35 #include "transport.h"
  36
  37 #define VERBOSE GNUNET_NO
  38
  39 /**
  40  * How long until we give up on transmitting the message?
  41  */
  42 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30)
  43
  44 /**
  45  * Our public key.
  46  */
  47 static struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded my_public_key;
  48
  49 /**
  50  * Our identity.
  51  */
  52 static struct GNUNET_PeerIdentity my_identity;
  53
  54 /**
  55  * Our private key.
  56  */
  57 static struct GNUNET_CRYPTO_RsaPrivateKey *my_private_key;
  58
  59 /**
  60  * Our configuration.
  61  */
  62 const struct GNUNET_CONFIGURATION_Handle *cfg;
  63
  64 /**
  65  * Number of neighbours we'd like to have.
  66  */
  67 static uint32_t max_connect_per_transport;
  68
  69 /**
  70  * Environment for this plugin.
  71  */
  72 struct GNUNET_TRANSPORT_PluginEnvironment env;
  73
  74 /**
  75  *handle for the api provided by this plugin
  76  */
  77 struct GNUNET_TRANSPORT_PluginFunctions *api;
  78
  79 /**
  80  * Did the test pass or fail?
  81  */
  82 static int ok;
  83
  84 /**
  85  */
  86 static void
  87 receive (void *cls, const struct GNUNET_PeerIdentity *peer,
  88          const struct GNUNET_MessageHeader *message, uint32_t distance,
  89          const char *sender_address, size_t sender_address_len)
  90 {
  91   /* do nothing */
  92 }
  93
  94 void
  95 notify_address (void *cls, const char *name, const void *addr, size_t addrlen,
  96                 struct GNUNET_TIME_Relative expires)
  97 {
  98 }
  99
 100 /**
 101  * Function called when the service shuts
 102  * down.  Unloads our plugins.
 103  *
 104  * @param cls closure
 105  * @param cfg configuration to use
 106  */
 107 static void
 108 unload_plugins (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg)
 109 {
 110   GNUNET_assert (NULL ==
 111                  GNUNET_PLUGIN_unload ("libgnunet_plugin_transport_tcp", api));
 112   if (my_private_key != NULL)
 113     GNUNET_CRYPTO_rsa_key_free (my_private_key);
 114
 115 }
 116
 117
 118 static void
 119 unload_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 120 {
 121   struct GNUNET_CONFIGURATION_Handle *cfg = cls;
 122
 123   unload_plugins (NULL, cfg);
 124 }
 125
 126
 127 /**
 128  * Simple example test that invokes
 129  * the "validate" function of the plugin
 130  * and tries to see if the plugin would
 131  * succeed to validate its own address.
 132  * (This test is not well-written since
 133  *  we hand-compile the address which
 134  *  kind-of works for TCP but would not
 135  *  work for other plugins; we should ask
 136  *  the plugin about its address instead...).
 137  */
 138 /* FIXME: this is TCP/UDP-specific and won't work
 139    for HTTP/SMTP/DV; we should instead use an
 140    address that we get from the plugin itself
 141    (if it is willing/able to give us one...) */
 142 static void
 143 test_validation ()
 144 {
 145   struct sockaddr_in soaddr;
 146
 147   memset (&soaddr, 0, sizeof (soaddr));
 148 #if HAVE_SOCKADDR_IN_SIN_LEN
 149   soaddr.sin_len = sizeof (soaddr);
 150 #endif
 151   soaddr.sin_family = AF_INET;
 152   soaddr.sin_port = htons (2368 /* FIXME: get from config! */ );
 153   soaddr.sin_addr.s_addr = htonl (INADDR_LOOPBACK);
 154   GNUNET_assert (GNUNET_OK ==
 155                  api->check_address (api->cls, &soaddr, sizeof (soaddr)));
 156   ok = 0;
 157   GNUNET_SCHEDULER_add_continuation (&unload_task, (void *) cfg,
 158                                      GNUNET_SCHEDULER_REASON_PREREQ_DONE);
 159 }
 160
 161
 162 static void
 163 setup_plugin_environment ()
 164 {
 165   env.cfg = cfg;
 166   env.my_identity = &my_identity;
 167   env.cls = &env;
 168   env.receive = &receive;
 169   env.notify_address = &notify_address;
 170   env.max_connections = max_connect_per_transport;
 171 }
 172
 173
 174 /**
 175  * Runs the test.
 176  *
 177  * @param cls closure
 178  * @param c configuration to use
 179  */
 180 static void
 181 run (void *cls, char *const *args, const char *cfgfile,
 182      const struct GNUNET_CONFIGURATION_Handle *c)
 183 {
 184   unsigned long long tneigh;
 185   char *keyfile;
 186   char *libname;
 187
 188   cfg = c;
 189   /* parse configuration */
 190   if ((GNUNET_OK !=
 191        GNUNET_CONFIGURATION_get_value_number (c, "TRANSPORT", "NEIGHBOUR_LIMIT",
 192                                               &tneigh)) ||
 193       (GNUNET_OK !=
 194        GNUNET_CONFIGURATION_get_value_filename (c, "GNUNETD", "HOSTKEY",
 195                                                 &keyfile)))
 196   {
 197     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 198                 _
 199                 ("Transport service is lacking key configuration settings.  Exiting.\n"));
 200     GNUNET_SCHEDULER_shutdown (s);
 201     return;
 202   }
 203   max_connect_per_transport = (uint32_t) tneigh;
 204   my_private_key = GNUNET_CRYPTO_rsa_key_create_from_file (keyfile);
 205   GNUNET_free (keyfile);
 206   if (my_private_key == NULL)
 207   {
 208     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 209                 _("Transport service could not access hostkey.  Exiting.\n"));
 210     GNUNET_SCHEDULER_shutdown (s);
 211     return;
 212   }
 213   GNUNET_CRYPTO_rsa_key_get_public (my_private_key, &my_public_key);
 214   GNUNET_CRYPTO_hash (&my_public_key, sizeof (my_public_key),
 215                       &my_identity.hashPubKey);
 216
 217
 218
 219   /* load plugins... */
 220   setup_plugin_environment ();
 221   GNUNET_log (GNUNET_ERROR_TYPE_INFO, _("Loading tcp transport plugin\n"));
 222   GNUNET_asprintf (&libname, "libgnunet_plugin_transport_tcp");
 223
 224   api = GNUNET_PLUGIN_load (libname, &env);
 225   GNUNET_free (libname);
 226   if (api == NULL)
 227   {
 228     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 229                 _("Failed to load transport plugin for tcp\n"));
 230     /* FIXME: set some error code for main */
 231     return;
 232   }
 233   test_validation ();
 234 }
 235
 236
 237 /**
 238  * The main function for the transport service.
 239  *
 240  * @param argc number of arguments from the command line
 241  * @param argv command line arguments
 242  * @return 0 ok, 1 on error
 243  */
 244 int
 245 main (int argc, char *const *argv)
 246 {
 247   static struct GNUNET_GETOPT_CommandLineOption options[] = {
 248     GNUNET_GETOPT_OPTION_END
 249   };
 250   int ret;
 251
 252   char *const argv_prog[] = {
 253     "test_plugin_transport",
 254     "-c",
 255     "test_plugin_transport_data.conf",
 256     "-L",
 257 #if VERBOSE
 258     "DEBUG",
 259 #else
 260     "WARNING",
 261 #endif
 262     NULL
 263   };
 264   GNUNET_log_setup ("test-plugin-transport",
 265 #if VERBOSE
 266                     "DEBUG",
 267 #else
 268                     "WARNING",
 269 #endif
 270                     NULL);
 271   ok = 1;                       /* set to fail */
 272   ret =
 273       (GNUNET_OK ==
 274        GNUNET_PROGRAM_run (5, argv_prog, "test-plugin-transport", "testcase",
 275                            options, &run, NULL)) ? ok : 1;
 276   GNUNET_DISK_directory_remove ("/tmp/test-gnunetd-plugin-transport");
 277   return ret;
 278 }
 279
 280 /* end of test_plugin_transport.c */