123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325 |
- /*
- This file is part of GNUnet.
- Copyright (C) 2008--2013 GNUnet e.V.
- 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., 51 Franklin Street, Fifth Floor,
- Boston, MA 02110-1301, USA.
- */
- /**
- * @file testbed/gnunet-testbed-profiler.c
- * @brief Profiling driver for the testbed.
- * @author Sree Harsha Totakura <sreeharsha@totakura.in>
- */
- #include "platform.h"
- #include "gnunet_util_lib.h"
- #include "gnunet_testbed_service.h"
- #include "testbed_api_hosts.h"
- /**
- * Generic loggins shorthand
- */
- #define LOG(kind,...) \
- GNUNET_log (kind, __VA_ARGS__)
- /**
- * Handle to global configuration
- */
- struct GNUNET_CONFIGURATION_Handle *cfg;
- /**
- * Peer linking - topology operation
- */
- struct GNUNET_TESTBED_Operation *topology_op;
- /**
- * Name of the file with the hosts to run the test over (configuration option).
- * It will be NULL if ENABLE_LL is set
- */
- static char *hosts_file;
- /**
- * Abort task identifier
- */
- static struct GNUNET_SCHEDULER_Task *abort_task;
- /**
- * Global event mask for all testbed events
- */
- uint64_t event_mask;
- /**
- * Number of peers to be started by the profiler
- */
- static unsigned int num_peers;
- /**
- * Number of timeout failures to tolerate
- */
- static unsigned int num_cont_fails;
- /**
- * Continuous failures during overlay connect operations
- */
- static unsigned int cont_fails;
- /**
- * Links which are successfully established
- */
- static unsigned int established_links;
- /**
- * Links which are not successfully established
- */
- static unsigned int failed_links;
- /**
- * Global testing status
- */
- static int result;
- /**
- * Are we running non interactively
- */
- static int noninteractive;
- /**
- * Shutdown nicely
- *
- * @param cls NULL
- */
- static void
- do_shutdown (void *cls)
- {
- if (NULL != abort_task)
- {
- GNUNET_SCHEDULER_cancel (abort_task);
- abort_task = NULL;
- }
- if (NULL != cfg)
- {
- GNUNET_CONFIGURATION_destroy (cfg);
- cfg = NULL;
- }
- }
- /**
- * abort task to run on test timed out
- *
- * @param cls NULL
- */
- static void
- do_abort (void *cls)
- {
- abort_task = NULL;
- LOG (GNUNET_ERROR_TYPE_WARNING,
- "Aborting\n");
- result = GNUNET_SYSERR;
- GNUNET_SCHEDULER_shutdown ();
- }
- /**
- * Function to print summary about how many overlay links we have made and how
- * many failed
- */
- static void
- print_overlay_links_summary ()
- {
- static int printed_already;
- if (GNUNET_YES == printed_already)
- return;
- printed_already = GNUNET_YES;
- printf ("%u links succeeded\n", established_links);
- printf ("%u links failed due to timeouts\n", failed_links);
- }
- /**
- * Controller event callback
- *
- * @param cls NULL
- * @param event the controller event
- */
- static void
- controller_event_cb (void *cls,
- const struct GNUNET_TESTBED_EventInformation *event)
- {
- switch (event->type)
- {
- case GNUNET_TESTBED_ET_OPERATION_FINISHED:
- /* Control reaches here when a peer linking operation fails */
- if (NULL != event->details.operation_finished.emsg)
- {
- printf ("F");
- fflush (stdout);
- failed_links++;
- if (++cont_fails > num_cont_fails)
- {
- printf ("\nAborting due to very high failure rate\n");
- print_overlay_links_summary ();
- if (NULL != abort_task)
- GNUNET_SCHEDULER_cancel (abort_task);
- abort_task = GNUNET_SCHEDULER_add_now (&do_abort, NULL);
- return;
- }
- }
- break;
- case GNUNET_TESTBED_ET_CONNECT:
- {
- if (0 != cont_fails)
- cont_fails--;
- if (0 == established_links)
- printf ("Establishing links. Please wait\n");
- printf (".");
- fflush (stdout);
- established_links++;
- }
- break;
- default:
- GNUNET_break (0);
- }
- }
- /**
- * Signature of a main function for a testcase.
- *
- * @param cls closure
- * @param h the run handle
- * @param num_peers number of peers in 'peers'
- * @param peers handle to peers run in the testbed
- * @param links_succeeded the number of overlay link connection attempts that
- * succeeded
- * @param links_failed the number of overlay link
- */
- static void
- test_run (void *cls,
- struct GNUNET_TESTBED_RunHandle *h,
- unsigned int num_peers, struct GNUNET_TESTBED_Peer **peers,
- unsigned int links_succeeded,
- unsigned int links_failed)
- {
- result = GNUNET_OK;
- fprintf (stdout, "\n");
- print_overlay_links_summary ();
- GNUNET_SCHEDULER_add_shutdown (&do_shutdown, NULL);
- if (noninteractive)
- {
- GNUNET_SCHEDULER_cancel (abort_task);
- abort_task = NULL;
- return;
- }
- #if (!ENABLE_SUPERMUC)
- fprintf (stdout, "Testbed running, waiting for keystroke to shut down\n");
- fflush (stdout);
- (void) getc (stdin);
- #endif
- fprintf (stdout, "Shutting down. Please wait\n");
- fflush (stdout);
- GNUNET_SCHEDULER_shutdown ();
- }
- /**
- * Main function that will be run by the scheduler.
- *
- * @param cls closure
- * @param args remaining command-line arguments
- * @param cfgfile name of the configuration file used (for saving, can be NULL!)
- * @param config configuration
- */
- static void
- run (void *cls, char *const *args, const char *cfgfile,
- const struct GNUNET_CONFIGURATION_Handle *config)
- {
- if (0 == num_peers)
- {
- LOG (GNUNET_ERROR_TYPE_ERROR, _("Exiting as the number of peers is %u\n"),
- num_peers);
- return;
- }
- cfg = GNUNET_CONFIGURATION_dup (config);
- event_mask = 0;
- event_mask |= (1LL << GNUNET_TESTBED_ET_CONNECT);
- event_mask |= (1LL << GNUNET_TESTBED_ET_OPERATION_FINISHED);
- GNUNET_TESTBED_run (hosts_file, cfg, num_peers, event_mask, controller_event_cb,
- NULL, &test_run, NULL);
- abort_task =
- GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &do_abort,
- NULL);
- }
- /**
- * Main function.
- *
- * @return 0 on success
- */
- int
- main (int argc, char *const *argv)
- {
- struct GNUNET_GETOPT_CommandLineOption options[] = {
- GNUNET_GETOPT_option_uint ('p',
- "num-peers",
- "COUNT",
- gettext_noop ("create COUNT number of peers"),
- &num_peers),
- GNUNET_GETOPT_option_uint ('e',
- "num-errors",
- "COUNT",
- gettext_noop ("tolerate COUNT number of continious timeout failures"),
- &num_cont_fails),
- GNUNET_GETOPT_option_flag ('n',
- "non-interactive",
- gettext_noop ("run profiler in non-interactive mode where upon "
- "testbed setup the profiler does not wait for a "
- "keystroke but continues to run until a termination "
- "signal is received"),
- &noninteractive),
- #if !ENABLE_SUPERMUC
- GNUNET_GETOPT_option_string ('H',
- "hosts",
- "FILENAME",
- gettext_noop ("name of the file with the login information for the testbed"),
- &hosts_file),
- #endif
- GNUNET_GETOPT_OPTION_END
- };
- const char *binaryHelp = "gnunet-testbed-profiler [OPTIONS]";
- int ret;
- if (GNUNET_OK != GNUNET_STRINGS_get_utf8_args (argc, argv, &argc, &argv))
- return 2;
- result = GNUNET_SYSERR;
- ret =
- GNUNET_PROGRAM_run (argc, argv, "gnunet-testbed-profiler", binaryHelp,
- options, &run, NULL);
- GNUNET_free ((void *) argv);
- if (GNUNET_OK != ret)
- return ret;
- if (GNUNET_OK != result)
- return 1;
- return 0;
- }
|