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.20 by tdb, Sat Feb 14 16:12:46 2004 UTC vs.
Revision 1.44 by tdb, Sun Apr 4 23:24:42 2004 UTC

#	Line 34 | Line 34
34		#include <string.h>
35		#endif
36
37	–	#include "tools.h"
37		#ifdef SOLARIS
38		#include <procfs.h>
39		#include <limits.h>
#	Line 43 | Line 42
42		#endif
43		#ifdef LINUX
44		#include <limits.h>
45	+	#include <unistd.h>
46	+	#include <sys/stat.h>
47	+	#include <fcntl.h>
48		#define PROC_LOCATION "/proc"
49		#define MAX_FILE_LENGTH PATH_MAX
50		#endif
#	Line 50 | Line 52
52		#include <stdlib.h>
53		#include <sys/param.h>
54		#include <sys/sysctl.h>
55	+	#if defined(FREEBSD) || defined(DFBSD)
56	+	#include <sys/user.h>
57	+	#else
58		#include <sys/proc.h>
59		#endif
60	<	#ifdef FREEBSD
61	<	#include <sys/user.h>
60	>	#include <string.h>
61	>	#include <paths.h>
62	>	#include <fcntl.h>
63	>	#include <limits.h>
64	>	#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
65	>	#include <kvm.h>
66	>	#include <tools.h>
67		#endif
68	+	#include <unistd.h>
69	+	#endif
70
71	<	process_stat_t *get_process_stats(){
72	<
73	<	static process_stat_t process_stat;
74	<
75	<	#if defined(SOLARIS) || defined(LINUX)
76	<	DIR *proc_dir;
77	<	struct dirent *dir_entry;
78	<	char filename[MAX_FILE_LENGTH];
79	<	FILE *f;
71	>	int get_proc_snapshot(proc_state_t **ps){
72	>	proc_state_t *proc_state = NULL;
73	>	proc_state_t *proc_state_ptr;
74	>	int proc_state_size = 0;
75	>	#ifdef ALLBSD
76	>	int mib[4];
77	>	size_t size;
78	>	struct kinfo_proc *kp_stats;
79	>	int procs, i;
80	>	char *proctitle, *proctitletmp;
81	>	#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
82	>	kvm_t *kvmd;
83	>	char **args;
84	>	int alloc;
85	>	#else
86	>	long buflen;
87	>	char *p;
88		#endif
69	–	#ifdef LINUX
70	–	char *line_ptr;
89		#endif
90	+	#if defined(SOLARIS) || defined(LINUX)
91	+	DIR *proc_dir;
92	+	struct dirent *dir_entry;
93	+	char filename[MAX_FILE_LENGTH];
94	+	FILE *f;
95		#ifdef SOLARIS
96		psinfo_t process_info;
97		#endif
98	<	#ifdef ALLBSD
99	<	int mib[3];
100	<	size_t size;
101	<	struct kinfo_proc *kp_stats;
102	<	int procs, i;
98	>	#ifdef LINUX
99	>	char s;
100	>	/* If someone has a executable of 4k filename length, they deserve to get it truncated :) */
101	>	char ps_name[4096];
102	>	char *ptr;
103	>	static char *psargs = NULL;
104	>	static int psarg_size = 0;
105	>	unsigned long stime, utime;
106	>	int x;
107	>	int fn;
108	>	int toread;
109	>	ssize_t size;
110	>	int t_read;
111		#endif
112
113	<	process_stat.sleeping=0;
114	<	process_stat.running=0;
115	<	process_stat.zombie=0;
85	<	process_stat.stopped=0;
86	<	process_stat.total=0;
113	>	if((proc_dir=opendir(PROC_LOCATION))==NULL){
114	>	return -1;
115	>	}
116
117	<	#if defined(SOLARIS) || defined(LINUX)
118	<	if((proc_dir=opendir(PROC_LOCATION))==NULL){
90	<	return NULL;
91	<	}
117	>	while((dir_entry=readdir(proc_dir))!=NULL){
118	>	if(atoi(dir_entry->d_name) == 0) continue;
119
93	–	while((dir_entry=readdir(proc_dir))!=NULL){
94	–	if(atoi(dir_entry->d_name) == 0) continue;
95	–
120		#ifdef SOLARIS
121	<	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
121	>	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
122		#endif
123		#ifdef LINUX
124	<	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/status", dir_entry->d_name);
124	>	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/stat", dir_entry->d_name);
125		#endif
126	+	if((f=fopen(filename, "r"))==NULL){
127	+	/* Open failed.. Process since vanished, or the path was too long.
128	+	* Ah well, move onwards to the next one */
129	+	continue;
130	+	}
131	+	#ifdef SOLARIS
132	+	fread(&process_info, sizeof(psinfo_t), 1, f);
133	+	#endif
134
135	<	if((f=fopen(filename, "r"))==NULL){
136	<	/* Open failed.. Process since vanished, or the path was too long.
137	<	* Ah well, move onwards to the next one */
138	<	continue;
139	<	}
135	>	proc_state = realloc(proc_state, (1+proc_state_size)*sizeof(proc_state_t));
136	>	proc_state_ptr = proc_state+proc_state_size;
137	>	#ifdef SOLARIS
138	>	proc_state_ptr->pid = process_info.pr_pid;
139	>	proc_state_ptr->parent = process_info.pr_ppid;
140	>	proc_state_ptr->pgid = process_info.pr_pgid;
141	>	proc_state_ptr->uid = process_info.pr_uid;
142	>	proc_state_ptr->euid = process_info.pr_euid;
143	>	proc_state_ptr->gid = process_info.pr_gid;
144	>	proc_state_ptr->egid = process_info.pr_egid;
145	>	proc_state_ptr->proc_size = (process_info.pr_size) * 1024;
146	>	proc_state_ptr->proc_resident = (process_info.pr_rssize) * 1024;
147	>	proc_state_ptr->time_spent = process_info.pr_time.tv_sec;
148	>	proc_state_ptr->cpu_percent = (process_info.pr_pctcpu * 100.0) / 0x8000;
149	>	proc_state_ptr->process_name = strdup(process_info.pr_fname);
150	>	proc_state_ptr->proctitle = strdup(process_info.pr_psargs);
151
152	<	#ifdef SOLARIS
153	<	fread(&process_info, sizeof(psinfo_t), 1, f);
154	<	if(process_info.pr_lwp.pr_state==1) process_stat.sleeping++;
155	<	if(process_info.pr_lwp.pr_state==2) process_stat.running++;
156	<	if(process_info.pr_lwp.pr_state==3) process_stat.zombie++;
114	<	if(process_info.pr_lwp.pr_state==4) process_stat.stopped++;
115	<	if(process_info.pr_lwp.pr_state==6) process_stat.running++;
152	>	if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SLEEPING;
153	>	if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = RUNNING;
154	>	if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = ZOMBIE;
155	>	if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = STOPPED;
156	>	if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = RUNNING;
157		#endif
158		#ifdef LINUX
159	<	if((line_ptr=f_read_line(f, "State:"))==NULL){
160	<	fclose(f);
161	<	continue;
159	>	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));
160	>	proc_state_ptr->proc_resident = proc_state_ptr->proc_resident * getpagesize();
161	>	if(s == 'S') proc_state_ptr->state = SLEEPING;
162	>	if(s == 'R') proc_state_ptr->state = RUNNING;
163	>	if(s == 'Z') proc_state_ptr->state = ZOMBIE;
164	>	if(s == 'T') proc_state_ptr->state = STOPPED;
165	>	if(s == 'D') proc_state_ptr->state = STOPPED;
166	>
167	>	/* pa_name[0] should = '(' */
168	>	ptr = strchr(&ps_name[1], ')');
169	>	if(ptr !=NULL) *ptr='\0';
170	>	proc_state_ptr->process_name = strdup(&ps_name[1]);
171	>
172	>	/* Need to do cpu */
173	>
174	>
175	>	/* proctitle */
176	>	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/cmdline", dir_entry->d_name);
177	>
178	>	if((fn=open(filename, O_RDONLY)) == -1){
179	>	/* Open failed.. Process since vanished, or the path was too long.
180	>	* Ah well, move onwards to the next one */
181	>	continue;
182	>	}
183	>	#define         PSARG_START_SIZE 128
184	>	if(psargs == NULL){
185	>	psargs = malloc(PSARG_START_SIZE);
186	>	psarg_size = PSARG_START_SIZE;
187		}
188	<	if((line_ptr=strchr(line_ptr, '\t'))==NULL){
189	<	fclose(f);
190	<	continue;
188	>	ptr = psargs;
189	>	t_read = 0;
190	>	toread = psarg_size;
191	>	while((size = read(fn, ptr, toread)) == toread){
192	>	psargs = realloc(psargs, (psarg_size + PSARG_START_SIZE));
193	>	ptr = psargs+psarg_size;
194	>	t_read = psarg_size;
195	>	psarg_size+=PSARG_START_SIZE;
196	>	toread = PSARG_START_SIZE;
197		}
198	<	line_ptr++;
199	<	if(line_ptr=='\0'){
200	<	fclose(f);
201	<	continue;
198	>	if(size != -1) t_read+=size;
199	>
200	>	ptr = psargs;
201	>	for(x=0; x<t_read; x++){
202	>	if (*ptr == '\0') *ptr = ' ';
203	>	ptr++;
204		}
205	+	/*  for safety sake */
206	+	psargs[t_read] = '\0';
207
208	<	if(*line_ptr=='S') process_stat.sleeping++;
133	<	if(*line_ptr=='R') process_stat.running++;
134	<	if(*line_ptr=='Z') process_stat.zombie++;
135	<	if(*line_ptr=='T') process_stat.stopped++;
136	<	if(*line_ptr=='D') process_stat.stopped++;
137	<	#endif
208	>	proc_state_ptr->proctitle = strdup(psargs);
209
139	–	fclose(f);
140	–	}
141	–	closedir(proc_dir);
210		#endif
211	+
212	+	proc_state_size++;
213	+
214	+	fclose(f);
215	+	}
216	+	closedir(proc_dir);
217	+	#endif
218	+
219		#ifdef ALLBSD
220		mib[0] = CTL_KERN;
221		mib[1] = KERN_PROC;
222		mib[2] = KERN_PROC_ALL;
223
224	<	if (sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
224	>	if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
225		return NULL;
226		}
227	<
227	>
228		procs = size / sizeof(struct kinfo_proc);
229
230		kp_stats = malloc(size);
231	<	if (kp_stats == NULL) {
231	>	if(kp_stats == NULL) {
232		return NULL;
233		}
234	<
235	<	if (sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
234	>
235	>	if(sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
236	>	free(kp_stats);
237		return NULL;
238		}
239
240	+	#if (defined(FREEBSD) && !defined(FREEBSD5)) || defined(DFBSD)
241	+	kvmd = get_kvm2();
242	+	#endif
243	+
244		for (i = 0; i < procs; i++) {
245	+	/* replace with something more sensible */
246	+	proc_state = realloc(proc_state,
247	+	(1+proc_state_size)*sizeof(proc_state_t));
248	+	if(proc_state == NULL ) {
249	+	return NULL;
250	+	}
251	+	proc_state_ptr = proc_state+proc_state_size;
252	+
253		#ifdef FREEBSD5
254	+	proc_state_ptr->process_name =
255	+	strdup(kp_stats[i].ki_comm);
256	+	#elif defined(DFBSD)
257	+	proc_state_ptr->process_name =
258	+	strdup(kp_stats[i].kp_thread.td_comm);
259	+	#else
260	+	proc_state_ptr->process_name =
261	+	strdup(kp_stats[i].kp_proc.p_comm);
262	+	#endif
263	+
264	+	#if defined(FREEBSD5) || defined(NETBSD) || defined(OPENBSD)
265	+	size = sizeof(buflen);
266	+
267	+	#ifdef FREEBSD5
268	+	if(sysctlbyname("kern.ps_arg_cache_limit", &buflen, &size, NULL, 0) < 0) {
269	+	return NULL;
270	+	}
271	+	#else
272	+	mib[1] = KERN_ARGMAX;
273	+
274	+	if(sysctl(mib, 2, &buflen, &size, NULL, 0) < 0) {
275	+	return NULL;
276	+	}
277	+	#endif
278	+
279	+	proctitle = malloc(buflen);
280	+	if(proctitle == NULL) {
281	+	return NULL;
282	+	}
283	+
284	+	size = buflen;
285	+
286	+	#ifdef FREEBSD5
287	+	mib[2] = KERN_PROC_ARGS;
288	+	mib[3] = kp_stats[i].ki_pid;
289	+	#else
290	+	mib[1] = KERN_PROC_ARGS;
291	+	mib[2] = kp_stats[i].kp_proc.p_pid;
292	+	mib[3] = KERN_PROC_ARGV;
293	+	#endif
294	+
295	+	if(sysctl(mib, 4, proctitle, &size, NULL, 0) < 0) {
296	+	free(proctitle);
297	+	proc_state_ptr->proctitle = NULL;
298	+	}
299	+	else if(size > 0) {
300	+	proc_state_ptr->proctitle = malloc(size+1);
301	+	if(proc_state_ptr->proctitle == NULL) {
302	+	return NULL;
303	+	}
304	+	p = proctitle;
305	+	proc_state_ptr->proctitle[0] = NULL;
306	+	do {
307	+	strlcat(proc_state_ptr->proctitle, p, size+1);
308	+	strlcat(proc_state_ptr->proctitle, " ", size+1);
309	+	p += strlen(p) + 1;
310	+	} while (p < proctitle + size);
311	+	free(proctitle);
312	+	/* remove trailing space */
313	+	proc_state_ptr->proctitle[strlen(proc_state_ptr->proctitle)-1] = \0;
314	+	}
315	+	else {
316	+	free(proctitle);
317	+	proc_state_ptr->proctitle = NULL;
318	+	}
319	+	#else
320	+	if(kvmd != NULL) {
321	+	args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
322	+	if(args != NULL) {
323	+	alloc = 1;
324	+	proctitle = malloc(alloc);
325	+	if(proctitle == NULL) {
326	+	return NULL;
327	+	}
328	+	while(*args != NULL) {
329	+	if(strlen(proctitle) + strlen(*args) >= alloc) {
330	+	alloc = (alloc + strlen(*args)) * 2;
331	+	proctitletmp = realloc(proctitle, alloc);
332	+	if(proctitletmp == NULL) {
333	+	return NULL;
334	+	}
335	+	proctitle = proctitletmp;
336	+	}
337	+	strlcat(proctitle, *args, alloc);
338	+	strlcat(proctitle, " ", alloc);
339	+	args++;
340	+	}
341	+	/* remove trailing space */
342	+	proctitle[strlen(proctitle)-1] = \0;
343	+	proc_state_ptr->proctitle = proctitle;
344	+	}
345	+	else {
346	+	proc_state_ptr->proctitle = NULL;
347	+	}
348	+	}
349	+	else {
350	+	proc_state_ptr->proctitle = NULL;
351	+	}
352	+	#endif
353	+
354	+	#ifdef FREEBSD5
355	+	proc_state_ptr->pid = kp_stats[i].ki_pid;
356	+	proc_state_ptr->parent = kp_stats[i].ki_ppid;
357	+	proc_state_ptr->pgid = kp_stats[i].ki_pgid;
358	+	#else
359	+	proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
360	+	proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
361	+	proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
362	+	#endif
363	+
364	+	#ifdef FREEBSD5
365	+	proc_state_ptr->uid = kp_stats[i].ki_ruid;
366	+	proc_state_ptr->euid = kp_stats[i].ki_uid;
367	+	proc_state_ptr->gid = kp_stats[i].ki_rgid;
368	+	proc_state_ptr->egid = kp_stats[i].ki_svgid;
369	+	#elif defined(DFBSD)
370	+	proc_state_ptr->uid = kp_stats[i].kp_eproc.e_ucred.cr_ruid;
371	+	proc_state_ptr->euid = kp_stats[i].kp_eproc.e_ucred.cr_svuid;
372	+	proc_state_ptr->gid = kp_stats[i].kp_eproc.e_ucred.cr_rgid;
373	+	proc_state_ptr->egid = kp_stats[i].kp_eproc.e_ucred.cr_svgid;
374	+	#else
375	+	proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
376	+	proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
377	+	proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
378	+	proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
379	+	#endif
380	+
381	+	#ifdef FREEBSD5
382	+	proc_state_ptr->proc_size = kp_stats[i].ki_size;
383	+	/* This is in pages */
384	+	proc_state_ptr->proc_resident =
385	+	kp_stats[i].ki_rssize * getpagesize();
386	+	/* This is in microseconds */
387	+	proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
388	+	proc_state_ptr->cpu_percent =
389	+	((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
390	+	proc_state_ptr->nice = kp_stats[i].ki_nice;
391	+	#else
392	+	proc_state_ptr->proc_size =
393	+	kp_stats[i].kp_eproc.e_vm.vm_map.size;
394	+	/* This is in pages */
395	+	proc_state_ptr->proc_resident =
396	+	kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
397	+	#if defined(NETBSD) || defined(OPENBSD)
398	+	proc_state_ptr->time_spent =
399	+	kp_stats[i].kp_proc.p_rtime.tv_sec;
400	+	#elif defined(DFBSD)
401	+	proc_state_ptr->time_spent =
402	+	( kp_stats[i].kp_thread.td_uticks +
403	+	kp_stats[i].kp_thread.td_sticks +
404	+	kp_stats[i].kp_thread.td_iticks ) / 1000000;
405	+	#else
406	+	/* This is in microseconds */
407	+	proc_state_ptr->time_spent =
408	+	kp_stats[i].kp_proc.p_runtime / 1000000;
409	+	#endif
410	+	proc_state_ptr->cpu_percent =
411	+	((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
412	+	proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
413	+	#endif
414	+
415	+	#ifdef FREEBSD5
416		switch (kp_stats[i].ki_stat) {
417		#else
418		switch (kp_stats[i].kp_proc.p_stat) {
#	Line 171 | Line 422 | process_stat_t *get_process_stats(){
422		#ifdef SONPROC
423		case SONPROC: /* NetBSD */
424		#endif
425	<	process_stat.running++;
425	>	proc_state_ptr->state = RUNNING;
426		break;
427		case SSLEEP:
428		#ifdef SWAIT
#	Line 180 | Line 431 | process_stat_t *get_process_stats(){
431		#ifdef SLOCK
432		case SLOCK: /* FreeBSD 5 */
433		#endif
434	<	process_stat.sleeping++;
434	>	proc_state_ptr->state = SLEEPING;
435		break;
436		case SSTOP:
437	<	process_stat.stopped++;
437	>	proc_state_ptr->state = STOPPED;
438		break;
439		case SZOMB:
440		#ifdef SDEAD
441		case SDEAD: /* OpenBSD & NetBSD */
442		#endif
443	<	process_stat.zombie++;
443	>	proc_state_ptr->state = ZOMBIE;
444		break;
445	<	}
445	>	default:
446	>	proc_state_ptr->state = UNKNOWN;
447	>	break;
448	>	}
449	>	proc_state_size++;
450		}
451	+
452	+	free(kp_stats);
453		#endif
454
455	<	free(kp_stats);
455	>	*ps = proc_state;
456	>	return proc_state_size;
457	>	}
458
459	<	#ifdef CYGWIN
460	<	return NULL;
461	<	#endif
459	>	process_stat_t *get_process_stats() {
460	>	static process_stat_t process_stat;
461	>	proc_state_t *ps;
462	>	int ps_size, x;
463
464	<	process_stat.total=process_stat.sleeping+process_stat.running+process_stat.zombie+process_stat.stopped;
464	>	process_stat.sleeping = 0;
465	>	process_stat.running = 0;
466	>	process_stat.zombie = 0;
467	>	process_stat.stopped = 0;
468	>	process_stat.total = 0;
469	>
470	>	ps_size = get_proc_snapshot(&ps);
471	>
472	>	if(ps_size == 0) {
473	>	return NULL;
474	>	}
475	>
476	>	for(x = 0; x < ps_size; x++) {
477	>	switch (ps->state) {
478	>	case RUNNING:
479	>	process_stat.running++;
480	>	break;
481	>	case SLEEPING:
482	>	process_stat.sleeping++;
483	>	break;
484	>	case STOPPED:
485	>	process_stat.stopped++;
486	>	break;
487	>	case ZOMBIE:
488	>	process_stat.zombie++;
489	>	break;
490	>	default:
491	>	/* currently no mapping for UNKNOWN in process_stat_t */
492	>	break;
493	>	}
494	>	ps++;
495	>	}
496	>
497	>	process_stat.total = ps_size;
498
499		return &process_stat;
500		}

Diff Legend

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