From e8b95ebf3e882a2342005cb812ba3d04bd74aa34 Mon Sep 17 00:00:00 2001
From: Evgeniy Stratonikov <evgeniy@nspcc.ru>
Date: Fri, 12 Mar 2021 14:19:04 +0300
Subject: [PATCH] oracle/neofs: change result types

1. `GetHash` returnes '0x'-prefixed little-endian hex-string.
2. `GetPayload` and `GetRange` assume result is UTF-8.
---
 pkg/services/oracle/neofs/neofs.go      | 19 ++++++++++++++++---
 pkg/services/oracle/neofs/neofs_test.go | 10 ++++++++++
 2 files changed, 26 insertions(+), 3 deletions(-)

diff --git a/pkg/services/oracle/neofs/neofs.go b/pkg/services/oracle/neofs/neofs.go
index 326853590..a767e5467 100644
--- a/pkg/services/oracle/neofs/neofs.go
+++ b/pkg/services/oracle/neofs/neofs.go
@@ -9,9 +9,11 @@ import (
 	"net/url"
 	"strconv"
 	"strings"
+	"unicode/utf8"
 
 	"github.com/gogo/protobuf/jsonpb"
 	"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
+	"github.com/nspcc-dev/neo-go/pkg/util"
 	"github.com/nspcc-dev/neofs-api-go/pkg/client"
 	"github.com/nspcc-dev/neofs-api-go/pkg/container"
 	"github.com/nspcc-dev/neofs-api-go/pkg/object"
@@ -104,7 +106,7 @@ func getPayload(ctx context.Context, c *client.Client, addr *object.Address) ([]
 	if err != nil {
 		return nil, err
 	}
-	return obj.Payload(), nil
+	return checkUTF8(obj.Payload())
 }
 
 func getRange(ctx context.Context, c *client.Client, addr *object.Address, ps ...string) ([]byte, error) {
@@ -115,7 +117,11 @@ func getRange(ctx context.Context, c *client.Client, addr *object.Address, ps ..
 	if err != nil {
 		return nil, err
 	}
-	return c.ObjectPayloadRangeData(ctx, new(client.RangeDataParams).WithAddress(addr).WithRange(r))
+	data, err := c.ObjectPayloadRangeData(ctx, new(client.RangeDataParams).WithAddress(addr).WithRange(r))
+	if err != nil {
+		return nil, err
+	}
+	return checkUTF8(data)
 }
 
 func getHeader(ctx context.Context, c *client.Client, addr *object.Address) ([]byte, error) {
@@ -149,7 +155,7 @@ func getHash(ctx context.Context, c *client.Client, addr *object.Address, ps ...
 	if len(hashes) == 0 {
 		return nil, fmt.Errorf("%w: empty response", ErrInvalidRange)
 	}
-	return hashes[0][:], nil
+	return util.Uint256(hashes[0]).MarshalJSON()
 }
 
 func parseRange(s string) (*object.Range, error) {
@@ -171,3 +177,10 @@ func parseRange(s string) (*object.Range, error) {
 	r.SetLength(length)
 	return r, nil
 }
+
+func checkUTF8(v []byte) ([]byte, error) {
+	if !utf8.Valid(v) {
+		return nil, errors.New("invalid UTF-8")
+	}
+	return v, nil
+}
diff --git a/pkg/services/oracle/neofs/neofs_test.go b/pkg/services/oracle/neofs/neofs_test.go
index 7685bec27..8f13cead5 100644
--- a/pkg/services/oracle/neofs/neofs_test.go
+++ b/pkg/services/oracle/neofs/neofs_test.go
@@ -81,3 +81,13 @@ func TestParseNeoFSURL(t *testing.T) {
 
 	}
 }
+
+func Test_checkUTF8(t *testing.T) {
+	_, err := checkUTF8([]byte{0xFF})
+	require.Error(t, err)
+
+	a := []byte{1, 2, 3}
+	b, err := checkUTF8(a)
+	require.NoError(t, err)
+	require.Equal(t, a, b)
+}