From 7dd62a923023aced8766f74ae9fb89c7745c4167 Mon Sep 17 00:00:00 2001 From: msutton Date: Sat, 16 Jul 2022 22:50:48 +0300 Subject: [PATCH] Change message size limits to 100MB --- cmd/kaspawallet/daemon/client/client.go | 10 ++++++++-- cmd/kaspawallet/daemon/server/server.go | 5 ++++- 2 files changed, 12 insertions(+), 3 deletions(-) diff --git a/cmd/kaspawallet/daemon/client/client.go b/cmd/kaspawallet/daemon/client/client.go index 6cfd38a32..1bf95d9dc 100644 --- a/cmd/kaspawallet/daemon/client/client.go +++ b/cmd/kaspawallet/daemon/client/client.go @@ -13,10 +13,16 @@ import ( // Connect connects to the kaspawalletd server, and returns the client instance func Connect(address string) (pb.KaspawalletdClient, func(), error) { // Connection is local, so 1 second timeout is sufficient - ctx, cancel := context.WithTimeout(context.Background(), time.Second) + ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second) defer cancel() - conn, err := grpc.DialContext(ctx, address, grpc.WithInsecure(), grpc.WithBlock()) + const maxMsgSize = 100_000_000 + conn, err := grpc.DialContext( + ctx, address, + grpc.WithInsecure(), + grpc.WithBlock(), + grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(maxMsgSize), grpc.MaxCallSendMsgSize(maxMsgSize))) + if err != nil { if errors.Is(err, context.DeadlineExceeded) { return nil, nil, errors.New("kaspawallet daemon is not running, start it with `kaspawallet start-daemon`") diff --git a/cmd/kaspawallet/daemon/server/server.go b/cmd/kaspawallet/daemon/server/server.go index d66ff8d65..aa51ccf55 100644 --- a/cmd/kaspawallet/daemon/server/server.go +++ b/cmd/kaspawallet/daemon/server/server.go @@ -96,7 +96,10 @@ func Start(params *dagconfig.Params, listen, rpcServer string, keysFilePath stri } }) - grpcServer := grpc.NewServer() + const maxMsgSize = 100_000_000 + grpcServer := grpc.NewServer( + grpc.MaxRecvMsgSize(maxMsgSize), + grpc.MaxSendMsgSize(maxMsgSize)) pb.RegisterKaspawalletdServer(grpcServer, serverInstance) spawn("grpcServer.Serve", func() {