Fork of the espurna firmware for `mhsw` switches
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

259 lines
6.8 KiB

7 years ago
  1. /*
  2. OTA MODULE
  3. Copyright (C) 2016-2018 by Xose Pérez <xose dot perez at gmail dot com>
  4. */
  5. #include "ArduinoOTA.h"
  6. // -----------------------------------------------------------------------------
  7. // Arduino OTA
  8. // -----------------------------------------------------------------------------
  9. void _otaConfigure() {
  10. ArduinoOTA.setPort(OTA_PORT);
  11. ArduinoOTA.setHostname(getSetting("hostname").c_str());
  12. #if USE_PASSWORD
  13. ArduinoOTA.setPassword(getSetting("adminPass", ADMIN_PASS).c_str());
  14. #endif
  15. }
  16. void _otaLoop() {
  17. ArduinoOTA.handle();
  18. }
  19. // -----------------------------------------------------------------------------
  20. // Terminal OTA
  21. // -----------------------------------------------------------------------------
  22. #if TERMINAL_SUPPORT
  23. #include <ESPAsyncTCP.h>
  24. AsyncClient * _ota_client;
  25. char * _ota_host;
  26. char * _ota_url;
  27. unsigned int _ota_port = 80;
  28. unsigned long _ota_size = 0;
  29. const char OTA_REQUEST_TEMPLATE[] PROGMEM =
  30. "GET %s HTTP/1.1\r\n"
  31. "Host: %s\r\n"
  32. "User-Agent: ESPurna\r\n"
  33. "Connection: close\r\n"
  34. "Content-Type: application/x-www-form-urlencoded\r\n"
  35. "Content-Length: 0\r\n\r\n\r\n";
  36. void _otaFrom(const char * host, unsigned int port, const char * url) {
  37. if (_ota_host) free(_ota_host);
  38. if (_ota_url) free(_ota_url);
  39. _ota_host = strdup(host);
  40. _ota_url = strdup(url);
  41. _ota_port = port;
  42. _ota_size = 0;
  43. if (_ota_client == NULL) {
  44. _ota_client = new AsyncClient();
  45. }
  46. _ota_client->onDisconnect([](void *s, AsyncClient *c) {
  47. DEBUG_MSG_P(PSTR("\n"));
  48. if (Update.end(true)){
  49. DEBUG_MSG_P(PSTR("[OTA] Success: %u bytes\n"), _ota_size);
  50. deferredReset(100, CUSTOM_RESET_OTA);
  51. } else {
  52. #ifdef DEBUG_PORT
  53. Update.printError(DEBUG_PORT);
  54. #endif
  55. eepromRotate(true);
  56. }
  57. DEBUG_MSG_P(PSTR("[OTA] Disconnected\n"));
  58. _ota_client->free();
  59. delete _ota_client;
  60. _ota_client = NULL;
  61. free(_ota_host);
  62. _ota_host = NULL;
  63. free(_ota_url);
  64. _ota_url = NULL;
  65. }, 0);
  66. _ota_client->onTimeout([](void *s, AsyncClient *c, uint32_t time) {
  67. _ota_client->close(true);
  68. }, 0);
  69. _ota_client->onData([](void * arg, AsyncClient * c, void * data, size_t len) {
  70. char * p = (char *) data;
  71. if (_ota_size == 0) {
  72. Update.runAsync(true);
  73. if (!Update.begin((ESP.getFreeSketchSpace() - 0x1000) & 0xFFFFF000)) {
  74. #ifdef DEBUG_PORT
  75. Update.printError(DEBUG_PORT);
  76. #endif
  77. }
  78. p = strstr((char *)data, "\r\n\r\n") + 4;
  79. len = len - (p - (char *) data);
  80. }
  81. if (!Update.hasError()) {
  82. if (Update.write((uint8_t *) p, len) != len) {
  83. #ifdef DEBUG_PORT
  84. Update.printError(DEBUG_PORT);
  85. #endif
  86. }
  87. }
  88. _ota_size += len;
  89. DEBUG_MSG_P(PSTR("[OTA] Progress: %u bytes\r"), _ota_size);
  90. }, NULL);
  91. _ota_client->onConnect([](void * arg, AsyncClient * client) {
  92. #if ASYNC_TCP_SSL_ENABLED
  93. if (443 == _ota_port) {
  94. uint8_t fp[20] = {0};
  95. sslFingerPrintArray(getSetting("otafp", OTA_GITHUB_FP).c_str(), fp);
  96. SSL * ssl = _ota_client->getSSL();
  97. if (ssl_match_fingerprint(ssl, fp) != SSL_OK) {
  98. DEBUG_MSG_P(PSTR("[OTA] Warning: certificate doesn't match\n"));
  99. }
  100. }
  101. #endif
  102. // Disabling EEPROM rotation to prevent writing to EEPROM after the upgrade
  103. eepromRotate(false);
  104. DEBUG_MSG_P(PSTR("[OTA] Downloading %s\n"), _ota_url);
  105. char buffer[strlen_P(OTA_REQUEST_TEMPLATE) + strlen(_ota_url) + strlen(_ota_host)];
  106. snprintf_P(buffer, sizeof(buffer), OTA_REQUEST_TEMPLATE, _ota_url, _ota_host);
  107. client->write(buffer);
  108. }, NULL);
  109. #if ASYNC_TCP_SSL_ENABLED
  110. bool connected = _ota_client->connect(host, port, 443 == port);
  111. #else
  112. bool connected = _ota_client->connect(host, port);
  113. #endif
  114. if (!connected) {
  115. DEBUG_MSG_P(PSTR("[OTA] Connection failed\n"));
  116. _ota_client->close(true);
  117. }
  118. }
  119. void _otaFrom(String url) {
  120. // Port from protocol
  121. unsigned int port = 80;
  122. if (url.startsWith("https://")) port = 443;
  123. url = url.substring(url.indexOf("/") + 2);
  124. // Get host
  125. String host = url.substring(0, url.indexOf("/"));
  126. // Explicit port
  127. int p = host.indexOf(":");
  128. if (p > 0) {
  129. port = host.substring(p + 1).toInt();
  130. host = host.substring(0, p);
  131. }
  132. // Get URL
  133. String uri = url.substring(url.indexOf("/"));
  134. _otaFrom(host.c_str(), port, uri.c_str());
  135. }
  136. void _otaInitCommands() {
  137. settingsRegisterCommand(F("OTA"), [](Embedis* e) {
  138. if (e->argc < 2) {
  139. DEBUG_MSG_P(PSTR("-ERROR: Wrong arguments\n"));
  140. } else {
  141. DEBUG_MSG_P(PSTR("+OK\n"));
  142. String url = String(e->argv[1]);
  143. _otaFrom(url);
  144. }
  145. });
  146. }
  147. #endif // TERMINAL_SUPPORT
  148. // -----------------------------------------------------------------------------
  149. void otaSetup() {
  150. _otaConfigure();
  151. #if WEB_SUPPORT
  152. wsOnAfterParseRegister(_otaConfigure);
  153. #endif
  154. #if TERMINAL_SUPPORT
  155. _otaInitCommands();
  156. #endif
  157. // Register loop
  158. espurnaRegisterLoop(_otaLoop);
  159. // -------------------------------------------------------------------------
  160. ArduinoOTA.onStart([]() {
  161. // Disabling EEPROM rotation to prevent writing to EEPROM after the upgrade
  162. eepromRotate(false);
  163. DEBUG_MSG_P(PSTR("[OTA] Start\n"));
  164. #if WEB_SUPPORT
  165. wsSend_P(PSTR("{\"message\": 2}"));
  166. #endif
  167. });
  168. ArduinoOTA.onEnd([]() {
  169. DEBUG_MSG_P(PSTR("\n"));
  170. DEBUG_MSG_P(PSTR("[OTA] Done, restarting...\n"));
  171. #if WEB_SUPPORT
  172. wsSend_P(PSTR("{\"action\": \"reload\"}"));
  173. #endif
  174. deferredReset(100, CUSTOM_RESET_OTA);
  175. });
  176. ArduinoOTA.onProgress([](unsigned int progress, unsigned int total) {
  177. DEBUG_MSG_P(PSTR("[OTA] Progress: %u%%\r"), (progress / (total / 100)));
  178. });
  179. ArduinoOTA.onError([](ota_error_t error) {
  180. #if DEBUG_SUPPORT
  181. DEBUG_MSG_P(PSTR("\n[OTA] Error #%u: "), error);
  182. if (error == OTA_AUTH_ERROR) DEBUG_MSG_P(PSTR("Auth Failed\n"));
  183. else if (error == OTA_BEGIN_ERROR) DEBUG_MSG_P(PSTR("Begin Failed\n"));
  184. else if (error == OTA_CONNECT_ERROR) DEBUG_MSG_P(PSTR("Connect Failed\n"));
  185. else if (error == OTA_RECEIVE_ERROR) DEBUG_MSG_P(PSTR("Receive Failed\n"));
  186. else if (error == OTA_END_ERROR) DEBUG_MSG_P(PSTR("End Failed\n"));
  187. #endif
  188. eepromRotate(true);
  189. });
  190. ArduinoOTA.begin();
  191. }