cflashmem.cpp

使用USB接口实现AVR单片机ISP升级.

/*
  cflashmem.cpp - part of flashtool for AVRUSBBoot, an USB bootloader for Atmel AVR controllers

  Thomas Fischl <[e-mail]>

  Creation Date..: 2006-03-18
  Last change....: 2006-06-25
*/
#include "cflashmem.h"
extern void UpdataMessageStr(char* mystring);
extern void UpdataMessageVal(unsigned int i);
CFlashmem::CFlashmem(unsigned int pagesize) {
  m_nPagesize = pagesize;
  m_nPagecount = 0;
  m_pFirstpage = NULL;
  m_pLastpage = NULL;
}

CFlashmem::~CFlashmem() {
  
}

CPage * CFlashmem::getFirstpage() {
  return m_pFirstpage;
}

CPage* CFlashmem::getPageToAddress(unsigned int nAddress) {
  unsigned int nBaseaddress = nAddress - (nAddress % m_nPagesize);
  
  if (m_pFirstpage == NULL) return NULL;

  if (m_pLastpage->getPageaddress() == nBaseaddress) return m_pLastpage;

  CPage* pPage = m_pFirstpage;
  while (pPage != NULL) {
    if (pPage->getPageaddress() == nBaseaddress) return pPage;
    pPage = pPage->getNext();
  }

  return NULL;
}

void CFlashmem::insertData(unsigned int nAddress, unsigned char bData) {
  
  CPage* pPage = getPageToAddress(nAddress);
  if (pPage == NULL) {

    pPage = new CPage(nAddress, m_nPagesize);
    m_nPagecount++;

    if (m_pLastpage != NULL) {
      m_pLastpage->setNext(pPage);
      pPage->setPrev(m_pLastpage);
    } else {
      m_pFirstpage = pPage;
    }
    m_pLastpage = pPage;

  }
  pPage->insert(nAddress, bData);
}

void CFlashmem::display() {
  CPage* pPage = m_pFirstpage;
  while (pPage != NULL) {
    //printf("Page Adresse: %d\n", pPage->getPageaddress());
	UpdataMessageStr("Page Adresse:");
    UpdataMessageVal(pPage->getPageaddress());
	UpdataMessageStr("\r\n");
    pPage->display();
    pPage = pPage->getNext();
  }
}


int sscanhex( unsigned char *str, unsigned int *hexout, int n )
{
  unsigned int hex = 0, x = 0;
  for(; n; n--){
    x = *str;
    if( x >= 'a' )
      x += 10 - 'a';
    else if( x >= 'A' )
      x += 10 - 'A';
    else
      x -= '0';
    if( x >= 16 )
      break;
    hex = hex * 16 + x;
    str++;
  }
  *hexout = hex;
  return n;					// 0 if all digits read
}


int readhex( FILE *fp, unsigned int *addr, unsigned char *data){
  /* Return value: 1..255	number of bytes
			0	end or segment record
		       -1	file end
		       -2	error or no HEX-File */
  char hexline[524];				// intel hex: max 255 byte
  unsigned char * hp = (unsigned char *) hexline;
  unsigned int byte;
  int i;
  unsigned int num;

  if( fgets( hexline, 524, fp ) == NULL )
    return -1;					// end of file
  if( *hp++ != ':' )
    return -2;                                  // no hex record
  if( sscanhex( hp, &num, 2 ))
    return -2;					// no hex number
  hp += 2;
  if( sscanhex( hp, addr, 4 ))
    return -2;
  hp += 4;
  if( sscanhex( hp, &byte, 2 ))
    return -2;
  if( byte != 0 )				// end or segment record
    return 0;
  for( i = num; i--; ){
    hp += 2;
    if( sscanhex( hp, &byte, 2 ))
      return -2;
    *data++ = byte;
  }
  return num;
}


void CFlashmem::readFromIHEX(char* filename) {
  assert(filename);
  
  FILE* fp;
  if ((fp = fopen(filename, "rb")) == NULL) {
    //printf("File %s open failed!\n", filename);
	  UpdataMessageStr("File ");
	  UpdataMessageStr(filename);
	  UpdataMessageStr(" open failed!");
	  UpdataMessageStr("\r\n");
    exit(1);
  };

  int i;
  unsigned int addr;
  unsigned char data[255];

  while( (i = readhex( fp, &addr, data )) >= 0 ){
    if ( i ) {
      for (int j = 0; j < i; j++) {
	insertData(addr + j, data[j]);
      }
    }
  }

  fclose(fp);
}

void *NoFailMalloc (size_t size)
{
	void *result;
	
	if ((result = malloc (size)) == NULL)
	{
		fprintf (stderr,"Can't allocate memory.\n");
		exit(1);
	}
	return (result);
}

void CFlashmem::readFromBIN(char* filename) {
	#define MEMORY_SIZE 1024*1024
	assert(filename);
	
	FILE* fp;
	if ((fp = fopen(filename, "rb")) == NULL) {
		//printf("File %s open failed!\n", filename);
		UpdataMessageStr("File ");
		UpdataMessageStr(filename);
		UpdataMessageStr(" open failed!");
		UpdataMessageStr("\r\n");
		exit(1);
	};
	
	unsigned char *Memory_Block;
	unsigned long int TotalBytes,i;

	fseek(fp,0,SEEK_END);
	TotalBytes=ftell(fp);
	if(TotalBytes>MEMORY_SIZE)
	{
		UpdataMessageStr("Bin file too large");
		UpdataMessageStr("\r\n");
		fclose(fp);
		exit(1);
	}
	fseek(fp,0,SEEK_SET);

	Memory_Block = (unsigned char *) NoFailMalloc(MEMORY_SIZE);
	memset (Memory_Block,0x00,MEMORY_SIZE);

	if(fread(Memory_Block,1,TotalBytes, fp)!=TotalBytes) /*判断是否读了TotalBytes个数*/		 
	{ 
		if(!feof(fp)) 
		{
			UpdataMessageStr("Read bin file error");
			UpdataMessageStr("\r\n");
            fclose(fp);             /*关闭文件*/
			free (Memory_Block);
			exit(1);
		}
	}
	for(i=0;i<TotalBytes;i++)
		insertData(i, Memory_Block[i]);
	free (Memory_Block);
	fclose(fp);
}