lib_xua/module_dfu/src/flash_interface.c

#include <xs1.h>
#include <flash.h>
#include <flashlib.h>
#include <print.h>
#include <string.h>
#include <xclib.h>

#ifndef FLASH_MAX_UPGRADE_SIZE 
#ifdef ARCH_G
#define FLASH_MAX_UPGRADE_SIZE 128 * 1024 // 128K default
#else
//#define FLASH_MAX_UPGRADE_SIZE 64 * 1024 // 64K default
#define FLASH_MAX_UPGRADE_SIZE 128 * 1024 // 64K default
#endif
#endif

#if 1
#define FLASH_ERROR() do { printstr("Error: line: "); printintln(__LINE__); __builtin_trap(); } while(0)
#else
#define FLASH_ERROR() do {} while(0)
#endif

static int flash_device_open = 0;
static fl_BootImageInfo factory_image;
static fl_BootImageInfo upgrade_image;

static int upgrade_image_valid = 0;
static int current_flash_subpage_index = 0;
static unsigned char current_flash_page_data[256];

int flash_cmd_enable_ports() __attribute__ ((weak));
int flash_cmd_enable_ports() {
  return 0;
}

int flash_cmd_disable_ports() __attribute__ ((weak));
int flash_cmd_disable_ports() {
  return 0;
}

void DFUCustomFlashEnable() __attribute__ ((weak));
void DFUCustomFlashEnable()
{
    return;
}


void DFUCustomFlashDisable() __attribute__ ((weak));
void DFUCustomFlashDisable()
{
    return;
}


int flash_cmd_init(void) {
  fl_BootImageInfo image;

  if (!flash_device_open) {
    if (flash_cmd_enable_ports());
      flash_device_open = 1;
  }

  if (!flash_device_open)
    return 0;

  // Disable flash protection
  fl_setProtection(0);

  if (fl_getFactoryImage(&image) != 0) {
    return 0;
  }
  factory_image = image;

  if (fl_getNextBootImage(&image) == 0) {
    upgrade_image_valid = 1;
    upgrade_image = image;
  }

  return 0;
}

int flash_cmd_deinit(void) {
  if (!flash_device_open)
    return 0;
 
  flash_cmd_disable_ports();
  flash_device_open = 0;
  return 0;
}

int flash_cmd_read_page(unsigned char *data) {
  if (!upgrade_image_valid) {
    *(unsigned int *)data = 1;
    return 4;
  }
  if (*(unsigned int *)data == 0) {
    fl_startImageRead(&upgrade_image);
  }
  current_flash_subpage_index = 0;
  if (fl_readImageRead(current_flash_page_data) == 0) {
    *(unsigned int *)data = 0;
  } else {
    *(unsigned int *)data = 1;
  }
  return 4;
}

int flash_cmd_read_page_data(unsigned char *data) {
  unsigned char *page_data_ptr = &current_flash_page_data[current_flash_subpage_index * 64];
  memcpy(data, page_data_ptr, 64);

  current_flash_subpage_index++;

  return 64;
}

#if 0
static int roundUpToSectorBoundary(unsigned address)
{
  unsigned curAddress = 0;
  unsigned numSectors = fl_getNumSectors();
  unsigned i;
  for (i = 0; i < numSectors; i++) {
    curAddress += fl_getSectorSize(i);
    if (curAddress >= address)
      return curAddress;
  }
  return address;
}
#endif

static void begin_write()
{
  int result;
  // TODO this will take a long time. To minimise the amount of time spent
  // paused on this operation it would be preferable to move to this to a
  // seperate command, e.g. start_write.
  do {
    result = fl_startImageAdd(&factory_image, FLASH_MAX_UPGRADE_SIZE, 0);
  } while (result > 0);
  if (result < 0)
    FLASH_ERROR();
}

static int pages_written = 0;

int flash_cmd_write_page(unsigned char *data) {
  unsigned int flag = *(unsigned int *)data;
  //printstr("flash_cmd_write_page(");
  //printint(flag);
  //printstr(")\n");

  if (upgrade_image_valid) {
    return 0;
  }

  switch (flag) {
  case 0:
    // First page.
    begin_write();
    pages_written = 0;
    // fallthrough
  case 1:
    // Do nothing.
    break;
  case 2:
    // Termination.
    if (fl_endWriteImage() != 0)
      FLASH_ERROR();
    // Sanity check
    fl_BootImageInfo image = factory_image;
    if (fl_getNextBootImage(&image) != 0)
      FLASH_ERROR();
    break;
  }
  current_flash_subpage_index = 0;

  return 0;
}

static int isAllOnes(unsigned char page[256])
{
  unsigned i;
  for (i = 0; i < 256; i++) {
    if (page[i] != 0xff)
      return 0;
  }
  return 1;
}

int flash_cmd_write_page_data(unsigned char *data) {
  unsigned char *page_data_ptr = &current_flash_page_data[current_flash_subpage_index * 64];
  //printstr("flash_cmd_write_page_data()\n");

  if (upgrade_image_valid) {
    return 0;
  }

  if (current_flash_subpage_index >= 4) {
    return 0;
  }

  memcpy(page_data_ptr, data, 64);

  current_flash_subpage_index++;

  if (current_flash_subpage_index == 4) {
    if (isAllOnes(data))
      FLASH_ERROR();
    if (fl_writeImagePage(current_flash_page_data) != 0) 
      FLASH_ERROR();
    pages_written++;
  }

  return 0;
}

int flash_cmd_erase_all(void) {
  if (upgrade_image_valid) {
    if (fl_deleteImage(&upgrade_image) != 0)
      FLASH_ERROR();
    upgrade_image_valid = 0;
  }
  return 0;
}

int flash_cmd_reboot(void) {
  //unsigned int pllVal;
  //read_sswitch_reg(0, 6, &pllVal);
  //write_sswitch_reg(0, 6, pllVal);
  return 0;
}