2 ** Licensed under the GPL v2, see the file LICENSE in this tarball
4 ** Based on nanotop.c from floppyfw project
6 ** Contact me: vda.linux@googlemail.com */
12 // disk_io: (3,0):(22272,17897,410702,4375,54750)
14 //TODO: use sysinfo libc call/syscall, if appropriate
15 // (faster than open/read/close):
16 // sysinfo({uptime=15017, loads=[5728, 15040, 16480]
17 // totalram=2107416576, freeram=211525632, sharedram=0, bufferram=157204480}
18 // totalswap=134209536, freeswap=134209536, procs=157})
23 typedef unsigned long long ullong;
25 enum { /* Preferably use powers of 2 */
26 PROC_MIN_FILE_SIZE = 256,
27 PROC_MAX_FILE_SIZE = 16 * 1024,
30 typedef struct proc_file {
36 static const char *const proc_name[] = {
37 "stat", // Must match the order of proc_file's!
46 // Sample generation flip-flop
48 // Linux 2.6? (otherwise assumes 2.4)
50 // 1 if sample delay is not an integer fraction of a second
51 smallint need_seconds;
53 const char *final_str;
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;
65 #define G (*ptr_to_globals)
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 )
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)
80 #define INIT_G() do { \
81 SET_PTR_TO_GLOBALS(xzalloc(sizeof(G))); \
82 cur_outbuf = outbuf; \
84 deltanz = delta = 1000000; \
87 // We depend on this being a char[], not char* - we take sizeof() of it
88 #define outbuf bb_common_bufsiz1
90 static inline void reset_outbuf(void)
95 static inline int outbuf_count(void)
97 return cur_outbuf - outbuf;
100 static void print_outbuf(void)
102 int sz = cur_outbuf - outbuf;
104 xwrite(STDOUT_FILENO, outbuf, sz);
109 static void put(const char *s)
112 if (sz > outbuf + sizeof(outbuf) - cur_outbuf)
113 sz = outbuf + sizeof(outbuf) - cur_outbuf;
114 memcpy(cur_outbuf, s, sz);
118 static void put_c(char c)
120 if (cur_outbuf < outbuf + sizeof(outbuf))
124 static void put_question_marks(int count)
130 static void readfile_z(proc_file *pf, const char* fname)
132 // open_read_close() will do two reads in order to be sure we are at EOF,
133 // and we don't need/want that.
141 buf = xmalloc(PROC_MIN_FILE_SIZE);
142 sz = PROC_MIN_FILE_SIZE;
145 fd = xopen(fname, O_RDONLY);
147 rdsz = read(fd, buf, sz-1);
150 if (rdsz == sz-1 && sz < PROC_MAX_FILE_SIZE) {
152 buf = xrealloc(buf, sz);
161 static const char* get_file(proc_file *pf)
163 if (pf->last_gen != gen) {
165 readfile_z(pf, proc_name[pf - &first_proc_file]);
170 static ullong read_after_slash(const char *p)
174 return strtoull(p+1, NULL, 10);
177 enum conv_type { conv_decimal, conv_slash };
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.
183 static int vrdval(const char* p, const char* key,
184 enum conv_type conv, ullong *vec, va_list arg_ptr)
194 indexnext = va_arg(arg_ptr, int);
196 while (*p == ' ' || *p == '\t') p++;
197 if (*p == '\n' || *p == '\0') break;
199 if (indexline == indexnext) { // read this value
200 *vec++ = conv==conv_decimal ?
201 strtoull(p, NULL, 10) :
203 indexnext = va_arg(arg_ptr, int);
205 while (*p > ' ') p++; // skip over value
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
214 static int rdval(const char* p, const char* key, ullong *vec, ...)
219 va_start(arg_ptr, vec);
220 result = vrdval(p, key, conv_decimal, vec, arg_ptr);
226 // Parses files with lines like "... ... ... 3/148 ...."
227 static int rdval_loadavg(const char* p, ullong *vec, ...)
232 va_start(arg_ptr, vec);
233 result = vrdval(p, "", conv_slash, vec, arg_ptr);
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
243 static int rdval_diskstats(const char* p, ullong *vec)
245 ullong rd = rd; // for compiler
251 while (*p == ' ' || *p == '\t') p++;
252 if (*p == '\0') break;
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++;
266 while (*p > ' ') p++; // skip over value
271 static void scale(ullong ul)
275 /* see http://en.wikipedia.org/wiki/Tera */
276 smart_ulltoa4(ul, buf, " kmgtpezy");
284 struct s_stat *next; \
285 void (*collect)(struct a *s); \
287 #define S_STAT_END(a) } a;
292 static void collect_literal(s_stat *s UNUSED_PARAM)
296 static s_stat* init_literal(void)
298 s_stat *s = xzalloc(sizeof(*s));
299 s->collect = collect_literal;
303 static s_stat* init_delay(const char *param)
305 delta = strtoul(param, NULL, 0) * 1000; /* param can be "" */
306 deltanz = delta > 0 ? delta : 1;
307 need_seconds = (1000000%deltanz) != 0;
311 static s_stat* init_cr(const char *param UNUSED_PARAM)
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
321 enum { CPU_FIELDCNT = 7 };
323 ullong old[CPU_FIELDCNT];
329 static void collect_cpu(cpu_stat *s)
331 ullong data[CPU_FIELDCNT] = { 0, 0, 0, 0, 0, 0, 0 };
332 unsigned frac[CPU_FIELDCNT] = { 0, 0, 0, 0, 0, 0, 0 };
335 int bar_sz = s->bar_sz;
339 if (rdval(get_file(&proc_stat), "cpu ", data, 1, 2, 3, 4, 5, 6, 7)) {
340 put_question_marks(bar_sz);
344 for (i = 0; i < CPU_FIELDCNT; i++) {
345 ullong old = s->old[i];
346 if (data[i] < old) old = data[i]; //sanitize
348 all += (data[i] -= old);
352 for (i = 0; i < CPU_FIELDCNT; i++) {
353 ullong t = bar_sz * data[i];
354 norm_all += data[i] = t / all;
358 while (norm_all < bar_sz) {
359 unsigned max = frac[0];
361 for (i = 1; i < CPU_FIELDCNT; i++) {
362 if (frac[i] > max) max = frac[i], pos = i;
364 frac[pos] = 0; //avoid bumping up same value twice
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
377 memset(bar, '?', bar_sz);
383 static s_stat* init_cpu(const char *param)
386 cpu_stat *s = xzalloc(sizeof(*s));
387 s->collect = collect_cpu;
388 sz = strtoul(param, NULL, 0); /* param can be "" */
389 if (sz < 10) sz = 10;
390 if (sz > 1000) sz = 1000;
391 s->bar = xzalloc(sz+1);
392 /*s->bar[sz] = '\0'; - xzalloc did it */
403 static void collect_int(int_stat *s)
408 if (rdval(get_file(&proc_stat), "intr", data, s->no)) {
409 put_question_marks(4);
414 if (data[0] < old) old = data[0]; //sanitize
416 scale(data[0] - old);
419 static s_stat* init_int(const char *param)
421 int_stat *s = xzalloc(sizeof(*s));
422 s->collect = collect_int;
423 if (param[0] == '\0') {
426 int n = xatoi_u(param);
437 static void collect_ctx(ctx_stat *s)
442 if (rdval(get_file(&proc_stat), "ctxt", data, 1)) {
443 put_question_marks(4);
448 if (data[0] < old) old = data[0]; //sanitize
450 scale(data[0] - old);
453 static s_stat* init_ctx(const char *param UNUSED_PARAM)
455 ctx_stat *s = xzalloc(sizeof(*s));
456 s->collect = collect_ctx;
466 static void collect_blk(blk_stat *s)
472 i = rdval_diskstats(get_file(&proc_diskstats), data);
474 i = rdval(get_file(&proc_stat), s->lookfor, data, 1, 2);
475 // Linux 2.4 reports bio in Kbytes, convert to sectors:
480 put_question_marks(9);
484 for (i=0; i<2; i++) {
485 ullong old = s->old[i];
486 if (data[i] < old) old = data[i]; //sanitize
490 scale(data[0]*512); // TODO: *sectorsize
495 static s_stat* init_blk(const char *param UNUSED_PARAM)
497 blk_stat *s = xzalloc(sizeof(*s));
498 s->collect = collect_blk;
506 S_STAT_END(fork_stat)
508 static void collect_thread_nr(fork_stat *s UNUSED_PARAM)
512 if (rdval_loadavg(get_file(&proc_loadavg), data, 4)) {
513 put_question_marks(4);
519 static void collect_fork(fork_stat *s)
524 if (rdval(get_file(&proc_stat), "processes", data, 1)) {
525 put_question_marks(4);
530 if (data[0] < old) old = data[0]; //sanitize
532 scale(data[0] - old);
535 static s_stat* init_fork(const char *param)
537 fork_stat *s = xzalloc(sizeof(*s));
539 s->collect = collect_thread_nr;
541 s->collect = collect_fork;
553 static void collect_if(if_stat *s)
558 if (rdval(get_file(&proc_net_dev), s->device_colon, data, 1, 3, 9, 11)) {
559 put_question_marks(10);
563 for (i=0; i<4; i++) {
564 ullong old = s->old[i];
565 if (data[i] < old) old = data[i]; //sanitize
569 put_c(data[1] ? '*' : ' ');
571 put_c(data[3] ? '*' : ' ');
575 static s_stat* init_if(const char *device)
577 if_stat *s = xzalloc(sizeof(*s));
579 if (!device || !device[0])
581 s->collect = collect_if;
584 s->device_colon = xasprintf("%s:", device);
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
600 //MemTotal: 773280 kB
601 //MemFree: 25912 kB - genuinely free
602 //Buffers: 320672 kB - cache
603 //Cached: 146396 kB - cache
606 //Inactive: 356892 kB
609 //LowTotal: 773280 kB
611 //SwapTotal: 131064 kB
612 //SwapFree: 131064 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
627 //Hugepagesize: 4096 kB
628 static void collect_mem(mem_stat *s)
636 if (rdval(get_file(&proc_meminfo), "MemTotal:", &m_total, 1)) {
637 put_question_marks(4);
641 scale(m_total << 10);
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)
650 put_question_marks(4);
654 m_free += m_bufs + m_cached + m_slab;
657 scale(m_free << 10); break;
659 scale((m_total - m_free) << 10); break;
663 static s_stat* init_mem(const char *param)
665 mem_stat *s = xzalloc(sizeof(*s));
666 s->collect = collect_mem;
675 static void collect_swp(swp_stat *s UNUSED_PARAM)
679 if (rdval(get_file(&proc_meminfo), "SwapTotal:", s_total, 1)
680 || rdval(proc_meminfo.file, "SwapFree:" , s_free, 1)
682 put_question_marks(4);
685 scale((s_total[0]-s_free[0]) << 10);
688 static s_stat* init_swp(const char *param UNUSED_PARAM)
690 swp_stat *s = xzalloc(sizeof(*s));
691 s->collect = collect_swp;
699 static void collect_fd(fd_stat *s UNUSED_PARAM)
703 if (rdval(get_file(&proc_sys_fs_filenr), "", data, 1, 2)) {
704 put_question_marks(4);
708 scale(data[0] - data[1]);
711 static s_stat* init_fd(const char *param UNUSED_PARAM)
713 fd_stat *s = xzalloc(sizeof(*s));
714 s->collect = collect_fd;
722 S_STAT_END(time_stat)
724 static void collect_time(time_stat *s)
726 char buf[sizeof("12:34:56.123456")];
728 int us = tv.tv_usec + s->scale/2;
729 time_t t = tv.tv_sec;
737 sprintf(buf, "%02d:%02d:%02d", tm->tm_hour, tm->tm_min, tm->tm_sec);
739 sprintf(buf+8, ".%0*d", s->prec, us / s->scale);
743 static s_stat* init_time(const char *param)
746 time_stat *s = xzalloc(sizeof(*s));
748 s->collect = collect_time;
749 prec = param[0] - '0';
750 if (prec < 0) prec = 0;
751 else if (prec > 6) prec = 6;
759 static void collect_info(s_stat *s)
770 typedef s_stat* init_func(const char *param);
772 static const char options[] ALIGN1 = "ncmsfixptbdr";
773 static init_func *const init_functions[] = {
788 int nmeter_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
789 int nmeter_main(int argc, char **argv)
792 s_stat *first = NULL;
804 if (open_read_close("version", buf, sizeof(buf)-1) > 0) {
805 buf[sizeof(buf)-1] = '\0';
806 is26 = (strstr(buf, " 2.4.") == NULL);
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]);
816 cur = strchr(cur, '%');
819 if (cur[1] == '%') { // %%
820 overlapping_strcpy(cur, cur + 1);
824 *cur++ = '\0'; // overwrite %
826 // format: %[foptstring]
828 p = strchr(options, cur[0]);
830 while (cur[0] != ']') {
835 *cur++ = '\0'; // overwrite [
839 while (cur[0] >= '0' && cur[0] <= '9')
843 p = strchr(options, cur[0]);
844 *cur++ = '\0'; // overwrite format char
848 s = init_functions[p-options](param);
851 /*s->next = NULL; - all initXXX funcs use xzalloc */
858 // %NNNNd or %r option. remove it from string
859 strcpy(prev + strlen(prev), cur);
866 /*s->next = NULL; - all initXXX funcs use xzalloc */
874 // Generate first samples but do not print them, they're bogus
878 gettimeofday(&tv, NULL);
879 usleep(delta > 1000000 ? 1000000 : delta - tv.tv_usec%deltanz);
883 gettimeofday(&tv, NULL);
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)
895 // can be commented out, will sacrifice sleep time precision a bit
896 gettimeofday(&tv, NULL);
898 rem = delta - ((ullong)tv.tv_sec*1000000 + tv.tv_usec) % deltanz;
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) {