psn-go/internal/decoders/chunk.go

package decoders

import (
	"encoding/binary"
	"errors"
)

type ChunkHeader struct {
	Id           uint16
	DataLen      uint16
	HasSubchunks bool
}

type Chunk struct {
	ChunkData []byte
	Header    ChunkHeader
}

func DecodeChunk(bytes []byte) (Chunk, error) {

	if len(bytes) < 4 {
		return Chunk{}, errors.New("chunk must be at least 4 bytes")
	}

	id := binary.LittleEndian.Uint16(bytes[0:2])
	lengthAndFlag := binary.LittleEndian.Uint16(bytes[2:4])

	data_len := lengthAndFlag

	has_subchunks := lengthAndFlag > 32768

	if has_subchunks {
		data_len = data_len - 32768
	}

	header := ChunkHeader{
		Id:           id,
		DataLen:      data_len,
		HasSubchunks: has_subchunks,
	}

	chunk_data := bytes[4 : 4+header.DataLen]

	return Chunk{
		Header:    header,
		ChunkData: chunk_data,
	}, nil
}