Clean-up: run-as: change space to tabs
[lttng-tools.git] / src / common / runas.c
index 9f30996e2ae77f6e2a8ad5a26ace9f00b0eda879..28bcff30a5c550ba7800f2b74aa249362d0c24ec 100644 (file)
@@ -1,20 +1,10 @@
 /*
- * Copyright (C) 2011 David Goulet <david.goulet@polymtl.ca>
- *                      Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
- *               2019 - Jérémie Galarneau <jeremie.galarneau@efficios.com>
+ * Copyright (C) 2011 David Goulet <david.goulet@polymtl.ca>
+ * Copyright (C) 2011 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
+ * Copyright (C) 2019 Jérémie Galarneau <jeremie.galarneau@efficios.com>
  *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License, version 2 only,
- * as published by the Free Software Foundation.
+ * SPDX-License-Identifier: GPL-2.0-only
  *
- * This program is distributed in the hope that it will be useful, but WITHOUT
- * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
- * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
- * more details.
- *
- * You should have received a copy of the GNU General Public License along
- * with this program; if not, write to the Free Software Foundation, Inc.,
- * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  */
 
 #define _LGPL_SOURCE
@@ -90,6 +80,7 @@ struct run_as_unlink_data {
 struct run_as_rmdir_data {
        int dirfd;
        char path[LTTNG_PATH_MAX];
+       int flags; /* enum lttng_directory_handle_rmdir_recursive_flags */
 } LTTNG_PACKED;
 
 struct run_as_extract_elf_symbol_offset_data {
@@ -350,22 +341,27 @@ int _mkdirat_recursive(struct run_as_data *data, struct run_as_ret *ret_value)
 {
        const char *path;
        mode_t mode;
-       struct lttng_directory_handle handle;
+       struct lttng_directory_handle *handle;
 
        path = data->u.mkdir.path;
        mode = data->u.mkdir.mode;
 
-       (void) lttng_directory_handle_init_from_dirfd(&handle,
-                       data->u.mkdir.dirfd);
+       handle = lttng_directory_handle_create_from_dirfd(data->u.mkdir.dirfd);
+       if (!handle) {
+               ret_value->_errno = errno;
+               ret_value->_error = true;
+               ret_value->u.ret = -1;
+               goto end;
+       }
        /* Ownership of dirfd is transferred to the handle. */
        data->u.mkdir.dirfd = -1;
        /* Safe to call as we have transitioned to the requested uid/gid. */
-       ret_value->u.ret =
-                       lttng_directory_handle_create_subdirectory_recursive(
-                                       &handle, path, mode);
+       ret_value->u.ret = lttng_directory_handle_create_subdirectory_recursive(
+                       handle, path, mode);
        ret_value->_errno = errno;
        ret_value->_error = (ret_value->u.ret) ? true : false;
-       lttng_directory_handle_fini(&handle);
+       lttng_directory_handle_put(handle);
+end:
        return ret_value->u.ret;
 }
 
@@ -374,22 +370,27 @@ int _mkdirat(struct run_as_data *data, struct run_as_ret *ret_value)
 {
        const char *path;
        mode_t mode;
-       struct lttng_directory_handle handle;
+       struct lttng_directory_handle *handle;
 
        path = data->u.mkdir.path;
        mode = data->u.mkdir.mode;
 
-       (void) lttng_directory_handle_init_from_dirfd(&handle,
-                       data->u.mkdir.dirfd);
+       handle = lttng_directory_handle_create_from_dirfd(data->u.mkdir.dirfd);
+       if (!handle) {
+               ret_value->u.ret = -1;
+               ret_value->_errno = errno;
+               ret_value->_error = true;
+               goto end;
+       }
        /* Ownership of dirfd is transferred to the handle. */
        data->u.mkdir.dirfd = -1;
        /* Safe to call as we have transitioned to the requested uid/gid. */
-       ret_value->u.ret =
-                       lttng_directory_handle_create_subdirectory(
-                                       &handle, path, mode);
+       ret_value->u.ret = lttng_directory_handle_create_subdirectory(
+                       handle, path, mode);
        ret_value->_errno = errno;
        ret_value->_error = (ret_value->u.ret) ? true : false;
-       lttng_directory_handle_fini(&handle);
+       lttng_directory_handle_put(handle);
+end:
        return ret_value->u.ret;
 }
 
@@ -397,14 +398,19 @@ static
 int _open(struct run_as_data *data, struct run_as_ret *ret_value)
 {
        int fd;
-       struct lttng_directory_handle handle;
+       struct lttng_directory_handle *handle;
 
-       (void) lttng_directory_handle_init_from_dirfd(&handle,
-                       data->u.open.dirfd);
+       handle = lttng_directory_handle_create_from_dirfd(data->u.open.dirfd);
+       if (!handle) {
+               ret_value->_errno = errno;
+               ret_value->_error = true;
+               ret_value->u.ret = -1;
+               goto end;
+       }
        /* Ownership of dirfd is transferred to the handle. */
        data->u.open.dirfd = -1;
 
-       fd = lttng_directory_handle_open_file(&handle,
+       fd = lttng_directory_handle_open_file(handle,
                        data->u.open.path, data->u.open.flags,
                        data->u.open.mode);
        if (fd < 0) {
@@ -417,64 +423,83 @@ int _open(struct run_as_data *data, struct run_as_ret *ret_value)
 
        ret_value->_errno = errno;
        ret_value->_error = fd < 0;
-       lttng_directory_handle_fini(&handle);
+       lttng_directory_handle_put(handle);
+end:
        return ret_value->u.ret;
 }
 
 static
 int _unlink(struct run_as_data *data, struct run_as_ret *ret_value)
 {
-       struct lttng_directory_handle handle;
+       struct lttng_directory_handle *handle;
 
-       (void) lttng_directory_handle_init_from_dirfd(&handle,
-                       data->u.unlink.dirfd);
+       handle = lttng_directory_handle_create_from_dirfd(data->u.unlink.dirfd);
+       if (!handle) {
+               ret_value->u.ret = -1;
+               ret_value->_errno = errno;
+               ret_value->_error = true;
+               goto end;
+       }
 
        /* Ownership of dirfd is transferred to the handle. */
        data->u.unlink.dirfd = -1;
 
-       ret_value->u.ret = lttng_directory_handle_unlink_file(&handle,
+       ret_value->u.ret = lttng_directory_handle_unlink_file(handle,
                        data->u.unlink.path);
        ret_value->_errno = errno;
        ret_value->_error = (ret_value->u.ret) ? true : false;
-       lttng_directory_handle_fini(&handle);
+       lttng_directory_handle_put(handle);
+end:
        return ret_value->u.ret;
 }
 
 static
 int _rmdir(struct run_as_data *data, struct run_as_ret *ret_value)
 {
-       struct lttng_directory_handle handle;
+       struct lttng_directory_handle *handle;
 
-       (void) lttng_directory_handle_init_from_dirfd(&handle,
-                       data->u.rmdir.dirfd);
+       handle = lttng_directory_handle_create_from_dirfd(data->u.rmdir.dirfd);
+       if (!handle) {
+               ret_value->u.ret = -1;
+               ret_value->_errno = errno;
+               ret_value->_error = true;
+               goto end;
+       }
 
        /* Ownership of dirfd is transferred to the handle. */
        data->u.rmdir.dirfd = -1;
 
        ret_value->u.ret = lttng_directory_handle_remove_subdirectory(
-                       &handle, data->u.rmdir.path);
+                       handle, data->u.rmdir.path);
        ret_value->_errno = errno;
        ret_value->_error = (ret_value->u.ret) ? true : false;
-       lttng_directory_handle_fini(&handle);
+       lttng_directory_handle_put(handle);
+end:
        return ret_value->u.ret;
 }
 
 static
 int _rmdir_recursive(struct run_as_data *data, struct run_as_ret *ret_value)
 {
-       struct lttng_directory_handle handle;
+       struct lttng_directory_handle *handle;
 
-       (void) lttng_directory_handle_init_from_dirfd(&handle,
-                       data->u.rmdir.dirfd);
+       handle = lttng_directory_handle_create_from_dirfd(data->u.rmdir.dirfd);
+       if (!handle) {
+               ret_value->u.ret = -1;
+               ret_value->_errno = errno;
+               ret_value->_error = true;
+               goto end;
+       }
 
        /* Ownership of dirfd is transferred to the handle. */
        data->u.rmdir.dirfd = -1;
 
        ret_value->u.ret = lttng_directory_handle_remove_subdirectory_recursive(
-                       &handle, data->u.rmdir.path);
+                       handle, data->u.rmdir.path, data->u.rmdir.flags);
        ret_value->_errno = errno;
        ret_value->_error = (ret_value->u.ret) ? true : false;
-       lttng_directory_handle_fini(&handle);
+       lttng_directory_handle_put(handle);
+end:
        return ret_value->u.ret;
 }
 
@@ -482,26 +507,35 @@ static
 int _rename(struct run_as_data *data, struct run_as_ret *ret_value)
 {
        const char *old_path, *new_path;
-       struct lttng_directory_handle old_handle, new_handle;
+       struct lttng_directory_handle *old_handle = NULL, *new_handle = NULL;
 
        old_path = data->u.rename.old_path;
        new_path = data->u.rename.new_path;
 
-       (void) lttng_directory_handle_init_from_dirfd(&old_handle,
+       old_handle = lttng_directory_handle_create_from_dirfd(
                        data->u.rename.dirfds[0]);
-       (void) lttng_directory_handle_init_from_dirfd(&new_handle,
+       if (!old_handle) {
+               ret_value->u.ret = -1;
+               goto end;
+       }
+       new_handle = lttng_directory_handle_create_from_dirfd(
                        data->u.rename.dirfds[1]);
+       if (!new_handle) {
+               ret_value->u.ret = -1;
+               goto end;
+       }
 
        /* Ownership of dirfds are transferred to the handles. */
        data->u.rename.dirfds[0] = data->u.rename.dirfds[1] = -1;
 
        /* Safe to call as we have transitioned to the requested uid/gid. */
        ret_value->u.ret = lttng_directory_handle_rename(
-                       &old_handle, old_path, &new_handle, new_path);
+                       old_handle, old_path, new_handle, new_path);
+end:
+       lttng_directory_handle_put(old_handle);
+       lttng_directory_handle_put(new_handle);
        ret_value->_errno = errno;
        ret_value->_error = (ret_value->u.ret) ? true : false;
-       lttng_directory_handle_fini(&old_handle);
-       lttng_directory_handle_fini(&new_handle);
        return ret_value->u.ret;
 }
 
@@ -511,15 +545,17 @@ int _extract_elf_symbol_offset(struct run_as_data *data,
                struct run_as_ret *ret_value)
 {
        int ret = 0;
+       uint64_t offset;
 
        ret_value->_error = false;
        ret = lttng_elf_get_symbol_offset(data->u.extract_elf_symbol_offset.fd,
                         data->u.extract_elf_symbol_offset.function,
-                        &ret_value->u.extract_elf_symbol_offset.offset);
+                        &offset);
        if (ret) {
                DBG("Failed to extract ELF function offset");
                ret_value->_error = true;
        }
+       ret_value->u.extract_elf_symbol_offset.offset = offset;
 
        return ret;
 }
@@ -631,9 +667,9 @@ int do_send_fds(int sock, const int *fds, unsigned int fd_count)
                        /* Return 0 as this is not a fatal error. */
                        return 0;
                }
-        }
+       }
 
-        len = lttcomm_send_fds_unix_sock(sock, fds, fd_count);
+       len = lttcomm_send_fds_unix_sock(sock, fds, fd_count);
        return len < 0 ? -1 : 0;
 }
 
@@ -659,9 +695,9 @@ int do_recv_fds(int sock, int *fds, unsigned int fd_count)
                        ERR("Invalid file descriptor received from worker (fd = %i)", fds[i]);
                        /* Return 0 as this is not a fatal error. */
                }
-        }
+       }
 end:
-        return ret;
+       return ret;
 }
 
 static
@@ -682,9 +718,9 @@ int send_fds_to_worker(const struct run_as_worker *worker,
                        ret = -1;
                        goto end;
                }
-        }
+       }
 
-        ret = do_send_fds(worker->sockpair[0], COMMAND_IN_FDS(data),
+       ret = do_send_fds(worker->sockpair[0], COMMAND_IN_FDS(data),
                        COMMAND_IN_FD_COUNT(data));
        if (ret < 0) {
                PERROR("Failed to send file descriptor to run-as worker");
@@ -794,10 +830,10 @@ static
 int handle_one_cmd(struct run_as_worker *worker)
 {
        int ret = 0;
-        struct run_as_data data = {};
-        ssize_t readlen, writelen;
-        struct run_as_ret sendret = {};
-        run_as_fct cmd;
+       struct run_as_data data = {};
+       ssize_t readlen, writelen;
+       struct run_as_ret sendret = {};
+       run_as_fct cmd;
        uid_t prev_euid;
 
        /*
@@ -1222,7 +1258,7 @@ int run_as_create_worker_no_lock(const char *procname,
                        ret = -1;
                }
                worker->sockpair[1] = -1;
-               free(worker->procname);
+               free(worker->procname);
                free(worker);
                LOG(ret ? PRINT_ERR : PRINT_DBG, "run_as worker exiting (ret = %d)", ret);
                exit(ret ? EXIT_FAILURE : EXIT_SUCCESS);
@@ -1301,7 +1337,7 @@ void run_as_destroy_worker_no_lock(void)
                if (WIFEXITED(status)) {
                        LOG(WEXITSTATUS(status) == 0 ? PRINT_DBG : PRINT_ERR,
                                        DEFAULT_RUN_AS_WORKER_NAME " terminated with status code %d",
-                                       WEXITSTATUS(status));
+                                       WEXITSTATUS(status));
                        break;
                } else if (WIFSIGNALED(status)) {
                        ERR(DEFAULT_RUN_AS_WORKER_NAME " was killed by signal %d",
@@ -1443,7 +1479,7 @@ error:
 
 LTTNG_HIDDEN
 int run_as_open(const char *path, int flags, mode_t mode, uid_t uid,
-                gid_t gid)
+               gid_t gid)
 {
        return run_as_openat(AT_FDCWD, path, flags, mode, uid, gid);
 }
@@ -1453,8 +1489,8 @@ int run_as_openat(int dirfd, const char *path, int flags, mode_t mode,
                uid_t uid, gid_t gid)
 {
        int ret;
-        struct run_as_data data = {};
-        struct run_as_ret run_as_ret = {};
+       struct run_as_data data = {};
+       struct run_as_ret run_as_ret = {};
 
        DBG3("openat() fd = %d%s, path = %s, flags = %X, mode = %d, uid %d, gid %d",
                        dirfd, dirfd == AT_FDCWD ? " (AT_FDCWD)" : "",
@@ -1537,13 +1573,13 @@ error:
 }
 
 LTTNG_HIDDEN
-int run_as_rmdir_recursive(const char *path, uid_t uid, gid_t gid)
+int run_as_rmdir_recursive(const char *path, uid_t uid, gid_t gid, int flags)
 {
-       return run_as_rmdirat_recursive(AT_FDCWD, path, uid, gid);
+       return run_as_rmdirat_recursive(AT_FDCWD, path, uid, gid, flags);
 }
 
 LTTNG_HIDDEN
-int run_as_rmdirat_recursive(int dirfd, const char *path, uid_t uid, gid_t gid)
+int run_as_rmdirat_recursive(int dirfd, const char *path, uid_t uid, gid_t gid, int flags)
 {
        int ret;
        struct run_as_data data = {};
@@ -1558,6 +1594,7 @@ int run_as_rmdirat_recursive(int dirfd, const char *path, uid_t uid, gid_t gid)
                goto error;
        }
        data.u.rmdir.dirfd = dirfd;
+       data.u.rmdir.flags = flags;
        run_as(dirfd == AT_FDCWD ? RUN_AS_RMDIR_RECURSIVE : RUN_AS_RMDIRAT_RECURSIVE,
                        &data, &run_as_ret, uid, gid);
        errno = run_as_ret._errno;
@@ -1613,7 +1650,7 @@ int run_as_extract_elf_symbol_offset(int fd, const char* function,
 {
        int ret;
        struct run_as_data data = {};
-        struct run_as_ret run_as_ret = {};
+       struct run_as_ret run_as_ret = {};
 
        DBG3("extract_elf_symbol_offset() on fd=%d and function=%s "
                        "with for uid %d and gid %d", fd, function,
This page took 0.029585 seconds and 5 git commands to generate.