aboutsummaryrefslogtreecommitdiffstats
path: root/Assembly.h
blob: ca3beaf5bd21a8f2f4ada75888649b17619df935 (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
/*
    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/>.
*/
/** @file CodeFragment.h
 * @author Gav Wood <i@gavwood.com>
 * @date 2014
 */

#pragma once

#include <iostream>
#include <libethsupport/Common.h>
#include <libethcore/Instruction.h>
#include "Exceptions.h"

namespace eth
{

enum AssemblyItemType { Operation, Push, PushString, PushTag, Tag, PushData };

class Assembly;

class AssemblyItem
{
    friend class Assembly;

public:
    AssemblyItem(u256 _push): m_type(Push), m_data(_push) {}
    AssemblyItem(AssemblyItemType _type, AssemblyItem const& _tag): m_type(_type), m_data(_tag.m_data) { assert(_type == PushTag); assert(_tag.m_type == Tag); }
    AssemblyItem(Instruction _i): m_type(Operation), m_data((byte)_i) {}
    AssemblyItem(AssemblyItemType _type, u256 _data): m_type(_type), m_data(_data) {}

    AssemblyItemType type() const { return m_type; }
    u256 data() const { return m_data; }

    std::ostream& streamOut(std::ostream& _out) const;

private:
    AssemblyItemType m_type;
    u256 m_data;
};

class Assembly
{
public:
    AssemblyItem newTag() { return AssemblyItem(Tag, m_usedTags++); }
    AssemblyItem newData(bytes const& _data) { auto h = sha3(_data); m_data[h] = _data; return AssemblyItem(PushData, h); }
    AssemblyItem newPushString(std::string const& _data) { auto b = asBytes(_data); auto h = sha3(b); m_data[h] = b; return AssemblyItem(PushString, h); }

    void append(AssemblyItem const& _i) { m_items.push_back(_i); }

    bytes assemble() const;
    void append(Assembly const& _a);
    std::ostream& streamOut(std::ostream& _out) const;

private:
    u256 m_usedTags = 0;
    std::vector<AssemblyItem> m_items;
    std::map<h256, bytes> m_data;
};

}