package container import ( "testing" "github.com/nspcc-dev/neofs-api-go/service" "github.com/nspcc-dev/neofs-crypto/test" "github.com/stretchr/testify/require" ) func TestRequestSign(t *testing.T) { sk := test.DecodeKey(0) type sigType interface { service.SignedDataWithToken service.SignKeyPairAccumulator service.SignKeyPairSource SetToken(*service.Token) } items := []struct { constructor func() sigType payloadCorrupt []func(sigType) }{ { // PutRequest constructor: func() sigType { return new(PutRequest) }, payloadCorrupt: []func(sigType){ func(s sigType) { req := s.(*PutRequest) id := req.GetMessageID() id[0]++ req.SetMessageID(id) }, func(s sigType) { req := s.(*PutRequest) req.SetCapacity(req.GetCapacity() + 1) }, func(s sigType) { req := s.(*PutRequest) owner := req.GetOwnerID() owner[0]++ req.SetOwnerID(owner) }, func(s sigType) { req := s.(*PutRequest) rules := req.GetRules() rules.ReplFactor++ req.SetRules(rules) }, func(s sigType) { req := s.(*PutRequest) req.SetBasicACL(req.GetBasicACL() + 1) }, }, }, { // DeleteRequest constructor: func() sigType { return new(DeleteRequest) }, payloadCorrupt: []func(sigType){ func(s sigType) { req := s.(*DeleteRequest) cid := req.GetCID() cid[0]++ req.SetCID(cid) }, }, }, { // GetRequest constructor: func() sigType { return new(GetRequest) }, payloadCorrupt: []func(sigType){ func(s sigType) { req := s.(*GetRequest) cid := req.GetCID() cid[0]++ req.SetCID(cid) }, }, }, } for _, item := range items { { // token corruptions v := item.constructor() token := new(service.Token) v.SetToken(token) require.NoError(t, service.SignDataWithSessionToken(sk, v)) require.NoError(t, service.VerifyAccumulatedSignaturesWithToken(v)) token.SetSessionKey(append(token.GetSessionKey(), 1)) require.Error(t, service.VerifyAccumulatedSignaturesWithToken(v)) } { // payload corruptions for _, corruption := range item.payloadCorrupt { v := item.constructor() require.NoError(t, service.SignDataWithSessionToken(sk, v)) require.NoError(t, service.VerifyAccumulatedSignaturesWithToken(v)) corruption(v) require.Error(t, service.VerifyAccumulatedSignaturesWithToken(v)) } } } }