Skip to content

Commit

Permalink
Add mocked_host library
Browse files Browse the repository at this point in the history
It takes the MockedHost implementation from evmone, converts it to C++11 and creates an interface library out of it.
  • Loading branch information
chfast committed Nov 19, 2019
1 parent 015e4b8 commit 22c2c52
Show file tree
Hide file tree
Showing 4 changed files with 338 additions and 0 deletions.
320 changes: 320 additions & 0 deletions include/evmc/mocked_host.hpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,320 @@
// EVMC: Ethereum Client-VM Connector API.
// Copyright 2019 The EVMC Authors.
// Licensed under the Apache License, Version 2.0.
#pragma once

#include <evmc/evmc.hpp>
#include <algorithm>
#include <iterator>
#include <string>
#include <unordered_map>
#include <vector>

namespace evmc
{
/// The string of bytes.
using bytes = std::basic_string<uint8_t>;

/// Mocked account.
struct MockedAccount
{
/// Extended value (by dirty flag) for account storage.
struct storage_value
{
/// The storage value.
bytes32 value;

/// True means this value has been modified already by the current transaction.
bool dirty{false};

/// Default constructor.
storage_value() noexcept = default;

/// Constructor.
storage_value(const bytes32& _value, bool _dirty = false) noexcept // NOLINT
: value{_value}, dirty{_dirty}
{}
};

/// The account code.
bytes code;

/// The code hash. Can be a value not related to the actual code.
bytes32 codehash;

/// The account balance.
uint256be balance;

/// The account storage map.
std::unordered_map<bytes32, storage_value> storage;

/// Helper method for setting balance by numeric type.
void set_balance(uint64_t x) noexcept
{
balance = uint256be{};
for (std::size_t i = 0; i < sizeof(x); ++i)
balance.bytes[sizeof(balance) - 1 - i] = static_cast<uint8_t>(x >> (8 * i));
}
};

/// Mocked EVMC Host implementation.
class MockedHost : public Host
{
public:
/// LOG record.
struct log_record
{
/// The address of the account which created the log.
address creator;

/// The data attached to the log.
bytes data;

/// The log topics.
std::vector<bytes32> topics;

/// Equal operator.
bool operator==(const log_record& other) const noexcept
{
return creator == other.creator && data == other.data &&
topics.size() == other.topics.size() &&
std::equal(topics.begin(), topics.end(), other.topics.begin());
}
};

/// SELFDESTRUCT record.
struct selfdestuct_record
{
/// The address of the account which has self-destructed.
address selfdestructed;

/// The address of the beneficiary account.
address beneficiary;

/// Equal operator.
bool operator==(const selfdestuct_record& other) const noexcept
{
return selfdestructed == other.selfdestructed && beneficiary == other.beneficiary;
}
};

/// The set of all accounts in the Host, organized by their addresses.
std::unordered_map<address, MockedAccount> accounts;

/// The EVMC transaction context to be returned by get_tx_context().
evmc_tx_context tx_context = {};

/// The block header hash value to be returned by get_block_hash().
bytes32 block_hash = {};

/// The call result to be returned by the call() method.
evmc_result call_result = {};

/// The record of all block numbers for which get_block_hash() was called.
std::vector<int64_t> recorded_blockhashes;

/// The record of all account accesses.
std::vector<address> recorded_account_accesses;

/// The maximum number of entries in recorded_account_accesses record.
/// This is arbitrary value useful in fuzzing when we don't want the record to explode.
static constexpr auto max_recorded_account_accesses = 200;

/// The record of all call messages requested in the call() method.
std::vector<evmc_message> recorded_calls;

/// The maximum number of entries in recorded_calls record.
/// This is arbitrary value useful in fuzzing when we don't want the record to explode.
static constexpr auto max_recorded_calls = 100;

/// The record of all LOGs passed to the emit_log() method.
std::vector<log_record> recorded_logs;

/// The record of all SELFDESTRUCTs from the selfdestruct() method.
std::vector<selfdestuct_record> recorded_selfdestructs;

protected:
/// The copy of call inputs for the recorded_calls record.
std::vector<bytes> m_recorded_calls_inputs;

/// Record an account access.
/// @param addr The address of the accessed account.
void record_account_access(const address& addr)
{
if (recorded_account_accesses.empty())
recorded_account_accesses.reserve(max_recorded_account_accesses);

if (recorded_account_accesses.size() < max_recorded_account_accesses)
recorded_account_accesses.emplace_back(addr);
}

/// Returns true if an account exists (EVMC Host method).
bool account_exists(const address& addr) noexcept override
{
record_account_access(addr);
return accounts.count(addr) != 0;
}

/// Get the account's storage value at the given key (EVMC Host method).
bytes32 get_storage(const address& addr, const bytes32& key) noexcept override
{
record_account_access(addr);

const auto account_iter = accounts.find(addr);
if (account_iter == accounts.end())
return {};

const auto storage_iter = account_iter->second.storage.find(key);
if (storage_iter != account_iter->second.storage.end())
return storage_iter->second.value;
return {};
}

/// Set the account's storage value (EVMC Host method).
evmc_storage_status set_storage(const address& addr,
const bytes32& key,
const bytes32& value) noexcept override
{
record_account_access(addr);
const auto it = accounts.find(addr);
if (it == accounts.end())
return EVMC_STORAGE_UNCHANGED;

auto& old = it->second.storage[key];

// Follow https://eips.ethereum.org/EIPS/eip-1283 specification.
// WARNING! This is not complete implementation as refund is not handled here.

if (old.value == value)
return EVMC_STORAGE_UNCHANGED;

evmc_storage_status status;
{
if (!old.dirty)
{
old.dirty = true;
if (!old.value)
status = EVMC_STORAGE_ADDED;
else if (value)
status = EVMC_STORAGE_MODIFIED;
else
status = EVMC_STORAGE_DELETED;
}
else
status = EVMC_STORAGE_MODIFIED_AGAIN;
}

old.value = value;
return status;
}

/// Get the account's balance (EVMC Host method).
uint256be get_balance(const address& addr) noexcept override
{
record_account_access(addr);
const auto it = accounts.find(addr);
if (it == accounts.end())
return {};

return it->second.balance;
}

/// Get the account's code size (EVMC host method).
size_t get_code_size(const address& addr) noexcept override
{
record_account_access(addr);
const auto it = accounts.find(addr);
if (it == accounts.end())
return 0;
return it->second.code.size();
}

/// Get the account's code hash (EVMC host method).
bytes32 get_code_hash(const address& addr) noexcept override
{
record_account_access(addr);
const auto it = accounts.find(addr);
if (it == accounts.end())
return {};
return it->second.codehash;
}

/// Copy the account's code to the given buffer (EVMC host method).
size_t copy_code(const address& addr,
size_t code_offset,
uint8_t* buffer_data,
size_t buffer_size) noexcept override
{
record_account_access(addr);
const auto it = accounts.find(addr);
if (it == accounts.end())
return 0;

const auto& code = it->second.code;

if (code_offset >= code.size())
return 0;

const auto n = std::min(buffer_size, code.size() - code_offset);

if (n > 0)
std::copy_n(&code[code_offset], n, buffer_data);
return n;
}

/// Selfdestruct the account (EVMC host method).
void selfdestruct(const address& addr, const address& beneficiary) noexcept override
{
record_account_access(addr);
recorded_selfdestructs.push_back({addr, beneficiary});
}

/// Call/create other contract (EVMC host method).
result call(const evmc_message& msg) noexcept override
{
record_account_access(msg.destination);

if (recorded_calls.empty())
{
recorded_calls.reserve(max_recorded_calls);
m_recorded_calls_inputs.reserve(max_recorded_calls); // Iterators will not invalidate.
}

if (recorded_calls.size() < max_recorded_calls)
{
recorded_calls.emplace_back(msg);
auto& call_msg = recorded_calls.back();
if (call_msg.input_size > 0)
{
m_recorded_calls_inputs.emplace_back(call_msg.input_data, call_msg.input_size);
const auto& input_copy = m_recorded_calls_inputs.back();
call_msg.input_data = input_copy.data();
}
}
return result{call_result};
}

/// Get transaction context (EVMC host method).
evmc_tx_context get_tx_context() noexcept override { return tx_context; }

/// Get the block header hash (EVMC host method).
bytes32 get_block_hash(int64_t block_number) noexcept override
{
recorded_blockhashes.emplace_back(block_number);
return block_hash;
}

/// Emit LOG (EVMC host method).
void emit_log(const address& addr,
const uint8_t* data,
size_t data_size,
const bytes32 topics[],
size_t topics_count) noexcept override
{
recorded_logs.push_back({addr, {data, data_size}, {}});
auto& record_topics = recorded_logs.back().topics;
record_topics.reserve(topics_count);
std::copy_n(topics, topics_count, std::back_inserter(record_topics));
}
};
} // namespace evmc
1 change: 1 addition & 0 deletions lib/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,3 +4,4 @@

add_subdirectory(instructions)
add_subdirectory(loader)
add_subdirectory(mocked_host)
15 changes: 15 additions & 0 deletions lib/mocked_host/CMakeLists.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,15 @@
# EVMC: Ethereum Client-VM Connector API.
# Copyright 2019 The EVMC Authors.
# Licensed under the Apache License, Version 2.0.

add_library(mocked_host INTERFACE)
target_sources(mocked_host INTERFACE $<BUILD_INTERFACE:${include_dir}/evmc/mocked_host.hpp>)

add_library(evmc::mocked_host ALIAS mocked_host)
target_include_directories(mocked_host INTERFACE
$<BUILD_INTERFACE:${include_dir}>$<INSTALL_INTERFACE:${CMAKE_INSTALL_INCLUDEDIR}>
)

if(EVMC_INSTALL)
install(TARGETS mocked_host EXPORT evmcTargets)
endif()
2 changes: 2 additions & 0 deletions test/integration/compilation/compilation_test.cxx
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@
#include <evmc/helpers.h>
#include <evmc/instructions.h>
#include <evmc/loader.h>
#include <evmc/mocked_host.hpp>
#include <evmc/utils.h>

// Include again to check if headers have proper include guards.
Expand All @@ -18,4 +19,5 @@
#include <evmc/helpers.h>
#include <evmc/instructions.h>
#include <evmc/loader.h>
#include <evmc/mocked_host.hpp>
#include <evmc/utils.h>

0 comments on commit 22c2c52

Please sign in to comment.