mirror of
https://github.com/DarkFlippers/unleashed-firmware
synced 2024-11-23 13:03:13 +00:00
4cee550cc6
* bt: disconnect first on profile change * bt keys: rework bt keys * saved struct: add payload size getter to API * bt: rework bt with new key storage API * bt: add keys storage operation to bt API * hid: save bt keys on sd card * bt: add unit tests for key storage * bt: working profile switch * bt: cleanup * bt hid: change keys storage path Co-authored-by: あく <alleteam@gmail.com>
110 lines
3.8 KiB
C
110 lines
3.8 KiB
C
#include <furi.h>
|
|
#include <furi_hal.h>
|
|
#include "../minunit.h"
|
|
|
|
#include <bt/bt_service/bt_keys_storage.h>
|
|
#include <storage/storage.h>
|
|
|
|
#define BT_TEST_KEY_STORAGE_FILE_PATH EXT_PATH("unit_tests/bt_test.keys")
|
|
#define BT_TEST_NVM_RAM_BUFF_SIZE (507 * 4) // The same as in ble NVM storage
|
|
|
|
typedef struct {
|
|
Storage* storage;
|
|
BtKeysStorage* bt_keys_storage;
|
|
uint8_t* nvm_ram_buff_dut;
|
|
uint8_t* nvm_ram_buff_ref;
|
|
} BtTest;
|
|
|
|
BtTest* bt_test = NULL;
|
|
|
|
void bt_test_alloc() {
|
|
bt_test = malloc(sizeof(BtTest));
|
|
bt_test->storage = furi_record_open(RECORD_STORAGE);
|
|
bt_test->nvm_ram_buff_dut = malloc(BT_TEST_NVM_RAM_BUFF_SIZE);
|
|
bt_test->nvm_ram_buff_ref = malloc(BT_TEST_NVM_RAM_BUFF_SIZE);
|
|
bt_test->bt_keys_storage = bt_keys_storage_alloc(BT_TEST_KEY_STORAGE_FILE_PATH);
|
|
bt_keys_storage_set_ram_params(
|
|
bt_test->bt_keys_storage, bt_test->nvm_ram_buff_dut, BT_TEST_NVM_RAM_BUFF_SIZE);
|
|
}
|
|
|
|
void bt_test_free() {
|
|
furi_assert(bt_test);
|
|
free(bt_test->nvm_ram_buff_ref);
|
|
free(bt_test->nvm_ram_buff_dut);
|
|
bt_keys_storage_free(bt_test->bt_keys_storage);
|
|
furi_record_close(RECORD_STORAGE);
|
|
free(bt_test);
|
|
bt_test = NULL;
|
|
}
|
|
|
|
static void bt_test_keys_storage_profile() {
|
|
// Emulate nvm change on initial connection
|
|
const int nvm_change_size_on_connection = 88;
|
|
for(size_t i = 0; i < nvm_change_size_on_connection; i++) {
|
|
bt_test->nvm_ram_buff_dut[i] = rand();
|
|
bt_test->nvm_ram_buff_ref[i] = bt_test->nvm_ram_buff_dut[i];
|
|
}
|
|
// Emulate update storage on initial connect
|
|
mu_assert(
|
|
bt_keys_storage_update(
|
|
bt_test->bt_keys_storage, bt_test->nvm_ram_buff_dut, nvm_change_size_on_connection),
|
|
"Failed to update key storage on initial connect");
|
|
memset(bt_test->nvm_ram_buff_dut, 0, BT_TEST_NVM_RAM_BUFF_SIZE);
|
|
mu_assert(bt_keys_storage_load(bt_test->bt_keys_storage), "Failed to load NVM");
|
|
mu_assert(
|
|
memcmp(
|
|
bt_test->nvm_ram_buff_ref, bt_test->nvm_ram_buff_dut, nvm_change_size_on_connection) ==
|
|
0,
|
|
"Wrong buffer loaded");
|
|
|
|
const int nvm_disconnect_update_offset = 84;
|
|
const int nvm_disconnect_update_size = 324;
|
|
const int nvm_total_size = nvm_change_size_on_connection -
|
|
(nvm_change_size_on_connection - nvm_disconnect_update_offset) +
|
|
nvm_disconnect_update_size;
|
|
// Emulate update storage on initial disconnect
|
|
for(size_t i = nvm_disconnect_update_offset;
|
|
i < nvm_disconnect_update_offset + nvm_disconnect_update_size;
|
|
i++) {
|
|
bt_test->nvm_ram_buff_dut[i] = rand();
|
|
bt_test->nvm_ram_buff_ref[i] = bt_test->nvm_ram_buff_dut[i];
|
|
}
|
|
mu_assert(
|
|
bt_keys_storage_update(
|
|
bt_test->bt_keys_storage,
|
|
&bt_test->nvm_ram_buff_dut[nvm_disconnect_update_offset],
|
|
nvm_disconnect_update_size),
|
|
"Failed to update key storage on initial disconnect");
|
|
memset(bt_test->nvm_ram_buff_dut, 0, BT_TEST_NVM_RAM_BUFF_SIZE);
|
|
mu_assert(bt_keys_storage_load(bt_test->bt_keys_storage), "Failed to load NVM");
|
|
mu_assert(
|
|
memcmp(bt_test->nvm_ram_buff_ref, bt_test->nvm_ram_buff_dut, nvm_total_size) == 0,
|
|
"Wrong buffer loaded");
|
|
}
|
|
|
|
static void bt_test_keys_remove_test_file() {
|
|
mu_assert(
|
|
storage_simply_remove(bt_test->storage, BT_TEST_KEY_STORAGE_FILE_PATH),
|
|
"Can't remove test file");
|
|
}
|
|
|
|
MU_TEST(bt_test_keys_storage_serial_profile) {
|
|
furi_assert(bt_test);
|
|
|
|
bt_test_keys_remove_test_file();
|
|
bt_test_keys_storage_profile();
|
|
bt_test_keys_remove_test_file();
|
|
}
|
|
|
|
MU_TEST_SUITE(test_bt) {
|
|
bt_test_alloc();
|
|
|
|
MU_RUN_TEST(bt_test_keys_storage_serial_profile);
|
|
|
|
bt_test_free();
|
|
}
|
|
|
|
int run_minunit_test_bt() {
|
|
MU_RUN_SUITE(test_bt);
|
|
return MU_EXIT_CODE;
|
|
}
|