Blame sysdeps/bsd/procaffinity.c

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