diff options
-rw-r--r-- | execute.c | 27 | ||||
-rw-r--r-- | execute.h | 1 | ||||
-rw-r--r-- | load-fragment.c | 3 |
3 files changed, 20 insertions, 11 deletions
@@ -30,6 +30,8 @@ #include <sys/un.h> #include <sys/prctl.h> #include <linux/sched.h> +#include <sys/types.h> +#include <sys/stat.h> #include "execute.h" #include "strv.h" @@ -130,7 +132,7 @@ static int shift_fds(int fds[], unsigned n_fds) { return 0; } -static int flags_fds(int fds[], unsigned n_fds) { +static int flags_fds(int fds[], unsigned n_fds, bool nonblock) { unsigned i; if (n_fds <= 0) @@ -138,7 +140,7 @@ static int flags_fds(int fds[], unsigned n_fds) { assert(fds); - /* Drops O_NONBLOCK and FD_CLOEXEC from the file flags */ + /* Drops/Sets O_NONBLOCK and FD_CLOEXEC from the file flags */ for (i = 0; i < n_fds; i++) { int flags; @@ -146,17 +148,20 @@ static int flags_fds(int fds[], unsigned n_fds) { if ((flags = fcntl(fds[i], F_GETFL, 0)) < 0) return -errno; - /* Since we are at it, let's make sure that nobody - * forgot setting O_NONBLOCK for all our fds */ + if (nonblock) + flags |= O_NONBLOCK; + else + flags &= ~O_NONBLOCK; - if (fcntl(fds[i], F_SETFL, flags &~O_NONBLOCK) < 0) + if (fcntl(fds[i], F_SETFL, flags) < 0) return -errno; + /* We unconditionally drop FD_CLOEXEC from the fds, + * since after all we want to pass these fds to our + * children */ if ((flags = fcntl(fds[i], F_GETFD, 0)) < 0) return -errno; - /* Also make sure nobody forgot O_CLOEXEC for all our - * fds */ if (fcntl(fds[i], F_SETFD, flags &~FD_CLOEXEC) < 0) return -errno; } @@ -393,7 +398,7 @@ int exec_spawn(const ExecCommand *command, const ExecContext *context, int *fds, if (close_fds(fds, n_fds) < 0 || shift_fds(fds, n_fds) < 0 || - flags_fds(fds, n_fds) < 0) { + flags_fds(fds, n_fds, context->non_blocking) < 0) { r = EXIT_FDS; goto fail; } @@ -550,10 +555,12 @@ void exec_context_dump(ExecContext *c, FILE* f, const char *prefix) { fprintf(f, "%sUMask: %04o\n" "%sWorkingDirectory: %s\n" - "%sRootDirectory: %s\n", + "%sRootDirectory: %s\n" + "%sNonBlocking: %s\n", prefix, c->umask, prefix, c->working_directory ? c->working_directory : "/", - prefix, c->root_directory ? c->root_directory : "/"); + prefix, c->root_directory ? c->root_directory : "/", + prefix, yes_no(c->non_blocking)); if (c->environment) for (e = c->environment; *e; e++) @@ -90,6 +90,7 @@ struct ExecContext { bool timer_slack_ns_set:1; bool cpu_sched_reset_on_fork; + bool non_blocking; ExecInput input; ExecOutput output; diff --git a/load-fragment.c b/load-fragment.c index 10972e43cd..ab7f830a5d 100644 --- a/load-fragment.c +++ b/load-fragment.c @@ -1071,7 +1071,8 @@ static int load_from_path(Unit *u, const char *path) { { "LimitMSGQUEUE", config_parse_limit, &(context).rlimit[RLIMIT_MSGQUEUE], section }, \ { "LimitNICE", config_parse_limit, &(context).rlimit[RLIMIT_NICE], section }, \ { "LimitRTPRIO", config_parse_limit, &(context).rlimit[RLIMIT_RTPRIO], section }, \ - { "LimitRTTIME", config_parse_limit, &(context).rlimit[RLIMIT_RTTIME], section } + { "LimitRTTIME", config_parse_limit, &(context).rlimit[RLIMIT_RTTIME], section }, \ + { "NonBlocking", config_parse_bool, &(context).non_blocking, section } const ConfigItem items[] = { { "Names", config_parse_names, u, "Meta" }, |