aboutsummaryrefslogtreecommitdiff
path: root/packfile/ingest/record_inflate.go
blob: b8eca25b59f2cf488e235d93250208d2cda56785 (about) (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
package ingest

import (
	"compress/zlib"
	"fmt"
	"io"

	"codeberg.org/lindenii/furgit/internal/intconv"
)

// inflateRecordPayload inflates one record's zlib payload from pack file.
func inflateRecordPayload(state *ingestState, idx int) ([]byte, error) {
	record := state.records[idx]
	if record.packedLen < uint64(record.headerLen) {
		return nil, &MalformedPackEntryError{Offset: record.offset, Reason: "entry packed span underflow"}
	}

	compressedOffset := record.offset + uint64(record.headerLen)
	compressedLen := record.packedLen - uint64(record.headerLen)

	compressedOffsetInt64, err := intconv.Uint64ToInt64(compressedOffset)
	if err != nil {
		return nil, err
	}

	compressedLenInt64, err := intconv.Uint64ToInt64(compressedLen)
	if err != nil {
		return nil, err
	}

	section := io.NewSectionReader(state.packFile, compressedOffsetInt64, compressedLenInt64)

	reader, err := zlib.NewReader(section)
	if err != nil {
		return nil, &MalformedPackEntryError{Offset: record.offset, Reason: fmt.Sprintf("open payload zlib: %v", err)}
	}

	defer func() { _ = reader.Close() }()

	out, err := io.ReadAll(reader)
	if err != nil {
		return nil, &MalformedPackEntryError{Offset: record.offset, Reason: fmt.Sprintf("inflate payload: %v", err)}
	}

	return out, nil
}