|
|
@@ -2,6 +2,7 @@
|
|
|
/*
|
|
|
* Auto-group scheduling implementation:
|
|
|
*/
|
|
|
+#include <linux/nospec.h>
|
|
|
#include "sched.h"
|
|
|
|
|
|
unsigned int __read_mostly sysctl_sched_autogroup_enabled = 1;
|
|
|
@@ -209,7 +210,7 @@ int proc_sched_autogroup_set_nice(struct task_struct *p, int nice)
|
|
|
static unsigned long next = INITIAL_JIFFIES;
|
|
|
struct autogroup *ag;
|
|
|
unsigned long shares;
|
|
|
- int err;
|
|
|
+ int err, idx;
|
|
|
|
|
|
if (nice < MIN_NICE || nice > MAX_NICE)
|
|
|
return -EINVAL;
|
|
|
@@ -227,7 +228,9 @@ int proc_sched_autogroup_set_nice(struct task_struct *p, int nice)
|
|
|
|
|
|
next = HZ / 10 + jiffies;
|
|
|
ag = autogroup_task_get(p);
|
|
|
- shares = scale_load(sched_prio_to_weight[nice + 20]);
|
|
|
+
|
|
|
+ idx = array_index_nospec(nice + 20, 40);
|
|
|
+ shares = scale_load(sched_prio_to_weight[idx]);
|
|
|
|
|
|
down_write(&ag->lock);
|
|
|
err = sched_group_set_shares(ag->tg, shares);
|