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