Skip to content

Commit

Permalink
Prevent FENTRY_CALL reordering
Browse files Browse the repository at this point in the history
FEntryInserter prepends FENTRY_CALL to the first basic block. In case
there are other instructions, PostRA Machine Instruction Scheduler can
move FENTRY_CALL call around. This actually occurs on SystemZ (see the
testcase). This is bad for the following reasons:

* FENTRY_CALL clobbers registers.
* Linux Kernel depends on whatever FENTRY_CALL expands to to be the very
  first instruction in the function.

Fix by adding isCall attribute to FENTRY_CALL, which prevents reordering
by making it a scheduling boundary for PostRA Machine Instruction
Scheduler.

Reviewed By: niravd

Differential Revision: https://reviews.llvm.org/D91218
  • Loading branch information
iii-i committed Dec 8, 2020
1 parent 5171b7b commit d58f112
Show file tree
Hide file tree
Showing 4 changed files with 43 additions and 0 deletions.
1 change: 1 addition & 0 deletions llvm/include/llvm/Target/Target.td
Expand Up @@ -1282,6 +1282,7 @@ def FENTRY_CALL : StandardPseudoInstruction {
let InOperandList = (ins);
let AsmString = "# FEntry call";
let usesCustomInserter = true;
let isCall = true;
let mayLoad = true;
let mayStore = true;
let hasSideEffects = true;
Expand Down
1 change: 1 addition & 0 deletions llvm/lib/CodeGen/MachineInstr.cpp
Expand Up @@ -713,6 +713,7 @@ bool MachineInstr::isCandidateForCallSiteEntry(QueryType Type) const {
case TargetOpcode::PATCHPOINT:
case TargetOpcode::STACKMAP:
case TargetOpcode::STATEPOINT:
case TargetOpcode::FENTRY_CALL:
return false;
}
return true;
Expand Down
22 changes: 22 additions & 0 deletions llvm/test/CodeGen/SystemZ/fentry-debug-info.ll
@@ -0,0 +1,22 @@
; Test that compiling with fentry and debuginfo works.
;
; RUN: llc %s -mtriple=s390x-linux-gnu -o - -verify-machineinstrs

define dso_local void @_Z1av() local_unnamed_addr #0 !dbg !7 {
entry:
ret void
}

attributes #0 = { norecurse nounwind readnone "fentry-call"="true" }

!llvm.dbg.cu = !{!0}
!llvm.module.flags = !{!3, !4}

!0 = distinct !DICompileUnit(language: DW_LANG_C_plus_plus_14, file: !1, producer: "clang version 12.0.0", isOptimized: true, runtimeVersion: 0, emissionKind: FullDebug, enums: !2, splitDebugInlining: false, nameTableKind: None)
!1 = !DIFile(filename: "a.cpp", directory: "/tmp")
!2 = !{}
!3 = !{i32 7, !"Dwarf Version", i32 4}
!4 = !{i32 2, !"Debug Info Version", i32 3}
!7 = distinct !DISubprogram(name: "a", linkageName: "_Z1av", scope: !1, file: !1, line: 1, type: !8, scopeLine: 1, flags: DIFlagPrototyped | DIFlagAllCallsDescribed, spFlags: DISPFlagDefinition | DISPFlagOptimized, unit: !0, retainedNodes: !2)
!8 = !DISubroutineType(types: !9)
!9 = !{null}
19 changes: 19 additions & 0 deletions llvm/test/CodeGen/SystemZ/fentry-no-reorder.ll
@@ -0,0 +1,19 @@
; RUN: llc %s -mtriple=s390x-linux-gnu -mcpu=zEC12 -o - -verify-machineinstrs \
; RUN: | FileCheck %s

@PawnTT = dso_local local_unnamed_addr global [2048 x i8] zeroinitializer, align 2

define dso_local void @clear_pawn_tt() local_unnamed_addr #0 {
entry:
call void @llvm.memset.p0i8.i64(i8* nonnull align 2 dereferenceable(2048) getelementptr inbounds ([2048 x i8], [2048 x i8]* @PawnTT, i64 0, i64 0), i8 0, i64 2048, i1 false)
ret void
}

declare void @llvm.memset.p0i8.i64(i8* nocapture writeonly, i8, i64, i1 immarg) #1

attributes #0 = { nofree nounwind writeonly "fentry-call"="true" }
attributes #1 = { argmemonly nofree nosync nounwind willreturn writeonly }

; CHECK: clear_pawn_tt: # @clear_pawn_tt
; CHECK-NEXT: # %bb.0:
; CHECK-NEXT: brasl %r0, __fentry__@PLT

0 comments on commit d58f112

Please sign in to comment.