4e708c41323f3ecae69722be5a790fd42ddf2c43
[oweals/gnunet.git] / 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_YES
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   GNUNET_log (GNUNET_ERROR_TYPE_INFO, 
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_setup ("test-nat-mini", "DEBUG", NULL);
86   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
87               "Requesting NAT redirection for port %u...\n",
88               PORT);
89   mini = GNUNET_NAT_mini_map_start (PORT,
90                                     GNUNET_YES /* tcp */,
91                                     &addr_callback, NULL);
92   if (NULL == mini)
93     {
94       GNUNET_log (GNUNET_ERROR_TYPE_INFO,
95                   "Could not start UPnP interaction\n");
96       return;
97     }
98   GNUNET_SCHEDULER_add_delayed (TIMEOUT, &stop, mini);
99 }
100
101
102 int
103 main (int argc, char *const argv[])
104 {
105   struct GNUNET_GETOPT_CommandLineOption options[] = {
106     GNUNET_GETOPT_OPTION_END
107   };
108
109   char *const argv_prog[] = {
110     "test-nat-mini",
111     "-c",
112     "test_nat_data.conf",
113     "-L",
114 #if VERBOSE
115     "DEBUG",
116 #else
117     "WARNING",
118 #endif
119     NULL
120   };
121
122   GNUNET_log_setup ("test-nat-mini",
123 #if VERBOSE
124                     "DEBUG",
125 #else
126                     "WARNING",
127 #endif
128                     NULL);
129
130   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
131               "UPnP test for NAT library, timeout set to %d seconds\n", TIMEOUT);
132   GNUNET_PROGRAM_run (5, argv_prog, "test-nat-mini",
133                       "nohelp", options, &run, NULL);
134   return 0;
135 }
136
137 /* end of test_nat_mini.c */