esp8266_deauther/esp8266_deauther/SSIDList.cpp

#include "SSIDList.h"

SSIDList::SSIDList() {
  if (listAdr + SSIDListLength * SSIDLength > 4096) Serial.println("WARNING: EEPROM overflow!");
}

void SSIDList::load() {
  len = EEPROM.read(lenAdr);
  if ( len < 0 || len > SSIDListLength) {
    clear();
    save();
  }
  for (int i = 0; i < len; i++) {
    for (int h = 0; h < SSIDLength; h++) {
      char _nextChar = EEPROM.read(listAdr + (i * SSIDLength) + h);
      names[i][h] = _nextChar;
    }
    encrypted[i] = EEPROM.read(encAdr + i);
  }
}

void SSIDList::clear() {
  len = 0;
}

void SSIDList::add(String name, bool enc) {
  if (len < SSIDListLength) {
    for (int i = 0; i < SSIDLength; i++) {
      if (i < name.length()) names[len][i] = name[i];
      else names[len][i] = 0x00;
    }
    encrypted[len] = enc;
    len++;
  }
}

void SSIDList::addClone(String name, int num, bool enc) {
  int _restSSIDLen = SSIDLength - name.length();
  String _apName;

  for (int c = 0; c < SSIDListLength && c < num; c++) {
    _apName = name;

    if (c < _restSSIDLen){
      for (int d = 0; d < _restSSIDLen - c; d++) _apName += " "; //e.g. "SAMPLEAP   "
    } else if (c < _restSSIDLen * 2) {
      //_apName = " " + _apName;
      _apName = "\u200B"+_apName;
      for (int d = 0; d < (_restSSIDLen - 3) - c / 2; d++) _apName += " "; //e.g. "SAMPLEAP   " (zero widtdh char at the beginning)
    } else if (c < _restSSIDLen * 3) {
      _apName = "." + _apName;
      for (int d = 0; d < (_restSSIDLen - 1) - c / 3; d++) _apName += " "; //e.g. ".SAMPLEAP   "
    } else {
      for (int d = 0; d < _restSSIDLen - 2; d++) _apName += " ";
      _apName += (String)c;//e.g. "SAMPLEAP        78"
    }
    add(_apName, enc);
  }
}

String SSIDList::get(int num) {
  String _name = "";
  for (int i = 0; i < SSIDLength; i++) {
    if (names[num][i] != 0x00) _name += names[num][i];
  }
  return _name;
}

bool SSIDList::isEncrypted(int num){
  return encrypted[num];
}

void SSIDList::remove(int num) {
  if (num >= 0 && num < len) {
    for (int i = num; i < len - 1; i++) {
      for (int h = 0; h < SSIDLength; h++) {
        names[i][h] = names[i + 1][h];
      }
      encrypted[i] = encrypted[i + 1];
    }
    len--;
  }
}

void SSIDList::save() {
  if (debug) Serial.print("saving SSIDList...");
  EEPROM.write(lenAdr, len);
  for (int i = 0; i < len; i++) {
    for (int h = 0; h < SSIDLength; h++) {
      EEPROM.write(listAdr + (i * SSIDLength) + h, names[i][h]);
    }
    EEPROM.write(encAdr + i, encrypted[i]);
  }
  EEPROM.commit();
  if (debug) Serial.println("done");
}

void SSIDList::_random() {
  String _rName;
  for (int i = len; i < SSIDListLength; i++) {
    _rName = "";
    for (int h = 0; h < SSIDLength; h++) _rName += letters[random(0, sizeof(letters))];
    add(_rName, random(2) > 0.5 );
  }
}