frostfs-node/pkg/innerring/processors/reputation/process_put.go

package reputation

import (
	"bytes"
	"encoding/hex"
	"errors"
	"fmt"

	"github.com/nspcc-dev/neofs-node/pkg/morph/client/reputation/wrapper"
	reputationEvent "github.com/nspcc-dev/neofs-node/pkg/morph/event/reputation"
	"github.com/nspcc-dev/neofs-node/pkg/services/reputation"
	apireputation "github.com/nspcc-dev/neofs-sdk-go/reputation"
	"go.uber.org/zap"
)

var errWrongManager = errors.New("got manager that is incorrect for peer")

func (rp *Processor) processPut(e *reputationEvent.Put) {
	if !rp.alphabetState.IsAlphabet() {
		rp.log.Info("non alphabet mode, ignore reputation put notification")
		return
	}

	epoch := e.Epoch()
	id := e.PeerID()
	value := e.Value()

	// check if epoch is valid
	currentEpoch := rp.epochState.EpochCounter()
	if epoch >= currentEpoch {
		rp.log.Info("ignore reputation value",
			zap.String("reason", "invalid epoch number"),
			zap.Uint64("trust_epoch", epoch),
			zap.Uint64("local_epoch", currentEpoch))

		return
	}

	// check signature
	if err := value.VerifySignature(); err != nil {
		rp.log.Info("ignore reputation value",
			zap.String("reason", "invalid signature"),
			zap.String("error", err.Error()))

		return
	}

	// check if manager is correct
	if err := rp.checkManagers(epoch, *value.Manager(), id); err != nil {
		rp.log.Info("ignore reputation value",
			zap.String("reason", "wrong manager"),
			zap.String("error", err.Error()))

		return
	}

	rp.approvePutReputation(e)
}

func (rp *Processor) checkManagers(e uint64, mng apireputation.PeerID, peer apireputation.PeerID) error {
	mm, err := rp.mngBuilder.BuildManagers(e, reputation.PeerIDFromBytes(peer.ToV2().GetPublicKey()))
	if err != nil {
		return fmt.Errorf("could not build managers: %w", err)
	}

	for _, m := range mm {
		// FIXME: do not use `ToV2` method outside neofs-api-go library
		if bytes.Equal(mng.ToV2().GetPublicKey(), m.PublicKey()) {
			return nil
		}
	}

	return errWrongManager
}

func (rp *Processor) approvePutReputation(e *reputationEvent.Put) {
	var (
		id  = e.PeerID()
		err error
	)

	if nr := e.NotaryRequest(); nr != nil {
		// put event was received via Notary service
		err = rp.reputationWrp.Morph().NotarySignAndInvokeTX(nr.MainTransaction)
	} else {
		args := wrapper.PutArgs{}
		args.SetEpoch(e.Epoch())
		args.SetPeerID(id)
		args.SetValue(e.Value())

		err = rp.reputationWrp.Put(args)
	}
	if err != nil {
		// FIXME: do not use `ToV2` method outside neofs-api-go library
		rp.log.Warn("can't send approval tx for reputation value",
			zap.String("peer_id", hex.EncodeToString(id.ToV2().GetPublicKey())),
			zap.String("error", err.Error()))
	}
}