From 9c841f1b0de6569cef5047e8ee49f47f808b3eae Mon Sep 17 00:00:00 2001 From: Matthias Wachs Date: Thu, 28 Mar 2013 12:14:50 +0000 Subject: modified test --- src/transport/test_transport_blacklisting.c | 338 ++++++++++++++++++++++++++++ 1 file changed, 338 insertions(+) create mode 100644 src/transport/test_transport_blacklisting.c (limited to 'src/transport/test_transport_blacklisting.c') diff --git a/src/transport/test_transport_blacklisting.c b/src/transport/test_transport_blacklisting.c new file mode 100644 index 000000000..8423c641e --- /dev/null +++ b/src/transport/test_transport_blacklisting.c @@ -0,0 +1,338 @@ +/* + This file is part of GNUnet. + (C) 2009, 2010, 2011 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 transport/transport_api_blacklisting.c + * @brief test for the blacklisting API + * stage 0: init + * stage 1: connect peers and stop + * stage 2: blacklist whole peer and connect + * stage 3: blacklist tcp and try connect + * + * @author Matthias Wachs + * + */ +#include "platform.h" +#include "gnunet_transport_service.h" +#include "transport-testing.h" + +char *test_name; + +struct PeerContext *p1; + +struct PeerContext *p2; + +static GNUNET_TRANSPORT_TESTING_ConnectRequest cc; + +struct GNUNET_TRANSPORT_TESTING_handle *tth; + +/** + * How long until we give up on transmitting the message? + */ +#define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 20) + +#define CONNECT_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 10) + + +static int stage; +static int ok; +static int connected; + +static GNUNET_SCHEDULER_TaskIdentifier die_task; + +static GNUNET_SCHEDULER_TaskIdentifier timeout_task; + +static GNUNET_SCHEDULER_TaskIdentifier stage_task; + +#if VERBOSE +#define OKPP do { ok++; FPRINTF (stderr, "Now at stage %u at %s:%u\n", ok, __FILE__, __LINE__); } while (0) +#else +#define OKPP do { ok++; } while (0) +#endif + +static void +run_stage (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc); + +static void +end (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) +{ + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Stopping\n"); + + if (die_task != GNUNET_SCHEDULER_NO_TASK) + { + GNUNET_SCHEDULER_cancel (die_task); + die_task = GNUNET_SCHEDULER_NO_TASK; + } + + if (timeout_task != GNUNET_SCHEDULER_NO_TASK) + { + GNUNET_SCHEDULER_cancel (timeout_task); + timeout_task = GNUNET_SCHEDULER_NO_TASK; + } + + if (stage_task != GNUNET_SCHEDULER_NO_TASK) + { + GNUNET_SCHEDULER_cancel (stage_task); + stage_task = GNUNET_SCHEDULER_NO_TASK; + } + + if (cc != NULL) + { + GNUNET_TRANSPORT_TESTING_connect_peers_cancel(tth, cc); + cc = NULL; + } + + if (p1 != NULL) + { + GNUNET_TRANSPORT_TESTING_stop_peer (tth, p1); + p1 = NULL; + } + if (p2 != NULL) + { + GNUNET_TRANSPORT_TESTING_stop_peer (tth, p2); + p2 = NULL; + } +} + +static void +end_badly (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) +{ + die_task = GNUNET_SCHEDULER_NO_TASK; + + if (timeout_task != GNUNET_SCHEDULER_NO_TASK) + { + GNUNET_SCHEDULER_cancel (timeout_task); + timeout_task = GNUNET_SCHEDULER_NO_TASK; + } + + if (stage_task != GNUNET_SCHEDULER_NO_TASK) + { + GNUNET_SCHEDULER_cancel (stage_task); + stage_task = GNUNET_SCHEDULER_NO_TASK; + } + + + if (cc != NULL) + { + GNUNET_TRANSPORT_TESTING_connect_peers_cancel(tth, cc); + cc = NULL; + } + if (p1 != NULL) + GNUNET_TRANSPORT_TESTING_stop_peer (tth, p1); + if (p2 != NULL) + GNUNET_TRANSPORT_TESTING_stop_peer (tth, p2); + + ok = GNUNET_SYSERR; +} + +static void +testing_connect_cb (struct PeerContext *p1, struct PeerContext *p2, void *cls) +{ + cc = NULL; + char *p1_c = GNUNET_strdup (GNUNET_i2s (&p1->id)); + + GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Peers connected: %u (%s) <-> %u (%s)\n", + p1->no, p1_c, p2->no, GNUNET_i2s (&p2->id)); + GNUNET_free (p1_c); + connected = GNUNET_YES; + stage_task = GNUNET_SCHEDULER_add_now (&run_stage, NULL); +} + +static void +connect_timeout (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) +{ + GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Peers not connected, next stage\n"); + timeout_task = GNUNET_SCHEDULER_NO_TASK; + stage_task = GNUNET_SCHEDULER_add_now (&run_stage, NULL); +} + +static int started; + +void +start_cb (struct PeerContext *p, void *cls) +{ + + started++; + GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Peer %u (`%s') started\n", p->no, + GNUNET_i2s (&p->id)); + + if (started != 2) + return; + + char *sender_c = GNUNET_strdup (GNUNET_i2s (&p1->id)); + + GNUNET_log (GNUNET_ERROR_TYPE_ERROR, + "Test tries to connect peer %u (`%s') -> peer %u (`%s')\n", + p1->no, sender_c, p2->no, GNUNET_i2s (&p2->id)); + GNUNET_free (sender_c); + + cc = GNUNET_TRANSPORT_TESTING_connect_peers (tth, p1, p2, &testing_connect_cb, + NULL); + +} + +static void +run_stage (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) +{ + stage_task = GNUNET_SCHEDULER_NO_TASK; + if (GNUNET_SCHEDULER_NO_TASK != die_task) + GNUNET_SCHEDULER_cancel (die_task); + die_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT, &end_badly, NULL); + GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Running stage %u\n", stage); + + if (0 == stage) + { + started = GNUNET_NO; + connected = GNUNET_NO; + if (0 == strcmp(test_name, "test_transport_blacklisting_no_bl")) + { + /* Try to connect peers successfully */ + p1 = GNUNET_TRANSPORT_TESTING_start_peer (tth, "test_transport_blacklisting_cfg_peer1.conf", 1, + NULL, NULL, NULL, &start_cb, NULL); + + p2 = GNUNET_TRANSPORT_TESTING_start_peer (tth, "test_transport_blacklisting_cfg_peer2.conf", 2, + NULL, NULL, NULL, &start_cb, NULL); + } + else if (0 == strcmp(test_name, "test_transport_blacklisting_outbound_bl_full")) + { + p1 = GNUNET_TRANSPORT_TESTING_start_peer (tth, "test_transport_blacklisting_cfg_blp_peer1_full.conf", 1, + NULL, NULL, NULL, &start_cb, NULL); + + p2 = GNUNET_TRANSPORT_TESTING_start_peer (tth, "test_transport_blacklisting_cfg_peer2.conf", 2, + NULL, NULL, NULL, &start_cb, NULL); + } + else if (0 == strcmp(test_name, "test_transport_blacklisting_outbound_bl_plugin")) + { + p1 = GNUNET_TRANSPORT_TESTING_start_peer (tth, "test_transport_blacklisting_cfg_blp_peer1_plugin.conf", 1, + NULL, NULL, NULL, &start_cb, NULL); + + p2 = GNUNET_TRANSPORT_TESTING_start_peer (tth, "test_transport_blacklisting_cfg_peer2.conf", 2, + NULL, NULL, NULL, &start_cb, NULL); + } + else if (0 == strcmp(test_name, "test_transport_blacklisting_inbound_bl_full")) + { + p1 = GNUNET_TRANSPORT_TESTING_start_peer (tth, "test_transport_blacklisting_cfg_peer1.conf", 1, + NULL, NULL, NULL, &start_cb, NULL); + + p2 = GNUNET_TRANSPORT_TESTING_start_peer (tth, "test_transport_blacklisting_cfg_blp_peer2_full.conf", 2, + NULL, NULL, NULL, &start_cb, NULL); + } + else if (0 == strcmp(test_name, "test_transport_blacklisting_inbound_bl_plugin")) + { + p1 = GNUNET_TRANSPORT_TESTING_start_peer (tth, "test_transport_blacklisting_cfg_peer1.conf", 1, + NULL, NULL, NULL, &start_cb, NULL); + + p2 = GNUNET_TRANSPORT_TESTING_start_peer (tth, "test_transport_blacklisting_cfg_blp_peer2_plugin.conf", 2, + NULL, NULL, NULL, &start_cb, NULL); + } + + timeout_task = GNUNET_SCHEDULER_add_delayed (CONNECT_TIMEOUT, &connect_timeout, NULL); + stage ++; + return; + } + + + if (cc != NULL) + { + GNUNET_TRANSPORT_TESTING_connect_peers_cancel(tth, cc); + cc = NULL; + } + + if (p1 != NULL) + { + GNUNET_TRANSPORT_TESTING_stop_peer (tth, p1); + p1 = NULL; + } + if (p2 != NULL) + { + GNUNET_TRANSPORT_TESTING_stop_peer (tth, p2); + p2 = NULL; + } + + GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Done in stage %u: Peers %s and %s!\n", stage, + (GNUNET_NO == started) ? "NOT STARTED" : "STARTED", + (GNUNET_YES == connected) ? "CONNECTED" : "NOT CONNECTED"); + + if (0 == strcmp(test_name, "test_transport_blacklisting_no_bl")) + { + if ((GNUNET_NO != started) && (GNUNET_YES == connected)) + ok = 0; + else + { + GNUNET_break (0); + ok = 1; + } + } + else + { + if ((GNUNET_NO != started) && (GNUNET_YES != connected)) + ok = 0; + else + { + ok = 1; + } + } + GNUNET_SCHEDULER_add_now (&end, NULL); +} + +static void +run (void *cls, char *const *args, const char *cfgfile, + const struct GNUNET_CONFIGURATION_Handle *cfg) +{ + connected = GNUNET_NO; + stage = 0; + GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Running test `%s'!\n", test_name); + stage_task = GNUNET_SCHEDULER_add_now (&run_stage, NULL); +} + + +int +main (int argc, char *argv0[]) +{ + ok = 1; + + GNUNET_TRANSPORT_TESTING_get_test_name (argv0[0], &test_name); + + GNUNET_log_setup ("test-transport-api-blacklisting", + "WARNING", + NULL); + + static char *const argv[] = { "date", + "-c", + "test_transport_api_data.conf", + NULL + }; + static struct GNUNET_GETOPT_CommandLineOption options[] = { + GNUNET_GETOPT_OPTION_END + }; + + tth = GNUNET_TRANSPORT_TESTING_init (); + + GNUNET_PROGRAM_run ((sizeof (argv) / sizeof (char *)) - 1, argv, + "test-transport-api-blacklisting", + "nohelp", options, &run, NULL); + + + GNUNET_TRANSPORT_TESTING_done (tth); + + return ok; +} + +/* end of transport_api_blacklisting.c */ -- cgit v1.2.3