From 8c3f8268b921e1f6fcf87627997112de73e09ffd Mon Sep 17 00:00:00 2001 From: Leonard Lyubich Date: Tue, 1 Sep 2020 16:31:57 +0300 Subject: [PATCH] [#130] sdk: Define object types Define Object type that represents the immutable NeoFS object. Define RawObject type that provides interface for convenient construction of NeoFS object. Signed-off-by: Leonard Lyubich --- pkg/object/object.go | 200 +++++++++++++++++++++++++++++++++++++++++++ pkg/object/raw.go | 53 ++++++++++++ 2 files changed, 253 insertions(+) create mode 100644 pkg/object/object.go create mode 100644 pkg/object/raw.go diff --git a/pkg/object/object.go b/pkg/object/object.go new file mode 100644 index 00000000..d86c965b --- /dev/null +++ b/pkg/object/object.go @@ -0,0 +1,200 @@ +package object + +import ( + "bytes" + "crypto/ecdsa" + "crypto/sha256" + + "github.com/nspcc-dev/neofs-api-go/pkg/container" + "github.com/nspcc-dev/neofs-api-go/pkg/owner" + "github.com/nspcc-dev/neofs-api-go/util/signature" + "github.com/nspcc-dev/neofs-api-go/v2/object" + "github.com/nspcc-dev/neofs-api-go/v2/refs" + signatureV2 "github.com/nspcc-dev/neofs-api-go/v2/signature" + crypto "github.com/nspcc-dev/neofs-crypto" + "github.com/pkg/errors" +) + +// Object represents NeoFS object that provides +// a convenient interface for working in isolation +// from the internal structure of an object. +// +// Object allows to work with the object in read-only +// mode as a reflection of the immutability of objects +// in the system. +type Object struct { + rwObject +} + +type rwObject struct { + fin bool + + id *ID + + key, sig []byte + + cid *container.ID + + ownerID *owner.ID + + payloadChecksum *refs.Checksum + + // TODO: add other fields +} + +// Verify checks if object structure is correct. +func (o *Object) Verify() error { + if o == nil { + return nil + } + + hdr := o.v2Header() + + data, err := hdr.StableMarshal(nil) + if err != nil { + return errors.Wrap(err, "could not marshal header") + } + + hdrChecksum := sha256.Sum256(data) + + if !bytes.Equal(hdrChecksum[:], o.id.val) { + return errors.New("invalid object identifier") + } + + if err := signature.VerifyDataWithSource( + signatureV2.StableMarshalerWrapper{ + SM: o.id.ToV2(), + }, + func() (key, sig []byte) { + return o.key, o.sig + }, + ); err != nil { + return errors.Wrap(err, "invalid object ID signature") + } + + return nil +} + +// ToV2 converts object to v2 Object message. +func (o *Object) ToV2() *object.Object { + obj, _ := o.rwObject.ToV2(nil) + + return obj +} + +func (o *rwObject) v2Header() *object.Header { + hV2 := new(object.Header) + hV2.SetContainerID(o.cid.ToV2()) + hV2.SetOwnerID(o.ownerID.ToV2()) + hV2.SetPayloadHash(o.payloadChecksum) + // TODO: set other fields + + return hV2 +} + +func (o *rwObject) complete(key *ecdsa.PrivateKey) (*object.Header, error) { + hdr := o.v2Header() + + hdrData, err := hdr.StableMarshal(nil) + if err != nil { + return nil, errors.Wrap(err, "could not marshal header") + } + + o.id = new(ID) + o.id.SetSHA256(sha256.Sum256(hdrData)) + + if err := signature.SignDataWithHandler( + key, + signatureV2.StableMarshalerWrapper{ + SM: o.id.ToV2(), + }, + func(key []byte, sig []byte) { + o.key, o.sig = key, sig + }, + ); err != nil { + return nil, errors.Wrap(err, "could sign object identifier") + } + + o.fin = true + + return hdr, nil +} + +// ToV2 calculates object identifier, signs structure and converts +// it to v2 Object message. +func (o *rwObject) ToV2(key *ecdsa.PrivateKey) (*object.Object, error) { + if o == nil { + return nil, nil + } + + var ( + hdr *object.Header + err error + ) + + if !o.fin { + if key == nil { + return nil, errors.Wrap(crypto.ErrEmptyPrivateKey, "could complete the object") + } + + if hdr, err = o.complete(key); err != nil { + return nil, errors.Wrapf(err, "could not complete the object") + } + } else { + hdr = o.v2Header() + } + + obj := new(object.Object) + obj.SetObjectID(o.id.ToV2()) + obj.SetHeader(hdr) + + sig := new(refs.Signature) + sig.SetKey(o.key) + sig.SetSign(o.sig) + obj.SetSignature(sig) + + return obj, nil +} + +// FromV2 converts v2 Object message to Object instance. +// +// Returns any error encountered which prevented the +// recovery of object data from the message. +func FromV2(oV2 *object.Object) (*Object, error) { + if oV2 == nil { + return nil, nil + } + + id, err := IDFromV2(oV2.GetObjectID()) + if err != nil { + return nil, errors.Wrap(err, "could not convert object ID") + } + + hdr := oV2.GetHeader() + + ownerID, err := owner.IDFromV2(hdr.GetOwnerID()) + if err != nil { + return nil, errors.Wrap(err, "could not convert owner ID") + } + + cid, err := container.IDFromV2(hdr.GetContainerID()) + if err != nil { + return nil, errors.Wrap(err, "could not convert container ID") + } + + // TODO: convert other fields + + sig := oV2.GetSignature() + + return &Object{ + rwObject: rwObject{ + fin: true, + id: id, + key: sig.GetKey(), + sig: sig.GetSign(), + cid: cid, + ownerID: ownerID, + payloadChecksum: hdr.GetPayloadHash(), + }, + }, nil +} diff --git a/pkg/object/raw.go b/pkg/object/raw.go new file mode 100644 index 00000000..c82ba5cc --- /dev/null +++ b/pkg/object/raw.go @@ -0,0 +1,53 @@ +package object + +import ( + "crypto/sha256" + + "github.com/nspcc-dev/neofs-api-go/pkg/container" + "github.com/nspcc-dev/neofs-api-go/pkg/owner" + "github.com/nspcc-dev/neofs-api-go/v2/refs" +) + +// RawObject represents NeoFS object that provides +// a convenient interface to fill in the fields of +// an object in isolation from its internal structure. +type RawObject struct { + rwObject +} + +func (o *RawObject) set(setter func()) { + o.fin = false + setter() +} + +// SetContainerID sets object's container identifier. +func (o *RawObject) SetContainerID(v *container.ID) { + if o != nil { + o.set(func() { + o.cid = v + }) + } +} + +// SetOwnerID sets identifier of the object's owner. +func (o *RawObject) SetOwnerID(v *owner.ID) { + if o != nil { + o.set(func() { + o.ownerID = v + }) + } +} + +// SetPayloadChecksumSHA256 sets payload checksum as a SHA256 checksum. +func (o *RawObject) SetPayloadChecksumSHA256(v [sha256.Size]byte) { + if o != nil { + o.set(func() { + if o.payloadChecksum == nil { + o.payloadChecksum = new(refs.Checksum) + } + + o.payloadChecksum.SetType(refs.SHA256) + o.payloadChecksum.SetSum(v[:]) + }) + } +}