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 static size_t
 537 send_dummy_data_task (void *cls, size_t size, void *buf)
 538 {
 539   int s = sizeof (struct TestMessage);
 540   struct TestMessage hdr;
 541
 542   hdr.header.size = htons (s);
 543   hdr.header.type = htons (GNUNET_MESSAGE_TYPE_TRANSPORT_ATS);
 544   if (force_rebuild)
 545     hdr.num = htonl (1);
 546   if (force_q_updates)
 547     hdr.num = htonl (2);
 548   memcpy (buf,&hdr, s);
 549   // GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Sent bytes: %i of %i\n", s, s);
 550   t  = NULL;
 551   return s;
 552 }
 553
 554
 555 static void
 556 send_task_f (void *cls,
 557              const struct GNUNET_SCHEDULER_TaskContext *tc)
 558 {
 559   send_task = GNUNET_SCHEDULER_NO_TASK;
 560   if ( (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) != 0)
 561     return;
 562
 563   if (t!=NULL)
 564     {
 565       GNUNET_TRANSPORT_notify_transmit_ready_cancel(t);
 566       t = NULL;
 567     }
 568
 569   if (send_msg == GNUNET_YES)
 570     t = GNUNET_TRANSPORT_notify_transmit_ready(th,
 571                                                &master_deamon->id,
 572                                                sizeof (struct TestMessage), 0,
 573                                                SEND_TIMEOUT,
 574                                                &send_dummy_data_task, NULL);
 575   send_task = GNUNET_SCHEDULER_add_delayed(
 576       GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_MILLISECONDS,1000),
 577       &send_task_f,
 578       NULL);
 579
 580 }
 581
 582 static void
 583 notify_connect (void *cls,
 584                 const struct GNUNET_PeerIdentity *peer,
 585                 const struct GNUNET_TRANSPORT_ATS_Information *ats,
 586                 uint32_t ats_count)
 587 {
 588   send_task = GNUNET_SCHEDULER_add_now(&send_task_f, NULL);
 589 }
 590
 591 static void
 592 notify_disconnect (void *cls,
 593                    const struct GNUNET_PeerIdentity *peer)
 594 {
 595   if (GNUNET_SCHEDULER_NO_TASK != send_task)
 596     {
 597       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 598                   "Disconnect event before transmission request could be scheduled!\n");
 599       GNUNET_SCHEDULER_cancel (send_task);
 600       send_task = GNUNET_SCHEDULER_NO_TASK;
 601     }
 602   if (NULL != t)
 603     {
 604       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 605                   "Disconnect event before transmission request could be completed!\n");
 606       GNUNET_TRANSPORT_notify_transmit_ready_cancel (t);
 607       t = NULL;
 608     }
 609 }
 610
 611 static void
 612 daemon_connect_cb(void *cls,
 613                   const struct GNUNET_PeerIdentity *first,
 614                   const struct GNUNET_PeerIdentity *second,
 615                   uint32_t distance,
 616                   const struct GNUNET_CONFIGURATION_Handle *first_cfg,
 617                   const struct GNUNET_CONFIGURATION_Handle *second_cfg,
 618                   struct GNUNET_TESTING_Daemon *first_daemon,
 619                   struct GNUNET_TESTING_Daemon *second_daemon,
 620                   const char *emsg)
 621 {
 622   char * firstc =  strdup(GNUNET_i2s(first));
 623   char * secondc =  strdup(GNUNET_i2s(second));
 624   connected++;
 625   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 626               "Connected peers `%s'<->`%s' (%i/%i)\n",
 627               firstc, secondc, connected, peers-1);
 628   GNUNET_free(firstc);
 629   GNUNET_free(secondc);
 630
 631   if ( ( (first_daemon == ping_deamon) ||
 632          (second_daemon == ping_deamon) ) &&
 633        (master_deamon != NULL) &&
 634        (ping_deamon != NULL) )
 635     {
 636       th = GNUNET_TRANSPORT_connect (ping_deamon->cfg,
 637                                      &ping_deamon->id,
 638                                      NULL, NULL,
 639                                      &notify_connect,
 640                                      &notify_disconnect);
 641       force_q_updates = GNUNET_YES;
 642       send_msg = GNUNET_YES;
 643     }
 644 }
 645
 646
 647
 648 static void
 649 daemon_start_cb (void *cls,
 650                  const struct GNUNET_PeerIdentity *id,
 651                  const struct GNUNET_CONFIGURATION_Handle *cfg,
 652                  struct GNUNET_TESTING_Daemon *d, const char *emsg)
 653 {
 654   if (id == NULL)
 655     {
 656       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 657                   "Start callback called with error (too long starting peers), aborting test!\n");
 658       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 659                   "Error from testing: `%s'\n");
 660       failed_peers++;
 661       if (failed_peers == peers_left)
 662         {
 663           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 664                       "Too many peers failed, ending test!\n");
 665           ok = 1;
 666         shutdown_peers ();
 667         }
 668       return;
 669     }
 670   peers_left--;
 671
 672   if (master_deamon == NULL)
 673     {
 674       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 675                   "Master peer `%s' '%s'\n",
 676                   GNUNET_i2s(id), d->cfgfile);
 677       master_deamon = d;
 678       stats = GNUNET_STATISTICS_create("transport", master_deamon->cfg);
 679       GNUNET_assert (stats != NULL);
 680       stats_task = GNUNET_SCHEDULER_add_now(&stats_get_task, NULL);
 681     }
 682   else
 683     {
 684       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 685                   "Connecting peer `%s'\n",
 686                   GNUNET_i2s(id), GNUNET_i2s(&master_deamon->id));
 687       GNUNET_TESTING_daemons_connect(d,
 688           master_deamon,
 689           TIMEOUT,
 690           0,
 691           GNUNET_YES,
 692           &daemon_connect_cb,
 693           NULL);
 694     }
 695
 696   if (peers_left == 0)
 697     {
 698       if (ping_deamon == NULL)
 699         {
 700           ping_deamon = d;
 701           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 702                       "Ping peer `%s' '%s'\n", GNUNET_i2s(id), d->cfgfile);
 703         }
 704
 705       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 706                   "All peers started successfully!\n");
 707       connect_peers();
 708       ok = 0;
 709     }
 710   else if (failed_peers == peers_left)
 711     {
 712       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 713                   "Too many peers failed, ending test!\n");
 714       shutdown_peers();
 715       ok = 1;
 716     }
 717 }
 718
 719
 720 static void
 721 run (void *cls,
 722      char *const *args,
 723      const char *cfgfile, const struct GNUNET_CONFIGURATION_Handle *cfg)
 724 {
 725   ok = 1;
 726   measurement_started = GNUNET_NO;
 727 #if VERBOSE
 728   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Starting %i peers.\n", peers);
 729 #endif
 730   peers_left = peers;
 731   pg = GNUNET_TESTING_daemons_start (cfg,
 732                                      peers_left, /* Total number of peers */
 733                                      peers_left, /* Number of outstanding connections */
 734                                      peers_left, /* Number of parallel ssh connections, or peers being started at once */
 735                                      TIMEOUT,
 736                                      NULL, NULL,
 737                                      &daemon_start_cb, NULL, NULL, NULL, NULL);
 738   GNUNET_assert (pg != NULL);
 739 }
 740
 741
 742 static int
 743 check ()
 744 {
 745   char *const argv[] = { "test-testing",
 746     "-c",
 747     config_file,
 748 #if VERBOSE
 749     "-L", "DEBUG",
 750 #endif
 751     NULL
 752   };
 753   struct GNUNET_GETOPT_CommandLineOption options[] = {
 754     GNUNET_GETOPT_OPTION_END
 755   };
 756   GNUNET_PROGRAM_run ((sizeof (argv) / sizeof (char *)) - 1,
 757                       argv, "test-transport-ats", "nohelp",
 758                       options, &run, &ok);
 759   return ok;
 760 }
 761
 762
 763 int
 764 main (int argc, char *argv[])
 765 {
 766   int ret;
 767
 768   GNUNET_log_setup ("test-transport-ats-multiple-peers",
 769 #if VERBOSE
 770                     "DEBUG",
 771 #else
 772                     "INFO",
 773 #endif
 774                     NULL);
 775
 776 #if !HAVE_LIBGLPK && !HAVE_GLPK
 777   GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 778               "GLPK not installed, exiting testcase\n");
 779   return 0;
 780 #endif
 781
 782   GNUNET_DISK_directory_remove ("/tmp/test-gnunet-testing");
 783   machine_parsable = GNUNET_NO;
 784   peers = NUM_PEERS;
 785   config_file = "test_transport_ats_4addr.conf";
 786
 787   int c = 0;
 788   if (argc >= 2)
 789     {
 790       for (c=0; c<argc; c++)
 791         {
 792           /* set peers */
 793           if ((strcmp(argv[c], "-p") == 0) && c < (argc-1))
 794             {
 795               peers = atoi(argv[c+1]);
 796               peers++;
 797             }
 798           /* set machine parsable */
 799           if (strcmp(argv[c], "-m") == 0)
 800             {
 801               machine_parsable = GNUNET_YES;
 802             }
 803           /* set config file */
 804           if ((strcmp(argv[c], "-c") == 0) && c < (argc-1))
 805             {
 806               config_file = argv[c+1];
 807             }
 808          }
 809     }
 810
 811
 812
 813   ret = check ();
 814   /**
 815    * Still need to remove the base testing directory here,
 816    * because group starts will create subdirectories under this
 817    * main dir.  However, we no longer need to sleep, as the
 818    * shutdown sequence won't return until everything is cleaned
 819    * up.
 820    */
 821   GNUNET_DISK_directory_remove ("/tmp/test-gnunet-testing");
 822   return ret;
 823 }
 824
 825 /* end of test_transport_ats_multiple_peers.c*/