[WIP] py3 error

JuryA · Apr 9, 2019 · c87d014 · c87d014
1 parent 68f6bca
commit c87d014
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/Windows/lazagne/softwares/browsers/chromium_based.py b/Windows/lazagne/softwares/browsers/chromium_based.py
@@ -1,5 +1,4 @@
 # -*- coding: utf-8 -*- 
-import ast
 import base64
 import json
 import os
@@ -90,7 +89,11 @@ def _export_credentials(self, db_path, is_yandex=False):
                 # Yandex passwords use a masterkey stored on windows credential manager
                 # https://yandex.com/support/browser-passwords-crypto/without-master.html
                 if is_yandex and yandex_enckey:
-                    p = ast.literal_eval(str(password))
+                    try:
+                        p = json.loads(str(password))
+                    except Exception:
+                        p = json.loads(password)
+
                     password = base64.b64decode(p['p'])
 
                     # Passwords are stored using AES-256-GCM algorithm
@@ -101,7 +104,8 @@ def _export_credentials(self, db_path, is_yandex=False):
                     # Failed...
                 else:
                     # Decrypt the Password
-                    password = Win32CryptUnprotectData(password, is_current_user=constant.is_current_user, user_dpapi=constant.user_dpapi)
+                    password = Win32CryptUnprotectData(password, is_current_user=constant.is_current_user,
+                                                       user_dpapi=constant.user_dpapi)
 
                 if not url and not login and not password:
                     continue