fix: add safe notice to user

bnb-chain · Oct 11, 2023 · ab25f77 · ab25f77
1 parent 3443acb
commit ab25f77
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/cmd/cmd_account.go b/cmd/cmd_account.go
@@ -250,10 +250,11 @@ func importKey(ctx *cli.Context) error {
 	}
 
 	// fetch password content
-	password, err = getPassword(ctx)
+	password, err = getPassword(ctx, true)
 	if err != nil {
 		return toCmdErr(err)
 	}
+	fmt.Println("- You must REMEMBER your password! Without the password, it's impossible to decrypt the key!")
 
 	// encrypt the private key
 	encryptContent, err = EncryptKey(key, password, EncryptScryptN, EncryptScryptP)
@@ -391,7 +392,7 @@ func createAccount(ctx *cli.Context) error {
 	}
 
 	// fetch password content
-	password, err = getPassword(ctx)
+	password, err = getPassword(ctx, true)
 	if err != nil {
 		return toCmdErr(err)
 	}
@@ -524,7 +525,7 @@ func parseKeystore(ctx *cli.Context) (string, string, error) {
 		return "", "", toCmdErr(err)
 	}
 	// fetch password content
-	password, err := getPassword(ctx)
+	password, err := getPassword(ctx, false)
 	if err != nil {
 		return "", "", toCmdErr(err)
 	}

diff --git a/cmd/utils.go b/cmd/utils.go
@@ -338,7 +338,7 @@ func parseActions(ctx *cli.Context, resourceType ResourceType) ([]permTypes.Acti
 }
 
 // getPassword return the password content
-func getPassword(ctx *cli.Context) (string, error) {
+func getPassword(ctx *cli.Context, needNotice bool) (string, error) {
 	var filepath string
 	if passwordFile := ctx.String(passwordFileFlag); passwordFile != "" {
 		filepath = passwordFile
@@ -350,13 +350,18 @@ func getPassword(ctx *cli.Context) (string, error) {
 	}
 
 	fmt.Print("Please enter the passphrase now:")
+
 	bytePassword, err := term.ReadPassword(int(os.Stdin.Fd()))
 	if err != nil {
 		fmt.Println("read password err:", err)
 		return "", err
 	}
 	password := string(bytePassword)
 	fmt.Println()
+	if needNotice {
+		fmt.Println("- You must BACKUP your key file! Without the key, it's impossible to set transaction to greenfield!")
+		fmt.Println("- You must REMEMBER your password! Without the password, it's impossible to decrypt the key!")
+	}
 	return password, nil
 }