nrf24scan.c 42 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147
  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 "1.7"
  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
  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 = 1; // 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;// len in bytes, max 32
  65. uint8_t NRF_Sniff_payload_max = 28;
  66. uint8_t NRF_AA_OFF = 0; // Disable Auto Acknowledgement
  67. bool NRF_ERROR = 0;
  68. struct {
  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. } addrs;
  78. int8_t log_to_file = 0; // 0 - no, 1 - yes(new), 2 - append, -1 - only clear
  79. uint16_t log_arr_idx;
  80. uint16_t view_log_arr_idx = 0;
  81. uint16_t view_log_arr_x = 0;
  82. bool save_to_new_log = true;
  83. uint16_t last_packet_send = -1;
  84. uint8_t last_packet_send_st = 0;
  85. int16_t find_channel_period = 0; // sec
  86. uint8_t menu_selected = 0;
  87. uint32_t start_time;
  88. 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>
  89. uint8_t view_log_decode_CRC = 0; // CRC bytes - 1/2, 0 - none
  90. #define menu_selected_max 5
  91. enum {
  92. Menu_open_file = 0,
  93. Menu_enter_channel,
  94. Menu_enter_rate,
  95. Menu_enter_scan_period,
  96. Menu_log,
  97. Menu_ok
  98. };
  99. //#define MIN(a, b) ((a<b)?a:b)
  100. static uint8_t GetHexVal(char hex) {
  101. return (uint8_t)hex - ((uint8_t)hex < 58 ? 48 : ((uint8_t)hex < 97 ? 55 : 87));
  102. }
  103. // Return num bytes in array
  104. static uint8_t ConvertHexToArray(char * hex, uint8_t * array, uint8_t maxlen) {
  105. uint8_t len = 0;
  106. while(maxlen) {
  107. uint8_t ch = *hex++;
  108. if(ch == 0) break;
  109. if(ch < '0') continue;
  110. *array++ = (GetHexVal(ch) << 4) + GetHexVal(*hex++);
  111. len++;
  112. maxlen--;
  113. }
  114. return len;
  115. }
  116. static void add_to_str_hex_bytes(char *out, char *arr, int bytes)
  117. {
  118. if(bytes <= 0) return;
  119. out += strlen(out);
  120. do {
  121. snprintf(out, 3, "%02X", *arr++);
  122. out += 2;
  123. } while(--bytes);
  124. }
  125. static void add_to_furi_str_hex_bytes(FuriString* str, char *arr, int bytes)
  126. {
  127. if(!bytes) return;
  128. do {
  129. furi_string_cat_printf(str, "%02X", *arr++);
  130. } while(--bytes);
  131. }
  132. static void add_to_str_hex_bytes_shift_9b(char *out, char *arr, int bytes)
  133. {
  134. if(bytes <= 0) return;
  135. out += strlen(out);
  136. arr++; // +8b
  137. do {
  138. snprintf(out, 4, "%02X", ((uint8_t)(*arr << 1)) | (*(arr+1) >> 7));
  139. arr++;
  140. out += 2;
  141. } while(--bytes);
  142. }
  143. void clear_log()
  144. {
  145. log_arr_idx = 0;
  146. view_log_arr_idx = 0;
  147. last_packet_send = -1;
  148. }
  149. void allocate_log_array()
  150. {
  151. APP->log_arr = malloc(LOG_REC_SIZE * MAX_LOG_RECORDS);
  152. if(APP->log_arr == NULL) {
  153. FURI_LOG_E(TAG, "Not enouch memory: %d", LOG_REC_SIZE * MAX_LOG_RECORDS);
  154. strcpy(addr_file_name, "MEMORY LOW!");
  155. }
  156. clear_log();
  157. }
  158. void write_to_log_file(Storage* storage, bool f_settings)
  159. {
  160. if(log_arr_idx == 0 && !f_settings) return;
  161. Stream* file_stream = file_stream_alloc(storage);
  162. FuriString* str = furi_string_alloc();
  163. furi_string_set(str, SCAN_APP_PATH_FOLDER);
  164. furi_string_cat(str, "/");
  165. bool fl;
  166. if(f_settings) {
  167. furi_string_cat(str, SETTINGS_FILENAME);
  168. fl = file_stream_open(file_stream, furi_string_get_cstr(str), FSAM_READ_WRITE, FSOM_CREATE_ALWAYS);
  169. if(!fl) file_stream_close(file_stream);
  170. } else {
  171. furi_string_cat(str, LOG_FILENAME);
  172. furi_string_cat(str, LOG_FILEEXT);
  173. if(save_to_new_log) {
  174. int cnt = 1;
  175. do {
  176. fl = file_stream_open(file_stream, furi_string_get_cstr(str), FSAM_READ_WRITE, FSOM_CREATE_NEW);
  177. if(fl) break;
  178. file_stream_close(file_stream);
  179. furi_string_set(str, SCAN_APP_PATH_FOLDER);
  180. furi_string_cat(str, "/");
  181. furi_string_cat(str, LOG_FILENAME);
  182. furi_string_cat_printf(str, "-%02d", cnt);
  183. furi_string_cat(str, LOG_FILEEXT);
  184. } while(++cnt < 100);
  185. if(!fl) {
  186. FURI_LOG_E(TAG, "Failed to create new log file");
  187. notification_message(APP->notification, &sequence_blink_red_100);
  188. }
  189. } else {
  190. fl = file_stream_open(file_stream, furi_string_get_cstr(str), FSAM_READ_WRITE, FSOM_OPEN_APPEND);
  191. if(fl) {
  192. if(stream_size(file_stream) == 0) save_to_new_log = true;
  193. } else file_stream_close(file_stream);
  194. }
  195. }
  196. if(fl) {
  197. FURI_LOG_D(TAG, "Save to %s", furi_string_get_cstr(str));
  198. if(save_to_new_log || f_settings) {
  199. furi_string_set(str, what_to_do == 1 ? SettingsFld_Sniff : "");
  200. furi_string_cat_printf(str, "%s %d\n%s %d\n%s %d\n", SettingsFld_Rate, NRF_rate, SettingsFld_Ch, NRF_channel, SettingsFld_ESB, NRF_ESB);
  201. furi_string_cat_printf(str, "%s %d\n%s %d\n%s %d\n", SettingsFld_DPL, NRF_DPL, SettingsFld_CRC, NRF_CRC, SettingsFld_Payload, NRF_Payload);
  202. furi_string_cat_printf(str, "P0: ");
  203. add_to_furi_str_hex_bytes(str, (char*)addrs.addr_P0, addrs.addr_len); furi_string_cat(str, "\n");
  204. 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"); }
  205. if(addrs.addr_count > 2) { furi_string_cat_printf(str, "P2: "); furi_string_cat_printf(str, "%02X\n", addrs.addr_P2); }
  206. if(addrs.addr_count > 3) { furi_string_cat_printf(str, "P3: "); furi_string_cat_printf(str, "%02X\n", addrs.addr_P3); }
  207. if(addrs.addr_count > 4) { furi_string_cat_printf(str, "P4: "); furi_string_cat_printf(str, "%02X\n", addrs.addr_P4); }
  208. if(addrs.addr_count > 5) { furi_string_cat_printf(str, "P5: "); furi_string_cat_printf(str, "%02X\n", addrs.addr_P5); }
  209. if(!(fl = stream_write_string(file_stream, str) == furi_string_size(str))) {
  210. FURI_LOG_E(TAG, "Failed to write header to file!");
  211. notification_message(APP->notification, &sequence_blink_red_100);
  212. }
  213. }
  214. if(fl) {
  215. if(f_settings) {
  216. save_settings = 0;
  217. if(strcmp(addr_file_name, "NONE") == 0) strcpy(addr_file_name, SETTINGS_FILENAME);
  218. } else {
  219. int i = 0;
  220. for(; i < log_arr_idx; i++) {
  221. furi_string_reset(str);
  222. add_to_furi_str_hex_bytes(str, (char*)APP->log_arr + i * LOG_REC_SIZE, LOG_REC_SIZE);
  223. furi_string_cat(str, "\n");
  224. if(stream_write_string(file_stream, str) != furi_string_size(str)) {
  225. FURI_LOG_E(TAG, "Failed to write to file!");
  226. break;
  227. }
  228. }
  229. if(i == log_arr_idx) {
  230. notification_message(APP->notification, &sequence_blink_yellow_100);
  231. FURI_LOG_D(TAG, "File saved");
  232. }
  233. save_to_new_log = false;
  234. }
  235. }
  236. file_stream_close(file_stream);
  237. } else {
  238. FURI_LOG_E(TAG, "Failed to open file %s", furi_string_get_cstr(str));
  239. notification_message(APP->notification, &sequence_blink_red_100);
  240. }
  241. stream_free(file_stream);
  242. furi_string_free(str);
  243. }
  244. static bool select_settings_file(Stream* stream) {
  245. DialogsApp* dialogs = furi_record_open("dialogs");
  246. bool result = false;
  247. FuriString* path;
  248. path = furi_string_alloc();
  249. furi_string_set(path, SCAN_APP_PATH_FOLDER);
  250. DialogsFileBrowserOptions browser_options;
  251. dialog_file_browser_set_basic_options(&browser_options, ".txt", NULL);
  252. browser_options.hide_ext = false;
  253. bool ret = dialog_file_browser_show(dialogs, path, path, &browser_options);
  254. furi_record_close("dialogs");
  255. if(ret) {
  256. if(!file_stream_open(stream, furi_string_get_cstr(path), FSAM_READ, FSOM_OPEN_EXISTING)) {
  257. FURI_LOG_D(TAG, "Cannot open file \"%s\"", furi_string_get_cstr(path));
  258. file_stream_close(stream);
  259. } else {
  260. FURI_LOG_D(TAG, "Open file \"%s\"", furi_string_get_cstr(path));
  261. strncpy(addr_file_name, furi_string_get_cstr(path) + sizeof(SCAN_APP_PATH_FOLDER), sizeof(addr_file_name));
  262. result = true;
  263. }
  264. }
  265. furi_string_free(path);
  266. return result;
  267. }
  268. // 0 - success, otherwise an error
  269. static uint8_t load_settings_file(Stream* file_stream) {
  270. size_t file_size = 0;
  271. char* file_buf;
  272. uint8_t err = 5;
  273. file_size = stream_size(file_stream);
  274. if(file_size == (size_t)0) {
  275. FURI_LOG_D(TAG, "load failed. file_size: %d", file_size);
  276. return 1;
  277. }
  278. file_size = MIN(file_size, (size_t) LOG_REC_SIZE * MAX_LOG_RECORDS * 2 + 100);
  279. file_buf = malloc(file_size + 1);
  280. if(file_buf == NULL) {
  281. FURI_LOG_D(TAG, "Memory low, need: %d", file_size);
  282. return 2;
  283. }
  284. memset(file_buf, 0, file_size + 1);
  285. if(stream_read(file_stream, (uint8_t*)file_buf, file_size) == file_size) {
  286. FURI_LOG_D(TAG, "Loading settings file");
  287. char* line_ptr = file_buf;
  288. int16_t line_num = 0;
  289. memset((uint8_t*)&addrs, 0, sizeof(addrs));
  290. what_to_do = 2;
  291. bool log_loaded = false;
  292. while(line_ptr && (size_t)(line_ptr - file_buf) < file_size) {
  293. char* end_ptr = strstr((char*)line_ptr, "\n");
  294. if(end_ptr == NULL) end_ptr = file_buf + file_size; else *end_ptr = '\0';
  295. int line_len = end_ptr - line_ptr;
  296. if(*line_ptr == '\r' || line_len == 0) {
  297. line_ptr = end_ptr + 1;
  298. continue;
  299. }
  300. if(*(end_ptr - 1) < '0') {
  301. *(end_ptr - 1) = '\0';
  302. line_len--;
  303. }
  304. //FURI_LOG_D(TAG, " L#%d: [%d]%s", line_num, line_len, line_ptr);
  305. if(strncmp(line_ptr, SettingsFld_Rate, sizeof(SettingsFld_Rate)-1) == 0) {
  306. NRF_rate = atoi(line_ptr + sizeof(SettingsFld_Rate));
  307. } else if(strncmp(line_ptr, SettingsFld_Ch, sizeof(SettingsFld_Ch)-1) == 0) {
  308. NRF_channel = atoi(line_ptr + sizeof(SettingsFld_Ch));
  309. } else if(strncmp(line_ptr, SettingsFld_ESB, sizeof(SettingsFld_ESB)-1) == 0) {
  310. NRF_ESB = atoi(line_ptr + sizeof(SettingsFld_ESB));
  311. } else if(strncmp(line_ptr, SettingsFld_DPL, sizeof(SettingsFld_DPL)-1) == 0) {
  312. NRF_DPL = atoi(line_ptr + sizeof(SettingsFld_DPL));
  313. } else if(strncmp(line_ptr, SettingsFld_CRC, sizeof(SettingsFld_CRC)-1) == 0) {
  314. NRF_CRC = atoi(line_ptr + sizeof(SettingsFld_CRC));
  315. if(what_to_do == 1) view_log_decode_CRC = NRF_CRC;
  316. } else if(strncmp(line_ptr, SettingsFld_Payload, sizeof(SettingsFld_Payload)-1) == 0) {
  317. NRF_Payload = atoi(line_ptr + sizeof(SettingsFld_Payload));
  318. if(NRF_Payload == 0 || NRF_Payload > 32) NRF_Payload = 32;
  319. } else if(strncmp(line_ptr, SettingsFld_Sniff, sizeof(SettingsFld_Sniff)-1) == 0) {
  320. what_to_do = 1;
  321. } else if(*line_ptr == SettingsFld_Addr) {
  322. char a = *(++line_ptr);
  323. line_ptr += 3;
  324. switch(a) {
  325. case '0':
  326. addrs.addr_len = ConvertHexToArray(line_ptr, addrs.addr_P0, what_to_do == 1 ? 3 : 5);
  327. //FURI_LOG_D(TAG, " +Addr(%d): %02X%02X%02X...", addrs.addr_len, addrs.addr_P0[0], addrs.addr_P0[1], addrs.addr_P0[2]);
  328. if(addrs.addr_len >= 2) err = 0;
  329. break;
  330. case '1':
  331. ConvertHexToArray(line_ptr, addrs.addr_P1, what_to_do == 1 ? 3 : 5);
  332. //FURI_LOG_D(TAG, " +Addr: %02X%02X%02X...", addrs.addr_P0[1], addrs.addr_P1[1], addrs.addr_P1[2]);
  333. break;
  334. case '2':
  335. ConvertHexToArray(line_ptr, &addrs.addr_P2, 1);
  336. break;
  337. case '3':
  338. ConvertHexToArray(line_ptr, &addrs.addr_P3, 1);
  339. break;
  340. case '4':
  341. ConvertHexToArray(line_ptr, &addrs.addr_P4, 1);
  342. break;
  343. case '5':
  344. ConvertHexToArray(line_ptr, &addrs.addr_P5, 1);
  345. break;
  346. default:
  347. a = 0;
  348. break;
  349. }
  350. if(err == 0 && a) addrs.addr_count = a - '0' + 1;
  351. } else if(line_len >= (NRF_Payload + 1) * 2) { // data
  352. if(!log_loaded) {
  353. clear_log();
  354. what_to_do = 0;
  355. log_loaded = true;
  356. }
  357. if(log_arr_idx < MAX_LOG_RECORDS - 1) {
  358. ConvertHexToArray(line_ptr, APP->log_arr + log_arr_idx * LOG_REC_SIZE, LOG_REC_SIZE);
  359. log_arr_idx++;
  360. }
  361. }
  362. line_ptr = end_ptr + 1;
  363. line_num++;
  364. }
  365. } else {
  366. FURI_LOG_D(TAG, "load failed. file size: %d", file_size);
  367. err = 4;
  368. }
  369. free(file_buf);
  370. return err;
  371. }
  372. static void input_callback(InputEvent* input_event, FuriMessageQueue* event_queue) {
  373. furi_assert(event_queue);
  374. PluginEvent event = {.type = EventTypeKey, .input = *input_event};
  375. furi_message_queue_put(event_queue, &event, FuriWaitForever);
  376. }
  377. static void prepare_nrf24(bool fsend_packet)
  378. {
  379. uint8_t erx_addr = (1<<0); // Enable RX_P0
  380. if(!fsend_packet) {
  381. if(addrs.addr_count == 0) return;
  382. uint8_t payload = NRF_Payload;
  383. if(what_to_do == 1) { // SNIFF
  384. payload += 5 + NRF_CRC; // + addr_max + CRC
  385. if(NRF_ESB) payload += 2;
  386. if(payload > 32) payload = 32;
  387. nrf24_write_reg(nrf24_HANDLE, REG_CONFIG, 0x70); // Mask all interrupts
  388. nrf24_write_reg(nrf24_HANDLE, REG_STATUS, 0x70); // clear interrupts
  389. nrf24_write_reg(nrf24_HANDLE, REG_RF_SETUP, NRF_rate);
  390. nrf24_write_reg(nrf24_HANDLE, REG_SETUP_RETR, 0); // Automatic Retransmission
  391. nrf24_write_reg(nrf24_HANDLE, REG_EN_AA, 0); // Auto acknowledgement
  392. nrf24_write_reg(nrf24_HANDLE, REG_FEATURE, 0); // Enables the W_TX_PAYLOAD_NOACK command, Disable Payload with ACK, set Dynamic Payload
  393. } else {
  394. nrf24_write_reg(nrf24_HANDLE, REG_CONFIG, 0x70 | ((NRF_CRC == 1 ? 0b1000 : NRF_CRC == 2 ? 0b1100 : 0))); // Mask all interrupts
  395. nrf24_write_reg(nrf24_HANDLE, REG_STATUS, 0x70); // clear interrupts
  396. nrf24_write_reg(nrf24_HANDLE, REG_RF_SETUP, NRF_rate);
  397. nrf24_write_reg(nrf24_HANDLE, REG_SETUP_RETR, NRF_ESB ? 0x11 : 0); // Automatic Retransmission
  398. nrf24_write_reg(nrf24_HANDLE, REG_EN_AA, NRF_AA_OFF || !NRF_ESB ? 0 : 0x3F); // Auto acknowledgement
  399. 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
  400. }
  401. nrf24_set_maclen(nrf24_HANDLE, addrs.addr_len);
  402. nrf24_set_mac(REG_RX_ADDR_P0, addrs.addr_P0, addrs.addr_len);
  403. uint8_t tmp[5] = { 0 };
  404. nrf24_read_reg(nrf24_HANDLE, REG_RX_ADDR_P0, tmp, addrs.addr_len);
  405. for(uint8_t i = 0; i < addrs.addr_len / 2; i++) {
  406. uint8_t tb = tmp[i];
  407. tmp[i] = tmp[addrs.addr_len - i - 1];
  408. tmp[addrs.addr_len - i - 1] = tb;
  409. }
  410. NRF_ERROR = memcmp(addrs.addr_P0, tmp, addrs.addr_len) != 0;
  411. uint8_t dyn = 0;
  412. if(addrs.addr_count > 0 && NRF_DPL) dyn |= (1<<0);
  413. FURI_LOG_D(TAG, "Payload: %d", payload);
  414. nrf24_write_reg(nrf24_HANDLE, RX_PW_P0, payload);
  415. if(addrs.addr_count > 1) {
  416. nrf24_set_mac(REG_RX_ADDR_P1, addrs.addr_P1, addrs.addr_len);
  417. nrf24_write_reg(nrf24_HANDLE, RX_PW_P1, payload);
  418. if(NRF_DPL) dyn |= (1<<1);
  419. erx_addr |= (1<<1); // Enable RX_P1
  420. } else nrf24_write_reg(nrf24_HANDLE, RX_PW_P1, 0);
  421. if(addrs.addr_count > 2) {
  422. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P2, &addrs.addr_P2, 1);
  423. nrf24_write_reg(nrf24_HANDLE, RX_PW_P2, payload);
  424. if(NRF_DPL) dyn |= (1<<2);
  425. erx_addr |= (1<<2); // Enable RX_P2
  426. } else nrf24_write_reg(nrf24_HANDLE, RX_PW_P2, 0);
  427. if(addrs.addr_count > 3) {
  428. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P3, &addrs.addr_P3, 1);
  429. nrf24_write_reg(nrf24_HANDLE, RX_PW_P3, payload);
  430. if(NRF_DPL) dyn |= (1<<3);
  431. erx_addr |= (1<<3); // Enable RX_P3
  432. } else nrf24_write_reg(nrf24_HANDLE, RX_PW_P3, 0);
  433. if(addrs.addr_count > 4) {
  434. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P4, &addrs.addr_P4, 1);
  435. nrf24_write_reg(nrf24_HANDLE, RX_PW_P4, payload);
  436. if(NRF_DPL) dyn |= (1<<4);
  437. erx_addr |= (1<<4); // Enable RX_P4
  438. } else nrf24_write_reg(nrf24_HANDLE, RX_PW_P4, 0);
  439. if(addrs.addr_count > 5) {
  440. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P5, &addrs.addr_P5, 1);
  441. nrf24_write_reg(nrf24_HANDLE, RX_PW_P5, payload);
  442. if(NRF_DPL) dyn |= (1<<5);
  443. erx_addr |= (1<<5); // Enable RX_P5
  444. } else nrf24_write_reg(nrf24_HANDLE, RX_PW_P5, 0);
  445. nrf24_write_reg(nrf24_HANDLE, REG_DYNPD, dyn); // Enable dynamic payload reg
  446. nrf24_write_reg(nrf24_HANDLE, REG_EN_RXADDR, erx_addr);
  447. nrf24_write_reg(nrf24_HANDLE, REG_RF_CH, NRF_channel);
  448. }
  449. nrf24_flush_rx(nrf24_HANDLE);
  450. nrf24_flush_tx(nrf24_HANDLE);
  451. nrf24_set_idle(nrf24_HANDLE);
  452. }
  453. static void start_scanning()
  454. {
  455. FURI_LOG_D(TAG, "Start proc-%d: Ch=%d Rate=%d", what_to_do, NRF_channel, NRF_rate);
  456. if(what_to_do == 1) {
  457. NRF_Sniff_payload_max = 32 - 3 - (NRF_ESB ? 2 : 0) - NRF_CRC + (addrs.addr_len - 2);
  458. if(NRF_Payload > NRF_Sniff_payload_max) NRF_Payload = NRF_Sniff_payload_max;
  459. view_log_decode_CRC = NRF_CRC;
  460. }
  461. prepare_nrf24(false);
  462. if(NRF_ERROR) {
  463. FURI_LOG_E(TAG, "NRF R/W ERROR!");
  464. return;
  465. }
  466. nrf24_set_rx_mode(nrf24_HANDLE);
  467. start_time = furi_get_tick();
  468. }
  469. // start bitnum = 7
  470. uint16_t calc_crc(uint32_t crc, uint8_t *ptr, uint8_t bitnum, uint16_t bits)
  471. {
  472. //uint8_t bitnum = 7;
  473. uint32_t crc_high, polynom;
  474. if(view_log_decode_CRC == 2) {
  475. crc_high = (1<<16);
  476. polynom = 0x11021; // X^16+X^12+X^5+1
  477. } else {
  478. crc_high = (1<<8);
  479. polynom = 0x107; // x^8+x^2+x^1+1
  480. }
  481. while(bits--) {
  482. crc <<= 1;
  483. if(((crc & crc_high) != 0) ^ ((*ptr >> bitnum) & 1)) crc ^= polynom;
  484. if(bitnum == 0) {
  485. ptr++;
  486. bitnum = 7;
  487. } else bitnum--;
  488. }
  489. return crc & (view_log_decode_CRC == 2 ? 0xFFFF : 0xFF);
  490. }
  491. // shifted 1 bit right
  492. uint16_t get_shifted_crc(uint8_t *pcrc)
  493. {
  494. uint16_t crc = ((uint8_t)(*pcrc << 1)) | (*(pcrc+1) >> 7);
  495. if(view_log_decode_CRC == 2) {
  496. crc = (crc << 8) | (((uint8_t)(*(pcrc+1) << 1))) | (*(pcrc+2) >> 7);
  497. }
  498. return crc;
  499. }
  500. bool check_packet(uint8_t *pkt, uint8_t size)
  501. {
  502. for(uint8_t addr_size = 3; addr_size <= 5; addr_size++) {
  503. if(NRF_ESB){
  504. uint8_t b = *(pkt + addr_size) >> 2;
  505. if((b > NRF_Sniff_payload_max && b != 0x33)) continue;
  506. if(furi_log_get_level() == FuriLogLevelDebug && addr_size == 3) {
  507. char dbuf[70];
  508. dbuf[0] = 0;
  509. add_to_str_hex_bytes(dbuf, (char*)pkt, size);
  510. FURI_LOG_D(TAG, "PKT%d: %s (%d)", *(pkt - 1), dbuf, size);
  511. }
  512. if(b != 0x33) { // DPL
  513. uint16_t crc = view_log_decode_CRC == 2 ? 0xFFFF : 0xFF;
  514. crc = calc_crc(crc, pkt, 7, 9 + b * 8);
  515. FURI_LOG_D(TAG, "DCRC: %X - %X", crc, get_shifted_crc(pkt + b + 1));
  516. if(crc == get_shifted_crc(pkt + b + 1)) {
  517. *(pkt - 1) = ((b & 0x1F) << 3) + 0b100 + (addr_size - 2);
  518. FURI_LOG_D(TAG, "VALID: pl: %d, addr: %d", b, addr_size);
  519. return true;
  520. }
  521. } else {
  522. for(uint8_t i = 0; i < size - view_log_decode_CRC; i++) {
  523. uint16_t crc = view_log_decode_CRC == 2 ? 0xFFFF : 0xFF;
  524. crc = calc_crc(crc, pkt, 7, 9 + i * 8);
  525. FURI_LOG_D(TAG, "CRC: %X - %X", crc, get_shifted_crc(pkt + i + 1));
  526. if(crc == get_shifted_crc(pkt + i + 1)) {
  527. *(pkt - 1) = ((i & 0x1F) << 3) + 0b100 + (addr_size - 2);
  528. FURI_LOG_D(TAG, "VALID: pl: %d, addr: %d", i, addr_size);
  529. return true;
  530. }
  531. }
  532. }
  533. } else {
  534. for(uint8_t i = 0; i < size - view_log_decode_CRC; i++) {
  535. uint16_t crc = view_log_decode_CRC == 2 ? 0xFFFF : 0xFF;
  536. crc = calc_crc(crc, pkt, 7, i * 8);
  537. if((view_log_decode_CRC == 1 && crc == *(pkt + i + 1)) || (view_log_decode_CRC == 2 && crc == ((*(pkt + i + 1)<<8) | *(pkt + i + 2)))) {
  538. *(pkt - 1) = ((i & 0x1F) << 3) + 0b000 + (addr_size - 2);
  539. return true;
  540. }
  541. }
  542. }
  543. }
  544. return false;
  545. }
  546. bool nrf24_read_newpacket() {
  547. if(APP->log_arr == NULL) return false;
  548. bool found = false;
  549. uint8_t packetsize;
  550. uint8_t *ptr = APP->log_arr + log_arr_idx * LOG_REC_SIZE;
  551. uint8_t st = nrf24_rxpacket(nrf24_HANDLE, ptr + 2 + (what_to_do == 1 ? addrs.addr_len - 2 : 0), &packetsize, !NRF_DPL);
  552. if(st & RX_DR) {
  553. st = (st >> 1) & 7; // pipe #
  554. if(what_to_do == 1) {
  555. *ptr++ = NRF_channel | 0x80;
  556. *ptr++ = st; // pipe #
  557. if(addrs.addr_len > 2) {
  558. *ptr = st == 0 ? addrs.addr_P0[2] : st == 1 ? addrs.addr_P1[2] : st == 2 ? addrs.addr_P2 : st == 3 ? addrs.addr_P3 : st == 4 ? addrs.addr_P4 : addrs.addr_P5;
  559. }
  560. if(!check_packet(ptr, packetsize)) {
  561. if(addrs.addr_len > 2) return false; // skip if mac MSB added to preamble
  562. if(addrs.addr_count == 1 && addrs.addr_P0[1] == 0xAA) { // Shift packet right by one bit if preamble = 0xAA
  563. for(uint8_t i = packetsize - 1; i > 0; i--) {
  564. ptr[i] = ptr[i - 1] << 7 | ptr[i] >> 1;
  565. }
  566. *ptr >>= 1;
  567. //if((st == 0 && (addrs.addr_P0[1] & 1)) || (st == 1 && (addrs.addr_P1[1] & 1))) *ptr |= 0x80;
  568. if(!check_packet(ptr, packetsize)) return false;
  569. } else return false;
  570. }
  571. } else {
  572. *ptr++ = NRF_channel;
  573. *ptr++ = ((packetsize & 0x1F) << 3) | st; // payload size + pipe #
  574. }
  575. if(packetsize < 32) memset(ptr + packetsize, 0, 32 - packetsize);
  576. if(log_arr_idx < MAX_LOG_RECORDS - 1) {
  577. log_arr_idx++;
  578. } else {
  579. if(log_to_file == 1 || log_to_file == 2) {
  580. write_to_log_file(APP->storage, false);
  581. clear_log();
  582. } else {
  583. memmove(APP->log_arr, APP->log_arr + LOG_REC_SIZE, log_arr_idx * LOG_REC_SIZE);
  584. }
  585. }
  586. FURI_LOG_D(TAG, "Found packet #%d pipe %d", log_arr_idx, st);
  587. notification_message(APP->notification, &sequence_blink_white_100);
  588. found = true;
  589. }
  590. return found;
  591. }
  592. bool nrf24_send_packet()
  593. {
  594. if(log_arr_idx == 0) return false;
  595. prepare_nrf24(!what_to_do);
  596. uint8_t *ptr = APP->log_arr + view_log_arr_idx * LOG_REC_SIZE;
  597. nrf24_write_reg(nrf24_HANDLE, REG_RF_CH, *ptr & 0x7F);
  598. if(*ptr & 0x80) { // RAW packet
  599. //uint8_t pktinfo = *(ptr + 1);
  600. //nrf24_set_maclen(nrf24_HANDLE, (pktinfo & 0b11) + 2);
  601. nrf24_set_maclen(nrf24_HANDLE, 2);
  602. //if(pktinfo & 0b100) { // ESB
  603. nrf24_write_reg(nrf24_HANDLE, REG_SETUP_RETR, 0); // No Automatic Retransmission
  604. nrf24_write_reg(nrf24_HANDLE, REG_EN_AA, 0); // No Auto acknowledgement
  605. //}
  606. //uint8_t alen = (*(ptr + 2) & 0b11) + 2;
  607. uint8_t adr[2] = { 0x55, 0x55 }; // NOT TESTED!
  608. if(*(ptr + 2) & 0x80) adr[0] = adr[1] = 0xAA;
  609. nrf24_set_mac(REG_RX_ADDR_P0, adr, 2);
  610. nrf24_set_mac(REG_TX_ADDR, adr, 2);
  611. last_packet_send_st = nrf24_txpacket(nrf24_HANDLE, ptr + 2, 32, false);
  612. } else {
  613. nrf24_write_reg(nrf24_HANDLE, REG_SETUP_RETR, NRF_ESB ? 0x11 : 0); // Automatic Retransmission
  614. nrf24_write_reg(nrf24_HANDLE, REG_EN_AA, NRF_AA_OFF || !NRF_ESB ? 0 : 0x3F); // Auto acknowledgement
  615. uint8_t *adr;
  616. uint8_t a = *(ptr + 1) & 0b111;
  617. if(a < 2) {
  618. if(a == 0) adr = addrs.addr_P0; else adr = addrs.addr_P1;
  619. nrf24_set_mac(REG_RX_ADDR_P0, adr, addrs.addr_len);
  620. nrf24_set_mac(REG_TX_ADDR, adr, addrs.addr_len);
  621. } else {
  622. uint8_t buf[5];
  623. memcpy(buf, addrs.addr_P1, addrs.addr_len - 1);
  624. buf[addrs.addr_len - 1] = a == 2 ? addrs.addr_P2 : a == 3 ? addrs.addr_P3 : a == 4 ? addrs.addr_P4 : addrs.addr_P5;
  625. nrf24_set_mac(REG_RX_ADDR_P0, buf, addrs.addr_len);
  626. nrf24_set_mac(REG_TX_ADDR, buf, addrs.addr_len);
  627. }
  628. a = *(ptr + 1) >> 3;
  629. if(a == 0) a = 32;
  630. last_packet_send_st = nrf24_txpacket(nrf24_HANDLE, ptr + 2, a, false);
  631. }
  632. last_packet_send = view_log_arr_idx;
  633. notification_message(APP->notification, last_packet_send_st ? &sequence_blink_blue_100 : &sequence_blink_red_100);
  634. if(what_to_do) start_scanning();
  635. return last_packet_send_st;
  636. }
  637. static void render_callback(Canvas* const canvas, void* ctx) {
  638. const PluginState* plugin_state = acquire_mutex((ValueMutex*)ctx, 25);
  639. if(plugin_state == NULL) return;
  640. //canvas_draw_frame(canvas, 0, 0, 128, 64); // border around the edge of the screen
  641. if(what_doing == 0) {
  642. canvas_set_font(canvas, FontSecondary); // 8x10 font
  643. if(save_settings) snprintf(screen_buf, sizeof(screen_buf), "Save: %s", SETTINGS_FILENAME); // menu_selected = 0
  644. else snprintf(screen_buf, sizeof(screen_buf), "Load: %s", addr_file_name);
  645. canvas_draw_str(canvas, 10, 10, screen_buf);
  646. snprintf(screen_buf, sizeof(screen_buf), "Ch: %d", NRF_channel); // menu_selected = 1
  647. canvas_draw_str(canvas, 10, 20, screen_buf);
  648. if(NRF_ESB) {
  649. strcpy(screen_buf, "ESB");
  650. if(NRF_DPL) strcat(screen_buf, " DPL");
  651. canvas_draw_str(canvas, 80, 20, screen_buf);
  652. }
  653. if(NRF_AA_OFF) { canvas_draw_str(canvas, 61, 20, "AA"); canvas_draw_line(canvas, 60, 21, 72, 12); }
  654. snprintf(screen_buf, sizeof(screen_buf), "Rate: %sbps", NRF_rate == 2 ? "2M" : NRF_rate == 1 ? "1M" : "250K"); // menu_selected = 2
  655. canvas_draw_str(canvas, 10, 30, screen_buf);
  656. snprintf(screen_buf, sizeof(screen_buf), "Payload: %d", NRF_Payload);
  657. canvas_draw_str(canvas, 80, 30, screen_buf);
  658. strcpy(screen_buf, "Next Ch time: "); // menu_selected = 3
  659. if(find_channel_period == 0) strcat(screen_buf, "off"); else snprintf(screen_buf + strlen(screen_buf), sizeof(screen_buf), "%d s", find_channel_period);
  660. canvas_draw_str(canvas, 10, 40, screen_buf);
  661. if(NRF_CRC == 1) canvas_draw_str(canvas, 99, 40, "CRC1");
  662. else if(NRF_CRC == 2) canvas_draw_str(canvas, 99, 40, "CRC2");
  663. 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
  664. canvas_draw_str(canvas, 10, 50, screen_buf);
  665. if(what_to_do) { // menu_selected = 5
  666. if(NRF_ERROR) snprintf(screen_buf, sizeof(screen_buf), "nRF24L01+ R/W ERROR!");
  667. else {
  668. if(what_to_do == 1) snprintf(screen_buf, sizeof(screen_buf), "Start sniff");
  669. else snprintf(screen_buf, sizeof(screen_buf), "Start scan (pipes: %d)", addrs.addr_count);
  670. }
  671. } else snprintf(screen_buf, sizeof(screen_buf), "View log (pipes: %d)", addrs.addr_count);
  672. canvas_draw_str(canvas, 10, 60, screen_buf);
  673. canvas_draw_str(canvas, 0, menu_selected * 10 + 10, ">");
  674. } else if(what_doing == 1){
  675. canvas_set_font(canvas, FontBatteryPercent); // 5x7 font, 9 lines
  676. bool ch2 = false;
  677. screen_buf[0] = '\0';
  678. if(view_log_arr_x == 0) { strcat(screen_buf, " "); ch2 = true;
  679. } else {
  680. snprintf(screen_buf, sizeof(screen_buf), "<%d", view_log_arr_x);
  681. if(view_log_arr_x < VIEW_LOG_MAX_X) ch2 = true;
  682. }
  683. snprintf(screen_buf + strlen(screen_buf), sizeof(screen_buf), " %s ch: %d - %d.", what_to_do == 1 ? "Sniff" : what_to_do == 2 ? "Read" : "View", NRF_channel, log_arr_idx);
  684. canvas_draw_str(canvas, 0, 7, screen_buf);
  685. if(ch2) canvas_draw_str(canvas, 121, 7, ">");
  686. if(log_arr_idx) {
  687. if(view_log_arr_idx >= log_arr_idx) view_log_arr_idx = log_arr_idx - 1;
  688. uint16_t page = view_log_arr_idx & ~7;
  689. for(uint8_t i = 0; i < 8 && page + i < log_arr_idx; i++) {
  690. screen_buf[0] = (view_log_arr_idx & 7) != i ? ' ' : last_packet_send != view_log_arr_idx ? '>' : last_packet_send_st ? '*' : '!';
  691. screen_buf[1] = '\0';
  692. uint8_t *ptr = APP->log_arr + (page + i) * LOG_REC_SIZE;
  693. uint8_t channel = *ptr++;
  694. uint8_t *crcptr = NULL;
  695. uint8_t pre = 0;
  696. int count = 0;
  697. if(channel & 0x80) { // RAW packet: nn:>{.address..}-xxxxxxxx
  698. uint8_t pktinfo = *ptr++;
  699. bool _PCF = pktinfo & 0b100;
  700. uint8_t count = (pktinfo >> 3);
  701. uint8_t adrsize = (pktinfo & 0b11) + 2;
  702. uint8_t plen = adrsize + count;
  703. count -= view_log_arr_x;
  704. uint8_t max_width = VIEW_LOG_WIDTH_B;
  705. if(view_log_arr_x == 0) max_width -= 5;
  706. if(count > max_width) count = max_width;
  707. if(count > 0) {
  708. uint8_t *pcrc = ptr;
  709. uint16_t crc;
  710. crc = view_log_decode_CRC == 2 ? 0xFFFF : 0xFF;
  711. crc = calc_crc(crc, pcrc, 7, (_PCF? 9 : 0) + plen * 8);
  712. pcrc += plen;
  713. if(_PCF) { //ESB
  714. pcrc++;
  715. if(crc == get_shifted_crc(pcrc)) crcptr = pcrc;
  716. } else {
  717. if((view_log_decode_CRC == 1 && crc == *pcrc) || (view_log_decode_CRC == 2 && crc == ((*pcrc<<8) | *(pcrc+1)))) {
  718. crcptr = pcrc;
  719. }
  720. }
  721. if(view_log_arr_x == 0) {
  722. add_to_str_hex_bytes(screen_buf, (char*)ptr, adrsize);
  723. for(int8_t j = 5 - adrsize; j > 0; j--) strcat(screen_buf, " ");
  724. strcat(screen_buf, "-");
  725. pre += 5 * 2 + 1;
  726. } else {
  727. ptr += view_log_arr_x - 1;
  728. }
  729. ptr += adrsize;
  730. 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);
  731. }
  732. } else {
  733. uint8_t dpl = *ptr++;
  734. uint8_t pipe = dpl & 0b111;
  735. dpl >>= 3;
  736. if(dpl == 0) dpl = 32;
  737. count = dpl - view_log_arr_x;
  738. if(view_log_decode_PCF) count--;
  739. uint8_t max_width = VIEW_LOG_WIDTH_B;
  740. if(view_log_arr_x == 0) {
  741. if(addrs.addr_count > 1) max_width--;
  742. if(view_log_decode_PCF) max_width -= 2;
  743. }
  744. if(count > max_width) count = max_width;
  745. if(count > 0) {
  746. if(view_log_decode_CRC) {
  747. static uint16_t prev_addr_CRC;
  748. static int8_t prev_pipe = -1;
  749. uint8_t *pcrc = APP->log_arr + (page + i) * LOG_REC_SIZE + 2;
  750. uint16_t crc;
  751. if(prev_pipe == pipe) { crc = prev_addr_CRC;
  752. } else {
  753. crc = view_log_decode_CRC == 2 ? 0xFFFF : 0xFF;
  754. if(pipe <= 1) { crc = calc_crc(crc, pipe == 0 ? addrs.addr_P0 : addrs.addr_P1, 7, addrs.addr_len * 8);
  755. } else {
  756. crc = calc_crc(crc, addrs.addr_P1, 7, (addrs.addr_len - 1) * 8);
  757. crc = calc_crc(crc, pipe == 2 ? &addrs.addr_P2 : pipe == 3 ? &addrs.addr_P3 : pipe == 4 ? &addrs.addr_P4 : &addrs.addr_P5, 7, 8);
  758. }
  759. prev_addr_CRC = crc;
  760. prev_pipe = pipe;
  761. }
  762. if(view_log_decode_PCF) {
  763. crc = calc_crc(crc, pcrc++, 7, 9);
  764. if(crc == get_shifted_crc(pcrc)) crcptr = pcrc;
  765. if(crcptr == NULL) {
  766. for(int8_t j = 0; j < (int8_t)dpl - view_log_decode_CRC - 1; j++) {
  767. crc = calc_crc(crc, pcrc++, 6, 8);
  768. if(crc == get_shifted_crc(pcrc)) {
  769. crcptr = pcrc;
  770. break;
  771. }
  772. }
  773. }
  774. } else {
  775. for(int8_t j = 0; j < (int8_t)dpl - view_log_decode_CRC; j++) {
  776. crc = calc_crc(crc, pcrc++, 7, 8);
  777. if((view_log_decode_CRC == 1 && crc == *pcrc) || (view_log_decode_CRC == 2 && crc == ((*pcrc<<8) | *(pcrc+1)))) {
  778. crcptr = pcrc;
  779. break;
  780. }
  781. }
  782. }
  783. }
  784. }
  785. ptr += view_log_arr_x;
  786. if(max_width < VIEW_LOG_WIDTH_B) {
  787. pre += snprintf(screen_buf + 1, 10, "%X-", pipe);
  788. if(view_log_decode_PCF) {
  789. pre += snprintf(screen_buf + strlen(screen_buf), 10, "%02X%01X-", *ptr >> 2, ((*ptr & 3) << 1) | (*(ptr+1) >> 7));
  790. }
  791. }
  792. 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);
  793. }
  794. uint16_t y = 14 + i * 7;
  795. canvas_draw_str(canvas, 3 * 5, y, screen_buf);
  796. uint16_t x = snprintf(screen_buf, sizeof(screen_buf), "%d", page + i + 1);
  797. canvas_draw_str(canvas, 0, y, screen_buf);
  798. if(crcptr) { // 5x7 font, 9 lines
  799. canvas_draw_str(canvas, x * 5, y, "=");
  800. int n = crcptr - (uint8_t*)ptr - 1;
  801. if(n > -view_log_decode_CRC && n < count) {
  802. int len;
  803. x = (4 + pre) * 5;
  804. if(n < 0) {
  805. len = view_log_decode_CRC + n;
  806. n = 0;
  807. } else {
  808. len = MIN(view_log_decode_CRC, count - n);
  809. x += n * 2 * 5;
  810. canvas_draw_line(canvas, x - 1, y, x - 1, y - 1);
  811. }
  812. canvas_draw_line(canvas, x - 1, y, n = x + len * 2 * 5 - 1, y);
  813. canvas_draw_line(canvas, n, y, n, y - 1);
  814. }
  815. } else canvas_draw_str(canvas, x * 5, y, ":");
  816. }
  817. }
  818. } else {
  819. canvas_set_font(canvas, FontBatteryPercent); // 5x7 font, 9 lines
  820. if(addrs.addr_count > 0) {
  821. snprintf(screen_buf, sizeof(screen_buf), "P0: ");
  822. add_to_str_hex_bytes(screen_buf, (char*)addrs.addr_P0, addrs.addr_len);
  823. canvas_draw_str(canvas, 0, 1 * 7, screen_buf);
  824. }
  825. if(addrs.addr_count > 1) {
  826. snprintf(screen_buf, sizeof(screen_buf), "P1: ");
  827. add_to_str_hex_bytes(screen_buf, (char*)addrs.addr_P1, addrs.addr_len);
  828. canvas_draw_str(canvas, 0, 2 * 7, screen_buf);
  829. }
  830. if(addrs.addr_count > 2) {
  831. canvas_draw_str(canvas, 0, 3 * 7, "P2: ");
  832. snprintf(screen_buf, sizeof(screen_buf), "%02X", addrs.addr_P2);
  833. canvas_draw_str(canvas, (4 + (addrs.addr_len - 1) * 2) * 5, 3 * 7, screen_buf);
  834. }
  835. if(addrs.addr_count > 3) {
  836. canvas_draw_str(canvas, 0, 4 * 7, "P3: ");
  837. snprintf(screen_buf, sizeof(screen_buf), "%02X", addrs.addr_P3);
  838. canvas_draw_str(canvas, (4 + (addrs.addr_len - 1) * 2) * 5, 4 * 7, screen_buf);
  839. }
  840. if(addrs.addr_count > 4) {
  841. canvas_draw_str(canvas, 0, 5 * 7, "P4: ");
  842. snprintf(screen_buf, sizeof(screen_buf), "%02X", addrs.addr_P4);
  843. canvas_draw_str(canvas, (4 + (addrs.addr_len - 1) * 2) * 5, 5 * 7, screen_buf);
  844. }
  845. if(addrs.addr_count > 5) {
  846. canvas_draw_str(canvas, 0, 6 * 7, "P5: ");
  847. snprintf(screen_buf, sizeof(screen_buf), "%02X", addrs.addr_P5);
  848. canvas_draw_str(canvas, (4 + (addrs.addr_len - 1) * 2) * 5, 6 * 7, screen_buf);
  849. }
  850. if(log_arr_idx) {
  851. uint8_t *ptr = APP->log_arr + view_log_arr_idx * LOG_REC_SIZE;
  852. snprintf(screen_buf, sizeof(screen_buf), "Current Ch: %d, size: %d", *ptr & 0x7F, *(ptr + 1) >> 3);
  853. canvas_draw_str(canvas, 0, 7 * 7, screen_buf);
  854. }
  855. screen_buf[0] = 'v';
  856. strcpy(screen_buf + 1, VERSION);
  857. canvas_draw_str(canvas, 108, 7, screen_buf);
  858. if(view_log_decode_PCF || view_log_decode_CRC) {
  859. strcpy(screen_buf, "Decode: ");
  860. if(view_log_decode_PCF) strcat(screen_buf, "PCF ");
  861. if(view_log_decode_CRC == 1) strcat(screen_buf, "CRC1");
  862. else if(view_log_decode_CRC == 2) strcat(screen_buf, "CRC2");
  863. canvas_draw_str(canvas, 0, 64, screen_buf);
  864. }
  865. }
  866. release_mutex((ValueMutex*)ctx, plugin_state);
  867. }
  868. int32_t nrf24scan_app(void* p) {
  869. UNUSED(p);
  870. APP = malloc(sizeof(Nrf24Scan));
  871. APP->event_queue = furi_message_queue_alloc(8, sizeof(PluginEvent));
  872. APP->plugin_state = malloc(sizeof(PluginState));
  873. ValueMutex state_mutex;
  874. if(!init_mutex(&state_mutex, APP->plugin_state, sizeof(PluginState))) {
  875. furi_message_queue_free(APP->event_queue);
  876. FURI_LOG_E(TAG, "cannot create mutex");
  877. free(APP->plugin_state);
  878. return 255;
  879. }
  880. memset((uint8_t*)&addrs, 0, sizeof(addrs));
  881. nrf24_init();
  882. // Set system callbacks
  883. APP->view_port = view_port_alloc();
  884. view_port_draw_callback_set(APP->view_port, render_callback, &state_mutex);
  885. view_port_input_callback_set(APP->view_port, input_callback, APP->event_queue);
  886. // Open GUI and register view_port
  887. APP->gui = furi_record_open(RECORD_GUI);
  888. gui_add_view_port(APP->gui, APP->view_port, GuiLayerFullscreen);
  889. APP->notification = furi_record_open(RECORD_NOTIFICATION);
  890. APP->storage = furi_record_open(RECORD_STORAGE);
  891. storage_common_mkdir(APP->storage, SCAN_APP_PATH_FOLDER);
  892. Stream* file_stream = file_stream_alloc(APP->storage);
  893. FuriString* path = furi_string_alloc();
  894. furi_string_set(path, SCAN_APP_PATH_FOLDER);
  895. furi_string_cat(path, "/");
  896. furi_string_cat(path, SNIFF_FILENAME);
  897. if(file_stream_open(file_stream, furi_string_get_cstr(path), FSAM_READ, FSOM_OPEN_EXISTING)) {
  898. uint8_t err = load_settings_file(file_stream);
  899. if(!err) strncpy(addr_file_name, furi_string_get_cstr(path) + sizeof(SCAN_APP_PATH_FOLDER), sizeof(addr_file_name));
  900. else snprintf(addr_file_name, sizeof(addr_file_name), "LOAD ERROR#%d", err);
  901. } else {
  902. strcpy(addr_file_name, "NONE");
  903. if(what_to_do == 1) {
  904. addrs.addr_P0[0] = 0;
  905. addrs.addr_P0[1] = 0xAA;
  906. addrs.addr_len = 2;
  907. addrs.addr_count = 1;
  908. view_log_decode_CRC = NRF_CRC = 2;
  909. }
  910. }
  911. file_stream_close(file_stream);
  912. stream_free(file_stream);
  913. furi_string_free(path);
  914. allocate_log_array();
  915. PluginEvent event;
  916. for(bool processing = true; processing;) {
  917. FuriStatus event_status = furi_message_queue_get(APP->event_queue, &event, 100);
  918. PluginState* plugin_state = (PluginState*)acquire_mutex_block(&state_mutex);
  919. if(event_status == FuriStatusOk) {
  920. // press events
  921. if(event.type == EventTypeKey) {
  922. //FURI_LOG_D(TAG, "Key: %d Type: %d Sec: %u", event.input.key, event.input.type, event.input.sequence);
  923. switch(event.input.key) {
  924. case InputKeyUp:
  925. if(event.input.type == InputTypeShort || event.input.type == InputTypeRepeat) {
  926. if(what_doing == 0) {
  927. if(menu_selected > 0) menu_selected--; else menu_selected = menu_selected_max;
  928. } else if(what_doing == 1) {
  929. view_log_arr_idx -= event.input.type == InputTypeRepeat ? 10 : 1;
  930. if(view_log_arr_idx >= log_arr_idx) view_log_arr_idx = 0;
  931. }
  932. }
  933. break;
  934. case InputKeyDown:
  935. if(event.input.type == InputTypeShort || event.input.type == InputTypeRepeat) {
  936. if(what_doing == 0) {
  937. if(menu_selected < menu_selected_max) menu_selected++; else menu_selected = 0;
  938. } else if(what_doing == 1) {
  939. view_log_arr_idx += event.input.type == InputTypeRepeat ? 10 : 1;
  940. if(view_log_arr_idx >= log_arr_idx) view_log_arr_idx = log_arr_idx - 1;
  941. }
  942. }
  943. break;
  944. case InputKeyLeft:
  945. if(event.input.type == InputTypeShort || event.input.type == InputTypeRepeat) {
  946. if(what_doing == 0) {
  947. switch(menu_selected) {
  948. case Menu_enter_channel:
  949. NRF_channel -= event.input.type == InputTypeRepeat ? 10 : 1;
  950. if(NRF_channel > MAX_CHANNEL) NRF_channel = MAX_CHANNEL;
  951. break;
  952. case Menu_enter_rate:
  953. NRF_Payload -= event.input.type == InputTypeRepeat ? 10 : 1;
  954. if(NRF_Payload == 0 || NRF_Payload > 32) NRF_Payload = 1;
  955. break;
  956. case Menu_enter_scan_period:
  957. find_channel_period -= event.input.type == InputTypeRepeat ? 10 : 1;
  958. if(find_channel_period < 0) find_channel_period = 0;
  959. break;
  960. case Menu_log:
  961. if(--log_to_file < -1) log_to_file = 2;
  962. break;
  963. case Menu_ok:
  964. if(++what_to_do > 2) what_to_do = 0;
  965. break;
  966. }
  967. } else if(what_doing == 1) {
  968. if(view_log_arr_x > 0) view_log_arr_x--;
  969. } else if(what_doing == 2) {
  970. //if(NRF_ESB == 0)
  971. view_log_decode_PCF ^= 1;
  972. }
  973. }
  974. break;
  975. case InputKeyRight:
  976. if(event.input.type == InputTypeShort || event.input.type == InputTypeRepeat) {
  977. if(what_doing == 0) {
  978. switch(menu_selected) {
  979. case Menu_open_file:
  980. save_settings ^= 1;
  981. break;
  982. case Menu_enter_channel:
  983. NRF_channel += event.input.type == InputTypeRepeat ? 10 : 1;
  984. if(NRF_channel > MAX_CHANNEL) NRF_channel = 0;
  985. break;
  986. case Menu_enter_rate:
  987. NRF_Payload += event.input.type == InputTypeRepeat ? 10 : 1;
  988. if(NRF_Payload == 0 || NRF_Payload > 32) NRF_Payload = 32;
  989. break;
  990. case Menu_enter_scan_period:
  991. find_channel_period += event.input.type == InputTypeRepeat ? 10 : 1;
  992. break;
  993. case Menu_log:
  994. if(++log_to_file > 2) log_to_file = -1;
  995. break;
  996. case Menu_ok:
  997. if(++what_to_do > 2) what_to_do = 0;
  998. break;
  999. }
  1000. } else if(what_doing == 1) {
  1001. if(view_log_arr_x < VIEW_LOG_MAX_X) view_log_arr_x++;
  1002. } else if(what_doing == 2) {
  1003. if(++view_log_decode_CRC > 2) view_log_decode_CRC = 0;
  1004. }
  1005. }
  1006. break;
  1007. case InputKeyOk:
  1008. if(event.input.type == InputTypeShort) {
  1009. if(what_doing == 0) {
  1010. switch(menu_selected) {
  1011. case Menu_open_file:
  1012. if(save_settings) {
  1013. write_to_log_file(APP->storage, true);
  1014. } else {
  1015. file_stream = file_stream_alloc(APP->storage);
  1016. if(select_settings_file(file_stream)) {
  1017. uint8_t err = load_settings_file(file_stream);
  1018. if(!err) save_to_new_log = true; else snprintf(addr_file_name, sizeof(addr_file_name), "LOAD ERROR#%d", err);
  1019. file_stream_close(file_stream);
  1020. }
  1021. stream_free(file_stream);
  1022. }
  1023. break;
  1024. case Menu_enter_channel:
  1025. if(NRF_ESB) {
  1026. if(NRF_DPL) NRF_DPL = NRF_ESB = 0; else NRF_DPL = 1;
  1027. } else NRF_ESB = 1;
  1028. //if(NRF_ESB) view_log_decode_PCF = 0;
  1029. break;
  1030. case Menu_enter_rate:
  1031. NRF_rate++;
  1032. if(NRF_rate > 2) NRF_rate = 0;
  1033. break;
  1034. case Menu_enter_scan_period:
  1035. if(++NRF_CRC > 2) NRF_CRC = what_to_do == 1 ? 1 : 0;
  1036. break;
  1037. case Menu_ok:
  1038. if(what_to_do) {
  1039. if(addrs.addr_count || what_to_do == 1) {
  1040. if(log_to_file == -1) {
  1041. log_to_file = 0;
  1042. clear_log();
  1043. save_to_new_log = true;
  1044. } else if(log_to_file == 1) save_to_new_log = true;
  1045. start_scanning();
  1046. if(!NRF_ERROR) what_doing = 1;
  1047. }
  1048. } else what_doing = 1;
  1049. key_press_seq_ok = event.input.sequence;
  1050. break;
  1051. }
  1052. } else if(what_doing == 1) {
  1053. nrf24_send_packet();
  1054. }
  1055. } else if(event.input.type == InputTypeLong) {
  1056. if(what_doing == 0) {
  1057. if(menu_selected == Menu_enter_channel) {
  1058. NRF_AA_OFF ^= 1;
  1059. key_press_seq_ok = event.input.sequence;
  1060. } else if(menu_selected == Menu_log) { // Log
  1061. if(log_arr_idx && (log_to_file == 1 || log_to_file == 2)) {
  1062. write_to_log_file(APP->storage, false);
  1063. clear_log();
  1064. }
  1065. }
  1066. } else if(what_doing == 1) {
  1067. what_doing = 2;
  1068. }
  1069. }
  1070. break;
  1071. case InputKeyBack:
  1072. if(event.input.type == InputTypeLong) processing = false;
  1073. else if(event.input.type == InputTypeShort) {
  1074. if(what_doing) what_doing--;
  1075. if(what_doing == 0) nrf24_set_idle(nrf24_HANDLE);;
  1076. }
  1077. break;
  1078. default:
  1079. break;
  1080. }
  1081. }
  1082. }
  1083. if(what_doing && what_to_do) {
  1084. nrf24_read_newpacket();
  1085. if(find_channel_period && furi_get_tick() - start_time >= (uint32_t)find_channel_period * 1000UL) {
  1086. if(++NRF_channel > MAX_CHANNEL) NRF_channel = 0;
  1087. start_scanning();
  1088. }
  1089. }
  1090. view_port_update(APP->view_port);
  1091. release_mutex(&state_mutex, plugin_state);
  1092. }
  1093. nrf24_set_idle(nrf24_HANDLE);
  1094. if(log_arr_idx && (log_to_file == 1 || log_to_file == 2)) {
  1095. write_to_log_file(APP->storage, false);
  1096. }
  1097. nrf24_deinit();
  1098. view_port_enabled_set(APP->view_port, false);
  1099. gui_remove_view_port(APP->gui, APP->view_port);
  1100. furi_record_close(RECORD_GUI);
  1101. furi_record_close(RECORD_NOTIFICATION);
  1102. furi_record_close(RECORD_STORAGE);
  1103. view_port_free(APP->view_port);
  1104. furi_message_queue_free(APP->event_queue);
  1105. free(APP->plugin_state);
  1106. if(APP->log_arr) free(APP->log_arr);
  1107. free(APP);
  1108. return 0;
  1109. }