Index: linux-2.6.8.1-ck8/include/linux/sched.h =================================================================== --- linux-2.6.8.1-ck8.orig/include/linux/sched.h 2004-09-17 00:18:12.707961936 +1000 +++ linux-2.6.8.1-ck8/include/linux/sched.h 2004-09-17 00:21:39.748723632 +1000 @@ -127,6 +127,14 @@ extern unsigned long nr_iowait(void); #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.1-ck8/kernel/sched.c =================================================================== --- linux-2.6.8.1-ck8.orig/kernel/sched.c 2004-09-17 00:21:28.840491964 +1000 +++ linux-2.6.8.1-ck8/kernel/sched.c 2004-09-17 00:21:39.750723308 +1000 @@ -2479,7 +2479,7 @@ static void __setscheduler(struct task_s 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; @@ -2525,9 +2525,8 @@ static int setscheduler(pid_t pid, int p 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; } /* @@ -2537,12 +2536,11 @@ static int setscheduler(pid_t pid, int p 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))