173 lines · 3.5 KB · C
RawDownload
#include <u.h>#include <libc.h>#include "cpp.h" Includelist	includelist[NINCLUDE]; char	*objname; Once	incblocked[NONCE];int	nblocked = 0; static intoncecmp(Once *a, Once *b){	if(a->qid.path == b->qid.path)	if(a->qid.type == b->qid.type)	if(a->dev == b->dev)	if(a->type == b->type)		return 0;	return 1;} voiddoinclude(Tokenrow *trp){	char fname[256], iname[256], *p;	Includelist *ip;	int angled, len, fd, i;	Dir *d;	Once n; 	trp->tp += 1;	if (trp->tp>=trp->lp)		goto syntax;	if (trp->tp->type!=STRING && trp->tp->type!=LT) {		len = trp->tp - trp->bp;		expandrow(trp, "<include>");		trp->tp = trp->bp+len;	}	if (trp->tp->type==STRING) {		len = trp->tp->len-2;		if (len > sizeof(fname) - 1)			len = sizeof(fname) - 1;		strncpy(fname, (char*)trp->tp->t+1, len);		angled = 0;	} else if (trp->tp->type==LT) {		len = 0;		trp->tp++;		while (trp->tp->type!=GT) {			if (trp->tp>trp->lp || len+trp->tp->len+2 >= sizeof(fname))				goto syntax;			strncpy(fname+len, (char*)trp->tp->t, trp->tp->len);			len += trp->tp->len;			trp->tp++;		}		angled = 1;	} else		goto syntax;	trp->tp += 2;	if (trp->tp < trp->lp || len==0)		goto syntax;	fname[len] = '\0';	if (fname[0]=='/') {		fd = open(fname, 0);		strcpy(iname, fname);	} else {		fd = -1;		if (!angled) {			strcpy(iname, cursource->filename);			p = strrchr(iname, '/');			if (p != NULL) {				*p = '\0';				strcat(iname, "/");				strcat(iname, fname);				fd = open(iname, 0);			}		}		for (i=NINCLUDE-1; fd<0 && i>=0; i--) {			ip = &includelist[i];			if (ip->file==NULL || ip->deleted || (angled && ip->always==0))				continue;			if (strlen(fname)+strlen(ip->file)+2 > sizeof(iname))				continue;			strcpy(iname, ip->file);			strcat(iname, "/");			strcat(iname, fname);			fd = open(iname, 0);		}		if (fd<0 && angled) {			strcpy(iname, cursource->filename);			p = strrchr(iname, '/');			if(p != NULL) {				*p = '\0';				strcat(iname, "/");				strcat(iname, fname);				fd = open(iname, 0);			}		}	}	if ( Mflag>1 || !angled&&Mflag==1 ) {		write(1,objname,strlen(objname));		write(1,iname,strlen(iname));		write(1,"\n",1);	}	if (fd >= 0) {		d = dirfstat(fd);		if (d == nil)			error(FATAL, "Out of memory from dirfstat");		n.qid = d->qid;		n.type = d->type;		n.dev = d->dev;		free(d);		for (i=0; i<nblocked; i++)			if (oncecmp(incblocked+i, &n) == 0)				return;		if (++incdepth > 20)			error(FATAL, "#include too deeply nested");		setsource((char*)newstring((uchar*)iname, strlen(iname), 0), fd, NULL);		genline();	} else {		trp->tp = trp->bp+2;		error(FATAL, "Could not find include file %r", trp);	}	return;syntax:	error(ERROR, "Syntax error in #include");	return;} /* * Generate a line directive for cursource */voidgenline(void){	static Token ta = { UNCLASS, NULL, 0, 0 };	static Tokenrow tr = { &ta, &ta, &ta+1, 1 };	uchar *p; 	if(nolineinfo)		return; 	ta.t = p = (uchar*)outp;	strcpy((char*)p, "#line ");	p += sizeof("#line ")-1;	p = (uchar*)outnum((char*)p, cursource->line);	*p++ = ' '; *p++ = '"';	if (cursource->filename[0]!='/' && wd[0]) {		strcpy((char*)p, wd);		p += strlen(wd);		*p++ = '/';	}	strcpy((char*)p, cursource->filename);	p += strlen((char*)p);	*p++ = '"'; *p++ = '\n';	ta.len = (char*)p-outp;	outp = (char*)p;	tr.tp = tr.bp;	puttokens(&tr);} voidsetobjname(char *f){	int n = strlen(f);	objname = (char*)domalloc(n+5);	strcpy(objname,f);	if(objname[n-2]=='.'){		strcpy(objname+n-1,"$O: ");	}else{		strcpy(objname+n,"$O: ");	}}