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