From c7c8b25c09ceb3cb384d1d29a50374f0f7e43826 Mon Sep 17 00:00:00 2001 From: Ori Newman Date: Mon, 28 Dec 2020 12:53:11 +0200 Subject: [PATCH] Set stream max message size and increase the max message size to 1GB (#1300) --- app/protocol/flows/blockrelay/ibd.go | 1 + .../network/netadapter/server/grpcserver/grpc_server.go | 2 +- .../network/netadapter/server/grpcserver/p2pserver.go | 3 ++- 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/app/protocol/flows/blockrelay/ibd.go b/app/protocol/flows/blockrelay/ibd.go index 71d8c00ae..1d4dfb71d 100644 --- a/app/protocol/flows/blockrelay/ibd.go +++ b/app/protocol/flows/blockrelay/ibd.go @@ -53,6 +53,7 @@ func (flow *handleRelayInvsFlow) runIBDIfNotRunning(highHash *externalapi.Domain } if blockInfo.BlockStatus == externalapi.StatusHeaderOnly { + log.Infof("Checking if the suggested pruning point %s is compatible to the node DAG", msgIBDRootHash.Hash) isValid, err := flow.Domain().Consensus().IsValidPruningPoint(msgIBDRootHash.Hash) if err != nil { return err diff --git a/infrastructure/network/netadapter/server/grpcserver/grpc_server.go b/infrastructure/network/netadapter/server/grpcserver/grpc_server.go index 1b19c0ac4..2cf4407b2 100644 --- a/infrastructure/network/netadapter/server/grpcserver/grpc_server.go +++ b/infrastructure/network/netadapter/server/grpcserver/grpc_server.go @@ -18,7 +18,7 @@ type gRPCServer struct { } // MaxMessageSize is the max size allowed for a message -const MaxMessageSize = 1024 * 1024 * 10 // 10MB +const MaxMessageSize = 1024 * 1024 * 1024 // 1GB // newGRPCServer creates a gRPC server func newGRPCServer(listeningAddresses []string) *gRPCServer { diff --git a/infrastructure/network/netadapter/server/grpcserver/p2pserver.go b/infrastructure/network/netadapter/server/grpcserver/p2pserver.go index 73dfbcd26..7add4fdfd 100644 --- a/infrastructure/network/netadapter/server/grpcserver/p2pserver.go +++ b/infrastructure/network/netadapter/server/grpcserver/p2pserver.go @@ -47,7 +47,8 @@ func (p *p2pServer) Connect(address string) (server.Connection, error) { } client := protowire.NewP2PClient(gRPCClientConnection) - stream, err := client.MessageStream(context.Background(), grpc.UseCompressor(gzip.Name)) + stream, err := client.MessageStream(context.Background(), grpc.UseCompressor(gzip.Name), + grpc.MaxCallRecvMsgSize(MaxMessageSize), grpc.MaxCallSendMsgSize(MaxMessageSize)) if err != nil { return nil, errors.Wrapf(err, "error getting client stream for %s", address) }