如何使用Python实现zip文件密码破解

小编给大家分享一下如何使用Python实现zip文件密码破解，希望大家阅读完这篇文章之后都有所收获，下面让我们一起去探讨吧！

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) # 设置队列大小

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

看完了这篇文章，相信你对“如何使用Python实现zip文件密码破解”有了一定的了解，如果想了解更多相关知识，欢迎关注亿速云行业资讯频道，感谢各位的阅读！