path: root/CompilerStack.cpp

/*
    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
 * Full-stack compiler that converts a source code string to bytecode.
 */

#include <libsolidity/AST.h>
#include <libsolidity/Scanner.h>
#include <libsolidity/Parser.h>
#include <libsolidity/GlobalContext.h>
#include <libsolidity/NameAndTypeResolver.h>
#include <libsolidity/Compiler.h>
#include <libsolidity/CompilerStack.h>
#include <libsolidity/InterfaceHandler.h>

using namespace std;

namespace dev
{
namespace solidity
{

bool CompilerStack::addSource(string const& _name, string const& _content)
{
    bool existed = m_sources.count(_name);
    reset(true);
    m_sources[_name].scanner = make_shared<Scanner>(CharStream(_content), _name);
    return existed;
}

void CompilerStack::setSource(string const& _sourceCode)
{
    reset();
    addSource("", _sourceCode);
}

void CompilerStack::parse()
{
    for (auto& sourcePair: m_sources)
    {
        sourcePair.second.scanner->reset();
        sourcePair.second.ast = Parser().parse(sourcePair.second.scanner);
    }
    resolveImports();

    m_globalContext = make_shared<GlobalContext>();
    NameAndTypeResolver resolver(m_globalContext->getDeclarations());
    for (Source const* source: m_sourceOrder)
        resolver.registerDeclarations(*source->ast);
    for (Source const* source: m_sourceOrder)
        for (ASTPointer<ASTNode> const& node: source->ast->getNodes())
            if (ContractDefinition* contract = dynamic_cast<ContractDefinition*>(node.get()))
            {
                m_globalContext->setCurrentContract(*contract);
                resolver.updateDeclaration(*m_globalContext->getCurrentThis());
                resolver.resolveNamesAndTypes(*contract);
                m_contracts[contract->getName()].contract = contract;
            }
    for (Source const* source: m_sourceOrder)
        for (ASTPointer<ASTNode> const& node: source->ast->getNodes())
            if (ContractDefinition* contract = dynamic_cast<ContractDefinition*>(node.get()))
            {
                m_globalContext->setCurrentContract(*contract);
                resolver.updateDeclaration(*m_globalContext->getCurrentThis());
                resolver.checkTypeRequirements(*contract);
                m_contracts[contract->getName()].contract = contract;
            }
    m_parseSuccessful = true;
}

void CompilerStack::parse(string const& _sourceCode)
{
    setSource(_sourceCode);
    parse();
}

vector<string> CompilerStack::getContractNames() const
{
    if (!m_parseSuccessful)
        BOOST_THROW_EXCEPTION(CompilerError() << errinfo_comment("Parsing was not successful."));
    vector<string> contractNames;
    for (auto const& contract: m_contracts)
        contractNames.push_back(contract.first);
    return contractNames;
}

void CompilerStack::compile(bool _optimize)
{
    if (!m_parseSuccessful)
        parse();

    map<ContractDefinition const*, bytes const*> contractBytecode;
    for (Source const* source: m_sourceOrder)
        for (ASTPointer<ASTNode> const& node: source->ast->getNodes())
            if (ContractDefinition* contract = dynamic_cast<ContractDefinition*>(node.get()))
            {
                m_globalContext->setCurrentContract(*contract);
                shared_ptr<Compiler> compiler = make_shared<Compiler>(_optimize);
                compiler->compileContract(*contract, m_globalContext->getMagicVariables(),
                                          contractBytecode);
                Contract& compiledContract = m_contracts[contract->getName()];
                compiledContract.bytecode = compiler->getAssembledBytecode();
                compiledContract.compiler = move(compiler);
                contractBytecode[compiledContract.contract] = &compiledContract.bytecode;
            }
}

bytes const& CompilerStack::compile(string const& _sourceCode, bool _optimize)
{
    parse(_sourceCode);
    compile(_optimize);
    return getBytecode();
}

bytes const& CompilerStack::getBytecode(string const& _contractName) const
{
    return getContract(_contractName).bytecode;
}

void CompilerStack::streamAssembly(ostream& _outStream, string const& _contractName) const
{
    getContract(_contractName).compiler->streamAssembly(_outStream);
}

string const& CompilerStack::getInterface(string const& _contractName) const
{
    return getJsonDocumentation(_contractName, DocumentationType::ABI_INTERFACE);
}

string const& CompilerStack::getJsonDocumentation(string const& _contractName, DocumentationType _type) const
{
    if (!m_parseSuccessful)
        BOOST_THROW_EXCEPTION(CompilerError() << errinfo_comment("Parsing was not successful."));

    Contract const& contract = getContract(_contractName);

    std::unique_ptr<string const>* doc;
    switch (_type)
    {
    case DocumentationType::NATSPEC_USER:
        doc = &contract.userDocumentation;
        break;
    case DocumentationType::NATSPEC_DEV:
        doc = &contract.devDocumentation;
        break;
    case DocumentationType::ABI_INTERFACE:
        doc = &contract.interface;
        break;
    default:
        BOOST_THROW_EXCEPTION(InternalCompilerError() << errinfo_comment("Illegal documentation type."));
    }
    if (!*doc)
        *doc = contract.interfaceHandler->getDocumentation(*contract.contract, _type);
    return *(*doc);
}

Scanner const& CompilerStack::getScanner(string const& _sourceName) const
{
    return *getSource(_sourceName).scanner;
}

SourceUnit const& CompilerStack::getAST(string const& _sourceName) const
{
    return *getSource(_sourceName).ast;
}

ContractDefinition const& CompilerStack::getContractDefinition(string const& _contractName) const
{
    return *getContract(_contractName).contract;
}

bytes CompilerStack::staticCompile(std::string const& _sourceCode, bool _optimize)
{
    CompilerStack stack;
    return stack.compile(_sourceCode, _optimize);
}

void CompilerStack::reset(bool _keepSources)
{
    m_parseSuccessful = false;
    if (_keepSources)
        for (auto sourcePair: m_sources)
            sourcePair.second.reset();
    else
        m_sources.clear();
    m_globalContext.reset();
    m_sourceOrder.clear();
    m_contracts.clear();
}

void CompilerStack::resolveImports()
{
    // topological sorting (depth first search) of the import graph, cutting potential cycles
    vector<Source const*> sourceOrder;
    set<Source const*> sourcesSeen;

    function<void(Source const*)> toposort = [&](Source const* _source)
    {
        if (sourcesSeen.count(_source))
            return;
        sourcesSeen.insert(_source);
        for (ASTPointer<ASTNode> const& node: _source->ast->getNodes())
            if (ImportDirective const* import = dynamic_cast<ImportDirective*>(node.get()))
            {
                string const& id = import->getIdentifier();
                if (!m_sources.count(id))
                    BOOST_THROW_EXCEPTION(ParserError()
                                          << errinfo_sourceLocation(import->getLocation())
                                          << errinfo_comment("Source not found."));
                toposort(&m_sources[id]);
            }
        sourceOrder.push_back(_source);
    };

    for (auto const& sourcePair: m_sources)
        toposort(&sourcePair.second);

    swap(m_sourceOrder, sourceOrder);
}

CompilerStack::Contract const& CompilerStack::getContract(string const& _contractName) const
{
    if (m_contracts.empty())
        BOOST_THROW_EXCEPTION(CompilerError() << errinfo_comment("No compiled contracts found."));
    string contractName = _contractName;
    if (_contractName.empty())
        // try to find the "last contract"
        for (ASTPointer<ASTNode> const& node: m_sourceOrder.back()->ast->getNodes())
            if (auto contract = dynamic_cast<ContractDefinition const*>(node.get()))
                contractName = contract->getName();
    auto it = m_contracts.find(contractName);
    if (it == m_contracts.end())
        BOOST_THROW_EXCEPTION(CompilerError() << errinfo_comment("Contract " + _contractName + " not found."));
    return it->second;
}

CompilerStack::Source const& CompilerStack::getSource(string const& _sourceName) const
{
    auto it = m_sources.find(_sourceName);
    if (it == m_sources.end())
        BOOST_THROW_EXCEPTION(CompilerError() << errinfo_comment("Given source file not found."));
    return it->second;
}

CompilerStack::Contract::Contract(): interfaceHandler(make_shared<InterfaceHandler>()) {}

}
}