nrf24scan.c 43 KB

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