如何使用Python實現(xiàn)zip文件密碼破解

小編給大家分享一下如何使用Python實現(xiàn)zip文件密碼破解，希望大家閱讀完這篇文章之后都有所收獲，下面讓我們一起去探討吧！

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

import zipfile
 
# 創(chuàng)建文件句柄
file = zipfile.ZipFile("測試.zip", 'r')
# 提取壓縮文件中的內(nèi)容，注意密碼必須是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__':
    # 創(chuàng)建一個標志用于判斷密碼是否破解成功
    flag = True
    # 創(chuàng)建一個線程池
    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)

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

然后程序奔潰：

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

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

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

看完了這篇文章，相信你對“如何使用Python實現(xiàn)zip文件密碼破解”有了一定的了解，如果想了解更多相關(guān)知識，歡迎關(guān)注億速云行業(yè)資訊頻道，感謝各位的閱讀！