htop/darwin/DarwinProcessList.c

187 lines
5.3 KiB
C
Raw Normal View History

/*
htop - DarwinProcessList.c
(C) 2014 Hisham H. Muhammad
Released under the GNU GPL, see the COPYING file
in the source distribution for its full text.
*/
#include "ProcessList.h"
#include "DarwinProcess.h"
#include "DarwinProcessList.h"
#include <stdlib.h>
#include <string.h>
2015-07-13 06:17:14 +00:00
#include <unistd.h>
#include <stdio.h>
#include <libproc.h>
2015-07-13 21:53:46 +00:00
#include <mach/vm_page_size.h>
#include <sys/mman.h>
2015-07-14 02:02:40 +00:00
#include <utmpx.h>
/*{
2015-07-14 02:02:40 +00:00
#include "ProcessList.h"
2015-07-13 06:17:14 +00:00
#include <mach/mach_host.h>
#include <sys/sysctl.h>
typedef struct DarwinProcessList_ {
ProcessList super;
2015-07-13 21:53:46 +00:00
host_basic_info_data_t host_info;
2015-07-14 02:02:40 +00:00
vm_statistics64_data_t vm_stats;
2015-07-13 22:09:18 +00:00
processor_cpu_load_info_t prev_load;
processor_cpu_load_info_t curr_load;
2015-07-14 02:02:40 +00:00
uint64_t kernel_threads;
uint64_t user_threads;
2015-07-14 16:46:16 +00:00
uint64_t global_diff;
2015-07-13 06:17:14 +00:00
} DarwinProcessList;
}*/
2015-07-13 06:17:14 +00:00
void ProcessList_getHostInfo(host_basic_info_data_t *p) {
mach_msg_type_number_t info_size = HOST_BASIC_INFO_COUNT;
if(0 != host_info(mach_host_self(), HOST_BASIC_INFO, (host_info_t)p, &info_size)) {
fprintf(stderr, "Unable to retrieve host info\n");
exit(2);
}
}
2015-07-13 22:09:18 +00:00
void ProcessList_freeCPULoadInfo(processor_cpu_load_info_t *p) {
if(NULL != p && NULL != *p) {
2015-07-13 21:53:46 +00:00
if(0 != munmap(*p, vm_page_size)) {
fprintf(stderr, "Unable to free old CPU load information\n");
exit(8);
}
}
2015-07-13 22:09:18 +00:00
*p = NULL;
}
unsigned ProcessList_allocateCPULoadInfo(processor_cpu_load_info_t *p) {
mach_msg_type_number_t info_size = sizeof(processor_cpu_load_info_t);
unsigned cpu_count;
2015-07-14 16:46:16 +00:00
// TODO Improving the accuracy of the load counts woule help a lot.
2015-07-13 06:17:14 +00:00
if(0 != host_processor_info(mach_host_self(), PROCESSOR_CPU_LOAD_INFO, &cpu_count, (processor_info_array_t *)p, &info_size)) {
fprintf(stderr, "Unable to retrieve CPU info\n");
exit(4);
}
return cpu_count;
}
2015-07-14 02:02:40 +00:00
void ProcessList_getVMStats(vm_statistics64_t p) {
mach_msg_type_number_t info_size = HOST_VM_INFO64_COUNT;
if (host_statistics64(mach_host_self(), HOST_VM_INFO64, (host_info_t)p, &info_size) != 0)
err(9, "Unable to retrieve VM statistics\n");
2015-07-14 02:02:40 +00:00
}
2015-07-13 06:17:14 +00:00
struct kinfo_proc *ProcessList_getKInfoProcs(size_t *count) {
int mib[4] = { CTL_KERN, KERN_PROC, KERN_PROC_ALL, 0 };
struct kinfo_proc *processes = NULL;
/* Note the two calls to sysctl(). One to get length and one to get the
* data. This -does- mean that the second call could end up with a missing
* process entry or two.
*/
*count = 0;
if (sysctl(mib, 4, NULL, count, NULL, 0) < 0)
err(5, "Unable to get size of kproc_infos");
2015-07-13 06:17:14 +00:00
processes = malloc(*count);
if (processes == NULL)
err(6, "Out of memory for kproc_infos");
2015-07-13 06:17:14 +00:00
if (sysctl(mib, 4, processes, count, NULL, 0) < 0)
err(7, "Unable to get kinfo_procs");
2015-07-13 06:17:14 +00:00
*count = *count / sizeof(struct kinfo_proc);
return processes;
}
ProcessList* ProcessList_new(UsersTable* usersTable, Hashtable* pidWhiteList, uid_t userId) {
2015-07-13 06:17:14 +00:00
DarwinProcessList* this = calloc(1, sizeof(DarwinProcessList));
ProcessList_init(&this->super, Class(Process), usersTable, pidWhiteList, userId);
2015-07-14 02:02:40 +00:00
/* Initialize the CPU information */
2015-07-13 22:09:18 +00:00
this->super.cpuCount = ProcessList_allocateCPULoadInfo(&this->prev_load);
2015-07-13 21:53:46 +00:00
ProcessList_getHostInfo(&this->host_info);
2015-07-13 22:09:18 +00:00
ProcessList_allocateCPULoadInfo(&this->curr_load);
2015-07-13 06:17:14 +00:00
2015-07-14 02:02:40 +00:00
/* Initialize the VM statistics */
ProcessList_getVMStats(&this->vm_stats);
this->super.kernelThreads = 0;
this->super.userlandThreads = 0;
this->super.totalTasks = 0;
this->super.runningTasks = 0;
2015-07-13 06:17:14 +00:00
return &this->super;
}
void ProcessList_delete(ProcessList* this) {
ProcessList_done(this);
free(this);
}
void ProcessList_goThroughEntries(ProcessList* super) {
2015-07-13 06:17:14 +00:00
DarwinProcessList *dpl = (DarwinProcessList *)super;
bool preExisting = true;
2015-07-13 06:17:14 +00:00
struct kinfo_proc *ps;
size_t count;
2015-07-14 16:46:16 +00:00
DarwinProcess *proc;
2015-07-13 06:17:14 +00:00
struct timeval tv;
gettimeofday(&tv, NULL); /* Start processing time */
2015-07-14 02:02:40 +00:00
/* Update the global data (CPU times and VM stats) */
2015-07-13 22:09:18 +00:00
ProcessList_freeCPULoadInfo(&dpl->prev_load);
dpl->prev_load = dpl->curr_load;
ProcessList_allocateCPULoadInfo(&dpl->curr_load);
2015-07-14 02:02:40 +00:00
ProcessList_getVMStats(&dpl->vm_stats);
2015-07-14 16:46:16 +00:00
/* Get the time difference */
dpl->global_diff = 0;
for(int i = 0; i < dpl->super.cpuCount; ++i) {
for(size_t j = 0; j < CPU_STATE_MAX; ++j) {
dpl->global_diff += dpl->curr_load[i].cpu_ticks[j] - dpl->prev_load[i].cpu_ticks[j];
}
}
2015-07-14 02:02:40 +00:00
/* Clear the thread counts */
super->kernelThreads = 0;
super->userlandThreads = 0;
super->totalTasks = 0;
super->runningTasks = 0;
2015-07-13 21:53:46 +00:00
2015-07-13 06:17:14 +00:00
/* We use kinfo_procs for initial data since :
*
* 1) They always succeed.
* 2) The contain the basic information.
*
* We attempt to fill-in additional information with libproc.
*/
ps = ProcessList_getKInfoProcs(&count);
for(size_t i = 0; i < count; ++i) {
2015-07-14 16:46:16 +00:00
proc = (DarwinProcess *)ProcessList_getProcess(super, ps[i].kp_proc.p_pid, &preExisting, (Process_New)DarwinProcess_new);
2015-07-13 06:17:14 +00:00
2015-07-14 16:46:16 +00:00
DarwinProcess_setFromKInfoProc(&proc->super, ps + i, tv.tv_sec, preExisting);
DarwinProcess_setFromLibprocPidinfo(proc, dpl);
2015-07-14 02:02:40 +00:00
super->totalTasks += 1;
2015-07-13 06:17:14 +00:00
if(!preExisting) {
2015-07-14 16:46:16 +00:00
proc->super.user = UsersTable_getRef(super->usersTable, proc->super.st_uid);
2015-07-13 06:17:14 +00:00
2015-07-14 16:46:16 +00:00
ProcessList_add(super, &proc->super);
2015-07-13 06:17:14 +00:00
}
}
2015-07-13 06:17:14 +00:00
free(ps);
}