X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=src%2Ftestbed%2Ftestbed_api_testbed.c;h=2be565e63c182a5fe3292eb709bd8f402b516580;hb=8a2d1e6aedbf1bc95052e63ac67093b89385b0a1;hp=95f9d3a56f50c28b3b2dc04b38e136a2ed67feb7;hpb=949f21f03df9f08b0a73c3e47a798e63790edf7f;p=oweals%2Fgnunet.git diff --git a/src/testbed/testbed_api_testbed.c b/src/testbed/testbed_api_testbed.c index 95f9d3a56..2be565e63 100644 --- a/src/testbed/testbed_api_testbed.c +++ b/src/testbed/testbed_api_testbed.c @@ -1,6 +1,6 @@ /* This file is part of GNUnet - (C) 2008--2012 Christian Grothoff (and other contributing authors) + (C) 2008--2013 Christian Grothoff (and other contributing authors) GNUnet is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published @@ -28,6 +28,7 @@ #include "platform.h" #include "gnunet_util_lib.h" #include "gnunet_testbed_service.h" +#include "testbed_api.h" #include "testbed_api_peers.h" #include "testbed_api_hosts.h" #include "testbed_api_topology.h" @@ -49,10 +50,28 @@ */ #define DEFAULT_SETUP_TIMEOUT 300 + +/** + * Configuration section for testbed + */ +#define TESTBED_CONFIG_SECTION "testbed" + +/** + * Option string for the maximum number of edges a peer is permitted to have + * while generating scale free topology + */ +#define SCALE_FREE_CAP "SCALE_FREE_TOPOLOGY_CAP" + /** - * DLL of operations + * Option string for the number of edges to be established when adding a new + * node to the scale free network */ -struct DLLOperation +#define SCALE_FREE_M "SCALE_FREE_TOPOLOGY_M" + +/** + * Context information for the operation we start + */ +struct RunContextOperation { /** * The testbed operation handle @@ -62,22 +81,13 @@ struct DLLOperation /** * Context information for GNUNET_TESTBED_run() */ - struct RunContext *rc; + struct GNUNET_TESTBED_RunHandle *rc; /** * Closure */ void *cls; - /** - * The next pointer for DLL - */ - struct DLLOperation *next; - - /** - * The prev pointer for DLL - */ - struct DLLOperation *prev; }; @@ -125,10 +135,32 @@ enum State }; +/** + * Context for host compability checks + */ +struct CompatibilityCheckContext +{ + /** + * The run context + */ + struct GNUNET_TESTBED_RunHandle *rc; + + /** + * Handle for the compability check + */ + struct GNUNET_TESTBED_HostHabitableCheckHandle *h; + + /** + * Index of the host in the run context's hosts array + */ + unsigned int index; +}; + + /** * Testbed Run Handle */ -struct RunContext +struct GNUNET_TESTBED_RunHandle { /** * The controller handle @@ -178,14 +210,9 @@ struct RunContext void *test_master_cls; /** - * The head element of DLL operations + * A hashmap for operations started by us */ - struct DLLOperation *dll_op_head; - - /** - * The tail element of DLL operations - */ - struct DLLOperation *dll_op_tail; + struct GNUNET_CONTAINER_MultiHashMap32 *rcop_map; /** * An array of hosts loaded from the hostkeys file @@ -193,9 +220,9 @@ struct RunContext struct GNUNET_TESTBED_Host **hosts; /** - * The handle for whether a host is habitable or not + * Array of compatibility check contexts */ - struct GNUNET_TESTBED_HostHabitableCheckHandle **hc_handles; + struct CompatibilityCheckContext *hclist; /** * Array of peers which we create @@ -228,11 +255,6 @@ struct RunContext */ GNUNET_SCHEDULER_TaskIdentifier register_hosts_task; - /** - * Task to be run while shutting down - */ - GNUNET_SCHEDULER_TaskIdentifier shutdown_run_task; - /** * Task to be run of a timeout */ @@ -308,24 +330,138 @@ struct RunContext }; +/** + * Return a 32-bit key from a pointer + * + * @param rcop the pointer + * @return 32-bit key + */ +static uint32_t +rcop_key (void *rcop) +{ + return * ((uint32_t *) &rcop); +} + + +/** + * Context information used for finding a pointer in the rcop_map + */ +struct SearchContext +{ + /** + * The operation pointer to look for + */ + struct GNUNET_TESTBED_Operation *query; + + /** + * The Run context operation which has the operation being queried + */ + struct RunContextOperation *result; +}; + + +/** + * Iterator for searching over the elements matching a given query + * + * @param cls the SearchContext + * @param key the 32-bit key + * @param value the RunContextOperation element + * @return GNUNET_YES to continue iteration; GNUNET_NO to cancel it + */ +static int +search_iterator (void *cls, uint32_t key, void *value) +{ + struct RunContextOperation *rcop = value; + struct SearchContext *sc = cls; + + GNUNET_assert (NULL != rcop); + if (sc->query == rcop->op) + { + GNUNET_assert (NULL == sc->result); + sc->result = rcop; + return GNUNET_NO; + } + return GNUNET_YES; +} + + +/** + * Initiate a search for the given operation in the rcop_map + * + * @param rc the RunContext whose rcop_map will be searched for the given + * operation + * @param op the given operation to search for + * @return the matching RunContextOperation if found; NULL if not + */ +static struct RunContextOperation * +search_rcop (struct GNUNET_TESTBED_RunHandle *rc, struct GNUNET_TESTBED_Operation *op) +{ + struct SearchContext sc; + + sc.query = op; + sc.result = NULL; + if (GNUNET_SYSERR == + GNUNET_CONTAINER_multihashmap32_get_multiple (rc->rcop_map, + rcop_key (op), + &search_iterator, + &sc)) + { + GNUNET_assert (NULL != sc.result); + return sc.result; + } + return NULL; +} + + +/** + * Insert an RunContextOperation into the rcop_map of the given RunContext + * + * @param rc the RunContext into whose map is to be used for insertion + * @param rcop the RunContextOperation to insert + */ +static void +insert_rcop (struct GNUNET_TESTBED_RunHandle *rc, struct RunContextOperation *rcop) +{ + GNUNET_assert (GNUNET_OK == + GNUNET_CONTAINER_multihashmap32_put (rc->rcop_map, + rcop_key (rcop->op), rcop, + GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE)); +} + + +/** + * Remove a RunContextOperation from the rcop_map of the given RunContext + * + * @param rc the RunContext from whose map the given RunContextOperaton has to + * be removed + * @param rcop the RunContextOperation + */ +static void +remove_rcop (struct GNUNET_TESTBED_RunHandle *rc, struct RunContextOperation *rcop) +{ + GNUNET_assert (GNUNET_YES == + GNUNET_CONTAINER_multihashmap32_remove (rc->rcop_map, + rcop_key (rcop->op), + rcop)); +} + /** * Assuming all peers have been destroyed cleanup run handle * - * @param cls the run handle - * @param tc the task context from scheduler + * @param rc the run context */ static void -cleanup_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) +cleanup (struct GNUNET_TESTBED_RunHandle *rc) { - struct RunContext *rc = cls; unsigned int hid; GNUNET_assert (GNUNET_SCHEDULER_NO_TASK == rc->register_hosts_task); GNUNET_assert (NULL == rc->reg_handle); GNUNET_assert (NULL == rc->peers); - GNUNET_assert (NULL == rc->hc_handles); + GNUNET_assert (NULL == rc->hclist); GNUNET_assert (RC_PEERS_SHUTDOWN == rc->state); - GNUNET_assert (NULL == rc->dll_op_head); + GNUNET_assert (0 == GNUNET_CONTAINER_multihashmap32_size (rc->rcop_map)); + GNUNET_CONTAINER_multihashmap32_destroy (rc->rcop_map); if (NULL != rc->c) GNUNET_TESTBED_controller_disconnect (rc->c); if (NULL != rc->cproc) @@ -342,19 +478,50 @@ cleanup_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) GNUNET_free (rc); } + +/** + * Iterator for cleaning up elements from rcop_map + * + * @param cls the RunContext + * @param key the 32-bit key + * @param value the RunContextOperation element + * @return always GNUNET_YES + */ +static int +rcop_cleanup_iterator (void *cls, uint32_t key, void *value) +{ + struct GNUNET_TESTBED_RunHandle *rc = cls; + struct RunContextOperation *rcop = value; + + GNUNET_assert (rc == rcop->rc); + remove_rcop (rc, rcop); + GNUNET_TESTBED_operation_done (rcop->op); + GNUNET_free (rcop); + return GNUNET_YES; +} + + +/** + * Cancels operations and tasks which are assigned to the given run context + * + * @param rc the RunContext + */ static void -cleanup (struct RunContext *rc) +rc_cleanup_operations (struct GNUNET_TESTBED_RunHandle *rc) { - struct DLLOperation *dll_op; + struct CompatibilityCheckContext *hc; unsigned int nhost; - if (NULL != rc->hc_handles) + if (NULL != rc->hclist) { for (nhost = 0; nhost < rc->num_hosts; nhost++) - if (NULL != rc->hc_handles[nhost]) - GNUNET_TESTBED_is_host_habitable_cancel (rc->hc_handles[nhost]); - GNUNET_free (rc->hc_handles); - rc->hc_handles = NULL; + { + hc = &rc->hclist[nhost]; + if (NULL != hc->h) + GNUNET_TESTBED_is_host_habitable_cancel (hc->h); + } + GNUNET_free (rc->hclist); + rc->hclist = NULL; } /* Stop register hosts task if it is running */ if (GNUNET_SCHEDULER_NO_TASK != rc->register_hosts_task) @@ -367,11 +534,6 @@ cleanup (struct RunContext *rc) GNUNET_SCHEDULER_cancel (rc->timeout_task); rc->timeout_task = GNUNET_SCHEDULER_NO_TASK; } - if (GNUNET_SCHEDULER_NO_TASK != rc->interrupt_task) - { - GNUNET_SCHEDULER_cancel (rc->interrupt_task); - rc->interrupt_task = GNUNET_SCHEDULER_NO_TASK; - } if (NULL != rc->reg_handle) { GNUNET_TESTBED_cancel_registration (rc->reg_handle); @@ -383,76 +545,96 @@ cleanup (struct RunContext *rc) rc->topology_operation = NULL; } /* cancel any exiting operations */ - while (NULL != (dll_op = rc->dll_op_head)) - { - GNUNET_TESTBED_operation_done (dll_op->op); - GNUNET_CONTAINER_DLL_remove (rc->dll_op_head, rc->dll_op_tail, dll_op); - GNUNET_free (dll_op); - } + GNUNET_assert (GNUNET_SYSERR != + GNUNET_CONTAINER_multihashmap32_iterate (rc->rcop_map, + &rcop_cleanup_iterator, + rc)); } /** - * Stops the testbed run and releases any used resources + * Cancels the scheduled interrupt task * - * @param cls the tesbed run handle - * @param tc the task context from scheduler + * @param rc the run context + */ +static void +cancel_interrupt_task (struct GNUNET_TESTBED_RunHandle *rc) +{ + GNUNET_SCHEDULER_cancel (rc->interrupt_task); + rc->interrupt_task = GNUNET_SCHEDULER_NO_TASK; +} + + +/** + * This callback will be called when all the operations are completed + * (done/cancelled) + * + * @param cls run context */ static void -shutdown_run (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) +wait_op_completion (void *cls) { - struct RunContext *rc = cls; - struct DLLOperation *dll_op; + struct GNUNET_TESTBED_RunHandle *rc = cls; + struct RunContextOperation *rcop; - GNUNET_assert (GNUNET_SCHEDULER_NO_TASK != rc->shutdown_run_task); - rc->shutdown_run_task = GNUNET_SCHEDULER_NO_TASK; - GNUNET_assert (GNUNET_NO == rc->shutdown); - rc->shutdown = GNUNET_YES; - cleanup (rc); - if (NULL != rc->c) + if ( (NULL == rc->cproc) + || (NULL == rc->c) + || (GNUNET_YES == rc->shutdown) ) { if (NULL != rc->peers) { - dll_op = GNUNET_malloc (sizeof (struct DLLOperation)); - dll_op->op = GNUNET_TESTBED_shutdown_peers (rc->c, dll_op, NULL, NULL); - GNUNET_assert (NULL != dll_op->op); - DEBUG ("Shutting down peers\n"); - rc->pstart_time = GNUNET_TIME_absolute_get (); - GNUNET_CONTAINER_DLL_insert_tail (rc->dll_op_head, rc->dll_op_tail, - dll_op); - return; + GNUNET_free (rc->peers); + rc->peers = NULL; } + goto cleanup_; } - rc->state = RC_PEERS_SHUTDOWN; /* No peers are present so we consider the - * state where all peers are SHUTDOWN */ - GNUNET_SCHEDULER_add_now (&cleanup_task, rc); + if (NULL == rc->peers) + goto cleanup_; + rc->shutdown = GNUNET_YES; + rcop = GNUNET_new (struct RunContextOperation); + rcop->rc = rc; + rcop->op = GNUNET_TESTBED_shutdown_peers (rc->c, rcop, NULL, NULL); + GNUNET_assert (NULL != rcop->op); + DEBUG ("Shutting down peers\n"); + rc->pstart_time = GNUNET_TIME_absolute_get (); + insert_rcop (rc, rcop); + return; + + cleanup_: + rc->state = RC_PEERS_SHUTDOWN; + cancel_interrupt_task (rc); + cleanup (rc); } /** - * Function to shutdown now + * Task run upon interrupts (SIGINT, SIGTERM) and upon scheduler shutdown. * - * @param rc the RunContext + * @param cls the RunContext which has to be acted upon + * @param tc the scheduler task context */ -static void -shutdown_now (struct RunContext *rc) -{ - if (GNUNET_YES == rc->shutdown) - return; - if (GNUNET_SCHEDULER_NO_TASK != rc->shutdown_run_task) - GNUNET_SCHEDULER_cancel (rc->shutdown_run_task); - GNUNET_SCHEDULER_shutdown (); /* Trigger shutdown in programs using this API */ - rc->shutdown_run_task = GNUNET_SCHEDULER_add_now (&shutdown_run, rc); -} - - static void interrupt (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) { - struct RunContext *rc = cls; - - rc->interrupt_task = GNUNET_SCHEDULER_NO_TASK; - shutdown_now (rc); + struct GNUNET_TESTBED_RunHandle *rc = cls; + struct GNUNET_TESTBED_Controller *c = rc->c; + unsigned int size; + + /* reschedule */ + rc->interrupt_task = GNUNET_SCHEDULER_add_delayed + (GNUNET_TIME_UNIT_FOREVER_REL, &interrupt, rc); + rc_cleanup_operations (rc); + if ( (GNUNET_NO == rc->shutdown) + && (NULL != c) + && (0 != (size = GNUNET_CONTAINER_multihashmap32_size (c->opc_map)))) + { + LOG (GNUNET_ERROR_TYPE_WARNING, "Shutdown postponed as there are " + "%u operations currently active\n", size); + c->opcq_empty_cb = &wait_op_completion; + c->opcq_empty_cls = rc; + return; + } + wait_op_completion (rc); } @@ -464,7 +646,7 @@ interrupt (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) * @return the representation string; this is NOT reentrant */ static const char * -prof_time (struct RunContext *rc) +prof_time (struct GNUNET_TESTBED_RunHandle *rc) { struct GNUNET_TIME_Relative ptime; @@ -482,19 +664,20 @@ prof_time (struct RunContext *rc) static void start_peers_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) { - struct RunContext *rc = cls; - struct DLLOperation *dll_op; + struct GNUNET_TESTBED_RunHandle *rc = cls; + struct RunContextOperation *rcop; unsigned int peer; DEBUG ("Starting Peers\n"); rc->pstart_time = GNUNET_TIME_absolute_get (); for (peer = 0; peer < rc->num_peers; peer++) { - dll_op = GNUNET_malloc (sizeof (struct DLLOperation)); - dll_op->op = GNUNET_TESTBED_peer_start (NULL, rc->peers[peer], NULL, NULL); - GNUNET_assert (NULL != dll_op->op); - dll_op->cls = rc->peers[peer]; - GNUNET_CONTAINER_DLL_insert_tail (rc->dll_op_head, rc->dll_op_tail, dll_op); + rcop = GNUNET_new (struct RunContextOperation); + rcop->rc = rc; + rcop->op = GNUNET_TESTBED_peer_start (NULL, rc->peers[peer], NULL, NULL); + GNUNET_assert (NULL != rcop->op); + rcop->cls = rc->peers[peer]; + insert_rcop (rc, rcop); } rc->peer_count = 0; } @@ -512,21 +695,20 @@ start_peers_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) static void peer_create_cb (void *cls, struct GNUNET_TESTBED_Peer *peer, const char *emsg) { - struct DLLOperation *dll_op = cls; - struct RunContext *rc; - - GNUNET_assert (NULL != dll_op); - rc = dll_op->rc; - GNUNET_assert (NULL != rc); - GNUNET_CONTAINER_DLL_remove (rc->dll_op_head, rc->dll_op_tail, dll_op); - GNUNET_TESTBED_operation_done (dll_op->op); - GNUNET_free (dll_op); + struct RunContextOperation *rcop = cls; + struct GNUNET_TESTBED_RunHandle *rc; + + GNUNET_assert (NULL != rcop); + GNUNET_assert (NULL != (rc = rcop->rc)); + remove_rcop (rc, rcop); + GNUNET_TESTBED_operation_done (rcop->op); + GNUNET_free (rcop); if (NULL == peer) { if (NULL != emsg) LOG (GNUNET_ERROR_TYPE_ERROR, "Error while creating a peer: %s\n", emsg); - shutdown_now (rc); + GNUNET_SCHEDULER_shutdown (); return; } rc->peers[rc->peer_count] = peer; @@ -545,12 +727,12 @@ peer_create_cb (void *cls, struct GNUNET_TESTBED_Peer *peer, const char *emsg) * @param rc the RunContext */ static void -call_master (struct RunContext *rc) +call_master (struct GNUNET_TESTBED_RunHandle *rc) { GNUNET_SCHEDULER_cancel (rc->timeout_task); rc->timeout_task = GNUNET_SCHEDULER_NO_TASK; if (NULL != rc->test_master) - rc->test_master (rc->test_master_cls, rc->num_peers, rc->peers, + rc->test_master (rc->test_master_cls, rc, rc->num_peers, rc->peers, rc->links_succeeded, rc->links_failed); } @@ -568,7 +750,7 @@ static void topology_completion_callback (void *cls, unsigned int nsuccess, unsigned int nfailures) { - struct RunContext *rc = cls; + struct GNUNET_TESTBED_RunHandle *rc = cls; DEBUG ("Overlay topology generated in %s\n", prof_time (rc)); GNUNET_TESTBED_operation_done (rc->topology_operation); @@ -586,9 +768,9 @@ topology_completion_callback (void *cls, unsigned int nsuccess, * @param rc the RunContext */ static void -create_peers (struct RunContext *rc) +create_peers (struct GNUNET_TESTBED_RunHandle *rc) { - struct DLLOperation *dll_op; + struct RunContextOperation *rcop; unsigned int peer; DEBUG ("Creating peers\n"); @@ -599,16 +781,16 @@ create_peers (struct RunContext *rc) rc->peer_count = 0; for (peer = 0; peer < rc->num_peers; peer++) { - dll_op = GNUNET_malloc (sizeof (struct DLLOperation)); - dll_op->rc = rc; - dll_op->op = + rcop = GNUNET_new (struct RunContextOperation); + rcop->rc = rc; + rcop->op = GNUNET_TESTBED_peer_create (rc->c, (0 == rc->num_hosts) ? rc->h : rc->hosts[peer % rc->num_hosts], - rc->cfg, peer_create_cb, dll_op); - GNUNET_assert (NULL != dll_op->op); - GNUNET_CONTAINER_DLL_insert_tail (rc->dll_op_head, rc->dll_op_tail, dll_op); + rc->cfg, &peer_create_cb, rcop); + GNUNET_assert (NULL != rcop->op); + insert_rcop (rc, rcop); } } @@ -623,26 +805,26 @@ create_peers (struct RunContext *rc) static void event_cb (void *cls, const struct GNUNET_TESTBED_EventInformation *event) { - struct RunContext *rc = cls; - struct DLLOperation *dll_op; + struct GNUNET_TESTBED_RunHandle *rc = cls; + struct RunContextOperation *rcop; if (RC_INIT == rc->state) { switch (event->type) { case GNUNET_TESTBED_ET_OPERATION_FINISHED: - dll_op = event->op_cls; + rcop = event->op_cls; if (NULL != event->details.operation_finished.emsg) { LOG (GNUNET_ERROR_TYPE_ERROR, _("Linking controllers failed. Exiting")); - shutdown_now (rc); + GNUNET_SCHEDULER_shutdown (); } else rc->reg_hosts++; - GNUNET_assert (event->op == dll_op->op); - GNUNET_CONTAINER_DLL_remove (rc->dll_op_head, rc->dll_op_tail, dll_op); - GNUNET_TESTBED_operation_done (dll_op->op); - GNUNET_free (dll_op); + GNUNET_assert (event->op == rcop->op); + remove_rcop (rc, rcop); + GNUNET_TESTBED_operation_done (rcop->op); + GNUNET_free (rcop); if (rc->reg_hosts == rc->num_hosts) { rc->state = RC_LINKED; @@ -651,27 +833,23 @@ event_cb (void *cls, const struct GNUNET_TESTBED_EventInformation *event) return; default: GNUNET_break (0); - shutdown_now (rc); + GNUNET_SCHEDULER_shutdown (); return; } } - for (dll_op = rc->dll_op_head; NULL != dll_op; dll_op = dll_op->next) - { - if ((GNUNET_TESTBED_ET_OPERATION_FINISHED == event->type) && - (event->op == dll_op->op)) - break; - } - if (NULL == dll_op) + if (GNUNET_TESTBED_ET_OPERATION_FINISHED != event->type) + goto call_cc; + if (NULL == (rcop = search_rcop (rc, event->op))) goto call_cc; - GNUNET_CONTAINER_DLL_remove (rc->dll_op_head, rc->dll_op_tail, dll_op); - GNUNET_TESTBED_operation_done (dll_op->op); - GNUNET_free (dll_op); + remove_rcop (rc, rcop); + GNUNET_TESTBED_operation_done (rcop->op); + GNUNET_free (rcop); if ( (GNUNET_NO == rc->shutdown) && (NULL != event->details.operation_finished.emsg) ) { LOG (GNUNET_ERROR_TYPE_ERROR, "A operation has failed with error: %s\n", event->details.operation_finished.emsg); - shutdown_now (rc); + GNUNET_SCHEDULER_shutdown (); return; } GNUNET_assert (GNUNET_YES == rc->shutdown); @@ -684,7 +862,7 @@ event_cb (void *cls, const struct GNUNET_TESTBED_EventInformation *event) GNUNET_free_non_null (rc->peers); rc->peers = NULL; DEBUG ("Peers shut down in %s\n", prof_time (rc)); - GNUNET_SCHEDULER_add_now (&cleanup_task, rc); + GNUNET_SCHEDULER_shutdown (); break; default: GNUNET_assert (0); @@ -696,25 +874,24 @@ call_cc: rc->cc (rc->cc_cls, event); if (GNUNET_TESTBED_ET_PEER_START != event->type) return; - for (dll_op = rc->dll_op_head; NULL != dll_op; dll_op = dll_op->next) - if ((NULL != dll_op->cls) && - (event->details.peer_start.peer == dll_op->cls)) - break; - if (NULL == dll_op) /* Not our operation */ + if (NULL == (rcop = search_rcop (rc, event->op))) /* Not our operation */ return; - GNUNET_CONTAINER_DLL_remove (rc->dll_op_head, rc->dll_op_tail, dll_op); - GNUNET_TESTBED_operation_done (dll_op->op); - GNUNET_free (dll_op); + remove_rcop (rc, rcop); + GNUNET_TESTBED_operation_done (rcop->op); + GNUNET_free (rcop); rc->peer_count++; if (rc->peer_count < rc->num_peers) return; DEBUG ("%u peers started in %s\n", rc->num_peers, prof_time (rc)); if (GNUNET_TESTBED_TOPOLOGY_NONE != rc->topology) { - if ((GNUNET_TESTBED_TOPOLOGY_ERDOS_RENYI == rc->topology) || - (GNUNET_TESTBED_TOPOLOGY_SMALL_WORLD_RING == rc->topology) || - (GNUNET_TESTBED_TOPOLOGY_SMALL_WORLD == rc->topology)) + switch (rc->topology) { + case GNUNET_TESTBED_TOPOLOGY_NONE: + GNUNET_assert (0); + case GNUNET_TESTBED_TOPOLOGY_ERDOS_RENYI: + case GNUNET_TESTBED_TOPOLOGY_SMALL_WORLD_RING: + case GNUNET_TESTBED_TOPOLOGY_SMALL_WORLD: rc->topology_operation = GNUNET_TESTBED_overlay_configure_topology (NULL, rc->num_peers, rc->peers, &rc->num_oc, @@ -723,9 +900,8 @@ call_cc: rc->topology, rc->random_links, GNUNET_TESTBED_TOPOLOGY_OPTION_END); - } - else if (GNUNET_TESTBED_TOPOLOGY_FROM_FILE == rc->topology) - { + break; + case GNUNET_TESTBED_TOPOLOGY_FROM_FILE: GNUNET_assert (NULL != rc->topo_file); rc->topology_operation = GNUNET_TESTBED_overlay_configure_topology (NULL, rc->num_peers, @@ -735,8 +911,32 @@ call_cc: rc->topology, rc->topo_file, GNUNET_TESTBED_TOPOLOGY_OPTION_END); - } - else + break; + case GNUNET_TESTBED_TOPOLOGY_SCALE_FREE: + { + unsigned long long number; + unsigned int cap; + GNUNET_assert (GNUNET_OK == + GNUNET_CONFIGURATION_get_value_number (rc->cfg, TESTBED_CONFIG_SECTION, + SCALE_FREE_CAP, + &number)); + cap = (unsigned int) number; + GNUNET_assert (GNUNET_OK == + GNUNET_CONFIGURATION_get_value_number (rc->cfg, TESTBED_CONFIG_SECTION, + SCALE_FREE_M, + &number)); + rc->topology_operation = + GNUNET_TESTBED_overlay_configure_topology (NULL, rc->num_peers, + rc->peers, &rc->num_oc, + &topology_completion_callback, + rc, + rc->topology, + cap, /* uint16_t */ + (unsigned int) number, /* uint8_t */ + GNUNET_TESTBED_TOPOLOGY_OPTION_END); + } + break; + default: rc->topology_operation = GNUNET_TESTBED_overlay_configure_topology (NULL, rc->num_peers, rc->peers, &rc->num_oc, @@ -744,9 +944,10 @@ call_cc: rc, rc->topology, GNUNET_TESTBED_TOPOLOGY_OPTION_END); + } if (NULL == rc->topology_operation) LOG (GNUNET_ERROR_TYPE_WARNING, - "Not generating topology. Check number of peers\n"); + "Not generating a topology. Check number of peers\n"); else { DEBUG ("Creating overlay topology\n"); @@ -778,14 +979,14 @@ register_hosts (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc); static void host_registration_completion (void *cls, const char *emsg) { - struct RunContext *rc = cls; + struct GNUNET_TESTBED_RunHandle *rc = cls; rc->reg_handle = NULL; if (NULL != emsg) { LOG (GNUNET_ERROR_TYPE_WARNING, _("Host registration failed for a host. Error: %s\n"), emsg); - shutdown_now (rc); + GNUNET_SCHEDULER_shutdown (); return; } rc->register_hosts_task = GNUNET_SCHEDULER_add_now (®ister_hosts, rc); @@ -801,8 +1002,8 @@ host_registration_completion (void *cls, const char *emsg) static void register_hosts (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) { - struct RunContext *rc = cls; - struct DLLOperation *dll_op; + struct GNUNET_TESTBED_RunHandle *rc = cls; + struct RunContextOperation *rcop; unsigned int slave; rc->register_hosts_task = GNUNET_SCHEDULER_NO_TASK; @@ -812,14 +1013,13 @@ register_hosts (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) /* Start slaves */ for (slave = 0; slave < rc->num_hosts; slave++) { - dll_op = GNUNET_malloc (sizeof (struct DLLOperation)); - dll_op->rc = rc; - dll_op->op = - GNUNET_TESTBED_controller_link (dll_op, rc->c, rc->hosts[slave], - rc->h, rc->cfg, GNUNET_YES); - GNUNET_assert (NULL != dll_op->op); - GNUNET_CONTAINER_DLL_insert_tail (rc->dll_op_head, rc->dll_op_tail, - dll_op); + rcop = GNUNET_new (struct RunContextOperation); + rcop->rc = rc; + rcop->op = + GNUNET_TESTBED_controller_link (rcop, rc->c, rc->hosts[slave], + rc->h, GNUNET_YES); + GNUNET_assert (NULL != rcop->op); + insert_rcop (rc, rcop); } rc->reg_hosts = 0; return; @@ -844,27 +1044,15 @@ static void controller_status_cb (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg, int status) { - struct RunContext *rc = cls; + struct GNUNET_TESTBED_RunHandle *rc = cls; uint64_t event_mask; if (status != GNUNET_OK) { + rc->cproc = NULL; GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Controller crash detected. Shutting down.\n")); - rc->cproc = NULL; - cleanup (rc); - if (NULL != rc->peers) - { - GNUNET_free (rc->peers); - rc->peers = NULL; - } - if (GNUNET_YES == rc->shutdown) - { - rc->state = RC_PEERS_SHUTDOWN; - GNUNET_SCHEDULER_add_now (&cleanup_task, rc); - } - else - shutdown_now (rc); + GNUNET_SCHEDULER_shutdown (); return; } GNUNET_CONFIGURATION_destroy (rc->cfg); @@ -875,8 +1063,7 @@ controller_status_cb (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg, if (rc->topology < GNUNET_TESTBED_TOPOLOGY_NONE) event_mask |= GNUNET_TESTBED_ET_CONNECT; rc->c = - GNUNET_TESTBED_controller_connect (rc->cfg, rc->h, event_mask, &event_cb, - rc); + GNUNET_TESTBED_controller_connect (rc->h, event_mask, &event_cb, rc); if (0 < rc->num_hosts) { rc->reg_hosts = 0; @@ -905,7 +1092,7 @@ netint_proc (void *cls, const char *name, int isDefault, const struct sockaddr *addr, const struct sockaddr *broadcast_addr, const struct sockaddr *netmask, socklen_t addrlen) { - struct RunContext *rc = cls; + struct GNUNET_TESTBED_RunHandle *rc = cls; char hostip[NI_MAXHOST]; char *buf; @@ -940,17 +1127,16 @@ static void host_habitable_cb (void *cls, const struct GNUNET_TESTBED_Host *host, int status) { - struct RunContext *rc = cls; + struct CompatibilityCheckContext *hc = cls; + struct GNUNET_TESTBED_RunHandle *rc; struct GNUNET_TESTBED_Host **old_hosts; unsigned int nhost; - for (nhost = 0; nhost < rc->num_hosts; nhost++) - { - if (host == rc->hosts[nhost]) - break; - } - GNUNET_assert (nhost != rc->num_hosts); - rc->hc_handles[nhost] = NULL; + GNUNET_assert (NULL != (rc = hc->rc)); + nhost = hc->index; + GNUNET_assert (nhost <= rc->num_hosts); + GNUNET_assert (host == rc->hosts[nhost]); + hc->h = NULL; if (GNUNET_NO == status) { if ((NULL != host) && (NULL != GNUNET_TESTBED_host_get_hostname (host))) @@ -959,14 +1145,14 @@ host_habitable_cb (void *cls, const struct GNUNET_TESTBED_Host *host, else LOG (GNUNET_ERROR_TYPE_ERROR, _("Testbed cannot be started on localhost\n")); - shutdown_now (rc); + GNUNET_SCHEDULER_shutdown (); return; } rc->reg_hosts++; if (rc->reg_hosts < rc->num_hosts) return; - GNUNET_free (rc->hc_handles); - rc->hc_handles = NULL; + GNUNET_free (rc->hclist); + rc->hclist = NULL; rc->h = rc->hosts[0]; rc->num_hosts--; if (0 < rc->num_hosts) @@ -983,18 +1169,21 @@ host_habitable_cb (void *cls, const struct GNUNET_TESTBED_Host *host, GNUNET_free (rc->hosts); rc->hosts = NULL; } + GNUNET_TESTBED_host_resolve_ (rc->h); + for (nhost = 0; nhost < rc->num_hosts; nhost++) + GNUNET_TESTBED_host_resolve_ (rc->hosts[nhost]); GNUNET_OS_network_interfaces_list (netint_proc, rc); if (NULL == rc->trusted_ip) rc->trusted_ip = GNUNET_strdup ("127.0.0.1"); rc->cproc = - GNUNET_TESTBED_controller_start (rc->trusted_ip, rc->h, rc->cfg, + GNUNET_TESTBED_controller_start (rc->trusted_ip, rc->h, &controller_status_cb, rc); GNUNET_free (rc->trusted_ip); rc->trusted_ip = NULL; if (NULL == rc->cproc) { LOG (GNUNET_ERROR_TYPE_ERROR, _("Cannot start the master controller")); - shutdown_now (rc); + GNUNET_SCHEDULER_shutdown (); } } @@ -1008,13 +1197,13 @@ host_habitable_cb (void *cls, const struct GNUNET_TESTBED_Host *host, static void timeout_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) { - struct RunContext *rc = cls; - + struct GNUNET_TESTBED_RunHandle *rc = cls; + rc->timeout_task = GNUNET_SCHEDULER_NO_TASK; LOG (GNUNET_ERROR_TYPE_ERROR, _("Shutting down testbed due to timeout while setup.\n")); - shutdown_now (rc); + GNUNET_SCHEDULER_shutdown (); if (NULL != rc->test_master) - rc->test_master (rc->test_master_cls, 0, NULL, 0, 0); + rc->test_master (rc->test_master_cls, rc, 0, NULL, 0, 0); rc->test_master = NULL; } @@ -1054,17 +1243,18 @@ GNUNET_TESTBED_run (const char *host_filename, GNUNET_TESTBED_TestMaster test_master, void *test_master_cls) { - struct RunContext *rc; + struct GNUNET_TESTBED_RunHandle *rc; char *topology; + struct CompatibilityCheckContext *hc; struct GNUNET_TIME_Relative timeout; - unsigned long long random_links; + unsigned long long number; unsigned int hid; unsigned int nhost; GNUNET_assert (num_peers > 0); - rc = GNUNET_malloc (sizeof (struct RunContext)); + rc = GNUNET_new (struct GNUNET_TESTBED_RunHandle); rc->cfg = GNUNET_CONFIGURATION_dup (cfg); -#if ENABLE_LL +#if ENABLE_SUPERMUC rc->num_hosts = GNUNET_TESTBED_hosts_load_from_loadleveler (rc->cfg, &rc->hosts); if (0 == rc->num_hosts) @@ -1098,12 +1288,12 @@ GNUNET_TESTBED_run (const char *host_filename, rc->state = RC_INIT; rc->topology = GNUNET_TESTBED_TOPOLOGY_NONE; if (GNUNET_OK == - GNUNET_CONFIGURATION_get_value_string (rc->cfg, "testbed", + GNUNET_CONFIGURATION_get_value_string (rc->cfg, TESTBED_CONFIG_SECTION, "OVERLAY_TOPOLOGY", &topology)) { if (GNUNET_NO == GNUNET_TESTBED_topology_get_ (&rc->topology, topology)) { - GNUNET_log_config_invalid (GNUNET_ERROR_TYPE_ERROR, "testbed", + GNUNET_log_config_invalid (GNUNET_ERROR_TYPE_ERROR, TESTBED_CONFIG_SECTION, "OVERLAY_TOPLOGY", _ ("Specified topology must be supported by testbed")); @@ -1116,37 +1306,73 @@ GNUNET_TESTBED_run (const char *host_filename, case GNUNET_TESTBED_TOPOLOGY_SMALL_WORLD_RING: case GNUNET_TESTBED_TOPOLOGY_SMALL_WORLD: if (GNUNET_OK != - GNUNET_CONFIGURATION_get_value_number (rc->cfg, "testbed", + GNUNET_CONFIGURATION_get_value_number (rc->cfg, TESTBED_CONFIG_SECTION, "OVERLAY_RANDOM_LINKS", - &random_links)) + &number)) { /* OVERLAY option RANDOM & SMALL_WORLD_RING requires OVERLAY_RANDOM_LINKS * option to be set to the number of random links to be established */ - GNUNET_log_config_missing (GNUNET_ERROR_TYPE_ERROR, "testbed", + GNUNET_log_config_missing (GNUNET_ERROR_TYPE_ERROR, TESTBED_CONFIG_SECTION, "OVERLAY_RANDOM_LINKS"); goto error_cleanup; } - if (random_links > UINT32_MAX) + if (number > UINT32_MAX) { GNUNET_break (0); /* Too big number */ goto error_cleanup; } - rc->random_links = (unsigned int) random_links; + rc->random_links = (unsigned int) number; break; case GNUNET_TESTBED_TOPOLOGY_FROM_FILE: if (GNUNET_OK != - GNUNET_CONFIGURATION_get_value_string (rc->cfg, "testbed", + GNUNET_CONFIGURATION_get_value_filename (rc->cfg, TESTBED_CONFIG_SECTION, "OVERLAY_TOPOLOGY_FILE", &rc->topo_file)) { - GNUNET_log_config_missing (GNUNET_ERROR_TYPE_ERROR, "testbed", + GNUNET_log_config_missing (GNUNET_ERROR_TYPE_ERROR, TESTBED_CONFIG_SECTION, "OVERLAY_TOPOLOGY_FILE"); goto error_cleanup; } + goto warn_ignore; + case GNUNET_TESTBED_TOPOLOGY_SCALE_FREE: + if (GNUNET_OK != + GNUNET_CONFIGURATION_get_value_number (rc->cfg, TESTBED_CONFIG_SECTION, + SCALE_FREE_CAP, &number)) + { + GNUNET_log_config_missing (GNUNET_ERROR_TYPE_ERROR, TESTBED_CONFIG_SECTION, + SCALE_FREE_CAP); + goto error_cleanup; + } + if (UINT16_MAX < number) + { + LOG (GNUNET_ERROR_TYPE_ERROR, + _("Maximum number of edges a peer can have in a scale free topology" + " cannot be more than %u. Given `%s = %llu'"), UINT16_MAX, + SCALE_FREE_CAP, number); + goto error_cleanup; + } + if (GNUNET_OK != + GNUNET_CONFIGURATION_get_value_number (rc->cfg, TESTBED_CONFIG_SECTION, + SCALE_FREE_M, &number)) + { + GNUNET_log_config_missing (GNUNET_ERROR_TYPE_ERROR, TESTBED_CONFIG_SECTION, + SCALE_FREE_M); + goto error_cleanup; + } + if (UINT8_MAX < number) + { + LOG (GNUNET_ERROR_TYPE_ERROR, + _("The number of edges that can established when adding a new node" + " to scale free topology cannot be more than %u. Given `%s = %llu'"), + UINT8_MAX, SCALE_FREE_M, number); + goto error_cleanup; + } + goto warn_ignore; default: + warn_ignore: /* Warn if OVERLAY_RANDOM_LINKS is present that it will be ignored */ if (GNUNET_YES == - GNUNET_CONFIGURATION_have_value (rc->cfg, "testbed", + GNUNET_CONFIGURATION_have_value (rc->cfg, TESTBED_CONFIG_SECTION, "OVERLAY_RANDOM_LINKS")) LOG (GNUNET_ERROR_TYPE_WARNING, "Ignoring value of `OVERLAY_RANDOM_LINKS' in given configuration\n"); @@ -1154,37 +1380,42 @@ GNUNET_TESTBED_run (const char *host_filename, } if (0 != rc->num_hosts) { - rc->hc_handles = - GNUNET_malloc (sizeof (struct GNUNET_TESTBED_HostHabitableCheckHandle *) - * rc->num_hosts); + rc->hclist = GNUNET_malloc (sizeof (struct CompatibilityCheckContext) + * rc->num_hosts); for (nhost = 0; nhost < rc->num_hosts; nhost++) { - if (NULL == - (rc->hc_handles[nhost] = - GNUNET_TESTBED_is_host_habitable (rc->hosts[nhost], rc->cfg, - &host_habitable_cb, rc))) + hc = &rc->hclist[nhost]; + hc->index = nhost; + hc->rc = rc; + hc->h = GNUNET_TESTBED_is_host_habitable (rc->hosts[nhost], rc->cfg, + &host_habitable_cb, hc); + if (NULL == hc->h) { GNUNET_break (0); for (nhost = 0; nhost < rc->num_hosts; nhost++) - if (NULL != rc->hc_handles[nhost]) - GNUNET_TESTBED_is_host_habitable_cancel (rc->hc_handles[nhost]); - GNUNET_free (rc->hc_handles); - rc->hc_handles = NULL; + { + hc = &rc->hclist[nhost]; + if (NULL != hc->h) + GNUNET_TESTBED_is_host_habitable_cancel (hc->h); + } + GNUNET_free (rc->hclist); + rc->hclist = NULL; goto error_cleanup; } } } else rc->cproc = - GNUNET_TESTBED_controller_start ("127.0.0.1", rc->h, rc->cfg, + GNUNET_TESTBED_controller_start ("127.0.0.1", rc->h, &controller_status_cb, rc); - if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_time (cfg, "TESTBED", + if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_time (cfg, TESTBED_CONFIG_SECTION, "SETUP_TIMEOUT", &timeout)) { timeout = GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, DEFAULT_SETUP_TIMEOUT); } + rc->rcop_map = GNUNET_CONTAINER_multihashmap32_create (256); rc->timeout_task = GNUNET_SCHEDULER_add_delayed (timeout, &timeout_task, rc); rc->interrupt_task = @@ -1208,4 +1439,18 @@ error_cleanup: } +/** + * Obtain handle to the master controller from a testbed run. The handle + * returned should not be disconnected. + * + * @param h the testbed run handle + * @return handle to the master controller + */ +struct GNUNET_TESTBED_Controller * +GNUNET_TESTBED_run_get_controller_handle (struct GNUNET_TESTBED_RunHandle *h) +{ + return h->c; +} + + /* end of testbed_api_testbed.c */