Anton Nikiforov
8ee590794f
Signed-off-by: Anton Nikiforov <an.nikiforov@yadro.com> Signed-off-by: Evgenii Stratonikov <e.stratonikov@yadro.com>
70 lines
1.9 KiB
Go
70 lines
1.9 KiB
Go
package key
|
|
|
|
import (
|
|
"crypto/ecdsa"
|
|
"errors"
|
|
|
|
"github.com/TrueCloudLab/frostfs-node/cmd/frostfs-cli/internal/common"
|
|
"github.com/nspcc-dev/neo-go/cli/flags"
|
|
"github.com/nspcc-dev/neo-go/cli/input"
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/nspcc-dev/neo-go/pkg/wallet"
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
// Key-related errors.
|
|
var (
|
|
ErrFs = errors.New("unable to read file from given path")
|
|
ErrInvalidKey = errors.New("provided key is incorrect, only wallet or binary key supported")
|
|
ErrInvalidAddress = errors.New("--address option must be specified and valid")
|
|
ErrInvalidPassword = errors.New("invalid password for the encrypted key")
|
|
)
|
|
|
|
// FromWallet returns private key of the wallet account.
|
|
func FromWallet(cmd *cobra.Command, w *wallet.Wallet, addrStr string) (*ecdsa.PrivateKey, error) {
|
|
var (
|
|
addr util.Uint160
|
|
err error
|
|
)
|
|
|
|
if addrStr == "" {
|
|
common.PrintVerbose(cmd, "Using default wallet address")
|
|
addr = w.GetChangeAddress()
|
|
} else {
|
|
addr, err = flags.ParseAddress(addrStr)
|
|
if err != nil {
|
|
common.PrintVerbose(cmd, "Can't parse address: %s", addrStr)
|
|
return nil, ErrInvalidAddress
|
|
}
|
|
}
|
|
|
|
acc := w.GetAccount(addr)
|
|
if acc == nil {
|
|
common.PrintVerbose(cmd, "Can't find wallet account for %s", addrStr)
|
|
return nil, ErrInvalidAddress
|
|
}
|
|
|
|
pass, err := getPassword()
|
|
if err != nil {
|
|
common.PrintVerbose(cmd, "Can't read password: %v", err)
|
|
return nil, ErrInvalidPassword
|
|
}
|
|
|
|
if err := acc.Decrypt(pass, keys.NEP2ScryptParams()); err != nil {
|
|
common.PrintVerbose(cmd, "Can't decrypt account: %v", err)
|
|
return nil, ErrInvalidPassword
|
|
}
|
|
|
|
return &acc.PrivateKey().PrivateKey, nil
|
|
}
|
|
|
|
func getPassword() (string, error) {
|
|
// this check allows empty passwords
|
|
if viper.IsSet("password") {
|
|
return viper.GetString("password"), nil
|
|
}
|
|
|
|
return input.ReadPassword("Enter password > ")
|
|
}
|