Skip to content

Commit

Permalink
core: in execute, Never fail setting Nice priority
Browse files Browse the repository at this point in the history
Instead, push to the closest possible Nice priority setting.

Replaces: #11397
  • Loading branch information
xnox authored and poettering committed Jan 2, 2020
1 parent 6bbeef2 commit 3909020
Show file tree
Hide file tree
Showing 3 changed files with 64 additions and 5 deletions.
57 changes: 57 additions & 0 deletions src/basic/process-util.c
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@
#include "alloc-util.h"
#include "architecture.h"
#include "env-util.h"
#include "errno-util.h"
#include "escape.h"
#include "fd-util.h"
#include "fileio.h"
Expand Down Expand Up @@ -1527,6 +1528,62 @@ int pidfd_get_pid(int fd, pid_t *ret) {
return parse_pid(p, ret);
}

static int rlimit_to_nice(rlim_t limit) {
if (limit <= 1)
return PRIO_MAX-1; /* i.e. 19 */

if (limit >= -PRIO_MIN + PRIO_MAX)
return PRIO_MIN; /* i.e. -20 */

return PRIO_MAX - (int) limit;
}

int setpriority_closest(int priority) {
int current, limit, saved_errno;
struct rlimit highest;

/* Try to set requested nice level */
if (setpriority(PRIO_PROCESS, 0, priority) >= 0)
return 1;

/* Permission failed */
saved_errno = -errno;
if (!ERRNO_IS_PRIVILEGE(saved_errno))
return saved_errno;

errno = 0;
current = getpriority(PRIO_PROCESS, 0);
if (errno != 0)
return -errno;

if (priority == current)
return 1;

/* Hmm, we'd expect that raising the nice level from our status quo would always work. If it doesn't,
* then the whole setpriority() system call is blocked to us, hence let's propagate the error
* right-away */
if (priority > current)
return saved_errno;

if (getrlimit(RLIMIT_NICE, &highest) < 0)
return -errno;

limit = rlimit_to_nice(highest.rlim_cur);

/* We are already less nice than limit allows us */
if (current < limit) {
log_debug("Cannot raise nice level, permissions and the resource limit do not allow it.");
return 0;
}

/* Push to the allowed limit */
if (setpriority(PRIO_PROCESS, 0, limit) < 0)
return -errno;

log_debug("Cannot set requested nice level (%i), used next best (%i).", priority, limit);
return 0;
}

static const char *const ioprio_class_table[] = {
[IOPRIO_CLASS_NONE] = "none",
[IOPRIO_CLASS_RT] = "realtime",
Expand Down
2 changes: 2 additions & 0 deletions src/basic/process-util.h
Original file line number Diff line number Diff line change
Expand Up @@ -200,3 +200,5 @@ assert_cc(TASKS_MAX <= (unsigned long) PID_T_MAX);
})

int pidfd_get_pid(int fd, pid_t *ret);

int setpriority_closest(int priority);
10 changes: 5 additions & 5 deletions src/core/execute.c
Original file line number Diff line number Diff line change
Expand Up @@ -3236,11 +3236,11 @@ static int exec_child(
}
}

if (context->nice_set)
if (setpriority(PRIO_PROCESS, 0, context->nice) < 0) {
*exit_status = EXIT_NICE;
return log_unit_error_errno(unit, errno, "Failed to set up process scheduling priority (nice level): %m");
}
if (context->nice_set) {
r = setpriority_closest(context->nice);
if (r < 0)
return log_unit_error_errno(unit, r, "Failed to set up process scheduling priority (nice level): %m");
}

if (context->cpu_sched_set) {
struct sched_param param = {
Expand Down

0 comments on commit 3909020

Please sign in to comment.