flipper-zero-tutorials/subghz/apps/genie-recorder/genie_file.c

530 lines
16 KiB
C
Raw Normal View History

2024-02-04 19:13:58 +00:00
#include "genie_file.h"
2023-09-28 19:56:39 +00:00
#include <storage/storage.h>
#include <flipper_format.h>
#define GENIE_APPS_DATA_FOLDER EXT_PATH("apps_data")
#define GENIE_SAVE_FOLDER \
GENIE_APPS_DATA_FOLDER "/" \
"genie"
#define GENIE_SAVE_NAME "keys"
#define GENIE_SAVE_EXTENSION ".txt"
// Should match application.fam, fap_version.
#define GENIE_MAJOR_VERSION 2
#define GENIE_MINOR_VERSION 5
#define GENIE_MAGIC_CODE 0x472A
2023-09-28 19:56:39 +00:00
#ifdef TAG
#undef TAG
#endif
#define TAG "GenieFile"
2024-02-04 19:13:58 +00:00
struct GenieFile {
uint32_t key_hi;
uint32_t key_lo; // sn
uint16_t last_sent;
uint16_t rec_count;
};
2023-09-28 19:56:39 +00:00
static void ensure_dir_exists(Storage* storage, char* dir) {
if(!storage_dir_exists(storage, dir)) {
FURI_LOG_I(TAG, "Creating directory: %s", dir);
storage_simply_mkdir(storage, dir);
} else {
FURI_LOG_D(TAG, "Directory exists: %s", dir);
}
}
static void ensure_save_folder_exists(Storage* storage) {
ensure_dir_exists(storage, GENIE_APPS_DATA_FOLDER);
ensure_dir_exists(storage, GENIE_SAVE_FOLDER);
}
void genie_file_init() {
Storage* storage = furi_record_open(RECORD_STORAGE);
2023-10-11 21:27:41 +00:00
ensure_save_folder_exists(storage);
furi_record_close(RECORD_STORAGE);
}
static uint16_t storage_file_read16(File* file) {
uint16_t read = 0;
char buffer[2] = {0};
storage_file_read(file, buffer, 2);
read |= (buffer[0] << 8);
read |= buffer[1];
return read;
}
static bool storage_file_write16(File* file, uint16_t data) {
char buffer[2] = {0};
buffer[0] = (data >> 8) & 0xFF;
buffer[1] = data & 0xFF;
return storage_file_write(file, buffer, 2) == 2;
}
static uint32_t storage_file_read32(File* file) {
uint32_t read = 0;
char buffer[4] = {0};
storage_file_read(file, buffer, 4);
read = (buffer[0] << 24);
read |= (buffer[1] << 16);
read |= (buffer[2] << 8);
read |= buffer[3];
return read;
}
static bool storage_file_write32(File* file, uint32_t data) {
char buffer[4] = {0};
buffer[0] = (data >> 24) & 0xFF;
buffer[1] = (data >> 16) & 0xFF;
buffer[2] = (data >> 8) & 0xFF;
buffer[3] = data & 0xFF;
return storage_file_write(file, buffer, 4) == 4;
}
enum {
GENIE_MAGIC = 0, // 2 bytes
GENIE_VERSION = 2, // 2 bytes
GENIE_SN = 4, // 4 bytes
GENIE_LAST_SENT = 8, // 2 bytes
GENIE_REC_COUNT = 10, // 2 bytes
GENIE_RESERVED = 12, // 4 bytes
GENIE_DATA = 16, // 64K bytes
2024-02-04 19:13:58 +00:00
} genie_file_layout;
static void genie_create_file(Storage* storage, char* name, uint32_t low) {
File* file = storage_file_alloc(storage);
if(!file) {
FURI_LOG_E(TAG, "Failed to alloc file");
return;
}
if(storage_file_open(file, name, FSAM_READ_WRITE, FSOM_CREATE_ALWAYS)) {
storage_file_write16(file, GENIE_MAGIC_CODE); // Magic value for Genie files.
storage_file_write16(file,
GENIE_MAJOR_VERSION << 8 | GENIE_MINOR_VERSION); // Version info
storage_file_write32(file, low); // Btn+SN
storage_file_write16(file, 0); // Last send (where to start looking for keys)
storage_file_write16(file, 0x0000); // Last recorded value (how many values we recorded)
storage_file_write32(file, 0x00000000); // Reserved data
char buffer[256] = {0}; // blocks of 0s.
for(int i = 0; i < 1024; i++) {
storage_file_write(file, buffer, 256);
}
}
storage_file_close(file);
storage_file_free(file);
}
static uint32_t hex_to_i32(const char* data) {
uint32_t value = 0;
for(int i = 0; i < 8; i++) {
value = value << 4;
value |= (data[i] >= 'a') ? 10 + data[i] - 'a' :
(data[i] >= 'A') ? 10 + data[i] - 'A' :
data[i] - '0';
}
return value;
}
2024-02-04 19:13:58 +00:00
GenieFile* genie_file_load(const char* path) {
GenieFile* genie_file = malloc(sizeof(GenieFile));
memset(genie_file, 0, sizeof(GenieFile));
bool load_success = false;
Storage* storage = furi_record_open(RECORD_STORAGE);
File* file = NULL;
do {
if(!storage) {
FURI_LOG_E(TAG, "Failed to access storage.");
break;
}
file = storage_file_alloc(storage);
if(!file) {
FURI_LOG_E(TAG, "Failed to allocate file.");
break;
}
if(!storage_file_exists(storage, path)) {
FURI_LOG_D(TAG, "File does not exist: %s", path);
break;
}
if(storage_file_open(file, path, FSAM_READ_WRITE, FSOM_OPEN_EXISTING)) {
if(!storage_file_seek(file, GENIE_VERSION, true)) {
FURI_LOG_E(TAG, "Failed to seek to GENIE_VERSION @ %d", GENIE_VERSION);
break;
}
uint16_t version = storage_file_read16(file);
if((version >> 8) > GENIE_MAJOR_VERSION) {
FURI_LOG_E(TAG, "Unsupported version: %04X", version);
break;
}
if(!storage_file_seek(file, GENIE_SN, true)) {
FURI_LOG_E(TAG, "Failed to seek to GENIE_SN @ %d", GENIE_SN);
break;
}
genie_file->key_lo = storage_file_read32(file);
if(!storage_file_seek(file, GENIE_LAST_SENT, true)) {
FURI_LOG_E(TAG, "Failed to seek to GENIE_LAST_SENT @ %d", GENIE_LAST_SENT);
break;
}
genie_file->last_sent = storage_file_read16(file);
if(!storage_file_seek(file, GENIE_REC_COUNT, true)) {
FURI_LOG_E(TAG, "Failed to seek to GENIE_REC_COUNT @ %d", GENIE_REC_COUNT);
break;
}
genie_file->rec_count = storage_file_read16(file);
if(!storage_file_seek(file, GENIE_DATA + (genie_file->last_sent * 4), true)) {
FURI_LOG_E(
TAG,
"Failed to seek to GENIE_DATA+last_sent*4 @ %d",
GENIE_DATA + (genie_file->last_sent * 4));
break;
}
genie_file->key_hi = storage_file_read32(file);
load_success = true;
}
} while(false);
if(file) {
storage_file_close(file);
storage_file_free(file);
}
furi_record_close(RECORD_STORAGE);
if(!load_success) {
free(genie_file);
genie_file = NULL;
}
return genie_file;
}
void genie_file_free(GenieFile* file) {
if(file) {
free(file);
}
}
uint32_t genie_file_get_key_hi(GenieFile* file) {
return file ? file->key_hi : 0;
}
uint32_t genie_file_get_key_lo(GenieFile* file) {
return file ? file->key_lo : 0;
}
uint16_t genie_file_get_last_sent(GenieFile* file) {
return file ? file->last_sent : 0;
}
void genie_file_set_last_sent(const char* genie_path, uint16_t last_sent) {
if(genie_path) {
Storage* storage = furi_record_open(RECORD_STORAGE);
File* file = NULL;
do {
if(!storage) {
FURI_LOG_E(TAG, "Failed to access storage.");
break;
}
file = storage_file_alloc(storage);
if(!file) {
FURI_LOG_E(TAG, "Failed to allocate file.");
break;
}
if(!storage_file_exists(storage, genie_path)) {
FURI_LOG_D(TAG, "File not found: %s", genie_path);
break;
}
if(storage_file_open(file, genie_path, FSAM_READ_WRITE, FSOM_OPEN_EXISTING)) {
if(!storage_file_seek(file, GENIE_VERSION, true)) {
FURI_LOG_E(TAG, "Failed to seek to GENIE_VERSION @ %d", GENIE_VERSION);
break;
}
uint16_t version = storage_file_read16(file);
if((version >> 8) > GENIE_MAJOR_VERSION) {
FURI_LOG_E(TAG, "Unsupported version: %04X", version);
break;
}
if(!storage_file_seek(file, GENIE_LAST_SENT, true)) {
FURI_LOG_E(TAG, "Failed to seek to GENIE_LAST_SENT @ %d", GENIE_LAST_SENT);
break;
}
if(!storage_file_write16(file, last_sent)) {
FURI_LOG_E(TAG, "Failed to set last sent count to %d.", last_sent);
break;
}
} else {
FURI_LOG_E(TAG, "Failed to open file");
break;
}
} while(false);
if(file) {
storage_file_close(file);
storage_file_free(file);
}
furi_record_close(RECORD_STORAGE);
}
}
uint16_t genie_file_get_rec_count(GenieFile* file) {
return file ? file->rec_count : 0;
}
uint16_t genie_rec_count_bin(uint32_t key_low) {
uint16_t count = 0;
char buffer[128] = {0};
snprintf(buffer, 128, "%s/%08lX.gne", GENIE_SAVE_FOLDER, key_low);
GenieFile* file = genie_file_load(buffer);
if(file) {
count = file->rec_count;
genie_file_free(file);
}
return count;
}
uint32_t genie_save_bin(const char* key) {
uint32_t result = 0;
uint32_t key_high = 0;
uint32_t key_low = 0;
key_high = hex_to_i32(key);
key_low = hex_to_i32(key + 8);
FURI_LOG_D(TAG, "Saving binary key. key: %s high: %08lX low: %08lX", key, key_high, key_low);
Storage* storage = furi_record_open(RECORD_STORAGE);
char buffer[128] = {0};
snprintf(buffer, 128, "%s/%08lX.gne", GENIE_SAVE_FOLDER, key_low);
File* file = NULL;
do {
if(!storage) {
FURI_LOG_E(TAG, "Failed to access storage.");
break;
}
file = storage_file_alloc(storage);
if(!file) {
FURI_LOG_E(TAG, "Failed to allocate file.");
break;
}
if(!storage_file_exists(storage, buffer)) {
FURI_LOG_D(TAG, "Creating file: %s", buffer);
genie_create_file(storage, buffer, key_low);
}
if(storage_file_open(file, buffer, FSAM_READ_WRITE, FSOM_OPEN_EXISTING)) {
if(!storage_file_seek(file, GENIE_VERSION, true)) {
FURI_LOG_E(TAG, "Failed to seek to GENIE_VERSION @ %d", GENIE_VERSION);
break;
}
uint16_t version = storage_file_read16(file);
if((version >> 8) > GENIE_MAJOR_VERSION) {
FURI_LOG_E(TAG, "Unsupported version: %04X", version);
break;
}
if(!storage_file_seek(file, GENIE_SN, true)) {
FURI_LOG_E(TAG, "Failed to seek to GENIE_SN @ %d", GENIE_SN);
break;
}
uint32_t low = storage_file_read32(file);
if(low != key_low) {
FURI_LOG_E(TAG, "Key mismatch. Expected: %08lX, got: %08lX", key_low, low);
break;
}
if(!storage_file_seek(file, GENIE_REC_COUNT, true)) {
FURI_LOG_E(TAG, "Failed to seek to GENIE_REC_COUNT @ %d", GENIE_REC_COUNT);
break;
}
uint16_t count = storage_file_read16(file);
if(!storage_file_seek(file, GENIE_DATA + (count * 4), true)) {
FURI_LOG_E(
TAG, "Failed to seek to GENIE_DATA+count*4 @ %d", GENIE_DATA + (count * 4));
break;
}
uint32_t existing = storage_file_read32(file);
if(existing != 0 && existing != key_high) {
if(count != 0xFFFF) {
FURI_LOG_E(
TAG,
"Key mismatch at %04X. Old: %08lX, New: %08lX",
count,
existing,
key_high);
result = count;
} else {
result = 0x10000;
}
break;
}
if(!storage_file_seek(file, GENIE_DATA + (count * 4), true)) {
FURI_LOG_E(
TAG, "Failed to seek to GENIE_DATA+count*4 @ %d", GENIE_DATA + (count * 4));
break;
}
if(!storage_file_write32(file, key_high)) {
FURI_LOG_E(TAG, "Failed to write to key_high to file: %08lX", key_high);
break;
}
if(!storage_file_seek(file, GENIE_REC_COUNT, true)) {
FURI_LOG_E(TAG, "Failed to seek to GENIE_REC_COUNT @ %d", GENIE_REC_COUNT);
break;
}
if(count != 0xFFFF) {
count++;
result = count;
if(!storage_file_write16(file, count)) {
FURI_LOG_E(TAG, "Failed to write count to file: %04X", count);
break;
}
} else {
result = 0x10000;
}
} else {
FURI_LOG_E(TAG, "Failed to open file");
break;
}
} while(false);
if(file) {
storage_file_close(file);
storage_file_free(file);
}
furi_record_close(RECORD_STORAGE);
return result;
}
bool genie_save(uint32_t count, FuriString* key) {
bool success = false;
2023-10-11 21:27:41 +00:00
Storage* storage = furi_record_open(RECORD_STORAGE);
2023-09-28 19:56:39 +00:00
char buffer[8 + 1 + 16 + 2 + 1] = {0};
snprintf(buffer, 8 + 1 + 16 + 2 + 1, "%08lX,%s\r\n", count, furi_string_get_cstr(key));
2023-09-28 19:56:39 +00:00
2023-10-11 21:27:41 +00:00
File* file = NULL;
do {
2023-10-11 21:27:41 +00:00
if(!storage) {
FURI_LOG_E(TAG, "Failed to access storage");
break;
}
2023-09-28 19:56:39 +00:00
2023-10-11 21:27:41 +00:00
file = storage_file_alloc(storage);
if(file && storage_file_open(
file,
(GENIE_SAVE_FOLDER "/" GENIE_SAVE_NAME GENIE_SAVE_EXTENSION),
FSAM_WRITE,
FSOM_OPEN_APPEND)) {
if(!storage_file_write(file, buffer, COUNT_OF(buffer) - 1)) {
2023-09-28 19:56:39 +00:00
FURI_LOG_E(TAG, "Failed to write to file");
break;
}
success = true;
2023-10-11 21:27:41 +00:00
} else {
FURI_LOG_E(TAG, "Failed to open file");
break;
2023-09-28 19:56:39 +00:00
}
} while(false);
2023-10-11 21:27:41 +00:00
if(file) {
storage_file_close(file);
storage_file_free(file);
}
2023-09-28 19:56:39 +00:00
2023-10-11 21:27:41 +00:00
furi_record_close(RECORD_STORAGE);
2023-09-28 19:56:39 +00:00
return success;
}
uint32_t genie_load() {
uint32_t count = 0;
2024-02-04 19:13:58 +00:00
uint32_t key_hi = 0;
uint32_t key_lo = 0;
2023-09-28 19:56:39 +00:00
Storage* storage = furi_record_open(RECORD_STORAGE);
FuriString* buffer = furi_string_alloc();
File* file = NULL;
do {
if(!storage) {
FURI_LOG_E(TAG, "Failed to access storage");
break;
}
ensure_save_folder_exists(storage);
file = storage_file_alloc(storage);
if(storage_file_open(
2023-10-11 21:27:41 +00:00
file,
(GENIE_SAVE_FOLDER "/" GENIE_SAVE_NAME GENIE_SAVE_EXTENSION),
FSAM_READ,
FSOM_OPEN_EXISTING)) {
FURI_LOG_D(TAG, "Reading file.");
2023-09-28 19:56:39 +00:00
char data[8 + 1 + 16 + 2 + 1] = {0};
2024-02-04 19:13:58 +00:00
int64_t offset = ((int64_t)storage_file_size(file)) - 120;
if(offset > 0) {
storage_file_seek(file, offset, true);
do {
storage_file_read(file, data, 1);
} while(data[0] != '\n');
}
2023-09-28 19:56:39 +00:00
while(storage_file_read(file, data, 8 + 1 + 16 + 2)) {
2024-02-04 19:13:58 +00:00
sscanf(data, "%08lX,%08lX%08lX", &count, &key_hi, &key_lo);
FURI_LOG_D(
TAG, "Read: %s, count: %ld, hi: %08lx, lo: %08lx", data, count, key_hi, key_lo);
2023-09-28 19:56:39 +00:00
}
}
} while(false);
if(file) {
storage_file_close(file);
2023-10-11 21:27:41 +00:00
storage_file_free(file);
2023-09-28 19:56:39 +00:00
}
furi_string_free(buffer);
furi_record_close(RECORD_STORAGE);
2024-02-04 19:13:58 +00:00
if(key_lo) {
uint16_t count_bin = genie_rec_count_bin(key_lo);
if(count_bin > 0) {
count = count_bin;
}
}
2023-09-28 19:56:39 +00:00
return count;
}