Add missing page erase.
[bertos.git] / bertos / fs / battfs.c
index 423319063abd9747b7256d1e5742ae0dd2da6cbe..9c72f74928c024bf331df8bde3d5118ba3d81482 100644 (file)
@@ -156,8 +156,8 @@ static bool battfs_writeHeader(struct BattFsSuper *disk, pgcnt_t page, struct Ba
         * Header is actually a footer, and so
         * resides at page end.
         */
-       if (disk->write(disk, page, disk->page_size - BATTFS_HEADER_LEN, buf, BATTFS_HEADER_LEN)
-           != BATTFS_HEADER_LEN)
+       if (!(disk->bufferWrite(disk, disk->page_size - BATTFS_HEADER_LEN, buf, BATTFS_HEADER_LEN)
+           == BATTFS_HEADER_LEN && disk->save(disk, page)))
        {
                LOG_ERR("Error: page[%d]\n", page);
                return false;
@@ -246,11 +246,10 @@ static bool countDiskFilePages(struct BattFsSuper *disk, pgoff_t *filelen_table)
  * inside file.
  * e.g. : at page array[0] you will find page address of the first page
  * of the first file (if present).
- * Free blocks are allocated after the last file, starting from invalid ones
- * and continuing with the marked free ones.
+ * Free blocks are allocated after the last file.
  *
  * \return true if ok, false on disk read errors.
- * \note The whole disk is scanned once.
+ * \note The whole disk is scanned at max twice.
  */
 static bool fillPageArray(struct BattFsSuper *disk, pgoff_t *filelen_table)
 {
@@ -343,6 +342,49 @@ static bool fillPageArray(struct BattFsSuper *disk, pgoff_t *filelen_table)
        return true;
 }
 
+
+/**
+ * Flush the current \a disk buffer.
+ * \return true if ok, false on errors.
+ */
+static bool battfs_flushBuffer(struct BattFsSuper *disk)
+{
+       if (disk->cache_dirty)
+       {
+               LOG_INFO("Flushing to disk page %d\n", disk->curr_page);
+               if (!disk->erase(disk, disk->curr_page))
+                       return false;
+
+               if (!disk->save(disk, disk->curr_page))
+                       return false;
+               disk->cache_dirty = false;
+       }
+       return true;
+}
+
+/**
+ * Load \a new_page from \a disk in disk page buffer.
+ * If a previuos page is still dirty in the buffer, will be
+ * flushed first.
+ * \return true if ok, false on errors.
+ */
+static bool battfs_loadPage(struct BattFsSuper *disk, pgcnt_t new_page)
+{
+       LOG_INFO("Loading page %d\n", new_page);
+       if (disk->curr_page == new_page)
+               return true;
+
+       if (!battfs_flushBuffer(disk))
+               return false;
+
+       if (!disk->load(disk, new_page))
+               return false;
+
+       disk->curr_page = new_page;
+       return true;
+}
+
+
 /**
  * Initialize and mount disk described by
  * \a disk.
@@ -364,7 +406,9 @@ bool battfs_init(struct BattFsSuper *disk)
 
        /* Disk open must set all of these */
        ASSERT(disk->read);
-       ASSERT(disk->write);
+       ASSERT(disk->load);
+       ASSERT(disk->bufferWrite);
+       ASSERT(disk->save);
        ASSERT(disk->erase);
        ASSERT(disk->close);
        ASSERT(disk->page_size);
@@ -396,6 +440,12 @@ bool battfs_init(struct BattFsSuper *disk)
                LOG_ERR("error filling page array\n");
                return false;
        }
+       #warning TODO: shuffle free blocks
+
+       /* Initialize page buffer cache */
+       disk->cache_dirty = false;
+       disk->curr_page = 0;
+       disk->load(disk, disk->curr_page);
 
        /* Init list for opened files. */
        LIST_INIT(&disk->file_opened_list);
@@ -408,9 +458,12 @@ bool battfs_init(struct BattFsSuper *disk)
  */
 static int battfs_flush(struct KFile *fd)
 {
-       (void)fd;
-       #warning TODO
-       return 0;
+       BattFs *fdb = BATTFS_CAST(fd);
+
+       if (battfs_flushBuffer(fdb->disk))
+               return 0;
+       else
+               return EOF;
 }
 
 /**
@@ -426,6 +479,69 @@ static int battfs_fileclose(struct KFile *fd)
        return 0;
 }
 
+
+/**
+ * Write to file \a fd \a size bytes from \a buf.
+ * \return The number of bytes written.
+ */
+static size_t battfs_write(struct KFile *fd, const void *_buf, size_t size)
+{
+       BattFs *fdb = BATTFS_CAST(fd);
+       const uint8_t *buf = (const uint8_t *)_buf;
+
+       size_t total_write = 0;
+       pgoff_t pg_offset;
+       pgaddr_t addr_offset;
+       pgaddr_t wr_len;
+
+       size = MIN((kfile_off_t)size, fd->size - fd->seek_pos);
+
+       while (size)
+       {
+               #warning TODO: outside EOF?
+
+               pg_offset = fd->seek_pos / (fdb->disk->page_size - BATTFS_HEADER_LEN);
+               addr_offset = fd->seek_pos % (fdb->disk->page_size - BATTFS_HEADER_LEN);
+               wr_len = MIN(size, (size_t)(fdb->disk->page_size - BATTFS_HEADER_LEN - addr_offset));
+
+
+               if (fdb->start[pg_offset] != fdb->disk->curr_page)
+               {
+                       if (!battfs_loadPage(fdb->disk, fdb->start[pg_offset]))
+                       {
+                               #warning TODO set error?
+                               return total_write;
+                       }
+
+                       /* Get a free page */
+                       fdb->disk->curr_page = fdb->disk->page_array[fdb->disk->free_page_start];
+                       movePages(fdb->disk, fdb->disk->free_page_start + 1, -1);
+
+                       /* Insert previous page in free blocks list */
+                       fdb->disk->page_array[fdb->disk->page_count - 1] = fdb->start[pg_offset];
+                       /* Assign new page */
+                       fdb->start[pg_offset] = fdb->disk->curr_page;
+                       #warning TODO: hdr have to be updated!
+               }
+
+
+               if (fdb->disk->bufferWrite(fdb->disk, addr_offset, buf, wr_len) != wr_len)
+               {
+                       #warning TODO set error?
+               }
+               fdb->disk->cache_dirty = true;
+
+               size -= wr_len;
+               fd->seek_pos += wr_len;
+               total_write += wr_len;
+               buf += wr_len;
+               #warning TODO: hdr have to be updated!
+       }
+       return total_write;
+
+}
+
+
 /**
  * Read from file \a fd \a size bytes in \a buf.
  * \return The number of bytes read.
@@ -448,6 +564,10 @@ static size_t battfs_read(struct KFile *fd, void *_buf, size_t size)
                addr_offset = fd->seek_pos % (fdb->disk->page_size - BATTFS_HEADER_LEN);
                read_len = MIN(size, (size_t)(fdb->disk->page_size - BATTFS_HEADER_LEN - addr_offset));
 
+               /* Flush current page if needed */
+               if (fdb->start[pg_offset] == fdb->disk->curr_page)
+                       battfs_flushBuffer(fdb->disk);
+
                /* Read from disk */
                if (fdb->disk->read(fdb->disk, fdb->start[pg_offset], addr_offset, buf, read_len) != read_len)
                {
@@ -471,10 +591,10 @@ static size_t battfs_read(struct KFile *fd, void *_buf, size_t size)
 static pgcnt_t *findFile(BattFsSuper *disk, inode_t inode)
 {
        BattFsPageHeader hdr;
-       pgcnt_t first = 0, page, last = disk->page_count -1;
+       pgcnt_t first = 0, page, last = disk->free_page_start;
        fcs_t fcs;
 
-       while (first <= last)
+       while (first < last)
        {
                page = (first + last) / 2;
 
@@ -506,19 +626,19 @@ bool battfs_fileExists(BattFsSuper *disk, inode_t inode)
  * in disk->page_array. Size is written in \a size.
  * \return true if all s ok, false on disk read errors.
  */
-static bool countFileSize(BattFsSuper *disk, pgcnt_t *start, inode_t inode, file_size_t *size)
+static file_size_t countFileSize(BattFsSuper *disk, pgcnt_t *start, inode_t inode)
 {
-       *size = 0;
+       file_size_t size = 0;
        BattFsPageHeader hdr;
 
        for (;;)
        {
                if (!battfs_readHeader(disk, *start++, &hdr))
-                       return false;
+                       return EOF;
                if (hdr.fcs == computeFcs(&hdr) && hdr.inode == inode)
-                       *size += hdr.fill;
+                       size += hdr.fill;
                else
-                       return true;
+                       return size;
        }
 }
 
@@ -551,7 +671,7 @@ bool battfs_fileopen(BattFsSuper *disk, BattFs *fd, inode_t inode, filemode_t mo
        }
 
        /* Fill file size */
-       if (!countFileSize(disk, fd->start, inode, &fd->fd.size))
+       if ((fd->fd.size = countFileSize(disk, fd->start, inode)) == EOF)
                return false;
 
        /* Reset seek position */
@@ -576,10 +696,10 @@ bool battfs_fileopen(BattFsSuper *disk, BattFs *fd, inode_t inode, filemode_t mo
        fd->fd.read = battfs_read;
        fd->fd.reopen = kfile_genericReopen;
        fd->fd.seek = kfile_genericSeek;
+       fd->fd.write = battfs_write;
 
-#warning TODO battfs_write, battfs_error, battfs_clearerr
+#warning TODO battfs_error, battfs_clearerr
 #if 0
-       fd->fd.write = battfs_write;
        fd->fd.error = battfs_error;
        fd->fd.clearerr = battfs_clearerr;
 #endif
@@ -608,10 +728,16 @@ bool battfs_close(struct BattFsSuper *disk)
        return disk->close(disk) && (res == 0);
 }
 
+#if UNIT_TEST
 bool battfs_writeTestBlock(struct BattFsSuper *disk, pgcnt_t page, inode_t inode, seq_t seq, fill_t fill, pgoff_t pgoff)
 {
        BattFsPageHeader hdr;
 
+       /* Reset page to all 0xff */
+       uint8_t buf[disk->page_size];
+       memset(buf, 0xFF, disk->page_size);
+       disk->bufferWrite(disk, 0, buf, disk->page_size);
+
        hdr.inode = inode;
        hdr.fill = fill;
        hdr.pgoff = pgoff;
@@ -626,3 +752,4 @@ bool battfs_writeTestBlock(struct BattFsSuper *disk, pgcnt_t page, inode_t inode
 
        return true;
 }
+#endif