1c30414a6c
Core changes: * avoid package-colliding variable naming * avoid using pointers to IDs where unnecessary * avoid using `idSDK` import alias pattern * use `EncodeToString` for protocol string calculation and `String` for printing Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
120 lines
3.4 KiB
Go
120 lines
3.4 KiB
Go
package container
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/nspcc-dev/neofs-api-go/v2/refs"
|
|
containerContract "github.com/nspcc-dev/neofs-contract/container"
|
|
core "github.com/nspcc-dev/neofs-node/pkg/core/container"
|
|
"github.com/nspcc-dev/neofs-node/pkg/morph/client"
|
|
apistatus "github.com/nspcc-dev/neofs-sdk-go/client/status"
|
|
"github.com/nspcc-dev/neofs-sdk-go/container"
|
|
cid "github.com/nspcc-dev/neofs-sdk-go/container/id"
|
|
neofscrypto "github.com/nspcc-dev/neofs-sdk-go/crypto"
|
|
"github.com/nspcc-dev/neofs-sdk-go/session"
|
|
)
|
|
|
|
type containerSource Client
|
|
|
|
func (x *containerSource) Get(cnr cid.ID) (*container.Container, error) {
|
|
return Get((*Client)(x), cnr)
|
|
}
|
|
|
|
// AsContainerSource provides container Source interface
|
|
// from Wrapper instance.
|
|
func AsContainerSource(w *Client) core.Source {
|
|
return (*containerSource)(w)
|
|
}
|
|
|
|
// Get marshals container ID, and passes it to Wrapper's Get method.
|
|
func Get(c *Client, cnr cid.ID) (*container.Container, error) {
|
|
binCnr := make([]byte, sha256.Size)
|
|
cnr.Encode(binCnr)
|
|
|
|
return c.Get(binCnr)
|
|
}
|
|
|
|
// Get reads the container from NeoFS system by binary identifier
|
|
// through Container contract call.
|
|
//
|
|
// If an empty slice is returned for the requested identifier,
|
|
// storage.ErrNotFound error is returned.
|
|
func (c *Client) Get(cid []byte) (*container.Container, error) {
|
|
prm := client.TestInvokePrm{}
|
|
prm.SetMethod(getMethod)
|
|
prm.SetArgs(cid)
|
|
|
|
res, err := c.client.TestInvoke(prm)
|
|
if err != nil {
|
|
if strings.Contains(err.Error(), containerContract.NotFoundError) {
|
|
var errNotFound apistatus.ContainerNotFound
|
|
|
|
return nil, errNotFound
|
|
}
|
|
return nil, fmt.Errorf("could not perform test invocation (%s): %w", getMethod, err)
|
|
} else if ln := len(res); ln != 1 {
|
|
return nil, fmt.Errorf("unexpected stack item count (%s): %d", getMethod, ln)
|
|
}
|
|
|
|
arr, err := client.ArrayFromStackItem(res[0])
|
|
if err != nil {
|
|
return nil, fmt.Errorf("could not get item array of container (%s): %w", getMethod, err)
|
|
}
|
|
|
|
if len(arr) != 4 {
|
|
return nil, fmt.Errorf("unexpected container stack item count (%s): %d", getMethod, len(arr))
|
|
}
|
|
|
|
cnrBytes, err := client.BytesFromStackItem(arr[0])
|
|
if err != nil {
|
|
return nil, fmt.Errorf("could not get byte array of container (%s): %w", getMethod, err)
|
|
}
|
|
|
|
sigBytes, err := client.BytesFromStackItem(arr[1])
|
|
if err != nil {
|
|
return nil, fmt.Errorf("could not get byte array of container signature (%s): %w", getMethod, err)
|
|
}
|
|
|
|
pub, err := client.BytesFromStackItem(arr[2])
|
|
if err != nil {
|
|
return nil, fmt.Errorf("could not get byte array of public key (%s): %w", getMethod, err)
|
|
}
|
|
|
|
tokBytes, err := client.BytesFromStackItem(arr[3])
|
|
if err != nil {
|
|
return nil, fmt.Errorf("could not get byte array of session token (%s): %w", getMethod, err)
|
|
}
|
|
|
|
cnr := container.New()
|
|
if err := cnr.Unmarshal(cnrBytes); err != nil {
|
|
// use other major version if there any
|
|
return nil, fmt.Errorf("can't unmarshal container: %w", err)
|
|
}
|
|
|
|
if len(tokBytes) > 0 {
|
|
var tok session.Container
|
|
|
|
err = tok.Unmarshal(tokBytes)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("could not unmarshal session token: %w", err)
|
|
}
|
|
|
|
cnr.SetSessionToken(&tok)
|
|
}
|
|
|
|
// FIXME(@cthulhu-rider): #1387 temp solution, later table structure won't have a signature
|
|
|
|
var sigV2 refs.Signature
|
|
sigV2.SetKey(pub)
|
|
sigV2.SetSign(sigBytes)
|
|
sigV2.SetScheme(refs.ECDSA_RFC6979_SHA256)
|
|
|
|
var sig neofscrypto.Signature
|
|
sig.ReadFromV2(sigV2)
|
|
|
|
cnr.SetSignature(&sig)
|
|
|
|
return cnr, nil
|
|
}
|