From 5434415c11af6ea91f61fe2e873cfc29afc5657f Mon Sep 17 00:00:00 2001 From: Rob Landley Date: Mon, 3 Dec 2012 16:16:06 -0600 Subject: Use sched_{set,get}affinity directly so musl doesn't have to provide wrappers. --- toys/other/taskset.c | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/toys/other/taskset.c b/toys/other/taskset.c index 967bf70f..2b067d4a 100644 --- a/toys/other/taskset.c +++ b/toys/other/taskset.c @@ -23,9 +23,11 @@ config TASKSET #define FOR_taskset #include "toys.h" -// Prototype for syscall wrappers sched.h refuses to give us -int sched_setaffinity(pid_t pid, size_t size, void *cpuset); -int sched_getaffinity(pid_t pid, size_t size, void *cpuset); +#include +#define sched_setaffinity(pid, size, cpuset) \ + syscall(__NR_sched_setaffinity, (pid_t)pid, (size_t)size, (void *)cpuset) +#define sched_getaffinity(pid, size, cpuset) \ + syscall(__NR_sched_getaffinity, (pid_t)pid, (size_t)size, (void *)cpuset) // mask is an array of long, which makes the layout a bit weird on big // endian systems but as long as it's consistent... @@ -40,7 +42,7 @@ static void do_taskset(pid_t pid, int quiet) if (!quiet) { int j = sizeof(toybuf), flag = 0; - if (sched_getaffinity(pid, sizeof(toybuf), (void *)mask)) + if (-1 == sched_getaffinity(pid, sizeof(toybuf), (void *)mask)) perror_exit(failed, "get", pid); printf("pid %d's %s affinity mask: ", pid, i ? "new" : "current"); @@ -70,7 +72,7 @@ static void do_taskset(pid_t pid, int quiet) mask[j/(2*sizeof(long))] |= digit << 4*(j&((2*sizeof(long))-1)); } - if (sched_setaffinity(pid, sizeof(toybuf), (void *)mask)) + if (-1 == sched_setaffinity(pid, sizeof(toybuf), (void *)mask)) perror_exit(failed, "set", pid); } } -- cgit v1.2.3