From 55781688cd64fc6b80b5b8cf6c67e4564466ac71 Mon Sep 17 00:00:00 2001
From: Guillermo Rodriguez <guillermo@telos.net>
Date: Sat, 24 Dec 2022 11:28:17 -0300
Subject: [PATCH] Testing unlimited msg size

---
 skynet/brain.py             | 4 ++--
 skynet/dgpu.py              | 2 +-
 skynet/frontend/__init__.py | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/skynet/brain.py b/skynet/brain.py
index 1d8d8a7..6ec745c 100644
--- a/skynet/brain.py
+++ b/skynet/brain.py
@@ -406,8 +406,8 @@ async def run_skynet(
             own_cert_string=tls_cert_data)
 
     with (
-        pynng.Rep0() as rpc_sock,
-        pynng.Bus0() as dgpu_bus
+        pynng.Rep0(recv_max_size=0) as rpc_sock,
+        pynng.Bus0(recv_max_size=0) as dgpu_bus
     ):
         async with open_database_connection(
             db_user, db_pass, db_host) as db_pool:
diff --git a/skynet/dgpu.py b/skynet/dgpu.py
index d002d59..1186630 100644
--- a/skynet/dgpu.py
+++ b/skynet/dgpu.py
@@ -177,7 +177,7 @@ async def open_dgpu_node(
                 ca_string=skynet_cert_data)
 
         logging.info(f'connecting to {dgpu_address}')
-        with pynng.Bus0() as dgpu_sock:
+        with pynng.Bus0(recv_max_size=0) as dgpu_sock:
             dgpu_sock.tls_config = tls_config
             dgpu_sock.dial(dgpu_address)
 
diff --git a/skynet/frontend/__init__.py b/skynet/frontend/__init__.py
index f238186..06a9623 100644
--- a/skynet/frontend/__init__.py
+++ b/skynet/frontend/__init__.py
@@ -70,7 +70,7 @@ async def open_skynet_rpc(
             own_cert_string=tls_cert_data,
             ca_string=skynet_cert_data)
 
-    with pynng.Req0() as sock:
+    with pynng.Req0(recv_max_size=0) as sock:
         if security:
             sock.tls_config = tls_config