| ... | ... | @@ -726,32 +726,38 @@ fn printSourceAtAddressMacOs(di: *DebugInfo, out_stream: var, address: usize, tt |
| 726 | 726 | } |
| 727 | 727 | |
| 728 | 728 | pub fn printSourceAtAddressPosix(debug_info: *DebugInfo, out_stream: var, address: usize, tty_config: TTY.Config) !void { |
| 729 | | const compile_unit = debug_info.findCompileUnit(address) catch { |
| 729 | // XXX Print as much as possible anyway |
| 730 | const module = try debug_info.lookupByAddress(address); |
| 731 | |
| 732 | const reloc_address = address - module.base_address; |
| 733 | warn("reloc {x} => {x}\n", .{ address, reloc_address }); |
| 734 | |
| 735 | if (module.dwarf.findCompileUnit(reloc_address) catch null) |compile_unit| { |
| 736 | const compile_unit_name = try compile_unit.die.getAttrString(&module.dwarf, DW.AT_name); |
| 737 | const symbol_name = module.dwarf.getSymbolName(reloc_address) orelse "???"; |
| 738 | const line_info = module.dwarf.getLineNumberInfo(compile_unit.*, reloc_address) catch |err| switch (err) { |
| 739 | error.MissingDebugInfo, error.InvalidDebugInfo => null, |
| 740 | else => return err, |
| 741 | }; |
| 742 | defer if (line_info) |li| li.deinit(); |
| 743 | |
| 730 | 744 | return printLineInfo( |
| 731 | 745 | out_stream, |
| 732 | | null, |
| 746 | line_info, |
| 733 | 747 | address, |
| 734 | | "???", |
| 735 | | "???", |
| 748 | symbol_name, |
| 749 | compile_unit_name, |
| 736 | 750 | tty_config, |
| 737 | 751 | printLineFromFileAnyOs, |
| 738 | 752 | ); |
| 739 | | }; |
| 740 | | |
| 741 | | const compile_unit_name = try compile_unit.die.getAttrString(debug_info, DW.AT_name); |
| 742 | | const symbol_name = debug_info.getSymbolName(address) orelse "???"; |
| 743 | | const line_info = debug_info.getLineNumberInfo(compile_unit.*, address) catch |err| switch (err) { |
| 744 | | error.MissingDebugInfo, error.InvalidDebugInfo => null, |
| 745 | | else => return err, |
| 746 | | }; |
| 747 | | defer if (line_info) |li| li.deinit(); |
| 753 | } |
| 748 | 754 | |
| 749 | | try printLineInfo( |
| 755 | return printLineInfo( |
| 750 | 756 | out_stream, |
| 751 | | line_info, |
| 757 | null, |
| 752 | 758 | address, |
| 753 | | symbol_name, |
| 754 | | compile_unit_name, |
| 759 | "???", |
| 760 | "???", |
| 755 | 761 | tty_config, |
| 756 | 762 | printLineFromFileAnyOs, |
| 757 | 763 | ); |
| ... | ... | @@ -824,6 +830,10 @@ pub fn openSelfDebugInfo(allocator: *mem.Allocator) !DebugInfo { |
| 824 | 830 | if (comptime std.Target.current.isDarwin()) { |
| 825 | 831 | return noasync openSelfDebugInfoMacOs(allocator); |
| 826 | 832 | } |
| 833 | if (builtin.os == .linux) { |
| 834 | _ = try allocator.create(u32); |
| 835 | return DebugInfo.init(allocator); |
| 836 | } |
| 827 | 837 | return noasync openSelfDebugInfoPosix(allocator); |
| 828 | 838 | } |
| 829 | 839 | |
| ... | ... | @@ -1190,7 +1200,113 @@ const MachoSymbol = struct { |
| 1190 | 1200 | } |
| 1191 | 1201 | }; |
| 1192 | 1202 | |
| 1193 | | pub const DebugInfo = switch (builtin.os) { |
| 1203 | pub const DebugInfo = struct { |
| 1204 | allocator: *mem.Allocator, |
| 1205 | address_map: std.StringHashMap(*ObjectDebugInfo), |
| 1206 | |
| 1207 | pub fn init(allocator: *mem.Allocator) DebugInfo { |
| 1208 | return DebugInfo{ |
| 1209 | .allocator = allocator, |
| 1210 | .address_map = std.StringHashMap(*ObjectDebugInfo).init(allocator), |
| 1211 | }; |
| 1212 | } |
| 1213 | |
| 1214 | pub fn deinit(self: *DebugInfo) void { |
| 1215 | self.address_map.deinit(); |
| 1216 | } |
| 1217 | |
| 1218 | pub fn lookupByAddress(self: *DebugInfo, address: usize) !*ObjectDebugInfo { |
| 1219 | const obj_di = try self.lookupModuleDl(address); |
| 1220 | return obj_di; |
| 1221 | } |
| 1222 | |
| 1223 | fn lookupModuleDl(self: *DebugInfo, address: usize) !*ObjectDebugInfo { |
| 1224 | var ctx = DIPContext{ .address = address }; |
| 1225 | |
| 1226 | warn("lookup {x}\n", .{address}); |
| 1227 | |
| 1228 | // XXX Locking? |
| 1229 | if (os.dl_iterate_phdr(DIPContext, dl_iterate_phdr_callback, &ctx) == 0) |
| 1230 | return error.DebugInfoNotFound; |
| 1231 | |
| 1232 | warn("found in \"{s}\"\n", .{ctx.name}); |
| 1233 | |
| 1234 | if (self.address_map.getValue(ctx.name)) |obj_di| { |
| 1235 | warn("cache hit!\n", .{}); |
| 1236 | return obj_di; |
| 1237 | } |
| 1238 | |
| 1239 | const exe_file = if (ctx.name.len > 0) |
| 1240 | try fs.openFileAbsolute(ctx.name, .{}) |
| 1241 | else |
| 1242 | try fs.openSelfExe(); |
| 1243 | defer exe_file.close(); |
| 1244 | |
| 1245 | const exe_len = math.cast(usize, try exe_file.getEndPos()) catch |
| 1246 | return error.DebugInfoTooLarge; |
| 1247 | const exe_mmap = try os.mmap( |
| 1248 | null, |
| 1249 | exe_len, |
| 1250 | os.PROT_READ, |
| 1251 | os.MAP_SHARED, |
| 1252 | exe_file.handle, |
| 1253 | 0, |
| 1254 | ); |
| 1255 | errdefer os.munmap(exe_mmap); |
| 1256 | |
| 1257 | const obj_di = try self.allocator.create(ObjectDebugInfo); |
| 1258 | errdefer self.allocator.destroy(obj_di); |
| 1259 | |
| 1260 | try self.address_map.putNoClobber(ctx.name, obj_di); |
| 1261 | |
| 1262 | obj_di.* = .{ |
| 1263 | .dwarf = try openElfDebugInfo(self.allocator, exe_mmap), |
| 1264 | .mapped_memory = exe_mmap, |
| 1265 | .base_address = ctx.base_address, |
| 1266 | }; |
| 1267 | |
| 1268 | return obj_di; |
| 1269 | } |
| 1270 | }; |
| 1271 | |
| 1272 | const DIPContext = struct { |
| 1273 | address: usize, |
| 1274 | base_address: usize = undefined, |
| 1275 | name: []const u8 = undefined, |
| 1276 | }; |
| 1277 | |
| 1278 | fn dl_iterate_phdr_callback(info: *os.dl_phdr_info, size: usize, context: ?*DIPContext) callconv(.C) i32 { |
| 1279 | const address = context.?.address; |
| 1280 | |
| 1281 | // The base address is too high |
| 1282 | if (address < info.dlpi_addr) |
| 1283 | return 0; |
| 1284 | |
| 1285 | const phdrs = info.dlpi_phdr[0..info.dlpi_phnum]; |
| 1286 | for (phdrs) |*phdr| { |
| 1287 | if (phdr.p_type != elf.PT_LOAD) continue; |
| 1288 | |
| 1289 | const seg_start = info.dlpi_addr + phdr.p_vaddr; |
| 1290 | const seg_end = seg_start + phdr.p_memsz; |
| 1291 | |
| 1292 | if (address > seg_start and address <= seg_end) { |
| 1293 | // Android libc uses NULL instead of an empty string to mark the |
| 1294 | // main program |
| 1295 | context.?.name = if (info.dlpi_name) |dlpi_name| |
| 1296 | mem.toSliceConst(u8, dlpi_name) |
| 1297 | else |
| 1298 | ""; |
| 1299 | context.?.base_address = info.dlpi_addr; |
| 1300 | // Stop the iteration |
| 1301 | return 1; |
| 1302 | } |
| 1303 | } |
| 1304 | |
| 1305 | // Continue the iteration |
| 1306 | return 0; |
| 1307 | } |
| 1308 | |
| 1309 | pub const ObjectDebugInfo = switch (builtin.os) { |
| 1194 | 1310 | .macosx, .ios, .watchos, .tvos => struct { |
| 1195 | 1311 | symbols: []const MachoSymbol, |
| 1196 | 1312 | strings: []const u8, |
| ... | ... | @@ -1213,6 +1329,11 @@ pub const DebugInfo = switch (builtin.os) { |
| 1213 | 1329 | sect_contribs: []pdb.SectionContribEntry, |
| 1214 | 1330 | modules: []Module, |
| 1215 | 1331 | }, |
| 1332 | .linux => struct { |
| 1333 | base_address: usize, |
| 1334 | dwarf: DW.DwarfInfo, |
| 1335 | mapped_memory: []u8, |
| 1336 | }, |
| 1216 | 1337 | else => DW.DwarfInfo, |
| 1217 | 1338 | }; |
| 1218 | 1339 | |
LemonBoy <thatlemon@gmail.com> 2020-02-09 11:55:43+01:00