Move m_storageOverlay and m_storageOriginal caches management into Ac…

…count.

libethereum/Account.cpp

@@ -20,8 +20,10 @@
  */
 
 #include "Account.h"
+#include "SecureTrieDB.h"
 #include "ValidationSchemes.h"
 #include <libdevcore/JsonUtils.h>
+#include <libdevcore/OverlayDB.h>
 #include <libethcore/ChainOperationParams.h>
 #include <libethcore/Precompiled.h>
 
@@ -39,6 +41,20 @@ void Account::setCode(bytes&& _code)
     m_codeHash = sha3(m_codeCache);
 }
 
+u256 Account::originalStorageValue(u256 const& _key, OverlayDB const& _db) const
+{
+    auto it = m_storageOriginal.find(_key);
+    if (it != m_storageOriginal.end())
+        return it->second;
+
+    // Not in the original values cache - go to the DB.
+    SecureTrieDB<h256, OverlayDB> const memdb(const_cast<OverlayDB*>(&_db), m_storageRoot);
+    std::string const payload = memdb.at(_key);
+    auto const value = payload.size() ? RLP(payload).toInt<u256>() : 0;
+    m_storageOriginal[_key] = value;
+    return value;
+}
+
 namespace js = json_spirit;
 
 namespace

libethereum/Account.h

@@ -22,13 +22,16 @@
 #pragma once
 
 #include <libdevcore/Common.h>
-#include <libdevcore/RLP.h>
-#include <libdevcore/TrieDB.h>
 #include <libdevcore/SHA3.h>
+#include <libdevcore/TrieCommon.h>
 #include <libethcore/Common.h>
 
+#include <boost/filesystem/path.hpp>
+
 namespace dev
 {
+class OverlayDB;
+
 namespace eth
 {
 
@@ -132,16 +135,24 @@ class Account
     /// the account.
     void setNonce(u256 const& _nonce) { m_nonce = _nonce; changed(); }
 
-
     /// @returns the root of the trie (whose nodes are stored in the state db externally to this class)
     /// which encodes the base-state of the account's storage (upon which the storage is overlaid).
     h256 baseRoot() const { assert(m_storageRoot); return m_storageRoot; }
 
+    u256 storageValue(u256 const& _key, OverlayDB const& _db) const
+    {
+        auto mit = m_storageOverlay.find(_key);
+        if (mit != m_storageOverlay.end())
+            return mit->second;
+
+        return originalStorageValue(_key, _db);
+    }
+
+    u256 originalStorageValue(u256 const& _key, OverlayDB const& _db) const;
+
     /// @returns the storage overlay as a simple hash map.
     std::unordered_map<u256, u256> const& storageOverlay() const { return m_storageOverlay; }
 
-    std::unordered_map<u256, u256> const& storageOriginal() const { return m_storageOriginal; }
-
     /// Set a key/value pair in the account's storage. This actually goes into the overlay, for committing
     /// to the trie later.
     void setStorage(u256 _p, u256 _v) { m_storageOverlay[_p] = _v; changed(); }
@@ -152,12 +163,6 @@ class Account
     /// Set the storage root.  Used when clearStorage() is reverted.
     void setStorageRoot(h256 const& _root) { m_storageOverlay.clear(); m_storageRoot = _root; changed(); }
 
-    /// Set a key/value pair in the account's storage to a value that is already present inside the
-    /// database.
-    void setStorageCache(u256 _p, u256 _v) const { m_storageOverlay[_p] = _v; }
-
-    void setStorageOriginal(u256 _p, u256 _v) const { m_storageOriginal[_p] = _v; }
-
     /// @returns the hash of the account's code.
     h256 codeHash() const { return m_codeHash; }
 

libethereum/SecureTrieDB.h

@@ -0,0 +1,35 @@
+/*
+    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/>.
+*/
+
+#pragma once
+
+#include <libdevcore/TrieDB.h>
+
+namespace dev
+{
+namespace eth
+{
+#if ETH_FATDB
+template <class KeyType, class DB>
+using SecureTrieDB = SpecificTrieDB<FatGenericTrieDB<DB>, KeyType>;
+#else
+template <class KeyType, class DB>
+using SecureTrieDB = SpecificTrieDB<HashedGenericTrieDB<DB>, KeyType>;
+#endif
+
+}  // namespace eth
+}  // namespace dev

libethereum/State.cpp

@@ -419,17 +419,7 @@ u256 State::storage(Address const& _id, u256 const& _key) const
 {
     if (Account const* a = account(_id))
     {
-        auto mit = a->storageOverlay().find(_key);
-        if (mit != a->storageOverlay().end())
-            return mit->second;
-
-        // Not in the storage cache - go to the DB.
-        SecureTrieDB<h256, OverlayDB> memdb(const_cast<OverlayDB*>(&m_db), a->baseRoot());          // promise we won't change the overlay! :)
-        string payload = memdb.at(_key);
-        u256 ret = payload.size() ? RLP(payload).toInt<u256>() : 0;
-        a->setStorageCache(_key, ret);
-        a->setStorageOriginal(_key, ret);
-        return ret;
+        return a->storageValue(_key, m_db);
     }
     else
         return 0;
@@ -440,25 +430,13 @@ void State::setStorage(Address const& _contract, u256 const& _key, u256 const& _
     m_changeLog.emplace_back(_contract, _key, storage(_contract, _key));
     Account& a = m_cache[_contract];
     a.setStorage(_key, _value);
-    if (a.storageOriginal().find(_key) == a.storageOriginal().end())
-    {
-        SecureTrieDB<h256, OverlayDB> memdb(const_cast<OverlayDB*>(&m_db), a.baseRoot());
-        string payload = memdb.at(_key);
-        u256 original = payload.size() ? RLP(payload).toInt<u256>() : 0;
-        a.setStorageOriginal(_key, original);
-    }
 }
 
 u256 State::originalStorageValue(Address const& _contract, u256 const& _key) const
 {
     if (Account const* a = account(_contract))
     {
-        auto it = a->storageOriginal().find(_key);
-        if (it != a->storageOriginal().end())
-            return it->second;
-
-        assert(a->storageOverlay().find(_key) == a->storageOverlay().end());
-        return storage(_contract, _key);
+        return a->originalStorageValue(_key, m_db);
     }
     else
         return 0;

libethereum/State.h

@@ -17,20 +17,20 @@
 
 #pragma once
 
-#include <array>
-#include <unordered_map>
+#include "Account.h"
+#include "GasPricer.h"
+#include "SecureTrieDB.h"
+#include "Transaction.h"
+#include "TransactionReceipt.h"
 #include <libdevcore/Common.h>
-#include <libdevcore/RLP.h>
-#include <libdevcore/TrieDB.h>
 #include <libdevcore/OverlayDB.h>
-#include <libethcore/Exceptions.h>
+#include <libdevcore/RLP.h>
 #include <libethcore/BlockHeader.h>
+#include <libethcore/Exceptions.h>
 #include <libethereum/CodeSizeCache.h>
 #include <libevm/ExtVMFace.h>
-#include "Account.h"
-#include "Transaction.h"
-#include "TransactionReceipt.h"
-#include "GasPricer.h"
+#include <array>
+#include <unordered_map>
 
 namespace dev
 {
@@ -73,15 +73,9 @@ enum class Permanence
 {
     Reverted,
     Committed,
-    Uncommitted	///< Uncommitted state for change log readings in tests.
+    Uncommitted  ///< Uncommitted state for change log readings in tests.
 };
 
-#if ETH_FATDB
-template <class KeyType, class DB> using SecureTrieDB = SpecificTrieDB<FatGenericTrieDB<DB>, KeyType>;
-#else
-template <class KeyType, class DB> using SecureTrieDB = SpecificTrieDB<HashedGenericTrieDB<DB>, KeyType>;
-#endif
-
 DEV_SIMPLE_EXCEPTION(InvalidAccountStartNonceInState);
 DEV_SIMPLE_EXCEPTION(IncorrectAccountStartNonceInState);
 
@@ -327,7 +321,7 @@ class State
     u256 const& requireAccountStartNonce() const;
     void noteAccountStartNonce(u256 const& _actual);
 
-    /// Create a savepoint in the state changelog.	///
+    /// Create a savepoint in the state changelog.  ///
     /// @return The savepoint index that can be used in rollback() function.
     size_t savepoint() const;
 
@@ -357,12 +351,19 @@ class State
     /// exception occurred.
     bool executeTransaction(Executive& _e, Transaction const& _t, OnOpFunc const& _onOp);
 
-    OverlayDB m_db;								///< Our overlay for the state tree.
-    SecureTrieDB<Address, OverlayDB> m_state;	///< Our state tree, as an OverlayDB DB.
-    mutable std::unordered_map<Address, Account> m_cache;	///< Our address cache. This stores the states of each address that has (or at least might have) been changed.
-    mutable std::vector<Address> m_unchangedCacheEntries;	///< Tracks entries in m_cache that can potentially be purged if it grows too large.
-    mutable std::set<Address> m_nonExistingAccountsCache;	///< Tracks addresses that are known to not exist.
-    AddressHash m_touched;						///< Tracks all addresses touched so far.
+    /// Our overlay for the state tree.
+    OverlayDB m_db;
+    /// Our state tree, as an OverlayDB DB.
+    SecureTrieDB<Address, OverlayDB> m_state;
+    /// Our address cache. This stores the states of each address that has (or at least might have)
+    /// been changed.
+    mutable std::unordered_map<Address, Account> m_cache;
+    /// Tracks entries in m_cache that can potentially be purged if it grows too large.
+    mutable std::vector<Address> m_unchangedCacheEntries;
+    /// Tracks addresses that are known to not exist.
+    mutable std::set<Address> m_nonExistingAccountsCache;
+    /// Tracks all addresses touched so far.
+    AddressHash m_touched;
 
     u256 m_accountStartNonce;