authorgravatar for kubkon@jakubkonka.comJakub Konka <kubkon@jakubkonka.com> 2023-01-20 12:10:16+01:00
committergravatar for kubkon@jakubkonka.comJakub Konka <kubkon@jakubkonka.com> 2023-01-20 12:10:16+01:00
log9d2711a3d9bfcacef2c23c9dd9c07f77ab110744
tree2dbd54c45a681d65dd0aa9d68b273102791dc4e4
parente17c551a2e3a0130339b6f1e5817fea2214c260f

macho: use ArrayHashMap for tracking of dyld runtime metadata

1 files changed, 31 insertions(+), 57 deletions(-)

src/link/MachO.zig+31-57
......@@ -238,10 +238,10 @@ const Entry = struct {
238238 }
239239};
240240
241const BindingTable = std.AutoHashMapUnmanaged(*Atom, std.ArrayListUnmanaged(Atom.Binding));
242const UnnamedConstTable = std.AutoHashMapUnmanaged(Module.Decl.Index, std.ArrayListUnmanaged(*Atom));
243const RebaseTable = std.AutoHashMapUnmanaged(*Atom, std.ArrayListUnmanaged(u32));
244const RelocationTable = std.AutoHashMapUnmanaged(*Atom, std.ArrayListUnmanaged(Relocation));
241const BindingTable = std.AutoArrayHashMapUnmanaged(*Atom, std.ArrayListUnmanaged(Atom.Binding));
242const UnnamedConstTable = std.AutoArrayHashMapUnmanaged(Module.Decl.Index, std.ArrayListUnmanaged(*Atom));
243const RebaseTable = std.AutoArrayHashMapUnmanaged(*Atom, std.ArrayListUnmanaged(u32));
244const RelocationTable = std.AutoArrayHashMapUnmanaged(*Atom, std.ArrayListUnmanaged(Relocation));
245245
246246const PendingUpdate = union(enum) {
247247 resolve_undef: u32,
......@@ -547,11 +547,8 @@ pub fn flushModule(self: *MachO, comp: *Compilation, prog_node: *std.Progress.No
547547
548548 try self.allocateSpecialSymbols();
549549
550 {
551 var it = self.relocs.keyIterator();
552 while (it.next()) |atom| {
553 try atom.*.resolveRelocations(self);
554 }
550 for (self.relocs.keys()) |atom| {
551 try atom.resolveRelocations(self);
555552 }
556553
557554 if (build_options.enable_logging) {
......@@ -1018,8 +1015,7 @@ fn writePtrWidthAtom(self: *MachO, atom: *Atom) !void {
10181015
10191016fn markRelocsDirtyByTarget(self: *MachO, target: SymbolWithLoc) void {
10201017 // TODO: reverse-lookup might come in handy here
1021 var it = self.relocs.valueIterator();
1022 while (it.next()) |relocs| {
1018 for (self.relocs.values()) |*relocs| {
10231019 for (relocs.items) |*reloc| {
10241020 if (!reloc.target.eql(target)) continue;
10251021 reloc.dirty = true;
......@@ -1028,8 +1024,7 @@ fn markRelocsDirtyByTarget(self: *MachO, target: SymbolWithLoc) void {
10281024}
10291025
10301026fn markRelocsDirtyByAddress(self: *MachO, addr: u64) void {
1031 var it = self.relocs.valueIterator();
1032 while (it.next()) |relocs| {
1027 for (self.relocs.values()) |*relocs| {
10331028 for (relocs.items) |*reloc| {
10341029 const target_atom = reloc.getTargetAtom(self) orelse continue;
10351030 const target_sym = target_atom.getSymbol(self);
......@@ -1785,47 +1780,32 @@ pub fn deinit(self: *MachO) void {
17851780 assert(self.decls.count() == 0);
17861781 }
17871782
1788 {
1789 var it = self.unnamed_const_atoms.valueIterator();
1790 while (it.next()) |atoms| {
1791 atoms.deinit(gpa);
1792 }
1793 self.unnamed_const_atoms.deinit(gpa);
1783 for (self.unnamed_const_atoms.values()) |*atoms| {
1784 atoms.deinit(gpa);
17941785 }
1786 self.unnamed_const_atoms.deinit(gpa);
17951787
17961788 self.atom_by_index_table.deinit(gpa);
17971789
1798 {
1799 var it = self.relocs.valueIterator();
1800 while (it.next()) |relocs| {
1801 relocs.deinit(gpa);
1802 }
1803 self.relocs.deinit(gpa);
1790 for (self.relocs.values()) |*relocs| {
1791 relocs.deinit(gpa);
18041792 }
1793 self.relocs.deinit(gpa);
18051794
1806 {
1807 var it = self.rebases.valueIterator();
1808 while (it.next()) |rebases| {
1809 rebases.deinit(gpa);
1810 }
1811 self.rebases.deinit(gpa);
1795 for (self.rebases.values()) |*rebases| {
1796 rebases.deinit(gpa);
18121797 }
1798 self.rebases.deinit(gpa);
18131799
1814 {
1815 var it = self.bindings.valueIterator();
1816 while (it.next()) |bindings| {
1817 bindings.deinit(gpa);
1818 }
1819 self.bindings.deinit(gpa);
1800 for (self.bindings.values()) |*bindings| {
1801 bindings.deinit(gpa);
18201802 }
1803 self.bindings.deinit(gpa);
18211804
1822 {
1823 var it = self.lazy_bindings.valueIterator();
1824 while (it.next()) |bindings| {
1825 bindings.deinit(gpa);
1826 }
1827 self.lazy_bindings.deinit(gpa);
1805 for (self.lazy_bindings.values()) |*bindings| {
1806 bindings.deinit(gpa);
18281807 }
1808 self.lazy_bindings.deinit(gpa);
18291809}
18301810
18311811fn freeAtom(self: *MachO, atom: *Atom) void {
......@@ -2580,13 +2560,13 @@ pub fn deleteExport(self: *MachO, exp: Export) void {
25802560}
25812561
25822562fn freeRelocationsForAtom(self: *MachO, atom: *Atom) void {
2583 var removed_relocs = self.relocs.fetchRemove(atom);
2563 var removed_relocs = self.relocs.fetchOrderedRemove(atom);
25842564 if (removed_relocs) |*relocs| relocs.value.deinit(self.base.allocator);
2585 var removed_rebases = self.rebases.fetchRemove(atom);
2565 var removed_rebases = self.rebases.fetchOrderedRemove(atom);
25862566 if (removed_rebases) |*rebases| rebases.value.deinit(self.base.allocator);
2587 var removed_bindings = self.bindings.fetchRemove(atom);
2567 var removed_bindings = self.bindings.fetchOrderedRemove(atom);
25882568 if (removed_bindings) |*bindings| bindings.value.deinit(self.base.allocator);
2589 var removed_lazy_bindings = self.lazy_bindings.fetchRemove(atom);
2569 var removed_lazy_bindings = self.lazy_bindings.fetchOrderedRemove(atom);
25902570 if (removed_lazy_bindings) |*lazy_bindings| lazy_bindings.value.deinit(self.base.allocator);
25912571}
25922572
......@@ -3199,11 +3179,8 @@ fn writeLinkeditSegmentData(self: *MachO) !void {
31993179fn collectRebaseData(self: *MachO, rebase: *Rebase) !void {
32003180 const gpa = self.base.allocator;
32013181 const slice = self.sections.slice();
3202 var it = self.rebases.keyIterator();
3203
3204 while (it.next()) |key_ptr| {
3205 const atom = key_ptr.*;
32063182
3183 for (self.rebases.keys()) |atom, i| {
32073184 log.debug(" ATOM(%{d}, '{s}')", .{ atom.sym_index, atom.getName(self) });
32083185
32093186 const sym = atom.getSymbol(self);
......@@ -3212,7 +3189,7 @@ fn collectRebaseData(self: *MachO, rebase: *Rebase) !void {
32123189
32133190 const base_offset = sym.n_value - seg.vmaddr;
32143191
3215 const rebases = self.rebases.get(atom).?;
3192 const rebases = self.rebases.values()[i];
32163193 try rebase.entries.ensureUnusedCapacity(gpa, rebases.items.len);
32173194
32183195 for (rebases.items) |offset| {
......@@ -3231,11 +3208,8 @@ fn collectRebaseData(self: *MachO, rebase: *Rebase) !void {
32313208fn collectBindData(self: *MachO, bind: anytype, raw_bindings: anytype) !void {
32323209 const gpa = self.base.allocator;
32333210 const slice = self.sections.slice();
3234 var it = raw_bindings.keyIterator();
3235
3236 while (it.next()) |key_ptr| {
3237 const atom = key_ptr.*;
32383211
3212 for (raw_bindings.keys()) |atom, i| {
32393213 log.debug(" ATOM(%{d}, '{s}')", .{ atom.sym_index, atom.getName(self) });
32403214
32413215 const sym = atom.getSymbol(self);
......@@ -3244,7 +3218,7 @@ fn collectBindData(self: *MachO, bind: anytype, raw_bindings: anytype) !void {
32443218
32453219 const base_offset = sym.n_value - seg.vmaddr;
32463220
3247 const bindings = raw_bindings.get(atom).?;
3221 const bindings = raw_bindings.values()[i];
32483222 try bind.entries.ensureUnusedCapacity(gpa, bindings.items.len);
32493223
32503224 for (bindings.items) |binding| {