summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorearonesty <erik@q32.com>2023-01-25 10:10:57 -0500
committerGitHub <noreply@github.com>2023-01-25 10:10:57 -0500
commiteb01d81cdfc559737c15b112bbde9abe452e4bad (patch)
treeaaf6bf59518ed2b973a395845eea2fff2c29089d
parent63b228837a27eda562c4335148c06c3ad0a89981 (diff)
downloadpysaml2-eb01d81cdfc559737c15b112bbde9abe452e4bad.tar.gz
Update entity.py
-rw-r--r--src/saml2/entity.py6
1 files changed, 5 insertions, 1 deletions
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)