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 (23 years, 7 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)
#	Content
1	#include <stdio.h>
2	#ifdef LINUX
3	#include <string.h>
4	#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
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	#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
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	#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
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	return xml_memory_stats;
159	}
160
161	int main(){
162	printf("%s\n", get_memory_stats());
163	exit(0);
164	}