From b96fa66b3bba1ea8e09fda0e994688409dbe30ce Mon Sep 17 00:00:00 2001 From: Kristaps Dzonsons Date: Sun, 22 Feb 2009 22:58:39 +0000 Subject: BROKEN BUILD: migrating to mmain stuff. --- mmain.c | 328 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 328 insertions(+) create mode 100644 mmain.c (limited to 'mmain.c') diff --git a/mmain.c b/mmain.c new file mode 100644 index 00000000..5e41683a --- /dev/null +++ b/mmain.c @@ -0,0 +1,328 @@ + /* $Id: mmain.c,v 1.1 2009/02/22 22:58:39 kristaps Exp $ */ +/* + * Copyright (c) 2008 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) /* Max input line 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. */ + u_long bufsz; /* Input buffer size. */ + char in[MAXPATHLEN]; /* Input file name. */ + int fdin; /* Input file desc. */ +}; + +extern char *__progname; + +static int getsopts(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 *); + +#ifdef __linux__ +extern int getsubopt(char **, char *const *, char **); +#endif + + +void +mmain_usage(const char *help) +{ + + warnx("usage: %s %s%s[-v] [-Wwarn...] [infile]", __progname, + help ? help : "", help ? " " : ""); +} + + +struct mmain * +mmain_alloc(void) +{ + struct mmain *p; + + if (NULL == (p = calloc(1, sizeof(struct mmain)))) + err(1, "malloc"); + + (void)strlcpy(p->in, "-", MAXPATHLEN); + p->fdin = STDIN_FILENO; + + return(p); +} + + +int +mmain_isopt(int c) +{ + + switch (c) { + case ('v'): + /* FALLTHROUGH */ + case ('W'): + return(1); + default: + break; + } + return(0); +} + + +int +mmain_getopt(struct mmain *p, int argc, + char *argv[], const char *help) +{ + int c; + + while (-1 != (c = getopt(argc, argv, ":vW:"))) + switch (c) { + case ('v'): + p->dbg++; + break; + case ('W'): + if ( ! getsopts(p, optarg)) + return(0); + break; + case ('?'): + break; + default: + mmain_usage(help); + return(0); + } + + argv += optind; + argc -= optind; + + if (0 == argc) + return(1); + + if (strlcpy(p->in, *argv++, MAXPATHLEN) < MAXPATHLEN) + return(1); + + warnx("filename too long"); + return(0); +} + + +__dead void +mmain_exit(struct mmain *p, int code) +{ + + if (p->mdoc) + mdoc_free(p->mdoc); + free(p); + exit(code); +} + + +struct mdoc * +mmain_mdoc(struct mmain *p) +{ + struct stat st; + int c; + struct mdoc_cb cb; + + cb.mdoc_err = msg_err; + cb.mdoc_warn = msg_warn; + cb.mdoc_msg = msg_msg; + + if (0 != strcmp(p->in, "-")) + if (-1 == (p->fdin = open(p->in, O_RDONLY, 0))) { + warn("%s", p->in); + return(0); + } + + /* Allocate a buffer to be BUFSIZ/block size. */ + + if (-1 == fstat(p->fdin, &st)) { + warn("%s", p->in); + p->bufsz = BUFSIZ; + } else + p->bufsz = MAX(st.st_blksize, BUFSIZ); + + p->buf = malloc(p->bufsz); + if (NULL == p->buf) + err(1, "malloc"); + + /* Allocate the parser. */ + + p->mdoc = mdoc_alloc(p, &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); +} + + +static int +getsopts(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: + return(0); + } + + return(1); +} + + +static int +parse(struct mmain *p) +{ + ssize_t sz, i; + size_t pos; + char line[MD_LINE_SZ]; + int lnn; + + /* + * This is a little more complicated than fgets. TODO: have + * some benchmarks that show it's faster (note that I want to + * check many, many manuals simultaneously, so speed is + * important). Fill a buffer (sized to the block size) with a + * single read, then parse \n-terminated lines into a line + * buffer, which is passed to the parser. Hard-code the line + * buffer to a particular size -- a reasonable assumption. + */ + + for (lnn = 1, 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 < sz; i++) { + if ('\n' != p->buf[i]) { + if (pos < sizeof(line)) { + line[(int)pos++] = p->buf[(int)i]; + continue; + } + warnx("%s: line %d too long", p->in, lnn); + return(0); + } + + line[(int)pos] = 0; + if ( ! mdoc_parseln(p->mdoc, lnn, line)) + return(0); + + lnn++; + pos = 0; + } + } + + return(mdoc_endparse(p->mdoc)); +} + + +static int +msg_err(void *arg, int line, int col, const char *msg) +{ + struct mmain *p; + + p = (struct mmain *)arg; + + warnx("%s:%d: error: %s (column %d)", + p->in, line, msg, col); + return(0); +} + + +static void +msg_msg(void *arg, int line, int col, const char *msg) +{ + struct mmain *p; + + p = (struct mmain *)arg; + + if (0 == p->dbg) + return; + + warnx("%s:%d: debug: %s (column %d)", + p->in, line, msg, col); +} + + +static int +msg_warn(void *arg, int line, int col, + enum mdoc_warn type, const char *msg) +{ + struct mmain *p; + + p = (struct mmain *)arg; + + switch (type) { + case (WARN_COMPAT): + if (p->warn & MD_WARN_COMPAT) + break; + return(1); + case (WARN_SYNTAX): + if (p->warn & MD_WARN_SYNTAX) + break; + return(1); + } + + warnx("%s:%d: warning: %s (column %d)", + p->in, line, msg, col); + + if ( ! (p->warn & MD_WARN_ERR)) + return(1); + + warnx("%s: considering warnings as errors", __progname); + return(0); +} -- cgit v1.2.3-56-ge451