harvey/sys/src/cmd/rc/here.c

#include "rc.h"
#include "exec.h"
#include "io.h"
#include "fns.h"

struct here *here, **ehere;
int ser = 0;
char tmp[] = "/tmp/here0000.0000";
char hex[] = "0123456789abcdef";

void psubst(io*, uchar*);
void pstrs(io*, word*);

void
hexnum(char *p, int n)
{
	*p++ = hex[(n>>12)&0xF];
	*p++ = hex[(n>>8)&0xF];
	*p++ = hex[(n>>4)&0xF];
	*p = hex[n&0xF];
}

tree*
heredoc(tree *tag)
{
	struct here *h = new(struct here);

	if(tag->type != WORD)
		yyerror("Bad here tag");
	h->next = 0;
	if(here)
		*ehere = h;
	else
		here = h;
	ehere = &h->next;
	h->tag = tag;
	hexnum(&tmp[9], getpid());
	hexnum(&tmp[14], ser++);
	h->name = strdup(tmp);
	return token(tmp, WORD);
}

/*
 * bug: lines longer than NLINE get split -- this can cause spurious
 * missubstitution, or a misrecognized EOF marker.
 */
#define	NLINE	4096

void
readhere(void)
{
	int c, subst;
	char *s, *tag;
	char line[NLINE+1];
	io *f;
	struct here *h, *nexth;

	for(h = here; h; h = nexth){
		subst = !h->tag->quoted;
		tag = h->tag->str;
		c = Creat(h->name);
		if(c < 0)
			yyerror("can't create here document");
		f = openfd(c);
		s = line;
		pprompt();
		while((c = rchr(runq->cmdfd)) != EOF){
			if(c == '\n' || s == &line[NLINE]){
				*s = '\0';
				if(tag && strcmp(line, tag) == 0)
					break;
				if(subst)
					psubst(f, (uchar *)line);
				else
					pstr(f, line);
				s = line;
				if(c == '\n'){
					pprompt();
					pchr(f, c);
				}else
					*s++ = c;
			}else
				*s++ = c;
		}
		flush(f);
		closeio(f);
		cleanhere(h->name);
		nexth = h->next;
		efree((char *)h);
	}
	here = 0;
	doprompt = 1;
}

void
psubst(io *f, uchar *s)
{
	int savec, n;
	uchar *t, *u;
	Rune r;
	word *star;

	while(*s){
		if(*s != '$'){		/* copy plain text rune */
			if(*s < Runeself)
				pchr(f, *s++);
			else{
				n = chartorune(&r, (char *)s);
				while(n-- > 0)
					pchr(f, *s++);
			}
		}else{			/* $something -- perform substitution */
			t = ++s;
			if(*t == '$')
				pchr(f, *t++);
			else{
				while(*t && idchr(*t))
					t++;
				savec = *t;
				*t = '\0';
				n = 0;
				for(u = s; *u && '0' <= *u && *u <= '9'; u++)
					n = n*10 + *u - '0';
				if(n && *u == '\0'){
					star = vlook("*")->val;
					if(star && 1 <= n && n <= count(star)){
						while(--n)
							star = star->next;
						pstr(f, star->word);
					}
				}else
					pstrs(f, vlook((char *)s)->val);
				*t = savec;
				if(savec == '^')
					t++;
			}
			s = t;
		}
	}
}

void
pstrs(io *f, word *a)
{
	if(a){
		while(a->next && a->next->word){
			pstr(f, a->word);
			pchr(f, ' ');
			a = a->next;
		}
		pstr(f, a->word);
	}
}