diff options
| -rw-r--r-- | src/sched/affinity.c | 19 |
1 files changed, 11 insertions, 8 deletions
diff --git a/src/sched/affinity.c b/src/sched/affinity.c index 737e41b5..948ece41 100644 --- a/src/sched/affinity.c +++ b/src/sched/affinity.c @@ -11,20 +11,23 @@ int sched_setaffinity(pid_t tid, size_t size, const cpu_set_t *set) int pthread_setaffinity_np(pthread_t td, size_t size, const cpu_set_t *set) { - return syscall(SYS_sched_setaffinity, td->tid, size, set); + return -__syscall(SYS_sched_setaffinity, td->tid, size, set); } -int sched_getaffinity(pid_t tid, size_t size, cpu_set_t *set) +static int do_getaffinity(pid_t tid, size_t size, cpu_set_t *set) { long ret = __syscall(SYS_sched_getaffinity, tid, size, set); - if (ret > 0) { - if (ret < size) memset((char *)set+ret, 0, size-ret); - ret = 0; - } - return __syscall_ret(ret); + if (ret < 0) return ret; + if (ret < size) memset((char *)set+ret, 0, size-ret); + return 0; +} + +int sched_getaffinity(pid_t tid, size_t size, cpu_set_t *set) +{ + return __syscall_ret(do_getaffinity(tid, size, set)); } int pthread_getaffinity_np(pthread_t td, size_t size, cpu_set_t *set) { - return sched_getaffinity(td->tid, size, set); + return -do_getaffinity(td->tid, size, set); } |