/**
* Read all processes to initialize the information tree.
* @param reference reference of ProcessTree
* @return treesize>0 if succeeded otherwise =0.
*/
int initprocesstree_sysdep(ProcessTree_T **reference) {
int i;
int treesize;
static kvm_t *kvm_handle;
ProcessTree_T *pt;
struct kinfo_proc *pinfo;
if (!(kvm_handle = kvm_open(NULL, _PATH_DEVNULL, NULL, O_RDONLY, prog))) {
LogError("system statistic error -- cannot initialize kvm interface\n");
return FALSE;
}
pinfo = kvm_getprocs(kvm_handle, KERN_PROC_ALL, 0, &treesize);
if (!pinfo || (treesize < 1)) {
LogError("system statistic error -- cannot get process tree\n");
kvm_close(kvm_handle);
return FALSE;
}
pt = xcalloc(sizeof(ProcessTree_T), treesize);
for (i = 0; i < treesize; i++) {
int j, flags;
char *procname = NULL;
char **args;
Buffer_T cmdline;
memset(&cmdline, 0, sizeof(Buffer_T));
#if (__FreeBSD_version > 500000)
pt[i].pid = pinfo[i].ki_pid;
pt[i].ppid = pinfo[i].ki_ppid;
pt[i].starttime = pinfo[i].ki_start.tv_sec;
pt[i].cputime = (long)(pinfo[i].ki_runtime / 100000);
pt[i].mem_kbyte = (unsigned long)(pinfo[i].ki_rssize * pagesize_kbyte);
flags = pinfo[i].ki_stat;
args = kvm_getargv(kvm_handle, &pinfo[i], 0);
procname = pinfo[i].ki_comm;
#else
pt[i].pid = pinfo[i].kp_proc.p_pid;
pt[i].ppid = pinfo[i].kp_eproc.e_ppid;
pt[i].starttime = pinfo[i].kp_eproc.e_stats.p_start.tv_sec;
pt[i].cputime = (long)(pinfo[i].kp_proc.p_runtime / 100000);
pt[i].mem_kbyte = (unsigned long)(pinfo[i].kp_eproc.e_vm.vm_rssize * pagesize_kbyte);
flags = pinfo[i].kp_proc.p_stat;
args = kvm_getargv(kvm_handle, &pinfo[i], 0);
procname = pinfo[i].kp_proc.p_comm;
#endif
if (flags == SZOMB)
pt[i].status_flag |= PROCESS_ZOMBIE;
pt[i].cpu_percent = 0;
pt[i].time = get_float_time();
if (args) {
for (j = 0; args[j]; j++)
Util_stringbuffer(&cmdline, args[j + 1] ? "%s " : "%s", args[j]);
pt[i].cmdline = cmdline.buf;
}
if (! pt[i].cmdline || ! *pt[i].cmdline)
pt[i].cmdline = xstrdup(procname);
}
*reference = pt;
kvm_close(kvm_handle);
return treesize;
}
/**
* Read all processes to initialize the information tree.
* @param reference reference of ProcessTree
* @return treesize > 0 if succeeded otherwise = 0.
*/
int initprocesstree_sysdep(ProcessTree_T **reference) {
int i;
int treesize;
char buf[_POSIX2_LINE_MAX];
size_t size = sizeof(maxslp);
int mib_proc2[6] = {CTL_KERN, KERN_PROC2, KERN_PROC_KTHREAD, 0, sizeof(struct kinfo_proc2), 0};
static int mib_maxslp[] = {CTL_VM, VM_MAXSLP};
ProcessTree_T *pt;
kvm_t *kvm_handle;
static struct kinfo_proc2 *pinfo;
if (sysctl(mib_maxslp, 2, &maxslp, &size, NULL, 0) < 0) {
LogError("system statistic error -- vm.maxslp failed");
return FALSE;
}
if (sysctl(mib_proc2, 6, NULL, &size, NULL, 0) == -1) {
LogError("system statistic error -- kern.proc2 #1 failed");
return FALSE;
}
size *= 2; // Add reserve for new processes which were created between calls of sysctl
pinfo = xcalloc(1, size);
mib_proc2[5] = (int)(size / sizeof(struct kinfo_proc2));
if (sysctl(mib_proc2, 6, pinfo, &size, NULL, 0) == -1) {
FREE(pinfo);
LogError("system statistic error -- kern.proc2 #2 failed");
return FALSE;
}
treesize = (int)(size / sizeof(struct kinfo_proc2));
pt = xcalloc(sizeof(ProcessTree_T), treesize);
if (! (kvm_handle = kvm_openfiles(NULL, NULL, NULL, KVM_NO_FILES, buf))) {
LogError("system statistic error -- kvm_openfiles failed: %s", buf);
return FALSE;
}
for (i = 0; i < treesize; i++) {
int j;
char **args;
Buffer_T cmdline;
pt[i].pid = pinfo[i].p_pid;
pt[i].ppid = pinfo[i].p_ppid;
pt[i].starttime = pinfo[i].p_ustart_sec;
pt[i].cputime = (long)((pinfo[i].p_rtime_sec * 10) + (pinfo[i].p_rtime_usec / 100000));
pt[i].cpu_percent = 0;
pt[i].mem_kbyte = (unsigned long)(pinfo[i].p_vm_rssize * pagesize_kbyte);
if (pinfo[i].p_stat == SZOMB)
pt[i].status_flag |= PROCESS_ZOMBIE; //FIXME: save system service flag too (kernel threads)
pt[i].time = get_float_time();
memset(&cmdline, 0, sizeof(Buffer_T));
if ((args = kvm_getargv2(kvm_handle, &pinfo[i], 0))) {
for (j = 0; args[j]; j++)
Util_stringbuffer(&cmdline, args[j + 1] ? "%s " : "%s", args[j]);
pt[i].cmdline = cmdline.buf;
}
if (! pt[i].cmdline || ! *pt[i].cmdline)
pt[i].cmdline = xstrdup(pinfo[i].p_comm);
}
FREE(pinfo);
kvm_close(kvm_handle);
*reference = pt;
return treesize;
}
