From ce293c982e1b6d9db775e02fefb8905caa4ee748 Mon Sep 17 00:00:00 2001 From: kcbanner Date: Thu, 2 Nov 2023 00:02:37 -0400 Subject: cbe: avoid collisions with builtins and intrinsics Changes: - Add `isMangledIdent` to determine if `fmtIdent` would make any edits to the identifier - Any function that has a mangled identifier is referred to using the mangled identifer within the current file, but if it is exported the first export will be with the non-mangled name. - Add `zig_import` to import a symbol under a different name - Add a level of indirection to float function names. Now, they are referred to as `zig_float_fn__`. The definitions in zig.h are wrapped with `zig_import` to import the symbol under the real name. The specific problem that sparked this change was the combination of `zig_libc_name_f80(name) __##name##x` with the input `fma`, resulting in `__fmax`, which is a new intrinsic in recent versions of cl.exe. With the above changes in place, compiler_rt can output the following: ``` static zig_weak_linkage_fn zig_f80 zig_e___fmax(zig_f80, zig_f80, zig_f80); zig_export(zig_weak_linkage_fn zig_f80 zig_e___fmax(zig_f80, zig_f80, zig_f80), __fmax, "__fmax"); ``` Within compiler_rt, `zig_e___fmax` is used to refer to the function, but consumers will import `__fmax`, which maps to their `zig_float_fn_f80_fma` definition from zig.h. --- src/codegen/c.zig | 71 ++++++++++++++++++++++++++++++++++++------------------- 1 file changed, 47 insertions(+), 24 deletions(-) (limited to 'src/codegen/c.zig') diff --git a/src/codegen/c.zig b/src/codegen/c.zig index 0178ffef9a..17aad238cb 100644 --- a/src/codegen/c.zig +++ b/src/codegen/c.zig @@ -258,6 +258,20 @@ pub fn fmtIdent(ident: []const u8) std.fmt.Formatter(formatIdent) { return .{ .data = ident }; } +// Returns true if `formatIdent` would make any edits to ident. +// This must be kept in sync with `formatIdent`. +pub fn isMangledIdent(ident: []const u8, solo: bool) bool { + if (solo and isReservedIdent(ident)) return true; + for (ident, 0..) |c, i| { + switch (c) { + 'a'...'z', 'A'...'Z', '_' => {}, + '0'...'9' => if (i == 0) return true, + else => return true, + } + } + return false; +} + /// This data is available when outputting .c code for a `InternPool.Index` /// that corresponds to `func`. /// It is not available when generating .h file. @@ -526,6 +540,7 @@ pub const DeclGen = struct { is_naked_fn: bool, /// This is a borrowed reference from `link.C`. fwd_decl: std.ArrayList(u8), + error_msg: ?*Module.ErrorMsg, ctypes: CType.Store, /// Keeps track of anonymous decls that need to be rendered before this @@ -1598,7 +1613,7 @@ pub const DeclGen = struct { switch (name) { .export_index => |export_index| try dg.renderDeclName(w, fn_decl_index, export_index), - .string => |string| try w.writeAll(string), + .string => |string| try w.print("{ }", .{fmtIdent(string)}), } try renderTypeSuffix( @@ -1813,9 +1828,17 @@ pub const DeclGen = struct { fn declIsGlobal(dg: *DeclGen, tv: TypedValue) bool { const mod = dg.module; return switch (mod.intern_pool.indexToKey(tv.val.ip_index)) { - .variable => |variable| mod.decl_exports.contains(variable.decl), + .variable => |variable| { + if (mod.decl_exports.get(variable.decl)) |exports| { + return !isMangledIdent(dg.module.intern_pool.stringToSlice(exports.items[0].opts.name), true); + } else return false; + }, .extern_func => true, - .func => |func| mod.decl_exports.contains(func.owner_decl), + .func => |func| { + if (mod.decl_exports.get(func.owner_decl)) |exports| { + return !isMangledIdent(dg.module.intern_pool.stringToSlice(exports.items[0].opts.name), true); + } else return false; + }, else => unreachable, }; } @@ -1925,9 +1948,9 @@ pub const DeclGen = struct { try mod.markDeclAlive(decl); if (mod.decl_exports.get(decl_index)) |exports| { - try writer.print("{}", .{exports.items[export_index].opts.name.fmt(&mod.intern_pool)}); + try writer.print("{ }", .{fmtIdent(mod.intern_pool.stringToSlice(exports.items[export_index].opts.name))}); } else if (decl.getExternDecl(mod).unwrap()) |extern_decl_index| { - try writer.print("{}", .{mod.declPtr(extern_decl_index).name.fmt(&mod.intern_pool)}); + try writer.print("{ }", .{fmtIdent(mod.intern_pool.stringToSlice(mod.declPtr(extern_decl_index).name))}); } else { // MSVC has a limit of 4095 character token length limit, and fmtIdent can (worst case), // expand to 3x the length of its input, but let's cut it off at a much shorter limit. @@ -2564,16 +2587,19 @@ fn genExports(o: *Object) !void { const fwd = o.dg.fwd_decl.writer(); const exports = mod.decl_exports.get(decl_index) orelse return; - if (exports.items.len < 2) return; + + const is_mangled = isMangledIdent(ip.stringToSlice(exports.items[0].opts.name), true); + if (exports.items.len < 2 and !is_mangled) return; switch (ip.indexToKey(tv.val.toIntern())) { .func => { - for (exports.items[1..], 1..) |@"export", i| { + const start_i = 1 - @intFromBool(is_mangled); + for (exports.items[start_i..], start_i..) |@"export", i| { try fwd.writeAll("zig_export("); if (exports.items[i].opts.linkage == .Weak) try fwd.writeAll("zig_weak_linkage_fn "); try o.dg.renderFunctionSignature(fwd, decl_index, .forward, .{ .export_index = @as(u32, @intCast(i)) }); - try fwd.print(", {s}, {s});\n", .{ - fmtStringLiteral(ip.stringToSlice(exports.items[0].opts.name), null), + try fwd.print(", { }, {s});\n", .{ + fmtIdent(ip.stringToSlice(exports.items[0].opts.name)), fmtStringLiteral(ip.stringToSlice(@"export".opts.name), null), }); } @@ -2583,7 +2609,8 @@ fn genExports(o: *Object) !void { unreachable; }, .variable => |variable| { - for (exports.items[1..], 1..) |@"export", i| { + const start_i = 1 - @intFromBool(is_mangled); + for (exports.items[start_i..], start_i..) |@"export", i| { try fwd.writeAll("zig_export("); if (exports.items[i].opts.linkage == .Weak) try fwd.writeAll("zig_weak_linkage "); const alias = ip.stringToSlice(@"export".opts.name); @@ -2595,8 +2622,8 @@ fn genExports(o: *Object) !void { decl.alignment, .complete, ); - try fwd.print(", {s}, {s});\n", .{ - fmtStringLiteral(ip.stringToSlice(exports.items[0].opts.name), null), + try fwd.print(", { }, {s});\n", .{ + fmtIdent(ip.stringToSlice(exports.items[0].opts.name)), fmtStringLiteral(alias, null), }); } @@ -6861,9 +6888,9 @@ fn airReduce(f: *Function, inst: Air.Inst.Index) !CValue { try f.writeCValue(writer, accum, .Other); switch (op) { .float_op => |func| { - try writer.writeAll(" = zig_libc_name_"); + try writer.writeAll(" = zig_float_fn_"); try f.object.dg.renderTypeForBuiltinFnName(writer, scalar_ty); - try writer.print("({s})(", .{func.operation}); + try writer.print("_{s}(", .{func.operation}); try f.writeCValue(writer, accum, .FunctionArgument); try writer.writeAll(", "); try f.writeCValue(writer, operand, .Other); @@ -7195,11 +7222,9 @@ fn unFloatOp(f: *Function, inst: Air.Inst.Index, operand: CValue, ty: Type, oper const v = try Vectorize.start(f, inst, writer, ty); try f.writeCValue(writer, local, .Other); try v.elem(f, writer); - try writer.writeAll(" = zig_libc_name_"); + try writer.writeAll(" = zig_float_fn_"); try f.object.dg.renderTypeForBuiltinFnName(writer, scalar_ty); - try writer.writeByte('('); - try writer.writeAll(operation); - try writer.writeAll(")("); + try writer.print("_{s}(", .{operation}); try f.writeCValue(writer, operand, .FunctionArgument); try v.elem(f, writer); try writer.writeAll(");\n"); @@ -7234,11 +7259,9 @@ fn airBinFloatOp(f: *Function, inst: Air.Inst.Index, operation: []const u8) !CVa const v = try Vectorize.start(f, inst, writer, inst_ty); try f.writeCValue(writer, local, .Other); try v.elem(f, writer); - try writer.writeAll(" = zig_libc_name_"); + try writer.writeAll(" = zig_float_fn_"); try f.object.dg.renderTypeForBuiltinFnName(writer, inst_scalar_ty); - try writer.writeByte('('); - try writer.writeAll(operation); - try writer.writeAll(")("); + try writer.print("_{s}(", .{operation}); try f.writeCValue(writer, lhs, .FunctionArgument); try v.elem(f, writer); try writer.writeAll(", "); @@ -7268,9 +7291,9 @@ fn airMulAdd(f: *Function, inst: Air.Inst.Index) !CValue { const v = try Vectorize.start(f, inst, writer, inst_ty); try f.writeCValue(writer, local, .Other); try v.elem(f, writer); - try writer.writeAll(" = zig_libc_name_"); + try writer.writeAll(" = zig_float_fn_"); try f.object.dg.renderTypeForBuiltinFnName(writer, inst_scalar_ty); - try writer.writeAll("(fma)("); + try writer.writeAll("_fma("); try f.writeCValue(writer, mulend1, .FunctionArgument); try v.elem(f, writer); try writer.writeAll(", "); -- cgit v1.2.3 From 192e9a315d92133b27d757f18bd4fd619976b755 Mon Sep 17 00:00:00 2001 From: kcbanner Date: Sat, 4 Nov 2023 19:37:19 -0400 Subject: cbe: add DeclVisibility and zig_extern_mangled to handle exporting mangled symbols under a different name --- lib/zig.h | 48 +++++++++++++++++-------------- src/codegen/c.zig | 84 ++++++++++++++++++++++++++++++++++++++++--------------- src/link/C.zig | 2 +- 3 files changed, 89 insertions(+), 45 deletions(-) (limited to 'src/codegen/c.zig') diff --git a/lib/zig.h b/lib/zig.h index ee4a338be7..c188b6e5c7 100644 --- a/lib/zig.h +++ b/lib/zig.h @@ -174,6 +174,12 @@ typedef char bool; #define zig_extern extern #endif +#if _MSC_VER +#define zig_extern_mangled zig_extern +#else +#define zig_extern_mangled static +#endif + #if zig_has_attribute(alias) #define zig_export(sig, symbol, name) zig_extern sig __attribute__((alias(#symbol))) #elif _MSC_VER @@ -190,14 +196,14 @@ typedef char bool; #if _MSC_VER #if _M_X64 -#define zig_import(sig, symbol, name) sig;\ +#define zig_import(sig, symbol, name) zig_extern sig;\ __pragma(comment(linker, "/alternatename:" #symbol "=" #name )) #else /*_M_X64 */ -#define zig_import(sig, symbol, name) sig;\ +#define zig_import(sig, symbol, name) zig_extern sig;\ __pragma(comment(linker, "/alternatename:_" #symbol "=_" #name )) #endif /*_M_X64 */ #else -#define zig_import(sig, symbol, name) zig_extern sig asm(#name); +#define zig_import(sig, symbol, name) zig_extern sig __asm(#name); #endif #define zig_expand_import(sig, symbol, name) zig_import(sig, symbol, name) @@ -3344,24 +3350,24 @@ zig_float_negate_builtin(128, zig_make_u128, (UINT64_C(1) << 63, UINT64_C(0))) zig_expand_concat(zig_float_binary_builtin_, zig_has_f##w)(f##w, sub, -) \ zig_expand_concat(zig_float_binary_builtin_, zig_has_f##w)(f##w, mul, *) \ zig_expand_concat(zig_float_binary_builtin_, zig_has_f##w)(f##w, div, /) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_sqrt(zig_f##w), zig_float_fn_f##w##_sqrt, zig_libc_name_f##w(sqrt)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_sin(zig_f##w), zig_float_fn_f##w##_sin, zig_libc_name_f##w(sin)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_cos(zig_f##w), zig_float_fn_f##w##_cos, zig_libc_name_f##w(cos)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_tan(zig_f##w), zig_float_fn_f##w##_tan, zig_libc_name_f##w(tan)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_exp(zig_f##w), zig_float_fn_f##w##_exp, zig_libc_name_f##w(exp)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_exp2(zig_f##w), zig_float_fn_f##w##_exp2, zig_libc_name_f##w(exp2)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_log(zig_f##w), zig_float_fn_f##w##_log, zig_libc_name_f##w(log)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_log2(zig_f##w), zig_float_fn_f##w##_log2, zig_libc_name_f##w(log2)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_log10(zig_f##w), zig_float_fn_f##w##_log10, zig_libc_name_f##w(log10)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_fabs(zig_f##w), zig_float_fn_f##w##_fabs, zig_libc_name_f##w(fabs)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_floor(zig_f##w), zig_float_fn_f##w##_floor, zig_libc_name_f##w(floor)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_ceil(zig_f##w), zig_float_fn_f##w##_ceil, zig_libc_name_f##w(ceil)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_round(zig_f##w), zig_float_fn_f##w##_round, zig_libc_name_f##w(round)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_trunc(zig_f##w), zig_float_fn_f##w##_trunc, zig_libc_name_f##w(trunc)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_fmod(zig_f##w, zig_f##w), zig_float_fn_f##w##_fmod, zig_libc_name_f##w(fmod)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_fmin(zig_f##w, zig_f##w), zig_float_fn_f##w##_fmin, zig_libc_name_f##w(fmin)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_fmax(zig_f##w, zig_f##w), zig_float_fn_f##w##_fmax, zig_libc_name_f##w(fmax)) \ - zig_expand_import(zig_extern zig_f##w zig_float_fn_f##w##_fma(zig_f##w, zig_f##w, zig_f##w), zig_float_fn_f##w##_fma, zig_libc_name_f##w(fma)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_sqrt(zig_f##w), zig_float_fn_f##w##_sqrt, zig_libc_name_f##w(sqrt)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_sin(zig_f##w), zig_float_fn_f##w##_sin, zig_libc_name_f##w(sin)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_cos(zig_f##w), zig_float_fn_f##w##_cos, zig_libc_name_f##w(cos)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_tan(zig_f##w), zig_float_fn_f##w##_tan, zig_libc_name_f##w(tan)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_exp(zig_f##w), zig_float_fn_f##w##_exp, zig_libc_name_f##w(exp)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_exp2(zig_f##w), zig_float_fn_f##w##_exp2, zig_libc_name_f##w(exp2)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_log(zig_f##w), zig_float_fn_f##w##_log, zig_libc_name_f##w(log)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_log2(zig_f##w), zig_float_fn_f##w##_log2, zig_libc_name_f##w(log2)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_log10(zig_f##w), zig_float_fn_f##w##_log10, zig_libc_name_f##w(log10)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_fabs(zig_f##w), zig_float_fn_f##w##_fabs, zig_libc_name_f##w(fabs)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_floor(zig_f##w), zig_float_fn_f##w##_floor, zig_libc_name_f##w(floor)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_ceil(zig_f##w), zig_float_fn_f##w##_ceil, zig_libc_name_f##w(ceil)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_round(zig_f##w), zig_float_fn_f##w##_round, zig_libc_name_f##w(round)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_trunc(zig_f##w), zig_float_fn_f##w##_trunc, zig_libc_name_f##w(trunc)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_fmod(zig_f##w, zig_f##w), zig_float_fn_f##w##_fmod, zig_libc_name_f##w(fmod)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_fmin(zig_f##w, zig_f##w), zig_float_fn_f##w##_fmin, zig_libc_name_f##w(fmin)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_fmax(zig_f##w, zig_f##w), zig_float_fn_f##w##_fmax, zig_libc_name_f##w(fmax)) \ + zig_expand_import(zig_f##w zig_float_fn_f##w##_fma(zig_f##w, zig_f##w, zig_f##w), zig_float_fn_f##w##_fma, zig_libc_name_f##w(fma)) \ \ static inline zig_f##w zig_div_trunc_f##w(zig_f##w lhs, zig_f##w rhs) { \ return zig_float_fn_f##w##_trunc(zig_div_f##w(lhs, rhs)); \ diff --git a/src/codegen/c.zig b/src/codegen/c.zig index 17aad238cb..735b58e2dd 100644 --- a/src/codegen/c.zig +++ b/src/codegen/c.zig @@ -272,6 +272,28 @@ pub fn isMangledIdent(ident: []const u8, solo: bool) bool { return false; } +const DeclVisibility = enum { + global, + global_mangled, + local, + + fn renderFwd(visibility: DeclVisibility, w: anytype) !void { + try w.writeAll(switch (visibility) { + .global => "zig_extern ", + // MSVC doesn't support exporting `static` functions, so they need special treatment + .global_mangled => "zig_extern_mangled ", + .local => "static ", + }); + } + + fn renderDef(visibility: DeclVisibility, w: anytype) !void { + return switch (visibility) { + .global => {}, + else => visibility.renderFwd(w), + }; + } +}; + /// This data is available when outputting .c code for a `InternPool.Index` /// that corresponds to `func`. /// It is not available when generating .h file. @@ -1825,19 +1847,25 @@ pub const DeclGen = struct { try renderTypeSuffix(dg.pass, store.*, mod, w, cty_idx, .suffix, .{}); } - fn declIsGlobal(dg: *DeclGen, tv: TypedValue) bool { + fn declVisibility(dg: *DeclGen, tv: TypedValue) DeclVisibility { const mod = dg.module; return switch (mod.intern_pool.indexToKey(tv.val.ip_index)) { .variable => |variable| { if (mod.decl_exports.get(variable.decl)) |exports| { - return !isMangledIdent(dg.module.intern_pool.stringToSlice(exports.items[0].opts.name), true); - } else return false; + return if (isMangledIdent(dg.module.intern_pool.stringToSlice(exports.items[0].opts.name), true)) + .global_mangled + else + .global; + } else return .local; }, - .extern_func => true, + .extern_func => .global, .func => |func| { if (mod.decl_exports.get(func.owner_decl)) |exports| { - return !isMangledIdent(dg.module.intern_pool.stringToSlice(exports.items[0].opts.name), true); - } else return false; + return if (isMangledIdent(dg.module.intern_pool.stringToSlice(exports.items[0].opts.name), true)) + .global_mangled + else + .global; + } else return .local; }, else => unreachable, }; @@ -1923,8 +1951,8 @@ pub const DeclGen = struct { fn renderFwdDecl(dg: *DeclGen, decl_index: Decl.Index, variable: InternPool.Key.Variable) !void { const decl = dg.module.declPtr(decl_index); const fwd = dg.fwd_decl.writer(); - const is_global = dg.declIsGlobal(.{ .ty = decl.ty, .val = decl.val }) or variable.is_extern; - try fwd.writeAll(if (is_global) "zig_extern " else "static "); + const visibility = if (variable.is_extern) .global else dg.declVisibility(.{ .ty = decl.ty, .val = decl.val }); + try visibility.renderFwd(fwd); const export_weak_linkage = if (dg.module.decl_exports.get(decl_index)) |exports| exports.items[0].opts.linkage == .Weak else @@ -2735,9 +2763,10 @@ pub fn genFunc(f: *Function) !void { o.code_header = std.ArrayList(u8).init(gpa); defer o.code_header.deinit(); - const is_global = o.dg.declIsGlobal(tv); + const visibility = o.dg.declVisibility(tv); const fwd_decl_writer = o.dg.fwd_decl.writer(); - try fwd_decl_writer.writeAll(if (is_global) "zig_extern " else "static "); + try visibility.renderFwd(fwd_decl_writer); + if (mod.decl_exports.get(decl_index)) |exports| if (exports.items[0].opts.linkage == .Weak) try fwd_decl_writer.writeAll("zig_weak_linkage_fn "); try o.dg.renderFunctionSignature(fwd_decl_writer, decl_index, .forward, .{ .export_index = 0 }); @@ -2745,7 +2774,7 @@ pub fn genFunc(f: *Function) !void { try genExports(o); try o.indent_writer.insertNewline(); - if (!is_global) try o.writer().writeAll("static "); + try visibility.renderDef(o.writer()); try o.dg.renderFunctionSignature(o.writer(), decl_index, .complete, .{ .export_index = 0 }); try o.writer().writeByte(' '); @@ -2829,9 +2858,9 @@ pub fn genDecl(o: *Object) !void { if (variable.is_extern) return; - const is_global = o.dg.declIsGlobal(tv) or variable.is_extern; + const visibility = if (variable.is_extern) .global else o.dg.declVisibility(tv); const w = o.writer(); - if (!is_global) try w.writeAll("static "); + try visibility.renderDef(w); if (variable.is_weak_linkage) try w.writeAll("zig_weak_linkage "); if (variable.is_threadlocal) try w.writeAll("zig_threadlocal "); if (mod.intern_pool.stringToSliceUnwrap(decl.@"linksection")) |s| @@ -2844,16 +2873,21 @@ pub fn genDecl(o: *Object) !void { try w.writeByte(';'); try o.indent_writer.insertNewline(); } else { - const is_global = o.dg.module.decl_exports.contains(decl_index); + const visibility: DeclVisibility = if (o.dg.module.decl_exports.get(decl_index)) |exports| b: { + break :b if (isMangledIdent(o.dg.module.intern_pool.stringToSlice(exports.items[0].opts.name), true)) + .global_mangled + else + .global; + } else .local; const decl_c_value = .{ .decl = decl_index }; - return genDeclValue(o, tv, is_global, decl_c_value, decl.alignment, decl.@"linksection"); + return genDeclValue(o, tv, visibility, decl_c_value, decl.alignment, decl.@"linksection"); } } pub fn genDeclValue( o: *Object, tv: TypedValue, - is_global: bool, + visibility: DeclVisibility, decl_c_value: CValue, alignment: Alignment, link_section: InternPool.OptionalNullTerminatedString, @@ -2861,12 +2895,13 @@ pub fn genDeclValue( const mod = o.dg.module; const fwd_decl_writer = o.dg.fwd_decl.writer(); - try fwd_decl_writer.writeAll(if (is_global) "zig_extern " else "static "); + try visibility.renderFwd(fwd_decl_writer); try o.dg.renderTypeAndName(fwd_decl_writer, tv.ty, decl_c_value, Const, alignment, .complete); try fwd_decl_writer.writeAll(";\n"); const w = o.writer(); - if (!is_global) try w.writeAll("static "); + try visibility.renderDef(w); + if (mod.intern_pool.stringToSliceUnwrap(link_section)) |s| try w.print("zig_linksection(\"{s}\", ", .{s}); try o.dg.renderTypeAndName(w, tv.ty, decl_c_value, Const, alignment, .complete); @@ -2891,11 +2926,14 @@ pub fn genHeader(dg: *DeclGen) error{ AnalysisFail, OutOfMemory }!void { switch (tv.ty.zigTypeTag(mod)) { .Fn => { - const is_global = dg.declIsGlobal(tv); - if (is_global) { - try writer.writeAll("zig_extern "); - try dg.renderFunctionSignature(writer, dg.pass.decl, .complete, .{ .export_index = 0 }); - try dg.fwd_decl.appendSlice(";\n"); + const visibility = dg.declVisibility(tv); + switch (visibility) { + .global, .global_mangled => { + try visibility.renderFwd(writer); + try dg.renderFunctionSignature(writer, dg.pass.decl, .complete, .{ .export_index = 0 }); + try dg.fwd_decl.appendSlice(";\n"); + }, + .local => {}, } }, else => {}, diff --git a/src/link/C.zig b/src/link/C.zig index 40dfc0771d..53d57c25cb 100644 --- a/src/link/C.zig +++ b/src/link/C.zig @@ -259,7 +259,7 @@ fn updateAnonDecl(self: *C, module: *Module, i: usize) !void { }; const c_value: codegen.CValue = .{ .constant = anon_decl }; const alignment: Alignment = self.aligned_anon_decls.get(anon_decl) orelse .none; - codegen.genDeclValue(&object, tv, false, c_value, alignment, .none) catch |err| switch (err) { + codegen.genDeclValue(&object, tv, .local, c_value, alignment, .none) catch |err| switch (err) { error.AnalysisFail => { @panic("TODO: C backend AnalysisFail on anonymous decl"); //try module.failed_decls.put(gpa, decl_index, object.dg.error_msg.?); -- cgit v1.2.3 From 26dabbf301ce4cd950cd66373dc17f0747c1e813 Mon Sep 17 00:00:00 2001 From: kcbanner Date: Sun, 5 Nov 2023 00:27:08 -0400 Subject: cbe: handle underscore prexfix on macos, don't mangle extern function names --- lib/zig.h | 4 ++++ src/codegen/c.zig | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) (limited to 'src/codegen/c.zig') diff --git a/lib/zig.h b/lib/zig.h index c188b6e5c7..032ad30053 100644 --- a/lib/zig.h +++ b/lib/zig.h @@ -203,7 +203,11 @@ typedef char bool; __pragma(comment(linker, "/alternatename:_" #symbol "=_" #name )) #endif /*_M_X64 */ #else +#if __APPLE__ +#define zig_import(sig, symbol, name) zig_extern sig __asm("_" #name); +#else /* __APPLE__ */ #define zig_import(sig, symbol, name) zig_extern sig __asm(#name); +#endif /* __APPLE__ */ #endif #define zig_expand_import(sig, symbol, name) zig_import(sig, symbol, name) diff --git a/src/codegen/c.zig b/src/codegen/c.zig index 735b58e2dd..46895d71a7 100644 --- a/src/codegen/c.zig +++ b/src/codegen/c.zig @@ -1978,7 +1978,7 @@ pub const DeclGen = struct { if (mod.decl_exports.get(decl_index)) |exports| { try writer.print("{ }", .{fmtIdent(mod.intern_pool.stringToSlice(exports.items[export_index].opts.name))}); } else if (decl.getExternDecl(mod).unwrap()) |extern_decl_index| { - try writer.print("{ }", .{fmtIdent(mod.intern_pool.stringToSlice(mod.declPtr(extern_decl_index).name))}); + try writer.print("{}", .{mod.declPtr(extern_decl_index).name.fmt(&mod.intern_pool)}); } else { // MSVC has a limit of 4095 character token length limit, and fmtIdent can (worst case), // expand to 3x the length of its input, but let's cut it off at a much shorter limit. -- cgit v1.2.3