7fc9d43b4e7eb0637f5ed9ef15efa633dae1063d
[oweals/gnunet.git] / src / transport / test_transport_api_reliability.c
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_reliability.c
22  * @brief base test case for transport implementations
23  *
24  * This test case serves as a base for tcp and http
25  * transport test cases to check that the transports
26  * achieve reliable message delivery.
27  */
28 #include "platform.h"
29 #include "gnunet_common.h"
30 #include "gnunet_hello_lib.h"
31 #include "gnunet_getopt_lib.h"
32 #include "gnunet_os_lib.h"
33 #include "gnunet_program_lib.h"
34 #include "gnunet_scheduler_lib.h"
35 #include "gnunet_server_lib.h"
36 #include "gnunet_transport_service.h"
37 #include "gauger.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  * Note that this value must not significantly exceed
49  * 'MAX_PENDING' in 'gnunet-service-transport.c', otherwise
50  * messages may be dropped even for a reliable transport.
51  */
52 #define TOTAL_MSGS (10000 * 2)
53
54 /**
55  * How long until we give up on transmitting the message?
56  */
57 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 1500)
58
59 #define MTYPE 12345
60
61 static struct PeerContext p1;
62
63 static struct PeerContext p2;
64
65 static int ok;
66
67 static int is_tcp;
68
69 static int is_tcp_nat;
70
71 static int is_http;
72
73 static int is_https;
74
75 static int is_udp;
76
77 static int is_unix;
78
79 static int is_wlan;
80
81 static int connected;
82
83 static unsigned long long total_bytes;
84
85 static struct GNUNET_TIME_Absolute start_time;
86
87 static GNUNET_SCHEDULER_TaskIdentifier die_task;
88
89 static GNUNET_SCHEDULER_TaskIdentifier tct;
90
91 struct GNUNET_TRANSPORT_TransmitHandle * th_p2;
92
93 static char * key_file_p1;
94 static char * cert_file_p1;
95
96 static char * key_file_p2;
97 static char * cert_file_p2;
98 static char *test_name;
99 static int msg_scheduled;
100 static int msg_sent;
101 static int msg_recv_expected;
102 static int msg_recv;
103
104 static int p1_hello_canceled;
105 static int p2_hello_canceled;
106
107 #if VERBOSE
108 #define OKPP do { ok++; fprintf (stderr, "Now at stage %u at %s:%u\n", ok, __FILE__, __LINE__); } while (0)
109 #else
110 #define OKPP do { ok++; } while (0)
111 #endif
112
113
114 static void
115 end ()
116 {
117   unsigned long long delta;
118   char *value_name;
119
120   if (die_task != GNUNET_SCHEDULER_NO_TASK)
121     GNUNET_SCHEDULER_cancel (die_task);
122   die_task = GNUNET_SCHEDULER_NO_TASK;
123 #if VERBOSE
124   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Disconnecting from transports!\n");
125 #endif
126
127   if (th_p2 != NULL)
128     GNUNET_TRANSPORT_notify_transmit_ready_cancel(th_p2);
129   th_p2 = NULL;
130
131   GNUNET_TRANSPORT_disconnect (p1.th);
132   GNUNET_TRANSPORT_disconnect (p2.th);
133 #if VERBOSE
134   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
135               "Transports disconnected, returning success!\n");
136 #endif
137   delta = GNUNET_TIME_absolute_get_duration (start_time).rel_value;
138   fprintf (stderr,
139            "\nThroughput was %llu kb/s\n",
140            total_bytes * 1000 / 1024 / delta);
141   GNUNET_asprintf(&value_name, "reliable_%s", test_name);
142   GAUGER ("TRANSPORT", value_name, (int)(total_bytes * 1000 / 1024 /delta), "kb/s");
143   GNUNET_free(value_name);
144   ok = 0;
145
146 }
147
148
149
150 static void
151 stop_arm (struct PeerContext *p)
152 {
153 #if START_ARM
154   if (0 != GNUNET_OS_process_kill (p->arm_proc, SIGTERM))
155     GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "kill");
156   GNUNET_OS_process_wait (p->arm_proc);
157   GNUNET_OS_process_close (p->arm_proc);
158   p->arm_proc = NULL;
159 #endif
160   GNUNET_CONFIGURATION_destroy (p->cfg);
161 }
162
163
164 static void
165 end_badly (void *cls,
166            const struct GNUNET_SCHEDULER_TaskContext *tc)
167 {
168   GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
169               "Reliability failed: Last message sent %u, Next message scheduled %u, Last message received %u, Message expected %u\n",
170               msg_sent, 
171               msg_scheduled, 
172               msg_recv, 
173               msg_recv_expected);
174   if (th_p2 != NULL)
175     GNUNET_TRANSPORT_notify_transmit_ready_cancel(th_p2);
176   th_p2 = NULL;
177
178   GNUNET_break (0);
179   GNUNET_TRANSPORT_disconnect (p1.th);
180   GNUNET_TRANSPORT_disconnect (p2.th);
181   if (GNUNET_SCHEDULER_NO_TASK != tct)
182     {
183       GNUNET_SCHEDULER_cancel (tct);
184       tct = GNUNET_SCHEDULER_NO_TASK;
185     }
186   ok = 1;
187 }
188
189
190 struct TestMessage
191 {
192   struct GNUNET_MessageHeader header;
193   uint32_t num;
194 };
195
196
197 static unsigned int
198 get_size (unsigned int iter)
199 {
200   unsigned int ret;
201
202   if (iter < 60000)
203     return iter + sizeof (struct TestMessage);
204   ret = (iter * iter * iter);
205   return sizeof (struct TestMessage) + (ret % 60000);
206 }
207
208
209 static void
210 notify_receive (void *cls,
211                 const struct GNUNET_PeerIdentity *peer,
212                 const struct GNUNET_MessageHeader *message,
213                 const struct GNUNET_TRANSPORT_ATS_Information *ats,
214                 uint32_t ats_count)
215 {
216   static int n;
217   unsigned int s;
218   char cbuf[GNUNET_SERVER_MAX_MESSAGE_SIZE - 1];
219   const struct TestMessage *hdr;
220
221   hdr = (const struct TestMessage*) message;
222   s = get_size (n);
223   if (MTYPE != ntohs (message->type))
224     return;
225   msg_recv_expected = n;
226   msg_recv = ntohl(hdr->num);
227   if (ntohs (message->size) != s)
228     {
229       GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
230                   "Expected message %u of size %u, got %u bytes of message %u\n",
231                   n, s,
232                   ntohs (message->size),
233                   ntohl (hdr->num));
234       if (die_task != GNUNET_SCHEDULER_NO_TASK)
235         GNUNET_SCHEDULER_cancel (die_task);
236       die_task = GNUNET_SCHEDULER_add_now (&end_badly, NULL);
237       return;
238     }
239   if (ntohl (hdr->num) != n)
240     {
241       GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
242                   "Expected message %u of size %u, got %u bytes of message %u\n",
243                   n, s,
244                   ntohs (message->size),
245                   ntohl (hdr->num));
246       if (die_task != GNUNET_SCHEDULER_NO_TASK)
247         GNUNET_SCHEDULER_cancel (die_task);
248       die_task = GNUNET_SCHEDULER_add_now (&end_badly, NULL);
249       return;
250     }
251   memset (cbuf, n, s - sizeof (struct TestMessage));
252   if (0 != memcmp (cbuf,
253                    &hdr[1],
254                    s - sizeof (struct TestMessage)))
255     {
256       GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
257                   "Expected message %u with bits %u, but body did not match\n",
258                   n, (unsigned char) n);
259       if (die_task != GNUNET_SCHEDULER_NO_TASK)
260         GNUNET_SCHEDULER_cancel (die_task);
261       die_task = GNUNET_SCHEDULER_add_now (&end_badly, NULL);
262       return;
263     }
264 #if VERBOSE
265   if (ntohl(hdr->num) % 5000 == 0)
266     {
267       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
268                   "Got message %u of size %u\n",
269                   ntohl (hdr->num),
270                   ntohs (message->size));
271     }
272 #endif
273   n++;
274   if (0 == (n % (TOTAL_MSGS/100)))
275     {
276       fprintf (stderr, ".");
277       if (die_task != GNUNET_SCHEDULER_NO_TASK)
278         GNUNET_SCHEDULER_cancel (die_task);
279       die_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT,
280                                                &end_badly,
281                                                NULL);
282     }
283   if (n == TOTAL_MSGS)
284     end ();
285 }
286
287
288 static size_t
289 notify_ready (void *cls, size_t size, void *buf)
290 {
291   static int n;
292   char *cbuf = buf;
293   struct TestMessage hdr;
294   unsigned int s;
295   unsigned int ret;
296
297   if (buf == NULL)
298     {
299       GNUNET_break (0);
300       ok = 42;
301       return 0;
302     }
303   th_p2 = NULL;
304   ret = 0;
305   s = get_size (n);
306   GNUNET_assert (size >= s);
307   GNUNET_assert (buf != NULL);
308   cbuf = buf;
309   do
310     {
311       hdr.header.size = htons (s);
312       hdr.header.type = htons (MTYPE);
313       hdr.num = htonl (n);
314       msg_sent = n;
315       memcpy (&cbuf[ret], &hdr, sizeof (struct TestMessage));
316       ret += sizeof (struct TestMessage);
317       memset (&cbuf[ret], n, s - sizeof (struct TestMessage));
318       ret += s - sizeof (struct TestMessage);
319 #if VERBOSE
320       if (n % 5000 == 0)
321         {
322           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
323                       "Sending message %u of size %u\n",
324                       n,
325                       s);
326         }
327 #endif
328       n++;
329       s = get_size (n);
330       if (0 == GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK, 16))
331         break; /* sometimes pack buffer full, sometimes not */
332     }
333   while (size - ret >= s);
334   if (n < TOTAL_MSGS)
335   {
336     if (th_p2 == NULL)
337       th_p2 = GNUNET_TRANSPORT_notify_transmit_ready (p2.th,
338                                             &p1.id,
339                                             s, 0, TIMEOUT,
340                                             &notify_ready,
341                                             NULL);
342     msg_scheduled = n;
343   }
344   if (n % 5000 == 0)
345     {
346       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
347                   "Returning total message block of size %u\n",
348                   ret);
349     }
350   total_bytes += ret;
351   return ret;
352 }
353
354
355 static void
356 notify_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer)
357 {
358   connected--;
359 #if VERBOSE
360   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
361               "Peer `%4s' disconnected (%p)!\n",
362               GNUNET_i2s (peer), cls);
363 #endif
364   if (th_p2 != NULL)
365     GNUNET_TRANSPORT_notify_transmit_ready_cancel(th_p2);
366 }
367
368
369 static void
370 exchange_hello_last (void *cls,
371                      const struct GNUNET_MessageHeader *message)
372 {
373   struct PeerContext *me = cls;
374
375   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
376               "Exchanging HELLO of size %d with peer (%s)!\n", 
377               (int) GNUNET_HELLO_size((const struct GNUNET_HELLO_Message *)message),
378               GNUNET_i2s (&me->id));
379   GNUNET_assert (message != NULL);
380   GNUNET_assert (GNUNET_OK ==
381                  GNUNET_HELLO_get_id ((const struct GNUNET_HELLO_Message *)
382                                       message, &me->id));
383   GNUNET_TRANSPORT_offer_hello (p1.th, message, NULL, NULL);
384 }
385
386
387
388 static void
389 exchange_hello (void *cls,
390                 const struct GNUNET_MessageHeader *message)
391 {
392   struct PeerContext *me = cls;
393
394   GNUNET_assert (message != NULL);
395   GNUNET_assert (GNUNET_OK ==
396                  GNUNET_HELLO_get_id ((const struct GNUNET_HELLO_Message *)
397                                       message, &me->id));
398   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
399               "Exchanging HELLO of size %d from peer %s!\n", 
400               (int) GNUNET_HELLO_size((const struct GNUNET_HELLO_Message *)message),
401               GNUNET_i2s (&me->id));
402   GNUNET_TRANSPORT_offer_hello (p2.th, message, NULL, NULL);
403 }
404
405
406 static void
407 notify_connect (void *cls,
408                 const struct GNUNET_PeerIdentity *peer,
409                 const struct GNUNET_TRANSPORT_ATS_Information *ats,
410                 uint32_t ats_count)
411 {
412   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
413               "Peer `%4s' connected to us (%p)!\n", 
414               GNUNET_i2s (peer), 
415               cls);
416   connected++;
417   if (cls == &p1)
418     {
419       GNUNET_TRANSPORT_set_quota (p1.th,
420                                   &p2.id,
421                                   GNUNET_BANDWIDTH_value_init (1024 * 1024 * 1024),
422                                   GNUNET_BANDWIDTH_value_init (1024 * 1024 * 1024));
423       start_time = GNUNET_TIME_absolute_get ();
424     }
425   else
426     {
427       GNUNET_TRANSPORT_set_quota (p2.th,
428                                   &p1.id,
429                                   GNUNET_BANDWIDTH_value_init (1024 * 1024 * 1024),
430                                   GNUNET_BANDWIDTH_value_init (1024 * 1024 * 1024));
431     }
432   if (2 == connected)
433     {
434       if (die_task != GNUNET_SCHEDULER_NO_TASK)
435         GNUNET_SCHEDULER_cancel (die_task);
436       if (tct != GNUNET_SCHEDULER_NO_TASK)
437         GNUNET_SCHEDULER_cancel (tct);
438       tct = GNUNET_SCHEDULER_NO_TASK;
439       if (p2_hello_canceled == GNUNET_NO)
440       {
441         GNUNET_TRANSPORT_get_hello_cancel (p2.th, &exchange_hello_last, &p2);
442         p2_hello_canceled = GNUNET_YES;
443       }
444       if (p1_hello_canceled == GNUNET_NO)
445       {
446         GNUNET_TRANSPORT_get_hello_cancel (p1.th, &exchange_hello, &p1);
447         p1_hello_canceled = GNUNET_YES;
448       }
449       die_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT,
450                                                &end_badly, NULL);
451       th_p2 = GNUNET_TRANSPORT_notify_transmit_ready (p2.th,
452                                               &p1.id,
453                                               get_size (0), 0, TIMEOUT,
454                                               &notify_ready,
455                                               NULL);
456       
457     }
458 }
459
460
461 static void
462 setup_peer (struct PeerContext *p, const char *cfgname)
463 {
464   p->cfg = GNUNET_CONFIGURATION_create ();
465   GNUNET_assert (GNUNET_OK == GNUNET_CONFIGURATION_load (p->cfg, cfgname));
466   if (GNUNET_CONFIGURATION_have_value (p->cfg,"PATHS", "SERVICEHOME"))
467       GNUNET_CONFIGURATION_get_value_string (p->cfg, "PATHS", "SERVICEHOME", &p->servicehome);
468   GNUNET_DISK_directory_remove (p->servicehome);
469
470 #if START_ARM
471   p->arm_proc = GNUNET_OS_start_process (NULL, NULL,
472                                         "gnunet-service-arm",
473                                         "gnunet-service-arm",
474 #if VERBOSE_ARM
475                                         "-L", "DEBUG",
476 #endif
477                                         "-c", cfgname, NULL);
478 #endif
479
480   if (is_https)
481     {
482       struct stat sbuf;
483       if (p==&p1)
484         {
485           if (GNUNET_CONFIGURATION_have_value (p->cfg,
486                                                "transport-https", "KEY_FILE"))
487             GNUNET_CONFIGURATION_get_value_string (p->cfg, "transport-https", "KEY_FILE", &key_file_p1);
488           if (key_file_p1 == NULL)
489             GNUNET_asprintf(&key_file_p1,"https_p1.key");
490           if (0 == stat (key_file_p1, &sbuf ))
491             {
492               if (0 == remove(key_file_p1))
493                 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
494                             "Successfully removed existing private key file `%s'\n",
495                             key_file_p1);
496               else
497                 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
498                             "Failed to remove private key file `%s'\n",
499                             key_file_p1);
500             }
501           if (GNUNET_CONFIGURATION_have_value (p->cfg,"transport-https", "CERT_FILE"))
502             GNUNET_CONFIGURATION_get_value_string (p->cfg, "transport-https", "CERT_FILE", &cert_file_p1);
503           if (cert_file_p1 == NULL)
504             GNUNET_asprintf(&cert_file_p1,"https_p1.cert");
505           if (0 == stat (cert_file_p1, &sbuf ))
506             {
507               if (0 == remove(cert_file_p1))
508                 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
509                             "Successfully removed existing certificate file `%s'\n",
510                             cert_file_p1);
511               else
512                 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
513                             "Failed to remove existing certificate file `%s'\n",
514                             cert_file_p1);
515             }
516         }
517       else if (p==&p2)
518         {
519           if (GNUNET_CONFIGURATION_have_value (p->cfg,
520                                                "transport-https", "KEY_FILE"))
521             GNUNET_CONFIGURATION_get_value_string (p->cfg, "transport-https", "KEY_FILE", &key_file_p2);
522           if (key_file_p2 == NULL)
523             GNUNET_asprintf(&key_file_p2,"https_p2.key");
524           if (0 == stat (key_file_p2, &sbuf ))
525             {
526               if (0 == remove(key_file_p2))
527                 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
528                             "Successfully removed existing private key file `%s'\n",
529                             key_file_p2);
530               else
531                 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
532                             "Failed to remove private key file `%s'\n",
533                             key_file_p2);
534             }
535           if (GNUNET_CONFIGURATION_have_value (p->cfg,"transport-https", "CERT_FILE"))
536             GNUNET_CONFIGURATION_get_value_string (p->cfg, "transport-https", "CERT_FILE", &cert_file_p2);
537           if (cert_file_p2 == NULL)
538             GNUNET_asprintf(&cert_file_p2,"https_p2.cert");
539           if (0 == stat (cert_file_p2, &sbuf ))
540             {
541               if (0 == remove(cert_file_p2))
542                 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
543                             "Successfully removed existing certificate file `%s'\n",
544                             cert_file_p2);
545               else
546                 GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
547                             "Failed to remove existing certificate file `%s'\n",
548                             cert_file_p2);
549             }
550         }
551     }
552   p->th = GNUNET_TRANSPORT_connect (p->cfg, NULL,
553                                     p,
554                                     &notify_receive,
555                                     &notify_connect,
556                                     &notify_disconnect);
557   GNUNET_assert (p->th != NULL);
558 }
559
560
561 /**
562  * Return the actual path to a file found in the current
563  * PATH environment variable.
564  *
565  * @param binary the name of the file to find
566  */
567 static char *
568 get_path_from_PATH (char *binary)
569 {
570   char *path;
571   char *pos;
572   char *end;
573   char *buf;
574   const char *p;
575
576   p = getenv ("PATH");
577   if (p == NULL)
578     {
579       GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
580                   _("PATH environment variable is unset.\n"));
581       return NULL;
582     }
583   path = GNUNET_strdup (p);     /* because we write on it */
584   buf = GNUNET_malloc (strlen (path) + 20);
585   pos = path;
586
587   while (NULL != (end = strchr (pos, PATH_SEPARATOR)))
588     {
589       *end = '\0';
590       sprintf (buf, "%s/%s", pos, binary);
591       if (GNUNET_DISK_file_test (buf) == GNUNET_YES)
592         {
593           GNUNET_free (path);
594           return buf;
595         }
596       pos = end + 1;
597     }
598   sprintf (buf, "%s/%s", pos, binary);
599   if (GNUNET_DISK_file_test (buf) == GNUNET_YES)
600     {
601       GNUNET_free (path);
602       return buf;
603     }
604   GNUNET_free (buf);
605   GNUNET_free (path);
606   return NULL;
607 }
608
609 /**
610  * Check whether the suid bit is set on a file.
611  * Attempts to find the file using the current
612  * PATH environment variable as a search path.
613  *
614  * @param binary the name of the file to check
615  *
616  * @return GNUNET_YES if the binary is found and
617  *         can be run properly, GNUNET_NO otherwise
618  */
619 static int
620 check_gnunet_nat_binary(char *binary)
621 {
622   struct stat statbuf;
623   char *p;
624 #ifdef MINGW
625   SOCKET rawsock;
626 #endif
627
628 #ifdef MINGW
629   char *binaryexe;
630   GNUNET_asprintf (&binaryexe, "%s.exe", binary);
631   p = get_path_from_PATH (binaryexe);
632   free (binaryexe);
633 #else
634   p = get_path_from_PATH (binary);
635 #endif
636   if (p == NULL)
637     {
638       GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
639                   _("Could not find binary `%s' in PATH!\n"),
640                   binary);
641       return GNUNET_NO;
642     }
643   if (0 != STAT (p, &statbuf))
644     {
645       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
646                   _("stat (%s) failed: %s\n"),
647                   p,
648                   STRERROR (errno));
649       GNUNET_free (p);
650       return GNUNET_SYSERR;
651     }
652   GNUNET_free (p);
653 #ifndef MINGW
654   if ( (0 != (statbuf.st_mode & S_ISUID)) &&
655        (statbuf.st_uid == 0) )
656     return GNUNET_YES;
657   return GNUNET_NO;
658 #else
659   rawsock = socket (AF_INET, SOCK_RAW, IPPROTO_ICMP);
660   if (INVALID_SOCKET == rawsock)
661     {
662       DWORD err = GetLastError ();
663       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
664                   "socket (AF_INET, SOCK_RAW, IPPROTO_ICMP) have failed! GLE = %d\n", err);
665       return GNUNET_NO; /* not running as administrator */
666     }
667   closesocket (rawsock);
668   return GNUNET_YES;
669 #endif
670 }
671
672
673 static void
674 try_connect (void *cls,
675              const struct GNUNET_SCHEDULER_TaskContext *tc)
676 {
677   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
678               "Asking peers to connect...\n");
679   GNUNET_TRANSPORT_try_connect (p2.th,
680                                 &p1.id);
681   GNUNET_TRANSPORT_try_connect (p1.th,
682                                 &p2.id);
683   tct = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS,
684                                       &try_connect,
685                                       NULL);
686 }
687
688
689 static void
690 run (void *cls,
691      char *const *args,
692      const char *cfgfile, const struct GNUNET_CONFIGURATION_Handle *cfg)
693 {
694   GNUNET_assert (ok == 1);
695   OKPP;
696   die_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT,
697                                            &end_badly,
698                                            NULL);
699   if (is_tcp)
700     {
701       setup_peer (&p1, "test_transport_api_tcp_peer1.conf");
702       setup_peer (&p2, "test_transport_api_tcp_peer2.conf");
703     }
704   else if (is_http)
705     {
706       setup_peer (&p1, "test_transport_api_rel_http_peer1.conf");
707       setup_peer (&p2, "test_transport_api_rel_http_peer2.conf");
708     }
709   else if (is_https)
710     {
711       setup_peer (&p1, "test_transport_api_rel_https_peer1.conf");
712       setup_peer (&p2, "test_transport_api_rel_https_peer2.conf");
713     }
714   else if (is_udp)
715     {
716       setup_peer (&p1, "test_transport_api_udp_peer1.conf");
717       setup_peer (&p2, "test_transport_api_udp_peer2.conf");
718     }
719   else if (is_unix)
720     {
721       setup_peer (&p1, "test_transport_api_unix_peer1.conf");
722       setup_peer (&p2, "test_transport_api_unix_peer2.conf");
723     }
724   else if (is_tcp_nat)
725     {
726       setup_peer (&p1, "test_transport_api_tcp_nat_peer1.conf");
727       setup_peer (&p2, "test_transport_api_tcp_nat_peer2.conf");
728     }
729   else if (is_wlan)
730     {
731       setup_peer (&p1, "test_transport_api_wlan_peer1.conf");
732       setup_peer (&p2, "test_transport_api_wlan_peer2.conf");
733     }
734   else
735     GNUNET_assert (0);
736   GNUNET_assert(p1.th != NULL);
737   GNUNET_assert(p2.th != NULL);
738   GNUNET_TRANSPORT_get_hello (p1.th, &exchange_hello, &p1);
739   p1_hello_canceled = GNUNET_NO;
740   GNUNET_TRANSPORT_get_hello (p2.th, &exchange_hello_last, &p2);
741   p2_hello_canceled = GNUNET_NO;
742   tct = GNUNET_SCHEDULER_add_now (&try_connect, NULL);
743 }
744
745
746 static int
747 check ()
748 {
749   char *const argv[] = { "test-transport-api-reliability",
750     "-c",
751     "test_transport_api_data.conf",
752 #if VERBOSE
753     "-L", "DEBUG",
754 #endif
755     NULL
756   };
757   struct GNUNET_GETOPT_CommandLineOption options[] = {
758     GNUNET_GETOPT_OPTION_END
759   };
760
761 #if WRITECONFIG
762   setTransportOptions("test_transport_api_data.conf");
763 #endif
764   ok = 1;
765
766   if ((GNUNET_YES == is_tcp_nat) && (check_gnunet_nat_binary("gnunet-nat-server") != GNUNET_YES))
767     {
768       GNUNET_log(GNUNET_ERROR_TYPE_WARNING,
769                  "Not running NAT test case, binaries not properly installed.\n");
770       return 0;
771     }
772
773   GNUNET_PROGRAM_run ((sizeof (argv) / sizeof (char *)) - 1,
774                       argv, "test-transport-api-reliability", "nohelp",
775                       options, &run, &ok);
776   stop_arm (&p1);
777   stop_arm (&p2);
778
779   if (is_https)
780   {
781     struct stat sbuf;
782     if (0 == stat (cert_file_p1, &sbuf ))
783     {
784       if (0 == remove(cert_file_p1))
785         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, 
786                     "Successfully removed existing certificate file `%s'\n",
787                     cert_file_p1);
788       else
789         GNUNET_log (GNUNET_ERROR_TYPE_ERROR, 
790                     "Failed to remove certfile `%s'\n",
791                     cert_file_p1);
792     }
793
794     if (0 == stat (key_file_p1, &sbuf ))
795     {
796       if (0 == remove(key_file_p1))
797         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, 
798                     "Successfully removed private key file `%s'\n",
799                     key_file_p1);
800       else
801         GNUNET_log (GNUNET_ERROR_TYPE_ERROR, 
802                     "Failed to private key file `%s'\n",key_file_p1);
803     }
804
805     if (0 == stat (cert_file_p2, &sbuf ))
806     {
807       if (0 == remove(cert_file_p2))
808         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, 
809                     "Successfully removed existing certificate file `%s'\n",
810                     cert_file_p2);
811       else
812         GNUNET_log (GNUNET_ERROR_TYPE_ERROR, 
813                     "Failed to remove certfile `%s'\n",cert_file_p2);
814     }
815
816     if (0 == stat (key_file_p2, &sbuf ))
817     {
818       if (0 == remove(key_file_p2))
819         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, 
820                     "Successfully removed private key file `%s'\n",
821                     key_file_p2);
822       else
823         GNUNET_log (GNUNET_ERROR_TYPE_ERROR, 
824                     "Failed to private key file `%s'\n",
825                     key_file_p2);
826     }
827     GNUNET_free(key_file_p1);
828     GNUNET_free(key_file_p2);
829     GNUNET_free(cert_file_p1);
830     GNUNET_free(cert_file_p2);
831     GNUNET_free(p1.servicehome);
832     GNUNET_free(p2.servicehome);
833   }
834
835   return ok;
836 }
837
838
839 int
840 main (int argc, char *argv[])
841 {
842   int ret;
843 #ifdef MINGW
844   return GNUNET_SYSERR;
845 #endif
846
847   if (strstr(argv[0], "tcp_nat") != NULL)
848     {
849       is_tcp_nat = GNUNET_YES;
850       GNUNET_asprintf(&test_name, "tcp_nat");
851     }
852   else if (strstr(argv[0], "tcp") != NULL)
853     {
854       is_tcp = GNUNET_YES;
855       GNUNET_asprintf(&test_name, "tcp");
856     }
857   else if (strstr(argv[0], "https") != NULL)
858     {
859       is_https = GNUNET_YES;
860       GNUNET_asprintf(&test_name, "https");
861     }
862   else if (strstr(argv[0], "http") != NULL)
863     {
864       is_http = GNUNET_YES;
865       GNUNET_asprintf(&test_name, "http");
866     }
867   else if (strstr(argv[0], "udp") != NULL)
868     {
869       is_udp = GNUNET_YES;
870       GNUNET_asprintf(&test_name, "udp");
871     }
872   else if (strstr(argv[0], "unix") != NULL)
873     {
874       is_unix = GNUNET_YES;
875       GNUNET_asprintf(&test_name, "unix");
876     }
877   else if (strstr(argv[0], "wlan") != NULL)
878     {
879        is_wlan = GNUNET_YES;
880     }
881   GNUNET_log_setup ("test-transport-api-reliability",
882 #if VERBOSE
883                     "DEBUG",
884 #else
885                     "WARNING",
886 #endif
887                     NULL);
888   ret = check ();
889   GNUNET_DISK_directory_remove (p1.servicehome);
890   GNUNET_DISK_directory_remove (p2.servicehome);
891   return ret;
892 }
893
894 /* end of test_transport_api_reliability.c */