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.2 by tdb, Fri Feb 28 22:59:35 2003 UTC vs.
Revision 1.47 by tdb, Mon Apr 5 00:21:29 2004 UTC

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

Diff Legend

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