Python zipfile 库修复

Question

我有一个问题。我制作了一个简单的 zip 文件，密码为 12345。 现在，当我尝试使用蛮力提取密码时，zipfile 选择了错误的密码。它说它找到了密码 aaln0，但提取的文件完全是空的。有没有办法“修复”图书馆？或者有替代品吗？谢谢

程序代码：

#!/usr/bin/env python
import itertools
import threading
import argparse
import time
import zipfile
import string
global found
found = False

def extract_zip(zFile, password):
    """
    Extract archive with password
    """
    try:
        zFile.extractall(pwd=password)
        write("[+] Password found:", password, "\n")
        global found
        found = True
    except Exception, e:
        pass

def write(*args):
    print "[%s] %s" % (time.ctime(), " ".join(args))

def main_loop(zFile, length):
    """
    Main loop
    """
    write("[*] Python Brute-Force zip cracker")
    write("[*] Zipfile: %s; password length: %s" % (zFile, length))
    try:
        zfile = zipfile.ZipFile(zFile)
    except:
        write("Cannot open zip file")
        exit(1)
    for combo in itertools.imap(''.join, itertools.product(string.letters + string.digits,
                                                            repeat=length)):
        if found:
            break
        thread = threading.Thread(target=extract_zip, args=(zfile, combo))
        thread.start()
    if not found:
        write("[-] Password not found")

def main():
    """
    Main function
    """
    parser = argparse.ArgumentParser(usage="brute-force-zipcracker.py -f <zipfile> -l <password length>")
    parser.add_argument("-f", "--zipfile", help="specify zip file", type=str)
    parser.add_argument("-l", "--length", type=int, help="password length", default=5)
    args = parser.parse_args()
    if (args.zipfile == None):
        print parser.usage
        exit(0)

    main_loop(args.zipfile, args.length)

if __name__ == '__main__':
    main()

Answer 1

首先，你正在做：

for combo in itertools.imap(...):
    if found:
        break
    thread = ...
    thread.start()
if not found:
    ...

请看一下。
found 是在一个线程中定义的，但是您正在启动多个线程，并且您全局希望它将在其中一个线程中设置。 如何确保正确的线程完成了正确的工作？如果其中一个线程出现误报并且您不费心从每个单独的线程中检索值怎么办。小心你的线程！

其次， 如果线程没有及时完成您的combo 循环，您将最终进入if not found，因为线程尚未完成运行以找到您要查找的内容，特别是如果您有更大的 zip-需要几秒钟才能完成的文件（成功的密码将开始解压缩文件，这可能需要几分钟，并且在该过程完成之前不会设置found）。

最后得到用于保护这个 zip 文件的参数会很整洁。

编辑：

您也可以通过以下格式向我们提供更多信息：

zFile.debug(3)
zFile.testzip()
zFile.extractall(pwd=password)

还有来自zipfile.ZipInfo(filename)的其他有用的东西

然后去解决

#!/usr/bin/env python
import itertools
import argparse
import zipfile
import string

def extract_zip(filename, password):
    try:
        zFile = zipefile.ZipFile(filename)
        zFile.extractall(pwd=password)
        return True
    except zipfile.BadZipFile:
        return False

def main_loop(filename, length):
    print("[*] Python Brute-Force zip cracker")
    print("[*] Zipfile: %s; password length: %s" % (zFile, length))
    cracked = False
    for combo in itertools.imap(''.join, itertools.product(string.letters + string.digits, repeat=length)):
        cracked = extract_zip(filename, combo)
        if cracked:
            print('Yaay your password is:',combo)
            break
    if not cracked:
        print('Sorry, no luck..')
def main():
    parser = argparse.ArgumentParser(usage="brute-force-zipcracker.py -f <zipfile> -l <password length>")
    parser.add_argument("-f", "--zipfile", help="specify zip file", type=str)
    parser.add_argument("-l", "--length", type=int, help="password length", default=5)
    args = parser.parse_args()
    if (args.zipfile == None):
        print parser.usage
        exit(0)

    main_loop(args.zipfile, args.length)

if __name__ == '__main__':
    main()