diff options
Diffstat (limited to 'libevmasm/ConstantOptimiser.h')
| -rw-r--r-- | libevmasm/ConstantOptimiser.h | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/libevmasm/ConstantOptimiser.h b/libevmasm/ConstantOptimiser.h index 4f12c49f..8a37b3cb 100644 --- a/libevmasm/ConstantOptimiser.h +++ b/libevmasm/ConstantOptimiser.h @@ -21,10 +21,13 @@ #pragma once -#include <vector> +#include <libevmasm/Exceptions.h> + #include <libdevcore/CommonData.h> #include <libdevcore/CommonIO.h> +#include <vector> + namespace dev { namespace eth @@ -130,6 +133,8 @@ public: ConstantOptimisationMethod(_params, _value) { m_routine = findRepresentation(m_value); + if (!checkRepresentation(m_value)) + BOOST_THROW_EXCEPTION(AssemblyException()); } virtual bigint gasNeeded() override { return gasNeeded(m_routine); } @@ -141,6 +146,8 @@ public: protected: /// Tries to recursively find a way to compute @a _value. AssemblyItems findRepresentation(u256 const& _value); + /// Recomputes the value from the calculated representation and checks for correctness. + bool checkRepresentation(u256 const& _value); bigint gasNeeded(AssemblyItems const& _routine); /// Counter for the complexity of optimization, will stop when it reaches zero. |