diff --git a/tools/unencrypted-keys-gen/main.go b/tools/unencrypted-keys-gen/main.go index 5e1425449a..85624d4cff 100644 --- a/tools/unencrypted-keys-gen/main.go +++ b/tools/unencrypted-keys-gen/main.go @@ -13,6 +13,7 @@ import ( var ( numKeys = flag.Int("num-keys", 0, "Number of validator private/withdrawal keys to generate") + startIndex = flag.Uint64("start-index", 0, "Start index for the determinstic keygen algorithm") outputJSON = flag.String("output-json", "", "JSON file to write output to") overwrite = flag.Bool("overwrite", false, "If the key file exists, it will be overwritten") ) @@ -53,18 +54,18 @@ func main() { } }() - ctnr := generateUnencryptedKeys() + ctnr := generateUnencryptedKeys(*startIndex) if err := SaveUnencryptedKeysToFile(file, ctnr); err != nil { log.Fatal(err) } } -func generateUnencryptedKeys() *UnencryptedKeysContainer { +func generateUnencryptedKeys(startIndex uint64) *UnencryptedKeysContainer { ctnr := &UnencryptedKeysContainer{ Keys: make([]*UnencryptedKeys, *numKeys), } - sks, _, err := interop.DeterministicallyGenerateKeys(0 /*startIndex*/, uint64(*numKeys)) + sks, _, err := interop.DeterministicallyGenerateKeys(startIndex, uint64(*numKeys)) if err != nil { panic(err) diff --git a/tools/unencrypted-keys-gen/main_test.go b/tools/unencrypted-keys-gen/main_test.go index 728e445e55..b473343e8a 100644 --- a/tools/unencrypted-keys-gen/main_test.go +++ b/tools/unencrypted-keys-gen/main_test.go @@ -8,7 +8,7 @@ import ( ) func TestSavesUnencryptedKeys(t *testing.T) { - ctnr := generateUnencryptedKeys() + ctnr := generateUnencryptedKeys(0 /* start index */) buf := new(bytes.Buffer) if err := SaveUnencryptedKeysToFile(buf, ctnr); err != nil { t.Fatal(err)