p-wc.cc

#include "u-lib.hh"

static int unparse_counts(char* buf, size_t bufsz, int mode,
                          const size_t* counts, const char* fname) {
    const char* sep = strlen(fname) ? " " : "";
    if (mode < 0) {
        return snprintf(buf, bufsz, "%8zu %7zu %7zu%s%s\n",
                        counts[0], counts[1], counts[2], sep, fname);
    } else if (*sep) {
        return snprintf(buf, bufsz, "%8zu%s%s\n",
                        counts[mode], sep, fname);
    } else {
        return snprintf(buf, bufsz, "%zu\n", counts[mode]);
    }
}

void process_main(int argc, char** argv) {
    static char buf[4096];

    int mode = -1;
    int argno = 1;
    if (argno < argc) {
        if (strcmp(argv[argno], "-c") == 0) {
            mode = 0;
            ++argno;
        } else if (strcmp(argv[argno], "-w") == 0) {
            mode = 1;
            ++argno;
        } else if (strcmp(argv[argno], "-l") == 0) {
            mode = 2;
            ++argno;
        } else if (argv[argno][0] == '-' && argv[argno][1]) {
            sys_write(2, "wc: bad argument\n", 17);
            sys_exit(1);
        }
    }

    size_t totals[3] = {0, 0, 0};
    unsigned nfiles = 0;

    while (argno < argc || nfiles == 0) {
        int fd = 0;
        bool newfd = false;
        if (argno < argc && strcmp(argv[argno], "-") != 0) {
            fd = sys_open(argv[argno], OF_READ);
            if (fd < 0) {
                dprintf(2, "%s: error %d\n", argv[argno], fd);
                sys_exit(1);
            }
            newfd = true;
        }

        size_t counts[3] = {0, 0, 0};
        bool inword = false;
        while (true) {
            ssize_t n = sys_read(fd, buf, sizeof(buf));
            if (n == 0 || (n < 0 && n != E_AGAIN)) {
                break;
            }
            for (ssize_t i = 0; i < n; ++i) {
                ++counts[0];
                if (!inword && !isspace(buf[i])) {
                    ++counts[1];
                }
                inword = !isspace(buf[i]);
                if (buf[i] == '\n') {
                    ++counts[2];
                }
            }
        }

        if (newfd) {
            sys_close(fd);
        }

        int n = unparse_counts(buf, sizeof(buf), mode, counts,
                               argno < argc ? argv[argno] : "");
        ssize_t w = sys_write(1, buf, n);
        assert(w == n);

        for (int i = 0; i < 3; ++i) {
            totals[i] += counts[i];
        }
        ++nfiles;
        ++argno;
    }

    if (nfiles > 1) {
        int n = unparse_counts(buf, sizeof(buf), mode, totals, "total");
        ssize_t w = sys_write(1, buf, n);
        assert(w == n);
    }

    sys_exit(0);
}