Skip to content

Commit 480d60e

Browse files
committed
PYTHON-3017 Properly check for closed KMS connections (#790)
(cherry picked from commit 99aab1b)
1 parent 54c87ba commit 480d60e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

pymongo/encryption.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,6 +130,8 @@ def kms_request(self, kms_context):
130130
conn.sendall(message)
131131
while kms_context.bytes_needed > 0:
132132
data = conn.recv(kms_context.bytes_needed)
133+
if not data:
134+
raise OSError('KMS connection closed')
133135
kms_context.feed(data)
134136
finally:
135137
conn.close()

0 commit comments

Comments
 (0)