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