ViewVC Help
View File | Revision Log | Show Annotations | Revision Graph | Root Listing
root/i-scream/projects/libstatgrab/src/libstatgrab/cpu_stats.c
Revision: 1.28
Committed: Tue Jun 17 15:52:52 2008 UTC (15 years, 11 months ago) by tdb
Content type: text/plain
Branch: MAIN
CVS Tags: LIBSTATGRAB_0_17
Changes since 1.27: +14 -5 lines
Log Message:
Linux version 2.5.41 or later provides iowait data in /proc/stat. Use
it if it's available.

File Contents

# User Rev Content
1 tdb 1.16 /*
2 tdb 1.21 * i-scream libstatgrab
3 tdb 1.10 * http://www.i-scream.org
4 tdb 1.16 * Copyright (C) 2000-2004 i-scream
5 pajs 1.1 *
6 tdb 1.16 * 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 pajs 1.1 *
11 tdb 1.16 * This library is distributed in the hope that it will be useful,
12 pajs 1.1 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 tdb 1.16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14     * Lesser General Public License for more details.
15 pajs 1.1 *
16 tdb 1.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 02111-1307 USA
19 tdb 1.17 *
20 tdb 1.28 * $Id: cpu_stats.c,v 1.27 2006/10/09 13:52:06 tdb Exp $
21 pajs 1.1 */
22    
23     #ifdef HAVE_CONFIG_H
24     #include "config.h"
25     #endif
26    
27     #include <time.h>
28     #include "statgrab.h"
29 tdb 1.22 #include "tools.h"
30 pajs 1.1 #ifdef SOLARIS
31     #include <kstat.h>
32     #include <sys/sysinfo.h>
33 pajs 1.6 #include <string.h>
34 pajs 1.1 #endif
35 ats 1.14 #if defined(LINUX) || defined(CYGWIN)
36 pajs 1.7 #include <stdio.h>
37     #endif
38 tdb 1.19 #if defined(FREEBSD) || defined(DFBSD)
39 pajs 1.9 #include <sys/sysctl.h>
40     #include <sys/dkstat.h>
41     #endif
42 ats 1.13 #ifdef NETBSD
43     #include <sys/types.h>
44     #include <sys/param.h>
45     #include <sys/sysctl.h>
46     #include <sys/sched.h>
47     #endif
48 tdb 1.18 #ifdef OPENBSD
49     #include <sys/param.h>
50     #include <sys/sysctl.h>
51     #include <sys/dkstat.h>
52     #endif
53 tdb 1.25 #ifdef HPUX
54     #include <sys/param.h>
55     #include <sys/pstat.h>
56     #include <sys/dk.h>
57     #endif
58 tdb 1.26 #ifdef WIN32
59     #include <pdh.h>
60     #include "win32.h"
61     #endif
62 pajs 1.1
63 ats 1.20 static sg_cpu_stats cpu_now;
64 pajs 1.1 static int cpu_now_uninit=1;
65    
66 ats 1.20 sg_cpu_stats *sg_get_cpu_stats(){
67 pajs 1.1
68 tdb 1.25 #ifdef HPUX
69     struct pst_dynamic pstat_dynamic;
70     int i;
71     #endif
72 pajs 1.1 #ifdef SOLARIS
73 tdb 1.22 kstat_ctl_t *kc;
74     kstat_t *ksp;
75 pajs 1.7 cpu_stat_t cs;
76     #endif
77 tdb 1.15 #if defined(LINUX) || defined(CYGWIN)
78 pajs 1.7 FILE *f;
79 tdb 1.28 int proc_stat_cpu;
80 pajs 1.7 #endif
81 ats 1.13 #ifdef ALLBSD
82 tdb 1.19 #if defined(NETBSD) || defined(OPENBSD)
83 ats 1.13 int mib[2];
84     #endif
85     #ifdef NETBSD
86     u_int64_t cp_time[CPUSTATES];
87     #else
88 pajs 1.9 long cp_time[CPUSTATES];
89 ats 1.13 #endif
90 pajs 1.9 size_t size;
91     #endif
92 tdb 1.22
93 pajs 1.1 cpu_now.user=0;
94 pajs 1.9 /* Not stored in linux or freebsd */
95 pajs 1.1 cpu_now.iowait=0;
96     cpu_now.kernel=0;
97     cpu_now.idle=0;
98 tdb 1.26 /* Not stored in linux, freebsd, hpux or windows */
99 pajs 1.1 cpu_now.swap=0;
100     cpu_now.total=0;
101 tdb 1.26 /* Not stored in solaris or windows */
102 pajs 1.1 cpu_now.nice=0;
103    
104 tdb 1.25 #ifdef HPUX
105     if (pstat_getdynamic(&pstat_dynamic, sizeof(pstat_dynamic), 1, 0) == -1) {
106     sg_set_error_with_errno(SG_ERROR_PSTAT, "pstat_dynamic");
107     return NULL;
108     }
109     cpu_now.user = pstat_dynamic.psd_cpu_time[CP_USER];
110     cpu_now.iowait = pstat_dynamic.psd_cpu_time[CP_WAIT];
111     cpu_now.kernel = pstat_dynamic.psd_cpu_time[CP_SSYS] + pstat_dynamic.psd_cpu_time[CP_SYS];
112     cpu_now.idle = pstat_dynamic.psd_cpu_time[CP_IDLE];
113     cpu_now.nice = pstat_dynamic.psd_cpu_time[CP_NICE];
114     for (i = 0; i < PST_MAX_CPUSTATES; i++) {
115     cpu_now.total += pstat_dynamic.psd_cpu_time[i];
116     }
117     #endif
118 pajs 1.7 #ifdef SOLARIS
119 tdb 1.22 if ((kc = kstat_open()) == NULL) {
120     sg_set_error(SG_ERROR_KSTAT_OPEN, NULL);
121     return NULL;
122     }
123     for (ksp = kc->kc_chain; ksp!=NULL; ksp = ksp->ks_next) {
124     if ((strcmp(ksp->ks_module, "cpu_stat")) != 0) continue;
125     if (kstat_read(kc, ksp, &cs) == -1) {
126     continue;
127     }
128 pajs 1.5 cpu_now.user+=(long long)cs.cpu_sysinfo.cpu[CPU_USER];
129     cpu_now.kernel+=(long long)cs.cpu_sysinfo.cpu[CPU_KERNEL];
130     cpu_now.idle+=(long long)cs.cpu_sysinfo.cpu[CPU_IDLE];
131 tdb 1.27 cpu_now.iowait+=(long long)cs.cpu_sysinfo.wait[W_IO]+(long long)cs.cpu_sysinfo.wait[W_PIO];
132     cpu_now.swap+=(long long)cs.cpu_sysinfo.wait[W_SWAP];
133 pajs 1.1 }
134    
135     cpu_now.total=cpu_now.user+cpu_now.iowait+cpu_now.kernel+cpu_now.idle+cpu_now.swap;
136 pajs 1.7
137 tdb 1.22 kstat_close(kc);
138 pajs 1.7 #endif
139 ats 1.14 #if defined(LINUX) || defined(CYGWIN)
140 pajs 1.7 if ((f=fopen("/proc/stat", "r" ))==NULL) {
141 ats 1.24 sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/stat");
142 pajs 1.7 return NULL;
143     }
144 tdb 1.28
145 pajs 1.7 /* The very first line should be cpu */
146 tdb 1.28 proc_stat_cpu = fscanf(f, "cpu %lld %lld %lld %lld %lld", \
147 pajs 1.7 &cpu_now.user, \
148     &cpu_now.nice, \
149     &cpu_now.kernel, \
150 tdb 1.28 &cpu_now.idle, \
151     &cpu_now.iowait);
152    
153     fclose(f);
154    
155     if (proc_stat_cpu < 4 || proc_stat_cpu > 5) {
156 tdb 1.22 sg_set_error(SG_ERROR_PARSE, "cpu");
157 pajs 1.7 return NULL;
158     }
159    
160 tdb 1.28 /* older linux doesn't give iowait */
161     if (proc_stat_cpu == 4) {
162     cpu_now.iowait = 0;
163     }
164 pajs 1.1
165 pajs 1.7 cpu_now.total=cpu_now.user+cpu_now.nice+cpu_now.kernel+cpu_now.idle;
166 pajs 1.9 #endif
167 ats 1.13 #ifdef ALLBSD
168 tdb 1.19 #if defined(FREEBSD) || defined(DFBSD)
169 ats 1.12 size = sizeof cp_time;
170 ats 1.11 if (sysctlbyname("kern.cp_time", &cp_time, &size, NULL, 0) < 0){
171 ats 1.24 sg_set_error_with_errno(SG_ERROR_SYSCTLBYNAME, "kern.cp_time");
172 pajs 1.9 return NULL;
173     }
174 ats 1.13 #else
175     mib[0] = CTL_KERN;
176 tdb 1.18 #ifdef NETBSD
177 ats 1.13 mib[1] = KERN_CP_TIME;
178 tdb 1.18 #else
179     mib[1] = KERN_CPTIME;
180     #endif
181 ats 1.13 size = sizeof cp_time;
182     if (sysctl(mib, 2, &cp_time, &size, NULL, 0) < 0) {
183 tdb 1.22 #ifdef NETBSD
184 ats 1.24 sg_set_error_with_errno(SG_ERROR_SYSCTL,
185     "CTL_KERN.KERN_CP_TIME");
186 tdb 1.22 #else
187 ats 1.24 sg_set_error_with_errno(SG_ERROR_SYSCTL,
188     "CTL_KERN.KERN_CPTIME");
189 tdb 1.22 #endif
190 ats 1.13 return NULL;
191     }
192     #endif
193 pajs 1.9
194     cpu_now.user=cp_time[CP_USER];
195     cpu_now.nice=cp_time[CP_NICE];
196     cpu_now.kernel=cp_time[CP_SYS];
197     cpu_now.idle=cp_time[CP_IDLE];
198    
199     cpu_now.total=cpu_now.user+cpu_now.nice+cpu_now.kernel+cpu_now.idle;
200    
201 pajs 1.1 #endif
202 tdb 1.26 #ifdef WIN32
203     sg_set_error(SG_ERROR_UNSUPPORTED, "Win32");
204     return NULL;
205     #endif
206 pajs 1.1
207     cpu_now.systime=time(NULL);
208 pajs 1.7 cpu_now_uninit=0;
209    
210 pajs 1.1
211     return &cpu_now;
212     }
213    
214 ats 1.20 sg_cpu_stats *sg_get_cpu_stats_diff(){
215     static sg_cpu_stats cpu_diff;
216     sg_cpu_stats cpu_then, *cpu_tmp;
217 pajs 1.1
218     if (cpu_now_uninit){
219 ats 1.20 if((cpu_tmp=sg_get_cpu_stats())==NULL){
220 tdb 1.22 /* Should sg_get_cpu_stats fail */
221 pajs 1.1 return NULL;
222     }
223     return cpu_tmp;
224     }
225    
226    
227 tdb 1.22 cpu_then.user=cpu_now.user;
228     cpu_then.kernel=cpu_now.kernel;
229     cpu_then.idle=cpu_now.idle;
230     cpu_then.iowait=cpu_now.iowait;
231     cpu_then.swap=cpu_now.swap;
232     cpu_then.nice=cpu_now.nice;
233     cpu_then.total=cpu_now.total;
234     cpu_then.systime=cpu_now.systime;
235 pajs 1.1
236 ats 1.20 if((cpu_tmp=sg_get_cpu_stats())==NULL){
237 pajs 1.1 return NULL;
238     }
239    
240     cpu_diff.user = cpu_now.user - cpu_then.user;
241     cpu_diff.kernel = cpu_now.kernel - cpu_then.kernel;
242     cpu_diff.idle = cpu_now.idle - cpu_then.idle;
243     cpu_diff.iowait = cpu_now.iowait - cpu_then.iowait;
244     cpu_diff.swap = cpu_now.swap - cpu_then.swap;
245     cpu_diff.nice = cpu_now.nice - cpu_then.nice;
246     cpu_diff.total = cpu_now.total - cpu_then.total;
247     cpu_diff.systime = cpu_now.systime - cpu_then.systime;
248    
249     return &cpu_diff;
250     }
251    
252 ats 1.20 sg_cpu_percents *sg_get_cpu_percents(){
253     static sg_cpu_percents cpu_usage;
254 tdb 1.26 #ifndef WIN32
255 ats 1.20 sg_cpu_stats *cs_ptr;
256 pajs 1.1
257 ats 1.20 cs_ptr=sg_get_cpu_stats_diff();
258 pajs 1.1 if(cs_ptr==NULL){
259     return NULL;
260     }
261    
262     cpu_usage.user = ((float)cs_ptr->user / (float)cs_ptr->total)*100;
263     cpu_usage.kernel = ((float)cs_ptr->kernel / (float)cs_ptr->total)*100;
264     cpu_usage.idle = ((float)cs_ptr->idle / (float)cs_ptr->total)*100;
265     cpu_usage.iowait = ((float)cs_ptr->iowait / (float)cs_ptr->total)*100;
266     cpu_usage.swap = ((float)cs_ptr->swap / (float)cs_ptr->total)*100;
267     cpu_usage.nice = ((float)cs_ptr->nice / (float)cs_ptr->total)*100;
268     cpu_usage.time_taken = cs_ptr->systime;
269 tdb 1.26 #else
270     double result;
271    
272     if(read_counter_double(SG_WIN32_PROC_USER, &result)) {
273     sg_set_error(SG_ERROR_PDHREAD, PDH_USER);
274     return NULL;
275     }
276     cpu_usage.user = (float)result;
277     if(read_counter_double(SG_WIN32_PROC_PRIV, &result)) {
278     sg_set_error(SG_ERROR_PDHREAD, PDH_PRIV);
279     return NULL;
280     }
281     cpu_usage.kernel = (float)result;
282     if(read_counter_double(SG_WIN32_PROC_IDLE, &result)) {
283     sg_set_error(SG_ERROR_PDHREAD, PDH_IDLE);
284     return NULL;
285     }
286     /* win2000 does not have an idle counter, but does have %activity
287     * so convert it to idle */
288     cpu_usage.idle = 100 - (float)result;
289     if(read_counter_double(SG_WIN32_PROC_INT, &result)) {
290     sg_set_error(SG_ERROR_PDHREAD, PDH_INTER);
291     return NULL;
292     }
293     cpu_usage.iowait = (float)result;
294     #endif
295 pajs 1.1
296     return &cpu_usage;
297     }
298