Merge pull request #48 from mitmproxy/mitmproxy_issue_341

Adjust netlib to mitmproxy changes

1 files changed, 8 insertions, 0 deletions

diff --git a/netlib/tcp.py b/netlib/tcp.py
index a5b9af22..2704eeae 100644
--- a/netlib/tcp.py
+++ b/netlib/tcp.py
@@ -216,10 +216,16 @@ class Address(object):
     def use_ipv6(self, b):
         self.family = socket.AF_INET6 if b else socket.AF_INET
 
+    def __repr__(self):
+        return repr(self.address)
+
     def __eq__(self, other):
         other = Address.wrap(other)
         return (self.address, self.family) == (other.address, other.family)
 
+    def __ne__(self, other):
+        return not self.__eq__(other)
+
 
 class _Connection(object):
     def get_current_cipher(self):
@@ -313,6 +319,8 @@ class TCPClient(_Connection):
             if self.source_address:
                 connection.bind(self.source_address())
             connection.connect(self.address())
+            if not self.source_address:
+                self.source_address = Address(connection.getsockname())
             self.rfile = Reader(connection.makefile('rb', self.rbufsize))
             self.wfile = Writer(connection.makefile('wb', self.wbufsize))
         except (socket.error, IOError), err: