src/libstatgrab/cpu_stats.c

/* 
 * i-scream central monitoring system
 * http://www.i-scream.org
 * Copyright (C) 2000-2003 i-scream
 * 
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <time.h>
#include "statgrab.h"
#ifdef SOLARIS
#include <kstat.h>
#include <sys/sysinfo.h>
#include <string.h>
#endif
#ifdef LINUX
#include <stdio.h>
#endif
#ifdef FREEBSD
#include <sys/sysctl.h>
#include <sys/dkstat.h>
#endif
#ifdef NETBSD
#include <sys/types.h>
#include <sys/param.h>
#include <sys/sysctl.h>
#include <sys/sched.h>
#endif

static cpu_states_t cpu_now;
static int cpu_now_uninit=1;

cpu_states_t *get_cpu_totals(){

#ifdef SOLARIS
        kstat_ctl_t *kc;
        kstat_t *ksp;
        cpu_stat_t cs;
#endif
#ifdef LINUX
        FILE *f;
#endif
#ifdef ALLBSD
#ifndef FREEBSD
        int mib[2];
#endif
#ifdef NETBSD
        u_int64_t cp_time[CPUSTATES];
#else
        long cp_time[CPUSTATES];
#endif
        size_t size;
#endif
        
        cpu_now.user=0;
        /* Not stored in linux or freebsd */
        cpu_now.iowait=0;
        cpu_now.kernel=0;
        cpu_now.idle=0;
        /* Not stored in linux or freebsd */
        cpu_now.swap=0;
        cpu_now.total=0;
        /* Not stored in solaris */
        cpu_now.nice=0;

#ifdef SOLARIS
        if ((kc = kstat_open()) == NULL) {
                return NULL;
        }
        for (ksp = kc->kc_chain; ksp!=NULL; ksp = ksp->ks_next) {
                if ((strcmp(ksp->ks_module, "cpu_stat")) != 0) continue;
                if (kstat_read(kc, ksp, &cs) == -1) {
                        continue;
                }
                cpu_now.user+=(long long)cs.cpu_sysinfo.cpu[CPU_USER];
                cpu_now.iowait+=(long long)cs.cpu_sysinfo.cpu[CPU_WAIT];
                cpu_now.kernel+=(long long)cs.cpu_sysinfo.cpu[CPU_KERNEL];
                cpu_now.idle+=(long long)cs.cpu_sysinfo.cpu[CPU_IDLE];
                cpu_now.swap+=(long long)cs.cpu_sysinfo.cpu[CPU_STATES];
        }

        cpu_now.total=cpu_now.user+cpu_now.iowait+cpu_now.kernel+cpu_now.idle+cpu_now.swap;
        
        kstat_close(kc);
#endif
#ifdef LINUX
        if ((f=fopen("/proc/stat", "r" ))==NULL) {
                return NULL;
        }
        /* The very first line should be cpu */
        if((fscanf(f, "cpu %lld %lld %lld %lld", \
                &cpu_now.user, \
                &cpu_now.nice, \
                &cpu_now.kernel, \
                &cpu_now.idle)) != 4){
                fclose(f);
                return NULL;
        }

        fclose(f);

        cpu_now.total=cpu_now.user+cpu_now.nice+cpu_now.kernel+cpu_now.idle;
#endif
#ifdef ALLBSD
#ifdef FREEBSD
        size = sizeof cp_time;
        if (sysctlbyname("kern.cp_time", &cp_time, &size, NULL, 0) < 0){
                return NULL;
        }
#else
        mib[0] = CTL_KERN;
        mib[1] = KERN_CP_TIME;
        size = sizeof cp_time;
        if (sysctl(mib, 2, &cp_time, &size, NULL, 0) < 0) {
                return NULL;
        }
#endif

        cpu_now.user=cp_time[CP_USER];
        cpu_now.nice=cp_time[CP_NICE];
        cpu_now.kernel=cp_time[CP_SYS];
        cpu_now.idle=cp_time[CP_IDLE];
        
        cpu_now.total=cpu_now.user+cpu_now.nice+cpu_now.kernel+cpu_now.idle;

#endif

        cpu_now.systime=time(NULL);
        cpu_now_uninit=0;


        return &cpu_now;
}

cpu_states_t *get_cpu_diff(){
        static cpu_states_t cpu_diff;
        cpu_states_t cpu_then, *cpu_tmp;

        if (cpu_now_uninit){
                if((cpu_tmp=get_cpu_totals())==NULL){
                /* Should get_cpu_totals fail */
                        return NULL;
                }
                return cpu_tmp;
        }


        cpu_then.user=cpu_now.user;
        cpu_then.kernel=cpu_now.kernel;
        cpu_then.idle=cpu_now.idle;
        cpu_then.iowait=cpu_now.iowait;
        cpu_then.swap=cpu_now.swap;
        cpu_then.nice=cpu_now.nice;
        cpu_then.total=cpu_now.total;
        cpu_then.systime=cpu_now.systime;

        if((cpu_tmp=get_cpu_totals())==NULL){
                return NULL;
        }

        cpu_diff.user = cpu_now.user - cpu_then.user;
        cpu_diff.kernel = cpu_now.kernel - cpu_then.kernel;
        cpu_diff.idle = cpu_now.idle - cpu_then.idle;
        cpu_diff.iowait = cpu_now.iowait - cpu_then.iowait;
        cpu_diff.swap = cpu_now.swap - cpu_then.swap;
        cpu_diff.nice = cpu_now.nice - cpu_then.nice;
        cpu_diff.total = cpu_now.total - cpu_then.total;
        cpu_diff.systime = cpu_now.systime - cpu_then.systime;

        return &cpu_diff;
}

cpu_percent_t *cpu_percent_usage(){
        static cpu_percent_t cpu_usage;
        cpu_states_t *cs_ptr;

        cs_ptr=get_cpu_diff();
        if(cs_ptr==NULL){
                return NULL;
        }

        cpu_usage.user =  ((float)cs_ptr->user / (float)cs_ptr->total)*100;
        cpu_usage.kernel =  ((float)cs_ptr->kernel / (float)cs_ptr->total)*100;
        cpu_usage.idle = ((float)cs_ptr->idle / (float)cs_ptr->total)*100;
        cpu_usage.iowait = ((float)cs_ptr->iowait / (float)cs_ptr->total)*100;
        cpu_usage.swap = ((float)cs_ptr->swap / (float)cs_ptr->total)*100;
        cpu_usage.nice = ((float)cs_ptr->nice / (float)cs_ptr->total)*100;
        cpu_usage.time_taken = cs_ptr->systime;

        return &cpu_usage;

}

Revision:	1.13
Committed:	Sun Oct 19 02:03:02 2003 UTC (20 years, 7 months ago) by ats
Content type:	text/plain
Branch:	MAIN
CVS Tags:	LIBSTATGRAB_0_7
Changes since 1.12:	+23 -1 lines
Log Message:	Initial support for NetBSD. This adds NetBSD support for everything except diskio stats (since they're even more disturbingly complex to get at on NetBSD than the three OSs we already support). Tested against NetBSD 1.6 on i386.
#	User	Rev	Content
1	pajs	1.1	/*
2			* i-scream central monitoring system
3	tdb	1.10	* http://www.i-scream.org
4			* Copyright (C) 2000-2003 i-scream
5	pajs	1.1	*
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.
10			*
11			* This program 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.
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.
19			*/
20
21			#ifdef HAVE_CONFIG_H
22			#include "config.h"
23			#endif
24
25			#include <time.h>
26			#include "statgrab.h"
27			#ifdef SOLARIS
28			#include <kstat.h>
29			#include <sys/sysinfo.h>
30	pajs	1.6	#include <string.h>
31	pajs	1.1	#endif
32	pajs	1.7	#ifdef LINUX
33			#include <stdio.h>
34			#endif
35	pajs	1.9	#ifdef FREEBSD
36			#include <sys/sysctl.h>
37			#include <sys/dkstat.h>
38			#endif
39	ats	1.13	#ifdef NETBSD
40			#include <sys/types.h>
41			#include <sys/param.h>
42			#include <sys/sysctl.h>
43			#include <sys/sched.h>
44			#endif
45	pajs	1.1
46			static cpu_states_t cpu_now;
47			static int cpu_now_uninit=1;
48
49			cpu_states_t *get_cpu_totals(){
50
51			#ifdef SOLARIS
52			kstat_ctl_t *kc;
53			kstat_t *ksp;
54	pajs	1.7	cpu_stat_t cs;
55			#endif
56			#ifdef LINUX
57			FILE *f;
58			#endif
59	ats	1.13	#ifdef ALLBSD
60			#ifndef FREEBSD
61			int mib[2];
62			#endif
63			#ifdef NETBSD
64			u_int64_t cp_time[CPUSTATES];
65			#else
66	pajs	1.9	long cp_time[CPUSTATES];
67	ats	1.13	#endif
68	pajs	1.9	size_t size;
69			#endif
70	pajs	1.7
71	pajs	1.1	cpu_now.user=0;
72	pajs	1.9	/* Not stored in linux or freebsd */
73	pajs	1.1	cpu_now.iowait=0;
74			cpu_now.kernel=0;
75			cpu_now.idle=0;
76	pajs	1.9	/* Not stored in linux or freebsd */
77	pajs	1.1	cpu_now.swap=0;
78			cpu_now.total=0;
79			/* Not stored in solaris */
80			cpu_now.nice=0;
81
82	pajs	1.7	#ifdef SOLARIS
83	pajs	1.1	if ((kc = kstat_open()) == NULL) {
84			return NULL;
85			}
86			for (ksp = kc->kc_chain; ksp!=NULL; ksp = ksp->ks_next) {
87			if ((strcmp(ksp->ks_module, "cpu_stat")) != 0) continue;
88			if (kstat_read(kc, ksp, &cs) == -1) {
89			continue;
90			}
91	pajs	1.5	cpu_now.user+=(long long)cs.cpu_sysinfo.cpu[CPU_USER];
92			cpu_now.iowait+=(long long)cs.cpu_sysinfo.cpu[CPU_WAIT];
93			cpu_now.kernel+=(long long)cs.cpu_sysinfo.cpu[CPU_KERNEL];
94			cpu_now.idle+=(long long)cs.cpu_sysinfo.cpu[CPU_IDLE];
95			cpu_now.swap+=(long long)cs.cpu_sysinfo.cpu[CPU_STATES];
96	pajs	1.1	}
97
98			cpu_now.total=cpu_now.user+cpu_now.iowait+cpu_now.kernel+cpu_now.idle+cpu_now.swap;
99	pajs	1.7
100			kstat_close(kc);
101			#endif
102			#ifdef LINUX
103			if ((f=fopen("/proc/stat", "r" ))==NULL) {
104			return NULL;
105			}
106			/* The very first line should be cpu */
107			if((fscanf(f, "cpu %lld %lld %lld %lld", \
108			&cpu_now.user, \
109			&cpu_now.nice, \
110			&cpu_now.kernel, \
111			&cpu_now.idle)) != 4){
112	pajs	1.8	fclose(f);
113	pajs	1.7	return NULL;
114			}
115
116			fclose(f);
117	pajs	1.1
118	pajs	1.7	cpu_now.total=cpu_now.user+cpu_now.nice+cpu_now.kernel+cpu_now.idle;
119	pajs	1.9	#endif
120	ats	1.13	#ifdef ALLBSD
121	pajs	1.9	#ifdef FREEBSD
122	ats	1.12	size = sizeof cp_time;
123	ats	1.11	if (sysctlbyname("kern.cp_time", &cp_time, &size, NULL, 0) < 0){
124	pajs	1.9	return NULL;
125			}
126	ats	1.13	#else
127			mib[0] = CTL_KERN;
128			mib[1] = KERN_CP_TIME;
129			size = sizeof cp_time;
130			if (sysctl(mib, 2, &cp_time, &size, NULL, 0) < 0) {
131			return NULL;
132			}
133			#endif
134	pajs	1.9
135			cpu_now.user=cp_time[CP_USER];
136			cpu_now.nice=cp_time[CP_NICE];
137			cpu_now.kernel=cp_time[CP_SYS];
138			cpu_now.idle=cp_time[CP_IDLE];
139
140			cpu_now.total=cpu_now.user+cpu_now.nice+cpu_now.kernel+cpu_now.idle;
141
142	pajs	1.1	#endif
143
144			cpu_now.systime=time(NULL);
145	pajs	1.7	cpu_now_uninit=0;
146
147	pajs	1.1
148			return &cpu_now;
149			}
150
151			cpu_states_t *get_cpu_diff(){
152			static cpu_states_t cpu_diff;
153			cpu_states_t cpu_then, *cpu_tmp;
154
155			if (cpu_now_uninit){
156			if((cpu_tmp=get_cpu_totals())==NULL){
157			/* Should get_cpu_totals fail */
158			return NULL;
159			}
160			return cpu_tmp;
161			}
162
163
164			cpu_then.user=cpu_now.user;
165			cpu_then.kernel=cpu_now.kernel;
166			cpu_then.idle=cpu_now.idle;
167			cpu_then.iowait=cpu_now.iowait;
168			cpu_then.swap=cpu_now.swap;
169			cpu_then.nice=cpu_now.nice;
170			cpu_then.total=cpu_now.total;
171			cpu_then.systime=cpu_now.systime;
172
173			if((cpu_tmp=get_cpu_totals())==NULL){
174			return NULL;
175			}
176
177			cpu_diff.user = cpu_now.user - cpu_then.user;
178			cpu_diff.kernel = cpu_now.kernel - cpu_then.kernel;
179			cpu_diff.idle = cpu_now.idle - cpu_then.idle;
180			cpu_diff.iowait = cpu_now.iowait - cpu_then.iowait;
181			cpu_diff.swap = cpu_now.swap - cpu_then.swap;
182			cpu_diff.nice = cpu_now.nice - cpu_then.nice;
183			cpu_diff.total = cpu_now.total - cpu_then.total;
184			cpu_diff.systime = cpu_now.systime - cpu_then.systime;
185
186			return &cpu_diff;
187			}
188
189			cpu_percent_t *cpu_percent_usage(){
190			static cpu_percent_t cpu_usage;
191			cpu_states_t *cs_ptr;
192
193			cs_ptr=get_cpu_diff();
194			if(cs_ptr==NULL){
195			return NULL;
196			}
197
198			cpu_usage.user = ((float)cs_ptr->user / (float)cs_ptr->total)*100;
199			cpu_usage.kernel = ((float)cs_ptr->kernel / (float)cs_ptr->total)*100;
200			cpu_usage.idle = ((float)cs_ptr->idle / (float)cs_ptr->total)*100;
201			cpu_usage.iowait = ((float)cs_ptr->iowait / (float)cs_ptr->total)*100;
202			cpu_usage.swap = ((float)cs_ptr->swap / (float)cs_ptr->total)*100;
203			cpu_usage.nice = ((float)cs_ptr->nice / (float)cs_ptr->total)*100;
204			cpu_usage.time_taken = cs_ptr->systime;
205
206			return &cpu_usage;
207
208			}
209