From fc0da3c8fcb69e49a9e26cea3b482ff3102381d1 Mon Sep 17 00:00:00 2001
From: Leonard Lyubich <leonard@nspcc.ru>
Date: Wed, 6 May 2020 14:15:07 +0300
Subject: [PATCH] object: implement signing payload methods on DeleteRequest
 message

---
 object/sign.go      | 77 ++++++++++++++++++++++++++-------------------
 object/sign_test.go | 16 ++++++++++
 2 files changed, 60 insertions(+), 33 deletions(-)

diff --git a/object/sign.go b/object/sign.go
index 6cece1b..acb9901 100644
--- a/object/sign.go
+++ b/object/sign.go
@@ -4,27 +4,23 @@ import (
 	"io"
 )
 
-// SignedData returns marshaled payload of the Put request.
+// SignedData returns payload bytes of the request.
 //
 // If payload is nil, ErrHeaderNotFound returns.
 func (m PutRequest) SignedData() ([]byte, error) {
-	r := m.GetR()
-	if r == nil {
+	sz := m.SignedDataSize()
+	if sz < 0 {
 		return nil, ErrHeaderNotFound
 	}
 
-	data := make([]byte, r.Size())
+	data := make([]byte, sz)
 
-	if _, err := r.MarshalTo(data); err != nil {
-		return nil, err
-	}
-
-	return data, nil
+	return data, m.ReadSignedData(data)
 }
 
-// ReadSignedData copies marshaled payload of the Put request to passed buffer.
+// ReadSignedData copies payload bytes to passed buffer.
 //
-// If payload is nil, ErrHeaderNotFound returns.
+// If the buffer size is insufficient, io.ErrUnexpectedEOF returns.
 func (m PutRequest) ReadSignedData(p []byte) error {
 	r := m.GetR()
 	if r == nil {
@@ -48,16 +44,14 @@ func (m PutRequest) SignedDataSize() int {
 	return r.Size()
 }
 
-// SignedData returns marshaled Address field.
-//
-// Resulting error is always nil.
+// SignedData returns payload bytes of the request.
 func (m GetRequest) SignedData() ([]byte, error) {
-	addr := m.GetAddress()
+	data := make([]byte, m.SignedDataSize())
 
-	return addressBytes(addr), nil
+	return data, m.ReadSignedData(data)
 }
 
-// ReadSignedData copies marshaled Address field to passed buffer.
+// ReadSignedData copies payload bytes to passed buffer.
 //
 // If the buffer size is insufficient, io.ErrUnexpectedEOF returns.
 func (m GetRequest) ReadSignedData(p []byte) error {
@@ -74,29 +68,19 @@ func (m GetRequest) ReadSignedData(p []byte) error {
 	return nil
 }
 
-// SignedDataSize returns the size of object address.
+// SignedDataSize returns payload size of the request.
 func (m GetRequest) SignedDataSize() int {
 	return addressSize(m.GetAddress())
 }
 
-// SignedData returns marshaled Address field.
-//
-// Resulting error is always nil.
+// SignedData returns payload bytes of the request.
 func (m HeadRequest) SignedData() ([]byte, error) {
-	sz := addressSize(m.Address)
+	data := make([]byte, m.SignedDataSize())
 
-	data := make([]byte, sz+1)
-
-	if m.GetFullHeaders() {
-		data[0] = 1
-	}
-
-	copy(data[1:], addressBytes(m.Address))
-
-	return data, nil
+	return data, m.ReadSignedData(data)
 }
 
-// ReadSignedData copies marshaled Address field to passed buffer.
+// ReadSignedData copies payload bytes to passed buffer.
 //
 // If the buffer size is insufficient, io.ErrUnexpectedEOF returns.
 func (m HeadRequest) ReadSignedData(p []byte) error {
@@ -115,11 +99,38 @@ func (m HeadRequest) ReadSignedData(p []byte) error {
 	return nil
 }
 
-// SignedDataSize returns the size of object address.
+// SignedDataSize returns payload size of the request.
 func (m HeadRequest) SignedDataSize() int {
 	return addressSize(m.Address) + 1
 }
 
+// SignedData returns payload bytes of the request.
+func (m DeleteRequest) SignedData() ([]byte, error) {
+	data := make([]byte, m.SignedDataSize())
+
+	return data, m.ReadSignedData(data)
+}
+
+// ReadSignedData copies payload bytes to passed buffer.
+//
+// If the buffer size is insufficient, io.ErrUnexpectedEOF returns.
+func (m DeleteRequest) ReadSignedData(p []byte) error {
+	if len(p) < m.SignedDataSize() {
+		return io.ErrUnexpectedEOF
+	}
+
+	off := copy(p, m.OwnerID.Bytes())
+
+	copy(p[off:], addressBytes(m.Address))
+
+	return nil
+}
+
+// SignedDataSize returns payload size of the request.
+func (m DeleteRequest) SignedDataSize() int {
+	return m.OwnerID.Size() + addressSize(m.Address)
+}
+
 func addressSize(addr Address) int {
 	return addr.CID.Size() + addr.ObjectID.Size()
 }
diff --git a/object/sign_test.go b/object/sign_test.go
index c5159c6..d6357b9 100644
--- a/object/sign_test.go
+++ b/object/sign_test.go
@@ -73,6 +73,22 @@ func TestSignVerifyRequests(t *testing.T) {
 				},
 			},
 		},
+		{ // DeleteRequest
+			constructor: func() sigType {
+				return new(DeleteRequest)
+			},
+			payloadCorrupt: []func(sigType){
+				func(s sigType) {
+					s.(*DeleteRequest).OwnerID[0]++
+				},
+				func(s sigType) {
+					s.(*DeleteRequest).Address.CID[0]++
+				},
+				func(s sigType) {
+					s.(*DeleteRequest).Address.ObjectID[0]++
+				},
+			},
+		},
 	}
 
 	for _, item := range items {