]> git.cameronkatri.com Git - mandoc.git/blobdiff - main.c
In the architectures dropdown, move aviion, ia64, solbourne, and vax
[mandoc.git] / main.c
diff --git a/main.c b/main.c
index 574fc64a91a1c9b7612b091976e53206c2938d11..eb056230c4c64fff51a1d13fea84de04b6d56d3d 100644 (file)
--- a/main.c
+++ b/main.c
@@ -1,7 +1,7 @@
-/*     $Id: main.c,v 1.246 2015/10/10 13:21:18 schwarze Exp $ */
+/*     $Id: main.c,v 1.265 2016/04/14 20:55:48 schwarze Exp $ */
 /*
  * Copyright (c) 2008-2012 Kristaps Dzonsons <kristaps@bsd.lv>
- * Copyright (c) 2010-2012, 2014, 2015 Ingo Schwarze <schwarze@openbsd.org>
+ * Copyright (c) 2010-2012, 2014-2016 Ingo Schwarze <schwarze@openbsd.org>
  * Copyright (c) 2010 Joerg Sonnenberger <joerg@netbsd.org>
  *
  * Permission to use, copy, modify, and distribute this software for any
@@ -24,6 +24,9 @@
 
 #include <assert.h>
 #include <ctype.h>
+#if HAVE_ERR
+#include <err.h>
+#endif
 #include <errno.h>
 #include <fcntl.h>
 #include <glob.h>
@@ -32,6 +35,7 @@
 #include <stdint.h>
 #include <stdlib.h>
 #include <string.h>
+#include <time.h>
 #include <unistd.h>
 
 #include "mandoc_aux.h"
@@ -59,10 +63,6 @@ enum outmode {
        OUTMODE_ONE
 };
 
-typedef        void            (*out_mdoc)(void *, const struct roff_man *);
-typedef        void            (*out_man)(void *, const struct roff_man *);
-typedef        void            (*out_free)(void *);
-
 enum   outt {
        OUTT_ASCII = 0, /* -Tascii */
        OUTT_LOCALE,    /* -Tlocale */
@@ -77,13 +77,9 @@ enum outt {
 
 struct curparse {
        struct mparse    *mp;
-       struct mchars    *mchars;       /* character table */
        enum mandoclevel  wlevel;       /* ignore messages below this */
        int               wstop;        /* stop after a file with a warning */
        enum outt         outtype;      /* which output to use */
-       out_mdoc          outmdoc;      /* mdoc output ptr */
-       out_man           outman;       /* man output ptr */
-       out_free          outfree;      /* free output ptr */
        void             *outdata;      /* data for output */
        struct manoutput *outopts;      /* output options */
 };
@@ -112,7 +108,6 @@ static      int               woptions(struct curparse *, char *);
 static const int sec_prios[] = {1, 4, 5, 8, 6, 3, 7, 2, 9};
 static char              help_arg[] = "help";
 static char             *help_argv[] = {help_arg, NULL};
-static const char       *progname;
 static enum mandoclevel  rc;
 
 
@@ -123,34 +118,47 @@ main(int argc, char *argv[])
        struct curparse  curp;
        struct mansearch search;
        struct tag_files *tag_files;
+       const char      *progname;
        char            *auxpaths;
        char            *defos;
        unsigned char   *uc;
        struct manpage  *res, *resp;
        char            *conf_file, *defpaths;
-       size_t           isec, i, sz;
+       const char      *sec;
+       size_t           i, sz;
        int              prio, best_prio;
-       char             sec;
-       enum mandoclevel rctmp;
        enum outmode     outmode;
        int              fd;
        int              show_usage;
        int              options;
        int              use_pager;
+       int              status, signum;
        int              c;
+       pid_t            pager_pid, tc_pgid, man_pgid, pid;
 
+#if HAVE_PROGNAME
+       progname = getprogname();
+#else
        if (argc < 1)
-               progname = "mandoc";
+               progname = mandoc_strdup("mandoc");
        else if ((progname = strrchr(argv[0], '/')) == NULL)
                progname = argv[0];
        else
                ++progname;
+       setprogname(progname);
+#endif
 
 #if HAVE_SQLITE3
-       if (strcmp(progname, BINM_MAKEWHATIS) == 0)
+       if (strncmp(progname, "mandocdb", 8) == 0 ||
+           strcmp(progname, BINM_MAKEWHATIS) == 0)
                return mandocdb(argc, argv);
 #endif
 
+#if HAVE_PLEDGE
+       if (pledge("stdio rpath tmppath tty proc exec flock", NULL) == -1)
+               err((int)MANDOCLEVEL_SYSERR, "pledge");
+#endif
+
        /* Search options. */
 
        memset(&conf, 0, sizeof(conf));
@@ -207,15 +215,11 @@ main(int argc, char *argv[])
                        break;
                case 'I':
                        if (strncmp(optarg, "os=", 3)) {
-                               fprintf(stderr,
-                                   "%s: -I %s: Bad argument\n",
-                                   progname, optarg);
+                               warnx("-I %s: Bad argument", optarg);
                                return (int)MANDOCLEVEL_BADARG;
                        }
                        if (defos) {
-                               fprintf(stderr,
-                                   "%s: -I %s: Duplicate argument\n",
-                                   progname, optarg);
+                               warnx("-I %s: Duplicate argument", optarg);
                                return (int)MANDOCLEVEL_BADARG;
                        }
                        defos = mandoc_strdup(optarg + 3);
@@ -294,6 +298,12 @@ main(int argc, char *argv[])
            !isatty(STDOUT_FILENO))
                use_pager = 0;
 
+#if HAVE_PLEDGE
+       if (!use_pager)
+               if (pledge("stdio rpath flock", NULL) == -1)
+                       err((int)MANDOCLEVEL_SYSERR, "pledge");
+#endif
+
        /* Parse arguments. */
 
        if (argc > 0) {
@@ -364,9 +374,7 @@ main(int argc, char *argv[])
                                fs_search(&search, &conf.manpath,
                                    argc, argv, &res, &sz);
                        else
-                               fprintf(stderr,
-                                   "%s: nothing appropriate\n",
-                                   progname);
+                               warnx("nothing appropriate");
                }
 
                if (sz == 0) {
@@ -382,7 +390,7 @@ main(int argc, char *argv[])
 
                if (outmode == OUTMODE_ONE) {
                        argc = 1;
-                       best_prio = 10;
+                       best_prio = 20;
                } else if (outmode == OUTMODE_ALL)
                        argc = (int)sz;
 
@@ -398,11 +406,13 @@ main(int argc, char *argv[])
                                    res[i].output);
                        else if (outmode == OUTMODE_ONE) {
                                /* Search for the best section. */
-                               isec = strcspn(res[i].file, "123456789");
-                               sec = res[i].file[isec];
-                               if ('\0' == sec)
+                               sec = res[i].file;
+                               sec += strcspn(sec, "123456789");
+                               if (sec[0] == '\0')
                                        continue;
-                               prio = sec_prios[sec - '1'];
+                               prio = sec_prios[sec[0] - '1'];
+                               if (sec[1] != '/')
+                                       prio += 10;
                                if (prio >= best_prio)
                                        continue;
                                best_prio = prio;
@@ -422,12 +432,21 @@ main(int argc, char *argv[])
 
        /* mandoc(1) */
 
+#if HAVE_PLEDGE
+       if (use_pager) {
+               if (pledge("stdio rpath tmppath tty proc exec", NULL) == -1)
+                       err((int)MANDOCLEVEL_SYSERR, "pledge");
+       } else {
+               if (pledge("stdio rpath", NULL) == -1)
+                       err((int)MANDOCLEVEL_SYSERR, "pledge");
+       }
+#endif
+
        if (search.argmode == ARG_FILE && ! moptions(&options, auxpaths))
                return (int)MANDOCLEVEL_BADARG;
 
-       curp.mchars = mchars_alloc();
-       curp.mp = mparse_alloc(options, curp.wlevel, mmsg,
-           curp.mchars, defos);
+       mchars_alloc();
+       curp.mp = mparse_alloc(options, curp.wlevel, mmsg, defos);
 
        /*
         * Conditionally start up the lookaside buffer before parsing.
@@ -442,11 +461,7 @@ main(int argc, char *argv[])
        }
 
        while (argc > 0) {
-               rctmp = mparse_open(curp.mp, &fd,
-                   resp != NULL ? resp->file : *argv);
-               if (rc < rctmp)
-                       rc = rctmp;
-
+               fd = mparse_open(curp.mp, resp != NULL ? resp->file : *argv);
                if (fd != -1) {
                        if (use_pager) {
                                tag_files = tag_init();
@@ -465,7 +480,8 @@ main(int argc, char *argv[])
 
                        if (argc > 1 && curp.outtype <= OUTT_UTF8)
                                ascii_sepline(curp.outdata);
-               }
+               } else if (rc < MANDOCLEVEL_ERROR)
+                       rc = MANDOCLEVEL_ERROR;
 
                if (MANDOCLEVEL_OK != rc && curp.wstop)
                        break;
@@ -478,10 +494,26 @@ main(int argc, char *argv[])
                        mparse_reset(curp.mp);
        }
 
-       if (curp.outfree)
-               (*curp.outfree)(curp.outdata);
+       if (curp.outdata != NULL) {
+               switch (curp.outtype) {
+               case OUTT_HTML:
+                       html_free(curp.outdata);
+                       break;
+               case OUTT_UTF8:
+               case OUTT_LOCALE:
+               case OUTT_ASCII:
+                       ascii_free(curp.outdata);
+                       break;
+               case OUTT_PDF:
+               case OUTT_PS:
+                       pspdf_free(curp.outdata);
+                       break;
+               default:
+                       break;
+               }
+       }
        mparse_free(curp.mp);
-       mchars_free(curp.mchars);
+       mchars_free();
 
 out:
        if (search.argmode != ARG_FILE) {
@@ -502,7 +534,52 @@ out:
        if (tag_files != NULL) {
                fclose(stdout);
                tag_write();
-               waitpid(spawn_pager(tag_files), NULL, 0);
+               man_pgid = getpgid(0);
+               tag_files->tcpgid = man_pgid == getpid() ?
+                   getpgid(getppid()) : man_pgid;
+               pager_pid = 0;
+               signum = SIGSTOP;
+               for (;;) {
+
+                       /* Stop here until moved to the foreground. */
+
+                       tc_pgid = tcgetpgrp(STDIN_FILENO);
+                       if (tc_pgid != man_pgid) {
+                               if (tc_pgid == pager_pid) {
+                                       (void)tcsetpgrp(STDIN_FILENO,
+                                           man_pgid);
+                                       if (signum == SIGTTIN)
+                                               continue;
+                               } else
+                                       tag_files->tcpgid = tc_pgid;
+                               kill(0, signum);
+                               continue;
+                       }
+
+                       /* Once in the foreground, activate the pager. */
+
+                       if (pager_pid) {
+                               (void)tcsetpgrp(STDIN_FILENO, pager_pid);
+                               kill(pager_pid, SIGCONT);
+                       } else
+                               pager_pid = spawn_pager(tag_files);
+
+                       /* Wait for the pager to stop or exit. */
+
+                       while ((pid = waitpid(pager_pid, &status,
+                           WUNTRACED)) == -1 && errno == EINTR)
+                               continue;
+
+                       if (pid == -1) {
+                               warn("wait");
+                               rc = MANDOCLEVEL_SYSERR;
+                               break;
+                       }
+                       if (!WIFSTOPPED(status))
+                               break;
+
+                       signum = WSTOPSIG(status);
+               }
                tag_unlink();
        }
 
@@ -577,8 +654,7 @@ fs_lookup(const struct manpaths *paths, size_t ipath,
            paths->paths[ipath], sec, name);
        globres = glob(file, 0, NULL, &globinfo);
        if (globres != 0 && globres != GLOB_NOMATCH)
-               fprintf(stderr, "%s: %s: glob: %s\n",
-                   progname, file, strerror(errno));
+               warn("%s: glob", file);
        free(file);
        if (globres == 0)
                file = mandoc_strdup(*globinfo.gl_pathv);
@@ -588,8 +664,8 @@ fs_lookup(const struct manpaths *paths, size_t ipath,
 
 found:
 #if HAVE_SQLITE3
-       fprintf(stderr, "%s: outdated mandoc.db lacks %s(%s) entry, run "
-           "makewhatis %s\n", progname, name, sec, paths->paths[ipath]);
+       warnx("outdated mandoc.db lacks %s(%s) entry, run makewhatis %s",
+           name, sec, paths->paths[ipath]);
 #endif
        *res = mandoc_reallocarray(*res, ++*ressz, sizeof(struct manpage));
        page = *res + (*ressz - 1);
@@ -608,7 +684,7 @@ fs_search(const struct mansearch *cfg, const struct manpaths *paths,
        int argc, char **argv, struct manpage **res, size_t *ressz)
 {
        const char *const sections[] =
-           {"1", "8", "6", "2", "3", "3p", "5", "7", "4", "9"};
+           {"1", "8", "6", "2", "3", "5", "7", "4", "9", "3p"};
        const size_t nsec = sizeof(sections)/sizeof(sections[0]);
 
        size_t           ipath, isec, lastsz;
@@ -631,9 +707,7 @@ fs_search(const struct mansearch *cfg, const struct manpaths *paths,
                                        return;
                }
                if (*ressz == lastsz)
-                       fprintf(stderr,
-                           "%s: No entry for %s in the manual.\n",
-                           progname, *argv);
+                       warnx("No entry for %s in the manual.", *argv);
                lastsz = *ressz;
                argv++;
                argc--;
@@ -649,9 +723,11 @@ parse(struct curparse *curp, int fd, const char *file)
        /* Begin by parsing the file itself. */
 
        assert(file);
-       assert(fd >= -1);
+       assert(fd >= 0);
 
        rctmp = mparse_readfd(curp->mp, fd, file);
+       if (fd != STDIN_FILENO)
+               close(fd);
        if (rc < rctmp)
                rc = rctmp;
 
@@ -665,82 +741,83 @@ parse(struct curparse *curp, int fd, const char *file)
 
        /* If unset, allocate output dev now (if applicable). */
 
-       if ( ! (curp->outman && curp->outmdoc)) {
+       if (curp->outdata == NULL) {
                switch (curp->outtype) {
                case OUTT_HTML:
-                       curp->outdata = html_alloc(curp->mchars,
-                           curp->outopts);
-                       curp->outfree = html_free;
+                       curp->outdata = html_alloc(curp->outopts);
                        break;
                case OUTT_UTF8:
-                       curp->outdata = utf8_alloc(curp->mchars,
-                           curp->outopts);
-                       curp->outfree = ascii_free;
+                       curp->outdata = utf8_alloc(curp->outopts);
                        break;
                case OUTT_LOCALE:
-                       curp->outdata = locale_alloc(curp->mchars,
-                           curp->outopts);
-                       curp->outfree = ascii_free;
+                       curp->outdata = locale_alloc(curp->outopts);
                        break;
                case OUTT_ASCII:
-                       curp->outdata = ascii_alloc(curp->mchars,
-                           curp->outopts);
-                       curp->outfree = ascii_free;
+                       curp->outdata = ascii_alloc(curp->outopts);
                        break;
                case OUTT_PDF:
-                       curp->outdata = pdf_alloc(curp->mchars,
-                           curp->outopts);
-                       curp->outfree = pspdf_free;
+                       curp->outdata = pdf_alloc(curp->outopts);
                        break;
                case OUTT_PS:
-                       curp->outdata = ps_alloc(curp->mchars,
-                           curp->outopts);
-                       curp->outfree = pspdf_free;
+                       curp->outdata = ps_alloc(curp->outopts);
                        break;
                default:
                        break;
                }
+       }
+
+       mparse_result(curp->mp, &man, NULL);
+
+       /* Execute the out device, if it exists. */
 
+       if (man == NULL)
+               return;
+       if (man->macroset == MACROSET_MDOC) {
+               mdoc_validate(man);
                switch (curp->outtype) {
                case OUTT_HTML:
-                       curp->outman = html_man;
-                       curp->outmdoc = html_mdoc;
+                       html_mdoc(curp->outdata, man);
                        break;
                case OUTT_TREE:
-                       curp->outman = tree_man;
-                       curp->outmdoc = tree_mdoc;
+                       tree_mdoc(curp->outdata, man);
                        break;
                case OUTT_MAN:
-                       curp->outmdoc = man_mdoc;
-                       curp->outman = man_man;
+                       man_mdoc(curp->outdata, man);
                        break;
                case OUTT_PDF:
-                       /* FALLTHROUGH */
                case OUTT_ASCII:
-                       /* FALLTHROUGH */
                case OUTT_UTF8:
-                       /* FALLTHROUGH */
                case OUTT_LOCALE:
-                       /* FALLTHROUGH */
                case OUTT_PS:
-                       curp->outman = terminal_man;
-                       curp->outmdoc = terminal_mdoc;
+                       terminal_mdoc(curp->outdata, man);
+                       break;
+               default:
+                       break;
+               }
+       }
+       if (man->macroset == MACROSET_MAN) {
+               man_validate(man);
+               switch (curp->outtype) {
+               case OUTT_HTML:
+                       html_man(curp->outdata, man);
+                       break;
+               case OUTT_TREE:
+                       tree_man(curp->outdata, man);
+                       break;
+               case OUTT_MAN:
+                       man_man(curp->outdata, man);
+                       break;
+               case OUTT_PDF:
+               case OUTT_ASCII:
+               case OUTT_UTF8:
+               case OUTT_LOCALE:
+               case OUTT_PS:
+                       terminal_man(curp->outdata, man);
                        break;
                default:
                        break;
                }
        }
-
-       mparse_result(curp->mp, &man, NULL);
-
-       /* Execute the out device, if it exists. */
-
-       if (man == NULL)
-               return;
-       if (curp->outmdoc != NULL && man->macroset == MACROSET_MDOC)
-               (*curp->outmdoc)(curp->outdata, man);
-       if (curp->outman != NULL && man->macroset == MACROSET_MAN)
-               (*curp->outman)(curp->outdata, man);
 }
 
 static void
@@ -751,12 +828,12 @@ passthrough(const char *file, int fd, int synopsis_only)
 
        FILE            *stream;
        const char      *syscall;
-       char            *line;
-       size_t           len, off;
-       ssize_t          nw;
+       char            *line, *cp;
+       size_t           linesz;
        int              print;
 
-       fflush(stdout);
+       line = NULL;
+       linesz = 0;
 
        if ((stream = fdopen(fd, "r")) == NULL) {
                close(fd);
@@ -765,47 +842,42 @@ passthrough(const char *file, int fd, int synopsis_only)
        }
 
        print = 0;
-       while ((line = fgetln(stream, &len)) != NULL) {
+       while (getline(&line, &linesz, stream) != -1) {
+               cp = line;
                if (synopsis_only) {
                        if (print) {
-                               if ( ! isspace((unsigned char)*line))
+                               if ( ! isspace((unsigned char)*cp))
                                        goto done;
-                               while (len &&
-                                   isspace((unsigned char)*line)) {
-                                       line++;
-                                       len--;
-                               }
+                               while (isspace((unsigned char)*cp))
+                                       cp++;
                        } else {
-                               if ((len == sizeof(synb) &&
-                                    ! strncmp(line, synb, len - 1)) ||
-                                   (len == sizeof(synr) &&
-                                    ! strncmp(line, synr, len - 1)))
+                               if (strcmp(cp, synb) == 0 ||
+                                   strcmp(cp, synr) == 0)
                                        print = 1;
                                continue;
                        }
                }
-               for (off = 0; off < len; off += nw)
-                       if ((nw = write(STDOUT_FILENO, line + off,
-                           len - off)) == -1 || nw == 0) {
-                               fclose(stream);
-                               syscall = "write";
-                               goto fail;
-                       }
+               if (fputs(cp, stdout)) {
+                       fclose(stream);
+                       syscall = "fputs";
+                       goto fail;
+               }
        }
 
        if (ferror(stream)) {
                fclose(stream);
-               syscall = "fgetln";
+               syscall = "getline";
                goto fail;
        }
 
 done:
+       free(line);
        fclose(stream);
        return;
 
 fail:
-       fprintf(stderr, "%s: %s: SYSERR: %s: %s",
-           progname, file, syscall, strerror(errno));
+       free(line);
+       warn("%s: SYSERR: %s", file, syscall);
        if (rc < MANDOCLEVEL_SYSERR)
                rc = MANDOCLEVEL_SYSERR;
 }
@@ -823,8 +895,7 @@ koptions(int *options, char *arg)
        } else if ( ! strcmp(arg, "us-ascii")) {
                *options &= ~(MPARSE_UTF8 | MPARSE_LATIN1);
        } else {
-               fprintf(stderr, "%s: -K %s: Bad argument\n",
-                   progname, arg);
+               warnx("-K %s: Bad argument", arg);
                return 0;
        }
        return 1;
@@ -843,8 +914,7 @@ moptions(int *options, char *arg)
        else if (0 == strcmp(arg, "an"))
                *options |= MPARSE_MAN;
        else {
-               fprintf(stderr, "%s: -m %s: Bad argument\n",
-                   progname, arg);
+               warnx("-m %s: Bad argument", arg);
                return 0;
        }
 
@@ -877,8 +947,7 @@ toptions(struct curparse *curp, char *arg)
        else if (0 == strcmp(arg, "pdf"))
                curp->outtype = OUTT_PDF;
        else {
-               fprintf(stderr, "%s: -T %s: Bad argument\n",
-                   progname, arg);
+               warnx("-T %s: Bad argument", arg);
                return 0;
        }
 
@@ -906,7 +975,6 @@ woptions(struct curparse *curp, char *arg)
                        curp->wstop = 1;
                        break;
                case 1:
-                       /* FALLTHROUGH */
                case 2:
                        curp->wlevel = MANDOCLEVEL_WARNING;
                        break;
@@ -920,8 +988,7 @@ woptions(struct curparse *curp, char *arg)
                        curp->wlevel = MANDOCLEVEL_BADARG;
                        break;
                default:
-                       fprintf(stderr, "%s: -W %s: Bad argument\n",
-                           progname, o);
+                       warnx("-W %s: Bad argument", o);
                        return 0;
                }
        }
@@ -935,7 +1002,7 @@ mmsg(enum mandocerr t, enum mandoclevel lvl,
 {
        const char      *mparse_msg;
 
-       fprintf(stderr, "%s: %s:", progname, file);
+       fprintf(stderr, "%s: %s:", getprogname(), file);
 
        if (line)
                fprintf(stderr, "%d:%d:", line, col + 1);
@@ -954,6 +1021,7 @@ mmsg(enum mandocerr t, enum mandoclevel lvl,
 static pid_t
 spawn_pager(struct tag_files *tag_files)
 {
+       const struct timespec timeout = { 0, 100000000 };  /* 0.1s */
 #define MAX_PAGER_ARGS 16
        char            *argv[MAX_PAGER_ARGS];
        const char      *pager;
@@ -1001,25 +1069,32 @@ spawn_pager(struct tag_files *tag_files)
 
        switch (pager_pid = fork()) {
        case -1:
-               fprintf(stderr, "%s: fork: %s\n",
-                   progname, strerror(errno));
-               exit((int)MANDOCLEVEL_SYSERR);
+               err((int)MANDOCLEVEL_SYSERR, "fork");
        case 0:
                break;
        default:
+               (void)setpgid(pager_pid, 0);
+               (void)tcsetpgrp(STDIN_FILENO, pager_pid);
+#if HAVE_PLEDGE
+               if (pledge("stdio rpath tmppath tty proc", NULL) == -1)
+                       err((int)MANDOCLEVEL_SYSERR, "pledge");
+#endif
+               tag_files->pager_pid = pager_pid;
                return pager_pid;
        }
 
        /* The child process becomes the pager. */
 
-       if (dup2(tag_files->ofd, STDOUT_FILENO) == -1) {
-               fprintf(stderr, "pager: stdout: %s\n", strerror(errno));
-               exit((int)MANDOCLEVEL_SYSERR);
-       }
+       if (dup2(tag_files->ofd, STDOUT_FILENO) == -1)
+               err((int)MANDOCLEVEL_SYSERR, "pager stdout");
        close(tag_files->ofd);
        close(tag_files->tfd);
+
+       /* Do not start the pager before controlling the terminal. */
+
+       while (tcgetpgrp(STDIN_FILENO) != getpid())
+               nanosleep(&timeout, NULL);
+
        execvp(argv[0], argv);
-       fprintf(stderr, "%s: exec %s: %s\n",
-           progname, argv[0], strerror(errno));
-       exit((int)MANDOCLEVEL_SYSERR);
+       err((int)MANDOCLEVEL_SYSERR, "exec %s", argv[0]);
 }