xref: /dragonfly/contrib/tcsh-6/sh.set.c (revision 2038fb68)
1 /* $Header: /p/tcsh/cvsroot/tcsh/sh.set.c,v 3.70 2006/08/24 20:56:31 christos Exp $ */
2 /*
3  * sh.set.c: Setting and Clearing of variables
4  */
5 /*-
6  * Copyright (c) 1980, 1991 The Regents of the University of California.
7  * All rights reserved.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions and the following disclaimer.
14  * 2. Redistributions in binary form must reproduce the above copyright
15  *    notice, this list of conditions and the following disclaimer in the
16  *    documentation and/or other materials provided with the distribution.
17  * 3. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  */
33 #include "sh.h"
34 
35 RCSID("$tcsh: sh.set.c,v 3.70 2006/08/24 20:56:31 christos Exp $")
36 
37 #include "ed.h"
38 #include "tw.h"
39 
40 #ifdef HAVE_NL_LANGINFO
41 #include <langinfo.h>
42 #endif
43 
44 extern int GotTermCaps;
45 int numeof = 0;
46 
47 static	void		 update_vars	(Char *);
48 static	Char		*getinx		(Char *, int *);
49 static	void		 asx		(Char *, int, Char *);
50 static	struct varent 	*getvx		(Char *, int);
51 static	Char		*xset		(Char *, Char ***);
52 static	Char		*operate	(int, Char *, Char *);
53 static	void	 	 putn1		(unsigned);
54 static	struct varent	*madrof		(Char *, struct varent *);
55 static	void		 unsetv1	(struct varent *);
56 static	void		 balance	(struct varent *, int, int);
57 
58 /*
59  * C Shell
60  */
61 
62 static void
63 update_vars(Char *vp)
64 {
65     if (eq(vp, STRpath)) {
66 	struct varent *p = adrof(STRpath);
67 	if (p == NULL)
68 	    stderror(ERR_NAME | ERR_UNDVAR);
69 	else {
70 	    exportpath(p->vec);
71 	    dohash(NULL, NULL);
72 	}
73     }
74     else if (eq(vp, STRhistchars)) {
75 	Char *pn = varval(vp);
76 
77 	HIST = *pn++;
78 	HISTSUB = *pn;
79     }
80     else if (eq(vp, STRpromptchars)) {
81 	Char *pn = varval(vp);
82 
83 	PRCH = *pn++;
84 	PRCHROOT = *pn;
85     }
86     else if (eq(vp, STRhistlit)) {
87 	HistLit = 1;
88     }
89     else if (eq(vp, STRuser)) {
90 	tsetenv(STRKUSER, varval(vp));
91 	tsetenv(STRLOGNAME, varval(vp));
92     }
93     else if (eq(vp, STRgroup)) {
94 	tsetenv(STRKGROUP, varval(vp));
95     }
96     else if (eq(vp, STRwordchars)) {
97 	word_chars = varval(vp);
98     }
99     else if (eq(vp, STRloginsh)) {
100 	loginsh = 1;
101     }
102     else if (eq(vp, STRsymlinks)) {
103 	Char *pn = varval(vp);
104 
105 	if (eq(pn, STRignore))
106 	    symlinks = SYM_IGNORE;
107 	else if (eq(pn, STRexpand))
108 	    symlinks = SYM_EXPAND;
109 	else if (eq(pn, STRchase))
110 	    symlinks = SYM_CHASE;
111 	else
112 	    symlinks = 0;
113     }
114     else if (eq(vp, STRterm)) {
115 	Char *cp = varval(vp);
116 	tsetenv(STRKTERM, cp);
117 #ifdef DOESNT_WORK_RIGHT
118 	cp = getenv("TERMCAP");
119 	if (cp && (*cp != '/'))	/* if TERMCAP and not a path */
120 	    Unsetenv(STRTERMCAP);
121 #endif /* DOESNT_WORK_RIGHT */
122 	GotTermCaps = 0;
123 	if (noediting && Strcmp(cp, STRnetwork) != 0 &&
124 	    Strcmp(cp, STRunknown) != 0 && Strcmp(cp, STRdumb) != 0) {
125 	    editing = 1;
126 	    noediting = 0;
127 	    setNS(STRedit);
128 	}
129 	ed_Init();		/* reset the editor */
130     }
131     else if (eq(vp, STRhome)) {
132 	Char *cp, *canon;
133 
134 	cp = Strsave(varval(vp));	/* get the old value back */
135 	cleanup_push(cp, xfree);
136 
137 	/*
138 	 * convert to cononical pathname (possibly resolving symlinks)
139 	 */
140 	canon = dcanon(cp, cp);
141 	cleanup_ignore(cp);
142 	cleanup_until(cp);
143 	cleanup_push(canon, xfree);
144 
145 	setcopy(vp, canon, VAR_READWRITE);	/* have to save the new val */
146 
147 	/* and now mirror home with HOME */
148 	tsetenv(STRKHOME, canon);
149 	/* fix directory stack for new tilde home */
150 	dtilde();
151 	cleanup_until(canon);
152     }
153     else if (eq(vp, STRedit)) {
154 	editing = 1;
155 	noediting = 0;
156 	/* PWP: add more stuff in here later */
157     }
158     else if (eq(vp, STRshlvl)) {
159 	tsetenv(STRKSHLVL, varval(vp));
160     }
161     else if (eq(vp, STRignoreeof)) {
162 	Char *cp;
163 	numeof = 0;
164     	for ((cp = varval(STRignoreeof)); cp && *cp; cp++) {
165 	    if (!Isdigit(*cp)) {
166 		numeof = 0;
167 		break;
168 	    }
169 	    numeof = numeof * 10 + *cp - '0';
170 	}
171 	if (numeof <= 0) numeof = 26;	/* Sanity check */
172     }
173     else if (eq(vp, STRbackslash_quote)) {
174 	bslash_quote = 1;
175     }
176     else if (eq(vp, STRdirstack)) {
177 	dsetstack();
178     }
179     else if (eq(vp, STRrecognize_only_executables)) {
180 	tw_cmd_free();
181     }
182     else if (eq(vp, STRkillring)) {
183 	SetKillRing(getn(varval(vp)));
184     }
185 #ifndef HAVENOUTMP
186     else if (eq(vp, STRwatch)) {
187 	resetwatch();
188     }
189 #endif /* HAVENOUTMP */
190     else if (eq(vp, STRimplicitcd)) {
191 	implicit_cd = ((eq(varval(vp), STRverbose)) ? 2 : 1);
192     }
193 #ifdef COLOR_LS_F
194     else if (eq(vp, STRcolor)) {
195 	set_color_context();
196     }
197 #endif /* COLOR_LS_F */
198 #if defined(KANJI) && defined(SHORT_STRINGS) && defined(DSPMBYTE)
199     else if(eq(vp, CHECK_MBYTEVAR) || eq(vp, STRnokanji)) {
200 	update_dspmbyte_vars();
201     }
202 #endif
203 #ifdef NLS_CATALOGS
204     else if (eq(vp, STRcatalog)) {
205 	nlsclose();
206 	nlsinit();
207     }
208 #if defined(FILEC) && defined(TIOCSTI)
209     else if (eq(vp, STRfilec))
210 	filec = 1;
211 #endif
212 #endif /* NLS_CATALOGS */
213 }
214 
215 
216 /*ARGSUSED*/
217 void
218 doset(Char **v, struct command *c)
219 {
220     Char *p;
221     Char   *vp, op;
222     Char  **vecp;
223     int    hadsub;
224     int     subscr;
225     int	    flags = VAR_READWRITE;
226     int    first_match = 0;
227     int    last_match = 0;
228     int    changed = 0;
229 
230     USE(c);
231     v++;
232     do {
233 	changed = 0;
234 	/*
235 	 * Readonly addition From: Tim P. Starrin <noid@cyborg.larc.nasa.gov>
236 	 */
237 	if (*v && eq(*v, STRmr)) {
238 	    flags = VAR_READONLY;
239 	    v++;
240 	    changed = 1;
241 	}
242 	if (*v && eq(*v, STRmf) && !last_match) {
243 	    first_match = 1;
244 	    v++;
245 	    changed = 1;
246 	}
247 	if (*v && eq(*v, STRml) && !first_match) {
248 	    last_match = 1;
249 	    v++;
250 	    changed = 1;
251 	}
252     } while(changed);
253     p = *v++;
254     if (p == 0) {
255 	plist(&shvhed, flags);
256 	return;
257     }
258     do {
259 	hadsub = 0;
260 	vp = p;
261 	if (letter(*p))
262 	    for (; alnum(*p); p++)
263 		continue;
264 	if (vp == p || !letter(*vp))
265 	    stderror(ERR_NAME | ERR_VARBEGIN);
266 	if (*p == '[') {
267 	    hadsub++;
268 	    p = getinx(p, &subscr);
269 	}
270 	if ((op = *p) != 0) {
271 	    *p++ = 0;
272 	    if (*p == 0 && *v && **v == '(')
273 		p = *v++;
274 	}
275 	else if (*v && eq(*v, STRequal)) {
276 	    op = '=', v++;
277 	    if (*v)
278 		p = *v++;
279 	}
280 	if (op && op != '=')
281 	    stderror(ERR_NAME | ERR_SYNTAX);
282 	if (eq(p, STRLparen)) {
283 	    Char **e = v;
284 
285 	    if (hadsub)
286 		stderror(ERR_NAME | ERR_SYNTAX);
287 	    for (;;) {
288 		if (!*e)
289 		    stderror(ERR_NAME | ERR_MISSING, ')');
290 		if (**e == ')')
291 		    break;
292 		e++;
293 	    }
294 	    p = *e;
295 	    *e = 0;
296 	    vecp = saveblk(v);
297 	    if (first_match)
298 	       flags |= VAR_FIRST;
299 	    else if (last_match)
300 	       flags |= VAR_LAST;
301 
302 	    set1(vp, vecp, &shvhed, flags);
303 	    *e = p;
304 	    v = e + 1;
305 	}
306 	else if (hadsub) {
307 	    Char *copy;
308 
309 	    copy = Strsave(p);
310 	    cleanup_push(copy, xfree);
311 	    asx(vp, subscr, copy);
312 	    cleanup_ignore(copy);
313 	    cleanup_until(copy);
314 	}
315 	else
316 	    setv(vp, Strsave(p), flags);
317 	update_vars(vp);
318     } while ((p = *v++) != NULL);
319 }
320 
321 static Char *
322 getinx(Char *cp, int *ip)
323 {
324     *ip = 0;
325     *cp++ = 0;
326     while (*cp && Isdigit(*cp))
327 	*ip = *ip * 10 + *cp++ - '0';
328     if (*cp++ != ']')
329 	stderror(ERR_NAME | ERR_SUBSCRIPT);
330     return (cp);
331 }
332 
333 static void
334 asx(Char *vp, int subscr, Char *p)
335 {
336     struct varent *v = getvx(vp, subscr);
337     Char *prev;
338 
339     if (v->v_flags & VAR_READONLY)
340 	stderror(ERR_READONLY|ERR_NAME, v->v_name);
341     prev = v->vec[subscr - 1];
342     cleanup_push(prev, xfree);
343     v->vec[subscr - 1] = globone(p, G_APPEND);
344     cleanup_until(prev);
345 }
346 
347 static struct varent *
348 getvx(Char *vp, int subscr)
349 {
350     struct varent *v = adrof(vp);
351 
352     if (v == 0)
353 	udvar(vp);
354     if (subscr < 1 || subscr > blklen(v->vec))
355 	stderror(ERR_NAME | ERR_RANGE);
356     return (v);
357 }
358 
359 /*ARGSUSED*/
360 void
361 dolet(Char **v, struct command *dummy)
362 {
363     Char *p;
364     Char   *vp, c, op;
365     int    hadsub;
366     int     subscr;
367 
368     USE(dummy);
369     v++;
370     p = *v++;
371     if (p == 0) {
372 	prvars();
373 	return;
374     }
375     do {
376 	hadsub = 0;
377 	vp = p;
378 	if (letter(*p))
379 	    for (; alnum(*p); p++)
380 		continue;
381 	if (vp == p || !letter(*vp))
382 	    stderror(ERR_NAME | ERR_VARBEGIN);
383 	if (*p == '[') {
384 	    hadsub++;
385 	    p = getinx(p, &subscr);
386 	}
387 	if (*p == 0 && *v)
388 	    p = *v++;
389 	if ((op = *p) != 0)
390 	    *p++ = 0;
391 	else
392 	    stderror(ERR_NAME | ERR_ASSIGN);
393 
394 	/*
395 	 * if there is no expression after the '=' then print a "Syntax Error"
396 	 * message - strike
397 	 */
398 	if (*p == '\0' && *v == NULL)
399 	    stderror(ERR_NAME | ERR_ASSIGN);
400 
401 	vp = Strsave(vp);
402 	cleanup_push(vp, xfree);
403 	if (op == '=') {
404 	    c = '=';
405 	    p = xset(p, &v);
406 	}
407 	else {
408 	    c = *p++;
409 	    if (any("+-", c)) {
410 		if (c != op || *p)
411 		    stderror(ERR_NAME | ERR_UNKNOWNOP);
412 		p = Strsave(STR1);
413 	    }
414 	    else {
415 		if (any("<>", op)) {
416 		    if (c != op)
417 			stderror(ERR_NAME | ERR_UNKNOWNOP);
418 		    stderror(ERR_NAME | ERR_SYNTAX);
419 		}
420 		if (c != '=')
421 		    stderror(ERR_NAME | ERR_UNKNOWNOP);
422 		p = xset(p, &v);
423 	    }
424 	}
425 	cleanup_push(p, xfree);
426 	if (op == '=') {
427 	    if (hadsub)
428 		asx(vp, subscr, p);
429 	    else
430 		setv(vp, p, VAR_READWRITE);
431 	    cleanup_ignore(p);
432 	}
433 	else if (hadsub) {
434 	    struct varent *gv = getvx(vp, subscr);
435 	    Char *val;
436 
437 	    val = operate(op, gv->vec[subscr - 1], p);
438 	    cleanup_push(val, xfree);
439 	    asx(vp, subscr, val);
440 	    cleanup_ignore(val);
441 	    cleanup_until(val);
442 	}
443 	else {
444 	    Char *val;
445 
446 	    val = operate(op, varval(vp), p);
447 	    cleanup_push(val, xfree);
448 	    setv(vp, val, VAR_READWRITE);
449 	    cleanup_ignore(val);
450 	    cleanup_until(val);
451 	}
452 	update_vars(vp);
453 	cleanup_until(vp);
454     } while ((p = *v++) != NULL);
455 }
456 
457 static Char *
458 xset(Char *cp, Char ***vp)
459 {
460     Char *dp;
461 
462     if (*cp) {
463 	dp = Strsave(cp);
464 	--(*vp);
465 	xfree(** vp);
466 	**vp = dp;
467     }
468     return (putn(expr(vp)));
469 }
470 
471 static Char *
472 operate(int op, Char *vp, Char *p)
473 {
474     Char    opr[2];
475     Char   *vec[5];
476     Char **v = vec;
477     Char  **vecp = v;
478     int i;
479 
480     if (op != '=') {
481 	if (*vp)
482 	    *v++ = vp;
483 	opr[0] = op;
484 	opr[1] = 0;
485 	*v++ = opr;
486 	if (op == '<' || op == '>')
487 	    *v++ = opr;
488     }
489     *v++ = p;
490     *v++ = 0;
491     i = expr(&vecp);
492     if (*vecp)
493 	stderror(ERR_NAME | ERR_EXPRESSION);
494     return (putn(i));
495 }
496 
497 static Char *putp;
498 
499 Char   *
500 putn(int n)
501 {
502     Char nbuf[(CHAR_BIT * sizeof (n) + 2) / 3 + 2]; /* Enough even for octal */
503 
504     putp = nbuf;
505     if (n < 0) {
506 	n = -n;
507 	*putp++ = '-';
508     }
509     putn1(n);
510     *putp = 0;
511     return (Strsave(nbuf));
512 }
513 
514 static void
515 putn1(unsigned n)
516 {
517     if (n > 9)
518 	putn1(n / 10);
519     *putp++ = n % 10 + '0';
520 }
521 
522 int
523 getn(Char *cp)
524 {
525     int n;
526     int     sign;
527 
528     if (!cp)			/* PWP: extra error checking */
529 	stderror(ERR_NAME | ERR_BADNUM);
530 
531     sign = 0;
532     if (cp[0] == '+' && cp[1])
533 	cp++;
534     if (*cp == '-') {
535 	sign++;
536 	cp++;
537 	if (!Isdigit(*cp))
538 	    stderror(ERR_NAME | ERR_BADNUM);
539     }
540     n = 0;
541     while (Isdigit(*cp))
542 	n = n * 10 + *cp++ - '0';
543     if (*cp)
544 	stderror(ERR_NAME | ERR_BADNUM);
545     return (sign ? -n : n);
546 }
547 
548 Char   *
549 value1(Char *var, struct varent *head)
550 {
551     struct varent *vp;
552 
553     if (!var || !head)		/* PWP: extra error checking */
554 	return (STRNULL);
555 
556     vp = adrof1(var, head);
557     return ((vp == NULL || vp->vec == NULL || vp->vec[0] == NULL) ?
558 	STRNULL : vp->vec[0]);
559 }
560 
561 static struct varent *
562 madrof(Char *pat, struct varent *vp)
563 {
564     struct varent *vp1;
565 
566     for (vp = vp->v_left; vp; vp = vp->v_right) {
567 	if (vp->v_left && (vp1 = madrof(pat, vp)) != NULL)
568 	    return vp1;
569 	if (Gmatch(vp->v_name, pat))
570 	    return vp;
571     }
572     return vp;
573 }
574 
575 struct varent *
576 adrof1(const Char *name, struct varent *v)
577 {
578     int cmp;
579 
580     v = v->v_left;
581     while (v && ((cmp = *name - *v->v_name) != 0 ||
582 		 (cmp = Strcmp(name, v->v_name)) != 0))
583 	if (cmp < 0)
584 	    v = v->v_left;
585 	else
586 	    v = v->v_right;
587     return v;
588 }
589 
590 void
591 setcopy(const Char *var, const Char *val, int flags)
592 {
593     Char *copy;
594 
595     copy = Strsave(val);
596     cleanup_push(copy, xfree);
597     setv(var, copy, flags);
598     cleanup_ignore(copy);
599     cleanup_until(copy);
600 }
601 
602 /*
603  * The caller is responsible for putting value in a safe place
604  */
605 void
606 setv(const Char *var, Char *val, int flags)
607 {
608     Char **vec = xmalloc(2 * sizeof(Char **));
609 
610     vec[0] = val;
611     vec[1] = 0;
612     set1(var, vec, &shvhed, flags);
613 }
614 
615 void
616 set1(const Char *var, Char **vec, struct varent *head, int flags)
617 {
618     Char **oldv = vec;
619 
620     if ((flags & VAR_NOGLOB) == 0) {
621 	int gflag;
622 
623 	gflag = tglob(oldv);
624 	if (gflag) {
625 	    vec = globall(oldv, gflag);
626 	    if (vec == 0) {
627 		blkfree(oldv);
628 		stderror(ERR_NAME | ERR_NOMATCH);
629 	    }
630 	    blkfree(oldv);
631 	}
632     }
633     /*
634      * Uniqueness addition from: Michael Veksler <mveksler@vnet.ibm.com>
635      */
636     if ( flags & (VAR_FIRST | VAR_LAST) ) {
637 	/*
638 	 * Code for -f (VAR_FIRST) and -l (VAR_LAST) options.
639 	 * Method:
640 	 *  Delete all duplicate words leaving "holes" in the word array (vec).
641 	 *  Then remove the "holes", keeping the order of the words unchanged.
642 	 */
643 	if (vec && vec[0] && vec[1]) { /* more than one word ? */
644 	    int i, j;
645 	    int num_items;
646 
647 	    for (num_items = 0; vec[num_items]; num_items++)
648 	        continue;
649 	    if (flags & VAR_FIRST) {
650 		/* delete duplications, keeping first occurance */
651 		for (i = 1; i < num_items; i++)
652 		    for (j = 0; j < i; j++)
653 			/* If have earlier identical item, remove i'th item */
654 			if (vec[i] && vec[j] && Strcmp(vec[j], vec[i]) == 0) {
655 			    xfree(vec[i]);
656 			    vec[i] = NULL;
657 			    break;
658 			}
659 	    } else if (flags & VAR_LAST) {
660 	      /* delete duplications, keeping last occurance */
661 		for (i = 0; i < num_items - 1; i++)
662 		    for (j = i + 1; j < num_items; j++)
663 			/* If have later identical item, remove i'th item */
664 			if (vec[i] && vec[j] && Strcmp(vec[j], vec[i]) == 0) {
665 			    /* remove identical item (the first) */
666 			    xfree(vec[i]);
667 			    vec[i] = NULL;
668 			}
669 	    }
670 	    /* Compress items - remove empty items */
671 	    for (j = i = 0; i < num_items; i++)
672 	       if (vec[i])
673 		  vec[j++] = vec[i];
674 
675 	    /* NULL-fy remaining items */
676 	    for (; j < num_items; j++)
677 		 vec[j] = NULL;
678 	}
679 	/* don't let the attribute propagate */
680 	flags &= ~(VAR_FIRST|VAR_LAST);
681     }
682     setq(var, vec, head, flags);
683 }
684 
685 
686 void
687 setq(const Char *name, Char **vec, struct varent *p, int flags)
688 {
689     struct varent *c;
690     int f;
691 
692     f = 0;			/* tree hangs off the header's left link */
693     while ((c = p->v_link[f]) != 0) {
694 	if ((f = *name - *c->v_name) == 0 &&
695 	    (f = Strcmp(name, c->v_name)) == 0) {
696 	    if (c->v_flags & VAR_READONLY)
697 		stderror(ERR_READONLY|ERR_NAME, c->v_name);
698 	    blkfree(c->vec);
699 	    c->v_flags = flags;
700 	    trim(c->vec = vec);
701 	    return;
702 	}
703 	p = c;
704 	f = f > 0;
705     }
706     p->v_link[f] = c = xmalloc(sizeof(struct varent));
707     c->v_name = Strsave(name);
708     c->v_flags = flags;
709     c->v_bal = 0;
710     c->v_left = c->v_right = 0;
711     c->v_parent = p;
712     balance(p, f, 0);
713     trim(c->vec = vec);
714 }
715 
716 /*ARGSUSED*/
717 void
718 unset(Char **v, struct command *c)
719 {
720     int did_roe, did_edit;
721 
722     USE(c);
723     did_roe = adrof(STRrecognize_only_executables) != NULL;
724     did_edit = adrof(STRedit) != NULL;
725     unset1(v, &shvhed);
726 
727 #if defined(FILEC) && defined(TIOCSTI)
728     if (adrof(STRfilec) == 0)
729 	filec = 0;
730 #endif /* FILEC && TIOCSTI */
731 
732     if (adrof(STRhistchars) == 0) {
733 	HIST = '!';
734 	HISTSUB = '^';
735     }
736     if (adrof(STRignoreeof) == 0)
737 	numeof = 0;
738     if (adrof(STRpromptchars) == 0) {
739 	PRCH = '>';
740 	PRCHROOT = '#';
741     }
742     if (adrof(STRhistlit) == 0)
743 	HistLit = 0;
744     if (adrof(STRloginsh) == 0)
745 	loginsh = 0;
746     if (adrof(STRwordchars) == 0)
747 	word_chars = STR_WORD_CHARS;
748     if (adrof(STRedit) == 0)
749 	editing = 0;
750     if (adrof(STRbackslash_quote) == 0)
751 	bslash_quote = 0;
752     if (adrof(STRsymlinks) == 0)
753 	symlinks = 0;
754     if (adrof(STRimplicitcd) == 0)
755 	implicit_cd = 0;
756     if (adrof(STRkillring) == 0)
757 	SetKillRing(0);
758     if (did_edit && noediting && adrof(STRedit) == 0)
759 	noediting = 0;
760     if (did_roe && adrof(STRrecognize_only_executables) == 0)
761 	tw_cmd_free();
762 #ifdef COLOR_LS_F
763     if (adrof(STRcolor) == 0)
764 	set_color_context();
765 #endif /* COLOR_LS_F */
766 #if defined(KANJI) && defined(SHORT_STRINGS) && defined(DSPMBYTE)
767     update_dspmbyte_vars();
768 #endif
769 #ifdef NLS_CATALOGS
770     nlsclose();
771     nlsinit();
772 #endif /* NLS_CATALOGS */
773 }
774 
775 void
776 unset1(Char *v[], struct varent *head)
777 {
778     struct varent *vp;
779     int cnt;
780 
781     while (*++v) {
782 	cnt = 0;
783 	while ((vp = madrof(*v, head)) != NULL)
784 	    if (vp->v_flags & VAR_READONLY)
785 		stderror(ERR_READONLY|ERR_NAME, vp->v_name);
786 	    else
787 		unsetv1(vp), cnt++;
788 	if (cnt == 0)
789 	    setname(short2str(*v));
790     }
791 }
792 
793 void
794 unsetv(Char *var)
795 {
796     struct varent *vp;
797 
798     if ((vp = adrof1(var, &shvhed)) == 0)
799 	udvar(var);
800     unsetv1(vp);
801 }
802 
803 static void
804 unsetv1(struct varent *p)
805 {
806     struct varent *c, *pp;
807     int f;
808 
809     /*
810      * Free associated memory first to avoid complications.
811      */
812     blkfree(p->vec);
813     xfree(p->v_name);
814     /*
815      * If p is missing one child, then we can move the other into where p is.
816      * Otherwise, we find the predecessor of p, which is guaranteed to have no
817      * right child, copy it into p, and move it's left child into it.
818      */
819     if (p->v_right == 0)
820 	c = p->v_left;
821     else if (p->v_left == 0)
822 	c = p->v_right;
823     else {
824 	for (c = p->v_left; c->v_right; c = c->v_right)
825 	    continue;
826 	p->v_name = c->v_name;
827 	p->v_flags = c->v_flags;
828 	p->vec = c->vec;
829 	p = c;
830 	c = p->v_left;
831     }
832 
833     /*
834      * Move c into where p is.
835      */
836     pp = p->v_parent;
837     f = pp->v_right == p;
838     if ((pp->v_link[f] = c) != 0)
839 	c->v_parent = pp;
840     /*
841      * Free the deleted node, and rebalance.
842      */
843     xfree(p);
844     balance(pp, f, 1);
845 }
846 
847 void
848 setNS(Char *cp)
849 {
850     setcopy(cp, STRNULL, VAR_READWRITE);
851 }
852 
853 /*ARGSUSED*/
854 void
855 shift(Char **v, struct command *c)
856 {
857     struct varent *argv;
858     Char *name;
859 
860     USE(c);
861     v++;
862     name = *v;
863     if (name == 0)
864 	name = STRargv;
865     else
866 	(void) strip(name);
867     argv = adrof(name);
868     if (argv == NULL || argv->vec == NULL)
869 	udvar(name);
870     if (argv->vec[0] == 0)
871 	stderror(ERR_NAME | ERR_NOMORE);
872     lshift(argv->vec, 1);
873     update_vars(name);
874 }
875 
876 void
877 exportpath(Char **val)
878 {
879     struct Strbuf buf = Strbuf_INIT;
880     Char    	*exppath;
881 
882     if (val)
883 	while (*val) {
884 	    Strbuf_append(&buf, *val++);
885 	    if (*val == 0 || eq(*val, STRRparen))
886 		break;
887 	    Strbuf_append1(&buf, PATHSEP);
888 	}
889     exppath = Strbuf_finish(&buf);
890     cleanup_push(exppath, xfree);
891     tsetenv(STRKPATH, exppath);
892     cleanup_until(exppath);
893 }
894 
895 #ifndef lint
896  /*
897   * Lint thinks these have null effect
898   */
899  /* macros to do single rotations on node p */
900 # define rright(p) (\
901 	t = (p)->v_left,\
902 	(t)->v_parent = (p)->v_parent,\
903 	(((p)->v_left = t->v_right) != NULL) ?\
904 	    (t->v_right->v_parent = (p)) : 0,\
905 	(t->v_right = (p))->v_parent = t,\
906 	(p) = t)
907 # define rleft(p) (\
908 	t = (p)->v_right,\
909 	((t)->v_parent = (p)->v_parent,\
910 	((p)->v_right = t->v_left) != NULL) ? \
911 		(t->v_left->v_parent = (p)) : 0,\
912 	(t->v_left = (p))->v_parent = t,\
913 	(p) = t)
914 #else
915 static struct varent *
916 rleft(struct varent *p)
917 {
918     return (p);
919 }
920 static struct varent *
921 rright(struct varent *p)
922 {
923     return (p);
924 }
925 
926 #endif /* ! lint */
927 
928 
929 /*
930  * Rebalance a tree, starting at p and up.
931  * F == 0 means we've come from p's left child.
932  * D == 1 means we've just done a delete, otherwise an insert.
933  */
934 static void
935 balance(struct varent *p, int f, int d)
936 {
937     struct varent *pp;
938 
939 #ifndef lint
940     struct varent *t;	/* used by the rotate macros */
941 #endif /* !lint */
942     int ff;
943 #ifdef lint
944     ff = 0;	/* Sun's lint is dumb! */
945 #endif
946 
947     /*
948      * Ok, from here on, p is the node we're operating on; pp is it's parent; f
949      * is the branch of p from which we have come; ff is the branch of pp which
950      * is p.
951      */
952     for (; (pp = p->v_parent) != 0; p = pp, f = ff) {
953 	ff = pp->v_right == p;
954 	if (f ^ d) {		/* right heavy */
955 	    switch (p->v_bal) {
956 	    case -1:		/* was left heavy */
957 		p->v_bal = 0;
958 		break;
959 	    case 0:		/* was balanced */
960 		p->v_bal = 1;
961 		break;
962 	    case 1:		/* was already right heavy */
963 		switch (p->v_right->v_bal) {
964 		case 1:	/* single rotate */
965 		    pp->v_link[ff] = rleft(p);
966 		    p->v_left->v_bal = 0;
967 		    p->v_bal = 0;
968 		    break;
969 		case 0:	/* single rotate */
970 		    pp->v_link[ff] = rleft(p);
971 		    p->v_left->v_bal = 1;
972 		    p->v_bal = -1;
973 		    break;
974 		case -1:	/* double rotate */
975 		    (void) rright(p->v_right);
976 		    pp->v_link[ff] = rleft(p);
977 		    p->v_left->v_bal =
978 			p->v_bal < 1 ? 0 : -1;
979 		    p->v_right->v_bal =
980 			p->v_bal > -1 ? 0 : 1;
981 		    p->v_bal = 0;
982 		    break;
983 		default:
984 		    break;
985 		}
986 		break;
987 	    default:
988 		break;
989 	    }
990 	}
991 	else {			/* left heavy */
992 	    switch (p->v_bal) {
993 	    case 1:		/* was right heavy */
994 		p->v_bal = 0;
995 		break;
996 	    case 0:		/* was balanced */
997 		p->v_bal = -1;
998 		break;
999 	    case -1:		/* was already left heavy */
1000 		switch (p->v_left->v_bal) {
1001 		case -1:	/* single rotate */
1002 		    pp->v_link[ff] = rright(p);
1003 		    p->v_right->v_bal = 0;
1004 		    p->v_bal = 0;
1005 		    break;
1006 		case 0:	/* single rotate */
1007 		    pp->v_link[ff] = rright(p);
1008 		    p->v_right->v_bal = -1;
1009 		    p->v_bal = 1;
1010 		    break;
1011 		case 1:	/* double rotate */
1012 		    (void) rleft(p->v_left);
1013 		    pp->v_link[ff] = rright(p);
1014 		    p->v_left->v_bal =
1015 			p->v_bal < 1 ? 0 : -1;
1016 		    p->v_right->v_bal =
1017 			p->v_bal > -1 ? 0 : 1;
1018 		    p->v_bal = 0;
1019 		    break;
1020 		default:
1021 		    break;
1022 		}
1023 		break;
1024 	    default:
1025 		break;
1026 	    }
1027 	}
1028 	/*
1029 	 * If from insert, then we terminate when p is balanced. If from
1030 	 * delete, then we terminate when p is unbalanced.
1031 	 */
1032 	if ((p->v_bal == 0) ^ d)
1033 	    break;
1034     }
1035 }
1036 
1037 void
1038 plist(struct varent *p, int what)
1039 {
1040     struct varent *c;
1041     int len;
1042 
1043     for (;;) {
1044 	while (p->v_left)
1045 	    p = p->v_left;
1046 x:
1047 	if (p->v_parent == 0)	/* is it the header? */
1048 	    break;
1049 	if ((p->v_flags & what) != 0) {
1050 	    if (setintr) {
1051 		int old_pintr_disabled;
1052 
1053 		pintr_push_enable(&old_pintr_disabled);
1054 		cleanup_until(&old_pintr_disabled);
1055 	    }
1056 	    len = blklen(p->vec);
1057 	    xprintf("%S\t", p->v_name);
1058 	    if (len != 1)
1059 		xputchar('(');
1060 	    blkpr(p->vec);
1061 	    if (len != 1)
1062 		xputchar(')');
1063 	    xputchar('\n');
1064 	}
1065 	if (p->v_right) {
1066 	    p = p->v_right;
1067 	    continue;
1068 	}
1069 	do {
1070 	    c = p;
1071 	    p = p->v_parent;
1072 	} while (p->v_right == c);
1073 	goto x;
1074     }
1075 }
1076 
1077 #if defined(KANJI) && defined(SHORT_STRINGS) && defined(DSPMBYTE)
1078 extern int dspmbyte_ls;
1079 
1080 void
1081 update_dspmbyte_vars(void)
1082 {
1083     int lp, iskcode;
1084     Char *dstr1;
1085     struct varent *vp;
1086 
1087     /* if variable "nokanji" is set, multi-byte display is disabled */
1088     if ((vp = adrof(CHECK_MBYTEVAR)) && !adrof(STRnokanji)) {
1089 	_enable_mbdisp = 1;
1090 	dstr1 = vp->vec[0];
1091 	if(eq (dstr1, STRsjis))
1092 	    iskcode = 1;
1093 	else if (eq(dstr1, STReuc))
1094 	    iskcode = 2;
1095 	else if (eq(dstr1, STRbig5))
1096 	    iskcode = 3;
1097 	else if (eq(dstr1, STRutf8))
1098 	    iskcode = 4;
1099 	else if ((dstr1[0] - '0') >= 0 && (dstr1[0] - '0') <= 3) {
1100 	    iskcode = 0;
1101 	}
1102 	else {
1103 	    xprintf(CGETS(18, 2,
1104 	       "Warning: unknown multibyte display; using default(euc(JP))\n"));
1105 	    iskcode = 2;
1106 	}
1107 	if (dstr1 && vp->vec[1] && eq(vp->vec[1], STRls))
1108 	  dspmbyte_ls = 1;
1109 	else
1110 	  dspmbyte_ls = 0;
1111 	for (lp = 0; lp < 256 && iskcode > 0; lp++) {
1112 	    switch (iskcode) {
1113 	    case 1:
1114 		/* Shift-JIS */
1115 		_cmap[lp] = _cmap_mbyte[lp];
1116 		_mbmap[lp] = _mbmap_sjis[lp];
1117 		break;
1118 	    case 2:
1119 		/* 2 ... euc */
1120 		_cmap[lp] = _cmap_mbyte[lp];
1121 		_mbmap[lp] = _mbmap_euc[lp];
1122 		break;
1123 	    case 3:
1124 		/* 3 ... big5 */
1125 		_cmap[lp] = _cmap_mbyte[lp];
1126 		_mbmap[lp] = _mbmap_big5[lp];
1127 		break;
1128 	    case 4:
1129 		/* 4 ... utf8 */
1130 		_cmap[lp] = _cmap_mbyte[lp];
1131 		_mbmap[lp] = _mbmap_utf8[lp];
1132 		break;
1133 	    default:
1134 		xprintf(CGETS(18, 3,
1135 		    "Warning: unknown multibyte code %d; multibyte disabled\n"),
1136 		    iskcode);
1137 		_cmap[lp] = _cmap_c[lp];
1138 		_mbmap[lp] = 0;	/* Default map all 0 */
1139 		_enable_mbdisp = 0;
1140 		break;
1141 	    }
1142 	}
1143 	if (iskcode == 0) {
1144 	    /* check original table */
1145 	    if (Strlen(dstr1) != 256) {
1146 		xprintf(CGETS(18, 4,
1147        "Warning: Invalid multibyte table length (%d); multibyte disabled\n"),
1148 		    Strlen(dstr1));
1149 		_enable_mbdisp = 0;
1150 	    }
1151 	    for (lp = 0; lp < 256 && _enable_mbdisp == 1; lp++) {
1152 		if (!((dstr1[lp] - '0') >= 0 && (dstr1[lp] - '0') <= 3)) {
1153 		    xprintf(CGETS(18, 4,
1154 	   "Warning: bad multibyte code at offset +%d; multibyte diabled\n"),
1155 			lp);
1156 		    _enable_mbdisp = 0;
1157 		    break;
1158 		}
1159 	    }
1160 	    /* set original table */
1161 	    for (lp = 0; lp < 256; lp++) {
1162 		if (_enable_mbdisp == 1) {
1163 		    _cmap[lp] = _cmap_mbyte[lp];
1164 		    _mbmap[lp] = (unsigned short) ((dstr1[lp] - '0') & 0x0f);
1165 		}
1166 		else {
1167 		    _cmap[lp] = _cmap_c[lp];
1168 		    _mbmap[lp] = 0;	/* Default map all 0 */
1169 		}
1170 	    }
1171 	}
1172     }
1173     else {
1174 	for (lp = 0; lp < 256; lp++) {
1175 	    _cmap[lp] = _cmap_c[lp];
1176 	    _mbmap[lp] = 0;	/* Default map all 0 */
1177 	}
1178 	_enable_mbdisp = 0;
1179 	dspmbyte_ls = 0;
1180     }
1181 #ifdef MBYTEDEBUG	/* Sorry, use for beta testing */
1182     {
1183 	Char mbmapstr[300];
1184 	for (lp = 0; lp < 256; lp++)
1185 	    mbmapstr[lp] = _mbmap[lp] + '0';
1186 	mbmapstr[lp] = 0;
1187 	setcopy(STRmbytemap, mbmapstr, VAR_READWRITE);
1188     }
1189 #endif /* MBYTEMAP */
1190 }
1191 
1192 /* dspkanji/dspmbyte autosetting */
1193 /* PATCH IDEA FROM Issei.Suzuki VERY THANKS */
1194 void
1195 autoset_dspmbyte(const Char *pcp)
1196 {
1197     int i;
1198     static const struct dspm_autoset_Table {
1199 	Char *n;
1200 	Char *v;
1201     } dspmt[] = {
1202 	{ STRLANGEUCJP, STReuc },
1203 	{ STRLANGEUCKR, STReuc },
1204 	{ STRLANGEUCZH, STReuc },
1205 	{ STRLANGEUCJPB, STReuc },
1206 	{ STRLANGEUCKRB, STReuc },
1207 	{ STRLANGEUCZHB, STReuc },
1208 #ifdef linux
1209 	{ STRLANGEUCJPC, STReuc },
1210 #endif
1211 	{ STRLANGSJIS, STRsjis },
1212 	{ STRLANGSJISB, STRsjis },
1213 	{ STRLANGBIG5, STRbig5 },
1214 	{ STRstarutfstar8, STRutf8 },
1215 	{ NULL, NULL }
1216     };
1217 #if defined(HAVE_NL_LANGINFO) && defined(CODESET)
1218     static const struct dspm_autoset_Table dspmc[] = {
1219 	{ STRstarutfstar8, STRutf8 },
1220 	{ STReuc, STReuc },
1221 	{ STRGB2312, STReuc },
1222 	{ STRLANGBIG5, STRbig5 },
1223 	{ NULL, NULL }
1224     };
1225     Char *codeset;
1226 
1227     codeset = str2short(nl_langinfo(CODESET));
1228     if (*codeset != '\0') {
1229 	for (i = 0; dspmc[i].n; i++) {
1230 	    const Char *estr;
1231 	    if (dspmc[i].n[0] && t_pmatch(pcp, dspmc[i].n, &estr, 0) > 0) {
1232 		setcopy(CHECK_MBYTEVAR, dspmc[i].v, VAR_READWRITE);
1233 		update_dspmbyte_vars();
1234 		return;
1235 	    }
1236 	}
1237     }
1238 #endif
1239 
1240     if (*pcp == '\0')
1241 	return;
1242 
1243     for (i = 0; dspmt[i].n; i++) {
1244 	const Char *estr;
1245 	if (dspmt[i].n[0] && t_pmatch(pcp, dspmt[i].n, &estr, 0) > 0) {
1246 	    setcopy(CHECK_MBYTEVAR, dspmt[i].v, VAR_READWRITE);
1247 	    update_dspmbyte_vars();
1248 	    break;
1249 	}
1250     }
1251 }
1252 #endif
1253