src/nat/test_nat_mini.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2009, 2011 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 /**
  22  * Testcase for port redirection and public IP address retrieval.
  23  * This test never fails, because there need to be a NAT box set up for tha *
  24  * @file nat/test_nat_mini.c
  25  * @brief Testcase for NAT library - mini
  26  * @author Christian Grothoff
  27  *
  28  * TODO: actually use ARM to start resolver service to make DNS work!
  29  */
  30
  31 #include "platform.h"
  32 #include "gnunet_common.h"
  33 #include "gnunet_util_lib.h"
  34 #include "gnunet_program_lib.h"
  35 #include "gnunet_scheduler_lib.h"
  36 #include "gnunet_nat_lib.h"
  37
  38
  39 #define VERBOSE GNUNET_NO
  40
  41 /* Time to wait before stopping NAT, in seconds */
  42 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5)
  43
  44 /**
  45  * Function called on each address that the NAT service
  46  * believes to be valid for the transport.
  47  */
  48 static void
  49 addr_callback (void *cls, int add_remove,
  50                const struct sockaddr *addr, socklen_t addrlen)
  51 {
  52   fprintf (stderr,
  53            "Address changed: %s `%s' (%u bytes)\n",
  54            add_remove == GNUNET_YES ? "added" : "removed",
  55            GNUNET_a2s (addr, addrlen),
  56            (unsigned int) addrlen);
  57 }
  58
  59
  60 /**
  61  * Function that terminates the test.
  62  */
  63 static void
  64 stop (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
  65 {
  66   struct GNUNET_NAT_MiniHandle *mini = cls;
  67
  68   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
  69               "Stopping NAT and quitting...\n");
  70   GNUNET_NAT_mini_map_stop (mini);
  71 }
  72
  73 #define PORT 10000
  74
  75 /**
  76  * Main function run with scheduler.
  77  */
  78 static void
  79 run (void *cls,
  80      char *const *args,
  81      const char *cfgfile, const struct GNUNET_CONFIGURATION_Handle *cfg)
  82 {
  83   struct GNUNET_NAT_MiniHandle *mini;
  84
  85   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
  86               "Requesting NAT redirection for port %u...\n",
  87               PORT);
  88   mini = GNUNET_NAT_mini_map_start (PORT,
  89                                     GNUNET_YES /* tcp */,
  90                                     &addr_callback, NULL);
  91   if (NULL == mini)
  92     {
  93       GNUNET_log (GNUNET_ERROR_TYPE_INFO,
  94                   "Could not start UPnP interaction\n");
  95       return;
  96     }
  97   GNUNET_SCHEDULER_add_delayed (TIMEOUT, &stop, mini);
  98 }
  99
 100
 101 int
 102 main (int argc, char *const argv[])
 103 {
 104   struct GNUNET_GETOPT_CommandLineOption options[] = {
 105     GNUNET_GETOPT_OPTION_END
 106   };
 107
 108   char *const argv_prog[] = {
 109     "test-nat-mini",
 110     "-c",
 111     "test_nat_data.conf",
 112     "-L",
 113 #if VERBOSE
 114     "DEBUG",
 115 #else
 116     "WARNING",
 117 #endif
 118     NULL
 119   };
 120
 121   GNUNET_log_setup ("test-nat-mini",
 122 #if VERBOSE
 123                     "DEBUG",
 124 #else
 125                     "WARNING",
 126 #endif
 127                     NULL);
 128
 129   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
 130               "UPnP test for NAT library, timeout set to %d seconds\n", TIMEOUT);
 131   GNUNET_PROGRAM_run (5, argv_prog, "test-nat-mini",
 132                       "nohelp", options, &run, NULL);
 133   return 0;
 134 }
 135
 136 /* end of test_nat_mini.c */