之前在家里的老電腦中，發(fā)現(xiàn)一個(gè)加密zip壓縮包，由于時(shí)隔太久忘記密碼了，依稀記得密碼是6位字母加數(shù)字，網(wǎng)上下載了很多破解密碼的軟件都沒(méi)有效果，于是想到自己用Python寫(xiě)一個(gè)暴力破解密碼的腳本。

Python有一個(gè)內(nèi)置模塊zipfile可以干這個(gè)事情，測(cè)試一波,一個(gè)測(cè)試文件，設(shè)置解壓密碼為123。

import zipfile 
 
# 創(chuàng)建文件句柄 
file = zipfile.ZipFile("測(cè)試.zip", 'r') 
# 提取壓縮文件中的內(nèi)容，注意密碼必須是bytes格式，path表示提取到哪 
file.extractall(path='.', pwd='123'.encode('utf-8')) 

運(yùn)行效果如下圖所示，提取成功。

好了開(kāi)始破解老文件的密碼，為了提高速度我加了多線程最初的代碼：

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: 
        # 如果獲取不到異常說(shuō)明破解成功 
        print('密碼為：', f.pwd) 
        global flag 
        flag = False 
 
 
if __name__ == '__main__': 
    # 創(chuàng)建一個(gè)標(biāo)志用于判斷密碼是否破解成功 
    flag = True 
    # 創(chuàng)建一個(gè)線程池 
    pool = ThreadPoolExecutor(100) 
    nums = [str(i) for i in range(10)] 
    chrs = [chr(i) for i in range(65, 91)] 
    # 生成數(shù)字+字母的6位數(shù)密碼 
    password_lst = itertools.permutations(nums + chrs, 6) 
    # 創(chuàng)建文件句柄 
    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) 

這個(gè)代碼有個(gè)問(wèn)題，跑一會(huì)兒內(nèi)存就爆了!原因：ThreadPoolExecutor默認(rèn)使用的是無(wú)界隊(duì)列，嘗試密碼的速度跟不上生產(chǎn)密碼的速度，會(huì)把生產(chǎn)任務(wù)無(wú)限添加到隊(duì)列中。導(dǎo)致內(nèi)存被占滿。內(nèi)存直接飆到95：

然后程序奔潰：

看了一下源碼發(fā)現(xiàn)ThreadPoolExecutor內(nèi)部使用的是無(wú)界隊(duì)列，所以導(dǎo)致內(nèi)存直接飆滿，重寫(xiě)ThreadPoolExecutor類中的_work_queue屬性，將無(wú)界隊(duì)列改成有界隊(duì)列，這樣就不會(huì)出現(xiàn)內(nèi)存爆滿的問(wèn)題，看代碼：

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) # 設(shè)置隊(duì)列大小 

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