053611f21c2519948d041d9022bd934efff8f924
[oweals/gnunet.git] / src / transport / test_transport_ats_multiple_peers.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 testing/test_transport_ats_multiple_peers.c
22  * @brief testcase for ats functionality by starting multiple peers
23  */
24 #include "platform.h"
25 #include "gnunet_testing_lib.h"
26 #include "gnunet_transport_service.h"
27 #include "gnunet_scheduler_lib.h"
28 #include "gauger.h"
29 #include "transport_ats.h"
30
31 #define VERBOSE GNUNET_NO
32
33 #define NUM_PEERS 11
34 #define MEASUREMENTS 5
35
36 #define DELAY GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 300)
37 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 300)
38 #define SEND_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 1)
39
40 static int ok;
41
42 static int peers_left;
43
44 static int failed_peers;
45
46 static int measurement_started = GNUNET_NO;
47 static char * config_file;
48
49 static struct GNUNET_TESTING_PeerGroup *pg;
50
51 static GNUNET_SCHEDULER_TaskIdentifier shutdown_task;
52 static GNUNET_SCHEDULER_TaskIdentifier stats_task;
53 static GNUNET_SCHEDULER_TaskIdentifier send_task;
54 struct GNUNET_TESTING_Daemon * master_deamon;
55 struct GNUNET_TESTING_Daemon * ping_deamon;
56
57 struct GNUNET_STATISTICS_Handle * stats;
58
59 struct TEST_result
60 {
61   uint64_t timestamp;
62   uint64_t duration;
63   uint64_t mechs;
64   uint64_t peers;
65   uint64_t solution;
66   uint64_t state;
67 };
68
69 struct TestMessage
70 {
71   struct GNUNET_MessageHeader header;
72   uint32_t num;
73 };
74
75
76 static int count;
77 static int c_new;
78 static int c_unmodified;
79 static int c_modified;
80 static int connected;
81 static int peers;
82
83 static int force_q_updates;
84 static int force_rebuild;
85 static int send_msg;
86 static int machine_parsable;
87
88 static struct TEST_result results_new       [MEASUREMENTS+1];
89 static struct TEST_result results_modified  [MEASUREMENTS+1];
90 static struct TEST_result results_unmodified[MEASUREMENTS+1];
91 static struct TEST_result current;
92
93 static struct GNUNET_STATISTICS_GetHandle * s_solution;
94 static struct GNUNET_STATISTICS_GetHandle * s_time;
95 static struct GNUNET_STATISTICS_GetHandle * s_peers;
96 static struct GNUNET_STATISTICS_GetHandle * s_mechs;
97 static struct GNUNET_STATISTICS_GetHandle * s_duration;
98 static struct GNUNET_STATISTICS_GetHandle * s_invalid;
99 static struct GNUNET_STATISTICS_GetHandle * s_state;
100
101 struct GNUNET_TRANSPORT_TransmitHandle * t;
102 struct GNUNET_TRANSPORT_Handle * th;
103
104 /**
105  * Check whether peers successfully shut down.
106  */
107 static void
108 shutdown_callback (void *cls, const char *emsg)
109 {
110   if (emsg != NULL)
111     {
112 #if VERBOSE
113       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, 
114                   "Shutdown of peers failed!\n");
115 #endif
116       if (ok == 0)
117         ok = 666;
118     }
119   else
120     {
121 #if VERBOSE
122       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
123                   "All peers successfully shut down!\n");
124         if (stats != NULL)
125                 GNUNET_STATISTICS_destroy(stats, GNUNET_NO);
126         stats = NULL;
127 #endif
128     }
129 }
130
131 static void
132 shutdown_peers()
133 {
134   if (shutdown_task != GNUNET_SCHEDULER_NO_TASK)
135     {
136       GNUNET_SCHEDULER_cancel(shutdown_task);
137       shutdown_task = GNUNET_SCHEDULER_NO_TASK;
138     }
139   if (stats_task != GNUNET_SCHEDULER_NO_TASK)
140     {
141       GNUNET_SCHEDULER_cancel(stats_task);
142       stats_task = GNUNET_SCHEDULER_NO_TASK;
143     }
144   if (send_task != GNUNET_SCHEDULER_NO_TASK)
145     {
146       GNUNET_SCHEDULER_cancel(send_task);
147       send_task = GNUNET_SCHEDULER_NO_TASK;
148     }
149   
150   if (t != NULL)
151     {
152       GNUNET_TRANSPORT_notify_transmit_ready_cancel(t);
153       t = NULL;
154     }
155   GNUNET_TRANSPORT_disconnect(th);  
156   if (s_time != NULL)
157     {
158       GNUNET_STATISTICS_get_cancel(s_time);
159       s_time = NULL;
160     }
161   if (s_peers != NULL)
162     {
163       GNUNET_STATISTICS_get_cancel(s_peers);
164       s_peers = NULL;
165     }
166   if (s_mechs != NULL)
167     {
168       GNUNET_STATISTICS_get_cancel(s_mechs);
169       s_mechs = NULL;
170     }
171   if (s_solution != NULL)
172     {
173       GNUNET_STATISTICS_get_cancel(s_solution);
174       s_solution = NULL;
175     }
176   if (s_duration != NULL)
177     {
178       GNUNET_STATISTICS_get_cancel(s_duration);
179       s_duration = NULL;
180     }
181   if (s_invalid != NULL)
182     {
183       GNUNET_STATISTICS_get_cancel(s_invalid);
184       s_invalid = NULL;
185     }
186   if (s_state != NULL)
187     {
188       GNUNET_STATISTICS_get_cancel(s_state);
189       s_state = NULL;
190     }
191   GNUNET_TESTING_daemons_stop (pg, TIMEOUT, &shutdown_callback, NULL);
192 }
193
194 static void 
195 evaluate_measurements()
196 {
197   int c;
198   //int mechs = 0;
199   double average[3];
200   double stddev[3];
201   //char * output;
202   c = 1;
203   
204   //GNUNET_asprintf(&output, "p,%i,m,%i,",peers, MEASUREMENTS, results_modified[0].mechs,
205   
206   average[0] = 0.0;
207   for (c=0; c<c_new;c++)
208     {
209       average[0] += (double) results_new[c].duration;
210     }
211   average[0] /= c_new;
212   
213   stddev[0] = 0.0;
214   for (c=0; c<c_new;c++)
215     {
216       stddev[0] += (results_new[c].duration - average[0]) *
217           (results_new[c].duration - average[0]);
218     }
219   stddev[0] /= c_new;
220   stddev[0] = sqrt (stddev[0]);
221   if (!machine_parsable)
222     fprintf (stderr,
223              "new, %i measurements, average: %f stddev: %f\n",
224              c_new, average[0], stddev[0]);
225   
226   average[1] = 0.0;
227   for (c=0; c<c_modified;c++)
228     {
229       average[1] += (double) results_modified[c].duration;
230     }
231   average[1] /= c_modified;
232   
233   stddev[1] = 0.0;
234   for (c=0; c<c_modified;c++)
235     {
236       stddev[1] += (results_modified[c].duration - average[1]) *
237           (results_modified[c].duration - average[1]);
238     }
239   stddev[1] /= c_modified;
240   stddev[1] = sqrt (stddev[1]);
241   if (!machine_parsable) 
242     fprintf (stderr,
243              "modified, %i measurements, average: %f stddev: %f\n",
244              c_modified, average[1], stddev[1]);
245   
246   average[2] = 0.0;
247   for (c=0; c<c_unmodified;c++)
248     {
249       average[2] += (double) results_unmodified[c].duration;
250     }
251   average[2] /= c_unmodified;
252   stddev[2] = 0.0;
253   for (c=0; c<c_unmodified;c++)
254     {
255       stddev[2] += (results_unmodified[c].duration - average[2]) *
256           (results_unmodified[c].duration - average[2]);
257     }
258   stddev[2] /= c_unmodified;
259   stddev[2] = sqrt (stddev[2]);
260   
261   if (!machine_parsable) 
262     fprintf (stderr,
263              "unmodified, %i measurements, average: %f stddev: %f\n",
264              c_unmodified, average[2], stddev[2]);
265   
266   if (machine_parsable)
267     fprintf (stderr,
268              "peers,%i,mechs,%llu,"
269              "new,%i,%f,%f,"
270              "mod,%i,%f,%f,"
271              "unmod,%i,%f,%f\n",
272              peers-1, (unsigned long long) results_unmodified[0].mechs,
273              c_new, average[0], stddev[0],
274              c_modified, average[1], stddev[1],
275              c_unmodified, average[2], stddev[2]);
276   shutdown_peers();
277 }
278
279
280 static int 
281 stats_cb (void *cls,
282           const char *subsystem,
283           const char *name,
284           uint64_t value,
285           int is_persistent)
286 {
287   static int printed = GNUNET_NO;
288 #if VERBOSE_ATS
289   GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "%s = %llu\n", name ,value);
290 #endif
291   if (0 == strcmp (name,"ATS invalid solutions"))
292     {
293       if (stats_task != GNUNET_SCHEDULER_NO_TASK)
294         {
295           GNUNET_SCHEDULER_cancel(stats_task);
296           stats_task = GNUNET_SCHEDULER_NO_TASK;
297         }
298       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,"MLP produced invalid %llu result(s)!\n",
299           value);
300       shutdown_peers();
301       return GNUNET_SYSERR;
302     }
303   
304   if (0 == strcmp (name,"ATS solution"))
305     {
306       s_solution = NULL;
307     }
308   
309   if (0 == strcmp (name,"ATS peers"))
310     {
311       s_peers = NULL;
312     }
313   
314   if (0 == strcmp (name,"ATS mechanisms"))
315     {
316       s_mechs = NULL;
317     }
318   
319   if (0 == strcmp (name,"ATS duration"))
320     {
321       s_duration = NULL;
322     }
323   if (0 == strcmp (name,"ATS timestamp"))
324     {
325       s_time = NULL;
326     }
327   if (0 == strcmp (name,"ATS state"))
328     {
329       s_state = NULL;
330     }
331   
332   if ( (measurement_started == GNUNET_NO) && 
333        (0 == strcmp (name, "ATS peers")) && 
334        (value == peers-1) )
335     {
336       measurement_started = GNUNET_YES;
337       count = 1;
338       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, 
339                   "All %llu peers connected\n", 
340                   value);
341 #if !VERBOSE
342       if (! machine_parsable)   
343         fprintf(stderr, "%i", count);
344 #endif
345     }
346   
347   if (measurement_started == GNUNET_YES)
348     {
349       // GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "%s == %llu\n", name ,value);
350       if (0 == strcmp (name,"ATS timestamp"))
351         {
352           if (current.timestamp == 0)
353             {
354               printed = GNUNET_NO;
355               current.timestamp = value;
356             }
357           if (current.timestamp == value)
358             {
359               printed = GNUNET_YES;
360             }
361           if (current.timestamp != value)
362             {
363               if (current.state == ATS_NEW)
364                 {
365                   if (c_new < MEASUREMENTS)
366                     {
367                       results_new[c_new] = current;
368                       c_new++;
369                     }
370                   else
371                     {
372                       force_rebuild = GNUNET_NO;
373                       force_q_updates = GNUNET_NO;
374                       send_msg = GNUNET_NO;
375                     }
376                 }
377               if (current.state == ATS_UNMODIFIED)
378                 {
379                   if (c_unmodified < MEASUREMENTS)
380                     {
381                       results_unmodified[c_unmodified] = current;
382                       c_unmodified++;
383                     }
384                   
385                 }
386               if (current.state == ATS_QUALITY_UPDATED)
387                 {
388                   if (c_modified < MEASUREMENTS)
389                     {
390                       results_modified[c_modified] = current;
391                       c_modified++;
392                     }
393                   else
394                     {
395                       force_q_updates = GNUNET_NO;
396                       force_rebuild = GNUNET_YES;
397                     }
398                 }
399               count ++;
400 #if VERBOSE
401               fprintf (stderr,
402                        "(new: %i / modified: %i / unmodified: %i) of %i \n", 
403                        c_new, c_modified, c_unmodified , MEASUREMENTS);
404 #endif
405               if ((c_modified >= MEASUREMENTS) &&
406                   (c_new >= MEASUREMENTS) &&
407                   (c_unmodified >= MEASUREMENTS))
408                 {
409 #if !VERBOSE
410                   if (!machine_parsable)
411                     fprintf(stdout, "\n");
412 #endif
413                   if (stats_task != GNUNET_SCHEDULER_NO_TASK)
414                     {
415                       GNUNET_SCHEDULER_cancel(stats_task);
416                       stats_task = GNUNET_SCHEDULER_NO_TASK;
417                     }
418                   evaluate_measurements();
419                   return GNUNET_SYSERR;
420                 }
421               
422               printed = GNUNET_NO;
423               current.timestamp = value;
424 #if !VERBOSE
425               if (! machine_parsable)
426                 fprintf(stderr, "..%i", count);
427 #endif
428               return GNUNET_OK;
429             }
430         }
431       
432       if (0 == strcmp (name,"ATS solution"))
433         {
434           current.solution = value;
435           if (printed == GNUNET_NO)
436             GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "[%i] %s: %llu \n",
437                 count, name, value);
438         }
439       
440       if (0 == strcmp (name,"ATS peers"))
441         {
442           current.peers = value;
443           if (printed == GNUNET_NO)
444             GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "[%i] %s: %llu \n",
445                 count, name, value);
446         }
447       
448       if (0 == strcmp (name,"ATS mechanisms"))
449         {
450           current.mechs = value;
451           if (printed == GNUNET_NO) 
452             GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "[%i] %s: %llu \n",
453                 count, name, value);
454         }
455       
456       if (0 == strcmp (name,"ATS duration"))
457         {
458           current.duration = value;
459           if (printed == GNUNET_NO) 
460             GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "[%i] %s: %llu \n",
461                 count, name, value);
462         }
463       if (0 == strcmp (name,"ATS state"))
464         {
465           current.state = value;
466           char * cont;
467           if (value == ATS_NEW)
468             cont = "NEW";
469           if (value == ATS_COST_UPDATED)
470             cont = "C_UPDATED";
471           if (value == ATS_QUALITY_UPDATED)
472             cont = "Q_UPDATED";
473           if (value == ATS_QUALITY_COST_UPDATED)
474             cont = "QC_UPDATED";
475           if (value == ATS_UNMODIFIED)
476             cont = "UNMODIFIED";
477           if (printed == GNUNET_NO) 
478             GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, 
479                         "[%i] ATS state: %s\n", count, cont);
480         }
481     }
482   return GNUNET_OK;
483 }
484
485
486 static void
487 stats_get_task (void *cls,
488                 const struct GNUNET_SCHEDULER_TaskContext *tc)
489 {
490   stats_task = GNUNET_SCHEDULER_NO_TASK;
491   if ( (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) != 0)
492     return;
493   
494   s_time = GNUNET_STATISTICS_get (stats, "transport", "ATS timestamp",
495       TIMEOUT, NULL, &stats_cb, NULL);
496   s_solution = GNUNET_STATISTICS_get (stats, "transport", "ATS solution",
497       TIMEOUT, NULL, &stats_cb, NULL);
498   s_duration = GNUNET_STATISTICS_get (stats, "transport","ATS duration",
499       TIMEOUT, NULL, &stats_cb, NULL);
500   s_peers = GNUNET_STATISTICS_get (stats, "transport", "ATS peers",
501       TIMEOUT, NULL, &stats_cb, NULL);
502   s_mechs = GNUNET_STATISTICS_get (stats, "transport", "ATS mechanisms",
503       TIMEOUT, NULL, &stats_cb, NULL);
504   s_invalid = GNUNET_STATISTICS_get (stats, "transport", "ATS invalid solutions",
505       TIMEOUT, NULL, &stats_cb, NULL);
506   s_state = GNUNET_STATISTICS_get (stats, "transport", "ATS state",
507       TIMEOUT, NULL, &stats_cb, NULL);
508   
509   stats_task = GNUNET_SCHEDULER_add_delayed(
510       GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MILLISECONDS, 100),
511       &stats_get_task,
512       NULL);
513 }
514
515
516 static void
517 delay (void *cls,
518        const struct GNUNET_SCHEDULER_TaskContext *tc)
519 {
520   shutdown_task = GNUNET_SCHEDULER_NO_TASK;
521   if ( (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) != 0)
522     return;
523 #if VERBOSE
524   GNUNET_log (GNUNET_ERROR_TYPE_ERROR, 
525               "Delay over\n");
526 #endif
527   shutdown_peers ();
528 }
529
530 static void
531 connect_peers()
532 {
533   shutdown_task = GNUNET_SCHEDULER_add_delayed(DELAY, &delay, NULL);
534 }
535
536
537 /* To make compiler happy */
538 void dummy(void)
539 {
540   struct ATS_quality_metric * q = qm;
541   q = NULL;
542   struct ATS_ressource * r = ressources;
543   r = NULL;
544 }
545
546 static size_t 
547 send_dummy_data_task (void *cls, size_t size, void *buf)
548 {
549   int s = sizeof (struct TestMessage);
550   struct TestMessage hdr;
551   
552   hdr.header.size = htons (s);
553   hdr.header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_ATS);
554   if (force_rebuild)
555     hdr.num = htonl (1);
556   if (force_q_updates)
557     hdr.num = htonl (2);
558   memcpy (buf,&hdr, s);
559   // GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Sent bytes: %i of %i\n", s, s);
560   t  = NULL;
561   return s;
562 }
563
564
565 static void 
566 send_task_f (void *cls,
567              const struct GNUNET_SCHEDULER_TaskContext *tc)
568 {
569   send_task = GNUNET_SCHEDULER_NO_TASK;
570   if ( (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) != 0)
571     return;
572   
573   if (t!=NULL)
574     {
575       GNUNET_TRANSPORT_notify_transmit_ready_cancel(t);
576       t = NULL;
577     }
578
579   if (send_msg == GNUNET_YES)
580     t = GNUNET_TRANSPORT_notify_transmit_ready(th, 
581                                                &master_deamon->id, 
582                                                sizeof (struct TestMessage), 0, 
583                                                SEND_TIMEOUT, 
584                                                &send_dummy_data_task, NULL);
585   send_task = GNUNET_SCHEDULER_add_delayed(
586       GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_MILLISECONDS,1000),
587       &send_task_f,
588       NULL);
589
590 }
591
592 static void
593 notify_connect (void *cls,
594                 const struct GNUNET_PeerIdentity *peer,
595                 const struct GNUNET_TRANSPORT_ATS_Information *ats,
596                 uint32_t ats_count)
597 {
598   send_task = GNUNET_SCHEDULER_add_now(&send_task_f, NULL);
599 }
600
601 static void
602 notify_disconnect (void *cls,
603                    const struct GNUNET_PeerIdentity *peer)
604 {
605   if (GNUNET_SCHEDULER_NO_TASK != send_task)
606     {
607       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
608                   "Disconnect event before transmission request could be scheduled!\n");
609       GNUNET_SCHEDULER_cancel (send_task);
610       send_task = GNUNET_SCHEDULER_NO_TASK;
611     }
612   if (NULL != t)
613     {
614       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
615                   "Disconnect event before transmission request could be completed!\n");
616       GNUNET_TRANSPORT_notify_transmit_ready_cancel (t);
617       t = NULL;
618     }
619 }
620
621 static void 
622 daemon_connect_cb(void *cls,
623                   const struct GNUNET_PeerIdentity *first,
624                   const struct GNUNET_PeerIdentity *second,
625                   uint32_t distance,
626                   const struct GNUNET_CONFIGURATION_Handle *first_cfg,
627                   const struct GNUNET_CONFIGURATION_Handle *second_cfg,
628                   struct GNUNET_TESTING_Daemon *first_daemon,
629                   struct GNUNET_TESTING_Daemon *second_daemon,
630                   const char *emsg)
631 {
632   char * firstc =  strdup(GNUNET_i2s(first));
633   char * secondc =  strdup(GNUNET_i2s(second));
634   connected++;
635   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
636               "Connected peers `%s'<->`%s' (%i/%i)\n", 
637               firstc, secondc, connected, peers-1);
638   GNUNET_free(firstc);
639   GNUNET_free(secondc);
640   
641   if ( ( (first_daemon == ping_deamon) || 
642          (second_daemon == ping_deamon) ) && 
643        (master_deamon != NULL) && 
644        (ping_deamon != NULL) )
645     {
646       th = GNUNET_TRANSPORT_connect (ping_deamon->cfg,
647                                      &ping_deamon->id, 
648                                      NULL, NULL,
649                                      &notify_connect, 
650                                      &notify_disconnect);
651       force_q_updates = GNUNET_YES;
652       send_msg = GNUNET_YES;
653     }
654 }
655
656
657
658 static void
659 daemon_start_cb (void *cls,
660                  const struct GNUNET_PeerIdentity *id,
661                  const struct GNUNET_CONFIGURATION_Handle *cfg,
662                  struct GNUNET_TESTING_Daemon *d, const char *emsg)
663 {
664   if (id == NULL)
665     {
666       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
667                   "Start callback called with error (too long starting peers), aborting test!\n");
668       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
669                   "Error from testing: `%s'\n");
670       failed_peers++;
671       if (failed_peers == peers_left)
672         {
673           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
674                       "Too many peers failed, ending test!\n");
675           ok = 1;
676         shutdown_peers ();
677         }
678       return;
679     }
680   peers_left--;
681
682   if (master_deamon == NULL)
683     {
684       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, 
685                   "Master peer `%s' '%s'\n",
686                   GNUNET_i2s(id), d->cfgfile);      
687       master_deamon = d;
688       stats = GNUNET_STATISTICS_create("transport", master_deamon->cfg);
689       GNUNET_assert (stats != NULL);
690       stats_task = GNUNET_SCHEDULER_add_now(&stats_get_task, NULL);
691     }
692   else
693     {
694       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, 
695                   "Connecting peer `%s'\n", 
696                   GNUNET_i2s(id), GNUNET_i2s(&master_deamon->id));
697       GNUNET_TESTING_daemons_connect(d,
698           master_deamon,
699           TIMEOUT,
700           0,
701           GNUNET_YES,
702           &daemon_connect_cb,
703           NULL);
704     }
705   
706   if (peers_left == 0)
707     {
708       if (ping_deamon == NULL)
709         {
710           ping_deamon = d;
711           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, 
712                       "Ping peer `%s' '%s'\n", GNUNET_i2s(id), d->cfgfile);
713         }
714       
715       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
716                   "All peers started successfully!\n");
717       connect_peers();
718       ok = 0;
719     }
720   else if (failed_peers == peers_left)
721     {
722       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
723                   "Too many peers failed, ending test!\n");
724       shutdown_peers();
725       ok = 1;
726     }
727 }
728
729
730 static void
731 run (void *cls,
732      char *const *args,
733      const char *cfgfile, const struct GNUNET_CONFIGURATION_Handle *cfg)
734 {
735   ok = 1;
736   measurement_started = GNUNET_NO;
737 #if VERBOSE
738   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Starting %i peers.\n", peers);
739 #endif
740   peers_left = peers;
741   pg = GNUNET_TESTING_daemons_start (cfg,
742                                      peers_left, /* Total number of peers */
743                                      peers_left, /* Number of outstanding connections */
744                                      peers_left, /* Number of parallel ssh connections, or peers being started at once */
745                                      TIMEOUT,
746                                      NULL, NULL,
747                                      &daemon_start_cb, NULL, NULL, NULL, NULL);
748   GNUNET_assert (pg != NULL);
749 }
750
751
752 static int
753 check ()
754 {
755   char *const argv[] = { "test-testing",
756     "-c",
757     config_file,
758 #if VERBOSE
759     "-L", "DEBUG",
760 #endif
761     NULL
762   };
763   struct GNUNET_GETOPT_CommandLineOption options[] = {
764     GNUNET_GETOPT_OPTION_END
765   };
766   GNUNET_PROGRAM_run ((sizeof (argv) / sizeof (char *)) - 1,
767                       argv, "test-transport-ats", "nohelp",
768                       options, &run, &ok);
769   return ok;
770 }
771
772
773 int
774 main (int argc, char *argv[])
775 {
776   int ret;
777
778   GNUNET_log_setup ("test-transport-ats-multiple-peers",
779 #if VERBOSE
780                     "DEBUG",
781 #else
782                     "INFO",
783 #endif
784                     NULL);
785
786 #if !HAVE_LIBGLPK && !HAVE_GLPK
787   GNUNET_log (GNUNET_ERROR_TYPE_ERROR, 
788               "GLPK not installed, exiting testcase\n");
789   return 0;
790 #endif
791
792   GNUNET_DISK_directory_remove ("/tmp/test-gnunet-testing");
793   machine_parsable = GNUNET_NO;
794   peers = NUM_PEERS;
795   config_file = "test_transport_ats_4addr.conf";
796
797   int c = 0;
798   if (argc >= 2)
799     {
800       for (c=0; c<argc; c++)
801         {
802           /* set peers */
803           if ((strcmp(argv[c], "-p") == 0) && c < (argc-1))
804             {
805               peers = atoi(argv[c+1]);
806               peers++;
807             }
808           /* set machine parsable */
809           if (strcmp(argv[c], "-m") == 0)
810             {
811               machine_parsable = GNUNET_YES;
812             }
813           /* set config file */
814           if ((strcmp(argv[c], "-c") == 0) && c < (argc-1))
815             {
816               config_file = argv[c+1];
817             }
818          }
819     }
820
821
822
823   ret = check ();
824   /**
825    * Still need to remove the base testing directory here,
826    * because group starts will create subdirectories under this
827    * main dir.  However, we no longer need to sleep, as the
828    * shutdown sequence won't return until everything is cleaned
829    * up.
830    */
831   GNUNET_DISK_directory_remove ("/tmp/test-gnunet-testing");
832   return ret;
833 }
834
835 /* end of test_transport_ats_multiple_peers.c*/