shithub: scc

Info  •  Files  •  Log  •  Branches

Add For statement

--- a/stmt.c

+++ b/stmt.c

@@ -22,12 +22,13 @@

 	return install(s, NS_LABEL);

 }

-static Node *

+static void

 stmtexp(void)

 {

-	Node *np = expr();

+	if (accept(';'))

+		return;

+	emitexp(expr());

 	expect(';');

-	return np;

 }

 static Node *

@@ -59,6 +60,34 @@

 }

 static void

+For(Symbol *lswitch)

+{

+	Symbol *begin= label(NULL), *cond = label(NULL), *end = label(NULL);

+	Node *econd = NULL, *einc = NULL;

+

+	expect(FOR);

+	expect('(');

+	stmtexp();

+

+	if (yytoken != ';')

+		econd = expr();

+	expect(';');

+	if (yytoken != ')')

+		einc = expr();

+	expect(')');

+

+	emitjump(cond, NULL);

+	emitbloop();

+	emitlabel(begin);

+	stmt(begin, end, lswitch);

+	if (einc)

+		emitexp(einc);

+	emitlabel(cond);

+	emitjump(begin, econd);

+	emiteloop();

+}

+

+static void

 Return(void)

 {

 	Node *np;

@@ -65,7 +94,8 @@

 	Type *tp = curfun->type->type;

 	expect(RETURN);

-	np = stmtexp();

+	np = expr();

+	expect(';');

 	if (np->type != tp) {

 		if (tp == voidtype)

 			warn(1, "function returning void returns a value");

@@ -101,7 +131,8 @@

 	case '{': compound(lbreak, lcont, lswitch); break;

 	case RETURN: Return(); break;

 	case WHILE: While(lswitch); break;

-	default: emitexp(stmtexp()); break;

+	case FOR: For(lswitch); break;

+	default: stmtexp(); break;

 	}

 }

-- 

⑨