X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=src%2Fcommon%2Frunas.c;h=9d9091638a5a9910c8cec2b6810357ca087e4499;hb=d3c04b7c1cbb447e82858125c64c4f69160d8023;hp=c146f652dfa324ef86c19ca2362c8f7c04ffe25c;hpb=6c1c0768320135c6936c371b09731851b508c023;p=lttng-tools.git

diff --git a/src/common/runas.c b/src/common/runas.c
index c146f652d..9d9091638 100644
--- a/src/common/runas.c
+++ b/src/common/runas.c
@@ -35,6 +35,7 @@
 #include <common/utils.h>
 #include <common/compat/mman.h>
 #include <common/compat/clone.h>
+#include <common/compat/getenv.h>
 
 #include "runas.h"
 
@@ -78,6 +79,14 @@ struct run_as_open_data {
 	mode_t mode;
 };
 
+struct run_as_unlink_data {
+	const char *path;
+};
+
+struct run_as_recursive_rmdir_data {
+	const char *path;
+};
+
 #ifdef VALGRIND
 static
 int use_clone(void)
@@ -88,7 +97,7 @@ int use_clone(void)
 static
 int use_clone(void)
 {
-	return !getenv("LTTNG_DEBUG_NOCLONE");
+	return !lttng_secure_getenv("LTTNG_DEBUG_NOCLONE");
 }
 #endif
 
@@ -129,6 +138,34 @@ int _open(void *_data)
 	return open(data->path, data->flags, data->mode);
 }
 
+static
+int _unlink(void *_data)
+{
+	int ret;
+	struct run_as_unlink_data *data = _data;
+
+	ret = unlink(data->path);
+	if (ret < 0) {
+		ret = -errno;
+	}
+
+	return ret;
+}
+
+static
+int _recursive_rmdir(void *_data)
+{
+	int ret;
+	struct run_as_recursive_rmdir_data *data = _data;
+
+	ret = utils_recursive_rmdir(data->path);
+	if (ret < 0) {
+		ret = -errno;
+	}
+
+	return ret;
+}
+
 static
 int child_run_as(void *_data)
 {
@@ -340,3 +377,25 @@ int run_as_open(const char *path, int flags, mode_t mode, uid_t uid, gid_t gid)
 	data.mode = mode;
 	return run_as(_open, &data, uid, gid);
 }
+
+LTTNG_HIDDEN
+int run_as_unlink(const char *path, uid_t uid, gid_t gid)
+{
+	struct run_as_unlink_data data;
+
+	DBG3("unlink() %s with for uid %d and gid %d",
+			path, uid, gid);
+	data.path = path;
+	return run_as(_unlink, &data, uid, gid);
+}
+
+LTTNG_HIDDEN
+int run_as_recursive_rmdir(const char *path, uid_t uid, gid_t gid)
+{
+	struct run_as_recursive_rmdir_data data;
+
+	DBG3("recursive_rmdir() %s with for uid %d and gid %d",
+			path, uid, gid);
+	data.path = path;
+	return run_as(_recursive_rmdir, &data, uid, gid);
+}