123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198 |
- #include "BootloaderHID.h"
- static bool RunBootloader = true;
- uint16_t MagicBootKey ATTR_NO_INIT;
- void Application_Jump_Check(void)
- {
-
- if ((MCUSR & (1 << WDRF)) && (MagicBootKey == MAGIC_BOOT_KEY))
- {
-
- MCUSR &= ~(1 << WDRF);
- wdt_disable();
-
- MagicBootKey = 0;
-
- ((void (*)(void))0x0000)();
- }
- }
- int main(void)
- {
-
- SetupHardware();
-
- GlobalInterruptEnable();
- while (RunBootloader)
- USB_USBTask();
-
- USB_Detach();
-
- MagicBootKey = MAGIC_BOOT_KEY;
-
- wdt_enable(WDTO_250MS);
- for (;;);
- }
- static void SetupHardware(void)
- {
-
- MCUSR &= ~(1 << WDRF);
- wdt_disable();
-
- clock_prescale_set(clock_div_1);
-
- MCUCR = (1 << IVCE);
- MCUCR = (1 << IVSEL);
-
- USB_Init();
- }
- void EVENT_USB_Device_ConfigurationChanged(void)
- {
-
- Endpoint_ConfigureEndpoint(HID_IN_EPADDR, EP_TYPE_INTERRUPT, HID_IN_EPSIZE, 1);
- }
- void EVENT_USB_Device_ControlRequest(void)
- {
-
- if ((USB_ControlRequest.bmRequestType & (CONTROL_REQTYPE_TYPE | CONTROL_REQTYPE_RECIPIENT)) !=
- (REQTYPE_CLASS | REQREC_INTERFACE))
- {
- return;
- }
-
- switch (USB_ControlRequest.bRequest)
- {
- case HID_REQ_SetReport:
- Endpoint_ClearSETUP();
-
- while (!(Endpoint_IsOUTReceived()));
-
- #if (FLASHEND > 0xFFFF)
- uint32_t PageAddress = ((uint32_t)Endpoint_Read_16_LE() << 8);
- #else
- uint16_t PageAddress = Endpoint_Read_16_LE();
- #endif
-
- #if (FLASHEND > 0xFFFF)
- if ((uint16_t)(PageAddress >> 8) == COMMAND_STARTAPPLICATION)
- #else
- if (PageAddress == COMMAND_STARTAPPLICATION)
- #endif
- {
- RunBootloader = false;
- }
- else if (PageAddress < BOOT_START_ADDR)
- {
-
- boot_page_erase(PageAddress);
- boot_spm_busy_wait();
-
- for (uint8_t PageWord = 0; PageWord < (SPM_PAGESIZE / 2); PageWord++)
- {
-
- if (!(Endpoint_BytesInEndpoint()))
- {
- Endpoint_ClearOUT();
- while (!(Endpoint_IsOUTReceived()));
- }
-
- boot_page_fill(PageAddress + ((uint16_t)PageWord << 1), Endpoint_Read_16_LE());
- }
-
- boot_page_write(PageAddress);
- boot_spm_busy_wait();
-
- boot_rww_enable();
- }
- Endpoint_ClearOUT();
- Endpoint_ClearStatusStage();
- break;
- }
- }
|