/*
   This file is part of LibGTop 2.0.

   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., 51 Franklin Street, Fifth Floor,
   Boston, MA 02110-1301, USA.
*/

#include <config.h>
#include <unistd.h>

#include <glibtop.h>
#include <glibtop/error.h>
#include <glibtop/proclist.h>

#include <glibtop_suid.h>

#include <sys/param.h>
#include <sys/sysctl.h>  

static const unsigned long _glibtop_sysdeps_proclist =
(1L << GLIBTOP_PROCLIST_TOTAL) + (1L << GLIBTOP_PROCLIST_NUMBER) +
(1L << GLIBTOP_PROCLIST_SIZE);

/* Init function. */

void
_glibtop_init_proclist_p (glibtop *server)
{
	server->sysdeps.proclist = _glibtop_sysdeps_proclist;
}

pid_t *
glibtop_get_proclist_p (glibtop *server, glibtop_proclist *buf,
			gint64 which, gint64 arg)
{
	unsigned count, total, i;
	pid_t *pids_chain;
	int mib[4];
	struct kinfo_proc *kp;
	size_t length;

	glibtop_init_p (server, (1 << GLIBTOP_SYSDEPS_PROCLIST), 0);

	memset (buf, 0, sizeof (glibtop_proclist));

	mib [0] = CTL_KERN;
	mib [1] = KERN_PROC;
	mib [2] = (int)(which & GLIBTOP_KERN_PROC_MASK);
	mib [3] = (int)arg;

	if (sysctl (mib, 4, NULL, &length, NULL, 0) < 0) {
		glibtop_warn_io_r (server, "sysctl (proclist)");
		return NULL;
	}
	if ((kp = (struct kinfo_proc *) g_malloc(length)) == NULL) {
		glibtop_warn_io_r (server, "malloc (proclist)");
		return NULL;
	}
	if (sysctl (mib, 4, kp, &length, NULL, 0) < 0) {
		glibtop_warn_io_r (server, "sysctl (proclist)");
		g_free (kp);
		return NULL;
	}

	count = length / sizeof (struct kinfo_proc);
	pids_chain = g_malloc(count * sizeof (unsigned));
	if (pids_chain ==  NULL) {
		glibtop_warn_io_r (server, "g_realloc (proclist)");
		g_free (kp);
		return NULL;
	}

	for (total = 0, i = 0; i < count; i++) {
		if ((which & GLIBTOP_EXCLUDE_IDLE) &&
		    (kp [i].kp_proc.p_stat != SRUN))
			continue;
		if ((which & GLIBTOP_EXCLUDE_SYSTEM) &&
			 (kp [i].kp_eproc.e_pcred.p_ruid == 0))
			continue;
		pids_chain [total++] = (unsigned) kp [i].kp_proc.p_pid;
	}

	g_free (kp);

	buf->number = total;
	buf->size = sizeof (unsigned);
	buf->total = total * sizeof (unsigned);
	buf->flags = _glibtop_sysdeps_proclist;

	return pids_chain;
}