source: MondoRescue/branches/3.0/mindi-busybox/miscutils/ttysize.c@ 2899

/* vi: set sw=4 ts=4: */
/*
 * Replacement for "stty size", which is awkward for shell script use.
 * - Allows to request width, height, or both, in any order.
 * - Does not complain on error, but returns width 80, height 24.
 * - Size: less than 200 bytes
 *
 * Copyright (C) 2007 by Denys Vlasenko <vda.linux@googlemail.com>
 *
 * Licensed under GPLv2, see file LICENSE in this source tree.
 */
#include "libbb.h"

int ttysize_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int ttysize_main(int argc UNUSED_PARAM, char **argv)
{
    unsigned w, h;
    struct winsize wsz;

    w = 80;
    h = 24;
    if (!ioctl(0, TIOCGWINSZ, &wsz)) {
        w = wsz.ws_col;
        h = wsz.ws_row;
    }

    if (!argv[1]) {
        printf("%u %u", w, h);
    } else {
        const char *fmt, *arg;

        fmt = "%u %u" + 3; /* "%u" */
        while ((arg = *++argv) != NULL) {
            char c = arg[0];
            if (c == 'w')
                printf(fmt, w);
            if (c == 'h')
                printf(fmt, h);
            fmt = "%u %u" + 2; /* " %u" */
        }
    }
    bb_putchar('\n');
    return 0;
}