From fbadf04511389c4a0687ba5e9baf0ecebdbb07f1 Mon Sep 17 00:00:00 2001 From: Lennart Poettering Date: Mon, 23 Dec 2013 19:10:11 +0100 Subject: bus: when getting a kdbus connection into a container wait first for child, then read message There's no EOF generated for AF_UNIX/SOCK_DGRAM sockets, hence let's wait for the child first to see if it succeeded, only then read the socket. --- src/libsystemd-bus/bus-container.c | 20 ++++++++++---------- src/machine/machinectl.c | 18 +++++++++--------- src/shared/logs-show.c | 8 ++++---- 3 files changed, 23 insertions(+), 23 deletions(-) diff --git a/src/libsystemd-bus/bus-container.c b/src/libsystemd-bus/bus-container.c index 1989afa82a..d330363b6f 100644 --- a/src/libsystemd-bus/bus-container.c +++ b/src/libsystemd-bus/bus-container.c @@ -205,6 +205,16 @@ int bus_container_connect_kernel(sd_bus *b) { close_nointr_nofail(pair[1]); pair[1] = -1; + r = wait_for_terminate(child, &si); + if (r < 0) + return r; + + if (si.si_code != CLD_EXITED) + return -EIO; + + if (si.si_status != EXIT_SUCCESS) + return -EIO; + if (recvmsg(pair[0], &mh, MSG_NOSIGNAL|MSG_CMSG_CLOEXEC) < 0) return -errno; @@ -224,16 +234,6 @@ int bus_container_connect_kernel(sd_bus *b) { fd = fds[0]; } - r = wait_for_terminate(child, &si); - if (r < 0) - return r; - - if (si.si_code != CLD_EXITED) - return -EIO; - - if (si.si_status != EXIT_SUCCESS) - return -EIO; - b->input_fd = b->output_fd = fd; fd = -1; diff --git a/src/machine/machinectl.c b/src/machine/machinectl.c index a2af87c17d..ab9060ab99 100644 --- a/src/machine/machinectl.c +++ b/src/machine/machinectl.c @@ -451,6 +451,15 @@ static int openpt_in_namespace(pid_t pid, int flags) { close_nointr_nofail(pair[1]); pair[1] = -1; + r = wait_for_terminate(child, &si); + if (r < 0 || si.si_code != CLD_EXITED || si.si_status != EXIT_SUCCESS || master < 0) { + + if (master >= 0) + close_nointr_nofail(master); + + return r < 0 ? r : -EIO; + } + if (recvmsg(pair[0], &mh, MSG_NOSIGNAL|MSG_CMSG_CLOEXEC) < 0) return -errno; @@ -470,15 +479,6 @@ static int openpt_in_namespace(pid_t pid, int flags) { master = fds[0]; } - r = wait_for_terminate(child, &si); - if (r < 0 || si.si_code != CLD_EXITED || si.si_status != EXIT_SUCCESS || master < 0) { - - if (master >= 0) - close_nointr_nofail(master); - - return r < 0 ? r : -EIO; - } - return master; } diff --git a/src/shared/logs-show.c b/src/shared/logs-show.c index 0f27c4ee41..61c3652bd5 100644 --- a/src/shared/logs-show.c +++ b/src/shared/logs-show.c @@ -1174,14 +1174,14 @@ static int get_boot_id_for_machine(const char *machine, sd_id128_t *boot_id) { close_nointr_nofail(pair[1]); pair[1] = -1; - k = recv(pair[0], buf, 36, 0); - if (k != 36) - return -EIO; - r = wait_for_terminate(child, &si); if (r < 0 || si.si_code != CLD_EXITED || si.si_status != EXIT_SUCCESS) return r < 0 ? r : -EIO; + k = recv(pair[0], buf, 36, 0); + if (k != 36) + return -EIO; + buf[36] = 0; r = sd_id128_from_string(buf, boot_id); if (r < 0) -- cgit v1.2.3-54-g00ecf