flash/nor: Allow CFI memory read/write functions be overriden

[openocd.git] / src / flash / nor / cfi.c

diff --git a/src/flash/nor/cfi.c b/src/flash/nor/cfi.c
index 14f9b113040d1305051eba4129d2f9cc89f39480..0437930b25483514a4a45c5a42b4a5c6d047bdc7 100644 (file)
--- a/src/flash/nor/cfi.c
+++ b/src/flash/nor/cfi.c
@@ -134,15 +134,25 @@ static inline uint32_t flash_address(struct flash_bank *bank, int sector, uint32
 static int cfi_target_write_memory(struct flash_bank *bank, target_addr_t addr,
                                   uint32_t count, const uint8_t *buffer)
 {
-       return target_write_memory(bank->target, addr, bank->bus_width,
-                                  count, buffer);
+       struct cfi_flash_bank *cfi_info = bank->driver_priv;
+       if (cfi_info->write_mem) {
+               return cfi_info->write_mem(bank, addr, count, buffer);
+       } else {
+               return target_write_memory(bank->target, addr, bank->bus_width,
+                                          count, buffer);
+       }
 }
 
 static int cfi_target_read_memory(struct flash_bank *bank, target_addr_t addr,
                                  uint32_t count, uint8_t *buffer)
 {
-       return target_read_memory(bank->target, addr, bank->bus_width,
-                                 count, buffer);
+       struct cfi_flash_bank *cfi_info = bank->driver_priv;
+       if (cfi_info->read_mem) {
+               return cfi_info->read_mem(bank, addr, count, buffer);
+       } else {
+               return target_read_memory(bank->target, addr, bank->bus_width,
+                                         count, buffer);
+       }
 }
 
 static void cfi_command(struct flash_bank *bank, uint8_t cmd, uint8_t *cmd_buf)