Enhance gRPC message handling by setting max message size to 64MB

- Updated gRPC client and server configurations to support larger message sizes, ensuring compatibility with large configurations and user data.
- Added constants for max receive and send message sizes in both the client and server setup.

Author: x0sina

controller/rpc/rpc_test.go

@@ -64,8 +64,14 @@ func TestMain(m *testing.M) {
 	}
 
 	creds := credentials.NewClientTLSFromCert(certPool, "")
+	// Set max message size to 64MB to match server configuration
+	const maxMsgSize = 64 * 1024 * 1024 // 64MB
 	opts := []grpc.DialOption{
 		grpc.WithTransportCredentials(creds),
+		grpc.WithDefaultCallOptions(
+			grpc.MaxCallRecvMsgSize(maxMsgSize),
+			grpc.MaxCallSendMsgSize(maxMsgSize),
+		),
 	}
 
 	conn, err := grpc.NewClient(addr, opts...)

controller/rpc/service.go

@@ -4,13 +4,14 @@ import (
 	"context"
 	"crypto/tls"
 	"fmt"
+	"log"
+	"net"
+
 	"github.com/pasarguard/node/common"
 	"github.com/pasarguard/node/config"
 	"github.com/pasarguard/node/controller"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials"
-	"log"
-	"net"
 )
 
 type Service struct {
@@ -30,8 +31,12 @@ func StartGRPCListener(tlsConfig *tls.Config, addr string, cfg *config.Config) (
 	creds := credentials.NewTLS(tlsConfig)
 
 	// Create the gRPC server with conditional middleware
+	// Set max message size to 64MB to handle large configs and user data
+	const maxMsgSize = 64 * 1024 * 1024 // 64MB
 	grpcServer := grpc.NewServer(
 		grpc.Creds(creds),
+		grpc.MaxRecvMsgSize(maxMsgSize),
+		grpc.MaxSendMsgSize(maxMsgSize),
 		grpc.UnaryInterceptor(ConditionalMiddleware(s)),
 		grpc.StreamInterceptor(ConditionalStreamMiddleware(s)),
 	)