├── examples ├── WiFi101_OTA │ ├── arduino_secrets.h │ └── WiFi101_OTA.ino └── WiFi101_SD_OTA │ ├── arduino_secrets.h │ └── WiFi101_SD_OTA.ino ├── .codespellrc ├── library.properties ├── .github ├── dependabot.yml └── workflows │ ├── spell-check.yml │ ├── check-arduino.yml │ ├── report-size-deltas.yml │ ├── compile-examples.yml │ └── sync-labels.yml ├── keywords.txt ├── docs ├── readme.md └── api.md ├── src ├── OTAStorage.h ├── SDStorage.h ├── InternalStorage.h ├── SerialFlashStorage.h ├── SDStorage.cpp ├── WiFi101OTA.h ├── SerialFlashStorage.cpp ├── InternalStorage.cpp └── WiFi101OTA.cpp └── README.adoc /examples/WiFi101_OTA/arduino_secrets.h: -------------------------------------------------------------------------------- 1 | #define SECRET_SSID "" 2 | #define SECRET_PASS "" 3 | 4 | -------------------------------------------------------------------------------- /examples/WiFi101_SD_OTA/arduino_secrets.h: -------------------------------------------------------------------------------- 1 | #define SECRET_SSID "" 2 | #define SECRET_PASS "" 3 | 4 | -------------------------------------------------------------------------------- /.codespellrc: -------------------------------------------------------------------------------- 1 | # See: https://github.com/codespell-project/codespell#using-a-config-file 2 | [codespell] 3 | # In the event of a false positive, add the problematic word, in all lowercase, to a comma-separated list here: 4 | ignore-words-list = , 5 | check-filenames = 6 | check-hidden = 7 | skip = ./.git 8 | -------------------------------------------------------------------------------- /library.properties: -------------------------------------------------------------------------------- 1 | name=WiFi101OTA 2 | version=1.1.0 3 | author=Arduino 4 | maintainer=Arduino 5 | sentence=Update sketches on your board over WiFi 6 | paragraph=Requires an Arduino SAMD board 7 | category=Other 8 | url=http://www.arduino.cc/en/Reference/WiFi101OTA 9 | architectures=samd 10 | depends=SerialFlash, WiFi101 11 | -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- 1 | # See: https://docs.github.com/en/github/administering-a-repository/configuration-options-for-dependency-updates#about-the-dependabotyml-file 2 | version: 2 3 | 4 | updates: 5 | # Configure check for outdated GitHub Actions actions in workflows. 6 | # See: https://docs.github.com/en/github/administering-a-repository/keeping-your-actions-up-to-date-with-dependabot 7 | - package-ecosystem: github-actions 8 | directory: / # Check the repository's workflows under /.github/workflows/ 9 | schedule: 10 | interval: daily 11 | -------------------------------------------------------------------------------- /keywords.txt: -------------------------------------------------------------------------------- 1 | ####################################### 2 | # Syntax Coloring Map For WiFi101OTA 3 | ####################################### 4 | 5 | ####################################### 6 | # Datatypes (KEYWORD1) 7 | ####################################### 8 | 9 | WiFi101OTA KEYWORD1 10 | WiFiOTA KEYWORD1 11 | 12 | InternalStorage KEYWORD1 13 | SDStorage KEYWORD1 14 | 15 | ####################################### 16 | # Methods and Functions (KEYWORD2) 17 | ####################################### 18 | 19 | begin KEYWORD2 20 | poll KEYWORD2 21 | 22 | ####################################### 23 | # Constants (LITERAL1) 24 | ####################################### 25 | -------------------------------------------------------------------------------- /.github/workflows/spell-check.yml: -------------------------------------------------------------------------------- 1 | name: Spell Check 2 | 3 | # See: https://docs.github.com/en/free-pro-team@latest/actions/reference/events-that-trigger-workflows 4 | on: 5 | push: 6 | pull_request: 7 | schedule: 8 | # Run every Tuesday at 8 AM UTC to catch new misspelling detections resulting from dictionary updates. 9 | - cron: "0 8 * * TUE" 10 | workflow_dispatch: 11 | repository_dispatch: 12 | 13 | jobs: 14 | spellcheck: 15 | runs-on: ubuntu-latest 16 | 17 | steps: 18 | - name: Checkout repository 19 | uses: actions/checkout@v4 20 | 21 | - name: Spell check 22 | uses: codespell-project/actions-codespell@master 23 | -------------------------------------------------------------------------------- /docs/readme.md: -------------------------------------------------------------------------------- 1 | # WiFi101 OTA library 2 | 3 | This library allows you to have utilize OTA features [MKR1000](https://store-usa.arduino.cc/products/arduino-mkr1000-wifi) and the [WiFi Shield 101](https://docs.arduino.cc/retired/shields/arduino-wifi-shield-101). The sketch to be loaded Over The Air can be stored on two media: 4 | 5 | - the internal flash; 6 | - an external SD card; 7 | 8 | In the first case the new sketch can't exceed the half of the total memory (256/2 = 128kB) while in the second case you have all the MCU's flash available. 9 | 10 | To use this library 11 | 12 | ``` 13 | #include 14 | #include 15 | #include 16 | #include 17 | #include 18 | ``` -------------------------------------------------------------------------------- /.github/workflows/check-arduino.yml: -------------------------------------------------------------------------------- 1 | name: Check Arduino 2 | 3 | # See: https://docs.github.com/en/free-pro-team@latest/actions/reference/events-that-trigger-workflows 4 | on: 5 | push: 6 | pull_request: 7 | schedule: 8 | # Run every Tuesday at 8 AM UTC to catch breakage caused by new rules added to Arduino Lint. 9 | - cron: "0 8 * * TUE" 10 | workflow_dispatch: 11 | repository_dispatch: 12 | 13 | jobs: 14 | lint: 15 | runs-on: ubuntu-latest 16 | 17 | steps: 18 | - name: Checkout repository 19 | uses: actions/checkout@v4 20 | 21 | - name: Arduino Lint 22 | uses: arduino/arduino-lint-action@v1 23 | with: 24 | compliance: specification 25 | library-manager: update 26 | # Always use this setting for official repositories. Remove for 3rd party projects. 27 | official: true 28 | project-type: library 29 | -------------------------------------------------------------------------------- /.github/workflows/report-size-deltas.yml: -------------------------------------------------------------------------------- 1 | name: Report Size Deltas 2 | 3 | # See: https://docs.github.com/en/free-pro-team@latest/actions/reference/events-that-trigger-workflows 4 | on: 5 | push: 6 | paths: 7 | - ".github/workflows/report-size-deltas.yml" 8 | schedule: 9 | # Run at the minimum interval allowed by GitHub Actions. 10 | # Note: GitHub Actions periodically has outages which result in workflow failures. 11 | # In this event, the workflows will start passing again once the service recovers. 12 | - cron: "*/5 * * * *" 13 | workflow_dispatch: 14 | repository_dispatch: 15 | 16 | jobs: 17 | report: 18 | runs-on: ubuntu-latest 19 | steps: 20 | - name: Comment size deltas reports to PRs 21 | uses: arduino/report-size-deltas@v1 22 | with: 23 | # Regex matching the names of the workflow artifacts created by the "Compile Examples" workflow 24 | sketches-reports-source: ^sketches-report-.+ 25 | -------------------------------------------------------------------------------- /src/OTAStorage.h: -------------------------------------------------------------------------------- 1 | /* 2 | Copyright (c) 2017 Arduino LLC. All right reserved. 3 | 4 | This library is free software; you can redistribute it and/or 5 | modify it under the terms of the GNU Lesser General Public 6 | License as published by the Free Software Foundation; either 7 | version 2.1 of the License, or (at your option) any later version. 8 | 9 | This library is distributed in the hope that it will be useful, 10 | but WITHOUT ANY WARRANTY; without even the implied warranty of 11 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 12 | See the GNU Lesser General Public License for more details. 13 | 14 | You should have received a copy of the GNU Lesser General Public 15 | License along with this library; if not, write to the Free Software 16 | Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA 17 | */ 18 | 19 | #ifndef _OTA_STORAGE_H_INCLUDED 20 | #define _OTA_STORAGE_H_INCLUDED 21 | 22 | class OTAStorage { 23 | public: 24 | virtual int open(int length) = 0; 25 | virtual size_t write(uint8_t) = 0; 26 | virtual void close() = 0; 27 | virtual void clear() = 0; 28 | virtual void apply() = 0; 29 | 30 | virtual long maxSize() { 31 | return ((256 * 1024) - 0x2000); 32 | } 33 | }; 34 | 35 | #endif 36 | -------------------------------------------------------------------------------- /src/SDStorage.h: -------------------------------------------------------------------------------- 1 | /* 2 | Copyright (c) 2017 Arduino LLC. All right reserved. 3 | 4 | This library is free software; you can redistribute it and/or 5 | modify it under the terms of the GNU Lesser General Public 6 | License as published by the Free Software Foundation; either 7 | version 2.1 of the License, or (at your option) any later version. 8 | 9 | This library is distributed in the hope that it will be useful, 10 | but WITHOUT ANY WARRANTY; without even the implied warranty of 11 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 12 | See the GNU Lesser General Public License for more details. 13 | 14 | You should have received a copy of the GNU Lesser General Public 15 | License along with this library; if not, write to the Free Software 16 | Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA 17 | */ 18 | 19 | #ifndef _SD_STORAGE_H_INCLUDED 20 | #define _SD_STORAGE_H_INCLUDED 21 | 22 | #include 23 | 24 | #include "OTAStorage.h" 25 | 26 | #ifndef SDCARD_SS_PIN 27 | #define SDCARD_SS_PIN 4 28 | #endif 29 | 30 | class SDStorageClass : public OTAStorage { 31 | public: 32 | virtual int open(int length); 33 | virtual size_t write(uint8_t); 34 | virtual void close(); 35 | virtual void clear(); 36 | virtual void apply(); 37 | 38 | private: 39 | File _file; 40 | }; 41 | 42 | extern SDStorageClass SDStorage; 43 | 44 | #endif 45 | -------------------------------------------------------------------------------- /src/InternalStorage.h: -------------------------------------------------------------------------------- 1 | /* 2 | Copyright (c) 2017 Arduino LLC. All right reserved. 3 | 4 | This library is free software; you can redistribute it and/or 5 | modify it under the terms of the GNU Lesser General Public 6 | License as published by the Free Software Foundation; either 7 | version 2.1 of the License, or (at your option) any later version. 8 | 9 | This library is distributed in the hope that it will be useful, 10 | but WITHOUT ANY WARRANTY; without even the implied warranty of 11 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 12 | See the GNU Lesser General Public License for more details. 13 | 14 | You should have received a copy of the GNU Lesser General Public 15 | License along with this library; if not, write to the Free Software 16 | Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA 17 | */ 18 | 19 | #ifndef _INTERNAL_STORAGE_H_INCLUDED 20 | #define _INTERNAL_STORAGE_H_INCLUDED 21 | 22 | #include "OTAStorage.h" 23 | 24 | class InternalStorageClass : public OTAStorage { 25 | public: 26 | virtual int open(int length); 27 | virtual size_t write(uint8_t); 28 | virtual void close(); 29 | virtual void clear(); 30 | virtual void apply(); 31 | virtual long maxSize(); 32 | 33 | private: 34 | union { 35 | uint32_t u32; 36 | uint8_t u8[4]; 37 | } _addressData; 38 | 39 | int _writeIndex; 40 | uint32_t* _writeAddress; 41 | }; 42 | 43 | extern InternalStorageClass InternalStorage; 44 | 45 | #endif 46 | -------------------------------------------------------------------------------- /src/SerialFlashStorage.h: -------------------------------------------------------------------------------- 1 | /* 2 | Copyright (c) 2017 Arduino LLC. All right reserved. 3 | 4 | This library is free software; you can redistribute it and/or 5 | modify it under the terms of the GNU Lesser General Public 6 | License as published by the Free Software Foundation; either 7 | version 2.1 of the License, or (at your option) any later version. 8 | 9 | This library is distributed in the hope that it will be useful, 10 | but WITHOUT ANY WARRANTY; without even the implied warranty of 11 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 12 | See the GNU Lesser General Public License for more details. 13 | 14 | You should have received a copy of the GNU Lesser General Public 15 | License along with this library; if not, write to the Free Software 16 | Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA 17 | */ 18 | 19 | #ifndef _SERIALFLASH_STORAGE_H_INCLUDED 20 | #define _SERIALFLASH_STORAGE_H_INCLUDED 21 | 22 | #include 23 | 24 | #include "OTAStorage.h" 25 | 26 | #define SERIAL_FLASH_BUFFER_SIZE 64 27 | #define SERIAL_FLASH_CS 5 28 | 29 | class SerialFlashStorageClass : public OTAStorage { 30 | public: 31 | virtual int open(int length); 32 | virtual size_t write(uint8_t); 33 | virtual void close(); 34 | virtual void clear(); 35 | virtual void apply(); 36 | 37 | private: 38 | SerialFlashFile _file; 39 | }; 40 | 41 | extern SerialFlashStorageClass SerialFlashStorage; 42 | 43 | #endif 44 | -------------------------------------------------------------------------------- /src/SDStorage.cpp: -------------------------------------------------------------------------------- 1 | /* 2 | Copyright (c) 2017 Arduino LLC. All right reserved. 3 | 4 | This library is free software; you can redistribute it and/or 5 | modify it under the terms of the GNU Lesser General Public 6 | License as published by the Free Software Foundation; either 7 | version 2.1 of the License, or (at your option) any later version. 8 | 9 | This library is distributed in the hope that it will be useful, 10 | but WITHOUT ANY WARRANTY; without even the implied warranty of 11 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 12 | See the GNU Lesser General Public License for more details. 13 | 14 | You should have received a copy of the GNU Lesser General Public 15 | License along with this library; if not, write to the Free Software 16 | Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA 17 | */ 18 | 19 | #include "SDStorage.h" 20 | 21 | #define UPDATE_FILE "UPDATE.BIN" 22 | 23 | int SDStorageClass::open(int length) 24 | { 25 | (void)length; 26 | _file = SD.open(UPDATE_FILE, FILE_WRITE); 27 | 28 | if (!_file) { 29 | return 0; 30 | } 31 | 32 | return 1; 33 | } 34 | 35 | size_t SDStorageClass::write(uint8_t b) 36 | { 37 | return _file.write(b); 38 | } 39 | 40 | void SDStorageClass::close() 41 | { 42 | _file.close(); 43 | } 44 | 45 | void SDStorageClass::clear() 46 | { 47 | SD.remove(UPDATE_FILE); 48 | } 49 | 50 | void SDStorageClass::apply() 51 | { 52 | // just reset, SDU copies the data to flash 53 | NVIC_SystemReset(); 54 | } 55 | 56 | SDStorageClass SDStorage; 57 | -------------------------------------------------------------------------------- /src/WiFi101OTA.h: -------------------------------------------------------------------------------- 1 | /* 2 | Copyright (c) 2017 Arduino LLC. All right reserved. 3 | 4 | This library is free software; you can redistribute it and/or 5 | modify it under the terms of the GNU Lesser General Public 6 | License as published by the Free Software Foundation; either 7 | version 2.1 of the License, or (at your option) any later version. 8 | 9 | This library is distributed in the hope that it will be useful, 10 | but WITHOUT ANY WARRANTY; without even the implied warranty of 11 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 12 | See the GNU Lesser General Public License for more details. 13 | 14 | You should have received a copy of the GNU Lesser General Public 15 | License along with this library; if not, write to the Free Software 16 | Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA 17 | */ 18 | 19 | #ifndef _WIFI101_OTA_H_INCLUDED 20 | #define _WIFI101_OTA_H_INCLUDED 21 | 22 | #include 23 | 24 | #include "WiFi101.h" 25 | #include "WiFiUdp.h" 26 | 27 | #include "OTAStorage.h" 28 | #include "SDStorage.h" 29 | #include "InternalStorage.h" 30 | #include "SerialFlashStorage.h" 31 | 32 | class WiFiOTAClass { 33 | public: 34 | WiFiOTAClass(); 35 | 36 | void begin(const char* name, const char* password, OTAStorage& storage); 37 | void poll(); 38 | 39 | private: 40 | void pollMdns(); 41 | void pollServer(); 42 | void sendHttpResponse(Client& client, int code, const char* status); 43 | void flushRequestBody(Client& client, long contentLength); 44 | 45 | private: 46 | String _name; 47 | String _expectedAuthorization; 48 | OTAStorage* _storage; 49 | WiFiServer _server; 50 | WiFiUDP _mdnsSocket; 51 | 52 | uint32_t _lastMdnsResponseTime; 53 | }; 54 | 55 | extern WiFiOTAClass WiFiOTA; 56 | 57 | #endif 58 | -------------------------------------------------------------------------------- /README.adoc: -------------------------------------------------------------------------------- 1 | :repository-owner: arduino-libraries 2 | :repository-name: WiFi101OTA 3 | 4 | = {repository-name} Library for Arduino = 5 | 6 | image:https://github.com/{repository-owner}/{repository-name}/actions/workflows/check-arduino.yml/badge.svg["Check Arduino status", link="https://github.com/{repository-owner}/{repository-name}/actions/workflows/check-arduino.yml"] 7 | image:https://github.com/{repository-owner}/{repository-name}/actions/workflows/compile-examples.yml/badge.svg["Compile Examples status", link="https://github.com/{repository-owner}/{repository-name}/actions/workflows/compile-examples.yml"] 8 | image:https://github.com/{repository-owner}/{repository-name}/actions/workflows/spell-check.yml/badge.svg["Spell Check status", link="https://github.com/{repository-owner}/{repository-name}/actions/workflows/spell-check.yml"] 9 | 10 | This library allows you to update sketches on your board over WiFi. 11 | It requires an Arduino SAMD board like the Zero or MKR1000. 12 | 13 | For more information about this library please visit us at 14 | https://www.arduino.cc/en/Reference/WiFi101OTA 15 | 16 | == License == 17 | 18 | Copyright (c) Arduino LLC. All right reserved. 19 | 20 | This library is free software; you can redistribute it and/or 21 | modify it under the terms of the GNU Lesser General Public 22 | License as published by the Free Software Foundation; either 23 | version 2.1 of the License, or (at your option) any later version. 24 | 25 | This library is distributed in the hope that it will be useful, 26 | but WITHOUT ANY WARRANTY; without even the implied warranty of 27 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 28 | Lesser General Public License for more details. 29 | 30 | You should have received a copy of the GNU Lesser General Public 31 | License along with this library; if not, write to the Free Software 32 | Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA 33 | -------------------------------------------------------------------------------- /src/SerialFlashStorage.cpp: -------------------------------------------------------------------------------- 1 | /* 2 | Copyright (c) 2017 Arduino LLC. All right reserved. 3 | 4 | This library is free software; you can redistribute it and/or 5 | modify it under the terms of the GNU Lesser General Public 6 | License as published by the Free Software Foundation; either 7 | version 2.1 of the License, or (at your option) any later version. 8 | 9 | This library is distributed in the hope that it will be useful, 10 | but WITHOUT ANY WARRANTY; without even the implied warranty of 11 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 12 | See the GNU Lesser General Public License for more details. 13 | 14 | You should have received a copy of the GNU Lesser General Public 15 | License along with this library; if not, write to the Free Software 16 | Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA 17 | */ 18 | 19 | #include "SerialFlashStorage.h" 20 | 21 | #define UPDATE_FILE "UPDATE.BIN" 22 | 23 | int SerialFlashStorageClass::open(int contentLength) 24 | { 25 | if (!SerialFlash.begin(SERIAL_FLASH_CS)) { 26 | return 0; 27 | } 28 | 29 | while (!SerialFlash.ready()) {} 30 | 31 | if (SerialFlash.exists(UPDATE_FILE)) { 32 | SerialFlash.remove(UPDATE_FILE); 33 | } 34 | 35 | if (SerialFlash.create(UPDATE_FILE, contentLength)) { 36 | _file = SerialFlash.open(UPDATE_FILE); 37 | } 38 | 39 | if (!_file) { 40 | return 0; 41 | } 42 | 43 | return 1; 44 | } 45 | 46 | size_t SerialFlashStorageClass::write(uint8_t b) 47 | { 48 | while (!SerialFlash.ready()) {} 49 | int ret = _file.write(&b, 1); 50 | return ret; 51 | } 52 | 53 | void SerialFlashStorageClass::close() 54 | { 55 | _file.close(); 56 | } 57 | 58 | void SerialFlashStorageClass::clear() 59 | { 60 | SerialFlash.remove(UPDATE_FILE); 61 | } 62 | 63 | void SerialFlashStorageClass::apply() 64 | { 65 | // just reset, SDU copies the data to flash 66 | NVIC_SystemReset(); 67 | } 68 | 69 | SerialFlashStorageClass SerialFlashStorage; 70 | -------------------------------------------------------------------------------- /.github/workflows/compile-examples.yml: -------------------------------------------------------------------------------- 1 | name: Compile Examples 2 | 3 | # See: https://docs.github.com/en/free-pro-team@latest/actions/reference/events-that-trigger-workflows 4 | on: 5 | push: 6 | paths: 7 | - ".github/workflows/compile-examples.yml" 8 | - "library.properties" 9 | - "examples/**" 10 | - "src/**" 11 | pull_request: 12 | paths: 13 | - ".github/workflows/compile-examples.yml" 14 | - "library.properties" 15 | - "examples/**" 16 | - "src/**" 17 | schedule: 18 | # Run every Tuesday at 8 AM UTC to catch breakage caused by changes to external resources (libraries, platforms). 19 | - cron: "0 8 * * TUE" 20 | workflow_dispatch: 21 | repository_dispatch: 22 | 23 | jobs: 24 | build: 25 | name: ${{ matrix.board.fqbn }} 26 | runs-on: ubuntu-latest 27 | 28 | env: 29 | SKETCHES_REPORTS_PATH: sketches-reports 30 | 31 | strategy: 32 | fail-fast: false 33 | 34 | matrix: 35 | board: 36 | - fqbn: arduino:samd:mkr1000 37 | platforms: | 38 | - name: arduino:samd 39 | artifact-name-suffix: arduino-samd-mkr1000 40 | 41 | steps: 42 | - name: Checkout repository 43 | uses: actions/checkout@v4 44 | 45 | - name: Compile examples 46 | uses: arduino/compile-sketches@v1 47 | with: 48 | github-token: ${{ secrets.GITHUB_TOKEN }} 49 | fqbn: ${{ matrix.board.fqbn }} 50 | platforms: ${{ matrix.board.platforms }} 51 | libraries: | 52 | # Install the library from the local path. 53 | - source-path: ./ 54 | # Install library dependencies. 55 | - name: SD 56 | - name: SerialFlash 57 | - name: WiFi101 58 | sketch-paths: | 59 | - examples 60 | enable-deltas-report: true 61 | sketches-report-path: ${{ env.SKETCHES_REPORTS_PATH }} 62 | 63 | - name: Save sketches report as workflow artifact 64 | uses: actions/upload-artifact@v4 65 | with: 66 | if-no-files-found: error 67 | path: ${{ env.SKETCHES_REPORTS_PATH }} 68 | name: sketches-report-${{ matrix.board.artifact-name-suffix }} 69 | -------------------------------------------------------------------------------- /examples/WiFi101_OTA/WiFi101_OTA.ino: -------------------------------------------------------------------------------- 1 | /* 2 | 3 | This example connects to an WPA encrypted WiFi network. 4 | Then it prints the MAC address of the WiFi shield, 5 | the IP address obtained, and other network details. 6 | It then polls for sketch updates over WiFi. Sketches 7 | can be updated by selecting a network port from within 8 | the Arduino IDE: Tools -> Port -> Network Ports ... 9 | 10 | Circuit: 11 | * WiFi shield attached 12 | 13 | created 13 July 2010 14 | by dlf (Metodo2 srl) 15 | modified 31 May 2012 16 | by Tom Igoe 17 | modified 16 January 2017 18 | by Sandeep Mistry 19 | */ 20 | 21 | #include 22 | #include 23 | #include 24 | 25 | #include "arduino_secrets.h" 26 | ///////please enter your sensitive data in the Secret tab/arduino_secrets.h 27 | /////// WiFi Settings /////// 28 | char ssid[] = SECRET_SSID; // your network SSID (name) 29 | char pass[] = SECRET_PASS; // your network password 30 | 31 | int status = WL_IDLE_STATUS; 32 | 33 | void setup() { 34 | //Initialize serial: 35 | Serial.begin(9600); 36 | 37 | // check for the presence of the shield: 38 | if (WiFi.status() == WL_NO_SHIELD) { 39 | Serial.println("WiFi shield not present"); 40 | // don't continue: 41 | while (true); 42 | } 43 | 44 | // attempt to connect to WiFi network: 45 | while ( status != WL_CONNECTED) { 46 | Serial.print("Attempting to connect to SSID: "); 47 | Serial.println(ssid); 48 | // Connect to WPA/WPA2 network. Change this line if using open or WEP network: 49 | status = WiFi.begin(ssid, pass); 50 | } 51 | 52 | // start the WiFi OTA library with internal (flash) based storage 53 | WiFiOTA.begin("Arduino", "password", InternalStorage); 54 | 55 | // you're connected now, so print out the status: 56 | printWifiStatus(); 57 | } 58 | 59 | void loop() { 60 | // check for WiFi OTA updates 61 | WiFiOTA.poll(); 62 | 63 | // add your normal loop code below ... 64 | } 65 | 66 | void printWifiStatus() { 67 | // print the SSID of the network you're attached to: 68 | Serial.print("SSID: "); 69 | Serial.println(WiFi.SSID()); 70 | 71 | // print your WiFi shield's IP address: 72 | IPAddress ip = WiFi.localIP(); 73 | Serial.print("IP Address: "); 74 | Serial.println(ip); 75 | 76 | // print the received signal strength: 77 | long rssi = WiFi.RSSI(); 78 | Serial.print("signal strength (RSSI):"); 79 | Serial.print(rssi); 80 | Serial.println(" dBm"); 81 | } 82 | -------------------------------------------------------------------------------- /examples/WiFi101_SD_OTA/WiFi101_SD_OTA.ino: -------------------------------------------------------------------------------- 1 | /* 2 | 3 | This example connects to an WPA encrypted WiFi network. 4 | Then it prints the MAC address of the WiFi shield, 5 | the IP address obtained, and other network details. 6 | It then polls for sketch updates over WiFi. Sketches 7 | can be updated by selecting a network port from within 8 | the Arduino IDE: Tools -> Port -> Network Ports ... 9 | 10 | Circuit: 11 | * WiFi shield attached 12 | * SD shield attached 13 | 14 | created 13 July 2010 15 | by dlf (Metodo2 srl) 16 | modified 31 May 2012 17 | by Tom Igoe 18 | modified 16 January 2017 19 | by Sandeep Mistry 20 | */ 21 | 22 | #include 23 | #include 24 | #include 25 | #include 26 | #include 27 | 28 | #include "arduino_secrets.h" 29 | ///////please enter your sensitive data in the Secret tab/arduino_secrets.h 30 | /////// WiFi Settings /////// 31 | char ssid[] = SECRET_SSID; // your network SSID (name) 32 | char pass[] = SECRET_PASS; // your network password 33 | 34 | int status = WL_IDLE_STATUS; 35 | 36 | void setup() { 37 | //Initialize serial: 38 | Serial.begin(9600); 39 | 40 | // setup SD card 41 | Serial.print("Initializing SD card..."); 42 | if (!SD.begin(SDCARD_SS_PIN)) { 43 | Serial.println("initialization failed!"); 44 | // don't continue: 45 | while (true); 46 | } 47 | Serial.println("initialization done."); 48 | 49 | // check for the presence of the shield: 50 | if (WiFi.status() == WL_NO_SHIELD) { 51 | Serial.println("WiFi shield not present"); 52 | // don't continue: 53 | while (true); 54 | } 55 | 56 | // attempt to connect to WiFi network: 57 | while ( status != WL_CONNECTED) { 58 | Serial.print("Attempting to connect to SSID: "); 59 | Serial.println(ssid); 60 | // Connect to WPA/WPA2 network. Change this line if using open or WEP network: 61 | status = WiFi.begin(ssid, pass); 62 | } 63 | 64 | // start the WiFi OTA library with SD based storage 65 | WiFiOTA.begin("Arduino", "password", SDStorage); 66 | 67 | // you're connected now, so print out the status: 68 | printWifiStatus(); 69 | } 70 | 71 | void loop() { 72 | // check for WiFi OTA updates 73 | WiFiOTA.poll(); 74 | 75 | // add your normal loop code below ... 76 | } 77 | 78 | void printWifiStatus() { 79 | // print the SSID of the network you're attached to: 80 | Serial.print("SSID: "); 81 | Serial.println(WiFi.SSID()); 82 | 83 | // print your WiFi shield's IP address: 84 | IPAddress ip = WiFi.localIP(); 85 | Serial.print("IP Address: "); 86 | Serial.println(ip); 87 | 88 | // print the received signal strength: 89 | long rssi = WiFi.RSSI(); 90 | Serial.print("signal strength (RSSI):"); 91 | Serial.print(rssi); 92 | Serial.println(" dBm"); 93 | } 94 | -------------------------------------------------------------------------------- /docs/api.md: -------------------------------------------------------------------------------- 1 | # WiFi101 OTA library 2 | 3 | ## Methods 4 | 5 | ### `WiFiOTA.begin()` 6 | 7 | #### Description 8 | 9 | Initializes the WiFi101 OTA library specifying board's network name, password and storage media 10 | 11 | #### Syntax 12 | 13 | ``` 14 | WiFiOTA.begin(name, password, storage); 15 | ``` 16 | 17 | #### Parameters 18 | 19 | **name:** the board's name to visualize in the network ports list 20 | 21 | **password:** the password to insert for the remote upload 22 | 23 | **storage:** the storage media to be used. Can be one between: 24 | 25 | - InternalStorage 26 | - SDStorage 27 | 28 | #### Example 29 | 30 | ``` 31 | #include 32 | #include 33 | #include 34 | 35 | char ssid[] = "yourNetwork"; // your network SSID (name) 36 | char pass[] = "secretPassword"; // your network password 37 | 38 | int status = WL_IDLE_STATUS; 39 | 40 | void setup() { 41 | //Initialize serial: 42 | Serial.begin(9600); 43 | 44 | // check for the presence of the shield: 45 | if (WiFi.status() == WL_NO_SHIELD) { 46 | Serial.println("WiFi shield not present"); 47 | // don't continue: 48 | while (true); 49 | } 50 | 51 | // attempt to connect to WiFi network: 52 | while ( status != WL_CONNECTED) { 53 | Serial.print("Attempting to connect to SSID: "); 54 | Serial.println(ssid); 55 | // Connect to WPA/WPA2 network. Change this line if using open or WEP network: 56 | status = WiFi.begin(ssid, pass); 57 | } 58 | 59 | // start the WiFi OTA library with internal (flash) based storage 60 | WiFiOTA.begin("Arduino", "password", InternalStorage); 61 | } 62 | 63 | void loop() { 64 | // check for WiFi OTA updates 65 | WiFiOTA.poll(); 66 | } 67 | ``` 68 | 69 | ### `WiFiOTA.poll()` 70 | 71 | #### Description 72 | 73 | Checks for WiFi OTA updates 74 | 75 | #### Syntax 76 | 77 | ``` 78 | WiFiOTA.poll(); 79 | ``` 80 | 81 | #### Parameters 82 | 83 | None 84 | 85 | #### Example 86 | 87 | ``` 88 | #include 89 | #include 90 | #include 91 | 92 | char ssid[] = "yourNetwork"; // your network SSID (name) 93 | char pass[] = "secretPassword"; // your network password 94 | 95 | int status = WL_IDLE_STATUS; 96 | 97 | void setup() { 98 | //Initialize serial: 99 | Serial.begin(9600); 100 | 101 | // check for the presence of the shield: 102 | if (WiFi.status() == WL_NO_SHIELD) { 103 | Serial.println("WiFi shield not present"); 104 | // don't continue: 105 | while (true); 106 | } 107 | 108 | // attempt to connect to WiFi network: 109 | while ( status != WL_CONNECTED) { 110 | Serial.print("Attempting to connect to SSID: "); 111 | Serial.println(ssid); 112 | // Connect to WPA/WPA2 network. Change this line if using open or WEP network: 113 | status = WiFi.begin(ssid, pass); 114 | } 115 | 116 | // start the WiFi OTA library with internal (flash) based storage 117 | WiFiOTA.begin("Arduino", "password", InternalStorage); 118 | } 119 | 120 | void loop() { 121 | // check for WiFi OTA updates 122 | WiFiOTA.poll(); 123 | } 124 | ``` -------------------------------------------------------------------------------- /src/InternalStorage.cpp: -------------------------------------------------------------------------------- 1 | /* 2 | Copyright (c) 2017 Arduino LLC. All right reserved. 3 | 4 | This library is free software; you can redistribute it and/or 5 | modify it under the terms of the GNU Lesser General Public 6 | License as published by the Free Software Foundation; either 7 | version 2.1 of the License, or (at your option) any later version. 8 | 9 | This library is distributed in the hope that it will be useful, 10 | but WITHOUT ANY WARRANTY; without even the implied warranty of 11 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 12 | See the GNU Lesser General Public License for more details. 13 | 14 | You should have received a copy of the GNU Lesser General Public 15 | License along with this library; if not, write to the Free Software 16 | Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA 17 | */ 18 | 19 | #include 20 | 21 | #include "InternalStorage.h" 22 | 23 | #define PAGE_SIZE (64) 24 | #define PAGES (4096) 25 | #define MAX_FLASH (PAGE_SIZE * PAGES) 26 | #define ROW_SIZE (PAGE_SIZE * 4) 27 | 28 | #define SKETCH_START_ADDRESS (0x2000) 29 | #define MAX_PARTIONED_SKETCH_SIZE ((MAX_FLASH - SKETCH_START_ADDRESS) / 2) 30 | #define STORAGE_START_ADDRESS (SKETCH_START_ADDRESS + MAX_PARTIONED_SKETCH_SIZE) 31 | 32 | extern "C" { 33 | // these functions must be in RAM (.data) and NOT inlined 34 | // as they erase and copy the sketch data in flash 35 | 36 | __attribute__ ((long_call, noinline, section (".data#"))) 37 | static void eraseFlash(int address, int length) 38 | { 39 | for (int i = 0; i < length; i += ROW_SIZE) { 40 | NVMCTRL->ADDR.reg = ((uint32_t)(address + i)) / 2; 41 | NVMCTRL->CTRLA.reg = NVMCTRL_CTRLA_CMDEX_KEY | NVMCTRL_CTRLA_CMD_ER; 42 | 43 | while (!NVMCTRL->INTFLAG.bit.READY); 44 | } 45 | } 46 | 47 | __attribute__ ((long_call, noinline, section (".data#"))) 48 | static void copyFlashAndReset(int dest, int src, int length) 49 | { 50 | uint32_t* d = (uint32_t*)dest; 51 | uint32_t* s = (uint32_t*)src; 52 | 53 | eraseFlash(dest, length); 54 | 55 | for (int i = 0; i < length; i += 4) { 56 | *d++ = *s++; 57 | 58 | while (!NVMCTRL->INTFLAG.bit.READY); 59 | } 60 | 61 | NVIC_SystemReset(); 62 | } 63 | } 64 | 65 | int InternalStorageClass::open(int length) 66 | { 67 | (void)length; 68 | _writeIndex = 0; 69 | _writeAddress = (uint32_t*)STORAGE_START_ADDRESS; 70 | 71 | // enable auto page writes 72 | NVMCTRL->CTRLB.bit.MANW = 0; 73 | 74 | eraseFlash(STORAGE_START_ADDRESS, MAX_PARTIONED_SKETCH_SIZE); 75 | 76 | return 1; 77 | } 78 | 79 | size_t InternalStorageClass::write(uint8_t b) 80 | { 81 | _addressData.u8[_writeIndex] = b; 82 | _writeIndex++; 83 | 84 | if (_writeIndex == 4) { 85 | _writeIndex = 0; 86 | 87 | *_writeAddress = _addressData.u32; 88 | 89 | _writeAddress++; 90 | 91 | while (!NVMCTRL->INTFLAG.bit.READY); 92 | } 93 | 94 | return 1; 95 | } 96 | 97 | void InternalStorageClass::close() 98 | { 99 | while ((int)_writeAddress % PAGE_SIZE) { 100 | write(0xff); 101 | } 102 | } 103 | 104 | void InternalStorageClass::clear() 105 | { 106 | } 107 | 108 | void InternalStorageClass::apply() 109 | { 110 | // disable interrupts, as vector table will be erased during flash sequence 111 | noInterrupts(); 112 | 113 | copyFlashAndReset(SKETCH_START_ADDRESS, STORAGE_START_ADDRESS, MAX_PARTIONED_SKETCH_SIZE); 114 | } 115 | 116 | long InternalStorageClass::maxSize() 117 | { 118 | return MAX_PARTIONED_SKETCH_SIZE; 119 | } 120 | 121 | InternalStorageClass InternalStorage; 122 | -------------------------------------------------------------------------------- /.github/workflows/sync-labels.yml: -------------------------------------------------------------------------------- 1 | # Source: https://github.com/arduino/tooling-project-assets/blob/main/workflow-templates/sync-labels.md 2 | name: Sync Labels 3 | 4 | # See: https://docs.github.com/en/actions/reference/events-that-trigger-workflows 5 | on: 6 | push: 7 | paths: 8 | - ".github/workflows/sync-labels.ya?ml" 9 | - ".github/label-configuration-files/*.ya?ml" 10 | pull_request: 11 | paths: 12 | - ".github/workflows/sync-labels.ya?ml" 13 | - ".github/label-configuration-files/*.ya?ml" 14 | schedule: 15 | # Run daily at 8 AM UTC to sync with changes to shared label configurations. 16 | - cron: "0 8 * * *" 17 | workflow_dispatch: 18 | repository_dispatch: 19 | 20 | env: 21 | CONFIGURATIONS_FOLDER: .github/label-configuration-files 22 | CONFIGURATIONS_ARTIFACT: label-configuration-files 23 | 24 | jobs: 25 | check: 26 | runs-on: ubuntu-latest 27 | 28 | steps: 29 | - name: Checkout repository 30 | uses: actions/checkout@v4 31 | 32 | - name: Download JSON schema for labels configuration file 33 | id: download-schema 34 | uses: carlosperate/download-file-action@v2 35 | with: 36 | file-url: https://raw.githubusercontent.com/arduino/tooling-project-assets/main/workflow-templates/assets/sync-labels/arduino-tooling-gh-label-configuration-schema.json 37 | location: ${{ runner.temp }}/label-configuration-schema 38 | 39 | - name: Install JSON schema validator 40 | run: | 41 | sudo npm install \ 42 | --global \ 43 | ajv-cli \ 44 | ajv-formats 45 | 46 | - name: Validate local labels configuration 47 | run: | 48 | # See: https://github.com/ajv-validator/ajv-cli#readme 49 | ajv validate \ 50 | --all-errors \ 51 | -c ajv-formats \ 52 | -s "${{ steps.download-schema.outputs.file-path }}" \ 53 | -d "${{ env.CONFIGURATIONS_FOLDER }}/*.{yml,yaml}" 54 | 55 | download: 56 | needs: check 57 | runs-on: ubuntu-latest 58 | 59 | strategy: 60 | matrix: 61 | filename: 62 | # Filenames of the shared configurations to apply to the repository in addition to the local configuration. 63 | # https://github.com/arduino/tooling-project-assets/blob/main/workflow-templates/assets/sync-labels 64 | - universal.yml 65 | 66 | steps: 67 | - name: Download 68 | uses: carlosperate/download-file-action@v2 69 | with: 70 | file-url: https://raw.githubusercontent.com/arduino/tooling-project-assets/main/workflow-templates/assets/sync-labels/${{ matrix.filename }} 71 | 72 | - name: Pass configuration files to next job via workflow artifact 73 | uses: actions/upload-artifact@v4 74 | with: 75 | path: | 76 | *.yaml 77 | *.yml 78 | if-no-files-found: error 79 | name: ${{ env.CONFIGURATIONS_ARTIFACT }} 80 | 81 | sync: 82 | needs: download 83 | runs-on: ubuntu-latest 84 | 85 | steps: 86 | - name: Set environment variables 87 | run: | 88 | # See: https://docs.github.com/en/actions/reference/workflow-commands-for-github-actions#setting-an-environment-variable 89 | echo "MERGED_CONFIGURATION_PATH=${{ runner.temp }}/labels.yml" >> "$GITHUB_ENV" 90 | 91 | - name: Determine whether to dry run 92 | id: dry-run 93 | if: > 94 | github.event_name == 'pull_request' || 95 | ( 96 | ( 97 | github.event_name == 'push' || 98 | github.event_name == 'workflow_dispatch' 99 | ) && 100 | github.ref != format('refs/heads/{0}', github.event.repository.default_branch) 101 | ) 102 | run: | 103 | # Use of this flag in the github-label-sync command will cause it to only check the validity of the 104 | # configuration. 105 | echo "::set-output name=flag::--dry-run" 106 | 107 | - name: Checkout repository 108 | uses: actions/checkout@v4 109 | 110 | - name: Download configuration files artifact 111 | uses: actions/download-artifact@v4 112 | with: 113 | name: ${{ env.CONFIGURATIONS_ARTIFACT }} 114 | path: ${{ env.CONFIGURATIONS_FOLDER }} 115 | 116 | - name: Remove unneeded artifact 117 | uses: geekyeggo/delete-artifact@v5 118 | with: 119 | name: ${{ env.CONFIGURATIONS_ARTIFACT }} 120 | 121 | - name: Merge label configuration files 122 | run: | 123 | # Merge all configuration files 124 | shopt -s extglob 125 | cat "${{ env.CONFIGURATIONS_FOLDER }}"/*.@(yml|yaml) > "${{ env.MERGED_CONFIGURATION_PATH }}" 126 | 127 | - name: Install github-label-sync 128 | run: sudo npm install --global github-label-sync 129 | 130 | - name: Sync labels 131 | env: 132 | GITHUB_ACCESS_TOKEN: ${{ secrets.GITHUB_TOKEN }} 133 | run: | 134 | # See: https://github.com/Financial-Times/github-label-sync 135 | github-label-sync \ 136 | --labels "${{ env.MERGED_CONFIGURATION_PATH }}" \ 137 | ${{ steps.dry-run.outputs.flag }} \ 138 | ${{ github.repository }} 139 | -------------------------------------------------------------------------------- /src/WiFi101OTA.cpp: -------------------------------------------------------------------------------- 1 | /* 2 | Copyright (c) 2017 Arduino LLC. All right reserved. 3 | 4 | This library is free software; you can redistribute it and/or 5 | modify it under the terms of the GNU Lesser General Public 6 | License as published by the Free Software Foundation; either 7 | version 2.1 of the License, or (at your option) any later version. 8 | 9 | This library is distributed in the hope that it will be useful, 10 | but WITHOUT ANY WARRANTY; without even the implied warranty of 11 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 12 | See the GNU Lesser General Public License for more details. 13 | 14 | You should have received a copy of the GNU Lesser General Public 15 | License along with this library; if not, write to the Free Software 16 | Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA 17 | */ 18 | 19 | #include 20 | 21 | #include "WiFi101OTA.h" 22 | 23 | #if defined(ARDUINO_SAMD_ZERO) 24 | #define BOARD "arduino_zero_edbg" 25 | #elif defined(ARDUINO_SAMD_MKR1000) 26 | #define BOARD "mkr1000" 27 | #elif defined(ARDUINO_SAMD_MKRZERO) 28 | #define BOARD "mkrzero" 29 | #else 30 | #error "Unsupported board!" 31 | #endif 32 | 33 | #define BOARD_LENGTH (sizeof(BOARD) - 1) 34 | 35 | static String base64Encode(const String& in) 36 | { 37 | static const char* CODES = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/="; 38 | 39 | int b; 40 | String out; 41 | out.reserve((in.length()) * 4 / 3); 42 | 43 | for (unsigned int i = 0; i < in.length(); i += 3) { 44 | b = (in.charAt(i) & 0xFC) >> 2; 45 | out += CODES[b]; 46 | 47 | b = (in.charAt(i) & 0x03) << 4; 48 | if (i + 1 < in.length()) { 49 | b |= (in.charAt(i + 1) & 0xF0) >> 4; 50 | out += CODES[b]; 51 | b = (in.charAt(i + 1) & 0x0F) << 2; 52 | if (i + 2 < in.length()) { 53 | b |= (in.charAt(i + 2) & 0xC0) >> 6; 54 | out += CODES[b]; 55 | b = in.charAt(i + 2) & 0x3F; 56 | out += CODES[b]; 57 | } else { 58 | out += CODES[b]; 59 | out += '='; 60 | } 61 | } else { 62 | out += CODES[b]; 63 | out += "=="; 64 | } 65 | } 66 | 67 | return out; 68 | } 69 | 70 | WiFiOTAClass::WiFiOTAClass() : 71 | _storage(NULL), 72 | _server(65280), 73 | _lastMdnsResponseTime(0) 74 | { 75 | } 76 | 77 | void WiFiOTAClass::begin(const char* name, const char* password, OTAStorage& storage) 78 | { 79 | _name = name; 80 | _expectedAuthorization = "Basic " + base64Encode("arduino:" + String(password)); 81 | _storage = &storage; 82 | 83 | _server.begin(); 84 | 85 | _mdnsSocket.beginMulti(IPAddress(224, 0, 0, 251), 5353); 86 | } 87 | 88 | void WiFiOTAClass::poll() 89 | { 90 | pollMdns(); 91 | pollServer(); 92 | } 93 | 94 | void WiFiOTAClass::pollMdns() 95 | { 96 | int packetLength = _mdnsSocket.parsePacket(); 97 | 98 | if (packetLength <= 0) { 99 | return; 100 | } 101 | 102 | const byte ARDUINO_SERVICE_REQUEST[37] = { 103 | 0x00, 0x00, // transaction id 104 | 0x00, 0x00, // flags 105 | 0x00, 0x01, // questions 106 | 0x00, 0x00, // answer RRs 107 | 0x00, 0x00, // authority RRs 108 | 0x00, 0x00, // additional RRs 109 | 0x08, 110 | 0x5f, 0x61, 0x72, 0x64, 0x75, 0x69, 0x6e, 0x6f, // _arduino 111 | 0x04, 112 | 0x5f, 0x74, 0x63, 0x70, // _tcp 113 | 0x05, 114 | 0x6c, 0x6f, 0x63, 0x61, 0x6c, 0x00, // local 115 | 0x00, 0x0c, // PTR 116 | 0x00, 0x01 // Class IN 117 | }; 118 | 119 | if (packetLength != sizeof(ARDUINO_SERVICE_REQUEST)) { 120 | while (packetLength) { 121 | if (_mdnsSocket.available()) { 122 | packetLength--; 123 | _mdnsSocket.read(); 124 | } 125 | } 126 | return; 127 | } 128 | 129 | byte request[packetLength]; 130 | 131 | _mdnsSocket.read(request, sizeof(request)); 132 | 133 | if (memcmp(&request[2], &ARDUINO_SERVICE_REQUEST[2], packetLength - 2) != 0) { 134 | return; 135 | } 136 | 137 | if ((millis() - _lastMdnsResponseTime) < 1000) { 138 | // ignore request 139 | return; 140 | } 141 | _lastMdnsResponseTime = millis(); 142 | 143 | _mdnsSocket.beginPacket(IPAddress(224, 0, 0, 251), 5353); 144 | 145 | const byte responseHeader[] = { 146 | 0x00, 0x00, // transaction id 147 | 0x84, 0x00, // flags 148 | 0x00, 0x00, // questions 149 | 0x00, 0x04, // answers RRs 150 | 0x00, 0x00, // authority RRs 151 | 0x00, 0x00 // additional RRS 152 | }; 153 | _mdnsSocket.write(responseHeader, sizeof(responseHeader)); 154 | 155 | const byte ptrRecordStart[] = { 156 | 0x08, 157 | '_', 'a', 'r', 'd', 'u', 'i', 'n', 'o', 158 | 159 | 0x04, 160 | '_', 't', 'c', 'p', 161 | 162 | 0x05, 163 | 'l', 'o', 'c', 'a', 'l', 164 | 0x00, 165 | 166 | 0x00, 0x0c, // PTR 167 | 0x00, 0x01, // class IN 168 | 0x00, 0x00, 0x11, 0x94, // TTL 169 | 170 | 0x00, (byte)(_name.length() + 3), // length 171 | (byte)_name.length() 172 | }; 173 | 174 | const byte ptrRecordEnd[] = { 175 | 0xc0, 0x0c 176 | }; 177 | 178 | _mdnsSocket.write(ptrRecordStart, sizeof(ptrRecordStart)); 179 | _mdnsSocket.write(_name.c_str(), _name.length()); 180 | _mdnsSocket.write(ptrRecordEnd, sizeof(ptrRecordEnd)); 181 | 182 | const byte txtRecord[] = { 183 | 0xc0, 0x2b, 184 | 0x00, 0x10, // TXT strings 185 | 0x80, 0x01, // class 186 | 0x00, 0x00, 0x11, 0x94, // TTL 187 | 0x00, (50 + BOARD_LENGTH), 188 | 13, 189 | 's', 's', 'h', '_', 'u', 'p', 'l', 'o', 'a', 'd', '=', 'n', 'o', 190 | 12, 191 | 't', 'c', 'p', '_', 'c', 'h', 'e', 'c', 'k', '=', 'n', 'o', 192 | 15, 193 | 'a', 'u', 't', 'h', '_', 'u', 'p', 'l', 'o', 'a', 'd', '=', 'y', 'e', 's', 194 | (6 + BOARD_LENGTH), 195 | 'b', 'o', 'a', 'r', 'd', '=', 196 | }; 197 | _mdnsSocket.write(txtRecord, sizeof(txtRecord)); 198 | _mdnsSocket.write((byte*)BOARD, BOARD_LENGTH); 199 | 200 | const byte srvRecordStart[] = { 201 | 0xc0, 0x2b, 202 | 0x00, 0x21, // SRV 203 | 0x80, 0x01, // class 204 | 0x00, 0x00, 0x00, 0x78, // TTL 205 | 0x00, (byte)(_name.length() + 9), // length 206 | 0x00, 0x00, 207 | 0x00, 0x00, 208 | 0xff, 0x00, // port 209 | (byte)_name.length() 210 | }; 211 | 212 | const byte srvRecordEnd[] = { 213 | 0xc0, 0x1a 214 | }; 215 | 216 | _mdnsSocket.write(srvRecordStart, sizeof(srvRecordStart)); 217 | _mdnsSocket.write(_name.c_str(), _name.length()); 218 | _mdnsSocket.write(srvRecordEnd, sizeof(srvRecordEnd)); 219 | 220 | uint32_t localIp = WiFi.localIP(); 221 | 222 | byte aRecordNameOffset = sizeof(responseHeader) + 223 | sizeof(ptrRecordStart) + _name.length() + sizeof(ptrRecordEnd) + 224 | sizeof(txtRecord) + BOARD_LENGTH + 225 | sizeof(srvRecordStart) - 1; 226 | 227 | byte aRecord[] = { 228 | 0xc0, aRecordNameOffset, 229 | 230 | 0x00, 0x01, // A record 231 | 0x80, 0x01, // class 232 | 0x00, 0x00, 0x00, 0x78, // TTL 233 | 0x00, 0x04, 234 | 0xff, 0xff, 0xff, 0xff // IP 235 | }; 236 | memcpy(&aRecord[sizeof(aRecord) - 4], &localIp, sizeof(localIp)); 237 | _mdnsSocket.write(aRecord, sizeof(aRecord)); 238 | 239 | _mdnsSocket.endPacket(); 240 | } 241 | 242 | void WiFiOTAClass::pollServer() 243 | { 244 | WiFiClient client = _server.available(); 245 | 246 | if (client) { 247 | String request = client.readStringUntil('\n'); 248 | request.trim(); 249 | 250 | String header; 251 | long contentLength = -1; 252 | String authorization; 253 | 254 | do { 255 | header = client.readStringUntil('\n'); 256 | header.trim(); 257 | 258 | if (header.startsWith("Content-Length: ")) { 259 | header.remove(0, 16); 260 | 261 | contentLength = header.toInt(); 262 | } else if (header.startsWith("Authorization: ")) { 263 | header.remove(0, 15); 264 | 265 | authorization = header; 266 | } 267 | } while (header != ""); 268 | 269 | if (request != "POST /sketch HTTP/1.1") { 270 | flushRequestBody(client, contentLength); 271 | sendHttpResponse(client, 404, "Not Found"); 272 | return; 273 | } 274 | 275 | if (_expectedAuthorization != authorization) { 276 | flushRequestBody(client, contentLength); 277 | sendHttpResponse(client, 401, "Unauthorized"); 278 | return; 279 | } 280 | 281 | if (contentLength <= 0) { 282 | sendHttpResponse(client, 400, "Bad Request"); 283 | return; 284 | } 285 | 286 | if (_storage == NULL || !_storage->open(contentLength)) { 287 | flushRequestBody(client, contentLength); 288 | sendHttpResponse(client, 500, "Internal Server Error"); 289 | return; 290 | } 291 | 292 | if (contentLength > _storage->maxSize()) { 293 | _storage->close(); 294 | flushRequestBody(client, contentLength); 295 | sendHttpResponse(client, 413, "Payload Too Large"); 296 | return; 297 | } 298 | 299 | long read = 0; 300 | 301 | while (client.connected() && read < contentLength) { 302 | while (client.available()) { 303 | read++; 304 | 305 | _storage->write((char)client.read()); 306 | } 307 | } 308 | 309 | _storage->close(); 310 | 311 | if (read == contentLength) { 312 | sendHttpResponse(client, 200, "OK"); 313 | 314 | delay(500); 315 | 316 | // apply the update 317 | _storage->apply(); 318 | 319 | while (true); 320 | } else { 321 | 322 | sendHttpResponse(client, 414, "Payload size wrong"); 323 | _storage->clear(); 324 | 325 | delay(500); 326 | 327 | client.stop(); 328 | } 329 | } 330 | } 331 | 332 | void WiFiOTAClass::sendHttpResponse(Client& client, int code, const char* status) 333 | { 334 | while (client.available()) { 335 | client.read(); 336 | } 337 | 338 | client.print("HTTP/1.1 "); 339 | client.print(code); 340 | client.print(" "); 341 | client.println(status); 342 | client.println("Connection: close"); 343 | client.println(); 344 | client.stop(); 345 | } 346 | 347 | void WiFiOTAClass::flushRequestBody(Client& client, long contentLength) 348 | { 349 | long read = 0; 350 | 351 | while (client.connected() && read < contentLength) { 352 | if (client.available()) { 353 | read++; 354 | 355 | client.read(); 356 | } 357 | } 358 | } 359 | 360 | WiFiOTAClass WiFiOTA; 361 | --------------------------------------------------------------------------------