Compiler: move c2_assert to libs/libc/c2_assert.c2i

chqrlie · chqrlie · commit e15d3829daa8 · 2025-07-05T23:08:04.000+02:00
* compute the assert string and function call at parse time
* add c2_assert module in C library interface libs/libc/c2_assert.c2i
* update tests
diff --git a/analyser/module_analyser_stmt.c2 b/analyser/module_analyser_stmt.c2
@@ -374,6 +374,11 @@ fn void Analyser.analyseAssertStmt(Analyser* ma, Stmt* s) {
 
     Expr* inner = a.getInner();
     ma.checker.check(builtins[BuiltinKind.Bool], qt, a.getInner2(), inner.getLoc());
+
+    if (a.getCall()) {
+        qt = ma.analyseExpr(a.getCall2(), true, RHS);
+        if (qt.isInvalid()) return;
+    }
 }
 
 fn void Analyser.analyseReturnStmt(Analyser* ma, Stmt* s) {
diff --git a/ast/assert_stmt.c2 b/ast/assert_stmt.c2
@@ -22,33 +22,42 @@ import string_buffer;
 public type AssertStmt struct @(opaque) {
     Stmt base;
     Expr* inner;
+    Expr* call;
 }
 
 public fn AssertStmt* AssertStmt.create(ast_context.Context* c,
-                                          SrcLoc loc,
-                                          Expr* inner)
+                                        SrcLoc loc,
+                                        Expr* inner,
+                                        Expr* call)
 {
     AssertStmt* s = c.alloc(sizeof(AssertStmt));
     s.base.init(StmtKind.Assert, loc);
     s.inner = inner;
+    s.call = call;
 #if AstStatistics
     Stats.addStmt(StmtKind.Assert, sizeof(AssertStmt));
 #endif
     return s;
 }
 
 fn Stmt* AssertStmt.instantiate(AssertStmt* s, Instantiator* inst) {
-    AssertStmt* s2 = AssertStmt.create(inst.c, s.base.loc, s.inner.instantiate(inst));
+    AssertStmt* s2 = AssertStmt.create(inst.c, s.base.loc, s.inner.instantiate(inst),
+                                       s.call ? s.call.instantiate(inst) : nil);
     return (Stmt*)s2;
 }
 
 public fn Expr* AssertStmt.getInner(const AssertStmt* s) { return s.inner; }
 
 public fn Expr** AssertStmt.getInner2(AssertStmt* s) { return &s.inner; }
 
+public fn Expr* AssertStmt.getCall(const AssertStmt* s) { return s.call; }
+
+public fn Expr** AssertStmt.getCall2(AssertStmt* s) { return &s.call; }
+
 fn void AssertStmt.print(const AssertStmt* s, string_buffer.Buf* out, u32 indent) {
     s.base.printKind(out, indent);
     out.newline();
     s.inner.print(out, indent + 1);
+    if (s.call) s.call.print(out, indent + 1);
 }
 
diff --git a/ast/utils.c2 b/ast/utils.c2
@@ -39,7 +39,7 @@ static_assert(80, sizeof(FunctionDecl));
 static_assert(8, sizeof(Stmt));
 static_assert(12, sizeof(GotoStmt));
 static_assert(24, sizeof(LabelStmt));
-static_assert(16, sizeof(AssertStmt));
+static_assert(24, sizeof(AssertStmt));
 static_assert(8, sizeof(DeclStmt));
 static_assert(16, sizeof(SwitchStmt));
 static_assert(24, sizeof(AsmStmt));
diff --git a/compiler/compiler.c2 b/compiler/compiler.c2
@@ -300,7 +300,8 @@ fn void Compiler.build(Compiler* c,
                                 c.astPool,
                                 c.builder,
                                 &c.kwinfo,
-                                target.getFeatures());
+                                target.getFeatures(),
+                                c.target.hasAsserts());
 
     ast.initialize(c.context, c.astPool, c.targetInfo.intWidth / 8, color.useColor());
 
diff --git a/generator/ast_visitor.c2 b/generator/ast_visitor.c2
@@ -213,6 +213,8 @@ fn void Visitor.handleStmt(Visitor* v, Stmt* s) {
     case Assert:
         AssertStmt* a = cast<AssertStmt*>(s);
         v.handleExpr(a.getInner());
+        Expr* call = a.getCall();
+        if (call) v.handleExpr(call);
         break;
     }
 }
diff --git a/generator/c/c2i_generator_stmt.c2 b/generator/c/c2i_generator_stmt.c2
@@ -207,23 +207,10 @@ fn void Generator.emitStmt(Generator* gen, ast.Stmt* s, u32 indent, bool newline
         //gen.emitAsmStmt(cast<AsmStmt*>(s), indent);
         break;
     case Assert:
-#if 0
-        // TODO: generate actual assert statement?
         AssertStmt* a = cast<AssertStmt*>(s);
-        source_mgr.Location loc = gen.sm.getLocation(a.getLoc());
-        const char* funcname = gen.cur_function.asDecl().getFullName();
-
-        out.add("(");
-        Expr* inner = a.getInner();
-        gen.emitExpr(out, inner);
-        out.print(") || c2_assert(\"%s\", %d, \"%s\", \"", loc.filename, loc.line, funcname);
-        // encode expression as a string
-        string_buffer.Buf* str = string_buffer.create(128, false, 0);
-        inner.printLiteral(str);
-        out.encodeBytes(str.data(), str.size(), '"');
-        str.free();
-        out.add("\");\n");
-#endif
+        out.add("assert(");
+        gen.emitExpr(a.getInner());
+        out.add(");\n");
         break;
     }
 }
diff --git a/generator/c/c_generator.c2 b/generator/c/c_generator.c2
@@ -1376,17 +1376,6 @@ const char[] C_types =
     #define ARRAY_SIZE(x) (sizeof(x)/sizeof((x)[0]))
     ```;
 
-const char[] C2_assert =
-    ```c
-    int dprintf(int fd, const char *format, ...);
-    void abort(void);
-    static int c2_assert(const char* filename, int line, const char* funcname, const char* condstr) {
-        dprintf(2, "%s:%d: function %s: Assertion failed: %s\n", filename, line, funcname, condstr);
-        abort();
-        return 0;
-    }
-    ```;
-
 const char[] C2_strswitch =
     ```c
     static int c2_strswitch(const char* s1, const char* s2) {
@@ -1426,10 +1415,6 @@ fn void Generator.emit_external_header(Generator* gen, bool enable_asserts, cons
 #endif
     out.add("#define to_container(type, member, ptr) ((type*)((char*)(ptr) - offsetof(type, member)))\n\n");
 
-    if (enable_asserts) {
-        out.add(C2_assert);
-    }
-
     // for switch(str):
     // String format: strings are prefixed by a length byte and concatenated as a single character array.
     // This minimizes memory use, cache misses and avoids extra symbols.
diff --git a/generator/c/c_generator_stmt.c2 b/generator/c/c_generator_stmt.c2
@@ -16,7 +16,6 @@
 module c_generator;
 
 import ast local;
-import source_mgr;
 import string_buffer;
 
 fn void Generator.emitVarDecl(Generator* gen, VarDecl* vd, string_buffer.Buf* out, bool emit_init, bool first) {
@@ -242,19 +241,14 @@ fn void Generator.emitStmt(Generator* gen, Stmt* s, u32 indent, bool newline) {
         if (!gen.enable_asserts) break;
 
         AssertStmt* a = cast<AssertStmt*>(s);
-        source_mgr.Location loc = gen.sm.locate(s.getLoc());
-        const char* funcname = gen.cur_function.asDecl().getFullName();
-
-        out.add("(");
-        Expr* inner = a.getInner();
-        gen.emitExpr(out, inner);
-        out.print(") || c2_assert(\"%s\", %d, \"%s\", \"", loc.filename, loc.line, funcname);
-        // encode expression as a string
-        string_buffer.Buf* str = string_buffer.create(128, false, 0);
-        inner.printLiteral(str);
-        out.encodeBytes(str.data(), str.size(), '"');
-        str.free();
-        out.add("\");\n");
+        Expr* call = a.getCall();
+        if (call) {
+            out.add("(");
+            gen.emitExpr(out, a.getInner());
+            out.add(") || ");
+            gen.emitExpr(out, call);
+            out.add(";\n");
+        }
         break;
     }
 }
diff --git a/libs/libc/c2_assert.c2i b/libs/libc/c2_assert.c2i
@@ -0,0 +1,14 @@
+module c2_assert;
+
+import stdio local;
+import stdlib local;
+
+public fn i32 c2_assert_fail(const char* filename @(auto_file),
+                             u32 line @(auto_line),
+                             const char* funcname @(auto_func),
+                             const char* condstr)
+{
+    dprintf(2, "%s:%d: function %s: Assertion failed: %s\n", filename, line, funcname, condstr);
+    abort();
+    return 0;
+}
diff --git a/libs/libc/manifest.yaml b/libs/libc/manifest.yaml
@@ -40,4 +40,4 @@ modules:
     - sys_utsname
     - uio
     - unistd
-
+    - c2_assert
diff --git a/parser/ast_builder.c2 b/parser/ast_builder.c2
@@ -873,8 +873,8 @@ public fn SwitchCase* Builder.actOnCase(Builder* b,
                              conds, num_conds, stmts, num_stmts);
 }
 
-public fn Stmt* Builder.actOnAssertStmt(Builder* b, SrcLoc loc, Expr* inner) {
-    return (Stmt*)AssertStmt.create(b.context, loc, inner);
+public fn Stmt* Builder.actOnAssertStmt(Builder* b, SrcLoc loc, Expr* inner, Expr* call) {
+    return (Stmt*)AssertStmt.create(b.context, loc, inner, call);
 }
 
 public fn Stmt* Builder.actOnBreakStmt(Builder* b, SrcLoc loc) {
diff --git a/parser/c2_parser.c2 b/parser/c2_parser.c2
@@ -53,6 +53,7 @@ public type Parser struct @(opaque) {
     const string_list.List* features;
     const keywords.Info* kwinfo;
     bool is_interface;
+    bool has_asserts;
     u32 va_list_idx;
     u32 varargs_idx;
     u32 stdarg_idx;
@@ -66,7 +67,8 @@ public fn Parser* create(SourceMgr* sm,
                            string_pool.Pool* pool,
                            ast_builder.Builder* builder,
                            const keywords.Info* kwinfo,
-                           const string_list.List* features)
+                           const string_list.List* features,
+                           bool has_asserts)
 {
     Parser* p = calloc(1, sizeof(Parser));
     p.sm = sm;
@@ -75,6 +77,7 @@ public fn Parser* create(SourceMgr* sm,
     p.builder = builder;
     p.features = features;
     p.kwinfo = kwinfo;
+    p.has_asserts = has_asserts;
     p.va_list_idx = pool.addStr("va_list", true);
     p.varargs_idx = pool.addStr("varargs", true);
     p.stdarg_idx = pool.addStr("stdarg", true);
diff --git a/parser/c2_parser_stmt.c2 b/parser/c2_parser_stmt.c2
@@ -21,6 +21,7 @@ import constants local;
 import token local;
 import src_loc local;
 import string_list;
+import string_buffer;
 
 fn Stmt* Parser.parseStmt(Parser* p) {
     // TODO use Jump Table (combined one for multiple purposes?)
@@ -390,10 +391,36 @@ fn Stmt* Parser.parseAssertStmt(Parser* p) {
     SrcLoc loc = p.tok.loc;
     p.consumeToken();
     p.expectAndConsume(Kind.LParen);
+    SrcLoc eloc = p.tok.loc;
     Expr* inner = p.parseExpr();
+    u32 elen = p.prev_loc - eloc;
     p.expectAndConsume(Kind.RParen);
+    u32 loc_end = p.prev_loc;
     p.expectAndConsume(Kind.Semicolon);
-    return p.builder.actOnAssertStmt(loc, inner);
+
+    Expr* call = nil;
+    if (p.has_asserts) {
+        // add c2_assert.c2_assert_fail func designator
+        Ref[2] ref;
+        ref[0].loc = loc;
+        ref[0].name_idx = p.pool.addStr("c2_assert", true);
+        ref[1].loc = loc;
+        ref[1].name_idx = p.pool.addStr("c2_assert_fail", true);
+        p.addImplicitImport(ref[0].name_idx, false);
+        Expr* func = p.builder.actOnMemberExpr(nil, ref, 2);
+
+        // encode expression as a string
+        string_buffer.Buf* buf = p.tokenizer.buf;
+        buf.clear();
+        inner.printLiteral(buf);
+        u32 msg_len = buf.size();
+        u32 msg_idx = p.pool.add(buf.data(), msg_len, true);
+        Expr* arg = p.builder.actOnStringLiteral(eloc, elen, msg_idx, msg_len);
+
+        // create call expression `c2_assert.c2_assert_fail("expression")
+        call = p.builder.actOnCallExpr(loc, loc_end, func, &arg, 1);
+    }
+    return p.builder.actOnAssertStmt(loc, inner, call);
 }
 
 fn Stmt* Parser.parseBreakStmt(Parser* p) {
diff --git a/test/c_generator/stmts/assert.c2t b/test/c_generator/stmts/assert.c2t
@@ -19,8 +19,8 @@ int32_t main(void);
 
 int32_t main(void)
 {
-   (test_a) || c2_assert("file1.c2", 7, "test.main", "a");
-   (test_p) || c2_assert("file1.c2", 8, "test.main", "p");
+   (test_a) || c2_assert_fail("file1.c2", 7, "test.main", "a");
+   (test_p) || c2_assert_fail("file1.c2", 8, "test.main", "p");
    return 0;
 }
 

Original file line number	Diff line number	Diff line change
`@@ -213,6 +213,8 @@ fn void Visitor.handleStmt(Visitor* v, Stmt* s) {`
`213`	`213`	`case Assert:`
`214`	`214`	`AssertStmt* a = cast<AssertStmt*>(s);`
`215`	`215`	`v.handleExpr(a.getInner());`
	`216`	`+ Expr* call = a.getCall();`
	`217`	`+ if (call) v.handleExpr(call);`
`216`	`218`	`break;`
`217`	`219`	`}`
`218`	`220`	`}`