mirror of
https://github.com/vacp2p/linea-monorepo.git
synced 2026-01-09 23:47:55 -05:00
543aedd23f
* refactor: disallow explicit dictionaries for anything other than v0 decompression setup * refactor: pass around dictionary stores instead of individual dictionaries * feat: add new dictionary --------- Co-authored-by: Arya Tabaie <15056835+Tabaie@users.noreply.github.com>
67 lines
1.9 KiB
Go
67 lines
1.9 KiB
Go
package blob
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"os"
|
|
|
|
"github.com/consensys/linea-monorepo/prover/lib/compressor/blob/dictionary"
|
|
"github.com/consensys/linea-monorepo/prover/lib/compressor/blob/encode"
|
|
v0 "github.com/consensys/linea-monorepo/prover/lib/compressor/blob/v0"
|
|
v1 "github.com/consensys/linea-monorepo/prover/lib/compressor/blob/v1"
|
|
"github.com/ethereum/go-ethereum/rlp"
|
|
)
|
|
|
|
func GetVersion(blob []byte) uint16 {
|
|
if len(blob) < 3 {
|
|
return 0
|
|
}
|
|
|
|
if blob[0] == 0x3f && blob[1] == 0xff && blob[2]&0xc0 == 0xc0 {
|
|
return 1
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func LoadDict(dictPath string) ([]byte, error) {
|
|
return os.ReadFile(dictPath)
|
|
}
|
|
|
|
// DecompressBlob takes in a Linea blob and outputs an RLP encoded list of RLP encoded blocks.
|
|
// Due to information loss during pre-compression encoding, two pieces of information are represented "hackily":
|
|
// The block hash is in the ParentHash field.
|
|
// The transaction from address is in the signature.R field.
|
|
func DecompressBlob(blob []byte, dictStore dictionary.Store) ([]byte, error) {
|
|
vsn := GetVersion(blob)
|
|
var (
|
|
blockDecoder func(*bytes.Reader) (encode.DecodedBlockData, error)
|
|
blocks [][]byte
|
|
err error
|
|
)
|
|
switch vsn {
|
|
case 0:
|
|
_, _, blocks, err = v0.DecompressBlob(blob, dictStore)
|
|
blockDecoder = v0.DecodeBlockFromUncompressed
|
|
case 1:
|
|
_, _, blocks, _, err = v1.DecompressBlob(blob, dictStore)
|
|
blockDecoder = v1.DecodeBlockFromUncompressed
|
|
default:
|
|
return nil, errors.New("unrecognized blob version")
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
blocksSerialized := make([][]byte, len(blocks))
|
|
var decodedBlock encode.DecodedBlockData
|
|
for i, block := range blocks {
|
|
if decodedBlock, err = blockDecoder(bytes.NewReader(block)); err != nil {
|
|
return nil, err
|
|
}
|
|
if blocksSerialized[i], err = rlp.EncodeToBytes(decodedBlock.ToStd()); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return rlp.EncodeToBytes(blocksSerialized)
|
|
}
|