Merge pull request #4 from rtitmuss/check_remote_address

Drop udp packets from unknown hosts
yozik04 · May 31, 2022 · 566e793 · 566e793
2 parents e110209 + e45d007
commit 566e793
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/nibe/connection/nibegw.py b/nibe/connection/nibegw.py
@@ -61,6 +61,12 @@ def connection_made(self, transport):
         self._transport = transport
 
     def datagram_received(self, data, addr):
+        if addr[0] != self._remote_ip:
+            logger.warning(
+                f"Ignoring packet from unknown host: {addr}"
+            )
+            return
+
         logger.debug(f"Received {hexlify(data)} from {addr}")
         try:
             msg = Response.parse(data)