From deddd8958a9d128a3a248163ead0996700b389c3 Mon Sep 17 00:00:00 2001
From: Sree Harsha Totakura <totakura@in.tum.de>
Date: Sun, 1 Jul 2012 15:37:23 +0000
Subject: [PATCH] link controller handler

---
 src/testbed/gnunet-service-testbed.c | 81 +++++++++++++++++++++++++++-
 src/testbed/testbed.h                |  5 ++
 src/testbed/testbed_api.c            |  1 +
 3 files changed, 85 insertions(+), 2 deletions(-)

diff --git a/src/testbed/gnunet-service-testbed.c b/src/testbed/gnunet-service-testbed.c
index 7f36485e7..7c0122159 100644
--- a/src/testbed/gnunet-service-testbed.c
+++ b/src/testbed/gnunet-service-testbed.c
@@ -27,6 +27,7 @@
 #include "platform.h"
 #include "gnunet_service_lib.h"
 #include "gnunet_server_lib.h"
+#include <zlib.h>
 
 #include "testbed.h"
 #include "gnunet_testbed_service.h"
@@ -413,14 +414,14 @@ handle_configure_shared_service (void *cls,
                                  struct GNUNET_SERVER_Client *client,
                                  const struct GNUNET_MessageHeader *message)
 {
-  struct GNUNET_TESTBED_ConfigureSharedServiceMessage *msg;
+  const struct GNUNET_TESTBED_ConfigureSharedServiceMessage *msg;
   struct SharedService *ss;
   char *service_name;
   struct GNUNET_HashCode hash;
   uint16_t msg_size;
   uint16_t service_name_size;
     
-  msg = (struct GNUNET_TESTBED_ConfigureSharedServiceMessage *) message;
+  msg = (const struct GNUNET_TESTBED_ConfigureSharedServiceMessage *) message;
   msg_size = ntohs (message->size);
   if (msg_size <= sizeof (struct GNUNET_TESTBED_ConfigureSharedServiceMessage))
   {
@@ -466,6 +467,80 @@ handle_configure_shared_service (void *cls,
 }
 
 
+/**
+ * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_LCONTROLLERS message
+ *
+ * @param cls NULL
+ * @param client identification of the client
+ * @param message the actual message
+ */
+static void 
+handle_link_controllers (void *cls,
+                         struct GNUNET_SERVER_Client *client,
+                         const struct GNUNET_MessageHeader *message)
+{
+  const struct GNUNET_TESTBED_ControllerLinkMessage *msg;
+  struct GNUNET_CONFIGURATION_Handle *cfg;
+  char *config;  
+  uLongf dest_size;
+  size_t config_size;
+  uint32_t delegated_host_id;
+  uint32_t slave_host_id;
+  uint16_t msize;
+   
+  msize = ntohs (message->size);
+  if (sizeof (struct GNUNET_TESTBED_ControllerLinkMessage) >= msize)
+  {
+    GNUNET_break (0);
+    GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
+    return;
+  }
+  msg = (const struct GNUNET_TESTBED_ControllerLinkMessage *) message;
+  delegated_host_id = ntohl (msg->delegated_host_id);
+  if ((delegated_host_id >= host_list_size) || 
+      (NULL == host_list[delegated_host_id]))
+  {
+    GNUNET_break (0);           /* Delegated host not registered with us */
+    GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
+    return;
+  }
+  slave_host_id = ntohl (msg->slave_host_id);
+  if ((slave_host_id >= host_list_size) || (NULL == host_list[slave_host_id]))
+  {
+    GNUNET_break (0);           /* Slave host not registered with us */
+    GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
+    return;
+  }
+  config_size = ntohs (msg->config_size);
+  config = GNUNET_malloc (config_size);
+  dest_size = (uLongf) config_size;
+  msize -= sizeof (struct GNUNET_TESTBED_ControllerLinkMessage);
+  if (Z_OK != uncompress ((Bytef *) config, &dest_size,
+                          (const Bytef *) &msg[1], (uLong) msize))
+  {
+    GNUNET_break (0);           /* Compression error */
+    GNUNET_free (config);
+    GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
+    return;
+  }
+  GNUNET_assert (config_size == dest_size);
+  cfg = GNUNET_CONFIGURATION_create ();
+  if (GNUNET_OK != GNUNET_CONFIGURATION_deserialize (cfg, config, config_size,
+                                                     GNUNET_NO))
+  {
+    GNUNET_break (0);           /* Configuration parsing error */
+    GNUNET_break (config);
+    GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
+    return;
+  }
+  GNUNET_free (config);
+  /* FIXME: start the slave controller if is_subordinate is yes; else store it
+  in routing */
+
+  GNUNET_CONFIGURATION_destroy (cfg);
+}
+
+
 /**
  * Iterator over hash map entries.
  *
@@ -567,6 +642,8 @@ testbed_run (void *cls,
       {&handle_add_host, NULL, GNUNET_MESSAGE_TYPE_TESTBED_ADDHOST, 0},
       {&handle_configure_shared_service, NULL,
        GNUNET_MESSAGE_TYPE_TESTBED_SERVICESHARE, 0},
+      {&handle_link_controllers, NULL,
+       GNUNET_MESSAGE_TYPE_TESTBED_LCONTROLLERS, 0},
       {NULL}
     };
 
diff --git a/src/testbed/testbed.h b/src/testbed/testbed.h
index 4e5d90c1a..6d71f6c51 100644
--- a/src/testbed/testbed.h
+++ b/src/testbed/testbed.h
@@ -168,6 +168,11 @@ struct GNUNET_TESTBED_ControllerLinkMessage
    */
   uint32_t slave_host_id GNUNET_PACKED;
 
+  /**
+   * The size of the uncompressed configuration
+   */
+  uint16_t config_size GNUNET_PACKED;
+
   /**
    * Set to 1 if the receiving controller is the master controller for
    * the slave host (and thus responsible for starting it?). 0 if not
diff --git a/src/testbed/testbed_api.c b/src/testbed/testbed_api.c
index 20e6810af..6e9fc491f 100644
--- a/src/testbed/testbed_api.c
+++ b/src/testbed/testbed_api.c
@@ -659,6 +659,7 @@ GNUNET_TESTBED_controller_link (struct GNUNET_TESTBED_Controller *master,
   msg->header.size = htons (msg_size);
   msg->delegated_host_id = htonl (GNUNET_TESTBED_host_get_id_ (delegated_host));
   msg->slave_host_id = htonl (GNUNET_TESTBED_host_get_id_ (slave_host));
+  msg->config_size = htons ((uint16_t) config_size);
   msg->is_subordinate = (GNUNET_YES == is_subordinate) ? 1 : 0;
   queue_message (master, (struct GNUNET_MessageHeader *) msg);
 }
-- 
2.25.1