assembly.s



  .text
  
SPMCR = 0x57   ;RWW区忙标志，读RWW区允许，允许写程序存储区

; void write_page (unsigned int adr, unsigned char function);
; bits 8:15 adr addresses the page...(must setup RAMPZ beforehand!!!)
_write_page::
    XCALL __WAIT_SPMEN__
    movw    r30, r16        ;move address to z pointer (R31 = ZH, R30 = ZL)
    STS     SPMCR, R18      ;argument 2 decides function
    SPM                     ;perform pagewrite
    RET

;-----------------------------------------

; void fill_temp_buffer (unsigned int data, unsigned int adr);
; bits 7:1 in adr addresses the word in the page... (2=first word, 4=second word etc..)
_fill_temp_buffer::
    XCALL __WAIT_SPMEN__
    movw    r30, r18        ;move adress to z pointer (R31=ZH R30=ZL)
    movw    r0, r16         ;move data to reg 0 and 1
    LDI     R19, 0x01
    STS     SPMCR, R19
    SPM                     ;Store program memory
    RET     
 
;----------------------------------------- 

;unsigned char read_flash(unsigned int add);
_read_flash::
    mov r31,r17
    mov r30,r16
    lpm r16,z
    clr r17
    ret 
  
;unsigned int read_program_memory (unsigned int adr ,unsigned char cmd);
_read_program_memory::
    movw    r30, r16        ;move adress to z pointer
    SBRC    R18, 0          ;read lockbits? (second argument = 0x09)
    STS     SPMCR, R18      ;if so, place second argument in SPMEN register
    LPM     r16, Z+
    LPM     r17, Z
    RET
 
;-----------------------------------------
       
_enableRWW::
 XCALL __WAIT_SPMEN__
    LDI R27,0x11
    STS SPMCR,R27
    SPM
    RET  
 
;-----------------------------------------          

__WAIT_SPMEN__:
    LDS     R27,SPMCR       ; load SPMCR to R27
    SBRC    R27,0           ; check SPMEN flag
    RJMP    __WAIT_SPMEN__  ; wait for SPMEN flag cleared       
    RET
 
;-----------------------------------------



assembly.h



void write_page (unsigned int adr, unsigned char function);
void fill_temp_buffer (unsigned int data,unsigned int adr);
unsigned int read_program_memory (unsigned int adr,unsigned char cmd);
//void write_lock_bits (unsigned char val);
void enableRWW(void);
unsigned char read_flash(unsigned int add);




main.c


/******************************************************************************
Atmega16 BootLoad程序 
日  期：2004年的最后一天

/*****************************************************************************/
#include <string.h>
#include <macros.h>
#include "assembly.h"//包含汇编代码头文件

//×××××××××××××××××××××××××××××××××××××××××××××××××××××××××
//请根据目标板选择芯片型号
//#define  ChipType     Atmega8      
#define  ChipType     Atmega16      
//#define  ChipType     Atmega32      
//#define  ChipType     Atmega64      
//#define  ChipType     Atmega128      
//#define  ChipType     Atmega8535      
//*********************************************************
//选择BOOT区大小
//#define  BootSize     'a'    //128  
//#define  BootSize     'b'    //256
//#define  BootSize       'c'    //512
//#define  BootSize     'd'    //1024
//#define  BootSize     'e'    //2048（不知道是否有2048字BOOT空间的芯片）
//选择BootLoad版本号
//#define  BootVer        'f'    //1.0版本
//#define  BootVer      'g'    //1.1版本
//#define  BootVer      'h'    //1.2版本
//#define  BootVer      'i'    //1.3版本
//#define  BootVer      'j'    //1.4版本
//#define  BootVer      'k'    //1.5版本
//#define  BootVer      'l'    //1.6版本
//#define  BootVer      'm'    //1.7版本
//#define  BootVer      'n'    //1.8版本
//#define  BootVer      'o'    //1.9版本
//#define  BootVer      'p'    //2.0版本(应该是最终版本了）
//**********************************************************

#define  Atmega16       0x31

//*****************************************************************************
//#define  InteClk                 //是否使用内部时钟
//#define  OscAddress     0x1fff   //时钟校准值存储地址
//#define  OscAddress     0x3fff   //时钟校准值存储地址
//*****************************************************************************
//8时钟下的波特率设置
//#define  BAU  103    //4800
//#define  BAU  51       //9600
//#define  BAU  34     //14400
//#define  BAU  25     //19200

//16M时钟下的波特率设置
#define  BAU  103   //9600

//使用的EEPROM
#define EEPROM_UPDATE_0X00_0X06 7
#define EEPROM_REPEAT_ENTER_BOOT_TIMES 0x07
#define EEPROM_DEBUG0 0x08
#define EEPROM_DEBUG1 0x09
#define ERASE_FLASH_PAGE_NUM 112//擦除页数


const unsigned char ucEepromWriteSuccess[EEPROM_UPDATE_0X00_0X06]={"success"};
const unsigned char ucEepromWriteUpdates[EEPROM_UPDATE_0X00_0X06]={"updates"};


//*****************************************************************************

#if (ChipType == Atmega16)
  #include  "iom16v.h"
  #define   PageByte     128   
  #define   AddressShift   7
#endif

void FlashLoad(void);
unsigned char GetPageNumber(void);
void ExecCode(void);
char GetOnePageData(void);
unsigned char WriteFlash(void);
char CheckFlash(void);
void SendChar(unsigned char c);
void delay(void);                 //1ms延时函数
//unsigned char RecChar(void);
static unsigned int GetCRC16Code(unsigned char* pCalcBuf, unsigned char ucSize);


//unsigned char PageBuffer[PageByte+2];
unsigned int  PageAddress=0;

#define BUFLEN 200
char uRxData;
char ucRecvBuf[BUFLEN];
unsigned int uiBufLen=0;
unsigned int uiStartBufDateLen=0;


//读EEPROM函数
static void EEPROM_Write(unsigned int Add,unsigned char Data)
{
 if(Add<512) //如果地址大于512，无效
 {
  while(EECR&BIT(EEWE)); //查询上一次EEPROM写操作是否完成
  EEAR=Add;    //写入地址
  EEDR=Data;    //写入数据
  EECR|=BIT(EEMWE);  //EEPROM写主机使能：准备
  EECR|=BIT(EEWE);  //EEPROM写使能：写入
 }
}

//读EEPROM函数
static unsigned char EEPROM_Read(unsigned int Add)
{
 unsigned char Temp=0;
 if(Add<512)  //如果地址大于512，读取0值
 {
  while(EECR&BIT(EEWE)); //查询上一次EEPROM写操作是否完成
  EEAR=Add;    //写入地址
  EECR|=BIT(EERE);  //EEPROM读使能：读出
  Temp=EEDR;    //读取数据
 } 
 return Temp;    //返回数据
}

#if 0
void PrintfStringToArm(char * str)
{
 int i=0;
 for(i=0;i<strlen(str);i++)
 {
  SendChar(*(str+i));
 }
}
#endif


/*****************************************************************************/
//Flash编程                            
/*****************************************************************************/
void FlashLoad(void)
{
 unsigned int i;

 unsigned char ucPageDateSuccessFlag=0;
 SendChar('!');
 while (1)
 {
  if(!GetPageNumber())//获取页地址
  {
   return;
  }
  else
  {
   if (PageAddress == 0xff80)//写结束
   {
    SendChar('$');
    CLI();
    
    for(i=0;i<EEPROM_UPDATE_0X00_0X06;i++)
    {
     EEPROM_Write(i,ucEepromWriteSuccess[i]);
    }
    SEI();

    return;
   }
   
   ucPageDateSuccessFlag=GetOnePageData();

   if (ucPageDateSuccessFlag==1)//获取一页正确的数据
   {
    if(WriteFlash())
    {
     if (CheckFlash())
      SendChar('!');
     else
      SendChar('@');
    }
    else
    {
     return;
    }
   }
   else if(ucPageDateSuccessFlag==2)//获取一页错误的数据
   {   
    SendChar('@');
   }
   else//超时
   {
    return;
   }
  }
 }

 //SendChar('%');
}

/*****************************************************************************/
unsigned char GetPageNumber(void)
{
 unsigned int i;
 unsigned char j; 
 unsigned int uiAddress;

 for(j=0;j<100;j++)
 {
  for(i=0; (i+4)<=uiBufLen; i++)
  {
   if(memcmp(ucRecvBuf+i,"##",2)==0)//"##"后紧跟页地址
   {
    memcpy(&uiAddress,ucRecvBuf+i+2,2);
    
    PageAddress=(uiAddress<<AddressShift);//验证了Z7~Z13(见中文版Datesheet P245) 

    uiStartBufDateLen=i+4;//页数据的偏移位
   
    return 1;   
   }
  }
  delay();
 }
 return 0;
}

static unsigned int GetCRC16Code(unsigned char* pCalcBuf, unsigned char ucSize)
{ 
 unsigned char i = 0;
 unsigned char j = 0;
 unsigned int usReturnValue = 0xFFFF;
 unsigned int usNew = 0xA001;
 
 for (i=0; i<ucSize; ++i)
 {
  usReturnValue ^= pCalcBuf[i];
  for(j=0; j<8; ++j)
  {
   if(usReturnValue & 0x0001)
   {
    usReturnValue >>= 1;
    usReturnValue ^= usNew;
   }
   else
   {
    usReturnValue >>= 1;
   }
  }
 }
 return usReturnValue;
}


/*****************************************************************************/
char GetOnePageData(void)
{
 unsigned char i;
 unsigned int LocalCheckSum = 0;
 unsigned int CheckSum = 0;

 for(i=0;i<200;i++)
 {
  if(uiBufLen>=(uiStartBufDateLen+PageByte+2))
  {
   //for(i=0;i<(PageByte+2);i++)
   //{
   // PageBuffer[i]=ucRecvBuf[i+uiStartBufDateLen];
   //}

   //LocalCheckSum=GetCRC16Code(PageBuffer, PageByte);
   LocalCheckSum=GetCRC16Code(&ucRecvBuf[uiStartBufDateLen-2], PageByte+2);//将页地址也一起校验，共校验PageByte+2个字节
   
   //memcpy(&CheckSum,&PageBuffer[PageByte],2);
   memcpy(&CheckSum,&ucRecvBuf[uiStartBufDateLen+PageByte],2);
   
   if (LocalCheckSum == CheckSum)
   {
    return 1;
   }
   else
   {
    CLI();
    
    memset(ucRecvBuf,0,BUFLEN);
    
    uiBufLen=0;

    SEI();

    return 2;
   }
  }
  delay();
 }

 return 0;
}

/*****************************************************************************/
unsigned char  WriteFlash(void)
{
 unsigned int i;
 unsigned int TempInt;
 
 for (i=0;i<PageByte;i+=2)
 {

  //TempInt=PageBuffer[i]+(PageBuffer[i+1]<<8);//写程序是按‘字‘写的，一页128字节，64个字
  TempInt=ucRecvBuf[uiStartBufDateLen+i]+(ucRecvBuf[uiStartBufDateLen+i+1]<<8);

  fill_temp_buffer(TempInt,i);    //call asm routine.见汇编代码，将一页的数据存储好
 }
 
 if(PageAddress==0x0000)
 {
  if(ucRecvBuf[uiStartBufDateLen]==0xFF)
  {
   //CLI();
    
   //EEPROM_Write(EEPROM_DEBUG1,0xD1);

   //SEI();

   return 0;
   
  }
  else//擦除数据，当为首页时，一致性擦除0-103页所有数据
  {
   for(;PageAddress<ERASE_FLASH_PAGE_NUM*128;PageAddress+=128)
   {    
    write_page(PageAddress,0x03);  //0x03,Bit1=1,页擦除，Bit0=1,存储程序存储器使能(SPMEN)
   }
   PageAddress=0x0000;
  }
 }
 else if(PageAddress==0x3700)//当为110页首地址时，一致性擦除110-111页的数据
 {
  for(;PageAddress<0x3800;PageAddress+=128)
  {   
   write_page(PageAddress,0x03);  //0x03,Bit1=1,页擦除，Bit0=1,存储程序存储器使能(SPMEN)
  }
  PageAddress=0x3700;
 }

 //write_page(PageAddress,0x03);       //0x03,Bit1=1,页擦除，Bit0=1,存储程序存储器使能(SPMEN)
 i=0;
 if((PageAddress < 0x3800)&&(PageAddress>=i))//从112页开始，为Bootloader程序，不允许写
 {
  write_page(PageAddress,0x05);       //0x05,Bit2=1,页写入，Bit0=1,存储程序存储器使能(SPMEN)
  
  enableRWW();

  return 1;
 }
 else
 {
  return 0;
 }
}

/*****************************************************************************/

char CheckFlash(void)
{
 unsigned char i;       
 unsigned int TempInt;
 unsigned int TempInt2;

 for (i=0;i<PageByte;i+=2)
 {
  TempInt = read_program_memory(PageAddress + i,0x00);
  //TempInt2 = PageBuffer[i] +(PageBuffer[i+1]<<8);
  TempInt2=ucRecvBuf[uiStartBufDateLen+i]+(ucRecvBuf[uiStartBufDateLen+i+1]<<8);
  if (TempInt != TempInt2)//烧写校验错误
  {
   CLI();
   
   memset(ucRecvBuf,0,BUFLEN);
   
   uiBufLen=0;
   
   SEI();
   
   return 0;
  }
 }
 CLI();
 
 memset(ucRecvBuf,0,BUFLEN);
 
 uiBufLen=0;
 
 SEI();
 
 return 1;
}

//串口接收完毕中断函数
//串口接收完毕中断触发声明
#pragma interrupt_handler  Usart0RecvdIsr:iv_USART_RX
void Usart0RecvdIsr(void)
{

  UCSRB&=~ BIT(RXCIE);  //接收完毕中断不使能
  
  uRxData=UDR;   //读取UDR的数据

  if(uiBufLen<BUFLEN)
  { 
   ucRecvBuf[uiBufLen]=uRxData;

   uiBufLen++;  
  }
  else
  {
   uiBufLen=0;
  }
  
  UCSRB|=BIT(RXCIE);  //接收完毕中断使能
}

 
/*****************************************************************************/
void SendChar(unsigned char c)
{ 
 while (!(UCSRA&(1<<UDRE)));
 UDR=c;
}


void delay(void)  //45-50ms左右
{
     unsigned int i,j;
     for(i=0;i<100;i++)
     for(j=0;j<1000;j++);
}


/*****************************************************************************/
void ExecCode(void)
{
 WDTCR=0x0F;//跳出前开看门狗2014-02-18添加

 GICR=0X01;           //中断向量表移位允许
 GICR=0X00;           //中断向量指向应用程序区
 
 asm("jmp 0x0000");             
}

void main()
{

 unsigned char i;
 unsigned char j;
 unsigned char ucEepromRepeatEnterBootTimes;
 unsigned char ucEepromRead[EEPROM_UPDATE_0X00_0X06];//={0};
 unsigned char bAppErrFlag=0;
 
 CLI();
 
 #if 0//2014-02-18屏蔽
 for(i=0; i<30;i++)//延时2s左右，等MCU电源稳定
 {
  delay();
 }
 #endif

 for(i=0;i<EEPROM_UPDATE_0X00_0X06;i++)
 {
  ucEepromRead[i]=EEPROM_Read(i);
 }
 

 if(memcmp(ucEepromRead,"updates",EEPROM_UPDATE_0X00_0X06)==0) 
 {
 
 AppErr: 
  
  TIMSK=0x00;//必须复位，否则从应用区跳转时会出问题
  WDTCR=0x00;
  
  GICR=(1<<IVCE);     //中断向量表移位允许
  GICR=(1<<IVSEL);     //中断向量指向Boot区

  
  UCSRC=(1<<URSEL)|(1<<UCSZ1)|(1<<UCSZ0); //8位数据＋1位停止
  //UCSRB = (1<<RXEN)|(1<<TXEN);   //允许串口发送和接收
  UCSRB=(1<<RXEN)|(1<<TXEN)|(1<<RXCIE);//允许收发,打开接收中断
  UBRR = BAU;
  
  SEI();

  for(i=0;i<100;i++)//大约4-5s钟
  {
   SendChar('[');    //通知PC机，BOOT下载准备就绪

   for(j=0;j<uiBufLen;j++)
   {
    if(ucRecvBuf[j]==']')
    {
      goto UpDateMcu;       
    }       
   }

   delay();     //延时等待PC机响应
   
   if(bAppErrFlag)//应用程序异常
   {
    i=0;
   }
  }

  
  //CLI();
  //EEPROM_Write(EEPROM_DEBUG0,0xD0);//入口超时
  //SEI();
  goto EXIT;
  
 }
 #if 1
 else
 {
  if(read_program_memory (0x0000,0x00)==0xFFFF)//判断应用程序首页是否异常
  {  
   #if 0
   for(i=0; i<EEPROM_UPDATE_0X00_0X06;i++)
   {
    EEPROM_Write(i,ucEepromWriteUpdates[i]);
   }
   #endif
   
   delay(); 

   if(read_program_memory (0x0000,0x00)==0xFFFF)//判断应用程序首页是否异常
   {
   
    bAppErrFlag = 1;
    
    goto AppErr;
   }
  }

  ucEepromRepeatEnterBootTimes=EEPROM_Read(EEPROM_REPEAT_ENTER_BOOT_TIMES);
  
  if(ucEepromRepeatEnterBootTimes<0xFD)
  {
   for(i=0; i<EEPROM_UPDATE_0X00_0X06;i++)
   {
    EEPROM_Write(i,ucEepromWriteUpdates[i]);
   }
  }
  else
  {  
   #if 1
   ucEepromRepeatEnterBootTimes--;
   
   EEPROM_Write(EEPROM_REPEAT_ENTER_BOOT_TIMES,ucEepromRepeatEnterBootTimes);
   #endif
  }
  
  goto EXIT;
 }

 #endif
 
UpDateMcu: 

 FlashLoad();
EXIT: 
 ExecCode();
}