Changes to FlashWriteSize (#14)

Co-authored-by: Paulus Schoutsen <paulus@home-assistant.io>

src/const.ts

@@ -18,8 +18,8 @@ export const flashSizes = {
   "16MB": 0x90,
 };
 
-export const FLASH_WRITE_SIZE = 0x200;
-export const ESP32S2_FLASH_WRITE_SIZE = 0x400;
+export const FLASH_WRITE_SIZE = 0x400;
+export const STUB_FLASH_WRITE_SIZE = 0x4000;
 export const FLASH_SECTOR_SIZE = 0x1000; // Flash sector size, minimum unit of erase.
 export const ESP_ROM_BAUD = 115200;
 

src/esp_loader.ts

@@ -7,7 +7,6 @@ import {
   DEFAULT_TIMEOUT,
   ERASE_REGION_TIMEOUT_PER_MB,
   ESP32S2_DATAREGVALUE,
-  ESP32S2_FLASH_WRITE_SIZE,
   ESP32_DATAREGVALUE,
   ESP8266_DATAREGVALUE,
   ESP_CHANGE_BAUDRATE,
@@ -24,6 +23,7 @@ import {
   ESP_SYNC,
   FLASH_SECTOR_SIZE,
   FLASH_WRITE_SIZE,
+  STUB_FLASH_WRITE_SIZE,
   MEM_END_ROM_TIMEOUT,
   ROM_INVALID_RECV_MSG,
   SYNC_PACKET,
@@ -508,8 +508,8 @@ export class ESPLoader extends EventTarget {
    * Get the Flash write size based on the chip
    */
   getFlashWriteSize() {
-    if (this.chipFamily == CHIP_FAMILY_ESP32S2) {
-      return ESP32S2_FLASH_WRITE_SIZE;
+    if (this.IS_STUB) {
+      return STUB_FLASH_WRITE_SIZE;
     }
     return FLASH_WRITE_SIZE;
   }