/* 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
}