Support ReplicateV2

eacl/table_test.go

@@ -242,7 +242,7 @@ func TestConstructTable(t *testing.T) {
 	require.Equal(t, anyValidRecords, tbl.Records())
 	require.Zero(t, tbl.GetCID())
 	require.EqualValues(t, 2, tbl.Version().Major())
-	require.EqualValues(t, 26, tbl.Version().Minor())
+	require.EqualValues(t, 27, tbl.Version().Minor())
 }
 
 func TestNewTableForContainer(t *testing.T) {
@@ -252,5 +252,5 @@ func TestNewTableForContainer(t *testing.T) {
 	cnr2 := tbl.GetCID()
 	require.Equal(t, cnr, cnr2)
 	require.EqualValues(t, 2, tbl.Version().Major())
-	require.EqualValues(t, 26, tbl.Version().Minor())
+	require.EqualValues(t, 27, tbl.Version().Minor())
 }

proto/object/encoding.go

@@ -1293,3 +1293,73 @@ const (
 	FieldReplicateResponseStatus
 	FieldReplicateResponseObjectSignature
 )
+
+// Field numbers of [ReplicateV2Request_Init] message.
+const (
+	_ = iota
+	FieldReplicateV2RequestInitObject
+	FieldReplicateV2RequestInitSignature
+	FieldReplicateV2RequestInitSignObject
+)
+
+// CalculateReplicateV2InitLength calculates length of ReplicateV2 request oneof
+// init message with given fields.
+func CalculateReplicateV2InitLength(objLen int, sigLen int, signObject bool) int {
+	ln := protoencoding.SizeEmbeddedLENField(FieldReplicateV2RequestInitObject, objLen)
+	ln += protoencoding.SizeEmbeddedLENField(FieldReplicateV2RequestInitSignature, sigLen)
+	ln += protoencoding.SizeBool(FieldReplicateV2RequestInitSignObject, signObject)
+	return ln
+}
+
+// WriteReplicateV2Init writes ReplicateV2 request init oneof field with given
+// fields into buf. Returns number of bytes written.
+func WriteReplicateV2Init(buf []byte, objLen int, writeObjFn protoencoding.WriteMessageFunc, sigLen int, writeSigFn protoencoding.WriteMessageFunc, signObject bool) int {
+	off := protoencoding.WriteMessageField(buf, FieldReplicateV2RequestInitObject, objLen, writeObjFn)
+	off += protoencoding.WriteMessageField(buf[off:], FieldReplicateV2RequestInitSignature, sigLen, writeSigFn)
+	off += protoencoding.MarshalToBool(buf[off:], FieldReplicateV2RequestInitSignObject, signObject)
+	return off
+}
+
+// Field numbers of [ReplicateV2Request] message.
+const (
+	_ = iota
+	FieldReplicateV2RequestInit
+	FieldReplicateV2RequestChunk
+)
+
+// CalculateReplicateV2InitRequestLength calculates length of initial
+// ReplicateV2 request message with given fields.
+func CalculateReplicateV2InitRequestLength(initLen int) int {
+	return protoencoding.SizeEmbeddedLENField(FieldReplicateV2RequestInit, initLen)
+}
+
+// CalculateReplicateV2ChunkRequestLength calculates length of chunk ReplicateV2
+// request message with given fields.
+func CalculateReplicateV2ChunkRequestLength(chunk []byte) int {
+	return protoencoding.SizeBytes(FieldReplicateV2RequestChunk, chunk)
+}
+
+// WriteReplicateV2InitRequest writes initial ReplicateV2 request with given
+// fields into buf. Returns number of bytes written.
+func WriteReplicateV2InitRequest(buf []byte, objLen int, writeObjFn protoencoding.WriteMessageFunc, nodeSigLen int, writeNodeSigFn protoencoding.WriteMessageFunc, signObject bool) int {
+	ln := CalculateReplicateV2InitLength(objLen, nodeSigLen, signObject)
+	if ln == 0 {
+		return 0
+	}
+	off := protoencoding.WriteTagAndLength(buf, FieldReplicateV2RequestInit, ln)
+	off += WriteReplicateV2Init(buf[off:], objLen, writeObjFn, nodeSigLen, writeNodeSigFn, signObject)
+	return off
+}
+
+// WriteReplicateV2ChunkRequest writes chunk ReplicateV2 request with given
+// length and fields into buf. Returns number of bytes written.
+func WriteReplicateV2ChunkRequest(buf []byte, chunk []byte) int {
+	return protoencoding.MarshalToBytes(buf, FieldReplicateV2RequestChunk, chunk)
+}
+
+// Field numbers of [ReplicateV2Response] message.
+const (
+	_ = iota
+	FieldReplicateV2ResponseStatus
+	FieldReplicateV2ResponseObjectSignature
+)

proto/object/encoding_test.go

@@ -1,13 +1,16 @@
 package object_test
 
 import (
+	"bytes"
 	"testing"
 
+	"github.com/nspcc-dev/neofs-sdk-go/internal/testutil"
 	protoencoding "github.com/nspcc-dev/neofs-sdk-go/proto/encoding"
 	prototest "github.com/nspcc-dev/neofs-sdk-go/proto/internal/test"
 	"github.com/nspcc-dev/neofs-sdk-go/proto/object"
 	"github.com/nspcc-dev/neofs-sdk-go/proto/refs"
 	"github.com/stretchr/testify/require"
+	"google.golang.org/protobuf/proto"
 )
 
 // returns random object.Range with all non-zero fields.
@@ -469,3 +472,57 @@ func TestSearchV2Response_Body_MarshalStable(t *testing.T) {
 		},
 	})
 }
+
+func TestWriteReplicateV2InitRequest(t *testing.T) {
+	obj := &object.Object{
+		ObjectId:  prototest.RandObjectID(),
+		Signature: prototest.RandSignature(),
+		Header:    randHeader(),
+	}
+	objHdrLen := obj.MarshaledSize()
+	sig := prototest.RandSignature()
+	sigLen := sig.MarshaledSize()
+	const signObject = true
+
+	initLen := object.CalculateReplicateV2InitLength(objHdrLen, sigLen, signObject)
+
+	ln := object.CalculateReplicateV2InitRequestLength(initLen)
+
+	buf := make([]byte, ln)
+
+	writeObjFn := protoencoding.WriteStablyMarshalledMessageFunc(obj)
+	writeSigFn := protoencoding.WriteStablyMarshalledMessageFunc(sig)
+
+	n := object.WriteReplicateV2InitRequest(buf, objHdrLen, writeObjFn, sigLen, writeSigFn, signObject)
+
+	require.EqualValues(t, ln, n)
+
+	var req object.ReplicateV2Request
+	require.NoError(t, proto.Unmarshal(buf, &req))
+
+	init := req.GetInit()
+	require.NotNil(t, init)
+
+	require.True(t, proto.Equal(obj, init.Object))
+	require.True(t, proto.Equal(sig, init.Signature))
+	require.EqualValues(t, signObject, init.SignObject)
+}
+
+func TestWriteReplicateV2ChunkRequest(t *testing.T) {
+	chunk := testutil.RandByteSlice(1024)
+
+	ln := object.CalculateReplicateV2ChunkRequestLength(chunk)
+
+	buf := make([]byte, ln)
+
+	n := object.WriteReplicateV2ChunkRequest(buf, chunk)
+
+	require.EqualValues(t, ln, n)
+
+	var req object.ReplicateV2Request
+	require.NoError(t, proto.Unmarshal(buf, &req))
+
+	gotChunk := req.GetPayloadChunk()
+	require.NotNil(t, gotChunk)
+	require.True(t, bytes.Equal(chunk, gotChunk))
+}