Add Micro-Pro DFU Bootloader from Stanley Lio

[pub/lufa.git] / Bootloaders / DFU / BootloaderDFU.c

diff --git a/Bootloaders/DFU/BootloaderDFU.c b/Bootloaders/DFU/BootloaderDFU.c
index 0ac9af9..1f86aee 100644 (file)
--- a/Bootloaders/DFU/BootloaderDFU.c
+++ b/Bootloaders/DFU/BootloaderDFU.c
@@ -142,6 +142,9 @@ void Application_Jump_Check(void)
        }
 }
 
+
+static volatile bool stayinbootloader;
+
 /** Main program entry point. This routine configures the hardware required by the bootloader, then continuously
  *  runs the bootloader processing routine until instructed to soft-exit, or hard-reset via the watchdog to start
  *  the loaded application code.
@@ -152,14 +155,42 @@ int main(void)
        SetupHardware();
 
        /* Turn on first LED on the board to indicate that the bootloader has started */
-       LEDs_SetAllLEDs(LEDS_LED1);
+       //LEDs_SetAllLEDs(LEDS_LED1);
 
        /* Enable global interrupts so that the USB stack can function */
        GlobalInterruptEnable();
 
        /* Run the USB management task while the bootloader is supposed to be running */
+       /*if bit_is_clear(PINB,PB5)     // PB5 is Digital 9 on Arduino Pro Micro
+       {
+               loop_until_bit_is_set(PINB,PB5);
+
+               while ((RunBootloader || WaitForExit) && bit_is_set(PINB,PB5))
+                       USB_USBTask();
+
+               loop_until_bit_is_clear(PINB,PB5);
+       }*/
+
+       stayinbootloader = false;
+
+       uint16_t i = 0;
        while (RunBootloader || WaitForExit)
-         USB_USBTask();
+       {
+               USB_USBTask();
+
+               if (!stayinbootloader)
+               {
+                       _delay_ms(1);
+                       if (i++ > 5000)
+                       {
+                               break;
+                       }
+               }
+               else
+               {
+                       i = 0;
+               }
+       }
 
        /* Reset configured hardware back to their original states for the user application */
        ResetHardware();
@@ -184,7 +215,11 @@ static void SetupHardware(void)
 
        /* Initialize the USB and other board hardware drivers */
        USB_Init();
-       LEDs_Init();
+       //LEDs_Init();
+       DDRB = 1;
+       PORTB = _BV(PB5);
+       DDRD = 0b00100000;
+       PORTD = 0;
 
        /* Bootloader active LED toggle timer initialization */
        TIMSK1 = (1 << TOIE1);
@@ -196,7 +231,11 @@ static void ResetHardware(void)
 {
        /* Shut down the USB and other board hardware drivers */
        USB_Disable();
-       LEDs_Disable();
+       //LEDs_Disable();
+       DDRB = 0;
+       PORTB = 0;
+       DDRD = 0;
+       PORTD = 0;
        
        /* Disable Bootloader active LED toggle timer */
        TIMSK1 = 0;
@@ -210,7 +249,10 @@ static void ResetHardware(void)
 /** ISR to periodically toggle the LEDs on the board to indicate that the bootloader is active. */
 ISR(TIMER1_OVF_vect, ISR_BLOCK)
 {
-       LEDs_ToggleLEDs(LEDS_LED1 | LEDS_LED2);
+       //LEDs_ToggleLEDs(LEDS_LED1 | LEDS_LED2);
+       PORTB &= ~_BV(PB0);
+       _delay_ms(5);
+       PORTB |= _BV(PB0);
 }
 
 /** Event handler for the USB_ControlRequest event. This is used to catch and process control requests sent to
@@ -226,8 +268,13 @@ void EVENT_USB_Device_ControlRequest(void)
                return;
        }
 
+stayinbootloader = true;
+
        /* Activity - toggle indicator LEDs */
-       LEDs_ToggleLEDs(LEDS_LED1 | LEDS_LED2);
+       //LEDs_ToggleLEDs(LEDS_LED1 | LEDS_LED2);
+       PORTB &= ~_BV(PB0);
+       _delay_ms(5);
+       PORTB |= _BV(PB0);
 
        /* Get the size of the command and data from the wLength value */
        SentCommand.DataSize = USB_ControlRequest.wLength;