/* This file is part of GNUnet. (C) 2009, 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 transport/test_transport_api.c * @brief base test case for transport implementations * * This test case serves as a base for tcp, udp, and udp-nat * transport test cases. Based on the executable being run * the correct test case will be performed. Conservation of * C code apparently. */ #include "platform.h" #include "gnunet_common.h" #include "gnunet_hello_lib.h" #include "gnunet_getopt_lib.h" #include "gnunet_os_lib.h" #include "gnunet_program_lib.h" #include "gnunet_scheduler_lib.h" #include "gnunet_transport_service.h" #include "transport.h" #include "transport-testing.h" #define VERBOSE GNUNET_NO #define VERBOSE_ARM GNUNET_NO #define START_ARM GNUNET_YES /** * How long until we give up on transmitting the message? */ #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30) /** * How long until we give up on transmitting the message? */ #define TIMEOUT_TRANSMIT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 60) #define MTYPE 12345 static struct PeerContext p1; static struct PeerContext p2; static int ok; static int is_tcp; static GNUNET_SCHEDULER_TaskIdentifier die_task; static GNUNET_SCHEDULER_TaskIdentifier tct; #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 stop_arm (struct PeerContext *p) { #if START_ARM if (NULL != p->arm_proc) { if (0 != GNUNET_OS_process_kill (p->arm_proc, SIGTERM)) GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "kill"); GNUNET_OS_process_wait (p->arm_proc); GNUNET_OS_process_close (p->arm_proc); p->arm_proc = NULL; } #endif GNUNET_CONFIGURATION_destroy (p->cfg); } static void exchange_hello_last (void *cls, const struct GNUNET_MessageHeader *message) { struct PeerContext *me = cls; GNUNET_assert (message != NULL); GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Exchanging HELLO of size %d with peer (%s)!\n", (int) GNUNET_HELLO_size((const struct GNUNET_HELLO_Message *)message), GNUNET_i2s (&me->id)); GNUNET_assert (GNUNET_OK == GNUNET_HELLO_get_id ((const struct GNUNET_HELLO_Message *) message, &me->id)); GNUNET_TRANSPORT_offer_hello (p1.th, message, NULL, NULL); } static void exchange_hello (void *cls, const struct GNUNET_MessageHeader *message) { struct PeerContext *me = cls; GNUNET_assert (message != NULL); GNUNET_assert (GNUNET_OK == GNUNET_HELLO_get_id ((const struct GNUNET_HELLO_Message *) message, &me->id)); GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Exchanging HELLO of size %d from peer %s!\n", (int) GNUNET_HELLO_size((const struct GNUNET_HELLO_Message *)message), GNUNET_i2s (&me->id)); GNUNET_TRANSPORT_offer_hello (p2.th, message, NULL, NULL); } static void end_badly (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) { GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Disconnecting from transports!\n"); GNUNET_break (0); die_task = GNUNET_SCHEDULER_NO_TASK; GNUNET_TRANSPORT_get_hello_cancel (p2.th, &exchange_hello_last, &p2); GNUNET_TRANSPORT_get_hello_cancel (p1.th, &exchange_hello, &p1); GNUNET_TRANSPORT_disconnect (p1.th); GNUNET_TRANSPORT_disconnect (p2.th); if (GNUNET_SCHEDULER_NO_TASK != tct) { GNUNET_SCHEDULER_cancel (tct); tct = GNUNET_SCHEDULER_NO_TASK; } ok = 1; } static void end_normally (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) { ok = 60; die_task = GNUNET_SCHEDULER_NO_TASK; GNUNET_TRANSPORT_get_hello_cancel (p2.th, &exchange_hello_last, &p2); GNUNET_TRANSPORT_get_hello_cancel (p1.th, &exchange_hello, &p1); GNUNET_TRANSPORT_disconnect (p1.th); GNUNET_TRANSPORT_disconnect (p2.th); if (GNUNET_SCHEDULER_NO_TASK != tct) { GNUNET_SCHEDULER_cancel (tct); tct = GNUNET_SCHEDULER_NO_TASK; } ok = 0; } static void notify_connect (void *cls, const struct GNUNET_PeerIdentity *peer, const struct GNUNET_TRANSPORT_ATS_Information *ats, uint32_t ats_count) { GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Peer `%4s' connected to us (%p)!\n", GNUNET_i2s (peer), cls); if (cls == &p1) { GNUNET_assert (ok >= 2); OKPP; OKPP; if (GNUNET_SCHEDULER_NO_TASK != die_task) GNUNET_SCHEDULER_cancel (die_task); if (GNUNET_SCHEDULER_NO_TASK != tct) GNUNET_SCHEDULER_cancel (tct); tct = GNUNET_SCHEDULER_NO_TASK; die_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT_TRANSMIT, &end_normally, NULL); } } static void notify_disconnect (void *cls, const struct GNUNET_PeerIdentity *peer) { if ( (ok != 60) && (GNUNET_SCHEDULER_NO_TASK != die_task) ) { GNUNET_SCHEDULER_cancel (die_task); die_task = GNUNET_SCHEDULER_add_now (&end_badly, NULL); } GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Peer `%4s' disconnected (%p)!\n", GNUNET_i2s (peer), cls); } static void setup_peer (struct PeerContext *p, const char *cfgname) { p->cfg = GNUNET_CONFIGURATION_create (); GNUNET_assert (GNUNET_OK == GNUNET_CONFIGURATION_load (p->cfg, cfgname)); if (GNUNET_CONFIGURATION_have_value (p->cfg,"PATHS", "SERVICEHOME")) { GNUNET_assert (GNUNET_OK == GNUNET_CONFIGURATION_get_value_string (p->cfg, "PATHS", "SERVICEHOME", &p->servicehome)); GNUNET_DISK_directory_remove (p->servicehome); } #if START_ARM p->arm_proc = GNUNET_OS_start_process (NULL, NULL, "gnunet-service-arm", "gnunet-service-arm", #if VERBOSE_ARM if (is_https) { struct stat sbuf; if (0 == stat (cert_file_p1, &sbuf )) { if (0 == remove(cert_file_p1)) GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Successfully removed existing certificate file `%s'\n",cert_file_p1); else GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Failed to remove certfile `%s'\n",cert_file_p1); } if (0 == stat (key_file_p1, &sbuf )) { if (0 == remove(key_file_p1)) GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Successfully removed private key file `%s'\n",key_file_p1); else GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Failed to private key file `%s'\n",key_file_p1); } if (0 == stat (cert_file_p2, &sbuf )) { if (0 == remove(cert_file_p2)) GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Successfully removed existing certificate file `%s'\n",cert_file_p2); else GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Failed to remove certfile `%s'\n",cert_file_p2); } if (0 == stat (key_file_p2, &sbuf )) { if (0 == remove(key_file_p2)) GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Successfully removed private key file `%s'\n",key_file_p2); else GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Failed to private key file `%s'\n",key_file_p2); } GNUNET_free(key_file_p1); GNUNET_free(key_file_p2); GNUNET_free(cert_file_p1); GNUNET_free(cert_file_p2); } "-L", "DEBUG", #endif "-c", cfgname, NULL); #endif p->th = GNUNET_TRANSPORT_connect (p->cfg, NULL, p, NULL, ¬ify_connect, ¬ify_disconnect); GNUNET_assert (p->th != NULL); } static void try_connect (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) { GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Asking peers to connect...\n"); /* FIXME: 'pX.id' may still be all-zeros here... */ GNUNET_TRANSPORT_try_connect (p2.th, &p1.id); GNUNET_TRANSPORT_try_connect (p1.th, &p2.id); tct = GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_SECONDS, &try_connect, NULL); } static void run (void *cls, char *const *args, const char *cfgfile, const struct GNUNET_CONFIGURATION_Handle *cfg) { GNUNET_assert (ok == 1); OKPP; die_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT, &end_badly, NULL); if (is_tcp) { setup_peer (&p1, "test_transport_api_multiaddress_tcp_peer1.conf"); setup_peer (&p2, "test_transport_api_multiaddress_tcp_peer2.conf"); } GNUNET_assert(p1.th != NULL); GNUNET_assert(p2.th != NULL); GNUNET_TRANSPORT_get_hello (p1.th, &exchange_hello, &p1); GNUNET_TRANSPORT_get_hello (p2.th, &exchange_hello_last, &p2); tct = GNUNET_SCHEDULER_add_now (&try_connect, NULL); } static int check () { static char *const argv[] = { "test-transport-api", "-c", "test_transport_api_data.conf", #if VERBOSE "-L", "DEBUG", #endif NULL }; static struct GNUNET_GETOPT_CommandLineOption options[] = { GNUNET_GETOPT_OPTION_END }; #if WRITECONFIG setTransportOptions("test_transport_api_data.conf"); #endif ok = 1; GNUNET_PROGRAM_run ((sizeof (argv) / sizeof (char *)) - 1, argv, "test-transport-api", "nohelp", options, &run, &ok); stop_arm (&p1); stop_arm (&p2); if (p1.servicehome != NULL) { GNUNET_DISK_directory_remove (p1.servicehome); GNUNET_free(p1.servicehome); } if (p2.servicehome != NULL) { GNUNET_DISK_directory_remove (p2.servicehome); GNUNET_free(p2.servicehome); } return ok; } int main (int argc, char *argv[]) { int ret; GNUNET_log_setup ("test-transport-api", #if VERBOSE "DEBUG", #else "WARNING", #endif NULL); if (strstr(argv[0], "tcp") != NULL) { is_tcp = GNUNET_YES; } ret = check (); return ret; } /* end of test_transport_api.c */