Configurable age range represented by the colour coding in the HTML

[sgt/agedu] / agedu.c

/*
 * Main program for agedu.
 */

#define _GNU_SOURCE
#include <stdio.h>
#include <errno.h>
#include <stdarg.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <time.h>
#include <assert.h>

#include <unistd.h>
#include <sys/types.h>
#include <fcntl.h>
#include <sys/mman.h>
#include <termios.h>
#include <sys/ioctl.h>
#include <fnmatch.h>

#include "du.h"
#include "trie.h"
#include "index.h"
#include "malloc.h"
#include "html.h"
#include "httpd.h"

#define PNAME "agedu"

#define lenof(x) (sizeof((x))/sizeof(*(x)))

void fatal(const char *fmt, ...)
{
    va_list ap;
    fprintf(stderr, "%s: ", PNAME);
    va_start(ap, fmt);
    vfprintf(stderr, fmt, ap);
    va_end(ap);
    fprintf(stderr, "\n");
    exit(1);
}

struct inclusion_exclusion {
    int type;
    const char *wildcard;
    int path;
};

struct ctx {
    triebuild *tb;
    dev_t datafile_dev, filesystem_dev;
    ino_t datafile_ino;
    time_t last_output_update;
    int progress, progwidth;
    struct inclusion_exclusion *inex;
    int ninex;
    int crossfs;
};

static int gotdata(void *vctx, const char *pathname, const struct stat64 *st)
{
    struct ctx *ctx = (struct ctx *)vctx;
    struct trie_file file;
    time_t t;
    int i, include;
    const char *filename;

    /*
     * Filter out our own data file.
     */
    if (st->st_dev == ctx->datafile_dev && st->st_ino == ctx->datafile_ino)
        return 0;

    /*
     * Don't cross the streams^W^Wany file system boundary.
     */
    if (!ctx->crossfs && st->st_dev != ctx->filesystem_dev)
        return 0;

    file.blocks = st->st_blocks;
    file.atime = st->st_atime;

    /*
     * Filter based on wildcards.
     */
    include = 1;
    filename = strrchr(pathname, '/');
    if (!filename)
        filename = pathname;
    else
        filename++;
    for (i = 0; i < ctx->ninex; i++) {
        if (fnmatch(ctx->inex[i].wildcard,
                    ctx->inex[i].path ? pathname : filename, 0) == 0)
            include = ctx->inex[i].type;
    }
    if (include == -1)
        return 0;                      /* ignore this entry and any subdirs */
    if (include == 0) {
        /*
         * Here we are supposed to be filtering an entry out, but
         * still recursing into it if it's a directory. However,
         * we can't actually leave out any directory whose
         * subdirectories we then look at. So we cheat, in that
         * case, by setting the size to zero.
         */
        if (!S_ISDIR(st->st_mode))
            return 0;                  /* just ignore */
        else
            file.blocks = 0;
    }

    triebuild_add(ctx->tb, pathname, &file);

    t = time(NULL);
    if (t != ctx->last_output_update) {
        if (ctx->progress) {
            fprintf(stderr, "%-*.*s\r", ctx->progwidth, ctx->progwidth,
                    pathname);
            fflush(stderr);
        }
        ctx->last_output_update = t;
    }

    return 1;
}

static void text_query(const void *mappedfile, const char *querydir,
                       time_t t, int depth)
{
    size_t maxpathlen;
    char *pathbuf;
    unsigned long xi1, xi2;
    unsigned long long s1, s2;

    maxpathlen = trie_maxpathlen(mappedfile);
    pathbuf = snewn(maxpathlen + 1, char);

    /*
     * We want to query everything between the supplied filename
     * (inclusive) and that filename with a ^A on the end
     * (exclusive). So find the x indices for each.
     */
    sprintf(pathbuf, "%s\001", querydir);
    xi1 = trie_before(mappedfile, querydir);
    xi2 = trie_before(mappedfile, pathbuf);

    /*
     * Now do the lookups in the age index.
     */
    s1 = index_query(mappedfile, xi1, t);
    s2 = index_query(mappedfile, xi2, t);

    if (s1 == s2)
        return;                        /* no space taken up => no display */

    /* Display in units of 2 512-byte blocks = 1Kb */
    printf("%-11llu %s\n", (s2 - s1) / 2, querydir);

    if (depth > 0) {
        /*
         * Now scan for first-level subdirectories and report
         * those too.
         */
        xi1++;
        while (xi1 < xi2) {
            trie_getpath(mappedfile, xi1, pathbuf);
            text_query(mappedfile, pathbuf, t, depth-1);
            strcat(pathbuf, "\001");
            xi1 = trie_before(mappedfile, pathbuf);
        }
    }
}

/*
 * Largely frivolous way to define all my command-line options. I
 * present here a parametric macro which declares a series of
 * _logical_ option identifiers, and for each one declares zero or
 * more short option characters and zero or more long option
 * words. Then I repeatedly invoke that macro with its arguments
 * defined to be various other macros, which allows me to
 * variously:
 * 
 *  - define an enum allocating a distinct integer value to each
 *    logical option id
 *  - define a string consisting of precisely all the short option
 *    characters
 *  - define a string array consisting of all the long option
 *    strings
 *  - define (with help from auxiliary enums) integer arrays
 *    parallel to both of the above giving the logical option id
 *    for each physical short and long option
 *  - define an array indexed by logical option id indicating
 *    whether the option in question takes a value
 *  - define a function which prints out brief online help for all
 *    the options.
 *
 * It's not at all clear to me that this trickery is actually
 * particularly _efficient_ - it still, after all, requires going
 * linearly through the option list at run time and doing a
 * strcmp, whereas in an ideal world I'd have liked the lists of
 * long and short options to be pre-sorted so that a binary search
 * or some other more efficient lookup was possible. (Not that
 * asymptotic algorithmic complexity is remotely vital in option
 * parsing, but if I were doing this in, say, Lisp or something
 * with an equivalently powerful preprocessor then once I'd had
 * the idea of preparing the option-parsing data structures at
 * compile time I would probably have made the effort to prepare
 * them _properly_. I could have Perl generate me a source file
 * from some sort of description, I suppose, but that would seem
 * like overkill. And in any case, it's more of a challenge to
 * achieve as much as possible by cunning use of cpp and enum than
 * to just write some sensible and logical code in a Turing-
 * complete language. I said it was largely frivolous :-)
 *
 * This approach does have the virtue that it brings together the
 * option ids, option spellings and help text into a single
 * combined list and defines them all in exactly one place. If I
 * want to add a new option, or a new spelling for an option, I
 * only have to modify the main OPTHELP macro below and then add
 * code to process the new logical id.
 *
 * (Though, really, even that isn't ideal, since it still involves
 * modifying the source file in more than one place. In a
 * _properly_ ideal world, I'd be able to interleave the option
 * definitions with the code fragments that process them. And then
 * not bother defining logical identifiers for them at all - those
 * would be automatically generated, since I wouldn't have any
 * need to specify them manually in another part of the code.)
 */

#define OPTHELP(NOVAL, VAL, SHORT, LONG, HELPPFX, HELPARG, HELPLINE, HELPOPT) \
    HELPPFX("usage") HELPLINE("agedu [options] action") \
    HELPPFX("actions") \
    VAL(SCAN) SHORT(s) LONG(scan) \
        HELPARG("directory") HELPOPT("scan and index a directory") \
    NOVAL(DUMP) SHORT(d) LONG(dump) HELPOPT("dump the index file") \
    VAL(TEXT) SHORT(t) LONG(text) \
        HELPARG("subdir") HELPOPT("print a plain text report on a subdirectory") \
    VAL(HTML) SHORT(H) LONG(html) \
        HELPARG("subdir") HELPOPT("print an HTML report on a subdirectory") \
    NOVAL(HTTPD) SHORT(w) LONG(web) LONG(server) LONG(httpd) \
        HELPOPT("serve reports from a temporary web server") \
    HELPPFX("options") \
    VAL(DATAFILE) SHORT(f) LONG(file) \
        HELPARG("filename") HELPOPT("[all modes] specify index file") \
    NOVAL(PROGRESS) LONG(progress) LONG(scan_progress) \
        HELPOPT("[--scan] report progress on stderr") \
    NOVAL(NOPROGRESS) LONG(no_progress) LONG(no_scan_progress) \
        HELPOPT("[--scan] do not report progress") \
    NOVAL(TTYPROGRESS) LONG(tty_progress) LONG(tty_scan_progress) \
                       LONG(progress_tty) LONG(scan_progress_tty) \
        HELPOPT("[--scan] report progress if stderr is a tty") \
    NOVAL(CROSSFS) LONG(cross_fs) \
        HELPOPT("[--scan] cross filesystem boundaries") \
    NOVAL(NOCROSSFS) LONG(no_cross_fs) \
        HELPOPT("[--scan] stick to one filesystem") \
    VAL(INCLUDE) LONG(include) \
        HELPARG("wildcard") HELPOPT("[--scan] include files matching pattern") \
    VAL(INCLUDEPATH) LONG(include_path) \
        HELPARG("wildcard") HELPOPT("[--scan] include pathnames matching pattern") \
    VAL(EXCLUDE) LONG(exclude) \
        HELPARG("wildcard") HELPOPT("[--scan] exclude files matching pattern") \
    VAL(EXCLUDEPATH) LONG(exclude_path) \
        HELPARG("wildcard") HELPOPT("[--scan] exclude pathnames matching pattern") \
    VAL(PRUNE) LONG(prune) \
        HELPARG("wildcard") HELPOPT("[--scan] prune files matching pattern") \
    VAL(PRUNEPATH) LONG(prune_path) \
        HELPARG("wildcard") HELPOPT("[--scan] prune pathnames matching pattern") \
    VAL(MINAGE) SHORT(a) LONG(age) LONG(min_age) LONG(minimum_age) \
        HELPARG("age") HELPOPT("[--text] include only files older than this") \
    VAL(AGERANGE) SHORT(r) LONG(age_range) LONG(range) LONG(ages) \
        HELPARG("age[-age]") HELPOPT("[--html,--web] set limits of colour coding") \
    VAL(AUTH) LONG(auth) LONG(http_auth) LONG(httpd_auth) \
              LONG(server_auth) LONG(web_auth) \
        HELPARG("type") HELPOPT("[--web] specify HTTP authentication method") \
    HELPPFX("also") \
    NOVAL(HELP) SHORT(h) LONG(help) HELPOPT("display this help text") \
    NOVAL(VERSION) SHORT(V) LONG(version) HELPOPT("report version number") \
    NOVAL(LICENCE) LONG(licence) LONG(license) \
        HELPOPT("display (MIT) licence text") \

#define IGNORE(x)
#define DEFENUM(x) OPT_ ## x,
#define ZERO(x) 0,
#define ONE(x) 1,
#define STRING(x) #x ,
#define STRINGNOCOMMA(x) #x
#define SHORTNEWOPT(x) SHORTtmp_ ## x = OPT_ ## x,
#define SHORTTHISOPT(x) SHORTtmp2_ ## x, SHORTVAL_ ## x = SHORTtmp2_ ## x - 1,
#define SHORTOPTVAL(x) SHORTVAL_ ## x,
#define SHORTTMP(x) SHORTtmp3_ ## x,
#define LONGNEWOPT(x) LONGtmp_ ## x = OPT_ ## x,
#define LONGTHISOPT(x) LONGtmp2_ ## x, LONGVAL_ ## x = LONGtmp2_ ## x - 1,
#define LONGOPTVAL(x) LONGVAL_ ## x,
#define LONGTMP(x) SHORTtmp3_ ## x,

#define OPTIONS(NOVAL, VAL, SHORT, LONG) \
    OPTHELP(NOVAL, VAL, SHORT, LONG, IGNORE, IGNORE, IGNORE, IGNORE)

enum { OPTIONS(DEFENUM,DEFENUM,IGNORE,IGNORE) NOPTIONS };
enum { OPTIONS(IGNORE,IGNORE,SHORTTMP,IGNORE) NSHORTOPTS };
enum { OPTIONS(IGNORE,IGNORE,IGNORE,LONGTMP) NLONGOPTS };
static const int opthasval[NOPTIONS] = {OPTIONS(ZERO,ONE,IGNORE,IGNORE)};
static const char shortopts[] = {OPTIONS(IGNORE,IGNORE,STRINGNOCOMMA,IGNORE)};
static const char *const longopts[] = {OPTIONS(IGNORE,IGNORE,IGNORE,STRING)};
enum { OPTIONS(SHORTNEWOPT,SHORTNEWOPT,SHORTTHISOPT,IGNORE) };
enum { OPTIONS(LONGNEWOPT,LONGNEWOPT,IGNORE,LONGTHISOPT) };
static const int shortvals[] = {OPTIONS(IGNORE,IGNORE,SHORTOPTVAL,IGNORE)};
static const int longvals[] = {OPTIONS(IGNORE,IGNORE,IGNORE,LONGOPTVAL)};

static void usage(FILE *fp)
{
    char longbuf[80];
    const char *prefix, *shortopt, *longopt, *optarg;
    int i, optex;

#define HELPRESET prefix = shortopt = longopt = optarg = NULL, optex = -1
#define HELPNOVAL(s) optex = 0;
#define HELPVAL(s) optex = 1;
#define HELPSHORT(s) if (!shortopt) shortopt = "-" #s;
#define HELPLONG(s) if (!longopt) { \
    strcpy(longbuf, "--" #s); longopt = longbuf; \
    for (i = 0; longbuf[i]; i++) if (longbuf[i] == '_') longbuf[i] = '-'; }
#define HELPPFX(s) prefix = s;
#define HELPARG(s) optarg = s;
#define HELPLINE(s) assert(optex == -1); \
    fprintf(fp, "%7s%c %s\n", prefix?prefix:"", prefix?':':' ', s); \
    HELPRESET;
#define HELPOPT(s) assert((optex == 1 && optarg) || (optex == 0 && !optarg)); \
    assert(shortopt || longopt); \
    i = fprintf(fp, "%7s%c %s%s%s%s%s", prefix?prefix:"", prefix?':':' ', \
        shortopt?shortopt:"", shortopt&&longopt?", ":"", longopt?longopt:"", \
        optarg?" ":"", optarg?optarg:""); \
    fprintf(fp, "%*s %s\n", i<32?32-i:0,"",s); HELPRESET;

    HELPRESET;
    OPTHELP(HELPNOVAL, HELPVAL, HELPSHORT, HELPLONG,
            HELPPFX, HELPARG, HELPLINE, HELPOPT);

#undef HELPRESET
#undef HELPNOVAL
#undef HELPVAL
#undef HELPSHORT
#undef HELPLONG
#undef HELPPFX
#undef HELPARG
#undef HELPLINE
#undef HELPOPT
}

static time_t parse_age(time_t now, const char *agestr)
{
    time_t t;
    struct tm tm;
    int nunits;
    char unit[2];

    t = now;

    if (2 != sscanf(agestr, "%d%1[DdWwMmYy]", &nunits, unit)) {
        fprintf(stderr, "%s: age specification should be a number followed by"
                " one of d,w,m,y\n", PNAME);
        exit(1);
    }

    if (unit[0] == 'd') {
        t -= 86400 * nunits;
    } else if (unit[0] == 'w') {
        t -= 86400 * 7 * nunits;
    } else {
        int ym;

        tm = *localtime(&t);
        ym = tm.tm_year * 12 + tm.tm_mon;

        if (unit[0] == 'm')
            ym -= nunits;
        else
            ym -= 12 * nunits;

        tm.tm_year = ym / 12;
        tm.tm_mon = ym % 12;

        t = mktime(&tm);
    }

    return t;
}

int main(int argc, char **argv)
{
    int fd, count;
    struct ctx actx, *ctx = &actx;
    struct stat st;
    off_t totalsize, realsize;
    void *mappedfile;
    triewalk *tw;
    indexbuild *ib;
    const struct trie_file *tf;
    char *filename = "agedu.dat";
    char *scandir = NULL;
    char *querydir = NULL;
    int doing_opts = 1;
    enum { USAGE, TEXT, HTML, SCAN, DUMP, HTTPD } mode = USAGE;
    time_t now = time(NULL);
    time_t textcutoff = now, htmlnewest = now, htmloldest = now;
    int htmlautoagerange = 1;
    int auth = HTTPD_AUTH_MAGIC | HTTPD_AUTH_BASIC;
    int progress = 1;
    struct inclusion_exclusion *inex = NULL;
    int ninex = 0, inexsize = 0;
    int crossfs = 0;

#ifdef DEBUG_MAD_OPTION_PARSING_MACROS
    {
        static const char *const optnames[NOPTIONS] = {
            OPTIONS(STRING,STRING,IGNORE,IGNORE)
        };
        int i;
        for (i = 0; i < NSHORTOPTS; i++)
            printf("-%c == %s [%s]\n", shortopts[i], optnames[shortvals[i]],
                   opthasval[shortvals[i]] ? "value" : "no value");
        for (i = 0; i < NLONGOPTS; i++)
            printf("--%s == %s [%s]\n", longopts[i], optnames[longvals[i]],
                   opthasval[longvals[i]] ? "value" : "no value");
    }
#endif

    while (--argc > 0) {
        char *p = *++argv;

        if (doing_opts && *p == '-') {
            int wordstart = 1;

            if (!strcmp(p, "--")) {
                doing_opts = 0;
                continue;
            }

            p++;
            while (*p) {
                int optid = -1;
                int i;
                char *optval;

                if (wordstart && *p == '-') {
                    /*
                     * GNU-style long option.
                     */
                    p++;
                    optval = strchr(p, '=');
                    if (optval)
                        *optval++ = '\0';

                    for (i = 0; i < NLONGOPTS; i++) {
                        const char *opt = longopts[i], *s = p;
                        int match = 1;
                        /*
                         * The underscores in the option names
                         * defined above may be given by the user
                         * as underscores or dashes, or omitted
                         * entirely.
                         */
                        while (*opt) {
                            if (*opt == '_') {
                                if (*s == '-' || *s == '_')
                                    s++;
                            } else {
                                if (*opt != *s) {
                                    match = 0;
                                    break;
                                }
                                s++;
                            }
                            opt++;
                        }
                        if (match && !*s) {
                            optid = longvals[i];
                            break;
                        }
                    }

                    if (optid < 0) {
                        fprintf(stderr, "%s: unrecognised option '--%s'\n",
                                PNAME, p);
                        return 1;
                    }

                    if (!opthasval[optid]) {
                        if (optval) {
                            fprintf(stderr, "%s: unexpected argument to option"
                                    " '--%s'\n", PNAME, p);
                            return 1;
                        }
                    } else {
                        if (!optval) {
                            if (--argc > 0) {
                                optval = *++argv;
                            } else {
                                fprintf(stderr, "%s: option '--%s' expects"
                                        " an argument\n", PNAME, p);
                                return 1;
                            }
                        }
                    }

                    p += strlen(p);    /* finished with this argument word */
                } else {
                    /*
                     * Short option.
                     */
                    char c = *p++;

                    for (i = 0; i < NSHORTOPTS; i++)
                        if (c == shortopts[i]) {
                            optid = shortvals[i];
                            break;
                        }

                    if (optid < 0) {
                        fprintf(stderr, "%s: unrecognised option '-%c'\n",
                                PNAME, c);
                        return 1;
                    }

                    if (opthasval[optid]) {
                        if (*p) {
                            optval = p;
                            p += strlen(p);
                        } else if (--argc > 0) {
                            optval = *++argv;
                        } else {
                            fprintf(stderr, "%s: option '-%c' expects"
                                    " an argument\n", PNAME, c);
                            return 1;
                        }
                    } else {
                        optval = NULL;
                    }
                }

                wordstart = 0;

                /*
                 * Now actually process the option.
                 */
                switch (optid) {
                  case OPT_HELP:
                    usage(stdout);
                    return 0;
                  case OPT_VERSION:
                    printf("FIXME: version();\n");
                    return 0;
                  case OPT_LICENCE:
                    printf("FIXME: licence();\n");
                    return 0;
                  case OPT_SCAN:
                    mode = SCAN;
                    scandir = optval;
                    break;
                  case OPT_DUMP:
                    mode = DUMP;
                    break;
                  case OPT_TEXT:
                    querydir = optval;
                    mode = TEXT;
                    break;
                  case OPT_HTML:
                    mode = HTML;
                    querydir = optval;
                    break;
                  case OPT_HTTPD:
                    mode = HTTPD;
                    break;
                  case OPT_PROGRESS:
                    progress = 2;
                    break;
                  case OPT_NOPROGRESS:
                    progress = 0;
                    break;
                  case OPT_TTYPROGRESS:
                    progress = 1;
                    break;
                  case OPT_CROSSFS:
                    crossfs = 1;
                    break;
                  case OPT_NOCROSSFS:
                    crossfs = 0;
                    break;
                  case OPT_DATAFILE:
                    filename = optval;
                    break;
                  case OPT_MINAGE:
                    textcutoff = parse_age(now, optval);
                    break;
                  case OPT_AGERANGE:
                    if (!strcmp(optval, "auto")) {
                        htmlautoagerange = 1;
                    } else {
                        char *q = optval + strcspn(optval, "-:");
                        if (*q)
                            *q++ = '\0';
                        htmloldest = parse_age(now, optval);
                        htmlnewest = *q ? parse_age(now, q) : now;
                        htmlautoagerange = 0;
                    }
                    break;
                  case OPT_AUTH:
                    if (!strcmp(optval, "magic"))
                        auth = HTTPD_AUTH_MAGIC;
                    else if (!strcmp(optval, "basic"))
                        auth = HTTPD_AUTH_BASIC;
                    else if (!strcmp(optval, "none"))
                        auth = HTTPD_AUTH_NONE;
                    else if (!strcmp(optval, "default"))
                        auth = HTTPD_AUTH_MAGIC | HTTPD_AUTH_BASIC;
                    else if (!strcmp(optval, "help") ||
                             !strcmp(optval, "list")) {
                        printf("agedu: supported HTTP authentication types"
                               " are:\n"
                               "       magic      use Linux /proc/net/tcp to"
                               " determine owner of peer socket\n"
                               "       basic      HTTP Basic username and"
                               " password authentication\n"
                               "       default    use 'magic' if possible, "
                               " otherwise fall back to 'basic'\n"
                               "       none       unauthenticated HTTP (if"
                               " the data file is non-confidential)\n");
                        return 0;
                    } else {
                        fprintf(stderr, "%s: unrecognised authentication"
                                " type '%s'\n%*s  options are 'magic',"
                                " 'basic', 'none', 'default'\n",
                                PNAME, optval, (int)strlen(PNAME), "");
                        return 1;
                    }
                    break;
                  case OPT_INCLUDE:
                  case OPT_INCLUDEPATH:
                  case OPT_EXCLUDE:
                  case OPT_EXCLUDEPATH:
                  case OPT_PRUNE:
                  case OPT_PRUNEPATH:
                    if (ninex >= inexsize) {
                        inexsize = ninex * 3 / 2 + 16;
                        inex = sresize(inex, inexsize,
                                       struct inclusion_exclusion);
                    }
                    inex[ninex].path = (optid == OPT_INCLUDEPATH ||
                                        optid == OPT_EXCLUDEPATH ||
                                        optid == OPT_PRUNEPATH);
                    inex[ninex].type = (optid == OPT_INCLUDE ? 1 :
                                        optid == OPT_INCLUDEPATH ? 1 :
                                        optid == OPT_EXCLUDE ? 0 :
                                        optid == OPT_EXCLUDEPATH ? 0 :
                                        optid == OPT_PRUNE ? -1 :
                                        /* optid == OPT_PRUNEPATH ? */ -1);
                    inex[ninex].wildcard = optval;
                    ninex++;
                    break;
                }
            }
        } else {
            fprintf(stderr, "%s: unexpected argument '%s'\n", PNAME, p);
            return 1;
        }
    }

    if (mode == USAGE) {
        usage(stderr);
        return 1;
    } else if (mode == SCAN) {

        fd = open(filename, O_RDWR | O_TRUNC | O_CREAT, S_IRWXU);
        if (fd < 0) {
            fprintf(stderr, "%s: %s: open: %s\n", PNAME, filename,
                    strerror(errno));
            return 1;
        }

        if (stat(scandir, &st) < 0) {
            fprintf(stderr, "%s: %s: stat: %s\n", PNAME, scandir,
                    strerror(errno));
            return 1;
        }
        ctx->filesystem_dev = crossfs ? 0 : st.st_dev;

        if (fstat(fd, &st) < 0) {
            perror("agedu: fstat");
            return 1;
        }
        ctx->datafile_dev = st.st_dev;
        ctx->datafile_ino = st.st_ino;
        ctx->inex = inex;
        ctx->ninex = ninex;
        ctx->crossfs = crossfs;

        ctx->last_output_update = time(NULL);

        /* progress==1 means report progress only if stderr is a tty */
        if (progress == 1)
            progress = isatty(2) ? 2 : 0;
        ctx->progress = progress;
        {
            struct winsize ws;
            if (progress && ioctl(2, TIOCGWINSZ, &ws) == 0)
                ctx->progwidth = ws.ws_col - 1;
            else
                ctx->progwidth = 79;
        }

        /*
         * Scan the directory tree, and write out the trie component
         * of the data file.
         */
        ctx->tb = triebuild_new(fd);
        du(scandir, gotdata, ctx);
        count = triebuild_finish(ctx->tb);
        triebuild_free(ctx->tb);

        if (ctx->progress) {
            fprintf(stderr, "%-*s\r", ctx->progwidth, "");
            fflush(stderr);
        }

        /*
         * Work out how much space the cumulative index trees will
         * take; enlarge the file, and memory-map it.
         */
        if (fstat(fd, &st) < 0) {
            perror("agedu: fstat");
            return 1;
        }

        printf("Built pathname index, %d entries, %ju bytes\n", count,
               (intmax_t)st.st_size);

        totalsize = index_compute_size(st.st_size, count);

        if (lseek(fd, totalsize-1, SEEK_SET) < 0) {
            perror("agedu: lseek");
            return 1;
        }
        if (write(fd, "\0", 1) < 1) {
            perror("agedu: write");
            return 1;
        }

        printf("Upper bound on index file size = %ju bytes\n",
               (intmax_t)totalsize);

        mappedfile = mmap(NULL, totalsize, PROT_READ|PROT_WRITE,MAP_SHARED, fd, 0);
        if (!mappedfile) {
            perror("agedu: mmap");
            return 1;
        }

        ib = indexbuild_new(mappedfile, st.st_size, count);
        tw = triewalk_new(mappedfile);
        while ((tf = triewalk_next(tw, NULL)) != NULL)
            indexbuild_add(ib, tf);
        triewalk_free(tw);
        realsize = indexbuild_realsize(ib);
        indexbuild_free(ib);

        munmap(mappedfile, totalsize);
        ftruncate(fd, realsize);
        close(fd);
        printf("Actual index file size = %ju bytes\n", (intmax_t)realsize);
    } else if (mode == TEXT) {
        size_t pathlen;

        fd = open(filename, O_RDONLY);
        if (fd < 0) {
            fprintf(stderr, "%s: %s: open: %s\n", PNAME, filename,
                    strerror(errno));
            return 1;
        }
        if (fstat(fd, &st) < 0) {
            perror("agedu: fstat");
            return 1;
        }
        totalsize = st.st_size;
        mappedfile = mmap(NULL, totalsize, PROT_READ, MAP_SHARED, fd, 0);
        if (!mappedfile) {
            perror("agedu: mmap");
            return 1;
        }

        /*
         * Trim trailing slash, just in case.
         */
        pathlen = strlen(querydir);
        if (pathlen > 0 && querydir[pathlen-1] == '/')
            querydir[--pathlen] = '\0';

        text_query(mappedfile, querydir, textcutoff, 1);
    } else if (mode == HTML) {
        size_t pathlen;
        struct html_config cfg;
        unsigned long xi;
        char *html;

        fd = open(filename, O_RDONLY);
        if (fd < 0) {
            fprintf(stderr, "%s: %s: open: %s\n", PNAME, filename,
                    strerror(errno));
            return 1;
        }
        if (fstat(fd, &st) < 0) {
            perror("agedu: fstat");
            return 1;
        }
        totalsize = st.st_size;
        mappedfile = mmap(NULL, totalsize, PROT_READ, MAP_SHARED, fd, 0);
        if (!mappedfile) {
            perror("agedu: mmap");
            return 1;
        }

        /*
         * Trim trailing slash, just in case.
         */
        pathlen = strlen(querydir);
        if (pathlen > 0 && querydir[pathlen-1] == '/')
            querydir[--pathlen] = '\0';

        xi = trie_before(mappedfile, querydir);
        cfg.format = NULL;
        cfg.autoage = htmlautoagerange;
        cfg.oldest = htmloldest;
        cfg.newest = htmlnewest;
        html = html_query(mappedfile, xi, &cfg);
        fputs(html, stdout);
    } else if (mode == DUMP) {
        size_t maxpathlen;
        char *buf;

        fd = open(filename, O_RDONLY);
        if (fd < 0) {
            fprintf(stderr, "%s: %s: open: %s\n", PNAME, filename,
                    strerror(errno));
            return 1;
        }
        if (fstat(fd, &st) < 0) {
            perror("agedu: fstat");
            return 1;
        }
        totalsize = st.st_size;
        mappedfile = mmap(NULL, totalsize, PROT_READ, MAP_SHARED, fd, 0);
        if (!mappedfile) {
            perror("agedu: mmap");
            return 1;
        }

        maxpathlen = trie_maxpathlen(mappedfile);
        buf = snewn(maxpathlen, char);

        tw = triewalk_new(mappedfile);
        while ((tf = triewalk_next(tw, buf)) != NULL) {
            printf("%s: %llu %llu\n", buf, tf->blocks, tf->atime);
        }
        triewalk_free(tw);
    } else if (mode == HTTPD) {
        struct html_config cfg;

        fd = open(filename, O_RDONLY);
        if (fd < 0) {
            fprintf(stderr, "%s: %s: open: %s\n", PNAME, filename,
                    strerror(errno));
            return 1;
        }
        if (fstat(fd, &st) < 0) {
            perror("agedu: fstat");
            return 1;
        }
        totalsize = st.st_size;
        mappedfile = mmap(NULL, totalsize, PROT_READ, MAP_SHARED, fd, 0);
        if (!mappedfile) {
            perror("agedu: mmap");
            return 1;
        }

        cfg.format = NULL;
        cfg.autoage = htmlautoagerange;
        cfg.oldest = htmloldest;
        cfg.newest = htmlnewest;
        run_httpd(mappedfile, auth, &cfg);
    }

    return 0;
}