#include <u.h>#include <libc.h>#include <stdio.h>#include "cpp.h" extern	int getopt(int, char **, char *);extern	char	*optarg;extern	int	optind;int	verbose;int	Mflag;int	Cplusplus;Nlist	*kwdefined;char	wd[128]; #define	NLSIZE	128 Nlist	*nlist[NLSIZE]; struct	kwtab {	char	*kw;	int	val;	int	flag;} kwtab[] = {	"if",		KIF,		ISKW,	"ifdef",	KIFDEF,		ISKW,	"ifndef",	KIFNDEF,	ISKW,	"elif",		KELIF,		ISKW,	"else",		KELSE,		ISKW,	"endif",	KENDIF,		ISKW,	"include",	KINCLUDE,	ISKW,	"define",	KDEFINE,	ISKW,	"undef",	KUNDEF,		ISKW,	"line",		KLINE,		ISKW,	"error",	KERROR,		ISKW,	"pragma",	KPRAGMA,	ISKW,	"eval",		KEVAL,		ISKW,	"defined",	KDEFINED,	ISDEFINED+ISUNCHANGE,	"__LINE__",	KLINENO,	ISMAC+ISUNCHANGE,	"__FILE__",	KFILE,		ISMAC+ISUNCHANGE,	"__DATE__",	KDATE,		ISMAC+ISUNCHANGE,	"__TIME__",	KTIME,		ISMAC+ISUNCHANGE,	"__STDC__",	KSTDC,		ISUNCHANGE,	NULL}; unsigned long	namebit[077+1];Nlist 	*np; voidsetup(int argc, char **argv){	struct kwtab *kp;	Nlist *np;	Token t;	int fd, i;	char *fp, *dp;	Tokenrow tr;	char *objtype;	static char nbuf[40];	static Token deftoken[1] = {{ NAME, 0, 0, 0, 7, (uchar*)"defined" }};	static Tokenrow deftr = { deftoken, deftoken, deftoken+1, 1 }; 	for (kp=kwtab; kp->kw; kp++) {		t.t = (uchar*)kp->kw;		t.len = strlen(kp->kw);		np = lookup(&t, 1);		np->flag = kp->flag;		np->val = kp->val;		if (np->val == KDEFINED) {			kwdefined = np;			np->val = NAME;			np->vp = &deftr;			np->ap = 0;		}	}	/*	 * For Plan 9, search /objtype/include, then /sys/include	 * (Note that includelist is searched from high end to low	 */	if ((objtype = getenv("objtype"))){		sprintf(nbuf, "/%s/include", objtype);		includelist[1].file = nbuf;		includelist[1].always = 1;	} else {		includelist[1].file = NULL;		error(WARNING, "Unknown $objtype");	}	if (getwd(wd, sizeof(wd))==0)		wd[0] = '\0';	includelist[0].file = "/sys/include";	includelist[0].always = 1;	setsource("", -1, 0);	ARGBEGIN {		case 'N':			for (i=0; i<NINCLUDE; i++)				if (includelist[i].always==1)					includelist[i].deleted = 1;			break;		case 'I':			for (i=NINCLUDE-2; i>=0; i--) {				if (includelist[i].file==NULL) {					includelist[i].always = 1;					includelist[i].file = ARGF();					break;				}			}			if (i<0)				error(FATAL, "Too many -I directives");			break;		case 'D':		case 'U':			setsource("<cmdarg>", -1, ARGF());			maketokenrow(3, &tr);			gettokens(&tr, 1);			doadefine(&tr, ARGC());			unsetsource();			break;		case 'M':			Mflag++;			break;		case 'V':			verbose++;			break;		case '+':			Cplusplus++;			break;		case 'W':			clearwstab();			break;		default:			error(FATAL, "Unknown argument");			break;	} ARGEND	dp = ".";	fp = "<stdin>";	fd = 0;	if (argc > 0) {		if ((fp = strrchr(argv[0], '/')) != NULL) {			int len = fp - argv[0];			dp = (char*)newstring((uchar*)argv[0], len+1, 0);			dp[len] = '\0';		}		fp = (char*)newstring((uchar*)argv[0], strlen(argv[0]), 0);		if ((fd = open(fp, 0)) <= 0)			error(FATAL, "Can't open input file %s", fp);	}	if (argc > 1) {		int fdo = create(argv[1], 1, 0666);		if (fdo<0)			error(FATAL, "Can't open output file %s", argv[1]);		dup(fdo, 1);	}	if (Mflag)		setobjname(fp);	includelist[NINCLUDE-1].always = 0;	includelist[NINCLUDE-1].file = dp;	setsource(fp, fd, NULL);} Nlist *lookup(Token *tp, int install){	unsigned int h;	Nlist *np;	uchar *cp, *cpe; 	h = 0;	for (cp=tp->t, cpe=cp+tp->len; cp<cpe; )		h += *cp++;	h %= NLSIZE;	np = nlist[h];	while (np) {		if (*tp->t==*np->name && tp->len==np->len 		 && strncmp((char*)tp->t, (char*)np->name, tp->len)==0)			return np;		np = np->next;	}	if (install) {		np = new(Nlist);		np->val = 0;		np->vp = NULL;		np->ap = NULL;		np->flag = 0;		np->len = tp->len;		np->name = newstring(tp->t, tp->len, 0);		np->next = nlist[h];		nlist[h] = np;		quickset(tp->t[0], tp->len>1? tp->t[1]:0);		return np;	}	return NULL;}