diff --git a/c_src/cmd_key.c b/c_src/cmd_key.c index 96206c4c..d66ac7c5 100644 --- a/c_src/cmd_key.c +++ b/c_src/cmd_key.c @@ -1,4 +1,5 @@ #include +#include #include #include @@ -6,6 +7,7 @@ #include "libbcachefs/checksum.h" #include "crypto.h" #include "libbcachefs.h" +#include "tools-util.h" static void unlock_usage(void) { @@ -16,6 +18,7 @@ static void unlock_usage(void) " -c Check if a device is encrypted\n" " -k (session|user|user_session)\n" " Keyring to add to (default: user)\n" + " -f Keyfile to read from (disables password prompt)\n" " -h Display this help and exit\n" "Report bugs to "); } @@ -24,9 +27,12 @@ int cmd_unlock(int argc, char *argv[]) { const char *keyring = "user"; bool check = false; + const char *key_file_path = NULL; + char *passphrase = NULL; + int opt; - while ((opt = getopt(argc, argv, "ck:h")) != -1) + while ((opt = getopt(argc, argv, "cf:k:h")) != -1) switch (opt) { case 'c': check = true; @@ -34,6 +40,9 @@ int cmd_unlock(int argc, char *argv[]) case 'k': keyring = strdup(optarg); break; + case 'f': + key_file_path = strdup(optarg); + break; case 'h': unlock_usage(); exit(EXIT_SUCCESS); @@ -62,8 +71,11 @@ int cmd_unlock(int argc, char *argv[]) if (check) exit(EXIT_SUCCESS); - - char *passphrase = read_passphrase("Enter passphrase: "); + if (key_file_path){ + passphrase = read_file_str(AT_FDCWD, key_file_path); + } else { + passphrase = read_passphrase("Enter passphrase: "); + } bch2_add_key(sb.sb, "user", keyring, passphrase);