From 8f28e4fb4da7f43c3165a5af9c19dca94656cb92 Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Thu, 23 Jan 2025 09:44:15 -1000 Subject: [PATCH] guards for no ipv6 --- wsdiscovery/threaded.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/wsdiscovery/threaded.py b/wsdiscovery/threaded.py index 8b4b5e7..21c7ff9 100644 --- a/wsdiscovery/threaded.py +++ b/wsdiscovery/threaded.py @@ -433,14 +433,14 @@ def addSourceAddr(self, addr): version = ipaddress.ip_address(addr).version if version == 4: self._networkingThread_v4.addSourceAddr(addr) - elif version == 6: + elif version == 6 and self._networkingThread_v6 is not None: self._networkingThread_v6.addSourceAddr(addr) def removeSourceAddr(self, addr): version = ipaddress.ip_address(addr).version if version == 4: self._networkingThread_v4.removeSourceAddr(addr) - elif version == 6: + elif version == 6 and self._networkingThread_v6 is not None: self._networkingThread_v6.removeSourceAddr(addr) def sendUnicastMessage(self, env, host, port, initialDelay=0, @@ -448,8 +448,9 @@ def sendUnicastMessage(self, env, host, port, initialDelay=0, "handle unicast message sending" self._networkingThread_v4.addUnicastMessage(env, host, port, initialDelay, unicast_num) - self._networkingThread_v6.addUnicastMessage(env, host, port, - initialDelay, unicast_num) + if self._networkingThread_v6 is not None: + self._networkingThread_v6.addUnicastMessage(env, host, port, + initialDelay, unicast_num) def sendMulticastMessage(self, env, initialDelay=0, multicast_num=MULTICAST_UDP_REPEAT): @@ -459,7 +460,8 @@ def sendMulticastMessage(self, env, initialDelay=0, MULTICAST_PORT, initialDelay, multicast_num) - self._networkingThread_v6.addMulticastMessage(env, + if self._networkingThread_v6 is not None: + self._networkingThread_v6.addMulticastMessage(env, MULTICAST_IPV6_ADDRESS, MULTICAST_PORT, initialDelay,