blob: 7836a90d51f2ea6e62c9c4db5c049d71f36b9b51 [file] [log] [blame]
Denis Vlasenko1db39b22006-10-11 20:59:02 +00001/*
Denys Vlasenko0ef64bd2010-08-16 20:14:46 +02002 * Licensed under GPLv2, see file LICENSE in this source tree.
3 *
4 * Based on nanotop.c from floppyfw project
5 *
Denys Vlasenkobdaea462010-08-16 20:42:14 +02006 * Contact me: vda.linux@googlemail.com
Denys Vlasenko0ef64bd2010-08-16 20:14:46 +02007 */
Denis Vlasenko1db39b22006-10-11 20:59:02 +00008
9//TODO:
10// simplify code
11// /proc/locks
12// /proc/stat:
13// disk_io: (3,0):(22272,17897,410702,4375,54750)
14// btime 1059401962
Denis Vlasenkoa6163ca2007-06-17 00:35:15 +000015//TODO: use sysinfo libc call/syscall, if appropriate
16// (faster than open/read/close):
17// sysinfo({uptime=15017, loads=[5728, 15040, 16480]
18// totalram=2107416576, freeram=211525632, sharedram=0, bufferram=157204480}
19// totalswap=134209536, freeswap=134209536, procs=157})
Denis Vlasenko1db39b22006-10-11 20:59:02 +000020
Denis Vlasenkoa6163ca2007-06-17 00:35:15 +000021#include "libbb.h"
Denis Vlasenko1db39b22006-10-11 20:59:02 +000022
23typedef unsigned long long ullong;
Denis Vlasenko1db39b22006-10-11 20:59:02 +000024
Denis Vlasenkobfc93a32008-09-26 11:12:09 +000025enum { /* Preferably use powers of 2 */
26 PROC_MIN_FILE_SIZE = 256,
27 PROC_MAX_FILE_SIZE = 16 * 1024,
28};
Denis Vlasenko1db39b22006-10-11 20:59:02 +000029
30typedef struct proc_file {
Denis Vlasenko1db39b22006-10-11 20:59:02 +000031 char *file;
Denis Vlasenkobfc93a32008-09-26 11:12:09 +000032 int file_sz;
Denis Vlasenko199c0d52007-05-30 14:48:38 +000033 smallint last_gen;
Denis Vlasenko1db39b22006-10-11 20:59:02 +000034} proc_file;
35
Denis Vlasenko199c0d52007-05-30 14:48:38 +000036static const char *const proc_name[] = {
37 "stat", // Must match the order of proc_file's!
38 "loadavg",
39 "net/dev",
40 "meminfo",
41 "diskstats",
Denis Vlasenko6ca409e2007-08-12 20:58:27 +000042 "sys/fs/file-nr"
Denis Vlasenko199c0d52007-05-30 14:48:38 +000043};
44
45struct globals {
46 // Sample generation flip-flop
47 smallint gen;
48 // Linux 2.6? (otherwise assumes 2.4)
49 smallint is26;
50 // 1 if sample delay is not an integer fraction of a second
51 smallint need_seconds;
52 char *cur_outbuf;
53 const char *final_str;
54 int delta;
55 int deltanz;
56 struct timeval tv;
57#define first_proc_file proc_stat
58 proc_file proc_stat; // Must match the order of proc_name's!
59 proc_file proc_loadavg;
60 proc_file proc_net_dev;
61 proc_file proc_meminfo;
62 proc_file proc_diskstats;
63 proc_file proc_sys_fs_filenr;
64};
65#define G (*ptr_to_globals)
66#define gen (G.gen )
67#define is26 (G.is26 )
68#define need_seconds (G.need_seconds )
69#define cur_outbuf (G.cur_outbuf )
70#define final_str (G.final_str )
71#define delta (G.delta )
72#define deltanz (G.deltanz )
73#define tv (G.tv )
74#define proc_stat (G.proc_stat )
75#define proc_loadavg (G.proc_loadavg )
76#define proc_net_dev (G.proc_net_dev )
77#define proc_meminfo (G.proc_meminfo )
78#define proc_diskstats (G.proc_diskstats )
79#define proc_sys_fs_filenr (G.proc_sys_fs_filenr)
Denis Vlasenko199c0d52007-05-30 14:48:38 +000080#define INIT_G() do { \
Denis Vlasenko574f2f42008-02-27 18:41:59 +000081 SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
82 cur_outbuf = outbuf; \
83 final_str = "\n"; \
84 deltanz = delta = 1000000; \
85} while (0)
Denis Vlasenko1db39b22006-10-11 20:59:02 +000086
Denis Vlasenkoef4bb262007-06-04 12:21:53 +000087// We depend on this being a char[], not char* - we take sizeof() of it
88#define outbuf bb_common_bufsiz1
89
Denis Vlasenko1db39b22006-10-11 20:59:02 +000090static inline void reset_outbuf(void)
91{
92 cur_outbuf = outbuf;
93}
94
95static inline int outbuf_count(void)
96{
97 return cur_outbuf - outbuf;
98}
99
100static void print_outbuf(void)
101{
102 int sz = cur_outbuf - outbuf;
103 if (sz > 0) {
Bernhard Reutner-Fischer5e25ddb2008-05-19 09:48:17 +0000104 xwrite(STDOUT_FILENO, outbuf, sz);
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000105 cur_outbuf = outbuf;
106 }
107}
108
109static void put(const char *s)
110{
111 int sz = strlen(s);
112 if (sz > outbuf + sizeof(outbuf) - cur_outbuf)
113 sz = outbuf + sizeof(outbuf) - cur_outbuf;
114 memcpy(cur_outbuf, s, sz);
115 cur_outbuf += sz;
116}
117
118static void put_c(char c)
119{
120 if (cur_outbuf < outbuf + sizeof(outbuf))
121 *cur_outbuf++ = c;
122}
123
124static void put_question_marks(int count)
125{
126 while (count--)
127 put_c('?');
128}
129
Denis Vlasenkobfc93a32008-09-26 11:12:09 +0000130static void readfile_z(proc_file *pf, const char* fname)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000131{
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000132// open_read_close() will do two reads in order to be sure we are at EOF,
133// and we don't need/want that.
Denis Vlasenkobfc93a32008-09-26 11:12:09 +0000134 int fd;
135 int sz, rdsz;
136 char *buf;
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000137
Denis Vlasenkobfc93a32008-09-26 11:12:09 +0000138 sz = pf->file_sz;
139 buf = pf->file;
140 if (!buf) {
141 buf = xmalloc(PROC_MIN_FILE_SIZE);
142 sz = PROC_MIN_FILE_SIZE;
143 }
144 again:
145 fd = xopen(fname, O_RDONLY);
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000146 buf[0] = '\0';
Denis Vlasenkobfc93a32008-09-26 11:12:09 +0000147 rdsz = read(fd, buf, sz-1);
Denis Vlasenkobc2fd372008-06-30 07:33:52 +0000148 close(fd);
Denis Vlasenkobfc93a32008-09-26 11:12:09 +0000149 if (rdsz > 0) {
150 if (rdsz == sz-1 && sz < PROC_MAX_FILE_SIZE) {
151 sz *= 2;
152 buf = xrealloc(buf, sz);
153 goto again;
154 }
155 buf[rdsz] = '\0';
156 }
157 pf->file_sz = sz;
158 pf->file = buf;
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000159}
160
161static const char* get_file(proc_file *pf)
162{
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000163 if (pf->last_gen != gen) {
164 pf->last_gen = gen;
Denis Vlasenkobfc93a32008-09-26 11:12:09 +0000165 readfile_z(pf, proc_name[pf - &first_proc_file]);
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000166 }
167 return pf->file;
168}
169
Denis Vlasenkodadfb492008-07-29 10:16:05 +0000170static ullong read_after_slash(const char *p)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000171{
172 p = strchr(p, '/');
173 if (!p) return 0;
174 return strtoull(p+1, NULL, 10);
175}
176
177enum conv_type { conv_decimal, conv_slash };
178
179// Reads decimal values from line. Values start after key, for example:
180// "cpu 649369 0 341297 4336769..." - key is "cpu" here.
181// Values are stored in vec[]. arg_ptr has list of positions
182// we are interested in: for example: 1,2,5 - we want 1st, 2nd and 5th value.
183static int vrdval(const char* p, const char* key,
184 enum conv_type conv, ullong *vec, va_list arg_ptr)
185{
186 int indexline;
187 int indexnext;
188
189 p = strstr(p, key);
190 if (!p) return 1;
191
192 p += strlen(key);
193 indexline = 1;
194 indexnext = va_arg(arg_ptr, int);
195 while (1) {
196 while (*p == ' ' || *p == '\t') p++;
197 if (*p == '\n' || *p == '\0') break;
198
199 if (indexline == indexnext) { // read this value
200 *vec++ = conv==conv_decimal ?
201 strtoull(p, NULL, 10) :
202 read_after_slash(p);
203 indexnext = va_arg(arg_ptr, int);
204 }
205 while (*p > ' ') p++; // skip over value
206 indexline++;
207 }
208 return 0;
209}
210
211// Parses files with lines like "cpu0 21727 0 15718 1813856 9461 10485 0 0":
212// rdval(file_contents, "string_to_find", result_vector, value#, value#...)
213// value# start with 1
214static int rdval(const char* p, const char* key, ullong *vec, ...)
215{
216 va_list arg_ptr;
217 int result;
218
219 va_start(arg_ptr, vec);
220 result = vrdval(p, key, conv_decimal, vec, arg_ptr);
221 va_end(arg_ptr);
222
223 return result;
224}
225
226// Parses files with lines like "... ... ... 3/148 ...."
227static int rdval_loadavg(const char* p, ullong *vec, ...)
228{
229 va_list arg_ptr;
230 int result;
231
232 va_start(arg_ptr, vec);
233 result = vrdval(p, "", conv_slash, vec, arg_ptr);
234 va_end(arg_ptr);
235
236 return result;
237}
238
239// Parses /proc/diskstats
240// 1 2 3 4 5 6(rd) 7 8 9 10(wr) 11 12 13 14
241// 3 0 hda 51292 14441 841783 926052 25717 79650 843256 3029804 0 148459 3956933
242// 3 1 hda1 0 0 0 0 <- ignore if only 4 fields
243static int rdval_diskstats(const char* p, ullong *vec)
244{
Denis Vlasenko259c9a62008-06-28 23:42:03 +0000245 ullong rd = rd; // for compiler
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000246 int indexline = 0;
247 vec[0] = 0;
248 vec[1] = 0;
249 while (1) {
250 indexline++;
251 while (*p == ' ' || *p == '\t') p++;
252 if (*p == '\0') break;
253 if (*p == '\n') {
254 indexline = 0;
255 p++;
256 continue;
257 }
258 if (indexline == 6) {
259 rd = strtoull(p, NULL, 10);
260 } else if (indexline == 10) {
261 vec[0] += rd; // TODO: *sectorsize (don't know how to find out sectorsize)
262 vec[1] += strtoull(p, NULL, 10);
263 while (*p != '\n' && *p != '\0') p++;
264 continue;
265 }
266 while (*p > ' ') p++; // skip over value
267 }
268 return 0;
269}
270
271static void scale(ullong ul)
272{
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000273 char buf[5];
Denis Vlasenko56ea65c2008-01-06 03:26:53 +0000274
275 /* see http://en.wikipedia.org/wiki/Tera */
276 smart_ulltoa4(ul, buf, " kmgtpezy");
277 buf[4] = '\0';
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000278 put(buf);
279}
280
281
282#define S_STAT(a) \
283typedef struct a { \
284 struct s_stat *next; \
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200285 void (*collect)(struct a *s) FAST_FUNC; \
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000286 const char *label;
287#define S_STAT_END(a) } a;
288
289S_STAT(s_stat)
290S_STAT_END(s_stat)
291
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200292static void FAST_FUNC collect_literal(s_stat *s UNUSED_PARAM)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000293{
294}
295
296static s_stat* init_literal(void)
297{
Denis Vlasenkobefd5ea2008-06-28 23:33:02 +0000298 s_stat *s = xzalloc(sizeof(*s));
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000299 s->collect = collect_literal;
300 return (s_stat*)s;
301}
302
303static s_stat* init_delay(const char *param)
304{
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000305 delta = strtoul(param, NULL, 0) * 1000; /* param can be "" */
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000306 deltanz = delta > 0 ? delta : 1;
307 need_seconds = (1000000%deltanz) != 0;
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000308 return NULL;
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000309}
310
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000311static s_stat* init_cr(const char *param UNUSED_PARAM)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000312{
313 final_str = "\r";
314 return (s_stat*)0;
315}
316
317
318// user nice system idle iowait irq softirq (last 3 only in 2.6)
319//cpu 649369 0 341297 4336769 11640 7122 1183
320//cpuN 649369 0 341297 4336769 11640 7122 1183
321enum { CPU_FIELDCNT = 7 };
322S_STAT(cpu_stat)
323 ullong old[CPU_FIELDCNT];
324 int bar_sz;
325 char *bar;
326S_STAT_END(cpu_stat)
327
328
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200329static void FAST_FUNC collect_cpu(cpu_stat *s)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000330{
331 ullong data[CPU_FIELDCNT] = { 0, 0, 0, 0, 0, 0, 0 };
332 unsigned frac[CPU_FIELDCNT] = { 0, 0, 0, 0, 0, 0, 0 };
333 ullong all = 0;
334 int norm_all = 0;
335 int bar_sz = s->bar_sz;
336 char *bar = s->bar;
337 int i;
338
339 if (rdval(get_file(&proc_stat), "cpu ", data, 1, 2, 3, 4, 5, 6, 7)) {
340 put_question_marks(bar_sz);
341 return;
342 }
343
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000344 for (i = 0; i < CPU_FIELDCNT; i++) {
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000345 ullong old = s->old[i];
346 if (data[i] < old) old = data[i]; //sanitize
347 s->old[i] = data[i];
348 all += (data[i] -= old);
349 }
350
351 if (all) {
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000352 for (i = 0; i < CPU_FIELDCNT; i++) {
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000353 ullong t = bar_sz * data[i];
354 norm_all += data[i] = t / all;
355 frac[i] = t % all;
356 }
357
358 while (norm_all < bar_sz) {
359 unsigned max = frac[0];
360 int pos = 0;
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000361 for (i = 1; i < CPU_FIELDCNT; i++) {
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000362 if (frac[i] > max) max = frac[i], pos = i;
363 }
364 frac[pos] = 0; //avoid bumping up same value twice
365 data[pos]++;
366 norm_all++;
367 }
368
369 memset(bar, '.', bar_sz);
370 memset(bar, 'S', data[2]); bar += data[2]; //sys
371 memset(bar, 'U', data[0]); bar += data[0]; //usr
372 memset(bar, 'N', data[1]); bar += data[1]; //nice
373 memset(bar, 'D', data[4]); bar += data[4]; //iowait
374 memset(bar, 'I', data[5]); bar += data[5]; //irq
375 memset(bar, 'i', data[6]); bar += data[6]; //softirq
376 } else {
377 memset(bar, '?', bar_sz);
378 }
379 put(s->bar);
380}
381
382
383static s_stat* init_cpu(const char *param)
384{
385 int sz;
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000386 cpu_stat *s = xzalloc(sizeof(*s));
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000387 s->collect = collect_cpu;
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000388 sz = strtoul(param, NULL, 0); /* param can be "" */
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000389 if (sz < 10) sz = 10;
390 if (sz > 1000) sz = 1000;
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000391 s->bar = xzalloc(sz+1);
392 /*s->bar[sz] = '\0'; - xzalloc did it */
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000393 s->bar_sz = sz;
394 return (s_stat*)s;
395}
396
397
398S_STAT(int_stat)
399 ullong old;
400 int no;
401S_STAT_END(int_stat)
402
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200403static void FAST_FUNC collect_int(int_stat *s)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000404{
405 ullong data[1];
406 ullong old;
407
408 if (rdval(get_file(&proc_stat), "intr", data, s->no)) {
409 put_question_marks(4);
410 return;
411 }
412
413 old = s->old;
414 if (data[0] < old) old = data[0]; //sanitize
415 s->old = data[0];
416 scale(data[0] - old);
417}
418
419static s_stat* init_int(const char *param)
420{
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000421 int_stat *s = xzalloc(sizeof(*s));
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000422 s->collect = collect_int;
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000423 if (param[0] == '\0') {
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000424 s->no = 1;
425 } else {
Denys Vlasenko77832482010-08-12 14:14:45 +0200426 int n = xatoi_positive(param);
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000427 s->no = n + 2;
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000428 }
429 return (s_stat*)s;
430}
431
432
433S_STAT(ctx_stat)
434 ullong old;
435S_STAT_END(ctx_stat)
436
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200437static void FAST_FUNC collect_ctx(ctx_stat *s)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000438{
439 ullong data[1];
440 ullong old;
441
442 if (rdval(get_file(&proc_stat), "ctxt", data, 1)) {
443 put_question_marks(4);
444 return;
445 }
446
447 old = s->old;
448 if (data[0] < old) old = data[0]; //sanitize
449 s->old = data[0];
450 scale(data[0] - old);
451}
452
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000453static s_stat* init_ctx(const char *param UNUSED_PARAM)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000454{
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000455 ctx_stat *s = xzalloc(sizeof(*s));
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000456 s->collect = collect_ctx;
457 return (s_stat*)s;
458}
459
460
461S_STAT(blk_stat)
462 const char* lookfor;
463 ullong old[2];
464S_STAT_END(blk_stat)
465
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200466static void FAST_FUNC collect_blk(blk_stat *s)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000467{
468 ullong data[2];
469 int i;
470
471 if (is26) {
472 i = rdval_diskstats(get_file(&proc_diskstats), data);
473 } else {
474 i = rdval(get_file(&proc_stat), s->lookfor, data, 1, 2);
475 // Linux 2.4 reports bio in Kbytes, convert to sectors:
476 data[0] *= 2;
477 data[1] *= 2;
478 }
479 if (i) {
480 put_question_marks(9);
481 return;
482 }
483
484 for (i=0; i<2; i++) {
485 ullong old = s->old[i];
486 if (data[i] < old) old = data[i]; //sanitize
487 s->old[i] = data[i];
488 data[i] -= old;
489 }
490 scale(data[0]*512); // TODO: *sectorsize
491 put_c(' ');
492 scale(data[1]*512);
493}
494
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000495static s_stat* init_blk(const char *param UNUSED_PARAM)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000496{
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000497 blk_stat *s = xzalloc(sizeof(*s));
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000498 s->collect = collect_blk;
499 s->lookfor = "page";
500 return (s_stat*)s;
501}
502
503
504S_STAT(fork_stat)
505 ullong old;
506S_STAT_END(fork_stat)
507
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200508static void FAST_FUNC collect_thread_nr(fork_stat *s UNUSED_PARAM)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000509{
510 ullong data[1];
511
512 if (rdval_loadavg(get_file(&proc_loadavg), data, 4)) {
513 put_question_marks(4);
514 return;
515 }
516 scale(data[0]);
517}
518
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200519static void FAST_FUNC collect_fork(fork_stat *s)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000520{
521 ullong data[1];
522 ullong old;
523
524 if (rdval(get_file(&proc_stat), "processes", data, 1)) {
525 put_question_marks(4);
526 return;
527 }
528
529 old = s->old;
530 if (data[0] < old) old = data[0]; //sanitize
531 s->old = data[0];
532 scale(data[0] - old);
533}
534
535static s_stat* init_fork(const char *param)
536{
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000537 fork_stat *s = xzalloc(sizeof(*s));
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000538 if (*param == 'n') {
539 s->collect = collect_thread_nr;
540 } else {
541 s->collect = collect_fork;
542 }
543 return (s_stat*)s;
544}
545
546
547S_STAT(if_stat)
548 ullong old[4];
549 const char *device;
550 char *device_colon;
551S_STAT_END(if_stat)
552
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200553static void FAST_FUNC collect_if(if_stat *s)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000554{
555 ullong data[4];
556 int i;
557
558 if (rdval(get_file(&proc_net_dev), s->device_colon, data, 1, 3, 9, 11)) {
559 put_question_marks(10);
560 return;
561 }
562
563 for (i=0; i<4; i++) {
564 ullong old = s->old[i];
565 if (data[i] < old) old = data[i]; //sanitize
566 s->old[i] = data[i];
567 data[i] -= old;
568 }
569 put_c(data[1] ? '*' : ' ');
570 scale(data[0]);
571 put_c(data[3] ? '*' : ' ');
572 scale(data[2]);
573}
574
575static s_stat* init_if(const char *device)
576{
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000577 if_stat *s = xzalloc(sizeof(*s));
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000578
579 if (!device || !device[0])
580 bb_show_usage();
581 s->collect = collect_if;
582
583 s->device = device;
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000584 s->device_colon = xasprintf("%s:", device);
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000585 return (s_stat*)s;
586}
587
588
589S_STAT(mem_stat)
590 char opt;
591S_STAT_END(mem_stat)
592
593// "Memory" value should not include any caches.
594// IOW: neither "ls -laR /" nor heavy read/write activity
595// should affect it. We'd like to also include any
596// long-term allocated kernel-side mem, but it is hard
597// to figure out. For now, bufs, cached & slab are
598// counted as "free" memory
599//2.6.16:
600//MemTotal: 773280 kB
601//MemFree: 25912 kB - genuinely free
602//Buffers: 320672 kB - cache
603//Cached: 146396 kB - cache
604//SwapCached: 0 kB
605//Active: 183064 kB
606//Inactive: 356892 kB
607//HighTotal: 0 kB
608//HighFree: 0 kB
609//LowTotal: 773280 kB
610//LowFree: 25912 kB
611//SwapTotal: 131064 kB
612//SwapFree: 131064 kB
613//Dirty: 48 kB
614//Writeback: 0 kB
615//Mapped: 96620 kB
616//Slab: 200668 kB - takes 7 Mb on my box fresh after boot,
617// but includes dentries and inodes
618// (== can take arbitrary amount of mem)
619//CommitLimit: 517704 kB
620//Committed_AS: 236776 kB
621//PageTables: 1248 kB
622//VmallocTotal: 516052 kB
623//VmallocUsed: 3852 kB
624//VmallocChunk: 512096 kB
625//HugePages_Total: 0
626//HugePages_Free: 0
627//Hugepagesize: 4096 kB
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200628static void FAST_FUNC collect_mem(mem_stat *s)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000629{
630 ullong m_total = 0;
631 ullong m_free = 0;
632 ullong m_bufs = 0;
633 ullong m_cached = 0;
634 ullong m_slab = 0;
635
636 if (rdval(get_file(&proc_meminfo), "MemTotal:", &m_total, 1)) {
637 put_question_marks(4);
638 return;
639 }
Denis Vlasenko8a2e4212008-02-29 07:46:19 +0000640 if (s->opt == 't') {
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000641 scale(m_total << 10);
642 return;
643 }
644
645 if (rdval(proc_meminfo.file, "MemFree:", &m_free , 1)
646 || rdval(proc_meminfo.file, "Buffers:", &m_bufs , 1)
647 || rdval(proc_meminfo.file, "Cached:", &m_cached, 1)
648 || rdval(proc_meminfo.file, "Slab:", &m_slab , 1)
649 ) {
650 put_question_marks(4);
651 return;
652 }
653
654 m_free += m_bufs + m_cached + m_slab;
Denis Vlasenkobf0a2012006-12-26 10:42:51 +0000655 switch (s->opt) {
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000656 case 'f':
657 scale(m_free << 10); break;
658 default:
659 scale((m_total - m_free) << 10); break;
660 }
661}
662
663static s_stat* init_mem(const char *param)
664{
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000665 mem_stat *s = xzalloc(sizeof(*s));
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000666 s->collect = collect_mem;
667 s->opt = param[0];
668 return (s_stat*)s;
669}
670
671
672S_STAT(swp_stat)
673S_STAT_END(swp_stat)
674
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200675static void FAST_FUNC collect_swp(swp_stat *s UNUSED_PARAM)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000676{
677 ullong s_total[1];
678 ullong s_free[1];
679 if (rdval(get_file(&proc_meminfo), "SwapTotal:", s_total, 1)
680 || rdval(proc_meminfo.file, "SwapFree:" , s_free, 1)
681 ) {
682 put_question_marks(4);
683 return;
684 }
685 scale((s_total[0]-s_free[0]) << 10);
686}
687
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000688static s_stat* init_swp(const char *param UNUSED_PARAM)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000689{
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000690 swp_stat *s = xzalloc(sizeof(*s));
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000691 s->collect = collect_swp;
692 return (s_stat*)s;
693}
694
695
696S_STAT(fd_stat)
697S_STAT_END(fd_stat)
698
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200699static void FAST_FUNC collect_fd(fd_stat *s UNUSED_PARAM)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000700{
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000701 ullong data[2];
702
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000703 if (rdval(get_file(&proc_sys_fs_filenr), "", data, 1, 2)) {
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000704 put_question_marks(4);
705 return;
706 }
707
708 scale(data[0] - data[1]);
709}
710
Denis Vlasenkoa60f84e2008-07-05 09:18:54 +0000711static s_stat* init_fd(const char *param UNUSED_PARAM)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000712{
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000713 fd_stat *s = xzalloc(sizeof(*s));
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000714 s->collect = collect_fd;
715 return (s_stat*)s;
716}
717
718
719S_STAT(time_stat)
720 int prec;
721 int scale;
722S_STAT_END(time_stat)
723
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200724static void FAST_FUNC collect_time(time_stat *s)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000725{
726 char buf[sizeof("12:34:56.123456")];
727 struct tm* tm;
728 int us = tv.tv_usec + s->scale/2;
729 time_t t = tv.tv_sec;
730
731 if (us >= 1000000) {
732 t++;
733 us -= 1000000;
734 }
735 tm = localtime(&t);
736
737 sprintf(buf, "%02d:%02d:%02d", tm->tm_hour, tm->tm_min, tm->tm_sec);
738 if (s->prec)
739 sprintf(buf+8, ".%0*d", s->prec, us / s->scale);
740 put(buf);
741}
742
743static s_stat* init_time(const char *param)
744{
745 int prec;
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000746 time_stat *s = xzalloc(sizeof(*s));
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000747
748 s->collect = collect_time;
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000749 prec = param[0] - '0';
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000750 if (prec < 0) prec = 0;
751 else if (prec > 6) prec = 6;
752 s->prec = prec;
753 s->scale = 1;
754 while (prec++ < 6)
755 s->scale *= 10;
756 return (s_stat*)s;
757}
758
Denys Vlasenkod5f1b1b2009-06-05 12:06:05 +0200759static void FAST_FUNC collect_info(s_stat *s)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000760{
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000761 gen ^= 1;
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000762 while (s) {
763 put(s->label);
764 s->collect(s);
765 s = s->next;
766 }
767}
768
769
770typedef s_stat* init_func(const char *param);
771
Denis Vlasenko6ca409e2007-08-12 20:58:27 +0000772static const char options[] ALIGN1 = "ncmsfixptbdr";
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000773static init_func *const init_functions[] = {
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000774 init_if,
775 init_cpu,
776 init_mem,
777 init_swp,
778 init_fd,
779 init_int,
780 init_ctx,
781 init_fork,
782 init_time,
783 init_blk,
784 init_delay,
Denis Vlasenko6ca409e2007-08-12 20:58:27 +0000785 init_cr
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000786};
787
Denis Vlasenko9b49a5e2007-10-11 10:05:36 +0000788int nmeter_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
Denys Vlasenko2ec91ae2010-01-04 14:15:38 +0100789int nmeter_main(int argc UNUSED_PARAM, char **argv)
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000790{
791 char buf[32];
792 s_stat *first = NULL;
793 s_stat *last = NULL;
794 s_stat *s;
795 char *cur, *prev;
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000796
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000797 INIT_G();
798
799 xchdir("/proc");
800
Denys Vlasenko2ec91ae2010-01-04 14:15:38 +0100801 if (!argv[1])
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000802 bb_show_usage();
803
Denis Vlasenkocd785fb2008-08-14 21:57:43 +0000804 if (open_read_close("version", buf, sizeof(buf)-1) > 0) {
805 buf[sizeof(buf)-1] = '\0';
806 is26 = (strstr(buf, " 2.4.") == NULL);
807 }
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000808
809 // Can use argv[1] directly, but this will mess up
810 // parameters as seen by e.g. ps. Making a copy...
811 cur = xstrdup(argv[1]);
812 while (1) {
813 char *param, *p;
814 prev = cur;
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000815 again:
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000816 cur = strchr(cur, '%');
817 if (!cur)
818 break;
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000819 if (cur[1] == '%') { // %%
Denis Vlasenko0f293b92008-07-22 20:16:55 +0000820 overlapping_strcpy(cur, cur + 1);
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000821 cur++;
822 goto again;
823 }
824 *cur++ = '\0'; // overwrite %
825 if (cur[0] == '[') {
826 // format: %[foptstring]
827 cur++;
828 p = strchr(options, cur[0]);
829 param = cur+1;
830 while (cur[0] != ']') {
831 if (!cur[0])
832 bb_show_usage();
833 cur++;
834 }
835 *cur++ = '\0'; // overwrite [
836 } else {
837 // format: %NNNNNNf
838 param = cur;
839 while (cur[0] >= '0' && cur[0] <= '9')
840 cur++;
841 if (!cur[0])
842 bb_show_usage();
843 p = strchr(options, cur[0]);
844 *cur++ = '\0'; // overwrite format char
845 }
846 if (!p)
847 bb_show_usage();
848 s = init_functions[p-options](param);
849 if (s) {
850 s->label = prev;
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000851 /*s->next = NULL; - all initXXX funcs use xzalloc */
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000852 if (!first)
853 first = s;
854 else
855 last->next = s;
856 last = s;
857 } else {
858 // %NNNNd or %r option. remove it from string
859 strcpy(prev + strlen(prev), cur);
860 cur = prev;
861 }
862 }
863 if (prev[0]) {
864 s = init_literal();
865 s->label = prev;
Denis Vlasenko333aa0c2008-06-28 23:30:37 +0000866 /*s->next = NULL; - all initXXX funcs use xzalloc */
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000867 if (!first)
868 first = s;
869 else
870 last->next = s;
871 last = s;
872 }
873
874 // Generate first samples but do not print them, they're bogus
875 collect_info(first);
876 reset_outbuf();
877 if (delta >= 0) {
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000878 gettimeofday(&tv, NULL);
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000879 usleep(delta > 1000000 ? 1000000 : delta - tv.tv_usec%deltanz);
880 }
881
882 while (1) {
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000883 gettimeofday(&tv, NULL);
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000884 collect_info(first);
885 put(final_str);
886 print_outbuf();
887
888 // Negative delta -> no usleep at all
889 // This will hog the CPU but you can have REALLY GOOD
890 // time resolution ;)
891 // TODO: detect and avoid useless updates
892 // (like: nothing happens except time)
893 if (delta >= 0) {
894 int rem;
895 // can be commented out, will sacrifice sleep time precision a bit
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000896 gettimeofday(&tv, NULL);
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000897 if (need_seconds)
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000898 rem = delta - ((ullong)tv.tv_sec*1000000 + tv.tv_usec) % deltanz;
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000899 else
900 rem = delta - tv.tv_usec%deltanz;
901 // Sometimes kernel wakes us up just a tiny bit earlier than asked
902 // Do not go to very short sleep in this case
903 if (rem < delta/128) {
904 rem += delta;
905 }
906 usleep(rem);
907 }
908 }
909
Denis Vlasenko199c0d52007-05-30 14:48:38 +0000910 /*return 0;*/
Denis Vlasenko1db39b22006-10-11 20:59:02 +0000911}