Method embed_data

StegaPy/plugin/randlsb/random_lsb_plugin.py:38–63

嵌入数据到封面图像

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