Calculate potfile from -o output for hashcat m22000

o2pot.py

#!/usr/bin/python3
# Calculate potfile from -o output for hashcat m22000
# Useful for -a9 mode, where the cracker, as of v6.2.6 cracker disables potfile output

import os
import sys
from hashlib import pbkdf2_hmac

if len(sys.argv) != 2 or not os.path.exists(sys.argv[1]):
    print(f'Usage: {sys.argv[0]} [hashcat.mac]')
    sys.exit(1)

with open(sys.argv[1], 'r', encoding='utf-8') as f:
    for line in f:
        line = line.rstrip('\r\n')

        arr = line.split(':', 4)
        if len(arr) != 5:
            continue

        if arr[4].startswith('$HEX[') and arr[4].endswith(']'):
            try:
                key = bytes.fromhex(arr[4][5:-1])
            except ValueError:
                continue
        else:
            key = arr[4].encode()

        print(f'{pbkdf2_hmac("sha1", key, arr[3].encode(), 4096, 32).hex()}*{arr[3].encode().hex()}:{arr[4]}')