Python 如何破解加密 zip 文件的密碼！

今天的文章來(lái)自【盞茶作酒】同學(xué)。這位同學(xué)在老電腦中發(fā)現(xiàn)了一個(gè)加密的 zip 文件，于是用 Python 破解了文件密碼。在破解的過(guò)程中出現(xiàn)了內(nèi)存爆炸的問(wèn)題，通過(guò)閱讀 Python 源代碼找到了解決方案。

之前在家里的老電腦中，發(fā)現(xiàn)一個(gè)加密zip壓縮包，由于時(shí)隔太久忘記密碼了，依稀記得密碼是6位字母加數(shù)字，網(wǎng)上下載了很多破解密碼的軟件都沒(méi)有效果，于是想到自己用Python寫一個(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)存直接飆滿，重寫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ì)列大小

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