FastbootMenu: add ability to set active slot

Requires developer mode to be enabled.

Change-Id: Iac972efac622bc4ed94645161ae2120ac5003533
Co-developed-by: Alexander Martinz <[email protected]>
Signed-off-by: Alexander Martinz <[email protected]>

Author: kcxt

QcomModulePkg/Include/Library/DrawUI.h

@@ -115,6 +115,8 @@ typedef enum {
   FFBM,
   QMMI,
   NOACTION,
+  SET_ACTIVE_SLOT_A,
+  SET_ACTIVE_SLOT_B,
   OPTION_ACTION_MAX,
 } OPTION_ITEM_ACTION;
 

QcomModulePkg/Library/BootLib/FastbootMenu.c

@@ -42,6 +42,8 @@
 
 STATIC OPTION_MENU_INFO gMenuInfo;
 
+#define OPTIONS_COUNT_DEVELOPER_MODE 2
+
 STATIC MENU_MSG_INFO mFastbootOptionTitle[] = {
     {{"START"},
      BIG_FACTOR,
@@ -87,6 +89,21 @@ STATIC MENU_MSG_INFO mFastbootOptionTitle[] = {
      0,
      QMMI},
 #endif
+    /* Developer mode options */
+    {{"Activate slot _a"},
+     BIG_FACTOR,
+     BGR_ORANGE,
+     BGR_BLACK,
+     OPTION_ITEM,
+     0,
+     SET_ACTIVE_SLOT_A},
+    {{"Activate slot _b"},
+     BIG_FACTOR,
+     BGR_ORANGE,
+     BGR_BLACK,
+     OPTION_ITEM,
+     0,
+     SET_ACTIVE_SLOT_B},
 };
 
 #define FASTBOOT_MSG_INDEX_HEADER                  0
@@ -283,6 +300,7 @@ FastbootMenuShowScreen (OPTION_MENU_INFO *OptionMenuInfo)
   EFI_STATUS Status = EFI_SUCCESS;
   UINT32 Location = 0;
   UINT32 OptionItem = 0;
+  UINT32 OptionItemCount = 0;
   UINT32 Height = 0;
   UINT32 i = 0;
   UINT32 j = 0;
@@ -417,7 +435,11 @@ FastbootMenuShowScreen (OPTION_MENU_INFO *OptionMenuInfo)
   }
 
   OptionMenuInfo->Info.MenuType = DISPLAY_MENU_FASTBOOT;
-  OptionMenuInfo->Info.OptionNum = ARRAY_SIZE (mFastbootOptionTitle);
+
+  OptionItemCount = ARRAY_SIZE (mFastbootOptionTitle);
+  if (!IsDeveloperModeEnabled ())
+    OptionItemCount -= OPTIONS_COUNT_DEVELOPER_MODE;
+  OptionMenuInfo->Info.OptionNum = OptionItemCount;
 
   return Status;
 }

QcomModulePkg/Library/BootLib/MenuKeysDetection.c

@@ -105,6 +105,8 @@ UpdateDeviceStatus (OPTION_MENU_INFO *MsgInfo, INTN Reason)
   EFI_STATUS Status = EFI_SUCCESS;
   EFI_GUID Ptype = gEfiMiscPartitionGuid;
   MemCardType CardType = UNKNOWN;
+  CHAR16 SuffixUnicode[MAX_SLOT_SUFFIX_SZ] = u"";
+  Slot BootSlot;
 
   /* Clear the screen */
   gST->ConOut->ClearScreen (gST->ConOut);
@@ -158,6 +160,26 @@ UpdateDeviceStatus (OPTION_MENU_INFO *MsgInfo, INTN Reason)
     }
     RebootDevice (NORMAL_MODE);
     break;
+  case SET_ACTIVE_SLOT_A:
+    AsciiStrToUnicodeStr ("_a", SuffixUnicode);
+    StrnCpyS (BootSlot.Suffix, ARRAY_SIZE (BootSlot.Suffix), SuffixUnicode,
+            StrLen (SuffixUnicode));
+    Status = SetActiveSlot (&BootSlot, TRUE);
+    if (Status != EFI_SUCCESS) {
+      DEBUG ((EFI_D_ERROR, "Failed to set the active slot to _a\n"));
+    }
+    RebootDevice (FASTBOOT_MODE);
+    break;
+  case SET_ACTIVE_SLOT_B:
+    AsciiStrToUnicodeStr ("_b", SuffixUnicode);
+    StrnCpyS (BootSlot.Suffix, ARRAY_SIZE (BootSlot.Suffix), SuffixUnicode,
+            StrLen (SuffixUnicode));
+    Status = SetActiveSlot (&BootSlot, TRUE);
+    if (Status != EFI_SUCCESS) {
+      DEBUG ((EFI_D_ERROR, "Failed to set the active slot to _b\n"));
+    }
+    RebootDevice (FASTBOOT_MODE);
+    break;
   }
 }