X-Git-Url: https://codewiz.org/gitweb?a=blobdiff_plain;f=bertos%2Fio%2Fkblock.c;h=7355deeded97884c6c010decd54a54f7580d8a83;hb=8d34c3b835bcdf6f01d3c15631abc62a713608fd;hp=32c36cf356b5a10cc7188f774da589b7e2abe296;hpb=1e69f4f5c7650a866f6231c48e2b41f31a583025;p=bertos.git

diff --git a/bertos/io/kblock.c b/bertos/io/kblock.c
index 32c36cf3..7355deed 100644
--- a/bertos/io/kblock.c
+++ b/bertos/io/kblock.c
@@ -38,35 +38,181 @@
 
 #include "kblock.h"
 
+#define LOG_LEVEL LOG_LVL_ERR
+#define LOG_FORMAT LOG_FMT_VERBOSE
 
-static void *kblock_swMap(struct KBlock *b, size_t offset, size_t size)
+#include <cfg/log.h>
+#include <string.h>
+
+INLINE size_t kblock_readDirect(struct KBlock *b, block_idx_t index, void *buf, size_t offset, size_t size)
 {
-	return (kblock_readBuf(b, b->priv.pagebuf, offset, size) == size) ? b->priv.pagebuf : NULL;
+	KB_ASSERT_METHOD(b, readDirect);
+	ASSERT(index < b->blk_cnt);
+	return b->priv.vt->readDirect(b, b->priv.blk_start + index, buf, offset, size);
 }
 
+INLINE int kblock_writeBlock(struct KBlock *b, block_idx_t index, const void *buf)
+{
+	KB_ASSERT_METHOD(b, writeBlock);
+	ASSERT(index < b->blk_cnt);
+	return b->priv.vt->writeBlock(b, b->priv.blk_start + index, buf);
+}
 
-static int kblock_swUnmap(struct KBlock *b, size_t offset, size_t size)
+INLINE size_t kblock_readBuf(struct KBlock *b, void *buf, size_t offset, size_t size)
 {
-	return (kblock_writeBuf(b, b->priv.pagebuf, offset, size) == size) ? 0 : EOF;
+	KB_ASSERT_METHOD(b, readBuf);
+	ASSERT(offset + size <= b->blk_size);
+
+	return b->priv.vt->readBuf(b, buf, offset, size);
 }
 
+INLINE size_t kblock_writeBuf(struct KBlock *b, const void *buf, size_t offset, size_t size)
+{
+	KB_ASSERT_METHOD(b, writeBuf);
+	ASSERT(offset + size <= b->blk_size);
+	return b->priv.vt->writeBuf(b, buf, offset, size);
+}
 
-void *kblock_unsupportedMap(struct KBlock *b, UNUSED_ARG(size_t, offset), UNUSED_ARG(size_t, size))
+INLINE int kblock_load(struct KBlock *b, block_idx_t index)
 {
-	LOG_WARN("This driver does not support block mapping: use kblock_addMapping() to add generic mapping functionality.\n");
-	b->priv.flags |= BV(KBS_ERR_MAP_NOT_AVAILABLE);
-	return NULL;
+	KB_ASSERT_METHOD(b, load);
+	ASSERT(index < b->blk_cnt);
+
+	LOG_INFO("index %d\n", index);
+	return b->priv.vt->load(b, b->priv.blk_start + index);
+}
+
+INLINE int kblock_store(struct KBlock *b, block_idx_t index)
+{
+	KB_ASSERT_METHOD(b, store);
+	ASSERT(index < b->blk_cnt);
+
+	LOG_INFO("index %d\n", index);
+	return b->priv.vt->store(b, b->priv.blk_start + index);
+}
+
+INLINE void kblock_setDirty(struct KBlock *b, bool dirty)
+{
+	if (dirty)
+		b->priv.flags |= KB_CACHE_DIRTY;
+	else
+		b->priv.flags &= ~KB_CACHE_DIRTY;
+}
+
+
+
+size_t kblock_read(struct KBlock *b, block_idx_t idx, void *buf, size_t offset, size_t size)
+{
+	ASSERT(b);
+	ASSERT(buf);
+	ASSERT(offset + size <= b->blk_size);
+	LOG_INFO("blk_idx %d, offset %d, size %d\n", idx, offset, size);
+
+	if (kblock_buffered(b) && idx == b->priv.curr_blk)
+		return kblock_readBuf(b, buf, offset, size);
+	else
+		return kblock_readDirect(b, idx, buf, offset, size);
+}
+
+
+int kblock_flush(struct KBlock *b)
+{
+	ASSERT(b);
+
+	if (!kblock_buffered(b))
+		return 0;
+
+	if (kblock_cacheDirty(b))
+	{
+		LOG_INFO("flushing block %d\n", b->priv.curr_blk);
+		if (kblock_store(b, b->priv.curr_blk) == 0)
+			kblock_setDirty(b, false);
+		else
+			return EOF;
+	}
+	return 0;
+}
+
+
+static bool kblock_loadPage(struct KBlock *b, block_idx_t idx)
+{
+	ASSERT(b);
+
+	if (idx != b->priv.curr_blk)
+	{
+		LOG_INFO("loading block %d\n", idx);
+		if (kblock_flush(b) != 0 || kblock_load(b, idx) != 0)
+				return false;
+
+		b->priv.curr_blk = idx;
+	}
+	return true;
+}
+
+
+size_t kblock_write(struct KBlock *b, block_idx_t idx, const void *buf, size_t offset, size_t size)
+{
+	ASSERT(b);
+	ASSERT(buf);
+	ASSERT(idx < b->blk_cnt);
+	ASSERT(offset + size <= b->blk_size);
+
+	LOG_INFO("blk_idx %d, offset %d, size %d\n", idx, offset, size);
+
+	if (kblock_buffered(b))
+	{
+		if (!kblock_loadPage(b, idx))
+			return 0;
+
+		kblock_setDirty(b, true);
+		return kblock_writeBuf(b, buf, offset, size);
+	}
+	else
+	{
+		ASSERT(offset == 0);
+		ASSERT(size == b->blk_size);
+		return (kblock_writeBlock(b, idx, buf) == 0) ? size : 0;
+	}
+}
+
+int kblock_copy(struct KBlock *b, block_idx_t src, block_idx_t dest)
+{
+	ASSERT(b);
+	ASSERT(src < b->blk_cnt);
+	ASSERT(dest < b->blk_cnt);
+	ASSERT(kblock_buffered(b));
+
+	if (!kblock_loadPage(b, src))
+		return EOF;
+
+	b->priv.curr_blk = dest;
+	kblock_setDirty(b, true);
+	return 0;
+}
+
+int kblock_swLoad(struct KBlock *b, block_idx_t index)
+{
+	return (kblock_readDirect(b, index, b->priv.buf, 0, b->blk_size) == b->blk_size) ? 0 : EOF;
+}
+
+int kblock_swStore(struct KBlock *b, block_idx_t index)
+{
+	return kblock_writeBlock(b, index, b->priv.buf);
+}
+
+size_t kblock_swReadBuf(struct KBlock *b, void *buf, size_t offset, size_t size)
+{
+	ASSERT(buf);
+	ASSERT(offset + size <= b->blk_size);
+
+	memcpy(buf, (uint8_t *)b->priv.buf + offset, size);
+	return size;
 }
 
-void kblock_addMapping(struct KBlock *dev, void *buf, size_t size)
+size_t kblock_swWriteBuf(struct KBlock *b, const void *buf, size_t offset, size_t size)
 {
 	ASSERT(buf);
-	ASSERT(size);
-	ASSERT(dev);
-	
-	dev->vt->map = kblock_swMap;
-	dev->vt->unmap = kblock_swUnmap;
-
-	dev->priv.pagebuf = buf;
-	dev->priv.pagebuf_size = size;
+	ASSERT(offset + size <= b->blk_size);
+	memcpy((uint8_t *)b->priv.buf + offset, buf, size);
+	return size;
 }