Adds hooks for service (svc)
[src/app-framework-binder.git] / src / main.c
index 18f5e5e..f5512a2 100644 (file)
  */
 
 #define _GNU_SOURCE
-#define NO_BINDING_VERBOSE_MACRO
+#define AFB_BINDING_PRAGMA_NO_VERBOSE_MACRO
 
 #include <stdlib.h>
 #include <stdio.h>
+#include <stdint.h>
+#include <signal.h>
 #include <string.h>
 #include <unistd.h>
 #include <fcntl.h>
 #include <sys/stat.h>
 #include <sys/wait.h>
 
-#include <systemd/sd-event.h>
-#include <systemd/sd-daemon.h>
+#include <json-c/json.h>
 
-#include <afb/afb-binding.h>
+#include <systemd/sd-daemon.h>
 
 #include "afb-config.h"
 #include "afb-hswitch.h"
-#include "afb-apis.h"
+#include "afb-apiset.h"
 #include "afb-api-so.h"
 #include "afb-api-dbus.h"
 #include "afb-api-ws.h"
 #include "afb-hsrv.h"
-#include "afb-context.h"
 #include "afb-hreq.h"
+#include "afb-xreq.h"
 #include "jobs.h"
 #include "afb-session.h"
 #include "verbose.h"
 #include "afb-common.h"
+#include "afb-monitor.h"
 #include "afb-hook.h"
 #include "sd-fds.h"
 
@@ -54,6 +56,8 @@
 */
 #define SELF_PGROUP 1
 
+struct afb_apiset *main_apiset;
+
 static struct afb_config *config;
 static pid_t childpid;
 
@@ -79,12 +83,12 @@ static struct afb_config_list *run_for_list(struct afb_config_list *list,
 
 static int run_start(void *closure, char *value)
 {
-       int (*starter) (const char *value) = closure;
-       return starter(value) >= 0;
+       int (*starter) (const char *value, struct afb_apiset *apiset) = closure;
+       return starter(value, main_apiset) >= 0;
 }
 
-static void start_list(struct afb_config_list *list,
-                      int (*starter) (const char *value), const char *message)
+static void apiset_start_list(struct afb_config_list *list,
+                      int (*starter) (const char *value, struct afb_apiset *apiset), const char *message)
 {
        list = run_for_list(list, run_start, starter);
        if (list) {
@@ -109,6 +113,7 @@ static void exit_handler()
                killpg(0, SIGTERM);
        else if (childpid > 0)
                killpg(childpid, SIGTERM);
+       exit(0);
 }
 
 static void on_sigterm(int signum, siginfo_t *info, void *uctx)
@@ -211,11 +216,11 @@ static int init_alias(void *closure, char *spec)
 static int init_http_server(struct afb_hsrv *hsrv)
 {
        if (!afb_hsrv_add_handler
-           (hsrv, config->rootapi, afb_hswitch_websocket_switch, NULL, 20))
+           (hsrv, config->rootapi, afb_hswitch_websocket_switch, main_apiset, 20))
                return 0;
 
        if (!afb_hsrv_add_handler
-           (hsrv, config->rootapi, afb_hswitch_apis, NULL, 10))
+           (hsrv, config->rootapi, afb_hswitch_apis, main_apiset, 10))
                return 0;
 
        if (run_for_list(config->aliases, init_alias, hsrv))
@@ -430,6 +435,97 @@ static int execute_command()
        return -1;
 }
 
+/*---------------------------------------------------------
+ | startup calls
+ +--------------------------------------------------------- */
+
+struct startup_req
+{
+       struct afb_xreq xreq;
+       char *api;
+       char *verb;
+       struct afb_config_list *current;
+       struct afb_session *session;
+};
+
+static void startup_call_reply(struct afb_xreq *xreq, int iserror, struct json_object *obj)
+{
+       struct startup_req *sreq = CONTAINER_OF_XREQ(struct startup_req, xreq);
+
+       if (!iserror)
+               NOTICE("startup call %s returned %s", sreq->current->value, json_object_get_string(obj));
+       else {
+               ERROR("startup call %s ERROR! %s", sreq->current->value, json_object_get_string(obj));
+               exit(1);
+       }
+}
+
+static void startup_call_current(struct startup_req *sreq);
+
+static void startup_call_unref(struct afb_xreq *xreq)
+{
+       struct startup_req *sreq = CONTAINER_OF_XREQ(struct startup_req, xreq);
+
+       free(sreq->api);
+       free(sreq->verb);
+       json_object_put(sreq->xreq.json);
+       sreq->current = sreq->current->next;
+       if (sreq->current)
+               startup_call_current(sreq);
+       else {
+               afb_session_close(sreq->session);
+               afb_session_unref(sreq->session);
+               free(sreq);
+       }
+}
+
+static struct afb_xreq_query_itf startup_xreq_itf =
+{
+       .reply = startup_call_reply,
+       .unref = startup_call_unref
+};
+
+static void startup_call_current(struct startup_req *sreq)
+{
+       char *api, *verb, *json;
+
+       api = sreq->current->value;
+       verb = strchr(api, '/');
+       if (verb) {
+               json = strchr(verb, ':');
+               if (json) {
+                       afb_xreq_init(&sreq->xreq, &startup_xreq_itf);
+                       afb_context_init(&sreq->xreq.context, sreq->session, NULL);
+                       sreq->xreq.context.validated = 1;
+                       sreq->api = strndup(api, verb - api);
+                       sreq->verb = strndup(verb + 1, json - verb - 1);
+                       sreq->xreq.api = sreq->api;
+                       sreq->xreq.verb = sreq->verb;
+                       sreq->xreq.json = json_tokener_parse(json + 1);
+                       if (sreq->api && sreq->verb && sreq->xreq.json) {
+                               afb_xreq_process(&sreq->xreq, main_apiset);
+                               return;
+                       }
+               }
+       }
+       ERROR("Bad call specification %s", sreq->current->value);
+       exit(1);
+}
+
+static void run_startup_calls()
+{
+       struct afb_config_list *list;
+       struct startup_req *sreq;
+
+       list = config->calls;
+       if (list) {
+               sreq = calloc(1, sizeof *sreq);
+               sreq->session = afb_session_create("startup", 3600);
+               sreq->current = list;
+               startup_call_current(sreq);
+       }
+}
+
 /*---------------------------------------------------------
  | job for starting the daemon
  +--------------------------------------------------------- */
@@ -456,32 +552,42 @@ static void start()
        }
 
        /* configure the daemon */
-       afb_apis_set_timeout(config->apiTimeout);
        afb_session_init(config->nbSessionMax, config->cntxTimeout, config->token);
        if (!afb_hreq_init_cookie(config->httpdPort, config->rootapi, config->cntxTimeout)) {
                ERROR("initialisation of cookies failed");
                goto error;
        }
+       main_apiset = afb_apiset_create("main", config->apiTimeout);
+       if (!main_apiset) {
+               ERROR("can't create main api set");
+               goto error;
+       }
+       if (afb_monitor_init() < 0) {
+               ERROR("failed to setup monitor");
+               goto error;
+       }
 
        /* install hooks */
        if (config->tracereq)
                afb_hook_create_xreq(NULL, NULL, NULL, config->tracereq, NULL, NULL);
        if (config->traceditf)
                afb_hook_create_ditf(NULL, config->traceditf, NULL, NULL);
+       if (config->tracesvc)
+               afb_hook_create_svc(NULL, config->tracesvc, NULL, NULL);
 
        /* load bindings */
-       start_list(config->dbus_clients, afb_api_dbus_add_client, "the afb-dbus client");
-       start_list(config->ws_clients, afb_api_ws_add_client, "the afb-websocket client");
-       start_list(config->ldpaths, afb_api_so_add_pathset, "the binding path set");
-       start_list(config->so_bindings, afb_api_so_add_binding, "the binding");
+       apiset_start_list(config->dbus_clients, afb_api_dbus_add_client, "the afb-dbus client");
+       apiset_start_list(config->ws_clients, afb_api_ws_add_client, "the afb-websocket client");
+       apiset_start_list(config->ldpaths, afb_api_so_add_pathset, "the binding path set");
+       apiset_start_list(config->so_bindings, afb_api_so_add_binding, "the binding");
 
-       start_list(config->dbus_servers, afb_api_dbus_add_server, "the afb-dbus service");
-       start_list(config->ws_servers, afb_api_ws_add_server, "the afb-websocket service");
+       apiset_start_list(config->dbus_servers, afb_api_dbus_add_server, "the afb-dbus service");
+       apiset_start_list(config->ws_servers, afb_api_ws_add_server, "the afb-websocket service");
 
        DEBUG("Init config done");
 
        /* start the services */
-       if (afb_apis_start_all_services(1) < 0)
+       if (afb_apiset_start_all_services(main_apiset, 1) < 0)
                goto error;
 
        /* start the HTTP server */
@@ -497,6 +603,9 @@ static void start()
 
        /* ready */
        sd_notify(1, "READY=1");
+
+       /* run the startup calls */
+       run_startup_calls();
        return;
 error:
        exit(1);