diff --git a/lib/DebugInfo/DWARFDebugArangeSet.cpp b/lib/DebugInfo/DWARFDebugArangeSet.cpp index a3e2e62104c..148b770bd84 100644 --- a/lib/DebugInfo/DWARFDebugArangeSet.cpp +++ b/lib/DebugInfo/DWARFDebugArangeSet.cpp @@ -72,6 +72,13 @@ DWARFDebugArangeSet::extract(DataExtractor data, uint32_t *offset_ptr) { Header.AddrSize = data.getU8(offset_ptr); Header.SegSize = data.getU8(offset_ptr); + // Perform basic validation of the header fields. + if (!data.isValidOffsetForDataOfSize(Offset, Header.Length) || + (Header.AddrSize != 4 && Header.AddrSize != 8)) { + clear(); + return false; + } + // The first tuple following the header in each set begins at an offset // that is a multiple of the size of a single tuple (that is, twice the // size of an address). The header is padded, if necessary, to the diff --git a/tools/llvm-dwarfdump/llvm-dwarfdump.cpp b/tools/llvm-dwarfdump/llvm-dwarfdump.cpp index 076dcd10fb0..ef9a47b9594 100644 --- a/tools/llvm-dwarfdump/llvm-dwarfdump.cpp +++ b/tools/llvm-dwarfdump/llvm-dwarfdump.cpp @@ -62,13 +62,17 @@ static void DumpInput(const StringRef &Filename) { i->getName(name); StringRef data; i->getContents(data); - if (name.endswith("debug_info")) + + if (name.startswith("__DWARF,")) + name = name.substr(8); // Skip "__DWARF," prefix. + name = name.substr(name.find_first_not_of("._")); // Skip . and _ prefixes. + if (name == "debug_info") DebugInfoSection = data; - else if (name.endswith("debug_abbrev")) + else if (name == "debug_abbrev") DebugAbbrevSection = data; - else if (name.endswith("debug_line")) + else if (name == "debug_line") DebugLineSection = data; - else if (name.endswith("debug_aranges")) + else if (name == "debug_aranges") DebugArangesSection = data; }