1
mirror of https://github.com/DarkFlippers/unleashed-firmware.git synced 2025-12-12 04:34:43 +04:00

Merge remote-tracking branch 'OFW/porta/cli-event-ordering' into dev

This commit is contained in:
MX
2025-04-08 17:50:48 +03:00
7 changed files with 61 additions and 56 deletions

4
.gitignore vendored
View File

@@ -73,3 +73,7 @@ PVS-Studio.log
# JS packages # JS packages
node_modules/ node_modules/
# cli_perf script output in case of errors
/block.bin
/return_block.bin

View File

@@ -50,7 +50,7 @@ struct CliVcp {
PipeSide* own_pipe; PipeSide* own_pipe;
PipeSide* shell_pipe; PipeSide* shell_pipe;
bool is_currently_transmitting; volatile bool is_currently_transmitting;
size_t previous_tx_length; size_t previous_tx_length;
CliRegistry* main_registry; CliRegistry* main_registry;
@@ -106,6 +106,7 @@ static void cli_vcp_signal_internal_event(CliVcp* cli_vcp, CliVcpInternalEvent e
static void cli_vcp_cdc_tx_done(void* context) { static void cli_vcp_cdc_tx_done(void* context) {
CliVcp* cli_vcp = context; CliVcp* cli_vcp = context;
cli_vcp->is_currently_transmitting = false;
cli_vcp_signal_internal_event(cli_vcp, CliVcpInternalEventTxDone); cli_vcp_signal_internal_event(cli_vcp, CliVcpInternalEventTxDone);
} }
@@ -195,6 +196,16 @@ static void cli_vcp_internal_event_happened(void* context) {
CliVcpInternalEvent event = furi_thread_flags_wait(CliVcpInternalEventAll, FuriFlagWaitAny, 0); CliVcpInternalEvent event = furi_thread_flags_wait(CliVcpInternalEventAll, FuriFlagWaitAny, 0);
furi_check(!(event & FuriFlagError)); furi_check(!(event & FuriFlagError));
if(event & CliVcpInternalEventRx) {
VCP_TRACE(TAG, "Rx");
cli_vcp_maybe_receive_data(cli_vcp);
}
if(event & CliVcpInternalEventTxDone) {
VCP_TRACE(TAG, "TxDone");
cli_vcp_maybe_send_data(cli_vcp);
}
if(event & CliVcpInternalEventDisconnected) { if(event & CliVcpInternalEventDisconnected) {
if(!cli_vcp->is_connected) return; if(!cli_vcp->is_connected) return;
FURI_LOG_D(TAG, "Disconnected"); FURI_LOG_D(TAG, "Disconnected");
@@ -234,17 +245,6 @@ static void cli_vcp_internal_event_happened(void* context) {
cli_main_motd, NULL, cli_vcp->shell_pipe, cli_vcp->main_registry, &cli_main_ext_config); cli_main_motd, NULL, cli_vcp->shell_pipe, cli_vcp->main_registry, &cli_main_ext_config);
cli_shell_start(cli_vcp->shell); cli_shell_start(cli_vcp->shell);
} }
if(event & CliVcpInternalEventRx) {
VCP_TRACE(TAG, "Rx");
cli_vcp_maybe_receive_data(cli_vcp);
}
if(event & CliVcpInternalEventTxDone) {
VCP_TRACE(TAG, "TxDone");
cli_vcp->is_currently_transmitting = false;
cli_vcp_maybe_send_data(cli_vcp);
}
} }
// ============ // ============

View File

@@ -3,16 +3,16 @@
#include <toolbox/pipe.h> #include <toolbox/pipe.h>
#include <storage/storage.h> #include <storage/storage.h>
#define TAG "cli" #define TAG "CliRegistry"
struct CliRegistry { struct CliRegistry {
CliCommandTree_t commands; CliCommandDict_t commands;
FuriMutex* mutex; FuriMutex* mutex;
}; };
CliRegistry* cli_registry_alloc(void) { CliRegistry* cli_registry_alloc(void) {
CliRegistry* registry = malloc(sizeof(CliRegistry)); CliRegistry* registry = malloc(sizeof(CliRegistry));
CliCommandTree_init(registry->commands); CliCommandDict_init(registry->commands);
registry->mutex = furi_mutex_alloc(FuriMutexTypeRecursive); registry->mutex = furi_mutex_alloc(FuriMutexTypeRecursive);
return registry; return registry;
} }
@@ -20,7 +20,7 @@ CliRegistry* cli_registry_alloc(void) {
void cli_registry_free(CliRegistry* registry) { void cli_registry_free(CliRegistry* registry) {
furi_check(furi_mutex_acquire(registry->mutex, FuriWaitForever) == FuriStatusOk); furi_check(furi_mutex_acquire(registry->mutex, FuriWaitForever) == FuriStatusOk);
furi_mutex_free(registry->mutex); furi_mutex_free(registry->mutex);
CliCommandTree_clear(registry->commands); CliCommandDict_clear(registry->commands);
free(registry); free(registry);
} }
@@ -61,7 +61,7 @@ void cli_registry_add_command_ex(
}; };
furi_check(furi_mutex_acquire(registry->mutex, FuriWaitForever) == FuriStatusOk); furi_check(furi_mutex_acquire(registry->mutex, FuriWaitForever) == FuriStatusOk);
CliCommandTree_set_at(registry->commands, name_str, command); CliCommandDict_set_at(registry->commands, name_str, command);
furi_check(furi_mutex_release(registry->mutex) == FuriStatusOk); furi_check(furi_mutex_release(registry->mutex) == FuriStatusOk);
furi_string_free(name_str); furi_string_free(name_str);
@@ -79,7 +79,7 @@ void cli_registry_delete_command(CliRegistry* registry, const char* name) {
} while(name_replace != FURI_STRING_FAILURE); } while(name_replace != FURI_STRING_FAILURE);
furi_check(furi_mutex_acquire(registry->mutex, FuriWaitForever) == FuriStatusOk); furi_check(furi_mutex_acquire(registry->mutex, FuriWaitForever) == FuriStatusOk);
CliCommandTree_erase(registry->commands, name_str); CliCommandDict_erase(registry->commands, name_str);
furi_check(furi_mutex_release(registry->mutex) == FuriStatusOk); furi_check(furi_mutex_release(registry->mutex) == FuriStatusOk);
furi_string_free(name_str); furi_string_free(name_str);
@@ -91,7 +91,7 @@ bool cli_registry_get_command(
CliRegistryCommand* result) { CliRegistryCommand* result) {
furi_assert(registry); furi_assert(registry);
furi_check(furi_mutex_acquire(registry->mutex, FuriWaitForever) == FuriStatusOk); furi_check(furi_mutex_acquire(registry->mutex, FuriWaitForever) == FuriStatusOk);
CliRegistryCommand* data = CliCommandTree_get(registry->commands, command); CliRegistryCommand* data = CliCommandDict_get(registry->commands, command);
if(data) *result = *data; if(data) *result = *data;
furi_check(furi_mutex_release(registry->mutex) == FuriStatusOk); furi_check(furi_mutex_release(registry->mutex) == FuriStatusOk);
@@ -103,16 +103,14 @@ void cli_registry_remove_external_commands(CliRegistry* registry) {
furi_check(registry); furi_check(registry);
furi_check(furi_mutex_acquire(registry->mutex, FuriWaitForever) == FuriStatusOk); furi_check(furi_mutex_acquire(registry->mutex, FuriWaitForever) == FuriStatusOk);
// FIXME FL-3977: memory leak somewhere within this function CliCommandDict_t internal_cmds;
CliCommandDict_init(internal_cmds);
CliCommandTree_t internal_cmds;
CliCommandTree_init(internal_cmds);
for for
M_EACH(item, registry->commands, CliCommandTree_t) { M_EACH(item, registry->commands, CliCommandDict_t) {
if(!(item->value_ptr->flags & CliCommandFlagExternal)) if(!(item->value.flags & CliCommandFlagExternal))
CliCommandTree_set_at(internal_cmds, *item->key_ptr, *item->value_ptr); CliCommandDict_set_at(internal_cmds, item->key, item->value);
} }
CliCommandTree_move(registry->commands, internal_cmds); CliCommandDict_move(registry->commands, internal_cmds);
furi_check(furi_mutex_release(registry->mutex) == FuriStatusOk); furi_check(furi_mutex_release(registry->mutex) == FuriStatusOk);
} }
@@ -148,7 +146,7 @@ void cli_registry_reload_external_commands(
.execute_callback = NULL, .execute_callback = NULL,
.flags = CliCommandFlagExternal, .flags = CliCommandFlagExternal,
}; };
CliCommandTree_set_at(registry->commands, plugin_name, command); CliCommandDict_set_at(registry->commands, plugin_name, command);
} }
furi_string_free(plugin_name); furi_string_free(plugin_name);
@@ -172,7 +170,7 @@ void cli_registry_unlock(CliRegistry* registry) {
furi_mutex_release(registry->mutex); furi_mutex_release(registry->mutex);
} }
CliCommandTree_t* cli_registry_get_commands(CliRegistry* registry) { CliCommandDict_t* cli_registry_get_commands(CliRegistry* registry) {
furi_assert(registry); furi_assert(registry);
return &registry->commands; return &registry->commands;
} }

View File

@@ -6,7 +6,7 @@
#pragma once #pragma once
#include <furi.h> #include <furi.h>
#include <m-bptree.h> #include <m-dict.h>
#include "cli_registry.h" #include "cli_registry.h"
#ifdef __cplusplus #ifdef __cplusplus
@@ -22,19 +22,9 @@ typedef struct {
size_t stack_depth; size_t stack_depth;
} CliRegistryCommand; } CliRegistryCommand;
#define CLI_COMMANDS_TREE_RANK 4 DICT_DEF2(CliCommandDict, FuriString*, FURI_STRING_OPLIST, CliRegistryCommand, M_POD_OPLIST);
// -V:BPTREE_DEF2:1103 #define M_OPL_CliCommandDict_t() DICT_OPLIST(CliCommandDict, FURI_STRING_OPLIST, M_POD_OPLIST)
// -V:BPTREE_DEF2:524
BPTREE_DEF2(
CliCommandTree,
CLI_COMMANDS_TREE_RANK,
FuriString*,
FURI_STRING_OPLIST,
CliRegistryCommand,
M_POD_OPLIST);
#define M_OPL_CliCommandTree_t() BPTREE_OPLIST2(CliCommandTree, FURI_STRING_OPLIST, M_POD_OPLIST)
bool cli_registry_get_command( bool cli_registry_get_command(
CliRegistry* registry, CliRegistry* registry,
@@ -48,7 +38,7 @@ void cli_registry_unlock(CliRegistry* registry);
/** /**
* @warning Surround calls to this function with `cli_registry_[un]lock` * @warning Surround calls to this function with `cli_registry_[un]lock`
*/ */
CliCommandTree_t* cli_registry_get_commands(CliRegistry* registry); CliCommandDict_t* cli_registry_get_commands(CliRegistry* registry);
#ifdef __cplusplus #ifdef __cplusplus
} }

View File

@@ -103,15 +103,15 @@ void cli_command_help(PipeSide* pipe, FuriString* args, void* context) {
printf("Available commands:\r\n" ANSI_FG_GREEN); printf("Available commands:\r\n" ANSI_FG_GREEN);
cli_registry_lock(registry); cli_registry_lock(registry);
CliCommandTree_t* commands = cli_registry_get_commands(registry); CliCommandDict_t* commands = cli_registry_get_commands(registry);
size_t commands_count = CliCommandTree_size(*commands); size_t commands_count = CliCommandDict_size(*commands);
CliCommandTree_it_t iterator; CliCommandDict_it_t iterator;
CliCommandTree_it(iterator, *commands); CliCommandDict_it(iterator, *commands);
for(size_t i = 0; i < commands_count; i++) { for(size_t i = 0; i < commands_count; i++) {
const CliCommandTree_itref_t* item = CliCommandTree_cref(iterator); const CliCommandDict_itref_t* item = CliCommandDict_cref(iterator);
printf("%-30s", furi_string_get_cstr(*item->key_ptr)); printf("%-30s", furi_string_get_cstr(item->key));
CliCommandTree_next(iterator); CliCommandDict_next(iterator);
if(i % columns == columns - 1) printf("\r\n"); if(i % columns == columns - 1) printf("\r\n");
} }
@@ -474,5 +474,6 @@ void cli_shell_join(CliShell* shell) {
void cli_shell_set_prompt(CliShell* shell, const char* prompt) { void cli_shell_set_prompt(CliShell* shell, const char* prompt) {
furi_check(shell); furi_check(shell);
furi_check(furi_thread_get_state(shell->thread) == FuriThreadStateStopped);
shell->prompt = prompt; shell->prompt = prompt;
} }

View File

@@ -111,10 +111,10 @@ void cli_shell_completions_fill_variants(CliShellCompletions* completions) {
if(segment.type == CliShellCompletionSegmentTypeCommand) { if(segment.type == CliShellCompletionSegmentTypeCommand) {
CliRegistry* registry = completions->registry; CliRegistry* registry = completions->registry;
cli_registry_lock(registry); cli_registry_lock(registry);
CliCommandTree_t* commands = cli_registry_get_commands(registry); CliCommandDict_t* commands = cli_registry_get_commands(registry);
for for
M_EACH(registered_command, *commands, CliCommandTree_t) { M_EACH(registered_command, *commands, CliCommandDict_t) {
FuriString* command_name = *registered_command->key_ptr; FuriString* command_name = registered_command->key;
if(furi_string_start_with(command_name, input)) { if(furi_string_start_with(command_name, input)) {
CommandCompletions_push_back(completions->variants, command_name); CommandCompletions_push_back(completions->variants, command_name);
} }

View File

@@ -32,16 +32,27 @@ def main():
bytes_to_send = args.length bytes_to_send = args.length
block_size = 1024 block_size = 1024
success = True
while bytes_to_send: while bytes_to_send:
actual_size = min(block_size, bytes_to_send) actual_size = min(block_size, bytes_to_send)
# can't use 0x03 because that's ASCII ETX, or Ctrl+C # can't use 0x03 because that's ASCII ETX, or Ctrl+C
block = bytes([randint(4, 255) for _ in range(actual_size)]) # block = bytes([randint(4, 255) for _ in range(actual_size)])
block = bytes([4 + (i // 64) for i in range(actual_size)])
port.write(block) port.write(block)
return_block = port.read(actual_size) return_block = port.read(actual_size)
if return_block != block: if return_block != block:
logger.error("Incorrect block received") with open("block.bin", "wb") as f:
f.write(block)
with open("return_block.bin", "wb") as f:
f.write(return_block)
logger.error(
"Incorrect block received. Saved to `block.bin' and `return_block.bin'."
)
logger.error(f"{bytes_to_send} bytes left. Aborting.")
success = False
break break
bytes_to_send -= actual_size bytes_to_send -= actual_size
@@ -49,7 +60,8 @@ def main():
end_time = time() end_time = time()
delta = end_time - start_time delta = end_time - start_time
speed = args.length / delta speed = args.length / delta
print(f"Speed: {speed/1024:.2f} KiB/s") if success:
print(f"Speed: {speed/1024:.2f} KiB/s")
port.write(b"\x03") # Ctrl+C port.write(b"\x03") # Ctrl+C
port.close() port.close()