X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=ltt-sessiond%2Fcontext.h;h=872e77e7bbd67c40d631e848cddff0af5d85d74e;hb=d6b189345152f407360bc287997d38d30054bd2c;hp=c8cd627761f13cdc3a98a06fe714cd83325decd4;hpb=1e307fab325060d9db4e989c0fdc3cddf46f50f6;p=lttng-tools.git

diff --git a/ltt-sessiond/context.h b/ltt-sessiond/context.h
index c8cd62776..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.h"
 
-int add_kernel_context(struct ltt_kernel_session *ksession,
-		struct lttng_kernel_context *kctx, char *event_name,
-		char *channel_name);
+#include "trace-kernel.h"
+#include "trace-ust.h"
+
+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 */