From 9265e31e65b16b520a41dbfd62f76d3898dd03b5 Mon Sep 17 00:00:00 2001 From: Leonard Lyubich Date: Fri, 11 Dec 2020 13:02:32 +0300 Subject: [PATCH] [#243] services/object: Fix lost tokens when generating requests Signed-off-by: Leonard Lyubich --- pkg/services/object/delete/exec.go | 5 +++++ pkg/services/object/delete/prm.go | 7 +++++++ pkg/services/object/delete/util.go | 2 ++ pkg/services/object/delete/v2/util.go | 18 ++++++++++++++++++ pkg/services/object/get/v2/util.go | 18 +++++++----------- pkg/services/object/search/v2/util.go | 19 ++++++++----------- 6 files changed, 47 insertions(+), 22 deletions(-) diff --git a/pkg/services/object/delete/exec.go b/pkg/services/object/delete/exec.go index 7e3706401..1afc8bbe4 100644 --- a/pkg/services/object/delete/exec.go +++ b/pkg/services/object/delete/exec.go @@ -4,6 +4,7 @@ import ( "context" "crypto/ecdsa" + "github.com/nspcc-dev/neofs-api-go/pkg/client" "github.com/nspcc-dev/neofs-api-go/pkg/container" objectSDK "github.com/nspcc-dev/neofs-api-go/pkg/object" "github.com/nspcc-dev/neofs-node/pkg/core/object" @@ -74,6 +75,10 @@ func (exec *execCtx) commonParameters() *util.CommonPrm { return exec.prm.common } +func (exec execCtx) callOptions() []client.CallOption { + return exec.prm.callOpts +} + func (exec *execCtx) newAddress(id *objectSDK.ID) *objectSDK.Address { a := objectSDK.NewAddress() a.SetObjectID(id) diff --git a/pkg/services/object/delete/prm.go b/pkg/services/object/delete/prm.go index bae89262f..779cd2e77 100644 --- a/pkg/services/object/delete/prm.go +++ b/pkg/services/object/delete/prm.go @@ -13,6 +13,8 @@ type Prm struct { common *util.CommonPrm + callOpts []client.CallOption + client.DeleteObjectParams } @@ -25,3 +27,8 @@ func (p *Prm) SetCommonParameters(common *util.CommonPrm) { func (p *Prm) SetPrivateKey(key *ecdsa.PrivateKey) { p.key = key } + +// SetRemoteCallOptions sets call options of remote client calls. +func (p *Prm) SetRemoteCallOptions(opts ...client.CallOption) { + p.callOpts = opts +} diff --git a/pkg/services/object/delete/util.go b/pkg/services/object/delete/util.go index ce0cf49eb..7a5da82d1 100644 --- a/pkg/services/object/delete/util.go +++ b/pkg/services/object/delete/util.go @@ -30,6 +30,7 @@ func (w *headSvcWrapper) headAddress(exec *execCtx, addr *objectSDK.Address) (*o p.SetHeaderWriter(wr) p.WithRawFlag(true) p.WithAddress(addr) + p.SetRemoteCallOptions(exec.callOptions()...) err := (*getsvc.Service)(w).Head(exec.context(), p) if err != nil { @@ -88,6 +89,7 @@ func (w *searchSvcWrapper) splitMembers(exec *execCtx) ([]*objectSDK.ID, error) p.SetPrivateKey(exec.key()) p.WithContainerID(exec.containerID()) p.WithSearchFilters(fs) + p.SetRemoteCallOptions(exec.callOptions()...) err := (*searchsvc.Service)(w).Search(exec.context(), p) if err != nil { diff --git a/pkg/services/object/delete/v2/util.go b/pkg/services/object/delete/v2/util.go index 019f368b4..c09cae632 100644 --- a/pkg/services/object/delete/v2/util.go +++ b/pkg/services/object/delete/v2/util.go @@ -1,6 +1,7 @@ package deletesvc import ( + "github.com/nspcc-dev/neofs-api-go/pkg/client" "github.com/nspcc-dev/neofs-api-go/pkg/object" "github.com/nspcc-dev/neofs-api-go/pkg/token" objectV2 "github.com/nspcc-dev/neofs-api-go/v2/object" @@ -24,6 +25,7 @@ func (s *Service) toPrm(req *objectV2.DeleteRequest, respBody *objectV2.DeleteRe p := new(deletesvc.Prm) p.SetPrivateKey(key) p.SetCommonParameters(commonParameters(meta)) + p.SetRemoteCallOptions(remoteCallOptionsFromMeta(meta)...) body := req.GetBody() p.WithAddress(object.NewAddressFromV2(body.GetAddress())) @@ -51,3 +53,19 @@ func commonParameters(meta *session.RequestMetaHeader) *util.CommonPrm { return prm } + +// can be shared accross all services +func remoteCallOptionsFromMeta(meta *session.RequestMetaHeader) []client.CallOption { + opts := make([]client.CallOption, 0, 3) + opts = append(opts, client.WithTTL(meta.GetTTL()-1)) + + if tok := meta.GetBearerToken(); tok != nil { + opts = append(opts, client.WithBearer(token.NewBearerTokenFromV2(tok))) + } + + if tok := meta.GetSessionToken(); tok != nil { + opts = append(opts, client.WithSession(token.NewSessionTokenFromV2(tok))) + } + + return opts +} diff --git a/pkg/services/object/get/v2/util.go b/pkg/services/object/get/v2/util.go index 8569efdc5..710c0ee14 100644 --- a/pkg/services/object/get/v2/util.go +++ b/pkg/services/object/get/v2/util.go @@ -4,7 +4,6 @@ import ( "crypto/sha256" "hash" - "github.com/nspcc-dev/neofs-api-go/pkg" "github.com/nspcc-dev/neofs-api-go/pkg/client" objectSDK "github.com/nspcc-dev/neofs-api-go/pkg/object" "github.com/nspcc-dev/neofs-api-go/pkg/token" @@ -145,18 +144,15 @@ func (s *Service) toHeadPrm(req *objectV2.HeadRequest, resp *objectV2.HeadRespon // can be shared accross all services func remoteCallOptionsFromMeta(meta *session.RequestMetaHeader) []client.CallOption { - xHdrs := meta.GetXHeaders() + opts := make([]client.CallOption, 0, 3) + opts = append(opts, client.WithTTL(meta.GetTTL()-1)) - opts := make([]client.CallOption, 0, 3+len(xHdrs)) + if tok := meta.GetBearerToken(); tok != nil { + opts = append(opts, client.WithBearer(token.NewBearerTokenFromV2(tok))) + } - opts = append(opts, - client.WithBearer(token.NewBearerTokenFromV2(meta.GetBearerToken())), - client.WithSession(token.NewSessionTokenFromV2(meta.GetSessionToken())), - client.WithTTL(meta.GetTTL()-1), - ) - - for i := range xHdrs { - opts = append(opts, client.WithXHeader(pkg.NewXHeaderFromV2(xHdrs[i]))) + if tok := meta.GetSessionToken(); tok != nil { + opts = append(opts, client.WithSession(token.NewSessionTokenFromV2(tok))) } return opts diff --git a/pkg/services/object/search/v2/util.go b/pkg/services/object/search/v2/util.go index ad3652e40..c6ef8409b 100644 --- a/pkg/services/object/search/v2/util.go +++ b/pkg/services/object/search/v2/util.go @@ -1,7 +1,6 @@ package searchsvc import ( - "github.com/nspcc-dev/neofs-api-go/pkg" "github.com/nspcc-dev/neofs-api-go/pkg/client" "github.com/nspcc-dev/neofs-api-go/pkg/container" objectSDK "github.com/nspcc-dev/neofs-api-go/pkg/object" @@ -38,24 +37,22 @@ func (s *Service) toPrm(req *objectV2.SearchRequest, stream objectSvc.SearchStre // can be shared accross all services func remoteCallOptionsFromMeta(meta *session.RequestMetaHeader) []client.CallOption { - xHdrs := meta.GetXHeaders() + opts := make([]client.CallOption, 0, 3) + opts = append(opts, client.WithTTL(meta.GetTTL()-1)) - opts := make([]client.CallOption, 0, 3+len(xHdrs)) + if tok := meta.GetBearerToken(); tok != nil { + opts = append(opts, client.WithBearer(token.NewBearerTokenFromV2(tok))) + } - opts = append(opts, - client.WithBearer(token.NewBearerTokenFromV2(meta.GetBearerToken())), - client.WithSession(token.NewSessionTokenFromV2(meta.GetSessionToken())), - client.WithTTL(meta.GetTTL()-1), - ) - - for i := range xHdrs { - opts = append(opts, client.WithXHeader(pkg.NewXHeaderFromV2(xHdrs[i]))) + if tok := meta.GetSessionToken(); tok != nil { + opts = append(opts, client.WithSession(token.NewSessionTokenFromV2(tok))) } return opts } func commonParameters(meta *session.RequestMetaHeader) *util.CommonPrm { + return new(util.CommonPrm). WithLocalOnly(meta.GetTTL() <= 1) }