From 50db2424a6f0f148914ec06e4c13e46a7a5bc533 Mon Sep 17 00:00:00 2001
From: Andrew Stitcher <astitcher@apache.org>
Date: Fri, 12 Aug 2011 22:32:41 +0000
Subject: QPID-3407: Add IPv6 connections to python

git-svn-id: https://svn.apache.org/repos/asf/qpid/trunk@1157276 13f79535-47bb-0310-9956-ffa450edef68
---
 qpid/python/qpid/util.py | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

(limited to 'qpid/python')

diff --git a/qpid/python/qpid/util.py b/qpid/python/qpid/util.py
index e62bebdf24..82d5044264 100644
--- a/qpid/python/qpid/util.py
+++ b/qpid/python/qpid/util.py
@@ -39,12 +39,17 @@ except ImportError:
       self.sock.close()
 
 def connect(host, port):
-  sock = socket.socket()
-  sock.connect((host, port))
-  sock.setblocking(1)
-  # XXX: we could use this on read, but we'd have to put write in a
-  # loop as well
-  # sock.settimeout(1)
+  for res in socket.getaddrinfo(host, port, 0, socket.SOCK_STREAM):
+    af, socktype, proto, canonname, sa = res
+    sock = socket.socket(af, socktype, proto)
+    try:
+      sock.connect(sa)
+      break
+    except socket.error, msg:
+      sock.close
+  else:
+    # If we got here then we couldn't connect (yet)
+    raise
   return sock
 
 def listen(host, port, predicate = lambda: True, bound = lambda: None):
-- 
cgit v1.2.1