X-Git-Url: https://codewiz.org/gitweb?a=blobdiff_plain;f=bertos%2Ffs%2Fbattfs.c;h=7d30231f68726c0f4f03299780a27ba038f785e5;hb=4ce77ecc12cbc60cdb8717755006f6f461e6f6ad;hp=59b03ae4f6eb0c4b846c15e4c57475e0400df0fd;hpb=802cec6861e5f1988f23085b2a6849d86368628c;p=bertos.git

diff --git a/bertos/fs/battfs.c b/bertos/fs/battfs.c
index 59b03ae4..7d30231f 100644
--- a/bertos/fs/battfs.c
+++ b/bertos/fs/battfs.c
@@ -50,6 +50,19 @@
 
 #include <string.h> /* memset, memmove */
 
+#if LOG_LEVEL >= LOG_LVL_INFO
+static void dumpPageArray(struct BattFsSuper *disk)
+{
+	kprintf("Page array dump, free_page_start %d:", disk->free_page_start);
+	for (pgcnt_t i = 0; i < disk->page_count; i++)
+	{
+		if (!(i % 16))
+			kputchar('\n');
+		kprintf("%04d ", disk->page_array[i]);
+	}
+	kputchar('\n');
+}
+#endif
 
 /**
  * Convert from memory representation to disk structure.
@@ -114,12 +127,27 @@ static fcs_t computeFcs(struct BattFsPageHeader *hdr)
 	return cks;
 }
 
+/**
+ * Read from disk.
+ * If available, the cache will be used.
+ * \return the number of bytes read.
+ */
+static size_t diskRead(struct BattFsSuper *disk, pgcnt_t page, pgaddr_t addr, void *buf, size_t size)
+{
+	/* Try to read from cache */
+	if (page == disk->curr_page)
+		return disk->bufferRead(disk, addr, buf, size);
+	/* Read from disk */
+	else
+		return disk->read(disk, page, addr, buf, size);
+}
+
 
 /**
- * Read header of page \a page.
+ * Read header of \a page in \a hdr.
  * \return true on success, false otherwise.
  */
-static bool battfs_readHeader(struct BattFsSuper *disk, pgcnt_t page, struct BattFsPageHeader *hdr)
+static bool readHdr(struct BattFsSuper *disk, pgcnt_t page, struct BattFsPageHeader *hdr)
 {
 	uint8_t buf[BATTFS_HEADER_LEN];
 	/*
@@ -127,10 +155,10 @@ static bool battfs_readHeader(struct BattFsSuper *disk, pgcnt_t page, struct Bat
 	 * Header is actually a footer, and so
 	 * resides at page end.
 	 */
-	if (disk->read(disk, page, disk->page_size - BATTFS_HEADER_LEN, buf, BATTFS_HEADER_LEN)
+	if (diskRead(disk, page, disk->data_size, buf, BATTFS_HEADER_LEN)
 	    != BATTFS_HEADER_LEN)
 	{
-		LOG_ERR("Error: page[%d]\n", page);
+		LOG_ERR("page[%d]\n", page);
 		return false;
 	}
 
@@ -141,30 +169,48 @@ static bool battfs_readHeader(struct BattFsSuper *disk, pgcnt_t page, struct Bat
 }
 
 /**
- * Write header of page \a page.
+ * Set header on current \a disk page buffer.
  * \return true on success, false otherwise.
  */
-static bool battfs_writeHeader(struct BattFsSuper *disk, pgcnt_t page, struct BattFsPageHeader *hdr)
+static bool setBufferHdr(struct BattFsSuper *disk, struct BattFsPageHeader *hdr)
 {
 	uint8_t buf[BATTFS_HEADER_LEN];
 
+	#warning FIXME:refactor computeFcs to save time and stack
+	hdr->fcs = computeFcs(hdr);
 	/* Fill buffer */
 	battfs_to_disk(hdr, buf);
 
 	/*
-	 * write header to disk.
+	 * write header to buffer.
 	 * Header is actually a footer, and so
 	 * resides at page end.
 	 */
-	if (!(disk->bufferWrite(disk, disk->page_size - BATTFS_HEADER_LEN, buf, BATTFS_HEADER_LEN)
-	    == BATTFS_HEADER_LEN && disk->save(disk, page)))
+	if (disk->bufferWrite(disk, disk->data_size, buf, BATTFS_HEADER_LEN)
+	    != BATTFS_HEADER_LEN)
 	{
-		LOG_ERR("Error: page[%d]\n", page);
+		LOG_ERR("writing to buffer\n");
 		return false;
 	}
 	return true;
 }
 
+static bool getBufferHdr(struct BattFsSuper *disk, struct BattFsPageHeader *hdr)
+{
+	uint8_t buf[BATTFS_HEADER_LEN];
+
+	if (disk->bufferRead(disk, disk->data_size, buf, BATTFS_HEADER_LEN)
+	    != BATTFS_HEADER_LEN)
+	{
+		LOG_ERR("reading from buffer\n");
+		return false;
+	}
+
+	disk_to_battfs(buf, hdr);
+
+	return true;
+}
+
 /**
  * Count the number of pages from
  * inode 0 to \a inode in \a filelen_table.
@@ -186,6 +232,7 @@ static pgcnt_t countPages(pgoff_t *filelen_table, inode_t inode)
 static void movePages(struct BattFsSuper *disk, pgcnt_t src, int offset)
 {
 	pgcnt_t dst = src + offset;
+	LOG_INFO("src %d, offset %d, size %ld\n", src, offset, (disk->page_count - MAX(dst, src)) * sizeof(pgcnt_t));
 	memmove(&disk->page_array[dst], &disk->page_array[src], (disk->page_count - MAX(dst, src)) * sizeof(pgcnt_t));
 
 	if (offset < 0)
@@ -213,16 +260,16 @@ static bool countDiskFilePages(struct BattFsSuper *disk, pgoff_t *filelen_table)
 	/* Count the number of disk page per file */
 	for (pgcnt_t page = 0; page < disk->page_count; page++)
 	{
-		if (!battfs_readHeader(disk, page, &hdr))
+		if (!readHdr(disk, page, &hdr))
 			return false;
 
 		/* Increase free space */
-		disk->free_bytes += disk->page_size - BATTFS_HEADER_LEN;
+		disk->free_bytes += disk->data_size;
 
 		/* Check header FCS */
 		if (hdr.fcs == computeFcs(&hdr))
 		{
-			ASSERT(hdr.fill <= disk->page_size - BATTFS_HEADER_LEN);
+			ASSERT(hdr.fill <= disk->data_size);
 
 			/* Page is valid and is owned by a file */
 			filelen_table[hdr.inode]++;
@@ -258,7 +305,7 @@ static bool fillPageArray(struct BattFsSuper *disk, pgoff_t *filelen_table)
 	/* Fill page allocation array */
 	for (pgcnt_t page = 0; page < disk->page_count; page++)
 	{
-		if (!battfs_readHeader(disk, page, &hdr))
+		if (!readHdr(disk, page, &hdr))
 			return false;
 
 		/* Check header FCS */
@@ -276,7 +323,7 @@ static bool fillPageArray(struct BattFsSuper *disk, pgoff_t *filelen_table)
 			{
 				BattFsPageHeader hdr_prv;
 
-				if (!battfs_readHeader(disk, disk->page_array[array_pos], &hdr_prv))
+				if (!readHdr(disk, disk->page_array[array_pos], &hdr_prv))
 					return false;
 
 				/* Check header FCS */
@@ -342,6 +389,50 @@ 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 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)
+			&& 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. The new page header loaded will be put in \a new_hdr.
+ * \return true if ok, false on errors.
+ */
+static bool loadPage(struct BattFsSuper *disk, pgcnt_t new_page, BattFsPageHeader *new_hdr)
+{
+	if (disk->curr_page == new_page)
+		return getBufferHdr(disk, new_hdr);
+
+	LOG_INFO("Loading page %d\n", new_page);
+
+	if (!(flushBuffer(disk)
+		&& disk->load(disk, new_page)
+		&& getBufferHdr(disk, new_hdr)))
+		return false;
+
+	disk->curr_page = new_page;
+
+	return true;
+}
+
+
 /**
  * Initialize and mount disk described by
  * \a disk.
@@ -365,10 +456,13 @@ bool battfs_init(struct BattFsSuper *disk)
 	ASSERT(disk->read);
 	ASSERT(disk->load);
 	ASSERT(disk->bufferWrite);
+	ASSERT(disk->bufferRead);
 	ASSERT(disk->save);
 	ASSERT(disk->erase);
 	ASSERT(disk->close);
-	ASSERT(disk->page_size);
+	ASSERT(disk->page_size > BATTFS_HEADER_LEN);
+	/* Fill page_size with the usable space */
+	disk->data_size = disk->page_size - BATTFS_HEADER_LEN;
 	ASSERT(disk->page_count);
 	ASSERT(disk->page_count < PAGE_UNSET_SENTINEL - 1);
 	ASSERT(disk->page_array);
@@ -376,12 +470,17 @@ bool battfs_init(struct BattFsSuper *disk)
 	memset(filelen_table, 0, BATTFS_MAX_FILES * sizeof(pgoff_t));
 
 	disk->free_bytes = 0;
-	disk->disk_size = (disk_size_t)(disk->page_size - BATTFS_HEADER_LEN) * disk->page_count;
+	disk->disk_size = (disk_size_t)disk->data_size * disk->page_count;
+
+	/* Initialize page buffer cache */
+	disk->cache_dirty = false;
+	disk->curr_page = 0;
+	disk->load(disk, disk->curr_page);
 
 	/* Count pages per file */
 	if (!countDiskFilePages(disk, filelen_table))
 	{
-		LOG_ERR("error counting file pages\n");
+		LOG_ERR("counting file pages\n");
 		return false;
 	}
 
@@ -394,24 +493,88 @@ bool battfs_init(struct BattFsSuper *disk)
 	/* Fill page allocation array using filelen_table */
 	if (!fillPageArray(disk, filelen_table))
 	{
-		LOG_ERR("error filling page array\n");
+		LOG_ERR("filling page array\n");
 		return false;
 	}
-
+	#if LOG_LEVEL >= LOG_LVL_INFO
+		dumpPageArray(disk);
+	#endif
+	#warning TODO: shuffle free blocks
+	//#if LOG_LEVEL > LOG_LVL_INFO
+	//	dumpPageArray(disk);
+	//#endif
 	/* Init list for opened files. */
 	LIST_INIT(&disk->file_opened_list);
 	return true;
 }
 
+/**
+ * Check the filesystem.
+ * \return true if ok, false on errors.
+ */
+bool battfs_fsck(struct BattFsSuper *disk)
+{
+	#define FSCHECK(cond) do { if(!(cond)) { LOG_ERR("\"" #cond "\"\n"); return false; } } while (0)
+
+	FSCHECK(disk->free_page_start <= disk->page_count);
+	FSCHECK(disk->data_size < disk->page_size);
+	FSCHECK(disk->free_bytes <= disk->disk_size);
+
+	disk_size_t free_bytes = 0;
+	BattFsPageHeader hdr, prev_hdr;
+	inode_t files = 0;
+	pgcnt_t page_used = 0;
+
+	bool start = true;
+
+	for (pgcnt_t page = 0; page < disk->page_count; page++)
+	{
+		FSCHECK(readHdr(disk, disk->page_array[page], &hdr));
+		free_bytes += disk->data_size;
+
+		if (page < disk->free_page_start)
+		{
+			FSCHECK(computeFcs(&hdr) == hdr.fcs);
+			page_used++;
+			free_bytes -= hdr.fill;
+			if (hdr.inode != prev_hdr.inode || start)
+			{
+				if (LIKELY(!start))
+					FSCHECK(hdr.inode > prev_hdr.inode);
+				else
+					start = false;
+
+				FSCHECK(hdr.pgoff == 0);
+				files++;
+			}
+			else
+			{
+				FSCHECK(hdr.fill != 0);
+				FSCHECK(prev_hdr.fill == disk->data_size);
+				FSCHECK(hdr.pgoff == prev_hdr.pgoff + 1);
+			}
+			prev_hdr = hdr;
+		}
+	}
+
+	FSCHECK(page_used == disk->free_page_start);
+	FSCHECK(free_bytes == disk->free_bytes);
+
+	return true;
+}
+
 /**
  * Flush file \a fd.
  * \return 0 if ok, EOF on errors.
  */
 static int battfs_flush(struct KFile *fd)
 {
-	(void)fd;
-	#warning TODO
-	return 0;
+	BattFs *fdb = BATTFS_CAST(fd);
+
+	if (flushBuffer(fdb->disk))
+		return 0;
+	else
+		return EOF;
 }
 
 /**
@@ -427,6 +590,189 @@ static int battfs_fileclose(struct KFile *fd)
 	return 0;
 }
 
+
+static bool getNewPage(struct BattFsSuper *disk, pgcnt_t new_pos, inode_t inode, pgoff_t pgoff, BattFsPageHeader *new_hdr)
+{
+	if (SPACE_OVER(disk))
+	{
+		#warning TODO space over!
+		LOG_ERR("No disk space available!\n");
+		return false;
+	}
+	flushBuffer(disk);
+	LOG_INFO("Getting new page %d, pos %d\n", disk->page_array[disk->free_page_start], new_pos);
+	disk->curr_page = disk->page_array[disk->free_page_start++];
+	memmove(&disk->page_array[new_pos + 1], &disk->page_array[new_pos], (disk->free_page_start - new_pos - 1) * sizeof(pgcnt_t));
+
+	Node *n;
+	/* Move following file start point one position ahead. */
+	FOREACH_NODE(n, &disk->file_opened_list)
+	{
+		BattFs *file = containerof(n, BattFs, link);
+		if (file->inode > inode)
+		{
+			LOG_INFO("Move file %d start pos\n", file->inode);
+			file->start++;
+		}
+	}
+
+	disk->page_array[new_pos] = disk->curr_page;
+	disk->cache_dirty = true;
+
+	new_hdr->inode = inode;
+	new_hdr->pgoff =  pgoff;
+	new_hdr->fill = 0;
+	new_hdr->seq = 0;
+	return setBufferHdr(disk, new_hdr);
+}
+
+/**
+ * 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;
+	BattFsPageHeader curr_hdr;
+
+	if (fd->seek_pos < 0)
+		return total_write;
+
+	if (fd->seek_pos > fd->size)
+	{
+		/* Handle writing when seek pos if far over EOF */
+		if (!loadPage(fdb->disk, fdb->start[fdb->max_off], &curr_hdr))
+		{
+				#warning TODO set error?
+				return total_write;
+		}
+
+		/* Fill unused space of first page with 0s */
+		uint8_t dummy = 0;
+		pgaddr_t zero_bytes = MIN(fd->seek_pos - fd->size, fdb->disk->data_size - curr_hdr.fill);
+		while (zero_bytes--)
+		{
+			if (fdb->disk->bufferWrite(fdb->disk, curr_hdr.fill, &dummy, 1) != 1)
+			{
+				#warning TODO set error?
+			}
+			curr_hdr.fill++;
+			fd->size++;
+			fdb->disk->free_bytes--;
+			fdb->disk->cache_dirty = true;
+		}
+		setBufferHdr(fdb->disk, &curr_hdr);
+
+		/* Allocate the missing pages first. */
+		pgoff_t missing_pages = fd->seek_pos / fdb->disk->data_size - fdb->max_off;
+
+		if (missing_pages)
+		{
+			LOG_INFO("missing pages: %d\n", missing_pages);
+			flushBuffer(fdb->disk);
+
+			/* Fill page buffer with 0 to avoid filling unused pages with garbage */
+			for (pgaddr_t off = 0; off < fdb->disk->data_size; off++)
+			{
+				if (fdb->disk->bufferWrite(fdb->disk, off, &dummy, 1) != 1)
+				{
+					#warning TODO set error?
+				}
+			}
+
+			while (missing_pages--)
+			{
+				zero_bytes = MIN((kfile_off_t)fdb->disk->data_size, fd->seek_pos - fd->size);
+				/* Get the new page needed */
+				if (!getNewPage(fdb->disk, (fdb->start - fdb->disk->page_array) + fdb->max_off + 1, fdb->inode, fdb->max_off + 1, &curr_hdr))
+					return total_write;
+
+				/* Update size and free space left */
+				fd->size += zero_bytes;
+				fdb->disk->free_bytes -= zero_bytes;
+
+				curr_hdr.fill = zero_bytes;
+				setBufferHdr(fdb->disk, &curr_hdr);
+
+				fdb->max_off++;
+			}
+		}
+	}
+	else if (!getBufferHdr(fdb->disk, &curr_hdr))
+		return total_write;
+
+	while (size)
+	{
+		pg_offset = fd->seek_pos / fdb->disk->data_size;
+		addr_offset = fd->seek_pos % fdb->disk->data_size;
+		wr_len = MIN(size, (size_t)(fdb->disk->data_size - addr_offset));
+
+		/* Handle write outside EOF */
+		if (pg_offset > fdb->max_off)
+		{
+			LOG_INFO("New page needed, pg_offset %d, pos %d\n", pg_offset, (fdb->start - fdb->disk->page_array) + pg_offset);
+			if (!getNewPage(fdb->disk, (fdb->start - fdb->disk->page_array) + pg_offset, fdb->inode, pg_offset, &curr_hdr))
+				return total_write;
+			fdb->max_off = pg_offset;
+		}
+		/* Handle cache load of a new page*/
+		else if (fdb->start[pg_offset] != fdb->disk->curr_page)
+		{
+			if (SPACE_OVER(fdb->disk))
+			{
+				LOG_ERR("No disk space available!\n");
+				return total_write;
+			}
+			LOG_INFO("Re-writing page %d to %d\n", fdb->start[pg_offset], fdb->disk->page_array[fdb->disk->free_page_start]);
+			if (!loadPage(fdb->disk, fdb->start[pg_offset], &curr_hdr))
+			{
+				#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 */
+			LOG_INFO("Setting page %d as free\n", fdb->start[pg_offset]);
+			fdb->disk->page_array[fdb->disk->page_count - 1] = fdb->start[pg_offset];
+			/* Assign new page */
+			fdb->start[pg_offset] = fdb->disk->curr_page;
+			curr_hdr.seq++;
+		}
+
+		//LOG_INFO("writing to buffer for page %d, offset %d, size %d\n", fdb->disk->curr_page, addr_offset, wr_len);
+		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;
+		fill_t fill_delta = MAX((int32_t)(addr_offset + wr_len) - curr_hdr.fill, (int32_t)0);
+		fdb->disk->free_bytes -= fill_delta;
+		fd->size += fill_delta;
+		curr_hdr.fill += fill_delta;
+
+		if (!setBufferHdr(fdb->disk, &curr_hdr))
+			return total_write;
+
+		//LOG_INFO("free_bytes %d, seek_pos %d, size %d, curr_hdr.fill %d\n", fdb->disk->free_bytes, fd->seek_pos, fd->size, curr_hdr.fill);
+	}
+	return total_write;
+}
+
+
 /**
  * Read from file \a fd \a size bytes in \a buf.
  * \return The number of bytes read.
@@ -441,20 +787,30 @@ static size_t battfs_read(struct KFile *fd, void *_buf, size_t size)
 	pgaddr_t addr_offset;
 	pgaddr_t read_len;
 
-	size = MIN((kfile_off_t)size, fd->size - fd->seek_pos);
+	if (fd->seek_pos < 0)
+		return total_read;
+
+	size = MIN((kfile_off_t)size, MAX(fd->size - fd->seek_pos, 0));
 
 	while (size)
 	{
-		pg_offset = fd->seek_pos / (fdb->disk->page_size - BATTFS_HEADER_LEN);
-		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));
+		pg_offset = fd->seek_pos / fdb->disk->data_size;
+		addr_offset = fd->seek_pos % fdb->disk->data_size;
+		read_len = MIN(size, (size_t)(fdb->disk->data_size - addr_offset));
 
+		//LOG_INFO("reading from page %d, offset %d, size %d\n", fdb->start[pg_offset], addr_offset, read_len);
 		/* Read from disk */
-		if (fdb->disk->read(fdb->disk, fdb->start[pg_offset], addr_offset, buf, read_len) != read_len)
+		if (diskRead(fdb->disk, fdb->start[pg_offset], addr_offset, buf, read_len) != read_len)
 		{
 			#warning TODO set error?
 		}
 
+		#if _DEBUG
+			BattFsPageHeader hdr;
+			readHdr(fdb->disk, fdb->start[pg_offset], &hdr);
+			ASSERT(hdr.inode == fdb->inode);
+		#endif
+
 		size -= read_len;
 		fd->seek_pos += read_len;
 		total_read += read_len;
@@ -466,32 +822,40 @@ static size_t battfs_read(struct KFile *fd, void *_buf, size_t size)
 
 /**
  * Search file \a inode in \a disk using a binary search.
- * \return pointer to file start in disk->page_array
- * if file exists, NULL otherwise.
+ * \a last is filled with array offset of file start
+ * in disk->page_array if file is found, otherwise
+ * \a last is filled with the correct insert position
+ * for creating a file with the given \a inode.
+ * \return true if file is found, false otherwisr.
  */
-static pgcnt_t *findFile(BattFsSuper *disk, inode_t inode)
+static bool findFile(BattFsSuper *disk, inode_t inode, pgcnt_t *last)
 {
 	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;
-
-		if (!battfs_readHeader(disk, disk->page_array[page], &hdr))
-			return NULL;
-
+		page = (first + *last) / 2;
+		LOG_INFO("first %d, last %d, page %d\n", first, *last, page);
+		if (!readHdr(disk, disk->page_array[page], &hdr))
+			return false;
+		LOG_INFO("inode read: %d\n", hdr.inode);
 		fcs = computeFcs(&hdr);
 		if (hdr.fcs == fcs && hdr.inode == inode)
-			return (&disk->page_array[page]) - hdr.pgoff;
+		{
+			*last = page - hdr.pgoff;
+			LOG_INFO("Found: %d\n", *last);
+			return true;
+		}
 		else if (hdr.fcs == fcs && hdr.inode < inode)
 			first = page + 1;
 		else
-			last = page - 1;
+			*last = page - 1;
 	}
-
-	return NULL;
+	LOG_INFO("Not found: last %d\n", *last);
+	return false;
 }
 
 /**
@@ -499,7 +863,8 @@ static pgcnt_t *findFile(BattFsSuper *disk, inode_t inode)
  */
 bool battfs_fileExists(BattFsSuper *disk, inode_t inode)
 {
-	return findFile(disk, inode) != NULL;
+	pgcnt_t dummy;
+	return findFile(disk, inode, &dummy);
 }
 
 /**
@@ -507,20 +872,21 @@ 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 (;;)
+	while (start < &disk->page_array[disk->free_page_start])
 	{
-		if (!battfs_readHeader(disk, *start++, &hdr))
-			return false;
+		if (!readHdr(disk, *start++, &hdr))
+			return EOF;
 		if (hdr.fcs == computeFcs(&hdr) && hdr.inode == inode)
-			*size += hdr.fill;
+			size += hdr.fill;
 		else
-			return true;
+			break;
 	}
+	return size;
 }
 
 /**
@@ -535,25 +901,24 @@ bool battfs_fileopen(BattFsSuper *disk, BattFs *fd, inode_t inode, filemode_t mo
 	memset(fd, 0, sizeof(*fd));
 
 	/* Search file start point in disk page array */
-	fd->start = findFile(disk, inode);
-	if (fd->start == NULL)
+	pgcnt_t start_pos;
+	if (!findFile(disk, inode, &start_pos))
 	{
+		LOG_INFO("file %d not found\n", inode);
 		if (!(mode & BATTFS_CREATE))
 			return false;
-
-		/* File does not exist, create it */
+		/* Create the file */
 		BattFsPageHeader hdr;
-		hdr.inode = inode;
-		hdr.seq = 0;
-		hdr.fill = 0;
-		hdr.pgoff = 0;
-		hdr.fcs = computeFcs(&hdr);
-		#warning TODO: get a free block and write on disk!
+		if (!(getNewPage(disk, start_pos, inode, 0, &hdr)))
+			return false;
 	}
+	fd->start = &disk->page_array[start_pos];
+	LOG_INFO("Start pos %d\n", start_pos);
 
 	/* Fill file size */
-	if (!countFileSize(disk, fd->start, inode, &fd->fd.size))
+	if ((fd->fd.size = countFileSize(disk, fd->start, inode)) == EOF)
 		return false;
+	fd->max_off = fd->fd.size / disk->data_size;
 
 	/* Reset seek position */
 	fd->fd.seek_pos = 0;
@@ -577,10 +942,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
@@ -610,17 +975,23 @@ bool battfs_close(struct BattFsSuper *disk)
 }
 
 #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;
 	hdr.seq = seq;
 	hdr.fcs = computeFcs(&hdr);
 
-	if (!battfs_writeHeader(disk, page, &hdr))
+	if (!(setBufferHdr(disk, &hdr) && disk->save(disk, page)))
 	{
 		LOG_ERR("error writing hdr\n");
 		return false;