Add hooking of daemon interface

Change-Id: I5840609f4300ade98412e8527ac5ce1f1e8c0c59
Signed-off-by: José Bollo <jose.bollo@iot.bzh>

1 files changed, 72 insertions, 13 deletions

diff --git a/src/afb-hook.h b/src/afb-hook.h
index 53804d40..46dd55b4 100644
--- a/src/afb-hook.h
+++ b/src/afb-hook.h
@@ -17,6 +17,22 @@
 
 #pragma once
 
+struct req;
+struct afb_context;
+struct json_object;
+struct afb_arg;
+struct afb_event;
+struct afb_session;
+struct afb_xreq;
+struct afb_ditf;
+struct sd_bus;
+struct sd_event;
+
+/*********************************************************
+* section hooking xreq
+*********************************************************/
+struct afb_hook_xreq;
+
 /* individual flags */
 #define afb_hook_flag_req_begin			0x000001
 #define afb_hook_flag_req_end			0x000002
@@ -61,16 +77,6 @@
 #define afb_hook_flags_req_extra	(afb_hook_flags_req_common|afb_hook_flags_req_ref|afb_hook_flags_req_context)
 #define afb_hook_flags_req_all		(afb_hook_flags_req_extra|afb_hook_flags_req_internal)
 
-struct req;
-struct afb_context;
-struct json_object;
-struct afb_arg;
-struct afb_event;
-struct afb_session;
-struct afb_xreq;
-
-struct afb_hook;
-
 struct afb_hook_xreq_itf {
 	void (*hook_xreq_begin)(void * closure, const struct afb_xreq *xreq);
 	void (*hook_xreq_end)(void * closure, const struct afb_xreq *xreq);
@@ -96,9 +102,9 @@ struct afb_hook_xreq_itf {
 
 extern void afb_hook_init_xreq(struct afb_xreq *xreq);
 
-extern struct afb_hook *afb_hook_xreq_create(const char *api, const char *verb, struct afb_session *session, unsigned flags, struct afb_hook_xreq_itf *itf, void *closure);
-extern struct afb_hook *afb_hook_addref(struct afb_hook *spec);
-extern void afb_hook_unref(struct afb_hook *spec);
+extern struct afb_hook_xreq *afb_hook_create_xreq(const char *api, const char *verb, struct afb_session *session, int flags, struct afb_hook_xreq_itf *itf, void *closure);
+extern struct afb_hook_xreq *afb_hook_addref_xreq(struct afb_hook_xreq *spec);
+extern void afb_hook_unref_xreq(struct afb_hook_xreq *spec);
 
 /* hooks for xreq */
 extern void afb_hook_xreq_begin(const struct afb_xreq *xreq);
@@ -122,3 +128,56 @@ extern void afb_hook_xreq_subcall_result(const struct afb_xreq *xreq, int status
 extern void afb_hook_xreq_subcallsync(const struct afb_xreq *xreq, const char *api, const char *verb, struct json_object *args);
 extern int afb_hook_xreq_subcallsync_result(const struct afb_xreq *xreq, int status, struct json_object *result);
 
+/*********************************************************
+* section hooking ditf
+*********************************************************/
+
+#define afb_hook_flag_ditf_vverbose			0x000001
+#define afb_hook_flag_ditf_event_make			0x000002
+#define afb_hook_flag_ditf_event_broadcast_before	0x000004
+#define afb_hook_flag_ditf_event_broadcast_after	0x000008
+#define afb_hook_flag_ditf_get_event_loop		0x000010
+#define afb_hook_flag_ditf_get_user_bus			0x000020
+#define afb_hook_flag_ditf_get_system_bus		0x000040
+#define afb_hook_flag_ditf_rootdir_get_fd		0x000080
+#define afb_hook_flag_ditf_rootdir_open_locale		0x000100
+
+#define afb_hook_flags_ditf_common	(afb_hook_flag_ditf_vverbose\
+					|afb_hook_flag_ditf_event_make\
+					|afb_hook_flag_ditf_event_broadcast_before\
+					|afb_hook_flag_ditf_event_broadcast_after)
+#define afb_hook_flags_ditf_extra	(afb_hook_flag_ditf_get_event_loop\
+					|afb_hook_flag_ditf_get_user_bus\
+					|afb_hook_flag_ditf_get_system_bus\
+					|afb_hook_flag_ditf_rootdir_get_fd\
+					|afb_hook_flag_ditf_rootdir_open_locale)
+
+#define afb_hook_flags_ditf_all		(afb_hook_flags_ditf_common|afb_hook_flags_ditf_extra)
+
+struct afb_hook_ditf_itf {
+       void (*hook_ditf_event_broadcast_before)(void *closure, const struct afb_ditf *ditf, const char *name, struct json_object *object);
+       void (*hook_ditf_event_broadcast_after)(void *closure, const struct afb_ditf *ditf, const char *name, struct json_object *object, int result);
+       void (*hook_ditf_get_event_loop)(void *closure, const struct afb_ditf *ditf, struct sd_event *result);
+       void (*hook_ditf_get_user_bus)(void *closure, const struct afb_ditf *ditf, struct sd_bus *result);
+       void (*hook_ditf_get_system_bus)(void *closure, const struct afb_ditf *ditf, struct sd_bus *result);
+       void (*hook_ditf_vverbose)(void*closure, const struct afb_ditf *ditf, int level, const char *file, int line, const char *function, const char *fmt, va_list args);
+       void (*hook_ditf_event_make)(void *closure, const struct afb_ditf *ditf, const char *name, struct afb_event result);
+       void (*hook_ditf_rootdir_get_fd)(void *closure, const struct afb_ditf *ditf, int result);
+       void (*hook_ditf_rootdir_open_locale)(void *closure, const struct afb_ditf *ditf, const char *filename, int flags, const char *locale, int result);
+};
+
+extern void afb_hook_ditf_event_broadcast_before(const struct afb_ditf *ditf, const char *name, struct json_object *object);
+extern int afb_hook_ditf_event_broadcast_after(const struct afb_ditf *ditf, const char *name, struct json_object *object, int result);
+extern struct sd_event *afb_hook_ditf_get_event_loop(const struct afb_ditf *ditf, struct sd_event *result);
+extern struct sd_bus *afb_hook_ditf_get_user_bus(const struct afb_ditf *ditf, struct sd_bus *result);
+extern struct sd_bus *afb_hook_ditf_get_system_bus(const struct afb_ditf *ditf, struct sd_bus *result);
+extern void afb_hook_ditf_vverbose(const struct afb_ditf *ditf, int level, const char *file, int line, const char *function, const char *fmt, va_list args);
+extern struct afb_event afb_hook_ditf_event_make(const struct afb_ditf *ditf, const char *name, struct afb_event result);
+extern int afb_hook_ditf_rootdir_get_fd(const struct afb_ditf *ditf, int result);
+extern int afb_hook_ditf_rootdir_open_locale(const struct afb_ditf *ditf, const char *filename, int flags, const char *locale, int result);
+
+extern int afb_hook_flags_ditf(const char *api);
+extern struct afb_hook_ditf *afb_hook_create_ditf(const char *api, int flags, struct afb_hook_ditf_itf *itf, void *closure);
+extern struct afb_hook_ditf *afb_hook_addref_ditf(struct afb_hook_ditf *hook);
+extern void afb_hook_unref_ditf(struct afb_hook_ditf *hook);
+