path: root/checkRandomVMTest.cpp

/*
    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 checkRandomTest.cpp
 * @author Christoph Jentzsch <jentzsch.simulationsoftware@gmail.com>
 * @date 2015
 * Check a random test and return 0/1 for success or failure. To be used for efficiency in the random test simulation.
 */

#include <libdevcore/Common.h>
#include <libdevcore/Assertions.h>
#include <libdevcore/Exceptions.h>
#include <libdevcore/Log.h>
#include <libevm/VMFactory.h>
#include "vm.h"
#pragma GCC diagnostic ignored "-Wunused-parameter"

using namespace std;
using namespace json_spirit;
using namespace dev::test;
using namespace dev;

bool doVMTest(mValue& _v);

int main(int argc, char *argv[])
{
    g_logVerbosity = 0;
    bool ret = false;

    try
    {
        mValue v;
        string s;
        for (int i = 1; i < argc; ++i)
            s += argv[i];
        if (asserts(s.length() > 0))
        {
            cout << "Content of argument is empty\n";
            return 1;
        }
        read_string(s, v);
        ret = doVMTest(v);
    }
    catch (Exception const& _e)
    {
        cout << "Failed test with Exception: " << diagnostic_information(_e) << endl;
        ret = false;
    }
    catch (std::exception const& _e)
    {
        cout << "Failed test with Exception: " << _e.what() << endl;
        ret = false;
    }
    return ret;
}

bool doVMTest(mValue& _v)
{
    eth::VMFactory::setKind(eth::VMKind::JIT);

    for (auto& i: _v.get_obj())
    {
        cnote << i.first;
        mObject& o = i.second.get_obj();

        assert(o.count("env") > 0);
        assert(o.count("pre") > 0);
        assert(o.count("exec") > 0);

        FakeExtVM fev;
        fev.importEnv(o["env"].get_obj());
        fev.importState(o["pre"].get_obj());

        fev.importExec(o["exec"].get_obj());
        if (fev.code.empty())
        {
            fev.thisTxCode = get<3>(fev.addresses.at(fev.myAddress));
            fev.code = fev.thisTxCode;
        }

        bytes output;
        u256 gas;
        bool vmExceptionOccured = false;
        try
        {
            auto vm = eth::VMFactory::create(fev.gas);
            output = vm->go(fev, fev.simpleTrace()).toBytes();
            gas = vm->gas();
        }
        catch (eth::VMException)
        {
            cnote << "Safe VM Exception";
            vmExceptionOccured = true;
        }
        catch (Exception const& _e)
        {
            cnote << "VM did throw an exception: " << diagnostic_information(_e);
            cnote << "Failed VM Test with Exception: " << _e.what();
            return 1;
        }
        catch (std::exception const& _e)
        {
            cnote << "VM did throw an exception: " << _e.what();
            cnote << "Failed VM Test with Exception: " << _e.what();
            return 1;
        }

        // delete null entries in storage for the sake of comparison
        for (auto  &a: fev.addresses)
        {
            vector<u256> keystoDelete;
            for (auto &s: get<2>(a.second))
            {
                if (s.second == 0)
                    keystoDelete.push_back(s.first);
            }
            for (auto const key: keystoDelete )
            {
                get<2>(a.second).erase(key);
            }
        }

        if (o.count("post") > 0)    // No exceptions expected
        {
            if (asserts(!vmExceptionOccured) || asserts(o.count("post") > 0) || asserts(o.count("callcreates") > 0) || asserts(o.count("out") > 0) || asserts(o.count("gas") > 0) || asserts(o.count("logs") > 0))
                return 1;

            dev::test::FakeExtVM test;
            test.importState(o["post"].get_obj());
            test.importCallCreates(o["callcreates"].get_array());
            test.sub.logs = importLog(o["logs"].get_array());

            //checkOutput(output, o);
            int j = 0;
            if (o["out"].type() == array_type)
                for (auto const& d: o["out"].get_array())
                {
                    if (asserts(output[j] == toInt(d)))
                    {
                        cout << "Output byte [" << j << "] different!";
                        return 1;
                    }
                    ++j;
                }
            else if (o["out"].get_str().find("0x") == 0)
            {
                if (asserts(output == fromHex(o["out"].get_str().substr(2))))
                    return 1;
            }
            else
            {
                if (asserts(output == fromHex(o["out"].get_str())))
                    return 1;
            }

            if (asserts(toInt(o["gas"]) == gas))
                return 1;

            auto& expectedAddrs = test.addresses;
            auto& resultAddrs = fev.addresses;
            for (auto&& expectedPair : expectedAddrs)
            {
                auto& expectedAddr = expectedPair.first;
                auto resultAddrIt = resultAddrs.find(expectedAddr);
                if (resultAddrIt == resultAddrs.end())
                {
                    cout << "Missing expected address " << expectedAddr;
                    return 1;
                }
                else
                {
                    auto& expectedState = expectedPair.second;
                    auto& resultState = resultAddrIt->second;
                    if (asserts(std::get<0>(expectedState) == std::get<0>(resultState)))
                    {
                        cout << expectedAddr << ": incorrect balance " << std::get<0>(resultState) << ", expected " << std::get<0>(expectedState);
                        return 1;
                    }
                    if (asserts(std::get<1>(expectedState) == std::get<1>(resultState)))
                    {
                        cout << expectedAddr << ": incorrect txCount " << std::get<1>(resultState) << ", expected " << std::get<1>(expectedState);
                        return 1;
                    }
                    if (asserts(std::get<3>(expectedState) == std::get<3>(resultState)))
                    {
                        cout << expectedAddr << ": incorrect code";
                        return 1;
                    }

                    //checkStorage(std::get<2>(expectedState), std::get<2>(resultState), expectedAddr);
                    for (auto&& expectedStorePair : std::get<2>(expectedState))
                    {
                        auto& expectedStoreKey = expectedStorePair.first;
                        auto resultStoreIt = std::get<2>(resultState).find(expectedStoreKey);
                        if (resultStoreIt == std::get<2>(resultState).end())
                        {
                            cout << expectedAddr << ": missing store key " << expectedStoreKey << endl;
                            return 1;
                        }
                        else
                        {
                            auto& expectedStoreValue = expectedStorePair.second;
                            auto& resultStoreValue = resultStoreIt->second;
                            if (asserts(expectedStoreValue == resultStoreValue))
                            {
                                cout << expectedAddr << ": store[" << expectedStoreKey << "] = " << resultStoreValue << ", expected " << expectedStoreValue << endl;
                                return 1;
                            }
                        }
                    }
                    if (assertsEqual(std::get<2>(resultState).size(), std::get<2>(expectedState).size()))
                        return 1;
                    for (auto&& resultStorePair: std::get<2>(resultState))
                    {
                        if (!std::get<2>(expectedState).count(resultStorePair.first))
                        {
                            cout << expectedAddr << ": unexpected store key " << resultStorePair.first << endl;
                            return 1;
                        }
                    }
                }
            }

            //checkAddresses<std::map<Address, std::tuple<u256, u256, std::map<u256, u256>, bytes> > >(test.addresses, fev.addresses);
            for (auto& resultPair : fev.addresses)
            {
                auto& resultAddr = resultPair.first;
                auto expectedAddrIt = test.addresses.find(resultAddr);
                if (expectedAddrIt == test.addresses.end())
                {
                    cout << "Missing result address " << resultAddr << endl;
                    return 1;
                }
            }
            if (asserts(test.addresses == fev.addresses))
                return 1;

            if (asserts(test.callcreates == fev.callcreates))
                return 1;

            //checkCallCreates(fev.callcreates, test.callcreates);
            {
                if (assertsEqual(test.callcreates.size(), fev.callcreates.size()))
                    return 1;

                for (size_t i = 0; i < test.callcreates.size(); ++i)
                {
                    if (asserts(test.callcreates[i].data() == fev.callcreates[i].data()))
                        return 1;
                    if (asserts(test.callcreates[i].receiveAddress() == fev.callcreates[i].receiveAddress()))
                        return 1;
                    if (asserts(test.callcreates[i].gas() == fev.callcreates[i].gas()))
                        return 1;
                    if (asserts(test.callcreates[i].value() == fev.callcreates[i].value()))
                        return 1;
                }
            }

            //checkLog(fev.sub.logs, test.sub.logs);
            {
                if (assertsEqual(fev.sub.logs.size(), test.sub.logs.size()))
                    return 1;

                for (size_t i = 0; i < fev.sub.logs.size(); ++i)
                {
                    if (assertsEqual(fev.sub.logs[i].address, test.sub.logs[i].address))
                        return 1;
                    if (assertsEqual(fev.sub.logs[i].topics, test.sub.logs[i].topics))
                        return 1;
                    if (asserts(fev.sub.logs[i].data == test.sub.logs[i].data))
                        return 1;
                }
            }

        }
        else    // Exception expected
        {
            if (asserts(vmExceptionOccured))
                return 1;
        }
    }
    // test passed
    return 0;
}