Tasmota/tasmota/xdrv_62_improv.ino

349 lines
12 KiB
Arduino
Raw Normal View History

/*
xdrv_62_improv.ino - IMPROV support for Tasmota
SPDX-FileCopyrightText: 2022 Theo Arends
SPDX-License-Identifier: GPL-3.0-only
*/
#ifdef USE_IMPROV
/*********************************************************************************************\
* Serial implementation of IMPROV for initial wifi configuration using esp-web-tools
*
* See https://esphome.github.io/esp-web-tools/ and https://www.improv-wifi.com/serial/
\*********************************************************************************************/
#define XDRV_62 62
#define IMPROV_WIFI_TIMEOUT 30 // Max seconds wait for wifi connection after reconfig
//#define IMPROV_DEBUG
enum ImprovError {
IMPROV_ERROR_NONE = 0x00,
IMPROV_ERROR_INVALID_RPC = 0x01,
IMPROV_ERROR_UNKNOWN_RPC = 0x02,
IMPROV_ERROR_UNABLE_TO_CONNECT = 0x03,
IMPROV_ERROR_NOT_AUTHORIZED = 0x04,
IMPROV_ERROR_UNKNOWN = 0xFF,
};
enum ImprovState {
IMPROV_STATE_STOPPED = 0x00,
IMPROV_STATE_AWAITING_AUTHORIZATION = 0x01,
IMPROV_STATE_AUTHORIZED = 0x02,
IMPROV_STATE_PROVISIONING = 0x03,
IMPROV_STATE_PROVISIONED = 0x04,
};
enum ImprovCommand {
IMPROV_UNKNOWN = 0x00,
IMPROV_WIFI_SETTINGS = 0x01,
IMPROV_GET_CURRENT_STATE = 0x02,
IMPROV_GET_DEVICE_INFO = 0x03,
IMPROV_GET_WIFI_NETWORKS = 0x04,
IMPROV_BAD_CHECKSUM = 0xFF,
};
enum ImprovSerialType {
IMPROV_TYPE_CURRENT_STATE = 0x01,
IMPROV_TYPE_ERROR_STATE = 0x02,
IMPROV_TYPE_RPC = 0x03,
IMPROV_TYPE_RPC_RESPONSE = 0x04
};
static const uint8_t IMPROV_SERIAL_VERSION = 1;
struct IMPROV {
uint8_t wifi_timeout;
uint8_t seriallog_level;
2022-04-03 13:48:30 +01:00
uint8_t version;
} Improv;
/*********************************************************************************************/
void ImprovWriteData(uint8_t* data, uint32_t size) {
data[0] = 'I';
data[1] = 'M';
data[2] = 'P';
data[3] = 'R';
data[4] = 'O';
data[5] = 'V';
data[6] = IMPROV_SERIAL_VERSION; // 0x01
uint8_t checksum = 0x00;
for (uint32_t i = 0; i < size -1; i++) {
checksum += data[i];
}
data[size -1] = checksum;
AddLog(LOG_LEVEL_DEBUG_MORE, PSTR("IMP: Send '%*_H'"), size, data);
for (uint32_t i = 0; i < size; i++) {
Serial.write(data[i]);
}
Serial.write('\n');
}
void ImprovSendCmndState(uint32_t command, uint32_t state) {
uint8_t data[11];
data[7] = command;
data[8] = 1;
data[9] = state;
ImprovWriteData(data, sizeof(data));
}
void ImprovSendState(uint32_t state) {
#ifdef IMPROV_DEBUG
AddLog(LOG_LEVEL_DEBUG, PSTR("IMP: State %d"), state);
#endif
RtcSettings.improv_state = state;
ImprovSendCmndState(IMPROV_TYPE_CURRENT_STATE, state); // 0x01
}
void ImprovSendError(uint32_t error) {
#ifdef IMPROV_DEBUG
AddLog(LOG_LEVEL_DEBUG, PSTR("IMP: Error %d"), error);
#endif
ImprovSendCmndState(IMPROV_TYPE_ERROR_STATE, error); // 0x02
}
void ImprovSendResponse(uint8_t* response, uint32_t size) {
uint8_t data[9 + size];
data[7] = IMPROV_TYPE_RPC_RESPONSE; // 0x04
data[8] = size -1;
memcpy(data +9, response, size);
2022-04-02 13:17:33 +01:00
data[10] = size -3; // Total length of strings following
if (data[10]) {
// Replace '\n' (= lf) with string length
2022-04-02 14:04:10 +01:00
// 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 ...
// I M P R O V ve ty le co pl lf T a s m o t a lf 1 1 . 0 . 0 . 5 lf ...
// I M P R O V ve ty le co pl l1 T a s m o t a l2 1 1 . 0 . 0 . 5 lf ...
2022-04-02 13:17:33 +01:00
uint32_t str_pos = 11;
for (uint32_t i = 12; i < sizeof(data); i++) {
if ('\n' == data[i]) {
data[str_pos] = i - str_pos -1; // Replace lf with string length
str_pos = i;
}
}
}
ImprovWriteData(data, sizeof(data));
}
void ImprovSendSetting(uint32_t command) {
char data[100];
uint32_t len = 0;
#ifdef USE_WEBSERVER
2022-04-02 13:17:33 +01:00
len = ext_snprintf_P(data, sizeof(data), PSTR("01\nhttp://%_I:%d\n"), (uint32_t)WiFi.localIP(), WEB_PORT);
len -= 3;
#endif // USE_WEBSERVER
data[0] = command;
ImprovSendResponse((uint8_t*)data, len +3);
}
2022-04-03 13:48:30 +01:00
void ImprovReceived(void) {
uint32_t command = TasmotaGlobal.serial_in_buffer[9];
switch (command) {
case IMPROV_WIFI_SETTINGS: { // 0x01
// if (RtcSettings.improv_state != IMPROV_STATE_AUTHORIZED) {
// ImprovSendError(IMPROV_ERROR_NOT_AUTHORIZED); // 0x04
// } else {
// 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
// I M P R O V vs ty le co dl sl s s i d pl p a s s w o r d cr
uint32_t ssid_length = TasmotaGlobal.serial_in_buffer[11];
uint32_t ssid_end = 12 + ssid_length;
uint32_t pass_length = TasmotaGlobal.serial_in_buffer[ssid_end];
uint32_t pass_start = ssid_end + 1;
uint32_t pass_end = pass_start + pass_length;
TasmotaGlobal.serial_in_buffer[ssid_end] = '\0';
char* ssid = &TasmotaGlobal.serial_in_buffer[12];
TasmotaGlobal.serial_in_buffer[pass_end] = '\0';
char* password = &TasmotaGlobal.serial_in_buffer[pass_start];
#ifdef IMPROV_DEBUG
2022-04-03 13:48:30 +01:00
AddLog(LOG_LEVEL_DEBUG, PSTR("IMP: Ssid '%s', Password '%s'"), ssid, password);
#endif // IMPROV_DEBUG
2022-04-03 13:48:30 +01:00
Improv.wifi_timeout = IMPROV_WIFI_TIMEOUT; // Set WiFi connect timeout
ImprovSendState(IMPROV_STATE_PROVISIONING);
Settings->flag4.network_wifi = 1; // Enable WiFi
char cmnd[TOPSZ];
snprintf_P(cmnd, sizeof(cmnd), PSTR(D_CMND_BACKLOG "0 " D_CMND_SSID "1 %s;" D_CMND_PASSWORD "1 %s"), ssid, password);
ExecuteCommand(cmnd, SRC_SERIAL); // Set SSID and Password and restart
// }
break;
}
case IMPROV_GET_CURRENT_STATE: { // 0x02
ImprovSendState(RtcSettings.improv_state);
if (IMPROV_STATE_PROVISIONED == RtcSettings.improv_state) {
ImprovSendSetting(command);
}
2022-04-03 13:48:30 +01:00
break;
}
case IMPROV_GET_DEVICE_INFO: { // 0x03
char data[200];
uint32_t len = snprintf_P(data, sizeof(data), PSTR("01\nTasmota\n%s\n%s\n%s\n"),
TasmotaGlobal.version, GetDeviceHardware().c_str(), SettingsText(SET_DEVICENAME));
data[0] = command;
ImprovSendResponse((uint8_t*)data, len);
break;
}
case IMPROV_GET_WIFI_NETWORKS: { // 0x04
char data[200];
int n = WiFi.scanNetworks(false, false); // Wait for scan result, hide hidden
if (n) {
int indices[n];
// Sort RSSI - strongest first
for (uint32_t i = 0; i < n; i++) { indices[i] = i; }
for (uint32_t i = 0; i < n; i++) {
for (uint32_t j = i + 1; j < n; j++) {
if (WiFi.RSSI(indices[j]) > WiFi.RSSI(indices[i])) {
std::swap(indices[i], indices[j]);
}
}
2022-04-03 13:48:30 +01:00
}
// Remove duplicate SSIDs - IMPROV does not distinguish between channels so no need to keep them
for (uint32_t i = 0; i < n; i++) {
if (-1 == indices[i]) { continue; }
String cssid = WiFi.SSID(indices[i]);
for (uint32_t j = i + 1; j < n; j++) {
if (cssid == WiFi.SSID(indices[j])) {
indices[j] = -1; // Set dup aps to index -1
}
}
}
2022-04-03 13:48:30 +01:00
// Send networks
for (uint32_t i = 0; i < n; i++) {
if (-1 == indices[i]) { continue; } // Skip dups
String ssid_copy = WiFi.SSID(indices[i]);
if (!ssid_copy.length()) { ssid_copy = F("no_name"); }
int32_t rssi = WiFi.RSSI(indices[i]);
bool encryption = (ENC_TYPE_NONE == WiFi.encryptionType(indices[i]));
// Send each ssid separately to avoid overflowing the buffer
uint32_t len = snprintf_P(data, sizeof(data), PSTR("01\n%s\n%d\n%s\n"),
ssid_copy.c_str(), rssi, (encryption)?"NO":"YES");
data[0] = command;
ImprovSendResponse((uint8_t*)data, len);
}
}
2022-04-03 13:48:30 +01:00
// Send empty response to signify the end of the list.
data[0] = command;
ImprovSendResponse((uint8_t*)data, 3); // Empty string
break;
}
/*
2022-04-03 13:48:30 +01:00
case IMPROV_BAD_CHECKSUM: { // 0xFF
break;
}
2022-04-03 13:48:30 +01:00
*/
default:
ImprovSendError(IMPROV_ERROR_UNKNOWN_RPC); // 0x02 - Unknown payload
}
}
/*********************************************************************************************/
bool ImprovSerialInput(void) {
// Check if received data is IMPROV data
if (6 == TasmotaGlobal.serial_in_byte_counter) {
TasmotaGlobal.serial_in_buffer[TasmotaGlobal.serial_in_byte_counter] = 0;
if (!strcmp_P(TasmotaGlobal.serial_in_buffer, PSTR("IMPROV"))) {
2022-04-03 13:48:30 +01:00
if (IMPROV_SERIAL_VERSION == TasmotaGlobal.serial_in_byte) {
Improv.seriallog_level = TasmotaGlobal.seriallog_level;
TasmotaGlobal.seriallog_level = 0; // Disable seriallogging interfering with IMPROV
Improv.version = IMPROV_SERIAL_VERSION;
}
}
}
2022-04-03 13:48:30 +01:00
if (IMPROV_SERIAL_VERSION == Improv.version) {
TasmotaGlobal.serial_in_buffer[TasmotaGlobal.serial_in_byte_counter++] = TasmotaGlobal.serial_in_byte;
// 0 1 2 3 4 5 6 7 8 9 10 11 8 + le +1
// I M P R O V ve ty le co pl data ... \n
// 49 4D 50 52 4F 56 01 03 xx yy zz ........ 0A
if (TasmotaGlobal.serial_in_byte_counter > 8) { // Wait for length
uint32_t data_len = TasmotaGlobal.serial_in_buffer[8];
if (TasmotaGlobal.serial_in_byte_counter > 10 + data_len) { // Receive including '\n'
AddLog(LOG_LEVEL_DEBUG_MORE, PSTR("IMP: Rcvd '%*_H'"), TasmotaGlobal.serial_in_byte_counter, TasmotaGlobal.serial_in_buffer);
uint32_t checksum_pos = TasmotaGlobal.serial_in_byte_counter -2;
uint8_t checksum = 0x00;
for (uint32_t i = 0; i < checksum_pos; i++) {
checksum += TasmotaGlobal.serial_in_buffer[i];
}
if (checksum != TasmotaGlobal.serial_in_buffer[checksum_pos]) {
ImprovSendError(IMPROV_ERROR_INVALID_RPC); // 0x01 - CRC error
}
else if (IMPROV_TYPE_RPC == TasmotaGlobal.serial_in_buffer[7]) {
uint32_t data_length = TasmotaGlobal.serial_in_buffer[10];
if (data_length == data_len - 2) {
ImprovReceived();
}
}
Improv.version = 0; // Done
TasmotaGlobal.seriallog_level = Improv.seriallog_level; // Restore seriallogging
return true;
}
}
2022-04-03 13:48:30 +01:00
TasmotaGlobal.serial_in_byte = 0;
}
return false;
}
void ImprovEverySecond(void) {
if (Improv.wifi_timeout) {
Improv.wifi_timeout--;
2022-04-03 13:48:30 +01:00
if (Improv.wifi_timeout < IMPROV_WIFI_TIMEOUT -3) { // Tasmota restarts after ssid or password change
2022-04-03 15:45:08 +01:00
if (WifiHasIP()) {
Improv.wifi_timeout = 0;
if (IMPROV_STATE_AUTHORIZED == RtcSettings.improv_state) {
RtcSettings.improv_state = IMPROV_STATE_PROVISIONED;
}
if (IMPROV_STATE_PROVISIONING == RtcSettings.improv_state) {
ImprovSendState(IMPROV_STATE_PROVISIONED);
ImprovSendSetting(IMPROV_WIFI_SETTINGS);
}
return;
}
}
if (!Improv.wifi_timeout) {
if (IMPROV_STATE_PROVISIONING == RtcSettings.improv_state) {
ImprovSendError(IMPROV_ERROR_UNABLE_TO_CONNECT); // 0x03 - WiFi connect timeout
ImprovSendState(IMPROV_STATE_AUTHORIZED);
}
}
}
}
void ImprovInit(void) {
if (!RtcSettings.improv_state) {
RtcSettings.improv_state = IMPROV_STATE_AUTHORIZED; // Power on state (persistent during restarts)
}
Improv.wifi_timeout = IMPROV_WIFI_TIMEOUT; // Try to update state after restart
#ifdef IMPROV_DEBUG
AddLog(LOG_LEVEL_DEBUG, PSTR("IMP: State %d"), RtcSettings.improv_state);
#endif // IMPROV_DEBUG
}
/*********************************************************************************************\
* Interface
\*********************************************************************************************/
bool Xdrv62(uint8_t function) {
bool result = false;
switch (function) {
case FUNC_EVERY_SECOND:
ImprovEverySecond();
break;
case FUNC_SERIAL:
result = ImprovSerialInput();
break;
case FUNC_PRE_INIT:
ImprovInit();
break;
}
return result;
}
#endif // USE_IMPROV