123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534 |
- /*
- This file is part of GNUnet
- (C) 2008--2013 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 testbed/test_testbed_api_2peers_1controller.c
- * @brief testcases for the testbed api: 2 peers are configured, started and
- * connected together. The 2 peer reside on a single controller.
- * @author Sree Harsha Totakura
- */
- #include "platform.h"
- #include "gnunet_util_lib.h"
- #include "gnunet_testing_lib.h"
- #include "gnunet_testbed_service.h"
- /**
- * Generic logging shortcut
- */
- #define LOG(kind,...) \
- GNUNET_log (kind, __VA_ARGS__)
- /**
- * Relative time seconds shorthand
- */
- #define TIME_REL_SECS(sec) \
- GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, sec)
- /**
- * Peer context
- */
- struct PeerContext
- {
- /**
- * The peer handle
- */
- struct GNUNET_TESTBED_Peer *peer;
- /**
- * Operations involving this peer
- */
- struct GNUNET_TESTBED_Operation *operation;
- /**
- * set to GNUNET_YES when peer is started
- */
- int is_running;
- };
- /**
- * Our localhost
- */
- static struct GNUNET_TESTBED_Host *host;
- /**
- * The controller process
- */
- static struct GNUNET_TESTBED_ControllerProc *cp;
- /**
- * The controller handle
- */
- static struct GNUNET_TESTBED_Controller *controller;
- /**
- * A neighbouring host
- */
- static struct GNUNET_TESTBED_Host *neighbour;
- /**
- * Handle for neighbour registration
- */
- static struct GNUNET_TESTBED_HostRegistrationHandle *reg_handle;
- /**
- * peer 1
- */
- static struct PeerContext peer1;
- /**
- * peer2
- */
- static struct PeerContext peer2;
- /**
- * Handle to configuration
- */
- static struct GNUNET_CONFIGURATION_Handle *cfg;
- /**
- * Handle to operations involving both peers
- */
- static struct GNUNET_TESTBED_Operation *common_operation;
- /**
- * Abort task identifier
- */
- static GNUNET_SCHEDULER_TaskIdentifier abort_task;
- /**
- * Delayed connect job identifier
- */
- static GNUNET_SCHEDULER_TaskIdentifier delayed_connect_task;
- /**
- * Different stages in testing
- */
- enum Stage
- {
- /**
- * Initial stage
- */
- INIT,
- /**
- * peers are created
- */
- PEERS_CREATED,
- /**
- * peers are started
- */
- PEERS_STARTED,
- /**
- * peers are connected
- */
- PEERS_CONNECTED,
- /**
- * Peers are connected once again (this should not fail as they are already connected)
- */
- PEERS_CONNECTED_2,
- /**
- * peers are stopped
- */
- PEERS_STOPPED,
- /**
- * Final success stage
- */
- SUCCESS
- };
- /**
- * The testing result
- */
- static enum Stage result;
- /**
- * shortcut to exit during failure
- */
- #define FAIL_TEST(cond) do { \
- if (!(cond)) { \
- GNUNET_break(0); \
- if (GNUNET_SCHEDULER_NO_TASK != abort_task) \
- GNUNET_SCHEDULER_cancel (abort_task); \
- abort_task = GNUNET_SCHEDULER_NO_TASK; \
- GNUNET_SCHEDULER_add_now (do_shutdown, NULL); \
- return; \
- } \
- } while (0)
- /**
- * Shutdown nicely
- *
- * @param cls NULL
- * @param tc the task context
- */
- static void
- do_shutdown (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
- {
- if (GNUNET_SCHEDULER_NO_TASK != abort_task)
- GNUNET_SCHEDULER_cancel (abort_task);
- if (GNUNET_SCHEDULER_NO_TASK != delayed_connect_task)
- GNUNET_SCHEDULER_cancel (delayed_connect_task);
- if (NULL != reg_handle)
- GNUNET_TESTBED_cancel_registration (reg_handle);
- GNUNET_TESTBED_controller_disconnect (controller);
- GNUNET_CONFIGURATION_destroy (cfg);
- if (NULL != cp)
- GNUNET_TESTBED_controller_stop (cp);
- GNUNET_TESTBED_host_destroy (neighbour);
- GNUNET_TESTBED_host_destroy (host);
- }
- /**
- * abort task to run on test timed out
- *
- * @param cls NULL
- * @param tc the task context
- */
- static void
- do_abort (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
- {
- LOG (GNUNET_ERROR_TYPE_WARNING, "Test timedout -- Aborting\n");
- abort_task = GNUNET_SCHEDULER_NO_TASK;
- do_shutdown (cls, tc);
- }
- /**
- * Callback to be called when an operation is completed
- *
- * @param cls the callback closure from functions generating an operation
- * @param op the operation that has been finished
- * @param emsg error message in case the operation has failed; will be NULL if
- * operation has executed successfully.
- */
- static void
- op_comp_cb (void *cls, struct GNUNET_TESTBED_Operation *op, const char *emsg);
- /**
- * task for delaying a connect
- *
- * @param cls NULL
- * @param tc the task context
- */
- static void
- do_delayed_connect (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
- {
- delayed_connect_task = GNUNET_SCHEDULER_NO_TASK;
- FAIL_TEST (NULL == common_operation);
- common_operation =
- GNUNET_TESTBED_overlay_connect (NULL, &op_comp_cb, NULL, peer1.peer,
- peer2.peer);
- }
- /**
- * Callback to be called when an operation is completed
- *
- * @param cls the callback closure from functions generating an operation
- * @param op the operation that has been finished
- * @param emsg error message in case the operation has failed; will be NULL if
- * operation has executed successfully.
- */
- static void
- op_comp_cb (void *cls, struct GNUNET_TESTBED_Operation *op, const char *emsg)
- {
- FAIL_TEST (common_operation == op);
- switch (result)
- {
- case PEERS_STARTED:
- FAIL_TEST (NULL == peer1.operation);
- FAIL_TEST (NULL == peer2.operation);
- FAIL_TEST (NULL != common_operation);
- break;
- case PEERS_CONNECTED:
- FAIL_TEST (NULL == peer1.operation);
- FAIL_TEST (NULL == peer2.operation);
- FAIL_TEST (NULL != common_operation);
- break;
- default:
- FAIL_TEST (0);
- }
- }
- /**
- * Signature of the event handler function called by the
- * respective event controller.
- *
- * @param cls closure
- * @param event information about the event
- */
- static void
- controller_cb (void *cls, const struct GNUNET_TESTBED_EventInformation *event)
- {
- switch (event->type)
- {
- case GNUNET_TESTBED_ET_OPERATION_FINISHED: /* Will be reached when we destroy peers */
- FAIL_TEST (PEERS_STOPPED == result);
- FAIL_TEST (NULL == event->op_cls);
- FAIL_TEST (NULL == event->details.operation_finished.emsg);
- FAIL_TEST (NULL == event->details.operation_finished.generic);
- if (event->op == peer1.operation)
- {
- GNUNET_TESTBED_operation_done (peer1.operation);
- peer1.operation = NULL;
- peer1.peer = NULL;
- }
- else if (event->op == peer2.operation)
- {
- GNUNET_TESTBED_operation_done (peer2.operation);
- peer2.operation = NULL;
- peer2.peer = NULL;
- }
- else
- FAIL_TEST (0);
- if ((NULL == peer1.peer) && (NULL == peer2.peer))
- {
- result = SUCCESS;
- GNUNET_SCHEDULER_add_now (&do_shutdown, NULL);
- }
- break;
- case GNUNET_TESTBED_ET_PEER_START:
- FAIL_TEST (INIT == result);
- FAIL_TEST (event->details.peer_start.host == host);
- if (event->details.peer_start.peer == peer1.peer)
- {
- peer1.is_running = GNUNET_YES;
- GNUNET_TESTBED_operation_done (peer1.operation);
- peer1.operation = NULL;
- }
- else if (event->details.peer_start.peer == peer2.peer)
- {
- peer2.is_running = GNUNET_YES;
- GNUNET_TESTBED_operation_done (peer2.operation);
- peer2.operation = NULL;
- }
- else
- FAIL_TEST (0);
- if ((GNUNET_YES == peer1.is_running) && (GNUNET_YES == peer2.is_running))
- {
- result = PEERS_STARTED;
- common_operation =
- GNUNET_TESTBED_overlay_connect (NULL, &op_comp_cb, NULL, peer1.peer,
- peer2.peer);
- }
- break;
- case GNUNET_TESTBED_ET_PEER_STOP:
- FAIL_TEST (PEERS_CONNECTED_2 == result);
- if (event->details.peer_stop.peer == peer1.peer)
- {
- peer1.is_running = GNUNET_NO;
- GNUNET_TESTBED_operation_done (peer1.operation);
- peer1.operation = GNUNET_TESTBED_peer_destroy (peer1.peer);
- }
- else if (event->details.peer_stop.peer == peer2.peer)
- {
- peer2.is_running = GNUNET_NO;
- GNUNET_TESTBED_operation_done (peer2.operation);
- peer2.operation = GNUNET_TESTBED_peer_destroy (peer2.peer);
- }
- else
- FAIL_TEST (0);
- if ((GNUNET_NO == peer1.is_running) && (GNUNET_NO == peer2.is_running))
- result = PEERS_STOPPED;
- break;
- case GNUNET_TESTBED_ET_CONNECT:
- switch (result)
- {
- case PEERS_STARTED:
- FAIL_TEST (NULL == peer1.operation);
- FAIL_TEST (NULL == peer2.operation);
- FAIL_TEST (NULL != common_operation);
- FAIL_TEST ((event->details.peer_connect.peer1 == peer1.peer) &&
- (event->details.peer_connect.peer2 == peer2.peer));
- GNUNET_TESTBED_operation_done (common_operation);
- common_operation = NULL;
- result = PEERS_CONNECTED;
- LOG (GNUNET_ERROR_TYPE_DEBUG, "Peers connected\n");
- delayed_connect_task =
- GNUNET_SCHEDULER_add_delayed (TIME_REL_SECS (3), &do_delayed_connect,
- NULL);
- break;
- case PEERS_CONNECTED:
- FAIL_TEST (NULL == peer1.operation);
- FAIL_TEST (NULL == peer2.operation);
- FAIL_TEST (NULL != common_operation);
- GNUNET_TESTBED_operation_done (common_operation);
- common_operation = NULL;
- result = PEERS_CONNECTED_2;
- LOG (GNUNET_ERROR_TYPE_DEBUG, "Peers connected again\n");
- peer1.operation = GNUNET_TESTBED_peer_stop (NULL, peer1.peer, NULL, NULL);
- peer2.operation = GNUNET_TESTBED_peer_stop (NULL, peer2.peer, NULL, NULL);
- break;
- default:
- FAIL_TEST (0);
- }
- break;
- default:
- FAIL_TEST (0);
- };
- }
- /**
- * Functions of this signature are called when a peer has been successfully
- * created
- *
- * @param cls the closure from GNUNET_TESTBED_peer_create()
- * @param peer the handle for the created peer; NULL on any error during
- * creation
- * @param emsg NULL if peer is not NULL; else MAY contain the error description
- */
- static void
- peer_create_cb (void *cls, struct GNUNET_TESTBED_Peer *peer, const char *emsg)
- {
- struct PeerContext *pc = cls;
- FAIL_TEST (NULL != pc->operation);
- FAIL_TEST (NULL != peer);
- FAIL_TEST (NULL == pc->peer);
- pc->peer = peer;
- GNUNET_TESTBED_operation_done (pc->operation);
- pc->operation = GNUNET_TESTBED_peer_start (NULL, pc->peer, NULL, NULL);
- }
- /**
- * Callback which will be called to after a host registration succeeded or failed
- *
- * @param cls the host which has been registered
- * @param emsg the error message; NULL if host registration is successful
- */
- static void
- registration_comp (void *cls, const char *emsg)
- {
- FAIL_TEST (cls == neighbour);
- reg_handle = NULL;
- peer1.operation =
- GNUNET_TESTBED_peer_create (controller, host, cfg, &peer_create_cb,
- &peer1);
- peer2.operation =
- GNUNET_TESTBED_peer_create (controller, host, cfg, &peer_create_cb,
- &peer2);
- FAIL_TEST (NULL != peer1.operation);
- FAIL_TEST (NULL != peer2.operation);
- }
- /**
- * Callback to signal successfull startup of the controller process
- *
- * @param cls the closure from GNUNET_TESTBED_controller_start()
- * @param cfg the configuration with which the controller has been started;
- * NULL if status is not GNUNET_OK
- * @param status GNUNET_OK if the startup is successfull; GNUNET_SYSERR if not,
- * GNUNET_TESTBED_controller_stop() shouldn't be called in this case
- */
- static void
- status_cb (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg_, int status)
- {
- uint64_t event_mask;
- if (GNUNET_OK != status)
- {
- cp = NULL;
- FAIL_TEST (0);
- }
- event_mask = 0;
- event_mask |= (1L << GNUNET_TESTBED_ET_PEER_START);
- event_mask |= (1L << GNUNET_TESTBED_ET_PEER_STOP);
- event_mask |= (1L << GNUNET_TESTBED_ET_CONNECT);
- event_mask |= (1L << GNUNET_TESTBED_ET_OPERATION_FINISHED);
- controller =
- GNUNET_TESTBED_controller_connect (host, event_mask, &controller_cb,
- NULL);
- FAIL_TEST (NULL != controller);
- neighbour = GNUNET_TESTBED_host_create ("localhost", NULL, cfg, 0);
- FAIL_TEST (NULL != neighbour);
- reg_handle =
- GNUNET_TESTBED_register_host (controller, neighbour, ®istration_comp,
- neighbour);
- FAIL_TEST (NULL != reg_handle);
- }
- /**
- * Main run function.
- *
- * @param cls NULL
- * @param args arguments passed to GNUNET_PROGRAM_run
- * @param cfgfile the path to configuration file
- * @param cfg the configuration file handle
- */
- static void
- run (void *cls, char *const *args, const char *cfgfile,
- const struct GNUNET_CONFIGURATION_Handle *config)
- {
- cfg = GNUNET_CONFIGURATION_dup (config);
- host = GNUNET_TESTBED_host_create (NULL, NULL, cfg, 0);
- FAIL_TEST (NULL != host);
- cp = GNUNET_TESTBED_controller_start ("127.0.0.1", host, status_cb,
- NULL);
- abort_task =
- GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_relative_multiply
- (GNUNET_TIME_UNIT_MINUTES, 3), &do_abort,
- NULL);
- }
- /**
- * Main function
- */
- int
- main (int argc, char **argv)
- {
- int ret;
- char *const argv2[] = { "test_testbed_api_2peers_1controller",
- "-c", "test_testbed_api.conf",
- NULL
- };
- struct GNUNET_GETOPT_CommandLineOption options[] = {
- GNUNET_GETOPT_OPTION_END
- };
- result = INIT;
- ret =
- GNUNET_PROGRAM_run ((sizeof (argv2) / sizeof (char *)) - 1, argv2,
- "test_testbed_api_2peers_1controller", "nohelp",
- options, &run, NULL);
- if ((GNUNET_OK != ret) || (SUCCESS != result))
- return 1;
- return 0;
- }
- /* end of test_testbed_api_2peers_1controller.c */
|