#include "ibutton_protocols.h" #include #include "ibutton_key_i.h" #include "protocols/protocol_group_defs.h" #define IBUTTON_FILE_TYPE "Flipper iButton key" #define IBUTTON_PROTOCOL_KEY_V1 "Key type" #define IBUTTON_PROTOCOL_KEY_V2 "Protocol" #define IBUTTON_CURRENT_FORMAT_VERSION 2U #define GET_PROTOCOL_GROUP(id) \ iButtonProtocolGroupInfo info; \ ibutton_protocols_get_group_by_id(protocols, (id), &info); #define GROUP_BASE (info.base) #define GROUP_DATA (info.group) #define PROTOCOL_ID (info.id) struct iButtonProtocols { iButtonProtocolGroupData** group_datas; }; typedef struct { const iButtonProtocolGroupBase* base; iButtonProtocolGroupData* group; iButtonProtocolLocalId id; } iButtonProtocolGroupInfo; static void ibutton_protocols_get_group_by_id( iButtonProtocols* protocols, iButtonProtocolId id, iButtonProtocolGroupInfo* info) { iButtonProtocolLocalId local_id = id; for(iButtonProtocolGroupId i = 0; i < iButtonProtocolGroupMax; ++i) { if(local_id < (signed)ibutton_protocol_groups[i]->protocol_count) { info->base = ibutton_protocol_groups[i]; info->group = protocols->group_datas[i]; info->id = local_id; return; } else { local_id -= ibutton_protocol_groups[i]->protocol_count; } } furi_crash(); } iButtonProtocols* ibutton_protocols_alloc(void) { iButtonProtocols* protocols = malloc(sizeof(iButtonProtocols*)); protocols->group_datas = malloc(sizeof(iButtonProtocolGroupData*) * iButtonProtocolGroupMax); for(iButtonProtocolGroupId i = 0; i < iButtonProtocolGroupMax; ++i) { protocols->group_datas[i] = ibutton_protocol_groups[i]->alloc(); } return protocols; } void ibutton_protocols_free(iButtonProtocols* protocols) { furi_check(protocols); for(iButtonProtocolGroupId i = 0; i < iButtonProtocolGroupMax; ++i) { ibutton_protocol_groups[i]->free(protocols->group_datas[i]); } free(protocols->group_datas); free(protocols); } uint32_t ibutton_protocols_get_protocol_count(void) { uint32_t count = 0; for(iButtonProtocolGroupId i = 0; i < iButtonProtocolGroupMax; ++i) { count += ibutton_protocol_groups[i]->protocol_count; } return count; } iButtonProtocolId ibutton_protocols_get_id_by_name(iButtonProtocols* protocols, const char* name) { furi_check(protocols); furi_check(name); iButtonProtocolLocalId offset = 0; for(iButtonProtocolGroupId i = 0; i < iButtonProtocolGroupMax; ++i) { iButtonProtocolLocalId local_id; if(ibutton_protocol_groups[i]->get_id_by_name(protocols->group_datas[i], &local_id, name)) { return local_id + offset; } offset += ibutton_protocol_groups[i]->protocol_count; } return iButtonProtocolIdInvalid; } uint32_t ibutton_protocols_get_features(iButtonProtocols* protocols, iButtonProtocolId id) { furi_check(protocols); GET_PROTOCOL_GROUP(id); return GROUP_BASE->get_features(GROUP_DATA, PROTOCOL_ID); } size_t ibutton_protocols_get_max_data_size(iButtonProtocols* protocols) { furi_check(protocols); size_t max_size = 0; for(iButtonProtocolGroupId i = 0; i < iButtonProtocolGroupMax; ++i) { const size_t current_max_size = ibutton_protocol_groups[i]->get_max_data_size(protocols->group_datas[i]); if(current_max_size > max_size) { max_size = current_max_size; } } return max_size; } const char* ibutton_protocols_get_manufacturer(iButtonProtocols* protocols, iButtonProtocolId id) { furi_check(protocols); GET_PROTOCOL_GROUP(id); return GROUP_BASE->get_manufacturer(GROUP_DATA, PROTOCOL_ID); } const char* ibutton_protocols_get_name(iButtonProtocols* protocols, iButtonProtocolId id) { furi_check(protocols); GET_PROTOCOL_GROUP(id); return GROUP_BASE->get_name(GROUP_DATA, PROTOCOL_ID); } bool ibutton_protocols_read(iButtonProtocols* protocols, iButtonKey* key) { furi_check(protocols); furi_check(key); iButtonProtocolLocalId id = iButtonProtocolIdInvalid; iButtonProtocolData* data = ibutton_key_get_protocol_data(key); iButtonProtocolLocalId offset = 0; for(iButtonProtocolGroupId i = 0; i < iButtonProtocolGroupMax; ++i) { if(ibutton_protocol_groups[i]->read(protocols->group_datas[i], data, &id)) { id += offset; break; } offset += ibutton_protocol_groups[i]->protocol_count; } ibutton_key_set_protocol_id(key, id); return id != iButtonProtocolIdInvalid; } bool ibutton_protocols_write_id(iButtonProtocols* protocols, iButtonKey* key) { furi_check(protocols); furi_check(key); const iButtonProtocolId id = ibutton_key_get_protocol_id(key); iButtonProtocolData* data = ibutton_key_get_protocol_data(key); GET_PROTOCOL_GROUP(id); return GROUP_BASE->write_id(GROUP_DATA, data, PROTOCOL_ID); } bool ibutton_protocols_write_copy(iButtonProtocols* protocols, iButtonKey* key) { furi_check(protocols); furi_check(key); const iButtonProtocolId id = ibutton_key_get_protocol_id(key); iButtonProtocolData* data = ibutton_key_get_protocol_data(key); GET_PROTOCOL_GROUP(id); return GROUP_BASE->write_copy(GROUP_DATA, data, PROTOCOL_ID); } void ibutton_protocols_emulate_start(iButtonProtocols* protocols, iButtonKey* key) { furi_check(protocols); furi_check(key); const iButtonProtocolId id = ibutton_key_get_protocol_id(key); iButtonProtocolData* data = ibutton_key_get_protocol_data(key); GET_PROTOCOL_GROUP(id); GROUP_BASE->emulate_start(GROUP_DATA, data, PROTOCOL_ID); } void ibutton_protocols_emulate_stop(iButtonProtocols* protocols, iButtonKey* key) { furi_check(protocols); furi_check(key); const iButtonProtocolId id = ibutton_key_get_protocol_id(key); iButtonProtocolData* data = ibutton_key_get_protocol_data(key); GET_PROTOCOL_GROUP(id); GROUP_BASE->emulate_stop(GROUP_DATA, data, PROTOCOL_ID); } bool ibutton_protocols_save( iButtonProtocols* protocols, const iButtonKey* key, const char* file_name) { furi_check(protocols); furi_check(key); furi_check(file_name); const iButtonProtocolId id = ibutton_key_get_protocol_id(key); const iButtonProtocolData* data = ibutton_key_get_protocol_data(key); bool success = false; Storage* storage = furi_record_open(RECORD_STORAGE); FlipperFormat* ff = flipper_format_buffered_file_alloc(storage); do { const char* protocol_name = ibutton_protocols_get_name(protocols, id); if(!flipper_format_buffered_file_open_always(ff, file_name)) break; if(!flipper_format_write_header_cstr(ff, IBUTTON_FILE_TYPE, IBUTTON_CURRENT_FORMAT_VERSION)) break; if(!flipper_format_write_string_cstr(ff, IBUTTON_PROTOCOL_KEY_V2, protocol_name)) break; GET_PROTOCOL_GROUP(id); if(!GROUP_BASE->save(GROUP_DATA, data, PROTOCOL_ID, ff)) break; success = true; } while(false); flipper_format_free(ff); furi_record_close(RECORD_STORAGE); return success; } bool ibutton_protocols_load(iButtonProtocols* protocols, iButtonKey* key, const char* file_name) { furi_check(protocols); furi_check(key); furi_check(file_name); iButtonProtocolData* data = ibutton_key_get_protocol_data(key); bool success = false; Storage* storage = furi_record_open(RECORD_STORAGE); FlipperFormat* ff = flipper_format_buffered_file_alloc(storage); FuriString* tmp = furi_string_alloc(); do { if(!flipper_format_buffered_file_open_existing(ff, file_name)) break; uint32_t version; if(!flipper_format_read_header(ff, tmp, &version)) break; if(!furi_string_equal(tmp, IBUTTON_FILE_TYPE)) break; if(version == 1) { if(!flipper_format_read_string(ff, IBUTTON_PROTOCOL_KEY_V1, tmp)) break; } else if(version == 2) { if(!flipper_format_read_string(ff, IBUTTON_PROTOCOL_KEY_V2, tmp)) break; } else { break; } const iButtonProtocolId id = ibutton_protocols_get_id_by_name(protocols, furi_string_get_cstr(tmp)); ibutton_key_set_protocol_id(key, id); GET_PROTOCOL_GROUP(id); if(!GROUP_BASE->load(GROUP_DATA, data, PROTOCOL_ID, version, ff)) break; success = true; } while(false); flipper_format_free(ff); furi_string_free(tmp); furi_record_close(RECORD_STORAGE); return success; } void ibutton_protocols_render_uid( iButtonProtocols* protocols, const iButtonKey* key, FuriString* result) { const iButtonProtocolId id = ibutton_key_get_protocol_id(key); const iButtonProtocolData* data = ibutton_key_get_protocol_data(key); GET_PROTOCOL_GROUP(id); GROUP_BASE->render_uid(GROUP_DATA, data, PROTOCOL_ID, result); } void ibutton_protocols_render_data( iButtonProtocols* protocols, const iButtonKey* key, FuriString* result) { furi_check(protocols); furi_check(key); furi_check(result); const iButtonProtocolId id = ibutton_key_get_protocol_id(key); const iButtonProtocolData* data = ibutton_key_get_protocol_data(key); GET_PROTOCOL_GROUP(id); GROUP_BASE->render_data(GROUP_DATA, data, PROTOCOL_ID, result); } void ibutton_protocols_render_brief_data( iButtonProtocols* protocols, const iButtonKey* key, FuriString* result) { furi_check(protocols); furi_check(key); furi_check(result); const iButtonProtocolId id = ibutton_key_get_protocol_id(key); const iButtonProtocolData* data = ibutton_key_get_protocol_data(key); GET_PROTOCOL_GROUP(id); GROUP_BASE->render_brief_data(GROUP_DATA, data, PROTOCOL_ID, result); } void ibutton_protocols_render_error( iButtonProtocols* protocols, const iButtonKey* key, FuriString* result) { furi_check(protocols); furi_check(key); furi_check(result); const iButtonProtocolId id = ibutton_key_get_protocol_id(key); const iButtonProtocolData* data = ibutton_key_get_protocol_data(key); GET_PROTOCOL_GROUP(id); GROUP_BASE->render_error(GROUP_DATA, data, PROTOCOL_ID, result); } bool ibutton_protocols_is_valid(iButtonProtocols* protocols, const iButtonKey* key) { furi_check(protocols); furi_check(key); const iButtonProtocolId id = ibutton_key_get_protocol_id(key); const iButtonProtocolData* data = ibutton_key_get_protocol_data(key); GET_PROTOCOL_GROUP(id); return GROUP_BASE->is_valid(GROUP_DATA, data, PROTOCOL_ID); } void ibutton_protocols_get_editable_data( iButtonProtocols* protocols, const iButtonKey* key, iButtonEditableData* editable) { furi_check(protocols); furi_check(key); furi_check(editable); const iButtonProtocolId id = ibutton_key_get_protocol_id(key); iButtonProtocolData* data = ibutton_key_get_protocol_data(key); GET_PROTOCOL_GROUP(id); GROUP_BASE->get_editable_data(GROUP_DATA, data, PROTOCOL_ID, editable); } void ibutton_protocols_apply_edits(iButtonProtocols* protocols, const iButtonKey* key) { furi_check(protocols); furi_check(key); const iButtonProtocolId id = ibutton_key_get_protocol_id(key); iButtonProtocolData* data = ibutton_key_get_protocol_data(key); GET_PROTOCOL_GROUP(id); GROUP_BASE->apply_edits(GROUP_DATA, data, PROTOCOL_ID); }