home *** CD-ROM | disk | FTP | other *** search
/ Big Green CD 8 / BGCD_8_Dev.iso / NEXTSTEP / UNIX / Shells / zsh-3.0.5-MIHS / src / Src / subst.c < prev    next >
Encoding:
C/C++ Source or Header  |  1997-09-25  |  37.1 KB  |  1,774 lines
  1. /*
  2.  * $Id: subst.c,v 2.52 1996/10/15 20:16:35 hzoli Exp $
  3.  *
  4.  * subst.c - various substitutions
  5.  *
  6.  * This file is part of zsh, the Z shell.
  7.  *
  8.  * Copyright (c) 1992-1996 Paul Falstad
  9.  * All rights reserved.
  10.  *
  11.  * Permission is hereby granted, without written agreement and without
  12.  * license or royalty fees, to use, copy, modify, and distribute this
  13.  * software and to distribute modified versions of this software for any
  14.  * purpose, provided that the above copyright notice and the following
  15.  * two paragraphs appear in all copies of this software.
  16.  *
  17.  * In no event shall Paul Falstad or the Zsh Development Group be liable
  18.  * to any party for direct, indirect, special, incidental, or consequential
  19.  * damages arising out of the use of this software and its documentation,
  20.  * even if Paul Falstad and the Zsh Development Group have been advised of
  21.  * the possibility of such damage.
  22.  *
  23.  * Paul Falstad and the Zsh Development Group specifically disclaim any
  24.  * warranties, including, but not limited to, the implied warranties of
  25.  * merchantability and fitness for a particular purpose.  The software
  26.  * provided hereunder is on an "as is" basis, and Paul Falstad and the
  27.  * Zsh Development Group have no obligation to provide maintenance,
  28.  * support, updates, enhancements, or modifications.
  29.  *
  30.  */
  31.  
  32. #include "zsh.h"
  33.  
  34. /* Do substitutions before fork. These are:
  35.  *  - Process substitution: <(...), >(...), =(...)
  36.  *  - Parameter substitution
  37.  *  - Command substitution
  38.  * Followed by
  39.  *  - Quote removal
  40.  *  - Brace expansion
  41.  *  - Tilde and equals substitution
  42.  *
  43.  * Bits 0 and 1 of flags are used in filesub.
  44.  * bit 0 is set when we are doing MAGIC_EQUALSUBST or normal
  45.  *     assignment but not a typeset.
  46.  * bit 1 is set on a real assignment (both typeset and normal).
  47.  * bit 2 is a flag to paramsubst (single word sub)
  48.  */
  49.  
  50. /**/
  51. void
  52. prefork(LinkList list, int flags)
  53. {
  54.     LinkNode node;
  55.  
  56.     MUSTUSEHEAP("prefork");
  57.     for (node = firstnode(list); node; incnode(node)) {
  58.     char *str, *str3;
  59.  
  60.     str = str3 = (char *)getdata(node);
  61.     if ((*str == Inang || *str == Outang || *str == Equals) &&
  62.         str[1] == Inpar) {
  63.         if (*str == Inang || *str == Outang)
  64.         setdata(node, (void *) getproc(str));    /* <(...) or >(...) */
  65.         else
  66.         setdata(node, (void *) getoutputfile(str));    /* =(...) */
  67.         if (!getdata(node))
  68.         return;
  69.     } else {
  70.         if (isset(SHFILEEXPANSION))
  71.         filesub((char **)getaddrdata(node), flags & 3);
  72.         if (!(node = stringsubst(list, node, flags & 4)))
  73.         return;
  74.     }
  75.     }
  76.     for (node = firstnode(list); node; incnode(node)) {
  77.     if (*(char *)getdata(node)) {
  78.         remnulargs(getdata(node));
  79.         if (unset(IGNOREBRACES) && !(flags & 4))
  80.         while (hasbraces(getdata(node)))
  81.             xpandbraces(list, &node);
  82.         if (unset(SHFILEEXPANSION))
  83.         filesub((char **)getaddrdata(node), flags & 3);
  84.     } else if (!(flags & 4))
  85.         uremnode(list, node);
  86.     if (errflag)
  87.         return;
  88.     }
  89. }
  90.  
  91. /**/
  92. LinkNode
  93. stringsubst(LinkList list, LinkNode node, int ssub)
  94. {
  95.     int qt;
  96.     char *str3 = (char *)getdata(node);
  97.     char *str  = str3;
  98.  
  99.     while (!errflag && *str) {
  100.     if ((qt = *str == Qstring) || *str == String)
  101.         if (str[1] == Inpar) {
  102.         str++;
  103.         goto comsub;
  104.         } else if (str[1] == Inbrack) {
  105.         /* $[...] */
  106.         char *str2 = str;
  107.         str2++;
  108.         if (skipparens(Inbrack, Outbrack, &str2)) {
  109.             zerr("closing bracket missing", NULL, 0);
  110.             return NULL;
  111.         }
  112.         str2[-1] = *str = '\0';
  113.         str = arithsubst(str + 2, &str3, str2);
  114.         setdata(node, (void *) str3);
  115.         continue;
  116.         } else {
  117.         node = paramsubst(list, node, &str, qt, ssub);
  118.         if (errflag || !node)
  119.             return NULL;
  120.         str3 = (char *)getdata(node);
  121.         continue;
  122.         }
  123.     else if ((qt = *str == Qtick) || *str == Tick)
  124.       comsub: {
  125.         LinkList pl;
  126.         char *s, *str2 = str;
  127.         char endchar;
  128.         int l1, l2;
  129.  
  130.         if (*str == Inpar) {
  131.         endchar = Outpar;
  132.         str[-1] = '\0';
  133.         if (skipparens(Inpar, Outpar, &str))
  134.             DPUTS(1, "Oops. parse error in command substitution");
  135.         str--;
  136.         } else {
  137.         endchar = *str;
  138.         *str = '\0';
  139.  
  140.         while (*++str != endchar)
  141.             DPUTS(!*str, "Oops. parse error in command substitution");
  142.         }
  143.         *str++ = '\0';
  144.         if (endchar == Outpar && str2[1] == '(' && str[-2] == ')') {
  145.         /* Math substitution of the form $((...)) */
  146.         str = arithsubst(str2 + 1, &str3, str);
  147.         setdata(node, (void *) str3);
  148.         continue;
  149.         }
  150.  
  151.         /* It is a command substitution, which will be parsed again   *
  152.          * by the lexer, so we untokenize it first, but we cannot use *
  153.          * untokenize() since in the case of `...` some Bnulls should *
  154.          * be left unchanged.  Note that the lexer doesn't tokenize   *
  155.          * the body of a command substitution so if there are some    *
  156.          * tokens here they are from a ${(e)~...} substitution.       */
  157.         for (str = str2; *++str; )
  158.         if (itok(*str) && *str != Nularg &&
  159.             !(endchar != Outpar && *str == Bnull &&
  160.               (str[1] == '$' || str[1] == '\\' || str[1] == '`' ||
  161.                (qt && str[1] == '"'))))
  162.             *str = ztokens[*str - Pound];
  163.         str++;
  164.         if (!(pl = getoutput(str2 + 1, qt || ssub))) {
  165.         zerr("parse error in command substitution", NULL, 0);
  166.         return NULL;
  167.         }
  168.         if (endchar == Outpar)
  169.         str2--;
  170.         if (!(s = (char *) ugetnode(pl))) {
  171.         str = strcpy(str2, str);
  172.         continue;
  173.         }
  174.         if (!qt && ssub && isset(GLOBSUBST))
  175.         tokenize(s);
  176.         l1 = str2 - str3;
  177.         l2 = strlen(s);
  178.         if (nonempty(pl)) {
  179.         LinkNode n = lastnode(pl);
  180.         str2 = (char *) ncalloc(l1 + l2 + 1);
  181.         strcpy(str2, str3);
  182.         strcpy(str2 + l1, s);
  183.         setdata(node, str2);
  184.         insertlinklist(pl, node, list);
  185.         s = (char *) getdata(node = n);
  186.         l1 = 0;
  187.         l2 = strlen(s);
  188.         }
  189.         str2 = (char *) ncalloc(l1 + l2 + strlen(str) + 1);
  190.         if (l1)
  191.         strcpy(str2, str3);
  192.         strcpy(str2 + l1, s);
  193.         str = strcpy(str2 + l1 + l2, str);
  194.         str3 = str2;
  195.         setdata(node, str3);
  196.         continue;
  197.     }
  198.     str++;
  199.     }
  200.     return errflag ? NULL : node;
  201. }
  202.  
  203. /**/
  204. void
  205. globlist(LinkList list)
  206. {
  207.     LinkNode node, next;
  208.  
  209.     badcshglob = 0;
  210.     for (node = firstnode(list); !errflag && node; node = next) {
  211.     next = nextnode(node);
  212.     glob(list, node);
  213.     }
  214.     if (badcshglob == 1)
  215.     zerr("no match", NULL, 0);
  216. }
  217.  
  218. /* perform substitution on a single word */
  219.  
  220. /**/
  221. void
  222. singsub(char **s)
  223. {
  224.     LinkList foo;
  225.  
  226.     foo = newlinklist();
  227.     addlinknode(foo, *s);
  228.     prefork(foo, 4);
  229.     if (errflag)
  230.     return;
  231.     *s = (char *) ugetnode(foo);
  232.     DPUTS(nonempty(foo), "BUG: singsub() produced more than one word!");
  233. }
  234.  
  235. /* Perform substitution on a single word. Unlike with singsub, the      *
  236.  * result can have more than one words. A single word result is sroted  *
  237.  * in *s and *isarr is set to zero; otherwise *isarr is set to 1 and    *
  238.  * the result is stored in *a. If `a' is zero a multiple word result is *
  239.  * joined using sep or the IFS parameter if sep is zero and the result  *
  240.  * is returned in *s.  The return value is true iff the expansion       *
  241.  * resulted in an empty list                                            */
  242.  
  243. /**/
  244. int
  245. multsub(char **s, char ***a, int *isarr, char *sep)
  246. {
  247.     LinkList foo;
  248.     int l;
  249.     char **r, **p;
  250.  
  251.     foo = newlinklist();
  252.     addlinknode(foo, *s);
  253.     prefork(foo, 0);
  254.     if (errflag) {
  255.     if (isarr)
  256.         *isarr = 0;
  257.     return 0;
  258.     }
  259.     if ((l = countlinknodes(foo)) > 1) {
  260.     p = r = ncalloc((l + 1) * sizeof(char*));
  261.     while (nonempty(foo))
  262.         *p++ = (char *)ugetnode(foo);
  263.     *p = NULL;
  264.     if (a) {
  265.         *a = r;
  266.         *isarr = 1;
  267.         return 0;
  268.     }
  269.     *s = sepjoin(r, NULL);
  270.     return 0;
  271.     }
  272.     if (l)
  273.     *s = (char *) ugetnode(foo);
  274.     else
  275.     *s = dupstring("");
  276.     if (isarr)
  277.     *isarr = 0;
  278.     return !l;
  279. }
  280.  
  281. /* ~, = subs: assign = 2 => typeset; assign = 1 => something that looks
  282.     like an assignment but may not be; assign = 3 => normal assignment */
  283.  
  284. /**/
  285. void
  286. filesub(char **namptr, int assign)
  287. {
  288.     char *sub = NULL, *str, *ptr;
  289.     int len;
  290.  
  291.     filesubstr(namptr, assign);
  292.  
  293.     if (!assign)
  294.     return;
  295.  
  296.     if (assign < 3)
  297.     if ((*namptr)[1] && (sub = strchr(*namptr + 1, Equals))) {
  298.         if (assign == 1)
  299.         for (ptr = *namptr; ptr != sub; ptr++)
  300.             if (!iident(*ptr) && !INULL(*ptr))
  301.             return;
  302.         str = sub + 1;
  303.         if ((sub[1] == Tilde || sub[1] == Equals) && filesubstr(&str, assign)) {
  304.         sub[1] = '\0';
  305.         *namptr = dyncat(*namptr, str);
  306.         }
  307.     } else
  308.         return;
  309.  
  310.     ptr = *namptr;
  311.     while ((sub = strchr(ptr, ':'))) {
  312.     str = sub + 1;
  313.     len = sub - *namptr;
  314.     if ((sub[1] == Tilde || sub[1] == Equals) && filesubstr(&str, assign)) {
  315.         sub[1] = '\0';
  316.         *namptr = dyncat(*namptr, str);
  317.     }
  318.     ptr = *namptr + len + 1;
  319.     }
  320. }
  321.  
  322. /**/
  323. int
  324. filesubstr(char **namptr, int assign)
  325. {
  326. #define isend(c) ( !(c) || (c)=='/' || (c)==Inpar || (assign && (c)==':') )
  327. #define isend2(c) ( !(c) || (c)==Inpar || (assign && (c)==':') )
  328.     char *str = *namptr;
  329.  
  330.     if (*str == Tilde && str[1] != '=' && str[1] != Equals) {
  331.     char *ptr;
  332.     int val;
  333.  
  334.     val = zstrtol(str + 1, &ptr, 10);
  335.     if (isend(str[1])) {   /* ~ */
  336.         *namptr = dyncat(home, str + 1);
  337.         return 1;
  338.     } else if (str[1] == '+' && isend(str[2])) {   /* ~+ */
  339.         *namptr = dyncat(pwd, str + 2);
  340.         return 1;
  341.     } else if (str[1] == '-' && isend(str[2])) {   /* ~- */
  342.         char *tmp;
  343.         *namptr = dyncat((tmp = oldpwd) ? tmp : pwd, str + 2);
  344.         return 1;
  345.     } else if (!inblank(str[1]) && isend(*ptr) &&
  346.            (!idigit(str[1]) || (ptr - str < 4))) {
  347.         char *ds;
  348.  
  349.         if (val < 0)
  350.         val = -val;
  351.         ds = dstackent(str[1], val);
  352.         if (!ds)
  353.         return 0;
  354.         *namptr = dyncat(ds, ptr);
  355.         return 1;
  356.     } else if (iuser(str[1])) {   /* ~foo */
  357.         char *ptr, *hom, save;
  358.  
  359.         for (ptr = ++str; *ptr && iuser(*ptr); ptr++);
  360.         save = *ptr;
  361.         if (!isend(save))
  362.         return 0;
  363.         *ptr = 0;
  364.         if (!(hom = getnameddir(str))) {
  365.         if (isset(NOMATCH))
  366.             zerr("no such user or named directory: %s", str, 0);
  367.         *ptr = save;
  368.         return 0;
  369.         }
  370.         *ptr = save;
  371.         *namptr = dyncat(hom, ptr);
  372.         return 1;
  373.     }
  374.     } else if (*str == Equals && isset(EQUALS) && str[1]) {   /* =foo */
  375.     char sav, *pp, *cnam;
  376.  
  377.     for (pp = str + 1; !isend2(*pp); pp++);
  378.     sav = *pp;
  379.     *pp = 0;
  380.     if (!(cnam = findcmd(str + 1))) {
  381.         Alias a = (Alias) aliastab->getnode(aliastab, str + 1);
  382.         
  383.         if (a)
  384.         cnam = ztrdup(a->text);
  385.         else {
  386.         if (isset(NOMATCH))
  387.             zerr("%s not found", str + 1, 0);
  388.         return 0;
  389.         }
  390.     }
  391.     *namptr = dupstring(cnam);
  392.     zsfree(cnam);
  393.     if (sav) {
  394.         *pp = sav;
  395.         *namptr = dyncat(*namptr, pp);
  396.     }
  397.     return 1;
  398.     }
  399.     return 0;
  400. #undef isend
  401. #undef isend2
  402. }
  403.  
  404. /**/
  405. char *
  406. strcatsub(char **d, char *pb, char *pe, char *src, int l, char *s, int glbsub)
  407. {
  408.     int pl = pe - pb;
  409.     char *dest = ncalloc(pl + l + (s ? strlen(s) : 0) + 1);
  410.  
  411.     *d = dest;
  412.     strncpy(dest, pb, pl);
  413.     dest += pl;
  414.     strcpy(dest, src);
  415.     if (glbsub)
  416.     tokenize(dest);
  417.     dest += l;
  418.     if (s)
  419.     strcpy(dest, s);
  420.     return dest;
  421. }
  422.  
  423. typedef int (*CompareFn) _((const void *, const void *));
  424.  
  425. /**/
  426. int
  427. strpcmp(const void *a, const void *b)
  428. {
  429. #ifdef HAVE_STRCOLL
  430.     return strcoll(*(char **)a, *(char **)b);
  431. #else
  432.     return strcmp(*(char **)a, *(char **)b);
  433. #endif
  434. }
  435.  
  436. /**/
  437. int
  438. invstrpcmp(const void *a, const void *b)
  439. {
  440. #ifdef HAVE_STRCOLL
  441.     return -strcoll(*(char **)a, *(char **)b);
  442. #else
  443.     return -strcmp(*(char **)a, *(char **)b);
  444. #endif
  445. }
  446.  
  447. /**/
  448. int
  449. cstrpcmp(const void *a, const void *b)
  450. {
  451. #ifdef HAVE_STRCOLL
  452. # ifdef __GNUC__
  453.     char c[strlen(*(char **) a) + 1];
  454.     char d[strlen(*(char **) b) + 1];
  455. # else
  456.     char *c = halloc(strlen(*(char **) a) + 1);
  457.     char *d = halloc(strlen(*(char **) b) + 1);
  458. # endif
  459.     char *s, *t;
  460.     int   cmp;
  461.  
  462.     for (s = *(char **) a, t = c; (*t++ = tulower(*s++)););
  463.     for (s = *(char **) b, t = d; (*t++ = tulower(*s++)););
  464.  
  465.     cmp = strcoll(c, d);
  466.  
  467.     return cmp;
  468. #else
  469.     char *c = *(char **)a, *d = *(char **)b;
  470.  
  471.     for (; *c && tulower(*c) == tulower(*d); c++, d++);
  472.  
  473.     return (int)STOUC(tulower(*c)) - (int)STOUC(tulower(*d));
  474. #endif
  475. }
  476.  
  477. /**/
  478. int
  479. invcstrpcmp(const void *a, const void *b)
  480. {
  481. #ifdef HAVE_STRCOLL
  482. # ifdef __GNUC__
  483.     char c[strlen(*(char **) a) + 1];
  484.     char d[strlen(*(char **) b) + 1];
  485. # else
  486.     char *c = halloc(strlen(*(char **) a) + 1);
  487.     char *d = halloc(strlen(*(char **) b) + 1);
  488. # endif
  489.     char *s, *t;
  490.     int   cmp;
  491.  
  492.     for (s = *(char **) a, t = c; (*t++ = tulower(*s++)););
  493.     for (s = *(char **) b, t = d; (*t++ = tulower(*s++)););
  494.  
  495.     cmp = strcoll(c, d);
  496.  
  497.     return -cmp;
  498. #else
  499.     char *c = *(char **)a, *d = *(char **)b;
  500.  
  501.     for (; *c && tulower(*c) == tulower(*d); c++, d++);
  502.  
  503.     return (int)STOUC(tulower(*d)) - (int)STOUC(tulower(*c));
  504. #endif
  505. }
  506.  
  507. /**/
  508. char *
  509. dopadding(char *str, int prenum, int postnum, char *preone, char *postone, char *premul, char *postmul)
  510. {
  511.     char def[3], *ret, *t, *r;
  512.     int ls, ls2, lpreone, lpostone, lpremul, lpostmul, lr, f, m, c, cc;
  513.  
  514.     def[0] = *ifs ? *ifs : ' ';
  515.     def[1] = *ifs == Meta ? ifs[1] ^ 32 : '\0';
  516.     def[2] = '\0';
  517.     if (preone && !*preone)
  518.     preone = def;
  519.     if (postone && !*postone)
  520.     postone = def;
  521.     if (!premul || !*premul)
  522.     premul = def;
  523.     if (!postmul || !*postmul)
  524.     postmul = def;
  525.  
  526.     ls = strlen(str);
  527.     lpreone = preone ? strlen(preone) : 0;
  528.     lpostone = postone ? strlen(postone) : 0;
  529.     lpremul = strlen(premul);
  530.     lpostmul = strlen(postmul);
  531.  
  532.     lr = prenum + postnum;
  533.  
  534.     if (lr == ls)
  535.     return str;
  536.  
  537.     r = ret = (char *)halloc(lr + 1);
  538.  
  539.     if (prenum) {
  540.     if (postnum) {
  541.         ls2 = ls / 2;
  542.  
  543.         f = prenum - ls2;
  544.         if (f <= 0)
  545.         for (str -= f, c = prenum; c--; *r++ = *str++);
  546.         else {
  547.         if (f <= lpreone)
  548.             for (c = f, t = preone + lpreone - f; c--; *r++ = *t++);
  549.         else {
  550.             f -= lpreone;
  551.             if ((m = f % lpremul))
  552.             for (c = m, t = premul + lpremul - m; c--; *r++ = *t++);
  553.             for (cc = f / lpremul; cc--;)
  554.             for (c = lpremul, t = premul; c--; *r++ = *t++);
  555.             for (c = lpreone; c--; *r++ = *preone++);
  556.         }
  557.         for (c = ls2; c--; *r++ = *str++);
  558.         }
  559.         ls2 = ls - ls2;
  560.         f = postnum - ls2;
  561.         if (f <= 0)
  562.         for (c = postnum; c--; *r++ = *str++);
  563.         else {
  564.         for (c = ls2; c--; *r++ = *str++);
  565.         if (f <= lpostone)
  566.             for (c = f; c--; *r++ = *postone++);
  567.         else {
  568.             f -= lpostone;
  569.             for (c = lpostone; c--; *r++ = *postone++);
  570.             for (cc = f / lpostmul; cc--;)
  571.             for (c = lpostmul, t = postmul; c--; *r++ = *t++);
  572.             if ((m = f % lpostmul))
  573.             for (; m--; *r++ = *postmul++);
  574.         }
  575.         }
  576.     } else {
  577.         f = prenum - ls;
  578.         if (f <= 0)
  579.         for (c = prenum, str -= f; c--; *r++ = *str++);
  580.         else {
  581.         if (f <= lpreone)
  582.             for (c = f, t = preone + lpreone - f; c--; *r++ = *t++);
  583.         else {
  584.             f -= lpreone;
  585.             if ((m = f % lpremul))
  586.             for (c = m, t = premul + lpremul - m; c--; *r++ = *t++);
  587.             for (cc = f / lpremul; cc--;)
  588.             for (c = lpremul, t = premul; c--; *r++ = *t++);
  589.             for (c = lpreone; c--; *r++ = *preone++);
  590.         }
  591.         for (c = ls; c--; *r++ = *str++);
  592.         }
  593.     }
  594.     } else if (postnum) {
  595.     f = postnum - ls;
  596.     if (f <= 0)
  597.         for (c = postnum; c--; *r++ = *str++);
  598.     else {
  599.         for (c = ls; c--; *r++ = *str++);
  600.         if (f <= lpostone)
  601.         for (c = f; c--; *r++ = *postone++);
  602.         else {
  603.         f -= lpostone;
  604.         for (c = lpostone; c--; *r++ = *postone++);
  605.         for (cc = f / lpostmul; cc--;)
  606.             for (c = lpostmul, t = postmul; c--; *r++ = *t++);
  607.         if ((m = f % lpostmul))
  608.             for (; m--; *r++ = *postmul++);
  609.         }
  610.     }
  611.     }
  612.     *r = '\0';
  613.  
  614.     return ret;
  615. }
  616.  
  617. /**/
  618. char *
  619. get_strarg(char *s)
  620. {
  621.     char t = *s++;
  622.  
  623.     if (!t)
  624.     return s - 1;
  625.  
  626.     switch (t) {
  627.     case '(':
  628.     t = ')';
  629.     break;
  630.     case '[':
  631.     t = ']';
  632.     break;
  633.     case '{':
  634.     t = '}';
  635.     break;
  636.     case '<':
  637.     t = '>';
  638.     break;
  639.     case Inpar:
  640.     t = Outpar;
  641.     break;
  642.     case Inang:
  643.     t = Outang;
  644.     break;
  645.     case Inbrace:
  646.     t = Outbrace;
  647.     break;
  648.     case Inbrack:
  649.     t = Outbrack;
  650.     break;
  651.     }
  652.  
  653.     while (*s && *s != t)
  654.     s++;
  655.  
  656.     return s;
  657. }
  658.  
  659. /**/
  660. int
  661. get_intarg(char **s)
  662. {
  663.     char *t = get_strarg(*s + 1);
  664.     char *p, sav;
  665.     long ret;
  666.  
  667.     if (!*t)
  668.     return -1;
  669.     sav = *t;
  670.     *t = '\0';
  671.     p = dupstring(*s + 2);
  672.     *s = t;
  673.     *t = sav;
  674.     if (parsestr(p))
  675.     return -1;
  676.     singsub(&p);
  677.     if (errflag)
  678.     return -1;
  679.     ret = matheval(p);
  680.     if (errflag)
  681.     return -1;
  682.     if (ret < 0)
  683.     ret = -ret;
  684.     return ret < 0 ? -ret : ret;
  685. }
  686.  
  687. /* parameter substitution */
  688.  
  689. #define    isstring(c) ((c) == '$' || (char)(c) == String || (char)(c) == Qstring)
  690. #define isbrack(c)  ((c) == '[' || (char)(c) == Inbrack)
  691.  
  692. /**/
  693. LinkNode
  694. paramsubst(LinkList l, LinkNode n, char **str, int qt, int ssub)
  695. {
  696.     char *aptr = *str;
  697.     char *s = aptr, *u, *idbeg, *idend, *ostr = (char *) getdata(n);
  698.     int colf;            /* != 0 means we found a colon after the name */
  699.     int doub = 0;        /* != 0 means we have %%, not %, or ##, not # */
  700.     int isarr = 0;
  701.     int plan9 = isset(RCEXPANDPARAM);
  702.     int globsubst = isset(GLOBSUBST);
  703.     int getlen = 0;
  704.     int whichlen = 0;
  705.     int chkset = 0;
  706.     int vunset = 0;
  707.     int spbreak = isset(SHWORDSPLIT) && !ssub && !qt;
  708.     char *val = NULL, **aval = NULL;
  709.     unsigned int fwidth = 0;
  710.     Value v;
  711.     int flags = 0;
  712.     int flnum = 0;
  713.     int substr = 0;
  714.     int sortit = 0, casind = 0;
  715.     int casmod = 0;
  716.     char *sep = NULL, *spsep = NULL;
  717.     char *premul = NULL, *postmul = NULL, *preone = NULL, *postone = NULL;
  718.     long prenum = 0, postnum = 0;
  719.     int copied = 0;
  720.     int arrasg = 0;
  721.     int eval = 0;
  722.     int nojoin = 0;
  723.     char inbrace = 0;        /* != 0 means ${...}, otherwise $... */
  724.  
  725.     *s++ = '\0';
  726.     if (!ialnum(*s) && *s != '#' && *s != Pound && *s != '-' &&
  727.     *s != '!' && *s != '$' && *s != String && *s != Qstring &&
  728.     *s != '?' && *s != Quest && *s != '_' &&
  729.     *s != '*' && *s != Star && *s != '@' && *s != '{' &&
  730.     *s != Inbrace && *s != '=' && *s != Equals && *s != Hat &&
  731.     *s != '^' && *s != '~' && *s != Tilde && *s != '+') {
  732.     s[-1] = '$';
  733.     *str = s;
  734.     return n;
  735.     }
  736.     DPUTS(*s == '{', "BUG: inbrace == '{' in paramsubst()");
  737.     if (*s == Inbrace) {
  738.     inbrace = 1;
  739.     s++;
  740.     if (*s == '(' || *s == Inpar) {
  741.         char *t, sav;
  742.         int tt = 0;
  743.         long num;
  744.         int escapes = 0;
  745.         int klen;
  746. #define UNTOK_AND_ESCAPE(X) {\
  747.         untokenize(X = dupstring(s + 1));\
  748.         if (escapes) {\
  749.             X = getkeystring(X, &klen, 3, NULL);\
  750.             X = metafy(X, klen, META_USEHEAP);\
  751.         }\
  752.         }
  753.  
  754.         for (s++; *s != ')' && *s != Outpar; s++, tt = 0) {
  755.         switch (*s) {
  756.         case ')':
  757.         case Outpar:
  758.             break;
  759.         case 'A':
  760.             arrasg = 1;
  761.             break;
  762.         case '@':
  763.             nojoin = 1;
  764.             break;
  765.         case 'M':
  766.             flags |= 8;
  767.             break;
  768.         case 'R':
  769.             flags |= 16;
  770.             break;
  771.         case 'B':
  772.             flags |= 32;
  773.             break;
  774.         case 'E':
  775.             flags |= 64;
  776.             break;
  777.         case 'N':
  778.             flags |= 128;
  779.             break;
  780.         case 'S':
  781.             substr = 1;
  782.             break;
  783.         case 'I':
  784.             flnum = get_intarg(&s);
  785.             if (flnum < 0)
  786.             goto flagerr;
  787.             break;
  788.  
  789.         case 'L':
  790.             casmod = 2;
  791.             break;
  792.         case 'U':
  793.             casmod = 1;
  794.             break;
  795.         case 'C':
  796.             casmod = 3;
  797.             break;
  798.  
  799.         case 'o':
  800.             sortit = 1;
  801.             break;
  802.         case 'O':
  803.             sortit = 2;
  804.             break;
  805.         case 'i':
  806.             casind = 1;
  807.             break;
  808.         case 'e':
  809.             eval = 1;
  810.             break;
  811.  
  812.         case 'c':
  813.             whichlen = 1;
  814.             break;
  815.         case 'w':
  816.             whichlen = 2;
  817.             break;
  818.         case 'W':
  819.             whichlen = 3;
  820.             break;
  821.  
  822.         case 'f':
  823.             spsep = "\n";
  824.             break;
  825.         case 'F':
  826.             sep = "\n";
  827.             break;
  828.  
  829.         case 's':
  830.             tt = 1;
  831.         /* fall through */
  832.         case 'j':
  833.             t = get_strarg(++s);
  834.             if (*t) {
  835.             sav = *t;
  836.             *t = '\0';
  837.             if (tt)
  838.                 UNTOK_AND_ESCAPE(spsep)
  839.             else
  840.                 UNTOK_AND_ESCAPE(sep)
  841.             *t = sav;
  842.             s = t;
  843.             } else
  844.             goto flagerr;
  845.             break;
  846.  
  847.         case 'l':
  848.             tt = 1;
  849.         /* fall through */
  850.         case 'r':
  851.             sav = s[1];
  852.             num = get_intarg(&s);
  853.             if (num < 0)
  854.             goto flagerr;
  855.             if (tt)
  856.             prenum = num;
  857.             else
  858.             postnum = num;
  859.             if (s[1] != sav)
  860.             break;
  861.             t = get_strarg(++s);
  862.             if (!*t)
  863.             goto flagerr;
  864.             sav = *t;
  865.             *t = '\0';
  866.             if (tt)
  867.             UNTOK_AND_ESCAPE(premul)
  868.             else
  869.             UNTOK_AND_ESCAPE(postmul)
  870.             *t = sav;
  871.             sav = *s;
  872.             s = t + 1;
  873.             if (*s != sav) {
  874.             s--;
  875.             break;
  876.             }
  877.             t = get_strarg(s);
  878.             if (!*t)
  879.             goto flagerr;
  880.             sav = *t;
  881.             *t = '\0';
  882.             if (tt)
  883.             UNTOK_AND_ESCAPE(preone)
  884.             else
  885.             UNTOK_AND_ESCAPE(postone)
  886.             *t = sav;
  887.             s = t;
  888.             break;
  889.  
  890.         case 'p':
  891.             escapes = 1;
  892.             break;
  893.  
  894.         default:
  895.           flagerr:
  896.             zerr("error in flags", NULL, 0);
  897.             return NULL;
  898.         }
  899.         }
  900.         s++;
  901.     }
  902.     }
  903.     if (sortit)
  904.     sortit += (casind << 1);
  905.  
  906.     if (!premul)
  907.     premul = " ";
  908.     if (!postmul)
  909.     postmul = " ";
  910.  
  911.     for (;;) {
  912.     if (*s == '^' || *s == Hat) {
  913.         if (*++s == '^' || *s == Hat) {
  914.         plan9 = 0;
  915.         s++;
  916.         } else
  917.         plan9 = 1;
  918.     } else if (*s == '=' || *s == Equals) {
  919.         if (*++s == '=' || *s == Equals) {
  920.         spbreak = 0;
  921.         s++;
  922.         } else
  923.         spbreak = 1;
  924.     } else if ((*s == '#' || *s == Pound) &&
  925.            (iident(s[1])
  926.             || s[1] == '*' || s[1] == Star || s[1] == '@'
  927.             || (isstring(s[1]) && (s[2] == Inbrace || s[2] == Inpar))))
  928.         getlen = 1 + whichlen, s++;
  929.     else if (*s == '~' || *s == Tilde) {
  930.         if (*++s == '~' || *s == Tilde) {
  931.         globsubst = 0;
  932.         s++;
  933.         } else
  934.         globsubst = 1;
  935.     } else if (*s == '+')
  936.         if (iident(s[1]))
  937.         chkset = 1, s++;
  938.         else if (!inbrace) {
  939.         *aptr = '$';
  940.         *str = aptr + 1;
  941.         return n;
  942.         } else {
  943.         zerr("bad substitution", NULL, 0);
  944.         return NULL;
  945.         }
  946.     else
  947.         break;
  948.     }
  949.     globsubst = globsubst && !qt;
  950.  
  951.     idbeg = s;
  952.     if (s[-1] && isstring(*s) && (s[1] == Inbrace || s[1] == Inpar)) {
  953.     int sav;
  954.     int quoted = *s == Qstring;
  955.  
  956.     val = s++;
  957.     skipparens(*s, *s == Inpar ? Outpar : Outbrace, &s);
  958.     sav = *s;
  959.     *s = 0;
  960.     if (multsub(&val, &aval, &isarr, NULL) && quoted) {
  961.         isarr = -1;
  962.         aval = alloc(sizeof(char *));
  963.     }
  964.     if (isarr)
  965.         isarr = -1;
  966.     copied = 1;
  967.     *s = sav;
  968.     v = (Value) NULL;
  969.     } else if (!(v = getvalue(&s, (unset(KSHARRAYS) || inbrace) ? 1 : -1)))
  970.     vunset = 1;
  971.     while (v || ((inbrace || (unset(KSHARRAYS) && vunset)) && isbrack(*s))) {
  972.     if (!v) {
  973.         Param pm;
  974.         char *os = s;
  975.  
  976.         if (!isbrack(*s))
  977.         break;
  978.         if (vunset) {
  979.         val = dupstring("");
  980.         isarr = 0;
  981.         }
  982.         pm = createparam(nulstring, isarr ? PM_ARRAY : PM_SCALAR);
  983.         if (isarr)
  984.         pm->u.arr = aval;
  985.         else
  986.         pm->u.str = val;
  987.         v = (Value) hcalloc(sizeof *v);
  988.         v->isarr = isarr;
  989.         v->pm = pm;
  990.         v->b = -1;
  991.         if (getindex(&s, v) || s == os)
  992.         break;
  993.     }
  994.     if ((isarr = v->isarr))
  995.         aval = getarrvalue(v);
  996.     else {
  997.         if (v->pm->flags & PM_ARRAY) {
  998.         int tmplen = arrlen(v->pm->gets.afn(v->pm));
  999.  
  1000.         if (v->a < 0)
  1001.             v->a += tmplen + v->inv;
  1002.         if (!v->inv && (v->a >= tmplen || v->a < 0))
  1003.             vunset = 1;
  1004.         }
  1005.         if (!vunset) {
  1006.         val = getstrvalue(v);
  1007.         fwidth = v->pm->ct ? v->pm->ct : strlen(val);
  1008.         switch (v->pm->flags & (PM_LEFT | PM_RIGHT_B | PM_RIGHT_Z)) {
  1009.             char *t;
  1010.             unsigned int t0;
  1011.  
  1012.         case PM_LEFT:
  1013.         case PM_LEFT | PM_RIGHT_Z:
  1014.             t = val;
  1015.             if (v->pm->flags & PM_RIGHT_Z)
  1016.             while (*t == '0')
  1017.                 t++;
  1018.             else
  1019.             while (iblank(*t))
  1020.                 t++;
  1021.             val = (char *)ncalloc(fwidth + 1);
  1022.             val[fwidth] = '\0';
  1023.             if ((t0 = strlen(t)) > fwidth)
  1024.             t0 = fwidth;
  1025.             memset(val, ' ', fwidth);
  1026.             strncpy(val, t, t0);
  1027.             break;
  1028.         case PM_RIGHT_B:
  1029.         case PM_RIGHT_Z:
  1030.         case PM_RIGHT_Z | PM_RIGHT_B:
  1031.             if (strlen(val) < fwidth) {
  1032.             t = (char *)ncalloc(fwidth + 1);
  1033.             memset(t, (v->pm->flags & PM_RIGHT_B) ? ' ' : '0', fwidth);
  1034.             if ((t0 = strlen(val)) > fwidth)
  1035.                 t0 = fwidth;
  1036.             strcpy(t + (fwidth - t0), val);
  1037.             val = t;
  1038.             } else {
  1039.             t = (char *)ncalloc(fwidth + 1);
  1040.             t[fwidth] = '\0';
  1041.             strncpy(t, val + strlen(val) - fwidth, fwidth);
  1042.             val = t;
  1043.             }
  1044.             break;
  1045.         }
  1046.         switch (v->pm->flags & (PM_LOWER | PM_UPPER)) {
  1047.             char *t;
  1048.  
  1049.         case PM_LOWER:
  1050.             t = val;
  1051.             for (; *t; t++)
  1052.             *t = tulower(*t);
  1053.             break;
  1054.         case PM_UPPER:
  1055.             t = val;
  1056.             for (; *t; t++)
  1057.             *t = tuupper(*t);
  1058.             break;
  1059.         }
  1060.         }
  1061.     }
  1062.     v = NULL;
  1063.     if (!inbrace)
  1064.         break;
  1065.     }
  1066.     if (isarr) {
  1067.     if (nojoin)
  1068.         isarr = -1;
  1069.     if (qt && !getlen && isarr > 0) {
  1070.         val = sepjoin(aval, sep);
  1071.         isarr = 0;
  1072.     }
  1073.     }
  1074.  
  1075.     idend = s;
  1076.     if ((colf = *s == ':'))
  1077.     s++;
  1078.  
  1079.     /* Check for ${..?..} or ${..=..} or one of those. *
  1080.      * Only works if the name is in braces.            */
  1081.  
  1082.     if (inbrace && (*s == '-' ||
  1083.             *s == '+' ||
  1084.             *s == ':' ||
  1085.             *s == '=' || *s == Equals ||
  1086.             *s == '%' ||
  1087.             *s == '#' || *s == Pound ||
  1088.             *s == '?' || *s == Quest)) {
  1089.     int bct;
  1090.  
  1091.     if (!flnum)
  1092.         flnum++;
  1093.     if (*s == '%')
  1094.         flags |= 1;
  1095.  
  1096.     /* Check for ${..%%..} or ${..##..} */
  1097.     if ((*s == '%' || *s == '#' || *s == Pound) && *s == s[1]) {
  1098.         s++;
  1099.         doub = 1;
  1100.     }
  1101.     u = s + 1;
  1102.  
  1103.     flags |= (doub << 1) | (substr << 2) | (colf << 8);
  1104.     if (!(flags & 0xf8))
  1105.         flags |= 16;
  1106.  
  1107.     for (bct = 1; bct && *++s;) {
  1108.         if (*s == Inbrace)
  1109.         bct++;
  1110.         else if (*s == Outbrace)
  1111.         bct--;
  1112.     }
  1113.  
  1114.     if (bct)
  1115.         goto noclosebrace;
  1116.  
  1117.     if (*s)
  1118.         *s++ = '\0';
  1119.     if (colf && !vunset)
  1120.         vunset = (isarr) ? !*aval : !*val || (*val == Nularg && !val[1]);
  1121.  
  1122.     switch (u[-1]) {
  1123.     case '+':
  1124.         if (vunset) {
  1125.         val = dupstring("");
  1126.         copied = 1;
  1127.         isarr = 0;
  1128.         break;
  1129.         }
  1130.         vunset = 1;
  1131.     /* Fall Through! */
  1132.     case '-':
  1133.         if (vunset) {
  1134.         val = dupstring(u);
  1135.         multsub(&val, &aval, &isarr, NULL);
  1136.         copied = 1;
  1137.         }
  1138.         break;
  1139.     case ':':
  1140.         if (*u != '=' && *u != Equals)
  1141.         goto noclosebrace;
  1142.         vunset = 1;
  1143.         u++;
  1144.         /* Fall through */
  1145.     case '=':
  1146.     case Equals:
  1147.         if (vunset) {
  1148.         char sav = *idend;
  1149.         int l;
  1150.  
  1151.         *idend = '\0';
  1152.         val = dupstring(u);
  1153.         isarr = 0;
  1154.         if (spsep || spbreak || !arrasg)
  1155.             multsub(&val, NULL, NULL, sep);
  1156.         else
  1157.             multsub(&val, &aval, &isarr, NULL);
  1158.         if (arrasg) {
  1159.             char *arr[2], **t, **a, **p;
  1160.             if (spsep || spbreak) {
  1161.             aval = sepsplit(val, spsep, 0);
  1162.             isarr = 2;
  1163.             sep = spsep = NULL;
  1164.             spbreak = 0;
  1165.             l = arrlen(aval);
  1166.             if (l && !*(aval[l-1]))
  1167.                 l--;
  1168.             if (l && !**aval)
  1169.                 l--, t = aval + 1;
  1170.             else
  1171.                 t = aval;
  1172.             } else if (!isarr) {
  1173.             arr[0] = val;
  1174.             arr[1] = NULL;
  1175.             t = aval = arr;
  1176.             l = 1;
  1177.             } else
  1178.             l = arrlen(aval), t = aval;
  1179.             p = a = zalloc(sizeof(char *) * (l + 1));
  1180.             while (l--) {
  1181.             untokenize(*t);
  1182.             *p++ = ztrdup(*t++);
  1183.             }
  1184.             *p++ = NULL;
  1185.             setaparam(idbeg, a);
  1186.         } else {
  1187.             untokenize(val);
  1188.             setsparam(idbeg, ztrdup(val));
  1189.         }
  1190.         *idend = sav;
  1191.         copied = 1;
  1192.         }
  1193.         break;
  1194.     case '?':
  1195.     case Quest:
  1196.         if (vunset) {
  1197.         char *msg;
  1198.  
  1199.         *idend = '\0';
  1200.         msg = tricat(idbeg, ": ", *u ? u : "parameter not set");
  1201.         zerr("%s", msg, 0);
  1202.         zsfree(msg);
  1203.         if (!interact)
  1204.             exit(1);
  1205.         return NULL;
  1206.         }
  1207.         break;
  1208.     case '%':
  1209.     case '#':
  1210.     case Pound:
  1211.         if (qt)
  1212.         if (parse_subst_string(u)) {
  1213.             zerr("parse error in ${...%c...} substitution",
  1214.              NULL, u[-1]);
  1215.             return NULL;
  1216.         }
  1217.         singsub(&u);
  1218.  
  1219.         if (!vunset && isarr) {
  1220.         char **ap = aval;
  1221.         char **pp = aval = (char **)ncalloc(sizeof(char *) * (arrlen(aval) + 1));
  1222.  
  1223.         while ((*pp = *ap++)) {
  1224.             if (getmatch(pp, u, flags, flnum))
  1225.             pp++;
  1226.         }
  1227.         copied = 1;
  1228.         } else {
  1229.         if (vunset)
  1230.             val = dupstring("");
  1231.         getmatch(&val, u, flags, flnum);
  1232.         copied = 1;
  1233.         }
  1234.         break;
  1235.     }
  1236.     } else {            /* no ${...=...} or anything, but possible modifiers. */
  1237.     if (chkset) {
  1238.         val = dupstring(vunset ? "0" : "1");
  1239.         isarr = 0;
  1240.     } else if (vunset) {
  1241.         if (unset(UNSET)) {
  1242.         *idend = '\0';
  1243.         zerr("%s: parameter not set", idbeg, 0);
  1244.         return NULL;
  1245.         }
  1246.         val = dupstring("");
  1247.     }
  1248.     if (colf) {
  1249.         s--;
  1250.         if (unset(KSHARRAYS) || inbrace) {
  1251.         if (!isarr)
  1252.             modify(&val, &s);
  1253.         else {
  1254.             char *ss;
  1255.             char **ap = aval;
  1256.             char **pp = aval = (char **)ncalloc(sizeof(char *) * (arrlen(aval) + 1));
  1257.  
  1258.             while ((*pp = *ap++)) {
  1259.             ss = s;
  1260.             modify(pp++, &ss);
  1261.             }
  1262.             if (pp == aval) {
  1263.             char *t = "";
  1264.             ss = s;
  1265.             modify(&t, &ss);
  1266.             }
  1267.             s = ss;
  1268.         }
  1269.         if (inbrace && *s != Outbrace) {
  1270.             if (*s == ':' && !imeta(s[1]))
  1271.             zerr("unrecognized modifier `%c'", NULL, s[1]);
  1272.             else
  1273.             zerr("unrecognized modifier", NULL, 0);
  1274.             return NULL;
  1275.         }
  1276.         }
  1277.     }
  1278.     if (inbrace) {
  1279.         if (*s != Outbrace) {
  1280.           noclosebrace:
  1281.         zerr("closing brace expected", NULL, 0);
  1282.         return NULL;
  1283.         }
  1284.         s++;
  1285.     }
  1286.     }
  1287.     if (errflag)
  1288.     return NULL;
  1289.     if (getlen) {
  1290.     long len = 0;
  1291.     char buf[14];
  1292.  
  1293.     if (isarr) {
  1294.         char **ctr;
  1295.         int sl = sep ? ztrlen(sep) : 1;
  1296.  
  1297.         if (getlen == 1)
  1298.         for (ctr = aval; *ctr; ctr++, len++);
  1299.         else if (getlen == 2) {
  1300.         if (*aval)
  1301.             for (len = -sl, ctr = aval;
  1302.              len += sl + ztrlen(*ctr), *++ctr;);
  1303.         }
  1304.         else
  1305.         for (ctr = aval;
  1306.              *ctr;
  1307.              len += wordcount(*ctr, spsep, getlen > 3), ctr++);
  1308.     } else {
  1309.         if (getlen < 3)
  1310.         len = ztrlen(val);
  1311.         else
  1312.         len = wordcount(val, spsep, getlen > 3);
  1313.     }
  1314.  
  1315.     sprintf(buf, "%ld", len);
  1316.     val = dupstring(buf);
  1317.     isarr = 0;
  1318.     }
  1319.     if (isarr > 0 && !plan9 && (!aval || !aval[0])) {
  1320.     val = dupstring("");
  1321.     isarr = 0;
  1322.     } else if (isarr && aval && aval[0] && !aval[1]) {
  1323.     val = aval[0];
  1324.     isarr = 0;
  1325.     }
  1326.     /* ssub is true when we are called from singsub (via prefork).
  1327.      * It means that we must join arrays and should not split words. */
  1328.     if (ssub || spbreak || spsep || sep) {
  1329.     if (isarr)
  1330.         val = sepjoin(aval, sep), isarr = 0;
  1331.     if (!ssub && (spbreak || spsep)) {
  1332.         aval = sepsplit(val, spsep, 0);
  1333.         if (!aval || !aval[0])
  1334.         val = dupstring("");
  1335.         else if (!aval[1])
  1336.         val = aval[0];
  1337.         else
  1338.         isarr = 2;
  1339.     }
  1340.     }
  1341.     if (casmod) {
  1342.     if (isarr) {
  1343.         char **ap;
  1344.  
  1345.         if (!copied)
  1346.         aval = arrdup(aval), copied = 1;
  1347.         ap = aval;
  1348.  
  1349.         if (casmod == 1)
  1350.         for (; *ap; ap++)
  1351.             makeuppercase(ap);
  1352.         else if (casmod == 2)
  1353.         for (; *ap; ap++)
  1354.             makelowercase(ap);
  1355.         else
  1356.         for (; *ap; ap++)
  1357.             makecapitals(ap);
  1358.  
  1359.     } else {
  1360.         if (!copied)
  1361.         val = dupstring(val), copied = 1;
  1362.         if (casmod == 1)
  1363.         makeuppercase(&val);
  1364.         else if (casmod == 2)
  1365.         makelowercase(&val);
  1366.         else
  1367.         makecapitals(&val);
  1368.     }
  1369.     }
  1370.     if (isarr) {
  1371.     char *x;
  1372.     char *y;
  1373.     int xlen;
  1374.     int i;
  1375.     LinkNode on = n;
  1376.  
  1377.     if (!aval[0] && !plan9) {
  1378.         if (aptr > (char *) getdata(n) &&
  1379.         aptr[-1] == Dnull && *s == Dnull)
  1380.         *--aptr = '\0', s++;
  1381.         y = (char *)ncalloc((aptr - ostr) + strlen(s) + 1);
  1382.         strcpy(y, ostr);
  1383.         *str = y + (aptr - ostr);
  1384.         strcpy(*str, s);
  1385.         setdata(n, y);
  1386.         return n;
  1387.     }
  1388.     if (sortit) {
  1389.         static CompareFn sortfn[] = {
  1390.         strpcmp, invstrpcmp, cstrpcmp, invcstrpcmp
  1391.         };
  1392.  
  1393.         if (!copied)
  1394.         aval = arrdup(aval);
  1395.  
  1396.         i = arrlen(aval);
  1397.         if (i && (*aval[i-1] || --i))
  1398.         qsort(aval, i, sizeof(char *), sortfn[sortit-1]);
  1399.     }
  1400.     if (plan9) {
  1401.         LinkList tl = newlinklist();
  1402.         LinkNode tn;
  1403.  
  1404.         *--s = Marker;
  1405.         addlinknode(tl, s);
  1406.         if (!eval && !stringsubst(tl, firstnode(tl), ssub))
  1407.         return NULL;
  1408.         *str = aptr;
  1409.         tn = firstnode(tl);
  1410.         while ((x = *aval++)) {
  1411.         if (prenum || postnum)
  1412.             x = dopadding(x, prenum, postnum, preone, postone,
  1413.                   premul, postmul);
  1414.         if (eval && parsestr(x))
  1415.             return NULL;
  1416.         xlen = strlen(x);
  1417.         for (tn = firstnode(tl);
  1418.              tn && *(y = (char *) getdata(tn)) == Marker;
  1419.              incnode(tn)) {
  1420.             strcatsub(&y, ostr, aptr, x, xlen, y + 1, globsubst);
  1421.             if (qt && !*y && isarr != 2)
  1422.             y = dupstring(nulstring);
  1423.             if (plan9)
  1424.             setdata(n, (void *) y), plan9 = 0;
  1425.             else
  1426.             insertlinknode(l, n, (void *) y), incnode(n);
  1427.         }
  1428.         }
  1429.         for (; tn; incnode(tn)) {
  1430.         y = (char *) getdata(tn);
  1431.         if (*y == Marker)
  1432.             continue;
  1433.         if (qt && !*y && isarr != 2)
  1434.             y = dupstring(nulstring);
  1435.         if (plan9)
  1436.             setdata(n, (void *) y), plan9 = 0;
  1437.         else
  1438.             insertlinknode(l, n, (void *) y), incnode(n);
  1439.         }
  1440.         if (plan9) {
  1441.         uremnode(l, n);
  1442.         return NULL;
  1443.         }
  1444.     } else {
  1445.         x = aval[0];
  1446.         if (prenum || postnum)
  1447.         x = dopadding(x, prenum, postnum, preone, postone,
  1448.                   premul, postmul);
  1449.         if (eval && parsestr(x))
  1450.         return NULL;
  1451.         xlen = strlen(x);
  1452.         strcatsub(&y, ostr, aptr, x, xlen, NULL, globsubst);
  1453.         if (qt && !*y && isarr != 2)
  1454.         y = dupstring(nulstring);
  1455.         setdata(n, (void *) y);
  1456.  
  1457.         i = 1;
  1458.         /* aval[1] is non-null here */
  1459.         while (aval[i + 1]) {
  1460.         x = aval[i++];
  1461.         if (prenum || postnum)
  1462.             x = dopadding(x, prenum, postnum, preone, postone,
  1463.                   premul, postmul);
  1464.         if (eval && parsestr(x))
  1465.             return NULL;
  1466.         if (qt && !*x && isarr != 2)
  1467.             y = dupstring(nulstring);
  1468.         else {
  1469.             y = dupstring(x);
  1470.             if (globsubst)
  1471.             tokenize(y);
  1472.         }
  1473.         insertlinknode(l, n, (void *) y), incnode(n);
  1474.         }
  1475.  
  1476.         x = aval[i];
  1477.         if (prenum || postnum)
  1478.         x = dopadding(x, prenum, postnum, preone, postone,
  1479.                   premul, postmul);
  1480.         if (eval && parsestr(x))
  1481.         return NULL;
  1482.         xlen = strlen(x);
  1483.         *str = strcatsub(&y, aptr, aptr, x, xlen, s, globsubst);
  1484.         if (qt && !*y && isarr != 2)
  1485.         y = dupstring(nulstring);
  1486.         insertlinknode(l, n, (void *) y), incnode(n);
  1487.     }
  1488.     if (eval)
  1489.         n = on;
  1490.     } else {
  1491.     int xlen;
  1492.     char *x;
  1493.     char *y;
  1494.  
  1495.     x = val;
  1496.     if (prenum || postnum)
  1497.         x = dopadding(x, prenum, postnum, preone, postone,
  1498.               premul, postmul);
  1499.     if (eval && parsestr(x))
  1500.         return NULL;
  1501.     xlen = strlen(x);
  1502.     *str = strcatsub(&y, ostr, aptr, x, xlen, s, globsubst);
  1503.     if (qt && !*y && isarr != 2)
  1504.         y = dupstring(nulstring);
  1505.     setdata(n, (void *) y);
  1506.     }
  1507.     if (eval)
  1508.     *str = (char *) getdata(n);
  1509.  
  1510.     return n;
  1511. }
  1512.  
  1513. /*
  1514.  * Arithmetic substitution: `a' is the string to be evaluated, `bptr'
  1515.  * points to the beginning of the string containing it.  The tail of
  1516.  * the string is given by `rest'. *bptr is modified with the substituted
  1517.  * string. The function returns a pointer to the tail in the substituted
  1518.  * string.
  1519.  */
  1520.  
  1521. /**/
  1522. char *
  1523. arithsubst(char *a, char **bptr, char *rest)
  1524. {
  1525.     char *s = *bptr, *t, buf[DIGBUFSIZE];
  1526.     char *b = buf;
  1527.     long v;
  1528.  
  1529.     singsub(&a);
  1530.     v = matheval(a);
  1531.     sprintf(buf, "%ld", v);
  1532.     t = *bptr = (char *)ncalloc(strlen(*bptr) + strlen(buf) + strlen(rest) + 1);
  1533.     t--;
  1534.     while ((*++t = *s++));
  1535.     t--;
  1536.     while ((*++t = *b++));
  1537.     strcat(t, rest);
  1538.     return t;
  1539. }
  1540.  
  1541. /**/
  1542. void
  1543. modify(char **str, char **ptr)
  1544. {
  1545.     char *ptr1, *ptr2, *ptr3, del, *lptr, c, *test, *sep, *t, *tt, tc, *e;
  1546.     char *copy, *all, *tmp, sav;
  1547.     int gbal, wall, rec, al, nl;
  1548.  
  1549.     test = NULL;
  1550.  
  1551.     if (**ptr == ':')
  1552.     *str = dupstring(*str);
  1553.  
  1554.     while (**ptr == ':') {
  1555.     lptr = *ptr;
  1556.     (*ptr)++;
  1557.     wall = gbal = 0;
  1558.     rec = 1;
  1559.     c = '\0';
  1560.     sep = NULL;
  1561.  
  1562.     for (; !c && **ptr;) {
  1563.         switch (**ptr) {
  1564.         case 'h':
  1565.         case 'r':
  1566.         case 'e':
  1567.         case 't':
  1568.         case 'l':
  1569.         case 'u':
  1570.         c = **ptr;
  1571.         break;
  1572.  
  1573.         case 's':
  1574.         c = **ptr;
  1575.         (*ptr)++;
  1576.         ptr1 = *ptr;
  1577.         del = *ptr1++;
  1578.         for (ptr2 = ptr1; *ptr2 != del && *ptr2; ptr2++);
  1579.         if (!*ptr2) {
  1580.             zerr("bad substitution", NULL, 0);
  1581.             return;
  1582.         }
  1583.         *ptr2++ = '\0';
  1584.         for (ptr3 = ptr2; *ptr3 != del && *ptr3; ptr3++);
  1585.         if ((sav = *ptr3))
  1586.             *ptr3++ = '\0';
  1587.         if (*ptr1) {
  1588.             zsfree(hsubl);
  1589.             hsubl = ztrdup(ptr1);
  1590.         }
  1591.         if (!hsubl) {
  1592.             zerr("no previous substitution", NULL, 0);
  1593.             return;
  1594.         }
  1595.         zsfree(hsubr);
  1596.         for (tt = hsubl; *tt; tt++)
  1597.             if (INULL(*tt))
  1598.             chuck(tt--);
  1599.         untokenize(hsubl);
  1600.         for (tt = hsubr = ztrdup(ptr2); *tt; tt++)
  1601.             if (INULL(*tt))
  1602.             chuck(tt--);
  1603.         ptr2[-1] = del;
  1604.         if (sav)
  1605.             ptr3[-1] = sav;
  1606.         *ptr = ptr3 - 1;
  1607.         break;
  1608.  
  1609.         case '&':
  1610.         c = 's';
  1611.         break;
  1612.  
  1613.         case 'g':
  1614.         (*ptr)++;
  1615.         gbal = 1;
  1616.         break;
  1617.  
  1618.         case 'w':
  1619.         wall = 1;
  1620.         (*ptr)++;
  1621.         break;
  1622.         case 'W':
  1623.         wall = 1;
  1624.         (*ptr)++;
  1625.         ptr1 = get_strarg(ptr2 = *ptr);
  1626.         if ((sav = *ptr1))
  1627.             *ptr1 = '\0';
  1628.         sep = dupstring(ptr2 + 1);
  1629.         if (sav)
  1630.             *ptr1 = sav;
  1631.         *ptr = ptr1 + 1;
  1632.         c = '\0';
  1633.         break;
  1634.  
  1635.         case 'f':
  1636.         rec = -1;
  1637.         (*ptr)++;
  1638.         break;
  1639.         case 'F':
  1640.         rec = get_intarg(ptr);
  1641.         (*ptr)++;
  1642.         break;
  1643.         default:
  1644.         *ptr = lptr;
  1645.         return;
  1646.         }
  1647.     }
  1648.     (*ptr)++;
  1649.     if (!c) {
  1650.         *ptr = lptr;
  1651.         return;
  1652.     }
  1653.     if (rec < 0)
  1654.         test = dupstring(*str);
  1655.  
  1656.     while (rec--) {
  1657.         if (wall) {
  1658.         al = 0;
  1659.         all = NULL;
  1660.         for (t = e = *str; (tt = findword(&e, sep));) {
  1661.             tc = *e;
  1662.             *e = '\0';
  1663.             copy = dupstring(tt);
  1664.             *e = tc;
  1665.             switch (c) {
  1666.             case 'h':
  1667.             remtpath(©);
  1668.             break;
  1669.             case 'r':
  1670.             remtext(©);
  1671.             break;
  1672.             case 'e':
  1673.             rembutext(©);
  1674.             break;
  1675.             case 't':
  1676.             remlpaths(©);
  1677.             break;
  1678.             case 'l':
  1679.             downcase(©);
  1680.             break;
  1681.             case 'u':
  1682.             upcase(©);
  1683.             break;
  1684.             case 's':
  1685.             if (hsubl && hsubr)
  1686.                 subst(©, hsubl, hsubr, gbal);
  1687.             break;
  1688.             }
  1689.             tc = *tt;
  1690.             *tt = '\0';
  1691.             nl = al + strlen(t) + strlen(copy);
  1692.             ptr1 = tmp = (char *)halloc(nl + 1);
  1693.             if (all)
  1694.             for (ptr2 = all; *ptr2;)
  1695.                 *ptr1++ = *ptr2++;
  1696.             for (ptr2 = t; *ptr2;)
  1697.             *ptr1++ = *ptr2++;
  1698.             *tt = tc;
  1699.             for (ptr2 = copy; *ptr2;)
  1700.             *ptr1++ = *ptr2++;
  1701.             *ptr1 = '\0';
  1702.             al = nl;
  1703.             all = tmp;
  1704.             t = e;
  1705.         }
  1706.         *str = all;
  1707.  
  1708.         } else {
  1709.         switch (c) {
  1710.         case 'h':
  1711.             remtpath(str);
  1712.             break;
  1713.         case 'r':
  1714.             remtext(str);
  1715.             break;
  1716.         case 'e':
  1717.             rembutext(str);
  1718.             break;
  1719.         case 't':
  1720.             remlpaths(str);
  1721.             break;
  1722.         case 'l':
  1723.             downcase(str);
  1724.             break;
  1725.         case 'u':
  1726.             upcase(str);
  1727.             break;
  1728.         case 's':
  1729.             if (hsubl && hsubr) {
  1730.             char *oldstr = *str;
  1731.  
  1732.             subst(str, hsubl, hsubr, gbal);
  1733.             if (*str != oldstr) {
  1734.                 *str = dupstring(oldstr = *str);
  1735.                 zsfree(oldstr);
  1736.             }
  1737.             }
  1738.             break;
  1739.         }
  1740.         }
  1741.         if (rec < 0) {
  1742.         if (!strcmp(test, *str))
  1743.             rec = 0;
  1744.         else
  1745.             test = dupstring(*str);
  1746.         }
  1747.     }
  1748.     }
  1749. }
  1750.  
  1751. /* get a directory stack entry */
  1752.  
  1753. /**/
  1754. char *
  1755. dstackent(char ch, int val)
  1756. {
  1757.     int backwards;
  1758.     LinkNode end=(LinkNode)dirstack, n;
  1759.  
  1760.     backwards = ch == (isset(PUSHDMINUS) ? '+' : '-');
  1761.     if(!backwards && !val--)
  1762.     return pwd;
  1763.     if (backwards)
  1764.     for (n=lastnode(dirstack); n != end && val; val--, n=prevnode(n));
  1765.     else
  1766.     for (end=NULL, n=firstnode(dirstack); n && val; val--, n=nextnode(n));
  1767.     if (n == end) {
  1768.     if (isset(NOMATCH))
  1769.         zerr("not enough directory stack entries.", NULL, 0);
  1770.     return NULL;
  1771.     }
  1772.     return (char *)getdata(n);
  1773. }
  1774.