Adds hooks for service (svc)

[src/app-framework-binder.git] / src / afb-svc.c

diff --git a/src/afb-svc.c b/src/afb-svc.c
index ac01291..97312e6 100644 (file)
--- a/src/afb-svc.c
+++ b/src/afb-svc.c
@@ -34,26 +34,12 @@
 #include "afb-xreq.h"
 #include "afb-cred.h"
 #include "afb-apiset.h"
+#include "afb-hook.h"
 #include "jobs.h"
 #include "verbose.h"
 
-/*
- * Structure for recording service
- */
-struct afb_svc
-{
-       /* session of the service */
-       struct afb_session *session;
-
-       /* the apiset for the service */
-       struct afb_apiset *apiset;
 
-       /* event listener of the service or NULL */
-       struct afb_evt_listener *listener;
-
-       /* on event callback for the service */
-       void (*on_event)(const char *event, struct json_object *object);
-};
+#define HOOK(x,...) if((svc)->hookflags & afb_hook_flag_svc_##x) afb_hook_svc_##x(__VA_ARGS__)
 
 /*
  * Structure for requests initiated by the service
@@ -128,6 +114,7 @@ static void svc_free(struct afb_svc *svc)
  * Allocates a new service
  */
 static struct afb_svc *afb_svc_alloc(
+                       const char *api,
                        struct afb_apiset *apiset,
                        int share_session
 )
@@ -142,6 +129,7 @@ static struct afb_svc *afb_svc_alloc(
        }
 
        /* instanciate the apiset */
+       svc->api = api;
        svc->apiset = afb_apiset_addref(apiset);
 
        /* instanciate the session */
@@ -160,6 +148,7 @@ static struct afb_svc *afb_svc_alloc(
                        goto error;
        }
 
+       svc->hookflags = afb_hook_flags_svc(svc->api);
        return svc;
 
 error:
@@ -171,17 +160,18 @@ error:
  * Creates a new service
  */
 struct afb_svc *afb_svc_create_v1(
-               struct afb_apiset *apiset,
-               int share_session,
-               int (*start)(struct afb_service service),
-               void (*on_event)(const char *event, struct json_object *object)
+                       const char *api,
+                       struct afb_apiset *apiset,
+                       int share_session,
+                       int (*start)(struct afb_service service),
+                       void (*on_event)(const char *event, struct json_object *object)
 )
 {
        int rc;
        struct afb_svc *svc;
 
        /* allocates the svc handler */
-       svc = afb_svc_alloc(apiset, share_session);
+       svc = afb_svc_alloc(api, apiset, share_session);
        if (svc == NULL)
                goto error;
 
@@ -194,9 +184,13 @@ struct afb_svc *afb_svc_create_v1(
        }
 
        /* initialises the svc now */
-       rc = start(to_afb_service(svc));
-       if (rc < 0)
-               goto error;
+       if (start) {
+               HOOK(start_before, svc);
+               rc = start(to_afb_service(svc));
+               HOOK(start_after, svc, rc);
+               if (rc < 0)
+                       goto error;
+       }
 
        return svc;
 
@@ -209,6 +203,7 @@ error:
  * Creates a new service
  */
 struct afb_svc *afb_svc_create_v2(
+                       const char *api,
                        struct afb_apiset *apiset,
                        int share_session,
                        int (*start)(),
@@ -220,7 +215,7 @@ struct afb_svc *afb_svc_create_v2(
        struct afb_svc *svc;
 
        /* allocates the svc handler */
-       svc = afb_svc_alloc(apiset, share_session);
+       svc = afb_svc_alloc(api, apiset, share_session);
        if (svc == NULL)
                goto error;
        data->service = to_afb_service(svc);
@@ -235,7 +230,9 @@ struct afb_svc *afb_svc_create_v2(
 
        /* starts the svc if needed */
        if (start) {
+               HOOK(start_before, svc);
                rc = start();
+               HOOK(start_after, svc, rc);
                if (rc < 0)
                        goto error;
        }
@@ -247,13 +244,21 @@ error:
        return NULL;
 }
 
+void afb_svc_update_hook(struct afb_svc *svc)
+{
+       svc->hookflags = afb_hook_flags_svc(svc->api);
+}
+
 /*
  * Propagates the event to the service
  */
 static void svc_on_event(void *closure, const char *event, int eventid, struct json_object *object)
 {
        struct afb_svc *svc = closure;
+
+       HOOK(on_event_before, svc, event, eventid, object);
        svc->on_event(event, object);
+       HOOK(on_event_after, svc, event, eventid, object);
        json_object_put(object);
 }
 
@@ -275,7 +280,6 @@ static struct svc_req *svcreq_create(struct afb_svc *svc, const char *api, const
                afb_xreq_init(&svcreq->xreq, &afb_svc_xreq_itf);
                afb_context_init(&svcreq->xreq.context, svc->session, NULL);
                svcreq->xreq.context.validated = 1;
-               svcreq->xreq.cred = afb_cred_current();
                copy = (char*)&svcreq[1];
                memcpy(copy, api, lenapi);
                svcreq->xreq.api = copy;
@@ -316,7 +320,9 @@ static void svcreq_reply(struct afb_xreq *xreq, int iserror, json_object *obj)
 {
        struct svc_req *svcreq = CONTAINER_OF_XREQ(struct svc_req, xreq);
        if (svcreq->callback) {
+               struct afb_svc *svc = svcreq->svc;
                svcreq->callback(svcreq->closure, iserror, obj);
+               HOOK(call_result, svc, iserror, obj);
                json_object_put(obj);
        } else {
                svcreq->iserror = iserror;
@@ -348,6 +354,8 @@ static void svc_call(void *closure, const char *api, const char *verb, struct js
        struct svc_req *svcreq;
        struct json_object *ierr;
 
+       HOOK(call, svc, api, verb, args);
+
        /* allocates the request */
        svcreq = svcreq_create(svc, api, verb, args);
        if (svcreq == NULL) {
@@ -355,6 +363,7 @@ static void svc_call(void *closure, const char *api, const char *verb, struct js
                json_object_put(args);
                ierr = afb_msg_json_internal_error();
                callback(cbclosure, 1, ierr);
+               HOOK(call_result, svc, 1, ierr);
                json_object_put(ierr);
                return;
        }
@@ -375,6 +384,8 @@ static int svc_call_sync(void *closure, const char *api, const char *verb, struc
        struct svc_req *svcreq;
        int rc;
 
+       HOOK(callsync, svc, api, verb, args);
+
        /* allocates the request */
        svcreq = svcreq_create(svc, api, verb, args);
        if (svcreq == NULL) {
@@ -382,19 +393,20 @@ static int svc_call_sync(void *closure, const char *api, const char *verb, struc
                errno = ENOMEM;
                json_object_put(args);
                *result = afb_msg_json_internal_error();
-               return -1;
+               rc = 0;
+       } else {
+               /* initialises the request */
+               svcreq->jobloop = NULL;
+               svcreq->callback = NULL;
+               svcreq->result = NULL;
+               svcreq->iserror = 1;
+               afb_xreq_addref(&svcreq->xreq);
+               rc = jobs_enter(NULL, 0, svcreq_sync_enter, svcreq);
+               rc = rc >= 0 && !svcreq->iserror;
+               *result = (rc || svcreq->result) ? svcreq->result : afb_msg_json_internal_error();
+               afb_xreq_unref(&svcreq->xreq);
        }
-
-       /* initialises the request */
-       svcreq->jobloop = NULL;
-       svcreq->callback = NULL;
-       svcreq->result = NULL;
-       svcreq->iserror = 1;
-       afb_xreq_addref(&svcreq->xreq);
-       rc = jobs_enter(NULL, 0, svcreq_sync_enter, svcreq);
-       rc = rc >= 0 && !svcreq->iserror;
-       *result = (rc || svcreq->result) ? svcreq->result : afb_msg_json_internal_error();
-       afb_xreq_unref(&svcreq->xreq);
+       HOOK(callsync_result, svc, rc, *result);
        return rc;
 }