Update gpt.8:
[dragonfly.git] / contrib / tcsh-6 / sh.set.c
blob8253e02067331a80eded27c79d673005e3712141
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.
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.
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.
33 #include "sh.h"
35 RCSID("$tcsh: sh.set.c,v 3.70 2006/08/24 20:56:31 christos Exp $")
37 #include "ed.h"
38 #include "tw.h"
40 #ifdef HAVE_NL_LANGINFO
41 #include <langinfo.h>
42 #endif
44 extern int GotTermCaps;
45 int numeof = 0;
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 exportpath (Char **);
57 static void balance (struct varent *, int, int);
60 * C Shell
63 static void
64 update_vars(Char *vp)
66 if (eq(vp, STRpath)) {
67 struct varent *p = adrof(STRpath);
68 if (p == NULL)
69 stderror(ERR_NAME | ERR_UNDVAR);
70 else {
71 exportpath(p->vec);
72 dohash(NULL, NULL);
75 else if (eq(vp, STRhistchars)) {
76 Char *pn = varval(vp);
78 HIST = *pn++;
79 HISTSUB = *pn;
81 else if (eq(vp, STRpromptchars)) {
82 Char *pn = varval(vp);
84 PRCH = *pn++;
85 PRCHROOT = *pn;
87 else if (eq(vp, STRhistlit)) {
88 HistLit = 1;
90 else if (eq(vp, STRuser)) {
91 tsetenv(STRKUSER, varval(vp));
92 tsetenv(STRLOGNAME, varval(vp));
94 else if (eq(vp, STRgroup)) {
95 tsetenv(STRKGROUP, varval(vp));
97 else if (eq(vp, STRwordchars)) {
98 word_chars = varval(vp);
100 else if (eq(vp, STRloginsh)) {
101 loginsh = 1;
103 else if (eq(vp, STRsymlinks)) {
104 Char *pn = varval(vp);
106 if (eq(pn, STRignore))
107 symlinks = SYM_IGNORE;
108 else if (eq(pn, STRexpand))
109 symlinks = SYM_EXPAND;
110 else if (eq(pn, STRchase))
111 symlinks = SYM_CHASE;
112 else
113 symlinks = 0;
115 else if (eq(vp, STRterm)) {
116 Char *cp = varval(vp);
117 tsetenv(STRKTERM, cp);
118 #ifdef DOESNT_WORK_RIGHT
119 cp = getenv("TERMCAP");
120 if (cp && (*cp != '/')) /* if TERMCAP and not a path */
121 Unsetenv(STRTERMCAP);
122 #endif /* DOESNT_WORK_RIGHT */
123 GotTermCaps = 0;
124 if (noediting && Strcmp(cp, STRnetwork) != 0 &&
125 Strcmp(cp, STRunknown) != 0 && Strcmp(cp, STRdumb) != 0) {
126 editing = 1;
127 noediting = 0;
128 setNS(STRedit);
130 ed_Init(); /* reset the editor */
132 else if (eq(vp, STRhome)) {
133 Char *cp, *canon;
135 cp = Strsave(varval(vp)); /* get the old value back */
136 cleanup_push(cp, xfree);
139 * convert to cononical pathname (possibly resolving symlinks)
141 canon = dcanon(cp, cp);
142 cleanup_ignore(cp);
143 cleanup_until(cp);
144 cleanup_push(canon, xfree);
146 setcopy(vp, canon, VAR_READWRITE); /* have to save the new val */
148 /* and now mirror home with HOME */
149 tsetenv(STRKHOME, canon);
150 /* fix directory stack for new tilde home */
151 dtilde();
152 cleanup_until(canon);
154 else if (eq(vp, STRedit)) {
155 editing = 1;
156 noediting = 0;
157 /* PWP: add more stuff in here later */
159 else if (eq(vp, STRshlvl)) {
160 tsetenv(STRKSHLVL, varval(vp));
162 else if (eq(vp, STRignoreeof)) {
163 Char *cp;
164 numeof = 0;
165 for ((cp = varval(STRignoreeof)); cp && *cp; cp++) {
166 if (!Isdigit(*cp)) {
167 numeof = 0;
168 break;
170 numeof = numeof * 10 + *cp - '0';
172 if (numeof <= 0) numeof = 26; /* Sanity check */
174 else if (eq(vp, STRbackslash_quote)) {
175 bslash_quote = 1;
177 else if (eq(vp, STRdirstack)) {
178 dsetstack();
180 else if (eq(vp, STRrecognize_only_executables)) {
181 tw_cmd_free();
183 else if (eq(vp, STRkillring)) {
184 SetKillRing(getn(varval(vp)));
186 #ifndef HAVENOUTMP
187 else if (eq(vp, STRwatch)) {
188 resetwatch();
190 #endif /* HAVENOUTMP */
191 else if (eq(vp, STRimplicitcd)) {
192 implicit_cd = ((eq(varval(vp), STRverbose)) ? 2 : 1);
194 #ifdef COLOR_LS_F
195 else if (eq(vp, STRcolor)) {
196 set_color_context();
198 #endif /* COLOR_LS_F */
199 #if defined(KANJI) && defined(SHORT_STRINGS) && defined(DSPMBYTE)
200 else if(eq(vp, CHECK_MBYTEVAR) || eq(vp, STRnokanji)) {
201 update_dspmbyte_vars();
203 #endif
204 #ifdef NLS_CATALOGS
205 else if (eq(vp, STRcatalog)) {
206 nlsclose();
207 nlsinit();
209 #if defined(FILEC) && defined(TIOCSTI)
210 else if (eq(vp, STRfilec))
211 filec = 1;
212 #endif
213 #endif /* NLS_CATALOGS */
217 /*ARGSUSED*/
218 void
219 doset(Char **v, struct command *c)
221 Char *p;
222 Char *vp, op;
223 Char **vecp;
224 int hadsub;
225 int subscr;
226 int flags = VAR_READWRITE;
227 int first_match = 0;
228 int last_match = 0;
229 int changed = 0;
231 USE(c);
232 v++;
233 do {
234 changed = 0;
236 * Readonly addition From: Tim P. Starrin <noid@cyborg.larc.nasa.gov>
238 if (*v && eq(*v, STRmr)) {
239 flags = VAR_READONLY;
240 v++;
241 changed = 1;
243 if (*v && eq(*v, STRmf) && !last_match) {
244 first_match = 1;
245 v++;
246 changed = 1;
248 if (*v && eq(*v, STRml) && !first_match) {
249 last_match = 1;
250 v++;
251 changed = 1;
253 } while(changed);
254 p = *v++;
255 if (p == 0) {
256 plist(&shvhed, flags);
257 return;
259 do {
260 hadsub = 0;
261 vp = p;
262 if (letter(*p))
263 for (; alnum(*p); p++)
264 continue;
265 if (vp == p || !letter(*vp))
266 stderror(ERR_NAME | ERR_VARBEGIN);
267 if (*p == '[') {
268 hadsub++;
269 p = getinx(p, &subscr);
271 if ((op = *p) != 0) {
272 *p++ = 0;
273 if (*p == 0 && *v && **v == '(')
274 p = *v++;
276 else if (*v && eq(*v, STRequal)) {
277 op = '=', v++;
278 if (*v)
279 p = *v++;
281 if (op && op != '=')
282 stderror(ERR_NAME | ERR_SYNTAX);
283 if (eq(p, STRLparen)) {
284 Char **e = v;
286 if (hadsub)
287 stderror(ERR_NAME | ERR_SYNTAX);
288 for (;;) {
289 if (!*e)
290 stderror(ERR_NAME | ERR_MISSING, ')');
291 if (**e == ')')
292 break;
293 e++;
295 p = *e;
296 *e = 0;
297 vecp = saveblk(v);
298 if (first_match)
299 flags |= VAR_FIRST;
300 else if (last_match)
301 flags |= VAR_LAST;
303 set1(vp, vecp, &shvhed, flags);
304 *e = p;
305 v = e + 1;
307 else if (hadsub) {
308 Char *copy;
310 copy = Strsave(p);
311 cleanup_push(copy, xfree);
312 asx(vp, subscr, copy);
313 cleanup_ignore(copy);
314 cleanup_until(copy);
316 else
317 setv(vp, Strsave(p), flags);
318 update_vars(vp);
319 } while ((p = *v++) != NULL);
322 static Char *
323 getinx(Char *cp, int *ip)
325 *ip = 0;
326 *cp++ = 0;
327 while (*cp && Isdigit(*cp))
328 *ip = *ip * 10 + *cp++ - '0';
329 if (*cp++ != ']')
330 stderror(ERR_NAME | ERR_SUBSCRIPT);
331 return (cp);
334 static void
335 asx(Char *vp, int subscr, Char *p)
337 struct varent *v = getvx(vp, subscr);
338 Char *prev;
340 if (v->v_flags & VAR_READONLY)
341 stderror(ERR_READONLY|ERR_NAME, v->v_name);
342 prev = v->vec[subscr - 1];
343 cleanup_push(prev, xfree);
344 v->vec[subscr - 1] = globone(p, G_APPEND);
345 cleanup_until(prev);
348 static struct varent *
349 getvx(Char *vp, int subscr)
351 struct varent *v = adrof(vp);
353 if (v == 0)
354 udvar(vp);
355 if (subscr < 1 || subscr > blklen(v->vec))
356 stderror(ERR_NAME | ERR_RANGE);
357 return (v);
360 /*ARGSUSED*/
361 void
362 dolet(Char **v, struct command *dummy)
364 Char *p;
365 Char *vp, c, op;
366 int hadsub;
367 int subscr;
369 USE(dummy);
370 v++;
371 p = *v++;
372 if (p == 0) {
373 prvars();
374 return;
376 do {
377 hadsub = 0;
378 vp = p;
379 if (letter(*p))
380 for (; alnum(*p); p++)
381 continue;
382 if (vp == p || !letter(*vp))
383 stderror(ERR_NAME | ERR_VARBEGIN);
384 if (*p == '[') {
385 hadsub++;
386 p = getinx(p, &subscr);
388 if (*p == 0 && *v)
389 p = *v++;
390 if ((op = *p) != 0)
391 *p++ = 0;
392 else
393 stderror(ERR_NAME | ERR_ASSIGN);
396 * if there is no expression after the '=' then print a "Syntax Error"
397 * message - strike
399 if (*p == '\0' && *v == NULL)
400 stderror(ERR_NAME | ERR_ASSIGN);
402 vp = Strsave(vp);
403 cleanup_push(vp, xfree);
404 if (op == '=') {
405 c = '=';
406 p = xset(p, &v);
408 else {
409 c = *p++;
410 if (any("+-", c)) {
411 if (c != op || *p)
412 stderror(ERR_NAME | ERR_UNKNOWNOP);
413 p = Strsave(STR1);
415 else {
416 if (any("<>", op)) {
417 if (c != op)
418 stderror(ERR_NAME | ERR_UNKNOWNOP);
419 stderror(ERR_NAME | ERR_SYNTAX);
421 if (c != '=')
422 stderror(ERR_NAME | ERR_UNKNOWNOP);
423 p = xset(p, &v);
426 cleanup_push(p, xfree);
427 if (op == '=') {
428 if (hadsub)
429 asx(vp, subscr, p);
430 else
431 setv(vp, p, VAR_READWRITE);
432 cleanup_ignore(p);
434 else if (hadsub) {
435 struct varent *gv = getvx(vp, subscr);
436 Char *val;
438 val = operate(op, gv->vec[subscr - 1], p);
439 cleanup_push(val, xfree);
440 asx(vp, subscr, val);
441 cleanup_ignore(val);
442 cleanup_until(val);
444 else {
445 Char *val;
447 val = operate(op, varval(vp), p);
448 cleanup_push(val, xfree);
449 setv(vp, val, VAR_READWRITE);
450 cleanup_ignore(val);
451 cleanup_until(val);
453 update_vars(vp);
454 cleanup_until(vp);
455 } while ((p = *v++) != NULL);
458 static Char *
459 xset(Char *cp, Char ***vp)
461 Char *dp;
463 if (*cp) {
464 dp = Strsave(cp);
465 --(*vp);
466 xfree(** vp);
467 **vp = dp;
469 return (putn(expr(vp)));
472 static Char *
473 operate(int op, Char *vp, Char *p)
475 Char opr[2];
476 Char *vec[5];
477 Char **v = vec;
478 Char **vecp = v;
479 int i;
481 if (op != '=') {
482 if (*vp)
483 *v++ = vp;
484 opr[0] = op;
485 opr[1] = 0;
486 *v++ = opr;
487 if (op == '<' || op == '>')
488 *v++ = opr;
490 *v++ = p;
491 *v++ = 0;
492 i = expr(&vecp);
493 if (*vecp)
494 stderror(ERR_NAME | ERR_EXPRESSION);
495 return (putn(i));
498 static Char *putp;
500 Char *
501 putn(int n)
503 Char nbuf[(CHAR_BIT * sizeof (n) + 2) / 3 + 2]; /* Enough even for octal */
505 putp = nbuf;
506 if (n < 0) {
507 n = -n;
508 *putp++ = '-';
510 putn1(n);
511 *putp = 0;
512 return (Strsave(nbuf));
515 static void
516 putn1(unsigned n)
518 if (n > 9)
519 putn1(n / 10);
520 *putp++ = n % 10 + '0';
524 getn(Char *cp)
526 int n;
527 int sign;
529 if (!cp) /* PWP: extra error checking */
530 stderror(ERR_NAME | ERR_BADNUM);
532 sign = 0;
533 if (cp[0] == '+' && cp[1])
534 cp++;
535 if (*cp == '-') {
536 sign++;
537 cp++;
538 if (!Isdigit(*cp))
539 stderror(ERR_NAME | ERR_BADNUM);
541 n = 0;
542 while (Isdigit(*cp))
543 n = n * 10 + *cp++ - '0';
544 if (*cp)
545 stderror(ERR_NAME | ERR_BADNUM);
546 return (sign ? -n : n);
549 Char *
550 value1(Char *var, struct varent *head)
552 struct varent *vp;
554 if (!var || !head) /* PWP: extra error checking */
555 return (STRNULL);
557 vp = adrof1(var, head);
558 return ((vp == NULL || vp->vec == NULL || vp->vec[0] == NULL) ?
559 STRNULL : vp->vec[0]);
562 static struct varent *
563 madrof(Char *pat, struct varent *vp)
565 struct varent *vp1;
567 for (vp = vp->v_left; vp; vp = vp->v_right) {
568 if (vp->v_left && (vp1 = madrof(pat, vp)) != NULL)
569 return vp1;
570 if (Gmatch(vp->v_name, pat))
571 return vp;
573 return vp;
576 struct varent *
577 adrof1(const Char *name, struct varent *v)
579 int cmp;
581 v = v->v_left;
582 while (v && ((cmp = *name - *v->v_name) != 0 ||
583 (cmp = Strcmp(name, v->v_name)) != 0))
584 if (cmp < 0)
585 v = v->v_left;
586 else
587 v = v->v_right;
588 return v;
591 void
592 setcopy(const Char *var, const Char *val, int flags)
594 Char *copy;
596 copy = Strsave(val);
597 cleanup_push(copy, xfree);
598 setv(var, copy, flags);
599 cleanup_ignore(copy);
600 cleanup_until(copy);
604 * The caller is responsible for putting value in a safe place
606 void
607 setv(const Char *var, Char *val, int flags)
609 Char **vec = xmalloc(2 * sizeof(Char **));
611 vec[0] = val;
612 vec[1] = 0;
613 set1(var, vec, &shvhed, flags);
616 void
617 set1(const Char *var, Char **vec, struct varent *head, int flags)
619 Char **oldv = vec;
621 if ((flags & VAR_NOGLOB) == 0) {
622 int gflag;
624 gflag = tglob(oldv);
625 if (gflag) {
626 vec = globall(oldv, gflag);
627 if (vec == 0) {
628 blkfree(oldv);
629 stderror(ERR_NAME | ERR_NOMATCH);
631 blkfree(oldv);
635 * Uniqueness addition from: Michael Veksler <mveksler@vnet.ibm.com>
637 if ( flags & (VAR_FIRST | VAR_LAST) ) {
639 * Code for -f (VAR_FIRST) and -l (VAR_LAST) options.
640 * Method:
641 * Delete all duplicate words leaving "holes" in the word array (vec).
642 * Then remove the "holes", keeping the order of the words unchanged.
644 if (vec && vec[0] && vec[1]) { /* more than one word ? */
645 int i, j;
646 int num_items;
648 for (num_items = 0; vec[num_items]; num_items++)
649 continue;
650 if (flags & VAR_FIRST) {
651 /* delete duplications, keeping first occurance */
652 for (i = 1; i < num_items; i++)
653 for (j = 0; j < i; j++)
654 /* If have earlier identical item, remove i'th item */
655 if (vec[i] && vec[j] && Strcmp(vec[j], vec[i]) == 0) {
656 xfree(vec[i]);
657 vec[i] = NULL;
658 break;
660 } else if (flags & VAR_LAST) {
661 /* delete duplications, keeping last occurance */
662 for (i = 0; i < num_items - 1; i++)
663 for (j = i + 1; j < num_items; j++)
664 /* If have later identical item, remove i'th item */
665 if (vec[i] && vec[j] && Strcmp(vec[j], vec[i]) == 0) {
666 /* remove identical item (the first) */
667 xfree(vec[i]);
668 vec[i] = NULL;
671 /* Compress items - remove empty items */
672 for (j = i = 0; i < num_items; i++)
673 if (vec[i])
674 vec[j++] = vec[i];
676 /* NULL-fy remaining items */
677 for (; j < num_items; j++)
678 vec[j] = NULL;
680 /* don't let the attribute propagate */
681 flags &= ~(VAR_FIRST|VAR_LAST);
683 setq(var, vec, head, flags);
687 void
688 setq(const Char *name, Char **vec, struct varent *p, int flags)
690 struct varent *c;
691 int f;
693 f = 0; /* tree hangs off the header's left link */
694 while ((c = p->v_link[f]) != 0) {
695 if ((f = *name - *c->v_name) == 0 &&
696 (f = Strcmp(name, c->v_name)) == 0) {
697 if (c->v_flags & VAR_READONLY)
698 stderror(ERR_READONLY|ERR_NAME, c->v_name);
699 blkfree(c->vec);
700 c->v_flags = flags;
701 trim(c->vec = vec);
702 return;
704 p = c;
705 f = f > 0;
707 p->v_link[f] = c = xmalloc(sizeof(struct varent));
708 c->v_name = Strsave(name);
709 c->v_flags = flags;
710 c->v_bal = 0;
711 c->v_left = c->v_right = 0;
712 c->v_parent = p;
713 balance(p, f, 0);
714 trim(c->vec = vec);
717 /*ARGSUSED*/
718 void
719 unset(Char **v, struct command *c)
721 int did_roe, did_edit;
723 USE(c);
724 did_roe = adrof(STRrecognize_only_executables) != NULL;
725 did_edit = adrof(STRedit) != NULL;
726 unset1(v, &shvhed);
728 #if defined(FILEC) && defined(TIOCSTI)
729 if (adrof(STRfilec) == 0)
730 filec = 0;
731 #endif /* FILEC && TIOCSTI */
733 if (adrof(STRhistchars) == 0) {
734 HIST = '!';
735 HISTSUB = '^';
737 if (adrof(STRignoreeof) == 0)
738 numeof = 0;
739 if (adrof(STRpromptchars) == 0) {
740 PRCH = '>';
741 PRCHROOT = '#';
743 if (adrof(STRhistlit) == 0)
744 HistLit = 0;
745 if (adrof(STRloginsh) == 0)
746 loginsh = 0;
747 if (adrof(STRwordchars) == 0)
748 word_chars = STR_WORD_CHARS;
749 if (adrof(STRedit) == 0)
750 editing = 0;
751 if (adrof(STRbackslash_quote) == 0)
752 bslash_quote = 0;
753 if (adrof(STRsymlinks) == 0)
754 symlinks = 0;
755 if (adrof(STRimplicitcd) == 0)
756 implicit_cd = 0;
757 if (adrof(STRkillring) == 0)
758 SetKillRing(0);
759 if (did_edit && noediting && adrof(STRedit) == 0)
760 noediting = 0;
761 if (did_roe && adrof(STRrecognize_only_executables) == 0)
762 tw_cmd_free();
763 #ifdef COLOR_LS_F
764 if (adrof(STRcolor) == 0)
765 set_color_context();
766 #endif /* COLOR_LS_F */
767 #if defined(KANJI) && defined(SHORT_STRINGS) && defined(DSPMBYTE)
768 update_dspmbyte_vars();
769 #endif
770 #ifdef NLS_CATALOGS
771 nlsclose();
772 nlsinit();
773 #endif /* NLS_CATALOGS */
776 void
777 unset1(Char *v[], struct varent *head)
779 struct varent *vp;
780 int cnt;
782 while (*++v) {
783 cnt = 0;
784 while ((vp = madrof(*v, head)) != NULL)
785 if (vp->v_flags & VAR_READONLY)
786 stderror(ERR_READONLY|ERR_NAME, vp->v_name);
787 else
788 unsetv1(vp), cnt++;
789 if (cnt == 0)
790 setname(short2str(*v));
794 void
795 unsetv(Char *var)
797 struct varent *vp;
799 if ((vp = adrof1(var, &shvhed)) == 0)
800 udvar(var);
801 unsetv1(vp);
804 static void
805 unsetv1(struct varent *p)
807 struct varent *c, *pp;
808 int f;
811 * Free associated memory first to avoid complications.
813 blkfree(p->vec);
814 xfree(p->v_name);
816 * If p is missing one child, then we can move the other into where p is.
817 * Otherwise, we find the predecessor of p, which is guaranteed to have no
818 * right child, copy it into p, and move it's left child into it.
820 if (p->v_right == 0)
821 c = p->v_left;
822 else if (p->v_left == 0)
823 c = p->v_right;
824 else {
825 for (c = p->v_left; c->v_right; c = c->v_right)
826 continue;
827 p->v_name = c->v_name;
828 p->v_flags = c->v_flags;
829 p->vec = c->vec;
830 p = c;
831 c = p->v_left;
835 * Move c into where p is.
837 pp = p->v_parent;
838 f = pp->v_right == p;
839 if ((pp->v_link[f] = c) != 0)
840 c->v_parent = pp;
842 * Free the deleted node, and rebalance.
844 xfree(p);
845 balance(pp, f, 1);
848 void
849 setNS(Char *cp)
851 setcopy(cp, STRNULL, VAR_READWRITE);
854 /*ARGSUSED*/
855 void
856 shift(Char **v, struct command *c)
858 struct varent *argv;
859 Char *name;
861 USE(c);
862 v++;
863 name = *v;
864 if (name == 0)
865 name = STRargv;
866 else
867 (void) strip(name);
868 argv = adrof(name);
869 if (argv == NULL || argv->vec == NULL)
870 udvar(name);
871 if (argv->vec[0] == 0)
872 stderror(ERR_NAME | ERR_NOMORE);
873 lshift(argv->vec, 1);
874 update_vars(name);
877 static void
878 exportpath(Char **val)
880 struct Strbuf buf = Strbuf_INIT;
881 Char *exppath;
883 if (val)
884 while (*val) {
885 Strbuf_append(&buf, *val++);
886 if (*val == 0 || eq(*val, STRRparen))
887 break;
888 Strbuf_append1(&buf, PATHSEP);
890 exppath = Strbuf_finish(&buf);
891 cleanup_push(exppath, xfree);
892 tsetenv(STRKPATH, exppath);
893 cleanup_until(exppath);
896 #ifndef lint
898 * Lint thinks these have null effect
900 /* macros to do single rotations on node p */
901 # define rright(p) (\
902 t = (p)->v_left,\
903 (t)->v_parent = (p)->v_parent,\
904 (((p)->v_left = t->v_right) != NULL) ?\
905 (t->v_right->v_parent = (p)) : 0,\
906 (t->v_right = (p))->v_parent = t,\
907 (p) = t)
908 # define rleft(p) (\
909 t = (p)->v_right,\
910 ((t)->v_parent = (p)->v_parent,\
911 ((p)->v_right = t->v_left) != NULL) ? \
912 (t->v_left->v_parent = (p)) : 0,\
913 (t->v_left = (p))->v_parent = t,\
914 (p) = t)
915 #else
916 static struct varent *
917 rleft(struct varent *p)
919 return (p);
921 static struct varent *
922 rright(struct varent *p)
924 return (p);
927 #endif /* ! lint */
931 * Rebalance a tree, starting at p and up.
932 * F == 0 means we've come from p's left child.
933 * D == 1 means we've just done a delete, otherwise an insert.
935 static void
936 balance(struct varent *p, int f, int d)
938 struct varent *pp;
940 #ifndef lint
941 struct varent *t; /* used by the rotate macros */
942 #endif /* !lint */
943 int ff;
944 #ifdef lint
945 ff = 0; /* Sun's lint is dumb! */
946 #endif
949 * Ok, from here on, p is the node we're operating on; pp is it's parent; f
950 * is the branch of p from which we have come; ff is the branch of pp which
951 * is p.
953 for (; (pp = p->v_parent) != 0; p = pp, f = ff) {
954 ff = pp->v_right == p;
955 if (f ^ d) { /* right heavy */
956 switch (p->v_bal) {
957 case -1: /* was left heavy */
958 p->v_bal = 0;
959 break;
960 case 0: /* was balanced */
961 p->v_bal = 1;
962 break;
963 case 1: /* was already right heavy */
964 switch (p->v_right->v_bal) {
965 case 1: /* single rotate */
966 pp->v_link[ff] = rleft(p);
967 p->v_left->v_bal = 0;
968 p->v_bal = 0;
969 break;
970 case 0: /* single rotate */
971 pp->v_link[ff] = rleft(p);
972 p->v_left->v_bal = 1;
973 p->v_bal = -1;
974 break;
975 case -1: /* double rotate */
976 (void) rright(p->v_right);
977 pp->v_link[ff] = rleft(p);
978 p->v_left->v_bal =
979 p->v_bal < 1 ? 0 : -1;
980 p->v_right->v_bal =
981 p->v_bal > -1 ? 0 : 1;
982 p->v_bal = 0;
983 break;
984 default:
985 break;
987 break;
988 default:
989 break;
992 else { /* left heavy */
993 switch (p->v_bal) {
994 case 1: /* was right heavy */
995 p->v_bal = 0;
996 break;
997 case 0: /* was balanced */
998 p->v_bal = -1;
999 break;
1000 case -1: /* was already left heavy */
1001 switch (p->v_left->v_bal) {
1002 case -1: /* single rotate */
1003 pp->v_link[ff] = rright(p);
1004 p->v_right->v_bal = 0;
1005 p->v_bal = 0;
1006 break;
1007 case 0: /* single rotate */
1008 pp->v_link[ff] = rright(p);
1009 p->v_right->v_bal = -1;
1010 p->v_bal = 1;
1011 break;
1012 case 1: /* double rotate */
1013 (void) rleft(p->v_left);
1014 pp->v_link[ff] = rright(p);
1015 p->v_left->v_bal =
1016 p->v_bal < 1 ? 0 : -1;
1017 p->v_right->v_bal =
1018 p->v_bal > -1 ? 0 : 1;
1019 p->v_bal = 0;
1020 break;
1021 default:
1022 break;
1024 break;
1025 default:
1026 break;
1030 * If from insert, then we terminate when p is balanced. If from
1031 * delete, then we terminate when p is unbalanced.
1033 if ((p->v_bal == 0) ^ d)
1034 break;
1038 void
1039 plist(struct varent *p, int what)
1041 struct varent *c;
1042 int len;
1044 for (;;) {
1045 while (p->v_left)
1046 p = p->v_left;
1048 if (p->v_parent == 0) /* is it the header? */
1049 break;
1050 if ((p->v_flags & what) != 0) {
1051 if (setintr) {
1052 int old_pintr_disabled;
1054 pintr_push_enable(&old_pintr_disabled);
1055 cleanup_until(&old_pintr_disabled);
1057 len = blklen(p->vec);
1058 xprintf("%S\t", p->v_name);
1059 if (len != 1)
1060 xputchar('(');
1061 blkpr(p->vec);
1062 if (len != 1)
1063 xputchar(')');
1064 xputchar('\n');
1066 if (p->v_right) {
1067 p = p->v_right;
1068 continue;
1070 do {
1071 c = p;
1072 p = p->v_parent;
1073 } while (p->v_right == c);
1074 goto x;
1078 #if defined(KANJI) && defined(SHORT_STRINGS) && defined(DSPMBYTE)
1079 extern int dspmbyte_ls;
1081 void
1082 update_dspmbyte_vars(void)
1084 int lp, iskcode;
1085 Char *dstr1;
1086 struct varent *vp;
1088 /* if variable "nokanji" is set, multi-byte display is disabled */
1089 if ((vp = adrof(CHECK_MBYTEVAR)) && !adrof(STRnokanji)) {
1090 _enable_mbdisp = 1;
1091 dstr1 = vp->vec[0];
1092 if(eq (dstr1, STRsjis))
1093 iskcode = 1;
1094 else if (eq(dstr1, STReuc))
1095 iskcode = 2;
1096 else if (eq(dstr1, STRbig5))
1097 iskcode = 3;
1098 else if (eq(dstr1, STRutf8))
1099 iskcode = 4;
1100 else if ((dstr1[0] - '0') >= 0 && (dstr1[0] - '0') <= 3) {
1101 iskcode = 0;
1103 else {
1104 xprintf(CGETS(18, 2,
1105 "Warning: unknown multibyte display; using default(euc(JP))\n"));
1106 iskcode = 2;
1108 if (dstr1 && vp->vec[1] && eq(vp->vec[1], STRls))
1109 dspmbyte_ls = 1;
1110 else
1111 dspmbyte_ls = 0;
1112 for (lp = 0; lp < 256 && iskcode > 0; lp++) {
1113 switch (iskcode) {
1114 case 1:
1115 /* Shift-JIS */
1116 _cmap[lp] = _cmap_mbyte[lp];
1117 _mbmap[lp] = _mbmap_sjis[lp];
1118 break;
1119 case 2:
1120 /* 2 ... euc */
1121 _cmap[lp] = _cmap_mbyte[lp];
1122 _mbmap[lp] = _mbmap_euc[lp];
1123 break;
1124 case 3:
1125 /* 3 ... big5 */
1126 _cmap[lp] = _cmap_mbyte[lp];
1127 _mbmap[lp] = _mbmap_big5[lp];
1128 break;
1129 case 4:
1130 /* 4 ... utf8 */
1131 _cmap[lp] = _cmap_mbyte[lp];
1132 _mbmap[lp] = _mbmap_utf8[lp];
1133 break;
1134 default:
1135 xprintf(CGETS(18, 3,
1136 "Warning: unknown multibyte code %d; multibyte disabled\n"),
1137 iskcode);
1138 _cmap[lp] = _cmap_c[lp];
1139 _mbmap[lp] = 0; /* Default map all 0 */
1140 _enable_mbdisp = 0;
1141 break;
1144 if (iskcode == 0) {
1145 /* check original table */
1146 if (Strlen(dstr1) != 256) {
1147 xprintf(CGETS(18, 4,
1148 "Warning: Invalid multibyte table length (%d); multibyte disabled\n"),
1149 Strlen(dstr1));
1150 _enable_mbdisp = 0;
1152 for (lp = 0; lp < 256 && _enable_mbdisp == 1; lp++) {
1153 if (!((dstr1[lp] - '0') >= 0 && (dstr1[lp] - '0') <= 3)) {
1154 xprintf(CGETS(18, 4,
1155 "Warning: bad multibyte code at offset +%d; multibyte diabled\n"),
1156 lp);
1157 _enable_mbdisp = 0;
1158 break;
1161 /* set original table */
1162 for (lp = 0; lp < 256; lp++) {
1163 if (_enable_mbdisp == 1) {
1164 _cmap[lp] = _cmap_mbyte[lp];
1165 _mbmap[lp] = (unsigned short) ((dstr1[lp] - '0') & 0x0f);
1167 else {
1168 _cmap[lp] = _cmap_c[lp];
1169 _mbmap[lp] = 0; /* Default map all 0 */
1174 else {
1175 for (lp = 0; lp < 256; lp++) {
1176 _cmap[lp] = _cmap_c[lp];
1177 _mbmap[lp] = 0; /* Default map all 0 */
1179 _enable_mbdisp = 0;
1180 dspmbyte_ls = 0;
1182 #ifdef MBYTEDEBUG /* Sorry, use for beta testing */
1184 Char mbmapstr[300];
1185 for (lp = 0; lp < 256; lp++)
1186 mbmapstr[lp] = _mbmap[lp] + '0';
1187 mbmapstr[lp] = 0;
1188 setcopy(STRmbytemap, mbmapstr, VAR_READWRITE);
1190 #endif /* MBYTEMAP */
1193 /* dspkanji/dspmbyte autosetting */
1194 /* PATCH IDEA FROM Issei.Suzuki VERY THANKS */
1195 void
1196 autoset_dspmbyte(const Char *pcp)
1198 int i;
1199 static const struct dspm_autoset_Table {
1200 Char *n;
1201 Char *v;
1202 } dspmt[] = {
1203 { STRLANGEUCJP, STReuc },
1204 { STRLANGEUCKR, STReuc },
1205 { STRLANGEUCZH, STReuc },
1206 { STRLANGEUCJPB, STReuc },
1207 { STRLANGEUCKRB, STReuc },
1208 { STRLANGEUCZHB, STReuc },
1209 #ifdef linux
1210 { STRLANGEUCJPC, STReuc },
1211 #endif
1212 { STRLANGSJIS, STRsjis },
1213 { STRLANGSJISB, STRsjis },
1214 { STRLANGBIG5, STRbig5 },
1215 { STRstarutfstar8, STRutf8 },
1216 { NULL, NULL }
1218 #if defined(HAVE_NL_LANGINFO) && defined(CODESET)
1219 static const struct dspm_autoset_Table dspmc[] = {
1220 { STRstarutfstar8, STRutf8 },
1221 { STReuc, STReuc },
1222 { STRGB2312, STReuc },
1223 { STRLANGBIG5, STRbig5 },
1224 { NULL, NULL }
1226 Char *codeset;
1228 codeset = str2short(nl_langinfo(CODESET));
1229 if (*codeset != '\0') {
1230 for (i = 0; dspmc[i].n; i++) {
1231 const Char *estr;
1232 if (dspmc[i].n[0] && t_pmatch(pcp, dspmc[i].n, &estr, 0) > 0) {
1233 setcopy(CHECK_MBYTEVAR, dspmc[i].v, VAR_READWRITE);
1234 update_dspmbyte_vars();
1235 return;
1239 #endif
1241 if (*pcp == '\0')
1242 return;
1244 for (i = 0; dspmt[i].n; i++) {
1245 const Char *estr;
1246 if (dspmt[i].n[0] && t_pmatch(pcp, dspmt[i].n, &estr, 0) > 0) {
1247 setcopy(CHECK_MBYTEVAR, dspmt[i].v, VAR_READWRITE);
1248 update_dspmbyte_vars();
1249 break;
1253 #endif