Actually compiling return codes!

Wilfred · Jul 12, 2014 · 65f3a01 · 65f3a01
1 parent e54125c
commit 65f3a01
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 3 deletions.
diff --git a/minimal_c.l b/minimal_c.l
@@ -13,7 +13,12 @@
 "("           { return '('; }
 ")"           { return ')'; }
 ";"           { return ';'; }
-[0-9]+        { return NUMBER; }
+[0-9]+        {
+                /* TODO: check numbers are in the legal range, and don't start with 0. */
+                printf("yytext: %s\n", yytext);
+                printf("yylval: %d\n", atoi(yytext));
+                yylval=atoi(yytext); return NUMBER;
+              }
 "return"      { return RETURN; }
 
 "int"         { return TYPE; }

diff --git a/minimal_c.y b/minimal_c.y
@@ -14,6 +14,9 @@ int yywrap()
 
 extern FILE *yyin;
 
+// Shameful hack. We should build a proper AST and traverse it.
+static int return_code;
+
 void write_skeleton() {
     FILE *out = fopen("out.s", "wb");
 
@@ -22,8 +25,9 @@ void write_skeleton() {
     fprintf(out, "    .global _start\n\n");
     fprintf(out, "_start:\n");
 
-    // Exit code of zero.
-    fprintf(out, "    movl    $0, %%ebx\n");
+    // Exit code as specified.
+    // TODO: convert to hex properly.
+    fprintf(out, "    movl    $%d, %%ebx\n", return_code);
 
     fprintf(out, "    movl    $1, %%eax\n");
     fprintf(out, "    int     $0x80\n");
@@ -73,5 +77,6 @@ function:
 
 expression:
 	RETURN NUMBER ';'
+        { return_code = $2; }
         ;
 
diff --git a/test_programs/two.c b/test_programs/two.c
@@ -0,0 +1,3 @@
+int main() {
+    return 2;
+}