228 lines
8.1 KiB
C++
228 lines
8.1 KiB
C++
/****************************************************************************
|
|
* Tu May 22 21:23:51 2020
|
|
* Copyright 2020 Dirk Brosswick
|
|
* Email: dirk.brosswick@googlemail.com
|
|
****************************************************************************/
|
|
|
|
/*
|
|
* 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 2 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, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
*/
|
|
|
|
#include <WiFi.h>
|
|
#include <WiFiClient.h>
|
|
#include <Update.h>
|
|
#include <SPIFFS.h>
|
|
#include <AsyncTCP.h>
|
|
#include <ESPAsyncWebServer.h>
|
|
#include <SPIFFSEditor.h>
|
|
#include <ESP32SSDP.h>
|
|
|
|
#include "webserver.h"
|
|
#include "config.h"
|
|
#include "gui/screenshot.h"
|
|
|
|
AsyncWebServer asyncserver( WEBSERVERPORT );
|
|
TaskHandle_t _WEBSERVER_Task;
|
|
|
|
|
|
void handleUpdate( AsyncWebServerRequest *request, const String& filename, size_t index, uint8_t *data, size_t len, bool final) {
|
|
|
|
if (!index){
|
|
/*
|
|
* if filename includes spiffs, update the spiffs partition
|
|
*/
|
|
int cmd = (filename.indexOf("spiffs") > 0) ? U_SPIFFS : U_FLASH;
|
|
if (!Update.begin(UPDATE_SIZE_UNKNOWN, cmd)) {
|
|
Update.printError(Serial);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Write Data an type message if fail
|
|
*/
|
|
if (Update.write(data, len) != len) {
|
|
Update.printError(Serial);
|
|
}
|
|
|
|
/*
|
|
* After write Update restart
|
|
*/
|
|
if (final) {
|
|
AsyncWebServerResponse *response = request->beginResponse(302, "text/plain", "Please wait while the switch reboots");
|
|
response->addHeader("Refresh", "20");
|
|
response->addHeader("Location", "/");
|
|
request->send(response);
|
|
if (!Update.end(true)){
|
|
Update.printError(Serial);
|
|
} else {
|
|
Serial.println("Update complete");
|
|
Serial.flush();
|
|
ESP.restart();
|
|
}
|
|
}
|
|
}
|
|
|
|
/*
|
|
*
|
|
*/
|
|
void asyncwebserver_start(void){
|
|
|
|
asyncserver.on("/info", HTTP_GET, [](AsyncWebServerRequest *request) {
|
|
String message("Firmwarestand: " __DATE__ " " __TIME__ "\nGCC-Version: " __VERSION__ "\n");
|
|
message = message + "Heap: " + ESP.getFreeHeap() + " bytes used of " + ESP.getHeapSize() + " bytes total\nHeap low water mark: " + ESP.getMinFreeHeap() + " bytes available\nPsram: " + ESP.getFreePsram() + " bytes used of " + ESP.getPsramSize() + " bytes available\nCurrent battery voltage: " + TTGOClass::getWatch()->power->getBattVoltage() / 1000 + " Volts";
|
|
request->send(200, "text/plain", message.c_str());
|
|
});
|
|
|
|
asyncserver.on("/shot", HTTP_GET, [](AsyncWebServerRequest * request) {
|
|
request->send(200, "text/plain", "screen is taken\r\n" );
|
|
screenshot_take();
|
|
screenshot_save();
|
|
});
|
|
|
|
asyncserver.addHandler(new SPIFFSEditor(SPIFFS));
|
|
asyncserver.serveStatic("/", SPIFFS, "/").setDefaultFile("index.htm");
|
|
|
|
asyncserver.onNotFound([](AsyncWebServerRequest *request){
|
|
Serial.printf( "NOT_FOUND: ");
|
|
if(request->method() == HTTP_GET)
|
|
Serial.printf( "GET");
|
|
else if(request->method() == HTTP_POST)
|
|
Serial.printf( "POST");
|
|
else if(request->method() == HTTP_DELETE)
|
|
Serial.printf( "DELETE");
|
|
else if(request->method() == HTTP_PUT)
|
|
Serial.printf( "PUT");
|
|
else if(request->method() == HTTP_PATCH)
|
|
Serial.printf( "PATCH");
|
|
else if(request->method() == HTTP_HEAD)
|
|
Serial.printf( "HEAD");
|
|
else if(request->method() == HTTP_OPTIONS)
|
|
Serial.printf( "OPTIONS");
|
|
else
|
|
Serial.printf( "UNKNOWN");
|
|
Serial.printf( " http://%s%s\n", request->host().c_str(), request->url().c_str());
|
|
|
|
if(request->contentLength()){
|
|
Serial.printf( "_CONTENT_TYPE: %s\n", request->contentType().c_str());
|
|
Serial.printf( "_CONTENT_LENGTH: %u\n", request->contentLength());
|
|
}
|
|
|
|
int headers = request->headers();
|
|
int i;
|
|
for(i=0;i<headers;i++){
|
|
AsyncWebHeader* h = request->getHeader(i);
|
|
Serial.printf( "_HEADER[%s]: %s\n", h->name().c_str(), h->value().c_str());
|
|
}
|
|
|
|
int params = request->params();
|
|
for(i=0;i<params;i++){
|
|
AsyncWebParameter* p = request->getParam(i);
|
|
if(p->isFile()){
|
|
Serial.printf( "_FILE[%s]: %s, size: %u\n", p->name().c_str(), p->value().c_str(), p->size());
|
|
} else if(p->isPost()){
|
|
Serial.printf( "_POST[%s]: %s\n", p->name().c_str(), p->value().c_str());
|
|
} else {
|
|
Serial.printf( "_GET[%s]: %s\n", p->name().c_str(), p->value().c_str());
|
|
}
|
|
}
|
|
request->send(404);
|
|
});
|
|
|
|
asyncserver.onFileUpload([](AsyncWebServerRequest *request, const String& filename, size_t index, uint8_t *data, size_t len, bool final){
|
|
if(!index)
|
|
Serial.printf( "UploadStart: %s\n", filename.c_str());
|
|
Serial.printf("%s", (const char*)data);
|
|
if(final)
|
|
Serial.printf( "UploadEnd: %s (%u)\n", filename.c_str(), index+len);
|
|
});
|
|
|
|
asyncserver.onRequestBody([](AsyncWebServerRequest *request, uint8_t *data, size_t len, size_t index, size_t total){
|
|
if(!index) {
|
|
Serial.printf( "BodyStart: %u\n", total);
|
|
}
|
|
Serial.printf( "%s", (const char*)data);
|
|
if(index + len == total) {
|
|
Serial.printf( "BodyEnd: %u\n", total);
|
|
}
|
|
});
|
|
|
|
asyncserver.on("/reset", HTTP_GET, []( AsyncWebServerRequest * request ) {
|
|
request->send(200, "text/plain", "Reset\r\n" );
|
|
delay(3000);
|
|
ESP.restart();
|
|
});
|
|
|
|
asyncserver.on("/update", HTTP_POST,
|
|
[](AsyncWebServerRequest *request) {},
|
|
[](AsyncWebServerRequest *request, const String& filename, size_t index, uint8_t *data, size_t len, bool final) { handleUpdate(request, filename, index, data, len, final); }
|
|
);
|
|
|
|
asyncserver.on("/description.xml", HTTP_GET, [](AsyncWebServerRequest *request) {
|
|
byte mac[6];
|
|
WiFi.macAddress(mac);
|
|
char tmp[12 + 1];
|
|
snprintf(tmp, sizeof(tmp), "%02X%02X%02X%02X%02X%02X", mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
|
|
String MacStrPart = String(tmp).substring(6);
|
|
|
|
String xmltext = String("<?xml version=\"1.0\"?>\n") +
|
|
"<root xmlns=\"urn:schemas-upnp-org:device-1-0\">\n"
|
|
"<specVersion>\n"
|
|
"\t<major>1</major>\n"
|
|
"\t<minor>0</minor>\n"
|
|
"</specVersion>\n"
|
|
"<URLBase>http://" + WiFi.localIP().toString() + "/</URLBase>\n"
|
|
"<device>\n"
|
|
"\t<deviceType>upnp:rootdevice</deviceType>\n"
|
|
|
|
/*this is the icon name in Windows*/
|
|
/*"\t<friendlyName>" + WiFi.getHostname() + "</friendlyName>\n"*/
|
|
"\t<friendlyName>" + DEV_NAME + " " + MacStrPart + "</friendlyName>\n" /*because the hostename is 'Espressif' */
|
|
|
|
"\t<presentationURL>/</presentationURL>\n"
|
|
"\t<manufacturer>" + "Dirk Broßwick (sharandac)" + "</manufacturer>\n"
|
|
"\t<manufacturerURL>https://github.com/sharandac/My-TTGO-Watch</manufacturerURL>\n"
|
|
"\t<modelName>" + DEV_INFO + "</modelName>\n"
|
|
|
|
"\t<modelNumber>" + WiFi.getHostname() + "</modelNumber>\n"
|
|
"\t<modelURL>" +
|
|
"/" + "</modelURL>\n"
|
|
|
|
"\t<serialNumber>Build: " + __FIRMWARE__ + "</serialNumber>\n"
|
|
//The last six bytes of the UUID are the hardware address of the first Ethernet adapter in the system the UUID was generated on.
|
|
"\t<UDN>uuid:38323636-4558-4DDA-9188-CDA0E6" + MacStrPart + "</UDN>\n"
|
|
"</device>\n"
|
|
"</root>\r\n"
|
|
"\r\n";
|
|
|
|
request->send(200, "text/xml", xmltext);
|
|
});
|
|
|
|
//Upnp / SSDP presentation - Multicast - link to description.xml
|
|
SSDP.setSchemaURL("description.xml");
|
|
SSDP.setHTTPPort( UPNPPORT );
|
|
SSDP.setURL("/");
|
|
SSDP.setDeviceType("upnp:rootdevice");
|
|
SSDP.begin();
|
|
|
|
asyncserver.begin();
|
|
|
|
log_i("enable webserver and ssdp");
|
|
}
|
|
|
|
void asyncwebserver_end(void) {
|
|
SSDP.end();
|
|
asyncserver.end();
|
|
log_i("disable webserver and ssdp");
|
|
} |