forked from TrueCloudLab/frostfs-node
99 lines
2.6 KiB
Go
99 lines
2.6 KiB
Go
package reputation
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/hex"
|
|
"errors"
|
|
"fmt"
|
|
|
|
repClient "github.com/nspcc-dev/neofs-node/pkg/morph/client/reputation"
|
|
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: #1147 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 := 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 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()))
|
|
}
|
|
}
|