ihost/libstatgrab/memory_stats.c

/* 
 * i-scream central monitoring system
 * http://www.i-scream.org.uk
 * Copyright (C) 2000-2002 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.
 */

#include <stdio.h>
#ifdef LINUX
#include <string.h>
#endif
#ifdef SOLARIS
#include <unistd.h>
#include <kstat.h>
#endif
#ifdef FREEBSD
#include <sys/types.h>
#include <sys/sysctl.h>
#include <unistd.h>
#endif
#include "ukcprog.h"

char *get_memory_stats(){
        long long total=0;
        long long free=0;
        long long used=0;
        long long cache=0;
        char *xml_memory_stats;
#ifdef LINUX
        FILE *f;
        char *line;
        long long buffers;
#endif
#ifdef SOLARIS
        kstat_ctl_t *kc;
        kstat_t *ksp;
        kstat_named_t *kn;
        u_long totalmem;
        int pagesize;
#endif
#ifdef FREEBSD
#define TOTAL_MEM_NAME "hw.physmem"
#define FREE_MEM_NAME "vm.stats.vm.v_free_count"
#define INACTIVE_MEM_NAME "vm.stats.vm.v_inactive_count"
#define CACHE_MEM_NAME "vm.stats.vm.v_cache_count"
        int pagesize;
        long inactive;
        size_t size;
#endif

#ifdef LINUX
        if ((f=fopen("/proc/meminfo", "r" ))==NULL) {
                errf("Failed to open memory stats (%m)");
                return NULL;
        }
        while((line=fpgetline(f)) != NULL){
                if (((strncmp(line,"Mem:",4)) == 0)) {
                        if((sscanf(line, "%*s %lld %lld %lld %*s %lld %lld", &total, &used, &free, &buffers, &cache)) != 5){
                                errf("Failed to read memory details (%m)");
                                return NULL;
                        }
                        break;
                }
        }
        if ((fclose(f)) != 0) {
                errf("Failed to close file (%m)");
                return NULL;
        }

        cache=cache+buffers;
        used=total-(free+cache);

        total=(total/1024)/1024;
        used=(used/1024)/1024;
        free=(free/1024)/1024;
        cache=(free/1024)/1024;
#endif
#ifdef SOLARIS
        if((pagesize=sysconf(_SC_PAGESIZE)) == -1){
                errf("Failed to get pagesize (%m)");
                return NULL;    
        }

        if((totalmem=sysconf(_SC_PHYS_PAGES)) == -1){
                errf("Failed to get total memory");
                return NULL;
        }

        if ((kc = kstat_open()) == NULL) {
                errf("kstat_open failure (%m)");
                return NULL;
        }
        if((ksp=kstat_lookup(kc, "unix", 0, "system_pages")) == NULL){
                errf("kstat lookup for \"system_pages\" (%m)");
                return NULL;
        }
        if (kstat_read(kc, ksp, 0) == -1) {
                errf("Failed to read kstat details (%m)");
                return NULL;
        }
        if((kn=kstat_data_lookup(ksp, "freemem")) == NULL){
                errf("Failed to get free memory (%m)");
                return NULL;
        }
        if((kstat_close(kc)) != 0){
                errf("Failed to close kstat control structure (%m)");
                return NULL;
        }       
        total=((totalmem*pagesize)/1024)/1024;
        free=(((kn->value.ul)*pagesize)/1024)/1024;
        used=total-free;
#endif
#ifdef FREEBSD
        if ((pagesize=getpagesize()) == -1){
                errf("Failed to get page size (%m)");
                return NULL;
        }

        if (sysctlbyname(TOTAL_MEM_NAME, NULL, &size, NULL, NULL) < 0){
                errf("sysctlbyname failed to get total memory (%m)");
                return NULL;
        }
        if (sysctlbyname(TOTAL_MEM_NAME, &total, &size, NULL, NULL) < 0){
                errf("Failed to get total memory stats (%m)");
                return NULL;
        }

        if (sysctlbyname(FREE_MEM_NAME, NULL, &size, NULL, NULL) < 0){
                errf("sysctlbyname failed to get free memory (%m)");
                return NULL;
        }
        if (sysctlbyname(FREE_MEM_NAME, &free, &size, NULL, NULL) < 0){
                errf("Failed to get free memory stats (%m)");
                return NULL;
        }

        if (sysctlbyname(INACTIVE_MEM_NAME, NULL, &size, NULL, NULL) < 0){
                errf("sysctlbyname failed to get inactive memory (%m)");
                return NULL;
        }
        if (sysctlbyname(INACTIVE_MEM_NAME, &inactive, &size, NULL, NULL) < 0){
                errf("Failed to get inactive memory stats (%m)");
                return NULL;
        }

        if (sysctlbyname(CACHE_MEM_NAME, NULL, &size, NULL, NULL) < 0){
                errf("sysctlbyname failed to get cache memory (%m)");
                return NULL;
        }
        if (sysctlbyname(CACHE_MEM_NAME, &cache, &size, NULL, NULL) < 0){
                errf("Failed to get cache memory stats (%m)");
                return NULL;
        }

        total/=(1024*1024);
        free=((free*pagesize)/(1024*1024))+((inactive*pagesize)/(1024*1024));
        cache=((cache*pagesize)/(1024*1024));
        
        /* used should also be cache+active+wired but its less effort this way :) */
        used=total-free;
          
#endif

        
        if((xml_memory_stats=strf("<memory><total>%ld</total><free>%ld</free><used>%ld</used><cache>%ld</cache></memory>", total, free, used, cache)) == NULL){
                errf("strf failed (%m)");
                return NULL;
        }
        return xml_memory_stats;
}
Revision:	1.8
Committed:	Tue May 21 16:47:12 2002 UTC (23 years, 7 months ago) by tdb
Content type:	text/plain
Branch:	MAIN
Changes since 1.7:	+1 -0 lines
Log Message:	Added URL to GPL headers.
#	Content
1	/*
2	* i-scream central monitoring system
3	* http://www.i-scream.org.uk
4	* Copyright (C) 2000-2002 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.
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	#include <stdio.h>
22	#ifdef LINUX
23	#include <string.h>
24	#endif
25	#ifdef SOLARIS
26	#include <unistd.h>
27	#include <kstat.h>
28	#endif
29	#ifdef FREEBSD
30	#include <sys/types.h>
31	#include <sys/sysctl.h>
32	#include <unistd.h>
33	#endif
34	#include "ukcprog.h"
35
36	char *get_memory_stats(){
37	long long total=0;
38	long long free=0;
39	long long used=0;
40	long long cache=0;
41	char *xml_memory_stats;
42	#ifdef LINUX
43	FILE *f;
44	char *line;
45	long long buffers;
46	#endif
47	#ifdef SOLARIS
48	kstat_ctl_t *kc;
49	kstat_t *ksp;
50	kstat_named_t *kn;
51	u_long totalmem;
52	int pagesize;
53	#endif
54	#ifdef FREEBSD
55	#define TOTAL_MEM_NAME "hw.physmem"
56	#define FREE_MEM_NAME "vm.stats.vm.v_free_count"
57	#define INACTIVE_MEM_NAME "vm.stats.vm.v_inactive_count"
58	#define CACHE_MEM_NAME "vm.stats.vm.v_cache_count"
59	int pagesize;
60	long inactive;
61	size_t size;
62	#endif
63
64	#ifdef LINUX
65	if ((f=fopen("/proc/meminfo", "r" ))==NULL) {
66	errf("Failed to open memory stats (%m)");
67	return NULL;
68	}
69	while((line=fpgetline(f)) != NULL){
70	if (((strncmp(line,"Mem:",4)) == 0)) {
71	if((sscanf(line, "%s %lld %lld %lld %s %lld %lld", &total, &used, &free, &buffers, &cache)) != 5){
72	errf("Failed to read memory details (%m)");
73	return NULL;
74	}
75	break;
76	}
77	}
78	if ((fclose(f)) != 0) {
79	errf("Failed to close file (%m)");
80	return NULL;
81	}
82
83	cache=cache+buffers;
84	used=total-(free+cache);
85
86	total=(total/1024)/1024;
87	used=(used/1024)/1024;
88	free=(free/1024)/1024;
89	cache=(free/1024)/1024;
90	#endif
91	#ifdef SOLARIS
92	if((pagesize=sysconf(_SC_PAGESIZE)) == -1){
93	errf("Failed to get pagesize (%m)");
94	return NULL;
95	}
96
97	if((totalmem=sysconf(_SC_PHYS_PAGES)) == -1){
98	errf("Failed to get total memory");
99	return NULL;
100	}
101
102	if ((kc = kstat_open()) == NULL) {
103	errf("kstat_open failure (%m)");
104	return NULL;
105	}
106	if((ksp=kstat_lookup(kc, "unix", 0, "system_pages")) == NULL){
107	errf("kstat lookup for \"system_pages\" (%m)");
108	return NULL;
109	}
110	if (kstat_read(kc, ksp, 0) == -1) {
111	errf("Failed to read kstat details (%m)");
112	return NULL;
113	}
114	if((kn=kstat_data_lookup(ksp, "freemem")) == NULL){
115	errf("Failed to get free memory (%m)");
116	return NULL;
117	}
118	if((kstat_close(kc)) != 0){
119	errf("Failed to close kstat control structure (%m)");
120	return NULL;
121	}
122	total=((totalmem*pagesize)/1024)/1024;
123	free=(((kn->value.ul)*pagesize)/1024)/1024;
124	used=total-free;
125	#endif
126	#ifdef FREEBSD
127	if ((pagesize=getpagesize()) == -1){
128	errf("Failed to get page size (%m)");
129	return NULL;
130	}
131
132	if (sysctlbyname(TOTAL_MEM_NAME, NULL, &size, NULL, NULL) < 0){
133	errf("sysctlbyname failed to get total memory (%m)");
134	return NULL;
135	}
136	if (sysctlbyname(TOTAL_MEM_NAME, &total, &size, NULL, NULL) < 0){
137	errf("Failed to get total memory stats (%m)");
138	return NULL;
139	}
140
141	if (sysctlbyname(FREE_MEM_NAME, NULL, &size, NULL, NULL) < 0){
142	errf("sysctlbyname failed to get free memory (%m)");
143	return NULL;
144	}
145	if (sysctlbyname(FREE_MEM_NAME, &free, &size, NULL, NULL) < 0){
146	errf("Failed to get free memory stats (%m)");
147	return NULL;
148	}
149
150	if (sysctlbyname(INACTIVE_MEM_NAME, NULL, &size, NULL, NULL) < 0){
151	errf("sysctlbyname failed to get inactive memory (%m)");
152	return NULL;
153	}
154	if (sysctlbyname(INACTIVE_MEM_NAME, &inactive, &size, NULL, NULL) < 0){
155	errf("Failed to get inactive memory stats (%m)");
156	return NULL;
157	}
158
159	if (sysctlbyname(CACHE_MEM_NAME, NULL, &size, NULL, NULL) < 0){
160	errf("sysctlbyname failed to get cache memory (%m)");
161	return NULL;
162	}
163	if (sysctlbyname(CACHE_MEM_NAME, &cache, &size, NULL, NULL) < 0){
164	errf("Failed to get cache memory stats (%m)");
165	return NULL;
166	}
167
168	total/=(1024*1024);
169	free=((freepagesize)/(10241024))+((inactivepagesize)/(10241024));
170	cache=((cachepagesize)/(10241024));
171
172	/* used should also be cache+active+wired but its less effort this way :) */
173	used=total-free;
174
175	#endif
176
177
178	if((xml_memory_stats=strf("<memory><total>%ld</total><free>%ld</free><used>%ld</used><cache>%ld</cache></memory>", total, free, used, cache)) == NULL){
179	errf("strf failed (%m)");
180	return NULL;
181	}
182	return xml_memory_stats;
183	}