LBRYFoundation/LBRY-Vault
1
0
Fork 0
mirror of https://github.com/LBRYFoundation/LBRY-Vault.git synced 2025-08-31 01:11:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

network: make MAX_INCOMING_MSG_SIZE configurable from config

Browse source 
requested in https://github.com/spesmilo/electrum/issues/4315#issuecomment-698730778
This commit is contained in:
SomberNight 2020-09-25 11:15:04 +02:00
parent 52bd0eb1a6
commit 5a7c3dc4d0
No known key found for this signature in database
GPG key ID: B33B5F232C6271E9
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
electrum/interface.py
View file

@ -126,13 +126,13 @@ def assert_list_or_tuple(val: Any) -> None:
class NotificationSession(RPCSession): class NotificationSession(RPCSession):
def __init__(self, *args, **kwargs): def __init__(self, *args, interface: 'Interface', **kwargs):
super(NotificationSession, self).__init__(*args, **kwargs) super(NotificationSession, self).__init__(*args, **kwargs)
self.subscriptions = defaultdict(list) self.subscriptions = defaultdict(list)
self.cache = {} self.cache = {}
self.default_timeout = NetworkTimeout.Generic.NORMAL self.default_timeout = NetworkTimeout.Generic.NORMAL
self._msg_counter = itertools.count(start=1) self._msg_counter = itertools.count(start=1)
self.interface = None # type: Optional[Interface] self.interface = interface # type: Interface
self.cost_hard_limit = 0 # disable aiorpcx resource limits self.cost_hard_limit = 0 # disable aiorpcx resource limits
async def handle_request(self, request): async def handle_request(self, request):
@ -209,7 +209,9 @@ class NotificationSession(RPCSession):
def default_framer(self): def default_framer(self):
# overridden so that max_size can be customized # overridden so that max_size can be customized
return NewlineFramer(max_size=MAX_INCOMING_MSG_SIZE) max_size = int(self.interface.network.config.get('network_max_incoming_msg_size',
MAX_INCOMING_MSG_SIZE))
return NewlineFramer(max_size=max_size)
class NetworkException(Exception): pass class NetworkException(Exception): pass
@ -605,7 +607,8 @@ class Interface(Logger):
return self.network.default_server == self.server return self.network.default_server == self.server
async def open_session(self, sslc, exit_early=False): async def open_session(self, sslc, exit_early=False):
async with _RSClient(session_factory=NotificationSession, session_factory = lambda *args, iface=self, **kwargs: NotificationSession(*args, **kwargs, interface=iface)
async with _RSClient(session_factory=session_factory,
host=self.host, port=self.port, host=self.host, port=self.port,
ssl=sslc, proxy=self.proxy) as session: ssl=sslc, proxy=self.proxy) as session:
self.session = session # type: NotificationSession self.session = session # type: NotificationSession