Home | History | Annotate | Line # | Download | only in sh
var.c revision 1.60
      1  1.60       kre /*	$NetBSD: var.c,v 1.60 2017/06/17 10:46:34 kre Exp $	*/
      2  1.12       cgd 
      3   1.1       cgd /*-
      4   1.5       jtc  * Copyright (c) 1991, 1993
      5   1.5       jtc  *	The Regents of the University of California.  All rights reserved.
      6   1.1       cgd  *
      7   1.1       cgd  * This code is derived from software contributed to Berkeley by
      8   1.1       cgd  * Kenneth Almquist.
      9   1.1       cgd  *
     10   1.1       cgd  * Redistribution and use in source and binary forms, with or without
     11   1.1       cgd  * modification, are permitted provided that the following conditions
     12   1.1       cgd  * are met:
     13   1.1       cgd  * 1. Redistributions of source code must retain the above copyright
     14   1.1       cgd  *    notice, this list of conditions and the following disclaimer.
     15   1.1       cgd  * 2. Redistributions in binary form must reproduce the above copyright
     16   1.1       cgd  *    notice, this list of conditions and the following disclaimer in the
     17   1.1       cgd  *    documentation and/or other materials provided with the distribution.
     18  1.33       agc  * 3. Neither the name of the University nor the names of its contributors
     19   1.1       cgd  *    may be used to endorse or promote products derived from this software
     20   1.1       cgd  *    without specific prior written permission.
     21   1.1       cgd  *
     22   1.1       cgd  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     23   1.1       cgd  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     24   1.1       cgd  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     25   1.1       cgd  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     26   1.1       cgd  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     27   1.1       cgd  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     28   1.1       cgd  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     29   1.1       cgd  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     30   1.1       cgd  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     31   1.1       cgd  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     32   1.1       cgd  * SUCH DAMAGE.
     33   1.1       cgd  */
     34   1.1       cgd 
     35  1.19  christos #include <sys/cdefs.h>
     36   1.1       cgd #ifndef lint
     37  1.12       cgd #if 0
     38  1.13  christos static char sccsid[] = "@(#)var.c	8.3 (Berkeley) 5/4/95";
     39  1.12       cgd #else
     40  1.60       kre __RCSID("$NetBSD: var.c,v 1.60 2017/06/17 10:46:34 kre Exp $");
     41  1.12       cgd #endif
     42   1.1       cgd #endif /* not lint */
     43   1.1       cgd 
     44  1.58       kre #include <stdio.h>
     45  1.13  christos #include <unistd.h>
     46  1.13  christos #include <stdlib.h>
     47  1.42  christos #include <string.h>
     48  1.21      fair #include <paths.h>
     49  1.41  christos #include <limits.h>
     50  1.13  christos 
     51   1.1       cgd /*
     52   1.1       cgd  * Shell variables.
     53   1.1       cgd  */
     54   1.1       cgd 
     55   1.1       cgd #include "shell.h"
     56   1.1       cgd #include "output.h"
     57   1.1       cgd #include "expand.h"
     58   1.1       cgd #include "nodes.h"	/* for other headers */
     59   1.1       cgd #include "eval.h"	/* defines cmdenviron */
     60   1.1       cgd #include "exec.h"
     61   1.1       cgd #include "syntax.h"
     62   1.1       cgd #include "options.h"
     63  1.40  christos #include "builtins.h"
     64   1.1       cgd #include "mail.h"
     65   1.1       cgd #include "var.h"
     66   1.1       cgd #include "memalloc.h"
     67   1.1       cgd #include "error.h"
     68   1.1       cgd #include "mystring.h"
     69  1.15  christos #include "parser.h"
     70  1.32       dsl #include "show.h"
     71  1.17  christos #ifndef SMALL
     72  1.13  christos #include "myhistedit.h"
     73  1.13  christos #endif
     74   1.1       cgd 
     75  1.35       dsl #ifdef SMALL
     76   1.1       cgd #define VTABSIZE 39
     77  1.35       dsl #else
     78  1.35       dsl #define VTABSIZE 517
     79  1.35       dsl #endif
     80   1.1       cgd 
     81   1.1       cgd 
     82   1.1       cgd struct varinit {
     83   1.1       cgd 	struct var *var;
     84   1.1       cgd 	int flags;
     85  1.23  christos 	const char *text;
     86  1.57       kre 	union var_func_union v_u;
     87   1.1       cgd };
     88  1.57       kre #define	func v_u.set_func
     89  1.57       kre #define	rfunc v_u.ref_func
     90  1.57       kre 
     91  1.57       kre char *get_lineno(struct var *);
     92   1.1       cgd 
     93  1.39  dholland struct localvar *localvars;
     94   1.1       cgd 
     95  1.17  christos #ifndef SMALL
     96   1.5       jtc struct var vhistsize;
     97  1.18  christos struct var vterm;
     98   1.7       cgd #endif
     99   1.1       cgd struct var vifs;
    100   1.1       cgd struct var vmail;
    101   1.1       cgd struct var vmpath;
    102   1.1       cgd struct var vpath;
    103   1.1       cgd struct var vps1;
    104   1.1       cgd struct var vps2;
    105  1.32       dsl struct var vps4;
    106  1.49  christos struct var vvers;
    107  1.14  christos struct var voptind;
    108  1.56       kre struct var line_num;
    109   1.1       cgd 
    110  1.57       kre struct var line_num;
    111  1.57       kre int line_number;
    112  1.57       kre int funclinebase = 0;
    113  1.57       kre int funclineabs = 0;
    114  1.57       kre 
    115  1.48  christos char ifs_default[] = " \t\n";
    116  1.48  christos 
    117   1.1       cgd const struct varinit varinit[] = {
    118  1.17  christos #ifndef SMALL
    119  1.14  christos 	{ &vhistsize,	VSTRFIXED|VTEXTFIXED|VUNSET,	"HISTSIZE=",
    120  1.57       kre 	   { .set_func= sethistsize } },
    121   1.7       cgd #endif
    122  1.14  christos 	{ &vifs,	VSTRFIXED|VTEXTFIXED,		"IFS= \t\n",
    123  1.57       kre 	   { NULL } },
    124  1.14  christos 	{ &vmail,	VSTRFIXED|VTEXTFIXED|VUNSET,	"MAIL=",
    125  1.57       kre 	   { NULL } },
    126  1.14  christos 	{ &vmpath,	VSTRFIXED|VTEXTFIXED|VUNSET,	"MAILPATH=",
    127  1.57       kre 	   { NULL } },
    128  1.49  christos 	{ &vvers,	VSTRFIXED|VTEXTFIXED|VNOEXPORT, "NETBSD_SHELL=",
    129  1.57       kre 	   { NULL } },
    130  1.26       cgd 	{ &vpath,	VSTRFIXED|VTEXTFIXED,		"PATH=" _PATH_DEFPATH,
    131  1.57       kre 	   { .set_func= changepath } },
    132  1.15  christos 	/*
    133   1.1       cgd 	 * vps1 depends on uid
    134   1.1       cgd 	 */
    135  1.14  christos 	{ &vps2,	VSTRFIXED|VTEXTFIXED,		"PS2=> ",
    136  1.57       kre 	   { NULL } },
    137  1.32       dsl 	{ &vps4,	VSTRFIXED|VTEXTFIXED,		"PS4=+ ",
    138  1.57       kre 	   { NULL } },
    139  1.18  christos #ifndef SMALL
    140  1.14  christos 	{ &vterm,	VSTRFIXED|VTEXTFIXED|VUNSET,	"TERM=",
    141  1.57       kre 	   { .set_func= setterm } },
    142   1.1       cgd #endif
    143  1.32       dsl 	{ &voptind,	VSTRFIXED|VTEXTFIXED|VNOFUNC,	"OPTIND=1",
    144  1.57       kre 	   { .set_func= getoptsreset } },
    145  1.57       kre 	{ &line_num,	VSTRFIXED|VTEXTFIXED|VFUNCREF,	"LINENO=1",
    146  1.57       kre 	   { .ref_func= get_lineno } },
    147  1.14  christos 	{ NULL,	0,				NULL,
    148  1.57       kre 	   { NULL } }
    149   1.1       cgd };
    150   1.1       cgd 
    151   1.1       cgd struct var *vartab[VTABSIZE];
    152   1.1       cgd 
    153  1.35       dsl STATIC int strequal(const char *, const char *);
    154  1.35       dsl STATIC struct var *find_var(const char *, struct var ***, int *);
    155   1.1       cgd 
    156   1.1       cgd /*
    157   1.1       cgd  * Initialize the varable symbol tables and import the environment
    158   1.1       cgd  */
    159   1.1       cgd 
    160   1.1       cgd #ifdef mkinit
    161  1.47  christos INCLUDE <stdio.h>
    162  1.47  christos INCLUDE <unistd.h>
    163   1.1       cgd INCLUDE "var.h"
    164  1.49  christos INCLUDE "version.h"
    165  1.27  christos MKINIT char **environ;
    166   1.1       cgd INIT {
    167   1.1       cgd 	char **envp;
    168  1.47  christos 	char buf[64];
    169   1.1       cgd 
    170   1.1       cgd 	initvar();
    171   1.1       cgd 	for (envp = environ ; *envp ; envp++) {
    172   1.1       cgd 		if (strchr(*envp, '=')) {
    173   1.1       cgd 			setvareq(*envp, VEXPORT|VTEXTFIXED);
    174   1.1       cgd 		}
    175   1.1       cgd 	}
    176  1.47  christos 
    177  1.47  christos 	/*
    178  1.53       kre 	 * Set variables which override anything read from environment.
    179  1.53       kre 	 *
    180  1.47  christos 	 * PPID is readonly
    181  1.53       kre 	 * Always default IFS
    182  1.53       kre 	 * NETBSD_SHELL is a constant (readonly), and is never exported
    183  1.57       kre 	 * LINENO is simply magic...
    184  1.47  christos 	 */
    185  1.47  christos 	snprintf(buf, sizeof(buf), "%d", (int)getppid());
    186  1.47  christos 	setvar("PPID", buf, VREADONLY);
    187  1.48  christos 	setvar("IFS", ifs_default, VTEXTFIXED);
    188  1.53       kre 
    189  1.53       kre 	setvar("NETBSD_SHELL", NETBSD_SHELL
    190  1.53       kre #ifdef BUILD_DATE
    191  1.53       kre 		" BUILD:" BUILD_DATE
    192  1.53       kre #endif
    193  1.53       kre #ifdef DEBUG
    194  1.53       kre 		" DEBUG"
    195  1.53       kre #endif
    196  1.53       kre #if !defined(JOBS) || JOBS == 0
    197  1.53       kre 		" -JOBS"
    198  1.53       kre #endif
    199  1.53       kre #ifndef DO_SHAREDVFORK
    200  1.53       kre 		" -VFORK"
    201  1.53       kre #endif
    202  1.53       kre #ifdef SMALL
    203  1.53       kre 		" SMALL"
    204  1.53       kre #endif
    205  1.53       kre #ifdef TINY
    206  1.53       kre 		" TINY"
    207  1.53       kre #endif
    208  1.53       kre #ifdef OLD_TTY_DRIVER
    209  1.53       kre 		" OLD_TTY"
    210  1.53       kre #endif
    211  1.53       kre #ifdef SYSV
    212  1.53       kre 		" SYSV"
    213  1.53       kre #endif
    214  1.53       kre #ifndef BSD
    215  1.53       kre 		" -BSD"
    216  1.53       kre #endif
    217  1.55       kre #ifdef BOGUS_NOT_COMMAND
    218  1.55       kre 		" BOGUS_NOT"
    219  1.55       kre #endif
    220  1.53       kre 		    , VTEXTFIXED|VREADONLY|VNOEXPORT);
    221  1.56       kre 
    222  1.56       kre 	setvar("LINENO", "1", VTEXTFIXED);
    223   1.1       cgd }
    224   1.1       cgd #endif
    225   1.1       cgd 
    226   1.1       cgd 
    227   1.1       cgd /*
    228   1.1       cgd  * This routine initializes the builtin variables.  It is called when the
    229   1.1       cgd  * shell is initialized and again when a shell procedure is spawned.
    230   1.1       cgd  */
    231   1.1       cgd 
    232   1.1       cgd void
    233  1.30  christos initvar(void)
    234  1.30  christos {
    235   1.1       cgd 	const struct varinit *ip;
    236   1.1       cgd 	struct var *vp;
    237   1.1       cgd 	struct var **vpp;
    238   1.1       cgd 
    239   1.1       cgd 	for (ip = varinit ; (vp = ip->var) != NULL ; ip++) {
    240  1.35       dsl 		if (find_var(ip->text, &vpp, &vp->name_len) != NULL)
    241  1.35       dsl 			continue;
    242  1.35       dsl 		vp->next = *vpp;
    243  1.35       dsl 		*vpp = vp;
    244  1.35       dsl 		vp->text = strdup(ip->text);
    245  1.35       dsl 		vp->flags = ip->flags;
    246  1.57       kre 		vp->v_u = ip->v_u;
    247   1.1       cgd 	}
    248   1.1       cgd 	/*
    249   1.1       cgd 	 * PS1 depends on uid
    250   1.1       cgd 	 */
    251  1.35       dsl 	if (find_var("PS1", &vpp, &vps1.name_len) == NULL) {
    252   1.1       cgd 		vps1.next = *vpp;
    253   1.1       cgd 		*vpp = &vps1;
    254   1.1       cgd 		vps1.flags = VSTRFIXED|VTEXTFIXED;
    255  1.44  christos 		vps1.text = NULL;
    256  1.44  christos 		choose_ps1();
    257   1.1       cgd 	}
    258   1.1       cgd }
    259   1.1       cgd 
    260  1.44  christos void
    261  1.44  christos choose_ps1(void)
    262  1.44  christos {
    263  1.44  christos 	free(vps1.text);
    264  1.44  christos 	vps1.text = strdup(geteuid() ? "PS1=$ " : "PS1=# ");
    265  1.44  christos }
    266  1.44  christos 
    267   1.1       cgd /*
    268  1.14  christos  * Safe version of setvar, returns 1 on success 0 on failure.
    269  1.14  christos  */
    270  1.14  christos 
    271  1.14  christos int
    272  1.30  christos setvarsafe(const char *name, const char *val, int flags)
    273  1.14  christos {
    274  1.14  christos 	struct jmploc jmploc;
    275  1.59       kre 	struct jmploc * const savehandler = handler;
    276  1.38  christos 	int volatile err = 0;
    277  1.14  christos 
    278  1.14  christos 	if (setjmp(jmploc.loc))
    279  1.14  christos 		err = 1;
    280  1.14  christos 	else {
    281  1.14  christos 		handler = &jmploc;
    282  1.14  christos 		setvar(name, val, flags);
    283  1.14  christos 	}
    284  1.14  christos 	handler = savehandler;
    285  1.14  christos 	return err;
    286  1.14  christos }
    287  1.14  christos 
    288  1.14  christos /*
    289   1.1       cgd  * Set the value of a variable.  The flags argument is ored with the
    290   1.1       cgd  * flags of the variable.  If val is NULL, the variable is unset.
    291   1.1       cgd  */
    292   1.1       cgd 
    293   1.1       cgd void
    294  1.30  christos setvar(const char *name, const char *val, int flags)
    295  1.10       cgd {
    296  1.23  christos 	const char *p;
    297  1.23  christos 	const char *q;
    298  1.23  christos 	char *d;
    299   1.1       cgd 	int len;
    300   1.1       cgd 	int namelen;
    301   1.1       cgd 	char *nameeq;
    302   1.1       cgd 	int isbad;
    303   1.1       cgd 
    304   1.1       cgd 	isbad = 0;
    305   1.1       cgd 	p = name;
    306  1.15  christos 	if (! is_name(*p))
    307   1.1       cgd 		isbad = 1;
    308  1.15  christos 	p++;
    309   1.1       cgd 	for (;;) {
    310   1.1       cgd 		if (! is_in_name(*p)) {
    311   1.1       cgd 			if (*p == '\0' || *p == '=')
    312   1.1       cgd 				break;
    313   1.1       cgd 			isbad = 1;
    314   1.1       cgd 		}
    315   1.1       cgd 		p++;
    316   1.1       cgd 	}
    317   1.1       cgd 	namelen = p - name;
    318   1.1       cgd 	if (isbad)
    319   1.5       jtc 		error("%.*s: bad variable name", namelen, name);
    320   1.1       cgd 	len = namelen + 2;		/* 2 is space for '=' and '\0' */
    321   1.1       cgd 	if (val == NULL) {
    322   1.1       cgd 		flags |= VUNSET;
    323   1.1       cgd 	} else {
    324   1.1       cgd 		len += strlen(val);
    325   1.1       cgd 	}
    326  1.23  christos 	d = nameeq = ckmalloc(len);
    327   1.1       cgd 	q = name;
    328   1.1       cgd 	while (--namelen >= 0)
    329  1.23  christos 		*d++ = *q++;
    330  1.23  christos 	*d++ = '=';
    331  1.23  christos 	*d = '\0';
    332   1.1       cgd 	if (val)
    333  1.23  christos 		scopy(val, d);
    334   1.1       cgd 	setvareq(nameeq, flags);
    335   1.1       cgd }
    336   1.1       cgd 
    337   1.1       cgd 
    338   1.1       cgd 
    339   1.1       cgd /*
    340   1.1       cgd  * Same as setvar except that the variable and value are passed in
    341   1.1       cgd  * the first argument as name=value.  Since the first argument will
    342   1.1       cgd  * be actually stored in the table, it should not be a string that
    343   1.1       cgd  * will go away.
    344   1.1       cgd  */
    345   1.1       cgd 
    346   1.1       cgd void
    347  1.30  christos setvareq(char *s, int flags)
    348  1.10       cgd {
    349   1.1       cgd 	struct var *vp, **vpp;
    350  1.35       dsl 	int nlen;
    351   1.1       cgd 
    352  1.49  christos 	if (aflag && !(flags & VNOEXPORT))
    353  1.28     bjh21 		flags |= VEXPORT;
    354  1.35       dsl 	vp = find_var(s, &vpp, &nlen);
    355  1.35       dsl 	if (vp != NULL) {
    356  1.60       kre 		if (vp->flags & VREADONLY) {
    357  1.60       kre 			if ((flags & (VTEXTFIXED|VSTACK)) == 0)
    358  1.60       kre 				ckfree(s);
    359  1.60       kre 			if (flags & VNOERROR)
    360  1.60       kre 				return;
    361  1.35       dsl 			error("%.*s: is read only", vp->name_len, s);
    362  1.60       kre 		}
    363  1.60       kre 		if (flags & VNOSET) {
    364  1.60       kre 			if ((flags & (VTEXTFIXED|VSTACK)) == 0)
    365  1.60       kre 				ckfree(s);
    366  1.35       dsl 			return;
    367  1.60       kre 		}
    368  1.57       kre 
    369  1.35       dsl 		INTOFF;
    370  1.14  christos 
    371  1.57       kre 		if (vp->func && !(vp->flags & VFUNCREF) && !(flags & VNOFUNC))
    372  1.35       dsl 			(*vp->func)(s + vp->name_len + 1);
    373  1.14  christos 
    374  1.35       dsl 		if ((vp->flags & (VTEXTFIXED|VSTACK)) == 0)
    375  1.35       dsl 			ckfree(vp->text);
    376  1.14  christos 
    377  1.35       dsl 		vp->flags &= ~(VTEXTFIXED|VSTACK|VUNSET);
    378  1.49  christos 		if (flags & VNOEXPORT)
    379  1.49  christos 			vp->flags &= ~VEXPORT;
    380  1.54       kre 		if (vp->flags & VNOEXPORT)
    381  1.54       kre 			flags &= ~VEXPORT;
    382  1.35       dsl 		vp->flags |= flags & ~VNOFUNC;
    383  1.35       dsl 		vp->text = s;
    384  1.35       dsl 
    385  1.35       dsl 		/*
    386  1.35       dsl 		 * We could roll this to a function, to handle it as
    387  1.35       dsl 		 * a regular variable function callback, but why bother?
    388  1.35       dsl 		 */
    389  1.35       dsl 		if (vp == &vmpath || (vp == &vmail && ! mpathset()))
    390  1.35       dsl 			chkmail(1);
    391  1.57       kre 
    392  1.35       dsl 		INTON;
    393  1.35       dsl 		return;
    394   1.1       cgd 	}
    395   1.1       cgd 	/* not found */
    396  1.60       kre 	if (flags & VNOSET) {
    397  1.60       kre 		if ((flags & (VTEXTFIXED|VSTACK)) == 0)
    398  1.60       kre 			ckfree(s);
    399  1.30  christos 		return;
    400  1.60       kre 	}
    401   1.1       cgd 	vp = ckmalloc(sizeof (*vp));
    402  1.57       kre 	vp->flags = flags & ~(VNOFUNC|VFUNCREF);
    403   1.1       cgd 	vp->text = s;
    404  1.35       dsl 	vp->name_len = nlen;
    405   1.1       cgd 	vp->next = *vpp;
    406  1.14  christos 	vp->func = NULL;
    407   1.1       cgd 	*vpp = vp;
    408   1.1       cgd }
    409   1.1       cgd 
    410   1.1       cgd 
    411   1.1       cgd 
    412   1.1       cgd /*
    413   1.1       cgd  * Process a linked list of variable assignments.
    414   1.1       cgd  */
    415   1.1       cgd 
    416   1.1       cgd void
    417  1.30  christos listsetvar(struct strlist *list, int flags)
    418  1.30  christos {
    419   1.1       cgd 	struct strlist *lp;
    420   1.1       cgd 
    421   1.1       cgd 	INTOFF;
    422   1.1       cgd 	for (lp = list ; lp ; lp = lp->next) {
    423  1.30  christos 		setvareq(savestr(lp->text), flags);
    424   1.1       cgd 	}
    425   1.1       cgd 	INTON;
    426   1.1       cgd }
    427   1.1       cgd 
    428  1.32       dsl void
    429  1.32       dsl listmklocal(struct strlist *list, int flags)
    430  1.32       dsl {
    431  1.32       dsl 	struct strlist *lp;
    432  1.32       dsl 
    433  1.32       dsl 	for (lp = list ; lp ; lp = lp->next)
    434  1.32       dsl 		mklocal(lp->text, flags);
    435  1.32       dsl }
    436   1.1       cgd 
    437   1.1       cgd 
    438   1.1       cgd /*
    439   1.1       cgd  * Find the value of a variable.  Returns NULL if not set.
    440   1.1       cgd  */
    441   1.1       cgd 
    442   1.1       cgd char *
    443  1.30  christos lookupvar(const char *name)
    444  1.30  christos {
    445   1.1       cgd 	struct var *v;
    446   1.1       cgd 
    447  1.35       dsl 	v = find_var(name, NULL, NULL);
    448  1.35       dsl 	if (v == NULL || v->flags & VUNSET)
    449  1.35       dsl 		return NULL;
    450  1.57       kre 	if (v->rfunc && (v->flags & VFUNCREF) != 0)
    451  1.57       kre 		return (*v->rfunc)(v) + v->name_len + 1;
    452  1.35       dsl 	return v->text + v->name_len + 1;
    453   1.1       cgd }
    454   1.1       cgd 
    455   1.1       cgd 
    456   1.1       cgd 
    457   1.1       cgd /*
    458   1.1       cgd  * Search the environment of a builtin command.  If the second argument
    459   1.1       cgd  * is nonzero, return the value of a variable even if it hasn't been
    460   1.1       cgd  * exported.
    461   1.1       cgd  */
    462   1.1       cgd 
    463   1.1       cgd char *
    464  1.30  christos bltinlookup(const char *name, int doall)
    465  1.10       cgd {
    466   1.1       cgd 	struct strlist *sp;
    467   1.1       cgd 	struct var *v;
    468   1.1       cgd 
    469   1.1       cgd 	for (sp = cmdenviron ; sp ; sp = sp->next) {
    470  1.35       dsl 		if (strequal(sp->text, name))
    471   1.1       cgd 			return strchr(sp->text, '=') + 1;
    472   1.1       cgd 	}
    473  1.35       dsl 
    474  1.35       dsl 	v = find_var(name, NULL, NULL);
    475  1.35       dsl 
    476  1.35       dsl 	if (v == NULL || v->flags & VUNSET || (!doall && !(v->flags & VEXPORT)))
    477  1.35       dsl 		return NULL;
    478  1.57       kre 	if (v->rfunc && (v->flags & VFUNCREF) != 0)
    479  1.57       kre 		return (*v->rfunc)(v) + v->name_len + 1;
    480  1.35       dsl 	return v->text + v->name_len + 1;
    481   1.1       cgd }
    482   1.1       cgd 
    483   1.1       cgd 
    484   1.1       cgd 
    485   1.1       cgd /*
    486   1.1       cgd  * Generate a list of exported variables.  This routine is used to construct
    487   1.1       cgd  * the third argument to execve when executing a program.
    488   1.1       cgd  */
    489   1.1       cgd 
    490   1.1       cgd char **
    491  1.30  christos environment(void)
    492  1.30  christos {
    493   1.1       cgd 	int nenv;
    494   1.1       cgd 	struct var **vpp;
    495   1.1       cgd 	struct var *vp;
    496  1.23  christos 	char **env;
    497  1.23  christos 	char **ep;
    498   1.1       cgd 
    499   1.1       cgd 	nenv = 0;
    500   1.1       cgd 	for (vpp = vartab ; vpp < vartab + VTABSIZE ; vpp++) {
    501   1.1       cgd 		for (vp = *vpp ; vp ; vp = vp->next)
    502  1.51       kre 			if ((vp->flags & (VEXPORT|VUNSET)) == VEXPORT)
    503   1.1       cgd 				nenv++;
    504   1.1       cgd 	}
    505  1.58       kre 	CTRACE(DBG_VARS, ("environment: %d vars to export\n", nenv));
    506   1.1       cgd 	ep = env = stalloc((nenv + 1) * sizeof *env);
    507   1.1       cgd 	for (vpp = vartab ; vpp < vartab + VTABSIZE ; vpp++) {
    508   1.1       cgd 		for (vp = *vpp ; vp ; vp = vp->next)
    509  1.57       kre 			if ((vp->flags & (VEXPORT|VUNSET)) == VEXPORT) {
    510  1.57       kre 				if (vp->rfunc && (vp->flags & VFUNCREF))
    511  1.57       kre 					*ep++ = (*vp->rfunc)(vp);
    512  1.57       kre 				else
    513  1.57       kre 					*ep++ = vp->text;
    514  1.58       kre 				VTRACE(DBG_VARS, ("environment: %s\n", ep[-1]));
    515  1.57       kre 			}
    516   1.1       cgd 	}
    517   1.1       cgd 	*ep = NULL;
    518   1.1       cgd 	return env;
    519   1.1       cgd }
    520   1.1       cgd 
    521   1.1       cgd 
    522   1.1       cgd /*
    523   1.1       cgd  * Called when a shell procedure is invoked to clear out nonexported
    524   1.1       cgd  * variables.  It is also necessary to reallocate variables of with
    525   1.1       cgd  * VSTACK set since these are currently allocated on the stack.
    526   1.1       cgd  */
    527   1.1       cgd 
    528   1.1       cgd #ifdef mkinit
    529  1.30  christos void shprocvar(void);
    530   1.1       cgd 
    531   1.1       cgd SHELLPROC {
    532   1.1       cgd 	shprocvar();
    533   1.1       cgd }
    534   1.1       cgd #endif
    535   1.1       cgd 
    536   1.1       cgd void
    537  1.30  christos shprocvar(void)
    538  1.30  christos {
    539   1.1       cgd 	struct var **vpp;
    540   1.1       cgd 	struct var *vp, **prev;
    541   1.1       cgd 
    542   1.1       cgd 	for (vpp = vartab ; vpp < vartab + VTABSIZE ; vpp++) {
    543   1.1       cgd 		for (prev = vpp ; (vp = *prev) != NULL ; ) {
    544   1.1       cgd 			if ((vp->flags & VEXPORT) == 0) {
    545   1.1       cgd 				*prev = vp->next;
    546   1.1       cgd 				if ((vp->flags & VTEXTFIXED) == 0)
    547   1.1       cgd 					ckfree(vp->text);
    548   1.1       cgd 				if ((vp->flags & VSTRFIXED) == 0)
    549   1.1       cgd 					ckfree(vp);
    550   1.1       cgd 			} else {
    551   1.1       cgd 				if (vp->flags & VSTACK) {
    552   1.1       cgd 					vp->text = savestr(vp->text);
    553   1.1       cgd 					vp->flags &=~ VSTACK;
    554   1.1       cgd 				}
    555   1.1       cgd 				prev = &vp->next;
    556   1.1       cgd 			}
    557   1.1       cgd 		}
    558   1.1       cgd 	}
    559   1.1       cgd 	initvar();
    560   1.1       cgd }
    561   1.1       cgd 
    562   1.1       cgd 
    563   1.1       cgd 
    564   1.1       cgd /*
    565   1.1       cgd  * Command to list all variables which are set.  Currently this command
    566   1.1       cgd  * is invoked from the set command when the set command is called without
    567   1.1       cgd  * any variables.
    568   1.1       cgd  */
    569   1.1       cgd 
    570  1.30  christos void
    571  1.30  christos print_quoted(const char *p)
    572  1.30  christos {
    573  1.30  christos 	const char *q;
    574  1.30  christos 
    575  1.50       kre 	if (p[0] == '\0') {
    576  1.50       kre 		out1fmt("''");
    577  1.50       kre 		return;
    578  1.50       kre 	}
    579  1.30  christos 	if (strcspn(p, "|&;<>()$`\\\"' \t\n*?[]#~=%") == strlen(p)) {
    580  1.30  christos 		out1fmt("%s", p);
    581  1.30  christos 		return;
    582  1.30  christos 	}
    583  1.30  christos 	while (*p) {
    584  1.30  christos 		if (*p == '\'') {
    585  1.30  christos 			out1fmt("\\'");
    586  1.30  christos 			p++;
    587  1.30  christos 			continue;
    588  1.30  christos 		}
    589  1.42  christos 		q = strchr(p, '\'');
    590  1.30  christos 		if (!q) {
    591  1.30  christos 			out1fmt("'%s'", p );
    592  1.30  christos 			return;
    593  1.30  christos 		}
    594  1.31       agc 		out1fmt("'%.*s'", (int)(q - p), p );
    595  1.30  christos 		p = q;
    596  1.30  christos 	}
    597  1.30  christos }
    598  1.30  christos 
    599  1.30  christos static int
    600  1.30  christos sort_var(const void *v_v1, const void *v_v2)
    601  1.30  christos {
    602  1.30  christos 	const struct var * const *v1 = v_v1;
    603  1.30  christos 	const struct var * const *v2 = v_v2;
    604  1.52       kre 	char *t1 = (*v1)->text, *t2 = (*v2)->text;
    605  1.30  christos 
    606  1.52       kre 	if (*t1 == *t2) {
    607  1.52       kre 		char *p, *s;
    608  1.52       kre 
    609  1.52       kre 		STARTSTACKSTR(p);
    610  1.52       kre 
    611  1.52       kre 		/*
    612  1.52       kre 		 * note: if lengths are equal, strings must be different
    613  1.52       kre 		 * so we don't care which string we pick for the \0 in
    614  1.52       kre 		 * that case.
    615  1.52       kre 		 */
    616  1.52       kre 		if ((strchr(t1, '=') - t1) <= (strchr(t2, '=') - t2)) {
    617  1.52       kre 			s = t1;
    618  1.52       kre 			t1 = p;
    619  1.52       kre 		} else {
    620  1.52       kre 			s = t2;
    621  1.52       kre 			t2 = p;
    622  1.52       kre 		}
    623  1.52       kre 
    624  1.52       kre 		while (*s && *s != '=') {
    625  1.52       kre 			STPUTC(*s, p);
    626  1.52       kre 			s++;
    627  1.52       kre 		}
    628  1.52       kre 		STPUTC('\0', p);
    629  1.52       kre 	}
    630  1.52       kre 
    631  1.52       kre 	return strcoll(t1, t2);
    632  1.30  christos }
    633  1.30  christos 
    634  1.30  christos /*
    635  1.30  christos  * POSIX requires that 'set' (but not export or readonly) output the
    636  1.30  christos  * variables in lexicographic order - by the locale's collating order (sigh).
    637  1.30  christos  * Maybe we could keep them in an ordered balanced binary tree
    638  1.30  christos  * instead of hashed lists.
    639  1.30  christos  * For now just roll 'em through qsort for printing...
    640  1.30  christos  */
    641  1.30  christos 
    642   1.1       cgd int
    643  1.49  christos showvars(const char *name, int flag, int show_value, const char *xtra)
    644  1.10       cgd {
    645   1.1       cgd 	struct var **vpp;
    646   1.1       cgd 	struct var *vp;
    647  1.30  christos 	const char *p;
    648  1.30  christos 
    649  1.30  christos 	static struct var **list;	/* static in case we are interrupted */
    650  1.30  christos 	static int list_len;
    651  1.30  christos 	int count = 0;
    652  1.30  christos 
    653  1.30  christos 	if (!list) {
    654  1.30  christos 		list_len = 32;
    655  1.30  christos 		list = ckmalloc(list_len * sizeof *list);
    656  1.30  christos 	}
    657   1.1       cgd 
    658   1.1       cgd 	for (vpp = vartab ; vpp < vartab + VTABSIZE ; vpp++) {
    659   1.1       cgd 		for (vp = *vpp ; vp ; vp = vp->next) {
    660  1.30  christos 			if (flag && !(vp->flags & flag))
    661  1.30  christos 				continue;
    662  1.30  christos 			if (vp->flags & VUNSET && !(show_value & 2))
    663  1.30  christos 				continue;
    664  1.30  christos 			if (count >= list_len) {
    665  1.30  christos 				list = ckrealloc(list,
    666  1.30  christos 					(list_len << 1) * sizeof *list);
    667  1.30  christos 				list_len <<= 1;
    668  1.30  christos 			}
    669  1.30  christos 			list[count++] = vp;
    670   1.1       cgd 		}
    671   1.1       cgd 	}
    672  1.30  christos 
    673  1.30  christos 	qsort(list, count, sizeof *list, sort_var);
    674  1.30  christos 
    675  1.30  christos 	for (vpp = list; count--; vpp++) {
    676  1.30  christos 		vp = *vpp;
    677  1.30  christos 		if (name)
    678  1.30  christos 			out1fmt("%s ", name);
    679  1.49  christos 		if (xtra)
    680  1.49  christos 			out1fmt("%s ", xtra);
    681  1.57       kre 		p = vp->text;
    682  1.57       kre 		if (vp->rfunc && (vp->flags & VFUNCREF) != 0) {
    683  1.57       kre 			p = (*vp->rfunc)(vp);
    684  1.57       kre 			if (p == NULL)
    685  1.57       kre 				p = vp->text;
    686  1.57       kre 		}
    687  1.57       kre 		for ( ; *p != '=' ; p++)
    688  1.30  christos 			out1c(*p);
    689  1.30  christos 		if (!(vp->flags & VUNSET) && show_value) {
    690  1.30  christos 			out1fmt("=");
    691  1.30  christos 			print_quoted(++p);
    692  1.30  christos 		}
    693  1.30  christos 		out1c('\n');
    694  1.30  christos 	}
    695   1.1       cgd 	return 0;
    696   1.1       cgd }
    697   1.1       cgd 
    698   1.1       cgd 
    699   1.1       cgd 
    700   1.1       cgd /*
    701   1.1       cgd  * The export and readonly commands.
    702   1.1       cgd  */
    703   1.1       cgd 
    704   1.1       cgd int
    705  1.30  christos exportcmd(int argc, char **argv)
    706  1.10       cgd {
    707   1.1       cgd 	struct var *vp;
    708   1.1       cgd 	char *name;
    709  1.23  christos 	const char *p;
    710   1.1       cgd 	int flag = argv[0][0] == 'r'? VREADONLY : VEXPORT;
    711  1.49  christos 	int pflg = 0;
    712  1.49  christos 	int nflg = 0;
    713  1.49  christos 	int xflg = 0;
    714  1.49  christos 	int res;
    715  1.49  christos 	int c;
    716  1.51       kre 	int f;
    717  1.49  christos 
    718  1.49  christos 	while ((c = nextopt("npx")) != '\0') {
    719  1.49  christos 		switch (c) {
    720  1.49  christos 		case 'p':
    721  1.49  christos 			if (nflg)
    722  1.49  christos 				return 1;
    723  1.49  christos 			pflg = 3;
    724  1.49  christos 			break;
    725  1.49  christos 		case 'n':
    726  1.49  christos 			if (pflg || xflg || flag == VREADONLY)
    727  1.49  christos 				return 1;
    728  1.49  christos 			nflg = 1;
    729  1.49  christos 			break;
    730  1.49  christos 		case 'x':
    731  1.49  christos 			if (nflg || flag == VREADONLY)
    732  1.49  christos 				return 1;
    733  1.49  christos 			flag = VNOEXPORT;
    734  1.49  christos 			xflg = 1;
    735  1.49  christos 			break;
    736  1.49  christos 		default:
    737  1.49  christos 			return 1;
    738  1.49  christos 		}
    739  1.49  christos 	}
    740   1.1       cgd 
    741  1.49  christos 	if (nflg && *argptr == NULL)
    742  1.49  christos 		return 1;
    743  1.49  christos 
    744  1.49  christos 	if (pflg || *argptr == NULL) {
    745  1.49  christos 		showvars( pflg ? argv[0] : 0, flag, pflg,
    746  1.49  christos 		    pflg && xflg ? "-x" : NULL );
    747  1.30  christos 		return 0;
    748  1.30  christos 	}
    749  1.30  christos 
    750  1.49  christos 	res = 0;
    751  1.30  christos 	while ((name = *argptr++) != NULL) {
    752  1.51       kre 		f = flag;
    753  1.30  christos 		if ((p = strchr(name, '=')) != NULL) {
    754  1.30  christos 			p++;
    755  1.30  christos 		} else {
    756  1.35       dsl 			vp = find_var(name, NULL, NULL);
    757  1.35       dsl 			if (vp != NULL) {
    758  1.49  christos 				if (nflg)
    759  1.49  christos 					vp->flags &= ~flag;
    760  1.49  christos 				else if (flag&VEXPORT && vp->flags&VNOEXPORT)
    761  1.49  christos 					res = 1;
    762  1.49  christos 				else {
    763  1.49  christos 					vp->flags |= flag;
    764  1.49  christos 					if (flag == VNOEXPORT)
    765  1.49  christos 						vp->flags &= ~VEXPORT;
    766  1.49  christos 				}
    767  1.36     enami 				continue;
    768  1.51       kre 			} else
    769  1.51       kre 				f |= VUNSET;
    770   1.1       cgd 		}
    771  1.49  christos 		if (!nflg)
    772  1.51       kre 			setvar(name, p, f);
    773   1.1       cgd 	}
    774  1.49  christos 	return res;
    775   1.1       cgd }
    776   1.1       cgd 
    777   1.1       cgd 
    778   1.1       cgd /*
    779   1.1       cgd  * The "local" command.
    780   1.1       cgd  */
    781   1.1       cgd 
    782  1.10       cgd int
    783  1.30  christos localcmd(int argc, char **argv)
    784  1.10       cgd {
    785   1.1       cgd 	char *name;
    786  1.54       kre 	int c;
    787  1.54       kre 	int flags = 0;		/*XXX perhaps VUNSET from a -o option value */
    788   1.1       cgd 
    789   1.1       cgd 	if (! in_function())
    790   1.1       cgd 		error("Not in a function");
    791  1.54       kre 
    792  1.54       kre 	/* upper case options, as bash stole all the good ones ... */
    793  1.54       kre 	while ((c = nextopt("INx")) != '\0')
    794  1.54       kre 		switch (c) {
    795  1.54       kre 		case 'I':	flags &= ~VUNSET;	break;
    796  1.54       kre 		case 'N':	flags |= VUNSET;	break;
    797  1.54       kre 		case 'x':	flags |= VEXPORT;	break;
    798  1.54       kre 		}
    799  1.54       kre 
    800   1.1       cgd 	while ((name = *argptr++) != NULL) {
    801  1.54       kre 		mklocal(name, flags);
    802   1.1       cgd 	}
    803   1.1       cgd 	return 0;
    804   1.1       cgd }
    805   1.1       cgd 
    806   1.1       cgd 
    807   1.1       cgd /*
    808  1.37       snj  * Make a variable a local variable.  When a variable is made local, its
    809   1.1       cgd  * value and flags are saved in a localvar structure.  The saved values
    810   1.1       cgd  * will be restored when the shell function returns.  We handle the name
    811   1.1       cgd  * "-" as a special case.
    812   1.1       cgd  */
    813   1.1       cgd 
    814   1.1       cgd void
    815  1.32       dsl mklocal(const char *name, int flags)
    816  1.30  christos {
    817   1.1       cgd 	struct localvar *lvp;
    818   1.1       cgd 	struct var **vpp;
    819   1.1       cgd 	struct var *vp;
    820   1.1       cgd 
    821   1.1       cgd 	INTOFF;
    822   1.1       cgd 	lvp = ckmalloc(sizeof (struct localvar));
    823   1.1       cgd 	if (name[0] == '-' && name[1] == '\0') {
    824  1.23  christos 		char *p;
    825  1.30  christos 		p = ckmalloc(sizeof_optlist);
    826  1.30  christos 		lvp->text = memcpy(p, optlist, sizeof_optlist);
    827   1.1       cgd 		vp = NULL;
    828   1.1       cgd 	} else {
    829  1.35       dsl 		vp = find_var(name, &vpp, NULL);
    830   1.1       cgd 		if (vp == NULL) {
    831  1.54       kre 			flags &= ~VNOEXPORT;
    832   1.1       cgd 			if (strchr(name, '='))
    833  1.54       kre 				setvareq(savestr(name),
    834  1.54       kre 				    VSTRFIXED | (flags & ~VUNSET));
    835   1.1       cgd 			else
    836  1.29  christos 				setvar(name, NULL, VSTRFIXED|flags);
    837   1.1       cgd 			vp = *vpp;	/* the new variable */
    838   1.1       cgd 			lvp->text = NULL;
    839   1.1       cgd 			lvp->flags = VUNSET;
    840   1.1       cgd 		} else {
    841   1.1       cgd 			lvp->text = vp->text;
    842   1.1       cgd 			lvp->flags = vp->flags;
    843   1.1       cgd 			vp->flags |= VSTRFIXED|VTEXTFIXED;
    844  1.54       kre 			if (vp->flags & VNOEXPORT)
    845  1.54       kre 				flags &= ~VEXPORT;
    846  1.54       kre 			if (flags & (VNOEXPORT | VUNSET))
    847  1.54       kre 				vp->flags &= ~VEXPORT;
    848  1.54       kre 			flags &= ~VNOEXPORT;
    849  1.35       dsl 			if (name[vp->name_len] == '=')
    850  1.54       kre 				setvareq(savestr(name), flags & ~VUNSET);
    851  1.54       kre 			else if (flags & VUNSET)
    852  1.54       kre 				unsetvar(name, 0);
    853  1.54       kre 			else
    854  1.54       kre 				vp->flags |= flags & (VUNSET|VEXPORT);
    855  1.57       kre 
    856  1.57       kre 			if (vp == &line_num) {
    857  1.57       kre 				if (name[vp->name_len] == '=')
    858  1.57       kre 					funclinebase = funclineabs -1;
    859  1.57       kre 				else
    860  1.57       kre 					funclinebase = 0;
    861  1.57       kre 			}
    862   1.1       cgd 		}
    863   1.1       cgd 	}
    864   1.1       cgd 	lvp->vp = vp;
    865   1.1       cgd 	lvp->next = localvars;
    866   1.1       cgd 	localvars = lvp;
    867   1.1       cgd 	INTON;
    868   1.1       cgd }
    869   1.1       cgd 
    870   1.1       cgd 
    871   1.1       cgd /*
    872   1.1       cgd  * Called after a function returns.
    873   1.1       cgd  */
    874   1.1       cgd 
    875   1.1       cgd void
    876  1.30  christos poplocalvars(void)
    877  1.30  christos {
    878   1.1       cgd 	struct localvar *lvp;
    879   1.1       cgd 	struct var *vp;
    880   1.1       cgd 
    881   1.1       cgd 	while ((lvp = localvars) != NULL) {
    882   1.1       cgd 		localvars = lvp->next;
    883   1.1       cgd 		vp = lvp->vp;
    884  1.32       dsl 		TRACE(("poplocalvar %s", vp ? vp->text : "-"));
    885   1.1       cgd 		if (vp == NULL) {	/* $- saved */
    886  1.30  christos 			memcpy(optlist, lvp->text, sizeof_optlist);
    887   1.1       cgd 			ckfree(lvp->text);
    888  1.57       kre 			optschanged();
    889   1.1       cgd 		} else if ((lvp->flags & (VUNSET|VSTRFIXED)) == VUNSET) {
    890  1.30  christos 			(void)unsetvar(vp->text, 0);
    891   1.1       cgd 		} else {
    892  1.57       kre 			if (vp->func && (vp->flags & (VNOFUNC|VFUNCREF)) == 0)
    893  1.35       dsl 				(*vp->func)(lvp->text + vp->name_len + 1);
    894   1.1       cgd 			if ((vp->flags & VTEXTFIXED) == 0)
    895   1.1       cgd 				ckfree(vp->text);
    896   1.1       cgd 			vp->flags = lvp->flags;
    897   1.1       cgd 			vp->text = lvp->text;
    898   1.1       cgd 		}
    899   1.1       cgd 		ckfree(lvp);
    900   1.1       cgd 	}
    901   1.1       cgd }
    902   1.1       cgd 
    903   1.1       cgd 
    904  1.10       cgd int
    905  1.30  christos setvarcmd(int argc, char **argv)
    906  1.10       cgd {
    907   1.1       cgd 	if (argc <= 2)
    908   1.1       cgd 		return unsetcmd(argc, argv);
    909   1.1       cgd 	else if (argc == 3)
    910   1.1       cgd 		setvar(argv[1], argv[2], 0);
    911   1.1       cgd 	else
    912   1.1       cgd 		error("List assignment not implemented");
    913   1.1       cgd 	return 0;
    914   1.1       cgd }
    915   1.1       cgd 
    916   1.1       cgd 
    917   1.1       cgd /*
    918   1.1       cgd  * The unset builtin command.  We unset the function before we unset the
    919   1.1       cgd  * variable to allow a function to be unset when there is a readonly variable
    920   1.1       cgd  * with the same name.
    921   1.1       cgd  */
    922   1.1       cgd 
    923  1.10       cgd int
    924  1.30  christos unsetcmd(int argc, char **argv)
    925  1.10       cgd {
    926   1.1       cgd 	char **ap;
    927   1.5       jtc 	int i;
    928   1.5       jtc 	int flg_func = 0;
    929   1.5       jtc 	int flg_var = 0;
    930  1.54       kre 	int flg_x = 0;
    931   1.5       jtc 	int ret = 0;
    932   1.5       jtc 
    933  1.54       kre 	while ((i = nextopt("efvx")) != '\0') {
    934  1.54       kre 		switch (i) {
    935  1.54       kre 		case 'f':
    936   1.5       jtc 			flg_func = 1;
    937  1.54       kre 			break;
    938  1.54       kre 		case 'e':
    939  1.54       kre 		case 'x':
    940  1.54       kre 			flg_x = (2 >> (i == 'e'));
    941  1.54       kre 			/* FALLTHROUGH */
    942  1.54       kre 		case 'v':
    943  1.54       kre 			flg_var = 1;
    944  1.54       kre 			break;
    945  1.54       kre 		}
    946   1.5       jtc 	}
    947  1.54       kre 
    948   1.5       jtc 	if (flg_func == 0 && flg_var == 0)
    949   1.5       jtc 		flg_var = 1;
    950  1.15  christos 
    951   1.5       jtc 	for (ap = argptr; *ap ; ap++) {
    952   1.5       jtc 		if (flg_func)
    953   1.5       jtc 			ret |= unsetfunc(*ap);
    954   1.5       jtc 		if (flg_var)
    955  1.54       kre 			ret |= unsetvar(*ap, flg_x);
    956   1.1       cgd 	}
    957   1.5       jtc 	return ret;
    958   1.1       cgd }
    959   1.1       cgd 
    960   1.1       cgd 
    961   1.1       cgd /*
    962   1.1       cgd  * Unset the specified variable.
    963   1.1       cgd  */
    964   1.1       cgd 
    965  1.14  christos int
    966  1.30  christos unsetvar(const char *s, int unexport)
    967  1.30  christos {
    968   1.1       cgd 	struct var **vpp;
    969   1.1       cgd 	struct var *vp;
    970   1.1       cgd 
    971  1.35       dsl 	vp = find_var(s, &vpp, NULL);
    972  1.35       dsl 	if (vp == NULL)
    973  1.43  christos 		return 0;
    974  1.35       dsl 
    975  1.54       kre 	if (vp->flags & VREADONLY && !(unexport & 1))
    976  1.43  christos 		return 1;
    977  1.35       dsl 
    978  1.35       dsl 	INTOFF;
    979  1.54       kre 	if (unexport & 1) {
    980  1.35       dsl 		vp->flags &= ~VEXPORT;
    981  1.35       dsl 	} else {
    982  1.35       dsl 		if (vp->text[vp->name_len + 1] != '\0')
    983  1.35       dsl 			setvar(s, nullstr, 0);
    984  1.54       kre 		if (!(unexport & 2))
    985  1.54       kre 			vp->flags &= ~VEXPORT;
    986  1.35       dsl 		vp->flags |= VUNSET;
    987  1.54       kre 		if ((vp->flags&(VEXPORT|VSTRFIXED|VREADONLY|VNOEXPORT)) == 0) {
    988  1.35       dsl 			if ((vp->flags & VTEXTFIXED) == 0)
    989  1.35       dsl 				ckfree(vp->text);
    990  1.35       dsl 			*vpp = vp->next;
    991  1.35       dsl 			ckfree(vp);
    992   1.1       cgd 		}
    993   1.1       cgd 	}
    994  1.35       dsl 	INTON;
    995  1.35       dsl 	return 0;
    996   1.1       cgd }
    997   1.1       cgd 
    998   1.1       cgd 
    999   1.1       cgd /*
   1000   1.1       cgd  * Returns true if the two strings specify the same varable.  The first
   1001   1.1       cgd  * variable name is terminated by '='; the second may be terminated by
   1002   1.1       cgd  * either '=' or '\0'.
   1003   1.1       cgd  */
   1004   1.1       cgd 
   1005   1.1       cgd STATIC int
   1006  1.35       dsl strequal(const char *p, const char *q)
   1007  1.30  christos {
   1008   1.1       cgd 	while (*p == *q++) {
   1009   1.1       cgd 		if (*p++ == '=')
   1010   1.1       cgd 			return 1;
   1011   1.1       cgd 	}
   1012   1.1       cgd 	if (*p == '=' && *(q - 1) == '\0')
   1013   1.1       cgd 		return 1;
   1014   1.1       cgd 	return 0;
   1015   1.1       cgd }
   1016  1.35       dsl 
   1017  1.35       dsl /*
   1018  1.35       dsl  * Search for a variable.
   1019  1.35       dsl  * 'name' may be terminated by '=' or a NUL.
   1020  1.35       dsl  * vppp is set to the pointer to vp, or the list head if vp isn't found
   1021  1.35       dsl  * lenp is set to the number of charactets in 'name'
   1022  1.35       dsl  */
   1023  1.35       dsl 
   1024  1.35       dsl STATIC struct var *
   1025  1.35       dsl find_var(const char *name, struct var ***vppp, int *lenp)
   1026  1.35       dsl {
   1027  1.35       dsl 	unsigned int hashval;
   1028  1.35       dsl 	int len;
   1029  1.35       dsl 	struct var *vp, **vpp;
   1030  1.35       dsl 	const char *p = name;
   1031  1.35       dsl 
   1032  1.35       dsl 	hashval = 0;
   1033  1.35       dsl 	while (*p && *p != '=')
   1034  1.35       dsl 		hashval = 2 * hashval + (unsigned char)*p++;
   1035  1.35       dsl 	len = p - name;
   1036  1.35       dsl 
   1037  1.35       dsl 	if (lenp)
   1038  1.35       dsl 		*lenp = len;
   1039  1.35       dsl 	vpp = &vartab[hashval % VTABSIZE];
   1040  1.35       dsl 	if (vppp)
   1041  1.35       dsl 		*vppp = vpp;
   1042  1.35       dsl 
   1043  1.35       dsl 	for (vp = *vpp ; vp ; vpp = &vp->next, vp = *vpp) {
   1044  1.35       dsl 		if (vp->name_len != len)
   1045  1.35       dsl 			continue;
   1046  1.35       dsl 		if (memcmp(vp->text, name, len) != 0)
   1047  1.35       dsl 			continue;
   1048  1.35       dsl 		if (vppp)
   1049  1.35       dsl 			*vppp = vpp;
   1050  1.35       dsl 		return vp;
   1051  1.35       dsl 	}
   1052  1.35       dsl 	return NULL;
   1053  1.35       dsl }
   1054  1.57       kre 
   1055  1.57       kre char *
   1056  1.57       kre get_lineno(struct var *vp)
   1057  1.57       kre {
   1058  1.57       kre 	static char lineno_buf[8 + 14];
   1059  1.57       kre 	int ln = line_number;
   1060  1.57       kre 
   1061  1.57       kre 	if (vp->flags & VUNSET)
   1062  1.57       kre 		return NULL;
   1063  1.57       kre 
   1064  1.57       kre 	ln -= funclinebase;
   1065  1.57       kre 	snprintf(lineno_buf, sizeof(lineno_buf), "LINENO=%d", ln);;
   1066  1.57       kre 	return lineno_buf;
   1067  1.57       kre }
   1068