nrf24scan.c 38 KB

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