Index: linux-2.6.8-rc3/include/linux/sched.h =================================================================== --- linux-2.6.8-rc3.orig/include/linux/sched.h 2004-08-08 22:53:54.704741504 +1000 +++ linux-2.6.8-rc3/include/linux/sched.h 2004-08-08 22:57:47.139654619 +1000 @@ -127,6 +127,14 @@ #define SCHED_FIFO 1 #define SCHED_RR 2 +#define SCHED_MIN 0 +#define SCHED_MAX 2 + +#define SCHED_RANGE(policy) ((policy) >= SCHED_MIN && \ + (policy) <= SCHED_MAX) +#define SCHED_RT(policy) ((policy) == SCHED_FIFO || \ + (policy) == SCHED_RR) + struct sched_param { int sched_priority; }; Index: linux-2.6.8-rc3/kernel/sched.c =================================================================== --- linux-2.6.8-rc3.orig/kernel/sched.c 2004-08-08 22:55:48.012663027 +1000 +++ linux-2.6.8-rc3/kernel/sched.c 2004-08-08 22:57:47.174648971 +1000 @@ -2377,7 +2377,7 @@ BUG_ON(task_queued(p)); p->policy = policy; p->rt_priority = prio; - if (policy != SCHED_NORMAL) + if (SCHED_RT(policy)) p->prio = MAX_USER_RT_PRIO-1 - p->rt_priority; else p->prio = p->static_prio; @@ -2423,9 +2423,8 @@ policy = p->policy; else { retval = -EINVAL; - if (policy != SCHED_FIFO && policy != SCHED_RR && - policy != SCHED_NORMAL) - goto out_unlock; + if (!SCHED_RANGE(policy)) + goto out_unlock; } /* @@ -2435,12 +2434,11 @@ retval = -EINVAL; if (lp.sched_priority < 0 || lp.sched_priority > MAX_USER_RT_PRIO-1) goto out_unlock; - if ((policy == SCHED_NORMAL) != (lp.sched_priority == 0)) - goto out_unlock; + if (!SCHED_RT(policy) != (lp.sched_priority == 0)) + goto out_unlock; retval = -EPERM; - if ((policy == SCHED_FIFO || policy == SCHED_RR) && - !capable(CAP_SYS_NICE)) + if (SCHED_RT(policy) && !capable(CAP_SYS_NICE)) goto out_unlock; if ((current->euid != p->euid) && (current->euid != p->uid) && !capable(CAP_SYS_NICE))