*/
void *conn_status_cls;
+ /**
+ * ARM control message for the 'arm_termination_handler'
+ * with the continuation to call once the ARM shutdown is done.
+ */
+ struct ARMControlMessage *thm;
+
/**
* ID of the reconnect task (if any).
*/
}
+/**
+ * Handler for ARM 'termination' reply (failure to receive).
+ *
+ * @param cls our "struct GNUNET_ARM_Handle"
+ * @param msg expected to be NULL
+ */
+static void
+arm_termination_handler (void *cls, const struct GNUNET_MessageHeader *msg)
+{
+ struct GNUNET_ARM_Handle *h = cls;
+ struct ARMControlMessage *cm;
+
+ if (NULL != msg)
+ {
+ GNUNET_break (0);
+ GNUNET_CLIENT_receive (h->client, &arm_termination_handler, h,
+ GNUNET_TIME_UNIT_FOREVER_REL);
+ return;
+ }
+ cm = h->thm;
+ h->thm = NULL;
+ h->currently_down = GNUNET_YES;
+ GNUNET_CLIENT_disconnect (h->client);
+ h->client = NULL;
+ if (NULL != cm->result_cont)
+ cm->result_cont (cm->cont_cls,
+ GNUNET_ARM_REQUEST_SENT_OK,
+ (const char *) &cm->msg[1],
+ GNUNET_ARM_RESULT_STOPPED);
+ GNUNET_free (cm->msg);
+ GNUNET_free (cm);
+}
+
+
/**
* Handler for ARM replies.
*
GNUNET_free (cm);
return;
}
+ if ( (GNUNET_MESSAGE_TYPE_ARM_RESULT == ntohs (msg->type)) &&
+ (0 == strcasecmp ((const char *) &cm->msg[1],
+ "arm")) &&
+ (NULL != (res = (const struct GNUNET_ARM_ResultMessage *) msg)) &&
+ (GNUNET_ARM_RESULT_STOPPING == ntohl (res->result)) )
+ {
+ /* special case: if we are stopping 'gnunet-service-arm', we do not just
+ wait for the result message, but also wait for the service to close
+ the connection (and then we have to close our client handle as well);
+ this is done by installing a different receive handler, waiting for
+ the connection to go down */
+ if (NULL != h->thm)
+ {
+ GNUNET_break (0);
+ cm->result_cont (h->thm->cont_cls,
+ GNUNET_ARM_REQUEST_SENT_OK,
+ (const char *) &h->thm->msg[1],
+ GNUNET_ARM_RESULT_IS_NOT_KNOWN);
+ GNUNET_free (h->thm->msg);
+ GNUNET_free (h->thm);
+ }
+ h->thm = cm;
+ GNUNET_CLIENT_receive (h->client, &arm_termination_handler, h,
+ GNUNET_TIME_UNIT_FOREVER_REL);
+ return;
+ }
GNUNET_CLIENT_receive (h->client, &client_notify_handler, h,
GNUNET_TIME_UNIT_FOREVER_REL);
switch (ntohs (msg->type))
memcpy (&msg[1], service_name, slen);
cm->msg = msg;
LOG (GNUNET_ERROR_TYPE_DEBUG,
- "Inserting a control message into the queue. Timeout is %llu\n",
- GNUNET_TIME_absolute_get_remaining (cm->timeout).rel_value);
+ "Inserting a control message into the queue. Timeout is %s\n",
+ GNUNET_STRINGS_relative_time_to_string (GNUNET_TIME_absolute_get_remaining (cm->timeout),
+ GNUNET_NO));
GNUNET_CONTAINER_DLL_insert_tail (h->control_pending_head,
h->control_pending_tail, cm);
cm->timeout_task_id =
*/
void
GNUNET_ARM_request_service_stop (struct GNUNET_ARM_Handle *h,
- const char *service_name, struct GNUNET_TIME_Relative timeout,
- GNUNET_ARM_ResultCallback cont, void *cont_cls)
+ const char *service_name,
+ struct GNUNET_TIME_Relative timeout,
+ GNUNET_ARM_ResultCallback cont,
+ void *cont_cls)
{
LOG (GNUNET_ERROR_TYPE_DEBUG,
"Stopping service `%s' within %s\n",
*/
static int quiet;
+/**
+ * Monitor ARM activity.
+ */
+static int monitor;
+
/**
* Set if we should print a list of currently running services.
*/
/**
* Attempts to delete configuration file and SERVICEHOME
- * on arm shutdown provided the end and delete options
+ * on ARM shutdown provided the end and delete options
* were specified when gnunet-arm was run.
*/
static void
config_file, dir);
if (UNLINK (config_file) != 0)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Failed to remove configuration file %s\n"), config_file);
- }
-
+ {
+ GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
+ _("Failed to remove configuration file %s\n"), config_file);
+ }
if (GNUNET_DISK_directory_remove (dir) != GNUNET_OK)
- {
- GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
- _("Failed to remove servicehome directory %s\n"), dir);
-
- }
+ {
+ GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
+ _("Failed to remove servicehome directory %s\n"), dir);
+
+ }
}
static void
shutdown_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
{
- GNUNET_ARM_disconnect_and_free (h);
- GNUNET_ARM_monitor_disconnect_and_free (m);
- h = NULL;
- m = NULL;
+ if (NULL != h)
+ {
+ GNUNET_ARM_disconnect_and_free (h);
+ h = NULL;
+ }
+ if (NULL != m)
+ {
+ GNUNET_ARM_monitor_disconnect_and_free (m);
+ m = NULL;
+ }
if ((end == GNUNET_YES) && (delete == GNUNET_YES))
delete_files ();
GNUNET_CONFIGURATION_destroy (cfg);
case GNUNET_ARM_REQUEST_SENT_OK:
return _("Message was sent successfully");
case GNUNET_ARM_REQUEST_CONFIGURATION_ERROR:
- return _("Misconfiguration (can't connect to the ARM service)");
+ return _("Misconfiguration (can not connect to the ARM service)");
case GNUNET_ARM_REQUEST_DISCONNECTED:
return _("We disconnected from ARM before we could send a request");
case GNUNET_ARM_REQUEST_BUSY:
return _("ARM API is busy");
case GNUNET_ARM_REQUEST_TOO_LONG:
- return _("Request doesn't fit into a message");
+ return _("Request does not fit into a message");
case GNUNET_ARM_REQUEST_TIMEOUT:
return _("Request timed out");
}
case GNUNET_ARM_RESULT_START_FAILED:
return _("%s service failed to start");
case GNUNET_ARM_RESULT_IN_SHUTDOWN:
- return _("%s service can't be started because ARM is shutting down");
+ return _("%s service cannot be started because ARM is shutting down");
}
return _("%.s Unknown result code.");
}
/**
* Function called whenever we connect to or disconnect from ARM.
+ * Termiantes the process if we fail to connect to the service on
+ * our first attempt.
*
* @param cls closure
* @param connected GNUNET_YES if connected, GNUNET_NO if disconnected,
conn_status (void *cls,
int connected)
{
- if (GNUNET_SYSERR == connected)
+ static int once;
+
+ if ( (GNUNET_SYSERR == connected) &&
+ (0 == once) )
{
GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
_("Fatal error initializing ARM API.\n"));
GNUNET_SCHEDULER_shutdown ();
return;
}
+ once = 1;
}
+/**
+ * We have requested ARM to be started, this function
+ * is called with the result of the operation. Informs the
+ * use of the result; on success, we continue with the event
+ * loop, on failure we terminate the process.
+ *
+ * @param cls closure unused
+ * @param rs what happened to our request
+ * @param service name of the service we tried to start ("arm")
+ * @param result if the request was processed, this is the result
+ * according to ARM
+ */
static void
-term_callback (void *cls,
- enum GNUNET_ARM_RequestStatus rs, const char *service,
- enum GNUNET_ARM_Result result)
+start_callback (void *cls,
+ enum GNUNET_ARM_RequestStatus rs, const char *service,
+ enum GNUNET_ARM_Result result)
{
+ char *msg;
+
if (GNUNET_ARM_REQUEST_SENT_OK != rs)
{
- char *msg;
- GNUNET_asprintf (&msg,
- _("Failed to send a request to kill the `%s' service: %%s\n"),
- term);
+ GNUNET_asprintf (&msg, "%s", _("Failed to start the ARM service: %s\n"));
FPRINTF (stdout, msg, req_string (rs));
GNUNET_free (msg);
GNUNET_SCHEDULER_shutdown ();
+ return;
}
- if ((GNUNET_ARM_RESULT_STOPPED == result) ||
- (GNUNET_ARM_RESULT_IS_STOPPED_ALREADY == result))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Service %s shutdown successful\n", term);
- term = NULL;
- GNUNET_SCHEDULER_add_now (action_loop, NULL);
- }
- else
+ if ( (GNUNET_ARM_RESULT_STARTING != result) &&
+ (GNUNET_ARM_RESULT_IS_STARTED_ALREADY != result) )
{
- char *msg;
- GNUNET_asprintf (&msg, _("Failed to kill the `%s' service: %s\n"),
- term, ret_string (result));
- FPRINTF (stdout, msg, service);
+ GNUNET_asprintf (&msg, "%s", _("Failed to start the ARM service: %s\n"));
+ FPRINTF (stdout, msg, ret_string (result));
GNUNET_free (msg);
GNUNET_SCHEDULER_shutdown ();
- }
+ return;
+ }
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "ARM service [re]start successful\n");
+ start = 0;
+ GNUNET_SCHEDULER_add_now (action_loop, NULL);
}
+/**
+ * We have requested ARM to be stopped, this function
+ * is called with the result of the operation. Informs the
+ * use of the result; on success, we continue with the event
+ * loop, on failure we terminate the process.
+ *
+ * @param cls closure unused
+ * @param rs what happened to our request
+ * @param service name of the service we tried to start ("arm")
+ * @param result if the request was processed, this is the result
+ * according to ARM
+ */
static void
-end_callback (void *cls,
- enum GNUNET_ARM_RequestStatus rs, const char *service,
- enum GNUNET_ARM_Result result)
+stop_callback (void *cls,
+ enum GNUNET_ARM_RequestStatus rs, const char *service,
+ enum GNUNET_ARM_Result result)
{
+ char *msg;
+
if (GNUNET_ARM_REQUEST_SENT_OK != rs)
{
- char *msg;
-
GNUNET_asprintf (&msg, "%s",
_("Failed to send a stop request to the ARM service: %s\n"));
FPRINTF (stdout, msg, req_string (rs));
GNUNET_free (msg);
GNUNET_SCHEDULER_shutdown ();
+ return;
}
- if ((GNUNET_ARM_RESULT_STOPPING == result) ||
- (GNUNET_ARM_RESULT_STOPPED == result) ||
- (GNUNET_ARM_RESULT_IS_STOPPED_ALREADY == result))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "ARM service shutdown successful\n");
- end = 0;
- if (restart)
- {
- restart = 0;
- start = 1;
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Initiating an ARM restart\n");
- }
- GNUNET_SCHEDULER_add_now (action_loop, NULL);
- }
- else
+ if ((GNUNET_ARM_RESULT_STOPPING != result) &&
+ (GNUNET_ARM_RESULT_STOPPED != result) &&
+ (GNUNET_ARM_RESULT_IS_STOPPED_ALREADY != result))
{
- char *msg;
-
GNUNET_asprintf (&msg, "%s", _("Failed to stop the ARM service: %s\n"));
FPRINTF (stdout, msg, ret_string (result));
GNUNET_free (msg);
GNUNET_SCHEDULER_shutdown ();
+ return;
+ }
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "ARM service shutdown successful\n");
+ end = 0;
+ if (restart)
+ {
+ restart = 0;
+ start = 1;
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "Initiating an ARM restart\n");
}
+ GNUNET_SCHEDULER_add_now (&action_loop, NULL);
}
+/**
+ * We have requested a service to be started, this function
+ * is called with the result of the operation. Informs the
+ * use of the result; on success, we continue with the event
+ * loop, on failure we terminate the process.
+ *
+ * @param cls closure unused
+ * @param rs what happened to our request
+ * @param service name of the service we tried to start
+ * @param result if the request was processed, this is the result
+ * according to ARM
+ */
static void
-start_callback (void *cls,
- enum GNUNET_ARM_RequestStatus rs, const char *service,
- enum GNUNET_ARM_Result result)
+init_callback (void *cls,
+ enum GNUNET_ARM_RequestStatus rs, const char *service,
+ enum GNUNET_ARM_Result result)
{
char *msg;
-
+
if (GNUNET_ARM_REQUEST_SENT_OK != rs)
{
- GNUNET_asprintf (&msg, "%s", _("Failed to start the ARM service: %s\n"));
+ GNUNET_asprintf (&msg, _("Failed to send a request to start the `%s' service: %%s\n"), init);
FPRINTF (stdout, msg, req_string (rs));
GNUNET_free (msg);
GNUNET_SCHEDULER_shutdown ();
return;
}
- if (! ((GNUNET_ARM_RESULT_STARTING == result) ||
- (GNUNET_ARM_RESULT_IS_STARTED_ALREADY == result)) )
+ if ((GNUNET_ARM_RESULT_STARTING != result) &&
+ (GNUNET_ARM_RESULT_IS_STARTED_ALREADY != result))
{
- GNUNET_asprintf (&msg, "%s", _("Failed to start the ARM service: %s\n"));
- FPRINTF (stdout, msg, ret_string (result));
+ GNUNET_asprintf (&msg, _("Failed to start the `%s' service: %s\n"),
+ init, ret_string (result));
+ FPRINTF (stdout, msg, service);
GNUNET_free (msg);
GNUNET_SCHEDULER_shutdown ();
- }
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "ARM service [re]start successful\n");
- start = 0;
- GNUNET_SCHEDULER_add_now (action_loop, NULL);
+ return;
+ }
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "Service %s [re]started successfully\n",
+ init);
+ init = NULL;
+ GNUNET_SCHEDULER_add_now (&action_loop, NULL);
}
+/**
+ * We have requested a service to be stopped, this function
+ * is called with the result of the operation. Informs the
+ * use of the result; on success, we continue with the event
+ * loop, on failure we terminate the process.
+ *
+ * @param cls closure unused
+ * @param rs what happened to our request
+ * @param service name of the service we tried to start
+ * @param result if the request was processed, this is the result
+ * according to ARM
+ */
static void
-init_callback (void *cls,
+term_callback (void *cls,
enum GNUNET_ARM_RequestStatus rs, const char *service,
enum GNUNET_ARM_Result result)
{
+ char *msg;
if (GNUNET_ARM_REQUEST_SENT_OK != rs)
{
- char *msg;
-
- GNUNET_asprintf (&msg, _("Failed to send a request to start the `%s' service: %%s\n"), init);
+ GNUNET_asprintf (&msg,
+ _("Failed to send a request to kill the `%s' service: %%s\n"),
+ term);
FPRINTF (stdout, msg, req_string (rs));
GNUNET_free (msg);
GNUNET_SCHEDULER_shutdown ();
+ return;
}
- if ((GNUNET_ARM_RESULT_STARTING == result) ||
- (GNUNET_ARM_RESULT_IS_STARTED_ALREADY == result))
- {
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Service %s [re]start successful\n", init);
- init = NULL;
- GNUNET_SCHEDULER_add_now (action_loop, NULL);
- }
- else
+ if ((GNUNET_ARM_RESULT_STOPPED != result) &&
+ (GNUNET_ARM_RESULT_IS_STOPPED_ALREADY != result))
{
- char *msg;
- GNUNET_asprintf (&msg, _("Failed to start the `%s' service: %s\n"),
- init, ret_string (result));
+ GNUNET_asprintf (&msg,
+ _("Failed to kill the `%s' service: %s\n"),
+ term, ret_string (result));
FPRINTF (stdout, msg, service);
GNUNET_free (msg);
GNUNET_SCHEDULER_shutdown ();
+ return;
}
+
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+ "Service %s stopped successfully\n", term);
+ term = NULL;
+ GNUNET_SCHEDULER_add_now (&action_loop, NULL);
}
+/**
+ * Function called with the list of running services. Prints
+ * the list to stdout, then starts the event loop again.
+ * Prints an error message and terminates the process on errors.
+ *
+ * @param cls closure (unused)
+ * @param rs request status (success, failure, etc.)
+ * @param count number of services in the list
+ * @param list list of services that are running
+ */
static void
list_callback (void *cls,
enum GNUNET_ARM_RequestStatus rs, unsigned int count,
const char *const*list)
{
unsigned int i;
+
if (GNUNET_ARM_REQUEST_SENT_OK != rs)
{
char *msg;
+
GNUNET_asprintf (&msg, "%s", _("Failed to request a list of services: %s\n"));
FPRINTF (stdout, msg, req_string (rs));
GNUNET_free (msg);
if (NULL == list)
{
FPRINTF (stderr, "%s", _("Error communicating with ARM. ARM not running?\n"));
+ GNUNET_SCHEDULER_shutdown ();
return;
}
FPRINTF (stdout, "%s", _("Running services:\n"));
for (i = 0; i < count; i++)
FPRINTF (stdout, "%s\n", list[i]);
- GNUNET_SCHEDULER_add_now (action_loop, NULL);
+ GNUNET_SCHEDULER_add_now (&action_loop, NULL);
}
/**
- * Main action loop. Runs the various
- * jobs that we've been asked to do in order.
+ * Main action loop. Runs the various jobs that we've been asked to
+ * do, in order.
*
* @param cls closure, unused
* @param tc context, unused
if (NULL != term)
{
GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Termination action\n");
- GNUNET_ARM_request_service_stop (h, term, (0 ==
- timeout.rel_value) ? STOP_TIMEOUT : timeout,
- term_callback, NULL);
+ GNUNET_ARM_request_service_stop (h, term,
+ (0 == timeout.rel_value) ? STOP_TIMEOUT : timeout,
+ &term_callback, NULL);
return;
}
break;
if (end || restart)
{
GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "End action\n");
- GNUNET_ARM_request_service_stop (h, "arm", (0 ==
- timeout.rel_value) ? STOP_TIMEOUT_ARM : timeout,
- end_callback, NULL);
+ GNUNET_ARM_request_service_stop (h, "arm",
+ (0 == timeout.rel_value) ? STOP_TIMEOUT_ARM : timeout,
+ &stop_callback, NULL);
return;
}
break;
{
GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Initialization action\n");
GNUNET_ARM_request_service_start (h, init, GNUNET_OS_INHERIT_STD_NONE,
- (0 == timeout.rel_value) ? STOP_TIMEOUT : timeout,
- init_callback, NULL);
+ (0 == timeout.rel_value) ? STOP_TIMEOUT : timeout,
+ &init_callback, NULL);
return;
}
break;
if (list)
{
GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
- "Going to list all running services controlled by ARM.\n");
-
+ "Going to list all running services controlled by ARM.\n");
GNUNET_ARM_request_service_list (h,
- (0 == timeout.rel_value) ? LIST_TIMEOUT : timeout,
- list_callback, &list);
+ (0 == timeout.rel_value) ? LIST_TIMEOUT : timeout,
+ &list_callback, &list);
return;
- }
- /* Fall through */
+ }
+ break;
+ case 5:
+ if (monitor)
+ {
+ if (! quiet)
+ fprintf (stderr,
+ _("Now only monitoring, press CTRL-C to stop.\n"));
+ quiet = 0; /* does not make sense to stay quiet in monitor mode at this time */
+ return; /* done with tasks, just monitor */
+ }
+ break;
default: /* last phase */
GNUNET_SCHEDULER_shutdown ();
return;
const char *service, enum GNUNET_ARM_ServiceStatus status)
{
const char *msg;
+
switch (status)
{
case GNUNET_ARM_SERVICE_MONITORING_STARTED:
else
FPRINTF (stderr, _("Unknown status %u for service %s.\n"), status, service);
}
- GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got service %s status %u\n", service, status);
+ GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got service %s status %d\n", service, (int) status);
}
* @param c configuration
*/
static void
-run (void *cls, char *const *args, const char *cfgfile,
+run (void *cls,
+ char *const *args,
+ const char *cfgfile,
const struct GNUNET_CONFIGURATION_Handle *c)
{
char *armconfig;
else
GNUNET_free (armconfig);
}
- h = GNUNET_ARM_connect (cfg, &conn_status, NULL);
- if (NULL != h)
- {
+ if (NULL == (h = GNUNET_ARM_connect (cfg, &conn_status, NULL)))
+ return;
+ if (monitor)
m = GNUNET_ARM_monitor (cfg, &srv_status, NULL);
- if (NULL != m)
- {
- GNUNET_SCHEDULER_add_now (&action_loop, NULL);
- GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
- shutdown_task, NULL);
- }
- else
- {
- GNUNET_ARM_disconnect_and_free (h);
- h = NULL;
- }
- }
+ GNUNET_SCHEDULER_add_now (&action_loop, NULL);
+ GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL,
+ &shutdown_task, NULL);
}
{'d', "delete", NULL,
gettext_noop ("delete config file and directory on exit"),
GNUNET_NO, &GNUNET_GETOPT_set_one, &delete},
+ {'m', "monitor", NULL,
+ gettext_noop ("monitor ARM activities"),
+ GNUNET_NO, &GNUNET_GETOPT_set_one, &monitor},
{'q', "quiet", NULL, gettext_noop ("don't print status messages"),
GNUNET_NO, &GNUNET_GETOPT_set_one, &quiet},
{'T', "timeout", "MSECS",
projects / oweals / gnunet.git / commitdiff