X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=ltt-sessiond%2Fcontext.h;h=872e77e7bbd67c40d631e848cddff0af5d85d74e;hb=9674ce7a9c5218066b395d1f6d1f19f277849f95;hp=a3d82e0b7d5af55c5cc664d58e73f65bfa352bdd;hpb=f8f0c3d1337c4eda5fb76a323ffe6c21c926a21e;p=lttng-tools.git

diff --git a/ltt-sessiond/context.h b/ltt-sessiond/context.h
index a3d82e0b7..872e77e7b 100644
--- a/ltt-sessiond/context.h
+++ b/ltt-sessiond/context.h
@@ -18,12 +18,20 @@
 #ifndef _LTT_CONTEXT_H
 #define _LTT_CONTEXT_H
 
-#include <lttng-kernel.h>
+#include <lttng/lttng.h>
+#ifdef CONFIG_LTTNG_TOOLS_HAVE_UST
+#include <ust/lttng-ust-abi.h>
+#else
+#include "lttng-ust-abi.h"
+#endif
+
 
 #include "trace-kernel.h"
+#include "trace-ust.h"
 
-int add_kernel_context(struct ltt_kernel_session *ksession,
-		struct lttng_kernel_context *kctx, char *event_name,
-		char *channel_name);
+int context_kernel_add(struct ltt_kernel_session *ksession,
+		struct lttng_event_context *ctx, char *event_name, char *channel_name);
+int context_ust_add(struct ltt_ust_session *ustsession,
+		struct lttng_event_context *ctx, char *event_name, char *channel_name);
 
 #endif /* _LTT_CONTEXT_H */