nrf24scan.c 29 KB

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