cpu/x86/mtrr: allow temporary MTRR range during coreboot

Certain platforms have a poorly performing SPI prefetcher so even if
accessing MMIO BIOS once the fetch time can be impacted. Payload
loading is one example where it can be impacted. Therefore, add the
ability for a platform to reconfigure the currently running CPU's
variable MTRR settings for the duration of coreboot's execution.

The function mtrr_use_temp_range() is added which uses the previous
MTRR solution as a basis along with a new range and type to use.
A new solution is calculated with the updated settings and the
original solution is put back prior to exiting coreboot into the OS
or payload.

Using this patch on apollolake reduced depthcharge payload loading
by 75 ms.

BUG=chrome-os-partner:56656,chrome-os-partner:59682

Change-Id: If87ee6f88e0ab0a463eafa35f89a5f7a7ad0fb85
Signed-off-by: Aaron Durbin <adurbin@chromium.org>
Reviewed-on: https://review.coreboot.org/17371
Tested-by: build bot (Jenkins)
Reviewed-by: Furquan Shaikh <furquan@google.com>
Reviewed-by: Andrey Petrov <andrey.petrov@intel.com>
Reviewed-by: Paul Menzel <paulepanter@users.sourceforge.net>
diff --git a/src/cpu/x86/mtrr/mtrr.c b/src/cpu/x86/mtrr/mtrr.c
index 8033e79..3ee907f 100644
--- a/src/cpu/x86/mtrr/mtrr.c
+++ b/src/cpu/x86/mtrr/mtrr.c
@@ -154,6 +154,24 @@
 	return 1;
 }
 
+static void print_physical_address_space(const struct memranges *addr_space,
+					const char *identifier)
+{
+	const struct range_entry *r;
+
+	if (identifier)
+		printk(BIOS_DEBUG, "MTRR: %s Physical address space:\n",
+			identifier);
+	else
+		printk(BIOS_DEBUG, "MTRR: Physical address space:\n");
+
+	memranges_each_entry(r, addr_space)
+		printk(BIOS_DEBUG,
+		       "0x%016llx - 0x%016llx size 0x%08llx type %ld\n",
+		       range_entry_base(r), range_entry_end(r),
+		       range_entry_size(r), range_entry_tag(r));
+}
+
 static struct memranges *get_physical_address_space(void)
 {
 	static struct memranges *addr_space;
@@ -163,7 +181,6 @@
 	 *  uncacheable ranges, such as graphics memory, at resource insertion
 	 * time remove uncacheable regions from the cacheable ones. */
 	if (addr_space == NULL) {
-		struct range_entry *r;
 		unsigned long mask;
 		unsigned long match;
 
@@ -193,12 +210,7 @@
 		                           RANGE_TO_PHYS_ADDR(RANGE_4GB),
 		                           MTRR_TYPE_UNCACHEABLE);
 
-		printk(BIOS_DEBUG, "MTRR: Physical address space:\n");
-		memranges_each_entry(r, addr_space)
-			printk(BIOS_DEBUG,
-			       "0x%016llx - 0x%016llx size 0x%08llx type %ld\n",
-			       range_entry_base(r), range_entry_end(r),
-			       range_entry_size(r), range_entry_tag(r));
+		print_physical_address_space(addr_space, NULL);
 	}
 
 	return addr_space;
@@ -380,11 +392,10 @@
 
 static void clear_var_mtrr(int index)
 {
-	msr_t msr_val;
+	msr_t msr = { .lo = 0, .hi = 0 };
 
-	msr_val = rdmsr(MTRR_PHYS_MASK(index));
-	msr_val.lo &= ~MTRR_PHYS_MASK_VALID;
-	wrmsr(MTRR_PHYS_MASK(index), msr_val);
+	wrmsr(MTRR_PHYS_BASE(index), msr);
+	wrmsr(MTRR_PHYS_MASK(index), msr);
 }
 
 static void prep_var_mtrr(struct var_mtrr_state *var_state,
@@ -817,3 +828,60 @@
 
 	post_code(0x93);
 }
+
+static bool put_back_original_solution;
+
+void mtrr_use_temp_range(uintptr_t begin, size_t size, int type)
+{
+	const struct range_entry *r;
+	const struct memranges *orig;
+	struct var_mtrr_solution sol;
+	struct memranges addr_space;
+	const int above4gb = 1; /* Cover above 4GiB by default. */
+	int address_bits;
+
+	/* Make a copy of the original address space and tweak it with the
+	 * provided range. */
+	memranges_init_empty(&addr_space, NULL, 0);
+	orig = get_physical_address_space();
+	memranges_each_entry(r, orig) {
+		unsigned long tag = range_entry_tag(r);
+
+		/* Remove any special tags from original solution. */
+		tag &= ~MTRR_RANGE_UC_USE_HOLE;
+
+		/* Remove any write combining MTRRs from the temporary
+		 * solution as it just fragments the address space. */
+		if (tag == MTRR_TYPE_WRCOMB)
+			tag = MTRR_TYPE_UNCACHEABLE;
+
+		memranges_insert(&addr_space, range_entry_base(r),
+				range_entry_size(r), tag);
+	}
+
+	/* Place new range into the address space. */
+	memranges_insert(&addr_space, begin, size, type);
+
+	print_physical_address_space(&addr_space, "TEMPORARY");
+
+	/* Calculate a new solution with the updated address space. */
+	address_bits = cpu_phys_address_size();
+	memset(&sol, 0, sizeof(sol));
+	sol.mtrr_default_type =
+		calc_var_mtrrs(&addr_space, above4gb, address_bits);
+	prepare_var_mtrrs(&addr_space, sol.mtrr_default_type,
+				above4gb, address_bits, &sol);
+	commit_var_mtrrs(&sol);
+
+	memranges_teardown(&addr_space);
+	put_back_original_solution = true;
+}
+
+static void remove_temp_solution(void *unused)
+{
+	if (put_back_original_solution)
+		commit_var_mtrrs(&mtrr_global_solution);
+}
+
+BOOT_STATE_INIT_ENTRY(BS_OS_RESUME, BS_ON_ENTRY, remove_temp_solution, NULL);
+BOOT_STATE_INIT_ENTRY(BS_PAYLOAD_LOAD, BS_ON_EXIT, remove_temp_solution, NULL);