electrum

commit 787ac5fe994086b994524201fec4b63f876c7c85
parent c19f9ee755b46701eb07d80370effffebbd67394
Author: SomberNight <somber.night@protonmail.com>
Date:   Fri, 27 Dec 2019 17:43:12 +0100

interface: make changing max incoming msg size (1 MB) easier

Diffstat:
M
electrum/interface.py
 | 
8
+++++++-

1 file changed, 7 insertions(+), 1 deletion(-)
diff --git a/electrum/interface.py b/electrum/interface.py
@@ -36,7 +36,7 @@ import itertools
 import logging
 
 import aiorpcx
-from aiorpcx import RPCSession, Notification, NetAddress
+from aiorpcx import RPCSession, Notification, NetAddress, NewlineFramer
 from aiorpcx.curio import timeout_after, TaskTimeout
 from aiorpcx.jsonrpc import JSONRPC, CodeMessageError
 from aiorpcx.rawsocket import RSClient
@@ -62,6 +62,8 @@ ca_path = certifi.where()
 
 BUCKET_NAME_OF_ONION_SERVERS = 'onion'
 
+MAX_INCOMING_MSG_SIZE = 1_000_000  # in bytes
+
 
 class NetworkTimeout:
     # seconds
@@ -157,6 +159,10 @@ class NotificationSession(RPCSession):
         if self.interface.debug or self.interface.network.debug:
             self.interface.logger.debug(msg)
 
+    def default_framer(self):
+        # overridden so that max_size can be customized
+        return NewlineFramer(max_size=MAX_INCOMING_MSG_SIZE)
+
 
 class NetworkException(Exception): pass