792b4dba2c
* update github.com/blevesearch/bleve v2.0.2 -> v2.0.3 * github.com/denisenkom/go-mssqldb v0.9.0 -> v0.10.0 * github.com/editorconfig/editorconfig-core-go v2.4.1 -> v2.4.2 * github.com/go-chi/cors v1.1.1 -> v1.2.0 * github.com/go-git/go-billy v5.0.0 -> v5.1.0 * github.com/go-git/go-git v5.2.0 -> v5.3.0 * github.com/go-ldap/ldap v3.2.4 -> v3.3.0 * github.com/go-redis/redis v8.6.0 -> v8.8.2 * github.com/go-sql-driver/mysql v1.5.0 -> v1.6.0 * github.com/go-swagger/go-swagger v0.26.1 -> v0.27.0 * github.com/lib/pq v1.9.0 -> v1.10.1 * github.com/mattn/go-sqlite3 v1.14.6 -> v1.14.7 * github.com/go-testfixtures/testfixtures v3.5.0 -> v3.6.0 * github.com/issue9/identicon v1.0.1 -> v1.2.0 * github.com/klauspost/compress v1.11.8 -> v1.12.1 * github.com/mgechev/revive v1.0.3 -> v1.0.6 * github.com/microcosm-cc/bluemonday v1.0.7 -> v1.0.8 * github.com/niklasfasching/go-org v1.4.0 -> v1.5.0 * github.com/olivere/elastic v7.0.22 -> v7.0.24 * github.com/pelletier/go-toml v1.8.1 -> v1.9.0 * github.com/prometheus/client_golang v1.9.0 -> v1.10.0 * github.com/xanzy/go-gitlab v0.44.0 -> v0.48.0 * github.com/yuin/goldmark v1.3.3 -> v1.3.5 * github.com/6543/go-version v1.2.4 -> v1.3.1 * do github.com/lib/pq v1.10.0 -> v1.10.1 again ...
146 lines
4.2 KiB
Go
Vendored
146 lines
4.2 KiB
Go
Vendored
// Package zstd provides decompression of zstandard files.
|
|
//
|
|
// For advanced usage and examples, go to the README: https://github.com/klauspost/compress/tree/master/zstd#zstd
|
|
package zstd
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/binary"
|
|
"errors"
|
|
"log"
|
|
"math"
|
|
"math/bits"
|
|
)
|
|
|
|
// enable debug printing
|
|
const debug = false
|
|
|
|
// Enable extra assertions.
|
|
const debugAsserts = debug || false
|
|
|
|
// print sequence details
|
|
const debugSequences = false
|
|
|
|
// print detailed matching information
|
|
const debugMatches = false
|
|
|
|
// force encoder to use predefined tables.
|
|
const forcePreDef = false
|
|
|
|
// zstdMinMatch is the minimum zstd match length.
|
|
const zstdMinMatch = 3
|
|
|
|
// Reset the buffer offset when reaching this.
|
|
const bufferReset = math.MaxInt32 - MaxWindowSize
|
|
|
|
var (
|
|
// ErrReservedBlockType is returned when a reserved block type is found.
|
|
// Typically this indicates wrong or corrupted input.
|
|
ErrReservedBlockType = errors.New("invalid input: reserved block type encountered")
|
|
|
|
// ErrCompressedSizeTooBig is returned when a block is bigger than allowed.
|
|
// Typically this indicates wrong or corrupted input.
|
|
ErrCompressedSizeTooBig = errors.New("invalid input: compressed size too big")
|
|
|
|
// ErrBlockTooSmall is returned when a block is too small to be decoded.
|
|
// Typically returned on invalid input.
|
|
ErrBlockTooSmall = errors.New("block too small")
|
|
|
|
// ErrMagicMismatch is returned when a "magic" number isn't what is expected.
|
|
// Typically this indicates wrong or corrupted input.
|
|
ErrMagicMismatch = errors.New("invalid input: magic number mismatch")
|
|
|
|
// ErrWindowSizeExceeded is returned when a reference exceeds the valid window size.
|
|
// Typically this indicates wrong or corrupted input.
|
|
ErrWindowSizeExceeded = errors.New("window size exceeded")
|
|
|
|
// ErrWindowSizeTooSmall is returned when no window size is specified.
|
|
// Typically this indicates wrong or corrupted input.
|
|
ErrWindowSizeTooSmall = errors.New("invalid input: window size was too small")
|
|
|
|
// ErrDecoderSizeExceeded is returned if decompressed size exceeds the configured limit.
|
|
ErrDecoderSizeExceeded = errors.New("decompressed size exceeds configured limit")
|
|
|
|
// ErrUnknownDictionary is returned if the dictionary ID is unknown.
|
|
// For the time being dictionaries are not supported.
|
|
ErrUnknownDictionary = errors.New("unknown dictionary")
|
|
|
|
// ErrFrameSizeExceeded is returned if the stated frame size is exceeded.
|
|
// This is only returned if SingleSegment is specified on the frame.
|
|
ErrFrameSizeExceeded = errors.New("frame size exceeded")
|
|
|
|
// ErrCRCMismatch is returned if CRC mismatches.
|
|
ErrCRCMismatch = errors.New("CRC check failed")
|
|
|
|
// ErrDecoderClosed will be returned if the Decoder was used after
|
|
// Close has been called.
|
|
ErrDecoderClosed = errors.New("decoder used after Close")
|
|
|
|
// ErrDecoderNilInput is returned when a nil Reader was provided
|
|
// and an operation other than Reset/DecodeAll/Close was attempted.
|
|
ErrDecoderNilInput = errors.New("nil input provided as reader")
|
|
)
|
|
|
|
func println(a ...interface{}) {
|
|
if debug {
|
|
log.Println(a...)
|
|
}
|
|
}
|
|
|
|
func printf(format string, a ...interface{}) {
|
|
if debug {
|
|
log.Printf(format, a...)
|
|
}
|
|
}
|
|
|
|
// matchLenFast does matching, but will not match the last up to 7 bytes.
|
|
func matchLenFast(a, b []byte) int {
|
|
endI := len(a) & (math.MaxInt32 - 7)
|
|
for i := 0; i < endI; i += 8 {
|
|
if diff := load64(a, i) ^ load64(b, i); diff != 0 {
|
|
return i + bits.TrailingZeros64(diff)>>3
|
|
}
|
|
}
|
|
return endI
|
|
}
|
|
|
|
// matchLen returns the maximum length.
|
|
// a must be the shortest of the two.
|
|
// The function also returns whether all bytes matched.
|
|
func matchLen(a, b []byte) int {
|
|
b = b[:len(a)]
|
|
for i := 0; i < len(a)-7; i += 8 {
|
|
if diff := load64(a, i) ^ load64(b, i); diff != 0 {
|
|
return i + (bits.TrailingZeros64(diff) >> 3)
|
|
}
|
|
}
|
|
|
|
checked := (len(a) >> 3) << 3
|
|
a = a[checked:]
|
|
b = b[checked:]
|
|
for i := range a {
|
|
if a[i] != b[i] {
|
|
return i + checked
|
|
}
|
|
}
|
|
return len(a) + checked
|
|
}
|
|
|
|
func load3232(b []byte, i int32) uint32 {
|
|
return binary.LittleEndian.Uint32(b[i:])
|
|
}
|
|
|
|
func load6432(b []byte, i int32) uint64 {
|
|
return binary.LittleEndian.Uint64(b[i:])
|
|
}
|
|
|
|
func load64(b []byte, i int) uint64 {
|
|
return binary.LittleEndian.Uint64(b[i:])
|
|
}
|
|
|
|
type byter interface {
|
|
Bytes() []byte
|
|
Len() int
|
|
}
|
|
|
|
var _ byter = &bytes.Buffer{}
|