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.53 by ats, Mon Apr 5 18:10:11 2004 UTC vs.
Revision 1.64 by pajs, Wed Apr 7 15:46:34 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	>	fclose(f);
162		#endif
163
164		if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
#	Line 160 | Line 178 | sg_process_stats *sg_get_process_stats(int *entries){
178		proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
179		proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
180		proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
181	<	proc_state_ptr->process_name = strdup(process_info.pr_fname);
182	<	proc_state_ptr->proctitle = strdup(process_info.pr_psargs);
181	>	proc_state_ptr->nice = (int)process_info.pr_lwp.pr_nice - 20;
182	>	if (sg_update_string(&proc_state_ptr->process_name,
183	>	process_info.pr_fname) < 0) {
184	>	return NULL;
185	>	}
186	>	if (sg_update_string(&proc_state_ptr->proctitle,
187	>	process_info.pr_psargs) < 0) {
188	>	return NULL;
189	>	}
190
191	<	if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
192	<	if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
193	<	if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
194	<	if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
195	<	if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
191	>	if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
192	>	if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
193	>	if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
194	>	if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
195	>	if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
196		#endif
197		#ifdef LINUX
198	<	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));
199	<	proc_state_ptr->proc_resident = proc_state_ptr->proc_resident * getpagesize();
198	>	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));
199	>	/* +3 becuase man page says "Resident  Set Size: number of pages the process has in real memory, minus 3 for administrative purposes." */
200	>	proc_state_ptr->proc_resident = (proc_state_ptr->proc_resident + 3) * getpagesize();
201		if(s == 'S') proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
202		if(s == 'R') proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
203		if(s == 'Z') proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
#	Line 183 | Line 209 | sg_process_stats *sg_get_process_stats(int *entries){
209		if(ptr !=NULL) *ptr='\0';
210
211		if (sg_update_string(&proc_state_ptr->process_name,
212	<	&ps_name[1]) < 0) {
212	>	&ps_name[1]) < 0) {
213		return NULL;
214		}
215
216	<	/* Need to do cpu */
217	<
216	>	/* cpu */
217	>	proc_state_ptr->cpu_percent = (100.0 * (utime + stime)) / ((uptime * 100.0) - starttime);
218
219	<	fclose(f);
219	>	fclose(f);
220
221	+	/* uid / gid */
222	+	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/status", dir_entry->d_name);
223	+	if ((f=fopen(filename, "r")) == NULL) {
224	+	/* Open failed.. Process since vanished, or the path was too long.
225	+	* Ah well, move onwards to the next one */
226	+	continue;
227	+	}
228	+
229	+	if((ptr=sg_f_read_line(f, "Uid:"))==NULL){
230	+	fclose(f);
231	+	continue;
232	+	}
233	+	sscanf(ptr, "Uid:\t%d\t%d\t%*d\t%*d\n", &(proc_state_ptr->uid), &(proc_state_ptr->euid));
234	+
235	+	if((ptr=sg_f_read_line(f, "Gid:"))==NULL){
236	+	fclose(f);
237	+	continue;
238	+	}
239	+	sscanf(ptr, "Gid:\t%d\t%d\t%*d\t%*d\n", &(proc_state_ptr->gid), &(proc_state_ptr->egid));
240	+
241	+	fclose(f);
242	+
243		/* proctitle */
244		snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/cmdline", dir_entry->d_name);
245
246	<	if((fn=open(filename, O_RDONLY)) == -1){
247	<	/* Open failed.. Process since vanished, or the path was too long.
248	<	* Ah well, move onwards to the next one */
249	<	continue;
250	<	}
246	>	if((fn=open(filename, O_RDONLY)) == -1){
247	>	/* Open failed.. Process since vanished, or the path was too long.
248	>	* Ah well, move onwards to the next one */
249	>	continue;
250	>	}
251
252		#define READ_BLOCK_SIZE 128
253		len = 0;
#	Line 245 | Line 293 | sg_process_stats *sg_get_process_stats(int *entries){
293		#endif
294
295		proc_state_size++;
296	<	}
297	<	closedir(proc_dir);
296	>	}
297	>	closedir(proc_dir);
298		#endif
299
300		#ifdef ALLBSD
#	Line 260 | Line 308 | sg_process_stats *sg_get_process_stats(int *entries){
308
309		procs = size / sizeof(struct kinfo_proc);
310
311	<	kp_stats = malloc(size);
311	>	kp_stats = sg_malloc(size);
312		if(kp_stats == NULL) {
313		return NULL;
314		}
315	+	memset(kp_stats, 0, size);
316
317		if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
318		free(kp_stats);
#	Line 277 | Line 326 | sg_process_stats *sg_get_process_stats(int *entries){
326		for (i = 0; i < procs; i++) {
327		const char *name;
328
329	+	#ifdef FREEBSD5
330	+	if (kp_stats[i].ki_stat == 0) {
331	+	#else
332	+	if (kp_stats[i].kp_proc.p_stat == 0) {
333	+	#endif
334	+	/* FreeBSD 5 deliberately overallocates the array that
335	+	* the sysctl returns, so we'll get a few junk
336	+	* processes on the end that we have to ignore. (Search
337	+	* for "overestimate by 5 procs" in
338	+	* src/sys/kern/kern_proc.c for more details.) */
339	+	continue;
340	+	}
341	+
342		if (VECTOR_RESIZE(proc_state, proc_state_size + 1) < 0) {
343		return NULL;
344		}
#	Line 301 | Line 363 | sg_process_stats *sg_get_process_stats(int *entries){
363		return NULL;
364		}
365		#else
366	+	/* FIXME - this value can be too large on some of
367	+	the BSD's, which causes sysctl not to return
368	+	anything. Maybe we need something smaller? */
369		mib[1] = KERN_ARGMAX;
370
371		if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
#	Line 308 | Line 373 | sg_process_stats *sg_get_process_stats(int *entries){
373		}
374		#endif
375
376	<	proctitle = malloc(buflen);
376	>	proctitle = sg_malloc(buflen);
377		if(proctitle == NULL) {
378		return NULL;
379		}
#	Line 330 | Line 395 | sg_process_stats *sg_get_process_stats(int *entries){
395		proc_state_ptr->proctitle = NULL;
396		}
397		else if(size > 0) {
398	<	proc_state_ptr->proctitle = malloc(size+1);
398	>	proc_state_ptr->proctitle = sg_malloc(size+1);
399		if(proc_state_ptr->proctitle == NULL) {
400		return NULL;
401		}
#	Line 359 | Line 424 | sg_process_stats *sg_get_process_stats(int *entries){
424		argslen += strlen(*argsp) + 1;
425		argsp++;
426		}
427	<	proctitle = malloc(argslen + 1);
427	>	proctitle = sg_malloc(argslen + 1);
428		proctitle[0] = '\0';
429		if(proctitle == NULL) {
430		return NULL;
#	Line 443 | Line 508 | sg_process_stats *sg_get_process_stats(int *entries){
508		proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
509		#endif
510
511	+	#ifdef NETBSD2
512	+	{
513	+	size_t size;
514	+	int mib[5];
515	+
516	+	mib[0] = CTL_KERN;
517	+	mib[1] = KERN_LWP;
518	+	mib[2] = kp_stats[i].kp_proc.p_pid;
519	+	mib[3] = sizeof(struct kinfo_lwp);
520	+	mib[4] = 0;
521	+
522	+	if(sysctl(mib, 5, NULL, &size, NULL, 0) < 0) {
523	+	return NULL;
524	+	}
525	+
526	+	lwps = size / sizeof(struct kinfo_lwp);
527	+	mib[4] = lwps;
528	+
529	+	kl_stats = sg_malloc(size);
530	+	if(kl_stats == NULL) {
531	+	return NULL;
532	+	}
533	+
534	+	if(sysctl(mib, 5, kl_stats, &size, NULL, 0) < 0) {
535	+	return NULL;
536	+	}
537	+	}
538	+
539	+	switch(kp_stats[i].kp_proc.p_stat) {
540	+	case SIDL:
541	+	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
542	+	break;
543	+	case SACTIVE:
544	+	{
545	+	int i;
546	+
547	+	for(i = 0; i < lwps; i++) {
548	+	switch(kl_stats[i].l_stat) {
549	+	case LSONPROC:
550	+	case LSRUN:
551	+	proc_state_ptr->state = SG_PROCESS_STATE_RUNNING;
552	+	goto end;
553	+	case LSSLEEP:
554	+	proc_state_ptr->state = SG_PROCESS_STATE_SLEEPING;
555	+	goto end;
556	+	case LSSTOP:
557	+	case LSSUSPENDED:
558	+	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
559	+	goto end;
560	+	}
561	+	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
562	+	}
563	+	end: ;
564	+	}
565	+	break;
566	+	case SSTOP:
567	+	proc_state_ptr->state = SG_PROCESS_STATE_STOPPED;
568	+	break;
569	+	case SZOMB:
570	+	proc_state_ptr->state = SG_PROCESS_STATE_ZOMBIE;
571	+	break;
572	+	default:
573	+	proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
574	+	break;
575	+	}
576	+	#else
577		#ifdef FREEBSD5
578		switch (kp_stats[i].ki_stat) {
579		#else
#	Line 477 | Line 608 | sg_process_stats *sg_get_process_stats(int *entries){
608		proc_state_ptr->state = SG_PROCESS_STATE_UNKNOWN;
609		break;
610		}
611	+	#endif
612		proc_state_size++;
613		}
614

Diff Legend

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