#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, int domain);
#endif /* _LTT_CONTEXT_H */