zsh/Src/subst.c

2324 lines
50 KiB
C
Raw Normal View History

1999-04-15 20:05:38 +02:00
/*
* subst.c - various substitutions
*
* This file is part of zsh, the Z shell.
*
* Copyright (c) 1992-1997 Paul Falstad
* All rights reserved.
*
* Permission is hereby granted, without written agreement and without
* license or royalty fees, to use, copy, modify, and distribute this
* software and to distribute modified versions of this software for any
* purpose, provided that the above copyright notice and the following
* two paragraphs appear in all copies of this software.
*
* In no event shall Paul Falstad or the Zsh Development Group be liable
* to any party for direct, indirect, special, incidental, or consequential
* damages arising out of the use of this software and its documentation,
* even if Paul Falstad and the Zsh Development Group have been advised of
* the possibility of such damage.
*
* Paul Falstad and the Zsh Development Group specifically disclaim any
* warranties, including, but not limited to, the implied warranties of
* merchantability and fitness for a particular purpose. The software
* provided hereunder is on an "as is" basis, and Paul Falstad and the
* Zsh Development Group have no obligation to provide maintenance,
* support, updates, enhancements, or modifications.
*
*/
#include "zsh.mdh"
#include "subst.pro"
/**/
char nulstring[] = {Nularg, '\0'};
/* Do substitutions before fork. These are:
* - Process substitution: <(...), >(...), =(...)
* - Parameter substitution
* - Command substitution
* Followed by
* - Quote removal
* - Brace expansion
* - Tilde and equals substitution
*
* PF_* flags are defined in zsh.h
1999-04-15 20:05:38 +02:00
*/
/**/
mod_export void
1999-04-15 20:05:38 +02:00
prefork(LinkList list, int flags)
{
LinkNode node, stop = 0;
int keep = 0, asssub = (flags & PF_TYPESET) && isset(KSHTYPESET);
1999-04-15 20:05:38 +02:00
queue_signals();
1999-04-15 20:05:38 +02:00
for (node = firstnode(list); node; incnode(node)) {
char *str, c;
1999-04-15 20:05:38 +02:00
str = (char *)getdata(node);
if (((c = *str) == Inang || c == Outang || c == Equals) &&
1999-04-15 20:05:38 +02:00
str[1] == Inpar) {
if (c == Inang || c == Outang)
1999-04-15 20:05:38 +02:00
setdata(node, (void *) getproc(str)); /* <(...) or >(...) */
else
setdata(node, (void *) getoutputfile(str)); /* =(...) */
if (!getdata(node)) {
unqueue_signals();
1999-04-15 20:05:38 +02:00
return;
}
1999-04-15 20:05:38 +02:00
} else {
if (isset(SHFILEEXPANSION))
filesub((char **)getaddrdata(node),
flags & (PF_TYPESET|PF_ASSIGN));
2001-06-12 12:34:57 +02:00
if (!(node = stringsubst(list, node, flags & PF_SINGLE, asssub))) {
unqueue_signals();
1999-04-15 20:05:38 +02:00
return;
}
1999-04-15 20:05:38 +02:00
}
}
for (node = firstnode(list); node; incnode(node)) {
if (node == stop)
keep = 0;
1999-04-15 20:05:38 +02:00
if (*(char *)getdata(node)) {
remnulargs(getdata(node));
if (unset(IGNOREBRACES) && !(flags & PF_SINGLE)) {
if (!keep)
stop = nextnode(node);
while (hasbraces(getdata(node))) {
keep = 1;
1999-04-15 20:05:38 +02:00
xpandbraces(list, &node);
}
}
1999-04-15 20:05:38 +02:00
if (unset(SHFILEEXPANSION))
filesub((char **)getaddrdata(node),
flags & (PF_TYPESET|PF_ASSIGN));
} else if (!(flags & PF_SINGLE) && !keep)
1999-04-15 20:05:38 +02:00
uremnode(list, node);
if (errflag) {
unqueue_signals();
1999-04-15 20:05:38 +02:00
return;
}
1999-04-15 20:05:38 +02:00
}
unqueue_signals();
1999-04-15 20:05:38 +02:00
}
/**/
static LinkNode
2001-06-12 12:34:57 +02:00
stringsubst(LinkList list, LinkNode node, int ssub, int asssub)
1999-04-15 20:05:38 +02:00
{
int qt;
char *str3 = (char *)getdata(node);
char *str = str3, c;
1999-04-15 20:05:38 +02:00
while (!errflag && (c = *str)) {
if ((qt = c == Qstring) || c == String) {
if ((c = str[1]) == Inpar) {
if (!qt)
mult_isarr = 1;
1999-04-15 20:05:38 +02:00
str++;
goto comsub;
} else if (c == Inbrack) {
1999-04-15 20:05:38 +02:00
/* $[...] */
char *str2 = str;
str2++;
if (skipparens(Inbrack, Outbrack, &str2)) {
zerr("closing bracket missing", NULL, 0);
return NULL;
}
str2[-1] = *str = '\0';
str = arithsubst(str + 2, &str3, str2);
setdata(node, (void *) str3);
continue;
} else if (c == Snull) {
1999-04-15 20:05:38 +02:00
str = getkeystring(str, NULL, 4, NULL);
continue;
} else {
node = paramsubst(list, node, &str, qt, ssub);
if (errflag || !node)
return NULL;
str3 = (char *)getdata(node);
continue;
}
} else if ((qt = c == Qtick) || (c == Tick ? (mult_isarr = 1) : 0))
1999-04-15 20:05:38 +02:00
comsub: {
LinkList pl;
char *s, *str2 = str;
char endchar;
int l1, l2;
if (c == Inpar) {
1999-04-15 20:05:38 +02:00
endchar = Outpar;
str[-1] = '\0';
#ifdef DEBUG
1999-04-15 20:05:38 +02:00
if (skipparens(Inpar, Outpar, &str))
dputs("BUG: parse error in command substitution");
#else
skipparens(Inpar, Outpar, &str);
#endif
1999-04-15 20:05:38 +02:00
str--;
} else {
endchar = c;
1999-04-15 20:05:38 +02:00
*str = '\0';
while (*++str != endchar)
DPUTS(!*str, "BUG: parse error in command substitution");
}
*str++ = '\0';
if (endchar == Outpar && str2[1] == '(' && str[-2] == ')') {
/* Math substitution of the form $((...)) */
str[-2] = '\0';
str = arithsubst(str2 + 2, &str3, str);
1999-04-15 20:05:38 +02:00
setdata(node, (void *) str3);
continue;
}
/* It is a command substitution, which will be parsed again *
* by the lexer, so we untokenize it first, but we cannot use *
* untokenize() since in the case of `...` some Bnulls should *
* be left unchanged. Note that the lexer doesn't tokenize *
* the body of a command substitution so if there are some *
* tokens here they are from a ${(e)~...} substitution. */
for (str = str2; (c = *++str); )
if (itok(c) && c != Nularg &&
!(endchar != Outpar && c == Bnull &&
1999-04-15 20:05:38 +02:00
(str[1] == '$' || str[1] == '\\' || str[1] == '`' ||
(qt && str[1] == '"'))))
*str = ztokens[c - Pound];
1999-04-15 20:05:38 +02:00
str++;
if (!(pl = getoutput(str2 + 1, qt || ssub))) {
zerr("parse error in command substitution", NULL, 0);
return NULL;
}
if (endchar == Outpar)
str2--;
if (!(s = (char *) ugetnode(pl))) {
str = strcpy(str2, str);
continue;
}
if (!qt && ssub && isset(GLOBSUBST))
tokenize(s);
l1 = str2 - str3;
l2 = strlen(s);
if (nonempty(pl)) {
LinkNode n = lastnode(pl);
str2 = (char *) hcalloc(l1 + l2 + 1);
1999-04-15 20:05:38 +02:00
strcpy(str2, str3);
strcpy(str2 + l1, s);
setdata(node, str2);
insertlinklist(pl, node, list);
s = (char *) getdata(node = n);
l1 = 0;
l2 = strlen(s);
}
str2 = (char *) hcalloc(l1 + l2 + strlen(str) + 1);
1999-04-15 20:05:38 +02:00
if (l1)
strcpy(str2, str3);
strcpy(str2 + l1, s);
str = strcpy(str2 + l1 + l2, str);
str3 = str2;
setdata(node, str3);
continue;
2001-06-12 12:34:57 +02:00
} else if (asssub && ((c == '=') || c == Equals) && str != str3) {
/*
* We are in a normal argument which looks like an assignment
* and is to be treated like one, with no word splitting.
*/
ssub = 1;
1999-04-15 20:05:38 +02:00
}
str++;
}
return errflag ? NULL : node;
}
/**/
mod_export void
globlist(LinkList list, int nountok)
1999-04-15 20:05:38 +02:00
{
LinkNode node, next;
badcshglob = 0;
for (node = firstnode(list); !errflag && node; node = next) {
next = nextnode(node);
zglob(list, node, nountok);
1999-04-15 20:05:38 +02:00
}
if (badcshglob == 1)
zerr("no match", NULL, 0);
}
/* perform substitution on a single word */
/**/
mod_export void
1999-04-15 20:05:38 +02:00
singsub(char **s)
{
int omi = mult_isarr;
local_list1(foo);
1999-04-15 20:05:38 +02:00
init_list1(foo, *s);
prefork(&foo, PF_SINGLE);
mult_isarr = omi;
1999-04-15 20:05:38 +02:00
if (errflag)
return;
*s = (char *) ugetnode(&foo);
DPUTS(nonempty(&foo), "BUG: singsub() produced more than one word!");
1999-04-15 20:05:38 +02:00
}
/* Perform substitution on a single word. Unlike with singsub, the *
* result can have more than one word. A single word result is stored *
1999-04-15 20:05:38 +02:00
* in *s and *isarr is set to zero; otherwise *isarr is set to 1 and *
* the result is stored in *a. If `a' is zero a multiple word result is *
* joined using sep or the IFS parameter if sep is zero and the result *
* is returned in *s. The return value is true iff the expansion *
* resulted in an empty list. *
* The mult_isarr variable is used by paramsubst() to tell if it yields *
* an array. */
/**/
static int mult_isarr;
1999-04-15 20:05:38 +02:00
/**/
static int
multsub(char **s, char ***a, int *isarr, char *sep)
{
int l, omi = mult_isarr;
1999-04-15 20:05:38 +02:00
char **r, **p;
local_list1(foo);
1999-04-15 20:05:38 +02:00
mult_isarr = 0;
init_list1(foo, *s);
prefork(&foo, 0);
1999-04-15 20:05:38 +02:00
if (errflag) {
if (isarr)
*isarr = 0;
mult_isarr = omi;
1999-04-15 20:05:38 +02:00
return 0;
}
if ((l = countlinknodes(&foo))) {
p = r = hcalloc((l + 1) * sizeof(char*));
while (nonempty(&foo))
*p++ = (char *)ugetnode(&foo);
1999-04-15 20:05:38 +02:00
*p = NULL;
if (a && mult_isarr) {
1999-04-15 20:05:38 +02:00
*a = r;
*isarr = SCANPM_MATCHMANY;
mult_isarr = omi;
1999-04-15 20:05:38 +02:00
return 0;
}
*s = sepjoin(r, NULL, 1);
mult_isarr = omi;
if (isarr)
*isarr = 0;
1999-04-15 20:05:38 +02:00
return 0;
}
if (l)
*s = (char *) ugetnode(&foo);
1999-04-15 20:05:38 +02:00
else
*s = dupstring("");
if (isarr)
*isarr = 0;
mult_isarr = omi;
1999-04-15 20:05:38 +02:00
return !l;
}
/*
* ~, = subs: assign & PF_TYPESET => typeset or magic equals
* assign & PF_ASSIGN => normal assignment
*/
1999-04-15 20:05:38 +02:00
/**/
mod_export void
1999-04-15 20:05:38 +02:00
filesub(char **namptr, int assign)
{
2000-06-17 21:51:51 +02:00
char *eql = NULL, *sub = NULL, *str, *ptr;
1999-04-15 20:05:38 +02:00
int len;
filesubstr(namptr, assign);
if (!assign)
return;
if (assign & PF_TYPESET) {
2000-06-17 21:51:51 +02:00
if ((*namptr)[1] && (eql = sub = strchr(*namptr + 1, Equals))) {
1999-04-15 20:05:38 +02:00
str = sub + 1;
if ((sub[1] == Tilde || sub[1] == Equals) && filesubstr(&str, assign)) {
sub[1] = '\0';
*namptr = dyncat(*namptr, str);
}
} else
return;
}
1999-04-15 20:05:38 +02:00
ptr = *namptr;
while ((sub = strchr(ptr, ':'))) {
str = sub + 1;
len = sub - *namptr;
2000-06-17 21:51:51 +02:00
if (sub > eql &&
(sub[1] == Tilde || sub[1] == Equals) &&
filesubstr(&str, assign)) {
1999-04-15 20:05:38 +02:00
sub[1] = '\0';
*namptr = dyncat(*namptr, str);
}
ptr = *namptr + len + 1;
}
}
/**/
mod_export int
1999-04-15 20:05:38 +02:00
filesubstr(char **namptr, int assign)
{
#define isend(c) ( !(c) || (c)=='/' || (c)==Inpar || (assign && (c)==':') )
#define isend2(c) ( !(c) || (c)==Inpar || (assign && (c)==':') )
char *str = *namptr;
if (*str == Tilde && str[1] != '=' && str[1] != Equals) {
char *ptr;
int val;
val = zstrtol(str + 1, &ptr, 10);
if (isend(str[1])) { /* ~ */
*namptr = dyncat(home, str + 1);
return 1;
} else if (str[1] == '+' && isend(str[2])) { /* ~+ */
*namptr = dyncat(pwd, str + 2);
return 1;
} else if (str[1] == '-' && isend(str[2])) { /* ~- */
char *tmp;
*namptr = dyncat((tmp = oldpwd) ? tmp : pwd, str + 2);
return 1;
} else if (!inblank(str[1]) && isend(*ptr) &&
(!idigit(str[1]) || (ptr - str < 4))) {
char *ds;
if (val < 0)
val = -val;
ds = dstackent(str[1], val);
if (!ds)
return 0;
*namptr = dyncat(ds, ptr);
return 1;
} else if (iuser(str[1])) { /* ~foo */
char *ptr, *hom, save;
for (ptr = ++str; *ptr && iuser(*ptr); ptr++);
save = *ptr;
if (!isend(save))
return 0;
*ptr = 0;
if (!(hom = getnameddir(str))) {
if (isset(NOMATCH))
zerr("no such user or named directory: %s", str, 0);
*ptr = save;
return 0;
}
*ptr = save;
*namptr = dyncat(hom, ptr);
return 1;
}
} else if (*str == Equals && isset(EQUALS) && str[1]) { /* =foo */
char sav, *pp, *cnam;
for (pp = str + 1; !isend2(*pp); pp++);
sav = *pp;
*pp = 0;
if (!(cnam = findcmd(str + 1, 1))) {
if (isset(NOMATCH))
zerr("%s not found", str + 1, 0);
return 0;
1999-04-15 20:05:38 +02:00
}
*namptr = dupstring(cnam);
if (sav) {
*pp = sav;
*namptr = dyncat(*namptr, pp);
}
return 1;
}
return 0;
#undef isend
#undef isend2
}
/**/
static char *
strcatsub(char **d, char *pb, char *pe, char *src, int l, char *s, int glbsub,
int copied)
1999-04-15 20:05:38 +02:00
{
char *dest;
1999-04-15 20:05:38 +02:00
int pl = pe - pb;
if (!pl && (!s || !*s)) {
*d = dest = (copied ? src : dupstring(src));
if (glbsub)
tokenize(dest);
} else {
*d = dest = hcalloc(pl + l + (s ? strlen(s) : 0) + 1);
strncpy(dest, pb, pl);
dest += pl;
strcpy(dest, src);
if (glbsub)
tokenize(dest);
dest += l;
if (s)
strcpy(dest, s);
}
1999-04-15 20:05:38 +02:00
return dest;
}
typedef int (*CompareFn) _((const void *, const void *));
/**/
int
strpcmp(const void *a, const void *b)
{
#ifdef HAVE_STRCOLL
return strcoll(*(char **)a, *(char **)b);
#else
return strcmp(*(char **)a, *(char **)b);
#endif
}
/**/
int
invstrpcmp(const void *a, const void *b)
{
#ifdef HAVE_STRCOLL
return -strcoll(*(char **)a, *(char **)b);
#else
return -strcmp(*(char **)a, *(char **)b);
#endif
}
/**/
int
cstrpcmp(const void *a, const void *b)
{
#ifdef HAVE_STRCOLL
VARARR(char, c, strlen(*(char **) a) + 1);
VARARR(char, d, strlen(*(char **) b) + 1);
char *s, *t;
int cmp;
for (s = *(char **) a, t = c; (*t++ = tulower(*s++)););
for (s = *(char **) b, t = d; (*t++ = tulower(*s++)););
cmp = strcoll(c, d);
return cmp;
#else
char *c = *(char **)a, *d = *(char **)b;
for (; *c && tulower(*c) == tulower(*d); c++, d++);
return (int)STOUC(tulower(*c)) - (int)STOUC(tulower(*d));
#endif
}
/**/
int
invcstrpcmp(const void *a, const void *b)
{
#ifdef HAVE_STRCOLL
VARARR(char, c, strlen(*(char **) a) + 1);
VARARR(char, d, strlen(*(char **) b) + 1);
char *s, *t;
int cmp;
for (s = *(char **) a, t = c; (*t++ = tulower(*s++)););
for (s = *(char **) b, t = d; (*t++ = tulower(*s++)););
cmp = strcoll(c, d);
return -cmp;
#else
char *c = *(char **)a, *d = *(char **)b;
for (; *c && tulower(*c) == tulower(*d); c++, d++);
return (int)STOUC(tulower(*d)) - (int)STOUC(tulower(*c));
#endif
}
/**/
static char *
dopadding(char *str, int prenum, int postnum, char *preone, char *postone, char *premul, char *postmul)
{
char def[3], *ret, *t, *r;
int ls, ls2, lpreone, lpostone, lpremul, lpostmul, lr, f, m, c, cc;
def[0] = *ifs ? *ifs : ' ';
def[1] = *ifs == Meta ? ifs[1] ^ 32 : '\0';
def[2] = '\0';
if (preone && !*preone)
preone = def;
if (postone && !*postone)
postone = def;
if (!premul || !*premul)
premul = def;
if (!postmul || !*postmul)
postmul = def;
ls = strlen(str);
lpreone = preone ? strlen(preone) : 0;
lpostone = postone ? strlen(postone) : 0;
lpremul = strlen(premul);
lpostmul = strlen(postmul);
lr = prenum + postnum;
if (lr == ls)
return str;
r = ret = (char *)zhalloc(lr + 1);
1999-04-15 20:05:38 +02:00
if (prenum) {
if (postnum) {
ls2 = ls / 2;
f = prenum - ls2;
if (f <= 0)
for (str -= f, c = prenum; c--; *r++ = *str++);
else {
if (f <= lpreone)
for (c = f, t = preone + lpreone - f; c--; *r++ = *t++);
else {
f -= lpreone;
if ((m = f % lpremul))
for (c = m, t = premul + lpremul - m; c--; *r++ = *t++);
for (cc = f / lpremul; cc--;)
for (c = lpremul, t = premul; c--; *r++ = *t++);
for (c = lpreone; c--; *r++ = *preone++);
}
for (c = ls2; c--; *r++ = *str++);
}
ls2 = ls - ls2;
f = postnum - ls2;
if (f <= 0)
for (c = postnum; c--; *r++ = *str++);
else {
for (c = ls2; c--; *r++ = *str++);
if (f <= lpostone)
for (c = f; c--; *r++ = *postone++);
else {
f -= lpostone;
for (c = lpostone; c--; *r++ = *postone++);
for (cc = f / lpostmul; cc--;)
for (c = lpostmul, t = postmul; c--; *r++ = *t++);
if ((m = f % lpostmul))
for (; m--; *r++ = *postmul++);
}
}
} else {
f = prenum - ls;
if (f <= 0)
for (c = prenum, str -= f; c--; *r++ = *str++);
else {
if (f <= lpreone)
for (c = f, t = preone + lpreone - f; c--; *r++ = *t++);
else {
f -= lpreone;
if ((m = f % lpremul))
for (c = m, t = premul + lpremul - m; c--; *r++ = *t++);
for (cc = f / lpremul; cc--;)
for (c = lpremul, t = premul; c--; *r++ = *t++);
for (c = lpreone; c--; *r++ = *preone++);
}
for (c = ls; c--; *r++ = *str++);
}
}
} else if (postnum) {
f = postnum - ls;
if (f <= 0)
for (c = postnum; c--; *r++ = *str++);
else {
for (c = ls; c--; *r++ = *str++);
if (f <= lpostone)
for (c = f; c--; *r++ = *postone++);
else {
f -= lpostone;
for (c = lpostone; c--; *r++ = *postone++);
for (cc = f / lpostmul; cc--;)
for (c = lpostmul, t = postmul; c--; *r++ = *t++);
if ((m = f % lpostmul))
for (; m--; *r++ = *postmul++);
}
}
}
*r = '\0';
return ret;
}
/**/
char *
get_strarg(char *s)
{
char t = *s++;
if (!t)
return s - 1;
switch (t) {
case '(':
t = ')';
break;
case '[':
t = ']';
break;
case '{':
t = '}';
break;
case '<':
t = '>';
break;
case Inpar:
t = Outpar;
break;
case Inang:
t = Outang;
break;
case Inbrace:
t = Outbrace;
break;
case Inbrack:
t = Outbrack;
break;
}
while (*s && *s != t)
s++;
return s;
}
/**/
static int
get_intarg(char **s)
{
char *t = get_strarg(*s + 1);
char *p, sav;
zlong ret;
1999-04-15 20:05:38 +02:00
if (!*t)
return -1;
sav = *t;
*t = '\0';
p = dupstring(*s + 2);
*s = t;
*t = sav;
if (parsestr(p))
return -1;
singsub(&p);
if (errflag)
return -1;
ret = mathevali(p);
1999-04-15 20:05:38 +02:00
if (errflag)
return -1;
if (ret < 0)
ret = -ret;
return ret < 0 ? -ret : ret;
}
/* Parsing for the (e) flag. */
static int
subst_parse_str(char **sp, int single, int err)
{
char *s;
*sp = s = dupstring(*sp);
if (!(err ? parsestr(s) : parsestrnoerr(s))) {
if (!single) {
int qt = 0;
for (; *s; s++)
if (!qt) {
if (*s == Qstring)
*s = String;
else if (*s == Qtick)
*s = Tick;
} else if (*s == Dnull)
qt = !qt;
}
return 0;
}
return 1;
}
1999-04-15 20:05:38 +02:00
/* parameter substitution */
#define isstring(c) ((c) == '$' || (char)(c) == String || (char)(c) == Qstring)
#define isbrack(c) ((c) == '[' || (char)(c) == Inbrack)
/**/
LinkNode
paramsubst(LinkList l, LinkNode n, char **str, int qt, int ssub)
{
char *aptr = *str, c, cc;
1999-04-15 20:05:38 +02:00
char *s = aptr, *fstr, *idbeg, *idend, *ostr = (char *) getdata(n);
int colf; /* != 0 means we found a colon after the name */
int isarr = 0;
int plan9 = isset(RCEXPANDPARAM);
int globsubst = isset(GLOBSUBST);
int getlen = 0;
int whichlen = 0;
int chkset = 0;
int vunset = 0;
int wantt = 0;
1999-04-15 20:05:38 +02:00
int spbreak = isset(SHWORDSPLIT) && !ssub && !qt;
char *val = NULL, **aval = NULL;
unsigned int fwidth = 0;
struct value vbuf;
Value v = NULL;
1999-04-15 20:05:38 +02:00
int flags = 0;
int flnum = 0;
int sortit = 0, casind = 0;
int unique = 0;
1999-04-15 20:05:38 +02:00
int casmod = 0;
int quotemod = 0, quotetype = 0, quoteerr = 0;
int visiblemod = 0;
int shsplit = 0;
1999-04-15 20:05:38 +02:00
char *sep = NULL, *spsep = NULL;
char *premul = NULL, *postmul = NULL, *preone = NULL, *postone = NULL;
char *replstr = NULL; /* replacement string for /orig/repl */
zlong prenum = 0, postnum = 0;
1999-04-15 20:05:38 +02:00
int copied = 0;
int arrasg = 0;
int eval = 0;
int aspar = 0;
int presc = 0;
1999-04-15 20:05:38 +02:00
int nojoin = 0;
char inbrace = 0; /* != 0 means ${...}, otherwise $... */
char hkeys = 0;
char hvals = 0;
int subexp;
1999-04-15 20:05:38 +02:00
*s++ = '\0';
if (!ialnum(c = *s) && c != '#' && c != Pound && c != '-' &&
c != '!' && c != '$' && c != String && c != Qstring &&
c != '?' && c != Quest && c != '_' &&
c != '*' && c != Star && c != '@' && c != '{' &&
c != Inbrace && c != '=' && c != Equals && c != Hat &&
c != '^' && c != '~' && c != Tilde && c != '+') {
1999-04-15 20:05:38 +02:00
s[-1] = '$';
*str = s;
return n;
}
DPUTS(c == '{', "BUG: inbrace == '{' in paramsubst()");
if (c == Inbrace) {
1999-04-15 20:05:38 +02:00
inbrace = 1;
s++;
if ((c = *s) == '!' && s[1] != Outbrace && emulation == EMULATE_KSH) {
hkeys = SCANPM_WANTKEYS;
s++;
} else if (c == '(' || c == Inpar) {
1999-04-15 20:05:38 +02:00
char *t, sav;
int tt = 0;
zlong num;
1999-04-15 20:05:38 +02:00
int escapes = 0;
int klen;
#define UNTOK(C) (itok(C) ? ztokens[(C) - Pound] : (C))
1999-04-15 20:05:38 +02:00
#define UNTOK_AND_ESCAPE(X) {\
untokenize(X = dupstring(s + 1));\
if (escapes) {\
X = getkeystring(X, &klen, 3, NULL);\
X = metafy(X, klen, META_HREALLOC);\
}\
}
for (s++; (c = *s) != ')' && c != Outpar; s++, tt = 0) {
switch (c) {
1999-04-15 20:05:38 +02:00
case ')':
case Outpar:
break;
case 'A':
++arrasg;
1999-04-15 20:05:38 +02:00
break;
case '@':
nojoin = 1;
break;
case 'M':
flags |= SUB_MATCH;
1999-04-15 20:05:38 +02:00
break;
case 'R':
flags |= SUB_REST;
1999-04-15 20:05:38 +02:00
break;
case 'B':
flags |= SUB_BIND;
1999-04-15 20:05:38 +02:00
break;
case 'E':
flags |= SUB_EIND;
1999-04-15 20:05:38 +02:00
break;
case 'N':
flags |= SUB_LEN;
1999-04-15 20:05:38 +02:00
break;
case 'S':
flags |= SUB_SUBSTR;
1999-04-15 20:05:38 +02:00
break;
case 'I':
flnum = get_intarg(&s);
if (flnum < 0)
goto flagerr;
break;
case 'L':
casmod = 2;
break;
case 'U':
casmod = 1;
break;
case 'C':
casmod = 3;
break;
case 'o':
sortit = 1;
break;
case 'O':
sortit = 2;
break;
case 'i':
casind = 1;
break;
case 'V':
visiblemod++;
break;
case 'q':
quotemod++, quotetype++;
break;
case 'Q':
quotemod--;
break;
case 'X':
quoteerr = 1;
break;
1999-04-15 20:05:38 +02:00
case 'e':
eval = 1;
break;
case 'P':
aspar = 1;
break;
1999-04-15 20:05:38 +02:00
case 'c':
whichlen = 1;
break;
case 'w':
whichlen = 2;
break;
case 'W':
whichlen = 3;
break;
case 'f':
spsep = "\n";
break;
case 'F':
sep = "\n";
break;
case 's':
tt = 1;
/* fall through */
case 'j':
t = get_strarg(++s);
if (*t) {
sav = *t;
*t = '\0';
if (tt)
UNTOK_AND_ESCAPE(spsep)
else
UNTOK_AND_ESCAPE(sep)
*t = sav;
s = t;
} else
goto flagerr;
break;
case 'l':
tt = 1;
/* fall through */
case 'r':
sav = s[1];
num = get_intarg(&s);
if (num < 0)
goto flagerr;
if (tt)
prenum = num;
else
postnum = num;
if (UNTOK(s[1]) != UNTOK(sav))
1999-04-15 20:05:38 +02:00
break;
t = get_strarg(++s);
if (!*t)
goto flagerr;
sav = *t;
*t = '\0';
if (tt)
UNTOK_AND_ESCAPE(premul)
else
UNTOK_AND_ESCAPE(postmul)
*t = sav;
sav = *s;
s = t + 1;
if (UNTOK(*s) != UNTOK(sav)) {
1999-04-15 20:05:38 +02:00
s--;
break;
}
t = get_strarg(s);
if (!*t)
goto flagerr;
sav = *t;
*t = '\0';
if (tt)
UNTOK_AND_ESCAPE(preone)
else
UNTOK_AND_ESCAPE(postone)
*t = sav;
s = t;
break;
case 'p':
escapes = 1;
break;
case 'k':
hkeys = SCANPM_WANTKEYS;
break;
case 'v':
hvals = SCANPM_WANTVALS;
break;
case 't':
wantt = 1;
break;
case '%':
presc++;
break;
case 'z':
shsplit = 1;
break;
case 'u':
unique = 1;
break;
1999-04-15 20:05:38 +02:00
default:
flagerr:
zerr("error in flags", NULL, 0);
return NULL;
}
}
s++;
}
}
if (sortit)
sortit += (casind << 1);
if (!premul)
premul = " ";
if (!postmul)
postmul = " ";
for (;;) {
if ((c = *s) == '^' || c == Hat) {
if ((c = *++s) == '^' || c == Hat) {
1999-04-15 20:05:38 +02:00
plan9 = 0;
s++;
} else
plan9 = 1;
} else if ((c = *s) == '=' || c == Equals) {
if ((c = *++s) == '=' || c == Equals) {
1999-04-15 20:05:38 +02:00
spbreak = 0;
s++;
} else
2001-02-16 18:16:13 +01:00
spbreak = 2;
} else if ((c == '#' || c == Pound) &&
(iident(cc = s[1])
|| cc == '*' || cc == Star || cc == '@'
|| cc == '-' || (cc == ':' && s[2] == '-')
|| (isstring(cc) && (s[2] == Inbrace || s[2] == Inpar))))
1999-04-15 20:05:38 +02:00
getlen = 1 + whichlen, s++;
else if (c == '~' || c == Tilde) {
if ((c = *++s) == '~' || c == Tilde) {
1999-04-15 20:05:38 +02:00
globsubst = 0;
s++;
} else
globsubst = 1;
} else if (c == '+') {
if (iident(s[1]) || (aspar && isstring(s[1]) &&
(s[2] == Inbrace || s[2] == Inpar)))
1999-04-15 20:05:38 +02:00
chkset = 1, s++;
else if (!inbrace) {
*aptr = '$';
*str = aptr + 1;
return n;
} else {
zerr("bad substitution", NULL, 0);
return NULL;
}
} else if (inbrace && INULL(*s))
s++;
1999-04-15 20:05:38 +02:00
else
break;
}
globsubst = globsubst && !qt;
idbeg = s;
if ((subexp = (inbrace && s[-1] && isstring(*s) &&
(s[1] == Inbrace || s[1] == Inpar)))) {
1999-04-15 20:05:38 +02:00
int sav;
int quoted = *s == Qstring;
val = s++;
skipparens(*s, *s == Inpar ? Outpar : Outbrace, &s);
sav = *s;
*s = 0;
if (multsub(&val, (aspar ? NULL : &aval), &isarr, NULL) && quoted) {
1999-04-15 20:05:38 +02:00
isarr = -1;
aval = (char **) hcalloc(sizeof(char *));
aspar = 0;
} else if (aspar)
idbeg = val;
1999-04-15 20:05:38 +02:00
*s = sav;
while (INULL(*s))
s++;
1999-04-15 20:05:38 +02:00
v = (Value) NULL;
} else if (aspar) {
if ((v = fetchvalue(&vbuf, &s, 1, (qt ? SCANPM_DQUOTED : 0)))) {
val = idbeg = getstrvalue(v);
subexp = 1;
} else
vunset = 1;
}
if (!subexp || aspar) {
char *ov = val;
if (!(v = fetchvalue(&vbuf, (subexp ? &ov : &s),
(wantt ? -1 :
((unset(KSHARRAYS) || inbrace) ? 1 : -1)),
hkeys|hvals|
(arrasg ? SCANPM_ASSIGNING : 0)|
(qt ? SCANPM_DQUOTED : 0))) ||
(v->pm && (v->pm->flags & PM_UNSET)))
vunset = 1;
if (wantt) {
if (v && v->pm && !(v->pm->flags & PM_UNSET)) {
int f = v->pm->flags;
switch (PM_TYPE(f)) {
case PM_SCALAR: val = "scalar"; break;
case PM_ARRAY: val = "array"; break;
case PM_INTEGER: val = "integer"; break;
case PM_EFLOAT:
case PM_FFLOAT: val = "float"; break;
case PM_HASHED: val = "association"; break;
}
val = dupstring(val);
if (v->pm->level)
val = dyncat(val, "-local");
if (f & PM_LEFT)
val = dyncat(val, "-left");
if (f & PM_RIGHT_B)
val = dyncat(val, "-right_blanks");
if (f & PM_RIGHT_Z)
val = dyncat(val, "-right_zeros");
if (f & PM_LOWER)
val = dyncat(val, "-lower");
if (f & PM_UPPER)
val = dyncat(val, "-upper");
if (f & PM_READONLY)
val = dyncat(val, "-readonly");
if (f & PM_TAGGED)
val = dyncat(val, "-tag");
if (f & PM_EXPORTED)
val = dyncat(val, "-export");
if (f & PM_UNIQUE)
val = dyncat(val, "-unique");
if (f & PM_HIDE)
val = dyncat(val, "-hide");
2000-06-09 17:37:05 +02:00
if (f & PM_HIDE)
val = dyncat(val, "-hideval");
if (f & PM_SPECIAL)
val = dyncat(val, "-special");
vunset = 0;
} else
val = dupstring("");
v = NULL;
isarr = 0;
}
}
1999-04-15 20:05:38 +02:00
while (v || ((inbrace || (unset(KSHARRAYS) && vunset)) && isbrack(*s))) {
if (!v) {
Param pm;
char *os = s;
if (!isbrack(*s))
break;
if (vunset) {
val = dupstring("");
isarr = 0;
}
pm = createparam(nulstring, isarr ? PM_ARRAY : PM_SCALAR);
DPUTS(!pm, "BUG: parameter not created");
1999-04-15 20:05:38 +02:00
if (isarr)
pm->u.arr = aval;
else
pm->u.str = val;
v = (Value) hcalloc(sizeof *v);
v->isarr = isarr;
v->pm = pm;
v->end = -1;
if (getindex(&s, v, qt) || s == os)
1999-04-15 20:05:38 +02:00
break;
}
if ((isarr = v->isarr)) {
/* No way to get here with v->inv != 0, so getvaluearr() *
* is called by getarrvalue(); needn't test PM_HASHED. */
if (v->isarr == SCANPM_WANTINDEX) {
isarr = v->isarr = 0;
val = dupstring(v->pm->nam);
} else
aval = getarrvalue(v);
} else {
1999-04-15 20:05:38 +02:00
if (v->pm->flags & PM_ARRAY) {
int tmplen = arrlen(v->pm->gets.afn(v->pm));
if (v->start < 0)
v->start += tmplen + v->inv;
if (!v->inv && (v->start >= tmplen || v->start < 0))
1999-04-15 20:05:38 +02:00
vunset = 1;
}
if (!vunset) {
val = getstrvalue(v);
fwidth = v->pm->ct ? v->pm->ct : strlen(val);
switch (v->pm->flags & (PM_LEFT | PM_RIGHT_B | PM_RIGHT_Z)) {
char *t;
unsigned int t0;
case PM_LEFT:
case PM_LEFT | PM_RIGHT_Z:
t = val;
if (v->pm->flags & PM_RIGHT_Z)
while (*t == '0')
t++;
else
while (iblank(*t))
t++;
val = (char *) hcalloc(fwidth + 1);
1999-04-15 20:05:38 +02:00
val[fwidth] = '\0';
if ((t0 = strlen(t)) > fwidth)
t0 = fwidth;
memset(val, ' ', fwidth);
strncpy(val, t, t0);
break;
case PM_RIGHT_B:
case PM_RIGHT_Z:
case PM_RIGHT_Z | PM_RIGHT_B:
{
int zero = 1;
if (strlen(val) < fwidth) {
if (v->pm->flags & PM_RIGHT_Z) {
for (t = val; iblank(*t); t++);
if (!*t || !idigit(*t))
zero = 0;
}
t = (char *) hcalloc(fwidth + 1);
memset(t, (((v->pm->flags & PM_RIGHT_B) || !zero) ?
' ' : '0'), fwidth);
if ((t0 = strlen(val)) > fwidth)
t0 = fwidth;
strcpy(t + (fwidth - t0), val);
val = t;
} else {
t = (char *) hcalloc(fwidth + 1);
t[fwidth] = '\0';
strncpy(t, val + strlen(val) - fwidth, fwidth);
val = t;
}
1999-04-15 20:05:38 +02:00
}
break;
}
switch (v->pm->flags & (PM_LOWER | PM_UPPER)) {
char *t;
case PM_LOWER:
t = val;
for (; (c = *t); t++)
*t = tulower(c);
1999-04-15 20:05:38 +02:00
break;
case PM_UPPER:
t = val;
for (; (c = *t); t++)
*t = tuupper(c);
1999-04-15 20:05:38 +02:00
break;
}
}
}
v = NULL;
if (!inbrace)
break;
}
if (inbrace &&
(c = *s) != '-' && c != '+' && c != ':' && c != '%' && c != '/' &&
c != '=' && c != Equals &&
c != '#' && c != Pound &&
c != '?' && c != Quest &&
c != '}' && c != Outbrace) {
zerr("bad substitution", NULL, 0);
return NULL;
}
1999-04-15 20:05:38 +02:00
if (isarr) {
if (nojoin)
isarr = -1;
if (qt && !getlen && isarr > 0) {
val = sepjoin(aval, sep, 1);
1999-04-15 20:05:38 +02:00
isarr = 0;
}
}
idend = s;
if (inbrace)
while (INULL(*s))
s++;
1999-04-15 20:05:38 +02:00
if ((colf = *s == ':'))
s++;
/* fstr is to be the text following the substitution. If we have *
* braces, we look for it here, else we infer it later on. */
fstr = s;
if (inbrace) {
int bct;
for (bct = 1; (c = *fstr); fstr++) {
if (c == Inbrace)
1999-04-15 20:05:38 +02:00
bct++;
else if (c == Outbrace && !--bct)
1999-04-15 20:05:38 +02:00
break;
}
if (bct) {
noclosebrace:
zerr("closing brace expected", NULL, 0);
return NULL;
}
if (c)
1999-04-15 20:05:38 +02:00
*fstr++ = '\0';
}
/* Check for ${..?..} or ${..=..} or one of those. *
* Only works if the name is in braces. */
if (inbrace && ((c = *s) == '-' ||
c == '+' ||
c == ':' ||
c == '=' || c == Equals ||
c == '%' ||
c == '#' || c == Pound ||
c == '?' || c == Quest ||
c == '/')) {
1999-04-15 20:05:38 +02:00
if (!flnum)
flnum++;
if (c == '%')
flags |= SUB_END;
1999-04-15 20:05:38 +02:00
/* Check for ${..%%..} or ${..##..} */
if ((c == '%' || c == '#' || c == Pound) && c == s[1]) {
1999-04-15 20:05:38 +02:00
s++;
/* we have %%, not %, or ##, not # */
flags |= SUB_LONG;
1999-04-15 20:05:38 +02:00
}
s++;
if (s[-1] == '/') {
char *ptr;
/*
* previous flags are irrelevant, except for (S) which
* indicates shortest substring; else look for longest.
*/
flags = (flags & SUB_SUBSTR) ? 0 : SUB_LONG;
if ((c = *s) == '/') {
/* doubled, so replace all occurrences */
flags |= SUB_GLOBAL;
s++;
}
/* Check for anchored substitution */
if (c == '%') {
/* anchor at tail */
flags |= SUB_END;
s++;
} else if (c == '#' || c == Pound) {
/* anchor at head: this is the `normal' case in getmatch */
s++;
} else
flags |= SUB_SUBSTR;
/*
* Find the / marking the end of the search pattern.
* If there isn't one, we're just going to delete that,
* i.e. replace it with an empty string.
*
* This allows quotation of the slash with '\\/'. Why
* two? Well, for a non-quoted string we can check for
* Bnull+/, which is what you get from `\/', but inside
* double quotes the Bnull isn't there, so it's not
* consistent.
*/
for (ptr = s; (c = *ptr) && c != '/'; ptr++)
if (c == '\\' && ptr[1] == '/')
chuck(ptr);
replstr = (*ptr && ptr[1]) ? ptr+1 : "";
*ptr = '\0';
}
1999-04-15 20:05:38 +02:00
if (colf)
flags |= SUB_ALL;
/*
* With no special flags, i.e. just a # or % or whatever,
* the matched portion is removed and we keep the rest.
* We also want the rest when we're doing a substitution.
*/
if (!(flags & (SUB_MATCH|SUB_REST|SUB_BIND|SUB_EIND|SUB_LEN)))
flags |= SUB_REST;
1999-04-15 20:05:38 +02:00
if (colf && !vunset)
vunset = (isarr) ? !*aval : !*val || (*val == Nularg && !val[1]);
switch (s[-1]) {
case '+':
if (vunset) {
val = dupstring("");
copied = 1;
isarr = 0;
break;
}
vunset = 1;
/* Fall Through! */
case '-':
if (vunset) {
val = dupstring(s);
/*
* This is not good enough for sh emulation! Sh would
* split unquoted substrings, yet not split quoted ones
* (except according to $@ rules); but this leaves the
* unquoted substrings unsplit, and other code below
* for spbreak splits even within the quoted substrings.
*/
multsub(&val, (aspar ? NULL : &aval), &isarr, NULL);
1999-04-15 20:05:38 +02:00
copied = 1;
}
break;
case ':':
if (*s != '=' && *s != Equals)
goto noclosebrace;
vunset = 1;
s++;
/* Fall through */
case '=':
case Equals:
if (vunset) {
char sav = *idend;
int l;
*idend = '\0';
val = dupstring(s);
isarr = 0;
if (spsep || spbreak || !arrasg)
multsub(&val, NULL, NULL, sep);
else
multsub(&val, &aval, &isarr, NULL);
if (arrasg) {
char *arr[2], **t, **a, **p;
if (spsep || spbreak) {
aval = sepsplit(val, spsep, 0, 1);
1999-04-15 20:05:38 +02:00
isarr = 2;
l = arrlen(aval);
if (l && !*(aval[l-1]))
l--;
if (l && !**aval)
l--, t = aval + 1;
else
t = aval;
} else if (!isarr) {
if (!*val && arrasg > 1) {
arr[0] = NULL;
l = 0;
} else {
arr[0] = val;
arr[1] = NULL;
l = 1;
}
1999-04-15 20:05:38 +02:00
t = aval = arr;
} else
l = arrlen(aval), t = aval;
p = a = zalloc(sizeof(char *) * (l + 1));
while (l--) {
untokenize(*t);
*p++ = ztrdup(*t++);
}
*p++ = NULL;
if (arrasg > 1) {
Param pm = sethparam(idbeg, a);
if (pm)
aval = paramvalarr(pm->gets.hfn(pm), hkeys|hvals);
} else
setaparam(idbeg, a);
1999-04-15 20:05:38 +02:00
} else {
untokenize(val);
setsparam(idbeg, ztrdup(val));
}
*idend = sav;
copied = 1;
if (isarr) {
if (nojoin)
isarr = -1;
2001-02-16 18:16:13 +01:00
if (qt && !getlen && isarr > 0 && !spsep && spbreak < 2) {
val = sepjoin(aval, sep, 1);
isarr = 0;
}
2001-02-16 18:16:13 +01:00
sep = spsep = NULL;
spbreak = 0;
}
1999-04-15 20:05:38 +02:00
}
break;
case '?':
case Quest:
if (vunset) {
char *msg;
*idend = '\0';
msg = tricat(idbeg, ": ", *s ? s : "parameter not set");
zerr("%s", msg, 0);
zsfree(msg);
if (!interact)
exit(1);
return NULL;
}
break;
case '%':
case '#':
case Pound:
case '/':
/* This once was executed only `if (qt) ...'. But with that
* patterns in a expansion resulting from a ${(e)...} aren't
* tokenized even though this function thinks they are (it thinks
* they are because subst_parse_str() turns Qstring tokens
* into String tokens and for unquoted parameter expansions the
* lexer normally does tokenize patterns inside parameter
* expansions). */
{
int one = noerrs, oef = errflag, haserr;
if (!quoteerr)
noerrs = 1;
haserr = parse_subst_string(s);
noerrs = one;
if (!quoteerr) {
errflag = oef;
if (haserr)
tokenize(s);
} else if (haserr || errflag) {
1999-04-15 20:05:38 +02:00
zerr("parse error in ${...%c...} substitution",
NULL, s[-1]);
return NULL;
}
}
{
2000-05-08 10:58:37 +02:00
#if 0
/*
* This allows # and % to be at the start of
* a parameter in the substitution, which is
* a bit nasty, and can be done (although
* less efficiently) with anchors.
*/
char t = s[-1];
singsub(&s);
if (t == '/' && (flags & SUB_SUBSTR)) {
if ((c = *s) == '#' || c == '%') {
flags &= ~SUB_SUBSTR;
if (c == '%')
flags |= SUB_END;
s++;
} else if (c == '\\') {
s++;
}
}
#else
singsub(&s);
2000-05-08 10:58:37 +02:00
#endif
}
1999-04-15 20:05:38 +02:00
if (!vunset && isarr) {
getmatcharr(&aval, s, flags, flnum, replstr);
1999-04-15 20:05:38 +02:00
copied = 1;
} else {
if (vunset)
val = dupstring("");
getmatch(&val, s, flags, flnum, replstr);
1999-04-15 20:05:38 +02:00
copied = 1;
}
break;
}
} else { /* no ${...=...} or anything, but possible modifiers. */
if (chkset) {
val = dupstring(vunset ? "0" : "1");
isarr = 0;
} else if (vunset) {
if (unset(UNSET)) {
*idend = '\0';
zerr("%s: parameter not set", idbeg, 0);
return NULL;
}
val = dupstring("");
}
if (colf) {
s--;
if (unset(KSHARRAYS) || inbrace) {
if (!isarr)
modify(&val, &s);
else {
char *ss;
char **ap = aval;
char **pp = aval = (char **) hcalloc(sizeof(char *) *
(arrlen(aval) + 1));
1999-04-15 20:05:38 +02:00
while ((*pp = *ap++)) {
ss = s;
modify(pp++, &ss);
}
if (pp == aval) {
char *t = "";
ss = s;
modify(&t, &ss);
}
s = ss;
}
copied = 1;
1999-04-15 20:05:38 +02:00
if (inbrace && *s) {
if (*s == ':' && !imeta(s[1]))
zerr("unrecognized modifier `%c'", NULL, s[1]);
else
zerr("unrecognized modifier", NULL, 0);
return NULL;
}
}
}
if (!inbrace)
fstr = s;
}
if (errflag)
return NULL;
if (getlen) {
long len = 0;
char buf[14];
if (isarr) {
char **ctr;
int sl = sep ? ztrlen(sep) : 1;
if (getlen == 1)
for (ctr = aval; *ctr; ctr++, len++);
else if (getlen == 2) {
if (*aval)
for (len = -sl, ctr = aval;
len += sl + ztrlen(*ctr), *++ctr;);
}
else
for (ctr = aval;
*ctr;
len += wordcount(*ctr, spsep, getlen > 3), ctr++);
} else {
if (getlen < 3)
len = ztrlen(val);
else
len = wordcount(val, spsep, getlen > 3);
}
sprintf(buf, "%ld", len);
val = dupstring(buf);
isarr = 0;
}
mult_isarr = isarr;
1999-04-15 20:05:38 +02:00
if (isarr > 0 && !plan9 && (!aval || !aval[0])) {
val = dupstring("");
isarr = 0;
} else if (isarr && aval && aval[0] && !aval[1]) {
/* treat a one-element array as a scalar for purposes of *
* concatenation with surrounding text (some${param}thing) *
* and rc_expand_param handling. Note: mult_isarr (above) *
* propagates the true array type from nested expansions. */
1999-04-15 20:05:38 +02:00
val = aval[0];
isarr = 0;
}
/* ssub is true when we are called from singsub (via prefork).
* It means that we must join arrays and should not split words. */
if (ssub || spbreak || spsep || sep) {
if (isarr)
val = sepjoin(aval, sep, 1), isarr = 0;
1999-04-15 20:05:38 +02:00
if (!ssub && (spbreak || spsep)) {
aval = sepsplit(val, spsep, 0, 1);
1999-04-15 20:05:38 +02:00
if (!aval || !aval[0])
val = dupstring("");
else if (!aval[1])
val = aval[0];
else
isarr = 2;
}
mult_isarr = isarr;
1999-04-15 20:05:38 +02:00
}
if (casmod) {
if (isarr) {
char **ap;
if (!copied)
aval = arrdup(aval), copied = 1;
ap = aval;
if (casmod == 1)
for (; *ap; ap++)
makeuppercase(ap);
else if (casmod == 2)
for (; *ap; ap++)
makelowercase(ap);
else
for (; *ap; ap++)
makecapitals(ap);
} else {
if (!copied)
val = dupstring(val), copied = 1;
if (casmod == 1)
makeuppercase(&val);
else if (casmod == 2)
makelowercase(&val);
else
makecapitals(&val);
}
}
if (presc) {
int ops = opts[PROMPTSUBST], opb = opts[PROMPTBANG];
int opp = opts[PROMPTPERCENT], len;
if (presc < 2) {
opts[PROMPTPERCENT] = 1;
opts[PROMPTSUBST] = opts[PROMPTBANG] = 0;
}
if (isarr) {
char **ap;
if (!copied)
aval = arrdup(aval), copied = 1;
ap = aval;
for (; *ap; ap++) {
char *tmps;
unmetafy(*ap, &len);
untokenize(*ap);
tmps = unmetafy(promptexpand(metafy(*ap, len, META_NOALLOC),
0, NULL, NULL), &len);
*ap = dupstring(tmps);
free(tmps);
}
} else {
char *tmps;
if (!copied)
val = dupstring(val), copied = 1;
unmetafy(val, &len);
untokenize(val);
tmps = unmetafy(promptexpand(metafy(val, len, META_NOALLOC),
0, NULL, NULL), &len);
val = dupstring(tmps);
free(tmps);
}
opts[PROMPTSUBST] = ops;
opts[PROMPTBANG] = opb;
opts[PROMPTPERCENT] = opp;
}
if (quotemod) {
if (--quotetype > 3)
quotetype = 3;
if (isarr) {
char **ap;
if (!copied)
aval = arrdup(aval), copied = 1;
ap = aval;
if (quotemod > 0) {
if (quotetype) {
int sl;
char *tmp;
for (; *ap; ap++) {
int pre = quotetype != 3 ? 1 : 2;
tmp = bslashquote(*ap, NULL, quotetype);
sl = strlen(tmp);
*ap = (char *) zhalloc(pre + sl + 2);
strcpy((*ap) + pre, tmp);
ap[0][pre - 1] = ap[0][pre + sl] = (quotetype != 2 ? '\'' : '"');
ap[0][pre + sl + 1] = '\0';
if (quotetype == 3)
ap[0][0] = '$';
}
} else
for (; *ap; ap++)
*ap = bslashquote(*ap, NULL, 0);
} else {
int one = noerrs, oef = errflag, haserr = 0;
if (!quoteerr)
noerrs = 1;
for (; *ap; ap++) {
haserr |= parse_subst_string(*ap);
remnulargs(*ap);
untokenize(*ap);
}
noerrs = one;
if (!quoteerr)
errflag = oef;
else if (haserr || errflag) {
zerr("parse error in parameter value", NULL, 0);
return NULL;
}
}
} else {
if (!copied)
val = dupstring(val), copied = 1;
if (quotemod > 0) {
if (quotetype) {
int pre = quotetype != 3 ? 1 : 2;
int sl;
char *tmp;
tmp = bslashquote(val, NULL, quotetype);
sl = strlen(tmp);
val = (char *) zhalloc(pre + sl + 2);
strcpy(val + pre, tmp);
val[pre - 1] = val[pre + sl] = (quotetype != 2 ? '\'' : '"');
val[pre + sl + 1] = '\0';
if (quotetype == 3)
val[0] = '$';
} else
val = bslashquote(val, NULL, 0);
} else {
int one = noerrs, oef = errflag, haserr;
if (!quoteerr)
noerrs = 1;
haserr = parse_subst_string(val);
noerrs = one;
if (!quoteerr)
errflag = oef;
else if (haserr || errflag) {
zerr("parse error in parameter value", NULL, 0);
return NULL;
}
remnulargs(val);
untokenize(val);
}
}
}
if (visiblemod) {
if (isarr) {
char **ap;
if (!copied)
aval = arrdup(aval), copied = 1;
for (ap = aval; *ap; ap++)
*ap = nicedupstring(*ap);
} else {
if (!copied)
val = dupstring(val), copied = 1;
val = nicedupstring(val);
}
}
if (shsplit) {
LinkList list = NULL;
if (isarr) {
char **ap;
for (ap = aval; *ap; ap++)
list = bufferwords(list, *ap, NULL);
isarr = 0;
} else
list = bufferwords(NULL, val, NULL);
if (!list || !firstnode(list))
val = dupstring("");
else if (!nextnode(firstnode(list)))
val = getdata(firstnode(list));
else {
char **ap;
LinkNode node;
aval = ap = (char **) zhalloc((countlinknodes(list) + 1) *
sizeof(char *));
for (node = firstnode(list); node; incnode(node))
*ap++ = (char *) getdata(node);
*ap = NULL;
mult_isarr = isarr = 2;
}
copied = 1;
}
1999-04-15 20:05:38 +02:00
if (isarr) {
char *x;
char *y;
int xlen;
int i;
LinkNode on = n;
if (!aval[0] && !plan9) {
if (aptr > (char *) getdata(n) &&
aptr[-1] == Dnull && *fstr == Dnull)
*--aptr = '\0', fstr++;
y = (char *) hcalloc((aptr - ostr) + strlen(fstr) + 1);
1999-04-15 20:05:38 +02:00
strcpy(y, ostr);
*str = y + (aptr - ostr);
strcpy(*str, fstr);
setdata(n, y);
return n;
}
if (unique) {
2001-12-17 02:16:36 +01:00
if(!copied)
aval = arrdup(aval);
i = arrlen(aval);
if (i > 1)
zhuniqarray(aval);
}
1999-04-15 20:05:38 +02:00
if (sortit) {
static CompareFn sortfn[] = {
strpcmp, invstrpcmp, cstrpcmp, invcstrpcmp
};
if (!copied)
aval = arrdup(aval);
i = arrlen(aval);
if (i && (*aval[i-1] || --i))
qsort(aval, i, sizeof(char *), sortfn[sortit-1]);
}
if (plan9) {
LinkNode tn;
local_list1(tl);
1999-04-15 20:05:38 +02:00
*--fstr = Marker;
init_list1(tl, fstr);
2001-06-12 12:34:57 +02:00
if (!eval && !stringsubst(&tl, firstnode(&tl), ssub, 0))
1999-04-15 20:05:38 +02:00
return NULL;
*str = aptr;
tn = firstnode(&tl);
1999-04-15 20:05:38 +02:00
while ((x = *aval++)) {
if (prenum || postnum)
x = dopadding(x, prenum, postnum, preone, postone,
premul, postmul);
if (eval && subst_parse_str(&x, (qt && !nojoin), quoteerr))
1999-04-15 20:05:38 +02:00
return NULL;
xlen = strlen(x);
for (tn = firstnode(&tl);
1999-04-15 20:05:38 +02:00
tn && *(y = (char *) getdata(tn)) == Marker;
incnode(tn)) {
strcatsub(&y, ostr, aptr, x, xlen, y + 1, globsubst,
copied);
1999-04-15 20:05:38 +02:00
if (qt && !*y && isarr != 2)
y = dupstring(nulstring);
if (plan9)
setdata(n, (void *) y), plan9 = 0;
else
insertlinknode(l, n, (void *) y), incnode(n);
}
}
for (; tn; incnode(tn)) {
y = (char *) getdata(tn);
if (*y == Marker)
continue;
if (qt && !*y && isarr != 2)
y = dupstring(nulstring);
if (plan9)
setdata(n, (void *) y), plan9 = 0;
else
insertlinknode(l, n, (void *) y), incnode(n);
}
if (plan9) {
uremnode(l, n);
return n;
}
} else {
x = aval[0];
if (prenum || postnum)
x = dopadding(x, prenum, postnum, preone, postone,
premul, postmul);
if (eval && subst_parse_str(&x, (qt && !nojoin), quoteerr))
1999-04-15 20:05:38 +02:00
return NULL;
xlen = strlen(x);
strcatsub(&y, ostr, aptr, x, xlen, NULL, globsubst, copied);
1999-04-15 20:05:38 +02:00
if (qt && !*y && isarr != 2)
y = dupstring(nulstring);
setdata(n, (void *) y);
i = 1;
/* aval[1] is non-null here */
while (aval[i + 1]) {
x = aval[i++];
if (prenum || postnum)
x = dopadding(x, prenum, postnum, preone, postone,
premul, postmul);
if (eval && subst_parse_str(&x, (qt && !nojoin), quoteerr))
1999-04-15 20:05:38 +02:00
return NULL;
if (qt && !*x && isarr != 2)
y = dupstring(nulstring);
else {
y = dupstring(x);
if (globsubst)
tokenize(y);
}
insertlinknode(l, n, (void *) y), incnode(n);
}
x = aval[i];
if (prenum || postnum)
x = dopadding(x, prenum, postnum, preone, postone,
premul, postmul);
if (eval && subst_parse_str(&x, (qt && !nojoin), quoteerr))
1999-04-15 20:05:38 +02:00
return NULL;
xlen = strlen(x);
*str = strcatsub(&y, aptr, aptr, x, xlen, fstr, globsubst, copied);
1999-04-15 20:05:38 +02:00
if (qt && !*y && isarr != 2)
y = dupstring(nulstring);
insertlinknode(l, n, (void *) y), incnode(n);
}
if (eval)
n = on;
} else {
int xlen;
char *x;
char *y;
x = val;
if (prenum || postnum)
x = dopadding(x, prenum, postnum, preone, postone,
premul, postmul);
if (eval && subst_parse_str(&x, (qt && !nojoin), quoteerr))
1999-04-15 20:05:38 +02:00
return NULL;
xlen = strlen(x);
*str = strcatsub(&y, ostr, aptr, x, xlen, fstr, globsubst, copied);
if (qt && !*y)
1999-04-15 20:05:38 +02:00
y = dupstring(nulstring);
setdata(n, (void *) y);
}
if (eval)
*str = (char *) getdata(n);
return n;
}
/*
* Arithmetic substitution: `a' is the string to be evaluated, `bptr'
* points to the beginning of the string containing it. The tail of
* the string is given by `rest'. *bptr is modified with the substituted
* string. The function returns a pointer to the tail in the substituted
* string.
*/
/**/
static char *
arithsubst(char *a, char **bptr, char *rest)
{
char *s = *bptr, *t;
2001-04-28 19:38:00 +02:00
char buf[BDIGBUFSIZE], *b = buf;
mnumber v;
1999-04-15 20:05:38 +02:00
singsub(&a);
v = matheval(a);
if ((v.type & MN_FLOAT) && !outputradix)
b = convfloat(v.u.d, 0, 0, NULL);
else {
if (v.type & MN_FLOAT)
v.u.l = (zlong) v.u.d;
convbase(buf, v.u.l, outputradix);
}
t = *bptr = (char *) hcalloc(strlen(*bptr) + strlen(b) +
strlen(rest) + 1);
1999-04-15 20:05:38 +02:00
t--;
while ((*++t = *s++));
t--;
while ((*++t = *b++));
strcat(t, rest);
return t;
}
/**/
void
modify(char **str, char **ptr)
{
char *ptr1, *ptr2, *ptr3, del, *lptr, c, *test, *sep, *t, *tt, tc, *e;
char *copy, *all, *tmp, sav;
int gbal, wall, rec, al, nl;
test = NULL;
if (**ptr == ':')
*str = dupstring(*str);
while (**ptr == ':') {
lptr = *ptr;
(*ptr)++;
wall = gbal = 0;
rec = 1;
c = '\0';
sep = NULL;
for (; !c && **ptr;) {
switch (**ptr) {
case 'h':
case 'r':
case 'e':
case 't':
case 'l':
case 'u':
case 'q':
case 'Q':
1999-04-15 20:05:38 +02:00
c = **ptr;
break;
case 's':
c = **ptr;
(*ptr)++;
ptr1 = *ptr;
del = *ptr1++;
for (ptr2 = ptr1; *ptr2 != del && *ptr2; ptr2++);
if (!*ptr2) {
zerr("bad substitution", NULL, 0);
return;
}
*ptr2++ = '\0';
for (ptr3 = ptr2; *ptr3 != del && *ptr3; ptr3++);
if ((sav = *ptr3))
*ptr3++ = '\0';
if (*ptr1) {
zsfree(hsubl);
hsubl = ztrdup(ptr1);
}
1999-04-15 20:05:38 +02:00
if (!hsubl) {
zerr("no previous substitution", NULL, 0);
return;
}
zsfree(hsubr);
for (tt = hsubl; *tt; tt++)
if (INULL(*tt))
chuck(tt--);
untokenize(hsubl);
for (tt = hsubr = ztrdup(ptr2); *tt; tt++)
if (INULL(*tt))
chuck(tt--);
ptr2[-1] = del;
if (sav)
ptr3[-1] = sav;
*ptr = ptr3 - 1;
break;
case '&':
c = 's';
break;
case 'g':
(*ptr)++;
gbal = 1;
break;
case 'w':
wall = 1;
(*ptr)++;
break;
case 'W':
wall = 1;
(*ptr)++;
ptr1 = get_strarg(ptr2 = *ptr);
if ((sav = *ptr1))
*ptr1 = '\0';
sep = dupstring(ptr2 + 1);
if (sav)
*ptr1 = sav;
*ptr = ptr1 + 1;
c = '\0';
break;
case 'f':
rec = -1;
(*ptr)++;
break;
case 'F':
rec = get_intarg(ptr);
(*ptr)++;
break;
default:
*ptr = lptr;
return;
}
}
(*ptr)++;
if (!c) {
*ptr = lptr;
return;
}
if (rec < 0)
test = dupstring(*str);
while (rec--) {
if (wall) {
al = 0;
all = NULL;
for (t = e = *str; (tt = findword(&e, sep));) {
tc = *e;
*e = '\0';
copy = dupstring(tt);
*e = tc;
switch (c) {
case 'h':
remtpath(&copy);
break;
case 'r':
remtext(&copy);
break;
case 'e':
rembutext(&copy);
break;
case 't':
remlpaths(&copy);
break;
case 'l':
downcase(&copy);
break;
case 'u':
upcase(&copy);
break;
case 's':
if (hsubl && hsubr)
subst(&copy, hsubl, hsubr, gbal);
break;
case 'q':
copy = bslashquote(copy, NULL, 0);
break;
case 'Q':
{
int one = noerrs, oef = errflag;
noerrs = 1;
parse_subst_string(copy);
noerrs = one;
errflag = oef;
remnulargs(copy);
untokenize(copy);
}
break;
1999-04-15 20:05:38 +02:00
}
tc = *tt;
*tt = '\0';
nl = al + strlen(t) + strlen(copy);
ptr1 = tmp = (char *)zhalloc(nl + 1);
1999-04-15 20:05:38 +02:00
if (all)
for (ptr2 = all; *ptr2;)
*ptr1++ = *ptr2++;
for (ptr2 = t; *ptr2;)
*ptr1++ = *ptr2++;
*tt = tc;
for (ptr2 = copy; *ptr2;)
*ptr1++ = *ptr2++;
*ptr1 = '\0';
al = nl;
all = tmp;
t = e;
}
*str = all;
} else {
switch (c) {
case 'h':
remtpath(str);
break;
case 'r':
remtext(str);
break;
case 'e':
rembutext(str);
break;
case 't':
remlpaths(str);
break;
case 'l':
downcase(str);
break;
case 'u':
upcase(str);
break;
case 's':
if (hsubl && hsubr) {
char *oldstr = *str;
subst(str, hsubl, hsubr, gbal);
if (*str != oldstr) {
*str = dupstring(oldstr = *str);
zsfree(oldstr);
}
}
break;
case 'q':
*str = bslashquote(*str, NULL, 0);
break;
case 'Q':
{
int one = noerrs, oef = errflag;
noerrs = 1;
parse_subst_string(*str);
noerrs = one;
errflag = oef;
remnulargs(*str);
untokenize(*str);
}
break;
1999-04-15 20:05:38 +02:00
}
}
if (rec < 0) {
if (!strcmp(test, *str))
rec = 0;
else
test = dupstring(*str);
}
}
}
}
/* get a directory stack entry */
/**/
static char *
dstackent(char ch, int val)
{
int backwards;
LinkNode end=(LinkNode)dirstack, n;
backwards = ch == (isset(PUSHDMINUS) ? '+' : '-');
if(!backwards && !val--)
return pwd;
if (backwards)
for (n=lastnode(dirstack); n != end && val; val--, n=prevnode(n));
else
for (end=NULL, n=firstnode(dirstack); n && val; val--, n=nextnode(n));
if (n == end) {
if (backwards && !val)
return pwd;
1999-04-15 20:05:38 +02:00
if (isset(NOMATCH))
zerr("not enough directory stack entries.", NULL, 0);
return NULL;
}
return (char *)getdata(n);
}