Python有一个内置模块zipfile可以干这个事情,测试一波,一个测试文件,设置解压密码为123。
- import zipfile
-
- # 创建文件句柄
- file = zipfile.ZipFile("测试.zip", 'r')
- # 提取压缩文件中的内容,注意密码必须是bytes格式,path表示提取到哪
- file.extractall(path='.', pwd='123'.encode('utf-8'))
运行效果如下图所示,提取成功。
好了开始破解老文件的密码,为了提高速度我加了多线程最初的代码:
- import zipfile
- import itertools
- from concurrent.futures import ThreadPoolExecutor
-
- def extract(file, password):
- if not flag: return
- file.extractall(path='.', pwd=''.join(password).encode('utf-8'))
-
-
- def result(f):
- exception = f.exception()
- if not exception:
- # 如果获取不到异常说明破解成功
- print('密码为:', f.pwd)
- global flag
- flag = False
-
-
- if __name__ == '__main__':
- # 创建一个标志用于判断密码是否破解成功
- flag = True
- # 创建一个线程池
- pool = ThreadPoolExecutor(100)
- nums = [str(i) for i in range(10)]
- chrs = [chr(i) for i in range(65, 91)]
- # 生成数字+字母的6位数密码
- password_lst = itertools.permutations(nums + chrs, 6)
- # 创建文件句柄
- zfile = zipfile.ZipFile("加密文件.zip", 'r')
- for pwd in password_lst:
- if not flag: break
- f = pool.submit(extract, zfile, pwd)
- f.pwd = pwd
- f.pool = pool
- f.add_done_callback(result)
这个代码有个问题,跑一会儿内存就爆了!原因:ThreadPoolExecutor默认使用的是无界队列,尝试密码的速度跟不上生产密码的速度,会把生产任务无限添加到队列中。导致内存被占满。内存直接飙到95:
然后程序奔溃:
看了一下源码发现ThreadPoolExecutor内部使用的是无界队列,所以导致内存直接飙满,重写ThreadPoolExecutor类中的_work_queue属性,将无界队列改成有界队列,这样就不会出现内存爆满的问题,看代码:
- import queue
- from concurrent.futures import ThreadPoolExecutor
-
-
- class BoundedThreadPoolExecutor(ThreadPoolExecutor):
- def __init__(self, max_workers=None, thread_name_prefix=''):
- super().__init__(max_workers, thread_name_prefix)
- self._work_queue = queue.Queue(self._max_workers * 2) # 设置队列大小
最后破解成功,如下图所示。