BUUCTF刷题MISC[九] (89-96)

刷题随笔，不足的地方望大佬指正

import zipfile
import os
import shutil


def extract_zip_layer(zip_path, extract_dir):
    """
    解压单个ZIP文件（密码=文件名），返回解压后新的ZIP文件路径
    :param zip_path: 当前要解压的ZIP文件路径
    :param extract_dir: 解压目录
    :return: 新的ZIP文件路径/None（无新ZIP时）
    """
    # 1. 获取ZIP文件名（不含后缀）作为密码
    zip_name = os.path.splitext(os.path.basename(zip_path))[0]
    password = zip_name.encode('utf-8')  # ZIP密码需为bytes格式

    # 2. 创建解压目录（避免重复）
    os.makedirs(extract_dir, exist_ok=True)

    try:
        # 3. 解压ZIP文件（密码=文件名）
        with zipfile.ZipFile(zip_path, 'r') as zf:
            # 设置密码并解压所有文件
            zf.extractall(pwd=password, path=extract_dir)
            print(f" 解压成功：{zip_path} | 密码：{zip_name}")

            # 4. 查找解压后目录中的ZIP文件（找下一层）
            new_zip_path = None
            for root, dirs, files in os.walk(extract_dir):
                for file in files:
                    if file.lower().endswith('.zip'):
                        new_zip_path = os.path.join(root, file)
                        # 找到第一个ZIP就返回（千层ZIP通常只有一个下一层）
                        return new_zip_path
            # 无新ZIP文件，返回None
            return None
    except zipfile.BadZipFile:
        print(f" 错误：{zip_path} 不是有效的ZIP文件")
        return None
    except RuntimeError as e:
        if "password" in str(e).lower():
            print(f" 密码错误：{zip_path} | 尝试密码：{zip_name}")
        else:
            print(f" 解压失败：{zip_path} | 原因：{e}")
        return None
    except Exception as e:
        print(f" 处理失败：{zip_path} | 原因：{e}")
        return None


def extract_all_layers(initial_zip, base_extract_dir="zip_extract"):
    """
    递归解压千层ZIP，直到无新ZIP文件
    :param initial_zip: 初始ZIP文件路径
    :param base_extract_dir: 基础解压目录
    """
    current_zip = initial_zip
    layer = 1  # 记录解压层数

    # 检查初始ZIP是否存在
    if not os.path.exists(current_zip):
        print(f" 初始ZIP文件不存在：{current_zip}")
        return

    print(f"===== 开始解压千层ZIP（初始文件：{current_zip}）=====")

    while current_zip:
        # 每层解压到独立目录（避免文件覆盖）
        current_extract_dir = os.path.join(base_extract_dir, f"layer_{layer}")
        # 解压当前层，获取下一层ZIP路径
        next_zip = extract_zip_layer(current_zip, current_extract_dir)

        # 清理：将下一层ZIP移到基础目录（方便后续处理），避免多层嵌套查找
        if next_zip:
            new_next_zip = os.path.join(base_extract_dir, os.path.basename(next_zip))
            shutil.move(next_zip, new_next_zip)
            current_zip = new_next_zip
            layer += 1
            print(f" 进入下一层（第{layer}层）：{current_zip}")
        else:
            print(f"\n 解压完成！共解压 {layer} 层，最终文件在：{current_extract_dir}")
            current_zip = None


# ===================== 脚本运行入口 =====================
if __name__ == "__main__":
    # 配置：修改为你的初始ZIP文件路径
    INITIAL_ZIP_PATH = r"E:\attachment\0573.zip"

    # 开始递归解压
    extract_all_layers(INITIAL_ZIP_PATH)

    # 防止窗口闪退
    input("\n按回车键退出...")

打开是一个文本