src/testbed/gnunet-service-testbed.c

   1 /*
   2   This file is part of GNUnet.
   3   (C) 2012 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 2, 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 /**
  22  * @file testbed/gnunet-service-testbed.c
  23  * @brief implementation of the TESTBED service
  24  * @author Sree Harsha Totakura
  25  */
  26
  27 #include "platform.h"
  28 #include "gnunet_service_lib.h"
  29 #include "gnunet_server_lib.h"
  30 #include <zlib.h>
  31
  32 #include "gnunet_testbed_service.h"
  33 #include "testbed.h"
  34 #include "testbed_api.h"
  35 #include "testbed_api_hosts.h"
  36 #include "gnunet_testing_lib-new.h"
  37
  38 /**
  39  * Generic logging
  40  */
  41 #define LOG(kind,...)                           \
  42   GNUNET_log (kind, __VA_ARGS__)
  43
  44 /**
  45  * Debug logging
  46  */
  47 #define LOG_DEBUG(...)                          \
  48   LOG (GNUNET_ERROR_TYPE_DEBUG, __VA_ARGS__)
  49
  50
  51 #define LIST_GROW_STEP 10
  52
  53 struct Context
  54 {
  55   /**
  56    * The client handle associated with this context
  57    */
  58   struct GNUNET_SERVER_Client *client;
  59
  60   /**
  61    * The network address of the master controller
  62    */
  63   char *master_ip;
  64
  65   /**
  66    * The TESTING system handle for starting peers locally
  67    */
  68   struct GNUNET_TESTING_System *system;
  69
  70   /**
  71    * Event mask of event to be responded in this context
  72    */
  73   uint64_t event_mask;
  74
  75   /**
  76    * Our host id according to this context
  77    */
  78   uint32_t host_id;
  79 };
  80
  81
  82 /**
  83  * The message queue for sending messages to clients
  84  */
  85 struct MessageQueue
  86 {
  87   /**
  88    * The message to be sent
  89    */
  90   struct GNUNET_MessageHeader *msg;
  91
  92   /**
  93    * The client to send the message to
  94    */
  95   struct GNUNET_SERVER_Client *client;
  96
  97   /**
  98    * next pointer for DLL
  99    */
 100   struct MessageQueue *next;
 101
 102   /**
 103    * prev pointer for DLL
 104    */
 105   struct MessageQueue *prev;
 106 };
 107
 108
 109 /**
 110  * The structure for identifying a shared service
 111  */
 112 struct SharedService
 113 {
 114   /**
 115    * The name of the shared service
 116    */
 117   char *name;
 118
 119   /**
 120    * Number of shared peers per instance of the shared service
 121    */
 122   uint32_t num_shared;
 123
 124   /**
 125    * Number of peers currently sharing the service
 126    */
 127   uint32_t num_sharing;
 128 };
 129
 130
 131 /**
 132  * A routing entry
 133  */
 134 struct Route
 135 {
 136   /**
 137    * destination host
 138    */
 139   uint32_t dest;
 140
 141   /**
 142    * The host destination is reachable thru
 143    */
 144   uint32_t thru;
 145 };
 146
 147
 148 /**
 149  * Structure representing a connected(directly-linked) controller
 150  */
 151 struct Slave
 152 {
 153   /**
 154    * The controller process handle if we had started the controller
 155    */
 156   struct GNUNET_TESTBED_ControllerProc *controller_proc;
 157
 158   /**
 159    * The controller handle
 160    */
 161   struct GNUNET_TESTBED_Controller *controller;
 162
 163   /**
 164    * The id of the host this controller is running on
 165    */
 166   uint32_t host_id;
 167 };
 168
 169
 170 /**
 171  * Slave startup context
 172  */
 173 struct SlaveContext
 174 {
 175   /**
 176    * The slave corresponding to this context
 177    */
 178   struct Slave *slave;
 179
 180   /**
 181    * The configuration used as a template while startup
 182    */
 183   struct GNUNET_CONFIGURATION_Handle *cfg;
 184 };
 185
 186
 187 /**
 188  * States of LCFContext
 189  */
 190 enum LCFContextState
 191   {
 192     /**
 193      * The Context has been initialized; Nothing has been done on it
 194      */
 195     INIT,
 196
 197     /**
 198      * Delegated host has been registered at the forwarding controller
 199      */
 200     DELEGATED_HOST_REGISTERED,
 201
 202     /**
 203      * The context has been finished (may have error)
 204      */
 205     FINISHED
 206
 207   };
 208
 209
 210 /**
 211  * Link controllers request forwarding context
 212  */
 213 struct LCFContext
 214 {
 215   /**
 216    * The serialized and compressed configuration
 217    */
 218   char *sxcfg;
 219
 220   /**
 221    * The gateway which will pass the link message to delegated host
 222    */
 223   struct Slave *gateway;
 224
 225   /**
 226    * The host registration handle while registered hosts in this context
 227    */
 228   struct GNUNET_TESTBED_HostRegistrationHandle *rhandle;
 229
 230   /**
 231    * The size of the compressed serialized configuration
 232    */
 233   size_t sxcfg_size;
 234
 235   /**
 236    * The size of the uncompressed configuration
 237    */
 238   size_t scfg_size;
 239
 240   /**
 241    * Should the delegated host be started by the slave host?
 242    */
 243   int is_subordinate;
 244
 245   /**
 246    * The state of this context
 247    */
 248   enum LCFContextState state;
 249
 250   /**
 251    * The delegated host
 252    */
 253   uint32_t delegated_host_id;
 254
 255
 256 };
 257
 258
 259 /**
 260  * Structure of a queue entry in LCFContext request queue
 261  */
 262 struct LCFContextQueue
 263 {
 264   /**
 265    * The LCFContext
 266    */
 267   struct LCFContext *lcf;
 268
 269   /**
 270    * Head prt for DLL
 271    */
 272   struct LCFContextQueue *next;
 273
 274   /**
 275    * Tail ptr for DLL
 276    */
 277   struct LCFContextQueue *prev;
 278 };
 279
 280
 281 /**
 282  * A locally started peer
 283  */
 284 struct Peer
 285 {
 286   /**
 287    * The peer handle from testing API
 288    */
 289   struct GNUNET_TESTING_Peer *peer;
 290
 291   /**
 292    * The modified (by GNUNET_TESTING_peer_configure) configuration this peer is
 293    * configured with
 294    */
 295   struct GNUNET_CONFIGURATION_Handle *cfg;
 296
 297   /**
 298    * Our local reference id for this peer
 299    */
 300   uint32_t id;
 301
 302 };
 303
 304
 305 /**
 306  * The master context; generated with the first INIT message
 307  */
 308 static struct Context *master_context;
 309
 310 /***********/
 311 /* Handles */
 312 /***********/
 313
 314 /**
 315  * Wrapped stdin.
 316  */
 317 static struct GNUNET_DISK_FileHandle *fh;
 318
 319 /**
 320  * Current Transmit Handle; NULL if no notify transmit exists currently
 321  */
 322 static struct GNUNET_SERVER_TransmitHandle *transmit_handle;
 323
 324 /****************/
 325 /* Lists & Maps */
 326 /****************/
 327
 328 /**
 329  * The head for the LCF queue
 330  */
 331 static struct LCFContextQueue *lcfq_head;
 332
 333 /**
 334  * The tail for the LCF queue
 335  */
 336 static struct LCFContextQueue *lcfq_tail;
 337
 338 /**
 339  * The message queue head
 340  */
 341 static struct MessageQueue *mq_head;
 342
 343 /**
 344  * The message queue tail
 345  */
 346 static struct MessageQueue *mq_tail;
 347
 348 /**
 349  * Array of host list
 350  */
 351 static struct GNUNET_TESTBED_Host **host_list;
 352
 353 /**
 354  * A list of routes
 355  */
 356 static struct Route **route_list;
 357
 358 /**
 359  * A list of directly linked neighbours
 360  */
 361 static struct Slave **slave_list;
 362
 363 /**
 364  * A list of peers we own locally
 365  */
 366 static struct Peer **peer_list;
 367
 368 /**
 369  * The hashmap of shared services
 370  */
 371 static struct GNUNET_CONTAINER_MultiHashMap *ss_map;
 372
 373 /**
 374  * The size of the host list
 375  */
 376 static uint32_t host_list_size;
 377
 378 /**
 379  * The size of the route list
 380  */
 381 static uint32_t route_list_size;
 382
 383 /**
 384  * The size of directly linked neighbours list
 385  */
 386 static uint32_t slave_list_size;
 387
 388 /**
 389  * The size of the peer list
 390  */
 391 static uint32_t peer_list_size;
 392
 393 /*********/
 394 /* Tasks */
 395 /*********/
 396
 397 /**
 398  * The lcf_task handle
 399  */
 400 static GNUNET_SCHEDULER_TaskIdentifier lcf_proc_task_id;
 401
 402 /**
 403  * The shutdown task handle
 404  */
 405 static GNUNET_SCHEDULER_TaskIdentifier shutdown_task_id;
 406
 407 /******************/
 408 /* Testing System */
 409 /******************/
 410
 411 /**
 412  * Our configuration; we also use this as template for starting other controllers
 413  */
 414 static struct GNUNET_CONFIGURATION_Handle *config;
 415
 416
 417 /**
 418  * Function called to notify a client about the connection begin ready to queue
 419  * more data.  "buf" will be NULL and "size" zero if the connection was closed
 420  * for writing in the meantime.
 421  *
 422  * @param cls NULL
 423  * @param size number of bytes available in buf
 424  * @param buf where the callee should write the message
 425  * @return number of bytes written to buf
 426  */
 427 static size_t
 428 transmit_ready_notify (void *cls, size_t size, void *buf)
 429 {
 430   struct MessageQueue *mq_entry;
 431
 432   transmit_handle = NULL;
 433   mq_entry = mq_head;
 434   GNUNET_assert (NULL != mq_entry);
 435   if (0 == size)
 436     return 0;
 437   GNUNET_assert (ntohs (mq_entry->msg->size) <= size);
 438   size = ntohs (mq_entry->msg->size);
 439   memcpy (buf, mq_entry->msg, size);
 440   GNUNET_free (mq_entry->msg);
 441   GNUNET_CONTAINER_DLL_remove (mq_head, mq_tail, mq_entry);
 442   GNUNET_free (mq_entry);
 443   mq_entry = mq_head;
 444   if (NULL != mq_entry)
 445     transmit_handle =
 446       GNUNET_SERVER_notify_transmit_ready (mq_entry->client,
 447                                            ntohs (mq_entry->msg->size),
 448                                            GNUNET_TIME_UNIT_FOREVER_REL,
 449                                            &transmit_ready_notify, NULL);
 450   return size;
 451 }
 452
 453
 454 /**
 455  * Queues a message in send queue for sending to the service
 456  *
 457  * @param client the client to whom the queued message has to be sent
 458  * @param msg the message to queue
 459  */
 460 static void
 461 queue_message (struct GNUNET_SERVER_Client *client,
 462                struct GNUNET_MessageHeader *msg)
 463 {
 464   struct MessageQueue *mq_entry;
 465   uint16_t type;
 466   uint16_t size;
 467
 468   type = ntohs (msg->type);
 469   size = ntohs (msg->size);
 470   GNUNET_assert ((GNUNET_MESSAGE_TYPE_TESTBED_INIT <= type) &&
 471                  (GNUNET_MESSAGE_TYPE_TESTBED_MAX > type));
 472   mq_entry = GNUNET_malloc (sizeof (struct MessageQueue));
 473   mq_entry->msg = msg;
 474   mq_entry->client = client;
 475   LOG_DEBUG ( "Queueing message of type %u, size %u for sending\n", type,
 476               ntohs (msg->size));
 477   GNUNET_CONTAINER_DLL_insert_tail (mq_head, mq_tail, mq_entry);
 478   if (NULL == transmit_handle)
 479     transmit_handle =
 480       GNUNET_SERVER_notify_transmit_ready (client, size,
 481                                            GNUNET_TIME_UNIT_FOREVER_REL,
 482                                            &transmit_ready_notify, NULL);
 483 }
 484
 485
 486 /**
 487  * Similar to GNUNET_realloc; however clears tail part of newly allocated memory
 488  *
 489  * @param ptr the memory block to realloc
 490  * @param size the size of ptr
 491  * @param new_size the size to which ptr has to be realloc'ed
 492  * @return the newly reallocated memory block
 493  */
 494 static void *
 495 TESTBED_realloc (void *ptr, size_t size, size_t new_size)
 496 {
 497   ptr = GNUNET_realloc (ptr, new_size);
 498   if (new_size > size)
 499     ptr = memset (ptr + size, 0, new_size - size);
 500   return ptr;
 501 }
 502
 503
 504 /**
 505  * Function to add a host to the current list of known hosts
 506  *
 507  * @param host the host to add
 508  * @return GNUNET_OK on success; GNUNET_SYSERR on failure due to host-id
 509  *           already in use
 510  */
 511 static int
 512 host_list_add (struct GNUNET_TESTBED_Host *host)
 513 {
 514   uint32_t host_id;
 515
 516   host_id = GNUNET_TESTBED_host_get_id_ (host);
 517   if (host_list_size <= host_id)
 518   {
 519     host_list =
 520       TESTBED_realloc (host_list,
 521                        sizeof (struct GNUNET_TESTBED_Host *) * host_list_size,
 522                        sizeof (struct GNUNET_TESTBED_Host *) *
 523                        (host_list_size + LIST_GROW_STEP));
 524     host_list_size += LIST_GROW_STEP;
 525   }
 526   if (NULL != host_list[host_id])
 527   {
 528     LOG_DEBUG ("A host with id: %u already exists\n", host_id);
 529     return GNUNET_SYSERR;
 530   }
 531   host_list[host_id] = host;
 532   return GNUNET_OK;
 533 }
 534
 535
 536 /**
 537  * Adds a route to the route list
 538  *
 539  * @param route the route to add
 540  */
 541 static void
 542 route_list_add (struct Route *route)
 543 {
 544   if (route->dest >= route_list_size)
 545   {
 546     route_list =
 547       TESTBED_realloc (route_list,
 548                        sizeof (struct Route *) * route_list_size,
 549                        sizeof (struct Route *) *
 550                        (route_list_size + LIST_GROW_STEP));
 551     route_list_size += LIST_GROW_STEP;
 552   }
 553   GNUNET_assert (NULL == route_list[route->dest]);
 554   route_list[route->dest] = route;
 555 }
 556
 557
 558 /**
 559  * Adds a slave to the slave array
 560  *
 561  * @param slave the slave controller to add
 562  */
 563 static void
 564 slave_list_add (struct Slave *slave)
 565 {
 566   if (slave->host_id  >= slave_list_size)
 567   {
 568     slave_list = TESTBED_realloc (slave_list,
 569                                   sizeof (struct Slave *) *slave_list_size,
 570                                   sizeof (struct Slave *) *
 571                                   (slave_list_size + LIST_GROW_STEP));
 572     slave_list_size += LIST_GROW_STEP;
 573   }
 574   GNUNET_assert (NULL == slave_list[slave->host_id]);
 575   slave_list[slave->host_id] = slave;
 576 }
 577
 578
 579 /**
 580  * Adds a peer to the peer array
 581  *
 582  * @param peer the peer to add
 583  */
 584 static void
 585 peer_list_add (struct Peer *peer)
 586 {
 587   if (peer->id  >= peer_list_size)
 588   {
 589     peer_list = TESTBED_realloc (peer_list,
 590                                  sizeof (struct Peer *) * peer_list_size,
 591                                  sizeof (struct Peer *) *
 592                                  (peer_list_size + LIST_GROW_STEP));
 593     peer_list_size += LIST_GROW_STEP;
 594   }
 595   GNUNET_assert (NULL == peer_list[peer->id]);
 596   peer_list[peer->id] = peer;
 597 }
 598
 599
 600 /**
 601  * Routes message to a host given its host_id
 602  *
 603  * @param host_id the id of the destination host
 604  * @param msg the message to be routed
 605  */
 606 static void
 607 route_message (uint32_t host_id, const struct GNUNET_MessageHeader *msg)
 608 {
 609   GNUNET_break (0);
 610 }
 611
 612
 613 /**
 614  * The  Link Controller forwarding task
 615  *
 616  * @param cls the LCFContext
 617  * @param tc the Task context from scheduler
 618  */
 619 static void
 620 lcf_proc_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
 621
 622
 623 /**
 624  * Completion callback for host registrations while forwarding Link Controller messages
 625  *
 626  * @param cls the LCFContext
 627  * @param emsg the error message; NULL if host registration is successful
 628  */
 629 static void
 630 lcf_proc_cc (void *cls, const char *emsg)
 631 {
 632   struct LCFContext *lcf = cls;
 633
 634   lcf->rhandle = NULL;
 635   GNUNET_assert (GNUNET_SCHEDULER_NO_TASK == lcf_proc_task_id);
 636   switch (lcf->state)
 637   {
 638   case INIT:
 639     if (NULL != emsg)
 640     {
 641       LOG (GNUNET_ERROR_TYPE_WARNING,
 642            "Host registration failed with message: %s\n", emsg);
 643       lcf->state = FINISHED;
 644       lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcf);
 645       return;
 646     }
 647     lcf->state = DELEGATED_HOST_REGISTERED;
 648     lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcf);
 649     break;
 650   default:
 651     GNUNET_assert (0);          /* Shouldn't reach here */
 652   }
 653 }
 654
 655
 656 /**
 657  * The  Link Controller forwarding task
 658  *
 659  * @param cls the LCFContext
 660  * @param tc the Task context from scheduler
 661  */
 662 static void
 663 lcf_proc_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 664 {
 665   struct LCFContext *lcf = cls;
 666   struct LCFContextQueue *lcfq;
 667
 668   lcf_proc_task_id = GNUNET_SCHEDULER_NO_TASK;
 669   switch (lcf->state)
 670   {
 671   case INIT:
 672     if (GNUNET_NO ==
 673         GNUNET_TESTBED_is_host_registered_ (host_list[lcf->delegated_host_id],
 674                                             lcf->gateway->controller))
 675     {
 676       lcf->rhandle =
 677         GNUNET_TESTBED_register_host (lcf->gateway->controller,
 678                                       host_list[lcf->delegated_host_id],
 679                                       lcf_proc_cc, lcf);
 680     }
 681     else
 682     {
 683       lcf->state = DELEGATED_HOST_REGISTERED;
 684       lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcf);
 685     }
 686     break;
 687   case DELEGATED_HOST_REGISTERED:
 688     GNUNET_TESTBED_controller_link_2 (lcf->gateway->controller,
 689                                       host_list[lcf->delegated_host_id],
 690                                       host_list[lcf->gateway->host_id],
 691                                       lcf->sxcfg, lcf->sxcfg_size,
 692                                       lcf->scfg_size,
 693                                       lcf->is_subordinate);
 694     lcf->state = FINISHED;
 695   case FINISHED:
 696     lcfq = lcfq_head;
 697     GNUNET_assert (lcfq->lcf == lcf);
 698     GNUNET_free (lcf->sxcfg);
 699     GNUNET_free (lcf);
 700     GNUNET_CONTAINER_DLL_remove (lcfq_head, lcfq_tail, lcfq);
 701     GNUNET_free (lcfq);
 702     if (NULL != lcfq_head)
 703       lcf_proc_task_id =
 704         GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcfq_head->lcf);
 705   }
 706 }
 707
 708
 709 /**
 710  * Callback for event from slave controllers
 711  *
 712  * @param cls struct Slave *
 713  * @param event information about the event
 714  */
 715 static void
 716 slave_event_callback(void *cls,
 717                      const struct GNUNET_TESTBED_EventInformation *event)
 718 {
 719   GNUNET_break (0);
 720 }
 721
 722
 723 /**
 724  * Callback to signal successfull startup of the controller process
 725  *
 726  * @param cls the closure from GNUNET_TESTBED_controller_start()
 727  * @param cfg the configuration with which the controller has been started;
 728  *          NULL if status is not GNUNET_OK
 729  * @param status GNUNET_OK if the startup is successfull; GNUNET_SYSERR if not,
 730  *          GNUNET_TESTBED_controller_stop() shouldn't be called in this case
 731  */
 732 static void
 733 slave_status_callback (void *cls,
 734                        const struct GNUNET_CONFIGURATION_Handle *cfg,
 735                        int status)
 736 {
 737   struct SlaveContext *sc = cls;
 738
 739   if (GNUNET_SYSERR == status)
 740   {
 741     sc->slave->controller_proc = NULL;
 742     LOG (GNUNET_ERROR_TYPE_WARNING,
 743          "Unexpected slave shutdown\n");
 744     GNUNET_SCHEDULER_shutdown ();       /* We too shutdown */
 745     return;
 746   }
 747   GNUNET_CONFIGURATION_destroy (sc->cfg);
 748   sc->slave->controller =
 749     GNUNET_TESTBED_controller_connect (cfg, host_list[sc->slave->host_id],
 750                                        master_context->event_mask,
 751                                        &slave_event_callback, sc->slave);
 752   GNUNET_free (sc);
 753 }
 754
 755
 756 /**
 757  * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_INIT messages
 758  *
 759  * @param cls NULL
 760  * @param client identification of the client
 761  * @param message the actual message
 762  */
 763 static void
 764 handle_init (void *cls,
 765              struct GNUNET_SERVER_Client *client,
 766              const struct GNUNET_MessageHeader *message)
 767 {
 768   const struct GNUNET_TESTBED_InitMessage *msg;
 769   struct GNUNET_TESTBED_Host *host;
 770   void *addr;
 771   size_t addrlen;
 772
 773   if (NULL != master_context)
 774   {
 775     GNUNET_break (0);
 776     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 777     return;
 778   }
 779   msg = (const struct GNUNET_TESTBED_InitMessage *) message;
 780   master_context = GNUNET_malloc (sizeof (struct Context));
 781   master_context->client = client;
 782   master_context->host_id = ntohl (msg->host_id);
 783   GNUNET_assert (GNUNET_OK ==
 784                  GNUNET_SERVER_client_get_address (client, &addr, &addrlen));
 785   master_context->master_ip = GNUNET_malloc (NI_MAXHOST);
 786   if (0 != getnameinfo (addr, addrlen, master_context->master_ip, NI_MAXHOST,
 787                         NULL, 0, NI_NUMERICHOST))
 788   {
 789     LOG (GNUNET_ERROR_TYPE_WARNING,
 790          "Cannot determine the ip of master controller: %s\n", STRERROR (errno));
 791     GNUNET_free (addr);
 792     GNUNET_assert (0);
 793   }
 794   GNUNET_free (addr);
 795   LOG_DEBUG ("Master Controller IP: %s\n", master_context->master_ip);
 796   master_context->system =
 797     GNUNET_TESTING_system_create ("testbed", master_context->master_ip);
 798   host = GNUNET_TESTBED_host_create_with_id (master_context->host_id,
 799                                              NULL, NULL, 0);
 800   host_list_add (host);
 801   master_context->event_mask = GNUNET_ntohll (msg->event_mask);
 802   GNUNET_SERVER_client_keep (client);
 803   LOG_DEBUG ("Created master context with host ID: %u\n",
 804              master_context->host_id);
 805   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 806 }
 807
 808
 809 /**
 810  * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_ADDHOST messages
 811  *
 812  * @param cls NULL
 813  * @param client identification of the client
 814  * @param message the actual message
 815  */
 816 static void
 817 handle_add_host (void *cls,
 818                  struct GNUNET_SERVER_Client *client,
 819                  const struct GNUNET_MessageHeader *message)
 820 {
 821   struct GNUNET_TESTBED_Host *host;
 822   const struct GNUNET_TESTBED_AddHostMessage *msg;
 823   struct GNUNET_TESTBED_HostConfirmedMessage *reply;
 824   char *username;
 825   char *hostname;
 826   char *emsg;
 827   uint32_t host_id;
 828   uint16_t username_length;
 829   uint16_t hostname_length;
 830   uint16_t reply_size;
 831
 832   msg = (const struct GNUNET_TESTBED_AddHostMessage *) message;
 833   username_length = ntohs (msg->user_name_length);
 834   username_length = (0 == username_length) ? 0 : username_length + 1;
 835   username = (char *) &(msg[1]);
 836   hostname = username + username_length;
 837   if (ntohs (message->size) <=
 838       (sizeof (struct GNUNET_TESTBED_AddHostMessage) + username_length))
 839   {
 840     GNUNET_break (0);
 841     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 842     return;
 843   }
 844   hostname_length = ntohs (message->size)
 845     - (sizeof (struct GNUNET_TESTBED_AddHostMessage) + username_length);
 846   if (strlen (hostname) != hostname_length - 1)
 847   {
 848     GNUNET_break (0);
 849     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 850     return;
 851   }
 852   host_id = ntohl (msg->host_id);
 853   LOG_DEBUG ("Received ADDHOST message\n");
 854   LOG_DEBUG ("-------host id: %u\n", host_id);
 855   if (NULL != hostname) LOG_DEBUG ("-------hostname: %s\n", hostname);
 856   if (0 != username_length) LOG_DEBUG ("-------username: %s\n", username);
 857   else LOG_DEBUG ("-------username: NULL\n");
 858   LOG_DEBUG ("-------ssh port: %u\n", ntohs (msg->ssh_port));
 859   host = GNUNET_TESTBED_host_create_with_id (host_id, hostname, username,
 860                                              ntohs (msg->ssh_port));
 861   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 862   reply_size = sizeof (struct GNUNET_TESTBED_HostConfirmedMessage);
 863   if (GNUNET_OK != host_list_add (host))
 864   {
 865     /* We are unable to add a host */
 866     emsg = "A host exists with given host-id";
 867     LOG_DEBUG ("%s: %u", emsg, host_id);
 868     GNUNET_TESTBED_host_destroy (host);
 869     reply_size += strlen (emsg) + 1;
 870     reply = GNUNET_malloc (reply_size);
 871     memcpy (&reply[1], emsg, strlen (emsg) + 1);
 872   }
 873   else
 874     reply = GNUNET_malloc (reply_size);
 875   reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_ADDHOSTCONFIRM);
 876   reply->header.size = htons (reply_size);
 877   reply->host_id = htonl (host_id);
 878   queue_message (client, (struct GNUNET_MessageHeader *) reply);
 879 }
 880
 881
 882 /**
 883  * Iterator over hash map entries.
 884  *
 885  * @param cls closure
 886  * @param key current key code
 887  * @param value value in the hash map
 888  * @return GNUNET_YES if we should continue to
 889  *         iterate,
 890  *         GNUNET_NO if not.
 891  */
 892 int ss_exists_iterator (void *cls,
 893                         const struct GNUNET_HashCode * key,
 894                         void *value)
 895 {
 896   struct SharedService *queried_ss = cls;
 897   struct SharedService *ss = value;
 898
 899   if (0 == strcmp (ss->name, queried_ss->name))
 900     return GNUNET_NO;
 901   else
 902     return GNUNET_YES;
 903 }
 904
 905
 906 /**
 907  * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_ADDHOST messages
 908  *
 909  * @param cls NULL
 910  * @param client identification of the client
 911  * @param message the actual message
 912  */
 913 static void
 914 handle_configure_shared_service (void *cls,
 915                                  struct GNUNET_SERVER_Client *client,
 916                                  const struct GNUNET_MessageHeader *message)
 917 {
 918   const struct GNUNET_TESTBED_ConfigureSharedServiceMessage *msg;
 919   struct SharedService *ss;
 920   char *service_name;
 921   struct GNUNET_HashCode hash;
 922   uint16_t msg_size;
 923   uint16_t service_name_size;
 924
 925   msg = (const struct GNUNET_TESTBED_ConfigureSharedServiceMessage *) message;
 926   msg_size = ntohs (message->size);
 927   if (msg_size <= sizeof (struct GNUNET_TESTBED_ConfigureSharedServiceMessage))
 928   {
 929     GNUNET_break (0);
 930     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 931     return;
 932   }
 933   service_name_size = msg_size -
 934     sizeof (struct GNUNET_TESTBED_ConfigureSharedServiceMessage);
 935   service_name = (char *) &msg[1];
 936   if ('\0' != service_name[service_name_size - 1])
 937   {
 938     GNUNET_break (0);
 939     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 940     return;
 941   }
 942   LOG_DEBUG ("Received service sharing request for %s, with %d peers\n",
 943              service_name, ntohl (msg->num_peers));
 944   if (ntohl (msg->host_id) != master_context->host_id)
 945   {
 946     route_message (ntohl (msg->host_id), message);
 947     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 948     return;
 949   }
 950   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 951   ss = GNUNET_malloc (sizeof (struct SharedService));
 952   ss->name = strdup (service_name);
 953   ss->num_shared = ntohl (msg->num_peers);
 954   GNUNET_CRYPTO_hash (ss->name, service_name_size, &hash);
 955   if (GNUNET_SYSERR ==
 956       GNUNET_CONTAINER_multihashmap_get_multiple (ss_map, &hash,
 957                                                   &ss_exists_iterator, ss))
 958   {
 959     LOG (GNUNET_ERROR_TYPE_WARNING,
 960          "Service %s already configured as a shared service. "
 961          "Ignoring service sharing request \n", ss->name);
 962     GNUNET_free (ss->name);
 963     GNUNET_free (ss);
 964     return;
 965   }
 966   GNUNET_CONTAINER_multihashmap_put (ss_map, &hash, ss,
 967                                      GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
 968 }
 969
 970
 971 /**
 972  * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_LCONTROLLERS message
 973  *
 974  * @param cls NULL
 975  * @param client identification of the client
 976  * @param message the actual message
 977  */
 978 static void
 979 handle_link_controllers (void *cls,
 980                          struct GNUNET_SERVER_Client *client,
 981                          const struct GNUNET_MessageHeader *message)
 982 {
 983   const struct GNUNET_TESTBED_ControllerLinkMessage *msg;
 984   struct GNUNET_CONFIGURATION_Handle *cfg;
 985   struct SlaveContext *sc;
 986   struct LCFContextQueue *lcfq;
 987   struct Route *route;
 988   struct Route *new_route;
 989   char *config;
 990   uLongf dest_size;
 991   size_t config_size;
 992   uint32_t delegated_host_id;
 993   uint32_t slave_host_id;
 994   uint16_t msize;
 995
 996   if (NULL == master_context)
 997   {
 998     GNUNET_break (0);
 999     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1000     return;
1001   }
1002   msize = ntohs (message->size);
1003   if (sizeof (struct GNUNET_TESTBED_ControllerLinkMessage) >= msize)
1004   {
1005     GNUNET_break (0);
1006     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1007     return;
1008   }
1009   msg = (const struct GNUNET_TESTBED_ControllerLinkMessage *) message;
1010   delegated_host_id = ntohl (msg->delegated_host_id);
1011   if (delegated_host_id == master_context->host_id)
1012   {
1013     GNUNET_break (0);
1014     LOG (GNUNET_ERROR_TYPE_WARNING, "Trying to link ourselves\n");
1015     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1016     return;
1017   }
1018   if ((delegated_host_id >= host_list_size) ||
1019       (NULL == host_list[delegated_host_id]))
1020   {
1021     LOG (GNUNET_ERROR_TYPE_WARNING, "Delegated host not registered with us\n");
1022     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1023     return;
1024   }
1025   slave_host_id = ntohl (msg->slave_host_id);
1026   if ((slave_host_id >= host_list_size) || (NULL == host_list[slave_host_id]))
1027   {
1028     LOG (GNUNET_ERROR_TYPE_WARNING, "Slave host not registered with us\n");
1029     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1030     return;
1031   }
1032   if (slave_host_id == delegated_host_id)
1033   {
1034     LOG (GNUNET_ERROR_TYPE_WARNING, "Slave and delegated host are same\n");
1035     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1036     return;
1037   }
1038   msize -= sizeof (struct GNUNET_TESTBED_ControllerLinkMessage);
1039   config_size = ntohs (msg->config_size);
1040
1041   if (slave_host_id == master_context->host_id) /* Link from us */
1042   {
1043     struct Slave *slave;
1044
1045     if ((delegated_host_id < slave_list_size) &&
1046         (NULL != slave_list[delegated_host_id])) /* We have already added */
1047     {
1048       LOG (GNUNET_ERROR_TYPE_WARNING, "Host %u already connected\n",
1049            delegated_host_id);
1050       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1051       return;
1052     }
1053     config = GNUNET_malloc (config_size);
1054     dest_size = (uLongf) config_size;
1055     if (Z_OK != uncompress ((Bytef *) config, &dest_size,
1056                             (const Bytef *) &msg[1], (uLong) msize))
1057     {
1058       GNUNET_break (0);           /* Compression error */
1059       GNUNET_free (config);
1060       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1061       return;
1062     }
1063     if (config_size == dest_size)
1064     {
1065       LOG (GNUNET_ERROR_TYPE_WARNING, "Uncompressed config size mismatch\n");
1066       GNUNET_free (config);
1067       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1068     }
1069     cfg = GNUNET_CONFIGURATION_create (); /* Free here or in lcfcontext */
1070     if (GNUNET_OK != GNUNET_CONFIGURATION_deserialize (cfg, config, config_size,
1071                                                        GNUNET_NO))
1072     {
1073       GNUNET_break (0);           /* Configuration parsing error */
1074       GNUNET_free (config);
1075       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1076       return;
1077     }
1078     GNUNET_free (config);
1079     if ((delegated_host_id < slave_list_size) &&
1080         (NULL != slave_list[delegated_host_id]))
1081     {
1082       GNUNET_break (0);           /* Configuration parsing error */
1083       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1084       return;
1085     }
1086     slave = GNUNET_malloc (sizeof (struct Slave));
1087     slave->host_id = delegated_host_id;
1088     slave_list_add (slave);
1089     sc = GNUNET_malloc (sizeof (struct SlaveContext));
1090     sc->slave = slave;
1091     sc->cfg = cfg;
1092     if (1 == msg->is_subordinate)
1093     {
1094       slave->controller_proc =
1095         GNUNET_TESTBED_controller_start (master_context->master_ip,
1096                                          host_list[slave->host_id],
1097                                          cfg, &slave_status_callback,
1098                                          sc);
1099     }
1100     new_route = GNUNET_malloc (sizeof (struct Route));
1101     new_route->dest = delegated_host_id;
1102     new_route->thru = master_context->host_id;
1103     route_list_add (new_route);
1104     GNUNET_SERVER_receive_done (client, GNUNET_OK);
1105     return;
1106   }
1107
1108   /* Route the request */
1109   if (slave_host_id >= route_list_size)
1110   {
1111     LOG (GNUNET_ERROR_TYPE_WARNING, "No route towards slave host");
1112     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1113     return;
1114   }
1115   while (NULL != (route = route_list[slave_host_id]))
1116   {
1117     if (route->thru == master_context->host_id)
1118       break;
1119     slave_host_id = route->thru;
1120   }
1121   GNUNET_assert (NULL != route); /* because we add routes carefully */
1122   GNUNET_assert (route->dest < slave_list_size);
1123   GNUNET_assert (NULL != slave_list[route->dest]);
1124   lcfq = GNUNET_malloc (sizeof (struct LCFContextQueue));
1125   lcfq->lcf = GNUNET_malloc (sizeof (struct LCFContext));
1126   lcfq->lcf->delegated_host_id = delegated_host_id;
1127   lcfq->lcf->is_subordinate =
1128     (1 == msg->is_subordinate) ? GNUNET_YES : GNUNET_NO;
1129   lcfq->lcf->state = INIT;
1130   lcfq->lcf->gateway = slave_list[route->dest];
1131   lcfq->lcf->sxcfg_size = msize;
1132   lcfq->lcf->sxcfg = GNUNET_malloc (msize);
1133   lcfq->lcf->scfg_size = config_size;
1134   (void) memcpy (lcfq->lcf->sxcfg, &msg[1], msize);
1135   if (NULL == lcfq_head)
1136   {
1137     GNUNET_assert (GNUNET_SCHEDULER_NO_TASK == lcf_proc_task_id);
1138     GNUNET_CONTAINER_DLL_insert_tail (lcfq_head, lcfq_tail, lcfq);
1139     lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcfq);
1140   }
1141   else
1142     GNUNET_CONTAINER_DLL_insert_tail (lcfq_head, lcfq_tail, lcfq);
1143   GNUNET_SERVER_receive_done (client, GNUNET_OK);
1144   new_route = GNUNET_malloc (sizeof (struct Route));
1145   new_route->dest = delegated_host_id;
1146   new_route->thru = route->dest;
1147   route_list_add (new_route);
1148 }
1149
1150
1151 /**
1152  * Handler for GNUNET_MESSAGE_TYPE_TESTBED_CREATEPEER messages
1153  *
1154  * @param cls NULL
1155  * @param client identification of the client
1156  * @param message the actual message
1157  */
1158 static void
1159 handle_peer_create (void *cls,
1160                     struct GNUNET_SERVER_Client *client,
1161                     const struct GNUNET_MessageHeader *message)
1162 {
1163   const struct GNUNET_TESTBED_PeerCreateMessage *msg;
1164   struct GNUNET_TESTBED_PeerCreateSuccessEventMessage *reply;
1165   struct GNUNET_CONFIGURATION_Handle *cfg;
1166   char *config;
1167   size_t dest_size;
1168   int ret;
1169   uint32_t config_size;
1170   uint16_t msize;
1171
1172
1173   msize = ntohs (message->size);
1174   if (msize <= sizeof (struct GNUNET_TESTBED_PeerCreateMessage))
1175   {
1176     GNUNET_break (0);           /* We need configuration */
1177     GNUNET_SERVER_receive_done (client, GNUNET_OK);
1178     return;
1179   }
1180   msg = (const struct GNUNET_TESTBED_PeerCreateMessage *) message;
1181   if (ntohl (msg->host_id) == master_context->host_id)
1182   {
1183     struct Peer *peer;
1184     char *emsg;
1185
1186     /* We are responsible for this peer */
1187     msize -= sizeof (struct GNUNET_TESTBED_PeerCreateMessage);
1188     config_size = ntohl (msg->config_size);
1189     config = GNUNET_malloc (config_size);
1190     dest_size = config_size;
1191     if (Z_OK != (ret = uncompress ((Bytef *) config, (uLongf *) &dest_size,
1192                                    (const Bytef *) &msg[1], (uLong) msize)))
1193     {
1194       GNUNET_break (0);           /* uncompression error */
1195       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1196       return;
1197     }
1198     if (config_size != dest_size)
1199     {
1200       GNUNET_break (0);/* Uncompressed config size mismatch */
1201       GNUNET_free (config);
1202       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1203       return;
1204     }
1205     cfg = GNUNET_CONFIGURATION_create ();
1206     if (GNUNET_OK != GNUNET_CONFIGURATION_deserialize (cfg, config, config_size,
1207                                                        GNUNET_NO))
1208     {
1209       GNUNET_break (0);           /* Configuration parsing error */
1210       GNUNET_free (config);
1211       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1212       return;
1213     }
1214     GNUNET_free (config);
1215     peer = GNUNET_malloc (sizeof (struct Peer));
1216     peer->cfg = cfg;
1217     peer->id = ntohl (msg->peer_id);
1218     LOG_DEBUG ("Creating peer with id: %u\n", peer->id);
1219     peer->peer = GNUNET_TESTING_peer_configure (master_context->system, peer->cfg,
1220                                                 peer->id,
1221                                                 NULL /* Peer id */,
1222                                                 &emsg);
1223     if (NULL == peer->peer)
1224     {
1225       LOG (GNUNET_ERROR_TYPE_WARNING, "Configuring peer failed: %s\n", emsg);
1226       GNUNET_free (emsg);
1227       GNUNET_break (0);
1228       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1229       return;
1230     }
1231     peer_list_add (peer);
1232     reply = GNUNET_malloc (sizeof (struct GNUNET_TESTBED_PeerCreateSuccessEventMessage));
1233     reply->header.size = htons (sizeof (struct GNUNET_TESTBED_PeerCreateSuccessEventMessage));
1234     reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_PEERCREATESUCCESS);
1235     reply->peer_id = msg->peer_id;
1236     reply->operation_id = msg->operation_id;
1237     queue_message (client, &reply->header);
1238     GNUNET_SERVER_receive_done (client, GNUNET_OK);
1239     return;
1240   }
1241
1242   /* FIXME: Forward the peer to other host */
1243   GNUNET_break (0);
1244   GNUNET_SERVER_receive_done (client, GNUNET_OK);
1245 }
1246
1247
1248 /**
1249  * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_DESTROYPEER messages
1250  *
1251  * @param cls NULL
1252  * @param client identification of the client
1253  * @param message the actual message
1254  */
1255 static void
1256 handle_peer_destroy (void *cls,
1257                      struct GNUNET_SERVER_Client *client,
1258                      const struct GNUNET_MessageHeader *message)
1259 {
1260   const struct GNUNET_TESTBED_PeerDestroyMessage *msg;
1261   struct GNUNET_TESTBED_GenericOperationSuccessEventMessage *reply;
1262   uint32_t peer_id;
1263   uint32_t id;
1264   uint16_t reply_size;
1265
1266   msg = (const struct GNUNET_TESTBED_PeerDestroyMessage *) message;
1267   peer_id = ntohl (msg->peer_id);
1268   LOG_DEBUG ("Received peer destory on peer: %u and operation id: %ul\n",
1269              peer_id, GNUNET_ntohll (msg->operation_id));
1270   if ((peer_list_size <= peer_id) || (NULL == peer_list[peer_id]))
1271   {
1272     GNUNET_break (0);
1273     /* FIXME: Reply with failure event message or forward to slave controller */
1274     GNUNET_SERVER_receive_done (client, GNUNET_OK);
1275     return;
1276   }
1277   GNUNET_TESTING_peer_destroy (peer_list[peer_id]->peer);
1278   GNUNET_CONFIGURATION_destroy (peer_list[peer_id]->cfg);
1279   GNUNET_free (peer_list[peer_id]);
1280   peer_list[peer_id] = NULL;
1281   for (id = 0; id < LIST_GROW_STEP; id++)
1282   {
1283     if (((peer_id + id >= peer_list_size) ||
1284          (NULL != peer_list[peer_id])))
1285       break;
1286   }
1287   if (LIST_GROW_STEP == id)
1288   {
1289     peer_list_size -= LIST_GROW_STEP;
1290     peer_list = GNUNET_realloc (peer_list, peer_list_size);
1291   }
1292   reply_size =
1293     sizeof (struct GNUNET_TESTBED_GenericOperationSuccessEventMessage);
1294   reply = GNUNET_malloc (reply_size);
1295   reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_GENERICOPSUCCESS);
1296   reply->header.size = htons (reply_size);
1297   reply->operation_id = msg->operation_id;
1298   reply->event_type = htonl (GNUNET_TESTBED_ET_OPERATION_FINISHED);
1299   queue_message (client, &reply->header);
1300   GNUNET_SERVER_receive_done (client, GNUNET_OK);
1301 }
1302
1303
1304 /**
1305  * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_DESTROYPEER messages
1306  *
1307  * @param cls NULL
1308  * @param client identification of the client
1309  * @param message the actual message
1310  */
1311 static void
1312 handle_peer_start (void *cls,
1313                    struct GNUNET_SERVER_Client *client,
1314                    const struct GNUNET_MessageHeader *message)
1315 {
1316   const struct GNUNET_TESTBED_PeerStartMessage *msg;
1317   struct GNUNET_TESTBED_PeerEventMessage *reply;
1318   uint32_t peer_id;
1319
1320   msg = (const struct GNUNET_TESTBED_PeerStartMessage *) message;
1321   peer_id = ntohl (msg->peer_id);
1322   if ((peer_id >= peer_list_size)
1323       || (NULL == peer_list[peer_id]))
1324   {
1325     GNUNET_break (0);
1326     /* FIXME: reply with failure message or forward to slave controller */
1327     GNUNET_SERVER_receive_done (client, GNUNET_OK);
1328     return;
1329   }
1330   if (GNUNET_OK != GNUNET_TESTING_peer_start (peer_list[peer_id]->peer))
1331   {
1332     /* FIXME: return FAILURE message */
1333     GNUNET_break (0);
1334     GNUNET_SERVER_receive_done (client, GNUNET_OK);
1335     return;
1336   }
1337   reply = GNUNET_malloc (sizeof (struct GNUNET_TESTBED_PeerEventMessage));
1338   reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_PEEREVENT);
1339   reply->header.size = htons (sizeof (struct GNUNET_TESTBED_PeerEventMessage));
1340   reply->event_type = htonl (GNUNET_TESTBED_ET_PEER_START);
1341   reply->host_id = htonl (master_context->host_id);
1342   reply->peer_id = msg->peer_id;
1343   reply->operation_id = msg->operation_id;
1344   queue_message (client, &reply->header);
1345   GNUNET_SERVER_receive_done (client, GNUNET_OK);
1346 }
1347
1348
1349 /**
1350  * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_DESTROYPEER messages
1351  *
1352  * @param cls NULL
1353  * @param client identification of the client
1354  * @param message the actual message
1355  */
1356 static void
1357 handle_peer_stop (void *cls,
1358                   struct GNUNET_SERVER_Client *client,
1359                   const struct GNUNET_MessageHeader *message)
1360 {
1361   const struct GNUNET_TESTBED_PeerStopMessage *msg;
1362   struct GNUNET_TESTBED_PeerEventMessage *reply;
1363   uint32_t peer_id;
1364
1365   msg = (const struct GNUNET_TESTBED_PeerStopMessage *) message;
1366   peer_id = ntohl (msg->peer_id);
1367   if ((peer_id >= peer_list_size) || (NULL == peer_list[peer_id]))
1368   {
1369     GNUNET_break (0);           /* FIXME: route to slave? */
1370     GNUNET_SERVER_receive_done (client, GNUNET_OK);
1371     return;
1372   }
1373   if (GNUNET_OK != GNUNET_TESTING_peer_stop (peer_list[peer_id]->peer))
1374   {
1375     /* FIXME: return FAILURE message */
1376     GNUNET_break (0);
1377     GNUNET_SERVER_receive_done (client, GNUNET_OK);
1378     return;
1379   }
1380   reply = GNUNET_malloc (sizeof (struct GNUNET_TESTBED_PeerEventMessage));
1381   reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_PEEREVENT);
1382   reply->header.size = htons (sizeof (struct GNUNET_TESTBED_PeerEventMessage));
1383   reply->event_type = htonl (GNUNET_TESTBED_ET_PEER_STOP);
1384   reply->host_id = htonl (master_context->host_id);
1385   reply->peer_id = msg->peer_id;
1386   reply->operation_id = msg->operation_id;
1387   queue_message (client, &reply->header);
1388   GNUNET_SERVER_receive_done (client, GNUNET_OK);
1389 }
1390
1391
1392 /**
1393  * Handler for GNUNET_MESSAGE_TYPE_TESTBED_GETPEERCONFIG messages
1394  *
1395  * @param cls NULL
1396  * @param client identification of the client
1397  * @param message the actual message
1398  */
1399 static void
1400 handle_peer_get_config (void *cls,
1401                         struct GNUNET_SERVER_Client *client,
1402                         const struct GNUNET_MessageHeader *message)
1403 {
1404   const struct GNUNET_TESTBED_PeerGetConfigurationMessage *msg;
1405   struct GNUNET_TESTBED_PeerConfigurationInformationMessage *reply;
1406   char *config;
1407   char *xconfig;
1408   size_t c_size;
1409   size_t xc_size;
1410   uint32_t peer_id;
1411   uint16_t msize;
1412
1413   msg = (const struct GNUNET_TESTBED_PeerGetConfigurationMessage *) message;
1414   peer_id = ntohl (msg->peer_id);
1415   if ((peer_id >= peer_list_size) || (NULL == peer_list[peer_id]))
1416   {
1417     /* FIXME: return FAILURE message */
1418     GNUNET_break (0);
1419     GNUNET_SERVER_receive_done (client, GNUNET_OK);
1420   }
1421   config = GNUNET_CONFIGURATION_serialize (peer_list[peer_id]->cfg,
1422                                            &c_size);
1423   xc_size = GNUNET_TESTBED_compress_config_ (config, c_size, &xconfig);
1424   GNUNET_free (config);
1425   msize = xc_size + sizeof (struct
1426                             GNUNET_TESTBED_PeerConfigurationInformationMessage);
1427   reply = GNUNET_realloc (xconfig, msize);
1428   (void) memmove (&reply[1], reply, xc_size);
1429   reply->header.size = htons (msize);
1430   reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_PEERCONFIG);
1431   reply->peer_id = msg->peer_id;
1432   reply->operation_id = msg->operation_id;
1433   GNUNET_TESTING_peer_get_identity (peer_list[peer_id]->peer,
1434                                     &reply->peer_identity);
1435   reply->config_size = htons ((uint16_t) c_size);
1436   queue_message (client, &reply->header);
1437   GNUNET_SERVER_receive_done (client, GNUNET_OK);
1438 }
1439
1440
1441 /**
1442  * Iterator over hash map entries.
1443  *
1444  * @param cls closure
1445  * @param key current key code
1446  * @param value value in the hash map
1447  * @return GNUNET_YES if we should continue to
1448  *         iterate,
1449  *         GNUNET_NO if not.
1450  */
1451 static int
1452 ss_map_free_iterator (void *cls,
1453                       const struct GNUNET_HashCode * key, void *value)
1454 {
1455   struct SharedService *ss = value;
1456
1457   GNUNET_assert (GNUNET_YES ==
1458                  GNUNET_CONTAINER_multihashmap_remove (ss_map, key, value));
1459   GNUNET_free (ss->name);
1460   GNUNET_free (ss);
1461   return GNUNET_YES;
1462 }
1463
1464
1465 /**
1466  * Task to clean up and shutdown nicely
1467  *
1468  * @param cls NULL
1469  * @param tc the TaskContext from scheduler
1470  */
1471 static void
1472 shutdown_task (void *cls,
1473                const struct GNUNET_SCHEDULER_TaskContext *tc)
1474 {
1475   struct LCFContextQueue *lcfq;
1476   uint32_t id;
1477
1478   shutdown_task_id = GNUNET_SCHEDULER_NO_TASK;
1479   LOG (GNUNET_ERROR_TYPE_DEBUG, "Shutting down testbed service\n");
1480   (void) GNUNET_CONTAINER_multihashmap_iterate (ss_map, &ss_map_free_iterator,
1481                                                 NULL);
1482   GNUNET_CONTAINER_multihashmap_destroy (ss_map);
1483   if (NULL != fh)
1484   {
1485     GNUNET_DISK_file_close (fh);
1486     fh = NULL;
1487   }
1488   if (NULL != lcfq_head)
1489   {
1490     if (GNUNET_SCHEDULER_NO_TASK != lcf_proc_task_id)
1491     {
1492       GNUNET_SCHEDULER_cancel (lcf_proc_task_id);
1493       lcf_proc_task_id = GNUNET_SCHEDULER_NO_TASK;
1494     }
1495     if (NULL != lcfq_head->lcf->rhandle)
1496       GNUNET_TESTBED_cancel_registration (lcfq_head->lcf->rhandle);
1497   }
1498   GNUNET_assert (GNUNET_SCHEDULER_NO_TASK == lcf_proc_task_id);
1499   for (lcfq = lcfq_head; NULL != lcfq; lcfq = lcfq_head)
1500   {
1501     GNUNET_free (lcfq->lcf->sxcfg);
1502     GNUNET_free (lcfq->lcf);
1503     GNUNET_CONTAINER_DLL_remove (lcfq_head, lcfq_tail, lcfq);
1504     GNUNET_free (lcfq);
1505   }
1506   /* Clear peer list */
1507   for (id = 0; id < peer_list_size; id++)
1508     if (NULL != peer_list[id])
1509     {
1510       GNUNET_TESTING_peer_destroy (peer_list[id]->peer);
1511       GNUNET_CONFIGURATION_destroy (peer_list[id]->cfg);
1512       GNUNET_free (peer_list[id]);
1513     }
1514   GNUNET_free_non_null (peer_list);
1515   /* Clear host list */
1516   for (id = 0; id < host_list_size; id++)
1517     if (NULL != host_list[id])
1518       GNUNET_TESTBED_host_destroy (host_list[id]);
1519   GNUNET_free_non_null (host_list);
1520   /* Clear route list */
1521   for (id = 0; id < route_list_size; id++)
1522     if (NULL != route_list[id])
1523       GNUNET_free (route_list[id]);
1524   GNUNET_free_non_null (route_list);
1525   /* Clear slave_list */
1526   for (id = 0; id < slave_list_size; id++)
1527     if (NULL != slave_list[id])
1528     {
1529       GNUNET_assert (NULL != slave_list[id]->controller);
1530       GNUNET_TESTBED_controller_disconnect (slave_list[id]->controller);
1531       if (NULL != slave_list[id]->controller_proc)
1532         GNUNET_TESTBED_controller_stop (slave_list[id]->controller_proc);
1533     }
1534   if (NULL != master_context)
1535   {
1536     GNUNET_free_non_null (master_context->master_ip);
1537     if (NULL != master_context->system)
1538       GNUNET_TESTING_system_destroy (master_context->system, GNUNET_YES);
1539   }
1540 }
1541
1542
1543 /**
1544  * Debug shutdown task in case of stdin getting closed
1545  *
1546  * @param cls NULL
1547  * @param tc the TaskContext from scheduler
1548  */
1549 static void
1550 shutdown_task_ (void *cls,
1551                 const struct GNUNET_SCHEDULER_TaskContext *tc)
1552 {
1553   LOG (GNUNET_ERROR_TYPE_DEBUG, "STDIN closed ...\n");
1554   shutdown_task (cls, tc);
1555 }
1556
1557
1558 /**
1559  * Callback for client disconnect
1560  *
1561  * @param cls NULL
1562  * @param client the client which has disconnected
1563  */
1564 static void
1565 client_disconnect_cb (void *cls, struct GNUNET_SERVER_Client *client)
1566 {
1567   if (NULL == master_context)
1568     return;
1569   if (client == master_context->client)
1570   {
1571     LOG (GNUNET_ERROR_TYPE_DEBUG, "Master client disconnected\n");
1572     GNUNET_SERVER_client_drop (client);
1573     /* should not be needed as we're terminated by failure to read
1574        from stdin, but if stdin fails for some reason, this shouldn't
1575        hurt for now --- might need to revise this later if we ever
1576        decide that master connections might be temporarily down
1577        for some reason */
1578     GNUNET_SCHEDULER_shutdown ();
1579   }
1580 }
1581
1582
1583 /**
1584  * Testbed setup
1585  *
1586  * @param cls closure
1587  * @param server the initialized server
1588  * @param cfg configuration to use
1589  */
1590 static void
1591 testbed_run (void *cls,
1592              struct GNUNET_SERVER_Handle *server,
1593              const struct GNUNET_CONFIGURATION_Handle *cfg)
1594 {
1595   static const struct GNUNET_SERVER_MessageHandler message_handlers[] =
1596     {
1597       {&handle_init, NULL, GNUNET_MESSAGE_TYPE_TESTBED_INIT,
1598        sizeof (struct GNUNET_TESTBED_InitMessage)},
1599       {&handle_add_host, NULL, GNUNET_MESSAGE_TYPE_TESTBED_ADDHOST, 0},
1600       {&handle_configure_shared_service, NULL,
1601        GNUNET_MESSAGE_TYPE_TESTBED_SERVICESHARE, 0},
1602       {&handle_link_controllers, NULL,
1603        GNUNET_MESSAGE_TYPE_TESTBED_LCONTROLLERS, 0},
1604       {&handle_peer_create, NULL, GNUNET_MESSAGE_TYPE_TESTBED_CREATEPEER, 0},
1605       {&handle_peer_destroy, NULL, GNUNET_MESSAGE_TYPE_TESTBED_DESTROYPEER,
1606        sizeof (struct GNUNET_TESTBED_PeerDestroyMessage)},
1607       {&handle_peer_start, NULL, GNUNET_MESSAGE_TYPE_TESTBED_STARTPEER,
1608        sizeof (struct GNUNET_TESTBED_PeerStartMessage)},
1609       {&handle_peer_stop, NULL, GNUNET_MESSAGE_TYPE_TESTBED_STOPPEER,
1610        sizeof (struct GNUNET_TESTBED_PeerStopMessage)},
1611       {&handle_peer_get_config, NULL, GNUNET_MESSAGE_TYPE_TESTBED_GETPEERCONFIG,
1612        sizeof (struct GNUNET_TESTBED_PeerGetConfigurationMessage)},
1613       {NULL}
1614     };
1615
1616   config = GNUNET_CONFIGURATION_dup (cfg);
1617   GNUNET_SERVER_add_handlers (server,
1618                               message_handlers);
1619   GNUNET_SERVER_disconnect_notify (server,
1620                                    &client_disconnect_cb,
1621                                    NULL);
1622   ss_map = GNUNET_CONTAINER_multihashmap_create (5);
1623   fh = GNUNET_DISK_get_handle_from_native (stdin);
1624   if (NULL == fh)
1625     shutdown_task_id =
1626       GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
1627                                     &shutdown_task,
1628                                     NULL);
1629   else
1630     shutdown_task_id =
1631       GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL,
1632                                       fh,
1633                                       &shutdown_task_,
1634                                       NULL);
1635   LOG_DEBUG ("Testbed startup complete\n");
1636 }
1637
1638
1639 /**
1640  * The starting point of execution
1641  */
1642 int main (int argc, char *const *argv)
1643 {
1644   return
1645     (GNUNET_OK ==
1646      GNUNET_SERVICE_run (argc,
1647                          argv,
1648                          "testbed",
1649                          GNUNET_SERVICE_OPTION_NONE,
1650                          &testbed_run,
1651                          NULL)) ? 0 : 1;
1652 }