Home | History | Annotate | Download | only in toolbox
      1 /*
      2  * Copyright (c) 2008, The Android Open Source Project
      3  * All rights reserved.
      4  *
      5  * Redistribution and use in source and binary forms, with or without
      6  * modification, are permitted provided that the following conditions
      7  * are met:
      8  *  * Redistributions of source code must retain the above copyright
      9  *    notice, this list of conditions and the following disclaimer.
     10  *  * Redistributions in binary form must reproduce the above copyright
     11  *    notice, this list of conditions and the following disclaimer in
     12  *    the documentation and/or other materials provided with the
     13  *    distribution.
     14  *  * Neither the name of Google, Inc. nor the names of its contributors
     15  *    may be used to endorse or promote products derived from this
     16  *    software without specific prior written permission.
     17  *
     18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
     19  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
     20  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
     21  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
     22  * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
     23  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
     24  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
     25  * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
     26  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
     27  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
     28  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     29  * SUCH DAMAGE.
     30  */
     31 
     32 #include <errno.h>
     33 #include <stdio.h>
     34 #include <stdlib.h>
     35 #include <sys/param.h>
     36 #include <unistd.h>
     37 
     38 struct state {
     39     long procs_r;
     40     long procs_b;
     41 
     42     long mem_free;
     43     long mem_mapped;
     44     long mem_anon;
     45     long mem_slab;
     46 
     47     long sys_in;
     48     long sys_cs;
     49     long sys_flt;
     50 
     51     long cpu_us;
     52     long cpu_ni;
     53     long cpu_sy;
     54     long cpu_id;
     55     long cpu_wa;
     56     long cpu_ir;
     57     long cpu_si;
     58 };
     59 
     60 #define MAX_LINE 256
     61 
     62 char line[MAX_LINE];
     63 
     64 static void read_state(struct state *s);
     65 static int read_meminfo(struct state *s);
     66 static int read_stat(struct state *s);
     67 static int read_vmstat(struct state *s);
     68 static void print_header(void);
     69 static void print_line(struct state *old, struct state *new);
     70 static void usage(char *cmd);
     71 
     72 int vmstat_main(int argc, char *argv[]) {
     73     struct state s[2];
     74     int iterations, delay, header_interval;
     75     int toggle, count;
     76     int i;
     77 
     78     iterations = -1;
     79     delay = 1;
     80     header_interval = 20;
     81 
     82     for (i = 1; i < argc; i++) {
     83         if (!strcmp(argv[i], "-n")) {
     84             if (i >= argc - 1) {
     85                 fprintf(stderr, "Option -n requires an argument.\n");
     86                 exit(EXIT_FAILURE);
     87             }
     88             iterations = atoi(argv[++i]);
     89             continue;
     90         }
     91         if (!strcmp(argv[i], "-d")) {
     92             if (i >= argc - 1) {
     93                 fprintf(stderr, "Option -d requires an argument.\n");
     94                 exit(EXIT_FAILURE);
     95             }
     96             delay = atoi(argv[++i]);
     97             continue;
     98         }
     99         if (!strcmp(argv[i], "-r")) {
    100             if (i >= argc - 1) {
    101                 fprintf(stderr, "Option -r requires an argument.\n");
    102                 exit(EXIT_FAILURE);
    103             }
    104             header_interval = atoi(argv[++i]);
    105             continue;
    106         }
    107         if (!strcmp(argv[i], "-h")) {
    108             usage(argv[0]);
    109             exit(EXIT_SUCCESS);
    110         }
    111         fprintf(stderr, "Invalid argument \"%s\".\n", argv[i]);
    112         usage(argv[0]);
    113 	exit(EXIT_FAILURE);
    114     }
    115 
    116     toggle = 0;
    117     count = 0;
    118 
    119     if (!header_interval)
    120         print_header();
    121     read_state(&s[1 - toggle]);
    122     while ((iterations < 0) || (iterations-- > 0)) {
    123         sleep(delay);
    124         read_state(&s[toggle]);
    125         if (header_interval) {
    126             if (count == 0)
    127                 print_header();
    128             count = (count + 1) % header_interval;
    129         }
    130         print_line(&s[1 - toggle], &s[toggle]);
    131         toggle = 1 - toggle;
    132     }
    133 
    134     return 0;
    135 }
    136 
    137 static void read_state(struct state *s) {
    138     int error;
    139 
    140     error = read_meminfo(s);
    141     if (error) {
    142         fprintf(stderr, "vmstat: could not read /proc/meminfo: %s\n", strerror(error));
    143         exit(EXIT_FAILURE);
    144     }
    145 
    146     error = read_stat(s);
    147     if (error) {
    148         fprintf(stderr, "vmstat: could not read /proc/stat: %s\n", strerror(error));
    149         exit(EXIT_FAILURE);
    150     }
    151 
    152     error = read_vmstat(s);
    153     if (error) {
    154         fprintf(stderr, "vmstat: could not read /proc/vmstat: %s\n", strerror(error));
    155         exit(EXIT_FAILURE);
    156     }
    157 }
    158 
    159 static int read_meminfo(struct state *s) {
    160     FILE *f;
    161 
    162     f = fopen("/proc/meminfo", "r");
    163     if (!f) return errno;
    164 
    165     while (fgets(line, MAX_LINE, f)) {
    166         sscanf(line, "MemFree: %ld kB", &s->mem_free);
    167         sscanf(line, "AnonPages: %ld kB", &s->mem_anon);
    168         sscanf(line, "Mapped: %ld kB", &s->mem_mapped);
    169         sscanf(line, "Slab: %ld kB", &s->mem_slab);
    170     }
    171 
    172     fclose(f);
    173 
    174     return 0;
    175 }
    176 
    177 static int read_stat(struct state *s) {
    178     FILE *f;
    179 
    180     f = fopen("/proc/stat", "r");
    181     if (!f) return errno;
    182 
    183     while (fgets(line, MAX_LINE, f)) {
    184         if (!strncmp(line, "cpu ", 4)) {
    185             sscanf(line, "cpu  %ld %ld %ld %ld %ld %ld %ld",
    186                 &s->cpu_us, &s->cpu_ni, &s->cpu_sy, &s->cpu_id, &s->cpu_wa,
    187                 &s->cpu_ir, &s->cpu_si);
    188         }
    189         sscanf(line, "intr %ld", &s->sys_in);
    190         sscanf(line, "ctxt %ld", &s->sys_cs);
    191         sscanf(line, "procs_running %ld", &s->procs_r);
    192         sscanf(line, "procs_blocked %ld", &s->procs_b);
    193     }
    194 
    195     fclose(f);
    196 
    197     return 0;
    198 }
    199 
    200 static int read_vmstat(struct state *s) {
    201     FILE *f;
    202 
    203     f = fopen("/proc/vmstat", "r");
    204     if (!f) return errno;
    205 
    206     while (fgets(line, MAX_LINE, f)) {
    207         sscanf(line, "pgmajfault %ld", &s->sys_flt);
    208     }
    209 
    210     fclose(f);
    211 
    212     return 0;
    213 }
    214 
    215 static void print_header(void) {
    216     printf("%-5s  %-27s  %-14s  %-17s\n", "procs", "memory", "system", "cpu");
    217     printf("%2s %2s  %6s %6s %6s %6s  %4s %4s %4s  %2s %2s %2s %2s %2s %2s\n", "r", "b", "free", "mapped", "anon", "slab", "in", "cs", "flt", "us", "ni", "sy", "id", "wa", "ir");
    218 }
    219 
    220 /* Jiffies to percent conversion */
    221 #define JP(jif) ((jif) * 100 / (HZ))
    222 #define NORM(var) ((var) = (((var) > 99) ? (99) : (var)))
    223 
    224 static void print_line(struct state *old, struct state *new) {
    225     int us, ni, sy, id, wa, ir;
    226     us = JP(new->cpu_us - old->cpu_us); NORM(us);
    227     ni = JP(new->cpu_ni - old->cpu_ni); NORM(ni);
    228     sy = JP(new->cpu_sy - old->cpu_sy); NORM(sy);
    229     id = JP(new->cpu_id - old->cpu_id); NORM(id);
    230     wa = JP(new->cpu_wa - old->cpu_wa); NORM(wa);
    231     ir = JP(new->cpu_ir - old->cpu_ir); NORM(ir);
    232     printf("%2ld %2ld  %6ld %6ld %6ld %6ld  %4ld %4ld %4ld  %2d %2d %2d %2d %2d %2d\n",
    233         new->procs_r ? (new->procs_r - 1) : 0, new->procs_b,
    234         new->mem_free, new->mem_mapped, new->mem_anon, new->mem_slab,
    235         new->sys_in - old->sys_in, new->sys_cs - old->sys_cs, new->sys_flt - old->sys_flt,
    236         us, ni, sy, id, wa, ir);
    237 }
    238 
    239 static void usage(char *cmd) {
    240     fprintf(stderr, "Usage: %s [ -h ] [ -n iterations ] [ -d delay ] [ -r header_repeat ]\n"
    241                     "    -n iterations     How many rows of data to print.\n"
    242                     "    -d delay          How long to sleep between rows.\n"
    243                     "    -r header_repeat  How many rows to print before repeating\n"
    244                     "                      the header.  Zero means never repeat.\n"
    245                     "    -h                Displays this help screen.\n",
    246         cmd);
    247 }
    248