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.32 2004/04/04 11:02:30 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>
#include <kvm.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[3];
        size_t size;
        struct kinfo_proc *kp_stats;
        int procs, i, alloc;
        static kvm_t *kvmd;
        char **args;
        char *proctitle;
#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 %*lu %*lu %*lu %*lu %*lu %lu %lu %*ld %*ld %*ld %d %*ld %*ld %*lu %llu %llu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*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

#ifdef NETBSD
        kvmd = kvm_openfiles(NULL, NULL, NULL, O_RDONLY, NULL);
#else
        kvmd = kvm_openfiles(_PATH_DEVNULL, _PATH_DEVNULL, NULL, O_RDONLY, NULL);
#endif

        if(kvmd == NULL) return NULL;

        kp_stats = kvm_getprocs(kvmd, KERN_PROC_ALL, 0, &procs);

        if (kp_stats == NULL || procs < 0) {
                return NULL;
        }

        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);
#else
                proc_state_ptr->process_name =
                        strdup(kp_stats[i].kp_proc.p_comm);
#endif

                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)) << 1;
                                        proctitle = realloc(proctitle, alloc);
                                        if(proctitle == NULL) {
                                                return NULL;
                                        }
                                }
                                strncat(proctitle, *args, strlen(*args));
                                strncat(proctitle, " ", 1);
                                args++;
                        }
                        /* remove trailing space */
                        proctitle[strlen(proctitle)-1] = NULL;
                        proc_state_ptr->proctitle = proctitle;
                }
                else {
                        proc_state_ptr->proctitle =
                                malloc(strlen(proc_state_ptr->process_name)+4);
                        if(proc_state_ptr->proctitle == NULL) {
                                return NULL;
                        }
                        sprintf(proc_state_ptr->proctitle, " (%s)",
                                proc_state_ptr->process_name);
                }

#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;
#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();
                /* This is in microseconds */
#ifdef NETBSD
                proc_state_ptr->time_spent =
                        kp_stats[i].kp_proc.p_rtime.tv_sec;
#else
                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 == NULL) {
                return NULL;
        }

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

        process_stat.total = ps_size;

        return &process_stat;
}
Revision:	1.33
Committed:	Sun Apr 4 11:28:46 2004 UTC (20 years, 1 month ago) by tdb
Content type:	text/plain
Branch:	MAIN
Changes since 1.32:	+11 -1 lines
Log Message:	This adds support for NetBSD, although it currently needs elevated privileges. I will try and fix that.
#	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.33	* $Id: process_stats.c,v 1.32 2004/04/04 11:02:30 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			#include <kvm.h>
65	pajs	1.6	#endif
66	pajs	1.1
67	pajs	1.23	int get_proc_snapshot(proc_state_t **ps){
68			proc_state_t *proc_state = NULL;
69			proc_state_t *proc_state_ptr;
70			int proc_state_size = 0;
71	tdb	1.26	#ifdef ALLBSD
72			int mib[3];
73			size_t size;
74			struct kinfo_proc *kp_stats;
75	tdb	1.29	int procs, i, alloc;
76			static kvm_t *kvmd;
77			char **args;
78			char *proctitle;
79	tdb	1.26	#endif
80	pajs	1.24	#if defined(SOLARIS) \|\| defined(LINUX)
81	pajs	1.23	DIR *proc_dir;
82			struct dirent *dir_entry;
83			char filename[MAX_FILE_LENGTH];
84			FILE *f;
85	pajs	1.24	#ifdef SOLARIS
86	pajs	1.5	psinfo_t process_info;
87	pajs	1.24	#endif
88	pajs	1.25	#ifdef LINUX
89			char s;
90			/* If someone has a executable of 4k filename length, they deserve to get it truncated :) */
91			char ps_name[4096];
92			char *ptr;
93			static char *psargs = NULL;
94			static int psarg_size = 0;
95			unsigned long stime, utime;
96			int x;
97			int fn;
98			int toread;
99			ssize_t size;
100			int t_read;
101			#endif
102	pajs	1.1
103	pajs	1.23	if((proc_dir=opendir(PROC_LOCATION))==NULL){
104	pajs	1.25	return -1;
105	pajs	1.23	}
106
107			while((dir_entry=readdir(proc_dir))!=NULL){
108			if(atoi(dir_entry->d_name) == 0) continue;
109
110	pajs	1.24	#ifdef SOLARIS
111	pajs	1.23	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
112	pajs	1.24	#endif
113	pajs	1.25	#ifdef LINUX
114			snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/stat", dir_entry->d_name);
115			#endif
116	pajs	1.23	if((f=fopen(filename, "r"))==NULL){
117			/* Open failed.. Process since vanished, or the path was too long.
118			* Ah well, move onwards to the next one */
119			continue;
120			}
121	pajs	1.24	#ifdef SOLARIS
122	pajs	1.23	fread(&process_info, sizeof(psinfo_t), 1, f);
123	pajs	1.25	#endif
124	pajs	1.1
125	pajs	1.23	proc_state = realloc(proc_state, (1+proc_state_size)*sizeof(proc_state_t));
126			proc_state_ptr = proc_state+proc_state_size;
127	pajs	1.25	#ifdef SOLARIS
128	pajs	1.23	proc_state_ptr->pid = process_info.pr_pid;
129			proc_state_ptr->parent = process_info.pr_ppid;
130			proc_state_ptr->pgid = process_info.pr_pgid;
131			proc_state_ptr->uid = process_info.pr_uid;
132			proc_state_ptr->euid = process_info.pr_euid;
133			proc_state_ptr->gid = process_info.pr_gid;
134			proc_state_ptr->egid = process_info.pr_egid;
135			proc_state_ptr->proc_size = (process_info.pr_size) * 1024;
136			proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
137			proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
138			proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
139			proc_state_ptr->process_name = strdup(process_info.pr_fname);
140			proc_state_ptr->proctitle = strdup(process_info.pr_psargs);
141	pajs	1.24
142			if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SLEEPING;
143			if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = RUNNING;
144			if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = ZOMBIE;
145			if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = STOPPED;
146			if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = RUNNING;
147	pajs	1.25	#endif
148			#ifdef LINUX
149			x = fscanf(f, "%d %4096s %c %d %d %d %d %d %lu %lu %lu %lu %lu %lu %lu %ld %ld %ld %d %ld %ld %lu %llu %llu %lu %lu %lu %lu %lu %lu %lu %lu %lu %lu %lu %lu %lu %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));
150			proc_state_ptr->proc_resident = proc_state_ptr->proc_resident * getpagesize();
151			if(s == 'S') proc_state_ptr->state = SLEEPING;
152			if(s == 'R') proc_state_ptr->state = RUNNING;
153			if(s == 'Z') proc_state_ptr->state = ZOMBIE;
154			if(s == 'T') proc_state_ptr->state = STOPPED;
155			if(s == 'D') proc_state_ptr->state = STOPPED;
156
157			/* pa_name[0] should = '(' */
158			ptr = strchr(&ps_name[1], ')');
159			if(ptr !=NULL) *ptr='\0';
160			proc_state_ptr->process_name = strdup(&ps_name[1]);
161
162			/* Need to do cpu */
163
164
165			/* proctitle */
166			snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/cmdline", dir_entry->d_name);
167
168			if((fn=open(filename, O_RDONLY)) == -1){
169			/* Open failed.. Process since vanished, or the path was too long.
170			* Ah well, move onwards to the next one */
171			continue;
172			}
173			#define PSARG_START_SIZE 128
174			if(psargs == NULL){
175			psargs = malloc(PSARG_START_SIZE);
176			psarg_size = PSARG_START_SIZE;
177			}
178			ptr = psargs;
179			t_read = 0;
180			toread = psarg_size;
181			while((size = read(fn, ptr, toread)) == toread){
182			psargs = realloc(psargs, (psarg_size + PSARG_START_SIZE));
183			ptr = psargs+psarg_size;
184			t_read = psarg_size;
185			psarg_size+=PSARG_START_SIZE;
186			toread = PSARG_START_SIZE;
187			}
188			if(size != -1) t_read+=size;
189
190			ptr = psargs;
191			for(x=0; x<t_read; x++){
192			if (ptr == '\0') ptr = ' ';
193			ptr++;
194			}
195			/* for safety sake */
196			psargs[t_read] = '\0';
197
198			proc_state_ptr->proctitle = strdup(psargs);
199
200	pajs	1.24	#endif
201	pajs	1.23
202			proc_state_size++;
203	pajs	1.1
204	pajs	1.23	fclose(f);
205			}
206			closedir(proc_dir);
207	tdb	1.26	#endif
208
209			#ifdef ALLBSD
210	tdb	1.33
211			#ifdef NETBSD
212			kvmd = kvm_openfiles(NULL, NULL, NULL, O_RDONLY, NULL);
213			#else
214	tdb	1.29	kvmd = kvm_openfiles(_PATH_DEVNULL, _PATH_DEVNULL, NULL, O_RDONLY, NULL);
215	tdb	1.33	#endif
216	tdb	1.26
217	tdb	1.29	if(kvmd == NULL) return NULL;
218	tdb	1.26
219	tdb	1.29	kp_stats = kvm_getprocs(kvmd, KERN_PROC_ALL, 0, &procs);
220	tdb	1.26
221	tdb	1.29	if (kp_stats == NULL \|\| procs < 0) {
222	tdb	1.26	return NULL;
223			}
224
225			for (i = 0; i < procs; i++) {
226	tdb	1.29	/* replace with something more sensible */
227	tdb	1.30	proc_state = realloc(proc_state,
228			(1+proc_state_size)*sizeof(proc_state_t));
229	tdb	1.29	if(proc_state == NULL ) {
230			return NULL;
231			}
232	tdb	1.26	proc_state_ptr = proc_state+proc_state_size;
233	tdb	1.30
234			#ifdef FREEBSD5
235			proc_state_ptr->process_name =
236			strdup(kp_stats[i].ki_comm);
237			#else
238			proc_state_ptr->process_name =
239			strdup(kp_stats[i].kp_proc.p_comm);
240			#endif
241	tdb	1.29
242			args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
243			if(args != NULL) {
244			alloc = 1;
245			proctitle = malloc(alloc);
246			if(proctitle == NULL) {
247			return NULL;
248			}
249			while(*args != NULL) {
250			if(strlen(proctitle) + strlen(*args) >= alloc) {
251			alloc = (alloc + strlen(*args)) << 1;
252			proctitle = realloc(proctitle, alloc);
253			if(proctitle == NULL) {
254			return NULL;
255			}
256			}
257			strncat(proctitle, args, strlen(args));
258			strncat(proctitle, " ", 1);
259			args++;
260			}
261			/* remove trailing space */
262			proctitle[strlen(proctitle)-1] = NULL;
263			proc_state_ptr->proctitle = proctitle;
264			}
265			else {
266	tdb	1.30	proc_state_ptr->proctitle =
267			malloc(strlen(proc_state_ptr->process_name)+4);
268	tdb	1.29	if(proc_state_ptr->proctitle == NULL) {
269			return NULL;
270			}
271	tdb	1.30	sprintf(proc_state_ptr->proctitle, " (%s)",
272			proc_state_ptr->process_name);
273	tdb	1.29	}
274	tdb	1.26
275	tdb	1.30	#ifdef FREEBSD5
276	tdb	1.26	proc_state_ptr->pid = kp_stats[i].ki_pid;
277			proc_state_ptr->parent = kp_stats[i].ki_ppid;
278			proc_state_ptr->pgid = kp_stats[i].ki_pgid;
279	tdb	1.30	#else
280			proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
281			proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
282			proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
283			#endif
284	tdb	1.26
285	tdb	1.30	#ifdef FREEBSD5
286	tdb	1.26	proc_state_ptr->uid = kp_stats[i].ki_ruid;
287			proc_state_ptr->euid = kp_stats[i].ki_uid;
288			proc_state_ptr->gid = kp_stats[i].ki_rgid;
289			proc_state_ptr->egid = kp_stats[i].ki_svgid;
290	tdb	1.30	#else
291			proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
292			proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
293			proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
294			proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
295			#endif
296	tdb	1.26
297	tdb	1.30	#ifdef FREEBSD5
298	tdb	1.26	proc_state_ptr->proc_size = kp_stats[i].ki_size;
299			/* This is in pages */
300	tdb	1.30	proc_state_ptr->proc_resident =
301			kp_stats[i].ki_rssize * getpagesize();
302	tdb	1.28	/* This is in microseconds */
303	tdb	1.26	proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
304	tdb	1.30	proc_state_ptr->cpu_percent =
305			((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
306	tdb	1.26	proc_state_ptr->nice = kp_stats[i].ki_nice;
307	tdb	1.30	#else
308			proc_state_ptr->proc_size =
309			kp_stats[i].kp_eproc.e_vm.vm_map.size;
310			/* This is in pages */
311			proc_state_ptr->proc_resident =
312			kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
313			/* This is in microseconds */
314	tdb	1.33	#ifdef NETBSD
315			proc_state_ptr->time_spent =
316			kp_stats[i].kp_proc.p_rtime.tv_sec;
317			#else
318	tdb	1.30	proc_state_ptr->time_spent =
319			kp_stats[i].kp_proc.p_runtime / 1000000;
320	tdb	1.33	#endif
321	tdb	1.30	proc_state_ptr->cpu_percent =
322			((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
323			proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
324			#endif
325	tdb	1.26
326	tdb	1.30	#ifdef FREEBSD5
327	tdb	1.26	switch (kp_stats[i].ki_stat) {
328	tdb	1.30	#else
329			switch (kp_stats[i].kp_proc.p_stat) {
330			#endif
331	tdb	1.26	case SIDL:
332			case SRUN:
333			#ifdef SONPROC
334			case SONPROC: /* NetBSD */
335			#endif
336			proc_state_ptr->state = RUNNING;
337			break;
338			case SSLEEP:
339			#ifdef SWAIT
340			case SWAIT: /* FreeBSD 5 */
341			#endif
342			#ifdef SLOCK
343			case SLOCK: /* FreeBSD 5 */
344			#endif
345			proc_state_ptr->state = SLEEPING;
346			break;
347			case SSTOP:
348			proc_state_ptr->state = STOPPED;
349			break;
350			case SZOMB:
351			#ifdef SDEAD
352			case SDEAD: /* OpenBSD & NetBSD */
353			#endif
354			proc_state_ptr->state = ZOMBIE;
355	tdb	1.31	break;
356			default:
357			proc_state_ptr->state = UNKNOWN;
358	tdb	1.26	break;
359			}
360			proc_state_size++;
361			}
362
363			free(kp_stats);
364			#endif
365
366	pajs	1.23	*ps = proc_state;
367			return proc_state_size;
368	tdb	1.27	}
369
370			process_stat_t *get_process_stats() {
371			static process_stat_t process_stat;
372			proc_state_t *ps;
373			int ps_size, x;
374
375	tdb	1.30	process_stat.sleeping = 0;
376			process_stat.running = 0;
377			process_stat.zombie = 0;
378			process_stat.stopped = 0;
379			process_stat.total = 0;
380	tdb	1.27
381			ps_size = get_proc_snapshot(&ps);
382	tdb	1.32
383			if(ps_size == NULL) {
384			return NULL;
385			}
386	tdb	1.27
387			for(x = 0; x < ps_size; x++) {
388			switch (ps->state) {
389	tdb	1.30	/* currently no mapping for UNKNOWN in process_stat_t */
390	tdb	1.27	case RUNNING:
391			process_stat.running++;
392			break;
393			case SLEEPING:
394			process_stat.sleeping++;
395			break;
396			case STOPPED:
397			process_stat.stopped++;
398			break;
399			case ZOMBIE:
400			process_stat.zombie++;
401			break;
402			}
403			ps++;
404			}
405
406			process_stat.total = ps_size;
407
408			return &process_stat;
409	pajs	1.23	}