Class RandomLSBPlugin

StegaPy/plugin/randlsb/random_lsb_plugin.py:17–154

RandomLSB（随机最低有效位）隐写插件

class RandomLSBPlugin(StegaPyPlugin):
    """RandomLSB（随机最低有效位）隐写插件"""
    
    NAMESPACE = "RandomLSB"
    
    def __init__(self, config: Optional[LSBConfig] = None):
        """初始化RandomLSB插件"""
        super().__init__(config or LSBConfig())
    
    def get_name(self) -> str:
        """获取插件名称"""
        return "RandomLSB"
    
    def get_purposes(self) -> List[Purpose]:
        """获取插件支持的用途"""
        return [Purpose.DATA_HIDING]
    
    def get_description(self) -> str:
        """获取插件描述"""
        return "RandomLSB（随机最低有效位）隐写算法，使用随机序列将数据隐藏在图像像素中，提供更好的安全性"
    
    def embed_data(self, msg: bytes, msg_filename: Optional[str],
                   cover: Optional[bytes], cover_filename: Optional[str],
                   stego_filename: Optional[str]) -> bytes:
        """嵌入数据到封面图像"""
        try:
            # 如果没有提供封面图像，生成随机图像
            if cover is None:
                header_size = LSBDataHeader.get_max_header_size()
                num_pixels = int(header_size * 8 / 3.0)
                num_pixels += int(len(msg) * 8 / (3.0 * self.config.get_max_bits_used_per_channel()))
                image = ImageUtil.generate_random_image(num_pixels)
            else:
                image = ImageUtil.byte_array_to_image(cover, cover_filename)
            
            # 获取密码（如果有）
            password = self.config.get_password() if self.config else None
            
            # 使用RandomLSB输出流嵌入数据
            lsb_os = RandomLSBOutputStream(image, len(msg), msg_filename, self.config, password)
            lsb_os.write(msg)
            lsb_os.flush()
            image = lsb_os.get_image()
            
            return ImageUtil.image_to_byte_array(image, stego_filename)
        except Exception as e:
            raise StegaPyException(str(e), StegaPyErrors.UNHANDLED_EXCEPTION, self.NAMESPACE)
    
    def extract_msg_filename(self, stego_data: bytes,
                             stego_filename: Optional[str]) -> str:
        """从隐写数据中提取消息文件名"""
        try:
            image = ImageUtil.byte_array_to_image(stego_data, stego_filename)
            password = self.config.get_password() if self.config else None
            lsb_is = RandomLSBInputStream(image, self.config, password)
            return lsb_is.get_data_header().get_filename()
        except Exception as e:
            raise StegaPyException(str(e), StegaPyErrors.UNHANDLED_EXCEPTION, self.NAMESPACE)