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.23 by pajs, Sun Mar 28 18:04:14 2004 UTC vs.
Revision 1.35 by tdb, Sun Apr 4 15:56:53 2004 UTC

#	Line 42 | 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 54 | Line 57
57		#else
58		#include <sys/proc.h>
59		#endif
60	+	#include <string.h>
61	+	#include <paths.h>
62	+	#include <fcntl.h>
63	+	#include <limits.h>
64	+	#include <kvm.h>
65		#endif
66
67		int get_proc_snapshot(proc_state_t **ps){
68		proc_state_t *proc_state = NULL;
69		proc_state_t *proc_state_ptr;
70		int proc_state_size = 0;
71	<	#if defined(SOLARIS)
71	>	#ifdef ALLBSD
72	>	int mib[3];
73	>	size_t size;
74	>	struct kinfo_proc *kp_stats;
75	>	int procs, i, alloc;
76	>	static kvm_t *kvmd;
77	>	char **args;
78	>	char *proctitle;
79	>	#endif
80	>	#if defined(SOLARIS) || defined(LINUX)
81		DIR *proc_dir;
82		struct dirent *dir_entry;
83		char filename[MAX_FILE_LENGTH];
84		FILE *f;
85	+	#ifdef SOLARIS
86		psinfo_t process_info;
87	+	#endif
88	+	#ifdef LINUX
89	+	char s;
90	+	/* If someone has a executable of 4k filename length, they deserve to get it truncated :) */
91	+	char ps_name[4096];
92	+	char *ptr;
93	+	static char *psargs = NULL;
94	+	static int psarg_size = 0;
95	+	unsigned long stime, utime;
96	+	int x;
97	+	int fn;
98	+	int toread;
99	+	ssize_t size;
100	+	int t_read;
101	+	#endif
102
103		if((proc_dir=opendir(PROC_LOCATION))==NULL){
104	<	return NULL;
104	>	return -1;
105		}
106
107		while((dir_entry=readdir(proc_dir))!=NULL){
108		if(atoi(dir_entry->d_name) == 0) continue;
109
110	+	#ifdef SOLARIS
111		snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
112	<
112	>	#endif
113	>	#ifdef LINUX
114	>	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/stat", dir_entry->d_name);
115	>	#endif
116		if((f=fopen(filename, "r"))==NULL){
117		/* Open failed.. Process since vanished, or the path was too long.
118		* Ah well, move onwards to the next one */
119		continue;
120		}
121	+	#ifdef SOLARIS
122		fread(&process_info, sizeof(psinfo_t), 1, f);
123	+	#endif
124
125		proc_state = realloc(proc_state, (1+proc_state_size)*sizeof(proc_state_t));
87	–
126		proc_state_ptr = proc_state+proc_state_size;
127	<
127	>	#ifdef SOLARIS
128		proc_state_ptr->pid = process_info.pr_pid;
129		proc_state_ptr->parent = process_info.pr_ppid;
130		proc_state_ptr->pgid = process_info.pr_pgid;
#	Line 101 | Line 139 | int get_proc_snapshot(proc_state_t **ps){
139		proc_state_ptr->process_name = strdup(process_info.pr_fname);
140		proc_state_ptr->proctitle = strdup(process_info.pr_psargs);
141
142	<	proc_state_size++;
142	>	if(process_info.pr_lwp.pr_state==1) proc_state_ptr->state = SLEEPING;
143	>	if(process_info.pr_lwp.pr_state==2) proc_state_ptr->state = RUNNING;
144	>	if(process_info.pr_lwp.pr_state==3) proc_state_ptr->state = ZOMBIE;
145	>	if(process_info.pr_lwp.pr_state==4) proc_state_ptr->state = STOPPED;
146	>	if(process_info.pr_lwp.pr_state==6) proc_state_ptr->state = RUNNING;
147		#endif
148	+	#ifdef LINUX
149	+	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));
150	+	proc_state_ptr->proc_resident = proc_state_ptr->proc_resident * getpagesize();
151	+	if(s == 'S') proc_state_ptr->state = SLEEPING;
152	+	if(s == 'R') proc_state_ptr->state = RUNNING;
153	+	if(s == 'Z') proc_state_ptr->state = ZOMBIE;
154	+	if(s == 'T') proc_state_ptr->state = STOPPED;
155	+	if(s == 'D') proc_state_ptr->state = STOPPED;
156	+
157	+	/* pa_name[0] should = '(' */
158	+	ptr = strchr(&ps_name[1], ')');
159	+	if(ptr !=NULL) *ptr='\0';
160	+	proc_state_ptr->process_name = strdup(&ps_name[1]);
161
162	+	/* Need to do cpu */
163	+
164
165	+	/* proctitle */
166	+	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/cmdline", dir_entry->d_name);
167	+
168	+	if((fn=open(filename, O_RDONLY)) == -1){
169	+	/* Open failed.. Process since vanished, or the path was too long.
170	+	* Ah well, move onwards to the next one */
171	+	continue;
172	+	}
173	+	#define         PSARG_START_SIZE 128
174	+	if(psargs == NULL){
175	+	psargs = malloc(PSARG_START_SIZE);
176	+	psarg_size = PSARG_START_SIZE;
177	+	}
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;
189	+
190	+	ptr = psargs;
191	+	for(x=0; x<t_read; x++){
192	+	if (*ptr == '\0') *ptr = ' ';
193	+	ptr++;
194	+	}
195	+	/*  for safety sake */
196	+	psargs[t_read] = '\0';
197	+
198	+	proc_state_ptr->proctitle = strdup(psargs);
199	+
200	+	#endif
201	+
202	+	proc_state_size++;
203	+
204		fclose(f);
205		}
206		closedir(proc_dir);
207	<	*ps = proc_state;
207	>	#endif
208
209	+	#ifdef ALLBSD
210	+
211	+	#if defined(NETBSD) || defined(OPENBSD)
212	+	kvmd = kvm_openfiles(NULL, NULL, NULL, O_RDONLY, NULL);
213	+	#else
214	+	kvmd = kvm_openfiles(_PATH_DEVNULL, _PATH_DEVNULL, NULL, O_RDONLY, NULL);
215	+	#endif
216	+
217	+	if(kvmd == NULL) return NULL;
218	+
219	+	kp_stats = kvm_getprocs(kvmd, KERN_PROC_ALL, 0, &procs);
220	+
221	+	if (kp_stats == NULL || procs < 0) {
222	+	return NULL;
223	+	}
224	+
225	+	for (i = 0; i < procs; i++) {
226	+	/* replace with something more sensible */
227	+	proc_state = realloc(proc_state,
228	+	(1+proc_state_size)*sizeof(proc_state_t));
229	+	if(proc_state == NULL ) {
230	+	return NULL;
231	+	}
232	+	proc_state_ptr = proc_state+proc_state_size;
233	+
234	+	#ifdef FREEBSD5
235	+	proc_state_ptr->process_name =
236	+	strdup(kp_stats[i].ki_comm);
237	+	#elif defined(DFBSD)
238	+	proc_state_ptr->process_name =
239	+	strdup(kp_stats[i].kp_thread.td_comm);
240	+	#else
241	+	proc_state_ptr->process_name =
242	+	strdup(kp_stats[i].kp_proc.p_comm);
243	+	#endif
244	+
245	+	args = kvm_getargv(kvmd, &(kp_stats[i]), 0);
246	+	if(args != NULL) {
247	+	alloc = 1;
248	+	proctitle = malloc(alloc);
249	+	if(proctitle == NULL) {
250	+	return NULL;
251	+	}
252	+	while(*args != NULL) {
253	+	if(strlen(proctitle) + strlen(*args) >= alloc) {
254	+	alloc = (alloc + strlen(*args)) << 1;
255	+	proctitle = realloc(proctitle, alloc);
256	+	if(proctitle == NULL) {
257	+	return NULL;
258	+	}
259	+	}
260	+	strncat(proctitle, *args, strlen(*args));
261	+	strncat(proctitle, " ", 1);
262	+	args++;
263	+	}
264	+	/* remove trailing space */
265	+	proctitle[strlen(proctitle)-1] = NULL;
266	+	proc_state_ptr->proctitle = proctitle;
267	+	}
268	+	else {
269	+	proc_state_ptr->proctitle =
270	+	malloc(strlen(proc_state_ptr->process_name)+4);
271	+	if(proc_state_ptr->proctitle == NULL) {
272	+	return NULL;
273	+	}
274	+	sprintf(proc_state_ptr->proctitle, " (%s)",
275	+	proc_state_ptr->process_name);
276	+	}
277	+
278	+	#ifdef FREEBSD5
279	+	proc_state_ptr->pid = kp_stats[i].ki_pid;
280	+	proc_state_ptr->parent = kp_stats[i].ki_ppid;
281	+	proc_state_ptr->pgid = kp_stats[i].ki_pgid;
282	+	#else
283	+	proc_state_ptr->pid = kp_stats[i].kp_proc.p_pid;
284	+	proc_state_ptr->parent = kp_stats[i].kp_eproc.e_ppid;
285	+	proc_state_ptr->pgid = kp_stats[i].kp_eproc.e_pgid;
286	+	#endif
287	+
288	+	#ifdef FREEBSD5
289	+	proc_state_ptr->uid = kp_stats[i].ki_ruid;
290	+	proc_state_ptr->euid = kp_stats[i].ki_uid;
291	+	proc_state_ptr->gid = kp_stats[i].ki_rgid;
292	+	proc_state_ptr->egid = kp_stats[i].ki_svgid;
293	+	#elif defined(DFBSD)
294	+	proc_state_ptr->uid = kp_stats[i].kp_eproc.e_ucred.cr_ruid;
295	+	proc_state_ptr->euid = kp_stats[i].kp_eproc.e_ucred.cr_svuid;
296	+	proc_state_ptr->gid = kp_stats[i].kp_eproc.e_ucred.cr_rgid;
297	+	proc_state_ptr->egid = kp_stats[i].kp_eproc.e_ucred.cr_svgid;
298	+	#else
299	+	proc_state_ptr->uid = kp_stats[i].kp_eproc.e_pcred.p_ruid;
300	+	proc_state_ptr->euid = kp_stats[i].kp_eproc.e_pcred.p_svuid;
301	+	proc_state_ptr->gid = kp_stats[i].kp_eproc.e_pcred.p_rgid;
302	+	proc_state_ptr->egid = kp_stats[i].kp_eproc.e_pcred.p_svgid;
303	+	#endif
304	+
305	+	#ifdef FREEBSD5
306	+	proc_state_ptr->proc_size = kp_stats[i].ki_size;
307	+	/* This is in pages */
308	+	proc_state_ptr->proc_resident =
309	+	kp_stats[i].ki_rssize * getpagesize();
310	+	/* This is in microseconds */
311	+	proc_state_ptr->time_spent = kp_stats[i].ki_runtime / 1000000;
312	+	proc_state_ptr->cpu_percent =
313	+	((double)kp_stats[i].ki_pctcpu / FSCALE) * 100.0;
314	+	proc_state_ptr->nice = kp_stats[i].ki_nice;
315	+	#else
316	+	proc_state_ptr->proc_size =
317	+	kp_stats[i].kp_eproc.e_vm.vm_map.size;
318	+	/* This is in pages */
319	+	proc_state_ptr->proc_resident =
320	+	kp_stats[i].kp_eproc.e_vm.vm_rssize * getpagesize();
321	+	/* This is in microseconds */
322	+	#if defined(NETBSD) || defined(OPENBSD)
323	+	proc_state_ptr->time_spent =
324	+	kp_stats[i].kp_proc.p_rtime.tv_sec;
325	+	#elif defined(DFBSD)
326	+	proc_state_ptr->time_spent =
327	+	( kp_stats[i].kp_thread.td_uticks +
328	+	kp_stats[i].kp_thread.td_sticks +
329	+	kp_stats[i].kp_thread.td_iticks ) / 1000000;
330	+	#else
331	+	proc_state_ptr->time_spent =
332	+	kp_stats[i].kp_proc.p_runtime / 1000000;
333	+	#endif
334	+	proc_state_ptr->cpu_percent =
335	+	((double)kp_stats[i].kp_proc.p_pctcpu / FSCALE) * 100.0;
336	+	proc_state_ptr->nice = kp_stats[i].kp_proc.p_nice;
337	+	#endif
338	+
339	+	#ifdef FREEBSD5
340	+	switch (kp_stats[i].ki_stat) {
341	+	#else
342	+	switch (kp_stats[i].kp_proc.p_stat) {
343	+	#endif
344	+	case SIDL:
345	+	case SRUN:
346	+	#ifdef SONPROC
347	+	case SONPROC: /* NetBSD */
348	+	#endif
349	+	proc_state_ptr->state = RUNNING;
350	+	break;
351	+	case SSLEEP:
352	+	#ifdef SWAIT
353	+	case SWAIT: /* FreeBSD 5 */
354	+	#endif
355	+	#ifdef SLOCK
356	+	case SLOCK: /* FreeBSD 5 */
357	+	#endif
358	+	proc_state_ptr->state = SLEEPING;
359	+	break;
360	+	case SSTOP:
361	+	proc_state_ptr->state = STOPPED;
362	+	break;
363	+	case SZOMB:
364	+	#ifdef SDEAD
365	+	case SDEAD: /* OpenBSD & NetBSD */
366	+	#endif
367	+	proc_state_ptr->state = ZOMBIE;
368	+	break;
369	+	default:
370	+	proc_state_ptr->state = UNKNOWN;
371	+	break;
372	+	}
373	+	proc_state_size++;
374	+	}
375	+
376	+	free(kp_stats);
377	+	#endif
378	+
379	+	*ps = proc_state;
380		return proc_state_size;
381		}
382
383	+	process_stat_t *get_process_stats() {
384	+	static process_stat_t process_stat;
385	+	proc_state_t *ps;
386	+	int ps_size, x;
387	+
388	+	process_stat.sleeping = 0;
389	+	process_stat.running = 0;
390	+	process_stat.zombie = 0;
391	+	process_stat.stopped = 0;
392	+	process_stat.total = 0;
393	+
394	+	ps_size = get_proc_snapshot(&ps);
395	+
396	+	if(ps_size == NULL) {
397	+	return NULL;
398	+	}
399	+
400	+	for(x = 0; x < ps_size; x++) {
401	+	switch (ps->state) {
402	+	/* currently no mapping for UNKNOWN in process_stat_t */
403	+	case RUNNING:
404	+	process_stat.running++;
405	+	break;
406	+	case SLEEPING:
407	+	process_stat.sleeping++;
408	+	break;
409	+	case STOPPED:
410	+	process_stat.stopped++;
411	+	break;
412	+	case ZOMBIE:
413	+	process_stat.zombie++;
414	+	break;
415	+	}
416	+	ps++;
417	+	}
418	+
419	+	process_stat.total = ps_size;
420	+
421	+	return &process_stat;
422	+	}

Diff Legend

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