/*
 * old: "Copyright 1994 by Henry Ware <al172@yfn.ysu.edu>. Copyleft same year."
 * most code copyright 2002 Albert Cahalan
 *
 * 27/05/2003 (Fabian Frederick) : Add unit conversion + interface
 *				   Export proc/stat access to libproc
 *				   Adapt vmstat helpfile
 * 31/05/2003 (Fabian) : Add diskstat support (/libproc)
 * June 2003 (Fabian)  : -S <x> -s & -s -S <x> patch
 * June 2003 (Fabian)  : Adding diskstat against 3.1.9, slabinfo
 *			 patching 'header' in disk & slab
 * July 2003 (Fabian)  : Adding disk partition output
 *			 Adding disk table
 *			 Syncing help / usage
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */

#include <assert.h>
#include <ctype.h>
#include <dirent.h>
#include <errno.h>
#include <fcntl.h>
#include <getopt.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/ioctl.h>
#include <sys/types.h>
#include <termios.h>
#include <unistd.h>
#include <time.h>

#include "c.h"
#include "fileutils.h"
#include "nls.h"
#include "strutils.h"
#include <proc/procps.h>

#define UNIT_B        1
#define UNIT_k        1000
#define UNIT_K        1024
#define UNIT_m        1000000
#define UNIT_M        1048576

static unsigned long dataUnit = UNIT_K;
static char szDataUnit[3] = "K";

#define VMSTAT        0
#define DISKSTAT      0x00000001
#define VMSUMSTAT     0x00000002
#define SLABSTAT      0x00000004
#define PARTITIONSTAT 0x00000008
#define DISKSUMSTAT   0x00000010

static int statMode = VMSTAT;

/* "-a" means "show active/inactive" */
static int a_option;

/* "-w" means "wide output" */
static int w_option;

/* "-t" means "show timestamp" */
static int t_option;

static unsigned sleep_time = 1;
static int infinite_updates = 0;
static unsigned long num_updates =1;
/* window height */
static unsigned int height;
static unsigned int moreheaders = TRUE;

static void __attribute__ ((__noreturn__))
    usage(FILE * out)
{
	fputs(USAGE_HEADER, out);
	fprintf(out,
	      _(" %s [options] [delay [count]]\n"),
		program_invocation_short_name);
	fputs(USAGE_OPTIONS, out);
	fputs(_(" -a, --active           active/inactive memory\n"), out);
	fputs(_(" -f, --forks            number of forks since boot\n"), out);
	fputs(_(" -m, --slabs            slabinfo\n"), out);
	fputs(_(" -n, --one-header       do not redisplay header\n"), out);
	fputs(_(" -s, --stats            event counter statistics\n"), out);
	fputs(_(" -d, --disk             disk statistics\n"), out);
	fputs(_(" -D, --disk-sum         summarize disk statistics\n"), out);
	fputs(_(" -p, --partition <dev>  partition specific statistics\n"), out);
	fputs(_(" -S, --unit <char>      define display unit\n"), out);
	fputs(_(" -w, --wide             wide output\n"), out);
	fputs(_(" -t, --timestamp        show timestamp\n"), out);
	fputs(USAGE_SEPARATOR, out);
	fputs(USAGE_HELP, out);
	fputs(USAGE_VERSION, out);
	fprintf(out, USAGE_MAN_TAIL("vmstat(8)"));

	exit(out == stderr ? EXIT_FAILURE : EXIT_SUCCESS);
}

static void new_header(void)
{
	struct tm *tm_ptr;
	time_t the_time;
	char timebuf[32];

	/* Translation Hint: Translating folloging header & fields
	 * that follow (marked with max x chars) might not work,
	 * unless manual page is translated as well.  */
	const char *header =
	    _("procs -----------memory---------- ---swap-- -----io---- -system-- ------cpu-----");
	const char *wide_header =
	    _("procs -----------------------memory---------------------- ---swap-- -----io---- -system-- --------cpu--------");
	const char *timestamp_header = _(" -----timestamp-----");

	const char format[] =
	    "%2s %2s %6s %6s %6s %6s %4s %4s %5s %5s %4s %4s %2s %2s %2s %2s %2s";
	const char wide_format[] =
	    "%2s %2s %12s %12s %12s %12s %4s %4s %5s %5s %4s %4s %3s %3s %3s %3s %3s";


	printf("%s", w_option ? wide_header : header);

	if (t_option) {
		printf("%s", timestamp_header);
	}

	printf("\n");

	printf(
	    w_option ? wide_format : format,
	    /* Translation Hint: max 2 chars */
	     _("r"),
	    /* Translation Hint: max 2 chars */
	     _("b"),
	    /* Translation Hint: max 6 chars */
	     _("swpd"),
	    /* Translation Hint: max 6 chars */
	     _("free"),
	    /* Translation Hint: max 6 chars */
	     a_option ? _("inact") :
	    /* Translation Hint: max 6 chars */
			_("buff"),
	    /* Translation Hint: max 6 chars */
	     a_option ? _("active") :
	    /* Translation Hint: max 6 chars */
			_("cache"),
	    /* Translation Hint: max 4 chars */
	     _("si"),
	    /* Translation Hint: max 4 chars */
	     _("so"),
	    /* Translation Hint: max 5 chars */
	     _("bi"),
	    /* Translation Hint: max 5 chars */
	     _("bo"),
	    /* Translation Hint: max 4 chars */
	     _("in"),
	    /* Translation Hint: max 4 chars */
	     _("cs"),
	    /* Translation Hint: max 2 chars */
	     _("us"),
	    /* Translation Hint: max 2 chars */
	     _("sy"),
	    /* Translation Hint: max 2 chars */
	     _("id"),
	    /* Translation Hint: max 2 chars */
	     _("wa"),
	    /* Translation Hint: max 2 chars */
	     _("st"));

	if (t_option) {
		(void) time( &the_time );
		tm_ptr = localtime( &the_time );
		if (strftime(timebuf, sizeof(timebuf), "%Z", tm_ptr)) {
			timebuf[strlen(timestamp_header) - 1] = '\0';
		} else {
			timebuf[0] = '\0';
		}
		printf(" %*s", (int)(strlen(timestamp_header) - 1), timebuf);
	}

	printf("\n");
}


static unsigned long unitConvert(unsigned long size)
{
	float cvSize;
	cvSize = (float)size / dataUnit * ((statMode == SLABSTAT) ? 1 : 1024);
	return ((unsigned long)cvSize);
}

static enum stat_item First_stat_items[] = {
    PROCPS_STAT_SYS_PROC_RUNNING,
    PROCPS_STAT_SYS_PROC_BLOCKED,
    PROCPS_STAT_SYS_INTERRUPTS,
    PROCPS_STAT_SYS_CTX_SWITCHES,
    PROCPS_STAT_TIC_USER,
    PROCPS_STAT_TIC_NICE,
    PROCPS_STAT_TIC_SYSTEM,
    PROCPS_STAT_TIC_IRQ,
    PROCPS_STAT_TIC_SOFTIRQ,
    PROCPS_STAT_TIC_IDLE,
    PROCPS_STAT_TIC_IOWAIT,
    PROCPS_STAT_TIC_STOLEN
};

static enum stat_item Loop_stat_items[] = {
    PROCPS_STAT_SYS_PROC_RUNNING,
    PROCPS_STAT_SYS_PROC_BLOCKED,
    PROCPS_STAT_SYS_DELTA_INTERRUPTS,
    PROCPS_STAT_SYS_DELTA_CTX_SWITCHES,
    PROCPS_STAT_TIC_DELTA_USER,
    PROCPS_STAT_TIC_DELTA_NICE,
    PROCPS_STAT_TIC_DELTA_SYSTEM,
    PROCPS_STAT_TIC_DELTA_IRQ,
    PROCPS_STAT_TIC_DELTA_SOFTIRQ,
    PROCPS_STAT_TIC_DELTA_IDLE,
    PROCPS_STAT_TIC_DELTA_IOWAIT,
    PROCPS_STAT_TIC_DELTA_STOLEN
};

enum Rel_statitems {
    stat_PRU, stat_PBL,
    stat_INT, stat_CTX,
    stat_USR, stat_NIC, stat_SYS, stat_IRQ, stat_SRQ,
    stat_IDL, stat_IOW, stat_STO};

static enum meminfo_item Mem_items[] = {
    PROCPS_MEMINFO_SWAP_USED,
    PROCPS_MEMINFO_MEM_FREE,
    PROCPS_MEMINFO_MEM_ACTIVE,
    PROCPS_MEMINFO_MEM_INACTIVE,
    PROCPS_MEMINFO_MEM_BUFFERS,
    PROCPS_MEMINFO_MEM_CACHED
};

enum Rel_memitems {
    mem_SUS, mem_FREE,
    mem_ACT, mem_INA,
    mem_BUF, mem_CAC
};

static void new_format(void)
{
#define TICv(E) PROCPS_STAT_VAL(E, ull_int, stat_stack)
#define DTICv(E) PROCPS_STAT_VAL(E, sl_int, stat_stack)
#define SYSv(E) PROCPS_STAT_VAL(E, ul_int, stat_stack)
#define MEMv(E) PROCPS_STAT_VAL(E, ul_int, mem_stack)
#define DSYSv(E) PROCPS_STAT_VAL(E, s_int, stat_stack)
    const char format[] =
        "%2lu %2lu %6lu %6lu %6lu %6lu %4u %4u %5u %5u %4u %4u %2u %2u %2u %2u %2u";
    const char wide_format[] =
        "%2lu %2lu %12lu %12lu %12lu %12lu %4u %4u %5u %5u %4u %4u %3u %3u %3u %3u %3u";

    unsigned int tog = 0;    /* toggle switch for cleaner code */
    unsigned int i;
    long hz;
    long long cpu_use, cpu_sys, cpu_idl, cpu_iow, cpu_sto;
    long long Div, divo2;
    unsigned long pgpgin[2], pgpgout[2], pswpin[2] = {0,0}, pswpout[2];
    unsigned int sleep_half;
    unsigned long kb_per_page = sysconf(_SC_PAGESIZE) / 1024ul;
    int debt = 0;        /* handle idle ticks running backwards */
    struct tm *tm_ptr;
    time_t the_time;
    char timebuf[32];
    struct procps_vmstat *vm_info = NULL;
    struct procps_statinfo *sys_info = NULL;
    struct stat_stack *stat_stack;
    struct procps_meminfo *mem_info = NULL;
    struct meminfo_stack *mem_stack;

    sleep_half = (sleep_time / 2);
    hz = procps_hertz_get();
    new_header();

    if (procps_vmstat_new(&vm_info) < 0)
        xerrx(EXIT_FAILURE,
            _("Unable to create vmstat structure"));
    if (procps_stat_new(&sys_info) < 0)
        xerrx(EXIT_FAILURE,
            _("Unable to create system stat structure"));
    if (procps_meminfo_new(&mem_info) < 0)
        xerrx(EXIT_FAILURE, _("Unable to create meminfo structure"));

    if (t_option) {
        (void) time( &the_time );
        tm_ptr = localtime( &the_time );
        strftime(timebuf, sizeof(timebuf), "%Y-%m-%d %H:%M:%S", tm_ptr);
    }
    /* Do the initial fill */
    if ((stat_stack = procps_stat_select(sys_info, First_stat_items, 12)) ==
        NULL)
        xerrx(EXIT_FAILURE,
              _("Unable to select stat information"));
    cpu_use = TICv(stat_USR) + TICv(stat_NIC);
    cpu_sys = TICv(stat_SYS) + TICv(stat_IRQ) + TICv(stat_SRQ);
    cpu_idl = TICv(stat_IDL);
    cpu_iow = TICv(stat_IOW);
    cpu_sto = TICv(stat_STO);

    pgpgin[tog] = procps_vmstat_get(vm_info, PROCPS_VMSTAT_PGPGIN);
    pgpgout[tog] = procps_vmstat_get(vm_info, PROCPS_VMSTAT_PGPGOUT);
    pswpin[tog] = procps_vmstat_get(vm_info, PROCPS_VMSTAT_PSWPIN);
    pswpout[tog] = procps_vmstat_get(vm_info, PROCPS_VMSTAT_PSWPOUT);

    if ((mem_stack = procps_meminfo_select(mem_info, Mem_items, 6)) ==
        NULL)
        xerrx(EXIT_FAILURE,
              _("Unable to select memory information"));

    Div = cpu_use + cpu_sys + cpu_idl + cpu_iow + cpu_sto;
    if (!Div) {
        Div = 1;
        cpu_idl = 1;
    }
    divo2 = Div / 2UL;

    printf(w_option ? wide_format : format,
           SYSv(stat_PRU),
           SYSv(stat_PBL),
           unitConvert(MEMv(mem_SUS)),
           unitConvert(MEMv(mem_FREE)),
           unitConvert((a_option?MEMv(mem_INA):MEMv(mem_BUF))),
           unitConvert((a_option?MEMv(mem_ACT):MEMv(mem_CAC))),
           (unsigned)( (unitConvert(procps_vmstat_get(vm_info, PROCPS_VMSTAT_PSWPIN)  * kb_per_page) * hz + divo2) / Div ),
           (unsigned)( (unitConvert(procps_vmstat_get(vm_info, PROCPS_VMSTAT_PSWPOUT)  * kb_per_page) * hz + divo2) / Div ),
           (unsigned)( (procps_vmstat_get(vm_info, PROCPS_VMSTAT_PGPGIN) * hz + divo2) / Div ),
           (unsigned)( (procps_vmstat_get(vm_info, PROCPS_VMSTAT_PGPGOUT) * hz + divo2) / Div ),
           (unsigned)( (SYSv(stat_INT)           * hz + divo2) / Div ),
           (unsigned)( (SYSv(stat_CTX)           * hz + divo2) / Div ),
           (unsigned)( (100*cpu_use        + divo2) / Div ),
           (unsigned)( (100*cpu_sys        + divo2) / Div ),
           (unsigned)( (100*cpu_idl        + divo2) / Div ),
           (unsigned)( (100*cpu_iow        + divo2) / Div ),
           (unsigned)( (100*cpu_sto        + divo2) / Div )
    );

    if (t_option) {
        printf(" %s", timebuf);
    }

    printf("\n");

    /* main loop */
    for (i = 1; infinite_updates || i < num_updates; i++) {
        sleep(sleep_time);
        if (moreheaders && ((i % height) == 0))
            new_header();
        tog = !tog;

        if ((stat_stack = procps_stat_select(sys_info, Loop_stat_items, 12)) == NULL)
            xerrx(EXIT_FAILURE,
                  _("Unable to select stat information"));

        cpu_use = DTICv(stat_USR) + DTICv(stat_NIC);
        cpu_sys = DTICv(stat_SYS) + DTICv(stat_IRQ) + DTICv(stat_SRQ);
        cpu_idl = DTICv(stat_IDL);
        cpu_iow = DTICv(stat_IOW);
        cpu_sto = DTICv(stat_STO);
        pgpgin[tog] = procps_vmstat_get(vm_info, PROCPS_VMSTAT_PGPGIN);
        pgpgout[tog] = procps_vmstat_get(vm_info, PROCPS_VMSTAT_PGPGOUT);
        pswpin[tog] = procps_vmstat_get(vm_info, PROCPS_VMSTAT_PSWPIN);
        pswpout[tog] = procps_vmstat_get(vm_info, PROCPS_VMSTAT_PSWPOUT);

        if (t_option) {
            (void) time( &the_time );
            tm_ptr = localtime( &the_time );
            strftime(timebuf, sizeof(timebuf), "%Y-%m-%d %H:%M:%S", tm_ptr);
        }

        /* idle can run backwards for a moment -- kernel "feature" */
        if (debt) {
            cpu_idl = (int)cpu_idl + debt;
            debt = 0;
        }
        if ((int)cpu_idl < 0) {
            debt = (int)cpu_idl;
            cpu_idl = 0;
        }

        Div = cpu_use + cpu_sys + cpu_idl + cpu_iow + cpu_sto;
        if (!Div) Div = 1, cpu_idl = 1;
        divo2 = Div / 2UL;
        printf(w_option ? wide_format : format,
               SYSv(stat_PRU),
               SYSv(stat_PBL),
               unitConvert(MEMv(mem_SUS)),
               unitConvert(MEMv(mem_FREE)),
               unitConvert((a_option?MEMv(mem_INA):MEMv(mem_BUF))),
               unitConvert((a_option?MEMv(mem_ACT):MEMv(mem_CAC))),
               /*si */
               (unsigned)( ( unitConvert((pswpin [tog] - pswpin [!tog])*kb_per_page)+sleep_half )/sleep_time ),
               /* so */
               (unsigned)( ( unitConvert((pswpout[tog] - pswpout[!tog])*kb_per_page)+sleep_half )/sleep_time ),
               /* bi */
               (unsigned)( (  pgpgin [tog] - pgpgin [!tog]           +sleep_half )/sleep_time ),
               /* bo */
               (unsigned)( (  pgpgout[tog] - pgpgout[!tog]           +sleep_half )/sleep_time ),
               /* in */
               (unsigned)( (  DSYSv(stat_INT)           +sleep_half )/sleep_time ),
               /* cs */
               (unsigned)( (  DSYSv(stat_CTX)           +sleep_half )/sleep_time ),
               /* us */
               (unsigned)( (100*cpu_use+divo2)/Div ),
               /* sy */
               (unsigned)( (100*cpu_sys+divo2)/Div ),
               /* id */
               (unsigned)( (100*cpu_idl+divo2)/Div ),
               /* wa */
               (unsigned)( (100*cpu_iow+divo2)/Div ),
               /* st */
               (unsigned)( (100*cpu_sto+divo2)/Div )
        );

        if (t_option) {
            printf(" %s", timebuf);
        }

        printf("\n");
    }
    /* Cleanup */
    procps_stat_unref(&sys_info);
    procps_vmstat_unref(&vm_info);
    procps_meminfo_unref(&mem_info);
#undef TICv
#undef DTICv
#undef SYSv
#undef DSYSv
#undef MEMv
}

static void diskpartition_header(const char *partition_name)
{
	printf("%-10s %10s %10s %10s %10s\n",
	       partition_name,
       /* Translation Hint: Translating folloging disk partition
	* header fields that follow (marked with max x chars) might
	* not work, unless manual page is translated as well. */
	       /* Translation Hint: max 10 chars. The word is
	        * expected to be centralized, use spaces at the end
	        * to do that. */
	       _("reads  "),
	       /* Translation Hint: max 10 chars */
	       _("read sectors"),
	       /* Translation Hint: max 10 chars. The word is
	        * expected to be centralized, use spaces at the end
	        * to do that. */
	       _("writes   "),
	       /* Translation Hint: max 10 chars */
	       _("requested writes"));
}

static int diskpartition_format(const char *partition_name)
{
#define PARTGET(x) procps_diskstat_dev_get(disk_stat, (x), partid)
    struct procps_diskstat *disk_stat;
    const char format[] = "%20lu %10lu %10lu %10lu\n";
    int i, partid;

    if (procps_diskstat_new(&disk_stat) < 0)
        xerr(EXIT_FAILURE,
             _("Unable to create diskstat structure"));

    if (procps_diskstat_read(disk_stat) < 0)
        xerr(EXIT_FAILURE,
             _("Unable to read diskstat"));
    if ((partid = procps_diskstat_dev_getbyname(disk_stat, partition_name))
        < 0)
        xerrx(EXIT_FAILURE, _("Partition %s not found"), partition_name);

    diskpartition_header(partition_name);
    for (i=0; infinite_updates || i < num_updates ; i++) {
        if (procps_diskstat_read(disk_stat) < 0)
            xerr(EXIT_FAILURE,
                 _("Unable to read diskstat"));
        if ((partid = procps_diskstat_dev_getbyname(disk_stat, partition_name))
            < 0)
            xerrx(EXIT_FAILURE,
                  _("Partition %s not found"), partition_name);

        printf(format,
               PARTGET(PROCPS_DISKSTAT_READS),
               PARTGET(PROCPS_DISKSTAT_READ_SECTORS),
               PARTGET(PROCPS_DISKSTAT_WRITES),
               PARTGET(PROCPS_DISKSTAT_WRITE_SECTORS)
              );

        if (infinite_updates || i+1 < num_updates)
            sleep(sleep_time);
    }
    procps_diskstat_unref(&disk_stat);
    return 0;
}

static void diskheader(void)
{
    struct tm *tm_ptr;
    time_t the_time;
    char timebuf[32];

    /* Translation Hint: Translating folloging header & fields
     * that follow (marked with max x chars) might not work,
     * unless manual page is translated as well.  */
    const char *header =
        _("disk- ------------reads------------ ------------writes----------- -----IO------");
    const char *wide_header =
        _("disk- -------------------reads------------------- -------------------writes------------------ ------IO-------");
    const char *timestamp_header = _(" -----timestamp-----");

    const char format[] =
        "%5s %6s %6s %7s %7s %6s %6s %7s %7s %6s %6s";
    const char wide_format[] =
        "%5s %9s %9s %11s %11s %9s %9s %11s %11s %7s %7s";

    printf("%s", w_option ? wide_header : header);

    if (t_option) {
        printf("%s", timestamp_header);
    }

    printf("\n");

    printf(w_option ? wide_format : format,
           " ",
           /* Translation Hint: max 6 chars */
           _("total"),
           /* Translation Hint: max 6 chars */
           _("merged"),
           /* Translation Hint: max 7 chars */
           _("sectors"),
           /* Translation Hint: max 7 chars */
           _("ms"),
           /* Translation Hint: max 6 chars */
           _("total"),
           /* Translation Hint: max 6 chars */
           _("merged"),
           /* Translation Hint: max 7 chars */
           _("sectors"),
           /* Translation Hint: max 7 chars */
           _("ms"),
           /* Translation Hint: max 6 chars */
           _("cur"),
           /* Translation Hint: max 6 chars */
           _("sec"));

    if (t_option) {
        (void) time( &the_time );
        tm_ptr = localtime( &the_time );
        if (strftime(timebuf, sizeof(timebuf), "%Z", tm_ptr)) {
            timebuf[strlen(timestamp_header) - 1] = '\0';
        } else {
            timebuf[0] = '\0';
        }
        printf(" %*s", (int)(strlen(timestamp_header) - 1), timebuf);
    }

    printf("\n");
}

static void diskformat(void)
{
#define DSTAT(x) procps_diskstat_dev_get(disk_stat, (x), diskid)
    struct procps_diskstat *disk_stat;
    int i,diskid, disk_count;
    time_t the_time;
    struct tm *tm_ptr;
    char timebuf[32];
    const char format[] = "%-5s %6lu %6lu %7lu %7lu %6lu %6lu %7lu %7lu %6lu %6lu";
    const char wide_format[] = "%-5s %9lu %9lu %lu %lu %9lu %9lu %lu %lu %7lu %7lu";

    if (procps_diskstat_new(&disk_stat) < 0)
        xerr(EXIT_FAILURE,
             _("Unable to create diskstat structure"));

    if (!moreheaders)
        diskheader();
    for (i=0; infinite_updates || i < num_updates ; i++) {
        if (procps_diskstat_read(disk_stat) < 0)
            xerr(EXIT_FAILURE,
                 _("Unable to read diskstat data"));

        if (t_option) {
            (void) time( &the_time );
            tm_ptr = localtime( &the_time );
            strftime(timebuf, sizeof(timebuf), "%Y-%m-%d %H:%M:%S", tm_ptr);
        }
        disk_count = procps_diskstat_dev_count(disk_stat);
        for (diskid = 0; diskid < disk_count; diskid++) {
            if (procps_diskstat_dev_isdisk(disk_stat, diskid) != 1)
                continue; /* not a disk */
            if (moreheaders && ((diskid % height) == 0))
                diskheader();
            printf(w_option ? wide_format : format,
                   procps_diskstat_dev_getname(disk_stat, diskid),
                   DSTAT(PROCPS_DISKSTAT_READS),
                   DSTAT(PROCPS_DISKSTAT_READS_MERGED),
                   DSTAT(PROCPS_DISKSTAT_READ_SECTORS),
                   DSTAT(PROCPS_DISKSTAT_READ_TIME),
                   DSTAT(PROCPS_DISKSTAT_WRITES),
                   DSTAT(PROCPS_DISKSTAT_WRITES_MERGED),
                   DSTAT(PROCPS_DISKSTAT_WRITE_SECTORS),
                   DSTAT(PROCPS_DISKSTAT_WRITE_TIME),
                   DSTAT(PROCPS_DISKSTAT_IO_INPROGRESS) / 1000,
                   DSTAT(PROCPS_DISKSTAT_IO_TIME) / 1000);
            if (t_option)
                printf(" %s\n", timebuf);
            else
                printf("\n");
            fflush(stdout);
        }
        if (infinite_updates || i+1 < num_updates)
            sleep(sleep_time);
    }
#undef DSTAT
    procps_diskstat_unref(&disk_stat);
}

static void slabheader(void)
{
	printf("%-24s %6s %6s %6s %6s\n",
	/* Translation Hint: Translating folloging slab fields that
	 * follow (marked with max x chars) might not work, unless
	 * manual page is translated as well.  */
	       /* Translation Hint: max 24 chars */
	       _("Cache"),
	       /* Translation Hint: max 6 chars */
	       _("Num"),
	       /* Translation Hint: max 6 chars */
	       _("Total"),
	       /* Translation Hint: max 6 chars */
	       _("Size"),
	       /* Translation Hint: max 6 chars */
	       _("Pages"));
}

static void slabformat (void)
{
 #define CHAINS_ALLOC  150
 #define MAX_ITEMS (int)(sizeof(node_items) / sizeof(node_items[0]))
    struct procps_slabinfo *slab_info;
    struct slabnode_stack **v, *p;
    int i, j, nr_slabs;
    enum slabnode_item node_items[] = {
        PROCPS_SLABNODE_AOBJS,    PROCPS_SLABNODE_OBJS,
        PROCPS_SLABNODE_OBJ_SIZE, PROCPS_SLABNODE_OBJS_PER_SLAB,
        PROCPS_SLABNODE_NAME,     PROCPS_SLABNODE_stack_end };
    enum rel_enums {
        slab_AOBJS, slab_OBJS, slab_OSIZE, slab_OPS, slab_NAME };

    if (procps_slabinfo_new(&slab_info) < 0)
        xerrx(EXIT_FAILURE, _("Unable to create slabinfo structure"));
    if (!(v = procps_slabnode_stacks_alloc(slab_info, CHAINS_ALLOC, MAX_ITEMS, node_items)))
        xerrx(EXIT_FAILURE, _("Unable to allocate slabinfo nodes"));

    if (!moreheaders)
        slabheader();

    for (i = 0; infinite_updates || i < num_updates; i++) {
        // this next guy also performs the procps_slabnode_read() call
        if ((nr_slabs = procps_slabnode_stacks_fill(slab_info, v, CHAINS_ALLOC)) < 0)
            xerrx(EXIT_FAILURE, _("Unable to get slabinfo node data, requires root permission"));
        if (!(v = procps_slabnode_stacks_sort(slab_info, v, nr_slabs, PROCPS_SLABNODE_NAME)))
            xerrx(EXIT_FAILURE, _("Unable to sort slab nodes"));

        for (j = 0; j < nr_slabs; j++) {
            p = v[j];
            if (moreheaders && ((j % height) == 0))
                slabheader();
            printf("%-24.24s %6u %6u %6u %6u\n",
                p->head[slab_NAME ].result.str,
                p->head[slab_AOBJS].result.u_int,
                p->head[slab_OBJS ].result.u_int,
                p->head[slab_OSIZE].result.u_int,
                p->head[slab_OPS  ].result.u_int);
        }
        if (infinite_updates || i+1 < num_updates)
            sleep(sleep_time);
    }
    procps_slabinfo_unref(&slab_info);
 #undef CHAINS_ALLOC
 #undef MAX_ITEMS
}

static void disksum_format(void)
{
#define DSTAT(x) procps_diskstat_dev_get(disk_stat, (x), devid)
    struct procps_diskstat *disk_stat;

    if (procps_diskstat_new(&disk_stat) < 0)
        xerr(EXIT_FAILURE,
             _("Unable to create diskstat structure"));

    if (procps_diskstat_read(disk_stat) < 0)
        xerr(EXIT_FAILURE,
             _("Unable to read diskstat"));

    int devid, dev_count, disk_count, part_count ;
    unsigned long reads, merged_reads, read_sectors, milli_reading, writes,
                  merged_writes, written_sectors, milli_writing, inprogress_IO,
                  milli_spent_IO, weighted_milli_spent_IO;

    reads = merged_reads = read_sectors = milli_reading = writes =
        merged_writes = written_sectors = milli_writing = inprogress_IO =
        milli_spent_IO = weighted_milli_spent_IO = 0;
    disk_count = part_count = 0;

    if ((dev_count = procps_diskstat_dev_count(disk_stat)) < 0)
        xerr(EXIT_FAILURE,
             _("Unable to count diskstat devices"));

    for (devid=0; devid < dev_count; devid++) {
        if (procps_diskstat_dev_isdisk(disk_stat, devid) != 1) {
            part_count++;
            continue; /* not a disk */
        }
        disk_count++;
        reads += DSTAT(PROCPS_DISKSTAT_READS);
        merged_reads += DSTAT(PROCPS_DISKSTAT_READS_MERGED);
        read_sectors += DSTAT(PROCPS_DISKSTAT_READ_SECTORS);
        milli_reading += DSTAT(PROCPS_DISKSTAT_READ_TIME);
        writes += DSTAT(PROCPS_DISKSTAT_WRITES);
        merged_writes += DSTAT(PROCPS_DISKSTAT_WRITES_MERGED);
        written_sectors += DSTAT(PROCPS_DISKSTAT_WRITE_SECTORS);
        milli_writing += DSTAT(PROCPS_DISKSTAT_WRITE_TIME);
        inprogress_IO += DSTAT(PROCPS_DISKSTAT_IO_INPROGRESS) / 1000;
        milli_spent_IO += DSTAT(PROCPS_DISKSTAT_IO_TIME) / 1000;
        weighted_milli_spent_IO += DSTAT(PROCPS_DISKSTAT_IO_TIME) / 1000;
    }
    printf(_("%13d disks\n"), disk_count);
    printf(_("%13d partitions\n"), part_count);
    printf(_("%13lu reads\n"), reads);
    printf(_("%13lu merged reads\n"), merged_reads);
    printf(_("%13lu read sectors\n"), read_sectors);
    printf(_("%13lu milli reading\n"), milli_reading);
    printf(_("%13lu writes\n"), writes);
    printf(_("%13lu merged writes\n"), merged_writes);
    printf(_("%13lu written sectors\n"), written_sectors);
    printf(_("%13lu milli writing\n"), milli_writing);
    printf(_("%13lu inprogress IO\n"), inprogress_IO);
    printf(_("%13lu milli spent IO\n"), milli_spent_IO);
    printf(_("%13lu milli weighted IO\n"), weighted_milli_spent_IO);
#undef DSTAT
    procps_diskstat_unref(&disk_stat);
}

static enum stat_item Sum_stat_items[] = {
    PROCPS_STAT_TIC_USER,
    PROCPS_STAT_TIC_NICE,
    PROCPS_STAT_TIC_SYSTEM,
    PROCPS_STAT_TIC_IDLE,
    PROCPS_STAT_TIC_IOWAIT,
    PROCPS_STAT_TIC_IRQ,
    PROCPS_STAT_TIC_SOFTIRQ,
    PROCPS_STAT_TIC_STOLEN,
    PROCPS_STAT_TIC_GUEST,
    PROCPS_STAT_TIC_GUEST_NICE,
    PROCPS_STAT_SYS_CTX_SWITCHES,
    PROCPS_STAT_SYS_INTERRUPTS,
    PROCPS_STAT_SYS_TIME_OF_BOOT,
    PROCPS_STAT_SYS_PROC_CREATED
};

enum Rel_sumstatitems {
    sstat_USR, sstat_NIC, sstat_SYS, sstat_IDL,
    sstat_IOW, sstat_IRQ, sstat_SRQ, sstat_STO,
    sstat_GST, sstat_GNI,
    sstat_CTX, sstat_INT, sstat_TOB,
    sstat_PCR
};

static enum meminfo_item Sum_mem_items[] = {
    PROCPS_MEMINFO_MEM_TOTAL,
    PROCPS_MEMINFO_MEM_USED,
    PROCPS_MEMINFO_MEM_ACTIVE,
    PROCPS_MEMINFO_MEM_INACTIVE,
    PROCPS_MEMINFO_MEM_FREE,
    PROCPS_MEMINFO_MEM_BUFFERS,
    PROCPS_MEMINFO_MEM_CACHED,
    PROCPS_MEMINFO_SWAP_TOTAL,
    PROCPS_MEMINFO_SWAP_USED,
    PROCPS_MEMINFO_SWAP_FREE,
};

enum Rel_summemitems {
    smem_MTOT, smem_MUSE, smem_MACT, smem_MIAC, smem_MFRE,
    smem_MBUF, smem_MCAC,
    smem_STOT, smem_SUSE, smem_SFRE
};

static void sum_format(void)
{
#define TICv(E) PROCPS_STAT_VAL(E, ull_int, stat_stack)
#define SYSv(E) PROCPS_STAT_VAL(E, ul_int, stat_stack)
#define MEMv(E) unitConvert(PROCPS_STAT_VAL(E, ul_int, mem_stack))
    struct procps_statinfo *sys_info = NULL;
    struct procps_vmstat *vm_info = NULL;
    struct procps_meminfo *mem_info = NULL;
    struct stat_stack *stat_stack;
    struct meminfo_stack *mem_stack;

    if (procps_stat_new(&sys_info) < 0)
        xerrx(EXIT_FAILURE,
        _("Unable to create system stat structure"));
    if ((stat_stack = procps_stat_select(sys_info, Sum_stat_items, 14)) ==
        NULL)
        xerrx(EXIT_FAILURE,
              _("Unable to select stat information"));

    if (procps_vmstat_new(&vm_info) < 0)
        xerrx(EXIT_FAILURE,
        _("Unable to create vmstat structure"));
    if (procps_meminfo_new(&mem_info) < 0)
        xerrx(EXIT_FAILURE, _("Unable to create meminfo structure"));
    if ((mem_stack = procps_meminfo_select(mem_info, Sum_mem_items, 10)) ==
        NULL)
        xerrx(EXIT_FAILURE,
              _("Unable to select memory information"));

    printf(_("%13lu %s total memory\n"), MEMv(smem_MTOT), szDataUnit);
    printf(_("%13lu %s used memory\n"), MEMv(smem_MUSE), szDataUnit);
    printf(_("%13lu %s active memory\n"), MEMv(smem_MACT), szDataUnit);
    printf(_("%13lu %s inactive memory\n"), MEMv(smem_MIAC), szDataUnit);
    printf(_("%13lu %s free memory\n"), MEMv(smem_MFRE), szDataUnit);
    printf(_("%13lu %s buffer memory\n"), MEMv(smem_MBUF), szDataUnit);
    printf(_("%13lu %s swap cache\n"), MEMv(smem_MCAC), szDataUnit);
    printf(_("%13lu %s total swap\n"), MEMv(smem_STOT), szDataUnit);
    printf(_("%13lu %s used swap\n"), MEMv(smem_SUSE), szDataUnit);
    printf(_("%13lu %s free swap\n"), MEMv(smem_SFRE), szDataUnit);
    printf(_("%13lld non-nice user cpu ticks\n"), TICv(sstat_USR));
    printf(_("%13lld nice user cpu ticks\n"), TICv(sstat_NIC));
    printf(_("%13lld system cpu ticks\n"), TICv(sstat_SYS));
    printf(_("%13lld idle cpu ticks\n"), TICv(sstat_IDL));
    printf(_("%13lld IO-wait cpu ticks\n"), TICv(sstat_IOW));
    printf(_("%13lld IRQ cpu ticks\n"), TICv(sstat_IRQ));
    printf(_("%13lld softirq cpu ticks\n"), TICv(sstat_SRQ));
    printf(_("%13lld stolen cpu ticks\n"), TICv(sstat_STO));
    printf(_("%13lld non-nice guest cpu ticks\n"), TICv(sstat_GST));
    printf(_("%13lld nice guest cpu ticks\n"), TICv(sstat_GNI));
    printf(_("%13lu pages paged in\n"), procps_vmstat_get(vm_info, PROCPS_VMSTAT_PGPGIN));
    printf(_("%13lu pages paged out\n"), procps_vmstat_get(vm_info, PROCPS_VMSTAT_PGPGOUT));
    printf(_("%13lu pages swapped in\n"), procps_vmstat_get(vm_info, PROCPS_VMSTAT_PSWPIN));
    printf(_("%13lu pages swapped out\n"), procps_vmstat_get(vm_info, PROCPS_VMSTAT_PSWPOUT));
    printf(_("%13lu interrupts\n"), SYSv(sstat_INT));
    printf(_("%13lu CPU context switches\n"), SYSv(sstat_CTX));
    printf(_("%13lu boot time\n"), SYSv(sstat_TOB));
    printf(_("%13lu forks\n"), SYSv(sstat_PCR));

    /* Cleanup */
    procps_stat_unref(&sys_info);
    procps_vmstat_unref(&vm_info);
    procps_meminfo_unref(&mem_info);
#undef TICv
#undef SYSv
#undef MEMv
}

static void fork_format(void)
{
    struct procps_statinfo *sys_info = NULL;

    if (procps_stat_new(&sys_info) < 0)
	xerrx(EXIT_FAILURE,
		_("Unable to create system stat structure"));

    printf(_("%13lld forks\n"), procps_stat_get(sys_info, PROCPS_STAT_SYS_PROC_CREATED));
    /* Cleanup */
    procps_stat_unref(&sys_info);
}

static int winhi(void)
{
	struct winsize win;
	int rows = 24;

	if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &win) != -1 && 0 < win.ws_row)
		rows = win.ws_row;

	return rows;
}

int main(int argc, char *argv[])
{
	char *partition = NULL;
	int c;
	long tmp;

	static const struct option longopts[] = {
		{"active", no_argument, NULL, 'a'},
		{"forks", no_argument, NULL, 'f'},
		{"slabs", no_argument, NULL, 'm'},
		{"one-header", no_argument, NULL, 'n'},
		{"stats", no_argument, NULL, 's'},
		{"disk", no_argument, NULL, 'd'},
		{"disk-sum", no_argument, NULL, 'D'},
		{"partition", required_argument, NULL, 'p'},
		{"unit", required_argument, NULL, 'S'},
		{"wide", no_argument, NULL, 'w'},
		{"timestamp", no_argument, NULL, 't'},
		{"help", no_argument, NULL, 'h'},
		{"version", no_argument, NULL, 'V'},
		{NULL, 0, NULL, 0}
	};

#ifdef HAVE_PROGRAM_INVOCATION_NAME
	program_invocation_name = program_invocation_short_name;
#endif
	setlocale (LC_ALL, "");
	bindtextdomain(PACKAGE, LOCALEDIR);
	textdomain(PACKAGE);
	atexit(close_stdout);

	while ((c =
		getopt_long(argc, argv, "afmnsdDp:S:wthV", longopts,
			    NULL)) != EOF)
		switch (c) {
		case 'V':
			printf(PROCPS_NG_VERSION);
			return EXIT_SUCCESS;
		case 'h':
			usage(stdout);
		case 'd':
			statMode |= DISKSTAT;
			break;
		case 'a':
			/* active/inactive mode */
			a_option = 1;
			break;
		case 'f':
			/* FIXME: check for conflicting args */
			fork_format();
			exit(0);
		case 'm':
			statMode |= SLABSTAT;
			break;
		case 'D':
			statMode |= DISKSUMSTAT;
			break;
		case 'n':
			/* print only one header */
			moreheaders = FALSE;
			break;
		case 'p':
			statMode |= PARTITIONSTAT;
			partition = optarg;
			if (memcmp(partition, "/dev/", 5) == 0)
				partition += 5;
			break;
		case 'S':
			switch (optarg[0]) {
			case 'b':
			case 'B':
				dataUnit = UNIT_B;
				break;
			case 'k':
				dataUnit = UNIT_k;
				break;
			case 'K':
				dataUnit = UNIT_K;
				break;
			case 'm':
				dataUnit = UNIT_m;
				break;
			case 'M':
				dataUnit = UNIT_M;
				break;
			default:
				xerrx(EXIT_FAILURE,
				     /* Translation Hint: do not change argument characters */
				     _("-S requires k, K, m or M (default is KiB)"));
			}
			szDataUnit[0] = optarg[0];
			break;
		case 's':
			statMode |= VMSUMSTAT;
			break;
		case 'w':
			w_option = 1;
			break;
		case 't':
			t_option = 1;
			break;
		default:
			/* no other aguments defined yet. */
			usage(stderr);
		}

	if (optind < argc) {
		tmp = strtol_or_err(argv[optind++], _("failed to parse argument"));
		if (tmp < 1)
			xerrx(EXIT_FAILURE, _("delay must be positive integer"));
		else if (UINT_MAX < tmp)
			xerrx(EXIT_FAILURE, _("too large delay value"));
		sleep_time = tmp;
		infinite_updates = 1;
	}
    num_updates = 1;
	if (optind < argc) {
		num_updates = strtol_or_err(argv[optind++], _("failed to parse argument"));
		infinite_updates = 0;
	}
	if (optind < argc)
		usage(stderr);

	if (moreheaders) {
		int wheight = winhi() - 3;
		height = ((wheight > 0) ? wheight : 22);
	}
	setlinebuf(stdout);
	switch (statMode) {
	case (VMSTAT):
		new_format();
		break;
	case (VMSUMSTAT):
		sum_format();
		break;
	case (DISKSTAT):
		diskformat();
		break;
	case (PARTITIONSTAT):
		if (diskpartition_format(partition) == -1)
			printf(_("partition was not found\n"));
		break;
	case (SLABSTAT):
		slabformat();
		break;
	case (DISKSUMSTAT):
		disksum_format();
		break;
	default:
		usage(stderr);
		break;
	}
	return 0;
}