add ipv6 support

use function to determine AF_INET or AF_INET6 for socket connections
CommunityHoneyNetwork · May 18, 2020 · 322b926 · 322b926
1 parent 2787c03
commit 322b926
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/hpfeeds/blocking/queue.py b/hpfeeds/blocking/queue.py
@@ -2,6 +2,7 @@
 
 import os
 import socket
+from .utils import get_inet_version
 
 try:
     import queue
@@ -24,10 +25,10 @@ def __init__(self):
         if os.name == 'posix':
             self._putsocket, self._getsocket = socket.socketpair()
         else:
-            server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-            server.bind(('127.0.0.1', 0))
+            server = socket.socket(get_inet_version('localhost'), socket.SOCK_STREAM)
+            server.bind(('localhost', 0))
             server.listen(1)
-            self._putsocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+            self._putsocket = socket.socket(get_inet_version(server.getsockname()[0]), socket.SOCK_STREAM)
             self._putsocket.connect(server.getsockname())
             self._getsocket, _ = server.accept()
             server.close()