Lefteris Karapetsas
10 years ago
43 changed files with 6455 additions and 292 deletions
@ -0,0 +1,104 @@ |
|||
/*
|
|||
This file is part of cpp-ethereum. |
|||
|
|||
cpp-ethereum is free software: you can redistribute it and/or modify |
|||
it under the terms of the GNU General Public License as published by |
|||
the Free Software Foundation, either version 3 of the License, or |
|||
(at your option) any later version. |
|||
|
|||
cpp-ethereum is distributed in the hope that it will be useful, |
|||
but WITHOUT ANY WARRANTY; without even the implied warranty of |
|||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|||
GNU General Public License for more details. |
|||
|
|||
You should have received a copy of the GNU General Public License |
|||
along with cpp-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|||
*/ |
|||
/** @file GasMeter.cpp
|
|||
* @author Christian <c@ethdev.com> |
|||
* @date 2015 |
|||
*/ |
|||
|
|||
#include "GasMeter.h" |
|||
#include <libevmcore/Params.h> |
|||
|
|||
using namespace std; |
|||
using namespace dev; |
|||
using namespace dev::eth; |
|||
|
|||
GasMeter::GasConsumption& GasMeter::GasConsumption::operator+=(GasConsumption const& _other) |
|||
{ |
|||
isInfinite = isInfinite || _other.isInfinite; |
|||
if (isInfinite) |
|||
return *this; |
|||
bigint v = bigint(value) + _other.value; |
|||
if (v > std::numeric_limits<u256>::max()) |
|||
isInfinite = true; |
|||
else |
|||
value = u256(v); |
|||
return *this; |
|||
} |
|||
|
|||
GasMeter::GasConsumption GasMeter::estimateMax(AssemblyItem const& _item) |
|||
{ |
|||
switch (_item.type()) { |
|||
case Push: |
|||
case PushTag: |
|||
return runGas(Instruction::PUSH1); |
|||
case Tag: |
|||
return runGas(Instruction::JUMPDEST); |
|||
case Operation: |
|||
{ |
|||
GasConsumption gas = runGas(_item.instruction()); |
|||
switch (_item.instruction()) |
|||
{ |
|||
case Instruction::SSTORE: |
|||
// @todo logic can be improved
|
|||
gas += c_sstoreSetGas; |
|||
break; |
|||
case Instruction::SLOAD: |
|||
gas += c_sloadGas; |
|||
break; |
|||
case Instruction::MSTORE: |
|||
case Instruction::MSTORE8: |
|||
case Instruction::MLOAD: |
|||
case Instruction::RETURN: |
|||
case Instruction::SHA3: |
|||
case Instruction::CALLDATACOPY: |
|||
case Instruction::CODECOPY: |
|||
case Instruction::EXTCODECOPY: |
|||
case Instruction::LOG0: |
|||
case Instruction::LOG1: |
|||
case Instruction::LOG2: |
|||
case Instruction::LOG3: |
|||
case Instruction::LOG4: |
|||
case Instruction::CALL: |
|||
case Instruction::CALLCODE: |
|||
case Instruction::CREATE: |
|||
case Instruction::EXP: |
|||
// @todo logic can be improved
|
|||
gas = GasConsumption::infinite(); |
|||
break; |
|||
default: |
|||
break; |
|||
} |
|||
return gas; |
|||
break; |
|||
} |
|||
default: |
|||
break; |
|||
} |
|||
|
|||
return GasConsumption::infinite(); |
|||
} |
|||
|
|||
GasMeter::GasConsumption GasMeter::runGas(Instruction _instruction) |
|||
{ |
|||
if (_instruction == Instruction::JUMPDEST) |
|||
return GasConsumption(1); |
|||
|
|||
int tier = instructionInfo(_instruction).gasPriceTier; |
|||
return tier == InvalidTier ? GasConsumption::infinite() : c_tierStepGas[tier]; |
|||
} |
|||
|
|||
|
@ -0,0 +1,67 @@ |
|||
/*
|
|||
This file is part of cpp-ethereum. |
|||
|
|||
cpp-ethereum is free software: you can redistribute it and/or modify |
|||
it under the terms of the GNU General Public License as published by |
|||
the Free Software Foundation, either version 3 of the License, or |
|||
(at your option) any later version. |
|||
|
|||
cpp-ethereum is distributed in the hope that it will be useful, |
|||
but WITHOUT ANY WARRANTY; without even the implied warranty of |
|||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|||
GNU General Public License for more details. |
|||
|
|||
You should have received a copy of the GNU General Public License |
|||
along with cpp-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|||
*/ |
|||
/** @file GasMeter.cpp
|
|||
* @author Christian <c@ethdev.com> |
|||
* @date 2015 |
|||
*/ |
|||
|
|||
#pragma once |
|||
|
|||
#include <ostream> |
|||
#include <libevmasm/AssemblyItem.h> |
|||
|
|||
namespace dev |
|||
{ |
|||
namespace eth |
|||
{ |
|||
|
|||
/**
|
|||
* Class that helps computing the maximum gas consumption for instructions. |
|||
*/ |
|||
class GasMeter |
|||
{ |
|||
public: |
|||
struct GasConsumption |
|||
{ |
|||
GasConsumption(u256 _value = 0, bool _infinite = false): value(_value), isInfinite(_infinite) {} |
|||
static GasConsumption infinite() { return GasConsumption(0, true); } |
|||
|
|||
GasConsumption& operator+=(GasConsumption const& _otherS); |
|||
std::ostream& operator<<(std::ostream& _str) const; |
|||
|
|||
u256 value; |
|||
bool isInfinite; |
|||
}; |
|||
|
|||
/// Returns an upper bound on the gas consumed by the given instruction.
|
|||
GasConsumption estimateMax(AssemblyItem const& _item); |
|||
|
|||
private: |
|||
static GasConsumption runGas(Instruction _instruction); |
|||
}; |
|||
|
|||
inline std::ostream& operator<<(std::ostream& _str, GasMeter::GasConsumption const& _consumption) |
|||
{ |
|||
if (_consumption.isInfinite) |
|||
return _str << "inf"; |
|||
else |
|||
return _str << _consumption.value; |
|||
} |
|||
|
|||
|
|||
} |
|||
} |
@ -0,0 +1,110 @@ |
|||
/*
|
|||
This file is part of cpp-ethereum. |
|||
|
|||
cpp-ethereum is free software: you can redistribute it and/or modify |
|||
it under the terms of the GNU General Public License as published by |
|||
the Free Software Foundation, either version 3 of the License, or |
|||
(at your option) any later version. |
|||
|
|||
cpp-ethereum is distributed in the hope that it will be useful, |
|||
but WITHOUT ANY WARRANTY; without even the implied warranty of |
|||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|||
GNU General Public License for more details. |
|||
|
|||
You should have received a copy of the GNU General Public License |
|||
along with cpp-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|||
*/ |
|||
/**
|
|||
* @author Christian <c@ethdev.com> |
|||
* @date 2015 |
|||
* Gas consumption estimator working alongside the AST. |
|||
*/ |
|||
|
|||
#include "StructuralGasEstimator.h" |
|||
#include <map> |
|||
#include <functional> |
|||
#include <libsolidity/AST.h> |
|||
#include <libsolidity/ASTVisitor.h> |
|||
|
|||
using namespace std; |
|||
using namespace dev; |
|||
using namespace dev::eth; |
|||
using namespace dev::solidity; |
|||
|
|||
StructuralGasEstimator::ASTGasConsumptionSelfAccumulated StructuralGasEstimator::performEstimation( |
|||
AssemblyItems const& _items, |
|||
vector<ASTNode const*> const& _ast |
|||
) |
|||
{ |
|||
solAssert(std::count(_ast.begin(), _ast.end(), nullptr) == 0, ""); |
|||
map<SourceLocation, GasMeter::GasConsumption> particularCosts; |
|||
GasMeter meter; |
|||
|
|||
for (auto const& item: _items) |
|||
particularCosts[item.getLocation()] += meter.estimateMax(item); |
|||
|
|||
ASTGasConsumptionSelfAccumulated gasCosts; |
|||
auto onNode = [&](ASTNode const& _node) |
|||
{ |
|||
gasCosts[&_node][0] = gasCosts[&_node][1] = particularCosts[_node.getLocation()]; |
|||
return true; |
|||
}; |
|||
auto onEdge = [&](ASTNode const& _parent, ASTNode const& _child) |
|||
{ |
|||
gasCosts[&_parent][1] += gasCosts[&_child][1]; |
|||
}; |
|||
ASTReduce folder(onNode, onEdge); |
|||
for (ASTNode const* ast: _ast) |
|||
ast->accept(folder); |
|||
|
|||
return gasCosts; |
|||
} |
|||
|
|||
map<ASTNode const*, GasMeter::GasConsumption> StructuralGasEstimator::breakToStatementLevel( |
|||
ASTGasConsumptionSelfAccumulated const& _gasCosts, |
|||
vector<ASTNode const*> const& _roots |
|||
) |
|||
{ |
|||
solAssert(std::count(_roots.begin(), _roots.end(), nullptr) == 0, ""); |
|||
// first pass: statementDepth[node] is the distance from the deepend statement to node
|
|||
// in direction of the tree root (or undefined if not possible)
|
|||
map<ASTNode const*, int> statementDepth; |
|||
auto onNodeFirstPass = [&](ASTNode const& _node) |
|||
{ |
|||
if (dynamic_cast<Statement const*>(&_node)) |
|||
statementDepth[&_node] = 0; |
|||
return true; |
|||
}; |
|||
auto onEdgeFirstPass = [&](ASTNode const& _parent, ASTNode const& _child) |
|||
{ |
|||
if (statementDepth.count(&_child)) |
|||
statementDepth[&_parent] = max(statementDepth[&_parent], statementDepth[&_child] + 1); |
|||
}; |
|||
ASTReduce firstPass(onNodeFirstPass, onEdgeFirstPass); |
|||
for (ASTNode const* node: _roots) |
|||
node->accept(firstPass); |
|||
|
|||
// we use the location of a node if
|
|||
// - its statement depth is 0 or
|
|||
// - its statement depth is undefined but the parent's statement depth is at least 1
|
|||
map<ASTNode const*, GasMeter::GasConsumption> gasCosts; |
|||
auto onNodeSecondPass = [&](ASTNode const& _node) |
|||
{ |
|||
return statementDepth.count(&_node); |
|||
}; |
|||
auto onEdgeSecondPass = [&](ASTNode const& _parent, ASTNode const& _child) |
|||
{ |
|||
bool useNode = false; |
|||
if (statementDepth.count(&_child)) |
|||
useNode = statementDepth[&_child] == 0; |
|||
else |
|||
useNode = statementDepth.count(&_parent) && statementDepth.at(&_parent) > 0; |
|||
if (useNode) |
|||
gasCosts[&_child] = _gasCosts.at(&_child)[1]; |
|||
}; |
|||
ASTReduce secondPass(onNodeSecondPass, onEdgeSecondPass); |
|||
for (ASTNode const* node: _roots) |
|||
node->accept(secondPass); |
|||
// gasCosts should only contain non-overlapping locations
|
|||
return gasCosts; |
|||
} |
@ -0,0 +1,62 @@ |
|||
/*
|
|||
This file is part of cpp-ethereum. |
|||
|
|||
cpp-ethereum is free software: you can redistribute it and/or modify |
|||
it under the terms of the GNU General Public License as published by |
|||
the Free Software Foundation, either version 3 of the License, or |
|||
(at your option) any later version. |
|||
|
|||
cpp-ethereum is distributed in the hope that it will be useful, |
|||
but WITHOUT ANY WARRANTY; without even the implied warranty of |
|||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|||
GNU General Public License for more details. |
|||
|
|||
You should have received a copy of the GNU General Public License |
|||
along with cpp-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|||
*/ |
|||
/**
|
|||
* @author Christian <c@ethdev.com> |
|||
* @date 2015 |
|||
* Gas consumption estimator working alongside the AST. |
|||
*/ |
|||
|
|||
#pragma once |
|||
|
|||
#include <vector> |
|||
#include <map> |
|||
#include <array> |
|||
#include <libsolidity/ASTForward.h> |
|||
#include <libevmasm/GasMeter.h> |
|||
#include <libevmasm/Assembly.h> |
|||
|
|||
namespace dev |
|||
{ |
|||
namespace solidity |
|||
{ |
|||
|
|||
class StructuralGasEstimator |
|||
{ |
|||
public: |
|||
using ASTGasConsumption = std::map<ASTNode const*, eth::GasMeter::GasConsumption>; |
|||
using ASTGasConsumptionSelfAccumulated = |
|||
std::map<ASTNode const*, std::array<eth::GasMeter::GasConsumption, 2>>; |
|||
|
|||
/// Estimates the gas consumption for every assembly item in the given assembly and stores
|
|||
/// it by source location.
|
|||
/// @returns a mapping from each AST node to a pair of its particular and syntactically accumulated gas costs.
|
|||
ASTGasConsumptionSelfAccumulated performEstimation( |
|||
eth::AssemblyItems const& _items, |
|||
std::vector<ASTNode const*> const& _ast |
|||
); |
|||
/// @returns a mapping from nodes with non-overlapping source locations to gas consumptions such that
|
|||
/// the following source locations are part of the mapping:
|
|||
/// 1. source locations of statements that do not contain other statements
|
|||
/// 2. maximal source locations that do not overlap locations coming from the first rule
|
|||
ASTGasConsumption breakToStatementLevel( |
|||
ASTGasConsumptionSelfAccumulated const& _gasCosts, |
|||
std::vector<ASTNode const*> const& _roots |
|||
); |
|||
}; |
|||
|
|||
} |
|||
} |
@ -0,0 +1,98 @@ |
|||
/*
|
|||
This file is part of cpp-ethereum. |
|||
|
|||
cpp-ethereum is free software: you can redistribute it and/or modify |
|||
it under the terms of the GNU General Public License as published by |
|||
the Free Software Foundation, either version 3 of the License, or |
|||
(at your option) any later version. |
|||
|
|||
cpp-ethereum is distributed in the hope that it will be useful, |
|||
but WITHOUT ANY WARRANTY; without even the implied warranty of |
|||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|||
GNU General Public License for more details. |
|||
|
|||
You should have received a copy of the GNU General Public License |
|||
along with cpp-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|||
*/ |
|||
/**
|
|||
* @author Christian <c@ethdev.com> |
|||
* @date 2015 |
|||
* Unit tests for the gas estimator. |
|||
*/ |
|||
|
|||
#include <test/libsolidity/solidityExecutionFramework.h> |
|||
#include <libsolidity/AST.h> |
|||
#include <libsolidity/StructuralGasEstimator.h> |
|||
#include <libsolidity/SourceReferenceFormatter.h> |
|||
|
|||
using namespace std; |
|||
using namespace dev::eth; |
|||
using namespace dev::solidity; |
|||
|
|||
namespace dev |
|||
{ |
|||
namespace solidity |
|||
{ |
|||
namespace test |
|||
{ |
|||
|
|||
class GasMeterTestFramework: public ExecutionFramework |
|||
{ |
|||
public: |
|||
GasMeterTestFramework() { } |
|||
void compile(string const& _sourceCode) |
|||
{ |
|||
m_compiler.setSource(_sourceCode); |
|||
ETH_TEST_REQUIRE_NO_THROW(m_compiler.compile(), "Compiling contract failed"); |
|||
|
|||
StructuralGasEstimator estimator; |
|||
AssemblyItems const* items = m_compiler.getRuntimeAssemblyItems(""); |
|||
ASTNode const& sourceUnit = m_compiler.getAST(); |
|||
BOOST_REQUIRE(items != nullptr); |
|||
m_gasCosts = estimator.breakToStatementLevel( |
|||
estimator.performEstimation(*items, vector<ASTNode const*>({&sourceUnit})), |
|||
{&sourceUnit} |
|||
); |
|||
} |
|||
|
|||
protected: |
|||
dev::solidity::CompilerStack m_compiler; |
|||
map<ASTNode const*, eth::GasMeter::GasConsumption> m_gasCosts; |
|||
}; |
|||
|
|||
BOOST_FIXTURE_TEST_SUITE(GasMeterTests, GasMeterTestFramework) |
|||
|
|||
BOOST_AUTO_TEST_CASE(non_overlapping_filtered_costs) |
|||
{ |
|||
char const* sourceCode = R"( |
|||
contract test { |
|||
bytes x; |
|||
function f(uint a) returns (uint b) { |
|||
x.length = a; |
|||
for (; a < 200; ++a) { |
|||
x[a] = 9; |
|||
b = a * a; |
|||
} |
|||
return f(a - 1); |
|||
} |
|||
} |
|||
)"; |
|||
compile(sourceCode); |
|||
for (auto first = m_gasCosts.cbegin(); first != m_gasCosts.cend(); ++first) |
|||
{ |
|||
auto second = first; |
|||
for (++second; second != m_gasCosts.cend(); ++second) |
|||
if (first->first->getLocation().intersects(second->first->getLocation())) |
|||
{ |
|||
BOOST_CHECK_MESSAGE(false, "Source locations should not overlap!"); |
|||
SourceReferenceFormatter::printSourceLocation(cout, first->first->getLocation(), m_compiler.getScanner()); |
|||
SourceReferenceFormatter::printSourceLocation(cout, second->first->getLocation(), m_compiler.getScanner()); |
|||
} |
|||
} |
|||
} |
|||
|
|||
BOOST_AUTO_TEST_SUITE_END() |
|||
|
|||
} |
|||
} |
|||
} |
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
Loading…
Reference in new issue