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.67 by pajs, Thu Apr 8 12:40:30 2004 UTC vs.
Revision 1.82 by tdb, Mon Oct 9 14:47:58 2006 UTC

#	Line 33 | Line 33
33		#include <stdlib.h>
34		#include <sys/types.h>
35		#include <dirent.h>
36	–	#include <string.h>
36		#endif
37	+	#include <string.h>
38
39		#ifdef SOLARIS
40		#include <procfs.h>
#	Line 51 | Line 51
51		#define MAX_FILE_LENGTH PATH_MAX
52		#endif
53		#ifdef ALLBSD
54	+	#include <errno.h>
55		#include <stdlib.h>
56		#include <sys/param.h>
57		#include <sys/sysctl.h>
#	Line 71 | Line 72
72		#include <sys/lwp.h>
73		#endif
74		#endif
75	+	#ifdef HPUX
76	+	#include <sys/param.h>
77	+	#include <sys/pstat.h>
78	+	#include <unistd.h>
79	+	#define PROCESS_BATCH 30
80	+	#endif
81	+	#ifdef WIN32
82	+	#include <windows.h>
83	+	#include <psapi.h>
84	+	#endif
85
86		static void proc_state_init(sg_process_stats *s) {
87		s->process_name = NULL;
#	Line 87 | Line 98 | sg_process_stats *sg_get_process_stats(int *entries){
98		proc_state_init, proc_state_destroy);
99		int proc_state_size = 0;
100		sg_process_stats *proc_state_ptr;
101	+	#ifdef HPUX
102	+	struct pst_status pstat_procinfo[PROCESS_BATCH];
103	+	long procidx = 0;
104	+	long long pagesize;
105	+	int num, i;
106	+	#endif
107		#ifdef ALLBSD
108		int mib[4];
109		size_t size;
#	Line 99 | Line 116 | sg_process_stats *sg_get_process_stats(int *entries){
116		int argslen = 0;
117		#else
118		long buflen;
119	<	char *p;
119	>	char *p, *proctitletmp;
120		#endif
121		#ifdef NETBSD2
122		int lwps;
#	Line 126 | Line 143 | sg_process_stats *sg_get_process_stats(int *entries){
143		int len;
144		int rc;
145		time_t uptime;
146	+	long tickspersec;
147		#endif
148
149		#ifdef LINUX
150		if ((f=fopen("/proc/uptime", "r")) == NULL) {
151	<	sg_set_error(SG_ERROR_OPEN, "/proc/uptime");
151	>	sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/uptime");
152		return NULL;
153		}
154		if((fscanf(f,"%lu %*d",&uptime)) != 1){
#	Line 141 | Line 159 | sg_process_stats *sg_get_process_stats(int *entries){
159		#endif
160
161		if((proc_dir=opendir(PROC_LOCATION))==NULL){
162	<	sg_set_error(SG_ERROR_OPENDIR, PROC_LOCATION);
162	>	sg_set_error_with_errno(SG_ERROR_OPENDIR, PROC_LOCATION);
163		return NULL;
164		}
165
#	Line 191 | Line 209 | sg_process_stats *sg_get_process_stats(int *entries){
209		return NULL;
210		}
211
212	<	if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
213	<	if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
214	<	if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
215	<	if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
216	<	if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
212	>	switch (process_info.pr_lwp.pr_state) {
213	>	case 1:
214	>	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
215	>	break;
216	>	case 2:
217	>	case 5:
218	>	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
219	>	break;
220	>	case 3:
221	>	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
222	>	break;
223	>	case 4:
224	>	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
225	>	break;
226	>	}
227		#endif
228		#ifdef LINUX
229		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));
230		/* +3 becuase man page says "Resident  Set Size: number of pages the process has in real memory, minus 3 for administrative purposes." */
231		proc_state_ptr->proc_resident = (proc_state_ptr->proc_resident + 3) * getpagesize();
232	<	if(s == 'S') proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
233	<	if(s == 'R') proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
234	<	if(s == 'Z') proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
235	<	if(s == 'T') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
236	<	if(s == 'D') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
232	>	switch (s) {
233	>	case 'S':
234	>	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
235	>	break;
236	>	case 'R':
237	>	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
238	>	break;
239	>	case 'Z':
240	>	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
241	>	break;
242	>	case 'T':
243	>	case 'D':
244	>	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
245	>	break;
246	>	}
247
248		/* pa_name[0] should = '(' */
249		ptr = strchr(&ps_name[1], ')');
#	Line 218 | Line 256 | sg_process_stats *sg_get_process_stats(int *entries){
256
257		/* cpu */
258		proc_state_ptr->cpu_percent = (100.0 * (utime + stime)) / ((uptime * 100.0) - starttime);
259	+	tickspersec = sysconf (_SC_CLK_TCK);
260	+	if (tickspersec < 0) {
261	+	proc_state_ptr->time_spent = 0;
262	+	}
263	+	else {
264	+	proc_state_ptr->time_spent = (utime + stime) / tickspersec;
265	+	}
266
267		fclose(f);
268
#	Line 306 | Line 351 | sg_process_stats *sg_get_process_stats(int *entries){
351		mib[2] = KERN_PROC_ALL;
352
353		if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
354	<	sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
354	>	sg_set_error_with_errno(SG_ERROR_SYSCTL,
355	>	"CTL_KERN.KERN_PROC.KERN_PROC_ALL");
356		return NULL;
357		}
358
#	Line 319 | Line 365 | sg_process_stats *sg_get_process_stats(int *entries){
365		memset(kp_stats, 0, size);
366
367		if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
368	<	sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
368	>	sg_set_error_with_errno(SG_ERROR_SYSCTL,
369	>	"CTL_KERN.KERN_PROC.KERN_PROC_ALL");
370		free(kp_stats);
371		return NULL;
372		}
#	Line 361 | Line 408 | sg_process_stats *sg_get_process_stats(int *entries){
408		}
409
410		#if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD)
364	–	size = sizeof(buflen);
411
412		#ifdef FREEBSD5
367	–	if(sysctlbyname("kern.ps_arg_cache_limit", &buflen, &size, NULL, 0) < 0) {
368	–	sg_set_error(SG_ERROR_SYSCTLBYNAME, "kern.ps_arg_cache_limit");
369	–	return NULL;
370	–	}
371	–	#else
372	–	/* FIXME - this value can be too large on some of
373	–	the BSD's, which causes sysctl not to return
374	–	anything. Maybe we need something smaller? */
375	–	mib[1] = KERN_ARGMAX;
376	–
377	–	if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
378	–	sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_ARGMAX");
379	–	return NULL;
380	–	}
381	–	#endif
382	–
383	–	proctitle = sg_malloc(buflen);
384	–	if(proctitle == NULL) {
385	–	return NULL;
386	–	}
387	–
388	–	size = buflen;
389	–
390	–	#ifdef FREEBSD5
413		mib[2] = KERN_PROC_ARGS;
414		mib[3] = kp_stats[i].ki_pid;
415		#else
#	Line 397 | Line 419 | sg_process_stats *sg_get_process_stats(int *entries){
419		#endif
420
421		free(proc_state_ptr->proctitle);
422	<	if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
423	<	free(proctitle);
424	<	proc_state_ptr->proctitle = NULL;
425	<	}
426	<	else if(size > 0) {
422	>	proc_state_ptr->proctitle = NULL;
423	>
424	>	/* Starting size - we'll double this straight away */
425	>	#define PROCTITLE_START_SIZE 64
426	>	buflen = PROCTITLE_START_SIZE;
427	>	size = buflen;
428	>	proctitle = NULL;
429	>
430	>	do {
431	>	if((long) size >= buflen) {
432	>	buflen *= 2;
433	>	size = buflen;
434	>	proctitletmp = sg_realloc(proctitle, buflen);
435	>	if(proctitletmp == NULL) {
436	>	free(proctitle);
437	>	proctitle = NULL;
438	>	proc_state_ptr->proctitle = NULL;
439	>	size = 0;
440	>	break;
441	>	}
442	>	proctitle = proctitletmp;
443	>	bzero(proctitle, buflen);
444	>	}
445	>
446	>	if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
447	>	free(proctitle);
448	>	proctitle = NULL;
449	>	proc_state_ptr->proctitle = NULL;
450	>	size = 0;
451	>	break;
452	>	}
453	>	} while((long) size >= buflen);
454	>
455	>	if(size > 0) {
456		proc_state_ptr->proctitle = sg_malloc(size+1);
457		if(proc_state_ptr->proctitle == NULL) {
458		return NULL;
459		}
460		p = proctitle;
461	+	#ifdef OPENBSD
462	+	/* On OpenBSD, this value has the argv pointers (which
463	+	* are terminated by a NULL) at the front, so we have
464	+	* to skip over them to get to the strings. */
465	+	while (*(char ***)p != NULL) {
466	+	p += sizeof(char **);
467	+	}
468	+	p += sizeof(char **);
469	+	#endif
470		proc_state_ptr->proctitle[0] = '\0';
471		do {
472		sg_strlcat(proc_state_ptr->proctitle, p, size+1);
#	Line 414 | Line 474 | sg_process_stats *sg_get_process_stats(int *entries){
474		p += strlen(p) + 1;
475		} while (p < proctitle + size);
476		free(proctitle);
477	+	proctitle = NULL;
478		/* remove trailing space */
479		proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = '\0';
480		}
481		else {
482	<	free(proctitle);
482	>	if(proctitle != NULL) {
483	>	free(proctitle);
484	>	proctitle = NULL;
485	>	}
486		proc_state_ptr->proctitle = NULL;
487		}
488		#else
489		free(proc_state_ptr->proctitle);
490	+	proc_state_ptr->proctitle = NULL;
491		if(kvmd != NULL) {
492		args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
493		if(args != NULL) {
#	Line 527 | Line 592 | sg_process_stats *sg_get_process_stats(int *entries){
592		mib[4] = 0;
593
594		if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) {
595	<	sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.0");
595	>	sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.0");
596		return NULL;
597		}
598
#	Line 540 | Line 605 | sg_process_stats *sg_get_process_stats(int *entries){
605		}
606
607		if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) {
608	<	sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.buffersize");
608	>	sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.buffersize");
609		return NULL;
610		}
611		}
#	Line 582 | Line 647 | sg_process_stats *sg_get_process_stats(int *entries){
647		proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
648		break;
649		}
650	+
651	+	free(kl_stats);
652		#else
653		#ifdef FREEBSD5
654		switch (kp_stats[i].ki_stat) {
#	Line 624 | Line 691 | sg_process_stats *sg_get_process_stats(int *entries){
691		free(kp_stats);
692		#endif
693
694	+	#ifdef HPUX
695	+	if ((pagesize = sysconf(_SC_PAGESIZE)) == -1) {
696	+	sg_set_error_with_errno(SG_ERROR_SYSCONF, "_SC_PAGESIZE");
697	+	return NULL;
698	+	}
699	+
700	+	while (1) {
701	+	num = pstat_getproc(pstat_procinfo, sizeof pstat_procinfo[0],
702	+	PROCESS_BATCH, procidx);
703	+	if (num == -1) {
704	+	sg_set_error_with_errno(SG_ERROR_PSTAT,
705	+	"pstat_getproc");
706	+	return NULL;
707	+	} else if (num == 0) {
708	+	break;
709	+	}
710	+
711	+	for (i = 0; i < num; i++) {
712	+	struct pst_status *pi = &pstat_procinfo[i];
713	+
714	+	if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
715	+	return NULL;
716	+	}
717	+	proc_state_ptr = proc_state+proc_state_size;
718	+
719	+	proc_state_ptr->pid = pi->pst_pid;
720	+	proc_state_ptr->parent = pi->pst_ppid;
721	+	proc_state_ptr->pgid = pi->pst_pgrp;
722	+	proc_state_ptr->uid = pi->pst_uid;
723	+	proc_state_ptr->euid = pi->pst_euid;
724	+	proc_state_ptr->gid = pi->pst_gid;
725	+	proc_state_ptr->egid = pi->pst_egid;
726	+	proc_state_ptr->proc_size = (pi->pst_dsize + pi->pst_tsize + pi->pst_ssize) * pagesize;
727	+	proc_state_ptr->proc_resident = pi->pst_rssize * pagesize;
728	+	proc_state_ptr->time_spent = pi->pst_time;
729	+	proc_state_ptr->cpu_percent = (pi->pst_pctcpu * 100.0) / 0x8000;
730	+	proc_state_ptr->nice = pi->pst_nice;
731	+
732	+	if (sg_update_string(&proc_state_ptr->process_name,
733	+	pi->pst_ucomm) < 0) {
734	+	return NULL;
735	+	}
736	+	if (sg_update_string(&proc_state_ptr->proctitle,
737	+	pi->pst_cmd) < 0) {
738	+	return NULL;
739	+	}
740	+
741	+	switch (pi->pst_stat) {
742	+	case PS_SLEEP:
743	+	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
744	+	break;
745	+	case PS_RUN:
746	+	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
747	+	break;
748	+	case PS_STOP:
749	+	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
750	+	break;
751	+	case PS_ZOMBIE:
752	+	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
753	+	break;
754	+	case PS_IDLE:
755	+	case PS_OTHER:
756	+	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
757	+	break;
758	+	}
759	+
760	+	proc_state_size++;
761	+	}
762	+	procidx = pstat_procinfo[num - 1].pst_idx + 1;
763	+	}
764	+	#endif
765	+
766		#ifdef CYGWIN
767		sg_set_error(SG_ERROR_UNSUPPORTED, "Cygwin");
768		return NULL;
769		#endif
770	+	#ifdef WIN32
771	+	/* FIXME The data needed for this is probably do able with the
772	+	* "performance registry". Although using this appears to be a black
773	+	* art and closely guarded secret.
774	+	* This is not directly used in ihost, so not considered a priority */
775	+	sg_set_error(SG_ERROR_UNSUPPORTED, "Win32");
776	+	return NULL;
777	+	#endif
778
779		*entries = proc_state_size;
780		return proc_state;
#	Line 635 | Line 782 | sg_process_stats *sg_get_process_stats(int *entries){
782
783		sg_process_count *sg_get_process_count() {
784		static sg_process_count process_stat;
785	+	#ifndef WIN32
786		sg_process_stats *ps;
787		int ps_size, x;
788	+	#else
789	+	DWORD aProcesses[1024];
790	+	DWORD cbNeeded;
791	+	#endif
792
793		process_stat.sleeping = 0;
794		process_stat.running = 0;
#	Line 644 | Line 796 | sg_process_count *sg_get_process_count() {
796		process_stat.stopped = 0;
797		process_stat.total = 0;
798
799	+	#ifndef WIN32
800		ps = sg_get_process_stats(&ps_size);
801		if (ps == NULL) {
802		return NULL;
#	Line 672 | Line 825 | sg_process_count *sg_get_process_count() {
825		}
826
827		process_stat.total = ps_size;
828	+	#else
829	+	if (!EnumProcesses(aProcesses, sizeof(aProcesses), &cbNeeded))
830	+	return NULL;
831	+	process_stat.total = cbNeeded / sizeof(DWORD);
832	+	#endif
833
834		return &process_stat;
835		}
836
837	+	int sg_process_compare_name(const void *va, const void *vb) {
838	+	const sg_process_stats *a = (sg_process_stats *)va;
839	+	const sg_process_stats *b = (sg_process_stats *)vb;
840	+
841	+	return strcmp(a->process_name, b->process_name);
842	+	}
843	+
844		int sg_process_compare_pid(const void *va, const void *vb) {
845		const sg_process_stats *a = (sg_process_stats *)va;
846		const sg_process_stats *b = (sg_process_stats *)vb;
#	Line 767 | Line 932 | int sg_process_compare_time(const void *va, const void
932		}
933		}
934
770	–	void sg_process_sort(sg_process_stats *ps, int num_ps, sg_sort_method sm){
771	–	int (*sortby_ptr)(const void *a, const void *b);
772	–
773	–	switch(sm){
774	–	case SG_PS_PID:
775	–	sortby_ptr = sg_process_compare_pid;
776	–	break;
777	–	case SG_PS_UID:
778	–	sortby_ptr = sg_process_compare_uid;
779	–	break;
780	–	case SG_PS_GID:
781	–	sortby_ptr = sg_process_compare_gid;
782	–	break;
783	–	case SG_PS_SIZE:
784	–	sortby_ptr = sg_process_compare_size;
785	–	break;
786	–	case SG_PS_RES:
787	–	sortby_ptr = sg_process_compare_res;
788	–	break;
789	–	case SG_PS_CPU:
790	–	sortby_ptr = sg_process_compare_cpu;
791	–	break;
792	–	case SG_PS_TIME:
793	–	sortby_ptr = sg_process_compare_time;
794	–	break;
795	–	}
796	–
797	–	qsort(ps, num_ps, sizeof(*ps), sortby_ptr);
798	–
799	–	return;
800	–	}

Diff Legend

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