You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
116 lines
3.5 KiB
116 lines
3.5 KiB
from PIL import Image
|
|
import PIL
|
|
import pyzbar.pyzbar as pyzbar
|
|
import otpauth_migration_decoder.src.decoder
|
|
import urllib.parse
|
|
import time
|
|
import qrcode
|
|
|
|
def decode_qr_file(img_file: str) -> str:
|
|
img = Image.open(img_file)
|
|
decoded = pyzbar.decode(img)
|
|
return decoded[0].data.decode("utf8")
|
|
|
|
def decode_screenshot_img() -> str:
|
|
from PIL import ImageGrab
|
|
img = ImageGrab.grab()
|
|
decoded = pyzbar.decode(img)
|
|
return decoded[0].data.decode("utf8")
|
|
|
|
def decode_totp(totp_uri: str) -> dict:
|
|
account_start = totp_uri.split("totp/")[1]
|
|
account = account_start.split("?")[0]
|
|
|
|
uri_params = account_start.split("?")[1]
|
|
|
|
try:
|
|
issuer_start = uri_params.split("issuer=")[1]
|
|
issuer = issuer_start.split("&")[0]
|
|
except:
|
|
issuer = ""
|
|
|
|
try:
|
|
secret_start = uri_params.split("secret=")[1]
|
|
secret = secret_start.split("&")[0]
|
|
except:
|
|
secret = ""
|
|
|
|
try:
|
|
algorithm_start = uri_params.split("algorithm=")[1]
|
|
algorithm = algorithm_start.split("&")[0]
|
|
except:
|
|
algorithm = ""
|
|
|
|
try:
|
|
digits_start = uri_params.split("digits=")[1]
|
|
digits = digits_start.split("&")[0]
|
|
except:
|
|
digits = ""
|
|
|
|
issuer = urllib.parse.unquote_plus(issuer)
|
|
secret = urllib.parse.unquote_plus(secret)
|
|
algorithm = urllib.parse.unquote_plus(algorithm)
|
|
account = urllib.parse.unquote_plus(account)
|
|
digits = urllib.parse.unquote_plus(digits)
|
|
|
|
return {"issuer": issuer, "account": account, "algorithm": algorithm, "secret": secret, "digits": digits}
|
|
|
|
def generate_new_url(totp_details: dict) -> str:
|
|
issuer = urllib.parse.quote(totp_details["issuer"])
|
|
account = urllib.parse.quote(totp_details["account"])
|
|
algorithm = urllib.parse.quote(totp_details["algorithm"])
|
|
secret = urllib.parse.quote(totp_details["secret"])
|
|
digits = urllib.parse.quote(totp_details["digits"])
|
|
|
|
generated_uri = "otpauth://totp/%s?algorithm=%s&digits=%s&issuer=%s&secret=%s" % (
|
|
account,
|
|
algorithm,
|
|
digits,
|
|
issuer,
|
|
secret
|
|
)
|
|
return generated_uri
|
|
|
|
def generate_qr_code(totp_details: dict):
|
|
generated_uri = generate_new_url(totp_details)
|
|
img = qrcode.make(generated_uri)
|
|
print(generated_uri)
|
|
img.show()
|
|
|
|
|
|
def main():
|
|
print("Grabbing screenshot in 2 seconds...")
|
|
time.sleep(2)
|
|
decoded_data = decode_screenshot_img()
|
|
print(decoded_data)
|
|
|
|
totp_uri_list: list = []
|
|
if "otpauth-migration" in decoded_data:
|
|
totp_uri_list = otpauth_migration_decoder.src.decoder.start_decode_migration(decoded_data)
|
|
elif "otpauth://totp/" in decoded_data:
|
|
totp_uri_list.append(decoded_data)
|
|
else:
|
|
print("Error in QR code. Please check if it is proper Google Authenticator QR code.")
|
|
return
|
|
|
|
for uri in totp_uri_list:
|
|
try:
|
|
curr_totp_details = decode_totp(uri)
|
|
print(curr_totp_details)
|
|
except:
|
|
print("Error in parsing the following URI: %s" % uri)
|
|
|
|
# Checking for each item
|
|
print()
|
|
print()
|
|
print("Do you want to change: (separator is \":::\")")
|
|
print("%s:::%s ... " % (curr_totp_details["issuer"], curr_totp_details["account"]), end="")
|
|
updated_data = input()
|
|
updated_data_splitted = updated_data.split(":::")
|
|
if updated_data != "":
|
|
curr_totp_details["issuer"] = updated_data_splitted[0]
|
|
curr_totp_details["account"] = updated_data_splitted[1]
|
|
generate_qr_code(curr_totp_details)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|