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.1 by pajs, Tue Feb 18 19:28:30 2003 UTC vs.
Revision 1.20 by tdb, Sat Feb 14 16:12:46 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
25	–	#include <stdio.h>
28		#include "statgrab.h"
29	+	#if defined(SOLARIS) || defined(LINUX)
30	+	#include <stdio.h>
31		#include <stdlib.h>
32	<	#include <unistd.h>
32	>	#include <sys/types.h>
33	>	#include <dirent.h>
34	>	#include <string.h>
35	>	#endif
36
37	+	#include "tools.h"
38		#ifdef SOLARIS
39		#include <procfs.h>
40		#include <limits.h>
33	–	#include <sys/types.h>
34	–	#include <dirent.h>
35	–	#include <string.h>
41		#define PROC_LOCATION "/proc"
42		#define MAX_FILE_LENGTH PATH_MAX
43		#endif
44	+	#ifdef LINUX
45	+	#include <limits.h>
46	+	#define PROC_LOCATION "/proc"
47	+	#define MAX_FILE_LENGTH PATH_MAX
48	+	#endif
49	+	#ifdef ALLBSD
50	+	#include <stdlib.h>
51	+	#include <sys/param.h>
52	+	#include <sys/sysctl.h>
53	+	#include <sys/proc.h>
54	+	#endif
55	+	#ifdef FREEBSD
56	+	#include <sys/user.h>
57	+	#endif
58
59		process_stat_t *get_process_stats(){
60
61		static process_stat_t process_stat;
62
63	<	psinfo_t process_info;
63	>	#if defined(SOLARIS) || defined(LINUX)
64		DIR *proc_dir;
65		struct dirent *dir_entry;
66		char filename[MAX_FILE_LENGTH];
67		FILE *f;
68	+	#endif
69	+	#ifdef LINUX
70	+	char *line_ptr;
71	+	#endif
72	+	#ifdef SOLARIS
73	+	psinfo_t process_info;
74	+	#endif
75	+	#ifdef ALLBSD
76	+	int mib[3];
77	+	size_t size;
78	+	struct kinfo_proc *kp_stats;
79	+	int procs, i;
80	+	#endif
81
82		process_stat.sleeping=0;
83		process_stat.running=0;
84		process_stat.zombie=0;
85		process_stat.stopped=0;
86	<	process_stat.running=0;
86	>	process_stat.total=0;
87
88	+	#if defined(SOLARIS) || defined(LINUX)
89		if((proc_dir=opendir(PROC_LOCATION))==NULL){
90		return NULL;
91		}
92
93		while((dir_entry=readdir(proc_dir))!=NULL){
94		if(atoi(dir_entry->d_name) == 0) continue;
95	+
96	+	#ifdef SOLARIS
97		snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/psinfo", dir_entry->d_name);
98	+	#endif
99	+	#ifdef LINUX
100	+	snprintf(filename, MAX_FILE_LENGTH, "/proc/%s/status", dir_entry->d_name);
101	+	#endif
102
64	–	printf("Filename : %s\n", filename);
103		if((f=fopen(filename, "r"))==NULL){
104		/* Open failed.. Process since vanished, or the path was too long.
105		* Ah well, move onwards to the next one */
106		continue;
107		}
108
109	+	#ifdef SOLARIS
110		fread(&process_info, sizeof(psinfo_t), 1, f);
72	–	fclose(f);
73	–
111		if(process_info.pr_lwp.pr_state==1) process_stat.sleeping++;
112		if(process_info.pr_lwp.pr_state==2) process_stat.running++;
113		if(process_info.pr_lwp.pr_state==3) process_stat.zombie++;
114		if(process_info.pr_lwp.pr_state==4) process_stat.stopped++;
115		if(process_info.pr_lwp.pr_state==6) process_stat.running++;
116	+	#endif
117	+	#ifdef LINUX
118	+	if((line_ptr=f_read_line(f, "State:"))==NULL){
119	+	fclose(f);
120	+	continue;
121	+	}
122	+	if((line_ptr=strchr(line_ptr, '\t'))==NULL){
123	+	fclose(f);
124	+	continue;
125	+	}
126	+	line_ptr++;
127	+	if(line_ptr=='\0'){
128	+	fclose(f);
129	+	continue;
130	+	}
131	+
132	+	if(*line_ptr=='S') process_stat.sleeping++;
133	+	if(*line_ptr=='R') process_stat.running++;
134	+	if(*line_ptr=='Z') process_stat.zombie++;
135	+	if(*line_ptr=='T') process_stat.stopped++;
136	+	if(*line_ptr=='D') process_stat.stopped++;
137	+	#endif
138	+
139	+	fclose(f);
140		}
80	–
141		closedir(proc_dir);
142	+	#endif
143	+	#ifdef ALLBSD
144	+	mib[0] = CTL_KERN;
145	+	mib[1] = KERN_PROC;
146	+	mib[2] = KERN_PROC_ALL;
147	+
148	+	if (sysctl(mib, 3, NULL, &size, NULL, 0) < 0) {
149	+	return NULL;
150	+	}
151	+
152	+	procs = size / sizeof(struct kinfo_proc);
153	+
154	+	kp_stats = malloc(size);
155	+	if (kp_stats == NULL) {
156	+	return NULL;
157	+	}
158	+
159	+	if (sysctl(mib, 3, kp_stats, &size, NULL, 0) < 0) {
160	+	return NULL;
161	+	}
162	+
163	+	for (i = 0; i < procs; i++) {
164	+	#ifdef FREEBSD5
165	+	switch (kp_stats[i].ki_stat) {
166	+	#else
167	+	switch (kp_stats[i].kp_proc.p_stat) {
168	+	#endif
169	+	case SIDL:
170	+	case SRUN:
171	+	#ifdef SONPROC
172	+	case SONPROC: /* NetBSD */
173	+	#endif
174	+	process_stat.running++;
175	+	break;
176	+	case SSLEEP:
177	+	#ifdef SWAIT
178	+	case SWAIT: /* FreeBSD 5 */
179	+	#endif
180	+	#ifdef SLOCK
181	+	case SLOCK: /* FreeBSD 5 */
182	+	#endif
183	+	process_stat.sleeping++;
184	+	break;
185	+	case SSTOP:
186	+	process_stat.stopped++;
187	+	break;
188	+	case SZOMB:
189	+	#ifdef SDEAD
190	+	case SDEAD: /* OpenBSD & NetBSD */
191	+	#endif
192	+	process_stat.zombie++;
193	+	break;
194	+	}
195	+	}
196	+	#endif
197	+
198	+	free(kp_stats);
199	+
200	+	#ifdef CYGWIN
201	+	return NULL;
202	+	#endif
203	+
204	+	process_stat.total=process_stat.sleeping+process_stat.running+process_stat.zombie+process_stat.stopped;
205
206		return &process_stat;
207		}

Diff Legend

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