michael/scrypted
1
0
Fork 0
mirror of https://github.com/koush/scrypted.git synced 2026-06-20 16:40:30 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

server: simplify convoluted peer key

Browse Source
This commit is contained in:
Koushik Dutta
2024-09-03 22:31:51 -07:00
parent ad15fe3324
commit 6d0da449ad
2 changed files with 53 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
server/python/plugin_remote.py
View File

@@ -601,21 +601,24 @@ class PluginRemote:
return clusterPeerPromise
async def connectClusterPeer():
reader, writer = await asyncio.open_connection(address, port)
sourceAddress, sourcePort = writer.get_extra_info("sockname")
if (
sourceAddress != SCRYPTED_CLUSTER_ADDRESS
and sourceAddress != "127.0.0.1"
):
print("source address mismatch", sourceAddress)
sourceKey = getClusterPeerKey(sourceAddress, sourcePort)
rpcTransport = rpc_reader.RpcStreamTransport(reader, writer)
clusterPeer, peerReadLoop = await rpc_reader.prepare_peer_readloop(
self.loop, rpcTransport
)
clusterPeer.onProxySerialization = lambda value: onProxySerialization(
value, sourceKey
)
try:
reader, writer = await asyncio.open_connection(address, port)
sourceAddress, sourcePort = writer.get_extra_info("sockname")
if (
sourceAddress != SCRYPTED_CLUSTER_ADDRESS
and sourceAddress != "127.0.0.1"
):
print("source address mismatch", sourceAddress)
rpcTransport = rpc_reader.RpcStreamTransport(reader, writer)
clusterPeer, peerReadLoop = await rpc_reader.prepare_peer_readloop(
self.loop, rpcTransport
)
clusterPeer.onProxySerialization = lambda value: onProxySerialization(
value, clusterPeerKey
)
except:
clusterPeers.pop(clusterPeerKey)
raise
async def run_loop():
try: