Class LSBPlugin

StegaPy/plugin/lsb/lsb_plugin.py:16–151

LSB（最低有效位）隐写插件

class LSBPlugin(StegaPyPlugin):
    """LSB（最低有效位）隐写插件"""
    
    NAMESPACE = "LSB"
    
    def __init__(self, config: Optional[LSBConfig] = None):
        """初始化LSB插件"""
        super().__init__(config or LSBConfig())
    
    def get_name(self) -> str:
        """获取插件名称"""
        return "LSB"
    
    def get_purposes(self) -> List[Purpose]:
        """获取插件支持的用途"""
        return [Purpose.DATA_HIDING]
    
    def get_description(self) -> str:
        """获取插件描述"""
        return "LSB（最低有效位）隐写算法，将数据隐藏在图像像素的最低有效位中"
    
    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:
                # 计算需要的像素数
                from .lsb_data_header import LSBDataHeader
                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)
            
            # 使用LSB输出流嵌入数据
            lsb_os = LSBOutputStream(image, len(msg), msg_filename, self.config)
            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)
            lsb_is = LSBInputStream(image, self.config)
            return lsb_is.get_data_header().get_filename()
        except Exception as e:
            raise StegaPyException(str(e), StegaPyErrors.UNHANDLED_EXCEPTION, self.NAMESPACE)
    
    def extract_data(self, stego_data: bytes, stego_filename: Optional[str],