957 lines
		
	
	
		
			21 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			957 lines
		
	
	
		
			21 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*
 | 
						|
 * Licensed under GPLv2, see file LICENSE in this source tree.
 | 
						|
 *
 | 
						|
 * Based on nanotop.c from floppyfw project
 | 
						|
 *
 | 
						|
 * Contact me: vda.linux@googlemail.com
 | 
						|
 */
 | 
						|
 | 
						|
//config:config NMETER
 | 
						|
//config:	bool "nmeter"
 | 
						|
//config:	default y
 | 
						|
//config:	help
 | 
						|
//config:	  Prints selected system stats continuously, one line per update.
 | 
						|
 | 
						|
//applet:IF_NMETER(APPLET(nmeter, BB_DIR_USR_BIN, BB_SUID_DROP))
 | 
						|
 | 
						|
//kbuild:lib-$(CONFIG_NMETER) += nmeter.o
 | 
						|
 | 
						|
//usage:#define nmeter_trivial_usage
 | 
						|
//usage:       "[-d MSEC] FORMAT_STRING"
 | 
						|
//usage:#define nmeter_full_usage "\n\n"
 | 
						|
//usage:       "Monitor system in real time"
 | 
						|
//usage:     "\n"
 | 
						|
//usage:     "\n -d MSEC	Milliseconds between updates, default:1000, none:-1"
 | 
						|
//usage:     "\n"
 | 
						|
//usage:     "\nFormat specifiers:"
 | 
						|
//usage:     "\n %Nc or %[cN]	CPU. N - bar size (default:10)"
 | 
						|
//usage:     "\n		(displays: S:system U:user N:niced D:iowait I:irq i:softirq)"
 | 
						|
//usage:     "\n %[nINTERFACE]	Network INTERFACE"
 | 
						|
//usage:     "\n %m		Allocated memory"
 | 
						|
//usage:     "\n %[mf]		Free memory"
 | 
						|
//usage:     "\n %[mt]		Total memory"
 | 
						|
//usage:     "\n %s		Allocated swap"
 | 
						|
//usage:     "\n %f		Number of used file descriptors"
 | 
						|
//usage:     "\n %Ni		Total/specific IRQ rate"
 | 
						|
//usage:     "\n %x		Context switch rate"
 | 
						|
//usage:     "\n %p		Forks"
 | 
						|
//usage:     "\n %[pn]		# of processes"
 | 
						|
//usage:     "\n %b		Block io"
 | 
						|
//usage:     "\n %Nt		Time (with N decimal points)"
 | 
						|
//usage:     "\n %r		Print <cr> instead of <lf> at EOL"
 | 
						|
 | 
						|
//TODO:
 | 
						|
// simplify code
 | 
						|
// /proc/locks
 | 
						|
// /proc/stat:
 | 
						|
// disk_io: (3,0):(22272,17897,410702,4375,54750)
 | 
						|
// btime 1059401962
 | 
						|
//TODO: use sysinfo libc call/syscall, if appropriate
 | 
						|
// (faster than open/read/close):
 | 
						|
// sysinfo({uptime=15017, loads=[5728, 15040, 16480]
 | 
						|
//  totalram=2107416576, freeram=211525632, sharedram=0, bufferram=157204480}
 | 
						|
//  totalswap=134209536, freeswap=134209536, procs=157})
 | 
						|
 | 
						|
#include "libbb.h"
 | 
						|
#include "common_bufsiz.h"
 | 
						|
 | 
						|
typedef unsigned long long ullong;
 | 
						|
 | 
						|
enum {  /* Preferably use powers of 2 */
 | 
						|
	PROC_MIN_FILE_SIZE = 256,
 | 
						|
	PROC_MAX_FILE_SIZE = 16 * 1024,
 | 
						|
};
 | 
						|
 | 
						|
typedef struct proc_file {
 | 
						|
	char *file;
 | 
						|
	int file_sz;
 | 
						|
	smallint last_gen;
 | 
						|
} proc_file;
 | 
						|
 | 
						|
static const char *const proc_name[] = {
 | 
						|
	"stat",		// Must match the order of proc_file's!
 | 
						|
	"loadavg",
 | 
						|
	"net/dev",
 | 
						|
	"meminfo",
 | 
						|
	"diskstats",
 | 
						|
	"sys/fs/file-nr"
 | 
						|
};
 | 
						|
 | 
						|
struct globals {
 | 
						|
	// Sample generation flip-flop
 | 
						|
	smallint gen;
 | 
						|
	// Linux 2.6? (otherwise assumes 2.4)
 | 
						|
	smallint is26;
 | 
						|
	// 1 if sample delay is not an integer fraction of a second
 | 
						|
	smallint need_seconds;
 | 
						|
	char final_char;
 | 
						|
	char *cur_outbuf;
 | 
						|
	int delta;
 | 
						|
	unsigned deltanz;
 | 
						|
	struct timeval tv;
 | 
						|
#define first_proc_file proc_stat
 | 
						|
	proc_file proc_stat;	// Must match the order of proc_name's!
 | 
						|
	proc_file proc_loadavg;
 | 
						|
	proc_file proc_net_dev;
 | 
						|
	proc_file proc_meminfo;
 | 
						|
	proc_file proc_diskstats;
 | 
						|
	proc_file proc_sys_fs_filenr;
 | 
						|
};
 | 
						|
#define G (*ptr_to_globals)
 | 
						|
#define gen                (G.gen               )
 | 
						|
#define is26               (G.is26              )
 | 
						|
#define need_seconds       (G.need_seconds      )
 | 
						|
#define cur_outbuf         (G.cur_outbuf        )
 | 
						|
#define tv                 (G.tv                )
 | 
						|
#define proc_stat          (G.proc_stat         )
 | 
						|
#define proc_loadavg       (G.proc_loadavg      )
 | 
						|
#define proc_net_dev       (G.proc_net_dev      )
 | 
						|
#define proc_meminfo       (G.proc_meminfo      )
 | 
						|
#define proc_diskstats     (G.proc_diskstats    )
 | 
						|
#define proc_sys_fs_filenr (G.proc_sys_fs_filenr)
 | 
						|
#define outbuf bb_common_bufsiz1
 | 
						|
#define INIT_G() do { \
 | 
						|
	setup_common_bufsiz(); \
 | 
						|
	SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
 | 
						|
	cur_outbuf = outbuf; \
 | 
						|
	G.final_char = '\n'; \
 | 
						|
	G.deltanz = G.delta = 1000000; \
 | 
						|
} while (0)
 | 
						|
 | 
						|
static inline void reset_outbuf(void)
 | 
						|
{
 | 
						|
	cur_outbuf = outbuf;
 | 
						|
}
 | 
						|
 | 
						|
static inline int outbuf_count(void)
 | 
						|
{
 | 
						|
	return cur_outbuf - outbuf;
 | 
						|
}
 | 
						|
 | 
						|
static void print_outbuf(void)
 | 
						|
{
 | 
						|
	int sz = cur_outbuf - outbuf;
 | 
						|
	if (sz > 0) {
 | 
						|
		xwrite(STDOUT_FILENO, outbuf, sz);
 | 
						|
		cur_outbuf = outbuf;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void put(const char *s)
 | 
						|
{
 | 
						|
	char *p = cur_outbuf;
 | 
						|
	int sz = outbuf + COMMON_BUFSIZE - p;
 | 
						|
	while (*s && --sz >= 0)
 | 
						|
		*p++ = *s++;
 | 
						|
	cur_outbuf = p;
 | 
						|
}
 | 
						|
 | 
						|
static void put_c(char c)
 | 
						|
{
 | 
						|
	if (cur_outbuf < outbuf + COMMON_BUFSIZE)
 | 
						|
		*cur_outbuf++ = c;
 | 
						|
}
 | 
						|
 | 
						|
static void put_question_marks(int count)
 | 
						|
{
 | 
						|
	while (count--)
 | 
						|
		put_c('?');
 | 
						|
}
 | 
						|
 | 
						|
static void readfile_z(proc_file *pf, const char* fname)
 | 
						|
{
 | 
						|
// open_read_close() will do two reads in order to be sure we are at EOF,
 | 
						|
// and we don't need/want that.
 | 
						|
	int fd;
 | 
						|
	int sz, rdsz;
 | 
						|
	char *buf;
 | 
						|
 | 
						|
	sz = pf->file_sz;
 | 
						|
	buf = pf->file;
 | 
						|
	if (!buf) {
 | 
						|
		buf = xmalloc(PROC_MIN_FILE_SIZE);
 | 
						|
		sz = PROC_MIN_FILE_SIZE;
 | 
						|
	}
 | 
						|
 again:
 | 
						|
	fd = xopen(fname, O_RDONLY);
 | 
						|
	buf[0] = '\0';
 | 
						|
	rdsz = read(fd, buf, sz-1);
 | 
						|
	close(fd);
 | 
						|
	if (rdsz > 0) {
 | 
						|
		if (rdsz == sz-1 && sz < PROC_MAX_FILE_SIZE) {
 | 
						|
			sz *= 2;
 | 
						|
			buf = xrealloc(buf, sz);
 | 
						|
			goto again;
 | 
						|
		}
 | 
						|
		buf[rdsz] = '\0';
 | 
						|
	}
 | 
						|
	pf->file_sz = sz;
 | 
						|
	pf->file = buf;
 | 
						|
}
 | 
						|
 | 
						|
static const char* get_file(proc_file *pf)
 | 
						|
{
 | 
						|
	if (pf->last_gen != gen) {
 | 
						|
		pf->last_gen = gen;
 | 
						|
		readfile_z(pf, proc_name[pf - &first_proc_file]);
 | 
						|
	}
 | 
						|
	return pf->file;
 | 
						|
}
 | 
						|
 | 
						|
static ullong read_after_slash(const char *p)
 | 
						|
{
 | 
						|
	p = strchr(p, '/');
 | 
						|
	if (!p) return 0;
 | 
						|
	return strtoull(p+1, NULL, 10);
 | 
						|
}
 | 
						|
 | 
						|
enum conv_type {
 | 
						|
	conv_decimal = 0,
 | 
						|
	conv_slash = 1
 | 
						|
};
 | 
						|
 | 
						|
// Reads decimal values from line. Values start after key, for example:
 | 
						|
// "cpu  649369 0 341297 4336769..." - key is "cpu" here.
 | 
						|
// Values are stored in vec[].
 | 
						|
// posbits is a bit lit of positions we are interested in.
 | 
						|
// for example: 00100110 - we want 1st, 2nd and 5th value.
 | 
						|
// posbits.bit0 encodes conversion type.
 | 
						|
static int rdval(const char* p, const char* key, ullong *vec, long posbits)
 | 
						|
{
 | 
						|
	unsigned curpos;
 | 
						|
 | 
						|
	p = strstr(p, key);
 | 
						|
	if (!p) return 1;
 | 
						|
 | 
						|
	p += strlen(key);
 | 
						|
	curpos = 1 << 1;
 | 
						|
	while (1) {
 | 
						|
		while (*p == ' ' || *p == '\t') p++;
 | 
						|
		if (*p == '\n' || *p == '\0') break;
 | 
						|
 | 
						|
		if (curpos & posbits) { // read this value
 | 
						|
			*vec++ = (posbits & 1) == conv_decimal ?
 | 
						|
				strtoull(p, NULL, 10) :
 | 
						|
				read_after_slash(p);
 | 
						|
			posbits -= curpos;
 | 
						|
			if (posbits <= 1)
 | 
						|
				return 0;
 | 
						|
		}
 | 
						|
		while (*p > ' ') // skip over the value
 | 
						|
			p++;
 | 
						|
		curpos <<= 1;
 | 
						|
	}
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
// Parses files with lines like "... ... ... 3/148 ...."
 | 
						|
static int rdval_loadavg(const char* p, ullong *vec, long posbits)
 | 
						|
{
 | 
						|
	int result;
 | 
						|
	result = rdval(p, "", vec, posbits | conv_slash);
 | 
						|
	return result;
 | 
						|
}
 | 
						|
 | 
						|
// Parses /proc/diskstats
 | 
						|
//   1  2 3   4     5     6(rd)  7      8     9     10(wr) 11     12 13     14
 | 
						|
//   3  0 hda 51292 14441 841783 926052 25717 79650 843256 3029804 0 148459 3956933
 | 
						|
//   3  1 hda1 0 0 0 0 <- ignore if only 4 fields
 | 
						|
// Linux 3.0 (maybe earlier) started printing full stats for hda1 too.
 | 
						|
// Had to add code which skips such devices.
 | 
						|
static int rdval_diskstats(const char* p, ullong *vec)
 | 
						|
{
 | 
						|
	char devname[32];
 | 
						|
	unsigned devname_len = 0;
 | 
						|
	int value_idx = 0;
 | 
						|
 | 
						|
	vec[0] = 0;
 | 
						|
	vec[1] = 0;
 | 
						|
	while (1) {
 | 
						|
		value_idx++;
 | 
						|
		while (*p == ' ' || *p == '\t')
 | 
						|
			p++;
 | 
						|
		if (*p == '\0')
 | 
						|
			break;
 | 
						|
		if (*p == '\n') {
 | 
						|
			value_idx = 0;
 | 
						|
			p++;
 | 
						|
			continue;
 | 
						|
		}
 | 
						|
		if (value_idx == 3) {
 | 
						|
			char *end = strchrnul(p, ' ');
 | 
						|
			/* If this a hda1-like device (same prefix as last one + digit)? */
 | 
						|
			if (devname_len && strncmp(devname, p, devname_len) == 0 && isdigit(p[devname_len])) {
 | 
						|
				p = end;
 | 
						|
				goto skip_line; /* skip entire line */
 | 
						|
			}
 | 
						|
			/* It is not. Remember the name for future checks */
 | 
						|
			devname_len = end - p;
 | 
						|
			if (devname_len > sizeof(devname)-1)
 | 
						|
				devname_len = sizeof(devname)-1;
 | 
						|
			strncpy(devname, p, devname_len);
 | 
						|
			/* devname[devname_len] = '\0'; - not really needed */
 | 
						|
			p = end;
 | 
						|
		} else
 | 
						|
		if (value_idx == 6) {
 | 
						|
			// TODO: *sectorsize (don't know how to find out sectorsize)
 | 
						|
			vec[0] += strtoull(p, NULL, 10);
 | 
						|
		} else
 | 
						|
		if (value_idx == 10) {
 | 
						|
			// TODO: *sectorsize (don't know how to find out sectorsize)
 | 
						|
			vec[1] += strtoull(p, NULL, 10);
 | 
						|
 skip_line:
 | 
						|
			while (*p != '\n' && *p != '\0')
 | 
						|
				p++;
 | 
						|
			continue;
 | 
						|
		}
 | 
						|
		while ((unsigned char)(*p) > ' ') // skip over value
 | 
						|
			p++;
 | 
						|
	}
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
static void scale(ullong ul)
 | 
						|
{
 | 
						|
	char buf[5];
 | 
						|
 | 
						|
	/* see http://en.wikipedia.org/wiki/Tera */
 | 
						|
	smart_ulltoa4(ul, buf, " kmgtpezy")[0] = '\0';
 | 
						|
	put(buf);
 | 
						|
}
 | 
						|
 | 
						|
#define S_STAT(a) \
 | 
						|
typedef struct a { \
 | 
						|
	struct s_stat *next; \
 | 
						|
	void (*collect)(struct a *s) FAST_FUNC; \
 | 
						|
	const char *label;
 | 
						|
#define S_STAT_END(a) } a;
 | 
						|
 | 
						|
S_STAT(s_stat)
 | 
						|
S_STAT_END(s_stat)
 | 
						|
 | 
						|
static void FAST_FUNC collect_literal(s_stat *s UNUSED_PARAM)
 | 
						|
{
 | 
						|
}
 | 
						|
 | 
						|
static s_stat* init_literal(void)
 | 
						|
{
 | 
						|
	s_stat *s = xzalloc(sizeof(*s));
 | 
						|
	s->collect = collect_literal;
 | 
						|
	return (s_stat*)s;
 | 
						|
}
 | 
						|
 | 
						|
static s_stat* init_cr(const char *param UNUSED_PARAM)
 | 
						|
{
 | 
						|
	G.final_char = '\r';
 | 
						|
	return NULL;
 | 
						|
}
 | 
						|
 | 
						|
//     user nice system idle  iowait irq  softirq (last 3 only in 2.6)
 | 
						|
//cpu  649369 0 341297 4336769 11640 7122 1183
 | 
						|
//cpuN 649369 0 341297 4336769 11640 7122 1183
 | 
						|
enum { CPU_FIELDCNT = 7 };
 | 
						|
S_STAT(cpu_stat)
 | 
						|
	ullong old[CPU_FIELDCNT];
 | 
						|
	int bar_sz;
 | 
						|
	char bar[1];
 | 
						|
S_STAT_END(cpu_stat)
 | 
						|
 | 
						|
static void FAST_FUNC collect_cpu(cpu_stat *s)
 | 
						|
{
 | 
						|
	ullong data[CPU_FIELDCNT] = { 0, 0, 0, 0, 0, 0, 0 };
 | 
						|
	unsigned frac[CPU_FIELDCNT] = { 0, 0, 0, 0, 0, 0, 0 };
 | 
						|
	ullong all = 0;
 | 
						|
	int norm_all = 0;
 | 
						|
	int bar_sz = s->bar_sz;
 | 
						|
	char *bar = s->bar;
 | 
						|
	int i;
 | 
						|
 | 
						|
	if (rdval(get_file(&proc_stat), "cpu ", data, 0
 | 
						|
	    | (1 << 1)
 | 
						|
	    | (1 << 2)
 | 
						|
	    | (1 << 3)
 | 
						|
	    | (1 << 4)
 | 
						|
	    | (1 << 5)
 | 
						|
	    | (1 << 6)
 | 
						|
	    | (1 << 7))
 | 
						|
	) {
 | 
						|
		put_question_marks(bar_sz);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	for (i = 0; i < CPU_FIELDCNT; i++) {
 | 
						|
		ullong old = s->old[i];
 | 
						|
		if (data[i] < old) old = data[i];		//sanitize
 | 
						|
		s->old[i] = data[i];
 | 
						|
		all += (data[i] -= old);
 | 
						|
	}
 | 
						|
 | 
						|
	if (all) {
 | 
						|
		for (i = 0; i < CPU_FIELDCNT; i++) {
 | 
						|
			ullong t = bar_sz * data[i];
 | 
						|
			norm_all += data[i] = t / all;
 | 
						|
			frac[i] = t % all;
 | 
						|
		}
 | 
						|
 | 
						|
		while (norm_all < bar_sz) {
 | 
						|
			unsigned max = frac[0];
 | 
						|
			int pos = 0;
 | 
						|
			for (i = 1; i < CPU_FIELDCNT; i++) {
 | 
						|
				if (frac[i] > max) max = frac[i], pos = i;
 | 
						|
			}
 | 
						|
			frac[pos] = 0;	//avoid bumping up same value twice
 | 
						|
			data[pos]++;
 | 
						|
			norm_all++;
 | 
						|
		}
 | 
						|
 | 
						|
		memset(bar, '.', bar_sz);
 | 
						|
		memset(bar, 'S', data[2]); bar += data[2]; //sys
 | 
						|
		memset(bar, 'U', data[0]); bar += data[0]; //usr
 | 
						|
		memset(bar, 'N', data[1]); bar += data[1]; //nice
 | 
						|
		memset(bar, 'D', data[4]); bar += data[4]; //iowait
 | 
						|
		memset(bar, 'I', data[5]); bar += data[5]; //irq
 | 
						|
		memset(bar, 'i', data[6]); bar += data[6]; //softirq
 | 
						|
	} else {
 | 
						|
		memset(bar, '?', bar_sz);
 | 
						|
	}
 | 
						|
	put(s->bar);
 | 
						|
}
 | 
						|
 | 
						|
static s_stat* init_cpu(const char *param)
 | 
						|
{
 | 
						|
	int sz;
 | 
						|
	cpu_stat *s;
 | 
						|
	sz = strtoul(param, NULL, 0); /* param can be "" */
 | 
						|
	if (sz < 10) sz = 10;
 | 
						|
	if (sz > 1000) sz = 1000;
 | 
						|
	s = xzalloc(sizeof(*s) + sz);
 | 
						|
	/*s->bar[sz] = '\0'; - xzalloc did it */
 | 
						|
	s->bar_sz = sz;
 | 
						|
	s->collect = collect_cpu;
 | 
						|
	return (s_stat*)s;
 | 
						|
}
 | 
						|
 | 
						|
S_STAT(int_stat)
 | 
						|
	ullong old;
 | 
						|
	int no;
 | 
						|
S_STAT_END(int_stat)
 | 
						|
 | 
						|
static void FAST_FUNC collect_int(int_stat *s)
 | 
						|
{
 | 
						|
	ullong data[1];
 | 
						|
	ullong old;
 | 
						|
 | 
						|
	if (rdval(get_file(&proc_stat), "intr", data, 1 << s->no)) {
 | 
						|
		put_question_marks(4);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	old = s->old;
 | 
						|
	if (data[0] < old) old = data[0];		//sanitize
 | 
						|
	s->old = data[0];
 | 
						|
	scale(data[0] - old);
 | 
						|
}
 | 
						|
 | 
						|
static s_stat* init_int(const char *param)
 | 
						|
{
 | 
						|
	int_stat *s = xzalloc(sizeof(*s));
 | 
						|
	s->collect = collect_int;
 | 
						|
	if (param[0] == '\0') {
 | 
						|
		s->no = 1;
 | 
						|
	} else {
 | 
						|
		int n = xatoi_positive(param);
 | 
						|
		s->no = n + 2;
 | 
						|
	}
 | 
						|
	return (s_stat*)s;
 | 
						|
}
 | 
						|
 | 
						|
S_STAT(ctx_stat)
 | 
						|
	ullong old;
 | 
						|
S_STAT_END(ctx_stat)
 | 
						|
 | 
						|
static void FAST_FUNC collect_ctx(ctx_stat *s)
 | 
						|
{
 | 
						|
	ullong data[1];
 | 
						|
	ullong old;
 | 
						|
 | 
						|
	if (rdval(get_file(&proc_stat), "ctxt", data, 1 << 1)) {
 | 
						|
		put_question_marks(4);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	old = s->old;
 | 
						|
	if (data[0] < old) old = data[0];		//sanitize
 | 
						|
	s->old = data[0];
 | 
						|
	scale(data[0] - old);
 | 
						|
}
 | 
						|
 | 
						|
static s_stat* init_ctx(const char *param UNUSED_PARAM)
 | 
						|
{
 | 
						|
	ctx_stat *s = xzalloc(sizeof(*s));
 | 
						|
	s->collect = collect_ctx;
 | 
						|
	return (s_stat*)s;
 | 
						|
}
 | 
						|
 | 
						|
S_STAT(blk_stat)
 | 
						|
	const char* lookfor;
 | 
						|
	ullong old[2];
 | 
						|
S_STAT_END(blk_stat)
 | 
						|
 | 
						|
static void FAST_FUNC collect_blk(blk_stat *s)
 | 
						|
{
 | 
						|
	ullong data[2];
 | 
						|
	int i;
 | 
						|
 | 
						|
	if (is26) {
 | 
						|
		i = rdval_diskstats(get_file(&proc_diskstats), data);
 | 
						|
	} else {
 | 
						|
		i = rdval(get_file(&proc_stat), s->lookfor, data, 0
 | 
						|
				| (1 << 1)
 | 
						|
				| (1 << 2)
 | 
						|
		);
 | 
						|
		// Linux 2.4 reports bio in Kbytes, convert to sectors:
 | 
						|
		data[0] *= 2;
 | 
						|
		data[1] *= 2;
 | 
						|
	}
 | 
						|
	if (i) {
 | 
						|
		put_question_marks(9);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	for (i=0; i<2; i++) {
 | 
						|
		ullong old = s->old[i];
 | 
						|
		if (data[i] < old) old = data[i];		//sanitize
 | 
						|
		s->old[i] = data[i];
 | 
						|
		data[i] -= old;
 | 
						|
	}
 | 
						|
	scale(data[0]*512); // TODO: *sectorsize
 | 
						|
	put_c(' ');
 | 
						|
	scale(data[1]*512);
 | 
						|
}
 | 
						|
 | 
						|
static s_stat* init_blk(const char *param UNUSED_PARAM)
 | 
						|
{
 | 
						|
	blk_stat *s = xzalloc(sizeof(*s));
 | 
						|
	s->collect = collect_blk;
 | 
						|
	s->lookfor = "page";
 | 
						|
	return (s_stat*)s;
 | 
						|
}
 | 
						|
 | 
						|
S_STAT(fork_stat)
 | 
						|
	ullong old;
 | 
						|
S_STAT_END(fork_stat)
 | 
						|
 | 
						|
static void FAST_FUNC collect_thread_nr(fork_stat *s UNUSED_PARAM)
 | 
						|
{
 | 
						|
	ullong data[1];
 | 
						|
 | 
						|
	if (rdval_loadavg(get_file(&proc_loadavg), data, 1 << 4)) {
 | 
						|
		put_question_marks(4);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
	scale(data[0]);
 | 
						|
}
 | 
						|
 | 
						|
static void FAST_FUNC collect_fork(fork_stat *s)
 | 
						|
{
 | 
						|
	ullong data[1];
 | 
						|
	ullong old;
 | 
						|
 | 
						|
	if (rdval(get_file(&proc_stat), "processes", data, 1 << 1)) {
 | 
						|
		put_question_marks(4);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	old = s->old;
 | 
						|
	if (data[0] < old) old = data[0];	//sanitize
 | 
						|
	s->old = data[0];
 | 
						|
	scale(data[0] - old);
 | 
						|
}
 | 
						|
 | 
						|
static s_stat* init_fork(const char *param)
 | 
						|
{
 | 
						|
	fork_stat *s = xzalloc(sizeof(*s));
 | 
						|
	if (*param == 'n') {
 | 
						|
		s->collect = collect_thread_nr;
 | 
						|
	} else {
 | 
						|
		s->collect = collect_fork;
 | 
						|
	}
 | 
						|
	return (s_stat*)s;
 | 
						|
}
 | 
						|
 | 
						|
S_STAT(if_stat)
 | 
						|
	ullong old[4];
 | 
						|
	const char *device;
 | 
						|
	char *device_colon;
 | 
						|
S_STAT_END(if_stat)
 | 
						|
 | 
						|
static void FAST_FUNC collect_if(if_stat *s)
 | 
						|
{
 | 
						|
	ullong data[4];
 | 
						|
	int i;
 | 
						|
 | 
						|
	if (rdval(get_file(&proc_net_dev), s->device_colon, data, 0
 | 
						|
	    | (1 << 1)
 | 
						|
	    | (1 << 3)
 | 
						|
	    | (1 << 9)
 | 
						|
	    | (1 << 11))
 | 
						|
	) {
 | 
						|
		put_question_marks(10);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	for (i=0; i<4; i++) {
 | 
						|
		ullong old = s->old[i];
 | 
						|
		if (data[i] < old) old = data[i];		//sanitize
 | 
						|
		s->old[i] = data[i];
 | 
						|
		data[i] -= old;
 | 
						|
	}
 | 
						|
	put_c(data[1] ? '*' : ' ');
 | 
						|
	scale(data[0]);
 | 
						|
	put_c(data[3] ? '*' : ' ');
 | 
						|
	scale(data[2]);
 | 
						|
}
 | 
						|
 | 
						|
static s_stat* init_if(const char *device)
 | 
						|
{
 | 
						|
	if_stat *s = xzalloc(sizeof(*s));
 | 
						|
 | 
						|
	if (!device || !device[0])
 | 
						|
		bb_show_usage();
 | 
						|
	s->collect = collect_if;
 | 
						|
 | 
						|
	s->device = device;
 | 
						|
	s->device_colon = xasprintf("%s:", device);
 | 
						|
	return (s_stat*)s;
 | 
						|
}
 | 
						|
 | 
						|
S_STAT(mem_stat)
 | 
						|
	char opt;
 | 
						|
S_STAT_END(mem_stat)
 | 
						|
 | 
						|
// "Memory" value should not include any caches.
 | 
						|
// IOW: neither "ls -laR /" nor heavy read/write activity
 | 
						|
//      should affect it. We'd like to also include any
 | 
						|
//      long-term allocated kernel-side mem, but it is hard
 | 
						|
//      to figure out. For now, bufs, cached & slab are
 | 
						|
//      counted as "free" memory
 | 
						|
//2.6.16:
 | 
						|
//MemTotal:       773280 kB
 | 
						|
//MemFree:         25912 kB - genuinely free
 | 
						|
//Buffers:        320672 kB - cache
 | 
						|
//Cached:         146396 kB - cache
 | 
						|
//SwapCached:          0 kB
 | 
						|
//Active:         183064 kB
 | 
						|
//Inactive:       356892 kB
 | 
						|
//HighTotal:           0 kB
 | 
						|
//HighFree:            0 kB
 | 
						|
//LowTotal:       773280 kB
 | 
						|
//LowFree:         25912 kB
 | 
						|
//SwapTotal:      131064 kB
 | 
						|
//SwapFree:       131064 kB
 | 
						|
//Dirty:              48 kB
 | 
						|
//Writeback:           0 kB
 | 
						|
//Mapped:          96620 kB
 | 
						|
//Slab:           200668 kB - takes 7 Mb on my box fresh after boot,
 | 
						|
//                            but includes dentries and inodes
 | 
						|
//                            (== can take arbitrary amount of mem)
 | 
						|
//CommitLimit:    517704 kB
 | 
						|
//Committed_AS:   236776 kB
 | 
						|
//PageTables:       1248 kB
 | 
						|
//VmallocTotal:   516052 kB
 | 
						|
//VmallocUsed:      3852 kB
 | 
						|
//VmallocChunk:   512096 kB
 | 
						|
//HugePages_Total:     0
 | 
						|
//HugePages_Free:      0
 | 
						|
//Hugepagesize:     4096 kB
 | 
						|
static void FAST_FUNC collect_mem(mem_stat *s)
 | 
						|
{
 | 
						|
	ullong m_total = 0;
 | 
						|
	ullong m_free = 0;
 | 
						|
	ullong m_bufs = 0;
 | 
						|
	ullong m_cached = 0;
 | 
						|
	ullong m_slab = 0;
 | 
						|
 | 
						|
	if (rdval(get_file(&proc_meminfo), "MemTotal:", &m_total, 1 << 1)) {
 | 
						|
		put_question_marks(4);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
	if (s->opt == 't') {
 | 
						|
		scale(m_total << 10);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	if (rdval(proc_meminfo.file, "MemFree:", &m_free  , 1 << 1)
 | 
						|
	 || rdval(proc_meminfo.file, "Buffers:", &m_bufs  , 1 << 1)
 | 
						|
	 || rdval(proc_meminfo.file, "Cached:",  &m_cached, 1 << 1)
 | 
						|
	 || rdval(proc_meminfo.file, "Slab:",    &m_slab  , 1 << 1)
 | 
						|
	) {
 | 
						|
		put_question_marks(4);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	m_free += m_bufs + m_cached + m_slab;
 | 
						|
	switch (s->opt) {
 | 
						|
	case 'f':
 | 
						|
		scale(m_free << 10); break;
 | 
						|
	default:
 | 
						|
		scale((m_total - m_free) << 10); break;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static s_stat* init_mem(const char *param)
 | 
						|
{
 | 
						|
	mem_stat *s = xzalloc(sizeof(*s));
 | 
						|
	s->collect = collect_mem;
 | 
						|
	s->opt = param[0];
 | 
						|
	return (s_stat*)s;
 | 
						|
}
 | 
						|
 | 
						|
S_STAT(swp_stat)
 | 
						|
S_STAT_END(swp_stat)
 | 
						|
 | 
						|
static void FAST_FUNC collect_swp(swp_stat *s UNUSED_PARAM)
 | 
						|
{
 | 
						|
	ullong s_total[1];
 | 
						|
	ullong s_free[1];
 | 
						|
	if (rdval(get_file(&proc_meminfo), "SwapTotal:", s_total, 1 << 1)
 | 
						|
	 || rdval(proc_meminfo.file,       "SwapFree:" , s_free,  1 << 1)
 | 
						|
	) {
 | 
						|
		put_question_marks(4);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
	scale((s_total[0]-s_free[0]) << 10);
 | 
						|
}
 | 
						|
 | 
						|
static s_stat* init_swp(const char *param UNUSED_PARAM)
 | 
						|
{
 | 
						|
	swp_stat *s = xzalloc(sizeof(*s));
 | 
						|
	s->collect = collect_swp;
 | 
						|
	return (s_stat*)s;
 | 
						|
}
 | 
						|
 | 
						|
S_STAT(fd_stat)
 | 
						|
S_STAT_END(fd_stat)
 | 
						|
 | 
						|
static void FAST_FUNC collect_fd(fd_stat *s UNUSED_PARAM)
 | 
						|
{
 | 
						|
	ullong data[2];
 | 
						|
 | 
						|
	if (rdval(get_file(&proc_sys_fs_filenr), "", data, 0
 | 
						|
	    | (1 << 1)
 | 
						|
	    | (1 << 2))
 | 
						|
	) {
 | 
						|
		put_question_marks(4);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	scale(data[0] - data[1]);
 | 
						|
}
 | 
						|
 | 
						|
static s_stat* init_fd(const char *param UNUSED_PARAM)
 | 
						|
{
 | 
						|
	fd_stat *s = xzalloc(sizeof(*s));
 | 
						|
	s->collect = collect_fd;
 | 
						|
	return (s_stat*)s;
 | 
						|
}
 | 
						|
 | 
						|
S_STAT(time_stat)
 | 
						|
	unsigned prec;
 | 
						|
	unsigned scale;
 | 
						|
S_STAT_END(time_stat)
 | 
						|
 | 
						|
static void FAST_FUNC collect_time(time_stat *s)
 | 
						|
{
 | 
						|
	char buf[sizeof("12:34:56.123456")];
 | 
						|
	struct tm* tm;
 | 
						|
	unsigned us = tv.tv_usec + s->scale/2;
 | 
						|
	time_t t = tv.tv_sec;
 | 
						|
 | 
						|
	if (us >= 1000000) {
 | 
						|
		t++;
 | 
						|
		us -= 1000000;
 | 
						|
	}
 | 
						|
	tm = localtime(&t);
 | 
						|
 | 
						|
	sprintf(buf, "%02d:%02d:%02d", tm->tm_hour, tm->tm_min, tm->tm_sec);
 | 
						|
	if (s->prec)
 | 
						|
		sprintf(buf+8, ".%0*d", s->prec, us / s->scale);
 | 
						|
	put(buf);
 | 
						|
}
 | 
						|
 | 
						|
static s_stat* init_time(const char *param)
 | 
						|
{
 | 
						|
	int prec;
 | 
						|
	time_stat *s = xzalloc(sizeof(*s));
 | 
						|
 | 
						|
	s->collect = collect_time;
 | 
						|
	prec = param[0] - '0';
 | 
						|
	if (prec < 0) prec = 0;
 | 
						|
	else if (prec > 6) prec = 6;
 | 
						|
	s->prec = prec;
 | 
						|
	s->scale = 1;
 | 
						|
	while (prec++ < 6)
 | 
						|
		s->scale *= 10;
 | 
						|
	return (s_stat*)s;
 | 
						|
}
 | 
						|
 | 
						|
static void FAST_FUNC collect_info(s_stat *s)
 | 
						|
{
 | 
						|
	gen ^= 1;
 | 
						|
	while (s) {
 | 
						|
		put(s->label);
 | 
						|
		s->collect(s);
 | 
						|
		s = s->next;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
typedef s_stat* init_func(const char *param);
 | 
						|
 | 
						|
static const char options[] ALIGN1 = "ncmsfixptbr";
 | 
						|
static init_func *const init_functions[] = {
 | 
						|
	init_if,
 | 
						|
	init_cpu,
 | 
						|
	init_mem,
 | 
						|
	init_swp,
 | 
						|
	init_fd,
 | 
						|
	init_int,
 | 
						|
	init_ctx,
 | 
						|
	init_fork,
 | 
						|
	init_time,
 | 
						|
	init_blk,
 | 
						|
	init_cr
 | 
						|
};
 | 
						|
 | 
						|
int nmeter_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 | 
						|
int nmeter_main(int argc UNUSED_PARAM, char **argv)
 | 
						|
{
 | 
						|
	char buf[32];
 | 
						|
	s_stat *first = NULL;
 | 
						|
	s_stat *last = NULL;
 | 
						|
	s_stat *s;
 | 
						|
	char *opt_d;
 | 
						|
	char *cur, *prev;
 | 
						|
 | 
						|
	INIT_G();
 | 
						|
 | 
						|
	xchdir("/proc");
 | 
						|
 | 
						|
	if (open_read_close("version", buf, sizeof(buf)-1) > 0) {
 | 
						|
		buf[sizeof(buf)-1] = '\0';
 | 
						|
		is26 = (strstr(buf, " 2.4.") == NULL);
 | 
						|
	}
 | 
						|
 | 
						|
	if (getopt32(argv, "d:", &opt_d)) {
 | 
						|
		G.delta = xatoi(opt_d) * 1000;
 | 
						|
		G.deltanz = G.delta > 0 ? G.delta : 1;
 | 
						|
		need_seconds = (1000000 % G.deltanz) != 0;
 | 
						|
	}
 | 
						|
	argv += optind;
 | 
						|
 | 
						|
	if (!argv[0])
 | 
						|
		bb_show_usage();
 | 
						|
 | 
						|
	// Can use argv[0] directly, but this will mess up
 | 
						|
	// parameters as seen by e.g. ps. Making a copy...
 | 
						|
	cur = xstrdup(argv[0]);
 | 
						|
	while (1) {
 | 
						|
		char *param, *p;
 | 
						|
		prev = cur;
 | 
						|
 again:
 | 
						|
		cur = strchr(cur, '%');
 | 
						|
		if (!cur)
 | 
						|
			break;
 | 
						|
		if (cur[1] == '%') {	// %%
 | 
						|
			overlapping_strcpy(cur, cur + 1);
 | 
						|
			cur++;
 | 
						|
			goto again;
 | 
						|
		}
 | 
						|
		*cur++ = '\0';		// overwrite %
 | 
						|
		if (cur[0] == '[') {
 | 
						|
			// format: %[foptstring]
 | 
						|
			cur++;
 | 
						|
			p = strchr(options, cur[0]);
 | 
						|
			param = cur+1;
 | 
						|
			while (cur[0] != ']') {
 | 
						|
				if (!cur[0])
 | 
						|
					bb_show_usage();
 | 
						|
				cur++;
 | 
						|
			}
 | 
						|
			*cur++ = '\0';	// overwrite [
 | 
						|
		} else {
 | 
						|
			// format: %NNNNNNf
 | 
						|
			param = cur;
 | 
						|
			while (cur[0] >= '0' && cur[0] <= '9')
 | 
						|
				cur++;
 | 
						|
			if (!cur[0])
 | 
						|
				bb_show_usage();
 | 
						|
			p = strchr(options, cur[0]);
 | 
						|
			*cur++ = '\0';	// overwrite format char
 | 
						|
		}
 | 
						|
		if (!p)
 | 
						|
			bb_show_usage();
 | 
						|
		s = init_functions[p-options](param);
 | 
						|
		if (s) {
 | 
						|
			s->label = prev;
 | 
						|
			/*s->next = NULL; - all initXXX funcs use xzalloc */
 | 
						|
			if (!first)
 | 
						|
				first = s;
 | 
						|
			else
 | 
						|
				last->next = s;
 | 
						|
			last = s;
 | 
						|
		} else {
 | 
						|
			// %r option. remove it from string
 | 
						|
			overlapping_strcpy(prev + strlen(prev), cur);
 | 
						|
			cur = prev;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	if (prev[0]) {
 | 
						|
		s = init_literal();
 | 
						|
		s->label = prev;
 | 
						|
		/*s->next = NULL; - all initXXX funcs use xzalloc */
 | 
						|
		if (!first)
 | 
						|
			first = s;
 | 
						|
		else
 | 
						|
			last->next = s;
 | 
						|
		last = s;
 | 
						|
	}
 | 
						|
 | 
						|
	// Generate first samples but do not print them, they're bogus
 | 
						|
	collect_info(first);
 | 
						|
	reset_outbuf();
 | 
						|
	if (G.delta >= 0) {
 | 
						|
		gettimeofday(&tv, NULL);
 | 
						|
		usleep(G.delta > 1000000 ? 1000000 : G.delta - tv.tv_usec % G.deltanz);
 | 
						|
	}
 | 
						|
 | 
						|
	while (1) {
 | 
						|
		gettimeofday(&tv, NULL);
 | 
						|
		collect_info(first);
 | 
						|
		put_c(G.final_char);
 | 
						|
		print_outbuf();
 | 
						|
 | 
						|
		// Negative delta -> no usleep at all
 | 
						|
		// This will hog the CPU but you can have REALLY GOOD
 | 
						|
		// time resolution ;)
 | 
						|
		// TODO: detect and avoid useless updates
 | 
						|
		// (like: nothing happens except time)
 | 
						|
		if (G.delta >= 0) {
 | 
						|
			int rem;
 | 
						|
			// can be commented out, will sacrifice sleep time precision a bit
 | 
						|
			gettimeofday(&tv, NULL);
 | 
						|
			if (need_seconds)
 | 
						|
				rem = G.delta - ((ullong)tv.tv_sec*1000000 + tv.tv_usec) % G.deltanz;
 | 
						|
			else
 | 
						|
				rem = G.delta - (unsigned)tv.tv_usec % G.deltanz;
 | 
						|
			// Sometimes kernel wakes us up just a tiny bit earlier than asked
 | 
						|
			// Do not go to very short sleep in this case
 | 
						|
			if (rem < (unsigned)G.delta / 128) {
 | 
						|
				rem += G.delta;
 | 
						|
			}
 | 
						|
			usleep(rem);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	/*return 0;*/
 | 
						|
}
 |