]> git.proxmox.com Git - mirror_lxc.git/blobdiff - src/lxc/attach.c
attach: don't close stdout of getent
[mirror_lxc.git] / src / lxc / attach.c
index 4c01dea1802b15ac361761a5f566d248e816c46e..80c41fe263a164a9ed628fe87163c5b833f99e75 100644 (file)
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  */
 
-#define _GNU_SOURCE
+#ifndef _GNU_SOURCE
+#define _GNU_SOURCE 1
+#endif
 #include <errno.h>
 #include <fcntl.h>
 #include <grp.h>
 #include <linux/unistd.h>
 #include <pwd.h>
+#include <pthread.h>
 #include <signal.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include "lxcseccomp.h"
 #include "macro.h"
 #include "mainloop.h"
+#include "memory_utils.h"
 #include "namespace.h"
+#include "raw_syscalls.h"
+#include "syscall_wrappers.h"
 #include "terminal.h"
 #include "utils.h"
 
@@ -71,31 +77,28 @@ static lxc_attach_options_t attach_static_default_options = LXC_ATTACH_OPTIONS_D
 
 static struct lxc_proc_context_info *lxc_proc_get_context_info(pid_t pid)
 {
+       __do_free char *line = NULL;
+       __do_fclose FILE *proc_file = NULL;
        int ret;
        bool found;
-       FILE *proc_file;
        char proc_fn[LXC_PROC_STATUS_LEN];
+       struct lxc_proc_context_info *info;
        size_t line_bufsz = 0;
-       char *line = NULL;
-       struct lxc_proc_context_info *info = NULL;
 
        /* Read capabilities. */
        ret = snprintf(proc_fn, LXC_PROC_STATUS_LEN, "/proc/%d/status", pid);
        if (ret < 0 || ret >= LXC_PROC_STATUS_LEN)
-               goto on_error;
+               return NULL;
 
        proc_file = fopen(proc_fn, "r");
        if (!proc_file) {
-               SYSERROR("Could not open %s", proc_fn);
-               goto on_error;
+               SYSERROR("Failed to open %s", proc_fn);
+               return NULL;
        }
 
        info = calloc(1, sizeof(*info));
-       if (!info) {
-               SYSERROR("Could not allocate memory");
-               fclose(proc_file);
+       if (!info)
                return NULL;
-       }
 
        found = false;
 
@@ -107,13 +110,10 @@ static struct lxc_proc_context_info *lxc_proc_get_context_info(pid_t pid)
                }
        }
 
-       free(line);
-       fclose(proc_file);
-
        if (!found) {
-               ERROR("Could not read capability bounding set from %s",
-                     proc_fn);
-               goto on_error;
+               ERROR("Could not read capability bounding set from %s", proc_fn);
+               free(info);
+               return NULL;
        }
 
        info->lsm_label = lsm_process_label_get(pid);
@@ -121,22 +121,12 @@ static struct lxc_proc_context_info *lxc_proc_get_context_info(pid_t pid)
        memset(info->ns_fd, -1, sizeof(int) * LXC_NS_MAX);
 
        return info;
-
-on_error:
-       free(info);
-       return NULL;
 }
 
 static inline void lxc_proc_close_ns_fd(struct lxc_proc_context_info *ctx)
 {
-       int i;
-
-       for (i = 0; i < LXC_NS_MAX; i++) {
-               if (ctx->ns_fd[i] < 0)
-                       continue;
-
-               close(ctx->ns_fd[i]);
-               ctx->ns_fd[i] = -EBADF;
+       for (int i = 0; i < LXC_NS_MAX; i++) {
+               __do_close_prot_errno int fd ATTR_UNUSED = move_fd(ctx->ns_fd[i]);
        }
 }
 
@@ -167,8 +157,8 @@ static void lxc_proc_put_context_info(struct lxc_proc_context_info *ctx)
  */
 static int in_same_namespace(pid_t pid1, pid_t pid2, const char *ns)
 {
-       int ns_fd1 = -1, ns_fd2 = -1, ret = -1;
-       int saved_errno;
+       __do_close_prot_errno int ns_fd1 = -1, ns_fd2 = -1;
+       int ret = -1;
        struct stat ns_st1, ns_st2;
 
        ns_fd1 = lxc_preserve_ns(pid1, ns);
@@ -179,42 +169,27 @@ static int in_same_namespace(pid_t pid1, pid_t pid2, const char *ns)
                if (errno == ENOENT)
                        return -EINVAL;
 
-               goto out;
+               return -1;
        }
 
        ns_fd2 = lxc_preserve_ns(pid2, ns);
        if (ns_fd2 < 0)
-               goto out;
+               return -1;
 
        ret = fstat(ns_fd1, &ns_st1);
        if (ret < 0)
-               goto out;
+               return -1;
 
        ret = fstat(ns_fd2, &ns_st2);
        if (ret < 0)
-               goto out;
+               return -1;
 
        /* processes are in the same namespace */
-       if ((ns_st1.st_dev == ns_st2.st_dev ) && (ns_st1.st_ino == ns_st2.st_ino)) {
-               ret = -EINVAL;
-               goto out;
-       }
+       if ((ns_st1.st_dev == ns_st2.st_dev) && (ns_st1.st_ino == ns_st2.st_ino))
+               return -EINVAL;
 
        /* processes are in different namespaces */
-       ret = ns_fd2;
-       ns_fd2 = -1;
-
-out:
-       saved_errno = errno;
-
-       if (ns_fd1 >= 0)
-               close(ns_fd1);
-
-       if (ns_fd2 >= 0)
-               close(ns_fd2);
-
-       errno = saved_errno;
-       return ret;
+       return move_fd(ns_fd2);
 }
 
 static int lxc_attach_to_ns(pid_t pid, struct lxc_proc_context_info *ctx)
@@ -238,7 +213,7 @@ static int lxc_attach_to_ns(pid_t pid, struct lxc_proc_context_info *ctx)
        return 0;
 }
 
-static int lxc_attach_remount_sys_proc(void)
+int lxc_attach_remount_sys_proc(void)
 {
        int ret;
 
@@ -435,13 +410,14 @@ static int lxc_attach_set_environment(struct lxc_proc_context_info *init_ctx,
 
 static char *lxc_attach_getpwshell(uid_t uid)
 {
+       __do_free char *line = NULL;
+       __do_fclose FILE *pipe_f = NULL;
        int fd, ret;
        pid_t pid;
        int pipes[2];
-       FILE *pipe_f;
        bool found = false;
        size_t line_bufsz = 0;
-       char *line = NULL, *result = NULL;
+       char *result = NULL;
 
        /* We need to fork off a process that runs the getent program, and we
         * need to capture its output, so we use a pipe for that purpose.
@@ -483,7 +459,7 @@ static char *lxc_attach_getpwshell(uid_t uid)
                        close(STDERR_FILENO);
                } else {
                        (void)dup3(fd, STDIN_FILENO, O_CLOEXEC);
-                       (void)dup3(fd, STDOUT_FILENO, O_CLOEXEC);
+                       (void)dup3(fd, STDERR_FILENO, O_CLOEXEC);
                        close(fd);
                }
 
@@ -562,9 +538,6 @@ static char *lxc_attach_getpwshell(uid_t uid)
                found = true;
        }
 
-       free(line);
-       fclose(pipe_f);
-
        ret = wait_for_pid(pid);
        if (ret < 0) {
                free(result);
@@ -581,10 +554,10 @@ static char *lxc_attach_getpwshell(uid_t uid)
 
 static void lxc_attach_get_init_uidgid(uid_t *init_uid, gid_t *init_gid)
 {
-       FILE *proc_file;
+       __do_free char *line = NULL;
+       __do_fclose FILE *proc_file = NULL;
        char proc_fn[LXC_PROC_STATUS_LEN];
        int ret;
-       char *line = NULL;
        size_t line_bufsz = 0;
        long value = -1;
        uid_t uid = (uid_t)-1;
@@ -615,9 +588,6 @@ static void lxc_attach_get_init_uidgid(uid_t *init_uid, gid_t *init_gid)
                        break;
        }
 
-       fclose(proc_file);
-       free(line);
-
        /* Only override arguments if we found something. */
        if (uid != (uid_t)-1)
                *init_uid = uid;
@@ -632,14 +602,14 @@ static void lxc_attach_get_init_uidgid(uid_t *init_uid, gid_t *init_gid)
 
 static bool fetch_seccomp(struct lxc_container *c, lxc_attach_options_t *options)
 {
+       __do_free char *path = NULL;
        int ret;
        bool bret;
-       char *path;
 
        if (!(options->namespaces & CLONE_NEWNS) ||
            !(options->attach_flags & LXC_ATTACH_LSM)) {
-               free(c->lxc_conf->seccomp);
-               c->lxc_conf->seccomp = NULL;
+               free(c->lxc_conf->seccomp.seccomp);
+               c->lxc_conf->seccomp.seccomp = NULL;
                return true;
        }
 
@@ -662,7 +632,6 @@ static bool fetch_seccomp(struct lxc_container *c, lxc_attach_options_t *options
 
        /* Copy the value into the new lxc_conf. */
        bret = c->set_config_item(c, "lxc.seccomp.profile", path);
-       free(path);
        if (!bret)
                return false;
 
@@ -679,8 +648,7 @@ static bool fetch_seccomp(struct lxc_container *c, lxc_attach_options_t *options
 
 static bool no_new_privs(struct lxc_container *c, lxc_attach_options_t *options)
 {
-       bool bret;
-       char *val;
+       __do_free char *val = NULL;
 
        /* Remove current setting. */
        if (!c->set_config_item(c, "lxc.no_new_privs", "")) {
@@ -696,24 +664,18 @@ static bool no_new_privs(struct lxc_container *c, lxc_attach_options_t *options)
        }
 
        /* Set currently active setting. */
-       bret = c->set_config_item(c, "lxc.no_new_privs", val);
-       free(val);
-       return bret;
+       return c->set_config_item(c, "lxc.no_new_privs", val);
 }
 
 static signed long get_personality(const char *name, const char *lxcpath)
 {
-       char *p;
-       signed long ret;
+       __do_free char *p = NULL;
 
        p = lxc_cmd_get_config_item(name, "lxc.arch", lxcpath);
        if (!p)
                return -1;
 
-       ret = lxc_config_parse_arch(p);
-       free(p);
-
-       return ret;
+       return lxc_config_parse_arch(p);
 }
 
 struct attach_clone_payload {
@@ -727,15 +689,8 @@ struct attach_clone_payload {
 
 static void lxc_put_attach_clone_payload(struct attach_clone_payload *p)
 {
-       if (p->ipc_socket >= 0) {
-               close(p->ipc_socket);
-               p->ipc_socket = -EBADF;
-       }
-
-       if (p->terminal_slave_fd >= 0) {
-               close(p->terminal_slave_fd);
-               p->terminal_slave_fd = -EBADF;
-       }
+       __do_close_prot_errno int ipc_socket ATTR_UNUSED = p->ipc_socket;
+       __do_close_prot_errno int terminal_slave_fd ATTR_UNUSED = p->terminal_slave_fd;
 
        if (p->init_ctx) {
                lxc_proc_put_context_info(p->init_ctx);
@@ -745,7 +700,7 @@ static void lxc_put_attach_clone_payload(struct attach_clone_payload *p)
 
 static int attach_child_main(struct attach_clone_payload *payload)
 {
-       int fd, lsm_fd, ret;
+       int lsm_fd, ret;
        uid_t new_uid;
        gid_t new_gid;
        uid_t ns_root_uid = 0;
@@ -897,12 +852,18 @@ static int attach_child_main(struct attach_clone_payload *payload)
        }
 
        if (init_ctx->container && init_ctx->container->lxc_conf &&
-           init_ctx->container->lxc_conf->seccomp) {
-               ret = lxc_seccomp_load(init_ctx->container->lxc_conf);
+           init_ctx->container->lxc_conf->seccomp.seccomp) {
+               struct lxc_conf *conf = init_ctx->container->lxc_conf;
+
+               ret = lxc_seccomp_load(conf);
                if (ret < 0)
                        goto on_error;
 
                TRACE("Loaded seccomp profile");
+
+               ret = lxc_seccomp_send_notifier_fd(&conf->seccomp, payload->ipc_socket);
+               if (ret < 0)
+                       goto on_error;
        }
 
        close(payload->ipc_socket);
@@ -938,10 +899,11 @@ static int attach_child_main(struct attach_clone_payload *payload)
        if (options->stderr_fd > STDERR_FILENO)
                close(options->stderr_fd);
 
-       /* Try to remove FD_CLOEXEC flag from stdin/stdout/stderr, but also
+       /*
+        * Try to remove FD_CLOEXEC flag from stdin/stdout/stderr, but also
         * here, ignore errors.
         */
-       for (fd = STDIN_FILENO; fd <= STDERR_FILENO; fd++) {
+       for (int fd = STDIN_FILENO; fd <= STDERR_FILENO; fd++) {
                ret = fd_cloexec(fd, false);
                if (ret < 0) {
                        SYSERROR("Failed to clear FD_CLOEXEC from file descriptor %d", fd);
@@ -1028,43 +990,27 @@ static int lxc_attach_terminal_mainloop_init(struct lxc_terminal *terminal,
 
 static inline void lxc_attach_terminal_close_master(struct lxc_terminal *terminal)
 {
-       if (terminal->master < 0)
-               return;
-
-       close(terminal->master);
-       terminal->master = -EBADF;
+       close_prot_errno_disarm(terminal->master);
 }
 
 static inline void lxc_attach_terminal_close_slave(struct lxc_terminal *terminal)
 {
-       if (terminal->slave < 0)
-               return;
-
-       close(terminal->slave);
-       terminal->slave = -EBADF;
+       close_prot_errno_disarm(terminal->slave);
 }
 
 static inline void lxc_attach_terminal_close_peer(struct lxc_terminal *terminal)
 {
-       if (terminal->peer < 0)
-               return;
-
-       close(terminal->peer);
-       terminal->peer = -EBADF;
+       close_prot_errno_disarm(terminal->peer);
 }
 
 static inline void lxc_attach_terminal_close_log(struct lxc_terminal *terminal)
 {
-       if (terminal->log_fd < 0)
-               return;
-
-       close(terminal->log_fd);
-       terminal->log_fd = -EBADF;
+       close_prot_errno_disarm(terminal->log_fd);
 }
 
-int lxc_attach(const char *name, const char *lxcpath,
-              lxc_attach_exec_t exec_function, void *exec_payload,
-              lxc_attach_options_t *options, pid_t *attached_process)
+int lxc_attach(struct lxc_container *container, lxc_attach_exec_t exec_function,
+              void *exec_payload, lxc_attach_options_t *options,
+              pid_t *attached_process)
 {
        int i, ret, status;
        int ipc_sockets[2];
@@ -1074,6 +1020,7 @@ int lxc_attach(const char *name, const char *lxcpath,
        struct lxc_proc_context_info *init_ctx;
        struct lxc_terminal terminal;
        struct lxc_conf *conf;
+       char *name, *lxcpath;
        struct attach_clone_payload payload = {0};
 
        ret = access("/proc/self/ns", X_OK);
@@ -1082,21 +1029,34 @@ int lxc_attach(const char *name, const char *lxcpath,
                return -1;
        }
 
+       if (!container)
+               return minus_one_set_errno(EINVAL);
+
+       if (!lxc_container_get(container))
+               return minus_one_set_errno(EINVAL);
+
+       name = container->name;
+       lxcpath = container->config_path;
+
        if (!options)
                options = &attach_static_default_options;
 
        init_pid = lxc_cmd_get_init_pid(name, lxcpath);
        if (init_pid < 0) {
                ERROR("Failed to get init pid");
+               lxc_container_put(container);
                return -1;
        }
 
        init_ctx = lxc_proc_get_context_info(init_pid);
        if (!init_ctx) {
                ERROR("Failed to get context of init process: %ld", (long)init_pid);
+               lxc_container_put(container);
                return -1;
        }
 
+       init_ctx->container = container;
+
        personality = get_personality(name, lxcpath);
        if (init_ctx->personality < 0) {
                ERROR("Failed to get personality of the container");
@@ -1105,12 +1065,6 @@ int lxc_attach(const char *name, const char *lxcpath,
        }
        init_ctx->personality = personality;
 
-       init_ctx->container = lxc_container_new(name, lxcpath);
-       if (!init_ctx->container) {
-               lxc_proc_put_context_info(init_ctx);
-               return -1;
-       }
-
        if (!init_ctx->container->lxc_conf) {
                init_ctx->container->lxc_conf = lxc_conf_init();
                if (!init_ctx->container->lxc_conf) {
@@ -1214,7 +1168,7 @@ int lxc_attach(const char *name, const char *lxcpath,
         * just fork()s away without exec'ing directly after, the socket fd will
         * exist in the forked process from the other thread and any close() in
         * our own child process will not really cause the socket to close
-        * properly, potentiall causing the parent to hang.
+        * properly, potentially causing the parent to hang.
         *
         * For this reason, while IPC is still active, we have to use shutdown()
         * if the child exits prematurely in order to signal that the socket is
@@ -1346,10 +1300,10 @@ int lxc_attach(const char *name, const char *lxcpath,
                if ((options->namespaces & CLONE_NEWNS) &&
                    (options->attach_flags & LXC_ATTACH_LSM) &&
                    init_ctx->lsm_label) {
-                       int ret = -1;
                        int labelfd;
                        bool on_exec;
 
+                       ret = -1;
                        on_exec = options->attach_flags & LXC_ATTACH_LSM_EXEC ? true : false;
                        labelfd = lsm_process_label_fd_get(attached_pid, on_exec);
                        if (labelfd < 0)
@@ -1371,6 +1325,16 @@ int lxc_attach(const char *name, const char *lxcpath,
                        TRACE("Sent LSM label file descriptor %d to child", labelfd);
                }
 
+               if (conf && conf->seccomp.seccomp) {
+                       ret = lxc_seccomp_recv_notifier_fd(&conf->seccomp, ipc_sockets[0]);
+                       if (ret < 0)
+                               goto close_mainloop;
+
+                       ret = lxc_seccomp_add_notifier(name, lxcpath, &conf->seccomp);
+                       if (ret < 0)
+                               goto close_mainloop;
+               }
+
                /* We're done, the child process should now execute whatever it
                 * is that the user requested. The parent can now track it with
                 * waitpid() or similar.
@@ -1470,7 +1434,7 @@ int lxc_attach(const char *name, const char *lxcpath,
        payload.exec_function = exec_function;
        payload.exec_payload = exec_payload;
 
-       pid = lxc_raw_clone(CLONE_PARENT);
+       pid = lxc_raw_clone(CLONE_PARENT, NULL);
        if (pid < 0) {
                SYSERROR("Failed to clone attached process");
                shutdown(ipc_sockets[1], SHUT_RDWR);
@@ -1479,6 +1443,15 @@ int lxc_attach(const char *name, const char *lxcpath,
        }
 
        if (pid == 0) {
+               if (options->attach_flags & LXC_ATTACH_TERMINAL) {
+                       ret = pthread_sigmask(SIG_SETMASK,
+                                             &terminal.tty_state->oldmask, NULL);
+                       if (ret < 0) {
+                               SYSERROR("Failed to reset signal mask");
+                               _exit(EXIT_FAILURE);
+                       }
+               }
+
                ret = attach_child_main(&payload);
                if (ret < 0)
                        ERROR("Failed to exec");
@@ -1507,7 +1480,7 @@ int lxc_attach(const char *name, const char *lxcpath,
 
        /* The rest is in the hands of the initial and the attached process. */
        lxc_proc_put_context_info(init_ctx);
-       _exit(0);
+       _exit(EXIT_SUCCESS);
 }
 
 int lxc_attach_run_command(void *payload)
@@ -1533,11 +1506,11 @@ int lxc_attach_run_command(void *payload)
 
 int lxc_attach_run_shell(void* payload)
 {
+       __do_free char *buf = NULL;
        uid_t uid;
        struct passwd pwent;
        struct passwd *pwentp = NULL;
        char *user_shell;
-       char *buf;
        size_t bufsize;
        int ret;
 
@@ -1584,6 +1557,5 @@ int lxc_attach_run_shell(void* payload)
        if (!pwentp)
                free(user_shell);
 
-       free(buf);
        return -1;
 }