nrf24scan.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655
  1. //
  2. // Written by vad7, 20.11.2022.
  3. // ver. 1.0
  4. //
  5. #include "nrf24scan.h"
  6. #include <furi.h>
  7. #include <furi_hal.h>
  8. #include <gui/gui.h>
  9. #include <dialogs/dialogs.h>
  10. #include <input/input.h>
  11. #include <stdlib.h>
  12. #include <dolphin/dolphin.h>
  13. #include <nrf24.h>
  14. #define TAG "nrf24scan"
  15. #define MAX_CHANNEL 125
  16. #define MAX_ADDR 6
  17. #define SCAN_APP_PATH_FOLDER "/ext/nrf24scan"
  18. #define ADDR_FILENAME "addr.txt" // File format (1 parameter per line):
  19. // 0.25/1/2 - rate in Mbps
  20. // 0..125 - default channel
  21. // address P0 in hex (5 byte, LSB last)
  22. // address P1 in hex (5 byte, LSB last)
  23. // address P2, LSB in hex (1 byte)
  24. // address P3, LSB in hex (1 byte)
  25. // address P4, LSB in hex (1 byte)
  26. // address P5, LSB in hex (1 byte)
  27. // captured data in raw format, first byte = addr # 0..5
  28. // ... up to MAX_LOG_RECORDS
  29. #define LOG_FILENAME "log"
  30. #define LOG_FILEEXT ".txt"
  31. #define MAX_LOG_RECORDS 100
  32. #define LOG_REC_SIZE 33 // max packet size
  33. #define VIEW_LOG_MAX_X 22
  34. Nrf24Scan* APP;
  35. uint8_t what_doing = 0; // 0 - setup, 1 - view log
  36. uint8_t what_to_do = 1; // 0 - view, 1 - view & scan
  37. char screen_buf[64];
  38. char addr_file_name[32];
  39. uint8_t NRF_rate; // 0 - 250Kbps, 1 - 1Mbps, 2 - 2Mbps
  40. uint8_t NRF_channel; // 0..125
  41. struct {
  42. uint8_t addr_P0[5]; // MSB first
  43. uint8_t addr_P1[5]; // MSB first
  44. uint8_t addr_P2; // LSB only, MSB bytes equal addr_P1
  45. uint8_t addr_P3; // LSB only, MSB bytes equal addr_P1
  46. uint8_t addr_P4; // LSB only, MSB bytes equal addr_P1
  47. uint8_t addr_P5; // LSB only, MSB bytes equal addr_P1
  48. uint8_t addr_len; // 2..5
  49. uint8_t addr_count;
  50. } addrs;
  51. int8_t log_to_file = 0; // 0 - no, 1 - yes(new), 2 - append, -1 - only clear
  52. uint16_t log_arr_idx;
  53. uint16_t view_log_arr_idx = 0;
  54. uint16_t view_log_arr_x = 0;
  55. bool save_to_new_log = true;
  56. uint16_t last_packet_send = -1;
  57. uint8_t last_packet_send_st = 0;
  58. int16_t find_channel_period = 0; // sec
  59. uint8_t menu_selected = 0;
  60. uint32_t start_time;
  61. #define menu_selected_max 5
  62. enum {
  63. Menu_open_file = 0,
  64. Menu_enter_channel,
  65. Menu_enter_rate,
  66. Menu_enter_scan_period,
  67. Menu_log,
  68. Menu_ok
  69. };
  70. static uint8_t GetHexVal(char hex) {
  71. return (uint8_t)hex - ((uint8_t)hex < 58 ? 48 : ((uint8_t)hex < 97 ? 55 : 87));
  72. }
  73. // Return num bytes in array
  74. static uint8_t ConvertHexToArray(char * hex, uint8_t * array, uint8_t maxlen) {
  75. uint8_t len = 0;
  76. do {
  77. uint8_t ch = *hex++;
  78. if(ch == 0) break;
  79. if(ch < '0') continue;
  80. *array++ = (GetHexVal(ch) << 4) + GetHexVal(*hex++);
  81. len++;
  82. } while(--maxlen);
  83. return len;
  84. }
  85. static void add_to_str_hex_bytes(char *out, char *arr, int bytes)
  86. {
  87. if(!bytes) return;
  88. out += strlen(out);
  89. do {
  90. snprintf(out, 3, "%02X", *arr++);
  91. out += 2;
  92. } while(--bytes);
  93. }
  94. static void add_to_furi_str_hex_bytes(FuriString* str, char *arr, int bytes)
  95. {
  96. if(!bytes) return;
  97. do {
  98. furi_string_cat_printf(str, "%02X", *arr++);
  99. } while(--bytes);
  100. }
  101. void clear_log()
  102. {
  103. log_arr_idx = 0;
  104. view_log_arr_idx = 0;
  105. last_packet_send = -1;
  106. }
  107. void allocate_log_array()
  108. {
  109. APP->log_arr = malloc(LOG_REC_SIZE * MAX_LOG_RECORDS);
  110. if(APP->log_arr == NULL) {
  111. FURI_LOG_E(TAG, "Not enouch memory: %d", LOG_REC_SIZE * MAX_LOG_RECORDS);
  112. strcpy(addr_file_name, "MEMORY LOW!");
  113. }
  114. clear_log();
  115. }
  116. void write_to_log_file(Storage* storage)
  117. {
  118. if(log_arr_idx == 0) return;
  119. Stream* file_stream = file_stream_alloc(storage);
  120. FuriString* str = furi_string_alloc();
  121. furi_string_set(str, SCAN_APP_PATH_FOLDER);
  122. furi_string_cat(str, "/");
  123. furi_string_cat(str, LOG_FILENAME);
  124. furi_string_cat(str, LOG_FILEEXT);
  125. bool fl;
  126. if(save_to_new_log) {
  127. int cnt = 1;
  128. do {
  129. fl = file_stream_open(file_stream, furi_string_get_cstr(str), FSAM_READ_WRITE, FSOM_CREATE_NEW);
  130. if(fl) break;
  131. furi_string_set(str, SCAN_APP_PATH_FOLDER);
  132. furi_string_cat(str, "/");
  133. furi_string_cat(str, LOG_FILENAME);
  134. furi_string_cat_printf(str, "-%02d", cnt);
  135. furi_string_cat(str, LOG_FILEEXT);
  136. } while(++cnt < 100);
  137. if(!fl) {
  138. FURI_LOG_E(TAG, "Failed to create new log file");
  139. notification_message(APP->notification, &sequence_blink_red_100);
  140. }
  141. } else {
  142. fl = file_stream_open(file_stream, furi_string_get_cstr(str), FSAM_READ_WRITE, FSOM_OPEN_APPEND);
  143. if(stream_size(file_stream) == 0) save_to_new_log = true;
  144. }
  145. if(fl) {
  146. FURI_LOG_D(TAG, "Save to log %s", furi_string_get_cstr(str));
  147. if(save_to_new_log) {
  148. furi_string_printf(str, "%d\n%d\n", NRF_rate, NRF_channel);
  149. add_to_furi_str_hex_bytes(str, (char*)addrs.addr_P0, addrs.addr_len); furi_string_cat(str, "\n");
  150. if(addrs.addr_count > 1) { add_to_furi_str_hex_bytes(str, (char*)addrs.addr_P1, addrs.addr_len); furi_string_cat(str, "\n"); }
  151. if(addrs.addr_count > 2) furi_string_cat_printf(str, "%02X\n", addrs.addr_P2);
  152. if(addrs.addr_count > 3) furi_string_cat_printf(str, "%02X\n", addrs.addr_P3);
  153. if(addrs.addr_count > 4) furi_string_cat_printf(str, "%02X\n", addrs.addr_P4);
  154. if(addrs.addr_count > 5) furi_string_cat_printf(str, "%02X\n", addrs.addr_P5);
  155. if(!(fl = stream_write_string(file_stream, str) == furi_string_size(str))) {
  156. FURI_LOG_E(TAG, "Failed to write header to log!");
  157. notification_message(APP->notification, &sequence_blink_red_100);
  158. }
  159. }
  160. if(fl) {
  161. int i = 0;
  162. for(; i < log_arr_idx; i++) {
  163. furi_string_reset(str);
  164. add_to_furi_str_hex_bytes(str, (char*)APP->log_arr + i * LOG_REC_SIZE, LOG_REC_SIZE);
  165. furi_string_cat(str, "\n");
  166. if(stream_write_string(file_stream, str) != furi_string_size(str)) {
  167. FURI_LOG_E(TAG, "Failed to write to log!");
  168. break;
  169. }
  170. }
  171. if(i == log_arr_idx) {
  172. notification_message(APP->notification, &sequence_blink_yellow_100);
  173. FURI_LOG_D(TAG, "Log saved");
  174. }
  175. }
  176. save_to_new_log = false;
  177. } else {
  178. FURI_LOG_E(TAG, "Failed to open log file");
  179. notification_message(APP->notification, &sequence_blink_red_100);
  180. }
  181. //file_stream_close(file_stream);
  182. stream_free(file_stream);
  183. furi_string_free(str);
  184. }
  185. static bool select_settings_file(Stream* stream) {
  186. DialogsApp* dialogs = furi_record_open("dialogs");
  187. bool result = false;
  188. FuriString* path;
  189. path = furi_string_alloc();
  190. furi_string_set(path, SCAN_APP_PATH_FOLDER);
  191. DialogsFileBrowserOptions browser_options;
  192. dialog_file_browser_set_basic_options(&browser_options, ".txt", NULL);
  193. browser_options.hide_ext = false;
  194. bool ret = dialog_file_browser_show(dialogs, path, path, &browser_options);
  195. furi_record_close("dialogs");
  196. if(ret) {
  197. if(!file_stream_open(stream, furi_string_get_cstr(path), FSAM_READ, FSOM_OPEN_EXISTING)) {
  198. //file_stream_close(stream);
  199. FURI_LOG_D(TAG, "Cannot open file \"%s\"", furi_string_get_cstr(path));
  200. } else {
  201. FURI_LOG_D(TAG, "Open file \"%s\"", furi_string_get_cstr(path));
  202. strncpy(addr_file_name, furi_string_get_cstr(path) + sizeof(SCAN_APP_PATH_FOLDER), sizeof(addr_file_name));
  203. result = true;
  204. }
  205. }
  206. furi_string_free(path);
  207. return result;
  208. }
  209. // True - success
  210. static bool load_settings_file(Stream* file_stream) {
  211. size_t file_size = 0;
  212. char* file_buf;
  213. bool loaded = false;
  214. file_size = stream_size(file_stream);
  215. if(file_size == (size_t)0) {
  216. FURI_LOG_D(TAG, "load failed. file_size: %d", file_size);
  217. return loaded;
  218. }
  219. file_buf = malloc(file_size);
  220. memset(file_buf, 0, file_size);
  221. if(stream_read(file_stream, (uint8_t*)file_buf, file_size) == file_size) {
  222. FURI_LOG_D(TAG, "Loading settings file");
  223. char* line_ptr = file_buf;
  224. int16_t line_num = 0;
  225. memset((uint8_t*)&addrs, 0, sizeof(addrs));
  226. bool log_loaded = false;
  227. while(line_ptr && line_ptr - file_buf < file_size) {
  228. char* end_ptr = strstr((char*)line_ptr, "\n");
  229. if(end_ptr == NULL) end_ptr = file_buf + file_size; else *end_ptr = '\0';
  230. FURI_LOG_D(TAG, " L#%d: [%d]%s", line_num, end_ptr - line_ptr, line_ptr);
  231. int line_len = end_ptr - line_ptr;
  232. if(*line_ptr == '\r' || line_len == 0) {
  233. line_ptr = end_ptr + 1;
  234. continue;
  235. }
  236. if(*(end_ptr - 1) < '0') {
  237. *(end_ptr - 1) = '\0';
  238. line_len--;
  239. }
  240. if(line_len >= LOG_REC_SIZE * 2) { // data
  241. if(!log_loaded) {
  242. clear_log();
  243. what_to_do = 0;
  244. log_loaded = true;
  245. }
  246. if(log_arr_idx < MAX_LOG_RECORDS - 1) {
  247. ConvertHexToArray(line_ptr, APP->log_arr + log_arr_idx * LOG_REC_SIZE, LOG_REC_SIZE);
  248. log_arr_idx++;
  249. }
  250. } else if(addrs.addr_count) {
  251. ConvertHexToArray(line_ptr, addrs.addr_count==1? &addrs.addr_P1[0] : addrs.addr_count==2? &addrs.addr_P2 : addrs.addr_count==3? &addrs.addr_P3 : addrs.addr_count==4? &addrs.addr_P4: &addrs.addr_P5, addrs.addr_count==1? 5 : 1);
  252. FURI_LOG_D(TAG, " +Addr_LSB: %s", line_ptr);
  253. if(++addrs.addr_count == MAX_ADDR) break;
  254. } else if(line_len < 4) { // Rate or Channel
  255. if(line_num == 0) { // 1st line - Rate
  256. NRF_rate = atoi(line_ptr);
  257. FURI_LOG_D(TAG, " Rate: %d", NRF_rate);
  258. } else if(line_num == 1) { // second line - Channel
  259. NRF_channel = atoi(line_ptr);
  260. FURI_LOG_D(TAG, " Ch: %d", NRF_channel);
  261. }
  262. } else if(line_len <= 5*2){ // addresses
  263. addrs.addr_len = ConvertHexToArray(line_ptr, addrs.addr_P0, 5);
  264. FURI_LOG_D(TAG, " +Addr(%d): %02X%02X%02X...", addrs.addr_len, addrs.addr_P0[0], addrs.addr_P0[1], addrs.addr_P0[2]);
  265. loaded = true;
  266. if(++addrs.addr_count == MAX_ADDR) break;
  267. }
  268. line_ptr = end_ptr + 1;
  269. line_num++;
  270. }
  271. } else {
  272. FURI_LOG_D(TAG, "load failed. file size: %d", file_size);
  273. }
  274. free(file_buf);
  275. return loaded;
  276. }
  277. static void input_callback(InputEvent* input_event, FuriMessageQueue* event_queue) {
  278. furi_assert(event_queue);
  279. PluginEvent event = {.type = EventTypeKey, .input = *input_event};
  280. furi_message_queue_put(event_queue, &event, FuriWaitForever);
  281. }
  282. static void prepare_nrf24()
  283. {
  284. uint8_t addr[5];
  285. uint8_t erx_addr = (1<<0); // Enable RX_P0
  286. if(addrs.addr_count == 0) return;
  287. nrf24_write_reg(nrf24_HANDLE, REG_CONFIG, 0x00); // Stop nRF
  288. nrf24_write_reg(nrf24_HANDLE, REG_STATUS, 0x70); // clear interrupts
  289. nrf24_write_reg(nrf24_HANDLE, REG_DYNPD, 0x0); // disable shockburst
  290. nrf24_write_reg(nrf24_HANDLE, REG_EN_AA, 0x00); // Disable Shockburst
  291. nrf24_write_reg(nrf24_HANDLE, REG_FEATURE, 0x01); // Enables the W_TX_PAYLOAD_NOACK command, Disable Payload with ACK, Disable Dynamic Payload Length
  292. nrf24_set_maclen(nrf24_HANDLE, addrs.addr_len);
  293. for(int i = 0; i < addrs.addr_len; i++) addr[i] = addrs.addr_P0[addrs.addr_len - i - 1];
  294. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P0, &addr[0], addrs.addr_len);
  295. nrf24_write_reg(nrf24_HANDLE, RX_PW_P0, 32);
  296. if(addrs.addr_count > 1) {
  297. for(int i = 0; i < addrs.addr_len; i++) addr[i] = addrs.addr_P1[addrs.addr_len - i - 1];
  298. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P1, &addr[0], addrs.addr_len);
  299. nrf24_write_reg(nrf24_HANDLE, RX_PW_P1, 32);
  300. erx_addr |= (1<<1); // Enable RX_P1
  301. } else nrf24_write_reg(nrf24_HANDLE, RX_PW_P1, 0);
  302. if(addrs.addr_count > 2) {
  303. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P2, &addrs.addr_P2, 1);
  304. nrf24_write_reg(nrf24_HANDLE, RX_PW_P2, 32);
  305. erx_addr |= (1<<2); // Enable RX_P2
  306. } else nrf24_write_reg(nrf24_HANDLE, RX_PW_P2, 0);
  307. if(addrs.addr_count > 3) {
  308. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P3, &addrs.addr_P3, 1);
  309. nrf24_write_reg(nrf24_HANDLE, RX_PW_P3, 32);
  310. erx_addr |= (1<<3); // Enable RX_P3
  311. } else nrf24_write_reg(nrf24_HANDLE, RX_PW_P3, 0);
  312. if(addrs.addr_count > 4) {
  313. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P4, &addrs.addr_P4, 1);
  314. nrf24_write_reg(nrf24_HANDLE, RX_PW_P4, 32);
  315. erx_addr |= (1<<4); // Enable RX_P4
  316. } else nrf24_write_reg(nrf24_HANDLE, RX_PW_P4, 0);
  317. if(addrs.addr_count > 5) {
  318. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P5, &addrs.addr_P5, 1);
  319. nrf24_write_reg(nrf24_HANDLE, RX_PW_P5, 32);
  320. erx_addr |= (1<<5); // Enable RX_P5
  321. } else nrf24_write_reg(nrf24_HANDLE, RX_PW_P5, 0);
  322. nrf24_write_reg(nrf24_HANDLE, REG_STATUS, 0x50); // clear RX_DR, MAX_RT.
  323. nrf24_set_idle(nrf24_HANDLE);
  324. nrf24_flush_rx(nrf24_HANDLE);
  325. nrf24_flush_tx(nrf24_HANDLE);
  326. nrf24_write_reg(nrf24_HANDLE, REG_EN_RXADDR, erx_addr);
  327. nrf24_write_reg(nrf24_HANDLE, REG_RF_CH, NRF_channel);
  328. nrf24_write_reg(nrf24_HANDLE, REG_RF_SETUP, NRF_rate);
  329. // prime for RX, no checksum
  330. nrf24_write_reg(nrf24_HANDLE, REG_CONFIG, 0x01); // disable AA and CRC
  331. }
  332. static void start_scanning()
  333. {
  334. prepare_nrf24();
  335. // prime for RX, no checksum
  336. nrf24_write_reg(nrf24_HANDLE, REG_CONFIG, 0x03); // PWR_UP and PRIM_RX, disable AA and CRC
  337. furi_hal_gpio_write(nrf24_CE_PIN, true);
  338. furi_delay_ms(100);
  339. start_time = furi_get_tick();
  340. FURI_LOG_D(TAG, "Start scan: Ch=%d Rate=%d", NRF_channel, NRF_rate);
  341. }
  342. bool nrf24_read_newpacket() {
  343. if(APP->log_arr == NULL) return false;
  344. bool found = false;
  345. uint8_t packetsize;
  346. uint8_t packet[32] = {0};
  347. uint8_t status = nrf24_rxpacket(nrf24_HANDLE, packet, &packetsize, true);
  348. if(status & 0x40) {
  349. uint8_t *ptr = APP->log_arr + log_arr_idx * LOG_REC_SIZE;
  350. *ptr++ = (status >> 1) & 7; // pipe #
  351. memcpy(ptr, packet, packetsize);
  352. if(packetsize < 32) memset(ptr + packetsize, 0, 32 - packetsize);
  353. if(log_arr_idx < MAX_LOG_RECORDS - 1) {
  354. log_arr_idx++;
  355. } else {
  356. if(log_to_file == 1 || log_to_file == 2) {
  357. write_to_log_file(APP->storage);
  358. clear_log();
  359. } else {
  360. memcpy((void*)APP->log_arr, (void*)APP->log_arr + LOG_REC_SIZE, log_arr_idx * LOG_REC_SIZE);
  361. }
  362. }
  363. FURI_LOG_D(TAG, "Found packet #%d pipe %d", log_arr_idx, (status >> 1) & 7);
  364. notification_message(APP->notification, &sequence_blink_white_100);
  365. found = true;
  366. }
  367. return found;
  368. }
  369. bool nrf24_send_packet()
  370. {
  371. if(!what_to_do) prepare_nrf24();
  372. if(log_arr_idx == 0) return false;
  373. last_packet_send_st = nrf24_txpacket(nrf24_HANDLE, APP->log_arr + view_log_arr_idx * LOG_REC_SIZE + 1, 32, false);
  374. last_packet_send = view_log_arr_idx;
  375. notification_message(APP->notification, last_packet_send_st ? &sequence_blink_blue_100 : &sequence_blink_red_100);
  376. if(what_to_do) start_scanning();
  377. return last_packet_send_st;
  378. }
  379. static void render_callback(Canvas* const canvas, void* ctx) {
  380. const PluginState* plugin_state = acquire_mutex((ValueMutex*)ctx, 25);
  381. if(plugin_state == NULL) return;
  382. //canvas_draw_frame(canvas, 0, 0, 128, 64); // border around the edge of the screen
  383. if(what_doing == 0) {
  384. canvas_set_font(canvas, FontSecondary); // 8x10 font
  385. snprintf(screen_buf, sizeof(screen_buf), "Settings: %s", addr_file_name); // menu_selected = 0
  386. canvas_draw_str(canvas, 10, 10, screen_buf);
  387. snprintf(screen_buf, sizeof(screen_buf), "Channel: %d", NRF_channel); // menu_selected = 1
  388. canvas_draw_str(canvas, 10, 20, screen_buf);
  389. snprintf(screen_buf, sizeof(screen_buf), "Rate: %sbps", NRF_rate == 2 ? "2M" : NRF_rate == 1 ? "1M" : "250K"); // menu_selected = 2
  390. canvas_draw_str(canvas, 10, 30, screen_buf);
  391. strcpy(screen_buf, "Find channel period: "); // menu_selected = 3
  392. if(find_channel_period == 0) strcat(screen_buf, "off"); else snprintf(screen_buf + strlen(screen_buf), sizeof(screen_buf), "%d s", find_channel_period);
  393. canvas_draw_str(canvas, 10, 40, screen_buf);
  394. 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
  395. canvas_draw_str(canvas, 10, 50, screen_buf);
  396. snprintf(screen_buf, sizeof(screen_buf), "%s (pipes: %d)", what_to_do ? "Start scan" : "View log", addrs.addr_count); // menu_selected = 5
  397. canvas_draw_str(canvas, 10, 60, screen_buf);
  398. canvas_draw_str(canvas, 0, menu_selected * 10 + 10, ">");
  399. } else {
  400. canvas_set_font(canvas, FontBatteryPercent); // 5x7 font, 9 lines
  401. bool ch2 = false;
  402. screen_buf[0] = '\0';
  403. if(view_log_arr_x == 0) { strcat(screen_buf, " "); ch2 = true;
  404. } else {
  405. snprintf(screen_buf, sizeof(screen_buf), "<%d", view_log_arr_x);
  406. if(view_log_arr_x < VIEW_LOG_MAX_X) ch2 = true;
  407. }
  408. snprintf(screen_buf + strlen(screen_buf), sizeof(screen_buf), " %s ch: %d - %d.", what_to_do ? "Read" : "View", NRF_channel, log_arr_idx);
  409. canvas_draw_str(canvas, 0, 7, screen_buf);
  410. if(ch2) canvas_draw_str(canvas, 121, 7, ">");
  411. if(log_arr_idx) {
  412. uint16_t page = view_log_arr_idx & ~7;
  413. for(uint8_t i = 0; i < 8 && page + i < log_arr_idx; i++) {
  414. snprintf(screen_buf, sizeof(screen_buf), "%d:%c", page + i + 1, (view_log_arr_idx & 7) != i ? ' ' : last_packet_send != view_log_arr_idx ? '>' : last_packet_send_st ? '+' : '!');
  415. char *ptr = (char*)APP->log_arr + (page + i) * LOG_REC_SIZE + view_log_arr_x;
  416. if(view_log_arr_x == 0 && addrs.addr_count > 1) {
  417. snprintf(screen_buf + strlen(screen_buf), sizeof(screen_buf), "%d-", *ptr + 1);
  418. add_to_str_hex_bytes(screen_buf, ptr + 1, 9);
  419. } else {
  420. add_to_str_hex_bytes(screen_buf, ptr + 1, 10);
  421. }
  422. canvas_draw_str(canvas, 0, 14 + i * 7, screen_buf);
  423. }
  424. }
  425. }
  426. release_mutex((ValueMutex*)ctx, plugin_state);
  427. }
  428. int32_t nrf24scan_app(void* p) {
  429. UNUSED(p);
  430. APP = malloc(sizeof(Nrf24Scan));
  431. APP->event_queue = furi_message_queue_alloc(8, sizeof(PluginEvent));
  432. APP->plugin_state = malloc(sizeof(PluginState));
  433. ValueMutex state_mutex;
  434. if(!init_mutex(&state_mutex, APP->plugin_state, sizeof(PluginState))) {
  435. furi_message_queue_free(APP->event_queue);
  436. FURI_LOG_E(TAG, "cannot create mutex\r\n");
  437. free(APP->plugin_state);
  438. return 255;
  439. }
  440. memset((uint8_t*)&addrs, 0, sizeof(addrs));
  441. nrf24_init();
  442. // Set system callbacks
  443. APP->view_port = view_port_alloc();
  444. view_port_draw_callback_set(APP->view_port, render_callback, &state_mutex);
  445. view_port_input_callback_set(APP->view_port, input_callback, APP->event_queue);
  446. // Open GUI and register view_port
  447. APP->gui = furi_record_open(RECORD_GUI);
  448. gui_add_view_port(APP->gui, APP->view_port, GuiLayerFullscreen);
  449. APP->notification = furi_record_open(RECORD_NOTIFICATION);
  450. APP->storage = furi_record_open(RECORD_STORAGE);
  451. storage_common_mkdir(APP->storage, SCAN_APP_PATH_FOLDER);
  452. Stream* file_stream = file_stream_alloc(APP->storage);
  453. FuriString* path = furi_string_alloc();
  454. furi_string_set(path, SCAN_APP_PATH_FOLDER);
  455. furi_string_cat(path, "/");
  456. furi_string_cat(path, ADDR_FILENAME);
  457. if(file_stream_open(file_stream, furi_string_get_cstr(path), FSAM_READ, FSOM_OPEN_EXISTING)) {
  458. if(load_settings_file(file_stream)) strcpy(addr_file_name, ADDR_FILENAME); else strcpy(addr_file_name, "LOAD ERROR");
  459. } else {
  460. strcpy(addr_file_name, "NONE");
  461. }
  462. //file_stream_close(file_stream);
  463. stream_free(file_stream);
  464. furi_string_free(path);
  465. allocate_log_array();
  466. PluginEvent event;
  467. for(bool processing = true; processing;) {
  468. FuriStatus event_status = furi_message_queue_get(APP->event_queue, &event, 100);
  469. PluginState* plugin_state = (PluginState*)acquire_mutex_block(&state_mutex);
  470. if(event_status == FuriStatusOk) {
  471. // press events
  472. if(event.type == EventTypeKey) {
  473. switch(event.input.key) {
  474. case InputKeyUp:
  475. if(event.input.type == InputTypePress || event.input.type == InputTypeRepeat) {
  476. if(what_doing == 0) {
  477. if(menu_selected > 0) menu_selected--; else menu_selected = menu_selected_max;
  478. } else {
  479. view_log_arr_idx -= event.input.type == InputTypeRepeat ? 10 : 1;
  480. if(view_log_arr_idx >= log_arr_idx) view_log_arr_idx = 0;
  481. }
  482. }
  483. break;
  484. case InputKeyDown:
  485. if(event.input.type == InputTypePress || event.input.type == InputTypeRepeat) {
  486. if(what_doing == 0) {
  487. if(menu_selected < menu_selected_max) menu_selected++; else menu_selected = 0;
  488. } else {
  489. view_log_arr_idx += event.input.type == InputTypeRepeat ? 10 : 1;
  490. if(view_log_arr_idx >= log_arr_idx) view_log_arr_idx = log_arr_idx - 1;
  491. }
  492. }
  493. break;
  494. case InputKeyRight:
  495. if(event.input.type == InputTypePress || event.input.type == InputTypeRepeat) {
  496. if(what_doing == 0) {
  497. switch(menu_selected) {
  498. case Menu_enter_channel:
  499. NRF_channel += event.input.type == InputTypeRepeat ? 10 : 1;
  500. if(NRF_channel > MAX_CHANNEL) NRF_channel = 0;
  501. break;
  502. case Menu_enter_rate:
  503. NRF_rate++;
  504. if(NRF_rate > 2) NRF_rate = 0;
  505. break;
  506. case Menu_enter_scan_period:
  507. find_channel_period += event.input.type == InputTypeRepeat ? 10 : 1;
  508. break;
  509. case Menu_log:
  510. if(++log_to_file > 2) log_to_file = -1;
  511. break;
  512. case Menu_ok:
  513. what_to_do = !what_to_do;
  514. break;
  515. }
  516. } else {
  517. if(view_log_arr_x < VIEW_LOG_MAX_X) view_log_arr_x++;
  518. }
  519. }
  520. break;
  521. case InputKeyLeft:
  522. if(event.input.type == InputTypePress || event.input.type == InputTypeRepeat) {
  523. if(what_doing == 0) {
  524. switch(menu_selected) {
  525. case Menu_enter_channel:
  526. NRF_channel -= event.input.type == InputTypeRepeat ? 10 : 1;
  527. if(NRF_channel > MAX_CHANNEL) NRF_channel = MAX_CHANNEL;
  528. break;
  529. case Menu_enter_rate:
  530. NRF_rate--;
  531. if(NRF_rate > 2) NRF_rate = 2;
  532. break;
  533. case Menu_enter_scan_period:
  534. find_channel_period -= event.input.type == InputTypeRepeat ? 10 : 1;
  535. if(find_channel_period < 0) find_channel_period = 0;
  536. break;
  537. case Menu_log:
  538. if(--log_to_file < -1) log_to_file = 2;
  539. break;
  540. case Menu_ok:
  541. what_to_do = !what_to_do;
  542. break;
  543. }
  544. } else {
  545. if(view_log_arr_x > 0) view_log_arr_x--;
  546. }
  547. }
  548. break;
  549. case InputKeyOk:
  550. if(event.input.type == InputTypePress) {
  551. if(what_doing == 0) {
  552. if(menu_selected == Menu_open_file) { // File
  553. file_stream = file_stream_alloc(APP->storage);
  554. if(select_settings_file(file_stream)) {
  555. if(load_settings_file(file_stream)) save_to_new_log = true; else strcpy(addr_file_name, "LOAD ERROR");
  556. //file_stream_close(file_stream);
  557. }
  558. stream_free(file_stream);
  559. } else if(menu_selected == Menu_ok) { // Ok
  560. what_doing = !what_doing;
  561. if(what_doing) {
  562. if(what_to_do) {
  563. if(addrs.addr_count == 0) what_doing = 0;
  564. else {
  565. if(log_to_file == -1) {
  566. clear_log();
  567. save_to_new_log = true;
  568. } else if(log_to_file == 1) save_to_new_log = true;
  569. start_scanning();
  570. }
  571. }
  572. } else nrf24_set_idle(nrf24_HANDLE);
  573. }
  574. } else { // Send
  575. nrf24_send_packet();
  576. }
  577. } else if(event.input.type == InputTypeLong) {
  578. if(!what_doing) {
  579. if(menu_selected == Menu_log) { // Log
  580. if(log_arr_idx && (log_to_file == 1 || log_to_file == 2)) {
  581. write_to_log_file(APP->storage);
  582. clear_log();
  583. }
  584. }
  585. nrf24_set_idle(nrf24_HANDLE);
  586. }
  587. }
  588. break;
  589. case InputKeyBack:
  590. if(event.input.type == InputTypeLong) processing = false; else what_doing = 0;
  591. nrf24_set_idle(nrf24_HANDLE);
  592. break;
  593. default:
  594. break;
  595. }
  596. }
  597. }
  598. if(what_to_do) {
  599. nrf24_read_newpacket();
  600. if(find_channel_period && furi_get_tick() - start_time >= (uint32_t)find_channel_period * 1000UL) {
  601. if(++NRF_channel > MAX_CHANNEL) NRF_channel = 0;
  602. start_scanning();
  603. }
  604. }
  605. view_port_update(APP->view_port);
  606. release_mutex(&state_mutex, plugin_state);
  607. }
  608. if(log_arr_idx && (log_to_file == 1 || log_to_file == 2)) {
  609. write_to_log_file(APP->storage);
  610. }
  611. nrf24_deinit();
  612. view_port_enabled_set(APP->view_port, false);
  613. gui_remove_view_port(APP->gui, APP->view_port);
  614. furi_record_close(RECORD_GUI);
  615. furi_record_close(RECORD_NOTIFICATION);
  616. furi_record_close(RECORD_STORAGE);
  617. view_port_free(APP->view_port);
  618. furi_message_queue_free(APP->event_queue);
  619. free(APP->plugin_state);
  620. if(APP->log_arr) free(APP->log_arr);
  621. free(APP);
  622. return 0;
  623. }