|
@@ -87,6 +87,43 @@ static void furi_string_secure_free(FuriString* str) {
|
|
|
furi_string_free(str);
|
|
furi_string_free(str);
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
+static bool totp_cli_read_secret(Cli* cli, FuriString* out_str, bool mask_user_input) {
|
|
|
|
|
+ uint8_t c;
|
|
|
|
|
+ while(cli_read(cli, &c, 1) == 1) {
|
|
|
|
|
+ if(c == CliSymbolAsciiEsc) {
|
|
|
|
|
+ // Some keys generating escape-sequences
|
|
|
|
|
+ // We need to ignore them as we case about alpha-numerics only
|
|
|
|
|
+ uint8_t c2;
|
|
|
|
|
+ cli_read_timeout(cli, &c2, 1, 0);
|
|
|
|
|
+ cli_read_timeout(cli, &c2, 1, 0);
|
|
|
|
|
+ } else if(c == CliSymbolAsciiETX) {
|
|
|
|
|
+ TOTP_CLI_DELETE_CURRENT_LINE();
|
|
|
|
|
+ TOTP_CLI_PRINTF("Cancelled by user\r\n");
|
|
|
|
|
+ return false;
|
|
|
|
|
+ } else if((c >= '0' && c <= '9') || (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z')) {
|
|
|
|
|
+ if(mask_user_input) {
|
|
|
|
|
+ putc('*', stdout);
|
|
|
|
|
+ } else {
|
|
|
|
|
+ putc(c, stdout);
|
|
|
|
|
+ }
|
|
|
|
|
+ fflush(stdout);
|
|
|
|
|
+ furi_string_push_back(out_str, c);
|
|
|
|
|
+ } else if(c == CliSymbolAsciiBackspace || c == CliSymbolAsciiDel) {
|
|
|
|
|
+ size_t out_str_size = furi_string_size(out_str);
|
|
|
|
|
+ if(out_str_size > 0) {
|
|
|
|
|
+ TOTP_CLI_DELETE_LAST_CHAR();
|
|
|
|
|
+ furi_string_left(out_str, out_str_size - 1);
|
|
|
|
|
+ }
|
|
|
|
|
+ } else if(c == CliSymbolAsciiCR) {
|
|
|
|
|
+ cli_nl();
|
|
|
|
|
+ break;
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ TOTP_CLI_DELETE_LAST_LINE();
|
|
|
|
|
+ return true;
|
|
|
|
|
+}
|
|
|
|
|
+
|
|
|
void totp_cli_command_add_handle(PluginState* plugin_state, FuriString* args, Cli* cli) {
|
|
void totp_cli_command_add_handle(PluginState* plugin_state, FuriString* args, Cli* cli) {
|
|
|
FuriString* temp_str = furi_string_alloc();
|
|
FuriString* temp_str = furi_string_alloc();
|
|
|
TokenInfo* token_info = token_info_alloc();
|
|
TokenInfo* token_info = token_info_alloc();
|
|
@@ -148,44 +185,8 @@ void totp_cli_command_add_handle(PluginState* plugin_state, FuriString* args, Cl
|
|
|
// Reading token secret
|
|
// Reading token secret
|
|
|
furi_string_reset(temp_str);
|
|
furi_string_reset(temp_str);
|
|
|
TOTP_CLI_PRINTF("Enter token secret and confirm with [ENTER]\r\n");
|
|
TOTP_CLI_PRINTF("Enter token secret and confirm with [ENTER]\r\n");
|
|
|
-
|
|
|
|
|
- uint8_t c;
|
|
|
|
|
- while(cli_read(cli, &c, 1) == 1) {
|
|
|
|
|
- if(c == CliSymbolAsciiEsc) {
|
|
|
|
|
- // Some keys generating escape-sequences
|
|
|
|
|
- // We need to ignore them as we case about alpha-numerics only
|
|
|
|
|
- uint8_t c2;
|
|
|
|
|
- cli_read_timeout(cli, &c2, 1, 0);
|
|
|
|
|
- cli_read_timeout(cli, &c2, 1, 0);
|
|
|
|
|
- } else if(c == CliSymbolAsciiETX) {
|
|
|
|
|
- TOTP_CLI_DELETE_CURRENT_LINE();
|
|
|
|
|
- TOTP_CLI_PRINTF("Cancelled by user\r\n");
|
|
|
|
|
- furi_string_secure_free(temp_str);
|
|
|
|
|
- token_info_free(token_info);
|
|
|
|
|
- return;
|
|
|
|
|
- } else if((c >= '0' && c <= '9') || (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z')) {
|
|
|
|
|
- if(mask_user_input) {
|
|
|
|
|
- putc('*', stdout);
|
|
|
|
|
- } else {
|
|
|
|
|
- putc(c, stdout);
|
|
|
|
|
- }
|
|
|
|
|
- fflush(stdout);
|
|
|
|
|
- furi_string_push_back(temp_str, c);
|
|
|
|
|
- } else if(c == CliSymbolAsciiBackspace || c == CliSymbolAsciiDel) {
|
|
|
|
|
- size_t temp_str_size = furi_string_size(temp_str);
|
|
|
|
|
- if(temp_str_size > 0) {
|
|
|
|
|
- TOTP_CLI_DELETE_LAST_CHAR();
|
|
|
|
|
- furi_string_left(temp_str, temp_str_size - 1);
|
|
|
|
|
- }
|
|
|
|
|
- } else if(c == CliSymbolAsciiCR) {
|
|
|
|
|
- cli_nl();
|
|
|
|
|
- break;
|
|
|
|
|
- }
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- TOTP_CLI_DELETE_LAST_LINE();
|
|
|
|
|
-
|
|
|
|
|
- if(!totp_cli_ensure_authenticated(plugin_state, cli)) {
|
|
|
|
|
|
|
+ if(!totp_cli_read_secret(cli, temp_str, mask_user_input) ||
|
|
|
|
|
+ !totp_cli_ensure_authenticated(plugin_state, cli)) {
|
|
|
furi_string_secure_free(temp_str);
|
|
furi_string_secure_free(temp_str);
|
|
|
token_info_free(token_info);
|
|
token_info_free(token_info);
|
|
|
return;
|
|
return;
|
|
@@ -210,11 +211,7 @@ void totp_cli_command_add_handle(PluginState* plugin_state, FuriString* args, Cl
|
|
|
load_generate_token_scene = true;
|
|
load_generate_token_scene = true;
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- if(plugin_state->tokens_list == NULL) {
|
|
|
|
|
- plugin_state->tokens_list = list_init_head(token_info);
|
|
|
|
|
- } else {
|
|
|
|
|
- list_add(plugin_state->tokens_list, token_info);
|
|
|
|
|
- }
|
|
|
|
|
|
|
+ TOTP_LIST_INIT_OR_ADD(plugin_state->tokens_list, token_info);
|
|
|
plugin_state->tokens_count++;
|
|
plugin_state->tokens_count++;
|
|
|
totp_config_file_save_new_token(token_info);
|
|
totp_config_file_save_new_token(token_info);
|
|
|
|
|
|