From ec1e9d8562dd09b7384502feffd78c0d12d9aeca Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=A9dric=20Krier?= <ced@b2ck.com>
Date: Mon, 25 Feb 2008 21:19:45 +0100
Subject: [PATCH] IPv6 compliant

---
 tryton/pysocket.py | 24 ++++++++++++++++--------
 tryton/rpc.py      |  4 ++--
 2 files changed, 18 insertions(+), 10 deletions(-)

diff --git a/tryton/pysocket.py b/tryton/pysocket.py
index d3738cc37e..4792274f0d 100644
--- a/tryton/pysocket.py
+++ b/tryton/pysocket.py
@@ -25,12 +25,7 @@ class PySocketException(Exception):
 class PySocket:
 
     def __init__(self, sock=None):
-        if sock is None:
-            self.sock = socket.socket(
-            socket.AF_INET, socket.SOCK_STREAM)
-        else:
-            self.sock = sock
-        self.sock.settimeout(120)
+        self.sock = sock
         self.host = None
         self.port = None
 
@@ -40,8 +35,18 @@ class PySocket:
             host, port = buf.split(':')
         if host in DNS_CACHE:
             host = DNS_CACHE[host]
+        if not self.sock:
+            familly = socket.AF_INET
+            if socket.has_ipv6:
+                try:
+                    socket.getaddrinfo(host, int(port), socket.AF_INET6)
+                    familly = socket.AF_INET6
+                except:
+                    pass
+            self.sock = socket.socket(familly, socket.SOCK_STREAM)
+            self.sock.settimeout(120)
         self.sock.connect((host, int(port)))
-        DNS_CACHE[host], port = self.sock.getpeername()
+        DNS_CACHE[host], port = self.sock.getpeername()[:2]
         self.host = host
         self.port = port
 
@@ -58,8 +63,11 @@ class PySocket:
     def reconnect(self):
         if self.host and self.port:
             self.disconnect()
+            familly = socket.AF_INET
+            if socket.has_ipv6:
+                familly = socket.AF_INET6
             self.sock = socket.socket(
-            socket.AF_INET, socket.SOCK_STREAM)
+                familly, socket.SOCK_STREAM)
             self.sock.settimeout(120)
             self.sock.connect((self.host, int(self.port)))
 
diff --git a/tryton/rpc.py b/tryton/rpc.py
index 186323a33f..d8cb39be95 100644
--- a/tryton/rpc.py
+++ b/tryton/rpc.py
@@ -271,7 +271,7 @@ class RPCSession(object):
         return 1
 
     def list_db(self, url):
-        match = re.match('^(http[s]?://|socket://)([\w.\-]+):(\d{1,5})$',
+        match = re.match('^(http[s]?://|socket://)([\w:.\-]+):(\d{1,5})$',
                 url or '')
         if not match:
             return -1
@@ -294,7 +294,7 @@ class RPCSession(object):
                 return -1
 
     def db_exec_no_except(self, url, method, *args):
-        match = re.match('^(http[s]?://|socket://)([\w.\-]+):(\d{1,5})$',
+        match = re.match('^(http[s]?://|socket://)([\w:.\-]+):(\d{1,5})$',
                 url or '')
         if match.group(1) == 'http://' or match.group(1) == 'https://':
             sock = xmlrpclib.ServerProxy(url + '/xmlrpc/db')