158 lines
3.9 KiB
C++
158 lines
3.9 KiB
C++
#include <WiFi.h>
|
|
#include "WifiDevice.h"
|
|
#include "../PreferencesKeys.h"
|
|
#include "../Logger.h"
|
|
#include "../MqttTopics.h"
|
|
|
|
RTC_NOINIT_ATTR char WiFiDevice_reconfdetect[17];
|
|
|
|
WifiDevice::WifiDevice(const String& hostname, Preferences* _preferences)
|
|
: NetworkDevice(hostname)
|
|
{
|
|
_startAp = strcmp(WiFiDevice_reconfdetect, "reconfigure_wifi") == 0;
|
|
|
|
_restartOnDisconnect = _preferences->getBool(preference_restart_on_disconnect);
|
|
|
|
size_t caLength = _preferences->getString(preference_mqtt_ca,_ca,TLS_CA_MAX_SIZE);
|
|
size_t crtLength = _preferences->getString(preference_mqtt_crt,_cert,TLS_CERT_MAX_SIZE);
|
|
size_t keyLength = _preferences->getString(preference_mqtt_key,_key,TLS_KEY_MAX_SIZE);
|
|
|
|
if(caLength > 1) // length is 1 when empty
|
|
{
|
|
Log->println(F("MQTT over TLS."));
|
|
Log->println(_ca);
|
|
_wifiClientSecure = new WiFiClientSecure();
|
|
_wifiClientSecure->setCACert(_ca);
|
|
if(crtLength > 1 && keyLength > 1) // length is 1 when empty
|
|
{
|
|
Log->println(F("MQTT with client certificate."));
|
|
Log->println(_cert);
|
|
Log->println(_key);
|
|
_wifiClientSecure->setCertificate(_cert);
|
|
_wifiClientSecure->setPrivateKey(_key);
|
|
}
|
|
_mqttClient = new MqttClient(*_wifiClientSecure);
|
|
} else
|
|
{
|
|
Log->println(F("MQTT without TLS."));
|
|
_wifiClient = new WiFiClient();
|
|
_mqttClient = new MqttClient(*_wifiClient);
|
|
}
|
|
|
|
if(_preferences->getBool(preference_mqtt_log_enabled))
|
|
{
|
|
_path = new char[200];
|
|
memset(_path, 0, sizeof(_path));
|
|
|
|
String pathStr = _preferences->getString(preference_mqtt_lock_path);
|
|
pathStr.concat(mqtt_topic_log);
|
|
strcpy(_path, pathStr.c_str());
|
|
Log = new MqttLogger(*_mqttClient, _path, MqttLoggerMode::MqttAndSerial);
|
|
}
|
|
}
|
|
|
|
MqttClient *WifiDevice::mqttClient()
|
|
{
|
|
return _mqttClient;
|
|
}
|
|
|
|
void WifiDevice::initialize()
|
|
{
|
|
std::vector<const char *> wm_menu;
|
|
wm_menu.push_back("wifi");
|
|
wm_menu.push_back("exit");
|
|
// reduced tieout if ESP is set to restart on disconnect
|
|
_wm.setConfigPortalTimeout(_restartOnDisconnect ? 60 * 3 : 60 * 30);
|
|
_wm.setShowInfoUpdate(false);
|
|
_wm.setMenu(wm_menu);
|
|
_wm.setHostname(_hostname);
|
|
|
|
_wm.setAPCallback(clearRtcInitVar);
|
|
|
|
bool res = false;
|
|
|
|
if(_startAp)
|
|
{
|
|
Log->println(F("Opening WiFi configuration portal."));
|
|
res = _wm.startConfigPortal();
|
|
}
|
|
else
|
|
{
|
|
res = _wm.autoConnect(); // password protected ap
|
|
}
|
|
|
|
if(!res) {
|
|
Log->println(F("Failed to connect. Wait for ESP restart."));
|
|
delay(1000);
|
|
ESP.restart();
|
|
}
|
|
else {
|
|
Log->print(F("WiFi connected: "));
|
|
Log->println(WiFi.localIP().toString());
|
|
}
|
|
|
|
if(_restartOnDisconnect)
|
|
{
|
|
WiFi.onEvent([&](WiFiEvent_t event, WiFiEventInfo_t info)
|
|
{
|
|
if(event == ARDUINO_EVENT_WIFI_STA_DISCONNECTED)
|
|
{
|
|
onDisconnected();
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
void WifiDevice::reconfigure()
|
|
{
|
|
strcpy(WiFiDevice_reconfdetect, "reconfigure_wifi");
|
|
delay(200);
|
|
ESP.restart();
|
|
}
|
|
|
|
void WifiDevice::printError()
|
|
{
|
|
if(_wifiClientSecure != nullptr)
|
|
{
|
|
char lastError[100];
|
|
_wifiClientSecure->lastError(lastError,100);
|
|
Log->println(lastError);
|
|
}
|
|
Log->print(F("Free Heap: "));
|
|
Log->println(ESP.getFreeHeap());
|
|
}
|
|
|
|
bool WifiDevice::isConnected()
|
|
{
|
|
return WiFi.isConnected();
|
|
}
|
|
|
|
ReconnectStatus WifiDevice::reconnect()
|
|
{
|
|
delay(3000);
|
|
return isConnected() ? ReconnectStatus::Success : ReconnectStatus::Failure;
|
|
}
|
|
|
|
void WifiDevice::update()
|
|
{
|
|
|
|
}
|
|
|
|
void WifiDevice::onDisconnected()
|
|
{
|
|
if(millis() > 60000)
|
|
{
|
|
ESP.restart();
|
|
}
|
|
}
|
|
|
|
int8_t WifiDevice::signalStrength()
|
|
{
|
|
return WiFi.RSSI();
|
|
}
|
|
|
|
void WifiDevice::clearRtcInitVar(WiFiManager *)
|
|
{
|
|
memset(WiFiDevice_reconfdetect, 0, sizeof WiFiDevice_reconfdetect);
|
|
}
|