@@ -336,15 +336,15 @@ func NewDefaultConfigNNodes(binaryPath string, numNodes uint32) (network.Config,
336
336
netConfig .BinaryPath = binaryPath
337
337
if int (numNodes ) > constants .DefaultNumNodes {
338
338
toAdd := int (numNodes ) - constants .DefaultNumNodes
339
- keys , err := getNNodeKeys (toAdd )
339
+ keys , err := generateNNodeKeys (toAdd )
340
340
if err != nil {
341
341
return netConfig , err
342
342
}
343
343
for i := 0 ; i < toAdd ; i ++ {
344
344
netConfig .NodeConfigs = append (netConfig .NodeConfigs , node.Config {
345
- StakingKey : string (keys [i ].stakingKey ),
346
- StakingCert : string (keys [i ].stakingCert ),
347
- StakingSigningKey : base64 .StdEncoding .EncodeToString (keys [i ].blsKey ),
345
+ StakingKey : string (keys [i ].StakingKey ),
346
+ StakingCert : string (keys [i ].StakingCert ),
347
+ StakingSigningKey : base64 .StdEncoding .EncodeToString (keys [i ].BlsKey ),
348
348
IsBeacon : true ,
349
349
})
350
350
}
@@ -376,35 +376,49 @@ func NewDefaultConfig(binaryPath string) (network.Config, error) {
376
376
return NewDefaultConfigNNodes (binaryPath , constants .DefaultNumNodes )
377
377
}
378
378
379
- type nodeKeys struct {
380
- stakingKey [] byte
381
- stakingCert [] byte
382
- blsKey [] byte
379
+ type EncodedNodeKeys struct {
380
+ StakingKey string
381
+ StakingCert string
382
+ BlsKey string
383
383
}
384
384
385
- func getNodeKeys () (* nodeKeys , error ) {
386
- keys := nodeKeys {}
385
+ type NodeKeys struct {
386
+ StakingKey []byte
387
+ StakingCert []byte
388
+ BlsKey []byte
389
+ }
390
+
391
+ func encodeNodeKeys (key * NodeKeys ) * EncodedNodeKeys {
392
+ return & EncodedNodeKeys {
393
+ StakingKey : string (key .StakingKey ),
394
+ StakingCert : string (key .StakingCert ),
395
+ BlsKey : base64 .StdEncoding .EncodeToString (key .BlsKey ),
396
+ }
397
+ }
398
+
399
+ func generateNodeKeys () (* NodeKeys , error ) {
387
400
stakingCert , stakingKey , err := staking .NewCertAndKeyBytes ()
388
401
if err != nil {
389
402
return nil , fmt .Errorf ("couldn't generate staking Cert/Key: %w" , err )
390
403
}
391
- keys .stakingKey = stakingKey
392
- keys .stakingCert = stakingCert
393
404
key , err := bls .NewSecretKey ()
394
405
if err != nil {
395
406
return nil , fmt .Errorf ("couldn't generate new signing key: %w" , err )
396
407
}
397
- keys .blsKey = bls .SecretKeyToBytes (key )
398
- return & keys , nil
408
+ return & NodeKeys {
409
+ StakingKey : stakingKey ,
410
+ StakingCert : stakingCert ,
411
+ BlsKey : bls .SecretKeyToBytes (key ),
412
+ }, nil
399
413
}
400
414
401
- func getNNodeKeys (num int ) ([]* nodeKeys , error ) {
402
- nodesKeys := []* nodeKeys {}
415
+ func generateNNodeKeys (num int ) ([]* NodeKeys , error ) {
416
+ nodesKeys := []* NodeKeys {}
403
417
lock := sync.Mutex {}
404
418
eg := errgroup.Group {}
405
419
for i := 0 ; i < num ; i ++ {
406
420
eg .Go (func () error {
407
- keys , err := getNodeKeys ()
421
+ keys , err := generateNodeKeys ()
408
422
if err != nil {
409
423
return err
410
424
}
0 commit comments