摄影：产品经理
海鲜咖喱泡饭
之前在家里的老电脑中，发现一个加密zip压缩包，由于时隔太久忘记密码了，依稀记得密码是6位字母加数字，网上下载了很多破解密码的软件都没有效果，于是想到自己用Python写一个暴力破解密码的脚本。
Python有一个内置模块zipfile可以干这个事情，测试一波,一个测试文件，设置解压密码为123。
1import zipfile 2 3# 创建文件句柄 4file = zipfile.ZipFile("测试.zip", 'r') 5# 提取压缩文件中的内容，注意密码必须是bytes格式，path表示提取到哪 6file.extractall(path='.', pwd='123'.encode('utf-8')) 7
运行效果如下图所示，提取成功。
好了开始破解老文件的密码，为了提高速度我加了多线程最初的代码：
1import zipfile 2import itertools 3from concurrent.futures import ThreadPoolExecutor 4 5def extract(file, password): 6 if not flag: return 7 file.extractall(path='.', pwd=''.join(password).encode('utf-8')) 8 9 10def result(f): 11 exception = f.exception() 12 if not exception: 13 # 如果获取不到异常说明破解成功 14 print('密码为：', f.pwd) 15 global flag 16 flag = False 17 18 19if __name__ == '__main__': 20 # 创建一个标志用于判断密码是否破解成功 21 flag = True 22 # 创建一个线程池 23 pool = ThreadPoolExecutor(100) 24 nums = [str(i) for i in range(10)] 25 chrs = [chr(i) for i in range(65, 91)] 26 # 生成数字+字母的6位数密码 27 password_lst = itertools.permutations(nums + chrs, 6) 28 # 创建文件句柄 29 zfile = zipfile.ZipFile("加密文件.zip", 'r') 30 for pwd in password_lst: 31 if not flag: break 32 f = pool.submit(extract, zfile, pwd) 33 f.pwd = pwd 34 f.pool = pool 35 f.add_done_callback(result) 36
这个代码有个问题，跑一会儿内存就爆了！原因：ThreadPoolExecutor默认使用的是无界队列，尝试密码的速度跟不上生产密码的速度，会把生产任务无限添加到队列中。导致内存被占满。内存直接飙到95：
然后程序奔溃：
看了一下源码发现ThreadPoolExecutor内部使用的是无界队列，所以导致内存直接飙满，重写ThreadPoolExecutor类中的_work_queue属性，将无界队列改成有界队列，这样就不会出现内存爆满的问题，看代码：
1import queue 2from concurrent.futures import ThreadPoolExecutor 3 4 5class BoundedThreadPoolExecutor(ThreadPoolExecutor): 6 def __init__(self, max_workers=None, thread_name_prefix=''): 7 super().__init__(max_workers, thread_name_prefix) 8 self._work_queue = queue.Queue(self._max_workers * 2) # 设置队列大小 9 10
最后破解成功，如下图所示。
往期精彩文章推荐：
一篇文章带你解锁Python库中操作系统级别模块psutil （Python实战文）一篇文章教会你Arrow 时间库在项目中的实际应用 一篇文章带你了解Django ORM操作（高端篇）
欢迎各位大佬点击链接加入群聊【helloworld开发者社区】：https://jq.qq.com/?_wv=1027&k=mBlk6nzX进群交流IT技术热点。
本文转自 https://mp.weixin.qq.com/s/t6Ms8S8sSpU2MrlIOler-Q，如有侵权，请联系删除。