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.19 by tdb, Sat Feb 14 02:56:11 2004 UTC vs.
Revision 1.48 by ats, Mon Apr 5 01:41:25 2004 UTC

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

Diff Legend

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