From eeb0328f203cfb8990b3eee5ee31d931fc539ceb Mon Sep 17 00:00:00 2001
From: Rich Felker <dalias@aerifal.cx>
Date: Sat, 10 Aug 2013 21:15:11 -0400
Subject: add cpu affinity interfaces

this first commit just includes the CPU_* and sched_* interfaces, not
the pthread_* interfaces, which may be added later. simple
sanity-check testing has been done for the basic interfaces, but most
of the macros have not yet been tested.
---
 src/sched/sched_cpucount.c    | 11 +++++++++++
 src/sched/sched_getaffinity.c | 10 ++++++++++
 src/sched/sched_setaffinity.c |  8 ++++++++
 3 files changed, 29 insertions(+)
 create mode 100644 src/sched/sched_cpucount.c
 create mode 100644 src/sched/sched_getaffinity.c
 create mode 100644 src/sched/sched_setaffinity.c

(limited to 'src')

diff --git a/src/sched/sched_cpucount.c b/src/sched/sched_cpucount.c
new file mode 100644
index 00000000..94aa259e
--- /dev/null
+++ b/src/sched/sched_cpucount.c
@@ -0,0 +1,11 @@
+#define _GNU_SOURCE
+#include <sched.h>
+
+int __sched_cpucount(size_t size, const cpu_set_t *set)
+{
+	size_t i, j, cnt=0;
+	const unsigned char *p = (const void *)set;
+	for (i=0; i<size; i++) for (j=0; j<8; j++)
+		if (p[i] & (1<<j)) cnt++;
+	return cnt;
+}
diff --git a/src/sched/sched_getaffinity.c b/src/sched/sched_getaffinity.c
new file mode 100644
index 00000000..0aa4c65a
--- /dev/null
+++ b/src/sched/sched_getaffinity.c
@@ -0,0 +1,10 @@
+#define _GNU_SOURCE
+#include <sched.h>
+#include "syscall.h"
+
+int sched_getaffinity(pid_t tid, size_t size, cpu_set_t *set)
+{
+	long ret = __syscall(SYS_sched_getaffinity, tid, size, set);
+	if (ret > 0) ret = 0;
+	return __syscall_ret(ret);
+}
diff --git a/src/sched/sched_setaffinity.c b/src/sched/sched_setaffinity.c
new file mode 100644
index 00000000..4344df17
--- /dev/null
+++ b/src/sched/sched_setaffinity.c
@@ -0,0 +1,8 @@
+#define _GNU_SOURCE
+#include <sched.h>
+#include "syscall.h"
+
+int sched_setaffinity(pid_t tid, size_t size, const cpu_set_t *set)
+{
+	return syscall(SYS_sched_setaffinity, tid, size, set);
+}
-- 
cgit v1.2.1