summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--src/afb-hook.c195
-rw-r--r--src/afb-hook.h37
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)
*********************************************************/