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.33 by tdb, Sun Apr 4 11:28:46 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 61 | Line 63
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[3];
89	>	int mib[4];
90		size_t size;
91		struct kinfo_proc *kp_stats;
92	<	int procs, i, alloc;
76	<	static kvm_t *kvmd;
77	<	char **args;
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 90 | Line 114 | int get_proc_snapshot(proc_state_t **ps){
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	<	static char *psargs = NULL;
94	<	static int psarg_size = 0;
117	>	VECTOR_DECLARE_STATIC(psargs, char, 128, NULL, NULL);
118		unsigned long stime, utime;
119		int x;
120		int fn;
121	<	int toread;
122	<	ssize_t size;
100	<	int t_read;
121	>	int len;
122	>	int rc;
123		#endif
124
125		if((proc_dir=opendir(PROC_LOCATION))==NULL){
#	Line 122 | Line 144 | int get_proc_snapshot(proc_state_t **ps){
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	+
152		#ifdef SOLARIS
153		proc_state_ptr->pid = process_info.pr_pid;
154		proc_state_ptr->parent = process_info.pr_ppid;
#	Line 146 | Line 171 | int get_proc_snapshot(proc_state_t **ps){
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 %*lu %*lu %*lu %*lu %*lu %lu %lu %*ld %*ld %*ld %d %*ld %*ld %*lu %llu %llu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*lu %*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));
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;
#	Line 157 | Line 182 | int get_proc_snapshot(proc_state_t **ps){
182		/* pa_name[0] should = '(' */
183		ptr = strchr(&ps_name[1], ')');
184		if(ptr !=NULL) *ptr='\0';
160	–	proc_state_ptr->process_name = strdup(&ps_name[1]);
185
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
#	Line 170 | Line 201 | int get_proc_snapshot(proc_state_t **ps){
201		* Ah well, move onwards to the next one */
202		continue;
203		}
204	<	#define         PSARG_START_SIZE 128
205	<	if(psargs == NULL){
206	<	psargs = malloc(PSARG_START_SIZE);
207	<	psarg_size = PSARG_START_SIZE;
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		}
178	–	ptr = psargs;
179	–	t_read = 0;
180	–	toread = psarg_size;
181	–	while((size = read(fn, ptr, toread)) == toread){
182	–	psargs = realloc(psargs, (psarg_size + PSARG_START_SIZE));
183	–	ptr = psargs+psarg_size;
184	–	t_read = psarg_size;
185	–	psarg_size+=PSARG_START_SIZE;
186	–	toread = PSARG_START_SIZE;
187	–	}
188	–	if(size != -1) t_read+=size;
222
223	+	/* Turn \0s into spaces within the command line. */
224		ptr = psargs;
225	<	for(x=0; x<t_read; x++){
225	>	for(x = 0; x < len; x++) {
226		if (*ptr == '\0') *ptr = ' ';
227		ptr++;
228		}
229	<	/*  for safety sake */
230	<	psargs[t_read] = '\0';
229	>	/* for safety's sake */
230	>	psargs[len] = '\0';
231
232	<	proc_state_ptr->proctitle = strdup(psargs);
233	<
232	>	if (update_string(&proc_state_ptr->proctitle, psargs) == NULL) {
233	>	return -1;
234	>	}
235		#endif
236
237		proc_state_size++;
203	–
204	–	fclose(f);
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	<	#ifdef NETBSD
248	<	kvmd = kvm_openfiles(NULL, NULL, NULL, O_RDONLY, NULL);
249	<	#else
214	<	kvmd = kvm_openfiles(_PATH_DEVNULL, _PATH_DEVNULL, NULL, O_RDONLY, NULL);
215	<	#endif
247	>	if(sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
248	>	return -1;
249	>	}
250
251	<	if(kvmd == NULL) return NULL;
251	>	procs = size / sizeof(struct kinfo_proc);
252
253	<	kp_stats = kvm_getprocs(kvmd, KERN_PROC_ALL, 0, &procs);
253	>	kp_stats = malloc(size);
254	>	if(kp_stats == NULL) {
255	>	return -1;
256	>	}
257
258	<	if (kp_stats == NULL || procs < 0) {
259	<	return NULL;
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	<	/* replace with something more sensible */
269	<	proc_state = realloc(proc_state,
270	<	(1+proc_state_size)*sizeof(proc_state_t));
271	<	if(proc_state == NULL ) {
230	<	return NULL;
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	<	proc_state_ptr->process_name =
277	<	strdup(kp_stats[i].ki_comm);
276	>	name = kp_stats[i].ki_comm;
277	>	#elif defined(DFBSD)
278	>	name = kp_stats[i].kp_thread.td_comm;
279		#else
280	<	proc_state_ptr->process_name =
239	<	strdup(kp_stats[i].kp_proc.p_comm);
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	<	args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
287	<	if(args != NULL) {
288	<	alloc = 1;
289	<	proctitle = malloc(alloc);
290	<	if(proctitle == NULL) {
291	<	return NULL;
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	<	while(*args != NULL) {
328	<	if(strlen(proctitle) + strlen(*args) >= alloc) {
329	<	alloc = (alloc + strlen(*args)) << 1;
330	<	proctitle = realloc(proctitle, alloc);
331	<	if(proctitle == NULL) {
332	<	return NULL;
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	<	strncat(proctitle, *args, strlen(*args));
366	<	strncat(proctitle, " ", 1);
367	<	args++;
365	>	/* remove trailing space */
366	>	proctitle[strlen(proctitle)-1] = '\0';
367	>	proc_state_ptr->proctitle = proctitle;
368		}
369	<	/* remove trailing space */
370	<	proctitle[strlen(proctitle)-1] = NULL;
371	<	proc_state_ptr->proctitle = proctitle;
369	>	else {
370	>	proc_state_ptr->proctitle = NULL;
371	>	}
372		}
373		else {
374	<	proc_state_ptr->proctitle =
267	<	malloc(strlen(proc_state_ptr->process_name)+4);
268	<	if(proc_state_ptr->proctitle == NULL) {
269	<	return NULL;
270	<	}
271	<	sprintf(proc_state_ptr->proctitle, " (%s)",
272	<	proc_state_ptr->process_name);
374	>	proc_state_ptr->proctitle = NULL;
375		}
376	+	#endif
377
378		#ifdef FREEBSD5
379		proc_state_ptr->pid = kp_stats[i].ki_pid;
#	Line 287 | Line 390 | int get_proc_snapshot(proc_state_t **ps){
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;
#	Line 310 | Line 418 | int get_proc_snapshot(proc_state_t **ps){
418		/* This is in pages */
419		proc_state_ptr->proc_resident =
420		kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
421	<	/* This is in microseconds */
314	<	#ifdef NETBSD
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
#	Line 363 | Line 476 | int get_proc_snapshot(proc_state_t **ps){
476		free(kp_stats);
477		#endif
478
479	+	#ifdef CYGWIN
480	+	return -1;
481	+	#endif
482	+
483		*ps = proc_state;
484		return proc_state_size;
485		}
#	Line 380 | Line 497 | process_stat_t *get_process_stats() {
497
498		ps_size = get_proc_snapshot(&ps);
499
500	<	if(ps_size == NULL) {
500	>	if(ps_size < 0) {
501		return NULL;
502		}
503
504		for(x = 0; x < ps_size; x++) {
505		switch (ps->state) {
389	–	/* currently no mapping for UNKNOWN in process_stat_t */
506		case RUNNING:
507		process_stat.running++;
508		break;
#	Line 398 | Line 514 | process_stat_t *get_process_stats() {
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++;

Diff Legend

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