xref: /titanic_52/usr/src/cmd/awk/run.c (revision 83efe03b07f2c50424cb88af7a19d789200bf17c)
17c478bd9Sstevel@tonic-gate /*
27c478bd9Sstevel@tonic-gate  * CDDL HEADER START
37c478bd9Sstevel@tonic-gate  *
47c478bd9Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
5*83efe03bScf46844  * Common Development and Distribution License (the "License").
6*83efe03bScf46844  * You may not use this file except in compliance with the License.
77c478bd9Sstevel@tonic-gate  *
87c478bd9Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
97c478bd9Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
107c478bd9Sstevel@tonic-gate  * See the License for the specific language governing permissions
117c478bd9Sstevel@tonic-gate  * and limitations under the License.
127c478bd9Sstevel@tonic-gate  *
137c478bd9Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
147c478bd9Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
157c478bd9Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
167c478bd9Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
177c478bd9Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
187c478bd9Sstevel@tonic-gate  *
197c478bd9Sstevel@tonic-gate  * CDDL HEADER END
207c478bd9Sstevel@tonic-gate  */
217c478bd9Sstevel@tonic-gate 
227c478bd9Sstevel@tonic-gate /*
23*83efe03bScf46844  * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
247c478bd9Sstevel@tonic-gate  * Use is subject to license terms.
257c478bd9Sstevel@tonic-gate  */
267c478bd9Sstevel@tonic-gate 
271ee2e5faSnakanon /*	Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T	*/
281ee2e5faSnakanon /*	  All Rights Reserved  	*/
297c478bd9Sstevel@tonic-gate 
301ee2e5faSnakanon #pragma ident	"%Z%%M%	%I%	%E% SMI"
317c478bd9Sstevel@tonic-gate 
321ee2e5faSnakanon #define	tempfree(x, s)	if (istemp(x)) tfree(x, s)
331ee2e5faSnakanon 
347c478bd9Sstevel@tonic-gate #define	execute(p) r_execute(p)
357c478bd9Sstevel@tonic-gate 
367c478bd9Sstevel@tonic-gate #define	DEBUG
377c478bd9Sstevel@tonic-gate #include	"awk.h"
387c478bd9Sstevel@tonic-gate #include	<math.h>
397c478bd9Sstevel@tonic-gate #include	"y.tab.h"
407c478bd9Sstevel@tonic-gate #include	<stdio.h>
417c478bd9Sstevel@tonic-gate #include	<ctype.h>
427c478bd9Sstevel@tonic-gate #include	<setjmp.h>
437c478bd9Sstevel@tonic-gate #include	<time.h>
447c478bd9Sstevel@tonic-gate 
457c478bd9Sstevel@tonic-gate #ifndef	FOPEN_MAX
467c478bd9Sstevel@tonic-gate #define	FOPEN_MAX	15	/* max number of open files, from ANSI std. */
477c478bd9Sstevel@tonic-gate #endif
487c478bd9Sstevel@tonic-gate 
497c478bd9Sstevel@tonic-gate 
501ee2e5faSnakanon static jmp_buf env;
517c478bd9Sstevel@tonic-gate 
521ee2e5faSnakanon static	Cell	*r_execute(Node *);
531ee2e5faSnakanon static	Cell	*gettemp(char *), *copycell(Cell *);
541ee2e5faSnakanon static	FILE	*openfile(int, uchar *), *redirect(int, Node *);
557c478bd9Sstevel@tonic-gate 
567c478bd9Sstevel@tonic-gate int	paircnt;
577c478bd9Sstevel@tonic-gate Node	*winner = NULL;
581ee2e5faSnakanon 
591ee2e5faSnakanon static Cell	*tmps;
607c478bd9Sstevel@tonic-gate 
617c478bd9Sstevel@tonic-gate static Cell	truecell	= { OBOOL, BTRUE, 0, 0, 1.0, NUM };
627c478bd9Sstevel@tonic-gate Cell	*true	= &truecell;
637c478bd9Sstevel@tonic-gate static Cell	falsecell	= { OBOOL, BFALSE, 0, 0, 0.0, NUM };
647c478bd9Sstevel@tonic-gate Cell	*false	= &falsecell;
657c478bd9Sstevel@tonic-gate static Cell	breakcell	= { OJUMP, JBREAK, 0, 0, 0.0, NUM };
667c478bd9Sstevel@tonic-gate Cell	*jbreak	= &breakcell;
677c478bd9Sstevel@tonic-gate static Cell	contcell	= { OJUMP, JCONT, 0, 0, 0.0, NUM };
687c478bd9Sstevel@tonic-gate Cell	*jcont	= &contcell;
697c478bd9Sstevel@tonic-gate static Cell	nextcell	= { OJUMP, JNEXT, 0, 0, 0.0, NUM };
707c478bd9Sstevel@tonic-gate Cell	*jnext	= &nextcell;
717c478bd9Sstevel@tonic-gate static Cell	exitcell	= { OJUMP, JEXIT, 0, 0, 0.0, NUM };
727c478bd9Sstevel@tonic-gate Cell	*jexit	= &exitcell;
737c478bd9Sstevel@tonic-gate static Cell	retcell		= { OJUMP, JRET, 0, 0, 0.0, NUM };
747c478bd9Sstevel@tonic-gate Cell	*jret	= &retcell;
757c478bd9Sstevel@tonic-gate static Cell	tempcell	= { OCELL, CTEMP, 0, 0, 0.0, NUM };
767c478bd9Sstevel@tonic-gate 
777c478bd9Sstevel@tonic-gate Node	*curnode = NULL;	/* the node being executed, for debugging */
787c478bd9Sstevel@tonic-gate 
791ee2e5faSnakanon static	void	tfree(Cell *, char *);
801ee2e5faSnakanon static	void	closeall(void);
811ee2e5faSnakanon static	double	ipow(double, int);
821ee2e5faSnakanon 
831ee2e5faSnakanon void
841ee2e5faSnakanon run(Node *a)
857c478bd9Sstevel@tonic-gate {
861ee2e5faSnakanon 	(void) execute(a);
877c478bd9Sstevel@tonic-gate 	closeall();
887c478bd9Sstevel@tonic-gate }
897c478bd9Sstevel@tonic-gate 
901ee2e5faSnakanon static Cell *
911ee2e5faSnakanon r_execute(Node *u)
927c478bd9Sstevel@tonic-gate {
937c478bd9Sstevel@tonic-gate 	register Cell *(*proc)();
947c478bd9Sstevel@tonic-gate 	register Cell *x;
957c478bd9Sstevel@tonic-gate 	register Node *a;
967c478bd9Sstevel@tonic-gate 
977c478bd9Sstevel@tonic-gate 	if (u == NULL)
987c478bd9Sstevel@tonic-gate 		return (true);
997c478bd9Sstevel@tonic-gate 	for (a = u; ; a = a->nnext) {
1007c478bd9Sstevel@tonic-gate 		curnode = a;
1017c478bd9Sstevel@tonic-gate 		if (isvalue(a)) {
1027c478bd9Sstevel@tonic-gate 			x = (Cell *) (a->narg[0]);
1037c478bd9Sstevel@tonic-gate 			if ((x->tval & FLD) && !donefld)
1047c478bd9Sstevel@tonic-gate 				fldbld();
1057c478bd9Sstevel@tonic-gate 			else if ((x->tval & REC) && !donerec)
1067c478bd9Sstevel@tonic-gate 				recbld();
1077c478bd9Sstevel@tonic-gate 			return (x);
1087c478bd9Sstevel@tonic-gate 		}
1091ee2e5faSnakanon 		/* probably a Cell* but too risky to print */
1101ee2e5faSnakanon 		if (notlegal(a->nobj))
1117c478bd9Sstevel@tonic-gate 			ERROR "illegal statement" FATAL;
1127c478bd9Sstevel@tonic-gate 		proc = proctab[a->nobj-FIRSTTOKEN];
1137c478bd9Sstevel@tonic-gate 		x = (*proc)(a->narg, a->nobj);
1147c478bd9Sstevel@tonic-gate 		if ((x->tval & FLD) && !donefld)
1157c478bd9Sstevel@tonic-gate 			fldbld();
1167c478bd9Sstevel@tonic-gate 		else if ((x->tval & REC) && !donerec)
1177c478bd9Sstevel@tonic-gate 			recbld();
1187c478bd9Sstevel@tonic-gate 		if (isexpr(a))
1197c478bd9Sstevel@tonic-gate 			return (x);
1207c478bd9Sstevel@tonic-gate 		/* a statement, goto next statement */
1217c478bd9Sstevel@tonic-gate 		if (isjump(x))
1227c478bd9Sstevel@tonic-gate 			return (x);
1237c478bd9Sstevel@tonic-gate 		if (a->nnext == (Node *)NULL)
1247c478bd9Sstevel@tonic-gate 			return (x);
1257c478bd9Sstevel@tonic-gate 		tempfree(x, "execute");
1267c478bd9Sstevel@tonic-gate 	}
1277c478bd9Sstevel@tonic-gate }
1287c478bd9Sstevel@tonic-gate 
1291ee2e5faSnakanon /*ARGSUSED*/
1301ee2e5faSnakanon Cell *
1311ee2e5faSnakanon program(Node **a, int n)
1327c478bd9Sstevel@tonic-gate {
1337c478bd9Sstevel@tonic-gate 	register Cell *x;
1347c478bd9Sstevel@tonic-gate 
1357c478bd9Sstevel@tonic-gate 	if (setjmp(env) != 0)
1367c478bd9Sstevel@tonic-gate 		goto ex;
1377c478bd9Sstevel@tonic-gate 	if (a[0]) {		/* BEGIN */
1387c478bd9Sstevel@tonic-gate 		x = execute(a[0]);
1397c478bd9Sstevel@tonic-gate 		if (isexit(x))
1407c478bd9Sstevel@tonic-gate 			return (true);
1411ee2e5faSnakanon 		if (isjump(x)) {
1421ee2e5faSnakanon 			ERROR "illegal break, continue or next from BEGIN"
1431ee2e5faSnakanon 			    FATAL;
1441ee2e5faSnakanon 		}
1457c478bd9Sstevel@tonic-gate 		tempfree(x, "");
1467c478bd9Sstevel@tonic-gate 	}
1477c478bd9Sstevel@tonic-gate loop:
1487c478bd9Sstevel@tonic-gate 	if (a[1] || a[2])
1491ee2e5faSnakanon 		while (getrec(&record, &record_size) > 0) {
1507c478bd9Sstevel@tonic-gate 			x = execute(a[1]);
1517c478bd9Sstevel@tonic-gate 			if (isexit(x))
1527c478bd9Sstevel@tonic-gate 				break;
1537c478bd9Sstevel@tonic-gate 			tempfree(x, "");
1547c478bd9Sstevel@tonic-gate 		}
1557c478bd9Sstevel@tonic-gate ex:
1567c478bd9Sstevel@tonic-gate 	if (setjmp(env) != 0)
1577c478bd9Sstevel@tonic-gate 		goto ex1;
1587c478bd9Sstevel@tonic-gate 	if (a[2]) {		/* END */
1597c478bd9Sstevel@tonic-gate 		x = execute(a[2]);
1607c478bd9Sstevel@tonic-gate 		if (iscont(x))	/* read some more */
1617c478bd9Sstevel@tonic-gate 			goto loop;
1627c478bd9Sstevel@tonic-gate 		if (isbreak(x) || isnext(x))
1637c478bd9Sstevel@tonic-gate 			ERROR "illegal break or next from END" FATAL;
1647c478bd9Sstevel@tonic-gate 		tempfree(x, "");
1657c478bd9Sstevel@tonic-gate 	}
1667c478bd9Sstevel@tonic-gate ex1:
1677c478bd9Sstevel@tonic-gate 	return (true);
1687c478bd9Sstevel@tonic-gate }
1697c478bd9Sstevel@tonic-gate 
1707c478bd9Sstevel@tonic-gate struct Frame {
1717c478bd9Sstevel@tonic-gate 	int nargs;	/* number of arguments in this call */
1727c478bd9Sstevel@tonic-gate 	Cell *fcncell;	/* pointer to Cell for function */
1737c478bd9Sstevel@tonic-gate 	Cell **args;	/* pointer to array of arguments after execute */
1747c478bd9Sstevel@tonic-gate 	Cell *retval;	/* return value */
1757c478bd9Sstevel@tonic-gate };
1767c478bd9Sstevel@tonic-gate 
1777c478bd9Sstevel@tonic-gate #define	NARGS	30
1787c478bd9Sstevel@tonic-gate 
1797c478bd9Sstevel@tonic-gate struct Frame *frame = NULL; /* base of stack frames; dynamically allocated */
1807c478bd9Sstevel@tonic-gate int	nframe = 0;		/* number of frames allocated */
1817c478bd9Sstevel@tonic-gate struct Frame *fp = NULL;	/* frame pointer. bottom level unused */
1827c478bd9Sstevel@tonic-gate 
1831ee2e5faSnakanon /*ARGSUSED*/
1841ee2e5faSnakanon Cell *
1851ee2e5faSnakanon call(Node **a, int n)
1867c478bd9Sstevel@tonic-gate {
1871ee2e5faSnakanon 	static Cell newcopycell =
1881ee2e5faSnakanon 		{ OCELL, CCOPY, 0, (uchar *) "", 0.0, NUM|STR|DONTFREE };
1897c478bd9Sstevel@tonic-gate 	int i, ncall, ndef, freed = 0;
1907c478bd9Sstevel@tonic-gate 	Node *x;
1917c478bd9Sstevel@tonic-gate 	Cell *args[NARGS], *oargs[NARGS], *y, *z, *fcn;
1927c478bd9Sstevel@tonic-gate 	uchar *s;
1937c478bd9Sstevel@tonic-gate 
1947c478bd9Sstevel@tonic-gate 	fcn = execute(a[0]);	/* the function itself */
1957c478bd9Sstevel@tonic-gate 	s = fcn->nval;
1967c478bd9Sstevel@tonic-gate 	if (!isfunc(fcn))
1977c478bd9Sstevel@tonic-gate 		ERROR "calling undefined function %s", s FATAL;
1987c478bd9Sstevel@tonic-gate 	if (frame == NULL) {
1991ee2e5faSnakanon 		fp = frame = (struct Frame *)calloc(nframe += 100,
2001ee2e5faSnakanon 		    sizeof (struct Frame));
2011ee2e5faSnakanon 		if (frame == NULL) {
2021ee2e5faSnakanon 			ERROR "out of space for stack frames calling %s",
2031ee2e5faSnakanon 			    s FATAL;
2041ee2e5faSnakanon 		}
2057c478bd9Sstevel@tonic-gate 	}
2067c478bd9Sstevel@tonic-gate 	for (ncall = 0, x = a[1]; x != NULL; x = x->nnext) /* args in call */
2077c478bd9Sstevel@tonic-gate 		ncall++;
2087c478bd9Sstevel@tonic-gate 	ndef = (int)fcn->fval;			/* args in defn */
2091ee2e5faSnakanon 	dprintf(("calling %s, %d args (%d in defn), fp=%d\n",
2101ee2e5faSnakanon 	    s, ncall, ndef, fp-frame));
2111ee2e5faSnakanon 	if (ncall > ndef) {
2127c478bd9Sstevel@tonic-gate 		ERROR "function %s called with %d args, uses only %d",
2137c478bd9Sstevel@tonic-gate 		    s, ncall, ndef WARNING;
2141ee2e5faSnakanon 	}
2151ee2e5faSnakanon 	if (ncall + ndef > NARGS) {
2161ee2e5faSnakanon 		ERROR "function %s has %d arguments, limit %d",
2171ee2e5faSnakanon 		    s, ncall+ndef, NARGS FATAL;
2181ee2e5faSnakanon 	}
2191ee2e5faSnakanon 	for (i = 0, x = a[1]; x != NULL; i++, x = x->nnext) {
2201ee2e5faSnakanon 		/* get call args */
2217c478bd9Sstevel@tonic-gate 		dprintf(("evaluate args[%d], fp=%d:\n", i, fp-frame));
2227c478bd9Sstevel@tonic-gate 		y = execute(x);
2237c478bd9Sstevel@tonic-gate 		oargs[i] = y;
2247c478bd9Sstevel@tonic-gate 		dprintf(("args[%d]: %s %f <%s>, t=%o\n",
2251ee2e5faSnakanon 		    i, y->nval, y->fval,
2261ee2e5faSnakanon 		    isarr(y) ? "(array)" : (char *)y->sval, y->tval));
2271ee2e5faSnakanon 		if (isfunc(y)) {
2281ee2e5faSnakanon 			ERROR "can't use function %s as argument in %s",
2291ee2e5faSnakanon 			    y->nval, s FATAL;
2301ee2e5faSnakanon 		}
2317c478bd9Sstevel@tonic-gate 		if (isarr(y))
2327c478bd9Sstevel@tonic-gate 			args[i] = y;	/* arrays by ref */
2337c478bd9Sstevel@tonic-gate 		else
2347c478bd9Sstevel@tonic-gate 			args[i] = copycell(y);
2357c478bd9Sstevel@tonic-gate 		tempfree(y, "callargs");
2367c478bd9Sstevel@tonic-gate 	}
2377c478bd9Sstevel@tonic-gate 	for (; i < ndef; i++) { /* add null args for ones not provided */
2387c478bd9Sstevel@tonic-gate 		args[i] = gettemp("nullargs");
2397c478bd9Sstevel@tonic-gate 		*args[i] = newcopycell;
2407c478bd9Sstevel@tonic-gate 	}
2417c478bd9Sstevel@tonic-gate 	fp++;	/* now ok to up frame */
2427c478bd9Sstevel@tonic-gate 	if (fp >= frame + nframe) {
2437c478bd9Sstevel@tonic-gate 		int dfp = fp - frame;	/* old index */
2447c478bd9Sstevel@tonic-gate 		frame = (struct Frame *)
2457c478bd9Sstevel@tonic-gate 		    realloc(frame, (nframe += 100) * sizeof (struct Frame));
2467c478bd9Sstevel@tonic-gate 		if (frame == NULL)
2477c478bd9Sstevel@tonic-gate 			ERROR "out of space for stack frames in %s", s FATAL;
2487c478bd9Sstevel@tonic-gate 		fp = frame + dfp;
2497c478bd9Sstevel@tonic-gate 	}
2507c478bd9Sstevel@tonic-gate 	fp->fcncell = fcn;
2517c478bd9Sstevel@tonic-gate 	fp->args = args;
2527c478bd9Sstevel@tonic-gate 	fp->nargs = ndef;	/* number defined with (excess are locals) */
2537c478bd9Sstevel@tonic-gate 	fp->retval = gettemp("retval");
2547c478bd9Sstevel@tonic-gate 
2557c478bd9Sstevel@tonic-gate 	dprintf(("start exec of %s, fp=%d\n", s, fp-frame));
2561ee2e5faSnakanon 	/*LINTED align*/
2577c478bd9Sstevel@tonic-gate 	y = execute((Node *)(fcn->sval));	/* execute body */
2587c478bd9Sstevel@tonic-gate 	dprintf(("finished exec of %s, fp=%d\n", s, fp-frame));
2597c478bd9Sstevel@tonic-gate 
2607c478bd9Sstevel@tonic-gate 	for (i = 0; i < ndef; i++) {
2617c478bd9Sstevel@tonic-gate 		Cell *t = fp->args[i];
2627c478bd9Sstevel@tonic-gate 		if (isarr(t)) {
2637c478bd9Sstevel@tonic-gate 			if (t->csub == CCOPY) {
2647c478bd9Sstevel@tonic-gate 				if (i >= ncall) {
2657c478bd9Sstevel@tonic-gate 					freesymtab(t);
2667c478bd9Sstevel@tonic-gate 					t->csub = CTEMP;
2677c478bd9Sstevel@tonic-gate 				} else {
2687c478bd9Sstevel@tonic-gate 					oargs[i]->tval = t->tval;
2697c478bd9Sstevel@tonic-gate 					oargs[i]->tval &= ~(STR|NUM|DONTFREE);
2707c478bd9Sstevel@tonic-gate 					oargs[i]->sval = t->sval;
2717c478bd9Sstevel@tonic-gate 					tempfree(t, "oargsarr");
2727c478bd9Sstevel@tonic-gate 				}
2737c478bd9Sstevel@tonic-gate 			}
2747c478bd9Sstevel@tonic-gate 		} else {
2757c478bd9Sstevel@tonic-gate 			t->csub = CTEMP;
2767c478bd9Sstevel@tonic-gate 			tempfree(t, "fp->args");
2777c478bd9Sstevel@tonic-gate 			if (t == y) freed = 1;
2787c478bd9Sstevel@tonic-gate 		}
2797c478bd9Sstevel@tonic-gate 	}
2807c478bd9Sstevel@tonic-gate 	tempfree(fcn, "call.fcn");
2817c478bd9Sstevel@tonic-gate 	if (isexit(y) || isnext(y))
2821ee2e5faSnakanon 		return (y);
2831ee2e5faSnakanon 	if (!freed)
2841ee2e5faSnakanon 		tempfree(y, "fcn ret"); /* this can free twice! */
2857c478bd9Sstevel@tonic-gate 	z = fp->retval;			/* return value */
2861ee2e5faSnakanon 	dprintf(("%s returns %g |%s| %o\n",
2871ee2e5faSnakanon 	    s, getfval(z), getsval(z), z->tval));
2887c478bd9Sstevel@tonic-gate 	fp--;
2897c478bd9Sstevel@tonic-gate 	return (z);
2907c478bd9Sstevel@tonic-gate }
2917c478bd9Sstevel@tonic-gate 
2921ee2e5faSnakanon static Cell *
2931ee2e5faSnakanon copycell(Cell *x)	/* make a copy of a cell in a temp */
2947c478bd9Sstevel@tonic-gate {
2957c478bd9Sstevel@tonic-gate 	Cell *y;
2967c478bd9Sstevel@tonic-gate 
2977c478bd9Sstevel@tonic-gate 	y = gettemp("copycell");
2987c478bd9Sstevel@tonic-gate 	y->csub = CCOPY;	/* prevents freeing until call is over */
2997c478bd9Sstevel@tonic-gate 	y->nval = x->nval;
3007c478bd9Sstevel@tonic-gate 	y->sval = x->sval ? tostring(x->sval) : NULL;
3017c478bd9Sstevel@tonic-gate 	y->fval = x->fval;
3021ee2e5faSnakanon 	/* copy is not constant or field is DONTFREE right? */
3031ee2e5faSnakanon 	y->tval = x->tval & ~(CON|FLD|REC|DONTFREE);
3041ee2e5faSnakanon 	return (y);
3057c478bd9Sstevel@tonic-gate }
3067c478bd9Sstevel@tonic-gate 
3071ee2e5faSnakanon /*ARGSUSED*/
3081ee2e5faSnakanon Cell *
3091ee2e5faSnakanon arg(Node **a, int nnn)
3107c478bd9Sstevel@tonic-gate {
3117c478bd9Sstevel@tonic-gate 	int n;
3127c478bd9Sstevel@tonic-gate 
3137c478bd9Sstevel@tonic-gate 	n = (int)a[0];	/* argument number, counting from 0 */
3147c478bd9Sstevel@tonic-gate 	dprintf(("arg(%d), fp->nargs=%d\n", n, fp->nargs));
3151ee2e5faSnakanon 	if (n+1 > fp->nargs) {
3167c478bd9Sstevel@tonic-gate 		ERROR "argument #%d of function %s was not supplied",
3177c478bd9Sstevel@tonic-gate 		    n+1, fp->fcncell->nval FATAL;
3181ee2e5faSnakanon 	}
3191ee2e5faSnakanon 	return (fp->args[n]);
3207c478bd9Sstevel@tonic-gate }
3217c478bd9Sstevel@tonic-gate 
3221ee2e5faSnakanon Cell *
3231ee2e5faSnakanon jump(Node **a, int n)
3247c478bd9Sstevel@tonic-gate {
3257c478bd9Sstevel@tonic-gate 	register Cell *y;
3267c478bd9Sstevel@tonic-gate 
3277c478bd9Sstevel@tonic-gate 	switch (n) {
3287c478bd9Sstevel@tonic-gate 	case EXIT:
3297c478bd9Sstevel@tonic-gate 		if (a[0] != NULL) {
3307c478bd9Sstevel@tonic-gate 			y = execute(a[0]);
33112809310Snakanon 			errorflag = (int)getfval(y);
3327c478bd9Sstevel@tonic-gate 			tempfree(y, "");
3337c478bd9Sstevel@tonic-gate 		}
3347c478bd9Sstevel@tonic-gate 		longjmp(env, 1);
3351ee2e5faSnakanon 		/*NOTREACHED*/
3367c478bd9Sstevel@tonic-gate 	case RETURN:
3377c478bd9Sstevel@tonic-gate 		if (a[0] != NULL) {
3387c478bd9Sstevel@tonic-gate 			y = execute(a[0]);
3397c478bd9Sstevel@tonic-gate 			if ((y->tval & (STR|NUM)) == (STR|NUM)) {
3401ee2e5faSnakanon 				(void) setsval(fp->retval, getsval(y));
3417c478bd9Sstevel@tonic-gate 				fp->retval->fval = getfval(y);
3427c478bd9Sstevel@tonic-gate 				fp->retval->tval |= NUM;
3431ee2e5faSnakanon 			} else if (y->tval & STR)
3441ee2e5faSnakanon 				(void) setsval(fp->retval, getsval(y));
3457c478bd9Sstevel@tonic-gate 			else if (y->tval & NUM)
3461ee2e5faSnakanon 				(void) setfval(fp->retval, getfval(y));
3477c478bd9Sstevel@tonic-gate 			tempfree(y, "");
3487c478bd9Sstevel@tonic-gate 		}
3497c478bd9Sstevel@tonic-gate 		return (jret);
3507c478bd9Sstevel@tonic-gate 	case NEXT:
3517c478bd9Sstevel@tonic-gate 		return (jnext);
3527c478bd9Sstevel@tonic-gate 	case BREAK:
3537c478bd9Sstevel@tonic-gate 		return (jbreak);
3547c478bd9Sstevel@tonic-gate 	case CONTINUE:
3557c478bd9Sstevel@tonic-gate 		return (jcont);
3567c478bd9Sstevel@tonic-gate 	default:	/* can't happen */
3577c478bd9Sstevel@tonic-gate 		ERROR "illegal jump type %d", n FATAL;
3587c478bd9Sstevel@tonic-gate 	}
3591ee2e5faSnakanon 	/*NOTREACHED*/
3601ee2e5faSnakanon 	return (NULL);
3617c478bd9Sstevel@tonic-gate }
3627c478bd9Sstevel@tonic-gate 
3631ee2e5faSnakanon Cell *
3641ee2e5faSnakanon getline(Node **a, int n)
3657c478bd9Sstevel@tonic-gate {
3667c478bd9Sstevel@tonic-gate 	/* a[0] is variable, a[1] is operator, a[2] is filename */
3677c478bd9Sstevel@tonic-gate 	register Cell *r, *x;
3681ee2e5faSnakanon 	uchar *buf;
3697c478bd9Sstevel@tonic-gate 	FILE *fp;
3701ee2e5faSnakanon 	size_t len;
3717c478bd9Sstevel@tonic-gate 
3721ee2e5faSnakanon 	(void) fflush(stdout);	/* in case someone is waiting for a prompt */
3737c478bd9Sstevel@tonic-gate 	r = gettemp("");
3747c478bd9Sstevel@tonic-gate 	if (a[1] != NULL) {		/* getline < file */
3757c478bd9Sstevel@tonic-gate 		x = execute(a[2]);		/* filename */
3767c478bd9Sstevel@tonic-gate 		if ((int)a[1] == '|')	/* input pipe */
3777c478bd9Sstevel@tonic-gate 			a[1] = (Node *)LE;	/* arbitrary flag */
3787c478bd9Sstevel@tonic-gate 		fp = openfile((int)a[1], getsval(x));
3797c478bd9Sstevel@tonic-gate 		tempfree(x, "");
3801ee2e5faSnakanon 		buf = NULL;
3817c478bd9Sstevel@tonic-gate 		if (fp == NULL)
3827c478bd9Sstevel@tonic-gate 			n = -1;
3837c478bd9Sstevel@tonic-gate 		else
3841ee2e5faSnakanon 			n = readrec(&buf, &len, fp);
3851ee2e5faSnakanon 		if (n > 0) {
3861ee2e5faSnakanon 			if (a[0] != NULL) {	/* getline var <file */
3871ee2e5faSnakanon 				(void) setsval(execute(a[0]), buf);
3887c478bd9Sstevel@tonic-gate 			} else {			/* getline <file */
3897c478bd9Sstevel@tonic-gate 				if (!(recloc->tval & DONTFREE))
3907c478bd9Sstevel@tonic-gate 					xfree(recloc->sval);
3911ee2e5faSnakanon 				expand_buf(&record, &record_size, len);
3921ee2e5faSnakanon 				(void) memcpy(record, buf, len);
3931ee2e5faSnakanon 				record[len] = '\0';
3947c478bd9Sstevel@tonic-gate 				recloc->sval = record;
3957c478bd9Sstevel@tonic-gate 				recloc->tval = REC | STR | DONTFREE;
3967c478bd9Sstevel@tonic-gate 				donerec = 1; donefld = 0;
3977c478bd9Sstevel@tonic-gate 			}
3981ee2e5faSnakanon 		}
3991ee2e5faSnakanon 		if (buf != NULL)
4001ee2e5faSnakanon 			free(buf);
4017c478bd9Sstevel@tonic-gate 	} else {			/* bare getline; use current input */
4027c478bd9Sstevel@tonic-gate 		if (a[0] == NULL)	/* getline */
4031ee2e5faSnakanon 			n = getrec(&record, &record_size);
4047c478bd9Sstevel@tonic-gate 		else {			/* getline var */
4051ee2e5faSnakanon 			init_buf(&buf, &len, LINE_INCR);
4061ee2e5faSnakanon 			n = getrec(&buf, &len);
4071ee2e5faSnakanon 			(void) setsval(execute(a[0]), buf);
4081ee2e5faSnakanon 			free(buf);
4097c478bd9Sstevel@tonic-gate 		}
4107c478bd9Sstevel@tonic-gate 	}
4111ee2e5faSnakanon 	(void) setfval(r, (Awkfloat)n);
4121ee2e5faSnakanon 	return (r);
4137c478bd9Sstevel@tonic-gate }
4147c478bd9Sstevel@tonic-gate 
4151ee2e5faSnakanon /*ARGSUSED*/
4161ee2e5faSnakanon Cell *
4171ee2e5faSnakanon getnf(Node **a, int n)
4187c478bd9Sstevel@tonic-gate {
4197c478bd9Sstevel@tonic-gate 	if (donefld == 0)
4207c478bd9Sstevel@tonic-gate 		fldbld();
4211ee2e5faSnakanon 	return ((Cell *)a[0]);
4227c478bd9Sstevel@tonic-gate }
4237c478bd9Sstevel@tonic-gate 
4241ee2e5faSnakanon /*ARGSUSED*/
4251ee2e5faSnakanon Cell *
4261ee2e5faSnakanon array(Node **a, int n)
4277c478bd9Sstevel@tonic-gate {
4287c478bd9Sstevel@tonic-gate 	register Cell *x, *y, *z;
4297c478bd9Sstevel@tonic-gate 	register uchar *s;
4307c478bd9Sstevel@tonic-gate 	register Node *np;
4311ee2e5faSnakanon 	uchar	*buf;
4321ee2e5faSnakanon 	size_t	bsize, tlen, len, slen;
4337c478bd9Sstevel@tonic-gate 
4347c478bd9Sstevel@tonic-gate 	x = execute(a[0]);	/* Cell* for symbol table */
4351ee2e5faSnakanon 	init_buf(&buf, &bsize, LINE_INCR);
4361ee2e5faSnakanon 	buf[0] = '\0';
4371ee2e5faSnakanon 	tlen = 0;
4381ee2e5faSnakanon 	slen = strlen((char *)*SUBSEP);
4397c478bd9Sstevel@tonic-gate 	for (np = a[1]; np; np = np->nnext) {
4407c478bd9Sstevel@tonic-gate 		y = execute(np);	/* subscript */
4417c478bd9Sstevel@tonic-gate 		s = getsval(y);
4421ee2e5faSnakanon 		len = strlen((char *)s);
4431ee2e5faSnakanon 		expand_buf(&buf, &bsize, tlen + len + slen);
4441ee2e5faSnakanon 		(void) memcpy(&buf[tlen], s, len);
4451ee2e5faSnakanon 		tlen += len;
4461ee2e5faSnakanon 		if (np->nnext) {
4471ee2e5faSnakanon 			(void) memcpy(&buf[tlen], *SUBSEP, slen);
4481ee2e5faSnakanon 			tlen += slen;
4491ee2e5faSnakanon 		}
4501ee2e5faSnakanon 		buf[tlen] = '\0';
4517c478bd9Sstevel@tonic-gate 		tempfree(y, "");
4527c478bd9Sstevel@tonic-gate 	}
4537c478bd9Sstevel@tonic-gate 	if (!isarr(x)) {
4547c478bd9Sstevel@tonic-gate 		dprintf(("making %s into an array\n", x->nval));
4557c478bd9Sstevel@tonic-gate 		if (freeable(x))
4567c478bd9Sstevel@tonic-gate 			xfree(x->sval);
4577c478bd9Sstevel@tonic-gate 		x->tval &= ~(STR|NUM|DONTFREE);
4587c478bd9Sstevel@tonic-gate 		x->tval |= ARR;
4597c478bd9Sstevel@tonic-gate 		x->sval = (uchar *) makesymtab(NSYMTAB);
4607c478bd9Sstevel@tonic-gate 	}
4611ee2e5faSnakanon 	/*LINTED align*/
4621ee2e5faSnakanon 	z = setsymtab(buf, (uchar *)"", 0.0, STR|NUM, (Array *)x->sval);
4637c478bd9Sstevel@tonic-gate 	z->ctype = OCELL;
4647c478bd9Sstevel@tonic-gate 	z->csub = CVAR;
4657c478bd9Sstevel@tonic-gate 	tempfree(x, "");
4661ee2e5faSnakanon 	free(buf);
4677c478bd9Sstevel@tonic-gate 	return (z);
4687c478bd9Sstevel@tonic-gate }
4697c478bd9Sstevel@tonic-gate 
4701ee2e5faSnakanon /*ARGSUSED*/
4711ee2e5faSnakanon Cell *
4721ee2e5faSnakanon delete(Node **a, int n)
4737c478bd9Sstevel@tonic-gate {
4747c478bd9Sstevel@tonic-gate 	Cell *x, *y;
4757c478bd9Sstevel@tonic-gate 	Node *np;
4761ee2e5faSnakanon 	uchar *buf, *s;
4771ee2e5faSnakanon 	size_t bsize, tlen, slen, len;
4787c478bd9Sstevel@tonic-gate 
4797c478bd9Sstevel@tonic-gate 	x = execute(a[0]);	/* Cell* for symbol table */
4807c478bd9Sstevel@tonic-gate 	if (!isarr(x))
4811ee2e5faSnakanon 		return (true);
4821ee2e5faSnakanon 	init_buf(&buf, &bsize, LINE_INCR);
4831ee2e5faSnakanon 	buf[0] = '\0';
4841ee2e5faSnakanon 	tlen = 0;
4851ee2e5faSnakanon 	slen = strlen((char *)*SUBSEP);
4867c478bd9Sstevel@tonic-gate 	for (np = a[1]; np; np = np->nnext) {
4877c478bd9Sstevel@tonic-gate 		y = execute(np);	/* subscript */
4887c478bd9Sstevel@tonic-gate 		s = getsval(y);
4891ee2e5faSnakanon 		len = strlen((char *)s);
4901ee2e5faSnakanon 		expand_buf(&buf, &bsize, tlen + len + slen);
4911ee2e5faSnakanon 		(void) memcpy(&buf[tlen], s, len);
4921ee2e5faSnakanon 		tlen += len;
4931ee2e5faSnakanon 		if (np->nnext) {
4941ee2e5faSnakanon 			(void) memcpy(&buf[tlen], *SUBSEP, slen);
4951ee2e5faSnakanon 			tlen += slen;
4961ee2e5faSnakanon 		}
4971ee2e5faSnakanon 		buf[tlen] = '\0';
4987c478bd9Sstevel@tonic-gate 		tempfree(y, "");
4997c478bd9Sstevel@tonic-gate 	}
5007c478bd9Sstevel@tonic-gate 	freeelem(x, buf);
5017c478bd9Sstevel@tonic-gate 	tempfree(x, "");
5021ee2e5faSnakanon 	free(buf);
5031ee2e5faSnakanon 	return (true);
5047c478bd9Sstevel@tonic-gate }
5057c478bd9Sstevel@tonic-gate 
5061ee2e5faSnakanon /*ARGSUSED*/
5071ee2e5faSnakanon Cell *
5081ee2e5faSnakanon intest(Node **a, int n)
5097c478bd9Sstevel@tonic-gate {
5107c478bd9Sstevel@tonic-gate 	register Cell *x, *ap, *k;
5117c478bd9Sstevel@tonic-gate 	Node *p;
5121ee2e5faSnakanon 	uchar *buf;
5137c478bd9Sstevel@tonic-gate 	uchar *s;
5141ee2e5faSnakanon 	size_t bsize, tlen, slen, len;
5157c478bd9Sstevel@tonic-gate 
5167c478bd9Sstevel@tonic-gate 	ap = execute(a[1]);	/* array name */
5177c478bd9Sstevel@tonic-gate 	if (!isarr(ap))
5187c478bd9Sstevel@tonic-gate 		ERROR "%s is not an array", ap->nval FATAL;
5191ee2e5faSnakanon 	init_buf(&buf, &bsize, LINE_INCR);
5207c478bd9Sstevel@tonic-gate 	buf[0] = 0;
5211ee2e5faSnakanon 	tlen = 0;
5221ee2e5faSnakanon 	slen = strlen((char *)*SUBSEP);
5237c478bd9Sstevel@tonic-gate 	for (p = a[0]; p; p = p->nnext) {
5247c478bd9Sstevel@tonic-gate 		x = execute(p);	/* expr */
5257c478bd9Sstevel@tonic-gate 		s = getsval(x);
5261ee2e5faSnakanon 		len = strlen((char *)s);
5271ee2e5faSnakanon 		expand_buf(&buf, &bsize, tlen + len + slen);
5281ee2e5faSnakanon 		(void) memcpy(&buf[tlen], s, len);
5291ee2e5faSnakanon 		tlen += len;
5307c478bd9Sstevel@tonic-gate 		tempfree(x, "");
5311ee2e5faSnakanon 		if (p->nnext) {
5321ee2e5faSnakanon 			(void) memcpy(&buf[tlen], *SUBSEP, slen);
5331ee2e5faSnakanon 			tlen += slen;
5347c478bd9Sstevel@tonic-gate 		}
5351ee2e5faSnakanon 		buf[tlen] = '\0';
5361ee2e5faSnakanon 	}
5371ee2e5faSnakanon 	/*LINTED align*/
5387c478bd9Sstevel@tonic-gate 	k = lookup(buf, (Array *)ap->sval);
5397c478bd9Sstevel@tonic-gate 	tempfree(ap, "");
5401ee2e5faSnakanon 	free(buf);
5417c478bd9Sstevel@tonic-gate 	if (k == NULL)
5427c478bd9Sstevel@tonic-gate 		return (false);
5437c478bd9Sstevel@tonic-gate 	else
5447c478bd9Sstevel@tonic-gate 		return (true);
5457c478bd9Sstevel@tonic-gate }
5467c478bd9Sstevel@tonic-gate 
5477c478bd9Sstevel@tonic-gate 
5481ee2e5faSnakanon Cell *
5491ee2e5faSnakanon matchop(Node **a, int n)
5507c478bd9Sstevel@tonic-gate {
5517c478bd9Sstevel@tonic-gate 	register Cell *x, *y;
5527c478bd9Sstevel@tonic-gate 	register uchar *s, *t;
5537c478bd9Sstevel@tonic-gate 	register int i;
5547c478bd9Sstevel@tonic-gate 	fa *pfa;
5557c478bd9Sstevel@tonic-gate 	int (*mf)() = match, mode = 0;
5567c478bd9Sstevel@tonic-gate 
5577c478bd9Sstevel@tonic-gate 	if (n == MATCHFCN) {
5587c478bd9Sstevel@tonic-gate 		mf = pmatch;
5597c478bd9Sstevel@tonic-gate 		mode = 1;
5607c478bd9Sstevel@tonic-gate 	}
5617c478bd9Sstevel@tonic-gate 	x = execute(a[1]);
5627c478bd9Sstevel@tonic-gate 	s = getsval(x);
5637c478bd9Sstevel@tonic-gate 	if (a[0] == 0)
5647c478bd9Sstevel@tonic-gate 		i = (*mf)(a[2], s);
5657c478bd9Sstevel@tonic-gate 	else {
5667c478bd9Sstevel@tonic-gate 		y = execute(a[2]);
5677c478bd9Sstevel@tonic-gate 		t = getsval(y);
5687c478bd9Sstevel@tonic-gate 		pfa = makedfa(t, mode);
5697c478bd9Sstevel@tonic-gate 		i = (*mf)(pfa, s);
5707c478bd9Sstevel@tonic-gate 		tempfree(y, "");
5717c478bd9Sstevel@tonic-gate 	}
5727c478bd9Sstevel@tonic-gate 	tempfree(x, "");
5737c478bd9Sstevel@tonic-gate 	if (n == MATCHFCN) {
5747c478bd9Sstevel@tonic-gate 		int start = patbeg - s + 1;
5757c478bd9Sstevel@tonic-gate 		if (patlen < 0)
5767c478bd9Sstevel@tonic-gate 			start = 0;
5771ee2e5faSnakanon 		(void) setfval(rstartloc, (Awkfloat)start);
5781ee2e5faSnakanon 		(void) setfval(rlengthloc, (Awkfloat)patlen);
5797c478bd9Sstevel@tonic-gate 		x = gettemp("");
5807c478bd9Sstevel@tonic-gate 		x->tval = NUM;
5817c478bd9Sstevel@tonic-gate 		x->fval = start;
5821ee2e5faSnakanon 		return (x);
5837c478bd9Sstevel@tonic-gate 	} else if (n == MATCH && i == 1 || n == NOTMATCH && i == 0)
5847c478bd9Sstevel@tonic-gate 		return (true);
5857c478bd9Sstevel@tonic-gate 	else
5867c478bd9Sstevel@tonic-gate 		return (false);
5877c478bd9Sstevel@tonic-gate }
5887c478bd9Sstevel@tonic-gate 
5897c478bd9Sstevel@tonic-gate 
5901ee2e5faSnakanon Cell *
5911ee2e5faSnakanon boolop(Node **a, int n)
5927c478bd9Sstevel@tonic-gate {
5937c478bd9Sstevel@tonic-gate 	register Cell *x, *y;
5947c478bd9Sstevel@tonic-gate 	register int i;
5957c478bd9Sstevel@tonic-gate 
5967c478bd9Sstevel@tonic-gate 	x = execute(a[0]);
5977c478bd9Sstevel@tonic-gate 	i = istrue(x);
5987c478bd9Sstevel@tonic-gate 	tempfree(x, "");
5997c478bd9Sstevel@tonic-gate 	switch (n) {
6007c478bd9Sstevel@tonic-gate 	case BOR:
6011ee2e5faSnakanon 		if (i)
6021ee2e5faSnakanon 			return (true);
6037c478bd9Sstevel@tonic-gate 		y = execute(a[1]);
6047c478bd9Sstevel@tonic-gate 		i = istrue(y);
6057c478bd9Sstevel@tonic-gate 		tempfree(y, "");
6061ee2e5faSnakanon 		return (i ? true : false);
6077c478bd9Sstevel@tonic-gate 	case AND:
6081ee2e5faSnakanon 		if (!i)
6091ee2e5faSnakanon 			return (false);
6107c478bd9Sstevel@tonic-gate 		y = execute(a[1]);
6117c478bd9Sstevel@tonic-gate 		i = istrue(y);
6127c478bd9Sstevel@tonic-gate 		tempfree(y, "");
6131ee2e5faSnakanon 		return (i ? true : false);
6147c478bd9Sstevel@tonic-gate 	case NOT:
6151ee2e5faSnakanon 		return (i ? false : true);
6167c478bd9Sstevel@tonic-gate 	default:	/* can't happen */
6177c478bd9Sstevel@tonic-gate 		ERROR "unknown boolean operator %d", n FATAL;
6187c478bd9Sstevel@tonic-gate 	}
6197c478bd9Sstevel@tonic-gate 	/*NOTREACHED*/
6201ee2e5faSnakanon 	return (NULL);
6217c478bd9Sstevel@tonic-gate }
6227c478bd9Sstevel@tonic-gate 
6231ee2e5faSnakanon Cell *
6241ee2e5faSnakanon relop(Node **a, int n)
6257c478bd9Sstevel@tonic-gate {
6267c478bd9Sstevel@tonic-gate 	register int i;
6277c478bd9Sstevel@tonic-gate 	register Cell *x, *y;
6287c478bd9Sstevel@tonic-gate 	Awkfloat j;
6297c478bd9Sstevel@tonic-gate 
6307c478bd9Sstevel@tonic-gate 	x = execute(a[0]);
6317c478bd9Sstevel@tonic-gate 	y = execute(a[1]);
6327c478bd9Sstevel@tonic-gate 	if (x->tval&NUM && y->tval&NUM) {
6337c478bd9Sstevel@tonic-gate 		j = x->fval - y->fval;
6347c478bd9Sstevel@tonic-gate 		i = j < 0 ? -1: (j > 0 ? 1: 0);
6357c478bd9Sstevel@tonic-gate 	} else {
6361ee2e5faSnakanon 		i = strcmp((char *)getsval(x), (char *)getsval(y));
6377c478bd9Sstevel@tonic-gate 	}
6387c478bd9Sstevel@tonic-gate 	tempfree(x, "");
6397c478bd9Sstevel@tonic-gate 	tempfree(y, "");
6407c478bd9Sstevel@tonic-gate 	switch (n) {
6411ee2e5faSnakanon 	case LT:	return (i < 0 ? true : false);
6421ee2e5faSnakanon 	case LE:	return (i <= 0 ? true : false);
6431ee2e5faSnakanon 	case NE:	return (i != 0 ? true : false);
6441ee2e5faSnakanon 	case EQ:	return (i == 0 ? true : false);
6451ee2e5faSnakanon 	case GE:	return (i >= 0 ? true : false);
6461ee2e5faSnakanon 	case GT:	return (i > 0 ? true : false);
6477c478bd9Sstevel@tonic-gate 	default:	/* can't happen */
6487c478bd9Sstevel@tonic-gate 		ERROR "unknown relational operator %d", n FATAL;
6497c478bd9Sstevel@tonic-gate 	}
6507c478bd9Sstevel@tonic-gate 	/*NOTREACHED*/
6511ee2e5faSnakanon 	return (false);
6527c478bd9Sstevel@tonic-gate }
6537c478bd9Sstevel@tonic-gate 
6541ee2e5faSnakanon static void
6551ee2e5faSnakanon tfree(Cell *a, char *s)
6567c478bd9Sstevel@tonic-gate {
6571ee2e5faSnakanon 	if (dbg > 1) {
6581ee2e5faSnakanon 		(void) printf("## tfree %.8s %06lo %s\n",
6591ee2e5faSnakanon 		    s, (ulong_t)a, a->sval ? a->sval : (uchar *)"");
6601ee2e5faSnakanon 	}
6617c478bd9Sstevel@tonic-gate 	if (freeable(a))
6627c478bd9Sstevel@tonic-gate 		xfree(a->sval);
6637c478bd9Sstevel@tonic-gate 	if (a == tmps)
6647c478bd9Sstevel@tonic-gate 		ERROR "tempcell list is curdled" FATAL;
6657c478bd9Sstevel@tonic-gate 	a->cnext = tmps;
6667c478bd9Sstevel@tonic-gate 	tmps = a;
6677c478bd9Sstevel@tonic-gate }
6687c478bd9Sstevel@tonic-gate 
6691ee2e5faSnakanon static Cell *
6701ee2e5faSnakanon gettemp(char *s)
6711ee2e5faSnakanon {
6721ee2e5faSnakanon 	int i;
6737c478bd9Sstevel@tonic-gate 	register Cell *x;
6747c478bd9Sstevel@tonic-gate 
6757c478bd9Sstevel@tonic-gate 	if (!tmps) {
6767c478bd9Sstevel@tonic-gate 		tmps = (Cell *)calloc(100, sizeof (Cell));
6777c478bd9Sstevel@tonic-gate 		if (!tmps)
6787c478bd9Sstevel@tonic-gate 			ERROR "no space for temporaries" FATAL;
6797c478bd9Sstevel@tonic-gate 		for (i = 1; i < 100; i++)
6807c478bd9Sstevel@tonic-gate 			tmps[i-1].cnext = &tmps[i];
6817c478bd9Sstevel@tonic-gate 		tmps[i-1].cnext = 0;
6827c478bd9Sstevel@tonic-gate 	}
6837c478bd9Sstevel@tonic-gate 	x = tmps;
6847c478bd9Sstevel@tonic-gate 	tmps = x->cnext;
6857c478bd9Sstevel@tonic-gate 	*x = tempcell;
6861ee2e5faSnakanon 	if (dbg > 1)
6871ee2e5faSnakanon 		(void) printf("## gtemp %.8s %06lo\n", s, (ulong_t)x);
6887c478bd9Sstevel@tonic-gate 	return (x);
6897c478bd9Sstevel@tonic-gate }
6907c478bd9Sstevel@tonic-gate 
6911ee2e5faSnakanon /*ARGSUSED*/
6921ee2e5faSnakanon Cell *
6931ee2e5faSnakanon indirect(Node **a, int n)
6947c478bd9Sstevel@tonic-gate {
6957c478bd9Sstevel@tonic-gate 	register Cell *x;
6967c478bd9Sstevel@tonic-gate 	register int m;
6977c478bd9Sstevel@tonic-gate 	register uchar *s;
6987c478bd9Sstevel@tonic-gate 
6997c478bd9Sstevel@tonic-gate 	x = execute(a[0]);
70012809310Snakanon 	m = (int)getfval(x);
7011ee2e5faSnakanon 	if (m == 0 && !is_number(s = getsval(x)))	/* suspicion! */
7027c478bd9Sstevel@tonic-gate 		ERROR "illegal field $(%s)", s FATAL;
7037c478bd9Sstevel@tonic-gate 	tempfree(x, "");
7047c478bd9Sstevel@tonic-gate 	x = fieldadr(m);
7057c478bd9Sstevel@tonic-gate 	x->ctype = OCELL;
7067c478bd9Sstevel@tonic-gate 	x->csub = CFLD;
7077c478bd9Sstevel@tonic-gate 	return (x);
7087c478bd9Sstevel@tonic-gate }
7097c478bd9Sstevel@tonic-gate 
7101ee2e5faSnakanon /*ARGSUSED*/
7111ee2e5faSnakanon Cell *
7121ee2e5faSnakanon substr(Node **a, int nnn)
7137c478bd9Sstevel@tonic-gate {
7147c478bd9Sstevel@tonic-gate 	register int k, m, n;
7157c478bd9Sstevel@tonic-gate 	register uchar *s;
7167c478bd9Sstevel@tonic-gate 	int temp;
7177c478bd9Sstevel@tonic-gate 	register Cell *x, *y, *z;
7187c478bd9Sstevel@tonic-gate 
7197c478bd9Sstevel@tonic-gate 	x = execute(a[0]);
7207c478bd9Sstevel@tonic-gate 	y = execute(a[1]);
7217c478bd9Sstevel@tonic-gate 	if (a[2] != 0)
7227c478bd9Sstevel@tonic-gate 		z = execute(a[2]);
7237c478bd9Sstevel@tonic-gate 	s = getsval(x);
7241ee2e5faSnakanon 	k = strlen((char *)s) + 1;
7257c478bd9Sstevel@tonic-gate 	if (k <= 1) {
7267c478bd9Sstevel@tonic-gate 		tempfree(x, "");
7277c478bd9Sstevel@tonic-gate 		tempfree(y, "");
7287c478bd9Sstevel@tonic-gate 		if (a[2] != 0)
7297c478bd9Sstevel@tonic-gate 			tempfree(z, "");
7307c478bd9Sstevel@tonic-gate 		x = gettemp("");
7311ee2e5faSnakanon 		(void) setsval(x, (uchar *)"");
7327c478bd9Sstevel@tonic-gate 		return (x);
7337c478bd9Sstevel@tonic-gate 	}
73412809310Snakanon 	m = (int)getfval(y);
7357c478bd9Sstevel@tonic-gate 	if (m <= 0)
7367c478bd9Sstevel@tonic-gate 		m = 1;
7377c478bd9Sstevel@tonic-gate 	else if (m > k)
7387c478bd9Sstevel@tonic-gate 		m = k;
7397c478bd9Sstevel@tonic-gate 	tempfree(y, "");
7407c478bd9Sstevel@tonic-gate 	if (a[2] != 0) {
74112809310Snakanon 		n = (int)getfval(z);
7427c478bd9Sstevel@tonic-gate 		tempfree(z, "");
7437c478bd9Sstevel@tonic-gate 	} else
7447c478bd9Sstevel@tonic-gate 		n = k - 1;
7457c478bd9Sstevel@tonic-gate 	if (n < 0)
7467c478bd9Sstevel@tonic-gate 		n = 0;
7477c478bd9Sstevel@tonic-gate 	else if (n > k - m)
7487c478bd9Sstevel@tonic-gate 		n = k - m;
7497c478bd9Sstevel@tonic-gate 	dprintf(("substr: m=%d, n=%d, s=%s\n", m, n, s));
7507c478bd9Sstevel@tonic-gate 	y = gettemp("");
7517c478bd9Sstevel@tonic-gate 	temp = s[n + m - 1];	/* with thanks to John Linderman */
7527c478bd9Sstevel@tonic-gate 	s[n + m - 1] = '\0';
7531ee2e5faSnakanon 	(void) setsval(y, s + m - 1);
7547c478bd9Sstevel@tonic-gate 	s[n + m - 1] = temp;
7557c478bd9Sstevel@tonic-gate 	tempfree(x, "");
7567c478bd9Sstevel@tonic-gate 	return (y);
7577c478bd9Sstevel@tonic-gate }
7587c478bd9Sstevel@tonic-gate 
7591ee2e5faSnakanon /*ARGSUSED*/
7601ee2e5faSnakanon Cell *
7611ee2e5faSnakanon sindex(Node **a, int nnn)
7627c478bd9Sstevel@tonic-gate {
7637c478bd9Sstevel@tonic-gate 	register Cell *x, *y, *z;
7647c478bd9Sstevel@tonic-gate 	register uchar *s1, *s2, *p1, *p2, *q;
7657c478bd9Sstevel@tonic-gate 	Awkfloat v = 0.0;
7667c478bd9Sstevel@tonic-gate 
7677c478bd9Sstevel@tonic-gate 	x = execute(a[0]);
7687c478bd9Sstevel@tonic-gate 	s1 = getsval(x);
7697c478bd9Sstevel@tonic-gate 	y = execute(a[1]);
7707c478bd9Sstevel@tonic-gate 	s2 = getsval(y);
7717c478bd9Sstevel@tonic-gate 
7727c478bd9Sstevel@tonic-gate 	z = gettemp("");
7737c478bd9Sstevel@tonic-gate 	for (p1 = s1; *p1 != '\0'; p1++) {
7747c478bd9Sstevel@tonic-gate 		for (q = p1, p2 = s2; *p2 != '\0' && *q == *p2; q++, p2++)
7757c478bd9Sstevel@tonic-gate 			;
7767c478bd9Sstevel@tonic-gate 		if (*p2 == '\0') {
7777c478bd9Sstevel@tonic-gate 			v = (Awkfloat) (p1 - s1 + 1);	/* origin 1 */
7787c478bd9Sstevel@tonic-gate 			break;
7797c478bd9Sstevel@tonic-gate 		}
7807c478bd9Sstevel@tonic-gate 	}
7817c478bd9Sstevel@tonic-gate 	tempfree(x, "");
7827c478bd9Sstevel@tonic-gate 	tempfree(y, "");
7831ee2e5faSnakanon 	(void) setfval(z, v);
7847c478bd9Sstevel@tonic-gate 	return (z);
7857c478bd9Sstevel@tonic-gate }
7867c478bd9Sstevel@tonic-gate 
7871ee2e5faSnakanon void
7881ee2e5faSnakanon format(uchar **bufp, uchar *s, Node *a)
7897c478bd9Sstevel@tonic-gate {
7901ee2e5faSnakanon 	uchar *fmt;
7911ee2e5faSnakanon 	register uchar *os;
7927c478bd9Sstevel@tonic-gate 	register Cell *x;
7931ee2e5faSnakanon 	int flag = 0, len;
7941ee2e5faSnakanon 	uchar_t	*buf;
7951ee2e5faSnakanon 	size_t bufsize, fmtsize, cnt, tcnt, ret;
7967c478bd9Sstevel@tonic-gate 
7971ee2e5faSnakanon 	init_buf(&buf, &bufsize, LINE_INCR);
7981ee2e5faSnakanon 	init_buf(&fmt, &fmtsize, LINE_INCR);
7997c478bd9Sstevel@tonic-gate 	os = s;
8001ee2e5faSnakanon 	cnt = 0;
8017c478bd9Sstevel@tonic-gate 	while (*s) {
8027c478bd9Sstevel@tonic-gate 		if (*s != '%') {
8031ee2e5faSnakanon 			expand_buf(&buf, &bufsize, cnt);
8041ee2e5faSnakanon 			buf[cnt++] = *s++;
8057c478bd9Sstevel@tonic-gate 			continue;
8067c478bd9Sstevel@tonic-gate 		}
8077c478bd9Sstevel@tonic-gate 		if (*(s+1) == '%') {
8081ee2e5faSnakanon 			expand_buf(&buf, &bufsize, cnt);
8091ee2e5faSnakanon 			buf[cnt++] = '%';
8107c478bd9Sstevel@tonic-gate 			s += 2;
8117c478bd9Sstevel@tonic-gate 			continue;
8127c478bd9Sstevel@tonic-gate 		}
8131ee2e5faSnakanon 		for (tcnt = 0; ; s++) {
8141ee2e5faSnakanon 			expand_buf(&fmt, &fmtsize, tcnt);
8151ee2e5faSnakanon 			fmt[tcnt++] = *s;
8161ee2e5faSnakanon 			if (*s == '\0')
8171ee2e5faSnakanon 				break;
8187c478bd9Sstevel@tonic-gate 			if (isalpha(*s) && *s != 'l' && *s != 'h' && *s != 'L')
8197c478bd9Sstevel@tonic-gate 				break;	/* the ansi panoply */
8207c478bd9Sstevel@tonic-gate 			if (*s == '*') {
8217c478bd9Sstevel@tonic-gate 				if (a == NULL) {
8227c478bd9Sstevel@tonic-gate 					ERROR
8237c478bd9Sstevel@tonic-gate 		"not enough args in printf(%s) or sprintf(%s)", os, os FATAL;
8247c478bd9Sstevel@tonic-gate 				}
8257c478bd9Sstevel@tonic-gate 				x = execute(a);
8267c478bd9Sstevel@tonic-gate 				a = a->nnext;
8271ee2e5faSnakanon 				tcnt--;
8281ee2e5faSnakanon 				expand_buf(&fmt, &fmtsize, tcnt + 12);
8291ee2e5faSnakanon 				ret = sprintf((char *)&fmt[tcnt], "%d",
8301ee2e5faSnakanon 				    (int)getfval(x));
8311ee2e5faSnakanon 				tcnt += ret;
8327c478bd9Sstevel@tonic-gate 				tempfree(x, "");
8337c478bd9Sstevel@tonic-gate 			}
8347c478bd9Sstevel@tonic-gate 		}
8351ee2e5faSnakanon 		fmt[tcnt] = '\0';
8361ee2e5faSnakanon 
8377c478bd9Sstevel@tonic-gate 		switch (*s) {
8387c478bd9Sstevel@tonic-gate 		case 'f': case 'e': case 'g': case 'E': case 'G':
8397c478bd9Sstevel@tonic-gate 			flag = 1;
8407c478bd9Sstevel@tonic-gate 			break;
8417c478bd9Sstevel@tonic-gate 		case 'd': case 'i':
8427c478bd9Sstevel@tonic-gate 			flag = 2;
8431ee2e5faSnakanon 			if (*(s-1) == 'l')
8441ee2e5faSnakanon 				break;
8451ee2e5faSnakanon 			fmt[tcnt - 1] = 'l';
8461ee2e5faSnakanon 			expand_buf(&fmt, &fmtsize, tcnt);
8471ee2e5faSnakanon 			fmt[tcnt++] = 'd';
8481ee2e5faSnakanon 			fmt[tcnt] = '\0';
8497c478bd9Sstevel@tonic-gate 			break;
8507c478bd9Sstevel@tonic-gate 		case 'o': case 'x': case 'X': case 'u':
8517c478bd9Sstevel@tonic-gate 			flag = *(s-1) == 'l' ? 2 : 3;
8527c478bd9Sstevel@tonic-gate 			break;
8537c478bd9Sstevel@tonic-gate 		case 's':
8547c478bd9Sstevel@tonic-gate 			flag = 4;
8557c478bd9Sstevel@tonic-gate 			break;
8567c478bd9Sstevel@tonic-gate 		case 'c':
8577c478bd9Sstevel@tonic-gate 			flag = 5;
8587c478bd9Sstevel@tonic-gate 			break;
8597c478bd9Sstevel@tonic-gate 		default:
8607c478bd9Sstevel@tonic-gate 			flag = 0;
8617c478bd9Sstevel@tonic-gate 			break;
8627c478bd9Sstevel@tonic-gate 		}
8637c478bd9Sstevel@tonic-gate 		if (flag == 0) {
8641ee2e5faSnakanon 			len = strlen((char *)fmt);
8651ee2e5faSnakanon 			expand_buf(&buf, &bufsize, cnt + len);
8661ee2e5faSnakanon 			(void) memcpy(&buf[cnt], fmt, len);
8671ee2e5faSnakanon 			cnt += len;
8681ee2e5faSnakanon 			buf[cnt] = '\0';
8697c478bd9Sstevel@tonic-gate 			continue;
8707c478bd9Sstevel@tonic-gate 		}
8717c478bd9Sstevel@tonic-gate 		if (a == NULL) {
8727c478bd9Sstevel@tonic-gate 			ERROR
8737c478bd9Sstevel@tonic-gate 	"not enough args in printf(%s) or sprintf(%s)", os, os FATAL;
8747c478bd9Sstevel@tonic-gate 		}
8757c478bd9Sstevel@tonic-gate 		x = execute(a);
8767c478bd9Sstevel@tonic-gate 		a = a->nnext;
8771ee2e5faSnakanon 		for (;;) {
8781ee2e5faSnakanon 			/* make sure we have at least 1 byte space */
8791ee2e5faSnakanon 			expand_buf(&buf, &bufsize, cnt + 1);
8801ee2e5faSnakanon 			len = bufsize - cnt;
8817c478bd9Sstevel@tonic-gate 			switch (flag) {
8821ee2e5faSnakanon 			case 1:
8831ee2e5faSnakanon 				/*LINTED*/
8841ee2e5faSnakanon 				ret = snprintf((char *)&buf[cnt], len,
8851ee2e5faSnakanon 				    (char *)fmt, getfval(x));
8867c478bd9Sstevel@tonic-gate 				break;
8871ee2e5faSnakanon 			case 2:
8881ee2e5faSnakanon 				/*LINTED*/
8891ee2e5faSnakanon 				ret = snprintf((char *)&buf[cnt], len,
8901ee2e5faSnakanon 				    (char *)fmt, (long)getfval(x));
8911ee2e5faSnakanon 				break;
8921ee2e5faSnakanon 			case 3:
8931ee2e5faSnakanon 				/*LINTED*/
8941ee2e5faSnakanon 				ret = snprintf((char *)&buf[cnt], len,
8951ee2e5faSnakanon 				    (char *)fmt, (int)getfval(x));
8961ee2e5faSnakanon 				break;
8971ee2e5faSnakanon 			case 4:
8981ee2e5faSnakanon 				/*LINTED*/
8991ee2e5faSnakanon 				ret = snprintf((char *)&buf[cnt], len,
9001ee2e5faSnakanon 				    (char *)fmt, getsval(x));
9011ee2e5faSnakanon 				break;
9021ee2e5faSnakanon 			case 5:
9031ee2e5faSnakanon 				if (isnum(x)) {
9041ee2e5faSnakanon 					/*LINTED*/
9051ee2e5faSnakanon 					ret = snprintf((char *)&buf[cnt], len,
9061ee2e5faSnakanon 					    (char *)fmt, (int)getfval(x));
9071ee2e5faSnakanon 				} else {
9081ee2e5faSnakanon 					/*LINTED*/
9091ee2e5faSnakanon 					ret = snprintf((char *)&buf[cnt], len,
9101ee2e5faSnakanon 					    (char *)fmt, getsval(x)[0]);
9111ee2e5faSnakanon 				}
9121ee2e5faSnakanon 				break;
9131ee2e5faSnakanon 			default:
9141ee2e5faSnakanon 				ret = 0;
9151ee2e5faSnakanon 			}
9161ee2e5faSnakanon 			if (ret < len)
9171ee2e5faSnakanon 				break;
9181ee2e5faSnakanon 			expand_buf(&buf, &bufsize, cnt + ret);
9197c478bd9Sstevel@tonic-gate 		}
9207c478bd9Sstevel@tonic-gate 		tempfree(x, "");
9211ee2e5faSnakanon 		cnt += ret;
9227c478bd9Sstevel@tonic-gate 		s++;
9237c478bd9Sstevel@tonic-gate 	}
9241ee2e5faSnakanon 	buf[cnt] = '\0';
9257c478bd9Sstevel@tonic-gate 	for (; a; a = a->nnext)	/* evaluate any remaining args */
9261ee2e5faSnakanon 		(void) execute(a);
9271ee2e5faSnakanon 	*bufp = tostring(buf);
9281ee2e5faSnakanon 	free(buf);
9291ee2e5faSnakanon 	free(fmt);
9307c478bd9Sstevel@tonic-gate }
9317c478bd9Sstevel@tonic-gate 
9321ee2e5faSnakanon /*ARGSUSED*/
9331ee2e5faSnakanon Cell *
9341ee2e5faSnakanon asprintf(Node **a, int n)
9357c478bd9Sstevel@tonic-gate {
9367c478bd9Sstevel@tonic-gate 	register Cell *x;
9377c478bd9Sstevel@tonic-gate 	register Node *y;
9381ee2e5faSnakanon 	uchar *buf;
9397c478bd9Sstevel@tonic-gate 
9407c478bd9Sstevel@tonic-gate 	y = a[0]->nnext;
9417c478bd9Sstevel@tonic-gate 	x = execute(a[0]);
9421ee2e5faSnakanon 	format(&buf, getsval(x), y);
9437c478bd9Sstevel@tonic-gate 	tempfree(x, "");
9447c478bd9Sstevel@tonic-gate 	x = gettemp("");
9451ee2e5faSnakanon 	x->sval = buf;
9467c478bd9Sstevel@tonic-gate 	x->tval = STR;
9477c478bd9Sstevel@tonic-gate 	return (x);
9487c478bd9Sstevel@tonic-gate }
9497c478bd9Sstevel@tonic-gate 
9501ee2e5faSnakanon /*ARGSUSED*/
9511ee2e5faSnakanon Cell *
9521ee2e5faSnakanon aprintf(Node **a, int n)
9537c478bd9Sstevel@tonic-gate {
9547c478bd9Sstevel@tonic-gate 	FILE *fp;
9557c478bd9Sstevel@tonic-gate 	register Cell *x;
9567c478bd9Sstevel@tonic-gate 	register Node *y;
9571ee2e5faSnakanon 	uchar *buf;
9587c478bd9Sstevel@tonic-gate 
9597c478bd9Sstevel@tonic-gate 	y = a[0]->nnext;
9607c478bd9Sstevel@tonic-gate 	x = execute(a[0]);
9611ee2e5faSnakanon 	format(&buf, getsval(x), y);
9627c478bd9Sstevel@tonic-gate 	tempfree(x, "");
9637c478bd9Sstevel@tonic-gate 	if (a[1] == NULL)
9641ee2e5faSnakanon 		(void) fputs((char *)buf, stdout);
9657c478bd9Sstevel@tonic-gate 	else {
9667c478bd9Sstevel@tonic-gate 		fp = redirect((int)a[1], a[2]);
9671ee2e5faSnakanon 		(void) fputs((char *)buf, fp);
9681ee2e5faSnakanon 		(void) fflush(fp);
9697c478bd9Sstevel@tonic-gate 	}
9701ee2e5faSnakanon 	free(buf);
9717c478bd9Sstevel@tonic-gate 	return (true);
9727c478bd9Sstevel@tonic-gate }
9737c478bd9Sstevel@tonic-gate 
9741ee2e5faSnakanon Cell *
9751ee2e5faSnakanon arith(Node **a, int n)
9767c478bd9Sstevel@tonic-gate {
9777c478bd9Sstevel@tonic-gate 	Awkfloat i, j;
9781ee2e5faSnakanon 	double v;
9797c478bd9Sstevel@tonic-gate 	register Cell *x, *y, *z;
9807c478bd9Sstevel@tonic-gate 
9817c478bd9Sstevel@tonic-gate 	x = execute(a[0]);
9827c478bd9Sstevel@tonic-gate 	i = getfval(x);
9837c478bd9Sstevel@tonic-gate 	tempfree(x, "");
9847c478bd9Sstevel@tonic-gate 	if (n != UMINUS) {
9857c478bd9Sstevel@tonic-gate 		y = execute(a[1]);
9867c478bd9Sstevel@tonic-gate 		j = getfval(y);
9877c478bd9Sstevel@tonic-gate 		tempfree(y, "");
9887c478bd9Sstevel@tonic-gate 	}
9897c478bd9Sstevel@tonic-gate 	z = gettemp("");
9907c478bd9Sstevel@tonic-gate 	switch (n) {
9917c478bd9Sstevel@tonic-gate 	case ADD:
9927c478bd9Sstevel@tonic-gate 		i += j;
9937c478bd9Sstevel@tonic-gate 		break;
9947c478bd9Sstevel@tonic-gate 	case MINUS:
9957c478bd9Sstevel@tonic-gate 		i -= j;
9967c478bd9Sstevel@tonic-gate 		break;
9977c478bd9Sstevel@tonic-gate 	case MULT:
9987c478bd9Sstevel@tonic-gate 		i *= j;
9997c478bd9Sstevel@tonic-gate 		break;
10007c478bd9Sstevel@tonic-gate 	case DIVIDE:
10017c478bd9Sstevel@tonic-gate 		if (j == 0)
10027c478bd9Sstevel@tonic-gate 			ERROR "division by zero" FATAL;
10037c478bd9Sstevel@tonic-gate 		i /= j;
10047c478bd9Sstevel@tonic-gate 		break;
10057c478bd9Sstevel@tonic-gate 	case MOD:
10067c478bd9Sstevel@tonic-gate 		if (j == 0)
10077c478bd9Sstevel@tonic-gate 			ERROR "division by zero in mod" FATAL;
10081ee2e5faSnakanon 		(void) modf(i/j, &v);
10097c478bd9Sstevel@tonic-gate 		i = i - j * v;
10107c478bd9Sstevel@tonic-gate 		break;
10117c478bd9Sstevel@tonic-gate 	case UMINUS:
10127c478bd9Sstevel@tonic-gate 		i = -i;
10137c478bd9Sstevel@tonic-gate 		break;
10147c478bd9Sstevel@tonic-gate 	case POWER:
10157c478bd9Sstevel@tonic-gate 		if (j >= 0 && modf(j, &v) == 0.0) /* pos integer exponent */
10167c478bd9Sstevel@tonic-gate 			i = ipow(i, (int)j);
10177c478bd9Sstevel@tonic-gate 		else
10187c478bd9Sstevel@tonic-gate 			i = errcheck(pow(i, j), "pow");
10197c478bd9Sstevel@tonic-gate 		break;
10207c478bd9Sstevel@tonic-gate 	default:	/* can't happen */
10217c478bd9Sstevel@tonic-gate 		ERROR "illegal arithmetic operator %d", n FATAL;
10227c478bd9Sstevel@tonic-gate 	}
10231ee2e5faSnakanon 	(void) setfval(z, i);
10247c478bd9Sstevel@tonic-gate 	return (z);
10257c478bd9Sstevel@tonic-gate }
10267c478bd9Sstevel@tonic-gate 
10271ee2e5faSnakanon static double
10281ee2e5faSnakanon ipow(double x, int n)
10297c478bd9Sstevel@tonic-gate {
10307c478bd9Sstevel@tonic-gate 	double v;
10317c478bd9Sstevel@tonic-gate 
10327c478bd9Sstevel@tonic-gate 	if (n <= 0)
10331ee2e5faSnakanon 		return (1.0);
10347c478bd9Sstevel@tonic-gate 	v = ipow(x, n/2);
10357c478bd9Sstevel@tonic-gate 	if (n % 2 == 0)
10361ee2e5faSnakanon 		return (v * v);
10377c478bd9Sstevel@tonic-gate 	else
10381ee2e5faSnakanon 		return (x * v * v);
10397c478bd9Sstevel@tonic-gate }
10407c478bd9Sstevel@tonic-gate 
10411ee2e5faSnakanon Cell *
10421ee2e5faSnakanon incrdecr(Node **a, int n)
10437c478bd9Sstevel@tonic-gate {
10447c478bd9Sstevel@tonic-gate 	register Cell *x, *z;
10457c478bd9Sstevel@tonic-gate 	register int k;
10467c478bd9Sstevel@tonic-gate 	Awkfloat xf;
10477c478bd9Sstevel@tonic-gate 
10487c478bd9Sstevel@tonic-gate 	x = execute(a[0]);
10497c478bd9Sstevel@tonic-gate 	xf = getfval(x);
10507c478bd9Sstevel@tonic-gate 	k = (n == PREINCR || n == POSTINCR) ? 1 : -1;
10517c478bd9Sstevel@tonic-gate 	if (n == PREINCR || n == PREDECR) {
10521ee2e5faSnakanon 		(void) setfval(x, xf + k);
10537c478bd9Sstevel@tonic-gate 		return (x);
10547c478bd9Sstevel@tonic-gate 	}
10557c478bd9Sstevel@tonic-gate 	z = gettemp("");
10561ee2e5faSnakanon 	(void) setfval(z, xf);
10571ee2e5faSnakanon 	(void) setfval(x, xf + k);
10587c478bd9Sstevel@tonic-gate 	tempfree(x, "");
10597c478bd9Sstevel@tonic-gate 	return (z);
10607c478bd9Sstevel@tonic-gate }
10617c478bd9Sstevel@tonic-gate 
10621ee2e5faSnakanon Cell *
10631ee2e5faSnakanon assign(Node **a, int n)
10647c478bd9Sstevel@tonic-gate {
10657c478bd9Sstevel@tonic-gate 	register Cell *x, *y;
10667c478bd9Sstevel@tonic-gate 	Awkfloat xf, yf;
10671ee2e5faSnakanon 	double v;
10687c478bd9Sstevel@tonic-gate 
10697c478bd9Sstevel@tonic-gate 	y = execute(a[1]);
10707c478bd9Sstevel@tonic-gate 	x = execute(a[0]);	/* order reversed from before... */
10717c478bd9Sstevel@tonic-gate 	if (n == ASSIGN) {	/* ordinary assignment */
10727c478bd9Sstevel@tonic-gate 		if ((y->tval & (STR|NUM)) == (STR|NUM)) {
10731ee2e5faSnakanon 			(void) setsval(x, getsval(y));
10747c478bd9Sstevel@tonic-gate 			x->fval = getfval(y);
10757c478bd9Sstevel@tonic-gate 			x->tval |= NUM;
10761ee2e5faSnakanon 		} else if (y->tval & STR)
10771ee2e5faSnakanon 			(void) setsval(x, getsval(y));
10787c478bd9Sstevel@tonic-gate 		else if (y->tval & NUM)
10791ee2e5faSnakanon 			(void) setfval(x, getfval(y));
10807c478bd9Sstevel@tonic-gate 		else
10817c478bd9Sstevel@tonic-gate 			funnyvar(y, "read value of");
10827c478bd9Sstevel@tonic-gate 		tempfree(y, "");
10837c478bd9Sstevel@tonic-gate 		return (x);
10847c478bd9Sstevel@tonic-gate 	}
10857c478bd9Sstevel@tonic-gate 	xf = getfval(x);
10867c478bd9Sstevel@tonic-gate 	yf = getfval(y);
10877c478bd9Sstevel@tonic-gate 	switch (n) {
10887c478bd9Sstevel@tonic-gate 	case ADDEQ:
10897c478bd9Sstevel@tonic-gate 		xf += yf;
10907c478bd9Sstevel@tonic-gate 		break;
10917c478bd9Sstevel@tonic-gate 	case SUBEQ:
10927c478bd9Sstevel@tonic-gate 		xf -= yf;
10937c478bd9Sstevel@tonic-gate 		break;
10947c478bd9Sstevel@tonic-gate 	case MULTEQ:
10957c478bd9Sstevel@tonic-gate 		xf *= yf;
10967c478bd9Sstevel@tonic-gate 		break;
10977c478bd9Sstevel@tonic-gate 	case DIVEQ:
10987c478bd9Sstevel@tonic-gate 		if (yf == 0)
10997c478bd9Sstevel@tonic-gate 			ERROR "division by zero in /=" FATAL;
11007c478bd9Sstevel@tonic-gate 		xf /= yf;
11017c478bd9Sstevel@tonic-gate 		break;
11027c478bd9Sstevel@tonic-gate 	case MODEQ:
11037c478bd9Sstevel@tonic-gate 		if (yf == 0)
11047c478bd9Sstevel@tonic-gate 			ERROR "division by zero in %%=" FATAL;
11051ee2e5faSnakanon 		(void) modf(xf/yf, &v);
11067c478bd9Sstevel@tonic-gate 		xf = xf - yf * v;
11077c478bd9Sstevel@tonic-gate 		break;
11087c478bd9Sstevel@tonic-gate 	case POWEQ:
11097c478bd9Sstevel@tonic-gate 		if (yf >= 0 && modf(yf, &v) == 0.0) /* pos integer exponent */
11107c478bd9Sstevel@tonic-gate 			xf = ipow(xf, (int)yf);
11117c478bd9Sstevel@tonic-gate 		else
11127c478bd9Sstevel@tonic-gate 			xf = errcheck(pow(xf, yf), "pow");
11137c478bd9Sstevel@tonic-gate 		break;
11147c478bd9Sstevel@tonic-gate 	default:
11157c478bd9Sstevel@tonic-gate 		ERROR "illegal assignment operator %d", n FATAL;
11167c478bd9Sstevel@tonic-gate 		break;
11177c478bd9Sstevel@tonic-gate 	}
11187c478bd9Sstevel@tonic-gate 	tempfree(y, "");
11191ee2e5faSnakanon 	(void) setfval(x, xf);
11207c478bd9Sstevel@tonic-gate 	return (x);
11217c478bd9Sstevel@tonic-gate }
11227c478bd9Sstevel@tonic-gate 
11231ee2e5faSnakanon /*ARGSUSED*/
11241ee2e5faSnakanon Cell *
11251ee2e5faSnakanon cat(Node **a, int q)
11267c478bd9Sstevel@tonic-gate {
11277c478bd9Sstevel@tonic-gate 	register Cell *x, *y, *z;
11287c478bd9Sstevel@tonic-gate 	register int n1, n2;
11297c478bd9Sstevel@tonic-gate 	register uchar *s;
11307c478bd9Sstevel@tonic-gate 
11317c478bd9Sstevel@tonic-gate 	x = execute(a[0]);
11327c478bd9Sstevel@tonic-gate 	y = execute(a[1]);
11331ee2e5faSnakanon 	(void) getsval(x);
11341ee2e5faSnakanon 	(void) getsval(y);
11351ee2e5faSnakanon 	n1 = strlen((char *)x->sval);
11361ee2e5faSnakanon 	n2 = strlen((char *)y->sval);
11377c478bd9Sstevel@tonic-gate 	s = (uchar *)malloc(n1 + n2 + 1);
11381ee2e5faSnakanon 	if (s == NULL) {
11397c478bd9Sstevel@tonic-gate 		ERROR "out of space concatenating %.15s and %.15s",
11407c478bd9Sstevel@tonic-gate 		    x->sval, y->sval FATAL;
11411ee2e5faSnakanon 	}
11421ee2e5faSnakanon 	(void) strcpy((char *)s, (char *)x->sval);
11431ee2e5faSnakanon 	(void) strcpy((char *)s + n1, (char *)y->sval);
11447c478bd9Sstevel@tonic-gate 	tempfree(y, "");
11457c478bd9Sstevel@tonic-gate 	z = gettemp("");
11467c478bd9Sstevel@tonic-gate 	z->sval = s;
11477c478bd9Sstevel@tonic-gate 	z->tval = STR;
11487c478bd9Sstevel@tonic-gate 	tempfree(x, "");
11497c478bd9Sstevel@tonic-gate 	return (z);
11507c478bd9Sstevel@tonic-gate }
11517c478bd9Sstevel@tonic-gate 
11521ee2e5faSnakanon /*ARGSUSED*/
11531ee2e5faSnakanon Cell *
11541ee2e5faSnakanon pastat(Node **a, int n)
11557c478bd9Sstevel@tonic-gate {
11567c478bd9Sstevel@tonic-gate 	register Cell *x;
11577c478bd9Sstevel@tonic-gate 
11587c478bd9Sstevel@tonic-gate 	if (a[0] == 0)
11597c478bd9Sstevel@tonic-gate 		x = execute(a[1]);
11607c478bd9Sstevel@tonic-gate 	else {
11617c478bd9Sstevel@tonic-gate 		x = execute(a[0]);
11627c478bd9Sstevel@tonic-gate 		if (istrue(x)) {
11637c478bd9Sstevel@tonic-gate 			tempfree(x, "");
11647c478bd9Sstevel@tonic-gate 			x = execute(a[1]);
11657c478bd9Sstevel@tonic-gate 		}
11667c478bd9Sstevel@tonic-gate 	}
11671ee2e5faSnakanon 	return (x);
11687c478bd9Sstevel@tonic-gate }
11697c478bd9Sstevel@tonic-gate 
11701ee2e5faSnakanon /*ARGSUSED*/
11711ee2e5faSnakanon Cell *
11721ee2e5faSnakanon dopa2(Node **a, int n)
11737c478bd9Sstevel@tonic-gate {
11747c478bd9Sstevel@tonic-gate 	Cell	*x;
11757c478bd9Sstevel@tonic-gate 	int	pair;
11767c478bd9Sstevel@tonic-gate 	static int	*pairstack = NULL;
11777c478bd9Sstevel@tonic-gate 
11787c478bd9Sstevel@tonic-gate 	if (!pairstack) {
11797c478bd9Sstevel@tonic-gate 		/* first time */
11807c478bd9Sstevel@tonic-gate 		dprintf(("paircnt: %d\n", paircnt));
11817c478bd9Sstevel@tonic-gate 		pairstack = (int *)malloc(sizeof (int) * paircnt);
11827c478bd9Sstevel@tonic-gate 		if (!pairstack)
11837c478bd9Sstevel@tonic-gate 			ERROR "out of space in dopa2" FATAL;
11847c478bd9Sstevel@tonic-gate 		(void) memset(pairstack, 0, sizeof (int) * paircnt);
11857c478bd9Sstevel@tonic-gate 	}
11867c478bd9Sstevel@tonic-gate 
11877c478bd9Sstevel@tonic-gate 	pair = (int)a[3];
11887c478bd9Sstevel@tonic-gate 	if (pairstack[pair] == 0) {
11897c478bd9Sstevel@tonic-gate 		x = execute(a[0]);
11907c478bd9Sstevel@tonic-gate 		if (istrue(x))
11917c478bd9Sstevel@tonic-gate 			pairstack[pair] = 1;
11927c478bd9Sstevel@tonic-gate 		tempfree(x, "");
11937c478bd9Sstevel@tonic-gate 	}
11947c478bd9Sstevel@tonic-gate 	if (pairstack[pair] == 1) {
11957c478bd9Sstevel@tonic-gate 		x = execute(a[1]);
11967c478bd9Sstevel@tonic-gate 		if (istrue(x))
11977c478bd9Sstevel@tonic-gate 			pairstack[pair] = 0;
11987c478bd9Sstevel@tonic-gate 		tempfree(x, "");
11997c478bd9Sstevel@tonic-gate 		x = execute(a[2]);
12007c478bd9Sstevel@tonic-gate 		return (x);
12017c478bd9Sstevel@tonic-gate 	}
12027c478bd9Sstevel@tonic-gate 	return (false);
12037c478bd9Sstevel@tonic-gate }
12047c478bd9Sstevel@tonic-gate 
12051ee2e5faSnakanon /*ARGSUSED*/
12061ee2e5faSnakanon Cell *
12071ee2e5faSnakanon split(Node **a, int nnn)
12087c478bd9Sstevel@tonic-gate {
12097c478bd9Sstevel@tonic-gate 	Cell *x, *y, *ap;
12107c478bd9Sstevel@tonic-gate 	register uchar *s;
12117c478bd9Sstevel@tonic-gate 	register int sep;
12121ee2e5faSnakanon 	uchar *t, temp, num[11], *fs;
12137c478bd9Sstevel@tonic-gate 	int n, tempstat;
12147c478bd9Sstevel@tonic-gate 
12157c478bd9Sstevel@tonic-gate 	y = execute(a[0]);	/* source string */
12167c478bd9Sstevel@tonic-gate 	s = getsval(y);
12177c478bd9Sstevel@tonic-gate 	if (a[2] == 0)		/* fs string */
12187c478bd9Sstevel@tonic-gate 		fs = *FS;
12197c478bd9Sstevel@tonic-gate 	else if ((int)a[3] == STRING) {	/* split(str,arr,"string") */
12207c478bd9Sstevel@tonic-gate 		x = execute(a[2]);
12217c478bd9Sstevel@tonic-gate 		fs = getsval(x);
12227c478bd9Sstevel@tonic-gate 	} else if ((int)a[3] == REGEXPR)
12237c478bd9Sstevel@tonic-gate 		fs = (uchar *)"(regexpr)";	/* split(str,arr,/regexpr/) */
12247c478bd9Sstevel@tonic-gate 	else
12257c478bd9Sstevel@tonic-gate 		ERROR "illegal type of split()" FATAL;
12267c478bd9Sstevel@tonic-gate 	sep = *fs;
12277c478bd9Sstevel@tonic-gate 	ap = execute(a[1]);	/* array name */
12287c478bd9Sstevel@tonic-gate 	freesymtab(ap);
12297c478bd9Sstevel@tonic-gate 	dprintf(("split: s=|%s|, a=%s, sep=|%s|\n", s, ap->nval, fs));
12307c478bd9Sstevel@tonic-gate 	ap->tval &= ~STR;
12317c478bd9Sstevel@tonic-gate 	ap->tval |= ARR;
12327c478bd9Sstevel@tonic-gate 	ap->sval = (uchar *)makesymtab(NSYMTAB);
12337c478bd9Sstevel@tonic-gate 
12347c478bd9Sstevel@tonic-gate 	n = 0;
12351ee2e5faSnakanon 	if (*s != '\0' && strlen((char *)fs) > 1 || (int)a[3] == REGEXPR) {
12361ee2e5faSnakanon 		/* reg expr */
12377c478bd9Sstevel@tonic-gate 		fa *pfa;
12387c478bd9Sstevel@tonic-gate 		if ((int)a[3] == REGEXPR) {	/* it's ready already */
12397c478bd9Sstevel@tonic-gate 			pfa = (fa *)a[2];
12407c478bd9Sstevel@tonic-gate 		} else {
12417c478bd9Sstevel@tonic-gate 			pfa = makedfa(fs, 1);
12427c478bd9Sstevel@tonic-gate 		}
12437c478bd9Sstevel@tonic-gate 		if (nematch(pfa, s)) {
12447c478bd9Sstevel@tonic-gate 			tempstat = pfa->initstat;
12457c478bd9Sstevel@tonic-gate 			pfa->initstat = 2;
12467c478bd9Sstevel@tonic-gate 			do {
12477c478bd9Sstevel@tonic-gate 				n++;
12481ee2e5faSnakanon 				(void) sprintf((char *)num, "%d", n);
12497c478bd9Sstevel@tonic-gate 				temp = *patbeg;
12507c478bd9Sstevel@tonic-gate 				*patbeg = '\0';
12511ee2e5faSnakanon 				if (is_number(s)) {
12521ee2e5faSnakanon 					(void) setsymtab(num, s,
12531ee2e5faSnakanon 					    atof((char *)s),
12541ee2e5faSnakanon 					    /*LINTED align*/
12551ee2e5faSnakanon 					    STR|NUM, (Array *)ap->sval);
12561ee2e5faSnakanon 				} else {
12571ee2e5faSnakanon 					(void) setsymtab(num, s, 0.0,
12581ee2e5faSnakanon 					    /*LINTED align*/
12591ee2e5faSnakanon 					    STR, (Array *)ap->sval);
12601ee2e5faSnakanon 				}
12617c478bd9Sstevel@tonic-gate 				*patbeg = temp;
12627c478bd9Sstevel@tonic-gate 				s = patbeg + patlen;
12637c478bd9Sstevel@tonic-gate 				if (*(patbeg+patlen-1) == 0 || *s == 0) {
12647c478bd9Sstevel@tonic-gate 					n++;
12651ee2e5faSnakanon 					(void) sprintf((char *)num, "%d", n);
12661ee2e5faSnakanon 					(void) setsymtab(num, (uchar *)"", 0.0,
12671ee2e5faSnakanon 					    /*LINTED align*/
12681ee2e5faSnakanon 					    STR, (Array *)ap->sval);
12697c478bd9Sstevel@tonic-gate 					pfa->initstat = tempstat;
12707c478bd9Sstevel@tonic-gate 					goto spdone;
12717c478bd9Sstevel@tonic-gate 				}
12727c478bd9Sstevel@tonic-gate 			} while (nematch(pfa, s));
12737c478bd9Sstevel@tonic-gate 		}
12747c478bd9Sstevel@tonic-gate 		n++;
12751ee2e5faSnakanon 		(void) sprintf((char *)num, "%d", n);
12761ee2e5faSnakanon 		if (is_number(s)) {
12771ee2e5faSnakanon 			(void) setsymtab(num, s, atof((char *)s),
12781ee2e5faSnakanon 			    /*LINTED align*/
12791ee2e5faSnakanon 			    STR|NUM, (Array *)ap->sval);
12801ee2e5faSnakanon 		} else {
12811ee2e5faSnakanon 			/*LINTED align*/
12821ee2e5faSnakanon 			(void) setsymtab(num, s, 0.0, STR, (Array *)ap->sval);
12831ee2e5faSnakanon 		}
12847c478bd9Sstevel@tonic-gate spdone:
12857c478bd9Sstevel@tonic-gate 		pfa = NULL;
12867c478bd9Sstevel@tonic-gate 	} else if (sep == ' ') {
12877c478bd9Sstevel@tonic-gate 		for (n = 0; ; ) {
12887c478bd9Sstevel@tonic-gate 			while (*s == ' ' || *s == '\t' || *s == '\n')
12897c478bd9Sstevel@tonic-gate 				s++;
12907c478bd9Sstevel@tonic-gate 			if (*s == 0)
12917c478bd9Sstevel@tonic-gate 				break;
12927c478bd9Sstevel@tonic-gate 			n++;
12937c478bd9Sstevel@tonic-gate 			t = s;
12947c478bd9Sstevel@tonic-gate 			do
12957c478bd9Sstevel@tonic-gate 				s++;
12961ee2e5faSnakanon 			while (*s != ' ' && *s != '\t' &&
12971ee2e5faSnakanon 			    *s != '\n' && *s != '\0')
12981ee2e5faSnakanon 				;
12997c478bd9Sstevel@tonic-gate 			temp = *s;
13007c478bd9Sstevel@tonic-gate 			*s = '\0';
13011ee2e5faSnakanon 			(void) sprintf((char *)num, "%d", n);
13021ee2e5faSnakanon 			if (is_number(t)) {
13031ee2e5faSnakanon 				(void) setsymtab(num, t, atof((char *)t),
13041ee2e5faSnakanon 				    /*LINTED align*/
13051ee2e5faSnakanon 				    STR|NUM, (Array *)ap->sval);
13061ee2e5faSnakanon 			} else {
13071ee2e5faSnakanon 				(void) setsymtab(num, t, 0.0,
13081ee2e5faSnakanon 				    /*LINTED align*/
13091ee2e5faSnakanon 				    STR, (Array *)ap->sval);
13101ee2e5faSnakanon 			}
13117c478bd9Sstevel@tonic-gate 			*s = temp;
13127c478bd9Sstevel@tonic-gate 			if (*s != 0)
13137c478bd9Sstevel@tonic-gate 				s++;
13147c478bd9Sstevel@tonic-gate 		}
13157c478bd9Sstevel@tonic-gate 	} else if (*s != 0) {
13167c478bd9Sstevel@tonic-gate 		for (;;) {
13177c478bd9Sstevel@tonic-gate 			n++;
13187c478bd9Sstevel@tonic-gate 			t = s;
13197c478bd9Sstevel@tonic-gate 			while (*s != sep && *s != '\n' && *s != '\0')
13207c478bd9Sstevel@tonic-gate 				s++;
13217c478bd9Sstevel@tonic-gate 			temp = *s;
13227c478bd9Sstevel@tonic-gate 			*s = '\0';
13231ee2e5faSnakanon 			(void) sprintf((char *)num, "%d", n);
13241ee2e5faSnakanon 			if (is_number(t)) {
13251ee2e5faSnakanon 				(void) setsymtab(num, t, atof((char *)t),
13261ee2e5faSnakanon 				    /*LINTED align*/
13271ee2e5faSnakanon 				    STR|NUM, (Array *)ap->sval);
13281ee2e5faSnakanon 			} else {
13291ee2e5faSnakanon 				(void) setsymtab(num, t, 0.0,
13301ee2e5faSnakanon 				    /*LINTED align*/
13311ee2e5faSnakanon 				    STR, (Array *)ap->sval);
13321ee2e5faSnakanon 			}
13337c478bd9Sstevel@tonic-gate 			*s = temp;
13347c478bd9Sstevel@tonic-gate 			if (*s++ == 0)
13357c478bd9Sstevel@tonic-gate 				break;
13367c478bd9Sstevel@tonic-gate 		}
13377c478bd9Sstevel@tonic-gate 	}
13387c478bd9Sstevel@tonic-gate 	tempfree(ap, "");
13397c478bd9Sstevel@tonic-gate 	tempfree(y, "");
13407c478bd9Sstevel@tonic-gate 	if (a[2] != 0 && (int)a[3] == STRING)
13417c478bd9Sstevel@tonic-gate 		tempfree(x, "");
13427c478bd9Sstevel@tonic-gate 	x = gettemp("");
13437c478bd9Sstevel@tonic-gate 	x->tval = NUM;
13447c478bd9Sstevel@tonic-gate 	x->fval = n;
13457c478bd9Sstevel@tonic-gate 	return (x);
13467c478bd9Sstevel@tonic-gate }
13477c478bd9Sstevel@tonic-gate 
13481ee2e5faSnakanon /*ARGSUSED*/
13491ee2e5faSnakanon Cell *
13501ee2e5faSnakanon condexpr(Node **a, int n)
13517c478bd9Sstevel@tonic-gate {
13527c478bd9Sstevel@tonic-gate 	register Cell *x;
13537c478bd9Sstevel@tonic-gate 
13547c478bd9Sstevel@tonic-gate 	x = execute(a[0]);
13557c478bd9Sstevel@tonic-gate 	if (istrue(x)) {
13567c478bd9Sstevel@tonic-gate 		tempfree(x, "");
13577c478bd9Sstevel@tonic-gate 		x = execute(a[1]);
13587c478bd9Sstevel@tonic-gate 	} else {
13597c478bd9Sstevel@tonic-gate 		tempfree(x, "");
13607c478bd9Sstevel@tonic-gate 		x = execute(a[2]);
13617c478bd9Sstevel@tonic-gate 	}
13627c478bd9Sstevel@tonic-gate 	return (x);
13637c478bd9Sstevel@tonic-gate }
13647c478bd9Sstevel@tonic-gate 
13651ee2e5faSnakanon /*ARGSUSED*/
13661ee2e5faSnakanon Cell *
13671ee2e5faSnakanon ifstat(Node **a, int n)
13687c478bd9Sstevel@tonic-gate {
13697c478bd9Sstevel@tonic-gate 	register Cell *x;
13707c478bd9Sstevel@tonic-gate 
13717c478bd9Sstevel@tonic-gate 	x = execute(a[0]);
13727c478bd9Sstevel@tonic-gate 	if (istrue(x)) {
13737c478bd9Sstevel@tonic-gate 		tempfree(x, "");
13747c478bd9Sstevel@tonic-gate 		x = execute(a[1]);
13757c478bd9Sstevel@tonic-gate 	} else if (a[2] != 0) {
13767c478bd9Sstevel@tonic-gate 		tempfree(x, "");
13777c478bd9Sstevel@tonic-gate 		x = execute(a[2]);
13787c478bd9Sstevel@tonic-gate 	}
13797c478bd9Sstevel@tonic-gate 	return (x);
13807c478bd9Sstevel@tonic-gate }
13817c478bd9Sstevel@tonic-gate 
13821ee2e5faSnakanon /*ARGSUSED*/
13831ee2e5faSnakanon Cell *
13841ee2e5faSnakanon whilestat(Node **a, int n)
13857c478bd9Sstevel@tonic-gate {
13867c478bd9Sstevel@tonic-gate 	register Cell *x;
13877c478bd9Sstevel@tonic-gate 
13887c478bd9Sstevel@tonic-gate 	for (;;) {
13897c478bd9Sstevel@tonic-gate 		x = execute(a[0]);
13907c478bd9Sstevel@tonic-gate 		if (!istrue(x))
13917c478bd9Sstevel@tonic-gate 			return (x);
13927c478bd9Sstevel@tonic-gate 		tempfree(x, "");
13937c478bd9Sstevel@tonic-gate 		x = execute(a[1]);
13947c478bd9Sstevel@tonic-gate 		if (isbreak(x)) {
13957c478bd9Sstevel@tonic-gate 			x = true;
13967c478bd9Sstevel@tonic-gate 			return (x);
13977c478bd9Sstevel@tonic-gate 		}
13987c478bd9Sstevel@tonic-gate 		if (isnext(x) || isexit(x) || isret(x))
13997c478bd9Sstevel@tonic-gate 			return (x);
14007c478bd9Sstevel@tonic-gate 		tempfree(x, "");
14017c478bd9Sstevel@tonic-gate 	}
14027c478bd9Sstevel@tonic-gate }
14037c478bd9Sstevel@tonic-gate 
14041ee2e5faSnakanon /*ARGSUSED*/
14051ee2e5faSnakanon Cell *
14061ee2e5faSnakanon dostat(Node **a, int n)
14077c478bd9Sstevel@tonic-gate {
14087c478bd9Sstevel@tonic-gate 	register Cell *x;
14097c478bd9Sstevel@tonic-gate 
14107c478bd9Sstevel@tonic-gate 	for (;;) {
14117c478bd9Sstevel@tonic-gate 		x = execute(a[0]);
14127c478bd9Sstevel@tonic-gate 		if (isbreak(x))
14131ee2e5faSnakanon 			return (true);
14147c478bd9Sstevel@tonic-gate 		if (isnext(x) || isexit(x) || isret(x))
14157c478bd9Sstevel@tonic-gate 			return (x);
14167c478bd9Sstevel@tonic-gate 		tempfree(x, "");
14177c478bd9Sstevel@tonic-gate 		x = execute(a[1]);
14187c478bd9Sstevel@tonic-gate 		if (!istrue(x))
14197c478bd9Sstevel@tonic-gate 			return (x);
14207c478bd9Sstevel@tonic-gate 		tempfree(x, "");
14217c478bd9Sstevel@tonic-gate 	}
14227c478bd9Sstevel@tonic-gate }
14237c478bd9Sstevel@tonic-gate 
14241ee2e5faSnakanon /*ARGSUSED*/
14251ee2e5faSnakanon Cell *
14261ee2e5faSnakanon forstat(Node **a, int n)
14277c478bd9Sstevel@tonic-gate {
14287c478bd9Sstevel@tonic-gate 	register Cell *x;
14297c478bd9Sstevel@tonic-gate 
14307c478bd9Sstevel@tonic-gate 	x = execute(a[0]);
14317c478bd9Sstevel@tonic-gate 	tempfree(x, "");
14327c478bd9Sstevel@tonic-gate 	for (;;) {
14337c478bd9Sstevel@tonic-gate 		if (a[1] != 0) {
14347c478bd9Sstevel@tonic-gate 			x = execute(a[1]);
14351ee2e5faSnakanon 			if (!istrue(x))
14361ee2e5faSnakanon 				return (x);
14371ee2e5faSnakanon 			else
14381ee2e5faSnakanon 				tempfree(x, "");
14397c478bd9Sstevel@tonic-gate 		}
14407c478bd9Sstevel@tonic-gate 		x = execute(a[3]);
14417c478bd9Sstevel@tonic-gate 		if (isbreak(x))		/* turn off break */
14421ee2e5faSnakanon 			return (true);
14437c478bd9Sstevel@tonic-gate 		if (isnext(x) || isexit(x) || isret(x))
14447c478bd9Sstevel@tonic-gate 			return (x);
14457c478bd9Sstevel@tonic-gate 		tempfree(x, "");
14467c478bd9Sstevel@tonic-gate 		x = execute(a[2]);
14477c478bd9Sstevel@tonic-gate 		tempfree(x, "");
14487c478bd9Sstevel@tonic-gate 	}
14497c478bd9Sstevel@tonic-gate }
14507c478bd9Sstevel@tonic-gate 
14511ee2e5faSnakanon /*ARGSUSED*/
14521ee2e5faSnakanon Cell *
14531ee2e5faSnakanon instat(Node **a, int n)
14547c478bd9Sstevel@tonic-gate {
14557c478bd9Sstevel@tonic-gate 	register Cell *x, *vp, *arrayp, *cp, *ncp;
14567c478bd9Sstevel@tonic-gate 	Array *tp;
14577c478bd9Sstevel@tonic-gate 	int i;
14587c478bd9Sstevel@tonic-gate 
14597c478bd9Sstevel@tonic-gate 	vp = execute(a[0]);
14607c478bd9Sstevel@tonic-gate 	arrayp = execute(a[1]);
14617c478bd9Sstevel@tonic-gate 	if (!isarr(arrayp))
14627c478bd9Sstevel@tonic-gate 		ERROR "%s is not an array", arrayp->nval FATAL;
14631ee2e5faSnakanon 	/*LINTED align*/
14647c478bd9Sstevel@tonic-gate 	tp = (Array *)arrayp->sval;
14657c478bd9Sstevel@tonic-gate 	tempfree(arrayp, "");
14667c478bd9Sstevel@tonic-gate 	for (i = 0; i < tp->size; i++) { /* this routine knows too much */
14677c478bd9Sstevel@tonic-gate 		for (cp = tp->tab[i]; cp != NULL; cp = ncp) {
14681ee2e5faSnakanon 			(void) setsval(vp, cp->nval);
14697c478bd9Sstevel@tonic-gate 			ncp = cp->cnext;
14707c478bd9Sstevel@tonic-gate 			x = execute(a[2]);
14717c478bd9Sstevel@tonic-gate 			if (isbreak(x)) {
14727c478bd9Sstevel@tonic-gate 				tempfree(vp, "");
14731ee2e5faSnakanon 				return (true);
14747c478bd9Sstevel@tonic-gate 			}
14757c478bd9Sstevel@tonic-gate 			if (isnext(x) || isexit(x) || isret(x)) {
14767c478bd9Sstevel@tonic-gate 				tempfree(vp, "");
14777c478bd9Sstevel@tonic-gate 				return (x);
14787c478bd9Sstevel@tonic-gate 			}
14797c478bd9Sstevel@tonic-gate 			tempfree(x, "");
14807c478bd9Sstevel@tonic-gate 		}
14817c478bd9Sstevel@tonic-gate 	}
14821ee2e5faSnakanon 	return (true);
14837c478bd9Sstevel@tonic-gate }
14847c478bd9Sstevel@tonic-gate 
14851ee2e5faSnakanon /*ARGSUSED*/
14861ee2e5faSnakanon Cell *
14871ee2e5faSnakanon bltin(Node **a, int n)
14887c478bd9Sstevel@tonic-gate {
14897c478bd9Sstevel@tonic-gate 	register Cell *x, *y;
14907c478bd9Sstevel@tonic-gate 	Awkfloat u;
14917c478bd9Sstevel@tonic-gate 	register int t;
14921ee2e5faSnakanon 	uchar *p, *buf;
14937c478bd9Sstevel@tonic-gate 	Node *nextarg;
14947c478bd9Sstevel@tonic-gate 
14957c478bd9Sstevel@tonic-gate 	t = (int)a[0];
14967c478bd9Sstevel@tonic-gate 	x = execute(a[1]);
14977c478bd9Sstevel@tonic-gate 	nextarg = a[1]->nnext;
14987c478bd9Sstevel@tonic-gate 	switch (t) {
14997c478bd9Sstevel@tonic-gate 	case FLENGTH:
15001ee2e5faSnakanon 		u = (Awkfloat)strlen((char *)getsval(x)); break;
15017c478bd9Sstevel@tonic-gate 	case FLOG:
15027c478bd9Sstevel@tonic-gate 		u = errcheck(log(getfval(x)), "log"); break;
15037c478bd9Sstevel@tonic-gate 	case FINT:
15041ee2e5faSnakanon 		(void) modf(getfval(x), &u); break;
15057c478bd9Sstevel@tonic-gate 	case FEXP:
15067c478bd9Sstevel@tonic-gate 		u = errcheck(exp(getfval(x)), "exp"); break;
15077c478bd9Sstevel@tonic-gate 	case FSQRT:
15087c478bd9Sstevel@tonic-gate 		u = errcheck(sqrt(getfval(x)), "sqrt"); break;
15097c478bd9Sstevel@tonic-gate 	case FSIN:
15107c478bd9Sstevel@tonic-gate 		u = sin(getfval(x)); break;
15117c478bd9Sstevel@tonic-gate 	case FCOS:
15127c478bd9Sstevel@tonic-gate 		u = cos(getfval(x)); break;
15137c478bd9Sstevel@tonic-gate 	case FATAN:
15147c478bd9Sstevel@tonic-gate 		if (nextarg == 0) {
15151ee2e5faSnakanon 			ERROR "atan2 requires two arguments; returning 1.0"
15161ee2e5faSnakanon 			    WARNING;
15177c478bd9Sstevel@tonic-gate 			u = 1.0;
15187c478bd9Sstevel@tonic-gate 		} else {
15197c478bd9Sstevel@tonic-gate 			y = execute(a[1]->nnext);
15207c478bd9Sstevel@tonic-gate 			u = atan2(getfval(x), getfval(y));
15217c478bd9Sstevel@tonic-gate 			tempfree(y, "");
15227c478bd9Sstevel@tonic-gate 			nextarg = nextarg->nnext;
15237c478bd9Sstevel@tonic-gate 		}
15247c478bd9Sstevel@tonic-gate 		break;
15257c478bd9Sstevel@tonic-gate 	case FSYSTEM:
15261ee2e5faSnakanon 		/* in case something is buffered already */
15271ee2e5faSnakanon 		(void) fflush(stdout);
15281ee2e5faSnakanon 		/* 256 is unix-dep */
15291ee2e5faSnakanon 		u = (Awkfloat)system((char *)getsval(x)) / 256;
15307c478bd9Sstevel@tonic-gate 		break;
15317c478bd9Sstevel@tonic-gate 	case FRAND:
15327c478bd9Sstevel@tonic-gate 		u = (Awkfloat)(rand() % 32767) / 32767.0;
15337c478bd9Sstevel@tonic-gate 		break;
15347c478bd9Sstevel@tonic-gate 	case FSRAND:
15357c478bd9Sstevel@tonic-gate 		if (x->tval & REC)	/* no argument provided */
15367c478bd9Sstevel@tonic-gate 			u = time((time_t *)0);
15377c478bd9Sstevel@tonic-gate 		else
15387c478bd9Sstevel@tonic-gate 			u = getfval(x);
15397c478bd9Sstevel@tonic-gate 		srand((int)u); u = (int)u;
15407c478bd9Sstevel@tonic-gate 		break;
15417c478bd9Sstevel@tonic-gate 	case FTOUPPER:
15427c478bd9Sstevel@tonic-gate 	case FTOLOWER:
15431ee2e5faSnakanon 		buf = tostring(getsval(x));
15447c478bd9Sstevel@tonic-gate 		if (t == FTOUPPER) {
15457c478bd9Sstevel@tonic-gate 			for (p = buf; *p; p++)
15467c478bd9Sstevel@tonic-gate 				if (islower(*p))
15477c478bd9Sstevel@tonic-gate 					*p = toupper(*p);
15487c478bd9Sstevel@tonic-gate 		} else {
15497c478bd9Sstevel@tonic-gate 			for (p = buf; *p; p++)
15507c478bd9Sstevel@tonic-gate 				if (isupper(*p))
15517c478bd9Sstevel@tonic-gate 					*p = tolower(*p);
15527c478bd9Sstevel@tonic-gate 		}
15537c478bd9Sstevel@tonic-gate 		tempfree(x, "");
15547c478bd9Sstevel@tonic-gate 		x = gettemp("");
15551ee2e5faSnakanon 		(void) setsval(x, buf);
15561ee2e5faSnakanon 		free(buf);
15571ee2e5faSnakanon 		return (x);
15587c478bd9Sstevel@tonic-gate 	default:	/* can't happen */
15597c478bd9Sstevel@tonic-gate 		ERROR "illegal function type %d", t FATAL;
15607c478bd9Sstevel@tonic-gate 		break;
15617c478bd9Sstevel@tonic-gate 	}
15627c478bd9Sstevel@tonic-gate 	tempfree(x, "");
15637c478bd9Sstevel@tonic-gate 	x = gettemp("");
15641ee2e5faSnakanon 	(void) setfval(x, u);
15657c478bd9Sstevel@tonic-gate 	if (nextarg != 0) {
15667c478bd9Sstevel@tonic-gate 		ERROR "warning: function has too many arguments" WARNING;
15677c478bd9Sstevel@tonic-gate 		for (; nextarg; nextarg = nextarg->nnext)
15681ee2e5faSnakanon 			(void) execute(nextarg);
15697c478bd9Sstevel@tonic-gate 	}
15707c478bd9Sstevel@tonic-gate 	return (x);
15717c478bd9Sstevel@tonic-gate }
15727c478bd9Sstevel@tonic-gate 
15731ee2e5faSnakanon /*ARGSUSED*/
15741ee2e5faSnakanon Cell *
15751ee2e5faSnakanon print(Node **a, int n)
15767c478bd9Sstevel@tonic-gate {
15777c478bd9Sstevel@tonic-gate 	register Node *x;
15787c478bd9Sstevel@tonic-gate 	register Cell *y;
15797c478bd9Sstevel@tonic-gate 	FILE *fp;
15807c478bd9Sstevel@tonic-gate 
15817c478bd9Sstevel@tonic-gate 	if (a[1] == 0)
15827c478bd9Sstevel@tonic-gate 		fp = stdout;
15837c478bd9Sstevel@tonic-gate 	else
15847c478bd9Sstevel@tonic-gate 		fp = redirect((int)a[1], a[2]);
15857c478bd9Sstevel@tonic-gate 	for (x = a[0]; x != NULL; x = x->nnext) {
15867c478bd9Sstevel@tonic-gate 		y = execute(x);
15871ee2e5faSnakanon 		(void) fputs((char *)getsval(y), fp);
15887c478bd9Sstevel@tonic-gate 		tempfree(y, "");
15897c478bd9Sstevel@tonic-gate 		if (x->nnext == NULL)
15901ee2e5faSnakanon 			(void) fputs((char *)*ORS, fp);
15917c478bd9Sstevel@tonic-gate 		else
15921ee2e5faSnakanon 			(void) fputs((char *)*OFS, fp);
15937c478bd9Sstevel@tonic-gate 	}
15947c478bd9Sstevel@tonic-gate 	if (a[1] != 0)
15951ee2e5faSnakanon 		(void) fflush(fp);
15967c478bd9Sstevel@tonic-gate 	return (true);
15977c478bd9Sstevel@tonic-gate }
15987c478bd9Sstevel@tonic-gate 
15991ee2e5faSnakanon /*ARGSUSED*/
16001ee2e5faSnakanon Cell *
16011ee2e5faSnakanon nullproc(Node **a, int n)
16027c478bd9Sstevel@tonic-gate {
16031ee2e5faSnakanon 	return (0);
16041ee2e5faSnakanon }
16051ee2e5faSnakanon 
16061ee2e5faSnakanon struct {
16077c478bd9Sstevel@tonic-gate 	FILE	*fp;
16087c478bd9Sstevel@tonic-gate 	uchar	*fname;
16097c478bd9Sstevel@tonic-gate 	int	mode;	/* '|', 'a', 'w' */
16107c478bd9Sstevel@tonic-gate } files[FOPEN_MAX];
16117c478bd9Sstevel@tonic-gate 
16121ee2e5faSnakanon static FILE *
16131ee2e5faSnakanon redirect(int a, Node *b)
16147c478bd9Sstevel@tonic-gate {
16157c478bd9Sstevel@tonic-gate 	FILE *fp;
16167c478bd9Sstevel@tonic-gate 	Cell *x;
16177c478bd9Sstevel@tonic-gate 	uchar *fname;
16187c478bd9Sstevel@tonic-gate 
16197c478bd9Sstevel@tonic-gate 	x = execute(b);
16207c478bd9Sstevel@tonic-gate 	fname = getsval(x);
16217c478bd9Sstevel@tonic-gate 	fp = openfile(a, fname);
16227c478bd9Sstevel@tonic-gate 	if (fp == NULL)
16237c478bd9Sstevel@tonic-gate 		ERROR "can't open file %s", fname FATAL;
16247c478bd9Sstevel@tonic-gate 	tempfree(x, "");
16251ee2e5faSnakanon 	return (fp);
16267c478bd9Sstevel@tonic-gate }
16277c478bd9Sstevel@tonic-gate 
16281ee2e5faSnakanon static FILE *
16291ee2e5faSnakanon openfile(int a, uchar *s)
16307c478bd9Sstevel@tonic-gate {
16317c478bd9Sstevel@tonic-gate 	register int i, m;
16327c478bd9Sstevel@tonic-gate 	register FILE *fp;
16337c478bd9Sstevel@tonic-gate 
16347c478bd9Sstevel@tonic-gate 	if (*s == '\0')
16357c478bd9Sstevel@tonic-gate 		ERROR "null file name in print or getline" FATAL;
16361ee2e5faSnakanon 	for (i = 0; i < FOPEN_MAX; i++) {
16371ee2e5faSnakanon 		if (files[i].fname &&
16381ee2e5faSnakanon 		    strcmp((char *)s, (char *)files[i].fname) == 0) {
16391ee2e5faSnakanon 			if (a == files[i].mode ||
16401ee2e5faSnakanon 			    a == APPEND && files[i].mode == GT) {
16411ee2e5faSnakanon 				return (files[i].fp);
16421ee2e5faSnakanon 			}
16431ee2e5faSnakanon 		}
16441ee2e5faSnakanon 	}
16451ee2e5faSnakanon 	for (i = 0; i < FOPEN_MAX; i++) {
16467c478bd9Sstevel@tonic-gate 		if (files[i].fp == 0)
16477c478bd9Sstevel@tonic-gate 			break;
16481ee2e5faSnakanon 	}
16497c478bd9Sstevel@tonic-gate 	if (i >= FOPEN_MAX)
16507c478bd9Sstevel@tonic-gate 		ERROR "%s makes too many open files", s FATAL;
16511ee2e5faSnakanon 	(void) fflush(stdout);	/* force a semblance of order */
16527c478bd9Sstevel@tonic-gate 	m = a;
16537c478bd9Sstevel@tonic-gate 	if (a == GT) {
16547c478bd9Sstevel@tonic-gate 		fp = fopen((char *)s, "w");
16557c478bd9Sstevel@tonic-gate 	} else if (a == APPEND) {
16567c478bd9Sstevel@tonic-gate 		fp = fopen((char *)s, "a");
16577c478bd9Sstevel@tonic-gate 		m = GT;	/* so can mix > and >> */
16587c478bd9Sstevel@tonic-gate 	} else if (a == '|') {	/* output pipe */
16597c478bd9Sstevel@tonic-gate 		fp = popen((char *)s, "w");
16607c478bd9Sstevel@tonic-gate 	} else if (a == LE) {	/* input pipe */
16617c478bd9Sstevel@tonic-gate 		fp = popen((char *)s, "r");
16627c478bd9Sstevel@tonic-gate 	} else if (a == LT) {	/* getline <file */
16631ee2e5faSnakanon 		fp = strcmp((char *)s, "-") == 0 ?
16641ee2e5faSnakanon 		    stdin : fopen((char *)s, "r");	/* "-" is stdin */
16657c478bd9Sstevel@tonic-gate 	} else	/* can't happen */
16667c478bd9Sstevel@tonic-gate 		ERROR "illegal redirection" FATAL;
16677c478bd9Sstevel@tonic-gate 	if (fp != NULL) {
16687c478bd9Sstevel@tonic-gate 		files[i].fname = tostring(s);
16697c478bd9Sstevel@tonic-gate 		files[i].fp = fp;
16707c478bd9Sstevel@tonic-gate 		files[i].mode = m;
16717c478bd9Sstevel@tonic-gate 	}
16721ee2e5faSnakanon 	return (fp);
16737c478bd9Sstevel@tonic-gate }
16747c478bd9Sstevel@tonic-gate 
16751ee2e5faSnakanon /*ARGSUSED*/
16761ee2e5faSnakanon Cell *
16771ee2e5faSnakanon closefile(Node **a, int n)
16787c478bd9Sstevel@tonic-gate {
16797c478bd9Sstevel@tonic-gate 	register Cell *x;
16807c478bd9Sstevel@tonic-gate 	int i, stat;
16817c478bd9Sstevel@tonic-gate 
16827c478bd9Sstevel@tonic-gate 	x = execute(a[0]);
16831ee2e5faSnakanon 	(void) getsval(x);
16841ee2e5faSnakanon 	for (i = 0; i < FOPEN_MAX; i++) {
16851ee2e5faSnakanon 		if (files[i].fname &&
16861ee2e5faSnakanon 		    strcmp((char *)x->sval, (char *)files[i].fname) == 0) {
16871ee2e5faSnakanon 			if (ferror(files[i].fp)) {
16881ee2e5faSnakanon 				ERROR "i/o error occurred on %s",
16891ee2e5faSnakanon 				    files[i].fname WARNING;
16901ee2e5faSnakanon 			}
16917c478bd9Sstevel@tonic-gate 			if (files[i].mode == '|' || files[i].mode == LE)
16927c478bd9Sstevel@tonic-gate 				stat = pclose(files[i].fp);
16937c478bd9Sstevel@tonic-gate 			else
16947c478bd9Sstevel@tonic-gate 				stat = fclose(files[i].fp);
16951ee2e5faSnakanon 			if (stat == EOF) {
16961ee2e5faSnakanon 				ERROR "i/o error occurred closing %s",
16971ee2e5faSnakanon 				    files[i].fname WARNING;
16981ee2e5faSnakanon 			}
16997c478bd9Sstevel@tonic-gate 			xfree(files[i].fname);
17001ee2e5faSnakanon 			/* watch out for ref thru this */
17011ee2e5faSnakanon 			files[i].fname = NULL;
17027c478bd9Sstevel@tonic-gate 			files[i].fp = NULL;
17037c478bd9Sstevel@tonic-gate 		}
17041ee2e5faSnakanon 	}
17057c478bd9Sstevel@tonic-gate 	tempfree(x, "close");
17067c478bd9Sstevel@tonic-gate 	return (true);
17077c478bd9Sstevel@tonic-gate }
17087c478bd9Sstevel@tonic-gate 
17091ee2e5faSnakanon static void
17101ee2e5faSnakanon closeall(void)
17117c478bd9Sstevel@tonic-gate {
17127c478bd9Sstevel@tonic-gate 	int i, stat;
17137c478bd9Sstevel@tonic-gate 
17141ee2e5faSnakanon 	for (i = 0; i < FOPEN_MAX; i++) {
17157c478bd9Sstevel@tonic-gate 		if (files[i].fp) {
17161ee2e5faSnakanon 			if (ferror(files[i].fp)) {
17171ee2e5faSnakanon 				ERROR "i/o error occurred on %s",
17181ee2e5faSnakanon 				    files[i].fname WARNING;
17191ee2e5faSnakanon 			}
17207c478bd9Sstevel@tonic-gate 			if (files[i].mode == '|' || files[i].mode == LE)
17217c478bd9Sstevel@tonic-gate 				stat = pclose(files[i].fp);
17227c478bd9Sstevel@tonic-gate 			else
17237c478bd9Sstevel@tonic-gate 				stat = fclose(files[i].fp);
17241ee2e5faSnakanon 			if (stat == EOF) {
17251ee2e5faSnakanon 				ERROR "i/o error occurred while closing %s",
17261ee2e5faSnakanon 				    files[i].fname WARNING;
17271ee2e5faSnakanon 			}
17281ee2e5faSnakanon 		}
17297c478bd9Sstevel@tonic-gate 	}
17307c478bd9Sstevel@tonic-gate }
17317c478bd9Sstevel@tonic-gate 
17321ee2e5faSnakanon /*ARGSUSED*/
17331ee2e5faSnakanon Cell *
17341ee2e5faSnakanon sub(Node **a, int nnn)
17357c478bd9Sstevel@tonic-gate {
17361ee2e5faSnakanon 	register uchar *sptr;
17377c478bd9Sstevel@tonic-gate 	register Cell *x, *y, *result;
17381ee2e5faSnakanon 	uchar *buf, *t;
17397c478bd9Sstevel@tonic-gate 	fa *pfa;
17401ee2e5faSnakanon 	size_t	bsize, cnt, len;
17417c478bd9Sstevel@tonic-gate 
17427c478bd9Sstevel@tonic-gate 	x = execute(a[3]);	/* target string */
17437c478bd9Sstevel@tonic-gate 	t = getsval(x);
17447c478bd9Sstevel@tonic-gate 	if (a[0] == 0)
17457c478bd9Sstevel@tonic-gate 		pfa = (fa *)a[1];	/* regular expression */
17467c478bd9Sstevel@tonic-gate 	else {
17477c478bd9Sstevel@tonic-gate 		y = execute(a[1]);
17487c478bd9Sstevel@tonic-gate 		pfa = makedfa(getsval(y), 1);
17497c478bd9Sstevel@tonic-gate 		tempfree(y, "");
17507c478bd9Sstevel@tonic-gate 	}
17517c478bd9Sstevel@tonic-gate 	y = execute(a[2]);	/* replacement string */
17527c478bd9Sstevel@tonic-gate 	result = false;
17537c478bd9Sstevel@tonic-gate 	if (pmatch(pfa, t)) {
17541ee2e5faSnakanon 		init_buf(&buf, &bsize, LINE_INCR);
17551ee2e5faSnakanon 		cnt = 0;
17567c478bd9Sstevel@tonic-gate 		sptr = t;
17571ee2e5faSnakanon 		len = patbeg - sptr;
17581ee2e5faSnakanon 		if (len > 0) {
17591ee2e5faSnakanon 			expand_buf(&buf, &bsize, cnt + len);
17601ee2e5faSnakanon 			(void) memcpy(buf, sptr, len);
17611ee2e5faSnakanon 			cnt += len;
17621ee2e5faSnakanon 		}
17637c478bd9Sstevel@tonic-gate 		sptr = getsval(y);
17641ee2e5faSnakanon 		while (*sptr != 0) {
17651ee2e5faSnakanon 			expand_buf(&buf, &bsize, cnt);
1766*83efe03bScf46844 			if (*sptr == '\\' &&
1767*83efe03bScf46844 			    (*(sptr+1) == '&' || *(sptr+1) == '\\')) {
17687c478bd9Sstevel@tonic-gate 				sptr++;		/* skip \, */
1769*83efe03bScf46844 				buf[cnt++] = *sptr++; /* add & or \ */
17707c478bd9Sstevel@tonic-gate 			} else if (*sptr == '&') {
17711ee2e5faSnakanon 				expand_buf(&buf, &bsize, cnt + patlen);
17727c478bd9Sstevel@tonic-gate 				sptr++;
17731ee2e5faSnakanon 				(void) memcpy(&buf[cnt], patbeg, patlen);
17741ee2e5faSnakanon 				cnt += patlen;
17751ee2e5faSnakanon 			} else {
17761ee2e5faSnakanon 				buf[cnt++] = *sptr++;
17771ee2e5faSnakanon 			}
17781ee2e5faSnakanon 		}
17797c478bd9Sstevel@tonic-gate 		sptr = patbeg + patlen;
17801ee2e5faSnakanon 		if ((patlen == 0 && *patbeg) || (patlen && *(sptr-1))) {
17811ee2e5faSnakanon 			len = strlen((char *)sptr);
17821ee2e5faSnakanon 			expand_buf(&buf, &bsize, cnt + len);
17831ee2e5faSnakanon 			(void) memcpy(&buf[cnt], sptr, len);
17841ee2e5faSnakanon 			cnt += len;
17851ee2e5faSnakanon 		}
17861ee2e5faSnakanon 		buf[cnt] = '\0';
17871ee2e5faSnakanon 		(void) setsval(x, buf);
17881ee2e5faSnakanon 		free(buf);
17891ee2e5faSnakanon 		result = true;
17907c478bd9Sstevel@tonic-gate 	}
17917c478bd9Sstevel@tonic-gate 	tempfree(x, "");
17927c478bd9Sstevel@tonic-gate 	tempfree(y, "");
17931ee2e5faSnakanon 	return (result);
17947c478bd9Sstevel@tonic-gate }
17957c478bd9Sstevel@tonic-gate 
17961ee2e5faSnakanon /*ARGSUSED*/
17971ee2e5faSnakanon Cell *
17981ee2e5faSnakanon gsub(Node **a, int nnn)
17997c478bd9Sstevel@tonic-gate {
18007c478bd9Sstevel@tonic-gate 	register Cell *x, *y;
18011ee2e5faSnakanon 	register uchar *rptr, *sptr, *t;
18021ee2e5faSnakanon 	uchar *buf;
18037c478bd9Sstevel@tonic-gate 	register fa *pfa;
18047c478bd9Sstevel@tonic-gate 	int mflag, tempstat, num;
18051ee2e5faSnakanon 	size_t	bsize, cnt, len;
18067c478bd9Sstevel@tonic-gate 
18077c478bd9Sstevel@tonic-gate 	mflag = 0;	/* if mflag == 0, can replace empty string */
18087c478bd9Sstevel@tonic-gate 	num = 0;
18097c478bd9Sstevel@tonic-gate 	x = execute(a[3]);	/* target string */
18107c478bd9Sstevel@tonic-gate 	t = getsval(x);
18117c478bd9Sstevel@tonic-gate 	if (a[0] == 0)
18127c478bd9Sstevel@tonic-gate 		pfa = (fa *) a[1];	/* regular expression */
18137c478bd9Sstevel@tonic-gate 	else {
18147c478bd9Sstevel@tonic-gate 		y = execute(a[1]);
18157c478bd9Sstevel@tonic-gate 		pfa = makedfa(getsval(y), 1);
18167c478bd9Sstevel@tonic-gate 		tempfree(y, "");
18177c478bd9Sstevel@tonic-gate 	}
18187c478bd9Sstevel@tonic-gate 	y = execute(a[2]);	/* replacement string */
18197c478bd9Sstevel@tonic-gate 	if (pmatch(pfa, t)) {
18207c478bd9Sstevel@tonic-gate 		tempstat = pfa->initstat;
18217c478bd9Sstevel@tonic-gate 		pfa->initstat = 2;
18221ee2e5faSnakanon 		init_buf(&buf, &bsize, LINE_INCR);
18237c478bd9Sstevel@tonic-gate 		rptr = getsval(y);
18241ee2e5faSnakanon 		cnt = 0;
18257c478bd9Sstevel@tonic-gate 		do {
18261ee2e5faSnakanon 			if (patlen == 0 && *patbeg != 0) {
18271ee2e5faSnakanon 				/* matched empty string */
18287c478bd9Sstevel@tonic-gate 				if (mflag == 0) {	/* can replace empty */
18297c478bd9Sstevel@tonic-gate 					num++;
18307c478bd9Sstevel@tonic-gate 					sptr = rptr;
18311ee2e5faSnakanon 					while (*sptr != 0) {
18321ee2e5faSnakanon 						expand_buf(&buf, &bsize, cnt);
18331ee2e5faSnakanon 						if (*sptr == '\\' &&
1834*83efe03bScf46844 						    (*(sptr+1) == '&' ||
1835*83efe03bScf46844 						    *(sptr+1) == '\\')) {
18367c478bd9Sstevel@tonic-gate 							sptr++;
18371ee2e5faSnakanon 							buf[cnt++] = *sptr++;
18387c478bd9Sstevel@tonic-gate 						} else if (*sptr == '&') {
18391ee2e5faSnakanon 							expand_buf(&buf,
18401ee2e5faSnakanon 							    &bsize,
18411ee2e5faSnakanon 							    cnt + patlen);
18427c478bd9Sstevel@tonic-gate 							sptr++;
18431ee2e5faSnakanon 							(void) memcpy(&buf[cnt],
18441ee2e5faSnakanon 							    patbeg, patlen);
18451ee2e5faSnakanon 							cnt += patlen;
18461ee2e5faSnakanon 						} else {
18471ee2e5faSnakanon 							buf[cnt++] = *sptr++;
18481ee2e5faSnakanon 						}
18491ee2e5faSnakanon 					}
18507c478bd9Sstevel@tonic-gate 				}
18517c478bd9Sstevel@tonic-gate 				if (*t == 0)	/* at end */
18527c478bd9Sstevel@tonic-gate 					goto done;
18531ee2e5faSnakanon 				expand_buf(&buf, &bsize, cnt);
18541ee2e5faSnakanon 				buf[cnt++] = *t++;
18557c478bd9Sstevel@tonic-gate 				mflag = 0;
18561ee2e5faSnakanon 			} else {	/* matched nonempty string */
18577c478bd9Sstevel@tonic-gate 				num++;
18587c478bd9Sstevel@tonic-gate 				sptr = t;
18591ee2e5faSnakanon 				len = patbeg - sptr;
18601ee2e5faSnakanon 				if (len > 0) {
18611ee2e5faSnakanon 					expand_buf(&buf, &bsize, cnt + len);
18621ee2e5faSnakanon 					(void) memcpy(&buf[cnt], sptr, len);
18631ee2e5faSnakanon 					cnt += len;
18641ee2e5faSnakanon 				}
18657c478bd9Sstevel@tonic-gate 				sptr = rptr;
18661ee2e5faSnakanon 				while (*sptr != 0) {
18671ee2e5faSnakanon 					expand_buf(&buf, &bsize, cnt);
1868*83efe03bScf46844 					if (*sptr == '\\' &&
1869*83efe03bScf46844 					    (*(sptr+1) == '&' ||
1870*83efe03bScf46844 					    *(sptr+1) == '\\')) {
18717c478bd9Sstevel@tonic-gate 						sptr++;
18721ee2e5faSnakanon 						buf[cnt++] = *sptr++;
18737c478bd9Sstevel@tonic-gate 					} else if (*sptr == '&') {
18741ee2e5faSnakanon 						expand_buf(&buf, &bsize,
18751ee2e5faSnakanon 						    cnt + patlen);
18767c478bd9Sstevel@tonic-gate 						sptr++;
18771ee2e5faSnakanon 						(void) memcpy(&buf[cnt],
18781ee2e5faSnakanon 						    patbeg, patlen);
18791ee2e5faSnakanon 						cnt += patlen;
18801ee2e5faSnakanon 					} else {
18811ee2e5faSnakanon 						buf[cnt++] = *sptr++;
18821ee2e5faSnakanon 					}
18831ee2e5faSnakanon 				}
18847c478bd9Sstevel@tonic-gate 				t = patbeg + patlen;
18857c478bd9Sstevel@tonic-gate 				if ((*(t-1) == 0) || (*t == 0))
18867c478bd9Sstevel@tonic-gate 					goto done;
18877c478bd9Sstevel@tonic-gate 				mflag = 1;
18887c478bd9Sstevel@tonic-gate 			}
18897c478bd9Sstevel@tonic-gate 		} while (pmatch(pfa, t));
18907c478bd9Sstevel@tonic-gate 		sptr = t;
18911ee2e5faSnakanon 		len = strlen((char *)sptr);
18921ee2e5faSnakanon 		expand_buf(&buf, &bsize, len + cnt);
18931ee2e5faSnakanon 		(void) memcpy(&buf[cnt], sptr, len);
18941ee2e5faSnakanon 		cnt += len;
18951ee2e5faSnakanon 	done:
18961ee2e5faSnakanon 		buf[cnt] = '\0';
18971ee2e5faSnakanon 		(void) setsval(x, buf);
18981ee2e5faSnakanon 		free(buf);
18997c478bd9Sstevel@tonic-gate 		pfa->initstat = tempstat;
19007c478bd9Sstevel@tonic-gate 	}
19017c478bd9Sstevel@tonic-gate 	tempfree(x, "");
19027c478bd9Sstevel@tonic-gate 	tempfree(y, "");
19037c478bd9Sstevel@tonic-gate 	x = gettemp("");
19047c478bd9Sstevel@tonic-gate 	x->tval = NUM;
19057c478bd9Sstevel@tonic-gate 	x->fval = num;
19067c478bd9Sstevel@tonic-gate 	return (x);
19077c478bd9Sstevel@tonic-gate }
1908