mirror of
https://github.com/UpsilonNumworks/Upsilon.git
synced 2026-01-19 00:37:25 +01:00
83 lines
2.1 KiB
C++
83 lines
2.1 KiB
C++
#include <bootloader/boot.h>
|
|
#include <bootloader/slot.h>
|
|
#include <ion.h>
|
|
#include <ion/src/device/shared/drivers/reset.h>
|
|
#include <bootloader/interface.h>
|
|
|
|
#include <assert.h>
|
|
|
|
namespace Bootloader {
|
|
|
|
BootMode Boot::mode() {
|
|
// We use the exam mode driver as storage for the boot mode
|
|
uint8_t mode = Ion::ExamMode::FetchExamMode();
|
|
|
|
if (mode > 3)
|
|
return Unknown;
|
|
|
|
return (BootMode) mode;
|
|
}
|
|
|
|
void Boot::setMode(BootMode mode) {
|
|
BootMode currentMode = Boot::mode();
|
|
if (currentMode == mode)
|
|
return;
|
|
|
|
assert(mode != BootMode::Unknown);
|
|
int8_t deltaMode = (int8_t)mode - (int8_t)currentMode;
|
|
deltaMode = deltaMode < 0 ? deltaMode + 4 : deltaMode;
|
|
assert(deltaMode > 0);
|
|
Ion::ExamMode::IncrementExamMode(deltaMode);
|
|
}
|
|
|
|
__attribute__((noreturn)) void Boot::boot() {
|
|
assert(mode() != BootMode::Unknown);
|
|
|
|
if (!Slot::A().kernelHeader()->isValid() && !Slot::B().kernelHeader()->isValid()) {
|
|
// Bootloader if both invalid
|
|
bootloader();
|
|
} else if (!Slot::A().kernelHeader()->isValid()) {
|
|
// If slot A is invalid and B valid, boot B
|
|
setMode(BootMode::SlotB);
|
|
Slot::B().boot();
|
|
} else if (!Slot::B().kernelHeader()->isValid()) {
|
|
// If slot B is invalid and A valid, boot A
|
|
setMode(BootMode::SlotA);
|
|
Slot::A().boot();
|
|
} else {
|
|
// Both valid, boot the selected one
|
|
if (mode() == BootMode::SlotA) {
|
|
Slot::A().boot();
|
|
} else if (mode() == BootMode::SlotB) {
|
|
Slot::B().boot();
|
|
}
|
|
}
|
|
|
|
// Achivement unlocked: How Did We Get Here?
|
|
bootloader();
|
|
}
|
|
|
|
__attribute__ ((noreturn)) void Boot::bootloader() {
|
|
for(;;) {
|
|
// Draw the interfaces and infos
|
|
Bootloader::Interface::draw();
|
|
|
|
// Enable USB
|
|
Ion::USB::enable();
|
|
|
|
// Wait for the device to be enumerated
|
|
do {
|
|
// If we pressed back while waiting, reset.
|
|
uint64_t scan = Ion::Keyboard::scan();
|
|
if (scan == Ion::Keyboard::State(Ion::Keyboard::Key::Back)) {
|
|
Ion::Device::Reset::core();
|
|
}
|
|
} while (!Ion::USB::isEnumerated());
|
|
|
|
// Launch the DFU stack, allowing to press Back to quit and reset
|
|
Ion::USB::DFU(true);
|
|
}
|
|
}
|
|
|
|
}
|