Safety measure to prevent possible incorrect padding

Barski-lab · Nov 18, 2021 · ca14232 · ca14232
1 parent 2e75ef9
commit ca14232
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cwl_airflow/utilities/helpers.py b/cwl_airflow/utilities/helpers.py
@@ -82,13 +82,13 @@ def get_uncompressed(data_str, parse_as_yaml=None):
     try:
         uncompressed = gzip.decompress(
             base64.b64decode(
-                data_str.encode("utf-8")
+                data_str.encode("utf-8") + b'=='       # safety measure to prevent incorrect padding error
             )
         ).decode("utf-8")
     except Exception:
         uncompressed = zlib.decompress(
             base64.b64decode(
-                data_str.encode("utf-8")
+                data_str.encode("utf-8") + b'=='       # safety measure to prevent incorrect padding error
             )
         ).decode("utf-8")
     return load_yaml(uncompressed) if parse_as_yaml else uncompressed