From 1bcfe74862e93519a9572b62e4396380326247d2 Mon Sep 17 00:00:00 2001
From: Ryan Summers <ryan.summers@vertigo-designs.com>
Date: Thu, 12 Oct 2023 12:21:20 +0200
Subject: [PATCH] Cleaning up impl after review

---
 py/stabilizer/stream.py | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/py/stabilizer/stream.py b/py/stabilizer/stream.py
index 99cd0b9fb..27d817f83 100644
--- a/py/stabilizer/stream.py
+++ b/py/stabilizer/stream.py
@@ -89,9 +89,17 @@ class StabilizerStream(asyncio.DatagramProtocol):
 
     @classmethod
     async def open(cls, addr, port, broker, maxsize=1):
-        """Open a UDP multicast socket and start receiving frames"""
+        """Open a UDP socket and start receiving frames"""
         loop = asyncio.get_running_loop()
         sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
+        sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+
+        # Increase the OS UDP receive buffer size to 4 MiB so that latency
+        # spikes don't impact much. Achieving 4 MiB may require increasing
+        # the max allowed buffer size, e.g. via
+        # `sudo sysctl net.core.rmem_max=26214400` but nowadays the default
+        # max appears to be ~ 50 MiB already.
+        sock.setsockopt(socket.SOL_SOCKET, socket.SO_RCVBUF, 4 << 20)
 
         # We need to specify which interface to receive broadcasts from, or Windows may choose the
         # wrong one. Thus, use the broker address to figure out our local address for the interface
@@ -105,14 +113,6 @@ async def open(cls, addr, port, broker, maxsize=1):
         sock.bind(('', port))
 
         transport, protocol = await loop.create_datagram_endpoint(lambda: cls(maxsize), sock=sock)
-        # Increase the OS UDP receive buffer size to 4 MiB so that latency
-        # spikes don't impact much. Achieving 4 MiB may require increasing
-        # the max allowed buffer size, e.g. via
-        # `sudo sysctl net.core.rmem_max=26214400` but nowadays the default
-        # max appears to be ~ 50 MiB already.
-        sock = transport.get_extra_info("socket")
-        if sock is not None:
-            sock.setsockopt(socket.SOL_SOCKET, socket.SO_RCVBUF, 4 << 20)
         return transport, protocol
 
     def __init__(self, maxsize):