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

package reputation

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

	"git.frostfs.info/TrueCloudLab/frostfs-node/internal/logs"
	repClient "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/client/reputation"
	reputationEvent "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/event/reputation"
	apireputation "git.frostfs.info/TrueCloudLab/frostfs-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(logs.ReputationNonAlphabetModeIgnoreReputationPutNotification)
		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(logs.ReputationIgnoreReputationValue,
			zap.String("reason", "invalid epoch number"),
			zap.Uint64("trust_epoch", epoch),
			zap.Uint64("local_epoch", currentEpoch))

		return
	}

	// check signature
	if !value.VerifySignature() {
		rp.log.Info(logs.ReputationIgnoreReputationValue,
			zap.String("reason", "invalid signature"),
		)

		return
	}

	// check if manager is correct
	if err := rp.checkManagers(epoch, value.Manager(), id); err != nil {
		rp.log.Info(logs.ReputationIgnoreReputationValue,
			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, peer)
	if err != nil {
		return fmt.Errorf("could not build managers: %w", err)
	}

	for _, m := range mm {
		// FIXME: #1147 do not use `ToV2` method outside frostfs-api-go library
		if bytes.Equal(mng.PublicKey(), 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 := repClient.PutPrm{}
		args.SetEpoch(e.Epoch())
		args.SetPeerID(id)
		args.SetValue(e.Value())

		err = rp.reputationWrp.Put(args)
	}
	if err != nil {
		// FIXME: #1147 do not use `ToV2` method outside frostfs-api-go library
		rp.log.Warn(logs.ReputationCantSendApprovalTxForReputationValue,
			zap.String("peer_id", hex.EncodeToString(id.PublicKey())),
			zap.String("error", err.Error()))
	}
}