hooks: Allow to remove hooking (and/or trace)

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

diff --git a/src/afb-session.c b/src/afb-session.c
index adfed2a..70e8c3c 100644 (file)
--- a/src/afb-session.c
+++ b/src/afb-session.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2015-2018 "IoT.bzh"
+ * Copyright (C) 2015-2019 "IoT.bzh"
  * Author "Fulup Ar Foll"
  * Author: José Bollo <jose.bollo@iot.bzh>
  *
@@ -26,6 +26,7 @@
 #include <string.h>
 #include <uuid/uuid.h>
 #include <errno.h>
+#include <unistd.h>
 
 #include "afb-session.h"
 #include "afb-hook.h"
@@ -64,6 +65,7 @@ struct afb_session
        time_t expiration;      /**< expiration time of the token */
        pthread_mutex_t mutex;  /**< mutex of the session */
        struct cookie *cookies[COOKIECOUNT]; /**< cookies of the session */
+       char *lang;             /**< current language setting for the session */
        uint8_t closed: 1;      /**< is the session closed ? */
        uint8_t autoclose: 1;   /**< close the session when unreferenced */
        uint8_t notinset: 1;    /**< session removed from the set of sessions */
@@ -106,7 +108,56 @@ static inline time_t time_now()
 static void new_uuid(char uuid[SIZEUUID])
 {
        uuid_t newuuid;
+
+#if defined(USE_UUID_GENERATE)
        uuid_generate(newuuid);
+#else
+       struct timespec ts;
+       static uint16_t pid;
+       static uint16_t counter;
+       static char state[32];
+       static struct random_data rdata;
+
+       int32_t x;
+       clock_gettime(CLOCK_MONOTONIC_RAW, &ts);
+       if (pid == 0) {
+               pid = (uint16_t)getpid();
+               counter = (uint16_t)(ts.tv_nsec >> 8);
+               rdata.state = NULL;
+               initstate_r((((unsigned)pid) << 16) + ((unsigned)counter),
+                                       state, sizeof state, &rdata);
+       }
+       ts.tv_nsec ^= (long)ts.tv_sec;
+       if (++counter == 0)
+               counter = 1;
+
+       newuuid[0] = (char)(ts.tv_nsec >> 24);
+       newuuid[1] = (char)(ts.tv_nsec >> 16);
+       newuuid[2] = (char)(ts.tv_nsec >> 8);
+       newuuid[3] = (char)(ts.tv_nsec);
+
+       newuuid[4] = (char)(pid >> 8);
+       newuuid[5] = (char)(pid);
+
+       random_r(&rdata, &x);
+       newuuid[6] = (char)(((x >> 16) & 0x0f) | 0x40); /* pseudo-random version */
+       newuuid[7] = (char)(x >> 8);
+
+       random_r(&rdata, &x);
+       newuuid[8] = (char)(((x >> 16) & 0x3f) | 0x80); /* variant RFC4122 */
+       newuuid[9] = (char)(x >> 8);
+
+       random_r(&rdata, &x);
+       newuuid[10] = (char)(x >> 16);
+       newuuid[11] = (char)(x >> 8);
+
+       random_r(&rdata, &x);
+       newuuid[12] = (char)(x >> 16);
+       newuuid[13] = (char)(x >> 8);
+
+       newuuid[14] = (char)(counter >> 8);
+       newuuid[15] = (char)(counter);
+#endif
        uuid_unparse_lower(newuuid, uuid);
 }
 
@@ -189,8 +240,10 @@ static void session_close(struct afb_session *session)
                /* close it now */
                session->closed = 1;
 
+#if WITH_AFB_HOOK
                /* emit the hook */
                afb_hook_session_close(session);
+#endif
 
                /* release cookies */
                for (idx = 0 ; idx < COOKIECOUNT ; idx++) {
@@ -207,8 +260,11 @@ static void session_close(struct afb_session *session)
 /* destroy the 'session' */
 static void session_destroy (struct afb_session *session)
 {
+#if WITH_AFB_HOOK
        afb_hook_session_destroy(session);
+#endif
        pthread_mutex_destroy(&session->mutex);
+       free(session->lang);
        free(session);
 }
 
@@ -264,7 +320,9 @@ static struct afb_session *session_add(const char *uuid, int timeout, time_t now
                return NULL;
        }
 
+#if WITH_AFB_HOOK
        afb_hook_session_create(session);
+#endif
 
        return session;
 }
@@ -461,7 +519,9 @@ end:
 struct afb_session *afb_session_addref(struct afb_session *session)
 {
        if (session != NULL) {
+#if WITH_AFB_HOOK
                afb_hook_session_addref(session);
+#endif
                session_lock(session);
                session->refcount++;
                session_unlock(session);
@@ -475,7 +535,9 @@ void afb_session_unref(struct afb_session *session)
        if (session == NULL)
                return;
 
+#if WITH_AFB_HOOK
        afb_hook_session_unref(session);
+#endif
        session_lock(session);
        if (!--session->refcount) {
                if (session->autoclose)
@@ -524,7 +586,7 @@ int afb_session_check_token (struct afb_session *session, const char *token)
 {
        int r;
 
-       session_unlock(session);
+       session_lock(session);
        r = !session->closed
          && session->expiration >= NOW
          && !(session->token[0] && strcmp (token, session->token));
@@ -535,10 +597,12 @@ int afb_session_check_token (struct afb_session *session, const char *token)
 /* generate a new token and update client context */
 void afb_session_new_token (struct afb_session *session)
 {
-       session_unlock(session);
+       session_lock(session);
        new_uuid(session->token);
        session_update_expiration(session, NOW);
+#if WITH_AFB_HOOK
        afb_hook_session_renew(session);
+#endif
        session_unlock(session);
 }
 
@@ -688,3 +752,37 @@ int afb_session_set_cookie(struct afb_session *session, const void *key, void *v
        return -(value != afb_session_cookie(session, key, NULL, freecb, value, 1));
 }
 
+/**
+ * Set the language attached to the session
+ *
+ * @param session the session to set
+ * @param lang    the language specifiction to set to session
+ *
+ * @return 0 in case of success or -1 in case of error
+ */
+int afb_session_set_language(struct afb_session *session, const char *lang)
+{
+       char *oldl, *newl;
+
+       newl = strdup(lang);
+       if (newl == NULL)
+               return -1;
+
+       oldl = session->lang;
+       session->lang = newl;
+       free(oldl);
+       return 0;
+}
+
+/**
+ * Get the language attached to the session
+ *
+ * @param session the session to query
+ * @param lang    a default language specifiction
+ *
+ * @return the langauage specification to use for session
+ */
+const char *afb_session_get_language(struct afb_session *session, const char *lang)
+{
+       return session->lang ?: lang;
+}