llvm/test/Transforms/EntryExitInstrumenter/mcount.ll

; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
; RUN: opt -passes="function(ee-instrument),cgscc(inline),function(ee-instrument<post-inline>)" -S < %s | FileCheck %s

; Running the passes twice should not result in more instrumentation.
; RUN: opt -passes="function(ee-instrument),function(ee-instrument),cgscc(inline),function(ee-instrument<post-inline>),function(ee-instrument<post-inline>)" -S < %s | FileCheck %s

target datalayout = "E-m:e-i64:64-n32:64"
target triple = "powerpc64le-unknown-linux"

define void @leaf_function() #0 {
; CHECK-LABEL: define void @leaf_function() {
; CHECK-NEXT:    call void @mcount()
; CHECK-NEXT:    [[TMP1:%.*]] = call ptr @llvm.returnaddress.p0(i32 0)
; CHECK-NEXT:    call void @__cyg_profile_func_enter(ptr @leaf_function, ptr [[TMP1]])
; CHECK-NEXT:    [[TMP2:%.*]] = call ptr @llvm.returnaddress.p0(i32 0)
; CHECK-NEXT:    call void @__cyg_profile_func_exit(ptr @leaf_function, ptr [[TMP2]])
; CHECK-NEXT:    ret void
;
  ret void
}


define void @root_function() #0 {
; CHECK-LABEL: define void @root_function() {
; CHECK-NEXT:    call void @mcount()
; CHECK-NEXT:    [[TMP1:%.*]] = call ptr @llvm.returnaddress.p0(i32 0)
; CHECK-NEXT:    call void @__cyg_profile_func_enter(ptr @root_function, ptr [[TMP1]])
; CHECK-NEXT:    [[TMP2:%.*]] = call ptr @llvm.returnaddress.p0(i32 0)
; CHECK-NEXT:    call void @__cyg_profile_func_enter(ptr @leaf_function, ptr [[TMP2]])
; CHECK-NEXT:    [[TMP3:%.*]] = call ptr @llvm.returnaddress.p0(i32 0)
; CHECK-NEXT:    call void @__cyg_profile_func_exit(ptr @leaf_function, ptr [[TMP3]])
; CHECK-NEXT:    [[TMP4:%.*]] = call ptr @llvm.returnaddress.p0(i32 0)
; CHECK-NEXT:    call void @__cyg_profile_func_exit(ptr @root_function, ptr [[TMP4]])
; CHECK-NEXT:    ret void
;
  call void @leaf_function()
  ret void
}

; The mcount function has many different names.

define void @f1() #1 {
; CHECK-LABEL: define void @f1() {
; CHECK-NEXT:    call void @.mcount()
; CHECK-NEXT:    ret void
;
  ret void
}

define void @f2() #2 {
; CHECK-LABEL: define void @f2() {
; CHECK-NEXT:    call void @llvm.arm.gnu.eabi.mcount()
; CHECK-NEXT:    ret void
;
  ret void
}

define void @f3() #3 {
; CHECK-LABEL: define void @f3() {
; CHECK-NEXT:    call void @"\01_mcount"()
; CHECK-NEXT:    ret void
;
  ret void
}

define void @f4() #4 {
; CHECK-LABEL: define void @f4() {
; CHECK-NEXT:    call void @"\01mcount"()
; CHECK-NEXT:    ret void
;
  ret void
}

define void @f5() #5 {
; CHECK-LABEL: define void @f5() {
; CHECK-NEXT:    call void @__mcount()
; CHECK-NEXT:    ret void
;
  ret void
}

define void @f6() #6 {
; CHECK-LABEL: define void @f6() {
; CHECK-NEXT:    call void @_mcount()
; CHECK-NEXT:    ret void
;
  ret void
}

define void @f7() #7 {
; CHECK-LABEL: define void @f7() {
; CHECK-NEXT:    call void @__cyg_profile_func_enter_bare()
; CHECK-NEXT:    ret void
;
  ret void
}

; Treat musttail calls as terminators; inserting between the musttail call and
; ret is not allowed.
declare ptr @tailcallee()
define ptr @tailcaller() #8 {
; CHECK-LABEL: define ptr @tailcaller() {
; CHECK-NEXT:    [[TMP1:%.*]] = call ptr @llvm.returnaddress.p0(i32 0)
; CHECK-NEXT:    call void @__cyg_profile_func_exit(ptr @tailcaller, ptr [[TMP1]])
; CHECK-NEXT:    [[TMP2:%.*]] = musttail call ptr @tailcallee()
; CHECK-NEXT:    ret ptr [[TMP2]]
;
  %1 = musttail call ptr @tailcallee()
  ret ptr %1
}
define ptr @tailcaller2() #8 {
; CHECK-LABEL: define ptr @tailcaller2() {
; CHECK-NEXT:    [[TMP1:%.*]] = call ptr @llvm.returnaddress.p0(i32 0)
; CHECK-NEXT:    call void @__cyg_profile_func_exit(ptr @tailcaller2, ptr [[TMP1]])
; CHECK-NEXT:    [[TMP2:%.*]] = musttail call ptr @tailcallee()
; CHECK-NEXT:    ret ptr [[TMP2]]
;
  %1 = musttail call ptr @tailcallee()
  ret ptr %1
}

;; naked functions are not instrumented, otherwise the argument registers
;; and the return address register (if present) would be clobbered.
define void @naked() naked {
; CHECK-LABEL: define void @naked(
; CHECK-SAME: ) #[[ATTR0:[0-9]+]] {
; CHECK-NEXT:    ret void
;
  ret void
}

define available_externally void @always_inline() {
; CHECK-LABEL: define available_externally void @always_inline() {
; CHECK-NEXT:    ret void
;
  ret void
}

; The attributes are "consumed" when the instrumentation is inserted.
; CHECK: attributes
; CHECK-NOT: instrument-function

attributes #0 = { "instrument-function-entry-inlined"="mcount" "instrument-function-entry"="__cyg_profile_func_enter" "instrument-function-exit"="__cyg_profile_func_exit" }
attributes #1 = { "instrument-function-entry-inlined"=".mcount" }
attributes #2 = { "instrument-function-entry-inlined"="llvm.arm.gnu.eabi.mcount" }
attributes #3 = { "instrument-function-entry-inlined"="\01_mcount" }
attributes #4 = { "instrument-function-entry-inlined"="\01mcount" }
attributes #5 = { "instrument-function-entry-inlined"="__mcount" }
attributes #6 = { "instrument-function-entry-inlined"="_mcount" }
attributes #7 = { "instrument-function-entry-inlined"="__cyg_profile_func_enter_bare" }
attributes #8 = { "instrument-function-exit"="__cyg_profile_func_exit" }