[BACK]Return to main.c CVS log [TXT][DIR] Up to [cvsweb.bsd.lv] / mandoc

Annotation of mandoc/main.c, Revision 1.4

1.4     ! kristaps    1: /* $Id: main.c,v 1.3 2009/03/19 16:40:49 kristaps Exp $ */
1.1       kristaps    2: /*
                      3:  * Copyright (c) 2008, 2009 Kristaps Dzonsons <kristaps@openbsd.org>
                      4:  *
                      5:  * Permission to use, copy, modify, and distribute this software for any
                      6:  * purpose with or without fee is hereby granted, provided that the
                      7:  * above copyright notice and this permission notice appear in all
                      8:  * copies.
                      9:  *
                     10:  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL
                     11:  * WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED
                     12:  * WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE
                     13:  * AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
                     14:  * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
                     15:  * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER
                     16:  * TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
                     17:  * PERFORMANCE OF THIS SOFTWARE.
                     18:  */
                     19: #include <sys/stat.h>
                     20:
                     21: #include <assert.h>
                     22: #include <err.h>
                     23: #include <fcntl.h>
                     24: #include <stdio.h>
                     25: #include <stdlib.h>
                     26: #include <string.h>
                     27: #include <unistd.h>
                     28:
                     29: #include "mdoc.h"
                     30:
1.3       kristaps   31: #ifdef __linux__
                     32: extern int               getsubopt(char **, char * const *, char **);
                     33: # ifndef __dead
                     34: #  define __dead __attribute__((__noreturn__))
                     35: # endif
                     36: #endif
                     37:
1.1       kristaps   38: #define        WARN_WALL         0x03          /* All-warnings mask. */
                     39: #define        WARN_WCOMPAT     (1 << 0)       /* Compatibility warnings. */
                     40: #define        WARN_WSYNTAX     (1 << 1)       /* Syntax warnings. */
                     41: #define        WARN_WERR        (1 << 2)       /* Warnings->errors. */
                     42:
                     43: enum outt {
                     44:        OUTT_ASCII,
                     45:        OUTT_LATIN1,
                     46:        OUTT_UTF8,
                     47:        OUTT_TREE,
                     48:        OUTT_LINT
                     49: };
                     50:
                     51: typedef        int             (*out_run)(void *, const struct mdoc *);
                     52: typedef        void            (*out_free)(void *);
                     53:
                     54: extern char             *__progname;
                     55:
                     56: extern void             *ascii_alloc(void);
                     57: extern void             *latin1_alloc(void);
                     58: extern void             *utf8_alloc(void);
                     59: extern int               terminal_run(void *, const struct mdoc *);
                     60: extern int               tree_run(void *, const struct mdoc *);
                     61: extern void              terminal_free(void *);
                     62:
                     63: __dead static void       version(void);
                     64: __dead static void       usage(void);
                     65: static int               foptions(int *, char *);
                     66: static int               toptions(enum outt *, char *);
                     67: static int               woptions(int *, char *);
                     68: static int               merr(void *, int, int, const char *);
                     69: static int               mwarn(void *, int, int,
                     70:                                enum mdoc_warn, const char *);
                     71: static int               file(char **, size_t *, char **, size_t *,
                     72:                                const char *, struct mdoc *);
                     73: static int               fdesc(char **, size_t *, char **, size_t *,
                     74:                                const char *, int, struct mdoc *);
                     75:
                     76:
                     77: int
                     78: main(int argc, char *argv[])
                     79: {
                     80:        int              c, rc, fflags, wflags;
                     81:        struct mdoc_cb   cb;
                     82:        struct mdoc     *mdoc;
                     83:        char            *buf, *line;
                     84:        size_t           bufsz, linesz;
                     85:        void            *outdata;
                     86:        enum outt        outtype;
                     87:        out_run          outrun;
                     88:        out_free         outfree;
                     89:
                     90:        fflags = wflags = 0;
                     91:        outtype = OUTT_ASCII;
                     92:
                     93:        /* LINTED */
                     94:        while (-1 != (c = getopt(argc, argv, "f:VW:T:")))
                     95:                switch (c) {
                     96:                case ('f'):
                     97:                        if ( ! foptions(&fflags, optarg))
                     98:                                return(0);
                     99:                        break;
                    100:                case ('T'):
                    101:                        if ( ! toptions(&outtype, optarg))
                    102:                                return(0);
                    103:                        break;
                    104:                case ('W'):
                    105:                        if ( ! woptions(&wflags, optarg))
                    106:                                return(0);
                    107:                        break;
                    108:                case ('V'):
                    109:                        version();
                    110:                        /* NOTREACHED */
                    111:                default:
                    112:                        usage();
                    113:                        /* NOTREACHED */
                    114:                }
                    115:
                    116:        argc -= optind;
                    117:        argv += optind;
                    118:
                    119:        /*
                    120:         * Allocate the appropriate front-end.  Note that utf8, ascii
                    121:         * and latin1 all resolve to the terminal front-end with
                    122:         * different encodings (see terminal.c).  Not all frontends have
                    123:         * cleanup or alloc routines.
                    124:         */
                    125:
                    126:        switch (outtype) {
                    127:        case (OUTT_LATIN1):
                    128:                outdata = latin1_alloc();
                    129:                outrun = terminal_run;
                    130:                outfree = terminal_free;
                    131:                break;
                    132:        case (OUTT_UTF8):
                    133:                outdata = utf8_alloc();
                    134:                outrun = terminal_run;
                    135:                outfree = terminal_free;
                    136:                break;
                    137:        case (OUTT_TREE):
                    138:                outdata = NULL;
                    139:                outrun = tree_run;
                    140:                outfree = NULL;
                    141:                break;
                    142:        case (OUTT_LINT):
                    143:                outdata = NULL;
                    144:                outrun = NULL;
                    145:                outfree = NULL;
                    146:                break;
                    147:        default:
                    148:                outdata = ascii_alloc();
                    149:                outrun = terminal_run;
                    150:                outfree = terminal_free;
                    151:                break;
                    152:        }
                    153:
                    154:        /*
                    155:         * All callbacks route into here, where we print them onto the
                    156:         * screen.  XXX - for now, no path for debugging messages.
                    157:         */
                    158:
                    159:        cb.mdoc_msg = NULL;
                    160:        cb.mdoc_err = merr;
                    161:        cb.mdoc_warn = mwarn;
                    162:
                    163:        buf = line = NULL;
                    164:        bufsz = linesz = 0;
                    165:
                    166:        mdoc = mdoc_alloc(&wflags, fflags, &cb);
                    167:
1.4     ! kristaps  168:        /*
        !           169:         * Loop around available files.
        !           170:         */
1.1       kristaps  171:
1.4     ! kristaps  172:        if (NULL == *argv) {
        !           173:                c = fdesc(&line, &linesz, &buf, &bufsz,
        !           174:                                "stdin", STDIN_FILENO, mdoc);
        !           175:                rc = 0;
        !           176:                if (c && NULL == outrun)
        !           177:                        rc = 1;
        !           178:                else if (c && outrun && (*outrun)(outdata, mdoc))
        !           179:                        rc = 1;
        !           180:        } else {
        !           181:                while (*argv) {
        !           182:                        c = file(&line, &linesz, &buf,
        !           183:                                        &bufsz, *argv, mdoc);
        !           184:                        if ( ! c)
        !           185:                                break;
        !           186:                        if (outrun && ! (*outrun)(outdata, mdoc))
        !           187:                                break;
        !           188:                        /* Reset the parser for another file. */
        !           189:                        mdoc_reset(mdoc);
        !           190:                        argv++;
        !           191:                }
        !           192:                rc = NULL == *argv;
1.1       kristaps  193:        }
                    194:
                    195:        if (buf)
                    196:                free(buf);
                    197:        if (line)
                    198:                free(line);
                    199:        if (outfree)
                    200:                (*outfree)(outdata);
                    201:
                    202:        mdoc_free(mdoc);
                    203:
                    204:        return(rc ? EXIT_SUCCESS : EXIT_FAILURE);
                    205: }
                    206:
                    207:
                    208: __dead static void
                    209: version(void)
                    210: {
                    211:
                    212:        (void)printf("%s %s\n", __progname, VERSION);
                    213:        exit(0);
                    214:        /* NOTREACHED */
                    215: }
                    216:
                    217:
                    218: __dead static void
                    219: usage(void)
                    220: {
                    221:
                    222:        (void)fprintf(stderr, "usage: %s\n", __progname);
                    223:        exit(1);
                    224:        /* NOTREACHED */
                    225: }
                    226:
                    227:
                    228: static int
                    229: file(char **ln, size_t *lnsz, char **buf, size_t *bufsz,
                    230:                const char *file, struct mdoc *mdoc)
                    231: {
                    232:        int              fd, c;
                    233:
                    234:        if (-1 == (fd = open(file, O_RDONLY, 0))) {
                    235:                warn("%s", file);
                    236:                return(0);
                    237:        }
                    238:
                    239:        c = fdesc(ln, lnsz, buf, bufsz, file, fd, mdoc);
                    240:
                    241:        if (-1 == close(fd))
                    242:                warn("%s", file);
                    243:
                    244:        return(c);
                    245: }
                    246:
                    247:
                    248: static int
                    249: fdesc(char **lnp, size_t *lnsz, char **bufp, size_t *bufsz,
                    250:                const char *f, int fd, struct mdoc *mdoc)
                    251: {
                    252:        size_t           sz;
                    253:        ssize_t          ssz;
                    254:        struct stat      st;
                    255:        int              j, i, pos, lnn;
                    256:        char            *ln, *buf;
                    257:
                    258:        buf = *bufp;
                    259:        ln = *lnp;
                    260:
                    261:        /*
                    262:         * Two buffers: ln and buf.  buf is the input buffer, optimised
                    263:         * for each file's block size.  ln is a line buffer.  Both
                    264:         * growable, hence passed in by ptr-ptr.
                    265:         */
                    266:
                    267:        if (-1 == fstat(fd, &st)) {
                    268:                warnx("%s", f);
                    269:                sz = BUFSIZ;
                    270:        } else
                    271:                sz = (unsigned)BUFSIZ > st.st_blksize ?
                    272:                        (size_t)BUFSIZ : st.st_blksize;
                    273:
                    274:        if (sz > *bufsz) {
                    275:                if (NULL == (buf = realloc(buf, sz)))
                    276:                        err(1, "realloc");
                    277:                *bufp = buf;
                    278:                *bufsz = sz;
                    279:        }
                    280:
                    281:        /*
                    282:         * Fill buf with file blocksize and parse newlines into ln.
                    283:         */
                    284:
                    285:        for (lnn = 1, pos = 0; ; ) {
                    286:                if (-1 == (ssz = read(fd, buf, sz))) {
                    287:                        warn("%s", f);
                    288:                        return(0);
                    289:                } else if (0 == ssz)
                    290:                        break;
                    291:
                    292:                for (i = 0; i < (int)ssz; i++) {
                    293:                        if (pos >= (int)*lnsz) {
                    294:                                *lnsz += 256; /* Step-size. */
                    295:                                ln = realloc(ln, *lnsz);
                    296:                                if (NULL == ln)
                    297:                                        err(1, "realloc");
                    298:                                *lnp = ln;
                    299:                        }
                    300:
                    301:                        if ('\n' != buf[i]) {
                    302:                                ln[pos++] = buf[i];
                    303:                                continue;
                    304:                        }
                    305:
                    306:                        /* Check for CPP-escaped newline.  */
                    307:
                    308:                        if (pos > 0 && '\\' == ln[pos - 1]) {
                    309:                                for (j = pos - 1; j >= 0; j--)
                    310:                                        if ('\\' != ln[j])
                    311:                                                break;
                    312:
                    313:                                if ( ! ((pos - j) % 2)) {
                    314:                                        pos--;
                    315:                                        lnn++;
                    316:                                        continue;
                    317:                                }
                    318:                        }
                    319:
                    320:                        ln[pos] = 0;
                    321:                        if ( ! mdoc_parseln(mdoc, lnn, ln))
                    322:                                return(0);
                    323:                        lnn++;
                    324:                        pos = 0;
                    325:                }
                    326:        }
                    327:
                    328:        return(mdoc_endparse(mdoc));
                    329: }
                    330:
                    331:
                    332: static int
                    333: toptions(enum outt *tflags, char *arg)
                    334: {
                    335:
                    336:        if (0 == strcmp(arg, "ascii"))
                    337:                *tflags = OUTT_ASCII;
                    338:        else if (0 == strcmp(arg, "latin1"))
                    339:                *tflags = OUTT_LATIN1;
                    340:        else if (0 == strcmp(arg, "utf8"))
                    341:                *tflags = OUTT_UTF8;
                    342:        else if (0 == strcmp(arg, "lint"))
                    343:                *tflags = OUTT_LINT;
                    344:        else if (0 == strcmp(arg, "tree"))
                    345:                *tflags = OUTT_TREE;
                    346:        else {
                    347:                warnx("bad argument: -T%s", arg);
                    348:                return(0);
                    349:        }
                    350:
                    351:        return(1);
                    352: }
                    353:
                    354:
                    355: /*
                    356:  * Parse out the options for [-fopt...] setting compiler options.  These
                    357:  * can be comma-delimited or called again.
                    358:  */
                    359: static int
                    360: foptions(int *fflags, char *arg)
                    361: {
                    362:        char            *v;
                    363:        char            *toks[4];
                    364:
                    365:        toks[0] = "ign-scope";
                    366:        toks[1] = "ign-escape";
                    367:        toks[2] = "ign-macro";
                    368:        toks[3] = NULL;
                    369:
                    370:        while (*arg)
                    371:                switch (getsubopt(&arg, toks, &v)) {
                    372:                case (0):
                    373:                        *fflags |= MDOC_IGN_SCOPE;
                    374:                        break;
                    375:                case (1):
                    376:                        *fflags |= MDOC_IGN_ESCAPE;
                    377:                        break;
                    378:                case (2):
                    379:                        *fflags |= MDOC_IGN_MACRO;
                    380:                        break;
                    381:                default:
                    382:                        warnx("bad argument: -f%s", arg);
                    383:                        return(0);
                    384:                }
                    385:
                    386:        return(1);
                    387: }
                    388:
                    389:
                    390: /*
                    391:  * Parse out the options for [-Werr...], which sets warning modes.
                    392:  * These can be comma-delimited or called again.
                    393:  */
                    394: static int
                    395: woptions(int *wflags, char *arg)
                    396: {
                    397:        char            *v;
                    398:        char            *toks[5];
                    399:
                    400:        toks[0] = "all";
                    401:        toks[1] = "compat";
                    402:        toks[2] = "syntax";
                    403:        toks[3] = "error";
                    404:        toks[4] = NULL;
                    405:
                    406:        while (*arg)
                    407:                switch (getsubopt(&arg, toks, &v)) {
                    408:                case (0):
                    409:                        *wflags |= WARN_WALL;
                    410:                        break;
                    411:                case (1):
                    412:                        *wflags |= WARN_WCOMPAT;
                    413:                        break;
                    414:                case (2):
                    415:                        *wflags |= WARN_WSYNTAX;
                    416:                        break;
                    417:                case (3):
                    418:                        *wflags |= WARN_WERR;
                    419:                        break;
                    420:                default:
                    421:                        warnx("bad argument: -W%s", arg);
                    422:                        return(0);
                    423:                }
                    424:
                    425:        return(1);
                    426: }
                    427:
                    428:
1.2       kristaps  429: /* ARGSUSED */
1.1       kristaps  430: static int
                    431: merr(void *arg, int line, int col, const char *msg)
                    432: {
                    433:
                    434:        warnx("error: %s (line %d, column %d)", msg, line, col);
                    435:        return(0);
                    436: }
                    437:
                    438:
                    439: static int
                    440: mwarn(void *arg, int line, int col,
                    441:                enum mdoc_warn type, const char *msg)
                    442: {
                    443:        int              flags;
                    444:        char            *wtype;
                    445:
                    446:        flags = *(int *)arg;
                    447:        wtype = NULL;
                    448:
                    449:        switch (type) {
                    450:        case (WARN_COMPAT):
                    451:                wtype = "compat";
                    452:                if (flags & WARN_WCOMPAT)
                    453:                        break;
                    454:                return(1);
                    455:        case (WARN_SYNTAX):
                    456:                wtype = "syntax";
                    457:                if (flags & WARN_WSYNTAX)
                    458:                        break;
                    459:                return(1);
                    460:        }
                    461:
                    462:        assert(wtype);
                    463:        warnx("%s warning: %s (line %d, column %d)",
                    464:                        wtype, msg, line, col);
                    465:
                    466:        if ( ! (flags & WARN_WERR))
                    467:                return(1);
                    468:
                    469:        warnx("%s: considering warnings as errors",
                    470:                        __progname);
                    471:        return(0);
                    472: }
                    473:
                    474:

CVSweb