Skip to content

Commit

Permalink
fix: Fix socket leak in impersonated_credentials
Browse files Browse the repository at this point in the history
impersonated_credentials.Credentials.sign_bytes() created
a session that wasn't closed leaking a socket. This fixes
the issue by always closing the requests session after
a signing request is complete.

Fixes googleapis#1122
  • Loading branch information
stewartmiles committed Aug 26, 2022
1 parent bb5c979 commit 2dcb6d3
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions google/auth/impersonated_credentials.py
Original file line number Diff line number Diff line change
Expand Up @@ -288,9 +288,12 @@ def sign_bytes(self, message):

authed_session = AuthorizedSession(self._source_credentials)

response = authed_session.post(
url=iam_sign_endpoint, headers=headers, json=body
)
try:
response = authed_session.post(
url=iam_sign_endpoint, headers=headers, json=body
)
finally:
authed_session.close()

if response.status_code != http_client.OK:
raise exceptions.TransportError(
Expand Down

0 comments on commit 2dcb6d3

Please sign in to comment.