otaserver/otaserver.cpp at master · kublet/otaserver · GitHub

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
#pragma once

#include "otaserver.h"

WebServer server(80);


/***************************************************************************************
** Function name:           init
** Description:             Initialize OTA server and endpoint
***************************************************************************************/
void OTAServer::init() {
if (!MDNS.begin("esp32")) { //http://esp32.local
    Serial.println("Error setting up MDNS responder!");
    while (1) {
      delay(1000);
    }
  }
  Serial.println("mDNS responder started");

  /*handling uploading firmware file */
  server.on("/update", HTTP_POST, []() {
    server.sendHeader("Connection", "close");
    server.send(200, "text/plain", (Update.hasError()) ? "FAIL" : "OK");
    ESP.restart();
  }, []() {
    HTTPUpload& upload = server.upload();
    if (upload.status == UPLOAD_FILE_START) {
      Serial.printf("Update: %s\n", upload.filename.c_str());
      if (!Update.begin(UPDATE_SIZE_UNKNOWN)) { //start with max available size
        Update.printError(Serial);
      }
    } else if (upload.status == UPLOAD_FILE_WRITE) {
      /* flashing firmware to ESP*/
      if (Update.write(upload.buf, upload.currentSize) != upload.currentSize) {
        Update.printError(Serial);
      }
    } else if (upload.status == UPLOAD_FILE_END) {
      if (Update.end(true)) { //true to set the size to the current progress
        Serial.printf("Update Success: %u\nRebooting...\n", upload.totalSize);
      } else {
        Update.printError(Serial);
      }
    }
  });
}

/***************************************************************************************
** Function name:           start
** Description:             Start OTA server
***************************************************************************************/
void OTAServer::start() {
  server.begin();
}

/***************************************************************************************
** Function name:           run
** Description:             Initialize and start OTA server
***************************************************************************************/
void OTAServer::run() {
  init();
  start();
}

/***************************************************************************************
** Function name:           handle
** Description:             Handle incoming connections from client sending OTA firmware
***************************************************************************************/
void OTAServer::handle() {
  server.handleClient();
}

/***************************************************************************************
** Function name:           stop
** Description:             Stop OTA server
***************************************************************************************/
void OTAServer::stop() {
  server.stop();
}

/***************************************************************************************
** Function name:           connectWifi
** Description:             Connects to WiFi
***************************************************************************************/
void OTAServer::connectWifi() {
  preferences.begin("core", true);
  std::string ssid = preferences.getString("ssid").c_str();
  std::string pw = preferences.getString("pw").c_str();
  preferences.end();
  WiFi.mode(WIFI_STA);
  WiFi.begin(ssid.c_str(), pw.c_str());
  Serial.print("Connected to ");
  Serial.println(ssid.c_str());
  while((WiFi.status() != WL_CONNECTED)) {
    delay(500);
    Serial.print(".");
  }
  Serial.println(WiFi.localIP());
}