Skip to content

Commit

Permalink
[InstCombine] Do not fold 'and (sext (ashr X, Shift)), C' if Shift < 0
Browse files Browse the repository at this point in the history
The 'and (sext (ashr X, ShiftC)), C' --> 'lshr (sext X), ShiftC'
transformation would access out of bounds bits in APInt::getLowBitsSet
if the shift count was larger than X's bit width or if it was negative.

Fixes llvm#56424
  • Loading branch information
BertalanD committed Jul 7, 2022
1 parent 6656029 commit ef7aed3
Show file tree
Hide file tree
Showing 2 changed files with 29 additions and 1 deletion.
3 changes: 2 additions & 1 deletion llvm/lib/Transforms/InstCombine/InstCombineAndOrXor.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1808,7 +1808,8 @@ Instruction *InstCombinerImpl::visitAnd(BinaryOperator &I) {

unsigned Width = Ty->getScalarSizeInBits();
const APInt *ShiftC;
if (match(Op0, m_OneUse(m_SExt(m_AShr(m_Value(X), m_APInt(ShiftC)))))) {
if (match(Op0, m_OneUse(m_SExt(m_AShr(m_Value(X), m_APInt(ShiftC))))) &&
ShiftC->ult(Width)) {
if (*C == APInt::getLowBitsSet(Width, Width - ShiftC->getZExtValue())) {
// We are clearing high bits that were potentially set by sext+ashr:
// and (sext (ashr X, ShiftC)), C --> lshr (sext X), ShiftC
Expand Down
27 changes: 27 additions & 0 deletions llvm/test/Transforms/InstCombine/pr56424.ll
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; RUN: opt < %s --passes=instcombine -S | FileCheck %s

; This would crash if we didn't check for a negative shift.
; https://github.com/llvm/llvm-project/issues/56424
define i64 @PR56424(i1 %cond, i32 %arg) {
; CHECK-LABEL: @PR56424(
; CHECK-NEXT: entry:
; CHECK-NEXT: br i1 [[COND:%.*]], label [[IF_THEN:%.*]], label [[IF_END:%.*]]
; CHECK: if.then:
; CHECK-NEXT: br label [[IF_END]]
; CHECK: if.end:
; CHECK-NEXT: ret i64 0
;
entry:
br i1 %cond, label %if.then, label %if.end

if.then:
%shr = ashr i32 %arg, -2
%sext = sext i32 %shr to i64
br label %if.end

if.end:
%val = phi i64 [ %sext, %if.then ], [ 0, %entry ]
%and = and i64 -81, %val
ret i64 %and
}

0 comments on commit ef7aed3

Please sign in to comment.