X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=src%2Futil%2Ftest_service.c;h=5a4fe27d03055c390f794ab67881bd6c9b3bf2cb;hb=0ea8e006d5f5ef84e31e000607bd24a23f8fc1ed;hp=93cafb4d2883c2d168e369f468d8bfedea1a56c1;hpb=652e89b59ed2207c2c12172fdabcd6e659995c81;p=oweals%2Fgnunet.git diff --git a/src/util/test_service.c b/src/util/test_service.c index 93cafb4d2..5a4fe27d0 100644 --- a/src/util/test_service.c +++ b/src/util/test_service.c @@ -30,6 +30,7 @@ #include "gnunet_scheduler_lib.h" #include "gnunet_time_lib.h" + #define VERBOSE GNUNET_NO #define PORT 12435 @@ -40,16 +41,7 @@ static struct GNUNET_SCHEDULER_Handle *sched; static struct GNUNET_SERVICE_Context *sctx; -static void -end_it (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) -{ - struct GNUNET_CLIENT_Connection *client = cls; - - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Shutting down service\n"); - GNUNET_CLIENT_service_shutdown (client); - if (sctx != NULL) - GNUNET_SERVICE_stop (sctx); -} +static int ok = 1; static size_t @@ -62,23 +54,20 @@ build_msg (void *cls, size_t size, void *buf) GNUNET_assert (size >= sizeof (struct GNUNET_MessageHeader)); msg->type = htons (MY_TYPE); msg->size = htons (sizeof (struct GNUNET_MessageHeader)); - GNUNET_SCHEDULER_add_continuation (sched, - GNUNET_YES, - &end_it, - client, - GNUNET_SCHEDULER_REASON_PREREQ_DONE); + GNUNET_CLIENT_disconnect (client, GNUNET_NO); return sizeof (struct GNUNET_MessageHeader); } + static void ready (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) { const struct GNUNET_CONFIGURATION_Handle *cfg = cls; struct GNUNET_CLIENT_Connection *client; + GNUNET_assert (0 != (tc->reason & GNUNET_SCHEDULER_REASON_PREREQ_DONE)); GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Service confirmed running\n"); sched = tc->sched; - GNUNET_assert (0 != (tc->reason & GNUNET_SCHEDULER_REASON_PREREQ_DONE)); client = GNUNET_CLIENT_connect (tc->sched, "test_service", cfg); GNUNET_assert (client != NULL); GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, @@ -89,6 +78,14 @@ ready (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc) GNUNET_NO, &build_msg, client); } +static void +do_stop (void *cls, + const struct GNUNET_SCHEDULER_TaskContext *tc) +{ + GNUNET_SERVICE_stop (sctx); +} + + static void recv_cb (void *cls, struct GNUNET_SERVER_Client *client, @@ -96,6 +93,11 @@ recv_cb (void *cls, { GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Receiving client message...\n"); GNUNET_SERVER_receive_done (client, GNUNET_OK); + if (sctx != NULL) + GNUNET_SCHEDULER_add_now (sched, &do_stop, NULL); + else + GNUNET_SCHEDULER_shutdown (sched); + ok = 0; } static struct GNUNET_SERVER_MessageHandler myhandlers[] = { @@ -103,6 +105,9 @@ static struct GNUNET_SERVER_MessageHandler myhandlers[] = { {NULL, NULL, 0, 0} }; + + + static void runner (void *cls, struct GNUNET_SCHEDULER_Handle *sched, @@ -117,12 +122,6 @@ runner (void *cls, (void *) cfg); } -static void -term (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg) -{ - int *ok = cls; - *ok = 0; -} /** * Main method, starts scheduler with task1, @@ -131,7 +130,7 @@ term (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg) static int check () { - int ok = 1; + ok = 1; char *const argv[] = { "test_service", "-c", @@ -149,7 +148,8 @@ check () GNUNET_SERVICE_run (5, argv, "test_service", - &runner, &ok, &term, &ok)); + GNUNET_SERVICE_OPTION_NONE, + &runner, &ok)); GNUNET_assert (0 == ok); return ok; } @@ -193,7 +193,6 @@ runner6 (void *cls, static int check6 () { - int ok = 1; char *const argv[] = { "test_service6", "-c", @@ -211,43 +210,13 @@ check6 () GNUNET_SERVICE_run (5, argv, "test_service6", - &runner6, &ok, &term, &ok)); + GNUNET_SERVICE_OPTION_NONE, + &runner6, &ok)); GNUNET_assert (0 == ok); return ok; } -/** - * Main method, starts scheduler with task1, - * checks that "ok" is correct at the end. - */ -static int -check6d () -{ - int ok = 1; - char *const argv[] = { - "test_service6", - "-c", - "test_service_data.conf", - "-L", -#if VERBOSE - "DEBUG", -#else - "WARNING", -#endif - "-d", - NULL - }; - GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Starting V6 as daemon\n"); - GNUNET_assert (GNUNET_OK == - GNUNET_SERVICE_run (6, - argv, - "test_service6", - &runner6, &ok, &term, &ok)); - GNUNET_break (0 == ok); - return ok; -} - static void start_stop_main (void *cls, @@ -300,7 +269,7 @@ int main (int argc, char *argv[]) { int ret = 0; - struct GNUNET_NETWORK_Handle *s; + struct GNUNET_NETWORK_Handle *s = NULL; GNUNET_log_setup ("test-service", #if VERBOSE @@ -311,7 +280,11 @@ main (int argc, char *argv[]) NULL); ret += check (); ret += check (); + + // FIXME +#ifndef MINGW s = GNUNET_NETWORK_socket_create (PF_INET6, SOCK_STREAM, 0); +#endif if (NULL == s) { if ((errno == ENOBUFS) || @@ -328,7 +301,6 @@ main (int argc, char *argv[]) { GNUNET_break (GNUNET_OK == GNUNET_NETWORK_socket_close (s)); ret += check6 (); - ret += check6d (); /* with daemonization */ } ret += check_start_stop ();