forked from TrueCloudLab/frostfs-node
Alex Vanin
20de74a505
Due to source code relocation from GitHub. Signed-off-by: Alex Vanin <a.vanin@yadro.com>
98 lines
2.4 KiB
Go
98 lines
2.4 KiB
Go
package reputation
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/hex"
|
|
"errors"
|
|
"fmt"
|
|
|
|
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("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 !value.VerifySignature() {
|
|
rp.log.Info("ignore reputation value",
|
|
zap.String("reason", "invalid signature"),
|
|
)
|
|
|
|
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, 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("can't send approval tx for reputation value",
|
|
zap.String("peer_id", hex.EncodeToString(id.PublicKey())),
|
|
zap.String("error", err.Error()))
|
|
}
|
|
}
|