/* Copyright (C) 1998-99 Martin Baulig

   This file is part of LibGTop 1.0.

   Contributed by Martin Baulig <martin@home-of-linux.org>, April 1998.

   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 <glibtop.h>

#include <glibtop/error.h>

#include <glibtop/procstate.h>

#include <glibtop_suid.h>

#include <mach.h>

#include <mach/mach_types.h>

#include <mach/task_info.h>

static const unsigned long _glibtop_sysdeps_proc_state =

(1L << GLIBTOP_PROC_STATE_CMD) +

(1L << GLIBTOP_PROC_STATE_UID) + (1L << GLIBTOP_PROC_STATE_GID) +

(1L << GLIBTOP_PROC_STATE_RUID) + (1L << GLIBTOP_PROC_STATE_RGID);

/* Init function. */

void

_glibtop_init_proc_state_p (glibtop *server)

{

	server->sysdeps.proc_state = _glibtop_sysdeps_proc_state;

}

/* Provides detailed information about a process. */

void

glibtop_get_proc_state_p (glibtop *server, glibtop_proc_state *buf,

			  pid_t pid)

{

	struct tbl_procinfo procinfo;

	int minim_state = 99, ret;

	task_t thistask;

	glibtop_init_p (server, GLIBTOP_SYSDEPS_PROC_STATE, 0);

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

	/* !!! THE FOLLOWING CODE RUNS SUID ROOT - CHANGE WITH CAUTION !!! */

	glibtop_suid_enter (server);

	ret = table (TBL_PROCINFO, pid, (char *) &procinfo, 1,

		     sizeof (struct tbl_procinfo));

	glibtop_suid_leave (server);

	/* !!! END OF SUID ROOT PART !!! */

	if (ret != 1) return;

	/* Check whether the process actually exists. */

	if (procinfo.pi_status == PI_EMPTY) return;

	/* Check whether it is not a zombie. */

	if (procinfo.pi_status == PI_ZOMBIE) {

		buf->state = GLIBTOP_PROCESS_ZOMBIE;

		buf->flags = (1L << GLIBTOP_PROC_STATE_STATE);

		return;

	}

	g_strlcpy (buf->cmd, procinfo.pi_comm, sizeof buf->cmd);

	buf->uid = procinfo.pi_svuid;

	buf->gid = procinfo.pi_svgid;

	buf->ruid = procinfo.pi_ruid;

	buf->rgid = procinfo.pi_rgid;

	buf->flags = _glibtop_sysdeps_proc_state;

	/* !!! THE FOLLOWING CODE RUNS SUID ROOT - CHANGE WITH CAUTION !!! */

	glibtop_suid_enter (server);

	/* Get task structure. */

	ret = task_by_unix_pid (task_self(), procinfo.pi_pid, &thistask);

	if (ret == KERN_SUCCESS) {

		thread_array_t			threadarr;

		unsigned int			threadarr_l;

		thread_basic_info_t		threadinfo;

		thread_basic_info_data_t	threadinfodata;

		int				j;

		/* Get thread array. */

		(void) task_threads (thistask, &threadarr, &threadarr_l);

		threadinfo = &threadinfodata;

		for (j = 0; j < threadarr_l; j++) {

			unsigned int		threadinfo_l = THREAD_BASIC_INFO_COUNT;

			int			tret;

			tret = thread_info (threadarr [j], THREAD_BASIC_INFO,

					    (thread_info_t) threadinfo, &threadinfo_l);

			if (tret == KERN_SUCCESS) {

          			if (minim_state > threadinfo->run_state)

					minim_state=threadinfo->run_state;

			}

		}

	}

	glibtop_suid_leave (server);

	/* !!! END OF SUID ROOT PART !!! */

	if (ret != KERN_SUCCESS) return;

	switch (minim_state) {

	case TH_STATE_RUNNING:

		buf->state = GLIBTOP_PROCESS_RUNNING;

		break;

	case TH_STATE_UNINTERRUPTIBLE:

		buf->state = GLIBTOP_PROCESS_UNINTERRUPTIBLE;

		break;

	case TH_STATE_WAITING:

		buf->state = GLIBTOP_PROCESS_INTERRUPTIBLE;

		break;

	case TH_STATE_STOPPED:

	case TH_STATE_HALTED:

		buf->state = GLIBTOP_PROCESS_STOPPED;

		break;

	default:

		if (ret != KERN_SUCCESS)

			buf->state = GLIBTOP_PROCESS_ZOMBIE;

		break;

	}

	if (buf->state)

		buf->flags |= (1L << GLIBTOP_PROC_STATE_STATE);

}