forked from TrueCloudLab/frostfs-crypto
Add method to load PrivateKey
- from hex string - from wif format - from file path
This commit is contained in:
parent
5a95565a67
commit
d059767e79
2 changed files with 107 additions and 0 deletions
21
load.go
Normal file
21
load.go
Normal file
|
@ -0,0 +1,21 @@
|
||||||
|
package crypto
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/ecdsa"
|
||||||
|
"encoding/hex"
|
||||||
|
"io/ioutil"
|
||||||
|
|
||||||
|
"github.com/pkg/errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
func LoadPrivateKey(val string) (*ecdsa.PrivateKey, error) {
|
||||||
|
if data, err := ioutil.ReadFile(val); err == nil {
|
||||||
|
return UnmarshalPrivateKey(data)
|
||||||
|
} else if data, err = hex.DecodeString(val); err == nil {
|
||||||
|
return UnmarshalPrivateKey(data)
|
||||||
|
} else if key, err := WIFDecode(val); err == nil {
|
||||||
|
return key, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, errors.Errorf("unknown key format (%q), expect: hex-string, wif or file-path", val)
|
||||||
|
}
|
86
load_test.go
Normal file
86
load_test.go
Normal file
|
@ -0,0 +1,86 @@
|
||||||
|
package crypto
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/x509"
|
||||||
|
"encoding/hex"
|
||||||
|
"io/ioutil"
|
||||||
|
"os"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/nspcc-dev/neofs-crypto/test"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
func Test_LoadPrivateKey_FromWIF(t *testing.T) {
|
||||||
|
for i := 0; i < 10; i++ {
|
||||||
|
expected := test.DecodeKey(i)
|
||||||
|
|
||||||
|
wif, err := WIFEncode(expected)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
actual, err := LoadPrivateKey(wif)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
require.Equal(t, expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func Test_LoadPrivateKey_FromHexString(t *testing.T) {
|
||||||
|
for i := 0; i < 10; i++ {
|
||||||
|
expected := test.DecodeKey(i)
|
||||||
|
|
||||||
|
hs := hex.EncodeToString(expected.D.Bytes())
|
||||||
|
|
||||||
|
actual, err := LoadPrivateKey(hs)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
require.Equal(t, expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func Test_LoadPrivateKey_FromFile(t *testing.T) {
|
||||||
|
for i := 0; i < 10; i++ {
|
||||||
|
expected := test.DecodeKey(i)
|
||||||
|
|
||||||
|
file, err := ioutil.TempFile("", "_marshaled.key")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
require.NoError(t, file.Close())
|
||||||
|
require.NoError(t, os.Remove(file.Name()))
|
||||||
|
}()
|
||||||
|
|
||||||
|
data, err := x509.MarshalECPrivateKey(expected)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
_, err = file.Write(data)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
actual, err := LoadPrivateKey(file.Name())
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
require.Equal(t, expected, actual)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func Test_LoadPrivateKey_FromCompressedFormatFile(t *testing.T) {
|
||||||
|
for i := 0; i < 10; i++ {
|
||||||
|
expected := test.DecodeKey(i)
|
||||||
|
|
||||||
|
file, err := ioutil.TempFile("", "_compressed.key")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
defer func() {
|
||||||
|
require.NoError(t, file.Close())
|
||||||
|
require.NoError(t, os.Remove(file.Name()))
|
||||||
|
}()
|
||||||
|
|
||||||
|
_, err = file.Write(expected.D.Bytes())
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
actual, err := LoadPrivateKey(file.Name())
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
require.Equal(t, expected, actual)
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue