From 167306a56c6d1e0ed4475b41235e47d4812429e8 Mon Sep 17 00:00:00 2001
From: Simon Atanasyan <simon@atanasyan.com>
Date: Wed, 22 Nov 2017 12:34:29 +0000
Subject: [PATCH] [MIPS] Write PLT0 entry in case of linking N64 ABI code

llvm-svn: 318831
---
 lld/ELF/Arch/Mips.cpp          | 13 +++++++++++--
 lld/test/ELF/mips-26-n32-n64.s | 35 +++++++++++++++++++++++++++++++++++
 2 files changed, 46 insertions(+), 2 deletions(-)
 create mode 100644 lld/test/ELF/mips-26-n32-n64.s

diff --git a/lld/ELF/Arch/Mips.cpp b/lld/ELF/Arch/Mips.cpp
index 0c1ca5c..495e256 100644
--- a/lld/ELF/Arch/Mips.cpp
+++ b/lld/ELF/Arch/Mips.cpp
@@ -278,15 +278,24 @@ template <class ELFT> void MIPS<ELFT>::writePltHeader(uint8_t *Buf) const {
     write32<E>(Buf + 4, 0x8dd90000);  // lw    $25, %lo(&GOTPLT[0])($14)
     write32<E>(Buf + 8, 0x25ce0000);  // addiu $14, $14, %lo(&GOTPLT[0])
     write32<E>(Buf + 12, 0x030ec023); // subu  $24, $24, $14
+    write32<E>(Buf + 16, 0x03e07825); // move  $15, $31
+    write32<E>(Buf + 20, 0x0018c082); // srl   $24, $24, 2
+  } else if (ELFT::Is64Bits) {
+    write32<E>(Buf, 0x3c0e0000);      // lui   $14, %hi(&GOTPLT[0])
+    write32<E>(Buf + 4, 0xddd90000);  // ld    $25, %lo(&GOTPLT[0])($14)
+    write32<E>(Buf + 8, 0x25ce0000);  // addiu $14, $14, %lo(&GOTPLT[0])
+    write32<E>(Buf + 12, 0x030ec023); // subu  $24, $24, $14
+    write32<E>(Buf + 16, 0x03e07825); // move  $15, $31
+    write32<E>(Buf + 20, 0x0018c0c2); // srl   $24, $24, 3
   } else {
     write32<E>(Buf, 0x3c1c0000);      // lui   $28, %hi(&GOTPLT[0])
     write32<E>(Buf + 4, 0x8f990000);  // lw    $25, %lo(&GOTPLT[0])($28)
     write32<E>(Buf + 8, 0x279c0000);  // addiu $28, $28, %lo(&GOTPLT[0])
     write32<E>(Buf + 12, 0x031cc023); // subu  $24, $24, $28
+    write32<E>(Buf + 16, 0x03e07825); // move  $15, $31
+    write32<E>(Buf + 20, 0x0018c082); // srl   $24, $24, 2
   }
 
-  write32<E>(Buf + 16, 0x03e07825); // move  $15, $31
-  write32<E>(Buf + 20, 0x0018c082); // srl   $24, $24, 2
   write32<E>(Buf + 24, 0x0320f809); // jalr  $25
   write32<E>(Buf + 28, 0x2718fffe); // subu  $24, $24, 2
 
diff --git a/lld/test/ELF/mips-26-n32-n64.s b/lld/test/ELF/mips-26-n32-n64.s
new file mode 100644
index 0000000..2e24873
--- /dev/null
+++ b/lld/test/ELF/mips-26-n32-n64.s
@@ -0,0 +1,35 @@
+# Check R_MIPS_26 relocation handling in case of N64 ABIs.
+
+# RUN: llvm-mc -filetype=obj -triple=mips64-unknown-linux \
+# RUN:         %S/Inputs/mips-dynamic.s -o %t-so.o
+# RUN: ld.lld %t-so.o -shared -o %t.so
+# RUN: llvm-mc -filetype=obj -triple=mips64-unknown-linux %s -o %t.o
+# RUN: ld.lld %t.o %t.so -o %t.exe
+# RUN: llvm-objdump -d %t.exe | FileCheck %s
+
+# REQUIRES: mips
+
+# CHECK:      Disassembly of section .text:
+# CHECK-NEXT: __start:
+# CHECK-NEXT:    20000:       0c 00 80 0c     jal     131120
+# CHECK-NEXT:    20004:       00 00 00 00     nop
+# CHECK-NEXT: Disassembly of section .plt:
+# CHECK-NEXT: .plt:
+# CHECK-NEXT:    20010:       3c 0e 00 03     lui     $14, 3
+# CHECK-NEXT:    20014:       dd d9 00 08     ld      $25, 8($14)
+# CHECK-NEXT:    20018:       25 ce 00 08     addiu   $14, $14, 8
+# CHECK-NEXT:    2001c:       03 0e c0 23     subu    $24, $24, $14
+# CHECK-NEXT:    20020:       03 e0 78 25     move    $15, $ra
+# CHECK-NEXT:    20024:       00 18 c0 c2     srl     $24, $24, 3
+# CHECK-NEXT:    20028:       03 20 f8 09     jalr    $25
+# CHECK-NEXT:    2002c:       27 18 ff fe     addiu   $24, $24, -2
+# CHECK-NEXT:    20030:       3c 0f 00 03     lui     $15, 3
+# CHECK-NEXT:    20034:       8d f9 00 18     lw      $25, 24($15)
+# CHECK-NEXT:    20038:       03 20 00 08     jr      $25
+# CHECK-NEXT:    2003c:       25 f8 00 18     addiu   $24, $15, 24
+
+  .text
+  .option pic0
+  .global __start
+__start:
+  jal   foo0
-- 
2.7.4