blob: 1ccb1fb26a9395d4d8007c5328d42b4b20cf573c [file] [log] [blame]
From 7e0768329ca347e37c7bdb0da16b51cb3e7b7d8b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Martin=20Storsj=C3=B6?= <martin@martin.st>
Date: Sun, 9 May 2021 23:27:35 +0300
Subject: [PATCH] [LLD] [COFF] Fix including the personality function for DWARF
EH when linking with --gc-sections
Since c579a5b1d92a9bc2046d00ee2d427832e0f5ddec we don't traverse
.eh_frame when doing GC. But the exception handling personality
function needs to be included, and is only referenced from within
.eh_frame.
Differential Revision: https://reviews.llvm.org/D102138
---
COFF/Driver.cpp | 19 ++++++++++++++++++-
test/COFF/gc-dwarf-eh.s | 36 ++++++++++++++++++++++++++++++++++++
2 files changed, 54 insertions(+), 1 deletion(-)
create mode 100644 test/COFF/gc-dwarf-eh.s
diff --git a/COFF/Driver.cpp b/COFF/Driver.cpp
index e2374b52b4f9f..029148c4722bb 100644
--- a/COFF/Driver.cpp
+++ b/COFF/Driver.cpp
@@ -2226,8 +2226,25 @@ void LinkerDriver::linkerMain(ArrayRef<const char *> argsArr) {
config->printSymbolOrder = arg->getValue();
// Identify unreferenced COMDAT sections.
- if (config->doGC)
+ if (config->doGC) {
+ if (config->mingw) {
+ // markLive doesn't traverse .eh_frame, but the personality function is
+ // only reached that way. The proper solution would be to parse and
+ // traverse the .eh_frame section, like the ELF linker does.
+ // For now, just manually try to retain the known possible personality
+ // functions. This doesn't bring in more object files, but only marks
+ // functions that already have been included to be retained.
+ for (const char *n : {"__gxx_personality_v0", "__gcc_personality_v0"}) {
+ Defined *d = dyn_cast_or_null<Defined>(symtab->findUnderscore(n));
+ if (d && !d->isGCRoot) {
+ d->isGCRoot = true;
+ config->gcroot.push_back(d);
+ }
+ }
+ }
+
markLive(symtab->getChunks());
+ }
// Needs to happen after the last call to addFile().
convertResources();
diff --git a/test/COFF/gc-dwarf-eh.s b/test/COFF/gc-dwarf-eh.s
new file mode 100644
index 0000000000000..757aa671c7693
--- /dev/null
+++ b/test/COFF/gc-dwarf-eh.s
@@ -0,0 +1,36 @@
+# REQUIRES: x86
+
+# RUN: llvm-mc -triple=i686-windows-gnu %s -filetype=obj -o %t.obj
+# RUN: lld-link -lldmingw -lldmap:%t.map -out:%t.exe -opt:ref -entry:main %t.obj -verbose 2>&1 | FileCheck %s
+# RUN: FileCheck %s --check-prefix=MAP --input-file=%t.map
+
+# CHECK: Discarded _unused
+
+# MAP: In Symbol
+# MAP: gc-dwarf-eh.s.tmp.obj:(.text)
+# MAP: {{ ___gxx_personality_v0$}}
+
+ .def _main; .scl 2; .type 32; .endef
+ .section .text,"xr",one_only,_main
+ .globl _main
+ .cfi_startproc
+ .cfi_personality 0, ___gxx_personality_v0
+_main:
+ xorl %eax, %eax
+ ret
+ .cfi_endproc
+
+ .def ___gxx_personality_v0; .scl 2; .type 32; .endef
+ .section .text,"xr",one_only,___gxx_personality_v0
+ .globl ___gxx_personality_v0
+___gxx_personality_v0:
+ ret
+
+ .def _unused; .scl 2; .type 32; .endef
+ .section .text,"xr",one_only,_unused
+ .globl _unused
+ .cfi_startproc
+ .cfi_personality 0, ___gxx_personality_v0
+_unused:
+ ret
+ .cfi_endproc