X-Git-Url: http://git.lttng.org/?a=blobdiff_plain;f=src%2Flttng-context-vegid.c;h=c723411ea00e1b93fccd4c8589f7fad9dc8ecf8a;hb=a9d49ac600f171e2289c8c409587bbcf2eec9f99;hp=87c53baff039edbb3e4a05066257ce4c9a010dcb;hpb=2dc781e02eb156a76554ada092a181ab2916db57;p=lttng-modules.git

diff --git a/src/lttng-context-vegid.c b/src/lttng-context-vegid.c
index 87c53baf..c723411e 100644
--- a/src/lttng-context-vegid.c
+++ b/src/lttng-context-vegid.c
@@ -11,15 +11,16 @@
 
 #include <linux/module.h>
 #include <linux/sched.h>
+#include <linux/user_namespace.h>
+
 #include <lttng/events.h>
 #include <lttng/events-internal.h>
 #include <lttng/tracer.h>
 #include <ringbuffer/frontend_types.h>
 #include <wrapper/vmalloc.h>
-#include <wrapper/user_namespace.h>
 
 static
-size_t vegid_get_size(void *priv, struct lttng_probe_ctx *probe_ctx, size_t offset)
+size_t vegid_get_size(void *priv, struct lttng_kernel_probe_ctx *probe_ctx, size_t offset)
 {
 	size_t size = 0;
 
@@ -29,29 +30,28 @@ size_t vegid_get_size(void *priv, struct lttng_probe_ctx *probe_ctx, size_t offs
 }
 
 static
-void vegid_record(void *priv, struct lttng_probe_ctx *probe_ctx,
-		 struct lib_ring_buffer_ctx *ctx,
-		 struct lttng_channel *chan)
+void vegid_record(void *priv, struct lttng_kernel_probe_ctx *probe_ctx,
+		 struct lttng_kernel_ring_buffer_ctx *ctx,
+		 struct lttng_kernel_channel_buffer *chan)
 {
 	gid_t vegid;
 
-	vegid = lttng_current_vegid();
-	lib_ring_buffer_align_ctx(ctx, lttng_alignof(vegid));
-	chan->ops->event_write(ctx, &vegid, sizeof(vegid));
+	vegid = from_kgid_munged(current_user_ns(), current_egid());
+	chan->ops->event_write(ctx, &vegid, sizeof(vegid), lttng_alignof(vegid));
 }
 
 static
 void vegid_get_value(void *priv,
-		struct lttng_probe_ctx *lttng_probe_ctx,
+		struct lttng_kernel_probe_ctx *lttng_probe_ctx,
 		struct lttng_ctx_value *value)
 {
-	value->u.s64 = lttng_current_vegid();
+	value->u.s64 = from_kgid_munged(current_user_ns(), current_egid());
 }
 
 static const struct lttng_kernel_ctx_field *ctx_field = lttng_kernel_static_ctx_field(
 	lttng_kernel_static_event_field("vegid",
 		lttng_kernel_static_type_integer_from_type(gid_t, __BYTE_ORDER, 10),
-		false, false, false),
+		false, false),
 	vegid_get_size,
 	vegid_record,
 	vegid_get_value,