/*

   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/proctime.h>

#include <glibtop_suid.h>

#include <mach/mach_init.h>

#include <mach/mach_host.h>

#include <mach/mach_port.h>

#include <mach/task.h>

#include <mach/vm_map.h>

#include <sys/types.h>

#include <sys/sysctl.h>

static const unsigned long _glibtop_sysdeps_proc_time =

(1L << GLIBTOP_PROC_TIME_UTIME) + (1L << GLIBTOP_PROC_TIME_STIME) +

(1L << GLIBTOP_PROC_TIME_FREQUENCY) + (1L << GLIBTOP_PROC_TIME_START_TIME);

#define tv2sec(tv) (((u_int64_t) tv.seconds * 1000000) + (u_int64_t) tv.microseconds)

/* Init function. */

void

_glibtop_init_proc_time_p (glibtop *server)

{

	server->sysdeps.proc_time = _glibtop_sysdeps_proc_time;

}

/* Provides detailed information about a process. */

void

glibtop_get_proc_time_p (glibtop *server, glibtop_proc_time *buf,

			 pid_t pid)

{

	unsigned int count;

	task_basic_info_data_t ti;

	task_thread_times_info_data_t tti;

	task_port_t task;

	time_value_t utime = {0, 0}, stime = {0, 0};

	size_t length;

	struct kinfo_proc kinfo;

	int mib[] = { CTL_KERN, KERN_PROC, KERN_PROC_PID, 0 };

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

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

	mib[3] = pid;

	length = sizeof (kinfo);

	if (sysctl (mib, 4, &kinfo, &length, NULL, 0) < 0) {

		glibtop_suid_leave (server);

		return;

	}

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

	glibtop_suid_enter (server);

	if (task_for_pid (mach_task_self(), pid, &task) != KERN_SUCCESS) {

		glibtop_suid_leave (server);

		return;

	}

	glibtop_suid_leave (server);

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

	count = TASK_BASIC_INFO_COUNT;

	if (task_info(task, TASK_BASIC_INFO, (task_info_t)&ti, &count) != KERN_SUCCESS) {

		glibtop_warn_io_r (server, "task_info (proctime)");

		if (task != mach_task_self())

			mach_port_deallocate (mach_task_self(), task);

		return;

	}

	count = TASK_THREAD_TIMES_INFO_COUNT;

	if (task_info(task, TASK_THREAD_TIMES_INFO, (task_info_t)&tti, &count) != KERN_SUCCESS) {

		glibtop_warn_io_r (server, "task_info (proctime)");

		if (task != mach_task_self())

			mach_port_deallocate (mach_task_self(), task);

		return;

	}

	time_value_add (&utime, &ti.user_time);

	time_value_add (&stime, &ti.system_time);

	time_value_add (&utime, &tti.user_time);

	time_value_add (&stime, &tti.system_time);

	/* use the same frequency as cpu.c */

	buf->utime  = tv2sec (utime) / 10000;

	buf->stime  = tv2sec (stime) / 10000;

	buf->frequency = 1000000 / 10000;

	buf->start_time = kinfo.kp_proc.p_starttime.tv_sec;

	buf->flags  = _glibtop_sysdeps_proc_time;

}