forked from TrueCloudLab/frostfs-node
[#840] adm: Update FrostFS ID deploy arguments
Signed-off-by: Evgenii Stratonikov <e.stratonikov@yadro.com>
This commit is contained in:
parent
b2c63e57ba
commit
f1db468d48
3 changed files with 47 additions and 48 deletions
|
@ -5,38 +5,29 @@ import (
|
||||||
|
|
||||||
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
||||||
"github.com/nspcc-dev/neo-go/pkg/encoding/address"
|
"github.com/nspcc-dev/neo-go/pkg/encoding/address"
|
||||||
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
|
|
||||||
"github.com/nspcc-dev/neo-go/pkg/util"
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
||||||
"github.com/spf13/viper"
|
"github.com/spf13/viper"
|
||||||
)
|
)
|
||||||
|
|
||||||
// neo-go doesn't support []util.Uint160 type:
|
func getFrostfsIDAdmin(v *viper.Viper) (util.Uint160, bool, error) {
|
||||||
// https://github.com/nspcc-dev/neo-go/blob/v0.103.0/pkg/smartcontract/parameter.go#L262
|
admin := v.GetString(frostfsIDAdminConfigKey)
|
||||||
// Thus, return []smartcontract.Parameter.
|
if admin == "" {
|
||||||
func getFrostfsIDAuthorizedKeys(v *viper.Viper, defaultOwner util.Uint160) ([]smartcontract.Parameter, error) {
|
return util.Uint160{}, false, nil
|
||||||
var res []smartcontract.Parameter
|
|
||||||
res = append(res, smartcontract.Parameter{Type: smartcontract.Hash160Type, Value: defaultOwner})
|
|
||||||
|
|
||||||
ks := v.GetStringSlice(frostfsIDAuthorizedKeysConfigKey)
|
|
||||||
for i := range ks {
|
|
||||||
h, err := address.StringToUint160(ks[i])
|
|
||||||
if err == nil {
|
|
||||||
res = append(res, smartcontract.Parameter{Type: smartcontract.Hash160Type, Value: h})
|
|
||||||
continue
|
|
||||||
}
|
}
|
||||||
|
|
||||||
h, err = util.Uint160DecodeStringLE(ks[i])
|
h, err := address.StringToUint160(admin)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
res = append(res, smartcontract.Parameter{Type: smartcontract.Hash160Type, Value: h})
|
return h, true, nil
|
||||||
continue
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pk, err := keys.NewPublicKeyFromString(ks[i])
|
h, err = util.Uint160DecodeStringLE(admin)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
res = append(res, smartcontract.Parameter{Type: smartcontract.Hash160Type, Value: pk.GetScriptHash()})
|
return h, true, nil
|
||||||
continue
|
|
||||||
}
|
}
|
||||||
return nil, fmt.Errorf("frostfsid: #%d item in authorized_keys is invalid: '%s'", i, ks[i])
|
|
||||||
|
pk, err := keys.NewPublicKeyFromString(admin)
|
||||||
|
if err == nil {
|
||||||
|
return pk.GetScriptHash(), true, nil
|
||||||
}
|
}
|
||||||
return res, nil
|
return util.Uint160{}, true, fmt.Errorf("frostfsid: admin is invalid: '%s'", admin)
|
||||||
}
|
}
|
||||||
|
|
|
@ -6,8 +6,6 @@ import (
|
||||||
|
|
||||||
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
||||||
"github.com/nspcc-dev/neo-go/pkg/encoding/address"
|
"github.com/nspcc-dev/neo-go/pkg/encoding/address"
|
||||||
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
|
|
||||||
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
||||||
"github.com/spf13/viper"
|
"github.com/spf13/viper"
|
||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
)
|
)
|
||||||
|
@ -20,30 +18,36 @@ func TestFrostfsIDConfig(t *testing.T) {
|
||||||
pks[i] = pk
|
pks[i] = pk
|
||||||
}
|
}
|
||||||
|
|
||||||
v := viper.New()
|
fmts := []string{
|
||||||
v.Set("frostfsid.authorized_keys", []string{
|
|
||||||
pks[0].GetScriptHash().StringLE(),
|
pks[0].GetScriptHash().StringLE(),
|
||||||
address.Uint160ToString(pks[1].GetScriptHash()),
|
address.Uint160ToString(pks[1].GetScriptHash()),
|
||||||
hex.EncodeToString(pks[2].PublicKey().UncompressedBytes()),
|
hex.EncodeToString(pks[2].PublicKey().UncompressedBytes()),
|
||||||
hex.EncodeToString(pks[3].PublicKey().Bytes()),
|
hex.EncodeToString(pks[3].PublicKey().Bytes()),
|
||||||
})
|
}
|
||||||
|
|
||||||
comm := util.Uint160{1, 2, 3}
|
for i := range fmts {
|
||||||
actual, err := getFrostfsIDAuthorizedKeys(v, comm)
|
v := viper.New()
|
||||||
|
v.Set("frostfsid.admin", fmts[i])
|
||||||
|
|
||||||
|
actual, found, err := getFrostfsIDAdmin(v)
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
require.Equal(t, len(pks)+1, len(actual))
|
require.True(t, found)
|
||||||
require.Equal(t, smartcontract.Hash160Type, actual[0].Type)
|
require.Equal(t, pks[i].GetScriptHash(), actual)
|
||||||
require.Equal(t, comm, actual[0].Value)
|
|
||||||
for i := range pks {
|
|
||||||
require.Equal(t, smartcontract.Hash160Type, actual[i+1].Type)
|
|
||||||
require.Equal(t, pks[i].GetScriptHash(), actual[i+1].Value)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
t.Run("bad key", func(t *testing.T) {
|
t.Run("bad key", func(t *testing.T) {
|
||||||
v := viper.New()
|
v := viper.New()
|
||||||
v.Set("frostfsid.authorized_keys", []string{"abc"})
|
v.Set("frostfsid.admin", "abc")
|
||||||
|
|
||||||
_, err := getFrostfsIDAuthorizedKeys(v, comm)
|
_, found, err := getFrostfsIDAdmin(v)
|
||||||
require.Error(t, err)
|
require.Error(t, err)
|
||||||
|
require.True(t, found)
|
||||||
|
})
|
||||||
|
t.Run("missing key", func(t *testing.T) {
|
||||||
|
v := viper.New()
|
||||||
|
|
||||||
|
_, found, err := getFrostfsIDAdmin(v)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.False(t, found)
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
|
@ -47,7 +47,7 @@ const (
|
||||||
proxyContract = "proxy"
|
proxyContract = "proxy"
|
||||||
)
|
)
|
||||||
|
|
||||||
const frostfsIDAuthorizedKeysConfigKey = "frostfsid.authorized_keys"
|
const frostfsIDAdminConfigKey = "frostfsid.admin"
|
||||||
|
|
||||||
var (
|
var (
|
||||||
contractList = []string{
|
contractList = []string{
|
||||||
|
@ -538,12 +538,16 @@ func (c *initializeContext) getContractDeployData(ctrName string, keysParam []an
|
||||||
nnsCs.Hash,
|
nnsCs.Hash,
|
||||||
"container")
|
"container")
|
||||||
case frostfsIDContract:
|
case frostfsIDContract:
|
||||||
hs, err := getFrostfsIDAuthorizedKeys(viper.GetViper(), c.CommitteeAcc.PublicKey().GetScriptHash())
|
h, found, err := getFrostfsIDAdmin(viper.GetViper())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
items = append(items, hs)
|
if found {
|
||||||
|
items = append(items, h)
|
||||||
|
} else {
|
||||||
|
items = append(items, nil)
|
||||||
|
}
|
||||||
case netmapContract:
|
case netmapContract:
|
||||||
md := getDefaultNetmapContractConfigMap()
|
md := getDefaultNetmapContractConfigMap()
|
||||||
if method == updateMethodName {
|
if method == updateMethodName {
|
||||||
|
|
Loading…
Reference in a new issue