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.66 by pajs, Thu Apr 8 11:26:42 2004 UTC vs.
Revision 1.81 by tdb, Sat Sep 24 13:29:22 2005 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 130 | Line 147 | sg_process_stats *sg_get_process_stats(int *entries){
147
148		#ifdef LINUX
149		if ((f=fopen("/proc/uptime", "r")) == NULL) {
150	<	sg_set_error(SG_ERROR_OPEN, "/proc/uptime");
150	>	sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/uptime");
151		return NULL;
152		}
153		if((fscanf(f,"%lu %*d",&uptime)) != 1){
#	Line 141 | Line 158 | sg_process_stats *sg_get_process_stats(int *entries){
158		#endif
159
160		if((proc_dir=opendir(PROC_LOCATION))==NULL){
161	<	sg_set_error(SG_ERROR_OPENDIR, PROC_LOCATION);
161	>	sg_set_error_with_errno(SG_ERROR_OPENDIR, PROC_LOCATION);
162		return NULL;
163		}
164
#	Line 191 | Line 208 | sg_process_stats *sg_get_process_stats(int *entries){
208		return NULL;
209		}
210
211	<	if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
212	<	if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
213	<	if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
214	<	if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
215	<	if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
211	>	switch (process_info.pr_lwp.pr_state) {
212	>	case 1:
213	>	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
214	>	break;
215	>	case 2:
216	>	case 5:
217	>	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
218	>	break;
219	>	case 3:
220	>	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
221	>	break;
222	>	case 4:
223	>	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
224	>	break;
225	>	}
226		#endif
227		#ifdef LINUX
228		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));
229		/* +3 becuase man page says "Resident  Set Size: number of pages the process has in real memory, minus 3 for administrative purposes." */
230		proc_state_ptr->proc_resident = (proc_state_ptr->proc_resident + 3) * getpagesize();
231	<	if(s == 'S') proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
232	<	if(s == 'R') proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
233	<	if(s == 'Z') proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
234	<	if(s == 'T') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
235	<	if(s == 'D') proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
231	>	switch (s) {
232	>	case 'S':
233	>	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
234	>	break;
235	>	case 'R':
236	>	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
237	>	break;
238	>	case 'Z':
239	>	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
240	>	break;
241	>	case 'T':
242	>	case 'D':
243	>	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
244	>	break;
245	>	}
246
247		/* pa_name[0] should = '(' */
248		ptr = strchr(&ps_name[1], ')');
#	Line 306 | Line 343 | sg_process_stats *sg_get_process_stats(int *entries){
343		mib[2] = KERN_PROC_ALL;
344
345		if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
346	<	sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
346	>	sg_set_error_with_errno(SG_ERROR_SYSCTL,
347	>	"CTL_KERN.KERN_PROC.KERN_PROC_ALL");
348		return NULL;
349		}
350
#	Line 319 | Line 357 | sg_process_stats *sg_get_process_stats(int *entries){
357		memset(kp_stats, 0, size);
358
359		if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
360	<	sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_PROC.KERN_PROC_ALL");
360	>	sg_set_error_with_errno(SG_ERROR_SYSCTL,
361	>	"CTL_KERN.KERN_PROC.KERN_PROC_ALL");
362		free(kp_stats);
363		return NULL;
364		}
#	Line 361 | Line 400 | sg_process_stats *sg_get_process_stats(int *entries){
400		}
401
402		#if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD)
364	–	size = sizeof(buflen);
403
404		#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
405		mib[2] = KERN_PROC_ARGS;
406		mib[3] = kp_stats[i].ki_pid;
407		#else
#	Line 397 | Line 411 | sg_process_stats *sg_get_process_stats(int *entries){
411		#endif
412
413		free(proc_state_ptr->proctitle);
414	<	if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
415	<	free(proctitle);
416	<	proc_state_ptr->proctitle = NULL;
417	<	}
418	<	else if(size > 0) {
414	>	proc_state_ptr->proctitle = NULL;
415	>
416	>	/* Starting size - we'll double this straight away */
417	>	#define PROCTITLE_START_SIZE 64
418	>	buflen = PROCTITLE_START_SIZE;
419	>	size = buflen;
420	>	proctitle = NULL;
421	>
422	>	do {
423	>	if((long) size >= buflen) {
424	>	buflen *= 2;
425	>	size = buflen;
426	>	proctitletmp = sg_realloc(proctitle, buflen);
427	>	if(proctitletmp == NULL) {
428	>	free(proctitle);
429	>	proctitle = NULL;
430	>	proc_state_ptr->proctitle = NULL;
431	>	size = 0;
432	>	break;
433	>	}
434	>	proctitle = proctitletmp;
435	>	bzero(proctitle, buflen);
436	>	}
437	>
438	>	if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
439	>	free(proctitle);
440	>	proctitle = NULL;
441	>	proc_state_ptr->proctitle = NULL;
442	>	size = 0;
443	>	break;
444	>	}
445	>	} while((long) size >= buflen);
446	>
447	>	if(size > 0) {
448		proc_state_ptr->proctitle = sg_malloc(size+1);
449		if(proc_state_ptr->proctitle == NULL) {
450		return NULL;
451		}
452		p = proctitle;
453	+	#ifdef OPENBSD
454	+	/* On OpenBSD, this value has the argv pointers (which
455	+	* are terminated by a NULL) at the front, so we have
456	+	* to skip over them to get to the strings. */
457	+	while (*(char ***)p != NULL) {
458	+	p += sizeof(char **);
459	+	}
460	+	p += sizeof(char **);
461	+	#endif
462		proc_state_ptr->proctitle[0] = '\0';
463		do {
464		sg_strlcat(proc_state_ptr->proctitle, p, size+1);
#	Line 414 | Line 466 | sg_process_stats *sg_get_process_stats(int *entries){
466		p += strlen(p) + 1;
467		} while (p < proctitle + size);
468		free(proctitle);
469	+	proctitle = NULL;
470		/* remove trailing space */
471		proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = '\0';
472		}
473		else {
474	<	free(proctitle);
474	>	if(proctitle != NULL) {
475	>	free(proctitle);
476	>	proctitle = NULL;
477	>	}
478		proc_state_ptr->proctitle = NULL;
479		}
480		#else
481		free(proc_state_ptr->proctitle);
482	+	proc_state_ptr->proctitle = NULL;
483		if(kvmd != NULL) {
484		args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
485		if(args != NULL) {
#	Line 527 | Line 584 | sg_process_stats *sg_get_process_stats(int *entries){
584		mib[4] = 0;
585
586		if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) {
587	<	sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.0");
587	>	sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.0");
588		return NULL;
589		}
590
#	Line 540 | Line 597 | sg_process_stats *sg_get_process_stats(int *entries){
597		}
598
599		if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) {
600	<	sg_set_error(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.buffersize");
600	>	sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_KERN.KERN_LWP.pid.structsize.buffersize");
601		return NULL;
602		}
603		}
#	Line 582 | Line 639 | sg_process_stats *sg_get_process_stats(int *entries){
639		proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
640		break;
641		}
642	+
643	+	free(kl_stats);
644		#else
645		#ifdef FREEBSD5
646		switch (kp_stats[i].ki_stat) {
#	Line 624 | Line 683 | sg_process_stats *sg_get_process_stats(int *entries){
683		free(kp_stats);
684		#endif
685
686	+	#ifdef HPUX
687	+	if ((pagesize = sysconf(_SC_PAGESIZE)) == -1) {
688	+	sg_set_error_with_errno(SG_ERROR_SYSCONF, "_SC_PAGESIZE");
689	+	return NULL;
690	+	}
691	+
692	+	while (1) {
693	+	num = pstat_getproc(pstat_procinfo, sizeof pstat_procinfo[0],
694	+	PROCESS_BATCH, procidx);
695	+	if (num == -1) {
696	+	sg_set_error_with_errno(SG_ERROR_PSTAT,
697	+	"pstat_getproc");
698	+	return NULL;
699	+	} else if (num == 0) {
700	+	break;
701	+	}
702	+
703	+	for (i = 0; i < num; i++) {
704	+	struct pst_status *pi = &pstat_procinfo[i];
705	+
706	+	if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
707	+	return NULL;
708	+	}
709	+	proc_state_ptr = proc_state+proc_state_size;
710	+
711	+	proc_state_ptr->pid = pi->pst_pid;
712	+	proc_state_ptr->parent = pi->pst_ppid;
713	+	proc_state_ptr->pgid = pi->pst_pgrp;
714	+	proc_state_ptr->uid = pi->pst_uid;
715	+	proc_state_ptr->euid = pi->pst_euid;
716	+	proc_state_ptr->gid = pi->pst_gid;
717	+	proc_state_ptr->egid = pi->pst_egid;
718	+	proc_state_ptr->proc_size = (pi->pst_dsize + pi->pst_tsize + pi->pst_ssize) * pagesize;
719	+	proc_state_ptr->proc_resident = pi->pst_rssize * pagesize;
720	+	proc_state_ptr->time_spent = pi->pst_time;
721	+	proc_state_ptr->cpu_percent = (pi->pst_pctcpu * 100.0) / 0x8000;
722	+	proc_state_ptr->nice = pi->pst_nice;
723	+
724	+	if (sg_update_string(&proc_state_ptr->process_name,
725	+	pi->pst_ucomm) < 0) {
726	+	return NULL;
727	+	}
728	+	if (sg_update_string(&proc_state_ptr->proctitle,
729	+	pi->pst_cmd) < 0) {
730	+	return NULL;
731	+	}
732	+
733	+	switch (pi->pst_stat) {
734	+	case PS_SLEEP:
735	+	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
736	+	break;
737	+	case PS_RUN:
738	+	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
739	+	break;
740	+	case PS_STOP:
741	+	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
742	+	break;
743	+	case PS_ZOMBIE:
744	+	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
745	+	break;
746	+	case PS_IDLE:
747	+	case PS_OTHER:
748	+	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
749	+	break;
750	+	}
751	+
752	+	proc_state_size++;
753	+	}
754	+	procidx = pstat_procinfo[num - 1].pst_idx + 1;
755	+	}
756	+	#endif
757	+
758		#ifdef CYGWIN
759		sg_set_error(SG_ERROR_UNSUPPORTED, "Cygwin");
760		return NULL;
761		#endif
762	+	#ifdef WIN32
763	+	/* FIXME The data needed for this is probably do able with the
764	+	* "performance registry". Although using this appears to be a black
765	+	* art and closely guarded secret.
766	+	* This is not directly used in ihost, so not considered a priority */
767	+	sg_set_error(SG_ERROR_UNSUPPORTED, "Win32");
768	+	return NULL;
769	+	#endif
770
771		*entries = proc_state_size;
772		return proc_state;
#	Line 635 | Line 774 | sg_process_stats *sg_get_process_stats(int *entries){
774
775		sg_process_count *sg_get_process_count() {
776		static sg_process_count process_stat;
777	+	#ifndef WIN32
778		sg_process_stats *ps;
779		int ps_size, x;
780	+	#else
781	+	DWORD aProcesses[1024];
782	+	DWORD cbNeeded;
783	+	#endif
784
785		process_stat.sleeping = 0;
786		process_stat.running = 0;
#	Line 644 | Line 788 | sg_process_count *sg_get_process_count() {
788		process_stat.stopped = 0;
789		process_stat.total = 0;
790
791	+	#ifndef WIN32
792		ps = sg_get_process_stats(&ps_size);
793		if (ps == NULL) {
794		return NULL;
#	Line 672 | Line 817 | sg_process_count *sg_get_process_count() {
817		}
818
819		process_stat.total = ps_size;
820	+	#else
821	+	if (!EnumProcesses(aProcesses, sizeof(aProcesses), &cbNeeded))
822	+	return NULL;
823	+	process_stat.total = cbNeeded / sizeof(DWORD);
824	+	#endif
825
826		return &process_stat;
827		}
828
829	<	int sg_compare_pid(const void *va, const void *vb) {
829	>	int sg_process_compare_name(const void *va, const void *vb) {
830		const sg_process_stats *a = (sg_process_stats *)va;
831		const sg_process_stats *b = (sg_process_stats *)vb;
832
833	+	return strcmp(a->process_name, b->process_name);
834	+	}
835	+
836	+	int sg_process_compare_pid(const void *va, const void *vb) {
837	+	const sg_process_stats *a = (sg_process_stats *)va;
838	+	const sg_process_stats *b = (sg_process_stats *)vb;
839	+
840		if (a->pid < b->pid) {
841		return -1;
842		} else if (a->pid == b->pid) {
#	Line 689 | Line 846 | int sg_compare_pid(const void *va, const void *vb) {
846		}
847		}
848
849	<	int sg_compare_uid(const void *va, const void *vb) {
849	>	int sg_process_compare_uid(const void *va, const void *vb) {
850		const sg_process_stats *a = (sg_process_stats *)va;
851		const sg_process_stats *b = (sg_process_stats *)vb;
852
#	Line 702 | Line 859 | int sg_compare_uid(const void *va, const void *vb) {
859		}
860		}
861
862	<	int sg_compare_gid(const void *va, const void *vb) {
862	>	int sg_process_compare_gid(const void *va, const void *vb) {
863		const sg_process_stats *a = (sg_process_stats *)va;
864		const sg_process_stats *b = (sg_process_stats *)vb;
865
#	Line 715 | Line 872 | int sg_compare_gid(const void *va, const void *vb) {
872		}
873		}
874
875	<	int sg_compare_size(const void *va, const void *vb) {
875	>	int sg_process_compare_size(const void *va, const void *vb) {
876		const sg_process_stats *a = (sg_process_stats *)va;
877		const sg_process_stats *b = (sg_process_stats *)vb;
878
#	Line 728 | Line 885 | int sg_compare_size(const void *va, const void *vb) {
885		}
886		}
887
888	<	int sg_compare_res(const void *va, const void *vb) {
888	>	int sg_process_compare_res(const void *va, const void *vb) {
889		const sg_process_stats *a = (sg_process_stats *)va;
890		const sg_process_stats *b = (sg_process_stats *)vb;
891
#	Line 741 | Line 898 | int sg_compare_res(const void *va, const void *vb) {
898		}
899		}
900
901	<	int sg_compare_cpu(const void *va, const void *vb) {
901	>	int sg_process_compare_cpu(const void *va, const void *vb) {
902		const sg_process_stats *a = (sg_process_stats *)va;
903		const sg_process_stats *b = (sg_process_stats *)vb;
904
#	Line 754 | Line 911 | int sg_compare_cpu(const void *va, const void *vb) {
911		}
912		}
913
914	<	int sg_compare_time(const void *va, const void *vb) {
914	>	int sg_process_compare_time(const void *va, const void *vb) {
915		const sg_process_stats *a = (sg_process_stats *)va;
916		const sg_process_stats *b = (sg_process_stats *)vb;
917
#	Line 767 | Line 924 | int sg_compare_time(const void *va, const void *vb) {
924		}
925		}
926
770	–	void sg_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_compare_pid;
776	–	break;
777	–	case SG_PS_UID:
778	–	sortby_ptr = sg_compare_uid;
779	–	break;
780	–	case SG_PS_GID:
781	–	sortby_ptr = sg_compare_gid;
782	–	break;
783	–	case SG_PS_SIZE:
784	–	sortby_ptr = sg_compare_size;
785	–	break;
786	–	case SG_PS_RES:
787	–	sortby_ptr = sg_compare_res;
788	–	break;
789	–	case SG_PS_CPU:
790	–	sortby_ptr = sg_compare_cpu;
791	–	break;
792	–	case SG_PS_TIME:
793	–	sortby_ptr = sg_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