Class LSBDataHeader

StegaPy/plugin/lsb/lsb_data_header.py:9–185

LSB数据头类，用于存储嵌入数据的元信息

class LSBDataHeader:
    """LSB数据头类，用于存储嵌入数据的元信息"""
    
    # 数据头标记（9字节）
    DATA_STAMP = b"STEGAPY  "  # 9字节，StegaPy项目标记
    HEADER_VERSION = b'\x02'  # 1字节，版本2
    FIXED_HEADER_LENGTH = 8  # 固定头长度
    CRYPT_ALGO_LENGTH = 8  # 加密算法名称长度
    MAX_FILENAME_LENGTH = 255  # 最大文件名长度
    
    def __init__(self, data_length=0, channel_bits_used=1, filename=None, config=None):
        """初始化数据头
        
        Args:
            data_length: 数据长度（不包括头）
            channel_bits_used: 每个颜色通道使用的位数
            filename: 文件名
            config: StegaPyConfig配置对象
        """
        self.data_length = data_length
        self.channel_bits_used = channel_bits_used
        self.filename = filename or ""
        self.config = config
        
        if len(self.filename.encode('utf-8')) > self.MAX_FILENAME_LENGTH:
            raise ValueError(f"文件名编码后长度不能超过{self.MAX_FILENAME_LENGTH}字节")
    
    def get_data_length(self):
        """获取数据长度"""
        return self.data_length
    
    def get_filename(self):
        """获取文件名"""
        return self.filename
    
    def get_channel_bits_used(self):
        """获取每个通道使用的位数"""
        return self.channel_bits_used
    
    def to_bytes(self):
        """转换为字节数组"""
        filename_bytes = self.filename.encode('utf-8')
        filename_len = len(filename_bytes)
        
        if filename_len > self.MAX_FILENAME_LENGTH:
            raise ValueError(f"文件名编码后长度超过{self.MAX_FILENAME_LENGTH}字节")
        
        # 构建头数据
        header = bytearray()
        
        # 1. DATA_STAMP (9字节)
        header.extend(self.DATA_STAMP)
        
        # 2. HEADER_VERSION (1字节)
        header.extend(self.HEADER_VERSION)
        
        # 3. FIXED_HEADER (8字节)
        # dataLength (4字节，小端序)