X-Git-Url: https://gerrit.automotivelinux.org/gerrit/gitweb?a=blobdiff_plain;f=src%2Fsession.c;h=9944e764935d723b0bdb49a65a031493c886c4db;hb=c94e1ef88adc641ebbab57377b97f8a7b6a70d1b;hp=d08ca759b3b130833e444da26fbdb7f7b57ee768;hpb=1e8edd92a04281bddc86fff3db5b407943504aed;p=src%2Fapp-framework-binder.git

diff --git a/src/session.c b/src/session.c
index d08ca759..9944e764 100644
--- a/src/session.c
+++ b/src/session.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2015 "IoT.bzh"
+ * Copyright (C) 2015, 2016 "IoT.bzh"
  * Author "Fulup Ar Foll"
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
@@ -38,22 +38,25 @@ struct client_value
 	void (*free_value)(void*);
 };
 
-struct afb_event_listener_list
+struct cookie
 {
-	struct afb_event_listener_list *next;
-	struct afb_event_listener listener;
-	int refcount;
+	struct cookie *next;
+	const void *key;
+	void *value;
+	void (*free_value)(void*);
 };
 
 struct AFB_clientCtx
 {
 	unsigned refcount;
+	unsigned loa;
+	int timeout;
 	time_t expiration;    // expiration time of the token
 	time_t access;
 	char uuid[37];        // long term authentication of remote client
 	char token[37];       // short term authentication of remote client
 	struct client_value *values;
-	struct afb_event_listener_list *listeners;
+	struct cookie *cookies;
 };
 
 // Session UUID are store in a simple array [for 10 sessions this should be enough]
@@ -64,13 +67,22 @@ static struct {
   int max;
   int timeout;
   int apicount;
-  const char *initok;
+  char initok[37];
 } sessions;
 
+/* generate a uuid */
+static void new_uuid(char uuid[37])
+{
+	uuid_t newuuid;
+	uuid_generate(newuuid);
+	uuid_unparse_lower(newuuid, uuid);
+}
+
 // Free context [XXXX Should be protected again memory abort XXXX]
 static void ctxUuidFreeCB (struct AFB_clientCtx *client)
 {
 	int idx;
+	struct cookie *cookie;
 
 	// If application add a handle let's free it now
 	assert (client->values != NULL);
@@ -78,6 +90,16 @@ static void ctxUuidFreeCB (struct AFB_clientCtx *client)
 	// Free client handle with a standard Free function, with app callback or ignore it
 	for (idx=0; idx < sessions.apicount; idx ++)
 		ctxClientValueSet(client, idx, NULL, NULL);
+
+	// free cookies
+	cookie = client->cookies;
+	while (cookie != NULL) {
+		client->cookies = cookie->next;
+		if (cookie->value != NULL && cookie->free_value != NULL)
+			cookie->free_value(cookie->value);
+		free(cookie);
+		cookie = client->cookies;
+	}
 }
 
 // Create a new store in RAM, not that is too small it will be automatically extended
@@ -88,11 +110,15 @@ void ctxStoreInit (int max_session_count, int timeout, const char *initok, int c
 	sessions.max = max_session_count;
 	sessions.timeout = timeout;
 	sessions.apicount = context_count;
-	if (strlen(initok) >= sizeof(sessions.store[0]->token)) {
+	if (initok == NULL)
+		/* without token, a secret is made to forbid creation of sessions */
+		new_uuid(sessions.initok);
+	else if (strlen(initok) < sizeof(sessions.store[0]->token))
+		strcpy(sessions.initok, initok);
+	else {
 		ERROR("initial token '%s' too long (max length 36)", initok);
 		exit(1);
 	}
-	sessions.initok = initok;
 }
 
 static struct AFB_clientCtx *ctxStoreSearch (const char* uuid)
@@ -165,7 +191,15 @@ added:
 // Check if context timeout or not
 static int ctxStoreTooOld (struct AFB_clientCtx *ctx, time_t now)
 {
-    return ctx->expiration <= now;
+    assert (ctx != NULL);
+    return ctx->expiration < now;
+}
+
+// Check if context is active or not
+static int ctxIsActive (struct AFB_clientCtx *ctx, time_t now)
+{
+    assert (ctx != NULL);
+    return ctx->uuid[0] != 0 && ctx->expiration >= now;
 }
 
 // Loop on every entry and remove old context sessions.hash
@@ -183,31 +217,11 @@ static void ctxStoreCleanUp (time_t now)
 	}
 }
 
-// This function will return exiting client context or newly created client context
-struct AFB_clientCtx *ctxClientGetSession (const char *uuid, int *created)
+static struct AFB_clientCtx *new_context (const char *uuid, int timeout, time_t now)
 {
-	uuid_t newuuid;
 	struct AFB_clientCtx *clientCtx;
-	time_t now;
 
-	/* cleaning */
-	now = NOW;
-	ctxStoreCleanUp (now);
-
-	/* search for an existing one not too old */
-	if (uuid != NULL) {
-		if (strlen(uuid) >= sizeof clientCtx->uuid) {
-			errno = EINVAL;
-			goto error;
-		}
-		clientCtx = ctxStoreSearch(uuid);
-		if (clientCtx != NULL) {
-			*created = 0;
-			goto found;
-		}
-	}
-
-	/* returns a new one */
+	/* allocates a new one */
         clientCtx = calloc(1, sizeof(struct AFB_clientCtx) + ((unsigned)sessions.apicount * sizeof(*clientCtx->values)));
 	if (clientCtx == NULL) {
 		errno = ENOMEM;
@@ -217,24 +231,32 @@ struct AFB_clientCtx *ctxClientGetSession (const char *uuid, int *created)
 
 	/* generate the uuid */
 	if (uuid == NULL) {
-		uuid_generate(newuuid);
-		uuid_unparse_lower(newuuid, clientCtx->uuid);
+		new_uuid(clientCtx->uuid);
 	} else {
+		if (strlen(uuid) >= sizeof clientCtx->uuid) {
+			errno = EINVAL;
+			goto error2;
+		}
 		strcpy(clientCtx->uuid, uuid);
 	}
 
 	/* init the token */
 	strcpy(clientCtx->token, sessions.initok);
-	clientCtx->expiration = now + sessions.timeout;
+	clientCtx->timeout = timeout;
+	if (timeout != 0)
+		clientCtx->expiration = now + timeout;
+	else {
+		clientCtx->expiration = (time_t)(~(time_t)0);
+		if (clientCtx->expiration < 0)
+			clientCtx->expiration = (time_t)(((unsigned long long)clientCtx->expiration) >> 1);
+	}
 	if (!ctxStoreAdd (clientCtx)) {
 		errno = ENOMEM;
 		goto error2;
 	}
-	*created = 1;
 
-found:
 	clientCtx->access = now;
-	clientCtx->refcount++;
+	clientCtx->refcount = 1;
 	return clientCtx;
 
 error2:
@@ -243,6 +265,48 @@ error:
 	return NULL;
 }
 
+struct AFB_clientCtx *ctxClientCreate (const char *uuid, int timeout)
+{
+	time_t now;
+
+	/* cleaning */
+	now = NOW;
+	ctxStoreCleanUp (now);
+
+	/* search for an existing one not too old */
+	if (uuid != NULL && ctxStoreSearch(uuid) != NULL) {
+		errno = EEXIST;
+		return NULL;
+	}
+
+	return new_context(uuid, timeout, now);
+}
+
+// This function will return exiting client context or newly created client context
+struct AFB_clientCtx *ctxClientGetSession (const char *uuid, int *created)
+{
+	struct AFB_clientCtx *clientCtx;
+	time_t now;
+
+	/* cleaning */
+	now = NOW;
+	ctxStoreCleanUp (now);
+
+	/* search for an existing one not too old */
+	if (uuid != NULL) {
+		clientCtx = ctxStoreSearch(uuid);
+		if (clientCtx != NULL) {
+			*created = 0;
+			clientCtx->access = now;
+			clientCtx->refcount++;
+			return clientCtx;
+		}
+	}
+
+	*created = 1;
+	return new_context(uuid, sessions.timeout, now);
+}
+
 struct AFB_clientCtx *ctxClientAddRef(struct AFB_clientCtx *clientCtx)
 {
 	if (clientCtx != NULL)
@@ -257,6 +321,7 @@ void ctxClientUnref(struct AFB_clientCtx *clientCtx)
 		--clientCtx->refcount;
        		if (clientCtx->refcount == 0 && clientCtx->uuid[0] == 0) {
 			ctxStoreDel (clientCtx);
+			free(clientCtx);
 		}
 	}
 }
@@ -265,8 +330,14 @@ void ctxClientUnref(struct AFB_clientCtx *clientCtx)
 void ctxClientClose (struct AFB_clientCtx *clientCtx)
 {
 	assert(clientCtx != NULL);
-        ctxUuidFreeCB (clientCtx);
-	clientCtx->uuid[0] = 0;
+	if (clientCtx->uuid[0] != 0) {
+		clientCtx->uuid[0] = 0;
+	        ctxUuidFreeCB (clientCtx);
+       		if (clientCtx->refcount == 0) {
+			ctxStoreDel (clientCtx);
+			free(clientCtx);
+		}
+	}
 }
 
 // Sample Generic Ping Debug API
@@ -276,7 +347,7 @@ int ctxTokenCheck (struct AFB_clientCtx *clientCtx, const char *token)
 	assert(token != NULL);
 
 	// compare current token with previous one
-	if (ctxStoreTooOld (clientCtx, NOW))
+	if (!ctxIsActive (clientCtx, NOW))
 		return 0;
 
 	if (clientCtx->token[0] && strcmp (token, clientCtx->token) != 0)
@@ -288,113 +359,38 @@ int ctxTokenCheck (struct AFB_clientCtx *clientCtx, const char *token)
 // generate a new token and update client context
 void ctxTokenNew (struct AFB_clientCtx *clientCtx)
 {
-	uuid_t newuuid;
-
 	assert(clientCtx != NULL);
 
 	// Old token was valid let's regenerate a new one
-	uuid_generate(newuuid);         // create a new UUID
-	uuid_unparse_lower(newuuid, clientCtx->token);
+	new_uuid(clientCtx->token);
 
 	// keep track of time for session timeout and further clean up
-	clientCtx->expiration = NOW + sessions.timeout;
+	if (clientCtx->timeout != 0)
+		clientCtx->expiration = NOW + clientCtx->timeout;
 }
 
-int ctxClientEventListenerAdd(struct AFB_clientCtx *clientCtx, struct afb_event_listener listener)
-{
-	struct afb_event_listener_list *iter, **prv;
-
-	prv = &clientCtx->listeners;
-	for (;;) {
-		iter = *prv;
-		if (iter == NULL) {
-			iter = calloc(1, sizeof *iter);
-			if (iter == NULL) {
-				errno = ENOMEM;
-				return -1;
-			}
-			iter->listener = listener;
-			iter->refcount = 1;
-			*prv = iter;
-			return 0;
-		}
-		if (iter->listener.itf == listener.itf && iter->listener.closure == listener.closure) {
-			iter->refcount++;
-			return 0;
-		}
-		prv = &iter->next;
-	}
-}
-
-void ctxClientEventListenerRemove(struct AFB_clientCtx *clientCtx, struct afb_event_listener listener)
-{
-	struct afb_event_listener_list *iter, **prv;
-
-	prv = &clientCtx->listeners;
-	for (;;) {
-		iter = *prv;
-		if (iter == NULL)
-			return;
-		if (iter->listener.itf == listener.itf && iter->listener.closure == listener.closure) {
-			if (!--iter->refcount) {
-				*prv = iter->next;
-				free(iter);
-			}
-			return;
-		}
-		prv = &iter->next;
-	}
-}
-
-static int send(struct AFB_clientCtx *clientCtx, const char *event, struct json_object *object)
+const char *ctxClientGetUuid (struct AFB_clientCtx *clientCtx)
 {
-	struct afb_event_listener_list *iter;
-	int result;
-
-	result = 0;
-	iter = clientCtx->listeners;
-	while (iter != NULL) {
-		iter->listener.itf->send(iter->listener.closure, event, json_object_get(object));
-		result++;
-		iter = iter->next;
-	}
-
-	return result;
+	assert(clientCtx != NULL);
+	return clientCtx->uuid;
 }
 
-int ctxClientEventSend(struct AFB_clientCtx *clientCtx, const char *event, struct json_object *object)
+const char *ctxClientGetToken (struct AFB_clientCtx *clientCtx)
 {
-	long idx;
-	time_t now;
-	int result;
-
-	if (clientCtx != NULL)
-		result = send(clientCtx, event, object);
-	else {
-		result = 0;
-		now = NOW;
-		for (idx=0; idx < sessions.max; idx++) {
-			clientCtx = sessions.store[idx];
-			if (clientCtx != NULL && !ctxStoreTooOld(clientCtx, now)) {
-				clientCtx = ctxClientAddRef(clientCtx);
-				result += send(clientCtx, event, object);
-				ctxClientUnref(clientCtx);
-			}
-		}
-	}
-	return result;
+	assert(clientCtx != NULL);
+	return clientCtx->token;
 }
 
-const char *ctxClientGetUuid (struct AFB_clientCtx *clientCtx)
+unsigned ctxClientGetLOA (struct AFB_clientCtx *clientCtx)
 {
 	assert(clientCtx != NULL);
-	return clientCtx->uuid;
+	return clientCtx->loa;
 }
 
-const char *ctxClientGetToken (struct AFB_clientCtx *clientCtx)
+void ctxClientSetLOA (struct AFB_clientCtx *clientCtx, unsigned loa)
 {
 	assert(clientCtx != NULL);
-	return clientCtx->token;
+	clientCtx->loa = loa;
 }
 
 void *ctxClientValueGet(struct AFB_clientCtx *clientCtx, int index)
@@ -413,6 +409,52 @@ void ctxClientValueSet(struct AFB_clientCtx *clientCtx, int index, void *value,
 	assert(index < sessions.apicount);
 	prev = clientCtx->values[index];
 	clientCtx->values[index] = (struct client_value){.value = value, .free_value = free_value};
-	if (prev.value !=  NULL && prev.free_value != NULL)
+	if (prev.value != NULL && prev.value != value && prev.free_value != NULL)
 		prev.free_value(prev.value);
 }
+
+void *ctxClientCookieGet(struct AFB_clientCtx *clientCtx, const void *key)
+{
+	struct cookie *cookie;
+
+	cookie = clientCtx->cookies;
+	while(cookie != NULL) {
+		if (cookie->key == key)
+			return cookie->value;
+		cookie = cookie->next;
+	}
+	return NULL;
+}
+
+int ctxClientCookieSet(struct AFB_clientCtx *clientCtx, const void *key, void *value, void (*free_value)(void*))
+{
+	struct cookie *cookie;
+
+	/* search for a replacement */
+	cookie = clientCtx->cookies;
+	while(cookie != NULL) {
+		if (cookie->key == key) {
+			if (cookie->value != NULL && cookie->value != value && cookie->free_value != NULL)
+				cookie->free_value(cookie->value);
+			cookie->value = value;
+			cookie->free_value = free_value;
+			return 0;
+		}
+		cookie = cookie->next;
+	}
+
+	/* allocates */
+	cookie = malloc(sizeof *cookie);
+	if (cookie == NULL) {
+		errno = ENOMEM;
+		return -1;
+	}
+
+	cookie->key = key;
+	cookie->value = value;
+	cookie->free_value = free_value;
+	cookie->next = clientCtx->cookies;
+	clientCtx->cookies = cookie;
+	return 0;
+}
+