Mirrors/extract_otp_secrets
1
0
Fork 0
mirror of https://github.com/scito/extract_otp_secrets.git synced 2025-12-14 02:40:26 +01:00
Code Issues Packages Projects Releases 1 Wiki Activity

fix: improve error handling, abort on missing query parameter (data)

Browse Source
This commit is contained in:
scito
2021-02-13 16:54:12 +01:00
parent e43d9ae925
commit a7a6247f47
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
extract_otp_secret_keys.py
View File

@@ -87,8 +87,12 @@ def print_qr(data):
for line in (line.strip() for line in fileinput.input(args.infile)): for line in (line.strip() for line in fileinput.input(args.infile)):
if verbose: print(line) if verbose: print(line)
if line.startswith('#') or line == '': continue if line.startswith('#') or line == '': continue
if not line.startswith('otpauth-migration://'): print('\nWARN: line is not a otpauth-migration:// URL\ninput file: {}\nline "{}"\nProbably a wrong file was given'.format(args.infile, line))
parsed_url = urlparse(line) parsed_url = urlparse(line)
params = parse_qs(parsed_url.query) params = parse_qs(parsed_url.query)
if not 'data' in params:
print('\nERROR: no data query parameter in input URL\ninput file: {}\nline "{}"\nProbably a wrong file was given'.format(args.infile, line))
sys.exit(1)
data_encoded = params['data'][0] data_encoded = params['data'][0]
data = base64.b64decode(data_encoded) data = base64.b64decode(data_encoded)
payload = generated_python.google_auth_pb2.MigrationPayload() payload = generated_python.google_auth_pb2.MigrationPayload()