src/libstatgrab/process_stats.c

/*
 * i-scream central monitoring system
 * http://www.i-scream.org
 * Copyright (C) 2000-2004 i-scream
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 * 
 * This library 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
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
 * 02111-1307 USA
 *
 * $Id: process_stats.c,v 1.43 2004/04/04 22:48:50 tdb Exp $
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "statgrab.h"
#if defined(SOLARIS) || defined(LINUX)
#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <dirent.h>
#include <string.h>
#endif

#ifdef SOLARIS
#include <procfs.h>
#include <limits.h>
#define PROC_LOCATION "/proc"
#define MAX_FILE_LENGTH PATH_MAX
#endif
#ifdef LINUX
#include <limits.h>
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>
#define PROC_LOCATION "/proc"
#define MAX_FILE_LENGTH PATH_MAX
#endif
#ifdef ALLBSD
#include <stdlib.h>
#include <sys/param.h>
#include <sys/sysctl.h>
#if defined(FREEBSD) || defined(DFBSD)
#include <sys/user.h>
#else
#include <sys/proc.h>
#endif
#include <string.h>
#include <paths.h>
#include <fcntl.h>
#include <limits.h>
#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
#include <kvm.h>
#include <tools.h>
#endif
#include <unistd.h>
#endif

int get_proc_snapshot(proc_state_t **ps){
        proc_state_t *proc_state = NULL;
        proc_state_t *proc_state_ptr;
        int proc_state_size = 0;
#ifdef ALLBSD
        int mib[4];
        size_t size;
        struct kinfo_proc *kp_stats;
        int procs, i;
        char *proctitle, *proctitletmp;
#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
        kvm_t *kvmd;
        char **args;
        int alloc;
#else
        long buflen;
        char *p;
#endif
#endif
#if defined(SOLARIS) || defined(LINUX)
        DIR *proc_dir;
        struct dirent *dir_entry;
        char filename[MAX_FILE_LENGTH];
        FILE *f;
#ifdef SOLARIS
        psinfo_t process_info;
#endif
#ifdef LINUX
        char s;
        /* If someone has a executable of 4k filename length, they deserve to get it truncated :) */
        char ps_name[4096];
        char *ptr;
        static char *psargs = NULL;
        static int psarg_size = 0;
        unsigned long stime, utime;
        int x;
        int fn;
        int toread;
        ssize_t size;
        int t_read;
#endif

        if((proc_dir=opendir(PROC_LOCATION))==NULL){
                return -1;
        }

        while((dir_entry=readdir(proc_dir))!=NULL){
                if(atoi(dir_entry->d_name) == 0) continue;

#ifdef SOLARIS
                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
#endif
#ifdef LINUX
                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/stat", dir_entry->d_name);
#endif
                if((f=fopen(filename, "r"))==NULL){
                        /* Open failed.. Process since vanished, or the path was too long.
                         * Ah well, move onwards to the next one */
                        continue;
                }
#ifdef SOLARIS
                fread(&process_info, sizeof(psinfo_t), 1, f);
#endif

                proc_state = realloc(proc_state, (1+proc_state_size)*sizeof(proc_state_t));
                proc_state_ptr = proc_state+proc_state_size;
#ifdef SOLARIS          
                proc_state_ptr->pid = process_info.pr_pid;
                proc_state_ptr->parent = process_info.pr_ppid;
                proc_state_ptr->pgid = process_info.pr_pgid;
                proc_state_ptr->uid = process_info.pr_uid;
                proc_state_ptr->euid = process_info.pr_euid;
                proc_state_ptr->gid = process_info.pr_gid;
                proc_state_ptr->egid = process_info.pr_egid;
                proc_state_ptr->proc_size = (process_info.pr_size) * 1024;
                proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
                proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
                proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
                proc_state_ptr->process_name = strdup(process_info.pr_fname);
                proc_state_ptr->proctitle = strdup(process_info.pr_psargs);

                if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SLEEPING;
                if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = RUNNING; 
                if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = ZOMBIE; 
                if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = STOPPED; 
                if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = RUNNING; 
#endif
#ifdef LINUX
                x = fscanf(f, "%d %4096s %c %d %d %*d %*d %*d %*u %*u %*u %*u %*u %lu %lu %*d %*d %*d %d %*d %*d %*u %llu %llu %*u %*u %*u %*u %*u %*u %*u %*u %*u %*u %*u %*u %*u %*d %*d\n", &(proc_state_ptr->pid), ps_name, &s, &(proc_state_ptr->parent), &(proc_state_ptr->pgid), &utime, &stime, &(proc_state_ptr->nice), &(proc_state_ptr->proc_size), &(proc_state_ptr->proc_resident));
                proc_state_ptr->proc_resident = proc_state_ptr->proc_resident * getpagesize();
                if(s == 'S') proc_state_ptr->state = SLEEPING;
                if(s == 'R') proc_state_ptr->state = RUNNING;
                if(s == 'Z') proc_state_ptr->state = ZOMBIE;
                if(s == 'T') proc_state_ptr->state = STOPPED;
                if(s == 'D') proc_state_ptr->state = STOPPED;
        
                /* pa_name[0] should = '(' */
                ptr = strchr(&ps_name[1], ')'); 
                if(ptr !=NULL) *ptr='\0';
                proc_state_ptr->process_name = strdup(&ps_name[1]);

                /* Need to do cpu */
                

                /* proctitle */ 
                snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/cmdline", dir_entry->d_name);

                if((fn=open(filename, O_RDONLY)) == -1){
                        /* Open failed.. Process since vanished, or the path was too long.
                         * Ah well, move onwards to the next one */
                        continue;
                }
#define         PSARG_START_SIZE 128
                if(psargs == NULL){
                        psargs = malloc(PSARG_START_SIZE);
                        psarg_size = PSARG_START_SIZE;
                }
                ptr = psargs;   
                t_read = 0;
                toread = psarg_size;
                while((size = read(fn, ptr, toread)) == toread){
                        psargs = realloc(psargs, (psarg_size + PSARG_START_SIZE));
                        ptr = psargs+psarg_size;
                        t_read = psarg_size;
                        psarg_size+=PSARG_START_SIZE;
                        toread = PSARG_START_SIZE;
                }
                if(size != -1) t_read+=size;

                ptr = psargs;
                for(x=0; x<t_read; x++){
                        if (*ptr == '\0') *ptr = ' ';
                        ptr++;
                }
                /*  for safety sake */
                psargs[t_read] = '\0';

                proc_state_ptr->proctitle = strdup(psargs);

#endif

                proc_state_size++;

                fclose(f);
        }
        closedir(proc_dir);
#endif

#ifdef ALLBSD
        mib[0] = CTL_KERN;
        mib[1] = KERN_PROC;
        mib[2] = KERN_PROC_ALL;

        if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
                return NULL;
        }

        procs = size / sizeof(struct kinfo_proc);

        kp_stats = malloc(size);
        if(kp_stats == NULL) {
                return NULL;
        }

        if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
                free(kp_stats);
                return NULL;
        }

#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
        kvmd = get_kvm2();
#endif

        for (i = 0; i < procs; i++) {
                /* replace with something more sensible */
                proc_state = realloc(proc_state,
                                (1+proc_state_size)*sizeof(proc_state_t));
                if(proc_state == NULL ) {
                        return NULL;
                }
                proc_state_ptr = proc_state+proc_state_size;

#ifdef FREEBSD5
                proc_state_ptr->process_name =
                        strdup(kp_stats[i].ki_comm);
#elif defined(DFBSD)
                proc_state_ptr->process_name =
                        strdup(kp_stats[i].kp_thread.td_comm);
#else
                proc_state_ptr->process_name =
                        strdup(kp_stats[i].kp_proc.p_comm);
#endif

#if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD)
                size = sizeof(buflen);

#ifdef FREEBSD5
                if(sysctlbyname("kern.ps_arg_cache_limit", &buflen, &size, NULL, 0) < 0) {
                        return NULL;
                }
#else
                mib[1] = KERN_ARGMAX;

                if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
                        return NULL;
                }
#endif

                proctitle = malloc(buflen);
                if(proctitle == NULL) {
                        return NULL;
                }

                size = buflen;

#ifdef FREEBSD5
                mib[2] = KERN_PROC_ARGS;
                mib[3] = kp_stats[i].ki_pid;
#else
                mib[1] = KERN_PROC_ARGS;
                mib[2] = kp_stats[i].kp_proc.p_pid;
                mib[3] = KERN_PROC_ARGV;
#endif

                if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
                        free(proctitle);
                        proc_state_ptr->proctitle = NULL;
                }
                else if(size > 0) {
                        proc_state_ptr->proctitle = malloc(size+1);
                        if(proc_state_ptr->proctitle == NULL) {
                                return NULL;
                        }
                        p = proctitle;
                        proc_state_ptr->proctitle[0] = NULL;
                        do {
                                strlcat(proc_state_ptr->proctitle, p, size+1);
                                strlcat(proc_state_ptr->proctitle, " ", size+1);
                                p += strlen(p) + 1;
                        } while (p < proctitle + size);
                        free(proctitle);
                                /* remove trailing space */
                        proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = \0;
                }
                else {
                        free(proctitle);
                        proc_state_ptr->proctitle = NULL;
                }
#else
                if(kvmd != NULL) {
                        args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
                        if(args != NULL) {
                                alloc = 1;
                                proctitle = malloc(alloc);
                                if(proctitle == NULL) {
                                        return NULL;
                                }
                                while(*args != NULL) {
                                        if(strlen(proctitle) + strlen(*args) >= alloc) {
                                                alloc = (alloc + strlen(*args)) * 2;
                                                proctitletmp = realloc(proctitle, alloc);
                                                if(proctitletmp == NULL) {
                                                        return NULL;
                                                }
                                                proctitle = proctitletmp;
                                        }
                                        strlcat(proctitle, *args, alloc);
                                        strlcat(proctitle, " ", alloc);
                                        args++;
                                }
                                /* remove trailing space */
                                proctitle[strlen(proctitle)-1] = \0;
                                proc_state_ptr->proctitle = proctitle;
                        }
                        else {
                                proc_state_ptr->proctitle = NULL;
                        }
                }
                else {
                        proc_state_ptr->proctitle = NULL;
                }
#endif

#ifdef FREEBSD5
                proc_state_ptr->pid = kp_stats[i].ki_pid;
                proc_state_ptr->parent = kp_stats[i].ki_ppid;
                proc_state_ptr->pgid = kp_stats[i].ki_pgid;
#else
                proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
                proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
                proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
#endif

#ifdef FREEBSD5
                proc_state_ptr->uid = kp_stats[i].ki_ruid;
                proc_state_ptr->euid = kp_stats[i].ki_uid;
                proc_state_ptr->gid = kp_stats[i].ki_rgid;
                proc_state_ptr->egid = kp_stats[i].ki_svgid;
#elif defined(DFBSD)
                proc_state_ptr->uid = kp_stats[i].kp_eproc.e_ucred.cr_ruid;
                proc_state_ptr->euid = kp_stats[i].kp_eproc.e_ucred.cr_svuid;
                proc_state_ptr->gid = kp_stats[i].kp_eproc.e_ucred.cr_rgid;
                proc_state_ptr->egid = kp_stats[i].kp_eproc.e_ucred.cr_svgid;
#else
                proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
                proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
                proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
                proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
#endif

#ifdef FREEBSD5
                proc_state_ptr->proc_size = kp_stats[i].ki_size;
                /* This is in pages */
                proc_state_ptr->proc_resident =
                        kp_stats[i].ki_rssize * getpagesize();
                /* This is in microseconds */
                proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
                proc_state_ptr->cpu_percent =
                        ((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
                proc_state_ptr->nice = kp_stats[i].ki_nice;
#else
                proc_state_ptr->proc_size =
                        kp_stats[i].kp_eproc.e_vm.vm_map.size;
                /* This is in pages */
                proc_state_ptr->proc_resident =
                        kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
#if defined(NETBSD) || defined(OPENBSD)
                proc_state_ptr->time_spent =
                        kp_stats[i].kp_proc.p_rtime.tv_sec;
#elif defined(DFBSD)
                proc_state_ptr->time_spent = 
                        ( kp_stats[i].kp_thread.td_uticks +
                        kp_stats[i].kp_thread.td_sticks +
                        kp_stats[i].kp_thread.td_iticks ) / 1000000;
#else
                /* This is in microseconds */
                proc_state_ptr->time_spent =
                        kp_stats[i].kp_proc.p_runtime / 1000000;
#endif
                proc_state_ptr->cpu_percent =
                        ((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
                proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
#endif

#ifdef FREEBSD5
                switch (kp_stats[i].ki_stat) {
#else
                switch (kp_stats[i].kp_proc.p_stat) {
#endif
                case SIDL:
                case SRUN:
#ifdef SONPROC
                case SONPROC: /* NetBSD */
#endif
                        proc_state_ptr->state = RUNNING;
                        break;
                case SSLEEP:
#ifdef SWAIT
                case SWAIT: /* FreeBSD 5 */
#endif
#ifdef SLOCK
                case SLOCK: /* FreeBSD 5 */
#endif
                        proc_state_ptr->state = SLEEPING;
                        break;
                case SSTOP:
                        proc_state_ptr->state = STOPPED;
                        break;
                case SZOMB:
#ifdef SDEAD
                case SDEAD: /* OpenBSD & NetBSD */
#endif
                        proc_state_ptr->state = ZOMBIE;
                        break;
                default:
                        proc_state_ptr->state = UNKNOWN;
                        break;
                }
                proc_state_size++;
        }

        free(kp_stats);
#endif

        *ps = proc_state;
        return proc_state_size;
}

process_stat_t *get_process_stats() {
        static process_stat_t process_stat;
        proc_state_t *ps;
        int ps_size, x;

        process_stat.sleeping = 0;
        process_stat.running = 0;
        process_stat.zombie = 0;
        process_stat.stopped = 0;
        process_stat.total = 0;

        ps_size = get_proc_snapshot(&ps);

        if(ps_size == 0) {
                return NULL;
        }

        for(x = 0; x < ps_size; x++) {
                switch (ps->state) {
                case RUNNING:
                        process_stat.running++;
                        break;
                case SLEEPING:
                        process_stat.sleeping++;
                        break;
                case STOPPED:
                        process_stat.stopped++;
                        break;
                case ZOMBIE:
                        process_stat.zombie++;
                        break;
                default:
                        /* currently no mapping for UNKNOWN in process_stat_t */
                        break;
                }
                ps++;
        }

        process_stat.total = ps_size;

        return &process_stat;
}
Revision:	1.44
Committed:	Sun Apr 4 23:24:42 2004 UTC (20 years, 1 month ago) by tdb
Content type:	text/plain
Branch:	MAIN
Changes since 1.43:	+14 -12 lines
Log Message:	Fix a bunch of silly errors.
#	User	Rev	Content
1	tdb	1.17	/*
2	pajs	1.1	* i-scream central monitoring system
3	tdb	1.8	* http://www.i-scream.org
4	tdb	1.17	* Copyright (C) 2000-2004 i-scream
5	pajs	1.1	*
6	tdb	1.17	* This library is free software; you can redistribute it and/or
7			* modify it under the terms of the GNU Lesser General Public
8			* License as published by the Free Software Foundation; either
9			* version 2.1 of the License, or (at your option) any later version.
10	pajs	1.1	*
11	tdb	1.17	* This library is distributed in the hope that it will be useful,
12	pajs	1.1	* but WITHOUT ANY WARRANTY; without even the implied warranty of
13	tdb	1.17	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14			* Lesser General Public License for more details.
15	pajs	1.1	*
16	tdb	1.17	* You should have received a copy of the GNU Lesser General Public
17			* License along with this library; if not, write to the Free Software
18			* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
19			* 02111-1307 USA
20	tdb	1.18	*
21	tdb	1.44	* $Id: process_stats.c,v 1.43 2004/04/04 22:48:50 tdb Exp $
22	pajs	1.1	*/
23
24			#ifdef HAVE_CONFIG_H
25			#include "config.h"
26			#endif
27
28	pajs	1.6	#include "statgrab.h"
29			#if defined(SOLARIS) \|\| defined(LINUX)
30	pajs	1.1	#include <stdio.h>
31			#include <stdlib.h>
32	pajs	1.5	#include <sys/types.h>
33			#include <dirent.h>
34			#include <string.h>
35	pajs	1.6	#endif
36	pajs	1.1
37			#ifdef SOLARIS
38			#include <procfs.h>
39			#include <limits.h>
40	pajs	1.5	#define PROC_LOCATION "/proc"
41			#define MAX_FILE_LENGTH PATH_MAX
42			#endif
43			#ifdef LINUX
44	ats	1.16	#include <limits.h>
45	pajs	1.25	#include <unistd.h>
46			#include <sys/stat.h>
47			#include <fcntl.h>
48	pajs	1.1	#define PROC_LOCATION "/proc"
49			#define MAX_FILE_LENGTH PATH_MAX
50			#endif
51	ats	1.13	#ifdef ALLBSD
52	tdb	1.19	#include <stdlib.h>
53	pajs	1.6	#include <sys/param.h>
54			#include <sys/sysctl.h>
55	tdb	1.22	#if defined(FREEBSD) \|\| defined(DFBSD)
56			#include <sys/user.h>
57			#else
58	tdb	1.19	#include <sys/proc.h>
59	ats	1.13	#endif
60	tdb	1.29	#include <string.h>
61			#include <paths.h>
62			#include <fcntl.h>
63			#include <limits.h>
64	tdb	1.41	#if (defined(FREEBSD) && !defined(FREEBSD5)) \|\| defined(DFBSD)
65	tdb	1.29	#include <kvm.h>
66	tdb	1.43	#include <tools.h>
67	pajs	1.6	#endif
68	tdb	1.41	#include <unistd.h>
69			#endif
70	pajs	1.1
71	pajs	1.23	int get_proc_snapshot(proc_state_t **ps){
72			proc_state_t *proc_state = NULL;
73			proc_state_t *proc_state_ptr;
74			int proc_state_size = 0;
75	tdb	1.26	#ifdef ALLBSD
76	tdb	1.37	int mib[4];
77	tdb	1.26	size_t size;
78			struct kinfo_proc *kp_stats;
79	tdb	1.41	int procs, i;
80	tdb	1.44	char proctitle, proctitletmp;
81	tdb	1.41	#if (defined(FREEBSD) && !defined(FREEBSD5)) \|\| defined(DFBSD)
82	tdb	1.42	kvm_t *kvmd;
83	tdb	1.41	char **args;
84			int alloc;
85			#else
86	tdb	1.37	long buflen;
87			char *p;
88			#endif
89	tdb	1.26	#endif
90	pajs	1.24	#if defined(SOLARIS) \|\| defined(LINUX)
91	pajs	1.23	DIR *proc_dir;
92			struct dirent *dir_entry;
93			char filename[MAX_FILE_LENGTH];
94			FILE *f;
95	pajs	1.24	#ifdef SOLARIS
96	pajs	1.5	psinfo_t process_info;
97	pajs	1.24	#endif
98	pajs	1.25	#ifdef LINUX
99			char s;
100			/* If someone has a executable of 4k filename length, they deserve to get it truncated :) */
101			char ps_name[4096];
102			char *ptr;
103			static char *psargs = NULL;
104			static int psarg_size = 0;
105			unsigned long stime, utime;
106			int x;
107			int fn;
108			int toread;
109			ssize_t size;
110			int t_read;
111			#endif
112	pajs	1.1
113	pajs	1.23	if((proc_dir=opendir(PROC_LOCATION))==NULL){
114	pajs	1.25	return -1;
115	pajs	1.23	}
116
117			while((dir_entry=readdir(proc_dir))!=NULL){
118			if(atoi(dir_entry->d_name) == 0) continue;
119
120	pajs	1.24	#ifdef SOLARIS
121	pajs	1.23	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
122	pajs	1.24	#endif
123	pajs	1.25	#ifdef LINUX
124			snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/stat", dir_entry->d_name);
125			#endif
126	pajs	1.23	if((f=fopen(filename, "r"))==NULL){
127			/* Open failed.. Process since vanished, or the path was too long.
128			* Ah well, move onwards to the next one */
129			continue;
130			}
131	pajs	1.24	#ifdef SOLARIS
132	pajs	1.23	fread(&process_info, sizeof(psinfo_t), 1, f);
133	pajs	1.25	#endif
134	pajs	1.1
135	pajs	1.23	proc_state = realloc(proc_state, (1+proc_state_size)*sizeof(proc_state_t));
136			proc_state_ptr = proc_state+proc_state_size;
137	pajs	1.25	#ifdef SOLARIS
138	pajs	1.23	proc_state_ptr->pid = process_info.pr_pid;
139			proc_state_ptr->parent = process_info.pr_ppid;
140			proc_state_ptr->pgid = process_info.pr_pgid;
141			proc_state_ptr->uid = process_info.pr_uid;
142			proc_state_ptr->euid = process_info.pr_euid;
143			proc_state_ptr->gid = process_info.pr_gid;
144			proc_state_ptr->egid = process_info.pr_egid;
145			proc_state_ptr->proc_size = (process_info.pr_size) * 1024;
146			proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
147			proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
148			proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
149			proc_state_ptr->process_name = strdup(process_info.pr_fname);
150			proc_state_ptr->proctitle = strdup(process_info.pr_psargs);
151	pajs	1.24
152			if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SLEEPING;
153			if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = RUNNING;
154			if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = ZOMBIE;
155			if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = STOPPED;
156			if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = RUNNING;
157	pajs	1.25	#endif
158			#ifdef LINUX
159	ats	1.38	x = fscanf(f, "%d %4096s %c %d %d %d %d %d %u %u %u %u %u %lu %lu %d %d %d %d %d %d %u %llu %llu %u %u %u %u %u %u %u %u %u %u %u %u %u %d %*d\n", &(proc_state_ptr->pid), ps_name, &s, &(proc_state_ptr->parent), &(proc_state_ptr->pgid), &utime, &stime, &(proc_state_ptr->nice), &(proc_state_ptr->proc_size), &(proc_state_ptr->proc_resident));
160	pajs	1.25	proc_state_ptr->proc_resident = proc_state_ptr->proc_resident * getpagesize();
161			if(s == 'S') proc_state_ptr->state = SLEEPING;
162			if(s == 'R') proc_state_ptr->state = RUNNING;
163			if(s == 'Z') proc_state_ptr->state = ZOMBIE;
164			if(s == 'T') proc_state_ptr->state = STOPPED;
165			if(s == 'D') proc_state_ptr->state = STOPPED;
166
167			/* pa_name[0] should = '(' */
168			ptr = strchr(&ps_name[1], ')');
169			if(ptr !=NULL) *ptr='\0';
170			proc_state_ptr->process_name = strdup(&ps_name[1]);
171
172			/* Need to do cpu */
173
174
175			/* proctitle */
176			snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/cmdline", dir_entry->d_name);
177
178			if((fn=open(filename, O_RDONLY)) == -1){
179			/* Open failed.. Process since vanished, or the path was too long.
180			* Ah well, move onwards to the next one */
181			continue;
182			}
183			#define PSARG_START_SIZE 128
184			if(psargs == NULL){
185			psargs = malloc(PSARG_START_SIZE);
186			psarg_size = PSARG_START_SIZE;
187			}
188			ptr = psargs;
189			t_read = 0;
190			toread = psarg_size;
191			while((size = read(fn, ptr, toread)) == toread){
192			psargs = realloc(psargs, (psarg_size + PSARG_START_SIZE));
193			ptr = psargs+psarg_size;
194			t_read = psarg_size;
195			psarg_size+=PSARG_START_SIZE;
196			toread = PSARG_START_SIZE;
197			}
198			if(size != -1) t_read+=size;
199
200			ptr = psargs;
201			for(x=0; x<t_read; x++){
202			if (ptr == '\0') ptr = ' ';
203			ptr++;
204			}
205			/* for safety sake */
206			psargs[t_read] = '\0';
207
208			proc_state_ptr->proctitle = strdup(psargs);
209
210	pajs	1.24	#endif
211	pajs	1.23
212			proc_state_size++;
213	pajs	1.1
214	pajs	1.23	fclose(f);
215			}
216			closedir(proc_dir);
217	tdb	1.26	#endif
218
219			#ifdef ALLBSD
220	tdb	1.36	mib[0] = CTL_KERN;
221			mib[1] = KERN_PROC;
222			mib[2] = KERN_PROC_ALL;
223
224			if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
225			return NULL;
226			}
227
228			procs = size / sizeof(struct kinfo_proc);
229
230			kp_stats = malloc(size);
231			if(kp_stats == NULL) {
232			return NULL;
233			}
234
235			if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
236			free(kp_stats);
237			return NULL;
238			}
239	tdb	1.33
240	tdb	1.37	#if (defined(FREEBSD) && !defined(FREEBSD5)) \|\| defined(DFBSD)
241	tdb	1.42	kvmd = get_kvm2();
242	tdb	1.33	#endif
243	tdb	1.26
244			for (i = 0; i < procs; i++) {
245	tdb	1.29	/* replace with something more sensible */
246	tdb	1.30	proc_state = realloc(proc_state,
247			(1+proc_state_size)*sizeof(proc_state_t));
248	tdb	1.29	if(proc_state == NULL ) {
249			return NULL;
250			}
251	tdb	1.26	proc_state_ptr = proc_state+proc_state_size;
252	tdb	1.30
253			#ifdef FREEBSD5
254			proc_state_ptr->process_name =
255			strdup(kp_stats[i].ki_comm);
256	tdb	1.35	#elif defined(DFBSD)
257			proc_state_ptr->process_name =
258			strdup(kp_stats[i].kp_thread.td_comm);
259	tdb	1.30	#else
260			proc_state_ptr->process_name =
261			strdup(kp_stats[i].kp_proc.p_comm);
262			#endif
263	tdb	1.29
264	tdb	1.37	#if defined(FREEBSD5) \|\| defined(NETBSD) \|\| defined(OPENBSD)
265			size = sizeof(buflen);
266
267			#ifdef FREEBSD5
268			if(sysctlbyname("kern.ps_arg_cache_limit", &buflen, &size, NULL, 0) < 0) {
269			return NULL;
270			}
271			#else
272			mib[1] = KERN_ARGMAX;
273
274			if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
275			return NULL;
276			}
277			#endif
278
279			proctitle = malloc(buflen);
280			if(proctitle == NULL) {
281			return NULL;
282			}
283
284			size = buflen;
285
286			#ifdef FREEBSD5
287			mib[2] = KERN_PROC_ARGS;
288			mib[3] = kp_stats[i].ki_pid;
289			#else
290			mib[1] = KERN_PROC_ARGS;
291			mib[2] = kp_stats[i].kp_proc.p_pid;
292			mib[3] = KERN_PROC_ARGV;
293			#endif
294
295			if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
296			free(proctitle);
297			proc_state_ptr->proctitle = NULL;
298			}
299			else if(size > 0) {
300	tdb	1.44	proc_state_ptr->proctitle = malloc(size+1);
301	tdb	1.37	if(proc_state_ptr->proctitle == NULL) {
302	tdb	1.29	return NULL;
303			}
304	tdb	1.37	p = proctitle;
305			proc_state_ptr->proctitle[0] = NULL;
306			do {
307	tdb	1.44	strlcat(proc_state_ptr->proctitle, p, size+1);
308			strlcat(proc_state_ptr->proctitle, " ", size+1);
309	tdb	1.37	p += strlen(p) + 1;
310			} while (p < proctitle + size);
311			free(proctitle);
312	tdb	1.44	/* remove trailing space */
313			proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = \0;
314	tdb	1.37	}
315			else {
316			free(proctitle);
317			proc_state_ptr->proctitle = NULL;
318			}
319			#else
320			if(kvmd != NULL) {
321			args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
322			if(args != NULL) {
323			alloc = 1;
324			proctitle = malloc(alloc);
325			if(proctitle == NULL) {
326			return NULL;
327			}
328			while(*args != NULL) {
329			if(strlen(proctitle) + strlen(*args) >= alloc) {
330	tdb	1.44	alloc = (alloc + strlen(args)) 2;
331			proctitletmp = realloc(proctitle, alloc);
332			if(proctitletmp == NULL) {
333	tdb	1.37	return NULL;
334			}
335	tdb	1.44	proctitle = proctitletmp;
336	tdb	1.29	}
337	tdb	1.44	strlcat(proctitle, *args, alloc);
338			strlcat(proctitle, " ", alloc);
339	tdb	1.37	args++;
340	tdb	1.29	}
341	tdb	1.37	/* remove trailing space */
342	tdb	1.44	proctitle[strlen(proctitle)-1] = \0;
343	tdb	1.37	proc_state_ptr->proctitle = proctitle;
344			}
345			else {
346			proc_state_ptr->proctitle = NULL;
347	tdb	1.29	}
348			}
349			else {
350	tdb	1.37	proc_state_ptr->proctitle = NULL;
351	tdb	1.29	}
352	tdb	1.37	#endif
353	tdb	1.26
354	tdb	1.30	#ifdef FREEBSD5
355	tdb	1.26	proc_state_ptr->pid = kp_stats[i].ki_pid;
356			proc_state_ptr->parent = kp_stats[i].ki_ppid;
357			proc_state_ptr->pgid = kp_stats[i].ki_pgid;
358	tdb	1.30	#else
359			proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
360			proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
361			proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
362			#endif
363	tdb	1.26
364	tdb	1.30	#ifdef FREEBSD5
365	tdb	1.26	proc_state_ptr->uid = kp_stats[i].ki_ruid;
366			proc_state_ptr->euid = kp_stats[i].ki_uid;
367			proc_state_ptr->gid = kp_stats[i].ki_rgid;
368			proc_state_ptr->egid = kp_stats[i].ki_svgid;
369	tdb	1.35	#elif defined(DFBSD)
370			proc_state_ptr->uid = kp_stats[i].kp_eproc.e_ucred.cr_ruid;
371			proc_state_ptr->euid = kp_stats[i].kp_eproc.e_ucred.cr_svuid;
372			proc_state_ptr->gid = kp_stats[i].kp_eproc.e_ucred.cr_rgid;
373			proc_state_ptr->egid = kp_stats[i].kp_eproc.e_ucred.cr_svgid;
374	tdb	1.30	#else
375			proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
376			proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
377			proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
378			proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
379			#endif
380	tdb	1.26
381	tdb	1.30	#ifdef FREEBSD5
382	tdb	1.26	proc_state_ptr->proc_size = kp_stats[i].ki_size;
383			/* This is in pages */
384	tdb	1.30	proc_state_ptr->proc_resident =
385			kp_stats[i].ki_rssize * getpagesize();
386	tdb	1.28	/* This is in microseconds */
387	tdb	1.26	proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
388	tdb	1.30	proc_state_ptr->cpu_percent =
389			((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
390	tdb	1.26	proc_state_ptr->nice = kp_stats[i].ki_nice;
391	tdb	1.30	#else
392			proc_state_ptr->proc_size =
393			kp_stats[i].kp_eproc.e_vm.vm_map.size;
394			/* This is in pages */
395			proc_state_ptr->proc_resident =
396			kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
397	tdb	1.34	#if defined(NETBSD) \|\| defined(OPENBSD)
398	tdb	1.33	proc_state_ptr->time_spent =
399			kp_stats[i].kp_proc.p_rtime.tv_sec;
400	tdb	1.35	#elif defined(DFBSD)
401			proc_state_ptr->time_spent =
402			( kp_stats[i].kp_thread.td_uticks +
403			kp_stats[i].kp_thread.td_sticks +
404			kp_stats[i].kp_thread.td_iticks ) / 1000000;
405	tdb	1.33	#else
406	tdb	1.36	/* This is in microseconds */
407	tdb	1.30	proc_state_ptr->time_spent =
408			kp_stats[i].kp_proc.p_runtime / 1000000;
409	tdb	1.33	#endif
410	tdb	1.30	proc_state_ptr->cpu_percent =
411			((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
412			proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
413			#endif
414	tdb	1.26
415	tdb	1.30	#ifdef FREEBSD5
416	tdb	1.26	switch (kp_stats[i].ki_stat) {
417	tdb	1.30	#else
418			switch (kp_stats[i].kp_proc.p_stat) {
419			#endif
420	tdb	1.26	case SIDL:
421			case SRUN:
422			#ifdef SONPROC
423			case SONPROC: /* NetBSD */
424			#endif
425			proc_state_ptr->state = RUNNING;
426			break;
427			case SSLEEP:
428			#ifdef SWAIT
429			case SWAIT: /* FreeBSD 5 */
430			#endif
431			#ifdef SLOCK
432			case SLOCK: /* FreeBSD 5 */
433			#endif
434			proc_state_ptr->state = SLEEPING;
435			break;
436			case SSTOP:
437			proc_state_ptr->state = STOPPED;
438			break;
439			case SZOMB:
440			#ifdef SDEAD
441			case SDEAD: /* OpenBSD & NetBSD */
442			#endif
443			proc_state_ptr->state = ZOMBIE;
444	tdb	1.31	break;
445			default:
446			proc_state_ptr->state = UNKNOWN;
447	tdb	1.26	break;
448			}
449			proc_state_size++;
450			}
451
452			free(kp_stats);
453			#endif
454
455	pajs	1.23	*ps = proc_state;
456			return proc_state_size;
457	tdb	1.27	}
458
459			process_stat_t *get_process_stats() {
460			static process_stat_t process_stat;
461			proc_state_t *ps;
462			int ps_size, x;
463
464	tdb	1.30	process_stat.sleeping = 0;
465			process_stat.running = 0;
466			process_stat.zombie = 0;
467			process_stat.stopped = 0;
468			process_stat.total = 0;
469	tdb	1.27
470			ps_size = get_proc_snapshot(&ps);
471	tdb	1.32
472	ats	1.39	if(ps_size == 0) {
473	tdb	1.32	return NULL;
474			}
475	tdb	1.27
476			for(x = 0; x < ps_size; x++) {
477			switch (ps->state) {
478			case RUNNING:
479			process_stat.running++;
480			break;
481			case SLEEPING:
482			process_stat.sleeping++;
483			break;
484			case STOPPED:
485			process_stat.stopped++;
486			break;
487			case ZOMBIE:
488			process_stat.zombie++;
489	ats	1.40	break;
490			default:
491			/* currently no mapping for UNKNOWN in process_stat_t */
492	tdb	1.27	break;
493			}
494			ps++;
495			}
496
497			process_stat.total = ps_size;
498
499			return &process_stat;
500	pajs	1.23	}