From eb01d81cdfc559737c15b112bbde9abe452e4bad Mon Sep 17 00:00:00 2001 From: earonesty Date: Wed, 25 Jan 2023 10:10:57 -0500 Subject: Update entity.py --- src/saml2/entity.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/saml2/entity.py b/src/saml2/entity.py index 9d0d2dcf..8016b481 100644 --- a/src/saml2/entity.py +++ b/src/saml2/entity.py @@ -3,6 +3,7 @@ from binascii import hexlify import copy from hashlib import sha1 import logging +import zlib import requests @@ -444,7 +445,10 @@ class Entity(HTTPBase): if binding == BINDING_HTTP_REDIRECT: xmlstr = decode_base64_and_inflate(txt) elif binding == BINDING_HTTP_POST: - xmlstr = base64.b64decode(txt) + try: + xmlstr = decode_base64_and_inflate(txt) + except zlib.error: + xmlstr = base64.b64decode(txt) elif binding == BINDING_SOAP: func = getattr(soap, f"parse_soap_enveloped_saml_{msgtype}") xmlstr = func(txt) -- cgit v1.2.1