| ... | @@ -1,144 +0,0 @@ |
| 1 | const std = @import("std"); |
| 2 | const FileSource = std.Build.FileSource; |
| 3 | const Step = std.Build.Step; |
| 4 | |
| 5 | pub const requires_symlinks = true; |
| 6 | |
| 7 | pub fn build(b: *std.Build) void { |
| 8 | const test_step = b.step("test", "Test"); |
| 9 | b.default_step = test_step; |
| 10 | |
| 11 | // We force cross-compilation to ensure we always pick a generic CPU with |
| 12 | // constant set of CPU features. |
| 13 | const aarch64_macos = std.zig.CrossTarget{ |
| 14 | .cpu_arch = .aarch64, |
| 15 | .os_tag = .macos, |
| 16 | }; |
| 17 | |
| 18 | testUuid(b, test_step, .ReleaseSafe, aarch64_macos); |
| 19 | testUuid(b, test_step, .ReleaseFast, aarch64_macos); |
| 20 | testUuid(b, test_step, .ReleaseSmall, aarch64_macos); |
| 21 | |
| 22 | const x86_64_macos = std.zig.CrossTarget{ |
| 23 | .cpu_arch = .x86_64, |
| 24 | .os_tag = .macos, |
| 25 | }; |
| 26 | |
| 27 | testUuid(b, test_step, .ReleaseSafe, x86_64_macos); |
| 28 | testUuid(b, test_step, .ReleaseFast, x86_64_macos); |
| 29 | testUuid(b, test_step, .ReleaseSmall, x86_64_macos); |
| 30 | } |
| 31 | |
| 32 | fn testUuid( |
| 33 | b: *std.Build, |
| 34 | test_step: *std.Build.Step, |
| 35 | optimize: std.builtin.OptimizeMode, |
| 36 | target: std.zig.CrossTarget, |
| 37 | ) void { |
| 38 | // The calculated UUID value is independent of debug info and so it should |
| 39 | // stay the same across builds. |
| 40 | { |
| 41 | const dylib = simpleDylib(b, optimize, target); |
| 42 | const install_step = b.addInstallArtifact(dylib); |
| 43 | install_step.dest_sub_path = "test1.dylib"; |
| 44 | install_step.step.dependOn(&dylib.step); |
| 45 | } |
| 46 | { |
| 47 | const dylib = simpleDylib(b, optimize, target); |
| 48 | dylib.strip = true; |
| 49 | const install_step = b.addInstallArtifact(dylib); |
| 50 | install_step.dest_sub_path = "test2.dylib"; |
| 51 | install_step.step.dependOn(&dylib.step); |
| 52 | } |
| 53 | |
| 54 | const cmp_step = CompareUuid.create(b, "test1.dylib", "test2.dylib"); |
| 55 | test_step.dependOn(&cmp_step.step); |
| 56 | } |
| 57 | |
| 58 | fn simpleDylib( |
| 59 | b: *std.Build, |
| 60 | optimize: std.builtin.OptimizeMode, |
| 61 | target: std.zig.CrossTarget, |
| 62 | ) *std.Build.Step.Compile { |
| 63 | const dylib = b.addSharedLibrary(.{ |
| 64 | .name = "test", |
| 65 | .version = .{ .major = 1, .minor = 0, .patch = 0 }, |
| 66 | .optimize = optimize, |
| 67 | .target = target, |
| 68 | }); |
| 69 | dylib.addCSourceFile("test.c", &.{}); |
| 70 | dylib.linkLibC(); |
| 71 | return dylib; |
| 72 | } |
| 73 | |
| 74 | const CompareUuid = struct { |
| 75 | pub const base_id = .custom; |
| 76 | |
| 77 | step: Step, |
| 78 | lhs: []const u8, |
| 79 | rhs: []const u8, |
| 80 | |
| 81 | pub fn create(owner: *std.Build, lhs: []const u8, rhs: []const u8) *CompareUuid { |
| 82 | const self = owner.allocator.create(CompareUuid) catch @panic("OOM"); |
| 83 | self.* = CompareUuid{ |
| 84 | .step = Step.init(.{ |
| 85 | .id = base_id, |
| 86 | .name = owner.fmt("compare uuid: {s} and {s}", .{ |
| 87 | lhs, |
| 88 | rhs, |
| 89 | }), |
| 90 | .owner = owner, |
| 91 | .makeFn = make, |
| 92 | }), |
| 93 | .lhs = lhs, |
| 94 | .rhs = rhs, |
| 95 | }; |
| 96 | return self; |
| 97 | } |
| 98 | |
| 99 | fn make(step: *Step, prog_node: *std.Progress.Node) anyerror!void { |
| 100 | _ = prog_node; |
| 101 | const b = step.owner; |
| 102 | const self = @fieldParentPtr(CompareUuid, "step", step); |
| 103 | const gpa = b.allocator; |
| 104 | |
| 105 | var lhs_uuid: [16]u8 = undefined; |
| 106 | const lhs_path = b.getInstallPath(.lib, self.lhs); |
| 107 | try parseUuid(gpa, lhs_path, &lhs_uuid); |
| 108 | |
| 109 | var rhs_uuid: [16]u8 = undefined; |
| 110 | const rhs_path = b.getInstallPath(.lib, self.rhs); |
| 111 | try parseUuid(gpa, rhs_path, &rhs_uuid); |
| 112 | |
| 113 | try std.testing.expectEqualStrings(&lhs_uuid, &rhs_uuid); |
| 114 | } |
| 115 | |
| 116 | fn parseUuid(gpa: std.mem.Allocator, path: []const u8, uuid: *[16]u8) anyerror!void { |
| 117 | const max_bytes: usize = 20 * 1024 * 1024; |
| 118 | const data = try std.fs.cwd().readFileAllocOptions( |
| 119 | gpa, |
| 120 | path, |
| 121 | max_bytes, |
| 122 | null, |
| 123 | @alignOf(u64), |
| 124 | null, |
| 125 | ); |
| 126 | var stream = std.io.fixedBufferStream(data); |
| 127 | const reader = stream.reader(); |
| 128 | |
| 129 | const hdr = try reader.readStruct(std.macho.mach_header_64); |
| 130 | if (hdr.magic != std.macho.MH_MAGIC_64) { |
| 131 | return error.InvalidMagicNumber; |
| 132 | } |
| 133 | |
| 134 | var it = std.macho.LoadCommandIterator{ |
| 135 | .ncmds = hdr.ncmds, |
| 136 | .buffer = data[@sizeOf(std.macho.mach_header_64)..][0..hdr.sizeofcmds], |
| 137 | }; |
| 138 | const cmd = while (it.next()) |cmd| switch (cmd.cmd()) { |
| 139 | .UUID => break cmd.cast(std.macho.uuid_command).?, |
| 140 | else => {}, |
| 141 | } else return error.UuidLoadCommandNotFound; |
| 142 | std.mem.copy(u8, uuid, &cmd.uuid); |
| 143 | } |
| 144 | }; |
Jakub Konka <kubkon@jakubkonka.com> 2023-06-19 11:33:06+02:00