src/libstatgrab/memory_stats.c

/*
 * i-scream libstatgrab
 * http://www.i-scream.org
 * Copyright (C) 2000-2004 i-scream
 * 
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 * 
 * This library 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
 * Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
 * 02111-1307 USA
 *
 * $Id: memory_stats.c,v 1.31 2004/11/06 14:55:53 ats Exp $
 */

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

#include "statgrab.h"
#include "tools.h"
#ifdef SOLARIS
#include <unistd.h>
#include <kstat.h>
#endif
#if defined(LINUX) || defined(CYGWIN)
#include <stdio.h>
#include <string.h>
#endif
#if defined(FREEBSD) || defined(DFBSD)
#include <sys/types.h>
#include <sys/sysctl.h>
#include <unistd.h>
#endif
#if defined(NETBSD) || defined(OPENBSD)
#include <sys/param.h>
#include <sys/time.h>
#include <uvm/uvm.h>
#endif
#ifdef HPUX
#include <sys/param.h>
#include <sys/pstat.h>
#include <unistd.h>
#endif
#ifdef WIN32
#include <windows.h>
#include "win32.h"
#endif

sg_mem_stats *sg_get_mem_stats(){

        static sg_mem_stats mem_stat;

#ifdef HPUX
        struct pst_static *pstat_static;
        struct pst_dynamic pstat_dynamic;
        long long pagesize;
#endif
#ifdef SOLARIS
        kstat_ctl_t *kc;
        kstat_t *ksp;
        kstat_named_t *kn;
        long totalmem;
        int pagesize;
#endif
#if defined(LINUX) || defined(CYGWIN)
        char *line_ptr;
        unsigned long long value;
        FILE *f;
#endif
#if defined(FREEBSD) || defined(DFBSD)
        int mib[2];
        u_long physmem;
        size_t size;
        u_int free_count;
        u_int cache_count;
        u_int inactive_count;
        int pagesize;
#endif
#if defined(NETBSD) || defined(OPENBSD)
        struct uvmexp *uvm;
#endif
#ifdef WIN32
        MEMORYSTATUSEX memstats;
#endif

#ifdef HPUX
        if((pagesize=sysconf(_SC_PAGESIZE)) == -1){
                sg_set_error_with_errno(SG_ERROR_SYSCONF, "_SC_PAGESIZE");
                return NULL;
        }

        if (pstat_getdynamic(&pstat_dynamic, sizeof(pstat_dynamic), 1, 0) == -1) {
                sg_set_error_with_errno(SG_ERROR_PSTAT, "pstat_dynamic");
                return NULL;
        }
        pstat_static = sg_get_pstat_static();
        if (pstat_static == NULL) {
                return NULL;
        }

        /* FIXME Does this include swap? */
        mem_stat.total = ((long long) pstat_static->physical_memory) * pagesize;
        mem_stat.free = ((long long) pstat_dynamic.psd_free) * pagesize;
        mem_stat.used = mem_stat.total - mem_stat.free;
#endif
#ifdef SOLARIS
        if((pagesize=sysconf(_SC_PAGESIZE)) == -1){
                sg_set_error_with_errno(SG_ERROR_SYSCONF, "_SC_PAGESIZE");
                return NULL;    
        }

        if((totalmem=sysconf(_SC_PHYS_PAGES)) == -1){
                sg_set_error_with_errno(SG_ERROR_SYSCONF, "_SC_PHYS_PAGES");
                return NULL;
        }

        if ((kc = kstat_open()) == NULL) {
                sg_set_error(SG_ERROR_KSTAT_OPEN, NULL);
                return NULL;
        }
        if((ksp=kstat_lookup(kc, "unix", 0, "system_pages")) == NULL){
                sg_set_error(SG_ERROR_KSTAT_LOOKUP, "unix,0,system_pages");
                return NULL;
        }
        if (kstat_read(kc, ksp, 0) == -1) {
                sg_set_error(SG_ERROR_KSTAT_READ, NULL);
                return NULL;
        }
        if((kn=kstat_data_lookup(ksp, "freemem")) == NULL){
                sg_set_error(SG_ERROR_KSTAT_DATA_LOOKUP, "freemem");
                return NULL;
        }
        kstat_close(kc);

        mem_stat.total = (long long)totalmem * (long long)pagesize;
        mem_stat.free = ((long long)kn->value.ul) * (long long)pagesize;
        mem_stat.used = mem_stat.total - mem_stat.free;
#endif

#if defined(LINUX) || defined(CYGWIN)
        if ((f = fopen("/proc/meminfo", "r")) == NULL) {
                sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/meminfo");
                return NULL;
        }

        while ((line_ptr = sg_f_read_line(f, "")) != NULL) {
                if (sscanf(line_ptr, "%*s %llu kB", &value) != 1) {
                        continue;
                }
                value *= 1024;

                if (strncmp(line_ptr, "MemTotal:", 9) == 0) {
                        mem_stat.total = value;
                } else if (strncmp(line_ptr, "MemFree:", 8) == 0) {
                        mem_stat.free = value;
                } else if (strncmp(line_ptr, "Cached:", 7) == 0) {
                        mem_stat.cache = value;
                }
        }

        fclose(f);
        mem_stat.used = mem_stat.total - mem_stat.free;
#endif

#if defined(FREEBSD) || defined(DFBSD)
        /* Returns bytes */
        mib[0] = CTL_HW;
        mib[1] = HW_PHYSMEM;
        size = sizeof physmem;
        if (sysctl(mib, 2, &physmem, &size, NULL, 0) < 0) {
                sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_HW.HW_PHYSMEM");
                return NULL;
        }
        mem_stat.total = physmem;

        /*returns pages*/
        size = sizeof free_count;
        if (sysctlbyname("vm.stats.vm.v_free_count", &free_count, &size, NULL, 0) < 0){
                sg_set_error_with_errno(SG_ERROR_SYSCTLBYNAME,
                                        "vm.stats.vm.v_free_count");
                return NULL;
        }

        size = sizeof inactive_count;
        if (sysctlbyname("vm.stats.vm.v_inactive_count", &inactive_count , &size, NULL, 0) < 0){
                sg_set_error_with_errno(SG_ERROR_SYSCTLBYNAME,
                                        "vm.stats.vm.v_inactive_count");
                return NULL;
        }

        size = sizeof cache_count;
        if (sysctlbyname("vm.stats.vm.v_cache_count", &cache_count, &size, NULL, 0) < 0){
                sg_set_error_with_errno(SG_ERROR_SYSCTLBYNAME,
                                        "vm.stats.vm.v_cache_count");
                return NULL;
        }

        /* Because all the vm.stats returns pages, I need to get the page size.
         * After that I then need to multiple the anything that used vm.stats to
         * get the system statistics by pagesize 
         */
        pagesize = getpagesize();
        mem_stat.cache=cache_count*pagesize;

        /* Of couse nothing is ever that simple :) And I have inactive pages to
         * deal with too. So I'm going to add them to free memory :)
         */
        mem_stat.free=(free_count*pagesize)+(inactive_count*pagesize);
        mem_stat.used=physmem-mem_stat.free;
#endif

#if defined(NETBSD) || defined(OPENBSD)
        if ((uvm = sg_get_uvmexp()) == NULL) {
                return NULL;
        }

        mem_stat.total = uvm->pagesize * uvm->npages;
#ifdef NETBSD
        mem_stat.cache = uvm->pagesize * (uvm->filepages + uvm->execpages);
#else
        /* Can't find cache memory on OpenBSD */
        mem_stat.cache = 0;
#endif
        mem_stat.free = uvm->pagesize * (uvm->free + uvm->inactive);
        mem_stat.used = mem_stat.total - mem_stat.free;
#endif

#ifdef WIN32
        memstats.dwLength = sizeof(memstats);
        if (!GlobalMemoryStatusEx(&memstats)) {
                sg_set_error_with_errno(SG_ERROR_MEMSTATUS, NULL);
                return NULL;
        }
        mem_stat.free = memstats.ullAvailPhys;
        mem_stat.total = memstats.ullTotalPhys;
        mem_stat.used = mem_stat.total - mem_stat.free;
        if(read_counter_large(SG_WIN32_MEM_CACHE, &mem_stat.cache)) {
                mem_stat.cache = 0;
        }
#endif
        return &mem_stat;
}
Revision:	1.32
Committed:	Sat Sep 24 13:29:22 2005 UTC (18 years, 7 months ago) by tdb
Content type:	text/plain
Branch:	MAIN
CVS Tags:	LIBSTATGRAB_0_13
Changes since 1.31:	+21 -1 lines
Log Message:	Add WIN32 support via MINGW. We'll need to add stuff to the README file about what this requires to build. All the hard work done by: skel
#	User	Rev	Content
1	tdb	1.18	/*
2	tdb	1.25	* i-scream libstatgrab
3	tdb	1.7	* http://www.i-scream.org
4	tdb	1.18	* Copyright (C) 2000-2004 i-scream
5	pajs	1.1	*
6	tdb	1.18	* This library is free software; you can redistribute it and/or
7			* modify it under the terms of the GNU Lesser General Public
8			* License as published by the Free Software Foundation; either
9			* version 2.1 of the License, or (at your option) any later version.
10	pajs	1.1	*
11	tdb	1.18	* This library is distributed in the hope that it will be useful,
12	pajs	1.1	* but WITHOUT ANY WARRANTY; without even the implied warranty of
13	tdb	1.18	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14			* Lesser General Public License for more details.
15	pajs	1.1	*
16	tdb	1.18	* You should have received a copy of the GNU Lesser General Public
17			* License along with this library; if not, write to the Free Software
18			* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
19			* 02111-1307 USA
20	tdb	1.19	*
21	tdb	1.32	* $Id: memory_stats.c,v 1.31 2004/11/06 14:55:53 ats Exp $
22	pajs	1.1	*/
23
24			#ifdef HAVE_CONFIG_H
25			#include "config.h"
26			#endif
27
28			#include "statgrab.h"
29	ats	1.12	#include "tools.h"
30	pajs	1.1	#ifdef SOLARIS
31			#include <unistd.h>
32			#include <kstat.h>
33			#endif
34	ats	1.17	#if defined(LINUX) \|\| defined(CYGWIN)
35	pajs	1.5	#include <stdio.h>
36			#include <string.h>
37			#endif
38	tdb	1.23	#if defined(FREEBSD) \|\| defined(DFBSD)
39	pajs	1.6	#include <sys/types.h>
40			#include <sys/sysctl.h>
41			#include <unistd.h>
42			#endif
43	tdb	1.21	#if defined(NETBSD) \|\| defined(OPENBSD)
44	tdb	1.20	#include <sys/param.h>
45	ats	1.22	#include <sys/time.h>
46	tdb	1.20	#include <uvm/uvm.h>
47			#endif
48	tdb	1.30	#ifdef HPUX
49			#include <sys/param.h>
50			#include <sys/pstat.h>
51			#include <unistd.h>
52			#endif
53	tdb	1.32	#ifdef WIN32
54			#include <windows.h>
55			#include "win32.h"
56			#endif
57	pajs	1.1
58	ats	1.24	sg_mem_stats *sg_get_mem_stats(){
59	pajs	1.1
60	ats	1.24	static sg_mem_stats mem_stat;
61	pajs	1.1
62	tdb	1.30	#ifdef HPUX
63	ats	1.31	struct pst_static *pstat_static;
64	tdb	1.30	struct pst_dynamic pstat_dynamic;
65			long long pagesize;
66			#endif
67	pajs	1.1	#ifdef SOLARIS
68			kstat_ctl_t *kc;
69			kstat_t *ksp;
70			kstat_named_t *kn;
71			long totalmem;
72			int pagesize;
73			#endif
74	ats	1.17	#if defined(LINUX) \|\| defined(CYGWIN)
75	pajs	1.5	char *line_ptr;
76	ats	1.15	unsigned long long value;
77	pajs	1.5	FILE *f;
78			#endif
79	tdb	1.23	#if defined(FREEBSD) \|\| defined(DFBSD)
80	ats	1.12	int mib[2];
81	ats	1.14	u_long physmem;
82			size_t size;
83	tdb	1.8	u_int free_count;
84			u_int cache_count;
85			u_int inactive_count;
86	ats	1.12	int pagesize;
87			#endif
88	tdb	1.20	#if defined(NETBSD) \|\| defined(OPENBSD)
89	ats	1.12	struct uvmexp *uvm;
90			#endif
91	tdb	1.32	#ifdef WIN32
92			MEMORYSTATUSEX memstats;
93			#endif
94	pajs	1.1
95	tdb	1.30	#ifdef HPUX
96			if((pagesize=sysconf(_SC_PAGESIZE)) == -1){
97			sg_set_error_with_errno(SG_ERROR_SYSCONF, "_SC_PAGESIZE");
98			return NULL;
99			}
100
101			if (pstat_getdynamic(&pstat_dynamic, sizeof(pstat_dynamic), 1, 0) == -1) {
102			sg_set_error_with_errno(SG_ERROR_PSTAT, "pstat_dynamic");
103			return NULL;
104			}
105	ats	1.31	pstat_static = sg_get_pstat_static();
106			if (pstat_static == NULL) {
107	tdb	1.30	return NULL;
108			}
109
110	ats	1.31	/* FIXME Does this include swap? */
111			mem_stat.total = ((long long) pstat_static->physical_memory) * pagesize;
112	tdb	1.30	mem_stat.free = ((long long) pstat_dynamic.psd_free) * pagesize;
113			mem_stat.used = mem_stat.total - mem_stat.free;
114			#endif
115	pajs	1.1	#ifdef SOLARIS
116			if((pagesize=sysconf(_SC_PAGESIZE)) == -1){
117	ats	1.28	sg_set_error_with_errno(SG_ERROR_SYSCONF, "_SC_PAGESIZE");
118	pajs	1.1	return NULL;
119			}
120
121			if((totalmem=sysconf(_SC_PHYS_PAGES)) == -1){
122	ats	1.28	sg_set_error_with_errno(SG_ERROR_SYSCONF, "_SC_PHYS_PAGES");
123	pajs	1.1	return NULL;
124			}
125
126			if ((kc = kstat_open()) == NULL) {
127	tdb	1.27	sg_set_error(SG_ERROR_KSTAT_OPEN, NULL);
128	pajs	1.1	return NULL;
129			}
130			if((ksp=kstat_lookup(kc, "unix", 0, "system_pages")) == NULL){
131	tdb	1.27	sg_set_error(SG_ERROR_KSTAT_LOOKUP, "unix,0,system_pages");
132	pajs	1.1	return NULL;
133			}
134			if (kstat_read(kc, ksp, 0) == -1) {
135	tdb	1.27	sg_set_error(SG_ERROR_KSTAT_READ, NULL);
136	pajs	1.1	return NULL;
137			}
138			if((kn=kstat_data_lookup(ksp, "freemem")) == NULL){
139	tdb	1.27	sg_set_error(SG_ERROR_KSTAT_DATA_LOOKUP, "freemem");
140	pajs	1.1	return NULL;
141			}
142	tdb	1.26	kstat_close(kc);
143	pajs	1.2
144	pajs	1.1	mem_stat.total = (long long)totalmem * (long long)pagesize;
145			mem_stat.free = ((long long)kn->value.ul) * (long long)pagesize;
146			mem_stat.used = mem_stat.total - mem_stat.free;
147	pajs	1.5	#endif
148
149	ats	1.17	#if defined(LINUX) \|\| defined(CYGWIN)
150	ats	1.16	if ((f = fopen("/proc/meminfo", "r")) == NULL) {
151	ats	1.28	sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/meminfo");
152	pajs	1.5	return NULL;
153			}
154
155	ats	1.24	while ((line_ptr = sg_f_read_line(f, "")) != NULL) {
156	ats	1.16	if (sscanf(line_ptr, "%*s %llu kB", &value) != 1) {
157	ats	1.15	continue;
158			}
159			value *= 1024;
160
161			if (strncmp(line_ptr, "MemTotal:", 9) == 0) {
162			mem_stat.total = value;
163			} else if (strncmp(line_ptr, "MemFree:", 8) == 0) {
164			mem_stat.free = value;
165			} else if (strncmp(line_ptr, "Cached:", 7) == 0) {
166			mem_stat.cache = value;
167			}
168	pajs	1.5	}
169
170			fclose(f);
171	ats	1.15	mem_stat.used = mem_stat.total - mem_stat.free;
172	pajs	1.6	#endif
173
174	tdb	1.23	#if defined(FREEBSD) \|\| defined(DFBSD)
175	tdb	1.8	/* Returns bytes */
176	ats	1.12	mib[0] = CTL_HW;
177			mib[1] = HW_PHYSMEM;
178	ats	1.11	size = sizeof physmem;
179	ats	1.12	if (sysctl(mib, 2, &physmem, &size, NULL, 0) < 0) {
180	ats	1.28	sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_HW.HW_PHYSMEM");
181	pajs	1.6	return NULL;
182	ats	1.12	}
183			mem_stat.total = physmem;
184	pajs	1.6
185			/returns pages/
186	ats	1.11	size = sizeof free_count;
187	tdb	1.27	if (sysctlbyname("vm.stats.vm.v_free_count", &free_count, &size, NULL, 0) < 0){
188	ats	1.28	sg_set_error_with_errno(SG_ERROR_SYSCTLBYNAME,
189			"vm.stats.vm.v_free_count");
190	pajs	1.6	return NULL;
191	tdb	1.27	}
192	pajs	1.6
193	ats	1.11	size = sizeof inactive_count;
194	tdb	1.27	if (sysctlbyname("vm.stats.vm.v_inactive_count", &inactive_count , &size, NULL, 0) < 0){
195	ats	1.28	sg_set_error_with_errno(SG_ERROR_SYSCTLBYNAME,
196			"vm.stats.vm.v_inactive_count");
197	pajs	1.6	return NULL;
198	tdb	1.27	}
199	pajs	1.6
200	ats	1.11	size = sizeof cache_count;
201	tdb	1.27	if (sysctlbyname("vm.stats.vm.v_cache_count", &cache_count, &size, NULL, 0) < 0){
202	ats	1.28	sg_set_error_with_errno(SG_ERROR_SYSCTLBYNAME,
203			"vm.stats.vm.v_cache_count");
204	pajs	1.6	return NULL;
205	tdb	1.27	}
206	pajs	1.6
207	tdb	1.8	/* Because all the vm.stats returns pages, I need to get the page size.
208	tdb	1.27	* After that I then need to multiple the anything that used vm.stats to
209	tdb	1.8	* get the system statistics by pagesize
210	pajs	1.6	*/
211	ats	1.29	pagesize = getpagesize();
212	tdb	1.8	mem_stat.cache=cache_count*pagesize;
213
214			/* Of couse nothing is ever that simple :) And I have inactive pages to
215			* deal with too. So I'm going to add them to free memory :)
216	pajs	1.6	*/
217	tdb	1.8	mem_stat.free=(free_countpagesize)+(inactive_countpagesize);
218			mem_stat.used=physmem-mem_stat.free;
219	ats	1.12	#endif
220	ats	1.14
221	tdb	1.21	#if defined(NETBSD) \|\| defined(OPENBSD)
222	ats	1.24	if ((uvm = sg_get_uvmexp()) == NULL) {
223	ats	1.12	return NULL;
224			}
225	tdb	1.20
226	ats	1.14	mem_stat.total = uvm->pagesize * uvm->npages;
227	tdb	1.20	#ifdef NETBSD
228	ats	1.12	mem_stat.cache = uvm->pagesize * (uvm->filepages + uvm->execpages);
229	tdb	1.20	#else
230	tdb	1.21	/* Can't find cache memory on OpenBSD */
231	tdb	1.20	mem_stat.cache = 0;
232			#endif
233	ats	1.14	mem_stat.free = uvm->pagesize * (uvm->free + uvm->inactive);
234			mem_stat.used = mem_stat.total - mem_stat.free;
235	pajs	1.1	#endif
236
237	tdb	1.32	#ifdef WIN32
238			memstats.dwLength = sizeof(memstats);
239			if (!GlobalMemoryStatusEx(&memstats)) {
240			sg_set_error_with_errno(SG_ERROR_MEMSTATUS, NULL);
241			return NULL;
242			}
243			mem_stat.free = memstats.ullAvailPhys;
244			mem_stat.total = memstats.ullTotalPhys;
245			mem_stat.used = mem_stat.total - mem_stat.free;
246			if(read_counter_large(SG_WIN32_MEM_CACHE, &mem_stat.cache)) {
247			mem_stat.cache = 0;
248			}
249			#endif
250	pajs	1.1	return &mem_stat;
251			}