123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263 |
- #define INCLUDE_FROM_BOOTLOADER_MASSSTORAGE_C
- #include "BootloaderMassStorage.h"
- USB_ClassInfo_MS_Device_t Disk_MS_Interface =
- {
- .Config =
- {
- .InterfaceNumber = INTERFACE_ID_MassStorage,
- .DataINEndpoint =
- {
- .Address = MASS_STORAGE_IN_EPADDR,
- .Size = MASS_STORAGE_IO_EPSIZE,
- .Banks = 1,
- },
- .DataOUTEndpoint =
- {
- .Address = MASS_STORAGE_OUT_EPADDR,
- .Size = MASS_STORAGE_IO_EPSIZE,
- .Banks = 1,
- },
- .TotalLUNs = 1,
- },
- };
- bool RunBootloader = true;
- uint16_t MagicBootKey ATTR_NO_INIT;
- static uint8_t TicksSinceLastCommand = 0;
- void Application_Jump_Check(void)
- {
- bool JumpToApplication = false;
- #if (BOARD == BOARD_LEONARDO)
-
- PORTC |= (1 << 7);
- Delay_MS(10);
-
- JumpToApplication = ((PINC & (1 << 7)) != 0);
-
- PORTC &= ~(1 << 7);
- #elif ((BOARD == BOARD_XPLAIN) || (BOARD == BOARD_XPLAIN_REV1))
-
- JTAG_DISABLE();
-
- PORTF |= (1 << 4);
- Delay_MS(10);
-
- JumpToApplication = ((PINF & (1 << 4)) != 0);
-
- JTAG_ENABLE();
- #else
-
- if (boot_lock_fuse_bits_get(GET_HIGH_FUSE_BITS) & FUSE_BOOTRST)
- {
-
- if (!(MCUSR & (1 << EXTRF)) || (MagicBootKey == MAGIC_BOOT_KEY))
- JumpToApplication = true;
-
- MCUSR &= ~(1 << EXTRF);
- }
- else
- {
-
- if ((MCUSR & (1 << WDRF)) && (MagicBootKey == MAGIC_BOOT_KEY))
- JumpToApplication = true;
-
- MCUSR &= ~(1 << WDRF);
- }
- #endif
-
- bool ApplicationValid = (pgm_read_word_near(0) != 0xFFFF);
-
- if (JumpToApplication && ApplicationValid)
- {
-
- MCUSR &= ~(1 << WDRF);
- wdt_disable();
-
- MagicBootKey = 0;
-
- ((void (*)(void))0x0000)();
- }
- }
- int main(void)
- {
- SetupHardware();
- LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
- GlobalInterruptEnable();
- while (RunBootloader || TicksSinceLastCommand++ < 0xFF)
- {
- MS_Device_USBTask(&Disk_MS_Interface);
- 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);
-
- LEDs_Init();
- USB_Init();
-
- TIMSK1 = (1 << TOIE1);
- TCCR1B = ((1 << CS11) | (1 << CS10));
- }
- ISR(TIMER1_OVF_vect, ISR_BLOCK)
- {
- LEDs_ToggleLEDs(LEDS_LED1 | LEDS_LED2);
- }
- void EVENT_USB_Device_Connect(void)
- {
-
- LEDs_SetAllLEDs(LEDMASK_USB_ENUMERATING);
- }
- void EVENT_USB_Device_Disconnect(void)
- {
-
- LEDs_SetAllLEDs(LEDMASK_USB_NOTREADY);
- }
- void EVENT_USB_Device_ConfigurationChanged(void)
- {
- bool ConfigSuccess = true;
-
- ConfigSuccess &= MS_Device_ConfigureEndpoints(&Disk_MS_Interface);
-
- LEDs_SetAllLEDs(ConfigSuccess ? LEDMASK_USB_READY : LEDMASK_USB_ERROR);
- }
- void EVENT_USB_Device_ControlRequest(void)
- {
- MS_Device_ProcessControlRequest(&Disk_MS_Interface);
- }
- bool CALLBACK_MS_Device_SCSICommandReceived(USB_ClassInfo_MS_Device_t* const MSInterfaceInfo)
- {
- bool CommandSuccess;
- LEDs_SetAllLEDs(LEDMASK_USB_BUSY);
- CommandSuccess = SCSI_DecodeSCSICommand(MSInterfaceInfo);
- LEDs_SetAllLEDs(LEDMASK_USB_READY);
-
- TicksSinceLastCommand = 0;
- return CommandSuccess;
- }
|