; RUN: llc -mtriple=s390x-ibm-linux %s -o - | FileCheck %s
; RUN: llc -mtriple=s390x-ibm-linux -function-sections %s -o - | FileCheck %s
; RUN: llc -mtriple=s390x-ibm-linux -no-integrated-as -binutils-version=2.35 %s -o - | FileCheck --check-prefix=NOLINK %s
; RUN: llc -mtriple=s390x-ibm-linux -no-integrated-as -binutils-version=2.36 %s -o - | FileCheck %s
; RUN: llc -mtriple=s390x-ibm-linux -mcpu=zEC12 %s -o - | FileCheck %s --check-prefix=ZEC12

;; GNU ld < 2.36 did not support mixed SHF_LINK_ORDER and non-SHF_LINK_ORDER sections.
; NOLINK-NOT: "awo"

define i32 @f0() "patchable-function-entry"="0" {
; CHECK-LABEL: f0:
; CHECK-NEXT: .Lfunc_begin0:
; CHECK-NOT:   nopr
; CHECK:       lhi     %r2, 0
; CHECK-NOT:   .section __patchable_function_entries
  ret i32 0
}

define i32 @f1() "patchable-function-entry"="1" {
; CHECK-LABEL: f1:
; CHECK-NEXT: .Lfunc_begin1:
; CHECK:       nopr
; CHECK-NEXT:  lhi     %r2, 0
; CHECK:       .section __patchable_function_entries,"awo",@progbits,f1{{$}}
; CHECK-NEXT:  .p2align 3, 0x0
; CHECK-NEXT: .quad   .Lfunc_begin1
  ret i32 0
}

;; Without -function-sections, f2 is in the same text section as f1.
;; They share the __patchable_function_entries section.
;; With -function-sections, f1 and f2 are in different text sections.
;; Use separate __patchable_function_entries.
define void @f2() "patchable-function-entry"="2" {
; CHECK-LABEL: f2:
; CHECK-NEXT: .Lfunc_begin2:
; CHECK-COUNT-2: nopr
; CHECK-NEXT:  br      %r14
; CHECK:       .section __patchable_function_entries,"awo",@progbits,f2{{$}}
; CHECK-NEXT:  .p2align 3, 0x0
; CHECK-NEXT: .quad   .Lfunc_begin2
  ret void
}

$f3 = comdat any
define void @f3() "patchable-function-entry"="3" comdat {
; CHECK-LABEL: f3:
; CHECK-NEXT: .Lfunc_begin3:
; CHECK-COUNT-3: nopr
; CHECK-NEXT:  br      %r14
; CHECK:       .section __patchable_function_entries,"awoG",@progbits,f3,f3,comdat{{$}}
; CHECK-NEXT:  .p2align 3, 0x0
; CHECK-NEXT: .quad   .Lfunc_begin3
  ret void
}

$f5 = comdat any
define void @f5() "patchable-function-entry"="5" comdat {
; CHECK-LABEL: f5:
; CHECK-NEXT: .Lfunc_begin4:
; CHECK-COUNT-5: nopr
; CHECK-NEXT:  aghi    %r15, -176
; CHECK:       .section __patchable_function_entries,"awoG",@progbits,f5,f5,comdat{{$}}
; CHECK:       .p2align 3, 0x0
; CHECK-NEXT: .quad   .Lfunc_begin4
  %frame = alloca i8, i32 16
  ret void
}

;; -fpatchable-function-entry=3,2
;; "patchable-function-prefix" emits data before the function entry label.
define void @f3_2() "patchable-function-entry"="1" "patchable-function-prefix"="2" {
; CHECK-LABEL: .type f3_2,@function
; CHECK-NEXT: .Ltmp0:
; CHECK-NEXT:  nopr
; CHECK-NEXT:  nopr
; CHECK-NEXT: f3_2:  # @f3_2
; CHECK:      # %bb.0:
; CHECK-NEXT:  nopr
; CHECK-NEXT:  br      %r14
;; .size does not include the prefix.
; CHECK:      .Lfunc_end5:
; CHECK-NEXT: .size f3_2, .Lfunc_end5-f3_2
; CHECK:      .section __patchable_function_entries,"awo",@progbits,f3_2{{$}}
; CHECK:      .p2align 3, 0x0
; CHECK-NEXT: .quad   .Ltmp0
  ret void
}

;; When prefix data is used, arbitrarily place noprs after prefix data.
define void @prefix() "patchable-function-entry"="0" "patchable-function-prefix"="1" prefix i32 1 {
; CHECK-LABEL: .type prefix,@function
; CHECK-NEXT: .long 1 # 0x1
; CHECK:      .Ltmp1:
; CHECK:       nopr
; CHECK-NEXT: prefix:  # @prefix
;; Emit a __patchable_function_entries entry even if "patchable-function-entry" is 0.
; CHECK:      .section __patchable_function_entries,"awo",@progbits,prefix{{$}}
; CHECK:      .p2align        3, 0x0
; CHECK-NEXT: .quad   .Ltmp1
  ret void
}

;; Make sure patchable function entry is not moved around.
declare void @f4_1(i64)

define void @f4_2() "patchable-function-entry"="3" {
; ZEC12-LABEL: f4_2:
; ZEC12-NEXT:  [[ENTRY:.Lfunc_begin[0-9]+]]:
; ZEC12:       # %bb.0:
; ZEC12-NEXT:  nopr
; ZEC12-NEXT:  nopr
; ZEC12-NEXT:  nopr
; ZEC12-NEXT:  lghi
; ZEC12:       .section __patchable_function_entries,"awo",@progbits,f4_2
; ZEC12-NEXT:  .p2align 3, 0x0
; ZEC12-NEXT:  .quad [[ENTRY]]
  tail call void (i64) @f4_1(i64 1)
  ret void
}