src/mesh/test_mesh_small.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2011 Christian Grothoff (and other contributing authors)
   4
   5      GNUnet is free software; you can redistribute it and/or modify
   6      it under the terms of the GNU General Public License as published
   7      by the Free Software Foundation; either version 3, or (at your
   8      option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      General Public License for more details.
  14
  15      You should have received a copy of the GNU General Public License
  16      along with GNUnet; see the file COPYING.  If not, write to the
  17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  18      Boston, MA 02111-1307, USA.
  19 */
  20 /**
  21  * @file mesh/test_mesh_small.c
  22  *
  23  * @brief Test for the mesh service: retransmission of traffic.
  24  */
  25 #include <stdio.h>
  26 #include "platform.h"
  27 #include "mesh_test_lib.h"
  28 #include "gnunet_mesh_service.h"
  29 #include "gnunet_statistics_service.h"
  30 #include <gauger.h>
  31
  32
  33 /**
  34  * How namy messages to send
  35  */
  36 #define TOTAL_PACKETS 1000
  37
  38 /**
  39  * How long until we give up on connecting the peers?
  40  */
  41 #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 120)
  42
  43 /**
  44  * Time to wait for stuff that should be rather fast
  45  */
  46 #define SHORT_TIME GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 60)
  47
  48 /**
  49  * DIFFERENT TESTS TO RUN
  50  */
  51 #define SETUP 0
  52 #define FORWARD 1
  53 #define KEEPALIVE 2
  54 #define SPEED 3
  55 #define SPEED_ACK 4
  56 #define SPEED_REL 8
  57 #define P2P_SIGNAL 10
  58
  59 /**
  60  * Which test are we running?
  61  */
  62 static int test;
  63
  64 /**
  65  * String with test name
  66  */
  67 char *test_name;
  68
  69 /**
  70  * Flag to send traffic leaf->root in speed tests to test BCK_ACK logic.
  71  */
  72 static int test_backwards = GNUNET_NO;
  73
  74 /**
  75  * How many events have happened
  76  */
  77 static int ok;
  78
  79 /**
  80  * Number of events expected to conclude the test successfully.
  81  */
  82 int ok_goal;
  83
  84 /**
  85  * Size of each test packet
  86  */
  87 size_t size_payload = sizeof (struct GNUNET_MessageHeader) + sizeof (uint32_t);
  88
  89 /**
  90  * Operation to get peer ids.
  91  */
  92 struct GNUNET_TESTBED_Operation *t_op[2];
  93
  94 /**
  95  * Peer ids.
  96  */
  97 struct GNUNET_PeerIdentity *p_id[2];
  98
  99 /**
 100  * Peer ids counter.
 101  */
 102 unsigned int p_ids;
 103
 104 /**
 105  * Is the setup initialized?
 106  */
 107 static int initialized;
 108
 109 /**
 110  * Number of payload packes sent
 111  */
 112 static int data_sent;
 113
 114 /**
 115  * Number of payload packets received
 116  */
 117 static int data_received;
 118
 119 /**
 120  * Number of payload packed explicitly (app level) acknowledged
 121  */
 122 static int data_ack;
 123
 124 /**
 125  * Total number of currently running peers.
 126  */
 127 static unsigned long long peers_running;
 128
 129 /**
 130  * Test context (to shut down).
 131  */
 132 struct GNUNET_MESH_TEST_Context *test_ctx;
 133
 134 /**
 135  * Task called to disconnect peers.
 136  */
 137 static GNUNET_SCHEDULER_TaskIdentifier disconnect_task;
 138
 139 /**
 140  * Task To perform tests
 141  */
 142 static GNUNET_SCHEDULER_TaskIdentifier test_task;
 143
 144 /**
 145  * Task called to shutdown test.
 146  */
 147 static GNUNET_SCHEDULER_TaskIdentifier shutdown_handle;
 148
 149 /**
 150  * Mesh handle for the root peer
 151  */
 152 static struct GNUNET_MESH_Handle *h1;
 153
 154 /**
 155  * Mesh handle for the first leaf peer
 156  */
 157 static struct GNUNET_MESH_Handle *h2;
 158
 159 /**
 160  * Channel handle for the root peer
 161  */
 162 static struct GNUNET_MESH_Channel *ch;
 163
 164 /**
 165  * Channel handle for the dest peer
 166  */
 167 static struct GNUNET_MESH_Channel *incoming_ch;
 168
 169 /**
 170  * Time we started the data transmission (after channel has been established
 171  * and initilized).
 172  */
 173 static struct GNUNET_TIME_Absolute start_time;
 174
 175 static struct GNUNET_TESTBED_Peer **testbed_peers;
 176 static struct GNUNET_TESTBED_Operation *stats_op;
 177 static unsigned int ka_sent;
 178 static unsigned int ka_received;
 179
 180
 181 /**
 182  * Show the results of the test (banwidth acheived) and log them to GAUGER
 183  */
 184 static void
 185 show_end_data (void)
 186 {
 187   static struct GNUNET_TIME_Absolute end_time;
 188   static struct GNUNET_TIME_Relative total_time;
 189
 190   end_time = GNUNET_TIME_absolute_get();
 191   total_time = GNUNET_TIME_absolute_get_difference(start_time, end_time);
 192   FPRINTF (stderr, "\nResults of test \"%s\"\n", test_name);
 193   FPRINTF (stderr, "Test time %s\n",
 194            GNUNET_STRINGS_relative_time_to_string (total_time,
 195                                                    GNUNET_YES));
 196   FPRINTF (stderr, "Test bandwidth: %f kb/s\n",
 197            4 * TOTAL_PACKETS * 1.0 / (total_time.rel_value_us / 1000)); // 4bytes * ms
 198   FPRINTF (stderr, "Test throughput: %f packets/s\n\n",
 199            TOTAL_PACKETS * 1000.0 / (total_time.rel_value_us / 1000)); // packets * ms
 200   GAUGER ("MESH", test_name,
 201           TOTAL_PACKETS * 1000.0 / (total_time.rel_value_us / 1000),
 202           "packets/s");
 203 }
 204
 205
 206 /**
 207  * Shut down peergroup, clean up.
 208  *
 209  * @param cls Closure (unused).
 210  * @param tc Task Context.
 211  */
 212 static void
 213 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 214 {
 215   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Ending test.\n");
 216   shutdown_handle = GNUNET_SCHEDULER_NO_TASK;
 217 }
 218
 219
 220 /**
 221  * Disconnect from mesh services af all peers, call shutdown.
 222  *
 223  * @param cls Closure (unused).
 224  * @param tc Task Context.
 225  */
 226 static void
 227 disconnect_mesh_peers (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 228 {
 229   long line = (long) cls;
 230   unsigned int i;
 231
 232   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 233               "disconnecting mesh service of peers, called from line %ld\n",
 234               line);
 235   disconnect_task = GNUNET_SCHEDULER_NO_TASK;
 236   for (i = 0; i < 2; i++)
 237   {
 238     GNUNET_TESTBED_operation_done (t_op[i]);
 239   }
 240   if (NULL != ch)
 241   {
 242     GNUNET_MESH_channel_destroy (ch);
 243     ch = NULL;
 244   }
 245   if (NULL != incoming_ch)
 246   {
 247     GNUNET_MESH_channel_destroy (incoming_ch);
 248     incoming_ch = NULL;
 249   }
 250   GNUNET_MESH_TEST_cleanup (test_ctx);
 251   if (GNUNET_SCHEDULER_NO_TASK != shutdown_handle)
 252   {
 253     GNUNET_SCHEDULER_cancel (shutdown_handle);
 254   }
 255   shutdown_handle = GNUNET_SCHEDULER_add_now (&shutdown_task, NULL);
 256 }
 257
 258
 259 /**
 260  * Abort test: schedule disconnect and shutdown immediately
 261  *
 262  * @param line Line in the code the abort is requested from (__LINE__).
 263  */
 264 static void
 265 abort_test (long line)
 266 {
 267   if (disconnect_task != GNUNET_SCHEDULER_NO_TASK)
 268   {
 269     GNUNET_SCHEDULER_cancel (disconnect_task);
 270     disconnect_task = GNUNET_SCHEDULER_add_now (&disconnect_mesh_peers,
 271                                                 (void *) line);
 272   }
 273 }
 274
 275 /**
 276  * Transmit ready callback.
 277  *
 278  * @param cls Closure (message type).
 279  * @param size Size of the tranmist buffer.
 280  * @param buf Pointer to the beginning of the buffer.
 281  *
 282  * @return Number of bytes written to buf.
 283  */
 284 static size_t
 285 tmt_rdy (void *cls, size_t size, void *buf);
 286
 287
 288 /**
 289  * Task to schedule a new data transmission.
 290  *
 291  * @param cls Closure (peer #).
 292  * @param tc Task Context.
 293  */
 294 static void
 295 data_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 296 {
 297   struct GNUNET_MESH_TransmitHandle *th;
 298   struct GNUNET_MESH_Channel *channel;
 299
 300   if ((GNUNET_SCHEDULER_REASON_SHUTDOWN & tc->reason) != 0)
 301     return;
 302
 303   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Data task\n");
 304   if (GNUNET_YES == test_backwards)
 305   {
 306     channel = incoming_ch;
 307   }
 308   else
 309   {
 310     channel = ch;
 311   }
 312   th = GNUNET_MESH_notify_transmit_ready (channel, GNUNET_NO,
 313                                           GNUNET_TIME_UNIT_FOREVER_REL,
 314                                           size_payload, &tmt_rdy, (void *) 1L);
 315   if (NULL == th)
 316   {
 317     unsigned long i = (unsigned long) cls;
 318
 319     GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Retransmission\n");
 320     if (0 == i)
 321     {
 322       GNUNET_log (GNUNET_ERROR_TYPE_INFO, "  in 1 ms\n");
 323       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_MILLISECONDS,
 324                                     &data_task, (void *)1UL);
 325     }
 326     else
 327     {
 328       i++;
 329       GNUNET_log (GNUNET_ERROR_TYPE_INFO, "in %u ms\n", i);
 330       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply(
 331                                       GNUNET_TIME_UNIT_MILLISECONDS,
 332                                       i),
 333                                     &data_task, (void *)i);
 334     }
 335   }
 336 }
 337
 338
 339 /**
 340  * Transmit ready callback
 341  *
 342  * @param cls Closure (message type).
 343  * @param size Size of the buffer we have.
 344  * @param buf Buffer to copy data to.
 345  */
 346 size_t
 347 tmt_rdy (void *cls, size_t size, void *buf)
 348 {
 349   struct GNUNET_MessageHeader *msg = buf;
 350   uint32_t *data;
 351
 352   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 353               "tmt_rdy called, filling buffer\n");
 354   if (size < size_payload || NULL == buf)
 355   {
 356     GNUNET_break (ok >= ok_goal - 2);
 357     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 358                 "size %u, buf %p, data_sent %u, data_received %u\n",
 359                 size, buf, data_sent, data_received);
 360     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "ok %u, ok goal %u\n", ok, ok_goal);
 361
 362     return 0;
 363   }
 364   msg->size = htons (size);
 365   msg->type = htons ((long) cls);
 366   data = (uint32_t *) &msg[1];
 367   *data = htonl (data_sent);
 368   if (GNUNET_NO == initialized)
 369   {
 370     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 371                 "sending initializer\n");
 372   }
 373   else if (SPEED == test)
 374   {
 375     data_sent++;
 376     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 377               " Sent packet %d\n", data_sent);
 378     if (data_sent < TOTAL_PACKETS)
 379     {
 380       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 381               " Scheduling packet %d\n", data_sent + 1);
 382       GNUNET_SCHEDULER_add_now (&data_task, NULL);
 383     }
 384   }
 385
 386   return size_payload;
 387 }
 388
 389
 390 /**
 391  * Function is called whenever a message is received.
 392  *
 393  * @param cls closure (set from GNUNET_MESH_connect)
 394  * @param channel connection to the other end
 395  * @param channel_ctx place to store local state associated with the channel
 396  * @param message the actual message
 397  * @return GNUNET_OK to keep the connection open,
 398  *         GNUNET_SYSERR to close it (signal serious error)
 399  */
 400 int
 401 data_callback (void *cls, struct GNUNET_MESH_Channel *channel,
 402                void **channel_ctx,
 403                const struct GNUNET_MessageHeader *message)
 404 {
 405   long client = (long) cls;
 406   long expected_target_client;
 407   uint32_t *data;
 408
 409   ok++;
 410
 411   GNUNET_MESH_receive_done (channel);
 412
 413   if ((ok % 20) == 0)
 414   {
 415     if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
 416     {
 417       GNUNET_SCHEDULER_cancel (disconnect_task);
 418       disconnect_task = GNUNET_SCHEDULER_add_delayed (SHORT_TIME,
 419                                                       &disconnect_mesh_peers,
 420                                                       (void *) __LINE__);
 421     }
 422   }
 423
 424   switch (client)
 425   {
 426   case 0L:
 427     GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Root client got a message!\n");
 428     break;
 429   case 4L:
 430     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
 431                 "Leaf client %li got a message.\n",
 432                 client);
 433     break;
 434   default:
 435     GNUNET_assert (0);
 436     break;
 437   }
 438   GNUNET_log (GNUNET_ERROR_TYPE_INFO, " ok: (%d/%d)\n", ok, ok_goal);
 439   data = (uint32_t *) &message[1];
 440   GNUNET_log (GNUNET_ERROR_TYPE_INFO, " payload: (%u)\n", ntohl (*data));
 441   if (SPEED == test && GNUNET_YES == test_backwards)
 442   {
 443     expected_target_client = 0L;
 444   }
 445   else
 446   {
 447     expected_target_client = 4L;
 448   }
 449
 450   if (GNUNET_NO == initialized)
 451   {
 452     initialized = GNUNET_YES;
 453     start_time = GNUNET_TIME_absolute_get ();
 454     if (SPEED == test)
 455     {
 456       GNUNET_assert (4L == client);
 457       GNUNET_SCHEDULER_add_now (&data_task, NULL);
 458       return GNUNET_OK;
 459     }
 460   }
 461
 462   if (client == expected_target_client) // Normally 4
 463   {
 464     data_received++;
 465     GNUNET_log (GNUNET_ERROR_TYPE_INFO, " received data %u\n", data_received);
 466     if (SPEED != test || (ok_goal - 2) == ok)
 467     {
 468       GNUNET_MESH_notify_transmit_ready (channel, GNUNET_NO,
 469                                          GNUNET_TIME_UNIT_FOREVER_REL,
 470                                          size_payload, &tmt_rdy, (void *) 1L);
 471       return GNUNET_OK;
 472     }
 473     else
 474     {
 475       if (data_received < TOTAL_PACKETS)
 476         return GNUNET_OK;
 477     }
 478   }
 479   else // Normally 0
 480   {
 481     if (test == SPEED_ACK || test == SPEED)
 482     {
 483       data_ack++;
 484       GNUNET_log (GNUNET_ERROR_TYPE_INFO, " received ack %u\n", data_ack);
 485       GNUNET_MESH_notify_transmit_ready (channel, GNUNET_NO,
 486                                          GNUNET_TIME_UNIT_FOREVER_REL,
 487                                          size_payload, &tmt_rdy, (void *) 1L);
 488       if (data_ack < TOTAL_PACKETS && SPEED != test)
 489         return GNUNET_OK;
 490       if (ok == 2 && SPEED == test)
 491         return GNUNET_OK;
 492       show_end_data();
 493     }
 494     if (test == P2P_SIGNAL)
 495     {
 496       GNUNET_MESH_channel_destroy (incoming_ch);
 497       incoming_ch = NULL;
 498     }
 499     else
 500     {
 501       GNUNET_MESH_channel_destroy (ch);
 502       ch = NULL;
 503     }
 504   }
 505
 506   if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
 507   {
 508     GNUNET_SCHEDULER_cancel (disconnect_task);
 509     disconnect_task = GNUNET_SCHEDULER_add_delayed (SHORT_TIME,
 510                                                     &disconnect_mesh_peers,
 511                                                     (void *) __LINE__);
 512   }
 513
 514   return GNUNET_OK;
 515 }
 516
 517
 518
 519
 520 /**
 521  * Stats callback. Finish the stats testbed operation and when all stats have
 522  * been iterated, shutdown the test.
 523  *
 524  * @param cls closure
 525  * @param op the operation that has been finished
 526  * @param emsg error message in case the operation has failed; will be NULL if
 527  *          operation has executed successfully.
 528  */
 529 static void
 530 stats_cont (void *cls, struct GNUNET_TESTBED_Operation *op, const char *emsg)
 531 {
 532   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "stats_cont for peer %u\n", cls);
 533   GNUNET_TESTBED_operation_done (stats_op);
 534
 535   if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
 536     GNUNET_SCHEDULER_cancel (disconnect_task);
 537   disconnect_task = GNUNET_SCHEDULER_add_now (&disconnect_mesh_peers,
 538                                               (void *) __LINE__);
 539
 540 }
 541
 542
 543 /**
 544  * Process statistic values.
 545  *
 546  * @param cls closure
 547  * @param peer the peer the statistic belong to
 548  * @param subsystem name of subsystem that created the statistic
 549  * @param name the name of the datum
 550  * @param value the current value
 551  * @param is_persistent GNUNET_YES if the value is persistent, GNUNET_NO if not
 552  * @return GNUNET_OK to continue, GNUNET_SYSERR to abort iteration
 553  */
 554 static int
 555 stats_iterator (void *cls, const struct GNUNET_TESTBED_Peer *peer,
 556                 const char *subsystem, const char *name,
 557                 uint64_t value, int is_persistent)
 558 {
 559   static const char *s_sent = "# keepalives sent";
 560   static const char *s_recv = "# keepalives received";
 561   uint32_t i;
 562
 563   i = GNUNET_TESTBED_get_index (peer);
 564   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  %u - %s [%s]: %llu\n",
 565               i, subsystem, name, value);
 566   if (0 == strncmp (s_sent, name, strlen (s_sent)) && 0 == i)
 567     ka_sent = value;
 568
 569   if (0 == strncmp(s_recv, name, strlen (s_recv)) && 4 == i)
 570   {
 571     ka_received = value;
 572     GNUNET_log (GNUNET_ERROR_TYPE_INFO, " sent: %u, received: %u\n",
 573                 ka_sent, ka_received);
 574     if (ka_sent < 2 || ka_sent > ka_received + 1)
 575       ok--;
 576   }
 577
 578   return GNUNET_OK;
 579 }
 580
 581
 582 /**
 583  * Task check that keepalives were sent and received.
 584  *
 585  * @param cls Closure (NULL).
 586  * @param tc Task Context.
 587  */
 588 static void
 589 check_keepalives (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 590 {
 591   if ((GNUNET_SCHEDULER_REASON_SHUTDOWN & tc->reason) != 0)
 592     return;
 593
 594   disconnect_task = GNUNET_SCHEDULER_NO_TASK;
 595   GNUNET_log (GNUNET_ERROR_TYPE_INFO, "check keepalives\n");
 596   GNUNET_MESH_channel_destroy (ch);
 597   stats_op = GNUNET_TESTBED_get_statistics (5, testbed_peers,
 598                                             "mesh", NULL,
 599                                             stats_iterator, stats_cont, NULL);
 600 }
 601
 602
 603 /**
 604  * Handlers, for diverse services
 605  */
 606 static struct GNUNET_MESH_MessageHandler handlers[] = {
 607   {&data_callback, 1, sizeof (struct GNUNET_MessageHeader)},
 608   {NULL, 0, 0}
 609 };
 610
 611
 612 /**
 613  * Method called whenever another peer has added us to a channel
 614  * the other peer initiated.
 615  *
 616  * @param cls Closure.
 617  * @param channel New handle to the channel.
 618  * @param initiator Peer that started the channel.
 619  * @param port Port this channel is connected to.
 620  * @param options channel option flags
 621  * @return Initial channel context for the channel
 622  *         (can be NULL -- that's not an error).
 623  */
 624 static void *
 625 incoming_channel (void *cls, struct GNUNET_MESH_Channel *channel,
 626                  const struct GNUNET_PeerIdentity *initiator,
 627                  uint32_t port, enum GNUNET_MESH_ChannelOption options)
 628 {
 629   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
 630               "Incoming channel from %s to peer %d\n",
 631               GNUNET_i2s (initiator), (long) cls);
 632   ok++;
 633   GNUNET_log (GNUNET_ERROR_TYPE_INFO, " ok: %d\n", ok);
 634   if ((long) cls == 4L)
 635     incoming_ch = channel;
 636   else
 637   {
 638     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 639                 "Incoming channel for unknown client %lu\n", (long) cls);
 640     GNUNET_break(0);
 641   }
 642   if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
 643   {
 644     GNUNET_SCHEDULER_cancel (disconnect_task);
 645     if (KEEPALIVE == test)
 646     {
 647       struct GNUNET_TIME_Relative delay;
 648       delay = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS , 5);
 649       disconnect_task =
 650         GNUNET_SCHEDULER_add_delayed (delay, &check_keepalives, NULL);
 651     }
 652     else
 653       disconnect_task = GNUNET_SCHEDULER_add_delayed (SHORT_TIME,
 654                                                       &disconnect_mesh_peers,
 655                                                       (void *) __LINE__);
 656   }
 657
 658   return NULL;
 659 }
 660
 661 /**
 662  * Function called whenever an inbound channel is destroyed.  Should clean up
 663  * any associated state.
 664  *
 665  * @param cls closure (set from GNUNET_MESH_connect)
 666  * @param channel connection to the other end (henceforth invalid)
 667  * @param channel_ctx place where local state associated
 668  *                   with the channel is stored
 669  */
 670 static void
 671 channel_cleaner (void *cls, const struct GNUNET_MESH_Channel *channel,
 672                  void *channel_ctx)
 673 {
 674   long i = (long) cls;
 675
 676   GNUNET_log (GNUNET_ERROR_TYPE_INFO,
 677               "Incoming channel disconnected at peer %d\n",
 678               i);
 679   if (4L == i)
 680   {
 681     ok++;
 682     GNUNET_break (channel == incoming_ch);
 683     incoming_ch = NULL;
 684   }
 685   else if (0L == i)
 686   {
 687     if (P2P_SIGNAL == test)
 688     {
 689       ok ++;
 690     }
 691     GNUNET_break (channel == ch);
 692     ch = NULL;
 693   }
 694   else
 695     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 696                 "Unknown peer! %d\n", i);
 697   GNUNET_log (GNUNET_ERROR_TYPE_INFO, " ok: %d\n", ok);
 698
 699   if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
 700   {
 701     GNUNET_SCHEDULER_cancel (disconnect_task);
 702     disconnect_task = GNUNET_SCHEDULER_add_now (&disconnect_mesh_peers,
 703                                                 (void *) __LINE__);
 704   }
 705
 706   return;
 707 }
 708
 709
 710 /**
 711  * START THE TESTCASE ITSELF, AS WE ARE CONNECTED TO THE MESH SERVICES.
 712  *
 713  * Testcase continues when the root receives confirmation of connected peers,
 714  * on callback funtion ch.
 715  *
 716  * @param cls Closure (unsued).
 717  * @param tc Task Context.
 718  */
 719 static void
 720 do_test (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 721 {
 722   enum GNUNET_MESH_ChannelOption flags;
 723
 724   if ((GNUNET_SCHEDULER_REASON_SHUTDOWN & tc->reason) != 0)
 725     return;
 726
 727   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test_task\n");
 728
 729   if (GNUNET_SCHEDULER_NO_TASK != disconnect_task)
 730   {
 731     GNUNET_SCHEDULER_cancel (disconnect_task);
 732   }
 733
 734   flags = GNUNET_MESH_OPTION_DEFAULT;
 735   if (SPEED_REL == test)
 736   {
 737     test = SPEED;
 738     flags |= GNUNET_MESH_OPTION_RELIABLE;
 739   }
 740   ch = GNUNET_MESH_channel_create (h1, NULL, p_id[1], 1, flags);
 741
 742   disconnect_task = GNUNET_SCHEDULER_add_delayed (SHORT_TIME,
 743                                                   &disconnect_mesh_peers,
 744                                                   (void *) __LINE__);
 745   if (KEEPALIVE == test)
 746     return; /* Don't send any data. */
 747
 748   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 749               "Sending data initializer...\n");
 750   data_ack = 0;
 751   data_received = 0;
 752   data_sent = 0;
 753   GNUNET_MESH_notify_transmit_ready (ch, GNUNET_NO,
 754                                      GNUNET_TIME_UNIT_FOREVER_REL,
 755                                      size_payload, &tmt_rdy, (void *) 1L);
 756 }
 757
 758 /**
 759  * Callback to be called when the requested peer information is available
 760  *
 761  * @param cls the closure from GNUNET_TESTBED_peer_get_information()
 762  * @param op the operation this callback corresponds to
 763  * @param pinfo the result; will be NULL if the operation has failed
 764  * @param emsg error message if the operation has failed;
 765  *             NULL if the operation is successfull
 766  */
 767 static void
 768 pi_cb (void *cls,
 769        struct GNUNET_TESTBED_Operation *op,
 770        const struct GNUNET_TESTBED_PeerInformation *pinfo,
 771        const char *emsg)
 772 {
 773   long i = (long) cls;
 774
 775   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "id callback for %ld\n", i);
 776
 777   if (NULL == pinfo || NULL != emsg)
 778   {
 779     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "pi_cb: %s\n", emsg);
 780     abort_test (__LINE__);
 781     return;
 782   }
 783   p_id[i] = pinfo->result.id;
 784   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "  id: %s\n", GNUNET_i2s (p_id[i]));
 785   p_ids++;
 786   if (p_ids < 2)
 787     return;
 788   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got all IDs, starting test\n");
 789   test_task = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS,
 790                                             &do_test, NULL);
 791 }
 792
 793 /**
 794  * test main: start test when all peers are connected
 795  *
 796  * @param cls Closure.
 797  * @param ctx Argument to give to GNUNET_MESH_TEST_cleanup on test end.
 798  * @param num_peers Number of peers that are running.
 799  * @param peers Array of peers.
 800  * @param meshes Handle to each of the MESHs of the peers.
 801  */
 802 static void
 803 tmain (void *cls,
 804        struct GNUNET_MESH_TEST_Context *ctx,
 805        unsigned int num_peers,
 806        struct GNUNET_TESTBED_Peer **peers,
 807        struct GNUNET_MESH_Handle **meshes)
 808 {
 809   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "test main\n");
 810   ok = 0;
 811   test_ctx = ctx;
 812   peers_running = num_peers;
 813   testbed_peers = peers;
 814   h1 = meshes[0];
 815   h2 = meshes[num_peers - 1];
 816   disconnect_task = GNUNET_SCHEDULER_add_delayed (SHORT_TIME,
 817                                                   &disconnect_mesh_peers,
 818                                                   (void *) __LINE__);
 819   shutdown_handle = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
 820                                                   &shutdown_task, NULL);
 821   t_op[0] = GNUNET_TESTBED_peer_get_information (peers[0],
 822                                                  GNUNET_TESTBED_PIT_IDENTITY,
 823                                                  &pi_cb, (void *) 0L);
 824   t_op[1] = GNUNET_TESTBED_peer_get_information (peers[num_peers - 1],
 825                                                  GNUNET_TESTBED_PIT_IDENTITY,
 826                                                  &pi_cb, (void *) 1L);
 827   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "requested peer ids\n");
 828 }
 829
 830
 831 /**
 832  * Main: start test
 833  */
 834 int
 835 main (int argc, char *argv[])
 836 {
 837   initialized = GNUNET_NO;
 838   uint32_t ports[2];
 839   const char *config_file;
 840
 841   GNUNET_log_setup ("test", "DEBUG", NULL);
 842   config_file = "test_mesh.conf";
 843
 844   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Start\n");
 845   if (strstr (argv[0], "_small_forward") != NULL)
 846   {
 847     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "FORWARD\n");
 848     test = FORWARD;
 849     test_name = "unicast";
 850     ok_goal = 4;
 851   }
 852   else if (strstr (argv[0], "_small_signal") != NULL)
 853   {
 854     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "SIGNAL\n");
 855     test = P2P_SIGNAL;
 856     test_name = "signal";
 857     ok_goal = 4;
 858   }
 859   else if (strstr (argv[0], "_small_speed_ack") != NULL)
 860   {
 861     /* Test is supposed to generate the following callbacks:
 862      * 1 incoming channel (@dest)
 863      * TOTAL_PACKETS received data packet (@dest)
 864      * TOTAL_PACKETS received data packet (@orig)
 865      * 1 received channel destroy (@dest)
 866      */
 867     ok_goal = TOTAL_PACKETS * 2 + 2;
 868     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "SPEED_ACK\n");
 869     test = SPEED_ACK;
 870     test_name = "speed ack";
 871   }
 872   else if (strstr (argv[0], "_small_speed") != NULL)
 873   {
 874     /* Test is supposed to generate the following callbacks:
 875      * 1 incoming channel (@dest)
 876      * 1 initial packet (@dest)
 877      * TOTAL_PACKETS received data packet (@dest)
 878      * 1 received data packet (@orig)
 879      * 1 received channel destroy (@dest)
 880      */
 881     ok_goal = TOTAL_PACKETS + 4;
 882     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "SPEED\n");
 883     if (strstr (argv[0], "_reliable") != NULL)
 884     {
 885       test = SPEED_REL;
 886       test_name = "speed reliable";
 887       config_file = "test_mesh_drop.conf";
 888     }
 889     else
 890     {
 891       test = SPEED;
 892       test_name = "speed";
 893     }
 894   }
 895   else if (strstr (argv[0], "_keepalive") != NULL)
 896   {
 897     test = KEEPALIVE;
 898     /* Test is supposed to generate the following callbacks:
 899      * 1 incoming channel (@dest)
 900      * [wait]
 901      * 1 received channel destroy (@dest)
 902      */
 903     ok_goal = 2;
 904   }
 905   else
 906   {
 907     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "UNKNOWN\n");
 908     test = SETUP;
 909     ok_goal = 0;
 910   }
 911
 912   if (strstr (argv[0], "backwards") != NULL)
 913   {
 914     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "BACKWARDS (LEAF TO ROOT)\n");
 915     test_backwards = GNUNET_YES;
 916     GNUNET_asprintf (&test_name, "backwards %s", test_name);
 917   }
 918
 919   p_ids = 0;
 920   ports[0] = 1;
 921   ports[1] = 0;
 922   GNUNET_MESH_TEST_run ("test_mesh_small",
 923                         config_file,
 924                         5,
 925                         &tmain,
 926                         NULL, /* tmain cls */
 927                         &incoming_channel,
 928                         &channel_cleaner,
 929                         handlers,
 930                         ports);
 931
 932   if (ok_goal > ok)
 933   {
 934     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 935                 "FAILED! (%d/%d)\n", ok, ok_goal);
 936     return 1;
 937   }
 938   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "success\n");
 939   return 0;
 940 }
 941
 942 /* end of test_mesh_small.c */
 943