/*
This file is part of GNUnet.
- (C) 2009 Christian Grothoff (and other contributing authors)
+ (C) 2009, 2010, 2011 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 2, or (at your
+ 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
* @file arm/gnunet-service-arm.c
* @brief the automated restart manager service
* @author Christian Grothoff
- *
- * TODO:
- * - multiple start-stop requests with RC>1 can result
- * in UP/DOWN signals based on "pending" that are inaccurate...
- * => have list of clients waiting for a resolution instead of
- * giving instant (but incorrect) replies
- * - need to test auto-restart code on configuration changes;
- * - should refine restart code to check if *relevant* parts of the
- * configuration were changed (anything in the section for the service)
- * - should have a way to specify dependencies between services and
- * manage restarts of groups of services
*/
#include "platform.h"
-#include "gnunet_client_lib.h"
-#include "gnunet_getopt_lib.h"
-#include "gnunet_os_lib.h"
+#include "gnunet_util_lib.h"
+#include "gnunet_arm_service.h"
#include "gnunet_protocols.h"
-#include "gnunet_service_lib.h"
-#include "gnunet_signal_lib.h"
#include "arm.h"
-
-/**
- * Check for configuration file changes every 5s.
- */
-#define MAINT_FREQUENCY GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 5)
-
/**
* Threshold after which exponential backoff shouldn't increase (in ms); 30m
*/
-#define EXPONENTIAL_BACKOFF_THRESHOLD (1000 * 60 * 30)
+#define EXPONENTIAL_BACKOFF_THRESHOLD GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_MINUTES, 30)
/**
*/
struct ServiceList;
+
/**
- * Function to call if waitpid informs us that
- * a process has died.
- *
- * @param cls closure
- * @param pos entry in the service list of the process that died
+ * Record with information about a listen socket we have open.
*/
-typedef void (*CleanCallback) (void *cls, struct ServiceList * pos);
+struct ServiceListeningInfo
+{
+ /**
+ * This is a linked list.
+ */
+ struct ServiceListeningInfo *next;
+
+ /**
+ * This is a linked list.
+ */
+ struct ServiceListeningInfo *prev;
+
+ /**
+ * Address this socket is listening on.
+ */
+ struct sockaddr *service_addr;
+
+ /**
+ * Service this listen socket is for.
+ */
+ struct ServiceList *sl;
+
+ /**
+ * Number of bytes in 'service_addr'
+ */
+ socklen_t service_addr_len;
+
+ /**
+ * Our listening socket.
+ */
+ struct GNUNET_NETWORK_Handle *listen_socket;
+
+ /**
+ * Task doing the accepting.
+ */
+ GNUNET_SCHEDULER_TaskIdentifier accept_task;
+
+};
+
/**
* List of our services.
struct ServiceList
{
/**
- * This is a linked list.
+ * This is a doubly-linked list.
*/
struct ServiceList *next;
+ /**
+ * This is a doubly-linked list.
+ */
+ struct ServiceList *prev;
+
+ /**
+ * Linked list of listen sockets associated with this service.
+ */
+ struct ServiceListeningInfo *listen_head;
+
+ /**
+ * Linked list of listen sockets associated with this service.
+ */
+ struct ServiceListeningInfo *listen_tail;
+
/**
* Name of the service.
*/
char *config;
/**
- * Function to call upon kill completion (waitpid), NULL
+ * Client to notify upon kill completion (waitpid), NULL
* if we should simply restart the process.
*/
- CleanCallback kill_continuation;
+ struct GNUNET_SERVER_Client *killing_client;
/**
- * Closure for kill_continuation.
+ * Process structure pointer of the child.
*/
- void *kill_continuation_cls;
+ struct GNUNET_OS_Process *proc;
/**
- * Process ID of the child.
+ * Process exponential backoff time
*/
- pid_t pid;
+ struct GNUNET_TIME_Relative backoff;
/**
- * Last time the config of this service was
- * modified.
+ * Absolute time at which the process is scheduled to restart in case of death
*/
- time_t mtime;
+ struct GNUNET_TIME_Absolute restart_at;
- /* Process exponential backoff time */
- struct GNUNET_TIME_Relative backoff;
-
- /* Absolute time at which the process is scheduled to restart in case of death */
- struct GNUNET_TIME_Absolute restartAt;
+ /**
+ * Time we asked the service to shut down (used to calculate time it took
+ * the service to terminate).
+ */
+ struct GNUNET_TIME_Absolute killed_at;
/**
- * Reference counter (counts how many times we've been
- * asked to start the service). We only actually stop
- * it once rc hits zero.
+ * Is this service to be started by default (or did a client tell us explicitly
+ * to start it)? GNUNET_NO if the service is started only upon 'accept' on a
+ * listen socket or possibly explicitly by a client changing the value.
*/
- unsigned int rc;
+ int is_default;
+ /**
+ * Should we use pipes to signal this process? (YES for Java binaries and if we
+ * are on Windoze).
+ */
+ int pipe_control;
};
/**
* List of running services.
*/
-static struct ServiceList *running;
+static struct ServiceList *running_head;
/**
- * Our configuration
+ * List of running services.
*/
-static const struct GNUNET_CONFIGURATION_Handle *cfg;
+static struct ServiceList *running_tail;
/**
- * Our scheduler.
+ * Our configuration
*/
-static struct GNUNET_SCHEDULER_Handle *sched;
+static const struct GNUNET_CONFIGURATION_Handle *cfg;
/**
* Command to prepend to each actual command.
*/
static char *prefix_command;
+/**
+ * Option to append to each actual command.
+ */
+static char *final_option;
+
/**
* ID of task called whenever we get a SIGCHILD.
*/
*/
static GNUNET_SCHEDULER_TaskIdentifier child_restart_task;
-/**
- * Context for our SIGCHILD handler.
- */
-static struct GNUNET_SIGNAL_Context *shc_chld;
-
/**
* Pipe used to communicate shutdown via signal.
*/
static struct GNUNET_DISK_PipeHandle *sigpipe;
-/**
- * Reading end of the signal pipe.
- */
-static const struct GNUNET_DISK_FileHandle *pr;
-
/**
* Are we in shutdown mode?
*/
static int in_shutdown;
-
/**
* Handle to our server instance. Our server is a bit special in that
* its service is not immediately stopped once we get a shutdown
static struct GNUNET_SERVER_Handle *server;
+#include "do_start_process.c"
+
+
/**
- * If the configuration file changes, restart tasks that depended on that
- * option.
+ * Actually start the process for the given service.
*
- * @param cls closure, NULL if we need to self-restart
- * @param tc context
+ * @param sl identifies service to start
*/
-static void
-config_change_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+static void
+start_process (struct ServiceList *sl)
{
- struct ServiceList *pos;
- struct stat sbuf;
+ char *loprefix;
+ char *options;
+ char *optpos;
+ char *optend;
+ const char *next;
+ int use_debug;
+ char b;
+ char *val;
+ struct ServiceListeningInfo *sli;
+ SOCKTYPE *lsocks;
+ unsigned int ls;
+
+ /* calculate listen socket list */
+ lsocks = NULL;
+ ls = 0;
+ for (sli = sl->listen_head; NULL != sli; sli = sli->next)
+ {
+ GNUNET_array_append (lsocks, ls,
+ GNUNET_NETWORK_get_fd (sli->listen_socket));
+ if (sli->accept_task != GNUNET_SCHEDULER_NO_TASK)
+ {
+ GNUNET_SCHEDULER_cancel (sli->accept_task);
+ sli->accept_task = GNUNET_SCHEDULER_NO_TASK;
+ }
+ }
+#if WINDOWS
+ GNUNET_array_append (lsocks, ls, INVALID_SOCKET);
+#else
+ GNUNET_array_append (lsocks, ls, -1);
+#endif
- pos = running;
- while (pos != NULL)
+ /* obtain configuration */
+ if (GNUNET_OK !=
+ GNUNET_CONFIGURATION_get_value_string (cfg, sl->name, "PREFIX",
+ &loprefix))
+ loprefix = GNUNET_strdup (prefix_command);
+ if (GNUNET_OK !=
+ GNUNET_CONFIGURATION_get_value_string (cfg, sl->name, "OPTIONS",
+ &options))
{
- /* FIXME: this test for config change is a bit too coarse grained */
- if ( (0 == STAT (pos->config, &sbuf)) &&
- (pos->mtime < sbuf.st_mtime) &&
- (pos->pid != 0) )
+ options = GNUNET_strdup (final_option);
+ if (NULL == strstr (options, "%"))
{
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- _("Restarting service `%s' due to configuration file change.\n"));
- if (0 != PLIBC_KILL (pos->pid, SIGTERM))
- GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "kill");
- else
- pos->backoff = GNUNET_TIME_UNIT_MILLISECONDS;
+ /* replace '{}' with service name */
+ while (NULL != (optpos = strstr (options, "{}")))
+ {
+ optpos[0] = '%';
+ optpos[1] = 's';
+ GNUNET_asprintf (&optpos, options, sl->name);
+ GNUNET_free (options);
+ options = optpos;
+ }
+ /* replace '$PATH' with value associated with "PATH" */
+ while (NULL != (optpos = strstr (options, "$")))
+ {
+ optend = optpos + 1;
+ while (isupper ((unsigned char) *optend))
+ optend++;
+ b = *optend;
+ if ('\0' == b)
+ next = "";
+ else
+ next = optend + 1;
+ *optend = '\0';
+ if (GNUNET_OK !=
+ GNUNET_CONFIGURATION_get_value_string (cfg, "PATHS",
+ optpos + 1, &val))
+ val = GNUNET_strdup ("");
+ *optpos = '\0';
+ GNUNET_asprintf (&optpos, "%s%s%c%s", options, val, b, next);
+ GNUNET_free (options);
+ GNUNET_free (val);
+ options = optpos;
+ }
}
- pos = pos->next;
}
-}
+ use_debug = GNUNET_CONFIGURATION_get_value_yesno (cfg, sl->name, "DEBUG");
+ /* actually start process */
+#if DEBUG_ARM
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "Starting service `%s' using binary `%s' and configuration `%s'\n",
+ sl->name, sl->binary, sl->config);
+#endif
+ GNUNET_assert (NULL == sl->proc);
+ if (GNUNET_YES == use_debug)
+ sl->proc =
+ do_start_process (sl->pipe_control,
+ lsocks, loprefix, sl->binary, "-c", sl->config, "-L",
+ "DEBUG", options, NULL);
+ else
+ sl->proc =
+ do_start_process (sl->pipe_control,
+ lsocks, loprefix, sl->binary, "-c", sl->config,
+ options, NULL);
+ if (sl->proc == NULL)
+ GNUNET_log (GNUNET_ERROR_TYPE_ERROR, _("Failed to start service `%s'\n"),
+ sl->name);
+ else
+ GNUNET_log (GNUNET_ERROR_TYPE_INFO, _("Starting service `%s'\n"),
+ sl->name);
+ /* clean up */
+ GNUNET_free (loprefix);
+ GNUNET_free (options);
+ GNUNET_array_grow (lsocks, ls, 0);
+}
/**
static size_t
write_result (void *cls, size_t size, void *buf)
{
- uint16_t *res = cls;
- struct GNUNET_MessageHeader *msg;
+ enum GNUNET_ARM_ProcessStatus *res = cls;
+ struct GNUNET_ARM_ResultMessage *msg;
if (buf == NULL)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Could not send status result to client\n"));
- return 0; /* error, not much we can do */
- }
- GNUNET_assert (size >= sizeof (struct GNUNET_MessageHeader));
+ {
+ GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
+ _("Could not send status result to client\n"));
+ return 0; /* error, not much we can do */
+ }
+#if DEBUG_ARM
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "Sending status response %u to client\n", (unsigned int) *res);
+#endif
+ GNUNET_assert (size >= sizeof (struct GNUNET_ARM_ResultMessage));
msg = buf;
- msg->size = htons (sizeof (struct GNUNET_MessageHeader));
- msg->type = htons (*res);
+ msg->header.size = htons (sizeof (struct GNUNET_ARM_ResultMessage));
+ msg->header.type = htons (GNUNET_MESSAGE_TYPE_ARM_RESULT);
+ msg->status = htonl ((uint32_t) (*res));
GNUNET_free (res);
- return sizeof (struct GNUNET_MessageHeader);
+ return sizeof (struct GNUNET_ARM_ResultMessage);
}
+/**
+ * Transmit the list of running services.
+ *
+ * @param cls pointer to struct GNUNET_ARM_ListResultMessage with the message
+ * @param size number of bytes available in buf
+ * @param buf where to copy the message, NULL on error
+ * @return number of bytes copied to buf
+ */
+static size_t
+write_list_result (void *cls, size_t size, void *buf)
+{
+ struct GNUNET_ARM_ListResultMessage *msg = cls;
+ struct GNUNET_ARM_ListResultMessage *rslt;
+ size_t rslt_size;
+
+ if (buf == NULL)
+ {
+ GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
+ _("Could not send list result to client\n"));
+ return 0; /* error, not much we can do */
+ }
+
+ GNUNET_assert (size >= msg->header.size);
+ rslt = buf;
+ rslt->header.size = htons (msg->header.size);
+ rslt->header.type = htons (msg->header.type);
+ rslt->count = htons (msg->count);
+
+ size_t list_size = msg->header.size
+ - sizeof (struct GNUNET_ARM_ListResultMessage);
+ memcpy (&rslt[1], &msg[1], list_size);
+
+ rslt_size = msg->header.size;
+ GNUNET_free (msg);
+ return rslt_size;
+}
/**
* @return NULL if it was not found
*/
static void
-signal_result (struct GNUNET_SERVER_Client *client,
- const char *name, uint16_t result)
+signal_result (struct GNUNET_SERVER_Client *client, const char *name,
+ enum GNUNET_ARM_ProcessStatus result)
{
- uint16_t *res;
+ enum GNUNET_ARM_ProcessStatus *res;
if (NULL == client)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- _
- ("Not sending status result to client: no client known\n"));
- return;
- }
-#if DEBUG_ARM
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Telling client that service `%s' is now %s\n",
- name,
- result == GNUNET_MESSAGE_TYPE_ARM_IS_DOWN ? "down" : "up");
-#endif
- res = GNUNET_malloc (sizeof (uint16_t));
+ return;
+ /* FIXME: this is not super-clean yet... */
+ res = GNUNET_malloc (sizeof (enum GNUNET_ARM_ProcessStatus));
*res = result;
GNUNET_SERVER_notify_transmit_ready (client,
- sizeof (struct GNUNET_MessageHeader),
+ sizeof (struct
+ GNUNET_ARM_ResultMessage),
GNUNET_TIME_UNIT_FOREVER_REL,
&write_result, res);
+ GNUNET_SERVER_receive_done (client, GNUNET_OK);
}
/**
* Find the process with the given service
- * name in the given list, remove it and return it.
+ * name in the given list and return it.
*
* @param name which service entry to look up
* @return NULL if it was not found
*/
static struct ServiceList *
-find_name (const char *name)
+find_service (const char *name)
{
- struct ServiceList *pos;
- struct ServiceList *prev;
+ struct ServiceList *sl;
- pos = running;
- prev = NULL;
- while (pos != NULL)
+ sl = running_head;
+ while (sl != NULL)
{
- if (0 == strcmp (pos->name, name))
- {
- if (prev == NULL)
- running = pos->next;
- else
- prev->next = pos->next;
- pos->next = NULL;
- return pos;
- }
- prev = pos;
- pos = pos->next;
+ if (0 == strcasecmp (sl->name, name))
+ return sl;
+ sl = sl->next;
}
return NULL;
}
/**
- * Free an entry in the service list.
+ * First connection has come to the listening socket associated with the service,
+ * create the service in order to relay the incoming connection to it
*
- * @param pos entry to free
+ * @param cls callback data, struct ServiceListeningInfo describing a listen socket
+ * @param tc context
*/
static void
-free_entry (struct ServiceList *pos)
+accept_connection (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
{
- GNUNET_free_non_null (pos->config);
- GNUNET_free_non_null (pos->binary);
- GNUNET_free (pos->name);
- GNUNET_free (pos);
+ struct ServiceListeningInfo *sli = cls;
+ struct ServiceList *sl = sli->sl;
+
+ sli->accept_task = GNUNET_SCHEDULER_NO_TASK;
+ if (0 != (GNUNET_SCHEDULER_REASON_SHUTDOWN & tc->reason))
+ return;
+ start_process (sl);
}
/**
- * Actually start the process for the given service.
+ * Creating a listening socket for each of the service's addresses and
+ * wait for the first incoming connection to it
*
- * @param sl identifies service to start
+ * @param sa address associated with the service
+ * @param addr_len length of sa
+ * @param sl service entry for the service in question
*/
static void
-start_process (struct ServiceList *sl)
+create_listen_socket (struct sockaddr *sa, socklen_t addr_len,
+ struct ServiceList *sl)
{
- char *loprefix;
- char *options;
- char **argv;
- unsigned int argv_size;
- char *lopos;
- char *optpos;
- const char *firstarg;
- int use_debug;
+ const static int on = 1;
+ struct GNUNET_NETWORK_Handle *sock;
+ struct ServiceListeningInfo *sli;
- /* start service */
- if (GNUNET_OK !=
- GNUNET_CONFIGURATION_get_value_string (cfg,
- sl->name, "PREFIX", &loprefix))
- loprefix = GNUNET_strdup (prefix_command);
- if (GNUNET_OK !=
- GNUNET_CONFIGURATION_get_value_string (cfg,
- sl->name, "OPTIONS", &options))
- options = GNUNET_strdup ("");
- use_debug = GNUNET_CONFIGURATION_get_value_yesno (cfg, sl->name, "DEBUG");
-
- GNUNET_log (GNUNET_ERROR_TYPE_INFO, _("Starting service `%s'\n"), sl->name);
-#if DEBUG_ARM
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Starting service `%s' using binary `%s' and configuration `%s'\n",
- sl->name, sl->binary, sl->config);
-#endif
- argv_size = 6;
- if (use_debug)
- argv_size += 2;
- lopos = loprefix;
- while ('\0' != *lopos)
+ switch (sa->sa_family)
{
- if (*lopos == ' ')
- argv_size++;
- lopos++;
+ case AF_INET:
+ sock = GNUNET_NETWORK_socket_create (PF_INET, SOCK_STREAM, 0);
+ break;
+ case AF_INET6:
+ sock = GNUNET_NETWORK_socket_create (PF_INET6, SOCK_STREAM, 0);
+ break;
+ case AF_UNIX:
+ if (strcmp (GNUNET_a2s (sa, addr_len), "@") == 0) /* Do not bind to blank UNIX path! */
+ return;
+ sock = GNUNET_NETWORK_socket_create (PF_UNIX, SOCK_STREAM, 0);
+ break;
+ default:
+ GNUNET_break (0);
+ sock = NULL;
+ errno = EAFNOSUPPORT;
+ break;
}
- optpos = options;
- while ('\0' != *optpos)
+ if (NULL == sock)
{
- if (*optpos == ' ')
- argv_size++;
- optpos++;
+ GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+ _("Unable to create socket for service `%s': %s\n"),
+ sl->name, STRERROR (errno));
+ GNUNET_free (sa);
+ return;
}
- firstarg = NULL;
- argv = GNUNET_malloc (argv_size * sizeof (char *));
- argv_size = 0;
- lopos = loprefix;
+ if (GNUNET_NETWORK_socket_setsockopt
+ (sock, SOL_SOCKET, SO_REUSEADDR, &on, sizeof (on)) != GNUNET_OK)
+ GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR | GNUNET_ERROR_TYPE_BULK,
+ "setsockopt");
+#ifdef IPV6_V6ONLY
+ if ((sa->sa_family == AF_INET6) &&
+ (GNUNET_NETWORK_socket_setsockopt
+ (sock, IPPROTO_IPV6, IPV6_V6ONLY, &on, sizeof (on)) != GNUNET_OK))
+ GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR | GNUNET_ERROR_TYPE_BULK,
+ "setsockopt");
+#endif
- while ('\0' != *lopos)
+ if (GNUNET_NETWORK_socket_bind
+ (sock, (const struct sockaddr *) sa, addr_len) != GNUNET_OK)
{
- while (*lopos == ' ')
- lopos++;
- if (*lopos == '\0')
- continue;
- if (argv_size == 0)
- firstarg = lopos;
- argv[argv_size++] = lopos;
- while (('\0' != *lopos) && (' ' != *lopos))
- lopos++;
- if ('\0' == *lopos)
- continue;
- *lopos = '\0';
- lopos++;
- }
- if (argv_size == 0)
- firstarg = sl->binary;
- argv[argv_size++] = sl->binary;
- argv[argv_size++] = "-c";
- argv[argv_size++] = sl->config;
- if (GNUNET_YES == use_debug)
- {
- argv[argv_size++] = "-L";
- argv[argv_size++] = "DEBUG";
+ GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
+ _
+ ("Unable to bind listening socket for service `%s' to address `%s': %s\n"),
+ sl->name, GNUNET_a2s (sa, addr_len), STRERROR (errno));
+ GNUNET_break (GNUNET_OK == GNUNET_NETWORK_socket_close (sock));
+ GNUNET_free (sa);
+ return;
}
- optpos = options;
- while ('\0' != *optpos)
+ if (GNUNET_NETWORK_socket_listen (sock, 5) != GNUNET_OK)
{
- while (*optpos == ' ')
- optpos++;
- if (*optpos == '\0')
- continue;
- argv[argv_size++] = optpos;
- while (('\0' != *optpos) && (' ' != *optpos))
- optpos++;
- if ('\0' == *optpos)
- continue;
- *optpos = '\0';
- optpos++;
+ GNUNET_log_strerror (GNUNET_ERROR_TYPE_ERROR, "listen");
+ GNUNET_break (GNUNET_OK == GNUNET_NETWORK_socket_close (sock));
+ GNUNET_free (sa);
+ return;
}
- argv[argv_size++] = NULL;
- sl->pid = GNUNET_OS_start_process_v (firstarg, argv);
- /* FIXME: should check sl->pid */
- GNUNET_free (argv);
- GNUNET_free (loprefix);
- GNUNET_free (options);
+ GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+ _("ARM now monitors connections to service `%s' at `%s'\n"),
+ sl->name, GNUNET_a2s (sa, addr_len));
+ sli = GNUNET_malloc (sizeof (struct ServiceListeningInfo));
+ sli->service_addr = sa;
+ sli->service_addr_len = addr_len;
+ sli->listen_socket = sock;
+ sli->sl = sl;
+ sli->accept_task =
+ GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL, sock,
+ &accept_connection, sli);
+ GNUNET_CONTAINER_DLL_insert (sl->listen_head, sl->listen_tail, sli);
}
/**
- * Start the specified service.
+ * Remove and free an entry in the service list. Listen sockets
+ * must have already been cleaned up. Only to be called during shutdown.
*
- * @param client who is asking for this
- * @param servicename name of the service to start
+ * @param sl entry to free
*/
static void
-start_service (struct GNUNET_SERVER_Client *client, const char *servicename)
+free_service (struct ServiceList *sl)
{
- struct ServiceList *sl;
- char *binary;
- char *config;
- struct stat sbuf;
+ GNUNET_assert (GNUNET_YES == in_shutdown);
+ GNUNET_CONTAINER_DLL_remove (running_head, running_tail, sl);
+ GNUNET_assert (NULL == sl->listen_head);
+ GNUNET_free_non_null (sl->config);
+ GNUNET_free_non_null (sl->binary);
+ GNUNET_free (sl->name);
+ GNUNET_free (sl);
+}
- if (GNUNET_YES == in_shutdown)
+
+/**
+ * Handle START-message.
+ *
+ * @param cls closure (always NULL)
+ * @param client identification of the client
+ * @param message the actual message
+ * @return GNUNET_OK to keep the connection open,
+ * GNUNET_SYSERR to close it (signal serious error)
+ */
+static void
+handle_start (void *cls, struct GNUNET_SERVER_Client *client,
+ const struct GNUNET_MessageHeader *message)
+{
+ const char *servicename;
+ struct ServiceList *sl;
+ uint16_t size;
+
+ size = ntohs (message->size);
+ size -= sizeof (struct GNUNET_MessageHeader);
+ servicename = (const char *) &message[1];
+ if ((size == 0) || (servicename[size - 1] != '\0'))
{
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- _("ARM is shutting down, service `%s' not started.\n"),
- servicename);
- signal_result (client, servicename, GNUNET_MESSAGE_TYPE_ARM_IS_DOWN);
+ GNUNET_break (0);
+ GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
return;
}
- sl = find_name (servicename);
- if (sl != NULL)
+ if (GNUNET_YES == in_shutdown)
{
- /* already running, just increment RC */
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- _("Service `%s' already running.\n"), servicename);
- sl->rc++;
- sl->next = running;
- running = sl;
- signal_result (client, servicename, GNUNET_MESSAGE_TYPE_ARM_IS_UP);
+ signal_result (client, servicename, GNUNET_ARM_PROCESS_SHUTDOWN);
return;
}
- if (GNUNET_OK !=
- GNUNET_CONFIGURATION_get_value_string (cfg,
- servicename, "BINARY", &binary))
+ sl = find_service (servicename);
+ if (NULL == sl)
{
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Binary implementing service `%s' not known!\n"),
- servicename);
- signal_result (client, servicename, GNUNET_MESSAGE_TYPE_ARM_IS_DOWN);
+ signal_result (client, servicename, GNUNET_ARM_PROCESS_UNKNOWN);
return;
}
- if ((GNUNET_OK !=
- GNUNET_CONFIGURATION_get_value_filename (cfg,
- servicename,
- "CONFIG",
- &config)) ||
- (0 != STAT (config, &sbuf)))
+ sl->is_default = GNUNET_YES;
+ if (sl->proc != NULL)
{
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Configuration file `%s' for service `%s' not known!\n"),
- config, servicename);
- signal_result (client, servicename, GNUNET_MESSAGE_TYPE_ARM_IS_DOWN);
- GNUNET_free (binary);
- GNUNET_free_non_null (config);
+ signal_result (client, servicename, GNUNET_ARM_PROCESS_ALREADY_RUNNING);
return;
}
- sl = GNUNET_malloc (sizeof (struct ServiceList));
- sl->name = GNUNET_strdup (servicename);
- sl->next = running;
- sl->rc = 1;
- sl->binary = binary;
- sl->config = config;
- sl->mtime = sbuf.st_mtime;
- sl->backoff = GNUNET_TIME_UNIT_MILLISECONDS;
- sl->restartAt = GNUNET_TIME_UNIT_FOREVER_ABS;
-
- running = sl;
start_process (sl);
- if (NULL != client)
- signal_result (client, servicename, GNUNET_MESSAGE_TYPE_ARM_IS_UP);
+ signal_result (client, servicename, GNUNET_ARM_PROCESS_STARTING);
}
/**
- * Free the given entry in the service list and signal
- * the given client that the service is now down.
- *
- * @param cls pointer to the client ("struct GNUNET_SERVER_Client*")
- * @param pos entry for the service
- */
-static void
-free_and_signal (void *cls, struct ServiceList *pos)
-{
- struct GNUNET_SERVER_Client *client = cls;
- /* find_name will remove "pos" from the list! */
- GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Service `%s' stopped\n", pos->name);
- signal_result (client, pos->name, GNUNET_MESSAGE_TYPE_ARM_IS_DOWN);
- GNUNET_SERVER_receive_done (client, GNUNET_OK);
- GNUNET_SERVER_client_drop (client);
- free_entry (pos);
-}
-
-
-/**
- * Stop the specified service.
+ * Handle STOP-message.
*
- * @param client who is asking for this
- * @param servicename name of the service to stop
+ * @param cls closure (always NULL)
+ * @param client identification of the client
+ * @param message the actual message
+ * @return GNUNET_OK to keep the connection open,
+ * GNUNET_SYSERR to close it (signal serious error)
*/
static void
-stop_service (struct GNUNET_SERVER_Client *client, const char *servicename)
+handle_stop (void *cls, struct GNUNET_SERVER_Client *client,
+ const struct GNUNET_MessageHeader *message)
{
- struct ServiceList *pos;
+ struct ServiceList *sl;
+ const char *servicename;
+ uint16_t size;
+ size = ntohs (message->size);
+ size -= sizeof (struct GNUNET_MessageHeader);
+ servicename = (const char *) &message[1];
+ if ((size == 0) || (servicename[size - 1] != '\0'))
+ {
+ GNUNET_break (0);
+ GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
+ return;
+ }
GNUNET_log (GNUNET_ERROR_TYPE_INFO,
_("Preparing to stop `%s'\n"), servicename);
- pos = find_name (servicename);
- if (pos == NULL)
+ sl = find_service (servicename);
+ if (sl == NULL)
{
- signal_result (client, servicename, GNUNET_MESSAGE_TYPE_ARM_IS_UNKNOWN);
- GNUNET_SERVER_receive_done (client, GNUNET_OK);
+ signal_result (client, servicename, GNUNET_ARM_PROCESS_UNKNOWN);
return;
}
- if (pos->rc > 1)
+ sl->is_default = GNUNET_NO;
+ if (GNUNET_YES == in_shutdown)
{
- /* RC>1, just decrement RC */
- pos->rc--;
- pos->next = running;
- running = pos;
-#if DEBUG_ARM
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Service `%s' still used by %u clients, will keep it running!\n",
- servicename, pos->rc);
-#endif
- signal_result (client, servicename, GNUNET_MESSAGE_TYPE_ARM_IS_UP);
- GNUNET_SERVER_receive_done (client, GNUNET_OK);
+ /* shutdown in progress */
+ signal_result (client, servicename, GNUNET_ARM_PROCESS_SHUTDOWN);
return;
}
- if (pos->rc == 1)
- pos->rc--; /* decrement RC to zero */
- if (pos->kill_continuation != NULL)
+ if (sl->killing_client != NULL)
{
/* killing already in progress */
-#if DEBUG_ARM
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Service `%s' is already down\n", servicename);
-#endif
- signal_result (client, servicename, GNUNET_MESSAGE_TYPE_ARM_IS_DOWN);
- GNUNET_SERVER_receive_done (client, GNUNET_OK);
+ signal_result (client, servicename,
+ GNUNET_ARM_PROCESS_ALREADY_STOPPING);
return;
}
-
- if (GNUNET_YES == in_shutdown)
+ if (sl->proc == NULL)
{
-#if DEBUG_ARM
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Termination request already sent to `%s' (since ARM is in shutdown).\n",
- servicename);
-#endif
- signal_result (client, servicename, GNUNET_MESSAGE_TYPE_ARM_IS_DOWN);
- GNUNET_SERVER_receive_done (client, GNUNET_OK);
+ /* process is down */
+ signal_result (client, servicename, GNUNET_ARM_PROCESS_ALREADY_DOWN);
return;
}
-
-
#if DEBUG_ARM
GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
"Sending kill signal to service `%s', waiting for process to die.\n",
servicename);
#endif
- if (0 != PLIBC_KILL (pos->pid, SIGTERM))
+ sl->killed_at = GNUNET_TIME_absolute_get ();
+ if (0 != GNUNET_OS_process_kill (sl->proc, SIGTERM))
GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "kill");
- pos->next = running;
- running = pos;
- pos->kill_continuation = &free_and_signal;
- pos->kill_continuation_cls = client;
+ sl->killing_client = client;
GNUNET_SERVER_client_keep (client);
}
-
/**
- * Handle START-message.
+ * Handle LIST-message.
*
* @param cls closure (always NULL)
* @param client identification of the client
* @param message the actual message
- * @return GNUNET_OK to keep the connection open,
- * GNUNET_SYSERR to close it (signal serious error)
*/
static void
-handle_start (void *cls,
- struct GNUNET_SERVER_Client *client,
- const struct GNUNET_MessageHeader *message)
+handle_list (void *cls, struct GNUNET_SERVER_Client *client,
+ const struct GNUNET_MessageHeader *message)
{
- const char *servicename;
- uint16_t size;
-
- size = ntohs (message->size);
- size -= sizeof (struct GNUNET_MessageHeader);
- servicename = (const char *) &message[1];
- if ((size == 0) || (servicename[size - 1] != '\0'))
+ struct GNUNET_ARM_ListResultMessage *msg;
+ size_t string_list_size;
+ size_t total_size;
+ struct ServiceList *sl;
+ uint16_t count;
+
+ if (NULL == client)
+ return;
+
+ count = 0;
+ string_list_size = 0;
+ /* first count the running processes get their name's size */
+ for (sl = running_head; sl != NULL; sl = sl->next)
+ {
+ if (sl->proc != NULL)
{
- GNUNET_break (0);
- GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
- return;
+ string_list_size += strlen (sl->name);
+ string_list_size += strlen (sl->binary);
+ string_list_size += 4;
+ count++;
}
- start_service (client, servicename);
+ }
+ total_size = sizeof (struct GNUNET_ARM_ListResultMessage)
+ + string_list_size;
+ msg = GNUNET_malloc (total_size);
+ msg->header.size = total_size;
+ msg->header.type = GNUNET_MESSAGE_TYPE_ARM_LIST_RESULT;
+ msg->count = count;
+
+ char *pos = (char *)&msg[1];
+ for (sl = running_head; sl != NULL; sl = sl->next)
+ {
+ if (sl->proc != NULL)
+ {
+ size_t s = strlen (sl->name) + strlen (sl->binary) + 4;
+ GNUNET_snprintf(pos, s, "%s (%s)", sl->name, sl->binary);
+ pos += s;
+ }
+ }
+
+ GNUNET_SERVER_notify_transmit_ready (client,
+ msg->header.size,
+ GNUNET_TIME_UNIT_FOREVER_REL,
+ &write_list_result, msg);
GNUNET_SERVER_receive_done (client, GNUNET_OK);
}
-
/**
- * Handle STOP-message.
- *
- * @param cls closure (always NULL)
- * @param client identification of the client
- * @param message the actual message
- * @return GNUNET_OK to keep the connection open,
- * GNUNET_SYSERR to close it (signal serious error)
+ * We are done with everything. Stop remaining
+ * tasks, signal handler and the server.
*/
static void
-handle_stop (void *cls,
- struct GNUNET_SERVER_Client *client,
- const struct GNUNET_MessageHeader *message)
+do_shutdown ()
{
- const char *servicename;
- uint16_t size;
-
- size = ntohs (message->size);
- size -= sizeof (struct GNUNET_MessageHeader);
- servicename = (const char *) &message[1];
- if ((size == 0) || (servicename[size - 1] != '\0'))
+ if (NULL != server)
{
- GNUNET_break (0);
- GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
- return;
+ GNUNET_SERVER_destroy (server);
+ server = NULL;
+ }
+ if (GNUNET_SCHEDULER_NO_TASK != child_death_task)
+ {
+ GNUNET_SCHEDULER_cancel (child_death_task);
+ child_death_task = GNUNET_SCHEDULER_NO_TASK;
}
- stop_service (client, servicename);
}
shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
{
struct ServiceList *pos;
- struct ServiceList *next;
- struct ServiceList *prev;
-
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, _("Stopping all services\n"));
+ struct ServiceList *nxt;
+ struct ServiceListeningInfo *sli;
+
+ if (GNUNET_SCHEDULER_NO_TASK != child_restart_task)
+ {
+ GNUNET_SCHEDULER_cancel (child_restart_task);
+ child_restart_task = GNUNET_SCHEDULER_NO_TASK;
+ }
in_shutdown = GNUNET_YES;
- pos = running;
- prev = NULL;
- while (NULL != pos)
+ /* first, stop listening */
+ for (pos = running_head; NULL != pos; pos = pos->next)
{
- next = pos->next;
- if (pos->pid != 0)
+ while (NULL != (sli = pos->listen_head))
{
-#if DEBUG_ARM
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Sending SIGTERM to `%s'\n", pos->name);
-#endif
- if (0 != PLIBC_KILL (pos->pid, SIGTERM))
+ GNUNET_CONTAINER_DLL_remove (pos->listen_head,
+ pos->listen_tail, sli);
+ if (sli->accept_task != GNUNET_SCHEDULER_NO_TASK)
+ {
+ GNUNET_SCHEDULER_cancel (sli->accept_task);
+ sli->accept_task = GNUNET_SCHEDULER_NO_TASK;
+ }
+ GNUNET_break (GNUNET_OK ==
+ GNUNET_NETWORK_socket_close (sli->listen_socket));
+ GNUNET_free (sli->service_addr);
+ GNUNET_free (sli);
+ }
+ }
+ /* then, shutdown all existing service processes */
+ nxt = running_head;
+ while (NULL != (pos = nxt))
+ {
+ nxt = pos->next;
+ if (pos->proc != NULL)
+ {
+ GNUNET_log (GNUNET_ERROR_TYPE_INFO, "Stopping service `%s'\n",
+ pos->name);
+ pos->killed_at = GNUNET_TIME_absolute_get ();
+ if (0 != GNUNET_OS_process_kill (pos->proc, SIGTERM))
GNUNET_log_strerror (GNUNET_ERROR_TYPE_WARNING, "kill");
- prev = pos;
}
else
{
- if (prev == NULL)
- running = next;
- else
- prev->next = next;
- free_entry (pos);
+ free_service (pos);
}
- pos = next;
- }
- if (running == NULL)
- {
- GNUNET_SERVER_destroy (server);
- GNUNET_SIGNAL_handler_uninstall (shc_chld);
- shc_chld = NULL;
}
+ /* finally, should all service processes be already gone, terminate for real */
+ if (running_head == NULL)
+ do_shutdown ();
}
* @param tc context
*/
static void
-delayed_restart_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+delayed_restart_task (void *cls,
+ const struct GNUNET_SCHEDULER_TaskContext *tc)
{
- struct ServiceList *pos;
+ struct ServiceList *sl;
struct GNUNET_TIME_Relative lowestRestartDelay;
+ struct ServiceListeningInfo *sli;
child_restart_task = GNUNET_SCHEDULER_NO_TASK;
if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
return;
+ GNUNET_assert (GNUNET_NO == in_shutdown);
lowestRestartDelay = GNUNET_TIME_UNIT_FOREVER_REL;
/* check for services that need to be restarted due to
- configuration changes or because the last restart failed */
- pos = running;
- while (pos != NULL)
+ * configuration changes or because the last restart failed */
+ for (sl = running_head; NULL != sl; sl = sl->next)
{
- if ( (pos->pid == 0) &&
- (GNUNET_YES != in_shutdown) )
+ if (sl->proc == NULL)
{
- if (GNUNET_TIME_absolute_get_remaining (pos->restartAt).value == 0)
+ /* service is currently not running */
+ if (GNUNET_TIME_absolute_get_remaining (sl->restart_at).rel_value ==
+ 0)
{
- GNUNET_log (GNUNET_ERROR_TYPE_INFO,
- _("Restarting service `%s'.\n"), pos->name);
- start_process (pos);
+ /* restart is now allowed */
+ if (sl->is_default)
+ {
+ /* process should run by default, start immediately */
+ GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+ _("Restarting service `%s'.\n"), sl->name);
+ start_process (sl);
+ }
+ else
+ {
+ /* process is run on-demand, ensure it is re-started if there is demand */
+ for (sli = sl->listen_head; NULL != sli; sli = sli->next)
+ if (GNUNET_SCHEDULER_NO_TASK == sli->accept_task)
+ {
+ /* accept was actually paused, so start it again */
+ sli->accept_task =
+ GNUNET_SCHEDULER_add_read_net
+ (GNUNET_TIME_UNIT_FOREVER_REL, sli->listen_socket,
+ &accept_connection, sli);
+ }
+ }
}
else
{
- lowestRestartDelay
- = GNUNET_TIME_relative_min (lowestRestartDelay,
- GNUNET_TIME_absolute_get_remaining
- (pos->restartAt));
+ /* update calculation for earliest time to reactivate a service */
+ lowestRestartDelay =
+ GNUNET_TIME_relative_min (lowestRestartDelay,
+ GNUNET_TIME_absolute_get_remaining
+ (sl->restart_at));
}
}
- pos = pos->next;
}
- if (lowestRestartDelay.value != GNUNET_TIME_UNIT_FOREVER_REL.value)
- child_restart_task
- = GNUNET_SCHEDULER_add_delayed (sched,
- lowestRestartDelay,
- &delayed_restart_task,
- NULL);
+ if (lowestRestartDelay.rel_value != GNUNET_TIME_UNIT_FOREVER_REL.rel_value)
+ {
+#if DEBUG_ARM
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Will restart process in %llums\n",
+ (unsigned long long) lowestRestartDelay.rel_value);
+#endif
+ child_restart_task =
+ GNUNET_SCHEDULER_add_delayed_with_priority (lowestRestartDelay,
+ GNUNET_SCHEDULER_PRIORITY_IDLE,
+ &delayed_restart_task, NULL);
+ }
}
/**
- *
+ * Task triggered whenever we receive a SIGCHLD (child
+ * process died).
*
* @param cls closure, NULL if we need to self-restart
* @param tc context
maint_child_death (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
{
struct ServiceList *pos;
- struct ServiceList *prev;
struct ServiceList *next;
+ struct ServiceListeningInfo *sli;
const char *statstr;
int statcode;
int ret;
- char c;
+ char c[16];
+ enum GNUNET_OS_ProcessStatusType statusType;
+ unsigned long statusCode;
+ const struct GNUNET_DISK_FileHandle *pr;
+ pr = GNUNET_DISK_pipe_handle (sigpipe, GNUNET_DISK_PIPE_END_READ);
child_death_task = GNUNET_SCHEDULER_NO_TASK;
- if (0 != (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
+ if (0 == (tc->reason & GNUNET_SCHEDULER_REASON_READ_READY))
{
+ /* shutdown scheduled us, ignore! */
child_death_task =
- GNUNET_SCHEDULER_add_read_file (sched, GNUNET_TIME_UNIT_FOREVER_REL, pr,
- &maint_child_death, NULL);
- return;
+ GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL,
+ pr, &maint_child_death, NULL);
+ return;
}
/* consume the signal */
- GNUNET_DISK_file_read (pr, &c, sizeof (c));
+ GNUNET_break (0 < GNUNET_DISK_file_read (pr, &c, sizeof (c)));
/* check for services that died (WAITPID) */
- prev = NULL;
- next = running;
+ next = running_head;
while (NULL != (pos = next))
{
- enum GNUNET_OS_ProcessStatusType statusType;
- unsigned long statusCode;
-
next = pos->next;
- if (pos->pid == 0)
- {
- prev = pos;
- continue;
- }
- if ((GNUNET_SYSERR == (ret = GNUNET_OS_process_status (pos->pid,
- &statusType,
- &statusCode))) ||
- ( (ret == GNUNET_NO) ||
- (statusType == GNUNET_OS_PROCESS_STOPPED) ||
- (statusType == GNUNET_OS_PROCESS_RUNNING)) )
- {
- prev = pos;
- continue;
- }
+
+ if (pos->proc == NULL)
+ {
+ if (GNUNET_YES == in_shutdown)
+ free_service (pos);
+ continue;
+ }
+ if ((GNUNET_SYSERR ==
+ (ret =
+ GNUNET_OS_process_status (pos->proc, &statusType, &statusCode)))
+ || ((ret == GNUNET_NO) || (statusType == GNUNET_OS_PROCESS_STOPPED)
+ || (statusType == GNUNET_OS_PROCESS_RUNNING)))
+ continue;
if (statusType == GNUNET_OS_PROCESS_EXITED)
- {
- statstr = _( /* process termination method */ "exit");
- statcode = statusCode;
- }
+ {
+ statstr = _( /* process termination method */ "exit");
+ statcode = statusCode;
+ }
else if (statusType == GNUNET_OS_PROCESS_SIGNALED)
- {
- statstr = _( /* process termination method */ "signal");
- statcode = statusCode;
- }
+ {
+ statstr = _( /* process termination method */ "signal");
+ statcode = statusCode;
+ }
else
+ {
+ statstr = _( /* process termination method */ "unknown");
+ statcode = 0;
+ }
+ if (0 != pos->killed_at.abs_value)
+ {
+ GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+ _("Service `%s' took %llu ms to terminate\n"),
+ pos->name,
+ GNUNET_TIME_absolute_get_duration (pos->killed_at).rel_value);
+ }
+ GNUNET_OS_process_close (pos->proc);
+ pos->proc = NULL;
+ if (NULL != pos->killing_client)
{
- statstr = _( /* process termination method */ "unknown");
- statcode = 0;
- }
- pos->pid = 0;
- if (NULL != pos->kill_continuation)
- {
- if (prev == NULL)
- running = next;
- else
- prev->next = next;
- pos->kill_continuation (pos->kill_continuation_cls, pos);
+ signal_result (pos->killing_client, pos->name,
+ GNUNET_ARM_PROCESS_DOWN);
+ GNUNET_SERVER_client_drop (pos->killing_client);
+ pos->killing_client = NULL;
+ /* process can still be re-started on-demand, ensure it is re-started if there is demand */
+ for (sli = pos->listen_head; NULL != sli; sli = sli->next)
+ {
+ GNUNET_break (GNUNET_SCHEDULER_NO_TASK == sli->accept_task);
+ sli->accept_task =
+ GNUNET_SCHEDULER_add_read_net (GNUNET_TIME_UNIT_FOREVER_REL,
+ sli->listen_socket,
+ &accept_connection, sli);
+ }
continue;
}
if (GNUNET_YES != in_shutdown)
{
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _
- ("Service `%s' terminated with status %s/%d, will try to restart it!\n"),
- pos->name, statstr, statcode);
- /* schedule restart */
- pos->restartAt
- = GNUNET_TIME_relative_to_absolute (pos->backoff);
- if (pos->backoff.value < EXPONENTIAL_BACKOFF_THRESHOLD)
- pos->backoff
- = GNUNET_TIME_relative_multiply (pos->backoff, 2);
+ if ((statusType == GNUNET_OS_PROCESS_EXITED) && (statcode == 0))
+ {
+ /* process terminated normally, allow restart at any time */
+ pos->restart_at.abs_value = 0;
+ }
+ else
+ {
+ if (0 == (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
+ GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+ _
+ ("Service `%s' terminated with status %s/%d, will restart in %llu ms\n"),
+ pos->name, statstr, statcode, pos->backoff.rel_value);
+ /* schedule restart */
+ pos->restart_at = GNUNET_TIME_relative_to_absolute (pos->backoff);
+ pos->backoff =
+ GNUNET_TIME_relative_min (EXPONENTIAL_BACKOFF_THRESHOLD,
+ GNUNET_TIME_relative_multiply
+ (pos->backoff, 2));
+ }
if (GNUNET_SCHEDULER_NO_TASK != child_restart_task)
- GNUNET_SCHEDULER_cancel (sched, child_restart_task);
- child_restart_task
- = GNUNET_SCHEDULER_add_now (sched,
- &delayed_restart_task,
- NULL);
+ GNUNET_SCHEDULER_cancel (child_restart_task);
+ child_restart_task =
+ GNUNET_SCHEDULER_add_with_priority
+ (GNUNET_SCHEDULER_PRIORITY_IDLE,
+ &delayed_restart_task, NULL);
}
-#if DEBUG_ARM
else
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Service `%s' terminated with status %s/%d\n",
- pos->name, statstr, statcode);
-#endif
- prev = pos;
- }
- if ( (running == NULL) &&
- (in_shutdown) )
- {
- GNUNET_SERVER_destroy (server);
- GNUNET_SIGNAL_handler_uninstall (shc_chld);
- shc_chld = NULL;
+ {
+ free_service (pos);
+ }
}
- else
+ child_death_task =
+ GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL,
+ pr, &maint_child_death, NULL);
+ if ((NULL == running_head) && (GNUNET_YES == in_shutdown))
+ do_shutdown ();
+}
+
+
+/**
+ * Transmit our shutdown acknowledgement to the client.
+ *
+ * @param cls the 'struct GNUNET_SERVER_Client'
+ * @param size number of bytes available in buf
+ * @param buf where to write the message
+ * @return number of bytes written
+ */
+static size_t
+transmit_shutdown_ack (void *cls, size_t size, void *buf)
+{
+ struct GNUNET_SERVER_Client *client = cls;
+ struct GNUNET_ARM_ResultMessage *msg;
+
+ if (size < sizeof (struct GNUNET_ARM_ResultMessage))
{
- child_death_task =
- GNUNET_SCHEDULER_add_read_file (sched, GNUNET_TIME_UNIT_FOREVER_REL, pr,
- &maint_child_death, NULL);
+ GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+ _("Failed to transmit shutdown ACK.\n"));
+ GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
+ return 0; /* client disconnected */
}
+ /* Make the connection flushing for the purpose of ACK transmitting,
+ * needed on W32 to ensure that the message is even received, harmless
+ * on other platforms... */
+ GNUNET_break (GNUNET_OK == GNUNET_SERVER_client_disable_corking (client));
+ msg = (struct GNUNET_ARM_ResultMessage *) buf;
+ msg->header.type = htons (GNUNET_MESSAGE_TYPE_ARM_RESULT);
+ msg->header.size = htons (sizeof (struct GNUNET_ARM_ResultMessage));
+ msg->status = htonl ((uint32_t) GNUNET_ARM_PROCESS_SHUTDOWN);
+ GNUNET_SERVER_receive_done (client, GNUNET_OK);
+ GNUNET_SERVER_client_drop (client);
+ return sizeof (struct GNUNET_ARM_ResultMessage);
}
/**
- * List of handlers for the messages understood by this service.
+ * Handler for SHUTDOWN message.
+ *
+ * @param cls closure (refers to service)
+ * @param client identification of the client
+ * @param message the actual message
*/
-static struct GNUNET_SERVER_MessageHandler handlers[] = {
- {&handle_start, NULL, GNUNET_MESSAGE_TYPE_ARM_START, 0},
- {&handle_stop, NULL, GNUNET_MESSAGE_TYPE_ARM_STOP, 0},
- {NULL, NULL, 0, 0}
-};
+static void
+handle_shutdown (void *cls, struct GNUNET_SERVER_Client *client,
+ const struct GNUNET_MessageHeader *message)
+{
+ GNUNET_SCHEDULER_shutdown ();
+ GNUNET_SERVER_client_keep (client);
+ GNUNET_SERVER_notify_transmit_ready (client,
+ sizeof (struct GNUNET_ARM_ResultMessage),
+ GNUNET_TIME_UNIT_FOREVER_REL,
+ &transmit_shutdown_ack, client);
+ GNUNET_SERVER_client_persist_ (client);
+}
/**
- * Signal handler called for signals that should cause us to shutdown.
+ * Signal handler called for SIGCHLD. Triggers the
+ * respective handler by writing to the trigger pipe.
*/
static void
sighandler_child_death ()
{
static char c;
+ int old_errno = errno; /* back-up errno */
+
+ GNUNET_break (1 ==
+ GNUNET_DISK_file_write (GNUNET_DISK_pipe_handle
+ (sigpipe, GNUNET_DISK_PIPE_END_WRITE),
+ &c, sizeof (c)));
+ errno = old_errno; /* restore errno */
+}
+
+
+/**
+ * Setup our service record for the given section in the configuration file
+ * (assuming the section is for a service).
+ *
+ * @param cls unused
+ * @param section a section in the configuration file
+ * @return GNUNET_OK (continue)
+ */
+static void
+setup_service (void *cls, const char *section)
+{
+ struct ServiceList *sl;
+ char *binary;
+ char *config;
+ struct stat sbuf;
+ struct sockaddr **addrs;
+ socklen_t *addr_lens;
+ int ret;
+ unsigned int i;
- GNUNET_DISK_file_write (GNUNET_DISK_pipe_handle
- (sigpipe, GNUNET_DISK_PIPE_END_WRITE), &c,
- sizeof (c));
+ if (strcasecmp (section, "arm") == 0)
+ return;
+ if (GNUNET_OK !=
+ GNUNET_CONFIGURATION_get_value_string (cfg, section, "BINARY", &binary))
+ {
+ /* not a service section */
+ return;
+ }
+ sl = find_service (section);
+ if (NULL != sl)
+ {
+ /* got the same section twice!? */
+ GNUNET_break (0);
+ return;
+ }
+ config = NULL;
+ if ((GNUNET_OK !=
+ GNUNET_CONFIGURATION_get_value_filename (cfg, section, "CONFIG",
+ &config)) ||
+ (0 != STAT (config, &sbuf)))
+ {
+ GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
+ _
+ ("Configuration file `%s' for service `%s' not valid: %s\n"),
+ config, section,
+ (config == NULL) ? _("option missing") : STRERROR (errno));
+ GNUNET_free (binary);
+ GNUNET_free_non_null (config);
+ return;
+ }
+ sl = GNUNET_malloc (sizeof (struct ServiceList));
+ sl->name = GNUNET_strdup (section);
+ sl->binary = binary;
+ sl->config = config;
+ sl->backoff = GNUNET_TIME_UNIT_MILLISECONDS;
+ sl->restart_at = GNUNET_TIME_UNIT_FOREVER_ABS;
+#if WINDOWS
+ sl->pipe_control = GNUNET_YES;
+#else
+ if (GNUNET_CONFIGURATION_have_value (cfg, section, "PIPECONTROL"))
+ sl->pipe_control = GNUNET_CONFIGURATION_get_value_yesno (cfg, section, "PIPECONTROL");
+#endif
+ GNUNET_CONTAINER_DLL_insert (running_head, running_tail, sl);
+ if (GNUNET_YES !=
+ GNUNET_CONFIGURATION_get_value_yesno (cfg, section, "AUTOSTART"))
+ return;
+ if (0 >= (ret = GNUNET_SERVICE_get_server_addresses (section, cfg,
+ &addrs, &addr_lens)))
+ return;
+ /* this will free (or capture) addrs[i] */
+ for (i = 0; i < ret; i++)
+ create_listen_socket (addrs[i], addr_lens[i], sl);
+ GNUNET_free (addrs);
+ GNUNET_free (addr_lens);
}
* Process arm requests.
*
* @param cls closure
- * @param s scheduler to use
* @param serv the initialized server
* @param c configuration to use
*/
static void
-run (void *cls,
- struct GNUNET_SCHEDULER_Handle *s,
- struct GNUNET_SERVER_Handle *serv,
+run (void *cls, struct GNUNET_SERVER_Handle *serv,
const struct GNUNET_CONFIGURATION_Handle *c)
{
+ static const struct GNUNET_SERVER_MessageHandler handlers[] = {
+ {&handle_start, NULL, GNUNET_MESSAGE_TYPE_ARM_START, 0},
+ {&handle_stop, NULL, GNUNET_MESSAGE_TYPE_ARM_STOP, 0},
+ {&handle_shutdown, NULL, GNUNET_MESSAGE_TYPE_ARM_SHUTDOWN,
+ sizeof (struct GNUNET_MessageHeader)},
+ {&handle_list, NULL, GNUNET_MESSAGE_TYPE_ARM_LIST,
+ sizeof (struct GNUNET_MessageHeader)},
+ {NULL, NULL, 0, 0}
+ };
char *defaultservices;
- char *pos;
+ const char *pos;
+ struct ServiceList *sl;
cfg = c;
- sched = s;
server = serv;
GNUNET_assert (serv != NULL);
- shc_chld = GNUNET_SIGNAL_handler_install (SIGCHLD, &sighandler_child_death);
- GNUNET_assert (sigpipe == NULL);
- sigpipe = GNUNET_DISK_pipe (GNUNET_NO);
- GNUNET_assert (sigpipe != NULL);
- pr = GNUNET_DISK_pipe_handle (sigpipe, GNUNET_DISK_PIPE_END_READ);
- GNUNET_assert (pr != NULL);
GNUNET_SERVER_ignore_shutdown (serv, GNUNET_YES);
- GNUNET_SCHEDULER_add_delayed (sched,
- GNUNET_TIME_UNIT_FOREVER_REL,
- &shutdown_task,
+ GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &shutdown_task,
NULL);
child_death_task =
- GNUNET_SCHEDULER_add_read_file (sched, GNUNET_TIME_UNIT_FOREVER_REL, pr,
+ GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL,
+ GNUNET_DISK_pipe_handle (sigpipe,
+ GNUNET_DISK_PIPE_END_READ),
&maint_child_death, NULL);
if (GNUNET_OK !=
- GNUNET_CONFIGURATION_get_value_string (cfg,
- "ARM",
- "GLOBAL_PREFIX",
+ GNUNET_CONFIGURATION_get_value_string (cfg, "ARM", "GLOBAL_PREFIX",
&prefix_command))
prefix_command = GNUNET_strdup ("");
+ if (GNUNET_OK !=
+ GNUNET_CONFIGURATION_get_value_string (cfg, "ARM", "GLOBAL_POSTFIX",
+ &final_option))
+ final_option = GNUNET_strdup ("");
+
+ GNUNET_CONFIGURATION_iterate_sections (cfg, &setup_service, NULL);
+
/* start default services... */
if (GNUNET_OK ==
- GNUNET_CONFIGURATION_get_value_string (cfg,
- "ARM",
- "DEFAULTSERVICES",
+ GNUNET_CONFIGURATION_get_value_string (cfg, "ARM", "DEFAULTSERVICES",
&defaultservices))
{
-#if DEBUG_ARM
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Starting default services `%s'\n", defaultservices);
-#endif
- pos = strtok (defaultservices, " ");
- while (pos != NULL)
+ GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+ _("Starting default services `%s'\n"), defaultservices);
+ if (0 < strlen (defaultservices))
{
- start_service (NULL, pos);
- pos = strtok (NULL, " ");
+ for (pos = strtok (defaultservices, " "); NULL != pos;
+ pos = strtok (NULL, " "))
+ {
+ sl = find_service (pos);
+ if (NULL == sl)
+ {
+ GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+ _
+ ("Default service `%s' not configured correctly!\n"),
+ pos);
+ continue;
+ }
+ sl->is_default = GNUNET_YES;
+ start_process (sl);
+ }
}
GNUNET_free (defaultservices);
}
else
{
-#if DEBUG_ARM
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "No default services configured.\n");
-#endif
+ GNUNET_log (GNUNET_ERROR_TYPE_INFO,
+ _
+ ("No default services configured, GNUnet will not really start right now.\n"));
}
/* process client requests */
GNUNET_SERVER_add_handlers (server, handlers);
-
- /* manage services */
- GNUNET_SCHEDULER_add_with_priority (sched,
- GNUNET_SCHEDULER_PRIORITY_IDLE,
- &config_change_task, NULL);
}
int
main (int argc, char *const *argv)
{
- return (GNUNET_OK ==
- GNUNET_SERVICE_run (argc,
- argv, "arm", GNUNET_YES, &run, NULL)) ? 0 : 1;
+ int ret;
+ struct GNUNET_SIGNAL_Context *shc_chld;
+
+ sigpipe = GNUNET_DISK_pipe (GNUNET_NO, GNUNET_NO, GNUNET_NO, GNUNET_NO);
+ GNUNET_assert (sigpipe != NULL);
+ shc_chld =
+ GNUNET_SIGNAL_handler_install (GNUNET_SIGCHLD, &sighandler_child_death);
+ ret =
+ (GNUNET_OK ==
+ GNUNET_SERVICE_run (argc, argv, "arm", GNUNET_YES, &run, NULL)) ? 0 : 1;
+ GNUNET_SIGNAL_handler_uninstall (shc_chld);
+ shc_chld = NULL;
+ GNUNET_DISK_pipe_close (sigpipe);
+ sigpipe = NULL;
+ return ret;
+}
+
+
+#ifdef LINUX
+#include <malloc.h>
+
+/**
+ * MINIMIZE heap size (way below 128k) since this process doesn't need much.
+ */
+void __attribute__ ((constructor)) GNUNET_ARM_memory_init ()
+{
+ mallopt (M_TRIM_THRESHOLD, 4 * 1024);
+ mallopt (M_TOP_PAD, 1 * 1024);
+ malloc_trim (0);
}
+#endif
+
/* end of gnunet-service-arm.c */