diff -Naurp linux-2.6.12-rc2-mm3.orig/include/linux/cpuset.h linux-2.6.12-rc2-mm3-0/include/linux/cpuset.h --- linux-2.6.12-rc2-mm3.orig/include/linux/cpuset.h 2005-05-11 19:44:42.000000000 +0530 +++ linux-2.6.12-rc2-mm3-0/include/linux/cpuset.h 2005-05-11 19:51:31.000000000 +0530 @@ -14,6 +14,10 @@ #ifdef CONFIG_CPUSETS +extern struct semaphore cpuset_sem; +#define lock_cpuset_sem() down(&cpuset_sem); +#define unlock_cpuset_sem() up(&cpuset_sem); + extern int cpuset_init(void); extern void cpuset_init_smp(void); extern void cpuset_fork(struct task_struct *p); @@ -29,6 +33,9 @@ extern char *cpuset_task_status_allowed( #else /* !CONFIG_CPUSETS */ +#define lock_cpuset_sem() do { } while (0) +#define unlock_cpuset_sem() do { } while (0) + static inline int cpuset_init(void) { return 0; } static inline void cpuset_init_smp(void) {} static inline void cpuset_fork(struct task_struct *p) {} diff -Naurp linux-2.6.12-rc2-mm3.orig/kernel/cpu.c linux-2.6.12-rc2-mm3-0/kernel/cpu.c --- linux-2.6.12-rc2-mm3.orig/kernel/cpu.c 2005-05-11 19:44:42.000000000 +0530 +++ linux-2.6.12-rc2-mm3-0/kernel/cpu.c 2005-05-11 19:46:38.000000000 +0530 @@ -10,6 +10,7 @@ #include #include #include +#include #include #include #include @@ -82,6 +83,7 @@ int cpu_down(unsigned int cpu) if ((err = lock_cpu_hotplug_interruptible()) != 0) return err; + lock_cpuset_sem(); if (num_online_cpus() == 1) { err = -EBUSY; @@ -145,6 +147,7 @@ out_thread: out_allowed: set_cpus_allowed(current, old_allowed); out: + unlock_cpuset_sem(); unlock_cpu_hotplug(); return err; } @@ -157,6 +160,7 @@ int __devinit cpu_up(unsigned int cpu) if ((ret = down_interruptible(&cpucontrol)) != 0) return ret; + lock_cpuset_sem(); if (cpu_online(cpu) || !cpu_present(cpu)) { ret = -EINVAL; @@ -184,6 +188,7 @@ out_notify: if (ret != 0) notifier_call_chain(&cpu_chain, CPU_UP_CANCELED, hcpu); out: + unlock_cpuset_sem(); up(&cpucontrol); return ret; } diff -Naurp linux-2.6.12-rc2-mm3.orig/kernel/cpuset.c linux-2.6.12-rc2-mm3-0/kernel/cpuset.c --- linux-2.6.12-rc2-mm3.orig/kernel/cpuset.c 2005-05-11 19:44:42.000000000 +0530 +++ linux-2.6.12-rc2-mm3-0/kernel/cpuset.c 2005-05-11 19:55:08.000000000 +0530 @@ -180,7 +180,7 @@ static struct super_block *cpuset_sb = N * for any special hack to ensure that top_cpuset cannot be deleted. */ -static DECLARE_MUTEX(cpuset_sem); +DECLARE_MUTEX(cpuset_sem); /* * A couple of forward declarations required, due to cyclic reference loop: @@ -1430,17 +1430,17 @@ void cpuset_exit(struct task_struct *tsk * attached to the specified @tsk. Guaranteed to return some non-empty * subset of cpu_online_map, even if this means going outside the * tasks cpuset. + * + * Call with cpuset_sem held. **/ cpumask_t cpuset_cpus_allowed(const struct task_struct *tsk) { cpumask_t mask; - down(&cpuset_sem); task_lock((struct task_struct *)tsk); guarantee_online_cpus(tsk->cpuset, &mask); task_unlock((struct task_struct *)tsk); - up(&cpuset_sem); return mask; } diff -Naurp linux-2.6.12-rc2-mm3.orig/kernel/sched.c linux-2.6.12-rc2-mm3-0/kernel/sched.c --- linux-2.6.12-rc2-mm3.orig/kernel/sched.c 2005-05-11 19:44:42.000000000 +0530 +++ linux-2.6.12-rc2-mm3-0/kernel/sched.c 2005-05-11 19:48:29.000000000 +0530 @@ -3608,11 +3608,13 @@ long sched_setaffinity(pid_t pid, cpumas cpumask_t cpus_allowed; lock_cpu_hotplug(); + lock_cpuset_sem(); read_lock(&tasklist_lock); p = find_process_by_pid(pid); if (!p) { read_unlock(&tasklist_lock); + unlock_cpuset_sem(); unlock_cpu_hotplug(); return -ESRCH; } @@ -3636,6 +3638,7 @@ long sched_setaffinity(pid_t pid, cpumas out_unlock: put_task_struct(p); + unlock_cpuset_sem(); unlock_cpu_hotplug(); return retval; }