diff options
Diffstat (limited to 'src/transport/test_transport_api_timeout.c')
-rw-r--r-- | src/transport/test_transport_api_timeout.c | 349 |
1 files changed, 349 insertions, 0 deletions
diff --git a/src/transport/test_transport_api_timeout.c b/src/transport/test_transport_api_timeout.c new file mode 100644 index 000000000..cec60e989 --- /dev/null +++ b/src/transport/test_transport_api_timeout.c | |||
@@ -0,0 +1,349 @@ | |||
1 | /* | ||
2 | This file is part of GNUnet. | ||
3 | (C) 2009, 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_timeout.c | ||
22 | * @brief test case for transport plugin implementations complying timeout | ||
23 | * settings | ||
24 | * | ||
25 | * | ||
26 | * This test case serves ensures that no peer disconnect events occurs | ||
27 | * while plugins are idle | ||
28 | */ | ||
29 | |||
30 | #include "platform.h" | ||
31 | #include "gnunet_common.h" | ||
32 | #include "gnunet_hello_lib.h" | ||
33 | #include "gnunet_getopt_lib.h" | ||
34 | #include "gnunet_os_lib.h" | ||
35 | #include "gnunet_program_lib.h" | ||
36 | #include "gnunet_scheduler_lib.h" | ||
37 | #include "gnunet_transport_service.h" | ||
38 | #include "transport.h" | ||
39 | #include "transport-testing.h" | ||
40 | |||
41 | #define VERBOSE GNUNET_NO | ||
42 | |||
43 | #define VERBOSE_ARM GNUNET_NO | ||
44 | |||
45 | #define START_ARM GNUNET_YES | ||
46 | |||
47 | /** | ||
48 | * How long until we give up on transmitting the message? | ||
49 | */ | ||
50 | #define WAIT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30) | ||
51 | |||
52 | #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 90) | ||
53 | |||
54 | #define MTYPE 12345 | ||
55 | |||
56 | static char *test_source; | ||
57 | |||
58 | static char *test_plugin; | ||
59 | |||
60 | static char *test_name; | ||
61 | |||
62 | static int ok; | ||
63 | |||
64 | static GNUNET_SCHEDULER_TaskIdentifier die_task; | ||
65 | |||
66 | static GNUNET_SCHEDULER_TaskIdentifier timer_task; | ||
67 | |||
68 | struct PeerContext *p1; | ||
69 | |||
70 | struct PeerContext *p2; | ||
71 | |||
72 | static GNUNET_TRANSPORT_TESTING_ConnectRequest cc; | ||
73 | |||
74 | struct GNUNET_TRANSPORT_TransmitHandle *th; | ||
75 | |||
76 | char *cfg_file_p1; | ||
77 | |||
78 | char *cfg_file_p2; | ||
79 | |||
80 | static struct GNUNET_TIME_Relative time_running; | ||
81 | |||
82 | static int shutdown_flag; | ||
83 | |||
84 | static int disconnects; | ||
85 | |||
86 | |||
87 | #if VERBOSE | ||
88 | #define OKPP do { ok++; fprintf (stderr, "Now at stage %u at %s:%u\n", ok, __FILE__, __LINE__); } while (0) | ||
89 | #else | ||
90 | #define OKPP do { ok++; } while (0) | ||
91 | #endif | ||
92 | |||
93 | |||
94 | static void | ||
95 | end () | ||
96 | { | ||
97 | GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Stopping peers\n"); | ||
98 | |||
99 | if (timer_task != GNUNET_SCHEDULER_NO_TASK) | ||
100 | { | ||
101 | GNUNET_SCHEDULER_cancel (timer_task); | ||
102 | timer_task = GNUNET_SCHEDULER_NO_TASK; | ||
103 | } | ||
104 | |||
105 | if (die_task != GNUNET_SCHEDULER_NO_TASK) | ||
106 | { | ||
107 | GNUNET_SCHEDULER_cancel (die_task); | ||
108 | die_task = GNUNET_SCHEDULER_NO_TASK; | ||
109 | } | ||
110 | |||
111 | if (th != NULL) | ||
112 | GNUNET_TRANSPORT_notify_transmit_ready_cancel (th); | ||
113 | th = NULL; | ||
114 | |||
115 | |||
116 | GNUNET_TRANSPORT_TESTING_stop_peer (p1); | ||
117 | GNUNET_TRANSPORT_TESTING_stop_peer (p2); | ||
118 | |||
119 | if (disconnects == 0) | ||
120 | ok = 0; | ||
121 | else | ||
122 | { | ||
123 | ok = disconnects; | ||
124 | GNUNET_log (GNUNET_ERROR_TYPE_ERROR, | ||
125 | "Fail! Had %u disconnects while waiting %llu seconds \n", | ||
126 | disconnects, WAIT.rel_value); | ||
127 | } | ||
128 | } | ||
129 | |||
130 | static void | ||
131 | end_badly (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
132 | { | ||
133 | die_task = GNUNET_SCHEDULER_NO_TASK; | ||
134 | |||
135 | GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Fail! Stopping peers\n"); | ||
136 | |||
137 | if (timer_task != GNUNET_SCHEDULER_NO_TASK) | ||
138 | { | ||
139 | GNUNET_SCHEDULER_cancel (timer_task); | ||
140 | timer_task = GNUNET_SCHEDULER_NO_TASK; | ||
141 | } | ||
142 | GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "1\n"); | ||
143 | if (cc != NULL) | ||
144 | GNUNET_TRANSPORT_TESTING_connect_peers_cancel (cc); | ||
145 | GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "2\n"); | ||
146 | if (th != NULL) | ||
147 | GNUNET_TRANSPORT_notify_transmit_ready_cancel (th); | ||
148 | th = NULL; | ||
149 | GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "3\n"); | ||
150 | if (p1 != NULL) | ||
151 | GNUNET_TRANSPORT_TESTING_stop_peer (p1); | ||
152 | GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "4\n"); | ||
153 | if (p2 != NULL) | ||
154 | GNUNET_TRANSPORT_TESTING_stop_peer (p2); | ||
155 | GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "5\n"); | ||
156 | ok = GNUNET_SYSERR; | ||
157 | } | ||
158 | |||
159 | |||
160 | static void | ||
161 | notify_receive (void *cls, const struct GNUNET_PeerIdentity *peer, | ||
162 | const struct GNUNET_MessageHeader *message, | ||
163 | const struct GNUNET_TRANSPORT_ATS_Information *ats, | ||
164 | uint32_t ats_count) | ||
165 | { | ||
166 | GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, | ||
167 | "Received message of type %d from peer %s!\n", | ||
168 | ntohs (message->type), GNUNET_i2s (peer)); | ||
169 | } | ||
170 | |||
171 | static void | ||
172 | notify_connect (void *cls, const struct GNUNET_PeerIdentity *peer, | ||
173 | const struct GNUNET_TRANSPORT_ATS_Information *ats, | ||
174 | uint32_t ats_count) | ||
175 | { | ||
176 | GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Peer `%4s' connected to us (%p)!\n", | ||
177 | GNUNET_i2s (peer), cls); | ||
178 | } | ||
179 | |||
180 | |||
181 | static void | ||
182 | notify_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer) | ||
183 | { | ||
184 | if (shutdown_flag != GNUNET_YES) | ||
185 | { | ||
186 | GNUNET_log (GNUNET_ERROR_TYPE_ERROR, | ||
187 | "FAIL! Peer `%4s' disconnected during waiting period!\n", | ||
188 | GNUNET_i2s (peer)); | ||
189 | disconnects++; | ||
190 | } | ||
191 | } | ||
192 | |||
193 | static void | ||
194 | timer (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) | ||
195 | { | ||
196 | static int percentage; | ||
197 | |||
198 | timer_task = GNUNET_SCHEDULER_NO_TASK; | ||
199 | |||
200 | if ((tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) != 0) | ||
201 | return; | ||
202 | |||
203 | percentage += 10; | ||
204 | time_running = | ||
205 | GNUNET_TIME_relative_add (time_running, | ||
206 | GNUNET_TIME_relative_divide (WAIT, 10)); | ||
207 | |||
208 | if (time_running.rel_value == | ||
209 | GNUNET_TIME_relative_max (time_running, WAIT).rel_value) | ||
210 | { | ||
211 | fprintf (stderr, "100%%\n"); | ||
212 | shutdown_flag = GNUNET_YES; | ||
213 | GNUNET_SCHEDULER_add_now (&end, NULL); | ||
214 | } | ||
215 | else | ||
216 | { | ||
217 | fprintf (stderr, "%u%%..", percentage); | ||
218 | timer_task = | ||
219 | GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_divide (WAIT, 10), | ||
220 | &timer, NULL); | ||
221 | } | ||
222 | } | ||
223 | |||
224 | static void | ||
225 | testing_connect_cb (struct PeerContext *p1, struct PeerContext *p2, void *cls) | ||
226 | { | ||
227 | cc = NULL; | ||
228 | char *p1_c = strdup (GNUNET_i2s (&p1->id)); | ||
229 | |||
230 | GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Peers connected: %s <-> %s\n", p1_c, | ||
231 | GNUNET_i2s (&p2->id)); | ||
232 | GNUNET_free (p1_c); | ||
233 | |||
234 | shutdown_flag = GNUNET_NO; | ||
235 | |||
236 | // FIXME: THIS IS REQUIRED! SEEMS TO BE A BUG! | ||
237 | timer_task = | ||
238 | GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_divide (WAIT, 10), | ||
239 | &timer, NULL); | ||
240 | } | ||
241 | |||
242 | static void | ||
243 | run (void *cls, char *const *args, const char *cfgfile, | ||
244 | const struct GNUNET_CONFIGURATION_Handle *cfg) | ||
245 | { | ||
246 | die_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT, &end_badly, NULL); | ||
247 | |||
248 | p1 = GNUNET_TRANSPORT_TESTING_start_peer (cfg_file_p1, ¬ify_receive, | ||
249 | ¬ify_connect, ¬ify_disconnect, | ||
250 | NULL); | ||
251 | p2 = GNUNET_TRANSPORT_TESTING_start_peer (cfg_file_p2, ¬ify_receive, | ||
252 | ¬ify_connect, ¬ify_disconnect, | ||
253 | NULL); | ||
254 | |||
255 | if ((p1 == NULL) || (p2 == NULL)) | ||
256 | { | ||
257 | GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Fail! Could not start peers!\n"); | ||
258 | if (die_task != GNUNET_SCHEDULER_NO_TASK) | ||
259 | GNUNET_SCHEDULER_cancel (die_task); | ||
260 | die_task = GNUNET_SCHEDULER_add_now (&end_badly, NULL); | ||
261 | return; | ||
262 | } | ||
263 | |||
264 | cc = GNUNET_TRANSPORT_TESTING_connect_peers (p1, p2, &testing_connect_cb, | ||
265 | NULL); | ||
266 | } | ||
267 | |||
268 | static int | ||
269 | check () | ||
270 | { | ||
271 | static char *const argv[] = { "test-transport-api-multiaddress", | ||
272 | "-c", | ||
273 | "test_transport_api_data.conf", | ||
274 | #if VERBOSE | ||
275 | "-L", "DEBUG", | ||
276 | #endif | ||
277 | NULL | ||
278 | }; | ||
279 | static struct GNUNET_GETOPT_CommandLineOption options[] = { | ||
280 | GNUNET_GETOPT_OPTION_END | ||
281 | }; | ||
282 | |||
283 | #if WRITECONFIG | ||
284 | setTransportOptions ("test_transport_api_data.conf"); | ||
285 | #endif | ||
286 | timer_task = GNUNET_SCHEDULER_NO_TASK; | ||
287 | |||
288 | ok = 1; | ||
289 | GNUNET_PROGRAM_run ((sizeof (argv) / sizeof (char *)) - 1, argv, | ||
290 | "test-transport-api-multiaddress", "nohelp", options, | ||
291 | &run, &ok); | ||
292 | |||
293 | return ok; | ||
294 | } | ||
295 | |||
296 | int | ||
297 | main (int argc, char *argv[]) | ||
298 | { | ||
299 | int ret; | ||
300 | int nat_res; | ||
301 | |||
302 | GNUNET_TRANSPORT_TESTING_get_test_source_name (__FILE__, &test_source); | ||
303 | GNUNET_TRANSPORT_TESTING_get_test_plugin_name (argv[0], test_source, | ||
304 | &test_plugin); | ||
305 | GNUNET_TRANSPORT_TESTING_get_test_name (argv[0], &test_name); | ||
306 | |||
307 | GNUNET_log_setup (test_name, | ||
308 | #if VERBOSE | ||
309 | "DEBUG", | ||
310 | #else | ||
311 | "WARNING", | ||
312 | #endif | ||
313 | NULL); | ||
314 | |||
315 | if ((strcmp (test_plugin, "tcp_nat") == 0) || | ||
316 | (strcmp (test_plugin, "udp_nat") == 0)) | ||
317 | { | ||
318 | nat_res = GNUNET_OS_check_helper_binary ("gnunet-nat-server"); | ||
319 | if (GNUNET_NO == nat_res) | ||
320 | { | ||
321 | GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Cannot run NAT test: `%s' %s \n", | ||
322 | "gnunet-nat-server", "SUID not set"); | ||
323 | return 0; | ||
324 | } | ||
325 | if (GNUNET_SYSERR == nat_res) | ||
326 | { | ||
327 | GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Cannot run NAT test: `%s' %s \n", | ||
328 | "gnunet-nat-server", "file not found"); | ||
329 | return 0; | ||
330 | } | ||
331 | } | ||
332 | |||
333 | GNUNET_TRANSPORT_TESTING_get_config_name (argv[0], &cfg_file_p1, 1); | ||
334 | GNUNET_TRANSPORT_TESTING_get_config_name (argv[0], &cfg_file_p2, 2); | ||
335 | |||
336 | ret = check (); | ||
337 | |||
338 | GNUNET_free (cfg_file_p1); | ||
339 | GNUNET_free (cfg_file_p2); | ||
340 | |||
341 | GNUNET_free (test_source); | ||
342 | GNUNET_free (test_plugin); | ||
343 | GNUNET_free (test_name); | ||
344 | |||
345 | |||
346 | return ret; | ||
347 | } | ||
348 | |||
349 | /* end of test_transport_api_timeout.c*/ | ||