diff -urN linux-2.4.19-pre7/arch/i386/kernel/entry.S linux/arch/i386/kernel/entry.S --- linux-2.4.19-pre7/arch/i386/kernel/entry.S Tue Apr 16 01:28:01 2002 +++ linux/arch/i386/kernel/entry.S Tue Apr 16 01:39:02 2002 @@ -637,8 +637,8 @@ .long SYMBOL_NAME(sys_tkill) .long SYMBOL_NAME(sys_ni_syscall) /* reserved for sendfile64 */ .long SYMBOL_NAME(sys_ni_syscall) /* 240 reserved for futex */ - .long SYMBOL_NAME(sys_ni_syscall) /* reserved for sched_setaffinity */ - .long SYMBOL_NAME(sys_ni_syscall) /* reserved for sched_getaffinity */ + .long SYMBOL_NAME(sys_sched_setaffinity) + .long SYMBOL_NAME(sys_sched_getaffinity) .rept NR_syscalls-(.-sys_call_table)/4 .long SYMBOL_NAME(sys_ni_syscall) diff -urN linux-2.4.19-pre7/include/linux/capability.h linux/include/linux/capability.h --- linux-2.4.19-pre7/include/linux/capability.h Tue Apr 16 01:25:12 2002 +++ linux/include/linux/capability.h Tue Apr 16 01:41:47 2002 @@ -243,6 +243,7 @@ /* Allow use of FIFO and round-robin (realtime) scheduling on own processes and setting the scheduling algorithm used by another process. */ +/* Allow setting cpu affinity on other processes */ #define CAP_SYS_NICE 23 diff -urN linux-2.4.19-pre7/kernel/sched.c linux/kernel/sched.c --- linux-2.4.19-pre7/kernel/sched.c Tue Apr 16 01:28:27 2002 +++ linux/kernel/sched.c Tue Apr 16 01:40:43 2002 @@ -1124,6 +1124,106 @@ return retval; } +/** + * sys_sched_setaffinity - set the cpu affinity of a process + * @pid: pid of the process + * @len: length of the bitmask pointed to by user_mask_ptr + * @user_mask_ptr: userspace pointer to the new cpu mask + */ +asmlinkage int sys_sched_setaffinity(pid_t pid, unsigned int len, + unsigned long *user_mask_ptr) +{ + unsigned long new_mask; + struct task_struct *p; + int retval, reschedule = 0; + + if (len < sizeof(new_mask)) + return -EINVAL; + + if (copy_from_user(&new_mask, user_mask_ptr, sizeof(new_mask))) + return -EFAULT; + + new_mask &= cpu_online_map; + if (!new_mask) + return -EINVAL; + + read_lock_irq(&tasklist_lock); + spin_lock(&runqueue_lock); + + retval = -ESRCH; + p = find_process_by_pid(pid); + if (!p) + goto out_unlock; + + retval = -EPERM; + if ((current->euid != p->euid) && (current->euid != p->uid) && + !capable(CAP_SYS_NICE)) + goto out_unlock; + + p->cpus_allowed = new_mask; + +#ifdef CONFIG_SMP + if (!(p->cpus_runnable & p->cpus_allowed)) { + if (p == current) + reschedule = 1; + else { + p->need_resched = 1; + smp_send_reschedule(p->processor); + } + } +#endif + + retval = 0; +out_unlock: + spin_unlock(&runqueue_lock); + read_unlock_irq(&tasklist_lock); + + if (reschedule) + schedule(); + + return retval; +} + +/** + * sys_sched_getaffinity - get the cpu affinity of a process + * @pid: pid of the process + * @len: length of the bitmask pointed to by user_mask_ptr + * @user_mask_ptr: userspace pointer to hold the current mask + */ +asmlinkage int sys_sched_getaffinity(pid_t pid, unsigned int len, + unsigned long *user_mask_ptr) +{ + unsigned long mask; + unsigned int real_len; + struct task_struct *p; + int retval; + + real_len = sizeof(mask); + + if (len < real_len) + return -EINVAL; + + read_lock_irq(&tasklist_lock); + spin_lock(&runqueue_lock); + + retval = -ESRCH; + p = find_process_by_pid(pid); + if (!p) + goto out_unlock; + + retval = 0; + mask = p->cpus_allowed & cpu_online_map; + +out_unlock: + spin_unlock(&runqueue_lock); + read_unlock_irq(&tasklist_lock); + if (retval) + return retval; + if (copy_to_user(user_mask_ptr, &mask, real_len)) + return -EFAULT; + return real_len; +} + static void show_task(struct task_struct * p) { unsigned long free = 0;