CommandLine.cpp 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675
  1. #include "CommandLine.h"
  2. CommandLine::CommandLine() {
  3. }
  4. void CommandLine::RunSetup() {
  5. Serial.println(this->ascii_art);
  6. Serial.println(F("\n\n--------------------------------\n"));
  7. Serial.println(F(" ESP32 Marauder \n"));
  8. Serial.println(" " + version_number + "\n");
  9. Serial.println(F(" By: justcallmekoko\n"));
  10. Serial.println(F("--------------------------------\n\n"));
  11. Serial.print("> ");
  12. }
  13. String CommandLine::getSerialInput() {
  14. String input = "";
  15. if (Serial.available() > 0)
  16. input = Serial.readStringUntil('\n');
  17. input.trim();
  18. return input;
  19. }
  20. void CommandLine::main(uint32_t currentTime) {
  21. String input = this->getSerialInput();
  22. this->runCommand(input);
  23. if (input != "")
  24. Serial.print("> ");
  25. }
  26. LinkedList<String> CommandLine::parseCommand(String input, char* delim) {
  27. LinkedList<String> cmd_args;
  28. if (input != "") {
  29. char fancy[input.length() + 1] = {};
  30. input.toCharArray(fancy, input.length() + 1);
  31. char* ptr = strtok(fancy, delim);
  32. while (ptr != NULL) {
  33. cmd_args.add(String(ptr));
  34. ptr = strtok(NULL, delim);
  35. }
  36. }
  37. return cmd_args;
  38. }
  39. int CommandLine::argSearch(LinkedList<String>* cmd_args_list, String key) {
  40. for (int i = 0; i < cmd_args_list->size(); i++) {
  41. if (cmd_args_list->get(i) == key)
  42. return i;
  43. }
  44. return -1;
  45. }
  46. bool CommandLine::checkValueExists(LinkedList<String>* cmd_args_list, int index) {
  47. if (index < cmd_args_list->size() - 1)
  48. return true;
  49. return false;
  50. }
  51. bool CommandLine::inRange(int max, int index) {
  52. if ((index >= 0) && (index < max))
  53. return true;
  54. return false;
  55. }
  56. bool CommandLine::apSelected() {
  57. for (int i = 0; i < access_points->size(); i++) {
  58. if (access_points->get(i).selected)
  59. return true;
  60. }
  61. return false;
  62. }
  63. bool CommandLine::hasSSIDs() {
  64. if (ssids->size() == 0)
  65. return false;
  66. return true;
  67. }
  68. void CommandLine::runCommand(String input) {
  69. if (input != "")
  70. Serial.println("#" + input);
  71. else
  72. return;
  73. LinkedList<String> cmd_args = this->parseCommand(input, " ");
  74. //// Admin commands
  75. // Help
  76. if (cmd_args.get(0) == HELP_CMD) {
  77. Serial.println(HELP_HEAD);
  78. Serial.println(HELP_CH_CMD);
  79. Serial.println(HELP_SETTINGS_CMD);
  80. Serial.println(HELP_CLEARAP_CMD_A);
  81. Serial.println(HELP_CLEARAP_CMD_B);
  82. Serial.println(HELP_REBOOT_CMD);
  83. Serial.println(HELP_UPDATE_CMD_A);
  84. Serial.println(HELP_UPDATE_CMD_B);
  85. // WiFi sniff/scan
  86. Serial.println(HELP_SCANAP_CMD);
  87. Serial.println(HELP_SCANSTA_CMD);
  88. Serial.println(HELP_SNIFF_RAW_CMD);
  89. Serial.println(HELP_SNIFF_BEACON_CMD);
  90. Serial.println(HELP_SNIFF_PROBE_CMD);
  91. Serial.println(HELP_SNIFF_PWN_CMD);
  92. Serial.println(HELP_SNIFF_ESP_CMD);
  93. Serial.println(HELP_SNIFF_DEAUTH_CMD);
  94. Serial.println(HELP_SNIFF_PMKID_CMD);
  95. Serial.println(HELP_STOPSCAN_CMD);
  96. // WiFi attack
  97. Serial.println(HELP_ATTACK_CMD);
  98. // WiFi Aux
  99. Serial.println(HELP_LIST_AP_CMD_A);
  100. Serial.println(HELP_LIST_AP_CMD_B);
  101. Serial.println(HELP_LIST_AP_CMD_C);
  102. Serial.println(HELP_SEL_CMD_A);
  103. Serial.println(HELP_SEL_CMD_B);
  104. Serial.println(HELP_SSID_CMD_A);
  105. Serial.println(HELP_SSID_CMD_B);
  106. // Bluetooth sniff/scan
  107. Serial.println(HELP_BT_SNIFF_CMD);
  108. Serial.println(HELP_BT_SKIM_CMD);
  109. Serial.println(HELP_FOOT);
  110. return;
  111. }
  112. // Stop Scan
  113. if (cmd_args.get(0) == STOPSCAN_CMD) {
  114. if (wifi_scan_obj.currentScanMode == OTA_UPDATE) {
  115. wifi_scan_obj.currentScanMode = WIFI_SCAN_OFF;
  116. //#ifdef HAS_SCREEN
  117. // menu_function_obj.changeMenu(menu_function_obj.updateMenu.parentMenu);
  118. //#endif
  119. WiFi.softAPdisconnect(true);
  120. web_obj.shutdownServer();
  121. return;
  122. }
  123. wifi_scan_obj.StartScan(WIFI_SCAN_OFF);
  124. Serial.println("Stopping WiFi tran/recv");
  125. // If we don't do this, the text and button coordinates will be off
  126. #ifdef HAS_SCREEN
  127. display_obj.tft.init();
  128. menu_function_obj.changeMenu(menu_function_obj.current_menu);
  129. #endif
  130. }
  131. // Channel command
  132. else if (cmd_args.get(0) == CH_CMD) {
  133. // Search for channel set arg
  134. int ch_set = this->argSearch(&cmd_args, "-s");
  135. if (cmd_args.size() == 1) {
  136. Serial.println("Current channel: " + (String)wifi_scan_obj.set_channel);
  137. }
  138. else if (ch_set != -1) {
  139. wifi_scan_obj.set_channel = cmd_args.get(ch_set + 1).toInt();
  140. wifi_scan_obj.changeChannel();
  141. Serial.println("Set channel: " + (String)wifi_scan_obj.set_channel);
  142. }
  143. }
  144. // Clear APs
  145. else if (cmd_args.get(0) == CLEARAP_CMD) {
  146. int ap_sw = this->argSearch(&cmd_args, "-a"); // APs
  147. int ss_sw = this->argSearch(&cmd_args, "-s"); // SSIDs
  148. if (ap_sw != -1)
  149. wifi_scan_obj.RunClearAPs();
  150. if (ss_sw != -1)
  151. wifi_scan_obj.RunClearSSIDs();
  152. }
  153. else if (cmd_args.get(0) == SETTINGS_CMD) {
  154. int ss_sw = this->argSearch(&cmd_args, "-s"); // Set setting
  155. int re_sw = this->argSearch(&cmd_args, "-r"); // Reset setting
  156. int en_sw = this->argSearch(&cmd_args, "enable"); // enable setting
  157. int da_sw = this->argSearch(&cmd_args, "disable"); // disable setting
  158. if (re_sw != -1) {
  159. settings_obj.createDefaultSettings(SPIFFS);
  160. return;
  161. }
  162. if (ss_sw == -1) {
  163. settings_obj.printJsonSettings(settings_obj.getSettingsString());
  164. }
  165. else {
  166. bool result = false;
  167. String setting_name = cmd_args.get(ss_sw + 1);
  168. if (en_sw != -1)
  169. result = settings_obj.saveSetting<bool>(setting_name, true);
  170. else if (da_sw != -1)
  171. result = settings_obj.saveSetting<bool>(setting_name, false);
  172. else {
  173. Serial.println("You did not properly enable/disable this setting.");
  174. return;
  175. }
  176. if (!result) {
  177. Serial.println("Could not successfully update setting \"" + setting_name + "\"");
  178. return;
  179. }
  180. }
  181. }
  182. else if (cmd_args.get(0) == REBOOT_CMD) {
  183. Serial.println("Rebooting...");
  184. ESP.restart();
  185. }
  186. //// WiFi/Bluetooth Scan/Attack commands
  187. if (!wifi_scan_obj.scanning()) {
  188. // AP Scan
  189. if (cmd_args.get(0) == SCANAP_CMD) {
  190. int full_sw = -1;
  191. #ifdef HAS_SCREEN
  192. display_obj.clearScreen();
  193. menu_function_obj.drawStatusBar();
  194. #endif
  195. if (full_sw == -1) {
  196. Serial.println("Starting AP scan. Stop with " + (String)STOPSCAN_CMD);
  197. wifi_scan_obj.StartScan(WIFI_SCAN_TARGET_AP, TFT_MAGENTA);
  198. }
  199. else {
  200. Serial.println("Starting Full AP scan. Stop with " + (String)STOPSCAN_CMD);
  201. wifi_scan_obj.StartScan(WIFI_SCAN_TARGET_AP_FULL, TFT_MAGENTA);
  202. }
  203. }
  204. // Raw sniff
  205. else if (cmd_args.get(0) == SNIFF_RAW_CMD) {
  206. Serial.println("Starting Raw sniff. Stop with " + (String)STOPSCAN_CMD);
  207. #ifdef HAS_SCREEN
  208. display_obj.clearScreen();
  209. menu_function_obj.drawStatusBar();
  210. #endif
  211. wifi_scan_obj.StartScan(WIFI_SCAN_RAW_CAPTURE, TFT_WHITE);
  212. }
  213. // Scan stations
  214. else if (cmd_args.get(0) == SCANSTA_CMD) {
  215. Serial.println("Starting Station scan. Stop with " + (String)STOPSCAN_CMD);
  216. #ifdef HAS_SCREEN
  217. display_obj.clearScreen();
  218. menu_function_obj.drawStatusBar();
  219. #endif
  220. wifi_scan_obj.StartScan(WIFI_SCAN_STATION, TFT_ORANGE);
  221. }
  222. // Beacon sniff
  223. else if (cmd_args.get(0) == SNIFF_BEACON_CMD) {
  224. Serial.println("Starting Beacon sniff. Stop with " + (String)STOPSCAN_CMD);
  225. #ifdef HAS_SCREEN
  226. display_obj.clearScreen();
  227. menu_function_obj.drawStatusBar();
  228. #endif
  229. wifi_scan_obj.StartScan(WIFI_SCAN_AP, TFT_MAGENTA);
  230. }
  231. // Probe sniff
  232. else if (cmd_args.get(0) == SNIFF_PROBE_CMD) {
  233. Serial.println("Starting Probe sniff. Stop with " + (String)STOPSCAN_CMD);
  234. #ifdef HAS_SCREEN
  235. display_obj.clearScreen();
  236. menu_function_obj.drawStatusBar();
  237. #endif
  238. wifi_scan_obj.StartScan(WIFI_SCAN_PROBE, TFT_MAGENTA);
  239. }
  240. // Deauth sniff
  241. else if (cmd_args.get(0) == SNIFF_DEAUTH_CMD) {
  242. Serial.println("Starting Deauth sniff. Stop with " + (String)STOPSCAN_CMD);
  243. #ifdef HAS_SCREEN
  244. display_obj.clearScreen();
  245. menu_function_obj.drawStatusBar();
  246. #endif
  247. wifi_scan_obj.StartScan(WIFI_SCAN_DEAUTH, TFT_RED);
  248. }
  249. // Pwn sniff
  250. else if (cmd_args.get(0) == SNIFF_PWN_CMD) {
  251. Serial.println("Starting Pwnagotchi sniff. Stop with " + (String)STOPSCAN_CMD);
  252. #ifdef HAS_SCREEN
  253. display_obj.clearScreen();
  254. menu_function_obj.drawStatusBar();
  255. #endif
  256. wifi_scan_obj.StartScan(WIFI_SCAN_PWN, TFT_MAGENTA);
  257. }
  258. // Espressif sniff
  259. else if (cmd_args.get(0) == SNIFF_ESP_CMD) {
  260. Serial.println("Starting Espressif device sniff. Stop with " + (String)STOPSCAN_CMD);
  261. #ifdef HAS_SCREEN
  262. display_obj.clearScreen();
  263. menu_function_obj.drawStatusBar();
  264. #endif
  265. wifi_scan_obj.StartScan(WIFI_SCAN_ESPRESSIF, TFT_MAGENTA);
  266. }
  267. // PMKID sniff
  268. else if (cmd_args.get(0) == SNIFF_PMKID_CMD) {
  269. int ch_sw = this->argSearch(&cmd_args, "-c");
  270. int d_sw = this->argSearch(&cmd_args, "-d"); // Deauth for pmkid
  271. if (ch_sw != -1) {
  272. wifi_scan_obj.set_channel = cmd_args.get(ch_sw + 1).toInt();
  273. wifi_scan_obj.changeChannel();
  274. Serial.println("Set channel: " + (String)wifi_scan_obj.set_channel);
  275. }
  276. if (d_sw == -1) {
  277. Serial.println("Starting PMKID sniff on channel " + (String)wifi_scan_obj.set_channel + ". Stop with " + (String)STOPSCAN_CMD);
  278. wifi_scan_obj.StartScan(WIFI_SCAN_EAPOL, TFT_VIOLET);
  279. }
  280. else {
  281. Serial.println("Starting PMKID sniff with deauthentication on channel " + (String)wifi_scan_obj.set_channel + ". Stop with " + (String)STOPSCAN_CMD);
  282. wifi_scan_obj.StartScan(WIFI_SCAN_ACTIVE_EAPOL, TFT_VIOLET);
  283. }
  284. }
  285. //// WiFi attack commands
  286. // attack
  287. if (cmd_args.get(0) == ATTACK_CMD) {
  288. int attack_type_switch = this->argSearch(&cmd_args, "-t"); // Required
  289. int list_beacon_sw = this->argSearch(&cmd_args, "-l");
  290. int rand_beacon_sw = this->argSearch(&cmd_args, "-r");
  291. int ap_beacon_sw = this->argSearch(&cmd_args, "-a");
  292. int src_addr_sw = this->argSearch(&cmd_args, "-s");
  293. int dst_addr_sw = this->argSearch(&cmd_args, "-d");
  294. if (attack_type_switch == -1) {
  295. Serial.println("You must specify an attack type");
  296. return;
  297. }
  298. else {
  299. String attack_type = cmd_args.get(attack_type_switch + 1);
  300. // Branch on attack type
  301. // Deauth
  302. if (attack_type == ATTACK_TYPE_DEAUTH) {
  303. if (dst_addr_sw == -1) {
  304. Serial.println("Sending to broadcast...");
  305. wifi_scan_obj.dst_mac = "ff:ff:ff:ff:ff:ff";
  306. }
  307. else {
  308. wifi_scan_obj.dst_mac = cmd_args.get(dst_addr_sw + 1);
  309. Serial.println("Sending to " + wifi_scan_obj.dst_mac + "...");
  310. }
  311. if (src_addr_sw == -1) {
  312. if (!this->apSelected()) {
  313. Serial.println("You don't have any targets selected. Use " + (String)SEL_CMD);
  314. return;
  315. }
  316. #ifdef HAS_SCREEN
  317. display_obj.clearScreen();
  318. menu_function_obj.drawStatusBar();
  319. #endif
  320. Serial.println("Starting Deauthentication attack. Stop with " + (String)STOPSCAN_CMD);
  321. wifi_scan_obj.StartScan(WIFI_ATTACK_DEAUTH, TFT_RED);
  322. }
  323. else {
  324. String src_mac_str = cmd_args.get(src_addr_sw + 1);
  325. sscanf(src_mac_str.c_str(), "%2hhx:%2hhx:%2hhx:%2hhx:%2hhx:%2hhx",
  326. &wifi_scan_obj.src_mac[0], &wifi_scan_obj.src_mac[1], &wifi_scan_obj.src_mac[2], &wifi_scan_obj.src_mac[3], &wifi_scan_obj.src_mac[4], &wifi_scan_obj.src_mac[5]);
  327. #ifdef HAS_SCREEN
  328. display_obj.clearScreen();
  329. menu_function_obj.drawStatusBar();
  330. #endif
  331. Serial.println("Starting Manual Deauthentication attack. Stop with " + (String)STOPSCAN_CMD);
  332. wifi_scan_obj.StartScan(WIFI_ATTACK_DEAUTH_MANUAL, TFT_RED);
  333. }
  334. }
  335. // Beacon
  336. else if (attack_type == ATTACK_TYPE_BEACON) {
  337. // spam by list
  338. if (list_beacon_sw != -1) {
  339. if (!this->hasSSIDs()) {
  340. Serial.println("You don't have any SSIDs in your list. Use " + (String)SSID_CMD);
  341. return;
  342. }
  343. #ifdef HAS_SCREEN
  344. display_obj.clearScreen();
  345. menu_function_obj.drawStatusBar();
  346. #endif
  347. Serial.println("Starting Beacon list spam. Stop with " + (String)STOPSCAN_CMD);
  348. wifi_scan_obj.StartScan(WIFI_ATTACK_BEACON_LIST, TFT_RED);
  349. }
  350. // spam with random
  351. else if (rand_beacon_sw != -1) {
  352. #ifdef HAS_SCREEN
  353. display_obj.clearScreen();
  354. menu_function_obj.drawStatusBar();
  355. #endif
  356. Serial.println("Starting random Beacon spam. Stop with " + (String)STOPSCAN_CMD);
  357. wifi_scan_obj.StartScan(WIFI_ATTACK_BEACON_SPAM, TFT_ORANGE);
  358. }
  359. // Spam from AP list
  360. else if (ap_beacon_sw != -1) {
  361. if (!this->apSelected()) {
  362. Serial.println("You don't have any targets selected. Use " + (String)SEL_CMD);
  363. return;
  364. }
  365. #ifdef HAS_SCREEN
  366. display_obj.clearScreen();
  367. menu_function_obj.drawStatusBar();
  368. #endif
  369. Serial.println("Starting Targeted AP Beacon spam. Stop with " + (String)STOPSCAN_CMD);
  370. wifi_scan_obj.StartScan(WIFI_ATTACK_AP_SPAM, TFT_MAGENTA);
  371. }
  372. else {
  373. Serial.println("You did not specify a beacon attack type");
  374. }
  375. }
  376. else if (attack_type == ATTACK_TYPE_PROBE) {
  377. if (!this->apSelected()) {
  378. Serial.println("You don't have any targets selected. Use " + (String)SEL_CMD);
  379. return;
  380. }
  381. Serial.println("Starting Probe spam. Stop with " + (String)STOPSCAN_CMD);
  382. #ifdef HAS_SCREEN
  383. display_obj.clearScreen();
  384. menu_function_obj.drawStatusBar();
  385. #endif
  386. wifi_scan_obj.StartScan(WIFI_ATTACK_AUTH, TFT_RED);
  387. }
  388. else if (attack_type == ATTACK_TYPE_RR) {
  389. Serial.println("Starting Rick Roll Beacon spam. Stop with " + (String)STOPSCAN_CMD);
  390. #ifdef HAS_SCREEN
  391. display_obj.clearScreen();
  392. menu_function_obj.drawStatusBar();
  393. #endif
  394. wifi_scan_obj.StartScan(WIFI_ATTACK_RICK_ROLL, TFT_YELLOW);
  395. }
  396. else {
  397. Serial.println("Attack type not properly defined");
  398. return;
  399. }
  400. }
  401. }
  402. //// Bluetooth scan/attack commands
  403. // Bluetooth scan
  404. if (cmd_args.get(0) == BT_SNIFF_CMD) {
  405. Serial.println("Starting Bluetooth scan. Stop with " + (String)STOPSCAN_CMD);
  406. #ifdef HAS_SCREEN
  407. display_obj.clearScreen();
  408. menu_function_obj.drawStatusBar();
  409. #endif
  410. wifi_scan_obj.StartScan(BT_SCAN_ALL, TFT_GREEN);
  411. }
  412. // Bluetooth CC Skimmer scan
  413. else if (cmd_args.get(0) == BT_SKIM_CMD) {
  414. Serial.println("Starting Bluetooth CC Skimmer scan. Stop with " + (String)STOPSCAN_CMD);
  415. #ifdef HAS_SCREEN
  416. display_obj.clearScreen();
  417. menu_function_obj.drawStatusBar();
  418. #endif
  419. wifi_scan_obj.StartScan(BT_SCAN_SKIMMERS, TFT_MAGENTA);
  420. }
  421. // Update command
  422. if (cmd_args.get(0) == UPDATE_CMD) {
  423. int w_sw = this->argSearch(&cmd_args, "-w"); // Web update
  424. int sd_sw = this->argSearch(&cmd_args, "-s"); // SD Update
  425. // Update via OTA
  426. if (w_sw != -1) {
  427. Serial.println("Starting Marauder OTA Update. Stop with " + (String)STOPSCAN_CMD);
  428. wifi_scan_obj.currentScanMode = OTA_UPDATE;
  429. //#ifdef HAS_SCREEN
  430. // menu_function_obj.changeMenu(menu_function_obj.updateMenu);
  431. //#endif
  432. web_obj.setupOTAupdate();
  433. }
  434. // Update via SD
  435. else if (sd_sw != -1) {
  436. if (!sd_obj.supported) {
  437. Serial.println("SD card is not connected. Cannot perform SD Update");
  438. return;
  439. }
  440. wifi_scan_obj.currentScanMode = OTA_UPDATE;
  441. sd_obj.runUpdate();
  442. }
  443. }
  444. }
  445. //// WiFi aux commands
  446. // List access points
  447. if (cmd_args.get(0) == LIST_AP_CMD) {
  448. int ap_sw = this->argSearch(&cmd_args, "-a");
  449. int ss_sw = this->argSearch(&cmd_args, "-s");
  450. int cl_sw = this->argSearch(&cmd_args, "-c");
  451. // List APs
  452. if (ap_sw != -1) {
  453. for (int i = 0; i < access_points->size(); i++) {
  454. if (access_points->get(i).selected)
  455. Serial.println("[" + (String)i + "] " + access_points->get(i).essid + " " + (String)access_points->get(i).rssi + " (selected)");
  456. else
  457. Serial.println("[" + (String)i + "] " + access_points->get(i).essid + " " + (String)access_points->get(i).rssi);
  458. }
  459. }
  460. // List SSIDs
  461. else if (ss_sw != -1) {
  462. for (int i = 0; i < ssids->size(); i++) {
  463. if (ssids->get(i).selected)
  464. Serial.println("[" + (String)i + "] " + ssids->get(i).essid + " (selected)");
  465. else
  466. Serial.println("[" + (String)i + "] " + ssids->get(i).essid);
  467. }
  468. }
  469. else if (cl_sw != -1) {
  470. char sta_mac[] = "00:00:00:00:00:00";
  471. for (int x = 0; x < access_points->size(); x++) {
  472. Serial.println("[" + (String)x + "] " + access_points->get(x).essid + " " + (String)access_points->get(x).rssi + ":");
  473. for (int i = 0; i < access_points->get(x).stations->size(); i++) {
  474. wifi_scan_obj.getMAC(sta_mac, stations->get(access_points->get(x).stations->get(i)).mac, 0);
  475. if (stations->get(i).selected) {
  476. Serial.print(" [" + (String)i + "] ");
  477. Serial.print(sta_mac);
  478. Serial.println(" (selected)");
  479. }
  480. else {
  481. Serial.print(" [" + (String)i + "] ");
  482. Serial.println(sta_mac);
  483. }
  484. }
  485. }
  486. }
  487. else {
  488. Serial.println("You did not specify which list to show");
  489. return;
  490. }
  491. }
  492. // Select access points or stations
  493. else if (cmd_args.get(0) == SEL_CMD) {
  494. // Get switches
  495. int ap_sw = this->argSearch(&cmd_args, "-a");
  496. int ss_sw = this->argSearch(&cmd_args, "-s");
  497. // select Access points
  498. if (ap_sw != -1) {
  499. // Get list of indices
  500. LinkedList<String> ap_index = this->parseCommand(cmd_args.get(ap_sw + 1), ",");
  501. // Select ALL APs
  502. if (cmd_args.get(ap_sw + 1) == "all") {
  503. for (int i = 0; i < access_points->size(); i++) {
  504. if (access_points->get(i).selected) {
  505. // Unselect "selected" ap
  506. AccessPoint new_ap = access_points->get(i);
  507. new_ap.selected = false;
  508. access_points->set(i, new_ap);
  509. }
  510. else {
  511. // Select "unselected" ap
  512. AccessPoint new_ap = access_points->get(i);
  513. new_ap.selected = true;
  514. access_points->set(i, new_ap);
  515. }
  516. }
  517. }
  518. // Select specific APs
  519. else {
  520. // Mark APs as selected
  521. for (int i = 0; i < ap_index.size(); i++) {
  522. int index = ap_index.get(i).toInt();
  523. if (!this->inRange(access_points->size(), index)) {
  524. Serial.println("Index not in range: " + (String)index);
  525. continue;
  526. }
  527. if (access_points->get(index).selected) {
  528. // Unselect "selected" ap
  529. AccessPoint new_ap = access_points->get(index);
  530. new_ap.selected = false;
  531. access_points->set(index, new_ap);
  532. }
  533. else {
  534. // Select "unselected" ap
  535. AccessPoint new_ap = access_points->get(index);
  536. new_ap.selected = true;
  537. access_points->set(index, new_ap);
  538. }
  539. }
  540. }
  541. }
  542. // select ssids
  543. else if (ss_sw != -1) {
  544. // Get list of indices
  545. LinkedList<String> ss_index = this->parseCommand(cmd_args.get(ss_sw + 1), ",");
  546. // Mark APs as selected
  547. for (int i = 0; i < ss_index.size(); i++) {
  548. int index = ss_index.get(i).toInt();
  549. if (!this->inRange(ssids->size(), index)) {
  550. Serial.println("Index not in range: " + (String)index);
  551. continue;
  552. }
  553. if (ssids->get(index).selected) {
  554. // Unselect "selected" ap
  555. ssid new_ssid = ssids->get(index);
  556. new_ssid.selected = false;
  557. ssids->set(index, new_ssid);
  558. }
  559. else {
  560. // Select "unselected" ap
  561. ssid new_ssid = ssids->get(index);
  562. new_ssid.selected = true;
  563. ssids->set(index, new_ssid);
  564. }
  565. }
  566. }
  567. else {
  568. Serial.println("You did not specify which list to select from");
  569. return;
  570. }
  571. }
  572. // SSID stuff
  573. else if (cmd_args.get(0) == SSID_CMD) {
  574. int add_sw = this->argSearch(&cmd_args, "-a");
  575. int gen_sw = this->argSearch(&cmd_args, "-g");
  576. int spc_sw = this->argSearch(&cmd_args, "-n");
  577. int rem_sw = this->argSearch(&cmd_args, "-r");
  578. // Add ssid
  579. if (add_sw != -1) {
  580. // Generate random
  581. if (gen_sw != -1) {
  582. int gen_count = cmd_args.get(gen_sw + 1).toInt();
  583. wifi_scan_obj.generateSSIDs(gen_count);
  584. }
  585. // Add specific
  586. else if (spc_sw != -1) {
  587. String essid = cmd_args.get(spc_sw + 1);
  588. wifi_scan_obj.addSSID(essid);
  589. }
  590. else {
  591. Serial.println("You did not specify how to add SSIDs");
  592. }
  593. }
  594. // Remove SSID
  595. else if (rem_sw != -1) {
  596. int index = cmd_args.get(rem_sw + 1).toInt();
  597. if (!this->inRange(ssids->size(), index)) {
  598. Serial.println("Index not in range: " + (String)index);
  599. return;
  600. }
  601. ssids->remove(index);
  602. }
  603. else {
  604. Serial.println("You did not specify whether to add or remove SSIDs");
  605. return;
  606. }
  607. }
  608. }