X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=librunas%2Frunas.c;h=0ffe52ea3792ad8f9e0e6a4d8933bff13c96d998;hb=6da9f915f29c93ac54ed62e7669df7c7e18a141d;hp=cbb70ee193469dfe35439b8735fcbdda9ebad938;hpb=6c9bfa35ac65810ce106433fb5721af842b75c78;p=lttng-tools.git

diff --git a/librunas/runas.c b/librunas/runas.c
index cbb70ee19..0ffe52ea3 100644
--- a/librunas/runas.c
+++ b/librunas/runas.c
@@ -28,14 +28,11 @@
 #include <unistd.h>
 #include <fcntl.h>
 #include <sched.h>
+#include <sys/mman.h>
 
 #include <lttngerr.h>
 
-/*
- * We allocate a stack of twice this size and point in the middle of it
- * to support archs with stack growing up and down.
- */
-#define CHILD_STACK_SIZE 2048
+#define CHILD_STACK_SIZE	10485760
 
 struct run_as_data {
 	int (*cmd)(void *data);
@@ -148,15 +145,19 @@ int child_run_as(void *_data)
 	 * cannot attach to this process with, e.g. ptrace, nor map this
 	 * process memory.
 	 */
-	ret = setegid(data->gid);
-	if (ret < 0) {
-		perror("setegid");
-		exit(EXIT_FAILURE);
+	if (data->gid != getegid()) {
+		ret = setegid(data->gid);
+		if (ret < 0) {
+			perror("setegid");
+			return EXIT_FAILURE;
+		}
 	}
-	ret = seteuid(data->uid);
-	if (ret < 0) {
-		perror("seteuid");
-		exit(EXIT_FAILURE);
+	if (data->uid != geteuid()) {
+		ret = seteuid(data->uid);
+		if (ret < 0) {
+			perror("seteuid");
+			return EXIT_FAILURE;
+		}
 	}
 	/*
 	 * Also set umask to 0 for mkdir executable bit.
@@ -171,13 +172,12 @@ int child_run_as(void *_data)
 				writeleft);
 		if (writelen < 0) {
 			perror("write");
-			exit(EXIT_FAILURE);
+			return EXIT_FAILURE;
 		}
 		writeleft -= writelen;
 		index += writelen;
 	} while (writeleft > 0);
-
-	exit(EXIT_SUCCESS);
+	return EXIT_SUCCESS;
 }
 
 static
@@ -187,9 +187,9 @@ int run_as(int (*cmd)(void *data), void *data, uid_t uid, gid_t gid)
 	int ret = 0;
 	int status;
 	pid_t pid;
-	char *child_stack;
 	int retval_pipe[2];
 	ssize_t readlen, readleft, index;
+	void *child_stack;
 	union {
 		int i;
 		char c[sizeof(int)];
@@ -206,10 +206,6 @@ int run_as(int (*cmd)(void *data), void *data, uid_t uid, gid_t gid)
 		}
 	}
 
-	child_stack = malloc(CHILD_STACK_SIZE * 2);
-	if (!child_stack) {
-		return -ENOMEM;
-	}
 	ret = pipe(retval_pipe);
 	if (ret < 0) {
 		perror("pipe");
@@ -220,13 +216,26 @@ int run_as(int (*cmd)(void *data), void *data, uid_t uid, gid_t gid)
 	run_as_data.uid = uid;
 	run_as_data.gid = gid;
 	run_as_data.retval_pipe = retval_pipe[1];	/* write end */
-
-	pid = clone(child_run_as, child_stack + CHILD_STACK_SIZE,
-		CLONE_FILES | SIGCHLD, &run_as_data, NULL);
+	child_stack = mmap(NULL, CHILD_STACK_SIZE,
+		PROT_WRITE | PROT_READ,
+		MAP_PRIVATE | MAP_GROWSDOWN | MAP_ANONYMOUS | MAP_STACK,
+		-1, 0);
+	if (child_stack == MAP_FAILED) {
+		perror("mmap");
+		ret = -ENOMEM;
+		goto close_pipe;
+	}
+	/*
+	 * Pointing to the middle of the stack to support architectures
+	 * where the stack grows up (HPPA).
+	 */
+	pid = clone(child_run_as, child_stack + (CHILD_STACK_SIZE / 2),
+		CLONE_FILES | SIGCHLD,
+		&run_as_data, NULL);
 	if (pid < 0) {
 		perror("clone");
 		ret = pid;
-		goto close_pipe;
+		goto unmap_stack;
 	}
 	/* receive return value */
 	readleft = sizeof(retval);
@@ -251,11 +260,15 @@ int run_as(int (*cmd)(void *data), void *data, uid_t uid, gid_t gid)
 		perror("wait");
 		ret = -1;
 	}
+unmap_stack:
+	ret = munmap(child_stack, CHILD_STACK_SIZE);
+	if (ret < 0) {
+		perror("munmap");
+	}
 close_pipe:
 	close(retval_pipe[0]);
 	close(retval_pipe[1]);
 end:
-	free(child_stack);
 	return retval.i;
 }