libc/mtdoutstream: Fix the in insufficient erase in byte write case

Signed-off-by: Xiang Xiao <xiaoxiang@xiaomi.com>
apache · Nov 4, 2024 · 11960fd · 11960fd
1 parent cccc413
commit 11960fd
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/libs/libc/stream/lib_mtdoutstream.c b/libs/libc/stream/lib_mtdoutstream.c
@@ -104,9 +104,12 @@ static int mtdoutstream_puts(FAR struct lib_outstream_s *self,
 #ifdef CONFIG_MTD_BYTE_WRITE
   if (inode->u.i_mtd->write != NULL)
     {
-      if (self->nput % erasesize == 0)
+      size_t sblock = (self->nput + erasesize - 1) / erasesize;
+      size_t eblock = (self->nput + len + erasesize - 1) / erasesize;
+
+      if (sblock != eblock)
         {
-          ret = MTD_ERASE(inode->u.i_mtd, self->nput / erasesize, 1);
+          ret = MTD_ERASE(inode->u.i_mtd, sblock, eblock - sblock);
           if (ret < 0)
             {
               return ret;