使用Python破解RAR文件密码的代码实例
from argparse import ArgumentParser
from itertools import chain, product
from os.path import exists
from string import digits, ascii_lowercase, ascii_uppercase, ascii_letters, printable
from subprocess import PIPE, Popen
from time import time
chars = (
# 默认的密码只来自数字 "0123456789"
digits
# 若需要更多的组合可加上如下
# 若要加上小写英文字母 "abcdefghijklmnopqrstuvwxyz" 的排列组合
# digits + ascii_lowercase
# 若要加上大小写英文字母 ascii_uppercase
# digits + ascii_lowercase + ascii_uppercase
# 若要加上标点符号和空白号，直接用string库下的 printable
# printable
)
special_chars = "();<>`|~"&'}]"
parser = ArgumentParser(description='Python combination generator to unrar')
parser.add_argument(
'--start',
help='Number of characters of the initial string [1 -> "a", 2 -> "aa"]',
type=int,
)
parser.add_argument(
'--stop',
help='Number of characters of the final string [3 -> "aaa"]',
type=int,
)
parser.add_argument(
'--verbose', help='Show combintations', default=False, required=False
)
parser.add_argument(
'--alphabet',
help='alternative chars to combinations',
default=chars,
required=False,
)
parser.add_argument('--file', help='.rar file [file.rar]', type=str)
args = parser.parse_args()
def generate_combinations(alphabet, length, start=1):
"""Generate combinations using alphabet."""
yield from (
''.join(string)
for string in chain.from_iterable(
product(alphabet, repeat=x) for x in range(start, length + 1)
)
)
def format(string):
"""Format chars to write them in shell."""
formated = map(
lambda char: char if char not in special_chars else f'\{char}', string
)
return ''.join(formated)
if __name__ == '__main__':
if not exists(args.file):
raise FileNotFoundError(args.file)
if args.stop < args.start:
raise Exception('Stop number is less than start')
start_time = time()
for combination in generate_combinations(args.alphabet, args.stop, args.start):
formated_combination = format(combination)
if args.verbose:
print(f'Trying: {combination}')
cmd = Popen(
f'unrar t -p{formated_combination} {args.file}'.split(),
stdout=PIPE,
stderr=PIPE,
)
out, err = cmd.communicate()
if 'All OK' in out.decode():
print(f'Password found: {combination}')
print(f'Time: {time() - start_time}')
exit()