gen4_poller.c 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648
  1. #include "core/log.h"
  2. #include "gen4_poller_i.h"
  3. #include <nfc/protocols/iso14443_3a/iso14443_3a.h>
  4. #include <nfc/protocols/iso14443_3a/iso14443_3a_poller.h>
  5. #include <nfc/helpers/nfc_util.h>
  6. #include <bit_lib/bit_lib.h>
  7. #include <nfc/nfc_poller.h>
  8. #include <furi/furi.h>
  9. #define GEN4_POLLER_THREAD_FLAG_DETECTED (1U << 0)
  10. typedef NfcCommand (*Gen4PollerStateHandler)(Gen4Poller* instance);
  11. typedef struct {
  12. NfcPoller* poller;
  13. uint32_t password;
  14. BitBuffer* tx_buffer;
  15. BitBuffer* rx_buffer;
  16. FuriThreadId thread_id;
  17. Gen4PollerError error;
  18. } Gen4PollerDetectContext;
  19. static const uint8_t gen4_poller_default_config[] = {0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02,
  20. 0x00, 0x09, 0x78, 0x00, 0x91, 0x02, 0xDA,
  21. 0xBC, 0x19, 0x10, 0x10, 0x11, 0x12, 0x13,
  22. 0x14, 0x15, 0x16, 0x04, 0x00, 0x08, 0x00};
  23. static const uint8_t gen4_poller_default_block_0[GEN4_POLLER_BLOCK_SIZE] =
  24. {0x00, 0x01, 0x02, 0x03, 0x04, 0x04, 0x08, 0x04, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00};
  25. static const uint8_t gen4_poller_default_empty_block[GEN4_POLLER_BLOCK_SIZE] = {0};
  26. static const uint8_t gen4_poller_default_sector_trailer_block[GEN4_POLLER_BLOCK_SIZE] =
  27. {0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0x07, 0x80, 0x69, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF};
  28. static bool gen4_poller_is_sector_trailer(uint8_t block_num) {
  29. uint8_t sec_tr_block_num = 0;
  30. if(block_num < 128) {
  31. sec_tr_block_num = block_num | 0x03;
  32. } else {
  33. sec_tr_block_num = block_num | 0x0f;
  34. }
  35. return block_num == sec_tr_block_num;
  36. }
  37. Gen4Poller* gen4_poller_alloc(Nfc* nfc) {
  38. furi_assert(nfc);
  39. Gen4Poller* instance = malloc(sizeof(Gen4Poller));
  40. instance->poller = nfc_poller_alloc(nfc, NfcProtocolIso14443_3a);
  41. instance->gen4_event.data = &instance->gen4_event_data;
  42. instance->tx_buffer = bit_buffer_alloc(GEN4_POLLER_MAX_BUFFER_SIZE);
  43. instance->rx_buffer = bit_buffer_alloc(GEN4_POLLER_MAX_BUFFER_SIZE);
  44. return instance;
  45. }
  46. void gen4_poller_free(Gen4Poller* instance) {
  47. furi_assert(instance);
  48. nfc_poller_free(instance->poller);
  49. bit_buffer_free(instance->tx_buffer);
  50. bit_buffer_free(instance->rx_buffer);
  51. free(instance);
  52. }
  53. void gen4_poller_set_password(Gen4Poller* instance, uint32_t password) {
  54. furi_assert(instance);
  55. instance->password = password;
  56. }
  57. NfcCommand gen4_poller_detect_callback(NfcGenericEvent event, void* context) {
  58. furi_assert(context);
  59. furi_assert(event.protocol == NfcProtocolIso14443_3a);
  60. furi_assert(event.instance);
  61. furi_assert(event.event_data);
  62. NfcCommand command = NfcCommandStop;
  63. Gen4PollerDetectContext* gen4_poller_detect_ctx = context;
  64. Iso14443_3aPoller* iso3_poller = event.instance;
  65. Iso14443_3aPollerEvent* iso3_event = event.event_data;
  66. gen4_poller_detect_ctx->error = Gen4PollerErrorTimeout;
  67. if(iso3_event->type == Iso14443_3aPollerEventTypeReady) {
  68. do {
  69. bit_buffer_append_byte(gen4_poller_detect_ctx->tx_buffer, GEN4_CMD_PREFIX);
  70. uint8_t pwd_arr[4] = {};
  71. bit_lib_num_to_bytes_be(gen4_poller_detect_ctx->password, COUNT_OF(pwd_arr), pwd_arr);
  72. bit_buffer_append_bytes(gen4_poller_detect_ctx->tx_buffer, pwd_arr, COUNT_OF(pwd_arr));
  73. bit_buffer_append_byte(gen4_poller_detect_ctx->tx_buffer, GEN4_CMD_GET_CFG);
  74. Iso14443_3aError error = iso14443_3a_poller_send_standard_frame(
  75. iso3_poller,
  76. gen4_poller_detect_ctx->tx_buffer,
  77. gen4_poller_detect_ctx->rx_buffer,
  78. GEN4_POLLER_MAX_FWT);
  79. if(error != Iso14443_3aErrorNone) {
  80. gen4_poller_detect_ctx->error = Gen4PollerErrorProtocol;
  81. break;
  82. }
  83. size_t rx_bytes = bit_buffer_get_size_bytes(gen4_poller_detect_ctx->rx_buffer);
  84. if((rx_bytes != 30) && (rx_bytes != 32)) {
  85. gen4_poller_detect_ctx->error = Gen4PollerErrorProtocol;
  86. break;
  87. }
  88. gen4_poller_detect_ctx->error = Gen4PollerErrorNone;
  89. } while(false);
  90. } else if(iso3_event->type == Iso14443_3aPollerEventTypeError) {
  91. gen4_poller_detect_ctx->error = Gen4PollerErrorTimeout;
  92. }
  93. furi_thread_flags_set(gen4_poller_detect_ctx->thread_id, GEN4_POLLER_THREAD_FLAG_DETECTED);
  94. return command;
  95. }
  96. Gen4PollerError gen4_poller_detect(Nfc* nfc, uint32_t password) {
  97. furi_assert(nfc);
  98. Gen4PollerDetectContext gen4_poller_detect_ctx = {};
  99. gen4_poller_detect_ctx.poller = nfc_poller_alloc(nfc, NfcProtocolIso14443_3a);
  100. gen4_poller_detect_ctx.password = password;
  101. gen4_poller_detect_ctx.tx_buffer = bit_buffer_alloc(GEN4_POLLER_MAX_BUFFER_SIZE);
  102. gen4_poller_detect_ctx.rx_buffer = bit_buffer_alloc(GEN4_POLLER_MAX_BUFFER_SIZE);
  103. gen4_poller_detect_ctx.thread_id = furi_thread_get_current_id();
  104. gen4_poller_detect_ctx.error = Gen4PollerErrorNone;
  105. nfc_poller_start(
  106. gen4_poller_detect_ctx.poller, gen4_poller_detect_callback, &gen4_poller_detect_ctx);
  107. uint32_t flags =
  108. furi_thread_flags_wait(GEN4_POLLER_THREAD_FLAG_DETECTED, FuriFlagWaitAny, FuriWaitForever);
  109. if(flags & GEN4_POLLER_THREAD_FLAG_DETECTED) {
  110. furi_thread_flags_clear(GEN4_POLLER_THREAD_FLAG_DETECTED);
  111. }
  112. nfc_poller_stop(gen4_poller_detect_ctx.poller);
  113. nfc_poller_free(gen4_poller_detect_ctx.poller);
  114. bit_buffer_free(gen4_poller_detect_ctx.tx_buffer);
  115. bit_buffer_free(gen4_poller_detect_ctx.rx_buffer);
  116. return gen4_poller_detect_ctx.error;
  117. }
  118. NfcCommand gen4_poller_idle_handler(Gen4Poller* instance) {
  119. NfcCommand command = NfcCommandContinue;
  120. instance->current_block = 0;
  121. memset(instance->config, 0, sizeof(instance->config));
  122. instance->gen4_event.type = Gen4PollerEventTypeCardDetected;
  123. command = instance->callback(instance->gen4_event, instance->context);
  124. instance->state = Gen4PollerStateRequestMode;
  125. return command;
  126. }
  127. NfcCommand gen4_poller_request_mode_handler(Gen4Poller* instance) {
  128. NfcCommand command = NfcCommandContinue;
  129. instance->gen4_event.type = Gen4PollerEventTypeRequestMode;
  130. command = instance->callback(instance->gen4_event, instance->context);
  131. if(instance->gen4_event_data.request_mode.mode == Gen4PollerModeWipe) {
  132. instance->state = Gen4PollerStateWipe;
  133. } else if(instance->gen4_event_data.request_mode.mode == Gen4PollerModeWrite) {
  134. instance->state = Gen4PollerStateRequestWriteData;
  135. } else if(instance->gen4_event_data.request_mode.mode == Gen4PollerModeSetPassword) {
  136. instance->state = Gen4PollerStateChangePassword;
  137. } else if(instance->gen4_event_data.request_mode.mode == Gen4PollerModeSetDefaultCFG) {
  138. instance->state = Gen4PollerStateSetDefaultConfig;
  139. } else if(instance->gen4_event_data.request_mode.mode == Gen4PollerModeGetCFG) {
  140. instance->state = Gen4PollerStateGetCurrentConfig;
  141. } else if(instance->gen4_event_data.request_mode.mode == Gen4PollerModeGetRevision) {
  142. instance->state = Gen4PollerStateGetRevision;
  143. } else {
  144. instance->state = Gen4PollerStateFail;
  145. }
  146. return command;
  147. }
  148. NfcCommand gen4_poller_wipe_handler(Gen4Poller* instance) {
  149. NfcCommand command = NfcCommandContinue;
  150. do {
  151. Gen4PollerError error = Gen4PollerErrorNone;
  152. if(instance->current_block == 0) {
  153. error = gen4_poller_set_config(
  154. instance,
  155. instance->password,
  156. gen4_poller_default_config,
  157. sizeof(gen4_poller_default_config),
  158. false);
  159. if(error != Gen4PollerErrorNone) {
  160. FURI_LOG_D(TAG, "Failed to set default config: %d", error);
  161. instance->state = Gen4PollerStateFail;
  162. break;
  163. }
  164. instance->password = 0;
  165. error = gen4_poller_write_block(
  166. instance, instance->password, instance->current_block, gen4_poller_default_block_0);
  167. if(error != Gen4PollerErrorNone) {
  168. FURI_LOG_D(TAG, "Failed to write 0 block: %d", error);
  169. instance->state = Gen4PollerStateFail;
  170. break;
  171. }
  172. } else if(instance->current_block < GEN4_POLLER_BLOCKS_TOTAL) {
  173. const uint8_t* block = gen4_poller_is_sector_trailer(instance->current_block) ?
  174. gen4_poller_default_sector_trailer_block :
  175. gen4_poller_default_empty_block;
  176. error = gen4_poller_write_block(
  177. instance, instance->password, instance->current_block, block);
  178. if(error != Gen4PollerErrorNone) {
  179. FURI_LOG_D(TAG, "Failed to write %d block: %d", instance->current_block, error);
  180. instance->state = Gen4PollerStateFail;
  181. break;
  182. }
  183. } else {
  184. instance->state = Gen4PollerStateSuccess;
  185. break;
  186. }
  187. instance->current_block++;
  188. } while(false);
  189. return command;
  190. }
  191. NfcCommand gen4_poller_request_write_data_handler(Gen4Poller* instance) {
  192. NfcCommand command = NfcCommandContinue;
  193. instance->gen4_event.type = Gen4PollerEventTypeRequestDataToWrite;
  194. command = instance->callback(instance->gen4_event, instance->context);
  195. instance->protocol = instance->gen4_event_data.request_data.protocol;
  196. instance->data = instance->gen4_event_data.request_data.data;
  197. if((instance->protocol == NfcProtocolMfClassic) ||
  198. (instance->protocol == NfcProtocolMfUltralight)) {
  199. instance->state = Gen4PollerStateWrite;
  200. } else {
  201. FURI_LOG_E(TAG, "Unsupported protocol");
  202. instance->state = Gen4PollerStateFail;
  203. }
  204. return command;
  205. }
  206. static NfcCommand gen4_poller_write_mf_classic(Gen4Poller* instance) {
  207. NfcCommand command = NfcCommandContinue;
  208. do {
  209. const MfClassicData* mfc_data = instance->data;
  210. const Iso14443_3aData* iso3_data = mfc_data->iso14443_3a_data;
  211. if(instance->current_block == 0) {
  212. instance->config[0] = 0x00;
  213. instance->total_blocks = mf_classic_get_total_block_num(mfc_data->type);
  214. if(iso3_data->uid_len == 4) {
  215. instance->config[1] = Gen4PollerUIDLengthSingle;
  216. } else if(iso3_data->uid_len == 7) {
  217. instance->config[1] = Gen4PollerUIDLengthDouble;
  218. } else {
  219. FURI_LOG_E(TAG, "Unsupported UID len: %d", iso3_data->uid_len);
  220. instance->state = Gen4PollerStateFail;
  221. break;
  222. }
  223. instance->config[6] = Gen4PollerShadowModeIgnore;
  224. instance->config[24] = iso3_data->atqa[0];
  225. instance->config[25] = iso3_data->atqa[1];
  226. instance->config[26] = iso3_data->sak;
  227. instance->config[27] = 0x00;
  228. instance->config[28] = instance->total_blocks - 1;
  229. instance->config[29] = 0x01;
  230. Gen4PollerError error = gen4_poller_set_config(
  231. instance, instance->password, instance->config, sizeof(instance->config), false);
  232. if(error != Gen4PollerErrorNone) {
  233. FURI_LOG_D(TAG, "Failed to write config: %d", error);
  234. instance->state = Gen4PollerStateFail;
  235. break;
  236. }
  237. }
  238. if(instance->current_block < instance->total_blocks) {
  239. FURI_LOG_D(TAG, "Writing block %d", instance->current_block);
  240. Gen4PollerError error = gen4_poller_write_block(
  241. instance,
  242. instance->password,
  243. instance->current_block,
  244. mfc_data->block[instance->current_block].data);
  245. if(error != Gen4PollerErrorNone) {
  246. FURI_LOG_D(TAG, "Failed to write %d block: %d", instance->current_block, error);
  247. instance->state = Gen4PollerStateFail;
  248. break;
  249. }
  250. } else {
  251. instance->state = Gen4PollerStateSuccess;
  252. break;
  253. }
  254. instance->current_block++;
  255. } while(false);
  256. return command;
  257. }
  258. static NfcCommand gen4_poller_write_mf_ultralight(Gen4Poller* instance) {
  259. NfcCommand command = NfcCommandContinue;
  260. do {
  261. const MfUltralightData* mfu_data = instance->data;
  262. const Iso14443_3aData* iso3_data = mfu_data->iso14443_3a_data;
  263. if(instance->current_block == 0) {
  264. instance->total_blocks = 64;
  265. instance->config[0] = 0x01;
  266. switch(mfu_data->type) {
  267. case MfUltralightTypeNTAG203:
  268. case MfUltralightTypeNTAG213:
  269. case MfUltralightTypeNTAG215:
  270. case MfUltralightTypeNTAG216:
  271. case MfUltralightTypeNTAGI2C1K:
  272. case MfUltralightTypeNTAGI2C2K:
  273. case MfUltralightTypeNTAGI2CPlus1K:
  274. case MfUltralightTypeNTAGI2CPlus2K:
  275. instance->config[27] = Gen4PollerUltralightModeNTAG;
  276. instance->total_blocks = 64 * 2;
  277. break;
  278. case MfUltralightTypeUL11:
  279. case MfUltralightTypeUL21:
  280. // UL-C?
  281. // UL?
  282. default:
  283. instance->config[27] = Gen4PollerUltralightModeUL_EV1;
  284. break;
  285. }
  286. if(iso3_data->uid_len == 4) {
  287. instance->config[1] = Gen4PollerUIDLengthSingle;
  288. } else if(iso3_data->uid_len == 7) {
  289. instance->config[1] = Gen4PollerUIDLengthDouble;
  290. } else {
  291. FURI_LOG_E(TAG, "Unsupported UID len: %d", iso3_data->uid_len);
  292. instance->state = Gen4PollerStateFail;
  293. break;
  294. }
  295. instance->config[6] = Gen4PollerShadowModeHighSpeedIgnore;
  296. instance->config[24] = iso3_data->atqa[0];
  297. instance->config[25] = iso3_data->atqa[1];
  298. instance->config[26] = iso3_data->sak;
  299. instance->config[27] = 0x00;
  300. instance->config[28] = instance->total_blocks - 1;
  301. instance->config[29] = 0x01;
  302. Gen4PollerError error = gen4_poller_set_config(
  303. instance, instance->password, instance->config, sizeof(instance->config), false);
  304. if(error != Gen4PollerErrorNone) {
  305. FURI_LOG_D(TAG, "Failed to write config: %d", error);
  306. instance->state = Gen4PollerStateFail;
  307. break;
  308. }
  309. }
  310. if(instance->current_block < mfu_data->pages_read) {
  311. FURI_LOG_D(
  312. TAG, "Writing page %zu / %zu", instance->current_block, mfu_data->pages_read);
  313. Gen4PollerError error = gen4_poller_write_block(
  314. instance,
  315. instance->password,
  316. instance->current_block,
  317. mfu_data->page[instance->current_block].data);
  318. if(error != Gen4PollerErrorNone) {
  319. FURI_LOG_D(TAG, "Failed to write %d page: %d", instance->current_block, error);
  320. instance->state = Gen4PollerStateFail;
  321. break;
  322. }
  323. instance->current_block++;
  324. } else {
  325. uint8_t block[GEN4_POLLER_BLOCK_SIZE] = {};
  326. bool write_success = true;
  327. for(size_t i = 0; i < 8; i++) {
  328. memcpy(block, &mfu_data->signature.data[i * 4], 4); //-V1086
  329. Gen4PollerError error =
  330. gen4_poller_write_block(instance, instance->password, 0xF2 + i, block);
  331. if(error != Gen4PollerErrorNone) {
  332. write_success = false;
  333. break;
  334. }
  335. }
  336. if(!write_success) {
  337. FURI_LOG_E(TAG, "Failed to write Signature");
  338. instance->state = Gen4PollerStateFail;
  339. break;
  340. }
  341. block[0] = mfu_data->version.header;
  342. block[1] = mfu_data->version.vendor_id;
  343. block[2] = mfu_data->version.prod_type;
  344. block[3] = mfu_data->version.prod_subtype;
  345. Gen4PollerError error =
  346. gen4_poller_write_block(instance, instance->password, 0xFA, block);
  347. if(error != Gen4PollerErrorNone) {
  348. FURI_LOG_E(TAG, "Failed to write 1st part Version");
  349. instance->state = Gen4PollerStateFail;
  350. break;
  351. }
  352. block[0] = mfu_data->version.prod_ver_major;
  353. block[1] = mfu_data->version.prod_ver_minor;
  354. block[2] = mfu_data->version.storage_size;
  355. block[3] = mfu_data->version.protocol_type;
  356. error = gen4_poller_write_block(instance, instance->password, 0xFB, block);
  357. if(error != Gen4PollerErrorNone) {
  358. FURI_LOG_E(TAG, "Failed to write 2nd part Version");
  359. instance->state = Gen4PollerStateFail;
  360. break;
  361. }
  362. // Password
  363. MfUltralightConfigPages* config_pages = NULL;
  364. mf_ultralight_get_config_page(mfu_data, &config_pages);
  365. block[0] = config_pages->password.data[0];
  366. block[1] = config_pages->password.data[1];
  367. block[2] = config_pages->password.data[2];
  368. block[3] = config_pages->password.data[3];
  369. error = gen4_poller_write_block(instance, instance->password, 0xE5, block);
  370. if(error != Gen4PollerErrorNone) {
  371. FURI_LOG_E(TAG, "Failed to write Password to sector E5");
  372. instance->state = Gen4PollerStateFail;
  373. break;
  374. }
  375. error = gen4_poller_write_block(instance, instance->password, 0xF0, block);
  376. if(error != Gen4PollerErrorNone) {
  377. FURI_LOG_E(TAG, "Failed to write Password to sector F0");
  378. instance->state = Gen4PollerStateFail;
  379. break;
  380. }
  381. // PACK
  382. block[0] = config_pages->pack.data[0];
  383. block[1] = config_pages->pack.data[1];
  384. block[2] = 0x00;
  385. block[3] = 0x00;
  386. error = gen4_poller_write_block(instance, instance->password, 0xE6, block);
  387. if(error != Gen4PollerErrorNone) {
  388. FURI_LOG_E(TAG, "Failed to write PACK to sector E6");
  389. instance->state = Gen4PollerStateFail;
  390. break;
  391. }
  392. error = gen4_poller_write_block(instance, instance->password, 0xF1, block);
  393. if(error != Gen4PollerErrorNone) {
  394. FURI_LOG_E(TAG, "Failed to write PACK to sector F1");
  395. instance->state = Gen4PollerStateFail;
  396. break;
  397. }
  398. instance->state = Gen4PollerStateSuccess;
  399. }
  400. } while(false);
  401. return command;
  402. }
  403. NfcCommand gen4_poller_write_handler(Gen4Poller* instance) {
  404. NfcCommand command = NfcCommandContinue;
  405. memcpy(instance->config, gen4_poller_default_config, sizeof(gen4_poller_default_config));
  406. uint8_t password_arr[4] = {};
  407. bit_lib_num_to_bytes_be(instance->password, sizeof(password_arr), password_arr);
  408. memcpy(&instance->config[2], password_arr, sizeof(password_arr));
  409. memset(&instance->config[7], 0, 17);
  410. if(instance->protocol == NfcProtocolMfClassic) {
  411. command = gen4_poller_write_mf_classic(instance);
  412. } else if(instance->protocol == NfcProtocolMfUltralight) {
  413. command = gen4_poller_write_mf_ultralight(instance);
  414. } else {
  415. furi_crash("Unsupported protocol to write");
  416. }
  417. return command;
  418. }
  419. NfcCommand gen4_poller_change_password_handler(Gen4Poller* instance) {
  420. NfcCommand command = NfcCommandContinue;
  421. do {
  422. instance->gen4_event.type = Gen4PollerEventTypeRequestNewPassword;
  423. command = instance->callback(instance->gen4_event, instance->context);
  424. if(command != NfcCommandContinue) break;
  425. uint32_t new_password = instance->gen4_event_data.request_password.password;
  426. Gen4PollerError error =
  427. gen4_poller_change_password(instance, instance->password, new_password);
  428. if(error != Gen4PollerErrorNone) {
  429. FURI_LOG_E(TAG, "Failed to change password: %d", error);
  430. instance->state = Gen4PollerStateFail;
  431. break;
  432. }
  433. instance->password = new_password;
  434. instance->state = Gen4PollerStateSuccess;
  435. } while(false);
  436. return command;
  437. }
  438. NfcCommand gen4_poller_set_default_cfg_handler(Gen4Poller* instance) {
  439. NfcCommand command = NfcCommandContinue;
  440. do {
  441. Gen4PollerError error = gen4_poller_set_config(
  442. instance,
  443. instance->password,
  444. gen4_poller_default_config,
  445. sizeof(gen4_poller_default_config),
  446. false);
  447. if(error != Gen4PollerErrorNone) {
  448. FURI_LOG_E(TAG, "Failed to set default config: %d", error);
  449. instance->state = Gen4PollerStateFail;
  450. break;
  451. }
  452. instance->state = Gen4PollerStateSuccess;
  453. } while(false);
  454. return command;
  455. }
  456. NfcCommand gen4_poller_get_current_cfg_handler(Gen4Poller* instance) {
  457. NfcCommand command = NfcCommandContinue;
  458. do {
  459. uint8_t the_config[32] = {};
  460. Gen4PollerError error = gen4_poller_get_config(instance, instance->password, the_config);
  461. if(error != Gen4PollerErrorNone) {
  462. FURI_LOG_E(TAG, "Failed to get current config: %d", error);
  463. instance->state = Gen4PollerStateFail;
  464. break;
  465. }
  466. // Copy config data to event data buffer
  467. memcpy(instance->gen4_event_data.display_config, the_config, sizeof(the_config));
  468. instance->state = Gen4PollerStateSuccess;
  469. } while(false);
  470. return command;
  471. }
  472. NfcCommand gen4_poller_get_revision_handler(Gen4Poller* instance) {
  473. NfcCommand command = NfcCommandContinue;
  474. do {
  475. uint8_t the_revision[5] = {0};
  476. Gen4PollerError error =
  477. gen4_poller_get_revision(instance, instance->password, the_revision);
  478. if(error != Gen4PollerErrorNone) {
  479. FURI_LOG_E(TAG, "Failed to get revision: %d", error);
  480. instance->state = Gen4PollerStateFail;
  481. break;
  482. }
  483. // Copy revision data to event data buffer
  484. memcpy(instance->gen4_event_data.revision_data, the_revision, sizeof(the_revision));
  485. instance->state = Gen4PollerStateSuccess;
  486. } while(false);
  487. return command;
  488. }
  489. NfcCommand gen4_poller_success_handler(Gen4Poller* instance) {
  490. NfcCommand command = NfcCommandContinue;
  491. instance->gen4_event.type = Gen4PollerEventTypeSuccess;
  492. command = instance->callback(instance->gen4_event, instance->context);
  493. if(command != NfcCommandStop) {
  494. furi_delay_ms(100);
  495. }
  496. return command;
  497. }
  498. NfcCommand gen4_poller_fail_handler(Gen4Poller* instance) {
  499. NfcCommand command = NfcCommandContinue;
  500. instance->gen4_event.type = Gen4PollerEventTypeFail;
  501. command = instance->callback(instance->gen4_event, instance->context);
  502. if(command != NfcCommandStop) {
  503. furi_delay_ms(100);
  504. }
  505. return command;
  506. }
  507. static const Gen4PollerStateHandler gen4_poller_state_handlers[Gen4PollerStateNum] = {
  508. [Gen4PollerStateIdle] = gen4_poller_idle_handler,
  509. [Gen4PollerStateRequestMode] = gen4_poller_request_mode_handler,
  510. [Gen4PollerStateRequestWriteData] = gen4_poller_request_write_data_handler,
  511. [Gen4PollerStateWrite] = gen4_poller_write_handler,
  512. [Gen4PollerStateWipe] = gen4_poller_wipe_handler,
  513. [Gen4PollerStateChangePassword] = gen4_poller_change_password_handler,
  514. [Gen4PollerStateSetDefaultConfig] = gen4_poller_set_default_cfg_handler,
  515. [Gen4PollerStateGetCurrentConfig] = gen4_poller_get_current_cfg_handler,
  516. [Gen4PollerStateGetRevision] = gen4_poller_get_revision_handler,
  517. [Gen4PollerStateSuccess] = gen4_poller_success_handler,
  518. [Gen4PollerStateFail] = gen4_poller_fail_handler,
  519. };
  520. static NfcCommand gen4_poller_callback(NfcGenericEvent event, void* context) {
  521. furi_assert(context);
  522. furi_assert(event.protocol == NfcProtocolIso14443_3a);
  523. furi_assert(event.event_data);
  524. furi_assert(event.instance);
  525. NfcCommand command = NfcCommandContinue;
  526. Gen4Poller* instance = context;
  527. instance->iso3_poller = event.instance;
  528. Iso14443_3aPollerEvent* iso3_event = event.event_data;
  529. if(iso3_event->type == Iso14443_3aPollerEventTypeReady) {
  530. command = gen4_poller_state_handlers[instance->state](instance);
  531. }
  532. return command;
  533. }
  534. void gen4_poller_start(Gen4Poller* instance, Gen4PollerCallback callback, void* context) {
  535. furi_assert(instance);
  536. furi_assert(callback);
  537. instance->callback = callback;
  538. instance->context = context;
  539. nfc_poller_start(instance->poller, gen4_poller_callback, instance);
  540. }
  541. void gen4_poller_stop(Gen4Poller* instance) {
  542. furi_assert(instance);
  543. nfc_poller_stop(instance->poller);
  544. }