[v2] Blob Metadata Store

core/data.go

@@ -5,11 +5,14 @@ import (
 	"errors"
 	"fmt"
 	"math/big"
+	"strconv"
 
 	"github.com/Layr-Labs/eigenda/common"
 	"github.com/Layr-Labs/eigenda/encoding"
+	"github.com/aws/aws-sdk-go-v2/service/dynamodb/types"
 	"github.com/consensys/gnark-crypto/ecc/bn254"
-	"github.com/ethereum/go-ethereum/crypto"
+	"github.com/ethereum/go-ethereum/accounts/abi"
+	"golang.org/x/crypto/sha3"
 )
 
 type AccountID = string
@@ -483,34 +486,80 @@ func (cb Bundles) FromEncodedBundles(eb EncodedBundles) (Bundles, error) {
 
 // PaymentMetadata represents the header information for a blob
 type PaymentMetadata struct {
-	// Existing fields
-	AccountID string
+	// AccountID is the ETH account address for the payer
+	AccountID string `json:"account_id"`
 
-	// New fields
-	BinIndex uint32
+	// BinIndex represents the range of time at which the dispersal is made
+	BinIndex uint32 `json:"bin_index"`
 	// TODO: we are thinking the contract can use uint128 for cumulative payment,
 	// but the definition on v2 uses uint64. Double check with team.
-	CumulativePayment *big.Int
+	CumulativePayment *big.Int `json:"cumulative_payment"`
 }
 
 // Hash returns the Keccak256 hash of the PaymentMetadata
-func (pm *PaymentMetadata) Hash() []byte {
-	// Create a byte slice to hold the serialized data
-	data := make([]byte, 0, len(pm.AccountID)+4+pm.CumulativePayment.BitLen()/8+1)
-
-	// Append AccountID
-	data = append(data, []byte(pm.AccountID)...)
-
-	// Append BinIndex
-	binIndexBytes := make([]byte, 4)
-	binary.BigEndian.PutUint32(binIndexBytes, pm.BinIndex)
-	data = append(data, binIndexBytes...)
-
-	// Append CumulativePayment
-	paymentBytes := pm.CumulativePayment.Bytes()
-	data = append(data, paymentBytes...)
+func (pm *PaymentMetadata) Hash() ([32]byte, error) {
+	blobHeaderType, err := abi.NewType("tuple", "", []abi.ArgumentMarshaling{
+		{
+			Name: "accountID",
+			Type: "string",
+		},
+		{
+			Name: "binIndex",
+			Type: "uint32",
+		},
+		{
+			Name: "cumulativePayment",
+			Type: "uint256",
+		},
+	})
+	if err != nil {
+		return [32]byte{}, err
+	}
+
+	arguments := abi.Arguments{
+		{
+			Type: blobHeaderType,
+		},
+	}
+
+	bytes, err := arguments.Pack(pm)
+	if err != nil {
+		return [32]byte{}, err
+	}
+
+	var hash [32]byte
+	hasher := sha3.NewLegacyKeccak256()
+	hasher.Write(bytes)
+	copy(hash[:], hasher.Sum(nil)[:32])
+
+	return hash, nil
+}
+
+func (pm *PaymentMetadata) MarshalDynamoDBAttributeValue() (types.AttributeValue, error) {
+	return &types.AttributeValueMemberM{
+		Value: map[string]types.AttributeValue{
+			"AccountID": &types.AttributeValueMemberS{Value: pm.AccountID},
+			"BinIndex":  &types.AttributeValueMemberN{Value: fmt.Sprintf("%d", pm.BinIndex)},
+			"CumulativePayment": &types.AttributeValueMemberN{
+				Value: pm.CumulativePayment.String(),
+			},
+		},
+	}, nil
+}
 
-	return crypto.Keccak256(data)
+func (pm *PaymentMetadata) UnmarshalDynamoDBAttributeValue(av types.AttributeValue) error {
+	m, ok := av.(*types.AttributeValueMemberM)
+	if !ok {
+		return fmt.Errorf("expected *types.AttributeValueMemberM, got %T", av)
+	}
+	pm.AccountID = m.Value["AccountID"].(*types.AttributeValueMemberS).Value
+	binIndex, err := strconv.ParseUint(m.Value["BinIndex"].(*types.AttributeValueMemberN).Value, 10, 32)
+	if err != nil {
+		return fmt.Errorf("failed to parse BinIndex: %w", err)
+	}
+	pm.BinIndex = uint32(binIndex)
+	pm.CumulativePayment, _ = new(big.Int).SetString(m.Value["CumulativePayment"].(*types.AttributeValueMemberN).Value, 10)
+	return nil
 }
 
 // OperatorInfo contains information about an operator which is stored on the blockchain state,

core/v2/assignment.go

@@ -8,7 +8,7 @@ import (
 	"github.com/Layr-Labs/eigenda/core"
 )
 
-func GetAssignments(state *core.OperatorState, blobVersion byte, quorum uint8) (map[core.OperatorID]Assignment, error) {
+func GetAssignments(state *core.OperatorState, blobVersion BlobVersion, quorum uint8) (map[core.OperatorID]Assignment, error) {
 
 	params, ok := ParametersMap[blobVersion]
 	if !ok {
@@ -81,7 +81,7 @@ func GetAssignments(state *core.OperatorState, blobVersion byte, quorum uint8) (
 
 }
 
-func GetAssignment(state *core.OperatorState, blobVersion byte, quorum core.QuorumID, id core.OperatorID) (Assignment, error) {
+func GetAssignment(state *core.OperatorState, blobVersion BlobVersion, quorum core.QuorumID, id core.OperatorID) (Assignment, error) {
 
 	assignments, err := GetAssignments(state, blobVersion, quorum)
 	if err != nil {
@@ -96,7 +96,7 @@ func GetAssignment(state *core.OperatorState, blobVersion byte, quorum core.Quor
 	return assignment, nil
 }
 
-func GetChunkLength(blobVersion byte, blobLength uint32) (uint32, error) {
+func GetChunkLength(blobVersion BlobVersion, blobLength uint32) (uint32, error) {
 
 	if blobLength == 0 {
 		return 0, fmt.Errorf("blob length must be greater than 0")

core/v2/assignment_test.go

@@ -20,7 +20,7 @@ func TestOperatorAssignmentsV2(t *testing.T) {
 	state := dat.GetTotalOperatorState(context.Background(), 0)
 	operatorState := state.OperatorState
 
-	blobVersion := byte(0)
+	blobVersion := corev2.BlobVersion(0)
 
 	assignments, err := corev2.GetAssignments(operatorState, blobVersion, 0)
 	assert.NoError(t, err)
@@ -90,7 +90,7 @@ func TestAssignmentWithTooManyOperators(t *testing.T) {
 
 	assert.Equal(t, len(state.Operators[0]), numOperators)
 
-	blobVersion := byte(0)
+	blobVersion := corev2.BlobVersion(0)
 
 	_, err = corev2.GetAssignments(state.OperatorState, blobVersion, 0)
 	assert.Error(t, err)
@@ -131,7 +131,7 @@ func FuzzOperatorAssignmentsV2(f *testing.F) {
 
 		state := dat.GetTotalOperatorState(context.Background(), 0)
 
-		blobVersion := byte(0)
+		blobVersion := corev2.BlobVersion(0)
 
 		assignments, err := corev2.GetAssignments(state.OperatorState, blobVersion, 0)
 		assert.NoError(t, err)
@@ -162,7 +162,7 @@ func FuzzOperatorAssignmentsV2(f *testing.F) {
 
 func TestChunkLength(t *testing.T) {
 
-	blobVersion := byte(0)
+	blobVersion := corev2.BlobVersion(0)
 
 	pairs := []struct {
 		blobLength  uint32
@@ -188,7 +188,7 @@ func TestChunkLength(t *testing.T) {
 
 func TestInvalidChunkLength(t *testing.T) {
 
-	blobVersion := byte(0)
+	blobVersion := corev2.BlobVersion(0)
 
 	invalidLengths := []uint32{
 		0,

core/v2/core_test.go

@@ -84,7 +84,7 @@ func makeTestComponents() (encoding.Prover, encoding.Verifier, error) {
 	return p, v, nil
 }
 
-func makeTestBlob(t *testing.T, p encoding.Prover, version uint8, refBlockNumber uint64, length int, quorums []core.QuorumID) (corev2.BlobCertificate, []byte) {
+func makeTestBlob(t *testing.T, p encoding.Prover, version corev2.BlobVersion, refBlockNumber uint64, length int, quorums []core.QuorumID) (corev2.BlobCertificate, []byte) {
 
 	data := make([]byte, length*31)
 	_, err := rand.Read(data)
@@ -101,7 +101,7 @@ func makeTestBlob(t *testing.T, p encoding.Prover, version uint8, refBlockNumber
 
 	header := corev2.BlobCertificate{
 		BlobHeader: corev2.BlobHeader{
-			Version:         version,
+			BlobVersion:     version,
 			QuorumNumbers:   quorums,
 			BlobCommitments: commitments,
 		},
@@ -148,7 +148,7 @@ func prepareBlobs(t *testing.T, operatorCount uint, headers []corev2.BlobCertifi
 
 		for _, quorum := range header.QuorumNumbers {
 
-			assignments, err := corev2.GetAssignments(state, header.Version, quorum)
+			assignments, err := corev2.GetAssignments(state, header.BlobVersion, quorum)
 			if err != nil {
 				t.Fatal(err)
 			}
@@ -238,7 +238,7 @@ func TestValidationSucceeds(t *testing.T) {
 
 	bn := uint64(0)
 
-	version := uint8(0)
+	version := corev2.BlobVersion(0)
 
 	pool := workerpool.New(1)