[llvm] 6a6b0e4 - [X86] Check the address in machine verifier
Shengchen Kan via llvm-commits
llvm-commits at lists.llvm.org
Wed Apr 27 19:05:45 PDT 2022
Author: Shengchen Kan
Date: 2022-04-28T10:05:39+08:00
New Revision: 6a6b0e4a63106c44162c9be8f2ec6fb9e7b5a0d8
URL: https://github.com/llvm/llvm-project/commit/6a6b0e4a63106c44162c9be8f2ec6fb9e7b5a0d8
DIFF: https://github.com/llvm/llvm-project/commit/6a6b0e4a63106c44162c9be8f2ec6fb9e7b5a0d8.diff
LOG: [X86] Check the address in machine verifier
1. The scale factor must be 1, 2, 4, 8
2. The displacement must fit in 32-bit signed integer
Noticed by: https://github.com/llvm/llvm-project/issues/55091
Reviewed By: pengfei
Differential Revision: https://reviews.llvm.org/D124455
Added:
llvm/test/CodeGen/MIR/X86/machine-verifier-address.mir
Modified:
llvm/lib/Target/X86/X86InstrInfo.cpp
llvm/lib/Target/X86/X86InstrInfo.h
Removed:
################################################################################
diff --git a/llvm/lib/Target/X86/X86InstrInfo.cpp b/llvm/lib/Target/X86/X86InstrInfo.cpp
index 2a5f01a83ff8..0c45094b6e2f 100644
--- a/llvm/lib/Target/X86/X86InstrInfo.cpp
+++ b/llvm/lib/Target/X86/X86InstrInfo.cpp
@@ -3619,6 +3619,35 @@ X86InstrInfo::getAddrModeFromMemoryOp(const MachineInstr &MemI,
return AM;
}
+bool X86InstrInfo::verifyInstruction(const MachineInstr &MI,
+ StringRef &ErrInfo) const {
+ Optional<ExtAddrMode> AMOrNone = getAddrModeFromMemoryOp(MI, nullptr);
+ if (!AMOrNone)
+ return true;
+
+ ExtAddrMode AM = *AMOrNone;
+
+ if (AM.ScaledReg != X86::NoRegister) {
+ switch (AM.Scale) {
+ case 1:
+ case 2:
+ case 4:
+ case 8:
+ break;
+ default:
+ ErrInfo = "Scale factor in address must be 1, 2, 4 or 8";
+ return false;
+ }
+ }
+ if (!isInt<32>(AM.Displacement)) {
+ ErrInfo = "Displacement in address must fit into 32-bit signed "
+ "integer";
+ return false;
+ }
+
+ return true;
+}
+
bool X86InstrInfo::getConstValDefinedInReg(const MachineInstr &MI,
const Register Reg,
int64_t &ImmVal) const {
diff --git a/llvm/lib/Target/X86/X86InstrInfo.h b/llvm/lib/Target/X86/X86InstrInfo.h
index 92e14832aede..4943d2152fd2 100644
--- a/llvm/lib/Target/X86/X86InstrInfo.h
+++ b/llvm/lib/Target/X86/X86InstrInfo.h
@@ -562,6 +562,8 @@ class X86InstrInfo final : public X86GenInstrInfo {
MachineBasicBlock::iterator &It, MachineFunction &MF,
outliner::Candidate &C) const override;
+ bool verifyInstruction(const MachineInstr &MI,
+ StringRef &ErrInfo) const override;
#define GET_INSTRINFO_HELPER_DECLS
#include "X86GenInstrInfo.inc"
diff --git a/llvm/test/CodeGen/MIR/X86/machine-verifier-address.mir b/llvm/test/CodeGen/MIR/X86/machine-verifier-address.mir
new file mode 100644
index 000000000000..228510a0ca81
--- /dev/null
+++ b/llvm/test/CodeGen/MIR/X86/machine-verifier-address.mir
@@ -0,0 +1,31 @@
+# RUN: not --crash llc -march=x86-64 -run-pass none -o /dev/null %s 2>&1 | FileCheck %s
+# This test ensures that the address is checked in machine verifier.
+
+---
+name: baz
+tracksRegLiveness: true
+body: |
+ bb.0:
+ successors: %bb.1(0x80000000)
+ liveins: $rdi, $xmm0
+
+ %1:vr128 = COPY $xmm0
+ %0:gr64 = COPY $rdi
+ %2:vr128 = COPY %1
+
+ bb.1:
+ successors: %bb.1(0x80000000)
+
+ %3:vr256 = AVX_SET0
+ %4:vr128 = VPSLLDri %2, 31
+ %5:vr256 = VPMOVSXDQYrr killed %4
+ %8:vr256 = IMPLICIT_DEF
+ ; CHECK: *** Bad machine code: Scale factor in address must be 1, 2, 4 or 8 ***
+ %6:vr256, %7:vr256 = VGATHERQPDYrm %3, %0, 16, killed %8, 0, $noreg, %5 :: (load unknown-size, align 8)
+ %9:vr128 = COPY %6.sub_xmm
+ ; CHECK: *** Bad machine code: Displacement in address must fit into 32-bit signed integer ***
+ VMOVLPDmr $noreg, 1, $noreg, 1111111111111, $noreg, killed %9 :: (store (s64) into `i64* undef`)
+ JMP_1 %bb.1
+ ; CHECK: LLVM ERROR: Found 2 machine code errors
+
+...
More information about the llvm-commits
mailing list