[PPC64] Allow R_PPC64_DTPREL* to preemptable local-dynamic symbols

Similar to D60945.

Differential Revision: https://reviews.llvm.org/D60994

llvm-svn: 358950

4 files changed, 38 insertions, 5 deletions

diff --git a/lld/ELF/Arch/PPC64.cpp b/lld/ELF/Arch/PPC64.cpp
index cf1d3dd478f..995c3f899d8 100644
--- a/lld/ELF/Arch/PPC64.cpp
+++ b/lld/ELF/Arch/PPC64.cpp
@@ -496,7 +496,7 @@ RelExpr PPC64::getRelExpr(RelType Type, const Symbol &S,
   case R_PPC64_DTPREL16_LO:
   case R_PPC64_DTPREL16_LO_DS:
   case R_PPC64_DTPREL64:
-    return R_ABS;
+    return R_DTPREL;
   case R_PPC64_TLSGD:
     return R_TLSDESC_CALL;
   case R_PPC64_TLSLD:
diff --git a/lld/ELF/InputSection.cpp b/lld/ELF/InputSection.cpp
index 9b5567b76b0..5f09d07a022 100644
--- a/lld/ELF/InputSection.cpp
+++ b/lld/ELF/InputSection.cpp
@@ -615,11 +615,10 @@ static uint64_t getRelocTargetVA(const InputFile *File, RelType Type, int64_t A,
                                  uint64_t P, const Symbol &Sym, RelExpr Expr) {
   switch (Expr) {
   case R_ABS:
+  case R_DTPREL:
   case R_RELAX_TLS_LD_TO_LE_ABS:
   case R_RELAX_GOT_PC_NOPIC:
     return Sym.getVA(A);
-  case R_DTPREL:
-    return Sym.getVA(A);
   case R_ADDEND:
     return A;
   case R_ARM_SBREL:
diff --git a/lld/ELF/Relocations.cpp b/lld/ELF/Relocations.cpp
index 2fe44fc6303..7408506ccf7 100644
--- a/lld/ELF/Relocations.cpp
+++ b/lld/ELF/Relocations.cpp
@@ -248,8 +248,7 @@ handleTlsRelocation(RelType Type, Symbol &Sym, InputSectionBase &C,
   }
 
   // Local-Dynamic relocs can be relaxed to Local-Exec.
-  // TODO Delete R_ABS after all R_*_DTPREL_* relocations migrate to R_DTPREL.
-  if ((Expr == R_ABS || Expr == R_DTPREL) && !Config->Shared) {
+  if (Expr == R_DTPREL && !Config->Shared) {
     C.Relocations.push_back(
         {Target->adjustRelaxExpr(Type, nullptr, R_RELAX_TLS_LD_TO_LE), Type,
          Offset, Addend, &Sym});
diff --git a/lld/test/ELF/ppc64-tls-ld-preemptable.s b/lld/test/ELF/ppc64-tls-ld-preemptable.s
new file mode 100644
index 00000000000..8070f2a1a03
--- /dev/null
+++ b/lld/test/ELF/ppc64-tls-ld-preemptable.s
@@ -0,0 +1,35 @@
+# REQUIRES: ppc
+# RUN: llvm-mc -filetype=obj -triple=ppc64le %s -o %t.o
+# RUN: ld.lld %t.o -shared -o %t.so
+# RUN: llvm-objdump -d --no-show-raw-insn %t.so | FileCheck %s
+# RUN: llvm-nm %t.so | FileCheck --check-prefix=NM %s
+# RUN: llvm-readelf -x .rodata %t.so | FileCheck --check-prefix=RODATA %s
+
+# 0x2a - 0x8000 = -32726
+# CHECK:      addis 4, 3, 0
+# CHECK-NEXT: addi 4, 4, -32726
+# CHECK-NEXT: lis 5, 0
+# CHECK-NEXT: ori 5, 5, 0
+
+# NM: 000000000000002a B i
+# RODATA: 2a000000 00000000
+
+# We used to error on R_PPC64_DTPREL* to preemptable symbols.
+# i is STB_GLOBAL and preemptable.
+.globl foo
+foo:
+  addis 4, 3, i@dtprel@ha # R_PPC64_DTPREL16_HA
+  addi 4, 4, i@dtprel@l   # R_PPC64_DTPREL16_LO
+
+  lis 5, i@dtprel@highesta # R_PPC64_DTPREL16_HIGHESTA
+  ori 5, 5, i@dtprel@highera # R_PPC64_DTPREL16_HIGHERA
+
+.section .rodata,"a",@progbits
+  .quad i@dtprel+32768
+
+.section .tbss,"awT",@nobits
+  .space 0x2a
+.globl i
+i:
+  .long 0
+  .size i, 4