xref: /linux/arch/x86/kernel/cpu/mtrr/generic.c (revision a7101d152665817bf7cafc47e7f5dcb1f54664fe)
1 /*
2  * This only handles 32bit MTRR on 32bit hosts. This is strictly wrong
3  * because MTRRs can span up to 40 bits (36bits on most modern x86)
4  */
5 #define DEBUG
6 
7 #include <linux/module.h>
8 #include <linux/init.h>
9 #include <linux/io.h>
10 #include <linux/mm.h>
11 
12 #include <asm/processor-flags.h>
13 #include <asm/cpufeature.h>
14 #include <asm/tlbflush.h>
15 #include <asm/mtrr.h>
16 #include <asm/msr.h>
17 #include <asm/pat.h>
18 
19 #include "mtrr.h"
20 
21 struct fixed_range_block {
22 	int base_msr;		/* start address of an MTRR block */
23 	int ranges;		/* number of MTRRs in this block  */
24 };
25 
26 static struct fixed_range_block fixed_range_blocks[] = {
27 	{ MSR_MTRRfix64K_00000, 1 }, /* one   64k MTRR  */
28 	{ MSR_MTRRfix16K_80000, 2 }, /* two   16k MTRRs */
29 	{ MSR_MTRRfix4K_C0000,  8 }, /* eight  4k MTRRs */
30 	{}
31 };
32 
33 static unsigned long smp_changes_mask;
34 static int mtrr_state_set;
35 u64 mtrr_tom2;
36 
37 struct mtrr_state_type mtrr_state;
38 EXPORT_SYMBOL_GPL(mtrr_state);
39 
40 /*
41  * BIOS is expected to clear MtrrFixDramModEn bit, see for example
42  * "BIOS and Kernel Developer's Guide for the AMD Athlon 64 and AMD
43  * Opteron Processors" (26094 Rev. 3.30 February 2006), section
44  * "13.2.1.2 SYSCFG Register": "The MtrrFixDramModEn bit should be set
45  * to 1 during BIOS initalization of the fixed MTRRs, then cleared to
46  * 0 for operation."
47  */
48 static inline void k8_check_syscfg_dram_mod_en(void)
49 {
50 	u32 lo, hi;
51 
52 	if (!((boot_cpu_data.x86_vendor == X86_VENDOR_AMD) &&
53 	      (boot_cpu_data.x86 >= 0x0f)))
54 		return;
55 
56 	rdmsr(MSR_K8_SYSCFG, lo, hi);
57 	if (lo & K8_MTRRFIXRANGE_DRAM_MODIFY) {
58 		printk(KERN_ERR FW_WARN "MTRR: CPU %u: SYSCFG[MtrrFixDramModEn]"
59 		       " not cleared by BIOS, clearing this bit\n",
60 		       smp_processor_id());
61 		lo &= ~K8_MTRRFIXRANGE_DRAM_MODIFY;
62 		mtrr_wrmsr(MSR_K8_SYSCFG, lo, hi);
63 	}
64 }
65 
66 /* Get the size of contiguous MTRR range */
67 static u64 get_mtrr_size(u64 mask)
68 {
69 	u64 size;
70 
71 	mask >>= PAGE_SHIFT;
72 	mask |= size_or_mask;
73 	size = -mask;
74 	size <<= PAGE_SHIFT;
75 	return size;
76 }
77 
78 /*
79  * Check and return the effective type for MTRR-MTRR type overlap.
80  * Returns 1 if the effective type is UNCACHEABLE, else returns 0
81  */
82 static int check_type_overlap(u8 *prev, u8 *curr)
83 {
84 	if (*prev == MTRR_TYPE_UNCACHABLE || *curr == MTRR_TYPE_UNCACHABLE) {
85 		*prev = MTRR_TYPE_UNCACHABLE;
86 		*curr = MTRR_TYPE_UNCACHABLE;
87 		return 1;
88 	}
89 
90 	if ((*prev == MTRR_TYPE_WRBACK && *curr == MTRR_TYPE_WRTHROUGH) ||
91 	    (*prev == MTRR_TYPE_WRTHROUGH && *curr == MTRR_TYPE_WRBACK)) {
92 		*prev = MTRR_TYPE_WRTHROUGH;
93 		*curr = MTRR_TYPE_WRTHROUGH;
94 	}
95 
96 	if (*prev != *curr) {
97 		*prev = MTRR_TYPE_UNCACHABLE;
98 		*curr = MTRR_TYPE_UNCACHABLE;
99 		return 1;
100 	}
101 
102 	return 0;
103 }
104 
105 /*
106  * Error/Semi-error returns:
107  * 0xFF - when MTRR is not enabled
108  * *repeat == 1 implies [start:end] spanned across MTRR range and type returned
109  *		corresponds only to [start:*partial_end].
110  *		Caller has to lookup again for [*partial_end:end].
111  */
112 static u8 __mtrr_type_lookup(u64 start, u64 end, u64 *partial_end, int *repeat)
113 {
114 	int i;
115 	u64 base, mask;
116 	u8 prev_match, curr_match;
117 
118 	*repeat = 0;
119 	if (!mtrr_state_set)
120 		return 0xFF;
121 
122 	if (!mtrr_state.enabled)
123 		return 0xFF;
124 
125 	/* Make end inclusive end, instead of exclusive */
126 	end--;
127 
128 	/* Look in fixed ranges. Just return the type as per start */
129 	if (mtrr_state.have_fixed && (start < 0x100000)) {
130 		int idx;
131 
132 		if (start < 0x80000) {
133 			idx = 0;
134 			idx += (start >> 16);
135 			return mtrr_state.fixed_ranges[idx];
136 		} else if (start < 0xC0000) {
137 			idx = 1 * 8;
138 			idx += ((start - 0x80000) >> 14);
139 			return mtrr_state.fixed_ranges[idx];
140 		} else if (start < 0x1000000) {
141 			idx = 3 * 8;
142 			idx += ((start - 0xC0000) >> 12);
143 			return mtrr_state.fixed_ranges[idx];
144 		}
145 	}
146 
147 	/*
148 	 * Look in variable ranges
149 	 * Look of multiple ranges matching this address and pick type
150 	 * as per MTRR precedence
151 	 */
152 	if (!(mtrr_state.enabled & 2))
153 		return mtrr_state.def_type;
154 
155 	prev_match = 0xFF;
156 	for (i = 0; i < num_var_ranges; ++i) {
157 		unsigned short start_state, end_state;
158 
159 		if (!(mtrr_state.var_ranges[i].mask_lo & (1 << 11)))
160 			continue;
161 
162 		base = (((u64)mtrr_state.var_ranges[i].base_hi) << 32) +
163 		       (mtrr_state.var_ranges[i].base_lo & PAGE_MASK);
164 		mask = (((u64)mtrr_state.var_ranges[i].mask_hi) << 32) +
165 		       (mtrr_state.var_ranges[i].mask_lo & PAGE_MASK);
166 
167 		start_state = ((start & mask) == (base & mask));
168 		end_state = ((end & mask) == (base & mask));
169 
170 		if (start_state != end_state) {
171 			/*
172 			 * We have start:end spanning across an MTRR.
173 			 * We split the region into
174 			 * either
175 			 * (start:mtrr_end) (mtrr_end:end)
176 			 * or
177 			 * (start:mtrr_start) (mtrr_start:end)
178 			 * depending on kind of overlap.
179 			 * Return the type for first region and a pointer to
180 			 * the start of second region so that caller will
181 			 * lookup again on the second region.
182 			 * Note: This way we handle multiple overlaps as well.
183 			 */
184 			if (start_state)
185 				*partial_end = base + get_mtrr_size(mask);
186 			else
187 				*partial_end = base;
188 
189 			if (unlikely(*partial_end <= start)) {
190 				WARN_ON(1);
191 				*partial_end = start + PAGE_SIZE;
192 			}
193 
194 			end = *partial_end - 1; /* end is inclusive */
195 			*repeat = 1;
196 		}
197 
198 		if ((start & mask) != (base & mask))
199 			continue;
200 
201 		curr_match = mtrr_state.var_ranges[i].base_lo & 0xff;
202 		if (prev_match == 0xFF) {
203 			prev_match = curr_match;
204 			continue;
205 		}
206 
207 		if (check_type_overlap(&prev_match, &curr_match))
208 			return curr_match;
209 	}
210 
211 	if (mtrr_tom2) {
212 		if (start >= (1ULL<<32) && (end < mtrr_tom2))
213 			return MTRR_TYPE_WRBACK;
214 	}
215 
216 	if (prev_match != 0xFF)
217 		return prev_match;
218 
219 	return mtrr_state.def_type;
220 }
221 
222 /*
223  * Returns the effective MTRR type for the region
224  * Error return:
225  * 0xFF - when MTRR is not enabled
226  */
227 u8 mtrr_type_lookup(u64 start, u64 end)
228 {
229 	u8 type, prev_type;
230 	int repeat;
231 	u64 partial_end;
232 
233 	type = __mtrr_type_lookup(start, end, &partial_end, &repeat);
234 
235 	/*
236 	 * Common path is with repeat = 0.
237 	 * However, we can have cases where [start:end] spans across some
238 	 * MTRR range. Do repeated lookups for that case here.
239 	 */
240 	while (repeat) {
241 		prev_type = type;
242 		start = partial_end;
243 		type = __mtrr_type_lookup(start, end, &partial_end, &repeat);
244 
245 		if (check_type_overlap(&prev_type, &type))
246 			return type;
247 	}
248 
249 	return type;
250 }
251 
252 /* Get the MSR pair relating to a var range */
253 static void
254 get_mtrr_var_range(unsigned int index, struct mtrr_var_range *vr)
255 {
256 	rdmsr(MTRRphysBase_MSR(index), vr->base_lo, vr->base_hi);
257 	rdmsr(MTRRphysMask_MSR(index), vr->mask_lo, vr->mask_hi);
258 }
259 
260 /* Fill the MSR pair relating to a var range */
261 void fill_mtrr_var_range(unsigned int index,
262 		u32 base_lo, u32 base_hi, u32 mask_lo, u32 mask_hi)
263 {
264 	struct mtrr_var_range *vr;
265 
266 	vr = mtrr_state.var_ranges;
267 
268 	vr[index].base_lo = base_lo;
269 	vr[index].base_hi = base_hi;
270 	vr[index].mask_lo = mask_lo;
271 	vr[index].mask_hi = mask_hi;
272 }
273 
274 static void get_fixed_ranges(mtrr_type *frs)
275 {
276 	unsigned int *p = (unsigned int *)frs;
277 	int i;
278 
279 	k8_check_syscfg_dram_mod_en();
280 
281 	rdmsr(MSR_MTRRfix64K_00000, p[0], p[1]);
282 
283 	for (i = 0; i < 2; i++)
284 		rdmsr(MSR_MTRRfix16K_80000 + i, p[2 + i * 2], p[3 + i * 2]);
285 	for (i = 0; i < 8; i++)
286 		rdmsr(MSR_MTRRfix4K_C0000 + i, p[6 + i * 2], p[7 + i * 2]);
287 }
288 
289 void mtrr_save_fixed_ranges(void *info)
290 {
291 	if (cpu_has_mtrr)
292 		get_fixed_ranges(mtrr_state.fixed_ranges);
293 }
294 
295 static unsigned __initdata last_fixed_start;
296 static unsigned __initdata last_fixed_end;
297 static mtrr_type __initdata last_fixed_type;
298 
299 static void __init print_fixed_last(void)
300 {
301 	if (!last_fixed_end)
302 		return;
303 
304 	pr_debug("  %05X-%05X %s\n", last_fixed_start,
305 		 last_fixed_end - 1, mtrr_attrib_to_str(last_fixed_type));
306 
307 	last_fixed_end = 0;
308 }
309 
310 static void __init update_fixed_last(unsigned base, unsigned end,
311 				     mtrr_type type)
312 {
313 	last_fixed_start = base;
314 	last_fixed_end = end;
315 	last_fixed_type = type;
316 }
317 
318 static void __init
319 print_fixed(unsigned base, unsigned step, const mtrr_type *types)
320 {
321 	unsigned i;
322 
323 	for (i = 0; i < 8; ++i, ++types, base += step) {
324 		if (last_fixed_end == 0) {
325 			update_fixed_last(base, base + step, *types);
326 			continue;
327 		}
328 		if (last_fixed_end == base && last_fixed_type == *types) {
329 			last_fixed_end = base + step;
330 			continue;
331 		}
332 		/* new segments: gap or different type */
333 		print_fixed_last();
334 		update_fixed_last(base, base + step, *types);
335 	}
336 }
337 
338 static void prepare_set(void);
339 static void post_set(void);
340 
341 static void __init print_mtrr_state(void)
342 {
343 	unsigned int i;
344 	int high_width;
345 
346 	pr_debug("MTRR default type: %s\n",
347 		 mtrr_attrib_to_str(mtrr_state.def_type));
348 	if (mtrr_state.have_fixed) {
349 		pr_debug("MTRR fixed ranges %sabled:\n",
350 			 mtrr_state.enabled & 1 ? "en" : "dis");
351 		print_fixed(0x00000, 0x10000, mtrr_state.fixed_ranges + 0);
352 		for (i = 0; i < 2; ++i)
353 			print_fixed(0x80000 + i * 0x20000, 0x04000,
354 				    mtrr_state.fixed_ranges + (i + 1) * 8);
355 		for (i = 0; i < 8; ++i)
356 			print_fixed(0xC0000 + i * 0x08000, 0x01000,
357 				    mtrr_state.fixed_ranges + (i + 3) * 8);
358 
359 		/* tail */
360 		print_fixed_last();
361 	}
362 	pr_debug("MTRR variable ranges %sabled:\n",
363 		 mtrr_state.enabled & 2 ? "en" : "dis");
364 	high_width = (__ffs64(size_or_mask) - (32 - PAGE_SHIFT) + 3) / 4;
365 
366 	for (i = 0; i < num_var_ranges; ++i) {
367 		if (mtrr_state.var_ranges[i].mask_lo & (1 << 11))
368 			pr_debug("  %u base %0*X%05X000 mask %0*X%05X000 %s\n",
369 				 i,
370 				 high_width,
371 				 mtrr_state.var_ranges[i].base_hi,
372 				 mtrr_state.var_ranges[i].base_lo >> 12,
373 				 high_width,
374 				 mtrr_state.var_ranges[i].mask_hi,
375 				 mtrr_state.var_ranges[i].mask_lo >> 12,
376 				 mtrr_attrib_to_str(mtrr_state.var_ranges[i].base_lo & 0xff));
377 		else
378 			pr_debug("  %u disabled\n", i);
379 	}
380 	if (mtrr_tom2)
381 		pr_debug("TOM2: %016llx aka %lldM\n", mtrr_tom2, mtrr_tom2>>20);
382 }
383 
384 /* Grab all of the MTRR state for this CPU into *state */
385 void __init get_mtrr_state(void)
386 {
387 	struct mtrr_var_range *vrs;
388 	unsigned long flags;
389 	unsigned lo, dummy;
390 	unsigned int i;
391 
392 	vrs = mtrr_state.var_ranges;
393 
394 	rdmsr(MSR_MTRRcap, lo, dummy);
395 	mtrr_state.have_fixed = (lo >> 8) & 1;
396 
397 	for (i = 0; i < num_var_ranges; i++)
398 		get_mtrr_var_range(i, &vrs[i]);
399 	if (mtrr_state.have_fixed)
400 		get_fixed_ranges(mtrr_state.fixed_ranges);
401 
402 	rdmsr(MSR_MTRRdefType, lo, dummy);
403 	mtrr_state.def_type = (lo & 0xff);
404 	mtrr_state.enabled = (lo & 0xc00) >> 10;
405 
406 	if (amd_special_default_mtrr()) {
407 		unsigned low, high;
408 
409 		/* TOP_MEM2 */
410 		rdmsr(MSR_K8_TOP_MEM2, low, high);
411 		mtrr_tom2 = high;
412 		mtrr_tom2 <<= 32;
413 		mtrr_tom2 |= low;
414 		mtrr_tom2 &= 0xffffff800000ULL;
415 	}
416 
417 	print_mtrr_state();
418 
419 	mtrr_state_set = 1;
420 
421 	/* PAT setup for BP. We need to go through sync steps here */
422 	local_irq_save(flags);
423 	prepare_set();
424 
425 	pat_init();
426 
427 	post_set();
428 	local_irq_restore(flags);
429 }
430 
431 /* Some BIOS's are messed up and don't set all MTRRs the same! */
432 void __init mtrr_state_warn(void)
433 {
434 	unsigned long mask = smp_changes_mask;
435 
436 	if (!mask)
437 		return;
438 	if (mask & MTRR_CHANGE_MASK_FIXED)
439 		pr_warning("mtrr: your CPUs had inconsistent fixed MTRR settings\n");
440 	if (mask & MTRR_CHANGE_MASK_VARIABLE)
441 		pr_warning("mtrr: your CPUs had inconsistent variable MTRR settings\n");
442 	if (mask & MTRR_CHANGE_MASK_DEFTYPE)
443 		pr_warning("mtrr: your CPUs had inconsistent MTRRdefType settings\n");
444 
445 	printk(KERN_INFO "mtrr: probably your BIOS does not setup all CPUs.\n");
446 	printk(KERN_INFO "mtrr: corrected configuration.\n");
447 }
448 
449 /*
450  * Doesn't attempt to pass an error out to MTRR users
451  * because it's quite complicated in some cases and probably not
452  * worth it because the best error handling is to ignore it.
453  */
454 void mtrr_wrmsr(unsigned msr, unsigned a, unsigned b)
455 {
456 	if (wrmsr_safe(msr, a, b) < 0) {
457 		printk(KERN_ERR
458 			"MTRR: CPU %u: Writing MSR %x to %x:%x failed\n",
459 			smp_processor_id(), msr, a, b);
460 	}
461 }
462 
463 /**
464  * set_fixed_range - checks & updates a fixed-range MTRR if it
465  *		     differs from the value it should have
466  * @msr: MSR address of the MTTR which should be checked and updated
467  * @changed: pointer which indicates whether the MTRR needed to be changed
468  * @msrwords: pointer to the MSR values which the MSR should have
469  */
470 static void set_fixed_range(int msr, bool *changed, unsigned int *msrwords)
471 {
472 	unsigned lo, hi;
473 
474 	rdmsr(msr, lo, hi);
475 
476 	if (lo != msrwords[0] || hi != msrwords[1]) {
477 		mtrr_wrmsr(msr, msrwords[0], msrwords[1]);
478 		*changed = true;
479 	}
480 }
481 
482 /**
483  * generic_get_free_region - Get a free MTRR.
484  * @base: The starting (base) address of the region.
485  * @size: The size (in bytes) of the region.
486  * @replace_reg: mtrr index to be replaced; set to invalid value if none.
487  *
488  * Returns: The index of the region on success, else negative on error.
489  */
490 int
491 generic_get_free_region(unsigned long base, unsigned long size, int replace_reg)
492 {
493 	unsigned long lbase, lsize;
494 	mtrr_type ltype;
495 	int i, max;
496 
497 	max = num_var_ranges;
498 	if (replace_reg >= 0 && replace_reg < max)
499 		return replace_reg;
500 
501 	for (i = 0; i < max; ++i) {
502 		mtrr_if->get(i, &lbase, &lsize, &ltype);
503 		if (lsize == 0)
504 			return i;
505 	}
506 
507 	return -ENOSPC;
508 }
509 
510 static void generic_get_mtrr(unsigned int reg, unsigned long *base,
511 			     unsigned long *size, mtrr_type *type)
512 {
513 	unsigned int mask_lo, mask_hi, base_lo, base_hi;
514 	unsigned int tmp, hi;
515 
516 	/*
517 	 * get_mtrr doesn't need to update mtrr_state, also it could be called
518 	 * from any cpu, so try to print it out directly.
519 	 */
520 	get_cpu();
521 
522 	rdmsr(MTRRphysMask_MSR(reg), mask_lo, mask_hi);
523 
524 	if ((mask_lo & 0x800) == 0) {
525 		/*  Invalid (i.e. free) range */
526 		*base = 0;
527 		*size = 0;
528 		*type = 0;
529 		goto out_put_cpu;
530 	}
531 
532 	rdmsr(MTRRphysBase_MSR(reg), base_lo, base_hi);
533 
534 	/* Work out the shifted address mask: */
535 	tmp = mask_hi << (32 - PAGE_SHIFT) | mask_lo >> PAGE_SHIFT;
536 	mask_lo = size_or_mask | tmp;
537 
538 	/* Expand tmp with high bits to all 1s: */
539 	hi = fls(tmp);
540 	if (hi > 0) {
541 		tmp |= ~((1<<(hi - 1)) - 1);
542 
543 		if (tmp != mask_lo) {
544 			printk(KERN_WARNING "mtrr: your BIOS has configured an incorrect mask, fixing it.\n");
545 			add_taint(TAINT_FIRMWARE_WORKAROUND);
546 			mask_lo = tmp;
547 		}
548 	}
549 
550 	/*
551 	 * This works correctly if size is a power of two, i.e. a
552 	 * contiguous range:
553 	 */
554 	*size = -mask_lo;
555 	*base = base_hi << (32 - PAGE_SHIFT) | base_lo >> PAGE_SHIFT;
556 	*type = base_lo & 0xff;
557 
558 out_put_cpu:
559 	put_cpu();
560 }
561 
562 /**
563  * set_fixed_ranges - checks & updates the fixed-range MTRRs if they
564  *		      differ from the saved set
565  * @frs: pointer to fixed-range MTRR values, saved by get_fixed_ranges()
566  */
567 static int set_fixed_ranges(mtrr_type *frs)
568 {
569 	unsigned long long *saved = (unsigned long long *)frs;
570 	bool changed = false;
571 	int block = -1, range;
572 
573 	k8_check_syscfg_dram_mod_en();
574 
575 	while (fixed_range_blocks[++block].ranges) {
576 		for (range = 0; range < fixed_range_blocks[block].ranges; range++)
577 			set_fixed_range(fixed_range_blocks[block].base_msr + range,
578 					&changed, (unsigned int *)saved++);
579 	}
580 
581 	return changed;
582 }
583 
584 /*
585  * Set the MSR pair relating to a var range.
586  * Returns true if changes are made.
587  */
588 static bool set_mtrr_var_ranges(unsigned int index, struct mtrr_var_range *vr)
589 {
590 	unsigned int lo, hi;
591 	bool changed = false;
592 
593 	rdmsr(MTRRphysBase_MSR(index), lo, hi);
594 	if ((vr->base_lo & 0xfffff0ffUL) != (lo & 0xfffff0ffUL)
595 	    || (vr->base_hi & (size_and_mask >> (32 - PAGE_SHIFT))) !=
596 		(hi & (size_and_mask >> (32 - PAGE_SHIFT)))) {
597 
598 		mtrr_wrmsr(MTRRphysBase_MSR(index), vr->base_lo, vr->base_hi);
599 		changed = true;
600 	}
601 
602 	rdmsr(MTRRphysMask_MSR(index), lo, hi);
603 
604 	if ((vr->mask_lo & 0xfffff800UL) != (lo & 0xfffff800UL)
605 	    || (vr->mask_hi & (size_and_mask >> (32 - PAGE_SHIFT))) !=
606 		(hi & (size_and_mask >> (32 - PAGE_SHIFT)))) {
607 		mtrr_wrmsr(MTRRphysMask_MSR(index), vr->mask_lo, vr->mask_hi);
608 		changed = true;
609 	}
610 	return changed;
611 }
612 
613 static u32 deftype_lo, deftype_hi;
614 
615 /**
616  * set_mtrr_state - Set the MTRR state for this CPU.
617  *
618  * NOTE: The CPU must already be in a safe state for MTRR changes.
619  * RETURNS: 0 if no changes made, else a mask indicating what was changed.
620  */
621 static unsigned long set_mtrr_state(void)
622 {
623 	unsigned long change_mask = 0;
624 	unsigned int i;
625 
626 	for (i = 0; i < num_var_ranges; i++) {
627 		if (set_mtrr_var_ranges(i, &mtrr_state.var_ranges[i]))
628 			change_mask |= MTRR_CHANGE_MASK_VARIABLE;
629 	}
630 
631 	if (mtrr_state.have_fixed && set_fixed_ranges(mtrr_state.fixed_ranges))
632 		change_mask |= MTRR_CHANGE_MASK_FIXED;
633 
634 	/*
635 	 * Set_mtrr_restore restores the old value of MTRRdefType,
636 	 * so to set it we fiddle with the saved value:
637 	 */
638 	if ((deftype_lo & 0xff) != mtrr_state.def_type
639 	    || ((deftype_lo & 0xc00) >> 10) != mtrr_state.enabled) {
640 
641 		deftype_lo = (deftype_lo & ~0xcff) | mtrr_state.def_type |
642 			     (mtrr_state.enabled << 10);
643 		change_mask |= MTRR_CHANGE_MASK_DEFTYPE;
644 	}
645 
646 	return change_mask;
647 }
648 
649 
650 static unsigned long cr4;
651 static DEFINE_RAW_SPINLOCK(set_atomicity_lock);
652 
653 /*
654  * Since we are disabling the cache don't allow any interrupts,
655  * they would run extremely slow and would only increase the pain.
656  *
657  * The caller must ensure that local interrupts are disabled and
658  * are reenabled after post_set() has been called.
659  */
660 static void prepare_set(void) __acquires(set_atomicity_lock)
661 {
662 	unsigned long cr0;
663 
664 	/*
665 	 * Note that this is not ideal
666 	 * since the cache is only flushed/disabled for this CPU while the
667 	 * MTRRs are changed, but changing this requires more invasive
668 	 * changes to the way the kernel boots
669 	 */
670 
671 	raw_spin_lock(&set_atomicity_lock);
672 
673 	/* Enter the no-fill (CD=1, NW=0) cache mode and flush caches. */
674 	cr0 = read_cr0() | X86_CR0_CD;
675 	write_cr0(cr0);
676 	wbinvd();
677 
678 	/* Save value of CR4 and clear Page Global Enable (bit 7) */
679 	if (cpu_has_pge) {
680 		cr4 = read_cr4();
681 		write_cr4(cr4 & ~X86_CR4_PGE);
682 	}
683 
684 	/* Flush all TLBs via a mov %cr3, %reg; mov %reg, %cr3 */
685 	__flush_tlb();
686 
687 	/* Save MTRR state */
688 	rdmsr(MSR_MTRRdefType, deftype_lo, deftype_hi);
689 
690 	/* Disable MTRRs, and set the default type to uncached */
691 	mtrr_wrmsr(MSR_MTRRdefType, deftype_lo & ~0xcff, deftype_hi);
692 	wbinvd();
693 }
694 
695 static void post_set(void) __releases(set_atomicity_lock)
696 {
697 	/* Flush TLBs (no need to flush caches - they are disabled) */
698 	__flush_tlb();
699 
700 	/* Intel (P6) standard MTRRs */
701 	mtrr_wrmsr(MSR_MTRRdefType, deftype_lo, deftype_hi);
702 
703 	/* Enable caches */
704 	write_cr0(read_cr0() & 0xbfffffff);
705 
706 	/* Restore value of CR4 */
707 	if (cpu_has_pge)
708 		write_cr4(cr4);
709 	raw_spin_unlock(&set_atomicity_lock);
710 }
711 
712 static void generic_set_all(void)
713 {
714 	unsigned long mask, count;
715 	unsigned long flags;
716 
717 	local_irq_save(flags);
718 	prepare_set();
719 
720 	/* Actually set the state */
721 	mask = set_mtrr_state();
722 
723 	/* also set PAT */
724 	pat_init();
725 
726 	post_set();
727 	local_irq_restore(flags);
728 
729 	/* Use the atomic bitops to update the global mask */
730 	for (count = 0; count < sizeof mask * 8; ++count) {
731 		if (mask & 0x01)
732 			set_bit(count, &smp_changes_mask);
733 		mask >>= 1;
734 	}
735 
736 }
737 
738 /**
739  * generic_set_mtrr - set variable MTRR register on the local CPU.
740  *
741  * @reg: The register to set.
742  * @base: The base address of the region.
743  * @size: The size of the region. If this is 0 the region is disabled.
744  * @type: The type of the region.
745  *
746  * Returns nothing.
747  */
748 static void generic_set_mtrr(unsigned int reg, unsigned long base,
749 			     unsigned long size, mtrr_type type)
750 {
751 	unsigned long flags;
752 	struct mtrr_var_range *vr;
753 
754 	vr = &mtrr_state.var_ranges[reg];
755 
756 	local_irq_save(flags);
757 	prepare_set();
758 
759 	if (size == 0) {
760 		/*
761 		 * The invalid bit is kept in the mask, so we simply
762 		 * clear the relevant mask register to disable a range.
763 		 */
764 		mtrr_wrmsr(MTRRphysMask_MSR(reg), 0, 0);
765 		memset(vr, 0, sizeof(struct mtrr_var_range));
766 	} else {
767 		vr->base_lo = base << PAGE_SHIFT | type;
768 		vr->base_hi = (base & size_and_mask) >> (32 - PAGE_SHIFT);
769 		vr->mask_lo = -size << PAGE_SHIFT | 0x800;
770 		vr->mask_hi = (-size & size_and_mask) >> (32 - PAGE_SHIFT);
771 
772 		mtrr_wrmsr(MTRRphysBase_MSR(reg), vr->base_lo, vr->base_hi);
773 		mtrr_wrmsr(MTRRphysMask_MSR(reg), vr->mask_lo, vr->mask_hi);
774 	}
775 
776 	post_set();
777 	local_irq_restore(flags);
778 }
779 
780 int generic_validate_add_page(unsigned long base, unsigned long size,
781 			      unsigned int type)
782 {
783 	unsigned long lbase, last;
784 
785 	/*
786 	 * For Intel PPro stepping <= 7
787 	 * must be 4 MiB aligned and not touch 0x70000000 -> 0x7003FFFF
788 	 */
789 	if (is_cpu(INTEL) && boot_cpu_data.x86 == 6 &&
790 	    boot_cpu_data.x86_model == 1 &&
791 	    boot_cpu_data.x86_mask <= 7) {
792 		if (base & ((1 << (22 - PAGE_SHIFT)) - 1)) {
793 			pr_warning("mtrr: base(0x%lx000) is not 4 MiB aligned\n", base);
794 			return -EINVAL;
795 		}
796 		if (!(base + size < 0x70000 || base > 0x7003F) &&
797 		    (type == MTRR_TYPE_WRCOMB
798 		     || type == MTRR_TYPE_WRBACK)) {
799 			pr_warning("mtrr: writable mtrr between 0x70000000 and 0x7003FFFF may hang the CPU.\n");
800 			return -EINVAL;
801 		}
802 	}
803 
804 	/*
805 	 * Check upper bits of base and last are equal and lower bits are 0
806 	 * for base and 1 for last
807 	 */
808 	last = base + size - 1;
809 	for (lbase = base; !(lbase & 1) && (last & 1);
810 	     lbase = lbase >> 1, last = last >> 1)
811 		;
812 	if (lbase != last) {
813 		pr_warning("mtrr: base(0x%lx000) is not aligned on a size(0x%lx000) boundary\n", base, size);
814 		return -EINVAL;
815 	}
816 	return 0;
817 }
818 
819 static int generic_have_wrcomb(void)
820 {
821 	unsigned long config, dummy;
822 	rdmsr(MSR_MTRRcap, config, dummy);
823 	return config & (1 << 10);
824 }
825 
826 int positive_have_wrcomb(void)
827 {
828 	return 1;
829 }
830 
831 /*
832  * Generic structure...
833  */
834 const struct mtrr_ops generic_mtrr_ops = {
835 	.use_intel_if		= 1,
836 	.set_all		= generic_set_all,
837 	.get			= generic_get_mtrr,
838 	.get_free_region	= generic_get_free_region,
839 	.set			= generic_set_mtrr,
840 	.validate_add_page	= generic_validate_add_page,
841 	.have_wrcomb		= generic_have_wrcomb,
842 };
843