[[359077]]

 之前在家里的老電腦中，發現一個加密zip壓縮包，由于時隔太久忘記密碼了，依稀記得密碼是6位字母加數字，網上下載了很多破解密碼的軟件都沒有效果，于是想到自己用Python寫一個暴力破解密碼的腳本。

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) # 設置隊列大小 

最后破解成功，如下圖所示。