forked from TrueCloudLab/frostfs-node
102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package netmap
|
|
|
|
import (
|
|
"crypto/elliptic"
|
|
"fmt"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-contract/netmap"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/client"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/event"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/state"
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
"github.com/nspcc-dev/neo-go/pkg/network/payload"
|
|
)
|
|
|
|
type UpdatePeer struct {
|
|
PubKey *keys.PublicKey
|
|
|
|
State netmap.NodeState
|
|
|
|
// For notary notifications only.
|
|
// Contains raw transactions of notary request.
|
|
Request *payload.P2PNotaryRequest
|
|
}
|
|
|
|
// MorphEvent implements Neo:Morph Event interface.
|
|
func (UpdatePeer) MorphEvent() {}
|
|
|
|
// Online returns true if node's state is requested to be switched
|
|
// to "online".
|
|
func (s UpdatePeer) Online() bool {
|
|
return s.State == netmap.NodeStateOnline
|
|
}
|
|
|
|
// Maintenance returns true if node's state is requested to be switched
|
|
// to "maintenance".
|
|
func (s UpdatePeer) Maintenance() bool {
|
|
return s.State == netmap.NodeStateMaintenance
|
|
}
|
|
|
|
func (s UpdatePeer) PublicKey() *keys.PublicKey {
|
|
return s.PubKey
|
|
}
|
|
|
|
// NotaryRequest returns raw notary request if notification
|
|
// was received via notary service. Otherwise, returns nil.
|
|
func (s UpdatePeer) NotaryRequest() *payload.P2PNotaryRequest {
|
|
return s.Request
|
|
}
|
|
|
|
func (s *UpdatePeer) decodeState(state int64) error {
|
|
switch s.State = netmap.NodeState(state); s.State {
|
|
default:
|
|
return fmt.Errorf("unsupported node state %d", state)
|
|
case
|
|
netmap.NodeStateOffline,
|
|
netmap.NodeStateOnline,
|
|
netmap.NodeStateMaintenance:
|
|
return nil
|
|
}
|
|
}
|
|
|
|
const expectedItemNumUpdatePeer = 2
|
|
|
|
func ParseUpdatePeer(e *state.ContainedNotificationEvent) (event.Event, error) {
|
|
var (
|
|
ev UpdatePeer
|
|
err error
|
|
)
|
|
|
|
params, err := event.ParseStackArray(e)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("could not parse stack items from notify event: %w", err)
|
|
}
|
|
|
|
if ln := len(params); ln != expectedItemNumUpdatePeer {
|
|
return nil, event.WrongNumberOfParameters(expectedItemNumUpdatePeer, ln)
|
|
}
|
|
|
|
// parse public key
|
|
key, err := client.BytesFromStackItem(params[1])
|
|
if err != nil {
|
|
return nil, fmt.Errorf("could not get public key: %w", err)
|
|
}
|
|
|
|
ev.PubKey, err = keys.NewPublicKeyFromBytes(key, elliptic.P256())
|
|
if err != nil {
|
|
return nil, fmt.Errorf("could not parse public key: %w", err)
|
|
}
|
|
|
|
// parse node status
|
|
st, err := client.IntFromStackItem(params[0])
|
|
if err != nil {
|
|
return nil, fmt.Errorf("could not get node status: %w", err)
|
|
}
|
|
|
|
err = ev.decodeState(st)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return ev, nil
|
|
}
|