nrf24scan.c 51 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395
  1. //
  2. // Written by vad7, 20.11.2022.
  3. //
  4. #include "nrf24scan.h"
  5. #include <furi.h>
  6. #include <furi_hal.h>
  7. #include <gui/gui.h>
  8. #include <dialogs/dialogs.h>
  9. #include <input/input.h>
  10. #include <stdlib.h>
  11. #include <dolphin/dolphin.h>
  12. #include <nrf24.h>
  13. #include <u8g2.h>
  14. #define TAG "nrf24scan"
  15. #define VERSION "2.0"
  16. #define MAX_CHANNEL 125
  17. #define MAX_ADDR 6
  18. #define SCAN_APP_PATH_FOLDER "/ext/nrf24scan"
  19. #define SETTINGS_FILENAME "addresses.txt"// Settings file format (1 parameter per line):
  20. // SNIFF - if present then sniff mode
  21. // Rate: 0/1/2 - rate in Mbps (=0.25/1/2)
  22. // Ch: 0..125 - default channel
  23. // ESB: 0/1 (1 - Enhanced ShockBurst)
  24. // DPL: 0/1 (1 - Dynamic Payload Length)
  25. // CRC: 0/1/2 (CRC length)
  26. // Payload: 1..32 (bytes)
  27. // P0: address P0 in hex (5 byte, LSB last)
  28. // P1: address P1 in hex (5 byte, LSB last)
  29. // P2: address P2, LSB in hex (1 byte)
  30. // P3: address P3, LSB in hex (1 byte)
  31. // P4: address P4, LSB in hex (1 byte)
  32. // P5: address P5, LSB in hex (1 byte)
  33. // captured data:
  34. // first byte = { RAW packet flag (0x80/0x00) } + { channel number }
  35. // second byte = { Payload len 5 bits, 0 = 32 } + {{ RAW packet: ESB flag 0x04/0x00 + address size-2 if RAW packet } or { pipe #(0..5) }},
  36. // ... up to MAX_LOG_RECORDS-1
  37. #define SNIFF_FILENAME "sniff.txt" // settings for sniff mode
  38. #define LOG_FILENAME "log"
  39. #define LOG_FILEEXT ".txt"
  40. #define MAX_LOG_RECORDS 100
  41. #define LOG_REC_SIZE 34 // max packet size
  42. #define VIEW_LOG_MAX_X 22
  43. #define VIEW_LOG_WIDTH_B 10 // bytes
  44. const char SettingsFld_Rate[] = "Rate:";
  45. const char SettingsFld_Ch[] = "Ch:";
  46. const char SettingsFld_ESB[] = "ESB:";
  47. const char SettingsFld_DPL[] = "DPL:";
  48. const char SettingsFld_CRC[] = "CRC:";
  49. const char SettingsFld_Payload[] = "Payload:";
  50. const char SettingsFld_Sniff[] = "SNIFF";
  51. char SettingsFld_Addr = 'P';
  52. Nrf24Scan* APP;
  53. uint8_t what_doing = 0; // 0 - setup, 1 - view log, 2 - view addresses
  54. uint8_t what_to_do = 1; // 0 - view, 1 - view & sniff, 2 - view & read, 3 - view & read selected addr
  55. uint32_t key_press_seq_ok = 0;
  56. uint8_t save_settings = 0;
  57. char screen_buf[64];
  58. char addr_file_name[32];
  59. uint8_t NRF_rate = 2; // 0 - 250Kbps, 1 - 1Mbps, 2 - 2Mbps
  60. uint8_t NRF_channel = 0;// 0..125
  61. uint8_t NRF_ESB = 1; // 0 - ShockBurst, 1 - Enhanced ShockBurst
  62. uint8_t NRF_DPL = 0; // 1 - Dynamic Payload Length
  63. uint8_t NRF_CRC = 2; // 1 - No, 1 - CRC 1byte, 2 - CRC 2byte
  64. uint8_t NRF_Payload = 32;// Payload len in bytes or Minimum payload in sniff mode, 0..32
  65. uint8_t NRF_Payload_sniff_min = 0;
  66. uint8_t NRF_AA_OFF = 0; // Disable Auto Acknowledgement
  67. bool NRF_ERROR = 0;
  68. struct ADDRS {
  69. uint8_t addr_P0[5]; // MSB first
  70. uint8_t addr_P1[5]; // MSB first
  71. uint8_t addr_P2; // LSB only, MSB bytes equal addr_P1
  72. uint8_t addr_P3; // LSB only, MSB bytes equal addr_P1
  73. uint8_t addr_P4; // LSB only, MSB bytes equal addr_P1
  74. uint8_t addr_P5; // LSB only, MSB bytes equal addr_P1
  75. uint8_t addr_len; // 2..5
  76. uint8_t addr_count;
  77. };
  78. struct ADDRS addrs;
  79. struct ADDRS addrs_sniff;
  80. bool sniff_loaded = 0;
  81. int16_t found_total;
  82. int16_t view_found = 0;
  83. int8_t log_to_file = 0; // 0 - no, 1 - yes(new), 2 - append, -1 - only clear
  84. uint16_t log_arr_idx;
  85. uint16_t view_log_arr_idx = 0;
  86. uint16_t view_log_arr_x = 0;
  87. bool save_to_new_log = true;
  88. uint16_t last_packet_send = -1;
  89. uint8_t last_packet_send_st = 0;
  90. int16_t find_channel_period = 0; // sec
  91. uint8_t menu_selected = 0;
  92. uint32_t start_time;
  93. uint8_t view_log_decode_PCF = 0; // view log: 1 - decode packet control field (9b) when ESB off. After pipe # (hex): <Payload len 6b><PID_2b+NO_ACK_1b>
  94. uint8_t view_log_decode_CRC = 0; // CRC bytes - 1/2, 0 - none
  95. #define menu_selected_max 5
  96. enum {
  97. Menu_open_file = 0,
  98. Menu_enter_channel,
  99. Menu_enter_rate,
  100. Menu_enter_scan_period,
  101. Menu_log,
  102. Menu_ok
  103. };
  104. //#define MIN(a, b) ((a<b)?a:b)
  105. static uint8_t GetHexVal(char hex) {
  106. return (uint8_t)hex - ((uint8_t)hex < 58 ? 48 : ((uint8_t)hex < 97 ? 55 : 87));
  107. }
  108. // Return num bytes in array
  109. static uint8_t ConvertHexToArray(char * hex, uint8_t * array, uint8_t maxlen) {
  110. uint8_t len = 0;
  111. while(maxlen) {
  112. uint8_t ch = *hex++;
  113. if(ch == 0) break;
  114. if(ch < '0') continue;
  115. *array++ = (GetHexVal(ch) << 4) + GetHexVal(*hex++);
  116. len++;
  117. maxlen--;
  118. }
  119. return len;
  120. }
  121. static void add_to_str_hex_bytes(char *out, char *arr, int bytes)
  122. {
  123. if(bytes <= 0) return;
  124. out += strlen(out);
  125. do {
  126. snprintf(out, 3, "%02X", *arr++);
  127. out += 2;
  128. } while(--bytes);
  129. }
  130. static void add_to_furi_str_hex_bytes(FuriString* str, char *arr, int bytes)
  131. {
  132. if(!bytes) return;
  133. do {
  134. furi_string_cat_printf(str, "%02X", *arr++);
  135. } while(--bytes);
  136. }
  137. static void add_to_str_hex_bytes_shift_9b(char *out, char *arr, int bytes)
  138. {
  139. if(bytes <= 0) return;
  140. out += strlen(out);
  141. arr++; // +8b
  142. do {
  143. snprintf(out, 4, "%02X", ((uint8_t)(*arr << 1)) | (*(arr+1) >> 7));
  144. arr++;
  145. out += 2;
  146. } while(--bytes);
  147. }
  148. void clear_log()
  149. {
  150. log_arr_idx = 0;
  151. view_log_arr_idx = 0;
  152. last_packet_send = -1;
  153. found_total = 0;
  154. view_found = 0;
  155. }
  156. void write_to_log_file(Storage* storage, bool f_settings)
  157. {
  158. if(log_arr_idx == 0 && !f_settings) return;
  159. Stream* file_stream = file_stream_alloc(storage);
  160. FuriString* str = furi_string_alloc();
  161. furi_string_set(str, SCAN_APP_PATH_FOLDER);
  162. furi_string_cat(str, "/");
  163. bool fl;
  164. if(f_settings) {
  165. furi_string_cat(str, SETTINGS_FILENAME);
  166. fl = file_stream_open(file_stream, furi_string_get_cstr(str), FSAM_READ_WRITE, FSOM_CREATE_ALWAYS);
  167. if(!fl) file_stream_close(file_stream);
  168. } else {
  169. furi_string_cat(str, LOG_FILENAME);
  170. furi_string_cat(str, LOG_FILEEXT);
  171. if(save_to_new_log) {
  172. int cnt = 1;
  173. do {
  174. fl = file_stream_open(file_stream, furi_string_get_cstr(str), FSAM_READ_WRITE, FSOM_CREATE_NEW);
  175. if(fl) break;
  176. file_stream_close(file_stream);
  177. furi_string_set(str, SCAN_APP_PATH_FOLDER);
  178. furi_string_cat(str, "/");
  179. furi_string_cat(str, LOG_FILENAME);
  180. furi_string_cat_printf(str, "-%02d", cnt);
  181. furi_string_cat(str, LOG_FILEEXT);
  182. } while(++cnt < 100);
  183. if(!fl) {
  184. FURI_LOG_E(TAG, "Failed to create new log file");
  185. notification_message(APP->notification, &sequence_blink_red_100);
  186. }
  187. } else {
  188. fl = file_stream_open(file_stream, furi_string_get_cstr(str), FSAM_READ_WRITE, FSOM_OPEN_APPEND);
  189. if(fl) {
  190. if(stream_size(file_stream) == 0) save_to_new_log = true;
  191. } else file_stream_close(file_stream);
  192. }
  193. }
  194. if(fl) {
  195. FURI_LOG_D(TAG, "Save to %s", furi_string_get_cstr(str));
  196. if(save_to_new_log || f_settings) {
  197. //if(what_to_do == 1) furi_string_printf(str, "%s\n", SettingsFld_Sniff); else furi_string_reset(str);
  198. furi_string_printf(str, "%s %d\n%s %d\n%s %d\n", SettingsFld_Rate, NRF_rate, SettingsFld_Ch, NRF_channel, SettingsFld_ESB, NRF_ESB);
  199. furi_string_cat_printf(str, "%s %d\n%s %d\n%s %d\n", SettingsFld_DPL, NRF_DPL, SettingsFld_CRC, NRF_CRC, SettingsFld_Payload, what_to_do == 1 ? NRF_Payload_sniff_min : NRF_Payload);
  200. furi_string_cat_printf(str, "P0: ");
  201. add_to_furi_str_hex_bytes(str, (char*)addrs.addr_P0, addrs.addr_len); furi_string_cat(str, "\n");
  202. if(addrs.addr_count > 1) { furi_string_cat_printf(str, "P1: "); add_to_furi_str_hex_bytes(str, (char*)addrs.addr_P1, addrs.addr_len); furi_string_cat(str, "\n"); }
  203. if(addrs.addr_count > 2) { furi_string_cat_printf(str, "P2: "); furi_string_cat_printf(str, "%02X\n", addrs.addr_P2); }
  204. if(addrs.addr_count > 3) { furi_string_cat_printf(str, "P3: "); furi_string_cat_printf(str, "%02X\n", addrs.addr_P3); }
  205. if(addrs.addr_count > 4) { furi_string_cat_printf(str, "P4: "); furi_string_cat_printf(str, "%02X\n", addrs.addr_P4); }
  206. if(addrs.addr_count > 5) { furi_string_cat_printf(str, "P5: "); furi_string_cat_printf(str, "%02X\n", addrs.addr_P5); }
  207. if(!(fl = stream_write_string(file_stream, str) == furi_string_size(str))) {
  208. FURI_LOG_E(TAG, "Failed to write header to file!");
  209. notification_message(APP->notification, &sequence_blink_red_100);
  210. }
  211. }
  212. if(fl) {
  213. if(f_settings) {
  214. save_settings = 0;
  215. if(strcmp(addr_file_name, "NONE") == 0) strcpy(addr_file_name, SETTINGS_FILENAME);
  216. } else {
  217. int i = 0;
  218. for(; i < log_arr_idx; i++) {
  219. furi_string_reset(str);
  220. uint8_t *ptr = APP->log_arr + i * LOG_REC_SIZE;
  221. int len;
  222. if(ptr[0] & 0x80) { // RAW
  223. len = (ptr[1] & 0b11)+2 + ((ptr[1] & 0b100) ? 2 : 0) + (ptr[1] >> 3) + 2; // addr + PCF? + payload + crcmax
  224. } else {
  225. len = (ptr[1] >> 3);
  226. if(len == 0) len = 32;
  227. }
  228. //if(len < NRF_Payload) len = NRF_Payload;
  229. add_to_furi_str_hex_bytes(str, (char*)ptr, len + 2);
  230. furi_string_cat(str, "\n");
  231. if(stream_write_string(file_stream, str) != furi_string_size(str)) {
  232. FURI_LOG_E(TAG, "Failed to write to file!");
  233. break;
  234. }
  235. }
  236. if(i == log_arr_idx) {
  237. notification_message(APP->notification, &sequence_blink_yellow_100);
  238. FURI_LOG_D(TAG, "File saved");
  239. }
  240. save_to_new_log = false;
  241. }
  242. }
  243. file_stream_close(file_stream);
  244. } else {
  245. FURI_LOG_E(TAG, "Failed to open file %s", furi_string_get_cstr(str));
  246. notification_message(APP->notification, &sequence_blink_red_100);
  247. }
  248. stream_free(file_stream);
  249. furi_string_free(str);
  250. }
  251. static bool select_settings_file(Stream* stream) {
  252. DialogsApp* dialogs = furi_record_open("dialogs");
  253. bool result = false;
  254. FuriString* path;
  255. path = furi_string_alloc();
  256. furi_string_set(path, SCAN_APP_PATH_FOLDER);
  257. DialogsFileBrowserOptions browser_options;
  258. dialog_file_browser_set_basic_options(&browser_options, ".txt", NULL);
  259. browser_options.hide_ext = false;
  260. bool ret = dialog_file_browser_show(dialogs, path, path, &browser_options);
  261. furi_record_close("dialogs");
  262. if(ret) {
  263. if(!file_stream_open(stream, furi_string_get_cstr(path), FSAM_READ, FSOM_OPEN_EXISTING)) {
  264. FURI_LOG_D(TAG, "Cannot open file \"%s\"", furi_string_get_cstr(path));
  265. file_stream_close(stream);
  266. } else {
  267. FURI_LOG_D(TAG, "Open file \"%s\"", furi_string_get_cstr(path));
  268. strncpy(addr_file_name, furi_string_get_cstr(path) + sizeof(SCAN_APP_PATH_FOLDER), sizeof(addr_file_name));
  269. result = true;
  270. }
  271. }
  272. furi_string_free(path);
  273. return result;
  274. }
  275. // 0 - success, otherwise an error
  276. static uint8_t load_settings_file(Stream* file_stream) {
  277. size_t file_size = 0;
  278. char* file_buf;
  279. uint8_t err = 5;
  280. file_size = stream_size(file_stream);
  281. if(file_size == (size_t)0) {
  282. FURI_LOG_D(TAG, "load failed. file_size: %d", file_size);
  283. return 1;
  284. }
  285. file_size = MIN(file_size, (size_t) LOG_REC_SIZE * MAX_LOG_RECORDS * 2 + 100);
  286. file_buf = malloc(file_size + 1);
  287. if(file_buf == NULL) {
  288. FURI_LOG_D(TAG, "Memory low, need: %d", file_size);
  289. return 2;
  290. }
  291. memset(file_buf, 0, file_size + 1);
  292. if(stream_read(file_stream, (uint8_t*)file_buf, file_size) == file_size) {
  293. FURI_LOG_D(TAG, "Loading settings file");
  294. char* line_ptr = file_buf;
  295. int16_t line_num = 0;
  296. what_to_do = 2;
  297. sniff_loaded = 0;
  298. bool log_loaded = false;
  299. while(line_ptr && (size_t)(line_ptr - file_buf) < file_size) {
  300. char* end_ptr = strstr((char*)line_ptr, "\n");
  301. if(end_ptr == NULL) end_ptr = file_buf + file_size; else *end_ptr = '\0';
  302. int line_len = end_ptr - line_ptr;
  303. if(*line_ptr == '\r' || line_len == 0) {
  304. line_ptr = end_ptr + 1;
  305. continue;
  306. }
  307. if(*(end_ptr - 1) < '0') {
  308. *(end_ptr - 1) = '\0';
  309. line_len--;
  310. }
  311. //FURI_LOG_D(TAG, " L#%d: [%d]%s", line_num, line_len, line_ptr);
  312. if(strncmp(line_ptr, SettingsFld_Rate, sizeof(SettingsFld_Rate)-1) == 0) {
  313. NRF_rate = atoi(line_ptr + sizeof(SettingsFld_Rate));
  314. } else if(strncmp(line_ptr, SettingsFld_Ch, sizeof(SettingsFld_Ch)-1) == 0) {
  315. NRF_channel = atoi(line_ptr + sizeof(SettingsFld_Ch));
  316. } else if(strncmp(line_ptr, SettingsFld_ESB, sizeof(SettingsFld_ESB)-1) == 0) {
  317. NRF_ESB = atoi(line_ptr + sizeof(SettingsFld_ESB));
  318. } else if(strncmp(line_ptr, SettingsFld_DPL, sizeof(SettingsFld_DPL)-1) == 0) {
  319. NRF_DPL = atoi(line_ptr + sizeof(SettingsFld_DPL));
  320. } else if(strncmp(line_ptr, SettingsFld_CRC, sizeof(SettingsFld_CRC)-1) == 0) {
  321. NRF_CRC = atoi(line_ptr + sizeof(SettingsFld_CRC));
  322. if(what_to_do == 1) view_log_decode_CRC = NRF_CRC;
  323. } else if(strncmp(line_ptr, SettingsFld_Payload, sizeof(SettingsFld_Payload)-1) == 0) {
  324. uint8_t pld = atoi(line_ptr + sizeof(SettingsFld_Payload));
  325. if(pld > 32) pld = 32;
  326. if(sniff_loaded) {
  327. NRF_Payload_sniff_min = pld;
  328. } else {
  329. if(pld == 0) pld = 32;
  330. NRF_Payload = pld;
  331. }
  332. } else if(strncmp(line_ptr, SettingsFld_Sniff, sizeof(SettingsFld_Sniff)-1) == 0) {
  333. what_to_do = 1;
  334. sniff_loaded = 1;
  335. } else if(*line_ptr == SettingsFld_Addr) {
  336. char a = *(++line_ptr);
  337. struct ADDRS *adr = sniff_loaded ? &addrs_sniff : &addrs;
  338. line_ptr += 3;
  339. switch(a) {
  340. case '0':
  341. memset(adr, 0, sizeof(addrs));
  342. adr->addr_len = ConvertHexToArray(line_ptr, adr->addr_P0, sniff_loaded ? 3 : 5);
  343. if(adr->addr_len >= 2) err = 0;
  344. break;
  345. case '1':
  346. ConvertHexToArray(line_ptr, adr->addr_P1, what_to_do == 1 ? 3 : 5);
  347. break;
  348. case '2':
  349. ConvertHexToArray(line_ptr, &adr->addr_P2, 1);
  350. break;
  351. case '3':
  352. ConvertHexToArray(line_ptr, &adr->addr_P3, 1);
  353. break;
  354. case '4':
  355. ConvertHexToArray(line_ptr, &adr->addr_P4, 1);
  356. break;
  357. case '5':
  358. ConvertHexToArray(line_ptr, &adr->addr_P5, 1);
  359. break;
  360. default:
  361. a = 0;
  362. break;
  363. }
  364. if(err == 0 && a) adr->addr_count = a - '0' + 1;
  365. } else if(line_len >= 3 * 2) { // data
  366. if(!log_loaded) {
  367. clear_log();
  368. what_to_do = 0;
  369. log_loaded = true;
  370. }
  371. if(log_arr_idx < MAX_LOG_RECORDS - 1) {
  372. ConvertHexToArray(line_ptr, APP->log_arr + log_arr_idx * LOG_REC_SIZE, LOG_REC_SIZE);
  373. log_arr_idx++;
  374. }
  375. }
  376. line_ptr = end_ptr + 1;
  377. line_num++;
  378. }
  379. } else {
  380. FURI_LOG_D(TAG, "load failed. file size: %d", file_size);
  381. err = 4;
  382. }
  383. free(file_buf);
  384. return err;
  385. }
  386. static void input_callback(InputEvent* input_event, FuriMessageQueue* event_queue) {
  387. furi_assert(event_queue);
  388. PluginEvent event = {.type = EventTypeKey, .input = *input_event};
  389. furi_message_queue_put(event_queue, &event, FuriWaitForever);
  390. }
  391. static void prepare_nrf24(bool fsend_packet)
  392. {
  393. nrf24_write_reg(nrf24_HANDLE, REG_STATUS, 0x70); // clear interrupts
  394. nrf24_write_reg(nrf24_HANDLE, REG_RF_SETUP, NRF_rate);
  395. uint8_t erx_addr = (1<<0); // Enable RX_P0
  396. struct ADDRS *adr = what_to_do == 1 ? &addrs_sniff : &addrs;
  397. if(!fsend_packet) {
  398. if(adr->addr_count == 0) return;
  399. uint8_t payload = NRF_Payload;
  400. if(what_to_do == 1) { // SNIFF
  401. //payload += 5 + NRF_CRC; // + addr_max + CRC
  402. //if(NRF_ESB) payload += 2;
  403. //if(payload > 32) payload = 32;
  404. payload = 32;
  405. nrf24_write_reg(nrf24_HANDLE, REG_CONFIG, 0x70); // Mask all interrupts
  406. nrf24_write_reg(nrf24_HANDLE, REG_SETUP_RETR, 0); // Automatic Retransmission
  407. nrf24_write_reg(nrf24_HANDLE, REG_EN_AA, 0); // Auto acknowledgement
  408. nrf24_write_reg(nrf24_HANDLE, REG_FEATURE, 0); // Enables the W_TX_PAYLOAD_NOACK command, Disable Payload with ACK, set Dynamic Payload
  409. } else {
  410. nrf24_write_reg(nrf24_HANDLE, REG_CONFIG, 0x70 | ((NRF_CRC == 1 ? 0b1000 : NRF_CRC == 2 ? 0b1100 : 0))); // Mask all interrupts
  411. nrf24_write_reg(nrf24_HANDLE, REG_SETUP_RETR, NRF_ESB ? 0x11 : 0); // Automatic Retransmission
  412. nrf24_write_reg(nrf24_HANDLE, REG_EN_AA, NRF_AA_OFF || !NRF_ESB ? 0 : 0x3F); // Auto acknowledgement
  413. nrf24_write_reg(nrf24_HANDLE, REG_FEATURE, NRF_DPL ? 4+1 : 1); // Enables the W_TX_PAYLOAD_NOACK command, Disable Payload with ACK, set Dynamic Payload
  414. }
  415. nrf24_set_maclen(nrf24_HANDLE, adr->addr_len);
  416. nrf24_set_mac(REG_RX_ADDR_P0, adr->addr_P0, adr->addr_len);
  417. uint8_t tmp[5] = { 0 };
  418. nrf24_read_reg(nrf24_HANDLE, REG_RX_ADDR_P0, tmp, adr->addr_len);
  419. for(uint8_t i = 0; i < adr->addr_len / 2; i++) {
  420. uint8_t tb = tmp[i];
  421. tmp[i] = tmp[adr->addr_len - i - 1];
  422. tmp[adr->addr_len - i - 1] = tb;
  423. }
  424. NRF_ERROR = memcmp(adr->addr_P0, tmp, adr->addr_len) != 0;
  425. FURI_LOG_D(TAG, "Payload: %d", payload);
  426. nrf24_write_reg(nrf24_HANDLE, RX_PW_P0, payload);
  427. if(adr->addr_count > 1) {
  428. nrf24_set_mac(REG_RX_ADDR_P1, adr->addr_P1, adr->addr_len);
  429. nrf24_write_reg(nrf24_HANDLE, RX_PW_P1, payload);
  430. erx_addr |= (1<<1); // Enable RX_P1
  431. } else nrf24_write_reg(nrf24_HANDLE, RX_PW_P1, 0);
  432. if(adr->addr_count > 2) {
  433. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P2, &adr->addr_P2, 1);
  434. nrf24_write_reg(nrf24_HANDLE, RX_PW_P2, payload);
  435. erx_addr |= (1<<2); // Enable RX_P2
  436. } else nrf24_write_reg(nrf24_HANDLE, RX_PW_P2, 0);
  437. if(adr->addr_count > 3) {
  438. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P3, &adr->addr_P3, 1);
  439. nrf24_write_reg(nrf24_HANDLE, RX_PW_P3, payload);
  440. erx_addr |= (1<<3); // Enable RX_P3
  441. } else nrf24_write_reg(nrf24_HANDLE, RX_PW_P3, 0);
  442. if(adr->addr_count > 4) {
  443. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P4, &adr->addr_P4, 1);
  444. nrf24_write_reg(nrf24_HANDLE, RX_PW_P4, payload);
  445. erx_addr |= (1<<4); // Enable RX_P4
  446. } else nrf24_write_reg(nrf24_HANDLE, RX_PW_P4, 0);
  447. if(adr->addr_count > 5) {
  448. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P5, &adr->addr_P5, 1);
  449. nrf24_write_reg(nrf24_HANDLE, RX_PW_P5, payload);
  450. erx_addr |= (1<<5); // Enable RX_P5
  451. } else nrf24_write_reg(nrf24_HANDLE, RX_PW_P5, 0);
  452. nrf24_write_reg(nrf24_HANDLE, REG_DYNPD, NRF_DPL ? 0x3F : 0); // Enable dynamic payload reg
  453. nrf24_write_reg(nrf24_HANDLE, REG_EN_RXADDR, erx_addr);
  454. nrf24_write_reg(nrf24_HANDLE, REG_RF_CH, NRF_channel);
  455. }
  456. nrf24_flush_rx(nrf24_HANDLE);
  457. nrf24_flush_tx(nrf24_HANDLE);
  458. nrf24_set_idle(nrf24_HANDLE);
  459. }
  460. void correct_NRF_Payload_sniff_min()
  461. {
  462. uint8_t pld = 32 - 3 - (NRF_ESB ? 2 : 0) - NRF_CRC + (addrs_sniff.addr_len - 2);
  463. if(NRF_Payload_sniff_min > pld) NRF_Payload_sniff_min = pld;
  464. }
  465. static void start_scanning()
  466. {
  467. FURI_LOG_D(TAG, "Start proc-%d: Ch=%d Rate=%d", what_to_do, NRF_channel, NRF_rate);
  468. if(what_to_do == 1) { // SNIFF
  469. correct_NRF_Payload_sniff_min();
  470. view_log_decode_CRC = NRF_CRC;
  471. } else if(sniff_loaded) { // Switch from sniff to scan/view
  472. }
  473. prepare_nrf24(false);
  474. if(NRF_ERROR) {
  475. FURI_LOG_E(TAG, "NRF R/W ERROR!");
  476. return;
  477. }
  478. nrf24_set_rx_mode(nrf24_HANDLE);
  479. start_time = furi_get_tick();
  480. }
  481. /*
  482. bool check_addr_found(uint8_t *pkt)
  483. {
  484. uint8_t idx = 255;
  485. if(addrs_found.addr_count > 0 && memcmp(addrs_found.addr_P0, pkt, addrs_found.addr_len) == 0) { idx = 0; goto x_end; }
  486. if(addrs_found.addr_count > 1 && memcmp(addrs_found.addr_P1, pkt, addrs_found.addr_len - 1) == 0) {
  487. if(addrs_found.addr_P1[addrs_found.addr_len - 1] == pkt[addrs_found.addr_len - 1]) { idx = 1; goto x_end; }
  488. if(addrs_found.addr_count > 2 && addrs_found.addr_P2 == pkt[addrs_found.addr_len - 1]) { idx = 2; goto x_end; }
  489. if(addrs_found.addr_count > 3 && addrs_found.addr_P3 == pkt[addrs_found.addr_len - 1]) { idx = 3; goto x_end; }
  490. if(addrs_found.addr_count > 4 && addrs_found.addr_P4 == pkt[addrs_found.addr_len - 1]) { idx = 4; goto x_end; }
  491. if(addrs_found.addr_count > 5 && addrs_found.addr_P5 == pkt[addrs_found.addr_len - 1]) { idx = 5; goto x_end; }
  492. }
  493. x_end:
  494. if(idx < sizeof(found_total) / sizeof(found_total[0])) {
  495. found_total[idx]++;
  496. return true;
  497. } else return false;
  498. }
  499. */
  500. // start bitnum = 7
  501. uint32_t calc_crc(uint32_t crc, uint8_t *ptr, uint8_t bitnum, uint16_t bits)
  502. {
  503. //uint8_t bitnum = 7;
  504. uint32_t crc_high, polynom;
  505. if(view_log_decode_CRC == 2) {
  506. crc_high = (1<<16);
  507. polynom = 0x1021; // X^16+X^12+X^5+1 => 0x11021 & 0xFFFF = 0x1021
  508. } else {
  509. crc_high = (1<<8);
  510. polynom = 0x07; // x^8+x^2+x^1+1 => 0x107 & 0xFF = 0x07
  511. }
  512. while(bits--) {
  513. crc <<= 1;
  514. if(((crc & crc_high) != 0) ^ ((*ptr >> bitnum) & 1)) crc ^= polynom;
  515. if(bitnum == 0) {
  516. ptr++;
  517. bitnum = 7;
  518. } else bitnum--;
  519. }
  520. return crc & (view_log_decode_CRC == 2 ? 0xFFFF : 0xFF);
  521. }
  522. // shifted 1 bit right
  523. uint32_t get_shifted_crc(uint8_t *pcrc)
  524. {
  525. uint32_t crc = ((*pcrc << 1) & 0xFF) | (*(pcrc+1) >> 7);
  526. if(view_log_decode_CRC == 2) {
  527. crc = (crc << 8) | ((*(pcrc+1) << 1) & 0xFF) | (*(pcrc+2) >> 7);
  528. }
  529. return crc;
  530. }
  531. bool check_packet(uint8_t *pkt, uint16_t size)
  532. {
  533. if(size < 3 || size > 32) return false;
  534. uint8_t b = *pkt;
  535. if(b == 0x55 || b == 0xAA || b == 0x00 || b == 0xFF) return false; // skip pkt when address begin with
  536. uint32_t prevcrc = 0xFFFFFFFF;
  537. bool found = false;
  538. uint8_t addr_size = 3;
  539. for(; addr_size <= 5; addr_size++) {
  540. if(NRF_ESB){
  541. uint8_t _payload = *(pkt + addr_size) >> 2;
  542. if((_payload > size - addr_size - 2 - view_log_decode_CRC && _payload != 0x33)) continue;
  543. uint8_t *p = pkt + addr_size;
  544. if(prevcrc == 0xFFFFFFFF) {
  545. prevcrc = calc_crc(view_log_decode_CRC == 2 ? 0xFFFF : 0xFF, pkt, 7, 3 * 8); // crc for smallest addr
  546. }
  547. uint32_t crc;
  548. if(addr_size > 3) crc = calc_crc(prevcrc, p - (addr_size - 3), 7, 8 * (addr_size - 3)); else crc = prevcrc;
  549. if(_payload != 0x33) { // DPL
  550. crc = calc_crc(crc, p, 7, 9 + _payload * 8);
  551. if(crc == get_shifted_crc(p + _payload + 1)) {
  552. *(pkt - 1) = ((_payload & 0x1F) << 3) + 0b100 + (addr_size - 2);
  553. FURI_LOG_D(TAG, "VALID CRC %X: dpl: %d, addr: %d", (uint16_t)crc, _payload, addr_size);
  554. found = true;
  555. break;
  556. }
  557. } else {
  558. crc = calc_crc(crc, p++, 7, 9); // PCF
  559. if(crc == get_shifted_crc(p)) {
  560. _payload = 0;
  561. found = true;
  562. } else {
  563. for(uint8_t i = 1; i < size - addr_size - view_log_decode_CRC; i++) {
  564. crc = calc_crc(crc, p++, 6, 8);
  565. if(crc == get_shifted_crc(p)) {
  566. _payload = i;
  567. found = true;
  568. break;
  569. }
  570. }
  571. }
  572. if(found) {
  573. *(pkt - 1) = ((_payload & 0x1F) << 3) + 0b100 + (addr_size - 2);
  574. FURI_LOG_D(TAG, "VALID CRC %X: pl: %d, addr: %d", (uint16_t)crc, _payload, addr_size);
  575. break;
  576. }
  577. }
  578. } else {
  579. uint8_t *p;
  580. if(addr_size == 3) {
  581. prevcrc = calc_crc(view_log_decode_CRC == 2 ? 0xFFFF : 0xFF, pkt, 7, 3 * 8); // crc for smallest addr
  582. p = pkt + addr_size;
  583. } else {
  584. p = pkt + addr_size - 1;
  585. prevcrc = calc_crc(prevcrc, p++, 7, 8);
  586. }
  587. uint32_t crc = prevcrc;
  588. if((view_log_decode_CRC == 1 && crc == *p) || (view_log_decode_CRC == 2 && crc == (uint32_t)((*p<<8) | *(p+1)))) {
  589. *(pkt - 1) = ((0 & 0x1F) << 3) + 0b000 + (addr_size - 2);
  590. FURI_LOG_D(TAG, "VALID CRC %X: pl: %d, addr: %d", (uint16_t)crc, 0, addr_size);
  591. found = true;
  592. break;
  593. }
  594. for(uint8_t i = 1; i <= size - addr_size - view_log_decode_CRC; i++) {
  595. crc = calc_crc(crc, p++, 7, 8);
  596. if((view_log_decode_CRC == 1 && crc == *p) || (view_log_decode_CRC == 2 && crc == (uint32_t)((*p<<8) | *(p+1)))) {
  597. *(pkt - 1) = ((i & 0x1F) << 3) + 0b000 + (addr_size - 2);
  598. FURI_LOG_D(TAG, "VALID CRC %X: pl: %d, addr: %d", (uint16_t)crc, i, addr_size);
  599. found = true;
  600. break;
  601. }
  602. }
  603. if(found) break;
  604. }
  605. }
  606. if(found) {
  607. if(furi_log_get_level() == FuriLogLevelDebug) {
  608. char dbuf[65];
  609. dbuf[0] = 0;
  610. add_to_str_hex_bytes(dbuf, (char*)pkt, size);
  611. FURI_LOG_D(TAG, "PKT%02X: %s (%d)", *(pkt - 1), dbuf, size);
  612. }
  613. int16_t i = 0;
  614. for(; i < found_total; i++) {
  615. if(APP->found[i].addr_size != addr_size) continue;
  616. if(memcmp(APP->found[i].addr, pkt, addr_size) == 0) break;
  617. }
  618. if(i != found_total) { // found
  619. APP->found[i].total++;
  620. } else {
  621. pkt--;
  622. uint8_t *p = APP->log_arr;
  623. for(i = 0; i < log_arr_idx; i++, p += LOG_REC_SIZE) {
  624. if(*p & 0x80 && (*(p+1) & 0b11) + 2 == addr_size && pkt != p) {
  625. if(memcmp(p, pkt, addr_size) == 0) break;
  626. }
  627. }
  628. if(i != log_arr_idx && found_total < MAX_LOG_RECORDS) { // found -> 2
  629. memset(&APP->found[found_total], 0, sizeof(struct FOUND));
  630. memcpy(APP->found[found_total].addr, pkt, addr_size);
  631. APP->found[found_total].addr_size = addr_size;
  632. APP->found[found_total].total = 2;
  633. if(found_total == 0) view_found = 0;
  634. found_total++;
  635. }
  636. }
  637. /*
  638. if(addrs_found.addr_count == 0) {
  639. memcpy(addrs_found.addr_P0, pkt, addr_size);
  640. addrs_found.addr_len = addr_size;
  641. found_total[0]++;
  642. addrs_found.addr_count++;
  643. } else if(addr_size == addrs_found.addr_len) {
  644. if(!check_addr_found(pkt)) {
  645. if(addrs_found.addr_count == 1) {
  646. memcpy(addrs_found.addr_P1, pkt, addr_size);
  647. found_total[1]++;
  648. addrs_found.addr_count++;
  649. } else if(addrs_found.addr_count == 2) {
  650. if(memcmp(addrs_found.addr_P1, pkt, addr_size - 1) == 0) {
  651. addrs_found.addr_P2 = pkt[addr_size - 1];
  652. found_total[2]++;
  653. addrs_found.addr_count++;
  654. } else if(memcmp(addrs_found.addr_P0, pkt, addr_size - 1) == 0) {
  655. uint8_t tmp[5];
  656. memcpy(tmp, addrs_found.addr_P1, addr_size); // swap P0-P1
  657. memcpy(addrs_found.addr_P1, addrs_found.addr_P0, addr_size);
  658. memcpy(addrs_found.addr_P0, tmp, addr_size);
  659. uint32_t n = found_total[0];
  660. found_total[0] = found_total[1];
  661. found_total[1] = n;
  662. addrs_found.addr_P2 = pkt[addr_size - 1];
  663. found_total[2]++;
  664. addrs_found.addr_count++;
  665. }
  666. } else if(addrs_found.addr_count >= 3) {
  667. if(memcmp(addrs_found.addr_P1, pkt, addr_size - 1) == 0) {
  668. if(addrs_found.addr_count == 3) {
  669. addrs_found.addr_P3 = pkt[addr_size - 1];
  670. found_total[3]++;
  671. addrs_found.addr_count++;
  672. } else if(addrs_found.addr_count == 4) {
  673. addrs_found.addr_P4 = pkt[addr_size - 1];
  674. found_total[4]++;
  675. addrs_found.addr_count++;
  676. } else if(addrs_found.addr_count == 5) {
  677. addrs_found.addr_P5 = pkt[addr_size - 1];
  678. found_total[5]++;
  679. addrs_found.addr_count++;
  680. }
  681. }
  682. }
  683. }
  684. }
  685. */
  686. }
  687. return found;
  688. }
  689. bool nrf24_read_newpacket() {
  690. if(APP->log_arr == NULL) return false;
  691. bool found = false;
  692. uint8_t packetsize;
  693. uint8_t *ptr = APP->log_arr + log_arr_idx * LOG_REC_SIZE;
  694. uint8_t st;
  695. /* test pkts
  696. static int iii = 0;
  697. char ppp[][65] = { "54A545109411544BAAE50110A3282512A9A1152A565B22AAA48AB751A5",
  698. "C8C8C0CE7A81018007202FFFFC",
  699. "EAEC8C8C2CE3C0101006FB737A",
  700. "BEBFFFEC8C8C1CC00542AF7CFF7DBEAFE3397FEAFEF1DDFA4AEF7FDBB7CDEABC",
  701. "FEAAAABEAAFEAAC8C8C28E1C810080490ABAF7FEEB76B7FDFEF7DFFB47FB97FE",
  702. "A8AAC8C8C1CE20163DF7DFFD00",
  703. "AFFEEFEC8C8C2CE4001010062F037F9BFFDF1DAD5EDBEF55DD9AB535FCB67F55",
  704. "AC8C8C1CE5F8102000D503D7ABF",
  705. "EE03080B4712555555550E80",
  706. "C8C8C41385818280127100",
  707. "AAC8C8C3CE05818280119000"
  708. "AC8C8C413858182801271000",
  709. "AAC8C8C40B0305028542"
  710. };
  711. if(iii < 13) {
  712. ConvertHexToArray(ppp[iii], ptr + 2, 32);
  713. st = RX_DR;
  714. packetsize = 32;
  715. iii++;
  716. } else
  717. //*/
  718. st = nrf24_rxpacket(nrf24_HANDLE, ptr + 2 + (what_to_do == 1 ? addrs_sniff.addr_len - 2 : 0), &packetsize, what_to_do == 1 ? 32 : !NRF_DPL);
  719. if(st & RX_DR) {
  720. st = (st >> 1) & 7; // pipe #
  721. if(what_to_do == 1) { // SNIFF
  722. *ptr++ = NRF_channel | 0x80;
  723. *ptr++ = st; // pipe #
  724. if(addrs_sniff.addr_len > 2) {
  725. *ptr = st == 0 ? addrs_sniff.addr_P0[2] : st == 1 ? addrs_sniff.addr_P1[2] : st == 2 ? addrs_sniff.addr_P2 : st == 3 ? addrs_sniff.addr_P3 : st == 4 ? addrs_sniff.addr_P4 : addrs_sniff.addr_P5;
  726. }
  727. if(!check_packet(ptr, packetsize)) {
  728. if(addrs_sniff.addr_len > 2) return false; // skip if mac MSB added to preamble
  729. uint8_t shifted = 0;
  730. uint8_t shift_max = (32 - 3 - NRF_Payload_sniff_min - NRF_CRC) * 8 - 1;
  731. while(shifted++ < shift_max) { // Shift packet left by one bit if minimum payload fits
  732. uint8_t i = 0;
  733. for(; i < packetsize - 1; i++) ptr[i] = (ptr[i] << 1) | (ptr[i + 1] >> 7);
  734. ptr[i] <<= 1;
  735. if(check_packet(ptr, packetsize - (shifted >> 3) - 1)) goto x_valid;
  736. }
  737. return false;
  738. }
  739. } else {
  740. *ptr++ = NRF_channel;
  741. *ptr++ = ((packetsize & 0x1F) << 3) | st; // payload size + pipe #
  742. }
  743. x_valid:
  744. if(packetsize < 32) memset(ptr + packetsize, 0, 32 - packetsize);
  745. if(log_arr_idx < MAX_LOG_RECORDS - 1) {
  746. log_arr_idx++;
  747. } else {
  748. if(log_to_file == 1 || log_to_file == 2) {
  749. write_to_log_file(APP->storage, false);
  750. clear_log();
  751. } else {
  752. memmove(APP->log_arr, APP->log_arr + LOG_REC_SIZE, log_arr_idx * LOG_REC_SIZE);
  753. }
  754. }
  755. FURI_LOG_D(TAG, "Found packet #%d pipe %d", log_arr_idx, st);
  756. notification_message(APP->notification, &sequence_blink_white_100);
  757. found = true;
  758. }
  759. return found;
  760. }
  761. bool nrf24_send_packet()
  762. {
  763. if(log_arr_idx == 0) return false;
  764. prepare_nrf24(!what_to_do);
  765. uint8_t *ptr = APP->log_arr + view_log_arr_idx * LOG_REC_SIZE;
  766. nrf24_write_reg(nrf24_HANDLE, REG_RF_CH, *ptr & 0x7F);
  767. if(*ptr & 0x80) { // RAW packet
  768. //uint8_t pktinfo = *(ptr + 1);
  769. //nrf24_set_maclen(nrf24_HANDLE, (pktinfo & 0b11) + 2);
  770. //if(pktinfo & 0b100) { // ESB
  771. nrf24_write_reg(nrf24_HANDLE, REG_SETUP_RETR, 0); // No Automatic Retransmission
  772. nrf24_write_reg(nrf24_HANDLE, REG_EN_AA, 0); // No Auto acknowledgement
  773. //}
  774. //uint8_t alen = (*(ptr + 2) & 0b11) + 2;
  775. uint8_t adr[2];
  776. adr[0] = ptr[2];
  777. adr[1] = ptr[3];
  778. nrf24_set_maclen(nrf24_HANDLE, 2);
  779. nrf24_set_mac(REG_RX_ADDR_P0, adr, 2);
  780. nrf24_set_mac(REG_TX_ADDR, adr, 2);
  781. last_packet_send_st = nrf24_txpacket(nrf24_HANDLE, ptr + 2 + 2, 32 - 2, false);
  782. } else {
  783. nrf24_write_reg(nrf24_HANDLE, REG_SETUP_RETR, NRF_ESB ? 0x11 : 0); // Automatic Retransmission
  784. nrf24_write_reg(nrf24_HANDLE, REG_EN_AA, NRF_AA_OFF || !NRF_ESB ? 0 : 0x3F); // Auto acknowledgement
  785. uint8_t *adr;
  786. uint8_t a = *(ptr + 1) & 0b111;
  787. if(a < 2) {
  788. if(a == 0) adr = addrs.addr_P0; else adr = addrs.addr_P1;
  789. nrf24_set_mac(REG_RX_ADDR_P0, adr, addrs.addr_len);
  790. nrf24_set_mac(REG_TX_ADDR, adr, addrs.addr_len);
  791. } else {
  792. uint8_t buf[5];
  793. memcpy(buf, addrs.addr_P1, addrs.addr_len - 1);
  794. buf[addrs.addr_len - 1] = a == 2 ? addrs.addr_P2 : a == 3 ? addrs.addr_P3 : a == 4 ? addrs.addr_P4 : addrs.addr_P5;
  795. nrf24_set_mac(REG_RX_ADDR_P0, buf, addrs.addr_len);
  796. nrf24_set_mac(REG_TX_ADDR, buf, addrs.addr_len);
  797. }
  798. a = *(ptr + 1) >> 3;
  799. if(a == 0) a = 32;
  800. nrf24_write_reg(nrf24_HANDLE, REG_CONFIG, 0x70 | ((NRF_CRC == 1 ? 0b1000 : NRF_CRC == 2 ? 0b1100 : 0))); // Mask all interrupts
  801. nrf24_write_reg(nrf24_HANDLE, REG_DYNPD, NRF_DPL ? 0x3F : 0); // Enable dynamic payload reg
  802. last_packet_send_st = nrf24_txpacket(nrf24_HANDLE, ptr + 2, a, false);
  803. }
  804. last_packet_send = view_log_arr_idx;
  805. notification_message(APP->notification, last_packet_send_st ? &sequence_blink_blue_100 : &sequence_blink_red_100);
  806. if(what_to_do) start_scanning();
  807. return last_packet_send_st;
  808. }
  809. static void render_callback(Canvas* const canvas, void* ctx) {
  810. const PluginState* plugin_state = acquire_mutex((ValueMutex*)ctx, 25);
  811. if(plugin_state == NULL) return;
  812. //canvas_draw_frame(canvas, 0, 0, 128, 64); // border around the edge of the screen
  813. if(what_doing == 0) {
  814. canvas_set_font(canvas, FontSecondary); // 8x10 font, 6 lines
  815. if(save_settings) snprintf(screen_buf, sizeof(screen_buf), "Save: %s", SETTINGS_FILENAME); // menu_selected = 0
  816. else snprintf(screen_buf, sizeof(screen_buf), "Load: %s", addr_file_name);
  817. canvas_draw_str(canvas, 10, 10, screen_buf);
  818. snprintf(screen_buf, sizeof(screen_buf), "Ch: %d", NRF_channel); // menu_selected = 1
  819. canvas_draw_str(canvas, 10, 20, screen_buf);
  820. if(NRF_ESB) {
  821. strcpy(screen_buf, "ESB");
  822. if(NRF_DPL) strcat(screen_buf, " DPL");
  823. canvas_draw_str(canvas, 78, 20, screen_buf);
  824. }
  825. if(NRF_AA_OFF) { canvas_draw_str(canvas, 61, 20, "AA"); canvas_draw_line(canvas, 60, 21, 72, 12); }
  826. snprintf(screen_buf, sizeof(screen_buf), "Rate: %sbps", NRF_rate == 2 ? "2M" : NRF_rate == 1 ? "1M" : "250K"); // menu_selected = 2
  827. canvas_draw_str(canvas, 10, 30, screen_buf);
  828. if(what_to_do == 1) snprintf(screen_buf, sizeof(screen_buf), "Min Payl: %d", NRF_Payload_sniff_min);
  829. else snprintf(screen_buf, sizeof(screen_buf), "Payload: %d", NRF_Payload);
  830. canvas_draw_str(canvas, 78, 30, screen_buf);
  831. strcpy(screen_buf, "Next Ch time: "); // menu_selected = 3
  832. if(find_channel_period == 0) strcat(screen_buf, "off"); else snprintf(screen_buf + strlen(screen_buf), sizeof(screen_buf), "%d s", find_channel_period);
  833. canvas_draw_str(canvas, 10, 40, screen_buf);
  834. if(NRF_CRC == 1) canvas_draw_str(canvas, 99, 40, "CRC1");
  835. else if(NRF_CRC == 2) canvas_draw_str(canvas, 99, 40, "CRC2");
  836. snprintf(screen_buf, sizeof(screen_buf), "Log: %s", log_to_file == 0 ? "No" : log_to_file == 1 ? "Yes" : log_to_file == 2 ? "Append" : "Clear");// menu_selected = 4
  837. canvas_draw_str(canvas, 10, 50, screen_buf);
  838. if(what_to_do) { // menu_selected = 5
  839. if(NRF_ERROR) snprintf(screen_buf, sizeof(screen_buf), "nRF24L01+ R/W ERROR!");
  840. else {
  841. if(what_to_do == 1) snprintf(screen_buf, sizeof(screen_buf), "Start sniff");
  842. else {
  843. uint8_t *p;
  844. if(what_to_do == 3 && log_arr_idx && *(p = APP->log_arr + view_log_arr_idx * LOG_REC_SIZE) & 0x80) { // +RAW
  845. snprintf(screen_buf, sizeof(screen_buf), "Start read: ");
  846. add_to_str_hex_bytes(screen_buf, (char*)p + 2, (*(p + 1) & 0b11) + 2);
  847. } else snprintf(screen_buf, sizeof(screen_buf), "Start scan (pipes: %d)", addrs.addr_count);
  848. }
  849. }
  850. } else snprintf(screen_buf, sizeof(screen_buf), "View log (pipes: %d)", addrs.addr_count);
  851. canvas_draw_str(canvas, 10, 60, screen_buf);
  852. canvas_draw_str(canvas, 0, menu_selected * 10 + 10, ">");
  853. } else if(what_doing == 1){
  854. canvas_set_font(canvas, FontBatteryPercent); // 5x7 font, 9 lines
  855. bool ch2 = false;
  856. screen_buf[0] = '\0';
  857. if(view_log_arr_x == 0) { strcat(screen_buf, " "); ch2 = true;
  858. } else {
  859. snprintf(screen_buf, sizeof(screen_buf), "<%d", view_log_arr_x);
  860. if(view_log_arr_x < VIEW_LOG_MAX_X) ch2 = true;
  861. }
  862. snprintf(screen_buf + strlen(screen_buf), sizeof(screen_buf), " %s ch: %d - %d.", what_to_do == 1 ? "Sniff" : what_to_do == 0 ? "View" : "Read", NRF_channel, log_arr_idx);
  863. canvas_draw_str(canvas, 0, 7, screen_buf);
  864. if(ch2) canvas_draw_str(canvas, 121, 7, ">");
  865. if(log_arr_idx) {
  866. if(view_log_arr_idx >= log_arr_idx) view_log_arr_idx = log_arr_idx - 1;
  867. uint16_t page = view_log_arr_idx & ~7;
  868. for(uint8_t i = 0; i < 8 && page + i < log_arr_idx; i++) {
  869. screen_buf[0] = (view_log_arr_idx & 7) != i ? ' ' : last_packet_send != view_log_arr_idx ? '>' : last_packet_send_st ? '*' : '!';
  870. screen_buf[1] = '\0';
  871. uint8_t *ptr = APP->log_arr + (page + i) * LOG_REC_SIZE;
  872. uint8_t channel = *ptr++;
  873. uint8_t *crcptr = NULL;
  874. uint8_t pre = 0;
  875. int count = 0;
  876. if(channel & 0x80) { // RAW packet: nn:>{.address..}-xxxxxxxx
  877. uint8_t pktinfo = *ptr++;
  878. bool _PCF = pktinfo & 0b100;
  879. uint8_t plen = count = (pktinfo >> 3);
  880. uint8_t adrsize = (pktinfo & 0b11) + 2;
  881. plen += adrsize;
  882. count += view_log_decode_CRC;
  883. if(view_log_arr_x > 0) count -= view_log_arr_x - 1;
  884. uint8_t max_width = VIEW_LOG_WIDTH_B;
  885. if(view_log_arr_x == 0) max_width -= 5;
  886. if(count > max_width) count = max_width;
  887. if(count > 0) {
  888. uint8_t *pcrc = ptr;
  889. uint32_t crc;
  890. crc = view_log_decode_CRC == 2 ? 0xFFFF : 0xFF;
  891. crc = calc_crc(crc, pcrc, 7, (_PCF? 9 : 0) + plen * 8);
  892. pcrc += plen;
  893. if(_PCF) { //ESB
  894. pcrc++;
  895. if(crc == get_shifted_crc(pcrc)) crcptr = pcrc;
  896. } else {
  897. if((view_log_decode_CRC == 1 && crc == *pcrc) || (view_log_decode_CRC == 2 && crc == (uint32_t)((*pcrc<<8) | *(pcrc+1)))) {
  898. crcptr = pcrc;
  899. }
  900. }
  901. if(view_log_arr_x == 0) {
  902. add_to_str_hex_bytes(screen_buf, (char*)ptr, adrsize);
  903. for(int8_t j = 5 - adrsize; j > 0; j--) strcat(screen_buf, " ");
  904. strcat(screen_buf, "-");
  905. pre += 5 * 2 + 1;
  906. } else {
  907. ptr += view_log_arr_x - 1;
  908. }
  909. ptr += adrsize;
  910. if(_PCF) add_to_str_hex_bytes_shift_9b(screen_buf, (char*)ptr++, count); else add_to_str_hex_bytes(screen_buf, (char*)ptr, count);
  911. }
  912. } else {
  913. uint8_t dpl = *ptr++;
  914. uint8_t pipe = dpl & 0b111;
  915. dpl >>= 3;
  916. if(dpl == 0) dpl = 32;
  917. count = dpl - view_log_arr_x;
  918. if(view_log_decode_PCF) count--;
  919. uint8_t max_width = VIEW_LOG_WIDTH_B;
  920. if(view_log_arr_x == 0) {
  921. if(addrs.addr_count > 1) max_width--;
  922. if(view_log_decode_PCF) max_width -= 2;
  923. }
  924. if(count > max_width) count = max_width;
  925. if(count > 0) {
  926. if(view_log_decode_CRC) {
  927. static uint16_t prev_addr_CRC;
  928. static int8_t prev_pipe = -1;
  929. uint8_t *pcrc = ptr;
  930. uint32_t crc;
  931. if(prev_pipe == pipe) { crc = prev_addr_CRC;
  932. } else {
  933. crc = view_log_decode_CRC == 2 ? 0xFFFF : 0xFF;
  934. if(pipe <= 1) { crc = calc_crc(crc, pipe == 0 ? addrs.addr_P0 : addrs.addr_P1, 7, addrs.addr_len * 8);
  935. } else {
  936. crc = calc_crc(crc, addrs.addr_P1, 7, (addrs.addr_len - 1) * 8);
  937. crc = calc_crc(crc, pipe == 2 ? &addrs.addr_P2 : pipe == 3 ? &addrs.addr_P3 : pipe == 4 ? &addrs.addr_P4 : &addrs.addr_P5, 7, 8);
  938. }
  939. prev_addr_CRC = crc;
  940. prev_pipe = pipe;
  941. }
  942. if(view_log_decode_PCF) {
  943. crc = calc_crc(crc, pcrc++, 7, 9);
  944. if(crc == get_shifted_crc(pcrc)) crcptr = pcrc;
  945. if(crcptr == NULL) {
  946. for(int8_t j = 0; j < (int8_t)dpl - view_log_decode_CRC - 1; j++) {
  947. crc = calc_crc(crc, pcrc++, 6, 8);
  948. if(crc == get_shifted_crc(pcrc)) {
  949. crcptr = pcrc;
  950. break;
  951. }
  952. }
  953. }
  954. } else {
  955. for(int8_t j = 0; j < (int8_t)dpl - view_log_decode_CRC; j++) {
  956. crc = calc_crc(crc, pcrc++, 7, 8);
  957. if((view_log_decode_CRC == 1 && crc == *pcrc) || (view_log_decode_CRC == 2 && crc == (uint32_t)((*pcrc<<8) | *(pcrc+1)))) {
  958. crcptr = pcrc;
  959. break;
  960. }
  961. }
  962. }
  963. }
  964. }
  965. ptr += view_log_arr_x;
  966. if(max_width < VIEW_LOG_WIDTH_B) {
  967. pre += snprintf(screen_buf + 1, 10, "%X-", pipe);
  968. if(view_log_decode_PCF) {
  969. pre += snprintf(screen_buf + strlen(screen_buf), 10, "%02X%01X-", *ptr >> 2, ((*ptr & 3) << 1) | (*(ptr+1) >> 7));
  970. }
  971. }
  972. if(view_log_decode_PCF) add_to_str_hex_bytes_shift_9b(screen_buf, (char*)ptr++, count); else add_to_str_hex_bytes(screen_buf, (char*)ptr, count);
  973. }
  974. uint16_t y = 14 + i * 7;
  975. canvas_draw_str(canvas, 3 * 5, y, screen_buf);
  976. uint16_t x = snprintf(screen_buf, sizeof(screen_buf), "%d", page + i + 1);
  977. canvas_draw_str(canvas, 0, y, screen_buf);
  978. if(crcptr) { // 5x7 font, 9 lines
  979. canvas_draw_str(canvas, x * 5, y, "=");
  980. int n = crcptr - (uint8_t*)ptr;
  981. if(n > -view_log_decode_CRC && n < count) {
  982. int len;
  983. x = (4 + pre) * 5;
  984. if(n < 0) {
  985. len = view_log_decode_CRC + n;
  986. n = 0;
  987. } else {
  988. len = MIN(view_log_decode_CRC, count - n);
  989. x += n * 2 * 5;
  990. canvas_draw_line(canvas, x - 1, y, x - 1, y - 1);
  991. }
  992. canvas_draw_line(canvas, x - 1, y, n = x + len * 2 * 5 - 1, y);
  993. canvas_draw_line(canvas, n, y, n, y - 1);
  994. }
  995. } else canvas_draw_str(canvas, x * 5, y, ":");
  996. }
  997. }
  998. } else {
  999. canvas_set_font(canvas, FontBatteryPercent); // 5x7 font, 9 lines
  1000. if(view_found >= 0) {
  1001. snprintf(screen_buf, 50, "Found > 1: %d", found_total);
  1002. canvas_draw_str(canvas, 0, 1 * 7, screen_buf);
  1003. int16_t idx = view_found * 7;
  1004. for(uint8_t i = 0; i < 7; i++) {
  1005. if(idx++ >= found_total) break;
  1006. snprintf(screen_buf, 16, "%d. ", idx);
  1007. add_to_str_hex_bytes(screen_buf, (char*)APP->found[idx].addr, APP->found[idx].addr_size);
  1008. if(APP->found[idx].addr_size == 3) strcat(screen_buf, " "); else if(APP->found[idx].addr_size == 4) strcat(screen_buf, " ");
  1009. snprintf(screen_buf + strlen(screen_buf), 16, " - %d", APP->found[idx].total);
  1010. canvas_draw_str(canvas, 0, 2 + i * 7, screen_buf);
  1011. }
  1012. } else {
  1013. struct ADDRS *a;
  1014. if(what_to_do == 1) {
  1015. a = &addrs_sniff;
  1016. canvas_draw_str(canvas, 0, 1 * 7, "Sniff prefix:");
  1017. } else {
  1018. a = &addrs;
  1019. canvas_draw_str(canvas, 0, 1 * 7, "Addresses:");
  1020. }
  1021. if(a->addr_count > 0) {
  1022. snprintf(screen_buf, sizeof(screen_buf), "P0: ");
  1023. add_to_str_hex_bytes(screen_buf, (char*)a->addr_P0, a->addr_len);
  1024. canvas_draw_str(canvas, 0, 2 * 7, screen_buf);
  1025. }
  1026. if(a->addr_count > 1) {
  1027. snprintf(screen_buf, sizeof(screen_buf), "P1: ");
  1028. add_to_str_hex_bytes(screen_buf, (char*)a->addr_P1, a->addr_len);
  1029. canvas_draw_str(canvas, 0, 3 * 7, screen_buf);
  1030. }
  1031. if(a->addr_count > 2) {
  1032. canvas_draw_str(canvas, 0, 4 * 7, "P2: ");
  1033. snprintf(screen_buf, sizeof(screen_buf), "%02X", a->addr_P2);
  1034. canvas_draw_str(canvas, (4 + (a->addr_len - 1) * 2) * 5, 4 * 7, screen_buf);
  1035. }
  1036. if(a->addr_count > 3) {
  1037. canvas_draw_str(canvas, 0, 5 * 7, "P3: ");
  1038. snprintf(screen_buf, sizeof(screen_buf), "%02X", a->addr_P3);
  1039. canvas_draw_str(canvas, (4 + (a->addr_len - 1) * 2) * 5, 5 * 7, screen_buf);
  1040. }
  1041. if(a->addr_count > 4) {
  1042. canvas_draw_str(canvas, 0, 6 * 7, "P4: ");
  1043. snprintf(screen_buf, sizeof(screen_buf), "%02X", a->addr_P4);
  1044. canvas_draw_str(canvas, (4 + (a->addr_len - 1) * 2) * 5, 6 * 7, screen_buf);
  1045. }
  1046. if(a->addr_count > 5) {
  1047. canvas_draw_str(canvas, 0, 7 * 7, "P5: ");
  1048. snprintf(screen_buf, sizeof(screen_buf), "%02X", a->addr_P5);
  1049. canvas_draw_str(canvas, (4 + (a->addr_len - 1) * 2) * 5, 7 * 7, screen_buf);
  1050. }
  1051. }
  1052. if(log_arr_idx) {
  1053. uint8_t *ptr = APP->log_arr + view_log_arr_idx * LOG_REC_SIZE;
  1054. uint8_t pktinfo = *(ptr + 1);
  1055. snprintf(screen_buf, 32, ">Ch: %d L: %d", *ptr & 0x7F, pktinfo >> 3);
  1056. if(*ptr & 0x80) {
  1057. strcat(screen_buf, " RAW");
  1058. if(pktinfo & 0b100) {
  1059. snprintf(screen_buf + strlen(screen_buf), 16, " ESB %s", *(ptr + 2 + (pktinfo & 0b11) + 2) >> 2 != 0x33 ? "DPL" : "");
  1060. }
  1061. }
  1062. canvas_draw_str(canvas, 0, 8 * 7, screen_buf);
  1063. }
  1064. screen_buf[0] = 'v';
  1065. strcpy(screen_buf + 1, VERSION);
  1066. canvas_draw_str(canvas, 104, 7, screen_buf);
  1067. if(view_log_decode_PCF || view_log_decode_CRC) {
  1068. strcpy(screen_buf, "Decode: ");
  1069. if(view_log_decode_PCF) strcat(screen_buf, "ESB ");
  1070. if(view_log_decode_CRC == 1) strcat(screen_buf, "CRC1");
  1071. else if(view_log_decode_CRC == 2) strcat(screen_buf, "CRC2");
  1072. canvas_draw_str(canvas, 0, 64, screen_buf);
  1073. }
  1074. }
  1075. release_mutex((ValueMutex*)ctx, plugin_state);
  1076. }
  1077. int32_t nrf24scan_app(void* p) {
  1078. UNUSED(p);
  1079. APP = malloc(sizeof(Nrf24Scan));
  1080. APP->event_queue = furi_message_queue_alloc(8, sizeof(PluginEvent));
  1081. APP->plugin_state = malloc(sizeof(PluginState));
  1082. ValueMutex state_mutex;
  1083. if(!init_mutex(&state_mutex, APP->plugin_state, sizeof(PluginState))) {
  1084. furi_message_queue_free(APP->event_queue);
  1085. FURI_LOG_E(TAG, "cannot create mutex");
  1086. free(APP->plugin_state);
  1087. return 255;
  1088. }
  1089. APP->log_arr = malloc(LOG_REC_SIZE * MAX_LOG_RECORDS);
  1090. if(APP->log_arr == NULL) {
  1091. FURI_LOG_E(TAG, "Not enouch memory: %d", LOG_REC_SIZE * MAX_LOG_RECORDS);
  1092. strcpy(addr_file_name, "MEMORY LOW!");
  1093. }
  1094. clear_log();
  1095. APP->found = malloc(sizeof(struct FOUND) * MAX_LOG_RECORDS);
  1096. if(APP->found == NULL) {
  1097. FURI_LOG_E(TAG, "Not enouch memory: %d", sizeof(struct FOUND) * MAX_LOG_RECORDS);
  1098. strcpy(addr_file_name, "MEMORY LOW!!");
  1099. }
  1100. memset((uint8_t*)&addrs, 0, sizeof(addrs));
  1101. memset((uint8_t*)&addrs_sniff, 0, sizeof(addrs_sniff));
  1102. nrf24_init();
  1103. // Set system callbacks
  1104. APP->view_port = view_port_alloc();
  1105. view_port_draw_callback_set(APP->view_port, render_callback, &state_mutex);
  1106. view_port_input_callback_set(APP->view_port, input_callback, APP->event_queue);
  1107. // Open GUI and register view_port
  1108. APP->gui = furi_record_open(RECORD_GUI);
  1109. gui_add_view_port(APP->gui, APP->view_port, GuiLayerFullscreen);
  1110. APP->notification = furi_record_open(RECORD_NOTIFICATION);
  1111. APP->storage = furi_record_open(RECORD_STORAGE);
  1112. storage_common_mkdir(APP->storage, SCAN_APP_PATH_FOLDER);
  1113. Stream* file_stream = file_stream_alloc(APP->storage);
  1114. FuriString* path = furi_string_alloc();
  1115. furi_string_set(path, SCAN_APP_PATH_FOLDER);
  1116. furi_string_cat(path, "/");
  1117. furi_string_cat(path, SNIFF_FILENAME);
  1118. if(file_stream_open(file_stream, furi_string_get_cstr(path), FSAM_READ, FSOM_OPEN_EXISTING)) {
  1119. uint8_t err = load_settings_file(file_stream);
  1120. if(!err) strncpy(addr_file_name, furi_string_get_cstr(path) + sizeof(SCAN_APP_PATH_FOLDER), sizeof(addr_file_name));
  1121. else snprintf(addr_file_name, sizeof(addr_file_name), "LOAD ERROR#%d", err);
  1122. } else {
  1123. strcpy(addr_file_name, "NONE");
  1124. if(what_to_do == 1) {
  1125. addrs.addr_P0[0] = 0;
  1126. addrs.addr_P0[1] = 0x55;
  1127. addrs.addr_len = 2;
  1128. addrs.addr_count = 1;
  1129. view_log_decode_CRC = NRF_CRC = 2;
  1130. NRF_Payload_sniff_min = 0; // Min
  1131. }
  1132. }
  1133. file_stream_close(file_stream);
  1134. stream_free(file_stream);
  1135. furi_string_free(path);
  1136. PluginEvent event;
  1137. for(bool processing = true; processing;) {
  1138. FuriStatus event_status = furi_message_queue_get(APP->event_queue, &event, 100);
  1139. PluginState* plugin_state = (PluginState*)acquire_mutex_block(&state_mutex);
  1140. if(event_status == FuriStatusOk) {
  1141. // press events
  1142. if(event.type == EventTypeKey) {
  1143. //FURI_LOG_D(TAG, "Key: %d Type: %d Sec: %u", event.input.key, event.input.type, event.input.sequence);
  1144. switch(event.input.key) {
  1145. case InputKeyUp:
  1146. if(event.input.type == InputTypeShort || event.input.type == InputTypeRepeat) {
  1147. if(what_doing == 0) {
  1148. if(menu_selected > 0) menu_selected--; else menu_selected = menu_selected_max;
  1149. } else if(what_doing == 1) {
  1150. view_log_arr_idx -= event.input.type == InputTypeRepeat ? 10 : 1;
  1151. if(view_log_arr_idx >= log_arr_idx) view_log_arr_idx = 0;
  1152. } else if(what_doing == 2) {
  1153. if(view_found > -1) view_found--;
  1154. }
  1155. }
  1156. break;
  1157. case InputKeyDown:
  1158. if(event.input.type == InputTypeShort || event.input.type == InputTypeRepeat) {
  1159. if(what_doing == 0) {
  1160. if(menu_selected < menu_selected_max) menu_selected++; else menu_selected = 0;
  1161. } else if(what_doing == 1) {
  1162. view_log_arr_idx += event.input.type == InputTypeRepeat ? 10 : 1;
  1163. if(view_log_arr_idx >= log_arr_idx) view_log_arr_idx = log_arr_idx - 1;
  1164. } else if(what_doing == 2) {
  1165. if(view_found < found_total / 7) view_found++;
  1166. }
  1167. }
  1168. break;
  1169. case InputKeyLeft:
  1170. if(event.input.type == InputTypeShort || event.input.type == InputTypeRepeat) {
  1171. if(what_doing == 0) {
  1172. switch(menu_selected) {
  1173. case Menu_enter_channel:
  1174. NRF_channel -= event.input.type == InputTypeRepeat ? 10 : 1;
  1175. if(NRF_channel > MAX_CHANNEL) NRF_channel = MAX_CHANNEL;
  1176. break;
  1177. case Menu_enter_rate:
  1178. if(what_to_do == 1) { // SNIFF
  1179. NRF_Payload_sniff_min -= event.input.type == InputTypeRepeat ? 10 : 1;
  1180. correct_NRF_Payload_sniff_min();
  1181. } else {
  1182. NRF_Payload -= event.input.type == InputTypeRepeat ? 10 : 1;
  1183. if(NRF_Payload > 32) NRF_Payload = 0;
  1184. }
  1185. break;
  1186. case Menu_enter_scan_period:
  1187. find_channel_period -= event.input.type == InputTypeRepeat ? 10 : 1;
  1188. if(find_channel_period < 0) find_channel_period = 0;
  1189. break;
  1190. case Menu_log:
  1191. if(--log_to_file < -1) log_to_file = 2;
  1192. break;
  1193. case Menu_ok:
  1194. if(--what_to_do > 3) what_to_do = 3;
  1195. break;
  1196. }
  1197. } else if(what_doing == 1) {
  1198. if(view_log_arr_x > 0) view_log_arr_x--;
  1199. } else if(what_doing == 2) {
  1200. //if(NRF_ESB == 0)
  1201. view_log_decode_PCF ^= 1;
  1202. }
  1203. }
  1204. break;
  1205. case InputKeyRight:
  1206. if(event.input.type == InputTypeShort || event.input.type == InputTypeRepeat) {
  1207. if(what_doing == 0) {
  1208. switch(menu_selected) {
  1209. case Menu_open_file:
  1210. save_settings ^= 1;
  1211. break;
  1212. case Menu_enter_channel:
  1213. NRF_channel += event.input.type == InputTypeRepeat ? 10 : 1;
  1214. if(NRF_channel > MAX_CHANNEL) NRF_channel = 0;
  1215. break;
  1216. case Menu_enter_rate:
  1217. if(what_to_do == 1) { // SNIFF
  1218. NRF_Payload_sniff_min += event.input.type == InputTypeRepeat ? 10 : 1;
  1219. correct_NRF_Payload_sniff_min();
  1220. } else {
  1221. NRF_Payload += event.input.type == InputTypeRepeat ? 10 : 1;
  1222. if(NRF_Payload > 32) NRF_Payload = 32;
  1223. }
  1224. break;
  1225. case Menu_enter_scan_period:
  1226. find_channel_period += event.input.type == InputTypeRepeat ? 10 : 1;
  1227. break;
  1228. case Menu_log:
  1229. if(++log_to_file > 2) log_to_file = -1;
  1230. break;
  1231. case Menu_ok:
  1232. if(++what_to_do > 3) what_to_do = 0;
  1233. break;
  1234. }
  1235. } else if(what_doing == 1) {
  1236. if(view_log_arr_x < VIEW_LOG_MAX_X) view_log_arr_x++;
  1237. } else if(what_doing == 2) {
  1238. if(++view_log_decode_CRC > 2) view_log_decode_CRC = 0;
  1239. }
  1240. }
  1241. break;
  1242. case InputKeyOk:
  1243. if(event.input.type == InputTypeShort) {
  1244. if(what_doing == 0) {
  1245. switch(menu_selected) {
  1246. case Menu_open_file:
  1247. if(save_settings) {
  1248. write_to_log_file(APP->storage, true);
  1249. } else {
  1250. file_stream = file_stream_alloc(APP->storage);
  1251. if(select_settings_file(file_stream)) {
  1252. uint8_t err = load_settings_file(file_stream);
  1253. if(!err) save_to_new_log = true; else snprintf(addr_file_name, sizeof(addr_file_name), "LOAD ERROR#%d", err);
  1254. file_stream_close(file_stream);
  1255. menu_selected = Menu_ok;
  1256. }
  1257. stream_free(file_stream);
  1258. }
  1259. break;
  1260. case Menu_enter_channel:
  1261. if(what_to_do == 1) {
  1262. if(NRF_ESB) NRF_DPL = NRF_ESB = 0; else NRF_ESB = 1;
  1263. } else {
  1264. if(NRF_ESB) {
  1265. if(NRF_DPL) NRF_DPL = NRF_ESB = 0; else NRF_DPL = 1;
  1266. } else NRF_ESB = 1;
  1267. }
  1268. break;
  1269. case Menu_enter_rate:
  1270. NRF_rate++;
  1271. if(NRF_rate > 2) NRF_rate = 0;
  1272. break;
  1273. case Menu_enter_scan_period:
  1274. if(++NRF_CRC > 2) NRF_CRC = what_to_do == 1 ? 1 : 0;
  1275. break;
  1276. case Menu_ok:
  1277. if(what_to_do) {
  1278. if(addrs.addr_count || what_to_do == 1) {
  1279. if(log_to_file == -1) {
  1280. log_to_file = 0;
  1281. clear_log();
  1282. save_to_new_log = true;
  1283. } else if(log_to_file == 1) save_to_new_log = true;
  1284. start_scanning();
  1285. if(!NRF_ERROR) what_doing = 1;
  1286. }
  1287. } else what_doing = 1;
  1288. key_press_seq_ok = event.input.sequence;
  1289. break;
  1290. }
  1291. } else if(what_doing == 1) {
  1292. what_doing = 2;
  1293. } else if(what_doing == 2) {
  1294. what_doing = 1;
  1295. }
  1296. } else if(event.input.type == InputTypeLong) {
  1297. if(what_doing == 0) {
  1298. if(menu_selected == Menu_enter_channel) {
  1299. NRF_AA_OFF ^= 1;
  1300. key_press_seq_ok = event.input.sequence;
  1301. } else if(menu_selected == Menu_log) { // Log
  1302. if(log_arr_idx && (log_to_file == 1 || log_to_file == 2)) {
  1303. write_to_log_file(APP->storage, false);
  1304. clear_log();
  1305. }
  1306. }
  1307. } else if(what_doing == 1 || what_doing == 2) {
  1308. nrf24_send_packet();
  1309. }
  1310. }
  1311. break;
  1312. case InputKeyBack:
  1313. if(event.input.type == InputTypeLong) processing = false;
  1314. else if(event.input.type == InputTypeShort) {
  1315. if(what_doing) what_doing--;
  1316. if(what_doing == 0) {
  1317. nrf24_set_idle(nrf24_HANDLE);
  1318. }
  1319. }
  1320. break;
  1321. default:
  1322. break;
  1323. }
  1324. }
  1325. }
  1326. if(what_doing && what_to_do) {
  1327. nrf24_read_newpacket();
  1328. if(find_channel_period && furi_get_tick() - start_time >= (uint32_t)find_channel_period * 1000UL) {
  1329. if(++NRF_channel > MAX_CHANNEL) NRF_channel = 0;
  1330. start_scanning();
  1331. }
  1332. }
  1333. view_port_update(APP->view_port);
  1334. release_mutex(&state_mutex, plugin_state);
  1335. }
  1336. nrf24_set_idle(nrf24_HANDLE);
  1337. if(log_arr_idx && (log_to_file == 1 || log_to_file == 2)) {
  1338. write_to_log_file(APP->storage, false);
  1339. }
  1340. nrf24_deinit();
  1341. view_port_enabled_set(APP->view_port, false);
  1342. gui_remove_view_port(APP->gui, APP->view_port);
  1343. furi_record_close(RECORD_GUI);
  1344. furi_record_close(RECORD_NOTIFICATION);
  1345. furi_record_close(RECORD_STORAGE);
  1346. view_port_free(APP->view_port);
  1347. furi_message_queue_free(APP->event_queue);
  1348. free(APP->plugin_state);
  1349. if(APP->log_arr) free(APP->log_arr);
  1350. if(APP->found) free(APP->found);
  1351. free(APP);
  1352. return 0;
  1353. }