solidity/vm.cpp

444 lines
11 KiB
C++
Raw Normal View History

/*
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 state.cpp
* @author Gav Wood <i@gavwood.com>
* @date 2014
* State test functions.
*/
2014-02-21 19:18:30 +00:00
#include <fstream>
2014-04-24 10:08:01 +00:00
#include <cstdint>
2014-04-23 14:08:11 +00:00
#include <libethcore/Log.h>
#include <libethereum/ExtVMFace.h>
#include <libethereum/Transaction.h>
#include <libethereum/VM.h>
#include <libethereum/Instruction.h>
#include "JsonSpiritHeaders.h"
#include <boost/test/unit_test.hpp>
using namespace std;
2014-02-21 19:18:30 +00:00
using namespace json_spirit;
using namespace eth;
namespace eth { namespace test {
class FakeExtVM: public ExtVMFace
{
public:
2014-02-21 19:18:30 +00:00
FakeExtVM()
{}
FakeExtVM(BlockInfo const& _previousBlock, BlockInfo const& _currentBlock):
ExtVMFace(Address(), Address(), Address(), 0, 1, bytesConstRef(), bytesConstRef(), _previousBlock, _currentBlock)
2014-02-21 19:18:30 +00:00
{}
u256 store(u256 _n)
{
return get<3>(addresses[myAddress])[_n];
}
void setStore(u256 _n, u256 _v)
{
get<3>(addresses[myAddress])[_n] = _v;
}
u256 balance(Address _a) { return get<0>(addresses[_a]); }
void subBalance(u256 _a) { get<0>(addresses[myAddress]) -= _a; }
u256 txCount(Address _a) { return get<1>(addresses[_a]); }
void suicide(Address _a)
{
get<0>(addresses[_a]) += get<0>(addresses[myAddress]);
addresses.erase(myAddress);
}
2014-03-20 01:29:40 +00:00
void transact(Transaction& _t)
2014-02-21 19:18:30 +00:00
{
if (get<0>(addresses[myAddress]) >= _t.value)
{
get<0>(addresses[myAddress]) -= _t.value;
get<1>(addresses[myAddress])++;
// get<0>(addresses[_t.receiveAddress]) += _t.value;
txs.push_back(_t);
}
}
h160 create(u256 _endowment, u256* _gas, bytesConstRef _init)
{
Transaction t;
t.value = _endowment;
t.gasPrice = gasPrice;
2014-03-31 01:49:36 +00:00
t.gas = *_gas;
t.data = _init.toBytes();
txs.push_back(t);
return right160(t.sha3(false));
}
2014-04-04 19:24:38 +00:00
bool call(Address _receiveAddress, u256 _value, bytesConstRef _data, u256* _gas, bytesRef _out)
2014-02-21 19:18:30 +00:00
{
Transaction t;
2014-04-04 19:24:38 +00:00
t.value = _value;
t.gasPrice = gasPrice;
2014-03-31 01:49:36 +00:00
t.gas = *_gas;
2014-04-04 19:24:38 +00:00
t.data = _data.toVector();
t.receiveAddress = _receiveAddress;
txs.push_back(t);
2014-04-04 19:24:38 +00:00
(void)_out;
return true;
2014-02-21 19:18:30 +00:00
}
2014-04-04 19:24:38 +00:00
void setTransaction(Address _caller, u256 _value, u256 _gasPrice, bytes const& _data)
2014-02-21 19:18:30 +00:00
{
2014-04-04 19:24:38 +00:00
caller = origin = _caller;
value = _value;
data = &_data;
gasPrice = _gasPrice;
2014-02-21 19:18:30 +00:00
}
2014-03-31 01:49:36 +00:00
void setContract(Address _myAddress, u256 _myBalance, u256 _myNonce, bytes const& _code, map<u256, u256> const& _storage)
2014-02-21 19:18:30 +00:00
{
myAddress = _myAddress;
2014-03-31 01:49:36 +00:00
set(myAddress, _myBalance, _myNonce, _code, _storage);
2014-02-21 19:18:30 +00:00
}
2014-03-31 01:49:36 +00:00
void set(Address _a, u256 _myBalance, u256 _myNonce, bytes const& _code, map<u256, u256> const& _storage)
2014-02-21 19:18:30 +00:00
{
get<0>(addresses[_a]) = _myBalance;
get<1>(addresses[_a]) = _myNonce;
get<2>(addresses[_a]) = 0;
2014-03-31 01:49:36 +00:00
get<3>(addresses[_a]) = _storage;
get<4>(addresses[_a]) = _code;
}
void reset(u256 _myBalance, u256 _myNonce, map<u256, u256> const& _storage)
{
txs.clear();
addresses.clear();
set(myAddress, _myBalance, _myNonce, get<4>(addresses[myAddress]), _storage);
2014-02-21 19:18:30 +00:00
}
mObject exportEnv()
{
mObject ret;
ret["previousHash"] = toString(previousBlock.hash);
ret["previousNonce"] = toString(previousBlock.nonce);
push(ret, "currentDifficulty", currentBlock.difficulty);
push(ret, "currentTimestamp", currentBlock.timestamp);
ret["currentCoinbase"] = toString(currentBlock.coinbaseAddress);
return ret;
}
void importEnv(mObject& _o)
{
BOOST_REQUIRE(_o.count("previousHash") > 0 );
BOOST_REQUIRE(_o.count("previousNonce") > 0 );
BOOST_REQUIRE(_o.count("currentDifficulty") > 0 );
BOOST_REQUIRE(_o.count("currentTimestamp") > 0 );
BOOST_REQUIRE(_o.count("currentCoinbase") > 0 );
2014-02-21 19:18:30 +00:00
previousBlock.hash = h256(_o["previousHash"].get_str());
previousBlock.nonce = h256(_o["previousNonce"].get_str());
currentBlock.difficulty = toInt(_o["currentDifficulty"]);
currentBlock.timestamp = toInt(_o["currentTimestamp"]);
currentBlock.coinbaseAddress = Address(_o["currentCoinbase"].get_str());
}
static u256 toInt(mValue const& _v)
{
switch (_v.type())
{
case str_type: return u256(_v.get_str());
case int_type: return (u256)_v.get_uint64();
case bool_type: return (u256)(uint64_t)_v.get_bool();
case real_type: return (u256)(uint64_t)_v.get_real();
default: cwarn << "Bad type for scalar: " << _v.type();
}
return 0;
}
static byte toByte(mValue const& _v)
{
switch (_v.type())
{
case str_type: return (byte)stoi(_v.get_str());
case int_type: return (byte)_v.get_uint64();
case bool_type: return (byte)_v.get_bool();
case real_type: return (byte)_v.get_real();
default: cwarn << "Bad type for scalar: " << _v.type();
}
return 0;
}
2014-02-21 19:18:30 +00:00
static void push(mObject& o, string const& _n, u256 _v)
{
if (_v < (u256)1 << 64)
o[_n] = (uint64_t)_v;
else
o[_n] = toString(_v);
}
static void push(mArray& a, u256 _v)
{
if (_v < (u256)1 << 64)
a.push_back((uint64_t)_v);
else
a.push_back(toString(_v));
}
mObject exportState()
{
mObject ret;
for (auto const& a: addresses)
{
mObject o;
push(o, "balance", get<0>(a.second));
push(o, "nonce", get<1>(a.second));
mObject store;
string curKey;
u256 li = 0;
mArray curVal;
for (auto const& s: get<3>(a.second))
{
if (!li || s.first > li + 8)
{
if (li)
store[curKey] = curVal;
li = s.first;
curKey = toString(li);
curVal = mArray();
}
else
for (; li != s.first; ++li)
curVal.push_back(0);
push(curVal, s.second);
++li;
}
if (li)
{
store[curKey] = curVal;
o["store"] = store;
}
ret[toString(a.first)] = o;
}
return ret;
}
void importState(mObject& _o)
{
for (auto const& i: _o)
{
mObject o = i.second.get_obj();
BOOST_REQUIRE(o.count("balance") > 0 );
BOOST_REQUIRE(o.count("nonce") > 0 );
BOOST_REQUIRE(o.count("store") > 0 );
2014-02-21 19:18:30 +00:00
auto& a = addresses[Address(i.first)];
get<0>(a) = toInt(o["balance"]);
get<1>(a) = toInt(o["nonce"]);
if (o.count("store"))
for (auto const& j: o["store"].get_obj())
{
u256 adr(j.first);
for (auto const& k: j.second.get_array())
get<3>(a)[adr++] = toInt(k);
}
if (o.count("code"))
{
2014-03-31 01:49:36 +00:00
bytes e;
bytes d = compileLisp(o["code"].get_str(), false, e);
get<4>(a) = d;
2014-02-21 19:18:30 +00:00
}
}
}
mObject exportExec()
{
2014-02-21 19:18:30 +00:00
mObject ret;
ret["address"] = toString(myAddress);
2014-04-04 19:24:38 +00:00
ret["caller"] = toString(caller);
ret["origin"] = toString(origin);
push(ret, "value", value);
push(ret, "gasPrice", gasPrice);
2014-02-21 19:18:30 +00:00
mArray d;
2014-04-04 19:24:38 +00:00
for (auto const& i: data)
2014-02-21 19:18:30 +00:00
push(d, i);
ret["data"] = d;
return ret;
}
void importExec(mObject& _o)
{
BOOST_REQUIRE(_o.count("address")> 0);
BOOST_REQUIRE(_o.count("caller") > 0);
BOOST_REQUIRE(_o.count("origin") > 0);
BOOST_REQUIRE(_o.count("value") > 0);
BOOST_REQUIRE(_o.count("gasPrice") > 0);
BOOST_REQUIRE(_o.count("data") > 0 );
2014-02-21 19:18:30 +00:00
myAddress = Address(_o["address"].get_str());
2014-04-04 19:24:38 +00:00
caller = Address(_o["caller"].get_str());
origin = Address(_o["origin"].get_str());
value = toInt(_o["value"]);
gasPrice = toInt(_o["gasPrice"]);
thisTxData.clear();
2014-02-21 19:18:30 +00:00
for (auto const& j: _o["data"].get_array())
thisTxData.push_back(toByte(j));
2014-04-04 19:24:38 +00:00
data = &thisTxData;
2014-02-21 19:18:30 +00:00
}
mArray exportTxs()
{
mArray ret;
for (Transaction const& tx: txs)
{
mObject o;
o["destination"] = toString(tx.receiveAddress);
push(o, "value", tx.value);
mArray d;
for (auto const& i: tx.data)
push(d, i);
o["data"] = d;
ret.push_back(o);
}
return ret;
}
void importTxs(mArray& _txs)
{
for (mValue& v: _txs)
{
auto tx = v.get_obj();
BOOST_REQUIRE(tx.count("destination") > 0);
BOOST_REQUIRE(tx.count("value") > 0 );
BOOST_REQUIRE(tx.count("data") > 0 );
2014-02-21 19:18:30 +00:00
Transaction t;
t.receiveAddress = Address(tx["destination"].get_str());
t.value = toInt(tx["value"]);
for (auto const& j: tx["data"].get_array())
t.data.push_back(toByte(j));
2014-02-21 19:18:30 +00:00
txs.push_back(t);
}
}
2014-03-31 01:49:36 +00:00
map<Address, tuple<u256, u256, u256, map<u256, u256>, bytes>> addresses;
Transactions txs;
bytes thisTxData;
};
void doTests(json_spirit::mValue& v, bool _fillin)
2014-02-21 19:18:30 +00:00
{
for (auto& i: v.get_obj())
2014-02-21 19:18:30 +00:00
{
cnote << i.first;
mObject& o = i.second.get_obj();
BOOST_REQUIRE( o.count("env") > 0 );
BOOST_REQUIRE( o.count("pre") > 0 );
BOOST_REQUIRE( o.count("exec") > 0 );
2014-02-21 19:18:30 +00:00
VM vm;
eth::test::FakeExtVM fev;
2014-02-21 19:18:30 +00:00
fev.importEnv(o["env"].get_obj());
fev.importState(o["pre"].get_obj());
if (_fillin)
o["pre"] = mValue(fev.exportState());
bytes output;
2014-02-21 19:18:30 +00:00
for (auto i: o["exec"].get_array())
{
fev.importExec(i.get_obj());
output = vm.go(fev).toBytes();
2014-02-21 19:18:30 +00:00
}
if (_fillin)
{
o["post"] = mValue(fev.exportState());
o["txs"] = fev.exportTxs();
mArray df;
for (auto const& i: output)
FakeExtVM::push(df, i);
o["out"] = df;
2014-02-21 19:18:30 +00:00
}
else
{
BOOST_REQUIRE( o.count("post") > 0 );
BOOST_REQUIRE( o.count("txs") > 0 );
BOOST_REQUIRE( o.count("out") > 0 );
eth::test::FakeExtVM test;
2014-02-21 19:18:30 +00:00
test.importState(o["post"].get_obj());
test.importTxs(o["txs"].get_array());
int i = 0;
for (auto const& d: o["out"].get_array())
{
BOOST_CHECK_MESSAGE( output[i] == FakeExtVM::toInt(d), "Output byte [" << i << "] different!");
++i;
}
BOOST_CHECK( test.addresses == fev.addresses);
BOOST_CHECK( test.txs == fev.txs );
2014-02-21 19:18:30 +00:00
}
}
}
} } // Namespace Close
BOOST_AUTO_TEST_CASE(vm_tests)
{
try
{
cnote << "Testing VM...";
json_spirit::mValue v;
string s = asString(contents("../../tests/vmtests.json"));
BOOST_REQUIRE_MESSAGE( s.length() > 0, "Contents of 'vmtests.json' is empty. Have you cloned the 'tests' repo branch develop?" );
json_spirit::read_string(s, v);
eth::test::doTests(v, false);
}
catch( std::exception& e)
{
BOOST_ERROR("Failed VM Test with Exception: " << e.what());
}
}
2014-02-21 19:18:30 +00:00
#if 0
2014-02-21 19:18:30 +00:00
string makeTestCase()
{
json_spirit::mObject o;
VM vm;
BlockInfo pb;
pb.hash = sha3("previousHash");
pb.nonce = sha3("previousNonce");
BlockInfo cb = pb;
cb.difficulty = 256;
cb.timestamp = 1;
cb.coinbaseAddress = toAddress(sha3("coinbase"));
FakeExtVM fev(pb, cb, 0);
2014-03-31 01:49:36 +00:00
bytes init;
fev.setContract(toAddress(sha3("contract")), ether, 0, compileLisp("(suicide (txsender))", false, init), map<u256, u256>());
2014-02-21 19:18:30 +00:00
o["env"] = fev.exportEnv();
o["pre"] = fev.exportState();
fev.setTransaction(toAddress(sha3("sender")), ether, finney, bytes());
2014-02-21 19:18:30 +00:00
mArray execs;
execs.push_back(fev.exportExec());
o["exec"] = execs;
vm.go(fev);
2014-02-21 19:18:30 +00:00
o["post"] = fev.exportState();
o["txs"] = fev.exportTxs();
2014-02-21 19:18:30 +00:00
return json_spirit::write_string(json_spirit::mValue(o), true);
}
};
}
#endif