Method encrypt

StegaPy/util/crypto_util.py:55–85

加密数据

(self, data)

        return kdf.derive(password.encode())
    
    def encrypt(self, data):
        """加密数据"""
        try:
            # 生成随机IV
            iv = os.urandom(16)
            
            # 创建加密器
            if self.algorithm.startswith('AES'):
                cipher = Cipher(
                    algorithms.AES(self.key),
                    modes.CBC(iv),
                    backend=default_backend()
                )
            else:
                raise ValueError(f"不支持的算法: {self.algorithm}")
            
            encryptor = cipher.encryptor()
            
            # 填充数据
            padder = padding.PKCS7(128).padder()
            padded_data = padder.update(data)
            padded_data += padder.finalize()
            
            # 加密
            ciphertext = encryptor.update(padded_data) + encryptor.finalize()
            
            # 返回: IV长度(1字节) + IV + 加密数据
            result = bytes([len(iv)]) + iv + ciphertext
            return result
        except Exception as e:
            raise Exception(f"加密失败: {str(e)}")
    
    def decrypt(self, data):
        """解密数据"""