Bug fix for the different between key file and hex key

This commit is contained in:
KnugiHK
2023-12-17 14:40:08 +08:00
parent 3d155fb48f
commit 481656fdeb

View File

@@ -1,5 +1,6 @@
#!/usr/bin/python3 #!/usr/bin/python3
import io
import os import os
import sqlite3 import sqlite3
import shutil import shutil
@@ -267,7 +268,8 @@ def main():
if args.wab: if args.wab:
wab = open(args.wab, "rb").read() wab = open(args.wab, "rb").read()
error_wa = extract.decrypt_backup(wab, key, contact_db, crypt, args.showkey, DbType.CONTACT) error_wa = extract.decrypt_backup(wab, key, contact_db, crypt, args.showkey, DbType.CONTACT)
key.seek(0) if isinstance(key, io.IOBase):
key.seek(0)
else: else:
error_wa = 0 error_wa = 0
error_message = extract.decrypt_backup(db, key, msg_db, crypt, args.showkey, DbType.MESSAGE) error_message = extract.decrypt_backup(db, key, msg_db, crypt, args.showkey, DbType.MESSAGE)