StateImporter.cpp

// Aleth: Ethereum C++ client, tools and libraries.
// Copyright 2015-2019 Aleth Authors.
// Licensed under the GNU General Public License, Version 3.
#include "StateImporter.h"

#include <libdevcore/OverlayDB.h>
#include <libdevcore/RLP.h>
#include <libdevcore/TrieDB.h>

namespace dev
{
namespace eth
{

namespace
{

class StateImporter: public StateImporterFace
{
public:
	explicit StateImporter(OverlayDB& _stateDb): m_trie(&_stateDb) { m_trie.init(); }

	void importAccount(h256 const& _addressHash, u256 const& _nonce, u256 const& _balance, std::map<h256, bytes> const& _storage, h256 const& _codeHash) override
	{
		RLPStream s(4);
		s << _nonce << _balance;

		h256 const storageRoot = isAccountImported(_addressHash) ? accountStorageRoot(_addressHash) : EmptyTrie;
		if (_storage.empty())
			s.append(storageRoot);
		else
		{
			SpecificTrieDB<GenericTrieDB<OverlayDB>, h256> storageDB(m_trie.db(), storageRoot);
			for (auto const& hashAndValue: _storage)
				storageDB.insert(hashAndValue.first, hashAndValue.second);

			s.append(storageDB.root());
		}

		s << _codeHash;

		m_trie.insert(_addressHash, &s.out());
	}

	h256 importCode(bytesConstRef _code) override
	{
		h256 const hash = sha3(_code);
		m_trie.db()->insert(hash, _code);
		return hash;
	}

	void commitStateDatabase() override { m_trie.db()->commit(); }

	bool isAccountImported(h256 const& _addressHash) const override { return m_trie.contains(_addressHash); }

	h256 stateRoot() const override { return m_trie.root(); }

	std::string lookupCode(h256 const& _hash) const override { return m_trie.db()->lookup(_hash); }

private:
	// can be used only with already imported accounts
	h256 accountStorageRoot(h256 const& _addressHash) const
	{
		std::string const account = m_trie.at(_addressHash);
		assert(!account.empty());
		
		RLP accountRlp(account);
		if (accountRlp.itemCount() < 3)
			BOOST_THROW_EXCEPTION(InvalidAccountInTheDatabase());

		return accountRlp[2].toHash<h256>(RLP::VeryStrict);
	}

	SpecificTrieDB<GenericTrieDB<OverlayDB>, h256> m_trie;
};

}

std::unique_ptr<StateImporterFace> createStateImporter(OverlayDB& _stateDb)
{
	return std::unique_ptr<StateImporterFace>(new StateImporter(_stateDb));
}

}
}