mirror of
https://github.com/anyproto/any-sync.git
synced 2025-06-11 10:18:08 +09:00
87 lines
2.9 KiB
Go
87 lines
2.9 KiB
Go
package crypto
|
|
|
|
import (
|
|
"crypto/ecdsa"
|
|
"crypto/rand"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/anyproto/go-slip10"
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/common/hexutil"
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMnemonic(t *testing.T) {
|
|
phrase, err := NewMnemonicGenerator().WithWordCount(12)
|
|
require.NoError(t, err)
|
|
parts := strings.Split(string(phrase), " ")
|
|
require.Equal(t, 12, len(parts))
|
|
res, err := phrase.DeriveKeys(0)
|
|
require.NoError(t, err)
|
|
bytes := make([]byte, 64)
|
|
_, err = rand.Read(bytes)
|
|
require.NoError(t, err)
|
|
|
|
// testing signing with keys
|
|
for _, k := range []PrivKey{res.MasterKey, res.Identity, res.OldAccountKey} {
|
|
sign, err := k.Sign(bytes)
|
|
require.NoError(t, err)
|
|
res, err := k.GetPublic().Verify(bytes, sign)
|
|
require.NoError(t, err)
|
|
require.True(t, res)
|
|
}
|
|
|
|
// testing derivation
|
|
masterKey, err := genKey(res.MasterNode)
|
|
require.NoError(t, err)
|
|
require.True(t, res.MasterKey.Equals(masterKey))
|
|
identityNode, err := res.MasterNode.Derive(slip10.FirstHardenedIndex)
|
|
require.NoError(t, err)
|
|
identityKey, err := genKey(identityNode)
|
|
require.NoError(t, err)
|
|
require.True(t, res.Identity.Equals(identityKey))
|
|
oldAccountRes, err := phrase.deriveForPath(true, 0, anytypeAccountOldPrefix)
|
|
require.NoError(t, err)
|
|
require.True(t, res.OldAccountKey.Equals(oldAccountRes.MasterKey))
|
|
|
|
// testing Ethereum derivation:
|
|
var phrase2 Mnemonic = "tag volcano eight thank tide danger coast health above argue embrace heavy"
|
|
res, err = phrase2.DeriveKeys(0)
|
|
require.NoError(t, err)
|
|
|
|
// get address by public key
|
|
publicKey := res.EthereumIdentity.Public()
|
|
publicKeyECDSA, ok := publicKey.(*ecdsa.PublicKey)
|
|
require.Equal(t, true, ok)
|
|
ethAddress := crypto.PubkeyToAddress(*publicKeyECDSA)
|
|
|
|
require.Equal(t, common.HexToAddress("0xC49926C4124cEe1cbA0Ea94Ea31a6c12318df947"), ethAddress)
|
|
}
|
|
|
|
func TestMnemonic_ethereumKeyFromMnemonic(t *testing.T) {
|
|
var badPphrase Mnemonic = "tag volcano"
|
|
_, _, err := badPphrase.ethereumKeyFromMnemonic(0, defaultEthereumDerivation)
|
|
require.Error(t, err)
|
|
|
|
// good
|
|
var phrase Mnemonic = "tag volcano eight thank tide danger coast health above argue embrace heavy"
|
|
|
|
addr, pk, err := phrase.ethereumKeyFromMnemonic(0, defaultEthereumDerivation)
|
|
require.NoError(t, err)
|
|
require.Equal(t, common.HexToAddress("0xC49926C4124cEe1cbA0Ea94Ea31a6c12318df947"), addr)
|
|
|
|
// what wallet.PrivateKeyHex(account) does
|
|
bytes := crypto.FromECDSA(pk)
|
|
pkStr := hexutil.Encode(bytes)[2:]
|
|
require.Equal(t, "63e21d10fd50155dbba0e7d3f7431a400b84b4c2ac1ee38872f82448fe3ecfb9", pkStr)
|
|
|
|
addr, pk, err = phrase.ethereumKeyFromMnemonic(1, defaultEthereumDerivation)
|
|
require.NoError(t, err)
|
|
require.Equal(t, common.HexToAddress("0x8230645aC28A4EdD1b0B53E7Cd8019744E9dD559"), addr)
|
|
|
|
bytes = crypto.FromECDSA(pk)
|
|
pkStr = hexutil.Encode(bytes)[2:]
|
|
require.Equal(t, "b31048b0aa87649bdb9016c0ee28c788ddfc45e52cd71cc0da08c47cb4390ae7", pkStr)
|
|
}
|