bootloader.c

Go to the documentation of this file.

/*----------------------------------------------------------------------------*/
/*----------------------------------------------------------------------------*/
//#############################################################################
//                                        IMPORTED DECLARATIONS
//#############################################################################

#include <Environment.h>
#include <bite.h>
#include <serial.h>
#include <hal_wait.h>
#include <set_hardware.h>
#include <cc2420.h>

#include <reprogrammation.h>

//#############################################################################
//                     CODE FUNCTIONS
//#############################################################################

//#############################################################################
//#############################################################################

void init ( void )
{       
        WDTCTL = WDTPW|WDTHOLD;   // Stop watchdog during initialization   
  
        setup_msp430();   // Setup CPU's clock
  setup_ports();    // Setup input and output port
  halSpiInit();     // Setup UART1 to SPI master
        
  halWait(TIME_CONSTANT); // Wait until SPI is ready
  
        clock_init();     // Setup timer B to generate time base for software clock
        cc2420_Init();    // Intialization of radio chip
  serial_init();    // UART0 setup to RS232 mode
  
        srand (time (NULL)); // initialise the random number generator
  
  halWait(TIME_CONSTANT); // Wait
        
        cc2420_change_channel(CHANNEL_EXPLOITATION); // switch to exploitation channel
        
        LED_TEST1_ON; // to say the device is on
        
        _EINT();
}

//#############################################################################
//#############################################################################

void main ( void )
{ 
  static char soft_buffer[SIZE_SOFT_BUFFER];   // buffer for write entire segment on flash memory
  static UINT8 prev_pckt_id,i,type,load_state=0,time_out=0,nb_try=0;
  UINT16 nb_paquet,start_segment,current_valid_indice=0,*ptr,packet_id=0;
  static JUST_CRC buf_ack;
  static DATA_PACKET buf_pack;
  static REQ_PACKET buf_req;
  
  init (); // init msp430, cc2420, timer B ...
  
  serial_puts("Bootloader RF\n");
    
  while(1){
    serial_puts("REQ VERS\n");  
        
    do{
      our_id = (UINT16) (rand() / (double)RAND_MAX * MAX_UINT16); // generate a dynamic identifier for negociation with server
    }while(our_id == BROADCAST_ID);   // verify choisen id is not broadcast adress
    
    wakeup_cc2420();
    if(check_online_code_version()){ // ask online last code version and compare with onchip code version
         
      load_state = 1; // update loading progress : we have writed info program memory with new software informations
                      // we can't return to old program now

      cc2420_change_channel(CHANNEL_REPROG);      
      
      MAC_send((char*)&buf_ack,TYPE_ACK,SEND_UNICAST); // send ACK
      
      nb_paquet = (((*ptr_info_prog).end_addr - (*ptr_info_prog).start_addr ) / SIZE_DATA_BUFFER)+1; // compute number of packet data that will be send
      start_segment = ((FLASH_BASE - (*ptr_info_prog).start_addr) / SEGMENT_SIZE) + 1;  // compute start point where new soft will be writed
      
      // initialize packet current and previous identifers
      prev_pckt_id = 0;time_out=0;packet_id=0;current_valid_indice=0;
      
      while(packet_id < nb_paquet && time_out < ERROR_THRESHOLD){ // recept all data packet 
        
        if(MAC_recept((char*)&buf_pack, &type , WAIT_ANSWER_CLIENT,DONT_MEMORISE_ADRESS) > 0 && type == TYPE_DATA && buf_pack.id == prev_pckt_id + 1){
            
            packet_id = buf_pack.id;i=0;
          
            //**************write data on soft_buffer***************************
            while ( current_valid_indice < SIZE_SOFT_BUFFER && i < SIZE_DATA_BUFFER ){
              soft_buffer[current_valid_indice] = buf_pack.data[i];
              current_valid_indice++;
              i++;            
            }
          
            if ( current_valid_indice == SIZE_SOFT_BUFFER && i == SIZE_DATA_BUFFER){  // data buffer match exactly on soft buffer
              erase_flash_segment(start_segment);
              flash_write(soft_buffer,SIZE_SOFT_BUFFER,start_segment,0);   // then empty soft buffer copying it on flash memory
              current_valid_indice=0;
              start_segment--;
            
            } else if ( i != SIZE_DATA_BUFFER ){ // soft buffer is full
                erase_flash_segment(start_segment);
                flash_write(soft_buffer,SIZE_SOFT_BUFFER,start_segment,0);   // then empty soft buffer copying it on flash memory
                current_valid_indice=0;
                start_segment--;
            
                // finish to copy packet data on soft buffer
                while ( i < SIZE_DATA_BUFFER ){
                  soft_buffer[current_valid_indice] = buf_pack.data[i];
                  current_valid_indice++;
                  i++;
                }
            }// else : packet data buffer is empty althougth soft buffer not full
          
            //*******************************************************************
          
            if(packet_id == nb_paquet){ // if it's the last packet : flush software buffer
          
              erase_flash_segment(start_segment);
              flash_write(soft_buffer,current_valid_indice,start_segment,0);   // then empty soft buffer copying it on flash memory
              
              load_state=2; // update loading progress : we have completed new software loading
                            // we can boot on it now
            }
            //*******************************************************************
          
            prev_pckt_id++;
          
            }else{  // packet just received is not next packet expected : request another packet from server
              time_out++;                                             // to count packet transmit error
              buf_req.id = prev_pckt_id + 1;                          // prepare request packet
              halWait(MAX_HAL_WAIT/5);                                // wait server is listenning RF channel
              MAC_send((char*)&buf_req,TYPE_REQ_PCKT,SEND_UNICAST);   // send request to server
            }
      } 
  }
  // switch to exploitation channel
  cc2420_change_channel(CHANNEL_EXPLOITATION);
  cc2420_powerdown(); // put radio chip in low power mode

  // boot on new program or bootloader if there is no program on memory
  if( load_state == 0 && nb_try < NB_REPROG_TRY){ // if no program was loaded : reboot on bootloader one time again

    nb_try++; // count how much time we have tried to load new software
    
    // future upgrade : can we put MSP in LPM here ?
    
    for (i=0; i <= 40 ; i++) // wait next updating session
      halWait(MAX_HAL_WAIT);
    
  }else if(load_state == 1 ){ // a program has begun to be loaded but fail before ending
                              // we haven't any information on old software then we have
                              // to try reload it until we sucess
    
    erase_info_memory(1); // erase information of canceled downloaded program
    
    // future upgrade : can we put MSP in LPM here ?
    
    for (i=0; i <= 40 ; i++) // wait next updating session
      halWait(MAX_HAL_WAIT);
    
  }else{  // a new software have been downloaded then we have to update interrupt
          // vector to reboot on new program or old one if no program has been download
        
    _DINT();                      // disable all maskable interrupt
    erase_interrupt_vector();     // erase interrupt vector
    ptr = (UINT16*)ptr_info_prog; // prepare pointer to copy info memory on IT vector
    for (i=0; i <= 15 ; i++){     // write back interrupts vectors

      if (*ptr != 0xFFFF)     // check information memory to find interrupts adress
        write_interrupt_vector(*ptr , i*2u - 2u);  // write adress
   
      ptr++;  // move pointer on next field of information memory
    }
    WDTCTL = WDTHOLD; // to cause device reboot
  }
  
  } 
}

---