Update libmilter.py

diciu · diciu · commit 712bff813ba3 · 2015-12-27T10:42:54.000+02:00
Uses getaddrinfo to allow both IPv4 and IPv6 connections.
diff --git a/libmilter.py b/libmilter.py
@@ -1264,8 +1264,10 @@ def unregister(self , fileno):
     def run(self):
         global DEFERRED_REG
         if self.sockStr.lower().startswith('inet:'):
-            junk , ip , port = self.sockStr.split(':')
-            self.sock = socket.socket(socket.AF_INET , socket.SOCK_STREAM)
+            ip = self.sockStr[5:self.sockStr.rfind(':')]
+            port = self.sockStr[self.sockStr.rfind(':')+1:]
+            (family, socktype, proto, canonname, sockaddr)=socket.getaddrinfo(ip, None)[0]
+            self.sock = socket.socket(family , socket.SOCK_STREAM)
             self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
             self.sock.bind((ip , int(port)))
         else:
