summaryrefslogtreecommitdiff
path: root/sysutils/libgtop/files/procaffinity.c
diff options
context:
space:
mode:
Diffstat (limited to 'sysutils/libgtop/files/procaffinity.c')
-rw-r--r--sysutils/libgtop/files/procaffinity.c87
1 files changed, 87 insertions, 0 deletions
diff --git a/sysutils/libgtop/files/procaffinity.c b/sysutils/libgtop/files/procaffinity.c
new file mode 100644
index 00000000000..141341d9e7b
--- /dev/null
+++ b/sysutils/libgtop/files/procaffinity.c
@@ -0,0 +1,87 @@
+/* Copyright (C) 2008 Juan Romero Pardines <xtraeme@gmail.com>
+
+ This file is part of LibGTop 2.
+
+ LibGTop is free software; you can redistribute it and/or modify it
+ under the terms of the GNU General Public License as published by
+ the Free Software Foundation; either version 2 of the License,
+ or (at your option) any later version.
+
+ LibGTop is distributed in the hope that it will be useful, but WITHOUT
+ ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
+ for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with LibGTop; see the file COPYING. If not, write to the
+ Free Software Foundation, Inc., 59 Temple Place - Suite 330,
+ Boston, MA 02111-1307, USA.
+*/
+
+#include <config.h>
+#include <glibtop/procaffinity.h>
+#include <glibtop/error.h>
+
+#include <sys/param.h>
+#if __NetBSD_Version__ >= 499007200 /* 4.99.72 to play safe */
+#include <sched.h>
+#endif
+
+void
+_glibtop_init_proc_affinity_s(glibtop *server)
+{
+ server->sysdeps.proc_affinity =
+ (1 << GLIBTOP_PROC_AFFINITY_NUMBER) |
+ (1 << GLIBTOP_PROC_AFFINITY_ALL);
+
+}
+
+
+guint16 *
+glibtop_get_proc_affinity_s(glibtop *server, glibtop_proc_affinity *buf, pid_t pid)
+{
+#if __NetBSD_Version__ >= 499007200
+ cpuset_t *cpuset;
+ size_t i;
+ GArray* cpus;
+
+ memset(buf, 0, sizeof *buf);
+
+ /* create the cpuset(3) */
+ cpuset = cpuset_create();
+ if (cpuset == NULL) {
+ glibtop_error_r(server, "cpuset_getaffinity (cpuset_create) failed");
+ return NULL;
+ }
+
+ /* get the affinity for _this_ pid, I assume that we always want lid 0 */
+ if ((_sched_getaffinity(pid, 0, cpuset_size(cpuset), cpuset)) != 0) {
+ cpuset_destroy(cpuset);
+ glibtop_error_r(server, "cpuset_getaffinity (_sched_getaffinity) failed");
+ return NULL;
+ }
+
+ cpus = g_array_new(FALSE, FALSE, sizeof(guint16));
+
+ for (i = 0; i < MIN(cpuset_size(cpuset), (size_t)(server->ncpu + 1)); i++) {
+ if (cpuset_isset(i, cpuset)) {
+ guint16 n = i;
+ g_array_append_val(cpus, n);
+ }
+ }
+
+ buf->number = cpus->len;
+ buf->all = (cpus->len == (size_t)(server->ncpu + 1));
+ buf->flags = (1 << GLIBTOP_PROC_AFFINITY_NUMBER)
+ | (1 << GLIBTOP_PROC_AFFINITY_ALL);
+
+ /* we are done */
+ cpuset_destroy(cpuset);
+
+ return (guint16*) g_array_free(cpus, FALSE);
+#else
+ memset(buf, 0, sizeof *buf);
+
+ return NULL;
+#endif
+}