Subversion Repositories or1k

#include "common.h"

#include "support.h"

#include "flash.h"

#include "net.h"

#include "uart.h"

#include "spr_defs.h"

#ifndef MAX_IMAGES

#define MAX_IMAGES 20

#endif

extern unsigned long fprog_addr;

extern char *tftp_filename;

static flash_cfg_struct __attribute__ ((section(".config"))) gcfg = { 0, 0, 0, 0 };

#define FLASH_IMAGES_BASE 0xf0300000

/* Buffer Address == for example internal RAM */

#define BUF_BASE_ADDR   0x00400000 -  0x00042000

#define BUF_LENGTH      4096

#define ALIGN(addr,size) ((addr + (size-1))&(~(size-1)))

void copy_and_boot(unsigned long src,

                   unsigned long dst,

                   unsigned long len,

                   int tx_next)

{

  __asm__ __volatile__("   ;\

        l.addi  r8,r0,0x1  ;\

        l.mtspr r0,r8,0x11  ;\

        l.nop               ;\

        l.nop               ;\

        l.nop               ;\

        l.nop               ;\

        l.nop               ;\

2:                          ;\

        l.sfgeu r4,r5       ;\

        l.bf    1f          ;\

        l.nop               ;\

        l.lwz   r8,0(r3)    ;\

        l.sw    0(r4),r8    ;\

        l.addi  r3,r3,4     ;\

        l.j     2b          ;\

        l.addi  r4,r4,4     ;\

1:      l.sw    0x0(r0),r6  ;\

        l.ori   r8,r0,0x100 ;\

        l.jr    r8          ;\

        l.nop");

}

/* WARNING: stack and non-const globals should not be used in this function -- it may corrupt what have we loaded;

   start_addr should be 0xffffffff if only copying should be made

   no return, when start_addr != 0xffffffff, if successful */

int copy_memory_run (register unsigned long src_addr, register unsigned long dst_addr, register unsigned long length, register int erase, register unsigned long start_addr)

{

  unsigned long i, flags;

  register char *dst = (char *) dst_addr;

  register const char *src = (const char *) src_addr;

  if (dst_addr >= FLASH_BASE_ADDR) {

    if (dst_addr + length >= FLASH_BASE_ADDR + FLASH_SIZE) {

      printf ("error: region does not fit into flash.\n");

       return 1;

    }

#ifndef CFG_IN_FLASH

    fl_program (src_addr, dst_addr, length, erase, 1 /* do verify */);

#else

    /* we must disable interrupts! */

    flags=mfspr(SPR_SR);

    mtspr(SPR_SR,flags & ~(SPR_SR_TEE | SPR_SR_IEE));

    printf("Unlocking flash... ");

    for(i = 0; i < length; i += FLASH_BLOCK_SIZE)

      fl_ext_unlock(dst_addr + i);

    printf("done\n");

    printf("Erasing flash... ");

    for(i = 0; i < length; i += FLASH_BLOCK_SIZE)

      fl_ext_erase(dst_addr+i);

    printf("done\n");

    printf("Programing flash:\n\t");

    for (i = 0; i < length; i += INC_ADDR) {

      if(((i+INC_ADDR) % 1000) == 0)

        printf("#");

      if((i % (65*1000)) == 0)

        printf("\n\t");

      if (fl_ext_program (dst_addr + i, reg_read(src_addr + i))) {

        printf("error programing at 0x%08lx!\n", dst_addr+i);

        return 1;

      }

    }

    printf("Verifying flash... ");

    for(i = 0; i < length; i += INC_ADDR) {

      if( reg_read(dst_addr+i) != reg_read(src_addr + i)) {

        printf ("error at %08lx: %08lx != %08lx\n", src_addr + i, reg_read(src_addr + i), reg_read(dst_addr + i));

        return 1;

      }

    }

    printf("OK!\n");

    mtspr(SPR_SR, flags);

#endif

    if(start_addr == 0xffffffff)

      return 0;

  }

  else {

    while (length--) *dst++ = *src++;

    if (start_addr == 0xffffffff)

      return 0;

  }

  /* Run the program */

  ((void (*)(void)) start_addr)();

  return 0; /* just to satisfy the cc */

}

void bf_jump(unsigned long addr)

{

  asm("l.jr   r3");

  asm("l.nop  0x0");

}

int boot_flash_cmd(int argc, char *argv[])

{

  unsigned long addr,val,jaddr;

  addr = 17;

  val = 0;

  /* clear SR */

  asm("l.mtspr %0,%1,0": : "r" (addr), "r" (val));

  /* jump */

  if(argc == 0)

    bf_jump(FLASH_BASE_ADDR+0x100);

  else {

    jaddr = strtoul(argv[0], 0, 0);

    bf_jump(jaddr);

  }

  return 0;

}

void

init_load (void)

{

#ifdef CFG_IN_FLASH

  copy_memory_run((unsigned long)&fl_word_program, (unsigned long)&fprog_addr,

                  95, 0, 0xffffffff);

  copy_memory_run((unsigned long)&fl_block_erase, (unsigned long)&fprog_addr+96,

                  119, 0, 0xffffffff);

  copy_memory_run((unsigned long)&fl_unlock_one_block,

                  (unsigned long)&fprog_addr+96+120,

                  115, 0, 0xffffffff);

  fl_ext_program = (t_fl_ext_program)&fprog_addr;

  fl_ext_erase = (t_fl_erase)&fprog_addr+96;

  fl_ext_unlock = (t_fl_erase)&fprog_addr+96+120;

#if 0

  printf("fl_word_program(): 0x%x\tfl_ext_program(): 0x%x\n",

         &fl_word_program, fl_ext_program);

  printf("fl_block_erase: 0x%x\tfl_ext_erase(): 0x%x\n",

         &fl_block_erase, fl_ext_erase);

  printf("fl_unlock_one_block(): 0x%x\tfl_ext_unlock(): 0x%x\n",

         &fl_unlock_one_block, fl_ext_unlock);

#endif

#else /* not CFG_IN_FLASH */

  fl_ext_program = (t_fl_ext_program)&fl_word_program;

  fl_ext_erase = (t_fl_erase)&fl_block_erase;

  fl_ext_unlock = (t_fl_erase)&fl_unlock_one_block;

#endif /* CFG_IN_FLASH */

  global.ip = gcfg.eth_ip;

  global.gw_ip = gcfg.eth_gw;

  global.mask = gcfg.eth_mask;

  global.srv_ip = gcfg.tftp_srv_ip;

  tftp_filename = "boot.img";

  /*memcpy(tftp_filename, gcfg.tftp_filename, strlen(gcfg.tftp_filename));

    tftp_filename[strlen(gcfg.tftp_filename)] = '\0';*/

}

int tftp_cmd (int argc, char *argv[])

{

  switch (argc) {

    case 0: tftp_filename = "boot.img";

                  break;

    case 3: global.src_addr = strtoul (argv[2], 0, 0);

    case 2: global.srv_ip = parse_ip (argv[1]);

    case 1: tftp_filename = &argv[0][0];

            break;

  }

  NetLoop(TFTP);

  return 0;

}

int tftp_conf_cmd(int argc, char *argv[])

{

  switch(argc) {

  case 0:

    printf("Image filename: %s", tftp_filename);

    printf("\nSrc addr: 0x%lx", global.src_addr);

    printf("\nServer IP: %s", inet_ntoa(global.srv_ip));

    return 0;

  case 3:

    global.src_addr = strtoul(argv[2], 0, 0);

    global.srv_ip = inet_aton(argv[1]);

    tftp_filename = argv[0];

    tftp_filename[strlen(argv[0])] = '\0';

    break;

  case 2:

    global.srv_ip = inet_aton(argv[1]);

    tftp_filename = argv[0];

    break;

  case 1:

    tftp_filename = argv[0];

    break;

  }

  return 0;

}

void save_global_cfg(flash_cfg_struct cfg)

{

  unsigned long dst = (unsigned long)&gcfg, src = (unsigned long)&cfg;

  unsigned long i, end, flags;

  end = (unsigned long)&cfg + sizeof(flash_cfg_struct);

  printf("Saving global cfg from 0x%lx (end: 0x%lx) to 0x%lx...", src, end, dst);

  /* we must disable interrupts! */

  flags=mfspr(SPR_SR);

  mtspr(SPR_SR,flags & ~(SPR_SR_TEE | SPR_SR_IEE));

  /*  printf("Unlocking... ");*/

  for(i = 0; (src+i <= end); i += FLASH_BLOCK_SIZE) {

    fl_ext_unlock(dst+i);

  }

  /*  printf("done\n");*/

  /*  printf("Erasing... ");*/

  for(i = 0; (src+i <= end); i += FLASH_BLOCK_SIZE)

    fl_ext_erase(dst);

  /*  printf("done\n");*/

  /*  printf("Programing... ");*/

  for(i = 0; (src+i <= end); i +=INC_ADDR) {

    if(fl_ext_program(dst+i, reg_read(src+i))) {

      printf("Error ocurred while saving.\n");

      return;

    }

  }

  printf("done\n");

  /* and than enable it back */

  mtspr(SPR_SR, flags);

  return;

}

int save_conf_cmd(int argc, char *argv[])

{

  flash_cfg_struct newCfg;

  newCfg = gcfg;

  newCfg.eth_ip = global.ip;

  newCfg.eth_mask = global.mask;

  newCfg.eth_gw = global.gw_ip;

  newCfg.tftp_srv_ip = global.srv_ip;

  /*  memcpy(newCfg.tftp_filename, tftp_filename, strlen(tftp_filename));*/

  save_global_cfg(newCfg);

  return 0;

}

int copy_cmd (int argc, char *argv[])

{

  switch (argc) {

    case 3: global.src_addr = strtoul (argv[2], 0, 0);

    case 2: global.length = strtoul (argv[2], 0, 0);

    case 1: global.src_addr = strtoul (argv[2], 0, 0);

    case 0: return copy_memory_run (global.src_addr, global.dst_addr, global.length,

                                    global.erase_method, 0xffffffff);

  }

  return -1;

}

void

images_info(void)

{

  int i;

  printf("Number of images: 0x%lx\n", gcfg.img_number);

  for(i = 0; i < gcfg.img_number; i++)

    printf("%d. image size: 0x%lx (at 0x%08lx)\n", i+1,

           gcfg.img_length[i], gcfg.img_start_addr[i]);

}

/*

 * get_good_addr()

 *

 * Here we try to find the most suitable place for our image. We search for

 * a hole between images, that is big enough (but as small as possible).

 *

 */

unsigned long

get_good_addr(unsigned int size)

{

  unsigned long start_addr[MAX_IMAGES], end_addr[MAX_IMAGES];

  unsigned long free[MAX_IMAGES], st_addr[MAX_IMAGES];

  unsigned long tmpval;

  unsigned int i = 0, j;

  flash_cfg_struct myCfg;

  myCfg = gcfg;

  /* we are full */

  if(gcfg.img_number == MAX_IMAGES)

    return 0xffffffff;

  if(gcfg.img_number == 0)

    return FLASH_IMAGES_BASE;

  for(i = 0; i < MAX_IMAGES; i++) {

    start_addr[i] = 0;

    end_addr[i] = 0;

    free[i] = 0;

    st_addr[i] = 0;

  }

  for(i = 0; i < myCfg.img_number; i++) {

    start_addr[i] = myCfg.img_start_addr[i];

    end_addr[i] = ALIGN((myCfg.img_start_addr[i] + myCfg.img_length[i]),

                        FLASH_BLOCK_SIZE);

  }

  /*  printf("\n");

  for(i = 0; i < myCfg.img_number; i++)

    printf("start: 0x%08x, end: 0x%08x\n", start_addr[i], end_addr[i]);

    printf("\n");*/

  /* bubble sorting by start_addr */

  for(j = myCfg.img_number - 1; j > 0; j--)

    for(i = 0; i < j; i++)

      if(start_addr[i] > start_addr[i+1]) {

        tmpval = start_addr[i];

        start_addr[i] = start_addr[i+1];

        start_addr[i+1] = tmpval;

        tmpval = end_addr[i];

        end_addr[i] = end_addr[i+1];

        end_addr[i+1] = tmpval;

      }

  /*  for(i = 0; i < myCfg.img_number; i++)

    printf("start: 0x%08x, end: 0x%08x\n", start_addr[i], end_addr[i]);

    printf("\n");*/

  /* now we calculate free space betwens segments */

  for(i = 1; i < myCfg.img_number; i++) {

    st_addr[i] = end_addr[i - 1];

    free[i] = start_addr[i] - end_addr[i - 1];

  }

  /* here we calcuta first position (starting with FLASH_IMAGES_BASE)... */

  st_addr[0] = FLASH_IMAGES_BASE + 0;

  free[0] = start_addr[0] - FLASH_IMAGES_BASE;

  /* ... and last one (ending with FLASH_IMAGES_BASE + FLASH_SIZE). */

  st_addr[myCfg.img_number] = end_addr[myCfg.img_number-1];

  free[myCfg.img_number] = (FLASH_IMAGES_BASE + FLASH_SIZE) - end_addr[myCfg.img_number-1];

  /*  for(i = 0; i < myCfg.img_number+1; i++)

    printf("start: 0x%08x, free: %x\n", st_addr[i], free[i]);

    printf("\n");*/

  /* yet another bubble sort by free (space) */

  for(j = myCfg.img_number; j > 0; j--)

    for(i = 0; i < j; i++)

      if(free[i] > free[i+1]) {

        tmpval = free[i];

        free[i] = free[i+1];

        free[i+1] = tmpval;

        tmpval = st_addr[i];

        st_addr[i] = st_addr[i+1];

        st_addr[i+1] = tmpval;

      }

  /*  for(i = 0; i < myCfg.img_number+1; i++)

    printf("start: 0x%08x, free: %x\n", st_addr[i], free[i]);

    printf("\n");*/

  /* now we pick the smallest but just big enough for our size */

  for(i = 0; i <= myCfg.img_number; i++)

    if(free[i] >= size)

      return st_addr[i];

  /* there is not enough space (in one segment) left */

  return 0;

}

unsigned long

prepare_img_data(unsigned int num, unsigned int size)

{

  int i;

  unsigned long addr=0;

  flash_cfg_struct newCfg;

  newCfg = gcfg;

  if(newCfg.img_number >= MAX_IMAGES) {

    printf("Maximum images exceeded: %d\n", MAX_IMAGES);

    return 0xffffffff;

  }

  newCfg.img_number++;

  if((num > newCfg.img_number) || (num == 0))

    num = newCfg.img_number;

  addr = get_good_addr(size);

  if(addr == 0x00) {

    printf("Can not find suitable place in flash. (None of free segments are big enough)\n");

    return 0xffffffff;

  }

  if(num < newCfg.img_number)

    for(i=newCfg.img_number-1; i >= num; i--) {

      newCfg.img_length[i] = newCfg.img_length[i-1];

      newCfg.img_start_addr[i] = newCfg.img_start_addr[i-1];

    }

  newCfg.img_length[num-1] = size;

  newCfg.img_start_addr[num-1] = addr;

  save_global_cfg(newCfg);

  return addr;

}

int

del_image_cmd(int argc, char *argv[])

{

  unsigned num, i;

  flash_cfg_struct newCfg = gcfg;

  newCfg.img_number = gcfg.img_number;

  for(i = 0; i < MAX_IMAGES; i++)

    newCfg.img_length[i] = gcfg.img_length[i];

  printf("Number of images available: 0x%lx\n", newCfg.img_number);

  if(argc == 0) {

    newCfg.img_number = 0;

    for(i = 0; i < MAX_IMAGES; i++) {

      newCfg.img_length[i] = 0;

      newCfg.img_start_addr[i] = 0;

    }

    save_global_cfg(newCfg);

    return 0;

  }

  else {

    num = strtoul(argv[0], 0, 0);

  }

  if(newCfg.img_number == 0) {

    printf("Nothing to delete!\n");

    return 0;

  }

  if((num == 0) || (num > newCfg.img_number))

    num = newCfg.img_number;

  for(i=num-1; i < newCfg.img_number; i++) {

    newCfg.img_length[i] = newCfg.img_length[i+1];

    newCfg.img_start_addr[i] = newCfg.img_start_addr[i+1];

  }

  newCfg.img_number--;

  save_global_cfg(newCfg);

  return 0;

}

int

boot_cmd(int argc, char *argv[])

{

  int num;

  extern int tx_next;

  if(argc == 0) {

    images_info();

    return 0;

  }

  num = strtoul(argv[0],0,0);

  if(gcfg.img_number < num) {

    printf("There are only %lu images, you requested %d!\n", gcfg.img_number, num);

    return -1;

  }

  printf("Copying image number %d from 0x%lx, size: 0x%lx...",

         num, gcfg.img_start_addr[num-1], gcfg.img_length[num-1]);

  printf("booting...\n");

  copy_and_boot(gcfg.img_start_addr[num-1], 0x0, gcfg.img_length[num-1], tx_next);

  return 0;

}

int mGetData(unsigned long);

int sboot_cmd (int argc, char *argv[])

{

  int copied;

  unsigned int num = 0xffffffff, addr = 0x0;

  switch(argc) {

  case 0:

    num = 0xffffffff;

    break;

  case 1:

    num = strtoul(argv[0], 0, 0);

    break;

  }

  copied = mGetData(global.src_addr);

  if(copied <= 0) {

    printf("sboot: error while getting the image!");

    return -1;

  }

  printf("image size: 0x%x\n", copied);

  if(num != 0xffffffff) {

    addr = prepare_img_data(num, copied);

    if(addr == 0xffffffff)

      printf("Image not written to flash!\n");

    else {

      printf("Copying image to flash, image number: %d, dst_addr: 0x%x\n",

             num, addr);

      copy_memory_run(global.src_addr, gcfg.img_start_addr[num-1], copied, 2, 0xffffffff);

    }

  }

  return 0;

}

int tboot_cmd (int argc, char *argv[])

{

  int copied;

  unsigned int num = 0xffffffff, addr = 0x0;

  extern int tx_next;

  switch (argc) {

  case 0:

    num = 0xffffffff;

    break;

  case 1:

    printf("argv[0] %p\n", argv[0]);

    num = strtoul(argv[0], 0, 0);

    printf("num %d\n", num);

    break;

  }

  //  global.src_addr = (unsigned long)0x0;

  copied =NetLoop(TFTP);

  if (copied <= 0) {

    printf("tboot: error while getting the image '%s'",

           tftp_filename);

    return -1;

  }

  if(num != 0xffffffff) {

    addr = prepare_img_data(num, copied);

    if(addr == 0xffffffff)

      printf("Image not written to flash!\n");

    else {

      printf("Copying image to flash, image number: %d, dst_addr: 0x%x\n",

             num, addr);

      copy_memory_run(global.src_addr, gcfg.img_start_addr[num-1], copied, 2, 0xffffffff);

    }

  }

  /* the point of no return */

  printf("tboot: copying 0x%lx -> 0x0, image size 0x%x...\n",

         global.src_addr, copied);

  printf("tboot: jumping to 0x100, booting image ...\n");

  copy_and_boot(global.src_addr, 0x0, 0x0 + copied, tx_next);

  return 0;

}

void module_load_init (void)

{

  register_command ("tftp", "[<file> [<srv_ip> [<src_addr>]]]",  "TFTP download", tftp_cmd);

  register_command ("tftp_conf", "[ <file> [ <srv_ip> [ <src_addr>]]]", "TFTP configuration", tftp_conf_cmd);

  register_command ("copy", "[<dst_addr> [<src_addr [<length>]]]", "Copy memory", copy_cmd);

  register_command ("tboot", "[<image number>]", "Bootstrap image downloaded via tftp", tboot_cmd);

  register_command ("sboot", "[<image number>]", "Bootstrap image downloaded via serial (Y/X modem)", sboot_cmd);

  register_command ("boot", "[<image number>]", "Bootstrap image copied from flash.", boot_cmd);

  register_command ("del_image", "[<image number>]", "Delete image", del_image_cmd);

  register_command ("save_conf", "", "Save current configuration into flash", save_conf_cmd);

  register_command ("boot_flash", "[<start_addr>]", "Boot image from <start_addr> (default from flash)", boot_flash_cmd);

  init_load();

}