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.24 by pajs, Sun Mar 28 18:22:58 2004 UTC vs.
Revision 1.49 by ats, Mon Apr 5 01:44:20 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 42 | 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 54 | Line 59
59		#else
60		#include <sys/proc.h>
61		#endif
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	+	static void proc_state_init(proc_state_t *s) {
74	+	s->process_name = NULL;
75	+	s->proctitle = NULL;
76	+	}
77	+
78	+	static void proc_state_destroy(proc_state_t *s) {
79	+	free(s->process_name);
80	+	free(s->proctitle);
81	+	}
82	+
83		int get_proc_snapshot(proc_state_t **ps){
84	<	proc_state_t *proc_state = NULL;
85	<	proc_state_t *proc_state_ptr;
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
103	+	#endif
104		#if defined(SOLARIS) || defined(LINUX)
105		DIR *proc_dir;
106		struct dirent *dir_entry;
#	Line 68 | Line 109 | int get_proc_snapshot(proc_state_t **ps){
109		#ifdef SOLARIS
110		psinfo_t process_info;
111		#endif
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		if((proc_dir=opendir(PROC_LOCATION))==NULL){
126	<	return NULL;
126	>	return -1;
127		}
128
129		while((dir_entry=readdir(proc_dir))!=NULL){
#	Line 79 | Line 132 | int get_proc_snapshot(proc_state_t **ps){
132		#ifdef SOLARIS
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/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 */
#	Line 86 | Line 142 | int get_proc_snapshot(proc_state_t **ps){
142		}
143		#ifdef SOLARIS
144		fread(&process_info, sizeof(psinfo_t), 1, f);
145	+	#endif
146
147	<	proc_state = realloc(proc_state, (1+proc_state_size)*sizeof(proc_state_t));
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	<
151	>
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;
#	Line 110 | Line 170 | int get_proc_snapshot(proc_state_t **ps){
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	+	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	<	proc_state_size++;
187	<	#endif
186	>	if (update_string(&proc_state_ptr->process_name,
187	>	&ps_name[1]) == NULL) {
188	>	return -1;
189	>	}
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	+
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 (update_string(&proc_state_ptr->proctitle, psargs) == NULL) {
233	+	return -1;
234	+	}
235	+	#endif
236	+
237	+	proc_state_size++;
238		}
239		closedir(proc_dir);
240	<	*ps = proc_state;
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 -1;
249	+	}
250	+
251	+	procs = size / sizeof(struct kinfo_proc);
252	+
253	+	kp_stats = malloc(size);
254	+	if(kp_stats == NULL) {
255	+	return -1;
256	+	}
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] = '\0';
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	+	proctitle[0] = '\0';
353	+	while(*args != NULL) {
354	+	if(strlen(proctitle) + strlen(*args) >= alloc) {
355	+	alloc = (alloc + strlen(*args)) * 2;
356	+	proctitletmp = realloc(proctitle, alloc);
357	+	if(proctitletmp == NULL) {
358	+	free(proctitle);
359	+	return -1;
360	+	}
361	+	proctitle = proctitletmp;
362	+	}
363	+	strlcat(proctitle, *args, alloc);
364	+	strlcat(proctitle, " ", alloc);
365	+	args++;
366	+	}
367	+	/* remove trailing space */
368	+	proctitle[strlen(proctitle)-1] = '\0';
369	+	proc_state_ptr->proctitle = proctitle;
370	+	}
371	+	else {
372	+	proc_state_ptr->proctitle = NULL;
373	+	}
374	+	}
375	+	else {
376	+	proc_state_ptr->proctitle = NULL;
377	+	}
378	+	#endif
379	+
380	+	#ifdef FREEBSD5
381	+	proc_state_ptr->pid = kp_stats[i].ki_pid;
382	+	proc_state_ptr->parent = kp_stats[i].ki_ppid;
383	+	proc_state_ptr->pgid = kp_stats[i].ki_pgid;
384	+	#else
385	+	proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
386	+	proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
387	+	proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
388	+	#endif
389	+
390	+	#ifdef FREEBSD5
391	+	proc_state_ptr->uid = kp_stats[i].ki_ruid;
392	+	proc_state_ptr->euid = kp_stats[i].ki_uid;
393	+	proc_state_ptr->gid = kp_stats[i].ki_rgid;
394	+	proc_state_ptr->egid = kp_stats[i].ki_svgid;
395	+	#elif defined(DFBSD)
396	+	proc_state_ptr->uid = kp_stats[i].kp_eproc.e_ucred.cr_ruid;
397	+	proc_state_ptr->euid = kp_stats[i].kp_eproc.e_ucred.cr_svuid;
398	+	proc_state_ptr->gid = kp_stats[i].kp_eproc.e_ucred.cr_rgid;
399	+	proc_state_ptr->egid = kp_stats[i].kp_eproc.e_ucred.cr_svgid;
400	+	#else
401	+	proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
402	+	proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
403	+	proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
404	+	proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
405	+	#endif
406	+
407	+	#ifdef FREEBSD5
408	+	proc_state_ptr->proc_size = kp_stats[i].ki_size;
409	+	/* This is in pages */
410	+	proc_state_ptr->proc_resident =
411	+	kp_stats[i].ki_rssize * getpagesize();
412	+	/* This is in microseconds */
413	+	proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
414	+	proc_state_ptr->cpu_percent =
415	+	((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
416	+	proc_state_ptr->nice = kp_stats[i].ki_nice;
417	+	#else
418	+	proc_state_ptr->proc_size =
419	+	kp_stats[i].kp_eproc.e_vm.vm_map.size;
420	+	/* This is in pages */
421	+	proc_state_ptr->proc_resident =
422	+	kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
423	+	#if defined(NETBSD) || defined(OPENBSD)
424	+	proc_state_ptr->time_spent =
425	+	kp_stats[i].kp_proc.p_rtime.tv_sec;
426	+	#elif defined(DFBSD)
427	+	proc_state_ptr->time_spent =
428	+	( kp_stats[i].kp_thread.td_uticks +
429	+	kp_stats[i].kp_thread.td_sticks +
430	+	kp_stats[i].kp_thread.td_iticks ) / 1000000;
431	+	#else
432	+	/* This is in microseconds */
433	+	proc_state_ptr->time_spent =
434	+	kp_stats[i].kp_proc.p_runtime / 1000000;
435	+	#endif
436	+	proc_state_ptr->cpu_percent =
437	+	((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
438	+	proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
439	+	#endif
440	+
441	+	#ifdef FREEBSD5
442	+	switch (kp_stats[i].ki_stat) {
443	+	#else
444	+	switch (kp_stats[i].kp_proc.p_stat) {
445	+	#endif
446	+	case SIDL:
447	+	case SRUN:
448	+	#ifdef SONPROC
449	+	case SONPROC: /* NetBSD */
450	+	#endif
451	+	proc_state_ptr->state = RUNNING;
452	+	break;
453	+	case SSLEEP:
454	+	#ifdef SWAIT
455	+	case SWAIT: /* FreeBSD 5 */
456	+	#endif
457	+	#ifdef SLOCK
458	+	case SLOCK: /* FreeBSD 5 */
459	+	#endif
460	+	proc_state_ptr->state = SLEEPING;
461	+	break;
462	+	case SSTOP:
463	+	proc_state_ptr->state = STOPPED;
464	+	break;
465	+	case SZOMB:
466	+	#ifdef SDEAD
467	+	case SDEAD: /* OpenBSD & NetBSD */
468	+	#endif
469	+	proc_state_ptr->state = ZOMBIE;
470	+	break;
471	+	default:
472	+	proc_state_ptr->state = UNKNOWN;
473	+	break;
474	+	}
475	+	proc_state_size++;
476	+	}
477	+
478	+	free(kp_stats);
479	+	#endif
480	+
481	+	#ifdef CYGWIN
482	+	return -1;
483	+	#endif
484	+
485	+	*ps = proc_state;
486		return proc_state_size;
487		}
488
489	+	process_stat_t *get_process_stats() {
490	+	static process_stat_t process_stat;
491	+	proc_state_t *ps;
492	+	int ps_size, x;
493	+
494	+	process_stat.sleeping = 0;
495	+	process_stat.running = 0;
496	+	process_stat.zombie = 0;
497	+	process_stat.stopped = 0;
498	+	process_stat.total = 0;
499	+
500	+	ps_size = get_proc_snapshot(&ps);
501	+
502	+	if(ps_size < 0) {
503	+	return NULL;
504	+	}
505	+
506	+	for(x = 0; x < ps_size; x++) {
507	+	switch (ps->state) {
508	+	case RUNNING:
509	+	process_stat.running++;
510	+	break;
511	+	case SLEEPING:
512	+	process_stat.sleeping++;
513	+	break;
514	+	case STOPPED:
515	+	process_stat.stopped++;
516	+	break;
517	+	case ZOMBIE:
518	+	process_stat.zombie++;
519	+	break;
520	+	default:
521	+	/* currently no mapping for UNKNOWN in process_stat_t */
522	+	break;
523	+	}
524	+	ps++;
525	+	}
526	+
527	+	process_stat.total = ps_size;
528	+
529	+	return &process_stat;
530	+	}

Diff Legend

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