Rune Harlyk
112 lines
3.7 KiB
C++
112 lines
3.7 KiB
C++
#include <ap_service.h>
|
|
#include <communication/webserver.h>
|
|
|
|
static const char *TAG = "APService";
|
|
|
|
APService::APService() : _dnsServer(nullptr), _lastManaged(0), _reconfigureAp(false), _recoveryMode(false) {}
|
|
|
|
APService::~APService() {
|
|
if (_dnsServer) {
|
|
delete _dnsServer;
|
|
_dnsServer = nullptr;
|
|
}
|
|
}
|
|
|
|
void APService::begin() {
|
|
_settings = EventBus::instance().peek<APSettings>();
|
|
_settingsHandle = EventBus::instance().subscribe<APSettings>([this](const APSettings &s) {
|
|
_settings = s;
|
|
reconfigureAP();
|
|
});
|
|
}
|
|
|
|
void APService::registerRoutes(WebServer &s) {
|
|
s.on("/api/ap/status", HTTP_GET, [this](httpd_req_t *request) { return getStatusProto(request); });
|
|
}
|
|
|
|
esp_err_t APService::getStatusProto(httpd_req_t *request) {
|
|
api_Response res = api_Response_init_zero;
|
|
res.status_code = 200;
|
|
res.which_payload = api_Response_ap_status_tag;
|
|
statusProto(res.payload.ap_status);
|
|
return WebServer::send(request, 200, res, api_Response_fields);
|
|
}
|
|
|
|
void APService::statusProto(api_APStatus &proto) {
|
|
proto.status = getAPNetworkStatus();
|
|
proto.ip_address = static_cast<uint32_t>(WiFi.softAPIP());
|
|
std::string mac = WiFi.softAPmacAddress();
|
|
strncpy(proto.mac_address, mac.c_str(), sizeof(proto.mac_address) - 1);
|
|
proto.mac_address[sizeof(proto.mac_address) - 1] = '\0';
|
|
proto.station_num = WiFi.softAPgetStationNum();
|
|
}
|
|
|
|
APNetworkStatus APService::getAPNetworkStatus() {
|
|
wifi_mode_t currentWiFiMode = WiFi.getMode();
|
|
bool apActive = currentWiFiMode == WIFI_MODE_AP || currentWiFiMode == WIFI_MODE_APSTA;
|
|
if (apActive && _settings.provision_mode != AP_MODE_ALWAYS && WiFi.status() == WL_CONNECTED) {
|
|
return LINGERING;
|
|
}
|
|
return apActive ? ACTIVE : INACTIVE;
|
|
}
|
|
|
|
void APService::reconfigureAP() {
|
|
_lastManaged = esp_timer_get_time() / 1000 - MANAGE_NETWORK_DELAY;
|
|
_reconfigureAp = true;
|
|
_recoveryMode = false;
|
|
}
|
|
|
|
void APService::recoveryMode() {
|
|
ESP_LOGI(TAG, "Recovery Mode needed");
|
|
_lastManaged = esp_timer_get_time() / 1000 - MANAGE_NETWORK_DELAY;
|
|
_recoveryMode = true;
|
|
_reconfigureAp = true;
|
|
}
|
|
|
|
void APService::loop() {
|
|
EXECUTE_EVERY_N_MS(MANAGE_NETWORK_DELAY, manageAP());
|
|
handleDNS();
|
|
}
|
|
|
|
void APService::manageAP() {
|
|
wifi_mode_t currentWiFiMode = WiFi.getMode();
|
|
if (_settings.provision_mode == AP_MODE_ALWAYS ||
|
|
(_settings.provision_mode == AP_MODE_DISCONNECTED && WiFi.status() != WL_CONNECTED) || _recoveryMode) {
|
|
if (_reconfigureAp || currentWiFiMode == WIFI_MODE_NULL || currentWiFiMode == WIFI_MODE_STA) {
|
|
startAP();
|
|
}
|
|
} else if ((currentWiFiMode == WIFI_MODE_AP || currentWiFiMode == WIFI_MODE_APSTA) &&
|
|
(_reconfigureAp || !WiFi.softAPgetStationNum())) {
|
|
stopAP();
|
|
}
|
|
_reconfigureAp = false;
|
|
}
|
|
|
|
void APService::startAP() {
|
|
ESP_LOGI(TAG, "Starting software access point: %s", _settings.ssid);
|
|
WiFi.softAPConfig(IPAddress(_settings.local_ip), IPAddress(_settings.gateway_ip), IPAddress(_settings.subnet_mask));
|
|
WiFi.softAP(_settings.ssid, _settings.password, _settings.channel, _settings.ssid_hidden, _settings.max_clients);
|
|
#if CONFIG_IDF_TARGET_ESP32C3
|
|
WiFi.setTxPower(8);
|
|
#endif
|
|
if (!_dnsServer) {
|
|
IPAddress apIp = WiFi.softAPIP();
|
|
ESP_LOGI(TAG, "Starting captive portal on %s", apIp.toString().c_str());
|
|
_dnsServer = new DNSServer;
|
|
_dnsServer->start(DNS_PORT, "*", apIp);
|
|
}
|
|
}
|
|
|
|
void APService::stopAP() {
|
|
if (_dnsServer) {
|
|
ESP_LOGI(TAG, "Stopping captive portal");
|
|
_dnsServer->stop();
|
|
delete _dnsServer;
|
|
_dnsServer = nullptr;
|
|
}
|
|
ESP_LOGI(TAG, "Stopping AP");
|
|
WiFi.softAPdisconnect(true);
|
|
}
|
|
|
|
void APService::handleDNS() {}
|