[2/N][zero serialization] Make Batcher operate on chunks without ser/…

…deser (Layr-Labs#700)

api/clients/mock/node_client.go

@@ -54,9 +54,18 @@ func (c *MockNodeClient) GetChunks(
 ) {
 	args := c.Called(opID, opInfo, batchHeaderHash, blobIndex)
 	encodedBlob := (args.Get(0)).(core.EncodedBlob)
+	chunks, err := encodedBlob.EncodedBundlesByOperator[opID][quorumID].ToFrames()
+	if err != nil {
+		chunksChan <- clients.RetrievedChunks{
+			OperatorID: opID,
+			Err:        err,
+			Chunks:     nil,
+		}
+
+	}
 	chunksChan <- clients.RetrievedChunks{
 		OperatorID: opID,
 		Err:        nil,
-		Chunks:     encodedBlob.BundlesByOperator[opID][quorumID],
+		Chunks:     chunks,
 	}
 }

api/clients/retrieval_client_test.go

@@ -60,8 +60,8 @@ var (
 	retrievalClient   clients.RetrievalClient
 	blobHeader        *core.BlobHeader
 	encodedBlob       core.EncodedBlob = core.EncodedBlob{
-		BlobHeader:        nil,
-		BundlesByOperator: make(map[core.OperatorID]core.Bundles),
+		BlobHeader:               nil,
+		EncodedBundlesByOperator: make(map[core.OperatorID]core.EncodedBundles),
 	}
 	batchHeaderHash        [32]byte
 	batchRoot              [32]byte
@@ -198,7 +198,11 @@ func setup(t *testing.T) {
 		bundles := make(map[core.QuorumID]core.Bundle, len(blobHeader.QuorumInfos))
 		bundles[quorumID] = chunks[assignment.StartIndex : assignment.StartIndex+assignment.NumChunks]
 		encodedBlob.BlobHeader = blobHeader
-		encodedBlob.BundlesByOperator[id] = bundles
+		eb, err := core.Bundles(bundles).ToEncodedBundles()
+		if err != nil {
+			t.Fatal(err)
+		}
+		encodedBlob.EncodedBundlesByOperator[id] = eb
 	}
 
 }

core/data.go

@@ -87,6 +87,49 @@ func (cd *ChunksData) Size() uint64 {
 	return size
 }
 
+func (cd *ChunksData) FromFrames(fr []*encoding.Frame) (*ChunksData, error) {
+	if len(fr) == 0 {
+		return nil, errors.New("no frame is provided")
+	}
+	var c ChunksData
+	c.Format = GnarkChunkEncodingFormat
+	c.ChunkLen = fr[0].Length()
+	c.Chunks = make([][]byte, 0, len(fr))
+	for _, f := range fr {
+		bytes, err := f.SerializeGnark()
+		if err != nil {
+			return nil, err
+		}
+		c.Chunks = append(c.Chunks, bytes)
+	}
+	return &c, nil
+}
+
+func (cd *ChunksData) ToFrames() ([]*encoding.Frame, error) {
+	frames := make([]*encoding.Frame, 0, len(cd.Chunks))
+	switch cd.Format {
+	case GobChunkEncodingFormat:
+		for _, data := range cd.Chunks {
+			fr, err := new(encoding.Frame).Deserialize(data)
+			if err != nil {
+				return nil, err
+			}
+			frames = append(frames, fr)
+		}
+	case GnarkChunkEncodingFormat:
+		for _, data := range cd.Chunks {
+			fr, err := new(encoding.Frame).DeserializeGnark(data)
+			if err != nil {
+				return nil, err
+			}
+			frames = append(frames, fr)
+		}
+	default:
+		return nil, fmt.Errorf("invalid chunk encoding format: %v", cd.Format)
+	}
+	return frames, nil
+}
+
 func (cd *ChunksData) FlattenToBundle() ([]byte, error) {
 	// Only Gnark coded chunks are dispersed as a byte array.
 	// Gob coded chunks are not flattened.
@@ -128,8 +171,9 @@ func (cd *ChunksData) ToGobFormat() (*ChunksData, error) {
 		gobChunks = append(gobChunks, gob)
 	}
 	return &ChunksData{
-		Chunks: gobChunks,
-		Format: GobChunkEncodingFormat,
+		Chunks:   gobChunks,
+		Format:   GobChunkEncodingFormat,
+		ChunkLen: cd.ChunkLen,
 	}, nil
 }
 
@@ -153,8 +197,9 @@ func (cd *ChunksData) ToGnarkFormat() (*ChunksData, error) {
 		gnarkChunks = append(gnarkChunks, gnark)
 	}
 	return &ChunksData{
-		Chunks: gnarkChunks,
-		Format: GnarkChunkEncodingFormat,
+		Chunks:   gnarkChunks,
+		Format:   GnarkChunkEncodingFormat,
+		ChunkLen: cd.ChunkLen,
 	}, nil
 }
 
@@ -266,6 +311,8 @@ type BatchHeader struct {
 type EncodedBlob struct {
 	BlobHeader        *BlobHeader
 	BundlesByOperator map[OperatorID]Bundles
+	// EncodedBundlesByOperator is bundles in encoded format (not deserialized)
+	EncodedBundlesByOperator map[OperatorID]EncodedBundles
 }
 
 // A Bundle is the collection of chunks associated with a single blob, for a single operator and a single quorum.
@@ -274,12 +321,23 @@ type Bundle []*encoding.Frame
 // Bundles is the collection of bundles associated with a single blob and a single operator.
 type Bundles map[QuorumID]Bundle
 
+// This is similar to Bundle, but tracks chunks in encoded format (i.e. not deserialized).
+type EncodedBundles map[QuorumID]*ChunksData
+
 // BlobMessage is the message that is sent to DA nodes. It contains the blob header and the associated chunk bundles.
 type BlobMessage struct {
 	BlobHeader *BlobHeader
 	Bundles    Bundles
 }
 
+// This is similar to BlobMessage, but keep the commitments and chunks in encoded format
+// (i.e. not deserialized)
+type EncodedBlobMessage struct {
+	// TODO(jianoaix): Change the commitments to encoded format.
+	BlobHeader     *BlobHeader
+	EncodedBundles map[QuorumID]*ChunksData
+}
+
 func (b Bundle) Size() uint64 {
 	size := uint64(0)
 	for _, chunk := range b {
@@ -388,3 +446,27 @@ func (cb Bundles) Size() uint64 {
 	}
 	return size
 }
+
+func (cb Bundles) ToEncodedBundles() (EncodedBundles, error) {
+	eb := make(EncodedBundles)
+	for quorum, bundle := range cb {
+		cd, err := new(ChunksData).FromFrames(bundle)
+		if err != nil {
+			return nil, err
+		}
+		eb[quorum] = cd
+	}
+	return eb, nil
+}
+
+func (cb Bundles) FromEncodedBundles(eb EncodedBundles) (Bundles, error) {
+	c := make(Bundles)
+	for quorum, chunkData := range eb {
+		fr, err := chunkData.ToFrames()
+		if err != nil {
+			return nil, err
+		}
+		c[quorum] = fr
+	}
+	return c, nil
+}

core/data_test.go

@@ -33,6 +33,52 @@ func createBundle(t *testing.T, numFrames, numCoeffs, seed int) core.Bundle {
 	return frames
 }
 
+func createChunksData(t *testing.T, seed int) (core.Bundle, *core.ChunksData, *core.ChunksData) {
+	bundle := createBundle(t, 64, 64, seed)
+	gobChunks := make([][]byte, len(bundle))
+	gnarkChunks := make([][]byte, len(bundle))
+	for i, frame := range bundle {
+		gobChunk, err := frame.Serialize()
+		assert.Nil(t, err)
+		gobChunks[i] = gobChunk
+
+		gnarkChunk, err := frame.SerializeGnark()
+		assert.Nil(t, err)
+		gnarkChunks[i] = gnarkChunk
+	}
+	gob := &core.ChunksData{
+		Chunks:   gobChunks,
+		Format:   core.GobChunkEncodingFormat,
+		ChunkLen: 64,
+	}
+	gnark := &core.ChunksData{
+		Chunks:   gnarkChunks,
+		Format:   core.GnarkChunkEncodingFormat,
+		ChunkLen: 64,
+	}
+	return bundle, gob, gnark
+}
+
+func checkChunksDataEquivalence(t *testing.T, cd1, cd2 *core.ChunksData) {
+	assert.Equal(t, cd1.Format, cd2.Format)
+	assert.Equal(t, cd1.ChunkLen, cd2.ChunkLen)
+	assert.Equal(t, len(cd1.Chunks), len(cd2.Chunks))
+	for i, c1 := range cd1.Chunks {
+		assert.True(t, bytes.Equal(c1, cd2.Chunks[i]))
+	}
+}
+
+func checkBundleEquivalence(t *testing.T, b1, b2 core.Bundle) {
+	assert.Equal(t, len(b1), len(b2))
+	for i := 0; i < len(b1); i++ {
+		assert.True(t, b1[i].Proof.Equal(&b2[i].Proof))
+		assert.Equal(t, len(b1[i].Coeffs), len(b2[i].Coeffs))
+		for j := 0; j < len(b1[i].Coeffs); j++ {
+			assert.True(t, b1[i].Coeffs[j].Equal(&b2[i].Coeffs[j]))
+		}
+	}
+}
+
 func TestInvalidBundleSer(t *testing.T) {
 	b1 := createBundle(t, 1, 0, 0)
 	_, err := b1.Serialize()
@@ -86,41 +132,38 @@ func TestBundleEncoding(t *testing.T) {
 		assert.Nil(t, err)
 		decoded, err := new(core.Bundle).Deserialize(bytes)
 		assert.Nil(t, err)
-		assert.Equal(t, len(bundle), len(decoded))
-		for i := 0; i < len(bundle); i++ {
-			assert.True(t, bundle[i].Proof.Equal(&decoded[i].Proof))
-			assert.Equal(t, len(bundle[i].Coeffs), len(decoded[i].Coeffs))
-			for j := 0; j < len(bundle[i].Coeffs); j++ {
-				assert.True(t, bundle[i].Coeffs[j].Equal(&decoded[i].Coeffs[j]))
-			}
-		}
+		checkBundleEquivalence(t, bundle, decoded)
 	}
 }
 
-func createChunksData(t *testing.T, seed int) (core.Bundle, *core.ChunksData, *core.ChunksData) {
-	bundle := createBundle(t, 64, 64, seed)
-	gobChunks := make([][]byte, len(bundle))
-	gnarkChunks := make([][]byte, len(bundle))
-	for i, frame := range bundle {
-		gobChunk, err := frame.Serialize()
+func TestEncodedBundles(t *testing.T) {
+	numTrials := 16
+	for i := 0; i < numTrials; i++ {
+		bundles := core.Bundles(map[core.QuorumID]core.Bundle{
+			0: createBundle(t, 64, 64, i),
+			1: createBundle(t, 64, 64, i+numTrials),
+		})
+		// ToEncodedBundles
+		ec, err := bundles.ToEncodedBundles()
 		assert.Nil(t, err)
-		gobChunks[i] = gobChunk
-
-		gnarkChunk, err := frame.SerializeGnark()
+		assert.Equal(t, len(ec), len(bundles))
+		for quorum, bundle := range bundles {
+			cd, ok := ec[quorum]
+			assert.True(t, ok)
+			fr, err := cd.ToFrames()
+			assert.Nil(t, err)
+			checkBundleEquivalence(t, fr, bundle)
+		}
+		// FromEncodedBundles
+		bundles2, err := new(core.Bundles).FromEncodedBundles(ec)
 		assert.Nil(t, err)
-		gnarkChunks[i] = gnarkChunk
-	}
-	gob := &core.ChunksData{
-		Chunks:   gobChunks,
-		Format:   core.GobChunkEncodingFormat,
-		ChunkLen: 64,
-	}
-	gnark := &core.ChunksData{
-		Chunks:   gnarkChunks,
-		Format:   core.GnarkChunkEncodingFormat,
-		ChunkLen: 64,
+		assert.Equal(t, len(bundles2), len(bundles))
+		for quorum, bundle := range bundles {
+			b, ok := bundles2[quorum]
+			assert.True(t, ok)
+			checkBundleEquivalence(t, b, bundle)
+		}
 	}
-	return bundle, gob, gnark
 }
 
 func TestChunksData(t *testing.T) {
@@ -136,26 +179,31 @@ func TestChunksData(t *testing.T) {
 		assert.Equal(t, convertedGob, gob)
 		convertedGob, err = gnark.ToGobFormat()
 		assert.Nil(t, err)
-		assert.Equal(t, len(gob.Chunks), len(convertedGob.Chunks))
-		for i := 0; i < len(gob.Chunks); i++ {
-			assert.True(t, bytes.Equal(gob.Chunks[i], convertedGob.Chunks[i]))
-		}
+		checkChunksDataEquivalence(t, gob, convertedGob)
 		// ToGnarkFormat
 		convertedGnark, err := gnark.ToGnarkFormat()
 		assert.Nil(t, err)
 		assert.Equal(t, convertedGnark, gnark)
 		convertedGnark, err = gob.ToGnarkFormat()
 		assert.Nil(t, err)
-		assert.Equal(t, len(gnark.Chunks), len(convertedGnark.Chunks))
-		for i := 0; i < len(gnark.Chunks); i++ {
-			assert.True(t, bytes.Equal(gnark.Chunks[i], convertedGnark.Chunks[i]))
-		}
+		checkChunksDataEquivalence(t, gnark, convertedGnark)
 		// FlattenToBundle
 		bytesFromChunksData, err := gnark.FlattenToBundle()
 		assert.Nil(t, err)
 		bytesFromBundle, err := bundle.Serialize()
 		assert.Nil(t, err)
 		assert.True(t, bytes.Equal(bytesFromChunksData, bytesFromBundle))
+		// FromFrames
+		cd, err := new(core.ChunksData).FromFrames(bundle)
+		assert.Nil(t, err)
+		checkChunksDataEquivalence(t, cd, gnark)
+		// ToFrames
+		fr1, err := gob.ToFrames()
+		assert.Nil(t, err)
+		checkBundleEquivalence(t, bundle, fr1)
+		fr2, err := gnark.ToFrames()
+		assert.Nil(t, err)
+		checkBundleEquivalence(t, bundle, fr2)
 		// Invalid cases
 		gnark.Chunks[0] = gnark.Chunks[0][1:]
 		_, err = gnark.FlattenToBundle()