[BACK]Return to eval.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / m4

Annotation of src/usr.bin/m4/eval.c, Revision 1.64

1.64    ! espie       1: /*     $OpenBSD: eval.c,v 1.63 2006/03/24 08:03:44 espie Exp $ */
1.7       millert     2: /*     $NetBSD: eval.c,v 1.7 1996/11/10 21:21:29 pk Exp $      */
1.1       deraadt     3:
                      4: /*
                      5:  * Copyright (c) 1989, 1993
                      6:  *     The Regents of the University of California.  All rights reserved.
                      7:  *
                      8:  * This code is derived from software contributed to Berkeley by
                      9:  * Ozan Yigit at York University.
                     10:  *
                     11:  * Redistribution and use in source and binary forms, with or without
                     12:  * modification, are permitted provided that the following conditions
                     13:  * are met:
                     14:  * 1. Redistributions of source code must retain the above copyright
                     15:  *    notice, this list of conditions and the following disclaimer.
                     16:  * 2. Redistributions in binary form must reproduce the above copyright
                     17:  *    notice, this list of conditions and the following disclaimer in the
                     18:  *    documentation and/or other materials provided with the distribution.
1.47      millert    19:  * 3. Neither the name of the University nor the names of its contributors
1.1       deraadt    20:  *    may be used to endorse or promote products derived from this software
                     21:  *    without specific prior written permission.
                     22:  *
                     23:  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
                     24:  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
                     25:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
                     26:  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
                     27:  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
                     28:  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
                     29:  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
                     30:  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
                     31:  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
                     32:  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
                     33:  * SUCH DAMAGE.
                     34:  */
                     35:
                     36: /*
                     37:  * eval.c
                     38:  * Facility: m4 macro processor
                     39:  * by: oz
                     40:  */
                     41:
                     42: #include <sys/types.h>
1.63      espie      43: #include <err.h>
1.1       deraadt    44: #include <errno.h>
1.56      espie      45: #include <limits.h>
1.1       deraadt    46: #include <unistd.h>
                     47: #include <stdio.h>
                     48: #include <stdlib.h>
1.13      espie      49: #include <stddef.h>
1.1       deraadt    50: #include <string.h>
1.6       millert    51: #include <fcntl.h>
1.1       deraadt    52: #include "mdef.h"
                     53: #include "stdd.h"
                     54: #include "extern.h"
                     55: #include "pathnames.h"
                     56:
1.43      millert    57: static void    dodefn(const char *);
                     58: static void    dopushdef(const char *, const char *);
                     59: static void    dodump(const char *[], int);
                     60: static void    dotrace(const char *[], int, int);
                     61: static void    doifelse(const char *[], int);
                     62: static int     doincl(const char *);
                     63: static int     dopaste(const char *);
                     64: static void    dochq(const char *[], int);
                     65: static void    dochc(const char *[], int);
1.55      espie      66: static void    dom4wrap(const char *);
1.43      millert    67: static void    dodiv(int);
                     68: static void    doundiv(const char *[], int);
                     69: static void    dosub(const char *[], int);
                     70: static void    map(char *, const char *, const char *, const char *);
                     71: static const char *handledash(char *, char *, const char *);
                     72: static void    expand_builtin(const char *[], int, int);
                     73: static void    expand_macro(const char *[], int);
1.49      espie      74: static void    dump_one_def(const char *, struct macro_definition *);
1.30      espie      75:
1.34      espie      76: unsigned long  expansion_id;
1.30      espie      77:
1.1       deraadt    78: /*
1.30      espie      79:  * eval - eval all macros and builtins calls
1.39      espie      80:  *       argc - number of elements in argv.
                     81:  *       argv - element vector :
                     82:  *                     argv[0] = definition of a user
1.49      espie      83:  *                               macro or NULL if built-in.
1.39      espie      84:  *                     argv[1] = name of the macro or
                     85:  *                               built-in.
                     86:  *                     argv[2] = parameters to user-defined
                     87:  *                        .      macro or built-in.
                     88:  *                        .
                     89:  *
                     90:  * A call in the form of macro-or-builtin() will result in:
                     91:  *                     argv[0] = nullstr
                     92:  *                     argv[1] = macro-or-builtin
                     93:  *                     argv[2] = nullstr
                     94:  *
                     95:  * argc is 3 for macro-or-builtin() and 2 for macro-or-builtin
1.30      espie      96:  */
                     97: void
1.50      espie      98: eval(const char *argv[], int argc, int td, int is_traced)
1.30      espie      99: {
1.59      espie     100:        size_t mark = SIZE_MAX;
1.35      espie     101:
1.34      espie     102:        expansion_id++;
1.30      espie     103:        if (td & RECDEF)
1.60      espie     104:                m4errx(1, "expanding recursive definition for %s.", argv[1]);
1.50      espie     105:        if (is_traced)
1.35      espie     106:                mark = trace(argv, argc, infile+ilevel);
1.30      espie     107:        if (td == MACRTYPE)
                    108:                expand_macro(argv, argc);
                    109:        else
                    110:                expand_builtin(argv, argc, td);
1.59      espie     111:        if (mark != SIZE_MAX)
1.35      espie     112:                finish_trace(mark);
1.30      espie     113: }
                    114:
                    115: /*
                    116:  * expand_builtin - evaluate built-in macros.
1.1       deraadt   117:  */
                    118: void
1.44      espie     119: expand_builtin(const char *argv[], int argc, int td)
1.1       deraadt   120: {
1.17      espie     121:        int c, n;
1.39      espie     122:        int ac;
1.1       deraadt   123:        static int sysval = 0;
                    124:
                    125: #ifdef DEBUG
                    126:        printf("argc = %d\n", argc);
                    127:        for (n = 0; n < argc; n++)
                    128:                printf("argv[%d] = %s\n", n, argv[n]);
1.42      espie     129:        fflush(stdout);
1.1       deraadt   130: #endif
1.22      espie     131:
1.1       deraadt   132:  /*
                    133:   * if argc == 3 and argv[2] is null, then we
                    134:   * have macro-or-builtin() type call. We adjust
                    135:   * argc to avoid further checking..
                    136:   */
1.58      espie     137:  /* we keep the initial value for those built-ins that differentiate
                    138:   * between builtin() and builtin.
                    139:   */
1.39      espie     140:        ac = argc;
                    141:
1.64    ! espie     142:        if (argc == 3 && !*(argv[2]) && !mimic_gnu)
1.1       deraadt   143:                argc--;
                    144:
1.22      espie     145:        switch (td & TYPEMASK) {
1.1       deraadt   146:
                    147:        case DEFITYPE:
                    148:                if (argc > 2)
                    149:                        dodefine(argv[2], (argc > 3) ? argv[3] : null);
                    150:                break;
                    151:
                    152:        case PUSDTYPE:
                    153:                if (argc > 2)
                    154:                        dopushdef(argv[2], (argc > 3) ? argv[3] : null);
                    155:                break;
                    156:
                    157:        case DUMPTYPE:
                    158:                dodump(argv, argc);
                    159:                break;
                    160:
1.37      espie     161:        case TRACEONTYPE:
                    162:                dotrace(argv, argc, 1);
                    163:                break;
                    164:
                    165:        case TRACEOFFTYPE:
                    166:                dotrace(argv, argc, 0);
                    167:                break;
                    168:
1.1       deraadt   169:        case EXPRTYPE:
                    170:        /*
                    171:         * doexpr - evaluate arithmetic
                    172:         * expression
                    173:         */
1.56      espie     174:        {
                    175:                int base = 10;
                    176:                int maxdigits = 0;
                    177:                const char *errstr;
                    178:
                    179:                if (argc > 3) {
                    180:                        base = strtonum(argv[3], 2, 36, &errstr);
                    181:                        if (errstr) {
1.60      espie     182:                                m4errx(1, "expr: base %s invalid.", argv[3]);
1.56      espie     183:                        }
                    184:                }
                    185:                if (argc > 4) {
                    186:                        maxdigits = strtonum(argv[4], 0, INT_MAX, &errstr);
                    187:                        if (errstr) {
1.60      espie     188:                                m4errx(1, "expr: maxdigits %s invalid.", argv[4]);
1.56      espie     189:                        }
                    190:                }
1.1       deraadt   191:                if (argc > 2)
1.56      espie     192:                        pbnumbase(expr(argv[2]), base, maxdigits);
1.1       deraadt   193:                break;
1.56      espie     194:        }
1.1       deraadt   195:
                    196:        case IFELTYPE:
                    197:                if (argc > 4)
                    198:                        doifelse(argv, argc);
                    199:                break;
                    200:
                    201:        case IFDFTYPE:
                    202:        /*
                    203:         * doifdef - select one of two
                    204:         * alternatives based on the existence of
                    205:         * another definition
                    206:         */
                    207:                if (argc > 3) {
1.49      espie     208:                        if (lookup_macro_definition(argv[2]) != NULL)
1.1       deraadt   209:                                pbstr(argv[3]);
                    210:                        else if (argc > 4)
                    211:                                pbstr(argv[4]);
                    212:                }
                    213:                break;
                    214:
                    215:        case LENGTYPE:
                    216:        /*
                    217:         * dolen - find the length of the
                    218:         * argument
                    219:         */
1.18      espie     220:                pbnum((argc > 2) ? strlen(argv[2]) : 0);
1.1       deraadt   221:                break;
                    222:
                    223:        case INCRTYPE:
                    224:        /*
                    225:         * doincr - increment the value of the
                    226:         * argument
                    227:         */
                    228:                if (argc > 2)
                    229:                        pbnum(atoi(argv[2]) + 1);
                    230:                break;
                    231:
                    232:        case DECRTYPE:
                    233:        /*
                    234:         * dodecr - decrement the value of the
                    235:         * argument
                    236:         */
                    237:                if (argc > 2)
                    238:                        pbnum(atoi(argv[2]) - 1);
                    239:                break;
                    240:
                    241:        case SYSCTYPE:
                    242:        /*
                    243:         * dosys - execute system command
                    244:         */
1.54      robert    245:                if (argc > 2) {
                    246:                        fflush(stdout);
1.1       deraadt   247:                        sysval = system(argv[2]);
1.54      robert    248:                }
1.1       deraadt   249:                break;
                    250:
                    251:        case SYSVTYPE:
                    252:        /*
                    253:         * dosysval - return value of the last
                    254:         * system call.
                    255:         *
                    256:         */
                    257:                pbnum(sysval);
                    258:                break;
                    259:
1.27      espie     260:        case ESYSCMDTYPE:
                    261:                if (argc > 2)
                    262:                        doesyscmd(argv[2]);
                    263:                break;
1.1       deraadt   264:        case INCLTYPE:
                    265:                if (argc > 2)
                    266:                        if (!doincl(argv[2]))
1.24      espie     267:                                err(1, "%s at line %lu: include(%s)",
                    268:                                    CURRENT_NAME, CURRENT_LINE, argv[2]);
1.1       deraadt   269:                break;
                    270:
                    271:        case SINCTYPE:
                    272:                if (argc > 2)
                    273:                        (void) doincl(argv[2]);
                    274:                break;
                    275: #ifdef EXTENDED
                    276:        case PASTTYPE:
                    277:                if (argc > 2)
                    278:                        if (!dopaste(argv[2]))
1.24      espie     279:                                err(1, "%s at line %lu: paste(%s)",
                    280:                                    CURRENT_NAME, CURRENT_LINE, argv[2]);
1.1       deraadt   281:                break;
                    282:
                    283:        case SPASTYPE:
                    284:                if (argc > 2)
                    285:                        (void) dopaste(argv[2]);
1.61      espie     286:                break;
                    287:        case FORMATTYPE:
1.62      espie     288:                doformat(argv, argc);
1.1       deraadt   289:                break;
                    290: #endif
                    291:        case CHNQTYPE:
1.58      espie     292:                dochq(argv, ac);
1.1       deraadt   293:                break;
                    294:
                    295:        case CHNCTYPE:
1.58      espie     296:                dochc(argv, argc);
1.1       deraadt   297:                break;
                    298:
                    299:        case SUBSTYPE:
                    300:        /*
                    301:         * dosub - select substring
                    302:         *
                    303:         */
                    304:                if (argc > 3)
                    305:                        dosub(argv, argc);
                    306:                break;
                    307:
                    308:        case SHIFTYPE:
                    309:        /*
                    310:         * doshift - push back all arguments
                    311:         * except the first one (i.e. skip
                    312:         * argv[2])
                    313:         */
                    314:                if (argc > 3) {
                    315:                        for (n = argc - 1; n > 3; n--) {
1.10      deraadt   316:                                pbstr(rquote);
1.1       deraadt   317:                                pbstr(argv[n]);
1.10      deraadt   318:                                pbstr(lquote);
1.58      espie     319:                                pushback(COMMA);
1.1       deraadt   320:                        }
1.10      deraadt   321:                        pbstr(rquote);
1.1       deraadt   322:                        pbstr(argv[3]);
1.10      deraadt   323:                        pbstr(lquote);
1.1       deraadt   324:                }
                    325:                break;
                    326:
                    327:        case DIVRTYPE:
                    328:                if (argc > 2 && (n = atoi(argv[2])) != 0)
                    329:                        dodiv(n);
                    330:                else {
                    331:                        active = stdout;
                    332:                        oindex = 0;
                    333:                }
                    334:                break;
                    335:
                    336:        case UNDVTYPE:
                    337:                doundiv(argv, argc);
                    338:                break;
                    339:
                    340:        case DIVNTYPE:
                    341:        /*
                    342:         * dodivnum - return the number of
                    343:         * current output diversion
                    344:         */
                    345:                pbnum(oindex);
                    346:                break;
                    347:
                    348:        case UNDFTYPE:
                    349:        /*
                    350:         * doundefine - undefine a previously
                    351:         * defined macro(s) or m4 keyword(s).
                    352:         */
                    353:                if (argc > 2)
                    354:                        for (n = 2; n < argc; n++)
1.49      espie     355:                                macro_undefine(argv[n]);
1.1       deraadt   356:                break;
                    357:
                    358:        case POPDTYPE:
                    359:        /*
                    360:         * dopopdef - remove the topmost
                    361:         * definitions of macro(s) or m4
                    362:         * keyword(s).
                    363:         */
                    364:                if (argc > 2)
                    365:                        for (n = 2; n < argc; n++)
1.49      espie     366:                                macro_popdef(argv[n]);
1.1       deraadt   367:                break;
                    368:
                    369:        case MKTMTYPE:
                    370:        /*
                    371:         * dotemp - create a temporary file
                    372:         */
1.16      espie     373:                if (argc > 2) {
                    374:                        int fd;
1.20      espie     375:                        char *temp;
                    376:
                    377:                        temp = xstrdup(argv[2]);
1.16      espie     378:
1.20      espie     379:                        fd = mkstemp(temp);
1.16      espie     380:                        if (fd == -1)
1.24      espie     381:                                err(1,
                    382:            "%s at line %lu: couldn't make temp file %s",
                    383:            CURRENT_NAME, CURRENT_LINE, argv[2]);
1.16      espie     384:                        close(fd);
1.20      espie     385:                        pbstr(temp);
                    386:                        free(temp);
1.16      espie     387:                }
1.1       deraadt   388:                break;
                    389:
                    390:        case TRNLTYPE:
                    391:        /*
                    392:         * dotranslit - replace all characters in
                    393:         * the source string that appears in the
                    394:         * "from" string with the corresponding
                    395:         * characters in the "to" string.
                    396:         */
                    397:                if (argc > 3) {
1.42      espie     398:                        char *temp;
                    399:
1.51      espie     400:                        temp = xalloc(strlen(argv[2])+1, NULL);
1.1       deraadt   401:                        if (argc > 4)
                    402:                                map(temp, argv[2], argv[3], argv[4]);
                    403:                        else
                    404:                                map(temp, argv[2], argv[3], null);
                    405:                        pbstr(temp);
1.42      espie     406:                        free(temp);
1.8       deraadt   407:                } else if (argc > 2)
1.1       deraadt   408:                        pbstr(argv[2]);
                    409:                break;
                    410:
                    411:        case INDXTYPE:
                    412:        /*
                    413:         * doindex - find the index of the second
                    414:         * argument string in the first argument
                    415:         * string. -1 if not present.
                    416:         */
                    417:                pbnum((argc > 3) ? indx(argv[2], argv[3]) : -1);
                    418:                break;
                    419:
                    420:        case ERRPTYPE:
                    421:        /*
                    422:         * doerrp - print the arguments to stderr
                    423:         * file
                    424:         */
                    425:                if (argc > 2) {
                    426:                        for (n = 2; n < argc; n++)
                    427:                                fprintf(stderr, "%s ", argv[n]);
                    428:                        fprintf(stderr, "\n");
                    429:                }
                    430:                break;
                    431:
                    432:        case DNLNTYPE:
                    433:        /*
                    434:         * dodnl - eat-up-to and including
                    435:         * newline
                    436:         */
                    437:                while ((c = gpbc()) != '\n' && c != EOF)
                    438:                        ;
                    439:                break;
                    440:
                    441:        case M4WRTYPE:
                    442:        /*
                    443:         * dom4wrap - set up for
                    444:         * wrap-up/wind-down activity
                    445:         */
1.55      espie     446:                if (argc > 2)
                    447:                        dom4wrap(argv[2]);
1.1       deraadt   448:                break;
                    449:
                    450:        case EXITTYPE:
                    451:        /*
                    452:         * doexit - immediate exit from m4.
                    453:         */
                    454:                killdiv();
                    455:                exit((argc > 2) ? atoi(argv[2]) : 0);
                    456:                break;
                    457:
                    458:        case DEFNTYPE:
                    459:                if (argc > 2)
                    460:                        for (n = 2; n < argc; n++)
                    461:                                dodefn(argv[n]);
                    462:                break;
                    463:
1.25      espie     464:        case INDIRTYPE: /* Indirect call */
                    465:                if (argc > 2)
                    466:                        doindir(argv, argc);
                    467:                break;
                    468:
                    469:        case BUILTINTYPE: /* Builtins only */
                    470:                if (argc > 2)
                    471:                        dobuiltin(argv, argc);
                    472:                break;
                    473:
                    474:        case PATSTYPE:
                    475:                if (argc > 2)
                    476:                        dopatsubst(argv, argc);
                    477:                break;
                    478:        case REGEXPTYPE:
                    479:                if (argc > 2)
                    480:                        doregexp(argv, argc);
                    481:                break;
                    482:        case LINETYPE:
                    483:                doprintlineno(infile+ilevel);
                    484:                break;
                    485:        case FILENAMETYPE:
                    486:                doprintfilename(infile+ilevel);
                    487:                break;
1.23      espie     488:        case SELFTYPE:
                    489:                pbstr(rquote);
                    490:                pbstr(argv[1]);
                    491:                pbstr(lquote);
                    492:                break;
1.1       deraadt   493:        default:
1.60      espie     494:                m4errx(1, "eval: major botch.");
1.1       deraadt   495:                break;
                    496:        }
                    497: }
                    498:
                    499: /*
1.30      espie     500:  * expand_macro - user-defined macro expansion
1.1       deraadt   501:  */
                    502: void
1.44      espie     503: expand_macro(const char *argv[], int argc)
1.1       deraadt   504: {
1.20      espie     505:        const char *t;
                    506:        const char *p;
1.17      espie     507:        int n;
                    508:        int argno;
1.1       deraadt   509:
                    510:        t = argv[0];                   /* defn string as a whole */
                    511:        p = t;
                    512:        while (*p)
                    513:                p++;
                    514:        p--;                           /* last character of defn */
                    515:        while (p > t) {
                    516:                if (*(p - 1) != ARGFLAG)
1.58      espie     517:                        PUSHBACK(*p);
1.1       deraadt   518:                else {
                    519:                        switch (*p) {
                    520:
                    521:                        case '#':
                    522:                                pbnum(argc - 2);
                    523:                                break;
                    524:                        case '0':
                    525:                        case '1':
                    526:                        case '2':
                    527:                        case '3':
                    528:                        case '4':
                    529:                        case '5':
                    530:                        case '6':
                    531:                        case '7':
                    532:                        case '8':
                    533:                        case '9':
                    534:                                if ((argno = *p - '0') < argc - 1)
                    535:                                        pbstr(argv[argno + 1]);
                    536:                                break;
                    537:                        case '*':
1.41      espie     538:                                if (argc > 2) {
                    539:                                        for (n = argc - 1; n > 2; n--) {
                    540:                                                pbstr(argv[n]);
1.58      espie     541:                                                pushback(COMMA);
1.41      espie     542:                                        }
                    543:                                        pbstr(argv[2]);
                    544:                                }
1.1       deraadt   545:                                break;
1.7       millert   546:                         case '@':
1.41      espie     547:                                if (argc > 2) {
                    548:                                        for (n = argc - 1; n > 2; n--) {
                    549:                                                pbstr(rquote);
                    550:                                                pbstr(argv[n]);
                    551:                                                pbstr(lquote);
1.58      espie     552:                                                pushback(COMMA);
1.41      espie     553:                                        }
                    554:                                        pbstr(rquote);
                    555:                                        pbstr(argv[2]);
                    556:                                        pbstr(lquote);
                    557:                                }
1.7       millert   558:                                 break;
1.1       deraadt   559:                        default:
1.58      espie     560:                                PUSHBACK(*p);
                    561:                                PUSHBACK('$');
1.1       deraadt   562:                                break;
                    563:                        }
                    564:                        p--;
                    565:                }
                    566:                p--;
                    567:        }
                    568:        if (p == t)                    /* do last character */
1.58      espie     569:                PUSHBACK(*p);
1.1       deraadt   570: }
                    571:
1.48      espie     572:
1.1       deraadt   573: /*
1.48      espie     574:  * dodefine - install definition in the table
                    575:  */
                    576: void
                    577: dodefine(const char *name, const char *defn)
                    578: {
1.64    ! espie     579:        if (!*name && !mimic_gnu)
1.60      espie     580:                m4errx(1, "null definition.");
1.64    ! espie     581:        else
        !           582:                macro_define(name, defn);
1.1       deraadt   583: }
                    584:
                    585: /*
                    586:  * dodefn - push back a quoted definition of
                    587:  *      the given name.
                    588:  */
1.20      espie     589: static void
1.44      espie     590: dodefn(const char *name)
1.1       deraadt   591: {
1.49      espie     592:        struct macro_definition *p;
1.1       deraadt   593:
1.49      espie     594:        if ((p = lookup_macro_definition(name)) != NULL) {
1.48      espie     595:                if ((p->type & TYPEMASK) == MACRTYPE) {
1.40      espie     596:                        pbstr(rquote);
                    597:                        pbstr(p->defn);
                    598:                        pbstr(lquote);
1.48      espie     599:                } else {
                    600:                        pbstr(p->defn);
1.31      espie     601:                        pbstr(BUILTIN_MARKER);
                    602:                }
1.1       deraadt   603:        }
                    604: }
                    605:
                    606: /*
                    607:  * dopushdef - install a definition in the hash table
                    608:  *      without removing a previous definition. Since
                    609:  *      each new entry is entered in *front* of the
                    610:  *      hash bucket, it hides a previous definition from
                    611:  *      lookup.
                    612:  */
1.20      espie     613: static void
1.44      espie     614: dopushdef(const char *name, const char *defn)
1.1       deraadt   615: {
1.64    ! espie     616:        if (!*name && !mimic_gnu)
1.60      espie     617:                m4errx(1, "null definition.");
1.64    ! espie     618:        else
        !           619:                macro_pushdef(name, defn);
1.1       deraadt   620: }
                    621:
                    622: /*
1.32      espie     623:  * dump_one_def - dump the specified definition.
                    624:  */
                    625: static void
1.49      espie     626: dump_one_def(const char *name, struct macro_definition *p)
1.32      espie     627: {
1.53      espie     628:        if (!traceout)
                    629:                traceout = stderr;
1.33      espie     630:        if (mimic_gnu) {
                    631:                if ((p->type & TYPEMASK) == MACRTYPE)
1.49      espie     632:                        fprintf(traceout, "%s:\t%s\n", name, p->defn);
1.33      espie     633:                else {
1.49      espie     634:                        fprintf(traceout, "%s:\t<%s>\n", name, p->defn);
1.33      espie     635:                }
                    636:        } else
1.49      espie     637:                fprintf(traceout, "`%s'\t`%s'\n", name, p->defn);
1.32      espie     638: }
                    639:
                    640: /*
1.1       deraadt   641:  * dodumpdef - dump the specified definitions in the hash
                    642:  *      table to stderr. If nothing is specified, the entire
                    643:  *      hash table is dumped.
                    644:  */
1.20      espie     645: static void
1.44      espie     646: dodump(const char *argv[], int argc)
1.1       deraadt   647: {
1.17      espie     648:        int n;
1.49      espie     649:        struct macro_definition *p;
1.1       deraadt   650:
                    651:        if (argc > 2) {
                    652:                for (n = 2; n < argc; n++)
1.49      espie     653:                        if ((p = lookup_macro_definition(argv[n])) != NULL)
                    654:                                dump_one_def(argv[n], p);
                    655:        } else
                    656:                macro_for_all(dump_one_def);
1.37      espie     657: }
                    658:
                    659: /*
                    660:  * dotrace - mark some macros as traced/untraced depending upon on.
                    661:  */
                    662: static void
1.44      espie     663: dotrace(const char *argv[], int argc, int on)
1.37      espie     664: {
                    665:        int n;
                    666:
                    667:        if (argc > 2) {
                    668:                for (n = 2; n < argc; n++)
                    669:                        mark_traced(argv[n], on);
                    670:        } else
                    671:                mark_traced(NULL, on);
1.1       deraadt   672: }
                    673:
                    674: /*
                    675:  * doifelse - select one of two alternatives - loop.
                    676:  */
1.20      espie     677: static void
1.44      espie     678: doifelse(const char *argv[], int argc)
1.1       deraadt   679: {
                    680:        cycle {
                    681:                if (STREQ(argv[2], argv[3]))
                    682:                        pbstr(argv[4]);
                    683:                else if (argc == 6)
                    684:                        pbstr(argv[5]);
                    685:                else if (argc > 6) {
                    686:                        argv += 3;
                    687:                        argc -= 3;
                    688:                        continue;
                    689:                }
                    690:                break;
                    691:        }
                    692: }
                    693:
                    694: /*
                    695:  * doinclude - include a given file.
                    696:  */
1.20      espie     697: static int
1.44      espie     698: doincl(const char *ifile)
1.1       deraadt   699: {
                    700:        if (ilevel + 1 == MAXINP)
1.60      espie     701:                m4errx(1, "too many include files.");
1.24      espie     702:        if (fopen_trypath(infile+ilevel+1, ifile) != NULL) {
1.1       deraadt   703:                ilevel++;
                    704:                bbase[ilevel] = bufbase = bp;
                    705:                return (1);
1.8       deraadt   706:        } else
1.1       deraadt   707:                return (0);
                    708: }
                    709:
                    710: #ifdef EXTENDED
                    711: /*
                    712:  * dopaste - include a given file without any
                    713:  *           macro processing.
                    714:  */
1.20      espie     715: static int
1.44      espie     716: dopaste(const char *pfile)
1.1       deraadt   717: {
                    718:        FILE *pf;
1.17      espie     719:        int c;
1.1       deraadt   720:
                    721:        if ((pf = fopen(pfile, "r")) != NULL) {
1.45      espie     722:                if (synch_lines)
                    723:                    fprintf(active, "#line 1 \"%s\"\n", pfile);
1.1       deraadt   724:                while ((c = getc(pf)) != EOF)
                    725:                        putc(c, active);
                    726:                (void) fclose(pf);
1.45      espie     727:                emit_synchline();
1.1       deraadt   728:                return (1);
1.8       deraadt   729:        } else
1.1       deraadt   730:                return (0);
                    731: }
                    732: #endif
                    733:
1.58      espie     734: /*
                    735:  * dochq - change quote characters
                    736:  */
1.39      espie     737: static void
1.58      espie     738: dochq(const char *argv[], int ac)
1.39      espie     739: {
                    740:        if (ac == 2) {
1.58      espie     741:                lquote[0] = LQUOTE; lquote[1] = EOS;
                    742:                rquote[0] = RQUOTE; rquote[1] = EOS;
1.39      espie     743:        } else {
                    744:                strlcpy(lquote, argv[2], sizeof(lquote));
1.58      espie     745:                if (ac > 3) {
1.39      espie     746:                        strlcpy(rquote, argv[3], sizeof(rquote));
1.58      espie     747:                } else {
                    748:                        rquote[0] = ECOMMT; rquote[1] = EOS;
                    749:                }
1.39      espie     750:        }
                    751: }
                    752:
1.1       deraadt   753: /*
1.58      espie     754:  * dochc - change comment characters
1.1       deraadt   755:  */
1.20      espie     756: static void
1.58      espie     757: dochc(const char *argv[], int argc)
1.1       deraadt   758: {
1.58      espie     759: /* XXX Note that there is no difference between no argument and a single
                    760:  * empty argument.
                    761:  */
                    762:        if (argc == 2) {
1.39      espie     763:                scommt[0] = EOS;
                    764:                ecommt[0] = EOS;
                    765:        } else {
1.58      espie     766:                strlcpy(scommt, argv[2], sizeof(scommt));
                    767:                if (argc == 3) {
                    768:                        ecommt[0] = ECOMMT; ecommt[1] = EOS;
                    769:                } else {
1.39      espie     770:                        strlcpy(ecommt, argv[3], sizeof(ecommt));
1.1       deraadt   771:                }
                    772:        }
1.55      espie     773: }
                    774:
                    775: /*
                    776:  * dom4wrap - expand text at EOF
                    777:  */
                    778: static void
                    779: dom4wrap(const char *text)
                    780: {
                    781:        if (wrapindex >= maxwraps) {
                    782:                if (maxwraps == 0)
                    783:                        maxwraps = 16;
                    784:                else
                    785:                        maxwraps *= 2;
                    786:                m4wraps = xrealloc(m4wraps, maxwraps * sizeof(*m4wraps),
                    787:                   "too many m4wraps");
                    788:        }
                    789:        m4wraps[wrapindex++] = xstrdup(text);
1.1       deraadt   790: }
                    791:
                    792: /*
                    793:  * dodivert - divert the output to a temporary file
                    794:  */
1.20      espie     795: static void
1.44      espie     796: dodiv(int n)
1.1       deraadt   797: {
1.6       millert   798:        int fd;
                    799:
1.8       deraadt   800:        oindex = n;
1.28      espie     801:        if (n >= maxout) {
                    802:                if (mimic_gnu)
                    803:                        resizedivs(n + 10);
                    804:                else
                    805:                        n = 0;          /* bitbucket */
                    806:        }
                    807:
                    808:        if (n < 0)
1.1       deraadt   809:                n = 0;                 /* bitbucket */
                    810:        if (outfile[n] == NULL) {
1.13      espie     811:                char fname[] = _PATH_DIVNAME;
                    812:
                    813:                if ((fd = mkstemp(fname)) < 0 ||
                    814:                        (outfile[n] = fdopen(fd, "w+")) == NULL)
                    815:                                err(1, "%s: cannot divert", fname);
                    816:                if (unlink(fname) == -1)
                    817:                        err(1, "%s: cannot unlink", fname);
1.1       deraadt   818:        }
                    819:        active = outfile[n];
                    820: }
                    821:
                    822: /*
                    823:  * doundivert - undivert a specified output, or all
                    824:  *              other outputs, in numerical order.
                    825:  */
1.20      espie     826: static void
1.44      espie     827: doundiv(const char *argv[], int argc)
1.1       deraadt   828: {
1.17      espie     829:        int ind;
                    830:        int n;
1.1       deraadt   831:
                    832:        if (argc > 2) {
                    833:                for (ind = 2; ind < argc; ind++) {
1.56      espie     834:                        const char *errstr;
                    835:                        n = strtonum(argv[ind], 1, INT_MAX, &errstr);
                    836:                        if (errstr) {
                    837:                                if (errno == EINVAL && mimic_gnu)
                    838:                                        getdivfile(argv[ind]);
                    839:                        } else {
                    840:                                if (n < maxout && outfile[n] != NULL)
                    841:                                        getdiv(n);
                    842:                        }
1.1       deraadt   843:                }
                    844:        }
                    845:        else
1.28      espie     846:                for (n = 1; n < maxout; n++)
1.1       deraadt   847:                        if (outfile[n] != NULL)
                    848:                                getdiv(n);
                    849: }
                    850:
                    851: /*
                    852:  * dosub - select substring
                    853:  */
1.20      espie     854: static void
1.44      espie     855: dosub(const char *argv[], int argc)
1.1       deraadt   856: {
1.20      espie     857:        const char *ap, *fc, *k;
1.17      espie     858:        int nc;
1.1       deraadt   859:
1.29      espie     860:        ap = argv[2];                  /* target string */
1.1       deraadt   861: #ifdef EXPR
1.29      espie     862:        fc = ap + expr(argv[3]);       /* first char */
1.1       deraadt   863: #else
1.29      espie     864:        fc = ap + atoi(argv[3]);       /* first char */
1.1       deraadt   865: #endif
1.29      espie     866:        nc = strlen(fc);
                    867:        if (argc >= 5)
1.1       deraadt   868: #ifdef EXPR
1.29      espie     869:                nc = min(nc, expr(argv[4]));
1.1       deraadt   870: #else
1.29      espie     871:                nc = min(nc, atoi(argv[4]));
1.1       deraadt   872: #endif
                    873:        if (fc >= ap && fc < ap + strlen(ap))
1.29      espie     874:                for (k = fc + nc - 1; k >= fc; k--)
1.58      espie     875:                        pushback(*k);
1.1       deraadt   876: }
                    877:
                    878: /*
                    879:  * map:
                    880:  * map every character of s1 that is specified in from
                    881:  * into s3 and replace in s. (source s1 remains untouched)
                    882:  *
                    883:  * This is a standard implementation of map(s,from,to) function of ICON
                    884:  * language. Within mapvec, we replace every character of "from" with
                    885:  * the corresponding character in "to". If "to" is shorter than "from",
                    886:  * than the corresponding entries are null, which means that those
                    887:  * characters dissapear altogether. Furthermore, imagine
                    888:  * map(dest, "sourcestring", "srtin", "rn..*") type call. In this case,
                    889:  * `s' maps to `r', `r' maps to `n' and `n' maps to `*'. Thus, `s'
                    890:  * ultimately maps to `*'. In order to achieve this effect in an efficient
                    891:  * manner (i.e. without multiple passes over the destination string), we
                    892:  * loop over mapvec, starting with the initial source character. if the
                    893:  * character value (dch) in this location is different than the source
                    894:  * character (sch), sch becomes dch, once again to index into mapvec, until
                    895:  * the character value stabilizes (i.e. sch = dch, in other words
                    896:  * mapvec[n] == n). Even if the entry in the mapvec is null for an ordinary
                    897:  * character, it will stabilize, since mapvec[0] == 0 at all times. At the
                    898:  * end, we restore mapvec* back to normal where mapvec[n] == n for
                    899:  * 0 <= n <= 127. This strategy, along with the restoration of mapvec, is
                    900:  * about 5 times faster than any algorithm that makes multiple passes over
                    901:  * destination string.
                    902:  */
1.20      espie     903: static void
1.44      espie     904: map(char *dest, const char *src, const char *from, const char *to)
1.1       deraadt   905: {
1.20      espie     906:        const char *tmp;
1.19      espie     907:        unsigned char sch, dch;
1.26      espie     908:        static char frombis[257];
                    909:        static char tobis[257];
1.19      espie     910:        static unsigned char mapvec[256] = {
                    911:            0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18,
                    912:            19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35,
                    913:            36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52,
                    914:            53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69,
                    915:            70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86,
                    916:            87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102,
                    917:            103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115,
                    918:            116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128,
                    919:            129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141,
                    920:            142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154,
                    921:            155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167,
                    922:            168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180,
                    923:            181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193,
                    924:            194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206,
                    925:            207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219,
                    926:            220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232,
                    927:            233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245,
                    928:            246, 247, 248, 249, 250, 251, 252, 253, 254, 255
1.1       deraadt   929:        };
                    930:
                    931:        if (*src) {
1.26      espie     932:                if (mimic_gnu) {
                    933:                        /*
                    934:                         * expand character ranges on the fly
                    935:                         */
                    936:                        from = handledash(frombis, frombis + 256, from);
                    937:                        to = handledash(tobis, tobis + 256, to);
                    938:                }
1.1       deraadt   939:                tmp = from;
                    940:        /*
                    941:         * create a mapping between "from" and
                    942:         * "to"
                    943:         */
                    944:                while (*from)
1.19      espie     945:                        mapvec[(unsigned char)(*from++)] = (*to) ?
                    946:                                (unsigned char)(*to++) : 0;
1.1       deraadt   947:
                    948:                while (*src) {
1.19      espie     949:                        sch = (unsigned char)(*src++);
1.1       deraadt   950:                        dch = mapvec[sch];
                    951:                        while (dch != sch) {
                    952:                                sch = dch;
                    953:                                dch = mapvec[sch];
                    954:                        }
1.19      espie     955:                        if ((*dest = (char)dch))
1.1       deraadt   956:                                dest++;
                    957:                }
                    958:        /*
                    959:         * restore all the changed characters
                    960:         */
                    961:                while (*tmp) {
1.19      espie     962:                        mapvec[(unsigned char)(*tmp)] = (unsigned char)(*tmp);
1.1       deraadt   963:                        tmp++;
                    964:                }
                    965:        }
1.19      espie     966:        *dest = '\0';
1.1       deraadt   967: }
1.26      espie     968:
                    969:
                    970: /*
                    971:  * handledash:
                    972:  *  use buffer to copy the src string, expanding character ranges
                    973:  * on the way.
                    974:  */
                    975: static const char *
1.44      espie     976: handledash(char *buffer, char *end, const char *src)
1.26      espie     977: {
                    978:        char *p;
                    979:
                    980:        p = buffer;
                    981:        while(*src) {
                    982:                if (src[1] == '-' && src[2]) {
                    983:                        unsigned char i;
1.56      espie     984:                        if ((unsigned char)src[0] <= (unsigned char)src[2]) {
                    985:                                for (i = (unsigned char)src[0];
                    986:                                    i <= (unsigned char)src[2]; i++) {
                    987:                                        *p++ = i;
                    988:                                        if (p == end) {
                    989:                                                *p = '\0';
                    990:                                                return buffer;
                    991:                                        }
                    992:                                }
                    993:                        } else {
                    994:                                for (i = (unsigned char)src[0];
                    995:                                    i >= (unsigned char)src[2]; i--) {
                    996:                                        *p++ = i;
                    997:                                        if (p == end) {
                    998:                                                *p = '\0';
                    999:                                                return buffer;
                   1000:                                        }
1.26      espie    1001:                                }
                   1002:                        }
                   1003:                        src += 3;
                   1004:                } else
                   1005:                        *p++ = *src++;
                   1006:                if (p == end)
                   1007:                        break;
                   1008:        }
                   1009:        *p = '\0';
                   1010:        return buffer;
                   1011: }