Imported Upstream version 0.5.3

2 files changed, 14 insertions, 3 deletions

diff --git a/docker/client.py b/docker/client.py
index 1ff43cc..4ff91d8 100644
--- a/docker/client.py
+++ b/docker/client.py
@@ -259,9 +259,20 @@ class Client(requests.Session):
     def _get_raw_response_socket(self, response):
         self._raise_for_status(response)
         if six.PY3:
-            return response.raw._fp.fp.raw._sock
+            sock = response.raw._fp.fp.raw._sock
         else:
-            return response.raw._fp.fp._sock
+            sock = response.raw._fp.fp._sock
+        try:
+            # Keep a reference to the response to stop it being garbage
+            # collected. If the response is garbage collected, it will close
+            # TLS sockets.
+            sock._response = response
+        except AttributeError:
+            # UNIX sockets can't have attributes set on them, but that's fine
+            # because we won't be doing TLS over them
+            pass
+
+        return sock
 
     def _stream_helper(self, response):
         """Generator for data coming from a chunked-encoded HTTP response."""
diff --git a/docker/version.py b/docker/version.py
index abe1e1c..f4c4f3f 100644
--- a/docker/version.py
+++ b/docker/version.py
@@ -1 +1 @@
-version = "0.5.1"
+version = "0.5.3"