Fix: run-as thread deadlock on itself in restart error path
[lttng-tools.git] / src / common / runas.c
index 863e824283fccfb159fad9150d859317e596618d..adbbe3e97a8446b514f5ac7ec308f0d217730115 100644 (file)
 #include <assert.h>
 #include <signal.h>
 
+#include <common/lttng-kernel.h>
 #include <common/common.h>
 #include <common/utils.h>
 #include <common/compat/getenv.h>
 #include <common/compat/prctl.h>
 #include <common/unix.h>
 #include <common/defaults.h>
+#include <common/lttng-elf.h>
+
+#include <lttng/constant.h>
 
 #include "runas.h"
 
@@ -64,6 +68,15 @@ struct run_as_rmdir_recursive_data {
        char path[PATH_MAX];
 };
 
+struct run_as_extract_elf_symbol_offset_data {
+       char function[LTTNG_SYMBOL_NAME_LEN];
+};
+
+struct run_as_extract_sdt_probe_offsets_data {
+       char probe_name[LTTNG_SYMBOL_NAME_LEN];
+       char provider_name[LTTNG_SYMBOL_NAME_LEN];
+};
+
 struct run_as_mkdir_ret {
        int ret;
 };
@@ -80,12 +93,23 @@ struct run_as_rmdir_recursive_ret {
        int ret;
 };
 
+struct run_as_extract_elf_symbol_offset_ret {
+       uint64_t offset;
+};
+
+struct run_as_extract_sdt_probe_offsets_ret {
+       uint32_t num_offset;
+       uint64_t offsets[LTTNG_KERNEL_MAX_UPROBE_NUM];
+};
+
 enum run_as_cmd {
        RUN_AS_MKDIR,
        RUN_AS_OPEN,
        RUN_AS_UNLINK,
        RUN_AS_RMDIR_RECURSIVE,
        RUN_AS_MKDIR_RECURSIVE,
+       RUN_AS_EXTRACT_ELF_SYMBOL_OFFSET,
+       RUN_AS_EXTRACT_SDT_PROBE_OFFSETS,
 };
 
 struct run_as_data {
@@ -96,6 +120,8 @@ struct run_as_data {
                struct run_as_open_data open;
                struct run_as_unlink_data unlink;
                struct run_as_rmdir_recursive_data rmdir_recursive;
+               struct run_as_extract_elf_symbol_offset_data extract_elf_symbol_offset;
+               struct run_as_extract_sdt_probe_offsets_data extract_sdt_probe_offsets;
        } u;
        uid_t uid;
        gid_t gid;
@@ -123,6 +149,8 @@ struct run_as_ret {
                struct run_as_open_ret open;
                struct run_as_unlink_ret unlink;
                struct run_as_rmdir_recursive_ret rmdir_recursive;
+               struct run_as_extract_elf_symbol_offset_ret extract_elf_symbol_offset;
+               struct run_as_extract_sdt_probe_offsets_ret extract_sdt_probe_offsets;
        } u;
        int _errno;
        bool _error;
@@ -190,7 +218,7 @@ int _open(struct run_as_data *data, struct run_as_ret *ret_value)
        ret_value->u.open.ret = open(data->u.open.path, data->u.open.flags, data->u.open.mode);
        ret_value->fd = ret_value->u.open.ret;
        ret_value->_errno = errno;
-       ret_value->_error = (ret_value->u.open.ret) ? true : false;
+       ret_value->_error = ret_value->u.open.ret < 0;
        return ret_value->u.open.ret;
 }
 
@@ -212,6 +240,83 @@ int _rmdir_recursive(struct run_as_data *data, struct run_as_ret *ret_value)
        return ret_value->u.rmdir_recursive.ret;
 }
 
+#ifdef HAVE_ELF_H
+static
+int _extract_elf_symbol_offset(struct run_as_data *data,
+               struct run_as_ret *ret_value)
+{
+       int ret = 0;
+       ret_value->_error = false;
+
+       ret = lttng_elf_get_symbol_offset(data->fd,
+                        data->u.extract_elf_symbol_offset.function,
+                        &ret_value->u.extract_elf_symbol_offset.offset);
+       if (ret) {
+               DBG("Failed to extract ELF function offset");
+               ret_value->_error = true;
+       }
+
+       return ret;
+}
+
+static
+int _extract_sdt_probe_offsets(struct run_as_data *data,
+               struct run_as_ret *ret_value)
+{
+       int ret = 0;
+       uint64_t *offsets = NULL;
+       uint32_t num_offset;
+
+       ret_value->_error = false;
+
+       /* On success, this call allocates the offsets paramater. */
+       ret = lttng_elf_get_sdt_probe_offsets(data->fd,
+                       data->u.extract_sdt_probe_offsets.provider_name,
+                       data->u.extract_sdt_probe_offsets.probe_name,
+                       &offsets, &num_offset);
+
+       if (ret) {
+               DBG("Failed to extract SDT probe offsets");
+               ret_value->_error = true;
+               goto end;
+       }
+
+       if (num_offset <= 0 || num_offset > LTTNG_KERNEL_MAX_UPROBE_NUM) {
+               DBG("Wrong number of probes.");
+               ret = -1;
+               ret_value->_error = true;
+               goto free_offset;
+       }
+
+       /* Copy the content of the offsets array to the ret struct. */
+       memcpy(ret_value->u.extract_sdt_probe_offsets.offsets,
+                       offsets, num_offset * sizeof(uint64_t));
+
+       ret_value->u.extract_sdt_probe_offsets.num_offset = num_offset;
+
+free_offset:
+       free(offsets);
+end:
+       return ret;
+}
+#else
+static
+int _extract_elf_symbol_offset(struct run_as_data *data,
+               struct run_as_ret *ret_value)
+{
+       ERR("Unimplemented runas command RUN_AS_EXTRACT_ELF_SYMBOL_OFFSET");
+       return -1;
+}
+
+static
+int _extract_sdt_probe_offsets(struct run_as_data *data,
+               struct run_as_ret *ret_value)
+{
+       ERR("Unimplemented runas command RUN_AS_EXTRACT_SDT_PROBE_OFFSETS");
+       return -1;
+}
+#endif
+
 static
 run_as_fct run_as_enum_to_fct(enum run_as_cmd cmd)
 {
@@ -226,6 +331,10 @@ run_as_fct run_as_enum_to_fct(enum run_as_cmd cmd)
                return _rmdir_recursive;
        case RUN_AS_MKDIR_RECURSIVE:
                return _mkdir_recursive;
+       case RUN_AS_EXTRACT_ELF_SYMBOL_OFFSET:
+               return _extract_elf_symbol_offset;
+       case RUN_AS_EXTRACT_SDT_PROBE_OFFSETS:
+               return _extract_sdt_probe_offsets;
        default:
                ERR("Unknown command %d", (int) cmd);
                return NULL;
@@ -238,7 +347,8 @@ int do_send_fd(int sock, int fd)
        ssize_t len;
 
        if (fd < 0) {
-               ERR("Invalid file description");
+               ERR("Attempt to send invalid file descriptor to master (fd = %i)", fd);
+               /* Return 0 as this is not a fatal error. */
                return 0;
        }
 
@@ -255,11 +365,6 @@ int do_recv_fd(int sock, int *fd)
 {
        ssize_t len;
 
-       if (*fd < 0) {
-               ERR("Invalid file description");
-               return 0;
-       }
-
        len = lttcomm_recv_fds_unix_sock(sock, fd, 1);
 
        if (!len) {
@@ -268,6 +373,12 @@ int do_recv_fd(int sock, int *fd)
                PERROR("lttcomm_recv_fds_unix_sock");
                return -1;
        }
+       if (*fd < 0) {
+               ERR("Invalid file descriptor received from worker (fd = %i)", *fd);
+               /* Return 0 as this is not a fatal error. */
+               return 0;
+       }
+
        return 0;
 }
 
@@ -277,10 +388,18 @@ int send_fd_to_worker(struct run_as_worker *worker, enum run_as_cmd cmd, int fd)
        int ret = 0;
 
        switch (cmd) {
+       case RUN_AS_EXTRACT_ELF_SYMBOL_OFFSET:
+       case RUN_AS_EXTRACT_SDT_PROBE_OFFSETS:
+               break;
        default:
                return 0;
        }
 
+       if (fd < 0) {
+               ERR("Refusing to send invalid fd to worker (fd = %i)", fd);
+               return -1;
+       }
+
        ret = do_send_fd(worker->sockpair[0], fd);
        if (ret < 0) {
                PERROR("do_send_fd");
@@ -302,6 +421,10 @@ int send_fd_to_master(struct run_as_worker *worker, enum run_as_cmd cmd, int fd)
                return 0;
        }
 
+       if (fd < 0) {
+               DBG("Not sending file descriptor to master as it is invalid (fd = %i)", fd);
+               return 0;
+       }
        ret = do_send_fd(worker->sockpair[1], fd);
        if (ret < 0) {
                PERROR("do_send_fd error");
@@ -343,6 +466,9 @@ int recv_fd_from_master(struct run_as_worker *worker, enum run_as_cmd cmd, int *
        int ret = 0;
 
        switch (cmd) {
+       case RUN_AS_EXTRACT_ELF_SYMBOL_OFFSET:
+       case RUN_AS_EXTRACT_SDT_PROBE_OFFSETS:
+               break;
        default:
                return 0;
        }
@@ -362,10 +488,16 @@ int cleanup_received_fd(enum run_as_cmd cmd, int fd)
        int ret = 0;
 
        switch (cmd) {
+       case RUN_AS_EXTRACT_ELF_SYMBOL_OFFSET:
+       case RUN_AS_EXTRACT_SDT_PROBE_OFFSETS:
+               break;
        default:
                return 0;
        }
 
+       if (fd < 0) {
+               return 0;
+       }
        ret = close(fd);
        if (ret < 0) {
                PERROR("close error");
@@ -374,6 +506,7 @@ int cleanup_received_fd(enum run_as_cmd cmd, int fd)
 
        return ret;
 }
+
 /*
  * Return < 0 on error, 0 if OK, 1 on hangup.
  */
@@ -387,6 +520,9 @@ int handle_one_cmd(struct run_as_worker *worker)
        run_as_fct cmd;
        uid_t prev_euid;
 
+       memset(&sendret, 0, sizeof(sendret));
+       sendret.fd = -1;
+
        /*
         * Stage 1: Receive run_as_data struct from the master.
         * The structure contains the command type and all the parameters needed for
@@ -427,6 +563,8 @@ int handle_one_cmd(struct run_as_worker *worker)
        if (data.gid != getegid()) {
                ret = setegid(data.gid);
                if (ret < 0) {
+                       sendret._error = true;
+                       sendret._errno = errno;
                        PERROR("setegid");
                        goto write_return;
                }
@@ -434,6 +572,8 @@ int handle_one_cmd(struct run_as_worker *worker)
        if (data.uid != prev_euid) {
                ret = seteuid(data.uid);
                if (ret < 0) {
+                       sendret._error = true;
+                       sendret._errno = errno;
                        PERROR("seteuid");
                        goto write_return;
                }
@@ -609,6 +749,12 @@ int run_as_cmd(struct run_as_worker *worker,
                PERROR("Error reading response from run_as");
                ret = -1;
                ret_value->_errno = errno;
+               goto end;
+       }
+
+       if (ret_value->_error) {
+               /* Skip stage 5 on error as there will be no fd to receive. */
+               goto end;
        }
 
        /*
@@ -652,160 +798,6 @@ end:
        return ret;
 }
 
-static
-int run_as_restart_worker(struct run_as_worker *worker)
-{
-       int ret = 0;
-       char *procname = NULL;
-
-       procname = worker->procname;
-
-       /* Close socket to run_as worker process and clean up the zombie process */
-       run_as_destroy_worker();
-
-       /* Create a new run_as worker process*/
-       ret = run_as_create_worker(procname);
-       if (ret < 0 ) {
-               ERR("Restarting the worker process failed");
-               ret = -1;
-               goto err;
-       }
-err:
-       return ret;
-}
-
-static
-int run_as(enum run_as_cmd cmd, struct run_as_data *data,
-                  struct run_as_ret *ret_value, uid_t uid, gid_t gid)
-{
-       int ret, saved_errno;
-
-       if (use_clone()) {
-               DBG("Using run_as worker");
-               pthread_mutex_lock(&worker_lock);
-               assert(global_worker);
-
-               ret = run_as_cmd(global_worker, cmd, data, ret_value, uid, gid);
-               saved_errno = ret_value->_errno;
-
-               pthread_mutex_unlock(&worker_lock);
-               /*
-                * If the worker thread crashed the errno is set to EIO. we log
-                * the error and  start a new worker process.
-                */
-               if (ret == -1 && saved_errno == EIO) {
-                       DBG("Socket closed unexpectedly... "
-                                       "Restarting the worker process");
-                       ret = run_as_restart_worker(global_worker);
-
-                       if (ret == -1) {
-                               ERR("Failed to restart worker process.");
-                               goto err;
-                       }
-               }
-       } else {
-               DBG("Using run_as without worker");
-               ret = run_as_noworker(cmd, data, ret_value, uid, gid);
-       }
-err:
-       return ret;
-}
-
-LTTNG_HIDDEN
-int run_as_mkdir_recursive(const char *path, mode_t mode, uid_t uid, gid_t gid)
-{
-       struct run_as_data data;
-       struct run_as_ret ret;
-
-       memset(&data, 0, sizeof(data));
-       memset(&ret, 0, sizeof(ret));
-       DBG3("mkdir() recursive %s with mode %d for uid %d and gid %d",
-                       path, (int) mode, (int) uid, (int) gid);
-       strncpy(data.u.mkdir.path, path, PATH_MAX - 1);
-       data.u.mkdir.path[PATH_MAX - 1] = '\0';
-       data.u.mkdir.mode = mode;
-
-       run_as(RUN_AS_MKDIR_RECURSIVE, &data, &ret, uid, gid);
-       errno = ret._errno;
-       return ret.u.mkdir.ret;
-}
-
-LTTNG_HIDDEN
-int run_as_mkdir(const char *path, mode_t mode, uid_t uid, gid_t gid)
-{
-       struct run_as_data data;
-       struct run_as_ret ret;
-
-       memset(&data, 0, sizeof(data));
-       memset(&ret, 0, sizeof(ret));
-
-       DBG3("mkdir() %s with mode %d for uid %d and gid %d",
-                       path, (int) mode, (int) uid, (int) gid);
-       strncpy(data.u.mkdir.path, path, PATH_MAX - 1);
-       data.u.mkdir.path[PATH_MAX - 1] = '\0';
-       data.u.mkdir.mode = mode;
-       run_as(RUN_AS_MKDIR, &data, &ret, uid, gid);
-       errno = ret._errno;
-       return ret.u.mkdir.ret;
-}
-
-LTTNG_HIDDEN
-int run_as_open(const char *path, int flags, mode_t mode, uid_t uid, gid_t gid)
-{
-       struct run_as_data data;
-       struct run_as_ret ret;
-
-       memset(&data, 0, sizeof(data));
-       memset(&ret, 0, sizeof(ret));
-
-       DBG3("open() %s with flags %X mode %d for uid %d and gid %d",
-                       path, flags, (int) mode, (int) uid, (int) gid);
-       strncpy(data.u.open.path, path, PATH_MAX - 1);
-       data.u.open.path[PATH_MAX - 1] = '\0';
-       data.u.open.flags = flags;
-       data.u.open.mode = mode;
-       run_as(RUN_AS_OPEN, &data, &ret, uid, gid);
-       errno = ret._errno;
-       ret.u.open.ret = ret.fd;
-       return ret.u.open.ret;
-}
-
-LTTNG_HIDDEN
-int run_as_unlink(const char *path, uid_t uid, gid_t gid)
-{
-       struct run_as_data data;
-       struct run_as_ret ret;
-
-       memset(&data, 0, sizeof(data));
-       memset(&ret, 0, sizeof(ret));
-
-       DBG3("unlink() %s with for uid %d and gid %d",
-                       path, (int) uid, (int) gid);
-       strncpy(data.u.unlink.path, path, PATH_MAX - 1);
-       data.u.unlink.path[PATH_MAX - 1] = '\0';
-       run_as(RUN_AS_UNLINK, &data, &ret, uid, gid);
-       errno = ret._errno;
-       return ret.u.unlink.ret;
-}
-
-LTTNG_HIDDEN
-int run_as_rmdir_recursive(const char *path, uid_t uid, gid_t gid)
-{
-       struct run_as_data data;
-       struct run_as_ret ret;
-
-       memset(&data, 0, sizeof(data));
-       memset(&ret, 0, sizeof(ret));
-
-       DBG3("rmdir_recursive() %s with for uid %d and gid %d",
-                       path, (int) uid, (int) gid);
-       strncpy(data.u.rmdir_recursive.path, path, PATH_MAX - 1);
-       data.u.rmdir_recursive.path[PATH_MAX - 1] = '\0';
-       run_as(RUN_AS_RMDIR_RECURSIVE, &data, &ret, uid, gid);
-       errno = ret._errno;
-       return ret.u.rmdir_recursive.ret;
-}
-
 static
 int reset_sighandler(void)
 {
@@ -877,8 +869,10 @@ end:
        return ret;
 }
 
-LTTNG_HIDDEN
-int run_as_create_worker(char *procname)
+static
+int run_as_create_worker_no_lock(const char *procname,
+               post_fork_cleanup_cb clean_up_func,
+               void *clean_up_user_data)
 {
        pid_t pid;
        int i, ret = 0;
@@ -886,7 +880,6 @@ int run_as_create_worker(char *procname)
        struct run_as_ret recvret;
        struct run_as_worker *worker;
 
-       pthread_mutex_lock(&worker_lock);
        assert(!global_worker);
        if (!use_clone()) {
                /*
@@ -901,7 +894,11 @@ int run_as_create_worker(char *procname)
                ret = -ENOMEM;
                goto end;
        }
-       worker->procname = procname;
+       worker->procname = strdup(procname);
+       if (!worker->procname) {
+               ret = -ENOMEM;
+               goto error_procname_alloc;
+       }
        /* Create unix socket. */
        if (lttcomm_create_anon_unix_socketpair(worker->sockpair) < 0) {
                ret = -1;
@@ -920,9 +917,13 @@ int run_as_create_worker(char *procname)
                reset_sighandler();
 
                set_worker_sighandlers();
+               if (clean_up_func) {
+                       if (clean_up_func(clean_up_user_data) < 0) {
+                               ERR("Run-as post-fork clean-up failed, exiting.");
+                               exit(EXIT_FAILURE);
+                       }
+               }
 
-               /* The child has no use for this lock. */
-               pthread_mutex_unlock(&worker_lock);
                /* Just close, no shutdown. */
                if (close(worker->sockpair[0])) {
                        PERROR("close");
@@ -946,6 +947,8 @@ int run_as_create_worker(char *procname)
                        ret = -1;
                }
                worker->sockpair[1] = -1;
+               free(worker->procname);
+               free(worker);
                LOG(ret ? PRINT_ERR : PRINT_DBG, "run_as worker exiting (ret = %d)", ret);
                exit(ret ? EXIT_FAILURE : EXIT_SUCCESS);
        } else {
@@ -971,7 +974,6 @@ int run_as_create_worker(char *procname)
                global_worker = worker;
        }
 end:
-       pthread_mutex_unlock(&worker_lock);
        return ret;
 
        /* Error handling. */
@@ -986,20 +988,20 @@ error_fork:
                worker->sockpair[i] = -1;
        }
 error_sock:
+       free(worker->procname);
+error_procname_alloc:
        free(worker);
-       pthread_mutex_unlock(&worker_lock);
        return ret;
 }
 
-LTTNG_HIDDEN
-void run_as_destroy_worker(void)
+static
+void run_as_destroy_worker_no_lock(void)
 {
        struct run_as_worker *worker = global_worker;
 
        DBG("Destroying run_as worker");
-       pthread_mutex_lock(&worker_lock);
        if (!worker) {
-               goto end;
+               return;
        }
        /* Close unix socket */
        DBG("Closing run_as worker socket");
@@ -1032,8 +1034,256 @@ void run_as_destroy_worker(void)
                        break;
                }
        }
+       free(worker->procname);
        free(worker);
        global_worker = NULL;
-end:
+}
+
+static
+int run_as_restart_worker(struct run_as_worker *worker)
+{
+       int ret = 0;
+       char *procname = NULL;
+
+       procname = worker->procname;
+
+       /* Close socket to run_as worker process and clean up the zombie process */
+       run_as_destroy_worker_no_lock();
+
+       /* Create a new run_as worker process*/
+       ret = run_as_create_worker_no_lock(procname, NULL, NULL);
+       if (ret < 0 ) {
+               ERR("Restarting the worker process failed");
+               ret = -1;
+               goto err;
+       }
+err:
+       return ret;
+}
+
+static
+int run_as(enum run_as_cmd cmd, struct run_as_data *data,
+                  struct run_as_ret *ret_value, uid_t uid, gid_t gid)
+{
+       int ret, saved_errno;
+
+       pthread_mutex_lock(&worker_lock);
+       if (use_clone()) {
+               DBG("Using run_as worker");
+
+               assert(global_worker);
+
+               ret = run_as_cmd(global_worker, cmd, data, ret_value, uid, gid);
+               saved_errno = ret_value->_errno;
+
+               /*
+                * If the worker thread crashed the errno is set to EIO. we log
+                * the error and  start a new worker process.
+                */
+               if (ret == -1 && saved_errno == EIO) {
+                       DBG("Socket closed unexpectedly... "
+                                       "Restarting the worker process");
+                       ret = run_as_restart_worker(global_worker);
+                       if (ret == -1) {
+                               ERR("Failed to restart worker process.");
+                               goto err;
+                       }
+               }
+       } else {
+               DBG("Using run_as without worker");
+               ret = run_as_noworker(cmd, data, ret_value, uid, gid);
+       }
+err:
+       pthread_mutex_unlock(&worker_lock);
+       return ret;
+}
+
+LTTNG_HIDDEN
+int run_as_mkdir_recursive(const char *path, mode_t mode, uid_t uid, gid_t gid)
+{
+       struct run_as_data data;
+       struct run_as_ret ret;
+
+       memset(&data, 0, sizeof(data));
+       memset(&ret, 0, sizeof(ret));
+       DBG3("mkdir() recursive %s with mode %d for uid %d and gid %d",
+                       path, (int) mode, (int) uid, (int) gid);
+       strncpy(data.u.mkdir.path, path, PATH_MAX - 1);
+       data.u.mkdir.path[PATH_MAX - 1] = '\0';
+       data.u.mkdir.mode = mode;
+
+       run_as(RUN_AS_MKDIR_RECURSIVE, &data, &ret, uid, gid);
+       errno = ret._errno;
+       return ret.u.mkdir.ret;
+}
+
+LTTNG_HIDDEN
+int run_as_mkdir(const char *path, mode_t mode, uid_t uid, gid_t gid)
+{
+       struct run_as_data data;
+       struct run_as_ret ret;
+
+       memset(&data, 0, sizeof(data));
+       memset(&ret, 0, sizeof(ret));
+
+       DBG3("mkdir() %s with mode %d for uid %d and gid %d",
+                       path, (int) mode, (int) uid, (int) gid);
+       strncpy(data.u.mkdir.path, path, PATH_MAX - 1);
+       data.u.mkdir.path[PATH_MAX - 1] = '\0';
+       data.u.mkdir.mode = mode;
+       run_as(RUN_AS_MKDIR, &data, &ret, uid, gid);
+       errno = ret._errno;
+       return ret.u.mkdir.ret;
+}
+
+LTTNG_HIDDEN
+int run_as_open(const char *path, int flags, mode_t mode, uid_t uid, gid_t gid)
+{
+       struct run_as_data data;
+       struct run_as_ret ret;
+
+       memset(&data, 0, sizeof(data));
+       memset(&ret, 0, sizeof(ret));
+
+       DBG3("open() %s with flags %X mode %d for uid %d and gid %d",
+                       path, flags, (int) mode, (int) uid, (int) gid);
+       strncpy(data.u.open.path, path, PATH_MAX - 1);
+       data.u.open.path[PATH_MAX - 1] = '\0';
+       data.u.open.flags = flags;
+       data.u.open.mode = mode;
+       run_as(RUN_AS_OPEN, &data, &ret, uid, gid);
+       errno = ret._errno;
+       ret.u.open.ret = ret.fd;
+       return ret.u.open.ret;
+}
+
+LTTNG_HIDDEN
+int run_as_unlink(const char *path, uid_t uid, gid_t gid)
+{
+       struct run_as_data data;
+       struct run_as_ret ret;
+
+       memset(&data, 0, sizeof(data));
+       memset(&ret, 0, sizeof(ret));
+
+       DBG3("unlink() %s with for uid %d and gid %d",
+                       path, (int) uid, (int) gid);
+       strncpy(data.u.unlink.path, path, PATH_MAX - 1);
+       data.u.unlink.path[PATH_MAX - 1] = '\0';
+       run_as(RUN_AS_UNLINK, &data, &ret, uid, gid);
+       errno = ret._errno;
+       return ret.u.unlink.ret;
+}
+
+LTTNG_HIDDEN
+int run_as_rmdir_recursive(const char *path, uid_t uid, gid_t gid)
+{
+       struct run_as_data data;
+       struct run_as_ret ret;
+
+       memset(&data, 0, sizeof(data));
+       memset(&ret, 0, sizeof(ret));
+
+       DBG3("rmdir_recursive() %s with for uid %d and gid %d",
+                       path, (int) uid, (int) gid);
+       strncpy(data.u.rmdir_recursive.path, path, PATH_MAX - 1);
+       data.u.rmdir_recursive.path[PATH_MAX - 1] = '\0';
+       run_as(RUN_AS_RMDIR_RECURSIVE, &data, &ret, uid, gid);
+       errno = ret._errno;
+       return ret.u.rmdir_recursive.ret;
+}
+
+LTTNG_HIDDEN
+int run_as_extract_elf_symbol_offset(int fd, const char* function,
+               uid_t uid, gid_t gid, uint64_t *offset)
+{
+       struct run_as_data data;
+       struct run_as_ret ret;
+
+       memset(&data, 0, sizeof(data));
+       memset(&ret, 0, sizeof(ret));
+
+       DBG3("extract_elf_symbol_offset() on fd=%d and function=%s "
+               "with for uid %d and gid %d", fd, function, (int) uid, (int) gid);
+
+       data.fd = fd;
+
+       strncpy(data.u.extract_elf_symbol_offset.function, function, LTTNG_SYMBOL_NAME_LEN - 1);
+
+       data.u.extract_elf_symbol_offset.function[LTTNG_SYMBOL_NAME_LEN - 1] = '\0';
+
+       run_as(RUN_AS_EXTRACT_ELF_SYMBOL_OFFSET, &data, &ret, uid, gid);
+
+       errno = ret._errno;
+
+       if (ret._error) {
+               return -1;
+       }
+
+       *offset = ret.u.extract_elf_symbol_offset.offset;
+       return 0;
+}
+
+LTTNG_HIDDEN
+int run_as_extract_sdt_probe_offsets(int fd, const char* provider_name,
+               const char* probe_name, uid_t uid, gid_t gid,
+               uint64_t **offsets, uint32_t *num_offset)
+{
+       struct run_as_data data;
+       struct run_as_ret ret;
+
+       memset(&data, 0, sizeof(data));
+       memset(&ret, 0, sizeof(ret));
+
+       DBG3("extract_sdt_probe_offsets() on fd=%d, probe_name=%s and "
+               "provider_name=%s with for uid %d and gid %d", fd, probe_name,
+               provider_name, (int) uid, (int) gid);
+
+       data.fd = fd;
+
+       strncpy(data.u.extract_sdt_probe_offsets.probe_name, probe_name, LTTNG_SYMBOL_NAME_LEN - 1);
+       strncpy(data.u.extract_sdt_probe_offsets.provider_name, provider_name, LTTNG_SYMBOL_NAME_LEN - 1);
+
+       data.u.extract_sdt_probe_offsets.probe_name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0';
+       data.u.extract_sdt_probe_offsets.provider_name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0';
+
+       run_as(RUN_AS_EXTRACT_SDT_PROBE_OFFSETS, &data, &ret, uid, gid);
+
+       errno = ret._errno;
+
+       if (ret._error) {
+               return -1;
+       }
+
+       *num_offset = ret.u.extract_sdt_probe_offsets.num_offset;
+
+       *offsets = zmalloc(*num_offset * sizeof(uint64_t));
+       if (!*offsets) {
+               return -ENOMEM;
+       }
+
+       memcpy(*offsets, ret.u.extract_sdt_probe_offsets.offsets, *num_offset * sizeof(uint64_t));
+       return 0;
+}
+
+LTTNG_HIDDEN
+int run_as_create_worker(const char *procname,
+               post_fork_cleanup_cb clean_up_func,
+               void *clean_up_user_data)
+{
+       int ret;
+
+       pthread_mutex_lock(&worker_lock);
+       ret = run_as_create_worker_no_lock(procname, clean_up_func,
+                       clean_up_user_data);
+       pthread_mutex_unlock(&worker_lock);
+       return ret;
+}
+
+LTTNG_HIDDEN
+void run_as_destroy_worker(void)
+{
+       pthread_mutex_lock(&worker_lock);
+       run_as_destroy_worker_no_lock();
        pthread_mutex_unlock(&worker_lock);
 }
This page took 0.032129 seconds and 4 git commands to generate.