Clean-up: replace space by tabs
[lttng-tools.git] / src / common / runas.c
index 4bdb43af38da4fbf822944237caa18cfcbe30926..e30871c5c8fc4f8b4fb30d42c64fdd60292b3034 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
@@ -351,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;
 }
 
@@ -375,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;
 }
 
@@ -398,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) {
@@ -418,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, data->u.rmdir.flags);
+                       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;
 }
 
@@ -483,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;
 }
 
@@ -512,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;
 }
@@ -632,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;
 }
 
This page took 0.026656 seconds and 4 git commands to generate.