nrf24scan.c 71 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667
  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 "2.2"
  16. #define MAX_CHANNEL 125
  17. #define MAX_ADDR 6
  18. #define SCAN_APP_PATH_FOLDER STORAGE_APP_DATA_PATH_PREFIX
  19. #define SETTINGS_FILENAME "addresses.txt" // Settings file format (1 parameter per line):
  20. // SNIFF - if present then sniff mode
  21. // Rate: 0/1/2 - rate in Mbps (=0.25/1/2)
  22. // Ch: 0..125 - default channel
  23. // ESB: 0/1 (1 - Enhanced ShockBurst)
  24. // DPL: 0/1 (1 - Dynamic Payload Length)
  25. // CRC: 0/1/2 (CRC length)
  26. // Payload: 1..32 (bytes)
  27. // P0: address P0 in hex (5 byte, LSB last)
  28. // P1: address P1 in hex (5 byte, LSB last)
  29. // P2: address P2, LSB in hex (1 byte)
  30. // P3: address P3, LSB in hex (1 byte)
  31. // P4: address P4, LSB in hex (1 byte)
  32. // P5: address P5, LSB in hex (1 byte)
  33. // captured data:
  34. // first byte = { RAW packet flag (0x80/0x00) } + { channel number }
  35. // second byte = { Payload len 5 bits, 0 = 32 } + {{ RAW packet: ESB flag 0x04/0x00 + address size-2 if RAW packet } or { pipe #(0..5) }},
  36. // ... up to MAX_LOG_RECORDS-1
  37. #define SNIFF_FILENAME "sniff.txt" // settings for sniff mode
  38. #define LOG_FILENAME "log"
  39. #define LOG_FILEEXT ".txt"
  40. #define MAX_LOG_RECORDS 200
  41. #define MAX_FOUND_RECORDS 70
  42. #define LOG_REC_SIZE 34 // max packet size
  43. #define VIEW_LOG_MAX_X 22
  44. #define VIEW_LOG_WIDTH_B 10 // bytes
  45. const char SettingsFld_Rate[] = "Rate:";
  46. const char SettingsFld_Ch[] = "Ch:";
  47. const char SettingsFld_ESB[] = "ESB:";
  48. const char SettingsFld_DPL[] = "DPL:";
  49. const char SettingsFld_CRC[] = "CRC:";
  50. const char SettingsFld_Payload[] = "Payload:";
  51. const char SettingsFld_Sniff[] = "SNIFF";
  52. char SettingsFld_Addr = 'P';
  53. Nrf24Scan* APP;
  54. uint8_t what_doing = 0; // 0 - setup, 1 - view log, 2 - view addresses
  55. uint8_t what_to_do =
  56. 1; // 0 - view, 1 - view & sniff, 2 - view & read, 3 - view & read selected addr
  57. uint8_t save_settings = 0;
  58. char screen_buf[64];
  59. char addr_file_name[32];
  60. uint8_t NRF_rate = 2; // 0 - 250Kbps, 1 - 1Mbps, 2 - 2Mbps
  61. uint8_t NRF_channel = 0; // 0..125
  62. uint8_t NRF_ESB = 1; // 0 - ShockBurst, 1 - Enhanced ShockBurst
  63. uint8_t NRF_DPL = 0; // 1 - Dynamic Payload Length
  64. uint8_t NRF_CRC = 2; // 1 - No, 1 - CRC 1byte, 2 - CRC 2byte
  65. uint8_t NRF_Payload = 32; // Payload len in bytes or Minimum payload in sniff mode, 0..32
  66. uint8_t NRF_Payload_sniff_min = 0;
  67. uint8_t NRF_AA_OFF = 0; // Disable Auto Acknowledgement
  68. bool NRF_ERROR = 0;
  69. bool NRF_BOARD_POWER_5V = false;
  70. struct ADDRS {
  71. uint8_t addr_P0[5]; // MSB first
  72. uint8_t addr_P1[5]; // MSB first
  73. uint8_t addr_P2; // LSB only, MSB bytes equal addr_P1
  74. uint8_t addr_P3; // LSB only, MSB bytes equal addr_P1
  75. uint8_t addr_P4; // LSB only, MSB bytes equal addr_P1
  76. uint8_t addr_P5; // LSB only, MSB bytes equal addr_P1
  77. uint8_t addr_len; // 2..5
  78. uint8_t addr_count;
  79. };
  80. struct ADDRS addrs;
  81. struct ADDRS addrs_sniff;
  82. bool sniff_loaded = 0;
  83. int16_t found_total;
  84. int16_t view_found;
  85. int8_t log_to_file = 0; // 0 - no, 1 - yes(new), 2 - append, -1 - only clear
  86. uint16_t log_arr_idx;
  87. uint16_t view_log_arr_idx = 0;
  88. uint16_t view_log_arr_x = 0;
  89. bool save_to_new_log = true;
  90. uint16_t last_packet_send = -1;
  91. uint8_t last_packet_send_st = 0;
  92. int16_t find_channel_period = 0; // sec
  93. uint8_t menu_selected = 0;
  94. uint32_t start_time;
  95. uint8_t view_log_decode_PCF =
  96. 0; // view log: 1 - decode packet control field (9b) when ESB off. After pipe # (hex): <Payload len 6b><PID_2b+NO_ACK_1b>
  97. uint8_t view_log_decode_CRC = 0; // CRC bytes - 1/2, 0 - none
  98. #define menu_selected_max 5
  99. enum {
  100. Menu_open_file = 0,
  101. Menu_enter_channel,
  102. Menu_enter_rate,
  103. Menu_enter_scan_period,
  104. Menu_log,
  105. Menu_ok
  106. };
  107. //#define MIN(a, b) ((a<b)?a:b)
  108. static uint8_t GetHexVal(char hex) {
  109. return (uint8_t)hex - ((uint8_t)hex < 58 ? 48 : ((uint8_t)hex < 97 ? 55 : 87));
  110. }
  111. // Return num bytes in array
  112. static uint8_t ConvertHexToArray(char* hex, uint8_t* array, uint8_t maxlen) {
  113. uint8_t len = 0;
  114. while(maxlen) {
  115. uint8_t ch = *hex++;
  116. if(ch == 0) break;
  117. if(ch < '0') continue;
  118. *array++ = (GetHexVal(ch) << 4) + GetHexVal(*hex++);
  119. len++;
  120. maxlen--;
  121. }
  122. return len;
  123. }
  124. static void add_to_str_hex_bytes(char* out, char* arr, int bytes) {
  125. if(bytes <= 0) return;
  126. out += strlen(out);
  127. do {
  128. snprintf(out, 3, "%02X", *arr++);
  129. out += 2;
  130. } while(--bytes);
  131. }
  132. static void add_to_furi_str_hex_bytes(FuriString* str, char* arr, int bytes) {
  133. if(!bytes) return;
  134. do {
  135. furi_string_cat_printf(str, "%02X", *arr++);
  136. } while(--bytes);
  137. }
  138. static void add_to_str_hex_bytes_shift_9b(char* out, char* arr, int bytes) {
  139. if(bytes <= 0) return;
  140. out += strlen(out);
  141. arr++; // +8b
  142. do {
  143. snprintf(out, 4, "%02X", ((uint8_t)(*arr << 1)) | (*(arr + 1) >> 7));
  144. arr++;
  145. out += 2;
  146. } while(--bytes);
  147. }
  148. void clear_log() {
  149. log_arr_idx = 0;
  150. view_log_arr_idx = 0;
  151. last_packet_send = -1;
  152. found_total = 0;
  153. view_found = -1;
  154. }
  155. void write_to_log_file(Storage* storage, bool f_settings) {
  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(
  165. file_stream, furi_string_get_cstr(str), FSAM_READ_WRITE, FSOM_CREATE_ALWAYS);
  166. if(!fl) file_stream_close(file_stream);
  167. } else {
  168. furi_string_cat(str, LOG_FILENAME);
  169. furi_string_cat(str, LOG_FILEEXT);
  170. if(save_to_new_log) {
  171. int cnt = 1;
  172. do {
  173. fl = file_stream_open(
  174. file_stream, furi_string_get_cstr(str), FSAM_READ_WRITE, FSOM_CREATE_NEW);
  175. if(fl) break;
  176. file_stream_close(file_stream);
  177. furi_string_set(str, SCAN_APP_PATH_FOLDER);
  178. furi_string_cat(str, "/");
  179. furi_string_cat(str, LOG_FILENAME);
  180. furi_string_cat_printf(str, "-%02d", cnt);
  181. furi_string_cat(str, LOG_FILEEXT);
  182. } while(++cnt < 100);
  183. if(!fl) {
  184. FURI_LOG_E(TAG, "Failed to create new log file");
  185. notification_message(APP->notification, &sequence_blink_red_100);
  186. }
  187. } else {
  188. fl = file_stream_open(
  189. file_stream, furi_string_get_cstr(str), FSAM_READ_WRITE, FSOM_OPEN_APPEND);
  190. if(fl) {
  191. if(stream_size(file_stream) == 0) save_to_new_log = true;
  192. } else
  193. file_stream_close(file_stream);
  194. }
  195. }
  196. if(fl) {
  197. FURI_LOG_D(TAG, "Save to %s", furi_string_get_cstr(str));
  198. if(save_to_new_log || f_settings) {
  199. if(what_to_do == 1)
  200. furi_string_printf(str, "%s\n", SettingsFld_Sniff);
  201. else
  202. furi_string_reset(str);
  203. furi_string_cat_printf(
  204. str,
  205. "%s %d\n%s %d\n%s %d\n",
  206. SettingsFld_Rate,
  207. NRF_rate,
  208. SettingsFld_Ch,
  209. NRF_channel,
  210. SettingsFld_ESB,
  211. NRF_ESB);
  212. furi_string_cat_printf(
  213. str,
  214. "%s %d\n%s %d\n%s %d\n",
  215. SettingsFld_DPL,
  216. NRF_DPL,
  217. SettingsFld_CRC,
  218. NRF_CRC,
  219. SettingsFld_Payload,
  220. what_to_do == 1 ? NRF_Payload_sniff_min : NRF_Payload);
  221. if(addrs.addr_count > 0) {
  222. furi_string_cat_printf(str, "P0: ");
  223. add_to_furi_str_hex_bytes(str, (char*)addrs.addr_P0, addrs.addr_len);
  224. furi_string_cat(str, "\n");
  225. }
  226. if(addrs.addr_count > 1) {
  227. furi_string_cat_printf(str, "P1: ");
  228. add_to_furi_str_hex_bytes(str, (char*)addrs.addr_P1, addrs.addr_len);
  229. furi_string_cat(str, "\n");
  230. }
  231. if(addrs.addr_count > 2) {
  232. furi_string_cat_printf(str, "P2: ");
  233. furi_string_cat_printf(str, "%02X\n", addrs.addr_P2);
  234. }
  235. if(addrs.addr_count > 3) {
  236. furi_string_cat_printf(str, "P3: ");
  237. furi_string_cat_printf(str, "%02X\n", addrs.addr_P3);
  238. }
  239. if(addrs.addr_count > 4) {
  240. furi_string_cat_printf(str, "P4: ");
  241. furi_string_cat_printf(str, "%02X\n", addrs.addr_P4);
  242. }
  243. if(addrs.addr_count > 5) {
  244. furi_string_cat_printf(str, "P5: ");
  245. furi_string_cat_printf(str, "%02X\n", addrs.addr_P5);
  246. }
  247. if(!(fl = stream_write_string(file_stream, str) == furi_string_size(str))) {
  248. FURI_LOG_E(TAG, "Failed to write header to file!");
  249. notification_message(APP->notification, &sequence_blink_red_100);
  250. }
  251. }
  252. if(fl) {
  253. if(f_settings) {
  254. save_settings = 0;
  255. if(strcmp(addr_file_name, "NONE") == 0) strcpy(addr_file_name, SETTINGS_FILENAME);
  256. } else {
  257. int i = 0;
  258. for(; i < log_arr_idx; i++) {
  259. furi_string_reset(str);
  260. uint8_t* ptr = APP->log_arr + i * LOG_REC_SIZE;
  261. int len;
  262. if(ptr[0] & 0x80) { // RAW
  263. len = (ptr[1] & 0b11) + 2 + ((ptr[1] & 0b100) ? 2 : 0) + (ptr[1] >> 3) +
  264. 2; // addr + PCF? + payload + crcmax
  265. } else {
  266. len = (ptr[1] >> 3);
  267. if(len == 0) len = 32;
  268. }
  269. //if(len < NRF_Payload) len = NRF_Payload;
  270. add_to_furi_str_hex_bytes(str, (char*)ptr, len + 2);
  271. furi_string_cat(str, "\n");
  272. if(stream_write_string(file_stream, str) != furi_string_size(str)) {
  273. FURI_LOG_E(TAG, "Failed to write to file!");
  274. break;
  275. }
  276. }
  277. if(i == log_arr_idx) {
  278. notification_message(APP->notification, &sequence_blink_yellow_100);
  279. FURI_LOG_D(TAG, "File saved");
  280. }
  281. save_to_new_log = false;
  282. }
  283. }
  284. file_stream_close(file_stream);
  285. } else {
  286. FURI_LOG_E(TAG, "Failed to open file %s", furi_string_get_cstr(str));
  287. notification_message(APP->notification, &sequence_blink_red_100);
  288. }
  289. stream_free(file_stream);
  290. furi_string_free(str);
  291. }
  292. static bool select_settings_file(Stream* stream) {
  293. DialogsApp* dialogs = furi_record_open("dialogs");
  294. bool result = false;
  295. FuriString* path;
  296. path = furi_string_alloc();
  297. furi_string_set(path, SCAN_APP_PATH_FOLDER);
  298. DialogsFileBrowserOptions browser_options;
  299. dialog_file_browser_set_basic_options(&browser_options, ".txt", NULL);
  300. browser_options.hide_ext = false;
  301. bool ret = dialog_file_browser_show(dialogs, path, path, &browser_options);
  302. furi_record_close("dialogs");
  303. if(ret) {
  304. if(!file_stream_open(stream, furi_string_get_cstr(path), FSAM_READ, FSOM_OPEN_EXISTING)) {
  305. FURI_LOG_D(TAG, "Cannot open file \"%s\"", furi_string_get_cstr(path));
  306. file_stream_close(stream);
  307. } else {
  308. FURI_LOG_D(TAG, "Open file \"%s\"", furi_string_get_cstr(path));
  309. strncpy(
  310. addr_file_name,
  311. furi_string_get_cstr(path) + sizeof(SCAN_APP_PATH_FOLDER),
  312. sizeof(addr_file_name));
  313. result = true;
  314. }
  315. }
  316. furi_string_free(path);
  317. return result;
  318. }
  319. // 0 - success, otherwise an error
  320. static uint8_t load_settings_file(Stream* file_stream) {
  321. size_t file_size = 0;
  322. char* file_buf;
  323. uint8_t err = 5;
  324. file_size = stream_size(file_stream);
  325. if(file_size == (size_t)0) {
  326. FURI_LOG_D(TAG, "load failed. file_size: %d", file_size);
  327. return 1;
  328. }
  329. file_size = MIN(file_size, (size_t)LOG_REC_SIZE * MAX_LOG_RECORDS * 2 + 100);
  330. file_buf = malloc(file_size + 1);
  331. if(file_buf == NULL) {
  332. FURI_LOG_D(TAG, "Memory low, need: %d", file_size);
  333. return 2;
  334. }
  335. memset(file_buf, 0, file_size + 1);
  336. if(stream_read(file_stream, (uint8_t*)file_buf, file_size) == file_size) {
  337. FURI_LOG_D(TAG, "Loading settings file");
  338. char* line_ptr = file_buf;
  339. int16_t line_num = 0;
  340. what_to_do = 2;
  341. sniff_loaded = 0;
  342. bool log_loaded = false;
  343. while(line_ptr && (size_t)(line_ptr - file_buf) < file_size) {
  344. char* end_ptr = strstr((char*)line_ptr, "\n");
  345. if(end_ptr == NULL)
  346. end_ptr = file_buf + file_size;
  347. else
  348. *end_ptr = '\0';
  349. int line_len = end_ptr - line_ptr;
  350. if(*line_ptr == '\r' || line_len == 0) {
  351. line_ptr = end_ptr + 1;
  352. continue;
  353. }
  354. if(*(end_ptr - 1) < '0') {
  355. *(end_ptr - 1) = '\0';
  356. line_len--;
  357. }
  358. //FURI_LOG_D(TAG, " L#%d: [%d]%s", line_num, line_len, line_ptr);
  359. if(strncmp(line_ptr, SettingsFld_Rate, sizeof(SettingsFld_Rate) - 1) == 0) {
  360. NRF_rate = atoi(line_ptr + sizeof(SettingsFld_Rate));
  361. } else if(strncmp(line_ptr, SettingsFld_Ch, sizeof(SettingsFld_Ch) - 1) == 0) {
  362. NRF_channel = atoi(line_ptr + sizeof(SettingsFld_Ch));
  363. } else if(strncmp(line_ptr, SettingsFld_ESB, sizeof(SettingsFld_ESB) - 1) == 0) {
  364. NRF_ESB = atoi(line_ptr + sizeof(SettingsFld_ESB));
  365. } else if(strncmp(line_ptr, SettingsFld_DPL, sizeof(SettingsFld_DPL) - 1) == 0) {
  366. NRF_DPL = atoi(line_ptr + sizeof(SettingsFld_DPL));
  367. } else if(strncmp(line_ptr, SettingsFld_CRC, sizeof(SettingsFld_CRC) - 1) == 0) {
  368. NRF_CRC = atoi(line_ptr + sizeof(SettingsFld_CRC));
  369. if(what_to_do == 1) view_log_decode_CRC = NRF_CRC;
  370. } else if(strncmp(line_ptr, SettingsFld_Payload, sizeof(SettingsFld_Payload) - 1) == 0) {
  371. uint8_t pld = atoi(line_ptr + sizeof(SettingsFld_Payload));
  372. if(pld > 32) pld = 32;
  373. if(sniff_loaded) {
  374. NRF_Payload_sniff_min = pld;
  375. } else {
  376. if(pld == 0) pld = 32;
  377. NRF_Payload = pld;
  378. }
  379. } else if(strncmp(line_ptr, SettingsFld_Sniff, sizeof(SettingsFld_Sniff) - 1) == 0) {
  380. what_to_do = 1;
  381. sniff_loaded = 1;
  382. } else if(*line_ptr == SettingsFld_Addr) {
  383. char a = *(++line_ptr);
  384. struct ADDRS* adr = sniff_loaded ? &addrs_sniff : &addrs;
  385. line_ptr += 3;
  386. switch(a) {
  387. case '0':
  388. memset(adr, 0, sizeof(addrs));
  389. adr->addr_len =
  390. ConvertHexToArray(line_ptr, adr->addr_P0, sniff_loaded ? 3 : 5);
  391. if(adr->addr_len >= 2) err = 0;
  392. break;
  393. case '1':
  394. ConvertHexToArray(line_ptr, adr->addr_P1, what_to_do == 1 ? 3 : 5);
  395. break;
  396. case '2':
  397. ConvertHexToArray(line_ptr, &adr->addr_P2, 1);
  398. break;
  399. case '3':
  400. ConvertHexToArray(line_ptr, &adr->addr_P3, 1);
  401. break;
  402. case '4':
  403. ConvertHexToArray(line_ptr, &adr->addr_P4, 1);
  404. break;
  405. case '5':
  406. ConvertHexToArray(line_ptr, &adr->addr_P5, 1);
  407. break;
  408. default:
  409. a = 0;
  410. break;
  411. }
  412. if(err == 0 && a) adr->addr_count = a - '0' + 1;
  413. } else if(line_len >= 3 * 2) { // data
  414. if(!log_loaded) {
  415. log_loaded = true;
  416. clear_log();
  417. what_to_do = 0;
  418. }
  419. if(log_arr_idx < MAX_LOG_RECORDS - 1) {
  420. if(ConvertHexToArray(
  421. line_ptr, APP->log_arr + log_arr_idx * LOG_REC_SIZE, LOG_REC_SIZE) > 0)
  422. err = 0;
  423. log_arr_idx++;
  424. }
  425. }
  426. line_ptr = end_ptr + 1;
  427. line_num++;
  428. }
  429. } else {
  430. FURI_LOG_D(TAG, "load failed. file size: %d", file_size);
  431. err = 4;
  432. }
  433. free(file_buf);
  434. return err;
  435. }
  436. static void input_callback(InputEvent* input_event, void* ctx) {
  437. furi_assert(ctx);
  438. FuriMessageQueue* event_queue = ctx;
  439. PluginEvent event = {.type = EventTypeKey, .input = *input_event};
  440. furi_message_queue_put(event_queue, &event, FuriWaitForever);
  441. }
  442. void check_add_addr(uint8_t* pkt) {
  443. if(addrs.addr_count > 0 && memcmp(addrs.addr_P0, pkt, addrs.addr_len) == 0) return;
  444. if(addrs.addr_count > 1 && memcmp(addrs.addr_P1, pkt, addrs.addr_len - 1) == 0) {
  445. if(addrs.addr_P1[addrs.addr_len - 1] == pkt[addrs.addr_len - 1]) return;
  446. if(addrs.addr_count > 2 && addrs.addr_P2 == pkt[addrs.addr_len - 1]) return;
  447. if(addrs.addr_count > 3 && addrs.addr_P3 == pkt[addrs.addr_len - 1]) return;
  448. if(addrs.addr_count > 4 && addrs.addr_P4 == pkt[addrs.addr_len - 1]) return;
  449. if(addrs.addr_count > 5 && addrs.addr_P5 == pkt[addrs.addr_len - 1]) return;
  450. }
  451. if(addrs.addr_count == 1)
  452. memcpy(addrs.addr_P1, pkt, addrs.addr_len);
  453. else if(addrs.addr_count == 2)
  454. addrs.addr_P2 = pkt[addrs.addr_len - 1];
  455. else if(addrs.addr_count == 3)
  456. addrs.addr_P3 = pkt[addrs.addr_len - 1];
  457. else if(addrs.addr_count == 4)
  458. addrs.addr_P4 = pkt[addrs.addr_len - 1];
  459. else if(addrs.addr_count == 5)
  460. addrs.addr_P5 = pkt[addrs.addr_len - 1];
  461. addrs.addr_count++;
  462. }
  463. static void prepare_nrf24(bool fsend_packet) {
  464. nrf24_write_reg(nrf24_HANDLE, REG_STATUS, 0x70); // clear interrupts
  465. nrf24_write_reg(
  466. nrf24_HANDLE,
  467. REG_RF_SETUP,
  468. (NRF_rate == 0 ? 0b00100000 :
  469. NRF_rate == 1 ? 0 :
  470. 0b00001000) |
  471. 0b111); // +TX high power
  472. uint8_t erx_addr = (1 << 0); // Enable RX_P0
  473. struct ADDRS* adr = what_to_do == 1 ? &addrs_sniff : &addrs;
  474. if(!fsend_packet) {
  475. uint8_t payload = NRF_Payload;
  476. uint8_t* rec = APP->log_arr + view_log_arr_idx * LOG_REC_SIZE;
  477. uint8_t addr_size = (*(rec + 1) & 0b11) + 2;
  478. bool setup_from_log = false;
  479. if(what_to_do >= 2) {
  480. if(log_arr_idx && (*rec & 0x80)) {
  481. setup_from_log = true;
  482. memcpy(addrs.addr_P0, rec + 2, addr_size);
  483. addrs.addr_count = 1;
  484. addrs.addr_len = addr_size;
  485. payload = *(rec + 1) >> 3;
  486. if(what_to_do == 2) {
  487. uint8_t* p = APP->log_arr + 2;
  488. int16_t i = 0;
  489. for(i = 0; i < log_arr_idx; i++, p += LOG_REC_SIZE) {
  490. if((*(p - 2) & 0x80) && (*(p - 1) & 0b11) + 2 == addr_size &&
  491. rec + 2 != p) {
  492. if(memcmp(p, addrs.addr_P0, addr_size - 1) == 0) {
  493. check_add_addr(p);
  494. if(addrs.addr_count >= 6) break;
  495. }
  496. }
  497. }
  498. }
  499. }
  500. }
  501. if(what_to_do == 1) { // SNIFF
  502. payload = 32;
  503. nrf24_write_reg(nrf24_HANDLE, REG_CONFIG, 0x70); // Mask all interrupts
  504. nrf24_write_reg(nrf24_HANDLE, REG_SETUP_RETR, 0); // Automatic Retransmission
  505. nrf24_write_reg(nrf24_HANDLE, REG_EN_AA, 0); // Auto acknowledgement
  506. nrf24_write_reg(
  507. nrf24_HANDLE,
  508. REG_FEATURE,
  509. 0); // Enables the W_TX_PAYLOAD_NOACK command, Disable Payload with ACK, set Dynamic Payload
  510. // EN_DYN_ACK(0x01) for W_TX_PAYLOAD_NOACK cmd broke AA on some fake nRF24l01+ modules
  511. } else if(setup_from_log) { // Scan
  512. nrf24_write_reg(
  513. nrf24_HANDLE,
  514. REG_CONFIG,
  515. 0x70 | ((NRF_CRC == 1 ? 0b1000 :
  516. NRF_CRC == 2 ? 0b1100 :
  517. 0))); // Mask all interrupts
  518. nrf24_write_reg(nrf24_HANDLE, REG_RF_CH, *rec & 0x7F);
  519. nrf24_write_reg(
  520. nrf24_HANDLE,
  521. REG_FEATURE,
  522. *(rec + 2 + addr_size) >> 2 != 0x33 ?
  523. 4 :
  524. 0); // Enables the W_TX_PAYLOAD_NOACK command, Disable Payload with ACK, set Dynamic Payload
  525. if(*(rec + 1) & 0b100) { // ESB
  526. nrf24_write_reg(nrf24_HANDLE, REG_SETUP_RETR, 0x01); // Automatic Retransmission
  527. nrf24_write_reg(nrf24_HANDLE, REG_EN_AA, 0x3F); // Auto acknowledgement
  528. } else {
  529. nrf24_write_reg(nrf24_HANDLE, REG_SETUP_RETR, 0); // Automatic Retransmission
  530. nrf24_write_reg(nrf24_HANDLE, REG_EN_AA, 0); // Auto acknowledgement
  531. }
  532. } else {
  533. nrf24_write_reg(
  534. nrf24_HANDLE,
  535. REG_CONFIG,
  536. 0x70 | ((NRF_CRC == 1 ? 0b1000 :
  537. NRF_CRC == 2 ? 0b1100 :
  538. 0))); // Mask all interrupts
  539. nrf24_write_reg(
  540. nrf24_HANDLE, REG_SETUP_RETR, NRF_ESB ? 0x01 : 0); // Automatic Retransmission
  541. nrf24_write_reg(
  542. nrf24_HANDLE,
  543. REG_EN_AA,
  544. NRF_AA_OFF || !NRF_ESB ? 0 : 0x3F); // Auto acknowledgement
  545. nrf24_write_reg(
  546. nrf24_HANDLE,
  547. REG_FEATURE,
  548. NRF_DPL ?
  549. 4 :
  550. 0); // Enables the W_TX_PAYLOAD_NOACK command, Disable Payload with ACK, set Dynamic Payload
  551. nrf24_write_reg(
  552. nrf24_HANDLE, REG_DYNPD, NRF_DPL ? 0x3F : 0); // Enable dynamic payload reg
  553. nrf24_write_reg(nrf24_HANDLE, REG_RF_CH, NRF_channel);
  554. }
  555. if(adr->addr_count == 0) return;
  556. nrf24_write_reg(nrf24_HANDLE, RX_PW_P0, payload);
  557. nrf24_set_maclen(nrf24_HANDLE, adr->addr_len);
  558. nrf24_set_mac(REG_RX_ADDR_P0, adr->addr_P0, adr->addr_len);
  559. uint8_t tmp[5] = {0};
  560. nrf24_read_reg(nrf24_HANDLE, REG_RX_ADDR_P0, tmp, adr->addr_len);
  561. for(uint8_t i = 0; i < adr->addr_len / 2; i++) {
  562. uint8_t tb = tmp[i];
  563. tmp[i] = tmp[adr->addr_len - i - 1];
  564. tmp[adr->addr_len - i - 1] = tb;
  565. }
  566. NRF_ERROR = memcmp(adr->addr_P0, tmp, adr->addr_len) != 0;
  567. FURI_LOG_D(TAG, "Payload: %d", payload);
  568. nrf24_write_reg(nrf24_HANDLE, RX_PW_P0, payload);
  569. if(adr->addr_count > 1) {
  570. nrf24_set_mac(REG_RX_ADDR_P1, adr->addr_P1, adr->addr_len);
  571. nrf24_write_reg(nrf24_HANDLE, RX_PW_P1, payload);
  572. erx_addr |= (1 << 1); // Enable RX_P1
  573. } else
  574. nrf24_write_reg(nrf24_HANDLE, RX_PW_P1, 0);
  575. if(adr->addr_count > 2) {
  576. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P2, &adr->addr_P2, 1);
  577. nrf24_write_reg(nrf24_HANDLE, RX_PW_P2, payload);
  578. erx_addr |= (1 << 2); // Enable RX_P2
  579. } else
  580. nrf24_write_reg(nrf24_HANDLE, RX_PW_P2, 0);
  581. if(adr->addr_count > 3) {
  582. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P3, &adr->addr_P3, 1);
  583. nrf24_write_reg(nrf24_HANDLE, RX_PW_P3, payload);
  584. erx_addr |= (1 << 3); // Enable RX_P3
  585. } else
  586. nrf24_write_reg(nrf24_HANDLE, RX_PW_P3, 0);
  587. if(adr->addr_count > 4) {
  588. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P4, &adr->addr_P4, 1);
  589. nrf24_write_reg(nrf24_HANDLE, RX_PW_P4, payload);
  590. erx_addr |= (1 << 4); // Enable RX_P4
  591. } else
  592. nrf24_write_reg(nrf24_HANDLE, RX_PW_P4, 0);
  593. if(adr->addr_count > 5) {
  594. nrf24_write_buf_reg(nrf24_HANDLE, REG_RX_ADDR_P5, &adr->addr_P5, 1);
  595. nrf24_write_reg(nrf24_HANDLE, RX_PW_P5, payload);
  596. erx_addr |= (1 << 5); // Enable RX_P5
  597. } else
  598. nrf24_write_reg(nrf24_HANDLE, RX_PW_P5, 0);
  599. nrf24_write_reg(nrf24_HANDLE, REG_EN_RXADDR, erx_addr);
  600. }
  601. nrf24_flush_rx(nrf24_HANDLE);
  602. nrf24_flush_tx(nrf24_HANDLE);
  603. nrf24_set_idle(nrf24_HANDLE);
  604. }
  605. void correct_NRF_Payload_sniff_min() {
  606. uint8_t pld = 32 - 3 - (NRF_ESB ? 2 : 0) - NRF_CRC + (addrs_sniff.addr_len - 2);
  607. if(NRF_Payload_sniff_min > pld) NRF_Payload_sniff_min = pld;
  608. }
  609. static void start_scanning() {
  610. FURI_LOG_D(TAG, "Start proc-%d: Ch=%d Rate=%d", what_to_do, NRF_channel, NRF_rate);
  611. if(what_to_do == 1) { // SNIFF
  612. correct_NRF_Payload_sniff_min();
  613. view_log_decode_CRC = NRF_CRC;
  614. } else if(sniff_loaded) { // Switch from sniff to scan/view
  615. // to do...
  616. }
  617. prepare_nrf24(false);
  618. if(NRF_ERROR) {
  619. FURI_LOG_E(TAG, "NRF R/W ERROR!");
  620. return;
  621. }
  622. nrf24_set_rx_mode(nrf24_HANDLE);
  623. start_time = furi_get_tick();
  624. }
  625. // start bitnum = 7
  626. uint32_t calc_crc(uint32_t crc, uint8_t* ptr, uint8_t bitnum, uint16_t bits) {
  627. //uint8_t bitnum = 7;
  628. uint32_t crc_high, polynom;
  629. if(view_log_decode_CRC == 2) {
  630. crc_high = (1 << 16);
  631. polynom = 0x1021; // X^16+X^12+X^5+1 => 0x11021 & 0xFFFF = 0x1021
  632. } else {
  633. crc_high = (1 << 8);
  634. polynom = 0x07; // x^8+x^2+x^1+1 => 0x107 & 0xFF = 0x07
  635. }
  636. while(bits--) {
  637. crc <<= 1;
  638. if(((crc & crc_high) != 0) ^ ((*ptr >> bitnum) & 1)) crc ^= polynom;
  639. if(bitnum == 0) {
  640. ptr++;
  641. bitnum = 7;
  642. } else
  643. bitnum--;
  644. }
  645. return crc & (view_log_decode_CRC == 2 ? 0xFFFF : 0xFF);
  646. }
  647. // shifted 1 bit right
  648. uint32_t get_shifted_crc(uint8_t* pcrc) {
  649. uint32_t crc = ((*pcrc << 1) & 0xFF) | (*(pcrc + 1) >> 7);
  650. if(view_log_decode_CRC == 2) {
  651. crc = (crc << 8) | ((*(pcrc + 1) << 1) & 0xFF) | (*(pcrc + 2) >> 7);
  652. }
  653. return crc;
  654. }
  655. bool check_packet(uint8_t* pkt, uint16_t size) {
  656. if(size < 3 || size > 32) return false;
  657. uint8_t b = *pkt;
  658. if(b == 0x55 || b == 0xAA || b == 0x00 || b == 0xFF)
  659. return false; // skip pkt when address begin with
  660. uint32_t prevcrc = 0xFFFFFFFF;
  661. bool found = false;
  662. uint8_t addr_size = 3;
  663. for(; addr_size <= 5; addr_size++) {
  664. if(NRF_ESB) {
  665. uint8_t _payload = *(pkt + addr_size) >> 2;
  666. if((_payload > size - addr_size - 2 - view_log_decode_CRC && _payload != 0x33))
  667. continue;
  668. uint8_t* p = pkt + addr_size;
  669. if(prevcrc == 0xFFFFFFFF) {
  670. prevcrc = calc_crc(
  671. view_log_decode_CRC == 2 ? 0xFFFF : 0xFF,
  672. pkt,
  673. 7,
  674. 3 * 8); // crc for smallest addr
  675. }
  676. uint32_t crc;
  677. if(addr_size > 3)
  678. crc = calc_crc(prevcrc, p - (addr_size - 3), 7, 8 * (addr_size - 3));
  679. else
  680. crc = prevcrc;
  681. if(_payload != 0x33) { // DPL
  682. crc = calc_crc(crc, p, 7, 9 + _payload * 8);
  683. if(crc == get_shifted_crc(p + _payload + 1)) {
  684. *(pkt - 1) = ((_payload & 0x1F) << 3) + 0b100 + (addr_size - 2);
  685. FURI_LOG_D(
  686. TAG, "VALID CRC %X: dpl: %d, addr: %d", (uint16_t)crc, _payload, addr_size);
  687. found = true;
  688. break;
  689. }
  690. } else {
  691. crc = calc_crc(crc, p++, 7, 9); // PCF
  692. if(crc == get_shifted_crc(p)) {
  693. _payload = 0;
  694. found = true;
  695. } else {
  696. for(uint8_t i = 1; i < size - addr_size - view_log_decode_CRC; i++) {
  697. crc = calc_crc(crc, p++, 6, 8);
  698. if(crc == get_shifted_crc(p)) {
  699. _payload = i;
  700. found = true;
  701. break;
  702. }
  703. }
  704. }
  705. if(found) {
  706. *(pkt - 1) = ((_payload & 0x1F) << 3) + 0b100 + (addr_size - 2);
  707. FURI_LOG_D(
  708. TAG, "VALID CRC %X: pl: %d, addr: %d", (uint16_t)crc, _payload, addr_size);
  709. break;
  710. }
  711. }
  712. } else {
  713. uint8_t* p;
  714. if(addr_size == 3) {
  715. prevcrc = calc_crc(
  716. view_log_decode_CRC == 2 ? 0xFFFF : 0xFF,
  717. pkt,
  718. 7,
  719. 3 * 8); // crc for smallest addr
  720. p = pkt + addr_size;
  721. } else {
  722. p = pkt + addr_size - 1;
  723. prevcrc = calc_crc(prevcrc, p++, 7, 8);
  724. }
  725. uint32_t crc = prevcrc;
  726. if((view_log_decode_CRC == 1 && crc == *p) ||
  727. (view_log_decode_CRC == 2 && crc == (uint32_t)((*p << 8) | *(p + 1)))) {
  728. *(pkt - 1) = ((0 & 0x1F) << 3) + 0b000 + (addr_size - 2);
  729. FURI_LOG_D(TAG, "VALID CRC %X: pl: %d, addr: %d", (uint16_t)crc, 0, addr_size);
  730. found = true;
  731. break;
  732. }
  733. for(uint8_t i = 1; i <= size - addr_size - view_log_decode_CRC; i++) {
  734. crc = calc_crc(crc, p++, 7, 8);
  735. if((view_log_decode_CRC == 1 && crc == *p) ||
  736. (view_log_decode_CRC == 2 && crc == (uint32_t)((*p << 8) | *(p + 1)))) {
  737. *(pkt - 1) = ((i & 0x1F) << 3) + 0b000 + (addr_size - 2);
  738. FURI_LOG_D(TAG, "VALID CRC %X: pl: %d, addr: %d", (uint16_t)crc, i, addr_size);
  739. found = true;
  740. break;
  741. }
  742. }
  743. if(found) break;
  744. }
  745. }
  746. if(found) {
  747. if(furi_log_get_level() == FuriLogLevelDebug) {
  748. char dbuf[65];
  749. dbuf[0] = 0;
  750. add_to_str_hex_bytes(dbuf, (char*)pkt, size);
  751. FURI_LOG_D(TAG, "PKT%02X: %s (%d)", *(pkt - 1), dbuf, size);
  752. }
  753. int16_t i = 0;
  754. for(; i < found_total; i++) {
  755. if(APP->found[i].addr_size != addr_size) continue;
  756. if(memcmp(APP->found[i].addr, pkt, addr_size) == 0) break;
  757. }
  758. if(i != found_total) { // found
  759. APP->found[i].total++;
  760. } else {
  761. uint8_t* p = APP->log_arr + 2;
  762. for(i = 0; i < log_arr_idx; i++, p += LOG_REC_SIZE) {
  763. if((*(p - 2) & 0x80) && (*(p - 1) & 0b11) + 2 == addr_size && pkt != p) {
  764. if(memcmp(p, pkt, addr_size) == 0) break;
  765. }
  766. }
  767. if(i != log_arr_idx && found_total < MAX_FOUND_RECORDS) { // found -> 2
  768. memset(&APP->found[found_total], 0, sizeof(struct FOUND));
  769. memcpy(APP->found[found_total].addr, pkt, addr_size);
  770. APP->found[found_total].addr_size = addr_size;
  771. APP->found[found_total].total = 2;
  772. if(found_total == 0) view_found = 0;
  773. found_total++;
  774. }
  775. }
  776. }
  777. return found;
  778. }
  779. bool nrf24_read_newpacket() {
  780. if(APP->log_arr == NULL) return false;
  781. bool found = false;
  782. uint8_t packetsize;
  783. uint8_t* ptr = APP->log_arr + log_arr_idx * LOG_REC_SIZE;
  784. uint8_t st;
  785. /* test pkts
  786. static int iii = 0;
  787. char ppp[][65] = { "54A545109411544BAAE50110A3282512A9A1152A565B22AAA48AB751A5",
  788. "C8C8C0CE7A81018007202FFFFC",
  789. "EAEC8C8C2CE3C0101006FB737A",
  790. "BEBFFFEC8C8C1CC00542AF7CFF7DBEAFE3397FEAFEF1DDFA4AEF7FDBB7CDEABC",
  791. "FEAAAABEAAFEAAC8C8C28E1C810080490ABAF7FEEB76B7FDFEF7DFFB47FB97FE",
  792. "A8AAC8C8C1CE20163DF7DFFD00",
  793. "AFFEEFEC8C8C2CE4001010062F037F9BFFDF1DAD5EDBEF55DD9AB535FCB67F55",
  794. "AC8C8C1CE5F8102000D503D7ABF",
  795. "EE03080B4712555555550E80",
  796. "C8C8C41385818280127100",
  797. "AAC8C8C3CE05818280119000"
  798. "AC8C8C413858182801271000",
  799. "AAC8C8C40B0305028542"
  800. };
  801. if(iii < 13) {
  802. ConvertHexToArray(ppp[iii], ptr + 2, 32);
  803. st = RX_DR;
  804. packetsize = 32;
  805. iii++;
  806. } else
  807. //*/
  808. st = nrf24_rxpacket(
  809. nrf24_HANDLE,
  810. ptr + 2 + (what_to_do == 1 ? addrs_sniff.addr_len - 2 : 0),
  811. &packetsize,
  812. what_to_do == 1 ? 32 : !NRF_DPL);
  813. if(st & RX_DR) {
  814. st = (st >> 1) & 7; // pipe #
  815. if(what_to_do == 1) { // SNIFF
  816. *ptr++ = NRF_channel | 0x80;
  817. *ptr++ = st; // pipe #
  818. if(addrs_sniff.addr_len > 2) {
  819. *ptr = st == 0 ? addrs_sniff.addr_P0[2] :
  820. st == 1 ? addrs_sniff.addr_P1[2] :
  821. st == 2 ? addrs_sniff.addr_P2 :
  822. st == 3 ? addrs_sniff.addr_P3 :
  823. st == 4 ? addrs_sniff.addr_P4 :
  824. addrs_sniff.addr_P5;
  825. }
  826. if(!check_packet(ptr, packetsize)) {
  827. if(addrs_sniff.addr_len > 2) return false; // skip if mac MSB added to preamble
  828. uint8_t shifted = 0;
  829. uint8_t shift_max = (32 - 3 - NRF_Payload_sniff_min - NRF_CRC) * 8 - 1;
  830. while(shifted++ <
  831. shift_max) { // Shift packet left by one bit if minimum payload fits
  832. uint8_t i = 0;
  833. for(; i < packetsize - 1; i++)
  834. ptr[i] = (ptr[i] << 1) | (ptr[i + 1] >> 7);
  835. ptr[i] <<= 1;
  836. if(check_packet(ptr, packetsize - (shifted >> 3) - 1)) goto x_valid;
  837. }
  838. return false;
  839. }
  840. } else {
  841. *ptr++ = NRF_channel;
  842. *ptr++ = ((packetsize & 0x1F) << 3) | st; // payload size + pipe #
  843. }
  844. x_valid:
  845. if(packetsize < 32) memset(ptr + packetsize, 0, 32 - packetsize);
  846. if(log_arr_idx < MAX_LOG_RECORDS - 1) {
  847. log_arr_idx++;
  848. } else {
  849. if(log_to_file == 1 || log_to_file == 2) {
  850. write_to_log_file(APP->storage, false);
  851. clear_log();
  852. } else {
  853. memmove(APP->log_arr, APP->log_arr + LOG_REC_SIZE, log_arr_idx * LOG_REC_SIZE);
  854. }
  855. }
  856. FURI_LOG_D(TAG, "Found packet #%d pipe %d", log_arr_idx, st);
  857. notification_message(APP->notification, &sequence_blink_white_100);
  858. found = true;
  859. }
  860. return found;
  861. }
  862. bool nrf24_send_packet() {
  863. if(log_arr_idx == 0) return false;
  864. prepare_nrf24(!what_to_do);
  865. uint8_t* ptr = APP->log_arr + view_log_arr_idx * LOG_REC_SIZE;
  866. nrf24_write_reg(nrf24_HANDLE, REG_RF_CH, *ptr & 0x7F);
  867. if(*ptr & 0x80) { // RAW packet
  868. //uint8_t pktinfo = *(ptr + 1);
  869. //nrf24_set_maclen(nrf24_HANDLE, (pktinfo & 0b11) + 2);
  870. //if(pktinfo & 0b100) { // ESB
  871. nrf24_write_reg(nrf24_HANDLE, REG_SETUP_RETR, 0); // No Automatic Retransmission
  872. nrf24_write_reg(nrf24_HANDLE, REG_EN_AA, 0); // No Auto acknowledgement
  873. //}
  874. //uint8_t alen = (*(ptr + 2) & 0b11) + 2;
  875. uint8_t adr[2];
  876. adr[0] = ptr[2];
  877. adr[1] = ptr[3];
  878. nrf24_set_maclen(nrf24_HANDLE, 2);
  879. nrf24_set_mac(REG_RX_ADDR_P0, adr, 2);
  880. nrf24_set_mac(REG_TX_ADDR, adr, 2);
  881. last_packet_send_st = nrf24_txpacket(nrf24_HANDLE, ptr + 2 + 2, 32 - 2, true);
  882. } else {
  883. nrf24_write_reg(
  884. nrf24_HANDLE, REG_SETUP_RETR, NRF_ESB ? 0x11 : 0); // Automatic Retransmission
  885. nrf24_write_reg(
  886. nrf24_HANDLE, REG_EN_AA, NRF_AA_OFF || !NRF_ESB ? 0 : 0x3F); // Auto acknowledgement
  887. uint8_t* adr;
  888. uint8_t a = *(ptr + 1) & 0b111;
  889. if(a < 2) {
  890. if(a == 0)
  891. adr = addrs.addr_P0;
  892. else
  893. adr = addrs.addr_P1;
  894. nrf24_set_mac(REG_RX_ADDR_P0, adr, addrs.addr_len);
  895. nrf24_set_mac(REG_TX_ADDR, adr, addrs.addr_len);
  896. } else {
  897. uint8_t buf[5];
  898. memcpy(buf, addrs.addr_P1, addrs.addr_len - 1);
  899. buf[addrs.addr_len - 1] = a == 2 ? addrs.addr_P2 :
  900. a == 3 ? addrs.addr_P3 :
  901. a == 4 ? addrs.addr_P4 :
  902. addrs.addr_P5;
  903. nrf24_set_mac(REG_RX_ADDR_P0, buf, addrs.addr_len);
  904. nrf24_set_mac(REG_TX_ADDR, buf, addrs.addr_len);
  905. }
  906. a = *(ptr + 1) >> 3;
  907. if(a == 0) a = 32;
  908. nrf24_write_reg(
  909. nrf24_HANDLE,
  910. REG_CONFIG,
  911. 0x70 | ((NRF_CRC == 1 ? 0b1000 :
  912. NRF_CRC == 2 ? 0b1100 :
  913. 0))); // Mask all interrupts
  914. nrf24_write_reg(nrf24_HANDLE, REG_DYNPD, NRF_DPL ? 0x3F : 0); // Enable dynamic payload reg
  915. last_packet_send_st = nrf24_txpacket(nrf24_HANDLE, ptr + 2, a, true);
  916. }
  917. last_packet_send = view_log_arr_idx;
  918. notification_message(
  919. APP->notification,
  920. last_packet_send_st ? &sequence_blink_blue_100 : &sequence_blink_red_100);
  921. if(what_to_do) start_scanning();
  922. return last_packet_send_st;
  923. }
  924. static void render_callback(Canvas* const canvas, void* ctx) {
  925. const PluginState* plugin_state = ctx;
  926. if(plugin_state == NULL) return;
  927. if(furi_mutex_acquire(plugin_state->mutex, 25) != FuriStatusOk) return;
  928. //canvas_draw_frame(canvas, 0, 0, 128, 64); // border around the edge of the screen
  929. if(what_doing == 0) {
  930. canvas_set_font(canvas, FontSecondary); // 8x10 font, 6 lines
  931. if(save_settings)
  932. snprintf(
  933. screen_buf, sizeof(screen_buf), "Save: %s", SETTINGS_FILENAME); // menu_selected = 0
  934. else
  935. snprintf(screen_buf, sizeof(screen_buf), "Load: %s", addr_file_name);
  936. canvas_draw_str(canvas, 10, 10, screen_buf);
  937. snprintf(screen_buf, sizeof(screen_buf), "Ch: %d", NRF_channel); // menu_selected = 1
  938. canvas_draw_str(canvas, 10, 20, screen_buf);
  939. if(NRF_ESB) {
  940. strcpy(screen_buf, "ESB");
  941. if(NRF_DPL) strcat(screen_buf, " DPL");
  942. canvas_draw_str(canvas, 78, 20, screen_buf);
  943. }
  944. if(NRF_AA_OFF) {
  945. canvas_draw_str(canvas, 61, 20, "AA");
  946. canvas_draw_line(canvas, 60, 21, 72, 12);
  947. }
  948. snprintf(
  949. screen_buf,
  950. sizeof(screen_buf),
  951. "Rate: %sbps",
  952. NRF_rate == 2 ? "2M" :
  953. NRF_rate == 1 ? "1M" :
  954. "250K"); // menu_selected = 2
  955. canvas_draw_str(canvas, 10, 30, screen_buf);
  956. if(what_to_do == 1)
  957. snprintf(screen_buf, sizeof(screen_buf), "Min Payl: %d", NRF_Payload_sniff_min);
  958. else if(what_to_do >= 2) {
  959. uint8_t* p = APP->log_arr + view_log_arr_idx * LOG_REC_SIZE;
  960. snprintf(
  961. screen_buf,
  962. sizeof(screen_buf),
  963. "Payload: %d",
  964. log_arr_idx && (*p & 0x80) ? *(p + 1) >> 3 : NRF_Payload);
  965. } else
  966. snprintf(screen_buf, sizeof(screen_buf), "Payload: %d", NRF_Payload);
  967. canvas_draw_str(canvas, 78, 30, screen_buf);
  968. strcpy(screen_buf, "Next Ch time: "); // menu_selected = 3
  969. if(find_channel_period == 0)
  970. strcat(screen_buf, "off");
  971. else
  972. snprintf(
  973. screen_buf + strlen(screen_buf), sizeof(screen_buf), "%d s", find_channel_period);
  974. canvas_draw_str(canvas, 10, 40, screen_buf);
  975. if(NRF_CRC == 1)
  976. canvas_draw_str(canvas, 99, 40, "CRC1");
  977. else if(NRF_CRC == 2)
  978. canvas_draw_str(canvas, 99, 40, "CRC2");
  979. snprintf(
  980. screen_buf,
  981. sizeof(screen_buf),
  982. "Log: %s",
  983. log_to_file == 0 ? "No" :
  984. log_to_file == 1 ? "Yes" :
  985. log_to_file == 2 ? "Append" :
  986. "Clear"); // menu_selected = 4
  987. canvas_draw_str(canvas, 10, 50, screen_buf);
  988. if(what_to_do) { // menu_selected = 5
  989. if(NRF_ERROR)
  990. snprintf(screen_buf, sizeof(screen_buf), "nRF24L01+ r/w ERROR!");
  991. else {
  992. if(what_to_do == 1)
  993. snprintf(screen_buf, sizeof(screen_buf), "Start sniff");
  994. else {
  995. uint8_t* p = APP->log_arr + view_log_arr_idx * LOG_REC_SIZE;
  996. if(log_arr_idx && (*p & 0x80)) { // +RAW
  997. snprintf(screen_buf, sizeof(screen_buf), "Start read: ");
  998. add_to_str_hex_bytes(screen_buf, (char*)p + 2, (*(p + 1) & 0b11) + 2);
  999. if(what_to_do == 2) strcpy(screen_buf + strlen(screen_buf) - 2, "* ");
  1000. } else
  1001. snprintf(
  1002. screen_buf,
  1003. sizeof(screen_buf),
  1004. "Start scan (pipes: %d)",
  1005. addrs.addr_count);
  1006. }
  1007. }
  1008. } else
  1009. snprintf(screen_buf, sizeof(screen_buf), "View log (pipes: %d)", addrs.addr_count);
  1010. canvas_draw_str(canvas, 10, 60, screen_buf);
  1011. canvas_draw_str(canvas, 0, menu_selected * 10 + 10, ">");
  1012. } else if(what_doing == 1) {
  1013. canvas_set_font(canvas, FontBatteryPercent); // 5x7 font, 9 lines
  1014. bool ch2 = false;
  1015. screen_buf[0] = '\0';
  1016. if(view_log_arr_x == 0) {
  1017. strcat(screen_buf, " ");
  1018. ch2 = true;
  1019. } else {
  1020. snprintf(screen_buf, sizeof(screen_buf), "<%d", view_log_arr_x);
  1021. if(view_log_arr_x < VIEW_LOG_MAX_X) ch2 = true;
  1022. }
  1023. snprintf(
  1024. screen_buf + strlen(screen_buf),
  1025. sizeof(screen_buf),
  1026. " %s ch: %d - %d.",
  1027. what_to_do == 1 ? "Sniff" :
  1028. what_to_do == 0 ? "View" :
  1029. "Read",
  1030. NRF_channel,
  1031. log_arr_idx);
  1032. canvas_draw_str(canvas, 0, 7, screen_buf);
  1033. if(ch2) canvas_draw_str(canvas, 121, 7, ">");
  1034. if(log_arr_idx) {
  1035. if(view_log_arr_idx >= log_arr_idx) view_log_arr_idx = log_arr_idx - 1;
  1036. uint16_t page = view_log_arr_idx & ~7;
  1037. for(uint8_t i = 0; i < 8 && page + i < log_arr_idx; i++) {
  1038. screen_buf[0] = (view_log_arr_idx & 7) != i ? ' ' :
  1039. last_packet_send != view_log_arr_idx ? '>' :
  1040. last_packet_send_st ? '*' :
  1041. '!';
  1042. screen_buf[1] = '\0';
  1043. uint8_t* ptr = APP->log_arr + (page + i) * LOG_REC_SIZE;
  1044. uint8_t channel = *ptr++;
  1045. uint8_t* crcptr = NULL;
  1046. uint8_t pre = 0;
  1047. int count = 0;
  1048. if(channel & 0x80) { // RAW packet: nn:>{.address..}-xxxxxxxx
  1049. uint8_t pktinfo = *ptr++;
  1050. bool _PCF = pktinfo & 0b100;
  1051. uint8_t plen = count = (pktinfo >> 3);
  1052. uint8_t adrsize = (pktinfo & 0b11) + 2;
  1053. plen += adrsize;
  1054. count += view_log_decode_CRC;
  1055. if(view_log_arr_x > 0) count -= view_log_arr_x - 1;
  1056. uint8_t max_width = VIEW_LOG_WIDTH_B;
  1057. if(view_log_arr_x == 0) max_width -= 5;
  1058. if(count > max_width) count = max_width;
  1059. if(count > 0) {
  1060. uint8_t* pcrc = ptr;
  1061. uint32_t crc;
  1062. crc = view_log_decode_CRC == 2 ? 0xFFFF : 0xFF;
  1063. crc = calc_crc(crc, pcrc, 7, (_PCF ? 9 : 0) + plen * 8);
  1064. pcrc += plen;
  1065. if(_PCF) { //ESB
  1066. pcrc++;
  1067. if(crc == get_shifted_crc(pcrc)) crcptr = pcrc;
  1068. } else {
  1069. if((view_log_decode_CRC == 1 && crc == *pcrc) ||
  1070. (view_log_decode_CRC == 2 &&
  1071. crc == (uint32_t)((*pcrc << 8) | *(pcrc + 1)))) {
  1072. crcptr = pcrc;
  1073. }
  1074. }
  1075. if(view_log_arr_x == 0) {
  1076. add_to_str_hex_bytes(screen_buf, (char*)ptr, adrsize);
  1077. for(int8_t j = 5 - adrsize; j > 0; j--)
  1078. strcat(screen_buf, " ");
  1079. strcat(screen_buf, "-");
  1080. pre += 5 * 2 + 1;
  1081. } else {
  1082. ptr += view_log_arr_x - 1;
  1083. }
  1084. ptr += adrsize;
  1085. if(_PCF)
  1086. add_to_str_hex_bytes_shift_9b(screen_buf, (char*)ptr++, count);
  1087. else
  1088. add_to_str_hex_bytes(screen_buf, (char*)ptr, count);
  1089. }
  1090. } else {
  1091. uint8_t dpl = *ptr++;
  1092. uint8_t pipe = dpl & 0b111;
  1093. dpl >>= 3;
  1094. if(dpl == 0) dpl = 32;
  1095. count = dpl - view_log_arr_x;
  1096. if(view_log_decode_PCF) count--;
  1097. uint8_t max_width = VIEW_LOG_WIDTH_B;
  1098. if(view_log_arr_x == 0) {
  1099. if(addrs.addr_count > 1) max_width--;
  1100. if(view_log_decode_PCF) max_width -= 2;
  1101. }
  1102. if(count > max_width) count = max_width;
  1103. if(count > 0) {
  1104. if(view_log_decode_CRC) {
  1105. static uint16_t prev_addr_CRC;
  1106. static int8_t prev_pipe = -1;
  1107. uint8_t* pcrc = ptr;
  1108. uint32_t crc;
  1109. if(prev_pipe == pipe) {
  1110. crc = prev_addr_CRC;
  1111. } else {
  1112. crc = view_log_decode_CRC == 2 ? 0xFFFF : 0xFF;
  1113. if(pipe <= 1) {
  1114. crc = calc_crc(
  1115. crc,
  1116. pipe == 0 ? addrs.addr_P0 : addrs.addr_P1,
  1117. 7,
  1118. addrs.addr_len * 8);
  1119. } else {
  1120. crc =
  1121. calc_crc(crc, addrs.addr_P1, 7, (addrs.addr_len - 1) * 8);
  1122. crc = calc_crc(
  1123. crc,
  1124. pipe == 2 ? &addrs.addr_P2 :
  1125. pipe == 3 ? &addrs.addr_P3 :
  1126. pipe == 4 ? &addrs.addr_P4 :
  1127. &addrs.addr_P5,
  1128. 7,
  1129. 8);
  1130. }
  1131. prev_addr_CRC = crc;
  1132. prev_pipe = pipe;
  1133. }
  1134. if(view_log_decode_PCF) {
  1135. crc = calc_crc(crc, pcrc++, 7, 9);
  1136. if(crc == get_shifted_crc(pcrc)) crcptr = pcrc;
  1137. if(crcptr == NULL) {
  1138. for(int8_t j = 0; j < (int8_t)dpl - view_log_decode_CRC - 1;
  1139. j++) {
  1140. crc = calc_crc(crc, pcrc++, 6, 8);
  1141. if(crc == get_shifted_crc(pcrc)) {
  1142. crcptr = pcrc;
  1143. break;
  1144. }
  1145. }
  1146. }
  1147. } else {
  1148. for(int8_t j = 0; j < (int8_t)dpl - view_log_decode_CRC; j++) {
  1149. crc = calc_crc(crc, pcrc++, 7, 8);
  1150. if((view_log_decode_CRC == 1 && crc == *pcrc) ||
  1151. (view_log_decode_CRC == 2 &&
  1152. crc == (uint32_t)((*pcrc << 8) | *(pcrc + 1)))) {
  1153. crcptr = pcrc;
  1154. break;
  1155. }
  1156. }
  1157. }
  1158. }
  1159. }
  1160. ptr += view_log_arr_x;
  1161. if(max_width < VIEW_LOG_WIDTH_B) {
  1162. pre += snprintf(screen_buf + 1, 10, "%X-", pipe);
  1163. if(view_log_decode_PCF) {
  1164. pre += snprintf(
  1165. screen_buf + strlen(screen_buf),
  1166. 10,
  1167. "%02X%01X-",
  1168. *ptr >> 2,
  1169. ((*ptr & 3) << 1) | (*(ptr + 1) >> 7));
  1170. }
  1171. }
  1172. if(view_log_decode_PCF)
  1173. add_to_str_hex_bytes_shift_9b(screen_buf, (char*)ptr++, count);
  1174. else
  1175. add_to_str_hex_bytes(screen_buf, (char*)ptr, count);
  1176. }
  1177. uint16_t y = 14 + i * 7;
  1178. canvas_draw_str(canvas, 3 * 5, y, screen_buf);
  1179. uint16_t x = page + i + 1;
  1180. if(x > 99) {
  1181. snprintf(screen_buf, 16, "%d", x);
  1182. canvas_draw_str(canvas, 1, y, screen_buf + 1);
  1183. canvas_draw_frame(canvas, 0, y - 2, 1, 2);
  1184. x = 2;
  1185. } else {
  1186. x = snprintf(screen_buf, 16, "%d", x);
  1187. canvas_draw_str(canvas, 0, y, screen_buf);
  1188. }
  1189. if(crcptr) { // 5x7 font, 9 lines
  1190. canvas_draw_str(canvas, x * 5, y, "=");
  1191. int n = crcptr - (uint8_t*)ptr;
  1192. if(n > -view_log_decode_CRC && n < count) {
  1193. int len;
  1194. x = (4 + pre) * 5;
  1195. if(n < 0) {
  1196. len = view_log_decode_CRC + n;
  1197. n = 0;
  1198. } else {
  1199. len = MIN(view_log_decode_CRC, count - n);
  1200. x += n * 2 * 5;
  1201. canvas_draw_line(canvas, x - 1, y, x - 1, y - 1);
  1202. }
  1203. canvas_draw_line(canvas, x - 1, y, n = x + len * 2 * 5 - 1, y);
  1204. canvas_draw_line(canvas, n, y, n, y - 1);
  1205. }
  1206. } else
  1207. canvas_draw_str(canvas, x * 5, y, ":");
  1208. }
  1209. }
  1210. } else {
  1211. canvas_set_font(canvas, FontBatteryPercent); // 5x7 font, 9 lines
  1212. if(view_found >= 0) {
  1213. snprintf(screen_buf, 50, "Found > 1: %d", found_total);
  1214. canvas_draw_str(canvas, 0, 1 * 7, screen_buf);
  1215. int16_t idx = view_found * 7;
  1216. for(uint8_t i = 0; i < 7; i++, idx++) {
  1217. if(idx >= found_total) break;
  1218. snprintf(screen_buf, 16, "%d. ", idx + 1);
  1219. add_to_str_hex_bytes(
  1220. screen_buf, (char*)APP->found[idx].addr, APP->found[idx].addr_size);
  1221. if(APP->found[idx].addr_size == 3)
  1222. strcat(screen_buf, " ");
  1223. else if(APP->found[idx].addr_size == 4)
  1224. strcat(screen_buf, " ");
  1225. snprintf(screen_buf + strlen(screen_buf), 16, " - %d", APP->found[idx].total);
  1226. canvas_draw_str(canvas, 0, (2 + i) * 7, screen_buf);
  1227. }
  1228. } else {
  1229. struct ADDRS* a;
  1230. if(what_to_do == 1) {
  1231. a = &addrs_sniff;
  1232. canvas_draw_str(canvas, 0, 1 * 7, "Sniff prefix:");
  1233. } else {
  1234. a = &addrs;
  1235. canvas_draw_str(canvas, 0, 1 * 7, "Addresses:");
  1236. }
  1237. if(a->addr_count > 0) {
  1238. snprintf(screen_buf, sizeof(screen_buf), "P0: ");
  1239. add_to_str_hex_bytes(screen_buf, (char*)a->addr_P0, a->addr_len);
  1240. canvas_draw_str(canvas, 0, 2 * 7, screen_buf);
  1241. }
  1242. if(a->addr_count > 1) {
  1243. snprintf(screen_buf, sizeof(screen_buf), "P1: ");
  1244. add_to_str_hex_bytes(screen_buf, (char*)a->addr_P1, a->addr_len);
  1245. canvas_draw_str(canvas, 0, 3 * 7, screen_buf);
  1246. }
  1247. if(a->addr_count > 2) {
  1248. canvas_draw_str(canvas, 0, 4 * 7, "P2: ");
  1249. snprintf(screen_buf, sizeof(screen_buf), "%02X", a->addr_P2);
  1250. canvas_draw_str(canvas, (4 + (a->addr_len - 1) * 2) * 5, 4 * 7, screen_buf);
  1251. }
  1252. if(a->addr_count > 3) {
  1253. canvas_draw_str(canvas, 0, 5 * 7, "P3: ");
  1254. snprintf(screen_buf, sizeof(screen_buf), "%02X", a->addr_P3);
  1255. canvas_draw_str(canvas, (4 + (a->addr_len - 1) * 2) * 5, 5 * 7, screen_buf);
  1256. }
  1257. if(a->addr_count > 4) {
  1258. canvas_draw_str(canvas, 0, 6 * 7, "P4: ");
  1259. snprintf(screen_buf, sizeof(screen_buf), "%02X", a->addr_P4);
  1260. canvas_draw_str(canvas, (4 + (a->addr_len - 1) * 2) * 5, 6 * 7, screen_buf);
  1261. }
  1262. if(a->addr_count > 5) {
  1263. canvas_draw_str(canvas, 0, 7 * 7, "P5: ");
  1264. snprintf(screen_buf, sizeof(screen_buf), "%02X", a->addr_P5);
  1265. canvas_draw_str(canvas, (4 + (a->addr_len - 1) * 2) * 5, 7 * 7, screen_buf);
  1266. }
  1267. }
  1268. if(log_arr_idx) {
  1269. uint8_t* ptr = APP->log_arr + view_log_arr_idx * LOG_REC_SIZE;
  1270. uint8_t pktinfo = *(ptr + 1);
  1271. snprintf(screen_buf, 32, ">Ch: %d L: %d", *ptr & 0x7F, pktinfo >> 3);
  1272. if(*ptr & 0x80) {
  1273. strcat(screen_buf, " RAW");
  1274. if(pktinfo & 0b100) {
  1275. snprintf(
  1276. screen_buf + strlen(screen_buf),
  1277. 16,
  1278. " ESB %s",
  1279. *(ptr + 2 + (pktinfo & 0b11) + 2) >> 2 != 0x33 ? "DPL" : "");
  1280. }
  1281. }
  1282. canvas_draw_str(canvas, 0, 8 * 7, screen_buf);
  1283. }
  1284. screen_buf[0] = 'v';
  1285. strcpy(screen_buf + 1, VERSION);
  1286. canvas_draw_str(canvas, 105, 7, screen_buf);
  1287. if(view_log_decode_PCF || view_log_decode_CRC) {
  1288. strcpy(screen_buf, "Decode: ");
  1289. if(view_log_decode_PCF) strcat(screen_buf, "ESB ");
  1290. if(view_log_decode_CRC == 1)
  1291. strcat(screen_buf, "CRC1");
  1292. else if(view_log_decode_CRC == 2)
  1293. strcat(screen_buf, "CRC2");
  1294. canvas_draw_str(canvas, 0, 64, screen_buf);
  1295. }
  1296. }
  1297. furi_mutex_release(plugin_state->mutex);
  1298. }
  1299. int32_t nrf24scan_app(void* p) {
  1300. UNUSED(p);
  1301. APP = malloc(sizeof(Nrf24Scan));
  1302. APP->event_queue = furi_message_queue_alloc(8, sizeof(PluginEvent));
  1303. PluginState* plugin_state = malloc(sizeof(PluginState));
  1304. plugin_state->mutex = furi_mutex_alloc(FuriMutexTypeNormal);
  1305. if(!plugin_state->mutex) {
  1306. furi_message_queue_free(APP->event_queue);
  1307. FURI_LOG_E(TAG, "cannot create mutex");
  1308. free(plugin_state);
  1309. return 255;
  1310. }
  1311. APP->log_arr = malloc(LOG_REC_SIZE * MAX_LOG_RECORDS);
  1312. if(APP->log_arr == NULL) {
  1313. FURI_LOG_E(TAG, "Not enouch memory: %d", LOG_REC_SIZE * MAX_LOG_RECORDS);
  1314. strcpy(addr_file_name, "MEMORY LOW!");
  1315. }
  1316. clear_log();
  1317. APP->found = malloc(sizeof(struct FOUND) * MAX_FOUND_RECORDS);
  1318. if(APP->found == NULL) {
  1319. FURI_LOG_E(TAG, "Not enouch memory: %d", sizeof(struct FOUND) * MAX_FOUND_RECORDS);
  1320. strcpy(addr_file_name, "MEMORY LOW!!");
  1321. }
  1322. memset((uint8_t*)&addrs, 0, sizeof(addrs));
  1323. memset((uint8_t*)&addrs_sniff, 0, sizeof(addrs_sniff));
  1324. if(!furi_hal_power_is_otg_enabled()) {
  1325. furi_hal_power_enable_otg();
  1326. NRF_BOARD_POWER_5V = true;
  1327. furi_delay_ms(100);
  1328. }
  1329. nrf24_init();
  1330. // Set system callbacks
  1331. APP->view_port = view_port_alloc();
  1332. view_port_draw_callback_set(APP->view_port, render_callback, plugin_state);
  1333. view_port_input_callback_set(APP->view_port, input_callback, APP->event_queue);
  1334. // Open GUI and register view_port
  1335. APP->gui = furi_record_open(RECORD_GUI);
  1336. gui_add_view_port(APP->gui, APP->view_port, GuiLayerFullscreen);
  1337. APP->notification = furi_record_open(RECORD_NOTIFICATION);
  1338. APP->storage = furi_record_open(RECORD_STORAGE);
  1339. storage_common_migrate(APP->storage, EXT_PATH("nrf24scan"), SCAN_APP_PATH_FOLDER);
  1340. storage_common_mkdir(APP->storage, SCAN_APP_PATH_FOLDER);
  1341. Stream* file_stream = file_stream_alloc(APP->storage);
  1342. FuriString* path = furi_string_alloc();
  1343. furi_string_set(path, SCAN_APP_PATH_FOLDER);
  1344. furi_string_cat(path, "/");
  1345. furi_string_cat(path, SNIFF_FILENAME);
  1346. if(file_stream_open(file_stream, furi_string_get_cstr(path), FSAM_READ, FSOM_OPEN_EXISTING)) {
  1347. uint8_t err = load_settings_file(file_stream);
  1348. if(!err)
  1349. strncpy(
  1350. addr_file_name,
  1351. furi_string_get_cstr(path) + sizeof(SCAN_APP_PATH_FOLDER),
  1352. sizeof(addr_file_name));
  1353. else
  1354. snprintf(addr_file_name, sizeof(addr_file_name), "LOAD ERROR#%d", err);
  1355. } else {
  1356. strcpy(addr_file_name, "NONE");
  1357. if(what_to_do == 1) {
  1358. addrs.addr_P0[0] = 0;
  1359. addrs.addr_P0[1] = 0x55;
  1360. addrs.addr_len = 2;
  1361. addrs.addr_count = 1;
  1362. view_log_decode_CRC = NRF_CRC = 2;
  1363. NRF_Payload_sniff_min = 0; // Min
  1364. }
  1365. }
  1366. file_stream_close(file_stream);
  1367. stream_free(file_stream);
  1368. furi_string_free(path);
  1369. PluginEvent event;
  1370. for(bool processing = true; processing;) {
  1371. FuriStatus event_status = furi_message_queue_get(APP->event_queue, &event, 100);
  1372. furi_mutex_acquire(plugin_state->mutex, FuriWaitForever);
  1373. if(event_status == FuriStatusOk) {
  1374. // press events
  1375. if(event.type == EventTypeKey) {
  1376. //FURI_LOG_D(TAG, "Key: %d Type: %d Sec: %u", event.input.key, event.input.type, event.input.sequence);
  1377. switch(event.input.key) {
  1378. case InputKeyUp:
  1379. if(event.input.type == InputTypeShort || event.input.type == InputTypeRepeat) {
  1380. if(what_doing == 0) {
  1381. if(menu_selected > 0)
  1382. menu_selected--;
  1383. else
  1384. menu_selected = menu_selected_max;
  1385. } else if(what_doing == 1) {
  1386. view_log_arr_idx -= event.input.type == InputTypeRepeat ? 10 : 1;
  1387. if(view_log_arr_idx >= log_arr_idx) view_log_arr_idx = 0;
  1388. } else if(what_doing == 2) {
  1389. if(view_found > -1) view_found--;
  1390. }
  1391. }
  1392. break;
  1393. case InputKeyDown:
  1394. if(event.input.type == InputTypeShort || event.input.type == InputTypeRepeat) {
  1395. if(what_doing == 0) {
  1396. if(menu_selected < menu_selected_max)
  1397. menu_selected++;
  1398. else
  1399. menu_selected = 0;
  1400. } else if(what_doing == 1) {
  1401. if(log_arr_idx == 0)
  1402. view_log_arr_idx = 0;
  1403. else {
  1404. view_log_arr_idx += event.input.type == InputTypeRepeat ? 10 : 1;
  1405. if(view_log_arr_idx >= log_arr_idx)
  1406. view_log_arr_idx = log_arr_idx - 1;
  1407. }
  1408. } else if(what_doing == 2) {
  1409. if(view_found < found_total / 7) view_found++;
  1410. }
  1411. }
  1412. break;
  1413. case InputKeyLeft:
  1414. if(event.input.type == InputTypeShort || event.input.type == InputTypeRepeat) {
  1415. if(what_doing == 0) {
  1416. switch(menu_selected) {
  1417. case Menu_enter_channel:
  1418. NRF_channel -= event.input.type == InputTypeRepeat ? 10 : 1;
  1419. if(NRF_channel > MAX_CHANNEL) NRF_channel = MAX_CHANNEL;
  1420. break;
  1421. case Menu_enter_rate:
  1422. if(what_to_do == 1) { // SNIFF
  1423. NRF_Payload_sniff_min -=
  1424. event.input.type == InputTypeRepeat ? 10 : 1;
  1425. correct_NRF_Payload_sniff_min();
  1426. } else {
  1427. NRF_Payload -= event.input.type == InputTypeRepeat ? 10 : 1;
  1428. if(NRF_Payload > 32) NRF_Payload = 0;
  1429. }
  1430. break;
  1431. case Menu_enter_scan_period:
  1432. find_channel_period -= event.input.type == InputTypeRepeat ? 10 :
  1433. 1;
  1434. if(find_channel_period < 0) find_channel_period = 0;
  1435. break;
  1436. case Menu_log:
  1437. if(--log_to_file < -1) log_to_file = 2;
  1438. break;
  1439. case Menu_ok:
  1440. if(--what_to_do > 3) what_to_do = 3;
  1441. break;
  1442. }
  1443. } else if(what_doing == 1) {
  1444. if(view_log_arr_x > 0) view_log_arr_x--;
  1445. } else if(what_doing == 2) {
  1446. //if(NRF_ESB == 0)
  1447. view_log_decode_PCF ^= 1;
  1448. }
  1449. }
  1450. break;
  1451. case InputKeyRight:
  1452. if(event.input.type == InputTypeShort || event.input.type == InputTypeRepeat) {
  1453. if(what_doing == 0) {
  1454. switch(menu_selected) {
  1455. case Menu_open_file:
  1456. save_settings ^= 1;
  1457. break;
  1458. case Menu_enter_channel:
  1459. NRF_channel += event.input.type == InputTypeRepeat ? 10 : 1;
  1460. if(NRF_channel > MAX_CHANNEL) NRF_channel = 0;
  1461. break;
  1462. case Menu_enter_rate:
  1463. if(what_to_do == 1) { // SNIFF
  1464. NRF_Payload_sniff_min +=
  1465. event.input.type == InputTypeRepeat ? 10 : 1;
  1466. correct_NRF_Payload_sniff_min();
  1467. } else {
  1468. NRF_Payload += event.input.type == InputTypeRepeat ? 10 : 1;
  1469. if(NRF_Payload > 32) NRF_Payload = 32;
  1470. }
  1471. break;
  1472. case Menu_enter_scan_period:
  1473. find_channel_period += event.input.type == InputTypeRepeat ? 10 :
  1474. 1;
  1475. break;
  1476. case Menu_log:
  1477. if(++log_to_file > 2) log_to_file = -1;
  1478. break;
  1479. case Menu_ok:
  1480. if(++what_to_do > 3) what_to_do = 0;
  1481. break;
  1482. }
  1483. } else if(what_doing == 1) {
  1484. if(view_log_arr_x < VIEW_LOG_MAX_X) view_log_arr_x++;
  1485. } else if(what_doing == 2) {
  1486. if(++view_log_decode_CRC > 2) view_log_decode_CRC = 0;
  1487. }
  1488. }
  1489. break;
  1490. case InputKeyOk:
  1491. if(event.input.type == InputTypeShort) {
  1492. if(what_doing == 0) {
  1493. switch(menu_selected) {
  1494. case Menu_open_file:
  1495. if(save_settings) {
  1496. write_to_log_file(APP->storage, true);
  1497. } else {
  1498. file_stream = file_stream_alloc(APP->storage);
  1499. if(select_settings_file(file_stream)) {
  1500. uint8_t err = load_settings_file(file_stream);
  1501. if(!err)
  1502. save_to_new_log = true;
  1503. else
  1504. snprintf(
  1505. addr_file_name,
  1506. sizeof(addr_file_name),
  1507. "LOAD ERROR#%d",
  1508. err);
  1509. file_stream_close(file_stream);
  1510. menu_selected = Menu_ok;
  1511. }
  1512. stream_free(file_stream);
  1513. }
  1514. break;
  1515. case Menu_enter_channel:
  1516. if(what_to_do == 1) {
  1517. if(NRF_ESB)
  1518. NRF_DPL = NRF_ESB = 0;
  1519. else
  1520. NRF_ESB = 1;
  1521. } else {
  1522. if(NRF_ESB) {
  1523. if(NRF_DPL)
  1524. NRF_DPL = NRF_ESB = 0;
  1525. else
  1526. NRF_DPL = 1;
  1527. } else
  1528. NRF_ESB = 1;
  1529. }
  1530. break;
  1531. case Menu_enter_rate:
  1532. NRF_rate++;
  1533. if(NRF_rate > 2) NRF_rate = 0;
  1534. break;
  1535. case Menu_enter_scan_period:
  1536. if(++NRF_CRC > 2) NRF_CRC = what_to_do == 1 ? 1 : 0;
  1537. break;
  1538. case Menu_ok:
  1539. if(what_to_do) {
  1540. if((addrs.addr_count ||
  1541. (what_to_do >= 2 && log_arr_idx &&
  1542. *(APP->log_arr + view_log_arr_idx * LOG_REC_SIZE) &
  1543. 0x80)) ||
  1544. what_to_do == 1) {
  1545. if(log_to_file == -1) {
  1546. log_to_file = 0;
  1547. clear_log();
  1548. save_to_new_log = true;
  1549. } else if(log_to_file == 1)
  1550. save_to_new_log = true;
  1551. start_scanning();
  1552. if(!NRF_ERROR) what_doing = 1;
  1553. }
  1554. } else
  1555. what_doing = 1;
  1556. break;
  1557. }
  1558. } else if(what_doing == 1) {
  1559. what_doing = 2;
  1560. } else if(what_doing == 2) {
  1561. what_doing = 1;
  1562. }
  1563. } else if(event.input.type == InputTypeLong) {
  1564. if(what_doing == 0) {
  1565. if(menu_selected == Menu_enter_channel) {
  1566. NRF_AA_OFF ^= 1;
  1567. } else if(menu_selected == Menu_log) { // Log
  1568. if(log_arr_idx && (log_to_file == 1 || log_to_file == 2)) {
  1569. write_to_log_file(APP->storage, false);
  1570. clear_log();
  1571. }
  1572. }
  1573. } else if(what_doing == 1 || what_doing == 2) {
  1574. nrf24_send_packet();
  1575. }
  1576. }
  1577. break;
  1578. case InputKeyBack:
  1579. if(event.input.type == InputTypeLong)
  1580. processing = false;
  1581. else if(event.input.type == InputTypeShort) {
  1582. if(what_doing) what_doing--;
  1583. if(what_doing == 0) {
  1584. nrf24_set_idle(nrf24_HANDLE);
  1585. }
  1586. }
  1587. break;
  1588. default:
  1589. break;
  1590. }
  1591. }
  1592. }
  1593. if(what_doing && what_to_do) {
  1594. nrf24_read_newpacket();
  1595. if(find_channel_period &&
  1596. furi_get_tick() - start_time >= (uint32_t)find_channel_period * 1000UL) {
  1597. if(++NRF_channel > MAX_CHANNEL) NRF_channel = 0;
  1598. start_scanning();
  1599. }
  1600. }
  1601. furi_mutex_release(plugin_state->mutex);
  1602. view_port_update(APP->view_port);
  1603. }
  1604. nrf24_set_idle(nrf24_HANDLE);
  1605. if(log_arr_idx && (log_to_file == 1 || log_to_file == 2)) {
  1606. write_to_log_file(APP->storage, false);
  1607. }
  1608. nrf24_deinit();
  1609. if(NRF_BOARD_POWER_5V) furi_hal_power_disable_otg();
  1610. view_port_enabled_set(APP->view_port, false);
  1611. gui_remove_view_port(APP->gui, APP->view_port);
  1612. furi_record_close(RECORD_GUI);
  1613. furi_record_close(RECORD_NOTIFICATION);
  1614. furi_record_close(RECORD_STORAGE);
  1615. view_port_free(APP->view_port);
  1616. furi_message_queue_free(APP->event_queue);
  1617. if(APP->log_arr) free(APP->log_arr);
  1618. if(APP->found) free(APP->found);
  1619. furi_mutex_free(plugin_state->mutex);
  1620. free(plugin_state);
  1621. free(APP);
  1622. return 0;
  1623. }