[#11] services: Implement universal Sign/Verify service

Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>

pkg/services/accounting/sign.go

@@ -5,39 +5,29 @@ import (
 	"crypto/ecdsa"
 
 	"github.com/nspcc-dev/neofs-api-go/v2/accounting"
-	"github.com/nspcc-dev/neofs-api-go/v2/signature"
+	"github.com/nspcc-dev/neofs-node/pkg/services/util"
-	"github.com/pkg/errors"
 )
 
 type signService struct {
-	key *ecdsa.PrivateKey
+	unarySigService *util.UnarySignService
-
-	svc accounting.Service
 }
 
 func NewSignService(key *ecdsa.PrivateKey, svc accounting.Service) accounting.Service {
 	return &signService{
-		key: key,
+		unarySigService: util.NewUnarySignService(
-		svc: svc,
+			key,
+			func(ctx context.Context, req interface{}) (interface{}, error) {
+				return svc.Balance(ctx, req.(*accounting.BalanceRequest))
+			},
+		),
 	}
 }
 
 func (s *signService) Balance(ctx context.Context, req *accounting.BalanceRequest) (*accounting.BalanceResponse, error) {
-	// verify request signatures
+	resp, err := s.unarySigService.HandleUnaryRequest(ctx, req)
-	if err := signature.VerifyServiceMessage(req); err != nil {
-		return nil, errors.Wrap(err, "could not verify request")
-	}
-
-	// process request
-	resp, err := s.svc.Balance(ctx, req)
 	if err != nil {
 		return nil, err
 	}
 
-	// sign the response
+	return resp.(*accounting.BalanceResponse), nil
-	if err := signature.SignServiceMessage(s.key, resp); err != nil {
-		return nil, errors.Wrap(err, "could not sign response")
-	}
-
-	return resp, nil
 }

pkg/services/util/sign.go

@@ -0,0 +1,44 @@
+package util
+
+import (
+	"context"
+	"crypto/ecdsa"
+
+	"github.com/nspcc-dev/neofs-api-go/v2/signature"
+	"github.com/pkg/errors"
+)
+
+type UnaryHandler func(context.Context, interface{}) (interface{}, error)
+
+type UnarySignService struct {
+	key *ecdsa.PrivateKey
+
+	unaryHandler UnaryHandler
+}
+
+func NewUnarySignService(key *ecdsa.PrivateKey, handler UnaryHandler) *UnarySignService {
+	return &UnarySignService{
+		key:          key,
+		unaryHandler: handler,
+	}
+}
+
+func (s *UnarySignService) HandleUnaryRequest(ctx context.Context, req interface{}) (interface{}, error) {
+	// verify request signatures
+	if err := signature.VerifyServiceMessage(req); err != nil {
+		return nil, errors.Wrap(err, "could not verify request")
+	}
+
+	// process request
+	resp, err := s.unaryHandler(ctx, req)
+	if err != nil {
+		return nil, errors.Wrap(err, "could not handle request")
+	}
+
+	// sign the response
+	if err := signature.SignServiceMessage(s.key, resp); err != nil {
+		return nil, errors.Wrap(err, "could not sign response")
+	}
+
+	return resp, nil
+}