Implement starknet_getStorageProof RPC

core/state.go

@@ -42,6 +42,10 @@
 	ContractNonce(addr *felt.Felt) (*felt.Felt, error)
 	ContractStorage(addr, key *felt.Felt) (*felt.Felt, error)
 	Class(classHash *felt.Felt) (*DeclaredClass, error)
+
+	ClassTrie() (*trie.Trie, error)
+	ContractTrie() (*trie.Trie, error)
+	ContractStorageTrie(addr *felt.Felt) (*trie.Trie, error)
 }
 
 type State struct {
@@ -124,6 +128,21 @@
 	return crypto.PoseidonArray(stateVersion, storageRoot, classesRoot), nil
 }
 
+func (s *State) ClassTrie() (*trie.Trie, error) {
+	// We don't need to call the closer function here because we are only reading the trie
+	tr, _, err := s.classesTrie()
+	return tr, err
+}
+
+func (s *State) ContractTrie() (*trie.Trie, error) {
+	tr, _, err := s.storage()
+	return tr, err
+}
+
+func (s *State) ContractStorageTrie(addr *felt.Felt) (*trie.Trie, error) {
+	return storage(addr, s.txn)
+}
+
 // storage returns a [core.Trie] that represents the Starknet global state in the given Txn context.
 func (s *State) storage() (*trie.Trie, func() error, error) {
 	return s.globalTrie(db.StateTrie, trie.NewTriePedersen)

core/state_snapshot.go

@@ -4,9 +4,12 @@
 	"errors"
 
 	"github.com/NethermindEth/juno/core/felt"
+	"github.com/NethermindEth/juno/core/trie"
 	"github.com/NethermindEth/juno/db"
 )
 
+var ErrHistoricalTrieNotSupported = errors.New("cannot support historical trie")
+
 type stateSnapshot struct {
 	blockNumber uint64
 	state       StateHistoryReader
@@ -87,3 +90,15 @@
 	}
 	return declaredClass, nil
 }
+
+func (s *stateSnapshot) ClassTrie() (*trie.Trie, error) {
+	return nil, ErrHistoricalTrieNotSupported
+}
+
+func (s *stateSnapshot) ContractTrie() (*trie.Trie, error) {
+	return nil, ErrHistoricalTrieNotSupported
+}
+
+func (s *stateSnapshot) ContractStorageTrie(addr *felt.Felt) (*trie.Trie, error) {
+	return nil, ErrHistoricalTrieNotSupported
+}

core/trie/trie.go

@@ -718,6 +718,10 @@ func (t *Trie) Dump() {
 	t.dump(0, nil)
 }
 
+func (t *Trie) HashFn() crypto.HashFn {
+	return t.hash
+}
+
 // Try to print a [Trie] in a somewhat human-readable form
 /*
 Todo: create more meaningful representation of trie. In the current format string, storage is being