1 files changed, 15 insertions, 0 deletions
diff --git a/tccgen.c b/tccgen.c
index b4f97f4..fa03daf 100644
--- a/tccgen.c
+++ b/tccgen.c
@@ -5770,6 +5770,21 @@ static void gen_function(Sym *sym)
     /* push a dummy symbol to enable local sym storage */
     sym_push2(&local_stack, SYM_FIELD, 0, 0);
     gfunc_prolog(&sym->type);
+#ifdef CONFIG_TCC_BCHECK
+    if (tcc_state->do_bound_check
+        && !strcmp(get_tok_str(sym->v, NULL), "main")) {
+        int i;
+
+        sym = local_stack;
+        for (i = 0, sym = local_stack; i < 2; i++, sym = sym->prev) {
+            if (sym->v & SYM_FIELD || sym->prev->v & SYM_FIELD)
+                break;
+            vpush_global_sym(&func_old_type, TOK___bound_main_arg);
+            vset(&sym->type, sym->r, sym->c);
+            gfunc_call(1);
+        }
+    }
+#endif
     rsym = 0;
     block(NULL, NULL, NULL, NULL, 0, 0);
     gsym(rsym);