index : gsoc2013-epiphany	archive-integration archive-integration-rebase master
	Google Summer of Code 2013 - GNOME - Archive Integration workspace

aboutsummaryrefslogtreecommitdiffstats

log msg author committer range

path: root/.cvsignore

	Commit message (Expand)	Author	Age	Files	Lines
*	Check for broken context event button.	Christian Persch	2005-01-23	1	-0/+2
*	More stamp magic.	Christian Persch	2005-01-09	1	-1/+1
*	Shut up	Christian Persch	2004-12-02	1	-0/+6
*	Shut up	Christian Persch	2004-08-25	1	-0/+1
*	*** empty log message ***	Marco Pesenti Gritti	2004-06-15	1	-0/+1
*	*** empty log message ***Release117	Marco Pesenti Gritti	2004-02-02	1	-0/+2
*	Initial revision	Marco Pesenti Gritti	2002-12-31	1	-0/+14

generated by cgit (git 2.34.1) at 2024-11-15 04:17:05 +0800

/*
    This file is part of solidity.

    solidity 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.

    solidity 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 solidity.  If not, see <http://www.gnu.org/licenses/>.
*/
/** @file Assembly.cpp
 * @author Gav Wood <i@gavwood.com>
 * @date 2014
 */

#include "Assembly.h"

#include <libevmasm/CommonSubexpressionEliminator.h>
#include <libevmasm/ControlFlowGraph.h>
#include <libevmasm/PeepholeOptimiser.h>
#include <libevmasm/BlockDeduplicator.h>
#include <libevmasm/ConstantOptimiser.h>
#include <libevmasm/GasMeter.h>

#include <fstream>
#include <json/json.h>

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

void Assembly::append(Assembly const& _a)
{
    auto newDeposit = m_deposit + _a.deposit();
    for (AssemblyItem i: _a.m_items)
    {
        if (i.type() == Tag || i.type() == PushTag)
            i.setData(i.data() + m_usedTags);
        else if (i.type() == PushSub || i.type() == PushSubSize)
            i.setData(i.data() + m_subs.size());
        append(i);
    }
    m_deposit = newDeposit;
    m_usedTags += _a.m_usedTags;
    for (auto const& i: _a.m_data)
        m_data.insert(i);
    for (auto const& i: _a.m_strings)
        m_strings.insert(i);
    m_subs += _a.m_subs;
    for (auto const& lib: _a.m_libraries)
        m_libraries.insert(lib);
}

void Assembly::append(Assembly const& _a, int _deposit)
{
    assertThrow(_deposit <= _a.m_deposit, InvalidDeposit, "");

    append(_a);
    while (_deposit++ < _a.m_deposit)
        append(Instruction::POP);
}

unsigned Assembly::bytesRequired(unsigned subTagSize) const
{
    for (unsigned tagSize = subTagSize; true; ++tagSize)
    {
        unsigned ret = 1;
        for (auto const& i: m_data)
            ret += i.second.size();

        for (AssemblyItem const& i: m_items)
            ret += i.bytesRequired(tagSize);
        if (dev::bytesRequired(ret) <= tagSize)
            return ret;
    }
}

namespace
{

string locationFromSources(StringMap const& _sourceCodes, SourceLocation const& _location)
{
    if (_location.isEmpty() || _sourceCodes.empty() || _location.start >= _location.end || _location.start < 0)
        return "";

    auto it = _sourceCodes.find(*_location.sourceName);
    if (it == _sourceCodes.end())
        return "";

    string const& source = it->second;
    if (size_t(_location.start) >= source.size())
        return "";

    string cut = source.substr(_location.start, _location.end - _location.start);
    auto newLinePos = cut.find_first_of("\n");
    if (newLinePos != string::npos)
        cut = cut.substr(0, newLinePos) + "...";

    return cut;
}

class Functionalizer
{
public:
    Functionalizer (ostream& _out, string const& _prefix, StringMap const& _sourceCodes):
        m_out(_out), m_prefix(_prefix), m_sourceCodes(_sourceCodes)
    {}

    void feed(AssemblyItem const& _item)
    {
        if (!_item.location().isEmpty() && _item.location() != m_location)
        {
            flush();
            m_location = _item.location();
            printLocation();
        }
        if (!(
            _item.canBeFunctional() &&
            _item.returnValues() <= 1 &&
            _item.arguments() <= int(m_pending.size())
        ))
        {
            flush();
            m_out << m_prefix << (_item.type() == Tag ? "" : "  ") << _item.toAssemblyText() << endl;
            return;
        }
        string expression = _item.toAssemblyText();
        if (_item.arguments() > 0)
        {
            expression += "(";
            for (int i = 0; i < _item.arguments(); ++i)
            {
                expression += m_pending.back();
                m_pending.pop_back();
                if (i + 1 < _item.arguments())
                    expression += ", ";
            }
            expression += ")";
        }

        m_pending.push_back(expression);
        if (_item.returnValues() != 1)
            flush();
    }

    void flush()
    {
        for (string const& expression: m_pending)
            m_out << m_prefix << "  " << expression << endl;
        m_pending.clear();
    }

    void printLocation()
    {
        if (!m_location.sourceName && m_location.isEmpty())
            return;
        m_out << m_prefix << "    /*";
        if (m_location.sourceName)
            m_out << " \"" + *m_location.sourceName + "\"";
        if (!m_location.isEmpty())
            m_out << ":" << to_string(m_location.start) + ":" + to_string(m_location.end);
        m_out << "  " << locationFromSources(m_sourceCodes, m_location);
        m_out << " */" << endl;
    }

private:
    strings m_pending;
    SourceLocation m_location;

    ostream& m_out;
    string const& m_prefix;
    StringMap const& m_sourceCodes;
};

}

ostream& Assembly::streamAsm(ostream& _out, string const& _prefix, StringMap const& _sourceCodes) const
{
    Functionalizer f(_out, _prefix, _sourceCodes);

    for (auto const& i: m_items)
        f.feed(i);
    f.flush();

    if (!m_data.empty() || !m_subs.empty())
    {
        _out << _prefix << "stop" << endl;
        for (auto const& i: m_data)
            if (u256(i.first) >= m_subs.size())
                _out << _prefix << "data_" << toHex(u256(i.first)) << " " << toHex(i.second) << endl;

        for (size_t i = 0; i < m_subs.size(); ++i)
        {
            _out << endl << _prefix << "sub_" << i << ": assembly {\n";
            m_subs[i]->streamAsm(_out, _prefix + "    ", _sourceCodes);
            _out << _prefix << "}" << endl;
        }
    }

    if (m_auxiliaryData.size() > 0)
        _out << endl << _prefix << "auxdata: 0x" << toHex(m_auxiliaryData) << endl;

    return _out;
}

Json::Value Assembly::createJsonValue(string _name, int _begin, int _end, string _value, string _jumpType) const
{
    Json::Value value;
    value["name"] = _name;
    value["begin"] = _begin;
    value["end"] = _end;
    if (!_value.empty())
        value["value"] = _value;
    if (!_jumpType.empty())
        value["jumpType"] = _jumpType;
    return value;
}

string toStringInHex(u256 _value)
{
    std::stringstream hexStr;
    hexStr << hex << _value;
    return hexStr.str();
}

Json::Value Assembly::streamAsmJson(ostream& _out, StringMap const& _sourceCodes) const
{
    Json::Value root;

    Json::Value collection(Json::arrayValue);
    for (AssemblyItem const& i: m_items)
    {
        switch (i.type())
        {
        case Operation:
            collection.append(
                createJsonValue(instructionInfo(i.instruction()).name, i.location().start, i.location().end, i.getJumpTypeAsString()));
            break;
        case Push:
            collection.append(
                createJsonValue("PUSH", i.location().start, i.location().end, toStringInHex(i.data()), i.getJumpTypeAsString()));
            break;
        case PushString:
            collection.append(
                createJsonValue("PUSH tag", i.location().start, i.location().end, m_strings.at((h256)i.data())));
            break;
        case PushTag:
            if (i.data() == 0)
                collection.append(
                    createJsonValue("PUSH [ErrorTag]", i.location().start, i.location().end, ""));
            else
                collection.append(
                    createJsonValue("PUSH [tag]", i.location().start, i.location().end, string(i.data())));
            break;
        case PushSub:
            collection.append(
                createJsonValue("PUSH [$]", i.location().start, i.location().end, dev::toString(h256(i.data()))));
            break;
        case PushSubSize:
            collection.append(
                createJsonValue("PUSH #[$]", i.location().start, i.location().end, dev::toString(h256(i.data()))));
            break;
        case PushProgramSize:
            collection.append(
                createJsonValue("PUSHSIZE", i.location().start, i.location().end));
            break;
        case PushLibraryAddress:
            collection.append(
                createJsonValue("PUSHLIB", i.location().start, i.location().end, m_libraries.at(h256(i.data())))
            );
            break;
        case Tag:
            collection.append(
                createJsonValue("tag", i.location().start, i.location().end, string(i.data())));
            collection.append(
                createJsonValue("JUMPDEST", i.location().start, i.location().end));
            break;
        case PushData:
            collection.append(createJsonValue("PUSH data", i.location().start, i.location().end, toStringInHex(i.data())));
            break;
        default:
            BOOST_THROW_EXCEPTION(InvalidOpcode());
        }
    }

    root[".code"] = collection;

    if (!m_data.empty() || !m_subs.empty())
    {
        Json::Value data;
        for (auto const& i: m_data)
            if (u256(i.first) >= m_subs.size())
                data[toStringInHex((u256)i.first)] = toHex(i.second);

        for (size_t i = 0; i < m_subs.size(); ++i)
        {
            std::stringstream hexStr;
            hexStr << hex << i;
            data[hexStr.str()] = m_subs[i]->stream(_out, "", _sourceCodes, true);
        }
        root[".data"] = data;
    }

    if (m_auxiliaryData.size() > 0)
        root[".auxdata"] = toHex(m_auxiliaryData);

    _out << root;

    return root;
}

Json::Value Assembly::stream(ostream& _out, string const& _prefix, StringMap const& _sourceCodes, bool _inJsonFormat) const
{
    if (_inJsonFormat)
        return streamAsmJson(_out, _sourceCodes);
    else
    {
        streamAsm(_out, _prefix, _sourceCodes);
        return Json::Value();
    }
}

AssemblyItem const& Assembly::append(AssemblyItem const& _i)
{
    assertThrow(m_deposit >= 0, AssemblyException, "");
    m_deposit += _i.deposit();
    m_items.push_back(_i);
    if (m_items.back().location().isEmpty() && !m_currentSourceLocation.isEmpty())
        m_items.back().setLocation(m_currentSourceLocation);
    return back();
}

AssemblyItem Assembly::newPushLibraryAddress(string const& _identifier)
{
    h256 h(dev::keccak256(_identifier));
    m_libraries[h] = _identifier;
    return AssemblyItem(PushLibraryAddress, h);
}

void Assembly::injectStart(AssemblyItem const& _i)
{
    m_items.insert(m_items.begin(), _i);
}

Assembly& Assembly::optimise(bool _enable, bool _isCreation, size_t _runs)
{
    OptimiserSettings settings;
    settings.isCreation = _isCreation;
    settings.runPeephole = true;
    if (_enable)
    {
        settings.runDeduplicate = true;
        settings.runCSE = true;
        settings.runConstantOptimiser = true;
    }
    settings.expectedExecutionsPerDeployment = _runs;
    optimiseInternal(settings);
    return *this;
}


Assembly& Assembly::optimise(OptimiserSettings _settings)
{
    optimiseInternal(_settings);
    return *this;
}

map<u256, u256> Assembly::optimiseInternal(OptimiserSettings _settings)
{
    // Run optimisation for sub-assemblies.
    for (size_t subId = 0; subId < m_subs.size(); ++subId)
    {
        OptimiserSettings settings = _settings;
        // Disable creation mode for sub-assemblies.
        settings.isCreation = false;
        map<u256, u256> subTagReplacements = m_subs[subId]->optimiseInternal(settings);
        // Apply the replacements (can be empty).
        BlockDeduplicator::applyTagReplacement(m_items, subTagReplacements, subId);
    }

    map<u256, u256> tagReplacements;
    // Iterate until no new optimisation possibilities are found.
    for (unsigned count = 1; count > 0;)
    {
        count = 0;

        if (_settings.runPeephole)
        {
            PeepholeOptimiser peepOpt(m_items);
            while (peepOpt.optimise())
                count++;
        }

        // This only modifies PushTags, we have to run again to actually remove code.
        if (_settings.runDeduplicate)
        {
            BlockDeduplicator dedup(m_items);
            if (dedup.deduplicate())
            {
                tagReplacements.insert(dedup.replacedTags().begin(), dedup.replacedTags().end());
                count++;
            }
        }

        if (_settings.runCSE)
        {
            // Control flow graph optimization has been here before but is disabled because it
            // assumes we only jump to tags that are pushed. This is not the case anymore with
            // function types that can be stored in storage.
            AssemblyItems optimisedItems;

            auto iter = m_items.begin();
            while (iter != m_items.end())
            {
                KnownState emptyState;
                CommonSubexpressionEliminator eliminator(emptyState);
                auto orig = iter;
                iter = eliminator.feedItems(iter, m_items.end());
                bool shouldReplace = false;
                AssemblyItems optimisedChunk;
                try
                {
                    optimisedChunk = eliminator.getOptimizedItems();
                    shouldReplace = (optimisedChunk.size() < size_t(iter - orig));
                }
                catch (StackTooDeepException const&)