Subversion Repositories openrisc_2011-10-31

#include "common.h"

#include "support.h"

#include "net.h"

#include "uart.h"

#include "spr-defs.h"

#include "flash.h"

#define SOH    0x01

#define STX    0x02

#define EOT    0x04

#define ACK    0x06

#define NAK    0x15

#define CAN    0x18

#define C      0x43

#define CPMEOF 0x1A

#define RETRY   10

#define TIMEOUT 3

/* update CRC */

unsigned short

updcrc(register int c, register unsigned int crc)

{

  register int count;

  for (count=8; --count>=0;) {

    if (crc & 0x8000) {

      crc <<= 1;

      crc += (((c<<=1) & 0400)  !=  0);

      crc ^= 0x1021;

    }

    else {

      crc <<= 1;

      crc += (((c<<=1) & 0400)  !=  0);

    }

  }

  return crc;

}

static thand_f *mTimeHandler;

static unsigned long mTimeValue;

static int mTimeoutCount;

static unsigned long src_addr;

unsigned int length;

unsigned int modemMode = 0, bLen = 128;

unsigned short mycrc;

unsigned long bno = 0;

static void mStartTimeout(void);

void

mSetTimeout(int iv, thand_f *f)

{

  if(iv == 0)

    mTimeHandler = (thand_f *)0;

  else {

    mTimeHandler = f;

    mTimeValue = get_timer(0) + iv;

  }

}

static void

mStartTimeout(void)

{

  if(++mTimeoutCount >= RETRY) {

    printf("...retry counter exceeded, quitting...\n");

    return;

  } else {

    printf(".");

    mSetTimeout(TIMEOUT * TICKS_PER_SEC, mStartTimeout);

    uart_putc(C);

  }

}

static void

mReceiveTimeout(void)

{

  if(++mTimeoutCount >= RETRY) {

    uart_putc(NAK);

    printf("...");

    return;

  } else {

    mSetTimeout(TIMEOUT * TICKS_PER_SEC, mReceiveTimeout);

    uart_putc(NAK);

  }

}

int

getBlock(unsigned char t)

{

  unsigned int i = 0, j = 0;

  unsigned char mybuf[133];

  unsigned char bNo, nBno;

  unsigned long dst_addr;

  unsigned char flags;

  flags = UART_LSR_FE | UART_LSR_PE | UART_LSR_OE | UART_LSR_BI; /*frame,parity,overrun errors */

  mycrc = 0;

  switch(t) {

  case SOH:

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

      if((REG8(UART_BASE + UART_LSR) & flags) == flags) {

        uart_putc(CAN);

      }

      mybuf[i] = uart_getc();

    }

    bNo = mybuf[0]; /* packet id */

    nBno = mybuf[1]; /* neg. packet id */

    if((bNo == 0x00) && (nBno == 0xff) && (bno == 0)) { /* start block */

      modemMode = 2; /* ymodem */

      uart_putc(ACK);

      uart_putc(C);

      return 1;

    }

    else if((0xff-bNo) == nBno) { /* data block */

      for(i = 2, j = 0; i < 130; i++, j++) {

        length++;

        mycrc = updcrc(mybuf[i], mycrc);

        dst_addr = src_addr+(bno*0x8000)+((bNo-1)*0x80)+j;

        REG8(dst_addr) = mybuf[i];

      }

      mycrc = updcrc(mybuf[130], mycrc);

      mycrc = updcrc(mybuf[131], mycrc);

      if(mycrc == 0) /* CRC match! */ {

        uart_putc(ACK);

        for(i=0;i < 128; i+=4) {

        /*      for(j=0; j<4; j++) {

          tmp = tmp << 8;

          tmp |= mybuf[i+j+2];

        }

        dst_addr = src_addr+(bno*0x8000)+((bNo-1)*128)+i;

        fl_word_program(dst_addr, tmp);*/

        }

        if(bNo == 0xff)

          bno++;

        return 1;

      }

      else {

        uart_putc(NAK);

        return -1;

      }

    }

    else { /* packet id didn't match neg packet id! */

      uart_putc(NAK);

      return -1;

    }

    return 1;

    break;

  case EOT:

    if(modemMode == 2) {/* ymodem */

      uart_putc(NAK);

      if(uart_getc() == EOT) {

        uart_putc(ACK);

        uart_putc(C);

      }

      else

        uart_putc(ACK);

    } else /* zmodem */

      uart_putc(ACK);

    return 0;

    break;

  default:

    /* Unknown header */

    uart_putc(NAK);

    return -1;

  }

}

int

mGetData(unsigned long saddr)

{

  int retval = 1;

  unsigned char c;

  length = 0;

  src_addr = saddr;

  modemMode = 1;

  bno = 0;

  printf("src_addr: 0x%lx\n", src_addr);

  if(fl_init() != 0) {

    printf("Flash init failed!\n");

    return(-1);

  }

#if 0

  printf("Unlocking flash...");

  for (i = 0, c = FLASH_BASE_ADDR; i < (FLASH_SIZE / FLASH_BLOCK_SIZE);

       i++, c += FLASH_BLOCK_SIZE)

    if (fl_unlock_one_block (c)) return 1;

  printf("done\n");

  printf("Erasing flash...");

  for (i = 0, c = FLASH_BASE_ADDR; i < (FLASH_SIZE / FLASH_BLOCK_SIZE);

       i++, c += FLASH_BLOCK_SIZE)

    if (fl_block_erase (c)) return 1;

    printf ("done\n");

#endif

  printf("Waiting...");

  mTimeoutCount = 0;

  mSetTimeout(TIMEOUT * TICKS_PER_SEC, mStartTimeout);

  while(1) {

    if(mTimeHandler && (get_timer(0) > mTimeValue)) {

      thand_f *x;

      x = mTimeHandler;

      mTimeHandler = (thand_f *)0;

      (*x)();

    }

    c = uart_testc();

    if(c != 0)

      break;

  }

  while(retval != 0) {

    retval = getBlock(c);

    if(retval != 0)

      c = uart_getc();

  }

  if(modemMode == 2) {

    c = uart_getc();

    retval = getBlock(c); /* last 'dummy' block for YModem */

    printf("... protocol: YModem, ");

  }

  else

    printf("... protocol: ZModem, ");

  return length;

}