mirrors
/
esp8266_deauther
mirror of https://github.com/SpacehuhnTech/esp8266_deauther.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity
esp8266_deauther/esp8266_deauther/SerialInterface.cpp

1091 lines
33 KiB
C++
Raw Blame History

#include "SerialInterface.h"
/*
A short introduction for the reader:
This class is a huuuuuuuge mess of if statements!!!!!1!eleven
But it works and I tried to make it as compact as possible.
If someone was able to make an Arduino framework for parsing an input string in different commands with certain rules,
required- and optional parameters of different datatypes to run different callback functions... shoot me a quick message on GitHub or Twitter: @spacehuhn!
PS: the framework has to be good in both CPU performance and RAM usage, otherwise it's useless. Good luck!
I already wasted way too much time trying to come up with the best way of programming this, so I will just keep it that way for now.
Also: NEVER CHANGE A RUNNING SYSTEM!
*/
SerialInterface::SerialInterface() {
list = new LinkedList<String>;
}
void SerialInterface::load() {
checkFile(execPath, String());
executing = true;
}
void SerialInterface::load(String filepath) {
execPath = filepath;
load();
}
void SerialInterface::enable() {
enabled = true;
prntln(CLI_SERIAL_ENABLED);
}
void SerialInterface::disable() {
enabled = true;
prntln(CLI_SERIAL_DISABLED);
}
void SerialInterface::error(String message) {
prnt(CLI_ERROR);
prntln(message);
}
void SerialInterface::parameterError(String parameter) {
prnt(CLI_ERROR_PARAMETER);
prnt(parameter);
prntln(DOUBLEQUOTES);
}
bool SerialInterface::isInt(String str) {
if (eqls(str,STR_TRUE) || eqls(str,STR_FALSE))
return true;
for (int i = 0; i < str.length(); i++)
if (!isDigit(str.charAt(i))) return false;
return true;
}
int SerialInterface::toInt(String str) {
if (eqls(str,STR_TRUE))
return 1;
else if (eqls(str,STR_FALSE))
return 0;
else
return str.toInt();
}
uint32_t SerialInterface::getTime(String time) {
int value = time.toInt();
if (value < 0) value = -value;
if (time.substring(time.length() - 1).equalsIgnoreCase(String(S)))
value *= 1000;
else if (time.substring(time.length() - 3).equalsIgnoreCase(str(STR_MIN)) || time.charAt(time.length() - 1) == M)
value *= 60000;
return value;
}
bool SerialInterface::eqlsCMD(int i, const char* keyword) {
return eqls(list->get(i).c_str(), keyword);
}
void SerialInterface::stopScript() {
if (continuously) {
continuously = false;
prnt(CLI_STOPPED_SCRIPT);
prnt(execPath);
prntln(DOUBLEQUOTES);
prnt(SPACE);
prnt(CLI_CONTINUOUSLY);
}
}
void SerialInterface::update() {
if (executing) {
if (execPath.charAt(0) != SLASH) execPath = SLASH + execPath;
prnt(CLI_EXECUTING);
prntln(execPath);
File f = SPIFFS.open(execPath, "r");
if (f.size() > 0) {
String line;
char tmp;
while (f.available()) {
tmp = f.read();
if (tmp == NEWLINE) {
runCommands(line);
line = String();
} else {
line += tmp;
}
}
runCommand(line);
}
f.close();
executing = false;
if(continuously) prntln(CLI_SCRIPT_DONE_CONTINUE);
else prntln(CLI_SCRIPT_DONE);
loopTime = currentTime;
} else {
if (enabled && Serial.available() > 0)
runCommands(Serial.readStringUntil(NEWLINE));
if (continuously) {
if (currentTime - loopTime > continueTime)
executing = true;
}
}
}
void SerialInterface::runCommands(String input) {
String tmp;
for (int i = 0; i < input.length(); i++) {
if (input.charAt(i) == SEMICOLON && input.charAt(i + 1) == SEMICOLON && input.charAt(i - 1) != BACKSLASH) {
runCommand(tmp);
tmp = String();
i++;
} else {
tmp += input.charAt(i);
}
}
tmp.replace(BACKSLASH + SEMICOLON + SEMICOLON, SEMICOLON + SEMICOLON);
if (tmp.length() > 0) runCommand(tmp);
}
void SerialInterface::runCommand(String input) {
input.replace(String(NEWLINE), String());
input.replace(String(CARRIAGERETURN), String());
list->clear();
// parse/split input in list
String tmp;
bool withinQuotes = false;
for (int i = 0; i < input.length() && i < 512; i++) {
if ((input.charAt(i) == SPACE && input.charAt(i - 1) != BACKSLASH && !withinQuotes) || input.charAt(i) == CARRIAGERETURN || input.charAt(i) == NEWLINE) {
//tmp.toLowerCase();
if (tmp.length() > 0) list->add(tmp);
tmp = String();
} else if(input.charAt(i) == DOUBLEQUOTES && (input.charAt(i-1) != BACKSLASH || (input.charAt(i-1) == BACKSLASH && input.charAt(i-2) == BACKSLASH))){
withinQuotes = !withinQuotes;
if(tmp.length() == 0 && !withinQuotes) tmp += SPACE;
} else if (input.charAt(i) != BACKSLASH || (input.charAt(i) == BACKSLASH && input.charAt(i - 1) == BACKSLASH)) {
tmp += input.charAt(i);
}
}
// add whatever is left from the loop above
if (tmp.length() > 0) list->add(tmp);
// stop when input is empty/invalid
if (list->size() == 0) return;
// print comments
if (list->get(0) == str(CLI_COMMENT)) {
prntln(input);
return;
}
// print command
prnt(CLI_INPUT_PREFIX);
prntln(input);
if (list->size() == 0) return;
// ===== HELP ===== //
if (eqlsCMD(0, CLI_HELP)) {
prntln(CLI_HELP_HEADER);
prntln(CLI_HELP_HELP);
prntln(CLI_HELP_SCAN);
prntln(CLI_HELP_SHOW);
prntln(CLI_HELP_SELECT);
prntln(CLI_HELP_DESELECT);
prntln(CLI_HELP_SSID_A);
prntln(CLI_HELP_SSID_B);
prntln(CLI_HELP_SSID_C);
prntln(CLI_HELP_NAME_A);
prntln(CLI_HELP_NAME_B);
prntln(CLI_HELP_NAME_C);
prntln(CLI_HELP_SET_NAME);
prntln(CLI_HELP_ENABLE_RANDOM);
prntln(CLI_HELP_DISABLE_RANDOM);
prntln(CLI_HELP_LOAD);
prntln(CLI_HELP_SAVE);
prntln(CLI_HELP_REMOVE_A);
prntln(CLI_HELP_REMOVE_B);
prntln(CLI_HELP_ATTACK);
prntln(CLI_HELP_ATTACK_STATUS);
prntln(CLI_HELP_STOP);
prntln(CLI_HELP_SYSINFO);
prntln(CLI_HELP_CLEAR);
prntln(CLI_HELP_FORMAT);
prntln(CLI_HELP_PRINT);
prntln(CLI_HELP_DELETE);
prntln(CLI_HELP_REPLACE);
prntln(CLI_HELP_COPY);
prntln(CLI_HELP_RENAME);
prntln(CLI_HELP_RUN);
prntln(CLI_HELP_WRITE);
prntln(CLI_HELP_GET);
prntln(CLI_HELP_SET);
prntln(CLI_HELP_RESET);
prntln(CLI_HELP_CHICKEN);
prntln(CLI_HELP_REBOOT);
prntln(CLI_HELP_INFO);
prntln(CLI_HELP_COMMENT);
prntln(CLI_HELP_SEND_DEAUTH);
prntln(CLI_HELP_SEND_BEACON);
prntln(CLI_HELP_SEND_PROBE);
prntln(CLI_HELP_LED_A);
prntln(CLI_HELP_LED_B);
prntln(CLI_HELP_LED_ENABLE);
prntln(CLI_HELP_DRAW);
prntln(CLI_HELP_SCREEN_ON);
prntln(CLI_HELP_SCREEN_MODE);
prntln(CLI_HELP_FOOTER);
}
// ===== SCAN ===== //
// scan [<mode>] [-t <time>] [-c <continue-time>] [-ch <channel>]
else if (eqlsCMD(0, CLI_SCAN)) {
uint8_t scanMode = SCAN_MODE_ALL;
uint8_t nextmode = SCAN_MODE_OFF;
uint8_t channel = wifi_channel;
bool channelHop = true;
uint32_t time = 15000;
uint32_t continueTime = 10000;
for (int i = 1; i < list->size(); i++) {
if (eqlsCMD(i, CLI_AP)) scanMode = SCAN_MODE_APS;
else if (eqlsCMD(i, CLI_STATION)) scanMode = SCAN_MODE_STATIONS;
else if (eqlsCMD(i, CLI_ALL)) scanMode = SCAN_MODE_ALL;
else if (eqlsCMD(i, CLI_WIFI)) scanMode = SCAN_MODE_SNIFFER;
else if (eqlsCMD(i, CLI_TIME)) {
i++;
time = getTime(list->get(i));
} else if (eqlsCMD(i, CLI_CONTINUE)) {
i++;
nextmode = scanMode;
continueTime = getTime(list->get(i));
} else if (eqlsCMD(i, CLI_CHANNEL)) {
i++;
if(!eqlsCMD(i, CLI_ALL)){
channelHop = false;
channel = list->get(i).toInt();
}
} else {
parameterError(list->get(i));
}
}
scan.start(scanMode, time, nextmode, continueTime, channelHop, channel);
}
// ===== SHOW ===== //
else if (eqlsCMD(0, CLI_SHOW)) {
// show selected [<all/aps/stations/names/ssids>]
if (eqlsCMD(1, CLI_SELECT)) {
if (list->size() > 2) {
for (int i = 2; i < list->size(); i++) {
if (eqlsCMD(i, CLI_AP)) accesspoints.printSelected();
else if (eqlsCMD(i, CLI_STATION)) stations.printSelected();
else if (eqlsCMD(i, CLI_NAME)) names.printSelected();
else if (eqlsCMD(i, CLI_ALL)) scan.printSelected();
else parameterError(list->get(i));
}
} else {
scan.printSelected();
}
}
// show [<all/aps/stations/names/ssids>]
else {
if (list->size() > 1) {
for (int i = 1; i < list->size(); i++) {
if (eqlsCMD(i, CLI_AP)) accesspoints.printAll();
else if (eqlsCMD(i, CLI_STATION)) stations.printAll();
else if (eqlsCMD(i, CLI_NAME)) names.printAll();
else if (eqlsCMD(i, CLI_SSID)) ssids.printAll();
else if (eqlsCMD(i, CLI_ALL)) scan.printAll();
else parameterError(list->get(i));
}
} else {
scan.printAll();
}
}
}
// ===== (DE)SELECT ===== //
// select [<type>] [<id>]
// deselect [<type>] [<id>]
else if (eqlsCMD(0, CLI_SELECT) || eqlsCMD(0, CLI_DESELECT)) {
bool select = eqlsCMD(0, CLI_SELECT);
int mode; // aps = 0, stations = 1, names = 2
int id; // -1 = all, -2 name string
if (list->size() == 1 || eqlsCMD(1, CLI_ALL)) {
select ? scan.selectAll() : scan.deselectAll();
return;
}
if (list->size() == 2 || eqlsCMD(2, CLI_ALL)) id = -1;
else if (!isInt(list->get(2))) id = -2;
else id = list->get(2).toInt();
if (eqlsCMD(1, CLI_AP)) mode = 0;
else if (eqlsCMD(1, CLI_STATION)) mode = 1;
else if (eqlsCMD(1, CLI_NAME)) mode = 2;
else parameterError(list->get(1));
if (id >= 0) {
if (mode == 0) select ? accesspoints.select(id) : accesspoints.deselect(id);
else if (mode == 1) select ? stations.select(id) : stations.deselect(id);
else if (mode == 2) select ? names.select(id) : names.deselect(id);
} else if (id == -1) {
if (mode == 0) select ? accesspoints.selectAll() : accesspoints.deselectAll();
else if (mode == 1) select ? stations.selectAll() : stations.deselectAll();
else if (mode == 2) select ? names.selectAll() : names.deselectAll();
} else if (id == -2 && mode == 2) {
select ? names.select(list->get(2)) : names.deselect(list->get(2));
} else {
parameterError(list->get(1) + SPACE + list->get(2));
}
}
// ===== ADD ===== //
else if (list->size() >= 3 && eqlsCMD(0, CLI_ADD) && eqlsCMD(1, CLI_SSID)) {
// add ssid -s [-f]
if(eqlsCMD(2, CLI_SELECT)){
bool force = eqlsCMD(3, CLI_FORCE);
ssids.cloneSelected(force);
}
// add ssid <ssid> [-wpa2] [-cl <clones>] [-f]
// add ssid -ap <id> [-cl <clones>] [-f]
else{
String ssid = list->get(2);
bool wpa2 = false;
bool force = false;
int clones = 1;
int i = 3;
if (eqlsCMD(2, CLI_AP)) {
ssid = accesspoints.getSSID(list->get(3).toInt());
wpa2 = accesspoints.getEncStr(list->get(3).toInt()) != " - ";
i = 4;
}
while (i < list->size()) {
if (eqlsCMD(i, CLI_WPA2)) wpa2 = true;
else if (eqlsCMD(i, CLI_FORCE)) force = true;
else if (eqlsCMD(i, CLI_CLONES)) {
clones = list->get(i + 1).toInt();
i++;
} else parameterError(list->get(i));
i++;
}
ssids.add(ssid, wpa2, clones, force);
}
}
// add name <name> [-ap <id>] [-s] [-f]
// add name <name> [-st <id>] [-s] [-f]
// add name <name> [-m <mac>] [-ch <channel>] [-b <bssid>] [-s] [-f]
else if (list->size() >= 3 && eqlsCMD(0, CLI_ADD) && eqlsCMD(1, CLI_NAME)) {
String name = list->get(2);
String mac;
uint8_t channel = wifi_channel;
String bssid;
bool selected = false;
bool force = false;
for (int i = 3; i < list->size(); i++) {
if (eqlsCMD(i, CLI_MAC)) mac = list->get(i + 1);
else if (eqlsCMD(i, CLI_AP)) mac = accesspoints.getMacStr(list->get(i + 1).toInt());
else if (eqlsCMD(i, CLI_STATION)) {
mac = stations.getMacStr(list->get(i + 1).toInt());
bssid = accesspoints.getMacStr(stations.getAP(list->get(i + 1).toInt()));
}
else if (eqlsCMD(i, CLI_CHANNEL)) channel = (uint8_t)list->get(i + 1).toInt();
else if (eqlsCMD(i, CLI_BSSID)) bssid = list->get(i + 1);
else if (eqlsCMD(i, CLI_SELECT)){
selected = true;
i--;
} else if (eqlsCMD(i, CLI_FORCE)){
force = true;
i--;
} else {
parameterError(list->get(i));
i--;
}
i++;
}
if (name.length() == 0) prntln(CLI_ERROR_NAME_LEN);
else if (mac.length() == 0) prntln(CLI_ERROR_MAC_LEN);
else names.add(mac, name, bssid, channel, selected, force);
}
// ===== SET NAME ==== //
// set name <id> <newname>
else if (list->size() == 4 && eqlsCMD(0, CLI_SET) && eqlsCMD(1, CLI_NAME)) {
names.setName(list->get(2).toInt(), list->get(3));
}
// ===== REPLACE ===== //
// replace name <id> [-n <name>} [-m <mac>] [-ch <channel>] [-b <bssid>] [-s]
else if (list->size() >= 4 && eqlsCMD(0, CLI_REPLACE) && eqlsCMD(1, CLI_NAME)) {
int id = list->get(2).toInt();
String name = names.getName(id);
String mac = names.getMacStr(id);
uint8_t channel = names.getCh(id);
String bssid = names.getBssidStr(id);
bool selected = names.getSelected(id);
for (int i = 3; i < list->size(); i++) {
if (eqlsCMD(i, CLI_NAME)) name = list->get(i + 1);
else if (eqlsCMD(i, CLI_MAC)) mac = list->get(i + 1);
else if (eqlsCMD(i, CLI_CHANNEL)) channel = (uint8_t)list->get(i + 1).toInt();
else if (eqlsCMD(i, CLI_BSSID)) bssid = list->get(i + 1);
else if (eqlsCMD(i, CLI_SELECT)){
selected = true;
i--;
} else {
parameterError(list->get(i));
i--;
}
i++;
}
names.replace(id,mac, name, bssid, channel, selected);
}
// replace ssid <id> [-n <name>} [-wpa2]
else if (list->size() >= 3 && eqlsCMD(0, CLI_REPLACE) && eqlsCMD(1, CLI_SSID)) {
int id = list->get(2).toInt();
String name = ssids.getName(id);
bool wpa2 = false;
for (int i = 3; i < list->size(); i++) {
if (eqlsCMD(i, CLI_NAME)){
name = list->get(i + 1);
i++;
}else if(eqlsCMD(i, CLI_WPA2)){
wpa2 = true;
}
}
ssids.replace(id,name, wpa2);
}
// ===== REMOVE ===== //
// remove <type> [-a]
// remove <type> <id>
else if (list->size() >= 2 && eqlsCMD(0, CLI_REMOVE)) {
if (list->size() == 2 || (eqlsCMD(2, CLI_ALL))) {
if (eqlsCMD(1, CLI_SSID)) ssids.removeAll();
else if (eqlsCMD(1, CLI_NAME)) names.removeAll();
else if (eqlsCMD(1, CLI_AP)) accesspoints.removeAll();
else if (eqlsCMD(1, CLI_STATION)) stations.removeAll();
else parameterError(list->get(1));
} else {
if (eqlsCMD(1, CLI_SSID)) ssids.remove(list->get(2).toInt());
else if (eqlsCMD(1, CLI_NAME)) names.remove(list->get(2).toInt());
else if (eqlsCMD(1, CLI_AP)) accesspoints.remove(list->get(2).toInt());
else if (eqlsCMD(1, CLI_STATION)) stations.remove(list->get(2).toInt());
else parameterError(list->get(1));
}
}
// ===== RANDOM ===== //
// enable random <interval>
else if (eqlsCMD(0, CLI_ENABLE) && eqlsCMD(1, CLI_RANDOM) && list->size() == 3) {
ssids.enableRandom(getTime(list->get(2)));
}
// disable random
else if (eqlsCMD(0, CLI_DISABLE) && eqlsCMD(1, CLI_RANDOM)) {
ssids.disableRandom();
}
// ====== RICE ===== //
// => have you tried putting it into a bowl rice?
// funny command to mess with people, please don't share the info thanks <3
else if (eqlsCMD(0, CLI_RICE)) {
prntln(CLI_RICE_START);
uint32_t i = 0;
uint8_t multi = 1;
uint8_t end = random(80, 99);
if (list->get(1).toInt() > 1) multi = list->get(1).toInt();
while (true) {
if (i % 10 == 0 && i > 0) {
char s[100];
sprintf(s,str(CLI_RICE_OUTPUT).c_str(), i / 10);
prnt(String(s));
if ((i / 10) == end) {
prnt(CLI_RICE_ERROR);
prnt(String(random(16, 255), HEX));
prnt(String(random(16, 255), HEX));
prnt(String(random(16, 255), HEX));
prntln(String(random(16, 255), HEX));
for (int i = 1; i <= 32; i++) {
for (int i = 0; i < 2; i++) {
for (int i = 1; i <= 8; i++) {
prnt(String(random(16, 255), HEX));
prnt(SPACE);
}
prnt(SPACE);
}
prntln();
}
ESP.reset();
} else if ((i / 10) % 10 == 0) {
prnt(CLI_RICE_MEM);
prnt(String(random(16, 255), HEX));
prnt(String(random(16, 255), HEX));
prnt(String(random(16, 255), HEX));
prntln(String(random(16, 255), HEX));
}
}
prnt(POINT);
delay(100 * multi * multi);
i++;
}
}
// ===== LOAD/SAVE ===== //
// save [<type>] [<file>]
// load [<type>] [<file>]
else if ((eqlsCMD(0, CLI_LOAD) || eqlsCMD(0, CLI_SAVE)) && list->size() >= 1 && list->size() <= 3) {
bool load = eqlsCMD(0, CLI_LOAD);
if (list->size() == 1 || eqlsCMD(1, CLI_ALL)) {
load ? ssids.load() : ssids.save(false);
load ? names.load() : names.save(false);
load ? settings.load() : settings.save(false);
if(!load) scan.save(false);
return;
}
if (list->size() == 3) { // Todo: check if -f or filename
if (eqlsCMD(1, CLI_SSID)) load ? ssids.load(list->get(2)) : ssids.save(true, list->get(2));
else if (eqlsCMD(1, CLI_NAME)) load ? names.load(list->get(2)) : names.save(true, list->get(2));
else if (eqlsCMD(1, CLI_SETTING)) load ? settings.load(list->get(2)) : settings.save(true, list->get(2));
else parameterError(list->get(1));
} else {
if (eqlsCMD(1, CLI_SSID)) load ? ssids.load() : ssids.save(false);
else if (eqlsCMD(1, CLI_NAME)) load ? names.load() : names.save(false);
else if (eqlsCMD(1, CLI_SETTING)) load ? settings.load() : settings.save(false);
else if ((eqlsCMD(1, CLI_SCAN) || eqlsCMD(1, CLI_AP) || eqlsCMD(1, CLI_STATION)) && !load) scan.save(false);
else parameterError(list->get(1));
}
}
// ===== ATTACK ===== //
// attack [-b] [-d] [-da] [p] [-t <timeout>]
// attack status [<on/off>]
else if (eqlsCMD(0, CLI_ATTACK)) {
if (eqlsCMD(1, CLI_STATUS)) {
if (list->size() == 2) {
attack.status();
} else {
if (eqlsCMD(2, CLI_ON)) attack.enableOutput();
else if (eqlsCMD(2, CLI_OFF)) attack.disableOutput();
else parameterError(list->get(2));
}
return;
}
bool beacon = false;
bool deauth = false;
bool deauthAll = false;
bool probe = false;
bool output = true;
uint32_t timeout = settings.getAttackTimeout() * 1000;
for (int i = 1; i < list->size(); i++) {
if (eqlsCMD(i, CLI_BEACON)) beacon = true;
else if (eqlsCMD(i, CLI_DEAUTH)) deauth = true;
else if (eqlsCMD(i, CLI_DEAUTHALL)) deauthAll = true;
else if (eqlsCMD(i, CLI_PROBE)) probe = true;
else if (eqlsCMD(i, CLI_NOOUTPUT)) output = false;
else if (eqlsCMD(i, CLI_TIMEOUT)) {
timeout = getTime(list->get(i + 1));
i++;
}
else parameterError(list->get(i));
}
attack.start(beacon, deauth, deauthAll, probe, output, timeout);
}
// ===== GET/SET ===== //
// get <setting>
else if (eqlsCMD(0, CLI_GET) && list->size() == 2) {
prntln(settings.get(list->get(1).c_str()));
}
// set <setting> <value>
else if (eqlsCMD(0, CLI_SET) && list->size() == 3) {
settings.set(list->get(1).c_str(), list->get(2));
}
// ====== CHICKEN ===== //
else if (eqlsCMD(0, CLI_CHICKEN)) {
prntln(CLI_CHICKEN_OUTPUT);
}
// ===== STOP ===== //
// stop [<mode>]
else if (eqlsCMD(0, CLI_STOP)) {
led.setMode(LED_MODE_IDLE, true);
if (list->size() >= 2 && !(eqlsCMD(1, CLI_ALL))) {
for (int i = 1; i < list->size(); i++) {
if (eqlsCMD(i, CLI_SCAN)) scan.stop();
else if (eqlsCMD(i, CLI_ATTACK)) attack.stop();
else if (eqlsCMD(i, CLI_SCRIPT)) stopScript();
else parameterError(list->get(i));
}
} else {
scan.stop();
attack.stop();
stopScript();
}
}
// ===== SYSTEM ===== //
// sysinfo
else if (eqlsCMD(0, CLI_SYSINFO)) {
prntln(CLI_SYSTEM_INFO);
char s[150];
sprintf(s,str(CLI_SYSTEM_OUTPUT).c_str(), 81920 - system_get_free_heap_size(), 100 - system_get_free_heap_size() / (81920 / 100), system_get_free_heap_size(), system_get_free_heap_size() / (81920 / 100), 81920);
prntln(String(s));
prnt(CLI_SYSTEM_CHANNEL);
prntln(settings.getChannel());
uint8_t mac[6];
prnt(CLI_SYSTEM_AP_MAC);
wifi_get_macaddr(0x01, mac);
prntln(macToStr(mac));
prnt(CLI_SYSTEM_ST_MAC);
wifi_get_macaddr(0x00, mac);
prntln(macToStr(mac));
FSInfo fs_info;
SPIFFS.info(fs_info);
sprintf(s,str(CLI_SYSTEM_RAM_OUT).c_str(), fs_info.usedBytes, fs_info.usedBytes / (fs_info.totalBytes / 100), fs_info.totalBytes - fs_info.usedBytes, (fs_info.totalBytes - fs_info.usedBytes) / (fs_info.totalBytes / 100), fs_info.totalBytes);
prnt(String(s));
sprintf(s,str(CLI_SYSTEM_SPIFFS_OUT).c_str(), fs_info.blockSize, fs_info.pageSize);
prnt(String(s));
prntln(CLI_FILES);
Dir dir = SPIFFS.openDir(String(SLASH));
while (dir.next()) {
prnt(String(SPACE) + String(SPACE) + dir.fileName() + String(SPACE));
File f = dir.openFile("r");
prnt(int(f.size()));
prntln(str(CLI_BYTES));
}
printWifiStatus();
prntln(CLI_SYSTEM_FOOTER);
}
// ===== RESET ===== //
// reset
else if (eqlsCMD(0, CLI_RESET)) {
settings.reset();
}
// ===== CLEAR ===== //
// clear
else if (eqlsCMD(0, CLI_CLEAR)) {
for (int i = 0; i < 100; i++)
prnt(HASHSIGN);
for (int i = 0; i < 60; i++)
prntln();
}
// ===== REBOOT ===== //
// reboot
else if (eqlsCMD(0, CLI_REBOOT)) {
ESP.reset();
}
// ===== FORMAT ==== //
// format
else if (eqlsCMD(0, CLI_FORMAT)) {
SPIFFS.format();
prntln(CLI_CLEARED_SPIFFS);
}
// ===== DELETE ==== //
// delete <file> [<lineFrom>] [<lineTo>]
else if (list->size() >= 2 && eqlsCMD(0, CLI_DELETE)) {
if (list->size() == 2) {
// remove whole file
if (removeFile(list->get(1))) {
prnt(CLI_REMOVED);
prntln(list->get(1));
} else {
prnt(CLI_ERROR_REMOVING);
prntln(list->get(1));
}
} else {
// remove certain lines
int beginLine = list->get(2).toInt();
int endLine = list->size() == 4 ? list->get(3).toInt() : beginLine;
if (removeLines(list->get(1), beginLine, endLine)) {
prnt(CLI_REMOVING_LINES);
prnt(beginLine);
prnt(String(SPACE) + String(DASH) + String(SPACE));
prnt(endLine);
prntln(String(SPACE) + list->get(1));
} else {
prnt(CLI_ERROR_REMOVING);
prntln(list->get(1));
}
}
}
// ===== COPY ==== //
// delete <file> <newfile>
else if (list->size() == 3 && eqlsCMD(0, CLI_COPY)) {
if(copyFile(list->get(1), list->get(2))){
prntln(CLI_COPIED_FILES);
}else{
prntln(CLI_ERROR_COPYING);
}
}
// ===== RENAME ==== //
// delete <file> <newfile>
else if (list->size() == 3 && eqlsCMD(0, CLI_RENAME)) {
if(renameFile(list->get(1), list->get(2))){
prntln(CLI_RENAMED_FILE);
}else{
prntln(CLI_ERROR_RENAMING_FILE);
}
}
// ===== WRITE ==== //
// write <file> <commands>
else if (list->size() >= 3 && eqlsCMD(0, CLI_WRITE)) {
String tmp = String();
for (int i = 2; i < list->size(); i++) {
tmp += list->get(i);
if (i < list->size() - 1) tmp += SPACE;
}
prnt(CLI_WRITTEN);
prnt(tmp);
prnt(CLI_TO);
prntln(list->get(1));
tmp += NEWLINE;
appendFile(list->get(1), tmp);
}
// ===== REPLACE ==== //
// replace <file> <line> <new-content>
else if (list->size() >= 4 && eqlsCMD(0, CLI_REPLACE)) {
int line = list->get(2).toInt();
String tmp = String();
for (int i = 3; i < list->size(); i++) {
tmp += list->get(i);
if (i < list->size() - 1) tmp += SPACE;
}
if (replaceLine(list->get(1), line, tmp)) {
prnt(CLI_REPLACED_LINE);
prnt(line);
prnt(CLI_WITH);
prntln(list->get(1));
} else {
prnt(CLI_ERROR_REPLACING_LINE);
prntln(list->get(1));
}
}
// ===== RUN ==== //
// run <file> [continue <num>]
else if (!executing && list->size() >= 2 && eqlsCMD(0, CLI_RUN)) {
executing = true;
execPath = list->get(1);
if (eqlsCMD(2, CLI_CONTINUE)) {
continuously = true;
continueTime = 10000;
if(list->size() == 3){
continueTime = getTime(list->get(3));
if(continueTime < 1000) continueTime = 1000;
}
}
}
// ===== PRINT ==== //
// print <file> [<lines>]
else if (list->size() >= 2 && eqlsCMD(0, CLI_PRINT)) {
readFileToSerial(list->get(1), eqlsCMD(2, CLI_LINE));
prntln();
}
// ===== INFO ===== //
// info
else if (eqlsCMD(0, CLI_INFO)) {
prntln(CLI_INFO_HEADER);
prnt(CLI_INFO_SOFTWARE);
prntln(settings.getVersion());
prntln(CLI_INFO_COPYRIGHT);
prntln(CLI_INFO_LICENSE);
prntln(CLI_INFO_ADDON);
prntln(CLI_INFO_HEADER);
}
// ===== SEND ===== //
// send deauth <apMac> <stMac> <rason> <channel>
else if (eqlsCMD(0, CLI_SEND) && list->size() == 6 && eqlsCMD(1, CLI_DEAUTH)) {
uint8_t apMac[6];
uint8_t stMac[6];
strToMac(list->get(2), apMac);
strToMac(list->get(3), stMac);
uint8_t reason = list->get(4).toInt();
uint8_t channel = list->get(5).toInt();
prnt(CLI_DEAUTHING);
prnt(macToStr(apMac));
prnt(CLI_ARROW);
prntln(macToStr(stMac));
attack.deauthDevice(apMac, stMac, reason, channel);
}
// send beacon <mac> <ssid> <ch> [wpa2]
else if (eqlsCMD(0, CLI_SEND) && list->size() >= 5 && eqlsCMD(1, CLI_BEACON)) {
uint8_t mac[6];
strToMac(list->get(2), mac);
uint8_t channel = list->get(4).toInt();
String ssid = list->get(3);
for (int i = ssid.length(); i < 32; i++)
ssid += SPACE;
prnt(CLI_SENDING_BEACON);
prnt(list->get(3));
prntln(DOUBLEQUOTES);
attack.sendBeacon(mac, ssid.c_str(), channel, eqlsCMD(5, CLI_WPA2));
}
// send probe <mac> <ssid> <ch>
else if (eqlsCMD(0, CLI_SEND) && list->size() == 5 && eqlsCMD(1, CLI_PROBE)) {
uint8_t mac[6];
strToMac(list->get(2), mac);
uint8_t channel = list->get(4).toInt();
String ssid = list->get(3);
for (int i = ssid.length(); i < 32; i++)
ssid += SPACE;
prnt(CLI_SENDING_PROBE);
prnt(list->get(3));
prntln(DOUBLEQUOTES);
attack.sendProbe(mac, ssid.c_str(), channel);
}
// send custom <packet>
else if (eqlsCMD(0, CLI_SEND) && eqlsCMD(1, CLI_CUSTOM)) {
String packetStr = list->get(2);
packetStr.replace(String(DOUBLEQUOTES), String());
uint16_t counter = 0;
uint16_t packetSize = packetStr.length() / 2;
uint8_t packet[packetSize];
for (int i = 0; i < packetSize; i++)
packet[i] = strtoul((packetStr.substring(i * 2, i * 2 + 2)).c_str(), NULL, 16);
if (attack.sendPacket(packet, packetSize, &counter, wifi_channel, 10)) prntln(CLI_CUSTOM_SENT);
else prntln(CLI_CUSTOM_FAILED);
}
// ===== LED ===== //
// led <r> <g> <b> [<brightness>]
else if (list->size() >= 4 && list->size() <= 5 && eqlsCMD(0, CLI_LED)) {
if (list->size() == 4)
led.setColor(list->get(1).toInt(), list->get(2).toInt(), list->get(3).toInt());
else
led.setColor(list->get(1).toInt(), list->get(2).toInt(), list->get(3).toInt(), list->get(4).toInt());
}
// led <#rrggbb> [<brightness>]
else if (list->size() >= 2 && list->size() <= 3 && eqlsCMD(0, CLI_LED) && list->get(1).charAt(0) == HASHSIGN) {
uint8_t c[3];
strToColor(list->get(1), c);
if (list->size() == 2)
led.setColor(c[0], c[1], c[2]);
else
led.setColor(c[0], c[1], c[2], list->get(2).toInt());
}
// led <enable/disable>
else if (list->size() == 2 && eqlsCMD(0, CLI_LED)) {
if (eqlsCMD(1, CLI_ENABLE))
led.tempEnable();
else if (eqlsCMD(1, CLI_DISABLE))
led.tempDisable();
else
parameterError(list->get(1));
}
// ===== DELAY ===== //
else if (list->size() == 2 && eqlsCMD(0, CLI_DELAY)) {
uint32_t endTime = currentTime + getTime(list->get(1));
while (currentTime < endTime) {
// ------- loop function ----- //
currentTime = millis();
wifiUpdate(); // manage access point
scan.update(); // run scan
attack.update(); // run attacks
ssids.update(); // run random mode, if enabled
led.update(); // update LED color
// auto-save
if (settings.getAutosave() && currentTime - autosaveTime > settings.getAutosaveTime()) {
autosaveTime = currentTime;
names.save(false);
ssids.save(false);
settings.save(false);
}
// ------- loop function end ----- //
yield();
}
}
// ===== DRAW ===== //
else if (eqlsCMD(0, CLI_DRAW)) {
int height = 25;
int width = 2;
if (list->size() >= 2) height = list->get(1).toInt();
if (list->size() >= 3) width = list->get(2).toInt();
double scale = scan.getScaleFactor(height);
prnt(String(DASH) + String(DASH) + String(DASH) + String(DASH) + String(VERTICALBAR)); // ----|
for (int j = 0; j < SCAN_PACKET_LIST_SIZE; j++) {
for (int k = 0; k < width; k++) prnt(EQUALS);
}
prntln(VERTICALBAR);
for (int i = height; i >= 0; i--) {
char s[200];
if (i == height) sprintf(s,str(CLI_DRAW_OUTPUT).c_str(), scan.getMaxPacket() > height ? scan.getMaxPacket() : height);
else if (i == height / 2) sprintf(s,str(CLI_DRAW_OUTPUT).c_str(), scan.getMaxPacket() > height ? scan.getMaxPacket()/2 : height/2);
else if (i == 0) sprintf(s,str(CLI_DRAW_OUTPUT).c_str(), 0);
else{
s[0] = SPACE;
s[1] = SPACE;
s[2] = SPACE;
s[3] = SPACE;
s[4] = ENDOFLINE;
}
prnt(String(s));
prnt(VERTICALBAR);
for (int j = 0; j < SCAN_PACKET_LIST_SIZE; j++) {
if (scan.getPackets(j)*scale > i) {
for (int k = 0; k < width; k++) prnt(HASHSIGN);
} else {
for (int k = 0; k < width; k++) prnt(SPACE);
}
}
prntln(VERTICALBAR);
}
prnt(String(DASH) + String(DASH) + String(DASH) + String(DASH) + String(VERTICALBAR)); // ----|
for (int j = 0; j < SCAN_PACKET_LIST_SIZE; j++) {
for (int k = 0; k < width; k++) prnt(EQUALS);
}
prntln(VERTICALBAR);
prnt(String(SPACE) + String(SPACE) + String(SPACE) + String(SPACE) + String(VERTICALBAR));
for (int j = 0; j < SCAN_PACKET_LIST_SIZE; j++) {
char s[6];
String helper = String(PERCENT)+DASH+(String)width+D;
if(j==0) sprintf(s,helper.c_str(),SCAN_PACKET_LIST_SIZE-1);
else if(j==SCAN_PACKET_LIST_SIZE/2) sprintf(s,helper.c_str(),SCAN_PACKET_LIST_SIZE/2);
else if(j==SCAN_PACKET_LIST_SIZE-1) sprintf(s,helper.c_str(),0);
else{
int k;
for (k = 0; k < width; k++) s[k] = SPACE;
s[k] = ENDOFLINE;
}
prnt(s);
}
prntln(VERTICALBAR);
}
// ===== START/STOP AP ===== //
// startap
else if (eqlsCMD(0, CLI_STARTAP)) {
String path = str(W_WEBINTERFACE);
String ssid = settings.getSSID();
String password = settings.getPassword();
int ch = wifi_channel;
bool hidden = settings.getHidden();
bool captivePortal = settings.getCaptivePortal();;
for (int i = 1; i < list->size(); i++) {
if (eqlsCMD(i, CLI_PATH)) {
path = list->get(i + 1);
i++;
} else if (eqlsCMD(i, CLI_SSID)) {
ssid = list->get(i + 1);
i++;
} else if (eqlsCMD(i, CLI_PASSWORD)) {
password = list->get(i + 1);
i++;
} else if (eqlsCMD(i, CLI_CHANNEL)) {
ch = list->get(i + 1).toInt();
i++;
} else if (eqlsCMD(i, CLI_HIDDEN)) {
hidden = true;
} else if (eqlsCMD(i, CLI_CAPTIVEPORTAL)) {
captivePortal = true;
} else {
parameterError(list->get(1));
}
}
startAP(path, ssid, password, ch, hidden, captivePortal);
}
//stopap
else if (eqlsCMD(0, CLI_STOPAP)) {
stopAP();
}
// ===== SCREEN ===== //
// screen mode <menu/packetmonitor/buttontest/loading>
else if(eqlsCMD(0, CLI_SCREEN) && eqlsCMD(1,CLI_MODE)){
if(eqlsCMD(2,CLI_MODE_BUTTONTEST)) displayUI.mode = SCREEN_MODE_BUTTON_TEST;
else if(eqlsCMD(2,CLI_MODE_PACKETMONITOR)) displayUI.mode = SCREEN_MODE_PACKETMONITOR;
else if(eqlsCMD(2,CLI_MODE_LOADINGSCREEN)) displayUI.mode = SCREEN_MODE_LOADSCAN;
else if(eqlsCMD(2,CLI_MODE_MENU)) displayUI.mode = SCREEN_MODE_MENU;
else parameterError(list->get(2));
prntln(CLI_CHANGED_SCREEN);
}
// screen <on/off>
else if(eqlsCMD(0, CLI_SCREEN) && (eqlsCMD(1,CLI_ON) || eqlsCMD(1,CLI_OFF))){
if(eqlsCMD(1,CLI_ON)){
displayUI.on();
} else if(eqlsCMD(1,CLI_OFF)){
displayUI.off();
}
}
// ===== NOT FOUND ===== //
else {
prnt(CLI_ERROR_NOT_FOUND_A);
prnt(input);
prntln(CLI_ERROR_NOT_FOUND_B);
// some debug stuff
/*
Serial.println(list->get(0));
for(int i=0;i<input.length();i++){
Serial.print(input.charAt(i), HEX);
Serial.print(' ');
}
*/
}
}
Reference in New Issue View Git Blame Copy Permalink