91 lines
2.6 KiB
C++
91 lines
2.6 KiB
C++
/*
|
|
MIT License
|
|
|
|
Copyright (c) 2021-22 Magnus
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in all
|
|
copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
SOFTWARE.
|
|
*/
|
|
#ifndef SRC_WIFI_HPP_
|
|
#define SRC_WIFI_HPP_
|
|
|
|
#if defined (ESP8266)
|
|
#include <ESP8266WiFi.h>
|
|
#else // defined (ESP32)
|
|
#include <WiFiClient.h>
|
|
#include <WiFiClientSecure.h>
|
|
#endif
|
|
|
|
#define WIFI_DEFAULT_SSID "GravityMon" // Name of created SSID
|
|
#define WIFI_DEFAULT_PWD "password" // Password for created SSID
|
|
#define WIFI_MDNS "gravitymon" // Prefix for MDNS name
|
|
|
|
// tcp cleanup, to avoid memory crash.
|
|
struct tcp_pcb;
|
|
extern struct tcp_pcb* tcp_tw_pcbs;
|
|
extern "C" void tcp_abort(struct tcp_pcb* pcb);
|
|
|
|
class WifiConnection {
|
|
private:
|
|
// WIFI
|
|
WiFiClient _client;
|
|
WiFiClientSecure _secureClient;
|
|
|
|
// OTA
|
|
bool _newFirmware = false;
|
|
bool parseFirmwareVersionString(int (&num)[3], const char* version);
|
|
void downloadFile(const char* fname);
|
|
void connectAsync();
|
|
bool waitForConnection(int maxTime = 20);
|
|
|
|
public:
|
|
// WIFI
|
|
void init();
|
|
|
|
bool connect();
|
|
bool disconnect();
|
|
bool isConnected();
|
|
bool isDoubleResetDetected();
|
|
void stopDoubleReset();
|
|
bool hasConfig();
|
|
String getIPAddress();
|
|
void startPortal();
|
|
void loop();
|
|
|
|
WiFiClient& getWifiClient() { return _client; }
|
|
WiFiClientSecure& getWifiClientSecure() { return _secureClient; }
|
|
void closeWifiClient() {
|
|
_client.stop();
|
|
_secureClient.stop();
|
|
|
|
// Cleanup memory allocated by open tcp connetions.
|
|
while (tcp_tw_pcbs) tcp_abort(tcp_tw_pcbs);
|
|
}
|
|
|
|
// OTA
|
|
bool updateFirmware();
|
|
bool checkFirmwareVersion();
|
|
};
|
|
|
|
// Global instance created
|
|
extern WifiConnection myWifi;
|
|
|
|
#endif // SRC_WIFI_HPP_
|
|
|
|
// EOF
|