Implement goto statement support

Lexer/Parser: add T_goto; parse goto identifier; and identifier: labels. Function-scoped labels with forward references via backpatching. Duplicate labels error; undefined labels error; unused labels warning.
IR/CFG: introduce OP_label and OP_jump; update human-readable and graph dumps.
Lower goto via an if (1) wrapper to keep CFG well-formed (then: jmp label; else: fallthrough).
Label tracking: label_t table and backpatch list (MAX_LABELS=256), reset per function.
SSA/semantics: add dominance-based check to warn when goto crosses variable initialization.
Notes: no computed goto (GNU extension).

Author: tobiichi3227

src/defs.h

@@ -20,6 +20,7 @@
 #define MAX_LOCALS 1600
 #define MAX_FIELDS 64
 #define MAX_TYPES 256
+#define MAX_LABELS 256
 #define MAX_IR_INSTR 80000
 #define MAX_BB_PRED 128
 #define MAX_BB_DOM_SUCC 64
@@ -179,6 +180,7 @@ typedef enum {
     T_break,
     T_default,
     T_continue,
+    T_goto,
     T_const, /* const qualifier */
     /* C pre-processor directives */
     T_cppd_include,
@@ -270,6 +272,7 @@ typedef enum {
     OP_branch,   /* conditional jump */
     OP_jump,     /* unconditional jump */
     OP_func_ret, /* returned value */
+    OP_label, /* for goto label */
 
     /* function pointer */
     OP_address_of_func, /* resolve function entry */
@@ -567,6 +570,13 @@ struct ref_block {
  * type, parameters) with SSA-related information (e.g., basic blocks, control
  * flow) to support parsing, analysis, optimization, and code generation.
  */
+
+typedef struct {
+    char label_name[MAX_ID_LEN];
+    basic_block_t *bb;
+    bool used;
+} label_t;
+
 struct func {
     /* Syntatic info */
     var_t return_def;

src/globals.c

@@ -1481,6 +1481,14 @@ void dump_bb_insn(func_t *func, basic_block_t *bb, bool *at_func_start)
             printf("br %%%s, %s, %s", rs1->var_name, bb->then_->bb_label_name,
                    bb->else_->bb_label_name);
             break;
+        case OP_jump:
+            print_indent(1);
+            printf("jmp %s", bb->next->bb_label_name);
+            break;
+        case OP_label:
+            print_indent(0);
+            printf("%s:", insn->str);
+            break;
         case OP_push:
             print_indent(1);
             printf("push %%%s", rs1->var_name);

src/lexer.c

@@ -85,6 +85,7 @@ void lex_init_keywords()
         {"break", T_break},
         {"default", T_default},
         {"continue", T_continue},
+        {"goto", T_goto},
         {"union", T_union},
         {"const", T_const},
     };
@@ -786,6 +787,8 @@ token_t lex_token_impl(bool aliasing)
                     keyword = T_enum;
             } else if (!memcmp(token_str, "case", 4))
                 keyword = T_case;
+            else if (!memcmp(token_str, "goto", 4))
+                keyword = T_goto;
             break;
 
         case 5: /* 5-letter keywords: while, break, union, const */

src/parser.c

@@ -26,6 +26,12 @@ int break_exit_idx = 0;
 basic_block_t *continue_bb[MAX_NESTING];
 int continue_pos_idx = 0;
 
+/* Label utilities */
+label_t labels[MAX_LABELS];
+int label_idx = 0;
+basic_block_t *backpatch_bb[MAX_LABELS];
+int backpatch_bb_idx = 0;
+
 /* stack of the operands of 3AC */
 var_t *operand_stack[MAX_OPERAND_STACK_SIZE];
 int operand_stack_idx = 0;
@@ -40,12 +46,30 @@ void parse_array_init(var_t *var,
                       basic_block_t **bb,
                       bool emit_code);
 
+
+label_t *find_label(char *name)
+{
+    for (int i = 0; i < label_idx; i++) {
+        if (!strcmp(name, labels[i].label_name))
+            return &labels[i];
+    }
+    return NULL;
+}
+
+void add_label(char *name, basic_block_t *bb)
+{
+    label_t *l = &labels[label_idx++];
+    strncpy(l->label_name, name, MAX_ID_LEN);
+    l->bb = bb;
+}
+
 char *gen_name_to(char *buf)
 {
     sprintf(buf, ".t%d", global_var_idx++);
     return buf;
 }
 
+
 var_t *require_var(block_t *blk)
 {
     var_list_t *var_list = &blk->locals;
@@ -997,6 +1021,60 @@ basic_block_t *handle_while_statement(block_t *parent, basic_block_t *bb)
     return else_;
 }
 
+
+
+basic_block_t *handle_goto_statement(block_t *parent, basic_block_t *bb)
+{
+    /* Since a goto splits the current program into two basic blocks and makes
+     * the subsequent basic block unreachable, this causes problems for later
+     * CFG operations. Therefore, we create a fake if that always executes to
+     * wrap the goto, and connect the unreachable basic block to the else
+     * branch. Finally, return this else block.
+     *
+     * after:
+     * code1;
+     * goto label;
+     * code2;
+     *
+     * before:
+     * code1;
+     * if (1) goto label;
+     * code2;
+     */
+
+    char token[MAX_ID_LEN];
+    if (lex_peek(T_identifier, token)) {
+        lex_expect(T_identifier);
+        lex_expect(T_semicolon);
+
+        basic_block_t *fake_if = bb_create(parent);
+        bb_connect(bb, fake_if, NEXT);
+        var_t *val = require_var(parent);
+        gen_name_to(val->var_name);
+        val->init_val = 1;
+        add_insn(parent, fake_if, OP_load_constant, val, NULL, NULL, 0, NULL);
+        add_insn(parent, fake_if, OP_branch, NULL, val, NULL, 0, NULL);
+
+        basic_block_t *then_ = bb_create(parent);
+        basic_block_t *else_ = bb_create(parent);
+        bb_connect(fake_if, then_, THEN);
+        bb_connect(fake_if, else_, ELSE);
+
+        label_t *label = find_label(token);
+        add_insn(parent, then_, OP_jump, NULL, NULL, NULL, 0, token);
+        if (label) {
+            label->used = true;
+            bb_connect(then_, label->bb, NEXT);
+        } else {
+            backpatch_bb[backpatch_bb_idx++] = then_;
+        }
+        return else_;
+    } else {
+        error("Expected identifier after 'goto'");
+        return NULL;
+    }
+}
+
 basic_block_t *handle_struct_variable_decl(block_t *parent,
                                            basic_block_t *bb,
                                            char *token)
@@ -4169,6 +4247,9 @@ basic_block_t *read_body_statement(block_t *parent, basic_block_t *bb)
         return do_while_end;
     }
 
+    if (lex_accept(T_goto))
+        return handle_goto_statement(parent, bb);
+
     /* empty statement */
     if (lex_accept(T_semicolon))
         return bb;
@@ -4753,6 +4834,22 @@ basic_block_t *read_body_statement(block_t *parent, basic_block_t *bb)
         return bb;
     }
 
+    if (lex_peek(T_identifier, token)) {
+        lex_accept(T_identifier);
+        if (lex_accept(T_colon)) {
+            label_t *l = find_label(token);
+            if (l) {
+                error("label redefinition");
+                return NULL;
+            }
+            basic_block_t *n = bb_create(parent);
+            bb_connect(bb, n, NEXT);
+            add_label(token, n);
+            add_insn(parent, n, OP_label, NULL, NULL, NULL, 0, token);
+            return n;
+        }
+    }
+
     error("Unrecognized statement token");
     return NULL;
 }
@@ -4794,6 +4891,27 @@ void read_func_body(func_t *func)
     basic_block_t *body = read_code_block(func, NULL, NULL, func->bbs);
     if (body)
         bb_connect(body, func->exit, NEXT);
+
+    for (int i = 0; i < backpatch_bb_idx; i++) {
+        basic_block_t *bb = backpatch_bb[i];
+        insn_t *g = bb->insn_list.tail;
+        label_t *label = find_label(g->str);
+        if (!label) {
+            error("goto label undefined");
+        } else {
+            label->used = true;
+            bb_connect(bb, label->bb, NEXT);
+        }
+    }
+
+    for (int i = 0; i < label_idx; i++) {
+        label_t *label = &labels[i];
+        if (!label->used)
+            printf("Warning: unused label %s\n", label->label_name);
+    }
+
+    backpatch_bb_idx = 0;
+    label_idx = 0;
 }
 
 /* if first token is type */

src/ssa.c

@@ -929,6 +929,65 @@ void unwind_phi(void)
     }
 }
 
+bool is_dominate(basic_block_t *pred, basic_block_t *succ)
+{
+    int i;
+    bool found = false;
+    for (i = 0; i < MAX_BB_DOM_SUCC; i++) {
+        if (!pred->dom_next[i])
+            break;
+        if (pred->dom_next[i] == succ) {
+            found = true;
+            break;
+        }
+        found |= is_dominate(pred->dom_next[i], succ);
+    }
+
+    return found;
+}
+
+void bb_check_var_cross_init(func_t *func, basic_block_t *bb)
+{
+    UNUSED(func);
+
+    /*
+     * For any variable, the basic block that defines it must dominate all the
+     * basic blocks where it is used; otherwise, it is an invalid cross-block
+     * initialization.
+     */
+
+    for (insn_t *insn = bb->insn_list.head; insn; insn = insn->next) {
+        if (insn->opcode == OP_allocat) {
+            var_t *var = insn->rd;
+            ref_block_t *ref;
+            for (ref = var->ref_block_list.head; ref; ref = ref->next) {
+                if (ref->bb == bb)
+                    continue;
+                if (!is_dominate(bb, ref->bb))
+                    printf("Warning: Variable '%s' cross-initialized\n",
+                           var->var_name);
+            }
+        }
+    }
+}
+
+void check_var_cross_initialization()
+{
+    bb_traversal_args_t *args = arena_alloc_traversal_args();
+    for (func_t *func = FUNC_LIST.head; func; func = func->next) {
+        /* Skip function declarations without bodies */
+        if (!func->bbs)
+            continue;
+
+        args->func = func;
+        args->bb = func->bbs;
+
+        func->visited++;
+        args->postorder_cb = bb_check_var_cross_init;
+        bb_forward_traversal(args);
+    }
+}
+
 #ifdef __SHECC__
 #else
 void bb_dump_connection(FILE *fd,
@@ -1112,6 +1171,12 @@ void bb_dump(FILE *fd, func_t *func, basic_block_t *bb)
                 sprintf(str, "<BRANCH %s<SUB>%d</SUB>>", insn->rs1->var_name,
                         insn->rs1->subscript);
                 break;
+            case OP_jump:
+                sprintf(str, "<JUMP>");
+                break;
+            case OP_label:
+                sprintf(str, "<LABEL>");
+                break;
             case OP_push:
                 sprintf(str, "<PUSH %s<SUB>%d</SUB>>", insn->rs1->var_name,
                         insn->rs1->subscript);
@@ -1281,6 +1346,9 @@ void ssa_build(void)
     build_df();
 
     solve_globals();
+
+    check_var_cross_initialization();
+
     solve_phi_insertion();
     solve_phi_params();