2 This file is part of GNUnet.
3 (C) 2001, 2002, 2004, 2005, 2006, 2007, 2009 Christian Grothoff (and other contributing authors)
5 GNUnet is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published
7 by the Free Software Foundation; either version 3, or (at your
8 option) any later version.
10 GNUnet is distributed in the hope that it will be useful, but
11 WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 General Public License for more details.
15 You should have received a copy of the GNU General Public License
16 along with GNUnet; see the file COPYING. If not, write to the
17 Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18 Boston, MA 02111-1307, USA.
23 * @file testing/gnunet-testing-run-service.c
24 * @brief tool to start a service for testing
25 * @author Florian Dold
27 * Start a peer, running only the service specified on the command line.
28 * Outputs the path to the temporary configuration file to stdout.
30 * The peer will run until this program is killed,
31 * or stdin is closed. When reading the character 'r' from stdin, the running service is
32 * restarted with the same configuration.
34 * This executable is intended to be used by gnunet-java, in order to reliably
35 * start and stop services for test cases.
39 #include "gnunet_getopt_lib.h"
40 #include "gnunet_program_lib.h"
41 #include "gnunet_util_lib.h"
42 #include "gnunet_signal_lib.h"
43 #include "gnunet_testing_lib-new.h"
44 #include "gnunet_os_lib.h"
47 #define LOG(kind,...) \
48 GNUNET_log_from (kind, "gnunet-testing", __VA_ARGS__)
54 static struct GNUNET_DISK_FileHandle *fh;
59 static char *tmpfilename;
64 static GNUNET_SCHEDULER_TaskIdentifier tid;
69 static struct GNUNET_TESTING_Peer *my_peer;
75 * Cleanup called by signal handlers and when stdin is closed.
76 * Removes the temporary file.
79 * @param tc scheduler context
82 cleanup (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
84 if (NULL != tmpfilename)
86 if (0 != UNLINK (tmpfilename))
87 GNUNET_log_strerror_file (GNUNET_ERROR_TYPE_WARNING, "unlink", tmpfilename);
89 if (GNUNET_SCHEDULER_NO_TASK != tid)
91 GNUNET_SCHEDULER_cancel (tid);
92 tid = GNUNET_SCHEDULER_NO_TASK;
96 GNUNET_DISK_file_close (fh);
103 * Called whenever we can read stdin non-blocking
106 * @param tc scheduler context
109 stdin_cb (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
113 tid = GNUNET_SCHEDULER_NO_TASK;
114 if (0 != (GNUNET_SCHEDULER_REASON_SHUTDOWN & tc->reason))
116 GNUNET_assert (0 != (GNUNET_SCHEDULER_REASON_READ_READY & tc->reason));
122 GNUNET_SCHEDULER_shutdown ();
125 GNUNET_TESTING_peer_stop (my_peer);
126 GNUNET_TESTING_peer_start (my_peer);
127 printf ("restarted\n");
132 /* ignore whitespace */
135 fprintf (stderr, _("Unknown command, use 'q' to quit or 'r' to restart peer\n"));
138 tid = GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL, &fh, &stdin_cb, NULL);
143 * Main function called by the testing library.
144 * Executed inside a running scheduler.
147 * @param cfg configuration of the peer that was started
148 * @param peer handle to the peer
151 testing_main (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg,
152 const struct GNUNET_TESTING_Peer *peer)
154 my_peer = (struct GNUNET_TESTING_Peer *) peer;
155 if (NULL == (tmpfilename = GNUNET_DISK_mktemp ("gnunet-testing")))
162 GNUNET_CONFIGURATION_write ((struct GNUNET_CONFIGURATION_Handle *) cfg, tmpfilename))
167 printf("%s\n", tmpfilename);
169 GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREFER_REL, &cleanup);
170 fh = GNUNET_DISK_get_handle_from_native (stdin);
171 tid = GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL, &fh, &stdin_cb, NULL);
178 * @param argc number of arguments from the command line
179 * @param argv command line arguments
180 * @return 0 ok, 1 on error
183 main (int argc, char *const *argv)
185 static char *cfg_name;
186 static char *srv_name;
187 static const struct GNUNET_GETOPT_CommandLineOption options[] = {
188 {'c', "config", "FILENAME",
189 gettext_noop ("name of the configuration file to use"), 1,
190 &GNUNET_GETOPT_set_string, &cfg_name},
191 {'s', "service", "SERVICE",
192 gettext_noop ("name of the service to run"), 1,
193 &GNUNET_GETOPT_set_string, &srv_name},
194 GNUNET_GETOPT_OPTION_HELP ("tool to start a service for testing"),
195 GNUNET_GETOPT_OPTION_END
200 GNUNET_GETOPT_run("gnunet-testing-run-service", options, argc, argv))
202 ret = GNUNET_TESTING_service_run_restartable ("gnunet_service_test", srv_name,
203 cfg_name, &testing_main, NULL);