From db12420c99eaf2f3c03743e2d927d37aa16009d6 Mon Sep 17 00:00:00 2001 From: Alex Vanin Date: Thu, 20 Aug 2020 12:43:47 +0300 Subject: [PATCH] Move version and signature structures to refs package Signed-off-by: Alex Vanin --- v2/container/convert.go | 20 ++++----- v2/container/grpc/service.go | 8 ++-- v2/container/grpc/types.go | 3 +- v2/container/marshal_test.go | 7 ++- v2/container/types.go | 30 ++++++------- v2/object/convert.go | 24 +++++----- v2/object/grpc/service.go | 4 +- v2/object/grpc/types.go | 8 ++-- v2/object/marshal_test.go | 8 ++-- v2/object/types.go | 36 +++++++-------- v2/refs/convert.go | 52 ++++++++++++++++++++++ v2/refs/grpc/types.go | 28 ++++++++++++ v2/refs/marshal.go | 86 ++++++++++++++++++++++++++++++++++++ v2/refs/marshal_test.go | 48 ++++++++++++++++++++ v2/refs/types.go | 64 +++++++++++++++++++++++++++ v2/signature/sign.go | 23 +++++----- 16 files changed, 363 insertions(+), 86 deletions(-) diff --git a/v2/container/convert.go b/v2/container/convert.go index b5d829a..34d5b97 100644 --- a/v2/container/convert.go +++ b/v2/container/convert.go @@ -43,7 +43,7 @@ func ContainerToGRPCMessage(c *Container) *container.Container { m := new(container.Container) m.SetVersion( - service.VersionToGRPCMessage(c.GetVersion()), + refs.VersionToGRPCMessage(c.GetVersion()), ) m.SetOwnerId( @@ -78,7 +78,7 @@ func ContainerFromGRPCMessage(m *container.Container) *Container { c := new(Container) c.SetVersion( - service.VersionFromGRPCMessage(m.GetVersion()), + refs.VersionFromGRPCMessage(m.GetVersion()), ) c.SetOwnerID( @@ -117,7 +117,7 @@ func PutRequestBodyToGRPCMessage(r *PutRequestBody) *container.PutRequest_Body { ) m.SetSignature( - service.SignatureToGRPCMessage(r.GetSignature()), + refs.SignatureToGRPCMessage(r.GetSignature()), ) return m @@ -135,7 +135,7 @@ func PutRequestBodyFromGRPCMessage(m *container.PutRequest_Body) *PutRequestBody ) r.SetSignature( - service.SignatureFromGRPCMessage(m.GetSignature()), + refs.SignatureFromGRPCMessage(m.GetSignature()), ) return r @@ -365,7 +365,7 @@ func DeleteRequestBodyToGRPCMessage(r *DeleteRequestBody) *container.DeleteReque ) m.SetSignature( - service.SignatureToGRPCMessage(r.GetSignature()), + refs.SignatureToGRPCMessage(r.GetSignature()), ) return m @@ -383,7 +383,7 @@ func DeleteRequestBodyFromGRPCMessage(m *container.DeleteRequest_Body) *DeleteRe ) r.SetSignature( - service.SignatureFromGRPCMessage(m.GetSignature()), + refs.SignatureFromGRPCMessage(m.GetSignature()), ) return r @@ -615,7 +615,7 @@ func SetExtendedACLRequestBodyToGRPCMessage(r *SetExtendedACLRequestBody) *conta ) m.SetSignature( - service.SignatureToGRPCMessage(r.GetSignature())) + refs.SignatureToGRPCMessage(r.GetSignature())) return m } @@ -632,7 +632,7 @@ func SetExtendedACLRequestBodyFromGRPCMessage(m *container.SetExtendedACLRequest ) r.SetSignature( - service.SignatureFromGRPCMessage(m.GetSignature()), + refs.SignatureFromGRPCMessage(m.GetSignature()), ) return r @@ -794,7 +794,7 @@ func GetExtendedACLResponseBodyToGRPCMessage(r *GetExtendedACLResponseBody) *con ) m.SetSignature( - service.SignatureToGRPCMessage(r.GetSignature()), + refs.SignatureToGRPCMessage(r.GetSignature()), ) return m @@ -812,7 +812,7 @@ func GetExtendedACLResponseBodyFromGRPCMessage(m *container.GetExtendedACLRespon ) r.SetSignature( - service.SignatureFromGRPCMessage(m.GetSignature()), + refs.SignatureFromGRPCMessage(m.GetSignature()), ) return r diff --git a/v2/container/grpc/service.go b/v2/container/grpc/service.go index f92501f..4da5e0f 100644 --- a/v2/container/grpc/service.go +++ b/v2/container/grpc/service.go @@ -14,7 +14,7 @@ func (m *PutRequest_Body) SetContainer(v *Container) { } // SetSignature sets signature of the container structure. -func (m *PutRequest_Body) SetSignature(v *service.Signature) { +func (m *PutRequest_Body) SetSignature(v *refs.Signature) { if m != nil { m.Signature = v } @@ -77,7 +77,7 @@ func (m *DeleteRequest_Body) SetContainerId(v *refs.ContainerID) { } // SetSignature sets signature of the container identifier. -func (m *DeleteRequest_Body) SetSignature(v *service.Signature) { +func (m *DeleteRequest_Body) SetSignature(v *refs.Signature) { if m != nil { m.Signature = v } @@ -243,7 +243,7 @@ func (m *SetExtendedACLRequest_Body) SetEacl(v *acl.EACLTable) { } // SetSignature sets signature of the eACL table. -func (m *SetExtendedACLRequest_Body) SetSignature(v *service.Signature) { +func (m *SetExtendedACLRequest_Body) SetSignature(v *refs.Signature) { if m != nil { m.Signature = v } @@ -327,7 +327,7 @@ func (m *GetExtendedACLResponse_Body) SetEacl(v *acl.EACLTable) { } // SetSignature sets signature of the eACL table. -func (m *GetExtendedACLResponse_Body) SetSignature(v *service.Signature) { +func (m *GetExtendedACLResponse_Body) SetSignature(v *refs.Signature) { if m != nil { m.Signature = v } diff --git a/v2/container/grpc/types.go b/v2/container/grpc/types.go index 5e51e3c..9fed6b2 100644 --- a/v2/container/grpc/types.go +++ b/v2/container/grpc/types.go @@ -3,7 +3,6 @@ package container import ( netmap "github.com/nspcc-dev/neofs-api-go/v2/netmap/grpc" refs "github.com/nspcc-dev/neofs-api-go/v2/refs/grpc" - service "github.com/nspcc-dev/neofs-api-go/v2/service/grpc" ) // SetKey sets key to the container attribute. @@ -56,7 +55,7 @@ func (m *Container) SetPlacementPolicy(v *netmap.PlacementPolicy) { } // SetVersion sets version of the container. -func (m *Container) SetVersion(v *service.Version) { +func (m *Container) SetVersion(v *refs.Version) { if m != nil { m.Version = v } diff --git a/v2/container/marshal_test.go b/v2/container/marshal_test.go index f46bb31..19e5d6e 100644 --- a/v2/container/marshal_test.go +++ b/v2/container/marshal_test.go @@ -8,7 +8,6 @@ import ( "github.com/nspcc-dev/neofs-api-go/v2/container" grpc "github.com/nspcc-dev/neofs-api-go/v2/container/grpc" "github.com/nspcc-dev/neofs-api-go/v2/refs" - "github.com/nspcc-dev/neofs-api-go/v2/service" "github.com/stretchr/testify/require" ) @@ -248,7 +247,7 @@ func generateContainer(n string) *container.Container { owner := new(refs.OwnerID) owner.SetValue([]byte("Owner ID")) - version := new(service.Version) + version := new(refs.Version) version.SetMajor(2) version.SetMinor(0) @@ -267,8 +266,8 @@ func generateContainer(n string) *container.Container { return cnr } -func generateSignature(k, v string) *service.Signature { - sig := new(service.Signature) +func generateSignature(k, v string) *refs.Signature { + sig := new(refs.Signature) sig.SetKey([]byte(k)) sig.SetSign([]byte(v)) diff --git a/v2/container/types.go b/v2/container/types.go index 033273f..79d97c4 100644 --- a/v2/container/types.go +++ b/v2/container/types.go @@ -12,7 +12,7 @@ type Attribute struct { } type Container struct { - version *service.Version + version *refs.Version ownerID *refs.OwnerID @@ -28,7 +28,7 @@ type Container struct { type PutRequestBody struct { cnr *Container - sig *service.Signature + sig *refs.Signature } type PutRequest struct { @@ -78,7 +78,7 @@ type GetResponse struct { type DeleteRequestBody struct { cid *refs.ContainerID - sig *service.Signature + sig *refs.Signature } type DeleteRequest struct { @@ -126,7 +126,7 @@ type ListResponse struct { type SetExtendedACLRequestBody struct { eacl *acl.Table - sig *service.Signature + sig *refs.Signature } type SetExtendedACLRequest struct { @@ -162,7 +162,7 @@ type GetExtendedACLRequest struct { type GetExtendedACLResponseBody struct { eacl *acl.Table - sig *service.Signature + sig *refs.Signature } type GetExtendedACLResponse struct { @@ -201,7 +201,7 @@ func (a *Attribute) SetValue(v string) { } } -func (c *Container) GetVersion() *service.Version { +func (c *Container) GetVersion() *refs.Version { if c != nil { return c.version } @@ -209,7 +209,7 @@ func (c *Container) GetVersion() *service.Version { return nil } -func (c *Container) SetVersion(v *service.Version) { +func (c *Container) SetVersion(v *refs.Version) { if c != nil { c.version = v } @@ -299,7 +299,7 @@ func (r *PutRequestBody) SetContainer(v *Container) { } } -func (r *PutRequestBody) GetSignature() *service.Signature { +func (r *PutRequestBody) GetSignature() *refs.Signature { if r != nil { return r.sig } @@ -307,7 +307,7 @@ func (r *PutRequestBody) GetSignature() *service.Signature { return nil } -func (r *PutRequestBody) SetSignature(v *service.Signature) { +func (r *PutRequestBody) SetSignature(v *refs.Signature) { if r != nil { r.sig = v } @@ -537,7 +537,7 @@ func (r *DeleteRequestBody) SetContainerID(v *refs.ContainerID) { } } -func (r *DeleteRequestBody) GetSignature() *service.Signature { +func (r *DeleteRequestBody) GetSignature() *refs.Signature { if r != nil { return r.sig } @@ -545,7 +545,7 @@ func (r *DeleteRequestBody) GetSignature() *service.Signature { return nil } -func (r *DeleteRequestBody) SetSignature(v *service.Signature) { +func (r *DeleteRequestBody) SetSignature(v *refs.Signature) { if r != nil { r.sig = v } @@ -761,7 +761,7 @@ func (r *SetExtendedACLRequestBody) SetEACL(v *acl.Table) { } } -func (r *SetExtendedACLRequestBody) GetSignature() *service.Signature { +func (r *SetExtendedACLRequestBody) GetSignature() *refs.Signature { if r != nil { return r.sig } @@ -769,7 +769,7 @@ func (r *SetExtendedACLRequestBody) GetSignature() *service.Signature { return nil } -func (r *SetExtendedACLRequestBody) SetSignature(v *service.Signature) { +func (r *SetExtendedACLRequestBody) SetSignature(v *refs.Signature) { if r != nil { r.sig = v } @@ -929,7 +929,7 @@ func (r *GetExtendedACLResponseBody) SetEACL(v *acl.Table) { } } -func (r *GetExtendedACLResponseBody) GetSignature() *service.Signature { +func (r *GetExtendedACLResponseBody) GetSignature() *refs.Signature { if r != nil { return r.sig } @@ -937,7 +937,7 @@ func (r *GetExtendedACLResponseBody) GetSignature() *service.Signature { return nil } -func (r *GetExtendedACLResponseBody) SetSignature(v *service.Signature) { +func (r *GetExtendedACLResponseBody) SetSignature(v *refs.Signature) { if r != nil { r.sig = v } diff --git a/v2/object/convert.go b/v2/object/convert.go index 3f7b85a..949bb13 100644 --- a/v2/object/convert.go +++ b/v2/object/convert.go @@ -33,7 +33,7 @@ func ShortHeaderToGRPCMessage(h *ShortHeader) *object.ShortHeader { m := new(object.ShortHeader) m.SetVersion( - service.VersionToGRPCMessage(h.GetVersion()), + refs.VersionToGRPCMessage(h.GetVersion()), ) m.SetCreationEpoch(h.GetCreationEpoch()) @@ -59,7 +59,7 @@ func ShortHeaderFromGRPCMessage(m *object.ShortHeader) *ShortHeader { h := new(ShortHeader) h.SetVersion( - service.VersionFromGRPCMessage(m.GetVersion()), + refs.VersionFromGRPCMessage(m.GetVersion()), ) h.SetCreationEpoch(m.GetCreationEpoch()) @@ -119,7 +119,7 @@ func SplitHeaderToGRPCMessage(h *SplitHeader) *object.Header_Split { ) m.SetParentSignature( - service.SignatureToGRPCMessage(h.GetParentSignature()), + refs.SignatureToGRPCMessage(h.GetParentSignature()), ) m.SetParentHeader( @@ -154,7 +154,7 @@ func SplitHeaderFromGRPCMessage(m *object.Header_Split) *SplitHeader { ) h.SetParentSignature( - service.SignatureFromGRPCMessage(m.GetParentSignature()), + refs.SignatureFromGRPCMessage(m.GetParentSignature()), ) h.SetParentHeader( @@ -181,7 +181,7 @@ func HeaderToGRPCMessage(h *Header) *object.Header { m := new(object.Header) m.SetVersion( - service.VersionToGRPCMessage(h.GetVersion()), + refs.VersionToGRPCMessage(h.GetVersion()), ) m.SetContainerId( @@ -236,7 +236,7 @@ func HeaderFromGRPCMessage(m *object.Header) *Header { h := new(Header) h.SetVersion( - service.VersionFromGRPCMessage(m.GetVersion()), + refs.VersionFromGRPCMessage(m.GetVersion()), ) h.SetContainerID( @@ -295,7 +295,7 @@ func ObjectToGRPCMessage(o *Object) *object.Object { ) m.SetSignature( - service.SignatureToGRPCMessage(o.GetSignature()), + refs.SignatureToGRPCMessage(o.GetSignature()), ) m.SetHeader( @@ -319,7 +319,7 @@ func ObjectFromGRPCMessage(m *object.Object) *Object { ) o.SetSignature( - service.SignatureFromGRPCMessage(m.GetSignature()), + refs.SignatureFromGRPCMessage(m.GetSignature()), ) o.SetHeader( @@ -407,7 +407,7 @@ func GetObjectPartInitToGRPCMessage(r *GetObjectPartInit) *object.GetResponse_Bo ) m.SetSignature( - service.SignatureToGRPCMessage(r.GetSignature()), + refs.SignatureToGRPCMessage(r.GetSignature()), ) m.SetHeader( @@ -429,7 +429,7 @@ func GetObjectPartInitFromGRPCMessage(m *object.GetResponse_Body_Init) *GetObjec ) r.SetSignature( - service.SignatureFromGRPCMessage(m.GetSignature()), + refs.SignatureFromGRPCMessage(m.GetSignature()), ) r.SetHeader( @@ -555,7 +555,7 @@ func PutObjectPartInitToGRPCMessage(r *PutObjectPartInit) *object.PutRequest_Bod ) m.SetSignature( - service.SignatureToGRPCMessage(r.GetSignature()), + refs.SignatureToGRPCMessage(r.GetSignature()), ) m.SetHeader( @@ -579,7 +579,7 @@ func PutObjectPartInitFromGRPCMessage(m *object.PutRequest_Body_Init) *PutObject ) r.SetSignature( - service.SignatureFromGRPCMessage(m.GetSignature()), + refs.SignatureFromGRPCMessage(m.GetSignature()), ) r.SetHeader( diff --git a/v2/object/grpc/service.go b/v2/object/grpc/service.go index 9dda1cb..04cebfa 100644 --- a/v2/object/grpc/service.go +++ b/v2/object/grpc/service.go @@ -48,7 +48,7 @@ func (m *GetResponse_Body_Init) SetObjectId(v *refs.ObjectID) { } // SetSignature sets signature of the object identifier. -func (m *GetResponse_Body_Init) SetSignature(v *service.Signature) { +func (m *GetResponse_Body_Init) SetSignature(v *refs.Signature) { if m != nil { m.Signature = v } @@ -122,7 +122,7 @@ func (m *PutRequest_Body_Init) SetObjectId(v *refs.ObjectID) { } // SetSignature sets signature of the object identifier. -func (m *PutRequest_Body_Init) SetSignature(v *service.Signature) { +func (m *PutRequest_Body_Init) SetSignature(v *refs.Signature) { if m != nil { m.Signature = v } diff --git a/v2/object/grpc/types.go b/v2/object/grpc/types.go index 3600765..24216ab 100644 --- a/v2/object/grpc/types.go +++ b/v2/object/grpc/types.go @@ -34,7 +34,7 @@ func (m *Header_Split) SetPrevious(v *refs.ObjectID) { } // SetParentSignature sets signature of the parent object header. -func (m *Header_Split) SetParentSignature(v *service.Signature) { +func (m *Header_Split) SetParentSignature(v *refs.Signature) { if m != nil { m.ParentSignature = v } @@ -76,7 +76,7 @@ func (m *Header) SetCreationEpoch(v uint64) { } // SetVersion sets version of the object format. -func (m *Header) SetVersion(v *service.Version) { +func (m *Header) SetVersion(v *refs.Version) { if m != nil { m.Version = v } @@ -139,7 +139,7 @@ func (m *Object) SetObjectId(v *refs.ObjectID) { } // SetSignature sets signature of the object identifier. -func (m *Object) SetSignature(v *service.Signature) { +func (m *Object) SetSignature(v *refs.Signature) { if m != nil { m.Signature = v } @@ -160,7 +160,7 @@ func (m *Object) SetPayload(v []byte) { } // SetVersion sets version of the object. -func (m *ShortHeader) SetVersion(v *service.Version) { +func (m *ShortHeader) SetVersion(v *refs.Version) { if m != nil { m.Version = v } diff --git a/v2/object/marshal_test.go b/v2/object/marshal_test.go index fd0eefe..02480bd 100644 --- a/v2/object/marshal_test.go +++ b/v2/object/marshal_test.go @@ -381,16 +381,16 @@ func generateContainerID(id string) *refs.ContainerID { return cid } -func generateSignature(k, v string) *service.Signature { - sig := new(service.Signature) +func generateSignature(k, v string) *refs.Signature { + sig := new(refs.Signature) sig.SetKey([]byte(k)) sig.SetSign([]byte(v)) return sig } -func generateVersion(maj, min uint32) *service.Version { - version := new(service.Version) +func generateVersion(maj, min uint32) *refs.Version { + version := new(refs.Version) version.SetMajor(maj) version.SetMinor(min) diff --git a/v2/object/types.go b/v2/object/types.go index 7999cee..5f421fc 100644 --- a/v2/object/types.go +++ b/v2/object/types.go @@ -10,7 +10,7 @@ type Type uint32 type MatchType uint32 type ShortHeader struct { - version *service.Version + version *refs.Version creatEpoch uint64 @@ -28,7 +28,7 @@ type Attribute struct { type SplitHeader struct { par, prev *refs.ObjectID - parSig *service.Signature + parSig *refs.Signature parHdr *Header @@ -36,7 +36,7 @@ type SplitHeader struct { } type Header struct { - version *service.Version + version *refs.Version cid *refs.ContainerID @@ -60,7 +60,7 @@ type Header struct { type Object struct { objectID *refs.ObjectID - idSig *service.Signature + idSig *refs.Signature header *Header @@ -88,7 +88,7 @@ type GetObjectPart interface { type GetObjectPartInit struct { id *refs.ObjectID - sig *service.Signature + sig *refs.Signature hdr *Header } @@ -116,7 +116,7 @@ type PutObjectPart interface { type PutObjectPartInit struct { id *refs.ObjectID - sig *service.Signature + sig *refs.Signature hdr *Header @@ -318,7 +318,7 @@ const ( MatchStringEqual ) -func (h *ShortHeader) GetVersion() *service.Version { +func (h *ShortHeader) GetVersion() *refs.Version { if h != nil { return h.version } @@ -326,7 +326,7 @@ func (h *ShortHeader) GetVersion() *service.Version { return nil } -func (h *ShortHeader) SetVersion(v *service.Version) { +func (h *ShortHeader) SetVersion(v *refs.Version) { if h != nil { h.version = v } @@ -444,7 +444,7 @@ func (h *SplitHeader) SetPrevious(v *refs.ObjectID) { } } -func (h *SplitHeader) GetParentSignature() *service.Signature { +func (h *SplitHeader) GetParentSignature() *refs.Signature { if h != nil { return h.parSig } @@ -452,7 +452,7 @@ func (h *SplitHeader) GetParentSignature() *service.Signature { return nil } -func (h *SplitHeader) SetParentSignature(v *service.Signature) { +func (h *SplitHeader) SetParentSignature(v *refs.Signature) { if h != nil { h.parSig = v } @@ -486,7 +486,7 @@ func (h *SplitHeader) SetChildren(v []*refs.ObjectID) { } } -func (h *Header) GetVersion() *service.Version { +func (h *Header) GetVersion() *refs.Version { if h != nil { return h.version } @@ -494,7 +494,7 @@ func (h *Header) GetVersion() *service.Version { return nil } -func (h *Header) SetVersion(v *service.Version) { +func (h *Header) SetVersion(v *refs.Version) { if h != nil { h.version = v } @@ -654,7 +654,7 @@ func (o *Object) SetObjectID(v *refs.ObjectID) { } } -func (o *Object) GetSignature() *service.Signature { +func (o *Object) GetSignature() *refs.Signature { if o != nil { return o.idSig } @@ -662,7 +662,7 @@ func (o *Object) GetSignature() *service.Signature { return nil } -func (o *Object) SetSignature(v *service.Signature) { +func (o *Object) SetSignature(v *refs.Signature) { if o != nil { o.idSig = v } @@ -780,7 +780,7 @@ func (r *GetObjectPartInit) SetObjectID(v *refs.ObjectID) { } } -func (r *GetObjectPartInit) GetSignature() *service.Signature { +func (r *GetObjectPartInit) GetSignature() *refs.Signature { if r != nil { return r.sig } @@ -788,7 +788,7 @@ func (r *GetObjectPartInit) GetSignature() *service.Signature { return nil } -func (r *GetObjectPartInit) SetSignature(v *service.Signature) { +func (r *GetObjectPartInit) SetSignature(v *refs.Signature) { if r != nil { r.sig = v } @@ -896,7 +896,7 @@ func (r *PutObjectPartInit) SetObjectID(v *refs.ObjectID) { } } -func (r *PutObjectPartInit) GetSignature() *service.Signature { +func (r *PutObjectPartInit) GetSignature() *refs.Signature { if r != nil { return r.sig } @@ -904,7 +904,7 @@ func (r *PutObjectPartInit) GetSignature() *service.Signature { return nil } -func (r *PutObjectPartInit) SetSignature(v *service.Signature) { +func (r *PutObjectPartInit) SetSignature(v *refs.Signature) { if r != nil { r.sig = v } diff --git a/v2/refs/convert.go b/v2/refs/convert.go index 4a84077..00971be 100644 --- a/v2/refs/convert.go +++ b/v2/refs/convert.go @@ -139,3 +139,55 @@ func ChecksumFromGRPCMessage(m *refs.Checksum) *Checksum { return c } + +func VersionToGRPCMessage(v *Version) *refs.Version { + if v == nil { + return nil + } + + msg := new(refs.Version) + + msg.SetMajor(v.GetMajor()) + msg.SetMinor(v.GetMinor()) + + return msg +} + +func VersionFromGRPCMessage(m *refs.Version) *Version { + if m == nil { + return nil + } + + v := new(Version) + + v.SetMajor(m.GetMajor()) + v.SetMinor(m.GetMinor()) + + return v +} + +func SignatureToGRPCMessage(s *Signature) *refs.Signature { + if s == nil { + return nil + } + + m := new(refs.Signature) + + m.SetKey(s.GetKey()) + m.SetSign(s.GetSign()) + + return m +} + +func SignatureFromGRPCMessage(m *refs.Signature) *Signature { + if m == nil { + return nil + } + + s := new(Signature) + + s.SetKey(m.GetKey()) + s.SetSign(m.GetSign()) + + return s +} diff --git a/v2/refs/grpc/types.go b/v2/refs/grpc/types.go index e55504d..b071dc8 100644 --- a/v2/refs/grpc/types.go +++ b/v2/refs/grpc/types.go @@ -48,3 +48,31 @@ func (m *Checksum) SetSum(v []byte) { m.Sum = v } } + +// SetMajor sets major version number. +func (m *Version) SetMajor(v uint32) { + if m != nil { + m.Major = v + } +} + +// SetMinor sets minor version number. +func (m *Version) SetMinor(v uint32) { + if m != nil { + m.Minor = v + } +} + +// SetKey sets public key in a binary format. +func (m *Signature) SetKey(v []byte) { + if m != nil { + m.Key = v + } +} + +// SetSign sets signature. +func (m *Signature) SetSign(v []byte) { + if m != nil { + m.Sign = v + } +} diff --git a/v2/refs/marshal.go b/v2/refs/marshal.go index 7c1f524..7d52c60 100644 --- a/v2/refs/marshal.go +++ b/v2/refs/marshal.go @@ -16,6 +16,12 @@ const ( checksumTypeField = 1 checksumValueField = 2 + + signatureKeyField = 1 + signatureValueField = 2 + + versionMajorField = 1 + versionMinorField = 2 ) func (o *OwnerID) StableMarshal(buf []byte) ([]byte, error) { @@ -173,3 +179,83 @@ func (c *Checksum) StableSize() (size int) { return size } + +func (s *Signature) StableMarshal(buf []byte) ([]byte, error) { + if s == nil { + return []byte{}, nil + } + + if buf == nil { + buf = make([]byte, s.StableSize()) + } + + var ( + offset, n int + err error + ) + + n, err = proto.BytesMarshal(signatureKeyField, buf[offset:], s.key) + if err != nil { + return nil, err + } + + offset += n + + _, err = proto.BytesMarshal(signatureValueField, buf[offset:], s.sign) + if err != nil { + return nil, err + } + + return buf, nil +} + +func (s *Signature) StableSize() (size int) { + if s == nil { + return 0 + } + + size += proto.BytesSize(signatureKeyField, s.key) + size += proto.BytesSize(signatureValueField, s.sign) + + return size +} + +func (v *Version) StableMarshal(buf []byte) ([]byte, error) { + if v == nil { + return []byte{}, nil + } + + if buf == nil { + buf = make([]byte, v.StableSize()) + } + + var ( + offset, n int + err error + ) + + n, err = proto.UInt32Marshal(versionMajorField, buf[offset:], v.major) + if err != nil { + return nil, err + } + + offset += n + + _, err = proto.UInt32Marshal(versionMinorField, buf[offset:], v.minor) + if err != nil { + return nil, err + } + + return buf, nil +} + +func (v *Version) StableSize() (size int) { + if v == nil { + return 0 + } + + size += proto.UInt32Size(versionMajorField, v.major) + size += proto.UInt32Size(versionMinorField, v.minor) + + return size +} diff --git a/v2/refs/marshal_test.go b/v2/refs/marshal_test.go index 441174e..f4e1350 100644 --- a/v2/refs/marshal_test.go +++ b/v2/refs/marshal_test.go @@ -106,3 +106,51 @@ func TestChecksum_StableMarshal(t *testing.T) { require.Equal(t, checksumFrom, checksumTo) }) } + +func TestSignature_StableMarshal(t *testing.T) { + signatureFrom := generateSignature("Public Key", "Signature") + transport := new(grpc.Signature) + + t.Run("non empty", func(t *testing.T) { + wire, err := signatureFrom.StableMarshal(nil) + require.NoError(t, err) + + err = transport.Unmarshal(wire) + require.NoError(t, err) + + signatureTo := refs.SignatureFromGRPCMessage(transport) + require.Equal(t, signatureFrom, signatureTo) + }) +} + +func TestVersion_StableMarshal(t *testing.T) { + versionFrom := generateVersion(2, 0) + transport := new(grpc.Version) + + t.Run("non empty", func(t *testing.T) { + wire, err := versionFrom.StableMarshal(nil) + require.NoError(t, err) + + err = transport.Unmarshal(wire) + require.NoError(t, err) + + versionTo := refs.VersionFromGRPCMessage(transport) + require.Equal(t, versionFrom, versionTo) + }) +} + +func generateSignature(k, v string) *refs.Signature { + sig := new(refs.Signature) + sig.SetKey([]byte(k)) + sig.SetSign([]byte(v)) + + return sig +} + +func generateVersion(maj, min uint32) *refs.Version { + version := new(refs.Version) + version.SetMajor(maj) + version.SetMinor(min) + + return version +} diff --git a/v2/refs/types.go b/v2/refs/types.go index e368b62..0b90268 100644 --- a/v2/refs/types.go +++ b/v2/refs/types.go @@ -26,6 +26,14 @@ type Checksum struct { type ChecksumType uint32 +type Signature struct { + key, sign []byte +} + +type Version struct { + major, minor uint32 +} + const ( UnknownChecksum ChecksumType = iota TillichZemor @@ -129,3 +137,59 @@ func (c *Checksum) SetSum(v []byte) { c.sum = v } } + +func (s *Signature) GetKey() []byte { + if s != nil { + return s.key + } + + return nil +} + +func (s *Signature) SetKey(v []byte) { + if s != nil { + s.key = v + } +} + +func (s *Signature) GetSign() []byte { + if s != nil { + return s.sign + } + + return nil +} + +func (s *Signature) SetSign(v []byte) { + if s != nil { + s.sign = v + } +} + +func (v *Version) GetMajor() uint32 { + if v != nil { + return v.major + } + + return 0 +} + +func (v *Version) SetMajor(val uint32) { + if v != nil { + v.major = val + } +} + +func (v *Version) GetMinor() uint32 { + if v != nil { + return v.minor + } + + return 0 +} + +func (v *Version) SetMinor(val uint32) { + if v != nil { + v.minor = val + } +} diff --git a/v2/signature/sign.go b/v2/signature/sign.go index 086ce0d..2cb768a 100644 --- a/v2/signature/sign.go +++ b/v2/signature/sign.go @@ -8,6 +8,7 @@ import ( "github.com/nspcc-dev/neofs-api-go/v2/accounting" "github.com/nspcc-dev/neofs-api-go/v2/container" "github.com/nspcc-dev/neofs-api-go/v2/object" + "github.com/nspcc-dev/neofs-api-go/v2/refs" "github.com/nspcc-dev/neofs-api-go/v2/service" "github.com/nspcc-dev/neofs-api-go/v2/session" "github.com/pkg/errors" @@ -42,12 +43,12 @@ type metaHeader interface { type verificationHeader interface { stableMarshaler - GetBodySignature() *service.Signature - SetBodySignature(*service.Signature) - GetMetaSignature() *service.Signature - SetMetaSignature(*service.Signature) - GetOriginSignature() *service.Signature - SetOriginSignature(*service.Signature) + GetBodySignature() *refs.Signature + SetBodySignature(*refs.Signature) + GetMetaSignature() *refs.Signature + SetMetaSignature(*refs.Signature) + GetOriginSignature() *refs.Signature + SetOriginSignature(*refs.Signature) setOrigin(stableMarshaler) getOrigin() verificationHeader @@ -129,14 +130,14 @@ func (s stableMarshalerWrapper) SignedDataSize() int { return 0 } -func keySignatureHandler(s *service.Signature) signature.KeySignatureHandler { +func keySignatureHandler(s *refs.Signature) signature.KeySignatureHandler { return func(key []byte, sig []byte) { s.SetKey(key) s.SetSign(sig) } } -func keySignatureSource(s *service.Signature) signature.KeySignatureSource { +func keySignatureSource(s *refs.Signature) signature.KeySignatureSource { return func() ([]byte, []byte) { return s.GetKey(), s.GetSign() } @@ -204,8 +205,8 @@ func SignServiceMessage(key *ecdsa.PrivateKey, msg interface{}) error { return nil } -func signServiceMessagePart(key *ecdsa.PrivateKey, part stableMarshaler, sigWrite func(*service.Signature)) error { - sig := new(service.Signature) +func signServiceMessagePart(key *ecdsa.PrivateKey, part stableMarshaler, sigWrite func(*refs.Signature)) error { + sig := new(refs.Signature) // sign part if err := signature.SignDataWithHandler( @@ -280,7 +281,7 @@ func verifyMatryoshkaLevel(body stableMarshaler, meta metaHeader, verify verific return verifyMatryoshkaLevel(body, meta.getOrigin(), origin) } -func verifyServiceMessagePart(part stableMarshaler, sigRdr func() *service.Signature) error { +func verifyServiceMessagePart(part stableMarshaler, sigRdr func() *refs.Signature) error { return signature.VerifyDataWithSource( &stableMarshalerWrapper{part}, keySignatureSource(sigRdr()),