esp8266_deauther/esp8266_deauther/APScan.cpp

#include "APScan.h"

APScan::APScan(){
  
}

bool APScan::start(){
    if(debug){
      Serial.println("starting AP scan...");
      Serial.println("MAC - Ch - RSSI - Encrypt. - SSID - Hidden");// - Vendor");
    }
    aps._clear();
    for(int i=0;i<maxAPScanResults;i++) selected[i] = false;
    results = WiFi.scanNetworks(false, true); // lets scanNetworks return hidden APs. (async = false & show_hidden = true)
    
    for(int i=0;i<results && i<maxAPScanResults;i++){
      Mac _ap;
      _ap.set(WiFi.BSSID(i)[0],WiFi.BSSID(i)[1],WiFi.BSSID(i)[2],WiFi.BSSID(i)[3],WiFi.BSSID(i)[4],WiFi.BSSID(i)[5]);
      aps.add(_ap);
      channels[i] = WiFi.channel(i);
      rssi[i] = WiFi.RSSI(i);
      encryption[i] = WiFi.encryptionType(i);
      hidden[i] = WiFi.isHidden(i);
      String _ssid = WiFi.SSID(i);
      _ssid.replace("\"","\\\"");
      _ssid.toCharArray(names[i],33);
      //data_getVendor(WiFi.BSSID(i)[0],WiFi.BSSID(i)[1],WiFi.BSSID(i)[2]).toCharArray(vendors[i],9);
      if(debug){
        _ap._print();
        Serial.print(" - ");
        Serial.print(channels[i]);
        Serial.print(" - ");
        Serial.print(rssi[i]);
        Serial.print(" - ");
        Serial.print(getEncryption(encryption[i]));
        Serial.print(" - ");
        Serial.print(names[i]);
        Serial.print(" - ");
        Serial.print(hidden[i]);
        //Serial.print(" - ");
        //Serial.print(vendors[i]);
        Serial.println();
      }
    }

    //for debugging the APScan crash bug
    /*if(debug){
      for(int i=results;i<maxAPScanResults;i++){
        Mac _ap;
        _ap.set(random(255),random(255),random(255),random(255),random(255),random(255));
        aps.add(_ap);
        channels[i] = random(1,12);
        rssi[i] = random(-30,-90);
        encryption[i] = ENC_TYPE_NONE;
        String _ssid = "test_dbeJwq3tPtJsuWtgULgShD9dxXV";
        _ssid.toCharArray(names[i],33);

        _ap._print();
        Serial.print(" - ");
        Serial.print(channels[i]);
        Serial.print(" - ");
        Serial.print(rssi[i]);
        Serial.print(" - ");
        Serial.print(getEncryption(encryption[i]));
        Serial.print(" - ");
        Serial.print(names[i]);
        Serial.println();

        results++;
      }
    }*/
    
    if(debug) Serial.println("scan done");
    if(debug) Serial.println(getResults());
    return true;
}

String APScan::getEncryption(int code){
  switch (code) {
    case ENC_TYPE_NONE:
      return "none";
      break;
    case ENC_TYPE_WEP:
      return "WEP";
      break;
    case ENC_TYPE_TKIP:
      return "WPA";
      break;
    case ENC_TYPE_CCMP:
      return "WPA2";
      break;
    case ENC_TYPE_AUTO:
      return "WPA*";
      break;
  }
  return "?";
}

String APScan::getAPName(int num){ 
    if(isHidden(num)) return "* Hidden SSID *";
    return names[num]; 
}
String APScan::getAPEncryption(int num){ return getEncryption(encryption[num]); }
//String APScan::getAPVendor(int num){ return vendors[num]; }
String APScan::getAPMac(int num){ return aps._get(num).toString(); }
bool APScan::getAPSelected(int num){ return selected[num]; }
bool APScan::isHidden(int num){ return hidden[num]; }
int APScan::getAPRSSI(int num){ return rssi[num]; }
int APScan::getAPChannel(int num){ return channels[num]; }

int APScan::getFirstTarget(){
  for(int i=0;i<maxAPScanResults;i++){
    if(isSelected(i)) return i;
  }
  return -1;
}

String APScan::getResults(){
  if(debug) Serial.print("getting AP scan result JSON ");
  String json = "{ \"aps\":[ ";
  for(int i=0;i<results && i<maxAPScanResults;i++){
    if(debug) Serial.print(".");
    json += "{";
    json += "\"i\":"+(String)i+",";
    json += "\"c\":"+(String)getAPChannel(i)+",";
    json += "\"m\":\""+getAPMac(i)+"\",";
    json += "\"ss\":\""+getAPName(i)+"\",";
    json += "\"r\":"+(String)getAPRSSI(i)+",";
    json += "\"e\":"+(String)encryption[i]+",";
    //json += "\"v\":\""+getAPVendor(i)+"\",";
    json += "\"se\":"+(String)getAPSelected(i);
    json += "}";
    if((i!=results-1) && (i!=maxAPScanResults-1)) json += ",";
  }
  json += "] }";
  if(debug) Serial.println("done");
  return json;
}

void APScan::select(int num){
  if(debug) Serial.println("seect "+(String)num+" - "+!selected[num]);
  if(selected[num]){
    selected[num] = false;
    selectedSum--;
  }else{
    selected[num] = true;
    selectedSum++;
  }
}

bool APScan::isSelected(int num){
  return selected[num];
}