From 21bdc82fb5f74e89dfa9a13f821ffa9e37faaaa2 Mon Sep 17 00:00:00 2001
From: Leonard Lyubich <leonard@nspcc.ru>
Date: Thu, 1 Oct 2020 20:12:53 +0300
Subject: [PATCH] [#64] core/object: Implement tombstone content messages

Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
---
 pkg/core/object/tombstone.go      | 163 ++++++++++++++++++++++++++++++
 pkg/core/object/tombstone_test.go |  43 ++++++++
 2 files changed, 206 insertions(+)
 create mode 100644 pkg/core/object/tombstone.go
 create mode 100644 pkg/core/object/tombstone_test.go

diff --git a/pkg/core/object/tombstone.go b/pkg/core/object/tombstone.go
new file mode 100644
index 000000000..924d4690a
--- /dev/null
+++ b/pkg/core/object/tombstone.go
@@ -0,0 +1,163 @@
+package object
+
+import (
+	"encoding/binary"
+	"fmt"
+	"io"
+
+	"github.com/nspcc-dev/neofs-api-go/pkg/object"
+	"github.com/nspcc-dev/neofs-api-go/util/proto"
+	"github.com/nspcc-dev/neofs-api-go/v2/refs"
+	"github.com/pkg/errors"
+)
+
+// FIXME: replace this code to neofs-api-go
+
+const addrListField = 1
+
+type TombstoneContent TombstoneContentV2
+
+type TombstoneContentV2 struct {
+	addrList []*refs.Address
+}
+
+func NewTombstoneContentFromV2(cV2 *TombstoneContentV2) *TombstoneContent {
+	return (*TombstoneContent)(cV2)
+}
+
+func NewTombstoneContent() *TombstoneContent {
+	return NewTombstoneContentFromV2(new(TombstoneContentV2))
+}
+
+func (c *TombstoneContent) MarshalBinary() ([]byte, error) {
+	return c.ToV2().StableMarshal(nil)
+}
+
+func (c *TombstoneContent) SetAddressList(v ...*object.Address) {
+	if c != nil {
+		addrV2 := make([]*refs.Address, 0, len(v))
+
+		for i := range v {
+			addrV2 = append(addrV2, v[i].ToV2())
+		}
+
+		(*TombstoneContentV2)(c).SetAddressList(addrV2)
+	}
+}
+
+func (c *TombstoneContent) GetAddressList() []*object.Address {
+	if c != nil {
+		addrV2 := (*TombstoneContentV2)(c).GetAddressList()
+		addr := make([]*object.Address, 0, len(addrV2))
+
+		for i := range addrV2 {
+			addr = append(addr, object.NewAddressFromV2(addrV2[i]))
+		}
+
+		return addr
+	}
+
+	return nil
+}
+
+func (c *TombstoneContent) ToV2() *TombstoneContentV2 {
+	return (*TombstoneContentV2)(c)
+}
+
+func (c *TombstoneContentV2) StableMarshal(buf []byte) ([]byte, error) {
+	if c == nil {
+		return make([]byte, 0), nil
+	}
+
+	if buf == nil {
+		buf = make([]byte, c.StableSize())
+	}
+
+	offset := 0
+
+	for i := range c.addrList {
+		n, err := proto.NestedStructureMarshal(addrListField, buf[offset:], c.addrList[i])
+		if err != nil {
+			return nil, err
+		}
+
+		offset += n
+	}
+
+	return buf, nil
+}
+
+func (c *TombstoneContentV2) StableSize() (sz int) {
+	if c == nil {
+		return
+	}
+
+	for i := range c.addrList {
+		sz += proto.NestedStructureSize(addrListField, c.addrList[i])
+	}
+
+	return
+}
+
+func (c *TombstoneContentV2) StableUnmarshal(data []byte) error {
+	c.addrList = c.addrList[:0]
+
+	for len(data) > 0 {
+		expPrefix, _ := proto.NestedStructurePrefix(addrListField)
+
+		prefix, ln := binary.Uvarint(data)
+		if ln <= 0 {
+			return io.ErrUnexpectedEOF
+		} else if expPrefix != prefix {
+			return errors.Errorf("wrong prefix %d", prefix)
+		}
+
+		data = data[ln:]
+
+		sz, ln := binary.Uvarint(data)
+		if ln <= 0 {
+			return io.ErrUnexpectedEOF
+		}
+
+		data = data[ln:]
+
+		addr := new(refs.Address)
+		if err := addr.StableUnmarshal(data[:sz]); err != nil {
+			fmt.Println(err)
+			return err
+		}
+
+		c.addrList = append(c.addrList, addr)
+
+		data = data[sz:]
+	}
+
+	return nil
+}
+
+func (c *TombstoneContentV2) SetAddressList(v []*refs.Address) {
+	if c != nil {
+		c.addrList = v
+	}
+}
+
+func (c *TombstoneContentV2) GetAddressList() []*refs.Address {
+	if c != nil {
+		return c.addrList
+	}
+
+	return nil
+}
+
+func TombstoneContentFromBytes(data []byte) (*TombstoneContent, error) {
+	if len(data) == 0 {
+		return nil, nil
+	}
+
+	cV2 := new(TombstoneContentV2)
+	if err := cV2.StableUnmarshal(data); err != nil {
+		return nil, err
+	}
+
+	return NewTombstoneContentFromV2(cV2), nil
+}
diff --git a/pkg/core/object/tombstone_test.go b/pkg/core/object/tombstone_test.go
new file mode 100644
index 000000000..7ca935c6a
--- /dev/null
+++ b/pkg/core/object/tombstone_test.go
@@ -0,0 +1,43 @@
+package object
+
+import (
+	"crypto/sha256"
+	"testing"
+
+	"github.com/nspcc-dev/neofs-api-go/pkg/container"
+	"github.com/nspcc-dev/neofs-api-go/pkg/object"
+	"github.com/stretchr/testify/require"
+)
+
+func TestTombstoneContent_MarshalBinary(t *testing.T) {
+	cid1 := container.NewID()
+	cid1.SetSHA256([sha256.Size]byte{1, 2})
+
+	id1 := object.NewID()
+	id1.SetSHA256([sha256.Size]byte{3, 4})
+
+	addr1 := object.NewAddress()
+	addr1.SetObjectID(id1)
+	addr1.SetContainerID(cid1)
+
+	cid2 := container.NewID()
+	cid2.SetSHA256([sha256.Size]byte{5, 6})
+
+	id2 := object.NewID()
+	id2.SetSHA256([sha256.Size]byte{7, 8})
+
+	addr2 := object.NewAddress()
+	addr2.SetObjectID(id2)
+	addr2.SetContainerID(cid2)
+
+	c := NewTombstoneContent()
+	c.SetAddressList(addr1, addr2)
+
+	data, err := c.MarshalBinary()
+	require.NoError(t, err)
+
+	c2, err := TombstoneContentFromBytes(data)
+	require.NoError(t, err)
+
+	require.Equal(t, c, c2)
+}