forked from TrueCloudLab/frostfs-node
[#990] nodeValidators: Add subnet entrance validator
Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
This commit is contained in:
parent
b6db699b6e
commit
a079a8f727
2 changed files with 82 additions and 0 deletions
|
@ -0,0 +1,41 @@
|
||||||
|
package subnet
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
morphsubnet "github.com/nspcc-dev/neofs-node/pkg/morph/client/subnet"
|
||||||
|
"github.com/nspcc-dev/neofs-sdk-go/netmap"
|
||||||
|
subnetid "github.com/nspcc-dev/neofs-sdk-go/subnet/id"
|
||||||
|
)
|
||||||
|
|
||||||
|
// VerifyAndUpdate calls subnet contract's `NodeAllowed` method.
|
||||||
|
// Removes subnets that have not been approved by the contract.
|
||||||
|
func (v *Validator) VerifyAndUpdate(n *netmap.NodeInfo) error {
|
||||||
|
prm := morphsubnet.NodeAllowedPrm{}
|
||||||
|
|
||||||
|
err := n.IterateSubnets(func(id subnetid.ID) error {
|
||||||
|
rawSubnetID, err := id.Marshal()
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("could not marshal subnetwork ID: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
prm.SetID(rawSubnetID)
|
||||||
|
prm.SetNode(n.PublicKey())
|
||||||
|
|
||||||
|
res, err := v.subnetClient.NodeAllowed(prm)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("could not call `NodeAllowed` contract method: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !res.Allowed() {
|
||||||
|
return netmap.ErrRemoveSubnet
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("could not verify subnet entrance of the node: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
|
@ -0,0 +1,41 @@
|
||||||
|
package subnet
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
|
||||||
|
morphsubnet "github.com/nspcc-dev/neofs-node/pkg/morph/client/subnet"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Validator is an utility that verifies node subnet
|
||||||
|
// allowance.
|
||||||
|
//
|
||||||
|
// For correct operation, Validator must be created
|
||||||
|
// using the constructor (New). After successful creation,
|
||||||
|
// the Validator is immediately ready to work through API.
|
||||||
|
type Validator struct {
|
||||||
|
subnetClient *morphsubnet.Client
|
||||||
|
}
|
||||||
|
|
||||||
|
// Prm groups the required parameters of the Validator's constructor.
|
||||||
|
//
|
||||||
|
// All values must comply with the requirements imposed on them.
|
||||||
|
// Passing incorrect parameter values will result in constructor
|
||||||
|
// failure (error or panic depending on the implementation).
|
||||||
|
type Prm struct {
|
||||||
|
SubnetClient *morphsubnet.Client
|
||||||
|
}
|
||||||
|
|
||||||
|
// New creates a new instance of the Validator.
|
||||||
|
//
|
||||||
|
// The created Validator does not require additional
|
||||||
|
// initialization and is completely ready for work.
|
||||||
|
func New(prm Prm) (*Validator, error) {
|
||||||
|
switch {
|
||||||
|
case prm.SubnetClient == nil:
|
||||||
|
return nil, errors.New("ir/nodeValidator: subnet client is not set")
|
||||||
|
}
|
||||||
|
|
||||||
|
return &Validator{
|
||||||
|
subnetClient: prm.SubnetClient,
|
||||||
|
}, nil
|
||||||
|
}
|
Loading…
Reference in a new issue