ViewVC Help

View File | Revision Log | Show Annotations | Revision Graph | Root Listing

root/i-scream/projects/libstatgrab/src/libstatgrab/process_stats.c

Comparing projects/libstatgrab/src/libstatgrab/process_stats.c (file contents):
Revision 1.14 by tdb, Mon Nov 10 21:07:04 2003 UTC vs.
Revision 1.75 by ats, Fri Sep 10 15:15:00 2004 UTC

#	Line 1 | Line 1
1	<	/*
2	<	* i-scream central monitoring system
1	>	/*
2	>	* i-scream libstatgrab
3		* http://www.i-scream.org
4	<	* Copyright (C) 2000-2003 i-scream
4	>	* Copyright (C) 2000-2004 i-scream
5		*
6	<	* This program is free software; you can redistribute it and/or
7	<	* modify it under the terms of the GNU General Public License
8	<	* as published by the Free Software Foundation; either version 2
9	<	* of the License, or (at your option) any later version.
6	>	* 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		*
11	<	* This program is distributed in the hope that it will be useful,
11	>	* This library is distributed in the hope that it will be useful,
12		* but WITHOUT ANY WARRANTY; without even the implied warranty of
13	<	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14	<	* GNU General Public License for more details.
13	>	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14	>	* Lesser General Public License for more details.
15		*
16	<	* You should have received a copy of the GNU General Public License
17	<	* along with this program; if not, write to the Free Software
18	<	* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
16	>	* 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	>	*
21	>	* $Id$
22		*/
23
24		#ifdef HAVE_CONFIG_H
#	Line 23 | Line 26
26		#endif
27
28		#include "statgrab.h"
29	+	#include "tools.h"
30	+	#include "vector.h"
31		#if defined(SOLARIS) || defined(LINUX)
32		#include <stdio.h>
33		#include <stdlib.h>
#	Line 31 | Line 36
36		#include <string.h>
37		#endif
38
34	–	#include "tools.h"
39		#ifdef SOLARIS
40		#include <procfs.h>
41		#include <limits.h>
#	Line 39 | Line 43
43		#define MAX_FILE_LENGTH PATH_MAX
44		#endif
45		#ifdef LINUX
42	–	#ifdef CYGWIN
46		#include <limits.h>
47	<	#else
48	<	#include <linux/limits.h>
49	<	#endif
47	>	#include <unistd.h>
48	>	#include <sys/stat.h>
49	>	#include <fcntl.h>
50		#define PROC_LOCATION "/proc"
51		#define MAX_FILE_LENGTH PATH_MAX
52		#endif
53		#ifdef ALLBSD
54	<	#include <kvm.h>
54	>	#include <errno.h>
55	>	#include <stdlib.h>
56		#include <sys/param.h>
57		#include <sys/sysctl.h>
58	<	#endif
55	<	#ifdef FREEBSD
58	>	#if defined(FREEBSD) || defined(DFBSD)
59		#include <sys/user.h>
60	+	#else
61	+	#include <sys/proc.h>
62		#endif
63	+	#include <string.h>
64	+	#include <paths.h>
65	+	#include <fcntl.h>
66	+	#include <limits.h>
67	+	#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
68	+	#include <kvm.h>
69	+	#endif
70	+	#include <unistd.h>
71	+	#ifdef NETBSD2
72	+	#include <sys/lwp.h>
73	+	#endif
74	+	#endif
75
76	<	process_stat_t *get_process_stats(){
76	>	static void proc_state_init(sg_process_stats *s) {
77	>	s->process_name = NULL;
78	>	s->proctitle = NULL;
79	>	}
80
81	<	static process_stat_t process_stat;
81	>	static void proc_state_destroy(sg_process_stats *s) {
82	>	free(s->process_name);
83	>	free(s->proctitle);
84	>	}
85
86	+	sg_process_stats *sg_get_process_stats(int *entries){
87	+	VECTOR_DECLARE_STATIC(proc_state, sg_process_stats, 64,
88	+	proc_state_init, proc_state_destroy);
89	+	int proc_state_size = 0;
90	+	sg_process_stats *proc_state_ptr;
91	+	#ifdef ALLBSD
92	+	int mib[4];
93	+	size_t size;
94	+	struct kinfo_proc *kp_stats;
95	+	int procs, i;
96	+	char *proctitle;
97	+	#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
98	+	kvm_t *kvmd;
99	+	char **args, **argsp;
100	+	int argslen = 0;
101	+	#else
102	+	long buflen;
103	+	char *p, *proctitletmp;
104	+	#endif
105	+	#ifdef NETBSD2
106	+	int lwps;
107	+	struct kinfo_lwp *kl_stats;
108	+	#endif
109	+	#endif
110		#if defined(SOLARIS) || defined(LINUX)
111	<	DIR *proc_dir;
111	>	DIR *proc_dir;
112		struct dirent *dir_entry;
113		char filename[MAX_FILE_LENGTH];
114		FILE *f;
68	–	#endif
69	–	#ifdef LINUX
70	–	char *line_ptr;
71	–	#endif
115		#ifdef SOLARIS
116		psinfo_t process_info;
117		#endif
118	<	#ifdef ALLBSD
119	<	struct kinfo_proc *kp_stats;
120	<	kvm_t *kvmd;
121	<	int procs;
118	>	#ifdef LINUX
119	>	char s;
120	>	/* If someone has a executable of 4k filename length, they deserve to get it truncated :) */
121	>	char ps_name[4096];
122	>	char *ptr;
123	>	VECTOR_DECLARE_STATIC(psargs, char, 128, NULL, NULL);
124	>	unsigned long stime, utime, starttime;
125	>	int x;
126	>	int fn;
127	>	int len;
128	>	int rc;
129	>	time_t uptime;
130		#endif
131
132	<	process_stat.sleeping=0;
133	<	process_stat.running=0;
134	<	process_stat.zombie=0;
135	<	process_stat.stopped=0;
136	<	process_stat.total=0;
132	>	#ifdef LINUX
133	>	if ((f=fopen("/proc/uptime", "r")) == NULL) {
134	>	sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/uptime");
135	>	return NULL;
136	>	}
137	>	if((fscanf(f,"%lu %*d",&uptime)) != 1){
138	>	sg_set_error(SG_ERROR_PARSE, NULL);
139	>	return NULL;
140	>	}
141	>	fclose(f);
142	>	#endif
143
87	–	#if defined(SOLARIS) || defined(LINUX)
144		if((proc_dir=opendir(PROC_LOCATION))==NULL){
145	+	sg_set_error_with_errno(SG_ERROR_OPENDIR, PROC_LOCATION);
146		return NULL;
147		}
148
#	Line 96 | Line 153 | process_stat_t *get_process_stats(){
153		snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
154		#endif
155		#ifdef LINUX
156	<	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/status", dir_entry->d_name);
156	>	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/stat", dir_entry->d_name);
157		#endif
101	–
158		if((f=fopen(filename, "r"))==NULL){
159	<	/* Open failed.. Process since vanished, or the path was too long.
159	>	/* Open failed.. Process since vanished, or the path was too long.
160		* Ah well, move onwards to the next one */
161		continue;
162		}
107	–
163		#ifdef SOLARIS
164		fread(&process_info, sizeof(psinfo_t), 1, f);
165	<	if(process_info.pr_lwp.pr_state==1) process_stat.sleeping++;
111	<	if(process_info.pr_lwp.pr_state==2) process_stat.running++;
112	<	if(process_info.pr_lwp.pr_state==3) process_stat.zombie++;
113	<	if(process_info.pr_lwp.pr_state==4) process_stat.stopped++;
114	<	if(process_info.pr_lwp.pr_state==6) process_stat.running++;
165	>	fclose(f);
166		#endif
167	+
168	+	if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
169	+	return NULL;
170	+	}
171	+	proc_state_ptr = proc_state+proc_state_size;
172	+
173	+	#ifdef SOLARIS
174	+	proc_state_ptr->pid = process_info.pr_pid;
175	+	proc_state_ptr->parent = process_info.pr_ppid;
176	+	proc_state_ptr->pgid = process_info.pr_pgid;
177	+	proc_state_ptr->uid = process_info.pr_uid;
178	+	proc_state_ptr->euid = process_info.pr_euid;
179	+	proc_state_ptr->gid = process_info.pr_gid;
180	+	proc_state_ptr->egid = process_info.pr_egid;
181	+	proc_state_ptr->proc_size = (process_info.pr_size) * 1024;
182	+	proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
183	+	proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
184	+	proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
185	+	proc_state_ptr->nice = (int)process_info.pr_lwp.pr_nice - 20;
186	+	if (sg_update_string(&proc_state_ptr->process_name,
187	+	process_info.pr_fname) < 0) {
188	+	return NULL;
189	+	}
190	+	if (sg_update_string(&proc_state_ptr->proctitle,
191	+	process_info.pr_psargs) < 0) {
192	+	return NULL;
193	+	}
194	+
195	+	switch (process_info.pr_lwp.pr_state) {
196	+	case 1:
197	+	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
198	+	break;
199	+	case 2:
200	+	case 5:
201	+	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
202	+	break;
203	+	case 3:
204	+	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
205	+	break;
206	+	case 4:
207	+	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
208	+	break;
209	+	}
210	+	#endif
211		#ifdef LINUX
212	<	if((line_ptr=f_read_line(f, "State:"))==NULL){
213	<	fclose(f);
212	>	x = fscanf(f, "%d %4096s %c %d %d %*d %*d %*d %*u %*u %*u %*u %*u %lu %lu %*d %*d %*d %d %*d %*d %lu %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), &starttime, &(proc_state_ptr->proc_size), &(proc_state_ptr->proc_resident));
213	>	/* +3 becuase man page says "Resident  Set Size: number of pages the process has in real memory, minus 3 for administrative purposes." */
214	>	proc_state_ptr->proc_resident = (proc_state_ptr->proc_resident + 3) * getpagesize();
215	>	switch (s) {
216	>	case 'S':
217	>	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
218	>	break;
219	>	case 'R':
220	>	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
221	>	break;
222	>	case 'Z':
223	>	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
224	>	break;
225	>	case 'T':
226	>	case 'D':
227	>	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
228	>	break;
229	>	}
230	>
231	>	/* pa_name[0] should = '(' */
232	>	ptr = strchr(&ps_name[1], ')');
233	>	if(ptr !=NULL) *ptr='\0';
234	>
235	>	if (sg_update_string(&proc_state_ptr->process_name,
236	>	&ps_name[1]) < 0) {
237	>	return NULL;
238	>	}
239	>
240	>	/* cpu */
241	>	proc_state_ptr->cpu_percent = (100.0 * (utime + stime)) / ((uptime * 100.0) - starttime);
242	>
243	>	fclose(f);
244	>
245	>	/* uid / gid */
246	>	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/status", dir_entry->d_name);
247	>	if ((f=fopen(filename, "r")) == NULL) {
248	>	/* Open failed.. Process since vanished, or the path was too long.
249	>	* Ah well, move onwards to the next one */
250		continue;
251		}
252	<	if((line_ptr=strchr(line_ptr, '\t'))==NULL){
252	>
253	>	if((ptr=sg_f_read_line(f, "Uid:"))==NULL){
254		fclose(f);
255		continue;
256		}
257	<	line_ptr++;
258	<	if(line_ptr=='\0'){
257	>	sscanf(ptr, "Uid:\t%d\t%d\t%*d\t%*d\n", &(proc_state_ptr->uid), &(proc_state_ptr->euid));
258	>
259	>	if((ptr=sg_f_read_line(f, "Gid:"))==NULL){
260		fclose(f);
261		continue;
262		}
263	+	sscanf(ptr, "Gid:\t%d\t%d\t%*d\t%*d\n", &(proc_state_ptr->gid), &(proc_state_ptr->egid));
264
131	–	if(*line_ptr=='S') process_stat.sleeping++;
132	–	if(*line_ptr=='R') process_stat.running++;
133	–	if(*line_ptr=='Z') process_stat.zombie++;
134	–	if(*line_ptr=='T') process_stat.stopped++;
135	–	if(*line_ptr=='D') process_stat.stopped++;
136	–	#endif
137	–
265		fclose(f);
266	+
267	+	/* proctitle */
268	+	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/cmdline", dir_entry->d_name);
269	+
270	+	if((fn=open(filename, O_RDONLY)) == -1){
271	+	/* Open failed.. Process since vanished, or the path was too long.
272	+	* Ah well, move onwards to the next one */
273	+	continue;
274	+	}
275	+
276	+	#define READ_BLOCK_SIZE 128
277	+	len = 0;
278	+	do {
279	+	if (VECTOR_RESIZE(psargs, len + READ_BLOCK_SIZE) < 0) {
280	+	return NULL;
281	+	}
282	+	rc = read(fn, psargs + len, READ_BLOCK_SIZE);
283	+	if (rc > 0) {
284	+	len += rc;
285	+	}
286	+	} while (rc == READ_BLOCK_SIZE);
287	+	close(fn);
288	+
289	+	if (rc == -1) {
290	+	/* Read failed; move on. */
291	+	continue;
292	+	}
293	+
294	+	/* Turn \0s into spaces within the command line. */
295	+	ptr = psargs;
296	+	for(x = 0; x < len; x++) {
297	+	if (*ptr == '\0') *ptr = ' ';
298	+	ptr++;
299	+	}
300	+
301	+	if (len == 0) {
302	+	/* We want psargs to be NULL. */
303	+	if (VECTOR_RESIZE(psargs, 0) < 0) {
304	+	return NULL;
305	+	}
306	+	} else {
307	+	/* Not empty, so append a \0. */
308	+	if (VECTOR_RESIZE(psargs, len + 1) < 0) {
309	+	return NULL;
310	+	}
311	+	psargs[len] = '\0';
312	+	}
313	+
314	+	if (sg_update_string(&proc_state_ptr->proctitle, psargs) < 0) {
315	+	return NULL;
316	+	}
317	+	#endif
318	+
319	+	proc_state_size++;
320		}
321		closedir(proc_dir);
322		#endif
323	+
324		#ifdef ALLBSD
325	<	if((kvmd = get_kvm()) == NULL){
325	>	mib[0] = CTL_KERN;
326	>	mib[1] = KERN_PROC;
327	>	mib[2] = KERN_PROC_ALL;
328	>
329	>	if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
330	>	sg_set_error_with_errno(SG_ERROR_SYSCTL,
331	>	"CTL_KERN.KERN_PROC.KERN_PROC_ALL");
332		return NULL;
333		}
334
335	<	kp_stats=kvm_getprocs(kvmd, KERN_PROC_ALL, 0, &procs);
335	>	procs = size / sizeof(struct kinfo_proc);
336
337	<	while(procs--){
337	>	kp_stats = sg_malloc(size);
338	>	if(kp_stats == NULL) {
339	>	return NULL;
340	>	}
341	>	memset(kp_stats, 0, size);
342	>
343	>	if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
344	>	sg_set_error_with_errno(SG_ERROR_SYSCTL,
345	>	"CTL_KERN.KERN_PROC.KERN_PROC_ALL");
346	>	free(kp_stats);
347	>	return NULL;
348	>	}
349	>
350	>	#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
351	>	kvmd = sg_get_kvm2();
352	>	#endif
353	>
354	>	for (i = 0; i < procs; i++) {
355	>	const char *name;
356	>
357		#ifdef FREEBSD5
358	<	if (kp_stats[procs].ki_stat == SSLEEP) process_stat.sleeping++;
152	<	if (kp_stats[procs].ki_stat == SWAIT) process_stat.sleeping++;
153	<	if (kp_stats[procs].ki_stat == SLOCK) process_stat.sleeping++;
154	<	if (kp_stats[procs].ki_stat == SRUN) process_stat.running++;
155	<	if (kp_stats[procs].ki_stat == SIDL) process_stat.running++;
156	<	if (kp_stats[procs].ki_stat == SZOMB) process_stat.zombie++;
157	<	if (kp_stats[procs].ki_stat == SSTOP) process_stat.stopped++;
158	<
358	>	if (kp_stats[i].ki_stat == 0) {
359		#else
360	<	if (kp_stats[procs].kp_proc.p_stat == SSLEEP) process_stat.sleeping++;
161	<	if (kp_stats[procs].kp_proc.p_stat == SRUN) process_stat.running++;
162	<	if (kp_stats[procs].kp_proc.p_stat == SIDL) process_stat.running++;
163	<	if (kp_stats[procs].kp_proc.p_stat == SZOMB) process_stat.zombie++;
164	<	if (kp_stats[procs].kp_proc.p_stat == SSTOP) process_stat.stopped++;
360	>	if (kp_stats[i].kp_proc.p_stat == 0) {
361		#endif
362	+	/* FreeBSD 5 deliberately overallocates the array that
363	+	* the sysctl returns, so we'll get a few junk
364	+	* processes on the end that we have to ignore. (Search
365	+	* for "overestimate by 5 procs" in
366	+	* src/sys/kern/kern_proc.c for more details.) */
367	+	continue;
368	+	}
369	+
370	+	if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
371	+	return NULL;
372	+	}
373	+	proc_state_ptr = proc_state+proc_state_size;
374	+
375	+	#ifdef FREEBSD5
376	+	name = kp_stats[i].ki_comm;
377	+	#elif defined(DFBSD)
378	+	name = kp_stats[i].kp_thread.td_comm;
379	+	#else
380	+	name = kp_stats[i].kp_proc.p_comm;
381	+	#endif
382	+	if (sg_update_string(&proc_state_ptr->process_name, name) < 0) {
383	+	return NULL;
384	+	}
385	+
386	+	#if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD)
387	+
388	+	#ifdef FREEBSD5
389	+	mib[2] = KERN_PROC_ARGS;
390	+	mib[3] = kp_stats[i].ki_pid;
391	+	#else
392	+	mib[1] = KERN_PROC_ARGS;
393	+	mib[2] = kp_stats[i].kp_proc.p_pid;
394	+	mib[3] = KERN_PROC_ARGV;
395	+	#endif
396	+
397	+	free(proc_state_ptr->proctitle);
398	+	proc_state_ptr->proctitle = NULL;
399	+
400	+	/* Starting size - we'll double this straight away */
401	+	#define PROCTITLE_START_SIZE 64
402	+	buflen = PROCTITLE_START_SIZE;
403	+	size = buflen;
404	+	proctitle = NULL;
405	+
406	+	do {
407	+	if(size >= buflen) {
408	+	buflen *= 2;
409	+	size = buflen;
410	+	proctitletmp = sg_realloc(proctitle, buflen);
411	+	if(proctitletmp == NULL) {
412	+	free(proctitle);
413	+	proctitle = NULL;
414	+	proc_state_ptr->proctitle = NULL;
415	+	size = 0;
416	+	break;
417	+	}
418	+	proctitle = proctitletmp;
419	+	bzero(proctitle, buflen);
420	+	}
421	+
422	+	if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
423	+	free(proctitle);
424	+	proctitle = NULL;
425	+	proc_state_ptr->proctitle = NULL;
426	+	size = 0;
427	+	break;
428	+	}
429	+	} while(size >= buflen);
430	+
431	+	if(size > 0) {
432	+	proc_state_ptr->proctitle = sg_malloc(size+1);
433	+	if(proc_state_ptr->proctitle == NULL) {
434	+	return NULL;
435	+	}
436	+	p = proctitle;
437	+	#ifdef OPENBSD
438	+	/* On OpenBSD, this value has the argv pointers (which
439	+	* are terminated by a NULL) at the front, so we have
440	+	* to skip over them to get to the strings. */
441	+	while (*(char ***)p != NULL) {
442	+	p += sizeof(char **);
443	+	}
444	+	p += sizeof(char **);
445	+	#endif
446	+	proc_state_ptr->proctitle[0] = '\0';
447	+	do {
448	+	sg_strlcat(proc_state_ptr->proctitle, p, size+1);
449	+	sg_strlcat(proc_state_ptr->proctitle, " ", size+1);
450	+	p += strlen(p) + 1;
451	+	} while (p < proctitle + size);
452	+	free(proctitle);
453	+	proctitle = NULL;
454	+	/* remove trailing space */
455	+	proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = '\0';
456	+	}
457	+	else {
458	+	if(proctitle != NULL) {
459	+	free(proctitle);
460	+	proctitle = NULL;
461	+	}
462	+	proc_state_ptr->proctitle = NULL;
463	+	}
464	+	#else
465	+	free(proc_state_ptr->proctitle);
466	+	proc_state_ptr->proctitle = NULL;
467	+	if(kvmd != NULL) {
468	+	args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
469	+	if(args != NULL) {
470	+	argsp = args;
471	+	while(*argsp != NULL) {
472	+	argslen += strlen(*argsp) + 1;
473	+	argsp++;
474	+	}
475	+	proctitle = sg_malloc(argslen + 1);
476	+	proctitle[0] = '\0';
477	+	if(proctitle == NULL) {
478	+	return NULL;
479	+	}
480	+	while(*args != NULL) {
481	+	sg_strlcat(proctitle, *args, argslen + 1);
482	+	sg_strlcat(proctitle, " ", argslen + 1);
483	+	args++;
484	+	}
485	+	/* remove trailing space */
486	+	proctitle[strlen(proctitle)-1] = '\0';
487	+	proc_state_ptr->proctitle = proctitle;
488	+	}
489	+	else {
490	+	proc_state_ptr->proctitle = NULL;
491	+	}
492	+	}
493	+	else {
494	+	proc_state_ptr->proctitle = NULL;
495	+	}
496	+	#endif
497	+
498	+	#ifdef FREEBSD5
499	+	proc_state_ptr->pid = kp_stats[i].ki_pid;
500	+	proc_state_ptr->parent = kp_stats[i].ki_ppid;
501	+	proc_state_ptr->pgid = kp_stats[i].ki_pgid;
502	+	#else
503	+	proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
504	+	proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
505	+	proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
506	+	#endif
507	+
508	+	#ifdef FREEBSD5
509	+	proc_state_ptr->uid = kp_stats[i].ki_ruid;
510	+	proc_state_ptr->euid = kp_stats[i].ki_uid;
511	+	proc_state_ptr->gid = kp_stats[i].ki_rgid;
512	+	proc_state_ptr->egid = kp_stats[i].ki_svgid;
513	+	#elif defined(DFBSD)
514	+	proc_state_ptr->uid = kp_stats[i].kp_eproc.e_ucred.cr_ruid;
515	+	proc_state_ptr->euid = kp_stats[i].kp_eproc.e_ucred.cr_svuid;
516	+	proc_state_ptr->gid = kp_stats[i].kp_eproc.e_ucred.cr_rgid;
517	+	proc_state_ptr->egid = kp_stats[i].kp_eproc.e_ucred.cr_svgid;
518	+	#else
519	+	proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
520	+	proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
521	+	proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
522	+	proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
523	+	#endif
524	+
525	+	#ifdef FREEBSD5
526	+	proc_state_ptr->proc_size = kp_stats[i].ki_size;
527	+	/* This is in pages */
528	+	proc_state_ptr->proc_resident =
529	+	kp_stats[i].ki_rssize * getpagesize();
530	+	/* This is in microseconds */
531	+	proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
532	+	proc_state_ptr->cpu_percent =
533	+	((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
534	+	proc_state_ptr->nice = kp_stats[i].ki_nice;
535	+	#else
536	+	proc_state_ptr->proc_size =
537	+	kp_stats[i].kp_eproc.e_vm.vm_map.size;
538	+	/* This is in pages */
539	+	proc_state_ptr->proc_resident =
540	+	kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
541	+	#if defined(NETBSD) || defined(OPENBSD)
542	+	proc_state_ptr->time_spent =
543	+	kp_stats[i].kp_proc.p_rtime.tv_sec;
544	+	#elif defined(DFBSD)
545	+	proc_state_ptr->time_spent =
546	+	( kp_stats[i].kp_thread.td_uticks +
547	+	kp_stats[i].kp_thread.td_sticks +
548	+	kp_stats[i].kp_thread.td_iticks ) / 1000000;
549	+	#else
550	+	/* This is in microseconds */
551	+	proc_state_ptr->time_spent =
552	+	kp_stats[i].kp_proc.p_runtime / 1000000;
553	+	#endif
554	+	proc_state_ptr->cpu_percent =
555	+	((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
556	+	proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
557	+	#endif
558	+
559	+	#ifdef NETBSD2
560	+	{
561	+	size_t size;
562	+	int mib[5];
563	+
564	+	mib[0] = CTL_KERN;
565	+	mib[1] = KERN_LWP;
566	+	mib[2] = kp_stats[i].kp_proc.p_pid;
567	+	mib[3] = sizeof(struct kinfo_lwp);
568	+	mib[4] = 0;
569	+
570	+	if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) {
571	+	sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.0");
572	+	return NULL;
573	+	}
574	+
575	+	lwps = size / sizeof(struct kinfo_lwp);
576	+	mib[4] = lwps;
577	+
578	+	kl_stats = sg_malloc(size);
579	+	if(kl_stats == NULL) {
580	+	return NULL;
581	+	}
582	+
583	+	if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) {
584	+	sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.buffersize");
585	+	return NULL;
586	+	}
587	+	}
588	+
589	+	switch(kp_stats[i].kp_proc.p_stat) {
590	+	case SIDL:
591	+	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
592	+	break;
593	+	case SACTIVE:
594	+	{
595	+	int i;
596	+
597	+	for(i = 0; i < lwps; i++) {
598	+	switch(kl_stats[i].l_stat) {
599	+	case LSONPROC:
600	+	case LSRUN:
601	+	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
602	+	goto end;
603	+	case LSSLEEP:
604	+	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
605	+	goto end;
606	+	case LSSTOP:
607	+	case LSSUSPENDED:
608	+	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
609	+	goto end;
610	+	}
611	+	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
612	+	}
613	+	end: ;
614	+	}
615	+	break;
616	+	case SSTOP:
617	+	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
618	+	break;
619	+	case SZOMB:
620	+	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
621	+	break;
622	+	default:
623	+	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
624	+	break;
625	+	}
626	+
627	+	free(kl_stats);
628	+	#else
629	+	#ifdef FREEBSD5
630	+	switch (kp_stats[i].ki_stat) {
631	+	#else
632	+	switch (kp_stats[i].kp_proc.p_stat) {
633	+	#endif
634	+	case SIDL:
635	+	case SRUN:
636	+	#ifdef SONPROC
637	+	case SONPROC: /* NetBSD */
638	+	#endif
639	+	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
640	+	break;
641	+	case SSLEEP:
642	+	#ifdef SWAIT
643	+	case SWAIT: /* FreeBSD 5 */
644	+	#endif
645	+	#ifdef SLOCK
646	+	case SLOCK: /* FreeBSD 5 */
647	+	#endif
648	+	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
649	+	break;
650	+	case SSTOP:
651	+	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
652	+	break;
653	+	case SZOMB:
654	+	#ifdef SDEAD
655	+	case SDEAD: /* OpenBSD & NetBSD */
656	+	#endif
657	+	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
658	+	break;
659	+	default:
660	+	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
661	+	break;
662	+	}
663	+	#endif
664	+	proc_state_size++;
665		}
666	+
667	+	free(kp_stats);
668		#endif
669
670	<	process_stat.total=process_stat.sleeping+process_stat.running+process_stat.zombie+process_stat.stopped;
670	>	#ifdef CYGWIN
671	>	sg_set_error(SG_ERROR_UNSUPPORTED, "Cygwin");
672	>	return NULL;
673	>	#endif
674
675	+	*entries = proc_state_size;
676	+	return proc_state;
677	+	}
678	+
679	+	sg_process_count *sg_get_process_count() {
680	+	static sg_process_count process_stat;
681	+	sg_process_stats *ps;
682	+	int ps_size, x;
683	+
684	+	process_stat.sleeping = 0;
685	+	process_stat.running = 0;
686	+	process_stat.zombie = 0;
687	+	process_stat.stopped = 0;
688	+	process_stat.total = 0;
689	+
690	+	ps = sg_get_process_stats(&ps_size);
691	+	if (ps == NULL) {
692	+	return NULL;
693	+	}
694	+
695	+	for(x = 0; x < ps_size; x++) {
696	+	switch (ps->state) {
697	+	case SG_PROCESS_STATE_RUNNING:
698	+	process_stat.running++;
699	+	break;
700	+	case SG_PROCESS_STATE_SLEEPING:
701	+	process_stat.sleeping++;
702	+	break;
703	+	case SG_PROCESS_STATE_STOPPED:
704	+	process_stat.stopped++;
705	+	break;
706	+	case SG_PROCESS_STATE_ZOMBIE:
707	+	process_stat.zombie++;
708	+	break;
709	+	default:
710	+	/* currently no mapping for SG_PROCESS_STATE_UNKNOWN in
711	+	* sg_process_count */
712	+	break;
713	+	}
714	+	ps++;
715	+	}
716	+
717	+	process_stat.total = ps_size;
718	+
719		return &process_stat;
720		}
721	+
722	+	int sg_process_compare_name(const void *va, const void *vb) {
723	+	const sg_process_stats *a = (sg_process_stats *)va;
724	+	const sg_process_stats *b = (sg_process_stats *)vb;
725	+
726	+	return strcmp(a->process_name, b->process_name);
727	+	}
728	+
729	+	int sg_process_compare_pid(const void *va, const void *vb) {
730	+	const sg_process_stats *a = (sg_process_stats *)va;
731	+	const sg_process_stats *b = (sg_process_stats *)vb;
732	+
733	+	if (a->pid < b->pid) {
734	+	return -1;
735	+	} else if (a->pid == b->pid) {
736	+	return 0;
737	+	} else {
738	+	return 1;
739	+	}
740	+	}
741	+
742	+	int sg_process_compare_uid(const void *va, const void *vb) {
743	+	const sg_process_stats *a = (sg_process_stats *)va;
744	+	const sg_process_stats *b = (sg_process_stats *)vb;
745	+
746	+	if (a->uid < b->uid) {
747	+	return -1;
748	+	} else if (a->uid == b->uid) {
749	+	return 0;
750	+	} else {
751	+	return 1;
752	+	}
753	+	}
754	+
755	+	int sg_process_compare_gid(const void *va, const void *vb) {
756	+	const sg_process_stats *a = (sg_process_stats *)va;
757	+	const sg_process_stats *b = (sg_process_stats *)vb;
758	+
759	+	if (a->gid < b->gid) {
760	+	return -1;
761	+	} else if (a->gid == b->gid) {
762	+	return 0;
763	+	} else {
764	+	return 1;
765	+	}
766	+	}
767	+
768	+	int sg_process_compare_size(const void *va, const void *vb) {
769	+	const sg_process_stats *a = (sg_process_stats *)va;
770	+	const sg_process_stats *b = (sg_process_stats *)vb;
771	+
772	+	if (a->proc_size < b->proc_size) {
773	+	return -1;
774	+	} else if (a->proc_size == b->proc_size) {
775	+	return 0;
776	+	} else {
777	+	return 1;
778	+	}
779	+	}
780	+
781	+	int sg_process_compare_res(const void *va, const void *vb) {
782	+	const sg_process_stats *a = (sg_process_stats *)va;
783	+	const sg_process_stats *b = (sg_process_stats *)vb;
784	+
785	+	if (a->proc_resident < b->proc_resident) {
786	+	return -1;
787	+	} else if (a->proc_resident == b->proc_resident) {
788	+	return 0;
789	+	} else {
790	+	return 1;
791	+	}
792	+	}
793	+
794	+	int sg_process_compare_cpu(const void *va, const void *vb) {
795	+	const sg_process_stats *a = (sg_process_stats *)va;
796	+	const sg_process_stats *b = (sg_process_stats *)vb;
797	+
798	+	if (a->cpu_percent < b->cpu_percent) {
799	+	return -1;
800	+	} else if (a->cpu_percent == b->cpu_percent) {
801	+	return 0;
802	+	} else {
803	+	return 1;
804	+	}
805	+	}
806	+
807	+	int sg_process_compare_time(const void *va, const void *vb) {
808	+	const sg_process_stats *a = (sg_process_stats *)va;
809	+	const sg_process_stats *b = (sg_process_stats *)vb;
810	+
811	+	if (a->time_spent < b->time_spent) {
812	+	return -1;
813	+	} else if (a->time_spent == b->time_spent) {
814	+	return 0;
815	+	} else {
816	+	return 1;
817	+	}
818	+	}
819	+

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines