[BOLT] Update skipRelocation for aarch64

[yota9]

Summary

The ld might relax ADRP+ADD or ADRP+LDR sequences to the ADR+NOP, add
the new case to the skipRelocation for aarch64.

Vladislav Khmelevsky,
Advanced Software Technology Lab, Huawei

[yota9]

@rafaelauler Gentle ping

[rafauler]

bolt/lib/Core/Relocation.cpp

227

By ld, do you mean GNU ld? How is the test testing for that? (since it uses lld)

[yota9]

bolt/lib/Core/Relocation.cpp

227

No, I meant generic linker, not bfd. The lld has this optimization too (https://reviews.llvm.org/D117614)

[rafauler]

bolt/test/runtime/plt-lld.test

1–2

Is it possible to add a check by dumping the binary symbol table to verify that these are properly updated? having the test as is (checking if a binary runs) is quite opaque for reviewers and makes it a bit hard for us to be helpful and understand what is going on and what your patch is fixing. If we treat these relocations that were relaxed to ADR as real relocs (meaning, we do not have your patch), what does it happen? Does BOLT crash? Do PLT pointers get messed up?

[rafauler]

bolt/lib/Core/Relocation.cpp

227

Can you replace ld by "linker", then?

[yota9]

bolt/test/runtime/plt-lld.test

1–2

I will add extra test now.

what does it happen?

The pointers in code are messed up, so we will have runtime problems. E.g.
Let's suppose the GOT relocations were relaxed to the local symbol definition

1036c:       d503201f        nop
                     1036c: R_AARCH64_ADR_GOT_PAGE   foo2
10370:       10000080        adr     x0, 10380 <foo2>
                     10370: R_AARCH64_LD64_GOT_LO12_NC       foo2

After bolt without this patch we will have:

400024:       90ffe000        adrp    x0, 0 <_DYNAMIC-0x20388>
400028:       91000000        add     x0, x0, #0x0

Because the extracted value was messed up (we were expecting other instruction)

[yota9]

Add test

[yota9]

Nit

[rafauler]

Looks good, thanks!

bolt/lib/Core/Relocation.cpp

 
 bool skipRelocationProcessX86(uint64_t Type, uint64_t Contents) {
   return false;
 }
 
 bool skipRelocationProcessAArch64(uint64_t Type, uint64_t Contents) {
   auto IsMov = [](uint64_t Contents) -> bool {
     // The bits 28-23 are 0b100101
-    if ((Contents & 0x1f800000) == 0x12800000)
-      return true;
-    return false;
+    return (Contents & 0x1f800000) == 0x12800000;
   };
 
   auto IsB = [](uint64_t Contents) -> bool {
     // The bits 31-26 are 0b000101
-    if ((Contents & 0xfc000000) == 0x14000000)
-      return true;
-    return false;
+    return (Contents & 0xfc000000) == 0x14000000;
+  };
+
+  auto IsAdr = [](uint64_t Contents) -> bool {
+    // The bits 31-24 are 0b0xx10000
+    return (Contents & 0x9f000000) == 0x10000000;
   };
 
   auto IsNop = [](uint64_t Contents) -> bool { return Contents == 0xd503201f; };
 
   // The linker might eliminate the instruction and replace it with NOP, ignore
   if (IsNop(Contents))
     return true;
 
bool skipRelocationProcessAArch64(uint64_t Type, uint64_t Contents) {
   case ELF::R_AARCH64_LD64_GOT_LO12_NC:
   case ELF::R_AARCH64_TLSIE_LD64_GOTTPREL_LO12_NC:
   case ELF::R_AARCH64_TLSDESC_LD64_LO12: {
     if (IsB(Contents))
       return true;
   }
   }
 
+  // The ld might relax ADRP+ADD or ADRP+LDR sequences to the ADR+NOP
+  switch (Type) {
+  default:
+    break;
+  case ELF::R_AARCH64_ADR_PREL_PG_HI21:
+  case ELF::R_AARCH64_ADD_ABS_LO12_NC:
+  case ELF::R_AARCH64_ADR_GOT_PAGE:
+  case ELF::R_AARCH64_LD64_GOT_LO12_NC:
+    if (IsAdr(Contents))
+      return true;
+  }
+
   return false;
 }
 
 uint64_t adjustValueX86(uint64_t Type, uint64_t Value, uint64_t PC) {
   switch (Type) {
   default:
     llvm_unreachable("not supported relocation");
   case ELF::R_X86_64_32:

bolt/test/runtime/plt-lld.test

 // This test checks that the pointers to PLT are properly updated.
 // The test is using lld linker.
 
-// RUN: %clang %cflags %p/../Inputs/plt.c -fuse-ld=lld \
-// RUN:    -o %t.lld.exe -Wl,-q
-// RUN: llvm-bolt %t.lld.exe -o %t.lld.bolt.exe -use-old-text=0 -lite=0
-// RUN: %t.lld.bolt.exe | FileCheck %s
+// Non-PIE:
+RUN: %clang %cflags -no-pie %p/../Inputs/plt.c -fuse-ld=lld \
+RUN:    -o %t.lld.exe -Wl,-q
+RUN: llvm-bolt %t.lld.exe -o %t.lld.bolt.exe -use-old-text=0 -lite=0
+RUN: %t.lld.bolt.exe | FileCheck %s
 
-// CHECK: Test completed
+// PIE:
+RUN: %clang %cflags -fPIC -pie %p/../Inputs/plt.c -fuse-ld=lld \
+RUN:    -o %t.lld.pie.exe -Wl,-q
+RUN: llvm-bolt %t.lld.pie.exe -o %t.lld.bolt.pie.exe -use-old-text=0 -lite=0
+RUN: %t.lld.bolt.pie.exe | FileCheck %s
+
+CHECK: Test completed