forked from plapointe6/EspMQTTClient
-
Notifications
You must be signed in to change notification settings - Fork 0
/
ESP32HTTPUpdateServer.h
103 lines (86 loc) · 3.11 KB
/
ESP32HTTPUpdateServer.h
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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#ifndef ESP32_HTTP_UPDATE_SERVER_H
#define ESP32_HTTP_UPDATE_SERVER_H
/*
Based on the HTTP update exemple of ESP32 core
*/
#include <WebServer.h>
#include <Update.h>
#define ESP32_WEB_UPDATE_HTML "<html><body><form method='POST' action='' enctype='multipart/form-data'><input type='file' name='update'><input type='submit' value='Update'></form></body></html>"
#define ESP32_WEB_UPDATE_SUCCESS_RESPONSE "<META http-equiv=\"refresh\" content=\"10;URL=/\">Update Success! Rebooting...\n"
class ESP32HTTPUpdateServer
{
private:
WebServer* _server;
String _username;
String _password;
bool _serialDebugging;
public:
ESP32HTTPUpdateServer(bool serialDebugging = false)
{
_server = NULL;
_username = "";
_password = "";
}
void setup(WebServer* server, const char* path = "/", const char* username = "", const char* password = "")
{
_server = server;
_username = username;
_password = password;
// Get of the index handling
_server->on(path, HTTP_GET, [&]() {
// Force authentication if a user and password are defined
if (_username.length() > 0 && _password.length() > 0 && !_server->authenticate(_username.c_str(), _password.c_str()))
return _server->requestAuthentication();
_server->sendHeader("Connection", "close");
_server->send(200, "text/html", ESP32_WEB_UPDATE_HTML);
});
// Post of the file handling
_server->on(path, HTTP_POST, [&]() {
_server->client().setNoDelay(true);
_server->send_P(200, "text/html", (Update.hasError()) ? "FAIL" : ESP32_WEB_UPDATE_SUCCESS_RESPONSE);
delay(100);
_server->client().stop();
ESP.restart();
}, [&]() {
HTTPUpload& upload = _server->upload();
if (upload.status == UPLOAD_FILE_START)
{
// Check if we are authenticated
if (!(_username.length() == 0 || _password.length() == 0 || _server->authenticate(_username.c_str(), _password.c_str())))
{
if (_serialDebugging)
Serial.printf("Unauthenticated Update\n");
return;
}
// Debugging message for upload start
if (_serialDebugging)
{
Serial.setDebugOutput(true);
Serial.printf("Update: %s\n", upload.filename.c_str());
}
// Starting update
bool error = Update.begin(UPDATE_SIZE_UNKNOWN);
if (_serialDebugging && error)
Update.printError(Serial);
}
else if (upload.status == UPLOAD_FILE_WRITE)
{
if (Update.write(upload.buf, upload.currentSize) != upload.currentSize && _serialDebugging)
Update.printError(Serial);
}
else if (upload.status == UPLOAD_FILE_END)
{
if (Update.end(true) && _serialDebugging)
Serial.printf("Update Success: %u\nRebooting...\n", upload.totalSize);
else if(_serialDebugging)
Update.printError(Serial);
if(_serialDebugging)
Serial.setDebugOutput(false);
}
else if(_serialDebugging)
Serial.printf("Update Failed Unexpectedly (likely broken connection): status=%d\n", upload.status);
});
_server->begin();
}
};
#endif