src/rps/test_rps.c

   1 /*
   2      This file is part of GNUnet.
   3      Copyright (C) 2009, 2012 GNUnet e.V.
   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., 51 Franklin Street, Fifth Floor,
  18      Boston, MA 02110-1301, USA.
  19 */
  20 /**
  21  * @file rps/test_rps_multipeer.c
  22  * @brief Testcase for the random peer sampling service.  Starts
  23  *        a peergroup with a given number of peers, then waits to
  24  *        receive size pushes/pulls from each peer.  Expects to wait
  25  *        for one message from each peer.
  26  */
  27 #include "platform.h"
  28 #include "gnunet_util_lib.h"
  29 #include "gnunet_testbed_service.h"
  30
  31 #include "gnunet_rps_service.h"
  32 #include "rps-test_util.h"
  33 #include "gnunet-service-rps_sampler_elem.h"
  34
  35 #include <inttypes.h>
  36
  37
  38 /**
  39  * How many peers do we start?
  40  */
  41 uint32_t num_peers;
  42
  43 /**
  44  * How long do we run the test?
  45  */
  46 //#define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30)
  47 static struct GNUNET_TIME_Relative timeout;
  48
  49
  50 /**
  51  * Portion of malicious peers
  52  */
  53 static double portion = .1;
  54
  55 /**
  56  * Type of malicious peer to test
  57  */
  58 static unsigned int mal_type = 0;
  59
  60 /**
  61  * Handles to all of the running peers
  62  */
  63 static struct GNUNET_TESTBED_Peer **testbed_peers;
  64
  65
  66 /**
  67  * Operation map entry
  68  */
  69 struct OpListEntry
  70 {
  71   /**
  72    * DLL next ptr
  73    */
  74   struct OpListEntry *next;
  75
  76   /**
  77    * DLL prev ptr
  78    */
  79   struct OpListEntry *prev;
  80
  81   /**
  82    * The testbed operation
  83    */
  84   struct GNUNET_TESTBED_Operation *op;
  85
  86   /**
  87    * Depending on whether we start or stop NSE service at the peer set this to 1
  88    * or -1
  89    */
  90   int delta;
  91
  92   /**
  93    * Index of the regarding peer
  94    */
  95   unsigned int index;
  96 };
  97
  98 /**
  99  * OpList DLL head
 100  */
 101 static struct OpListEntry *oplist_head;
 102
 103 /**
 104  * OpList DLL tail
 105  */
 106 static struct OpListEntry *oplist_tail;
 107
 108
 109 /**
 110  * A pending reply: A request was sent and the reply is pending.
 111  */
 112 struct PendingReply
 113 {
 114   /**
 115    * DLL next,prev ptr
 116    */
 117   struct PendingReply *next;
 118   struct PendingReply *prev;
 119
 120   /**
 121    * Handle to the request we are waiting for
 122    */
 123   struct GNUNET_RPS_Request_Handle *req_handle;
 124
 125   /**
 126    * The peer that requested
 127    */
 128   struct RPSPeer *rps_peer;
 129 };
 130
 131
 132 /**
 133  * A pending request: A request was not made yet but is scheduled for later.
 134  */
 135 struct PendingRequest
 136 {
 137   /**
 138    * DLL next,prev ptr
 139    */
 140   struct PendingRequest *next;
 141   struct PendingRequest *prev;
 142
 143   /**
 144    * Handle to the request we are waiting for
 145    */
 146   struct GNUNET_SCHEDULER_Task *request_task;
 147
 148   /**
 149    * The peer that requested
 150    */
 151   struct RPSPeer *rps_peer;
 152 };
 153
 154
 155 /**
 156  * Information we track for each peer.
 157  */
 158 struct RPSPeer
 159 {
 160   /**
 161    * Index of the peer.
 162    */
 163   unsigned int index;
 164
 165   /**
 166    * Handle for RPS connect operation.
 167    */
 168   struct GNUNET_TESTBED_Operation *op;
 169
 170   /**
 171    * Handle to RPS service.
 172    */
 173   struct GNUNET_RPS_Handle *rps_handle;
 174
 175   /**
 176    * ID of the peer.
 177    */
 178   struct GNUNET_PeerIdentity *peer_id;
 179
 180   /**
 181    * A request handle to check for an request
 182    */
 183   //struct GNUNET_RPS_Request_Handle *req_handle;
 184
 185   /**
 186    * Peer on- or offline?
 187    */
 188   int online;
 189
 190   /**
 191    * Number of Peer IDs to request
 192    */
 193   unsigned int num_ids_to_request;
 194
 195   /**
 196    * Pending requests DLL
 197    */
 198   struct PendingRequest *pending_req_head;
 199   struct PendingRequest *pending_req_tail;
 200
 201   /**
 202    * Number of pending requests
 203    */
 204   unsigned int num_pending_reqs;
 205
 206   /**
 207    * Pending replies DLL
 208    */
 209   struct PendingReply *pending_rep_head;
 210   struct PendingReply *pending_rep_tail;
 211
 212   /**
 213    * Number of pending replies
 214    */
 215   unsigned int num_pending_reps;
 216
 217   /**
 218    * Number of received PeerIDs
 219    */
 220   unsigned int num_recv_ids;
 221 };
 222
 223
 224 /**
 225  * Information for all the peers.
 226  */
 227 static struct RPSPeer *rps_peers;
 228
 229 /**
 230  * Peermap to get the index of a given peer ID quick.
 231  */
 232 static struct GNUNET_CONTAINER_MultiPeerMap *peer_map;
 233
 234 /**
 235  * IDs of the peers.
 236  */
 237 static struct GNUNET_PeerIdentity *rps_peer_ids;
 238
 239 /**
 240  * ID of the targeted peer.
 241  */
 242 static struct GNUNET_PeerIdentity *target_peer;
 243
 244 /**
 245  * ID of the peer that requests for the evaluation.
 246  */
 247 static struct RPSPeer *eval_peer;
 248
 249 /**
 250  * Number of online peers.
 251  */
 252 static unsigned int num_peers_online;
 253
 254 /**
 255  * Return value from 'main'.
 256  */
 257 static int ok;
 258
 259
 260 /**
 261  * Identifier for the churn task that runs periodically
 262  */
 263 static struct GNUNET_SCHEDULER_Task *churn_task;
 264
 265 /**
 266  * Identifier for the churn task that runs periodically
 267  */
 268 static struct GNUNET_SCHEDULER_Task *shutdown_task;
 269
 270
 271 /**
 272  * Called to initialise the given RPSPeer
 273  */
 274 typedef void (*InitPeer) (struct RPSPeer *rps_peer);
 275
 276 /**
 277  * Called directly after connecting to the service
 278  */
 279 typedef void (*PreTest) (void *cls, struct GNUNET_RPS_Handle *h);
 280
 281 /**
 282  * Called from within #rps_connect_complete_cb ()
 283  * Executes functions to test the api/service
 284  */
 285 typedef void (*MainTest) (struct RPSPeer *rps_peer);
 286
 287 /**
 288  * Callback called once the requested random peers are available
 289  */
 290 typedef void (*ReplyHandle) (void *cls,
 291                              uint64_t n,
 292                              const struct GNUNET_PeerIdentity *recv_peers);
 293
 294 /**
 295  * Called directly before disconnecting from the service
 296  */
 297 typedef void (*PostTest) (void *cls, struct GNUNET_RPS_Handle *h);
 298
 299 /**
 300  * Function called after disconnect to evaluate test success
 301  */
 302 typedef int (*EvaluationCallback) (void);
 303
 304
 305 /**
 306  * Structure to define a single test
 307  */
 308 struct SingleTestRun
 309 {
 310   /**
 311    * Name of the test
 312    */
 313   char *name;
 314
 315   /**
 316    * Called to initialise peer
 317    */
 318   InitPeer init_peer;
 319
 320   /**
 321    * Called directly after connecting to the service
 322    */
 323   PreTest pre_test;
 324
 325   /**
 326    * Function to execute the functions to be tested
 327    */
 328   MainTest main_test;
 329
 330   /**
 331    * Callback called once the requested peers are available
 332    */
 333   ReplyHandle reply_handle;
 334
 335   /**
 336    * Called directly before disconnecting from the service
 337    */
 338   PostTest post_test;
 339
 340   /**
 341    * Function to evaluate the test results
 342    */
 343   EvaluationCallback eval_cb;
 344
 345   /**
 346    * Request interval
 347    */
 348   uint32_t request_interval;
 349
 350   /**
 351    * Number of Requests to make.
 352    */
 353   uint32_t num_requests;
 354 } cur_test_run;
 355
 356 /**
 357  * Are we shutting down?
 358  */
 359 static int in_shutdown;
 360
 361 /**
 362  * Append arguments to file
 363  */
 364 static void
 365 tofile_ (const char *file_name, char *line)
 366 {
 367   struct GNUNET_DISK_FileHandle *f;
 368   /* char output_buffer[512]; */
 369   size_t size;
 370   /* int size; */
 371   size_t size2;
 372
 373   if (NULL == (f = GNUNET_DISK_file_open (file_name,
 374                                           GNUNET_DISK_OPEN_APPEND |
 375                                           GNUNET_DISK_OPEN_WRITE |
 376                                           GNUNET_DISK_OPEN_CREATE,
 377                                           GNUNET_DISK_PERM_USER_READ |
 378                                           GNUNET_DISK_PERM_USER_WRITE |
 379                                           GNUNET_DISK_PERM_GROUP_READ |
 380                                           GNUNET_DISK_PERM_OTHER_READ)))
 381   {
 382     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 383                 "Not able to open file %s\n",
 384                 file_name);
 385     return;
 386   }
 387   /* size = GNUNET_snprintf (output_buffer,
 388                           sizeof (output_buffer),
 389                           "%llu %s\n",
 390                           GNUNET_TIME_absolute_get ().abs_value_us,
 391                           line);
 392   if (0 > size)
 393   {
 394     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 395                 "Failed to write string to buffer (size: %i)\n",
 396                 size);
 397     return;
 398   } */
 399
 400   size = strlen (line) * sizeof (char);
 401
 402   size2 = GNUNET_DISK_file_write (f, line, size);
 403   if (size != size2)
 404   {
 405     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 406                 "Unable to write to file! (Size: %u, size2: %u)\n",
 407                 size,
 408                 size2);
 409     return;
 410   }
 411
 412   if (GNUNET_YES != GNUNET_DISK_file_close (f))
 413     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 414                 "Unable to close file\n");
 415 }
 416
 417 /**
 418  * This function is used to facilitate writing important information to disk
 419  */
 420 #define tofile(file_name, ...) do {\
 421   char tmp_buf[512];\
 422     int size;\
 423     size = GNUNET_snprintf(tmp_buf,sizeof(tmp_buf),__VA_ARGS__);\
 424     if (0 > size)\
 425       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,\
 426                      "Failed to create tmp_buf\n");\
 427     else\
 428       tofile_(file_name,tmp_buf);\
 429   } while (0);
 430
 431
 432 /**
 433  * Write the ids and their according index in the given array to a file
 434  * Unused
 435  */
 436 /* static void
 437 ids_to_file (char *file_name,
 438              struct GNUNET_PeerIdentity *peer_ids,
 439              unsigned int num_peer_ids)
 440 {
 441   unsigned int i;
 442
 443   for (i=0 ; i < num_peer_ids ; i++)
 444   {
 445     to_file (file_name,
 446              "%u\t%s",
 447              i,
 448              GNUNET_i2s_full (&peer_ids[i]));
 449   }
 450 } */
 451
 452 /**
 453  * Test the success of a single test
 454  */
 455 static int
 456 evaluate (void)
 457 {
 458   unsigned int i;
 459   int tmp_ok;
 460
 461   tmp_ok = 1;
 462
 463   for (i = 0; i < num_peers; i++)
 464   {
 465     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 466         "%u. peer [%s] received %u of %u expected peer_ids: %i\n",
 467         i,
 468         GNUNET_i2s (rps_peers[i].peer_id),
 469         rps_peers[i].num_recv_ids,
 470         rps_peers[i].num_ids_to_request,
 471         (rps_peers[i].num_ids_to_request == rps_peers[i].num_recv_ids));
 472     tmp_ok &= (rps_peers[i].num_ids_to_request == rps_peers[i].num_recv_ids);
 473   }
 474   return tmp_ok? 0 : 1;
 475 }
 476
 477
 478 /**
 479  * Creates an oplist entry and adds it to the oplist DLL
 480  */
 481 static struct OpListEntry *
 482 make_oplist_entry ()
 483 {
 484   struct OpListEntry *entry;
 485
 486   entry = GNUNET_new (struct OpListEntry);
 487   GNUNET_CONTAINER_DLL_insert_tail (oplist_head, oplist_tail, entry);
 488   return entry;
 489 }
 490
 491
 492 /**
 493  * Task run on timeout to shut everything down.
 494  */
 495 static void
 496 shutdown_op (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 497 {
 498   unsigned int i;
 499
 500   in_shutdown = GNUNET_YES;
 501   if (NULL != churn_task)
 502   {
 503     GNUNET_SCHEDULER_cancel (churn_task);
 504     churn_task = NULL;
 505   }
 506   for (i = 0; i < num_peers; i++)
 507     if (NULL != rps_peers[i].op)
 508       GNUNET_TESTBED_operation_done (rps_peers[i].op);
 509   GNUNET_SCHEDULER_shutdown ();
 510 }
 511
 512
 513 /**
 514  * Seed peers.
 515  */
 516   void
 517 seed_peers (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 518 {
 519   unsigned int amount;
 520   struct RPSPeer *peer = (struct RPSPeer *) cls;
 521   unsigned int i;
 522
 523   // TODO if malicious don't seed mal peers
 524   amount = round (.5 * num_peers);
 525
 526   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Seeding peers:\n");
 527   for (i = 0 ; i < amount ; i++)
 528     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Seeding %u. peer: %s\n",
 529                 i,
 530                 GNUNET_i2s (&rps_peer_ids[i]));
 531
 532   GNUNET_RPS_seed_ids (peer->rps_handle, amount, rps_peer_ids);
 533 }
 534
 535 /**
 536  * Seed peers.
 537  */
 538   void
 539 seed_peers_big (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 540 {
 541   struct RPSPeer *peer = (struct RPSPeer *) cls;
 542   unsigned int seed_msg_size;
 543   uint32_t num_peers_max;
 544   unsigned int amount;
 545   unsigned int i;
 546
 547   seed_msg_size = 8; /* sizeof (struct GNUNET_RPS_CS_SeedMessage) */
 548   num_peers_max = (GNUNET_SERVER_MAX_MESSAGE_SIZE - seed_msg_size) /
 549     sizeof (struct GNUNET_PeerIdentity);
 550   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 551       "Peers that fit in one seed msg; %u\n",
 552       num_peers_max);
 553   amount = num_peers_max + (0.5 * num_peers_max);
 554   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 555       "Seeding many (%u) peers:\n",
 556       amount);
 557   struct GNUNET_PeerIdentity ids_to_seed[amount];
 558   for (i = 0; i < amount; i++)
 559   {
 560     ids_to_seed[i] = *peer->peer_id;
 561     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Seeding %u. peer: %s\n",
 562                 i,
 563                 GNUNET_i2s (&ids_to_seed[i]));
 564   }
 565
 566   GNUNET_RPS_seed_ids (peer->rps_handle, amount, ids_to_seed);
 567 }
 568
 569 /**
 570  * Get the id of peer i.
 571  */
 572   void
 573 info_cb (void *cb_cls,
 574          struct GNUNET_TESTBED_Operation *op,
 575          const struct GNUNET_TESTBED_PeerInformation *pinfo,
 576          const char *emsg)
 577 {
 578   struct OpListEntry *entry = (struct OpListEntry *) cb_cls;
 579
 580   if (NULL == pinfo || NULL != emsg)
 581   {
 582     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Got Error: %s\n", emsg);
 583     GNUNET_TESTBED_operation_done (entry->op);
 584     return;
 585   }
 586
 587   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 588               "Peer %u is %s\n",
 589               entry->index,
 590               GNUNET_i2s (pinfo->result.id));
 591
 592   rps_peer_ids[entry->index] = *(pinfo->result.id);
 593   rps_peers[entry->index].peer_id = &rps_peer_ids[entry->index];
 594
 595   GNUNET_CONTAINER_multipeermap_put (peer_map,
 596       &rps_peer_ids[entry->index],
 597       &rps_peers[entry->index],
 598       GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
 599   tofile ("/tmp/rps/peer_ids",
 600            "%u\t%s\n",
 601            entry->index,
 602            GNUNET_i2s_full (&rps_peer_ids[entry->index]));
 603
 604   GNUNET_CONTAINER_DLL_remove (oplist_head, oplist_tail, entry);
 605   GNUNET_TESTBED_operation_done (entry->op);
 606   GNUNET_free (entry);
 607 }
 608
 609
 610 /**
 611  * Callback to be called when RPS service connect operation is completed
 612  *
 613  * @param cls the callback closure from functions generating an operation
 614  * @param op the operation that has been finished
 615  * @param ca_result the RPS service handle returned from rps_connect_adapter
 616  * @param emsg error message in case the operation has failed; will be NULL if
 617  *          operation has executed successfully.
 618  */
 619 static void
 620 rps_connect_complete_cb (void *cls,
 621                          struct GNUNET_TESTBED_Operation *op,
 622                          void *ca_result,
 623                          const char *emsg)
 624 {
 625   struct RPSPeer *rps_peer = cls;
 626   struct GNUNET_RPS_Handle *rps = ca_result;
 627
 628   rps_peer->rps_handle = rps;
 629   rps_peer->online = GNUNET_YES;
 630   num_peers_online++;
 631
 632   GNUNET_assert (op == rps_peer->op);
 633   if (NULL != emsg)
 634   {
 635     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 636                 "Failed to connect to RPS service: %s\n",
 637                 emsg);
 638     ok = 1;
 639     GNUNET_SCHEDULER_shutdown ();
 640     return;
 641   }
 642
 643   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Started client successfully\n");
 644
 645   cur_test_run.main_test (rps_peer);
 646 }
 647
 648
 649 /**
 650  * Adapter function called to establish a connection to
 651  * the RPS service.
 652  *
 653  * @param cls closure
 654  * @param cfg configuration of the peer to connect to; will be available until
 655  *          GNUNET_TESTBED_operation_done() is called on the operation returned
 656  *          from GNUNET_TESTBED_service_connect()
 657  * @return service handle to return in 'op_result', NULL on error
 658  */
 659 static void *
 660 rps_connect_adapter (void *cls,
 661                                  const struct GNUNET_CONFIGURATION_Handle *cfg)
 662 {
 663   struct GNUNET_RPS_Handle *h;
 664
 665   h = GNUNET_RPS_connect (cfg);
 666
 667   if (NULL != cur_test_run.pre_test)
 668     cur_test_run.pre_test (cls, h);
 669
 670   return h;
 671 }
 672
 673
 674 /**
 675  * Adapter function called to destroy connection to
 676  * RPS service.
 677  *
 678  * @param cls closure
 679  * @param op_result service handle returned from the connect adapter
 680  */
 681 static void
 682 rps_disconnect_adapter (void *cls,
 683                                           void *op_result)
 684 {
 685   struct RPSPeer *peer = cls;
 686   struct GNUNET_RPS_Handle *h = op_result;
 687   GNUNET_assert (NULL != peer);
 688   GNUNET_RPS_disconnect (h);
 689   peer->rps_handle = NULL;
 690 }
 691
 692
 693 /***********************************************************************
 694  * Definition of tests
 695 ***********************************************************************/
 696
 697 // TODO check whether tests can be stopped earlier
 698 static int
 699 default_eval_cb (void)
 700 {
 701   return evaluate ();
 702 }
 703
 704 static int
 705 no_eval (void)
 706 {
 707   return 0;
 708 }
 709
 710 /**
 711  * Initialise given RPSPeer
 712  */
 713 static void default_init_peer (struct RPSPeer *rps_peer)
 714 {
 715   rps_peer->num_ids_to_request = 1;
 716 }
 717
 718 /**
 719  * Callback to call on receipt of a reply
 720  *
 721  * @param cls closure
 722  * @param n number of peers
 723  * @param recv_peers the received peers
 724  */
 725 static void
 726 default_reply_handle (void *cls,
 727                       uint64_t n,
 728                       const struct GNUNET_PeerIdentity *recv_peers)
 729 {
 730   struct RPSPeer *rps_peer;
 731   struct PendingReply *pending_rep = (struct PendingReply *) cls;
 732   unsigned int i;
 733
 734   rps_peer = pending_rep->rps_peer;
 735   GNUNET_CONTAINER_DLL_remove (rps_peer->pending_rep_head,
 736                                rps_peer->pending_rep_tail,
 737                                pending_rep);
 738   rps_peer->num_pending_reps--;
 739   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 740               "[%s] got %" PRIu64 " peers:\n",
 741               GNUNET_i2s (rps_peer->peer_id),
 742               n);
 743
 744   for (i = 0; i < n; i++)
 745   {
 746     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 747                 "%u: %s\n",
 748                 i,
 749                 GNUNET_i2s (&recv_peers[i]));
 750
 751     rps_peer->num_recv_ids++;
 752   }
 753
 754   if (0 == evaluate ())
 755   {
 756     GNUNET_SCHEDULER_cancel (shutdown_task);
 757     shutdown_task = GNUNET_SCHEDULER_add_now (&shutdown_op, NULL);
 758   }
 759 }
 760
 761 /**
 762  * Request random peers.
 763  */
 764 static void
 765 request_peers (void *cls,
 766                const struct GNUNET_SCHEDULER_TaskContext *tc)
 767 {
 768   struct RPSPeer *rps_peer;
 769   struct PendingRequest *pending_req = (struct PendingRequest *) cls;
 770   struct PendingReply *pending_rep;
 771
 772   if (GNUNET_YES == in_shutdown)
 773     return;
 774   rps_peer = pending_req->rps_peer;
 775   GNUNET_assert (1 <= rps_peer->num_pending_reqs);
 776   GNUNET_CONTAINER_DLL_remove (rps_peer->pending_req_head,
 777                                rps_peer->pending_req_tail,
 778                                pending_req);
 779   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 780               "Requesting one peer\n");
 781   pending_rep = GNUNET_new (struct PendingReply);
 782   pending_rep->rps_peer = rps_peer;
 783   pending_rep->req_handle = GNUNET_RPS_request_peers (rps_peer->rps_handle,
 784       1,
 785       cur_test_run.reply_handle,
 786       pending_rep);
 787   GNUNET_CONTAINER_DLL_insert_tail (rps_peer->pending_rep_head,
 788                                     rps_peer->pending_rep_tail,
 789                                     pending_rep);
 790   rps_peer->num_pending_reps++;
 791   rps_peer->num_pending_reqs--;
 792 }
 793
 794 static void
 795 cancel_pending_req (struct PendingRequest *pending_req)
 796 {
 797   struct RPSPeer *rps_peer;
 798
 799   rps_peer = pending_req->rps_peer;
 800   GNUNET_CONTAINER_DLL_remove (rps_peer->pending_req_head,
 801                                rps_peer->pending_req_tail,
 802                                pending_req);
 803   rps_peer->num_pending_reqs--;
 804   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 805               "Cancelling pending request\n");
 806   GNUNET_SCHEDULER_cancel (pending_req->request_task);
 807   GNUNET_free (pending_req);
 808 }
 809
 810 static void
 811 cancel_request (struct PendingReply *pending_rep)
 812 {
 813   struct RPSPeer *rps_peer;
 814
 815   rps_peer = pending_rep->rps_peer;
 816   GNUNET_CONTAINER_DLL_remove (rps_peer->pending_rep_head,
 817                                rps_peer->pending_rep_tail,
 818                                pending_rep);
 819   rps_peer->num_pending_reps--;
 820   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 821               "Cancelling request\n");
 822   GNUNET_RPS_request_cancel (pending_rep->req_handle);
 823   GNUNET_free (pending_rep);
 824 }
 825
 826 /**
 827  * Cancel a request.
 828  */
 829 static void
 830 cancel_request_cb (void *cls,
 831                 const struct GNUNET_SCHEDULER_TaskContext *tc)
 832 {
 833   struct PendingReply *pending_rep;
 834   struct RPSPeer *rps_peer = (struct RPSPeer *) cls;
 835
 836   if (GNUNET_YES == in_shutdown)
 837     return;
 838   pending_rep = rps_peer->pending_rep_head;
 839   GNUNET_assert (1 <= rps_peer->num_pending_reps);
 840   cancel_request (pending_rep);
 841 }
 842
 843
 844 /**
 845  * Schedule requests for peer @a rps_peer that have neither been scheduled, nor
 846  * issued, nor replied
 847  */
 848 void
 849 schedule_missing_requests (struct RPSPeer *rps_peer)
 850 {
 851   unsigned int i;
 852   struct PendingRequest *pending_req;
 853
 854   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 855       "Scheduling %u - %u missing requests\n",
 856       rps_peer->num_ids_to_request,
 857       rps_peer->num_pending_reqs + rps_peer->num_pending_reps);
 858   GNUNET_assert (rps_peer->num_pending_reqs + rps_peer->num_pending_reps <=
 859       rps_peer->num_ids_to_request);
 860   for (i = rps_peer->num_pending_reqs + rps_peer->num_pending_reps;
 861        i < rps_peer->num_ids_to_request; i++)
 862   {
 863     pending_req = GNUNET_new (struct PendingRequest);
 864     pending_req->rps_peer = rps_peer;
 865     pending_req->request_task = GNUNET_SCHEDULER_add_delayed (
 866         GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS,
 867           cur_test_run.request_interval * i),
 868         request_peers,
 869         pending_req);
 870     GNUNET_CONTAINER_DLL_insert_tail (rps_peer->pending_req_head,
 871                                       rps_peer->pending_req_tail,
 872                                       pending_req);
 873     rps_peer->num_pending_reqs++;
 874   }
 875 }
 876
 877 void
 878 cancel_pending_req_rep (struct RPSPeer *rps_peer)
 879 {
 880   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 881       "Cancelling all (pending) requests.\n");
 882   while (NULL != rps_peer->pending_req_head)
 883     cancel_pending_req (rps_peer->pending_req_head);
 884   GNUNET_assert (0 == rps_peer->num_pending_reqs);
 885   while (NULL != rps_peer->pending_rep_head)
 886     cancel_request (rps_peer->pending_rep_head);
 887   GNUNET_assert (0 == rps_peer->num_pending_reps);
 888 }
 889
 890 /***********************************
 891  * MALICIOUS
 892 ***********************************/
 893
 894 /**
 895  * Initialise only non-mal RPSPeers
 896  */
 897 static void mal_init_peer (struct RPSPeer *rps_peer)
 898 {
 899   if (rps_peer->index >= round (portion * num_peers))
 900     rps_peer->num_ids_to_request = 1;
 901 }
 902
 903 static void
 904 mal_pre (void *cls, struct GNUNET_RPS_Handle *h)
 905 {
 906   #ifdef ENABLE_MALICIOUS
 907   uint32_t num_mal_peers;
 908   struct RPSPeer *rps_peer = (struct RPSPeer *) cls;
 909
 910   GNUNET_assert ( (1 >= portion) &&
 911                   (0 <  portion) );
 912   num_mal_peers = round (portion * num_peers);
 913
 914   if (rps_peer->index < num_mal_peers)
 915   {
 916     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 917                 "%u. peer [%s] of %" PRIu32 " malicious peers turning malicious\n",
 918                 rps_peer->index,
 919                 GNUNET_i2s (rps_peer->peer_id),
 920                 num_mal_peers);
 921
 922     GNUNET_RPS_act_malicious (h, mal_type, num_mal_peers,
 923                               rps_peer_ids, target_peer);
 924   }
 925   #endif /* ENABLE_MALICIOUS */
 926 }
 927
 928 static void
 929 mal_cb (struct RPSPeer *rps_peer)
 930 {
 931   uint32_t num_mal_peers;
 932
 933   #ifdef ENABLE_MALICIOUS
 934   GNUNET_assert ( (1 >= portion) &&
 935                   (0 <  portion) );
 936   num_mal_peers = round (portion * num_peers);
 937
 938   if (rps_peer->index >= num_mal_peers)
 939   { /* It's useless to ask a malicious peer about a random sample -
 940        it's not sampling */
 941     GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 2),
 942                                   seed_peers, rps_peer);
 943     schedule_missing_requests (rps_peer);
 944   }
 945   #endif /* ENABLE_MALICIOUS */
 946 }
 947
 948
 949 /***********************************
 950  * SINGLE_REQUEST
 951 ***********************************/
 952 static void
 953 single_req_cb (struct RPSPeer *rps_peer)
 954 {
 955   schedule_missing_requests (rps_peer);
 956 }
 957
 958 /***********************************
 959  * DELAYED_REQUESTS
 960 ***********************************/
 961 static void
 962 delay_req_cb (struct RPSPeer *rps_peer)
 963 {
 964   schedule_missing_requests (rps_peer);
 965 }
 966
 967 /***********************************
 968  * SEED
 969 ***********************************/
 970 static void
 971 seed_cb (struct RPSPeer *rps_peer)
 972 {
 973   GNUNET_SCHEDULER_add_delayed (
 974       GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 10),
 975       seed_peers, rps_peer);
 976 }
 977
 978 /***********************************
 979  * SEED_BIG
 980 ***********************************/
 981 static void
 982 seed_big_cb (struct RPSPeer *rps_peer)
 983 {
 984   // TODO test seeding > GNUNET_SERVER_MAX_MESSAGE_SIZE peers
 985   GNUNET_SCHEDULER_add_delayed (
 986       GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 2),
 987       seed_peers_big, rps_peer);
 988 }
 989
 990 /***********************************
 991  * SINGLE_PEER_SEED
 992 ***********************************/
 993 static void
 994 single_peer_seed_cb (struct RPSPeer *rps_peer)
 995 {
 996   // TODO
 997 }
 998
 999 /***********************************
1000  * SEED_REQUEST
1001 ***********************************/
1002 static void
1003 seed_req_cb (struct RPSPeer *rps_peer)
1004 {
1005   GNUNET_SCHEDULER_add_delayed (
1006       GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 2),
1007       seed_peers, rps_peer);
1008   schedule_missing_requests (rps_peer);
1009 }
1010
1011 //TODO start big mal
1012
1013 /***********************************
1014  * REQUEST_CANCEL
1015 ***********************************/
1016 static void
1017 req_cancel_cb (struct RPSPeer *rps_peer)
1018 {
1019   schedule_missing_requests (rps_peer);
1020   GNUNET_SCHEDULER_add_delayed (
1021       GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS,
1022                                      (cur_test_run.request_interval + 1)),
1023       cancel_request_cb, rps_peer);
1024 }
1025
1026 /***********************************
1027  * PROFILER
1028 ***********************************/
1029
1030 /**
1031  * Callback to be called when RPS service is started or stopped at peers
1032  *
1033  * @param cls NULL
1034  * @param op the operation handle
1035  * @param emsg NULL on success; otherwise an error description
1036  */
1037 static void
1038 churn_cb (void *cls,
1039           struct GNUNET_TESTBED_Operation *op,
1040           const char *emsg)
1041 {
1042   // FIXME
1043   struct OpListEntry *entry = cls;
1044
1045   GNUNET_TESTBED_operation_done (entry->op);
1046   if (NULL != emsg)
1047   {
1048     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Failed to start/stop RPS at a peer\n");
1049     GNUNET_SCHEDULER_shutdown ();
1050     return;
1051   }
1052   GNUNET_assert (0 != entry->delta);
1053
1054   num_peers_online += entry->delta;
1055
1056   if (0 > entry->delta)
1057   { /* Peer hopefully just went offline */
1058     if (GNUNET_YES != rps_peers[entry->index].online)
1059     {
1060       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1061                   "peer %s was expected to go offline but is still marked as online\n",
1062                   GNUNET_i2s (rps_peers[entry->index].peer_id));
1063       GNUNET_break (0);
1064     }
1065     else
1066     {
1067       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1068                   "peer %s probably went offline as expected\n",
1069                   GNUNET_i2s (rps_peers[entry->index].peer_id));
1070     }
1071     rps_peers[entry->index].online = GNUNET_NO;
1072   }
1073
1074   else if (0 < entry->delta)
1075   { /* Peer hopefully just went online */
1076     if (GNUNET_NO != rps_peers[entry->index].online)
1077     {
1078       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
1079                   "peer %s was expected to go online but is still marked as offline\n",
1080                   GNUNET_i2s (rps_peers[entry->index].peer_id));
1081       GNUNET_break (0);
1082     }
1083     else
1084     {
1085       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1086                   "peer %s probably went online as expected\n",
1087                   GNUNET_i2s (rps_peers[entry->index].peer_id));
1088       if (NULL != cur_test_run.pre_test)
1089       {
1090         cur_test_run.pre_test (&rps_peers[entry->index],
1091             rps_peers[entry->index].rps_handle);
1092         schedule_missing_requests (&rps_peers[entry->index]);
1093       }
1094     }
1095     rps_peers[entry->index].online = GNUNET_YES;
1096   }
1097
1098   GNUNET_CONTAINER_DLL_remove (oplist_head, oplist_tail, entry);
1099   GNUNET_free (entry);
1100   //if (num_peers_in_round[current_round] == peers_running)
1101   //  run_round ();
1102 }
1103
1104 static void
1105 manage_service_wrapper (unsigned int i, unsigned int j, int delta,
1106     double prob_go_on_off)
1107 {
1108   struct OpListEntry *entry;
1109   uint32_t prob;
1110
1111   prob = GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_WEAK,
1112                                    UINT32_MAX);
1113   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1114               "%u. selected peer (%u: %s) is %s.\n",
1115               i,
1116               j,
1117               GNUNET_i2s (rps_peers[j].peer_id),
1118               (0 > delta) ? "online" : "offline");
1119   if (prob < prob_go_on_off * UINT32_MAX)
1120   {
1121     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1122                 "%s goes %s\n",
1123                 GNUNET_i2s (rps_peers[j].peer_id),
1124                 (0 > delta) ? "offline" : "online");
1125
1126     if (0 > delta)
1127       cancel_pending_req_rep (&rps_peers[j]);
1128     entry = make_oplist_entry ();
1129     entry->delta = delta;
1130     entry->index = j;
1131     entry->op = GNUNET_TESTBED_peer_manage_service (NULL,
1132                                                     testbed_peers[j],
1133                                                     "rps",
1134                                                     &churn_cb,
1135                                                     entry,
1136                                                     (0 > delta) ? 0 : 1);
1137   }
1138 }
1139
1140 static void
1141 churn (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1142 {
1143   unsigned int i;
1144   unsigned int j;
1145   double portion_online;
1146   unsigned int *permut;
1147   double prob_go_offline;
1148   double portion_go_online;
1149   double portion_go_offline;
1150
1151   /* Compute the probability for an online peer to go offline
1152    * this round */
1153   portion_online = num_peers_online * 1.0 / num_peers;
1154   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1155               "Portion online: %f\n",
1156               portion_online);
1157   portion_go_online = ((1 - portion_online) * .5 * .66);
1158   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1159               "Portion that should go online: %f\n",
1160               portion_go_online);
1161   portion_go_offline = (portion_online + portion_go_online) - .75;
1162   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1163               "Portion that probably goes offline: %f\n",
1164               portion_go_offline);
1165   prob_go_offline = portion_go_offline / (portion_online * .5);
1166   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1167               "Probability of a selected online peer to go offline: %f\n",
1168               prob_go_offline);
1169
1170   permut = GNUNET_CRYPTO_random_permute (GNUNET_CRYPTO_QUALITY_WEAK,
1171                                          (unsigned int) num_peers);
1172
1173   /* Go over 50% randomly chosen peers */
1174   for (i = 0; i < .5 * num_peers; i++)
1175   {
1176     j = permut[i];
1177
1178     /* If online, shut down with certain probability */
1179     if (GNUNET_YES == rps_peers[j].online)
1180     {
1181       manage_service_wrapper (i, j, -1, prob_go_offline);
1182     }
1183
1184     /* If offline, restart with certain probability */
1185     else if (GNUNET_NO == rps_peers[j].online)
1186     {
1187       manage_service_wrapper (i, j, 1, 0.66);
1188     }
1189   }
1190
1191   GNUNET_free (permut);
1192
1193   churn_task = GNUNET_SCHEDULER_add_delayed (
1194         GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 2),
1195         churn,
1196         NULL);
1197 }
1198
1199
1200 /**
1201  * Initialise given RPSPeer
1202  */
1203 static void profiler_init_peer (struct RPSPeer *rps_peer)
1204 {
1205   if (num_peers - 1 == rps_peer->index)
1206     rps_peer->num_ids_to_request = cur_test_run.num_requests;
1207 }
1208
1209
1210 /**
1211  * Callback to call on receipt of a reply
1212  *
1213  * @param cls closure
1214  * @param n number of peers
1215  * @param recv_peers the received peers
1216  */
1217 static void
1218 profiler_reply_handle (void *cls,
1219                       uint64_t n,
1220                       const struct GNUNET_PeerIdentity *recv_peers)
1221 {
1222   struct RPSPeer *rps_peer;
1223   struct RPSPeer *rcv_rps_peer;
1224   char *file_name;
1225   char *file_name_dh;
1226   unsigned int i;
1227   struct PendingReply *pending_rep = (struct PendingReply *) cls;
1228
1229   rps_peer = pending_rep->rps_peer;
1230   file_name = "/tmp/rps/received_ids";
1231   file_name_dh = "/tmp/rps/diehard_input";
1232   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1233               "[%s] got %" PRIu64 " peers:\n",
1234               GNUNET_i2s (rps_peer->peer_id),
1235               n);
1236   for (i = 0; i < n; i++)
1237   {
1238     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1239                 "%u: %s\n",
1240                 i,
1241                 GNUNET_i2s (&recv_peers[i]));
1242     tofile (file_name,
1243              "%s\n",
1244              GNUNET_i2s_full (&recv_peers[i]));
1245     rcv_rps_peer = GNUNET_CONTAINER_multipeermap_get (peer_map, &recv_peers[i]);
1246     tofile (file_name_dh,
1247              "%" PRIu32 "\n",
1248              (uint32_t) rcv_rps_peer->index);
1249   }
1250   default_reply_handle (cls, n, recv_peers);
1251 }
1252
1253
1254 static void
1255 profiler_cb (struct RPSPeer *rps_peer)
1256 {
1257   /* Start churn */
1258   if (NULL == churn_task)
1259   {
1260     churn_task = GNUNET_SCHEDULER_add_delayed (
1261           GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5),
1262           churn,
1263           NULL);
1264   }
1265
1266   /* Only request peer ids at one peer.
1267    * (It's the before-last because last one is target of the focussed attack.)
1268    */
1269   if (eval_peer == rps_peer)
1270     schedule_missing_requests (rps_peer);
1271 }
1272
1273 /**
1274  * Function called from #profiler_eval with a filename.
1275  *
1276  * @param cls closure
1277  * @param filename complete filename (absolute path)
1278  * @return #GNUNET_OK to continue to iterate,
1279  *  #GNUNET_NO to stop iteration with no error,
1280  *  #GNUNET_SYSERR to abort iteration with error!
1281  */
1282 int
1283 file_name_cb (void *cls, const char *filename)
1284 {
1285   if (NULL != strstr (filename, "sampler_el"))
1286   {
1287     struct RPS_SamplerElement *s_elem;
1288     struct GNUNET_CRYPTO_AuthKey auth_key;
1289     const char *key_char;
1290     uint32_t i;
1291
1292     key_char = filename + 20; /* Length of "/tmp/rps/sampler_el-" */
1293     tofile (filename, "--------------------------\n");
1294
1295     auth_key = string_to_auth_key (key_char);
1296     s_elem = RPS_sampler_elem_create ();
1297     RPS_sampler_elem_set (s_elem, auth_key);
1298
1299     for (i = 0; i < num_peers; i++)
1300     {
1301       RPS_sampler_elem_next (s_elem, &rps_peer_ids[i]);
1302     }
1303   }
1304   return GNUNET_OK;
1305 }
1306
1307 /**
1308  * This is run after the test finished.
1309  *
1310  * Compute all perfect samples.
1311  */
1312 int
1313 profiler_eval (void)
1314 {
1315   /* Compute perfect sample for each sampler element */
1316   if (-1 == GNUNET_DISK_directory_scan ("/tmp/rps/", file_name_cb, NULL))
1317   {
1318     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Scan of directory failed\n");
1319   }
1320
1321   return evaluate ();
1322 }
1323
1324
1325 /***********************************************************************
1326  * /Definition of tests
1327 ***********************************************************************/
1328
1329
1330 /**
1331  * Actual "main" function for the testcase.
1332  *
1333  * @param cls closure
1334  * @param h the run handle
1335  * @param n_peers number of peers in 'peers'
1336  * @param peers handle to peers run in the testbed
1337  * @param links_succeeded the number of overlay link connection attempts that
1338  *          succeeded
1339  * @param links_failed the number of overlay link connection attempts that
1340  *          failed
1341  */
1342 static void
1343 run (void *cls,
1344      struct GNUNET_TESTBED_RunHandle *h,
1345      unsigned int n_peers,
1346      struct GNUNET_TESTBED_Peer **peers,
1347      unsigned int links_succeeded,
1348      unsigned int links_failed)
1349 {
1350   unsigned int i;
1351   struct OpListEntry *entry;
1352   uint32_t num_mal_peers;
1353
1354   testbed_peers = peers;
1355   num_peers_online = 0;
1356   for (i = 0; i < num_peers; i++)
1357   {
1358     entry = make_oplist_entry ();
1359     entry->index = i;
1360     rps_peers[i].index = i;
1361     if (NULL != cur_test_run.init_peer)
1362       cur_test_run.init_peer (&rps_peers[i]);
1363     entry->op = GNUNET_TESTBED_peer_get_information (peers[i],
1364                                                      GNUNET_TESTBED_PIT_IDENTITY,
1365                                                      &info_cb,
1366                                                      entry);
1367   }
1368
1369   num_mal_peers = round (portion * num_peers);
1370   GNUNET_assert (num_peers == n_peers);
1371   for (i = 0; i < n_peers; i++)
1372   {
1373     rps_peers[i].index = i;
1374     if ( (rps_peers[i].num_recv_ids < rps_peers[i].num_ids_to_request) ||
1375          (i < num_mal_peers) )
1376     {
1377       rps_peers[i].op =
1378         GNUNET_TESTBED_service_connect (&rps_peers[i],
1379                                         peers[i],
1380                                         "rps",
1381                                         &rps_connect_complete_cb,
1382                                         &rps_peers[i],
1383                                         &rps_connect_adapter,
1384                                         &rps_disconnect_adapter,
1385                                         &rps_peers[i]);
1386     }
1387   }
1388
1389   if (NULL != churn_task)
1390     GNUNET_SCHEDULER_cancel (churn_task);
1391   shutdown_task = GNUNET_SCHEDULER_add_delayed (timeout, &shutdown_op, NULL);
1392 }
1393
1394
1395 /**
1396  * Entry point for the testcase, sets up the testbed.
1397  *
1398  * @param argc unused
1399  * @param argv unused
1400  * @return 0 on success
1401  */
1402 int
1403 main (int argc, char *argv[])
1404 {
1405   int ret_value;
1406
1407   num_peers = 5;
1408   cur_test_run.name = "test-rps-default";
1409   cur_test_run.init_peer = default_init_peer;
1410   cur_test_run.pre_test = NULL;
1411   cur_test_run.reply_handle = default_reply_handle;
1412   cur_test_run.eval_cb = default_eval_cb;
1413   churn_task = NULL;
1414   timeout = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30);
1415
1416   if (strstr (argv[0], "malicious") != NULL)
1417   {
1418     cur_test_run.pre_test = mal_pre;
1419     cur_test_run.main_test = mal_cb;
1420     cur_test_run.init_peer = mal_init_peer;
1421
1422     if (strstr (argv[0], "_1") != NULL)
1423     {
1424       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Test malicious peer type 1\n");
1425       cur_test_run.name = "test-rps-malicious_1";
1426       mal_type = 1;
1427     }
1428     else if (strstr (argv[0], "_2") != NULL)
1429     {
1430       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Test malicious peer type 2\n");
1431       cur_test_run.name = "test-rps-malicious_2";
1432       mal_type = 2;
1433     }
1434     else if (strstr (argv[0], "_3") != NULL)
1435     {
1436       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Test malicious peer type 3\n");
1437       cur_test_run.name = "test-rps-malicious_3";
1438       mal_type = 3;
1439     }
1440   }
1441
1442   else if (strstr (argv[0], "_single_req") != NULL)
1443   {
1444     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Test single request\n");
1445     cur_test_run.name = "test-rps-single-req";
1446     cur_test_run.main_test = single_req_cb;
1447   }
1448
1449   else if (strstr (argv[0], "_delayed_reqs") != NULL)
1450   {
1451     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Test delayed requests\n");
1452     cur_test_run.name = "test-rps-delayed-reqs";
1453     cur_test_run.main_test = delay_req_cb;
1454   }
1455
1456   else if (strstr (argv[0], "_seed_big") != NULL)
1457   {
1458     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Test seeding (num_peers > GNUNET_SERVER_MAX_MESSAGE_SIZE)\n");
1459     num_peers = 1;
1460     cur_test_run.name = "test-rps-seed-big";
1461     cur_test_run.main_test = seed_big_cb;
1462     cur_test_run.eval_cb = no_eval;
1463     timeout = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 10);
1464   }
1465
1466   else if (strstr (argv[0], "_single_peer_seed") != NULL)
1467   {
1468     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Test seeding and requesting on a single peer\n");
1469     cur_test_run.name = "test-rps-single-peer-seed";
1470     cur_test_run.main_test = single_peer_seed_cb;
1471   }
1472
1473   else if (strstr (argv[0], "_seed_request") != NULL)
1474   {
1475     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Test seeding and requesting on multiple peers\n");
1476     cur_test_run.name = "test-rps-seed-request";
1477     cur_test_run.main_test = seed_req_cb;
1478   }
1479
1480   else if (strstr (argv[0], "_seed") != NULL)
1481   {
1482     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Test seeding\n");
1483     cur_test_run.name = "test-rps-seed";
1484     cur_test_run.main_test = seed_cb;
1485     cur_test_run.eval_cb = no_eval;
1486   }
1487
1488   else if (strstr (argv[0], "_req_cancel") != NULL)
1489   {
1490     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Test cancelling a request\n");
1491     cur_test_run.name = "test-rps-req-cancel";
1492     num_peers = 1;
1493     cur_test_run.main_test = req_cancel_cb;
1494     cur_test_run.eval_cb = no_eval;
1495     timeout = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 10);
1496   }
1497
1498   else if (strstr (argv[0], "profiler") != NULL)
1499   {
1500     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "This is the profiler\n");
1501     cur_test_run.name = "test-rps-profiler";
1502     num_peers = 10;
1503     mal_type = 3;
1504     cur_test_run.init_peer = profiler_init_peer;
1505     cur_test_run.pre_test = mal_pre;
1506     cur_test_run.main_test = profiler_cb;
1507     cur_test_run.reply_handle = profiler_reply_handle;
1508     cur_test_run.eval_cb = profiler_eval;
1509     cur_test_run.request_interval = 2;
1510     cur_test_run.num_requests = 5;
1511     timeout = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 90);
1512
1513     /* 'Clean' directory */
1514     (void) GNUNET_DISK_directory_remove ("/tmp/rps/");
1515     GNUNET_DISK_directory_create ("/tmp/rps/");
1516   }
1517
1518   rps_peers = GNUNET_new_array (num_peers, struct RPSPeer);
1519   peer_map = GNUNET_CONTAINER_multipeermap_create (num_peers, GNUNET_NO);
1520   rps_peer_ids = GNUNET_new_array (num_peers, struct GNUNET_PeerIdentity);
1521   if ( (2 == mal_type) ||
1522        (3 == mal_type))
1523     target_peer = &rps_peer_ids[num_peers - 2];
1524   if (profiler_eval == cur_test_run.eval_cb)
1525     eval_peer = &rps_peers[num_peers - 1];    /* FIXME: eval_peer could be a
1526                                                  malicious peer if not careful
1527                                                  with the malicious portion */
1528
1529   ok = 1;
1530   (void) GNUNET_TESTBED_test_run (cur_test_run.name,
1531                                   "test_rps.conf",
1532                                   num_peers,
1533                                   0, NULL, NULL,
1534                                   &run, NULL);
1535
1536   ret_value = cur_test_run.eval_cb();
1537   GNUNET_free (rps_peers );
1538   GNUNET_free (rps_peer_ids);
1539   GNUNET_CONTAINER_multipeermap_destroy (peer_map);
1540   return ret_value;
1541 }
1542
1543 /* end of test_rps_multipeer.c */