FastbootCmds: add oem command to enable mainline optimization

In order to better support non-android operating systems, such
as postmarketOS (@postmarketOS, https://postmarketos.org/), add
a custom oem command {en,dis}able-mainline-optimization,
which can be used to toggle different behavior to improve UX
when using such systems.

Right now it does not do anything on its own except printing
whether mainline optimization is enabled or not when executing
`fastboot oem device-info`.

Change-Id: Icf4323e1402e97a9f81045a5f3225385876e0fce
Signed-off-by: Alexander Martinz <[email protected]>

Author: amartinz

QcomModulePkg/Include/Library/DeviceInfo.h

@@ -56,6 +56,7 @@ typedef struct device_info {
   CHAR8 user_public_key[MAX_USER_KEY_SIZE];
   UINT64 rollback_index[MAX_VB_PARTITIONS];
   CHAR8 Display_Cmdline[MAX_DISPLAY_CMDLINE_LEN];
+  BOOLEAN is_mainline_optimization_enabled;
 } DeviceInfo;
 
 struct verified_boot_verity_mode {
@@ -72,6 +73,7 @@ BOOLEAN IsUnlocked (VOID);
 BOOLEAN IsUnlockCritical (VOID);
 BOOLEAN IsEnforcing (VOID);
 BOOLEAN IsChargingScreenEnable (VOID);
+BOOLEAN IsMainlineOptimizationEnabled (VOID);
 BOOLEAN IsUserPublicKeySet (VOID);
 VOID
 GetBootloaderVersion (CHAR8 *BootloaderVersion, UINT32 Len);
@@ -80,6 +82,8 @@ GetRadioVersion (CHAR8 *RadioVersion, UINT32 Len);
 EFI_STATUS
 EnableChargingScreen (BOOLEAN IsEnabled);
 EFI_STATUS
+EnableMainlineOptimization (BOOLEAN IsEnabled);
+EFI_STATUS
 EnableEnforcingMode (BOOLEAN IsEnabled);
 EFI_STATUS
 SetDeviceUnlockValue (UINT32 Type, BOOLEAN State);

QcomModulePkg/Library/BootLib/DeviceInfo.c

@@ -57,6 +57,11 @@ BOOLEAN IsChargingScreenEnable (VOID)
   return DevInfo.is_charger_screen_enabled;
 }
 
+BOOLEAN IsMainlineOptimizationEnabled (VOID)
+{
+  return DevInfo.is_mainline_optimization_enabled;
+}
+
 BOOLEAN IsUserPublicKeySet (VOID)
 {
   CHAR8 *UserKeyBuffer = NULL;
@@ -108,6 +113,24 @@ EnableChargingScreen (BOOLEAN IsEnabled)
   return Status;
 }
 
+EFI_STATUS
+EnableMainlineOptimization (BOOLEAN IsEnabled)
+{
+  EFI_STATUS Status = EFI_SUCCESS;
+
+  if (IsMainlineOptimizationEnabled () != IsEnabled) {
+    DevInfo.is_mainline_optimization_enabled = IsEnabled;
+    Status = ReadWriteDeviceInfo (WRITE_CONFIG, &DevInfo, sizeof (DevInfo));
+    if (Status != EFI_SUCCESS) {
+      DEBUG ((EFI_D_ERROR, "Error %a mainline optimization: %r\n",
+              (IsEnabled ? "Enabling" : "Disabling"), Status));
+      return Status;
+    }
+  }
+
+  return Status;
+}
+
 EFI_STATUS
 StoreDisplayCmdLine (CONST CHAR8 *CmdLine, UINT32 CmdLineLen)
 {
@@ -309,6 +332,7 @@ EFI_STATUS DeviceInfoInit (VOID)
     }
     DevInfo.is_charger_screen_enabled = FALSE;
     DevInfo.verity_mode = TRUE;
+    DevInfo.is_mainline_optimization_enabled = FALSE;
     Status =
         ReadWriteDeviceInfo (WRITE_CONFIG, (VOID *)&DevInfo, sizeof (DevInfo));
     if (Status != EFI_SUCCESS) {

QcomModulePkg/Library/FastbootLib/FastbootCmds.c

@@ -3043,6 +3043,34 @@ CmdOemDisableChargerScreen (CONST CHAR8 *Arg, VOID *Data, UINT32 Size)
   }
 }
 
+STATIC VOID
+CmdOemEnableMainlineOptimization (CONST CHAR8 *Arg, VOID *Data, UINT32 Size)
+{
+  EFI_STATUS Status;
+  DEBUG ((EFI_D_INFO, "Enabling mainline optimization\n"));
+
+  Status = EnableMainlineOptimization (TRUE);
+  if (Status != EFI_SUCCESS) {
+    FastbootFail ("Failed to enable mainline optimization");
+  } else {
+    FastbootOkay ("");
+  }
+}
+
+STATIC VOID
+CmdOemDisableMainlineOptimization (CONST CHAR8 *Arg, VOID *Data, UINT32 Size)
+{
+  EFI_STATUS Status;
+  DEBUG ((EFI_D_INFO, "Disabling mainline optimization\n"));
+
+  Status = EnableMainlineOptimization (FALSE);
+  if (Status != EFI_SUCCESS) {
+    FastbootFail ("Failed to disable mainline optimization");
+  } else {
+    FastbootOkay ("");
+  }
+}
+
 STATIC VOID
 CmdOemOffModeCharger (CONST CHAR8 *Arg, VOID *Data, UINT32 Size)
 {
@@ -3288,6 +3316,10 @@ CmdOemDevinfo (CONST CHAR8 *arg, VOID *data, UINT32 sz)
                IsUserPublicKeySet () ? "true" : "false");
   FastbootInfo (DeviceInfo);
   WaitForTransferComplete ();
+  AsciiSPrint (DeviceInfo, sizeof (DeviceInfo), "Mainline optimization enabled: %a",
+               IsMainlineOptimizationEnabled () ? "true" : "false");
+  FastbootInfo (DeviceInfo);
+  WaitForTransferComplete ();
   FastbootOkay ("");
 }
 
@@ -3731,6 +3763,8 @@ FastbootCommandSetup (IN VOID *Base, IN UINT64 Size)
 #endif
       {"oem enable-charger-screen", CmdOemEnableChargerScreen},
       {"oem disable-charger-screen", CmdOemDisableChargerScreen},
+      {"oem enable-mainline-optimization", CmdOemEnableMainlineOptimization},
+      {"oem disable-mainline-optimization", CmdOemDisableMainlineOptimization},
       {"oem off-mode-charge", CmdOemOffModeCharger},
       {"oem select-display-panel", CmdOemSelectDisplayPanel},
       {"oem device-info", CmdOemDevinfo},