Fix timeout logic when erasing regions.

esp-rs · Aug 20, 2023 · ad37cc3 · ad37cc3
1 parent cbc472d
commit ad37cc3
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/espflash/src/flasher/mod.rs b/espflash/src/flasher/mod.rs
@@ -855,10 +855,10 @@ impl Flasher {
     pub fn erase_region(&mut self, offset: u32, size: u32) -> Result<(), Error> {
         debug!("Erasing region of 0x{:x}B at 0x{:08x}", size, offset);
 
-        self.connection
-            .with_timeout(CommandType::EraseRegion.timeout(), |connection| {
-                connection.command(Command::EraseRegion { offset, size })
-            })?;
+        self.connection.with_timeout(
+            CommandType::EraseRegion.timeout_for_size(size),
+            |connection| connection.command(Command::EraseRegion { offset, size }),
+        )?;
         std::thread::sleep(Duration::from_secs_f32(0.05));
         self.connection.flush()?;
         Ok(())