bolt/test/runtime/X86/instrumentation-ind-calls.s

   1 # This reproduces a bug with instrumentation when trying to count calls
   2 # when the target address is computed with a reference to the stack pointer.
   3 # Our instrumentation code uses the stack to save registers to be
   4 # transparent with the instrumented code, but we end up updating the stack
   5 # pointer while doing so, which affects this target address calculation.
   6 # The solution is to temporarily fix RSP. Check that we correctly instrument
   7 # these cases.
   8
   9 # REQUIRES: system-linux,bolt-runtime
  10
  11 # RUN: llvm-mc -filetype=obj -triple x86_64-unknown-unknown \
  12 # RUN:   %s -o %t.o
  13 # RUN: %clang %cflags -no-pie %t.o -o %t.exe -Wl,-q
  14
  15 # RUN: llvm-bolt %t.exe --instrument --instrumentation-file=%t.fdata \
  16 # RUN:   -o %t.instrumented
  17
  18 # Instrumented program needs to finish returning zero
  19 # RUN: %t.instrumented arg1 arg2
  20
  21 # Test that the instrumented data makes sense
  22 # RUN:  llvm-bolt %t.exe -o %t.bolted --data %t.fdata \
  23 # RUN:    --reorder-blocks=ext-tsp --reorder-functions=hfsort+ \
  24 # RUN:    --print-only=main --print-finalized | FileCheck %s
  25
  26 # RUN: %t.bolted arg1 arg2
  27
  28 # Check that our indirect call has 1 hit recorded in the fdata file and that
  29 # this was processed correctly by BOLT
  30 # CHECK:         callq   *0x18(%rsp) # CallProfile: 1 (0 misses) :
  31 # CHECK-NEXT:        { targetFunc: 1 (0 misses) }
  32
  33   .text
  34   .globl  main
  35   .type main, %function
  36   .p2align  4
  37 main:
  38   pushq %rbp
  39   movq  %rsp, %rbp
  40   leaq targetFunc, %rax
  41   pushq %rax                  # We save the target function address in the stack
  42   subq  $0x18, %rsp           # Set up a dummy stack frame
  43   cmpl  $0x2, %edi
  44   jb    .LBBerror             # Add control flow so we don't have a trivial case
  45 .LBB2:
  46   callq *0x18(%rsp)           # Indirect call using %rsp
  47   addq $0x20, %rsp
  48   movq %rbp, %rsp
  49   pop %rbp
  50   retq
  51
  52 .LBBerror:
  53   addq $0x20, %rsp
  54   movq %rbp, %rsp
  55   pop %rbp
  56   movq $1, %rax               # Finish with an error if we go this path
  57   retq
  58   .size main, .-main
  59
  60   .globl targetFunc
  61   .type targetFunc, %function
  62   .p2align  4
  63 targetFunc:
  64   xorq %rax, %rax
  65   retq
  66   .size targetFunc, .-targetFunc