mirror of
https://github.com/DarkFlippers/unleashed-firmware
synced 2024-11-23 13:03:13 +00:00
427 lines
16 KiB
C
427 lines
16 KiB
C
#include "cli_commands.h"
|
|
#include <furi-hal.h>
|
|
#include <furi-hal-gpio.h>
|
|
#include <rtc.h>
|
|
#include <task-control-block.h>
|
|
#include <time.h>
|
|
#include <notification/notification-messages.h>
|
|
#include <shci.h>
|
|
|
|
#define ENCLAVE_SIGNATURE_KEY_SLOT 1
|
|
#define ENCLAVE_SIGNATURE_SIZE 16
|
|
static const uint8_t enclave_signature_iv[16] =
|
|
{0x32, 0xe6, 0xa7, 0x85, 0x20, 0xae, 0x0b, 0xf0, 0x00, 0xb6, 0x30, 0x9b, 0xd5, 0x42, 0x9e, 0xa6};
|
|
static const uint8_t enclave_signature_input[ENCLAVE_SIGNATURE_SIZE] =
|
|
{0xdc, 0x76, 0x15, 0x1e, 0x69, 0xe8, 0xdc, 0xd3, 0x4a, 0x71, 0x0b, 0x42, 0x71, 0xe0, 0xa9, 0x78};
|
|
static const uint8_t enclave_signature_expected[ENCLAVE_SIGNATURE_SIZE] =
|
|
{0x6b, 0x31, 0xc, 0xac, 0x3f, 0x68, 0x79, 0x76, 0x43, 0xc4, 0xfe, 0xe0, 0x25, 0x53, 0x64, 0xc7};
|
|
|
|
/*
|
|
* Device Info Command
|
|
* This command is intended to be used by humans and machines
|
|
* Keys and values format MUST NOT BE changed
|
|
*/
|
|
void cli_command_device_info(Cli* cli, string_t args, void* context) {
|
|
// Model name
|
|
printf("hardware_model : %s\r\n", furi_hal_version_get_model_name());
|
|
const char* name = furi_hal_version_get_name_ptr();
|
|
if(name) {
|
|
printf("hardware_name : %s\r\n", name);
|
|
}
|
|
|
|
// Unique ID
|
|
printf("hardware_uid : ");
|
|
const uint8_t* uid = furi_hal_version_uid();
|
|
for(size_t i = 0; i < furi_hal_version_uid_size(); i++) {
|
|
printf("%02X", uid[i]);
|
|
}
|
|
printf("\r\n");
|
|
|
|
// Board Revision
|
|
printf("hardware_ver : %d\r\n", furi_hal_version_get_hw_version());
|
|
printf("hardware_target : %d\r\n", furi_hal_version_get_hw_target());
|
|
printf("hardware_body : %d\r\n", furi_hal_version_get_hw_body());
|
|
printf("hardware_connect : %d\r\n", furi_hal_version_get_hw_connect());
|
|
printf("hardware_timestamp : %lu\r\n", furi_hal_version_get_hw_timestamp());
|
|
|
|
// Color and Region
|
|
printf("hardware_color : %d\r\n", furi_hal_version_get_hw_color());
|
|
printf("hardware_region : %d\r\n", furi_hal_version_get_hw_region());
|
|
|
|
// Bootloader Version
|
|
const Version* boot_version = furi_hal_version_get_boot_version();
|
|
if(boot_version) {
|
|
printf("boot_version : %s\r\n", version_get_version(boot_version));
|
|
printf("boot_commit : %s\r\n", version_get_githash(boot_version));
|
|
printf("boot_branch : %s\r\n", version_get_gitbranch(boot_version));
|
|
printf("boot_build_date : %s\r\n", version_get_builddate(boot_version));
|
|
}
|
|
|
|
// Firmware version
|
|
const Version* firmware_version = furi_hal_version_get_firmware_version();
|
|
if(firmware_version) {
|
|
printf("firmware_version : %s\r\n", version_get_version(firmware_version));
|
|
printf("firmware_target : %s\r\n", version_get_target(firmware_version));
|
|
printf("firmware_commit : %s\r\n", version_get_githash(firmware_version));
|
|
printf("firmware_branch : %s\r\n", version_get_gitbranch(firmware_version));
|
|
printf("firmware_build_date : %s\r\n", version_get_builddate(firmware_version));
|
|
}
|
|
|
|
WirelessFwInfo_t pWirelessInfo;
|
|
if(furi_hal_bt_is_alive() && SHCI_GetWirelessFwInfo(&pWirelessInfo) == SHCI_Success) {
|
|
printf("radio_alive : true\r\n");
|
|
// FUS Info
|
|
printf("radio_fus_major : %d\r\n", pWirelessInfo.FusVersionMajor);
|
|
printf("radio_fus_minor : %d\r\n", pWirelessInfo.FusVersionMinor);
|
|
printf("radio_fus_sub : %d\r\n", pWirelessInfo.FusVersionSub);
|
|
printf("radio_fus_sram2b : %dK\r\n", pWirelessInfo.FusMemorySizeSram2B);
|
|
printf("radio_fus_sram2a : %dK\r\n", pWirelessInfo.FusMemorySizeSram2A);
|
|
printf("radio_fus_flash : %dK\r\n", pWirelessInfo.FusMemorySizeFlash * 4);
|
|
// Stack Info
|
|
printf("radio_stack_type : %d\r\n", pWirelessInfo.StackType);
|
|
printf("radio_stack_major : %d\r\n", pWirelessInfo.VersionMajor);
|
|
printf("radio_stack_minor : %d\r\n", pWirelessInfo.VersionMinor);
|
|
printf("radio_stack_sub : %d\r\n", pWirelessInfo.VersionSub);
|
|
printf("radio_stack_branch : %d\r\n", pWirelessInfo.VersionBranch);
|
|
printf("radio_stack_release : %d\r\n", pWirelessInfo.VersionReleaseType);
|
|
printf("radio_stack_sram2b : %dK\r\n", pWirelessInfo.MemorySizeSram2B);
|
|
printf("radio_stack_sram2a : %dK\r\n", pWirelessInfo.MemorySizeSram2A);
|
|
printf("radio_stack_sram1 : %dK\r\n", pWirelessInfo.MemorySizeSram1);
|
|
printf("radio_stack_flash : %dK\r\n", pWirelessInfo.MemorySizeFlash * 4);
|
|
// Mac address
|
|
printf("radio_ble_mac : ");
|
|
const uint8_t* ble_mac = furi_hal_version_get_ble_mac();
|
|
for(size_t i = 0; i < 6; i++) {
|
|
printf("%02X", ble_mac[i]);
|
|
}
|
|
printf("\r\n");
|
|
|
|
// Signature verification
|
|
uint8_t buffer[ENCLAVE_SIGNATURE_SIZE];
|
|
bool enclave_valid = false;
|
|
if(furi_hal_crypto_store_load_key(ENCLAVE_SIGNATURE_KEY_SLOT, enclave_signature_iv)) {
|
|
if(furi_hal_crypto_encrypt(enclave_signature_input, buffer, ENCLAVE_SIGNATURE_SIZE)) {
|
|
enclave_valid =
|
|
memcmp(buffer, enclave_signature_expected, ENCLAVE_SIGNATURE_SIZE) == 0;
|
|
}
|
|
furi_hal_crypto_store_unload_key(ENCLAVE_SIGNATURE_KEY_SLOT);
|
|
}
|
|
printf("enclave_valid : %s\r\n", enclave_valid ? "true" : "false");
|
|
} else {
|
|
printf("radio_alive : false\r\n");
|
|
}
|
|
}
|
|
|
|
void cli_command_help(Cli* cli, string_t args, void* context) {
|
|
(void)args;
|
|
printf("Commands we have:");
|
|
|
|
// Command count
|
|
const size_t commands_count = CliCommandTree_size(cli->commands);
|
|
const size_t commands_count_mid = commands_count / 2 + commands_count % 2;
|
|
|
|
// Use 2 iterators from start and middle to show 2 columns
|
|
CliCommandTree_it_t it_left;
|
|
CliCommandTree_it(it_left, cli->commands);
|
|
CliCommandTree_it_t it_right;
|
|
CliCommandTree_it(it_right, cli->commands);
|
|
for(size_t i = 0; i < commands_count_mid; i++) CliCommandTree_next(it_right);
|
|
|
|
// Iterate throw tree
|
|
for(size_t i = 0; i < commands_count_mid; i++) {
|
|
printf("\r\n");
|
|
// Left Column
|
|
if(!CliCommandTree_end_p(it_left)) {
|
|
printf("%-30s", string_get_cstr(*CliCommandTree_ref(it_left)->key_ptr));
|
|
CliCommandTree_next(it_left);
|
|
}
|
|
// Right Column
|
|
if(!CliCommandTree_end_p(it_right)) {
|
|
printf(string_get_cstr(*CliCommandTree_ref(it_right)->key_ptr));
|
|
CliCommandTree_next(it_right);
|
|
}
|
|
};
|
|
|
|
if(string_size(args) > 0) {
|
|
cli_nl();
|
|
printf("Also I have no clue what '");
|
|
printf(string_get_cstr(args));
|
|
printf("' is.");
|
|
}
|
|
}
|
|
|
|
void cli_command_date(Cli* cli, string_t args, void* context) {
|
|
RTC_TimeTypeDef time;
|
|
RTC_DateTypeDef date;
|
|
|
|
if(string_size(args) > 0) {
|
|
uint16_t Hours, Minutes, Seconds, Month, Date, Year, WeekDay;
|
|
int ret = sscanf(
|
|
string_get_cstr(args),
|
|
"%hu:%hu:%hu %hu-%hu-%hu %hu",
|
|
&Hours,
|
|
&Minutes,
|
|
&Seconds,
|
|
&Month,
|
|
&Date,
|
|
&Year,
|
|
&WeekDay);
|
|
if(ret == 7) {
|
|
time.Hours = Hours;
|
|
time.Minutes = Minutes;
|
|
time.Seconds = Seconds;
|
|
time.DayLightSaving = RTC_DAYLIGHTSAVING_NONE;
|
|
time.StoreOperation = RTC_STOREOPERATION_RESET;
|
|
date.WeekDay = WeekDay;
|
|
date.Month = Month;
|
|
date.Date = Date;
|
|
date.Year = Year - 2000;
|
|
HAL_RTC_SetTime(&hrtc, &time, RTC_FORMAT_BIN);
|
|
HAL_RTC_SetDate(&hrtc, &date, RTC_FORMAT_BIN);
|
|
|
|
// Verification
|
|
HAL_RTC_GetTime(&hrtc, &time, RTC_FORMAT_BIN);
|
|
HAL_RTC_GetDate(&hrtc, &date, RTC_FORMAT_BIN);
|
|
printf(
|
|
"New time is: %.2d:%.2d:%.2d %.2d-%.2d-%.2d %d",
|
|
time.Hours,
|
|
time.Minutes,
|
|
time.Seconds,
|
|
date.Month,
|
|
date.Date,
|
|
2000 + date.Year,
|
|
date.WeekDay);
|
|
} else {
|
|
printf(
|
|
"Invalid time format, use `hh:mm:ss MM-DD-YYYY WD`. sscanf %d %s",
|
|
ret,
|
|
string_get_cstr(args));
|
|
return;
|
|
}
|
|
} else {
|
|
// TODO add get_datetime to core, not use HAL here
|
|
// READ ORDER MATTERS! Time then date.
|
|
HAL_RTC_GetTime(&hrtc, &time, RTC_FORMAT_BIN);
|
|
HAL_RTC_GetDate(&hrtc, &date, RTC_FORMAT_BIN);
|
|
printf(
|
|
"%.2d:%.2d:%.2d %.2d-%.2d-%.2d %d",
|
|
time.Hours,
|
|
time.Minutes,
|
|
time.Seconds,
|
|
date.Month,
|
|
date.Date,
|
|
2000 + date.Year,
|
|
date.WeekDay);
|
|
}
|
|
}
|
|
|
|
void cli_command_log(Cli* cli, string_t args, void* context) {
|
|
furi_stdglue_set_global_stdout_callback(cli_stdout_callback);
|
|
printf("Press any key to stop...\r\n");
|
|
cli_getc(cli);
|
|
furi_stdglue_set_global_stdout_callback(NULL);
|
|
}
|
|
|
|
void cli_command_vibro(Cli* cli, string_t args, void* context) {
|
|
if(!string_cmp(args, "0")) {
|
|
NotificationApp* notification = furi_record_open("notification");
|
|
notification_message_block(notification, &sequence_reset_vibro);
|
|
furi_record_close("notification");
|
|
} else if(!string_cmp(args, "1")) {
|
|
NotificationApp* notification = furi_record_open("notification");
|
|
notification_message_block(notification, &sequence_set_vibro_on);
|
|
furi_record_close("notification");
|
|
} else {
|
|
cli_print_usage("vibro", "<1|0>", string_get_cstr(args));
|
|
}
|
|
}
|
|
|
|
void cli_command_led(Cli* cli, string_t args, void* context) {
|
|
// Get first word as light name
|
|
NotificationMessage notification_led_message;
|
|
string_t light_name;
|
|
string_init(light_name);
|
|
size_t ws = string_search_char(args, ' ');
|
|
if(ws == STRING_FAILURE) {
|
|
cli_print_usage("led", "<r|g|b|bl> <0-255>", string_get_cstr(args));
|
|
string_clear(light_name);
|
|
return;
|
|
} else {
|
|
string_set_n(light_name, args, 0, ws);
|
|
string_right(args, ws);
|
|
string_strim(args);
|
|
}
|
|
// Check light name
|
|
if(!string_cmp(light_name, "r")) {
|
|
notification_led_message.type = NotificationMessageTypeLedRed;
|
|
} else if(!string_cmp(light_name, "g")) {
|
|
notification_led_message.type = NotificationMessageTypeLedGreen;
|
|
} else if(!string_cmp(light_name, "b")) {
|
|
notification_led_message.type = NotificationMessageTypeLedBlue;
|
|
} else if(!string_cmp(light_name, "bl")) {
|
|
notification_led_message.type = NotificationMessageTypeLedDisplay;
|
|
} else {
|
|
cli_print_usage("led", "<r|g|b|bl> <0-255>", string_get_cstr(args));
|
|
string_clear(light_name);
|
|
return;
|
|
}
|
|
string_clear(light_name);
|
|
// Read light value from the rest of the string
|
|
char* end_ptr;
|
|
uint32_t value = strtoul(string_get_cstr(args), &end_ptr, 0);
|
|
if(!(value < 256 && *end_ptr == '\0')) {
|
|
cli_print_usage("led", "<r|g|b|bl> <0-255>", string_get_cstr(args));
|
|
return;
|
|
}
|
|
|
|
// Set led value
|
|
notification_led_message.data.led.value = value;
|
|
|
|
// Form notification sequence
|
|
const NotificationSequence notification_sequence = {
|
|
¬ification_led_message,
|
|
NULL,
|
|
};
|
|
|
|
// Send notification
|
|
NotificationApp* notification = furi_record_open("notification");
|
|
notification_internal_message_block(notification, ¬ification_sequence);
|
|
furi_record_close("notification");
|
|
}
|
|
|
|
void cli_command_gpio_set(Cli* cli, string_t args, void* context) {
|
|
char pin_names[][4] = {
|
|
"PC0",
|
|
"PC1",
|
|
"PC3",
|
|
"PB2",
|
|
"PB3",
|
|
"PA4",
|
|
"PA6",
|
|
"PA7",
|
|
#ifdef DEBUG
|
|
"PA0",
|
|
"PB7",
|
|
"PB8",
|
|
"PB9"
|
|
#endif
|
|
};
|
|
GpioPin gpio[] = {
|
|
{.port = GPIOC, .pin = LL_GPIO_PIN_0},
|
|
{.port = GPIOC, .pin = LL_GPIO_PIN_1},
|
|
{.port = GPIOC, .pin = LL_GPIO_PIN_3},
|
|
{.port = GPIOB, .pin = LL_GPIO_PIN_2},
|
|
{.port = GPIOB, .pin = LL_GPIO_PIN_3},
|
|
{.port = GPIOA, .pin = LL_GPIO_PIN_4},
|
|
{.port = GPIOA, .pin = LL_GPIO_PIN_6},
|
|
{.port = GPIOA, .pin = LL_GPIO_PIN_7},
|
|
#ifdef DEBUG
|
|
{.port = GPIOA, .pin = LL_GPIO_PIN_0}, // IR_RX (PA0)
|
|
{.port = GPIOB, .pin = LL_GPIO_PIN_7}, // UART RX (PB7)
|
|
{.port = GPIOB, .pin = LL_GPIO_PIN_8}, // SPEAKER (PB8)
|
|
{.port = GPIOB, .pin = LL_GPIO_PIN_9}, // IR_TX (PB9)
|
|
#endif
|
|
};
|
|
uint8_t num = 0;
|
|
bool pin_found = false;
|
|
|
|
// Get first word as pin name
|
|
string_t pin_name;
|
|
string_init(pin_name);
|
|
size_t ws = string_search_char(args, ' ');
|
|
if(ws == STRING_FAILURE) {
|
|
cli_print_usage("gpio_set", "<pin_name> <0|1>", string_get_cstr(args));
|
|
string_clear(pin_name);
|
|
return;
|
|
} else {
|
|
string_set_n(pin_name, args, 0, ws);
|
|
string_right(args, ws);
|
|
string_strim(args);
|
|
}
|
|
// Search correct pin name
|
|
for(num = 0; num < sizeof(pin_names) / sizeof(char*); num++) {
|
|
if(!string_cmp(pin_name, pin_names[num])) {
|
|
pin_found = true;
|
|
break;
|
|
}
|
|
}
|
|
if(!pin_found) {
|
|
printf("Wrong pin name. Available pins: ");
|
|
for(uint8_t i = 0; i < sizeof(pin_names) / sizeof(char*); i++) {
|
|
printf("%s ", pin_names[i]);
|
|
}
|
|
string_clear(pin_name);
|
|
return;
|
|
}
|
|
string_clear(pin_name);
|
|
// Read "0" or "1" as second argument to set or reset pin
|
|
if(!string_cmp(args, "0")) {
|
|
LL_GPIO_SetPinMode(gpio[num].port, gpio[num].pin, LL_GPIO_MODE_OUTPUT);
|
|
LL_GPIO_SetPinOutputType(gpio[num].port, gpio[num].pin, LL_GPIO_OUTPUT_PUSHPULL);
|
|
LL_GPIO_ResetOutputPin(gpio[num].port, gpio[num].pin);
|
|
} else if(!string_cmp(args, "1")) {
|
|
#ifdef DEBUG
|
|
if(num == 8) { // PA0
|
|
printf(
|
|
"Setting PA0 pin HIGH with TSOP connected can damage IR receiver. Are you sure you want to continue? (y/n)?\r\n");
|
|
char c = cli_getc(cli);
|
|
if(c != 'y' && c != 'Y') {
|
|
printf("Cancelled.\r\n");
|
|
return;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
LL_GPIO_SetPinMode(gpio[num].port, gpio[num].pin, LL_GPIO_MODE_OUTPUT);
|
|
LL_GPIO_SetPinOutputType(gpio[num].port, gpio[num].pin, LL_GPIO_OUTPUT_PUSHPULL);
|
|
LL_GPIO_SetOutputPin(gpio[num].port, gpio[num].pin);
|
|
} else {
|
|
printf("Wrong 2nd argument. Use \"1\" to set, \"0\" to reset");
|
|
}
|
|
return;
|
|
}
|
|
|
|
void cli_command_ps(Cli* cli, string_t args, void* context) {
|
|
const uint8_t threads_num_max = 32;
|
|
osThreadId_t threads_id[threads_num_max];
|
|
uint8_t thread_num = osThreadEnumerate(threads_id, threads_num_max);
|
|
printf("%d threads in total:\r\n", thread_num);
|
|
printf("%-20s %-14s %-14s %s\r\n", "Name", "Stack start", "Stack alloc", "Stack watermark");
|
|
for(uint8_t i = 0; i < thread_num; i++) {
|
|
TaskControlBlock* tcb = (TaskControlBlock*)threads_id[i];
|
|
printf(
|
|
"%-20s 0x%-12lx %-14ld %ld\r\n",
|
|
osThreadGetName(threads_id[i]),
|
|
(uint32_t)tcb->pxStack,
|
|
(uint32_t)(tcb->pxEndOfStack - tcb->pxStack + 1) * sizeof(uint32_t),
|
|
osThreadGetStackSpace(threads_id[i]) * sizeof(uint32_t));
|
|
}
|
|
}
|
|
|
|
void cli_command_free(Cli* cli, string_t args, void* context) {
|
|
printf("Free heap size: %d\r\n", memmgr_get_free_heap());
|
|
printf("Minimum heap size: %d\r\n", memmgr_get_minimum_free_heap());
|
|
printf("Maximum heap block: %d\r\n", memmgr_heap_get_max_free_block());
|
|
}
|
|
|
|
void cli_command_free_blocks(Cli* cli, string_t args, void* context) {
|
|
memmgr_heap_printf_free_blocks();
|
|
}
|
|
|
|
void cli_commands_init(Cli* cli) {
|
|
cli_add_command(cli, "!", CliCommandFlagParallelSafe, cli_command_device_info, NULL);
|
|
cli_add_command(cli, "device_info", CliCommandFlagParallelSafe, cli_command_device_info, NULL);
|
|
|
|
cli_add_command(cli, "?", CliCommandFlagParallelSafe, cli_command_help, NULL);
|
|
cli_add_command(cli, "help", CliCommandFlagParallelSafe, cli_command_help, NULL);
|
|
|
|
cli_add_command(cli, "date", CliCommandFlagParallelSafe, cli_command_date, NULL);
|
|
cli_add_command(cli, "log", CliCommandFlagParallelSafe, cli_command_log, NULL);
|
|
cli_add_command(cli, "ps", CliCommandFlagParallelSafe, cli_command_ps, NULL);
|
|
cli_add_command(cli, "free", CliCommandFlagParallelSafe, cli_command_free, NULL);
|
|
cli_add_command(cli, "free_blocks", CliCommandFlagParallelSafe, cli_command_free_blocks, NULL);
|
|
|
|
cli_add_command(cli, "vibro", CliCommandFlagDefault, cli_command_vibro, NULL);
|
|
cli_add_command(cli, "led", CliCommandFlagDefault, cli_command_led, NULL);
|
|
cli_add_command(cli, "gpio_set", CliCommandFlagDefault, cli_command_gpio_set, NULL);
|
|
}
|