feat: allow disabling part proofs

celestiaorg · Oct 11, 2024 · 49befbd · 49befbd
1 parent 56bf584
commit 49befbd
Show file tree

Hide file tree

Showing 4 changed files with 291 additions and 116 deletions.
diff --git a/proto/tendermint/rpc/grpc/types.pb.go b/proto/tendermint/rpc/grpc/types.pb.go
diff --git a/proto/tendermint/rpc/grpc/types.proto b/proto/tendermint/rpc/grpc/types.proto
@@ -17,11 +17,13 @@ message RequestBroadcastTx {
 }
 
 message BlockByHashRequest {
-  bytes hash = 1;
+  bytes hash        = 1;
+  bool  prove_parts = 2;
 }
 
 message BlockByHeightRequest {
-  int64 height = 1;
+  int64 height      = 1;
+  bool  prove_parts = 2;
 }
 
 message BlockMetaByHashRequest {

diff --git a/rpc/grpc/api.go b/rpc/grpc/api.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"github.com/tendermint/tendermint/proto/tendermint/crypto"
 	"sync"
 	"time"
 
@@ -188,6 +189,9 @@ func (blockAPI *BlockAPI) BlockByHash(req *BlockByHashRequest, stream BlockAPI_B
 		if err != nil {
 			return err
 		}
+		if !req.ProveParts {
+			part.Proof = crypto.Proof{}
+		}
 		isLastPart := i == int(blockMeta.BlockID.PartSetHeader.Total)-1
 		err = stream.Send(&BlockByHashResponse{
 			BlockPart: part,
@@ -208,6 +212,9 @@ func (blockAPI *BlockAPI) BlockByHeight(req *BlockByHeightRequest, stream BlockA
 		if err != nil {
 			return err
 		}
+		if !req.ProveParts {
+			part.Proof = crypto.Proof{}
+		}
 		isLastPart := i == int(blockMeta.BlockID.PartSetHeader.Total)-1
 		err = stream.Send(&BlockByHeightResponse{
 			BlockPart: part,