forked from TrueCloudLab/frostfs-node
[#525] morph/container: Accept container session token in Put
`Put` method of latest `Container` contract accepts binary session token as an argument. Provide `PutArgs.SetSessionToken` method. Accept session token as a `[]byte` in `Wrapper.Put` method and attach it to `PutArgs`. Marshal session token from container in `wrapper.Put` function and pass it to the method. Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
This commit is contained in:
parent
1deb3f3d01
commit
db67a117f0
3 changed files with 17 additions and 4 deletions
|
@ -66,7 +66,7 @@ func (cp *Processor) checkPutContainer(e *containerEvent.Put) error {
|
|||
|
||||
func (cp *Processor) approvePutContainer(e *containerEvent.Put) {
|
||||
// FIXME: here we should bind key to owner if needed
|
||||
err := cp.cnrClient.Put(e.Container(), e.PublicKey(), e.Signature())
|
||||
err := cp.cnrClient.Put(e.Container(), e.PublicKey(), e.Signature(), nil)
|
||||
if err != nil {
|
||||
cp.log.Error("could not approve put container",
|
||||
zap.String("error", err.Error()),
|
||||
|
|
|
@ -36,6 +36,13 @@ func (p *PutArgs) SetSignature(v []byte) {
|
|||
p.sig = v
|
||||
}
|
||||
|
||||
// SetSessionToken sets token of the session
|
||||
// within which the container was created
|
||||
// in a binary format.
|
||||
func (p *PutArgs) SetSessionToken(v []byte) {
|
||||
p.token = v
|
||||
}
|
||||
|
||||
// Put invokes the call of put container method
|
||||
// of NeoFS Container contract.
|
||||
func (c *Client) Put(args PutArgs) error {
|
||||
|
|
|
@ -33,9 +33,14 @@ func Put(w *Wrapper, cnr *container.Container) (*cid.ID, error) {
|
|||
return nil, fmt.Errorf("can't marshal container: %w", err)
|
||||
}
|
||||
|
||||
binToken, err := cnr.SessionToken().Marshal()
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("could not marshal session token: %w", err)
|
||||
}
|
||||
|
||||
sig := cnr.Signature()
|
||||
|
||||
err = w.Put(data, sig.Key(), sig.Sign())
|
||||
err = w.Put(data, sig.Key(), sig.Sign(), binToken)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
@ -46,14 +51,14 @@ func Put(w *Wrapper, cnr *container.Container) (*cid.ID, error) {
|
|||
return id, nil
|
||||
}
|
||||
|
||||
// Put saves binary container with its key and signature
|
||||
// Put saves binary container with its session token, key and signature
|
||||
// in NeoFS system through Container contract call.
|
||||
//
|
||||
// Returns calculated container identifier and any error
|
||||
// encountered that caused the saving to interrupt.
|
||||
//
|
||||
// If TryNotary is provided, call notary contract.
|
||||
func (w *Wrapper) Put(cnr, key, sig []byte) error {
|
||||
func (w *Wrapper) Put(cnr, key, sig, token []byte) error {
|
||||
if len(sig) == 0 || len(key) == 0 {
|
||||
return errNilArgument
|
||||
}
|
||||
|
@ -63,6 +68,7 @@ func (w *Wrapper) Put(cnr, key, sig []byte) error {
|
|||
args.SetContainer(cnr)
|
||||
args.SetSignature(sig)
|
||||
args.SetPublicKey(key)
|
||||
args.SetSessionToken(token)
|
||||
|
||||
err := w.client.Put(args)
|
||||
if err != nil {
|
||||
|
|
Loading…
Reference in a new issue