src/testing/gnunet-testing-run-service.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2001, 2002, 2004, 2005, 2006, 2007, 2009 Christian Grothoff (and other contributing authors)
   4
   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.
   9
  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.
  14
  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.
  19 */
  20
  21
  22 /**
  23  * @file testing/gnunet-testing-run-service.c
  24  * @brief tool to start a service for testing
  25  * @author Florian Dold
  26  *
  27  * Start a peer, running only the service specified on the command line.
  28  * Outputs the path to the temporary configuration file to stdout.
  29  *
  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.
  33  *
  34  * This executable is intended to be used by gnunet-java, in order to reliably
  35  * start and stop services for test cases.
  36  */
  37
  38 #include "platform.h"
  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"
  45
  46
  47 static struct GNUNET_DISK_FileHandle fh;
  48 static char *tmpfilename = NULL;
  49 static GNUNET_SCHEDULER_TaskIdentifier tid = GNUNET_SCHEDULER_NO_TASK;
  50 static struct GNUNET_TESTING_Peer *my_peer = NULL;
  51
  52
  53 #define LOG(kind,...)                                           \
  54   GNUNET_log_from (kind, "gnunettestingnew", __VA_ARGS__)
  55
  56
  57 /**
  58  * Cleanup called by signal handlers and when stdin is closed.
  59  * Removes the temporary file with the configuration and shuts down the scheduler.
  60  */
  61 void
  62 cleanup (void)
  63 {
  64   if (NULL != tmpfilename)
  65   {
  66     remove (tmpfilename);
  67   }
  68   GNUNET_SCHEDULER_shutdown ();
  69 }
  70
  71 /**
  72  * Called whenever we can read stdin non-blocking
  73  */
  74 void
  75 stdin_cb (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
  76 {
  77   int c;
  78
  79   if (GNUNET_SCHEDULER_REASON_SHUTDOWN & tc->reason)
  80   {
  81       return;
  82   }
  83   if (GNUNET_SCHEDULER_REASON_READ_READY & tc->reason)
  84   {
  85     c = getchar ();
  86     if (EOF == c)
  87     {
  88       tid = GNUNET_SCHEDULER_NO_TASK;
  89       cleanup ();
  90     }
  91     else
  92     {
  93       if (c == 'r')
  94       {
  95         GNUNET_TESTING_peer_stop(my_peer);
  96         GNUNET_TESTING_peer_start(my_peer);
  97         printf("restarted\n");
  98         fflush(stdout);
  99       }
 100       tid = GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL, &fh, &stdin_cb, NULL);
 101     }
 102     return;
 103   }
 104   GNUNET_break (0);
 105 }
 106
 107 /**
 108  * Main function called by the testing library.
 109  * Executed inside a running scheduler.
 110  */
 111 void
 112 testing_main (void *cls, const struct GNUNET_CONFIGURATION_Handle *cfg,
 113               const struct GNUNET_TESTING_Peer *peer)
 114 {
 115   my_peer = peer;
 116   tmpfilename = tmpnam (NULL);
 117   if (NULL == tmpfilename)
 118   {
 119     GNUNET_break (0);
 120     cleanup ();
 121     return;
 122   }
 123
 124   if (GNUNET_SYSERR ==
 125           GNUNET_CONFIGURATION_write((struct GNUNET_CONFIGURATION_Handle *) cfg, tmpfilename))
 126   {
 127     GNUNET_break (0);
 128     return;
 129   }
 130
 131   printf("%s\n", tmpfilename);
 132   fflush(stdout);
 133
 134   GNUNET_break(NULL != GNUNET_SIGNAL_handler_install(SIGTERM, &cleanup));
 135   GNUNET_break(NULL != GNUNET_SIGNAL_handler_install(SIGINT, &cleanup));
 136
 137   fh.fd = 0; /* 0=stdin */
 138   tid = GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL, &fh, &stdin_cb, NULL);
 139 }
 140
 141
 142
 143 /**
 144  * The main function.
 145  *
 146  * @param argc number of arguments from the command line
 147  * @param argv command line arguments
 148  * @return 0 ok, 1 on error
 149  */
 150 int
 151 main (int argc, char *const *argv)
 152 {
 153   static const struct GNUNET_GETOPT_CommandLineOption options[] = {
 154     GNUNET_GETOPT_OPTION_HELP("tool to start a service for testing"),
 155     GNUNET_GETOPT_OPTION_END
 156   };
 157   int arg_start;
 158   int ret;
 159
 160   arg_start = GNUNET_GETOPT_run("gnunet-testing-run-service", options, argc, argv);
 161
 162   if (arg_start == GNUNET_SYSERR)
 163   {
 164     return 1;
 165   }
 166
 167   if (arg_start != 1 || argc != 2)
 168   {
 169     fprintf (stderr, "Invalid number of arguments\n");
 170     return 1;
 171   }
 172
 173   ret =  GNUNET_TESTING_service_run_restartable ("gnunet_service_test", argv[1],
 174                                                  NULL, &testing_main, NULL);
 175
 176   printf ("bye\n");
 177
 178   return ret;
 179 }