feat: enhance key generation with staking support

This commit is contained in:
Zach Kelling
2026-07-26 10:19:06 -07:00
committed by zeekay
parent eb270d8d50
commit 90b6856267
3 changed files with 274 additions and 10 deletions
+8
View File
@@ -0,0 +1,8 @@
-----BEGIN CERTIFICATE-----
MIIBETCBt6ADAgECAgEAMAoGCCqGSM49BAMCMAAwIBcNOTkxMjMxMDAwMDAwWhgP
MjEyNTEyMjMwMjM5NDBaMAAwWTATBgcqhkjOPQIBBggqhkjOPQMBBwNCAASS0Q9K
ihP2PHxZSo1hF44wDD1UzCCmVXmieAtBu9qv5LM4UFxYSeYC18vSyk5gvVxkSzd6
opESXvONuu7glmonoyAwHjAOBgNVHQ8BAf8EBAMCB4AwDAYDVR0TAQH/BAIwADAK
BggqhkjOPQQDAgNJADBGAiEAljK6W6COvjS2JfphiyYa7vSKktrTachOCoKpUGDH
90ICIQDnao4xD8nyU7ZAQm/9vLsiGjj1CL3AxlXM9ZtiY5rAoQ==
-----END CERTIFICATE-----
+5
View File
@@ -0,0 +1,5 @@
-----BEGIN PRIVATE KEY-----
MIGHAgEAMBMGByqGSM49AgEGCCqGSM49AwEHBG0wawIBAQQgvNkzVzWhxsq7Csv9
zPLZ3gnClFAJ1QOddbOU1J/Ex/ChRANCAASS0Q9KihP2PHxZSo1hF44wDD1UzCCm
VXmieAtBu9qv5LM4UFxYSeYC18vSyk5gvVxkSzd6opESXvONuu7glmon
-----END PRIVATE KEY-----
+261 -10
View File
@@ -10,9 +10,11 @@ package keys
import ( import (
"crypto/ecdsa" "crypto/ecdsa"
"crypto/elliptic"
"encoding/base64" "encoding/base64"
"encoding/hex" "encoding/hex"
"fmt" "fmt"
"math/big"
"os" "os"
"path/filepath" "path/filepath"
"strings" "strings"
@@ -20,6 +22,8 @@ import (
"github.com/luxfi/crypto/bls" "github.com/luxfi/crypto/bls"
"github.com/luxfi/crypto/bls/signer/localsigner" "github.com/luxfi/crypto/bls/signer/localsigner"
luxcrypto "github.com/luxfi/crypto/secp256k1" luxcrypto "github.com/luxfi/crypto/secp256k1"
"github.com/luxfi/go-bip32"
"github.com/luxfi/go-bip39"
"github.com/luxfi/ids" "github.com/luxfi/ids"
"github.com/luxfi/node/staking" "github.com/luxfi/node/staking"
"github.com/luxfi/node/vms/platformvm/signer" "github.com/luxfi/node/vms/platformvm/signer"
@@ -211,28 +215,42 @@ func (ks *KeyStore) Load(name string) (*ValidatorKey, error) {
func LoadFromDir(nodeDir string) (*ValidatorKey, error) { func LoadFromDir(nodeDir string) (*ValidatorKey, error) {
vk := &ValidatorKey{} vk := &ValidatorKey{}
// Load TLS cert - try modern path first // Load TLS cert - try modern path first, then legacy
certPath := filepath.Join(nodeDir, "staking", "staker.crt") certPath := filepath.Join(nodeDir, "staking", "staker.crt")
certPEM, err := os.ReadFile(certPath) certPEM, err := os.ReadFile(certPath)
if err != nil { if err != nil {
certPath = filepath.Join(nodeDir, "staker.crt") certPath = filepath.Join(nodeDir, "staker.crt")
certPEM, err = os.ReadFile(certPath) certPEM, err = os.ReadFile(certPath)
if err != nil {
return nil, fmt.Errorf("failed to read staker.crt: %w", err)
}
} }
vk.StakerCert = certPEM
// Load TLS key // Load TLS key
keyPath := filepath.Join(nodeDir, "staking", "staker.key") keyPath := filepath.Join(nodeDir, "staking", "staker.key")
keyPEM, err := os.ReadFile(keyPath) keyPEM, kerr := os.ReadFile(keyPath)
if err != nil { if kerr != nil {
keyPath = filepath.Join(nodeDir, "staker.key") keyPath = filepath.Join(nodeDir, "staker.key")
keyPEM, err = os.ReadFile(keyPath) keyPEM, kerr = os.ReadFile(keyPath)
}
// If TLS cert/key missing, generate them and persist
if err != nil || kerr != nil || len(certPEM) == 0 || len(keyPEM) == 0 {
fmt.Printf(" Generating TLS staking cert for %s\n", filepath.Base(nodeDir))
certPEM, keyPEM, err = staking.NewCertAndKeyBytes()
if err != nil { if err != nil {
return nil, fmt.Errorf("failed to read staker.key: %w", err) return nil, fmt.Errorf("failed to generate TLS cert: %w", err)
}
// Save to disk for future use
stakingDir := filepath.Join(nodeDir, "staking")
if err := os.MkdirAll(stakingDir, 0700); err != nil {
return nil, fmt.Errorf("failed to create staking dir: %w", err)
}
if err := os.WriteFile(filepath.Join(stakingDir, "staker.key"), keyPEM, 0600); err != nil {
return nil, fmt.Errorf("failed to write staker.key: %w", err)
}
if err := os.WriteFile(filepath.Join(stakingDir, "staker.crt"), certPEM, 0644); err != nil {
return nil, fmt.Errorf("failed to write staker.crt: %w", err)
} }
} }
vk.StakerCert = certPEM
vk.StakerKey = keyPEM vk.StakerKey = keyPEM
// Derive NodeID from TLS cert // Derive NodeID from TLS cert
@@ -309,7 +327,16 @@ func (ks *KeyStore) List() ([]string, error) {
var names []string var names []string
for _, entry := range entries { for _, entry := range entries {
if entry.IsDir() { if entry.IsDir() {
names = append(names, entry.Name()) name := entry.Name()
// Skip hidden directories (like .git) and non-node directories
if strings.HasPrefix(name, ".") {
continue
}
// Only include node* directories
if !strings.HasPrefix(name, "node") {
continue
}
names = append(names, name)
} }
} }
return names, nil return names, nil
@@ -370,3 +397,227 @@ func (vk *ValidatorKey) BLSPoPHex() string {
func (vk *ValidatorKey) CChainAddrHex() string { func (vk *ValidatorKey) CChainAddrHex() string {
return "0x" + hex.EncodeToString(vk.CChainAddr[:]) return "0x" + hex.EncodeToString(vk.CChainAddr[:])
} }
// DeriveValidatorsFromMnemonic derives N validator keys from a BIP39 mnemonic.
// Each validator uses BIP44 path m/44'/9000'/0'/0/{index} for the EC key.
// TLS staking certs and BLS keys are generated fresh (not deterministic from mnemonic).
// This is designed for runtime use - no files are written to disk.
func DeriveValidatorsFromMnemonic(mnemonic string, count int) ([]*ValidatorKey, error) {
if count <= 0 || count > 100 {
return nil, fmt.Errorf("invalid validator count: %d (must be 1-100)", count)
}
validators := make([]*ValidatorKey, count)
for i := 0; i < count; i++ {
vk, err := DeriveValidatorFromMnemonic(mnemonic, uint32(i))
if err != nil {
return nil, fmt.Errorf("failed to derive validator %d: %w", i, err)
}
validators[i] = vk
}
return validators, nil
}
// DeriveValidatorFromMnemonic derives a single validator key from mnemonic at given index.
// All keys (EC, TLS, BLS) are now derived deterministically from the mnemonic.
func DeriveValidatorFromMnemonic(mnemonic string, accountIndex uint32) (*ValidatorKey, error) {
vk := &ValidatorKey{}
// 1. Derive EC key from mnemonic using BIP44 path m/44'/9000'/0'/0/{index}
ecKeyBytes, err := deriveMnemonicKey(mnemonic, accountIndex)
if err != nil {
return nil, fmt.Errorf("failed to derive EC key: %w", err)
}
vk.ECPrivateKey = ecKeyBytes
// Derive P-chain and C-chain addresses
luxPrivKey, err := luxcrypto.ToPrivateKey(ecKeyBytes)
if err != nil {
return nil, fmt.Errorf("failed to create secp256k1 key: %w", err)
}
pubKey := luxPrivKey.PublicKey()
vk.PChainAddr = ids.ShortID(pubKey.Address())
vk.CChainAddr = pubkeyToAddress(pubKey.ToECDSA())
// 2. Derive TLS staking cert deterministically from mnemonic
// Use a separate derivation path: m/44'/9000'/1'/0/{index} for TLS keys
tlsKeyBytes, err := deriveMnemonicKeyForPath(mnemonic, 1, accountIndex) // account=1 for TLS
if err != nil {
return nil, fmt.Errorf("failed to derive TLS key seed: %w", err)
}
// Create P-256 private key from derived seed (TLS uses P-256, not secp256k1)
p256Key, err := deriveP256Key(tlsKeyBytes)
if err != nil {
return nil, fmt.Errorf("failed to derive P-256 key: %w", err)
}
certPEM, keyPEM, err := staking.NewCertAndKeyBytesFromKey(p256Key)
if err != nil {
return nil, fmt.Errorf("failed to generate TLS cert: %w", err)
}
vk.StakerCert = certPEM
vk.StakerKey = keyPEM
// Derive NodeID from TLS cert
tlsCert, err := staking.LoadTLSCertFromBytes(keyPEM, certPEM)
if err != nil {
return nil, fmt.Errorf("failed to parse TLS cert: %w", err)
}
stakingCert := &ids.Certificate{
Raw: tlsCert.Leaf.Raw,
PublicKey: tlsCert.Leaf.PublicKey,
}
vk.NodeID = ids.NodeIDFromCert(stakingCert)
// 3. Derive BLS signer key deterministically from mnemonic
// Use a separate derivation path: m/44'/9000'/2'/0/{index} for BLS keys
blsSeed, err := deriveMnemonicKeyForPath(mnemonic, 2, accountIndex) // account=2 for BLS
if err != nil {
return nil, fmt.Errorf("failed to derive BLS key seed: %w", err)
}
// Create BLS signer from seed using proper BLS key derivation (handles field order internally)
blsKey, err := localsigner.FromSeed(blsSeed)
if err != nil {
return nil, fmt.Errorf("failed to create BLS key from seed: %w", err)
}
vk.BLSSecretKey = blsKey.ToBytes()
pop, err := signer.NewProofOfPossession(blsKey)
if err != nil {
return nil, fmt.Errorf("failed to generate BLS PoP: %w", err)
}
vk.BLSPublicKey = pop.PublicKey[:]
vk.BLSPoP = pop.ProofOfPossession[:]
return vk, nil
}
// deriveP256Key creates an ECDSA P-256 private key from a 32-byte seed.
// This allows deterministic TLS key generation from mnemonic-derived seeds.
func deriveP256Key(seed []byte) (*ecdsa.PrivateKey, error) {
if len(seed) < 32 {
return nil, fmt.Errorf("seed must be at least 32 bytes")
}
// Use the seed as the private key scalar (reduced mod curve order)
curve := elliptic.P256()
k := new(big.Int).SetBytes(seed[:32])
k.Mod(k, curve.Params().N)
// Ensure k is not zero
if k.Sign() == 0 {
k.SetInt64(1)
}
priv := &ecdsa.PrivateKey{
PublicKey: ecdsa.PublicKey{
Curve: curve,
},
D: k,
}
priv.PublicKey.X, priv.PublicKey.Y = curve.ScalarBaseMult(k.Bytes())
return priv, nil
}
// deriveMnemonicKeyForPath derives a key using BIP44 path m/44'/9000'/{account}'/0/{index}
func deriveMnemonicKeyForPath(mnemonic string, account, index uint32) ([]byte, error) {
if !bip39.IsMnemonicValid(mnemonic) {
return nil, fmt.Errorf("invalid mnemonic phrase")
}
seed := bip39.NewSeed(mnemonic, "")
// Create master key from seed
masterKey, err := bip32.NewMasterKey(seed)
if err != nil {
return nil, fmt.Errorf("failed to create master key: %w", err)
}
// BIP-44 path: m/44'/9000'/{account}'/0/{index}
// m/44' (purpose)
key, err := masterKey.NewChildKey(bip32.FirstHardenedChild + 44)
if err != nil {
return nil, fmt.Errorf("failed to derive purpose: %w", err)
}
// m/44'/9000' (coin type for LUX)
key, err = key.NewChildKey(bip32.FirstHardenedChild + LUXCoinType)
if err != nil {
return nil, fmt.Errorf("failed to derive coin type: %w", err)
}
// m/44'/9000'/{account}' (account - 0=EC, 1=TLS, 2=BLS)
key, err = key.NewChildKey(bip32.FirstHardenedChild + account)
if err != nil {
return nil, fmt.Errorf("failed to derive account: %w", err)
}
// m/44'/9000'/{account}'/0 (change)
key, err = key.NewChildKey(0)
if err != nil {
return nil, fmt.Errorf("failed to derive change: %w", err)
}
// m/44'/9000'/{account}'/0/{index} (address index)
key, err = key.NewChildKey(index)
if err != nil {
return nil, fmt.Errorf("failed to derive address index: %w", err)
}
return key.Key, nil
}
// LUXCoinType is the BIP-44 coin type for LUX (9000')
const LUXCoinType = 9000
// deriveMnemonicKey derives an EC private key from mnemonic using BIP44 path m/44'/9000'/0'/0/{index}
func deriveMnemonicKey(mnemonic string, accountIndex uint32) ([]byte, error) {
if !bip39.IsMnemonicValid(mnemonic) {
return nil, fmt.Errorf("invalid mnemonic phrase")
}
seed := bip39.NewSeed(mnemonic, "")
// Create master key from seed
masterKey, err := bip32.NewMasterKey(seed)
if err != nil {
return nil, fmt.Errorf("failed to create master key: %w", err)
}
// BIP-44 path: m/44'/9000'/0'/0/{accountIndex}
// m/44' (purpose)
key, err := masterKey.NewChildKey(bip32.FirstHardenedChild + 44)
if err != nil {
return nil, fmt.Errorf("failed to derive purpose: %w", err)
}
// m/44'/9000' (coin type for LUX)
key, err = key.NewChildKey(bip32.FirstHardenedChild + LUXCoinType)
if err != nil {
return nil, fmt.Errorf("failed to derive coin type: %w", err)
}
// m/44'/9000'/0' (account)
key, err = key.NewChildKey(bip32.FirstHardenedChild + 0)
if err != nil {
return nil, fmt.Errorf("failed to derive account: %w", err)
}
// m/44'/9000'/0'/0 (change)
key, err = key.NewChildKey(0)
if err != nil {
return nil, fmt.Errorf("failed to derive change: %w", err)
}
// m/44'/9000'/0'/0/{accountIndex} (address index)
key, err = key.NewChildKey(accountIndex)
if err != nil {
return nil, fmt.Errorf("failed to derive address index: %w", err)
}
return key.Key, nil
}