Parse break, continue, return as builtins

1 files changed, 48 insertions, 9 deletions

diff --git a/parser.c b/parser.c
index 6309e65..1042edb 100644
--- a/parser.c
+++ b/parser.c
@@ -203,6 +203,12 @@ static void printtoken(FILE *stream,Token tok,const char *msg){
 #endif
 
 
+static const char *builtin_keywords[]={
+	"break", "continue", "return"
+};
+static const int nbuiltin_keywords=sizeof(builtin_keywords)/sizeof(builtin_keywords[0]);
+
+
 static AST* parseexpr(const char *source,int *reslen,int minprec,int maxprec);
 
 static AST* parseterm(const char *source,int *reslen){
@@ -380,10 +386,25 @@ static AST* parseterm(const char *source,int *reslen){
 				node=malloc(sizeof(AST));
 				if(!node)outofmem();
 				node->type=AST_CALL;
-				node->c.func=malloc(tok.len+1);
-				if(!node->c.func)outofmem();
-				memcpy(node->c.func,tok.str,tok.len);
-				node->c.func[tok.len]='\0';
+				int i;
+				for(i=0;i<nbuiltin_keywords;i++){
+					if(tok.len==(int)strlen(builtin_keywords[i])&&memcmp(builtin_keywords[i],tok.str,tok.len)==0){
+						break;
+					}
+				}
+				if(i<nbuiltin_keywords){
+					int prefixlen=strlen("__builtin_");
+					node->c.func=malloc(tok.len+prefixlen+1);
+					if(!node->c.func)outofmem();
+					memcpy(node->c.func,"__builtin_",prefixlen);
+					memcpy(node->c.func+prefixlen,tok.str,tok.len);
+					node->c.func[prefixlen+tok.len]='\0';
+				} else {
+					node->c.func=malloc(tok.len+1);
+					if(!node->c.func)outofmem();
+					memcpy(node->c.func,tok.str,tok.len);
+					node->c.func[tok.len]='\0';
+				}
 				node->c.nargs=nargs;
 				node->c.args=args;
 #undef FREEARGSRETNULL
@@ -391,11 +412,29 @@ static AST* parseterm(const char *source,int *reslen){
 				source=tempsource;
 				node=malloc(sizeof(AST));
 				if(!node)outofmem();
-				node->type=AST_VAR;
-				node->v.name=malloc(tok.len+1);
-				if(!node->v.name)outofmem();
-				memcpy(node->v.name,tok.str,tok.len);
-				node->v.name[tok.len]='\0';
+				int i;
+				for(i=0;i<nbuiltin_keywords;i++){
+					if(tok.len==(int)strlen(builtin_keywords[i])&&memcmp(builtin_keywords[i],tok.str,tok.len)==0){
+						break;
+					}
+				}
+				if(i<nbuiltin_keywords){
+					node->type=AST_CALL;
+					int prefixlen=strlen("__builtin_");
+					node->c.func=malloc(tok.len+prefixlen+1);
+					if(!node->c.func)outofmem();
+					memcpy(node->c.func,"__builtin_",prefixlen);
+					memcpy(node->c.func+prefixlen,tok.str,tok.len);
+					node->c.func[prefixlen+tok.len]='\0';
+					node->c.nargs=0;
+					node->c.args=malloc(1);
+				} else {
+					node->type=AST_VAR;
+					node->v.name=malloc(tok.len+1);
+					if(!node->v.name)outofmem();
+					memcpy(node->v.name,tok.str,tok.len);
+					node->v.name[tok.len]='\0';
+				}
 			}
 			break;
 		}