authorgravatar for kubkon@jakubkonka.comJakub Konka <kubkon@jakubkonka.com> 2022-09-09 22:29:50+02:00
committergravatar for kubkon@jakubkonka.comJakub Konka <kubkon@jakubkonka.com> 2022-09-09 22:29:50+02:00
logbac065c7cfb6f3aa0fe1cdf1334adb5a46d9a2af
tree17bb6f6eed2416214493a9b7f25e1860203ef7c2
parent8d44e031618a956a1b31c36b4c096a3000678a6b

coff: use global accessor abstractions from MachO

1 files changed, 52 insertions(+), 16 deletions(-)

src/link/Coff.zig+52-16
......@@ -1423,23 +1423,22 @@ fn resolveGlobalSymbol(self: *Coff, current: SymbolWithLoc) !void {
14231423 const sym = self.getSymbol(current);
14241424 const sym_name = self.getSymbolName(current);
14251425
1426 const global_index = self.resolver.get(sym_name) orelse {
1427 const name = try gpa.dupe(u8, sym_name);
1428 const global_index = try self.allocateGlobal();
1429 self.globals.items[global_index] = current;
1430 try self.resolver.putNoClobber(gpa, name, global_index);
1426 const gop = try self.getOrPutGlobalPtr(sym_name);
1427 if (!gop.found_existing) {
1428 gop.value_ptr.* = current;
14311429 if (sym.section_number == .UNDEFINED) {
1432 try self.unresolved.putNoClobber(gpa, global_index, false);
1430 try self.unresolved.putNoClobber(gpa, self.getGlobalIndex(sym_name).?, false);
14331431 }
14341432 return;
1435 };
1433 }
14361434
14371435 log.debug("TODO finish resolveGlobalSymbols implementation", .{});
14381436
14391437 if (sym.section_number == .UNDEFINED) return;
14401438
1441 _ = self.unresolved.swapRemove(global_index);
1442 self.globals.items[global_index] = current;
1439 _ = self.unresolved.swapRemove(self.getGlobalIndex(sym_name).?);
1440
1441 gop.value_ptr.* = current;
14431442}
14441443
14451444pub fn flush(self: *Coff, comp: *Compilation, prog_node: *std.Progress.Node) !void {
......@@ -1544,23 +1543,23 @@ pub fn getDeclVAddr(
15441543}
15451544
15461545pub fn getGlobalSymbol(self: *Coff, name: []const u8) !u32 {
1547 if (self.resolver.get(name)) |global_index| {
1548 return self.globals.items[global_index].sym_index;
1546 const gop = try self.getOrPutGlobalPtr(name);
1547
1548 if (gop.found_existing) {
1549 return gop.value_ptr.sym_index;
15491550 }
15501551
1551 const gpa = self.base.allocator;
15521552 const sym_index = try self.allocateSymbol();
1553 const global_index = try self.allocateGlobal();
15541553 const sym_loc = SymbolWithLoc{ .sym_index = sym_index, .file = null };
1555 self.globals.items[global_index] = sym_loc;
1554 gop.value_ptr.* = sym_loc;
15561555
1556 const gpa = self.base.allocator;
15571557 const sym_name = try gpa.dupe(u8, name);
15581558 const sym = self.getSymbolPtr(sym_loc);
15591559 try self.setSymbolName(sym, sym_name);
15601560 sym.storage_class = .EXTERNAL;
15611561
1562 try self.resolver.putNoClobber(gpa, sym_name, global_index);
1563 try self.unresolved.putNoClobber(gpa, global_index, true);
1562 try self.unresolved.putNoClobber(gpa, self.getGlobalIndex(name).?, true);
15641563
15651564 return sym_index;
15661565}
......@@ -2061,6 +2060,43 @@ pub fn getSymbolName(self: *const Coff, sym_loc: SymbolWithLoc) []const u8 {
20612060 return self.strtab.get(offset).?;
20622061}
20632062
2063/// Returns pointer to the global entry for `name` if one exists.
2064pub fn getGlobalPtr(self: *Coff, name: []const u8) ?*SymbolWithLoc {
2065 const global_index = self.resolver.get(name) orelse return null;
2066 return &self.globals.items[global_index];
2067}
2068
2069/// Returns the global entry for `name` if one exists.
2070pub fn getGlobal(self: *const Coff, name: []const u8) ?SymbolWithLoc {
2071 const global_index = self.resolver.get(name) orelse return null;
2072 return self.globals.items[global_index];
2073}
2074
2075/// Returns the index of the global entry for `name` if one exists.
2076pub fn getGlobalIndex(self: *const Coff, name: []const u8) ?u32 {
2077 return self.resolver.get(name);
2078}
2079
2080const GetOrPutGlobalPtrResult = struct {
2081 found_existing: bool,
2082 value_ptr: *SymbolWithLoc,
2083};
2084
2085/// Return pointer to the global entry for `name` if one exists.
2086/// Puts a new global entry for `name` if one doesn't exist, and
2087/// returns a pointer to it.
2088pub fn getOrPutGlobalPtr(self: *Coff, name: []const u8) !GetOrPutGlobalPtrResult {
2089 if (self.getGlobalPtr(name)) |ptr| {
2090 return GetOrPutGlobalPtrResult{ .found_existing = true, .value_ptr = ptr };
2091 }
2092 const gpa = self.base.allocator;
2093 const global_index = try self.allocateGlobal();
2094 const global_name = try gpa.dupe(u8, name);
2095 _ = try self.resolver.put(gpa, global_name, global_index);
2096 const ptr = &self.globals.items[global_index];
2097 return GetOrPutGlobalPtrResult{ .found_existing = false, .value_ptr = ptr };
2098}
2099
20642100/// Returns atom if there is an atom referenced by the symbol described by `sym_loc` descriptor.
20652101/// Returns null on failure.
20662102pub fn getAtomForSymbol(self: *Coff, sym_loc: SymbolWithLoc) ?*Atom {