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;
}