Python 使用 PyPDF2 实现 PDF 加密与密码破解

Python 使用 PyPDF2 实现 PDF 加密与密码破解 | 极客日志

PyPDF2

pip install PyPDF2

python -m pip show PyPDF2

import PyPDF2

def encrypt_pdf(old_path, new_path, password='1234'):
    """
    给 PDF 文件添加密码保护
    
    :param old_path: 待加密文件的绝对路径
    :param new_path: 加密后文件的保存路径
    :param password: 设置的密码字符串
    """
    try:
        with open(old_path, 'rb') as pdf_file:
            # 创建 PdfFileReader 对象
            pdf_reader = PyPDF2.PdfFileReader(pdf_file)
            # 创建 PdfFileWriter 对象
            pdf_writer = PyPDF2.PdfFileWriter()
            
            # 将所有页面添加到 Writer 中
            for page_num in range(pdf_reader.numPages):
                pdf_writer.addPage(pdf_reader.getPage(page_num))
            
            # 设置密码并加密
            pdf_writer.encrypt(user_password=password)
            
            # 写入新文件
            with open(new_path, 'wb') as result_pdf:
                pdf_writer.write(result_pdf)
            
            print(f'加密成功！文件已保存至：{new_path}')
            return True
    except Exception as e:
        print(f'加密失败：{str(e)}')
        return False

if __name__ == '__main__':
    # 示例调用
    # encrypt_pdf('input.pdf', 'output_encrypted.pdf', '8888')
    pass

def generate_passwords(length=4):
    """
    生成指定位数的纯数字密码列表
    """
    passwords = []
    start = 10**(length-1) if length > 1 else 0
    end = 10**length
    for i in range(start, end):
        passwords.append(str(i).zfill(length))
    return passwords

def decrypt_pdf(target_path, max_attempts=10000):
    """
    尝试暴力破解 PDF 密码
    
    :param target_path: 目标加密 PDF 路径
    :param max_attempts: 最大尝试次数
    """:
    try:
        with open(target_path, 'rb') as pdf_file:
            pdf_reader = PyPDF2.PdfFileReader(pdf_file)
            
            # 检查文件是否加密
            if not pdf_reader.isEncrypted:
                print('文件未加密，无需破解。')
                return
            
            print('开始尝试破解密码...')
            
            # 遍历密码空间
            for i in range(max_attempts):
                pwd = str(i).zfill(4)
                
                # 尝试解密
                status = pdf_reader.decrypt(pwd)
                
                if status > 0:  # 状态大于 0 表示解密成功
                    print(f'破解成功！密码为：{pwd}')
                    
                    # 可选：输出解密后的内容或保存副本
                    pdf_writer = PyPDF2.PdfFileWriter()
                    for pageNum in range(pdf_reader.numPages):
                        pdf_writer.addPage(pdf_reader.getPage(pageNum))
                    
                    output_path = target_path.replace('.pdf', '_decrypted.pdf')
                    with open(output_path, 'wb') as out_file:
                        pdf_writer.write(out_file)
                    print(f'已保存解密文件至：{output_path}')
                    break
                else:
                    # 为了减少日志噪音，可定期打印进度
                    if i % 1000 == 0 and i != 0:
                        print(f'尝试了 {i} 个密码...')
            else:
                print('未找到匹配的密码。')
                
    except FileNotFoundError:
        print('错误：找不到指定的 PDF 文件。')
    except Exception as e:
        print(f'发生未知错误：{str(e)}')

if __name__ == '__main__':
    # 示例调用
    # decrypt_pdf('encrypted.pdf')
    pass

import PyPDF2
import os

class PDFFileManager:
    def __init__(self):
        self.reader = None
        self.writer = None

    def encrypt(self, input_path, output_path, password='1234'):
        """加密 PDF"""
        if not os.path.exists(input_path):
            print(f'文件不存在：{input_path}')
            return
        
        with open(input_path, 'rb') as f:
            reader = PyPDF2.PdfFileReader(f)
            writer = PyPDF2.PdfFileWriter()
            for page in range(reader.numPages):
                writer.addPage(reader.getPage(page))
            writer.encrypt(password)
            
        with open(output_path, 'wb') as f:
            writer.write(f)
        print(f'加密完成：{output_path}')

    def brute_force_decrypt(self, input_path, max_digits=4):
        """暴力破解密码"""
        if not os.path.exists(input_path):
            print(f'文件不存在：{input_path}')
            return

        with open(input_path, 'rb') as f:
            reader = PyPDF2.PdfFileReader(f)
            if not reader.isEncrypted:
                print('文件未加密')
                return

            total_combinations = 10 ** max_digits
            print(f'总组合数：{total_combinations}')

            for i in range(total_combinations):
                pwd = str(i).zfill(max_digits)
                if reader.decrypt(pwd) > 0:
                    print(f'成功！密码：{pwd}')
                    # 保存解密结果
                    writer = PyPDF2.PdfFileWriter()
                    for p in range(reader.numPages):
                        writer.addPage(reader.getPage(p))
                    out_name = input_path.replace('.pdf', '_unlocked.pdf')
                    with open(out_name, 'wb') as f_out:
                        writer.write(f_out)
                    return
            print('破解失败')

if __name__ == '__main__':
    manager = PDFFileManager()
    # 测试加密
    # manager.encrypt('test.pdf', 'test_encrypted.pdf', '1314')
    # 测试破解
    # manager.brute_force_decrypt('test_encrypted.pdf')
    pass

Python 使用 PyPDF2 实现 PDF 加密与密码破解

Python 使用 PyPDF2 实现 PDF 加密与密码破解

前言

环境准备

1. 安装依赖库

2. 代码结构说明

PDF 加密原理简述

实现 PDF 加密

实现 PDF 密码破解（暴力破解）

1. 密码生成逻辑

2. 解密与验证函数

完整整合代码

安全分析与最佳实践

1. 密码强度风险

2. 法律与道德警示

3. 优化建议

总结

更多推荐文章

相关免费在线工具

Python 使用 PyPDF2 实现 PDF 加密与密码破解

Python 使用 PyPDF2 实现 PDF 加密与密码破解

前言

环境准备

1. 安装依赖库

2. 代码结构说明

PDF 加密原理简述

实现 PDF 加密

实现 PDF 密码破解（暴力破解）

1. 密码生成逻辑

2. 解密与验证函数

完整整合代码

安全分析与最佳实践

1. 密码强度风险

2. 法律与道德警示

3. 优化建议

总结

微信扫一扫，关注极客日志

更多推荐文章

相关免费在线工具