[clang][bytecode] Fix checking for integer overflow (#137962)
We need to evaluate both the True/False expressions of a conditional
operator as well as the LHS/RHS of a binary operator in more cases.
diff --git a/clang/lib/AST/ByteCode/ByteCodeEmitter.h b/clang/lib/AST/ByteCode/ByteCodeEmitter.h
index 5c74829..9e9dd5e 100644
--- a/clang/lib/AST/ByteCode/ByteCodeEmitter.h
+++ b/clang/lib/AST/ByteCode/ByteCodeEmitter.h
@@ -60,6 +60,7 @@
/// We're always emitting bytecode.
bool isActive() const { return true; }
+ bool checkingForUndefinedBehavior() const { return false; }
/// Callback for local registration.
Local createLocal(Descriptor *D);
diff --git a/clang/lib/AST/ByteCode/Compiler.cpp b/clang/lib/AST/ByteCode/Compiler.cpp
index fe8d05c..ae6574c 100644
--- a/clang/lib/AST/ByteCode/Compiler.cpp
+++ b/clang/lib/AST/ByteCode/Compiler.cpp
@@ -866,12 +866,14 @@
// Assignments require us to evalute the RHS first.
if (BO->getOpcode() == BO_Assign) {
- // We don't support assignments in C.
- if (!Ctx.getLangOpts().CPlusPlus)
- return this->emitInvalid(BO);
if (!visit(RHS) || !visit(LHS))
return false;
+
+ // We don't support assignments in C.
+ if (!Ctx.getLangOpts().CPlusPlus && !this->emitInvalid(BO))
+ return false;
+
if (!this->emitFlip(*LT, *RT, BO))
return false;
} else {
@@ -2367,8 +2369,19 @@
return false;
}
- if (!this->visitBool(Condition))
+ if (!this->visitBool(Condition)) {
+ // If the condition failed and we're checking for undefined behavior
+ // (which only happens with EvalEmitter) check the TrueExpr and FalseExpr
+ // as well.
+ if (this->checkingForUndefinedBehavior()) {
+ if (!this->discard(TrueExpr))
+ return false;
+ if (!this->discard(FalseExpr))
+ return false;
+ }
return false;
+ }
+
if (!this->jumpFalse(LabelFalse))
return false;
if (!visitChildExpr(TrueExpr))
diff --git a/clang/lib/AST/ByteCode/EvalEmitter.h b/clang/lib/AST/ByteCode/EvalEmitter.h
index f53f86c..18adf86 100644
--- a/clang/lib/AST/ByteCode/EvalEmitter.h
+++ b/clang/lib/AST/ByteCode/EvalEmitter.h
@@ -69,6 +69,9 @@
/// Since expressions can only jump forward, predicated execution is
/// used to deal with if-else statements.
bool isActive() const { return CurrentLabel == ActiveLabel; }
+ bool checkingForUndefinedBehavior() const {
+ return S.checkingForUndefinedBehavior();
+ }
/// Callback for registering a local.
Local createLocal(Descriptor *D);
diff --git a/clang/lib/AST/ByteCode/Interp.cpp b/clang/lib/AST/ByteCode/Interp.cpp
index 62b4495..1a7fc6c 100644
--- a/clang/lib/AST/ByteCode/Interp.cpp
+++ b/clang/lib/AST/ByteCode/Interp.cpp
@@ -1336,7 +1336,7 @@
return false;
}
- if (Off > Ptr.block()->getSize())
+ if ((Ptr.getByteOffset() + Off) >= Ptr.block()->getSize())
return false;
S.Stk.push<Pointer>(Ptr.atField(Off));
diff --git a/clang/test/Sema/integer-overflow.c b/clang/test/Sema/integer-overflow.c
index 3141443..30a47aa 100644
--- a/clang/test/Sema/integer-overflow.c
+++ b/clang/test/Sema/integer-overflow.c
@@ -1,4 +1,5 @@
// RUN: %clang_cc1 %s -verify -fsyntax-only -triple x86_64-pc-linux-gnu
+// RUN: %clang_cc1 %s -verify -fsyntax-only -triple x86_64-pc-linux-gnu -fexperimental-new-constant-interpreter
typedef unsigned long long uint64_t;
typedef unsigned int uint32_t;