Fix storage reset

rust-ethereum · Nov 18, 2023 · d2a9a9a · d2a9a9a
1 parent ba17d41
commit d2a9a9a
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/src/backend/in_memory.rs b/src/backend/in_memory.rs
@@ -174,12 +174,13 @@ impl RuntimeBackend for InMemoryBackend {
 	}
 
 	fn set_storage(&mut self, address: H160, index: H256, value: H256) -> Result<(), ExitError> {
-		self.current_layer_mut()
-			.state
-			.entry(address)
-			.or_default()
-			.storage
-			.insert(index, value);
+		let entry = self.current_layer_mut().state.entry(address).or_default();
+
+		if value == H256::default() {
+			entry.storage.remove(&index);
+		} else {
+			entry.storage.insert(index, value);
+		}
 		Ok(())
 	}