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.27 2004/04/07 21:08:40 tdb 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

sg_mem_stats *sg_get_mem_stats(){

        static sg_mem_stats mem_stat;

#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 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 
         */
        /* FIXME getpagesize() cannot actually fail */
        if ((pagesize=getpagesize()) == -1){
                sg_set_error(SG_ERROR_GETPAGESIZE, "NULL");
                return NULL;
        }

        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

        return &mem_stat;
}
Revision:	1.28
Committed:	Sun Jul 18 21:30:11 2004 UTC (19 years, 10 months ago) by ats
Content type:	text/plain
Branch:	MAIN
Changes since 1.27:	+12 -8 lines
Log Message:	Use sg_set_error_with_errno whenever errno's valid. Change the one user of SG_ERROR_ENOENT to SG_ERROR_SYSCTL instead (since that's what it should have been).
#	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	ats	1.28	* $Id: memory_stats.c,v 1.27 2004/04/07 21:08:40 tdb 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	pajs	1.1
49	ats	1.24	sg_mem_stats *sg_get_mem_stats(){
50	pajs	1.1
51	ats	1.24	static sg_mem_stats mem_stat;
52	pajs	1.1
53			#ifdef SOLARIS
54			kstat_ctl_t *kc;
55			kstat_t *ksp;
56			kstat_named_t *kn;
57			long totalmem;
58			int pagesize;
59			#endif
60	ats	1.17	#if defined(LINUX) \|\| defined(CYGWIN)
61	pajs	1.5	char *line_ptr;
62	ats	1.15	unsigned long long value;
63	pajs	1.5	FILE *f;
64			#endif
65	tdb	1.23	#if defined(FREEBSD) \|\| defined(DFBSD)
66	ats	1.12	int mib[2];
67	ats	1.14	u_long physmem;
68			size_t size;
69	tdb	1.8	u_int free_count;
70			u_int cache_count;
71			u_int inactive_count;
72	ats	1.12	int pagesize;
73			#endif
74	tdb	1.20	#if defined(NETBSD) \|\| defined(OPENBSD)
75	ats	1.12	struct uvmexp *uvm;
76			#endif
77	pajs	1.1
78			#ifdef SOLARIS
79			if((pagesize=sysconf(_SC_PAGESIZE)) == -1){
80	ats	1.28	sg_set_error_with_errno(SG_ERROR_SYSCONF, "_SC_PAGESIZE");
81	pajs	1.1	return NULL;
82			}
83
84			if((totalmem=sysconf(_SC_PHYS_PAGES)) == -1){
85	ats	1.28	sg_set_error_with_errno(SG_ERROR_SYSCONF, "_SC_PHYS_PAGES");
86	pajs	1.1	return NULL;
87			}
88
89			if ((kc = kstat_open()) == NULL) {
90	tdb	1.27	sg_set_error(SG_ERROR_KSTAT_OPEN, NULL);
91	pajs	1.1	return NULL;
92			}
93			if((ksp=kstat_lookup(kc, "unix", 0, "system_pages")) == NULL){
94	tdb	1.27	sg_set_error(SG_ERROR_KSTAT_LOOKUP, "unix,0,system_pages");
95	pajs	1.1	return NULL;
96			}
97			if (kstat_read(kc, ksp, 0) == -1) {
98	tdb	1.27	sg_set_error(SG_ERROR_KSTAT_READ, NULL);
99	pajs	1.1	return NULL;
100			}
101			if((kn=kstat_data_lookup(ksp, "freemem")) == NULL){
102	tdb	1.27	sg_set_error(SG_ERROR_KSTAT_DATA_LOOKUP, "freemem");
103	pajs	1.1	return NULL;
104			}
105	tdb	1.26	kstat_close(kc);
106	pajs	1.2
107	pajs	1.1	mem_stat.total = (long long)totalmem * (long long)pagesize;
108			mem_stat.free = ((long long)kn->value.ul) * (long long)pagesize;
109			mem_stat.used = mem_stat.total - mem_stat.free;
110	pajs	1.5	#endif
111
112	ats	1.17	#if defined(LINUX) \|\| defined(CYGWIN)
113	ats	1.16	if ((f = fopen("/proc/meminfo", "r")) == NULL) {
114	ats	1.28	sg_set_error_with_errno(SG_ERROR_OPEN, "/proc/meminfo");
115	pajs	1.5	return NULL;
116			}
117
118	ats	1.24	while ((line_ptr = sg_f_read_line(f, "")) != NULL) {
119	ats	1.16	if (sscanf(line_ptr, "%*s %llu kB", &value) != 1) {
120	ats	1.15	continue;
121			}
122			value *= 1024;
123
124			if (strncmp(line_ptr, "MemTotal:", 9) == 0) {
125			mem_stat.total = value;
126			} else if (strncmp(line_ptr, "MemFree:", 8) == 0) {
127			mem_stat.free = value;
128			} else if (strncmp(line_ptr, "Cached:", 7) == 0) {
129			mem_stat.cache = value;
130			}
131	pajs	1.5	}
132
133			fclose(f);
134	ats	1.15	mem_stat.used = mem_stat.total - mem_stat.free;
135	pajs	1.6	#endif
136
137	tdb	1.23	#if defined(FREEBSD) \|\| defined(DFBSD)
138	tdb	1.8	/* Returns bytes */
139	ats	1.12	mib[0] = CTL_HW;
140			mib[1] = HW_PHYSMEM;
141	ats	1.11	size = sizeof physmem;
142	ats	1.12	if (sysctl(mib, 2, &physmem, &size, NULL, 0) < 0) {
143	ats	1.28	sg_set_error_with_errno(SG_ERROR_SYSCTL, "CTL_HW.HW_PHYSMEM");
144	pajs	1.6	return NULL;
145	ats	1.12	}
146			mem_stat.total = physmem;
147	pajs	1.6
148			/returns pages/
149	ats	1.11	size = sizeof free_count;
150	tdb	1.27	if (sysctlbyname("vm.stats.vm.v_free_count", &free_count, &size, NULL, 0) < 0){
151	ats	1.28	sg_set_error_with_errno(SG_ERROR_SYSCTLBYNAME,
152			"vm.stats.vm.v_free_count");
153	pajs	1.6	return NULL;
154	tdb	1.27	}
155	pajs	1.6
156	ats	1.11	size = sizeof inactive_count;
157	tdb	1.27	if (sysctlbyname("vm.stats.vm.v_inactive_count", &inactive_count , &size, NULL, 0) < 0){
158	ats	1.28	sg_set_error_with_errno(SG_ERROR_SYSCTLBYNAME,
159			"vm.stats.vm.v_inactive_count");
160	pajs	1.6	return NULL;
161	tdb	1.27	}
162	pajs	1.6
163	ats	1.11	size = sizeof cache_count;
164	tdb	1.27	if (sysctlbyname("vm.stats.vm.v_cache_count", &cache_count, &size, NULL, 0) < 0){
165	ats	1.28	sg_set_error_with_errno(SG_ERROR_SYSCTLBYNAME,
166			"vm.stats.vm.v_cache_count");
167	pajs	1.6	return NULL;
168	tdb	1.27	}
169	pajs	1.6
170	tdb	1.8	/* Because all the vm.stats returns pages, I need to get the page size.
171	tdb	1.27	* After that I then need to multiple the anything that used vm.stats to
172	tdb	1.8	* get the system statistics by pagesize
173	pajs	1.6	*/
174	ats	1.28	/* FIXME getpagesize() cannot actually fail */
175	pajs	1.6	if ((pagesize=getpagesize()) == -1){
176	tdb	1.27	sg_set_error(SG_ERROR_GETPAGESIZE, "NULL");
177	pajs	1.6	return NULL;
178			}
179
180	tdb	1.8	mem_stat.cache=cache_count*pagesize;
181
182			/* Of couse nothing is ever that simple :) And I have inactive pages to
183			* deal with too. So I'm going to add them to free memory :)
184	pajs	1.6	*/
185	tdb	1.8	mem_stat.free=(free_countpagesize)+(inactive_countpagesize);
186			mem_stat.used=physmem-mem_stat.free;
187	ats	1.12	#endif
188	ats	1.14
189	tdb	1.21	#if defined(NETBSD) \|\| defined(OPENBSD)
190	ats	1.24	if ((uvm = sg_get_uvmexp()) == NULL) {
191	ats	1.12	return NULL;
192			}
193	tdb	1.20
194	ats	1.14	mem_stat.total = uvm->pagesize * uvm->npages;
195	tdb	1.20	#ifdef NETBSD
196	ats	1.12	mem_stat.cache = uvm->pagesize * (uvm->filepages + uvm->execpages);
197	tdb	1.20	#else
198	tdb	1.21	/* Can't find cache memory on OpenBSD */
199	tdb	1.20	mem_stat.cache = 0;
200			#endif
201	ats	1.14	mem_stat.free = uvm->pagesize * (uvm->free + uvm->inactive);
202			mem_stat.used = mem_stat.total - mem_stat.free;
203	pajs	1.1	#endif
204
205			return &mem_stat;
206			}