x86/e820: temporarily enable KHO scratch for memory below 1M

KHO kernels are special and use only scratch memory for memblock
allocations, but memory below 1M is ignored by kernel after early boot and
cannot be naturally marked as scratch.

To allow allocation of the real-mode trampoline and a few (if any) other
very early allocations from below 1M forcibly mark the memory below 1M as
scratch.

After real mode trampoline is allocated, clear that scratch marking.

Link: https://lkml.kernel.org/r/[e-mail]
Signed-off-by: Alexander Graf <[e-mail]>
Co-developed-by: Mike Rapoport (Microsoft) <[e-mail]>
Signed-off-by: Mike Rapoport (Microsoft) <[e-mail]>
Co-developed-by: Changyuan Lyu <[e-mail]>
Signed-off-by: Changyuan Lyu <[e-mail]>
Acked-by: Dave Hansen <[e-mail]>
Cc: Andy Lutomirski <[e-mail]>
Cc: Anthony Yznaga <[e-mail]>
Cc: Arnd Bergmann <[e-mail]>
Cc: Ashish Kalra <[e-mail]>
Cc: Ben Herrenschmidt <[e-mail]>
Cc: Borislav Betkov <[e-mail]>
Cc: Catalin Marinas <[e-mail]>
Cc: David Woodhouse <[e-mail]>
Cc: Eric Biederman <[e-mail]>
Cc: "H. Peter Anvin" <[e-mail]>
Cc: Ingo Molnar <[e-mail]>
Cc: James Gowans <[e-mail]>
Cc: Jason Gunthorpe <[e-mail]>
Cc: Jonathan Corbet <[e-mail]>
Cc: Krzysztof Kozlowski <[e-mail]>
Cc: Marc Rutland <[e-mail]>
Cc: Paolo Bonzini <[e-mail]>
Cc: Pasha Tatashin <[e-mail]>
Cc: Peter Zijlstra <[e-mail]>
Cc: Pratyush Yadav <[e-mail]>
Cc: Rob Herring <[e-mail]>
Cc: Saravana Kannan <[e-mail]>
Cc: Stanislav Kinsburskii <[e-mail]>
Cc: Steven Rostedt <[e-mail]>
Cc: Thomas Gleinxer <[e-mail]>
Cc: Thomas Lendacky <[e-mail]>
Cc: Will Deacon <[e-mail]>
Signed-off-by: Andrew Morton <[e-mail]>

arch/x86/kernel/e820.c

···
 	}
 
 	/*
+	 * At this point memblock is only allowed to allocate from memory
+	 * below 1M (aka ISA_END_ADDRESS) up until direct map is completely set
+	 * up in init_mem_mapping().
+	 *
+	 * KHO kernels are special and use only scratch memory for memblock
+	 * allocations, but memory below 1M is ignored by kernel after early
+	 * boot and cannot be naturally marked as scratch.
+	 *
+	 * To allow allocation of the real-mode trampoline and a few (if any)
+	 * other very early allocations from below 1M forcibly mark the memory
+	 * below 1M as scratch.
+	 *
+	 * After real mode trampoline is allocated, we clear that scratch
+	 * marking.
+	 */
+	memblock_mark_kho_scratch(0, SZ_1M);
+
+	/*
 	 * 32-bit systems are limited to 4BG of memory even with HIGHMEM and
 	 * to even less without it.
 	 * Discard memory after max_pfn - the actual limit detected at runtime.

arch/x86/realmode/init.c

···
 	 * setup_arch().
 	 */
 	memblock_reserve(0, SZ_1M);
+
+	memblock_clear_kho_scratch(0, SZ_1M);
 }
 
 static void __init sme_sev_setup_real_mode(struct trampoline_header *th)