From 97a661e39c1c7b67fda105102bef52b830620cc3 Mon Sep 17 00:00:00 2001
From: Dana Powers <dana.powers@gmail.com>
Date: Fri, 3 Mar 2017 12:08:49 -0800
Subject: Catch socket errors during ssl handshake

---
 kafka/conn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'kafka/conn.py')

diff --git a/kafka/conn.py b/kafka/conn.py
index ae3c55d..a678348 100644
--- a/kafka/conn.py
+++ b/kafka/conn.py
@@ -381,7 +381,7 @@ class BrokerConnection(object):
         # old ssl in python2.6 will swallow all SSLErrors here...
         except (SSLWantReadError, SSLWantWriteError):
             pass
-        except SSLZeroReturnError:
+        except (SSLZeroReturnError, ConnectionError):
             log.warning('SSL connection closed by server during handshake.')
             self.close(Errors.ConnectionError('SSL connection closed by server during handshake'))
         # Other SSLErrors will be raised to user
-- 
cgit v1.2.1