solidity/libsolidity/formal/VariableUsage.cpp

120 lines
3.7 KiB
C++
Raw Normal View History

2017-09-28 13:24:24 +00:00
/*
This file is part of solidity.
solidity 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.
solidity 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 solidity. If not, see <http://www.gnu.org/licenses/>.
*/
// SPDX-License-Identifier: GPL-3.0
2017-09-28 13:24:24 +00:00
#include <libsolidity/formal/VariableUsage.h>
#include <libsolidity/formal/BMC.h>
#include <libsolidity/formal/SMTEncoder.h>
#include <range/v3/view.hpp>
2019-04-01 09:10:28 +00:00
#include <algorithm>
2017-09-28 13:24:24 +00:00
using namespace std;
2019-12-11 16:31:36 +00:00
using namespace solidity;
using namespace solidity::util;
using namespace solidity::frontend;
using namespace solidity::frontend::smt;
2017-09-28 13:24:24 +00:00
set<VariableDeclaration const*> VariableUsage::touchedVariables(ASTNode const& _node, vector<CallableDeclaration const*> const& _outerCallstack)
{
m_touchedVariables.clear();
m_callStack.clear();
m_callStack += _outerCallstack;
if (!m_callStack.empty())
m_lastCall = m_callStack.back();
_node.accept(*this);
return m_touchedVariables;
}
2019-04-01 09:10:28 +00:00
void VariableUsage::endVisit(Identifier const& _identifier)
2017-09-28 13:24:24 +00:00
{
if (_identifier.annotation().willBeWrittenTo)
checkIdentifier(_identifier);
}
void VariableUsage::endVisit(IndexAccess const& _indexAccess)
{
if (_indexAccess.annotation().willBeWrittenTo)
{
/// identifier.annotation().willBeWrittenTo == false, that's why we
/// need to check that before.
auto identifier = dynamic_cast<Identifier const*>(SMTEncoder::leftmostBase(_indexAccess));
if (identifier)
checkIdentifier(*identifier);
}
2017-09-28 13:24:24 +00:00
}
2019-04-01 09:10:28 +00:00
void VariableUsage::endVisit(FunctionCall const& _funCall)
2017-09-28 13:24:24 +00:00
{
auto scopeContract = currentScopeContract();
2021-01-25 17:17:56 +00:00
if (m_inlineFunctionCalls(_funCall, scopeContract, m_currentContract))
if (auto funDef = SMTEncoder::functionCallToDefinition(_funCall, scopeContract, m_currentContract))
if (find(m_callStack.begin(), m_callStack.end(), funDef) == m_callStack.end())
funDef->accept(*this);
2019-04-01 09:10:28 +00:00
}
2017-09-28 13:24:24 +00:00
2019-04-01 09:10:28 +00:00
bool VariableUsage::visit(FunctionDefinition const& _function)
{
m_callStack.push_back(&_function);
2019-04-01 09:10:28 +00:00
return true;
}
2017-09-28 13:24:24 +00:00
2019-04-01 09:10:28 +00:00
void VariableUsage::endVisit(FunctionDefinition const&)
{
solAssert(!m_callStack.empty(), "");
m_callStack.pop_back();
2019-04-01 09:10:28 +00:00
}
2019-04-01 09:10:28 +00:00
void VariableUsage::endVisit(ModifierInvocation const& _modifierInv)
{
auto const& modifierDef = dynamic_cast<ModifierDefinition const*>(_modifierInv.name().annotation().referencedDeclaration);
2019-04-01 09:10:28 +00:00
if (modifierDef)
modifierDef->accept(*this);
}
2019-04-01 09:10:28 +00:00
void VariableUsage::endVisit(PlaceholderStatement const&)
{
solAssert(!m_callStack.empty(), "");
FunctionDefinition const* funDef = nullptr;
for (auto it = m_callStack.rbegin(); it != m_callStack.rend() && !funDef; ++it)
funDef = dynamic_cast<FunctionDefinition const*>(*it);
solAssert(funDef, "");
if (funDef->isImplemented())
funDef->body().accept(*this);
2019-04-01 09:10:28 +00:00
}
2017-09-28 13:24:24 +00:00
void VariableUsage::checkIdentifier(Identifier const& _identifier)
2019-04-01 09:10:28 +00:00
{
Declaration const* declaration = _identifier.annotation().referencedDeclaration;
solAssert(declaration, "");
if (VariableDeclaration const* varDecl = dynamic_cast<VariableDeclaration const*>(declaration))
{
if (!varDecl->isLocalVariable() || (m_lastCall && varDecl->functionOrModifierDefinition() == m_lastCall))
m_touchedVariables.insert(varDecl);
}
2017-09-28 13:24:24 +00:00
}
ContractDefinition const* VariableUsage::currentScopeContract()
{
for (auto&& f: m_callStack | ranges::views::reverse)
if (auto fun = dynamic_cast<FunctionDefinition const*>(f))
return fun->annotation().contract;
return nullptr;
}