aboutsummaryrefslogtreecommitdiffstats
path: root/libsolidity/Compiler.h
blob: 7b7cffcfb6452a1d17a89ad66d243e0f09c2cf12 (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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
/*
    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
 * Solidity AST to EVM bytecode compiler.
 */

#pragma once

#include <ostream>
#include <functional>
#include <libsolidity/ASTVisitor.h>
#include <libsolidity/CompilerContext.h>
#include <libevmasm/Assembly.h>

namespace dev {
namespace solidity {

class Compiler: private ASTConstVisitor
{
public:
    explicit Compiler(bool _optimize = false, unsigned _runs = 200):
        m_optimize(_optimize),
        m_optimizeRuns(_runs),
        m_returnTag(m_context.newTag())
    { }

    void compileContract(
        ContractDefinition const& _contract,
        std::map<ContractDefinition const*, eth::Assembly const*> const& _contracts
    );
    /// Compiles a contract that uses CALLCODE to call into a pre-deployed version of the given
    /// contract at runtime, but contains the full creation-time code.
    void compileClone(
        ContractDefinition const& _contract,
        std::map<ContractDefinition const*, eth::Assembly const*> const& _contracts
    );
    eth::Assembly const& assembly() { return m_context.assembly(); }
    eth::LinkerObject assembledObject() { return m_context.assembledObject(); }
    eth::LinkerObject runtimeObject() { return m_context.assembledRuntimeObject(m_runtimeSub); }
    /// @arg _sourceCodes is the map of input files to source code strings
    /// @arg _inJsonFromat shows whether the out should be in Json format
    Json::Value streamAssembly(std::ostream& _stream, StringMap const& _sourceCodes = StringMap(), bool _inJsonFormat = false) const
    {
        return m_context.streamAssembly(_stream, _sourceCodes, _inJsonFormat);
    }
    /// @returns Assembly items of the normal compiler context
    eth::AssemblyItems const& assemblyItems() const { return m_context.assembly().items(); }
    /// @returns Assembly items of the runtime compiler context
    eth::AssemblyItems const& runtimeAssemblyItems() const { return m_context.assembly().sub(m_runtimeSub).items(); }

    /// @returns the entry label of the given function. Might return an AssemblyItem of type
    /// UndefinedItem if it does not exist yet.
    eth::AssemblyItem functionEntryLabel(FunctionDefinition const& _function) const;

private:
    /// Registers the non-function objects inside the contract with the context and stores the basic
    /// information about the contract like the AST annotations.
    void initializeContext(
        ContractDefinition const& _contract,
        std::map<ContractDefinition const*, eth::Assembly const*> const& _compiledContracts
    );
    /// Adds the code that is run at creation time. Should be run after exchanging the run-time context
    /// with a new and initialized context. Adds the constructor code.
    void packIntoContractCreator(ContractDefinition const& _contract, CompilerContext const& _runtimeContext);
    /// Appends state variable initialisation and constructor code.
    void appendInitAndConstructorCode(ContractDefinition const& _contract);
    void appendBaseConstructor(FunctionDefinition const& _constructor);
    void appendConstructor(FunctionDefinition const& _constructor);
    void appendFunctionSelector(ContractDefinition const& _contract);
    /// Creates code that unpacks the arguments for the given function represented by a vector of TypePointers.
    /// From memory if @a _fromMemory is true, otherwise from call data.
    /// Expects source offset on the stack.
    void appendCalldataUnpacker(
        TypePointers const& _typeParameters,
        bool _fromMemory = false,
        u256 _startOffset = u256(-1)
    );
    void appendReturnValuePacker(TypePointers const& _typeParameters);

    void registerStateVariables(ContractDefinition const& _contract);
    void initializeStateVariables(ContractDefinition const& _contract);

    /// Initialises all memory arrays in the local variables to point to an empty location.
    void initialiseMemoryArrays(std::vector<VariableDeclaration const*> _variables);
    /// Pushes the initialised value of the given type to the stack. If the type is a memory
    /// reference type, allocates memory and pushes the memory pointer.
    /// Not to be used for storage references.
    void initialiseInMemory(Type const& _type);

    virtual bool visit(VariableDeclaration const& _variableDeclaration) override;
    virtual bool visit(FunctionDefinition const& _function) override;
    virtual bool visit(IfStatement const& _ifStatement) override;
    virtual bool visit(WhileStatement const& _whileStatement) override;
    virtual bool visit(ForStatement const& _forStatement) override;
    virtual bool visit(Continue const& _continue) override;
    virtual bool visit(Break const& _break) override;
    virtual bool visit(Return const& _return) override;
    virtual bool visit(Throw const& _throw) override;
    virtual bool visit(VariableDeclarationStatement const& _variableDeclarationStatement) override;
    virtual bool visit(ExpressionStatement const& _expressionStatement) override;
    virtual bool visit(PlaceholderStatement const&) override;

    /// Repeatedly visits all function which are referenced but which are not compiled yet.
    void appendFunctionsWithoutCode();

    /// Appends one layer of function modifier code of the current function, or the function
    /// body itself if the last modifier was reached.
    void appendModifierOrFunctionCode();

    void appendStackVariableInitialisation(VariableDeclaration const& _variable);
    void compileExpression(Expression const& _expression, TypePointer const& _targetType = TypePointer());

    /// @returns the runtime assembly for clone contracts.
    static eth::Assembly cloneRuntime();

    bool const m_optimize;
    unsigned const m_optimizeRuns;
    CompilerContext m_context;
    size_t m_runtimeSub = size_t(-1); ///< Identifier of the runtime sub-assembly
    CompilerContext m_runtimeContext;
    std::vector<eth::AssemblyItem> m_breakTags; ///< tag to jump to for a "break" statement
    std::vector<eth::AssemblyItem> m_continueTags; ///< tag to jump to for a "continue" statement
    eth::AssemblyItem m_returnTag; ///< tag to jump to for a "return" statement
    unsigned m_modifierDepth = 0;
    FunctionDefinition const* m_currentFunction = nullptr;
    unsigned m_stackCleanupForReturn = 0; ///< this number of stack elements need to be removed before jump to m_returnTag
    // arguments for base constructors, filled in derived-to-base order
    std::map<FunctionDefinition const*, std::vector<ASTPointer<Expression>> const*> m_baseArguments;
};

}
}