Use a relocation against the symbol if it is a PLT and the symbol is in another

section. Common because of linkonce sections.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@115718 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/lib/MC/ELFObjectWriter.cpp b/lib/MC/ELFObjectWriter.cpp
index d79029106b0599c7cd296eb3d0e25c1279f458e5..9f9681ae3efc3a646fc2fcbcaf05689bc47291c6 100644 (file)
--- a/lib/MC/ELFObjectWriter.cpp
+++ b/lib/MC/ELFObjectWriter.cpp
@@ -529,7 +529,8 @@ void ELFObjectWriterImpl::WriteSymbolTable(MCDataFragment *F,
 }
 
 static bool ShouldRelocOnSymbol(const MCSymbolData &SD,
-                                const MCValue &Target) {
+                                const MCValue &Target,
+                                const MCFragment &F) {
   const MCSymbol &Symbol = SD.getSymbol();
   if (Symbol.isUndefined())
     return true;
@@ -543,6 +544,14 @@ static bool ShouldRelocOnSymbol(const MCSymbolData &SD,
   if (SD.isExternal())
     return true;
 
+  const llvm::MCSymbolRefExpr& Ref = *Target.getSymA();
+  const MCSectionELF &Sec2 =
+    static_cast<const MCSectionELF&>(F.getParent()->getSection());
+
+  if (Ref.getKind() == MCSymbolRefExpr::VK_PLT &&
+      &Sec2 != &Section)
+    return true;
+
   return false;
 }
 
@@ -587,7 +596,7 @@ void ELFObjectWriterImpl::RecordRelocation(const MCAssembler &Asm,
       return;
     }
 
-    bool RelocOnSymbol = ShouldRelocOnSymbol(SD, Target);
+    bool RelocOnSymbol = ShouldRelocOnSymbol(SD, Target, *Fragment);
     if (!RelocOnSymbol) {
       Index = F->getParent()->getOrdinal();
 

diff --git a/test/MC/ELF/merge.s b/test/MC/ELF/merge.s
index 00ffd5ee9a82a2c07ccdaf1f2e202f1010b4a248..227d4140e16d571774b1c5de4feba47e729732a0 100644 (file)
--- a/test/MC/ELF/merge.s
+++ b/test/MC/ELF/merge.s
@@ -10,10 +10,14 @@
         movsd   .Lfoo(%rip), %xmm1
         movl   $.Lfoo, %edi
         movl   $.Lfoo+2, %edi
+        jmp    foo@PLT
 
         .section        .sec1,"aM",@progbits,16
 .Lfoo:
 
+        .section       bar,"ax",@progbits
+foo:
+
 // Section 4 is "sec1"
 // CHECK: # Section 4
 // CHECK-NEXT:  (('sh_name', 18) # '.sec1'
@@ -22,8 +26,12 @@
 // CHECK:      # Symbol 1
 // CHECK-NEXT: (('st_name', 1) # '.Lfoo'
 
-// Symbol number 5 is section 4
-// CHECK:        # Symbol 5
+// Symbol number 2 is foo
+// CHECK:      # Symbol 2
+// CHECK-NEXT: (('st_name', 7) # 'foo'
+
+// Symbol number 6 is section 4
+// CHECK:        # Symbol 6
 // CHECK-NEXT:    (('st_name', 0) # ''
 // CHECK-NEXT:     ('st_bind', 0)
 // CHECK-NEXT:     ('st_type', 3)
@@ -39,10 +47,10 @@
 // CHECK-NEXT:    ('r_addend',
 // CHECK-NEXT:   ),
 
-// Relocation 1 refers to symbol 5
+// Relocation 1 refers to symbol 6
 // CHECK-NEXT:  # Relocation 1
 // CHECK-NEXT: (('r_offset',
-// CHECK-NEXT:  ('r_sym', 5)
+// CHECK-NEXT:  ('r_sym', 6)
 // CHECK-NEXT:  ('r_type', 10)
 // CHECK-NEXT:  ('r_addend',
 // CHECK-NEXT: ),
@@ -54,4 +62,12 @@
 // CHECK-NEXT:    ('r_type', 10
 // CHECK-NEXT:    ('r_addend',
 // CHECK-NEXT:   ),
+
+// Relocation 3 refers to symbol 2
+// CHECK-NEXT:   # Relocation 3
+// CHECK-NEXT:   (('r_offset',
+// CHECK-NEXT:    ('r_sym', 2)
+// CHECK-NEXT:    ('r_type', 4
+// CHECK-NEXT:    ('r_addend',
+// CHECK-NEXT:   ),
 // CHECK-NEXT:  ])