From 18e92f7fc464dde7ac6c90878131ecf8584f9fce Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20Nowosielski?= <pawel@nethermind.io>
Date: Wed, 4 Sep 2024 21:57:08 +0200
Subject: [PATCH] protobuf changes

---
 p2p/starknet/p2p/proto/class.proto    | 4 ++++
 p2p/starknet/p2p/proto/snapshot.proto | 8 ++++++--
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/p2p/starknet/p2p/proto/class.proto b/p2p/starknet/p2p/proto/class.proto
index f68235350c..227adca4e7 100644
--- a/p2p/starknet/p2p/proto/class.proto
+++ b/p2p/starknet/p2p/proto/class.proto
@@ -61,3 +61,7 @@ message Classes {
     uint32   domain        = 1;
     repeated Class classes = 2;
 }
+
+message ClassHashesRequest {
+    repeated Hash class_hashes = 1;
+}
diff --git a/p2p/starknet/p2p/proto/snapshot.proto b/p2p/starknet/p2p/proto/snapshot.proto
index 47b5ae12e0..997c58c47a 100644
--- a/p2p/starknet/p2p/proto/snapshot.proto
+++ b/p2p/starknet/p2p/proto/snapshot.proto
@@ -58,6 +58,7 @@ message ContractRangeResponse {
         ContractRange range = 4;
         Fin           fin   = 5;
     }
+    PatriciaRangeProof range_proof = 6;
 }
 
 // duplicate of GetContractRange. Can introduce a 'type' instead.
@@ -77,6 +78,7 @@ message ClassRangeResponse {
         Classes classes = 4;
         Fin     fin     = 5;
     }
+    PatriciaRangeProof range_proof = 6;
 }
 
 // A position in some contract's state tree is identified by the state tree's root and the key in it
@@ -105,8 +107,10 @@ message ContractStorage {
 
 message ContractStorageResponse {
     optional Hash     state_root = 1; // may not appear if Fin is sent to end the whole response
+    optional Address contract_address = 2;
     oneof responses {
-        ContractStorage storage = 2;
-        Fin             fin     = 3;
+        ContractStorage storage = 3;
+        Fin             fin     = 4;
     }
+    PatriciaRangeProof range_proof = 5;
 }