Method embed_data

StegaPy/plugin/lsb/lsb_plugin.py:37–61

嵌入数据到封面图像

(self, msg: bytes, msg_filename: Optional[str],
                   cover: Optional[bytes], cover_filename: Optional[str],
                   stego_filename: Optional[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: