update-impl.h

#include <Arduino.h>
#include <coredecls.h>
#include <WiFiClient.h>
#include <WiFiServer.h>
#include <ESP8266WebServer.h>
#include <WiFiUdp.h>
#include <flash_hal.h>
#include <FS.h>
#include "StreamString.h"
#include "update.h"
#include "page.h"

namespace esp8266httpupdateserver {
using namespace esp8266webserver;

static const char successResponse[] PROGMEM =
  "<META http-equiv=\"refresh\" content=\"15;URL=/\">Update Success! Rebooting...";

template<typename ServerType>
ESP8266HTTPUpdateServerTemplate<ServerType>::ESP8266HTTPUpdateServerTemplate(bool serial_debug) {
  _serial_output = serial_debug;
  _server = NULL;
  _username = emptyString;
  _password = emptyString;
  _authenticated = false;
}

template<typename ServerType>
void ESP8266HTTPUpdateServerTemplate<ServerType>::setup(ESP8266WebServerTemplate<ServerType>* server, const String& path, const String& username, const String& password) {
  _server = server;
  _username = username;
  _password = password;

  // handler for the /update form page
  _server->on(path.c_str(), HTTP_GET, [&]() {
    if (_username != emptyString && _password != emptyString && !_server->authenticate(_username.c_str(), _password.c_str()))
      return _server->requestAuthentication();
    _server->send_P(200, PSTR("text/html"), index_html);
  });

  // handler for the /update form page - preflight options
  _server->on(
    path.c_str(), HTTP_OPTIONS, [&]() {
      _server->sendHeader("Access-Control-Allow-Headers", "*");
      _server->sendHeader("Access-Control-Allow-Origin", "*");
      _server->send(200, F("text/html"), String(F("y")));
    },
    [&]() {
      _authenticated = (_username == emptyString || _password == emptyString || _server->authenticate(_username.c_str(), _password.c_str()));
      if (!_authenticated) {
        if (_serial_output)
          Serial.printf("Unauthenticated Update\n");
        return;
      }
    });

  // handler for the /update form POST (once file upload finishes)
  _server->on(
    path.c_str(), HTTP_POST, [&]() {
      _server->sendHeader("Access-Control-Allow-Headers", "*");
      _server->sendHeader("Access-Control-Allow-Origin", "*");
      if (!_authenticated)
        return _server->requestAuthentication();
      if (Update.hasError()) {
        _server->send(200, F("text/html"), String(F("Update error: ")) + _updaterError);
      } else {
        _server->client().setNoDelay(true);
        _server->send_P(200, PSTR("text/html"), successResponse);
        delay(100);
        _server->client().stop();
        ESP.restart();
      }
    },
    [&]() {
      // handler for the file upload, gets the sketch bytes, and writes
      // them through the Update object
      HTTPUpload& upload = _server->upload();

      if (upload.status == UPLOAD_FILE_START) {
        _updaterError.clear();
        if (_serial_output)
          Serial.setDebugOutput(true);

        _authenticated = (_username == emptyString || _password == emptyString || _server->authenticate(_username.c_str(), _password.c_str()));
        if (!_authenticated) {
          if (_serial_output)
            Serial.printf("Unauthenticated Update\n");
          return;
        }

        if (_serial_output)
          Serial.printf("Update: %s\n", upload.filename.c_str());
        if (upload.name == "filesystem") {
          size_t fsSize = ((size_t)FS_end - (size_t)FS_start);
          close_all_fs();
          if (!Update.begin(fsSize, U_FS)) {  //start with max available size
            if (_serial_output) Update.printError(Serial);
          }
        } else {
          uint32_t maxSketchSpace = (ESP.getFreeSketchSpace() - 0x1000) & 0xFFFFF000;
          if (!Update.begin(maxSketchSpace, U_FLASH)) {  //start with max available size
            _setUpdaterError();
          }
        }
      } else if (_authenticated && upload.status == UPLOAD_FILE_WRITE && !_updaterError.length()) {
        if (_serial_output) Serial.printf(".");
        if (Update.write(upload.buf, upload.currentSize) != upload.currentSize) {
          _setUpdaterError();
        }
      } else if (_authenticated && upload.status == UPLOAD_FILE_END && !_updaterError.length()) {
        if (Update.end(true)) {  //true to set the size to the current progress
          if (_serial_output) Serial.printf("Update Success: %zu\nRebooting...\n", upload.totalSize);
        } else {
          _setUpdaterError();
        }
        if (_serial_output) Serial.setDebugOutput(false);
      } else if (_authenticated && upload.status == UPLOAD_FILE_ABORTED) {
        Update.end();
        if (_serial_output) Serial.println("Update was aborted");
      }
      esp_yield();
    });
}

template<typename ServerType>
void ESP8266HTTPUpdateServerTemplate<ServerType>::_setUpdaterError() {
  if (_serial_output) Update.printError(Serial);
  StreamString str;
  Update.printError(str);
  _updaterError = str.c_str();
}

};