aboutsummaryrefslogtreecommitdiffstats
path: root/BlockDeduplicator.cpp
blob: eadbe1b402d0bb3240d57a8831e8dbced2caabc6 (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
/*
    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 BlockDeduplicator.cpp
 * @author Christian <c@ethdev.com>
 * @date 2015
 * Unifies basic blocks that share content.
 */

#include <libevmasm/BlockDeduplicator.h>
#include <functional>
#include <libevmasm/AssemblyItem.h>
#include <libevmasm/SemanticInformation.h>

using namespace std;
using namespace dev;
using namespace dev::eth;


bool BlockDeduplicator::deduplicate()
{
    // Compares indices based on the suffix that starts there, ignoring tags and stopping at
    // opcodes that stop the control flow.
    function<bool(size_t, size_t)> comparator = [&](size_t _i, size_t _j)
    {
        if (_i == _j)
            return false;

        BlockIterator first(m_items.begin() + _i, m_items.end());
        BlockIterator second(m_items.begin() + _j, m_items.end());
        BlockIterator end(m_items.end(), m_items.end());

        if (first != end && (*first).type() == Tag)
            ++first;
        if (second != end && (*second).type() == Tag)
            ++second;

        return std::lexicographical_compare(first, end, second, end);
    };

    set<size_t, function<bool(size_t, size_t)>> blocksSeen(comparator);
    map<u256, u256> tagReplacement;
    for (size_t i = 0; i < m_items.size(); ++i)
    {
        if (m_items.at(i).type() != Tag)
            continue;
        auto it = blocksSeen.find(i);
        if (it == blocksSeen.end())
            blocksSeen.insert(i);
        else
            tagReplacement[m_items.at(i).data()] = m_items.at(*it).data();
    }

    bool ret = false;
    for (AssemblyItem& item: m_items)
        if (item.type() == PushTag && tagReplacement.count(item.data()))
        {
            ret = true;
            item.setData(tagReplacement.at(item.data()));
        }
    return ret;
}

BlockDeduplicator::BlockIterator& BlockDeduplicator::BlockIterator::operator++()
{
    if (it == end)
        return *this;
    if (SemanticInformation::altersControlFlow(*it) && *it != AssemblyItem(eth::Instruction::JUMPI))
        it = end;
    else
    {
        ++it;
        while (it != end && it->type() == Tag)
            ++it;
    }
    return *this;
}