aboutsummaryrefslogtreecommitdiffstats
path: root/crypto/keypair.go
blob: cc17328cb0411914896634ab7d5bfd691ad10be8 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
package crypto

import (
    "strings"

    "github.com/ethereum/go-ethereum/common"
    "github.com/ethereum/go-ethereum/crypto/secp256k1"
)

type KeyPair struct {
    PrivateKey []byte
    PublicKey  []byte
    address    []byte
    mnemonic   string
    // The associated account
    // account *StateObject
}

func GenerateNewKeyPair() *KeyPair {
    _, prv := secp256k1.GenerateKeyPair()
    keyPair, _ := NewKeyPairFromSec(prv) // swallow error, this one cannot err
    return keyPair
}

func NewKeyPairFromSec(seckey []byte) (*KeyPair, error) {
    pubkey, err := secp256k1.GeneratePubKey(seckey)
    if err != nil {
        return nil, err
    }

    return &KeyPair{PrivateKey: seckey, PublicKey: pubkey}, nil
}

func (k *KeyPair) Address() []byte {
    if k.address == nil {
        k.address = Sha3(k.PublicKey[1:])[12:]
    }
    return k.address
}

func (k *KeyPair) Mnemonic() string {
    if k.mnemonic == "" {
        k.mnemonic = strings.Join(MnemonicEncode(common.Bytes2Hex(k.PrivateKey)), " ")
    }
    return k.mnemonic
}

func (k *KeyPair) AsStrings() (string, string, string, string) {
    return k.Mnemonic(), common.Bytes2Hex(k.Address()), common.Bytes2Hex(k.PrivateKey), common.Bytes2Hex(k.PublicKey)
}