MHR-23: patches and fixes, also restricted disks to at least have 10G of

free space.

Signed-off-by: Amlal El Mahrouss <amlal.elmahrouss@icloud.com>

1 files changed, 9 insertions, 0 deletions

diff --git a/Private/NewBoot/BootKit/BootKit.hxx b/Private/NewBoot/BootKit/BootKit.hxx
index c8333b86..faba02fb 100644
--- a/Private/NewBoot/BootKit/BootKit.hxx
+++ b/Private/NewBoot/BootKit/BootKit.hxx
@@ -313,6 +313,15 @@ inline Boolean BDiskFormatFactory<BootDev>::Format(const char* partName,
 
     /// @note A catalog roughly equal to a sector.
 
+    constexpr auto cMinimumDiskSize = 10; // at minimum.
+
+    /// @note also look at EPM headers, for free part blocks.
+
+    if (GIB(fDiskDev.GetDiskSize()) < cMinimumDiskSize) {
+        EFI::ThrowError(L"Disk-Too-Tiny", L"Disk can't contain a New Filesystem partition.");
+        return false;
+    }
+
     partBlock->Version = kNewFSVersionInteger;
     partBlock->CatalogCount = blobCount;
     partBlock->Kind = kNewFSHardDrive;