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

---
 object/sign.go      | 76 +++++++++++++++++++++++++++++++++++++++++++++
 object/sign_test.go | 28 +++++++++++++++++
 2 files changed, 104 insertions(+)

diff --git a/object/sign.go b/object/sign.go
index 506fc95..3b97d62 100644
--- a/object/sign.go
+++ b/object/sign.go
@@ -1,6 +1,7 @@
 package object
 
 import (
+	"encoding/binary"
 	"io"
 )
 
@@ -161,6 +162,81 @@ func (m GetRangeRequest) SignedDataSize() int {
 	return (&m.Range).Size() + addressSize(m.GetAddress())
 }
 
+// SignedData returns payload bytes of the request.
+func (m GetRangeHashRequest) 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 GetRangeHashRequest) ReadSignedData(p []byte) error {
+	if len(p) < m.SignedDataSize() {
+		return io.ErrUnexpectedEOF
+	}
+
+	var off int
+
+	off += copy(p[off:], addressBytes(m.GetAddress()))
+
+	off += copy(p[off:], sliceBytes(m.GetSalt()))
+
+	copy(p[off:], rangeSetBytes(m.GetRanges()))
+
+	return nil
+}
+
+// SignedDataSize returns payload size of the request.
+func (m GetRangeHashRequest) SignedDataSize() int {
+	var sz int
+
+	sz += addressSize(m.GetAddress())
+
+	sz += rangeSetSize(m.GetRanges())
+
+	sz += sliceSize(m.GetSalt())
+
+	return sz
+}
+
+func sliceSize(v []byte) int {
+	return 4 + len(v)
+}
+
+func sliceBytes(v []byte) []byte {
+	data := make([]byte, sliceSize(v))
+
+	binary.BigEndian.PutUint32(data, uint32(len(v)))
+
+	copy(data[4:], v)
+
+	return data
+}
+
+func rangeSetSize(rs []Range) int {
+	return 4 + len(rs)*16 // two uint64 fields
+}
+
+func rangeSetBytes(rs []Range) []byte {
+	data := make([]byte, rangeSetSize(rs))
+
+	binary.BigEndian.PutUint32(data, uint32(len(rs)))
+
+	off := 4
+
+	for i := range rs {
+		binary.BigEndian.PutUint64(data[off:], rs[i].Offset)
+		off += 8
+
+		binary.BigEndian.PutUint64(data[off:], rs[i].Length)
+		off += 8
+	}
+
+	return data
+}
+
 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 f8450ef..8b9a011 100644
--- a/object/sign_test.go
+++ b/object/sign_test.go
@@ -108,6 +108,34 @@ func TestSignVerifyRequests(t *testing.T) {
 				},
 			},
 		},
+		{ // GetRangeHashRequest
+			constructor: func() sigType {
+				return &GetRangeHashRequest{
+					Ranges: []Range{{}},
+					Salt:   []byte{1, 2, 3},
+				}
+			},
+			payloadCorrupt: []func(sigType){
+				func(s sigType) {
+					s.(*GetRangeHashRequest).Address.CID[0]++
+				},
+				func(s sigType) {
+					s.(*GetRangeHashRequest).Address.ObjectID[0]++
+				},
+				func(s sigType) {
+					s.(*GetRangeHashRequest).Salt[0]++
+				},
+				func(s sigType) {
+					s.(*GetRangeHashRequest).Ranges[0].Length++
+				},
+				func(s sigType) {
+					s.(*GetRangeHashRequest).Ranges[0].Offset++
+				},
+				func(s sigType) {
+					s.(*GetRangeHashRequest).Ranges = nil
+				},
+			},
+		},
 	}
 
 	for _, item := range items {