package netmap import ( "crypto/elliptic" "fmt" "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/client" "github.com/nspcc-dev/neo-go/pkg/crypto/keys" "github.com/nspcc-dev/neo-go/pkg/vm/stackitem" ) // UpdateIRPrm groups parameters of UpdateInnerRing // invocation. type UpdateIRPrm struct { keys keys.PublicKeys client.InvokePrmOptional } // SetKeys sets new inner ring keys. func (u *UpdateIRPrm) SetKeys(keys keys.PublicKeys) { u.keys = keys } // UpdateInnerRing updates inner ring keys. func (c *Client) UpdateInnerRing(p UpdateIRPrm) error { args := make([][]byte, len(p.keys)) for i := range args { args[i] = p.keys[i].Bytes() } prm := client.InvokePrm{} prm.SetMethod(updateInnerRingMethod) prm.SetArgs(args) prm.InvokePrmOptional = p.InvokePrmOptional return c.client.Invoke(prm) } // GetInnerRingList return current IR list. func (c *Client) GetInnerRingList() (keys.PublicKeys, error) { invokePrm := client.TestInvokePrm{} invokePrm.SetMethod(innerRingListMethod) prms, err := c.client.TestInvoke(invokePrm) if err != nil { return nil, fmt.Errorf("could not perform test invocation (%s): %w", innerRingListMethod, err) } return irKeysFromStackItem(prms, innerRingListMethod) } func irKeysFromStackItem(stack []stackitem.Item, method string) (keys.PublicKeys, error) { if ln := len(stack); ln != 1 { return nil, fmt.Errorf("unexpected stack item count (%s): %d", method, ln) } irs, err := client.ArrayFromStackItem(stack[0]) if err != nil { return nil, fmt.Errorf("could not get stack item array from stack item (%s): %w", method, err) } irKeys := make(keys.PublicKeys, len(irs)) for i := range irs { irKeys[i], err = irKeyFromStackItem(irs[i]) if err != nil { return nil, err } } return irKeys, nil } const irNodeFixedPrmNumber = 1 func irKeyFromStackItem(prm stackitem.Item) (*keys.PublicKey, error) { prms, err := client.ArrayFromStackItem(prm) if err != nil { return nil, fmt.Errorf("could not get stack item array (IRNode): %w", err) } else if ln := len(prms); ln != irNodeFixedPrmNumber { return nil, fmt.Errorf( "unexpected stack item count (IRNode): expected %d, has %d", irNodeFixedPrmNumber, ln, ) } byteKey, err := client.BytesFromStackItem(prms[0]) if err != nil { return nil, fmt.Errorf("could not parse bytes from stack item (IRNode): %w", err) } return keys.NewPublicKeyFromBytes(byteKey, elliptic.P256()) }