bertos/cpu/avr/drv/flash_avr.c

   1 /**
   2  * \file
   3  * <!--
   4  * This file is part of BeRTOS.
   5  *
   6  * Bertos is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  19  *
  20  * As a special exception, you may use this file as part of a free software
  21  * library without restriction.  Specifically, if other files instantiate
  22  * templates or use macros or inline functions from this file, or you compile
  23  * this file and link it with other files to produce an executable, this
  24  * file does not by itself cause the resulting executable to be covered by
  25  * the GNU General Public License.  This exception does not however
  26  * invalidate any other reasons why the executable file might be covered by
  27  * the GNU General Public License.
  28  *
  29  * Copyright 2007 Develer S.r.l. (http://www.develer.com/)
  30  *
  31  * -->
  32  *
  33  * \brief Self programming routines.
  34  *
  35  * \version $Id$
  36  * \author Francesco Sacchi <batt@develer.com>
  37  * \author Daniele Basile <asterix@develer.com>
  38  *
  39  * This module implements a kfile-like access for Atmel avr
  40  * internal flash.
  41  * Internal flash writing access is controlled by BOOTSZ fuses, check
  42  * datasheet for details.
  43  */
  44
  45 #include "flash_avr.h"
  46
  47 #include "cfg/cfg_flash_avr.h"
  48 #include <cfg/macros.h> // MIN()
  49 #include <cfg/compiler.h>
  50 #include <cfg/debug.h>
  51 #include <cpu/irq.h>
  52
  53 #include <drv/wdt.h>
  54
  55 #include <kern/kfile.h>
  56
  57 #include <avr/io.h>
  58 #include <avr/boot.h>
  59 #include <avr/pgmspace.h>
  60
  61 #include <string.h>
  62
  63
  64
  65 /**
  66  * Definition of type for avr flash module.
  67  */
  68 typedef uint16_t avr_page_addr_t;
  69
  70
  71 /**
  72  * Temporary buffer cointaing data block to
  73  * write on flash.
  74  */
  75 static uint8_t page_buf[SPM_PAGESIZE];
  76
  77 /**
  78  * Flag for checking if current page is modified.
  79  */
  80 bool page_modified;
  81
  82
  83
  84 /*
  85  * Private avr flush funtion.
  86  *
  87  * Write current buffered page in flash memory (if modified).
  88  * This function erase flash memory page before writing.
  89  *
  90  * This function is only use internaly in this module.
  91  */
  92 static void flash_avr_flush(KFileFlashAvr *fd)
  93 {
  94         if (page_modified)
  95         {
  96                 kprintf("Flushing page %d\n", fd->curr_page);
  97
  98                 // Wait while the SPM instruction is busy.
  99                 boot_spm_busy_wait();
 100
 101                 kprintf("Filling temparary page buffer...");
 102
 103                 // Fill the temporary buffer of the AVR
 104                 for (avr_page_addr_t page_addr = 0; page_addr < SPM_PAGESIZE; page_addr += 2)
 105                 {
 106                         uint16_t word = ((uint16_t)page_buf[page_addr + 1] << 8) | page_buf[page_addr];
 107
 108                         ATOMIC(boot_page_fill(page_addr, word));
 109                 }
 110                 kprintf("Done.\n");
 111
 112                 wdt_reset();
 113
 114                 kprintf("Erasing page, addr %u...", fd->curr_page * SPM_PAGESIZE);
 115
 116                 /* Page erase */
 117                 ATOMIC(boot_page_erase(fd->curr_page * SPM_PAGESIZE));
 118
 119                 /* Wait until the memory is erased. */
 120                 boot_spm_busy_wait();
 121
 122                 kprintf("Done.\n");
 123                 kprintf("Writing page, addr %u...", fd->curr_page * SPM_PAGESIZE);
 124
 125                 /* Store buffer in flash page. */
 126                 ATOMIC(boot_page_write(fd->curr_page * SPM_PAGESIZE));
 127                 boot_spm_busy_wait();  // Wait while the SPM instruction is busy.
 128
 129                 /*
 130                 * Reenable RWW-section again. We need this if we want to jump back
 131                 * to the application after bootloading.
 132                 */
 133                 ATOMIC(boot_rww_enable());
 134
 135                 page_modified = false;
 136                 kprintf("Done.\n");
 137         }
 138 }
 139
 140
 141 /**
 142  * Flush avr flash function.
 143  *
 144  * Write current buffered page in flash memory (if modified).
 145  * This function erase flash memory page before writing.
 146  */
 147 static int flash_avr_kfileFlush(struct KFile *_fd)
 148 {
 149         KFileFlashAvr *fd = KFILEFLASHAVR(_fd);
 150         flash_avr_flush(fd);
 151         return 0;
 152 }
 153
 154
 155 /**
 156  * Check current page and if \a page is different, load it in
 157  * temporary buffer.
 158  */
 159 static void flash_avr_loadPage(KFileFlashAvr *fd, avr_page_t page)
 160 {
 161         if (page != fd->curr_page)
 162         {
 163                 flash_avr_flush(fd);
 164                 // Load page
 165                 memcpy_P(page_buf, (const char *)(page * SPM_PAGESIZE), SPM_PAGESIZE);
 166                 fd->curr_page = page;
 167                 kprintf("Loaded page %d\n", fd->curr_page);
 168         }
 169 }
 170
 171 /**
 172  * Write program memory.
 173  * Write \a size bytes from buffer \a _buf to file \a fd
 174  * \note Write operations are buffered.
 175  */
 176 static size_t flash_avr_write(struct KFile *_fd, const void *_buf, size_t size)
 177 {
 178         KFileFlashAvr *fd = KFILEFLASHAVR(_fd);
 179         const uint8_t *buf =(const uint8_t *)_buf;
 180
 181         avr_page_t page;
 182         avr_page_addr_t page_addr;
 183         size_t total_write = 0;
 184
 185
 186         ASSERT(fd->fd.seek_pos + (kfile_off_t)size <= (kfile_off_t)fd->fd.size);
 187         size = MIN((uint32_t)size, fd->fd.size - fd->fd.seek_pos);
 188
 189         kprintf("Writing at pos[%u]\n", fd->fd.seek_pos);
 190         while (size)
 191         {
 192                 page = fd->fd.seek_pos / SPM_PAGESIZE;
 193                 page_addr = fd->fd.seek_pos % SPM_PAGESIZE;
 194
 195                 flash_avr_loadPage(fd, page);
 196
 197                 size_t wr_len = MIN(size, SPM_PAGESIZE - page_addr);
 198                 memcpy(page_buf + page_addr, buf, wr_len);
 199                 page_modified = true;
 200
 201                 buf += wr_len;
 202                 fd->fd.seek_pos += wr_len;
 203                 size -= wr_len;
 204                 total_write += wr_len;
 205         }
 206         kprintf("written %u bytes\n", total_write);
 207         return total_write;
 208 }
 209
 210 /**
 211  * Open flash file \a fd
 212  * \a name and \a mode are unused, cause flash memory is
 213  * threated like one file.
 214  */
 215 static void flash_avr_open(struct KFileFlashAvr *fd)
 216 {
 217         fd->curr_page = 0;
 218         memcpy_P(page_buf, (const char *)(fd->curr_page * SPM_PAGESIZE), SPM_PAGESIZE);
 219
 220         fd->fd.seek_pos = 0;
 221         fd->fd.size = (uint16_t)(FLASHEND - CONFIG_FLASH_AVR_BOOTSIZE + 1);
 222         page_modified = false;
 223
 224         kprintf("Flash file opened\n");
 225 }
 226
 227 /**
 228  * Close file \a fd
 229  */
 230 static int flash_avr_close(struct KFile *_fd)
 231 {
 232         KFileFlashAvr *fd = KFILEFLASHAVR(_fd);
 233         flash_avr_flush(fd);
 234         kprintf("Flash file closed\n");
 235         return 0;
 236 }
 237
 238 /**
 239  * Reopen file \a fd
 240  */
 241 static struct KFile *flash_avr_reopen(struct KFile *fd)
 242 {
 243         KFileFlashAvr *_fd = KFILEFLASHAVR(fd);
 244         flash_avr_close(fd);
 245         flash_avr_open(_fd);
 246         return fd;
 247 }
 248
 249
 250 /**
 251  * Read from file \a fd \a size bytes and put it in buffer \a buf
 252  * \return the number of bytes read.
 253  */
 254 static size_t flash_avr_read(struct KFile *_fd, void *buf, size_t size)
 255 {
 256         KFileFlashAvr *fd = KFILEFLASHAVR(_fd);
 257         ASSERT(fd->fd.seek_pos + (kfile_off_t)size <= (kfile_off_t)fd->fd.size);
 258         size = MIN((uint32_t)size, fd->fd.size - fd->fd.seek_pos);
 259
 260         kprintf("Reading at pos[%u]\n", fd->fd.seek_pos);
 261         // Flush current buffered page (if modified).
 262         flash_avr_flush(fd);
 263
 264         /*
 265          * AVR pointers are 16 bits wide, this hack is needed to avoid
 266          * compiler warning, cause fd->seek_pos is a 32bit offset.
 267          */
 268         const uint8_t *pgm_addr = (const uint8_t *)0;
 269         pgm_addr += fd->fd.seek_pos;
 270
 271         memcpy_P(buf, pgm_addr, size);
 272         fd->fd.seek_pos += size;
 273         kprintf("Read %u bytes\n", size);
 274         return size;
 275 }
 276
 277 /**
 278  * Init AVR flash read/write file.
 279  */
 280 void flash_avr_init(struct KFileFlashAvr *fd)
 281 {
 282         memset(fd, 0, sizeof(*fd));
 283         DB(fd->fd._type = KFT_FLASHAVR);
 284
 285         // Set up flash programming functions.
 286         fd->fd.reopen = flash_avr_reopen;
 287         fd->fd.close = flash_avr_close;
 288         fd->fd.read = flash_avr_read;
 289         fd->fd.write = flash_avr_write;
 290         fd->fd.seek = kfile_genericSeek;
 291         fd->fd.flush = flash_avr_kfileFlush;
 292         fd->curr_page = 0;
 293
 294         flash_avr_open(fd);
 295 }
 296
 297