2 This file is part of GNUnet.
3 Copyright (C) 2009, 2010, 2012, 2013 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 * @brief API for accessing the ARM service
24 * @author Christian Grothoff
28 #include "gnunet_util_lib.h"
29 #include "gnunet_arm_service.h"
30 #include "gnunet_protocols.h"
33 #define LOG(kind,...) GNUNET_log_from (kind, "arm-api",__VA_ARGS__)
36 * Handle for interacting with ARM.
38 struct GNUNET_ARM_Handle
41 * Our control connection to the ARM service.
43 struct GNUNET_CLIENT_Connection *client;
46 * The configuration that we are using.
48 struct GNUNET_CONFIGURATION_Handle *cfg;
51 * Handle for our current transmission request.
53 struct GNUNET_CLIENT_TransmitHandle *cth;
56 * Head of doubly-linked list of pending requests.
58 struct ARMControlMessage *control_pending_head;
61 * Tail of doubly-linked list of pending requests.
63 struct ARMControlMessage *control_pending_tail;
66 * Head of doubly-linked list of sent requests.
68 struct ARMControlMessage *control_sent_head;
71 * Tail of doubly-linked list of sent requests.
73 struct ARMControlMessage *control_sent_tail;
76 * Callback to invoke on connection/disconnection.
78 GNUNET_ARM_ConnectionStatusCallback conn_status;
81 * Closure for conn_status.
83 void *conn_status_cls;
86 * ARM control message for the 'arm_termination_handler'
87 * with the continuation to call once the ARM shutdown is done.
89 struct ARMControlMessage *thm;
92 * ID of the reconnect task (if any).
94 struct GNUNET_SCHEDULER_Task * reconnect_task;
97 * Current delay we use for re-trying to connect to core.
99 struct GNUNET_TIME_Relative retry_backoff;
102 * Counter for request identifiers
104 uint64_t request_id_counter;
107 * Are we currently disconnected and hence unable to send?
109 unsigned char currently_down;
112 * GNUNET_YES if we're running a service test.
114 unsigned char service_test_is_active;
119 * Entry in a doubly-linked list of control messages to be transmitted
120 * to the arm service.
122 * The actual message is allocated at the end of this struct.
124 struct ARMControlMessage
127 * This is a doubly-linked list.
129 struct ARMControlMessage *next;
132 * This is a doubly-linked list.
134 struct ARMControlMessage *prev;
139 struct GNUNET_ARM_Handle *h;
144 struct GNUNET_ARM_Message *msg;
147 * Callback for service state change requests.
149 GNUNET_ARM_ResultCallback result_cont;
152 * Callback for service list requests.
154 GNUNET_ARM_ServiceListCallback list_cont;
157 * Closure for 'result_cont' or 'list_cont'.
162 * Timeout for the operation.
164 struct GNUNET_TIME_Absolute timeout;
167 * Task to run when request times out.
169 struct GNUNET_SCHEDULER_Task * timeout_task_id;
172 * Flags for passing std descriptors to ARM (when starting ARM).
174 enum GNUNET_OS_InheritStdioFlags std_inheritance;
177 * Type of the request expressed as a message type (start, stop or list).
186 * @param h arm handle
187 * @return GNUNET_OK on success, GNUNET_SYSERR on failure
190 reconnect_arm (struct GNUNET_ARM_Handle *h);
194 * Check the list of pending requests, send the next
197 * @param h arm handle
198 * @param ignore_currently_down transmit message even if not initialized?
201 trigger_next_request (struct GNUNET_ARM_Handle *h, int ignore_currently_down);
205 * Task scheduled to try to re-connect to arm.
207 * @param cls the 'struct GNUNET_ARM_Handle'
208 * @param tc task context
211 reconnect_arm_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
213 struct GNUNET_ARM_Handle *h = cls;
215 h->reconnect_task = NULL;
216 LOG (GNUNET_ERROR_TYPE_DEBUG, "Connecting to ARM service after delay\n");
222 * Close down any existing connection to the ARM service and
223 * try re-establishing it later.
225 * @param h our handle
228 reconnect_arm_later (struct GNUNET_ARM_Handle *h)
230 if (GNUNET_NO != h->currently_down)
234 GNUNET_CLIENT_notify_transmit_ready_cancel (h->cth);
237 if (NULL != h->client)
239 GNUNET_CLIENT_disconnect (h->client);
242 h->currently_down = GNUNET_YES;
243 GNUNET_assert (NULL == h->reconnect_task);
245 GNUNET_SCHEDULER_add_delayed (h->retry_backoff, &reconnect_arm_task, h);
246 /* Don't clear pending messages on disconnection, deliver them later
247 clear_pending_messages (h, GNUNET_ARM_REQUEST_DISCONNECTED);
248 GNUNET_assert (NULL == h->control_pending_head);
250 h->retry_backoff = GNUNET_TIME_STD_BACKOFF (h->retry_backoff);
251 if (NULL != h->conn_status)
252 h->conn_status (h->conn_status_cls, GNUNET_NO);
257 * Find a control message by its unique ID.
259 * @param h ARM handle
260 * @param id unique message ID to use for the lookup
261 * @return NULL if not found
263 static struct ARMControlMessage *
264 find_cm_by_id (struct GNUNET_ARM_Handle *h, uint64_t id)
266 struct ARMControlMessage *result;
267 for (result = h->control_sent_head; result; result = result->next)
268 if (id == result->msg->request_id)
275 * Handler for ARM 'termination' reply (failure to receive).
277 * @param cls our "struct GNUNET_ARM_Handle"
278 * @param msg expected to be NULL
281 arm_termination_handler (void *cls, const struct GNUNET_MessageHeader *msg)
283 struct GNUNET_ARM_Handle *h = cls;
284 struct ARMControlMessage *cm;
289 GNUNET_CLIENT_receive (h->client, &arm_termination_handler, h,
290 GNUNET_TIME_UNIT_FOREVER_REL);
295 h->currently_down = GNUNET_YES;
296 GNUNET_CLIENT_disconnect (h->client);
298 if (NULL != cm->result_cont)
299 cm->result_cont (cm->cont_cls,
300 GNUNET_ARM_REQUEST_SENT_OK,
301 (const char *) &cm->msg[1],
302 GNUNET_ARM_RESULT_STOPPED);
303 GNUNET_free (cm->msg);
309 * Handler for ARM replies.
311 * @param cls our `struct GNUNET_ARM_Handle`
312 * @param msg the message received from the arm service
315 client_notify_handler (void *cls,
316 const struct GNUNET_MessageHeader *msg)
318 struct GNUNET_ARM_Handle *h = cls;
319 const struct GNUNET_ARM_Message *arm_msg;
320 const struct GNUNET_ARM_ResultMessage *res;
321 const struct GNUNET_ARM_ListResultMessage *lres;
322 struct ARMControlMessage *cm;
326 enum GNUNET_ARM_Result result;
336 LOG (GNUNET_ERROR_TYPE_INFO,
337 _("Client was disconnected from arm service, trying to reconnect.\n"));
338 reconnect_arm_later (h);
341 msize = ntohs (msg->size);
342 LOG (GNUNET_ERROR_TYPE_DEBUG,
343 "Processing message of type %u and size %u from arm service\n",
344 ntohs (msg->type), msize);
345 if (msize < sizeof (struct GNUNET_ARM_Message))
348 reconnect_arm_later (h);
351 arm_msg = (const struct GNUNET_ARM_Message *) msg;
352 GNUNET_break (0 == ntohl (arm_msg->reserved));
353 id = GNUNET_ntohll (arm_msg->request_id);
354 cm = find_cm_by_id (h, id);
357 LOG (GNUNET_ERROR_TYPE_DEBUG,
358 "Message with unknown id %llu\n",
363 switch (ntohs (msg->type))
365 case GNUNET_MESSAGE_TYPE_ARM_RESULT:
366 if (msize < sizeof (struct GNUNET_ARM_ResultMessage))
372 case GNUNET_MESSAGE_TYPE_ARM_LIST_RESULT:
373 if (msize < sizeof (struct GNUNET_ARM_ListResultMessage))
380 lres = (const struct GNUNET_ARM_ListResultMessage *) msg;
381 rcount = ntohs (lres->count);
385 list = GNUNET_malloc (sizeof (const char *) * rcount);
386 pos = (const char *)&lres[1];
387 for (i = 0; i < rcount; i++)
389 const char *end = memchr (pos, 0, msize - size_check);
397 size_check += (end - pos) + 1;
400 if (GNUNET_YES == fail)
411 GNUNET_assert (NULL != cm->timeout_task_id);
412 GNUNET_SCHEDULER_cancel (cm->timeout_task_id);
413 GNUNET_CONTAINER_DLL_remove (h->control_sent_head,
414 h->control_sent_tail, cm);
415 if (GNUNET_YES == fail)
417 reconnect_arm_later (h);
418 GNUNET_free (cm->msg);
422 if ( (GNUNET_MESSAGE_TYPE_ARM_RESULT == ntohs (msg->type)) &&
423 (0 == strcasecmp ((const char *) &cm->msg[1],
425 (NULL != (res = (const struct GNUNET_ARM_ResultMessage *) msg)) &&
426 (GNUNET_ARM_RESULT_STOPPING == ntohl (res->result)) )
428 /* special case: if we are stopping 'gnunet-service-arm', we do not just
429 wait for the result message, but also wait for the service to close
430 the connection (and then we have to close our client handle as well);
431 this is done by installing a different receive handler, waiting for
432 the connection to go down */
436 cm->result_cont (h->thm->cont_cls,
437 GNUNET_ARM_REQUEST_SENT_OK,
438 (const char *) &h->thm->msg[1],
439 GNUNET_ARM_RESULT_IS_NOT_KNOWN);
440 GNUNET_free (h->thm->msg);
441 GNUNET_free (h->thm);
444 GNUNET_CLIENT_receive (h->client, &arm_termination_handler, h,
445 GNUNET_TIME_UNIT_FOREVER_REL);
448 GNUNET_CLIENT_receive (h->client, &client_notify_handler, h,
449 GNUNET_TIME_UNIT_FOREVER_REL);
450 switch (ntohs (msg->type))
452 case GNUNET_MESSAGE_TYPE_ARM_RESULT:
453 res = (const struct GNUNET_ARM_ResultMessage *) msg;
454 LOG (GNUNET_ERROR_TYPE_DEBUG,
455 "Received response from ARM for service `%s': %u\n",
456 (const char *) &cm->msg[1], ntohs (msg->type));
457 result = (enum GNUNET_ARM_Result) ntohl (res->result);
458 if (NULL != cm->result_cont)
459 cm->result_cont (cm->cont_cls, GNUNET_ARM_REQUEST_SENT_OK,
460 (const char *) &cm->msg[1], result);
462 case GNUNET_MESSAGE_TYPE_ARM_LIST_RESULT:
463 if (NULL != cm->list_cont)
464 cm->list_cont (cm->cont_cls, GNUNET_ARM_REQUEST_SENT_OK, rcount,
466 GNUNET_free_non_null (list);
469 GNUNET_free (cm->msg);
475 * Transmit the next message to the arm service.
477 * @param cls closure with the `struct GNUNET_ARM_Handle`
478 * @param size number of bytes available in @a buf
479 * @param buf where the callee should write the message
480 * @return number of bytes written to @a buf
483 transmit_arm_message (void *cls, size_t size, void *buf)
485 struct GNUNET_ARM_Handle *h = cls;
486 struct ARMControlMessage *cm;
487 struct GNUNET_ARM_Message *arm_msg;
489 int notify_connection;
492 notify_connection = GNUNET_NO;
493 LOG (GNUNET_ERROR_TYPE_DEBUG,
494 "transmit_arm_message is running with %p buffer of size %lu. ARM is known to be %s\n",
495 buf, size, h->currently_down ? "unconnected" : "connected");
496 GNUNET_assert (NULL == h->reconnect_task);
498 if ((GNUNET_YES == h->currently_down) && (NULL != buf))
500 h->currently_down = GNUNET_NO;
501 notify_connection = GNUNET_YES;
502 h->retry_backoff = GNUNET_TIME_UNIT_MILLISECONDS;
503 GNUNET_CLIENT_receive (h->client, &client_notify_handler, h,
504 GNUNET_TIME_UNIT_FOREVER_REL);
508 LOG (GNUNET_ERROR_TYPE_DEBUG,
509 "Transmission failed, initiating reconnect\n");
510 reconnect_arm_later (h);
513 if (NULL == (cm = h->control_pending_head))
515 LOG (GNUNET_ERROR_TYPE_DEBUG,
516 "Queue is empty, not sending anything\n");
520 GNUNET_assert (NULL != cm->msg);
521 msize = ntohs (cm->msg->header.size);
524 LOG (GNUNET_ERROR_TYPE_DEBUG,
525 "Request is too big (%u < %u), not sending it\n", size, msize);
526 trigger_next_request (h, GNUNET_NO);
531 if (0 == h->request_id_counter)
532 h->request_id_counter++;
533 request_id = h->request_id_counter++;
534 LOG (GNUNET_ERROR_TYPE_DEBUG,
535 "Transmitting control message with %u bytes of type %u to arm with id %llu\n",
536 (unsigned int) msize, (unsigned int) ntohs (cm->msg->header.type), request_id);
537 arm_msg->reserved = htonl (0);
538 arm_msg->request_id = GNUNET_htonll (request_id);
539 memcpy (buf, cm->msg, msize);
540 /* Otherwise we won't be able to find it later! */
541 arm_msg->request_id = request_id;
542 GNUNET_CONTAINER_DLL_remove (h->control_pending_head,
543 h->control_pending_tail, cm);
544 GNUNET_CONTAINER_DLL_insert_tail (h->control_sent_head,
545 h->control_sent_tail, cm);
546 /* Don't free msg, keep it around (kind of wasteful, but then we don't
547 * really have many messages to handle, and it'll be freed when it times
550 trigger_next_request (h, GNUNET_NO);
553 if ((GNUNET_YES == notify_connection) && (NULL != h->conn_status))
554 h->conn_status (h->conn_status_cls, GNUNET_YES);
560 * Check the list of pending requests, send the next
563 * @param h arm handle
564 * @param ignore_currently_down transmit message even if not initialized?
567 trigger_next_request (struct GNUNET_ARM_Handle *h, int ignore_currently_down)
571 msize = sizeof (struct GNUNET_MessageHeader);
572 if ((GNUNET_YES == h->currently_down) && (ignore_currently_down == GNUNET_NO))
574 LOG (GNUNET_ERROR_TYPE_DEBUG,
575 "ARM connection down, not processing queue\n");
580 LOG (GNUNET_ERROR_TYPE_DEBUG, "Request pending, not processing queue\n");
583 if (NULL != h->control_pending_head)
585 ntohs (h->control_pending_head->msg->header.size);
586 else if (GNUNET_NO == ignore_currently_down)
588 LOG (GNUNET_ERROR_TYPE_DEBUG,
589 "Request queue empty, not processing queue\n");
590 return; /* no pending message */
593 GNUNET_CLIENT_notify_transmit_ready (h->client, msize,
594 GNUNET_TIME_UNIT_FOREVER_REL,
595 GNUNET_NO, &transmit_arm_message, h);
602 * @param h arm handle
603 * @return GNUNET_OK on success, GNUNET_SYSERR on failure
606 reconnect_arm (struct GNUNET_ARM_Handle *h)
608 GNUNET_assert (NULL == h->client);
609 GNUNET_assert (GNUNET_YES == h->currently_down);
610 h->client = GNUNET_CLIENT_connect ("arm", h->cfg);
611 if (NULL == h->client)
613 LOG (GNUNET_ERROR_TYPE_DEBUG,
614 "arm_api, GNUNET_CLIENT_connect returned NULL\n");
615 if (NULL != h->conn_status)
616 h->conn_status (h->conn_status_cls, GNUNET_SYSERR);
617 return GNUNET_SYSERR;
619 LOG (GNUNET_ERROR_TYPE_DEBUG,
620 "arm_api, GNUNET_CLIENT_connect returned non-NULL\n");
621 trigger_next_request (h, GNUNET_YES);
627 * Set up a context for communicating with ARM, then
628 * start connecting to the ARM service using that context.
630 * @param cfg configuration to use (needed to contact ARM;
631 * the ARM service may internally use a different
632 * configuration to determine how to start the service).
633 * @param conn_status will be called when connecting/disconnecting
634 * @param cls closure for conn_status
635 * @return context to use for further ARM operations, NULL on error.
637 struct GNUNET_ARM_Handle *
638 GNUNET_ARM_connect (const struct GNUNET_CONFIGURATION_Handle *cfg,
639 GNUNET_ARM_ConnectionStatusCallback conn_status, void *cls)
641 struct GNUNET_ARM_Handle *h;
643 h = GNUNET_new (struct GNUNET_ARM_Handle);
644 h->cfg = GNUNET_CONFIGURATION_dup (cfg);
645 h->currently_down = GNUNET_YES;
646 h->reconnect_task = NULL;
647 h->conn_status = conn_status;
648 h->conn_status_cls = cls;
649 if (GNUNET_OK != reconnect_arm (h))
659 * Disconnect from the ARM service (if connected) and destroy the context.
661 * @param h the handle that was being used
664 GNUNET_ARM_disconnect_and_free (struct GNUNET_ARM_Handle *h)
666 struct ARMControlMessage *cm;
668 LOG (GNUNET_ERROR_TYPE_DEBUG, "Disconnecting from ARM service\n");
671 GNUNET_CLIENT_notify_transmit_ready_cancel (h->cth);
674 while ((NULL != (cm = h->control_pending_head))
675 || (NULL != (cm = h->control_sent_head)) )
677 if (NULL != h->control_pending_head)
678 GNUNET_CONTAINER_DLL_remove (h->control_pending_head,
679 h->control_pending_tail, cm);
681 GNUNET_CONTAINER_DLL_remove (h->control_sent_head,
682 h->control_sent_tail, cm);
683 GNUNET_assert (NULL != cm->timeout_task_id);
684 GNUNET_SCHEDULER_cancel (cm->timeout_task_id);
685 if (NULL != cm->result_cont)
686 cm->result_cont (cm->cont_cls, GNUNET_ARM_REQUEST_DISCONNECTED,
688 /* FIXME: What about list callback? */
689 GNUNET_free_non_null (cm->msg);
692 if (NULL != h->client)
694 GNUNET_CLIENT_disconnect (h->client);
697 if (NULL != h->reconnect_task)
699 GNUNET_SCHEDULER_cancel (h->reconnect_task);
700 h->reconnect_task = NULL;
702 if (GNUNET_NO == h->service_test_is_active)
704 GNUNET_CONFIGURATION_destroy (h->cfg);
711 * Message timed out. Remove it from the queue.
713 * @param cls the message (struct ARMControlMessage *)
714 * @param tc task context
717 control_message_timeout (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
719 struct ARMControlMessage *cm = cls;
720 struct GNUNET_ARM_Message *arm_msg;
722 LOG (GNUNET_ERROR_TYPE_DEBUG,
723 "Control message timed out\n");
725 if ((NULL == arm_msg) || (0 == arm_msg->request_id))
727 GNUNET_CONTAINER_DLL_remove (cm->h->control_pending_head,
728 cm->h->control_pending_tail, cm);
732 GNUNET_CONTAINER_DLL_remove (cm->h->control_sent_head,
733 cm->h->control_sent_tail, cm);
735 if (NULL != cm->result_cont)
736 cm->result_cont (cm->cont_cls, GNUNET_ARM_REQUEST_TIMEOUT, NULL, 0);
737 else if (NULL != cm->list_cont)
738 cm->list_cont (cm->cont_cls, GNUNET_ARM_REQUEST_TIMEOUT, 0, NULL);
739 GNUNET_free_non_null (cm->msg);
745 * A client specifically requested starting of ARM itself.
746 * This function is called with information about whether
747 * or not ARM is running; if it is, report success. If
748 * it is not, start the ARM process.
750 * @param cls the context for the request that we will report on (struct ARMControlMessage *)
751 * @param result GNUNET_YES if ARM is running
754 arm_service_report (void *cls,
757 struct ARMControlMessage *cm = cls;
758 struct GNUNET_ARM_Handle *h;
759 struct GNUNET_OS_Process *proc;
760 unsigned char test_is_active;
768 test_is_active = cm->h->service_test_is_active;
769 if ((GNUNET_YES == test_is_active) &&
770 (GNUNET_YES == result))
772 LOG (GNUNET_ERROR_TYPE_DEBUG,
773 "Looks like `%s' is already running.\n",
774 "gnunet-service-arm");
775 /* arm is running! */
777 cm->result_cont (cm->cont_cls,
778 GNUNET_ARM_REQUEST_SENT_OK, "arm",
779 GNUNET_ARM_RESULT_IS_STARTED_ALREADY);
781 if (GNUNET_NO == test_is_active)
783 /* User disconnected & destroyed ARM handle in the middle of
784 * the service test, so we kept the handle around until now.
786 GNUNET_CONFIGURATION_destroy (cm->h->cfg);
789 if ((GNUNET_YES == result) ||
790 (GNUNET_NO == test_is_active))
795 cm->h->service_test_is_active = GNUNET_NO;
796 LOG (GNUNET_ERROR_TYPE_DEBUG,
797 "Looks like `%s' is not running, will start it.\n",
798 "gnunet-service-arm");
799 if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_string (
800 cm->h->cfg, "arm", "PREFIX", &loprefix))
801 loprefix = GNUNET_strdup ("");
802 if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_string (
803 cm->h->cfg, "arm", "OPTIONS", &lopostfix))
804 lopostfix = GNUNET_strdup ("");
805 if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_string (
806 cm->h->cfg, "arm", "BINARY", &cbinary))
808 GNUNET_log_config_missing (GNUNET_ERROR_TYPE_WARNING, "arm", "BINARY");
810 cm->result_cont (cm->cont_cls,
811 GNUNET_ARM_REQUEST_SENT_OK, "arm",
812 GNUNET_ARM_RESULT_IS_NOT_KNOWN);
814 GNUNET_free (loprefix);
815 GNUNET_free (lopostfix);
818 if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_filename (
819 cm->h->cfg, "arm", "CONFIG", &config))
821 binary = GNUNET_OS_get_libexec_binary_path (cbinary);
822 GNUNET_asprintf ("edbinary,
825 GNUNET_free (cbinary);
826 if ((GNUNET_YES == GNUNET_CONFIGURATION_have_value (
827 cm->h->cfg, "TESTING", "WEAKRANDOM")) &&
828 (GNUNET_YES == GNUNET_CONFIGURATION_get_value_yesno (
829 cm->h->cfg, "TESTING", "WEAKRANDOM")) &&
830 (GNUNET_NO == GNUNET_CONFIGURATION_have_value (
831 cm->h->cfg, "TESTING", "HOSTFILE")))
833 /* Means we are ONLY running locally */
834 /* we're clearly running a test, don't daemonize */
836 proc = GNUNET_OS_start_process_s (GNUNET_NO, cm->std_inheritance,
837 NULL, loprefix, quotedbinary,
838 /* no daemonization! */
841 proc = GNUNET_OS_start_process_s (GNUNET_NO, cm->std_inheritance,
842 NULL, loprefix, quotedbinary, "-c", config,
843 /* no daemonization! */
849 proc = GNUNET_OS_start_process_s (GNUNET_NO, cm->std_inheritance,
850 NULL, loprefix, quotedbinary,
851 "-d", lopostfix, NULL);
853 proc = GNUNET_OS_start_process_s (GNUNET_NO, cm->std_inheritance,
854 NULL, loprefix, quotedbinary, "-c",
856 "-d", lopostfix, NULL);
858 GNUNET_free (binary);
859 GNUNET_free (quotedbinary);
860 GNUNET_free_non_null (config);
861 GNUNET_free (loprefix);
862 GNUNET_free (lopostfix);
866 cm->result_cont (cm->cont_cls, GNUNET_ARM_REQUEST_SENT_OK, "arm",
867 GNUNET_ARM_RESULT_START_FAILED);
872 cm->result_cont (cm->cont_cls, GNUNET_ARM_REQUEST_SENT_OK, "arm",
873 GNUNET_ARM_RESULT_STARTING);
874 GNUNET_OS_process_destroy (proc);
882 * Start or stop a service.
884 * @param h handle to ARM
885 * @param service_name name of the service
886 * @param timeout how long to wait before failing for good
887 * @param cb callback to invoke when service is ready
888 * @param cb_cls closure for callback
889 * @param type type of the request
892 change_service (struct GNUNET_ARM_Handle *h, const char *service_name,
893 struct GNUNET_TIME_Relative timeout, GNUNET_ARM_ResultCallback cb,
894 void *cb_cls, uint16_t type)
896 struct ARMControlMessage *cm;
898 struct GNUNET_ARM_Message *msg;
900 slen = strlen (service_name) + 1;
901 if (slen + sizeof (struct GNUNET_ARM_Message) >=
902 GNUNET_SERVER_MAX_MESSAGE_SIZE)
906 cb (cb_cls, GNUNET_ARM_REQUEST_TOO_LONG, NULL, 0);
909 LOG (GNUNET_ERROR_TYPE_DEBUG, "Requesting %s of service `%s'.\n",
910 (GNUNET_MESSAGE_TYPE_ARM_START == type) ? "start" : "termination",
912 cm = GNUNET_malloc (sizeof (struct ARMControlMessage) + slen);
914 cm->result_cont = cb;
915 cm->cont_cls = cb_cls;
916 cm->timeout = GNUNET_TIME_relative_to_absolute (timeout);
917 memcpy (&cm[1], service_name, slen);
918 msg = GNUNET_malloc (sizeof (struct GNUNET_ARM_Message) + slen);
919 msg->header.size = htons (sizeof (struct GNUNET_ARM_Message) + slen);
920 msg->header.type = htons (type);
921 msg->reserved = htonl (0);
922 memcpy (&msg[1], service_name, slen);
924 LOG (GNUNET_ERROR_TYPE_DEBUG,
925 "Inserting a control message into the queue. Timeout is %s\n",
926 GNUNET_STRINGS_relative_time_to_string (GNUNET_TIME_absolute_get_remaining (cm->timeout),
928 GNUNET_CONTAINER_DLL_insert_tail (h->control_pending_head,
929 h->control_pending_tail, cm);
930 cm->timeout_task_id =
931 GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_absolute_get_remaining
932 (cm->timeout), &control_message_timeout, cm);
933 trigger_next_request (h, GNUNET_NO);
938 * Request for a service to be started.
940 * @param h handle to ARM
941 * @param service_name name of the service
942 * @param std_inheritance inheritance of std streams
943 * @param timeout how long to wait before failing for good
944 * @param cont callback to invoke after request is sent or not sent
945 * @param cont_cls closure for callback
948 GNUNET_ARM_request_service_start (struct GNUNET_ARM_Handle *h,
949 const char *service_name,
950 enum GNUNET_OS_InheritStdioFlags std_inheritance,
951 struct GNUNET_TIME_Relative timeout,
952 GNUNET_ARM_ResultCallback cont,
955 struct ARMControlMessage *cm;
958 LOG (GNUNET_ERROR_TYPE_DEBUG,
959 "Asked to start service `%s' within %s\n", service_name,
960 GNUNET_STRINGS_relative_time_to_string (timeout, GNUNET_NO));
961 if (0 == strcasecmp ("arm", service_name))
964 * 1) We're connected to ARM already. Invoke the callback immediately.
965 * 2) We're not connected to ARM.
966 * Cancel any reconnection attempts temporarily, then perform
969 if (GNUNET_NO == h->currently_down)
971 LOG (GNUNET_ERROR_TYPE_DEBUG, "ARM is already running\n");
973 cont (cont_cls, GNUNET_ARM_REQUEST_SENT_OK, "arm", GNUNET_ARM_RESULT_IS_STARTED_ALREADY);
975 else if (GNUNET_NO == h->service_test_is_active)
979 GNUNET_CLIENT_notify_transmit_ready_cancel (h->cth);
982 if (NULL != h->client)
984 GNUNET_CLIENT_disconnect (h->client);
987 if (NULL != h->reconnect_task)
989 GNUNET_SCHEDULER_cancel (h->reconnect_task);
990 h->reconnect_task = NULL;
993 LOG (GNUNET_ERROR_TYPE_DEBUG,
994 "Not connected to ARM, will do a service test\n");
996 slen = strlen ("arm") + 1;
997 cm = GNUNET_malloc (sizeof (struct ARMControlMessage) + slen);
999 cm->result_cont = cont;
1000 cm->cont_cls = cont_cls;
1001 cm->timeout = GNUNET_TIME_relative_to_absolute (timeout);
1002 cm->std_inheritance = std_inheritance;
1003 memcpy (&cm[1], service_name, slen);
1004 h->service_test_is_active = GNUNET_YES;
1005 GNUNET_CLIENT_service_test ("arm", h->cfg, timeout, &arm_service_report,
1010 /* Service test is already running - tell user to chill out and try
1013 LOG (GNUNET_ERROR_TYPE_DEBUG, "Service test is already in progress, we're busy\n");
1015 cont (cont_cls, GNUNET_ARM_REQUEST_BUSY, NULL, 0);
1019 change_service (h, service_name, timeout, cont, cont_cls,
1020 GNUNET_MESSAGE_TYPE_ARM_START);
1025 * Request a service to be stopped.
1026 * Stopping arm itself will not invalidate its handle, and
1027 * ARM API will try to restore connection to the ARM service,
1028 * even if ARM connection was lost because you asked for ARM to be stopped.
1029 * Call GNUNET_ARM_disconnect_and_free () to free the handle and prevent
1030 * further connection attempts.
1032 * @param h handle to ARM
1033 * @param service_name name of the service
1034 * @param timeout how long to wait before failing for good
1035 * @param cont callback to invoke after request is sent or is not sent
1036 * @param cont_cls closure for callback
1039 GNUNET_ARM_request_service_stop (struct GNUNET_ARM_Handle *h,
1040 const char *service_name,
1041 struct GNUNET_TIME_Relative timeout,
1042 GNUNET_ARM_ResultCallback cont,
1045 LOG (GNUNET_ERROR_TYPE_DEBUG,
1046 "Stopping service `%s' within %s\n",
1048 GNUNET_STRINGS_relative_time_to_string (timeout, GNUNET_NO));
1049 change_service (h, service_name, timeout, cont, cont_cls,
1050 GNUNET_MESSAGE_TYPE_ARM_STOP);
1055 * Request a list of running services.
1057 * @param h handle to ARM
1058 * @param timeout how long to wait before failing for good
1059 * @param cont callback to invoke after request is sent or is not sent
1060 * @param cont_cls closure for callback
1063 GNUNET_ARM_request_service_list (struct GNUNET_ARM_Handle *h,
1064 struct GNUNET_TIME_Relative timeout,
1065 GNUNET_ARM_ServiceListCallback cont,
1068 struct ARMControlMessage *cm;
1069 struct GNUNET_ARM_Message *msg;
1071 LOG (GNUNET_ERROR_TYPE_DEBUG,
1072 "Requesting LIST from ARM service with timeout: %s\n",
1073 GNUNET_STRINGS_relative_time_to_string (timeout, GNUNET_YES));
1074 cm = GNUNET_new (struct ARMControlMessage);
1076 cm->list_cont = cont;
1077 cm->cont_cls = cont_cls;
1078 cm->timeout = GNUNET_TIME_relative_to_absolute (timeout);
1079 msg = GNUNET_malloc (sizeof (struct GNUNET_ARM_Message));
1080 msg->header.size = htons (sizeof (struct GNUNET_ARM_Message));
1081 msg->header.type = htons (GNUNET_MESSAGE_TYPE_ARM_LIST);
1082 msg->reserved = htonl (0);
1084 GNUNET_CONTAINER_DLL_insert_tail (h->control_pending_head,
1085 h->control_pending_tail, cm);
1086 cm->timeout_task_id =
1087 GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_absolute_get_remaining
1088 (cm->timeout), &control_message_timeout, cm);
1089 trigger_next_request (h, GNUNET_NO);
1093 /* end of arm_api.c */