roboot/avr/roboot_if_flash_avr.c

Go to the documentation of this file.

/* ================================================================================ *//* ================================================================================= */

#include "roboot.h"
#include <avr/boot.h>
#include <avr/pgmspace.h>

static uint32_t FlashUnlock(uint32_t page_address);
static void     FlashLock(uint32_t page_address);
static uint32_t FlashErasePage(uint32_t page_address);
static uint32_t FlashWriteData(uint32_t write_addr, const uint8_t *chunk, uint32_t chunk_len);

static uint8_t memcmp_prog(const uint8_t* s1, uint32_t s2, size_t len);
static uint8_t read_byte_prog(uint32_t addr);

const ROBOOT_FLASH_Operations_T flash_oper = { FlashUnlock, FlashLock, FlashErasePage, FlashWriteData };


// -----------------------------------------------------------------------------
//  FlashUnlock
// -----------------------------------------------------------------------------
static uint32_t FlashUnlock(uint32_t page_address)
{
    (void) page_address;

    return 0;

} /* FlashUnlock */


// -----------------------------------------------------------------------------
//  FlashLock
// -----------------------------------------------------------------------------
static void FlashLock(uint32_t page_address)
{
    (void) page_address;

} /* FlashLock */


// -----------------------------------------------------------------------------
//  FlashErasePage
// -----------------------------------------------------------------------------
static uint32_t FlashErasePage(uint32_t page_address)
{
    boot_page_erase(page_address);
    boot_spm_busy_wait();
    return 0;
} /* FlashErasePage */


// -----------------------------------------------------------------------------
//  FlashWriteData
// -----------------------------------------------------------------------------
static uint32_t FlashWriteData(uint32_t data_address, const uint8_t *chunk, uint32_t chunk_len)
{
    uint16_t i;
    const uint8_t *flash_data;
    uint16_t check_len;

    flash_data = chunk;

    // adjust chunk_len to even value (FLASH writing possible only for words)
    if (chunk_len & 1U)
    {
        chunk_len++;
    }

    while (chunk_len)
    {
        check_len = (chunk_len > SPM_PAGESIZE) ? SPM_PAGESIZE : chunk_len;
        for (i=0; i < check_len; i += 2)
        {
#if 0
            uint16_t w = *flash_data++;
            w += (*flash_data++) << 8;
            boot_page_fill(data_address+i, w);
#else
            // because GCC stores 16 bits integers in little endian format then it is
            // possible to directly convert byte address to word
            boot_page_fill(data_address+i, *(uint16_t*)flash_data);
            flash_data += 2;
#endif
        }

        boot_page_write(data_address);      // Store buffer in flash page.
        boot_spm_busy_wait();               // Wait until the memory is written.

        // verify just written data
        boot_rww_enable();
        if (memcmp_prog(chunk, data_address, check_len))
        {
            return 1;
        }

        chunk_len -= check_len;
        data_address += SPM_PAGESIZE;
        chunk += SPM_PAGESIZE;
    }

    return 0;
} /* FlashWriteData */


// -----------------------------------------------------------------------------
//  memcmp_prog
// -----------------------------------------------------------------------------
static uint8_t memcmp_prog(const uint8_t* s1, uint32_t s2, size_t len)
{
    while (len)
    {
        if (read_byte_prog(s2) != *s1)
        {
            return 1;
        }
        s1++;
        s2++;
        len--;
    }

    return 0;

} /* memcmp_prog */


// -----------------------------------------------------------------------------
//  read_byte_prog
// -----------------------------------------------------------------------------
static uint8_t read_byte_prog(uint32_t addr)
{
    uint8_t pgm_byte;

#if FLASHEND <= 0xFFFFUL
    pgm_byte = pgm_read_byte_near((uint16_t) addr);
#else
#if 0
    __asm__
    (
        "out %2, %C1"   "\n\t"
        "mov r31, %B1"  "\n\t"
        "mov r30, %A1"  "\n\t"
        "elpm %0, Z"    "\n\t"
        : "=r" (pgm_byte)
        : "r" (addr),
          "I" (_SFR_IO_ADDR(RAMPZ))
        : "r30", "r31"
    );
#else
    pgm_byte = pgm_read_byte_far(addr);
#endif
#endif /* #if FLASHEND <= 0xFFFFUL */

    return pgm_byte;

} /* read_byte_prog */

---