mirror of
https://github.com/ziglang/zig.git
synced 2026-01-06 05:25:10 +00:00
4135 lines
161 KiB
Zig
4135 lines
161 KiB
Zig
const std = @import("std");
|
|
const build_options = @import("build_options");
|
|
const assert = std.debug.assert;
|
|
const dwarf = std.dwarf;
|
|
const fs = std.fs;
|
|
const log = std.log.scoped(.link);
|
|
const macho = std.macho;
|
|
const math = std.math;
|
|
const mem = std.mem;
|
|
|
|
const aarch64 = @import("../../arch/aarch64/bits.zig");
|
|
const bind = @import("bind.zig");
|
|
const dead_strip = @import("dead_strip.zig");
|
|
const fat = @import("fat.zig");
|
|
const link = @import("../../link.zig");
|
|
const load_commands = @import("load_commands.zig");
|
|
const thunks = @import("thunks.zig");
|
|
const trace = @import("../../tracy.zig").trace;
|
|
const uuid = @import("uuid.zig");
|
|
|
|
const Allocator = mem.Allocator;
|
|
const Archive = @import("Archive.zig");
|
|
const Atom = @import("ZldAtom.zig");
|
|
const Cache = @import("../../Cache.zig");
|
|
const CodeSignature = @import("CodeSignature.zig");
|
|
const Compilation = @import("../../Compilation.zig");
|
|
const DwarfInfo = @import("DwarfInfo.zig");
|
|
const Dylib = @import("Dylib.zig");
|
|
const MachO = @import("../MachO.zig");
|
|
const LibStub = @import("../tapi.zig").LibStub;
|
|
const Object = @import("Object.zig");
|
|
const StringTable = @import("../strtab.zig").StringTable;
|
|
const Trie = @import("Trie.zig");
|
|
|
|
pub const Zld = struct {
|
|
gpa: Allocator,
|
|
file: fs.File,
|
|
page_size: u16,
|
|
options: *const link.Options,
|
|
|
|
objects: std.ArrayListUnmanaged(Object) = .{},
|
|
archives: std.ArrayListUnmanaged(Archive) = .{},
|
|
dylibs: std.ArrayListUnmanaged(Dylib) = .{},
|
|
dylibs_map: std.StringHashMapUnmanaged(u16) = .{},
|
|
referenced_dylibs: std.AutoArrayHashMapUnmanaged(u16, void) = .{},
|
|
|
|
segments: std.ArrayListUnmanaged(macho.segment_command_64) = .{},
|
|
sections: std.MultiArrayList(Section) = .{},
|
|
|
|
locals: std.ArrayListUnmanaged(macho.nlist_64) = .{},
|
|
globals: std.ArrayListUnmanaged(SymbolWithLoc) = .{},
|
|
|
|
entry_index: ?u32 = null,
|
|
mh_execute_header_index: ?u32 = null,
|
|
dso_handle_index: ?u32 = null,
|
|
dyld_stub_binder_index: ?u32 = null,
|
|
dyld_private_sym_index: ?u32 = null,
|
|
stub_helper_preamble_sym_index: ?u32 = null,
|
|
|
|
strtab: StringTable(.strtab) = .{},
|
|
|
|
tlv_ptr_entries: std.ArrayListUnmanaged(IndirectPointer) = .{},
|
|
tlv_ptr_table: std.AutoHashMapUnmanaged(SymbolWithLoc, u32) = .{},
|
|
|
|
got_entries: std.ArrayListUnmanaged(IndirectPointer) = .{},
|
|
got_table: std.AutoHashMapUnmanaged(SymbolWithLoc, u32) = .{},
|
|
|
|
stubs: std.ArrayListUnmanaged(IndirectPointer) = .{},
|
|
stubs_table: std.AutoHashMapUnmanaged(SymbolWithLoc, u32) = .{},
|
|
|
|
thunk_table: std.AutoHashMapUnmanaged(AtomIndex, thunks.ThunkIndex) = .{},
|
|
thunks: std.ArrayListUnmanaged(thunks.Thunk) = .{},
|
|
|
|
atoms: std.ArrayListUnmanaged(Atom) = .{},
|
|
|
|
fn parseObject(self: *Zld, path: []const u8) !bool {
|
|
const gpa = self.gpa;
|
|
const file = fs.cwd().openFile(path, .{}) catch |err| switch (err) {
|
|
error.FileNotFound => return false,
|
|
else => |e| return e,
|
|
};
|
|
defer file.close();
|
|
|
|
const name = try gpa.dupe(u8, path);
|
|
errdefer gpa.free(name);
|
|
const cpu_arch = self.options.target.cpu.arch;
|
|
const mtime: u64 = mtime: {
|
|
const stat = file.stat() catch break :mtime 0;
|
|
break :mtime @intCast(u64, @divFloor(stat.mtime, 1_000_000_000));
|
|
};
|
|
const file_stat = try file.stat();
|
|
const file_size = math.cast(usize, file_stat.size) orelse return error.Overflow;
|
|
const contents = try file.readToEndAllocOptions(gpa, file_size, file_size, @alignOf(u64), null);
|
|
|
|
var object = Object{
|
|
.name = name,
|
|
.mtime = mtime,
|
|
.contents = contents,
|
|
};
|
|
|
|
object.parse(gpa, cpu_arch) catch |err| switch (err) {
|
|
error.EndOfStream, error.NotObject => {
|
|
object.deinit(gpa);
|
|
return false;
|
|
},
|
|
else => |e| return e,
|
|
};
|
|
|
|
try self.objects.append(gpa, object);
|
|
|
|
return true;
|
|
}
|
|
|
|
fn parseArchive(self: *Zld, path: []const u8, force_load: bool) !bool {
|
|
const gpa = self.gpa;
|
|
const file = fs.cwd().openFile(path, .{}) catch |err| switch (err) {
|
|
error.FileNotFound => return false,
|
|
else => |e| return e,
|
|
};
|
|
errdefer file.close();
|
|
|
|
const name = try gpa.dupe(u8, path);
|
|
errdefer gpa.free(name);
|
|
const cpu_arch = self.options.target.cpu.arch;
|
|
const reader = file.reader();
|
|
const fat_offset = try fat.getLibraryOffset(reader, cpu_arch);
|
|
try reader.context.seekTo(fat_offset);
|
|
|
|
var archive = Archive{
|
|
.name = name,
|
|
.fat_offset = fat_offset,
|
|
.file = file,
|
|
};
|
|
|
|
archive.parse(gpa, reader) catch |err| switch (err) {
|
|
error.EndOfStream, error.NotArchive => {
|
|
archive.deinit(gpa);
|
|
return false;
|
|
},
|
|
else => |e| return e,
|
|
};
|
|
|
|
if (force_load) {
|
|
defer archive.deinit(gpa);
|
|
// Get all offsets from the ToC
|
|
var offsets = std.AutoArrayHashMap(u32, void).init(gpa);
|
|
defer offsets.deinit();
|
|
for (archive.toc.values()) |offs| {
|
|
for (offs.items) |off| {
|
|
_ = try offsets.getOrPut(off);
|
|
}
|
|
}
|
|
for (offsets.keys()) |off| {
|
|
const object = try archive.parseObject(gpa, cpu_arch, off);
|
|
try self.objects.append(gpa, object);
|
|
}
|
|
} else {
|
|
try self.archives.append(gpa, archive);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
const ParseDylibError = error{
|
|
OutOfMemory,
|
|
EmptyStubFile,
|
|
MismatchedCpuArchitecture,
|
|
UnsupportedCpuArchitecture,
|
|
EndOfStream,
|
|
} || fs.File.OpenError || std.os.PReadError || Dylib.Id.ParseError;
|
|
|
|
const DylibCreateOpts = struct {
|
|
syslibroot: ?[]const u8,
|
|
id: ?Dylib.Id = null,
|
|
dependent: bool = false,
|
|
needed: bool = false,
|
|
weak: bool = false,
|
|
};
|
|
|
|
fn parseDylib(
|
|
self: *Zld,
|
|
path: []const u8,
|
|
dependent_libs: anytype,
|
|
opts: DylibCreateOpts,
|
|
) ParseDylibError!bool {
|
|
const gpa = self.gpa;
|
|
const file = fs.cwd().openFile(path, .{}) catch |err| switch (err) {
|
|
error.FileNotFound => return false,
|
|
else => |e| return e,
|
|
};
|
|
defer file.close();
|
|
|
|
const cpu_arch = self.options.target.cpu.arch;
|
|
const file_stat = try file.stat();
|
|
var file_size = math.cast(usize, file_stat.size) orelse return error.Overflow;
|
|
|
|
const reader = file.reader();
|
|
const fat_offset = math.cast(usize, try fat.getLibraryOffset(reader, cpu_arch)) orelse
|
|
return error.Overflow;
|
|
try file.seekTo(fat_offset);
|
|
file_size -= fat_offset;
|
|
|
|
const contents = try file.readToEndAllocOptions(gpa, file_size, file_size, @alignOf(u64), null);
|
|
defer gpa.free(contents);
|
|
|
|
const dylib_id = @intCast(u16, self.dylibs.items.len);
|
|
var dylib = Dylib{ .weak = opts.weak };
|
|
|
|
dylib.parseFromBinary(
|
|
gpa,
|
|
cpu_arch,
|
|
dylib_id,
|
|
dependent_libs,
|
|
path,
|
|
contents,
|
|
) catch |err| switch (err) {
|
|
error.EndOfStream, error.NotDylib => {
|
|
try file.seekTo(0);
|
|
|
|
var lib_stub = LibStub.loadFromFile(gpa, file) catch {
|
|
dylib.deinit(gpa);
|
|
return false;
|
|
};
|
|
defer lib_stub.deinit();
|
|
|
|
try dylib.parseFromStub(
|
|
gpa,
|
|
self.options.target,
|
|
lib_stub,
|
|
dylib_id,
|
|
dependent_libs,
|
|
path,
|
|
);
|
|
},
|
|
else => |e| return e,
|
|
};
|
|
|
|
if (opts.id) |id| {
|
|
if (dylib.id.?.current_version < id.compatibility_version) {
|
|
log.warn("found dylib is incompatible with the required minimum version", .{});
|
|
log.warn(" dylib: {s}", .{id.name});
|
|
log.warn(" required minimum version: {}", .{id.compatibility_version});
|
|
log.warn(" dylib version: {}", .{dylib.id.?.current_version});
|
|
|
|
// TODO maybe this should be an error and facilitate auto-cleanup?
|
|
dylib.deinit(gpa);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
try self.dylibs.append(gpa, dylib);
|
|
try self.dylibs_map.putNoClobber(gpa, dylib.id.?.name, dylib_id);
|
|
|
|
const should_link_dylib_even_if_unreachable = blk: {
|
|
if (self.options.dead_strip_dylibs and !opts.needed) break :blk false;
|
|
break :blk !(opts.dependent or self.referenced_dylibs.contains(dylib_id));
|
|
};
|
|
|
|
if (should_link_dylib_even_if_unreachable) {
|
|
try self.referenced_dylibs.putNoClobber(gpa, dylib_id, {});
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
fn parseInputFiles(
|
|
self: *Zld,
|
|
files: []const []const u8,
|
|
syslibroot: ?[]const u8,
|
|
dependent_libs: anytype,
|
|
) !void {
|
|
for (files) |file_name| {
|
|
const full_path = full_path: {
|
|
var buffer: [fs.MAX_PATH_BYTES]u8 = undefined;
|
|
break :full_path try fs.realpath(file_name, &buffer);
|
|
};
|
|
log.debug("parsing input file path '{s}'", .{full_path});
|
|
|
|
if (try self.parseObject(full_path)) continue;
|
|
if (try self.parseArchive(full_path, false)) continue;
|
|
if (try self.parseDylib(full_path, dependent_libs, .{
|
|
.syslibroot = syslibroot,
|
|
})) continue;
|
|
|
|
log.debug("unknown filetype for positional input file: '{s}'", .{file_name});
|
|
}
|
|
}
|
|
|
|
fn parseAndForceLoadStaticArchives(self: *Zld, files: []const []const u8) !void {
|
|
for (files) |file_name| {
|
|
const full_path = full_path: {
|
|
var buffer: [fs.MAX_PATH_BYTES]u8 = undefined;
|
|
break :full_path try fs.realpath(file_name, &buffer);
|
|
};
|
|
log.debug("parsing and force loading static archive '{s}'", .{full_path});
|
|
|
|
if (try self.parseArchive(full_path, true)) continue;
|
|
log.debug("unknown filetype: expected static archive: '{s}'", .{file_name});
|
|
}
|
|
}
|
|
|
|
fn parseLibs(
|
|
self: *Zld,
|
|
lib_names: []const []const u8,
|
|
lib_infos: []const link.SystemLib,
|
|
syslibroot: ?[]const u8,
|
|
dependent_libs: anytype,
|
|
) !void {
|
|
for (lib_names) |lib, i| {
|
|
const lib_info = lib_infos[i];
|
|
log.debug("parsing lib path '{s}'", .{lib});
|
|
if (try self.parseDylib(lib, dependent_libs, .{
|
|
.syslibroot = syslibroot,
|
|
.needed = lib_info.needed,
|
|
.weak = lib_info.weak,
|
|
})) continue;
|
|
if (try self.parseArchive(lib, false)) continue;
|
|
|
|
log.debug("unknown filetype for a library: '{s}'", .{lib});
|
|
}
|
|
}
|
|
|
|
fn parseDependentLibs(self: *Zld, syslibroot: ?[]const u8, dependent_libs: anytype) !void {
|
|
// At this point, we can now parse dependents of dylibs preserving the inclusion order of:
|
|
// 1) anything on the linker line is parsed first
|
|
// 2) afterwards, we parse dependents of the included dylibs
|
|
// TODO this should not be performed if the user specifies `-flat_namespace` flag.
|
|
// See ld64 manpages.
|
|
var arena_alloc = std.heap.ArenaAllocator.init(self.gpa);
|
|
const arena = arena_alloc.allocator();
|
|
defer arena_alloc.deinit();
|
|
|
|
while (dependent_libs.readItem()) |*dep_id| {
|
|
defer dep_id.id.deinit(self.gpa);
|
|
|
|
if (self.dylibs_map.contains(dep_id.id.name)) continue;
|
|
|
|
const weak = self.dylibs.items[dep_id.parent].weak;
|
|
const has_ext = blk: {
|
|
const basename = fs.path.basename(dep_id.id.name);
|
|
break :blk mem.lastIndexOfScalar(u8, basename, '.') != null;
|
|
};
|
|
const extension = if (has_ext) fs.path.extension(dep_id.id.name) else "";
|
|
const without_ext = if (has_ext) blk: {
|
|
const index = mem.lastIndexOfScalar(u8, dep_id.id.name, '.') orelse unreachable;
|
|
break :blk dep_id.id.name[0..index];
|
|
} else dep_id.id.name;
|
|
|
|
for (&[_][]const u8{ extension, ".tbd" }) |ext| {
|
|
const with_ext = try std.fmt.allocPrint(arena, "{s}{s}", .{ without_ext, ext });
|
|
const full_path = if (syslibroot) |root| try fs.path.join(arena, &.{ root, with_ext }) else with_ext;
|
|
|
|
log.debug("trying dependency at fully resolved path {s}", .{full_path});
|
|
|
|
const did_parse_successfully = try self.parseDylib(full_path, dependent_libs, .{
|
|
.id = dep_id.id,
|
|
.syslibroot = syslibroot,
|
|
.dependent = true,
|
|
.weak = weak,
|
|
});
|
|
if (did_parse_successfully) break;
|
|
} else {
|
|
log.debug("unable to resolve dependency {s}", .{dep_id.id.name});
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn getOutputSection(self: *Zld, sect: macho.section_64) !?u8 {
|
|
const segname = sect.segName();
|
|
const sectname = sect.sectName();
|
|
const res: ?u8 = blk: {
|
|
if (mem.eql(u8, "__LLVM", segname)) {
|
|
log.debug("TODO LLVM section: type 0x{x}, name '{s},{s}'", .{
|
|
sect.flags, segname, sectname,
|
|
});
|
|
break :blk null;
|
|
}
|
|
|
|
if (sect.isCode()) {
|
|
break :blk self.getSectionByName("__TEXT", "__text") orelse try self.initSection(
|
|
"__TEXT",
|
|
"__text",
|
|
.{
|
|
.flags = macho.S_REGULAR |
|
|
macho.S_ATTR_PURE_INSTRUCTIONS |
|
|
macho.S_ATTR_SOME_INSTRUCTIONS,
|
|
},
|
|
);
|
|
}
|
|
|
|
if (sect.isDebug()) {
|
|
// TODO debug attributes
|
|
if (mem.eql(u8, "__LD", segname) and mem.eql(u8, "__compact_unwind", sectname)) {
|
|
log.debug("TODO compact unwind section: type 0x{x}, name '{s},{s}'", .{
|
|
sect.flags, segname, sectname,
|
|
});
|
|
}
|
|
break :blk null;
|
|
}
|
|
|
|
switch (sect.type()) {
|
|
macho.S_4BYTE_LITERALS,
|
|
macho.S_8BYTE_LITERALS,
|
|
macho.S_16BYTE_LITERALS,
|
|
=> {
|
|
break :blk self.getSectionByName("__TEXT", "__const") orelse try self.initSection(
|
|
"__TEXT",
|
|
"__const",
|
|
.{},
|
|
);
|
|
},
|
|
macho.S_CSTRING_LITERALS => {
|
|
if (mem.startsWith(u8, sectname, "__objc")) {
|
|
break :blk self.getSectionByName(segname, sectname) orelse try self.initSection(
|
|
segname,
|
|
sectname,
|
|
.{},
|
|
);
|
|
}
|
|
break :blk self.getSectionByName("__TEXT", "__cstring") orelse try self.initSection(
|
|
"__TEXT",
|
|
"__cstring",
|
|
.{ .flags = macho.S_CSTRING_LITERALS },
|
|
);
|
|
},
|
|
macho.S_MOD_INIT_FUNC_POINTERS,
|
|
macho.S_MOD_TERM_FUNC_POINTERS,
|
|
=> {
|
|
break :blk self.getSectionByName("__DATA_CONST", sectname) orelse try self.initSection(
|
|
"__DATA_CONST",
|
|
sectname,
|
|
.{ .flags = sect.flags },
|
|
);
|
|
},
|
|
macho.S_LITERAL_POINTERS,
|
|
macho.S_ZEROFILL,
|
|
macho.S_THREAD_LOCAL_VARIABLES,
|
|
macho.S_THREAD_LOCAL_VARIABLE_POINTERS,
|
|
macho.S_THREAD_LOCAL_REGULAR,
|
|
macho.S_THREAD_LOCAL_ZEROFILL,
|
|
=> {
|
|
break :blk self.getSectionByName(segname, sectname) orelse try self.initSection(
|
|
segname,
|
|
sectname,
|
|
.{ .flags = sect.flags },
|
|
);
|
|
},
|
|
macho.S_COALESCED => {
|
|
// TODO unwind info
|
|
if (mem.eql(u8, "__TEXT", segname) and mem.eql(u8, "__eh_frame", sectname)) {
|
|
log.debug("TODO eh frame section: type 0x{x}, name '{s},{s}'", .{
|
|
sect.flags, segname, sectname,
|
|
});
|
|
break :blk null;
|
|
}
|
|
break :blk self.getSectionByName(segname, sectname) orelse try self.initSection(
|
|
segname,
|
|
sectname,
|
|
.{},
|
|
);
|
|
},
|
|
macho.S_REGULAR => {
|
|
if (mem.eql(u8, segname, "__TEXT")) {
|
|
if (mem.eql(u8, sectname, "__rodata") or
|
|
mem.eql(u8, sectname, "__typelink") or
|
|
mem.eql(u8, sectname, "__itablink") or
|
|
mem.eql(u8, sectname, "__gosymtab") or
|
|
mem.eql(u8, sectname, "__gopclntab"))
|
|
{
|
|
break :blk self.getSectionByName("__DATA_CONST", "__const") orelse try self.initSection(
|
|
"__DATA_CONST",
|
|
"__const",
|
|
.{},
|
|
);
|
|
}
|
|
}
|
|
if (mem.eql(u8, segname, "__DATA")) {
|
|
if (mem.eql(u8, sectname, "__const") or
|
|
mem.eql(u8, sectname, "__cfstring") or
|
|
mem.eql(u8, sectname, "__objc_classlist") or
|
|
mem.eql(u8, sectname, "__objc_imageinfo"))
|
|
{
|
|
break :blk self.getSectionByName("__DATA_CONST", sectname) orelse
|
|
try self.initSection(
|
|
"__DATA_CONST",
|
|
sectname,
|
|
.{},
|
|
);
|
|
} else if (mem.eql(u8, sectname, "__data")) {
|
|
break :blk self.getSectionByName("__DATA", "__data") orelse
|
|
try self.initSection(
|
|
"__DATA",
|
|
"__data",
|
|
.{},
|
|
);
|
|
}
|
|
}
|
|
break :blk self.getSectionByName(segname, sectname) orelse try self.initSection(
|
|
segname,
|
|
sectname,
|
|
.{},
|
|
);
|
|
},
|
|
else => break :blk null,
|
|
}
|
|
};
|
|
return res;
|
|
}
|
|
|
|
pub fn addAtomToSection(self: *Zld, atom_index: AtomIndex) void {
|
|
const atom = self.getAtomPtr(atom_index);
|
|
const sym = self.getSymbol(atom.getSymbolWithLoc());
|
|
var section = self.sections.get(sym.n_sect - 1);
|
|
if (section.header.size > 0) {
|
|
const last_atom = self.getAtomPtr(section.last_atom_index);
|
|
last_atom.next_index = atom_index;
|
|
atom.prev_index = section.last_atom_index;
|
|
} else {
|
|
section.first_atom_index = atom_index;
|
|
}
|
|
section.last_atom_index = atom_index;
|
|
section.header.size += atom.size;
|
|
self.sections.set(sym.n_sect - 1, section);
|
|
}
|
|
|
|
pub fn createEmptyAtom(self: *Zld, sym_index: u32, size: u64, alignment: u32) !AtomIndex {
|
|
const gpa = self.gpa;
|
|
const index = @intCast(AtomIndex, self.atoms.items.len);
|
|
const atom = try self.atoms.addOne(gpa);
|
|
atom.* = Atom.empty;
|
|
atom.sym_index = sym_index;
|
|
atom.size = size;
|
|
atom.alignment = alignment;
|
|
|
|
log.debug("creating ATOM(%{d}) at index {d}", .{ sym_index, index });
|
|
|
|
return index;
|
|
}
|
|
|
|
pub fn createGotAtom(self: *Zld) !AtomIndex {
|
|
const sym_index = try self.allocateSymbol();
|
|
const atom_index = try self.createEmptyAtom(sym_index, @sizeOf(u64), 3);
|
|
const sym = self.getSymbolPtr(.{ .sym_index = sym_index });
|
|
sym.n_type = macho.N_SECT;
|
|
|
|
const sect_id = self.getSectionByName("__DATA_CONST", "__got") orelse
|
|
try self.initSection("__DATA_CONST", "__got", .{
|
|
.flags = macho.S_NON_LAZY_SYMBOL_POINTERS,
|
|
});
|
|
sym.n_sect = sect_id + 1;
|
|
|
|
self.addAtomToSection(atom_index);
|
|
|
|
return atom_index;
|
|
}
|
|
|
|
fn writeGotPointer(self: *Zld, got_index: u32, writer: anytype) !void {
|
|
const target_addr = blk: {
|
|
const entry = self.got_entries.items[got_index];
|
|
const sym = entry.getTargetSymbol(self);
|
|
break :blk sym.n_value;
|
|
};
|
|
try writer.writeIntLittle(u64, target_addr);
|
|
}
|
|
|
|
pub fn createTlvPtrAtom(self: *Zld) !AtomIndex {
|
|
const sym_index = try self.allocateSymbol();
|
|
const atom_index = try self.createEmptyAtom(sym_index, @sizeOf(u64), 3);
|
|
const sym = self.getSymbolPtr(.{ .sym_index = sym_index });
|
|
sym.n_type = macho.N_SECT;
|
|
|
|
const sect_id = (try self.getOutputSection(.{
|
|
.segname = makeStaticString("__DATA"),
|
|
.sectname = makeStaticString("__thread_ptrs"),
|
|
.flags = macho.S_THREAD_LOCAL_VARIABLE_POINTERS,
|
|
})).?;
|
|
sym.n_sect = sect_id + 1;
|
|
|
|
self.addAtomToSection(atom_index);
|
|
|
|
return atom_index;
|
|
}
|
|
|
|
fn createDyldStubBinderGotAtom(self: *Zld) !void {
|
|
const sym_index = self.dyld_stub_binder_index orelse return;
|
|
const gpa = self.gpa;
|
|
const target = SymbolWithLoc{ .sym_index = sym_index };
|
|
const atom_index = try self.createGotAtom();
|
|
const got_index = @intCast(u32, self.got_entries.items.len);
|
|
try self.got_entries.append(gpa, .{
|
|
.target = target,
|
|
.atom_index = atom_index,
|
|
});
|
|
try self.got_table.putNoClobber(gpa, target, got_index);
|
|
}
|
|
|
|
fn createDyldPrivateAtom(self: *Zld) !void {
|
|
if (self.dyld_stub_binder_index == null) return;
|
|
|
|
const sym_index = try self.allocateSymbol();
|
|
const atom_index = try self.createEmptyAtom(sym_index, @sizeOf(u64), 3);
|
|
const sym = self.getSymbolPtr(.{ .sym_index = sym_index });
|
|
sym.n_type = macho.N_SECT;
|
|
|
|
const sect_id = self.getSectionByName("__DATA", "__data") orelse try self.initSection("__DATA", "__data", .{});
|
|
sym.n_sect = sect_id + 1;
|
|
|
|
self.dyld_private_sym_index = sym_index;
|
|
|
|
self.addAtomToSection(atom_index);
|
|
}
|
|
|
|
fn createStubHelperPreambleAtom(self: *Zld) !void {
|
|
if (self.dyld_stub_binder_index == null) return;
|
|
|
|
const cpu_arch = self.options.target.cpu.arch;
|
|
const size: u64 = switch (cpu_arch) {
|
|
.x86_64 => 15,
|
|
.aarch64 => 6 * @sizeOf(u32),
|
|
else => unreachable,
|
|
};
|
|
const alignment: u32 = switch (cpu_arch) {
|
|
.x86_64 => 0,
|
|
.aarch64 => 2,
|
|
else => unreachable,
|
|
};
|
|
const sym_index = try self.allocateSymbol();
|
|
const atom_index = try self.createEmptyAtom(sym_index, size, alignment);
|
|
const sym = self.getSymbolPtr(.{ .sym_index = sym_index });
|
|
sym.n_type = macho.N_SECT;
|
|
|
|
const sect_id = self.getSectionByName("__TEXT", "__stub_helper") orelse
|
|
try self.initSection("__TEXT", "__stub_helper", .{
|
|
.flags = macho.S_REGULAR |
|
|
macho.S_ATTR_PURE_INSTRUCTIONS |
|
|
macho.S_ATTR_SOME_INSTRUCTIONS,
|
|
});
|
|
sym.n_sect = sect_id + 1;
|
|
|
|
self.stub_helper_preamble_sym_index = sym_index;
|
|
|
|
self.addAtomToSection(atom_index);
|
|
}
|
|
|
|
fn writeStubHelperPreambleCode(self: *Zld, writer: anytype) !void {
|
|
const cpu_arch = self.options.target.cpu.arch;
|
|
const source_addr = blk: {
|
|
const sym = self.getSymbol(.{ .sym_index = self.stub_helper_preamble_sym_index.? });
|
|
break :blk sym.n_value;
|
|
};
|
|
const dyld_private_addr = blk: {
|
|
const sym = self.getSymbol(.{ .sym_index = self.dyld_private_sym_index.? });
|
|
break :blk sym.n_value;
|
|
};
|
|
const dyld_stub_binder_got_addr = blk: {
|
|
const index = self.got_table.get(.{ .sym_index = self.dyld_stub_binder_index.? }).?;
|
|
const entry = self.got_entries.items[index];
|
|
break :blk entry.getAtomSymbol(self).n_value;
|
|
};
|
|
switch (cpu_arch) {
|
|
.x86_64 => {
|
|
try writer.writeAll(&.{ 0x4c, 0x8d, 0x1d });
|
|
{
|
|
const disp = try Atom.calcPcRelativeDisplacementX86(source_addr + 3, dyld_private_addr, 0);
|
|
try writer.writeIntLittle(i32, disp);
|
|
}
|
|
try writer.writeAll(&.{ 0x41, 0x53, 0xff, 0x25 });
|
|
{
|
|
const disp = try Atom.calcPcRelativeDisplacementX86(source_addr + 11, dyld_stub_binder_got_addr, 0);
|
|
try writer.writeIntLittle(i32, disp);
|
|
}
|
|
},
|
|
.aarch64 => {
|
|
{
|
|
const pages = Atom.calcNumberOfPages(source_addr, dyld_private_addr);
|
|
try writer.writeIntLittle(u32, aarch64.Instruction.adrp(.x17, pages).toU32());
|
|
}
|
|
{
|
|
const off = try Atom.calcPageOffset(dyld_private_addr, .arithmetic);
|
|
try writer.writeIntLittle(u32, aarch64.Instruction.add(.x17, .x17, off, false).toU32());
|
|
}
|
|
try writer.writeIntLittle(u32, aarch64.Instruction.stp(
|
|
.x16,
|
|
.x17,
|
|
aarch64.Register.sp,
|
|
aarch64.Instruction.LoadStorePairOffset.pre_index(-16),
|
|
).toU32());
|
|
{
|
|
const pages = Atom.calcNumberOfPages(source_addr + 12, dyld_stub_binder_got_addr);
|
|
try writer.writeIntLittle(u32, aarch64.Instruction.adrp(.x16, pages).toU32());
|
|
}
|
|
{
|
|
const off = try Atom.calcPageOffset(dyld_stub_binder_got_addr, .load_store_64);
|
|
try writer.writeIntLittle(u32, aarch64.Instruction.ldr(
|
|
.x16,
|
|
.x16,
|
|
aarch64.Instruction.LoadStoreOffset.imm(off),
|
|
).toU32());
|
|
}
|
|
try writer.writeIntLittle(u32, aarch64.Instruction.br(.x16).toU32());
|
|
},
|
|
else => unreachable,
|
|
}
|
|
}
|
|
|
|
pub fn createStubHelperAtom(self: *Zld) !AtomIndex {
|
|
const cpu_arch = self.options.target.cpu.arch;
|
|
const stub_size: u4 = switch (cpu_arch) {
|
|
.x86_64 => 10,
|
|
.aarch64 => 3 * @sizeOf(u32),
|
|
else => unreachable,
|
|
};
|
|
const alignment: u2 = switch (cpu_arch) {
|
|
.x86_64 => 0,
|
|
.aarch64 => 2,
|
|
else => unreachable,
|
|
};
|
|
|
|
const sym_index = try self.allocateSymbol();
|
|
const atom_index = try self.createEmptyAtom(sym_index, stub_size, alignment);
|
|
const sym = self.getSymbolPtr(.{ .sym_index = sym_index });
|
|
sym.n_sect = macho.N_SECT;
|
|
|
|
const sect_id = self.getSectionByName("__TEXT", "__stub_helper").?;
|
|
sym.n_sect = sect_id + 1;
|
|
|
|
self.addAtomToSection(atom_index);
|
|
|
|
return atom_index;
|
|
}
|
|
|
|
fn writeStubHelperCode(self: *Zld, atom_index: AtomIndex, writer: anytype) !void {
|
|
const cpu_arch = self.options.target.cpu.arch;
|
|
const source_addr = blk: {
|
|
const atom = self.getAtom(atom_index);
|
|
const sym = self.getSymbol(atom.getSymbolWithLoc());
|
|
break :blk sym.n_value;
|
|
};
|
|
const target_addr = blk: {
|
|
const sym = self.getSymbol(.{ .sym_index = self.stub_helper_preamble_sym_index.? });
|
|
break :blk sym.n_value;
|
|
};
|
|
switch (cpu_arch) {
|
|
.x86_64 => {
|
|
try writer.writeAll(&.{ 0x68, 0x0, 0x0, 0x0, 0x0, 0xe9 });
|
|
{
|
|
const disp = try Atom.calcPcRelativeDisplacementX86(source_addr + 6, target_addr, 0);
|
|
try writer.writeIntLittle(i32, disp);
|
|
}
|
|
},
|
|
.aarch64 => {
|
|
const stub_size: u4 = 3 * @sizeOf(u32);
|
|
const literal = blk: {
|
|
const div_res = try math.divExact(u64, stub_size - @sizeOf(u32), 4);
|
|
break :blk math.cast(u18, div_res) orelse return error.Overflow;
|
|
};
|
|
try writer.writeIntLittle(u32, aarch64.Instruction.ldrLiteral(
|
|
.w16,
|
|
literal,
|
|
).toU32());
|
|
{
|
|
const disp = try Atom.calcPcRelativeDisplacementArm64(source_addr + 4, target_addr);
|
|
try writer.writeIntLittle(u32, aarch64.Instruction.b(disp).toU32());
|
|
}
|
|
try writer.writeAll(&.{ 0x0, 0x0, 0x0, 0x0 });
|
|
},
|
|
else => unreachable,
|
|
}
|
|
}
|
|
|
|
pub fn createLazyPointerAtom(self: *Zld) !AtomIndex {
|
|
const sym_index = try self.allocateSymbol();
|
|
const atom_index = try self.createEmptyAtom(sym_index, @sizeOf(u64), 3);
|
|
const sym = self.getSymbolPtr(.{ .sym_index = sym_index });
|
|
sym.n_type = macho.N_SECT;
|
|
|
|
const sect_id = self.getSectionByName("__DATA", "__la_symbol_ptr") orelse
|
|
try self.initSection("__DATA", "__la_symbol_ptr", .{
|
|
.flags = macho.S_LAZY_SYMBOL_POINTERS,
|
|
});
|
|
sym.n_sect = sect_id + 1;
|
|
|
|
self.addAtomToSection(atom_index);
|
|
|
|
return atom_index;
|
|
}
|
|
|
|
fn writeLazyPointer(self: *Zld, stub_helper_index: u32, writer: anytype) !void {
|
|
const target_addr = blk: {
|
|
const sect_id = self.getSectionByName("__TEXT", "__stub_helper").?;
|
|
var atom_index = self.sections.items(.first_atom_index)[sect_id];
|
|
var count: u32 = 0;
|
|
while (count < stub_helper_index + 1) : (count += 1) {
|
|
const atom = self.getAtom(atom_index);
|
|
if (atom.next_index) |next_index| {
|
|
atom_index = next_index;
|
|
}
|
|
}
|
|
const atom = self.getAtom(atom_index);
|
|
const sym = self.getSymbol(atom.getSymbolWithLoc());
|
|
break :blk sym.n_value;
|
|
};
|
|
try writer.writeIntLittle(u64, target_addr);
|
|
}
|
|
|
|
pub fn createStubAtom(self: *Zld) !AtomIndex {
|
|
const cpu_arch = self.options.target.cpu.arch;
|
|
const alignment: u2 = switch (cpu_arch) {
|
|
.x86_64 => 0,
|
|
.aarch64 => 2,
|
|
else => unreachable, // unhandled architecture type
|
|
};
|
|
const stub_size: u4 = switch (cpu_arch) {
|
|
.x86_64 => 6,
|
|
.aarch64 => 3 * @sizeOf(u32),
|
|
else => unreachable, // unhandled architecture type
|
|
};
|
|
const sym_index = try self.allocateSymbol();
|
|
const atom_index = try self.createEmptyAtom(sym_index, stub_size, alignment);
|
|
const sym = self.getSymbolPtr(.{ .sym_index = sym_index });
|
|
sym.n_type = macho.N_SECT;
|
|
|
|
const sect_id = self.getSectionByName("__TEXT", "__stubs") orelse
|
|
try self.initSection("__TEXT", "__stubs", .{
|
|
.flags = macho.S_SYMBOL_STUBS |
|
|
macho.S_ATTR_PURE_INSTRUCTIONS |
|
|
macho.S_ATTR_SOME_INSTRUCTIONS,
|
|
.reserved2 = stub_size,
|
|
});
|
|
sym.n_sect = sect_id + 1;
|
|
|
|
self.addAtomToSection(atom_index);
|
|
|
|
return atom_index;
|
|
}
|
|
|
|
fn writeStubCode(self: *Zld, atom_index: AtomIndex, stub_index: u32, writer: anytype) !void {
|
|
const cpu_arch = self.options.target.cpu.arch;
|
|
const source_addr = blk: {
|
|
const atom = self.getAtom(atom_index);
|
|
const sym = self.getSymbol(atom.getSymbolWithLoc());
|
|
break :blk sym.n_value;
|
|
};
|
|
const target_addr = blk: {
|
|
// TODO: cache this at stub atom creation; they always go in pairs anyhow
|
|
const la_sect_id = self.getSectionByName("__DATA", "__la_symbol_ptr").?;
|
|
var la_atom_index = self.sections.items(.first_atom_index)[la_sect_id];
|
|
var count: u32 = 0;
|
|
while (count < stub_index) : (count += 1) {
|
|
const la_atom = self.getAtom(la_atom_index);
|
|
la_atom_index = la_atom.next_index.?;
|
|
}
|
|
const atom = self.getAtom(la_atom_index);
|
|
const sym = self.getSymbol(atom.getSymbolWithLoc());
|
|
break :blk sym.n_value;
|
|
};
|
|
switch (cpu_arch) {
|
|
.x86_64 => {
|
|
try writer.writeAll(&.{ 0xff, 0x25 });
|
|
{
|
|
const disp = try Atom.calcPcRelativeDisplacementX86(source_addr + 2, target_addr, 0);
|
|
try writer.writeIntLittle(i32, disp);
|
|
}
|
|
},
|
|
.aarch64 => {
|
|
{
|
|
const pages = Atom.calcNumberOfPages(source_addr, target_addr);
|
|
try writer.writeIntLittle(u32, aarch64.Instruction.adrp(.x16, pages).toU32());
|
|
}
|
|
{
|
|
const off = try Atom.calcPageOffset(target_addr, .load_store_64);
|
|
try writer.writeIntLittle(u32, aarch64.Instruction.ldr(
|
|
.x16,
|
|
.x16,
|
|
aarch64.Instruction.LoadStoreOffset.imm(off),
|
|
).toU32());
|
|
}
|
|
try writer.writeIntLittle(u32, aarch64.Instruction.br(.x16).toU32());
|
|
},
|
|
else => unreachable,
|
|
}
|
|
}
|
|
|
|
fn createTentativeDefAtoms(self: *Zld) !void {
|
|
const gpa = self.gpa;
|
|
|
|
for (self.globals.items) |global| {
|
|
const sym = self.getSymbolPtr(global);
|
|
if (!sym.tentative()) continue;
|
|
if (sym.n_desc == N_DEAD) continue;
|
|
|
|
log.debug("creating tentative definition for ATOM(%{d}, '{s}') in object({?})", .{
|
|
global.sym_index, self.getSymbolName(global), global.file,
|
|
});
|
|
|
|
// Convert any tentative definition into a regular symbol and allocate
|
|
// text blocks for each tentative definition.
|
|
const size = sym.n_value;
|
|
const alignment = (sym.n_desc >> 8) & 0x0f;
|
|
const n_sect = (try self.getOutputSection(.{
|
|
.segname = makeStaticString("__DATA"),
|
|
.sectname = makeStaticString("__bss"),
|
|
.flags = macho.S_ZEROFILL,
|
|
})).? + 1;
|
|
|
|
sym.* = .{
|
|
.n_strx = sym.n_strx,
|
|
.n_type = macho.N_SECT | macho.N_EXT,
|
|
.n_sect = n_sect,
|
|
.n_desc = 0,
|
|
.n_value = 0,
|
|
};
|
|
|
|
const atom_index = try self.createEmptyAtom(global.sym_index, size, alignment);
|
|
const atom = self.getAtomPtr(atom_index);
|
|
atom.file = global.file;
|
|
|
|
self.addAtomToSection(atom_index);
|
|
|
|
assert(global.getFile() != null);
|
|
const object = &self.objects.items[global.getFile().?];
|
|
try object.atoms.append(gpa, atom_index);
|
|
object.atom_by_index_table[global.sym_index] = atom_index;
|
|
}
|
|
}
|
|
|
|
fn resolveSymbolsInObject(self: *Zld, object_id: u16, resolver: *SymbolResolver) !void {
|
|
const object = &self.objects.items[object_id];
|
|
const in_symtab = object.in_symtab orelse return;
|
|
|
|
log.debug("resolving symbols in '{s}'", .{object.name});
|
|
|
|
var sym_index: u32 = 0;
|
|
while (sym_index < in_symtab.len) : (sym_index += 1) {
|
|
const sym = &object.symtab[sym_index];
|
|
const sym_name = object.getSymbolName(sym_index);
|
|
|
|
if (sym.stab()) {
|
|
log.err("unhandled symbol type: stab", .{});
|
|
log.err(" symbol '{s}'", .{sym_name});
|
|
log.err(" first definition in '{s}'", .{object.name});
|
|
return error.UnhandledSymbolType;
|
|
}
|
|
|
|
if (sym.indr()) {
|
|
log.err("unhandled symbol type: indirect", .{});
|
|
log.err(" symbol '{s}'", .{sym_name});
|
|
log.err(" first definition in '{s}'", .{object.name});
|
|
return error.UnhandledSymbolType;
|
|
}
|
|
|
|
if (sym.abs()) {
|
|
log.err("unhandled symbol type: absolute", .{});
|
|
log.err(" symbol '{s}'", .{sym_name});
|
|
log.err(" first definition in '{s}'", .{object.name});
|
|
return error.UnhandledSymbolType;
|
|
}
|
|
|
|
if (sym.sect() and !sym.ext()) {
|
|
log.debug("symbol '{s}' local to object {s}; skipping...", .{
|
|
sym_name,
|
|
object.name,
|
|
});
|
|
continue;
|
|
}
|
|
|
|
const sym_loc = SymbolWithLoc{ .sym_index = sym_index, .file = object_id };
|
|
|
|
const global_index = resolver.table.get(sym_name) orelse {
|
|
const gpa = self.gpa;
|
|
const global_index = @intCast(u32, self.globals.items.len);
|
|
try self.globals.append(gpa, sym_loc);
|
|
try resolver.table.putNoClobber(sym_name, global_index);
|
|
if (sym.undf() and !sym.tentative()) {
|
|
try resolver.unresolved.putNoClobber(global_index, {});
|
|
}
|
|
continue;
|
|
};
|
|
const global = &self.globals.items[global_index];
|
|
const global_sym = self.getSymbol(global.*);
|
|
|
|
// Cases to consider: sym vs global_sym
|
|
// 1. strong(sym) and strong(global_sym) => error
|
|
// 2. strong(sym) and weak(global_sym) => sym
|
|
// 3. strong(sym) and tentative(global_sym) => sym
|
|
// 4. strong(sym) and undf(global_sym) => sym
|
|
// 5. weak(sym) and strong(global_sym) => global_sym
|
|
// 6. weak(sym) and tentative(global_sym) => sym
|
|
// 7. weak(sym) and undf(global_sym) => sym
|
|
// 8. tentative(sym) and strong(global_sym) => global_sym
|
|
// 9. tentative(sym) and weak(global_sym) => global_sym
|
|
// 10. tentative(sym) and tentative(global_sym) => pick larger
|
|
// 11. tentative(sym) and undf(global_sym) => sym
|
|
// 12. undf(sym) and * => global_sym
|
|
//
|
|
// Reduces to:
|
|
// 1. strong(sym) and strong(global_sym) => error
|
|
// 2. * and strong(global_sym) => global_sym
|
|
// 3. weak(sym) and weak(global_sym) => global_sym
|
|
// 4. tentative(sym) and tentative(global_sym) => pick larger
|
|
// 5. undf(sym) and * => global_sym
|
|
// 6. else => sym
|
|
|
|
const sym_is_strong = sym.sect() and !(sym.weakDef() or sym.pext());
|
|
const global_is_strong = global_sym.sect() and !(global_sym.weakDef() or global_sym.pext());
|
|
const sym_is_weak = sym.sect() and (sym.weakDef() or sym.pext());
|
|
const global_is_weak = global_sym.sect() and (global_sym.weakDef() or global_sym.pext());
|
|
|
|
if (sym_is_strong and global_is_strong) {
|
|
log.err("symbol '{s}' defined multiple times", .{sym_name});
|
|
if (global.getFile()) |file| {
|
|
log.err(" first definition in '{s}'", .{self.objects.items[file].name});
|
|
}
|
|
log.err(" next definition in '{s}'", .{self.objects.items[object_id].name});
|
|
return error.MultipleSymbolDefinitions;
|
|
}
|
|
|
|
const update_global = blk: {
|
|
if (global_is_strong) break :blk false;
|
|
if (sym_is_weak and global_is_weak) break :blk false;
|
|
if (sym.tentative() and global_sym.tentative()) {
|
|
if (global_sym.n_value >= sym.n_value) break :blk false;
|
|
}
|
|
if (sym.undf() and !sym.tentative()) break :blk false;
|
|
break :blk true;
|
|
};
|
|
|
|
if (update_global) {
|
|
const global_object = &self.objects.items[global.getFile().?];
|
|
global_object.globals_lookup[global.sym_index] = global_index;
|
|
_ = resolver.unresolved.swapRemove(resolver.table.get(sym_name).?);
|
|
global.* = sym_loc;
|
|
} else {
|
|
object.globals_lookup[sym_index] = global_index;
|
|
}
|
|
}
|
|
}
|
|
|
|
fn resolveSymbolsInArchives(self: *Zld, resolver: *SymbolResolver) !void {
|
|
if (self.archives.items.len == 0) return;
|
|
|
|
const gpa = self.gpa;
|
|
const cpu_arch = self.options.target.cpu.arch;
|
|
var next_sym: usize = 0;
|
|
loop: while (next_sym < resolver.unresolved.count()) {
|
|
const global = self.globals.items[resolver.unresolved.keys()[next_sym]];
|
|
const sym_name = self.getSymbolName(global);
|
|
|
|
for (self.archives.items) |archive| {
|
|
// Check if the entry exists in a static archive.
|
|
const offsets = archive.toc.get(sym_name) orelse {
|
|
// No hit.
|
|
continue;
|
|
};
|
|
assert(offsets.items.len > 0);
|
|
|
|
const object_id = @intCast(u16, self.objects.items.len);
|
|
const object = try archive.parseObject(gpa, cpu_arch, offsets.items[0]);
|
|
try self.objects.append(gpa, object);
|
|
try self.resolveSymbolsInObject(object_id, resolver);
|
|
|
|
continue :loop;
|
|
}
|
|
|
|
next_sym += 1;
|
|
}
|
|
}
|
|
|
|
fn resolveSymbolsInDylibs(self: *Zld, resolver: *SymbolResolver) !void {
|
|
if (self.dylibs.items.len == 0) return;
|
|
|
|
var next_sym: usize = 0;
|
|
loop: while (next_sym < resolver.unresolved.count()) {
|
|
const global_index = resolver.unresolved.keys()[next_sym];
|
|
const global = self.globals.items[global_index];
|
|
const sym = self.getSymbolPtr(global);
|
|
const sym_name = self.getSymbolName(global);
|
|
|
|
for (self.dylibs.items) |dylib, id| {
|
|
if (!dylib.symbols.contains(sym_name)) continue;
|
|
|
|
const dylib_id = @intCast(u16, id);
|
|
if (!self.referenced_dylibs.contains(dylib_id)) {
|
|
try self.referenced_dylibs.putNoClobber(self.gpa, dylib_id, {});
|
|
}
|
|
|
|
const ordinal = self.referenced_dylibs.getIndex(dylib_id) orelse unreachable;
|
|
sym.n_type |= macho.N_EXT;
|
|
sym.n_desc = @intCast(u16, ordinal + 1) * macho.N_SYMBOL_RESOLVER;
|
|
|
|
if (dylib.weak) {
|
|
sym.n_desc |= macho.N_WEAK_REF;
|
|
}
|
|
|
|
assert(resolver.unresolved.swapRemove(global_index));
|
|
continue :loop;
|
|
}
|
|
|
|
next_sym += 1;
|
|
}
|
|
}
|
|
|
|
fn resolveSymbolsAtLoading(self: *Zld, resolver: *SymbolResolver) !void {
|
|
const is_lib = self.options.output_mode == .Lib;
|
|
const is_dyn_lib = self.options.link_mode == .Dynamic and is_lib;
|
|
const allow_undef = is_dyn_lib and (self.options.allow_shlib_undefined orelse false);
|
|
|
|
var next_sym: usize = 0;
|
|
while (next_sym < resolver.unresolved.count()) {
|
|
const global_index = resolver.unresolved.keys()[next_sym];
|
|
const global = self.globals.items[global_index];
|
|
const sym = self.getSymbolPtr(global);
|
|
const sym_name = self.getSymbolName(global);
|
|
|
|
if (sym.discarded()) {
|
|
sym.* = .{
|
|
.n_strx = 0,
|
|
.n_type = macho.N_UNDF,
|
|
.n_sect = 0,
|
|
.n_desc = 0,
|
|
.n_value = 0,
|
|
};
|
|
_ = resolver.unresolved.swapRemove(global_index);
|
|
continue;
|
|
} else if (allow_undef) {
|
|
const n_desc = @bitCast(
|
|
u16,
|
|
macho.BIND_SPECIAL_DYLIB_FLAT_LOOKUP * @intCast(i16, macho.N_SYMBOL_RESOLVER),
|
|
);
|
|
sym.n_type = macho.N_EXT;
|
|
sym.n_desc = n_desc;
|
|
_ = resolver.unresolved.swapRemove(global_index);
|
|
continue;
|
|
}
|
|
|
|
log.err("undefined reference to symbol '{s}'", .{sym_name});
|
|
if (global.getFile()) |file| {
|
|
log.err(" first referenced in '{s}'", .{self.objects.items[file].name});
|
|
}
|
|
|
|
next_sym += 1;
|
|
}
|
|
}
|
|
|
|
fn createMhExecuteHeaderSymbol(self: *Zld, resolver: *SymbolResolver) !void {
|
|
if (self.options.output_mode != .Exe) return;
|
|
if (resolver.table.get("__mh_execute_header")) |global_index| {
|
|
const global = self.globals.items[global_index];
|
|
const sym = self.getSymbol(global);
|
|
self.mh_execute_header_index = global_index;
|
|
if (!sym.undf() and !(sym.pext() or sym.weakDef())) return;
|
|
}
|
|
|
|
const gpa = self.gpa;
|
|
const sym_index = try self.allocateSymbol();
|
|
const sym_loc = SymbolWithLoc{ .sym_index = sym_index };
|
|
const sym = self.getSymbolPtr(sym_loc);
|
|
sym.n_strx = try self.strtab.insert(gpa, "__mh_execute_header");
|
|
sym.n_type = macho.N_SECT | macho.N_EXT;
|
|
sym.n_desc = macho.REFERENCED_DYNAMICALLY;
|
|
|
|
if (resolver.table.get("__mh_execute_header")) |global_index| {
|
|
const global = &self.globals.items[global_index];
|
|
const global_object = &self.objects.items[global.getFile().?];
|
|
global_object.globals_lookup[global.sym_index] = global_index;
|
|
global.* = sym_loc;
|
|
self.mh_execute_header_index = global_index;
|
|
} else {
|
|
const global_index = @intCast(u32, self.globals.items.len);
|
|
try self.globals.append(gpa, sym_loc);
|
|
self.mh_execute_header_index = global_index;
|
|
}
|
|
}
|
|
|
|
fn createDsoHandleSymbol(self: *Zld, resolver: *SymbolResolver) !void {
|
|
const global_index = resolver.table.get("___dso_handle") orelse return;
|
|
const global = &self.globals.items[global_index];
|
|
self.dso_handle_index = global_index;
|
|
if (!self.getSymbol(global.*).undf()) return;
|
|
|
|
const gpa = self.gpa;
|
|
const sym_index = try self.allocateSymbol();
|
|
const sym_loc = SymbolWithLoc{ .sym_index = sym_index };
|
|
const sym = self.getSymbolPtr(sym_loc);
|
|
sym.n_strx = try self.strtab.insert(gpa, "___dso_handle");
|
|
sym.n_type = macho.N_SECT | macho.N_EXT;
|
|
sym.n_desc = macho.N_WEAK_DEF;
|
|
|
|
const global_object = &self.objects.items[global.getFile().?];
|
|
global_object.globals_lookup[global.sym_index] = global_index;
|
|
_ = resolver.unresolved.swapRemove(resolver.table.get("___dso_handle").?);
|
|
global.* = sym_loc;
|
|
}
|
|
|
|
fn resolveDyldStubBinder(self: *Zld, resolver: *SymbolResolver) !void {
|
|
if (self.dyld_stub_binder_index != null) return;
|
|
if (resolver.unresolved.count() == 0) return; // no need for a stub binder if we don't have any imports
|
|
|
|
const gpa = self.gpa;
|
|
const sym_name = "dyld_stub_binder";
|
|
const sym_index = try self.allocateSymbol();
|
|
const sym_loc = SymbolWithLoc{ .sym_index = sym_index };
|
|
const sym = self.getSymbolPtr(sym_loc);
|
|
sym.n_strx = try self.strtab.insert(gpa, sym_name);
|
|
sym.n_type = macho.N_UNDF;
|
|
|
|
const global = SymbolWithLoc{ .sym_index = sym_index };
|
|
try self.globals.append(gpa, global);
|
|
|
|
for (self.dylibs.items) |dylib, id| {
|
|
if (!dylib.symbols.contains(sym_name)) continue;
|
|
|
|
const dylib_id = @intCast(u16, id);
|
|
if (!self.referenced_dylibs.contains(dylib_id)) {
|
|
try self.referenced_dylibs.putNoClobber(gpa, dylib_id, {});
|
|
}
|
|
|
|
const ordinal = self.referenced_dylibs.getIndex(dylib_id) orelse unreachable;
|
|
sym.n_type |= macho.N_EXT;
|
|
sym.n_desc = @intCast(u16, ordinal + 1) * macho.N_SYMBOL_RESOLVER;
|
|
self.dyld_stub_binder_index = sym_index;
|
|
|
|
break;
|
|
}
|
|
|
|
if (self.dyld_stub_binder_index == null) {
|
|
log.err("undefined reference to symbol '{s}'", .{sym_name});
|
|
return error.UndefinedSymbolReference;
|
|
}
|
|
}
|
|
|
|
pub fn deinit(self: *Zld) void {
|
|
const gpa = self.gpa;
|
|
|
|
self.tlv_ptr_entries.deinit(gpa);
|
|
self.tlv_ptr_table.deinit(gpa);
|
|
self.got_entries.deinit(gpa);
|
|
self.got_table.deinit(gpa);
|
|
self.stubs.deinit(gpa);
|
|
self.stubs_table.deinit(gpa);
|
|
self.thunk_table.deinit(gpa);
|
|
|
|
for (self.thunks.items) |*thunk| {
|
|
thunk.deinit(gpa);
|
|
}
|
|
self.thunks.deinit(gpa);
|
|
|
|
self.strtab.deinit(gpa);
|
|
self.locals.deinit(gpa);
|
|
self.globals.deinit(gpa);
|
|
|
|
for (self.objects.items) |*object| {
|
|
object.deinit(gpa);
|
|
}
|
|
self.objects.deinit(gpa);
|
|
for (self.archives.items) |*archive| {
|
|
archive.deinit(gpa);
|
|
}
|
|
self.archives.deinit(gpa);
|
|
for (self.dylibs.items) |*dylib| {
|
|
dylib.deinit(gpa);
|
|
}
|
|
self.dylibs.deinit(gpa);
|
|
self.dylibs_map.deinit(gpa);
|
|
self.referenced_dylibs.deinit(gpa);
|
|
|
|
self.segments.deinit(gpa);
|
|
self.sections.deinit(gpa);
|
|
self.atoms.deinit(gpa);
|
|
}
|
|
|
|
fn createSegments(self: *Zld) !void {
|
|
const pagezero_vmsize = self.options.pagezero_size orelse MachO.default_pagezero_vmsize;
|
|
const aligned_pagezero_vmsize = mem.alignBackwardGeneric(u64, pagezero_vmsize, self.page_size);
|
|
if (self.options.output_mode != .Lib and aligned_pagezero_vmsize > 0) {
|
|
if (aligned_pagezero_vmsize != pagezero_vmsize) {
|
|
log.warn("requested __PAGEZERO size (0x{x}) is not page aligned", .{pagezero_vmsize});
|
|
log.warn(" rounding down to 0x{x}", .{aligned_pagezero_vmsize});
|
|
}
|
|
try self.segments.append(self.gpa, .{
|
|
.cmdsize = @sizeOf(macho.segment_command_64),
|
|
.segname = makeStaticString("__PAGEZERO"),
|
|
.vmsize = aligned_pagezero_vmsize,
|
|
});
|
|
}
|
|
|
|
// __TEXT segment is non-optional
|
|
{
|
|
const protection = getSegmentMemoryProtection("__TEXT");
|
|
try self.segments.append(self.gpa, .{
|
|
.cmdsize = @sizeOf(macho.segment_command_64),
|
|
.segname = makeStaticString("__TEXT"),
|
|
.maxprot = protection,
|
|
.initprot = protection,
|
|
});
|
|
}
|
|
|
|
for (self.sections.items(.header)) |header, sect_id| {
|
|
if (header.size == 0) continue; // empty section
|
|
|
|
const segname = header.segName();
|
|
const segment_id = self.getSegmentByName(segname) orelse blk: {
|
|
log.debug("creating segment '{s}'", .{segname});
|
|
const segment_id = @intCast(u8, self.segments.items.len);
|
|
const protection = getSegmentMemoryProtection(segname);
|
|
try self.segments.append(self.gpa, .{
|
|
.cmdsize = @sizeOf(macho.segment_command_64),
|
|
.segname = makeStaticString(segname),
|
|
.maxprot = protection,
|
|
.initprot = protection,
|
|
});
|
|
break :blk segment_id;
|
|
};
|
|
const segment = &self.segments.items[segment_id];
|
|
segment.cmdsize += @sizeOf(macho.section_64);
|
|
segment.nsects += 1;
|
|
self.sections.items(.segment_index)[sect_id] = segment_id;
|
|
}
|
|
|
|
// __LINKEDIT always comes last
|
|
{
|
|
const protection = getSegmentMemoryProtection("__LINKEDIT");
|
|
try self.segments.append(self.gpa, .{
|
|
.cmdsize = @sizeOf(macho.segment_command_64),
|
|
.segname = makeStaticString("__LINKEDIT"),
|
|
.maxprot = protection,
|
|
.initprot = protection,
|
|
});
|
|
}
|
|
}
|
|
|
|
pub fn allocateSymbol(self: *Zld) !u32 {
|
|
try self.locals.ensureUnusedCapacity(self.gpa, 1);
|
|
log.debug(" (allocating symbol index {d})", .{self.locals.items.len});
|
|
const index = @intCast(u32, self.locals.items.len);
|
|
_ = self.locals.addOneAssumeCapacity();
|
|
self.locals.items[index] = .{
|
|
.n_strx = 0,
|
|
.n_type = 0,
|
|
.n_sect = 0,
|
|
.n_desc = 0,
|
|
.n_value = 0,
|
|
};
|
|
return index;
|
|
}
|
|
|
|
fn allocateSpecialSymbols(self: *Zld) !void {
|
|
for (&[_]?u32{
|
|
self.dso_handle_index,
|
|
self.mh_execute_header_index,
|
|
}) |maybe_index| {
|
|
const global_index = maybe_index orelse continue;
|
|
const global = self.globals.items[global_index];
|
|
if (global.getFile() != null) continue;
|
|
const name = self.getSymbolName(global);
|
|
const sym = self.getSymbolPtr(global);
|
|
const segment_index = self.getSegmentByName("__TEXT").?;
|
|
const seg = self.segments.items[segment_index];
|
|
sym.n_sect = 1;
|
|
sym.n_value = seg.vmaddr;
|
|
log.debug("allocating {s} at the start of {s}", .{
|
|
name,
|
|
seg.segName(),
|
|
});
|
|
}
|
|
}
|
|
|
|
fn writeAtoms(self: *Zld, reverse_lookups: [][]u32) !void {
|
|
const gpa = self.gpa;
|
|
const slice = self.sections.slice();
|
|
|
|
for (slice.items(.first_atom_index)) |first_atom_index, sect_id| {
|
|
const header = slice.items(.header)[sect_id];
|
|
var atom_index = first_atom_index;
|
|
|
|
if (header.isZerofill()) continue;
|
|
|
|
var buffer = std.ArrayList(u8).init(gpa);
|
|
defer buffer.deinit();
|
|
try buffer.ensureTotalCapacity(math.cast(usize, header.size) orelse return error.Overflow);
|
|
|
|
log.debug("writing atoms in {s},{s}", .{ header.segName(), header.sectName() });
|
|
|
|
var count: u32 = 0;
|
|
while (true) : (count += 1) {
|
|
const atom = self.getAtom(atom_index);
|
|
const this_sym = self.getSymbol(atom.getSymbolWithLoc());
|
|
const padding_size: usize = if (atom.next_index) |next_index| blk: {
|
|
const next_sym = self.getSymbol(self.getAtom(next_index).getSymbolWithLoc());
|
|
const size = next_sym.n_value - (this_sym.n_value + atom.size);
|
|
break :blk math.cast(usize, size) orelse return error.Overflow;
|
|
} else 0;
|
|
|
|
log.debug(" (adding ATOM(%{d}, '{s}') from object({?}) to buffer)", .{
|
|
atom.sym_index,
|
|
self.getSymbolName(atom.getSymbolWithLoc()),
|
|
atom.file,
|
|
});
|
|
if (padding_size > 0) {
|
|
log.debug(" (with padding {x})", .{padding_size});
|
|
}
|
|
|
|
const offset = buffer.items.len;
|
|
|
|
// TODO: move writing synthetic sections into a separate function
|
|
if (atom.getFile() == null) outer: {
|
|
if (self.dyld_private_sym_index) |sym_index| {
|
|
if (atom.sym_index == sym_index) {
|
|
buffer.appendSliceAssumeCapacity(&[_]u8{0} ** @sizeOf(u64));
|
|
break :outer;
|
|
}
|
|
}
|
|
switch (header.type()) {
|
|
macho.S_NON_LAZY_SYMBOL_POINTERS => {
|
|
try self.writeGotPointer(count, buffer.writer());
|
|
},
|
|
macho.S_LAZY_SYMBOL_POINTERS => {
|
|
try self.writeLazyPointer(count, buffer.writer());
|
|
},
|
|
macho.S_THREAD_LOCAL_VARIABLE_POINTERS => {
|
|
buffer.appendSliceAssumeCapacity(&[_]u8{0} ** @sizeOf(u64));
|
|
},
|
|
else => {
|
|
if (self.stub_helper_preamble_sym_index) |sym_index| {
|
|
if (sym_index == atom.sym_index) {
|
|
try self.writeStubHelperPreambleCode(buffer.writer());
|
|
break :outer;
|
|
}
|
|
}
|
|
if (header.type() == macho.S_SYMBOL_STUBS) {
|
|
try self.writeStubCode(atom_index, count, buffer.writer());
|
|
} else if (mem.eql(u8, header.sectName(), "__stub_helper")) {
|
|
try self.writeStubHelperCode(atom_index, buffer.writer());
|
|
} else if (header.isCode()) {
|
|
// A thunk
|
|
try thunks.writeThunkCode(self, atom_index, buffer.writer());
|
|
} else unreachable;
|
|
},
|
|
}
|
|
} else {
|
|
const code = Atom.getAtomCode(self, atom_index);
|
|
const relocs = Atom.getAtomRelocs(self, atom_index);
|
|
const size = math.cast(usize, atom.size) orelse return error.Overflow;
|
|
buffer.appendSliceAssumeCapacity(code);
|
|
try Atom.resolveRelocs(
|
|
self,
|
|
atom_index,
|
|
buffer.items[offset..][0..size],
|
|
relocs,
|
|
reverse_lookups[atom.getFile().?],
|
|
);
|
|
}
|
|
|
|
var i: usize = 0;
|
|
while (i < padding_size) : (i += 1) {
|
|
// TODO with NOPs
|
|
buffer.appendAssumeCapacity(0);
|
|
}
|
|
|
|
if (atom.next_index) |next_index| {
|
|
atom_index = next_index;
|
|
} else {
|
|
assert(buffer.items.len == header.size);
|
|
log.debug(" (writing at file offset 0x{x})", .{header.offset});
|
|
try self.file.pwriteAll(buffer.items, header.offset);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
fn pruneAndSortSections(self: *Zld) !void {
|
|
const gpa = self.gpa;
|
|
|
|
const SortSection = struct {
|
|
pub fn lessThan(_: void, lhs: Section, rhs: Section) bool {
|
|
return getSectionPrecedence(lhs.header) < getSectionPrecedence(rhs.header);
|
|
}
|
|
};
|
|
|
|
const slice = self.sections.slice();
|
|
var sections = std.ArrayList(Section).init(gpa);
|
|
defer sections.deinit();
|
|
try sections.ensureTotalCapacity(slice.len);
|
|
|
|
{
|
|
var i: u8 = 0;
|
|
while (i < slice.len) : (i += 1) {
|
|
const section = self.sections.get(i);
|
|
if (section.header.size == 0) {
|
|
log.debug("pruning section {s},{s}", .{
|
|
section.header.segName(),
|
|
section.header.sectName(),
|
|
});
|
|
continue;
|
|
}
|
|
sections.appendAssumeCapacity(section);
|
|
}
|
|
}
|
|
|
|
std.sort.sort(Section, sections.items, {}, SortSection.lessThan);
|
|
|
|
self.sections.shrinkRetainingCapacity(0);
|
|
for (sections.items) |out| {
|
|
self.sections.appendAssumeCapacity(out);
|
|
}
|
|
}
|
|
|
|
fn calcSectionSizes(self: *Zld, reverse_lookups: [][]u32) !void {
|
|
const slice = self.sections.slice();
|
|
for (slice.items(.header)) |*header, sect_id| {
|
|
if (header.size == 0) continue;
|
|
if (self.requiresThunks()) {
|
|
if (header.isCode() and !(header.type() == macho.S_SYMBOL_STUBS) and !mem.eql(u8, header.sectName(), "__stub_helper")) continue;
|
|
}
|
|
|
|
var atom_index = slice.items(.first_atom_index)[sect_id];
|
|
header.size = 0;
|
|
header.@"align" = 0;
|
|
|
|
while (true) {
|
|
const atom = self.getAtom(atom_index);
|
|
const atom_alignment = try math.powi(u32, 2, atom.alignment);
|
|
const atom_offset = mem.alignForwardGeneric(u64, header.size, atom_alignment);
|
|
const padding = atom_offset - header.size;
|
|
|
|
const sym = self.getSymbolPtr(atom.getSymbolWithLoc());
|
|
sym.n_value = atom_offset;
|
|
|
|
header.size += padding + atom.size;
|
|
header.@"align" = @max(header.@"align", atom.alignment);
|
|
|
|
if (atom.next_index) |next_index| {
|
|
atom_index = next_index;
|
|
} else break;
|
|
}
|
|
}
|
|
|
|
if (self.requiresThunks()) {
|
|
for (slice.items(.header)) |header, sect_id| {
|
|
if (!header.isCode()) continue;
|
|
if (header.type() == macho.S_SYMBOL_STUBS) continue;
|
|
if (mem.eql(u8, header.sectName(), "__stub_helper")) continue;
|
|
|
|
// Create jump/branch range extenders if needed.
|
|
try thunks.createThunks(self, @intCast(u8, sect_id), reverse_lookups);
|
|
}
|
|
}
|
|
}
|
|
|
|
fn allocateSegments(self: *Zld) !void {
|
|
for (self.segments.items) |*segment, segment_index| {
|
|
const is_text_segment = mem.eql(u8, segment.segName(), "__TEXT");
|
|
const base_size = if (is_text_segment) try load_commands.calcMinHeaderPad(self.gpa, self.options, .{
|
|
.segments = self.segments.items,
|
|
.dylibs = self.dylibs.items,
|
|
.referenced_dylibs = self.referenced_dylibs.keys(),
|
|
}) else 0;
|
|
try self.allocateSegment(@intCast(u8, segment_index), base_size);
|
|
}
|
|
}
|
|
|
|
fn getSegmentAllocBase(self: Zld, segment_index: u8) struct { vmaddr: u64, fileoff: u64 } {
|
|
if (segment_index > 0) {
|
|
const prev_segment = self.segments.items[segment_index - 1];
|
|
return .{
|
|
.vmaddr = prev_segment.vmaddr + prev_segment.vmsize,
|
|
.fileoff = prev_segment.fileoff + prev_segment.filesize,
|
|
};
|
|
}
|
|
return .{ .vmaddr = 0, .fileoff = 0 };
|
|
}
|
|
|
|
fn allocateSegment(self: *Zld, segment_index: u8, init_size: u64) !void {
|
|
const segment = &self.segments.items[segment_index];
|
|
|
|
if (mem.eql(u8, segment.segName(), "__PAGEZERO")) return; // allocated upon creation
|
|
|
|
const base = self.getSegmentAllocBase(segment_index);
|
|
segment.vmaddr = base.vmaddr;
|
|
segment.fileoff = base.fileoff;
|
|
segment.filesize = init_size;
|
|
segment.vmsize = init_size;
|
|
|
|
// Allocate the sections according to their alignment at the beginning of the segment.
|
|
const indexes = self.getSectionIndexes(segment_index);
|
|
var start = init_size;
|
|
|
|
const slice = self.sections.slice();
|
|
for (slice.items(.header)[indexes.start..indexes.end]) |*header, sect_id| {
|
|
var atom_index = slice.items(.first_atom_index)[indexes.start + sect_id];
|
|
|
|
const alignment = try math.powi(u32, 2, header.@"align");
|
|
const start_aligned = mem.alignForwardGeneric(u64, start, alignment);
|
|
const n_sect = @intCast(u8, indexes.start + sect_id + 1);
|
|
|
|
header.offset = if (header.isZerofill())
|
|
0
|
|
else
|
|
@intCast(u32, segment.fileoff + start_aligned);
|
|
header.addr = segment.vmaddr + start_aligned;
|
|
|
|
log.debug("allocating local symbols in sect({d}, '{s},{s}')", .{
|
|
n_sect,
|
|
header.segName(),
|
|
header.sectName(),
|
|
});
|
|
|
|
while (true) {
|
|
const atom = self.getAtom(atom_index);
|
|
const sym = self.getSymbolPtr(atom.getSymbolWithLoc());
|
|
sym.n_value += header.addr;
|
|
sym.n_sect = n_sect;
|
|
|
|
log.debug(" ATOM(%{d}, '{s}') @{x}", .{
|
|
atom.sym_index,
|
|
self.getSymbolName(atom.getSymbolWithLoc()),
|
|
sym.n_value,
|
|
});
|
|
|
|
if (atom.getFile() != null) {
|
|
// Update each symbol contained within the atom
|
|
var it = Atom.getInnerSymbolsIterator(self, atom_index);
|
|
while (it.next()) |sym_loc| {
|
|
const inner_sym = self.getSymbolPtr(sym_loc);
|
|
inner_sym.n_value = sym.n_value + Atom.calcInnerSymbolOffset(
|
|
self,
|
|
atom_index,
|
|
sym_loc.sym_index,
|
|
);
|
|
inner_sym.n_sect = n_sect;
|
|
}
|
|
|
|
// If there is a section alias, update it now too
|
|
if (Atom.getSectionAlias(self, atom_index)) |sym_loc| {
|
|
const alias = self.getSymbolPtr(sym_loc);
|
|
alias.n_value = sym.n_value;
|
|
alias.n_sect = n_sect;
|
|
}
|
|
}
|
|
|
|
if (atom.next_index) |next_index| {
|
|
atom_index = next_index;
|
|
} else break;
|
|
}
|
|
|
|
start = start_aligned + header.size;
|
|
|
|
if (!header.isZerofill()) {
|
|
segment.filesize = start;
|
|
}
|
|
segment.vmsize = start;
|
|
}
|
|
|
|
segment.filesize = mem.alignForwardGeneric(u64, segment.filesize, self.page_size);
|
|
segment.vmsize = mem.alignForwardGeneric(u64, segment.vmsize, self.page_size);
|
|
}
|
|
|
|
const InitSectionOpts = struct {
|
|
flags: u32 = macho.S_REGULAR,
|
|
reserved1: u32 = 0,
|
|
reserved2: u32 = 0,
|
|
};
|
|
|
|
fn initSection(
|
|
self: *Zld,
|
|
segname: []const u8,
|
|
sectname: []const u8,
|
|
opts: InitSectionOpts,
|
|
) !u8 {
|
|
const gpa = self.gpa;
|
|
log.debug("creating section '{s},{s}'", .{ segname, sectname });
|
|
const index = @intCast(u8, self.sections.slice().len);
|
|
try self.sections.append(gpa, .{
|
|
.segment_index = undefined,
|
|
.header = .{
|
|
.sectname = makeStaticString(sectname),
|
|
.segname = makeStaticString(segname),
|
|
.flags = opts.flags,
|
|
.reserved1 = opts.reserved1,
|
|
.reserved2 = opts.reserved2,
|
|
},
|
|
.first_atom_index = undefined,
|
|
.last_atom_index = undefined,
|
|
});
|
|
return index;
|
|
}
|
|
|
|
inline fn getSegmentPrecedence(segname: []const u8) u4 {
|
|
if (mem.eql(u8, segname, "__PAGEZERO")) return 0x0;
|
|
if (mem.eql(u8, segname, "__TEXT")) return 0x1;
|
|
if (mem.eql(u8, segname, "__DATA_CONST")) return 0x2;
|
|
if (mem.eql(u8, segname, "__DATA")) return 0x3;
|
|
if (mem.eql(u8, segname, "__LINKEDIT")) return 0x5;
|
|
return 0x4;
|
|
}
|
|
|
|
inline fn getSegmentMemoryProtection(segname: []const u8) macho.vm_prot_t {
|
|
if (mem.eql(u8, segname, "__PAGEZERO")) return macho.PROT.NONE;
|
|
if (mem.eql(u8, segname, "__TEXT")) return macho.PROT.READ | macho.PROT.EXEC;
|
|
if (mem.eql(u8, segname, "__LINKEDIT")) return macho.PROT.READ;
|
|
return macho.PROT.READ | macho.PROT.WRITE;
|
|
}
|
|
|
|
inline fn getSectionPrecedence(header: macho.section_64) u8 {
|
|
const segment_precedence: u4 = getSegmentPrecedence(header.segName());
|
|
const section_precedence: u4 = blk: {
|
|
if (header.isCode()) {
|
|
if (mem.eql(u8, "__text", header.sectName())) break :blk 0x0;
|
|
if (header.type() == macho.S_SYMBOL_STUBS) break :blk 0x1;
|
|
break :blk 0x2;
|
|
}
|
|
switch (header.type()) {
|
|
macho.S_NON_LAZY_SYMBOL_POINTERS,
|
|
macho.S_LAZY_SYMBOL_POINTERS,
|
|
=> break :blk 0x0,
|
|
macho.S_MOD_INIT_FUNC_POINTERS => break :blk 0x1,
|
|
macho.S_MOD_TERM_FUNC_POINTERS => break :blk 0x2,
|
|
macho.S_ZEROFILL => break :blk 0xf,
|
|
macho.S_THREAD_LOCAL_REGULAR => break :blk 0xd,
|
|
macho.S_THREAD_LOCAL_ZEROFILL => break :blk 0xe,
|
|
else => if (mem.eql(u8, "__eh_frame", header.sectName()))
|
|
break :blk 0xf
|
|
else
|
|
break :blk 0x3,
|
|
}
|
|
};
|
|
return (@intCast(u8, segment_precedence) << 4) + section_precedence;
|
|
}
|
|
|
|
fn writeSegmentHeaders(self: *Zld, ncmds: *u32, writer: anytype) !void {
|
|
for (self.segments.items) |seg, i| {
|
|
const indexes = self.getSectionIndexes(@intCast(u8, i));
|
|
var out_seg = seg;
|
|
out_seg.cmdsize = @sizeOf(macho.segment_command_64);
|
|
out_seg.nsects = 0;
|
|
|
|
// Update section headers count; any section with size of 0 is excluded
|
|
// since it doesn't have any data in the final binary file.
|
|
for (self.sections.items(.header)[indexes.start..indexes.end]) |header| {
|
|
if (header.size == 0) continue;
|
|
out_seg.cmdsize += @sizeOf(macho.section_64);
|
|
out_seg.nsects += 1;
|
|
}
|
|
|
|
if (out_seg.nsects == 0 and
|
|
(mem.eql(u8, out_seg.segName(), "__DATA_CONST") or
|
|
mem.eql(u8, out_seg.segName(), "__DATA"))) continue;
|
|
|
|
try writer.writeStruct(out_seg);
|
|
for (self.sections.items(.header)[indexes.start..indexes.end]) |header| {
|
|
if (header.size == 0) continue;
|
|
try writer.writeStruct(header);
|
|
}
|
|
|
|
ncmds.* += 1;
|
|
}
|
|
}
|
|
|
|
fn writeLinkeditSegmentData(self: *Zld, ncmds: *u32, lc_writer: anytype, reverse_lookups: [][]u32) !void {
|
|
try self.writeDyldInfoData(ncmds, lc_writer, reverse_lookups);
|
|
try self.writeFunctionStarts(ncmds, lc_writer);
|
|
try self.writeDataInCode(ncmds, lc_writer);
|
|
try self.writeSymtabs(ncmds, lc_writer);
|
|
|
|
const seg = self.getLinkeditSegmentPtr();
|
|
seg.vmsize = mem.alignForwardGeneric(u64, seg.filesize, self.page_size);
|
|
}
|
|
|
|
fn collectRebaseDataFromContainer(
|
|
self: *Zld,
|
|
sect_id: u8,
|
|
pointers: *std.ArrayList(bind.Pointer),
|
|
container: anytype,
|
|
) !void {
|
|
const slice = self.sections.slice();
|
|
const segment_index = slice.items(.segment_index)[sect_id];
|
|
const seg = self.getSegment(sect_id);
|
|
|
|
try pointers.ensureUnusedCapacity(container.items.len);
|
|
|
|
for (container.items) |entry| {
|
|
const target_sym = entry.getTargetSymbol(self);
|
|
if (target_sym.undf()) continue;
|
|
|
|
const atom_sym = entry.getAtomSymbol(self);
|
|
const base_offset = atom_sym.n_value - seg.vmaddr;
|
|
|
|
log.debug(" | rebase at {x}", .{base_offset});
|
|
|
|
pointers.appendAssumeCapacity(.{
|
|
.offset = base_offset,
|
|
.segment_id = segment_index,
|
|
});
|
|
}
|
|
}
|
|
|
|
fn collectRebaseData(self: *Zld, pointers: *std.ArrayList(bind.Pointer)) !void {
|
|
log.debug("collecting rebase data", .{});
|
|
|
|
// First, unpack GOT entries
|
|
if (self.getSectionByName("__DATA_CONST", "__got")) |sect_id| {
|
|
try self.collectRebaseDataFromContainer(sect_id, pointers, self.got_entries);
|
|
}
|
|
|
|
const slice = self.sections.slice();
|
|
|
|
// Next, unpact lazy pointers
|
|
// TODO: save la_ptr in a container so that we can re-use the helper
|
|
if (self.getSectionByName("__DATA", "__la_symbol_ptr")) |sect_id| {
|
|
const segment_index = slice.items(.segment_index)[sect_id];
|
|
const seg = self.getSegment(sect_id);
|
|
var atom_index = slice.items(.first_atom_index)[sect_id];
|
|
|
|
try pointers.ensureUnusedCapacity(self.stubs.items.len);
|
|
|
|
while (true) {
|
|
const atom = self.getAtom(atom_index);
|
|
const sym = self.getSymbol(atom.getSymbolWithLoc());
|
|
const base_offset = sym.n_value - seg.vmaddr;
|
|
|
|
log.debug(" | rebase at {x}", .{base_offset});
|
|
|
|
pointers.appendAssumeCapacity(.{
|
|
.offset = base_offset,
|
|
.segment_id = segment_index,
|
|
});
|
|
|
|
if (atom.next_index) |next_index| {
|
|
atom_index = next_index;
|
|
} else break;
|
|
}
|
|
}
|
|
|
|
// Finally, unpack the rest.
|
|
for (slice.items(.header)) |header, sect_id| {
|
|
switch (header.type()) {
|
|
macho.S_LITERAL_POINTERS,
|
|
macho.S_REGULAR,
|
|
macho.S_MOD_INIT_FUNC_POINTERS,
|
|
macho.S_MOD_TERM_FUNC_POINTERS,
|
|
=> {},
|
|
else => continue,
|
|
}
|
|
|
|
const segment_index = slice.items(.segment_index)[sect_id];
|
|
const segment = self.getSegment(@intCast(u8, sect_id));
|
|
if (segment.maxprot & macho.PROT.WRITE == 0) continue;
|
|
|
|
log.debug("{s},{s}", .{ header.segName(), header.sectName() });
|
|
|
|
const cpu_arch = self.options.target.cpu.arch;
|
|
var atom_index = slice.items(.first_atom_index)[sect_id];
|
|
|
|
while (true) {
|
|
const atom = self.getAtom(atom_index);
|
|
const sym = self.getSymbol(atom.getSymbolWithLoc());
|
|
|
|
const should_rebase = blk: {
|
|
if (self.dyld_private_sym_index) |sym_index| {
|
|
if (atom.getFile() == null and atom.sym_index == sym_index) break :blk false;
|
|
}
|
|
break :blk !sym.undf();
|
|
};
|
|
|
|
if (should_rebase) {
|
|
log.debug(" ATOM({d}, %{d}, '{s}')", .{ atom_index, atom.sym_index, self.getSymbolName(atom.getSymbolWithLoc()) });
|
|
|
|
const object = self.objects.items[atom.getFile().?];
|
|
const base_rel_offset: i32 = blk: {
|
|
const source_sym = object.getSourceSymbol(atom.sym_index) orelse break :blk 0;
|
|
const source_sect = object.getSourceSection(source_sym.n_sect - 1);
|
|
break :blk @intCast(i32, source_sym.n_value - source_sect.addr);
|
|
};
|
|
const relocs = Atom.getAtomRelocs(self, atom_index);
|
|
|
|
for (relocs) |rel| {
|
|
switch (cpu_arch) {
|
|
.aarch64 => {
|
|
const rel_type = @intToEnum(macho.reloc_type_arm64, rel.r_type);
|
|
if (rel_type != .ARM64_RELOC_UNSIGNED) continue;
|
|
if (rel.r_length != 3) continue;
|
|
},
|
|
.x86_64 => {
|
|
const rel_type = @intToEnum(macho.reloc_type_x86_64, rel.r_type);
|
|
if (rel_type != .X86_64_RELOC_UNSIGNED) continue;
|
|
if (rel.r_length != 3) continue;
|
|
},
|
|
else => unreachable,
|
|
}
|
|
|
|
const base_offset = @intCast(i32, sym.n_value - segment.vmaddr);
|
|
const rel_offset = rel.r_address - base_rel_offset;
|
|
const offset = @intCast(u64, base_offset + rel_offset);
|
|
log.debug(" | rebase at {x}", .{offset});
|
|
|
|
try pointers.append(.{
|
|
.offset = offset,
|
|
.segment_id = segment_index,
|
|
});
|
|
}
|
|
}
|
|
|
|
if (atom.next_index) |next_index| {
|
|
atom_index = next_index;
|
|
} else break;
|
|
}
|
|
}
|
|
}
|
|
|
|
fn collectBindDataFromContainer(
|
|
self: *Zld,
|
|
sect_id: u8,
|
|
pointers: *std.ArrayList(bind.Pointer),
|
|
container: anytype,
|
|
) !void {
|
|
const slice = self.sections.slice();
|
|
const segment_index = slice.items(.segment_index)[sect_id];
|
|
const seg = self.getSegment(sect_id);
|
|
|
|
try pointers.ensureUnusedCapacity(container.items.len);
|
|
|
|
for (container.items) |entry| {
|
|
const bind_sym_name = entry.getTargetSymbolName(self);
|
|
const bind_sym = entry.getTargetSymbol(self);
|
|
if (bind_sym.sect()) continue;
|
|
|
|
const sym = entry.getAtomSymbol(self);
|
|
const base_offset = sym.n_value - seg.vmaddr;
|
|
|
|
const dylib_ordinal = @divTrunc(@bitCast(i16, bind_sym.n_desc), macho.N_SYMBOL_RESOLVER);
|
|
var flags: u4 = 0;
|
|
log.debug(" | bind at {x}, import('{s}') in dylib({d})", .{
|
|
base_offset,
|
|
bind_sym_name,
|
|
dylib_ordinal,
|
|
});
|
|
if (bind_sym.weakRef()) {
|
|
log.debug(" | marking as weak ref ", .{});
|
|
flags |= @truncate(u4, macho.BIND_SYMBOL_FLAGS_WEAK_IMPORT);
|
|
}
|
|
pointers.appendAssumeCapacity(.{
|
|
.offset = base_offset,
|
|
.segment_id = segment_index,
|
|
.dylib_ordinal = dylib_ordinal,
|
|
.name = bind_sym_name,
|
|
.bind_flags = flags,
|
|
});
|
|
}
|
|
}
|
|
|
|
fn collectBindData(self: *Zld, pointers: *std.ArrayList(bind.Pointer), reverse_lookups: [][]u32) !void {
|
|
log.debug("collecting bind data", .{});
|
|
|
|
// First, unpack GOT section
|
|
if (self.getSectionByName("__DATA_CONST", "__got")) |sect_id| {
|
|
try self.collectBindDataFromContainer(sect_id, pointers, self.got_entries);
|
|
}
|
|
|
|
// Next, unpack TLV pointers section
|
|
if (self.getSectionByName("__DATA", "__thread_ptrs")) |sect_id| {
|
|
try self.collectBindDataFromContainer(sect_id, pointers, self.tlv_ptr_entries);
|
|
}
|
|
|
|
// Finally, unpack the rest.
|
|
const slice = self.sections.slice();
|
|
for (slice.items(.header)) |header, sect_id| {
|
|
switch (header.type()) {
|
|
macho.S_LITERAL_POINTERS,
|
|
macho.S_REGULAR,
|
|
macho.S_MOD_INIT_FUNC_POINTERS,
|
|
macho.S_MOD_TERM_FUNC_POINTERS,
|
|
=> {},
|
|
else => continue,
|
|
}
|
|
|
|
const segment_index = slice.items(.segment_index)[sect_id];
|
|
const segment = self.getSegment(@intCast(u8, sect_id));
|
|
if (segment.maxprot & macho.PROT.WRITE == 0) continue;
|
|
|
|
const cpu_arch = self.options.target.cpu.arch;
|
|
var atom_index = slice.items(.first_atom_index)[sect_id];
|
|
|
|
log.debug("{s},{s}", .{ header.segName(), header.sectName() });
|
|
|
|
while (true) {
|
|
const atom = self.getAtom(atom_index);
|
|
const sym = self.getSymbol(atom.getSymbolWithLoc());
|
|
|
|
log.debug(" ATOM({d}, %{d}, '{s}')", .{ atom_index, atom.sym_index, self.getSymbolName(atom.getSymbolWithLoc()) });
|
|
|
|
const should_bind = blk: {
|
|
if (self.dyld_private_sym_index) |sym_index| {
|
|
if (atom.getFile() == null and atom.sym_index == sym_index) break :blk false;
|
|
}
|
|
break :blk true;
|
|
};
|
|
|
|
if (should_bind) {
|
|
const object = self.objects.items[atom.getFile().?];
|
|
const base_rel_offset: i32 = blk: {
|
|
const source_sym = object.getSourceSymbol(atom.sym_index) orelse break :blk 0;
|
|
const source_sect = object.getSourceSection(source_sym.n_sect - 1);
|
|
break :blk @intCast(i32, source_sym.n_value - source_sect.addr);
|
|
};
|
|
const relocs = Atom.getAtomRelocs(self, atom_index);
|
|
|
|
for (relocs) |rel| {
|
|
switch (cpu_arch) {
|
|
.aarch64 => {
|
|
const rel_type = @intToEnum(macho.reloc_type_arm64, rel.r_type);
|
|
if (rel_type != .ARM64_RELOC_UNSIGNED) continue;
|
|
if (rel.r_length != 3) continue;
|
|
},
|
|
.x86_64 => {
|
|
const rel_type = @intToEnum(macho.reloc_type_x86_64, rel.r_type);
|
|
if (rel_type != .X86_64_RELOC_UNSIGNED) continue;
|
|
if (rel.r_length != 3) continue;
|
|
},
|
|
else => unreachable,
|
|
}
|
|
|
|
const global = Atom.parseRelocTarget(self, atom_index, rel, reverse_lookups[atom.getFile().?]);
|
|
const bind_sym_name = self.getSymbolName(global);
|
|
const bind_sym = self.getSymbol(global);
|
|
if (!bind_sym.undf()) continue;
|
|
|
|
const base_offset = @intCast(i32, sym.n_value - segment.vmaddr);
|
|
const rel_offset = rel.r_address - base_rel_offset;
|
|
const offset = @intCast(u64, base_offset + rel_offset);
|
|
|
|
const dylib_ordinal = @divTrunc(@bitCast(i16, bind_sym.n_desc), macho.N_SYMBOL_RESOLVER);
|
|
var flags: u4 = 0;
|
|
log.debug(" | bind at {x}, import('{s}') in dylib({d})", .{
|
|
base_offset,
|
|
bind_sym_name,
|
|
dylib_ordinal,
|
|
});
|
|
if (bind_sym.weakRef()) {
|
|
log.debug(" | marking as weak ref ", .{});
|
|
flags |= @truncate(u4, macho.BIND_SYMBOL_FLAGS_WEAK_IMPORT);
|
|
}
|
|
try pointers.append(.{
|
|
.offset = offset,
|
|
.segment_id = segment_index,
|
|
.dylib_ordinal = dylib_ordinal,
|
|
.name = bind_sym_name,
|
|
.bind_flags = flags,
|
|
});
|
|
}
|
|
}
|
|
if (atom.next_index) |next_index| {
|
|
atom_index = next_index;
|
|
} else break;
|
|
}
|
|
}
|
|
}
|
|
|
|
fn collectLazyBindData(self: *Zld, pointers: *std.ArrayList(bind.Pointer)) !void {
|
|
const sect_id = self.getSectionByName("__DATA", "__la_symbol_ptr") orelse return;
|
|
|
|
log.debug("collecting lazy bind data", .{});
|
|
|
|
const slice = self.sections.slice();
|
|
const segment_index = slice.items(.segment_index)[sect_id];
|
|
const seg = self.getSegment(sect_id);
|
|
var atom_index = slice.items(.first_atom_index)[sect_id];
|
|
|
|
// TODO: we actually don't need to store lazy pointer atoms as they are synthetically generated by the linker
|
|
try pointers.ensureUnusedCapacity(self.stubs.items.len);
|
|
|
|
var count: u32 = 0;
|
|
while (true) : (count += 1) {
|
|
const atom = self.getAtom(atom_index);
|
|
|
|
log.debug(" ATOM(%{d}, '{s}')", .{ atom.sym_index, self.getSymbolName(atom.getSymbolWithLoc()) });
|
|
|
|
const sym = self.getSymbol(atom.getSymbolWithLoc());
|
|
const base_offset = sym.n_value - seg.vmaddr;
|
|
|
|
const stub_entry = self.stubs.items[count];
|
|
const bind_sym = stub_entry.getTargetSymbol(self);
|
|
const bind_sym_name = stub_entry.getTargetSymbolName(self);
|
|
const dylib_ordinal = @divTrunc(@bitCast(i16, bind_sym.n_desc), macho.N_SYMBOL_RESOLVER);
|
|
var flags: u4 = 0;
|
|
log.debug(" | lazy bind at {x}, import('{s}') in dylib({d})", .{
|
|
base_offset,
|
|
bind_sym_name,
|
|
dylib_ordinal,
|
|
});
|
|
if (bind_sym.weakRef()) {
|
|
log.debug(" | marking as weak ref ", .{});
|
|
flags |= @truncate(u4, macho.BIND_SYMBOL_FLAGS_WEAK_IMPORT);
|
|
}
|
|
pointers.appendAssumeCapacity(.{
|
|
.offset = base_offset,
|
|
.segment_id = segment_index,
|
|
.dylib_ordinal = dylib_ordinal,
|
|
.name = bind_sym_name,
|
|
.bind_flags = flags,
|
|
});
|
|
|
|
if (atom.next_index) |next_index| {
|
|
atom_index = next_index;
|
|
} else break;
|
|
}
|
|
}
|
|
|
|
fn collectExportData(self: *Zld, trie: *Trie) !void {
|
|
const gpa = self.gpa;
|
|
|
|
// TODO handle macho.EXPORT_SYMBOL_FLAGS_REEXPORT and macho.EXPORT_SYMBOL_FLAGS_STUB_AND_RESOLVER.
|
|
log.debug("collecting export data", .{});
|
|
|
|
const segment_index = self.getSegmentByName("__TEXT").?;
|
|
const exec_segment = self.segments.items[segment_index];
|
|
const base_address = exec_segment.vmaddr;
|
|
|
|
if (self.options.output_mode == .Exe) {
|
|
for (&[_]SymbolWithLoc{
|
|
self.getEntryPoint(),
|
|
self.globals.items[self.mh_execute_header_index.?],
|
|
}) |global| {
|
|
const sym = self.getSymbol(global);
|
|
const sym_name = self.getSymbolName(global);
|
|
log.debug(" (putting '{s}' defined at 0x{x})", .{ sym_name, sym.n_value });
|
|
try trie.put(gpa, .{
|
|
.name = sym_name,
|
|
.vmaddr_offset = sym.n_value - base_address,
|
|
.export_flags = macho.EXPORT_SYMBOL_FLAGS_KIND_REGULAR,
|
|
});
|
|
}
|
|
} else {
|
|
assert(self.options.output_mode == .Lib);
|
|
for (self.globals.items) |global| {
|
|
const sym = self.getSymbol(global);
|
|
if (sym.undf()) continue;
|
|
if (sym.n_desc == N_DEAD) continue;
|
|
|
|
const sym_name = self.getSymbolName(global);
|
|
log.debug(" (putting '{s}' defined at 0x{x})", .{ sym_name, sym.n_value });
|
|
try trie.put(gpa, .{
|
|
.name = sym_name,
|
|
.vmaddr_offset = sym.n_value - base_address,
|
|
.export_flags = macho.EXPORT_SYMBOL_FLAGS_KIND_REGULAR,
|
|
});
|
|
}
|
|
}
|
|
|
|
try trie.finalize(gpa);
|
|
}
|
|
|
|
fn writeDyldInfoData(self: *Zld, ncmds: *u32, lc_writer: anytype, reverse_lookups: [][]u32) !void {
|
|
const gpa = self.gpa;
|
|
|
|
var rebase_pointers = std.ArrayList(bind.Pointer).init(gpa);
|
|
defer rebase_pointers.deinit();
|
|
try self.collectRebaseData(&rebase_pointers);
|
|
|
|
var bind_pointers = std.ArrayList(bind.Pointer).init(gpa);
|
|
defer bind_pointers.deinit();
|
|
try self.collectBindData(&bind_pointers, reverse_lookups);
|
|
|
|
var lazy_bind_pointers = std.ArrayList(bind.Pointer).init(gpa);
|
|
defer lazy_bind_pointers.deinit();
|
|
try self.collectLazyBindData(&lazy_bind_pointers);
|
|
|
|
var trie = Trie{};
|
|
defer trie.deinit(gpa);
|
|
try self.collectExportData(&trie);
|
|
|
|
const link_seg = self.getLinkeditSegmentPtr();
|
|
const rebase_off = mem.alignForwardGeneric(u64, link_seg.fileoff, @alignOf(u64));
|
|
assert(rebase_off == link_seg.fileoff);
|
|
const rebase_size = try bind.rebaseInfoSize(rebase_pointers.items);
|
|
log.debug("writing rebase info from 0x{x} to 0x{x}", .{ rebase_off, rebase_off + rebase_size });
|
|
|
|
const bind_off = mem.alignForwardGeneric(u64, rebase_off + rebase_size, @alignOf(u64));
|
|
const bind_size = try bind.bindInfoSize(bind_pointers.items);
|
|
log.debug("writing bind info from 0x{x} to 0x{x}", .{ bind_off, bind_off + bind_size });
|
|
|
|
const lazy_bind_off = mem.alignForwardGeneric(u64, bind_off + bind_size, @alignOf(u64));
|
|
const lazy_bind_size = try bind.lazyBindInfoSize(lazy_bind_pointers.items);
|
|
log.debug("writing lazy bind info from 0x{x} to 0x{x}", .{ lazy_bind_off, lazy_bind_off + lazy_bind_size });
|
|
|
|
const export_off = mem.alignForwardGeneric(u64, lazy_bind_off + lazy_bind_size, @alignOf(u64));
|
|
const export_size = trie.size;
|
|
log.debug("writing export trie from 0x{x} to 0x{x}", .{ export_off, export_off + export_size });
|
|
|
|
const needed_size = math.cast(usize, export_off + export_size - rebase_off) orelse return error.Overflow;
|
|
link_seg.filesize = needed_size;
|
|
|
|
var buffer = try gpa.alloc(u8, needed_size);
|
|
defer gpa.free(buffer);
|
|
mem.set(u8, buffer, 0);
|
|
|
|
var stream = std.io.fixedBufferStream(buffer);
|
|
const writer = stream.writer();
|
|
|
|
try bind.writeRebaseInfo(rebase_pointers.items, writer);
|
|
try stream.seekTo(bind_off - rebase_off);
|
|
|
|
try bind.writeBindInfo(bind_pointers.items, writer);
|
|
try stream.seekTo(lazy_bind_off - rebase_off);
|
|
|
|
try bind.writeLazyBindInfo(lazy_bind_pointers.items, writer);
|
|
try stream.seekTo(export_off - rebase_off);
|
|
|
|
_ = try trie.write(writer);
|
|
|
|
log.debug("writing dyld info from 0x{x} to 0x{x}", .{
|
|
rebase_off,
|
|
rebase_off + needed_size,
|
|
});
|
|
|
|
try self.file.pwriteAll(buffer, rebase_off);
|
|
|
|
const offset = math.cast(usize, lazy_bind_off - rebase_off) orelse return error.Overflow;
|
|
const size = math.cast(usize, lazy_bind_size) orelse return error.Overflow;
|
|
try self.populateLazyBindOffsetsInStubHelper(buffer[offset..][0..size]);
|
|
|
|
try lc_writer.writeStruct(macho.dyld_info_command{
|
|
.cmd = .DYLD_INFO_ONLY,
|
|
.cmdsize = @sizeOf(macho.dyld_info_command),
|
|
.rebase_off = @intCast(u32, rebase_off),
|
|
.rebase_size = @intCast(u32, rebase_size),
|
|
.bind_off = @intCast(u32, bind_off),
|
|
.bind_size = @intCast(u32, bind_size),
|
|
.weak_bind_off = 0,
|
|
.weak_bind_size = 0,
|
|
.lazy_bind_off = @intCast(u32, lazy_bind_off),
|
|
.lazy_bind_size = @intCast(u32, lazy_bind_size),
|
|
.export_off = @intCast(u32, export_off),
|
|
.export_size = @intCast(u32, export_size),
|
|
});
|
|
ncmds.* += 1;
|
|
}
|
|
|
|
fn populateLazyBindOffsetsInStubHelper(self: *Zld, buffer: []const u8) !void {
|
|
const gpa = self.gpa;
|
|
|
|
const stub_helper_section_index = self.getSectionByName("__TEXT", "__stub_helper") orelse return;
|
|
const la_symbol_ptr_section_index = self.getSectionByName("__DATA", "__la_symbol_ptr") orelse return;
|
|
|
|
if (self.stub_helper_preamble_sym_index == null) return;
|
|
|
|
const section = self.sections.get(stub_helper_section_index);
|
|
const last_atom_index = section.last_atom_index;
|
|
|
|
var table = std.AutoHashMap(i64, AtomIndex).init(gpa);
|
|
defer table.deinit();
|
|
|
|
{
|
|
var stub_atom_index = last_atom_index;
|
|
var laptr_atom_index = self.sections.items(.last_atom_index)[la_symbol_ptr_section_index];
|
|
|
|
const base_addr = blk: {
|
|
const segment_index = self.getSegmentByName("__DATA").?;
|
|
const seg = self.segments.items[segment_index];
|
|
break :blk seg.vmaddr;
|
|
};
|
|
|
|
while (true) {
|
|
const stub_atom = self.getAtom(stub_atom_index);
|
|
const laptr_atom = self.getAtom(laptr_atom_index);
|
|
const laptr_off = blk: {
|
|
const sym = self.getSymbolPtr(laptr_atom.getSymbolWithLoc());
|
|
break :blk @intCast(i64, sym.n_value - base_addr);
|
|
};
|
|
|
|
try table.putNoClobber(laptr_off, stub_atom_index);
|
|
|
|
if (laptr_atom.prev_index) |prev_index| {
|
|
laptr_atom_index = prev_index;
|
|
stub_atom_index = stub_atom.prev_index.?;
|
|
} else break;
|
|
}
|
|
}
|
|
|
|
var stream = std.io.fixedBufferStream(buffer);
|
|
var reader = stream.reader();
|
|
var offsets = std.ArrayList(struct { sym_offset: i64, offset: u32 }).init(gpa);
|
|
try offsets.append(.{ .sym_offset = undefined, .offset = 0 });
|
|
defer offsets.deinit();
|
|
var valid_block = false;
|
|
|
|
while (true) {
|
|
const inst = reader.readByte() catch |err| switch (err) {
|
|
error.EndOfStream => break,
|
|
};
|
|
const opcode: u8 = inst & macho.BIND_OPCODE_MASK;
|
|
|
|
switch (opcode) {
|
|
macho.BIND_OPCODE_DO_BIND => {
|
|
valid_block = true;
|
|
},
|
|
macho.BIND_OPCODE_DONE => {
|
|
if (valid_block) {
|
|
const offset = try stream.getPos();
|
|
try offsets.append(.{ .sym_offset = undefined, .offset = @intCast(u32, offset) });
|
|
}
|
|
valid_block = false;
|
|
},
|
|
macho.BIND_OPCODE_SET_SYMBOL_TRAILING_FLAGS_IMM => {
|
|
var next = try reader.readByte();
|
|
while (next != @as(u8, 0)) {
|
|
next = try reader.readByte();
|
|
}
|
|
},
|
|
macho.BIND_OPCODE_SET_SEGMENT_AND_OFFSET_ULEB => {
|
|
var inserted = offsets.pop();
|
|
inserted.sym_offset = try std.leb.readILEB128(i64, reader);
|
|
try offsets.append(inserted);
|
|
},
|
|
macho.BIND_OPCODE_SET_DYLIB_ORDINAL_ULEB => {
|
|
_ = try std.leb.readULEB128(u64, reader);
|
|
},
|
|
macho.BIND_OPCODE_SET_ADDEND_SLEB => {
|
|
_ = try std.leb.readILEB128(i64, reader);
|
|
},
|
|
else => {},
|
|
}
|
|
}
|
|
|
|
const header = self.sections.items(.header)[stub_helper_section_index];
|
|
const stub_offset: u4 = switch (self.options.target.cpu.arch) {
|
|
.x86_64 => 1,
|
|
.aarch64 => 2 * @sizeOf(u32),
|
|
else => unreachable,
|
|
};
|
|
var buf: [@sizeOf(u32)]u8 = undefined;
|
|
_ = offsets.pop();
|
|
|
|
while (offsets.popOrNull()) |bind_offset| {
|
|
const atom_index = table.get(bind_offset.sym_offset).?;
|
|
const atom = self.getAtom(atom_index);
|
|
const sym = self.getSymbol(atom.getSymbolWithLoc());
|
|
|
|
const file_offset = header.offset + sym.n_value - header.addr + stub_offset;
|
|
mem.writeIntLittle(u32, &buf, bind_offset.offset);
|
|
|
|
log.debug("writing lazy bind offset in stub helper of 0x{x} for symbol {s} at offset 0x{x}", .{
|
|
bind_offset.offset,
|
|
self.getSymbolName(atom.getSymbolWithLoc()),
|
|
file_offset,
|
|
});
|
|
|
|
try self.file.pwriteAll(&buf, file_offset);
|
|
}
|
|
}
|
|
|
|
const asc_u64 = std.sort.asc(u64);
|
|
|
|
fn writeFunctionStarts(self: *Zld, ncmds: *u32, lc_writer: anytype) !void {
|
|
const text_seg_index = self.getSegmentByName("__TEXT") orelse return;
|
|
const text_sect_index = self.getSectionByName("__TEXT", "__text") orelse return;
|
|
const text_seg = self.segments.items[text_seg_index];
|
|
|
|
const gpa = self.gpa;
|
|
|
|
// We need to sort by address first
|
|
var addresses = std.ArrayList(u64).init(gpa);
|
|
defer addresses.deinit();
|
|
try addresses.ensureTotalCapacityPrecise(self.globals.items.len);
|
|
|
|
for (self.globals.items) |global| {
|
|
const sym = self.getSymbol(global);
|
|
if (sym.undf()) continue;
|
|
if (sym.n_desc == N_DEAD) continue;
|
|
|
|
const sect_id = sym.n_sect - 1;
|
|
if (sect_id != text_sect_index) continue;
|
|
|
|
addresses.appendAssumeCapacity(sym.n_value);
|
|
}
|
|
|
|
std.sort.sort(u64, addresses.items, {}, asc_u64);
|
|
|
|
var offsets = std.ArrayList(u32).init(gpa);
|
|
defer offsets.deinit();
|
|
try offsets.ensureTotalCapacityPrecise(addresses.items.len);
|
|
|
|
var last_off: u32 = 0;
|
|
for (addresses.items) |addr| {
|
|
const offset = @intCast(u32, addr - text_seg.vmaddr);
|
|
const diff = offset - last_off;
|
|
|
|
if (diff == 0) continue;
|
|
|
|
offsets.appendAssumeCapacity(diff);
|
|
last_off = offset;
|
|
}
|
|
|
|
var buffer = std.ArrayList(u8).init(gpa);
|
|
defer buffer.deinit();
|
|
|
|
const max_size = @intCast(usize, offsets.items.len * @sizeOf(u64));
|
|
try buffer.ensureTotalCapacity(max_size);
|
|
|
|
for (offsets.items) |offset| {
|
|
try std.leb.writeULEB128(buffer.writer(), offset);
|
|
}
|
|
|
|
const link_seg = self.getLinkeditSegmentPtr();
|
|
const offset = mem.alignForwardGeneric(u64, link_seg.fileoff + link_seg.filesize, @alignOf(u64));
|
|
const needed_size = buffer.items.len;
|
|
link_seg.filesize = offset + needed_size - link_seg.fileoff;
|
|
|
|
log.debug("writing function starts info from 0x{x} to 0x{x}", .{ offset, offset + needed_size });
|
|
|
|
try self.file.pwriteAll(buffer.items, offset);
|
|
|
|
try lc_writer.writeStruct(macho.linkedit_data_command{
|
|
.cmd = .FUNCTION_STARTS,
|
|
.cmdsize = @sizeOf(macho.linkedit_data_command),
|
|
.dataoff = @intCast(u32, offset),
|
|
.datasize = @intCast(u32, needed_size),
|
|
});
|
|
ncmds.* += 1;
|
|
}
|
|
|
|
fn filterDataInCode(
|
|
dices: []const macho.data_in_code_entry,
|
|
start_addr: u64,
|
|
end_addr: u64,
|
|
) []const macho.data_in_code_entry {
|
|
const Predicate = struct {
|
|
addr: u64,
|
|
|
|
pub fn predicate(self: @This(), dice: macho.data_in_code_entry) bool {
|
|
return dice.offset >= self.addr;
|
|
}
|
|
};
|
|
|
|
const start = lsearch(macho.data_in_code_entry, dices, Predicate{ .addr = start_addr });
|
|
const end = lsearch(macho.data_in_code_entry, dices[start..], Predicate{ .addr = end_addr }) + start;
|
|
|
|
return dices[start..end];
|
|
}
|
|
|
|
fn writeDataInCode(self: *Zld, ncmds: *u32, lc_writer: anytype) !void {
|
|
var out_dice = std.ArrayList(macho.data_in_code_entry).init(self.gpa);
|
|
defer out_dice.deinit();
|
|
|
|
const text_sect_id = self.getSectionByName("__TEXT", "__text") orelse return;
|
|
const text_sect_header = self.sections.items(.header)[text_sect_id];
|
|
|
|
for (self.objects.items) |object| {
|
|
const dice = object.parseDataInCode() orelse continue;
|
|
try out_dice.ensureUnusedCapacity(dice.len);
|
|
|
|
for (object.atoms.items) |atom_index| {
|
|
const atom = self.getAtom(atom_index);
|
|
const sym = self.getSymbol(atom.getSymbolWithLoc());
|
|
const sect_id = sym.n_sect - 1;
|
|
if (sect_id != text_sect_id) {
|
|
continue;
|
|
}
|
|
|
|
const source_addr = if (object.getSourceSymbol(atom.sym_index)) |source_sym|
|
|
source_sym.n_value
|
|
else blk: {
|
|
const nbase = @intCast(u32, object.in_symtab.?.len);
|
|
const source_sect_id = @intCast(u16, atom.sym_index - nbase);
|
|
break :blk object.getSourceSection(source_sect_id).addr;
|
|
};
|
|
const filtered_dice = filterDataInCode(dice, source_addr, source_addr + atom.size);
|
|
const base = math.cast(u32, sym.n_value - text_sect_header.addr + text_sect_header.offset) orelse
|
|
return error.Overflow;
|
|
|
|
for (filtered_dice) |single| {
|
|
const offset = math.cast(u32, single.offset - source_addr + base) orelse
|
|
return error.Overflow;
|
|
out_dice.appendAssumeCapacity(.{
|
|
.offset = offset,
|
|
.length = single.length,
|
|
.kind = single.kind,
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
const seg = self.getLinkeditSegmentPtr();
|
|
const offset = mem.alignForwardGeneric(u64, seg.fileoff + seg.filesize, @alignOf(u64));
|
|
const needed_size = out_dice.items.len * @sizeOf(macho.data_in_code_entry);
|
|
seg.filesize = offset + needed_size - seg.fileoff;
|
|
|
|
log.debug("writing data-in-code from 0x{x} to 0x{x}", .{ offset, offset + needed_size });
|
|
|
|
try self.file.pwriteAll(mem.sliceAsBytes(out_dice.items), offset);
|
|
try lc_writer.writeStruct(macho.linkedit_data_command{
|
|
.cmd = .DATA_IN_CODE,
|
|
.cmdsize = @sizeOf(macho.linkedit_data_command),
|
|
.dataoff = @intCast(u32, offset),
|
|
.datasize = @intCast(u32, needed_size),
|
|
});
|
|
ncmds.* += 1;
|
|
}
|
|
|
|
fn writeSymtabs(self: *Zld, ncmds: *u32, lc_writer: anytype) !void {
|
|
var symtab_cmd = macho.symtab_command{
|
|
.cmdsize = @sizeOf(macho.symtab_command),
|
|
.symoff = 0,
|
|
.nsyms = 0,
|
|
.stroff = 0,
|
|
.strsize = 0,
|
|
};
|
|
var dysymtab_cmd = macho.dysymtab_command{
|
|
.cmdsize = @sizeOf(macho.dysymtab_command),
|
|
.ilocalsym = 0,
|
|
.nlocalsym = 0,
|
|
.iextdefsym = 0,
|
|
.nextdefsym = 0,
|
|
.iundefsym = 0,
|
|
.nundefsym = 0,
|
|
.tocoff = 0,
|
|
.ntoc = 0,
|
|
.modtaboff = 0,
|
|
.nmodtab = 0,
|
|
.extrefsymoff = 0,
|
|
.nextrefsyms = 0,
|
|
.indirectsymoff = 0,
|
|
.nindirectsyms = 0,
|
|
.extreloff = 0,
|
|
.nextrel = 0,
|
|
.locreloff = 0,
|
|
.nlocrel = 0,
|
|
};
|
|
var ctx = try self.writeSymtab(&symtab_cmd);
|
|
defer ctx.imports_table.deinit();
|
|
try self.writeDysymtab(ctx, &dysymtab_cmd);
|
|
try self.writeStrtab(&symtab_cmd);
|
|
try lc_writer.writeStruct(symtab_cmd);
|
|
try lc_writer.writeStruct(dysymtab_cmd);
|
|
ncmds.* += 2;
|
|
}
|
|
|
|
fn writeSymtab(self: *Zld, lc: *macho.symtab_command) !SymtabCtx {
|
|
const gpa = self.gpa;
|
|
|
|
var locals = std.ArrayList(macho.nlist_64).init(gpa);
|
|
defer locals.deinit();
|
|
|
|
for (self.objects.items) |object| {
|
|
for (object.atoms.items) |atom_index| {
|
|
const atom = self.getAtom(atom_index);
|
|
const sym_loc = atom.getSymbolWithLoc();
|
|
const sym = self.getSymbol(sym_loc);
|
|
if (sym.n_strx == 0) continue; // no name, skip
|
|
if (sym.ext()) continue; // an export lands in its own symtab section, skip
|
|
if (self.symbolIsTemp(sym_loc)) continue; // local temp symbol, skip
|
|
|
|
var out_sym = sym;
|
|
out_sym.n_strx = try self.strtab.insert(gpa, self.getSymbolName(sym_loc));
|
|
try locals.append(out_sym);
|
|
}
|
|
}
|
|
|
|
if (!self.options.strip) {
|
|
for (self.objects.items) |object| {
|
|
try self.generateSymbolStabs(object, &locals);
|
|
}
|
|
}
|
|
|
|
var exports = std.ArrayList(macho.nlist_64).init(gpa);
|
|
defer exports.deinit();
|
|
|
|
for (self.globals.items) |global| {
|
|
const sym = self.getSymbol(global);
|
|
if (sym.undf()) continue; // import, skip
|
|
if (sym.n_desc == N_DEAD) continue;
|
|
|
|
var out_sym = sym;
|
|
out_sym.n_strx = try self.strtab.insert(gpa, self.getSymbolName(global));
|
|
try exports.append(out_sym);
|
|
}
|
|
|
|
var imports = std.ArrayList(macho.nlist_64).init(gpa);
|
|
defer imports.deinit();
|
|
|
|
var imports_table = std.AutoHashMap(SymbolWithLoc, u32).init(gpa);
|
|
|
|
for (self.globals.items) |global| {
|
|
const sym = self.getSymbol(global);
|
|
if (!sym.undf()) continue; // not an import, skip
|
|
if (sym.n_desc == N_DEAD) continue;
|
|
|
|
const new_index = @intCast(u32, imports.items.len);
|
|
var out_sym = sym;
|
|
out_sym.n_strx = try self.strtab.insert(gpa, self.getSymbolName(global));
|
|
try imports.append(out_sym);
|
|
try imports_table.putNoClobber(global, new_index);
|
|
}
|
|
|
|
const nlocals = @intCast(u32, locals.items.len);
|
|
const nexports = @intCast(u32, exports.items.len);
|
|
const nimports = @intCast(u32, imports.items.len);
|
|
const nsyms = nlocals + nexports + nimports;
|
|
|
|
const seg = self.getLinkeditSegmentPtr();
|
|
const offset = mem.alignForwardGeneric(
|
|
u64,
|
|
seg.fileoff + seg.filesize,
|
|
@alignOf(macho.nlist_64),
|
|
);
|
|
const needed_size = nsyms * @sizeOf(macho.nlist_64);
|
|
seg.filesize = offset + needed_size - seg.fileoff;
|
|
|
|
var buffer = std.ArrayList(u8).init(gpa);
|
|
defer buffer.deinit();
|
|
try buffer.ensureTotalCapacityPrecise(needed_size);
|
|
buffer.appendSliceAssumeCapacity(mem.sliceAsBytes(locals.items));
|
|
buffer.appendSliceAssumeCapacity(mem.sliceAsBytes(exports.items));
|
|
buffer.appendSliceAssumeCapacity(mem.sliceAsBytes(imports.items));
|
|
|
|
log.debug("writing symtab from 0x{x} to 0x{x}", .{ offset, offset + needed_size });
|
|
try self.file.pwriteAll(buffer.items, offset);
|
|
|
|
lc.symoff = @intCast(u32, offset);
|
|
lc.nsyms = nsyms;
|
|
|
|
return SymtabCtx{
|
|
.nlocalsym = nlocals,
|
|
.nextdefsym = nexports,
|
|
.nundefsym = nimports,
|
|
.imports_table = imports_table,
|
|
};
|
|
}
|
|
|
|
fn writeStrtab(self: *Zld, lc: *macho.symtab_command) !void {
|
|
const seg = self.getLinkeditSegmentPtr();
|
|
const offset = mem.alignForwardGeneric(u64, seg.fileoff + seg.filesize, @alignOf(u64));
|
|
const needed_size = self.strtab.buffer.items.len;
|
|
seg.filesize = offset + needed_size - seg.fileoff;
|
|
|
|
log.debug("writing string table from 0x{x} to 0x{x}", .{ offset, offset + needed_size });
|
|
|
|
try self.file.pwriteAll(self.strtab.buffer.items, offset);
|
|
|
|
lc.stroff = @intCast(u32, offset);
|
|
lc.strsize = @intCast(u32, needed_size);
|
|
}
|
|
|
|
const SymtabCtx = struct {
|
|
nlocalsym: u32,
|
|
nextdefsym: u32,
|
|
nundefsym: u32,
|
|
imports_table: std.AutoHashMap(SymbolWithLoc, u32),
|
|
};
|
|
|
|
fn writeDysymtab(self: *Zld, ctx: SymtabCtx, lc: *macho.dysymtab_command) !void {
|
|
const gpa = self.gpa;
|
|
const nstubs = @intCast(u32, self.stubs.items.len);
|
|
const ngot_entries = @intCast(u32, self.got_entries.items.len);
|
|
const nindirectsyms = nstubs * 2 + ngot_entries;
|
|
const iextdefsym = ctx.nlocalsym;
|
|
const iundefsym = iextdefsym + ctx.nextdefsym;
|
|
|
|
const seg = self.getLinkeditSegmentPtr();
|
|
const offset = mem.alignForwardGeneric(u64, seg.fileoff + seg.filesize, @alignOf(u64));
|
|
const needed_size = nindirectsyms * @sizeOf(u32);
|
|
seg.filesize = offset + needed_size - seg.fileoff;
|
|
|
|
log.debug("writing indirect symbol table from 0x{x} to 0x{x}", .{ offset, offset + needed_size });
|
|
|
|
var buf = std.ArrayList(u8).init(gpa);
|
|
defer buf.deinit();
|
|
try buf.ensureTotalCapacity(needed_size);
|
|
const writer = buf.writer();
|
|
|
|
if (self.getSectionByName("__TEXT", "__stubs")) |sect_id| {
|
|
const stubs = &self.sections.items(.header)[sect_id];
|
|
stubs.reserved1 = 0;
|
|
for (self.stubs.items) |entry| {
|
|
const target_sym = entry.getTargetSymbol(self);
|
|
assert(target_sym.undf());
|
|
try writer.writeIntLittle(u32, iundefsym + ctx.imports_table.get(entry.target).?);
|
|
}
|
|
}
|
|
|
|
if (self.getSectionByName("__DATA_CONST", "__got")) |sect_id| {
|
|
const got = &self.sections.items(.header)[sect_id];
|
|
got.reserved1 = nstubs;
|
|
for (self.got_entries.items) |entry| {
|
|
const target_sym = entry.getTargetSymbol(self);
|
|
if (target_sym.undf()) {
|
|
try writer.writeIntLittle(u32, iundefsym + ctx.imports_table.get(entry.target).?);
|
|
} else {
|
|
try writer.writeIntLittle(u32, macho.INDIRECT_SYMBOL_LOCAL);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (self.getSectionByName("__DATA", "__la_symbol_ptr")) |sect_id| {
|
|
const la_symbol_ptr = &self.sections.items(.header)[sect_id];
|
|
la_symbol_ptr.reserved1 = nstubs + ngot_entries;
|
|
for (self.stubs.items) |entry| {
|
|
const target_sym = entry.getTargetSymbol(self);
|
|
assert(target_sym.undf());
|
|
try writer.writeIntLittle(u32, iundefsym + ctx.imports_table.get(entry.target).?);
|
|
}
|
|
}
|
|
|
|
assert(buf.items.len == needed_size);
|
|
try self.file.pwriteAll(buf.items, offset);
|
|
|
|
lc.nlocalsym = ctx.nlocalsym;
|
|
lc.iextdefsym = iextdefsym;
|
|
lc.nextdefsym = ctx.nextdefsym;
|
|
lc.iundefsym = iundefsym;
|
|
lc.nundefsym = ctx.nundefsym;
|
|
lc.indirectsymoff = @intCast(u32, offset);
|
|
lc.nindirectsyms = nindirectsyms;
|
|
}
|
|
|
|
fn writeCodeSignaturePadding(
|
|
self: *Zld,
|
|
code_sig: *CodeSignature,
|
|
ncmds: *u32,
|
|
lc_writer: anytype,
|
|
) !u32 {
|
|
const seg = self.getLinkeditSegmentPtr();
|
|
// Code signature data has to be 16-bytes aligned for Apple tools to recognize the file
|
|
// https://github.com/opensource-apple/cctools/blob/fdb4825f303fd5c0751be524babd32958181b3ed/libstuff/checkout.c#L271
|
|
const offset = mem.alignForwardGeneric(u64, seg.fileoff + seg.filesize, 16);
|
|
const needed_size = code_sig.estimateSize(offset);
|
|
seg.filesize = offset + needed_size - seg.fileoff;
|
|
seg.vmsize = mem.alignForwardGeneric(u64, seg.filesize, self.page_size);
|
|
log.debug("writing code signature padding from 0x{x} to 0x{x}", .{ offset, offset + needed_size });
|
|
// Pad out the space. We need to do this to calculate valid hashes for everything in the file
|
|
// except for code signature data.
|
|
try self.file.pwriteAll(&[_]u8{0}, offset + needed_size - 1);
|
|
|
|
try lc_writer.writeStruct(macho.linkedit_data_command{
|
|
.cmd = .CODE_SIGNATURE,
|
|
.cmdsize = @sizeOf(macho.linkedit_data_command),
|
|
.dataoff = @intCast(u32, offset),
|
|
.datasize = @intCast(u32, needed_size),
|
|
});
|
|
ncmds.* += 1;
|
|
|
|
return @intCast(u32, offset);
|
|
}
|
|
|
|
fn writeCodeSignature(
|
|
self: *Zld,
|
|
comp: *const Compilation,
|
|
code_sig: *CodeSignature,
|
|
offset: u32,
|
|
) !void {
|
|
const seg_id = self.getSegmentByName("__TEXT").?;
|
|
const seg = self.segments.items[seg_id];
|
|
|
|
var buffer = std.ArrayList(u8).init(self.gpa);
|
|
defer buffer.deinit();
|
|
try buffer.ensureTotalCapacityPrecise(code_sig.size());
|
|
try code_sig.writeAdhocSignature(comp, .{
|
|
.file = self.file,
|
|
.exec_seg_base = seg.fileoff,
|
|
.exec_seg_limit = seg.filesize,
|
|
.file_size = offset,
|
|
.output_mode = self.options.output_mode,
|
|
}, buffer.writer());
|
|
assert(buffer.items.len == code_sig.size());
|
|
|
|
log.debug("writing code signature from 0x{x} to 0x{x}", .{
|
|
offset,
|
|
offset + buffer.items.len,
|
|
});
|
|
|
|
try self.file.pwriteAll(buffer.items, offset);
|
|
}
|
|
|
|
/// Writes Mach-O file header.
|
|
fn writeHeader(self: *Zld, ncmds: u32, sizeofcmds: u32) !void {
|
|
var header: macho.mach_header_64 = .{};
|
|
header.flags = macho.MH_NOUNDEFS | macho.MH_DYLDLINK | macho.MH_PIE | macho.MH_TWOLEVEL;
|
|
|
|
switch (self.options.target.cpu.arch) {
|
|
.aarch64 => {
|
|
header.cputype = macho.CPU_TYPE_ARM64;
|
|
header.cpusubtype = macho.CPU_SUBTYPE_ARM_ALL;
|
|
},
|
|
.x86_64 => {
|
|
header.cputype = macho.CPU_TYPE_X86_64;
|
|
header.cpusubtype = macho.CPU_SUBTYPE_X86_64_ALL;
|
|
},
|
|
else => return error.UnsupportedCpuArchitecture,
|
|
}
|
|
|
|
switch (self.options.output_mode) {
|
|
.Exe => {
|
|
header.filetype = macho.MH_EXECUTE;
|
|
},
|
|
.Lib => {
|
|
// By this point, it can only be a dylib.
|
|
header.filetype = macho.MH_DYLIB;
|
|
header.flags |= macho.MH_NO_REEXPORTED_DYLIBS;
|
|
},
|
|
else => unreachable,
|
|
}
|
|
|
|
if (self.getSectionByName("__DATA", "__thread_vars")) |sect_id| {
|
|
header.flags |= macho.MH_HAS_TLV_DESCRIPTORS;
|
|
if (self.sections.items(.header)[sect_id].size > 0) {
|
|
header.flags |= macho.MH_HAS_TLV_DESCRIPTORS;
|
|
}
|
|
}
|
|
|
|
header.ncmds = ncmds;
|
|
header.sizeofcmds = sizeofcmds;
|
|
|
|
log.debug("writing Mach-O header {}", .{header});
|
|
|
|
try self.file.pwriteAll(mem.asBytes(&header), 0);
|
|
}
|
|
|
|
pub fn makeStaticString(bytes: []const u8) [16]u8 {
|
|
var buf = [_]u8{0} ** 16;
|
|
assert(bytes.len <= buf.len);
|
|
mem.copy(u8, &buf, bytes);
|
|
return buf;
|
|
}
|
|
|
|
pub inline fn getAtomPtr(self: *Zld, atom_index: AtomIndex) *Atom {
|
|
assert(atom_index < self.atoms.items.len);
|
|
return &self.atoms.items[atom_index];
|
|
}
|
|
|
|
pub inline fn getAtom(self: Zld, atom_index: AtomIndex) Atom {
|
|
assert(atom_index < self.atoms.items.len);
|
|
return self.atoms.items[atom_index];
|
|
}
|
|
|
|
fn getSegmentByName(self: Zld, segname: []const u8) ?u8 {
|
|
for (self.segments.items) |seg, i| {
|
|
if (mem.eql(u8, segname, seg.segName())) return @intCast(u8, i);
|
|
} else return null;
|
|
}
|
|
|
|
pub inline fn getSegment(self: Zld, sect_id: u8) macho.segment_command_64 {
|
|
const index = self.sections.items(.segment_index)[sect_id];
|
|
return self.segments.items[index];
|
|
}
|
|
|
|
pub inline fn getSegmentPtr(self: *Zld, sect_id: u8) *macho.segment_command_64 {
|
|
const index = self.sections.items(.segment_index)[sect_id];
|
|
return &self.segments.items[index];
|
|
}
|
|
|
|
pub inline fn getLinkeditSegmentPtr(self: *Zld) *macho.segment_command_64 {
|
|
assert(self.segments.items.len > 0);
|
|
const seg = &self.segments.items[self.segments.items.len - 1];
|
|
assert(mem.eql(u8, seg.segName(), "__LINKEDIT"));
|
|
return seg;
|
|
}
|
|
|
|
pub fn getSectionByName(self: Zld, segname: []const u8, sectname: []const u8) ?u8 {
|
|
// TODO investigate caching with a hashmap
|
|
for (self.sections.items(.header)) |header, i| {
|
|
if (mem.eql(u8, header.segName(), segname) and mem.eql(u8, header.sectName(), sectname))
|
|
return @intCast(u8, i);
|
|
} else return null;
|
|
}
|
|
|
|
pub fn getSectionIndexes(self: Zld, segment_index: u8) struct { start: u8, end: u8 } {
|
|
var start: u8 = 0;
|
|
const nsects = for (self.segments.items) |seg, i| {
|
|
if (i == segment_index) break @intCast(u8, seg.nsects);
|
|
start += @intCast(u8, seg.nsects);
|
|
} else 0;
|
|
return .{ .start = start, .end = start + nsects };
|
|
}
|
|
|
|
pub fn symbolIsTemp(self: *Zld, sym_with_loc: SymbolWithLoc) bool {
|
|
const sym = self.getSymbol(sym_with_loc);
|
|
if (!sym.sect()) return false;
|
|
if (sym.ext()) return false;
|
|
const sym_name = self.getSymbolName(sym_with_loc);
|
|
return mem.startsWith(u8, sym_name, "l") or mem.startsWith(u8, sym_name, "L");
|
|
}
|
|
|
|
/// Returns pointer-to-symbol described by `sym_with_loc` descriptor.
|
|
pub fn getSymbolPtr(self: *Zld, sym_with_loc: SymbolWithLoc) *macho.nlist_64 {
|
|
if (sym_with_loc.getFile()) |file| {
|
|
const object = &self.objects.items[file];
|
|
return &object.symtab[sym_with_loc.sym_index];
|
|
} else {
|
|
return &self.locals.items[sym_with_loc.sym_index];
|
|
}
|
|
}
|
|
|
|
/// Returns symbol described by `sym_with_loc` descriptor.
|
|
pub fn getSymbol(self: *Zld, sym_with_loc: SymbolWithLoc) macho.nlist_64 {
|
|
return self.getSymbolPtr(sym_with_loc).*;
|
|
}
|
|
|
|
/// Returns name of the symbol described by `sym_with_loc` descriptor.
|
|
pub fn getSymbolName(self: *Zld, sym_with_loc: SymbolWithLoc) []const u8 {
|
|
if (sym_with_loc.getFile()) |file| {
|
|
const object = self.objects.items[file];
|
|
return object.getSymbolName(sym_with_loc.sym_index);
|
|
} else {
|
|
const sym = self.locals.items[sym_with_loc.sym_index];
|
|
return self.strtab.get(sym.n_strx).?;
|
|
}
|
|
}
|
|
|
|
/// Returns GOT atom that references `sym_with_loc` if one exists.
|
|
/// Returns null otherwise.
|
|
pub fn getGotAtomIndexForSymbol(self: *Zld, sym_with_loc: SymbolWithLoc) ?AtomIndex {
|
|
const index = self.got_table.get(sym_with_loc) orelse return null;
|
|
const entry = self.got_entries.items[index];
|
|
return entry.atom_index;
|
|
}
|
|
|
|
/// Returns stubs atom that references `sym_with_loc` if one exists.
|
|
/// Returns null otherwise.
|
|
pub fn getStubsAtomIndexForSymbol(self: *Zld, sym_with_loc: SymbolWithLoc) ?AtomIndex {
|
|
const index = self.stubs_table.get(sym_with_loc) orelse return null;
|
|
const entry = self.stubs.items[index];
|
|
return entry.atom_index;
|
|
}
|
|
|
|
/// Returns TLV pointer atom that references `sym_with_loc` if one exists.
|
|
/// Returns null otherwise.
|
|
pub fn getTlvPtrAtomIndexForSymbol(self: *Zld, sym_with_loc: SymbolWithLoc) ?AtomIndex {
|
|
const index = self.tlv_ptr_table.get(sym_with_loc) orelse return null;
|
|
const entry = self.tlv_ptr_entries.items[index];
|
|
return entry.atom_index;
|
|
}
|
|
|
|
/// Returns symbol location corresponding to the set entrypoint.
|
|
/// Asserts output mode is executable.
|
|
pub fn getEntryPoint(self: Zld) SymbolWithLoc {
|
|
assert(self.options.output_mode == .Exe);
|
|
const global_index = self.entry_index.?;
|
|
return self.globals.items[global_index];
|
|
}
|
|
|
|
inline fn requiresThunks(self: Zld) bool {
|
|
return self.options.target.cpu.arch == .aarch64;
|
|
}
|
|
|
|
pub fn generateSymbolStabs(self: *Zld, object: Object, locals: *std.ArrayList(macho.nlist_64)) !void {
|
|
log.debug("generating stabs for '{s}'", .{object.name});
|
|
|
|
const gpa = self.gpa;
|
|
var debug_info = object.parseDwarfInfo();
|
|
|
|
var lookup = DwarfInfo.AbbrevLookupTable.init(gpa);
|
|
defer lookup.deinit();
|
|
try lookup.ensureUnusedCapacity(std.math.maxInt(u8));
|
|
|
|
// We assume there is only one CU.
|
|
var cu_it = debug_info.getCompileUnitIterator();
|
|
const compile_unit = while (try cu_it.next()) |cu| {
|
|
const offset = math.cast(usize, cu.cuh.debug_abbrev_offset) orelse return error.Overflow;
|
|
try debug_info.genAbbrevLookupByKind(offset, &lookup);
|
|
break cu;
|
|
} else {
|
|
log.debug("no compile unit found in debug info in {s}; skipping", .{object.name});
|
|
return;
|
|
};
|
|
|
|
var abbrev_it = compile_unit.getAbbrevEntryIterator(debug_info);
|
|
const cu_entry: DwarfInfo.AbbrevEntry = while (try abbrev_it.next(lookup)) |entry| switch (entry.tag) {
|
|
dwarf.TAG.compile_unit => break entry,
|
|
else => continue,
|
|
} else {
|
|
log.debug("missing DWARF_TAG_compile_unit tag in {s}; skipping", .{object.name});
|
|
return;
|
|
};
|
|
|
|
var maybe_tu_name: ?[]const u8 = null;
|
|
var maybe_tu_comp_dir: ?[]const u8 = null;
|
|
var attr_it = cu_entry.getAttributeIterator(debug_info, compile_unit.cuh);
|
|
|
|
while (try attr_it.next()) |attr| switch (attr.name) {
|
|
dwarf.AT.comp_dir => maybe_tu_comp_dir = attr.getString(debug_info, compile_unit.cuh) orelse continue,
|
|
dwarf.AT.name => maybe_tu_name = attr.getString(debug_info, compile_unit.cuh) orelse continue,
|
|
else => continue,
|
|
};
|
|
|
|
if (maybe_tu_name == null or maybe_tu_comp_dir == null) {
|
|
log.debug("missing DWARF_AT_comp_dir and DWARF_AT_name attributes {s}; skipping", .{object.name});
|
|
return;
|
|
}
|
|
|
|
const tu_name = maybe_tu_name.?;
|
|
const tu_comp_dir = maybe_tu_comp_dir.?;
|
|
|
|
// Open scope
|
|
try locals.ensureUnusedCapacity(3);
|
|
locals.appendAssumeCapacity(.{
|
|
.n_strx = try self.strtab.insert(gpa, tu_comp_dir),
|
|
.n_type = macho.N_SO,
|
|
.n_sect = 0,
|
|
.n_desc = 0,
|
|
.n_value = 0,
|
|
});
|
|
locals.appendAssumeCapacity(.{
|
|
.n_strx = try self.strtab.insert(gpa, tu_name),
|
|
.n_type = macho.N_SO,
|
|
.n_sect = 0,
|
|
.n_desc = 0,
|
|
.n_value = 0,
|
|
});
|
|
locals.appendAssumeCapacity(.{
|
|
.n_strx = try self.strtab.insert(gpa, object.name),
|
|
.n_type = macho.N_OSO,
|
|
.n_sect = 0,
|
|
.n_desc = 1,
|
|
.n_value = object.mtime,
|
|
});
|
|
|
|
var stabs_buf: [4]macho.nlist_64 = undefined;
|
|
|
|
var name_lookup: ?DwarfInfo.SubprogramLookupByName = if (object.header.flags & macho.MH_SUBSECTIONS_VIA_SYMBOLS == 0) blk: {
|
|
var name_lookup = DwarfInfo.SubprogramLookupByName.init(gpa);
|
|
errdefer name_lookup.deinit();
|
|
try name_lookup.ensureUnusedCapacity(@intCast(u32, object.atoms.items.len));
|
|
try debug_info.genSubprogramLookupByName(compile_unit, lookup, &name_lookup);
|
|
break :blk name_lookup;
|
|
} else null;
|
|
defer if (name_lookup) |*nl| nl.deinit();
|
|
|
|
for (object.atoms.items) |atom_index| {
|
|
const atom = self.getAtom(atom_index);
|
|
const stabs = try self.generateSymbolStabsForSymbol(
|
|
atom_index,
|
|
atom.getSymbolWithLoc(),
|
|
name_lookup,
|
|
&stabs_buf,
|
|
);
|
|
try locals.appendSlice(stabs);
|
|
|
|
var it = Atom.getInnerSymbolsIterator(self, atom_index);
|
|
while (it.next()) |sym_loc| {
|
|
const contained_stabs = try self.generateSymbolStabsForSymbol(
|
|
atom_index,
|
|
sym_loc,
|
|
name_lookup,
|
|
&stabs_buf,
|
|
);
|
|
try locals.appendSlice(contained_stabs);
|
|
}
|
|
}
|
|
|
|
// Close scope
|
|
try locals.append(.{
|
|
.n_strx = 0,
|
|
.n_type = macho.N_SO,
|
|
.n_sect = 0,
|
|
.n_desc = 0,
|
|
.n_value = 0,
|
|
});
|
|
}
|
|
|
|
fn generateSymbolStabsForSymbol(
|
|
self: *Zld,
|
|
atom_index: AtomIndex,
|
|
sym_loc: SymbolWithLoc,
|
|
lookup: ?DwarfInfo.SubprogramLookupByName,
|
|
buf: *[4]macho.nlist_64,
|
|
) ![]const macho.nlist_64 {
|
|
const gpa = self.gpa;
|
|
const object = self.objects.items[sym_loc.getFile().?];
|
|
const sym = self.getSymbol(sym_loc);
|
|
const sym_name = self.getSymbolName(sym_loc);
|
|
const header = self.sections.items(.header)[sym.n_sect - 1];
|
|
|
|
if (sym.n_strx == 0) return buf[0..0];
|
|
if (self.symbolIsTemp(sym_loc)) return buf[0..0];
|
|
|
|
if (!header.isCode()) {
|
|
// Since we are not dealing with machine code, it's either a global or a static depending
|
|
// on the linkage scope.
|
|
if (sym.sect() and sym.ext()) {
|
|
// Global gets an N_GSYM stab type.
|
|
buf[0] = .{
|
|
.n_strx = try self.strtab.insert(gpa, sym_name),
|
|
.n_type = macho.N_GSYM,
|
|
.n_sect = sym.n_sect,
|
|
.n_desc = 0,
|
|
.n_value = 0,
|
|
};
|
|
} else {
|
|
// Local static gets an N_STSYM stab type.
|
|
buf[0] = .{
|
|
.n_strx = try self.strtab.insert(gpa, sym_name),
|
|
.n_type = macho.N_STSYM,
|
|
.n_sect = sym.n_sect,
|
|
.n_desc = 0,
|
|
.n_value = sym.n_value,
|
|
};
|
|
}
|
|
return buf[0..1];
|
|
}
|
|
|
|
const size: u64 = size: {
|
|
if (object.header.flags & macho.MH_SUBSECTIONS_VIA_SYMBOLS != 0) {
|
|
break :size self.getAtom(atom_index).size;
|
|
}
|
|
|
|
// Since we don't have subsections to work with, we need to infer the size of each function
|
|
// the slow way by scanning the debug info for matching symbol names and extracting
|
|
// the symbol's DWARF_AT_low_pc and DWARF_AT_high_pc values.
|
|
const source_sym = object.getSourceSymbol(sym_loc.sym_index) orelse return buf[0..0];
|
|
const subprogram = lookup.?.get(sym_name[1..]) orelse return buf[0..0];
|
|
|
|
if (subprogram.addr <= source_sym.n_value and source_sym.n_value < subprogram.addr + subprogram.size) {
|
|
break :size subprogram.size;
|
|
} else {
|
|
log.debug("no stab found for {s}", .{sym_name});
|
|
return buf[0..0];
|
|
}
|
|
};
|
|
|
|
buf[0] = .{
|
|
.n_strx = 0,
|
|
.n_type = macho.N_BNSYM,
|
|
.n_sect = sym.n_sect,
|
|
.n_desc = 0,
|
|
.n_value = sym.n_value,
|
|
};
|
|
buf[1] = .{
|
|
.n_strx = try self.strtab.insert(gpa, sym_name),
|
|
.n_type = macho.N_FUN,
|
|
.n_sect = sym.n_sect,
|
|
.n_desc = 0,
|
|
.n_value = sym.n_value,
|
|
};
|
|
buf[2] = .{
|
|
.n_strx = 0,
|
|
.n_type = macho.N_FUN,
|
|
.n_sect = 0,
|
|
.n_desc = 0,
|
|
.n_value = size,
|
|
};
|
|
buf[3] = .{
|
|
.n_strx = 0,
|
|
.n_type = macho.N_ENSYM,
|
|
.n_sect = sym.n_sect,
|
|
.n_desc = 0,
|
|
.n_value = size,
|
|
};
|
|
|
|
return buf;
|
|
}
|
|
|
|
fn logSegments(self: *Zld) void {
|
|
log.debug("segments:", .{});
|
|
for (self.segments.items) |segment, i| {
|
|
log.debug(" segment({d}): {s} @{x} ({x}), sizeof({x})", .{
|
|
i,
|
|
segment.segName(),
|
|
segment.fileoff,
|
|
segment.vmaddr,
|
|
segment.vmsize,
|
|
});
|
|
}
|
|
}
|
|
|
|
fn logSections(self: *Zld) void {
|
|
log.debug("sections:", .{});
|
|
for (self.sections.items(.header)) |header, i| {
|
|
log.debug(" sect({d}): {s},{s} @{x} ({x}), sizeof({x})", .{
|
|
i + 1,
|
|
header.segName(),
|
|
header.sectName(),
|
|
header.offset,
|
|
header.addr,
|
|
header.size,
|
|
});
|
|
}
|
|
}
|
|
|
|
fn logSymAttributes(sym: macho.nlist_64, buf: []u8) []const u8 {
|
|
if (sym.sect()) {
|
|
buf[0] = 's';
|
|
}
|
|
if (sym.ext()) {
|
|
if (sym.weakDef() or sym.pext()) {
|
|
buf[1] = 'w';
|
|
} else {
|
|
buf[1] = 'e';
|
|
}
|
|
}
|
|
if (sym.tentative()) {
|
|
buf[2] = 't';
|
|
}
|
|
if (sym.undf()) {
|
|
buf[3] = 'u';
|
|
}
|
|
return buf[0..];
|
|
}
|
|
|
|
fn logSymtab(self: *Zld) void {
|
|
var buf: [4]u8 = undefined;
|
|
|
|
const scoped_log = std.log.scoped(.symtab);
|
|
|
|
scoped_log.debug("locals:", .{});
|
|
for (self.objects.items) |object, id| {
|
|
scoped_log.debug(" object({d}): {s}", .{ id, object.name });
|
|
if (object.in_symtab == null) continue;
|
|
for (object.symtab) |sym, sym_id| {
|
|
mem.set(u8, &buf, '_');
|
|
scoped_log.debug(" %{d}: {s} @{x} in sect({d}), {s}", .{
|
|
sym_id,
|
|
object.getSymbolName(@intCast(u32, sym_id)),
|
|
sym.n_value,
|
|
sym.n_sect,
|
|
logSymAttributes(sym, &buf),
|
|
});
|
|
}
|
|
}
|
|
scoped_log.debug(" object(-1)", .{});
|
|
for (self.locals.items) |sym, sym_id| {
|
|
if (sym.undf()) continue;
|
|
scoped_log.debug(" %{d}: {s} @{x} in sect({d}), {s}", .{
|
|
sym_id,
|
|
self.strtab.get(sym.n_strx).?,
|
|
sym.n_value,
|
|
sym.n_sect,
|
|
logSymAttributes(sym, &buf),
|
|
});
|
|
}
|
|
|
|
scoped_log.debug("exports:", .{});
|
|
for (self.globals.items) |global, i| {
|
|
const sym = self.getSymbol(global);
|
|
if (sym.undf()) continue;
|
|
if (sym.n_desc == N_DEAD) continue;
|
|
scoped_log.debug(" %{d}: {s} @{x} in sect({d}), {s} (def in object({?}))", .{
|
|
i,
|
|
self.getSymbolName(global),
|
|
sym.n_value,
|
|
sym.n_sect,
|
|
logSymAttributes(sym, &buf),
|
|
global.file,
|
|
});
|
|
}
|
|
|
|
scoped_log.debug("imports:", .{});
|
|
for (self.globals.items) |global, i| {
|
|
const sym = self.getSymbol(global);
|
|
if (!sym.undf()) continue;
|
|
if (sym.n_desc == N_DEAD) continue;
|
|
const ord = @divTrunc(sym.n_desc, macho.N_SYMBOL_RESOLVER);
|
|
scoped_log.debug(" %{d}: {s} @{x} in ord({d}), {s}", .{
|
|
i,
|
|
self.getSymbolName(global),
|
|
sym.n_value,
|
|
ord,
|
|
logSymAttributes(sym, &buf),
|
|
});
|
|
}
|
|
|
|
scoped_log.debug("GOT entries:", .{});
|
|
for (self.got_entries.items) |entry, i| {
|
|
const atom_sym = entry.getAtomSymbol(self);
|
|
const target_sym = entry.getTargetSymbol(self);
|
|
const target_sym_name = entry.getTargetSymbolName(self);
|
|
if (target_sym.undf()) {
|
|
scoped_log.debug(" {d}@{x} => import('{s}')", .{
|
|
i,
|
|
atom_sym.n_value,
|
|
target_sym_name,
|
|
});
|
|
} else {
|
|
scoped_log.debug(" {d}@{x} => local(%{d}) in object({?}) {s}", .{
|
|
i,
|
|
atom_sym.n_value,
|
|
entry.target.sym_index,
|
|
entry.target.file,
|
|
logSymAttributes(target_sym, buf[0..4]),
|
|
});
|
|
}
|
|
}
|
|
|
|
scoped_log.debug("__thread_ptrs entries:", .{});
|
|
for (self.tlv_ptr_entries.items) |entry, i| {
|
|
const atom_sym = entry.getAtomSymbol(self);
|
|
const target_sym = entry.getTargetSymbol(self);
|
|
const target_sym_name = entry.getTargetSymbolName(self);
|
|
assert(target_sym.undf());
|
|
scoped_log.debug(" {d}@{x} => import('{s}')", .{
|
|
i,
|
|
atom_sym.n_value,
|
|
target_sym_name,
|
|
});
|
|
}
|
|
|
|
scoped_log.debug("stubs entries:", .{});
|
|
for (self.stubs.items) |entry, i| {
|
|
const atom_sym = entry.getAtomSymbol(self);
|
|
const target_sym = entry.getTargetSymbol(self);
|
|
const target_sym_name = entry.getTargetSymbolName(self);
|
|
assert(target_sym.undf());
|
|
scoped_log.debug(" {d}@{x} => import('{s}')", .{
|
|
i,
|
|
atom_sym.n_value,
|
|
target_sym_name,
|
|
});
|
|
}
|
|
|
|
scoped_log.debug("thunks:", .{});
|
|
for (self.thunks.items) |thunk, i| {
|
|
scoped_log.debug(" thunk({d})", .{i});
|
|
for (thunk.lookup.keys()) |target, j| {
|
|
const target_sym = self.getSymbol(target);
|
|
const atom = self.getAtom(thunk.lookup.get(target).?);
|
|
const atom_sym = self.getSymbol(atom.getSymbolWithLoc());
|
|
scoped_log.debug(" {d}@{x} => thunk('{s}'@{x})", .{
|
|
j,
|
|
atom_sym.n_value,
|
|
self.getSymbolName(target),
|
|
target_sym.n_value,
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
fn logAtoms(self: *Zld) void {
|
|
log.debug("atoms:", .{});
|
|
const slice = self.sections.slice();
|
|
for (slice.items(.first_atom_index)) |first_atom_index, sect_id| {
|
|
var atom_index = first_atom_index;
|
|
const header = slice.items(.header)[sect_id];
|
|
|
|
log.debug("{s},{s}", .{ header.segName(), header.sectName() });
|
|
|
|
while (true) {
|
|
const atom = self.getAtom(atom_index);
|
|
self.logAtom(atom_index, log);
|
|
|
|
if (atom.next_index) |next_index| {
|
|
atom_index = next_index;
|
|
} else break;
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn logAtom(self: *Zld, atom_index: AtomIndex, logger: anytype) void {
|
|
if (!build_options.enable_logging) return;
|
|
|
|
const atom = self.getAtom(atom_index);
|
|
const sym = self.getSymbol(atom.getSymbolWithLoc());
|
|
const sym_name = self.getSymbolName(atom.getSymbolWithLoc());
|
|
logger.debug(" ATOM({d}, %{d}, '{s}') @ {x} (sizeof({x}), alignof({x})) in object({?}) in sect({d})", .{
|
|
atom_index,
|
|
atom.sym_index,
|
|
sym_name,
|
|
sym.n_value,
|
|
atom.size,
|
|
atom.alignment,
|
|
atom.file,
|
|
sym.n_sect,
|
|
});
|
|
|
|
if (atom.getFile() != null) {
|
|
var it = Atom.getInnerSymbolsIterator(self, atom_index);
|
|
while (it.next()) |sym_loc| {
|
|
const inner = self.getSymbol(sym_loc);
|
|
const inner_name = self.getSymbolName(sym_loc);
|
|
const offset = Atom.calcInnerSymbolOffset(self, atom_index, sym_loc.sym_index);
|
|
|
|
logger.debug(" (%{d}, '{s}') @ {x} ({x})", .{
|
|
sym_loc.sym_index,
|
|
inner_name,
|
|
inner.n_value,
|
|
offset,
|
|
});
|
|
}
|
|
|
|
if (Atom.getSectionAlias(self, atom_index)) |sym_loc| {
|
|
const alias = self.getSymbol(sym_loc);
|
|
const alias_name = self.getSymbolName(sym_loc);
|
|
|
|
logger.debug(" (%{d}, '{s}') @ {x} ({x})", .{
|
|
sym_loc.sym_index,
|
|
alias_name,
|
|
alias.n_value,
|
|
0,
|
|
});
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
pub const N_DEAD: u16 = @bitCast(u16, @as(i16, -1));
|
|
|
|
const Section = struct {
|
|
header: macho.section_64,
|
|
segment_index: u8,
|
|
first_atom_index: AtomIndex,
|
|
last_atom_index: AtomIndex,
|
|
};
|
|
|
|
pub const AtomIndex = u32;
|
|
|
|
const IndirectPointer = struct {
|
|
target: SymbolWithLoc,
|
|
atom_index: AtomIndex,
|
|
|
|
pub fn getTargetSymbol(self: @This(), zld: *Zld) macho.nlist_64 {
|
|
return zld.getSymbol(self.target);
|
|
}
|
|
|
|
pub fn getTargetSymbolName(self: @This(), zld: *Zld) []const u8 {
|
|
return zld.getSymbolName(self.target);
|
|
}
|
|
|
|
pub fn getAtomSymbol(self: @This(), zld: *Zld) macho.nlist_64 {
|
|
const atom = zld.getAtom(self.atom_index);
|
|
return zld.getSymbol(atom.getSymbolWithLoc());
|
|
}
|
|
};
|
|
|
|
pub const SymbolWithLoc = struct {
|
|
// Index into the respective symbol table.
|
|
sym_index: u32,
|
|
|
|
// -1 means it's a synthetic global.
|
|
file: i32 = -1,
|
|
|
|
pub inline fn getFile(self: SymbolWithLoc) ?u31 {
|
|
if (self.file == -1) return null;
|
|
return @intCast(u31, self.file);
|
|
}
|
|
|
|
pub inline fn eql(self: SymbolWithLoc, other: SymbolWithLoc) bool {
|
|
return self.file == other.file and self.sym_index == other.sym_index;
|
|
}
|
|
};
|
|
|
|
const SymbolResolver = struct {
|
|
arena: Allocator,
|
|
table: std.StringHashMap(u32),
|
|
unresolved: std.AutoArrayHashMap(u32, void),
|
|
};
|
|
|
|
pub fn linkWithZld(macho_file: *MachO, comp: *Compilation, prog_node: *std.Progress.Node) link.File.FlushError!void {
|
|
const tracy = trace(@src());
|
|
defer tracy.end();
|
|
|
|
const gpa = macho_file.base.allocator;
|
|
const options = &macho_file.base.options;
|
|
const target = options.target;
|
|
|
|
var arena_allocator = std.heap.ArenaAllocator.init(gpa);
|
|
defer arena_allocator.deinit();
|
|
const arena = arena_allocator.allocator();
|
|
|
|
const directory = options.emit.?.directory; // Just an alias to make it shorter to type.
|
|
const full_out_path = try directory.join(arena, &[_][]const u8{options.emit.?.sub_path});
|
|
|
|
// If there is no Zig code to compile, then we should skip flushing the output file because it
|
|
// will not be part of the linker line anyway.
|
|
const module_obj_path: ?[]const u8 = if (options.module != null) blk: {
|
|
try macho_file.flushModule(comp, prog_node);
|
|
|
|
if (fs.path.dirname(full_out_path)) |dirname| {
|
|
break :blk try fs.path.join(arena, &.{ dirname, macho_file.base.intermediary_basename.? });
|
|
} else {
|
|
break :blk macho_file.base.intermediary_basename.?;
|
|
}
|
|
} else null;
|
|
|
|
var sub_prog_node = prog_node.start("MachO Flush", 0);
|
|
sub_prog_node.activate();
|
|
sub_prog_node.context.refresh();
|
|
defer sub_prog_node.end();
|
|
|
|
const cpu_arch = target.cpu.arch;
|
|
const os_tag = target.os.tag;
|
|
const abi = target.abi;
|
|
const is_lib = options.output_mode == .Lib;
|
|
const is_dyn_lib = options.link_mode == .Dynamic and is_lib;
|
|
const is_exe_or_dyn_lib = is_dyn_lib or options.output_mode == .Exe;
|
|
const stack_size = options.stack_size_override orelse 0;
|
|
const is_debug_build = options.optimize_mode == .Debug;
|
|
const gc_sections = options.gc_sections orelse !is_debug_build;
|
|
|
|
const id_symlink_basename = "zld.id";
|
|
|
|
var man: Cache.Manifest = undefined;
|
|
defer if (!options.disable_lld_caching) man.deinit();
|
|
|
|
var digest: [Cache.hex_digest_len]u8 = undefined;
|
|
|
|
if (!options.disable_lld_caching) {
|
|
man = comp.cache_parent.obtain();
|
|
|
|
// We are about to obtain this lock, so here we give other processes a chance first.
|
|
macho_file.base.releaseLock();
|
|
|
|
comptime assert(Compilation.link_hash_implementation_version == 7);
|
|
|
|
for (options.objects) |obj| {
|
|
_ = try man.addFile(obj.path, null);
|
|
man.hash.add(obj.must_link);
|
|
}
|
|
for (comp.c_object_table.keys()) |key| {
|
|
_ = try man.addFile(key.status.success.object_path, null);
|
|
}
|
|
try man.addOptionalFile(module_obj_path);
|
|
// We can skip hashing libc and libc++ components that we are in charge of building from Zig
|
|
// installation sources because they are always a product of the compiler version + target information.
|
|
man.hash.add(stack_size);
|
|
man.hash.addOptional(options.pagezero_size);
|
|
man.hash.addOptional(options.search_strategy);
|
|
man.hash.addOptional(options.headerpad_size);
|
|
man.hash.add(options.headerpad_max_install_names);
|
|
man.hash.add(gc_sections);
|
|
man.hash.add(options.dead_strip_dylibs);
|
|
man.hash.add(options.strip);
|
|
man.hash.addListOfBytes(options.lib_dirs);
|
|
man.hash.addListOfBytes(options.framework_dirs);
|
|
link.hashAddSystemLibs(&man.hash, options.frameworks);
|
|
man.hash.addListOfBytes(options.rpath_list);
|
|
if (is_dyn_lib) {
|
|
man.hash.addOptionalBytes(options.install_name);
|
|
man.hash.addOptional(options.version);
|
|
}
|
|
link.hashAddSystemLibs(&man.hash, options.system_libs);
|
|
man.hash.addOptionalBytes(options.sysroot);
|
|
try man.addOptionalFile(options.entitlements);
|
|
|
|
// We don't actually care whether it's a cache hit or miss; we just need the digest and the lock.
|
|
_ = try man.hit();
|
|
digest = man.final();
|
|
|
|
var prev_digest_buf: [digest.len]u8 = undefined;
|
|
const prev_digest: []u8 = Cache.readSmallFile(
|
|
directory.handle,
|
|
id_symlink_basename,
|
|
&prev_digest_buf,
|
|
) catch |err| blk: {
|
|
log.debug("MachO Zld new_digest={s} error: {s}", .{
|
|
std.fmt.fmtSliceHexLower(&digest),
|
|
@errorName(err),
|
|
});
|
|
// Handle this as a cache miss.
|
|
break :blk prev_digest_buf[0..0];
|
|
};
|
|
if (mem.eql(u8, prev_digest, &digest)) {
|
|
// Hot diggity dog! The output binary is already there.
|
|
log.debug("MachO Zld digest={s} match - skipping invocation", .{
|
|
std.fmt.fmtSliceHexLower(&digest),
|
|
});
|
|
macho_file.base.lock = man.toOwnedLock();
|
|
return;
|
|
}
|
|
log.debug("MachO Zld prev_digest={s} new_digest={s}", .{
|
|
std.fmt.fmtSliceHexLower(prev_digest),
|
|
std.fmt.fmtSliceHexLower(&digest),
|
|
});
|
|
|
|
// We are about to change the output file to be different, so we invalidate the build hash now.
|
|
directory.handle.deleteFile(id_symlink_basename) catch |err| switch (err) {
|
|
error.FileNotFound => {},
|
|
else => |e| return e,
|
|
};
|
|
}
|
|
|
|
if (options.output_mode == .Obj) {
|
|
// LLD's MachO driver does not support the equivalent of `-r` so we do a simple file copy
|
|
// here. TODO: think carefully about how we can avoid this redundant operation when doing
|
|
// build-obj. See also the corresponding TODO in linkAsArchive.
|
|
const the_object_path = blk: {
|
|
if (options.objects.len != 0) {
|
|
break :blk options.objects[0].path;
|
|
}
|
|
|
|
if (comp.c_object_table.count() != 0)
|
|
break :blk comp.c_object_table.keys()[0].status.success.object_path;
|
|
|
|
if (module_obj_path) |p|
|
|
break :blk p;
|
|
|
|
// TODO I think this is unreachable. Audit this situation when solving the above TODO
|
|
// regarding eliding redundant object -> object transformations.
|
|
return error.NoObjectsToLink;
|
|
};
|
|
// This can happen when using --enable-cache and using the stage1 backend. In this case
|
|
// we can skip the file copy.
|
|
if (!mem.eql(u8, the_object_path, full_out_path)) {
|
|
try fs.cwd().copyFile(the_object_path, fs.cwd(), full_out_path, .{});
|
|
}
|
|
} else {
|
|
const page_size = macho_file.page_size;
|
|
const sub_path = options.emit.?.sub_path;
|
|
if (macho_file.base.file == null) {
|
|
macho_file.base.file = try directory.handle.createFile(sub_path, .{
|
|
.truncate = true,
|
|
.read = true,
|
|
.mode = link.determineMode(options.*),
|
|
});
|
|
}
|
|
var zld = Zld{
|
|
.gpa = gpa,
|
|
.file = macho_file.base.file.?,
|
|
.page_size = macho_file.page_size,
|
|
.options = options,
|
|
};
|
|
defer zld.deinit();
|
|
|
|
try zld.atoms.append(gpa, Atom.empty); // AtomIndex at 0 is reserved as null atom
|
|
try zld.strtab.buffer.append(gpa, 0);
|
|
|
|
var lib_not_found = false;
|
|
var framework_not_found = false;
|
|
|
|
// Positional arguments to the linker such as object files and static archives.
|
|
var positionals = std.ArrayList([]const u8).init(arena);
|
|
try positionals.ensureUnusedCapacity(options.objects.len);
|
|
|
|
var must_link_archives = std.StringArrayHashMap(void).init(arena);
|
|
try must_link_archives.ensureUnusedCapacity(options.objects.len);
|
|
|
|
for (options.objects) |obj| {
|
|
if (must_link_archives.contains(obj.path)) continue;
|
|
if (obj.must_link) {
|
|
_ = must_link_archives.getOrPutAssumeCapacity(obj.path);
|
|
} else {
|
|
_ = positionals.appendAssumeCapacity(obj.path);
|
|
}
|
|
}
|
|
|
|
for (comp.c_object_table.keys()) |key| {
|
|
try positionals.append(key.status.success.object_path);
|
|
}
|
|
|
|
if (module_obj_path) |p| {
|
|
try positionals.append(p);
|
|
}
|
|
|
|
if (comp.compiler_rt_lib) |lib| {
|
|
try positionals.append(lib.full_object_path);
|
|
}
|
|
|
|
// libc++ dep
|
|
if (options.link_libcpp) {
|
|
try positionals.append(comp.libcxxabi_static_lib.?.full_object_path);
|
|
try positionals.append(comp.libcxx_static_lib.?.full_object_path);
|
|
}
|
|
|
|
// Shared and static libraries passed via `-l` flag.
|
|
var candidate_libs = std.StringArrayHashMap(link.SystemLib).init(arena);
|
|
|
|
const system_lib_names = options.system_libs.keys();
|
|
for (system_lib_names) |system_lib_name| {
|
|
// By this time, we depend on these libs being dynamically linked libraries and not static libraries
|
|
// (the check for that needs to be earlier), but they could be full paths to .dylib files, in which
|
|
// case we want to avoid prepending "-l".
|
|
if (Compilation.classifyFileExt(system_lib_name) == .shared_library) {
|
|
try positionals.append(system_lib_name);
|
|
continue;
|
|
}
|
|
|
|
const system_lib_info = options.system_libs.get(system_lib_name).?;
|
|
try candidate_libs.put(system_lib_name, .{
|
|
.needed = system_lib_info.needed,
|
|
.weak = system_lib_info.weak,
|
|
});
|
|
}
|
|
|
|
var lib_dirs = std.ArrayList([]const u8).init(arena);
|
|
for (options.lib_dirs) |dir| {
|
|
if (try MachO.resolveSearchDir(arena, dir, options.sysroot)) |search_dir| {
|
|
try lib_dirs.append(search_dir);
|
|
} else {
|
|
log.warn("directory not found for '-L{s}'", .{dir});
|
|
}
|
|
}
|
|
|
|
var libs = std.StringArrayHashMap(link.SystemLib).init(arena);
|
|
|
|
// Assume ld64 default -search_paths_first if no strategy specified.
|
|
const search_strategy = options.search_strategy orelse .paths_first;
|
|
outer: for (candidate_libs.keys()) |lib_name| {
|
|
switch (search_strategy) {
|
|
.paths_first => {
|
|
// Look in each directory for a dylib (stub first), and then for archive
|
|
for (lib_dirs.items) |dir| {
|
|
for (&[_][]const u8{ ".tbd", ".dylib", ".a" }) |ext| {
|
|
if (try MachO.resolveLib(arena, dir, lib_name, ext)) |full_path| {
|
|
try libs.put(full_path, candidate_libs.get(lib_name).?);
|
|
continue :outer;
|
|
}
|
|
}
|
|
} else {
|
|
log.warn("library not found for '-l{s}'", .{lib_name});
|
|
lib_not_found = true;
|
|
}
|
|
},
|
|
.dylibs_first => {
|
|
// First, look for a dylib in each search dir
|
|
for (lib_dirs.items) |dir| {
|
|
for (&[_][]const u8{ ".tbd", ".dylib" }) |ext| {
|
|
if (try MachO.resolveLib(arena, dir, lib_name, ext)) |full_path| {
|
|
try libs.put(full_path, candidate_libs.get(lib_name).?);
|
|
continue :outer;
|
|
}
|
|
}
|
|
} else for (lib_dirs.items) |dir| {
|
|
if (try MachO.resolveLib(arena, dir, lib_name, ".a")) |full_path| {
|
|
try libs.put(full_path, candidate_libs.get(lib_name).?);
|
|
} else {
|
|
log.warn("library not found for '-l{s}'", .{lib_name});
|
|
lib_not_found = true;
|
|
}
|
|
}
|
|
},
|
|
}
|
|
}
|
|
|
|
if (lib_not_found) {
|
|
log.warn("Library search paths:", .{});
|
|
for (lib_dirs.items) |dir| {
|
|
log.warn(" {s}", .{dir});
|
|
}
|
|
}
|
|
|
|
try MachO.resolveLibSystem(arena, comp, options.sysroot, target, lib_dirs.items, &libs);
|
|
|
|
// frameworks
|
|
var framework_dirs = std.ArrayList([]const u8).init(arena);
|
|
for (options.framework_dirs) |dir| {
|
|
if (try MachO.resolveSearchDir(arena, dir, options.sysroot)) |search_dir| {
|
|
try framework_dirs.append(search_dir);
|
|
} else {
|
|
log.warn("directory not found for '-F{s}'", .{dir});
|
|
}
|
|
}
|
|
|
|
outer: for (options.frameworks.keys()) |f_name| {
|
|
for (framework_dirs.items) |dir| {
|
|
for (&[_][]const u8{ ".tbd", ".dylib", "" }) |ext| {
|
|
if (try MachO.resolveFramework(arena, dir, f_name, ext)) |full_path| {
|
|
const info = options.frameworks.get(f_name).?;
|
|
try libs.put(full_path, .{
|
|
.needed = info.needed,
|
|
.weak = info.weak,
|
|
});
|
|
continue :outer;
|
|
}
|
|
}
|
|
} else {
|
|
log.warn("framework not found for '-framework {s}'", .{f_name});
|
|
framework_not_found = true;
|
|
}
|
|
}
|
|
|
|
if (framework_not_found) {
|
|
log.warn("Framework search paths:", .{});
|
|
for (framework_dirs.items) |dir| {
|
|
log.warn(" {s}", .{dir});
|
|
}
|
|
}
|
|
|
|
if (options.verbose_link) {
|
|
var argv = std.ArrayList([]const u8).init(arena);
|
|
|
|
try argv.append("zig");
|
|
try argv.append("ld");
|
|
|
|
if (is_exe_or_dyn_lib) {
|
|
try argv.append("-dynamic");
|
|
}
|
|
|
|
if (is_dyn_lib) {
|
|
try argv.append("-dylib");
|
|
|
|
if (options.install_name) |install_name| {
|
|
try argv.append("-install_name");
|
|
try argv.append(install_name);
|
|
}
|
|
}
|
|
|
|
if (options.sysroot) |syslibroot| {
|
|
try argv.append("-syslibroot");
|
|
try argv.append(syslibroot);
|
|
}
|
|
|
|
for (options.rpath_list) |rpath| {
|
|
try argv.append("-rpath");
|
|
try argv.append(rpath);
|
|
}
|
|
|
|
if (options.pagezero_size) |pagezero_size| {
|
|
try argv.append("-pagezero_size");
|
|
try argv.append(try std.fmt.allocPrint(arena, "0x{x}", .{pagezero_size}));
|
|
}
|
|
|
|
if (options.search_strategy) |strat| switch (strat) {
|
|
.paths_first => try argv.append("-search_paths_first"),
|
|
.dylibs_first => try argv.append("-search_dylibs_first"),
|
|
};
|
|
|
|
if (options.headerpad_size) |headerpad_size| {
|
|
try argv.append("-headerpad_size");
|
|
try argv.append(try std.fmt.allocPrint(arena, "0x{x}", .{headerpad_size}));
|
|
}
|
|
|
|
if (options.headerpad_max_install_names) {
|
|
try argv.append("-headerpad_max_install_names");
|
|
}
|
|
|
|
if (gc_sections) {
|
|
try argv.append("-dead_strip");
|
|
}
|
|
|
|
if (options.dead_strip_dylibs) {
|
|
try argv.append("-dead_strip_dylibs");
|
|
}
|
|
|
|
if (options.entry) |entry| {
|
|
try argv.append("-e");
|
|
try argv.append(entry);
|
|
}
|
|
|
|
for (options.objects) |obj| {
|
|
try argv.append(obj.path);
|
|
}
|
|
|
|
for (comp.c_object_table.keys()) |key| {
|
|
try argv.append(key.status.success.object_path);
|
|
}
|
|
|
|
if (module_obj_path) |p| {
|
|
try argv.append(p);
|
|
}
|
|
|
|
if (comp.compiler_rt_lib) |lib| {
|
|
try argv.append(lib.full_object_path);
|
|
}
|
|
|
|
if (options.link_libcpp) {
|
|
try argv.append(comp.libcxxabi_static_lib.?.full_object_path);
|
|
try argv.append(comp.libcxx_static_lib.?.full_object_path);
|
|
}
|
|
|
|
try argv.append("-o");
|
|
try argv.append(full_out_path);
|
|
|
|
try argv.append("-lSystem");
|
|
try argv.append("-lc");
|
|
|
|
for (options.system_libs.keys()) |l_name| {
|
|
const info = options.system_libs.get(l_name).?;
|
|
const arg = if (info.needed)
|
|
try std.fmt.allocPrint(arena, "-needed-l{s}", .{l_name})
|
|
else if (info.weak)
|
|
try std.fmt.allocPrint(arena, "-weak-l{s}", .{l_name})
|
|
else
|
|
try std.fmt.allocPrint(arena, "-l{s}", .{l_name});
|
|
try argv.append(arg);
|
|
}
|
|
|
|
for (options.lib_dirs) |lib_dir| {
|
|
try argv.append(try std.fmt.allocPrint(arena, "-L{s}", .{lib_dir}));
|
|
}
|
|
|
|
for (options.frameworks.keys()) |framework| {
|
|
const info = options.frameworks.get(framework).?;
|
|
const arg = if (info.needed)
|
|
try std.fmt.allocPrint(arena, "-needed_framework {s}", .{framework})
|
|
else if (info.weak)
|
|
try std.fmt.allocPrint(arena, "-weak_framework {s}", .{framework})
|
|
else
|
|
try std.fmt.allocPrint(arena, "-framework {s}", .{framework});
|
|
try argv.append(arg);
|
|
}
|
|
|
|
for (options.framework_dirs) |framework_dir| {
|
|
try argv.append(try std.fmt.allocPrint(arena, "-F{s}", .{framework_dir}));
|
|
}
|
|
|
|
if (is_dyn_lib and (options.allow_shlib_undefined orelse false)) {
|
|
try argv.append("-undefined");
|
|
try argv.append("dynamic_lookup");
|
|
}
|
|
|
|
for (must_link_archives.keys()) |lib| {
|
|
try argv.append(try std.fmt.allocPrint(arena, "-force_load {s}", .{lib}));
|
|
}
|
|
|
|
Compilation.dump_argv(argv.items);
|
|
}
|
|
|
|
var dependent_libs = std.fifo.LinearFifo(struct {
|
|
id: Dylib.Id,
|
|
parent: u16,
|
|
}, .Dynamic).init(arena);
|
|
|
|
try zld.parseInputFiles(positionals.items, options.sysroot, &dependent_libs);
|
|
try zld.parseAndForceLoadStaticArchives(must_link_archives.keys());
|
|
try zld.parseLibs(libs.keys(), libs.values(), options.sysroot, &dependent_libs);
|
|
try zld.parseDependentLibs(options.sysroot, &dependent_libs);
|
|
|
|
var resolver = SymbolResolver{
|
|
.arena = arena,
|
|
.table = std.StringHashMap(u32).init(arena),
|
|
.unresolved = std.AutoArrayHashMap(u32, void).init(arena),
|
|
};
|
|
|
|
for (zld.objects.items) |_, object_id| {
|
|
try zld.resolveSymbolsInObject(@intCast(u16, object_id), &resolver);
|
|
}
|
|
|
|
try zld.resolveSymbolsInArchives(&resolver);
|
|
try zld.resolveDyldStubBinder(&resolver);
|
|
try zld.resolveSymbolsInDylibs(&resolver);
|
|
try zld.createMhExecuteHeaderSymbol(&resolver);
|
|
try zld.createDsoHandleSymbol(&resolver);
|
|
try zld.resolveSymbolsAtLoading(&resolver);
|
|
|
|
if (resolver.unresolved.count() > 0) {
|
|
return error.UndefinedSymbolReference;
|
|
}
|
|
if (lib_not_found) {
|
|
return error.LibraryNotFound;
|
|
}
|
|
if (framework_not_found) {
|
|
return error.FrameworkNotFound;
|
|
}
|
|
|
|
if (options.output_mode == .Exe) {
|
|
const entry_name = options.entry orelse "_main";
|
|
const global_index = resolver.table.get(entry_name) orelse {
|
|
log.err("entrypoint '{s}' not found", .{entry_name});
|
|
return error.MissingMainEntrypoint;
|
|
};
|
|
zld.entry_index = global_index;
|
|
}
|
|
|
|
for (zld.objects.items) |*object, object_id| {
|
|
try object.splitIntoAtoms(&zld, @intCast(u31, object_id));
|
|
}
|
|
|
|
var reverse_lookups: [][]u32 = try arena.alloc([]u32, zld.objects.items.len);
|
|
for (zld.objects.items) |object, i| {
|
|
reverse_lookups[i] = try object.createReverseSymbolLookup(arena);
|
|
}
|
|
|
|
if (gc_sections) {
|
|
try dead_strip.gcAtoms(&zld, reverse_lookups);
|
|
}
|
|
|
|
try zld.createDyldPrivateAtom();
|
|
try zld.createTentativeDefAtoms();
|
|
try zld.createStubHelperPreambleAtom();
|
|
|
|
for (zld.objects.items) |object| {
|
|
for (object.atoms.items) |atom_index| {
|
|
const atom = zld.getAtom(atom_index);
|
|
const sym = zld.getSymbol(atom.getSymbolWithLoc());
|
|
const header = zld.sections.items(.header)[sym.n_sect - 1];
|
|
if (header.isZerofill()) continue;
|
|
|
|
const relocs = Atom.getAtomRelocs(&zld, atom_index);
|
|
try Atom.scanAtomRelocs(&zld, atom_index, relocs, reverse_lookups[atom.getFile().?]);
|
|
}
|
|
}
|
|
|
|
try zld.createDyldStubBinderGotAtom();
|
|
|
|
try zld.calcSectionSizes(reverse_lookups);
|
|
try zld.pruneAndSortSections();
|
|
try zld.createSegments();
|
|
try zld.allocateSegments();
|
|
|
|
try zld.allocateSpecialSymbols();
|
|
|
|
if (build_options.enable_logging) {
|
|
zld.logSymtab();
|
|
zld.logSegments();
|
|
zld.logSections();
|
|
zld.logAtoms();
|
|
}
|
|
|
|
try zld.writeAtoms(reverse_lookups);
|
|
|
|
var lc_buffer = std.ArrayList(u8).init(arena);
|
|
const lc_writer = lc_buffer.writer();
|
|
|
|
var ncmds: u32 = 0;
|
|
|
|
try zld.writeLinkeditSegmentData(&ncmds, lc_writer, reverse_lookups);
|
|
|
|
// If the last section of __DATA segment is zerofill section, we need to ensure
|
|
// that the free space between the end of the last non-zerofill section of __DATA
|
|
// segment and the beginning of __LINKEDIT segment is zerofilled as the loader will
|
|
// copy-paste this space into memory for quicker zerofill operation.
|
|
if (zld.getSegmentByName("__DATA")) |data_seg_id| blk: {
|
|
var physical_zerofill_start: ?u64 = null;
|
|
const section_indexes = zld.getSectionIndexes(data_seg_id);
|
|
for (zld.sections.items(.header)[section_indexes.start..section_indexes.end]) |header| {
|
|
if (header.isZerofill() and header.size > 0) break;
|
|
physical_zerofill_start = header.offset + header.size;
|
|
} else break :blk;
|
|
const start = physical_zerofill_start orelse break :blk;
|
|
const linkedit = zld.getLinkeditSegmentPtr();
|
|
const size = math.cast(usize, linkedit.fileoff - start) orelse return error.Overflow;
|
|
if (size > 0) {
|
|
log.debug("zeroing out zerofill area of length {x} at {x}", .{ size, start });
|
|
var padding = try zld.gpa.alloc(u8, size);
|
|
defer zld.gpa.free(padding);
|
|
mem.set(u8, padding, 0);
|
|
try zld.file.pwriteAll(padding, start);
|
|
}
|
|
}
|
|
|
|
try load_commands.writeDylinkerLC(&ncmds, lc_writer);
|
|
|
|
if (zld.options.output_mode == .Exe) {
|
|
const seg_id = zld.getSegmentByName("__TEXT").?;
|
|
const seg = zld.segments.items[seg_id];
|
|
const global = zld.getEntryPoint();
|
|
const sym = zld.getSymbol(global);
|
|
try load_commands.writeMainLC(@intCast(u32, sym.n_value - seg.vmaddr), options, &ncmds, lc_writer);
|
|
} else {
|
|
assert(zld.options.output_mode == .Lib);
|
|
try load_commands.writeDylibIdLC(zld.gpa, zld.options, &ncmds, lc_writer);
|
|
}
|
|
|
|
try load_commands.writeRpathLCs(zld.gpa, zld.options, &ncmds, lc_writer);
|
|
try load_commands.writeSourceVersionLC(&ncmds, lc_writer);
|
|
try load_commands.writeBuildVersionLC(zld.options, &ncmds, lc_writer);
|
|
|
|
// Looking forward into the future, we will want to offer `-no_uuid` support in which case
|
|
// there will be nothing to backpatch.
|
|
const uuid_offset_backpatch: ?usize = blk: {
|
|
const index = lc_buffer.items.len;
|
|
var uuid_buf: [16]u8 = [_]u8{0} ** 16;
|
|
try load_commands.writeUuidLC(&uuid_buf, &ncmds, lc_writer);
|
|
break :blk index;
|
|
};
|
|
|
|
try load_commands.writeLoadDylibLCs(zld.dylibs.items, zld.referenced_dylibs.keys(), &ncmds, lc_writer);
|
|
|
|
const requires_codesig = blk: {
|
|
if (options.entitlements) |_| break :blk true;
|
|
if (cpu_arch == .aarch64 and (os_tag == .macos or abi == .simulator)) break :blk true;
|
|
break :blk false;
|
|
};
|
|
var codesig_offset: ?u32 = null;
|
|
var codesig: ?CodeSignature = if (requires_codesig) blk: {
|
|
// Preallocate space for the code signature.
|
|
// We need to do this at this stage so that we have the load commands with proper values
|
|
// written out to the file.
|
|
// The most important here is to have the correct vm and filesize of the __LINKEDIT segment
|
|
// where the code signature goes into.
|
|
var codesig = CodeSignature.init(page_size);
|
|
codesig.code_directory.ident = options.emit.?.sub_path;
|
|
if (options.entitlements) |path| {
|
|
try codesig.addEntitlements(gpa, path);
|
|
}
|
|
codesig_offset = try zld.writeCodeSignaturePadding(&codesig, &ncmds, lc_writer);
|
|
break :blk codesig;
|
|
} else null;
|
|
defer if (codesig) |*csig| csig.deinit(gpa);
|
|
|
|
var headers_buf = std.ArrayList(u8).init(arena);
|
|
try zld.writeSegmentHeaders(&ncmds, headers_buf.writer());
|
|
|
|
try zld.file.pwriteAll(headers_buf.items, @sizeOf(macho.mach_header_64));
|
|
try zld.file.pwriteAll(lc_buffer.items, @sizeOf(macho.mach_header_64) + headers_buf.items.len);
|
|
try zld.writeHeader(ncmds, @intCast(u32, lc_buffer.items.len + headers_buf.items.len));
|
|
|
|
if (uuid_offset_backpatch) |backpatch| {
|
|
const seg = zld.getLinkeditSegmentPtr();
|
|
const file_size = seg.fileoff + seg.filesize;
|
|
var uuid_buf: [16]u8 = undefined;
|
|
try uuid.calcMd5Hash(zld.gpa, zld.file, file_size, &uuid_buf);
|
|
const offset = @sizeOf(macho.mach_header_64) + headers_buf.items.len + backpatch + @sizeOf(macho.load_command);
|
|
try zld.file.pwriteAll(&uuid_buf, offset);
|
|
}
|
|
|
|
if (codesig) |*csig| {
|
|
try zld.writeCodeSignature(comp, csig, codesig_offset.?); // code signing always comes last
|
|
}
|
|
}
|
|
|
|
if (!options.disable_lld_caching) {
|
|
// Update the file with the digest. If it fails we can continue; it only
|
|
// means that the next invocation will have an unnecessary cache miss.
|
|
Cache.writeSmallFile(directory.handle, id_symlink_basename, &digest) catch |err| {
|
|
log.debug("failed to save linking hash digest file: {s}", .{@errorName(err)});
|
|
};
|
|
// Again failure here only means an unnecessary cache miss.
|
|
man.writeManifest() catch |err| {
|
|
log.debug("failed to write cache manifest when linking: {s}", .{@errorName(err)});
|
|
};
|
|
// We hang on to this lock so that the output file path can be used without
|
|
// other processes clobbering it.
|
|
macho_file.base.lock = man.toOwnedLock();
|
|
}
|
|
}
|
|
|
|
/// Binary search
|
|
pub fn bsearch(comptime T: type, haystack: []align(1) const T, predicate: anytype) usize {
|
|
if (!@hasDecl(@TypeOf(predicate), "predicate"))
|
|
@compileError("Predicate is required to define fn predicate(@This(), T) bool");
|
|
|
|
var min: usize = 0;
|
|
var max: usize = haystack.len;
|
|
while (min < max) {
|
|
const index = (min + max) / 2;
|
|
const curr = haystack[index];
|
|
if (predicate.predicate(curr)) {
|
|
min = index + 1;
|
|
} else {
|
|
max = index;
|
|
}
|
|
}
|
|
return min;
|
|
}
|
|
|
|
/// Linear search
|
|
pub fn lsearch(comptime T: type, haystack: []align(1) const T, predicate: anytype) usize {
|
|
if (!@hasDecl(@TypeOf(predicate), "predicate"))
|
|
@compileError("Predicate is required to define fn predicate(@This(), T) bool");
|
|
|
|
var i: usize = 0;
|
|
while (i < haystack.len) : (i += 1) {
|
|
if (predicate.predicate(haystack[i])) break;
|
|
}
|
|
return i;
|
|
}
|