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

Diff Legend

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