mirror of
https://github.com/UberGuidoZ/Flipper.git
synced 2024-12-23 06:50:12 +00:00
109 lines
3.6 KiB
C
109 lines
3.6 KiB
C
#include "cli_control.h"
|
|
|
|
#include <cli/cli.h>
|
|
#include <cli/cli_i.h>
|
|
#include <cli/cli_vcp.h>
|
|
#include "cligui_main_i.h"
|
|
#include <FreeRTOS.h>
|
|
|
|
volatile bool gotCallbackSet = false;
|
|
|
|
FuriStreamBuffer* tx_stream;
|
|
FuriStreamBuffer* rx_stream;
|
|
static FuriThread* volatile cliThread = NULL;
|
|
static void tx_handler_stdout(const char* buffer, size_t size) {
|
|
furi_stream_buffer_send(tx_stream, buffer, size, FuriWaitForever);
|
|
}
|
|
static void tx_handler(const uint8_t* buffer, size_t size) {
|
|
furi_thread_set_stdout_callback(tx_handler_stdout);
|
|
cliThread = furi_thread_get_current();
|
|
furi_stream_buffer_send(tx_stream, buffer, size, FuriWaitForever);
|
|
}
|
|
static size_t real_rx_handler(uint8_t* buffer, size_t size, uint32_t timeout) {
|
|
size_t rx_cnt = 0;
|
|
while(size > 0) {
|
|
size_t batch_size = size;
|
|
if(batch_size > 128) batch_size = 128;
|
|
size_t len = furi_stream_buffer_receive(rx_stream, buffer, batch_size, timeout);
|
|
if(len == 0) break;
|
|
size -= len;
|
|
buffer += len;
|
|
rx_cnt += len;
|
|
}
|
|
return rx_cnt;
|
|
}
|
|
|
|
static CliCommand_internal* getInternalCliCommand(Cli* cli, const char* name) {
|
|
FuriString* target_command = furi_string_alloc();
|
|
furi_string_set_str(target_command, name);
|
|
CliCommand_internal* command =
|
|
CliCommandTree_internal_get(((Cli_internal*)cli)->commands, target_command);
|
|
furi_string_free(target_command);
|
|
return command;
|
|
}
|
|
|
|
static void session_init(void) {
|
|
}
|
|
static void session_deinit(void) {
|
|
}
|
|
static bool session_connected(void) {
|
|
return true;
|
|
}
|
|
static CliSession session;
|
|
void latch_tx_handler() {
|
|
Cli* global_cli = furi_record_open(RECORD_CLI);
|
|
|
|
CliCommand_internal* help_command = getInternalCliCommand(global_cli, "help");
|
|
cliThread = help_command->context;
|
|
|
|
furi_thread_set_stdout_callback(tx_handler_stdout);
|
|
if(cliThread != NULL) {
|
|
((FuriThread_internal*)cliThread)->output.write_callback = &tx_handler_stdout;
|
|
}
|
|
|
|
rx_stream = furi_stream_buffer_alloc(128, 1);
|
|
tx_stream = furi_stream_buffer_alloc(128, 1);
|
|
|
|
session.tx = &tx_handler;
|
|
session.rx = &real_rx_handler;
|
|
session.tx_stdout = &tx_handler_stdout;
|
|
session.init = &session_init;
|
|
session.deinit = &session_deinit;
|
|
session.is_connected = &session_connected;
|
|
cli_session_close(global_cli);
|
|
cli_session_open(global_cli, &session);
|
|
// Unlock loader-lock
|
|
Loader* loader = furi_record_open(RECORD_LOADER);
|
|
Loader_internal* loader_i = (Loader_internal*)loader;
|
|
loader_i->lock_count = 0;
|
|
furi_record_close(RECORD_CLI);
|
|
furi_record_close(RECORD_LOADER);
|
|
}
|
|
void unlatch_tx_handler(bool persist) {
|
|
Cli* global_cli = furi_record_open(RECORD_CLI);
|
|
// Stash cliThread if not null
|
|
if(cliThread != NULL) {
|
|
CliCommand_internal* help_command = getInternalCliCommand(global_cli, "help");
|
|
help_command->context = cliThread;
|
|
}
|
|
// Switch to new session
|
|
if(persist) {
|
|
// Use dummy debug firmware function as is_connected
|
|
cli_vcp.is_connected = &furi_hal_version_do_i_belong_here;
|
|
} else {
|
|
// Send CTRL-C
|
|
char eot = 0x03;
|
|
furi_stream_buffer_send(rx_stream, &eot, 1, FuriWaitForever);
|
|
}
|
|
cli_session_open(global_cli, &cli_vcp);
|
|
furi_record_close(RECORD_CLI);
|
|
// Unblock waiting rx handler
|
|
furi_stream_buffer_send(rx_stream, "_", 1, FuriWaitForever);
|
|
// Reconfigure stdout_callback to cli_vcp
|
|
if(cliThread != NULL) {
|
|
((FuriThread_internal*)cliThread)->output.write_callback = cli_vcp.tx_stdout;
|
|
}
|
|
// At this point, all cli_vcp functions should be back.
|
|
furi_stream_buffer_free(rx_stream);
|
|
furi_stream_buffer_free(tx_stream);
|
|
} |