From 7ed5bc021addd80c81af4b89c2713983a1775fbf Mon Sep 17 00:00:00 2001
From: Guillaume Ballet <3272758+gballet@users.noreply.github.com>
Date: Tue, 19 Sep 2023 14:47:24 +0200
Subject: [PATCH] trie: add getter for preimage store in trie.Database (#28155)

---
 trie/database.go | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/trie/database.go b/trie/database.go
index 2915ff9484..535ad87d72 100644
--- a/trie/database.go
+++ b/trie/database.go
@@ -189,6 +189,15 @@ func (db *Database) WritePreimages() {
 	}
 }
 
+// Preimage retrieves a cached trie node pre-image from memory. If it cannot be
+// found cached, the method queries the persistent database for the content.
+func (db *Database) Preimage(hash common.Hash) []byte {
+	if db.preimages == nil {
+		return nil
+	}
+	return db.preimages.preimage(hash)
+}
+
 // Cap iteratively flushes old but still referenced trie nodes until the total
 // memory usage goes below the given threshold. The held pre-images accumulated
 // up to this point will be flushed in case the size exceeds the threshold.