ihost/libstatgrab/memory_stats.c

#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 total=0;
        long free=0;
        long used=0;
        long cache=0;
        char *xml_memory_stats;
#ifdef LINUX
        FILE *f;
        char *line;
        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 %ld %ld %ld %*s %ld %ld", &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;
        }
        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;
}

int main(){
        printf("%s\n", get_memory_stats());
        exit(0);
}
Revision:	1.2
Committed:	Thu May 16 10:34:23 2002 UTC (22 years, 6 months ago) by pajs
Content type:	text/plain
Branch:	MAIN
Changes since 1.1:	+111 -2 lines
Log Message:	Memory stats for all OS's Solaris doesn't report cache as i couldn't find where it was easily. Linux and freebsd, im not sure how sane the results are wrt used and free because of the cache/buffers situation. A better implimentation of what we are planning to report coming soon to a commit near you ;) (must be 15 or over)
#	User	Rev	Content
1	pajs	1.1	#include <stdio.h>
2	pajs	1.2	#ifdef LINUX
3	pajs	1.1	#include <string.h>
4	pajs	1.2	#endif
5			#ifdef SOLARIS
6			#include <unistd.h>
7			#include <kstat.h>
8			#endif
9			#ifdef FREEBSD
10			#include <sys/types.h>
11			#include <sys/sysctl.h>
12			#include <unistd.h>
13			#endif
14			#include <ukcprog.h>
15	pajs	1.1
16			char *get_memory_stats(){
17			long total=0;
18			long free=0;
19			long used=0;
20			long cache=0;
21			char *xml_memory_stats;
22			#ifdef LINUX
23			FILE *f;
24			char *line;
25			long buffers;
26			#endif
27	pajs	1.2	#ifdef SOLARIS
28			kstat_ctl_t *kc;
29			kstat_t *ksp;
30			kstat_named_t *kn;
31			u_long totalmem;
32			int pagesize;
33			#endif
34			#ifdef FREEBSD
35			#define TOTAL_MEM_NAME "hw.physmem"
36			#define FREE_MEM_NAME "vm.stats.vm.v_free_count"
37			#define INACTIVE_MEM_NAME "vm.stats.vm.v_inactive_count"
38			#define CACHE_MEM_NAME "vm.stats.vm.v_cache_count"
39			int pagesize;
40			long inactive;
41			size_t size;
42			#endif
43	pajs	1.1
44			#ifdef LINUX
45			if ((f=fopen("/proc/meminfo", "r" ))==NULL) {
46			errf("Failed to open memory stats (%m)");
47			return NULL;
48			}
49			while((line=fpgetline(f)) != NULL){
50			if (((strncmp(line,"Mem:",4)) == 0)) {
51			if((sscanf(line, "%s %ld %ld %ld %s %ld %ld", &total, &used, &free, &buffers, &cache)) != 5){
52			errf("Failed to read memory details (%m)");
53			return NULL;
54			}
55			break;
56			}
57			}
58			if ((fclose(f)) != 0) {
59			errf("Failed to close file (%m)");
60			return NULL;
61			}
62
63			cache=cache+buffers;
64			used=total-(free+cache);
65
66			total=(total/1024)/1024;
67			used=(used/1024)/1024;
68			free=(free/1024)/1024;
69			cache=(free/1024)/1024;
70			#endif
71	pajs	1.2	#ifdef SOLARIS
72			if((pagesize=sysconf(_SC_PAGESIZE)) == -1){
73			errf("Failed to get pagesize (%m)");
74			return NULL;
75			}
76
77			if((totalmem=sysconf(_SC_PHYS_PAGES)) == -1){
78			errf("Failed to get total memory");
79			return NULL;
80			}
81
82			if ((kc = kstat_open()) == NULL) {
83			errf("kstat_open failure (%m)");
84			return NULL;
85			}
86			if((ksp=kstat_lookup(kc, "unix", 0, "system_pages")) == NULL){
87			errf("kstat lookup for \"system_pages\" (%m)");
88			return NULL;
89			}
90			if (kstat_read(kc, ksp, 0) == -1) {
91			errf("Failed to read kstat details (%m)");
92			return NULL;
93			}
94			if((kn=kstat_data_lookup(ksp, "freemem")) == NULL){
95			errf("Failed to get free memory (%m)");
96			return NULL;
97			}
98			total=((totalmem*pagesize)/1024)/1024;
99			free=(((kn->value.ul)*pagesize)/1024)/1024;
100			used=total-free;
101			#endif
102			#ifdef FREEBSD
103			if ((pagesize=getpagesize()) == -1){
104			errf("Failed to get page size (%m)");
105			return NULL;
106			}
107
108			if (sysctlbyname(TOTAL_MEM_NAME, NULL, &size, NULL, NULL) < 0){
109			errf("sysctlbyname failed to get total memory (%m)");
110			return NULL;
111			}
112			if (sysctlbyname(TOTAL_MEM_NAME, &total, &size, NULL, NULL) < 0){
113			errf("Failed to get total memory stats (%m)");
114			return NULL;
115			}
116
117			if (sysctlbyname(FREE_MEM_NAME, NULL, &size, NULL, NULL) < 0){
118			errf("sysctlbyname failed to get free memory (%m)");
119			return NULL;
120			}
121			if (sysctlbyname(FREE_MEM_NAME, &free, &size, NULL, NULL) < 0){
122			errf("Failed to get free memory stats (%m)");
123			return NULL;
124			}
125
126			if (sysctlbyname(INACTIVE_MEM_NAME, NULL, &size, NULL, NULL) < 0){
127			errf("sysctlbyname failed to get inactive memory (%m)");
128			return NULL;
129			}
130			if (sysctlbyname(INACTIVE_MEM_NAME, &inactive, &size, NULL, NULL) < 0){
131			errf("Failed to get inactive memory stats (%m)");
132			return NULL;
133			}
134
135			if (sysctlbyname(CACHE_MEM_NAME, NULL, &size, NULL, NULL) < 0){
136			errf("sysctlbyname failed to get cache memory (%m)");
137			return NULL;
138			}
139			if (sysctlbyname(CACHE_MEM_NAME, &cache, &size, NULL, NULL) < 0){
140			errf("Failed to get cache memory stats (%m)");
141			return NULL;
142			}
143
144			total/=(1024*1024);
145			free=((freepagesize)/(10241024))+((inactivepagesize)/(10241024));
146			cache=((cachepagesize)/(10241024));
147
148			/* used should also be cache+active+wired but its less effort this way :) */
149			used=total-free;
150
151			#endif
152
153	pajs	1.1
154			if((xml_memory_stats=strf("<memory><total>%ld</total><free>%ld</free><used>%ld</used><cache>%ld</cache></memory>", total, free, used, cache)) == NULL){
155			errf("strf failed (%m)");
156			return NULL;
157			}
158	pajs	1.2	return xml_memory_stats;
159	pajs	1.1	}
160
161			int main(){
162			printf("%s\n", get_memory_stats());
163			exit(0);
164			}