2 This file is part of GNUnet
3 (C) 2008--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 3, 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/testbed_api_testbed.c
23 * @brief high-level testbed management
24 * @author Christian Grothoff
25 * @author Sree Harsha Totakura
29 #include "gnunet_util_lib.h"
30 #include "gnunet_testbed_service.h"
31 #include "testbed_api_peers.h"
32 #include "testbed_api_hosts.h"
33 #include "testbed_api_topology.h"
36 * Generic loggins shorthand
38 #define LOG(kind,...) \
39 GNUNET_log_from (kind, "testbed-api-testbed", __VA_ARGS__)
42 * Debug logging shortcut
45 LOG (GNUNET_ERROR_TYPE_DEBUG, __VA_ARGS__)
53 * The testbed operation handle
55 struct GNUNET_TESTBED_Operation *op;
58 * Context information for GNUNET_TESTBED_run()
60 struct RunContext *rc;
68 * The next pointer for DLL
70 struct DLLOperation *next;
73 * The prev pointer for DLL
75 struct DLLOperation *prev;
80 * States of RunContext
90 * Controllers on given hosts started and linked
95 * The testbed run is ready and the master callback can be called now. At this
96 * time the peers are all started and if a topology is provided in the
97 * configuration the topology would have been attempted
107 * Peers are destroyed
119 * The controller handle
121 struct GNUNET_TESTBED_Controller *c;
124 * The configuration of the controller. This is based on the cfg given to the
125 * function GNUNET_TESTBED_run(). We also use this config as a template while
128 struct GNUNET_CONFIGURATION_Handle *cfg;
131 * Handle to the host on which the controller runs
133 struct GNUNET_TESTBED_Host *h;
136 * The handle to the controller process
138 struct GNUNET_TESTBED_ControllerProc *cproc;
141 * The callback to use as controller callback
143 GNUNET_TESTBED_ControllerCallback cc;
146 * The pointer to the controller callback
151 * The trusted IP string
156 * TestMaster callback to call when testbed initialization is done
158 GNUNET_TESTBED_TestMaster test_master;
161 * The closure for the TestMaster callback
163 void *test_master_cls;
166 * The head element of DLL operations
168 struct DLLOperation *dll_op_head;
171 * The tail element of DLL operations
173 struct DLLOperation *dll_op_tail;
176 * An array of hosts loaded from the hostkeys file
178 struct GNUNET_TESTBED_Host **hosts;
181 * The handle for whether a host is habitable or not
183 struct GNUNET_TESTBED_HostHabitableCheckHandle **hc_handles;
186 * Array of peers which we create
188 struct GNUNET_TESTBED_Peer **peers;
191 * The topology generation operation. Will be null if no topology is set in
194 struct GNUNET_TESTBED_Operation *topology_operation;
197 * The file containing topology data. Only used if the topology is set to 'FROM_FILE'
202 * Host registration handle
204 struct GNUNET_TESTBED_HostRegistrationHandle *reg_handle;
207 * Profiling start time
209 struct GNUNET_TIME_Absolute pstart_time;
212 * Host registration task
214 GNUNET_SCHEDULER_TaskIdentifier register_hosts_task;
217 * Task to be run while shutting down
219 GNUNET_SCHEDULER_TaskIdentifier shutdown_run_task;
222 * The event mask for the controller
227 * State of this context
232 * The topology which has to be achieved with the peers started in this context
234 enum GNUNET_TESTBED_TopologyOption topology;
237 * Have we already shutdown
242 * Number of hosts in the given host file
244 unsigned int num_hosts;
247 * Number of registered hosts. Also used as a counter while checking
248 * habitabillity of hosts
250 unsigned int reg_hosts;
253 * Current peer count for an operation; Set this to 0 and increment for each
254 * successful operation on a peer
256 unsigned int peer_count;
259 * number of peers to start
261 unsigned int num_peers;
264 * counter to count overlay connect attempts. This counter includes both
265 * successful and failed overlay connects
267 unsigned int oc_count;
270 * Expected overlay connects. Should be zero if no topology is relavant
275 * Number of random links to established
277 unsigned int random_links;
283 * Function to return the string representation of the duration between current
284 * time and `pstart_time' in `RunContext'
286 * @param rc the RunContext
287 * @return the representation string; this is NOT reentrant
290 prof_time (struct RunContext *rc)
292 struct GNUNET_TIME_Relative ptime;
294 ptime = GNUNET_TIME_absolute_get_duration (rc->pstart_time);
295 return GNUNET_STRINGS_relative_time_to_string (ptime, GNUNET_YES);
300 * Task for starting peers
302 * @param cls the RunHandle
303 * @param tc the task context from scheduler
306 start_peers_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
308 struct RunContext *rc = cls;
309 struct DLLOperation *dll_op;
312 DEBUG ("Starting Peers\n");
313 rc->pstart_time = GNUNET_TIME_absolute_get ();
314 for (peer = 0; peer < rc->num_peers; peer++)
316 dll_op = GNUNET_malloc (sizeof (struct DLLOperation));
317 dll_op->op = GNUNET_TESTBED_peer_start (NULL, rc->peers[peer], NULL, NULL);
318 dll_op->cls = rc->peers[peer];
319 GNUNET_CONTAINER_DLL_insert_tail (rc->dll_op_head, rc->dll_op_tail, dll_op);
326 * Functions of this signature are called when a peer has been successfully
329 * @param cls the closure from GNUNET_TESTBED_peer_create()
330 * @param peer the handle for the created peer; NULL on any error during
332 * @param emsg NULL if peer is not NULL; else MAY contain the error description
335 peer_create_cb (void *cls, struct GNUNET_TESTBED_Peer *peer, const char *emsg)
337 struct DLLOperation *dll_op = cls;
338 struct RunContext *rc;
340 GNUNET_assert (NULL != dll_op);
342 GNUNET_assert (NULL != rc);
343 GNUNET_CONTAINER_DLL_remove (rc->dll_op_head, rc->dll_op_tail, dll_op);
344 GNUNET_TESTBED_operation_done (dll_op->op);
345 GNUNET_free (dll_op);
349 LOG (GNUNET_ERROR_TYPE_WARNING, "Error while creating a peer: %s\n",
351 /* FIXME: GNUNET_TESTBED_shutdown_run()? */
354 rc->peers[rc->peer_count] = peer;
356 if (rc->peer_count < rc->num_peers)
358 DEBUG ("%u peers created in %s\n", rc->num_peers, prof_time (rc));
359 GNUNET_SCHEDULER_add_now (&start_peers_task, rc);
364 * Assuming all peers have been destroyed cleanup run handle
366 * @param cls the run handle
367 * @param tc the task context from scheduler
370 cleanup_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
372 struct RunContext *rc = cls;
373 struct DLLOperation *dll_op;
376 GNUNET_assert (GNUNET_SCHEDULER_NO_TASK == rc->register_hosts_task);
377 GNUNET_assert (NULL == rc->reg_handle);
378 GNUNET_assert (NULL == rc->peers);
379 GNUNET_assert (NULL == rc->hc_handles);
380 GNUNET_assert (RC_PEERS_DESTROYED == rc->state);
381 if (NULL != rc->dll_op_head)
382 { /* cancel our pending operations */
383 while (NULL != (dll_op = rc->dll_op_head))
385 GNUNET_TESTBED_operation_done (dll_op->op);
386 GNUNET_CONTAINER_DLL_remove (rc->dll_op_head, rc->dll_op_tail, dll_op);
387 GNUNET_free (dll_op);
391 GNUNET_TESTBED_controller_disconnect (rc->c);
392 if (NULL != rc->cproc)
393 GNUNET_TESTBED_controller_stop (rc->cproc);
395 GNUNET_TESTBED_host_destroy (rc->h);
396 for (hid = 0; hid < rc->num_hosts; hid++)
397 GNUNET_TESTBED_host_destroy (rc->hosts[hid]);
398 GNUNET_free_non_null (rc->hosts);
400 GNUNET_CONFIGURATION_destroy (rc->cfg);
401 GNUNET_free_non_null (rc->topo_file);
402 GNUNET_free_non_null (rc->trusted_ip);
408 * Stops the testbed run and releases any used resources
410 * @param cls the tesbed run handle
411 * @param tc the task context from scheduler
414 shutdown_run (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
418 * Function to shutdown now
420 * @param rc the RunContext
423 shutdown_now (struct RunContext *rc)
425 if (GNUNET_YES == rc->shutdown)
427 if (GNUNET_SCHEDULER_NO_TASK != rc->shutdown_run_task)
428 GNUNET_SCHEDULER_cancel (rc->shutdown_run_task);
429 rc->shutdown_run_task = GNUNET_SCHEDULER_add_now (&shutdown_run, rc);
434 * Stops the testbed run and releases any used resources
436 * @param cls the tesbed run handle
437 * @param tc the task context from scheduler
440 shutdown_run (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
442 struct RunContext *rc = cls;
443 struct DLLOperation *dll_op;
444 int all_peers_destroyed;
448 GNUNET_assert (GNUNET_SCHEDULER_NO_TASK != rc->shutdown_run_task);
449 rc->shutdown_run_task = GNUNET_SCHEDULER_NO_TASK;
450 GNUNET_assert (GNUNET_NO == rc->shutdown);
451 rc->shutdown = GNUNET_YES;
452 if (NULL != rc->hc_handles)
454 for (nhost = 0; nhost < rc->num_hosts; nhost++)
455 if (NULL != rc->hc_handles[nhost])
456 GNUNET_TESTBED_is_host_habitable_cancel (rc->hc_handles[nhost]);
457 GNUNET_free (rc->hc_handles);
458 rc->hc_handles = NULL;
460 /* Stop register hosts task if it is running */
461 if (GNUNET_SCHEDULER_NO_TASK != rc->register_hosts_task)
463 GNUNET_SCHEDULER_cancel (rc->register_hosts_task);
464 rc->register_hosts_task = GNUNET_SCHEDULER_NO_TASK;
466 if (NULL != rc->reg_handle)
468 GNUNET_TESTBED_cancel_registration (rc->reg_handle);
469 rc->reg_handle = NULL;
473 if (NULL != rc->peers)
475 if (NULL != rc->topology_operation)
477 GNUNET_TESTBED_operation_done (rc->topology_operation);
478 rc->topology_operation = NULL;
480 if (RC_INIT == rc->state)
481 rc->state = RC_READY; /* Even though we haven't called the master callback */
483 /* Check if some peers are stopped */
484 for (peer = 0; peer < rc->num_peers; peer++)
486 if (NULL == rc->peers[peer])
488 if (PS_STOPPED != rc->peers[peer]->state)
491 if (peer == rc->num_peers)
493 /* All peers are stopped */
494 rc->state = RC_PEERS_STOPPED;
495 all_peers_destroyed = GNUNET_YES;
496 for (peer = 0; peer < rc->num_peers; peer++)
498 if (NULL == rc->peers[peer])
500 all_peers_destroyed = GNUNET_NO;
501 dll_op = GNUNET_malloc (sizeof (struct DLLOperation));
502 dll_op->op = GNUNET_TESTBED_peer_destroy (rc->peers[peer]);
503 GNUNET_CONTAINER_DLL_insert_tail (rc->dll_op_head, rc->dll_op_tail,
506 if (all_peers_destroyed == GNUNET_NO)
508 DEBUG ("Destroying peers\n");
509 rc->pstart_time = GNUNET_TIME_absolute_get ();
513 /* Some peers are stopped */
514 DEBUG ("Stopping peers\n");
515 rc->pstart_time = GNUNET_TIME_absolute_get ();
516 for (peer = 0; peer < rc->num_peers; peer++)
518 if ((NULL == rc->peers[peer]) || (PS_STARTED != rc->peers[peer]->state))
523 dll_op = GNUNET_malloc (sizeof (struct DLLOperation));
524 dll_op->op = GNUNET_TESTBED_peer_stop (rc->peers[peer], NULL, NULL);
525 dll_op->cls = rc->peers[peer];
526 GNUNET_CONTAINER_DLL_insert_tail (rc->dll_op_head, rc->dll_op_tail,
529 if (rc->peer_count != rc->num_peers)
531 GNUNET_free (rc->peers);
535 rc->state = RC_PEERS_DESTROYED; /* No peers are present so we consider the
536 * state where all peers are destroyed */
537 GNUNET_SCHEDULER_add_now (&cleanup_task, rc);
542 * Task to call master task
544 * @param cls the run context
545 * @param tc the task context
548 call_master (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
550 struct RunContext *rc = cls;
552 if (NULL != rc->topology_operation)
554 DEBUG ("Overlay topology generated in %s\n", prof_time (rc));
555 GNUNET_TESTBED_operation_done (rc->topology_operation);
556 rc->topology_operation = NULL;
558 if (NULL != rc->test_master)
559 rc->test_master (rc->test_master_cls, rc->num_peers, rc->peers);
564 * Function to create peers
566 * @param rc the RunContext
569 create_peers (struct RunContext *rc)
571 struct DLLOperation *dll_op;
574 DEBUG ("Creating peers\n");
575 rc->pstart_time = GNUNET_TIME_absolute_get ();
577 GNUNET_malloc (sizeof (struct GNUNET_TESTBED_Peer *) * rc->num_peers);
578 GNUNET_assert (NULL != rc->c);
580 for (peer = 0; peer < rc->num_peers; peer++)
582 dll_op = GNUNET_malloc (sizeof (struct DLLOperation));
585 GNUNET_TESTBED_peer_create (rc->c,
587 rc->num_hosts) ? rc->h : rc->hosts[peer %
589 rc->cfg, peer_create_cb, dll_op);
590 GNUNET_CONTAINER_DLL_insert_tail (rc->dll_op_head, rc->dll_op_tail, dll_op);
596 * Signature of the event handler function called by the
597 * respective event controller.
600 * @param event information about the event
603 event_cb (void *cls, const struct GNUNET_TESTBED_EventInformation *event)
605 struct RunContext *rc = cls;
606 struct DLLOperation *dll_op;
607 unsigned int peer_id;
609 if (RC_INIT == rc->state)
613 case GNUNET_TESTBED_ET_OPERATION_FINISHED:
614 dll_op = event->details.operation_finished.op_cls;
615 if (NULL != event->details.operation_finished.emsg)
617 LOG (GNUNET_ERROR_TYPE_ERROR, _("Linking controllers failed. Exiting"));
622 GNUNET_assert (event->details.operation_finished.operation == dll_op->op);
623 GNUNET_CONTAINER_DLL_remove (rc->dll_op_head, rc->dll_op_tail, dll_op);
624 GNUNET_TESTBED_operation_done (dll_op->op);
625 GNUNET_free (dll_op);
626 if (rc->reg_hosts == rc->num_hosts)
628 rc->state = RC_LINKED;
638 if (NULL != rc->topology_operation)
642 case GNUNET_TESTBED_ET_OPERATION_FINISHED:
643 case GNUNET_TESTBED_ET_CONNECT:
651 if (rc->oc_count == rc->num_oc)
653 rc->state = RC_READY;
654 GNUNET_SCHEDULER_add_continuation (&call_master, rc,
655 GNUNET_SCHEDULER_REASON_PREREQ_DONE);
659 for (dll_op = rc->dll_op_head; NULL != dll_op; dll_op = dll_op->next)
661 if ((GNUNET_TESTBED_ET_OPERATION_FINISHED == event->type) &&
662 (event->details.operation_finished.operation == dll_op->op))
664 if ((GNUNET_TESTBED_ET_PEER_STOP == event->type) &&
665 (event->details.peer_stop.peer == dll_op->cls))
670 GNUNET_CONTAINER_DLL_remove (rc->dll_op_head, rc->dll_op_tail, dll_op);
671 GNUNET_TESTBED_operation_done (dll_op->op);
672 GNUNET_free (dll_op);
674 if (rc->peer_count < rc->num_peers)
680 rc->state = RC_PEERS_STOPPED;
681 DEBUG ("Peers stopped in %s\n", prof_time (rc));
682 DEBUG ("Destroying peers\n");
683 rc->pstart_time = GNUNET_TIME_absolute_get ();
685 for (peer_id = 0; peer_id < rc->num_peers; peer_id++)
687 dll_op = GNUNET_malloc (sizeof (struct DLLOperation));
688 dll_op->op = GNUNET_TESTBED_peer_destroy (rc->peers[peer_id]);
689 GNUNET_CONTAINER_DLL_insert_tail (rc->dll_op_head, rc->dll_op_tail,
693 case RC_PEERS_STOPPED:
694 rc->state = RC_PEERS_DESTROYED;
695 GNUNET_free (rc->peers);
697 DEBUG ("Peers destroyed in %s\n", prof_time (rc));
698 GNUNET_SCHEDULER_add_now (&cleanup_task, rc);
706 if ((0 != (rc->event_mask & (1LL << event->type))) && (NULL != rc->cc))
707 rc->cc (rc->cc_cls, event);
708 if (GNUNET_TESTBED_ET_PEER_START != event->type)
710 for (dll_op = rc->dll_op_head; NULL != dll_op; dll_op = dll_op->next)
711 if ((NULL != dll_op->cls) &&
712 (event->details.peer_start.peer == dll_op->cls))
714 if (NULL == dll_op) /* Not our operation */
716 GNUNET_CONTAINER_DLL_remove (rc->dll_op_head, rc->dll_op_tail, dll_op);
717 GNUNET_TESTBED_operation_done (dll_op->op);
718 GNUNET_free (dll_op);
720 if (rc->peer_count < rc->num_peers)
722 DEBUG ("%u peers started in %s\n", rc->num_peers, prof_time (rc));
723 if (GNUNET_TESTBED_TOPOLOGY_NONE != rc->topology)
725 if ((GNUNET_TESTBED_TOPOLOGY_ERDOS_RENYI == rc->topology) ||
726 (GNUNET_TESTBED_TOPOLOGY_SMALL_WORLD_RING == rc->topology) ||
727 (GNUNET_TESTBED_TOPOLOGY_SMALL_WORLD == rc->topology))
729 rc->topology_operation =
730 GNUNET_TESTBED_overlay_configure_topology (NULL, rc->num_peers,
731 rc->peers, &rc->num_oc,
734 GNUNET_TESTBED_TOPOLOGY_OPTION_END);
736 else if (GNUNET_TESTBED_TOPOLOGY_FROM_FILE == rc->topology)
738 GNUNET_assert (NULL != rc->topo_file);
739 rc->topology_operation =
740 GNUNET_TESTBED_overlay_configure_topology (NULL, rc->num_peers,
741 rc->peers, &rc->num_oc,
744 GNUNET_TESTBED_TOPOLOGY_OPTION_END);
747 rc->topology_operation =
748 GNUNET_TESTBED_overlay_configure_topology (NULL, rc->num_peers,
749 rc->peers, &rc->num_oc,
751 GNUNET_TESTBED_TOPOLOGY_OPTION_END);
752 if (NULL == rc->topology_operation)
753 LOG (GNUNET_ERROR_TYPE_WARNING,
754 "Not generating topology. Check number of peers\n");
757 DEBUG ("Creating overlay topology\n");
758 rc->pstart_time = GNUNET_TIME_absolute_get ();
762 rc->state = RC_READY;
763 GNUNET_SCHEDULER_add_continuation (&call_master, rc,
764 GNUNET_SCHEDULER_REASON_PREREQ_DONE);
769 * Task to register all hosts available in the global host list
771 * @param cls the RunContext
772 * @param tc the scheduler task context
775 register_hosts (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc);
779 * Callback which will be called to after a host registration succeeded or failed
781 * @param cls the closure
782 * @param emsg the error message; NULL if host registration is successful
785 host_registration_completion (void *cls, const char *emsg)
787 struct RunContext *rc = cls;
789 rc->reg_handle = NULL;
792 LOG (GNUNET_ERROR_TYPE_WARNING,
793 _("Host registration failed for a host. Error: %s\n"), emsg);
797 rc->register_hosts_task = GNUNET_SCHEDULER_add_now (®ister_hosts, rc);
802 * Task to register all hosts available in the global host list
804 * @param cls RunContext
805 * @param tc the scheduler task context
808 register_hosts (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
810 struct RunContext *rc = cls;
811 struct DLLOperation *dll_op;
814 rc->register_hosts_task = GNUNET_SCHEDULER_NO_TASK;
815 if (rc->reg_hosts == rc->num_hosts)
817 DEBUG ("All hosts successfully registered\n");
819 for (slave = 0; slave < rc->num_hosts; slave++)
821 dll_op = GNUNET_malloc (sizeof (struct DLLOperation));
824 GNUNET_TESTBED_controller_link (dll_op, rc->c, rc->hosts[slave],
825 rc->h, rc->cfg, GNUNET_YES);
826 GNUNET_CONTAINER_DLL_insert_tail (rc->dll_op_head, rc->dll_op_tail,
833 GNUNET_TESTBED_register_host (rc->c, rc->hosts[rc->reg_hosts],
834 host_registration_completion, rc);
840 * Callback to signal successfull startup of the controller process
842 * @param cls the closure from GNUNET_TESTBED_controller_start()
843 * @param cfg the configuration with which the controller has been started;
844 * NULL if status is not GNUNET_OK
845 * @param status GNUNET_OK if the startup is successfull; GNUNET_SYSERR if not,
846 * GNUNET_TESTBED_controller_stop() shouldn't be called in this case
849 controller_status_cb (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg,
852 struct RunContext *rc = cls;
855 if (status != GNUNET_OK)
860 GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Testbed startup failed\n");
863 GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
864 "Controller crash detected. Shutting down.\n");
870 GNUNET_CONFIGURATION_destroy (rc->cfg);
871 rc->cfg = GNUNET_CONFIGURATION_dup (cfg);
872 event_mask = rc->event_mask;
873 event_mask |= (1LL << GNUNET_TESTBED_ET_PEER_STOP);
874 event_mask |= (1LL << GNUNET_TESTBED_ET_OPERATION_FINISHED);
875 event_mask |= (1LL << GNUNET_TESTBED_ET_PEER_START);
876 if (rc->topology < GNUNET_TESTBED_TOPOLOGY_NONE)
877 event_mask |= GNUNET_TESTBED_ET_CONNECT;
879 GNUNET_TESTBED_controller_connect (rc->cfg, rc->h, event_mask, &event_cb,
881 if (0 < rc->num_hosts)
884 rc->register_hosts_task = GNUNET_SCHEDULER_add_now (®ister_hosts, rc);
887 rc->state = RC_LINKED;
893 * Callback function invoked for each interface found.
896 * @param name name of the interface (can be NULL for unknown)
897 * @param isDefault is this presumably the default interface
898 * @param addr address of this interface (can be NULL for unknown or unassigned)
899 * @param broadcast_addr the broadcast address (can be NULL for unknown or unassigned)
900 * @param netmask the network mask (can be NULL for unknown or unassigned))
901 * @param addrlen length of the address
902 * @return GNUNET_OK to continue iteration, GNUNET_SYSERR to abort
905 netint_proc (void *cls, const char *name, int isDefault,
906 const struct sockaddr *addr, const struct sockaddr *broadcast_addr,
907 const struct sockaddr *netmask, socklen_t addrlen)
909 struct RunContext *rc = cls;
910 char hostip[NI_MAXHOST];
913 if (sizeof (struct sockaddr_in) != addrlen)
914 return GNUNET_OK; /* Only consider IPv4 for now */
916 getnameinfo (addr, addrlen, hostip, NI_MAXHOST, NULL, 0, NI_NUMERICHOST))
917 GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "getnameinfo");
918 if (NULL == rc->trusted_ip)
920 rc->trusted_ip = GNUNET_strdup (hostip);
923 (void) GNUNET_asprintf (&buf, "%s; %s", rc->trusted_ip, hostip);
924 GNUNET_free (rc->trusted_ip);
925 rc->trusted_ip = buf;
931 * Callbacks of this type are called by GNUNET_TESTBED_is_host_habitable to
932 * inform whether the given host is habitable or not. The Handle returned by
933 * GNUNET_TESTBED_is_host_habitable() is invalid after this callback is called
936 * @param host the host whose status is being reported; will be NULL if the host
937 * given to GNUNET_TESTBED_is_host_habitable() is NULL
938 * @param status GNUNET_YES if it is habitable; GNUNET_NO if not
941 host_habitable_cb (void *cls, const struct GNUNET_TESTBED_Host *host,
944 struct RunContext *rc = cls;
945 struct GNUNET_TESTBED_Host **old_hosts;
948 for (nhost = 0; nhost < rc->num_hosts; nhost++)
950 if (host == rc->hosts[nhost])
953 GNUNET_assert (nhost != rc->num_hosts);
954 rc->hc_handles[nhost] = NULL;
955 if (GNUNET_NO == status)
957 if ((NULL != host) && (NULL != GNUNET_TESTBED_host_get_hostname (host)))
958 LOG (GNUNET_ERROR_TYPE_ERROR, _("Host %s cannot start testbed\n"),
959 GNUNET_TESTBED_host_get_hostname (host));
961 LOG (GNUNET_ERROR_TYPE_ERROR,
962 _("Testbed cannot be started on localhost\n"));
967 if (rc->reg_hosts < rc->num_hosts)
969 GNUNET_free (rc->hc_handles);
970 rc->hc_handles = NULL;
971 rc->h = rc->hosts[0];
973 if (0 < rc->num_hosts)
975 old_hosts = rc->hosts;
976 rc->hosts = GNUNET_malloc (sizeof (struct GNUNET_TESTBED_Host *)
978 memcpy (rc->hosts, &old_hosts[1], (sizeof (struct GNUNET_TESTBED_Host *)
980 GNUNET_free (old_hosts);
984 GNUNET_free (rc->hosts);
987 GNUNET_OS_network_interfaces_list (netint_proc, rc);
988 if (NULL == rc->trusted_ip)
989 rc->trusted_ip = GNUNET_strdup ("127.0.0.1");
991 GNUNET_TESTBED_controller_start (rc->trusted_ip, rc->h, rc->cfg,
992 &controller_status_cb, rc);
993 GNUNET_free (rc->trusted_ip);
994 rc->trusted_ip = NULL;
995 if (NULL == rc->cproc)
997 LOG (GNUNET_ERROR_TYPE_ERROR, _("Cannot start the master controller"));
1004 * Convenience method for running a testbed with
1005 * a single call. Underlay and overlay topology
1006 * are configured using the "UNDERLAY" and "OVERLAY"
1007 * options in the "[testbed]" section of the configuration\
1008 * (with possible options given in "UNDERLAY_XXX" and/or
1011 * The testbed is to be terminated using a call to
1012 * "GNUNET_SCHEDULER_shutdown".
1014 * @param host_filename name of the file with the 'hosts', NULL
1015 * to run everything on 'localhost'
1016 * @param cfg configuration to use (for testbed, controller and peers)
1017 * @param num_peers number of peers to start; FIXME: maybe put that ALSO into cfg?
1018 * @param event_mask bit mask with set of events to call 'cc' for;
1019 * or-ed values of "1LL" shifted by the
1020 * respective 'enum GNUNET_TESTBED_EventType'
1021 * (i.e. "(1LL << GNUNET_TESTBED_ET_CONNECT) || ...")
1022 * @param cc controller callback to invoke on events; This callback is called
1023 * for all peer start events even if GNUNET_TESTBED_ET_PEER_START isn't
1024 * set in the event_mask as this is the only way get access to the
1025 * handle of each peer
1026 * @param cc_cls closure for cc
1027 * @param test_master this callback will be called once the test is ready
1028 * @param test_master_cls closure for 'test_master'.
1031 GNUNET_TESTBED_run (const char *host_filename,
1032 const struct GNUNET_CONFIGURATION_Handle *cfg,
1033 unsigned int num_peers, uint64_t event_mask,
1034 GNUNET_TESTBED_ControllerCallback cc, void *cc_cls,
1035 GNUNET_TESTBED_TestMaster test_master,
1036 void *test_master_cls)
1038 struct RunContext *rc;
1040 unsigned long long random_links;
1044 GNUNET_assert (num_peers > 0);
1045 rc = GNUNET_malloc (sizeof (struct RunContext));
1046 if (NULL != host_filename)
1049 GNUNET_TESTBED_hosts_load_from_file (host_filename, &rc->hosts);
1050 if (0 == rc->num_hosts)
1052 LOG (GNUNET_ERROR_TYPE_WARNING,
1053 _("No hosts loaded. Need at least one host\n"));
1058 rc->h = GNUNET_TESTBED_host_create (NULL, NULL, 0);
1059 rc->cfg = GNUNET_CONFIGURATION_dup (cfg);
1060 rc->num_peers = num_peers;
1061 rc->event_mask = event_mask;
1063 rc->cc_cls = cc_cls;
1064 rc->test_master = test_master;
1065 rc->test_master_cls = test_master_cls;
1066 rc->state = RC_INIT;
1067 rc->topology = GNUNET_TESTBED_TOPOLOGY_NONE;
1069 GNUNET_CONFIGURATION_get_value_string (rc->cfg, "testbed",
1070 "OVERLAY_TOPOLOGY", &topology))
1072 if (GNUNET_NO == GNUNET_TESTBED_topology_get_ (&rc->topology, topology))
1074 GNUNET_log_config_invalid (GNUNET_ERROR_TYPE_ERROR, "testbed",
1077 ("Specified topology must be supported by testbed"));
1079 GNUNET_free (topology);
1081 switch (rc->topology)
1083 case GNUNET_TESTBED_TOPOLOGY_ERDOS_RENYI:
1084 case GNUNET_TESTBED_TOPOLOGY_SMALL_WORLD_RING:
1085 case GNUNET_TESTBED_TOPOLOGY_SMALL_WORLD:
1087 GNUNET_CONFIGURATION_get_value_number (rc->cfg, "testbed",
1088 "OVERLAY_RANDOM_LINKS",
1091 /* OVERLAY option RANDOM & SMALL_WORLD_RING requires OVERLAY_RANDOM_LINKS
1092 * option to be set to the number of random links to be established */
1093 GNUNET_log_config_missing (GNUNET_ERROR_TYPE_ERROR, "testbed",
1094 "OVERLAY_RANDOM_LINKS");
1097 if (random_links > UINT32_MAX)
1099 GNUNET_break (0); /* Too big number */
1102 rc->random_links = (unsigned int) random_links;
1104 case GNUNET_TESTBED_TOPOLOGY_FROM_FILE:
1106 GNUNET_CONFIGURATION_get_value_string (rc->cfg, "testbed",
1107 "OVERLAY_TOPOLOGY_FILE",
1110 GNUNET_log_config_missing (GNUNET_ERROR_TYPE_ERROR, "testbed",
1111 "OVERLAY_TOPOLOGY_FILE");
1115 /* Warn if OVERLAY_RANDOM_LINKS is present that it will be ignored */
1117 GNUNET_CONFIGURATION_have_value (rc->cfg, "testbed",
1118 "OVERLAY_RANDOM_LINKS"))
1119 LOG (GNUNET_ERROR_TYPE_WARNING,
1120 "Ignoring value of `OVERLAY_RANDOM_LINKS' in given configuration\n");
1123 if (NULL != host_filename)
1126 GNUNET_malloc (sizeof (struct GNUNET_TESTBED_HostHabitableCheckHandle *)
1128 for (nhost = 0; nhost < rc->num_hosts; nhost++)
1131 (rc->hc_handles[nhost] =
1132 GNUNET_TESTBED_is_host_habitable (rc->hosts[nhost], rc->cfg,
1133 &host_habitable_cb, rc)))
1136 for (nhost = 0; nhost < rc->num_hosts; nhost++)
1137 if (NULL != rc->hc_handles[nhost])
1138 GNUNET_TESTBED_is_host_habitable_cancel (rc->hc_handles[nhost]);
1139 GNUNET_free (rc->hc_handles);
1140 rc->hc_handles = NULL;
1147 GNUNET_TESTBED_controller_start ("127.0.0.1", rc->h, rc->cfg,
1148 &controller_status_cb, rc);
1149 rc->shutdown_run_task =
1150 GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &shutdown_run,
1156 GNUNET_TESTBED_host_destroy (rc->h);
1157 if (NULL != rc->hosts)
1159 for (hid = 0; hid < rc->num_hosts; hid++)
1160 if (NULL != rc->hosts[hid])
1161 GNUNET_TESTBED_host_destroy (rc->hosts[hid]);
1162 GNUNET_free (rc->hosts);
1168 /* end of testbed_api_testbed.c */