Christian
10 years ago
15 changed files with 183 additions and 332 deletions
@ -1,105 +0,0 @@ |
|||||
|
|
||||
/*
|
|
||||
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 2014 |
|
||||
* Callgraph of functions inside a contract. |
|
||||
*/ |
|
||||
|
|
||||
#include <libsolidity/AST.h> |
|
||||
#include <libsolidity/CallGraph.h> |
|
||||
|
|
||||
using namespace std; |
|
||||
|
|
||||
namespace dev |
|
||||
{ |
|
||||
namespace solidity |
|
||||
{ |
|
||||
|
|
||||
void CallGraph::addNode(ASTNode const& _node) |
|
||||
{ |
|
||||
if (!m_nodesSeen.count(&_node)) |
|
||||
{ |
|
||||
m_workQueue.push(&_node); |
|
||||
m_nodesSeen.insert(&_node); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
set<FunctionDefinition const*> const& CallGraph::getCalls() |
|
||||
{ |
|
||||
computeCallGraph(); |
|
||||
return m_functionsSeen; |
|
||||
} |
|
||||
|
|
||||
void CallGraph::computeCallGraph() |
|
||||
{ |
|
||||
while (!m_workQueue.empty()) |
|
||||
{ |
|
||||
m_workQueue.front()->accept(*this); |
|
||||
m_workQueue.pop(); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
bool CallGraph::visit(Identifier const& _identifier) |
|
||||
{ |
|
||||
if (auto fun = dynamic_cast<FunctionDefinition const*>(_identifier.getReferencedDeclaration())) |
|
||||
{ |
|
||||
if (m_functionOverrideResolver) |
|
||||
fun = (*m_functionOverrideResolver)(fun->getName()); |
|
||||
solAssert(fun, "Error finding override for function " + fun->getName()); |
|
||||
addNode(*fun); |
|
||||
} |
|
||||
if (auto modifier = dynamic_cast<ModifierDefinition const*>(_identifier.getReferencedDeclaration())) |
|
||||
{ |
|
||||
if (m_modifierOverrideResolver) |
|
||||
modifier = (*m_modifierOverrideResolver)(modifier->getName()); |
|
||||
solAssert(modifier, "Error finding override for modifier " + modifier->getName()); |
|
||||
addNode(*modifier); |
|
||||
} |
|
||||
return true; |
|
||||
} |
|
||||
|
|
||||
bool CallGraph::visit(FunctionDefinition const& _function) |
|
||||
{ |
|
||||
m_functionsSeen.insert(&_function); |
|
||||
return true; |
|
||||
} |
|
||||
|
|
||||
bool CallGraph::visit(MemberAccess const& _memberAccess) |
|
||||
{ |
|
||||
// used for "BaseContract.baseContractFunction"
|
|
||||
if (_memberAccess.getExpression().getType()->getCategory() == Type::Category::TYPE) |
|
||||
{ |
|
||||
TypeType const& type = dynamic_cast<TypeType const&>(*_memberAccess.getExpression().getType()); |
|
||||
if (type.getMembers().getMemberType(_memberAccess.getMemberName())) |
|
||||
{ |
|
||||
ContractDefinition const& contract = dynamic_cast<ContractType const&>(*type.getActualType()) |
|
||||
.getContractDefinition(); |
|
||||
for (ASTPointer<FunctionDefinition> const& function: contract.getDefinedFunctions()) |
|
||||
if (function->getName() == _memberAccess.getMemberName()) |
|
||||
{ |
|
||||
addNode(*function); |
|
||||
return true; |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
return true; |
|
||||
} |
|
||||
|
|
||||
} |
|
||||
} |
|
||||
@ -1,69 +0,0 @@ |
|||||
/*
|
|
||||
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 2014 |
|
||||
* Callgraph of functions inside a contract. |
|
||||
*/ |
|
||||
|
|
||||
#include <set> |
|
||||
#include <queue> |
|
||||
#include <functional> |
|
||||
#include <boost/range/iterator_range.hpp> |
|
||||
#include <libsolidity/ASTVisitor.h> |
|
||||
|
|
||||
namespace dev |
|
||||
{ |
|
||||
namespace solidity |
|
||||
{ |
|
||||
|
|
||||
/**
|
|
||||
* Can be used to compute the graph of calls (or rather references) between functions of the same |
|
||||
* contract. Current functionality is limited to computing all functions that are directly |
|
||||
* or indirectly called by some functions. |
|
||||
*/ |
|
||||
class CallGraph: private ASTConstVisitor |
|
||||
{ |
|
||||
public: |
|
||||
using FunctionOverrideResolver = std::function<FunctionDefinition const*(std::string const&)>; |
|
||||
using ModifierOverrideResolver = std::function<ModifierDefinition const*(std::string const&)>; |
|
||||
|
|
||||
CallGraph(FunctionOverrideResolver const& _functionOverrideResolver, |
|
||||
ModifierOverrideResolver const& _modifierOverrideResolver): |
|
||||
m_functionOverrideResolver(&_functionOverrideResolver), |
|
||||
m_modifierOverrideResolver(&_modifierOverrideResolver) {} |
|
||||
|
|
||||
void addNode(ASTNode const& _node); |
|
||||
|
|
||||
std::set<FunctionDefinition const*> const& getCalls(); |
|
||||
|
|
||||
private: |
|
||||
virtual bool visit(FunctionDefinition const& _function) override; |
|
||||
virtual bool visit(Identifier const& _identifier) override; |
|
||||
virtual bool visit(MemberAccess const& _memberAccess) override; |
|
||||
|
|
||||
void computeCallGraph(); |
|
||||
|
|
||||
FunctionOverrideResolver const* m_functionOverrideResolver; |
|
||||
ModifierOverrideResolver const* m_modifierOverrideResolver; |
|
||||
std::set<ASTNode const*> m_nodesSeen; |
|
||||
std::set<FunctionDefinition const*> m_functionsSeen; |
|
||||
std::queue<ASTNode const*> m_workQueue; |
|
||||
}; |
|
||||
|
|
||||
} |
|
||||
} |
|
||||
Loading…
Reference in new issue