C言語の自作コンパイラを作成したいと思っています。以下のlex, yacc, cのコードを用いてwhile文を表現したいのですが、うまくいきません。test.c の T_WHILEの部分がおかしいのかなと思うのですが、どのように変更したらよいか教えていただきたいです。
test.c
#include <stdio.h> struct stab { int val; char name[20]; } stab[100]; int stabuse = 0; struct node { int type, left, right; } ntab[400]; int ntabuse = 1; #define T_STLIST 1 #define T_ASSIGN 2 #define T_READ 3 #define T_PRINT 4 #define T_ADD 5 #define T_SUB 6 #define T_MUL 7 #define T_DIV 8 #define T_REM 9 #define T_NUM 10 #define T_VAR 11 #define T_WHILE 12 #define T_IF 13 #define T_LT 14 #define T_GT 15 int lookup(char*); int node(int, int, int); void dotree(int); extern char *yytext; #include "y.tab.c" #include "lex.yy.c" int main() { yyparse(); return 0; } int lookup(char *s) { int i; for(i = 0; i < stabuse; ++i) if(strcmp(stab[i].name, s) == 0) return i; if(stabuse >= 99) { printf("table overflow.\n"); exit(1); } strcpy(stab[stabuse].name, s); return stabuse++; } int node(int t, int l, int r) { int i = ntabuse++; ntab[i].type = t; ntab[i].left = l; ntab[i].right = r; return i; } void dotree(int i) { int stk; printf(" .section .rodata\n"); printf(".Lprompt: .string\"> \"\n"); /* プロンプト */ printf(".Lread: .string\"%%ld\"\n"); /* 読み取り用書式 */ printf(".Lprint: .string\"%%ld\\n\"\n"); /* 書き出し用書式 */ printf(" .text\n"); printf(".global main\n"); printf("main:\n"); printf(" pushq %%rbp\n"); printf(" movq %%rsp,%%rbp\n"); stk = (8*stabuse + 15) / 16; /* 変数の個数*8 で 16 の倍数へ切り上げ */ stk *= 16; printf(" subq $%d,%%rsp\n", stk); emittree(i); printf(" leave\n"); printf(" ret\n"); } void emittree(int i) { static int labelno = 1; int l; switch(ntab[i].type) { case T_STLIST: if(ntab[i].left) emittree(ntab[i].left); emittree(ntab[i].right); break; case T_READ: printf(" movq $.Lprompt,%%rdi\n"); printf(" movq $0,%%rax\n"); /* 浮動小数点レジスタを使わない */ printf(" call printf\n"); printf(" leaq %d(%%rbp),%%rsi\n", -(ntab[i].left+1)*8); printf(" movq $.Lread,%%rdi\n"); printf(" movq $0,%%rax\n"); /* 浮動小数点レジスタを使わない */ printf(" call scanf\n"); break; case T_PRINT: emittree(ntab[i].left); printf(" popq %%rsi\n"); printf(" movq $.Lprint,%%rdi\n"); printf(" movq $0,%%rax\n"); /* 浮動小数点レジスタを使わない */ printf(" call printf\n"); break; case T_NUM: printf(" pushq $%d\n", ntab[i].left); break; case T_VAR: printf(" pushq %d(%%rbp)\n", -(ntab[i].left+1)*8); break; default: printf("NotImplemented: %d\n", ntab[i].type); break; case T_ASSIGN: emittree(ntab[i].right); printf(" popq %d(%%rbp)\n", -(ntab[i].left+1)*8); break; case T_ADD: emittree(ntab[i].left); emittree(ntab[i].right); printf(" popq %%rdx\n"); printf(" popq %%rax\n"); printf(" addq %%rdx,%%rax\n"); printf(" pushq %%rax\n"); break; case T_SUB: emittree(ntab[i].left); emittree(ntab[i].right); printf(" popq %%rdx\n"); printf(" popq %%rax\n"); printf(" subq %%rdx, %%rax\n"); printf(" pushq %%rax\n"); break; case T_MUL: emittree(ntab[i].left); emittree(ntab[i].right); printf(" popq %%rdx\n"); printf(" popq %%rax\n"); printf(" imulq %%rdx\n"); printf(" pushq %%rax\n"); break; case T_DIV: emittree(ntab[i].left); emittree(ntab[i].right); printf(" popq %%rsp\n"); printf(" popq %%rax\n"); printf(" movq $0,%%rdx\n"); printf(" idivq %%rsp\n"); printf(" pushq %%rax\n"); break; case T_REM: emittree(ntab[i].left); emittree(ntab[i].right); printf(" popq %%rsp\n"); printf(" popq %%rax\n"); printf(" movq $0,%%rdx\n"); printf(" idivq %%rsp\n"); printf(" pushq %%rdx\n"); case T_LT: emittree(ntab[i].left); emittree(ntab[i].right); printf(" popq %%rcx\n"); printf(" popq %%rax\n"); printf(" cmp %%rcx,%%rax\n"); printf(" jge "); break; case T_GT: emittree(ntab[i].left); emittree(ntab[i].right); printf(" popq %%rcx\n"); printf(" popq %%rax\n"); printf(" cmp %%rcx,%%rax\n"); printf(" jle "); break; case T_IF: l = labelno++; emittree(ntab[i].left); printf(".L%d\n", l); emittree(ntab[i].right); printf(".L%d:\n", l); break; case T_WHILE: // この部分 l = labelno++; // ラベル番号更新 printf(".L%d:\n", l); // ラベル l 生成 emittree(ntab[i].left); // 分岐条件判定 printf(".L%d\n", l+1); // 不成立なら l+1に移動 emittree(ntab[i].right); // 成立していたら、中身を実行 printf(".L%d\n", l); // 前のラベル l に戻る printf(".L%d:\n", l+1); // ラベル l+1 生成 break; } }
test.yacc
%token NUM; %token IDENT; %token READ; %token PRINT; %token WHILE; %token IF; %left '+' '-'; %left '*' '/'; %% prog : IDENT '{' stlist'}' { dotree($3); return 0; } ; stlist : { $$ = 0; } | stlist stat { $$ = node(T_STLIST, $1, $2); } ; stat : var '=' expr ';' { $$ = node(T_ASSIGN, $1, $3); } | READ var ';' { $$ = node(T_READ, $2, 0); } | PRINT expr ';' { $$ = node(T_PRINT, $2, 0); } | WHILE '(' cond ')' stat { $$ = node(T_WHILE, $3, $5); } | IF '(' cond ')' stat { $$ = node(T_IF, $3, $5); } | '{' stlist '}' { $$ = $2; } ; cond : expr '<' expr { $$ = node(T_LT, $1, $3); } | expr '>' expr { $$ = node(T_GT, $1, $3); } ; expr : term { $$ = $1; } | expr '+' term { $$ = node(T_ADD, $1, $3); } | expr '-' term { $$ = node(T_SUB, $1, $3); } ; term: prim{ $$ = $1; } | term '*' prim { $$ = node(T_MUL, $1, $3); } | term '/' prim { $$ = node(T_DIV, $1, $3); } | term '%' prim { $$ = node(T_REM, $1, $3); } ; prim : NUM { $$ = node(T_NUM, atoi(yytext), 0); } | var { $$ = node(T_VAR, $1, 0); } | '(' expr ')' { $$ = $2; } ; var : IDENT { $$ = lookup(yytext); } ;
test.lex
alpha [a-zA-Z] digit [0-9] white [\n\t ] %% while { return WHILE; } if { return IF; } read { return READ; } print { return PRINT; } {alpha}({alpha}|{digit})* { return IDENT; } {digit}+ { return NUM; } [-+()=;{}<>*/%] { return yytext[0]; } {white} { ; }
まだ回答がついていません
会員登録して回答してみよう