X-Git-Url: https://gerrit.automotivelinux.org/gerrit/gitweb?a=blobdiff_plain;f=src%2Fafb-session.c;h=61aa3f61914195551b0eecdbb0221696401f2814;hb=6f13ad1989875b5a0ce50b24211fd1fba093735f;hp=3befb92ec1c69ef63216876197aeab39431c4c69;hpb=5bffa91c6c719b828a317adfb913ce57dbb2c6f0;p=src%2Fapp-framework-binder.git diff --git a/src/afb-session.c b/src/afb-session.c index 3befb92e..61aa3f61 100644 --- a/src/afb-session.c +++ b/src/afb-session.c @@ -1,5 +1,5 @@ /* - * Copyright (C) 2015, 2016, 2017 "IoT.bzh" + * Copyright (C) 2015-2019 "IoT.bzh" * Author "Fulup Ar Foll" * Author: José Bollo * @@ -21,211 +21,223 @@ #include #include #include +#include +#include #include -#include -#include #include -#include - #include "afb-session.h" +#include "afb-hook.h" #include "verbose.h" +#include "pearson.h" +#include "uuid.h" -#define HEADCOUNT 16 -#define COOKEYCOUNT 8 -#define COOKEYMASK (COOKEYCOUNT - 1) +#define HEADCOUNT 16 +#define COOKIECOUNT 8 +#define COOKIEMASK (COOKIECOUNT - 1) -#define NOW (time(NULL)) +#define _MAXEXP_ ((time_t)(~(time_t)0)) +#define _MAXEXP2_ ((time_t)((((unsigned long long)_MAXEXP_) >> 1))) +#define MAX_EXPIRATION (_MAXEXP_ >= 0 ? _MAXEXP_ : _MAXEXP2_) +#define NOW (time_now()) +/** + * structure for a cookie added to sessions + */ struct cookie { - struct cookie *next; - const void *key; - void *value; - void (*freecb)(void*); + struct cookie *next; /**< link to next cookie */ + const void *key; /**< pointer key */ + void *value; /**< value */ + void (*freecb)(void*); /**< function to call when session is closed */ }; +/** + * structure for session + */ struct afb_session { - struct afb_session *next; /* link to the next */ - unsigned refcount; - int timeout; - time_t expiration; // expiration time of the token - pthread_mutex_t mutex; - char idx; - char uuid[37]; // long term authentication of remote client - char token[37]; // short term authentication of remote client - struct cookie *cookies[COOKEYCOUNT]; + struct afb_session *next; /**< link to the next */ + unsigned refcount; /**< count of reference to the session */ + int timeout; /**< timeout of the 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 */ + uuid_stringz_t uuid; /**< long term authentication of remote client */ + uuid_stringz_t token; /**< short term authentication of remote client */ }; -// Session UUID are store in a simple array [for 10 sessions this should be enough] +/** + * structure for managing sessions + */ static struct { - pthread_mutex_t mutex; // declare a mutex to protect hash table - struct afb_session *heads[HEADCOUNT]; // sessions - int count; // current number of sessions - int max; - int timeout; - char initok[37]; -} sessions; + int count; /**< current number of sessions */ + int max; /**< maximum count of sessions */ + int timeout; /**< common initial timeout */ + struct afb_session *heads[HEADCOUNT]; /**< sessions */ + uuid_stringz_t initok; /**< common initial token */ + pthread_mutex_t mutex; /**< declare a mutex to protect hash table */ +} sessions = { + .count = 0, + .max = 10, + .timeout = 3600, + .heads = { 0 }, + .initok = { 0 }, + .mutex = PTHREAD_MUTEX_INITIALIZER +}; -/* generate a uuid */ -static void new_uuid(char uuid[37]) +/** + * Get the actual raw time + */ +static inline time_t time_now() { - uuid_t newuuid; - uuid_generate(newuuid); - uuid_unparse_lower(newuuid, uuid); + struct timespec ts; + clock_gettime(CLOCK_MONOTONIC_RAW, &ts); + return ts.tv_sec; } -static inline void lock(struct afb_session *session) +/* lock the set of sessions for exclusive access */ +static inline void sessionset_lock() { - pthread_mutex_lock(&session->mutex); + pthread_mutex_lock(&sessions.mutex); } -static inline void unlock(struct afb_session *session) +/* unlock the set of sessions of exclusive access */ +static inline void sessionset_unlock() { - pthread_mutex_unlock(&session->mutex); + pthread_mutex_unlock(&sessions.mutex); } -// Free context [XXXX Should be protected again memory abort XXXX] -static void remove_all_cookies(struct afb_session *session) +/* + * search within the set of sessions the session of 'uuid'. + * 'hashidx' is the precomputed hash for 'uuid' + * return the session or NULL + */ +static struct afb_session *sessionset_search(const char *uuid, uint8_t hashidx) { - int idx; - struct cookie *cookie, *next; - - // free cookies - for (idx = 0 ; idx < COOKEYCOUNT ; idx++) { - cookie = session->cookies[idx]; - session->cookies[idx] = NULL; - while (cookie != NULL) { - next = cookie->next; - if (cookie->freecb != NULL) - cookie->freecb(cookie->value); - free(cookie); - cookie = next; - } - } -} + struct afb_session *session; -/* tiny hash function inspired from pearson */ -static int pearson4(const char *text) -{ - static uint8_t T[16] = { - 4, 1, 6, 0, 9, 14, 11, 5, - 2, 3, 12, 15, 10, 7, 8, 13 - }; - uint8_t r, c; + session = sessions.heads[hashidx]; + while (session && strcmp(uuid, session->uuid)) + session = session->next; - for (r = 0; (c = (uint8_t)*text) ; text++) { - r = T[r ^ (15 & c)]; - r = T[r ^ (c >> 4)]; - } - return r; // % HEADCOUNT; + return session; } -// Create a new store in RAM, not that is too small it will be automatically extended -void afb_session_init (int max_session_count, int timeout, const char *initok) +/* add 'session' to the set of sessions */ +static int sessionset_add(struct afb_session *session, uint8_t hashidx) { - pthread_mutex_init(&sessions.mutex, NULL); - sessions.max = max_session_count; - sessions.timeout = timeout; - if (initok == NULL) - /* without token, a secret is made to forbid creation of sessions */ - new_uuid(sessions.initok); - else if (strlen(initok) < sizeof sessions.initok) - strcpy(sessions.initok, initok); - else { - ERROR("initial token '%s' too long (max length %d)", initok, ((int)(sizeof sessions.initok)) - 1); - exit(1); + /* check availability */ + if (sessions.max && sessions.count >= sessions.max) { + errno = EBUSY; + return -1; } -} -const char *afb_session_initial_token() -{ - return sessions.initok; + /* add the session */ + session->next = sessions.heads[hashidx]; + sessions.heads[hashidx] = session; + sessions.count++; + return 0; } -static struct afb_session *search (const char* uuid, int idx) +/* make a new uuid not used in the set of sessions */ +static uint8_t sessionset_make_uuid (uuid_stringz_t uuid) { - struct afb_session *session; - - session = sessions.heads[idx]; - while (session && strcmp(uuid, session->uuid)) - session = session->next; + uint8_t hashidx; - return session; + do { + uuid_new_stringz(uuid); + hashidx = pearson4(uuid); + } while(sessionset_search(uuid, hashidx)); + return hashidx; } -static void destroy (struct afb_session *session) +/* lock the 'session' for exclusive access */ +static inline void session_lock(struct afb_session *session) { - struct afb_session **prv; - - assert (session != NULL); + pthread_mutex_lock(&session->mutex); +} - pthread_mutex_lock(&sessions.mutex); - prv = &sessions.heads[(int)session->idx]; - while (*prv) - if (*prv != session) - prv = &((*prv)->next); - else { - *prv = session->next; - sessions.count--; - pthread_mutex_destroy(&session->mutex); - free(session); - break; - } - pthread_mutex_unlock(&sessions.mutex); +/* unlock the 'session' of exclusive access */ +static inline void session_unlock(struct afb_session *session) +{ + pthread_mutex_unlock(&session->mutex); } -// Check if context timeout or not -static int is_expired (struct afb_session *ctx, time_t now) +/* close the 'session' */ +static void session_close(struct afb_session *session) { - assert (ctx != NULL); - return ctx->expiration < now; + int idx; + struct cookie *cookie; + + /* close only one time */ + if (!session->closed) { + /* 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++) { + while ((cookie = session->cookies[idx])) { + session->cookies[idx] = cookie->next; + if (cookie->freecb != NULL) + cookie->freecb(cookie->value); + free(cookie); + } + } + } } -// Check if context is active or not -static int is_active (struct afb_session *ctx, time_t now) +/* destroy the 'session' */ +static void session_destroy (struct afb_session *session) { - assert (ctx != NULL); - return ctx->uuid[0] != 0 && ctx->expiration >= now; +#if WITH_AFB_HOOK + afb_hook_session_destroy(session); +#endif + pthread_mutex_destroy(&session->mutex); + free(session->lang); + free(session); } -// Loop on every entry and remove old context sessions.hash -static time_t cleanup () +/* update expiration of 'session' according to 'now' */ +static void session_update_expiration(struct afb_session *session, time_t now) { - struct afb_session *session, *next; - int idx; - time_t now; + time_t expiration; - // Loop on Sessions Table and remove anything that is older than timeout - now = NOW; - for (idx = 0 ; idx < HEADCOUNT; idx++) { - session = sessions.heads[idx]; - while (session) { - next = session->next; - if (is_expired(session, now)) - afb_session_close(session); - session = next; - } - } - return now; + /* compute expiration */ + expiration = now + afb_session_timeout(session); + if (expiration < 0) + expiration = MAX_EXPIRATION; + + /* record the expiration */ + session->expiration = expiration; } -static struct afb_session *add_session (const char *uuid, int timeout, time_t now, int idx) +/* + * Add a new session with the 'uuid' (of 'hashidx') + * and the 'timeout' starting from 'now'. + * Add it to the set of sessions + * Return the created session + */ +static struct afb_session *session_add(const char *uuid, int timeout, time_t now, uint8_t hashidx) { struct afb_session *session; + /* check arguments */ if (!AFB_SESSION_TIMEOUT_IS_VALID(timeout) || (uuid && strlen(uuid) >= sizeof session->uuid)) { errno = EINVAL; return NULL; } - if (sessions.count >= sessions.max) { - errno = EBUSY; - return NULL; - } - /* allocates a new one */ session = calloc(1, sizeof *session); if (session == NULL) { @@ -233,55 +245,129 @@ static struct afb_session *add_session (const char *uuid, int timeout, time_t no return NULL; } + /* initialize */ pthread_mutex_init(&session->mutex, NULL); session->refcount = 1; strcpy(session->uuid, uuid); strcpy(session->token, sessions.initok); - - /* init timeout */ - if (timeout == AFB_SESSION_TIMEOUT_DEFAULT) - timeout = sessions.timeout; session->timeout = timeout; - session->expiration = now + timeout; - if (timeout == AFB_SESSION_TIMEOUT_INFINITE || session->expiration < 0) { - session->expiration = (time_t)(~(time_t)0); - if (session->expiration < 0) - session->expiration = (time_t)(((unsigned long long)session->expiration) >> 1); + session_update_expiration(session, now); + + /* add */ + if (sessionset_add(session, hashidx)) { + free(session); + return NULL; } - session->idx = (char)idx; - session->next = sessions.heads[idx]; - sessions.heads[idx] = session; - sessions.count++; +#if WITH_AFB_HOOK + afb_hook_session_create(session); +#endif return session; } -static struct afb_session *new_session (int timeout, time_t now) +/* Remove expired sessions and return current time (now) */ +static time_t sessionset_cleanup (int force) { + struct afb_session *session, **prv; int idx; - char uuid[37]; + time_t now; - do { - new_uuid(uuid); - idx = pearson4(uuid); - } while(search(uuid, idx)); - return add_session(uuid, timeout, now, idx); + /* Loop on Sessions Table and remove anything that is older than timeout */ + now = NOW; + for (idx = 0 ; idx < HEADCOUNT; idx++) { + prv = &sessions.heads[idx]; + while ((session = *prv)) { + session_lock(session); + if (force || session->expiration < now) + session_close(session); + if (!session->closed) + prv = &session->next; + else { + *prv = session->next; + sessions.count--; + session->notinset = 1; + if ( !session->refcount) { + session_destroy(session); + continue; + } + } + session_unlock(session); + } + } + return now; } -/* Creates a new session with 'timeout' */ -struct afb_session *afb_session_create (int timeout) +/** + * Initialize the session manager with a 'max_session_count', + * an initial common 'timeout' and an initial common token 'initok'. + * + * @param max_session_count maximum allowed session count in the same time + * @param timeout the initial default timeout of sessions + * @param initok the initial default token of sessions + * + */ +int afb_session_init (int max_session_count, int timeout, const char *initok) +{ + /* check parameters */ + if (initok && strlen(initok) >= sizeof sessions.initok) { + ERROR("initial token '%s' too long (max length %d)", + initok, ((int)(sizeof sessions.initok)) - 1); + errno = EINVAL; + return -1; + } + + /* init the sessionset (after cleanup) */ + sessionset_lock(); + sessionset_cleanup(1); + sessions.max = max_session_count; + sessions.timeout = timeout; + if (initok == NULL) + uuid_new_stringz(sessions.initok); + else + strcpy(sessions.initok, initok); + sessionset_unlock(); + return 0; +} + +/** + * Iterate the sessions and call 'callback' with + * the 'closure' for each session. + */ +void afb_session_foreach(void (*callback)(void *closure, struct afb_session *session), void *closure) { - time_t now; struct afb_session *session; + int idx; - /* cleaning */ - pthread_mutex_lock(&sessions.mutex); - now = cleanup(); - session = new_session(timeout, now); - pthread_mutex_unlock(&sessions.mutex); + /* Loop on Sessions Table and remove anything that is older than timeout */ + sessionset_lock(); + for (idx = 0 ; idx < HEADCOUNT; idx++) { + session = sessions.heads[idx]; + while (session) { + if (!session->closed) + callback(closure, session); + session = session->next; + } + } + sessionset_unlock(); +} - return session; +/** + * Cleanup the sessionset of its closed or expired sessions + */ +void afb_session_purge() +{ + sessionset_lock(); + sessionset_cleanup(0); + sessionset_unlock(); +} + +/** + * @return the initial token set at initialization + */ +const char *afb_session_initial_token() +{ + return sessions.initok; } /* Searchs the session of 'uuid' */ @@ -289,121 +375,185 @@ struct afb_session *afb_session_search (const char *uuid) { struct afb_session *session; - /* cleaning */ - pthread_mutex_lock(&sessions.mutex); - cleanup(); - session = search(uuid, pearson4(uuid)); - pthread_mutex_unlock(&sessions.mutex); + sessionset_lock(); + sessionset_cleanup(0); + session = sessionset_search(uuid, pearson4(uuid)); + session = afb_session_addref(session); + sessionset_unlock(); return session; } +/** + * Creates a new session with 'timeout' + */ +struct afb_session *afb_session_create (int timeout) +{ + return afb_session_get(NULL, timeout, NULL); +} + +/** + * Returns the timeout of 'session' in seconds + */ +int afb_session_timeout(struct afb_session *session) +{ + int timeout; + + /* compute timeout */ + timeout = session->timeout; + if (timeout == AFB_SESSION_TIMEOUT_DEFAULT) + timeout = sessions.timeout; + if (timeout < 0) + timeout = INT_MAX; + return timeout; +} + +/** + * Returns the second remaining before expiration of 'session' + */ +int afb_session_what_remains(struct afb_session *session) +{ + return (int)(session->expiration - NOW); +} + /* This function will return exiting session or newly created session */ struct afb_session *afb_session_get (const char *uuid, int timeout, int *created) { - int idx; + uuid_stringz_t _uuid_; + uint8_t hashidx; struct afb_session *session; time_t now; + int c; /* cleaning */ - pthread_mutex_lock(&sessions.mutex); - now = cleanup(); + sessionset_lock(); + now = sessionset_cleanup(0); /* search for an existing one not too old */ - if (!uuid) - session = new_session(timeout, now); - else { - idx = pearson4(uuid); - session = search(uuid, idx); - if (session != NULL) { - __atomic_add_fetch(&session->refcount, 1, __ATOMIC_RELAXED); - pthread_mutex_unlock(&sessions.mutex); - if (created) - *created = 0; - return session; + if (!uuid) { + hashidx = sessionset_make_uuid(_uuid_); + uuid = _uuid_; + } else { + hashidx = pearson4(uuid); + session = sessionset_search(uuid, hashidx); + if (session) { + /* session found */ + afb_session_addref(session); + c = 0; + goto end; } - session = add_session (uuid, timeout, now, idx); } - pthread_mutex_unlock(&sessions.mutex); - + /* create the session */ + session = session_add(uuid, timeout, now, hashidx); + c = 1; +end: + sessionset_unlock(); if (created) - *created = !!session; + *created = c; return session; } -/* increase the use count on the session */ +/* increase the use count on 'session' (can be NULL) */ struct afb_session *afb_session_addref(struct afb_session *session) { - if (session != NULL) - __atomic_add_fetch(&session->refcount, 1, __ATOMIC_RELAXED); + if (session != NULL) { +#if WITH_AFB_HOOK + afb_hook_session_addref(session); +#endif + session_lock(session); + session->refcount++; + session_unlock(session); + } return session; } -/* decrease the use count of the session */ +/* decrease the use count of 'session' (can be NULL) */ void afb_session_unref(struct afb_session *session) { - if (session != NULL) { - assert(session->refcount != 0); - if (!__atomic_sub_fetch(&session->refcount, 1, __ATOMIC_RELAXED)) { - if (session->uuid[0] == 0) - destroy (session); + if (session == NULL) + return; + +#if WITH_AFB_HOOK + afb_hook_session_unref(session); +#endif + session_lock(session); + if (!--session->refcount) { + if (session->autoclose) + session_close(session); + if (session->notinset) { + session_destroy(session); + return; } } + session_unlock(session); } -// Free Client Session Context +/* close 'session' */ void afb_session_close (struct afb_session *session) { - assert(session != NULL); - if (session->uuid[0] != 0) { - session->uuid[0] = 0; - remove_all_cookies(session); - if (session->refcount == 0) - destroy (session); - } + session_lock(session); + session_close(session); + session_unlock(session); } -// Sample Generic Ping Debug API -int afb_session_check_token (struct afb_session *session, const char *token) +/** + * Set the 'autoclose' flag of the 'session' + * + * A session whose autoclose flag is true will close as + * soon as it is no more referenced. + * + * @param session the session to set + * @param autoclose the value to set + */ +void afb_session_set_autoclose(struct afb_session *session, int autoclose) { - assert(session != NULL); - assert(token != NULL); - - // compare current token with previous one - if (!is_active (session, NOW)) - return 0; + session->autoclose = !!autoclose; +} - if (session->token[0] && strcmp (token, session->token) != 0) - return 0; +/* is 'session' closed? */ +int afb_session_is_closed (struct afb_session *session) +{ + return session->closed; +} - return 1; +/* + * check whether the token of 'session' is 'token' + * return 1 if true or 0 otherwise + */ +int afb_session_check_token (struct afb_session *session, const char *token) +{ + int r; + + session_lock(session); + r = !session->closed + && session->expiration >= NOW + && !(session->token[0] && strcmp (token, session->token)); + session_unlock(session); + return r; } -// generate a new token and update client context +/* generate a new token and update client context */ void afb_session_new_token (struct afb_session *session) { - assert(session != NULL); - - // Old token was valid let's regenerate a new one - new_uuid(session->token); - - // keep track of time for session timeout and further clean up - if (session->timeout != 0) - session->expiration = NOW + session->timeout; + session_lock(session); + uuid_new_stringz(session->token); + session_update_expiration(session, NOW); +#if WITH_AFB_HOOK + afb_hook_session_renew(session); +#endif + session_unlock(session); } /* Returns the uuid of 'session' */ const char *afb_session_uuid (struct afb_session *session) { - assert(session != NULL); return session->uuid; } /* Returns the token of 'session' */ const char *afb_session_token (struct afb_session *session) { - assert(session != NULL); return session->token; } @@ -416,7 +566,7 @@ static int cookeyidx(const void *key) { intptr_t x = (intptr_t)key; unsigned r = (unsigned)((x >> 5) ^ (x >> 15)); - return r & COOKEYMASK; + return r & COOKIEMASK; } /** @@ -424,12 +574,12 @@ static int cookeyidx(const void *key) * * The behaviour of this function depends on its parameters: * - * @param session the session - * @param key the key of the cookie - * @param makecb the creation function - * @param freecb the release function - * @param closure an argument - * @param replace a boolean enforcing replecement of the previous value + * @param session the session + * @param key the key of the cookie + * @param makecb the creation function or NULL + * @param freecb the release function or NULL + * @param closure an argument for makecb or the value if makecb==NULL + * @param replace a boolean enforcing replacement of the previous value * * @return the value of the cookie * @@ -453,7 +603,7 @@ void *afb_session_cookie(struct afb_session *session, const void *key, void *(*m idx = cookeyidx(key); /* lock session and search for the cookie of 'key' */ - lock(session); + session_lock(session); prv = &session->cookies[idx]; for (;;) { cookie = *prv; @@ -491,14 +641,14 @@ void *afb_session_cookie(struct afb_session *session, const void *key, void *(*m if (cookie->value != value && cookie->freecb) cookie->freecb(cookie->value); - /* store the value and its releaser */ - cookie->value = value; - cookie->freecb = freecb; - - /* but if both are NULL drop the cookie */ + /* if both value and freecb are NULL drop the cookie */ if (!value && !freecb) { *prv = cookie->next; free(cookie); + } else { + /* store the value and its releaser */ + cookie->value = value; + cookie->freecb = freecb; } } break; @@ -508,17 +658,70 @@ void *afb_session_cookie(struct afb_session *session, const void *key, void *(*m } /* unlock the session and return the value */ - unlock(session); + session_unlock(session); return value; } +/** + * Get the cookie of 'key' in the 'session'. + * + * @param session the session to search in + * @param key the key of the data to retrieve + * + * @return the data staored for the key or NULL if the key isn't found + */ void *afb_session_get_cookie(struct afb_session *session, const void *key) { return afb_session_cookie(session, key, NULL, NULL, NULL, 0); } +/** + * Set the cookie of 'key' in the 'session' to the 'value' that can be + * cleaned using 'freecb' (if not null). + * + * @param session the session to set + * @param key the key of the data to store + * @param value the value to store at key + * @param freecb a function to use when the cookie value is to remove (or null) + * + * @return 0 in case of success or -1 in case of error + */ int afb_session_set_cookie(struct afb_session *session, const void *key, void *value, void (*freecb)(void*)) { 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; +}