allow fetching poseidon hash in zktrie mode

scroll-tech · Jan 8, 2025 · 1ff0d01 · 1ff0d01
1 parent 3fcfbe3
commit 1ff0d01
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/core/state/state_object.go b/core/state/state_object.go
@@ -544,7 +544,10 @@ func (s *stateObject) setNonce(nonce uint64) {
 }
 
 func (s *stateObject) PoseidonCodeHash() []byte {
-	panic("shouldn't be called")
+	if !s.db.IsZktrie() {
+		panic("PoseidonCodeHash is only available in zktrie mode")
+	}
+	return s.data.PoseidonCodeHash
 }
 
 func (s *stateObject) KeccakCodeHash() []byte {