[lld] ec315a5 - [lld-macho] Fix LOH parsing segfault
Jez Ng via llvm-commits
llvm-commits at lists.llvm.org
Thu Jul 21 11:00:32 PDT 2022
Author: Jez Ng
Date: 2022-07-21T13:59:39-04:00
New Revision: ec315a5fa1c5880df0f6a25cede9282bb551c8a1
URL: https://github.com/llvm/llvm-project/commit/ec315a5fa1c5880df0f6a25cede9282bb551c8a1
DIFF: https://github.com/llvm/llvm-project/commit/ec315a5fa1c5880df0f6a25cede9282bb551c8a1.diff
LOG: [lld-macho] Fix LOH parsing segfault
`advanceSubsection()` didn't account for the possibility that a section
could have no subsections.
Reviewed By: #lld-macho, thakis, BertalanD
Differential Revision: https://reviews.llvm.org/D130288
Added:
lld/test/MachO/loh-parsing.s
Modified:
lld/MachO/InputFiles.cpp
Removed:
################################################################################
diff --git a/lld/MachO/InputFiles.cpp b/lld/MachO/InputFiles.cpp
index a2c27f1fd9f3..d39fa5feff1f 100644
--- a/lld/MachO/InputFiles.cpp
+++ b/lld/MachO/InputFiles.cpp
@@ -560,7 +560,7 @@ void ObjFile::parseOptimizationHints(ArrayRef<uint8_t> data) {
if (section == sections.end())
return;
++subsection;
- if (subsection == (*section)->subsections.end()) {
+ while (subsection == (*section)->subsections.end()) {
++section;
if (section == sections.end())
return;
@@ -582,6 +582,7 @@ void ObjFile::parseOptimizationHints(ArrayRef<uint8_t> data) {
if (section == sections.end())
break;
updateAddr();
+ assert(hintStart->offset0 >= subsectionBase);
}
}
diff --git a/lld/test/MachO/loh-parsing.s b/lld/test/MachO/loh-parsing.s
new file mode 100644
index 000000000000..aad1af359fe1
--- /dev/null
+++ b/lld/test/MachO/loh-parsing.s
@@ -0,0 +1,24 @@
+# REQUIRES: aarch64
+
+# RUN: llvm-mc -filetype=obj -triple=arm64-apple-darwin %s -o %t.o
+# RUN: %lld -arch arm64 -dylib %t.o -o /dev/null
+
+## Check that we parse the LOH & match it to its referent sections correctly,
+## even when there are other subsections that don't get parsed as regular
+## sections. (We would previously segfault.)
+## __debug_info is one such section that gets special-case handling.
+
+.text
+_foo:
+
+.section __DWARF,__debug_info,regular,debug
+
+## __StaticInit occurs after __debug_info in the input object file, so the
+## LOH-matching code will have to "walk" past __debug_info while searching for
+## __StaticInit. Thus this verifies that we can skip past __debug_info
+## correctly.
+.section __TEXT,__StaticInit
+L1: adrp x1, _foo at PAGE
+L2: ldr x1, [x1, _foo at PAGEOFF]
+
+.loh AdrpLdr L1, L2
More information about the llvm-commits
mailing list