@@ -26,6 +26,12 @@ int break_exit_idx = 0;
2626basic_block_t  * continue_bb [MAX_NESTING ];
2727int  continue_pos_idx  =  0 ;
2828
29+ /* Label utilities */ 
30+ label_t  labels [MAX_LABELS ];
31+ int  label_idx  =  0 ;
32+ basic_block_t  * backpatch_bb [MAX_LABELS ];
33+ int  backpatch_bb_idx  =  0 ;
34+ 
2935/* stack of the operands of 3AC */ 
3036var_t  * operand_stack [MAX_OPERAND_STACK_SIZE ];
3137int  operand_stack_idx  =  0 ;
@@ -40,6 +46,23 @@ void parse_array_init(var_t *var,
4046                      basic_block_t  * * bb ,
4147                      bool  emit_code );
4248
49+ 
50+ label_t  * find_label (char  * name )
51+ {
52+     for  (int  i  =  0 ; i  <  label_idx ; i ++ ) {
53+         if  (!strcmp (name , labels [i ].label_name ))
54+             return  & labels [i ];
55+     }
56+     return  NULL ;
57+ }
58+ 
59+ void  add_label (char  * name , basic_block_t  * bb )
60+ {
61+     label_t  * l  =  & labels [label_idx ++ ];
62+     strncpy (l -> label_name , name , MAX_ID_LEN );
63+     l -> bb  =  bb ;
64+ }
65+ 
4366char  * gen_name_to (char  * buf )
4467{
4568    sprintf (buf , ".t%d" , global_var_idx ++ );
@@ -997,6 +1020,57 @@ basic_block_t *handle_while_statement(block_t *parent, basic_block_t *bb)
9971020    return  else_ ;
9981021}
9991022
1023+ basic_block_t  * handle_goto_statement (block_t  * parent , basic_block_t  * bb )
1024+ {
1025+     /* Since a goto splits the current program into two basic blocks and makes 
1026+      * the subsequent basic block unreachable, this causes problems for later 
1027+      * CFG operations. Therefore, we create a fake if that always executes to 
1028+      * wrap the goto, and connect the unreachable basic block to the else 
1029+      * branch. Finally, return this else block. 
1030+      * 
1031+      * after: 
1032+      * code1; 
1033+      * goto label; 
1034+      * code2; 
1035+      * 
1036+      * before: 
1037+      * code1; 
1038+      * if (1) goto label; 
1039+      * code2; 
1040+      */ 
1041+ 
1042+     char  token [MAX_ID_LEN ];
1043+     if  (!lex_peek (T_identifier , token ))
1044+         error ("Expected identifier after 'goto'" );
1045+ 
1046+     lex_expect (T_identifier );
1047+     lex_expect (T_semicolon );
1048+ 
1049+     basic_block_t  * fake_if  =  bb_create (parent );
1050+     bb_connect (bb , fake_if , NEXT );
1051+     var_t  * val  =  require_var (parent );
1052+     gen_name_to (val -> var_name );
1053+     val -> init_val  =  1 ;
1054+     add_insn (parent , fake_if , OP_load_constant , val , NULL , NULL , 0 , NULL );
1055+     add_insn (parent , fake_if , OP_branch , NULL , val , NULL , 0 , NULL );
1056+ 
1057+     basic_block_t  * then_  =  bb_create (parent );
1058+     basic_block_t  * else_  =  bb_create (parent );
1059+     bb_connect (fake_if , then_ , THEN );
1060+     bb_connect (fake_if , else_ , ELSE );
1061+ 
1062+     add_insn (parent , then_ , OP_jump , NULL , NULL , NULL , 0 , token );
1063+     label_t  * label  =  find_label (token );
1064+     if  (label ) {
1065+         label -> used  =  true;
1066+         bb_connect (then_ , label -> bb , NEXT );
1067+         return  else_ ;
1068+     }
1069+ 
1070+     backpatch_bb [backpatch_bb_idx ++ ] =  then_ ;
1071+     return  else_ ;
1072+ }
1073+ 
10001074basic_block_t  * handle_struct_variable_decl (block_t  * parent ,
10011075                                           basic_block_t  * bb ,
10021076                                           char  * token )
@@ -4169,6 +4243,9 @@ basic_block_t *read_body_statement(block_t *parent, basic_block_t *bb)
41694243        return  do_while_end ;
41704244    }
41714245
4246+     if  (lex_accept (T_goto ))
4247+         return  handle_goto_statement (parent , bb );
4248+ 
41724249    /* empty statement */ 
41734250    if  (lex_accept (T_semicolon ))
41744251        return  bb ;
@@ -4753,6 +4830,21 @@ basic_block_t *read_body_statement(block_t *parent, basic_block_t *bb)
47534830        return  bb ;
47544831    }
47554832
4833+     if  (lex_peek (T_identifier , token )) {
4834+         lex_accept (T_identifier );
4835+         if  (lex_accept (T_colon )) {
4836+             label_t  * l  =  find_label (token );
4837+             if  (l )
4838+                 error ("label redefinition" );
4839+ 
4840+             basic_block_t  * n  =  bb_create (parent );
4841+             bb_connect (bb , n , NEXT );
4842+             add_label (token , n );
4843+             add_insn (parent , n , OP_label , NULL , NULL , NULL , 0 , token );
4844+             return  n ;
4845+         }
4846+     }
4847+ 
47564848    error ("Unrecognized statement token" );
47574849    return  NULL ;
47584850}
@@ -4794,6 +4886,28 @@ void read_func_body(func_t *func)
47944886    basic_block_t  * body  =  read_code_block (func , NULL , NULL , func -> bbs );
47954887    if  (body )
47964888        bb_connect (body , func -> exit , NEXT );
4889+ 
4890+     for  (int  i  =  0 ; i  <  backpatch_bb_idx ; i ++ ) {
4891+         basic_block_t  * bb  =  backpatch_bb [i ];
4892+         insn_t  * g  =  bb -> insn_list .tail ;
4893+         label_t  * label  =  find_label (g -> str );
4894+         if  (!label )
4895+             error ("goto label undefined" );
4896+ 
4897+         label -> used  =  true;
4898+         bb_connect (bb , label -> bb , NEXT );
4899+     }
4900+ 
4901+     for  (int  i  =  0 ; i  <  label_idx ; i ++ ) {
4902+         label_t  * label  =  & labels [i ];
4903+         if  (label -> used )
4904+             continue ;
4905+ 
4906+         printf ("Warning: unused label %s\n" , label -> label_name );
4907+     }
4908+ 
4909+     backpatch_bb_idx  =  0 ;
4910+     label_idx  =  0 ;
47974911}
47984912
47994913/* if first token is type */ 
0 commit comments