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