463 lines
14 KiB
C++
463 lines
14 KiB
C++
/*
|
|
* This file is part of the "bluetoothheater" distribution
|
|
* (https://gitlab.com/mrjones.id.au/bluetoothheater)
|
|
*
|
|
* Copyright (C) 2018 Ray Jones <ray@mrjones.id.au>
|
|
* Copyright (C) 2018 James Clark
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
|
|
#include "BTCWebServer.h"
|
|
#include "../Utility/DebugPort.h"
|
|
#include "../Protocol/TxManage.h"
|
|
#include "../Protocol/helpers.h"
|
|
#include "../cfg/pins.h"
|
|
#include "../cfg/BTCConfig.h"
|
|
#include "Index.h"
|
|
#include "../Utility/BTC_JSON.h"
|
|
#include "../Utility/Moderator.h"
|
|
#include <WiFiManager.h>
|
|
#if USE_SPIFFS == 1
|
|
#include <FS.H>
|
|
#include <SPIFFS.h>
|
|
#endif
|
|
#include "../Utility/NVStorage.h"
|
|
|
|
extern void ShowOTAScreen(int percent=0, bool webpdate=false);
|
|
|
|
extern WiFiManager wm;
|
|
|
|
File fsUploadFile; // a File object to temporarily store the received file
|
|
int SPIFFSupload = 0;
|
|
|
|
WebServer server(80);
|
|
WebSocketsServer webSocket = WebSocketsServer(81);
|
|
|
|
bool bRxWebData = false; // flags for OLED animation
|
|
bool bTxWebData = false;
|
|
bool bUpdateAccessed = false; // flag used to ensure web update always starts via /update. direct accesses to /updatenow will FAIL
|
|
|
|
const int led = 13;
|
|
|
|
#if USE_SPIFFS == 1
|
|
|
|
String getContentType(String filename) { // convert the file extension to the MIME type
|
|
if (filename.endsWith(".html")) return "text/html";
|
|
else if (filename.endsWith(".css")) return "text/css";
|
|
else if (filename.endsWith(".js")) return "application/javascript";
|
|
else if (filename.endsWith(".ico")) return "image/x-icon";
|
|
else if (filename.endsWith(".bin")) return "application/octet-stream";
|
|
return "text/plain";
|
|
}
|
|
|
|
bool handleFileRead(String path) { // send the right file to the client (if it exists)
|
|
DebugPort.println("handleFileRead: " + path);
|
|
if (path.endsWith("/")) path += "index.html"; // If a folder is requested, send the index file
|
|
String contentType = getContentType(path); // Get the MIME type
|
|
if (SPIFFS.exists(path)) { // If the file exists
|
|
File file = SPIFFS.open(path, "r"); // Open it
|
|
size_t sent = server.streamFile(file, contentType); // And send it to the client
|
|
file.close(); // Then close the file again
|
|
return true;
|
|
}
|
|
DebugPort.println("\tFile Not Found");
|
|
return false; // If the file doesn't exist, return false
|
|
}
|
|
|
|
/*void handleFavIcon() {
|
|
handleFileRead("/favicon.ico");
|
|
}*/
|
|
|
|
void handleBTCRoot() {
|
|
handleFileRead("/index.html");
|
|
}
|
|
#else
|
|
void handleBTCRoot() {
|
|
String s = MAIN_PAGE; //Read HTML contents
|
|
server.send(200, "text/html", s); //Send web page
|
|
}
|
|
#endif
|
|
|
|
void handleWMConfig() {
|
|
server.send(200, "text/plain", "Start Config Portal - Retaining credential");
|
|
DebugPort.println("Starting web portal for wifi config");
|
|
delay(500);
|
|
// wm.startWebPortal();
|
|
wifiEnterConfigPortal(true, false, 3000);
|
|
}
|
|
|
|
void handleReset() {
|
|
server.send(200, "text/plain", "Start Config Portal - Resetting Wifi credentials!");
|
|
DebugPort.println("diconnecting client and wifi, then rebooting");
|
|
delay(500);
|
|
wifiEnterConfigPortal(true, true, 3000);
|
|
}
|
|
|
|
void handleFormat() {
|
|
server.send(200, "text/plain", "Formatting SPIFFS partition!");
|
|
DebugPort.println("Formatting SPIFFS partition");
|
|
delay(500);
|
|
SPIFFS.format();
|
|
}
|
|
|
|
void handleBTCNotFound() {
|
|
digitalWrite(led, 1);
|
|
String message = "File Not Found\n\n";
|
|
message += "URI: ";
|
|
message += server.uri();
|
|
message += "\nMethod: ";
|
|
message += (server.method() == HTTP_GET) ? "GET" : "POST";
|
|
message += "\nArguments: ";
|
|
message += server.args();
|
|
message += "\n";
|
|
for (uint8_t i = 0; i < server.args(); i++) {
|
|
message += " " + server.argName(i) + ": " + server.arg(i) + "\n";
|
|
}
|
|
server.send(404, "text/plain", message);
|
|
digitalWrite(led, 0);
|
|
}
|
|
|
|
const char* serverIndex = R"=====(
|
|
<!DOCTYPE html>
|
|
<html lang="en">
|
|
<head>
|
|
<meta charset="utf-8"/>
|
|
<script>
|
|
// global variables
|
|
var sendSize;
|
|
var Socket;
|
|
|
|
function _(el) {
|
|
return document.getElementById(el);
|
|
}
|
|
function init() {
|
|
Socket = new WebSocket('ws://' + window.location.hostname + ':81/');
|
|
|
|
Socket.onmessage = function(event){
|
|
var response = JSON.parse(event.data);
|
|
var key;
|
|
for(key in response) {
|
|
console.log("JSON decode:", key, response[key]);
|
|
switch(key) {
|
|
case "progress":
|
|
// actual data bytes received as fed back via web socket
|
|
var bytes = response[key];
|
|
_("loaded_n_total").innerHTML = "Uploaded " + bytes + " bytes of " + sendSize;
|
|
var percent = Math.round( 100 * (bytes / sendSize));
|
|
_("progressBar").value = percent;
|
|
_("status").innerHTML = percent+"% uploaded.. please wait";
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
function uploadFile() {
|
|
_("cancel").hidden = true;
|
|
var file = _("file1").files[0];
|
|
sendSize = file.size;
|
|
var formdata = new FormData();
|
|
formdata.append("update", file);
|
|
var ajax = new XMLHttpRequest();
|
|
// progress is handled via websocket JSON sent from controller
|
|
// using server side progress only shows the buffer filling, not actual delivery.
|
|
ajax.addEventListener("load", completeHandler, false);
|
|
ajax.addEventListener("error", errorHandler, false);
|
|
ajax.addEventListener("abort", abortHandler, false);
|
|
ajax.open("POST", "/updatenow");
|
|
ajax.send(formdata);
|
|
}
|
|
function completeHandler(event) {
|
|
console.log(event);
|
|
_("status").innerHTML = event.target.responseText;
|
|
_("progressBar").value = 0;
|
|
_("loaded_n_total").innerHTML = "Uploaded " + sendSize + " bytes of " + sendSize;
|
|
setTimeout(function () {
|
|
window.location.assign("/");
|
|
}, 5000);
|
|
}
|
|
function errorHandler(event) {
|
|
console.log(event);
|
|
_("status").innerHTML = "Upload Failed";
|
|
}
|
|
function abortHandler(event) {
|
|
console.log(event);
|
|
_("status").innerHTML = "Upload Aborted";
|
|
}
|
|
</script>
|
|
<style>
|
|
body {font-family: Arial, Helvetica, sans-serif;}
|
|
</style>
|
|
<title>Afterburner firmware update</title>
|
|
</head>
|
|
<body onload="javascript:init()">
|
|
<h1>Afterburner firmware update</h1>
|
|
<form id='upload_form' method='POST' enctype='multipart/form-data'>
|
|
<input type='file' name='file1' id='file1'> <BR>
|
|
<input type='button' value='Update' onclick="uploadFile()">
|
|
<progress id="progressBar" value="0" max="100" style="width:300px;"></progress><BR>
|
|
<h3 id="status"></h3>
|
|
<p id="loaded_n_total"></p>
|
|
<BR>
|
|
<input type='button' onclick=window.location.assign("/") value='Cancel' id="cancel">
|
|
</form>
|
|
</body>
|
|
</html>
|
|
)=====";
|
|
|
|
const char* rootIndex = R"=====(
|
|
<!DOCTYPE html>
|
|
<script>
|
|
function init() {
|
|
window.location.assign("/");
|
|
}
|
|
</script
|
|
<html>
|
|
<head>
|
|
<title>HTML Meta Tag</title>
|
|
</head>
|
|
<body onload="javascript:init()">
|
|
<p>Root redirect</p>
|
|
</body>
|
|
</html>
|
|
)=====";
|
|
|
|
void rootRedirect()
|
|
{
|
|
// server.sendHeader("Connection", "close");
|
|
server.send(200, "text/html", rootIndex);
|
|
}
|
|
|
|
|
|
void initWebServer(void) {
|
|
|
|
Update
|
|
.onProgress([](unsigned int progress, unsigned int total) {
|
|
int percent = (progress / (total / 100));
|
|
DebugPort.printf("Progress: %u%%\r", percent);
|
|
DebugPort.handle(); // keep telnet spy alive
|
|
ShowOTAScreen(percent, true);
|
|
DebugPort.print("^");
|
|
});
|
|
|
|
if (MDNS.begin("Afterburner")) {
|
|
DebugPort.println("MDNS responder started");
|
|
}
|
|
|
|
// server.on("/", handleBTCRoot);
|
|
|
|
server.on("/wmconfig", handleWMConfig);
|
|
server.on("/resetwifi", handleReset);
|
|
server.on("/formatspiffs", handleFormat);
|
|
|
|
server.on("/tst", HTTP_GET, []() {
|
|
server.sendHeader("Location","/"); // reselect the update page
|
|
server.send(303);
|
|
// rootRedirect();
|
|
});
|
|
|
|
// Magical code originally shamelessly lifted from Arduino WebUpdate example, then modified
|
|
// This allows pushing new firmware to the ESP from a WEB BROWSER!
|
|
// Added authentication and a sequencing flag to ensure this is not bypassed
|
|
//
|
|
// Initial launch page
|
|
server.on("/update", HTTP_GET, []() {
|
|
sCredentials creds = NVstore.getCredentials();
|
|
if (!server.authenticate(creds.webUpdateUsername, creds.webUpdatePassword)) {
|
|
return server.requestAuthentication();
|
|
}
|
|
bUpdateAccessed = true;
|
|
server.sendHeader("Connection", "close");
|
|
server.sendHeader("Cache-Control", "no-cache");
|
|
server.send(200, "text/html", serverIndex);
|
|
// handleFileRead("/uploadfirmware.html");
|
|
});
|
|
server.on("/updatenow", HTTP_GET, []() { // handle attempts to just browse the /updatenow path - force redirect to root
|
|
rootRedirect();
|
|
});
|
|
// actual guts that manages the new firmware upload
|
|
server.on("/updatenow", HTTP_POST, []() {
|
|
// completion functionality
|
|
if(SPIFFSupload) {
|
|
if(SPIFFSupload == 1) {
|
|
server.send(200, "OK");
|
|
server.sendHeader("Location","/update"); // reselect the update page
|
|
server.send(303);
|
|
}
|
|
else {
|
|
server.send(500, "text/plain", "500: couldn't create file");
|
|
}
|
|
SPIFFSupload = 0;
|
|
}
|
|
else {
|
|
if(Update.hasError())
|
|
server.send(200, "text/plain", "FAIL - Afterburner will reboot shortly");
|
|
else
|
|
server.send(200, "OK - Afterburner will reboot shortly");
|
|
delay(1000);
|
|
ESP.restart(); // reboot
|
|
}
|
|
}, []() {
|
|
if(bUpdateAccessed) { // only allow progression via /update, attempts to directly access /updatenow will fail
|
|
HTTPUpload& upload = server.upload();
|
|
if (upload.status == UPLOAD_FILE_START) {
|
|
String filename = upload.filename;
|
|
DebugPort.setDebugOutput(true);
|
|
if(filename.endsWith(".bin")) {
|
|
DebugPort.printf("Update: %s\r\n", filename.c_str());
|
|
if (!Update.begin()) { //start with max available size
|
|
Update.printError(DebugPort);
|
|
}
|
|
}
|
|
else {
|
|
if(!filename.startsWith("/")) filename = "/"+filename;
|
|
DebugPort.printf("handleFileUpload Name: %s\r\n", filename.c_str());
|
|
fsUploadFile = SPIFFS.open(filename, "w"); // Open the file for writing in SPIFFS (create if it doesn't exist)
|
|
SPIFFSupload = fsUploadFile ? 1 : 2;
|
|
//filename = String();
|
|
}
|
|
} else if (upload.status == UPLOAD_FILE_WRITE) {
|
|
// handle upload segments
|
|
#if USE_SW_WATCHDOG == 1
|
|
feedWatchdog(); // we get stuck here for a while, don't let the watchdog bite!
|
|
#endif
|
|
if(upload.totalSize) {
|
|
char JSON[64];
|
|
sprintf(JSON, "{\"progress\":%d}", upload.totalSize);
|
|
sendWebServerString(JSON); // feedback proper byte count of update
|
|
}
|
|
DebugPort.print(".");
|
|
if(fsUploadFile) {
|
|
fsUploadFile.write(upload.buf, upload.currentSize); // Write the received bytes to the file
|
|
}
|
|
else {
|
|
if (Update.write(upload.buf, upload.currentSize) != upload.currentSize) {
|
|
Update.printError(DebugPort);
|
|
}
|
|
}
|
|
} else if (upload.status == UPLOAD_FILE_END) {
|
|
// handle end of upload
|
|
if(SPIFFSupload) {
|
|
if(fsUploadFile) {
|
|
fsUploadFile.close(); // Close the file again
|
|
DebugPort.printf("handleFileUpload Size: %d\r\n", upload.totalSize);
|
|
}
|
|
}
|
|
else {
|
|
if (Update.end(true)) { //true to set the size to the current progress
|
|
DebugPort.printf("Update Success: %u\r\nRebooting...\r\n", upload.totalSize);
|
|
} else {
|
|
Update.printError(DebugPort);
|
|
}
|
|
}
|
|
DebugPort.setDebugOutput(false);
|
|
bUpdateAccessed = false;
|
|
} else {
|
|
DebugPort.printf("Update Failed Unexpectedly (likely broken connection): status=%d\r\n", upload.status);
|
|
bUpdateAccessed = false;
|
|
}
|
|
}
|
|
else {
|
|
// attempt to POST without using /update - forced redirect to root
|
|
rootRedirect();
|
|
}
|
|
});
|
|
|
|
#if USE_SPIFFS == 1
|
|
// NOTE: this serves the default home page, and favicon.ico
|
|
server.onNotFound([]()
|
|
{ // If the client requests any URI
|
|
if (!handleFileRead(server.uri())) // send it if it exists
|
|
server.send(404, "text/plain", "404: Not Found"); // otherwise, respond with a 404 (Not Found) error
|
|
}
|
|
);
|
|
#else
|
|
server.onNotFound(handleBTCNotFound);
|
|
#endif
|
|
|
|
server.begin();
|
|
webSocket.begin();
|
|
webSocket.onEvent(webSocketEvent);
|
|
DebugPort.println("HTTP server started");
|
|
|
|
}
|
|
|
|
unsigned char cVal;
|
|
|
|
// called my main sketch loop()
|
|
bool doWebServer(void) {
|
|
webSocket.loop();
|
|
server.handleClient();
|
|
}
|
|
|
|
bool isWebServerClientChange()
|
|
{
|
|
static int prevNumClients = -1;
|
|
|
|
int numClients = webSocket.connectedClients();
|
|
if(numClients != prevNumClients) {
|
|
prevNumClients = numClients;
|
|
DebugPort.println("Changed number of web clients, should reset JSON moderator");
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool sendWebServerString(const char* Str)
|
|
{
|
|
unsigned long tStart = millis();
|
|
if(webSocket.connectedClients()) {
|
|
unsigned long tCon = millis() - tStart;
|
|
tStart = millis();
|
|
bTxWebData = true; // OLED tx data animation flag
|
|
webSocket.broadcastTXT(Str);
|
|
unsigned long tWeb = millis() - tStart;
|
|
// DebugPort.printf("Websend times : %ld,%ld\r\n", tCon, tWeb);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
|
|
void webSocketEvent(uint8_t num, WStype_t type, uint8_t * payload, size_t length) {
|
|
if (type == WStype_TEXT) {
|
|
bRxWebData = true;
|
|
char cmd[256];
|
|
memset(cmd, 0, 256);
|
|
for (int i = 0; i < length && i < 256; i++) {
|
|
cmd[i] = payload[i];
|
|
}
|
|
interpretJsonCommand(cmd); // send to the main heater controller decode routine
|
|
}
|
|
}
|
|
|
|
bool hasWebClientSpoken(bool reset)
|
|
{
|
|
bool retval = bRxWebData;
|
|
if(reset)
|
|
bRxWebData = false;
|
|
return retval;
|
|
}
|
|
|
|
bool hasWebServerSpoken(bool reset)
|
|
{
|
|
bool retval = bTxWebData;
|
|
if(reset)
|
|
bTxWebData = false;
|
|
return retval;
|
|
}
|
|
|