From 67097d85de6b9b816ee92e82ba964024aa90c0e3 Mon Sep 17 00:00:00 2001 From: Andrei <16517508+anvacaru@users.noreply.github.com> Date: Wed, 4 Dec 2024 16:48:24 +0200 Subject: [PATCH 1/2] add forgetBranch cheatcode --- src/IKontrolCheatsBase.sol | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/IKontrolCheatsBase.sol b/src/IKontrolCheatsBase.sol index bc6d2cf..2fc0b07 100644 --- a/src/IKontrolCheatsBase.sol +++ b/src/IKontrolCheatsBase.sol @@ -3,6 +3,7 @@ pragma solidity >=0.6.2 <0.9.0; pragma experimental ABIEncoderV2; interface KontrolCheatsBase { + enum ComparisonOperator { Equal, NotEqual, LessThanOrEqual, LessThan, GreaterThanOrEqual, GreaterThan } // Expects a call using the CALL opcode to an address with the specified calldata. function expectRegularCall(address,bytes calldata) external; // Expects a call using the CALL opcode to an address with the specified msg.value and calldata. @@ -35,4 +36,6 @@ interface KontrolCheatsBase { function freshBytes(uint256) external view returns (bytes memory); // Returns a symbolic address function freshAddress() external view returns (address); + // Removes a path constraint + function forgetBranch(uint256, ComparisonOperator, uint256) external; } From 2f732254b3f84ea75de2fc25d79da29adde13adf Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Petar=20Maksimovi=C4=87?= Date: Thu, 5 Dec 2024 17:09:06 +0000 Subject: [PATCH 2/2] Update src/IKontrolCheatsBase.sol --- src/IKontrolCheatsBase.sol | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/IKontrolCheatsBase.sol b/src/IKontrolCheatsBase.sol index 2fc0b07..9480d1b 100644 --- a/src/IKontrolCheatsBase.sol +++ b/src/IKontrolCheatsBase.sol @@ -36,6 +36,6 @@ interface KontrolCheatsBase { function freshBytes(uint256) external view returns (bytes memory); // Returns a symbolic address function freshAddress() external view returns (address); - // Removes a path constraint + // Removes a branching condition from the path constraints function forgetBranch(uint256, ComparisonOperator, uint256) external; }