X-Git-Url: https://codewiz.org/gitweb?a=blobdiff_plain;f=bertos%2Fcpu%2Fcortex-m3%2Fdrv%2Fflash_stm32.c;h=3b3dd79d2d42047da622ecdb9c4feaa6a87461b1;hb=574d08f7aba9cfec59053ddcb7d2af3ebd0fda64;hp=68b3cdaff630e58af8906c8ea8b023e8f34e362d;hpb=6b88c2593ab913484e8b8f88fd65a23831acf4fe;p=bertos.git

diff --git a/bertos/cpu/cortex-m3/drv/flash_stm32.c b/bertos/cpu/cortex-m3/drv/flash_stm32.c
index 68b3cdaf..3b3dd79d 100644
--- a/bertos/cpu/cortex-m3/drv/flash_stm32.c
+++ b/bertos/cpu/cortex-m3/drv/flash_stm32.c
@@ -165,6 +165,9 @@ INLINE bool stm32_writeWord(struct KBlock *blk, uint32_t addr, uint16_t data)
 
 static size_t stm32_flash_writeDirect(struct KBlock *blk, block_idx_t idx, const void *_buf, size_t offset, size_t size)
 {
+	ASSERT(offset == 0);
+	ASSERT(size == blk->blk_size);
+
 	if (!stm32_erasePage(blk, (idx * blk->blk_size)))
 		return 0;
 
@@ -173,7 +176,7 @@ static size_t stm32_flash_writeDirect(struct KBlock *blk, block_idx_t idx, const
 
 	while (size)
 	{
-		uint16_t data = ((*buf + 1) << 8) | *buf;
+		uint16_t data = (*(buf + 1) << 8) | *buf;
 		if (!stm32_writeWord(blk, addr, data))
 			return 0;
 
@@ -195,6 +198,8 @@ static const KBlockVTable flash_stm32_buffered_vt =
 	.load = kblock_swLoad,
 	.store = kblock_swStore,
 
+	.close = kblock_swClose,
+
 	.error = stm32_flash_error,
 	.clearerr = stm32_flash_clearerror,
 };
@@ -204,6 +209,8 @@ static const KBlockVTable flash_stm32_unbuffered_vt =
 	.readDirect = stm32_flash_readDirect,
 	.writeDirect = stm32_flash_writeDirect,
 
+	.close = kblock_swClose,
+
 	.error = stm32_flash_error,
 	.clearerr = stm32_flash_clearerror,
 };
@@ -227,16 +234,21 @@ static void common_init(Flash *fls)
 }
 
 
-void flash_hw_init(Flash *fls)
+void flash_hw_init(Flash *fls, int flags)
 {
 	common_init(fls);
 	fls->blk.priv.vt = &flash_stm32_buffered_vt;
-	fls->blk.priv.flags |= KB_BUFFERED | KB_PARTIAL_WRITE;
+	fls->blk.priv.flags |= KB_BUFFERED | KB_PARTIAL_WRITE | flags;
 	fls->blk.priv.buf = flash_buf;
+
+	/* Load the first block in the cache */
+	void *flash_start = 0x0;
+	memcpy(fls->blk.priv.buf, flash_start, fls->blk.blk_size);
 }
 
-void flash_hw_initUnbuffered(Flash *fls)
+void flash_hw_initUnbuffered(Flash *fls, int flags)
 {
 	common_init(fls);
 	fls->blk.priv.vt = &flash_stm32_unbuffered_vt;
+	fls->blk.priv.flags |= flags;
 }