mirror of
https://github.com/luxfi/node.git
synced 2026-10-30 15:24:05 +00:00
zeekay
7cf87b230b
The module already called itself github.com/luxfi/node while containing no node in Go: cmd/luxd was twenty-one lines that printed the size of a few types, and the Go column of the matrix measured a VM plugin. A repo that fetches the thing it claims to be has not replaced it, so the Go sources move in — 1515 files, the daemon and everything under it — and the build reaches outside this module for nothing. Rust and C++ sit beside it unchanged and are still measured against it by the differential. That is the point of one repo: the reference implementation and the two ports are the same checkout, so parity is a thing you can run rather than a thing you assert. Verified: builds, and the closure carries no lux-private — checked at build time rather than believed. The daemon starts twelve chains on a local network (P X C Q Z A B G K M F, plus D through its plugin) and reports healthy.
75 lines
2.2 KiB
Go
75 lines
2.2 KiB
Go
// Copyright (C) 2019-2025, Lux Industries Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
package compression
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
|
|
"github.com/klauspost/compress/zstd"
|
|
)
|
|
|
|
var _ Compressor = (*zstdCompressor)(nil)
|
|
|
|
func NewZstdCompressor(maxSize int64) (Compressor, error) {
|
|
return NewZstdCompressorWithLevel(maxSize, zstd.SpeedDefault)
|
|
}
|
|
|
|
func NewZstdCompressorWithLevel(maxSize int64, level zstd.EncoderLevel) (Compressor, error) {
|
|
if maxSize == math.MaxInt64 {
|
|
// "Decompress" creates "io.LimitReader" with max size + 1:
|
|
// if the max size + 1 overflows, "io.LimitReader" reads nothing
|
|
// returning 0 byte for the decompress call
|
|
// require max size < math.MaxInt64 to prevent int64 overflows
|
|
return nil, ErrInvalidMaxSizeCompressor
|
|
}
|
|
// Configure decoder with maximum decompressed size limit
|
|
decoder, err := zstd.NewReader(nil, zstd.WithDecoderMaxMemory(uint64(maxSize)))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &zstdCompressor{
|
|
maxSize: maxSize,
|
|
level: level,
|
|
decoder: decoder,
|
|
}, nil
|
|
}
|
|
|
|
type zstdCompressor struct {
|
|
maxSize int64
|
|
level zstd.EncoderLevel
|
|
decoder *zstd.Decoder
|
|
}
|
|
|
|
func (z *zstdCompressor) Compress(msg []byte) ([]byte, error) {
|
|
if int64(len(msg)) > z.maxSize {
|
|
return nil, fmt.Errorf("%w: (%d) > (%d)", ErrMsgTooLarge, len(msg), z.maxSize)
|
|
}
|
|
encoder, err := zstd.NewWriter(nil, zstd.WithEncoderLevel(z.level))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return encoder.EncodeAll(msg, nil), nil
|
|
}
|
|
|
|
func (z *zstdCompressor) Decompress(msg []byte) ([]byte, error) {
|
|
decompressed, err := z.decoder.DecodeAll(msg, nil)
|
|
if err != nil {
|
|
// If the decoder returns an error about size limit, wrap it with our error
|
|
if err.Error() == "decompressed size exceeds configured limit" {
|
|
return nil, fmt.Errorf("%w: decompression stopped due to size limit", ErrDecompressedMsgTooLarge)
|
|
}
|
|
return nil, err
|
|
}
|
|
if int64(len(decompressed)) > z.maxSize {
|
|
return nil, fmt.Errorf("%w: (%d) > (%d)", ErrDecompressedMsgTooLarge, len(decompressed), z.maxSize)
|
|
}
|
|
// Normalize nil to empty slice so compress/decompress roundtrip is
|
|
// consistent for empty input.
|
|
if decompressed == nil {
|
|
decompressed = []byte{}
|
|
}
|
|
return decompressed, nil
|
|
}
|