123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541 |
- /*
- This file is part of GNUnet
- (C) 2009, 2010, 2011, 2012 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 hostlist/test_gnunet_daemon_hostlist_learning.c
- * @brief test for gnunet_daemon_hostslist.c
- * @author Christian Grothoff
- */
- #include "platform.h"
- #include "gnunet_util_lib.h"
- #include "gnunet_arm_service.h"
- #include "gnunet_core_service.h"
- #include "gnunet_transport_service.h"
- #include "gnunet_resolver_service.h"
- #include "gnunet_statistics_service.h"
- #define MAX_URL_LEN 1000
- /**
- * How long until wait until testcases fails
- */
- #define TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 180)
- #define CHECK_INTERVAL GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 1)
- struct PeerContext
- {
- struct GNUNET_CONFIGURATION_Handle *cfg;
- struct GNUNET_TRANSPORT_Handle *th;
- struct GNUNET_MessageHeader *hello;
- struct GNUNET_CORE_Handle *core;
- struct GNUNET_STATISTICS_Handle *stats;
- struct GNUNET_OS_Process *arm_proc;
- };
- static int timeout;
- static int adv_sent;
- static int adv_arrived;
- static int learned_hostlist_saved;
- static int learned_hostlist_downloaded;
- static char *current_adv_uri;
- static const struct GNUNET_CONFIGURATION_Handle *cfg;
- static GNUNET_SCHEDULER_TaskIdentifier timeout_task;
- static GNUNET_SCHEDULER_TaskIdentifier check_task;
- static struct PeerContext adv_peer;
- static struct PeerContext learn_peer;
- static struct GNUNET_STATISTICS_GetHandle *download_stats;
- static struct GNUNET_STATISTICS_GetHandle *urisrecv_stat;
- static struct GNUNET_STATISTICS_GetHandle *advsent_stat;
- static void
- shutdown_testcase ()
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Shutdown testcase....\n");
- if (timeout_task != GNUNET_SCHEDULER_NO_TASK)
- {
- GNUNET_SCHEDULER_cancel (timeout_task);
- timeout_task = GNUNET_SCHEDULER_NO_TASK;
- }
- if (NULL != download_stats)
- {
- GNUNET_STATISTICS_get_cancel (download_stats);
- download_stats = NULL;
- }
- if (NULL != urisrecv_stat)
- {
- GNUNET_STATISTICS_get_cancel (urisrecv_stat);
- urisrecv_stat = NULL;
- }
- if (NULL != advsent_stat)
- {
- GNUNET_STATISTICS_get_cancel (advsent_stat);
- advsent_stat = NULL;
- }
- if (NULL != adv_peer.stats)
- {
- GNUNET_STATISTICS_destroy (adv_peer.stats, GNUNET_NO);
- adv_peer.stats = NULL;
- }
- if (NULL != learn_peer.stats)
- {
- GNUNET_STATISTICS_destroy (learn_peer.stats, GNUNET_NO);
- learn_peer.stats = NULL;
- }
- if (check_task != GNUNET_SCHEDULER_NO_TASK)
- {
- GNUNET_SCHEDULER_cancel (check_task);
- check_task = GNUNET_SCHEDULER_NO_TASK;
- }
- if (NULL != current_adv_uri)
- {
- GNUNET_free (current_adv_uri);
- current_adv_uri = NULL;
- }
- if (adv_peer.th != NULL)
- {
- GNUNET_TRANSPORT_disconnect (adv_peer.th);
- adv_peer.th = NULL;
- }
- if (learn_peer.th != NULL)
- {
- GNUNET_TRANSPORT_disconnect (learn_peer.th);
- learn_peer.th = NULL;
- }
- if (adv_peer.core != NULL)
- {
- GNUNET_CORE_disconnect (adv_peer.core);
- adv_peer.core = NULL;
- }
- if (learn_peer.core != NULL)
- {
- GNUNET_CORE_disconnect (learn_peer.core);
- learn_peer.core = NULL;
- }
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Killing hostlist server ARM process.\n");
- if (0 != GNUNET_OS_process_kill (adv_peer.arm_proc, GNUNET_TERM_SIG))
- GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "kill");
- if (GNUNET_OS_process_wait (adv_peer.arm_proc) != GNUNET_OK)
- GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "waitpid");
- GNUNET_OS_process_destroy (adv_peer.arm_proc);
- adv_peer.arm_proc = NULL;
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Killing hostlist client ARM process.\n");
- if (0 != GNUNET_OS_process_kill (learn_peer.arm_proc, GNUNET_TERM_SIG))
- GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "kill");
- if (GNUNET_OS_process_wait (learn_peer.arm_proc) != GNUNET_OK)
- GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "waitpid");
- GNUNET_OS_process_destroy (learn_peer.arm_proc);
- learn_peer.arm_proc = NULL;
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Shutdown complete....\n");
- }
- /**
- * Timeout, give up.
- */
- static void
- timeout_error (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
- {
- timeout_task = GNUNET_SCHEDULER_NO_TASK;
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
- "Timeout while executing testcase, test failed.\n");
- timeout = GNUNET_YES;
- shutdown_testcase ();
- }
- static void
- process_downloads_done (void *cls, int success)
- {
- download_stats = NULL;
- }
- static void
- do_shutdown (void *cls,
- const struct GNUNET_SCHEDULER_TaskContext *tc)
- {
- shutdown_testcase ();
- }
- static int
- process_downloads (void *cls, const char *subsystem, const char *name,
- uint64_t value, int is_persistent)
- {
- if ((value >= 2) && (learned_hostlist_downloaded == GNUNET_NO))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Peer has successfully downloaded advertised URI\n");
- learned_hostlist_downloaded = GNUNET_YES;
- if ((learned_hostlist_saved == GNUNET_YES) && (adv_sent == GNUNET_YES))
- {
- GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
- }
- }
- return GNUNET_OK;
- }
- static void
- process_uris_recv_done (void *cls, int success)
- {
- urisrecv_stat = NULL;
- }
- static int
- process_uris_recv (void *cls, const char *subsystem, const char *name,
- uint64_t value, int is_persistent)
- {
- if (((struct PeerContext *) cls == &learn_peer) && (value == 1) &&
- (learned_hostlist_saved == GNUNET_NO))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Peer has successfully saved advertised URI\n");
- learned_hostlist_saved = GNUNET_YES;
- if ((learned_hostlist_downloaded == GNUNET_YES) && (adv_sent == GNUNET_YES))
- {
- GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
- }
- }
- return GNUNET_OK;
- }
- static void
- process_adv_sent_done (void *cls, int success)
- {
- advsent_stat = NULL;
- }
- static int
- process_adv_sent (void *cls, const char *subsystem, const char *name,
- uint64_t value, int is_persistent)
- {
- if ((value >= 1) && (adv_sent == GNUNET_NO))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Server has successfully sent advertisement\n");
- adv_sent = GNUNET_YES;
- if ((learned_hostlist_downloaded == GNUNET_YES) &&
- (learned_hostlist_saved == GNUNET_YES))
- {
- GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
- }
- }
- return GNUNET_OK;
- }
- /**
- * Check the server statistics regularly
- */
- static void
- check_statistics (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
- {
- char *stat;
- check_task = GNUNET_SCHEDULER_NO_TASK;
- if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
- return;
- GNUNET_asprintf (&stat, gettext_noop ("# advertised URI `%s' downloaded"),
- current_adv_uri);
- if (NULL != learn_peer.stats)
- {
- if (NULL != download_stats)
- GNUNET_STATISTICS_get_cancel (download_stats);
- download_stats =
- GNUNET_STATISTICS_get (learn_peer.stats, "hostlist", stat,
- GNUNET_TIME_UNIT_MINUTES,
- &process_downloads_done, &process_downloads,
- &learn_peer);
- if (NULL != urisrecv_stat)
- GNUNET_STATISTICS_get_cancel (urisrecv_stat);
- urisrecv_stat =
- GNUNET_STATISTICS_get (learn_peer.stats, "hostlist",
- gettext_noop ("# advertised hostlist URIs"),
- GNUNET_TIME_UNIT_MINUTES,
- &process_uris_recv_done, &process_uris_recv,
- &learn_peer);
- }
- GNUNET_free (stat);
- if (NULL != adv_peer.stats)
- {
- if (NULL != advsent_stat)
- GNUNET_STATISTICS_get_cancel (advsent_stat);
- advsent_stat =
- GNUNET_STATISTICS_get (adv_peer.stats, "hostlist",
- gettext_noop ("# hostlist advertisements send"),
- GNUNET_TIME_UNIT_MINUTES, &process_adv_sent_done,
- &process_adv_sent, NULL);
- }
- check_task =
- GNUNET_SCHEDULER_add_delayed (CHECK_INTERVAL, &check_statistics, NULL);
- }
- /**
- * Core handler for p2p hostlist advertisements
- */
- static int
- ad_arrive_handler (void *cls, const struct GNUNET_PeerIdentity *peer,
- const struct GNUNET_MessageHeader *message)
- {
- char *hostname;
- char *expected_uri;
- unsigned long long port;
- const struct GNUNET_MessageHeader *incoming;
- const char *end;
- if (-1 ==
- GNUNET_CONFIGURATION_get_value_number (adv_peer.cfg, "HOSTLIST",
- "HTTPPORT", &port))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
- "Could not read advertising server's configuration\n");
- return GNUNET_SYSERR;
- }
- if (GNUNET_SYSERR ==
- GNUNET_CONFIGURATION_get_value_string (adv_peer.cfg, "HOSTLIST",
- "EXTERNAL_DNS_NAME", &hostname))
- hostname = GNUNET_RESOLVER_local_fqdn_get ();
- GNUNET_asprintf (&expected_uri, "http://%s:%u/",
- hostname != NULL ? hostname : "localhost",
- (unsigned int) port);
- incoming = (const struct GNUNET_MessageHeader *) message;
- end = (const char *) &incoming[1];
- if ('\0' !=
- end[ntohs (message->size) - sizeof (struct GNUNET_MessageHeader) - 1])
- {
- GNUNET_break (0);
- GNUNET_free (expected_uri);
- GNUNET_free_non_null (hostname);
- return GNUNET_SYSERR;
- }
- current_adv_uri = GNUNET_strdup (end);
- if (0 == strcmp (expected_uri, current_adv_uri))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Received hostlist advertisement with URI `%s' as expected\n",
- current_adv_uri);
- adv_arrived = GNUNET_YES;
- adv_sent = GNUNET_YES;
- }
- else
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
- "Expected URI `%s' and received URI `%s' differ\n",
- expected_uri, current_adv_uri);
- GNUNET_free (expected_uri);
- GNUNET_free_non_null (hostname);
- return GNUNET_OK;
- }
- /**
- * List of handlers if we are learning.
- */
- static struct GNUNET_CORE_MessageHandler learn_handlers[] = {
- {&ad_arrive_handler, GNUNET_MESSAGE_TYPE_HOSTLIST_ADVERTISEMENT, 0},
- {NULL, 0, 0}
- };
- static void
- setup_learn_peer (struct PeerContext *p, const char *cfgname)
- {
- char *filename;
- unsigned int result;
- char *binary;
- binary = GNUNET_OS_get_libexec_binary_path ("gnunet-service-arm");
- p->cfg = GNUNET_CONFIGURATION_create ();
- p->arm_proc =
- GNUNET_OS_start_process (GNUNET_YES, GNUNET_OS_INHERIT_STD_OUT_AND_ERR,
- NULL, NULL, NULL,
- binary,
- "gnunet-service-arm",
- "-c", cfgname, NULL);
- GNUNET_assert (GNUNET_OK == GNUNET_CONFIGURATION_load (p->cfg, cfgname));
- if (GNUNET_OK ==
- GNUNET_CONFIGURATION_get_value_string (p->cfg, "HOSTLIST", "HOSTLISTFILE",
- &filename))
- {
- if (GNUNET_YES == GNUNET_DISK_file_test (filename))
- {
- result = UNLINK (filename);
- if (result == 0)
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- _("Hostlist file `%s' was removed\n"), filename);
- }
- GNUNET_free (filename);
- }
- p->core =
- GNUNET_CORE_connect (p->cfg, NULL, NULL, NULL, NULL, NULL, GNUNET_NO,
- NULL, GNUNET_NO, learn_handlers);
- GNUNET_assert (NULL != p->core);
- p->stats = GNUNET_STATISTICS_create ("hostlist", p->cfg);
- GNUNET_assert (NULL != p->stats);
- GNUNET_free (binary);
- }
- static void
- setup_adv_peer (struct PeerContext *p, const char *cfgname)
- {
- char *binary;
- binary = GNUNET_OS_get_libexec_binary_path ("gnunet-service-arm");
- p->cfg = GNUNET_CONFIGURATION_create ();
- p->arm_proc =
- GNUNET_OS_start_process (GNUNET_YES, GNUNET_OS_INHERIT_STD_OUT_AND_ERR,
- NULL, NULL, NULL,
- binary,
- "gnunet-service-arm",
- "-c", cfgname, NULL);
- GNUNET_assert (GNUNET_OK == GNUNET_CONFIGURATION_load (p->cfg, cfgname));
- p->stats = GNUNET_STATISTICS_create ("hostlist", p->cfg);
- GNUNET_assert (NULL != p->stats);
- GNUNET_free (binary);
- }
- static void
- run (void *cls, char *const *args, const char *cfgfile,
- const struct GNUNET_CONFIGURATION_Handle *c)
- {
- timeout = GNUNET_NO;
- adv_sent = GNUNET_NO;
- adv_arrived = 0;
- learned_hostlist_saved = GNUNET_NO;
- learned_hostlist_downloaded = GNUNET_NO;
- cfg = c;
- setup_adv_peer (&adv_peer, "test_learning_adv_peer.conf");
- setup_learn_peer (&learn_peer, "test_learning_learn_peer.conf");
- timeout_task = GNUNET_SCHEDULER_add_delayed (TIMEOUT, &timeout_error, NULL);
- check_task =
- GNUNET_SCHEDULER_add_delayed (CHECK_INTERVAL, &check_statistics, NULL);
- }
- static int
- check ()
- {
- unsigned int failed;
- char *const argv[] = {
- "test-gnunet-daemon-hostlist-learning",
- "-c", "learning_data.conf",
- NULL
- };
- struct GNUNET_GETOPT_CommandLineOption options[] = {
- GNUNET_GETOPT_OPTION_END
- };
- GNUNET_PROGRAM_run ((sizeof (argv) / sizeof (char *)) - 1, argv,
- "test-gnunet-daemon-hostlist-learning", "nohelp", options,
- &run, NULL);
- failed = GNUNET_NO;
- if (timeout == GNUNET_YES)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Testcase timeout\n");
- failed = GNUNET_YES;
- }
- if (adv_arrived != GNUNET_YES)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
- "Learning peer did not receive advertisement from server\n");
- failed = GNUNET_YES;
- }
- if (learned_hostlist_saved == GNUNET_NO)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
- "Advertised hostlist was not saved in datastore\n");
- failed = GNUNET_YES;
- }
- if (learned_hostlist_downloaded == GNUNET_NO)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
- "Advertised hostlist could not be downloaded from server\n");
- failed = GNUNET_YES;
- }
- if (adv_sent == GNUNET_NO)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
- "Advertised was not sent from server to client\n");
- failed = GNUNET_YES;
- }
- if (GNUNET_YES == failed)
- return GNUNET_YES;
- return GNUNET_NO;
- }
- int
- main (int argc, char *argv[])
- {
- int ret;
- GNUNET_DISK_directory_remove ("/tmp/test-gnunet-hostlist-peer-1");
- GNUNET_DISK_directory_remove ("/tmp/test-gnunet-hostlist-peer-2");
- GNUNET_log_setup ("test-gnunet-daemon-hostlist",
- "WARNING",
- NULL);
- #if !WINDOWS
- system ("gnunet-peerinfo -s -c test_learning_adv_peer.conf > /dev/null");
- system ("gnunet-peerinfo -s -c test_learning_learn_peer.conf > /dev/null");
- #else
- system ("gnunet-peerinfo -s -c test_learning_adv_peer.conf > NUL");
- system ("gnunet-peerinfo -s -c test_learning_learn_peer.conf > NUL");
- #endif
- ret = check ();
- GNUNET_DISK_directory_remove ("/tmp/test-gnunet-hostlist-peer-1");
- GNUNET_DISK_directory_remove ("/tmp/test-gnunet-hostlist-peer-2");
- if (GNUNET_YES == GNUNET_DISK_file_test ("hostlists_learn_peer.file"))
- {
- if (0 == UNLINK ("hostlists_learn_peer.file"))
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- "Hostlist file hostlists_learn_peer.file was removed\n");
- }
- return ret;
- }
- /* end of test_gnunet_daemon_hostlist_learning.c */
|