path: root/libsolidity/inlineasm/AsmScopeFiller.cpp

/*
    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/>.
*/
/**
 * Module responsible for registering identifiers inside their scopes.
 */

#include <libsolidity/inlineasm/AsmScopeFiller.h>

#include <libsolidity/inlineasm/AsmData.h>
#include <libsolidity/inlineasm/AsmScope.h>

#include <libsolidity/interface/Exceptions.h>
#include <libsolidity/interface/Utils.h>

#include <boost/range/adaptor/reversed.hpp>

#include <memory>
#include <functional>

using namespace std;
using namespace dev;
using namespace dev::solidity;
using namespace dev::solidity::assembly;

ScopeFiller::ScopeFiller(ScopeFiller::Scopes& _scopes, ErrorList& _errors):
    m_scopes(_scopes), m_errors(_errors)
{
    m_currentScope = &scope(nullptr);
}

bool ScopeFiller::operator()(Label const& _item)
{
    if (!m_currentScope->registerLabel(_item.name))
    {
        //@TODO secondary location
        m_errors.push_back(make_shared<Error>(
            Error::Type::DeclarationError,
            "Label name " + _item.name + " already taken in this scope.",
            _item.location
        ));
        return false;
    }
    return true;
}

bool ScopeFiller::operator()(assembly::VariableDeclaration const& _varDecl)
{
    for (auto const& variable: _varDecl.variables)
        if (!registerVariable(variable, _varDecl.location, *m_currentScope))
            return false;
    return true;
}

bool ScopeFiller::operator()(assembly::FunctionDefinition const& _funDef)
{
    bool success = true;
    vector<Scope::JuliaType> arguments;
    for (auto const& _argument: _funDef.arguments)
        arguments.push_back(_argument.type);
    vector<Scope::JuliaType> returns;
    for (auto const& _return: _funDef.returns)
        returns.push_back(_return.type);
    if (!m_currentScope->registerFunction(_funDef.name, arguments, returns))
    {
        //@TODO secondary location
        m_errors.push_back(make_shared<Error>(
            Error::Type::DeclarationError,
            "Function name " + _funDef.name + " already taken in this scope.",
            _funDef.location
        ));
        success = false;
    }
    Scope& body = scope(&_funDef.body);
    body.superScope = m_currentScope;
    body.functionScope = true;
    for (auto const& var: _funDef.arguments + _funDef.returns)
        if (!registerVariable(var, _funDef.location, body))
            success = false;

    if (!(*this)(_funDef.body))
        success = false;

    return success;
}

bool ScopeFiller::operator()(Switch const& _switch)
{
    bool success = true;
    for (auto const& _case: _switch.cases)
        if (!(*this)(_case.body))
            success = false;
    return success;
}

bool ScopeFiller::operator()(Block const& _block)
{
    bool success = true;
    scope(&_block).superScope = m_currentScope;
    m_currentScope = &scope(&_block);

    for (auto const& s: _block.statements)
        if (!boost::apply_visitor(*this, s))
            success = false;

    m_currentScope = m_currentScope->superScope;
    return success;
}

bool ScopeFiller::registerVariable(TypedName const& _name, SourceLocation const& _location, Scope& _scope)
{
    if (!_scope.registerVariable(_name.name, _name.type))
    {
        //@TODO secondary location
        m_errors.push_back(make_shared<Error>(
            Error::Type::DeclarationError,
            "Variable name " + _name.name + " already taken in this scope.",
            _location
        ));
        return false;
    }
    return true;
}

Scope& ScopeFiller::scope(Block const* _block)
{
    auto& scope = m_scopes[_block];
    if (!scope)
        scope = make_shared<Scope>();
    return *scope;
}