aboutsummaryrefslogtreecommitdiffstats
path: root/CompilerContext.cpp
blob: cd22c4e8bfeaa373719c3236cd10a26b1775c1f8 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
/*
    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
 * Utilities for the solidity compiler.
 */

#include <utility>
#include <numeric>
#include <libsolidity/AST.h>
#include <libsolidity/Compiler.h>

using namespace std;

namespace dev {
namespace solidity {

void CompilerContext::addMagicGlobal(MagicVariableDeclaration const& _declaration)
{
    m_magicGlobals.insert(&_declaration);
}

void CompilerContext::addStateVariable(VariableDeclaration const& _declaration)
{
    m_stateVariables[&_declaration] = m_stateVariablesSize;
    m_stateVariablesSize += _declaration.getType()->getStorageSize();
}

void CompilerContext::addVariable(VariableDeclaration const& _declaration)
{
    m_localVariables[&_declaration] = m_localVariablesSize;
    m_localVariablesSize += _declaration.getType()->getSizeOnStack();
}

void CompilerContext::addAndInitializeVariable(VariableDeclaration const& _declaration)
{
    addVariable(_declaration);

    unsigned const size = _declaration.getType()->getSizeOnStack();
    for (unsigned i = 0; i < size; ++i)
        *this << u256(0);
    m_asm.adjustDeposit(-size);
}

void CompilerContext::addFunction(FunctionDefinition const& _function)
{
    m_functionEntryLabels.insert(std::make_pair(&_function, m_asm.newTag()));
}

bool CompilerContext::isLocalVariable(Declaration const* _declaration) const
{
    return m_localVariables.count(_declaration) > 0;
}

eth::AssemblyItem CompilerContext::getFunctionEntryLabel(FunctionDefinition const& _function) const
{
    auto res = m_functionEntryLabels.find(&_function);
    if (asserts(res != m_functionEntryLabels.end()))
        BOOST_THROW_EXCEPTION(InternalCompilerError() << errinfo_comment("Function entry label not found."));
    return res->second.tag();
}

unsigned CompilerContext::getBaseStackOffsetOfVariable(Declaration const& _declaration) const
{
    auto res = m_localVariables.find(&_declaration);
    if (asserts(res != m_localVariables.end()))
        BOOST_THROW_EXCEPTION(InternalCompilerError() << errinfo_comment("Variable not found on stack."));
    return m_localVariablesSize - res->second - 1;
}

unsigned CompilerContext::baseToCurrentStackOffset(unsigned _baseOffset) const
{
    return _baseOffset + m_asm.deposit();
}

u256 CompilerContext::getStorageLocationOfVariable(const Declaration& _declaration) const
{
    auto it = m_stateVariables.find(&_declaration);
    if (it == m_stateVariables.end())
        BOOST_THROW_EXCEPTION(InternalCompilerError() << errinfo_comment("Variable not found in storage."));
    return it->second;
}



}
}