unleashed-firmware/lib/nfc/nfc_device.c
gornekich d92b0a82cc
NFC refactoring (#3050)
"A long time ago in a galaxy far, far away...." we started NFC subsystem refactoring.

Starring:

- @gornekich - NFC refactoring project lead, architect, senior developer
- @gsurkov - architect, senior developer
- @RebornedBrain - senior developer

Supporting roles:

- @skotopes, @DrZlo13, @hedger - general architecture advisors, code review
- @Astrrra, @doomwastaken, @Hellitron, @ImagineVagon333 - quality assurance

Special thanks:

@bettse, @pcunning, @nxv, @noproto, @AloneLiberty and everyone else who has been helping us all this time and contributing valuable knowledges, ideas and source code.
2023-10-24 12:08:09 +09:00

356 lines
10 KiB
C

#include "nfc_device_i.h"
#include <storage/storage.h>
#include <flipper_format/flipper_format.h>
#include "nfc_common.h"
#include "protocols/nfc_device_defs.h"
#define NFC_FILE_HEADER "Flipper NFC device"
#define NFC_DEV_TYPE_ERROR "Protocol type mismatch"
#define NFC_DEVICE_UID_KEY "UID"
#define NFC_DEVICE_TYPE_KEY "Device type"
#define NFC_DEVICE_UID_MAX_LEN (10U)
NfcDevice* nfc_device_alloc() {
NfcDevice* instance = malloc(sizeof(NfcDevice));
instance->protocol = NfcProtocolInvalid;
return instance;
}
void nfc_device_free(NfcDevice* instance) {
furi_assert(instance);
nfc_device_clear(instance);
free(instance);
}
void nfc_device_clear(NfcDevice* instance) {
furi_assert(instance);
if(instance->protocol == NfcProtocolInvalid) {
furi_assert(instance->protocol_data == NULL);
} else if(instance->protocol < NfcProtocolNum) {
if(instance->protocol_data) {
nfc_devices[instance->protocol]->free(instance->protocol_data);
instance->protocol_data = NULL;
}
instance->protocol = NfcProtocolInvalid;
}
}
void nfc_device_reset(NfcDevice* instance) {
furi_assert(instance);
furi_assert(instance->protocol < NfcProtocolNum);
if(instance->protocol_data) {
nfc_devices[instance->protocol]->reset(instance->protocol_data);
}
}
NfcProtocol nfc_device_get_protocol(const NfcDevice* instance) {
furi_assert(instance);
return instance->protocol;
}
const NfcDeviceData* nfc_device_get_data(const NfcDevice* instance, NfcProtocol protocol) {
return nfc_device_get_data_ptr(instance, protocol);
}
const char* nfc_device_get_protocol_name(NfcProtocol protocol) {
furi_assert(protocol < NfcProtocolNum);
return nfc_devices[protocol]->protocol_name;
}
const char* nfc_device_get_name(const NfcDevice* instance, NfcDeviceNameType name_type) {
furi_assert(instance);
furi_assert(instance->protocol < NfcProtocolNum);
return nfc_devices[instance->protocol]->get_name(instance->protocol_data, name_type);
}
const uint8_t* nfc_device_get_uid(const NfcDevice* instance, size_t* uid_len) {
furi_assert(instance);
furi_assert(instance->protocol < NfcProtocolNum);
return nfc_devices[instance->protocol]->get_uid(instance->protocol_data, uid_len);
}
bool nfc_device_set_uid(NfcDevice* instance, const uint8_t* uid, size_t uid_len) {
furi_assert(instance);
furi_assert(instance->protocol < NfcProtocolNum);
return nfc_devices[instance->protocol]->set_uid(instance->protocol_data, uid, uid_len);
}
void nfc_device_set_data(
NfcDevice* instance,
NfcProtocol protocol,
const NfcDeviceData* protocol_data) {
furi_assert(instance);
furi_assert(protocol < NfcProtocolNum);
nfc_device_clear(instance);
instance->protocol = protocol;
instance->protocol_data = nfc_devices[protocol]->alloc();
nfc_devices[protocol]->copy(instance->protocol_data, protocol_data);
}
void nfc_device_copy_data(
const NfcDevice* instance,
NfcProtocol protocol,
NfcDeviceData* protocol_data) {
furi_assert(instance);
furi_assert(protocol < NfcProtocolNum);
furi_assert(protocol_data);
if(instance->protocol != protocol) {
furi_crash(NFC_DEV_TYPE_ERROR);
}
nfc_devices[protocol]->copy(protocol_data, instance->protocol_data);
}
bool nfc_device_is_equal_data(
const NfcDevice* instance,
NfcProtocol protocol,
const NfcDeviceData* protocol_data) {
furi_assert(instance);
furi_assert(protocol < NfcProtocolNum);
furi_assert(protocol_data);
return instance->protocol == protocol &&
nfc_devices[protocol]->is_equal(instance->protocol_data, protocol_data);
}
bool nfc_device_is_equal(const NfcDevice* instance, const NfcDevice* other) {
furi_assert(instance);
furi_assert(other);
return nfc_device_is_equal_data(instance, other->protocol, other->protocol_data);
}
void nfc_device_set_loading_callback(
NfcDevice* instance,
NfcLoadingCallback callback,
void* context) {
furi_assert(instance);
furi_assert(callback);
instance->loading_callback = callback;
instance->loading_callback_context = context;
}
bool nfc_device_save(NfcDevice* instance, const char* path) {
furi_assert(instance);
furi_assert(instance->protocol < NfcProtocolNum);
furi_assert(path);
bool saved = false;
Storage* storage = furi_record_open(RECORD_STORAGE);
FlipperFormat* ff = flipper_format_buffered_file_alloc(storage);
FuriString* temp_str = furi_string_alloc();
if(instance->loading_callback) {
instance->loading_callback(instance->loading_callback_context, true);
}
do {
// Open file
if(!flipper_format_buffered_file_open_always(ff, path)) break;
// Write header
if(!flipper_format_write_header_cstr(ff, NFC_FILE_HEADER, NFC_CURRENT_FORMAT_VERSION))
break;
// Write allowed device types
furi_string_printf(temp_str, "%s can be ", NFC_DEVICE_TYPE_KEY);
for(NfcProtocol protocol = 0; protocol < NfcProtocolNum; ++protocol) {
furi_string_cat(temp_str, nfc_devices[protocol]->protocol_name);
if(protocol < NfcProtocolNum - 1) {
furi_string_cat(temp_str, ", ");
}
}
if(!flipper_format_write_comment(ff, temp_str)) break;
// Write device type
if(!flipper_format_write_string_cstr(
ff, NFC_DEVICE_TYPE_KEY, nfc_devices[instance->protocol]->protocol_name))
break;
// Write UID
furi_string_printf(temp_str, "%s is common for all formats", NFC_DEVICE_UID_KEY);
if(!flipper_format_write_comment(ff, temp_str)) break;
size_t uid_len;
const uint8_t* uid = nfc_device_get_uid(instance, &uid_len);
if(!flipper_format_write_hex(ff, NFC_DEVICE_UID_KEY, uid, uid_len)) break;
// Write protocol-dependent data
if(!nfc_devices[instance->protocol]->save(instance->protocol_data, ff)) break;
saved = true;
} while(false);
if(instance->loading_callback) {
instance->loading_callback(instance->loading_callback_context, false);
}
furi_string_free(temp_str);
flipper_format_free(ff);
furi_record_close(RECORD_STORAGE);
return saved;
}
static bool nfc_device_load_uid(
FlipperFormat* ff,
uint8_t* uid,
uint32_t* uid_len,
const uint32_t uid_maxlen) {
bool loaded = false;
do {
uint32_t uid_len_current;
if(!flipper_format_get_value_count(ff, NFC_DEVICE_UID_KEY, &uid_len_current)) break;
if(uid_len_current > uid_maxlen) break;
if(!flipper_format_read_hex(ff, NFC_DEVICE_UID_KEY, uid, uid_len_current)) break;
*uid_len = uid_len_current;
loaded = true;
} while(false);
return loaded;
}
static bool nfc_device_load_unified(NfcDevice* instance, FlipperFormat* ff, uint32_t version) {
bool loaded = false;
FuriString* temp_str = furi_string_alloc();
do {
// Read Nfc device type
if(!flipper_format_read_string(ff, NFC_DEVICE_TYPE_KEY, temp_str)) break;
// Detect protocol
NfcProtocol protocol;
for(protocol = 0; protocol < NfcProtocolNum; ++protocol) {
if(furi_string_equal(temp_str, nfc_devices[protocol]->protocol_name)) {
break;
}
}
if(protocol == NfcProtocolNum) break;
nfc_device_clear(instance);
instance->protocol = protocol;
instance->protocol_data = nfc_devices[protocol]->alloc();
// Load UID
uint8_t uid[NFC_DEVICE_UID_MAX_LEN];
uint32_t uid_len;
if(!nfc_device_load_uid(ff, uid, &uid_len, NFC_DEVICE_UID_MAX_LEN)) break;
if(!nfc_device_set_uid(instance, uid, uid_len)) break;
// Load data
if(!nfc_devices[protocol]->load(instance->protocol_data, ff, version)) break;
loaded = true;
} while(false);
if(!loaded) {
nfc_device_clear(instance);
}
furi_string_free(temp_str);
return loaded;
}
static bool nfc_device_load_legacy(NfcDevice* instance, FlipperFormat* ff, uint32_t version) {
bool loaded = false;
FuriString* temp_str = furi_string_alloc();
do {
// Read Nfc device type
if(!flipper_format_read_string(ff, NFC_DEVICE_TYPE_KEY, temp_str)) break;
nfc_device_clear(instance);
// Detect protocol
for(NfcProtocol protocol = 0; protocol < NfcProtocolNum; protocol++) {
instance->protocol = protocol;
instance->protocol_data = nfc_devices[protocol]->alloc();
// Verify protocol
if(nfc_devices[protocol]->verify(instance->protocol_data, temp_str)) {
uint8_t uid[NFC_DEVICE_UID_MAX_LEN];
uint32_t uid_len;
// Load data
loaded = nfc_device_load_uid(ff, uid, &uid_len, NFC_DEVICE_UID_MAX_LEN) &&
nfc_device_set_uid(instance, uid, uid_len) &&
nfc_devices[protocol]->load(instance->protocol_data, ff, version);
break;
}
nfc_device_clear(instance);
}
} while(false);
furi_string_free(temp_str);
return loaded;
}
bool nfc_device_load(NfcDevice* instance, const char* path) {
furi_assert(instance);
furi_assert(path);
bool loaded = false;
Storage* storage = furi_record_open(RECORD_STORAGE);
FlipperFormat* ff = flipper_format_buffered_file_alloc(storage);
FuriString* temp_str;
temp_str = furi_string_alloc();
if(instance->loading_callback) {
instance->loading_callback(instance->loading_callback_context, true);
}
do {
if(!flipper_format_buffered_file_open_existing(ff, path)) break;
// Read and verify file header
uint32_t version = 0;
if(!flipper_format_read_header(ff, temp_str, &version)) break;
if(furi_string_cmp_str(temp_str, NFC_FILE_HEADER)) break;
if(version < NFC_MINIMUM_SUPPORTED_FORMAT_VERSION) break;
// Select loading method
loaded = (version < NFC_UNIFIED_FORMAT_VERSION) ?
nfc_device_load_legacy(instance, ff, version) :
nfc_device_load_unified(instance, ff, version);
} while(false);
if(instance->loading_callback) {
instance->loading_callback(instance->loading_callback_context, false);
}
furi_string_free(temp_str);
flipper_format_free(ff);
furi_record_close(RECORD_STORAGE);
return loaded;
}