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.51 by tdb, Mon Apr 5 11:21:49 2004 UTC vs.
Revision 1.60 by ats, Wed Apr 7 10:27:53 2004 UTC

#	Line 1 | Line 1
1		/*
2	<	* i-scream central monitoring system
2	>	* i-scream libstatgrab
3		* http://www.i-scream.org
4		* Copyright (C) 2000-2004 i-scream
5		*
#	Line 65 | Line 65
65		#include <limits.h>
66		#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
67		#include <kvm.h>
68	–	#include <tools.h>
68		#endif
69		#include <unistd.h>
70	+	#ifdef NETBSD2
71	+	#include <sys/lwp.h>
72		#endif
73	+	#endif
74
75	<	static void proc_state_init(proc_state_t *s) {
75	>	static void proc_state_init(sg_process_stats *s) {
76		s->process_name = NULL;
77		s->proctitle = NULL;
78		}
79
80	<	static void proc_state_destroy(proc_state_t *s) {
80	>	static void proc_state_destroy(sg_process_stats *s) {
81		free(s->process_name);
82		free(s->proctitle);
83		}
84
85	<	int get_proc_snapshot(proc_state_t **ps){
86	<	VECTOR_DECLARE_STATIC(proc_state, proc_state_t, 64,
85	>	sg_process_stats *sg_get_process_stats(int *entries){
86	>	VECTOR_DECLARE_STATIC(proc_state, sg_process_stats, 64,
87		proc_state_init, proc_state_destroy);
88		int proc_state_size = 0;
89	<	proc_state_t *proc_state_ptr;
89	>	sg_process_stats *proc_state_ptr;
90		#ifdef ALLBSD
91		int mib[4];
92		size_t size;
#	Line 99 | Line 101 | int get_proc_snapshot(proc_state_t **ps){
101		long buflen;
102		char *p;
103		#endif
104	+	#ifdef NETBSD2
105	+	int lwps;
106	+	struct kinfo_lwp *kl_stats;
107		#endif
108	+	#endif
109		#if defined(SOLARIS) || defined(LINUX)
110		DIR *proc_dir;
111		struct dirent *dir_entry;
#	Line 114 | Line 120 | int get_proc_snapshot(proc_state_t **ps){
120		char ps_name[4096];
121		char *ptr;
122		VECTOR_DECLARE_STATIC(psargs, char, 128, NULL, NULL);
123	<	unsigned long stime, utime;
123	>	unsigned long stime, utime, starttime;
124		int x;
125		int fn;
126		int len;
127		int rc;
128	+	time_t uptime;
129		#endif
130
131	+	#ifdef LINUX
132	+	if ((f=fopen("/proc/uptime", "r")) == NULL) {
133	+	return NULL;
134	+	}
135	+	if((fscanf(f,"%lu %*d",&uptime)) != 1){
136	+	return NULL;
137	+	}
138	+	fclose(f);
139	+	#endif
140	+
141		if((proc_dir=opendir(PROC_LOCATION))==NULL){
142	<	return -1;
142	>	return NULL;
143		}
144
145		while((dir_entry=readdir(proc_dir))!=NULL){
#	Line 144 | Line 161 | int get_proc_snapshot(proc_state_t **ps){
161		#endif
162
163		if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
164	<	return -1;
164	>	return NULL;
165		}
166		proc_state_ptr = proc_state+proc_state_size;
167
#	Line 160 | Line 177 | int get_proc_snapshot(proc_state_t **ps){
177		proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
178		proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
179		proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
180	<	proc_state_ptr->process_name = strdup(process_info.pr_fname);
181	<	proc_state_ptr->proctitle = strdup(process_info.pr_psargs);
180	>	if (sg_update_string(&proc_state_ptr->process_name,
181	>	process_info.pr_fname) < 0) {
182	>	return NULL;
183	>	}
184	>	if (sg_update_string(&proc_state_ptr->proctitle,
185	>	process_info.pr_psargs) < 0) {
186	>	return NULL;
187	>	}
188
189	<	if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SLEEPING;
190	<	if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = RUNNING;
191	<	if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = ZOMBIE;
192	<	if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = STOPPED;
193	<	if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = RUNNING;
189	>	if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
190	>	if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
191	>	if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
192	>	if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
193	>	if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
194		#endif
195		#ifdef LINUX
196	<	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));
197	<	proc_state_ptr->proc_resident = proc_state_ptr->proc_resident * getpagesize();
198	<	if(s == 'S') proc_state_ptr->state = SLEEPING;
199	<	if(s == 'R') proc_state_ptr->state = RUNNING;
200	<	if(s == 'Z') proc_state_ptr->state = ZOMBIE;
201	<	if(s == 'T') proc_state_ptr->state = STOPPED;
202	<	if(s == 'D') proc_state_ptr->state = STOPPED;
196	>	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));
197	>	/* +3 becuase man page says "Resident  Set Size: number of pages the process has in real memory, minus 3 for administrative purposes." */
198	>	proc_state_ptr->proc_resident = (proc_state_ptr->proc_resident + 3) * getpagesize();
199	>	if(s == 'S') proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
200	>	if(s == 'R') proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
201	>	if(s == 'Z') proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
202	>	if(s == 'T') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
203	>	if(s == 'D') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
204
205		/* pa_name[0] should = '(' */
206		ptr = strchr(&ps_name[1], ')');
207		if(ptr !=NULL) *ptr='\0';
208
209	<	if (update_string(&proc_state_ptr->process_name,
210	<	&ps_name[1]) == NULL) {
211	<	return -1;
209	>	if (sg_update_string(&proc_state_ptr->process_name,
210	>	&ps_name[1]) < 0) {
211	>	return NULL;
212		}
213
214	<	/* Need to do cpu */
215	<
214	>	/* cpu */
215	>	proc_state_ptr->cpu_percent = (100.0 * (utime + stime)) / ((uptime * 100.0) - starttime);
216	>	printf("%s  ut: %lu st: %lu up: %lu start: %lu\n", proc_state_ptr->process_name, utime, stime, uptime , starttime);
217
218		fclose(f);
219
#	Line 204 | Line 229 | int get_proc_snapshot(proc_state_t **ps){
229		#define READ_BLOCK_SIZE 128
230		len = 0;
231		do {
232	<	if (VECTOR_RESIZE(psargs, len + READ_BLOCK_SIZE + 1) < 0) {
233	<	return -1;
232	>	if (VECTOR_RESIZE(psargs, len + READ_BLOCK_SIZE) < 0) {
233	>	return NULL;
234		}
235		rc = read(fn, psargs + len, READ_BLOCK_SIZE);
236		if (rc > 0) {
#	Line 225 | Line 250 | int get_proc_snapshot(proc_state_t **ps){
250		if (*ptr == '\0') *ptr = ' ';
251		ptr++;
252		}
228	–	/* for safety's sake */
229	–	psargs[len] = '\0';
253
254	<	if (update_string(&proc_state_ptr->proctitle, psargs) == NULL) {
255	<	return -1;
254	>	if (len == 0) {
255	>	/* We want psargs to be NULL. */
256	>	if (VECTOR_RESIZE(psargs, 0) < 0) {
257	>	return NULL;
258	>	}
259	>	} else {
260	>	/* Not empty, so append a \0. */
261	>	if (VECTOR_RESIZE(psargs, len + 1) < 0) {
262	>	return NULL;
263	>	}
264	>	psargs[len] = '\0';
265		}
266	+
267	+	if (sg_update_string(&proc_state_ptr->proctitle, psargs) < 0) {
268	+	return NULL;
269	+	}
270		#endif
271
272		proc_state_size++;
#	Line 244 | Line 280 | int get_proc_snapshot(proc_state_t **ps){
280		mib[2] = KERN_PROC_ALL;
281
282		if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
283	<	return -1;
283	>	return NULL;
284		}
285
286		procs = size / sizeof(struct kinfo_proc);
287
288		kp_stats = malloc(size);
289		if(kp_stats == NULL) {
290	<	return -1;
290	>	return NULL;
291		}
292	+	memset(kp_stats, 0, size);
293
294		if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
295		free(kp_stats);
296	<	return -1;
296	>	return NULL;
297		}
298
299		#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
300	<	kvmd = get_kvm2();
300	>	kvmd = sg_get_kvm2();
301		#endif
302
303		for (i = 0; i < procs; i++) {
304		const char *name;
305
306	+	#ifdef FREEBSD5
307	+	if (kp_stats[i].ki_stat == 0) {
308	+	#else
309	+	if (kp_stats[i].kp_proc.p_stat == 0) {
310	+	#endif
311	+	/* FreeBSD 5 deliberately overallocates the array that
312	+	* the sysctl returns, so we'll get a few junk
313	+	* processes on the end that we have to ignore. (Search
314	+	* for "overestimate by 5 procs" in
315	+	* src/sys/kern/kern_proc.c for more details.) */
316	+	continue;
317	+	}
318	+
319		if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
320	<	return -1;
320	>	return NULL;
321		}
322		proc_state_ptr = proc_state+proc_state_size;
323
#	Line 278 | Line 328 | int get_proc_snapshot(proc_state_t **ps){
328		#else
329		name = kp_stats[i].kp_proc.p_comm;
330		#endif
331	<	if (update_string(&proc_state_ptr->process_name, name) == NULL) {
332	<	return -1;
331	>	if (sg_update_string(&proc_state_ptr->process_name, name) < 0) {
332	>	return NULL;
333		}
334
335		#if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD)
#	Line 287 | Line 337 | int get_proc_snapshot(proc_state_t **ps){
337
338		#ifdef FREEBSD5
339		if(sysctlbyname("kern.ps_arg_cache_limit", &buflen, &size, NULL, 0) < 0) {
340	<	return -1;
340	>	return NULL;
341		}
342		#else
343	+	/* FIXME - this value can be too large on some of
344	+	the BSD's, which causes sysctl not to return
345	+	anything. Maybe we need something smaller? */
346		mib[1] = KERN_ARGMAX;
347
348		if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
349	<	return -1;
349	>	return NULL;
350		}
351		#endif
352
353		proctitle = malloc(buflen);
354		if(proctitle == NULL) {
355	<	return -1;
355	>	return NULL;
356		}
357
358		size = buflen;
#	Line 321 | Line 374 | int get_proc_snapshot(proc_state_t **ps){
374		else if(size > 0) {
375		proc_state_ptr->proctitle = malloc(size+1);
376		if(proc_state_ptr->proctitle == NULL) {
377	<	return -1;
377	>	return NULL;
378		}
379		p = proctitle;
380		proc_state_ptr->proctitle[0] = '\0';
381		do {
382	<	strlcat(proc_state_ptr->proctitle, p, size+1);
383	<	strlcat(proc_state_ptr->proctitle, " ", size+1);
382	>	sg_strlcat(proc_state_ptr->proctitle, p, size+1);
383	>	sg_strlcat(proc_state_ptr->proctitle, " ", size+1);
384		p += strlen(p) + 1;
385		} while (p < proctitle + size);
386		free(proctitle);
#	Line 351 | Line 404 | int get_proc_snapshot(proc_state_t **ps){
404		proctitle = malloc(argslen + 1);
405		proctitle[0] = '\0';
406		if(proctitle == NULL) {
407	<	return -1;
407	>	return NULL;
408		}
409		while(*args != NULL) {
410	<	strlcat(proctitle, *args, argslen + 1);
411	<	strlcat(proctitle, " ", argslen + 1);
410	>	sg_strlcat(proctitle, *args, argslen + 1);
411	>	sg_strlcat(proctitle, " ", argslen + 1);
412		args++;
413		}
414		/* remove trailing space */
#	Line 432 | Line 485 | int get_proc_snapshot(proc_state_t **ps){
485		proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
486		#endif
487
488	+	#ifdef NETBSD2
489	+	{
490	+	size_t size;
491	+	int mib[5];
492	+
493	+	mib[0] = CTL_KERN;
494	+	mib[1] = KERN_LWP;
495	+	mib[2] = kp_stats[i].kp_proc.p_pid;
496	+	mib[3] = sizeof(struct kinfo_lwp);
497	+	mib[4] = 0;
498	+
499	+	if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) {
500	+	return NULL;
501	+	}
502	+
503	+	lwps = size / sizeof(struct kinfo_lwp);
504	+	mib[4] = lwps;
505	+
506	+	kl_stats = malloc(size);
507	+	if(kl_stats == NULL) {
508	+	return NULL;
509	+	}
510	+
511	+	if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) {
512	+	return NULL;
513	+	}
514	+	}
515	+
516	+	switch(kp_stats[i].kp_proc.p_stat) {
517	+	case SIDL:
518	+	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
519	+	break;
520	+	case SACTIVE:
521	+	{
522	+	int i;
523	+
524	+	for(i = 0; i < lwps; i++) {
525	+	switch(kl_stats[i].l_stat) {
526	+	case LSONPROC:
527	+	case LSRUN:
528	+	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
529	+	goto end;
530	+	case LSSLEEP:
531	+	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
532	+	goto end;
533	+	case LSSTOP:
534	+	case LSSUSPENDED:
535	+	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
536	+	goto end;
537	+	}
538	+	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
539	+	}
540	+	end: ;
541	+	}
542	+	break;
543	+	case SSTOP:
544	+	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
545	+	break;
546	+	case SZOMB:
547	+	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
548	+	break;
549	+	default:
550	+	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
551	+	break;
552	+	}
553	+	#else
554		#ifdef FREEBSD5
555		switch (kp_stats[i].ki_stat) {
556		#else
#	Line 442 | Line 561 | int get_proc_snapshot(proc_state_t **ps){
561		#ifdef SONPROC
562		case SONPROC: /* NetBSD */
563		#endif
564	<	proc_state_ptr->state = RUNNING;
564	>	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
565		break;
566		case SSLEEP:
567		#ifdef SWAIT
#	Line 451 | Line 570 | int get_proc_snapshot(proc_state_t **ps){
570		#ifdef SLOCK
571		case SLOCK: /* FreeBSD 5 */
572		#endif
573	<	proc_state_ptr->state = SLEEPING;
573	>	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
574		break;
575		case SSTOP:
576	<	proc_state_ptr->state = STOPPED;
576	>	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
577		break;
578		case SZOMB:
579		#ifdef SDEAD
580		case SDEAD: /* OpenBSD & NetBSD */
581		#endif
582	<	proc_state_ptr->state = ZOMBIE;
582	>	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
583		break;
584		default:
585	<	proc_state_ptr->state = UNKNOWN;
585	>	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
586		break;
587		}
588	+	#endif
589		proc_state_size++;
590		}
591
#	Line 473 | Line 593 | int get_proc_snapshot(proc_state_t **ps){
593		#endif
594
595		#ifdef CYGWIN
596	<	return -1;
596	>	return NULL;
597		#endif
598
599	<	*ps = proc_state;
600	<	return proc_state_size;
599	>	*entries = proc_state_size;
600	>	return proc_state;
601		}
602
603	<	process_stat_t *get_process_stats() {
604	<	static process_stat_t process_stat;
605	<	proc_state_t *ps;
603	>	sg_process_count *sg_get_process_count() {
604	>	static sg_process_count process_stat;
605	>	sg_process_stats *ps;
606		int ps_size, x;
607
608		process_stat.sleeping = 0;
#	Line 491 | Line 611 | process_stat_t *get_process_stats() {
611		process_stat.stopped = 0;
612		process_stat.total = 0;
613
614	<	ps_size = get_proc_snapshot(&ps);
615	<
496	<	if(ps_size < 0) {
614	>	ps = sg_get_process_stats(&ps_size);
615	>	if (ps == NULL) {
616		return NULL;
617		}
618
619		for(x = 0; x < ps_size; x++) {
620		switch (ps->state) {
621	<	case RUNNING:
621	>	case SG_PROCESS_STATE_RUNNING:
622		process_stat.running++;
623		break;
624	<	case SLEEPING:
624	>	case SG_PROCESS_STATE_SLEEPING:
625		process_stat.sleeping++;
626		break;
627	<	case STOPPED:
627	>	case SG_PROCESS_STATE_STOPPED:
628		process_stat.stopped++;
629		break;
630	<	case ZOMBIE:
630	>	case SG_PROCESS_STATE_ZOMBIE:
631		process_stat.zombie++;
632		break;
633		default:
634	<	/* currently no mapping for UNKNOWN in process_stat_t */
634	>	/* currently no mapping for SG_PROCESS_STATE_UNKNOWN in
635	>	* sg_process_count */
636		break;
637		}
638		ps++;

Diff Legend

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