afb-hook: Add hooks for session

Change-Id: I4fbd6bf57958abb366a742b12afb92bfc03ad299

2 files changed, 228 insertions, 4 deletions

diff --git a/src/afb-hook.c b/src/afb-hook.c
index b5dbe154..33cb91bf 100644
--- a/src/afb-hook.c
+++ b/src/afb-hook.c
@@ -47,10 +47,10 @@
 struct afb_hook_xreq {
 	struct afb_hook_xreq *next; /**< next hook */
 	unsigned refcount; /**< reference count */
+	unsigned flags; /**< hook flags */
 	char *api; /**< api hooked or NULL for any */
 	char *verb; /**< verb hooked or NULL for any */
 	struct afb_session *session; /**< session hooked or NULL if any */
-	unsigned flags; /**< hook flags */
 	struct afb_hook_xreq_itf *itf; /**< interface of hook */
 	void *closure; /**< closure for callbacks */
 };
@@ -61,8 +61,8 @@ struct afb_hook_xreq {
 struct afb_hook_ditf {
 	struct afb_hook_ditf *next; /**< next hook */
 	unsigned refcount; /**< reference count */
-	char *api; /**< api hooked or NULL for any */
 	unsigned flags; /**< hook flags */
+	char *api; /**< api hooked or NULL for any */
 	struct afb_hook_ditf_itf *itf; /**< interface of hook */
 	void *closure; /**< closure for callbacks */
 };
@@ -73,8 +73,8 @@ struct afb_hook_ditf {
 struct afb_hook_svc {
 	struct afb_hook_svc *next; /**< next hook */
 	unsigned refcount; /**< reference count */
-	char *api; /**< api hooked or NULL for any */
 	unsigned flags; /**< hook flags */
+	char *api; /**< api hooked or NULL for any */
 	struct afb_hook_svc_itf *itf; /**< interface of hook */
 	void *closure; /**< closure for callbacks */
 };
@@ -85,13 +85,25 @@ struct afb_hook_svc {
 struct afb_hook_evt {
 	struct afb_hook_evt *next; /**< next hook */
 	unsigned refcount; /**< reference count */
-	char *pattern; /**< event pattern name hooked or NULL for any */
 	unsigned flags; /**< hook flags */
+	char *pattern; /**< event pattern name hooked or NULL for any */
 	struct afb_hook_evt_itf *itf; /**< interface of hook */
 	void *closure; /**< closure for callbacks */
 };
 
 /**
+ * Definition of a hook for session
+ */
+struct afb_hook_session {
+	struct afb_hook_session *next; /**< next hook */
+	unsigned refcount; /**< reference count */
+	unsigned flags; /**< hook flags */
+	char *pattern; /**< event pattern name hooked or NULL for any */
+	struct afb_hook_session_itf *itf; /**< interface of hook */
+	void *closure; /**< closure for callbacks */
+};
+
+/**
  * Definition of a hook for global
  */
 struct afb_hook_global {
@@ -117,6 +129,9 @@ static struct afb_hook_svc *list_of_svc_hooks = NULL;
 /* list of hooks for evt */
 static struct afb_hook_evt *list_of_evt_hooks = NULL;
 
+/* list of hooks for session */
+static struct afb_hook_session *list_of_session_hooks = NULL;
+
 /* list of hooks for global */
 static struct afb_hook_global *list_of_global_hooks = NULL;
 
@@ -1430,6 +1445,178 @@ void afb_hook_unref_evt(struct afb_hook_evt *hook)
 }
 
 /******************************************************************************
+ * section: default callbacks for sessions (session)
+ *****************************************************************************/
+
+static void _hook_session_(struct afb_session *session, const char *format, ...)
+{
+	va_list ap;
+	va_start(ap, format);
+	_hook_("session-%s", format, ap, afb_session_uuid(session));
+	va_end(ap);
+}
+
+static void hook_session_create_default_cb(void *closure, const struct afb_hookid *hookid, struct afb_session *session)
+{
+	_hook_session_(session, "create -> token=%s", afb_session_token(session));
+}
+
+static void hook_session_close_default_cb(void *closure, const struct afb_hookid *hookid, struct afb_session *session)
+{
+	_hook_session_(session, "close");
+}
+
+static void hook_session_destroy_default_cb(void *closure, const struct afb_hookid *hookid, struct afb_session *session)
+{
+	_hook_session_(session, "destroy");
+}
+
+static void hook_session_renew_default_cb(void *closure, const struct afb_hookid *hookid, struct afb_session *session)
+{
+	_hook_session_(session, "renew -> token=%s", afb_session_token(session));
+}
+
+static void hook_session_addref_default_cb(void *closure, const struct afb_hookid *hookid, struct afb_session *session)
+{
+	_hook_session_(session, "addref");
+}
+
+static void hook_session_unref_default_cb(void *closure, const struct afb_hookid *hookid, struct afb_session *session)
+{
+	_hook_session_(session, "unref");
+}
+
+static struct afb_hook_session_itf hook_session_default_itf = {
+	.hook_session_create = hook_session_create_default_cb,
+	.hook_session_close = hook_session_close_default_cb,
+	.hook_session_destroy = hook_session_destroy_default_cb,
+	.hook_session_renew = hook_session_renew_default_cb,
+	.hook_session_addref = hook_session_addref_default_cb,
+	.hook_session_unref = hook_session_unref_default_cb
+};
+
+/******************************************************************************
+ * section: hooks for tracing sessions (session)
+ *****************************************************************************/
+
+#define _HOOK_SESSION_(what,...)   \
+	struct afb_hook_session *hook; \
+	struct afb_hookid hookid; \
+	const char *sessid = 0; \
+	pthread_rwlock_rdlock(&rwlock); \
+	init_hookid(&hookid); \
+	hook = list_of_session_hooks; \
+	while (hook) { \
+		if (hook->itf->hook_session_##what \
+		 && (hook->flags & afb_hook_flag_session_##what) != 0 \
+		 && (!hook->pattern || !fnmatch(hook->pattern, (sessid?:(sessid=afb_session_uuid(session))), FNM_CASEFOLD))) { \
+			hook->itf->hook_session_##what(hook->closure, &hookid, __VA_ARGS__); \
+		} \
+		hook = hook->next; \
+	} \
+	pthread_rwlock_unlock(&rwlock);
+
+void afb_hook_session_create(struct afb_session *session)
+{
+	_HOOK_SESSION_(create, session);
+}
+
+void afb_hook_session_close(struct afb_session *session)
+{
+	_HOOK_SESSION_(close, session);
+}
+
+void afb_hook_session_destroy(struct afb_session *session)
+{
+	_HOOK_SESSION_(destroy, session);
+}
+
+void afb_hook_session_renew(struct afb_session *session)
+{
+	_HOOK_SESSION_(renew, session);
+}
+
+void afb_hook_session_addref(struct afb_session *session)
+{
+	_HOOK_SESSION_(addref, session);
+}
+
+void afb_hook_session_unref(struct afb_session *session)
+{
+	_HOOK_SESSION_(unref, session);
+}
+
+
+/******************************************************************************
+ * section: hooking sessions (session)
+ *****************************************************************************/
+
+struct afb_hook_session *afb_hook_create_session(const char *pattern, int flags, struct afb_hook_session_itf *itf, void *closure)
+{
+	struct afb_hook_session *hook;
+
+	/* alloc the result */
+	hook = calloc(1, sizeof *hook);
+	if (hook == NULL)
+		return NULL;
+
+	/* get a copy of the names */
+	hook->pattern = pattern ? strdup(pattern) : NULL;
+	if (pattern && !hook->pattern) {
+		free(hook);
+		return NULL;
+	}
+
+	/* initialise the rest */
+	hook->refcount = 1;
+	hook->flags = flags;
+	hook->itf = itf ? itf : &hook_session_default_itf;
+	hook->closure = closure;
+
+	/* record the hook */
+	pthread_rwlock_wrlock(&rwlock);
+	hook->next = list_of_session_hooks;
+	list_of_session_hooks = hook;
+	pthread_rwlock_unlock(&rwlock);
+
+	/* returns it */
+	return hook;
+}
+
+struct afb_hook_session *afb_hook_addref_session(struct afb_hook_session *hook)
+{
+	pthread_rwlock_wrlock(&rwlock);
+	hook->refcount++;
+	pthread_rwlock_unlock(&rwlock);
+	return hook;
+}
+
+void afb_hook_unref_session(struct afb_hook_session *hook)
+{
+	struct afb_hook_session **prv;
+
+	if (hook) {
+		pthread_rwlock_wrlock(&rwlock);
+		if (--hook->refcount)
+			hook = NULL;
+		else {
+			/* unlink */
+			prv = &list_of_session_hooks;
+			while (*prv && *prv != hook)
+				prv = &(*prv)->next;
+			if(*prv)
+				*prv = hook->next;
+		}
+		pthread_rwlock_unlock(&rwlock);
+		if (hook) {
+			/* free */
+			free(hook->pattern);
+			free(hook);
+		}
+	}
+}
+
+/******************************************************************************
  * section: default callbacks for globals (global)
  *****************************************************************************/
 
diff --git a/src/afb-hook.h b/src/afb-hook.h
index c9ecb331..3ab98468 100644
--- a/src/afb-hook.h
+++ b/src/afb-hook.h
@@ -35,6 +35,7 @@ struct afb_hook_xreq;
 struct afb_hook_ditf;
 struct afb_hook_svc;
 struct afb_hook_evt;
+struct afb_hook_session;
 struct afb_hook_global;
 
 /*********************************************************
@@ -332,6 +333,42 @@ extern struct afb_hook_evt *afb_hook_addref_evt(struct afb_hook_evt *hook);
 extern void afb_hook_unref_evt(struct afb_hook_evt *hook);
 
 /*********************************************************
+* section hooking session (session interface)
+*********************************************************/
+
+#define afb_hook_flag_session_create			0x000001
+#define afb_hook_flag_session_close			0x000002
+#define afb_hook_flag_session_destroy			0x000004
+#define afb_hook_flag_session_renew			0x000008
+#define afb_hook_flag_session_addref			0x000010
+#define afb_hook_flag_session_unref			0x000020
+
+#define afb_hook_flags_session_common	(afb_hook_flag_session_create|afb_hook_flag_session_close\
+					|afb_hook_flag_session_renew)
+#define afb_hook_flags_session_all	(afb_hook_flags_session_common|afb_hook_flag_session_destroy\
+					|afb_hook_flag_session_addref|afb_hook_flag_session_unref)
+
+struct afb_hook_session_itf {
+	void (*hook_session_create)(void *closure, const struct afb_hookid *hookid, struct afb_session *session);
+	void (*hook_session_close)(void *closure, const struct afb_hookid *hookid, struct afb_session *session);
+	void (*hook_session_destroy)(void *closure, const struct afb_hookid *hookid, struct afb_session *session);
+	void (*hook_session_renew)(void *closure, const struct afb_hookid *hookid, struct afb_session *session);
+	void (*hook_session_addref)(void *closure, const struct afb_hookid *hookid, struct afb_session *session);
+	void (*hook_session_unref)(void *closure, const struct afb_hookid *hookid, struct afb_session *session);
+};
+
+extern void afb_hook_session_create(struct afb_session *session);
+extern void afb_hook_session_close(struct afb_session *session);
+extern void afb_hook_session_destroy(struct afb_session *session);
+extern void afb_hook_session_renew(struct afb_session *session);
+extern void afb_hook_session_addref(struct afb_session *session);
+extern void afb_hook_session_unref(struct afb_session *session);
+
+extern struct afb_hook_session *afb_hook_create_session(const char *pattern, int flags, struct afb_hook_session_itf *itf, void *closure);
+extern struct afb_hook_session *afb_hook_addref_session(struct afb_hook_session *hook);
+extern void afb_hook_unref_session(struct afb_hook_session *hook);
+
+/*********************************************************
 * section hooking global (global interface)
 *********************************************************/