shithub: scc

Download patch

ref: 8fb3ee51e2eacdf956288a5434dbd2cd57a161ec
parent: 7a32845fcd3b08eb599634ef26bfd963944d630c
author: Roberto E. Vargas Caballero <[email protected]>
date: Fri May 8 14:15:43 EDT 2015

Add include handler

It is the first version of the preprocessor handler, and it
does the correct work for include of local files.

--- a/cc1/cc1.h
+++ b/cc1/cc1.h
@@ -40,6 +40,7 @@
 		int i;
 		char *s;
 		uint8_t token;
+		void (*fun)(char *);
 	} u;
 	struct symbol *next;
 	struct symbol *hash;
@@ -96,6 +97,7 @@
 	NS_IDEN,
 	NS_TAG,
 	NS_LABEL,
+	NS_CPP,
 	NS_STRUCTS,
 	NR_NAMESPACES
 };
@@ -264,7 +266,7 @@
 extern void discard(void);
 extern char *filename(void);
 extern unsigned short fileline(void);
-extern void addinput(char *fname);
+extern bool addinput(char *fname);
 extern void delinput(void);
 #define accept(t) ((yytoken == (t)) ? next() : 0)
 
--- a/cc1/lex.c
+++ b/cc1/lex.c
@@ -1,4 +1,5 @@
 
+#include <assert.h>
 #include <inttypes.h>
 #include <setjmp.h>
 #include <stdio.h>
@@ -10,7 +11,7 @@
 #include "../inc/cc.h"
 #include "cc1.h"
 
-#define INPUTSIZ 128
+#define INPUTSIZ 120
 
 typedef struct input Input;
 
@@ -17,7 +18,7 @@
 struct input {
 	char *fname;
 	unsigned short nline;
-	int8_t cnt;
+	int cnt;
 	FILE *fp;
 	char *line, *ptr;
 	struct input *next;
@@ -35,32 +36,35 @@
 
 static Input *input;
 
-void
+bool
 addinput(char *fname)
 {
-	Input *ip, *oip = input;
+	Input *ip;
+	FILE *fp;
+	unsigned short nline = 1;
 
+	if (fname) {
+		if ((fp = fopen(fname, "r")) == NULL)
+			return 0;
+		fname = xstrdup(fname);
+	} else if (!input) {
+		fp = stdin;
+		fname = "(stdin)";
+	} else {
+		fname = input->fname;
+		nline = input->nline;
+		fp = NULL;
+	}
+
 	ip = xmalloc(sizeof(Input));
+	ip->fname = fname;
 	ip->next = input;
 	ip->line = NULL;
 	ip->cnt = 0;
-	ip->nline = 1;
+	ip->nline = nline;
+	ip->fp = fp;
 	input = ip;
-
-	if (fname) {
-		if ((ip->fp = fopen(fname, "r")) == NULL)
-			die("file '%s' not found", fname);
-		ip->fname = xstrdup(fname);
-		next();
-	} else if (!oip) {
-		ip->fp = stdin;
-		ip->fname = "(stdin)";
-		next();
-	} else {
-		ip->fname = oip->fname;
-		ip->fp = NULL;
-		ip->nline = input->nline;
-	}
+	return 1;
 }
 
 void
@@ -99,6 +103,110 @@
 		die("input file '%s' too long", input->fname);
 }
 
+/* TODO: preprocessor error must not rise recover */
+static void
+preprocessor(void)
+{
+	char str[IDENTSIZ+1], *p, *q;
+	unsigned short cnt, n;
+	Symbol *sym;
+
+	p = input->ptr;
+	q = &p[input->cnt-1];
+	while (q > p && isspace(*q))
+		++q;
+	while (isspace(*p))
+		++p;
+	for (q = p; isalpha(*q); ++q)
+		/* nothing */;
+	if ((n = q - p) > IDENTSIZ)
+		goto bad_directive;
+	strncpy(str, p, n);
+	str[n] = '\0';
+
+	/* discard this line for the lexer */
+	input->cnt = 0;
+	if ((sym = lookup(str, NS_CPP)) == NULL)
+		goto bad_directive;
+	(*sym->u.fun)(q);
+	return;
+
+bad_directive:
+	error("incorrect preprocessor directive");
+}
+
+void
+include(char *s)
+{
+	char fname[FILENAME_MAX], delim, c, *p;
+	size_t len;
+
+	while (isspace(*s))
+		++s;
+	if ((c = *s++) == '>')
+		delim = '>';
+	else if (c == '"')
+		delim = '"';
+	else
+		goto bad_include;
+
+	for (p = s; (c = *p) && c != delim; ++p)
+		/* nothing */;
+	if (c == '\0')
+		goto bad_include;
+
+	len = p - s;
+	if (delim == '"') {
+		if (len >= FILENAME_MAX)
+			goto too_long;
+		strncpy(fname, s, len);
+		fname[len] = '\0';
+		if (!addinput(fname))
+			goto not_found;
+		return;
+	}
+	abort();
+
+	return;
+
+not_found:
+	error("included file '%s' not found", fname);
+too_long:
+	error("file name in include too long");
+bad_include:
+	error("#include expects \"FILENAME\" or <FILENAME>");
+}
+
+void
+define(char *str)
+{
+	
+}
+
+void
+undef(char *str)
+{
+	fprintf(stderr, "Esto en un undef\n");
+}
+
+void
+ifdef(char *str)
+{
+	fprintf(stderr, "Esto en un ifdef\n");
+}
+
+void
+ifndef(char *str)
+{
+	fprintf(stderr, "Esto en un ifndef\n");
+}
+
+void
+endif(char *str)
+{
+	fprintf(stderr, "Esto en un endif\n");
+}
+
 static int
 readline(void)
 {
@@ -105,8 +213,12 @@
 	char *bp, *ptr;
 	uint8_t n;
 	int c;
-	FILE *fp = input->fp;
+	FILE *fp;
 
+repeat:
+	if (!input)
+		return EOF;
+	fp = input->fp;
 	if (!input->line)
 		input->line = xmalloc(INPUTSIZ);
 	ptr = input->line;
@@ -115,8 +227,10 @@
 		if (c == '\n')
 			newline();
 	}
-	if (c == EOF)
-		return EOF;
+	if (c == EOF) {
+		delinput();
+		goto repeat;
+	}
 	ungetc(c, fp);
 
 	for (bp = ptr; (c = getc(fp)) != EOF && c != '\n'; ) {
@@ -134,12 +248,22 @@
 	*bp = ' ';
 	input->cnt = bp - ptr;
 	input->ptr = ptr;
-	return *input->ptr++;
+
+	if ((c = *input->ptr++) == '#') {
+		*bp = '\0';
+		preprocessor();
+		goto repeat;
+	}
+	return c;
 }
 
 static int
 backchar(int c)
 {
+	if (!input) {
+		assert(c == EOF);
+		return c;
+	}
 	++input->cnt;
 	return *--input->ptr = c;
 }
--- a/cc1/main.c
+++ b/cc1/main.c
@@ -67,12 +67,10 @@
 
 	ikeywords();
 
-	addinput(*argv);
-	while (yytoken != EOFTOK)
-		extdecl();
-	delinput();
+	if (!addinput(*argv))
+		die("error opening input file '%s'", *argv);
+	for (next(); yytoken != EOFTOK; extdecl())
+		/* nothing */;
 
-	if (fclose(stdout))
-		die("error writing in output:%s", strerror(errno));
 	return 0;
 }
--- a/cc1/symbol.c
+++ b/cc1/symbol.c
@@ -107,10 +107,13 @@
 void
 ikeywords(void)
 {
-	static struct {
+	extern void include(char *), define(char *), undef(char *);
+	extern void ifdef(char *),  ifndef(char *), endif(char *);
+	static struct words {
 		char *str;
 		uint8_t token, value;
-	} *bp, buff[] = {
+		void (*fun)(char *);
+	} ccwords[] = {
 		{"auto", SCLASS, AUTO},
 		{"break", BREAK, BREAK},
 		{"_Bool", TYPE, BOOL},
@@ -146,13 +149,36 @@
 		{"volatile", TQUALIFIER, VOLATILE},
 		{"while", WHILE, WHILE},
 		{NULL, 0, 0},
+	}, cppwords[] = {
+		{"include", 0, 0, include},
+		{"define", 0, 0, define},
+		{"undef", 0, 0, undef},
+		{"ifdef", 0, 0, ifdef},
+		{"ifndef", 0, 0, ifndef},
+		{"endif", 0, 0, endif},
+		{NULL, 0, 0}
 	};
+	static struct wordlist {
+		struct words *words;
+		uint8_t ns;
+	} wordlist [] = {
+		{ccwords, NS_IDEN},
+		{cppwords, NS_CPP},
+		{NULL, 0}
+	};
+	struct wordlist *lp;
+	struct words *bp;
 	Symbol *sym;
 
-	for (bp = buff; bp->str; ++bp) {
-		sym = install(bp->str, NS_IDEN);
-		sym->token = bp->token;
-		sym->u.token = bp->value;
+	for (lp = wordlist; lp->words; ++lp) {
+		for (bp = lp->words; bp->str; ++bp) {
+			sym = install(bp->str, lp->ns);
+			sym->token = bp->token;
+			if (bp->fun)
+				sym->u.fun = bp->fun;
+			else
+				sym->u.token = bp->value;
+		}
 	}
 	globalcnt = 0;
 }