neoneo-go/pkg/network/compress.go
Roman Khimov dde0763840 network: fix compression handling functions
Turns out, Neo uses block compression and not a streaming lz4 format. And it
doesn't contain uncompressed payload size which makes guessing it somewhat
suboptimal.
2020-06-18 12:12:43 +03:00

29 lines
646 B
Go

package network
import (
"github.com/pierrec/lz4"
)
// compress compresses bytes using lz4.
func compress(source []byte) ([]byte, error) {
dest := make([]byte, lz4.CompressBlockBound(len(source)))
size, err := lz4.CompressBlock(source, dest, nil)
if err != nil {
return nil, err
}
return dest[:size], nil
}
// decompress decompresses bytes using lz4.
func decompress(source []byte) ([]byte, error) {
maxSize := len(source) * 255
if maxSize > PayloadMaxSize {
maxSize = PayloadMaxSize
}
dest := make([]byte, maxSize)
size, err := lz4.UncompressBlock(source, dest)
if err != nil {
return nil, err
}
return dest[:size], nil
}