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.50 by tdb, Mon Apr 5 09:33:38 2004 UTC vs.
Revision 1.55 by tdb, Tue Apr 6 13:40:41 2004 UTC

#	Line 68 | Line 68
68		#include <tools.h>
69		#endif
70		#include <unistd.h>
71	+	#ifdef NETBSD2
72	+	#include <sys/lwp.h>
73		#endif
74	+	#endif
75
76	<	static void proc_state_init(proc_state_t *s) {
76	>	static void proc_state_init(sg_process_stats *s) {
77		s->process_name = NULL;
78		s->proctitle = NULL;
79		}
80
81	<	static void proc_state_destroy(proc_state_t *s) {
81	>	static void proc_state_destroy(sg_process_stats *s) {
82		free(s->process_name);
83		free(s->proctitle);
84		}
85
86	<	int get_proc_snapshot(proc_state_t **ps){
87	<	VECTOR_DECLARE_STATIC(proc_state, proc_state_t, 64,
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	<	proc_state_t *proc_state_ptr;
90	>	sg_process_stats *proc_state_ptr;
91		#ifdef ALLBSD
92		int mib[4];
93		size_t size;
#	Line 99 | Line 102 | int get_proc_snapshot(proc_state_t **ps){
102		long buflen;
103		char *p;
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;
112		struct dirent *dir_entry;
#	Line 122 | Line 129 | int get_proc_snapshot(proc_state_t **ps){
129		#endif
130
131		if((proc_dir=opendir(PROC_LOCATION))==NULL){
132	<	return -1;
132	>	return NULL;
133		}
134
135		while((dir_entry=readdir(proc_dir))!=NULL){
#	Line 144 | Line 151 | int get_proc_snapshot(proc_state_t **ps){
151		#endif
152
153		if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
154	<	return -1;
154	>	return NULL;
155		}
156		proc_state_ptr = proc_state+proc_state_size;
157
#	Line 163 | Line 170 | int get_proc_snapshot(proc_state_t **ps){
170		proc_state_ptr->process_name = strdup(process_info.pr_fname);
171		proc_state_ptr->proctitle = strdup(process_info.pr_psargs);
172
173	<	if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SLEEPING;
174	<	if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = RUNNING;
175	<	if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = ZOMBIE;
176	<	if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = STOPPED;
177	<	if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = RUNNING;
173	>	if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
174	>	if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
175	>	if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
176	>	if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
177	>	if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
178		#endif
179		#ifdef LINUX
180		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));
181		proc_state_ptr->proc_resident = proc_state_ptr->proc_resident * getpagesize();
182	<	if(s == 'S') proc_state_ptr->state = SLEEPING;
183	<	if(s == 'R') proc_state_ptr->state = RUNNING;
184	<	if(s == 'Z') proc_state_ptr->state = ZOMBIE;
185	<	if(s == 'T') proc_state_ptr->state = STOPPED;
186	<	if(s == 'D') proc_state_ptr->state = STOPPED;
182	>	if(s == 'S') proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
183	>	if(s == 'R') proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
184	>	if(s == 'Z') proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
185	>	if(s == 'T') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
186	>	if(s == 'D') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
187
188		/* pa_name[0] should = '(' */
189		ptr = strchr(&ps_name[1], ')');
190		if(ptr !=NULL) *ptr='\0';
191
192	<	if (update_string(&proc_state_ptr->process_name,
193	<	&ps_name[1]) == NULL) {
194	<	return -1;
192	>	if (sg_update_string(&proc_state_ptr->process_name,
193	>	&ps_name[1]) < 0) {
194	>	return NULL;
195		}
196
197		/* Need to do cpu */
#	Line 204 | Line 211 | int get_proc_snapshot(proc_state_t **ps){
211		#define READ_BLOCK_SIZE 128
212		len = 0;
213		do {
214	<	if (VECTOR_RESIZE(psargs, len + READ_BLOCK_SIZE + 1) < 0) {
215	<	return -1;
214	>	if (VECTOR_RESIZE(psargs, len + READ_BLOCK_SIZE) < 0) {
215	>	return NULL;
216		}
217		rc = read(fn, psargs + len, READ_BLOCK_SIZE);
218		if (rc > 0) {
#	Line 225 | Line 232 | int get_proc_snapshot(proc_state_t **ps){
232		if (*ptr == '\0') *ptr = ' ';
233		ptr++;
234		}
228	–	/* for safety's sake */
229	–	psargs[len] = '\0';
235
236	<	if (update_string(&proc_state_ptr->proctitle, psargs) == NULL) {
237	<	return -1;
236	>	if (len == 0) {
237	>	/* We want psargs to be NULL. */
238	>	if (VECTOR_RESIZE(psargs, 0) < 0) {
239	>	return NULL;
240	>	}
241	>	} else {
242	>	/* Not empty, so append a \0. */
243	>	if (VECTOR_RESIZE(psargs, len + 1) < 0) {
244	>	return NULL;
245	>	}
246	>	psargs[len] = '\0';
247		}
248	+
249	+	if (sg_update_string(&proc_state_ptr->proctitle, psargs) < 0) {
250	+	return NULL;
251	+	}
252		#endif
253
254		proc_state_size++;
#	Line 244 | Line 262 | int get_proc_snapshot(proc_state_t **ps){
262		mib[2] = KERN_PROC_ALL;
263
264		if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
265	<	return -1;
265	>	return NULL;
266		}
267
268		procs = size / sizeof(struct kinfo_proc);
269
270		kp_stats = malloc(size);
271		if(kp_stats == NULL) {
272	<	return -1;
272	>	return NULL;
273		}
274	+	memset(kp_stats, 0, size);
275
276		if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
277		free(kp_stats);
278	<	return -1;
278	>	return NULL;
279		}
280
281		#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
282	<	kvmd = get_kvm2();
282	>	kvmd = sg_get_kvm2();
283		#endif
284
285		for (i = 0; i < procs; i++) {
286		const char *name;
287
288	+	#ifdef FREEBSD5
289	+	if (kp_stats[i].ki_stat == 0) {
290	+	#else
291	+	if (kp_stats[i].kp_proc.p_stat == 0) {
292	+	#endif
293	+	/* FreeBSD 5 deliberately overallocates the array that
294	+	* the sysctl returns, so we'll get a few junk
295	+	* processes on the end that we have to ignore. (Search
296	+	* for "overestimate by 5 procs" in
297	+	* src/sys/kern/kern_proc.c for more details.) */
298	+	continue;
299	+	}
300	+
301		if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
302	<	return -1;
302	>	return NULL;
303		}
304		proc_state_ptr = proc_state+proc_state_size;
305
#	Line 278 | Line 310 | int get_proc_snapshot(proc_state_t **ps){
310		#else
311		name = kp_stats[i].kp_proc.p_comm;
312		#endif
313	<	if (update_string(&proc_state_ptr->process_name, name) == NULL) {
314	<	return -1;
313	>	if (sg_update_string(&proc_state_ptr->process_name, name) < 0) {
314	>	return NULL;
315		}
316
317		#if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD)
#	Line 287 | Line 319 | int get_proc_snapshot(proc_state_t **ps){
319
320		#ifdef FREEBSD5
321		if(sysctlbyname("kern.ps_arg_cache_limit", &buflen, &size, NULL, 0) < 0) {
322	<	return -1;
322	>	return NULL;
323		}
324		#else
325		mib[1] = KERN_ARGMAX;
326
327		if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
328	<	return -1;
328	>	return NULL;
329		}
330		#endif
331
332		proctitle = malloc(buflen);
333		if(proctitle == NULL) {
334	<	return -1;
334	>	return NULL;
335		}
336
337		size = buflen;
#	Line 321 | Line 353 | int get_proc_snapshot(proc_state_t **ps){
353		else if(size > 0) {
354		proc_state_ptr->proctitle = malloc(size+1);
355		if(proc_state_ptr->proctitle == NULL) {
356	<	return -1;
356	>	return NULL;
357		}
358		p = proctitle;
359		proc_state_ptr->proctitle[0] = '\0';
360		do {
361	<	strlcat(proc_state_ptr->proctitle, p, size+1);
362	<	strlcat(proc_state_ptr->proctitle, " ", size+1);
361	>	sg_strlcat(proc_state_ptr->proctitle, p, size+1);
362	>	sg_strlcat(proc_state_ptr->proctitle, " ", size+1);
363		p += strlen(p) + 1;
364		} while (p < proctitle + size);
365		free(proctitle);
#	Line 345 | Line 377 | int get_proc_snapshot(proc_state_t **ps){
377		if(args != NULL) {
378		argsp = args;
379		while(*argsp != NULL) {
380	<	argslen += strlen(*args);
381	<	args++;
380	>	argslen += strlen(*argsp) + 1;
381	>	argsp++;
382		}
383		proctitle = malloc(argslen + 1);
384		proctitle[0] = '\0';
385		if(proctitle == NULL) {
386	<	return -1;
386	>	return NULL;
387		}
388		while(*args != NULL) {
389	<	strlcat(proctitle, *args, argslen);
390	<	strlcat(proctitle, " ", argslen);
389	>	sg_strlcat(proctitle, *args, argslen + 1);
390	>	sg_strlcat(proctitle, " ", argslen + 1);
391		args++;
392		}
393		/* remove trailing space */
#	Line 432 | Line 464 | int get_proc_snapshot(proc_state_t **ps){
464		proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
465		#endif
466
467	+	#ifdef NETBSD2
468	+	{
469	+	size_t size;
470	+	int mib[5];
471	+
472	+	mib[0] = CTL_KERN;
473	+	mib[1] = KERN_LWP;
474	+	mib[2] = kp_stats[i].kp_proc.p_pid;
475	+	mib[3] = sizeof(struct kinfo_lwp);
476	+	mib[4] = 0;
477	+
478	+	if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) {
479	+	return NULL;
480	+	}
481	+
482	+	lwps = size / sizeof(struct kinfo_lwp);
483	+	mib[4] = lwps;
484	+
485	+	kl_stats = malloc(size);
486	+	if(kl_stats == NULL) {
487	+	return NULL;
488	+	}
489	+
490	+	if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) {
491	+	return NULL;
492	+	}
493	+	}
494	+
495	+	switch(kp_stats[i].kp_proc.p_stat) {
496	+	case SIDL:
497	+	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
498	+	break;
499	+	case SACTIVE:
500	+	{
501	+	int i;
502	+
503	+	for(i = 0; i < lwps; i++) {
504	+	switch(kl_stats[i].l_stat) {
505	+	case LSONPROC:
506	+	case LSRUN:
507	+	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
508	+	goto end;
509	+	case LSSLEEP:
510	+	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
511	+	goto end;
512	+	case LSSTOP:
513	+	case LSSUSPENDED:
514	+	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
515	+	goto end;
516	+	}
517	+	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
518	+	}
519	+	end: ;
520	+	}
521	+	break;
522	+	case SSTOP:
523	+	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
524	+	break;
525	+	case SZOMB:
526	+	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
527	+	break;
528	+	default:
529	+	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
530	+	break;
531	+	}
532	+	#else
533		#ifdef FREEBSD5
534		switch (kp_stats[i].ki_stat) {
535		#else
#	Line 442 | Line 540 | int get_proc_snapshot(proc_state_t **ps){
540		#ifdef SONPROC
541		case SONPROC: /* NetBSD */
542		#endif
543	<	proc_state_ptr->state = RUNNING;
543	>	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
544		break;
545		case SSLEEP:
546		#ifdef SWAIT
#	Line 451 | Line 549 | int get_proc_snapshot(proc_state_t **ps){
549		#ifdef SLOCK
550		case SLOCK: /* FreeBSD 5 */
551		#endif
552	<	proc_state_ptr->state = SLEEPING;
552	>	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
553		break;
554		case SSTOP:
555	<	proc_state_ptr->state = STOPPED;
555	>	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
556		break;
557		case SZOMB:
558		#ifdef SDEAD
559		case SDEAD: /* OpenBSD & NetBSD */
560		#endif
561	<	proc_state_ptr->state = ZOMBIE;
561	>	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
562		break;
563		default:
564	<	proc_state_ptr->state = UNKNOWN;
564	>	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
565		break;
566		}
567	+	#endif
568		proc_state_size++;
569		}
570
#	Line 473 | Line 572 | int get_proc_snapshot(proc_state_t **ps){
572		#endif
573
574		#ifdef CYGWIN
575	<	return -1;
575	>	return NULL;
576		#endif
577
578	<	*ps = proc_state;
579	<	return proc_state_size;
578	>	*entries = proc_state_size;
579	>	return proc_state;
580		}
581
582	<	process_stat_t *get_process_stats() {
583	<	static process_stat_t process_stat;
584	<	proc_state_t *ps;
582	>	sg_process_count *sg_get_process_count() {
583	>	static sg_process_count process_stat;
584	>	sg_process_stats *ps;
585		int ps_size, x;
586
587		process_stat.sleeping = 0;
#	Line 491 | Line 590 | process_stat_t *get_process_stats() {
590		process_stat.stopped = 0;
591		process_stat.total = 0;
592
593	<	ps_size = get_proc_snapshot(&ps);
594	<
496	<	if(ps_size < 0) {
593	>	ps = sg_get_process_stats(&ps_size);
594	>	if (ps == NULL) {
595		return NULL;
596		}
597
598		for(x = 0; x < ps_size; x++) {
599		switch (ps->state) {
600	<	case RUNNING:
600	>	case SG_PROCESS_STATE_RUNNING:
601		process_stat.running++;
602		break;
603	<	case SLEEPING:
603	>	case SG_PROCESS_STATE_SLEEPING:
604		process_stat.sleeping++;
605		break;
606	<	case STOPPED:
606	>	case SG_PROCESS_STATE_STOPPED:
607		process_stat.stopped++;
608		break;
609	<	case ZOMBIE:
609	>	case SG_PROCESS_STATE_ZOMBIE:
610		process_stat.zombie++;
611		break;
612		default:
613	<	/* currently no mapping for UNKNOWN in process_stat_t */
613	>	/* currently no mapping for SG_PROCESS_STATE_UNKNOWN in
614	>	* sg_process_count */
615		break;
616		}
617		ps++;

Diff Legend

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