[bootloader] Compatibility with Omega 2.0

bootloader/boot.cpp

@@ -0,0 +1,82 @@
+#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);
+  }
+}
+
+}