diff options
Diffstat (limited to 'src/testing/test_testing_reconnect.c')
| -rw-r--r-- | src/testing/test_testing_reconnect.c | 249 |
1 files changed, 249 insertions, 0 deletions
diff --git a/src/testing/test_testing_reconnect.c b/src/testing/test_testing_reconnect.c new file mode 100644 index 0000000..bcee386 --- /dev/null +++ b/src/testing/test_testing_reconnect.c @@ -0,0 +1,249 @@ +/* + This file is part of GNUnet. + (C) 2010 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 testing/test_testing_reconnect.c + * @brief testcase for functions to connect two peers in testing.c + */ +#include "platform.h" +#include "gnunet_testing_lib.h" + +#define VERBOSE GNUNET_YES + +/** + * How long until we give up on connecting the peers? + */ +#define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 300) + +#define CONNECT_ATTEMPTS 3 + +static int ok; + +static struct GNUNET_TESTING_Daemon *d1; + +static struct GNUNET_TESTING_Daemon *d2; + +static struct GNUNET_CONFIGURATION_Handle *c1; + +static struct GNUNET_CONFIGURATION_Handle *c2; + +static struct GNUNET_TESTING_ConnectContext *cc; + +/** + * How many start-connect-stop iterations should we do? + */ +#define NUM_PHASES 2 + +static int phase; + +/** + * Run the next phase of starting daemons, connecting them and + * stopping them again. + */ +static void +run_phase (void); + +static void +end2_cb (void *cls, const char *emsg) +{ + + if (emsg != NULL) + { + GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Ending with error: %s\n", emsg); + ok = 1; + } + else + { + if (phase < NUM_PHASES) + { + FPRINTF (stderr, "%s", "."); + run_phase (); + return; + } + FPRINTF (stderr, "%s", ".\n"); +#if VERBOSE + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, + "Both daemons terminated, will now exit.\n"); +#endif + ok = 0; + } +} + +static void +end1_cb (void *cls, const char *emsg) +{ + if (emsg != NULL) + { + GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Stopping daemon 1 gave: %s\n", + emsg); + ok = 1; + } + else + { + ok = 0; + } + if (d2 != NULL) + { + GNUNET_TESTING_daemon_stop (d2, TIMEOUT, &end2_cb, NULL, + (phase == NUM_PHASES) ? GNUNET_YES : GNUNET_NO, + GNUNET_NO); + d2 = NULL; + } +} + +static void +finish_testing (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) +{ + GNUNET_TESTING_daemon_stop (d1, TIMEOUT, &end1_cb, NULL, + (phase == NUM_PHASES) ? GNUNET_YES : GNUNET_NO, + GNUNET_NO); + d1 = NULL; +} + + +static void +my_connect_complete (void *cls, const struct GNUNET_PeerIdentity *first, + const struct GNUNET_PeerIdentity *second, + unsigned int distance, + const struct GNUNET_CONFIGURATION_Handle *first_cfg, + const struct GNUNET_CONFIGURATION_Handle *second_cfg, + struct GNUNET_TESTING_Daemon *first_daemon, + struct GNUNET_TESTING_Daemon *second_daemon, + const char *emsg) +{ + cc = NULL; +#if VERBOSE + FPRINTF (stderr, "Peer %s ", GNUNET_i2s (first)); + FPRINTF (stderr, "connected to %s\n", GNUNET_i2s (second)); +#endif + GNUNET_SCHEDULER_add_now (&finish_testing, NULL); +} + + + + +static void +my_cb2 (void *cls, const struct GNUNET_PeerIdentity *id, + const struct GNUNET_CONFIGURATION_Handle *cfg, + struct GNUNET_TESTING_Daemon *d, const char *emsg) +{ + if (emsg != NULL) + { + GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Starting daemon 2 gave: %s\n", + emsg); + GNUNET_assert (0); + return; + } + GNUNET_assert (id != NULL); +#if VERBOSE + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Daemon `%s' started.\n", + GNUNET_i2s (id)); +#endif + cc = GNUNET_TESTING_daemons_connect (d1, d2, TIMEOUT, CONNECT_ATTEMPTS, + GNUNET_YES, &my_connect_complete, NULL); +} + + +static void +my_cb1 (void *cls, const struct GNUNET_PeerIdentity *id, + const struct GNUNET_CONFIGURATION_Handle *cfg, + struct GNUNET_TESTING_Daemon *d, const char *emsg) +{ + if (emsg != NULL) + { + GNUNET_log (GNUNET_ERROR_TYPE_WARNING, "Starting daemon 1 gave: %s\n", + emsg); + GNUNET_assert (0); + return; + } + GNUNET_assert (id != NULL); +#if VERBOSE + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Daemon `%s' started.\n", + GNUNET_i2s (id)); +#endif + d2 = GNUNET_TESTING_daemon_start (c2, TIMEOUT, GNUNET_NO, NULL, NULL, 0, NULL, + NULL, NULL, &my_cb2, NULL); + GNUNET_assert (d2 != NULL); +} + + +static void +run (void *cls, char *const *args, const char *cfgfile, + const struct GNUNET_CONFIGURATION_Handle *cfg) +{ + ok = 1; +#if VERBOSE + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Starting daemon.\n"); +#endif + c1 = GNUNET_CONFIGURATION_create (); + GNUNET_assert (GNUNET_OK == + GNUNET_CONFIGURATION_load (c1, + "test_testing_connect_peer1.conf")); + c2 = GNUNET_CONFIGURATION_create (); + GNUNET_assert (GNUNET_OK == + GNUNET_CONFIGURATION_load (c2, + "test_testing_connect_peer2.conf")); + run_phase (); +} + +static void +run_phase () +{ + phase++; + d1 = GNUNET_TESTING_daemon_start (c1, TIMEOUT, GNUNET_NO, NULL, NULL, 0, NULL, + NULL, NULL, &my_cb1, NULL); + GNUNET_assert (d1 != NULL); +} + +static int +check () +{ + char *const argv[] = { "test-testing-reconnect", + "-c", + "test_testing_data.conf", +#if VERBOSE + "-L", "DEBUG", +#endif + NULL + }; + struct GNUNET_GETOPT_CommandLineOption options[] = { + GNUNET_GETOPT_OPTION_END + }; + GNUNET_PROGRAM_run ((sizeof (argv) / sizeof (char *)) - 1, argv, + "test-testing-reconnect", "nohelp", options, &run, &ok); + return ok; +} + +int +main (int argc, char *argv[]) +{ + int ret; + + GNUNET_log_setup ("test-testing-reconnect", +#if VERBOSE + "DEBUG", +#else + "WARNING", +#endif + NULL); + ret = check (); + return ret; +} + +/* end of test_testing_reconnect.c */ |