2 This file is part of GNUnet.
3 (C) 2012 Christian Grothoff (and other contributing authors)
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.
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.
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.
22 * @file testbed/gnunet-service-testbed.c
23 * @brief implementation of the TESTBED service
24 * @author Sree Harsha Totakura
28 #include "gnunet_service_lib.h"
29 #include "gnunet_server_lib.h"
30 #include "gnunet_transport_service.h"
31 #include "gnunet_core_service.h"
32 #include "gnunet_hello_lib.h"
35 #include "gnunet_testbed_service.h"
37 #include "testbed_api.h"
38 #include "testbed_api_hosts.h"
39 #include "gnunet_testing_lib-new.h"
44 #define LOG(kind,...) \
45 GNUNET_log (kind, __VA_ARGS__)
50 #define LOG_DEBUG(...) \
51 LOG (GNUNET_ERROR_TYPE_DEBUG, __VA_ARGS__)
54 * By how much should the arrays lists grow
56 #define LIST_GROW_STEP 10
59 * Default timeout for operations which may take some time
61 #define TIMEOUT GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_SECONDS, 30)
64 * The main context information associated with the client which started us
69 * The client handle associated with this context
71 struct GNUNET_SERVER_Client *client;
74 * The network address of the master controller
79 * The TESTING system handle for starting peers locally
81 struct GNUNET_TESTING_System *system;
84 * Event mask of event to be responded in this context
89 * Our host id according to this context
96 * The message queue for sending messages to clients
101 * The message to be sent
103 struct GNUNET_MessageHeader *msg;
106 * The client to send the message to
108 struct GNUNET_SERVER_Client *client;
111 * next pointer for DLL
113 struct MessageQueue *next;
116 * prev pointer for DLL
118 struct MessageQueue *prev;
123 * The structure for identifying a shared service
128 * The name of the shared service
133 * Number of shared peers per instance of the shared service
138 * Number of peers currently sharing the service
140 uint32_t num_sharing;
155 * The host destination is reachable thru
162 * Structure representing a connected(directly-linked) controller
167 * The controller process handle if we had started the controller
169 struct GNUNET_TESTBED_ControllerProc *controller_proc;
172 * The controller handle
174 struct GNUNET_TESTBED_Controller *controller;
177 * The id of the host this controller is running on
184 * States of LCFContext
189 * The Context has been initialized; Nothing has been done on it
194 * Delegated host has been registered at the forwarding controller
196 DELEGATED_HOST_REGISTERED,
199 * The slave host has been registred at the forwarding controller
201 SLAVE_HOST_REGISTERED,
204 * The context has been finished (may have error)
211 * Link controllers request forwarding context
216 * The gateway which will pass the link message to delegated host
218 struct Slave *gateway;
221 * The controller link message that has to be forwarded to
223 struct GNUNET_TESTBED_ControllerLinkMessage *msg;
226 * The client which has asked to perform this operation
228 struct GNUNET_SERVER_Client *client;
231 * The host registration handle while registered hosts in this context
233 struct GNUNET_TESTBED_HostRegistrationHandle *rhandle;
236 * The id of the operation which created this context
238 uint64_t operation_id;
241 * The state of this context
243 enum LCFContextState state;
248 uint32_t delegated_host_id;
253 uint32_t slave_host_id;
259 * Structure of a queue entry in LCFContext request queue
261 struct LCFContextQueue
266 struct LCFContext *lcf;
271 struct LCFContextQueue *next;
276 struct LCFContextQueue *prev;
281 * A locally started peer
290 * The peer handle from testing API
292 struct GNUNET_TESTING_Peer *peer;
295 * The modified (by GNUNET_TESTING_peer_configure) configuration this
296 * peer is configured with
298 struct GNUNET_CONFIGURATION_Handle *cfg;
305 * The controller this peer is started through
307 struct GNUNET_TESTBED_Controller *controller;
314 * Our local reference id for this peer
319 * Is this peer local created?
327 * State information for overlay connect context
337 * Peer 1 has connected to peer0
339 OCC_STATE_PEER0_SUCCESS,
342 * Peer 2 has connected to peer1
344 OCC_STATE_PEER1_SUCCESS,
350 * Context information for connecting 2 peers in overlay
352 struct OverlayConnectContext
355 * The client which has requested for overlay connection
357 struct GNUNET_SERVER_Client *client;
360 * the peer which has to connect to the other peer
367 struct Peer *other_peer;
370 * Transport handle of the first peer to offer second peer's HELLO
372 struct GNUNET_TRANSPORT_Handle *p1th;
375 * Transport handle of other peer to get its HELLO
377 struct GNUNET_TRANSPORT_Handle *p2th;
380 * Core handles of the peer which has to connect to other peer
382 struct GNUNET_CORE_Handle *ch;
385 * HELLO of the other peer
387 struct GNUNET_MessageHeader *hello;
390 * Get hello handle for the other peer
392 struct GNUNET_TRANSPORT_GetHelloHandle *ghh;
395 * The error message we send if this overlay connect operation has timed out
400 * The peer identity of the first peer
402 struct GNUNET_PeerIdentity peer_identity;
405 * The peer identity of the other peer
407 struct GNUNET_PeerIdentity other_peer_identity;
410 * The id of the operation responsible for creating this context
415 * The id of the task for sending HELLO of peer 2 to peer 1 and ask peer 1 to
418 GNUNET_SCHEDULER_TaskIdentifier send_hello_task;
421 * The id of the overlay connect timeout task
423 GNUNET_SCHEDULER_TaskIdentifier timeout_task;
426 * State information for determining whose HELLOs have been successfully
435 * Context information for operations forwarded to subcontrollers
437 struct ForwardedOperationContext
440 * The generated operation context
442 struct OperationContext *opc;
445 * The client to which we have to reply
447 struct GNUNET_SERVER_Client *client;
455 * Task ID for the timeout task
457 GNUNET_SCHEDULER_TaskIdentifier timeout_task;
460 * The id of the operation that has been forwarded
462 uint64_t operation_id;
468 * Context information used while linking controllers
470 struct LinkControllersContext
473 * The client which initiated the link controller operation
475 struct GNUNET_SERVER_Client *client;
478 * The ID of the operation
480 uint64_t operation_id;
483 * Pointer to the slave handle if we are directly starting/connecting to the controller
491 * The master context; generated with the first INIT message
493 static struct Context *master_context;
500 * Current Transmit Handle; NULL if no notify transmit exists currently
502 static struct GNUNET_SERVER_TransmitHandle *transmit_handle;
509 * The head for the LCF queue
511 static struct LCFContextQueue *lcfq_head;
514 * The tail for the LCF queue
516 static struct LCFContextQueue *lcfq_tail;
519 * The message queue head
521 static struct MessageQueue *mq_head;
524 * The message queue tail
526 static struct MessageQueue *mq_tail;
531 static struct GNUNET_TESTBED_Host **host_list;
536 static struct Route **route_list;
539 * A list of directly linked neighbours
541 static struct Slave **slave_list;
544 * A list of peers we own locally
546 static struct Peer **peer_list;
549 * The hashmap of shared services
551 static struct GNUNET_CONTAINER_MultiHashMap *ss_map;
554 * The size of the host list
556 static uint32_t host_list_size;
559 * The size of the route list
561 static uint32_t route_list_size;
564 * The size of directly linked neighbours list
566 static uint32_t slave_list_size;
569 * The size of the peer list
571 static uint32_t peer_list_size;
578 * The lcf_task handle
580 static GNUNET_SCHEDULER_TaskIdentifier lcf_proc_task_id;
583 * The shutdown task handle
585 static GNUNET_SCHEDULER_TaskIdentifier shutdown_task_id;
589 * Function called to notify a client about the connection begin ready to queue
590 * more data. "buf" will be NULL and "size" zero if the connection was closed
591 * for writing in the meantime.
594 * @param size number of bytes available in buf
595 * @param buf where the callee should write the message
596 * @return number of bytes written to buf
599 transmit_ready_notify (void *cls, size_t size, void *buf)
601 struct MessageQueue *mq_entry;
603 transmit_handle = NULL;
605 GNUNET_assert (NULL != mq_entry);
608 GNUNET_assert (ntohs (mq_entry->msg->size) <= size);
609 size = ntohs (mq_entry->msg->size);
610 memcpy (buf, mq_entry->msg, size);
611 GNUNET_free (mq_entry->msg);
612 GNUNET_CONTAINER_DLL_remove (mq_head, mq_tail, mq_entry);
613 GNUNET_free (mq_entry);
615 if (NULL != mq_entry)
617 GNUNET_SERVER_notify_transmit_ready (mq_entry->client,
618 ntohs (mq_entry->msg->size),
619 GNUNET_TIME_UNIT_FOREVER_REL,
620 &transmit_ready_notify, NULL);
626 * Queues a message in send queue for sending to the service
628 * @param client the client to whom the queued message has to be sent
629 * @param msg the message to queue
632 queue_message (struct GNUNET_SERVER_Client *client,
633 struct GNUNET_MessageHeader *msg)
635 struct MessageQueue *mq_entry;
639 type = ntohs (msg->type);
640 size = ntohs (msg->size);
641 GNUNET_assert ((GNUNET_MESSAGE_TYPE_TESTBED_INIT <= type) &&
642 (GNUNET_MESSAGE_TYPE_TESTBED_MAX > type));
643 mq_entry = GNUNET_malloc (sizeof (struct MessageQueue));
645 mq_entry->client = client;
646 LOG_DEBUG ("Queueing message of type %u, size %u for sending\n", type,
648 GNUNET_CONTAINER_DLL_insert_tail (mq_head, mq_tail, mq_entry);
649 if (NULL == transmit_handle)
651 GNUNET_SERVER_notify_transmit_ready (client, size,
652 GNUNET_TIME_UNIT_FOREVER_REL,
653 &transmit_ready_notify, NULL);
658 * Similar to GNUNET_realloc; however clears tail part of newly allocated memory
660 * @param ptr the memory block to realloc
661 * @param size the size of ptr
662 * @param new_size the size to which ptr has to be realloc'ed
663 * @return the newly reallocated memory block
666 TESTBED_realloc (void *ptr, size_t size, size_t new_size)
668 ptr = GNUNET_realloc (ptr, new_size);
670 ptr = memset (ptr + size, 0, new_size - size);
676 * Function to add a host to the current list of known hosts
678 * @param host the host to add
679 * @return GNUNET_OK on success; GNUNET_SYSERR on failure due to host-id
683 host_list_add (struct GNUNET_TESTBED_Host *host)
687 host_id = GNUNET_TESTBED_host_get_id_ (host);
688 if (host_list_size <= host_id)
691 TESTBED_realloc (host_list,
692 sizeof (struct GNUNET_TESTBED_Host *) * host_list_size,
693 sizeof (struct GNUNET_TESTBED_Host *) *
694 (host_list_size + LIST_GROW_STEP));
695 host_list_size += LIST_GROW_STEP;
697 if (NULL != host_list[host_id])
699 LOG_DEBUG ("A host with id: %u already exists\n", host_id);
700 return GNUNET_SYSERR;
702 host_list[host_id] = host;
708 * Adds a route to the route list
710 * @param route the route to add
713 route_list_add (struct Route *route)
715 if (route->dest >= route_list_size)
718 TESTBED_realloc (route_list, sizeof (struct Route *) * route_list_size,
719 sizeof (struct Route *) * (route_list_size +
721 route_list_size += LIST_GROW_STEP;
723 GNUNET_assert (NULL == route_list[route->dest]);
724 route_list[route->dest] = route;
729 * Adds a slave to the slave array
731 * @param slave the slave controller to add
734 slave_list_add (struct Slave *slave)
736 if (slave->host_id >= slave_list_size)
739 TESTBED_realloc (slave_list, sizeof (struct Slave *) * slave_list_size,
740 sizeof (struct Slave *) * (slave_list_size +
742 slave_list_size += LIST_GROW_STEP;
744 GNUNET_assert (NULL == slave_list[slave->host_id]);
745 slave_list[slave->host_id] = slave;
750 * Adds a peer to the peer array
752 * @param peer the peer to add
755 peer_list_add (struct Peer *peer)
759 orig_size = peer_list_size;
760 if (peer->id >= peer_list_size)
762 while (peer->id >= peer_list_size)
763 peer_list_size += LIST_GROW_STEP;
765 TESTBED_realloc (peer_list, sizeof (struct Peer *) * orig_size,
766 sizeof (struct Peer *) * peer_list_size);
768 GNUNET_assert (NULL == peer_list[peer->id]);
769 peer_list[peer->id] = peer;
774 * Removes a the give peer from the peer array
776 * @param peer the peer to be removed
779 peer_list_remove (struct Peer *peer)
784 peer_list[peer->id] = NULL;
785 orig_size = peer_list_size;
786 while (peer_list_size >= LIST_GROW_STEP)
788 for (id = peer_list_size - 1;
789 (id >= peer_list_size - LIST_GROW_STEP) && (id != UINT32_MAX); id--)
790 if (NULL != peer_list[id])
792 if (id != ((peer_list_size - LIST_GROW_STEP) - 1))
794 peer_list_size -= LIST_GROW_STEP;
796 if (orig_size == peer_list_size)
799 GNUNET_realloc (peer_list, sizeof (struct Peer *) * peer_list_size);
804 * Finds the route with directly connected host as destination through which
805 * the destination host can be reached
807 * @param host_id the id of the destination host
808 * @return the route with directly connected destination host; NULL if no route
811 static struct Route *
812 find_dest_route (uint32_t host_id)
816 while (NULL != (route = route_list[host_id]))
818 if (route->thru == master_context->host_id)
820 host_id = route->thru;
827 * Routes message to a host given its host_id
829 * @param host_id the id of the destination host
830 * @param msg the message to be routed
833 route_message (uint32_t host_id, const struct GNUNET_MessageHeader *msg)
840 * Send operation failure message to client
842 * @param client the client to which the failure message has to be sent to
843 * @param operation_id the id of the failed operation
844 * @param emsg the error message; can be NULL
847 send_operation_fail_msg (struct GNUNET_SERVER_Client *client,
848 uint64_t operation_id, const char *emsg)
850 struct GNUNET_TESTBED_OperationFailureEventMessage *msg;
854 msize = sizeof (struct GNUNET_TESTBED_OperationFailureEventMessage);
855 emsg_len = (NULL == emsg) ? 0 : strlen (emsg) + 1;
857 msg = GNUNET_malloc (msize);
858 msg->header.size = htons (msize);
859 msg->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_OPERATIONEVENT);
860 msg->event_type = htonl (GNUNET_TESTBED_ET_OPERATION_FINISHED);
861 msg->operation_id = GNUNET_htonll (operation_id);
863 memcpy (&msg[1], emsg, emsg_len);
864 queue_message (client, &msg->header);
869 * Function to send generic operation success message to given client
871 * @param client the client to send the message to
872 * @param operation_id the id of the operation which was successful
875 send_operation_success_msg (struct GNUNET_SERVER_Client *client,
876 uint64_t operation_id)
878 struct GNUNET_TESTBED_GenericOperationSuccessEventMessage *msg;
881 msize = sizeof (struct GNUNET_TESTBED_GenericOperationSuccessEventMessage);
882 msg = GNUNET_malloc (msize);
883 msg->header.size = htons (msize);
884 msg->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_GENERICOPSUCCESS);
885 msg->operation_id = GNUNET_htonll (operation_id);
886 msg->event_type = htonl (GNUNET_TESTBED_ET_OPERATION_FINISHED);
887 queue_message (client, &msg->header);
892 * The Link Controller forwarding task
894 * @param cls the LCFContext
895 * @param tc the Task context from scheduler
898 lcf_proc_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
902 * Completion callback for host registrations while forwarding Link Controller messages
904 * @param cls the LCFContext
905 * @param emsg the error message; NULL if host registration is successful
908 lcf_proc_cc (void *cls, const char *emsg)
910 struct LCFContext *lcf = cls;
913 GNUNET_assert (GNUNET_SCHEDULER_NO_TASK == lcf_proc_task_id);
918 goto registration_error;
919 lcf->state = DELEGATED_HOST_REGISTERED;
920 lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcf);
922 case DELEGATED_HOST_REGISTERED:
924 goto registration_error;
925 lcf->state = SLAVE_HOST_REGISTERED;
926 lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcf);
929 GNUNET_assert (0); /* Shouldn't reach here */
934 LOG (GNUNET_ERROR_TYPE_WARNING, "Host registration failed with message: %s\n",
936 lcf->state = FINISHED;
937 lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcf);
942 * Callback to be called when forwarded link controllers operation is
943 * successfull. We have to relay the reply msg back to the client
945 * @param cls ForwardedOperationContext
946 * @param msg the peer create success message
949 forwarded_operation_reply_relay (void *cls,
950 const struct GNUNET_MessageHeader *msg)
952 struct ForwardedOperationContext *fopc = cls;
953 struct GNUNET_MessageHeader *dup_msg;
956 msize = ntohs (msg->size);
957 dup_msg = GNUNET_malloc (msize);
958 (void) memcpy (dup_msg, msg, msize);
959 queue_message (fopc->client, dup_msg);
960 GNUNET_SERVER_client_drop (fopc->client);
961 GNUNET_SCHEDULER_cancel (fopc->timeout_task);
967 * Task to free resources when forwarded link controllers has been timedout
969 * @param cls the ForwardedOperationContext
970 * @param tc the task context from scheduler
973 forwarded_operation_timeout (void *cls,
974 const struct GNUNET_SCHEDULER_TaskContext *tc)
976 struct ForwardedOperationContext *fopc = cls;
978 GNUNET_TESTBED_forward_operation_msg_cancel_ (fopc->opc);
979 send_operation_fail_msg (fopc->client, fopc->operation_id, "Timeout");
980 GNUNET_SERVER_client_drop (fopc->client);
986 * The Link Controller forwarding task
988 * @param cls the LCFContext
989 * @param tc the Task context from scheduler
992 lcf_proc_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
994 struct LCFContext *lcf = cls;
995 struct LCFContextQueue *lcfq;
996 struct ForwardedOperationContext *fopc;
998 lcf_proc_task_id = GNUNET_SCHEDULER_NO_TASK;
1003 GNUNET_TESTBED_is_host_registered_ (host_list[lcf->delegated_host_id],
1004 lcf->gateway->controller))
1007 GNUNET_TESTBED_register_host (lcf->gateway->controller,
1008 host_list[lcf->delegated_host_id],
1013 lcf->state = DELEGATED_HOST_REGISTERED;
1014 lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcf);
1017 case DELEGATED_HOST_REGISTERED:
1019 GNUNET_TESTBED_is_host_registered_ (host_list[lcf->slave_host_id],
1020 lcf->gateway->controller))
1023 GNUNET_TESTBED_register_host (lcf->gateway->controller,
1024 host_list[lcf->slave_host_id],
1029 lcf->state = SLAVE_HOST_REGISTERED;
1030 lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcf);
1033 case SLAVE_HOST_REGISTERED:
1034 fopc = GNUNET_malloc (sizeof (struct ForwardedOperationContext));
1035 fopc->client = lcf->client;
1036 fopc->operation_id = lcf->operation_id;
1038 GNUNET_TESTBED_forward_operation_msg_ (lcf->gateway->controller,
1041 &forwarded_operation_reply_relay,
1043 fopc->timeout_task =
1044 GNUNET_SCHEDULER_add_delayed (TIMEOUT, &forwarded_operation_timeout,
1046 lcf->state = FINISHED;
1047 lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcf);
1051 GNUNET_assert (lcfq->lcf == lcf);
1052 GNUNET_free (lcf->msg);
1054 GNUNET_CONTAINER_DLL_remove (lcfq_head, lcfq_tail, lcfq);
1056 if (NULL != lcfq_head)
1058 GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcfq_head->lcf);
1064 * Callback for event from slave controllers
1066 * @param cls struct Slave *
1067 * @param event information about the event
1070 slave_event_callback (void *cls,
1071 const struct GNUNET_TESTBED_EventInformation *event)
1078 * Callback to signal successfull startup of the controller process
1080 * @param cls the closure from GNUNET_TESTBED_controller_start()
1081 * @param cfg the configuration with which the controller has been started;
1082 * NULL if status is not GNUNET_OK
1083 * @param status GNUNET_OK if the startup is successfull; GNUNET_SYSERR if not,
1084 * GNUNET_TESTBED_controller_stop() shouldn't be called in this case
1087 slave_status_callback (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg,
1090 struct LinkControllersContext *lcc = cls;
1092 if (GNUNET_SYSERR == status)
1094 lcc->slave->controller_proc = NULL;
1095 LOG (GNUNET_ERROR_TYPE_WARNING, "Unexpected slave shutdown\n");
1096 GNUNET_SCHEDULER_shutdown (); /* We too shutdown */
1099 lcc->slave->controller =
1100 GNUNET_TESTBED_controller_connect (cfg, host_list[lcc->slave->host_id],
1101 master_context->event_mask,
1102 &slave_event_callback, lcc->slave);
1103 if (NULL != lcc->slave->controller)
1104 send_operation_success_msg (lcc->client, lcc->operation_id);
1106 send_operation_fail_msg (lcc->client, lcc->operation_id,
1107 "Could not connect to delegated controller");
1108 GNUNET_SERVER_client_drop (lcc->client);
1114 * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_INIT messages
1117 * @param client identification of the client
1118 * @param message the actual message
1121 handle_init (void *cls, struct GNUNET_SERVER_Client *client,
1122 const struct GNUNET_MessageHeader *message)
1124 const struct GNUNET_TESTBED_InitMessage *msg;
1125 struct GNUNET_TESTBED_Host *host;
1126 const char *controller_hostname;
1129 if (NULL != master_context)
1132 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1135 msg = (const struct GNUNET_TESTBED_InitMessage *) message;
1136 msize = ntohs (message->size);
1137 if (msize <= sizeof (struct GNUNET_TESTBED_InitMessage))
1140 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1143 msize -= sizeof (struct GNUNET_TESTBED_InitMessage);
1144 controller_hostname = (const char *) &msg[1];
1145 if ('\0' != controller_hostname[msize - 1])
1148 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1151 master_context = GNUNET_malloc (sizeof (struct Context));
1152 master_context->client = client;
1153 master_context->host_id = ntohl (msg->host_id);
1154 master_context->master_ip = GNUNET_strdup (controller_hostname);
1155 LOG_DEBUG ("Master Controller IP: %s\n", master_context->master_ip);
1156 master_context->system =
1157 GNUNET_TESTING_system_create ("testbed", master_context->master_ip);
1159 GNUNET_TESTBED_host_create_with_id (master_context->host_id, NULL, NULL,
1161 host_list_add (host);
1162 master_context->event_mask = GNUNET_ntohll (msg->event_mask);
1163 GNUNET_SERVER_client_keep (client);
1164 LOG_DEBUG ("Created master context with host ID: %u\n",
1165 master_context->host_id);
1166 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1171 * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_ADDHOST messages
1174 * @param client identification of the client
1175 * @param message the actual message
1178 handle_add_host (void *cls, struct GNUNET_SERVER_Client *client,
1179 const struct GNUNET_MessageHeader *message)
1181 struct GNUNET_TESTBED_Host *host;
1182 const struct GNUNET_TESTBED_AddHostMessage *msg;
1183 struct GNUNET_TESTBED_HostConfirmedMessage *reply;
1188 uint16_t username_length;
1189 uint16_t hostname_length;
1190 uint16_t reply_size;
1193 msg = (const struct GNUNET_TESTBED_AddHostMessage *) message;
1194 msize = ntohs (msg->header.size);
1195 username = (char *) &(msg[1]);
1196 username_length = ntohs (msg->user_name_length);
1197 GNUNET_assert (msize > (sizeof (struct GNUNET_TESTBED_AddHostMessage) + username_length + 1)); /* msg must contain hostname */
1198 if (0 != username_length)
1199 GNUNET_assert ('\0' == username[username_length]);
1200 username_length = (0 == username_length) ? 0 : username_length + 1;
1201 hostname = username + username_length;
1203 msize - (sizeof (struct GNUNET_TESTBED_AddHostMessage) + username_length);
1204 GNUNET_assert ('\0' == hostname[hostname_length - 1]);
1205 GNUNET_assert (strlen (hostname) == hostname_length - 1);
1206 host_id = ntohl (msg->host_id);
1207 LOG_DEBUG ("Received ADDHOST message\n");
1208 LOG_DEBUG ("-------host id: %u\n", host_id);
1209 if (NULL != hostname)
1210 LOG_DEBUG ("-------hostname: %s\n", hostname);
1211 if (0 != username_length)
1212 LOG_DEBUG ("-------username: %s\n", username);
1215 LOG_DEBUG ("-------username: NULL\n");
1218 LOG_DEBUG ("-------ssh port: %u\n", ntohs (msg->ssh_port));
1220 GNUNET_TESTBED_host_create_with_id (host_id, hostname, username,
1221 ntohs (msg->ssh_port));
1222 GNUNET_assert (NULL != host);
1223 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1224 reply_size = sizeof (struct GNUNET_TESTBED_HostConfirmedMessage);
1225 if (GNUNET_OK != host_list_add (host))
1227 /* We are unable to add a host */
1228 emsg = "A host exists with given host-id";
1229 LOG_DEBUG ("%s: %u", emsg, host_id);
1230 GNUNET_TESTBED_host_destroy (host);
1231 reply_size += strlen (emsg) + 1;
1232 reply = GNUNET_malloc (reply_size);
1233 memcpy (&reply[1], emsg, strlen (emsg) + 1);
1236 reply = GNUNET_malloc (reply_size);
1237 reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_ADDHOSTCONFIRM);
1238 reply->header.size = htons (reply_size);
1239 reply->host_id = htonl (host_id);
1240 queue_message (client, &reply->header);
1245 * Iterator over hash map entries.
1247 * @param cls closure
1248 * @param key current key code
1249 * @param value value in the hash map
1250 * @return GNUNET_YES if we should continue to
1255 ss_exists_iterator (void *cls, const struct GNUNET_HashCode *key, void *value)
1257 struct SharedService *queried_ss = cls;
1258 struct SharedService *ss = value;
1260 if (0 == strcmp (ss->name, queried_ss->name))
1268 * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_ADDHOST messages
1271 * @param client identification of the client
1272 * @param message the actual message
1275 handle_configure_shared_service (void *cls, struct GNUNET_SERVER_Client *client,
1276 const struct GNUNET_MessageHeader *message)
1278 const struct GNUNET_TESTBED_ConfigureSharedServiceMessage *msg;
1279 struct SharedService *ss;
1281 struct GNUNET_HashCode hash;
1283 uint16_t service_name_size;
1285 msg = (const struct GNUNET_TESTBED_ConfigureSharedServiceMessage *) message;
1286 msg_size = ntohs (message->size);
1287 if (msg_size <= sizeof (struct GNUNET_TESTBED_ConfigureSharedServiceMessage))
1290 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1294 msg_size - sizeof (struct GNUNET_TESTBED_ConfigureSharedServiceMessage);
1295 service_name = (char *) &msg[1];
1296 if ('\0' != service_name[service_name_size - 1])
1299 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1302 LOG_DEBUG ("Received service sharing request for %s, with %d peers\n",
1303 service_name, ntohl (msg->num_peers));
1304 if (ntohl (msg->host_id) != master_context->host_id)
1306 route_message (ntohl (msg->host_id), message);
1307 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1310 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1311 ss = GNUNET_malloc (sizeof (struct SharedService));
1312 ss->name = strdup (service_name);
1313 ss->num_shared = ntohl (msg->num_peers);
1314 GNUNET_CRYPTO_hash (ss->name, service_name_size, &hash);
1315 if (GNUNET_SYSERR ==
1316 GNUNET_CONTAINER_multihashmap_get_multiple (ss_map, &hash,
1317 &ss_exists_iterator, ss))
1319 LOG (GNUNET_ERROR_TYPE_WARNING,
1320 "Service %s already configured as a shared service. "
1321 "Ignoring service sharing request \n", ss->name);
1322 GNUNET_free (ss->name);
1326 GNUNET_CONTAINER_multihashmap_put (ss_map, &hash, ss,
1327 GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE);
1332 * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_LCONTROLLERS message
1335 * @param client identification of the client
1336 * @param message the actual message
1339 handle_link_controllers (void *cls, struct GNUNET_SERVER_Client *client,
1340 const struct GNUNET_MessageHeader *message)
1342 const struct GNUNET_TESTBED_ControllerLinkMessage *msg;
1343 struct GNUNET_CONFIGURATION_Handle *cfg;
1344 struct LCFContextQueue *lcfq;
1345 struct Route *route;
1346 struct Route *new_route;
1350 uint32_t delegated_host_id;
1351 uint32_t slave_host_id;
1354 if (NULL == master_context)
1357 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1360 msize = ntohs (message->size);
1361 if (sizeof (struct GNUNET_TESTBED_ControllerLinkMessage) >= msize)
1364 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1367 msg = (const struct GNUNET_TESTBED_ControllerLinkMessage *) message;
1368 delegated_host_id = ntohl (msg->delegated_host_id);
1369 if (delegated_host_id == master_context->host_id)
1372 LOG (GNUNET_ERROR_TYPE_WARNING, "Trying to link ourselves\n");
1373 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1376 if ((delegated_host_id >= host_list_size) ||
1377 (NULL == host_list[delegated_host_id]))
1379 LOG (GNUNET_ERROR_TYPE_WARNING, "Delegated host not registered with us\n");
1380 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1383 slave_host_id = ntohl (msg->slave_host_id);
1384 if ((slave_host_id >= host_list_size) || (NULL == host_list[slave_host_id]))
1386 LOG (GNUNET_ERROR_TYPE_WARNING, "Slave host not registered with us\n");
1387 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1390 if (slave_host_id == delegated_host_id)
1392 LOG (GNUNET_ERROR_TYPE_WARNING, "Slave and delegated host are same\n");
1393 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1397 if (slave_host_id == master_context->host_id) /* Link from us */
1399 struct Slave *slave;
1401 msize -= sizeof (struct GNUNET_TESTBED_ControllerLinkMessage);
1402 config_size = ntohs (msg->config_size);
1403 if ((delegated_host_id < slave_list_size) && (NULL != slave_list[delegated_host_id])) /* We have already added */
1405 LOG (GNUNET_ERROR_TYPE_WARNING, "Host %u already connected\n",
1407 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1410 config = GNUNET_malloc (config_size);
1411 dest_size = (uLongf) config_size;
1413 uncompress ((Bytef *) config, &dest_size, (const Bytef *) &msg[1],
1416 GNUNET_break (0); /* Compression error */
1417 GNUNET_free (config);
1418 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1421 if (config_size != dest_size)
1423 LOG (GNUNET_ERROR_TYPE_WARNING, "Uncompressed config size mismatch\n");
1424 GNUNET_free (config);
1425 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1428 cfg = GNUNET_CONFIGURATION_create (); /* Free here or in lcfcontext */
1430 GNUNET_CONFIGURATION_deserialize (cfg, config, config_size, GNUNET_NO))
1432 GNUNET_break (0); /* Configuration parsing error */
1433 GNUNET_free (config);
1434 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1437 GNUNET_free (config);
1438 if ((delegated_host_id < slave_list_size) &&
1439 (NULL != slave_list[delegated_host_id]))
1441 GNUNET_break (0); /* Configuration parsing error */
1442 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1445 slave = GNUNET_malloc (sizeof (struct Slave));
1446 slave->host_id = delegated_host_id;
1447 slave_list_add (slave);
1448 if (1 == msg->is_subordinate)
1450 struct LinkControllersContext *lcc;
1452 lcc = GNUNET_malloc (sizeof (struct LinkControllersContext));
1453 lcc->operation_id = GNUNET_ntohll (msg->operation_id);
1454 GNUNET_SERVER_client_keep (client);
1455 lcc->client = client;
1457 slave->controller_proc =
1458 GNUNET_TESTBED_controller_start (master_context->master_ip,
1459 host_list[slave->host_id], cfg,
1460 &slave_status_callback, lcc);
1465 GNUNET_TESTBED_controller_connect (cfg, host_list[slave->host_id],
1466 master_context->event_mask,
1467 &slave_event_callback, slave);
1468 if (NULL != slave->controller)
1469 send_operation_success_msg (client, GNUNET_ntohll (msg->operation_id));
1471 send_operation_fail_msg (client, GNUNET_ntohll (msg->operation_id),
1472 "Could not connect to delegated controller");
1474 GNUNET_CONFIGURATION_destroy (cfg);
1475 new_route = GNUNET_malloc (sizeof (struct Route));
1476 new_route->dest = delegated_host_id;
1477 new_route->thru = master_context->host_id;
1478 route_list_add (new_route);
1479 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1483 /* Route the request */
1484 if (slave_host_id >= route_list_size)
1486 LOG (GNUNET_ERROR_TYPE_WARNING, "No route towards slave host");
1487 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1490 lcfq = GNUNET_malloc (sizeof (struct LCFContextQueue));
1491 lcfq->lcf = GNUNET_malloc (sizeof (struct LCFContext));
1492 lcfq->lcf->delegated_host_id = delegated_host_id;
1493 lcfq->lcf->slave_host_id = slave_host_id;
1494 route = find_dest_route (slave_host_id);
1495 GNUNET_assert (NULL != route); /* because we add routes carefully */
1496 GNUNET_assert (route->dest < slave_list_size);
1497 GNUNET_assert (NULL != slave_list[route->dest]);
1498 lcfq->lcf->state = INIT;
1499 lcfq->lcf->operation_id = GNUNET_ntohll (msg->operation_id);
1500 lcfq->lcf->gateway = slave_list[route->dest];
1501 lcfq->lcf->msg = GNUNET_malloc (msize);
1502 (void) memcpy (lcfq->lcf->msg, msg, msize);
1503 GNUNET_SERVER_client_keep (client);
1504 lcfq->lcf->client = client;
1505 if (NULL == lcfq_head)
1507 GNUNET_assert (GNUNET_SCHEDULER_NO_TASK == lcf_proc_task_id);
1508 GNUNET_CONTAINER_DLL_insert_tail (lcfq_head, lcfq_tail, lcfq);
1509 lcf_proc_task_id = GNUNET_SCHEDULER_add_now (&lcf_proc_task, lcfq->lcf);
1512 GNUNET_CONTAINER_DLL_insert_tail (lcfq_head, lcfq_tail, lcfq);
1513 /* FIXME: Adding a new route should happen after the controllers are linked
1515 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1516 new_route = GNUNET_malloc (sizeof (struct Route));
1517 new_route->dest = delegated_host_id;
1518 new_route->thru = route->dest;
1519 route_list_add (new_route);
1524 * The task to be executed if the forwarded peer create operation has been
1527 * @param cls the FowardedOperationContext
1528 * @param tc the TaskContext from the scheduler
1531 peer_create_forward_timeout (void *cls,
1532 const struct GNUNET_SCHEDULER_TaskContext *tc)
1534 struct ForwardedOperationContext *fo_ctxt = cls;
1536 /* send error msg to client */
1537 send_operation_fail_msg (fo_ctxt->client, fo_ctxt->operation_id, "Timedout");
1538 GNUNET_SERVER_client_drop (fo_ctxt->client);
1539 GNUNET_TESTBED_forward_operation_msg_cancel_ (fo_ctxt->opc);
1540 GNUNET_free (fo_ctxt);
1545 * Callback to be called when forwarded peer create operation is
1546 * successfull. We have to relay the reply msg back to the client
1548 * @param cls ForwardedOperationContext
1549 * @param msg the peer create success message
1552 peer_create_success_cb (void *cls, const struct GNUNET_MessageHeader *msg)
1554 struct ForwardedOperationContext *fo_ctxt = cls;
1555 const struct GNUNET_TESTBED_PeerCreateSuccessEventMessage *success_msg;
1556 struct GNUNET_MessageHeader *dup_msg;
1560 GNUNET_SCHEDULER_cancel (fo_ctxt->timeout_task);
1561 if (ntohs (msg->type) == GNUNET_MESSAGE_TYPE_TESTBED_PEERCREATESUCCESS)
1564 (const struct GNUNET_TESTBED_PeerCreateSuccessEventMessage *) msg;
1565 peer = GNUNET_malloc (sizeof (struct Peer));
1566 peer->is_remote = GNUNET_YES;
1567 peer->id = ntohl (success_msg->peer_id);
1568 GNUNET_assert (NULL != fo_ctxt->cls);
1569 peer->details.remote.controller = fo_ctxt->cls;
1570 peer_list_add (peer);
1572 msize = ntohs (msg->size);
1573 dup_msg = GNUNET_malloc (msize);
1574 (void) memcpy (dup_msg, msg, msize);
1575 queue_message (fo_ctxt->client, dup_msg);
1576 GNUNET_SERVER_client_drop (fo_ctxt->client);
1577 GNUNET_free (fo_ctxt);
1583 * Handler for GNUNET_MESSAGE_TYPE_TESTBED_CREATEPEER messages
1586 * @param client identification of the client
1587 * @param message the actual message
1590 handle_peer_create (void *cls, struct GNUNET_SERVER_Client *client,
1591 const struct GNUNET_MessageHeader *message)
1593 const struct GNUNET_TESTBED_PeerCreateMessage *msg;
1594 struct GNUNET_TESTBED_PeerCreateSuccessEventMessage *reply;
1595 struct GNUNET_CONFIGURATION_Handle *cfg;
1596 struct ForwardedOperationContext *fo_ctxt;
1597 struct Route *route;
1602 uint32_t config_size;
1608 msize = ntohs (message->size);
1609 if (msize <= sizeof (struct GNUNET_TESTBED_PeerCreateMessage))
1611 GNUNET_break (0); /* We need configuration */
1612 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1615 msg = (const struct GNUNET_TESTBED_PeerCreateMessage *) message;
1616 host_id = ntohl (msg->host_id);
1617 peer_id = ntohl (msg->peer_id);
1618 if (UINT32_MAX == peer_id)
1620 send_operation_fail_msg (client, GNUNET_ntohll (msg->operation_id),
1621 "Cannot create peer with given ID");
1622 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1625 if (host_id == master_context->host_id)
1629 /* We are responsible for this peer */
1630 msize -= sizeof (struct GNUNET_TESTBED_PeerCreateMessage);
1631 config_size = ntohl (msg->config_size);
1632 config = GNUNET_malloc (config_size);
1633 dest_size = config_size;
1636 uncompress ((Bytef *) config, (uLongf *) & dest_size,
1637 (const Bytef *) &msg[1], (uLong) msize)))
1639 GNUNET_break (0); /* uncompression error */
1640 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1643 if (config_size != dest_size)
1645 GNUNET_break (0); /* Uncompressed config size mismatch */
1646 GNUNET_free (config);
1647 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1650 cfg = GNUNET_CONFIGURATION_create ();
1652 GNUNET_CONFIGURATION_deserialize (cfg, config, config_size, GNUNET_NO))
1654 GNUNET_break (0); /* Configuration parsing error */
1655 GNUNET_free (config);
1656 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1659 GNUNET_free (config);
1660 peer = GNUNET_malloc (sizeof (struct Peer));
1661 peer->is_remote = GNUNET_NO;
1662 peer->details.local.cfg = cfg;
1664 LOG_DEBUG ("Creating peer with id: %u\n", peer->id);
1665 peer->details.local.peer =
1666 GNUNET_TESTING_peer_configure (master_context->system,
1667 peer->details.local.cfg, peer->id,
1668 NULL /* Peer id */ ,
1670 if (NULL == peer->details.local.peer)
1672 LOG (GNUNET_ERROR_TYPE_WARNING, "Configuring peer failed: %s\n", emsg);
1676 GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1679 peer_list_add (peer);
1681 GNUNET_malloc (sizeof
1682 (struct GNUNET_TESTBED_PeerCreateSuccessEventMessage));
1683 reply->header.size =
1684 htons (sizeof (struct GNUNET_TESTBED_PeerCreateSuccessEventMessage));
1685 reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_PEERCREATESUCCESS);
1686 reply->peer_id = msg->peer_id;
1687 reply->operation_id = msg->operation_id;
1688 queue_message (client, &reply->header);
1689 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1693 /* Forward peer create request */
1694 route = find_dest_route (host_id);
1698 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1701 fo_ctxt = GNUNET_malloc (sizeof (struct ForwardedOperationContext));
1702 GNUNET_SERVER_client_keep (client);
1703 fo_ctxt->client = client;
1704 fo_ctxt->operation_id = GNUNET_ntohll (msg->operation_id);
1705 fo_ctxt->cls = slave_list[route->dest]->controller;
1707 GNUNET_TESTBED_forward_operation_msg_ (slave_list
1708 [route->dest]->controller,
1709 fo_ctxt->operation_id,
1711 peer_create_success_cb, fo_ctxt);
1712 fo_ctxt->timeout_task =
1713 GNUNET_SCHEDULER_add_delayed (TIMEOUT, &peer_create_forward_timeout,
1716 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1721 * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_DESTROYPEER messages
1724 * @param client identification of the client
1725 * @param message the actual message
1728 handle_peer_destroy (void *cls, struct GNUNET_SERVER_Client *client,
1729 const struct GNUNET_MessageHeader *message)
1731 const struct GNUNET_TESTBED_PeerDestroyMessage *msg;
1732 struct ForwardedOperationContext *fopc;
1736 msg = (const struct GNUNET_TESTBED_PeerDestroyMessage *) message;
1737 peer_id = ntohl (msg->peer_id);
1738 LOG_DEBUG ("Received peer destory on peer: %u and operation id: %ul\n",
1739 peer_id, GNUNET_ntohll (msg->operation_id));
1740 if ((peer_list_size <= peer_id) || (NULL == peer_list[peer_id]))
1742 LOG (GNUNET_ERROR_TYPE_ERROR,
1743 "Asked to destroy a non existent peer with id: %u\n", peer_id);
1744 send_operation_fail_msg (client, GNUNET_ntohll (msg->operation_id),
1745 "Peer doesn't exist");
1746 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1749 peer = peer_list[peer_id];
1750 if (GNUNET_YES == peer->is_remote)
1752 /* Forward the destory message to sub controller */
1753 fopc = GNUNET_malloc (sizeof (struct ForwardedOperationContext));
1754 GNUNET_SERVER_client_keep (client);
1755 fopc->client = client;
1756 fopc->operation_id = GNUNET_ntohll (msg->operation_id);
1758 GNUNET_TESTBED_forward_operation_msg_ (peer->details.remote.controller,
1759 fopc->operation_id, &msg->header,
1760 &forwarded_operation_reply_relay,
1762 fopc->timeout_task =
1763 GNUNET_SCHEDULER_add_delayed (TIMEOUT, &forwarded_operation_timeout,
1765 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1768 GNUNET_TESTING_peer_destroy (peer->details.local.peer);
1769 GNUNET_CONFIGURATION_destroy (peer->details.local.cfg);
1770 peer_list_remove (peer);
1772 send_operation_success_msg (client, GNUNET_ntohll (msg->operation_id));
1773 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1778 * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_DESTROYPEER messages
1781 * @param client identification of the client
1782 * @param message the actual message
1785 handle_peer_start (void *cls, struct GNUNET_SERVER_Client *client,
1786 const struct GNUNET_MessageHeader *message)
1788 const struct GNUNET_TESTBED_PeerStartMessage *msg;
1789 struct GNUNET_TESTBED_PeerEventMessage *reply;
1790 struct ForwardedOperationContext *fopc;
1794 msg = (const struct GNUNET_TESTBED_PeerStartMessage *) message;
1795 peer_id = ntohl (msg->peer_id);
1796 if ((peer_id >= peer_list_size) || (NULL == peer_list[peer_id]))
1799 LOG (GNUNET_ERROR_TYPE_ERROR,
1800 "Asked to start a non existent peer with id: %u\n", peer_id);
1801 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1804 peer = peer_list[peer_id];
1805 if (GNUNET_YES == peer->is_remote)
1807 fopc = GNUNET_malloc (sizeof (struct ForwardedOperationContext));
1808 GNUNET_SERVER_client_keep (client);
1809 fopc->client = client;
1810 fopc->operation_id = GNUNET_ntohll (msg->operation_id);
1812 GNUNET_TESTBED_forward_operation_msg_ (peer->details.remote.controller,
1813 fopc->operation_id, &msg->header,
1814 &forwarded_operation_reply_relay,
1816 fopc->timeout_task =
1817 GNUNET_SCHEDULER_add_delayed (TIMEOUT, &forwarded_operation_timeout,
1819 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1822 if (GNUNET_OK != GNUNET_TESTING_peer_start (peer->details.local.peer))
1824 send_operation_fail_msg (client, GNUNET_ntohll (msg->operation_id),
1826 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1829 reply = GNUNET_malloc (sizeof (struct GNUNET_TESTBED_PeerEventMessage));
1830 reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_PEEREVENT);
1831 reply->header.size = htons (sizeof (struct GNUNET_TESTBED_PeerEventMessage));
1832 reply->event_type = htonl (GNUNET_TESTBED_ET_PEER_START);
1833 reply->host_id = htonl (master_context->host_id);
1834 reply->peer_id = msg->peer_id;
1835 reply->operation_id = msg->operation_id;
1836 queue_message (client, &reply->header);
1837 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1842 * Message handler for GNUNET_MESSAGE_TYPE_TESTBED_DESTROYPEER messages
1845 * @param client identification of the client
1846 * @param message the actual message
1849 handle_peer_stop (void *cls, struct GNUNET_SERVER_Client *client,
1850 const struct GNUNET_MessageHeader *message)
1852 const struct GNUNET_TESTBED_PeerStopMessage *msg;
1853 struct GNUNET_TESTBED_PeerEventMessage *reply;
1854 struct ForwardedOperationContext *fopc;
1858 msg = (const struct GNUNET_TESTBED_PeerStopMessage *) message;
1859 peer_id = ntohl (msg->peer_id);
1860 if ((peer_id >= peer_list_size) || (NULL == peer_list[peer_id]))
1862 send_operation_fail_msg (client, GNUNET_ntohll (msg->operation_id),
1864 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1867 peer = peer_list[peer_id];
1868 if (GNUNET_YES == peer->is_remote)
1870 fopc = GNUNET_malloc (sizeof (struct ForwardedOperationContext));
1871 GNUNET_SERVER_client_keep (client);
1872 fopc->client = client;
1873 fopc->operation_id = GNUNET_ntohll (msg->operation_id);
1875 GNUNET_TESTBED_forward_operation_msg_ (peer->details.remote.controller,
1876 fopc->operation_id, &msg->header,
1877 &forwarded_operation_reply_relay,
1879 fopc->timeout_task =
1880 GNUNET_SCHEDULER_add_delayed (TIMEOUT, &forwarded_operation_timeout,
1882 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1885 if (GNUNET_OK != GNUNET_TESTING_peer_stop (peer->details.local.peer))
1887 send_operation_fail_msg (client, GNUNET_ntohll (msg->operation_id),
1888 "Peer not running");
1889 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1892 reply = GNUNET_malloc (sizeof (struct GNUNET_TESTBED_PeerEventMessage));
1893 reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_PEEREVENT);
1894 reply->header.size = htons (sizeof (struct GNUNET_TESTBED_PeerEventMessage));
1895 reply->event_type = htonl (GNUNET_TESTBED_ET_PEER_STOP);
1896 reply->host_id = htonl (master_context->host_id);
1897 reply->peer_id = msg->peer_id;
1898 reply->operation_id = msg->operation_id;
1899 queue_message (client, &reply->header);
1900 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1905 * Handler for GNUNET_MESSAGE_TYPE_TESTBED_GETPEERCONFIG messages
1908 * @param client identification of the client
1909 * @param message the actual message
1912 handle_peer_get_config (void *cls, struct GNUNET_SERVER_Client *client,
1913 const struct GNUNET_MessageHeader *message)
1915 const struct GNUNET_TESTBED_PeerGetConfigurationMessage *msg;
1916 struct GNUNET_TESTBED_PeerConfigurationInformationMessage *reply;
1925 msg = (const struct GNUNET_TESTBED_PeerGetConfigurationMessage *) message;
1926 peer_id = ntohl (msg->peer_id);
1927 if ((peer_id >= peer_list_size) || (NULL == peer_list[peer_id]))
1929 send_operation_fail_msg (client, GNUNET_ntohll (msg->operation_id),
1931 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1934 peer = peer_list[peer_id];
1935 if (GNUNET_YES == peer->is_remote)
1937 /* FIXME: forward to sub controller */
1939 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1943 GNUNET_CONFIGURATION_serialize (peer_list[peer_id]->details.local.cfg,
1945 xc_size = GNUNET_TESTBED_compress_config_ (config, c_size, &xconfig);
1946 GNUNET_free (config);
1949 sizeof (struct GNUNET_TESTBED_PeerConfigurationInformationMessage);
1950 reply = GNUNET_realloc (xconfig, msize);
1951 (void) memmove (&reply[1], reply, xc_size);
1952 reply->header.size = htons (msize);
1953 reply->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_PEERCONFIG);
1954 reply->peer_id = msg->peer_id;
1955 reply->operation_id = msg->operation_id;
1956 GNUNET_TESTING_peer_get_identity (peer_list[peer_id]->details.local.peer,
1957 &reply->peer_identity);
1958 reply->config_size = htons ((uint16_t) c_size);
1959 queue_message (client, &reply->header);
1960 GNUNET_SERVER_receive_done (client, GNUNET_OK);
1965 * Task for cleaing up overlay connect context structure
1967 * @param cls the overlay connect context
1968 * @param tc the task context
1971 occ_cleanup (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
1973 struct OverlayConnectContext *occ = cls;
1975 GNUNET_free_non_null (occ->emsg);
1976 GNUNET_free_non_null (occ->hello);
1977 if (GNUNET_SCHEDULER_NO_TASK != occ->send_hello_task)
1978 GNUNET_SCHEDULER_cancel (occ->send_hello_task);
1979 if (NULL != occ->ch)
1980 GNUNET_CORE_disconnect (occ->ch);
1981 if (NULL != occ->ghh)
1982 GNUNET_TRANSPORT_get_hello_cancel (occ->ghh);
1983 if (NULL != occ->p1th)
1984 GNUNET_TRANSPORT_disconnect (occ->p1th);
1985 if (NULL != occ->p2th)
1986 GNUNET_TRANSPORT_disconnect (occ->p2th);
1992 * Task which will be run when overlay connect request has been timed out
1994 * @param cls the OverlayConnectContext
1995 * @param tc the TaskContext
1998 timeout_overlay_connect (void *cls,
1999 const struct GNUNET_SCHEDULER_TaskContext *tc)
2001 struct OverlayConnectContext *occ = cls;
2003 occ->timeout_task = GNUNET_SCHEDULER_NO_TASK;
2004 send_operation_fail_msg (occ->client, occ->op_id, occ->emsg);
2005 GNUNET_SERVER_client_drop (occ->client);
2006 occ_cleanup (occ, tc);
2012 * Function called to notify transport users that another
2013 * peer connected to us.
2015 * @param cls closure
2016 * @param new_peer the peer that connected
2017 * @param ats performance data
2018 * @param ats_count number of entries in ats (excluding 0-termination)
2021 overlay_connect_notify (void *cls, const struct GNUNET_PeerIdentity *new_peer,
2022 const struct GNUNET_ATS_Information *ats,
2023 unsigned int ats_count)
2025 struct OverlayConnectContext *occ = cls;
2026 struct GNUNET_TESTBED_ConnectionEventMessage *msg;
2028 char *other_peer_str;
2030 LOG_DEBUG ("Overlay connect notify\n");
2032 memcmp (new_peer, &occ->peer_identity,
2033 sizeof (struct GNUNET_PeerIdentity)))
2035 new_peer_str = GNUNET_strdup (GNUNET_i2s (new_peer));
2036 other_peer_str = GNUNET_strdup (GNUNET_i2s (&occ->other_peer_identity));
2038 memcmp (new_peer, &occ->other_peer_identity,
2039 sizeof (struct GNUNET_PeerIdentity)))
2041 LOG_DEBUG ("Unexpected peer %4s connected to peer %4s\n", new_peer_str,
2043 GNUNET_free (new_peer_str);
2044 GNUNET_free (other_peer_str);
2047 GNUNET_free (new_peer_str);
2048 LOG_DEBUG ("Peer %4s connected to peer %4s\n", other_peer_str,
2049 GNUNET_i2s (&occ->peer_identity));
2050 GNUNET_free (other_peer_str);
2051 if (GNUNET_SCHEDULER_NO_TASK != occ->send_hello_task)
2053 GNUNET_SCHEDULER_cancel (occ->send_hello_task);
2054 occ->send_hello_task = GNUNET_SCHEDULER_NO_TASK;
2056 GNUNET_assert (GNUNET_SCHEDULER_NO_TASK != occ->timeout_task);
2057 GNUNET_SCHEDULER_cancel (occ->timeout_task);
2058 occ->timeout_task = GNUNET_SCHEDULER_NO_TASK;
2059 GNUNET_free_non_null (occ->emsg);
2061 GNUNET_TRANSPORT_disconnect (occ->p1th);
2063 /* Peer 1 has connected connect to peer2 - now send overlay connect success message */
2064 LOG_DEBUG ("Peers connected - Sending overlay connect success\n");
2065 msg = GNUNET_malloc (sizeof (struct GNUNET_TESTBED_ConnectionEventMessage));
2067 htons (sizeof (struct GNUNET_TESTBED_ConnectionEventMessage));
2068 msg->header.type = htons (GNUNET_MESSAGE_TYPE_TESTBED_PEERCONEVENT);
2069 msg->event_type = htonl (GNUNET_TESTBED_ET_CONNECT);
2070 msg->peer1 = htonl (occ->peer->id);
2071 msg->peer2 = htonl (occ->other_peer->id);
2072 msg->operation_id = GNUNET_htonll (occ->op_id);
2073 queue_message (occ->client, &msg->header);
2074 GNUNET_SERVER_client_drop (occ->client);
2075 GNUNET_SCHEDULER_add_now (&occ_cleanup, occ);
2080 send_hello (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
2082 struct OverlayConnectContext *occ = cls;
2083 char *other_peer_str;
2085 occ->send_hello_task = GNUNET_SCHEDULER_NO_TASK;
2086 if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
2088 GNUNET_assert (NULL != occ->hello);
2089 other_peer_str = GNUNET_strdup (GNUNET_i2s (&occ->other_peer_identity));
2090 LOG_DEBUG ("Offering HELLO of %s to %s\n", other_peer_str,
2091 GNUNET_i2s (&occ->peer_identity));
2092 GNUNET_free (other_peer_str);
2093 GNUNET_TRANSPORT_offer_hello (occ->p1th, occ->hello, NULL, NULL);
2094 GNUNET_TRANSPORT_try_connect (occ->p1th, &occ->other_peer_identity);
2095 occ->send_hello_task =
2096 GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS, &send_hello, occ);
2100 * Test for checking whether HELLO message is empty
2102 * @param cls empty flag to set
2103 * @param address the HELLO
2104 * @param expiration expiration of the HELLO
2108 test_address (void *cls, const struct GNUNET_HELLO_Address *address,
2109 struct GNUNET_TIME_Absolute expiration)
2119 * Function called whenever there is an update to the
2120 * HELLO of peers in the OverlayConnectClosure
2122 * @param cls closure
2123 * @param hello our updated HELLO
2126 hello_update_cb (void *cls, const struct GNUNET_MessageHeader *hello)
2128 struct OverlayConnectContext *occ = cls;
2132 msize = ntohs (hello->size);
2135 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
2136 "HELLO message of peer %s is of size 0\n",
2137 &occ->other_peer_identity);
2141 (void) GNUNET_HELLO_iterate_addresses ((const struct GNUNET_HELLO_Message *)
2142 hello, GNUNET_NO, &test_address,
2144 if (GNUNET_YES == empty)
2146 LOG_DEBUG ("HELLO of %s is empty\n",
2147 GNUNET_i2s (&occ->other_peer_identity));
2150 LOG_DEBUG ("Received HELLO of %s\n", GNUNET_i2s (&occ->other_peer_identity));
2151 occ->hello = GNUNET_malloc (msize);
2152 memcpy (occ->hello, hello, msize);
2153 GNUNET_TRANSPORT_get_hello_cancel (occ->ghh);
2155 GNUNET_TRANSPORT_disconnect (occ->p2th);
2157 GNUNET_free_non_null (occ->emsg);
2158 occ->emsg = GNUNET_strdup ("Timeout while offering HELLO to other peer");
2159 occ->send_hello_task = GNUNET_SCHEDULER_add_now (&send_hello, occ);
2164 * Function called after GNUNET_CORE_connect has succeeded (or failed
2165 * for good). Note that the private key of the peer is intentionally
2166 * not exposed here; if you need it, your process should try to read
2167 * the private key file directly (which should work if you are
2170 * @param cls closure
2171 * @param server handle to the server, NULL if we failed
2172 * @param my_identity ID of this peer, NULL if we failed
2175 core_startup_cb (void *cls, struct GNUNET_CORE_Handle *server,
2176 const struct GNUNET_PeerIdentity *my_identity)
2178 struct OverlayConnectContext *occ = cls;
2180 GNUNET_free_non_null (occ->emsg);
2182 memcpy (&occ->peer_identity, my_identity,
2183 sizeof (struct GNUNET_PeerIdentity));
2185 GNUNET_TRANSPORT_connect (occ->peer->details.local.cfg,
2186 &occ->peer_identity, NULL, NULL, NULL, NULL);
2187 /* Connect to the transport of 2nd peer and get its HELLO message */
2188 GNUNET_TESTING_peer_get_identity (occ->other_peer->details.local.peer,
2189 &occ->other_peer_identity);
2191 GNUNET_TRANSPORT_connect (occ->other_peer->details.local.cfg,
2192 &occ->other_peer_identity, NULL, NULL, NULL,
2194 if ((NULL == occ->p1th) || (NULL == occ->p2th))
2196 occ->emsg = GNUNET_strdup ("Cannot connect to TRANSPORTs of peers");
2197 GNUNET_SCHEDULER_cancel (occ->timeout_task);
2198 occ->timeout_task = GNUNET_SCHEDULER_add_now (&timeout_overlay_connect,
2202 LOG_DEBUG ("Acquiring HELLO of peer %s\n",
2203 GNUNET_i2s (&occ->other_peer_identity));
2204 occ->emsg = GNUNET_strdup ("Timeout while acquiring HELLO message");
2205 occ->ghh = GNUNET_TRANSPORT_get_hello (occ->p2th, &hello_update_cb, occ);
2210 * Handler for GNUNET_MESSAGE_TYPE_TESTBED_OLCONNECT messages
2213 * @param client identification of the client
2214 * @param message the actual message
2217 handle_overlay_connect (void *cls, struct GNUNET_SERVER_Client *client,
2218 const struct GNUNET_MessageHeader *message)
2220 const struct GNUNET_TESTBED_OverlayConnectMessage *msg;
2221 struct OverlayConnectContext *occ;
2223 struct GNUNET_CORE_MessageHandler no_handlers[] = {
2229 msg = (const struct GNUNET_TESTBED_OverlayConnectMessage *) message;
2230 p1 = ntohl (msg->peer1);
2231 p2 = ntohl (msg->peer2);
2232 GNUNET_assert (p1 < peer_list_size);
2233 GNUNET_assert (NULL != peer_list[p1]);
2234 GNUNET_assert (p2 < peer_list_size);
2235 GNUNET_assert (NULL != peer_list[p2]);
2236 /* FIXME: Add cases where we have to forward overlay connect message to sub
2238 occ = GNUNET_malloc (sizeof (struct OverlayConnectContext));
2239 GNUNET_SERVER_client_keep (client);
2240 occ->client = client;
2241 occ->state = OCC_STATE_INIT;
2242 occ->peer = peer_list[p1];
2243 occ->other_peer = peer_list[p2];
2244 occ->op_id = GNUNET_ntohll (msg->operation_id);
2246 GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
2247 (GNUNET_TIME_UNIT_SECONDS, 30),
2248 &timeout_overlay_connect, occ);
2249 /* Connect to the core of 1st peer and wait for the 2nd peer to connect */
2250 occ->emsg = GNUNET_strdup ("Timeout while connecting to CORE");
2252 GNUNET_CORE_connect (occ->peer->details.local.cfg, occ, &core_startup_cb,
2253 &overlay_connect_notify, NULL, NULL, GNUNET_NO, NULL,
2254 GNUNET_NO, no_handlers);
2255 GNUNET_SERVER_receive_done (client, GNUNET_OK);
2260 * Iterator over hash map entries.
2262 * @param cls closure
2263 * @param key current key code
2264 * @param value value in the hash map
2265 * @return GNUNET_YES if we should continue to
2270 ss_map_free_iterator (void *cls, const struct GNUNET_HashCode *key, void *value)
2272 struct SharedService *ss = value;
2274 GNUNET_assert (GNUNET_YES ==
2275 GNUNET_CONTAINER_multihashmap_remove (ss_map, key, value));
2276 GNUNET_free (ss->name);
2283 * Task to clean up and shutdown nicely
2286 * @param tc the TaskContext from scheduler
2289 shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
2291 struct LCFContextQueue *lcfq;
2294 shutdown_task_id = GNUNET_SCHEDULER_NO_TASK;
2295 LOG (GNUNET_ERROR_TYPE_DEBUG, "Shutting down testbed service\n");
2296 (void) GNUNET_CONTAINER_multihashmap_iterate (ss_map, &ss_map_free_iterator,
2298 GNUNET_CONTAINER_multihashmap_destroy (ss_map);
2299 if (NULL != lcfq_head)
2301 if (GNUNET_SCHEDULER_NO_TASK != lcf_proc_task_id)
2303 GNUNET_SCHEDULER_cancel (lcf_proc_task_id);
2304 lcf_proc_task_id = GNUNET_SCHEDULER_NO_TASK;
2306 if (NULL != lcfq_head->lcf->rhandle)
2307 GNUNET_TESTBED_cancel_registration (lcfq_head->lcf->rhandle);
2309 GNUNET_assert (GNUNET_SCHEDULER_NO_TASK == lcf_proc_task_id);
2310 for (lcfq = lcfq_head; NULL != lcfq; lcfq = lcfq_head)
2312 GNUNET_free (lcfq->lcf->msg);
2313 GNUNET_free (lcfq->lcf);
2314 GNUNET_CONTAINER_DLL_remove (lcfq_head, lcfq_tail, lcfq);
2317 /* Clear peer list */
2318 for (id = 0; id < peer_list_size; id++)
2319 if (NULL != peer_list[id])
2321 if (GNUNET_NO == peer_list[id]->is_remote)
2323 GNUNET_TESTING_peer_destroy (peer_list[id]->details.local.peer);
2324 GNUNET_CONFIGURATION_destroy (peer_list[id]->details.local.cfg);
2326 GNUNET_free (peer_list[id]);
2328 GNUNET_free_non_null (peer_list);
2329 /* Clear host list */
2330 for (id = 0; id < host_list_size; id++)
2331 if (NULL != host_list[id])
2332 GNUNET_TESTBED_host_destroy (host_list[id]);
2333 GNUNET_free_non_null (host_list);
2334 /* Clear route list */
2335 for (id = 0; id < route_list_size; id++)
2336 if (NULL != route_list[id])
2337 GNUNET_free (route_list[id]);
2338 GNUNET_free_non_null (route_list);
2339 /* Clear slave_list */
2340 for (id = 0; id < slave_list_size; id++)
2341 if (NULL != slave_list[id])
2343 if (NULL != slave_list[id]->controller)
2344 GNUNET_TESTBED_controller_disconnect (slave_list[id]->controller);
2345 if (NULL != slave_list[id]->controller_proc)
2346 GNUNET_TESTBED_controller_stop (slave_list[id]->controller_proc);
2348 if (NULL != master_context)
2350 GNUNET_free_non_null (master_context->master_ip);
2351 if (NULL != master_context->system)
2352 GNUNET_TESTING_system_destroy (master_context->system, GNUNET_YES);
2353 GNUNET_free (master_context);
2354 master_context = NULL;
2360 * Callback for client disconnect
2363 * @param client the client which has disconnected
2366 client_disconnect_cb (void *cls, struct GNUNET_SERVER_Client *client)
2368 if (NULL == master_context)
2370 if (client == master_context->client)
2372 LOG (GNUNET_ERROR_TYPE_DEBUG, "Master client disconnected\n");
2373 GNUNET_SERVER_client_drop (client);
2374 /* should not be needed as we're terminated by failure to read
2375 * from stdin, but if stdin fails for some reason, this shouldn't
2376 * hurt for now --- might need to revise this later if we ever
2377 * decide that master connections might be temporarily down
2378 * for some reason */
2379 //GNUNET_SCHEDULER_shutdown ();
2387 * @param cls closure
2388 * @param server the initialized server
2389 * @param cfg configuration to use
2392 testbed_run (void *cls, struct GNUNET_SERVER_Handle *server,
2393 const struct GNUNET_CONFIGURATION_Handle *cfg)
2395 static const struct GNUNET_SERVER_MessageHandler message_handlers[] = {
2396 {&handle_init, NULL, GNUNET_MESSAGE_TYPE_TESTBED_INIT, 0},
2397 {&handle_add_host, NULL, GNUNET_MESSAGE_TYPE_TESTBED_ADDHOST, 0},
2398 {&handle_configure_shared_service, NULL,
2399 GNUNET_MESSAGE_TYPE_TESTBED_SERVICESHARE, 0},
2400 {&handle_link_controllers, NULL,
2401 GNUNET_MESSAGE_TYPE_TESTBED_LCONTROLLERS, 0},
2402 {&handle_peer_create, NULL, GNUNET_MESSAGE_TYPE_TESTBED_CREATEPEER, 0},
2403 {&handle_peer_destroy, NULL, GNUNET_MESSAGE_TYPE_TESTBED_DESTROYPEER,
2404 sizeof (struct GNUNET_TESTBED_PeerDestroyMessage)},
2405 {&handle_peer_start, NULL, GNUNET_MESSAGE_TYPE_TESTBED_STARTPEER,
2406 sizeof (struct GNUNET_TESTBED_PeerStartMessage)},
2407 {&handle_peer_stop, NULL, GNUNET_MESSAGE_TYPE_TESTBED_STOPPEER,
2408 sizeof (struct GNUNET_TESTBED_PeerStopMessage)},
2409 {&handle_peer_get_config, NULL, GNUNET_MESSAGE_TYPE_TESTBED_GETPEERCONFIG,
2410 sizeof (struct GNUNET_TESTBED_PeerGetConfigurationMessage)},
2411 {&handle_overlay_connect, NULL, GNUNET_MESSAGE_TYPE_TESTBED_OLCONNECT,
2412 sizeof (struct GNUNET_TESTBED_OverlayConnectMessage)},
2416 GNUNET_SERVER_add_handlers (server, message_handlers);
2417 GNUNET_SERVER_disconnect_notify (server, &client_disconnect_cb, NULL);
2418 ss_map = GNUNET_CONTAINER_multihashmap_create (5);
2420 GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
2421 &shutdown_task, NULL);
2422 LOG_DEBUG ("Testbed startup complete\n");
2427 * The starting point of execution
2430 main (int argc, char *const *argv)
2432 //sleep (15); /* Debugging */
2433 return (GNUNET_OK ==
2434 GNUNET_SERVICE_run (argc, argv, "testbed", GNUNET_SERVICE_OPTION_NONE,
2435 &testbed_run, NULL)) ? 0 : 1;