diff options
Diffstat (limited to 'src/dht/test_dht_multipeer.c')
| -rw-r--r-- | src/dht/test_dht_multipeer.c | 859 |
1 files changed, 0 insertions, 859 deletions
diff --git a/src/dht/test_dht_multipeer.c b/src/dht/test_dht_multipeer.c deleted file mode 100644 index ab7d90e..0000000 --- a/src/dht/test_dht_multipeer.c +++ /dev/null @@ -1,859 +0,0 @@ -/* - This file is part of GNUnet. - (C) 2009 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 - by the Free Software Foundation; either version 3, or (at your - option) any later version. - - GNUnet is distributed in the hope that it will be useful, but - WITHOUT ANY WARRANTY; without even the implied warranty of - MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU - General Public License for more details. - - You should have received a copy of the GNU General Public License - along with GNUnet; see the file COPYING. If not, write to the - Free Software Foundation, Inc., 59 Temple Place - Suite 330, - Boston, MA 02111-1307, USA. -*/ -/** - * @file dht/test_dht_multipeer.c - * @brief testcase for testing DHT service with - * multiple peers. - */ -#include "platform.h" -#include "gnunet_testing_lib.h" -#include "gnunet_core_service.h" -#include "gnunet_dht_service.h" - -/* Timeout for entire testcase */ -#define TIMEOUT GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_MINUTES, 30) - -/* Timeout for waiting for replies to get requests */ -#define GET_TIMEOUT GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_SECONDS, 300) - -/* */ -#define START_DELAY GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_SECONDS, 30) - -/* Timeout for waiting for gets to complete */ -#define GET_DELAY GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_MILLISECONDS, 50) - -/* Timeout for waiting for puts to complete */ -#define PUT_DELAY GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_MILLISECONDS, 50) - -/* If number of peers not in config file, use this number */ -#define DEFAULT_NUM_PEERS 10 - -#define TEST_DATA_SIZE 8 - -#define MAX_OUTSTANDING_PUTS 100 - -#define MAX_OUTSTANDING_GETS 100 - -#define PATH_TRACKING GNUNET_NO - - - -struct TestPutContext -{ - /** - * This is a linked list - */ - struct TestPutContext *next; - - /** - * This is a linked list - */ - struct TestPutContext *prev; - - /** - * Handle to the first peers DHT service (via the API) - */ - struct GNUNET_DHT_Handle *dht_handle; - - /** - * Handle to the PUT peer daemon - */ - struct GNUNET_TESTING_Daemon *daemon; - - /** - * Identifier for this PUT - */ - uint32_t uid; - - /** - * Task handle for processing of the put. - */ - GNUNET_SCHEDULER_TaskIdentifier task; -}; - - -struct TestGetContext -{ - /** - * This is a linked list - */ - struct TestGetContext *next; - - /** - * This is a linked list - */ - struct TestGetContext *prev; - - /** - * Handle to the first peers DHT service (via the API) - */ - struct GNUNET_DHT_Handle *dht_handle; - - /** - * Handle for the DHT get request - */ - struct GNUNET_DHT_GetHandle *get_handle; - - /** - * Handle to the GET peer daemon - */ - struct GNUNET_TESTING_Daemon *daemon; - - /** - * Identifier for this GET - */ - uint32_t uid; - - /** - * Task for disconnecting DHT handles (and stopping GET) - */ - GNUNET_SCHEDULER_TaskIdentifier task; - - /** - * Whether or not this request has been fulfilled already. - */ - int succeeded; -}; - - -/** - * List of GETS to perform - */ -static struct TestGetContext *all_gets_head; - -/** - * List of GETS to perform - */ -static struct TestGetContext *all_gets_tail; - -/** - * List of PUTS to perform - */ -static struct TestPutContext *all_puts_head; - -/** - * List of PUTS to perform - */ -static struct TestPutContext *all_puts_tail; - -/** - * Handle to the set of all peers run for this test. - */ -static struct GNUNET_TESTING_PeerGroup *pg; - -/** - * Total number of peers to run, set based on config file. - */ -static unsigned long long num_peers; - -/** - * How many puts do we currently have in flight? - */ -static unsigned long long outstanding_puts; - -/** - * How many puts are done? - */ -static unsigned long long puts_completed; - -/** - * How many puts do we currently have in flight? - */ -static unsigned long long outstanding_gets; - -/** - * How many gets are done? - */ -static unsigned long long gets_completed; - -/** - * How many gets failed? - */ -static unsigned long long gets_failed; - -/** - * Directory to remove on shutdown. - */ -static char *test_directory; - -/** - * Option to use when routing. - */ -static enum GNUNET_DHT_RouteOption route_option; - -/** - * Task handle to use to schedule test failure / success. - */ -static GNUNET_SCHEDULER_TaskIdentifier die_task; - -/** - * Task handle to use to schedule test shutdown - */ -GNUNET_SCHEDULER_TaskIdentifier shutdown_task; - -/** - * Global return value (0 for success, anything else for failure) - */ -static int ok; - - -/** - * Check whether peers successfully shut down. - */ -static void -shutdown_callback (void *cls, const char *emsg) -{ - if (emsg != NULL) - { - FPRINTF (stderr, "Failed to shutdown testing topology: %s\n", emsg); - if (ok == 0) - ok = 2; - } - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Shutdown callback completed.\n"); -} - -static void -do_stop (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) -{ - if ((tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) == 0) - { - if (GNUNET_SCHEDULER_NO_TASK != shutdown_task) - { - GNUNET_SCHEDULER_cancel(shutdown_task); - shutdown_task = GNUNET_SCHEDULER_NO_TASK; - } - } - else - { - shutdown_task = GNUNET_SCHEDULER_NO_TASK ; - } - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Shutdown requested.\n"); - if (NULL != pg) - GNUNET_TESTING_daemons_stop (pg, TIMEOUT, &shutdown_callback, NULL); - pg = NULL; -} - - -/** - * Master context for 'stat_run'. - */ -struct StatMaster -{ - struct GNUNET_STATISTICS_Handle *stat; - unsigned int daemon; - unsigned int value; -}; - -struct StatValues -{ - const char *subsystem; - const char *name; - unsigned long long total; -}; - -/** - * Statistics we print out. - */ -static struct StatValues stats[] = { - {"core", "# bytes decrypted", 0}, - {"core", "# bytes encrypted", 0}, - {"core", "# type maps received", 0}, - {"core", "# session keys confirmed via PONG", 0}, - {"core", "# peers connected", 0}, - {"core", "# key exchanges initiated", 0}, - {"core", "# send requests dropped (disconnected)", 0}, - {"core", "# transmissions delayed due to corking", 0}, - {"core", "# messages discarded (expired prior to transmission)", 0}, - {"core", "# messages discarded (disconnected)", 0}, - {"core", "# discarded CORE_SEND requests", 0}, - {"core", "# discarded lower priority CORE_SEND requests", 0}, - {"transport", "# bytes received via TCP", 0}, - {"transport", "# bytes transmitted via TCP", 0}, - {"dht", "# PUT messages queued for transmission", 0}, - {"dht", "# P2P PUT requests received", 0}, - {"dht", "# GET messages queued for transmission", 0}, - {"dht", "# P2P GET requests received", 0}, - {"dht", "# RESULT messages queued for transmission", 0}, - {"dht", "# P2P RESULTS received", 0}, - {"dht", "# Queued messages discarded (peer disconnected)", 0}, - {"dht", "# Peers excluded from routing due to Bloomfilter", 0}, - {"dht", "# Peer selection failed", 0}, - {"dht", "# FIND PEER requests ignored due to Bloomfilter", 0}, - {"dht", "# FIND PEER requests ignored due to lack of HELLO", 0}, - {"dht", "# P2P FIND PEER requests processed", 0}, - {"dht", "# P2P GET requests ONLY routed", 0}, - {"dht", "# Preference updates given to core", 0}, - {"dht", "# REPLIES ignored for CLIENTS (no match)", 0}, - {"dht", "# GET requests from clients injected", 0}, - {"dht", "# GET requests received from clients", 0}, - {"dht", "# GET STOP requests received from clients", 0}, - {"dht", "# ITEMS stored in datacache", 0}, - {"dht", "# Good RESULTS found in datacache", 0}, - {"dht", "# GET requests given to datacache", 0}, - {NULL, NULL, 0} -}; - - -/** - * Callback function to process statistic values. - * - * @param cls closure - * @param subsystem name of subsystem that created the statistic - * @param name the name of the datum - * @param value the current value - * @param is_persistent GNUNET_YES if the value is persistent, GNUNET_NO if not - * @return GNUNET_OK to continue, GNUNET_SYSERR to abort iteration - */ -static int -print_stat (void *cls, const char *subsystem, const char *name, uint64_t value, - int is_persistent) -{ - struct StatMaster *sm = cls; - - stats[sm->value].total += value; - FPRINTF (stderr, "Peer %2u: %12s/%50s = %12llu\n", sm->daemon, subsystem, - name, (unsigned long long) value); - return GNUNET_OK; -} - - -/** - * Function that gathers stats from all daemons. - */ -static void -stat_run (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc); - - -/** - * Function called when GET operation on stats is done. - */ -static void -get_done (void *cls, int success) -{ - struct StatMaster *sm = cls; - - GNUNET_break (GNUNET_OK == success); - sm->value++; - GNUNET_SCHEDULER_add_now (&stat_run, sm); -} - - -/** - * Function that gathers stats from all daemons. - */ -static void -stat_run (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) -{ - struct StatMaster *sm = cls; - unsigned int i; - - die_task = GNUNET_SCHEDULER_NO_TASK; - if (stats[sm->value].name != NULL) - { - GNUNET_STATISTICS_get (sm->stat, -#if 0 - NULL, NULL, -#else - stats[sm->value].subsystem, stats[sm->value].name, -#endif - GNUNET_TIME_UNIT_FOREVER_REL, &get_done, &print_stat, - sm); - return; - } - GNUNET_STATISTICS_destroy (sm->stat, GNUNET_NO); - sm->value = 0; - sm->daemon++; - if (sm->daemon == num_peers) - { - GNUNET_free (sm); - i = 0; - while (stats[i].name != NULL) - { - FPRINTF (stderr, "Total : %12s/%50s = %12llu\n", stats[i].subsystem, - stats[i].name, (unsigned long long) stats[i].total); - i++; - } - die_task = GNUNET_SCHEDULER_add_now (&do_stop, NULL); - return; - } - sm->stat = - GNUNET_STATISTICS_create ("<driver>", - GNUNET_TESTING_daemon_get (pg, - sm->daemon)->cfg); - die_task = GNUNET_SCHEDULER_add_now (&stat_run, sm); -} - - -/** - * Function scheduled to be run on the successful completion of this - * testcase. - */ -static void -finish_testing (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) -{ - struct TestPutContext *test_put; - struct TestGetContext *test_get; - struct StatMaster *sm; - - die_task = GNUNET_SCHEDULER_NO_TASK; - while (NULL != (test_put = all_puts_head)) - { - if (test_put->task != GNUNET_SCHEDULER_NO_TASK) - GNUNET_SCHEDULER_cancel (test_put->task); - if (test_put->dht_handle != NULL) - GNUNET_DHT_disconnect (test_put->dht_handle); - GNUNET_CONTAINER_DLL_remove (all_puts_head, all_puts_tail, test_put); - GNUNET_free (test_put); - } - - while (NULL != (test_get = all_gets_head)) - { - if (test_get->task != GNUNET_SCHEDULER_NO_TASK) - GNUNET_SCHEDULER_cancel (test_get->task); - if (test_get->get_handle != NULL) - GNUNET_DHT_get_stop (test_get->get_handle); - if (test_get->dht_handle != NULL) - GNUNET_DHT_disconnect (test_get->dht_handle); - GNUNET_CONTAINER_DLL_remove (all_gets_head, all_gets_tail, test_get); - GNUNET_free (test_get); - } - sm = GNUNET_malloc (sizeof (struct StatMaster)); - sm->stat = - GNUNET_STATISTICS_create ("<driver>", - GNUNET_TESTING_daemon_get (pg, - sm->daemon)->cfg); - die_task = GNUNET_SCHEDULER_add_now (&stat_run, sm); -} - - -/** - * Check if the get_handle is being used, if so stop the request. Either - * way, schedule the end_badly_cont function which actually shuts down the - * test. - */ -static void -end_badly (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) -{ - const char *emsg = cls; - struct TestPutContext *test_put; - struct TestGetContext *test_get; - - die_task = GNUNET_SCHEDULER_NO_TASK; - FPRINTF (stderr, "Failing test with error: `%s'!\n", emsg); - while (NULL != (test_put = all_puts_head)) - { - if (test_put->task != GNUNET_SCHEDULER_NO_TASK) - GNUNET_SCHEDULER_cancel (test_put->task); - if (test_put->dht_handle != NULL) - GNUNET_DHT_disconnect (test_put->dht_handle); - GNUNET_CONTAINER_DLL_remove (all_puts_head, all_puts_tail, test_put); - GNUNET_free (test_put); - } - - while (NULL != (test_get = all_gets_head)) - { - if (test_get->task != GNUNET_SCHEDULER_NO_TASK) - GNUNET_SCHEDULER_cancel (test_get->task); - if (test_get->get_handle != NULL) - GNUNET_DHT_get_stop (test_get->get_handle); - if (test_get->dht_handle != NULL) - GNUNET_DHT_disconnect (test_get->dht_handle); - GNUNET_CONTAINER_DLL_remove (all_gets_head, all_gets_tail, test_get); - GNUNET_free (test_get); - } - ok = 1; - /* testing_peergroup will do that in its own end_badly() handler */ - /*GNUNET_TESTING_daemons_stop (pg, TIMEOUT, &shutdown_callback, NULL); */ - pg = NULL; -} - - -/** - * Task to release get handle. - */ -static void -get_stop_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) -{ - struct TestGetContext *test_get = cls; - GNUNET_HashCode search_key; /* Key stored under */ - char original_data[TEST_DATA_SIZE]; /* Made up data to store */ - - test_get->task = GNUNET_SCHEDULER_NO_TASK; - memset (original_data, test_get->uid, sizeof (original_data)); - GNUNET_CRYPTO_hash (original_data, TEST_DATA_SIZE, &search_key); - if (test_get->succeeded != GNUNET_YES) - { - gets_failed++; - FPRINTF (stderr, "Get from peer %s for key %s failed!\n", - GNUNET_i2s (&test_get->daemon->id), GNUNET_h2s (&search_key)); - } - GNUNET_assert (test_get->get_handle != NULL); - GNUNET_DHT_get_stop (test_get->get_handle); - test_get->get_handle = NULL; - - outstanding_gets--; /* GET is really finished */ - GNUNET_DHT_disconnect (test_get->dht_handle); - test_get->dht_handle = NULL; - - GNUNET_CONTAINER_DLL_remove (all_gets_head, all_gets_tail, test_get); - GNUNET_free (test_get); - if ((gets_failed > 10) && (outstanding_gets == 0)) - { - /* Had more than 10% failures */ - FPRINTF (stderr, "%llu gets succeeded, %llu gets failed!\n", gets_completed, - gets_failed); - GNUNET_SCHEDULER_cancel (die_task); - ok = 1; - die_task = - GNUNET_SCHEDULER_add_now (&finish_testing, "not all gets succeeded"); - return; - } - if ((gets_completed + gets_failed == num_peers * num_peers) && (outstanding_gets == 0)) /* All gets successful */ - { - FPRINTF (stderr, "%llu gets succeeded, %llu gets failed!\n", gets_completed, - gets_failed); - GNUNET_SCHEDULER_cancel (die_task); - ok = 0; - die_task = GNUNET_SCHEDULER_add_now (&finish_testing, NULL); - } -} - - -/** - * Iterator called if the GET request initiated returns a response. - * - * @param cls closure - * @param exp when will this value expire - * @param key key of the result - * @param type type of the result - * @param size number of bytes in data - * @param data pointer to the result data - */ -static void -get_result_iterator (void *cls, struct GNUNET_TIME_Absolute exp, - const GNUNET_HashCode * key, - const struct GNUNET_PeerIdentity *get_path, - unsigned int get_path_length, - const struct GNUNET_PeerIdentity *put_path, - unsigned int put_path_length, enum GNUNET_BLOCK_Type type, - size_t size, const void *data) -{ - struct TestGetContext *test_get = cls; - GNUNET_HashCode search_key; /* Key stored under */ - char original_data[TEST_DATA_SIZE]; /* Made up data to store */ - - memset (original_data, test_get->uid, sizeof (original_data)); - GNUNET_CRYPTO_hash (original_data, TEST_DATA_SIZE, &search_key); - if (test_get->succeeded == GNUNET_YES) - return; /* Get has already been successful, probably ending now */ - -#if PATH_TRACKING - if (put_path != NULL) - { - unsigned int i; - - FPRINTF (stderr, "PUT (%u) Path: ", test_get->uid); - for (i = 0; i < put_path_length; i++) - FPRINTF (stderr, "%s%s", i == 0 ? "" : "->", GNUNET_i2s (&put_path[i])); - FPRINTF (stderr, "%s", "\n"); - } - if (get_path != NULL) - { - unsigned int i; - - FPRINTF (stderr, "GET (%u) Path: ", test_get->uid); - for (i = 0; i < get_path_length; i++) - FPRINTF (stderr, "%s%s", i == 0 ? "" : "->", GNUNET_i2s (&get_path[i])); - FPRINTF (stderr, "%s%s\n", get_path_length > 0 ? "->" : "", - GNUNET_i2s (&test_get->daemon->id)); - } -#endif - - if ((0 != memcmp (&search_key, key, sizeof (GNUNET_HashCode))) || - (0 != memcmp (original_data, data, sizeof (original_data)))) - { - FPRINTF (stderr, "%s", "Key or data is not the same as was inserted!\n"); - return; - } - gets_completed++; - test_get->succeeded = GNUNET_YES; - GNUNET_SCHEDULER_cancel (test_get->task); - test_get->task = GNUNET_SCHEDULER_add_now (&get_stop_task, test_get); -} - - -/** - * Set up some data, and call API PUT function - */ -static void -do_get (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) -{ - struct TestGetContext *test_get = cls; - GNUNET_HashCode key; /* Made up key to store data under */ - char data[TEST_DATA_SIZE]; /* Made up data to store */ - - if (outstanding_gets > MAX_OUTSTANDING_GETS) - { - test_get->task = - GNUNET_SCHEDULER_add_delayed (GET_DELAY, &do_get, test_get); - return; - } - memset (data, test_get->uid, sizeof (data)); - GNUNET_CRYPTO_hash (data, TEST_DATA_SIZE, &key); - test_get->dht_handle = GNUNET_DHT_connect (test_get->daemon->cfg, 10); - GNUNET_assert (test_get->dht_handle != NULL); - outstanding_gets++; - test_get->get_handle = - GNUNET_DHT_get_start (test_get->dht_handle, - GNUNET_BLOCK_TYPE_TEST, &key, 1, route_option, NULL, - 0, &get_result_iterator, test_get); - test_get->task = - GNUNET_SCHEDULER_add_delayed (GET_TIMEOUT, &get_stop_task, test_get); -} - - -/** - * Task to release DHT handles for PUT - */ -static void -put_disconnect_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) -{ - struct TestPutContext *test_put = cls; - - test_put->task = GNUNET_SCHEDULER_NO_TASK; - GNUNET_DHT_disconnect (test_put->dht_handle); - test_put->dht_handle = NULL; - GNUNET_CONTAINER_DLL_remove (all_puts_head, all_puts_tail, test_put); - GNUNET_free (test_put); -} - - -/** - * Schedule the GET requests - */ -static void -start_gets (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) -{ - unsigned long long i; - unsigned long long j; - struct TestGetContext *test_get; - - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, - "Issuing %llu GETs\n", - (unsigned long long) (num_peers * num_peers)); - for (i = 0; i < num_peers; i++) - for (j = 0; j < num_peers; j++) - { - test_get = GNUNET_malloc (sizeof (struct TestGetContext)); - test_get->uid = i + j * num_peers; - test_get->daemon = GNUNET_TESTING_daemon_get (pg, j); - GNUNET_CONTAINER_DLL_insert (all_gets_head, all_gets_tail, test_get); - test_get->task = GNUNET_SCHEDULER_add_now (&do_get, test_get); - } -} - - -/** - * Called when the PUT request has been transmitted to the DHT service. - */ -static void -put_finished (void *cls, int success) -{ - struct TestPutContext *test_put = cls; - - outstanding_puts--; - puts_completed++; - if (GNUNET_SCHEDULER_NO_TASK != test_put->task) - { - GNUNET_SCHEDULER_cancel (test_put->task); - } - test_put->task = GNUNET_SCHEDULER_add_now (&put_disconnect_task, test_put); - if (puts_completed != num_peers * num_peers) - return; - - GNUNET_assert (outstanding_puts == 0); - GNUNET_SCHEDULER_add_delayed (START_DELAY, &start_gets, NULL); -} - - -/** - * Set up some data, and call API PUT function - */ -static void -do_put (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) -{ - struct TestPutContext *test_put = cls; - GNUNET_HashCode key; /* Made up key to store data under */ - char data[TEST_DATA_SIZE]; /* Made up data to store */ - - test_put->task = GNUNET_SCHEDULER_NO_TASK; - if (outstanding_puts > MAX_OUTSTANDING_PUTS) - { - test_put->task = - GNUNET_SCHEDULER_add_delayed (PUT_DELAY, &do_put, test_put); - return; - } - memset (data, test_put->uid, sizeof (data)); - GNUNET_CRYPTO_hash (data, TEST_DATA_SIZE, &key); - test_put->dht_handle = GNUNET_DHT_connect (test_put->daemon->cfg, 10); - GNUNET_assert (test_put->dht_handle != NULL); - outstanding_puts++; - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, - "PUT %u at `%s'\n", test_put->uid, - GNUNET_i2s (&test_put->daemon->id)); - GNUNET_DHT_put (test_put->dht_handle, &key, 1, route_option, - GNUNET_BLOCK_TYPE_TEST, sizeof (data), data, - GNUNET_TIME_UNIT_FOREVER_ABS, GNUNET_TIME_UNIT_FOREVER_REL, - &put_finished, test_put); - test_put->task = - GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, - &put_disconnect_task, test_put); -} - - -static void -run_dht_test (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) -{ - unsigned long long i; - struct TestPutContext *test_put; - - if ((tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN) != 0) - { - ok = 1; - return; - } -#if PATH_TRACKING - route_option = - GNUNET_DHT_RO_RECORD_ROUTE | GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE; -#else - route_option = GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE; -#endif - die_task = - GNUNET_SCHEDULER_add_delayed (TIMEOUT, &end_badly, - "from setup puts/gets"); - FPRINTF (stderr, "Issuing %llu PUTs (one per peer)\n", - (unsigned long long) (num_peers * num_peers)); - for (i = 0; i < num_peers * num_peers; i++) - { - test_put = GNUNET_malloc (sizeof (struct TestPutContext)); - test_put->uid = i; - test_put->daemon = GNUNET_TESTING_daemon_get (pg, i % num_peers); - test_put->task = GNUNET_SCHEDULER_add_now (&do_put, test_put); - GNUNET_CONTAINER_DLL_insert (all_puts_head, all_puts_tail, test_put); - } -} - - -/** - * This function is called once testing has finished setting up the topology. - * - * @param cls unused - * @param emsg variable is NULL on success (peers connected), and non-NULL on - * failure (peers failed to connect). - */ -static void -startup_done (void *cls, const char *emsg) -{ - if (emsg != NULL) - { - FPRINTF (stderr, "Failed to setup topology: %s\n", emsg); - die_task = GNUNET_SCHEDULER_add_now (&end_badly, "topology setup failed"); - return; - } - die_task = - GNUNET_SCHEDULER_add_delayed (START_DELAY, &run_dht_test, - "from setup puts/gets"); -} - - -static void -run (void *cls, char *const *args, const char *cfgfile, - const struct GNUNET_CONFIGURATION_Handle *cfg) -{ - /* Get path from configuration file */ - if (GNUNET_YES != - GNUNET_CONFIGURATION_get_value_string (cfg, "paths", "servicehome", - &test_directory)) - { - GNUNET_break (0); - ok = 404; - return; - } - if (GNUNET_SYSERR == - GNUNET_CONFIGURATION_get_value_number (cfg, "testing", "num_peers", - &num_peers)) - num_peers = DEFAULT_NUM_PEERS; - pg = GNUNET_TESTING_peergroup_start (cfg, num_peers, TIMEOUT, NULL, - &startup_done, NULL, NULL); - GNUNET_assert (NULL != pg); - shutdown_task = GNUNET_SCHEDULER_add_delayed(GNUNET_TIME_UNIT_FOREVER_REL, - &do_stop, NULL); -} - - -static int -check () -{ - int ret; - - /* Arguments for GNUNET_PROGRAM_run */ - char *const argv[] = { "test-dht-multipeer", /* Name to give running binary */ - "-c", - "test_dht_multipeer_data.conf", /* Config file to use */ - NULL - }; - struct GNUNET_GETOPT_CommandLineOption options[] = { - GNUNET_GETOPT_OPTION_END - }; - /* Run the run function as a new program */ - ret = - GNUNET_PROGRAM_run ((sizeof (argv) / sizeof (char *)) - 1, argv, - "test-dht-multipeer", "nohelp", options, &run, &ok); - if (ret != GNUNET_OK) - { - GNUNET_log (GNUNET_ERROR_TYPE_WARNING, - "`test-dht-multipeer': Failed with error code %d\n", ret); - } - return ok; -} - - -int -main (int argc, char *argv[]) -{ - int ret; - - GNUNET_log_setup ("test-dht-multipeer", - "WARNING", - NULL); - ret = check (); - /** - * Need to remove base directory, subdirectories taken care - * of by the testing framework. - */ - if (GNUNET_DISK_directory_remove (test_directory) != GNUNET_OK) - { - GNUNET_log (GNUNET_ERROR_TYPE_WARNING, - "Failed to remove testing directory %s\n", test_directory); - } - return ret; -} - -/* end of test_dht_multipeer.c */ |