xmem/get_mem.c

/*  $Id: get_mem.c,v 1.5 2018/12/26 18:46:04 bch Exp $
 *
 * xmem(1) utility adapted by Christian Barthel <bch@online.de>:
 *   - Added OpenBSD support
 *   - Added FreeBSD support
 *
 * get memory usage, from get_load.c derived
 *
 * Author:  Hans-Helmut Buehmann 20. Jan. 1996
 *
 * Modified for more recent kernels Helmut Geyer Oct. 1996
 */

#if __OpenBSD__

#include <sys/param.h>	/* DEV_BSIZE MAXCOMLEN PZERO */
#include <sys/sysctl.h>
#include <sys/swap.h>
#include <sys/mount.h>
#include <sys/proc.h>

#include <unistd.h>

#include <X11/Xos.h>
#include <X11/Intrinsic.h>
#include <stdio.h>
#include <stdlib.h>
#include <err.h>
#include "MemStripChart.h"

static int pageshift = -1;
#define pagetok(size) ((size) << pageshift)
#define LOG1024 10

static double allmem_kb = 0;

static void initPageShift(void)
{
	int pagesize = getpagesize();
	pageshift = 0;
	while (pagesize > 1) {
		pageshift++;
		pagesize >>= 1;
	}
	pageshift -= LOG1024;
}

static double getSwapFrac()
{
	struct swapent *swdev;
	int used, total, nswap, rnswap, i;

	nswap = swapctl(SWAP_NSWAP, 0, 0);
	if (nswap == 0)
		return 0;

	swdev = calloc(nswap, sizeof(*swdev));
	if (swdev == NULL)
		return 0;

	rnswap = swapctl(SWAP_STATS, swdev, nswap);
	if (rnswap == -1) {
		free(swdev);
		return 0;
	}

	/* if rnswap != nswap, then what? */

	/* Total things up */
	total = used = 0;
	for (i = 0; i < nswap; i++) {
		if (swdev[i].se_flags & SWF_ENABLE) {
			used += (swdev[i].se_inuse / (1024 / DEV_BSIZE));
			total += (swdev[i].se_nblks / (1024 / DEV_BSIZE));
		}
	}
	free(swdev);
	return (double)used/total;
}

void GetMemLoadPoint(Widget w, caddr_t closure, caddr_t call_data)
{
	static int bcstats_mib[] = {CTL_VFS, VFS_GENERIC, VFS_BCACHESTAT};
	static int uvmexp_mib[] = {CTL_VM, VM_UVMEXP};
	MemStripChartCallbackData ret;
	struct uvmexp uvmexp;
	struct bcachestats bcstats;
	size_t size;

	if (pageshift < 0)
		initPageShift();

	size = sizeof(uvmexp);
	if (sysctl(uvmexp_mib, 2, &uvmexp, &size, NULL, 0) < 0) {
		warn("sysctl failed");
		bzero(&uvmexp, sizeof(uvmexp));
	}
	size = sizeof(bcstats);
	if (sysctl(bcstats_mib, 3, &bcstats, &size, NULL, 0) < 0) {
		warn("sysctl failed");
		bzero(&bcstats, sizeof(bcstats));
	}


	allmem_kb = pagetok(uvmexp.npages);
	ret.cached = pagetok(bcstats.numbufpages) / allmem_kb;
	ret.free = pagetok(uvmexp.free) / allmem_kb;
	ret.buffer = 0;
	ret.code = (1 - ret.cached - ret.free);
	ret.swap = getSwapFrac();

#ifdef DEBUG
	printf("%lf  %lf %lf  %lf %lf\n",
	    allmem_kb,
	    ret.code,
	    ret.cached,
	    ret.free,
	    ret.swap);
#endif

	memcpy(call_data, &ret, sizeof(MemStripChartCallbackData));
}

#endif

/* ------------------------------------------------------------------- */

#if __FreeBSD__

/*  $Id: get_mem_fbsd.c,v 1.1 2018/12/26 18:46:16 bch Exp $
 *
 * Author: Christian Barthel <bch@online.de>
 */


#include <sys/types.h>	/* DEV_BSIZE MAXCOMLEN PZERO */
#include <sys/sysctl.h>

#include <unistd.h>
#include <kvm.h>

#include <X11/Xos.h>
#include <X11/Intrinsic.h>
#include <stdio.h>
#include <stdlib.h>
#include <err.h>
#include "MemStripChart.h"

static int pageshift = -1;
static int pagesize  = 0;
static kvm_t *kd = 0;
#define pagetok(size) ((size) << pageshift)
#define LOG1024 10

static unsigned int allmem_pages = 0;

static void initPageShift(void)
{
	pagesize = getpagesize();
	size_t len = sizeof(allmem_pages);
	int mib[len];
	pageshift = 0;
	while (pagesize > 1) {
		pageshift++;
		pagesize >>= 1;
	}
	pageshift -= LOG1024;

	kd = kvm_open(NULL, "/dev/null" ,NULL, O_RDONLY, "kvm_open");

	if (sysctlbyname("vm.stats.vm.v_page_count", &allmem_pages, &len, NULL, 0) == -1)
			perror("sysctl");
#ifdef DEBUG
	printf("page_count: %d\n", allmem_pages);
#endif
}

static double getSwapFrac()
{
		/* XXX Borrowed from top(1) code: */
		int n;
		struct kvm_swap swapary[1];
		static int pagesize = 0;
		static unsigned long swap_maxpages = 0;
		size_t sz = sizeof(swap_maxpages);

		int retavail = 0;
		int retfree = 0;


		n = kvm_getswapinfo(kd, swapary, 1, 0);
		if (n < 0 || swapary[0].ksw_total == 0)
				return (0);

		if (swap_maxpages == 0)
				sysctlbyname("vm.swap_maxpages",
					     &swap_maxpages, &sz, NULL, 0);

		if ( swapary[0].ksw_total > swap_maxpages )
				swapary[0].ksw_total = swap_maxpages;

		retavail = swapary[0].ksw_total;
		retfree = swapary[0].ksw_total - swapary[0].ksw_used;

#ifdef DEBUG
		printf("swap: %d %d %d %d\n", retavail,
		       retavail, swapary[0].ksw_total,
		       swapary[0].ksw_used);
#endif
		return (double)(retavail-retfree)/(double)retavail;
}

void GetMemLoadPoint(Widget w, caddr_t closure, caddr_t call_data)
{
	MemStripChartCallbackData ret;
	int a, b;
	size_t sz = sizeof(int);
	double cache = 0;

	if (pageshift < 0)
		initPageShift();

	sysctlbyname("vm.stats.vm.v_wire_count", &a, &sz, NULL, 0);
	ret.code = (double)a / (double)allmem_pages;

	sysctlbyname("vm.stats.vm.v_active_count", &a, &sz, NULL, 0);
	ret.buffer = (double)a / (double)allmem_pages;

	/* v_cache_count and v_inactive_count are treated similarly */
	sysctlbyname("vm.stats.vm.v_inactive_count", &a, &sz, NULL, 0);
	sysctlbyname("vm.stats.vm.v_cache_count", &b, &sz, NULL, 0);
	ret.cached = (double)(a + b) / allmem_pages;


	sysctlbyname("vm.stats.vm.v_free_count", &a, &sz, NULL, 0);
	ret.free = (double)a / allmem_pages;
	ret.swap = getSwapFrac();

#ifdef DEBUG
	printf("%u  %lf %lf  %lf %lf\n", allmem_pages,
				 ret.code, ret.cached, ret.free, ret.swap);
#endif
	memcpy(call_data, &ret, sizeof(MemStripChartCallbackData));
}

#endif