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.54 by ats, Mon Apr 5 19:55:27 2004 UTC vs.
Revision 1.63 by tdb, Wed Apr 7 14:53:40 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(sg_process_stats *s) {
76		s->process_name = NULL;
#	Line 82 | Line 84 | static void proc_state_destroy(sg_process_stats *s) {
84
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);
87	>	proc_state_init, proc_state_destroy);
88		int proc_state_size = 0;
89		sg_process_stats *proc_state_ptr;
90		#ifdef ALLBSD
#	Line 99 | Line 101 | sg_process_stats *sg_get_process_stats(int *entries){
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;
112	<	char filename[MAX_FILE_LENGTH];
113	<	FILE *f;
110	>	DIR *proc_dir;
111	>	struct dirent *dir_entry;
112	>	char filename[MAX_FILE_LENGTH];
113	>	FILE *f;
114		#ifdef SOLARIS
115		psinfo_t process_info;
116		#endif
#	Line 114 | Line 120 | sg_process_stats *sg_get_process_stats(int *entries){
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	<	if((proc_dir=opendir(PROC_LOCATION))==NULL){
132	<	return NULL;
133	<	}
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	<	while((dir_entry=readdir(proc_dir))!=NULL){
142	<	if(atoi(dir_entry->d_name) == 0) continue;
141	>	if((proc_dir=opendir(PROC_LOCATION))==NULL){
142	>	return NULL;
143	>	}
144
145	+	while((dir_entry=readdir(proc_dir))!=NULL){
146	+	if(atoi(dir_entry->d_name) == 0) continue;
147	+
148		#ifdef SOLARIS
149	<	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
149	>	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
150		#endif
151		#ifdef LINUX
152		snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/stat", dir_entry->d_name);
153		#endif
154	<	if((f=fopen(filename, "r"))==NULL){
155	<	/* Open failed.. Process since vanished, or the path was too long.
156	<	* Ah well, move onwards to the next one */
157	<	continue;
158	<	}
154	>	if((f=fopen(filename, "r"))==NULL){
155	>	/* Open failed.. Process since vanished, or the path was too long.
156	>	* Ah well, move onwards to the next one */
157	>	continue;
158	>	}
159		#ifdef SOLARIS
160	<	fread(&process_info, sizeof(psinfo_t), 1, f);
160	>	fread(&process_info, sizeof(psinfo_t), 1, f);
161		#endif
162
163		if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
#	Line 160 | Line 177 | sg_process_stats *sg_get_process_stats(int *entries){
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 = 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;
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();
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;
#	Line 183 | Line 207 | sg_process_stats *sg_get_process_stats(int *entries){
207		if(ptr !=NULL) *ptr='\0';
208
209		if (sg_update_string(&proc_state_ptr->process_name,
210	<	&ps_name[1]) < 0) {
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
217	<	fclose(f);
217	>	fclose(f);
218
219	+	/* uid / gid */
220	+	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/status", dir_entry->d_name);
221	+	if ((f=fopen(filename, "r")) == NULL) {
222	+	/* Open failed.. Process since vanished, or the path was too long.
223	+	* Ah well, move onwards to the next one */
224	+	continue;
225	+	}
226	+
227	+	if((ptr=sg_f_read_line(f, "Uid:"))==NULL){
228	+	fclose(f);
229	+	continue;
230	+	}
231	+	sscanf(ptr, "Uid:\t%d\t%d\t%*d\t%*d\n", &(proc_state_ptr->uid), &(proc_state_ptr->euid));
232	+
233	+	if((ptr=sg_f_read_line(f, "Gid:"))==NULL){
234	+	fclose(f);
235	+	continue;
236	+	}
237	+	sscanf(ptr, "Gid:\t%d\t%d\t%*d\t%*d\n", &(proc_state_ptr->gid), &(proc_state_ptr->egid));
238	+
239	+	fclose(f);
240	+
241		/* proctitle */
242		snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/cmdline", dir_entry->d_name);
243
244	<	if((fn=open(filename, O_RDONLY)) == -1){
245	<	/* Open failed.. Process since vanished, or the path was too long.
246	<	* Ah well, move onwards to the next one */
247	<	continue;
248	<	}
244	>	if((fn=open(filename, O_RDONLY)) == -1){
245	>	/* Open failed.. Process since vanished, or the path was too long.
246	>	* Ah well, move onwards to the next one */
247	>	continue;
248	>	}
249
250		#define READ_BLOCK_SIZE 128
251		len = 0;
#	Line 245 | Line 291 | sg_process_stats *sg_get_process_stats(int *entries){
291		#endif
292
293		proc_state_size++;
294	<	}
295	<	closedir(proc_dir);
294	>	}
295	>	closedir(proc_dir);
296		#endif
297
298		#ifdef ALLBSD
#	Line 260 | Line 306 | sg_process_stats *sg_get_process_stats(int *entries){
306
307		procs = size / sizeof(struct kinfo_proc);
308
309	<	kp_stats = malloc(size);
309	>	kp_stats = sg_malloc(size);
310		if(kp_stats == NULL) {
311		return NULL;
312		}
#	Line 278 | Line 324 | sg_process_stats *sg_get_process_stats(int *entries){
324		for (i = 0; i < procs; i++) {
325		const char *name;
326
327	+	#ifdef FREEBSD5
328		if (kp_stats[i].ki_stat == 0) {
329	+	#else
330	+	if (kp_stats[i].kp_proc.p_stat == 0) {
331	+	#endif
332		/* FreeBSD 5 deliberately overallocates the array that
333		* the sysctl returns, so we'll get a few junk
334		* processes on the end that we have to ignore. (Search
#	Line 311 | Line 361 | sg_process_stats *sg_get_process_stats(int *entries){
361		return NULL;
362		}
363		#else
364	+	/* FIXME - this value can be too large on some of
365	+	the BSD's, which causes sysctl not to return
366	+	anything. Maybe we need something smaller? */
367		mib[1] = KERN_ARGMAX;
368
369		if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
#	Line 318 | Line 371 | sg_process_stats *sg_get_process_stats(int *entries){
371		}
372		#endif
373
374	<	proctitle = malloc(buflen);
374	>	proctitle = sg_malloc(buflen);
375		if(proctitle == NULL) {
376		return NULL;
377		}
#	Line 340 | Line 393 | sg_process_stats *sg_get_process_stats(int *entries){
393		proc_state_ptr->proctitle = NULL;
394		}
395		else if(size > 0) {
396	<	proc_state_ptr->proctitle = malloc(size+1);
396	>	proc_state_ptr->proctitle = sg_malloc(size+1);
397		if(proc_state_ptr->proctitle == NULL) {
398		return NULL;
399		}
#	Line 369 | Line 422 | sg_process_stats *sg_get_process_stats(int *entries){
422		argslen += strlen(*argsp) + 1;
423		argsp++;
424		}
425	<	proctitle = malloc(argslen + 1);
425	>	proctitle = sg_malloc(argslen + 1);
426		proctitle[0] = '\0';
427		if(proctitle == NULL) {
428		return NULL;
#	Line 453 | Line 506 | sg_process_stats *sg_get_process_stats(int *entries){
506		proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
507		#endif
508
509	+	#ifdef NETBSD2
510	+	{
511	+	size_t size;
512	+	int mib[5];
513	+
514	+	mib[0] = CTL_KERN;
515	+	mib[1] = KERN_LWP;
516	+	mib[2] = kp_stats[i].kp_proc.p_pid;
517	+	mib[3] = sizeof(struct kinfo_lwp);
518	+	mib[4] = 0;
519	+
520	+	if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) {
521	+	return NULL;
522	+	}
523	+
524	+	lwps = size / sizeof(struct kinfo_lwp);
525	+	mib[4] = lwps;
526	+
527	+	kl_stats = sg_malloc(size);
528	+	if(kl_stats == NULL) {
529	+	return NULL;
530	+	}
531	+
532	+	if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) {
533	+	return NULL;
534	+	}
535	+	}
536	+
537	+	switch(kp_stats[i].kp_proc.p_stat) {
538	+	case SIDL:
539	+	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
540	+	break;
541	+	case SACTIVE:
542	+	{
543	+	int i;
544	+
545	+	for(i = 0; i < lwps; i++) {
546	+	switch(kl_stats[i].l_stat) {
547	+	case LSONPROC:
548	+	case LSRUN:
549	+	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
550	+	goto end;
551	+	case LSSLEEP:
552	+	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
553	+	goto end;
554	+	case LSSTOP:
555	+	case LSSUSPENDED:
556	+	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
557	+	goto end;
558	+	}
559	+	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
560	+	}
561	+	end: ;
562	+	}
563	+	break;
564	+	case SSTOP:
565	+	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
566	+	break;
567	+	case SZOMB:
568	+	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
569	+	break;
570	+	default:
571	+	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
572	+	break;
573	+	}
574	+	#else
575		#ifdef FREEBSD5
576		switch (kp_stats[i].ki_stat) {
577		#else
#	Line 487 | Line 606 | sg_process_stats *sg_get_process_stats(int *entries){
606		proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
607		break;
608		}
609	+	#endif
610		proc_state_size++;
611		}
612

Diff Legend

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