Cleanup: Move wrapper/ headers to include/wrapper/

[lttng-modules.git] / include / wrapper / user_namespace.h

diff --git a/include/wrapper/user_namespace.h b/include/wrapper/user_namespace.h
new file mode 100644 (file)
index 0000000..5e0ab5a
--- /dev/null
+++ b/include/wrapper/user_namespace.h
@@ -0,0 +1,116 @@
+/* SPDX-License-Identifier: (GPL-2.0-only or LGPL-2.1-only)
+ *
+ * wrapper/user_namespace.h
+ *
+ * Copyright (C) 2019 Michael Jeanson <mjeanson@efficios.com>
+ *
+ */
+
+#ifndef _LTTNG_WRAPPER_USER_NAMESPACE_H
+#define _LTTNG_WRAPPER_USER_NAMESPACE_H
+
+#include <linux/version.h>
+#include <linux/user_namespace.h>
+
+
+#if LINUX_VERSION_CODE >= KERNEL_VERSION(3, 5, 0)
+
+#define lttng_current_xxuid(xxx)                               \
+       (from_kuid_munged(&init_user_ns, current_##xxx()))
+
+#define lttng_current_vxxuid(xxx)                              \
+       (from_kuid_munged(current_user_ns(), current_##xxx()))
+
+#define lttng_current_xxgid(xxx)                               \
+       (from_kgid_munged(&init_user_ns, current_##xxx()))
+
+#define lttng_current_vxxgid(xxx)                              \
+       (from_kgid_munged(current_user_ns(), current_##xxx()))
+
+static inline
+uid_t lttng_task_vuid(struct task_struct *p, struct user_namespace *ns)
+{
+       uid_t uid;
+       kuid_t kuid;
+
+       kuid = task_cred_xxx(p, uid);
+       uid = from_kuid_munged(ns, kuid);
+
+       return uid;
+}
+
+static inline
+gid_t lttng_task_vgid(struct task_struct *p, struct user_namespace *ns)
+{
+       gid_t gid;
+       kgid_t kgid;
+
+       kgid = task_cred_xxx(p, gid);
+       gid = from_kgid_munged(ns, kgid);
+
+       return gid;
+}
+
+#else
+
+#define lttng_current_xxuid(xxx)       (current_##xxx())
+
+#define lttng_current_vxxuid(xxx)                                      \
+       (user_ns_map_uid(current_user_ns(), current_cred(), current_##xxx()))
+
+#define lttng_current_xxgid(xxx)       (current_##xxx())
+
+#define lttng_current_vxxgid(xxx)                                      \
+       (user_ns_map_gid(current_user_ns(), current_cred(), current_##xxx()))
+
+static inline
+uid_t lttng_task_vuid(struct task_struct *p, struct user_namespace *ns)
+{
+       uid_t uid;
+
+       /*
+        * __task_cred requires the RCU readlock be held
+        */
+       rcu_read_lock();
+       uid = user_ns_map_uid(ns, __task_cred(p), __task_cred(p)->uid);
+       rcu_read_unlock();
+
+       return uid;
+}
+
+static inline
+gid_t lttng_task_vgid(struct task_struct *p, struct user_namespace *ns)
+{
+       gid_t gid;
+
+       /*
+        * __task_cred requires the RCU readlock be held
+        */
+       rcu_read_lock();
+       gid = user_ns_map_gid(ns, __task_cred(p), __task_cred(p)->gid);
+       rcu_read_unlock();
+
+       return gid;
+}
+
+#endif
+
+#define lttng_current_uid()    (lttng_current_xxuid(uid))
+#define lttng_current_euid()   (lttng_current_xxuid(euid))
+#define lttng_current_suid()   (lttng_current_xxuid(suid))
+#define lttng_current_fsuid()  (lttng_current_xxuid(fsuid))
+#define lttng_current_gid()    (lttng_current_xxgid(gid))
+#define lttng_current_egid()   (lttng_current_xxgid(egid))
+#define lttng_current_sgid()   (lttng_current_xxgid(sgid))
+#define lttng_current_fsgid()  (lttng_current_xxgid(fsgid))
+
+#define lttng_current_vuid()   (lttng_current_vxxuid(uid))
+#define lttng_current_veuid()  (lttng_current_vxxuid(euid))
+#define lttng_current_vsuid()  (lttng_current_vxxuid(suid))
+#define lttng_current_vfsuid() (lttng_current_vxxuid(fsuid))
+#define lttng_current_vgid()   (lttng_current_vxxgid(gid))
+#define lttng_current_vegid()  (lttng_current_vxxgid(egid))
+#define lttng_current_vsgid()  (lttng_current_vxxgid(sgid))
+#define lttng_current_vfsgid() (lttng_current_vxxgid(fsgid))
+
+#endif /* _LTTNG_WRAPPER_USER_NAMESPACE_H */