Replace base64 decoding errors with BadData.

pallets · Apr 13, 2016 · ec69312 · ec69312
1 parent f1ecd77
commit ec69312
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/itsdangerous.py b/itsdangerous.py
@@ -207,7 +207,12 @@ def base64_decode(string):
     The result is also a bytestring.
     """
     string = want_bytes(string, encoding='ascii', errors='ignore')
-    return base64.urlsafe_b64decode(string + b'=' * (-len(string) % 4))
+    string += b'=' * (-len(string) % 4)
+
+    try:
+        return base64.urlsafe_b64decode(string)
+    except (TypeError, ValueError):
+        raise BadData('Invalid base64-encoded data')
 
 
 _int64_struct = struct.Struct('>Q')