/* $Id: mmain.c,v 1.15 2009/03/19 11:49:00 kristaps Exp $ */ /* * Copyright (c) 2008, 2009 Kristaps Dzonsons * * Permission to use, copy, modify, and distribute this software for any * purpose with or without fee is hereby granted, provided that the * above copyright notice and this permission notice appear in all * copies. * * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL * WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED * WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE * AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER * TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR * PERFORMANCE OF THIS SOFTWARE. */ #include #include #include #include #include #include #include #include #include #include #include "mmain.h" #define MD_LINE_SZ (256) /* Input line step-size. */ struct mmain { int warn; /* Warning flags. */ #define MD_WARN_SYNTAX (1 << 0) /* Show syntax warnings. */ #define MD_WARN_COMPAT (1 << 1) /* Show compat warnings. */ #define MD_WARN_ALL (0x03) /* Show all warnings. */ #define MD_WARN_ERR (1 << 2) /* Make warnings->errors. */ int dbg; /* Debug level. */ struct mdoc *mdoc; /* Active parser. */ char *buf; /* Input buffer. */ size_t bufsz; /* Input buffer size. */ const char *in; /* Input file name. */ int fdin; /* Input file desc. */ int pflags; /* Parse flags. */ }; extern char *__progname; static void usage(const char *, const char *); static int optswarn(struct mmain *, char *); static int optsopt(struct mmain *, char *); static int parse(struct mmain *); static void msg_msg(void *, int, int, const char *); static int msg_err(void *, int, int, const char *); static int msg_warn(void *, int, int, enum mdoc_warn, const char *); /* * Print our and our caller's usage message. */ void usage(const char *help, const char *args) { warnx("usage: %s %s%s[-v] [-foption...] [-Wwarn...]%s%s", __progname, help ? help : "", help ? " " : "", args ? " " : "", args ? args : ""); } /* * Allocate the convenience library and initialise some values. */ struct mmain * mmain_alloc(void) { struct mmain *p; if (NULL == (p = calloc(1, sizeof(struct mmain)))) err(1, "malloc"); return(p); } /* * Parse command-line options. Accepts a small (<28 char) opstring "u" * parameter (e.g. "ho:") or NULL, a corresponding "help" string (e.g. * "[-h] [-o output]" or NULL, a callback function for parsed arguments * and an opaque pointer argument for that function. */ int mmain_getopt(struct mmain *p, int argc, char *argv[], const char *help, const char *args, const char *u, void *arg, int (*getopt_cb)(void *, int, char *)) { int c; char opts[32]; /* XXX */ size_t sz; extern int optind; sz = strlcpy(opts, "VvW:f:", 32); assert(sz < 32); if (u) { sz = strlcat(opts, u, 32); assert(sz < 32); } optind = 1; /* LINTED */ while (-1 != (c = getopt(argc, argv, opts))) switch (c) { case ('f'): if ( ! optsopt(p, optarg)) mmain_exit(p, 1); break; case ('v'): p->dbg++; break; case ('V'): (void)printf("%s %s\n", __progname, VERSION); mmain_exit(p, 0); /* NOTREACHED */ case ('W'): if ( ! optswarn(p, optarg)) mmain_exit(p, 1); break; case ('?'): usage(help, args); mmain_exit(p, 1); /* NOTREACHED */ default: assert(getopt_cb); if ((*getopt_cb)(arg, c, optarg)) break; mmain_exit(p, 1); /* NOTREACHED */ } return(optind); } void mmain_reset(struct mmain *p) { if (p->mdoc) mdoc_free(p->mdoc); p->mdoc = NULL; } void mmain_free(struct mmain *p) { if (p->mdoc) mdoc_free(p->mdoc); free(p); } __dead void mmain_exit(struct mmain *p, int code) { mmain_free(p); exit(code); } void mmain_prepare(struct mmain *p, const char *in) { struct stat st; if ((p->in = in)) { if (-1 == (p->fdin = open(p->in, O_RDONLY, 0))) { warn("%s", p->in); mmain_exit(p, 1); } } else { p->fdin = STDIN_FILENO; p->in = "-"; } if (-1 == fstat(p->fdin, &st)) { warn("%s", p->in); p->bufsz = BUFSIZ; } else p->bufsz = (unsigned)BUFSIZ > st.st_blksize ? (size_t)BUFSIZ : st.st_blksize; if (NULL == (p->buf = realloc(p->buf, p->bufsz))) err(1, "realloc"); } struct mdoc * mmain_process(struct mmain *p) { int c; struct mdoc_cb cb; /* XXX - in mmain_alloc.*/ cb.mdoc_err = msg_err; cb.mdoc_warn = msg_warn; cb.mdoc_msg = msg_msg; /* Allocate the parser. */ p->mdoc = mdoc_alloc(p, p->pflags, &cb); /* Parse the input file. */ c = parse(p); free(p->buf); if (STDIN_FILENO != p->fdin) if (-1 == close(p->fdin)) warn("%s", p->in); return(c ? p->mdoc : NULL); } struct mdoc * mmain_mdoc(struct mmain *p, const char *in) { mmain_prepare(p, in); return(mmain_process(p)); } static int optsopt(struct mmain *p, char *arg) { char *v; char *toks[] = { "ign-scope", "ign-escape", "ign-macro", NULL }; while (*arg) switch (getsubopt(&arg, toks, &v)) { case (0): p->pflags |= MDOC_IGN_SCOPE; break; case (1): p->pflags |= MDOC_IGN_ESCAPE; break; case (2): p->pflags |= MDOC_IGN_MACRO; break; default: warnx("unknown -f argument"); return(0); } return(1); } static int optswarn(struct mmain *p, char *arg) { char *v; char *toks[] = { "all", "compat", "syntax", "error", NULL }; while (*arg) switch (getsubopt(&arg, toks, &v)) { case (0): p->warn |= MD_WARN_ALL; break; case (1): p->warn |= MD_WARN_COMPAT; break; case (2): p->warn |= MD_WARN_SYNTAX; break; case (3): p->warn |= MD_WARN_ERR; break; default: warnx("unknown -W argument"); return(0); } return(1); } static int parse(struct mmain *p) { ssize_t sz; int j, i, pos, len, lnn; char *ln; for (ln = NULL, lnn = 1, len = pos = 0; ; ) { if (-1 == (sz = read(p->fdin, p->buf, p->bufsz))) { warn("%s", p->in); return(0); } else if (0 == sz) break; for (i = 0; i < (int)sz; i++) { if (pos >= len) { len += MD_LINE_SZ; ln = realloc(ln, (size_t)len); if (NULL == ln) err(1, "realloc"); } if ('\n' != p->buf[i]) { ln[pos++] = p->buf[i]; continue; } /* Check for escaped newline. */ if (pos > 0 && '\\' == ln[pos - 1]) { for (j = pos - 1; j >= 0; j--) if ('\\' != ln[j]) break; if ( ! ((pos - j) % 2)) { pos--; lnn++; continue; } } ln[pos] = 0; if ( ! mdoc_parseln(p->mdoc, lnn, ln)) { free(ln); return(0); } lnn++; pos = 0; } } if (ln) free(ln); if (pos > 0) warnx("%s: file not eof-terminated", p->in); return(mdoc_endparse(p->mdoc)); }