path: root/Src/loop.c

/*
 * loop.c - loop execution
 *
 * This file is part of zsh, the Z shell.
 *
 * Copyright (c) 1992-1997 Paul Falstad
 * All rights reserved.
 *
 * Permission is hereby granted, without written agreement and without
 * license or royalty fees, to use, copy, modify, and distribute this
 * software and to distribute modified versions of this software for any
 * purpose, provided that the above copyright notice and the following
 * two paragraphs appear in all copies of this software.
 *
 * In no event shall Paul Falstad or the Zsh Development Group be liable
 * to any party for direct, indirect, special, incidental, or consequential
 * damages arising out of the use of this software and its documentation,
 * even if Paul Falstad and the Zsh Development Group have been advised of
 * the possibility of such damage.
 *
 * Paul Falstad and the Zsh Development Group specifically disclaim any
 * warranties, including, but not limited to, the implied warranties of
 * merchantability and fitness for a particular purpose.  The software
 * provided hereunder is on an "as is" basis, and Paul Falstad and the
 * Zsh Development Group have no obligation to provide maintenance,
 * support, updates, enhancements, or modifications.
 *
 */

#include "zsh.mdh"
#include "loop.pro"

/* # of nested loops we are in */
 
/**/
int loops;
 
/* # of continue levels */
 
/**/
int contflag;
 
/* # of break levels */
 
/**/
mod_export int breaks;

/**/
int
execfor(Cmd cmd, LinkList args, int flags)
{
    Forcmd node;
    char *str;
    zlong val = 0;

    node = cmd->u.forcmd;
    if (node->condition) {
	str = dupstring(node->name);
	singsub(&str);
	if (isset(XTRACE)) {
	    char *str2 = dupstring(str);
	    untokenize(str2);
	    printprompt4();
	    fprintf(stderr, "%s\n", str2);
	    fflush(stderr);
	}
	if (!errflag)
	    matheval(str);
	if (errflag)
	    return lastval = errflag;
    } else if (!node->inflag) {
	char **x;

	args = newlinklist();
	for (x = pparams; *x; x++)
	    addlinknode(args, dupstring(*x));
    }
    lastval = 0;
    loops++;
    pushheap();
    cmdpush(CS_FOR);
    for (;;) {
	if (node->condition) {
	    str = dupstring(node->condition);
	    singsub(&str);
	    if (!errflag) {
		while (iblank(*str))
		    str++;
		if (*str) {
		    if (isset(XTRACE)) {
			printprompt4();
			fprintf(stderr, "%s\n", str);
			fflush(stderr);
		    }
		    val = mathevali(str);
		} else
		    val = 1;
	    }
	    if (errflag) {
		if (breaks)
		    breaks--;
		lastval = 1;
		break;
	    }
	    if (!val)
		break;
	} else {
	    if (!args || !(str = (char *) ugetnode(args)))
		break;
	    if (isset(XTRACE)) {
		printprompt4();
		fprintf(stderr, "%s=%s\n", node->name, str);
		fflush(stderr);
	    }
	    setsparam(node->name, ztrdup(str));
	}
	execlist(node->list, 1,
		 (flags & CFLAG_EXEC) && args && empty(args));
	if (breaks) {
	    breaks--;
	    if (breaks || !contflag)
		break;
	    contflag = 0;
	}
	if (node->condition && !errflag) {
	    str = dupstring(node->advance);
	    if (isset(XTRACE)) {
		printprompt4();
		fprintf(stderr, "%s\n", str);
		fflush(stderr);
	    }
	    singsub(&str);
	    if (!errflag)
		matheval(str);
	}
	if (errflag) {
	    if (breaks)
		breaks--;
	    lastval = 1;
	    break;
	}
	freeheap();
    }
    popheap();
    cmdpop();
    loops--;
    return lastval;
}

/**/
int
execselect(Cmd cmd, LinkList args, int flags)
{
    Forcmd node;
    char *str, *s;
    LinkNode n;
    int i, usezle;
    FILE *inp;
    size_t more;

    node = cmd->u.forcmd;
    if (!node->inflag) {
	char **x;

	args = newlinklist();
	for (x = pparams; *x; x++)
	    addlinknode(args, dupstring(*x));
    }
    if (!args || empty(args))
	return 1;
    loops++;
    lastval = 0;
    pushheap();
    cmdpush(CS_SELECT);
    usezle = interact && SHTTY != -1 && isset(USEZLE);
    inp = fdopen(dup(usezle ? SHTTY : 0), "r");
    more = selectlist(args, 0);
    for (;;) {
	for (;;) {
	    if (empty(bufstack)) {
	    	if (usezle) {
		    int oef = errflag;

		    isfirstln = 1;
		    str = (char *)zleread(prompt3, NULL, 0);
		    if (errflag)
			str = NULL;
		    errflag = oef;
	    	} else {
		    str = promptexpand(prompt3, 0, NULL, NULL);
		    zputs(str, stderr);
		    free(str);
		    fflush(stderr);
		    str = fgets(zalloc(256), 256, inp);
	    	}
	    } else
		str = (char *)getlinknode(bufstack);
	    if (!str || errflag) {
		if (breaks)
		    breaks--;
		fprintf(stderr, "\n");
		fflush(stderr);
		goto done;
	    }
	    if ((s = strchr(str, '\n')))
		*s = '\0';
	    if (*str)
	      break;
	    more = selectlist(args, more);
	}
	setsparam("REPLY", ztrdup(str));
	i = atoi(str);
	if (!i)
	    str = "";
	else {
	    for (i--, n = firstnode(args); n && i; incnode(n), i--);
	    if (n)
		str = (char *) getdata(n);
	    else
		str = "";
	}
	setsparam(node->name, ztrdup(str));
	execlist(node->list, 1, 0);
	freeheap();
	if (breaks) {
	    breaks--;
	    if (breaks || !contflag)
		break;
	    contflag = 0;
	}
	if (errflag)
	    break;
    }
  done:
    cmdpop();
    popheap();
    fclose(inp);
    loops--;
    return lastval;
}

/* And this is used to print select lists. */

/**/
size_t
selectlist(LinkList l, size_t start)
{
    size_t longest = 1, fct, fw = 0, colsz, t0, t1, ct;
    LinkNode n;
    char **arr, **ap;

    trashzle();
    ct = countlinknodes(l);
    ap = arr = (char **)alloc((countlinknodes(l) + 1) * sizeof(char **));

    for (n = (LinkNode) firstnode(l); n; incnode(n))
	*ap++ = (char *)getdata(n);
    *ap = NULL;
    for (ap = arr; *ap; ap++)
	if (strlen(*ap) > longest)
	    longest = strlen(*ap);
    t0 = ct;
    longest++;
    while (t0)
	t0 /= 10, longest++;
    /* to compensate for added ')' */
    fct = (columns - 1) / (longest + 3);
    if (fct == 0)
	fct = 1;
    else
	fw = (columns - 1) / fct;
    colsz = (ct + fct - 1) / fct;
    for (t1 = start; t1 != colsz && t1 - start < lines - 2; t1++) {
	ap = arr + t1;
	do {
	    int t2 = strlen(*ap) + 2, t3;

	    fprintf(stderr, "%d) %s", t3 = ap - arr + 1, *ap);
	    while (t3)
		t2++, t3 /= 10;
	    for (; t2 < fw; t2++)
		fputc(' ', stderr);
	    for (t0 = colsz; t0 && *ap; t0--, ap++);
	}
	while (*ap);
	fputc('\n', stderr);
    }

 /* Below is a simple attempt at doing it the Korn Way..
       ap = arr;
       t0 = 0;
       do {
           t0++;
           fprintf(stderr,"%d) %s\n",t0,*ap);
           ap++;
       }
       while (*ap);*/
    fflush(stderr);

    return t1 < colsz ? t1 : 0;
}

/**/
int
execwhile(Cmd cmd, LinkList args, int flags)
{
    struct whilecmd *node;
    int olderrexit, oldval;

    olderrexit = noerrexit;
    node = cmd->u.whilecmd;
    oldval = 0;
    pushheap();
    cmdpush(node->cond ? CS_UNTIL : CS_WHILE);
    loops++;
    for (;;) {
	noerrexit = 1;
	execlist(node->cont, 1, 0);
	noerrexit = olderrexit;
	if (!((lastval == 0) ^ node->cond)) {
	    if (breaks)
		breaks--;
	    lastval = oldval;
	    break;
	}
	execlist(node->loop, 1, 0);
	if (breaks) {
	    breaks--;
	    if (breaks || !contflag)
		break;
	    contflag = 0;
	}
	freeheap();
	if (errflag) {
	    lastval = 1;
	    break;
	}
	oldval = lastval;
    }
    cmdpop();
    popheap();
    loops--;
    return lastval;
}

/**/
int
execrepeat(Cmd cmd, LinkList args, int flags)
{
    int count;

    lastval = 0;
    if (!args || empty(args) || nextnode(firstnode(args))) {
	zerr("bad argument for repeat", NULL, 0);
	return 1;
    }
    count = atoi(peekfirst(args));
    pushheap();
    cmdpush(CS_REPEAT);
    loops++;
    while (count-- > 0) {
	execlist(cmd->u.list, 1, 0);
	freeheap();
	if (breaks) {
	    breaks--;
	    if (breaks || !contflag)
		break;
	    contflag = 0;
	}
	if (errflag) {
	    lastval = 1;
	    break;
	}
    }
    cmdpop();
    popheap();
    loops--;
    return lastval;
}

/**/
int
execif(Cmd cmd, LinkList args, int flags)
{
    struct ifcmd *node;
    int olderrexit, s = 0;
    List *i, *t;

    olderrexit = noerrexit;
    node = cmd->u.ifcmd;
    i = node->ifls;
    t = node->thenls;

    if (!noerrexit)
	noerrexit = 1;
    while (*i) {
	cmdpush(s ? CS_ELIF : CS_IF);
	execlist(*i, 1, 0);
	cmdpop();
	if (!lastval)
	    break;
	s = 1;
	i++;
	t++;
    }
    noerrexit = olderrexit;

    if (*t) {
	cmdpush(*i ? (s ? CS_ELIFTHEN : CS_IFTHEN) : CS_ELSE);
	execlist(*t, 1, flags & CFLAG_EXEC);
	cmdpop();
    } else
	lastval = 0;

    return lastval;
}

/**/
int
execcase(Cmd cmd, LinkList args, int flags)
{
    struct casecmd *node;
    char *word;
    List *l;
    char **p;
    Patprog *pp, pprog;
    int save;

    node = cmd->u.casecmd;
    l = node->lists;
    p = node->pats;
    pp = node->progs;

    word = dupstring(*p++);
    singsub(&word);
    untokenize(word);
    lastval = 0;

    if (node) {
	cmdpush(CS_CASE);
	while (*p) {
	    char *pat = NULL, *opat;

	    pprog = NULL;
	    save = 0;

	    if (isset(XTRACE)) {
		char *pat2;

		opat = pat = dupstring(*p + 1);
		singsub(&pat);
		save = (!strcmp(pat, opat) && *pp != dummy_patprog2);

		pat2 = dupstring(pat);
		untokenize(pat2);
		printprompt4();
		fprintf(stderr, "case %s (%s)\n", word, pat2);
		fflush(stderr);
	    }
	    if (*pp != dummy_patprog1 && *pp != dummy_patprog2)
		pprog = *pp;

	    if (!pprog) {
		if (!pat) {
		    opat = pat = dupstring(*p + 1);
		    singsub(&pat);
		    save = (!strcmp(pat, opat) && *pp != dummy_patprog2);
		}
		if (!(pprog = patcompile(pat, (save ? PAT_ZDUP : PAT_STATIC),
				      NULL)))
		    zerr("bad pattern: %s", pat, 0);
		else if (save)
		    *pp = pprog;
	    }
	    if (pprog && pattry(pprog, word)) {
		do {
		    execlist(*l++, 1, **p == ';' && (flags & CFLAG_EXEC));
		} while(**p++ == '&' && *p);
		break;
	    }
	    p++;
	    pp++;
	    l++;
	}
	cmdpop();
    }
    return lastval;
}