Sync to OpenBSD:

[mandoc.git] / roff.c

diff --git a/roff.c b/roff.c
index 07e765036deb827a6a2ad5a9e8d688dfe833cfbc..c42643cc5f9e5000fbb89cc6139fb7f8ed531803 100644 (file)
--- a/roff.c
+++ b/roff.c
@@ -1,4 +1,4 @@
-/*     $Id: roff.c,v 1.101 2010/09/04 18:31:44 kristaps Exp $ */
+/*     $Id: roff.c,v 1.108 2010/12/21 01:46:44 schwarze Exp $ */

 /*
  * Copyright (c) 2010 Kristaps Dzonsons <kristaps@bsd.lv>
  * Copyright (c) 2010 Ingo Schwarze <schwarze@openbsd.org>
 /*
  * Copyright (c) 2010 Kristaps Dzonsons <kristaps@bsd.lv>
  * Copyright (c) 2010 Ingo Schwarze <schwarze@openbsd.org>


@@ -7,9 +7,9 @@
  * purpose with or without fee is hereby granted, provided that the above
  * copyright notice and this permission notice appear in all copies.
  *
  * 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
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHORS DISCLAIM ALL WARRANTIES

  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF

- * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHORS 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
  * 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


@@ -45,6 +45,7 @@
 #endif
 
 enum   rofft {
 #endif
 
 enum   rofft {

+       ROFF_ad,

        ROFF_am,
        ROFF_ami,
        ROFF_am1,
        ROFF_am,
        ROFF_ami,
        ROFF_am1,


@@ -53,14 +54,19 @@ enum        rofft {
        ROFF_de1,
        ROFF_ds,
        ROFF_el,
        ROFF_de1,
        ROFF_ds,
        ROFF_el,

+       ROFF_hy,

        ROFF_ie,
        ROFF_if,
        ROFF_ig,
        ROFF_ie,
        ROFF_if,
        ROFF_ig,

+       ROFF_ne,
+       ROFF_nh,
+       ROFF_nr,

        ROFF_rm,
        ROFF_rm,

+       ROFF_so,

        ROFF_tr,
        ROFF_cblock,
        ROFF_ccond, /* FIXME: remove this. */
        ROFF_tr,
        ROFF_cblock,
        ROFF_ccond, /* FIXME: remove this. */

-       ROFF_nr,
+       ROFF_USERDEF,

        ROFF_MAX
 };
 
        ROFF_MAX
 };
 


@@ -83,7 +89,8 @@ struct        roff {
        enum roffrule    rstack[RSTACK_MAX]; /* stack of !`ie' rules */
        int              rstackpos; /* position in rstack */
        struct regset   *regs; /* read/writable registers */
        enum roffrule    rstack[RSTACK_MAX]; /* stack of !`ie' rules */
        int              rstackpos; /* position in rstack */
        struct regset   *regs; /* read/writable registers */

-       struct roffstr  *first_string;
+       struct roffstr  *first_string; /* user-defined strings & macros */
+       const char      *current_string; /* value of last called user macro */

 };
 
 struct roffnode {
 };
 
 struct roffnode {


@@ -91,6 +98,7 @@ struct        roffnode {
        struct roffnode *parent; /* up one in stack */
        int              line; /* parse line */
        int              col; /* parse col */
        struct roffnode *parent; /* up one in stack */
        int              line; /* parse line */
        int              col; /* parse col */

+       char            *name; /* node name, e.g. macro name */

        char            *end; /* end-rules: custom token */
        int              endspan; /* end-rules: next-line or infty */
        enum roffrule    rule; /* current evaluation rule */
        char            *end; /* end-rules: custom token */
        int              endspan; /* end-rules: next-line or infty */
        enum roffrule    rule; /* current evaluation rule */


@@ -130,13 +138,15 @@ static    enum roffrule    roff_evalcond(const char *, int *);
 static void             roff_freestr(struct roff *);
 static const char      *roff_getstrn(const struct roff *, 
                                const char *, size_t);
 static void             roff_freestr(struct roff *);
 static const char      *roff_getstrn(const struct roff *, 
                                const char *, size_t);

-static enum rofferr     roff_line(ROFF_ARGS);
+static enum rofferr     roff_line_ignore(ROFF_ARGS);
+static enum rofferr     roff_line_error(ROFF_ARGS);

 static enum rofferr     roff_nr(ROFF_ARGS);
 static int              roff_res(struct roff *, 
                                char **, size_t *, int);
 static void             roff_setstr(struct roff *,
 static enum rofferr     roff_nr(ROFF_ARGS);
 static int              roff_res(struct roff *, 
                                char **, size_t *, int);
 static void             roff_setstr(struct roff *,

-                               const char *, const char *);
-static char            *roff_strdup(const char *);
+                               const char *, const char *, int);
+static enum rofferr     roff_so(ROFF_ARGS);
+static enum rofferr     roff_userdef(ROFF_ARGS);

 
 /* See roff_hash_find() */
 
 
 /* See roff_hash_find() */
 


@@ -147,6 +157,7 @@ static      char            *roff_strdup(const char *);
 static struct roffmac  *hash[HASHWIDTH];
 
 static struct roffmac   roffs[ROFF_MAX] = {
 static struct roffmac  *hash[HASHWIDTH];
 
 static struct roffmac   roffs[ROFF_MAX] = {

+       { "ad", roff_line_ignore, NULL, NULL, 0, NULL },

        { "am", roff_block, roff_block_text, roff_block_sub, 0, NULL },
        { "ami", roff_block, roff_block_text, roff_block_sub, 0, NULL },
        { "am1", roff_block, roff_block_text, roff_block_sub, 0, NULL },
        { "am", roff_block, roff_block_text, roff_block_sub, 0, NULL },
        { "ami", roff_block, roff_block_text, roff_block_sub, 0, NULL },
        { "am1", roff_block, roff_block_text, roff_block_sub, 0, NULL },


@@ -155,24 +166,29 @@ static    struct roffmac   roffs[ROFF_MAX] = {
        { "de1", roff_block, roff_block_text, roff_block_sub, 0, NULL },
        { "ds", roff_ds, NULL, NULL, 0, NULL },
        { "el", roff_cond, roff_cond_text, roff_cond_sub, ROFFMAC_STRUCT, NULL },
        { "de1", roff_block, roff_block_text, roff_block_sub, 0, NULL },
        { "ds", roff_ds, NULL, NULL, 0, NULL },
        { "el", roff_cond, roff_cond_text, roff_cond_sub, ROFFMAC_STRUCT, NULL },

+       { "hy", roff_line_ignore, NULL, NULL, 0, NULL },

        { "ie", roff_cond, roff_cond_text, roff_cond_sub, ROFFMAC_STRUCT, NULL },
        { "if", roff_cond, roff_cond_text, roff_cond_sub, ROFFMAC_STRUCT, NULL },
        { "ig", roff_block, roff_block_text, roff_block_sub, 0, NULL },
        { "ie", roff_cond, roff_cond_text, roff_cond_sub, ROFFMAC_STRUCT, NULL },
        { "if", roff_cond, roff_cond_text, roff_cond_sub, ROFFMAC_STRUCT, NULL },
        { "ig", roff_block, roff_block_text, roff_block_sub, 0, NULL },

-       { "rm", roff_line, NULL, NULL, 0, NULL },
-       { "tr", roff_line, NULL, NULL, 0, NULL },
+       { "ne", roff_line_ignore, NULL, NULL, 0, NULL },
+       { "nh", roff_line_ignore, NULL, NULL, 0, NULL },
+       { "nr", roff_nr, NULL, NULL, 0, NULL },
+       { "rm", roff_line_error, NULL, NULL, 0, NULL },
+       { "so", roff_so, NULL, NULL, 0, NULL },
+       { "tr", roff_line_ignore, NULL, NULL, 0, NULL },

        { ".", roff_cblock, NULL, NULL, 0, NULL },
        { "\\}", roff_ccond, NULL, NULL, 0, NULL },
        { ".", roff_cblock, NULL, NULL, 0, NULL },
        { "\\}", roff_ccond, NULL, NULL, 0, NULL },

-       { "nr", roff_nr, NULL, NULL, 0, NULL },
+       { NULL, roff_userdef, NULL, NULL, 0, NULL },

 };
 
 static void             roff_free1(struct roff *);
 };
 
 static void             roff_free1(struct roff *);

-static enum rofft       roff_hash_find(const char *);
+static enum rofft       roff_hash_find(const char *, size_t);

 static void             roff_hash_init(void);
 static void             roffnode_cleanscope(struct roff *);
 static void             roff_hash_init(void);
 static void             roffnode_cleanscope(struct roff *);

-static void             roffnode_push(struct roff *, 
-                               enum rofft, int, int);
+static void             roffnode_push(struct roff *, enum rofft,
+                               const char *, int, int);

 static void             roffnode_pop(struct roff *);
 static void             roffnode_pop(struct roff *);

-static enum rofft       roff_parse(const char *, int *);
+static enum rofft       roff_parse(struct roff *, const char *, int *);

 static int              roff_parse_nat(const char *, unsigned int *);
 
 /* See roff_hash_find() */
 static int              roff_parse_nat(const char *, unsigned int *);
 
 /* See roff_hash_find() */


@@ -184,7 +200,7 @@ roff_hash_init(void)
        struct roffmac   *n;
        int               buc, i;
 
        struct roffmac   *n;
        int               buc, i;
 

-       for (i = 0; i < (int)ROFF_MAX; i++) {
+       for (i = 0; i < (int)ROFF_USERDEF; i++) {

                assert(roffs[i].name[0] >= ASCII_LO);
                assert(roffs[i].name[0] <= ASCII_HI);
 
                assert(roffs[i].name[0] >= ASCII_LO);
                assert(roffs[i].name[0] <= ASCII_HI);
 


@@ -205,7 +221,7 @@ roff_hash_init(void)
  * the nil-terminated string name could be found.
  */
 static enum rofft
  * the nil-terminated string name could be found.
  */
 static enum rofft

-roff_hash_find(const char *p)
+roff_hash_find(const char *p, size_t s)

 {
        int              buc;
        struct roffmac  *n;
 {
        int              buc;
        struct roffmac  *n;


@@ -225,7 +241,7 @@ roff_hash_find(const char *p)
        if (NULL == (n = hash[buc]))
                return(ROFF_MAX);
        for ( ; n; n = n->next)
        if (NULL == (n = hash[buc]))
                return(ROFF_MAX);
        for ( ; n; n = n->next)

-               if (0 == strcmp(n->name, p))
+               if (0 == strncmp(n->name, p, s) && '\0' == n->name[(int)s])

                        return((enum rofft)(n - roffs));
 
        return(ROFF_MAX);
                        return((enum rofft)(n - roffs));
 
        return(ROFF_MAX);


@@ -250,8 +266,8 @@ roffnode_pop(struct roff *r)
 
        ROFF_DEBUG("roff: popping scope\n");
        r->last = r->last->parent;
 
        ROFF_DEBUG("roff: popping scope\n");
        r->last = r->last->parent;

-       if (p->end)
-               free(p->end);
+       free(p->name);
+       free(p->end);

        free(p);
 }
 
        free(p);
 }
 


@@ -261,12 +277,15 @@ roffnode_pop(struct roff *r)
  * removed with roffnode_pop().
  */
 static void
  * removed with roffnode_pop().
  */
 static void

-roffnode_push(struct roff *r, enum rofft tok, int line, int col)
+roffnode_push(struct roff *r, enum rofft tok, const char *name,
+               int line, int col)

 {
        struct roffnode *p;
 
        p = mandoc_calloc(1, sizeof(struct roffnode));
        p->tok = tok;
 {
        struct roffnode *p;
 
        p = mandoc_calloc(1, sizeof(struct roffnode));
        p->tok = tok;

+       if (name)
+               p->name = mandoc_strdup(name);

        p->parent = r->last;
        p->line = line;
        p->col = col;
        p->parent = r->last;
        p->line = line;
        p->col = col;


@@ -327,15 +346,40 @@ roff_alloc(struct regset *regs, void *data, const mandocmsg msg)
 static int
 roff_res(struct roff *r, char **bufp, size_t *szp, int pos)
 {
 static int
 roff_res(struct roff *r, char **bufp, size_t *szp, int pos)
 {

-       const char      *cp, *cpp, *st, *res;
+       const char      *stesc; /* start of an escape sequence ('\\') */
+       const char      *stnam; /* start of the name, after "[(*" */
+       const char      *cp;    /* end of the name, e.g. before ']' */
+       const char      *res;   /* the string to be substituted */

        int              i, maxl;
        size_t           nsz;
        char            *n;
 
        int              i, maxl;
        size_t           nsz;
        char            *n;
 

-       /* LINTED */
-       for (cp = &(*bufp)[pos]; (cpp = strstr(cp, "\\*")); cp++) {
-               cp = cpp + 2;
+       /* Search for a leading backslash and save a pointer to it. */
+
+       cp = *bufp + pos;
+       while (NULL != (cp = strchr(cp, '\\'))) {
+               stesc = cp++;
+
+               /*
+                * The second character must be an asterisk.
+                * If it isn't, skip it anyway:  It is escaped,
+                * so it can't start another escape sequence.
+                */
+
+               if ('\0' == *cp)
+                       return(1);
+               if ('*' != *cp++)
+                       continue;
+
+               /*
+                * The third character decides the length
+                * of the name of the string.
+                * Save a pointer to the name.
+                */
+

                switch (*cp) {
                switch (*cp) {

+               case ('\0'):
+                       return(1);

                case ('('):
                        cp++;
                        maxl = 2;
                case ('('):
                        cp++;
                        maxl = 2;


@@ -348,8 +392,9 @@ roff_res(struct roff *r, char **bufp, size_t *szp, int pos)
                        maxl = 1;
                        break;
                }
                        maxl = 1;
                        break;
                }

+               stnam = cp;

 
 

-               st = cp;
+               /* Advance to the end of the name. */

 
                for (i = 0; 0 == maxl || i < maxl; i++, cp++) {
                        if ('\0' == *cp)
 
                for (i = 0; 0 == maxl || i < maxl; i++, cp++) {
                        if ('\0' == *cp)


@@ -358,21 +403,26 @@ roff_res(struct roff *r, char **bufp, size_t *szp, int pos)
                                break;
                }
 
                                break;
                }
 

-               res = roff_getstrn(r, st, (size_t)i);
+               /*
+                * Retrieve the replacement string; if it is
+                * undefined, resume searching for escapes.
+                */
+
+               res = roff_getstrn(r, stnam, (size_t)i);

 
                if (NULL == res) {
                        cp -= maxl ? 1 : 0;
                        continue;
                }
 
 
                if (NULL == res) {
                        cp -= maxl ? 1 : 0;
                        continue;
                }
 

+               /* Replace the escape sequence by the string. */
+

                ROFF_DEBUG("roff: splicing reserved: [%.*s]\n", i, st);
 
                nsz = *szp + strlen(res) + 1;
                n = mandoc_malloc(nsz);
 
                ROFF_DEBUG("roff: splicing reserved: [%.*s]\n", i, st);
 
                nsz = *szp + strlen(res) + 1;
                n = mandoc_malloc(nsz);
 

-               *n = '\0';
-
-               strlcat(n, *bufp, (size_t)(cpp - *bufp + 1));
+               strlcpy(n, *bufp, (size_t)(stesc - *bufp + 1));

                strlcat(n, res, nsz);
                strlcat(n, cp + (maxl ? 0 : 1), nsz);
 
                strlcat(n, res, nsz);
                strlcat(n, cp + (maxl ? 0 : 1), nsz);
 


@@ -400,7 +450,7 @@ roff_parseln(struct roff *r, int ln, char **bufp,
         */
 
        if (r->first_string && ! roff_res(r, bufp, szp, pos))
         */
 
        if (r->first_string && ! roff_res(r, bufp, szp, pos))

-               return(ROFF_RERUN);
+               return(ROFF_REPARSE);

 
        /*
         * First, if a scope is open and we're not a macro, pass the
 
        /*
         * First, if a scope is open and we're not a macro, pass the


@@ -441,11 +491,13 @@ roff_parseln(struct roff *r, int ln, char **bufp,
         */
 
        ppos = pos;
         */
 
        ppos = pos;

-       if (ROFF_MAX == (t = roff_parse(*bufp, &pos)))
+       if (ROFF_MAX == (t = roff_parse(r, *bufp, &pos)))

                return(ROFF_CONT);
 
                return(ROFF_CONT);
 

-       ROFF_DEBUG("roff: intercept new-scope: %s, [%s]\n", 
-                       roffs[t].name, &(*bufp)[pos]);
+       ROFF_DEBUG("roff: intercept new-scope: [%s], [%s]\n", 
+                       ROFF_USERDEF == t ? r->current_string : roffs[t].name, 
+                       &(*bufp)[pos]);
+

        assert(roffs[t].proc);
        return((*roffs[t].proc)
                        (r, t, bufp, szp, 
        assert(roffs[t].proc);
        return((*roffs[t].proc)
                        (r, t, bufp, szp, 


@@ -469,35 +521,28 @@ roff_endparse(struct roff *r)
  * form of ".foo xxx" in the usual way.
  */
 static enum rofft
  * form of ".foo xxx" in the usual way.
  */
 static enum rofft

-roff_parse(const char *buf, int *pos)
+roff_parse(struct roff *r, const char *buf, int *pos)

 {
 {

-       int              j;
-       char             mac[5];
+       const char      *mac;
+       size_t           maclen;

        enum rofft       t;
 
        assert(ROFF_CTL(buf[*pos]));
        (*pos)++;
 
        enum rofft       t;
 
        assert(ROFF_CTL(buf[*pos]));
        (*pos)++;
 

-       while (buf[*pos] && (' ' == buf[*pos] || '\t' == buf[*pos]))
+       while (' ' == buf[*pos] || '\t' == buf[*pos])

                (*pos)++;
 
        if ('\0' == buf[*pos])
                return(ROFF_MAX);
 
                (*pos)++;
 
        if ('\0' == buf[*pos])
                return(ROFF_MAX);
 

-       for (j = 0; j < 4; j++, (*pos)++)
-               if ('\0' == (mac[j] = buf[*pos]))
-                       break;
-               else if (' ' == buf[*pos] || (j && '\\' == buf[*pos]))
-                       break;
-
-       if (j == 4 || j < 1)
-               return(ROFF_MAX);
-
-       mac[j] = '\0';
+       mac = buf + *pos;
+       maclen = strcspn(mac, " \\\t\0");

 
 

-       if (ROFF_MAX == (t = roff_hash_find(mac)))
-               return(t);
+       t = (r->current_string = roff_getstrn(r, mac, maclen))
+           ? ROFF_USERDEF : roff_hash_find(mac, maclen);

 
 

+       *pos += maclen;

        while (buf[*pos] && ' ' == buf[*pos])
                (*pos)++;
 
        while (buf[*pos] && ' ' == buf[*pos])
                (*pos)++;
 


@@ -549,11 +594,10 @@ roff_cblock(ROFF_ARGS)
        case (ROFF_am1):
                /* FALLTHROUGH */
        case (ROFF_de):
        case (ROFF_am1):
                /* FALLTHROUGH */
        case (ROFF_de):

+               /* ROFF_de1 is remapped to ROFF_de in roff_block(). */

                /* FALLTHROUGH */
        case (ROFF_dei):
                /* FALLTHROUGH */
                /* FALLTHROUGH */
        case (ROFF_dei):
                /* FALLTHROUGH */

-       case (ROFF_de1):
-               /* FALLTHROUGH */

        case (ROFF_ig):
                break;
        default:
        case (ROFF_ig):
                break;
        default:


@@ -631,25 +675,55 @@ roff_block(ROFF_ARGS)
 {
        int             sv;
        size_t          sz;
 {
        int             sv;
        size_t          sz;

+       char            *name;
+
+       name = NULL;
+
+       if (ROFF_ig != tok) {
+               if ('\0' == (*bufp)[pos]) {
+                       (*r->msg)(MANDOCERR_NOARGS, r->data, ln, ppos, NULL);
+                       return(ROFF_IGN);
+               }
+
+               /*
+                * Re-write `de1', since we don't really care about
+                * groff's strange compatibility mode, into `de'.
+                */
+
+               if (ROFF_de1 == tok)
+                       tok = ROFF_de;
+               if (ROFF_de == tok)
+                       name = *bufp + pos;
+               else
+                       (*r->msg)(MANDOCERR_REQUEST, r->data, ln, ppos,
+                           roffs[tok].name);

 
 

-       if (ROFF_ig != tok && '\0' == (*bufp)[pos]) {
-               if ( ! (*r->msg)(MANDOCERR_NOARGS, r->data, ln, ppos, NULL))
-                       return(ROFF_ERR);
-               return(ROFF_IGN);
-       } else if (ROFF_ig != tok) {

                while ((*bufp)[pos] && ' ' != (*bufp)[pos])
                        pos++;
                while ((*bufp)[pos] && ' ' != (*bufp)[pos])
                        pos++;

+

                while (' ' == (*bufp)[pos])
                while (' ' == (*bufp)[pos])

-                       pos++;
+                       (*bufp)[pos++] = '\0';

        }
 
        }
 

-       roffnode_push(r, tok, ln, ppos);
+       roffnode_push(r, tok, name, ln, ppos);
+
+       /*
+        * At the beginning of a `de' macro, clear the existing string
+        * with the same name, if there is one.  New content will be
+        * added from roff_block_text() in multiline mode.
+        */
+
+       if (ROFF_de == tok)
+               roff_setstr(r, name, "", 0);

 
        if ('\0' == (*bufp)[pos])
                return(ROFF_IGN);
 
 
        if ('\0' == (*bufp)[pos])
                return(ROFF_IGN);
 

+       /* If present, process the custom end-of-line marker. */
+

        sv = pos;
        sv = pos;

-       while ((*bufp)[pos] && ' ' != (*bufp)[pos] && 
+       while ((*bufp)[pos] &&
+                       ' ' != (*bufp)[pos] && 

                        '\t' != (*bufp)[pos])
                pos++;
 
                        '\t' != (*bufp)[pos])
                pos++;
 


@@ -671,8 +745,7 @@ roff_block(ROFF_ARGS)
        r->last->end[(int)sz] = '\0';
 
        if ((*bufp)[pos])
        r->last->end[(int)sz] = '\0';
 
        if ((*bufp)[pos])

-               if ( ! (*r->msg)(MANDOCERR_ARGSLOST, r->data, ln, pos, NULL))
-                       return(ROFF_ERR);
+               (*r->msg)(MANDOCERR_ARGSLOST, r->data, ln, pos, NULL);

 
        return(ROFF_IGN);
 }
 
        return(ROFF_IGN);
 }


@@ -710,7 +783,7 @@ roff_block_sub(ROFF_ARGS)
                        roffnode_pop(r);
                        roffnode_cleanscope(r);
 
                        roffnode_pop(r);
                        roffnode_cleanscope(r);
 

-                       if (ROFF_MAX != roff_parse(*bufp, &pos))
+                       if (ROFF_MAX != roff_parse(r, *bufp, &pos))

                                return(ROFF_RERUN);
                        return(ROFF_IGN);
                }
                                return(ROFF_RERUN);
                        return(ROFF_IGN);
                }


@@ -722,11 +795,17 @@ roff_block_sub(ROFF_ARGS)
         */
 
        ppos = pos;
         */
 
        ppos = pos;

-       t = roff_parse(*bufp, &pos);
+       t = roff_parse(r, *bufp, &pos);

 
 

-       /* If we're not a comment-end, then throw it away. */
-       if (ROFF_cblock != t)
+       /*
+        * Macros other than block-end are only significant
+        * in `de' blocks; elsewhere, simply throw them away.
+        */
+       if (ROFF_cblock != t) {
+               if (ROFF_de == tok)
+                       roff_setstr(r, r->last->name, *bufp + ppos, 1);

                return(ROFF_IGN);
                return(ROFF_IGN);

+       }

 
        assert(roffs[t].proc);
        return((*roffs[t].proc)(r, t, bufp, szp, 
 
        assert(roffs[t].proc);
        return((*roffs[t].proc)(r, t, bufp, szp, 


@@ -739,6 +818,9 @@ static enum rofferr
 roff_block_text(ROFF_ARGS)
 {
 
 roff_block_text(ROFF_ARGS)
 {
 

+       if (ROFF_de == tok)
+               roff_setstr(r, r->last->name, *bufp + pos, 1);
+

        return(ROFF_IGN);
 }
 
        return(ROFF_IGN);
 }
 


@@ -749,7 +831,6 @@ roff_cond_sub(ROFF_ARGS)
 {
        enum rofft       t;
        enum roffrule    rr;
 {
        enum rofft       t;
        enum roffrule    rr;

-       struct roffnode *l;

 
        ppos = pos;
        rr = r->last->rule;
 
        ppos = pos;
        rr = r->last->rule;


@@ -759,13 +840,12 @@ roff_cond_sub(ROFF_ARGS)
         * continue. 
         */
 
         * continue. 
         */
 

-       l = r->last;

        roffnode_cleanscope(r);
 
        roffnode_cleanscope(r);
 

-       if (ROFF_MAX == (t = roff_parse(*bufp, &pos))) {
+       if (ROFF_MAX == (t = roff_parse(r, *bufp, &pos))) {

                if ('\\' == (*bufp)[pos] && '}' == (*bufp)[pos + 1])
                        return(roff_ccond
                if ('\\' == (*bufp)[pos] && '}' == (*bufp)[pos + 1])
                        return(roff_ccond

-                               (r, ROFF_ccond, bufp, szp, 
+                               (r, ROFF_ccond, bufp, szp,

                                 ln, pos, pos + 2, offs));
                return(ROFFRULE_DENY == rr ? ROFF_IGN : ROFF_CONT);
        }
                                 ln, pos, pos + 2, offs));
                return(ROFFRULE_DENY == rr ? ROFF_IGN : ROFF_CONT);
        }


@@ -840,15 +920,22 @@ roff_evalcond(const char *v, int *pos)
        return(ROFFRULE_DENY);
 }
 
        return(ROFFRULE_DENY);
 }
 

-

 /* ARGSUSED */
 static enum rofferr
 /* ARGSUSED */
 static enum rofferr

-roff_line(ROFF_ARGS)
+roff_line_ignore(ROFF_ARGS)

 {
 
        return(ROFF_IGN);
 }
 
 {
 
        return(ROFF_IGN);
 }
 

+/* ARGSUSED */
+static enum rofferr
+roff_line_error(ROFF_ARGS)
+{
+
+       (*r->msg)(MANDOCERR_REQUEST, r->data, ln, ppos, roffs[tok].name);
+       return(ROFF_IGN);
+}

 
 /* ARGSUSED */
 static enum rofferr
 
 /* ARGSUSED */
 static enum rofferr


@@ -891,12 +978,11 @@ roff_cond(ROFF_ARGS)
         */
 
        if ('\0' == (*bufp)[pos] && sv != pos) {
         */
 
        if ('\0' == (*bufp)[pos] && sv != pos) {

-               if ((*r->msg)(MANDOCERR_NOARGS, r->data, ln, ppos, NULL))
-                       return(ROFF_IGN);
-               return(ROFF_ERR);
+               (*r->msg)(MANDOCERR_NOARGS, r->data, ln, ppos, NULL);
+               return(ROFF_IGN);

        }
 
        }
 

-       roffnode_push(r, tok, ln, ppos);
+       roffnode_push(r, tok, NULL, ln, ppos);

 
        r->last->rule = rule;
 
 
        r->last->rule = rule;
 


@@ -993,7 +1079,7 @@ roff_ds(ROFF_ARGS)
                string++;
 
        /* The rest is the value. */
                string++;
 
        /* The rest is the value. */

-       roff_setstr(r, name, string);
+       roff_setstr(r, name, string, 0);

        return(ROFF_IGN);
 }
 
        return(ROFF_IGN);
 }
 


@@ -1040,50 +1126,190 @@ roff_nr(ROFF_ARGS)
        return(ROFF_IGN);
 }
 
        return(ROFF_IGN);
 }
 

-
-static char *
-roff_strdup(const char *name)
+/* ARGSUSED */
+static enum rofferr
+roff_so(ROFF_ARGS)

 {
 {

-       char            *namecopy, *sv;
+       char *name;

 
 

-       /* 
-        * This isn't a nice simple mandoc_strdup() because we must
-        * handle roff's stupid double-escape rule. 
+       (*r->msg)(MANDOCERR_SO, r->data, ln, ppos, NULL);
+
+       /*
+        * Handle `so'.  Be EXTREMELY careful, as we shouldn't be
+        * opening anything that's not in our cwd or anything beneath
+        * it.  Thus, explicitly disallow traversing up the file-system
+        * or using absolute paths.

         */
         */

-       sv = namecopy = mandoc_malloc(strlen(name) + 1);
-       while (*name) {
-               if ('\\' == *name && '\\' == *(name + 1))
-                       name++;
-               *namecopy++ = *name++;
+
+       name = *bufp + pos;
+       if ('/' == *name || strstr(name, "../") || strstr(name, "/..")) {
+               (*r->msg)(MANDOCERR_SOPATH, r->data, ln, pos, NULL);
+               return(ROFF_ERR);

        }
 
        }
 

-       *namecopy = '\0';
-       return(sv);
+       *offs = pos;
+       return(ROFF_SO);

 }
 
 }
 

+/* ARGSUSED */
+static enum rofferr
+roff_userdef(ROFF_ARGS)
+{
+       const char       *arg[9];
+       char             *cp, *n1, *n2;
+       int               i, quoted, pairs;
+
+       /*
+        * Collect pointers to macro argument strings
+        * and null-terminate them.
+        */
+       cp = *bufp + pos;
+       for (i = 0; i < 9; i++) {
+               /* Quoting can only start with a new word. */
+               if ('"' == *cp) {
+                       quoted = 1;
+                       cp++;
+               } else
+                       quoted = 0;
+               arg[i] = cp;
+               for (pairs = 0; '\0' != *cp; cp++) {
+                       /* Unquoted arguments end at blanks. */
+                       if (0 == quoted) {
+                               if (' ' == *cp)
+                                       break;
+                               continue;
+                       }
+                       /* After pairs of quotes, move left. */
+                       if (pairs)
+                               cp[-pairs] = cp[0];
+                       /* Pairs of quotes do not end words, ... */
+                       if ('"' == cp[0] && '"' == cp[1]) {
+                               pairs++;
+                               cp++;
+                               continue;
+                       }
+                       /* ... but solitary quotes do. */
+                       if ('"' != *cp)
+                               continue;
+                       if (pairs)
+                               cp[-pairs] = '\0';
+                       *cp = ' ';
+                       break;
+               }
+               /* Last argument; the remaining ones are empty strings. */
+               if ('\0' == *cp)
+                       continue;
+               /* Null-terminate argument and move to the next one. */
+               *cp++ = '\0';
+               while (' ' == *cp)
+                       cp++;
+       }

 
 

+       /*
+        * Expand macro arguments.
+        */
+       *szp = 0;
+       n1 = cp = mandoc_strdup(r->current_string);
+       while (NULL != (cp = strstr(cp, "\\$"))) {
+               i = cp[2] - '1';
+               if (0 > i || 8 < i) {
+                       /* Not an argument invocation. */
+                       cp += 2;
+                       continue;
+               }
+
+               *szp = strlen(n1) - 3 + strlen(arg[i]) + 1;
+               n2 = mandoc_malloc(*szp);
+
+               strlcpy(n2, n1, (size_t)(cp - n1 + 1));
+               strlcat(n2, arg[i], *szp);
+               strlcat(n2, cp + 3, *szp);
+
+               cp = n2 + (cp - n1);
+               free(n1);
+               n1 = n2;
+       }
+
+       /*
+        * Replace the macro invocation
+        * by the expanded macro.
+        */
+       free(*bufp);
+       *bufp = n1;
+       if (0 == *szp)
+               *szp = strlen(*bufp) + 1;
+
+       return(*szp > 1 && '\n' == (*bufp)[(int)*szp - 2] ?
+          ROFF_REPARSE : ROFF_APPEND);
+}
+
+/*
+ * Store *string into the user-defined string called *name.
+ * In multiline mode, append to an existing entry and append '\n';
+ * else replace the existing entry, if there is one.
+ * To clear an existing entry, call with (*r, *name, NULL, 0).
+ */

 static void
 static void

-roff_setstr(struct roff *r, const char *name, const char *string)
+roff_setstr(struct roff *r, const char *name, const char *string,
+       int multiline)

 {
        struct roffstr   *n;
 {
        struct roffstr   *n;

-       char             *namecopy;
+       char             *c;
+       size_t            oldch, newch;

 
 

+       /* Search for an existing string with the same name. */

        n = r->first_string;
        while (n && strcmp(name, n->name))
                n = n->next;
 
        if (NULL == n) {
        n = r->first_string;
        while (n && strcmp(name, n->name))
                n = n->next;
 
        if (NULL == n) {

-               namecopy = mandoc_strdup(name);
+               /* Create a new string table entry. */

                n = mandoc_malloc(sizeof(struct roffstr));
                n = mandoc_malloc(sizeof(struct roffstr));

-               n->name = namecopy;
+               n->name = mandoc_strdup(name);
+               n->string = NULL;

                n->next = r->first_string;
                r->first_string = n;
                n->next = r->first_string;
                r->first_string = n;

-       } else
+       } else if (0 == multiline) {
+               /* In multiline mode, append; else replace. */

                free(n->string);
                free(n->string);

+               n->string = NULL;
+       }
+
+       if (NULL == string)
+               return;
+
+       /*
+        * One additional byte for the '\n' in multiline mode,
+        * and one for the terminating '\0'.
+        */
+       newch = strlen(string) + (multiline ? 2 : 1);
+       if (NULL == n->string) {
+               n->string = mandoc_malloc(newch);
+               *n->string = '\0';
+               oldch = 0;
+       } else {
+               oldch = strlen(n->string);
+               n->string = mandoc_realloc(n->string, oldch + newch);
+       }
+
+       /* Skip existing content in the destination buffer. */
+       c = n->string + oldch;
+
+       /* Append new content to the destination buffer. */
+       while (*string) {
+               /*
+                * Rudimentary roff copy mode:
+                * Handle escaped backslashes.
+                */
+               if ('\\' == *string && '\\' == *(string + 1))
+                       string++;
+               *c++ = *string++;
+       }

 
 

-       /* Don't use mandoc_strdup: clean out double-escapes. */
-       n->string = string ? roff_strdup(string) : NULL;
-       ROFF_DEBUG("roff: new symbol: [%s] = [%s]\n", name, n->string);
+       /* Append terminating bytes. */
+       if (multiline)
+               *c++ = '\n';
+       *c = '\0';

 }
 
 
 }