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;

        total=(total/1024)/1024;
        used=(used/1024)/1024;
        free=(free/1024)/1024;
        cache=(cache/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

        /* casting these from a long long to a long is a KLUDGE! please fix it :) */    
        if((xml_memory_stats=strf("<memory><total>%ld</total><free>%ld</free><used>%ld</used><cache>%ld</cache></memory>", (long)total, (long)free, (long)used, (long)cache)) == NULL){
                errf("strf failed (%m)");
                return NULL;
        }
        return xml_memory_stats;
}
Revision:	1.9
Committed:	Tue May 28 14:40:41 2002 UTC (23 years, 5 months ago) by tdb
Content type:	text/plain
Branch:	MAIN
Changes since 1.8:	+4 -4 lines
Log Message:	A few bug fixes. Firstly on linux the cache value being sent was actually the same as the free value ;) Secondly, on linux used was not including the cache value. To clarify, total=used+free. cache is merely and indication of how much of the used is being used for cache. The final bug fix was in the XML generation. The values were all "long long", but were being treated as "long" in the strf. Unfortunately strf doesn't seem to know about "long long" values, so we just cast them to long's ;) It's hacky, but works for now.
#	User	Rev	Content
1	tdb	1.5	/*
2			* i-scream central monitoring system
3	tdb	1.8	* http://www.i-scream.org.uk
4	tdb	1.5	* 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	pajs	1.1	#include <stdio.h>
22	pajs	1.2	#ifdef LINUX
23	pajs	1.1	#include <string.h>
24	pajs	1.2	#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	tdb	1.3	#include "ukcprog.h"
35	pajs	1.1
36			char *get_memory_stats(){
37	pajs	1.6	long long total=0;
38			long long free=0;
39			long long used=0;
40			long long cache=0;
41	pajs	1.1	char *xml_memory_stats;
42			#ifdef LINUX
43			FILE *f;
44			char *line;
45	pajs	1.7	long long buffers;
46	pajs	1.1	#endif
47	pajs	1.2	#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	pajs	1.1
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	pajs	1.7	if((sscanf(line, "%s %lld %lld %lld %s %lld %lld", &total, &used, &free, &buffers, &cache)) != 5){
72	pajs	1.1	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	tdb	1.9	used=total-free;
85	pajs	1.1
86			total=(total/1024)/1024;
87			used=(used/1024)/1024;
88			free=(free/1024)/1024;
89	tdb	1.9	cache=(cache/1024)/1024;
90	pajs	1.1	#endif
91	pajs	1.2	#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	pajs	1.4	if((kstat_close(kc)) != 0){
119			errf("Failed to close kstat control structure (%m)");
120			return NULL;
121			}
122	pajs	1.2	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	tdb	1.9	/* casting these from a long long to a long is a KLUDGE! please fix it :) */
178			if((xml_memory_stats=strf("<memory><total>%ld</total><free>%ld</free><used>%ld</used><cache>%ld</cache></memory>", (long)total, (long)free, (long)used, (long)cache)) == NULL){
179	pajs	1.1	errf("strf failed (%m)");
180			return NULL;
181			}
182	pajs	1.2	return xml_memory_stats;
183	pajs	1.1	}