xref: /freebsd-13-stable/sys/amd64/amd64/pmap.c (revision 8490d5ec8e0e642de1cf8d1ed474bdd45edb7dbc)
1 /*-
2  * SPDX-License-Identifier: BSD-4-Clause
3  *
4  * Copyright (c) 1991 Regents of the University of California.
5  * All rights reserved.
6  * Copyright (c) 1994 John S. Dyson
7  * All rights reserved.
8  * Copyright (c) 1994 David Greenman
9  * All rights reserved.
10  * Copyright (c) 2003 Peter Wemm
11  * All rights reserved.
12  * Copyright (c) 2005-2010 Alan L. Cox <alc@cs.rice.edu>
13  * All rights reserved.
14  *
15  * This code is derived from software contributed to Berkeley by
16  * the Systems Programming Group of the University of Utah Computer
17  * Science Department and William Jolitz of UUNET Technologies Inc.
18  *
19  * Redistribution and use in source and binary forms, with or without
20  * modification, are permitted provided that the following conditions
21  * are met:
22  * 1. Redistributions of source code must retain the above copyright
23  *    notice, this list of conditions and the following disclaimer.
24  * 2. Redistributions in binary form must reproduce the above copyright
25  *    notice, this list of conditions and the following disclaimer in the
26  *    documentation and/or other materials provided with the distribution.
27  * 3. All advertising materials mentioning features or use of this software
28  *    must display the following acknowledgement:
29  *	This product includes software developed by the University of
30  *	California, Berkeley and its contributors.
31  * 4. Neither the name of the University nor the names of its contributors
32  *    may be used to endorse or promote products derived from this software
33  *    without specific prior written permission.
34  *
35  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
36  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
37  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
38  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
39  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
40  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
41  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
42  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
43  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
44  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
45  * SUCH DAMAGE.
46  *
47  *	from:	@(#)pmap.c	7.7 (Berkeley)	5/12/91
48  */
49 /*-
50  * Copyright (c) 2003 Networks Associates Technology, Inc.
51  * Copyright (c) 2014-2020 The FreeBSD Foundation
52  * All rights reserved.
53  *
54  * This software was developed for the FreeBSD Project by Jake Burkholder,
55  * Safeport Network Services, and Network Associates Laboratories, the
56  * Security Research Division of Network Associates, Inc. under
57  * DARPA/SPAWAR contract N66001-01-C-8035 ("CBOSS"), as part of the DARPA
58  * CHATS research program.
59  *
60  * Portions of this software were developed by
61  * Konstantin Belousov <kib@FreeBSD.org> under sponsorship from
62  * the FreeBSD Foundation.
63  *
64  * Redistribution and use in source and binary forms, with or without
65  * modification, are permitted provided that the following conditions
66  * are met:
67  * 1. Redistributions of source code must retain the above copyright
68  *    notice, this list of conditions and the following disclaimer.
69  * 2. Redistributions in binary form must reproduce the above copyright
70  *    notice, this list of conditions and the following disclaimer in the
71  *    documentation and/or other materials provided with the distribution.
72  *
73  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
74  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
75  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
76  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
77  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
78  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
79  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
80  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
81  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
82  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
83  * SUCH DAMAGE.
84  */
85 
86 #define	AMD64_NPT_AWARE
87 
88 #include <sys/cdefs.h>
89 /*
90  *	Manages physical address maps.
91  *
92  *	Since the information managed by this module is
93  *	also stored by the logical address mapping module,
94  *	this module may throw away valid virtual-to-physical
95  *	mappings at almost any time.  However, invalidations
96  *	of virtual-to-physical mappings must be done as
97  *	requested.
98  *
99  *	In order to cope with hardware architectures which
100  *	make virtual-to-physical map invalidates expensive,
101  *	this module may delay invalidate or reduced protection
102  *	operations until such time as they are actually
103  *	necessary.  This module is given full information as
104  *	to which processors are currently using which maps,
105  *	and to when physical maps must be made correct.
106  */
107 
108 #include "opt_ddb.h"
109 #include "opt_pmap.h"
110 #include "opt_vm.h"
111 
112 #include <sys/param.h>
113 #include <sys/asan.h>
114 #include <sys/bitstring.h>
115 #include <sys/bus.h>
116 #include <sys/systm.h>
117 #include <sys/counter.h>
118 #include <sys/kernel.h>
119 #include <sys/ktr.h>
120 #include <sys/lock.h>
121 #include <sys/malloc.h>
122 #include <sys/mman.h>
123 #include <sys/mutex.h>
124 #include <sys/proc.h>
125 #include <sys/rangeset.h>
126 #include <sys/rwlock.h>
127 #include <sys/sbuf.h>
128 #include <sys/smr.h>
129 #include <sys/sx.h>
130 #include <sys/turnstile.h>
131 #include <sys/vmem.h>
132 #include <sys/vmmeter.h>
133 #include <sys/sched.h>
134 #include <sys/sysctl.h>
135 #include <sys/smp.h>
136 #ifdef DDB
137 #include <sys/kdb.h>
138 #include <ddb/ddb.h>
139 #endif
140 
141 #include <vm/vm.h>
142 #include <vm/vm_param.h>
143 #include <vm/vm_kern.h>
144 #include <vm/vm_page.h>
145 #include <vm/vm_map.h>
146 #include <vm/vm_object.h>
147 #include <vm/vm_extern.h>
148 #include <vm/vm_pageout.h>
149 #include <vm/vm_pager.h>
150 #include <vm/vm_phys.h>
151 #include <vm/vm_radix.h>
152 #include <vm/vm_reserv.h>
153 #include <vm/vm_dumpset.h>
154 #include <vm/uma.h>
155 
156 #include <machine/asan.h>
157 #include <machine/intr_machdep.h>
158 #include <x86/apicvar.h>
159 #include <x86/ifunc.h>
160 #include <machine/cpu.h>
161 #include <machine/cputypes.h>
162 #include <machine/intr_machdep.h>
163 #include <machine/md_var.h>
164 #include <machine/pcb.h>
165 #include <machine/specialreg.h>
166 #ifdef SMP
167 #include <machine/smp.h>
168 #endif
169 #include <machine/sysarch.h>
170 #include <machine/tss.h>
171 
172 #ifdef NUMA
173 #define	PMAP_MEMDOM	MAXMEMDOM
174 #else
175 #define	PMAP_MEMDOM	1
176 #endif
177 
178 static __inline boolean_t
pmap_type_guest(pmap_t pmap)179 pmap_type_guest(pmap_t pmap)
180 {
181 
182 	return ((pmap->pm_type == PT_EPT) || (pmap->pm_type == PT_RVI));
183 }
184 
185 static __inline boolean_t
pmap_emulate_ad_bits(pmap_t pmap)186 pmap_emulate_ad_bits(pmap_t pmap)
187 {
188 
189 	return ((pmap->pm_flags & PMAP_EMULATE_AD_BITS) != 0);
190 }
191 
192 static __inline pt_entry_t
pmap_valid_bit(pmap_t pmap)193 pmap_valid_bit(pmap_t pmap)
194 {
195 	pt_entry_t mask;
196 
197 	switch (pmap->pm_type) {
198 	case PT_X86:
199 	case PT_RVI:
200 		mask = X86_PG_V;
201 		break;
202 	case PT_EPT:
203 		if (pmap_emulate_ad_bits(pmap))
204 			mask = EPT_PG_EMUL_V;
205 		else
206 			mask = EPT_PG_READ;
207 		break;
208 	default:
209 		panic("pmap_valid_bit: invalid pm_type %d", pmap->pm_type);
210 	}
211 
212 	return (mask);
213 }
214 
215 static __inline pt_entry_t
pmap_rw_bit(pmap_t pmap)216 pmap_rw_bit(pmap_t pmap)
217 {
218 	pt_entry_t mask;
219 
220 	switch (pmap->pm_type) {
221 	case PT_X86:
222 	case PT_RVI:
223 		mask = X86_PG_RW;
224 		break;
225 	case PT_EPT:
226 		if (pmap_emulate_ad_bits(pmap))
227 			mask = EPT_PG_EMUL_RW;
228 		else
229 			mask = EPT_PG_WRITE;
230 		break;
231 	default:
232 		panic("pmap_rw_bit: invalid pm_type %d", pmap->pm_type);
233 	}
234 
235 	return (mask);
236 }
237 
238 static pt_entry_t pg_g;
239 
240 static __inline pt_entry_t
pmap_global_bit(pmap_t pmap)241 pmap_global_bit(pmap_t pmap)
242 {
243 	pt_entry_t mask;
244 
245 	switch (pmap->pm_type) {
246 	case PT_X86:
247 		mask = pg_g;
248 		break;
249 	case PT_RVI:
250 	case PT_EPT:
251 		mask = 0;
252 		break;
253 	default:
254 		panic("pmap_global_bit: invalid pm_type %d", pmap->pm_type);
255 	}
256 
257 	return (mask);
258 }
259 
260 static __inline pt_entry_t
pmap_accessed_bit(pmap_t pmap)261 pmap_accessed_bit(pmap_t pmap)
262 {
263 	pt_entry_t mask;
264 
265 	switch (pmap->pm_type) {
266 	case PT_X86:
267 	case PT_RVI:
268 		mask = X86_PG_A;
269 		break;
270 	case PT_EPT:
271 		if (pmap_emulate_ad_bits(pmap))
272 			mask = EPT_PG_READ;
273 		else
274 			mask = EPT_PG_A;
275 		break;
276 	default:
277 		panic("pmap_accessed_bit: invalid pm_type %d", pmap->pm_type);
278 	}
279 
280 	return (mask);
281 }
282 
283 static __inline pt_entry_t
pmap_modified_bit(pmap_t pmap)284 pmap_modified_bit(pmap_t pmap)
285 {
286 	pt_entry_t mask;
287 
288 	switch (pmap->pm_type) {
289 	case PT_X86:
290 	case PT_RVI:
291 		mask = X86_PG_M;
292 		break;
293 	case PT_EPT:
294 		if (pmap_emulate_ad_bits(pmap))
295 			mask = EPT_PG_WRITE;
296 		else
297 			mask = EPT_PG_M;
298 		break;
299 	default:
300 		panic("pmap_modified_bit: invalid pm_type %d", pmap->pm_type);
301 	}
302 
303 	return (mask);
304 }
305 
306 static __inline pt_entry_t
pmap_pku_mask_bit(pmap_t pmap)307 pmap_pku_mask_bit(pmap_t pmap)
308 {
309 
310 	return (pmap->pm_type == PT_X86 ? X86_PG_PKU_MASK : 0);
311 }
312 
313 #if !defined(DIAGNOSTIC)
314 #ifdef __GNUC_GNU_INLINE__
315 #define PMAP_INLINE	__attribute__((__gnu_inline__)) inline
316 #else
317 #define PMAP_INLINE	extern inline
318 #endif
319 #else
320 #define PMAP_INLINE
321 #endif
322 
323 #ifdef PV_STATS
324 #define PV_STAT(x)	do { x ; } while (0)
325 #else
326 #define PV_STAT(x)	do { } while (0)
327 #endif
328 
329 #undef pa_index
330 #ifdef NUMA
331 #define	pa_index(pa)	({					\
332 	KASSERT((pa) <= vm_phys_segs[vm_phys_nsegs - 1].end,	\
333 	    ("address %lx beyond the last segment", (pa)));	\
334 	(pa) >> PDRSHIFT;					\
335 })
336 #define	pa_to_pmdp(pa)	(&pv_table[pa_index(pa)])
337 #define	pa_to_pvh(pa)	(&(pa_to_pmdp(pa)->pv_page))
338 #define	PHYS_TO_PV_LIST_LOCK(pa)	({			\
339 	struct rwlock *_lock;					\
340 	if (__predict_false((pa) > pmap_last_pa))		\
341 		_lock = &pv_dummy_large.pv_lock;		\
342 	else							\
343 		_lock = &(pa_to_pmdp(pa)->pv_lock);		\
344 	_lock;							\
345 })
346 #else
347 #define	pa_index(pa)	((pa) >> PDRSHIFT)
348 #define	pa_to_pvh(pa)	(&pv_table[pa_index(pa)])
349 
350 #define	NPV_LIST_LOCKS	MAXCPU
351 
352 #define	PHYS_TO_PV_LIST_LOCK(pa)	\
353 			(&pv_list_locks[pa_index(pa) % NPV_LIST_LOCKS])
354 #endif
355 
356 #define	CHANGE_PV_LIST_LOCK_TO_PHYS(lockp, pa)	do {	\
357 	struct rwlock **_lockp = (lockp);		\
358 	struct rwlock *_new_lock;			\
359 							\
360 	_new_lock = PHYS_TO_PV_LIST_LOCK(pa);		\
361 	if (_new_lock != *_lockp) {			\
362 		if (*_lockp != NULL)			\
363 			rw_wunlock(*_lockp);		\
364 		*_lockp = _new_lock;			\
365 		rw_wlock(*_lockp);			\
366 	}						\
367 } while (0)
368 
369 #define	CHANGE_PV_LIST_LOCK_TO_VM_PAGE(lockp, m)	\
370 			CHANGE_PV_LIST_LOCK_TO_PHYS(lockp, VM_PAGE_TO_PHYS(m))
371 
372 #define	RELEASE_PV_LIST_LOCK(lockp)		do {	\
373 	struct rwlock **_lockp = (lockp);		\
374 							\
375 	if (*_lockp != NULL) {				\
376 		rw_wunlock(*_lockp);			\
377 		*_lockp = NULL;				\
378 	}						\
379 } while (0)
380 
381 #define	VM_PAGE_TO_PV_LIST_LOCK(m)	\
382 			PHYS_TO_PV_LIST_LOCK(VM_PAGE_TO_PHYS(m))
383 
384 struct pmap kernel_pmap_store;
385 
386 vm_offset_t virtual_avail;	/* VA of first avail page (after kernel bss) */
387 vm_offset_t virtual_end;	/* VA of last avail page (end of kernel AS) */
388 
389 int nkpt;
390 SYSCTL_INT(_machdep, OID_AUTO, nkpt, CTLFLAG_RD, &nkpt, 0,
391     "Number of kernel page table pages allocated on bootup");
392 
393 static int ndmpdp;
394 vm_paddr_t dmaplimit;
395 vm_offset_t kernel_vm_end = VM_MIN_KERNEL_ADDRESS;
396 pt_entry_t pg_nx;
397 
398 static SYSCTL_NODE(_vm, OID_AUTO, pmap, CTLFLAG_RD | CTLFLAG_MPSAFE, 0,
399     "VM/pmap parameters");
400 
401 static int pg_ps_enabled = 1;
402 SYSCTL_INT(_vm_pmap, OID_AUTO, pg_ps_enabled, CTLFLAG_RDTUN | CTLFLAG_NOFETCH,
403     &pg_ps_enabled, 0, "Are large page mappings enabled?");
404 
405 int __read_frequently la57 = 0;
406 SYSCTL_INT(_vm_pmap, OID_AUTO, la57, CTLFLAG_RDTUN | CTLFLAG_NOFETCH,
407     &la57, 0,
408     "5-level paging for host is enabled");
409 
410 static bool
pmap_is_la57(pmap_t pmap)411 pmap_is_la57(pmap_t pmap)
412 {
413 	if (pmap->pm_type == PT_X86)
414 		return (la57);
415 	return (false);		/* XXXKIB handle EPT */
416 }
417 
418 #define	PAT_INDEX_SIZE	8
419 static int pat_index[PAT_INDEX_SIZE];	/* cache mode to PAT index conversion */
420 
421 static u_int64_t	KPTphys;	/* phys addr of kernel level 1 */
422 static u_int64_t	KPDphys;	/* phys addr of kernel level 2 */
423 static u_int64_t	KPDPphys;	/* phys addr of kernel level 3 */
424 u_int64_t		KPML4phys;	/* phys addr of kernel level 4 */
425 u_int64_t		KPML5phys;	/* phys addr of kernel level 5,
426 					   if supported */
427 
428 #ifdef KASAN
429 static uint64_t		KASANPDPphys;
430 #endif
431 
432 static pml4_entry_t	*kernel_pml4;
433 static u_int64_t	DMPDphys;	/* phys addr of direct mapped level 2 */
434 static u_int64_t	DMPDPphys;	/* phys addr of direct mapped level 3 */
435 static int		ndmpdpphys;	/* number of DMPDPphys pages */
436 
437 vm_paddr_t		kernphys;	/* phys addr of start of bootstrap data */
438 vm_paddr_t		KERNend;	/* and the end */
439 
440 /*
441  * pmap_mapdev support pre initialization (i.e. console)
442  */
443 #define	PMAP_PREINIT_MAPPING_COUNT	8
444 static struct pmap_preinit_mapping {
445 	vm_paddr_t	pa;
446 	vm_offset_t	va;
447 	vm_size_t	sz;
448 	int		mode;
449 } pmap_preinit_mapping[PMAP_PREINIT_MAPPING_COUNT];
450 static int pmap_initialized;
451 
452 /*
453  * Data for the pv entry allocation mechanism.
454  * Updates to pv_invl_gen are protected by the pv list lock but reads are not.
455  */
456 #ifdef NUMA
457 static __inline int
pc_to_domain(struct pv_chunk * pc)458 pc_to_domain(struct pv_chunk *pc)
459 {
460 
461 	return (vm_phys_domain(DMAP_TO_PHYS((vm_offset_t)pc)));
462 }
463 #else
464 static __inline int
pc_to_domain(struct pv_chunk * pc __unused)465 pc_to_domain(struct pv_chunk *pc __unused)
466 {
467 
468 	return (0);
469 }
470 #endif
471 
472 struct pv_chunks_list {
473 	struct mtx pvc_lock;
474 	TAILQ_HEAD(pch, pv_chunk) pvc_list;
475 	int active_reclaims;
476 } __aligned(CACHE_LINE_SIZE);
477 
478 struct pv_chunks_list __exclusive_cache_line pv_chunks[PMAP_MEMDOM];
479 
480 #ifdef	NUMA
481 struct pmap_large_md_page {
482 	struct rwlock   pv_lock;
483 	struct md_page  pv_page;
484 	u_long pv_invl_gen;
485 };
486 __exclusive_cache_line static struct pmap_large_md_page pv_dummy_large;
487 #define pv_dummy pv_dummy_large.pv_page
488 __read_mostly static struct pmap_large_md_page *pv_table;
489 __read_mostly vm_paddr_t pmap_last_pa;
490 #else
491 static struct rwlock __exclusive_cache_line pv_list_locks[NPV_LIST_LOCKS];
492 static u_long pv_invl_gen[NPV_LIST_LOCKS];
493 static struct md_page *pv_table;
494 static struct md_page pv_dummy;
495 #endif
496 
497 /*
498  * All those kernel PT submaps that BSD is so fond of
499  */
500 pt_entry_t *CMAP1 = NULL;
501 caddr_t CADDR1 = 0;
502 static vm_offset_t qframe = 0;
503 static struct mtx qframe_mtx;
504 
505 static int pmap_flags = PMAP_PDE_SUPERPAGE;	/* flags for x86 pmaps */
506 
507 static vmem_t *large_vmem;
508 static u_int lm_ents;
509 #define	PMAP_ADDRESS_IN_LARGEMAP(va)	((va) >= LARGEMAP_MIN_ADDRESS && \
510 	(va) < LARGEMAP_MIN_ADDRESS + NBPML4 * (u_long)lm_ents)
511 
512 int pmap_pcid_enabled = 1;
513 SYSCTL_INT(_vm_pmap, OID_AUTO, pcid_enabled, CTLFLAG_RDTUN | CTLFLAG_NOFETCH,
514     &pmap_pcid_enabled, 0, "Is TLB Context ID enabled ?");
515 int invpcid_works = 0;
516 SYSCTL_INT(_vm_pmap, OID_AUTO, invpcid_works, CTLFLAG_RD, &invpcid_works, 0,
517     "Is the invpcid instruction available ?");
518 int pmap_pcid_invlpg_workaround = 0;
519 SYSCTL_INT(_vm_pmap, OID_AUTO, pcid_invlpg_workaround,
520     CTLFLAG_RDTUN | CTLFLAG_NOFETCH,
521     &pmap_pcid_invlpg_workaround, 0,
522     "Enable small core PCID/INVLPG workaround");
523 int pmap_pcid_invlpg_workaround_uena = 1;
524 
525 int __read_frequently pti = 0;
526 SYSCTL_INT(_vm_pmap, OID_AUTO, pti, CTLFLAG_RDTUN | CTLFLAG_NOFETCH,
527     &pti, 0,
528     "Page Table Isolation enabled");
529 static vm_object_t pti_obj;
530 static pml4_entry_t *pti_pml4;
531 static vm_pindex_t pti_pg_idx;
532 static bool pti_finalized;
533 
534 struct pmap_pkru_range {
535 	struct rs_el	pkru_rs_el;
536 	u_int		pkru_keyidx;
537 	int		pkru_flags;
538 };
539 
540 static uma_zone_t pmap_pkru_ranges_zone;
541 static bool pmap_pkru_same(pmap_t pmap, vm_offset_t sva, vm_offset_t eva);
542 static pt_entry_t pmap_pkru_get(pmap_t pmap, vm_offset_t va);
543 static void pmap_pkru_on_remove(pmap_t pmap, vm_offset_t sva, vm_offset_t eva);
544 static void *pkru_dup_range(void *ctx, void *data);
545 static void pkru_free_range(void *ctx, void *node);
546 static int pmap_pkru_copy(pmap_t dst_pmap, pmap_t src_pmap);
547 static int pmap_pkru_deassign(pmap_t pmap, vm_offset_t sva, vm_offset_t eva);
548 static void pmap_pkru_deassign_all(pmap_t pmap);
549 
550 static COUNTER_U64_DEFINE_EARLY(pcid_save_cnt);
551 SYSCTL_COUNTER_U64(_vm_pmap, OID_AUTO, pcid_save_cnt, CTLFLAG_RD,
552     &pcid_save_cnt, "Count of saved TLB context on switch");
553 
554 static LIST_HEAD(, pmap_invl_gen) pmap_invl_gen_tracker =
555     LIST_HEAD_INITIALIZER(&pmap_invl_gen_tracker);
556 static struct mtx invl_gen_mtx;
557 /* Fake lock object to satisfy turnstiles interface. */
558 static struct lock_object invl_gen_ts = {
559 	.lo_name = "invlts",
560 };
561 static struct pmap_invl_gen pmap_invl_gen_head = {
562 	.gen = 1,
563 	.next = NULL,
564 };
565 static u_long pmap_invl_gen = 1;
566 static int pmap_invl_waiters;
567 static struct callout pmap_invl_callout;
568 static bool pmap_invl_callout_inited;
569 
570 #define	PMAP_ASSERT_NOT_IN_DI() \
571     KASSERT(pmap_not_in_di(), ("DI already started"))
572 
573 static bool
pmap_di_locked(void)574 pmap_di_locked(void)
575 {
576 	int tun;
577 
578 	if ((cpu_feature2 & CPUID2_CX16) == 0)
579 		return (true);
580 	tun = 0;
581 	TUNABLE_INT_FETCH("vm.pmap.di_locked", &tun);
582 	return (tun != 0);
583 }
584 
585 static int
sysctl_pmap_di_locked(SYSCTL_HANDLER_ARGS)586 sysctl_pmap_di_locked(SYSCTL_HANDLER_ARGS)
587 {
588 	int locked;
589 
590 	locked = pmap_di_locked();
591 	return (sysctl_handle_int(oidp, &locked, 0, req));
592 }
593 SYSCTL_PROC(_vm_pmap, OID_AUTO, di_locked, CTLTYPE_INT | CTLFLAG_RDTUN |
594     CTLFLAG_MPSAFE, 0, 0, sysctl_pmap_di_locked, "",
595     "Locked delayed invalidation");
596 
597 static bool pmap_not_in_di_l(void);
598 static bool pmap_not_in_di_u(void);
599 DEFINE_IFUNC(, bool, pmap_not_in_di, (void))
600 {
601 
602 	return (pmap_di_locked() ? pmap_not_in_di_l : pmap_not_in_di_u);
603 }
604 
605 static bool
pmap_not_in_di_l(void)606 pmap_not_in_di_l(void)
607 {
608 	struct pmap_invl_gen *invl_gen;
609 
610 	invl_gen = &curthread->td_md.md_invl_gen;
611 	return (invl_gen->gen == 0);
612 }
613 
614 static void
pmap_thread_init_invl_gen_l(struct thread * td)615 pmap_thread_init_invl_gen_l(struct thread *td)
616 {
617 	struct pmap_invl_gen *invl_gen;
618 
619 	invl_gen = &td->td_md.md_invl_gen;
620 	invl_gen->gen = 0;
621 }
622 
623 static void
pmap_delayed_invl_wait_block(u_long * m_gen,u_long * invl_gen)624 pmap_delayed_invl_wait_block(u_long *m_gen, u_long *invl_gen)
625 {
626 	struct turnstile *ts;
627 
628 	ts = turnstile_trywait(&invl_gen_ts);
629 	if (*m_gen > atomic_load_long(invl_gen))
630 		turnstile_wait(ts, NULL, TS_SHARED_QUEUE);
631 	else
632 		turnstile_cancel(ts);
633 }
634 
635 static void
pmap_delayed_invl_finish_unblock(u_long new_gen)636 pmap_delayed_invl_finish_unblock(u_long new_gen)
637 {
638 	struct turnstile *ts;
639 
640 	turnstile_chain_lock(&invl_gen_ts);
641 	ts = turnstile_lookup(&invl_gen_ts);
642 	if (new_gen != 0)
643 		pmap_invl_gen = new_gen;
644 	if (ts != NULL) {
645 		turnstile_broadcast(ts, TS_SHARED_QUEUE);
646 		turnstile_unpend(ts);
647 	}
648 	turnstile_chain_unlock(&invl_gen_ts);
649 }
650 
651 /*
652  * Start a new Delayed Invalidation (DI) block of code, executed by
653  * the current thread.  Within a DI block, the current thread may
654  * destroy both the page table and PV list entries for a mapping and
655  * then release the corresponding PV list lock before ensuring that
656  * the mapping is flushed from the TLBs of any processors with the
657  * pmap active.
658  */
659 static void
pmap_delayed_invl_start_l(void)660 pmap_delayed_invl_start_l(void)
661 {
662 	struct pmap_invl_gen *invl_gen;
663 	u_long currgen;
664 
665 	invl_gen = &curthread->td_md.md_invl_gen;
666 	PMAP_ASSERT_NOT_IN_DI();
667 	mtx_lock(&invl_gen_mtx);
668 	if (LIST_EMPTY(&pmap_invl_gen_tracker))
669 		currgen = pmap_invl_gen;
670 	else
671 		currgen = LIST_FIRST(&pmap_invl_gen_tracker)->gen;
672 	invl_gen->gen = currgen + 1;
673 	LIST_INSERT_HEAD(&pmap_invl_gen_tracker, invl_gen, link);
674 	mtx_unlock(&invl_gen_mtx);
675 }
676 
677 /*
678  * Finish the DI block, previously started by the current thread.  All
679  * required TLB flushes for the pages marked by
680  * pmap_delayed_invl_page() must be finished before this function is
681  * called.
682  *
683  * This function works by bumping the global DI generation number to
684  * the generation number of the current thread's DI, unless there is a
685  * pending DI that started earlier.  In the latter case, bumping the
686  * global DI generation number would incorrectly signal that the
687  * earlier DI had finished.  Instead, this function bumps the earlier
688  * DI's generation number to match the generation number of the
689  * current thread's DI.
690  */
691 static void
pmap_delayed_invl_finish_l(void)692 pmap_delayed_invl_finish_l(void)
693 {
694 	struct pmap_invl_gen *invl_gen, *next;
695 
696 	invl_gen = &curthread->td_md.md_invl_gen;
697 	KASSERT(invl_gen->gen != 0, ("missed invl_start"));
698 	mtx_lock(&invl_gen_mtx);
699 	next = LIST_NEXT(invl_gen, link);
700 	if (next == NULL)
701 		pmap_delayed_invl_finish_unblock(invl_gen->gen);
702 	else
703 		next->gen = invl_gen->gen;
704 	LIST_REMOVE(invl_gen, link);
705 	mtx_unlock(&invl_gen_mtx);
706 	invl_gen->gen = 0;
707 }
708 
709 static bool
pmap_not_in_di_u(void)710 pmap_not_in_di_u(void)
711 {
712 	struct pmap_invl_gen *invl_gen;
713 
714 	invl_gen = &curthread->td_md.md_invl_gen;
715 	return (((uintptr_t)invl_gen->next & PMAP_INVL_GEN_NEXT_INVALID) != 0);
716 }
717 
718 static void
pmap_thread_init_invl_gen_u(struct thread * td)719 pmap_thread_init_invl_gen_u(struct thread *td)
720 {
721 	struct pmap_invl_gen *invl_gen;
722 
723 	invl_gen = &td->td_md.md_invl_gen;
724 	invl_gen->gen = 0;
725 	invl_gen->next = (void *)PMAP_INVL_GEN_NEXT_INVALID;
726 }
727 
728 static bool
pmap_di_load_invl(struct pmap_invl_gen * ptr,struct pmap_invl_gen * out)729 pmap_di_load_invl(struct pmap_invl_gen *ptr, struct pmap_invl_gen *out)
730 {
731 	uint64_t new_high, new_low, old_high, old_low;
732 	char res;
733 
734 	old_low = new_low = 0;
735 	old_high = new_high = (uintptr_t)0;
736 
737 	__asm volatile("lock;cmpxchg16b\t%1"
738 	    : "=@cce" (res), "+m" (*ptr), "+a" (old_low), "+d" (old_high)
739 	    : "b"(new_low), "c" (new_high)
740 	    : "memory", "cc");
741 	if (res == 0) {
742 		if ((old_high & PMAP_INVL_GEN_NEXT_INVALID) != 0)
743 			return (false);
744 		out->gen = old_low;
745 		out->next = (void *)old_high;
746 	} else {
747 		out->gen = new_low;
748 		out->next = (void *)new_high;
749 	}
750 	return (true);
751 }
752 
753 static bool
pmap_di_store_invl(struct pmap_invl_gen * ptr,struct pmap_invl_gen * old_val,struct pmap_invl_gen * new_val)754 pmap_di_store_invl(struct pmap_invl_gen *ptr, struct pmap_invl_gen *old_val,
755     struct pmap_invl_gen *new_val)
756 {
757 	uint64_t new_high, new_low, old_high, old_low;
758 	char res;
759 
760 	new_low = new_val->gen;
761 	new_high = (uintptr_t)new_val->next;
762 	old_low = old_val->gen;
763 	old_high = (uintptr_t)old_val->next;
764 
765 	__asm volatile("lock;cmpxchg16b\t%1"
766 	    : "=@cce" (res), "+m" (*ptr), "+a" (old_low), "+d" (old_high)
767 	    : "b"(new_low), "c" (new_high)
768 	    : "memory", "cc");
769 	return (res);
770 }
771 
772 static COUNTER_U64_DEFINE_EARLY(pv_page_count);
773 SYSCTL_COUNTER_U64(_vm_pmap, OID_AUTO, pv_page_count, CTLFLAG_RD,
774     &pv_page_count, "Current number of allocated pv pages");
775 
776 static COUNTER_U64_DEFINE_EARLY(user_pt_page_count);
777 SYSCTL_COUNTER_U64(_vm_pmap, OID_AUTO, user_pt_page_count, CTLFLAG_RD,
778     &user_pt_page_count,
779     "Current number of allocated page table pages for userspace");
780 
781 static COUNTER_U64_DEFINE_EARLY(kernel_pt_page_count);
782 SYSCTL_COUNTER_U64(_vm_pmap, OID_AUTO, kernel_pt_page_count, CTLFLAG_RD,
783     &kernel_pt_page_count,
784     "Current number of allocated page table pages for the kernel");
785 
786 #ifdef PV_STATS
787 
788 static COUNTER_U64_DEFINE_EARLY(invl_start_restart);
789 SYSCTL_COUNTER_U64(_vm_pmap, OID_AUTO, invl_start_restart,
790     CTLFLAG_RD, &invl_start_restart,
791     "Number of delayed TLB invalidation request restarts");
792 
793 static COUNTER_U64_DEFINE_EARLY(invl_finish_restart);
794 SYSCTL_COUNTER_U64(_vm_pmap, OID_AUTO, invl_finish_restart, CTLFLAG_RD,
795     &invl_finish_restart,
796     "Number of delayed TLB invalidation completion restarts");
797 
798 static int invl_max_qlen;
799 SYSCTL_INT(_vm_pmap, OID_AUTO, invl_max_qlen, CTLFLAG_RD,
800     &invl_max_qlen, 0,
801     "Maximum delayed TLB invalidation request queue length");
802 #endif
803 
804 #define di_delay	locks_delay
805 
806 static void
pmap_delayed_invl_start_u(void)807 pmap_delayed_invl_start_u(void)
808 {
809 	struct pmap_invl_gen *invl_gen, *p, prev, new_prev;
810 	struct thread *td;
811 	struct lock_delay_arg lda;
812 	uintptr_t prevl;
813 	u_char pri;
814 #ifdef PV_STATS
815 	int i, ii;
816 #endif
817 
818 	td = curthread;
819 	invl_gen = &td->td_md.md_invl_gen;
820 	PMAP_ASSERT_NOT_IN_DI();
821 	lock_delay_arg_init(&lda, &di_delay);
822 	invl_gen->saved_pri = 0;
823 	pri = td->td_base_pri;
824 	if (pri > PVM) {
825 		thread_lock(td);
826 		pri = td->td_base_pri;
827 		if (pri > PVM) {
828 			invl_gen->saved_pri = pri;
829 			sched_prio(td, PVM);
830 		}
831 		thread_unlock(td);
832 	}
833 again:
834 	PV_STAT(i = 0);
835 	for (p = &pmap_invl_gen_head;; p = prev.next) {
836 		PV_STAT(i++);
837 		prevl = (uintptr_t)atomic_load_ptr(&p->next);
838 		if ((prevl & PMAP_INVL_GEN_NEXT_INVALID) != 0) {
839 			PV_STAT(counter_u64_add(invl_start_restart, 1));
840 			lock_delay(&lda);
841 			goto again;
842 		}
843 		if (prevl == 0)
844 			break;
845 		prev.next = (void *)prevl;
846 	}
847 #ifdef PV_STATS
848 	if ((ii = invl_max_qlen) < i)
849 		atomic_cmpset_int(&invl_max_qlen, ii, i);
850 #endif
851 
852 	if (!pmap_di_load_invl(p, &prev) || prev.next != NULL) {
853 		PV_STAT(counter_u64_add(invl_start_restart, 1));
854 		lock_delay(&lda);
855 		goto again;
856 	}
857 
858 	new_prev.gen = prev.gen;
859 	new_prev.next = invl_gen;
860 	invl_gen->gen = prev.gen + 1;
861 
862 	/* Formal fence between store to invl->gen and updating *p. */
863 	atomic_thread_fence_rel();
864 
865 	/*
866 	 * After inserting an invl_gen element with invalid bit set,
867 	 * this thread blocks any other thread trying to enter the
868 	 * delayed invalidation block.  Do not allow to remove us from
869 	 * the CPU, because it causes starvation for other threads.
870 	 */
871 	critical_enter();
872 
873 	/*
874 	 * ABA for *p is not possible there, since p->gen can only
875 	 * increase.  So if the *p thread finished its di, then
876 	 * started a new one and got inserted into the list at the
877 	 * same place, its gen will appear greater than the previously
878 	 * read gen.
879 	 */
880 	if (!pmap_di_store_invl(p, &prev, &new_prev)) {
881 		critical_exit();
882 		PV_STAT(counter_u64_add(invl_start_restart, 1));
883 		lock_delay(&lda);
884 		goto again;
885 	}
886 
887 	/*
888 	 * There we clear PMAP_INVL_GEN_NEXT_INVALID in
889 	 * invl_gen->next, allowing other threads to iterate past us.
890 	 * pmap_di_store_invl() provides fence between the generation
891 	 * write and the update of next.
892 	 */
893 	invl_gen->next = NULL;
894 	critical_exit();
895 }
896 
897 static bool
pmap_delayed_invl_finish_u_crit(struct pmap_invl_gen * invl_gen,struct pmap_invl_gen * p)898 pmap_delayed_invl_finish_u_crit(struct pmap_invl_gen *invl_gen,
899     struct pmap_invl_gen *p)
900 {
901 	struct pmap_invl_gen prev, new_prev;
902 	u_long mygen;
903 
904 	/*
905 	 * Load invl_gen->gen after setting invl_gen->next
906 	 * PMAP_INVL_GEN_NEXT_INVALID.  This prevents larger
907 	 * generations to propagate to our invl_gen->gen.  Lock prefix
908 	 * in atomic_set_ptr() worked as seq_cst fence.
909 	 */
910 	mygen = atomic_load_long(&invl_gen->gen);
911 
912 	if (!pmap_di_load_invl(p, &prev) || prev.next != invl_gen)
913 		return (false);
914 
915 	KASSERT(prev.gen < mygen,
916 	    ("invalid di gen sequence %lu %lu", prev.gen, mygen));
917 	new_prev.gen = mygen;
918 	new_prev.next = (void *)((uintptr_t)invl_gen->next &
919 	    ~PMAP_INVL_GEN_NEXT_INVALID);
920 
921 	/* Formal fence between load of prev and storing update to it. */
922 	atomic_thread_fence_rel();
923 
924 	return (pmap_di_store_invl(p, &prev, &new_prev));
925 }
926 
927 static void
pmap_delayed_invl_finish_u(void)928 pmap_delayed_invl_finish_u(void)
929 {
930 	struct pmap_invl_gen *invl_gen, *p;
931 	struct thread *td;
932 	struct lock_delay_arg lda;
933 	uintptr_t prevl;
934 
935 	td = curthread;
936 	invl_gen = &td->td_md.md_invl_gen;
937 	KASSERT(invl_gen->gen != 0, ("missed invl_start: gen 0"));
938 	KASSERT(((uintptr_t)invl_gen->next & PMAP_INVL_GEN_NEXT_INVALID) == 0,
939 	    ("missed invl_start: INVALID"));
940 	lock_delay_arg_init(&lda, &di_delay);
941 
942 again:
943 	for (p = &pmap_invl_gen_head; p != NULL; p = (void *)prevl) {
944 		prevl = (uintptr_t)atomic_load_ptr(&p->next);
945 		if ((prevl & PMAP_INVL_GEN_NEXT_INVALID) != 0) {
946 			PV_STAT(counter_u64_add(invl_finish_restart, 1));
947 			lock_delay(&lda);
948 			goto again;
949 		}
950 		if ((void *)prevl == invl_gen)
951 			break;
952 	}
953 
954 	/*
955 	 * It is legitimate to not find ourself on the list if a
956 	 * thread before us finished its DI and started it again.
957 	 */
958 	if (__predict_false(p == NULL)) {
959 		PV_STAT(counter_u64_add(invl_finish_restart, 1));
960 		lock_delay(&lda);
961 		goto again;
962 	}
963 
964 	critical_enter();
965 	atomic_set_ptr((uintptr_t *)&invl_gen->next,
966 	    PMAP_INVL_GEN_NEXT_INVALID);
967 	if (!pmap_delayed_invl_finish_u_crit(invl_gen, p)) {
968 		atomic_clear_ptr((uintptr_t *)&invl_gen->next,
969 		    PMAP_INVL_GEN_NEXT_INVALID);
970 		critical_exit();
971 		PV_STAT(counter_u64_add(invl_finish_restart, 1));
972 		lock_delay(&lda);
973 		goto again;
974 	}
975 	critical_exit();
976 	if (atomic_load_int(&pmap_invl_waiters) > 0)
977 		pmap_delayed_invl_finish_unblock(0);
978 	if (invl_gen->saved_pri != 0) {
979 		thread_lock(td);
980 		sched_prio(td, invl_gen->saved_pri);
981 		thread_unlock(td);
982 	}
983 }
984 
985 #ifdef DDB
DB_SHOW_COMMAND(di_queue,pmap_di_queue)986 DB_SHOW_COMMAND(di_queue, pmap_di_queue)
987 {
988 	struct pmap_invl_gen *p, *pn;
989 	struct thread *td;
990 	uintptr_t nextl;
991 	bool first;
992 
993 	for (p = &pmap_invl_gen_head, first = true; p != NULL; p = pn,
994 	    first = false) {
995 		nextl = (uintptr_t)atomic_load_ptr(&p->next);
996 		pn = (void *)(nextl & ~PMAP_INVL_GEN_NEXT_INVALID);
997 		td = first ? NULL : __containerof(p, struct thread,
998 		    td_md.md_invl_gen);
999 		db_printf("gen %lu inv %d td %p tid %d\n", p->gen,
1000 		    (nextl & PMAP_INVL_GEN_NEXT_INVALID) != 0, td,
1001 		    td != NULL ? td->td_tid : -1);
1002 	}
1003 }
1004 #endif
1005 
1006 #ifdef PV_STATS
1007 static COUNTER_U64_DEFINE_EARLY(invl_wait);
1008 SYSCTL_COUNTER_U64(_vm_pmap, OID_AUTO, invl_wait,
1009     CTLFLAG_RD, &invl_wait,
1010     "Number of times DI invalidation blocked pmap_remove_all/write");
1011 
1012 static COUNTER_U64_DEFINE_EARLY(invl_wait_slow);
1013 SYSCTL_COUNTER_U64(_vm_pmap, OID_AUTO, invl_wait_slow, CTLFLAG_RD,
1014      &invl_wait_slow, "Number of slow invalidation waits for lockless DI");
1015 
1016 #endif
1017 
1018 #ifdef NUMA
1019 static u_long *
pmap_delayed_invl_genp(vm_page_t m)1020 pmap_delayed_invl_genp(vm_page_t m)
1021 {
1022 	vm_paddr_t pa;
1023 	u_long *gen;
1024 
1025 	pa = VM_PAGE_TO_PHYS(m);
1026 	if (__predict_false((pa) > pmap_last_pa))
1027 		gen = &pv_dummy_large.pv_invl_gen;
1028 	else
1029 		gen = &(pa_to_pmdp(pa)->pv_invl_gen);
1030 
1031 	return (gen);
1032 }
1033 #else
1034 static u_long *
pmap_delayed_invl_genp(vm_page_t m)1035 pmap_delayed_invl_genp(vm_page_t m)
1036 {
1037 
1038 	return (&pv_invl_gen[pa_index(VM_PAGE_TO_PHYS(m)) % NPV_LIST_LOCKS]);
1039 }
1040 #endif
1041 
1042 static void
pmap_delayed_invl_callout_func(void * arg __unused)1043 pmap_delayed_invl_callout_func(void *arg __unused)
1044 {
1045 
1046 	if (atomic_load_int(&pmap_invl_waiters) == 0)
1047 		return;
1048 	pmap_delayed_invl_finish_unblock(0);
1049 }
1050 
1051 static void
pmap_delayed_invl_callout_init(void * arg __unused)1052 pmap_delayed_invl_callout_init(void *arg __unused)
1053 {
1054 
1055 	if (pmap_di_locked())
1056 		return;
1057 	callout_init(&pmap_invl_callout, 1);
1058 	pmap_invl_callout_inited = true;
1059 }
1060 SYSINIT(pmap_di_callout, SI_SUB_CPU + 1, SI_ORDER_ANY,
1061     pmap_delayed_invl_callout_init, NULL);
1062 
1063 /*
1064  * Ensure that all currently executing DI blocks, that need to flush
1065  * TLB for the given page m, actually flushed the TLB at the time the
1066  * function returned.  If the page m has an empty PV list and we call
1067  * pmap_delayed_invl_wait(), upon its return we know that no CPU has a
1068  * valid mapping for the page m in either its page table or TLB.
1069  *
1070  * This function works by blocking until the global DI generation
1071  * number catches up with the generation number associated with the
1072  * given page m and its PV list.  Since this function's callers
1073  * typically own an object lock and sometimes own a page lock, it
1074  * cannot sleep.  Instead, it blocks on a turnstile to relinquish the
1075  * processor.
1076  */
1077 static void
pmap_delayed_invl_wait_l(vm_page_t m)1078 pmap_delayed_invl_wait_l(vm_page_t m)
1079 {
1080 	u_long *m_gen;
1081 #ifdef PV_STATS
1082 	bool accounted = false;
1083 #endif
1084 
1085 	m_gen = pmap_delayed_invl_genp(m);
1086 	while (*m_gen > pmap_invl_gen) {
1087 #ifdef PV_STATS
1088 		if (!accounted) {
1089 			counter_u64_add(invl_wait, 1);
1090 			accounted = true;
1091 		}
1092 #endif
1093 		pmap_delayed_invl_wait_block(m_gen, &pmap_invl_gen);
1094 	}
1095 }
1096 
1097 static void
pmap_delayed_invl_wait_u(vm_page_t m)1098 pmap_delayed_invl_wait_u(vm_page_t m)
1099 {
1100 	u_long *m_gen;
1101 	struct lock_delay_arg lda;
1102 	bool fast;
1103 
1104 	fast = true;
1105 	m_gen = pmap_delayed_invl_genp(m);
1106 	lock_delay_arg_init(&lda, &di_delay);
1107 	while (*m_gen > atomic_load_long(&pmap_invl_gen_head.gen)) {
1108 		if (fast || !pmap_invl_callout_inited) {
1109 			PV_STAT(counter_u64_add(invl_wait, 1));
1110 			lock_delay(&lda);
1111 			fast = false;
1112 		} else {
1113 			/*
1114 			 * The page's invalidation generation number
1115 			 * is still below the current thread's number.
1116 			 * Prepare to block so that we do not waste
1117 			 * CPU cycles or worse, suffer livelock.
1118 			 *
1119 			 * Since it is impossible to block without
1120 			 * racing with pmap_delayed_invl_finish_u(),
1121 			 * prepare for the race by incrementing
1122 			 * pmap_invl_waiters and arming a 1-tick
1123 			 * callout which will unblock us if we lose
1124 			 * the race.
1125 			 */
1126 			atomic_add_int(&pmap_invl_waiters, 1);
1127 
1128 			/*
1129 			 * Re-check the current thread's invalidation
1130 			 * generation after incrementing
1131 			 * pmap_invl_waiters, so that there is no race
1132 			 * with pmap_delayed_invl_finish_u() setting
1133 			 * the page generation and checking
1134 			 * pmap_invl_waiters.  The only race allowed
1135 			 * is for a missed unblock, which is handled
1136 			 * by the callout.
1137 			 */
1138 			if (*m_gen >
1139 			    atomic_load_long(&pmap_invl_gen_head.gen)) {
1140 				callout_reset(&pmap_invl_callout, 1,
1141 				    pmap_delayed_invl_callout_func, NULL);
1142 				PV_STAT(counter_u64_add(invl_wait_slow, 1));
1143 				pmap_delayed_invl_wait_block(m_gen,
1144 				    &pmap_invl_gen_head.gen);
1145 			}
1146 			atomic_add_int(&pmap_invl_waiters, -1);
1147 		}
1148 	}
1149 }
1150 
1151 DEFINE_IFUNC(, void, pmap_thread_init_invl_gen, (struct thread *))
1152 {
1153 
1154 	return (pmap_di_locked() ? pmap_thread_init_invl_gen_l :
1155 	    pmap_thread_init_invl_gen_u);
1156 }
1157 
1158 DEFINE_IFUNC(static, void, pmap_delayed_invl_start, (void))
1159 {
1160 
1161 	return (pmap_di_locked() ? pmap_delayed_invl_start_l :
1162 	    pmap_delayed_invl_start_u);
1163 }
1164 
1165 DEFINE_IFUNC(static, void, pmap_delayed_invl_finish, (void))
1166 {
1167 
1168 	return (pmap_di_locked() ? pmap_delayed_invl_finish_l :
1169 	    pmap_delayed_invl_finish_u);
1170 }
1171 
1172 DEFINE_IFUNC(static, void, pmap_delayed_invl_wait, (vm_page_t))
1173 {
1174 
1175 	return (pmap_di_locked() ? pmap_delayed_invl_wait_l :
1176 	    pmap_delayed_invl_wait_u);
1177 }
1178 
1179 /*
1180  * Mark the page m's PV list as participating in the current thread's
1181  * DI block.  Any threads concurrently using m's PV list to remove or
1182  * restrict all mappings to m will wait for the current thread's DI
1183  * block to complete before proceeding.
1184  *
1185  * The function works by setting the DI generation number for m's PV
1186  * list to at least the DI generation number of the current thread.
1187  * This forces a caller of pmap_delayed_invl_wait() to block until
1188  * current thread calls pmap_delayed_invl_finish().
1189  */
1190 static void
pmap_delayed_invl_page(vm_page_t m)1191 pmap_delayed_invl_page(vm_page_t m)
1192 {
1193 	u_long gen, *m_gen;
1194 
1195 	rw_assert(VM_PAGE_TO_PV_LIST_LOCK(m), RA_WLOCKED);
1196 	gen = curthread->td_md.md_invl_gen.gen;
1197 	if (gen == 0)
1198 		return;
1199 	m_gen = pmap_delayed_invl_genp(m);
1200 	if (*m_gen < gen)
1201 		*m_gen = gen;
1202 }
1203 
1204 /*
1205  * Crashdump maps.
1206  */
1207 static caddr_t crashdumpmap;
1208 
1209 /*
1210  * Internal flags for pmap_enter()'s helper functions.
1211  */
1212 #define	PMAP_ENTER_NORECLAIM	0x1000000	/* Don't reclaim PV entries. */
1213 #define	PMAP_ENTER_NOREPLACE	0x2000000	/* Don't replace mappings. */
1214 
1215 /*
1216  * Internal flags for pmap_mapdev_internal() and
1217  * pmap_change_props_locked().
1218  */
1219 #define	MAPDEV_FLUSHCACHE	0x00000001	/* Flush cache after mapping. */
1220 #define	MAPDEV_SETATTR		0x00000002	/* Modify existing attrs. */
1221 #define	MAPDEV_ASSERTVALID	0x00000004	/* Assert mapping validity. */
1222 
1223 TAILQ_HEAD(pv_chunklist, pv_chunk);
1224 
1225 static void	free_pv_chunk(struct pv_chunk *pc);
1226 static void	free_pv_chunk_batch(struct pv_chunklist *batch);
1227 static void	free_pv_entry(pmap_t pmap, pv_entry_t pv);
1228 static pv_entry_t get_pv_entry(pmap_t pmap, struct rwlock **lockp);
1229 static int	popcnt_pc_map_pq(uint64_t *map);
1230 static vm_page_t reclaim_pv_chunk(pmap_t locked_pmap, struct rwlock **lockp);
1231 static void	reserve_pv_entries(pmap_t pmap, int needed,
1232 		    struct rwlock **lockp);
1233 static void	pmap_pv_demote_pde(pmap_t pmap, vm_offset_t va, vm_paddr_t pa,
1234 		    struct rwlock **lockp);
1235 static bool	pmap_pv_insert_pde(pmap_t pmap, vm_offset_t va, pd_entry_t pde,
1236 		    u_int flags, struct rwlock **lockp);
1237 #if VM_NRESERVLEVEL > 0
1238 static void	pmap_pv_promote_pde(pmap_t pmap, vm_offset_t va, vm_paddr_t pa,
1239 		    struct rwlock **lockp);
1240 #endif
1241 static void	pmap_pvh_free(struct md_page *pvh, pmap_t pmap, vm_offset_t va);
1242 static pv_entry_t pmap_pvh_remove(struct md_page *pvh, pmap_t pmap,
1243 		    vm_offset_t va);
1244 
1245 static void	pmap_abort_ptp(pmap_t pmap, vm_offset_t va, vm_page_t mpte);
1246 static int pmap_change_props_locked(vm_offset_t va, vm_size_t size,
1247     vm_prot_t prot, int mode, int flags);
1248 static boolean_t pmap_demote_pde(pmap_t pmap, pd_entry_t *pde, vm_offset_t va);
1249 static boolean_t pmap_demote_pde_locked(pmap_t pmap, pd_entry_t *pde,
1250     vm_offset_t va, struct rwlock **lockp);
1251 static boolean_t pmap_demote_pdpe(pmap_t pmap, pdp_entry_t *pdpe,
1252     vm_offset_t va);
1253 static bool	pmap_enter_2mpage(pmap_t pmap, vm_offset_t va, vm_page_t m,
1254 		    vm_prot_t prot, struct rwlock **lockp);
1255 static int	pmap_enter_pde(pmap_t pmap, vm_offset_t va, pd_entry_t newpde,
1256 		    u_int flags, vm_page_t m, struct rwlock **lockp);
1257 static vm_page_t pmap_enter_quick_locked(pmap_t pmap, vm_offset_t va,
1258     vm_page_t m, vm_prot_t prot, vm_page_t mpte, struct rwlock **lockp);
1259 static void pmap_fill_ptp(pt_entry_t *firstpte, pt_entry_t newpte);
1260 static int pmap_insert_pt_page(pmap_t pmap, vm_page_t mpte, bool promoted);
1261 static void pmap_invalidate_cache_range_selfsnoop(vm_offset_t sva,
1262     vm_offset_t eva);
1263 static void pmap_invalidate_cache_range_all(vm_offset_t sva,
1264     vm_offset_t eva);
1265 static void pmap_invalidate_pde_page(pmap_t pmap, vm_offset_t va,
1266 		    pd_entry_t pde);
1267 static void pmap_kenter_attr(vm_offset_t va, vm_paddr_t pa, int mode);
1268 static vm_page_t pmap_large_map_getptp_unlocked(void);
1269 static vm_paddr_t pmap_large_map_kextract(vm_offset_t va);
1270 #if VM_NRESERVLEVEL > 0
1271 static void pmap_promote_pde(pmap_t pmap, pd_entry_t *pde, vm_offset_t va,
1272     struct rwlock **lockp);
1273 #endif
1274 static boolean_t pmap_protect_pde(pmap_t pmap, pd_entry_t *pde, vm_offset_t sva,
1275     vm_prot_t prot);
1276 static void pmap_pte_props(pt_entry_t *pte, u_long bits, u_long mask);
1277 static void pmap_pti_add_kva_locked(vm_offset_t sva, vm_offset_t eva,
1278     bool exec);
1279 static pdp_entry_t *pmap_pti_pdpe(vm_offset_t va);
1280 static pd_entry_t *pmap_pti_pde(vm_offset_t va);
1281 static void pmap_pti_wire_pte(void *pte);
1282 static int pmap_remove_pde(pmap_t pmap, pd_entry_t *pdq, vm_offset_t sva,
1283     struct spglist *free, struct rwlock **lockp);
1284 static int pmap_remove_pte(pmap_t pmap, pt_entry_t *ptq, vm_offset_t sva,
1285     pd_entry_t ptepde, struct spglist *free, struct rwlock **lockp);
1286 static vm_page_t pmap_remove_pt_page(pmap_t pmap, vm_offset_t va);
1287 static void pmap_remove_page(pmap_t pmap, vm_offset_t va, pd_entry_t *pde,
1288     struct spglist *free);
1289 static bool	pmap_remove_ptes(pmap_t pmap, vm_offset_t sva, vm_offset_t eva,
1290 		    pd_entry_t *pde, struct spglist *free,
1291 		    struct rwlock **lockp);
1292 static boolean_t pmap_try_insert_pv_entry(pmap_t pmap, vm_offset_t va,
1293     vm_page_t m, struct rwlock **lockp);
1294 static void pmap_update_pde(pmap_t pmap, vm_offset_t va, pd_entry_t *pde,
1295     pd_entry_t newpde);
1296 static void pmap_update_pde_invalidate(pmap_t, vm_offset_t va, pd_entry_t pde);
1297 
1298 static pd_entry_t *pmap_alloc_pde(pmap_t pmap, vm_offset_t va, vm_page_t *pdpgp,
1299 		struct rwlock **lockp);
1300 static vm_page_t pmap_allocpte_alloc(pmap_t pmap, vm_pindex_t ptepindex,
1301 		struct rwlock **lockp, vm_offset_t va);
1302 static vm_page_t pmap_allocpte_nosleep(pmap_t pmap, vm_pindex_t ptepindex,
1303 		struct rwlock **lockp, vm_offset_t va);
1304 static vm_page_t pmap_allocpte(pmap_t pmap, vm_offset_t va,
1305 		struct rwlock **lockp);
1306 
1307 static void _pmap_unwire_ptp(pmap_t pmap, vm_offset_t va, vm_page_t m,
1308     struct spglist *free);
1309 static int pmap_unuse_pt(pmap_t, vm_offset_t, pd_entry_t, struct spglist *);
1310 
1311 static vm_page_t pmap_alloc_pt_page(pmap_t, vm_pindex_t, int);
1312 static void pmap_free_pt_page(pmap_t, vm_page_t, bool);
1313 
1314 /********************/
1315 /* Inline functions */
1316 /********************/
1317 
1318 /*
1319  * Return a non-clipped indexes for a given VA, which are page table
1320  * pages indexes at the corresponding level.
1321  */
1322 static __inline vm_pindex_t
pmap_pde_pindex(vm_offset_t va)1323 pmap_pde_pindex(vm_offset_t va)
1324 {
1325 	return (va >> PDRSHIFT);
1326 }
1327 
1328 static __inline vm_pindex_t
pmap_pdpe_pindex(vm_offset_t va)1329 pmap_pdpe_pindex(vm_offset_t va)
1330 {
1331 	return (NUPDE + (va >> PDPSHIFT));
1332 }
1333 
1334 static __inline vm_pindex_t
pmap_pml4e_pindex(vm_offset_t va)1335 pmap_pml4e_pindex(vm_offset_t va)
1336 {
1337 	return (NUPDE + NUPDPE + (va >> PML4SHIFT));
1338 }
1339 
1340 static __inline vm_pindex_t
pmap_pml5e_pindex(vm_offset_t va)1341 pmap_pml5e_pindex(vm_offset_t va)
1342 {
1343 	return (NUPDE + NUPDPE + NUPML4E + (va >> PML5SHIFT));
1344 }
1345 
1346 static __inline pml4_entry_t *
pmap_pml5e(pmap_t pmap,vm_offset_t va)1347 pmap_pml5e(pmap_t pmap, vm_offset_t va)
1348 {
1349 
1350 	MPASS(pmap_is_la57(pmap));
1351 	return (&pmap->pm_pmltop[pmap_pml5e_index(va)]);
1352 }
1353 
1354 static __inline pml4_entry_t *
pmap_pml5e_u(pmap_t pmap,vm_offset_t va)1355 pmap_pml5e_u(pmap_t pmap, vm_offset_t va)
1356 {
1357 
1358 	MPASS(pmap_is_la57(pmap));
1359 	return (&pmap->pm_pmltopu[pmap_pml5e_index(va)]);
1360 }
1361 
1362 static __inline pml4_entry_t *
pmap_pml5e_to_pml4e(pml5_entry_t * pml5e,vm_offset_t va)1363 pmap_pml5e_to_pml4e(pml5_entry_t *pml5e, vm_offset_t va)
1364 {
1365 	pml4_entry_t *pml4e;
1366 
1367 	/* XXX MPASS(pmap_is_la57(pmap); */
1368 	pml4e = (pml4_entry_t *)PHYS_TO_DMAP(*pml5e & PG_FRAME);
1369 	return (&pml4e[pmap_pml4e_index(va)]);
1370 }
1371 
1372 /* Return a pointer to the PML4 slot that corresponds to a VA */
1373 static __inline pml4_entry_t *
pmap_pml4e(pmap_t pmap,vm_offset_t va)1374 pmap_pml4e(pmap_t pmap, vm_offset_t va)
1375 {
1376 	pml5_entry_t *pml5e;
1377 	pml4_entry_t *pml4e;
1378 	pt_entry_t PG_V;
1379 
1380 	if (pmap_is_la57(pmap)) {
1381 		pml5e = pmap_pml5e(pmap, va);
1382 		PG_V = pmap_valid_bit(pmap);
1383 		if ((*pml5e & PG_V) == 0)
1384 			return (NULL);
1385 		pml4e = (pml4_entry_t *)PHYS_TO_DMAP(*pml5e & PG_FRAME);
1386 	} else {
1387 		pml4e = pmap->pm_pmltop;
1388 	}
1389 	return (&pml4e[pmap_pml4e_index(va)]);
1390 }
1391 
1392 static __inline pml4_entry_t *
pmap_pml4e_u(pmap_t pmap,vm_offset_t va)1393 pmap_pml4e_u(pmap_t pmap, vm_offset_t va)
1394 {
1395 	MPASS(!pmap_is_la57(pmap));
1396 	return (&pmap->pm_pmltopu[pmap_pml4e_index(va)]);
1397 }
1398 
1399 /* Return a pointer to the PDP slot that corresponds to a VA */
1400 static __inline pdp_entry_t *
pmap_pml4e_to_pdpe(pml4_entry_t * pml4e,vm_offset_t va)1401 pmap_pml4e_to_pdpe(pml4_entry_t *pml4e, vm_offset_t va)
1402 {
1403 	pdp_entry_t *pdpe;
1404 
1405 	pdpe = (pdp_entry_t *)PHYS_TO_DMAP(*pml4e & PG_FRAME);
1406 	return (&pdpe[pmap_pdpe_index(va)]);
1407 }
1408 
1409 /* Return a pointer to the PDP slot that corresponds to a VA */
1410 static __inline pdp_entry_t *
pmap_pdpe(pmap_t pmap,vm_offset_t va)1411 pmap_pdpe(pmap_t pmap, vm_offset_t va)
1412 {
1413 	pml4_entry_t *pml4e;
1414 	pt_entry_t PG_V;
1415 
1416 	PG_V = pmap_valid_bit(pmap);
1417 	pml4e = pmap_pml4e(pmap, va);
1418 	if (pml4e == NULL || (*pml4e & PG_V) == 0)
1419 		return (NULL);
1420 	return (pmap_pml4e_to_pdpe(pml4e, va));
1421 }
1422 
1423 /* Return a pointer to the PD slot that corresponds to a VA */
1424 static __inline pd_entry_t *
pmap_pdpe_to_pde(pdp_entry_t * pdpe,vm_offset_t va)1425 pmap_pdpe_to_pde(pdp_entry_t *pdpe, vm_offset_t va)
1426 {
1427 	pd_entry_t *pde;
1428 
1429 	KASSERT((*pdpe & PG_PS) == 0,
1430 	    ("%s: pdpe %#lx is a leaf", __func__, *pdpe));
1431 	pde = (pd_entry_t *)PHYS_TO_DMAP(*pdpe & PG_FRAME);
1432 	return (&pde[pmap_pde_index(va)]);
1433 }
1434 
1435 /* Return a pointer to the PD slot that corresponds to a VA */
1436 static __inline pd_entry_t *
pmap_pde(pmap_t pmap,vm_offset_t va)1437 pmap_pde(pmap_t pmap, vm_offset_t va)
1438 {
1439 	pdp_entry_t *pdpe;
1440 	pt_entry_t PG_V;
1441 
1442 	PG_V = pmap_valid_bit(pmap);
1443 	pdpe = pmap_pdpe(pmap, va);
1444 	if (pdpe == NULL || (*pdpe & PG_V) == 0)
1445 		return (NULL);
1446 	KASSERT((*pdpe & PG_PS) == 0,
1447 	    ("pmap_pde for 1G page, pmap %p va %#lx", pmap, va));
1448 	return (pmap_pdpe_to_pde(pdpe, va));
1449 }
1450 
1451 /* Return a pointer to the PT slot that corresponds to a VA */
1452 static __inline pt_entry_t *
pmap_pde_to_pte(pd_entry_t * pde,vm_offset_t va)1453 pmap_pde_to_pte(pd_entry_t *pde, vm_offset_t va)
1454 {
1455 	pt_entry_t *pte;
1456 
1457 	KASSERT((*pde & PG_PS) == 0,
1458 	    ("%s: pde %#lx is a leaf", __func__, *pde));
1459 	pte = (pt_entry_t *)PHYS_TO_DMAP(*pde & PG_FRAME);
1460 	return (&pte[pmap_pte_index(va)]);
1461 }
1462 
1463 /* Return a pointer to the PT slot that corresponds to a VA */
1464 static __inline pt_entry_t *
pmap_pte(pmap_t pmap,vm_offset_t va)1465 pmap_pte(pmap_t pmap, vm_offset_t va)
1466 {
1467 	pd_entry_t *pde;
1468 	pt_entry_t PG_V;
1469 
1470 	PG_V = pmap_valid_bit(pmap);
1471 	pde = pmap_pde(pmap, va);
1472 	if (pde == NULL || (*pde & PG_V) == 0)
1473 		return (NULL);
1474 	if ((*pde & PG_PS) != 0)	/* compat with i386 pmap_pte() */
1475 		return ((pt_entry_t *)pde);
1476 	return (pmap_pde_to_pte(pde, va));
1477 }
1478 
1479 static __inline void
pmap_resident_count_adj(pmap_t pmap,int count)1480 pmap_resident_count_adj(pmap_t pmap, int count)
1481 {
1482 
1483 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
1484 	KASSERT(pmap->pm_stats.resident_count + count >= 0,
1485 	    ("pmap %p resident count underflow %ld %d", pmap,
1486 	    pmap->pm_stats.resident_count, count));
1487 	pmap->pm_stats.resident_count += count;
1488 }
1489 
1490 static __inline void
pmap_pt_page_count_pinit(pmap_t pmap,int count)1491 pmap_pt_page_count_pinit(pmap_t pmap, int count)
1492 {
1493 	KASSERT(pmap->pm_stats.resident_count + count >= 0,
1494 	    ("pmap %p resident count underflow %ld %d", pmap,
1495 	    pmap->pm_stats.resident_count, count));
1496 	pmap->pm_stats.resident_count += count;
1497 }
1498 
1499 static __inline void
pmap_pt_page_count_adj(pmap_t pmap,int count)1500 pmap_pt_page_count_adj(pmap_t pmap, int count)
1501 {
1502 	if (pmap == kernel_pmap)
1503 		counter_u64_add(kernel_pt_page_count, count);
1504 	else {
1505 		if (pmap != NULL)
1506 			pmap_resident_count_adj(pmap, count);
1507 		counter_u64_add(user_pt_page_count, count);
1508 	}
1509 }
1510 
1511 pt_entry_t vtoptem __read_mostly = ((1ul << (NPTEPGSHIFT + NPDEPGSHIFT +
1512     NPDPEPGSHIFT + NPML4EPGSHIFT)) - 1) << 3;
1513 vm_offset_t PTmap __read_mostly = (vm_offset_t)P4Tmap;
1514 
1515 PMAP_INLINE pt_entry_t *
vtopte(vm_offset_t va)1516 vtopte(vm_offset_t va)
1517 {
1518 	KASSERT(va >= VM_MAXUSER_ADDRESS, ("vtopte on a uva/gpa 0x%0lx", va));
1519 
1520 	return ((pt_entry_t *)(PTmap + ((va >> (PAGE_SHIFT - 3)) & vtoptem)));
1521 }
1522 
1523 pd_entry_t vtopdem __read_mostly = ((1ul << (NPDEPGSHIFT + NPDPEPGSHIFT +
1524     NPML4EPGSHIFT)) - 1) << 3;
1525 vm_offset_t PDmap __read_mostly = (vm_offset_t)P4Dmap;
1526 
1527 static __inline pd_entry_t *
vtopde(vm_offset_t va)1528 vtopde(vm_offset_t va)
1529 {
1530 	KASSERT(va >= VM_MAXUSER_ADDRESS, ("vtopde on a uva/gpa 0x%0lx", va));
1531 
1532 	return ((pt_entry_t *)(PDmap + ((va >> (PDRSHIFT - 3)) & vtopdem)));
1533 }
1534 
1535 static u_int64_t
allocpages(vm_paddr_t * firstaddr,int n)1536 allocpages(vm_paddr_t *firstaddr, int n)
1537 {
1538 	u_int64_t ret;
1539 
1540 	ret = *firstaddr;
1541 	bzero((void *)ret, n * PAGE_SIZE);
1542 	*firstaddr += n * PAGE_SIZE;
1543 	return (ret);
1544 }
1545 
1546 CTASSERT(powerof2(NDMPML4E));
1547 
1548 /* number of kernel PDP slots */
1549 #define	NKPDPE(ptpgs)		howmany(ptpgs, NPDEPG)
1550 
1551 static void
nkpt_init(vm_paddr_t addr)1552 nkpt_init(vm_paddr_t addr)
1553 {
1554 	int pt_pages;
1555 
1556 #ifdef NKPT
1557 	pt_pages = NKPT;
1558 #else
1559 	pt_pages = howmany(addr - kernphys, NBPDR) + 1; /* +1 for 2M hole @0 */
1560 	pt_pages += NKPDPE(pt_pages);
1561 
1562 	/*
1563 	 * Add some slop beyond the bare minimum required for bootstrapping
1564 	 * the kernel.
1565 	 *
1566 	 * This is quite important when allocating KVA for kernel modules.
1567 	 * The modules are required to be linked in the negative 2GB of
1568 	 * the address space.  If we run out of KVA in this region then
1569 	 * pmap_growkernel() will need to allocate page table pages to map
1570 	 * the entire 512GB of KVA space which is an unnecessary tax on
1571 	 * physical memory.
1572 	 *
1573 	 * Secondly, device memory mapped as part of setting up the low-
1574 	 * level console(s) is taken from KVA, starting at virtual_avail.
1575 	 * This is because cninit() is called after pmap_bootstrap() but
1576 	 * before vm_mem_init() and pmap_init(). 20MB for a frame buffer
1577 	 * is not uncommon.
1578 	 */
1579 	pt_pages += 32;		/* 64MB additional slop. */
1580 #endif
1581 	nkpt = pt_pages;
1582 }
1583 
1584 /*
1585  * Returns the proper write/execute permission for a physical page that is
1586  * part of the initial boot allocations.
1587  *
1588  * If the page has kernel text, it is marked as read-only. If the page has
1589  * kernel read-only data, it is marked as read-only/not-executable. If the
1590  * page has only read-write data, it is marked as read-write/not-executable.
1591  * If the page is below/above the kernel range, it is marked as read-write.
1592  *
1593  * This function operates on 2M pages, since we map the kernel space that
1594  * way.
1595  */
1596 static inline pt_entry_t
bootaddr_rwx(vm_paddr_t pa)1597 bootaddr_rwx(vm_paddr_t pa)
1598 {
1599 	/*
1600 	 * The kernel is loaded at a 2MB-aligned address, and memory below that
1601 	 * need not be executable.  The .bss section is padded to a 2MB
1602 	 * boundary, so memory following the kernel need not be executable
1603 	 * either.  Preloaded kernel modules have their mapping permissions
1604 	 * fixed up by the linker.
1605 	 */
1606 	if (pa < trunc_2mpage(kernphys + btext - KERNSTART) ||
1607 	    pa >= trunc_2mpage(kernphys + _end - KERNSTART))
1608 		return (X86_PG_RW | pg_nx);
1609 
1610 	/*
1611 	 * The linker should ensure that the read-only and read-write
1612 	 * portions don't share the same 2M page, so this shouldn't
1613 	 * impact read-only data. However, in any case, any page with
1614 	 * read-write data needs to be read-write.
1615 	 */
1616 	if (pa >= trunc_2mpage(kernphys + brwsection - KERNSTART))
1617 		return (X86_PG_RW | pg_nx);
1618 
1619 	/*
1620 	 * Mark any 2M page containing kernel text as read-only. Mark
1621 	 * other pages with read-only data as read-only and not executable.
1622 	 * (It is likely a small portion of the read-only data section will
1623 	 * be marked as read-only, but executable. This should be acceptable
1624 	 * since the read-only protection will keep the data from changing.)
1625 	 * Note that fixups to the .text section will still work until we
1626 	 * set CR0.WP.
1627 	 */
1628 	if (pa < round_2mpage(kernphys + etext - KERNSTART))
1629 		return (0);
1630 	return (pg_nx);
1631 }
1632 
1633 static void
create_pagetables(vm_paddr_t * firstaddr)1634 create_pagetables(vm_paddr_t *firstaddr)
1635 {
1636 	pd_entry_t *pd_p;
1637 	pdp_entry_t *pdp_p;
1638 	pml4_entry_t *p4_p;
1639 	uint64_t DMPDkernphys;
1640 	vm_paddr_t pax;
1641 #ifdef KASAN
1642 	pt_entry_t *pt_p;
1643 	uint64_t KASANPDphys, KASANPTphys, KASANphys;
1644 	vm_offset_t kasankernbase;
1645 	int kasankpdpi, kasankpdi, nkasanpte;
1646 #endif
1647 	int i, j, ndm1g, nkpdpe, nkdmpde;
1648 
1649 	/* Allocate page table pages for the direct map */
1650 	ndmpdp = howmany(ptoa(Maxmem), NBPDP);
1651 	if (ndmpdp < 4)		/* Minimum 4GB of dirmap */
1652 		ndmpdp = 4;
1653 	ndmpdpphys = howmany(ndmpdp, NPDPEPG);
1654 	if (ndmpdpphys > NDMPML4E) {
1655 		/*
1656 		 * Each NDMPML4E allows 512 GB, so limit to that,
1657 		 * and then readjust ndmpdp and ndmpdpphys.
1658 		 */
1659 		printf("NDMPML4E limits system to %d GB\n", NDMPML4E * 512);
1660 		Maxmem = atop(NDMPML4E * NBPML4);
1661 		ndmpdpphys = NDMPML4E;
1662 		ndmpdp = NDMPML4E * NPDEPG;
1663 	}
1664 	DMPDPphys = allocpages(firstaddr, ndmpdpphys);
1665 	ndm1g = 0;
1666 	if ((amd_feature & AMDID_PAGE1GB) != 0) {
1667 		/*
1668 		 * Calculate the number of 1G pages that will fully fit in
1669 		 * Maxmem.
1670 		 */
1671 		ndm1g = ptoa(Maxmem) >> PDPSHIFT;
1672 
1673 		/*
1674 		 * Allocate 2M pages for the kernel. These will be used in
1675 		 * place of the one or more 1G pages from ndm1g that maps
1676 		 * kernel memory into DMAP.
1677 		 */
1678 		nkdmpde = howmany((vm_offset_t)brwsection - KERNSTART +
1679 		    kernphys - rounddown2(kernphys, NBPDP), NBPDP);
1680 		DMPDkernphys = allocpages(firstaddr, nkdmpde);
1681 	}
1682 	if (ndm1g < ndmpdp)
1683 		DMPDphys = allocpages(firstaddr, ndmpdp - ndm1g);
1684 	dmaplimit = (vm_paddr_t)ndmpdp << PDPSHIFT;
1685 
1686 	/* Allocate pages */
1687 	KPML4phys = allocpages(firstaddr, 1);
1688 	KPDPphys = allocpages(firstaddr, NKPML4E);
1689 #ifdef KASAN
1690 	KASANPDPphys = allocpages(firstaddr, NKASANPML4E);
1691 	KASANPDphys = allocpages(firstaddr, 1);
1692 #endif
1693 
1694 	/*
1695 	 * Allocate the initial number of kernel page table pages required to
1696 	 * bootstrap.  We defer this until after all memory-size dependent
1697 	 * allocations are done (e.g. direct map), so that we don't have to
1698 	 * build in too much slop in our estimate.
1699 	 *
1700 	 * Note that when NKPML4E > 1, we have an empty page underneath
1701 	 * all but the KPML4I'th one, so we need NKPML4E-1 extra (zeroed)
1702 	 * pages.  (pmap_enter requires a PD page to exist for each KPML4E.)
1703 	 */
1704 	nkpt_init(*firstaddr);
1705 	nkpdpe = NKPDPE(nkpt);
1706 
1707 	KPTphys = allocpages(firstaddr, nkpt);
1708 	KPDphys = allocpages(firstaddr, nkpdpe);
1709 
1710 #ifdef KASAN
1711 	nkasanpte = howmany(nkpt, KASAN_SHADOW_SCALE);
1712 	KASANPTphys = allocpages(firstaddr, nkasanpte);
1713 	KASANphys = allocpages(firstaddr, nkasanpte * NPTEPG);
1714 #endif
1715 
1716 	/*
1717 	 * Connect the zero-filled PT pages to their PD entries.  This
1718 	 * implicitly maps the PT pages at their correct locations within
1719 	 * the PTmap.
1720 	 */
1721 	pd_p = (pd_entry_t *)KPDphys;
1722 	for (i = 0; i < nkpt; i++)
1723 		pd_p[i] = (KPTphys + ptoa(i)) | X86_PG_RW | X86_PG_V;
1724 
1725 	/*
1726 	 * Map from start of the kernel in physical memory (staging
1727 	 * area) to the end of loader preallocated memory using 2MB
1728 	 * pages.  This replaces some of the PD entries created above.
1729 	 * For compatibility, identity map 2M at the start.
1730 	 */
1731 	pd_p[0] = X86_PG_V | PG_PS | pg_g | X86_PG_M | X86_PG_A |
1732 	    X86_PG_RW | pg_nx;
1733 	for (i = 1, pax = kernphys; pax < KERNend; i++, pax += NBPDR) {
1734 		/* Preset PG_M and PG_A because demotion expects it. */
1735 		pd_p[i] = pax | X86_PG_V | PG_PS | pg_g | X86_PG_M |
1736 		    X86_PG_A | bootaddr_rwx(pax);
1737 	}
1738 
1739 	/*
1740 	 * Because we map the physical blocks in 2M pages, adjust firstaddr
1741 	 * to record the physical blocks we've actually mapped into kernel
1742 	 * virtual address space.
1743 	 */
1744 	if (*firstaddr < round_2mpage(KERNend))
1745 		*firstaddr = round_2mpage(KERNend);
1746 
1747 	/* And connect up the PD to the PDP (leaving room for L4 pages) */
1748 	pdp_p = (pdp_entry_t *)(KPDPphys + ptoa(KPML4I - KPML4BASE));
1749 	for (i = 0; i < nkpdpe; i++)
1750 		pdp_p[i + KPDPI] = (KPDphys + ptoa(i)) | X86_PG_RW | X86_PG_V;
1751 
1752 #ifdef KASAN
1753 	kasankernbase = kasan_md_addr_to_shad(KERNBASE);
1754 	kasankpdpi = pmap_pdpe_index(kasankernbase);
1755 	kasankpdi = pmap_pde_index(kasankernbase);
1756 
1757 	pdp_p = (pdp_entry_t *)KASANPDPphys;
1758 	pdp_p[kasankpdpi] = (KASANPDphys | X86_PG_RW | X86_PG_V | pg_nx);
1759 
1760 	pd_p = (pd_entry_t *)KASANPDphys;
1761 	for (i = 0; i < nkasanpte; i++)
1762 		pd_p[i + kasankpdi] = (KASANPTphys + ptoa(i)) | X86_PG_RW |
1763 		    X86_PG_V | pg_nx;
1764 
1765 	pt_p = (pt_entry_t *)KASANPTphys;
1766 	for (i = 0; i < nkasanpte * NPTEPG; i++)
1767 		pt_p[i] = (KASANphys + ptoa(i)) | X86_PG_RW | X86_PG_V |
1768 		    X86_PG_M | X86_PG_A | pg_nx;
1769 #endif
1770 
1771 	/*
1772 	 * Now, set up the direct map region using 2MB and/or 1GB pages.  If
1773 	 * the end of physical memory is not aligned to a 1GB page boundary,
1774 	 * then the residual physical memory is mapped with 2MB pages.  Later,
1775 	 * if pmap_mapdev{_attr}() uses the direct map for non-write-back
1776 	 * memory, pmap_change_attr() will demote any 2MB or 1GB page mappings
1777 	 * that are partially used.
1778 	 */
1779 	pd_p = (pd_entry_t *)DMPDphys;
1780 	for (i = NPDEPG * ndm1g, j = 0; i < NPDEPG * ndmpdp; i++, j++) {
1781 		pd_p[j] = (vm_paddr_t)i << PDRSHIFT;
1782 		/* Preset PG_M and PG_A because demotion expects it. */
1783 		pd_p[j] |= X86_PG_RW | X86_PG_V | PG_PS | pg_g |
1784 		    X86_PG_M | X86_PG_A | pg_nx;
1785 	}
1786 	pdp_p = (pdp_entry_t *)DMPDPphys;
1787 	for (i = 0; i < ndm1g; i++) {
1788 		pdp_p[i] = (vm_paddr_t)i << PDPSHIFT;
1789 		/* Preset PG_M and PG_A because demotion expects it. */
1790 		pdp_p[i] |= X86_PG_RW | X86_PG_V | PG_PS | pg_g |
1791 		    X86_PG_M | X86_PG_A | pg_nx;
1792 	}
1793 	for (j = 0; i < ndmpdp; i++, j++) {
1794 		pdp_p[i] = DMPDphys + ptoa(j);
1795 		pdp_p[i] |= X86_PG_RW | X86_PG_V | pg_nx;
1796 	}
1797 
1798 	/*
1799 	 * Instead of using a 1G page for the memory containing the kernel,
1800 	 * use 2M pages with read-only and no-execute permissions.  (If using 1G
1801 	 * pages, this will partially overwrite the PDPEs above.)
1802 	 */
1803 	if (ndm1g > 0) {
1804 		pd_p = (pd_entry_t *)DMPDkernphys;
1805 		for (i = 0, pax = rounddown2(kernphys, NBPDP);
1806 		    i < NPDEPG * nkdmpde; i++, pax += NBPDR) {
1807 			pd_p[i] = pax | X86_PG_V | PG_PS | pg_g | X86_PG_M |
1808 			    X86_PG_A | pg_nx | bootaddr_rwx(pax);
1809 		}
1810 		j = rounddown2(kernphys, NBPDP) >> PDPSHIFT;
1811 		for (i = 0; i < nkdmpde; i++) {
1812 			pdp_p[i + j] = (DMPDkernphys + ptoa(i)) |
1813 			    X86_PG_RW | X86_PG_V | pg_nx;
1814 		}
1815 	}
1816 
1817 	/* And recursively map PML4 to itself in order to get PTmap */
1818 	p4_p = (pml4_entry_t *)KPML4phys;
1819 	p4_p[PML4PML4I] = KPML4phys;
1820 	p4_p[PML4PML4I] |= X86_PG_RW | X86_PG_V | pg_nx;
1821 
1822 #ifdef KASAN
1823 	/* Connect the KASAN shadow map slots up to the PML4. */
1824 	for (i = 0; i < NKASANPML4E; i++) {
1825 		p4_p[KASANPML4I + i] = KASANPDPphys + ptoa(i);
1826 		p4_p[KASANPML4I + i] |= X86_PG_RW | X86_PG_V | pg_nx;
1827 	}
1828 #endif
1829 
1830 	/* Connect the Direct Map slots up to the PML4. */
1831 	for (i = 0; i < ndmpdpphys; i++) {
1832 		p4_p[DMPML4I + i] = DMPDPphys + ptoa(i);
1833 		p4_p[DMPML4I + i] |= X86_PG_RW | X86_PG_V | pg_nx;
1834 	}
1835 
1836 	/* Connect the KVA slots up to the PML4 */
1837 	for (i = 0; i < NKPML4E; i++) {
1838 		p4_p[KPML4BASE + i] = KPDPphys + ptoa(i);
1839 		p4_p[KPML4BASE + i] |= X86_PG_RW | X86_PG_V;
1840 	}
1841 
1842 	kernel_pml4 = (pml4_entry_t *)PHYS_TO_DMAP(KPML4phys);
1843 }
1844 
1845 /*
1846  *	Bootstrap the system enough to run with virtual memory.
1847  *
1848  *	On amd64 this is called after mapping has already been enabled
1849  *	and just syncs the pmap module with what has already been done.
1850  *	[We can't call it easily with mapping off since the kernel is not
1851  *	mapped with PA == VA, hence we would have to relocate every address
1852  *	from the linked base (virtual) address "KERNBASE" to the actual
1853  *	(physical) address starting relative to 0]
1854  */
1855 void
pmap_bootstrap(vm_paddr_t * firstaddr)1856 pmap_bootstrap(vm_paddr_t *firstaddr)
1857 {
1858 	vm_offset_t va;
1859 	pt_entry_t *pte, *pcpu_pte;
1860 	struct region_descriptor r_gdt;
1861 	uint64_t cr4, pcpu_phys;
1862 	u_long res;
1863 	int i;
1864 
1865 	KERNend = *firstaddr;
1866 	res = atop(KERNend - (vm_paddr_t)kernphys);
1867 
1868 	if (!pti)
1869 		pg_g = X86_PG_G;
1870 
1871 	/*
1872 	 * Create an initial set of page tables to run the kernel in.
1873 	 */
1874 	create_pagetables(firstaddr);
1875 
1876 	pcpu_phys = allocpages(firstaddr, MAXCPU);
1877 
1878 	/*
1879 	 * Add a physical memory segment (vm_phys_seg) corresponding to the
1880 	 * preallocated kernel page table pages so that vm_page structures
1881 	 * representing these pages will be created.  The vm_page structures
1882 	 * are required for promotion of the corresponding kernel virtual
1883 	 * addresses to superpage mappings.
1884 	 */
1885 	vm_phys_early_add_seg(KPTphys, KPTphys + ptoa(nkpt));
1886 
1887 	/*
1888 	 * Account for the virtual addresses mapped by create_pagetables().
1889 	 */
1890 	virtual_avail = (vm_offset_t)KERNSTART + round_2mpage(KERNend -
1891 	    (vm_paddr_t)kernphys);
1892 	virtual_end = VM_MAX_KERNEL_ADDRESS;
1893 
1894 	/*
1895 	 * Enable PG_G global pages, then switch to the kernel page
1896 	 * table from the bootstrap page table.  After the switch, it
1897 	 * is possible to enable SMEP and SMAP since PG_U bits are
1898 	 * correct now.
1899 	 */
1900 	cr4 = rcr4();
1901 	cr4 |= CR4_PGE;
1902 	load_cr4(cr4);
1903 	load_cr3(KPML4phys);
1904 	if (cpu_stdext_feature & CPUID_STDEXT_SMEP)
1905 		cr4 |= CR4_SMEP;
1906 	if (cpu_stdext_feature & CPUID_STDEXT_SMAP)
1907 		cr4 |= CR4_SMAP;
1908 	load_cr4(cr4);
1909 
1910 	/*
1911 	 * Initialize the kernel pmap (which is statically allocated).
1912 	 * Count bootstrap data as being resident in case any of this data is
1913 	 * later unmapped (using pmap_remove()) and freed.
1914 	 */
1915 	PMAP_LOCK_INIT(kernel_pmap);
1916 	kernel_pmap->pm_pmltop = kernel_pml4;
1917 	kernel_pmap->pm_cr3 = KPML4phys;
1918 	kernel_pmap->pm_ucr3 = PMAP_NO_CR3;
1919 	TAILQ_INIT(&kernel_pmap->pm_pvchunk);
1920 	kernel_pmap->pm_stats.resident_count = res;
1921 	kernel_pmap->pm_flags = pmap_flags;
1922 
1923 	/*
1924 	 * The kernel pmap is always active on all CPUs.  Once CPUs are
1925 	 * enumerated, the mask will be set equal to all_cpus.
1926 	 */
1927 	CPU_FILL(&kernel_pmap->pm_active);
1928 
1929  	/*
1930 	 * Initialize the TLB invalidations generation number lock.
1931 	 */
1932 	mtx_init(&invl_gen_mtx, "invlgn", NULL, MTX_DEF);
1933 
1934 	/*
1935 	 * Reserve some special page table entries/VA space for temporary
1936 	 * mapping of pages.
1937 	 */
1938 #define	SYSMAP(c, p, v, n)	\
1939 	v = (c)va; va += ((n)*PAGE_SIZE); p = pte; pte += (n);
1940 
1941 	va = virtual_avail;
1942 	pte = vtopte(va);
1943 
1944 	/*
1945 	 * Crashdump maps.  The first page is reused as CMAP1 for the
1946 	 * memory test.
1947 	 */
1948 	SYSMAP(caddr_t, CMAP1, crashdumpmap, MAXDUMPPGS)
1949 	CADDR1 = crashdumpmap;
1950 
1951 	SYSMAP(struct pcpu *, pcpu_pte, __pcpu, MAXCPU);
1952 	virtual_avail = va;
1953 
1954 	for (i = 0; i < MAXCPU; i++) {
1955 		pcpu_pte[i] = (pcpu_phys + ptoa(i)) | X86_PG_V | X86_PG_RW |
1956 		    pg_g | pg_nx | X86_PG_M | X86_PG_A;
1957 	}
1958 
1959 	/*
1960 	 * Re-initialize PCPU area for BSP after switching.
1961 	 * Make hardware use gdt and common_tss from the new PCPU.
1962 	 */
1963 	STAILQ_INIT(&cpuhead);
1964 	wrmsr(MSR_GSBASE, (uint64_t)&__pcpu[0]);
1965 	pcpu_init(&__pcpu[0], 0, sizeof(struct pcpu));
1966 	amd64_bsp_pcpu_init1(&__pcpu[0]);
1967 	amd64_bsp_ist_init(&__pcpu[0]);
1968 	__pcpu[0].pc_common_tss.tss_iobase = sizeof(struct amd64tss) +
1969 	    IOPERM_BITMAP_SIZE;
1970 	memcpy(__pcpu[0].pc_gdt, temp_bsp_pcpu.pc_gdt, NGDT *
1971 	    sizeof(struct user_segment_descriptor));
1972 	gdt_segs[GPROC0_SEL].ssd_base = (uintptr_t)&__pcpu[0].pc_common_tss;
1973 	ssdtosyssd(&gdt_segs[GPROC0_SEL],
1974 	    (struct system_segment_descriptor *)&__pcpu[0].pc_gdt[GPROC0_SEL]);
1975 	r_gdt.rd_limit = NGDT * sizeof(struct user_segment_descriptor) - 1;
1976 	r_gdt.rd_base = (long)__pcpu[0].pc_gdt;
1977 	lgdt(&r_gdt);
1978 	wrmsr(MSR_GSBASE, (uint64_t)&__pcpu[0]);
1979 	ltr(GSEL(GPROC0_SEL, SEL_KPL));
1980 	__pcpu[0].pc_dynamic = temp_bsp_pcpu.pc_dynamic;
1981 	__pcpu[0].pc_acpi_id = temp_bsp_pcpu.pc_acpi_id;
1982 
1983 	/*
1984 	 * Initialize the PAT MSR.
1985 	 * pmap_init_pat() clears and sets CR4_PGE, which, as a
1986 	 * side-effect, invalidates stale PG_G TLB entries that might
1987 	 * have been created in our pre-boot environment.
1988 	 */
1989 	pmap_init_pat();
1990 
1991 	/* Initialize TLB Context Id. */
1992 	if (pmap_pcid_enabled) {
1993 		for (i = 0; i < MAXCPU; i++) {
1994 			kernel_pmap->pm_pcids[i].pm_pcid = PMAP_PCID_KERN;
1995 			kernel_pmap->pm_pcids[i].pm_gen = 1;
1996 		}
1997 
1998 		/*
1999 		 * PMAP_PCID_KERN + 1 is used for initialization of
2000 		 * proc0 pmap.  The pmap' pcid state might be used by
2001 		 * EFIRT entry before first context switch, so it
2002 		 * needs to be valid.
2003 		 */
2004 		PCPU_SET(pcid_next, PMAP_PCID_KERN + 2);
2005 		PCPU_SET(pcid_gen, 1);
2006 
2007 		/*
2008 		 * pcpu area for APs is zeroed during AP startup.
2009 		 * pc_pcid_next and pc_pcid_gen are initialized by AP
2010 		 * during pcpu setup.
2011 		 */
2012 		load_cr4(rcr4() | CR4_PCIDE);
2013 	}
2014 }
2015 
2016 /*
2017  * Setup the PAT MSR.
2018  */
2019 void
pmap_init_pat(void)2020 pmap_init_pat(void)
2021 {
2022 	uint64_t pat_msr;
2023 	u_long cr0, cr4;
2024 	int i;
2025 
2026 	/* Bail if this CPU doesn't implement PAT. */
2027 	if ((cpu_feature & CPUID_PAT) == 0)
2028 		panic("no PAT??");
2029 
2030 	/* Set default PAT index table. */
2031 	for (i = 0; i < PAT_INDEX_SIZE; i++)
2032 		pat_index[i] = -1;
2033 	pat_index[PAT_WRITE_BACK] = 0;
2034 	pat_index[PAT_WRITE_THROUGH] = 1;
2035 	pat_index[PAT_UNCACHEABLE] = 3;
2036 	pat_index[PAT_WRITE_COMBINING] = 6;
2037 	pat_index[PAT_WRITE_PROTECTED] = 5;
2038 	pat_index[PAT_UNCACHED] = 2;
2039 
2040 	/*
2041 	 * Initialize default PAT entries.
2042 	 * Leave the indices 0-3 at the default of WB, WT, UC-, and UC.
2043 	 * Program 5 and 6 as WP and WC.
2044 	 *
2045 	 * Leave 4 and 7 as WB and UC.  Note that a recursive page table
2046 	 * mapping for a 2M page uses a PAT value with the bit 3 set due
2047 	 * to its overload with PG_PS.
2048 	 */
2049 	pat_msr = PAT_VALUE(0, PAT_WRITE_BACK) |
2050 	    PAT_VALUE(1, PAT_WRITE_THROUGH) |
2051 	    PAT_VALUE(2, PAT_UNCACHED) |
2052 	    PAT_VALUE(3, PAT_UNCACHEABLE) |
2053 	    PAT_VALUE(4, PAT_WRITE_BACK) |
2054 	    PAT_VALUE(5, PAT_WRITE_PROTECTED) |
2055 	    PAT_VALUE(6, PAT_WRITE_COMBINING) |
2056 	    PAT_VALUE(7, PAT_UNCACHEABLE);
2057 
2058 	/* Disable PGE. */
2059 	cr4 = rcr4();
2060 	load_cr4(cr4 & ~CR4_PGE);
2061 
2062 	/* Disable caches (CD = 1, NW = 0). */
2063 	cr0 = rcr0();
2064 	load_cr0((cr0 & ~CR0_NW) | CR0_CD);
2065 
2066 	/* Flushes caches and TLBs. */
2067 	wbinvd();
2068 	invltlb();
2069 
2070 	/* Update PAT and index table. */
2071 	wrmsr(MSR_PAT, pat_msr);
2072 
2073 	/* Flush caches and TLBs again. */
2074 	wbinvd();
2075 	invltlb();
2076 
2077 	/* Restore caches and PGE. */
2078 	load_cr0(cr0);
2079 	load_cr4(cr4);
2080 }
2081 
2082 vm_page_t
pmap_page_alloc_below_4g(bool zeroed)2083 pmap_page_alloc_below_4g(bool zeroed)
2084 {
2085 	return (vm_page_alloc_noobj_contig((zeroed ? VM_ALLOC_ZERO : 0),
2086 	    1, 0, (1ULL << 32), PAGE_SIZE, 0, VM_MEMATTR_DEFAULT));
2087 }
2088 
2089 extern const char la57_trampoline[], la57_trampoline_gdt_desc[],
2090     la57_trampoline_gdt[], la57_trampoline_end[];
2091 
2092 static void
pmap_bootstrap_la57(void * arg __unused)2093 pmap_bootstrap_la57(void *arg __unused)
2094 {
2095 	char *v_code;
2096 	pml5_entry_t *v_pml5;
2097 	pml4_entry_t *v_pml4;
2098 	pdp_entry_t *v_pdp;
2099 	pd_entry_t *v_pd;
2100 	pt_entry_t *v_pt;
2101 	vm_page_t m_code, m_pml4, m_pdp, m_pd, m_pt, m_pml5;
2102 	void (*la57_tramp)(uint64_t pml5);
2103 	struct region_descriptor r_gdt;
2104 
2105 	if ((cpu_stdext_feature2 & CPUID_STDEXT2_LA57) == 0)
2106 		return;
2107 	TUNABLE_INT_FETCH("vm.pmap.la57", &la57);
2108 	if (!la57)
2109 		return;
2110 
2111 	r_gdt.rd_limit = NGDT * sizeof(struct user_segment_descriptor) - 1;
2112 	r_gdt.rd_base = (long)__pcpu[0].pc_gdt;
2113 
2114 	m_code = pmap_page_alloc_below_4g(true);
2115 	v_code = (char *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(m_code));
2116 	m_pml5 = pmap_page_alloc_below_4g(true);
2117 	KPML5phys = VM_PAGE_TO_PHYS(m_pml5);
2118 	v_pml5 = (pml5_entry_t *)PHYS_TO_DMAP(KPML5phys);
2119 	m_pml4 = pmap_page_alloc_below_4g(true);
2120 	v_pml4 = (pdp_entry_t *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(m_pml4));
2121 	m_pdp = pmap_page_alloc_below_4g(true);
2122 	v_pdp = (pdp_entry_t *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(m_pdp));
2123 	m_pd = pmap_page_alloc_below_4g(true);
2124 	v_pd = (pdp_entry_t *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(m_pd));
2125 	m_pt = pmap_page_alloc_below_4g(true);
2126 	v_pt = (pt_entry_t *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(m_pt));
2127 
2128 	/*
2129 	 * Map m_code 1:1, it appears below 4G in KVA due to physical
2130 	 * address being below 4G.  Since kernel KVA is in upper half,
2131 	 * the pml4e should be zero and free for temporary use.
2132 	 */
2133 	kernel_pmap->pm_pmltop[pmap_pml4e_index(VM_PAGE_TO_PHYS(m_code))] =
2134 	    VM_PAGE_TO_PHYS(m_pdp) | X86_PG_V | X86_PG_RW | X86_PG_A |
2135 	    X86_PG_M;
2136 	v_pdp[pmap_pdpe_index(VM_PAGE_TO_PHYS(m_code))] =
2137 	    VM_PAGE_TO_PHYS(m_pd) | X86_PG_V | X86_PG_RW | X86_PG_A |
2138 	    X86_PG_M;
2139 	v_pd[pmap_pde_index(VM_PAGE_TO_PHYS(m_code))] =
2140 	    VM_PAGE_TO_PHYS(m_pt) | X86_PG_V | X86_PG_RW | X86_PG_A |
2141 	    X86_PG_M;
2142 	v_pt[pmap_pte_index(VM_PAGE_TO_PHYS(m_code))] =
2143 	    VM_PAGE_TO_PHYS(m_code) | X86_PG_V | X86_PG_RW | X86_PG_A |
2144 	    X86_PG_M;
2145 
2146 	/*
2147 	 * Add pml5 entry at top of KVA pointing to existing pml4 table,
2148 	 * entering all existing kernel mappings into level 5 table.
2149 	 */
2150 	v_pml5[pmap_pml5e_index(UPT_MAX_ADDRESS)] = KPML4phys | X86_PG_V |
2151 	    X86_PG_RW | X86_PG_A | X86_PG_M | pg_g;
2152 
2153 	/*
2154 	 * Add pml5 entry for 1:1 trampoline mapping after LA57 is turned on.
2155 	 */
2156 	v_pml5[pmap_pml5e_index(VM_PAGE_TO_PHYS(m_code))] =
2157 	    VM_PAGE_TO_PHYS(m_pml4) | X86_PG_V | X86_PG_RW | X86_PG_A |
2158 	    X86_PG_M;
2159 	v_pml4[pmap_pml4e_index(VM_PAGE_TO_PHYS(m_code))] =
2160 	    VM_PAGE_TO_PHYS(m_pdp) | X86_PG_V | X86_PG_RW | X86_PG_A |
2161 	    X86_PG_M;
2162 
2163 	/*
2164 	 * Copy and call the 48->57 trampoline, hope we return there, alive.
2165 	 */
2166 	bcopy(la57_trampoline, v_code, la57_trampoline_end - la57_trampoline);
2167 	*(u_long *)(v_code + 2 + (la57_trampoline_gdt_desc - la57_trampoline)) =
2168 	    la57_trampoline_gdt - la57_trampoline + VM_PAGE_TO_PHYS(m_code);
2169 	la57_tramp = (void (*)(uint64_t))VM_PAGE_TO_PHYS(m_code);
2170 	invlpg((vm_offset_t)la57_tramp);
2171 	la57_tramp(KPML5phys);
2172 
2173 	/*
2174 	 * gdt was necessary reset, switch back to our gdt.
2175 	 */
2176 	lgdt(&r_gdt);
2177 	wrmsr(MSR_GSBASE, (uint64_t)&__pcpu[0]);
2178 	load_ds(_udatasel);
2179 	load_es(_udatasel);
2180 	load_fs(_ufssel);
2181 	ssdtosyssd(&gdt_segs[GPROC0_SEL],
2182 	    (struct system_segment_descriptor *)&__pcpu[0].pc_gdt[GPROC0_SEL]);
2183 	ltr(GSEL(GPROC0_SEL, SEL_KPL));
2184 
2185 	/*
2186 	 * Now unmap the trampoline, and free the pages.
2187 	 * Clear pml5 entry used for 1:1 trampoline mapping.
2188 	 */
2189 	pte_clear(&v_pml5[pmap_pml5e_index(VM_PAGE_TO_PHYS(m_code))]);
2190 	invlpg((vm_offset_t)v_code);
2191 	vm_page_free(m_code);
2192 	vm_page_free(m_pdp);
2193 	vm_page_free(m_pd);
2194 	vm_page_free(m_pt);
2195 
2196 	/*
2197 	 * Recursively map PML5 to itself in order to get PTmap and
2198 	 * PDmap.
2199 	 */
2200 	v_pml5[PML5PML5I] = KPML5phys | X86_PG_RW | X86_PG_V | pg_nx;
2201 
2202 	vtoptem = ((1ul << (NPTEPGSHIFT + NPDEPGSHIFT + NPDPEPGSHIFT +
2203 	    NPML4EPGSHIFT + NPML5EPGSHIFT)) - 1) << 3;
2204 	PTmap = (vm_offset_t)P5Tmap;
2205 	vtopdem = ((1ul << (NPDEPGSHIFT + NPDPEPGSHIFT +
2206 	    NPML4EPGSHIFT + NPML5EPGSHIFT)) - 1) << 3;
2207 	PDmap = (vm_offset_t)P5Dmap;
2208 
2209 	kernel_pmap->pm_cr3 = KPML5phys;
2210 	kernel_pmap->pm_pmltop = v_pml5;
2211 	pmap_pt_page_count_adj(kernel_pmap, 1);
2212 }
2213 SYSINIT(la57, SI_SUB_KMEM, SI_ORDER_ANY, pmap_bootstrap_la57, NULL);
2214 
2215 /*
2216  *	Initialize a vm_page's machine-dependent fields.
2217  */
2218 void
pmap_page_init(vm_page_t m)2219 pmap_page_init(vm_page_t m)
2220 {
2221 
2222 	TAILQ_INIT(&m->md.pv_list);
2223 	m->md.pat_mode = PAT_WRITE_BACK;
2224 }
2225 
2226 static int pmap_allow_2m_x_ept;
2227 SYSCTL_INT(_vm_pmap, OID_AUTO, allow_2m_x_ept, CTLFLAG_RWTUN | CTLFLAG_NOFETCH,
2228     &pmap_allow_2m_x_ept, 0,
2229     "Allow executable superpage mappings in EPT");
2230 
2231 void
pmap_allow_2m_x_ept_recalculate(void)2232 pmap_allow_2m_x_ept_recalculate(void)
2233 {
2234 	/*
2235 	 * SKL002, SKL012S.  Since the EPT format is only used by
2236 	 * Intel CPUs, the vendor check is merely a formality.
2237 	 */
2238 	if (!(cpu_vendor_id != CPU_VENDOR_INTEL ||
2239 	    (cpu_ia32_arch_caps & IA32_ARCH_CAP_IF_PSCHANGE_MC_NO) != 0 ||
2240 	    (CPUID_TO_FAMILY(cpu_id) == 0x6 &&
2241 	    (CPUID_TO_MODEL(cpu_id) == 0x26 ||	/* Atoms */
2242 	    CPUID_TO_MODEL(cpu_id) == 0x27 ||
2243 	    CPUID_TO_MODEL(cpu_id) == 0x35 ||
2244 	    CPUID_TO_MODEL(cpu_id) == 0x36 ||
2245 	    CPUID_TO_MODEL(cpu_id) == 0x37 ||
2246 	    CPUID_TO_MODEL(cpu_id) == 0x86 ||
2247 	    CPUID_TO_MODEL(cpu_id) == 0x1c ||
2248 	    CPUID_TO_MODEL(cpu_id) == 0x4a ||
2249 	    CPUID_TO_MODEL(cpu_id) == 0x4c ||
2250 	    CPUID_TO_MODEL(cpu_id) == 0x4d ||
2251 	    CPUID_TO_MODEL(cpu_id) == 0x5a ||
2252 	    CPUID_TO_MODEL(cpu_id) == 0x5c ||
2253 	    CPUID_TO_MODEL(cpu_id) == 0x5d ||
2254 	    CPUID_TO_MODEL(cpu_id) == 0x5f ||
2255 	    CPUID_TO_MODEL(cpu_id) == 0x6e ||
2256 	    CPUID_TO_MODEL(cpu_id) == 0x7a ||
2257 	    CPUID_TO_MODEL(cpu_id) == 0x57 ||	/* Knights */
2258 	    CPUID_TO_MODEL(cpu_id) == 0x85))))
2259 		pmap_allow_2m_x_ept = 1;
2260 #ifndef BURN_BRIDGES
2261 	TUNABLE_INT_FETCH("hw.allow_2m_x_ept", &pmap_allow_2m_x_ept);
2262 #endif
2263 	TUNABLE_INT_FETCH("vm.pmap.allow_2m_x_ept", &pmap_allow_2m_x_ept);
2264 }
2265 
2266 static bool
pmap_allow_2m_x_page(pmap_t pmap,bool executable)2267 pmap_allow_2m_x_page(pmap_t pmap, bool executable)
2268 {
2269 
2270 	return (pmap->pm_type != PT_EPT || !executable ||
2271 	    !pmap_allow_2m_x_ept);
2272 }
2273 
2274 #ifdef NUMA
2275 static void
pmap_init_pv_table(void)2276 pmap_init_pv_table(void)
2277 {
2278 	struct pmap_large_md_page *pvd;
2279 	vm_size_t s;
2280 	long start, end, highest, pv_npg;
2281 	int domain, i, j, pages;
2282 
2283 	/*
2284 	 * We strongly depend on the size being a power of two, so the assert
2285 	 * is overzealous. However, should the struct be resized to a
2286 	 * different power of two, the code below needs to be revisited.
2287 	 */
2288 	CTASSERT((sizeof(*pvd) == 64));
2289 
2290 	/*
2291 	 * Calculate the size of the array.
2292 	 */
2293 	pmap_last_pa = vm_phys_segs[vm_phys_nsegs - 1].end;
2294 	pv_npg = howmany(pmap_last_pa, NBPDR);
2295 	s = (vm_size_t)pv_npg * sizeof(struct pmap_large_md_page);
2296 	s = round_page(s);
2297 	pv_table = (struct pmap_large_md_page *)kva_alloc(s);
2298 	if (pv_table == NULL)
2299 		panic("%s: kva_alloc failed\n", __func__);
2300 
2301 	/*
2302 	 * Iterate physical segments to allocate space for respective pages.
2303 	 */
2304 	highest = -1;
2305 	s = 0;
2306 	for (i = 0; i < vm_phys_nsegs; i++) {
2307 		end = vm_phys_segs[i].end / NBPDR;
2308 		domain = vm_phys_segs[i].domain;
2309 
2310 		if (highest >= end)
2311 			continue;
2312 
2313 		start = highest + 1;
2314 		pvd = &pv_table[start];
2315 
2316 		pages = end - start + 1;
2317 		s = round_page(pages * sizeof(*pvd));
2318 		highest = start + (s / sizeof(*pvd)) - 1;
2319 
2320 		for (j = 0; j < s; j += PAGE_SIZE) {
2321 			vm_page_t m = vm_page_alloc_noobj_domain(domain, 0);
2322 			if (m == NULL)
2323 				panic("failed to allocate PV table page");
2324 			pmap_qenter((vm_offset_t)pvd + j, &m, 1);
2325 		}
2326 
2327 		for (j = 0; j < s / sizeof(*pvd); j++) {
2328 			rw_init_flags(&pvd->pv_lock, "pmap pv list", RW_NEW);
2329 			TAILQ_INIT(&pvd->pv_page.pv_list);
2330 			pvd->pv_page.pv_gen = 0;
2331 			pvd->pv_page.pat_mode = 0;
2332 			pvd->pv_invl_gen = 0;
2333 			pvd++;
2334 		}
2335 	}
2336 	pvd = &pv_dummy_large;
2337 	rw_init_flags(&pvd->pv_lock, "pmap pv list dummy", RW_NEW);
2338 	TAILQ_INIT(&pvd->pv_page.pv_list);
2339 	pvd->pv_page.pv_gen = 0;
2340 	pvd->pv_page.pat_mode = 0;
2341 	pvd->pv_invl_gen = 0;
2342 }
2343 #else
2344 static void
pmap_init_pv_table(void)2345 pmap_init_pv_table(void)
2346 {
2347 	vm_size_t s;
2348 	long i, pv_npg;
2349 
2350 	/*
2351 	 * Initialize the pool of pv list locks.
2352 	 */
2353 	for (i = 0; i < NPV_LIST_LOCKS; i++)
2354 		rw_init(&pv_list_locks[i], "pmap pv list");
2355 
2356 	/*
2357 	 * Calculate the size of the pv head table for superpages.
2358 	 */
2359 	pv_npg = howmany(vm_phys_segs[vm_phys_nsegs - 1].end, NBPDR);
2360 
2361 	/*
2362 	 * Allocate memory for the pv head table for superpages.
2363 	 */
2364 	s = (vm_size_t)pv_npg * sizeof(struct md_page);
2365 	s = round_page(s);
2366 	pv_table = (struct md_page *)kmem_malloc(s, M_WAITOK | M_ZERO);
2367 	for (i = 0; i < pv_npg; i++)
2368 		TAILQ_INIT(&pv_table[i].pv_list);
2369 	TAILQ_INIT(&pv_dummy.pv_list);
2370 }
2371 #endif
2372 
2373 /*
2374  *	Initialize the pmap module.
2375  *
2376  *	Called by vm_mem_init(), to initialize any structures that the pmap
2377  *	system needs to map virtual memory.
2378  */
2379 void
pmap_init(void)2380 pmap_init(void)
2381 {
2382 	struct pmap_preinit_mapping *ppim;
2383 	vm_page_t m, mpte;
2384 	int error, i, ret, skz63;
2385 
2386 	/* L1TF, reserve page @0 unconditionally */
2387 	vm_page_blacklist_add(0, bootverbose);
2388 
2389 	/* Detect bare-metal Skylake Server and Skylake-X. */
2390 	if (vm_guest == VM_GUEST_NO && cpu_vendor_id == CPU_VENDOR_INTEL &&
2391 	    CPUID_TO_FAMILY(cpu_id) == 0x6 && CPUID_TO_MODEL(cpu_id) == 0x55) {
2392 		/*
2393 		 * Skylake-X errata SKZ63. Processor May Hang When
2394 		 * Executing Code In an HLE Transaction Region between
2395 		 * 40000000H and 403FFFFFH.
2396 		 *
2397 		 * Mark the pages in the range as preallocated.  It
2398 		 * seems to be impossible to distinguish between
2399 		 * Skylake Server and Skylake X.
2400 		 */
2401 		skz63 = 1;
2402 		TUNABLE_INT_FETCH("hw.skz63_enable", &skz63);
2403 		if (skz63 != 0) {
2404 			if (bootverbose)
2405 				printf("SKZ63: skipping 4M RAM starting "
2406 				    "at physical 1G\n");
2407 			for (i = 0; i < atop(0x400000); i++) {
2408 				ret = vm_page_blacklist_add(0x40000000 +
2409 				    ptoa(i), FALSE);
2410 				if (!ret && bootverbose)
2411 					printf("page at %#lx already used\n",
2412 					    0x40000000 + ptoa(i));
2413 			}
2414 		}
2415 	}
2416 
2417 	/* IFU */
2418 	pmap_allow_2m_x_ept_recalculate();
2419 
2420 	/*
2421 	 * Initialize the vm page array entries for the kernel pmap's
2422 	 * page table pages.
2423 	 */
2424 	PMAP_LOCK(kernel_pmap);
2425 	for (i = 0; i < nkpt; i++) {
2426 		mpte = PHYS_TO_VM_PAGE(KPTphys + (i << PAGE_SHIFT));
2427 		KASSERT(mpte >= vm_page_array &&
2428 		    mpte < &vm_page_array[vm_page_array_size],
2429 		    ("pmap_init: page table page is out of range"));
2430 		mpte->pindex = pmap_pde_pindex(KERNBASE) + i;
2431 		mpte->phys_addr = KPTphys + (i << PAGE_SHIFT);
2432 		mpte->ref_count = 1;
2433 
2434 		/*
2435 		 * Collect the page table pages that were replaced by a 2MB
2436 		 * page in create_pagetables().  They are zero filled.
2437 		 */
2438 		if ((i == 0 ||
2439 		    kernphys + ((vm_paddr_t)(i - 1) << PDRSHIFT) < KERNend) &&
2440 		    pmap_insert_pt_page(kernel_pmap, mpte, false))
2441 			panic("pmap_init: pmap_insert_pt_page failed");
2442 	}
2443 	PMAP_UNLOCK(kernel_pmap);
2444 	vm_wire_add(nkpt);
2445 
2446 	/*
2447 	 * If the kernel is running on a virtual machine, then it must assume
2448 	 * that MCA is enabled by the hypervisor.  Moreover, the kernel must
2449 	 * be prepared for the hypervisor changing the vendor and family that
2450 	 * are reported by CPUID.  Consequently, the workaround for AMD Family
2451 	 * 10h Erratum 383 is enabled if the processor's feature set does not
2452 	 * include at least one feature that is only supported by older Intel
2453 	 * or newer AMD processors.
2454 	 */
2455 	if (vm_guest != VM_GUEST_NO && (cpu_feature & CPUID_SS) == 0 &&
2456 	    (cpu_feature2 & (CPUID2_SSSE3 | CPUID2_SSE41 | CPUID2_AESNI |
2457 	    CPUID2_AVX | CPUID2_XSAVE)) == 0 && (amd_feature2 & (AMDID2_XOP |
2458 	    AMDID2_FMA4)) == 0)
2459 		workaround_erratum383 = 1;
2460 
2461 	/*
2462 	 * Are large page mappings enabled?
2463 	 */
2464 	TUNABLE_INT_FETCH("vm.pmap.pg_ps_enabled", &pg_ps_enabled);
2465 	if (pg_ps_enabled) {
2466 		KASSERT(MAXPAGESIZES > 1 && pagesizes[1] == 0,
2467 		    ("pmap_init: can't assign to pagesizes[1]"));
2468 		pagesizes[1] = NBPDR;
2469 		if ((amd_feature & AMDID_PAGE1GB) != 0) {
2470 			KASSERT(MAXPAGESIZES > 2 && pagesizes[2] == 0,
2471 			    ("pmap_init: can't assign to pagesizes[2]"));
2472 			pagesizes[2] = NBPDP;
2473 		}
2474 	}
2475 
2476 	/*
2477 	 * Initialize pv chunk lists.
2478 	 */
2479 	for (i = 0; i < PMAP_MEMDOM; i++) {
2480 		mtx_init(&pv_chunks[i].pvc_lock, "pmap pv chunk list", NULL, MTX_DEF);
2481 		TAILQ_INIT(&pv_chunks[i].pvc_list);
2482 	}
2483 	pmap_init_pv_table();
2484 
2485 	pmap_initialized = 1;
2486 	for (i = 0; i < PMAP_PREINIT_MAPPING_COUNT; i++) {
2487 		ppim = pmap_preinit_mapping + i;
2488 		if (ppim->va == 0)
2489 			continue;
2490 		/* Make the direct map consistent */
2491 		if (ppim->pa < dmaplimit && ppim->pa + ppim->sz <= dmaplimit) {
2492 			(void)pmap_change_attr(PHYS_TO_DMAP(ppim->pa),
2493 			    ppim->sz, ppim->mode);
2494 		}
2495 		if (!bootverbose)
2496 			continue;
2497 		printf("PPIM %u: PA=%#lx, VA=%#lx, size=%#lx, mode=%#x\n", i,
2498 		    ppim->pa, ppim->va, ppim->sz, ppim->mode);
2499 	}
2500 
2501 	mtx_init(&qframe_mtx, "qfrmlk", NULL, MTX_SPIN);
2502 	error = vmem_alloc(kernel_arena, PAGE_SIZE, M_BESTFIT | M_WAITOK,
2503 	    (vmem_addr_t *)&qframe);
2504 	if (error != 0)
2505 		panic("qframe allocation failed");
2506 
2507 	lm_ents = 8;
2508 	TUNABLE_INT_FETCH("vm.pmap.large_map_pml4_entries", &lm_ents);
2509 	if (lm_ents > LMEPML4I - LMSPML4I + 1)
2510 		lm_ents = LMEPML4I - LMSPML4I + 1;
2511 	if (bootverbose)
2512 		printf("pmap: large map %u PML4 slots (%lu GB)\n",
2513 		    lm_ents, (u_long)lm_ents * (NBPML4 / 1024 / 1024 / 1024));
2514 	if (lm_ents != 0) {
2515 		large_vmem = vmem_create("large", LARGEMAP_MIN_ADDRESS,
2516 		    (vmem_size_t)lm_ents * NBPML4, PAGE_SIZE, 0, M_WAITOK);
2517 		if (large_vmem == NULL) {
2518 			printf("pmap: cannot create large map\n");
2519 			lm_ents = 0;
2520 		}
2521 		for (i = 0; i < lm_ents; i++) {
2522 			m = pmap_large_map_getptp_unlocked();
2523 			/* XXXKIB la57 */
2524 			kernel_pml4[LMSPML4I + i] = X86_PG_V |
2525 			    X86_PG_RW | X86_PG_A | X86_PG_M | pg_nx |
2526 			    VM_PAGE_TO_PHYS(m);
2527 		}
2528 	}
2529 }
2530 
2531 SYSCTL_UINT(_vm_pmap, OID_AUTO, large_map_pml4_entries,
2532     CTLFLAG_RDTUN | CTLFLAG_NOFETCH, &lm_ents, 0,
2533     "Maximum number of PML4 entries for use by large map (tunable).  "
2534     "Each entry corresponds to 512GB of address space.");
2535 
2536 static SYSCTL_NODE(_vm_pmap, OID_AUTO, pde, CTLFLAG_RD | CTLFLAG_MPSAFE, 0,
2537     "2MB page mapping counters");
2538 
2539 static COUNTER_U64_DEFINE_EARLY(pmap_pde_demotions);
2540 SYSCTL_COUNTER_U64(_vm_pmap_pde, OID_AUTO, demotions,
2541     CTLFLAG_RD, &pmap_pde_demotions, "2MB page demotions");
2542 
2543 static COUNTER_U64_DEFINE_EARLY(pmap_pde_mappings);
2544 SYSCTL_COUNTER_U64(_vm_pmap_pde, OID_AUTO, mappings, CTLFLAG_RD,
2545     &pmap_pde_mappings, "2MB page mappings");
2546 
2547 static COUNTER_U64_DEFINE_EARLY(pmap_pde_p_failures);
2548 SYSCTL_COUNTER_U64(_vm_pmap_pde, OID_AUTO, p_failures, CTLFLAG_RD,
2549     &pmap_pde_p_failures, "2MB page promotion failures");
2550 
2551 static COUNTER_U64_DEFINE_EARLY(pmap_pde_promotions);
2552 SYSCTL_COUNTER_U64(_vm_pmap_pde, OID_AUTO, promotions, CTLFLAG_RD,
2553     &pmap_pde_promotions, "2MB page promotions");
2554 
2555 static SYSCTL_NODE(_vm_pmap, OID_AUTO, pdpe, CTLFLAG_RD | CTLFLAG_MPSAFE, 0,
2556     "1GB page mapping counters");
2557 
2558 static COUNTER_U64_DEFINE_EARLY(pmap_pdpe_demotions);
2559 SYSCTL_COUNTER_U64(_vm_pmap_pdpe, OID_AUTO, demotions, CTLFLAG_RD,
2560     &pmap_pdpe_demotions, "1GB page demotions");
2561 
2562 /***************************************************
2563  * Low level helper routines.....
2564  ***************************************************/
2565 
2566 static pt_entry_t
pmap_swap_pat(pmap_t pmap,pt_entry_t entry)2567 pmap_swap_pat(pmap_t pmap, pt_entry_t entry)
2568 {
2569 	int x86_pat_bits = X86_PG_PTE_PAT | X86_PG_PDE_PAT;
2570 
2571 	switch (pmap->pm_type) {
2572 	case PT_X86:
2573 	case PT_RVI:
2574 		/* Verify that both PAT bits are not set at the same time */
2575 		KASSERT((entry & x86_pat_bits) != x86_pat_bits,
2576 		    ("Invalid PAT bits in entry %#lx", entry));
2577 
2578 		/* Swap the PAT bits if one of them is set */
2579 		if ((entry & x86_pat_bits) != 0)
2580 			entry ^= x86_pat_bits;
2581 		break;
2582 	case PT_EPT:
2583 		/*
2584 		 * Nothing to do - the memory attributes are represented
2585 		 * the same way for regular pages and superpages.
2586 		 */
2587 		break;
2588 	default:
2589 		panic("pmap_switch_pat_bits: bad pm_type %d", pmap->pm_type);
2590 	}
2591 
2592 	return (entry);
2593 }
2594 
2595 boolean_t
pmap_is_valid_memattr(pmap_t pmap __unused,vm_memattr_t mode)2596 pmap_is_valid_memattr(pmap_t pmap __unused, vm_memattr_t mode)
2597 {
2598 
2599 	return (mode >= 0 && mode < PAT_INDEX_SIZE &&
2600 	    pat_index[(int)mode] >= 0);
2601 }
2602 
2603 /*
2604  * Determine the appropriate bits to set in a PTE or PDE for a specified
2605  * caching mode.
2606  */
2607 int
pmap_cache_bits(pmap_t pmap,int mode,boolean_t is_pde)2608 pmap_cache_bits(pmap_t pmap, int mode, boolean_t is_pde)
2609 {
2610 	int cache_bits, pat_flag, pat_idx;
2611 
2612 	if (!pmap_is_valid_memattr(pmap, mode))
2613 		panic("Unknown caching mode %d\n", mode);
2614 
2615 	switch (pmap->pm_type) {
2616 	case PT_X86:
2617 	case PT_RVI:
2618 		/* The PAT bit is different for PTE's and PDE's. */
2619 		pat_flag = is_pde ? X86_PG_PDE_PAT : X86_PG_PTE_PAT;
2620 
2621 		/* Map the caching mode to a PAT index. */
2622 		pat_idx = pat_index[mode];
2623 
2624 		/* Map the 3-bit index value into the PAT, PCD, and PWT bits. */
2625 		cache_bits = 0;
2626 		if (pat_idx & 0x4)
2627 			cache_bits |= pat_flag;
2628 		if (pat_idx & 0x2)
2629 			cache_bits |= PG_NC_PCD;
2630 		if (pat_idx & 0x1)
2631 			cache_bits |= PG_NC_PWT;
2632 		break;
2633 
2634 	case PT_EPT:
2635 		cache_bits = EPT_PG_IGNORE_PAT | EPT_PG_MEMORY_TYPE(mode);
2636 		break;
2637 
2638 	default:
2639 		panic("unsupported pmap type %d", pmap->pm_type);
2640 	}
2641 
2642 	return (cache_bits);
2643 }
2644 
2645 static int
pmap_cache_mask(pmap_t pmap,boolean_t is_pde)2646 pmap_cache_mask(pmap_t pmap, boolean_t is_pde)
2647 {
2648 	int mask;
2649 
2650 	switch (pmap->pm_type) {
2651 	case PT_X86:
2652 	case PT_RVI:
2653 		mask = is_pde ? X86_PG_PDE_CACHE : X86_PG_PTE_CACHE;
2654 		break;
2655 	case PT_EPT:
2656 		mask = EPT_PG_IGNORE_PAT | EPT_PG_MEMORY_TYPE(0x7);
2657 		break;
2658 	default:
2659 		panic("pmap_cache_mask: invalid pm_type %d", pmap->pm_type);
2660 	}
2661 
2662 	return (mask);
2663 }
2664 
2665 static int
pmap_pat_index(pmap_t pmap,pt_entry_t pte,bool is_pde)2666 pmap_pat_index(pmap_t pmap, pt_entry_t pte, bool is_pde)
2667 {
2668 	int pat_flag, pat_idx;
2669 
2670 	pat_idx = 0;
2671 	switch (pmap->pm_type) {
2672 	case PT_X86:
2673 	case PT_RVI:
2674 		/* The PAT bit is different for PTE's and PDE's. */
2675 		pat_flag = is_pde ? X86_PG_PDE_PAT : X86_PG_PTE_PAT;
2676 
2677 		if ((pte & pat_flag) != 0)
2678 			pat_idx |= 0x4;
2679 		if ((pte & PG_NC_PCD) != 0)
2680 			pat_idx |= 0x2;
2681 		if ((pte & PG_NC_PWT) != 0)
2682 			pat_idx |= 0x1;
2683 		break;
2684 	case PT_EPT:
2685 		if ((pte & EPT_PG_IGNORE_PAT) != 0)
2686 			panic("EPT PTE %#lx has no PAT memory type", pte);
2687 		pat_idx = (pte & EPT_PG_MEMORY_TYPE(0x7)) >> 3;
2688 		break;
2689 	}
2690 
2691 	/* See pmap_init_pat(). */
2692 	if (pat_idx == 4)
2693 		pat_idx = 0;
2694 	if (pat_idx == 7)
2695 		pat_idx = 3;
2696 
2697 	return (pat_idx);
2698 }
2699 
2700 bool
pmap_ps_enabled(pmap_t pmap)2701 pmap_ps_enabled(pmap_t pmap)
2702 {
2703 
2704 	return (pg_ps_enabled && (pmap->pm_flags & PMAP_PDE_SUPERPAGE) != 0);
2705 }
2706 
2707 static void
pmap_update_pde_store(pmap_t pmap,pd_entry_t * pde,pd_entry_t newpde)2708 pmap_update_pde_store(pmap_t pmap, pd_entry_t *pde, pd_entry_t newpde)
2709 {
2710 
2711 	switch (pmap->pm_type) {
2712 	case PT_X86:
2713 		break;
2714 	case PT_RVI:
2715 	case PT_EPT:
2716 		/*
2717 		 * XXX
2718 		 * This is a little bogus since the generation number is
2719 		 * supposed to be bumped up when a region of the address
2720 		 * space is invalidated in the page tables.
2721 		 *
2722 		 * In this case the old PDE entry is valid but yet we want
2723 		 * to make sure that any mappings using the old entry are
2724 		 * invalidated in the TLB.
2725 		 *
2726 		 * The reason this works as expected is because we rendezvous
2727 		 * "all" host cpus and force any vcpu context to exit as a
2728 		 * side-effect.
2729 		 */
2730 		atomic_add_long(&pmap->pm_eptgen, 1);
2731 		break;
2732 	default:
2733 		panic("pmap_update_pde_store: bad pm_type %d", pmap->pm_type);
2734 	}
2735 	pde_store(pde, newpde);
2736 }
2737 
2738 /*
2739  * After changing the page size for the specified virtual address in the page
2740  * table, flush the corresponding entries from the processor's TLB.  Only the
2741  * calling processor's TLB is affected.
2742  *
2743  * The calling thread must be pinned to a processor.
2744  */
2745 static void
pmap_update_pde_invalidate(pmap_t pmap,vm_offset_t va,pd_entry_t newpde)2746 pmap_update_pde_invalidate(pmap_t pmap, vm_offset_t va, pd_entry_t newpde)
2747 {
2748 	pt_entry_t PG_G;
2749 
2750 	if (pmap_type_guest(pmap))
2751 		return;
2752 
2753 	KASSERT(pmap->pm_type == PT_X86,
2754 	    ("pmap_update_pde_invalidate: invalid type %d", pmap->pm_type));
2755 
2756 	PG_G = pmap_global_bit(pmap);
2757 
2758 	if ((newpde & PG_PS) == 0)
2759 		/* Demotion: flush a specific 2MB page mapping. */
2760 		pmap_invlpg(pmap, va);
2761 	else if ((newpde & PG_G) == 0)
2762 		/*
2763 		 * Promotion: flush every 4KB page mapping from the TLB
2764 		 * because there are too many to flush individually.
2765 		 */
2766 		invltlb();
2767 	else {
2768 		/*
2769 		 * Promotion: flush every 4KB page mapping from the TLB,
2770 		 * including any global (PG_G) mappings.
2771 		 */
2772 		invltlb_glob();
2773 	}
2774 }
2775 
2776 /*
2777  * The amd64 pmap uses different approaches to TLB invalidation
2778  * depending on the kernel configuration, available hardware features,
2779  * and known hardware errata.  The kernel configuration option that
2780  * has the greatest operational impact on TLB invalidation is PTI,
2781  * which is enabled automatically on affected Intel CPUs.  The most
2782  * impactful hardware features are first PCID, and then INVPCID
2783  * instruction presence.  PCID usage is quite different for PTI
2784  * vs. non-PTI.
2785  *
2786  * * Kernel Page Table Isolation (PTI or KPTI) is used to mitigate
2787  *   the Meltdown bug in some Intel CPUs.  Under PTI, each user address
2788  *   space is served by two page tables, user and kernel.  The user
2789  *   page table only maps user space and a kernel trampoline.  The
2790  *   kernel trampoline includes the entirety of the kernel text but
2791  *   only the kernel data that is needed to switch from user to kernel
2792  *   mode.  The kernel page table maps the user and kernel address
2793  *   spaces in their entirety.  It is identical to the per-process
2794  *   page table used in non-PTI mode.
2795  *
2796  *   User page tables are only used when the CPU is in user mode.
2797  *   Consequently, some TLB invalidations can be postponed until the
2798  *   switch from kernel to user mode.  In contrast, the user
2799  *   space part of the kernel page table is used for copyout(9), so
2800  *   TLB invalidations on this page table cannot be similarly postponed.
2801  *
2802  *   The existence of a user mode page table for the given pmap is
2803  *   indicated by a pm_ucr3 value that differs from PMAP_NO_CR3, in
2804  *   which case pm_ucr3 contains the %cr3 register value for the user
2805  *   mode page table's root.
2806  *
2807  * * The pm_active bitmask indicates which CPUs currently have the
2808  *   pmap active.  A CPU's bit is set on context switch to the pmap, and
2809  *   cleared on switching off this CPU.  For the kernel page table,
2810  *   the pm_active field is immutable and contains all CPUs.  The
2811  *   kernel page table is always logically active on every processor,
2812  *   but not necessarily in use by the hardware, e.g., in PTI mode.
2813  *
2814  *   When requesting invalidation of virtual addresses with
2815  *   pmap_invalidate_XXX() functions, the pmap sends shootdown IPIs to
2816  *   all CPUs recorded as active in pm_active.  Updates to and reads
2817  *   from pm_active are not synchronized, and so they may race with
2818  *   each other.  Shootdown handlers are prepared to handle the race.
2819  *
2820  * * PCID is an optional feature of the long mode x86 MMU where TLB
2821  *   entries are tagged with the 'Process ID' of the address space
2822  *   they belong to.  This feature provides a limited namespace for
2823  *   process identifiers, 12 bits, supporting 4095 simultaneous IDs
2824  *   total.
2825  *
2826  *   Allocation of a PCID to a pmap is done by an algorithm described
2827  *   in section 15.12, "Other TLB Consistency Algorithms", of
2828  *   Vahalia's book "Unix Internals".  A PCID cannot be allocated for
2829  *   the whole lifetime of a pmap in pmap_pinit() due to the limited
2830  *   namespace.  Instead, a per-CPU, per-pmap PCID is assigned when
2831  *   the CPU is about to start caching TLB entries from a pmap,
2832  *   i.e., on the context switch that activates the pmap on the CPU.
2833  *
2834  *   The PCID allocator maintains a per-CPU, per-pmap generation
2835  *   count, pm_gen, which is incremented each time a new PCID is
2836  *   allocated.  On TLB invalidation, the generation counters for the
2837  *   pmap are zeroed, which signals the context switch code that the
2838  *   previously allocated PCID is no longer valid.  Effectively,
2839  *   zeroing any of these counters triggers a TLB shootdown for the
2840  *   given CPU/address space, due to the allocation of a new PCID.
2841  *
2842  *   Zeroing can be performed remotely.  Consequently, if a pmap is
2843  *   inactive on a CPU, then a TLB shootdown for that pmap and CPU can
2844  *   be initiated by an ordinary memory access to reset the target
2845  *   CPU's generation count within the pmap.  The CPU initiating the
2846  *   TLB shootdown does not need to send an IPI to the target CPU.
2847  *
2848  * * PTI + PCID.  The available PCIDs are divided into two sets: PCIDs
2849  *   for complete (kernel) page tables, and PCIDs for user mode page
2850  *   tables.  A user PCID value is obtained from the kernel PCID value
2851  *   by setting the highest bit, 11, to 1 (0x800 == PMAP_PCID_USER_PT).
2852  *
2853  *   User space page tables are activated on return to user mode, by
2854  *   loading pm_ucr3 into %cr3.  If the PCPU(ucr3_load_mask) requests
2855  *   clearing bit 63 of the loaded ucr3, this effectively causes
2856  *   complete invalidation of the user mode TLB entries for the
2857  *   current pmap.  In which case, local invalidations of individual
2858  *   pages in the user page table are skipped.
2859  *
2860  * * Local invalidation, all modes.  If the requested invalidation is
2861  *   for a specific address or the total invalidation of a currently
2862  *   active pmap, then the TLB is flushed using INVLPG for a kernel
2863  *   page table, and INVPCID(INVPCID_CTXGLOB)/invltlb_glob() for a
2864  *   user space page table(s).
2865  *
2866  *   If the INVPCID instruction is available, it is used to flush user
2867  *   entries from the kernel page table.
2868  *
2869  *   When PCID is enabled, the INVLPG instruction invalidates all TLB
2870  *   entries for the given page that either match the current PCID or
2871  *   are global. Since TLB entries for the same page under different
2872  *   PCIDs are unaffected, kernel pages which reside in all address
2873  *   spaces could be problematic.  We avoid the problem by creating
2874  *   all kernel PTEs with the global flag (PG_G) set, when PTI is
2875  *   disabled.
2876  *
2877  * * mode: PTI disabled, PCID present.  The kernel reserves PCID 0 for its
2878  *   address space, all other 4095 PCIDs are used for user mode spaces
2879  *   as described above.  A context switch allocates a new PCID if
2880  *   the recorded PCID is zero or the recorded generation does not match
2881  *   the CPU's generation, effectively flushing the TLB for this address space.
2882  *   Total remote invalidation is performed by zeroing pm_gen for all CPUs.
2883  *	local user page: INVLPG
2884  *	local kernel page: INVLPG
2885  *	local user total: INVPCID(CTX)
2886  *	local kernel total: INVPCID(CTXGLOB) or invltlb_glob()
2887  *	remote user page, inactive pmap: zero pm_gen
2888  *	remote user page, active pmap: zero pm_gen + IPI:INVLPG
2889  *	(Both actions are required to handle the aforementioned pm_active races.)
2890  *	remote kernel page: IPI:INVLPG
2891  *	remote user total, inactive pmap: zero pm_gen
2892  *	remote user total, active pmap: zero pm_gen + IPI:(INVPCID(CTX) or
2893  *          reload %cr3)
2894  *	(See note above about pm_active races.)
2895  *	remote kernel total: IPI:(INVPCID(CTXGLOB) or invltlb_glob())
2896  *
2897  * PTI enabled, PCID present.
2898  *	local user page: INVLPG for kpt, INVPCID(ADDR) or (INVLPG for ucr3)
2899  *          for upt
2900  *	local kernel page: INVLPG
2901  *	local user total: INVPCID(CTX) or reload %cr3 for kpt, clear PCID_SAVE
2902  *          on loading UCR3 into %cr3 for upt
2903  *	local kernel total: INVPCID(CTXGLOB) or invltlb_glob()
2904  *	remote user page, inactive pmap: zero pm_gen
2905  *	remote user page, active pmap: zero pm_gen + IPI:(INVLPG for kpt,
2906  *          INVPCID(ADDR) for upt)
2907  *	remote kernel page: IPI:INVLPG
2908  *	remote user total, inactive pmap: zero pm_gen
2909  *	remote user total, active pmap: zero pm_gen + IPI:(INVPCID(CTX) for kpt,
2910  *          clear PCID_SAVE on loading UCR3 into $cr3 for upt)
2911  *	remote kernel total: IPI:(INVPCID(CTXGLOB) or invltlb_glob())
2912  *
2913  *  No PCID.
2914  *	local user page: INVLPG
2915  *	local kernel page: INVLPG
2916  *	local user total: reload %cr3
2917  *	local kernel total: invltlb_glob()
2918  *	remote user page, inactive pmap: -
2919  *	remote user page, active pmap: IPI:INVLPG
2920  *	remote kernel page: IPI:INVLPG
2921  *	remote user total, inactive pmap: -
2922  *	remote user total, active pmap: IPI:(reload %cr3)
2923  *	remote kernel total: IPI:invltlb_glob()
2924  *  Since on return to user mode, the reload of %cr3 with ucr3 causes
2925  *  TLB invalidation, no specific action is required for user page table.
2926  *
2927  * EPT.  EPT pmaps do not map KVA, all mappings are userspace.
2928  * XXX TODO
2929  */
2930 
2931 #ifdef SMP
2932 /*
2933  * Interrupt the cpus that are executing in the guest context.
2934  * This will force the vcpu to exit and the cached EPT mappings
2935  * will be invalidated by the host before the next vmresume.
2936  */
2937 static __inline void
pmap_invalidate_ept(pmap_t pmap)2938 pmap_invalidate_ept(pmap_t pmap)
2939 {
2940 	smr_seq_t goal;
2941 	int ipinum;
2942 
2943 	sched_pin();
2944 	KASSERT(!CPU_ISSET(curcpu, &pmap->pm_active),
2945 	    ("pmap_invalidate_ept: absurd pm_active"));
2946 
2947 	/*
2948 	 * The TLB mappings associated with a vcpu context are not
2949 	 * flushed each time a different vcpu is chosen to execute.
2950 	 *
2951 	 * This is in contrast with a process's vtop mappings that
2952 	 * are flushed from the TLB on each context switch.
2953 	 *
2954 	 * Therefore we need to do more than just a TLB shootdown on
2955 	 * the active cpus in 'pmap->pm_active'. To do this we keep
2956 	 * track of the number of invalidations performed on this pmap.
2957 	 *
2958 	 * Each vcpu keeps a cache of this counter and compares it
2959 	 * just before a vmresume. If the counter is out-of-date an
2960 	 * invept will be done to flush stale mappings from the TLB.
2961 	 *
2962 	 * To ensure that all vCPU threads have observed the new counter
2963 	 * value before returning, we use SMR.  Ordering is important here:
2964 	 * the VMM enters an SMR read section before loading the counter
2965 	 * and after updating the pm_active bit set.  Thus, pm_active is
2966 	 * a superset of active readers, and any reader that has observed
2967 	 * the goal has observed the new counter value.
2968 	 */
2969 	atomic_add_long(&pmap->pm_eptgen, 1);
2970 
2971 	goal = smr_advance(pmap->pm_eptsmr);
2972 
2973 	/*
2974 	 * Force the vcpu to exit and trap back into the hypervisor.
2975 	 */
2976 	ipinum = pmap->pm_flags & PMAP_NESTED_IPIMASK;
2977 	ipi_selected(pmap->pm_active, ipinum);
2978 	sched_unpin();
2979 
2980 	/*
2981 	 * Ensure that all active vCPUs will observe the new generation counter
2982 	 * value before executing any more guest instructions.
2983 	 */
2984 	smr_wait(pmap->pm_eptsmr, goal);
2985 }
2986 
2987 static inline void
pmap_invalidate_preipi_pcid(pmap_t pmap)2988 pmap_invalidate_preipi_pcid(pmap_t pmap)
2989 {
2990 	u_int cpuid, i;
2991 
2992 	sched_pin();
2993 
2994 	cpuid = PCPU_GET(cpuid);
2995 	if (pmap != PCPU_GET(curpmap))
2996 		cpuid = 0xffffffff;	/* An impossible value */
2997 
2998 	CPU_FOREACH(i) {
2999 		if (cpuid != i)
3000 			pmap->pm_pcids[i].pm_gen = 0;
3001 	}
3002 
3003 	/*
3004 	 * The fence is between stores to pm_gen and the read of the
3005 	 * pm_active mask.  We need to ensure that it is impossible
3006 	 * for us to miss the bit update in pm_active and
3007 	 * simultaneously observe a non-zero pm_gen in
3008 	 * pmap_activate_sw(), otherwise TLB update is missed.
3009 	 * Without the fence, IA32 allows such an outcome.  Note that
3010 	 * pm_active is updated by a locked operation, which provides
3011 	 * the reciprocal fence.
3012 	 */
3013 	atomic_thread_fence_seq_cst();
3014 }
3015 
3016 static void
pmap_invalidate_preipi_nopcid(pmap_t pmap __unused)3017 pmap_invalidate_preipi_nopcid(pmap_t pmap __unused)
3018 {
3019 	sched_pin();
3020 }
3021 
3022 DEFINE_IFUNC(static, void, pmap_invalidate_preipi, (pmap_t))
3023 {
3024 	return (pmap_pcid_enabled ? pmap_invalidate_preipi_pcid :
3025 	    pmap_invalidate_preipi_nopcid);
3026 }
3027 
3028 static inline void
pmap_invalidate_page_pcid_cb(pmap_t pmap,vm_offset_t va,const bool invpcid_works1)3029 pmap_invalidate_page_pcid_cb(pmap_t pmap, vm_offset_t va,
3030     const bool invpcid_works1)
3031 {
3032 	struct invpcid_descr d;
3033 	uint64_t kcr3, ucr3;
3034 	uint32_t pcid;
3035 	u_int cpuid;
3036 
3037 	/*
3038 	 * Because pm_pcid is recalculated on a context switch, we
3039 	 * must ensure there is no preemption, not just pinning.
3040 	 * Otherwise, we might use a stale value below.
3041 	 */
3042 	CRITICAL_ASSERT(curthread);
3043 
3044 	/*
3045 	 * No need to do anything with user page tables invalidation
3046 	 * if there is no user page table, or invalidation is deferred
3047 	 * until the return to userspace.  ucr3_load_mask is stable
3048 	 * because we have preemption disabled.
3049 	 */
3050 	if (pmap->pm_ucr3 == PMAP_NO_CR3 ||
3051 	    PCPU_GET(ucr3_load_mask) != PMAP_UCR3_NOMASK)
3052 		return;
3053 
3054 	cpuid = PCPU_GET(cpuid);
3055 
3056 	pcid = pmap->pm_pcids[cpuid].pm_pcid;
3057 	if (invpcid_works1) {
3058 		d.pcid = pcid | PMAP_PCID_USER_PT;
3059 		d.pad = 0;
3060 		d.addr = va;
3061 		invpcid(&d, INVPCID_ADDR);
3062 	} else {
3063 		kcr3 = pmap->pm_cr3 | pcid | CR3_PCID_SAVE;
3064 		ucr3 = pmap->pm_ucr3 | pcid | PMAP_PCID_USER_PT | CR3_PCID_SAVE;
3065 		pmap_pti_pcid_invlpg(ucr3, kcr3, va);
3066 	}
3067 }
3068 
3069 static void
pmap_invalidate_page_pcid_invpcid_cb(pmap_t pmap,vm_offset_t va)3070 pmap_invalidate_page_pcid_invpcid_cb(pmap_t pmap, vm_offset_t va)
3071 {
3072 	pmap_invalidate_page_pcid_cb(pmap, va, true);
3073 }
3074 
3075 static void
pmap_invalidate_page_pcid_noinvpcid_cb(pmap_t pmap,vm_offset_t va)3076 pmap_invalidate_page_pcid_noinvpcid_cb(pmap_t pmap, vm_offset_t va)
3077 {
3078 	pmap_invalidate_page_pcid_cb(pmap, va, false);
3079 }
3080 
3081 static void
pmap_invalidate_page_nopcid_cb(pmap_t pmap __unused,vm_offset_t va __unused)3082 pmap_invalidate_page_nopcid_cb(pmap_t pmap __unused, vm_offset_t va __unused)
3083 {
3084 }
3085 
3086 DEFINE_IFUNC(static, void, pmap_invalidate_page_cb, (pmap_t, vm_offset_t))
3087 {
3088 	if (pmap_pcid_enabled)
3089 		return (invpcid_works ? pmap_invalidate_page_pcid_invpcid_cb :
3090 		    pmap_invalidate_page_pcid_noinvpcid_cb);
3091 	return (pmap_invalidate_page_nopcid_cb);
3092 }
3093 
3094 static void
pmap_invalidate_page_curcpu_cb(pmap_t pmap,vm_offset_t va,vm_offset_t addr2 __unused)3095 pmap_invalidate_page_curcpu_cb(pmap_t pmap, vm_offset_t va,
3096     vm_offset_t addr2 __unused)
3097 {
3098 	if (pmap == kernel_pmap) {
3099 		pmap_invlpg(kernel_pmap, va);
3100 	} else if (pmap == PCPU_GET(curpmap)) {
3101 		invlpg(va);
3102 		pmap_invalidate_page_cb(pmap, va);
3103 	}
3104 }
3105 
3106 void
pmap_invalidate_page(pmap_t pmap,vm_offset_t va)3107 pmap_invalidate_page(pmap_t pmap, vm_offset_t va)
3108 {
3109 	if (pmap_type_guest(pmap)) {
3110 		pmap_invalidate_ept(pmap);
3111 		return;
3112 	}
3113 
3114 	KASSERT(pmap->pm_type == PT_X86,
3115 	    ("pmap_invalidate_page: invalid type %d", pmap->pm_type));
3116 
3117 	pmap_invalidate_preipi(pmap);
3118 	smp_masked_invlpg(va, pmap, pmap_invalidate_page_curcpu_cb);
3119 }
3120 
3121 /* 4k PTEs -- Chosen to exceed the total size of Broadwell L2 TLB */
3122 #define	PMAP_INVLPG_THRESHOLD	(4 * 1024 * PAGE_SIZE)
3123 
3124 static void
pmap_invalidate_range_pcid_cb(pmap_t pmap,vm_offset_t sva,vm_offset_t eva,const bool invpcid_works1)3125 pmap_invalidate_range_pcid_cb(pmap_t pmap, vm_offset_t sva, vm_offset_t eva,
3126     const bool invpcid_works1)
3127 {
3128 	struct invpcid_descr d;
3129 	uint64_t kcr3, ucr3;
3130 	uint32_t pcid;
3131 	u_int cpuid;
3132 
3133 	CRITICAL_ASSERT(curthread);
3134 
3135 	if (pmap != PCPU_GET(curpmap) ||
3136 	    pmap->pm_ucr3 == PMAP_NO_CR3 ||
3137 	    PCPU_GET(ucr3_load_mask) != PMAP_UCR3_NOMASK)
3138 		return;
3139 
3140 	cpuid = PCPU_GET(cpuid);
3141 
3142 	pcid = pmap->pm_pcids[cpuid].pm_pcid;
3143 	if (invpcid_works1) {
3144 		d.pcid = pcid | PMAP_PCID_USER_PT;
3145 		d.pad = 0;
3146 		for (d.addr = sva; d.addr < eva; d.addr += PAGE_SIZE)
3147 			invpcid(&d, INVPCID_ADDR);
3148 	} else {
3149 		kcr3 = pmap->pm_cr3 | pcid | CR3_PCID_SAVE;
3150 		ucr3 = pmap->pm_ucr3 | pcid | PMAP_PCID_USER_PT | CR3_PCID_SAVE;
3151 		pmap_pti_pcid_invlrng(ucr3, kcr3, sva, eva);
3152 	}
3153 }
3154 
3155 static void
pmap_invalidate_range_pcid_invpcid_cb(pmap_t pmap,vm_offset_t sva,vm_offset_t eva)3156 pmap_invalidate_range_pcid_invpcid_cb(pmap_t pmap, vm_offset_t sva,
3157     vm_offset_t eva)
3158 {
3159 	pmap_invalidate_range_pcid_cb(pmap, sva, eva, true);
3160 }
3161 
3162 static void
pmap_invalidate_range_pcid_noinvpcid_cb(pmap_t pmap,vm_offset_t sva,vm_offset_t eva)3163 pmap_invalidate_range_pcid_noinvpcid_cb(pmap_t pmap, vm_offset_t sva,
3164     vm_offset_t eva)
3165 {
3166 	pmap_invalidate_range_pcid_cb(pmap, sva, eva, false);
3167 }
3168 
3169 static void
pmap_invalidate_range_nopcid_cb(pmap_t pmap __unused,vm_offset_t sva __unused,vm_offset_t eva __unused)3170 pmap_invalidate_range_nopcid_cb(pmap_t pmap __unused, vm_offset_t sva __unused,
3171     vm_offset_t eva __unused)
3172 {
3173 }
3174 
3175 DEFINE_IFUNC(static, void, pmap_invalidate_range_cb, (pmap_t, vm_offset_t,
3176     vm_offset_t))
3177 {
3178 	if (pmap_pcid_enabled)
3179 		return (invpcid_works ? pmap_invalidate_range_pcid_invpcid_cb :
3180 		    pmap_invalidate_range_pcid_noinvpcid_cb);
3181 	return (pmap_invalidate_range_nopcid_cb);
3182 }
3183 
3184 static void
pmap_invalidate_range_curcpu_cb(pmap_t pmap,vm_offset_t sva,vm_offset_t eva)3185 pmap_invalidate_range_curcpu_cb(pmap_t pmap, vm_offset_t sva, vm_offset_t eva)
3186 {
3187 	vm_offset_t addr;
3188 
3189 	if (pmap == kernel_pmap) {
3190 		if (PCPU_GET(pcid_invlpg_workaround)) {
3191 			struct invpcid_descr d = { 0 };
3192 
3193 			invpcid(&d, INVPCID_CTXGLOB);
3194 		} else {
3195 			for (addr = sva; addr < eva; addr += PAGE_SIZE)
3196 				invlpg(addr);
3197 		}
3198 	} else if (pmap == PCPU_GET(curpmap)) {
3199 		for (addr = sva; addr < eva; addr += PAGE_SIZE)
3200 			invlpg(addr);
3201 		pmap_invalidate_range_cb(pmap, sva, eva);
3202 	}
3203 }
3204 
3205 void
pmap_invalidate_range(pmap_t pmap,vm_offset_t sva,vm_offset_t eva)3206 pmap_invalidate_range(pmap_t pmap, vm_offset_t sva, vm_offset_t eva)
3207 {
3208 	if (eva - sva >= PMAP_INVLPG_THRESHOLD) {
3209 		pmap_invalidate_all(pmap);
3210 		return;
3211 	}
3212 
3213 	if (pmap_type_guest(pmap)) {
3214 		pmap_invalidate_ept(pmap);
3215 		return;
3216 	}
3217 
3218 	KASSERT(pmap->pm_type == PT_X86,
3219 	    ("pmap_invalidate_range: invalid type %d", pmap->pm_type));
3220 
3221 	pmap_invalidate_preipi(pmap);
3222 	smp_masked_invlpg_range(sva, eva, pmap,
3223 	    pmap_invalidate_range_curcpu_cb);
3224 }
3225 
3226 static inline void
pmap_invalidate_all_pcid_cb(pmap_t pmap,bool invpcid_works1)3227 pmap_invalidate_all_pcid_cb(pmap_t pmap, bool invpcid_works1)
3228 {
3229 	struct invpcid_descr d;
3230 	uint64_t kcr3;
3231 	uint32_t pcid;
3232 	u_int cpuid;
3233 
3234 	if (pmap == kernel_pmap) {
3235 		if (invpcid_works1) {
3236 			bzero(&d, sizeof(d));
3237 			invpcid(&d, INVPCID_CTXGLOB);
3238 		} else {
3239 			invltlb_glob();
3240 		}
3241 	} else if (pmap == PCPU_GET(curpmap)) {
3242 		CRITICAL_ASSERT(curthread);
3243 		cpuid = PCPU_GET(cpuid);
3244 
3245 		pcid = pmap->pm_pcids[cpuid].pm_pcid;
3246 		if (invpcid_works1) {
3247 			d.pcid = pcid;
3248 			d.pad = 0;
3249 			d.addr = 0;
3250 			invpcid(&d, INVPCID_CTX);
3251 		} else {
3252 			kcr3 = pmap->pm_cr3 | pcid;
3253 			load_cr3(kcr3);
3254 		}
3255 		if (pmap->pm_ucr3 != PMAP_NO_CR3)
3256 			PCPU_SET(ucr3_load_mask, ~CR3_PCID_SAVE);
3257 	}
3258 }
3259 
3260 static void
pmap_invalidate_all_pcid_invpcid_cb(pmap_t pmap)3261 pmap_invalidate_all_pcid_invpcid_cb(pmap_t pmap)
3262 {
3263 	pmap_invalidate_all_pcid_cb(pmap, true);
3264 }
3265 
3266 static void
pmap_invalidate_all_pcid_noinvpcid_cb(pmap_t pmap)3267 pmap_invalidate_all_pcid_noinvpcid_cb(pmap_t pmap)
3268 {
3269 	pmap_invalidate_all_pcid_cb(pmap, false);
3270 }
3271 
3272 static void
pmap_invalidate_all_nopcid_cb(pmap_t pmap)3273 pmap_invalidate_all_nopcid_cb(pmap_t pmap)
3274 {
3275 	if (pmap == kernel_pmap)
3276 		invltlb_glob();
3277 	else if (pmap == PCPU_GET(curpmap))
3278 		invltlb();
3279 }
3280 
3281 DEFINE_IFUNC(static, void, pmap_invalidate_all_cb, (pmap_t))
3282 {
3283 	if (pmap_pcid_enabled)
3284 		return (invpcid_works ? pmap_invalidate_all_pcid_invpcid_cb :
3285 		    pmap_invalidate_all_pcid_noinvpcid_cb);
3286 	return (pmap_invalidate_all_nopcid_cb);
3287 }
3288 
3289 static void
pmap_invalidate_all_curcpu_cb(pmap_t pmap,vm_offset_t addr1 __unused,vm_offset_t addr2 __unused)3290 pmap_invalidate_all_curcpu_cb(pmap_t pmap, vm_offset_t addr1 __unused,
3291     vm_offset_t addr2 __unused)
3292 {
3293 	pmap_invalidate_all_cb(pmap);
3294 }
3295 
3296 void
pmap_invalidate_all(pmap_t pmap)3297 pmap_invalidate_all(pmap_t pmap)
3298 {
3299 	if (pmap_type_guest(pmap)) {
3300 		pmap_invalidate_ept(pmap);
3301 		return;
3302 	}
3303 
3304 	KASSERT(pmap->pm_type == PT_X86,
3305 	    ("pmap_invalidate_all: invalid type %d", pmap->pm_type));
3306 
3307 	pmap_invalidate_preipi(pmap);
3308 	smp_masked_invltlb(pmap, pmap_invalidate_all_curcpu_cb);
3309 }
3310 
3311 static void
pmap_invalidate_cache_curcpu_cb(pmap_t pmap __unused,vm_offset_t va __unused,vm_offset_t addr2 __unused)3312 pmap_invalidate_cache_curcpu_cb(pmap_t pmap __unused, vm_offset_t va __unused,
3313     vm_offset_t addr2 __unused)
3314 {
3315 	wbinvd();
3316 }
3317 
3318 void
pmap_invalidate_cache(void)3319 pmap_invalidate_cache(void)
3320 {
3321 	sched_pin();
3322 	smp_cache_flush(pmap_invalidate_cache_curcpu_cb);
3323 }
3324 
3325 struct pde_action {
3326 	cpuset_t invalidate;	/* processors that invalidate their TLB */
3327 	pmap_t pmap;
3328 	vm_offset_t va;
3329 	pd_entry_t *pde;
3330 	pd_entry_t newpde;
3331 	u_int store;		/* processor that updates the PDE */
3332 };
3333 
3334 static void
pmap_update_pde_action(void * arg)3335 pmap_update_pde_action(void *arg)
3336 {
3337 	struct pde_action *act = arg;
3338 
3339 	if (act->store == PCPU_GET(cpuid))
3340 		pmap_update_pde_store(act->pmap, act->pde, act->newpde);
3341 }
3342 
3343 static void
pmap_update_pde_teardown(void * arg)3344 pmap_update_pde_teardown(void *arg)
3345 {
3346 	struct pde_action *act = arg;
3347 
3348 	if (CPU_ISSET(PCPU_GET(cpuid), &act->invalidate))
3349 		pmap_update_pde_invalidate(act->pmap, act->va, act->newpde);
3350 }
3351 
3352 /*
3353  * Change the page size for the specified virtual address in a way that
3354  * prevents any possibility of the TLB ever having two entries that map the
3355  * same virtual address using different page sizes.  This is the recommended
3356  * workaround for Erratum 383 on AMD Family 10h processors.  It prevents a
3357  * machine check exception for a TLB state that is improperly diagnosed as a
3358  * hardware error.
3359  */
3360 static void
pmap_update_pde(pmap_t pmap,vm_offset_t va,pd_entry_t * pde,pd_entry_t newpde)3361 pmap_update_pde(pmap_t pmap, vm_offset_t va, pd_entry_t *pde, pd_entry_t newpde)
3362 {
3363 	struct pde_action act;
3364 	cpuset_t active, other_cpus;
3365 	u_int cpuid;
3366 
3367 	sched_pin();
3368 	cpuid = PCPU_GET(cpuid);
3369 	other_cpus = all_cpus;
3370 	CPU_CLR(cpuid, &other_cpus);
3371 	if (pmap == kernel_pmap || pmap_type_guest(pmap))
3372 		active = all_cpus;
3373 	else {
3374 		active = pmap->pm_active;
3375 	}
3376 	if (CPU_OVERLAP(&active, &other_cpus)) {
3377 		act.store = cpuid;
3378 		act.invalidate = active;
3379 		act.va = va;
3380 		act.pmap = pmap;
3381 		act.pde = pde;
3382 		act.newpde = newpde;
3383 		CPU_SET(cpuid, &active);
3384 		smp_rendezvous_cpus(active,
3385 		    smp_no_rendezvous_barrier, pmap_update_pde_action,
3386 		    pmap_update_pde_teardown, &act);
3387 	} else {
3388 		pmap_update_pde_store(pmap, pde, newpde);
3389 		if (CPU_ISSET(cpuid, &active))
3390 			pmap_update_pde_invalidate(pmap, va, newpde);
3391 	}
3392 	sched_unpin();
3393 }
3394 #else /* !SMP */
3395 /*
3396  * Normal, non-SMP, invalidation functions.
3397  */
3398 void
pmap_invalidate_page(pmap_t pmap,vm_offset_t va)3399 pmap_invalidate_page(pmap_t pmap, vm_offset_t va)
3400 {
3401 	struct invpcid_descr d;
3402 	uint64_t kcr3, ucr3;
3403 	uint32_t pcid;
3404 
3405 	if (pmap->pm_type == PT_RVI || pmap->pm_type == PT_EPT) {
3406 		pmap->pm_eptgen++;
3407 		return;
3408 	}
3409 	KASSERT(pmap->pm_type == PT_X86,
3410 	    ("pmap_invalidate_range: unknown type %d", pmap->pm_type));
3411 
3412 	if (pmap == kernel_pmap || pmap == PCPU_GET(curpmap)) {
3413 		invlpg(va);
3414 		if (pmap == PCPU_GET(curpmap) && pmap_pcid_enabled &&
3415 		    pmap->pm_ucr3 != PMAP_NO_CR3) {
3416 			critical_enter();
3417 			pcid = pmap->pm_pcids[0].pm_pcid;
3418 			if (invpcid_works) {
3419 				d.pcid = pcid | PMAP_PCID_USER_PT;
3420 				d.pad = 0;
3421 				d.addr = va;
3422 				invpcid(&d, INVPCID_ADDR);
3423 			} else {
3424 				kcr3 = pmap->pm_cr3 | pcid | CR3_PCID_SAVE;
3425 				ucr3 = pmap->pm_ucr3 | pcid |
3426 				    PMAP_PCID_USER_PT | CR3_PCID_SAVE;
3427 				pmap_pti_pcid_invlpg(ucr3, kcr3, va);
3428 			}
3429 			critical_exit();
3430 		}
3431 	} else if (pmap_pcid_enabled)
3432 		pmap->pm_pcids[0].pm_gen = 0;
3433 }
3434 
3435 void
pmap_invalidate_range(pmap_t pmap,vm_offset_t sva,vm_offset_t eva)3436 pmap_invalidate_range(pmap_t pmap, vm_offset_t sva, vm_offset_t eva)
3437 {
3438 	struct invpcid_descr d;
3439 	vm_offset_t addr;
3440 	uint64_t kcr3, ucr3;
3441 
3442 	if (pmap->pm_type == PT_RVI || pmap->pm_type == PT_EPT) {
3443 		pmap->pm_eptgen++;
3444 		return;
3445 	}
3446 	KASSERT(pmap->pm_type == PT_X86,
3447 	    ("pmap_invalidate_range: unknown type %d", pmap->pm_type));
3448 
3449 	if (pmap == kernel_pmap || pmap == PCPU_GET(curpmap)) {
3450 		for (addr = sva; addr < eva; addr += PAGE_SIZE)
3451 			invlpg(addr);
3452 		if (pmap == PCPU_GET(curpmap) && pmap_pcid_enabled &&
3453 		    pmap->pm_ucr3 != PMAP_NO_CR3) {
3454 			critical_enter();
3455 			if (invpcid_works) {
3456 				d.pcid = pmap->pm_pcids[0].pm_pcid |
3457 				    PMAP_PCID_USER_PT;
3458 				d.pad = 0;
3459 				d.addr = sva;
3460 				for (; d.addr < eva; d.addr += PAGE_SIZE)
3461 					invpcid(&d, INVPCID_ADDR);
3462 			} else {
3463 				kcr3 = pmap->pm_cr3 | pmap->pm_pcids[0].
3464 				    pm_pcid | CR3_PCID_SAVE;
3465 				ucr3 = pmap->pm_ucr3 | pmap->pm_pcids[0].
3466 				    pm_pcid | PMAP_PCID_USER_PT | CR3_PCID_SAVE;
3467 				pmap_pti_pcid_invlrng(ucr3, kcr3, sva, eva);
3468 			}
3469 			critical_exit();
3470 		}
3471 	} else if (pmap_pcid_enabled) {
3472 		pmap->pm_pcids[0].pm_gen = 0;
3473 	}
3474 }
3475 
3476 void
pmap_invalidate_all(pmap_t pmap)3477 pmap_invalidate_all(pmap_t pmap)
3478 {
3479 	struct invpcid_descr d;
3480 	uint64_t kcr3, ucr3;
3481 
3482 	if (pmap->pm_type == PT_RVI || pmap->pm_type == PT_EPT) {
3483 		pmap->pm_eptgen++;
3484 		return;
3485 	}
3486 	KASSERT(pmap->pm_type == PT_X86,
3487 	    ("pmap_invalidate_all: unknown type %d", pmap->pm_type));
3488 
3489 	if (pmap == kernel_pmap) {
3490 		if (pmap_pcid_enabled && invpcid_works) {
3491 			bzero(&d, sizeof(d));
3492 			invpcid(&d, INVPCID_CTXGLOB);
3493 		} else {
3494 			invltlb_glob();
3495 		}
3496 	} else if (pmap == PCPU_GET(curpmap)) {
3497 		if (pmap_pcid_enabled) {
3498 			critical_enter();
3499 			if (invpcid_works) {
3500 				d.pcid = pmap->pm_pcids[0].pm_pcid;
3501 				d.pad = 0;
3502 				d.addr = 0;
3503 				invpcid(&d, INVPCID_CTX);
3504 				if (pmap->pm_ucr3 != PMAP_NO_CR3) {
3505 					d.pcid |= PMAP_PCID_USER_PT;
3506 					invpcid(&d, INVPCID_CTX);
3507 				}
3508 			} else {
3509 				kcr3 = pmap->pm_cr3 | pmap->pm_pcids[0].pm_pcid;
3510 				if (pmap->pm_ucr3 != PMAP_NO_CR3) {
3511 					ucr3 = pmap->pm_ucr3 | pmap->pm_pcids[
3512 					    0].pm_pcid | PMAP_PCID_USER_PT;
3513 					pmap_pti_pcid_invalidate(ucr3, kcr3);
3514 				} else
3515 					load_cr3(kcr3);
3516 			}
3517 			critical_exit();
3518 		} else {
3519 			invltlb();
3520 		}
3521 	} else if (pmap_pcid_enabled) {
3522 		pmap->pm_pcids[0].pm_gen = 0;
3523 	}
3524 }
3525 
3526 PMAP_INLINE void
pmap_invalidate_cache(void)3527 pmap_invalidate_cache(void)
3528 {
3529 
3530 	wbinvd();
3531 }
3532 
3533 static void
pmap_update_pde(pmap_t pmap,vm_offset_t va,pd_entry_t * pde,pd_entry_t newpde)3534 pmap_update_pde(pmap_t pmap, vm_offset_t va, pd_entry_t *pde, pd_entry_t newpde)
3535 {
3536 
3537 	pmap_update_pde_store(pmap, pde, newpde);
3538 	if (pmap == kernel_pmap || pmap == PCPU_GET(curpmap))
3539 		pmap_update_pde_invalidate(pmap, va, newpde);
3540 	else
3541 		pmap->pm_pcids[0].pm_gen = 0;
3542 }
3543 #endif /* !SMP */
3544 
3545 static void
pmap_invalidate_pde_page(pmap_t pmap,vm_offset_t va,pd_entry_t pde)3546 pmap_invalidate_pde_page(pmap_t pmap, vm_offset_t va, pd_entry_t pde)
3547 {
3548 
3549 	/*
3550 	 * When the PDE has PG_PROMOTED set, the 2MB page mapping was created
3551 	 * by a promotion that did not invalidate the 512 4KB page mappings
3552 	 * that might exist in the TLB.  Consequently, at this point, the TLB
3553 	 * may hold both 4KB and 2MB page mappings for the address range [va,
3554 	 * va + NBPDR).  Therefore, the entire range must be invalidated here.
3555 	 * In contrast, when PG_PROMOTED is clear, the TLB will not hold any
3556 	 * 4KB page mappings for the address range [va, va + NBPDR), and so a
3557 	 * single INVLPG suffices to invalidate the 2MB page mapping from the
3558 	 * TLB.
3559 	 */
3560 	if ((pde & PG_PROMOTED) != 0)
3561 		pmap_invalidate_range(pmap, va, va + NBPDR - 1);
3562 	else
3563 		pmap_invalidate_page(pmap, va);
3564 }
3565 
3566 DEFINE_IFUNC(, void, pmap_invalidate_cache_range,
3567     (vm_offset_t sva, vm_offset_t eva))
3568 {
3569 
3570 	if ((cpu_feature & CPUID_SS) != 0)
3571 		return (pmap_invalidate_cache_range_selfsnoop);
3572 	if ((cpu_feature & CPUID_CLFSH) != 0)
3573 		return (pmap_force_invalidate_cache_range);
3574 	return (pmap_invalidate_cache_range_all);
3575 }
3576 
3577 #define PMAP_CLFLUSH_THRESHOLD   (2 * 1024 * 1024)
3578 
3579 static void
pmap_invalidate_cache_range_check_align(vm_offset_t sva,vm_offset_t eva)3580 pmap_invalidate_cache_range_check_align(vm_offset_t sva, vm_offset_t eva)
3581 {
3582 
3583 	KASSERT((sva & PAGE_MASK) == 0,
3584 	    ("pmap_invalidate_cache_range: sva not page-aligned"));
3585 	KASSERT((eva & PAGE_MASK) == 0,
3586 	    ("pmap_invalidate_cache_range: eva not page-aligned"));
3587 }
3588 
3589 static void
pmap_invalidate_cache_range_selfsnoop(vm_offset_t sva,vm_offset_t eva)3590 pmap_invalidate_cache_range_selfsnoop(vm_offset_t sva, vm_offset_t eva)
3591 {
3592 
3593 	pmap_invalidate_cache_range_check_align(sva, eva);
3594 }
3595 
3596 void
pmap_force_invalidate_cache_range(vm_offset_t sva,vm_offset_t eva)3597 pmap_force_invalidate_cache_range(vm_offset_t sva, vm_offset_t eva)
3598 {
3599 
3600 	sva &= ~(vm_offset_t)(cpu_clflush_line_size - 1);
3601 
3602 	/*
3603 	 * XXX: Some CPUs fault, hang, or trash the local APIC
3604 	 * registers if we use CLFLUSH on the local APIC range.  The
3605 	 * local APIC is always uncached, so we don't need to flush
3606 	 * for that range anyway.
3607 	 */
3608 	if (pmap_kextract(sva) == lapic_paddr)
3609 		return;
3610 
3611 	if ((cpu_stdext_feature & CPUID_STDEXT_CLFLUSHOPT) != 0) {
3612 		/*
3613 		 * Do per-cache line flush.  Use a locked
3614 		 * instruction to insure that previous stores are
3615 		 * included in the write-back.  The processor
3616 		 * propagates flush to other processors in the cache
3617 		 * coherence domain.
3618 		 */
3619 		atomic_thread_fence_seq_cst();
3620 		for (; sva < eva; sva += cpu_clflush_line_size)
3621 			clflushopt(sva);
3622 		atomic_thread_fence_seq_cst();
3623 	} else {
3624 		/*
3625 		 * Writes are ordered by CLFLUSH on Intel CPUs.
3626 		 */
3627 		if (cpu_vendor_id != CPU_VENDOR_INTEL)
3628 			mfence();
3629 		for (; sva < eva; sva += cpu_clflush_line_size)
3630 			clflush(sva);
3631 		if (cpu_vendor_id != CPU_VENDOR_INTEL)
3632 			mfence();
3633 	}
3634 }
3635 
3636 static void
pmap_invalidate_cache_range_all(vm_offset_t sva,vm_offset_t eva)3637 pmap_invalidate_cache_range_all(vm_offset_t sva, vm_offset_t eva)
3638 {
3639 
3640 	pmap_invalidate_cache_range_check_align(sva, eva);
3641 	pmap_invalidate_cache();
3642 }
3643 
3644 /*
3645  * Remove the specified set of pages from the data and instruction caches.
3646  *
3647  * In contrast to pmap_invalidate_cache_range(), this function does not
3648  * rely on the CPU's self-snoop feature, because it is intended for use
3649  * when moving pages into a different cache domain.
3650  */
3651 void
pmap_invalidate_cache_pages(vm_page_t * pages,int count)3652 pmap_invalidate_cache_pages(vm_page_t *pages, int count)
3653 {
3654 	vm_offset_t daddr, eva;
3655 	int i;
3656 	bool useclflushopt;
3657 
3658 	useclflushopt = (cpu_stdext_feature & CPUID_STDEXT_CLFLUSHOPT) != 0;
3659 	if (count >= PMAP_CLFLUSH_THRESHOLD / PAGE_SIZE ||
3660 	    ((cpu_feature & CPUID_CLFSH) == 0 && !useclflushopt))
3661 		pmap_invalidate_cache();
3662 	else {
3663 		if (useclflushopt)
3664 			atomic_thread_fence_seq_cst();
3665 		else if (cpu_vendor_id != CPU_VENDOR_INTEL)
3666 			mfence();
3667 		for (i = 0; i < count; i++) {
3668 			daddr = PHYS_TO_DMAP(VM_PAGE_TO_PHYS(pages[i]));
3669 			eva = daddr + PAGE_SIZE;
3670 			for (; daddr < eva; daddr += cpu_clflush_line_size) {
3671 				if (useclflushopt)
3672 					clflushopt(daddr);
3673 				else
3674 					clflush(daddr);
3675 			}
3676 		}
3677 		if (useclflushopt)
3678 			atomic_thread_fence_seq_cst();
3679 		else if (cpu_vendor_id != CPU_VENDOR_INTEL)
3680 			mfence();
3681 	}
3682 }
3683 
3684 void
pmap_flush_cache_range(vm_offset_t sva,vm_offset_t eva)3685 pmap_flush_cache_range(vm_offset_t sva, vm_offset_t eva)
3686 {
3687 
3688 	pmap_invalidate_cache_range_check_align(sva, eva);
3689 
3690 	if ((cpu_stdext_feature & CPUID_STDEXT_CLWB) == 0) {
3691 		pmap_force_invalidate_cache_range(sva, eva);
3692 		return;
3693 	}
3694 
3695 	/* See comment in pmap_force_invalidate_cache_range(). */
3696 	if (pmap_kextract(sva) == lapic_paddr)
3697 		return;
3698 
3699 	atomic_thread_fence_seq_cst();
3700 	for (; sva < eva; sva += cpu_clflush_line_size)
3701 		clwb(sva);
3702 	atomic_thread_fence_seq_cst();
3703 }
3704 
3705 void
pmap_flush_cache_phys_range(vm_paddr_t spa,vm_paddr_t epa,vm_memattr_t mattr)3706 pmap_flush_cache_phys_range(vm_paddr_t spa, vm_paddr_t epa, vm_memattr_t mattr)
3707 {
3708 	pt_entry_t *pte;
3709 	vm_offset_t vaddr;
3710 	int error, pte_bits;
3711 
3712 	KASSERT((spa & PAGE_MASK) == 0,
3713 	    ("pmap_flush_cache_phys_range: spa not page-aligned"));
3714 	KASSERT((epa & PAGE_MASK) == 0,
3715 	    ("pmap_flush_cache_phys_range: epa not page-aligned"));
3716 
3717 	if (spa < dmaplimit) {
3718 		pmap_flush_cache_range(PHYS_TO_DMAP(spa), PHYS_TO_DMAP(MIN(
3719 		    dmaplimit, epa)));
3720 		if (dmaplimit >= epa)
3721 			return;
3722 		spa = dmaplimit;
3723 	}
3724 
3725 	pte_bits = pmap_cache_bits(kernel_pmap, mattr, 0) | X86_PG_RW |
3726 	    X86_PG_V;
3727 	error = vmem_alloc(kernel_arena, PAGE_SIZE, M_BESTFIT | M_WAITOK,
3728 	    &vaddr);
3729 	KASSERT(error == 0, ("vmem_alloc failed: %d", error));
3730 	pte = vtopte(vaddr);
3731 	for (; spa < epa; spa += PAGE_SIZE) {
3732 		sched_pin();
3733 		pte_store(pte, spa | pte_bits);
3734 		pmap_invlpg(kernel_pmap, vaddr);
3735 		/* XXXKIB atomic inside flush_cache_range are excessive */
3736 		pmap_flush_cache_range(vaddr, vaddr + PAGE_SIZE);
3737 		sched_unpin();
3738 	}
3739 	vmem_free(kernel_arena, vaddr, PAGE_SIZE);
3740 }
3741 
3742 /*
3743  *	Routine:	pmap_extract
3744  *	Function:
3745  *		Extract the physical page address associated
3746  *		with the given map/virtual_address pair.
3747  */
3748 vm_paddr_t
pmap_extract(pmap_t pmap,vm_offset_t va)3749 pmap_extract(pmap_t pmap, vm_offset_t va)
3750 {
3751 	pdp_entry_t *pdpe;
3752 	pd_entry_t *pde;
3753 	pt_entry_t *pte, PG_V;
3754 	vm_paddr_t pa;
3755 
3756 	pa = 0;
3757 	PG_V = pmap_valid_bit(pmap);
3758 	PMAP_LOCK(pmap);
3759 	pdpe = pmap_pdpe(pmap, va);
3760 	if (pdpe != NULL && (*pdpe & PG_V) != 0) {
3761 		if ((*pdpe & PG_PS) != 0)
3762 			pa = (*pdpe & PG_PS_FRAME) | (va & PDPMASK);
3763 		else {
3764 			pde = pmap_pdpe_to_pde(pdpe, va);
3765 			if ((*pde & PG_V) != 0) {
3766 				if ((*pde & PG_PS) != 0) {
3767 					pa = (*pde & PG_PS_FRAME) |
3768 					    (va & PDRMASK);
3769 				} else {
3770 					pte = pmap_pde_to_pte(pde, va);
3771 					pa = (*pte & PG_FRAME) |
3772 					    (va & PAGE_MASK);
3773 				}
3774 			}
3775 		}
3776 	}
3777 	PMAP_UNLOCK(pmap);
3778 	return (pa);
3779 }
3780 
3781 /*
3782  *	Routine:	pmap_extract_and_hold
3783  *	Function:
3784  *		Atomically extract and hold the physical page
3785  *		with the given pmap and virtual address pair
3786  *		if that mapping permits the given protection.
3787  */
3788 vm_page_t
pmap_extract_and_hold(pmap_t pmap,vm_offset_t va,vm_prot_t prot)3789 pmap_extract_and_hold(pmap_t pmap, vm_offset_t va, vm_prot_t prot)
3790 {
3791 	pdp_entry_t pdpe, *pdpep;
3792 	pd_entry_t pde, *pdep;
3793 	pt_entry_t pte, PG_RW, PG_V;
3794 	vm_page_t m;
3795 
3796 	m = NULL;
3797 	PG_RW = pmap_rw_bit(pmap);
3798 	PG_V = pmap_valid_bit(pmap);
3799 	PMAP_LOCK(pmap);
3800 
3801 	pdpep = pmap_pdpe(pmap, va);
3802 	if (pdpep == NULL || ((pdpe = *pdpep) & PG_V) == 0)
3803 		goto out;
3804 	if ((pdpe & PG_PS) != 0) {
3805 		if ((pdpe & PG_RW) == 0 && (prot & VM_PROT_WRITE) != 0)
3806 			goto out;
3807 		m = PHYS_TO_VM_PAGE((pdpe & PG_PS_FRAME) | (va & PDPMASK));
3808 		goto check_page;
3809 	}
3810 
3811 	pdep = pmap_pdpe_to_pde(pdpep, va);
3812 	if (pdep == NULL || ((pde = *pdep) & PG_V) == 0)
3813 		goto out;
3814 	if ((pde & PG_PS) != 0) {
3815 		if ((pde & PG_RW) == 0 && (prot & VM_PROT_WRITE) != 0)
3816 			goto out;
3817 		m = PHYS_TO_VM_PAGE((pde & PG_PS_FRAME) | (va & PDRMASK));
3818 		goto check_page;
3819 	}
3820 
3821 	pte = *pmap_pde_to_pte(pdep, va);
3822 	if ((pte & PG_V) == 0 ||
3823 	    ((pte & PG_RW) == 0 && (prot & VM_PROT_WRITE) != 0))
3824 		goto out;
3825 	m = PHYS_TO_VM_PAGE(pte & PG_FRAME);
3826 
3827 check_page:
3828 	if (m != NULL && !vm_page_wire_mapped(m))
3829 		m = NULL;
3830 out:
3831 	PMAP_UNLOCK(pmap);
3832 	return (m);
3833 }
3834 
3835 vm_paddr_t
pmap_kextract(vm_offset_t va)3836 pmap_kextract(vm_offset_t va)
3837 {
3838 	pd_entry_t pde;
3839 	vm_paddr_t pa;
3840 
3841 	if (va >= DMAP_MIN_ADDRESS && va < DMAP_MAX_ADDRESS) {
3842 		pa = DMAP_TO_PHYS(va);
3843 	} else if (PMAP_ADDRESS_IN_LARGEMAP(va)) {
3844 		pa = pmap_large_map_kextract(va);
3845 	} else {
3846 		pde = *vtopde(va);
3847 		if (pde & PG_PS) {
3848 			pa = (pde & PG_PS_FRAME) | (va & PDRMASK);
3849 		} else {
3850 			/*
3851 			 * Beware of a concurrent promotion that changes the
3852 			 * PDE at this point!  For example, vtopte() must not
3853 			 * be used to access the PTE because it would use the
3854 			 * new PDE.  It is, however, safe to use the old PDE
3855 			 * because the page table page is preserved by the
3856 			 * promotion.
3857 			 */
3858 			pa = *pmap_pde_to_pte(&pde, va);
3859 			pa = (pa & PG_FRAME) | (va & PAGE_MASK);
3860 		}
3861 	}
3862 	return (pa);
3863 }
3864 
3865 /***************************************************
3866  * Low level mapping routines.....
3867  ***************************************************/
3868 
3869 /*
3870  * Add a wired page to the kva.
3871  * Note: not SMP coherent.
3872  */
3873 PMAP_INLINE void
pmap_kenter(vm_offset_t va,vm_paddr_t pa)3874 pmap_kenter(vm_offset_t va, vm_paddr_t pa)
3875 {
3876 	pt_entry_t *pte;
3877 
3878 	pte = vtopte(va);
3879 	pte_store(pte, pa | pg_g | pg_nx | X86_PG_A | X86_PG_M |
3880 	    X86_PG_RW | X86_PG_V);
3881 }
3882 
3883 static __inline void
pmap_kenter_attr(vm_offset_t va,vm_paddr_t pa,int mode)3884 pmap_kenter_attr(vm_offset_t va, vm_paddr_t pa, int mode)
3885 {
3886 	pt_entry_t *pte;
3887 	int cache_bits;
3888 
3889 	pte = vtopte(va);
3890 	cache_bits = pmap_cache_bits(kernel_pmap, mode, 0);
3891 	pte_store(pte, pa | pg_g | pg_nx | X86_PG_A | X86_PG_M |
3892 	    X86_PG_RW | X86_PG_V | cache_bits);
3893 }
3894 
3895 /*
3896  * Remove a page from the kernel pagetables.
3897  * Note: not SMP coherent.
3898  */
3899 PMAP_INLINE void
pmap_kremove(vm_offset_t va)3900 pmap_kremove(vm_offset_t va)
3901 {
3902 	pt_entry_t *pte;
3903 
3904 	pte = vtopte(va);
3905 	pte_clear(pte);
3906 }
3907 
3908 /*
3909  *	Used to map a range of physical addresses into kernel
3910  *	virtual address space.
3911  *
3912  *	The value passed in '*virt' is a suggested virtual address for
3913  *	the mapping. Architectures which can support a direct-mapped
3914  *	physical to virtual region can return the appropriate address
3915  *	within that region, leaving '*virt' unchanged. Other
3916  *	architectures should map the pages starting at '*virt' and
3917  *	update '*virt' with the first usable address after the mapped
3918  *	region.
3919  */
3920 vm_offset_t
pmap_map(vm_offset_t * virt,vm_paddr_t start,vm_paddr_t end,int prot)3921 pmap_map(vm_offset_t *virt, vm_paddr_t start, vm_paddr_t end, int prot)
3922 {
3923 	return PHYS_TO_DMAP(start);
3924 }
3925 
3926 /*
3927  * Add a list of wired pages to the kva
3928  * this routine is only used for temporary
3929  * kernel mappings that do not need to have
3930  * page modification or references recorded.
3931  * Note that old mappings are simply written
3932  * over.  The page *must* be wired.
3933  * Note: SMP coherent.  Uses a ranged shootdown IPI.
3934  */
3935 void
pmap_qenter(vm_offset_t sva,vm_page_t * ma,int count)3936 pmap_qenter(vm_offset_t sva, vm_page_t *ma, int count)
3937 {
3938 	pt_entry_t *endpte, oldpte, pa, *pte;
3939 	vm_page_t m;
3940 	int cache_bits;
3941 
3942 	oldpte = 0;
3943 	pte = vtopte(sva);
3944 	endpte = pte + count;
3945 	while (pte < endpte) {
3946 		m = *ma++;
3947 		cache_bits = pmap_cache_bits(kernel_pmap, m->md.pat_mode, 0);
3948 		pa = VM_PAGE_TO_PHYS(m) | cache_bits;
3949 		if ((*pte & (PG_FRAME | X86_PG_PTE_CACHE)) != pa) {
3950 			oldpte |= *pte;
3951 			pte_store(pte, pa | pg_g | pg_nx | X86_PG_A |
3952 			    X86_PG_M | X86_PG_RW | X86_PG_V);
3953 		}
3954 		pte++;
3955 	}
3956 	if (__predict_false((oldpte & X86_PG_V) != 0))
3957 		pmap_invalidate_range(kernel_pmap, sva, sva + count *
3958 		    PAGE_SIZE);
3959 }
3960 
3961 /*
3962  * This routine tears out page mappings from the
3963  * kernel -- it is meant only for temporary mappings.
3964  * Note: SMP coherent.  Uses a ranged shootdown IPI.
3965  */
3966 void
pmap_qremove(vm_offset_t sva,int count)3967 pmap_qremove(vm_offset_t sva, int count)
3968 {
3969 	vm_offset_t va;
3970 
3971 	va = sva;
3972 	while (count-- > 0) {
3973 		KASSERT(va >= VM_MIN_KERNEL_ADDRESS, ("usermode va %lx", va));
3974 		pmap_kremove(va);
3975 		va += PAGE_SIZE;
3976 	}
3977 	pmap_invalidate_range(kernel_pmap, sva, va);
3978 }
3979 
3980 /***************************************************
3981  * Page table page management routines.....
3982  ***************************************************/
3983 /*
3984  * Schedule the specified unused page table page to be freed.  Specifically,
3985  * add the page to the specified list of pages that will be released to the
3986  * physical memory manager after the TLB has been updated.
3987  */
3988 static __inline void
pmap_add_delayed_free_list(vm_page_t m,struct spglist * free,boolean_t set_PG_ZERO)3989 pmap_add_delayed_free_list(vm_page_t m, struct spglist *free,
3990     boolean_t set_PG_ZERO)
3991 {
3992 
3993 	if (set_PG_ZERO)
3994 		m->flags |= PG_ZERO;
3995 	else
3996 		m->flags &= ~PG_ZERO;
3997 	SLIST_INSERT_HEAD(free, m, plinks.s.ss);
3998 }
3999 
4000 /*
4001  * Inserts the specified page table page into the specified pmap's collection
4002  * of idle page table pages.  Each of a pmap's page table pages is responsible
4003  * for mapping a distinct range of virtual addresses.  The pmap's collection is
4004  * ordered by this virtual address range.
4005  *
4006  * If "promoted" is false, then the page table page "mpte" must be zero filled.
4007  */
4008 static __inline int
pmap_insert_pt_page(pmap_t pmap,vm_page_t mpte,bool promoted)4009 pmap_insert_pt_page(pmap_t pmap, vm_page_t mpte, bool promoted)
4010 {
4011 
4012 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
4013 	mpte->valid = promoted ? VM_PAGE_BITS_ALL : 0;
4014 	return (vm_radix_insert(&pmap->pm_root, mpte));
4015 }
4016 
4017 /*
4018  * Removes the page table page mapping the specified virtual address from the
4019  * specified pmap's collection of idle page table pages, and returns it.
4020  * Otherwise, returns NULL if there is no page table page corresponding to the
4021  * specified virtual address.
4022  */
4023 static __inline vm_page_t
pmap_remove_pt_page(pmap_t pmap,vm_offset_t va)4024 pmap_remove_pt_page(pmap_t pmap, vm_offset_t va)
4025 {
4026 
4027 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
4028 	return (vm_radix_remove(&pmap->pm_root, pmap_pde_pindex(va)));
4029 }
4030 
4031 /*
4032  * Decrements a page table page's reference count, which is used to record the
4033  * number of valid page table entries within the page.  If the reference count
4034  * drops to zero, then the page table page is unmapped.  Returns TRUE if the
4035  * page table page was unmapped and FALSE otherwise.
4036  */
4037 static inline boolean_t
pmap_unwire_ptp(pmap_t pmap,vm_offset_t va,vm_page_t m,struct spglist * free)4038 pmap_unwire_ptp(pmap_t pmap, vm_offset_t va, vm_page_t m, struct spglist *free)
4039 {
4040 
4041 	--m->ref_count;
4042 	if (m->ref_count == 0) {
4043 		_pmap_unwire_ptp(pmap, va, m, free);
4044 		return (TRUE);
4045 	} else
4046 		return (FALSE);
4047 }
4048 
4049 static void
_pmap_unwire_ptp(pmap_t pmap,vm_offset_t va,vm_page_t m,struct spglist * free)4050 _pmap_unwire_ptp(pmap_t pmap, vm_offset_t va, vm_page_t m, struct spglist *free)
4051 {
4052 	pml5_entry_t *pml5;
4053 	pml4_entry_t *pml4;
4054 	pdp_entry_t *pdp;
4055 	pd_entry_t *pd;
4056 	vm_page_t pdpg, pdppg, pml4pg;
4057 
4058 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
4059 
4060 	/*
4061 	 * unmap the page table page
4062 	 */
4063 	if (m->pindex >= NUPDE + NUPDPE + NUPML4E) {
4064 		/* PML4 page */
4065 		MPASS(pmap_is_la57(pmap));
4066 		pml5 = pmap_pml5e(pmap, va);
4067 		*pml5 = 0;
4068 		if (pmap->pm_pmltopu != NULL && va <= VM_MAXUSER_ADDRESS) {
4069 			pml5 = pmap_pml5e_u(pmap, va);
4070 			*pml5 = 0;
4071 		}
4072 	} else if (m->pindex >= NUPDE + NUPDPE) {
4073 		/* PDP page */
4074 		pml4 = pmap_pml4e(pmap, va);
4075 		*pml4 = 0;
4076 		if (!pmap_is_la57(pmap) && pmap->pm_pmltopu != NULL &&
4077 		    va <= VM_MAXUSER_ADDRESS) {
4078 			pml4 = pmap_pml4e_u(pmap, va);
4079 			*pml4 = 0;
4080 		}
4081 	} else if (m->pindex >= NUPDE) {
4082 		/* PD page */
4083 		pdp = pmap_pdpe(pmap, va);
4084 		*pdp = 0;
4085 	} else {
4086 		/* PTE page */
4087 		pd = pmap_pde(pmap, va);
4088 		*pd = 0;
4089 	}
4090 	if (m->pindex < NUPDE) {
4091 		/* We just released a PT, unhold the matching PD */
4092 		pdpg = PHYS_TO_VM_PAGE(*pmap_pdpe(pmap, va) & PG_FRAME);
4093 		pmap_unwire_ptp(pmap, va, pdpg, free);
4094 	} else if (m->pindex < NUPDE + NUPDPE) {
4095 		/* We just released a PD, unhold the matching PDP */
4096 		pdppg = PHYS_TO_VM_PAGE(*pmap_pml4e(pmap, va) & PG_FRAME);
4097 		pmap_unwire_ptp(pmap, va, pdppg, free);
4098 	} else if (m->pindex < NUPDE + NUPDPE + NUPML4E && pmap_is_la57(pmap)) {
4099 		/* We just released a PDP, unhold the matching PML4 */
4100 		pml4pg = PHYS_TO_VM_PAGE(*pmap_pml5e(pmap, va) & PG_FRAME);
4101 		pmap_unwire_ptp(pmap, va, pml4pg, free);
4102 	}
4103 
4104 	pmap_pt_page_count_adj(pmap, -1);
4105 
4106 	/*
4107 	 * Put page on a list so that it is released after
4108 	 * *ALL* TLB shootdown is done
4109 	 */
4110 	pmap_add_delayed_free_list(m, free, TRUE);
4111 }
4112 
4113 /*
4114  * After removing a page table entry, this routine is used to
4115  * conditionally free the page, and manage the reference count.
4116  */
4117 static int
pmap_unuse_pt(pmap_t pmap,vm_offset_t va,pd_entry_t ptepde,struct spglist * free)4118 pmap_unuse_pt(pmap_t pmap, vm_offset_t va, pd_entry_t ptepde,
4119     struct spglist *free)
4120 {
4121 	vm_page_t mpte;
4122 
4123 	if (va >= VM_MAXUSER_ADDRESS)
4124 		return (0);
4125 	KASSERT(ptepde != 0, ("pmap_unuse_pt: ptepde != 0"));
4126 	mpte = PHYS_TO_VM_PAGE(ptepde & PG_FRAME);
4127 	return (pmap_unwire_ptp(pmap, va, mpte, free));
4128 }
4129 
4130 /*
4131  * Release a page table page reference after a failed attempt to create a
4132  * mapping.
4133  */
4134 static void
pmap_abort_ptp(pmap_t pmap,vm_offset_t va,vm_page_t mpte)4135 pmap_abort_ptp(pmap_t pmap, vm_offset_t va, vm_page_t mpte)
4136 {
4137 	struct spglist free;
4138 
4139 	SLIST_INIT(&free);
4140 	if (pmap_unwire_ptp(pmap, va, mpte, &free)) {
4141 		/*
4142 		 * Although "va" was never mapped, paging-structure caches
4143 		 * could nonetheless have entries that refer to the freed
4144 		 * page table pages.  Invalidate those entries.
4145 		 */
4146 		pmap_invalidate_page(pmap, va);
4147 		vm_page_free_pages_toq(&free, true);
4148 	}
4149 }
4150 
4151 void
pmap_pinit0(pmap_t pmap)4152 pmap_pinit0(pmap_t pmap)
4153 {
4154 	struct proc *p;
4155 	struct thread *td;
4156 	int i;
4157 
4158 	PMAP_LOCK_INIT(pmap);
4159 	pmap->pm_pmltop = kernel_pmap->pm_pmltop;
4160 	pmap->pm_pmltopu = NULL;
4161 	pmap->pm_cr3 = kernel_pmap->pm_cr3;
4162 	/* hack to keep pmap_pti_pcid_invalidate() alive */
4163 	pmap->pm_ucr3 = PMAP_NO_CR3;
4164 	vm_radix_init(&pmap->pm_root);
4165 	CPU_ZERO(&pmap->pm_active);
4166 	TAILQ_INIT(&pmap->pm_pvchunk);
4167 	bzero(&pmap->pm_stats, sizeof pmap->pm_stats);
4168 	pmap->pm_flags = pmap_flags;
4169 	CPU_FOREACH(i) {
4170 		pmap->pm_pcids[i].pm_pcid = PMAP_PCID_KERN + 1;
4171 		pmap->pm_pcids[i].pm_gen = 1;
4172 	}
4173 	pmap_activate_boot(pmap);
4174 	td = curthread;
4175 	if (pti) {
4176 		p = td->td_proc;
4177 		PROC_LOCK(p);
4178 		p->p_md.md_flags |= P_MD_KPTI;
4179 		PROC_UNLOCK(p);
4180 	}
4181 	pmap_thread_init_invl_gen(td);
4182 
4183 	if ((cpu_stdext_feature2 & CPUID_STDEXT2_PKU) != 0) {
4184 		pmap_pkru_ranges_zone = uma_zcreate("pkru ranges",
4185 		    sizeof(struct pmap_pkru_range), NULL, NULL, NULL, NULL,
4186 		    UMA_ALIGN_PTR, 0);
4187 	}
4188 }
4189 
4190 void
pmap_pinit_pml4(vm_page_t pml4pg)4191 pmap_pinit_pml4(vm_page_t pml4pg)
4192 {
4193 	pml4_entry_t *pm_pml4;
4194 	int i;
4195 
4196 	pm_pml4 = (pml4_entry_t *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(pml4pg));
4197 
4198 	/* Wire in kernel global address entries. */
4199 	for (i = 0; i < NKPML4E; i++) {
4200 		pm_pml4[KPML4BASE + i] = (KPDPphys + ptoa(i)) | X86_PG_RW |
4201 		    X86_PG_V;
4202 	}
4203 #ifdef KASAN
4204 	for (i = 0; i < NKASANPML4E; i++) {
4205 		pm_pml4[KASANPML4I + i] = (KASANPDPphys + ptoa(i)) | X86_PG_RW |
4206 		    X86_PG_V | pg_nx;
4207 	}
4208 #endif
4209 	for (i = 0; i < ndmpdpphys; i++) {
4210 		pm_pml4[DMPML4I + i] = (DMPDPphys + ptoa(i)) | X86_PG_RW |
4211 		    X86_PG_V;
4212 	}
4213 
4214 	/* install self-referential address mapping entry(s) */
4215 	pm_pml4[PML4PML4I] = VM_PAGE_TO_PHYS(pml4pg) | X86_PG_V | X86_PG_RW |
4216 	    X86_PG_A | X86_PG_M;
4217 
4218 	/* install large map entries if configured */
4219 	for (i = 0; i < lm_ents; i++)
4220 		pm_pml4[LMSPML4I + i] = kernel_pmap->pm_pmltop[LMSPML4I + i];
4221 }
4222 
4223 void
pmap_pinit_pml5(vm_page_t pml5pg)4224 pmap_pinit_pml5(vm_page_t pml5pg)
4225 {
4226 	pml5_entry_t *pm_pml5;
4227 
4228 	pm_pml5 = (pml5_entry_t *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(pml5pg));
4229 
4230 	/*
4231 	 * Add pml5 entry at top of KVA pointing to existing pml4 table,
4232 	 * entering all existing kernel mappings into level 5 table.
4233 	 */
4234 	pm_pml5[pmap_pml5e_index(UPT_MAX_ADDRESS)] = KPML4phys | X86_PG_V |
4235 	    X86_PG_RW | X86_PG_A | X86_PG_M | pg_g |
4236 	    pmap_cache_bits(kernel_pmap, VM_MEMATTR_DEFAULT, FALSE);
4237 
4238 	/*
4239 	 * Install self-referential address mapping entry.
4240 	 */
4241 	pm_pml5[PML5PML5I] = VM_PAGE_TO_PHYS(pml5pg) |
4242 	    X86_PG_RW | X86_PG_V | X86_PG_M | X86_PG_A |
4243 	    pmap_cache_bits(kernel_pmap, VM_MEMATTR_DEFAULT, FALSE);
4244 }
4245 
4246 static void
pmap_pinit_pml4_pti(vm_page_t pml4pgu)4247 pmap_pinit_pml4_pti(vm_page_t pml4pgu)
4248 {
4249 	pml4_entry_t *pm_pml4u;
4250 	int i;
4251 
4252 	pm_pml4u = (pml4_entry_t *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(pml4pgu));
4253 	for (i = 0; i < NPML4EPG; i++)
4254 		pm_pml4u[i] = pti_pml4[i];
4255 }
4256 
4257 static void
pmap_pinit_pml5_pti(vm_page_t pml5pgu)4258 pmap_pinit_pml5_pti(vm_page_t pml5pgu)
4259 {
4260 	pml5_entry_t *pm_pml5u;
4261 
4262 	pm_pml5u = (pml5_entry_t *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(pml5pgu));
4263 	pagezero(pm_pml5u);
4264 
4265 	/*
4266 	 * Add pml5 entry at top of KVA pointing to existing pml4 pti
4267 	 * table, entering all kernel mappings needed for usermode
4268 	 * into level 5 table.
4269 	 */
4270 	pm_pml5u[pmap_pml5e_index(UPT_MAX_ADDRESS)] =
4271 	    pmap_kextract((vm_offset_t)pti_pml4) |
4272 	    X86_PG_V | X86_PG_RW | X86_PG_A | X86_PG_M | pg_g |
4273 	    pmap_cache_bits(kernel_pmap, VM_MEMATTR_DEFAULT, FALSE);
4274 }
4275 
4276 /* Allocate a page table page and do related bookkeeping */
4277 static vm_page_t
pmap_alloc_pt_page(pmap_t pmap,vm_pindex_t pindex,int flags)4278 pmap_alloc_pt_page(pmap_t pmap, vm_pindex_t pindex, int flags)
4279 {
4280 	vm_page_t m;
4281 
4282 	m = vm_page_alloc_noobj(flags);
4283 	if (__predict_false(m == NULL))
4284 		return (NULL);
4285 	m->pindex = pindex;
4286 	pmap_pt_page_count_adj(pmap, 1);
4287 	return (m);
4288 }
4289 
4290 static void
pmap_free_pt_page(pmap_t pmap,vm_page_t m,bool zerofilled)4291 pmap_free_pt_page(pmap_t pmap, vm_page_t m, bool zerofilled)
4292 {
4293 	/*
4294 	 * This function assumes the page will need to be unwired,
4295 	 * even though the counterpart allocation in pmap_alloc_pt_page()
4296 	 * doesn't enforce VM_ALLOC_WIRED.  However, all current uses
4297 	 * of pmap_free_pt_page() require unwiring.  The case in which
4298 	 * a PT page doesn't require unwiring because its ref_count has
4299 	 * naturally reached 0 is handled through _pmap_unwire_ptp().
4300 	 */
4301 	vm_page_unwire_noq(m);
4302 	if (zerofilled)
4303 		vm_page_free_zero(m);
4304 	else
4305 		vm_page_free(m);
4306 
4307 	pmap_pt_page_count_adj(pmap, -1);
4308 }
4309 
4310 /*
4311  * Initialize a preallocated and zeroed pmap structure,
4312  * such as one in a vmspace structure.
4313  */
4314 int
pmap_pinit_type(pmap_t pmap,enum pmap_type pm_type,int flags)4315 pmap_pinit_type(pmap_t pmap, enum pmap_type pm_type, int flags)
4316 {
4317 	vm_page_t pmltop_pg, pmltop_pgu;
4318 	vm_paddr_t pmltop_phys;
4319 	int i;
4320 
4321 	bzero(&pmap->pm_stats, sizeof pmap->pm_stats);
4322 
4323 	/*
4324 	 * Allocate the page directory page.  Pass NULL instead of a
4325 	 * pointer to the pmap here to avoid calling
4326 	 * pmap_resident_count_adj() through pmap_pt_page_count_adj(),
4327 	 * since that requires pmap lock.  Instead do the accounting
4328 	 * manually.
4329 	 *
4330 	 * Note that final call to pmap_remove() optimization that
4331 	 * checks for zero resident_count is basically disabled by
4332 	 * accounting for top-level page.  But the optimization was
4333 	 * not effective since we started using non-managed mapping of
4334 	 * the shared page.
4335 	 */
4336 	pmltop_pg = pmap_alloc_pt_page(NULL, 0, VM_ALLOC_WIRED | VM_ALLOC_ZERO |
4337 	    VM_ALLOC_WAITOK);
4338 	pmap_pt_page_count_pinit(pmap, 1);
4339 
4340 	pmltop_phys = VM_PAGE_TO_PHYS(pmltop_pg);
4341 	pmap->pm_pmltop = (pml5_entry_t *)PHYS_TO_DMAP(pmltop_phys);
4342 
4343 	CPU_FOREACH(i) {
4344 		pmap->pm_pcids[i].pm_pcid = PMAP_PCID_NONE;
4345 		pmap->pm_pcids[i].pm_gen = 0;
4346 	}
4347 	pmap->pm_cr3 = PMAP_NO_CR3;	/* initialize to an invalid value */
4348 	pmap->pm_ucr3 = PMAP_NO_CR3;
4349 	pmap->pm_pmltopu = NULL;
4350 
4351 	pmap->pm_type = pm_type;
4352 
4353 	/*
4354 	 * Do not install the host kernel mappings in the nested page
4355 	 * tables. These mappings are meaningless in the guest physical
4356 	 * address space.
4357 	 * Install minimal kernel mappings in PTI case.
4358 	 */
4359 	switch (pm_type) {
4360 	case PT_X86:
4361 		pmap->pm_cr3 = pmltop_phys;
4362 		if (pmap_is_la57(pmap))
4363 			pmap_pinit_pml5(pmltop_pg);
4364 		else
4365 			pmap_pinit_pml4(pmltop_pg);
4366 		if ((curproc->p_md.md_flags & P_MD_KPTI) != 0) {
4367 			/*
4368 			 * As with pmltop_pg, pass NULL instead of a
4369 			 * pointer to the pmap to ensure that the PTI
4370 			 * page counted explicitly.
4371 			 */
4372 			pmltop_pgu = pmap_alloc_pt_page(NULL, 0,
4373 			    VM_ALLOC_WIRED | VM_ALLOC_WAITOK);
4374 			pmap_pt_page_count_pinit(pmap, 1);
4375 			pmap->pm_pmltopu = (pml4_entry_t *)PHYS_TO_DMAP(
4376 			    VM_PAGE_TO_PHYS(pmltop_pgu));
4377 			if (pmap_is_la57(pmap))
4378 				pmap_pinit_pml5_pti(pmltop_pgu);
4379 			else
4380 				pmap_pinit_pml4_pti(pmltop_pgu);
4381 			pmap->pm_ucr3 = VM_PAGE_TO_PHYS(pmltop_pgu);
4382 		}
4383 		if ((cpu_stdext_feature2 & CPUID_STDEXT2_PKU) != 0) {
4384 			rangeset_init(&pmap->pm_pkru, pkru_dup_range,
4385 			    pkru_free_range, pmap, M_NOWAIT);
4386 		}
4387 		break;
4388 	case PT_EPT:
4389 	case PT_RVI:
4390 		pmap->pm_eptsmr = smr_create("pmap", 0, 0);
4391 		break;
4392 	}
4393 
4394 	vm_radix_init(&pmap->pm_root);
4395 	CPU_ZERO(&pmap->pm_active);
4396 	TAILQ_INIT(&pmap->pm_pvchunk);
4397 	pmap->pm_flags = flags;
4398 	pmap->pm_eptgen = 0;
4399 
4400 	return (1);
4401 }
4402 
4403 int
pmap_pinit(pmap_t pmap)4404 pmap_pinit(pmap_t pmap)
4405 {
4406 
4407 	return (pmap_pinit_type(pmap, PT_X86, pmap_flags));
4408 }
4409 
4410 static void
pmap_allocpte_free_unref(pmap_t pmap,vm_offset_t va,pt_entry_t * pte)4411 pmap_allocpte_free_unref(pmap_t pmap, vm_offset_t va, pt_entry_t *pte)
4412 {
4413 	vm_page_t mpg;
4414 	struct spglist free;
4415 
4416 	mpg = PHYS_TO_VM_PAGE(*pte & PG_FRAME);
4417 	if (mpg->ref_count != 0)
4418 		return;
4419 	SLIST_INIT(&free);
4420 	_pmap_unwire_ptp(pmap, va, mpg, &free);
4421 	pmap_invalidate_page(pmap, va);
4422 	vm_page_free_pages_toq(&free, true);
4423 }
4424 
4425 static pml4_entry_t *
pmap_allocpte_getpml4(pmap_t pmap,struct rwlock ** lockp,vm_offset_t va,bool addref)4426 pmap_allocpte_getpml4(pmap_t pmap, struct rwlock **lockp, vm_offset_t va,
4427     bool addref)
4428 {
4429 	vm_pindex_t pml5index;
4430 	pml5_entry_t *pml5;
4431 	pml4_entry_t *pml4;
4432 	vm_page_t pml4pg;
4433 	pt_entry_t PG_V;
4434 	bool allocated;
4435 
4436 	if (!pmap_is_la57(pmap))
4437 		return (&pmap->pm_pmltop[pmap_pml4e_index(va)]);
4438 
4439 	PG_V = pmap_valid_bit(pmap);
4440 	pml5index = pmap_pml5e_index(va);
4441 	pml5 = &pmap->pm_pmltop[pml5index];
4442 	if ((*pml5 & PG_V) == 0) {
4443 		if (pmap_allocpte_nosleep(pmap, pmap_pml5e_pindex(va), lockp,
4444 		    va) == NULL)
4445 			return (NULL);
4446 		allocated = true;
4447 	} else {
4448 		allocated = false;
4449 	}
4450 	pml4 = (pml4_entry_t *)PHYS_TO_DMAP(*pml5 & PG_FRAME);
4451 	pml4 = &pml4[pmap_pml4e_index(va)];
4452 	if ((*pml4 & PG_V) == 0) {
4453 		pml4pg = PHYS_TO_VM_PAGE(*pml5 & PG_FRAME);
4454 		if (allocated && !addref)
4455 			pml4pg->ref_count--;
4456 		else if (!allocated && addref)
4457 			pml4pg->ref_count++;
4458 	}
4459 	return (pml4);
4460 }
4461 
4462 static pdp_entry_t *
pmap_allocpte_getpdp(pmap_t pmap,struct rwlock ** lockp,vm_offset_t va,bool addref)4463 pmap_allocpte_getpdp(pmap_t pmap, struct rwlock **lockp, vm_offset_t va,
4464     bool addref)
4465 {
4466 	vm_page_t pdppg;
4467 	pml4_entry_t *pml4;
4468 	pdp_entry_t *pdp;
4469 	pt_entry_t PG_V;
4470 	bool allocated;
4471 
4472 	PG_V = pmap_valid_bit(pmap);
4473 
4474 	pml4 = pmap_allocpte_getpml4(pmap, lockp, va, false);
4475 	if (pml4 == NULL)
4476 		return (NULL);
4477 
4478 	if ((*pml4 & PG_V) == 0) {
4479 		/* Have to allocate a new pdp, recurse */
4480 		if (pmap_allocpte_nosleep(pmap, pmap_pml4e_pindex(va), lockp,
4481 		    va) == NULL) {
4482 			if (pmap_is_la57(pmap))
4483 				pmap_allocpte_free_unref(pmap, va,
4484 				    pmap_pml5e(pmap, va));
4485 			return (NULL);
4486 		}
4487 		allocated = true;
4488 	} else {
4489 		allocated = false;
4490 	}
4491 	pdp = (pdp_entry_t *)PHYS_TO_DMAP(*pml4 & PG_FRAME);
4492 	pdp = &pdp[pmap_pdpe_index(va)];
4493 	if ((*pdp & PG_V) == 0) {
4494 		pdppg = PHYS_TO_VM_PAGE(*pml4 & PG_FRAME);
4495 		if (allocated && !addref)
4496 			pdppg->ref_count--;
4497 		else if (!allocated && addref)
4498 			pdppg->ref_count++;
4499 	}
4500 	return (pdp);
4501 }
4502 
4503 /*
4504  * The ptepindexes, i.e. page indices, of the page table pages encountered
4505  * while translating virtual address va are defined as follows:
4506  * - for the page table page (last level),
4507  *      ptepindex = pmap_pde_pindex(va) = va >> PDRSHIFT,
4508  *   in other words, it is just the index of the PDE that maps the page
4509  *   table page.
4510  * - for the page directory page,
4511  *      ptepindex = NUPDE (number of userland PD entries) +
4512  *          (pmap_pde_index(va) >> NPDEPGSHIFT)
4513  *   i.e. index of PDPE is put after the last index of PDE,
4514  * - for the page directory pointer page,
4515  *      ptepindex = NUPDE + NUPDPE + (pmap_pde_index(va) >> (NPDEPGSHIFT +
4516  *          NPML4EPGSHIFT),
4517  *   i.e. index of pml4e is put after the last index of PDPE,
4518  * - for the PML4 page (if LA57 mode is enabled),
4519  *      ptepindex = NUPDE + NUPDPE + NUPML4E + (pmap_pde_index(va) >>
4520  *          (NPDEPGSHIFT + NPML4EPGSHIFT + NPML5EPGSHIFT),
4521  *   i.e. index of pml5e is put after the last index of PML4E.
4522  *
4523  * Define an order on the paging entries, where all entries of the
4524  * same height are put together, then heights are put from deepest to
4525  * root.  Then ptexpindex is the sequential number of the
4526  * corresponding paging entry in this order.
4527  *
4528  * The values of NUPDE, NUPDPE, and NUPML4E are determined by the size of
4529  * LA57 paging structures even in LA48 paging mode. Moreover, the
4530  * ptepindexes are calculated as if the paging structures were 5-level
4531  * regardless of the actual mode of operation.
4532  *
4533  * The root page at PML4/PML5 does not participate in this indexing scheme,
4534  * since it is statically allocated by pmap_pinit() and not by pmap_allocpte().
4535  */
4536 static vm_page_t
pmap_allocpte_nosleep(pmap_t pmap,vm_pindex_t ptepindex,struct rwlock ** lockp,vm_offset_t va)4537 pmap_allocpte_nosleep(pmap_t pmap, vm_pindex_t ptepindex, struct rwlock **lockp,
4538     vm_offset_t va)
4539 {
4540 	vm_pindex_t pml5index, pml4index;
4541 	pml5_entry_t *pml5, *pml5u;
4542 	pml4_entry_t *pml4, *pml4u;
4543 	pdp_entry_t *pdp;
4544 	pd_entry_t *pd;
4545 	vm_page_t m, pdpg;
4546 	pt_entry_t PG_A, PG_M, PG_RW, PG_V;
4547 
4548 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
4549 
4550 	PG_A = pmap_accessed_bit(pmap);
4551 	PG_M = pmap_modified_bit(pmap);
4552 	PG_V = pmap_valid_bit(pmap);
4553 	PG_RW = pmap_rw_bit(pmap);
4554 
4555 	/*
4556 	 * Allocate a page table page.
4557 	 */
4558 	m = pmap_alloc_pt_page(pmap, ptepindex,
4559 	    VM_ALLOC_WIRED | VM_ALLOC_ZERO);
4560 	if (m == NULL)
4561 		return (NULL);
4562 
4563 	/*
4564 	 * Map the pagetable page into the process address space, if
4565 	 * it isn't already there.
4566 	 */
4567 	if (ptepindex >= NUPDE + NUPDPE + NUPML4E) {
4568 		MPASS(pmap_is_la57(pmap));
4569 
4570 		pml5index = pmap_pml5e_index(va);
4571 		pml5 = &pmap->pm_pmltop[pml5index];
4572 		KASSERT((*pml5 & PG_V) == 0,
4573 		    ("pmap %p va %#lx pml5 %#lx", pmap, va, *pml5));
4574 		*pml5 = VM_PAGE_TO_PHYS(m) | PG_U | PG_RW | PG_V | PG_A | PG_M;
4575 
4576 		if (pmap->pm_pmltopu != NULL && pml5index < NUPML5E) {
4577 			if (pmap->pm_ucr3 != PMAP_NO_CR3)
4578 				*pml5 |= pg_nx;
4579 
4580 			pml5u = &pmap->pm_pmltopu[pml5index];
4581 			*pml5u = VM_PAGE_TO_PHYS(m) | PG_U | PG_RW | PG_V |
4582 			    PG_A | PG_M;
4583 		}
4584 	} else if (ptepindex >= NUPDE + NUPDPE) {
4585 		pml4index = pmap_pml4e_index(va);
4586 		/* Wire up a new PDPE page */
4587 		pml4 = pmap_allocpte_getpml4(pmap, lockp, va, true);
4588 		if (pml4 == NULL) {
4589 			pmap_free_pt_page(pmap, m, true);
4590 			return (NULL);
4591 		}
4592 		KASSERT((*pml4 & PG_V) == 0,
4593 		    ("pmap %p va %#lx pml4 %#lx", pmap, va, *pml4));
4594 		*pml4 = VM_PAGE_TO_PHYS(m) | PG_U | PG_RW | PG_V | PG_A | PG_M;
4595 
4596 		if (!pmap_is_la57(pmap) && pmap->pm_pmltopu != NULL &&
4597 		    pml4index < NUPML4E) {
4598 			/*
4599 			 * PTI: Make all user-space mappings in the
4600 			 * kernel-mode page table no-execute so that
4601 			 * we detect any programming errors that leave
4602 			 * the kernel-mode page table active on return
4603 			 * to user space.
4604 			 */
4605 			if (pmap->pm_ucr3 != PMAP_NO_CR3)
4606 				*pml4 |= pg_nx;
4607 
4608 			pml4u = &pmap->pm_pmltopu[pml4index];
4609 			*pml4u = VM_PAGE_TO_PHYS(m) | PG_U | PG_RW | PG_V |
4610 			    PG_A | PG_M;
4611 		}
4612 	} else if (ptepindex >= NUPDE) {
4613 		/* Wire up a new PDE page */
4614 		pdp = pmap_allocpte_getpdp(pmap, lockp, va, true);
4615 		if (pdp == NULL) {
4616 			pmap_free_pt_page(pmap, m, true);
4617 			return (NULL);
4618 		}
4619 		KASSERT((*pdp & PG_V) == 0,
4620 		    ("pmap %p va %#lx pdp %#lx", pmap, va, *pdp));
4621 		*pdp = VM_PAGE_TO_PHYS(m) | PG_U | PG_RW | PG_V | PG_A | PG_M;
4622 	} else {
4623 		/* Wire up a new PTE page */
4624 		pdp = pmap_allocpte_getpdp(pmap, lockp, va, false);
4625 		if (pdp == NULL) {
4626 			pmap_free_pt_page(pmap, m, true);
4627 			return (NULL);
4628 		}
4629 		if ((*pdp & PG_V) == 0) {
4630 			/* Have to allocate a new pd, recurse */
4631 		  if (pmap_allocpte_nosleep(pmap, pmap_pdpe_pindex(va),
4632 		      lockp, va) == NULL) {
4633 				pmap_allocpte_free_unref(pmap, va,
4634 				    pmap_pml4e(pmap, va));
4635 				pmap_free_pt_page(pmap, m, true);
4636 				return (NULL);
4637 			}
4638 		} else {
4639 			/* Add reference to the pd page */
4640 			pdpg = PHYS_TO_VM_PAGE(*pdp & PG_FRAME);
4641 			pdpg->ref_count++;
4642 		}
4643 		pd = (pd_entry_t *)PHYS_TO_DMAP(*pdp & PG_FRAME);
4644 
4645 		/* Now we know where the page directory page is */
4646 		pd = &pd[pmap_pde_index(va)];
4647 		KASSERT((*pd & PG_V) == 0,
4648 		    ("pmap %p va %#lx pd %#lx", pmap, va, *pd));
4649 		*pd = VM_PAGE_TO_PHYS(m) | PG_U | PG_RW | PG_V | PG_A | PG_M;
4650 	}
4651 
4652 	return (m);
4653 }
4654 
4655 /*
4656  * This routine is called if the desired page table page does not exist.
4657  *
4658  * If page table page allocation fails, this routine may sleep before
4659  * returning NULL.  It sleeps only if a lock pointer was given.  Sleep
4660  * occurs right before returning to the caller. This way, we never
4661  * drop pmap lock to sleep while a page table page has ref_count == 0,
4662  * which prevents the page from being freed under us.
4663  */
4664 static vm_page_t
pmap_allocpte_alloc(pmap_t pmap,vm_pindex_t ptepindex,struct rwlock ** lockp,vm_offset_t va)4665 pmap_allocpte_alloc(pmap_t pmap, vm_pindex_t ptepindex, struct rwlock **lockp,
4666     vm_offset_t va)
4667 {
4668 	vm_page_t m;
4669 
4670 	m = pmap_allocpte_nosleep(pmap, ptepindex, lockp, va);
4671 	if (m == NULL && lockp != NULL) {
4672 		RELEASE_PV_LIST_LOCK(lockp);
4673 		PMAP_UNLOCK(pmap);
4674 		PMAP_ASSERT_NOT_IN_DI();
4675 		vm_wait(NULL);
4676 		PMAP_LOCK(pmap);
4677 	}
4678 	return (m);
4679 }
4680 
4681 static pd_entry_t *
pmap_alloc_pde(pmap_t pmap,vm_offset_t va,vm_page_t * pdpgp,struct rwlock ** lockp)4682 pmap_alloc_pde(pmap_t pmap, vm_offset_t va, vm_page_t *pdpgp,
4683     struct rwlock **lockp)
4684 {
4685 	pdp_entry_t *pdpe, PG_V;
4686 	pd_entry_t *pde;
4687 	vm_page_t pdpg;
4688 	vm_pindex_t pdpindex;
4689 
4690 	PG_V = pmap_valid_bit(pmap);
4691 
4692 retry:
4693 	pdpe = pmap_pdpe(pmap, va);
4694 	if (pdpe != NULL && (*pdpe & PG_V) != 0) {
4695 		pde = pmap_pdpe_to_pde(pdpe, va);
4696 		if (va < VM_MAXUSER_ADDRESS) {
4697 			/* Add a reference to the pd page. */
4698 			pdpg = PHYS_TO_VM_PAGE(*pdpe & PG_FRAME);
4699 			pdpg->ref_count++;
4700 		} else
4701 			pdpg = NULL;
4702 	} else if (va < VM_MAXUSER_ADDRESS) {
4703 		/* Allocate a pd page. */
4704 		pdpindex = pmap_pde_pindex(va) >> NPDPEPGSHIFT;
4705 		pdpg = pmap_allocpte_alloc(pmap, NUPDE + pdpindex, lockp, va);
4706 		if (pdpg == NULL) {
4707 			if (lockp != NULL)
4708 				goto retry;
4709 			else
4710 				return (NULL);
4711 		}
4712 		pde = (pd_entry_t *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(pdpg));
4713 		pde = &pde[pmap_pde_index(va)];
4714 	} else
4715 		panic("pmap_alloc_pde: missing page table page for va %#lx",
4716 		    va);
4717 	*pdpgp = pdpg;
4718 	return (pde);
4719 }
4720 
4721 static vm_page_t
pmap_allocpte(pmap_t pmap,vm_offset_t va,struct rwlock ** lockp)4722 pmap_allocpte(pmap_t pmap, vm_offset_t va, struct rwlock **lockp)
4723 {
4724 	vm_pindex_t ptepindex;
4725 	pd_entry_t *pd, PG_V;
4726 	vm_page_t m;
4727 
4728 	PG_V = pmap_valid_bit(pmap);
4729 
4730 	/*
4731 	 * Calculate pagetable page index
4732 	 */
4733 	ptepindex = pmap_pde_pindex(va);
4734 retry:
4735 	/*
4736 	 * Get the page directory entry
4737 	 */
4738 	pd = pmap_pde(pmap, va);
4739 
4740 	/*
4741 	 * This supports switching from a 2MB page to a
4742 	 * normal 4K page.
4743 	 */
4744 	if (pd != NULL && (*pd & (PG_PS | PG_V)) == (PG_PS | PG_V)) {
4745 		if (!pmap_demote_pde_locked(pmap, pd, va, lockp)) {
4746 			/*
4747 			 * Invalidation of the 2MB page mapping may have caused
4748 			 * the deallocation of the underlying PD page.
4749 			 */
4750 			pd = NULL;
4751 		}
4752 	}
4753 
4754 	/*
4755 	 * If the page table page is mapped, we just increment the
4756 	 * hold count, and activate it.
4757 	 */
4758 	if (pd != NULL && (*pd & PG_V) != 0) {
4759 		m = PHYS_TO_VM_PAGE(*pd & PG_FRAME);
4760 		m->ref_count++;
4761 	} else {
4762 		/*
4763 		 * Here if the pte page isn't mapped, or if it has been
4764 		 * deallocated.
4765 		 */
4766 		m = pmap_allocpte_alloc(pmap, ptepindex, lockp, va);
4767 		if (m == NULL && lockp != NULL)
4768 			goto retry;
4769 	}
4770 	return (m);
4771 }
4772 
4773 /***************************************************
4774  * Pmap allocation/deallocation routines.
4775  ***************************************************/
4776 
4777 /*
4778  * Release any resources held by the given physical map.
4779  * Called when a pmap initialized by pmap_pinit is being released.
4780  * Should only be called if the map contains no valid mappings.
4781  */
4782 void
pmap_release(pmap_t pmap)4783 pmap_release(pmap_t pmap)
4784 {
4785 	vm_page_t m;
4786 	int i;
4787 
4788 	KASSERT(vm_radix_is_empty(&pmap->pm_root),
4789 	    ("pmap_release: pmap %p has reserved page table page(s)",
4790 	    pmap));
4791 	KASSERT(CPU_EMPTY(&pmap->pm_active),
4792 	    ("releasing active pmap %p", pmap));
4793 
4794 	m = PHYS_TO_VM_PAGE(DMAP_TO_PHYS((vm_offset_t)pmap->pm_pmltop));
4795 
4796 	if (pmap_is_la57(pmap)) {
4797 		pmap->pm_pmltop[pmap_pml5e_index(UPT_MAX_ADDRESS)] = 0;
4798 		pmap->pm_pmltop[PML5PML5I] = 0;
4799 	} else {
4800 		for (i = 0; i < NKPML4E; i++)	/* KVA */
4801 			pmap->pm_pmltop[KPML4BASE + i] = 0;
4802 #ifdef KASAN
4803 		for (i = 0; i < NKASANPML4E; i++) /* KASAN shadow map */
4804 			pmap->pm_pmltop[KASANPML4I + i] = 0;
4805 #endif
4806 		for (i = 0; i < ndmpdpphys; i++)/* Direct Map */
4807 			pmap->pm_pmltop[DMPML4I + i] = 0;
4808 		pmap->pm_pmltop[PML4PML4I] = 0;	/* Recursive Mapping */
4809 		for (i = 0; i < lm_ents; i++)	/* Large Map */
4810 			pmap->pm_pmltop[LMSPML4I + i] = 0;
4811 	}
4812 
4813 	pmap_free_pt_page(NULL, m, true);
4814 	pmap_pt_page_count_pinit(pmap, -1);
4815 
4816 	if (pmap->pm_pmltopu != NULL) {
4817 		m = PHYS_TO_VM_PAGE(DMAP_TO_PHYS((vm_offset_t)pmap->
4818 		    pm_pmltopu));
4819 		pmap_free_pt_page(NULL, m, false);
4820 		pmap_pt_page_count_pinit(pmap, -1);
4821 	}
4822 	if (pmap->pm_type == PT_X86 &&
4823 	    (cpu_stdext_feature2 & CPUID_STDEXT2_PKU) != 0)
4824 		rangeset_fini(&pmap->pm_pkru);
4825 
4826 	KASSERT(pmap->pm_stats.resident_count == 0,
4827 	    ("pmap_release: pmap %p resident count %ld != 0",
4828 	    pmap, pmap->pm_stats.resident_count));
4829 }
4830 
4831 static int
kvm_size(SYSCTL_HANDLER_ARGS)4832 kvm_size(SYSCTL_HANDLER_ARGS)
4833 {
4834 	unsigned long ksize = VM_MAX_KERNEL_ADDRESS - VM_MIN_KERNEL_ADDRESS;
4835 
4836 	return sysctl_handle_long(oidp, &ksize, 0, req);
4837 }
4838 SYSCTL_PROC(_vm, OID_AUTO, kvm_size, CTLTYPE_LONG | CTLFLAG_RD | CTLFLAG_MPSAFE,
4839     0, 0, kvm_size, "LU",
4840     "Size of KVM");
4841 
4842 static int
kvm_free(SYSCTL_HANDLER_ARGS)4843 kvm_free(SYSCTL_HANDLER_ARGS)
4844 {
4845 	unsigned long kfree = VM_MAX_KERNEL_ADDRESS - kernel_vm_end;
4846 
4847 	return sysctl_handle_long(oidp, &kfree, 0, req);
4848 }
4849 SYSCTL_PROC(_vm, OID_AUTO, kvm_free, CTLTYPE_LONG | CTLFLAG_RD | CTLFLAG_MPSAFE,
4850     0, 0, kvm_free, "LU",
4851     "Amount of KVM free");
4852 
4853 /*
4854  * Allocate physical memory for the vm_page array and map it into KVA,
4855  * attempting to back the vm_pages with domain-local memory.
4856  */
4857 void
pmap_page_array_startup(long pages)4858 pmap_page_array_startup(long pages)
4859 {
4860 	pdp_entry_t *pdpe;
4861 	pd_entry_t *pde, newpdir;
4862 	vm_offset_t va, start, end;
4863 	vm_paddr_t pa;
4864 	long pfn;
4865 	int domain, i;
4866 
4867 	vm_page_array_size = pages;
4868 
4869 	start = VM_MIN_KERNEL_ADDRESS;
4870 	end = start + pages * sizeof(struct vm_page);
4871 	for (va = start; va < end; va += NBPDR) {
4872 		pfn = first_page + (va - start) / sizeof(struct vm_page);
4873 		domain = vm_phys_domain(ptoa(pfn));
4874 		pdpe = pmap_pdpe(kernel_pmap, va);
4875 		if ((*pdpe & X86_PG_V) == 0) {
4876 			pa = vm_phys_early_alloc(domain, PAGE_SIZE);
4877 			dump_add_page(pa);
4878 			pagezero((void *)PHYS_TO_DMAP(pa));
4879 			*pdpe = (pdp_entry_t)(pa | X86_PG_V | X86_PG_RW |
4880 			    X86_PG_A | X86_PG_M);
4881 		}
4882 		pde = pmap_pdpe_to_pde(pdpe, va);
4883 		if ((*pde & X86_PG_V) != 0)
4884 			panic("Unexpected pde");
4885 		pa = vm_phys_early_alloc(domain, NBPDR);
4886 		for (i = 0; i < NPDEPG; i++)
4887 			dump_add_page(pa + i * PAGE_SIZE);
4888 		newpdir = (pd_entry_t)(pa | X86_PG_V | X86_PG_RW | X86_PG_A |
4889 		    X86_PG_M | PG_PS | pg_g | pg_nx);
4890 		pde_store(pde, newpdir);
4891 	}
4892 	vm_page_array = (vm_page_t)start;
4893 }
4894 
4895 /*
4896  * grow the number of kernel page table entries, if needed
4897  */
4898 void
pmap_growkernel(vm_offset_t addr)4899 pmap_growkernel(vm_offset_t addr)
4900 {
4901 	vm_paddr_t paddr;
4902 	vm_page_t nkpg;
4903 	pd_entry_t *pde, newpdir;
4904 	pdp_entry_t *pdpe;
4905 	vm_offset_t end;
4906 
4907 	mtx_assert(&kernel_map->system_mtx, MA_OWNED);
4908 
4909 	/*
4910 	 * The kernel map covers two distinct regions of KVA: that used
4911 	 * for dynamic kernel memory allocations, and the uppermost 2GB
4912 	 * of the virtual address space.  The latter is used to map the
4913 	 * kernel and loadable kernel modules.  This scheme enables the
4914 	 * use of a special code generation model for kernel code which
4915 	 * takes advantage of compact addressing modes in machine code.
4916 	 *
4917 	 * Both regions grow upwards; to avoid wasting memory, the gap
4918 	 * in between is unmapped.  If "addr" is above "KERNBASE", the
4919 	 * kernel's region is grown, otherwise the kmem region is grown.
4920 	 *
4921 	 * The correctness of this action is based on the following
4922 	 * argument: vm_map_insert() allocates contiguous ranges of the
4923 	 * kernel virtual address space.  It calls this function if a range
4924 	 * ends after "kernel_vm_end".  If the kernel is mapped between
4925 	 * "kernel_vm_end" and "addr", then the range cannot begin at
4926 	 * "kernel_vm_end".  In fact, its beginning address cannot be less
4927 	 * than the kernel.  Thus, there is no immediate need to allocate
4928 	 * any new kernel page table pages between "kernel_vm_end" and
4929 	 * "KERNBASE".
4930 	 */
4931 	if (KERNBASE < addr) {
4932 		end = KERNBASE + nkpt * NBPDR;
4933 		if (end == 0)
4934 			return;
4935 	} else {
4936 		end = kernel_vm_end;
4937 	}
4938 
4939 	addr = roundup2(addr, NBPDR);
4940 	if (addr - 1 >= vm_map_max(kernel_map))
4941 		addr = vm_map_max(kernel_map);
4942 	if (addr <= end) {
4943 		/*
4944 		 * The grown region is already mapped, so there is
4945 		 * nothing to do.
4946 		 */
4947 		return;
4948 	}
4949 
4950 	kasan_shadow_map(end, addr - end);
4951 	while (end < addr) {
4952 		pdpe = pmap_pdpe(kernel_pmap, end);
4953 		if ((*pdpe & X86_PG_V) == 0) {
4954 			nkpg = pmap_alloc_pt_page(kernel_pmap,
4955 			    pmap_pdpe_pindex(end), VM_ALLOC_WIRED |
4956 			    VM_ALLOC_INTERRUPT | VM_ALLOC_ZERO);
4957 			if (nkpg == NULL)
4958 				panic("pmap_growkernel: no memory to grow kernel");
4959 			paddr = VM_PAGE_TO_PHYS(nkpg);
4960 			*pdpe = (pdp_entry_t)(paddr | X86_PG_V | X86_PG_RW |
4961 			    X86_PG_A | X86_PG_M);
4962 			continue; /* try again */
4963 		}
4964 		pde = pmap_pdpe_to_pde(pdpe, end);
4965 		if ((*pde & X86_PG_V) != 0) {
4966 			end = (end + NBPDR) & ~PDRMASK;
4967 			if (end - 1 >= vm_map_max(kernel_map)) {
4968 				end = vm_map_max(kernel_map);
4969 				break;
4970 			}
4971 			continue;
4972 		}
4973 
4974 		nkpg = pmap_alloc_pt_page(kernel_pmap, pmap_pde_pindex(end),
4975 		    VM_ALLOC_WIRED | VM_ALLOC_INTERRUPT | VM_ALLOC_ZERO);
4976 		if (nkpg == NULL)
4977 			panic("pmap_growkernel: no memory to grow kernel");
4978 		paddr = VM_PAGE_TO_PHYS(nkpg);
4979 		newpdir = paddr | X86_PG_V | X86_PG_RW | X86_PG_A | X86_PG_M;
4980 		pde_store(pde, newpdir);
4981 
4982 		end = (end + NBPDR) & ~PDRMASK;
4983 		if (end - 1 >= vm_map_max(kernel_map)) {
4984 			end = vm_map_max(kernel_map);
4985 			break;
4986 		}
4987 	}
4988 
4989 	if (end <= KERNBASE)
4990 		kernel_vm_end = end;
4991 	else
4992 		nkpt = howmany(end - KERNBASE, NBPDR);
4993 }
4994 
4995 /***************************************************
4996  * page management routines.
4997  ***************************************************/
4998 
4999 CTASSERT(sizeof(struct pv_chunk) == PAGE_SIZE);
5000 CTASSERT(_NPCM == 3);
5001 CTASSERT(_NPCPV == 168);
5002 
5003 static __inline struct pv_chunk *
pv_to_chunk(pv_entry_t pv)5004 pv_to_chunk(pv_entry_t pv)
5005 {
5006 
5007 	return ((struct pv_chunk *)((uintptr_t)pv & ~(uintptr_t)PAGE_MASK));
5008 }
5009 
5010 #define PV_PMAP(pv) (pv_to_chunk(pv)->pc_pmap)
5011 
5012 #define	PC_FREE0	0xfffffffffffffffful
5013 #define	PC_FREE1	0xfffffffffffffffful
5014 #define	PC_FREE2	0x000000fffffffffful
5015 
5016 static const uint64_t pc_freemask[_NPCM] = { PC_FREE0, PC_FREE1, PC_FREE2 };
5017 
5018 #ifdef PV_STATS
5019 
5020 static COUNTER_U64_DEFINE_EARLY(pc_chunk_count);
5021 SYSCTL_COUNTER_U64(_vm_pmap, OID_AUTO, pc_chunk_count, CTLFLAG_RD,
5022     &pc_chunk_count, "Current number of pv entry cnunks");
5023 
5024 static COUNTER_U64_DEFINE_EARLY(pc_chunk_allocs);
5025 SYSCTL_COUNTER_U64(_vm_pmap, OID_AUTO, pc_chunk_allocs, CTLFLAG_RD,
5026     &pc_chunk_allocs, "Total number of pv entry chunks allocated");
5027 
5028 static COUNTER_U64_DEFINE_EARLY(pc_chunk_frees);
5029 SYSCTL_COUNTER_U64(_vm_pmap, OID_AUTO, pc_chunk_frees, CTLFLAG_RD,
5030     &pc_chunk_frees, "Total number of pv entry chunks freed");
5031 
5032 static COUNTER_U64_DEFINE_EARLY(pc_chunk_tryfail);
5033 SYSCTL_COUNTER_U64(_vm_pmap, OID_AUTO, pc_chunk_tryfail, CTLFLAG_RD,
5034     &pc_chunk_tryfail,
5035     "Number of failed attempts to get a pv entry chunk page");
5036 
5037 static COUNTER_U64_DEFINE_EARLY(pv_entry_frees);
5038 SYSCTL_COUNTER_U64(_vm_pmap, OID_AUTO, pv_entry_frees, CTLFLAG_RD,
5039     &pv_entry_frees, "Total number of pv entries freed");
5040 
5041 static COUNTER_U64_DEFINE_EARLY(pv_entry_allocs);
5042 SYSCTL_COUNTER_U64(_vm_pmap, OID_AUTO, pv_entry_allocs, CTLFLAG_RD,
5043     &pv_entry_allocs, "Total number of pv entries allocated");
5044 
5045 static COUNTER_U64_DEFINE_EARLY(pv_entry_count);
5046 SYSCTL_COUNTER_U64(_vm_pmap, OID_AUTO, pv_entry_count, CTLFLAG_RD,
5047     &pv_entry_count, "Current number of pv entries");
5048 
5049 static COUNTER_U64_DEFINE_EARLY(pv_entry_spare);
5050 SYSCTL_COUNTER_U64(_vm_pmap, OID_AUTO, pv_entry_spare, CTLFLAG_RD,
5051     &pv_entry_spare, "Current number of spare pv entries");
5052 #endif
5053 
5054 static void
reclaim_pv_chunk_leave_pmap(pmap_t pmap,pmap_t locked_pmap,bool start_di)5055 reclaim_pv_chunk_leave_pmap(pmap_t pmap, pmap_t locked_pmap, bool start_di)
5056 {
5057 
5058 	if (pmap == NULL)
5059 		return;
5060 	pmap_invalidate_all(pmap);
5061 	if (pmap != locked_pmap)
5062 		PMAP_UNLOCK(pmap);
5063 	if (start_di)
5064 		pmap_delayed_invl_finish();
5065 }
5066 
5067 /*
5068  * We are in a serious low memory condition.  Resort to
5069  * drastic measures to free some pages so we can allocate
5070  * another pv entry chunk.
5071  *
5072  * Returns NULL if PV entries were reclaimed from the specified pmap.
5073  *
5074  * We do not, however, unmap 2mpages because subsequent accesses will
5075  * allocate per-page pv entries until repromotion occurs, thereby
5076  * exacerbating the shortage of free pv entries.
5077  */
5078 static vm_page_t
reclaim_pv_chunk_domain(pmap_t locked_pmap,struct rwlock ** lockp,int domain)5079 reclaim_pv_chunk_domain(pmap_t locked_pmap, struct rwlock **lockp, int domain)
5080 {
5081 	struct pv_chunks_list *pvc;
5082 	struct pv_chunk *pc, *pc_marker, *pc_marker_end;
5083 	struct pv_chunk_header pc_marker_b, pc_marker_end_b;
5084 	struct md_page *pvh;
5085 	pd_entry_t *pde;
5086 	pmap_t next_pmap, pmap;
5087 	pt_entry_t *pte, tpte;
5088 	pt_entry_t PG_G, PG_A, PG_M, PG_RW;
5089 	pv_entry_t pv;
5090 	vm_offset_t va;
5091 	vm_page_t m, m_pc;
5092 	struct spglist free;
5093 	uint64_t inuse;
5094 	int bit, field, freed;
5095 	bool start_di, restart;
5096 
5097 	PMAP_LOCK_ASSERT(locked_pmap, MA_OWNED);
5098 	KASSERT(lockp != NULL, ("reclaim_pv_chunk: lockp is NULL"));
5099 	pmap = NULL;
5100 	m_pc = NULL;
5101 	PG_G = PG_A = PG_M = PG_RW = 0;
5102 	SLIST_INIT(&free);
5103 	bzero(&pc_marker_b, sizeof(pc_marker_b));
5104 	bzero(&pc_marker_end_b, sizeof(pc_marker_end_b));
5105 	pc_marker = (struct pv_chunk *)&pc_marker_b;
5106 	pc_marker_end = (struct pv_chunk *)&pc_marker_end_b;
5107 
5108 	/*
5109 	 * A delayed invalidation block should already be active if
5110 	 * pmap_advise() or pmap_remove() called this function by way
5111 	 * of pmap_demote_pde_locked().
5112 	 */
5113 	start_di = pmap_not_in_di();
5114 
5115 	pvc = &pv_chunks[domain];
5116 	mtx_lock(&pvc->pvc_lock);
5117 	pvc->active_reclaims++;
5118 	TAILQ_INSERT_HEAD(&pvc->pvc_list, pc_marker, pc_lru);
5119 	TAILQ_INSERT_TAIL(&pvc->pvc_list, pc_marker_end, pc_lru);
5120 	while ((pc = TAILQ_NEXT(pc_marker, pc_lru)) != pc_marker_end &&
5121 	    SLIST_EMPTY(&free)) {
5122 		next_pmap = pc->pc_pmap;
5123 		if (next_pmap == NULL) {
5124 			/*
5125 			 * The next chunk is a marker.  However, it is
5126 			 * not our marker, so active_reclaims must be
5127 			 * > 1.  Consequently, the next_chunk code
5128 			 * will not rotate the pv_chunks list.
5129 			 */
5130 			goto next_chunk;
5131 		}
5132 		mtx_unlock(&pvc->pvc_lock);
5133 
5134 		/*
5135 		 * A pv_chunk can only be removed from the pc_lru list
5136 		 * when both pc_chunks_mutex is owned and the
5137 		 * corresponding pmap is locked.
5138 		 */
5139 		if (pmap != next_pmap) {
5140 			restart = false;
5141 			reclaim_pv_chunk_leave_pmap(pmap, locked_pmap,
5142 			    start_di);
5143 			pmap = next_pmap;
5144 			/* Avoid deadlock and lock recursion. */
5145 			if (pmap > locked_pmap) {
5146 				RELEASE_PV_LIST_LOCK(lockp);
5147 				PMAP_LOCK(pmap);
5148 				if (start_di)
5149 					pmap_delayed_invl_start();
5150 				mtx_lock(&pvc->pvc_lock);
5151 				restart = true;
5152 			} else if (pmap != locked_pmap) {
5153 				if (PMAP_TRYLOCK(pmap)) {
5154 					if (start_di)
5155 						pmap_delayed_invl_start();
5156 					mtx_lock(&pvc->pvc_lock);
5157 					restart = true;
5158 				} else {
5159 					pmap = NULL; /* pmap is not locked */
5160 					mtx_lock(&pvc->pvc_lock);
5161 					pc = TAILQ_NEXT(pc_marker, pc_lru);
5162 					if (pc == NULL ||
5163 					    pc->pc_pmap != next_pmap)
5164 						continue;
5165 					goto next_chunk;
5166 				}
5167 			} else if (start_di)
5168 				pmap_delayed_invl_start();
5169 			PG_G = pmap_global_bit(pmap);
5170 			PG_A = pmap_accessed_bit(pmap);
5171 			PG_M = pmap_modified_bit(pmap);
5172 			PG_RW = pmap_rw_bit(pmap);
5173 			if (restart)
5174 				continue;
5175 		}
5176 
5177 		/*
5178 		 * Destroy every non-wired, 4 KB page mapping in the chunk.
5179 		 */
5180 		freed = 0;
5181 		for (field = 0; field < _NPCM; field++) {
5182 			for (inuse = ~pc->pc_map[field] & pc_freemask[field];
5183 			    inuse != 0; inuse &= ~(1UL << bit)) {
5184 				bit = bsfq(inuse);
5185 				pv = &pc->pc_pventry[field * 64 + bit];
5186 				va = pv->pv_va;
5187 				pde = pmap_pde(pmap, va);
5188 				if ((*pde & PG_PS) != 0)
5189 					continue;
5190 				pte = pmap_pde_to_pte(pde, va);
5191 				if ((*pte & PG_W) != 0)
5192 					continue;
5193 				tpte = pte_load_clear(pte);
5194 				if ((tpte & PG_G) != 0)
5195 					pmap_invalidate_page(pmap, va);
5196 				m = PHYS_TO_VM_PAGE(tpte & PG_FRAME);
5197 				if ((tpte & (PG_M | PG_RW)) == (PG_M | PG_RW))
5198 					vm_page_dirty(m);
5199 				if ((tpte & PG_A) != 0)
5200 					vm_page_aflag_set(m, PGA_REFERENCED);
5201 				CHANGE_PV_LIST_LOCK_TO_VM_PAGE(lockp, m);
5202 				TAILQ_REMOVE(&m->md.pv_list, pv, pv_next);
5203 				m->md.pv_gen++;
5204 				if (TAILQ_EMPTY(&m->md.pv_list) &&
5205 				    (m->flags & PG_FICTITIOUS) == 0) {
5206 					pvh = pa_to_pvh(VM_PAGE_TO_PHYS(m));
5207 					if (TAILQ_EMPTY(&pvh->pv_list)) {
5208 						vm_page_aflag_clear(m,
5209 						    PGA_WRITEABLE);
5210 					}
5211 				}
5212 				pmap_delayed_invl_page(m);
5213 				pc->pc_map[field] |= 1UL << bit;
5214 				pmap_unuse_pt(pmap, va, *pde, &free);
5215 				freed++;
5216 			}
5217 		}
5218 		if (freed == 0) {
5219 			mtx_lock(&pvc->pvc_lock);
5220 			goto next_chunk;
5221 		}
5222 		/* Every freed mapping is for a 4 KB page. */
5223 		pmap_resident_count_adj(pmap, -freed);
5224 		PV_STAT(counter_u64_add(pv_entry_frees, freed));
5225 		PV_STAT(counter_u64_add(pv_entry_spare, freed));
5226 		PV_STAT(counter_u64_add(pv_entry_count, -freed));
5227 		TAILQ_REMOVE(&pmap->pm_pvchunk, pc, pc_list);
5228 		if (pc->pc_map[0] == PC_FREE0 && pc->pc_map[1] == PC_FREE1 &&
5229 		    pc->pc_map[2] == PC_FREE2) {
5230 			PV_STAT(counter_u64_add(pv_entry_spare, -_NPCPV));
5231 			PV_STAT(counter_u64_add(pc_chunk_count, -1));
5232 			PV_STAT(counter_u64_add(pc_chunk_frees, 1));
5233 			/* Entire chunk is free; return it. */
5234 			m_pc = PHYS_TO_VM_PAGE(DMAP_TO_PHYS((vm_offset_t)pc));
5235 			dump_drop_page(m_pc->phys_addr);
5236 			mtx_lock(&pvc->pvc_lock);
5237 			TAILQ_REMOVE(&pvc->pvc_list, pc, pc_lru);
5238 			break;
5239 		}
5240 		TAILQ_INSERT_HEAD(&pmap->pm_pvchunk, pc, pc_list);
5241 		mtx_lock(&pvc->pvc_lock);
5242 		/* One freed pv entry in locked_pmap is sufficient. */
5243 		if (pmap == locked_pmap)
5244 			break;
5245 next_chunk:
5246 		TAILQ_REMOVE(&pvc->pvc_list, pc_marker, pc_lru);
5247 		TAILQ_INSERT_AFTER(&pvc->pvc_list, pc, pc_marker, pc_lru);
5248 		if (pvc->active_reclaims == 1 && pmap != NULL) {
5249 			/*
5250 			 * Rotate the pv chunks list so that we do not
5251 			 * scan the same pv chunks that could not be
5252 			 * freed (because they contained a wired
5253 			 * and/or superpage mapping) on every
5254 			 * invocation of reclaim_pv_chunk().
5255 			 */
5256 			while ((pc = TAILQ_FIRST(&pvc->pvc_list)) != pc_marker) {
5257 				MPASS(pc->pc_pmap != NULL);
5258 				TAILQ_REMOVE(&pvc->pvc_list, pc, pc_lru);
5259 				TAILQ_INSERT_TAIL(&pvc->pvc_list, pc, pc_lru);
5260 			}
5261 		}
5262 	}
5263 	TAILQ_REMOVE(&pvc->pvc_list, pc_marker, pc_lru);
5264 	TAILQ_REMOVE(&pvc->pvc_list, pc_marker_end, pc_lru);
5265 	pvc->active_reclaims--;
5266 	mtx_unlock(&pvc->pvc_lock);
5267 	reclaim_pv_chunk_leave_pmap(pmap, locked_pmap, start_di);
5268 	if (m_pc == NULL && !SLIST_EMPTY(&free)) {
5269 		m_pc = SLIST_FIRST(&free);
5270 		SLIST_REMOVE_HEAD(&free, plinks.s.ss);
5271 		/* Recycle a freed page table page. */
5272 		m_pc->ref_count = 1;
5273 	}
5274 	vm_page_free_pages_toq(&free, true);
5275 	return (m_pc);
5276 }
5277 
5278 static vm_page_t
reclaim_pv_chunk(pmap_t locked_pmap,struct rwlock ** lockp)5279 reclaim_pv_chunk(pmap_t locked_pmap, struct rwlock **lockp)
5280 {
5281 	vm_page_t m;
5282 	int i, domain;
5283 
5284 	domain = PCPU_GET(domain);
5285 	for (i = 0; i < vm_ndomains; i++) {
5286 		m = reclaim_pv_chunk_domain(locked_pmap, lockp, domain);
5287 		if (m != NULL)
5288 			break;
5289 		domain = (domain + 1) % vm_ndomains;
5290 	}
5291 
5292 	return (m);
5293 }
5294 
5295 /*
5296  * free the pv_entry back to the free list
5297  */
5298 static void
free_pv_entry(pmap_t pmap,pv_entry_t pv)5299 free_pv_entry(pmap_t pmap, pv_entry_t pv)
5300 {
5301 	struct pv_chunk *pc;
5302 	int idx, field, bit;
5303 
5304 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
5305 	PV_STAT(counter_u64_add(pv_entry_frees, 1));
5306 	PV_STAT(counter_u64_add(pv_entry_spare, 1));
5307 	PV_STAT(counter_u64_add(pv_entry_count, -1));
5308 	pc = pv_to_chunk(pv);
5309 	idx = pv - &pc->pc_pventry[0];
5310 	field = idx / 64;
5311 	bit = idx % 64;
5312 	pc->pc_map[field] |= 1ul << bit;
5313 	if (pc->pc_map[0] != PC_FREE0 || pc->pc_map[1] != PC_FREE1 ||
5314 	    pc->pc_map[2] != PC_FREE2) {
5315 		/* 98% of the time, pc is already at the head of the list. */
5316 		if (__predict_false(pc != TAILQ_FIRST(&pmap->pm_pvchunk))) {
5317 			TAILQ_REMOVE(&pmap->pm_pvchunk, pc, pc_list);
5318 			TAILQ_INSERT_HEAD(&pmap->pm_pvchunk, pc, pc_list);
5319 		}
5320 		return;
5321 	}
5322 	TAILQ_REMOVE(&pmap->pm_pvchunk, pc, pc_list);
5323 	free_pv_chunk(pc);
5324 }
5325 
5326 static void
free_pv_chunk_dequeued(struct pv_chunk * pc)5327 free_pv_chunk_dequeued(struct pv_chunk *pc)
5328 {
5329 	vm_page_t m;
5330 
5331 	PV_STAT(counter_u64_add(pv_entry_spare, -_NPCPV));
5332 	PV_STAT(counter_u64_add(pc_chunk_count, -1));
5333 	PV_STAT(counter_u64_add(pc_chunk_frees, 1));
5334 	counter_u64_add(pv_page_count, -1);
5335 	/* entire chunk is free, return it */
5336 	m = PHYS_TO_VM_PAGE(DMAP_TO_PHYS((vm_offset_t)pc));
5337 	dump_drop_page(m->phys_addr);
5338 	vm_page_unwire_noq(m);
5339 	vm_page_free(m);
5340 }
5341 
5342 static void
free_pv_chunk(struct pv_chunk * pc)5343 free_pv_chunk(struct pv_chunk *pc)
5344 {
5345 	struct pv_chunks_list *pvc;
5346 
5347 	pvc = &pv_chunks[pc_to_domain(pc)];
5348 	mtx_lock(&pvc->pvc_lock);
5349 	TAILQ_REMOVE(&pvc->pvc_list, pc, pc_lru);
5350 	mtx_unlock(&pvc->pvc_lock);
5351 	free_pv_chunk_dequeued(pc);
5352 }
5353 
5354 static void
free_pv_chunk_batch(struct pv_chunklist * batch)5355 free_pv_chunk_batch(struct pv_chunklist *batch)
5356 {
5357 	struct pv_chunks_list *pvc;
5358 	struct pv_chunk *pc, *npc;
5359 	int i;
5360 
5361 	for (i = 0; i < vm_ndomains; i++) {
5362 		if (TAILQ_EMPTY(&batch[i]))
5363 			continue;
5364 		pvc = &pv_chunks[i];
5365 		mtx_lock(&pvc->pvc_lock);
5366 		TAILQ_FOREACH(pc, &batch[i], pc_list) {
5367 			TAILQ_REMOVE(&pvc->pvc_list, pc, pc_lru);
5368 		}
5369 		mtx_unlock(&pvc->pvc_lock);
5370 	}
5371 
5372 	for (i = 0; i < vm_ndomains; i++) {
5373 		TAILQ_FOREACH_SAFE(pc, &batch[i], pc_list, npc) {
5374 			free_pv_chunk_dequeued(pc);
5375 		}
5376 	}
5377 }
5378 
5379 /*
5380  * Returns a new PV entry, allocating a new PV chunk from the system when
5381  * needed.  If this PV chunk allocation fails and a PV list lock pointer was
5382  * given, a PV chunk is reclaimed from an arbitrary pmap.  Otherwise, NULL is
5383  * returned.
5384  *
5385  * The given PV list lock may be released.
5386  */
5387 static pv_entry_t
get_pv_entry(pmap_t pmap,struct rwlock ** lockp)5388 get_pv_entry(pmap_t pmap, struct rwlock **lockp)
5389 {
5390 	struct pv_chunks_list *pvc;
5391 	int bit, field;
5392 	pv_entry_t pv;
5393 	struct pv_chunk *pc;
5394 	vm_page_t m;
5395 
5396 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
5397 	PV_STAT(counter_u64_add(pv_entry_allocs, 1));
5398 retry:
5399 	pc = TAILQ_FIRST(&pmap->pm_pvchunk);
5400 	if (pc != NULL) {
5401 		for (field = 0; field < _NPCM; field++) {
5402 			if (pc->pc_map[field]) {
5403 				bit = bsfq(pc->pc_map[field]);
5404 				break;
5405 			}
5406 		}
5407 		if (field < _NPCM) {
5408 			pv = &pc->pc_pventry[field * 64 + bit];
5409 			pc->pc_map[field] &= ~(1ul << bit);
5410 			/* If this was the last item, move it to tail */
5411 			if (pc->pc_map[0] == 0 && pc->pc_map[1] == 0 &&
5412 			    pc->pc_map[2] == 0) {
5413 				TAILQ_REMOVE(&pmap->pm_pvchunk, pc, pc_list);
5414 				TAILQ_INSERT_TAIL(&pmap->pm_pvchunk, pc,
5415 				    pc_list);
5416 			}
5417 			PV_STAT(counter_u64_add(pv_entry_count, 1));
5418 			PV_STAT(counter_u64_add(pv_entry_spare, -1));
5419 			return (pv);
5420 		}
5421 	}
5422 	/* No free items, allocate another chunk */
5423 	m = vm_page_alloc_noobj(VM_ALLOC_WIRED);
5424 	if (m == NULL) {
5425 		if (lockp == NULL) {
5426 			PV_STAT(counter_u64_add(pc_chunk_tryfail, 1));
5427 			return (NULL);
5428 		}
5429 		m = reclaim_pv_chunk(pmap, lockp);
5430 		if (m == NULL)
5431 			goto retry;
5432 	} else
5433 		counter_u64_add(pv_page_count, 1);
5434 	PV_STAT(counter_u64_add(pc_chunk_count, 1));
5435 	PV_STAT(counter_u64_add(pc_chunk_allocs, 1));
5436 	dump_add_page(m->phys_addr);
5437 	pc = (void *)PHYS_TO_DMAP(m->phys_addr);
5438 	pc->pc_pmap = pmap;
5439 	pc->pc_map[0] = PC_FREE0 & ~1ul;	/* preallocated bit 0 */
5440 	pc->pc_map[1] = PC_FREE1;
5441 	pc->pc_map[2] = PC_FREE2;
5442 	pvc = &pv_chunks[vm_page_domain(m)];
5443 	mtx_lock(&pvc->pvc_lock);
5444 	TAILQ_INSERT_TAIL(&pvc->pvc_list, pc, pc_lru);
5445 	mtx_unlock(&pvc->pvc_lock);
5446 	pv = &pc->pc_pventry[0];
5447 	TAILQ_INSERT_HEAD(&pmap->pm_pvchunk, pc, pc_list);
5448 	PV_STAT(counter_u64_add(pv_entry_count, 1));
5449 	PV_STAT(counter_u64_add(pv_entry_spare, _NPCPV - 1));
5450 	return (pv);
5451 }
5452 
5453 /*
5454  * Returns the number of one bits within the given PV chunk map.
5455  *
5456  * The erratas for Intel processors state that "POPCNT Instruction May
5457  * Take Longer to Execute Than Expected".  It is believed that the
5458  * issue is the spurious dependency on the destination register.
5459  * Provide a hint to the register rename logic that the destination
5460  * value is overwritten, by clearing it, as suggested in the
5461  * optimization manual.  It should be cheap for unaffected processors
5462  * as well.
5463  *
5464  * Reference numbers for erratas are
5465  * 4th Gen Core: HSD146
5466  * 5th Gen Core: BDM85
5467  * 6th Gen Core: SKL029
5468  */
5469 static int
popcnt_pc_map_pq(uint64_t * map)5470 popcnt_pc_map_pq(uint64_t *map)
5471 {
5472 	u_long result, tmp;
5473 
5474 	__asm __volatile("xorl %k0,%k0;popcntq %2,%0;"
5475 	    "xorl %k1,%k1;popcntq %3,%1;addl %k1,%k0;"
5476 	    "xorl %k1,%k1;popcntq %4,%1;addl %k1,%k0"
5477 	    : "=&r" (result), "=&r" (tmp)
5478 	    : "m" (map[0]), "m" (map[1]), "m" (map[2]));
5479 	return (result);
5480 }
5481 
5482 /*
5483  * Ensure that the number of spare PV entries in the specified pmap meets or
5484  * exceeds the given count, "needed".
5485  *
5486  * The given PV list lock may be released.
5487  */
5488 static void
reserve_pv_entries(pmap_t pmap,int needed,struct rwlock ** lockp)5489 reserve_pv_entries(pmap_t pmap, int needed, struct rwlock **lockp)
5490 {
5491 	struct pv_chunks_list *pvc;
5492 	struct pch new_tail[PMAP_MEMDOM];
5493 	struct pv_chunk *pc;
5494 	vm_page_t m;
5495 	int avail, free, i;
5496 	bool reclaimed;
5497 
5498 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
5499 	KASSERT(lockp != NULL, ("reserve_pv_entries: lockp is NULL"));
5500 
5501 	/*
5502 	 * Newly allocated PV chunks must be stored in a private list until
5503 	 * the required number of PV chunks have been allocated.  Otherwise,
5504 	 * reclaim_pv_chunk() could recycle one of these chunks.  In
5505 	 * contrast, these chunks must be added to the pmap upon allocation.
5506 	 */
5507 	for (i = 0; i < PMAP_MEMDOM; i++)
5508 		TAILQ_INIT(&new_tail[i]);
5509 retry:
5510 	avail = 0;
5511 	TAILQ_FOREACH(pc, &pmap->pm_pvchunk, pc_list) {
5512 #ifndef __POPCNT__
5513 		if ((cpu_feature2 & CPUID2_POPCNT) == 0)
5514 			bit_count((bitstr_t *)pc->pc_map, 0,
5515 			    sizeof(pc->pc_map) * NBBY, &free);
5516 		else
5517 #endif
5518 		free = popcnt_pc_map_pq(pc->pc_map);
5519 		if (free == 0)
5520 			break;
5521 		avail += free;
5522 		if (avail >= needed)
5523 			break;
5524 	}
5525 	for (reclaimed = false; avail < needed; avail += _NPCPV) {
5526 		m = vm_page_alloc_noobj(VM_ALLOC_WIRED);
5527 		if (m == NULL) {
5528 			m = reclaim_pv_chunk(pmap, lockp);
5529 			if (m == NULL)
5530 				goto retry;
5531 			reclaimed = true;
5532 		} else
5533 			counter_u64_add(pv_page_count, 1);
5534 		PV_STAT(counter_u64_add(pc_chunk_count, 1));
5535 		PV_STAT(counter_u64_add(pc_chunk_allocs, 1));
5536 		dump_add_page(m->phys_addr);
5537 		pc = (void *)PHYS_TO_DMAP(m->phys_addr);
5538 		pc->pc_pmap = pmap;
5539 		pc->pc_map[0] = PC_FREE0;
5540 		pc->pc_map[1] = PC_FREE1;
5541 		pc->pc_map[2] = PC_FREE2;
5542 		TAILQ_INSERT_HEAD(&pmap->pm_pvchunk, pc, pc_list);
5543 		TAILQ_INSERT_TAIL(&new_tail[vm_page_domain(m)], pc, pc_lru);
5544 		PV_STAT(counter_u64_add(pv_entry_spare, _NPCPV));
5545 
5546 		/*
5547 		 * The reclaim might have freed a chunk from the current pmap.
5548 		 * If that chunk contained available entries, we need to
5549 		 * re-count the number of available entries.
5550 		 */
5551 		if (reclaimed)
5552 			goto retry;
5553 	}
5554 	for (i = 0; i < vm_ndomains; i++) {
5555 		if (TAILQ_EMPTY(&new_tail[i]))
5556 			continue;
5557 		pvc = &pv_chunks[i];
5558 		mtx_lock(&pvc->pvc_lock);
5559 		TAILQ_CONCAT(&pvc->pvc_list, &new_tail[i], pc_lru);
5560 		mtx_unlock(&pvc->pvc_lock);
5561 	}
5562 }
5563 
5564 /*
5565  * First find and then remove the pv entry for the specified pmap and virtual
5566  * address from the specified pv list.  Returns the pv entry if found and NULL
5567  * otherwise.  This operation can be performed on pv lists for either 4KB or
5568  * 2MB page mappings.
5569  */
5570 static __inline pv_entry_t
pmap_pvh_remove(struct md_page * pvh,pmap_t pmap,vm_offset_t va)5571 pmap_pvh_remove(struct md_page *pvh, pmap_t pmap, vm_offset_t va)
5572 {
5573 	pv_entry_t pv;
5574 
5575 	TAILQ_FOREACH(pv, &pvh->pv_list, pv_next) {
5576 		if (pmap == PV_PMAP(pv) && va == pv->pv_va) {
5577 			TAILQ_REMOVE(&pvh->pv_list, pv, pv_next);
5578 			pvh->pv_gen++;
5579 			break;
5580 		}
5581 	}
5582 	return (pv);
5583 }
5584 
5585 /*
5586  * After demotion from a 2MB page mapping to 512 4KB page mappings,
5587  * destroy the pv entry for the 2MB page mapping and reinstantiate the pv
5588  * entries for each of the 4KB page mappings.
5589  */
5590 static void
pmap_pv_demote_pde(pmap_t pmap,vm_offset_t va,vm_paddr_t pa,struct rwlock ** lockp)5591 pmap_pv_demote_pde(pmap_t pmap, vm_offset_t va, vm_paddr_t pa,
5592     struct rwlock **lockp)
5593 {
5594 	struct md_page *pvh;
5595 	struct pv_chunk *pc;
5596 	pv_entry_t pv;
5597 	vm_offset_t va_last;
5598 	vm_page_t m;
5599 	int bit, field;
5600 
5601 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
5602 	KASSERT((pa & PDRMASK) == 0,
5603 	    ("pmap_pv_demote_pde: pa is not 2mpage aligned"));
5604 	CHANGE_PV_LIST_LOCK_TO_PHYS(lockp, pa);
5605 
5606 	/*
5607 	 * Transfer the 2mpage's pv entry for this mapping to the first
5608 	 * page's pv list.  Once this transfer begins, the pv list lock
5609 	 * must not be released until the last pv entry is reinstantiated.
5610 	 */
5611 	pvh = pa_to_pvh(pa);
5612 	va = trunc_2mpage(va);
5613 	pv = pmap_pvh_remove(pvh, pmap, va);
5614 	KASSERT(pv != NULL, ("pmap_pv_demote_pde: pv not found"));
5615 	m = PHYS_TO_VM_PAGE(pa);
5616 	TAILQ_INSERT_TAIL(&m->md.pv_list, pv, pv_next);
5617 	m->md.pv_gen++;
5618 	/* Instantiate the remaining NPTEPG - 1 pv entries. */
5619 	PV_STAT(counter_u64_add(pv_entry_allocs, NPTEPG - 1));
5620 	va_last = va + NBPDR - PAGE_SIZE;
5621 	for (;;) {
5622 		pc = TAILQ_FIRST(&pmap->pm_pvchunk);
5623 		KASSERT(pc->pc_map[0] != 0 || pc->pc_map[1] != 0 ||
5624 		    pc->pc_map[2] != 0, ("pmap_pv_demote_pde: missing spare"));
5625 		for (field = 0; field < _NPCM; field++) {
5626 			while (pc->pc_map[field]) {
5627 				bit = bsfq(pc->pc_map[field]);
5628 				pc->pc_map[field] &= ~(1ul << bit);
5629 				pv = &pc->pc_pventry[field * 64 + bit];
5630 				va += PAGE_SIZE;
5631 				pv->pv_va = va;
5632 				m++;
5633 				KASSERT((m->oflags & VPO_UNMANAGED) == 0,
5634 			    ("pmap_pv_demote_pde: page %p is not managed", m));
5635 				TAILQ_INSERT_TAIL(&m->md.pv_list, pv, pv_next);
5636 				m->md.pv_gen++;
5637 				if (va == va_last)
5638 					goto out;
5639 			}
5640 		}
5641 		TAILQ_REMOVE(&pmap->pm_pvchunk, pc, pc_list);
5642 		TAILQ_INSERT_TAIL(&pmap->pm_pvchunk, pc, pc_list);
5643 	}
5644 out:
5645 	if (pc->pc_map[0] == 0 && pc->pc_map[1] == 0 && pc->pc_map[2] == 0) {
5646 		TAILQ_REMOVE(&pmap->pm_pvchunk, pc, pc_list);
5647 		TAILQ_INSERT_TAIL(&pmap->pm_pvchunk, pc, pc_list);
5648 	}
5649 	PV_STAT(counter_u64_add(pv_entry_count, NPTEPG - 1));
5650 	PV_STAT(counter_u64_add(pv_entry_spare, -(NPTEPG - 1)));
5651 }
5652 
5653 #if VM_NRESERVLEVEL > 0
5654 /*
5655  * After promotion from 512 4KB page mappings to a single 2MB page mapping,
5656  * replace the many pv entries for the 4KB page mappings by a single pv entry
5657  * for the 2MB page mapping.
5658  */
5659 static void
pmap_pv_promote_pde(pmap_t pmap,vm_offset_t va,vm_paddr_t pa,struct rwlock ** lockp)5660 pmap_pv_promote_pde(pmap_t pmap, vm_offset_t va, vm_paddr_t pa,
5661     struct rwlock **lockp)
5662 {
5663 	struct md_page *pvh;
5664 	pv_entry_t pv;
5665 	vm_offset_t va_last;
5666 	vm_page_t m;
5667 
5668 	KASSERT((pa & PDRMASK) == 0,
5669 	    ("pmap_pv_promote_pde: pa is not 2mpage aligned"));
5670 	CHANGE_PV_LIST_LOCK_TO_PHYS(lockp, pa);
5671 
5672 	/*
5673 	 * Transfer the first page's pv entry for this mapping to the 2mpage's
5674 	 * pv list.  Aside from avoiding the cost of a call to get_pv_entry(),
5675 	 * a transfer avoids the possibility that get_pv_entry() calls
5676 	 * reclaim_pv_chunk() and that reclaim_pv_chunk() removes one of the
5677 	 * mappings that is being promoted.
5678 	 */
5679 	m = PHYS_TO_VM_PAGE(pa);
5680 	va = trunc_2mpage(va);
5681 	pv = pmap_pvh_remove(&m->md, pmap, va);
5682 	KASSERT(pv != NULL, ("pmap_pv_promote_pde: pv not found"));
5683 	pvh = pa_to_pvh(pa);
5684 	TAILQ_INSERT_TAIL(&pvh->pv_list, pv, pv_next);
5685 	pvh->pv_gen++;
5686 	/* Free the remaining NPTEPG - 1 pv entries. */
5687 	va_last = va + NBPDR - PAGE_SIZE;
5688 	do {
5689 		m++;
5690 		va += PAGE_SIZE;
5691 		pmap_pvh_free(&m->md, pmap, va);
5692 	} while (va < va_last);
5693 }
5694 #endif /* VM_NRESERVLEVEL > 0 */
5695 
5696 /*
5697  * First find and then destroy the pv entry for the specified pmap and virtual
5698  * address.  This operation can be performed on pv lists for either 4KB or 2MB
5699  * page mappings.
5700  */
5701 static void
pmap_pvh_free(struct md_page * pvh,pmap_t pmap,vm_offset_t va)5702 pmap_pvh_free(struct md_page *pvh, pmap_t pmap, vm_offset_t va)
5703 {
5704 	pv_entry_t pv;
5705 
5706 	pv = pmap_pvh_remove(pvh, pmap, va);
5707 	KASSERT(pv != NULL, ("pmap_pvh_free: pv not found"));
5708 	free_pv_entry(pmap, pv);
5709 }
5710 
5711 /*
5712  * Conditionally create the PV entry for a 4KB page mapping if the required
5713  * memory can be allocated without resorting to reclamation.
5714  */
5715 static boolean_t
pmap_try_insert_pv_entry(pmap_t pmap,vm_offset_t va,vm_page_t m,struct rwlock ** lockp)5716 pmap_try_insert_pv_entry(pmap_t pmap, vm_offset_t va, vm_page_t m,
5717     struct rwlock **lockp)
5718 {
5719 	pv_entry_t pv;
5720 
5721 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
5722 	/* Pass NULL instead of the lock pointer to disable reclamation. */
5723 	if ((pv = get_pv_entry(pmap, NULL)) != NULL) {
5724 		pv->pv_va = va;
5725 		CHANGE_PV_LIST_LOCK_TO_VM_PAGE(lockp, m);
5726 		TAILQ_INSERT_TAIL(&m->md.pv_list, pv, pv_next);
5727 		m->md.pv_gen++;
5728 		return (TRUE);
5729 	} else
5730 		return (FALSE);
5731 }
5732 
5733 /*
5734  * Create the PV entry for a 2MB page mapping.  Always returns true unless the
5735  * flag PMAP_ENTER_NORECLAIM is specified.  If that flag is specified, returns
5736  * false if the PV entry cannot be allocated without resorting to reclamation.
5737  */
5738 static bool
pmap_pv_insert_pde(pmap_t pmap,vm_offset_t va,pd_entry_t pde,u_int flags,struct rwlock ** lockp)5739 pmap_pv_insert_pde(pmap_t pmap, vm_offset_t va, pd_entry_t pde, u_int flags,
5740     struct rwlock **lockp)
5741 {
5742 	struct md_page *pvh;
5743 	pv_entry_t pv;
5744 	vm_paddr_t pa;
5745 
5746 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
5747 	/* Pass NULL instead of the lock pointer to disable reclamation. */
5748 	if ((pv = get_pv_entry(pmap, (flags & PMAP_ENTER_NORECLAIM) != 0 ?
5749 	    NULL : lockp)) == NULL)
5750 		return (false);
5751 	pv->pv_va = va;
5752 	pa = pde & PG_PS_FRAME;
5753 	CHANGE_PV_LIST_LOCK_TO_PHYS(lockp, pa);
5754 	pvh = pa_to_pvh(pa);
5755 	TAILQ_INSERT_TAIL(&pvh->pv_list, pv, pv_next);
5756 	pvh->pv_gen++;
5757 	return (true);
5758 }
5759 
5760 /*
5761  * Fills a page table page with mappings to consecutive physical pages.
5762  */
5763 static void
pmap_fill_ptp(pt_entry_t * firstpte,pt_entry_t newpte)5764 pmap_fill_ptp(pt_entry_t *firstpte, pt_entry_t newpte)
5765 {
5766 	pt_entry_t *pte;
5767 
5768 	for (pte = firstpte; pte < firstpte + NPTEPG; pte++) {
5769 		*pte = newpte;
5770 		newpte += PAGE_SIZE;
5771 	}
5772 }
5773 
5774 /*
5775  * Tries to demote a 2MB page mapping.  If demotion fails, the 2MB page
5776  * mapping is invalidated.
5777  */
5778 static boolean_t
pmap_demote_pde(pmap_t pmap,pd_entry_t * pde,vm_offset_t va)5779 pmap_demote_pde(pmap_t pmap, pd_entry_t *pde, vm_offset_t va)
5780 {
5781 	struct rwlock *lock;
5782 	boolean_t rv;
5783 
5784 	lock = NULL;
5785 	rv = pmap_demote_pde_locked(pmap, pde, va, &lock);
5786 	if (lock != NULL)
5787 		rw_wunlock(lock);
5788 	return (rv);
5789 }
5790 
5791 static void
pmap_demote_pde_check(pt_entry_t * firstpte __unused,pt_entry_t newpte __unused)5792 pmap_demote_pde_check(pt_entry_t *firstpte __unused, pt_entry_t newpte __unused)
5793 {
5794 #ifdef INVARIANTS
5795 #ifdef DIAGNOSTIC
5796 	pt_entry_t *xpte, *ypte;
5797 
5798 	for (xpte = firstpte; xpte < firstpte + NPTEPG;
5799 	    xpte++, newpte += PAGE_SIZE) {
5800 		if ((*xpte & PG_FRAME) != (newpte & PG_FRAME)) {
5801 			printf("pmap_demote_pde: xpte %zd and newpte map "
5802 			    "different pages: found %#lx, expected %#lx\n",
5803 			    xpte - firstpte, *xpte, newpte);
5804 			printf("page table dump\n");
5805 			for (ypte = firstpte; ypte < firstpte + NPTEPG; ypte++)
5806 				printf("%zd %#lx\n", ypte - firstpte, *ypte);
5807 			panic("firstpte");
5808 		}
5809 	}
5810 #else
5811 	KASSERT((*firstpte & PG_FRAME) == (newpte & PG_FRAME),
5812 	    ("pmap_demote_pde: firstpte and newpte map different physical"
5813 	    " addresses"));
5814 #endif
5815 #endif
5816 }
5817 
5818 static void
pmap_demote_pde_abort(pmap_t pmap,vm_offset_t va,pd_entry_t * pde,pd_entry_t oldpde,struct rwlock ** lockp)5819 pmap_demote_pde_abort(pmap_t pmap, vm_offset_t va, pd_entry_t *pde,
5820     pd_entry_t oldpde, struct rwlock **lockp)
5821 {
5822 	struct spglist free;
5823 	vm_offset_t sva;
5824 
5825 	SLIST_INIT(&free);
5826 	sva = trunc_2mpage(va);
5827 	pmap_remove_pde(pmap, pde, sva, &free, lockp);
5828 	if ((oldpde & pmap_global_bit(pmap)) == 0)
5829 		pmap_invalidate_pde_page(pmap, sva, oldpde);
5830 	vm_page_free_pages_toq(&free, true);
5831 	CTR2(KTR_PMAP, "pmap_demote_pde: failure for va %#lx in pmap %p",
5832 	    va, pmap);
5833 }
5834 
5835 static boolean_t
pmap_demote_pde_locked(pmap_t pmap,pd_entry_t * pde,vm_offset_t va,struct rwlock ** lockp)5836 pmap_demote_pde_locked(pmap_t pmap, pd_entry_t *pde, vm_offset_t va,
5837     struct rwlock **lockp)
5838 {
5839 	pd_entry_t newpde, oldpde;
5840 	pt_entry_t *firstpte, newpte;
5841 	pt_entry_t PG_A, PG_G, PG_M, PG_PKU_MASK, PG_RW, PG_V;
5842 	vm_paddr_t mptepa;
5843 	vm_page_t mpte;
5844 	int PG_PTE_CACHE;
5845 	bool in_kernel;
5846 
5847 	PG_A = pmap_accessed_bit(pmap);
5848 	PG_G = pmap_global_bit(pmap);
5849 	PG_M = pmap_modified_bit(pmap);
5850 	PG_RW = pmap_rw_bit(pmap);
5851 	PG_V = pmap_valid_bit(pmap);
5852 	PG_PTE_CACHE = pmap_cache_mask(pmap, 0);
5853 	PG_PKU_MASK = pmap_pku_mask_bit(pmap);
5854 
5855 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
5856 	in_kernel = va >= VM_MAXUSER_ADDRESS;
5857 	oldpde = *pde;
5858 	KASSERT((oldpde & (PG_PS | PG_V)) == (PG_PS | PG_V),
5859 	    ("pmap_demote_pde: oldpde is missing PG_PS and/or PG_V"));
5860 
5861 	/*
5862 	 * Invalidate the 2MB page mapping and return "failure" if the
5863 	 * mapping was never accessed.
5864 	 */
5865 	if ((oldpde & PG_A) == 0) {
5866 		KASSERT((oldpde & PG_W) == 0,
5867 		    ("pmap_demote_pde: a wired mapping is missing PG_A"));
5868 		pmap_demote_pde_abort(pmap, va, pde, oldpde, lockp);
5869 		return (FALSE);
5870 	}
5871 
5872 	mpte = pmap_remove_pt_page(pmap, va);
5873 	if (mpte == NULL) {
5874 		KASSERT((oldpde & PG_W) == 0,
5875 		    ("pmap_demote_pde: page table page for a wired mapping"
5876 		    " is missing"));
5877 
5878 		/*
5879 		 * If the page table page is missing and the mapping
5880 		 * is for a kernel address, the mapping must belong to
5881 		 * the direct map.  Page table pages are preallocated
5882 		 * for every other part of the kernel address space,
5883 		 * so the direct map region is the only part of the
5884 		 * kernel address space that must be handled here.
5885 		 */
5886 		KASSERT(!in_kernel || (va >= DMAP_MIN_ADDRESS &&
5887 		    va < DMAP_MAX_ADDRESS),
5888 		    ("pmap_demote_pde: No saved mpte for va %#lx", va));
5889 
5890 		/*
5891 		 * If the 2MB page mapping belongs to the direct map
5892 		 * region of the kernel's address space, then the page
5893 		 * allocation request specifies the highest possible
5894 		 * priority (VM_ALLOC_INTERRUPT).  Otherwise, the
5895 		 * priority is normal.
5896 		 */
5897 		mpte = pmap_alloc_pt_page(pmap, pmap_pde_pindex(va),
5898 		    (in_kernel ? VM_ALLOC_INTERRUPT : 0) | VM_ALLOC_WIRED);
5899 
5900 		/*
5901 		 * If the allocation of the new page table page fails,
5902 		 * invalidate the 2MB page mapping and return "failure".
5903 		 */
5904 		if (mpte == NULL) {
5905 			pmap_demote_pde_abort(pmap, va, pde, oldpde, lockp);
5906 			return (FALSE);
5907 		}
5908 
5909 		if (!in_kernel)
5910 			mpte->ref_count = NPTEPG;
5911 	}
5912 	mptepa = VM_PAGE_TO_PHYS(mpte);
5913 	firstpte = (pt_entry_t *)PHYS_TO_DMAP(mptepa);
5914 	newpde = mptepa | PG_M | PG_A | (oldpde & PG_U) | PG_RW | PG_V;
5915 	KASSERT((oldpde & (PG_M | PG_RW)) != PG_RW,
5916 	    ("pmap_demote_pde: oldpde is missing PG_M"));
5917 	newpte = oldpde & ~PG_PS;
5918 	newpte = pmap_swap_pat(pmap, newpte);
5919 
5920 	/*
5921 	 * If the page table page is not leftover from an earlier promotion,
5922 	 * initialize it.
5923 	 */
5924 	if (vm_page_none_valid(mpte))
5925 		pmap_fill_ptp(firstpte, newpte);
5926 
5927 	pmap_demote_pde_check(firstpte, newpte);
5928 
5929 	/*
5930 	 * If the mapping has changed attributes, update the page table
5931 	 * entries.
5932 	 */
5933 	if ((*firstpte & PG_PTE_PROMOTE) != (newpte & PG_PTE_PROMOTE))
5934 		pmap_fill_ptp(firstpte, newpte);
5935 
5936 	/*
5937 	 * The spare PV entries must be reserved prior to demoting the
5938 	 * mapping, that is, prior to changing the PDE.  Otherwise, the state
5939 	 * of the PDE and the PV lists will be inconsistent, which can result
5940 	 * in reclaim_pv_chunk() attempting to remove a PV entry from the
5941 	 * wrong PV list and pmap_pv_demote_pde() failing to find the expected
5942 	 * PV entry for the 2MB page mapping that is being demoted.
5943 	 */
5944 	if ((oldpde & PG_MANAGED) != 0)
5945 		reserve_pv_entries(pmap, NPTEPG - 1, lockp);
5946 
5947 	/*
5948 	 * Demote the mapping.  This pmap is locked.  The old PDE has
5949 	 * PG_A set.  If the old PDE has PG_RW set, it also has PG_M
5950 	 * set.  Thus, there is no danger of a race with another
5951 	 * processor changing the setting of PG_A and/or PG_M between
5952 	 * the read above and the store below.
5953 	 */
5954 	if (workaround_erratum383)
5955 		pmap_update_pde(pmap, va, pde, newpde);
5956 	else
5957 		pde_store(pde, newpde);
5958 
5959 	/*
5960 	 * Invalidate a stale recursive mapping of the page table page.
5961 	 */
5962 	if (in_kernel)
5963 		pmap_invalidate_page(pmap, (vm_offset_t)vtopte(va));
5964 
5965 	/*
5966 	 * Demote the PV entry.
5967 	 */
5968 	if ((oldpde & PG_MANAGED) != 0)
5969 		pmap_pv_demote_pde(pmap, va, oldpde & PG_PS_FRAME, lockp);
5970 
5971 	counter_u64_add(pmap_pde_demotions, 1);
5972 	CTR2(KTR_PMAP, "pmap_demote_pde: success for va %#lx in pmap %p",
5973 	    va, pmap);
5974 	return (TRUE);
5975 }
5976 
5977 /*
5978  * pmap_remove_kernel_pde: Remove a kernel superpage mapping.
5979  */
5980 static void
pmap_remove_kernel_pde(pmap_t pmap,pd_entry_t * pde,vm_offset_t va)5981 pmap_remove_kernel_pde(pmap_t pmap, pd_entry_t *pde, vm_offset_t va)
5982 {
5983 	pd_entry_t newpde;
5984 	vm_paddr_t mptepa;
5985 	vm_page_t mpte;
5986 
5987 	KASSERT(pmap == kernel_pmap, ("pmap %p is not kernel_pmap", pmap));
5988 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
5989 	mpte = pmap_remove_pt_page(pmap, va);
5990 	if (mpte == NULL)
5991 		panic("pmap_remove_kernel_pde: Missing pt page.");
5992 
5993 	mptepa = VM_PAGE_TO_PHYS(mpte);
5994 	newpde = mptepa | X86_PG_M | X86_PG_A | X86_PG_RW | X86_PG_V;
5995 
5996 	/*
5997 	 * If this page table page was unmapped by a promotion, then it
5998 	 * contains valid mappings.  Zero it to invalidate those mappings.
5999 	 */
6000 	if (vm_page_any_valid(mpte))
6001 		pagezero((void *)PHYS_TO_DMAP(mptepa));
6002 
6003 	/*
6004 	 * Demote the mapping.
6005 	 */
6006 	if (workaround_erratum383)
6007 		pmap_update_pde(pmap, va, pde, newpde);
6008 	else
6009 		pde_store(pde, newpde);
6010 
6011 	/*
6012 	 * Invalidate a stale recursive mapping of the page table page.
6013 	 */
6014 	pmap_invalidate_page(pmap, (vm_offset_t)vtopte(va));
6015 }
6016 
6017 /*
6018  * pmap_remove_pde: do the things to unmap a superpage in a process
6019  */
6020 static int
pmap_remove_pde(pmap_t pmap,pd_entry_t * pdq,vm_offset_t sva,struct spglist * free,struct rwlock ** lockp)6021 pmap_remove_pde(pmap_t pmap, pd_entry_t *pdq, vm_offset_t sva,
6022     struct spglist *free, struct rwlock **lockp)
6023 {
6024 	struct md_page *pvh;
6025 	pd_entry_t oldpde;
6026 	vm_offset_t eva, va;
6027 	vm_page_t m, mpte;
6028 	pt_entry_t PG_G, PG_A, PG_M, PG_RW;
6029 
6030 	PG_G = pmap_global_bit(pmap);
6031 	PG_A = pmap_accessed_bit(pmap);
6032 	PG_M = pmap_modified_bit(pmap);
6033 	PG_RW = pmap_rw_bit(pmap);
6034 
6035 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
6036 	KASSERT((sva & PDRMASK) == 0,
6037 	    ("pmap_remove_pde: sva is not 2mpage aligned"));
6038 	oldpde = pte_load_clear(pdq);
6039 	if (oldpde & PG_W)
6040 		pmap->pm_stats.wired_count -= NBPDR / PAGE_SIZE;
6041 	if ((oldpde & PG_G) != 0)
6042 		pmap_invalidate_pde_page(kernel_pmap, sva, oldpde);
6043 	pmap_resident_count_adj(pmap, -NBPDR / PAGE_SIZE);
6044 	if (oldpde & PG_MANAGED) {
6045 		CHANGE_PV_LIST_LOCK_TO_PHYS(lockp, oldpde & PG_PS_FRAME);
6046 		pvh = pa_to_pvh(oldpde & PG_PS_FRAME);
6047 		pmap_pvh_free(pvh, pmap, sva);
6048 		eva = sva + NBPDR;
6049 		for (va = sva, m = PHYS_TO_VM_PAGE(oldpde & PG_PS_FRAME);
6050 		    va < eva; va += PAGE_SIZE, m++) {
6051 			if ((oldpde & (PG_M | PG_RW)) == (PG_M | PG_RW))
6052 				vm_page_dirty(m);
6053 			if (oldpde & PG_A)
6054 				vm_page_aflag_set(m, PGA_REFERENCED);
6055 			if (TAILQ_EMPTY(&m->md.pv_list) &&
6056 			    TAILQ_EMPTY(&pvh->pv_list))
6057 				vm_page_aflag_clear(m, PGA_WRITEABLE);
6058 			pmap_delayed_invl_page(m);
6059 		}
6060 	}
6061 	if (pmap == kernel_pmap) {
6062 		pmap_remove_kernel_pde(pmap, pdq, sva);
6063 	} else {
6064 		mpte = pmap_remove_pt_page(pmap, sva);
6065 		if (mpte != NULL) {
6066 			KASSERT(vm_page_all_valid(mpte),
6067 			    ("pmap_remove_pde: pte page not promoted"));
6068 			pmap_resident_count_adj(pmap, -1);
6069 			KASSERT(mpte->ref_count == NPTEPG,
6070 			    ("pmap_remove_pde: pte page ref count error"));
6071 			mpte->ref_count = 0;
6072 			pmap_add_delayed_free_list(mpte, free, FALSE);
6073 		}
6074 	}
6075 	return (pmap_unuse_pt(pmap, sva, *pmap_pdpe(pmap, sva), free));
6076 }
6077 
6078 /*
6079  * pmap_remove_pte: do the things to unmap a page in a process
6080  */
6081 static int
pmap_remove_pte(pmap_t pmap,pt_entry_t * ptq,vm_offset_t va,pd_entry_t ptepde,struct spglist * free,struct rwlock ** lockp)6082 pmap_remove_pte(pmap_t pmap, pt_entry_t *ptq, vm_offset_t va,
6083     pd_entry_t ptepde, struct spglist *free, struct rwlock **lockp)
6084 {
6085 	struct md_page *pvh;
6086 	pt_entry_t oldpte, PG_A, PG_M, PG_RW;
6087 	vm_page_t m;
6088 
6089 	PG_A = pmap_accessed_bit(pmap);
6090 	PG_M = pmap_modified_bit(pmap);
6091 	PG_RW = pmap_rw_bit(pmap);
6092 
6093 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
6094 	oldpte = pte_load_clear(ptq);
6095 	if (oldpte & PG_W)
6096 		pmap->pm_stats.wired_count -= 1;
6097 	pmap_resident_count_adj(pmap, -1);
6098 	if (oldpte & PG_MANAGED) {
6099 		m = PHYS_TO_VM_PAGE(oldpte & PG_FRAME);
6100 		if ((oldpte & (PG_M | PG_RW)) == (PG_M | PG_RW))
6101 			vm_page_dirty(m);
6102 		if (oldpte & PG_A)
6103 			vm_page_aflag_set(m, PGA_REFERENCED);
6104 		CHANGE_PV_LIST_LOCK_TO_VM_PAGE(lockp, m);
6105 		pmap_pvh_free(&m->md, pmap, va);
6106 		if (TAILQ_EMPTY(&m->md.pv_list) &&
6107 		    (m->flags & PG_FICTITIOUS) == 0) {
6108 			pvh = pa_to_pvh(VM_PAGE_TO_PHYS(m));
6109 			if (TAILQ_EMPTY(&pvh->pv_list))
6110 				vm_page_aflag_clear(m, PGA_WRITEABLE);
6111 		}
6112 		pmap_delayed_invl_page(m);
6113 	}
6114 	return (pmap_unuse_pt(pmap, va, ptepde, free));
6115 }
6116 
6117 /*
6118  * Remove a single page from a process address space
6119  */
6120 static void
pmap_remove_page(pmap_t pmap,vm_offset_t va,pd_entry_t * pde,struct spglist * free)6121 pmap_remove_page(pmap_t pmap, vm_offset_t va, pd_entry_t *pde,
6122     struct spglist *free)
6123 {
6124 	struct rwlock *lock;
6125 	pt_entry_t *pte, PG_V;
6126 
6127 	PG_V = pmap_valid_bit(pmap);
6128 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
6129 	if ((*pde & PG_V) == 0)
6130 		return;
6131 	pte = pmap_pde_to_pte(pde, va);
6132 	if ((*pte & PG_V) == 0)
6133 		return;
6134 	lock = NULL;
6135 	pmap_remove_pte(pmap, pte, va, *pde, free, &lock);
6136 	if (lock != NULL)
6137 		rw_wunlock(lock);
6138 	pmap_invalidate_page(pmap, va);
6139 }
6140 
6141 /*
6142  * Removes the specified range of addresses from the page table page.
6143  */
6144 static bool
pmap_remove_ptes(pmap_t pmap,vm_offset_t sva,vm_offset_t eva,pd_entry_t * pde,struct spglist * free,struct rwlock ** lockp)6145 pmap_remove_ptes(pmap_t pmap, vm_offset_t sva, vm_offset_t eva,
6146     pd_entry_t *pde, struct spglist *free, struct rwlock **lockp)
6147 {
6148 	pt_entry_t PG_G, *pte;
6149 	vm_offset_t va;
6150 	bool anyvalid;
6151 
6152 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
6153 	PG_G = pmap_global_bit(pmap);
6154 	anyvalid = false;
6155 	va = eva;
6156 	for (pte = pmap_pde_to_pte(pde, sva); sva != eva; pte++,
6157 	    sva += PAGE_SIZE) {
6158 		if (*pte == 0) {
6159 			if (va != eva) {
6160 				pmap_invalidate_range(pmap, va, sva);
6161 				va = eva;
6162 			}
6163 			continue;
6164 		}
6165 		if ((*pte & PG_G) == 0)
6166 			anyvalid = true;
6167 		else if (va == eva)
6168 			va = sva;
6169 		if (pmap_remove_pte(pmap, pte, sva, *pde, free, lockp)) {
6170 			sva += PAGE_SIZE;
6171 			break;
6172 		}
6173 	}
6174 	if (va != eva)
6175 		pmap_invalidate_range(pmap, va, sva);
6176 	return (anyvalid);
6177 }
6178 
6179 static void
pmap_remove1(pmap_t pmap,vm_offset_t sva,vm_offset_t eva,bool map_delete)6180 pmap_remove1(pmap_t pmap, vm_offset_t sva, vm_offset_t eva, bool map_delete)
6181 {
6182 	struct rwlock *lock;
6183 	vm_page_t mt;
6184 	vm_offset_t va_next;
6185 	pml5_entry_t *pml5e;
6186 	pml4_entry_t *pml4e;
6187 	pdp_entry_t *pdpe;
6188 	pd_entry_t ptpaddr, *pde;
6189 	pt_entry_t PG_G, PG_V;
6190 	struct spglist free;
6191 	int anyvalid;
6192 
6193 	PG_G = pmap_global_bit(pmap);
6194 	PG_V = pmap_valid_bit(pmap);
6195 
6196 	/*
6197 	 * If there are no resident pages besides the top level page
6198 	 * table page(s), there is nothing to do.  Kernel pmap always
6199 	 * accounts whole preloaded area as resident, which makes its
6200 	 * resident count > 2.
6201 	 * Perform an unsynchronized read.  This is, however, safe.
6202 	 */
6203 	if (pmap->pm_stats.resident_count <= 1 + (pmap->pm_pmltopu != NULL ?
6204 	    1 : 0))
6205 		return;
6206 
6207 	anyvalid = 0;
6208 	SLIST_INIT(&free);
6209 
6210 	pmap_delayed_invl_start();
6211 	PMAP_LOCK(pmap);
6212 	if (map_delete)
6213 		pmap_pkru_on_remove(pmap, sva, eva);
6214 
6215 	/*
6216 	 * special handling of removing one page.  a very
6217 	 * common operation and easy to short circuit some
6218 	 * code.
6219 	 */
6220 	if (sva + PAGE_SIZE == eva) {
6221 		pde = pmap_pde(pmap, sva);
6222 		if (pde && (*pde & PG_PS) == 0) {
6223 			pmap_remove_page(pmap, sva, pde, &free);
6224 			goto out;
6225 		}
6226 	}
6227 
6228 	lock = NULL;
6229 	for (; sva < eva; sva = va_next) {
6230 		if (pmap->pm_stats.resident_count == 0)
6231 			break;
6232 
6233 		if (pmap_is_la57(pmap)) {
6234 			pml5e = pmap_pml5e(pmap, sva);
6235 			if ((*pml5e & PG_V) == 0) {
6236 				va_next = (sva + NBPML5) & ~PML5MASK;
6237 				if (va_next < sva)
6238 					va_next = eva;
6239 				continue;
6240 			}
6241 			pml4e = pmap_pml5e_to_pml4e(pml5e, sva);
6242 		} else {
6243 			pml4e = pmap_pml4e(pmap, sva);
6244 		}
6245 		if ((*pml4e & PG_V) == 0) {
6246 			va_next = (sva + NBPML4) & ~PML4MASK;
6247 			if (va_next < sva)
6248 				va_next = eva;
6249 			continue;
6250 		}
6251 
6252 		va_next = (sva + NBPDP) & ~PDPMASK;
6253 		if (va_next < sva)
6254 			va_next = eva;
6255 		pdpe = pmap_pml4e_to_pdpe(pml4e, sva);
6256 		if ((*pdpe & PG_V) == 0)
6257 			continue;
6258 		if ((*pdpe & PG_PS) != 0) {
6259 			KASSERT(va_next <= eva,
6260 			    ("partial update of non-transparent 1G mapping "
6261 			    "pdpe %#lx sva %#lx eva %#lx va_next %#lx",
6262 			    *pdpe, sva, eva, va_next));
6263 			MPASS(pmap != kernel_pmap); /* XXXKIB */
6264 			MPASS((*pdpe & (PG_MANAGED | PG_G)) == 0);
6265 			anyvalid = 1;
6266 			*pdpe = 0;
6267 			pmap_resident_count_adj(pmap, -NBPDP / PAGE_SIZE);
6268 			mt = PHYS_TO_VM_PAGE(*pmap_pml4e(pmap, sva) & PG_FRAME);
6269 			pmap_unwire_ptp(pmap, sva, mt, &free);
6270 			continue;
6271 		}
6272 
6273 		/*
6274 		 * Calculate index for next page table.
6275 		 */
6276 		va_next = (sva + NBPDR) & ~PDRMASK;
6277 		if (va_next < sva)
6278 			va_next = eva;
6279 
6280 		pde = pmap_pdpe_to_pde(pdpe, sva);
6281 		ptpaddr = *pde;
6282 
6283 		/*
6284 		 * Weed out invalid mappings.
6285 		 */
6286 		if (ptpaddr == 0)
6287 			continue;
6288 
6289 		/*
6290 		 * Check for large page.
6291 		 */
6292 		if ((ptpaddr & PG_PS) != 0) {
6293 			/*
6294 			 * Are we removing the entire large page?  If not,
6295 			 * demote the mapping and fall through.
6296 			 */
6297 			if (sva + NBPDR == va_next && eva >= va_next) {
6298 				/*
6299 				 * The TLB entry for a PG_G mapping is
6300 				 * invalidated by pmap_remove_pde().
6301 				 */
6302 				if ((ptpaddr & PG_G) == 0)
6303 					anyvalid = 1;
6304 				pmap_remove_pde(pmap, pde, sva, &free, &lock);
6305 				continue;
6306 			} else if (!pmap_demote_pde_locked(pmap, pde, sva,
6307 			    &lock)) {
6308 				/* The large page mapping was destroyed. */
6309 				continue;
6310 			} else
6311 				ptpaddr = *pde;
6312 		}
6313 
6314 		/*
6315 		 * Limit our scan to either the end of the va represented
6316 		 * by the current page table page, or to the end of the
6317 		 * range being removed.
6318 		 */
6319 		if (va_next > eva)
6320 			va_next = eva;
6321 
6322 		if (pmap_remove_ptes(pmap, sva, va_next, pde, &free, &lock))
6323 			anyvalid = 1;
6324 	}
6325 	if (lock != NULL)
6326 		rw_wunlock(lock);
6327 out:
6328 	if (anyvalid)
6329 		pmap_invalidate_all(pmap);
6330 	PMAP_UNLOCK(pmap);
6331 	pmap_delayed_invl_finish();
6332 	vm_page_free_pages_toq(&free, true);
6333 }
6334 
6335 /*
6336  *	Remove the given range of addresses from the specified map.
6337  *
6338  *	It is assumed that the start and end are properly
6339  *	rounded to the page size.
6340  */
6341 void
pmap_remove(pmap_t pmap,vm_offset_t sva,vm_offset_t eva)6342 pmap_remove(pmap_t pmap, vm_offset_t sva, vm_offset_t eva)
6343 {
6344 	pmap_remove1(pmap, sva, eva, false);
6345 }
6346 
6347 /*
6348  *	Remove the given range of addresses as part of a logical unmap
6349  *	operation. This has the effect of calling pmap_remove(), but
6350  *	also clears any metadata that should persist for the lifetime
6351  *	of a logical mapping.
6352  */
6353 void
pmap_map_delete(pmap_t pmap,vm_offset_t sva,vm_offset_t eva)6354 pmap_map_delete(pmap_t pmap, vm_offset_t sva, vm_offset_t eva)
6355 {
6356 	pmap_remove1(pmap, sva, eva, true);
6357 }
6358 
6359 /*
6360  *	Routine:	pmap_remove_all
6361  *	Function:
6362  *		Removes this physical page from
6363  *		all physical maps in which it resides.
6364  *		Reflects back modify bits to the pager.
6365  *
6366  *	Notes:
6367  *		Original versions of this routine were very
6368  *		inefficient because they iteratively called
6369  *		pmap_remove (slow...)
6370  */
6371 
6372 void
pmap_remove_all(vm_page_t m)6373 pmap_remove_all(vm_page_t m)
6374 {
6375 	struct md_page *pvh;
6376 	pv_entry_t pv;
6377 	pmap_t pmap;
6378 	struct rwlock *lock;
6379 	pt_entry_t *pte, tpte, PG_A, PG_M, PG_RW;
6380 	pd_entry_t *pde;
6381 	vm_offset_t va;
6382 	struct spglist free;
6383 	int pvh_gen, md_gen;
6384 
6385 	KASSERT((m->oflags & VPO_UNMANAGED) == 0,
6386 	    ("pmap_remove_all: page %p is not managed", m));
6387 	SLIST_INIT(&free);
6388 	lock = VM_PAGE_TO_PV_LIST_LOCK(m);
6389 	pvh = (m->flags & PG_FICTITIOUS) != 0 ? &pv_dummy :
6390 	    pa_to_pvh(VM_PAGE_TO_PHYS(m));
6391 	rw_wlock(lock);
6392 retry:
6393 	while ((pv = TAILQ_FIRST(&pvh->pv_list)) != NULL) {
6394 		pmap = PV_PMAP(pv);
6395 		if (!PMAP_TRYLOCK(pmap)) {
6396 			pvh_gen = pvh->pv_gen;
6397 			rw_wunlock(lock);
6398 			PMAP_LOCK(pmap);
6399 			rw_wlock(lock);
6400 			if (pvh_gen != pvh->pv_gen) {
6401 				PMAP_UNLOCK(pmap);
6402 				goto retry;
6403 			}
6404 		}
6405 		va = pv->pv_va;
6406 		pde = pmap_pde(pmap, va);
6407 		(void)pmap_demote_pde_locked(pmap, pde, va, &lock);
6408 		PMAP_UNLOCK(pmap);
6409 	}
6410 	while ((pv = TAILQ_FIRST(&m->md.pv_list)) != NULL) {
6411 		pmap = PV_PMAP(pv);
6412 		if (!PMAP_TRYLOCK(pmap)) {
6413 			pvh_gen = pvh->pv_gen;
6414 			md_gen = m->md.pv_gen;
6415 			rw_wunlock(lock);
6416 			PMAP_LOCK(pmap);
6417 			rw_wlock(lock);
6418 			if (pvh_gen != pvh->pv_gen || md_gen != m->md.pv_gen) {
6419 				PMAP_UNLOCK(pmap);
6420 				goto retry;
6421 			}
6422 		}
6423 		PG_A = pmap_accessed_bit(pmap);
6424 		PG_M = pmap_modified_bit(pmap);
6425 		PG_RW = pmap_rw_bit(pmap);
6426 		pmap_resident_count_adj(pmap, -1);
6427 		pde = pmap_pde(pmap, pv->pv_va);
6428 		KASSERT((*pde & PG_PS) == 0, ("pmap_remove_all: found"
6429 		    " a 2mpage in page %p's pv list", m));
6430 		pte = pmap_pde_to_pte(pde, pv->pv_va);
6431 		tpte = pte_load_clear(pte);
6432 		if (tpte & PG_W)
6433 			pmap->pm_stats.wired_count--;
6434 		if (tpte & PG_A)
6435 			vm_page_aflag_set(m, PGA_REFERENCED);
6436 
6437 		/*
6438 		 * Update the vm_page_t clean and reference bits.
6439 		 */
6440 		if ((tpte & (PG_M | PG_RW)) == (PG_M | PG_RW))
6441 			vm_page_dirty(m);
6442 		pmap_unuse_pt(pmap, pv->pv_va, *pde, &free);
6443 		pmap_invalidate_page(pmap, pv->pv_va);
6444 		TAILQ_REMOVE(&m->md.pv_list, pv, pv_next);
6445 		m->md.pv_gen++;
6446 		free_pv_entry(pmap, pv);
6447 		PMAP_UNLOCK(pmap);
6448 	}
6449 	vm_page_aflag_clear(m, PGA_WRITEABLE);
6450 	rw_wunlock(lock);
6451 	pmap_delayed_invl_wait(m);
6452 	vm_page_free_pages_toq(&free, true);
6453 }
6454 
6455 /*
6456  * pmap_protect_pde: do the things to protect a 2mpage in a process
6457  */
6458 static boolean_t
pmap_protect_pde(pmap_t pmap,pd_entry_t * pde,vm_offset_t sva,vm_prot_t prot)6459 pmap_protect_pde(pmap_t pmap, pd_entry_t *pde, vm_offset_t sva, vm_prot_t prot)
6460 {
6461 	pd_entry_t newpde, oldpde;
6462 	vm_page_t m, mt;
6463 	boolean_t anychanged;
6464 	pt_entry_t PG_G, PG_M, PG_RW;
6465 
6466 	PG_G = pmap_global_bit(pmap);
6467 	PG_M = pmap_modified_bit(pmap);
6468 	PG_RW = pmap_rw_bit(pmap);
6469 
6470 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
6471 	KASSERT((sva & PDRMASK) == 0,
6472 	    ("pmap_protect_pde: sva is not 2mpage aligned"));
6473 	anychanged = FALSE;
6474 retry:
6475 	oldpde = newpde = *pde;
6476 	if ((prot & VM_PROT_WRITE) == 0) {
6477 		if ((oldpde & (PG_MANAGED | PG_M | PG_RW)) ==
6478 		    (PG_MANAGED | PG_M | PG_RW)) {
6479 			m = PHYS_TO_VM_PAGE(oldpde & PG_PS_FRAME);
6480 			for (mt = m; mt < &m[NBPDR / PAGE_SIZE]; mt++)
6481 				vm_page_dirty(mt);
6482 		}
6483 		newpde &= ~(PG_RW | PG_M);
6484 	}
6485 	if ((prot & VM_PROT_EXECUTE) == 0)
6486 		newpde |= pg_nx;
6487 	if (newpde != oldpde) {
6488 		/*
6489 		 * As an optimization to future operations on this PDE, clear
6490 		 * PG_PROMOTED.  The impending invalidation will remove any
6491 		 * lingering 4KB page mappings from the TLB.
6492 		 */
6493 		if (!atomic_cmpset_long(pde, oldpde, newpde & ~PG_PROMOTED))
6494 			goto retry;
6495 		if ((oldpde & PG_G) != 0)
6496 			pmap_invalidate_pde_page(kernel_pmap, sva, oldpde);
6497 		else
6498 			anychanged = TRUE;
6499 	}
6500 	return (anychanged);
6501 }
6502 
6503 /*
6504  *	Set the physical protection on the
6505  *	specified range of this map as requested.
6506  */
6507 void
pmap_protect(pmap_t pmap,vm_offset_t sva,vm_offset_t eva,vm_prot_t prot)6508 pmap_protect(pmap_t pmap, vm_offset_t sva, vm_offset_t eva, vm_prot_t prot)
6509 {
6510 	vm_page_t m;
6511 	vm_offset_t va_next;
6512 	pml4_entry_t *pml4e;
6513 	pdp_entry_t *pdpe;
6514 	pd_entry_t ptpaddr, *pde;
6515 	pt_entry_t *pte, PG_G, PG_M, PG_RW, PG_V;
6516 	pt_entry_t obits, pbits;
6517 	boolean_t anychanged;
6518 
6519 	KASSERT((prot & ~VM_PROT_ALL) == 0, ("invalid prot %x", prot));
6520 	if (prot == VM_PROT_NONE) {
6521 		pmap_remove(pmap, sva, eva);
6522 		return;
6523 	}
6524 
6525 	if ((prot & (VM_PROT_WRITE|VM_PROT_EXECUTE)) ==
6526 	    (VM_PROT_WRITE|VM_PROT_EXECUTE))
6527 		return;
6528 
6529 	PG_G = pmap_global_bit(pmap);
6530 	PG_M = pmap_modified_bit(pmap);
6531 	PG_V = pmap_valid_bit(pmap);
6532 	PG_RW = pmap_rw_bit(pmap);
6533 	anychanged = FALSE;
6534 
6535 	/*
6536 	 * Although this function delays and batches the invalidation
6537 	 * of stale TLB entries, it does not need to call
6538 	 * pmap_delayed_invl_start() and
6539 	 * pmap_delayed_invl_finish(), because it does not
6540 	 * ordinarily destroy mappings.  Stale TLB entries from
6541 	 * protection-only changes need only be invalidated before the
6542 	 * pmap lock is released, because protection-only changes do
6543 	 * not destroy PV entries.  Even operations that iterate over
6544 	 * a physical page's PV list of mappings, like
6545 	 * pmap_remove_write(), acquire the pmap lock for each
6546 	 * mapping.  Consequently, for protection-only changes, the
6547 	 * pmap lock suffices to synchronize both page table and TLB
6548 	 * updates.
6549 	 *
6550 	 * This function only destroys a mapping if pmap_demote_pde()
6551 	 * fails.  In that case, stale TLB entries are immediately
6552 	 * invalidated.
6553 	 */
6554 
6555 	PMAP_LOCK(pmap);
6556 	for (; sva < eva; sva = va_next) {
6557 		pml4e = pmap_pml4e(pmap, sva);
6558 		if (pml4e == NULL || (*pml4e & PG_V) == 0) {
6559 			va_next = (sva + NBPML4) & ~PML4MASK;
6560 			if (va_next < sva)
6561 				va_next = eva;
6562 			continue;
6563 		}
6564 
6565 		va_next = (sva + NBPDP) & ~PDPMASK;
6566 		if (va_next < sva)
6567 			va_next = eva;
6568 		pdpe = pmap_pml4e_to_pdpe(pml4e, sva);
6569 		if ((*pdpe & PG_V) == 0)
6570 			continue;
6571 		if ((*pdpe & PG_PS) != 0) {
6572 			KASSERT(va_next <= eva,
6573 			    ("partial update of non-transparent 1G mapping "
6574 			    "pdpe %#lx sva %#lx eva %#lx va_next %#lx",
6575 			    *pdpe, sva, eva, va_next));
6576 retry_pdpe:
6577 			obits = pbits = *pdpe;
6578 			MPASS((pbits & (PG_MANAGED | PG_G)) == 0);
6579 			MPASS(pmap != kernel_pmap); /* XXXKIB */
6580 			if ((prot & VM_PROT_WRITE) == 0)
6581 				pbits &= ~(PG_RW | PG_M);
6582 			if ((prot & VM_PROT_EXECUTE) == 0)
6583 				pbits |= pg_nx;
6584 
6585 			if (pbits != obits) {
6586 				if (!atomic_cmpset_long(pdpe, obits, pbits))
6587 					/* PG_PS cannot be cleared under us, */
6588 					goto retry_pdpe;
6589 				anychanged = TRUE;
6590 			}
6591 			continue;
6592 		}
6593 
6594 		va_next = (sva + NBPDR) & ~PDRMASK;
6595 		if (va_next < sva)
6596 			va_next = eva;
6597 
6598 		pde = pmap_pdpe_to_pde(pdpe, sva);
6599 		ptpaddr = *pde;
6600 
6601 		/*
6602 		 * Weed out invalid mappings.
6603 		 */
6604 		if (ptpaddr == 0)
6605 			continue;
6606 
6607 		/*
6608 		 * Check for large page.
6609 		 */
6610 		if ((ptpaddr & PG_PS) != 0) {
6611 			/*
6612 			 * Are we protecting the entire large page?  If not,
6613 			 * demote the mapping and fall through.
6614 			 */
6615 			if (sva + NBPDR == va_next && eva >= va_next) {
6616 				/*
6617 				 * The TLB entry for a PG_G mapping is
6618 				 * invalidated by pmap_protect_pde().
6619 				 */
6620 				if (pmap_protect_pde(pmap, pde, sva, prot))
6621 					anychanged = TRUE;
6622 				continue;
6623 			} else if (!pmap_demote_pde(pmap, pde, sva)) {
6624 				/*
6625 				 * The large page mapping was destroyed.
6626 				 */
6627 				continue;
6628 			}
6629 		}
6630 
6631 		if (va_next > eva)
6632 			va_next = eva;
6633 
6634 		for (pte = pmap_pde_to_pte(pde, sva); sva != va_next; pte++,
6635 		    sva += PAGE_SIZE) {
6636 retry:
6637 			obits = pbits = *pte;
6638 			if ((pbits & PG_V) == 0)
6639 				continue;
6640 
6641 			if ((prot & VM_PROT_WRITE) == 0) {
6642 				if ((pbits & (PG_MANAGED | PG_M | PG_RW)) ==
6643 				    (PG_MANAGED | PG_M | PG_RW)) {
6644 					m = PHYS_TO_VM_PAGE(pbits & PG_FRAME);
6645 					vm_page_dirty(m);
6646 				}
6647 				pbits &= ~(PG_RW | PG_M);
6648 			}
6649 			if ((prot & VM_PROT_EXECUTE) == 0)
6650 				pbits |= pg_nx;
6651 
6652 			if (pbits != obits) {
6653 				if (!atomic_cmpset_long(pte, obits, pbits))
6654 					goto retry;
6655 				if (obits & PG_G)
6656 					pmap_invalidate_page(pmap, sva);
6657 				else
6658 					anychanged = TRUE;
6659 			}
6660 		}
6661 	}
6662 	if (anychanged)
6663 		pmap_invalidate_all(pmap);
6664 	PMAP_UNLOCK(pmap);
6665 }
6666 
6667 #if VM_NRESERVLEVEL > 0
6668 static bool
pmap_pde_ept_executable(pmap_t pmap,pd_entry_t pde)6669 pmap_pde_ept_executable(pmap_t pmap, pd_entry_t pde)
6670 {
6671 
6672 	if (pmap->pm_type != PT_EPT)
6673 		return (false);
6674 	return ((pde & EPT_PG_EXECUTE) != 0);
6675 }
6676 
6677 /*
6678  * Tries to promote the 512, contiguous 4KB page mappings that are within a
6679  * single page table page (PTP) to a single 2MB page mapping.  For promotion
6680  * to occur, two conditions must be met: (1) the 4KB page mappings must map
6681  * aligned, contiguous physical memory and (2) the 4KB page mappings must have
6682  * identical characteristics.
6683  */
6684 static void
pmap_promote_pde(pmap_t pmap,pd_entry_t * pde,vm_offset_t va,struct rwlock ** lockp)6685 pmap_promote_pde(pmap_t pmap, pd_entry_t *pde, vm_offset_t va,
6686     struct rwlock **lockp)
6687 {
6688 	pd_entry_t newpde;
6689 	pt_entry_t *firstpte, oldpte, pa, *pte;
6690 	pt_entry_t PG_G, PG_A, PG_M, PG_RW, PG_V, PG_PKU_MASK;
6691 	vm_page_t mpte;
6692 	int PG_PTE_CACHE;
6693 
6694 	PG_A = pmap_accessed_bit(pmap);
6695 	PG_G = pmap_global_bit(pmap);
6696 	PG_M = pmap_modified_bit(pmap);
6697 	PG_V = pmap_valid_bit(pmap);
6698 	PG_RW = pmap_rw_bit(pmap);
6699 	PG_PKU_MASK = pmap_pku_mask_bit(pmap);
6700 	PG_PTE_CACHE = pmap_cache_mask(pmap, 0);
6701 
6702 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
6703 
6704 	/*
6705 	 * Examine the first PTE in the specified PTP.  Abort if this PTE is
6706 	 * either invalid, unused, or does not map the first 4KB physical page
6707 	 * within a 2MB page.
6708 	 */
6709 	firstpte = (pt_entry_t *)PHYS_TO_DMAP(*pde & PG_FRAME);
6710 	newpde = *firstpte;
6711 	if ((newpde & ((PG_FRAME & PDRMASK) | PG_A | PG_V)) != (PG_A | PG_V) ||
6712 	    !pmap_allow_2m_x_page(pmap, pmap_pde_ept_executable(pmap,
6713 	    newpde))) {
6714 		counter_u64_add(pmap_pde_p_failures, 1);
6715 		CTR2(KTR_PMAP, "pmap_promote_pde: failure for va %#lx"
6716 		    " in pmap %p", va, pmap);
6717 		return;
6718 	}
6719 setpde:
6720 	if ((newpde & (PG_M | PG_RW)) == PG_RW) {
6721 		/*
6722 		 * When PG_M is already clear, PG_RW can be cleared without
6723 		 * a TLB invalidation.
6724 		 */
6725 		if (!atomic_fcmpset_long(firstpte, &newpde, newpde & ~PG_RW))
6726 			goto setpde;
6727 		newpde &= ~PG_RW;
6728 	}
6729 
6730 	/*
6731 	 * Examine each of the other PTEs in the specified PTP.  Abort if this
6732 	 * PTE maps an unexpected 4KB physical page or does not have identical
6733 	 * characteristics to the first PTE.
6734 	 */
6735 	pa = (newpde & (PG_PS_FRAME | PG_A | PG_V)) + NBPDR - PAGE_SIZE;
6736 	for (pte = firstpte + NPTEPG - 1; pte > firstpte; pte--) {
6737 		oldpte = *pte;
6738 		if ((oldpte & (PG_FRAME | PG_A | PG_V)) != pa) {
6739 			counter_u64_add(pmap_pde_p_failures, 1);
6740 			CTR2(KTR_PMAP, "pmap_promote_pde: failure for va %#lx"
6741 			    " in pmap %p", va, pmap);
6742 			return;
6743 		}
6744 setpte:
6745 		if ((oldpte & (PG_M | PG_RW)) == PG_RW) {
6746 			/*
6747 			 * When PG_M is already clear, PG_RW can be cleared
6748 			 * without a TLB invalidation.
6749 			 */
6750 			if (!atomic_fcmpset_long(pte, &oldpte, oldpte & ~PG_RW))
6751 				goto setpte;
6752 			oldpte &= ~PG_RW;
6753 			CTR2(KTR_PMAP, "pmap_promote_pde: protect for va %#lx"
6754 			    " in pmap %p", (oldpte & PG_FRAME & PDRMASK) |
6755 			    (va & ~PDRMASK), pmap);
6756 		}
6757 		if ((oldpte & PG_PTE_PROMOTE) != (newpde & PG_PTE_PROMOTE)) {
6758 			counter_u64_add(pmap_pde_p_failures, 1);
6759 			CTR2(KTR_PMAP, "pmap_promote_pde: failure for va %#lx"
6760 			    " in pmap %p", va, pmap);
6761 			return;
6762 		}
6763 		pa -= PAGE_SIZE;
6764 	}
6765 
6766 	/*
6767 	 * Save the page table page in its current state until the PDE
6768 	 * mapping the superpage is demoted by pmap_demote_pde() or
6769 	 * destroyed by pmap_remove_pde().
6770 	 */
6771 	mpte = PHYS_TO_VM_PAGE(*pde & PG_FRAME);
6772 	KASSERT(mpte >= vm_page_array &&
6773 	    mpte < &vm_page_array[vm_page_array_size],
6774 	    ("pmap_promote_pde: page table page is out of range"));
6775 	KASSERT(mpte->pindex == pmap_pde_pindex(va),
6776 	    ("pmap_promote_pde: page table page's pindex is wrong "
6777 	    "mpte %p pidx %#lx va %#lx va pde pidx %#lx",
6778 	    mpte, mpte->pindex, va, pmap_pde_pindex(va)));
6779 	if (pmap_insert_pt_page(pmap, mpte, true)) {
6780 		counter_u64_add(pmap_pde_p_failures, 1);
6781 		CTR2(KTR_PMAP,
6782 		    "pmap_promote_pde: failure for va %#lx in pmap %p", va,
6783 		    pmap);
6784 		return;
6785 	}
6786 
6787 	/*
6788 	 * Promote the pv entries.
6789 	 */
6790 	if ((newpde & PG_MANAGED) != 0)
6791 		pmap_pv_promote_pde(pmap, va, newpde & PG_PS_FRAME, lockp);
6792 
6793 	/*
6794 	 * Propagate the PAT index to its proper position.
6795 	 */
6796 	newpde = pmap_swap_pat(pmap, newpde);
6797 
6798 	/*
6799 	 * Map the superpage.
6800 	 */
6801 	if (workaround_erratum383)
6802 		pmap_update_pde(pmap, va, pde, PG_PS | newpde);
6803 	else
6804 		pde_store(pde, PG_PROMOTED | PG_PS | newpde);
6805 
6806 	counter_u64_add(pmap_pde_promotions, 1);
6807 	CTR2(KTR_PMAP, "pmap_promote_pde: success for va %#lx"
6808 	    " in pmap %p", va, pmap);
6809 }
6810 #endif /* VM_NRESERVLEVEL > 0 */
6811 
6812 static int
pmap_enter_largepage(pmap_t pmap,vm_offset_t va,pt_entry_t newpte,int flags,int psind)6813 pmap_enter_largepage(pmap_t pmap, vm_offset_t va, pt_entry_t newpte, int flags,
6814     int psind)
6815 {
6816 	vm_page_t mp;
6817 	pt_entry_t origpte, *pml4e, *pdpe, *pde, pten, PG_V;
6818 
6819 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
6820 	KASSERT(psind > 0 && psind < MAXPAGESIZES && pagesizes[psind] != 0,
6821 	    ("psind %d unexpected", psind));
6822 	KASSERT(((newpte & PG_FRAME) & (pagesizes[psind] - 1)) == 0,
6823 	    ("unaligned phys address %#lx newpte %#lx psind %d",
6824 	    newpte & PG_FRAME, newpte, psind));
6825 	KASSERT((va & (pagesizes[psind] - 1)) == 0,
6826 	    ("unaligned va %#lx psind %d", va, psind));
6827 	KASSERT(va < VM_MAXUSER_ADDRESS,
6828 	    ("kernel mode non-transparent superpage")); /* XXXKIB */
6829 	KASSERT(va + pagesizes[psind] < VM_MAXUSER_ADDRESS,
6830 	    ("overflowing user map va %#lx psind %d", va, psind)); /* XXXKIB */
6831 
6832 	PG_V = pmap_valid_bit(pmap);
6833 
6834 restart:
6835 	if (!pmap_pkru_same(pmap, va, va + pagesizes[psind]))
6836 		return (KERN_PROTECTION_FAILURE);
6837 	pten = newpte;
6838 	if (va < VM_MAXUSER_ADDRESS && pmap->pm_type == PT_X86)
6839 		pten |= pmap_pkru_get(pmap, va);
6840 
6841 	if (psind == 2) {	/* 1G */
6842 		pml4e = pmap_pml4e(pmap, va);
6843 		if (pml4e == NULL || (*pml4e & PG_V) == 0) {
6844 			mp = pmap_allocpte_alloc(pmap, pmap_pml4e_pindex(va),
6845 			    NULL, va);
6846 			if (mp == NULL)
6847 				goto allocf;
6848 			pdpe = (pdp_entry_t *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(mp));
6849 			pdpe = &pdpe[pmap_pdpe_index(va)];
6850 			origpte = *pdpe;
6851 			MPASS(origpte == 0);
6852 		} else {
6853 			pdpe = pmap_pml4e_to_pdpe(pml4e, va);
6854 			KASSERT(pdpe != NULL, ("va %#lx lost pdpe", va));
6855 			origpte = *pdpe;
6856 			if ((origpte & PG_V) == 0) {
6857 				mp = PHYS_TO_VM_PAGE(*pml4e & PG_FRAME);
6858 				mp->ref_count++;
6859 			}
6860 		}
6861 		*pdpe = pten;
6862 	} else /* (psind == 1) */ {	/* 2M */
6863 		pde = pmap_pde(pmap, va);
6864 		if (pde == NULL) {
6865 			mp = pmap_allocpte_alloc(pmap, pmap_pdpe_pindex(va),
6866 			    NULL, va);
6867 			if (mp == NULL)
6868 				goto allocf;
6869 			pde = (pd_entry_t *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(mp));
6870 			pde = &pde[pmap_pde_index(va)];
6871 			origpte = *pde;
6872 			MPASS(origpte == 0);
6873 		} else {
6874 			origpte = *pde;
6875 			if ((origpte & PG_V) == 0) {
6876 				pdpe = pmap_pdpe(pmap, va);
6877 				MPASS(pdpe != NULL && (*pdpe & PG_V) != 0);
6878 				mp = PHYS_TO_VM_PAGE(*pdpe & PG_FRAME);
6879 				mp->ref_count++;
6880 			}
6881 		}
6882 		*pde = pten;
6883 	}
6884 	KASSERT((origpte & PG_V) == 0 || ((origpte & PG_PS) != 0 &&
6885 	    (origpte & PG_PS_FRAME) == (pten & PG_PS_FRAME)),
6886 	    ("va %#lx changing %s phys page origpte %#lx pten %#lx",
6887 	    va, psind == 2 ? "1G" : "2M", origpte, pten));
6888 	if ((pten & PG_W) != 0 && (origpte & PG_W) == 0)
6889 		pmap->pm_stats.wired_count += pagesizes[psind] / PAGE_SIZE;
6890 	else if ((pten & PG_W) == 0 && (origpte & PG_W) != 0)
6891 		pmap->pm_stats.wired_count -= pagesizes[psind] / PAGE_SIZE;
6892 	if ((origpte & PG_V) == 0)
6893 		pmap_resident_count_adj(pmap, pagesizes[psind] / PAGE_SIZE);
6894 
6895 	return (KERN_SUCCESS);
6896 
6897 allocf:
6898 	if ((flags & PMAP_ENTER_NOSLEEP) != 0)
6899 		return (KERN_RESOURCE_SHORTAGE);
6900 	PMAP_UNLOCK(pmap);
6901 	vm_wait(NULL);
6902 	PMAP_LOCK(pmap);
6903 	goto restart;
6904 }
6905 
6906 /*
6907  *	Insert the given physical page (p) at
6908  *	the specified virtual address (v) in the
6909  *	target physical map with the protection requested.
6910  *
6911  *	If specified, the page will be wired down, meaning
6912  *	that the related pte can not be reclaimed.
6913  *
6914  *	NB:  This is the only routine which MAY NOT lazy-evaluate
6915  *	or lose information.  That is, this routine must actually
6916  *	insert this page into the given map NOW.
6917  *
6918  *	When destroying both a page table and PV entry, this function
6919  *	performs the TLB invalidation before releasing the PV list
6920  *	lock, so we do not need pmap_delayed_invl_page() calls here.
6921  */
6922 int
pmap_enter(pmap_t pmap,vm_offset_t va,vm_page_t m,vm_prot_t prot,u_int flags,int8_t psind)6923 pmap_enter(pmap_t pmap, vm_offset_t va, vm_page_t m, vm_prot_t prot,
6924     u_int flags, int8_t psind)
6925 {
6926 	struct rwlock *lock;
6927 	pd_entry_t *pde;
6928 	pt_entry_t *pte, PG_G, PG_A, PG_M, PG_RW, PG_V;
6929 	pt_entry_t newpte, origpte;
6930 	pv_entry_t pv;
6931 	vm_paddr_t opa, pa;
6932 	vm_page_t mpte, om;
6933 	int rv;
6934 	boolean_t nosleep;
6935 
6936 	PG_A = pmap_accessed_bit(pmap);
6937 	PG_G = pmap_global_bit(pmap);
6938 	PG_M = pmap_modified_bit(pmap);
6939 	PG_V = pmap_valid_bit(pmap);
6940 	PG_RW = pmap_rw_bit(pmap);
6941 
6942 	va = trunc_page(va);
6943 	KASSERT(va <= VM_MAX_KERNEL_ADDRESS, ("pmap_enter: toobig"));
6944 	KASSERT(va < UPT_MIN_ADDRESS || va >= UPT_MAX_ADDRESS,
6945 	    ("pmap_enter: invalid to pmap_enter page table pages (va: 0x%lx)",
6946 	    va));
6947 	KASSERT((m->oflags & VPO_UNMANAGED) != 0 || !VA_IS_CLEANMAP(va),
6948 	    ("pmap_enter: managed mapping within the clean submap"));
6949 	if ((m->oflags & VPO_UNMANAGED) == 0)
6950 		VM_PAGE_OBJECT_BUSY_ASSERT(m);
6951 	KASSERT((flags & PMAP_ENTER_RESERVED) == 0,
6952 	    ("pmap_enter: flags %u has reserved bits set", flags));
6953 	pa = VM_PAGE_TO_PHYS(m);
6954 	newpte = (pt_entry_t)(pa | PG_A | PG_V);
6955 	if ((flags & VM_PROT_WRITE) != 0)
6956 		newpte |= PG_M;
6957 	if ((prot & VM_PROT_WRITE) != 0)
6958 		newpte |= PG_RW;
6959 	KASSERT((newpte & (PG_M | PG_RW)) != PG_M,
6960 	    ("pmap_enter: flags includes VM_PROT_WRITE but prot doesn't"));
6961 	if ((prot & VM_PROT_EXECUTE) == 0)
6962 		newpte |= pg_nx;
6963 	if ((flags & PMAP_ENTER_WIRED) != 0)
6964 		newpte |= PG_W;
6965 	if (va < VM_MAXUSER_ADDRESS)
6966 		newpte |= PG_U;
6967 	if (pmap == kernel_pmap)
6968 		newpte |= PG_G;
6969 	newpte |= pmap_cache_bits(pmap, m->md.pat_mode, psind > 0);
6970 
6971 	/*
6972 	 * Set modified bit gratuitously for writeable mappings if
6973 	 * the page is unmanaged. We do not want to take a fault
6974 	 * to do the dirty bit accounting for these mappings.
6975 	 */
6976 	if ((m->oflags & VPO_UNMANAGED) != 0) {
6977 		if ((newpte & PG_RW) != 0)
6978 			newpte |= PG_M;
6979 	} else
6980 		newpte |= PG_MANAGED;
6981 
6982 	lock = NULL;
6983 	PMAP_LOCK(pmap);
6984 	if ((flags & PMAP_ENTER_LARGEPAGE) != 0) {
6985 		KASSERT((m->oflags & VPO_UNMANAGED) != 0,
6986 		    ("managed largepage va %#lx flags %#x", va, flags));
6987 		rv = pmap_enter_largepage(pmap, va, newpte | PG_PS, flags,
6988 		    psind);
6989 		goto out;
6990 	}
6991 	if (psind == 1) {
6992 		/* Assert the required virtual and physical alignment. */
6993 		KASSERT((va & PDRMASK) == 0, ("pmap_enter: va unaligned"));
6994 		KASSERT(m->psind > 0, ("pmap_enter: m->psind < psind"));
6995 		rv = pmap_enter_pde(pmap, va, newpte | PG_PS, flags, m, &lock);
6996 		goto out;
6997 	}
6998 	mpte = NULL;
6999 
7000 	/*
7001 	 * In the case that a page table page is not
7002 	 * resident, we are creating it here.
7003 	 */
7004 retry:
7005 	pde = pmap_pde(pmap, va);
7006 	if (pde != NULL && (*pde & PG_V) != 0 && ((*pde & PG_PS) == 0 ||
7007 	    pmap_demote_pde_locked(pmap, pde, va, &lock))) {
7008 		pte = pmap_pde_to_pte(pde, va);
7009 		if (va < VM_MAXUSER_ADDRESS && mpte == NULL) {
7010 			mpte = PHYS_TO_VM_PAGE(*pde & PG_FRAME);
7011 			mpte->ref_count++;
7012 		}
7013 	} else if (va < VM_MAXUSER_ADDRESS) {
7014 		/*
7015 		 * Here if the pte page isn't mapped, or if it has been
7016 		 * deallocated.
7017 		 */
7018 		nosleep = (flags & PMAP_ENTER_NOSLEEP) != 0;
7019 		mpte = pmap_allocpte_alloc(pmap, pmap_pde_pindex(va),
7020 		    nosleep ? NULL : &lock, va);
7021 		if (mpte == NULL && nosleep) {
7022 			rv = KERN_RESOURCE_SHORTAGE;
7023 			goto out;
7024 		}
7025 		goto retry;
7026 	} else
7027 		panic("pmap_enter: invalid page directory va=%#lx", va);
7028 
7029 	origpte = *pte;
7030 	pv = NULL;
7031 	if (va < VM_MAXUSER_ADDRESS && pmap->pm_type == PT_X86)
7032 		newpte |= pmap_pkru_get(pmap, va);
7033 
7034 	/*
7035 	 * Is the specified virtual address already mapped?
7036 	 */
7037 	if ((origpte & PG_V) != 0) {
7038 		/*
7039 		 * Wiring change, just update stats. We don't worry about
7040 		 * wiring PT pages as they remain resident as long as there
7041 		 * are valid mappings in them. Hence, if a user page is wired,
7042 		 * the PT page will be also.
7043 		 */
7044 		if ((newpte & PG_W) != 0 && (origpte & PG_W) == 0)
7045 			pmap->pm_stats.wired_count++;
7046 		else if ((newpte & PG_W) == 0 && (origpte & PG_W) != 0)
7047 			pmap->pm_stats.wired_count--;
7048 
7049 		/*
7050 		 * Remove the extra PT page reference.
7051 		 */
7052 		if (mpte != NULL) {
7053 			mpte->ref_count--;
7054 			KASSERT(mpte->ref_count > 0,
7055 			    ("pmap_enter: missing reference to page table page,"
7056 			     " va: 0x%lx", va));
7057 		}
7058 
7059 		/*
7060 		 * Has the physical page changed?
7061 		 */
7062 		opa = origpte & PG_FRAME;
7063 		if (opa == pa) {
7064 			/*
7065 			 * No, might be a protection or wiring change.
7066 			 */
7067 			if ((origpte & PG_MANAGED) != 0 &&
7068 			    (newpte & PG_RW) != 0)
7069 				vm_page_aflag_set(m, PGA_WRITEABLE);
7070 			if (((origpte ^ newpte) & ~(PG_M | PG_A)) == 0)
7071 				goto unchanged;
7072 			goto validate;
7073 		}
7074 
7075 		/*
7076 		 * The physical page has changed.  Temporarily invalidate
7077 		 * the mapping.  This ensures that all threads sharing the
7078 		 * pmap keep a consistent view of the mapping, which is
7079 		 * necessary for the correct handling of COW faults.  It
7080 		 * also permits reuse of the old mapping's PV entry,
7081 		 * avoiding an allocation.
7082 		 *
7083 		 * For consistency, handle unmanaged mappings the same way.
7084 		 */
7085 		origpte = pte_load_clear(pte);
7086 		KASSERT((origpte & PG_FRAME) == opa,
7087 		    ("pmap_enter: unexpected pa update for %#lx", va));
7088 		if ((origpte & PG_MANAGED) != 0) {
7089 			om = PHYS_TO_VM_PAGE(opa);
7090 
7091 			/*
7092 			 * The pmap lock is sufficient to synchronize with
7093 			 * concurrent calls to pmap_page_test_mappings() and
7094 			 * pmap_ts_referenced().
7095 			 */
7096 			if ((origpte & (PG_M | PG_RW)) == (PG_M | PG_RW))
7097 				vm_page_dirty(om);
7098 			if ((origpte & PG_A) != 0) {
7099 				pmap_invalidate_page(pmap, va);
7100 				vm_page_aflag_set(om, PGA_REFERENCED);
7101 			}
7102 			CHANGE_PV_LIST_LOCK_TO_PHYS(&lock, opa);
7103 			pv = pmap_pvh_remove(&om->md, pmap, va);
7104 			KASSERT(pv != NULL,
7105 			    ("pmap_enter: no PV entry for %#lx", va));
7106 			if ((newpte & PG_MANAGED) == 0)
7107 				free_pv_entry(pmap, pv);
7108 			if ((om->a.flags & PGA_WRITEABLE) != 0 &&
7109 			    TAILQ_EMPTY(&om->md.pv_list) &&
7110 			    ((om->flags & PG_FICTITIOUS) != 0 ||
7111 			    TAILQ_EMPTY(&pa_to_pvh(opa)->pv_list)))
7112 				vm_page_aflag_clear(om, PGA_WRITEABLE);
7113 		} else {
7114 			/*
7115 			 * Since this mapping is unmanaged, assume that PG_A
7116 			 * is set.
7117 			 */
7118 			pmap_invalidate_page(pmap, va);
7119 		}
7120 		origpte = 0;
7121 	} else {
7122 		/*
7123 		 * Increment the counters.
7124 		 */
7125 		if ((newpte & PG_W) != 0)
7126 			pmap->pm_stats.wired_count++;
7127 		pmap_resident_count_adj(pmap, 1);
7128 	}
7129 
7130 	/*
7131 	 * Enter on the PV list if part of our managed memory.
7132 	 */
7133 	if ((newpte & PG_MANAGED) != 0) {
7134 		if (pv == NULL) {
7135 			pv = get_pv_entry(pmap, &lock);
7136 			pv->pv_va = va;
7137 		}
7138 		CHANGE_PV_LIST_LOCK_TO_PHYS(&lock, pa);
7139 		TAILQ_INSERT_TAIL(&m->md.pv_list, pv, pv_next);
7140 		m->md.pv_gen++;
7141 		if ((newpte & PG_RW) != 0)
7142 			vm_page_aflag_set(m, PGA_WRITEABLE);
7143 	}
7144 
7145 	/*
7146 	 * Update the PTE.
7147 	 */
7148 	if ((origpte & PG_V) != 0) {
7149 validate:
7150 		origpte = pte_load_store(pte, newpte);
7151 		KASSERT((origpte & PG_FRAME) == pa,
7152 		    ("pmap_enter: unexpected pa update for %#lx", va));
7153 		if ((newpte & PG_M) == 0 && (origpte & (PG_M | PG_RW)) ==
7154 		    (PG_M | PG_RW)) {
7155 			if ((origpte & PG_MANAGED) != 0)
7156 				vm_page_dirty(m);
7157 
7158 			/*
7159 			 * Although the PTE may still have PG_RW set, TLB
7160 			 * invalidation may nonetheless be required because
7161 			 * the PTE no longer has PG_M set.
7162 			 */
7163 		} else if ((origpte & PG_NX) != 0 || (newpte & PG_NX) == 0) {
7164 			/*
7165 			 * This PTE change does not require TLB invalidation.
7166 			 */
7167 			goto unchanged;
7168 		}
7169 		if ((origpte & PG_A) != 0)
7170 			pmap_invalidate_page(pmap, va);
7171 	} else
7172 		pte_store(pte, newpte);
7173 
7174 unchanged:
7175 
7176 #if VM_NRESERVLEVEL > 0
7177 	/*
7178 	 * If both the page table page and the reservation are fully
7179 	 * populated, then attempt promotion.
7180 	 */
7181 	if ((mpte == NULL || mpte->ref_count == NPTEPG) &&
7182 	    pmap_ps_enabled(pmap) &&
7183 	    (m->flags & PG_FICTITIOUS) == 0 &&
7184 	    vm_reserv_level_iffullpop(m) == 0)
7185 		pmap_promote_pde(pmap, pde, va, &lock);
7186 #endif
7187 
7188 	rv = KERN_SUCCESS;
7189 out:
7190 	if (lock != NULL)
7191 		rw_wunlock(lock);
7192 	PMAP_UNLOCK(pmap);
7193 	return (rv);
7194 }
7195 
7196 /*
7197  * Tries to create a read- and/or execute-only 2MB page mapping.  Returns true
7198  * if successful.  Returns false if (1) a page table page cannot be allocated
7199  * without sleeping, (2) a mapping already exists at the specified virtual
7200  * address, or (3) a PV entry cannot be allocated without reclaiming another
7201  * PV entry.
7202  */
7203 static bool
pmap_enter_2mpage(pmap_t pmap,vm_offset_t va,vm_page_t m,vm_prot_t prot,struct rwlock ** lockp)7204 pmap_enter_2mpage(pmap_t pmap, vm_offset_t va, vm_page_t m, vm_prot_t prot,
7205     struct rwlock **lockp)
7206 {
7207 	pd_entry_t newpde;
7208 	pt_entry_t PG_V;
7209 
7210 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
7211 	PG_V = pmap_valid_bit(pmap);
7212 	newpde = VM_PAGE_TO_PHYS(m) | pmap_cache_bits(pmap, m->md.pat_mode, 1) |
7213 	    PG_PS | PG_V;
7214 	if ((m->oflags & VPO_UNMANAGED) == 0)
7215 		newpde |= PG_MANAGED;
7216 	if ((prot & VM_PROT_EXECUTE) == 0)
7217 		newpde |= pg_nx;
7218 	if (va < VM_MAXUSER_ADDRESS)
7219 		newpde |= PG_U;
7220 	return (pmap_enter_pde(pmap, va, newpde, PMAP_ENTER_NOSLEEP |
7221 	    PMAP_ENTER_NOREPLACE | PMAP_ENTER_NORECLAIM, NULL, lockp) ==
7222 	    KERN_SUCCESS);
7223 }
7224 
7225 /*
7226  * Returns true if every page table entry in the specified page table page is
7227  * zero.
7228  */
7229 static bool
pmap_every_pte_zero(vm_paddr_t pa)7230 pmap_every_pte_zero(vm_paddr_t pa)
7231 {
7232 	pt_entry_t *pt_end, *pte;
7233 
7234 	KASSERT((pa & PAGE_MASK) == 0, ("pa is misaligned"));
7235 	pte = (pt_entry_t *)PHYS_TO_DMAP(pa);
7236 	for (pt_end = pte + NPTEPG; pte < pt_end; pte++) {
7237 		if (*pte != 0)
7238 			return (false);
7239 	}
7240 	return (true);
7241 }
7242 
7243 /*
7244  * Tries to create the specified 2MB page mapping.  Returns KERN_SUCCESS if
7245  * the mapping was created, and either KERN_FAILURE or KERN_RESOURCE_SHORTAGE
7246  * otherwise.  Returns KERN_FAILURE if PMAP_ENTER_NOREPLACE was specified and
7247  * a mapping already exists at the specified virtual address.  Returns
7248  * KERN_RESOURCE_SHORTAGE if PMAP_ENTER_NOSLEEP was specified and a page table
7249  * page allocation failed.  Returns KERN_RESOURCE_SHORTAGE if
7250  * PMAP_ENTER_NORECLAIM was specified and a PV entry allocation failed.
7251  *
7252  * The parameter "m" is only used when creating a managed, writeable mapping.
7253  */
7254 static int
pmap_enter_pde(pmap_t pmap,vm_offset_t va,pd_entry_t newpde,u_int flags,vm_page_t m,struct rwlock ** lockp)7255 pmap_enter_pde(pmap_t pmap, vm_offset_t va, pd_entry_t newpde, u_int flags,
7256     vm_page_t m, struct rwlock **lockp)
7257 {
7258 	struct spglist free;
7259 	pd_entry_t oldpde, *pde;
7260 	pt_entry_t PG_G, PG_RW, PG_V;
7261 	vm_page_t mt, pdpg;
7262 
7263 	KASSERT(pmap == kernel_pmap || (newpde & PG_W) == 0,
7264 	    ("pmap_enter_pde: cannot create wired user mapping"));
7265 	PG_G = pmap_global_bit(pmap);
7266 	PG_RW = pmap_rw_bit(pmap);
7267 	KASSERT((newpde & (pmap_modified_bit(pmap) | PG_RW)) != PG_RW,
7268 	    ("pmap_enter_pde: newpde is missing PG_M"));
7269 	PG_V = pmap_valid_bit(pmap);
7270 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
7271 
7272 	if (!pmap_allow_2m_x_page(pmap, pmap_pde_ept_executable(pmap,
7273 	    newpde))) {
7274 		CTR2(KTR_PMAP, "pmap_enter_pde: 2m x blocked for va %#lx"
7275 		    " in pmap %p", va, pmap);
7276 		return (KERN_FAILURE);
7277 	}
7278 	if ((pde = pmap_alloc_pde(pmap, va, &pdpg, (flags &
7279 	    PMAP_ENTER_NOSLEEP) != 0 ? NULL : lockp)) == NULL) {
7280 		CTR2(KTR_PMAP, "pmap_enter_pde: failure for va %#lx"
7281 		    " in pmap %p", va, pmap);
7282 		return (KERN_RESOURCE_SHORTAGE);
7283 	}
7284 
7285 	/*
7286 	 * If pkru is not same for the whole pde range, return failure
7287 	 * and let vm_fault() cope.  Check after pde allocation, since
7288 	 * it could sleep.
7289 	 */
7290 	if (!pmap_pkru_same(pmap, va, va + NBPDR)) {
7291 		pmap_abort_ptp(pmap, va, pdpg);
7292 		return (KERN_PROTECTION_FAILURE);
7293 	}
7294 	if (va < VM_MAXUSER_ADDRESS && pmap->pm_type == PT_X86) {
7295 		newpde &= ~X86_PG_PKU_MASK;
7296 		newpde |= pmap_pkru_get(pmap, va);
7297 	}
7298 
7299 	/*
7300 	 * If there are existing mappings, either abort or remove them.
7301 	 */
7302 	oldpde = *pde;
7303 	if ((oldpde & PG_V) != 0) {
7304 		KASSERT(pdpg == NULL || pdpg->ref_count > 1,
7305 		    ("pmap_enter_pde: pdpg's reference count is too low"));
7306 		if ((flags & PMAP_ENTER_NOREPLACE) != 0 && (va <
7307 		    VM_MAXUSER_ADDRESS || (oldpde & PG_PS) != 0 ||
7308 		    !pmap_every_pte_zero(oldpde & PG_FRAME))) {
7309 			if (pdpg != NULL)
7310 				pdpg->ref_count--;
7311 			CTR2(KTR_PMAP, "pmap_enter_pde: failure for va %#lx"
7312 			    " in pmap %p", va, pmap);
7313 			return (KERN_FAILURE);
7314 		}
7315 		/* Break the existing mapping(s). */
7316 		SLIST_INIT(&free);
7317 		if ((oldpde & PG_PS) != 0) {
7318 			/*
7319 			 * The reference to the PD page that was acquired by
7320 			 * pmap_alloc_pde() ensures that it won't be freed.
7321 			 * However, if the PDE resulted from a promotion, then
7322 			 * a reserved PT page could be freed.
7323 			 */
7324 			(void)pmap_remove_pde(pmap, pde, va, &free, lockp);
7325 			if ((oldpde & PG_G) == 0)
7326 				pmap_invalidate_pde_page(pmap, va, oldpde);
7327 		} else {
7328 			pmap_delayed_invl_start();
7329 			if (pmap_remove_ptes(pmap, va, va + NBPDR, pde, &free,
7330 			    lockp))
7331 		               pmap_invalidate_all(pmap);
7332 			pmap_delayed_invl_finish();
7333 		}
7334 		if (va < VM_MAXUSER_ADDRESS) {
7335 			vm_page_free_pages_toq(&free, true);
7336 			KASSERT(*pde == 0, ("pmap_enter_pde: non-zero pde %p",
7337 			    pde));
7338 		} else {
7339 			KASSERT(SLIST_EMPTY(&free),
7340 			    ("pmap_enter_pde: freed kernel page table page"));
7341 
7342 			/*
7343 			 * Both pmap_remove_pde() and pmap_remove_ptes() will
7344 			 * leave the kernel page table page zero filled.
7345 			 */
7346 			mt = PHYS_TO_VM_PAGE(*pde & PG_FRAME);
7347 			if (pmap_insert_pt_page(pmap, mt, false))
7348 				panic("pmap_enter_pde: trie insert failed");
7349 		}
7350 	}
7351 
7352 	if ((newpde & PG_MANAGED) != 0) {
7353 		/*
7354 		 * Abort this mapping if its PV entry could not be created.
7355 		 */
7356 		if (!pmap_pv_insert_pde(pmap, va, newpde, flags, lockp)) {
7357 			if (pdpg != NULL)
7358 				pmap_abort_ptp(pmap, va, pdpg);
7359 			CTR2(KTR_PMAP, "pmap_enter_pde: failure for va %#lx"
7360 			    " in pmap %p", va, pmap);
7361 			return (KERN_RESOURCE_SHORTAGE);
7362 		}
7363 		if ((newpde & PG_RW) != 0) {
7364 			for (mt = m; mt < &m[NBPDR / PAGE_SIZE]; mt++)
7365 				vm_page_aflag_set(mt, PGA_WRITEABLE);
7366 		}
7367 	}
7368 
7369 	/*
7370 	 * Increment counters.
7371 	 */
7372 	if ((newpde & PG_W) != 0)
7373 		pmap->pm_stats.wired_count += NBPDR / PAGE_SIZE;
7374 	pmap_resident_count_adj(pmap, NBPDR / PAGE_SIZE);
7375 
7376 	/*
7377 	 * Map the superpage.  (This is not a promoted mapping; there will not
7378 	 * be any lingering 4KB page mappings in the TLB.)
7379 	 */
7380 	pde_store(pde, newpde);
7381 
7382 	counter_u64_add(pmap_pde_mappings, 1);
7383 	CTR2(KTR_PMAP, "pmap_enter_pde: success for va %#lx in pmap %p",
7384 	    va, pmap);
7385 	return (KERN_SUCCESS);
7386 }
7387 
7388 /*
7389  * Maps a sequence of resident pages belonging to the same object.
7390  * The sequence begins with the given page m_start.  This page is
7391  * mapped at the given virtual address start.  Each subsequent page is
7392  * mapped at a virtual address that is offset from start by the same
7393  * amount as the page is offset from m_start within the object.  The
7394  * last page in the sequence is the page with the largest offset from
7395  * m_start that can be mapped at a virtual address less than the given
7396  * virtual address end.  Not every virtual page between start and end
7397  * is mapped; only those for which a resident page exists with the
7398  * corresponding offset from m_start are mapped.
7399  */
7400 void
pmap_enter_object(pmap_t pmap,vm_offset_t start,vm_offset_t end,vm_page_t m_start,vm_prot_t prot)7401 pmap_enter_object(pmap_t pmap, vm_offset_t start, vm_offset_t end,
7402     vm_page_t m_start, vm_prot_t prot)
7403 {
7404 	struct rwlock *lock;
7405 	vm_offset_t va;
7406 	vm_page_t m, mpte;
7407 	vm_pindex_t diff, psize;
7408 
7409 	VM_OBJECT_ASSERT_LOCKED(m_start->object);
7410 
7411 	psize = atop(end - start);
7412 	mpte = NULL;
7413 	m = m_start;
7414 	lock = NULL;
7415 	PMAP_LOCK(pmap);
7416 	while (m != NULL && (diff = m->pindex - m_start->pindex) < psize) {
7417 		va = start + ptoa(diff);
7418 		if ((va & PDRMASK) == 0 && va + NBPDR <= end &&
7419 		    m->psind == 1 && pmap_ps_enabled(pmap) &&
7420 		    pmap_enter_2mpage(pmap, va, m, prot, &lock))
7421 			m = &m[NBPDR / PAGE_SIZE - 1];
7422 		else
7423 			mpte = pmap_enter_quick_locked(pmap, va, m, prot,
7424 			    mpte, &lock);
7425 		m = TAILQ_NEXT(m, listq);
7426 	}
7427 	if (lock != NULL)
7428 		rw_wunlock(lock);
7429 	PMAP_UNLOCK(pmap);
7430 }
7431 
7432 /*
7433  * this code makes some *MAJOR* assumptions:
7434  * 1. Current pmap & pmap exists.
7435  * 2. Not wired.
7436  * 3. Read access.
7437  * 4. No page table pages.
7438  * but is *MUCH* faster than pmap_enter...
7439  */
7440 
7441 void
pmap_enter_quick(pmap_t pmap,vm_offset_t va,vm_page_t m,vm_prot_t prot)7442 pmap_enter_quick(pmap_t pmap, vm_offset_t va, vm_page_t m, vm_prot_t prot)
7443 {
7444 	struct rwlock *lock;
7445 
7446 	lock = NULL;
7447 	PMAP_LOCK(pmap);
7448 	(void)pmap_enter_quick_locked(pmap, va, m, prot, NULL, &lock);
7449 	if (lock != NULL)
7450 		rw_wunlock(lock);
7451 	PMAP_UNLOCK(pmap);
7452 }
7453 
7454 static vm_page_t
pmap_enter_quick_locked(pmap_t pmap,vm_offset_t va,vm_page_t m,vm_prot_t prot,vm_page_t mpte,struct rwlock ** lockp)7455 pmap_enter_quick_locked(pmap_t pmap, vm_offset_t va, vm_page_t m,
7456     vm_prot_t prot, vm_page_t mpte, struct rwlock **lockp)
7457 {
7458 	pt_entry_t newpte, *pte, PG_V;
7459 
7460 	KASSERT(!VA_IS_CLEANMAP(va) ||
7461 	    (m->oflags & VPO_UNMANAGED) != 0,
7462 	    ("pmap_enter_quick_locked: managed mapping within the clean submap"));
7463 	PG_V = pmap_valid_bit(pmap);
7464 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
7465 
7466 	/*
7467 	 * In the case that a page table page is not
7468 	 * resident, we are creating it here.
7469 	 */
7470 	if (va < VM_MAXUSER_ADDRESS) {
7471 		pdp_entry_t *pdpe;
7472 		pd_entry_t *pde;
7473 		vm_pindex_t ptepindex;
7474 
7475 		/*
7476 		 * Calculate pagetable page index
7477 		 */
7478 		ptepindex = pmap_pde_pindex(va);
7479 		if (mpte && (mpte->pindex == ptepindex)) {
7480 			mpte->ref_count++;
7481 		} else {
7482 			/*
7483 			 * If the page table page is mapped, we just increment
7484 			 * the hold count, and activate it.  Otherwise, we
7485 			 * attempt to allocate a page table page, passing NULL
7486 			 * instead of the PV list lock pointer because we don't
7487 			 * intend to sleep.  If this attempt fails, we don't
7488 			 * retry.  Instead, we give up.
7489 			 */
7490 			pdpe = pmap_pdpe(pmap, va);
7491 			if (pdpe != NULL && (*pdpe & PG_V) != 0) {
7492 				if ((*pdpe & PG_PS) != 0)
7493 					return (NULL);
7494 				pde = pmap_pdpe_to_pde(pdpe, va);
7495 				if ((*pde & PG_V) != 0) {
7496 					if ((*pde & PG_PS) != 0)
7497 						return (NULL);
7498 					mpte = PHYS_TO_VM_PAGE(*pde & PG_FRAME);
7499 					mpte->ref_count++;
7500 				} else {
7501 					mpte = pmap_allocpte_alloc(pmap,
7502 					    ptepindex, NULL, va);
7503 					if (mpte == NULL)
7504 						return (NULL);
7505 				}
7506 			} else {
7507 				mpte = pmap_allocpte_alloc(pmap, ptepindex,
7508 				    NULL, va);
7509 				if (mpte == NULL)
7510 					return (NULL);
7511 			}
7512 		}
7513 		pte = (pt_entry_t *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(mpte));
7514 		pte = &pte[pmap_pte_index(va)];
7515 	} else {
7516 		mpte = NULL;
7517 		pte = vtopte(va);
7518 	}
7519 	if (*pte) {
7520 		if (mpte != NULL)
7521 			mpte->ref_count--;
7522 		return (NULL);
7523 	}
7524 
7525 	/*
7526 	 * Enter on the PV list if part of our managed memory.
7527 	 */
7528 	if ((m->oflags & VPO_UNMANAGED) == 0 &&
7529 	    !pmap_try_insert_pv_entry(pmap, va, m, lockp)) {
7530 		if (mpte != NULL)
7531 			pmap_abort_ptp(pmap, va, mpte);
7532 		return (NULL);
7533 	}
7534 
7535 	/*
7536 	 * Increment counters
7537 	 */
7538 	pmap_resident_count_adj(pmap, 1);
7539 
7540 	newpte = VM_PAGE_TO_PHYS(m) | PG_V |
7541 	    pmap_cache_bits(pmap, m->md.pat_mode, 0);
7542 	if ((m->oflags & VPO_UNMANAGED) == 0)
7543 		newpte |= PG_MANAGED;
7544 	if ((prot & VM_PROT_EXECUTE) == 0)
7545 		newpte |= pg_nx;
7546 	if (va < VM_MAXUSER_ADDRESS)
7547 		newpte |= PG_U | pmap_pkru_get(pmap, va);
7548 	pte_store(pte, newpte);
7549 	return (mpte);
7550 }
7551 
7552 /*
7553  * Make a temporary mapping for a physical address.  This is only intended
7554  * to be used for panic dumps.
7555  */
7556 void *
pmap_kenter_temporary(vm_paddr_t pa,int i)7557 pmap_kenter_temporary(vm_paddr_t pa, int i)
7558 {
7559 	vm_offset_t va;
7560 
7561 	va = (vm_offset_t)crashdumpmap + (i * PAGE_SIZE);
7562 	pmap_kenter(va, pa);
7563 	pmap_invlpg(kernel_pmap, va);
7564 	return ((void *)crashdumpmap);
7565 }
7566 
7567 /*
7568  * This code maps large physical mmap regions into the
7569  * processor address space.  Note that some shortcuts
7570  * are taken, but the code works.
7571  */
7572 void
pmap_object_init_pt(pmap_t pmap,vm_offset_t addr,vm_object_t object,vm_pindex_t pindex,vm_size_t size)7573 pmap_object_init_pt(pmap_t pmap, vm_offset_t addr, vm_object_t object,
7574     vm_pindex_t pindex, vm_size_t size)
7575 {
7576 	pd_entry_t *pde;
7577 	pt_entry_t PG_A, PG_M, PG_RW, PG_V;
7578 	vm_paddr_t pa, ptepa;
7579 	vm_page_t p, pdpg;
7580 	int pat_mode;
7581 
7582 	PG_A = pmap_accessed_bit(pmap);
7583 	PG_M = pmap_modified_bit(pmap);
7584 	PG_V = pmap_valid_bit(pmap);
7585 	PG_RW = pmap_rw_bit(pmap);
7586 
7587 	VM_OBJECT_ASSERT_WLOCKED(object);
7588 	KASSERT(object->type == OBJT_DEVICE || object->type == OBJT_SG,
7589 	    ("pmap_object_init_pt: non-device object"));
7590 	if ((addr & (NBPDR - 1)) == 0 && (size & (NBPDR - 1)) == 0) {
7591 		if (!pmap_ps_enabled(pmap))
7592 			return;
7593 		if (!vm_object_populate(object, pindex, pindex + atop(size)))
7594 			return;
7595 		p = vm_page_lookup(object, pindex);
7596 		KASSERT(vm_page_all_valid(p),
7597 		    ("pmap_object_init_pt: invalid page %p", p));
7598 		pat_mode = p->md.pat_mode;
7599 
7600 		/*
7601 		 * Abort the mapping if the first page is not physically
7602 		 * aligned to a 2MB page boundary.
7603 		 */
7604 		ptepa = VM_PAGE_TO_PHYS(p);
7605 		if (ptepa & (NBPDR - 1))
7606 			return;
7607 
7608 		/*
7609 		 * Skip the first page.  Abort the mapping if the rest of
7610 		 * the pages are not physically contiguous or have differing
7611 		 * memory attributes.
7612 		 */
7613 		p = TAILQ_NEXT(p, listq);
7614 		for (pa = ptepa + PAGE_SIZE; pa < ptepa + size;
7615 		    pa += PAGE_SIZE) {
7616 			KASSERT(vm_page_all_valid(p),
7617 			    ("pmap_object_init_pt: invalid page %p", p));
7618 			if (pa != VM_PAGE_TO_PHYS(p) ||
7619 			    pat_mode != p->md.pat_mode)
7620 				return;
7621 			p = TAILQ_NEXT(p, listq);
7622 		}
7623 
7624 		/*
7625 		 * Map using 2MB pages.  Since "ptepa" is 2M aligned and
7626 		 * "size" is a multiple of 2M, adding the PAT setting to "pa"
7627 		 * will not affect the termination of this loop.
7628 		 */
7629 		PMAP_LOCK(pmap);
7630 		for (pa = ptepa | pmap_cache_bits(pmap, pat_mode, 1);
7631 		    pa < ptepa + size; pa += NBPDR) {
7632 			pde = pmap_alloc_pde(pmap, addr, &pdpg, NULL);
7633 			if (pde == NULL) {
7634 				/*
7635 				 * The creation of mappings below is only an
7636 				 * optimization.  If a page directory page
7637 				 * cannot be allocated without blocking,
7638 				 * continue on to the next mapping rather than
7639 				 * blocking.
7640 				 */
7641 				addr += NBPDR;
7642 				continue;
7643 			}
7644 			if ((*pde & PG_V) == 0) {
7645 				pde_store(pde, pa | PG_PS | PG_M | PG_A |
7646 				    PG_U | PG_RW | PG_V);
7647 				pmap_resident_count_adj(pmap, NBPDR / PAGE_SIZE);
7648 				counter_u64_add(pmap_pde_mappings, 1);
7649 			} else {
7650 				/* Continue on if the PDE is already valid. */
7651 				pdpg->ref_count--;
7652 				KASSERT(pdpg->ref_count > 0,
7653 				    ("pmap_object_init_pt: missing reference "
7654 				    "to page directory page, va: 0x%lx", addr));
7655 			}
7656 			addr += NBPDR;
7657 		}
7658 		PMAP_UNLOCK(pmap);
7659 	}
7660 }
7661 
7662 /*
7663  *	Clear the wired attribute from the mappings for the specified range of
7664  *	addresses in the given pmap.  Every valid mapping within that range
7665  *	must have the wired attribute set.  In contrast, invalid mappings
7666  *	cannot have the wired attribute set, so they are ignored.
7667  *
7668  *	The wired attribute of the page table entry is not a hardware
7669  *	feature, so there is no need to invalidate any TLB entries.
7670  *	Since pmap_demote_pde() for the wired entry must never fail,
7671  *	pmap_delayed_invl_start()/finish() calls around the
7672  *	function are not needed.
7673  */
7674 void
pmap_unwire(pmap_t pmap,vm_offset_t sva,vm_offset_t eva)7675 pmap_unwire(pmap_t pmap, vm_offset_t sva, vm_offset_t eva)
7676 {
7677 	vm_offset_t va_next;
7678 	pml4_entry_t *pml4e;
7679 	pdp_entry_t *pdpe;
7680 	pd_entry_t *pde;
7681 	pt_entry_t *pte, PG_V, PG_G;
7682 
7683 	PG_V = pmap_valid_bit(pmap);
7684 	PG_G = pmap_global_bit(pmap);
7685 	PMAP_LOCK(pmap);
7686 	for (; sva < eva; sva = va_next) {
7687 		pml4e = pmap_pml4e(pmap, sva);
7688 		if (pml4e == NULL || (*pml4e & PG_V) == 0) {
7689 			va_next = (sva + NBPML4) & ~PML4MASK;
7690 			if (va_next < sva)
7691 				va_next = eva;
7692 			continue;
7693 		}
7694 
7695 		va_next = (sva + NBPDP) & ~PDPMASK;
7696 		if (va_next < sva)
7697 			va_next = eva;
7698 		pdpe = pmap_pml4e_to_pdpe(pml4e, sva);
7699 		if ((*pdpe & PG_V) == 0)
7700 			continue;
7701 		if ((*pdpe & PG_PS) != 0) {
7702 			KASSERT(va_next <= eva,
7703 			    ("partial update of non-transparent 1G mapping "
7704 			    "pdpe %#lx sva %#lx eva %#lx va_next %#lx",
7705 			    *pdpe, sva, eva, va_next));
7706 			MPASS(pmap != kernel_pmap); /* XXXKIB */
7707 			MPASS((*pdpe & (PG_MANAGED | PG_G)) == 0);
7708 			atomic_clear_long(pdpe, PG_W);
7709 			pmap->pm_stats.wired_count -= NBPDP / PAGE_SIZE;
7710 			continue;
7711 		}
7712 
7713 		va_next = (sva + NBPDR) & ~PDRMASK;
7714 		if (va_next < sva)
7715 			va_next = eva;
7716 		pde = pmap_pdpe_to_pde(pdpe, sva);
7717 		if ((*pde & PG_V) == 0)
7718 			continue;
7719 		if ((*pde & PG_PS) != 0) {
7720 			if ((*pde & PG_W) == 0)
7721 				panic("pmap_unwire: pde %#jx is missing PG_W",
7722 				    (uintmax_t)*pde);
7723 
7724 			/*
7725 			 * Are we unwiring the entire large page?  If not,
7726 			 * demote the mapping and fall through.
7727 			 */
7728 			if (sva + NBPDR == va_next && eva >= va_next) {
7729 				atomic_clear_long(pde, PG_W);
7730 				pmap->pm_stats.wired_count -= NBPDR /
7731 				    PAGE_SIZE;
7732 				continue;
7733 			} else if (!pmap_demote_pde(pmap, pde, sva))
7734 				panic("pmap_unwire: demotion failed");
7735 		}
7736 		if (va_next > eva)
7737 			va_next = eva;
7738 		for (pte = pmap_pde_to_pte(pde, sva); sva != va_next; pte++,
7739 		    sva += PAGE_SIZE) {
7740 			if ((*pte & PG_V) == 0)
7741 				continue;
7742 			if ((*pte & PG_W) == 0)
7743 				panic("pmap_unwire: pte %#jx is missing PG_W",
7744 				    (uintmax_t)*pte);
7745 
7746 			/*
7747 			 * PG_W must be cleared atomically.  Although the pmap
7748 			 * lock synchronizes access to PG_W, another processor
7749 			 * could be setting PG_M and/or PG_A concurrently.
7750 			 */
7751 			atomic_clear_long(pte, PG_W);
7752 			pmap->pm_stats.wired_count--;
7753 		}
7754 	}
7755 	PMAP_UNLOCK(pmap);
7756 }
7757 
7758 /*
7759  *	Copy the range specified by src_addr/len
7760  *	from the source map to the range dst_addr/len
7761  *	in the destination map.
7762  *
7763  *	This routine is only advisory and need not do anything.
7764  */
7765 void
pmap_copy(pmap_t dst_pmap,pmap_t src_pmap,vm_offset_t dst_addr,vm_size_t len,vm_offset_t src_addr)7766 pmap_copy(pmap_t dst_pmap, pmap_t src_pmap, vm_offset_t dst_addr, vm_size_t len,
7767     vm_offset_t src_addr)
7768 {
7769 	struct rwlock *lock;
7770 	pml4_entry_t *pml4e;
7771 	pdp_entry_t *pdpe;
7772 	pd_entry_t *pde, srcptepaddr;
7773 	pt_entry_t *dst_pte, PG_A, PG_M, PG_V, ptetemp, *src_pte;
7774 	vm_offset_t addr, end_addr, va_next;
7775 	vm_page_t dst_pdpg, dstmpte, srcmpte;
7776 
7777 	if (dst_addr != src_addr)
7778 		return;
7779 
7780 	if (dst_pmap->pm_type != src_pmap->pm_type)
7781 		return;
7782 
7783 	/*
7784 	 * EPT page table entries that require emulation of A/D bits are
7785 	 * sensitive to clearing the PG_A bit (aka EPT_PG_READ). Although
7786 	 * we clear PG_M (aka EPT_PG_WRITE) concomitantly, the PG_U bit
7787 	 * (aka EPT_PG_EXECUTE) could still be set. Since some EPT
7788 	 * implementations flag an EPT misconfiguration for exec-only
7789 	 * mappings we skip this function entirely for emulated pmaps.
7790 	 */
7791 	if (pmap_emulate_ad_bits(dst_pmap))
7792 		return;
7793 
7794 	end_addr = src_addr + len;
7795 	lock = NULL;
7796 	if (dst_pmap < src_pmap) {
7797 		PMAP_LOCK(dst_pmap);
7798 		PMAP_LOCK(src_pmap);
7799 	} else {
7800 		PMAP_LOCK(src_pmap);
7801 		PMAP_LOCK(dst_pmap);
7802 	}
7803 
7804 	PG_A = pmap_accessed_bit(dst_pmap);
7805 	PG_M = pmap_modified_bit(dst_pmap);
7806 	PG_V = pmap_valid_bit(dst_pmap);
7807 
7808 	for (addr = src_addr; addr < end_addr; addr = va_next) {
7809 		KASSERT(addr < UPT_MIN_ADDRESS,
7810 		    ("pmap_copy: invalid to pmap_copy page tables"));
7811 
7812 		pml4e = pmap_pml4e(src_pmap, addr);
7813 		if (pml4e == NULL || (*pml4e & PG_V) == 0) {
7814 			va_next = (addr + NBPML4) & ~PML4MASK;
7815 			if (va_next < addr)
7816 				va_next = end_addr;
7817 			continue;
7818 		}
7819 
7820 		va_next = (addr + NBPDP) & ~PDPMASK;
7821 		if (va_next < addr)
7822 			va_next = end_addr;
7823 		pdpe = pmap_pml4e_to_pdpe(pml4e, addr);
7824 		if ((*pdpe & PG_V) == 0)
7825 			continue;
7826 		if ((*pdpe & PG_PS) != 0) {
7827 			KASSERT(va_next <= end_addr,
7828 			    ("partial update of non-transparent 1G mapping "
7829 			    "pdpe %#lx sva %#lx eva %#lx va_next %#lx",
7830 			    *pdpe, addr, end_addr, va_next));
7831 			MPASS((addr & PDPMASK) == 0);
7832 			MPASS((*pdpe & PG_MANAGED) == 0);
7833 			srcptepaddr = *pdpe;
7834 			pdpe = pmap_pdpe(dst_pmap, addr);
7835 			if (pdpe == NULL) {
7836 				if (pmap_allocpte_alloc(dst_pmap,
7837 				    pmap_pml4e_pindex(addr), NULL, addr) ==
7838 				    NULL)
7839 					break;
7840 				pdpe = pmap_pdpe(dst_pmap, addr);
7841 			} else {
7842 				pml4e = pmap_pml4e(dst_pmap, addr);
7843 				dst_pdpg = PHYS_TO_VM_PAGE(*pml4e & PG_FRAME);
7844 				dst_pdpg->ref_count++;
7845 			}
7846 			KASSERT(*pdpe == 0,
7847 			    ("1G mapping present in dst pmap "
7848 			    "pdpe %#lx sva %#lx eva %#lx va_next %#lx",
7849 			    *pdpe, addr, end_addr, va_next));
7850 			*pdpe = srcptepaddr & ~PG_W;
7851 			pmap_resident_count_adj(dst_pmap, NBPDP / PAGE_SIZE);
7852 			continue;
7853 		}
7854 
7855 		va_next = (addr + NBPDR) & ~PDRMASK;
7856 		if (va_next < addr)
7857 			va_next = end_addr;
7858 
7859 		pde = pmap_pdpe_to_pde(pdpe, addr);
7860 		srcptepaddr = *pde;
7861 		if (srcptepaddr == 0)
7862 			continue;
7863 
7864 		if (srcptepaddr & PG_PS) {
7865 			/*
7866 			 * We can only virtual copy whole superpages.
7867 			 */
7868 			if ((addr & PDRMASK) != 0 || addr + NBPDR > end_addr)
7869 				continue;
7870 			pde = pmap_alloc_pde(dst_pmap, addr, &dst_pdpg, NULL);
7871 			if (pde == NULL)
7872 				break;
7873 			if (*pde == 0 && ((srcptepaddr & PG_MANAGED) == 0 ||
7874 			    pmap_pv_insert_pde(dst_pmap, addr, srcptepaddr,
7875 			    PMAP_ENTER_NORECLAIM, &lock))) {
7876 				/*
7877 				 * We leave the dirty bit unchanged because
7878 				 * managed read/write superpage mappings are
7879 				 * required to be dirty.  However, managed
7880 				 * superpage mappings are not required to
7881 				 * have their accessed bit set, so we clear
7882 				 * it because we don't know if this mapping
7883 				 * will be used.
7884 				 */
7885 				srcptepaddr &= ~PG_W;
7886 				if ((srcptepaddr & PG_MANAGED) != 0)
7887 					srcptepaddr &= ~PG_A;
7888 				*pde = srcptepaddr;
7889 				pmap_resident_count_adj(dst_pmap, NBPDR /
7890 				    PAGE_SIZE);
7891 				counter_u64_add(pmap_pde_mappings, 1);
7892 			} else
7893 				pmap_abort_ptp(dst_pmap, addr, dst_pdpg);
7894 			continue;
7895 		}
7896 
7897 		srcptepaddr &= PG_FRAME;
7898 		srcmpte = PHYS_TO_VM_PAGE(srcptepaddr);
7899 		KASSERT(srcmpte->ref_count > 0,
7900 		    ("pmap_copy: source page table page is unused"));
7901 
7902 		if (va_next > end_addr)
7903 			va_next = end_addr;
7904 
7905 		src_pte = (pt_entry_t *)PHYS_TO_DMAP(srcptepaddr);
7906 		src_pte = &src_pte[pmap_pte_index(addr)];
7907 		dstmpte = NULL;
7908 		for (; addr < va_next; addr += PAGE_SIZE, src_pte++) {
7909 			ptetemp = *src_pte;
7910 
7911 			/*
7912 			 * We only virtual copy managed pages.
7913 			 */
7914 			if ((ptetemp & PG_MANAGED) == 0)
7915 				continue;
7916 
7917 			if (dstmpte != NULL) {
7918 				KASSERT(dstmpte->pindex ==
7919 				    pmap_pde_pindex(addr),
7920 				    ("dstmpte pindex/addr mismatch"));
7921 				dstmpte->ref_count++;
7922 			} else if ((dstmpte = pmap_allocpte(dst_pmap, addr,
7923 			    NULL)) == NULL)
7924 				goto out;
7925 			dst_pte = (pt_entry_t *)
7926 			    PHYS_TO_DMAP(VM_PAGE_TO_PHYS(dstmpte));
7927 			dst_pte = &dst_pte[pmap_pte_index(addr)];
7928 			if (*dst_pte == 0 &&
7929 			    pmap_try_insert_pv_entry(dst_pmap, addr,
7930 			    PHYS_TO_VM_PAGE(ptetemp & PG_FRAME), &lock)) {
7931 				/*
7932 				 * Clear the wired, modified, and accessed
7933 				 * (referenced) bits during the copy.
7934 				 */
7935 				*dst_pte = ptetemp & ~(PG_W | PG_M | PG_A);
7936 				pmap_resident_count_adj(dst_pmap, 1);
7937 			} else {
7938 				pmap_abort_ptp(dst_pmap, addr, dstmpte);
7939 				goto out;
7940 			}
7941 			/* Have we copied all of the valid mappings? */
7942 			if (dstmpte->ref_count >= srcmpte->ref_count)
7943 				break;
7944 		}
7945 	}
7946 out:
7947 	if (lock != NULL)
7948 		rw_wunlock(lock);
7949 	PMAP_UNLOCK(src_pmap);
7950 	PMAP_UNLOCK(dst_pmap);
7951 }
7952 
7953 int
pmap_vmspace_copy(pmap_t dst_pmap,pmap_t src_pmap)7954 pmap_vmspace_copy(pmap_t dst_pmap, pmap_t src_pmap)
7955 {
7956 	int error;
7957 
7958 	if (dst_pmap->pm_type != src_pmap->pm_type ||
7959 	    dst_pmap->pm_type != PT_X86 ||
7960 	    (cpu_stdext_feature2 & CPUID_STDEXT2_PKU) == 0)
7961 		return (0);
7962 	for (;;) {
7963 		if (dst_pmap < src_pmap) {
7964 			PMAP_LOCK(dst_pmap);
7965 			PMAP_LOCK(src_pmap);
7966 		} else {
7967 			PMAP_LOCK(src_pmap);
7968 			PMAP_LOCK(dst_pmap);
7969 		}
7970 		error = pmap_pkru_copy(dst_pmap, src_pmap);
7971 		/* Clean up partial copy on failure due to no memory. */
7972 		if (error == ENOMEM)
7973 			pmap_pkru_deassign_all(dst_pmap);
7974 		PMAP_UNLOCK(src_pmap);
7975 		PMAP_UNLOCK(dst_pmap);
7976 		if (error != ENOMEM)
7977 			break;
7978 		vm_wait(NULL);
7979 	}
7980 	return (error);
7981 }
7982 
7983 /*
7984  * Zero the specified hardware page.
7985  */
7986 void
pmap_zero_page(vm_page_t m)7987 pmap_zero_page(vm_page_t m)
7988 {
7989 	vm_offset_t va = PHYS_TO_DMAP(VM_PAGE_TO_PHYS(m));
7990 
7991 	pagezero((void *)va);
7992 }
7993 
7994 /*
7995  * Zero an area within a single hardware page.  off and size must not
7996  * cover an area beyond a single hardware page.
7997  */
7998 void
pmap_zero_page_area(vm_page_t m,int off,int size)7999 pmap_zero_page_area(vm_page_t m, int off, int size)
8000 {
8001 	vm_offset_t va = PHYS_TO_DMAP(VM_PAGE_TO_PHYS(m));
8002 
8003 	if (off == 0 && size == PAGE_SIZE)
8004 		pagezero((void *)va);
8005 	else
8006 		bzero((char *)va + off, size);
8007 }
8008 
8009 /*
8010  * Copy 1 specified hardware page to another.
8011  */
8012 void
pmap_copy_page(vm_page_t msrc,vm_page_t mdst)8013 pmap_copy_page(vm_page_t msrc, vm_page_t mdst)
8014 {
8015 	vm_offset_t src = PHYS_TO_DMAP(VM_PAGE_TO_PHYS(msrc));
8016 	vm_offset_t dst = PHYS_TO_DMAP(VM_PAGE_TO_PHYS(mdst));
8017 
8018 	pagecopy((void *)src, (void *)dst);
8019 }
8020 
8021 int unmapped_buf_allowed = 1;
8022 
8023 void
pmap_copy_pages(vm_page_t ma[],vm_offset_t a_offset,vm_page_t mb[],vm_offset_t b_offset,int xfersize)8024 pmap_copy_pages(vm_page_t ma[], vm_offset_t a_offset, vm_page_t mb[],
8025     vm_offset_t b_offset, int xfersize)
8026 {
8027 	void *a_cp, *b_cp;
8028 	vm_page_t pages[2];
8029 	vm_offset_t vaddr[2], a_pg_offset, b_pg_offset;
8030 	int cnt;
8031 	boolean_t mapped;
8032 
8033 	while (xfersize > 0) {
8034 		a_pg_offset = a_offset & PAGE_MASK;
8035 		pages[0] = ma[a_offset >> PAGE_SHIFT];
8036 		b_pg_offset = b_offset & PAGE_MASK;
8037 		pages[1] = mb[b_offset >> PAGE_SHIFT];
8038 		cnt = min(xfersize, PAGE_SIZE - a_pg_offset);
8039 		cnt = min(cnt, PAGE_SIZE - b_pg_offset);
8040 		mapped = pmap_map_io_transient(pages, vaddr, 2, FALSE);
8041 		a_cp = (char *)vaddr[0] + a_pg_offset;
8042 		b_cp = (char *)vaddr[1] + b_pg_offset;
8043 		bcopy(a_cp, b_cp, cnt);
8044 		if (__predict_false(mapped))
8045 			pmap_unmap_io_transient(pages, vaddr, 2, FALSE);
8046 		a_offset += cnt;
8047 		b_offset += cnt;
8048 		xfersize -= cnt;
8049 	}
8050 }
8051 
8052 /*
8053  * Returns true if the pmap's pv is one of the first
8054  * 16 pvs linked to from this page.  This count may
8055  * be changed upwards or downwards in the future; it
8056  * is only necessary that true be returned for a small
8057  * subset of pmaps for proper page aging.
8058  */
8059 boolean_t
pmap_page_exists_quick(pmap_t pmap,vm_page_t m)8060 pmap_page_exists_quick(pmap_t pmap, vm_page_t m)
8061 {
8062 	struct md_page *pvh;
8063 	struct rwlock *lock;
8064 	pv_entry_t pv;
8065 	int loops = 0;
8066 	boolean_t rv;
8067 
8068 	KASSERT((m->oflags & VPO_UNMANAGED) == 0,
8069 	    ("pmap_page_exists_quick: page %p is not managed", m));
8070 	rv = FALSE;
8071 	lock = VM_PAGE_TO_PV_LIST_LOCK(m);
8072 	rw_rlock(lock);
8073 	TAILQ_FOREACH(pv, &m->md.pv_list, pv_next) {
8074 		if (PV_PMAP(pv) == pmap) {
8075 			rv = TRUE;
8076 			break;
8077 		}
8078 		loops++;
8079 		if (loops >= 16)
8080 			break;
8081 	}
8082 	if (!rv && loops < 16 && (m->flags & PG_FICTITIOUS) == 0) {
8083 		pvh = pa_to_pvh(VM_PAGE_TO_PHYS(m));
8084 		TAILQ_FOREACH(pv, &pvh->pv_list, pv_next) {
8085 			if (PV_PMAP(pv) == pmap) {
8086 				rv = TRUE;
8087 				break;
8088 			}
8089 			loops++;
8090 			if (loops >= 16)
8091 				break;
8092 		}
8093 	}
8094 	rw_runlock(lock);
8095 	return (rv);
8096 }
8097 
8098 /*
8099  *	pmap_page_wired_mappings:
8100  *
8101  *	Return the number of managed mappings to the given physical page
8102  *	that are wired.
8103  */
8104 int
pmap_page_wired_mappings(vm_page_t m)8105 pmap_page_wired_mappings(vm_page_t m)
8106 {
8107 	struct rwlock *lock;
8108 	struct md_page *pvh;
8109 	pmap_t pmap;
8110 	pt_entry_t *pte;
8111 	pv_entry_t pv;
8112 	int count, md_gen, pvh_gen;
8113 
8114 	if ((m->oflags & VPO_UNMANAGED) != 0)
8115 		return (0);
8116 	lock = VM_PAGE_TO_PV_LIST_LOCK(m);
8117 	rw_rlock(lock);
8118 restart:
8119 	count = 0;
8120 	TAILQ_FOREACH(pv, &m->md.pv_list, pv_next) {
8121 		pmap = PV_PMAP(pv);
8122 		if (!PMAP_TRYLOCK(pmap)) {
8123 			md_gen = m->md.pv_gen;
8124 			rw_runlock(lock);
8125 			PMAP_LOCK(pmap);
8126 			rw_rlock(lock);
8127 			if (md_gen != m->md.pv_gen) {
8128 				PMAP_UNLOCK(pmap);
8129 				goto restart;
8130 			}
8131 		}
8132 		pte = pmap_pte(pmap, pv->pv_va);
8133 		if ((*pte & PG_W) != 0)
8134 			count++;
8135 		PMAP_UNLOCK(pmap);
8136 	}
8137 	if ((m->flags & PG_FICTITIOUS) == 0) {
8138 		pvh = pa_to_pvh(VM_PAGE_TO_PHYS(m));
8139 		TAILQ_FOREACH(pv, &pvh->pv_list, pv_next) {
8140 			pmap = PV_PMAP(pv);
8141 			if (!PMAP_TRYLOCK(pmap)) {
8142 				md_gen = m->md.pv_gen;
8143 				pvh_gen = pvh->pv_gen;
8144 				rw_runlock(lock);
8145 				PMAP_LOCK(pmap);
8146 				rw_rlock(lock);
8147 				if (md_gen != m->md.pv_gen ||
8148 				    pvh_gen != pvh->pv_gen) {
8149 					PMAP_UNLOCK(pmap);
8150 					goto restart;
8151 				}
8152 			}
8153 			pte = pmap_pde(pmap, pv->pv_va);
8154 			if ((*pte & PG_W) != 0)
8155 				count++;
8156 			PMAP_UNLOCK(pmap);
8157 		}
8158 	}
8159 	rw_runlock(lock);
8160 	return (count);
8161 }
8162 
8163 /*
8164  * Returns TRUE if the given page is mapped individually or as part of
8165  * a 2mpage.  Otherwise, returns FALSE.
8166  */
8167 boolean_t
pmap_page_is_mapped(vm_page_t m)8168 pmap_page_is_mapped(vm_page_t m)
8169 {
8170 	struct rwlock *lock;
8171 	boolean_t rv;
8172 
8173 	if ((m->oflags & VPO_UNMANAGED) != 0)
8174 		return (FALSE);
8175 	lock = VM_PAGE_TO_PV_LIST_LOCK(m);
8176 	rw_rlock(lock);
8177 	rv = !TAILQ_EMPTY(&m->md.pv_list) ||
8178 	    ((m->flags & PG_FICTITIOUS) == 0 &&
8179 	    !TAILQ_EMPTY(&pa_to_pvh(VM_PAGE_TO_PHYS(m))->pv_list));
8180 	rw_runlock(lock);
8181 	return (rv);
8182 }
8183 
8184 /*
8185  * Destroy all managed, non-wired mappings in the given user-space
8186  * pmap.  This pmap cannot be active on any processor besides the
8187  * caller.
8188  *
8189  * This function cannot be applied to the kernel pmap.  Moreover, it
8190  * is not intended for general use.  It is only to be used during
8191  * process termination.  Consequently, it can be implemented in ways
8192  * that make it faster than pmap_remove().  First, it can more quickly
8193  * destroy mappings by iterating over the pmap's collection of PV
8194  * entries, rather than searching the page table.  Second, it doesn't
8195  * have to test and clear the page table entries atomically, because
8196  * no processor is currently accessing the user address space.  In
8197  * particular, a page table entry's dirty bit won't change state once
8198  * this function starts.
8199  *
8200  * Although this function destroys all of the pmap's managed,
8201  * non-wired mappings, it can delay and batch the invalidation of TLB
8202  * entries without calling pmap_delayed_invl_start() and
8203  * pmap_delayed_invl_finish().  Because the pmap is not active on
8204  * any other processor, none of these TLB entries will ever be used
8205  * before their eventual invalidation.  Consequently, there is no need
8206  * for either pmap_remove_all() or pmap_remove_write() to wait for
8207  * that eventual TLB invalidation.
8208  */
8209 void
pmap_remove_pages(pmap_t pmap)8210 pmap_remove_pages(pmap_t pmap)
8211 {
8212 	pd_entry_t ptepde;
8213 	pt_entry_t *pte, tpte;
8214 	pt_entry_t PG_M, PG_RW, PG_V;
8215 	struct spglist free;
8216 	struct pv_chunklist free_chunks[PMAP_MEMDOM];
8217 	vm_page_t m, mpte, mt;
8218 	pv_entry_t pv;
8219 	struct md_page *pvh;
8220 	struct pv_chunk *pc, *npc;
8221 	struct rwlock *lock;
8222 	int64_t bit;
8223 	uint64_t inuse, bitmask;
8224 	int allfree, field, i, idx;
8225 #ifdef PV_STATS
8226 	int freed;
8227 #endif
8228 	boolean_t superpage;
8229 	vm_paddr_t pa;
8230 
8231 	/*
8232 	 * Assert that the given pmap is only active on the current
8233 	 * CPU.  Unfortunately, we cannot block another CPU from
8234 	 * activating the pmap while this function is executing.
8235 	 */
8236 	KASSERT(pmap == PCPU_GET(curpmap), ("non-current pmap %p", pmap));
8237 #ifdef INVARIANTS
8238 	{
8239 		cpuset_t other_cpus;
8240 
8241 		other_cpus = all_cpus;
8242 		critical_enter();
8243 		CPU_CLR(PCPU_GET(cpuid), &other_cpus);
8244 		CPU_AND(&other_cpus, &other_cpus, &pmap->pm_active);
8245 		critical_exit();
8246 		KASSERT(CPU_EMPTY(&other_cpus), ("pmap active %p", pmap));
8247 	}
8248 #endif
8249 
8250 	lock = NULL;
8251 	PG_M = pmap_modified_bit(pmap);
8252 	PG_V = pmap_valid_bit(pmap);
8253 	PG_RW = pmap_rw_bit(pmap);
8254 
8255 	for (i = 0; i < PMAP_MEMDOM; i++)
8256 		TAILQ_INIT(&free_chunks[i]);
8257 	SLIST_INIT(&free);
8258 	PMAP_LOCK(pmap);
8259 	TAILQ_FOREACH_SAFE(pc, &pmap->pm_pvchunk, pc_list, npc) {
8260 		allfree = 1;
8261 #ifdef PV_STATS
8262 		freed = 0;
8263 #endif
8264 		for (field = 0; field < _NPCM; field++) {
8265 			inuse = ~pc->pc_map[field] & pc_freemask[field];
8266 			while (inuse != 0) {
8267 				bit = bsfq(inuse);
8268 				bitmask = 1UL << bit;
8269 				idx = field * 64 + bit;
8270 				pv = &pc->pc_pventry[idx];
8271 				inuse &= ~bitmask;
8272 
8273 				pte = pmap_pdpe(pmap, pv->pv_va);
8274 				ptepde = *pte;
8275 				pte = pmap_pdpe_to_pde(pte, pv->pv_va);
8276 				tpte = *pte;
8277 				if ((tpte & (PG_PS | PG_V)) == PG_V) {
8278 					superpage = FALSE;
8279 					ptepde = tpte;
8280 					pte = (pt_entry_t *)PHYS_TO_DMAP(tpte &
8281 					    PG_FRAME);
8282 					pte = &pte[pmap_pte_index(pv->pv_va)];
8283 					tpte = *pte;
8284 				} else {
8285 					/*
8286 					 * Keep track whether 'tpte' is a
8287 					 * superpage explicitly instead of
8288 					 * relying on PG_PS being set.
8289 					 *
8290 					 * This is because PG_PS is numerically
8291 					 * identical to PG_PTE_PAT and thus a
8292 					 * regular page could be mistaken for
8293 					 * a superpage.
8294 					 */
8295 					superpage = TRUE;
8296 				}
8297 
8298 				if ((tpte & PG_V) == 0) {
8299 					panic("bad pte va %lx pte %lx",
8300 					    pv->pv_va, tpte);
8301 				}
8302 
8303 /*
8304  * We cannot remove wired pages from a process' mapping at this time
8305  */
8306 				if (tpte & PG_W) {
8307 					allfree = 0;
8308 					continue;
8309 				}
8310 
8311 				/* Mark free */
8312 				pc->pc_map[field] |= bitmask;
8313 
8314 				/*
8315 				 * Because this pmap is not active on other
8316 				 * processors, the dirty bit cannot have
8317 				 * changed state since we last loaded pte.
8318 				 */
8319 				pte_clear(pte);
8320 
8321 				if (superpage)
8322 					pa = tpte & PG_PS_FRAME;
8323 				else
8324 					pa = tpte & PG_FRAME;
8325 
8326 				m = PHYS_TO_VM_PAGE(pa);
8327 				KASSERT(m->phys_addr == pa,
8328 				    ("vm_page_t %p phys_addr mismatch %016jx %016jx",
8329 				    m, (uintmax_t)m->phys_addr,
8330 				    (uintmax_t)tpte));
8331 
8332 				KASSERT((m->flags & PG_FICTITIOUS) != 0 ||
8333 				    m < &vm_page_array[vm_page_array_size],
8334 				    ("pmap_remove_pages: bad tpte %#jx",
8335 				    (uintmax_t)tpte));
8336 
8337 				/*
8338 				 * Update the vm_page_t clean/reference bits.
8339 				 */
8340 				if ((tpte & (PG_M | PG_RW)) == (PG_M | PG_RW)) {
8341 					if (superpage) {
8342 						for (mt = m; mt < &m[NBPDR / PAGE_SIZE]; mt++)
8343 							vm_page_dirty(mt);
8344 					} else
8345 						vm_page_dirty(m);
8346 				}
8347 
8348 				CHANGE_PV_LIST_LOCK_TO_VM_PAGE(&lock, m);
8349 
8350 				if (superpage) {
8351 					pmap_resident_count_adj(pmap, -NBPDR / PAGE_SIZE);
8352 					pvh = pa_to_pvh(tpte & PG_PS_FRAME);
8353 					TAILQ_REMOVE(&pvh->pv_list, pv, pv_next);
8354 					pvh->pv_gen++;
8355 					if (TAILQ_EMPTY(&pvh->pv_list)) {
8356 						for (mt = m; mt < &m[NBPDR / PAGE_SIZE]; mt++)
8357 							if ((mt->a.flags & PGA_WRITEABLE) != 0 &&
8358 							    TAILQ_EMPTY(&mt->md.pv_list))
8359 								vm_page_aflag_clear(mt, PGA_WRITEABLE);
8360 					}
8361 					mpte = pmap_remove_pt_page(pmap, pv->pv_va);
8362 					if (mpte != NULL) {
8363 						KASSERT(vm_page_all_valid(mpte),
8364 						    ("pmap_remove_pages: pte page not promoted"));
8365 						pmap_resident_count_adj(pmap, -1);
8366 						KASSERT(mpte->ref_count == NPTEPG,
8367 						    ("pmap_remove_pages: pte page reference count error"));
8368 						mpte->ref_count = 0;
8369 						pmap_add_delayed_free_list(mpte, &free, FALSE);
8370 					}
8371 				} else {
8372 					pmap_resident_count_adj(pmap, -1);
8373 					TAILQ_REMOVE(&m->md.pv_list, pv, pv_next);
8374 					m->md.pv_gen++;
8375 					if ((m->a.flags & PGA_WRITEABLE) != 0 &&
8376 					    TAILQ_EMPTY(&m->md.pv_list) &&
8377 					    (m->flags & PG_FICTITIOUS) == 0) {
8378 						pvh = pa_to_pvh(VM_PAGE_TO_PHYS(m));
8379 						if (TAILQ_EMPTY(&pvh->pv_list))
8380 							vm_page_aflag_clear(m, PGA_WRITEABLE);
8381 					}
8382 				}
8383 				pmap_unuse_pt(pmap, pv->pv_va, ptepde, &free);
8384 #ifdef PV_STATS
8385 				freed++;
8386 #endif
8387 			}
8388 		}
8389 		PV_STAT(counter_u64_add(pv_entry_frees, freed));
8390 		PV_STAT(counter_u64_add(pv_entry_spare, freed));
8391 		PV_STAT(counter_u64_add(pv_entry_count, -freed));
8392 		if (allfree) {
8393 			TAILQ_REMOVE(&pmap->pm_pvchunk, pc, pc_list);
8394 			TAILQ_INSERT_TAIL(&free_chunks[pc_to_domain(pc)], pc, pc_list);
8395 		}
8396 	}
8397 	if (lock != NULL)
8398 		rw_wunlock(lock);
8399 	pmap_invalidate_all(pmap);
8400 	pmap_pkru_deassign_all(pmap);
8401 	free_pv_chunk_batch((struct pv_chunklist *)&free_chunks);
8402 	PMAP_UNLOCK(pmap);
8403 	vm_page_free_pages_toq(&free, true);
8404 }
8405 
8406 static boolean_t
pmap_page_test_mappings(vm_page_t m,boolean_t accessed,boolean_t modified)8407 pmap_page_test_mappings(vm_page_t m, boolean_t accessed, boolean_t modified)
8408 {
8409 	struct rwlock *lock;
8410 	pv_entry_t pv;
8411 	struct md_page *pvh;
8412 	pt_entry_t *pte, mask;
8413 	pt_entry_t PG_A, PG_M, PG_RW, PG_V;
8414 	pmap_t pmap;
8415 	int md_gen, pvh_gen;
8416 	boolean_t rv;
8417 
8418 	rv = FALSE;
8419 	lock = VM_PAGE_TO_PV_LIST_LOCK(m);
8420 	rw_rlock(lock);
8421 restart:
8422 	TAILQ_FOREACH(pv, &m->md.pv_list, pv_next) {
8423 		pmap = PV_PMAP(pv);
8424 		if (!PMAP_TRYLOCK(pmap)) {
8425 			md_gen = m->md.pv_gen;
8426 			rw_runlock(lock);
8427 			PMAP_LOCK(pmap);
8428 			rw_rlock(lock);
8429 			if (md_gen != m->md.pv_gen) {
8430 				PMAP_UNLOCK(pmap);
8431 				goto restart;
8432 			}
8433 		}
8434 		pte = pmap_pte(pmap, pv->pv_va);
8435 		mask = 0;
8436 		if (modified) {
8437 			PG_M = pmap_modified_bit(pmap);
8438 			PG_RW = pmap_rw_bit(pmap);
8439 			mask |= PG_RW | PG_M;
8440 		}
8441 		if (accessed) {
8442 			PG_A = pmap_accessed_bit(pmap);
8443 			PG_V = pmap_valid_bit(pmap);
8444 			mask |= PG_V | PG_A;
8445 		}
8446 		rv = (*pte & mask) == mask;
8447 		PMAP_UNLOCK(pmap);
8448 		if (rv)
8449 			goto out;
8450 	}
8451 	if ((m->flags & PG_FICTITIOUS) == 0) {
8452 		pvh = pa_to_pvh(VM_PAGE_TO_PHYS(m));
8453 		TAILQ_FOREACH(pv, &pvh->pv_list, pv_next) {
8454 			pmap = PV_PMAP(pv);
8455 			if (!PMAP_TRYLOCK(pmap)) {
8456 				md_gen = m->md.pv_gen;
8457 				pvh_gen = pvh->pv_gen;
8458 				rw_runlock(lock);
8459 				PMAP_LOCK(pmap);
8460 				rw_rlock(lock);
8461 				if (md_gen != m->md.pv_gen ||
8462 				    pvh_gen != pvh->pv_gen) {
8463 					PMAP_UNLOCK(pmap);
8464 					goto restart;
8465 				}
8466 			}
8467 			pte = pmap_pde(pmap, pv->pv_va);
8468 			mask = 0;
8469 			if (modified) {
8470 				PG_M = pmap_modified_bit(pmap);
8471 				PG_RW = pmap_rw_bit(pmap);
8472 				mask |= PG_RW | PG_M;
8473 			}
8474 			if (accessed) {
8475 				PG_A = pmap_accessed_bit(pmap);
8476 				PG_V = pmap_valid_bit(pmap);
8477 				mask |= PG_V | PG_A;
8478 			}
8479 			rv = (*pte & mask) == mask;
8480 			PMAP_UNLOCK(pmap);
8481 			if (rv)
8482 				goto out;
8483 		}
8484 	}
8485 out:
8486 	rw_runlock(lock);
8487 	return (rv);
8488 }
8489 
8490 /*
8491  *	pmap_is_modified:
8492  *
8493  *	Return whether or not the specified physical page was modified
8494  *	in any physical maps.
8495  */
8496 boolean_t
pmap_is_modified(vm_page_t m)8497 pmap_is_modified(vm_page_t m)
8498 {
8499 
8500 	KASSERT((m->oflags & VPO_UNMANAGED) == 0,
8501 	    ("pmap_is_modified: page %p is not managed", m));
8502 
8503 	/*
8504 	 * If the page is not busied then this check is racy.
8505 	 */
8506 	if (!pmap_page_is_write_mapped(m))
8507 		return (FALSE);
8508 	return (pmap_page_test_mappings(m, FALSE, TRUE));
8509 }
8510 
8511 /*
8512  *	pmap_is_prefaultable:
8513  *
8514  *	Return whether or not the specified virtual address is eligible
8515  *	for prefault.
8516  */
8517 boolean_t
pmap_is_prefaultable(pmap_t pmap,vm_offset_t addr)8518 pmap_is_prefaultable(pmap_t pmap, vm_offset_t addr)
8519 {
8520 	pd_entry_t *pde;
8521 	pt_entry_t *pte, PG_V;
8522 	boolean_t rv;
8523 
8524 	PG_V = pmap_valid_bit(pmap);
8525 	rv = FALSE;
8526 	PMAP_LOCK(pmap);
8527 	pde = pmap_pde(pmap, addr);
8528 	if (pde != NULL && (*pde & (PG_PS | PG_V)) == PG_V) {
8529 		pte = pmap_pde_to_pte(pde, addr);
8530 		rv = (*pte & PG_V) == 0;
8531 	}
8532 	PMAP_UNLOCK(pmap);
8533 	return (rv);
8534 }
8535 
8536 /*
8537  *	pmap_is_referenced:
8538  *
8539  *	Return whether or not the specified physical page was referenced
8540  *	in any physical maps.
8541  */
8542 boolean_t
pmap_is_referenced(vm_page_t m)8543 pmap_is_referenced(vm_page_t m)
8544 {
8545 
8546 	KASSERT((m->oflags & VPO_UNMANAGED) == 0,
8547 	    ("pmap_is_referenced: page %p is not managed", m));
8548 	return (pmap_page_test_mappings(m, TRUE, FALSE));
8549 }
8550 
8551 /*
8552  * Clear the write and modified bits in each of the given page's mappings.
8553  */
8554 void
pmap_remove_write(vm_page_t m)8555 pmap_remove_write(vm_page_t m)
8556 {
8557 	struct md_page *pvh;
8558 	pmap_t pmap;
8559 	struct rwlock *lock;
8560 	pv_entry_t next_pv, pv;
8561 	pd_entry_t *pde;
8562 	pt_entry_t oldpte, *pte, PG_M, PG_RW;
8563 	vm_offset_t va;
8564 	int pvh_gen, md_gen;
8565 
8566 	KASSERT((m->oflags & VPO_UNMANAGED) == 0,
8567 	    ("pmap_remove_write: page %p is not managed", m));
8568 
8569 	vm_page_assert_busied(m);
8570 	if (!pmap_page_is_write_mapped(m))
8571 		return;
8572 
8573 	lock = VM_PAGE_TO_PV_LIST_LOCK(m);
8574 	pvh = (m->flags & PG_FICTITIOUS) != 0 ? &pv_dummy :
8575 	    pa_to_pvh(VM_PAGE_TO_PHYS(m));
8576 	rw_wlock(lock);
8577 retry:
8578 	TAILQ_FOREACH_SAFE(pv, &pvh->pv_list, pv_next, next_pv) {
8579 		pmap = PV_PMAP(pv);
8580 		if (!PMAP_TRYLOCK(pmap)) {
8581 			pvh_gen = pvh->pv_gen;
8582 			rw_wunlock(lock);
8583 			PMAP_LOCK(pmap);
8584 			rw_wlock(lock);
8585 			if (pvh_gen != pvh->pv_gen) {
8586 				PMAP_UNLOCK(pmap);
8587 				goto retry;
8588 			}
8589 		}
8590 		PG_RW = pmap_rw_bit(pmap);
8591 		va = pv->pv_va;
8592 		pde = pmap_pde(pmap, va);
8593 		if ((*pde & PG_RW) != 0)
8594 			(void)pmap_demote_pde_locked(pmap, pde, va, &lock);
8595 		KASSERT(lock == VM_PAGE_TO_PV_LIST_LOCK(m),
8596 		    ("inconsistent pv lock %p %p for page %p",
8597 		    lock, VM_PAGE_TO_PV_LIST_LOCK(m), m));
8598 		PMAP_UNLOCK(pmap);
8599 	}
8600 	TAILQ_FOREACH(pv, &m->md.pv_list, pv_next) {
8601 		pmap = PV_PMAP(pv);
8602 		if (!PMAP_TRYLOCK(pmap)) {
8603 			pvh_gen = pvh->pv_gen;
8604 			md_gen = m->md.pv_gen;
8605 			rw_wunlock(lock);
8606 			PMAP_LOCK(pmap);
8607 			rw_wlock(lock);
8608 			if (pvh_gen != pvh->pv_gen ||
8609 			    md_gen != m->md.pv_gen) {
8610 				PMAP_UNLOCK(pmap);
8611 				goto retry;
8612 			}
8613 		}
8614 		PG_M = pmap_modified_bit(pmap);
8615 		PG_RW = pmap_rw_bit(pmap);
8616 		pde = pmap_pde(pmap, pv->pv_va);
8617 		KASSERT((*pde & PG_PS) == 0,
8618 		    ("pmap_remove_write: found a 2mpage in page %p's pv list",
8619 		    m));
8620 		pte = pmap_pde_to_pte(pde, pv->pv_va);
8621 		oldpte = *pte;
8622 		if (oldpte & PG_RW) {
8623 			while (!atomic_fcmpset_long(pte, &oldpte, oldpte &
8624 			    ~(PG_RW | PG_M)))
8625 				cpu_spinwait();
8626 			if ((oldpte & PG_M) != 0)
8627 				vm_page_dirty(m);
8628 			pmap_invalidate_page(pmap, pv->pv_va);
8629 		}
8630 		PMAP_UNLOCK(pmap);
8631 	}
8632 	rw_wunlock(lock);
8633 	vm_page_aflag_clear(m, PGA_WRITEABLE);
8634 	pmap_delayed_invl_wait(m);
8635 }
8636 
8637 static __inline boolean_t
safe_to_clear_referenced(pmap_t pmap,pt_entry_t pte)8638 safe_to_clear_referenced(pmap_t pmap, pt_entry_t pte)
8639 {
8640 
8641 	if (!pmap_emulate_ad_bits(pmap))
8642 		return (TRUE);
8643 
8644 	KASSERT(pmap->pm_type == PT_EPT, ("invalid pm_type %d", pmap->pm_type));
8645 
8646 	/*
8647 	 * XWR = 010 or 110 will cause an unconditional EPT misconfiguration
8648 	 * so we don't let the referenced (aka EPT_PG_READ) bit to be cleared
8649 	 * if the EPT_PG_WRITE bit is set.
8650 	 */
8651 	if ((pte & EPT_PG_WRITE) != 0)
8652 		return (FALSE);
8653 
8654 	/*
8655 	 * XWR = 100 is allowed only if the PMAP_SUPPORTS_EXEC_ONLY is set.
8656 	 */
8657 	if ((pte & EPT_PG_EXECUTE) == 0 ||
8658 	    ((pmap->pm_flags & PMAP_SUPPORTS_EXEC_ONLY) != 0))
8659 		return (TRUE);
8660 	else
8661 		return (FALSE);
8662 }
8663 
8664 /*
8665  *	pmap_ts_referenced:
8666  *
8667  *	Return a count of reference bits for a page, clearing those bits.
8668  *	It is not necessary for every reference bit to be cleared, but it
8669  *	is necessary that 0 only be returned when there are truly no
8670  *	reference bits set.
8671  *
8672  *	As an optimization, update the page's dirty field if a modified bit is
8673  *	found while counting reference bits.  This opportunistic update can be
8674  *	performed at low cost and can eliminate the need for some future calls
8675  *	to pmap_is_modified().  However, since this function stops after
8676  *	finding PMAP_TS_REFERENCED_MAX reference bits, it may not detect some
8677  *	dirty pages.  Those dirty pages will only be detected by a future call
8678  *	to pmap_is_modified().
8679  *
8680  *	A DI block is not needed within this function, because
8681  *	invalidations are performed before the PV list lock is
8682  *	released.
8683  */
8684 int
pmap_ts_referenced(vm_page_t m)8685 pmap_ts_referenced(vm_page_t m)
8686 {
8687 	struct md_page *pvh;
8688 	pv_entry_t pv, pvf;
8689 	pmap_t pmap;
8690 	struct rwlock *lock;
8691 	pd_entry_t oldpde, *pde;
8692 	pt_entry_t *pte, PG_A, PG_M, PG_RW;
8693 	vm_offset_t va;
8694 	vm_paddr_t pa;
8695 	int cleared, md_gen, not_cleared, pvh_gen;
8696 	struct spglist free;
8697 	boolean_t demoted;
8698 
8699 	KASSERT((m->oflags & VPO_UNMANAGED) == 0,
8700 	    ("pmap_ts_referenced: page %p is not managed", m));
8701 	SLIST_INIT(&free);
8702 	cleared = 0;
8703 	pa = VM_PAGE_TO_PHYS(m);
8704 	lock = PHYS_TO_PV_LIST_LOCK(pa);
8705 	pvh = (m->flags & PG_FICTITIOUS) != 0 ? &pv_dummy : pa_to_pvh(pa);
8706 	rw_wlock(lock);
8707 retry:
8708 	not_cleared = 0;
8709 	if ((pvf = TAILQ_FIRST(&pvh->pv_list)) == NULL)
8710 		goto small_mappings;
8711 	pv = pvf;
8712 	do {
8713 		if (pvf == NULL)
8714 			pvf = pv;
8715 		pmap = PV_PMAP(pv);
8716 		if (!PMAP_TRYLOCK(pmap)) {
8717 			pvh_gen = pvh->pv_gen;
8718 			rw_wunlock(lock);
8719 			PMAP_LOCK(pmap);
8720 			rw_wlock(lock);
8721 			if (pvh_gen != pvh->pv_gen) {
8722 				PMAP_UNLOCK(pmap);
8723 				goto retry;
8724 			}
8725 		}
8726 		PG_A = pmap_accessed_bit(pmap);
8727 		PG_M = pmap_modified_bit(pmap);
8728 		PG_RW = pmap_rw_bit(pmap);
8729 		va = pv->pv_va;
8730 		pde = pmap_pde(pmap, pv->pv_va);
8731 		oldpde = *pde;
8732 		if ((oldpde & (PG_M | PG_RW)) == (PG_M | PG_RW)) {
8733 			/*
8734 			 * Although "oldpde" is mapping a 2MB page, because
8735 			 * this function is called at a 4KB page granularity,
8736 			 * we only update the 4KB page under test.
8737 			 */
8738 			vm_page_dirty(m);
8739 		}
8740 		if ((oldpde & PG_A) != 0) {
8741 			/*
8742 			 * Since this reference bit is shared by 512 4KB
8743 			 * pages, it should not be cleared every time it is
8744 			 * tested.  Apply a simple "hash" function on the
8745 			 * physical page number, the virtual superpage number,
8746 			 * and the pmap address to select one 4KB page out of
8747 			 * the 512 on which testing the reference bit will
8748 			 * result in clearing that reference bit.  This
8749 			 * function is designed to avoid the selection of the
8750 			 * same 4KB page for every 2MB page mapping.
8751 			 *
8752 			 * On demotion, a mapping that hasn't been referenced
8753 			 * is simply destroyed.  To avoid the possibility of a
8754 			 * subsequent page fault on a demoted wired mapping,
8755 			 * always leave its reference bit set.  Moreover,
8756 			 * since the superpage is wired, the current state of
8757 			 * its reference bit won't affect page replacement.
8758 			 */
8759 			if ((((pa >> PAGE_SHIFT) ^ (pv->pv_va >> PDRSHIFT) ^
8760 			    (uintptr_t)pmap) & (NPTEPG - 1)) == 0 &&
8761 			    (oldpde & PG_W) == 0) {
8762 				if (safe_to_clear_referenced(pmap, oldpde)) {
8763 					atomic_clear_long(pde, PG_A);
8764 					pmap_invalidate_page(pmap, pv->pv_va);
8765 					demoted = FALSE;
8766 				} else if (pmap_demote_pde_locked(pmap, pde,
8767 				    pv->pv_va, &lock)) {
8768 					/*
8769 					 * Remove the mapping to a single page
8770 					 * so that a subsequent access may
8771 					 * repromote.  Since the underlying
8772 					 * page table page is fully populated,
8773 					 * this removal never frees a page
8774 					 * table page.
8775 					 */
8776 					demoted = TRUE;
8777 					va += VM_PAGE_TO_PHYS(m) - (oldpde &
8778 					    PG_PS_FRAME);
8779 					pte = pmap_pde_to_pte(pde, va);
8780 					pmap_remove_pte(pmap, pte, va, *pde,
8781 					    NULL, &lock);
8782 					pmap_invalidate_page(pmap, va);
8783 				} else
8784 					demoted = TRUE;
8785 
8786 				if (demoted) {
8787 					/*
8788 					 * The superpage mapping was removed
8789 					 * entirely and therefore 'pv' is no
8790 					 * longer valid.
8791 					 */
8792 					if (pvf == pv)
8793 						pvf = NULL;
8794 					pv = NULL;
8795 				}
8796 				cleared++;
8797 				KASSERT(lock == VM_PAGE_TO_PV_LIST_LOCK(m),
8798 				    ("inconsistent pv lock %p %p for page %p",
8799 				    lock, VM_PAGE_TO_PV_LIST_LOCK(m), m));
8800 			} else
8801 				not_cleared++;
8802 		}
8803 		PMAP_UNLOCK(pmap);
8804 		/* Rotate the PV list if it has more than one entry. */
8805 		if (pv != NULL && TAILQ_NEXT(pv, pv_next) != NULL) {
8806 			TAILQ_REMOVE(&pvh->pv_list, pv, pv_next);
8807 			TAILQ_INSERT_TAIL(&pvh->pv_list, pv, pv_next);
8808 			pvh->pv_gen++;
8809 		}
8810 		if (cleared + not_cleared >= PMAP_TS_REFERENCED_MAX)
8811 			goto out;
8812 	} while ((pv = TAILQ_FIRST(&pvh->pv_list)) != pvf);
8813 small_mappings:
8814 	if ((pvf = TAILQ_FIRST(&m->md.pv_list)) == NULL)
8815 		goto out;
8816 	pv = pvf;
8817 	do {
8818 		if (pvf == NULL)
8819 			pvf = pv;
8820 		pmap = PV_PMAP(pv);
8821 		if (!PMAP_TRYLOCK(pmap)) {
8822 			pvh_gen = pvh->pv_gen;
8823 			md_gen = m->md.pv_gen;
8824 			rw_wunlock(lock);
8825 			PMAP_LOCK(pmap);
8826 			rw_wlock(lock);
8827 			if (pvh_gen != pvh->pv_gen || md_gen != m->md.pv_gen) {
8828 				PMAP_UNLOCK(pmap);
8829 				goto retry;
8830 			}
8831 		}
8832 		PG_A = pmap_accessed_bit(pmap);
8833 		PG_M = pmap_modified_bit(pmap);
8834 		PG_RW = pmap_rw_bit(pmap);
8835 		pde = pmap_pde(pmap, pv->pv_va);
8836 		KASSERT((*pde & PG_PS) == 0,
8837 		    ("pmap_ts_referenced: found a 2mpage in page %p's pv list",
8838 		    m));
8839 		pte = pmap_pde_to_pte(pde, pv->pv_va);
8840 		if ((*pte & (PG_M | PG_RW)) == (PG_M | PG_RW))
8841 			vm_page_dirty(m);
8842 		if ((*pte & PG_A) != 0) {
8843 			if (safe_to_clear_referenced(pmap, *pte)) {
8844 				atomic_clear_long(pte, PG_A);
8845 				pmap_invalidate_page(pmap, pv->pv_va);
8846 				cleared++;
8847 			} else if ((*pte & PG_W) == 0) {
8848 				/*
8849 				 * Wired pages cannot be paged out so
8850 				 * doing accessed bit emulation for
8851 				 * them is wasted effort. We do the
8852 				 * hard work for unwired pages only.
8853 				 */
8854 				pmap_remove_pte(pmap, pte, pv->pv_va,
8855 				    *pde, &free, &lock);
8856 				pmap_invalidate_page(pmap, pv->pv_va);
8857 				cleared++;
8858 				if (pvf == pv)
8859 					pvf = NULL;
8860 				pv = NULL;
8861 				KASSERT(lock == VM_PAGE_TO_PV_LIST_LOCK(m),
8862 				    ("inconsistent pv lock %p %p for page %p",
8863 				    lock, VM_PAGE_TO_PV_LIST_LOCK(m), m));
8864 			} else
8865 				not_cleared++;
8866 		}
8867 		PMAP_UNLOCK(pmap);
8868 		/* Rotate the PV list if it has more than one entry. */
8869 		if (pv != NULL && TAILQ_NEXT(pv, pv_next) != NULL) {
8870 			TAILQ_REMOVE(&m->md.pv_list, pv, pv_next);
8871 			TAILQ_INSERT_TAIL(&m->md.pv_list, pv, pv_next);
8872 			m->md.pv_gen++;
8873 		}
8874 	} while ((pv = TAILQ_FIRST(&m->md.pv_list)) != pvf && cleared +
8875 	    not_cleared < PMAP_TS_REFERENCED_MAX);
8876 out:
8877 	rw_wunlock(lock);
8878 	vm_page_free_pages_toq(&free, true);
8879 	return (cleared + not_cleared);
8880 }
8881 
8882 /*
8883  *	Apply the given advice to the specified range of addresses within the
8884  *	given pmap.  Depending on the advice, clear the referenced and/or
8885  *	modified flags in each mapping and set the mapped page's dirty field.
8886  */
8887 void
pmap_advise(pmap_t pmap,vm_offset_t sva,vm_offset_t eva,int advice)8888 pmap_advise(pmap_t pmap, vm_offset_t sva, vm_offset_t eva, int advice)
8889 {
8890 	struct rwlock *lock;
8891 	pml4_entry_t *pml4e;
8892 	pdp_entry_t *pdpe;
8893 	pd_entry_t oldpde, *pde;
8894 	pt_entry_t *pte, PG_A, PG_G, PG_M, PG_RW, PG_V;
8895 	vm_offset_t va, va_next;
8896 	vm_page_t m;
8897 	bool anychanged;
8898 
8899 	if (advice != MADV_DONTNEED && advice != MADV_FREE)
8900 		return;
8901 
8902 	/*
8903 	 * A/D bit emulation requires an alternate code path when clearing
8904 	 * the modified and accessed bits below. Since this function is
8905 	 * advisory in nature we skip it entirely for pmaps that require
8906 	 * A/D bit emulation.
8907 	 */
8908 	if (pmap_emulate_ad_bits(pmap))
8909 		return;
8910 
8911 	PG_A = pmap_accessed_bit(pmap);
8912 	PG_G = pmap_global_bit(pmap);
8913 	PG_M = pmap_modified_bit(pmap);
8914 	PG_V = pmap_valid_bit(pmap);
8915 	PG_RW = pmap_rw_bit(pmap);
8916 	anychanged = false;
8917 	pmap_delayed_invl_start();
8918 	PMAP_LOCK(pmap);
8919 	for (; sva < eva; sva = va_next) {
8920 		pml4e = pmap_pml4e(pmap, sva);
8921 		if (pml4e == NULL || (*pml4e & PG_V) == 0) {
8922 			va_next = (sva + NBPML4) & ~PML4MASK;
8923 			if (va_next < sva)
8924 				va_next = eva;
8925 			continue;
8926 		}
8927 
8928 		va_next = (sva + NBPDP) & ~PDPMASK;
8929 		if (va_next < sva)
8930 			va_next = eva;
8931 		pdpe = pmap_pml4e_to_pdpe(pml4e, sva);
8932 		if ((*pdpe & PG_V) == 0)
8933 			continue;
8934 		if ((*pdpe & PG_PS) != 0)
8935 			continue;
8936 
8937 		va_next = (sva + NBPDR) & ~PDRMASK;
8938 		if (va_next < sva)
8939 			va_next = eva;
8940 		pde = pmap_pdpe_to_pde(pdpe, sva);
8941 		oldpde = *pde;
8942 		if ((oldpde & PG_V) == 0)
8943 			continue;
8944 		else if ((oldpde & PG_PS) != 0) {
8945 			if ((oldpde & PG_MANAGED) == 0)
8946 				continue;
8947 			lock = NULL;
8948 			if (!pmap_demote_pde_locked(pmap, pde, sva, &lock)) {
8949 				if (lock != NULL)
8950 					rw_wunlock(lock);
8951 
8952 				/*
8953 				 * The large page mapping was destroyed.
8954 				 */
8955 				continue;
8956 			}
8957 
8958 			/*
8959 			 * Unless the page mappings are wired, remove the
8960 			 * mapping to a single page so that a subsequent
8961 			 * access may repromote.  Choosing the last page
8962 			 * within the address range [sva, min(va_next, eva))
8963 			 * generally results in more repromotions.  Since the
8964 			 * underlying page table page is fully populated, this
8965 			 * removal never frees a page table page.
8966 			 */
8967 			if ((oldpde & PG_W) == 0) {
8968 				va = eva;
8969 				if (va > va_next)
8970 					va = va_next;
8971 				va -= PAGE_SIZE;
8972 				KASSERT(va >= sva,
8973 				    ("pmap_advise: no address gap"));
8974 				pte = pmap_pde_to_pte(pde, va);
8975 				KASSERT((*pte & PG_V) != 0,
8976 				    ("pmap_advise: invalid PTE"));
8977 				pmap_remove_pte(pmap, pte, va, *pde, NULL,
8978 				    &lock);
8979 				anychanged = true;
8980 			}
8981 			if (lock != NULL)
8982 				rw_wunlock(lock);
8983 		}
8984 		if (va_next > eva)
8985 			va_next = eva;
8986 		va = va_next;
8987 		for (pte = pmap_pde_to_pte(pde, sva); sva != va_next; pte++,
8988 		    sva += PAGE_SIZE) {
8989 			if ((*pte & (PG_MANAGED | PG_V)) != (PG_MANAGED | PG_V))
8990 				goto maybe_invlrng;
8991 			else if ((*pte & (PG_M | PG_RW)) == (PG_M | PG_RW)) {
8992 				if (advice == MADV_DONTNEED) {
8993 					/*
8994 					 * Future calls to pmap_is_modified()
8995 					 * can be avoided by making the page
8996 					 * dirty now.
8997 					 */
8998 					m = PHYS_TO_VM_PAGE(*pte & PG_FRAME);
8999 					vm_page_dirty(m);
9000 				}
9001 				atomic_clear_long(pte, PG_M | PG_A);
9002 			} else if ((*pte & PG_A) != 0)
9003 				atomic_clear_long(pte, PG_A);
9004 			else
9005 				goto maybe_invlrng;
9006 
9007 			if ((*pte & PG_G) != 0) {
9008 				if (va == va_next)
9009 					va = sva;
9010 			} else
9011 				anychanged = true;
9012 			continue;
9013 maybe_invlrng:
9014 			if (va != va_next) {
9015 				pmap_invalidate_range(pmap, va, sva);
9016 				va = va_next;
9017 			}
9018 		}
9019 		if (va != va_next)
9020 			pmap_invalidate_range(pmap, va, sva);
9021 	}
9022 	if (anychanged)
9023 		pmap_invalidate_all(pmap);
9024 	PMAP_UNLOCK(pmap);
9025 	pmap_delayed_invl_finish();
9026 }
9027 
9028 /*
9029  *	Clear the modify bits on the specified physical page.
9030  */
9031 void
pmap_clear_modify(vm_page_t m)9032 pmap_clear_modify(vm_page_t m)
9033 {
9034 	struct md_page *pvh;
9035 	pmap_t pmap;
9036 	pv_entry_t next_pv, pv;
9037 	pd_entry_t oldpde, *pde;
9038 	pt_entry_t *pte, PG_M, PG_RW;
9039 	struct rwlock *lock;
9040 	vm_offset_t va;
9041 	int md_gen, pvh_gen;
9042 
9043 	KASSERT((m->oflags & VPO_UNMANAGED) == 0,
9044 	    ("pmap_clear_modify: page %p is not managed", m));
9045 	vm_page_assert_busied(m);
9046 
9047 	if (!pmap_page_is_write_mapped(m))
9048 		return;
9049 	pvh = (m->flags & PG_FICTITIOUS) != 0 ? &pv_dummy :
9050 	    pa_to_pvh(VM_PAGE_TO_PHYS(m));
9051 	lock = VM_PAGE_TO_PV_LIST_LOCK(m);
9052 	rw_wlock(lock);
9053 restart:
9054 	TAILQ_FOREACH_SAFE(pv, &pvh->pv_list, pv_next, next_pv) {
9055 		pmap = PV_PMAP(pv);
9056 		if (!PMAP_TRYLOCK(pmap)) {
9057 			pvh_gen = pvh->pv_gen;
9058 			rw_wunlock(lock);
9059 			PMAP_LOCK(pmap);
9060 			rw_wlock(lock);
9061 			if (pvh_gen != pvh->pv_gen) {
9062 				PMAP_UNLOCK(pmap);
9063 				goto restart;
9064 			}
9065 		}
9066 		PG_M = pmap_modified_bit(pmap);
9067 		PG_RW = pmap_rw_bit(pmap);
9068 		va = pv->pv_va;
9069 		pde = pmap_pde(pmap, va);
9070 		oldpde = *pde;
9071 		/* If oldpde has PG_RW set, then it also has PG_M set. */
9072 		if ((oldpde & PG_RW) != 0 &&
9073 		    pmap_demote_pde_locked(pmap, pde, va, &lock) &&
9074 		    (oldpde & PG_W) == 0) {
9075 			/*
9076 			 * Write protect the mapping to a single page so that
9077 			 * a subsequent write access may repromote.
9078 			 */
9079 			va += VM_PAGE_TO_PHYS(m) - (oldpde & PG_PS_FRAME);
9080 			pte = pmap_pde_to_pte(pde, va);
9081 			atomic_clear_long(pte, PG_M | PG_RW);
9082 			vm_page_dirty(m);
9083 			pmap_invalidate_page(pmap, va);
9084 		}
9085 		PMAP_UNLOCK(pmap);
9086 	}
9087 	TAILQ_FOREACH(pv, &m->md.pv_list, pv_next) {
9088 		pmap = PV_PMAP(pv);
9089 		if (!PMAP_TRYLOCK(pmap)) {
9090 			md_gen = m->md.pv_gen;
9091 			pvh_gen = pvh->pv_gen;
9092 			rw_wunlock(lock);
9093 			PMAP_LOCK(pmap);
9094 			rw_wlock(lock);
9095 			if (pvh_gen != pvh->pv_gen || md_gen != m->md.pv_gen) {
9096 				PMAP_UNLOCK(pmap);
9097 				goto restart;
9098 			}
9099 		}
9100 		PG_M = pmap_modified_bit(pmap);
9101 		PG_RW = pmap_rw_bit(pmap);
9102 		pde = pmap_pde(pmap, pv->pv_va);
9103 		KASSERT((*pde & PG_PS) == 0, ("pmap_clear_modify: found"
9104 		    " a 2mpage in page %p's pv list", m));
9105 		pte = pmap_pde_to_pte(pde, pv->pv_va);
9106 		if ((*pte & (PG_M | PG_RW)) == (PG_M | PG_RW)) {
9107 			atomic_clear_long(pte, PG_M);
9108 			pmap_invalidate_page(pmap, pv->pv_va);
9109 		}
9110 		PMAP_UNLOCK(pmap);
9111 	}
9112 	rw_wunlock(lock);
9113 }
9114 
9115 /*
9116  * Miscellaneous support routines follow
9117  */
9118 
9119 /* Adjust the properties for a leaf page table entry. */
9120 static __inline void
pmap_pte_props(pt_entry_t * pte,u_long bits,u_long mask)9121 pmap_pte_props(pt_entry_t *pte, u_long bits, u_long mask)
9122 {
9123 	u_long opte, npte;
9124 
9125 	opte = *(u_long *)pte;
9126 	do {
9127 		npte = opte & ~mask;
9128 		npte |= bits;
9129 	} while (npte != opte && !atomic_fcmpset_long((u_long *)pte, &opte,
9130 	    npte));
9131 }
9132 
9133 /*
9134  * Map a set of physical memory pages into the kernel virtual
9135  * address space. Return a pointer to where it is mapped. This
9136  * routine is intended to be used for mapping device memory,
9137  * NOT real memory.
9138  */
9139 static void *
pmap_mapdev_internal(vm_paddr_t pa,vm_size_t size,int mode,int flags)9140 pmap_mapdev_internal(vm_paddr_t pa, vm_size_t size, int mode, int flags)
9141 {
9142 	struct pmap_preinit_mapping *ppim;
9143 	vm_offset_t va, offset;
9144 	vm_size_t tmpsize;
9145 	int i;
9146 
9147 	offset = pa & PAGE_MASK;
9148 	size = round_page(offset + size);
9149 	pa = trunc_page(pa);
9150 
9151 	if (!pmap_initialized) {
9152 		va = 0;
9153 		for (i = 0; i < PMAP_PREINIT_MAPPING_COUNT; i++) {
9154 			ppim = pmap_preinit_mapping + i;
9155 			if (ppim->va == 0) {
9156 				ppim->pa = pa;
9157 				ppim->sz = size;
9158 				ppim->mode = mode;
9159 				ppim->va = virtual_avail;
9160 				virtual_avail += size;
9161 				va = ppim->va;
9162 				break;
9163 			}
9164 		}
9165 		if (va == 0)
9166 			panic("%s: too many preinit mappings", __func__);
9167 	} else {
9168 		/*
9169 		 * If we have a preinit mapping, re-use it.
9170 		 */
9171 		for (i = 0; i < PMAP_PREINIT_MAPPING_COUNT; i++) {
9172 			ppim = pmap_preinit_mapping + i;
9173 			if (ppim->pa == pa && ppim->sz == size &&
9174 			    (ppim->mode == mode ||
9175 			    (flags & MAPDEV_SETATTR) == 0))
9176 				return ((void *)(ppim->va + offset));
9177 		}
9178 		/*
9179 		 * If the specified range of physical addresses fits within
9180 		 * the direct map window, use the direct map.
9181 		 */
9182 		if (pa < dmaplimit && pa + size <= dmaplimit) {
9183 			va = PHYS_TO_DMAP(pa);
9184 			if ((flags & MAPDEV_SETATTR) != 0) {
9185 				PMAP_LOCK(kernel_pmap);
9186 				i = pmap_change_props_locked(va, size,
9187 				    PROT_NONE, mode, flags);
9188 				PMAP_UNLOCK(kernel_pmap);
9189 			} else
9190 				i = 0;
9191 			if (!i)
9192 				return ((void *)(va + offset));
9193 		}
9194 		va = kva_alloc(size);
9195 		if (va == 0)
9196 			panic("%s: Couldn't allocate KVA", __func__);
9197 	}
9198 	for (tmpsize = 0; tmpsize < size; tmpsize += PAGE_SIZE)
9199 		pmap_kenter_attr(va + tmpsize, pa + tmpsize, mode);
9200 	pmap_invalidate_range(kernel_pmap, va, va + tmpsize);
9201 	if ((flags & MAPDEV_FLUSHCACHE) != 0)
9202 		pmap_invalidate_cache_range(va, va + tmpsize);
9203 	return ((void *)(va + offset));
9204 }
9205 
9206 void *
pmap_mapdev_attr(vm_paddr_t pa,vm_size_t size,int mode)9207 pmap_mapdev_attr(vm_paddr_t pa, vm_size_t size, int mode)
9208 {
9209 
9210 	return (pmap_mapdev_internal(pa, size, mode, MAPDEV_FLUSHCACHE |
9211 	    MAPDEV_SETATTR));
9212 }
9213 
9214 void *
pmap_mapdev(vm_paddr_t pa,vm_size_t size)9215 pmap_mapdev(vm_paddr_t pa, vm_size_t size)
9216 {
9217 
9218 	return (pmap_mapdev_attr(pa, size, PAT_UNCACHEABLE));
9219 }
9220 
9221 void *
pmap_mapdev_pciecfg(vm_paddr_t pa,vm_size_t size)9222 pmap_mapdev_pciecfg(vm_paddr_t pa, vm_size_t size)
9223 {
9224 
9225 	return (pmap_mapdev_internal(pa, size, PAT_UNCACHEABLE,
9226 	    MAPDEV_SETATTR));
9227 }
9228 
9229 void *
pmap_mapbios(vm_paddr_t pa,vm_size_t size)9230 pmap_mapbios(vm_paddr_t pa, vm_size_t size)
9231 {
9232 
9233 	return (pmap_mapdev_internal(pa, size, PAT_WRITE_BACK,
9234 	    MAPDEV_FLUSHCACHE));
9235 }
9236 
9237 void
pmap_unmapdev(vm_offset_t va,vm_size_t size)9238 pmap_unmapdev(vm_offset_t va, vm_size_t size)
9239 {
9240 	struct pmap_preinit_mapping *ppim;
9241 	vm_offset_t offset;
9242 	int i;
9243 
9244 	/* If we gave a direct map region in pmap_mapdev, do nothing */
9245 	if (va >= DMAP_MIN_ADDRESS && va < DMAP_MAX_ADDRESS)
9246 		return;
9247 	offset = va & PAGE_MASK;
9248 	size = round_page(offset + size);
9249 	va = trunc_page(va);
9250 	for (i = 0; i < PMAP_PREINIT_MAPPING_COUNT; i++) {
9251 		ppim = pmap_preinit_mapping + i;
9252 		if (ppim->va == va && ppim->sz == size) {
9253 			if (pmap_initialized)
9254 				return;
9255 			ppim->pa = 0;
9256 			ppim->va = 0;
9257 			ppim->sz = 0;
9258 			ppim->mode = 0;
9259 			if (va + size == virtual_avail)
9260 				virtual_avail = va;
9261 			return;
9262 		}
9263 	}
9264 	if (pmap_initialized) {
9265 		pmap_qremove(va, atop(size));
9266 		kva_free(va, size);
9267 	}
9268 }
9269 
9270 /*
9271  * Tries to demote a 1GB page mapping.
9272  */
9273 static boolean_t
pmap_demote_pdpe(pmap_t pmap,pdp_entry_t * pdpe,vm_offset_t va)9274 pmap_demote_pdpe(pmap_t pmap, pdp_entry_t *pdpe, vm_offset_t va)
9275 {
9276 	pdp_entry_t newpdpe, oldpdpe;
9277 	pd_entry_t *firstpde, newpde, *pde;
9278 	pt_entry_t PG_A, PG_M, PG_RW, PG_V;
9279 	vm_paddr_t pdpgpa;
9280 	vm_page_t pdpg;
9281 
9282 	PG_A = pmap_accessed_bit(pmap);
9283 	PG_M = pmap_modified_bit(pmap);
9284 	PG_V = pmap_valid_bit(pmap);
9285 	PG_RW = pmap_rw_bit(pmap);
9286 
9287 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
9288 	oldpdpe = *pdpe;
9289 	KASSERT((oldpdpe & (PG_PS | PG_V)) == (PG_PS | PG_V),
9290 	    ("pmap_demote_pdpe: oldpdpe is missing PG_PS and/or PG_V"));
9291 	pdpg = pmap_alloc_pt_page(pmap, va >> PDPSHIFT,
9292 	    VM_ALLOC_WIRED | VM_ALLOC_INTERRUPT);
9293 	if (pdpg  == NULL) {
9294 		CTR2(KTR_PMAP, "pmap_demote_pdpe: failure for va %#lx"
9295 		    " in pmap %p", va, pmap);
9296 		return (FALSE);
9297 	}
9298 	pdpgpa = VM_PAGE_TO_PHYS(pdpg);
9299 	firstpde = (pd_entry_t *)PHYS_TO_DMAP(pdpgpa);
9300 	newpdpe = pdpgpa | PG_M | PG_A | (oldpdpe & PG_U) | PG_RW | PG_V;
9301 	KASSERT((oldpdpe & PG_A) != 0,
9302 	    ("pmap_demote_pdpe: oldpdpe is missing PG_A"));
9303 	KASSERT((oldpdpe & (PG_M | PG_RW)) != PG_RW,
9304 	    ("pmap_demote_pdpe: oldpdpe is missing PG_M"));
9305 	newpde = oldpdpe;
9306 
9307 	/*
9308 	 * Initialize the page directory page.
9309 	 */
9310 	for (pde = firstpde; pde < firstpde + NPDEPG; pde++) {
9311 		*pde = newpde;
9312 		newpde += NBPDR;
9313 	}
9314 
9315 	/*
9316 	 * Demote the mapping.
9317 	 */
9318 	*pdpe = newpdpe;
9319 
9320 	/*
9321 	 * Invalidate a stale recursive mapping of the page directory page.
9322 	 */
9323 	pmap_invalidate_page(pmap, (vm_offset_t)vtopde(va));
9324 
9325 	counter_u64_add(pmap_pdpe_demotions, 1);
9326 	CTR2(KTR_PMAP, "pmap_demote_pdpe: success for va %#lx"
9327 	    " in pmap %p", va, pmap);
9328 	return (TRUE);
9329 }
9330 
9331 /*
9332  * Sets the memory attribute for the specified page.
9333  */
9334 void
pmap_page_set_memattr(vm_page_t m,vm_memattr_t ma)9335 pmap_page_set_memattr(vm_page_t m, vm_memattr_t ma)
9336 {
9337 
9338 	m->md.pat_mode = ma;
9339 
9340 	/*
9341 	 * If "m" is a normal page, update its direct mapping.  This update
9342 	 * can be relied upon to perform any cache operations that are
9343 	 * required for data coherence.
9344 	 */
9345 	if ((m->flags & PG_FICTITIOUS) == 0 &&
9346 	    pmap_change_attr(PHYS_TO_DMAP(VM_PAGE_TO_PHYS(m)), PAGE_SIZE,
9347 	    m->md.pat_mode))
9348 		panic("memory attribute change on the direct map failed");
9349 }
9350 
9351 void
pmap_page_set_memattr_noflush(vm_page_t m,vm_memattr_t ma)9352 pmap_page_set_memattr_noflush(vm_page_t m, vm_memattr_t ma)
9353 {
9354 	int error;
9355 
9356 	m->md.pat_mode = ma;
9357 
9358 	if ((m->flags & PG_FICTITIOUS) != 0)
9359 		return;
9360 	PMAP_LOCK(kernel_pmap);
9361 	error = pmap_change_props_locked(PHYS_TO_DMAP(VM_PAGE_TO_PHYS(m)),
9362 	    PAGE_SIZE, PROT_NONE, m->md.pat_mode, 0);
9363 	PMAP_UNLOCK(kernel_pmap);
9364 	if (error != 0)
9365 		panic("memory attribute change on the direct map failed");
9366 }
9367 
9368 /*
9369  * Changes the specified virtual address range's memory type to that given by
9370  * the parameter "mode".  The specified virtual address range must be
9371  * completely contained within either the direct map or the kernel map.  If
9372  * the virtual address range is contained within the kernel map, then the
9373  * memory type for each of the corresponding ranges of the direct map is also
9374  * changed.  (The corresponding ranges of the direct map are those ranges that
9375  * map the same physical pages as the specified virtual address range.)  These
9376  * changes to the direct map are necessary because Intel describes the
9377  * behavior of their processors as "undefined" if two or more mappings to the
9378  * same physical page have different memory types.
9379  *
9380  * Returns zero if the change completed successfully, and either EINVAL or
9381  * ENOMEM if the change failed.  Specifically, EINVAL is returned if some part
9382  * of the virtual address range was not mapped, and ENOMEM is returned if
9383  * there was insufficient memory available to complete the change.  In the
9384  * latter case, the memory type may have been changed on some part of the
9385  * virtual address range or the direct map.
9386  */
9387 int
pmap_change_attr(vm_offset_t va,vm_size_t size,int mode)9388 pmap_change_attr(vm_offset_t va, vm_size_t size, int mode)
9389 {
9390 	int error;
9391 
9392 	PMAP_LOCK(kernel_pmap);
9393 	error = pmap_change_props_locked(va, size, PROT_NONE, mode,
9394 	    MAPDEV_FLUSHCACHE);
9395 	PMAP_UNLOCK(kernel_pmap);
9396 	return (error);
9397 }
9398 
9399 /*
9400  * Changes the specified virtual address range's protections to those
9401  * specified by "prot".  Like pmap_change_attr(), protections for aliases
9402  * in the direct map are updated as well.  Protections on aliasing mappings may
9403  * be a subset of the requested protections; for example, mappings in the direct
9404  * map are never executable.
9405  */
9406 int
pmap_change_prot(vm_offset_t va,vm_size_t size,vm_prot_t prot)9407 pmap_change_prot(vm_offset_t va, vm_size_t size, vm_prot_t prot)
9408 {
9409 	int error;
9410 
9411 	/* Only supported within the kernel map. */
9412 	if (va < VM_MIN_KERNEL_ADDRESS)
9413 		return (EINVAL);
9414 
9415 	PMAP_LOCK(kernel_pmap);
9416 	error = pmap_change_props_locked(va, size, prot, -1,
9417 	    MAPDEV_ASSERTVALID);
9418 	PMAP_UNLOCK(kernel_pmap);
9419 	return (error);
9420 }
9421 
9422 static int
pmap_change_props_locked(vm_offset_t va,vm_size_t size,vm_prot_t prot,int mode,int flags)9423 pmap_change_props_locked(vm_offset_t va, vm_size_t size, vm_prot_t prot,
9424     int mode, int flags)
9425 {
9426 	vm_offset_t base, offset, tmpva;
9427 	vm_paddr_t pa_start, pa_end, pa_end1;
9428 	pdp_entry_t *pdpe;
9429 	pd_entry_t *pde, pde_bits, pde_mask;
9430 	pt_entry_t *pte, pte_bits, pte_mask;
9431 	int error;
9432 	bool changed;
9433 
9434 	PMAP_LOCK_ASSERT(kernel_pmap, MA_OWNED);
9435 	base = trunc_page(va);
9436 	offset = va & PAGE_MASK;
9437 	size = round_page(offset + size);
9438 
9439 	/*
9440 	 * Only supported on kernel virtual addresses, including the direct
9441 	 * map but excluding the recursive map.
9442 	 */
9443 	if (base < DMAP_MIN_ADDRESS)
9444 		return (EINVAL);
9445 
9446 	/*
9447 	 * Construct our flag sets and masks.  "bits" is the subset of
9448 	 * "mask" that will be set in each modified PTE.
9449 	 *
9450 	 * Mappings in the direct map are never allowed to be executable.
9451 	 */
9452 	pde_bits = pte_bits = 0;
9453 	pde_mask = pte_mask = 0;
9454 	if (mode != -1) {
9455 		pde_bits |= pmap_cache_bits(kernel_pmap, mode, true);
9456 		pde_mask |= X86_PG_PDE_CACHE;
9457 		pte_bits |= pmap_cache_bits(kernel_pmap, mode, false);
9458 		pte_mask |= X86_PG_PTE_CACHE;
9459 	}
9460 	if (prot != VM_PROT_NONE) {
9461 		if ((prot & VM_PROT_WRITE) != 0) {
9462 			pde_bits |= X86_PG_RW;
9463 			pte_bits |= X86_PG_RW;
9464 		}
9465 		if ((prot & VM_PROT_EXECUTE) == 0 ||
9466 		    va < VM_MIN_KERNEL_ADDRESS) {
9467 			pde_bits |= pg_nx;
9468 			pte_bits |= pg_nx;
9469 		}
9470 		pde_mask |= X86_PG_RW | pg_nx;
9471 		pte_mask |= X86_PG_RW | pg_nx;
9472 	}
9473 
9474 	/*
9475 	 * Pages that aren't mapped aren't supported.  Also break down 2MB pages
9476 	 * into 4KB pages if required.
9477 	 */
9478 	for (tmpva = base; tmpva < base + size; ) {
9479 		pdpe = pmap_pdpe(kernel_pmap, tmpva);
9480 		if (pdpe == NULL || *pdpe == 0) {
9481 			KASSERT((flags & MAPDEV_ASSERTVALID) == 0,
9482 			    ("%s: addr %#lx is not mapped", __func__, tmpva));
9483 			return (EINVAL);
9484 		}
9485 		if (*pdpe & PG_PS) {
9486 			/*
9487 			 * If the current 1GB page already has the required
9488 			 * properties, then we need not demote this page.  Just
9489 			 * increment tmpva to the next 1GB page frame.
9490 			 */
9491 			if ((*pdpe & pde_mask) == pde_bits) {
9492 				tmpva = trunc_1gpage(tmpva) + NBPDP;
9493 				continue;
9494 			}
9495 
9496 			/*
9497 			 * If the current offset aligns with a 1GB page frame
9498 			 * and there is at least 1GB left within the range, then
9499 			 * we need not break down this page into 2MB pages.
9500 			 */
9501 			if ((tmpva & PDPMASK) == 0 &&
9502 			    tmpva + PDPMASK < base + size) {
9503 				tmpva += NBPDP;
9504 				continue;
9505 			}
9506 			if (!pmap_demote_pdpe(kernel_pmap, pdpe, tmpva))
9507 				return (ENOMEM);
9508 		}
9509 		pde = pmap_pdpe_to_pde(pdpe, tmpva);
9510 		if (*pde == 0) {
9511 			KASSERT((flags & MAPDEV_ASSERTVALID) == 0,
9512 			    ("%s: addr %#lx is not mapped", __func__, tmpva));
9513 			return (EINVAL);
9514 		}
9515 		if (*pde & PG_PS) {
9516 			/*
9517 			 * If the current 2MB page already has the required
9518 			 * properties, then we need not demote this page.  Just
9519 			 * increment tmpva to the next 2MB page frame.
9520 			 */
9521 			if ((*pde & pde_mask) == pde_bits) {
9522 				tmpva = trunc_2mpage(tmpva) + NBPDR;
9523 				continue;
9524 			}
9525 
9526 			/*
9527 			 * If the current offset aligns with a 2MB page frame
9528 			 * and there is at least 2MB left within the range, then
9529 			 * we need not break down this page into 4KB pages.
9530 			 */
9531 			if ((tmpva & PDRMASK) == 0 &&
9532 			    tmpva + PDRMASK < base + size) {
9533 				tmpva += NBPDR;
9534 				continue;
9535 			}
9536 			if (!pmap_demote_pde(kernel_pmap, pde, tmpva))
9537 				return (ENOMEM);
9538 		}
9539 		pte = pmap_pde_to_pte(pde, tmpva);
9540 		if (*pte == 0) {
9541 			KASSERT((flags & MAPDEV_ASSERTVALID) == 0,
9542 			    ("%s: addr %#lx is not mapped", __func__, tmpva));
9543 			return (EINVAL);
9544 		}
9545 		tmpva += PAGE_SIZE;
9546 	}
9547 	error = 0;
9548 
9549 	/*
9550 	 * Ok, all the pages exist, so run through them updating their
9551 	 * properties if required.
9552 	 */
9553 	changed = false;
9554 	pa_start = pa_end = 0;
9555 	for (tmpva = base; tmpva < base + size; ) {
9556 		pdpe = pmap_pdpe(kernel_pmap, tmpva);
9557 		if (*pdpe & PG_PS) {
9558 			if ((*pdpe & pde_mask) != pde_bits) {
9559 				pmap_pte_props(pdpe, pde_bits, pde_mask);
9560 				changed = true;
9561 			}
9562 			if (tmpva >= VM_MIN_KERNEL_ADDRESS &&
9563 			    (*pdpe & PG_PS_FRAME) < dmaplimit) {
9564 				if (pa_start == pa_end) {
9565 					/* Start physical address run. */
9566 					pa_start = *pdpe & PG_PS_FRAME;
9567 					pa_end = pa_start + NBPDP;
9568 				} else if (pa_end == (*pdpe & PG_PS_FRAME))
9569 					pa_end += NBPDP;
9570 				else {
9571 					/* Run ended, update direct map. */
9572 					error = pmap_change_props_locked(
9573 					    PHYS_TO_DMAP(pa_start),
9574 					    pa_end - pa_start, prot, mode,
9575 					    flags);
9576 					if (error != 0)
9577 						break;
9578 					/* Start physical address run. */
9579 					pa_start = *pdpe & PG_PS_FRAME;
9580 					pa_end = pa_start + NBPDP;
9581 				}
9582 			}
9583 			tmpva = trunc_1gpage(tmpva) + NBPDP;
9584 			continue;
9585 		}
9586 		pde = pmap_pdpe_to_pde(pdpe, tmpva);
9587 		if (*pde & PG_PS) {
9588 			if ((*pde & pde_mask) != pde_bits) {
9589 				pmap_pte_props(pde, pde_bits, pde_mask);
9590 				changed = true;
9591 			}
9592 			if (tmpva >= VM_MIN_KERNEL_ADDRESS &&
9593 			    (*pde & PG_PS_FRAME) < dmaplimit) {
9594 				if (pa_start == pa_end) {
9595 					/* Start physical address run. */
9596 					pa_start = *pde & PG_PS_FRAME;
9597 					pa_end = pa_start + NBPDR;
9598 				} else if (pa_end == (*pde & PG_PS_FRAME))
9599 					pa_end += NBPDR;
9600 				else {
9601 					/* Run ended, update direct map. */
9602 					error = pmap_change_props_locked(
9603 					    PHYS_TO_DMAP(pa_start),
9604 					    pa_end - pa_start, prot, mode,
9605 					    flags);
9606 					if (error != 0)
9607 						break;
9608 					/* Start physical address run. */
9609 					pa_start = *pde & PG_PS_FRAME;
9610 					pa_end = pa_start + NBPDR;
9611 				}
9612 			}
9613 			tmpva = trunc_2mpage(tmpva) + NBPDR;
9614 		} else {
9615 			pte = pmap_pde_to_pte(pde, tmpva);
9616 			if ((*pte & pte_mask) != pte_bits) {
9617 				pmap_pte_props(pte, pte_bits, pte_mask);
9618 				changed = true;
9619 			}
9620 			if (tmpva >= VM_MIN_KERNEL_ADDRESS &&
9621 			    (*pte & PG_FRAME) < dmaplimit) {
9622 				if (pa_start == pa_end) {
9623 					/* Start physical address run. */
9624 					pa_start = *pte & PG_FRAME;
9625 					pa_end = pa_start + PAGE_SIZE;
9626 				} else if (pa_end == (*pte & PG_FRAME))
9627 					pa_end += PAGE_SIZE;
9628 				else {
9629 					/* Run ended, update direct map. */
9630 					error = pmap_change_props_locked(
9631 					    PHYS_TO_DMAP(pa_start),
9632 					    pa_end - pa_start, prot, mode,
9633 					    flags);
9634 					if (error != 0)
9635 						break;
9636 					/* Start physical address run. */
9637 					pa_start = *pte & PG_FRAME;
9638 					pa_end = pa_start + PAGE_SIZE;
9639 				}
9640 			}
9641 			tmpva += PAGE_SIZE;
9642 		}
9643 	}
9644 	if (error == 0 && pa_start != pa_end && pa_start < dmaplimit) {
9645 		pa_end1 = MIN(pa_end, dmaplimit);
9646 		if (pa_start != pa_end1)
9647 			error = pmap_change_props_locked(PHYS_TO_DMAP(pa_start),
9648 			    pa_end1 - pa_start, prot, mode, flags);
9649 	}
9650 
9651 	/*
9652 	 * Flush CPU caches if required to make sure any data isn't cached that
9653 	 * shouldn't be, etc.
9654 	 */
9655 	if (changed) {
9656 		pmap_invalidate_range(kernel_pmap, base, tmpva);
9657 		if ((flags & MAPDEV_FLUSHCACHE) != 0)
9658 			pmap_invalidate_cache_range(base, tmpva);
9659 	}
9660 	return (error);
9661 }
9662 
9663 /*
9664  * Demotes any mapping within the direct map region that covers more than the
9665  * specified range of physical addresses.  This range's size must be a power
9666  * of two and its starting address must be a multiple of its size.  Since the
9667  * demotion does not change any attributes of the mapping, a TLB invalidation
9668  * is not mandatory.  The caller may, however, request a TLB invalidation.
9669  */
9670 void
pmap_demote_DMAP(vm_paddr_t base,vm_size_t len,boolean_t invalidate)9671 pmap_demote_DMAP(vm_paddr_t base, vm_size_t len, boolean_t invalidate)
9672 {
9673 	pdp_entry_t *pdpe;
9674 	pd_entry_t *pde;
9675 	vm_offset_t va;
9676 	boolean_t changed;
9677 
9678 	if (len == 0)
9679 		return;
9680 	KASSERT(powerof2(len), ("pmap_demote_DMAP: len is not a power of 2"));
9681 	KASSERT((base & (len - 1)) == 0,
9682 	    ("pmap_demote_DMAP: base is not a multiple of len"));
9683 	if (len < NBPDP && base < dmaplimit) {
9684 		va = PHYS_TO_DMAP(base);
9685 		changed = FALSE;
9686 		PMAP_LOCK(kernel_pmap);
9687 		pdpe = pmap_pdpe(kernel_pmap, va);
9688 		if ((*pdpe & X86_PG_V) == 0)
9689 			panic("pmap_demote_DMAP: invalid PDPE");
9690 		if ((*pdpe & PG_PS) != 0) {
9691 			if (!pmap_demote_pdpe(kernel_pmap, pdpe, va))
9692 				panic("pmap_demote_DMAP: PDPE failed");
9693 			changed = TRUE;
9694 		}
9695 		if (len < NBPDR) {
9696 			pde = pmap_pdpe_to_pde(pdpe, va);
9697 			if ((*pde & X86_PG_V) == 0)
9698 				panic("pmap_demote_DMAP: invalid PDE");
9699 			if ((*pde & PG_PS) != 0) {
9700 				if (!pmap_demote_pde(kernel_pmap, pde, va))
9701 					panic("pmap_demote_DMAP: PDE failed");
9702 				changed = TRUE;
9703 			}
9704 		}
9705 		if (changed && invalidate)
9706 			pmap_invalidate_page(kernel_pmap, va);
9707 		PMAP_UNLOCK(kernel_pmap);
9708 	}
9709 }
9710 
9711 /*
9712  * Perform the pmap work for mincore(2).  If the page is not both referenced and
9713  * modified by this pmap, returns its physical address so that the caller can
9714  * find other mappings.
9715  */
9716 int
pmap_mincore(pmap_t pmap,vm_offset_t addr,vm_paddr_t * pap)9717 pmap_mincore(pmap_t pmap, vm_offset_t addr, vm_paddr_t *pap)
9718 {
9719 	pdp_entry_t *pdpe;
9720 	pd_entry_t *pdep;
9721 	pt_entry_t pte, PG_A, PG_M, PG_RW, PG_V;
9722 	vm_paddr_t pa;
9723 	int val;
9724 
9725 	PG_A = pmap_accessed_bit(pmap);
9726 	PG_M = pmap_modified_bit(pmap);
9727 	PG_V = pmap_valid_bit(pmap);
9728 	PG_RW = pmap_rw_bit(pmap);
9729 
9730 	PMAP_LOCK(pmap);
9731 	pte = 0;
9732 	pa = 0;
9733 	val = 0;
9734 	pdpe = pmap_pdpe(pmap, addr);
9735 	if (pdpe == NULL)
9736 		goto out;
9737 	if ((*pdpe & PG_V) != 0) {
9738 		if ((*pdpe & PG_PS) != 0) {
9739 			pte = *pdpe;
9740 			pa = ((pte & PG_PS_PDP_FRAME) | (addr & PDPMASK)) &
9741 			    PG_FRAME;
9742 			val = MINCORE_PSIND(2);
9743 		} else {
9744 			pdep = pmap_pde(pmap, addr);
9745 			if (pdep != NULL && (*pdep & PG_V) != 0) {
9746 				if ((*pdep & PG_PS) != 0) {
9747 					pte = *pdep;
9748 			/* Compute the physical address of the 4KB page. */
9749 					pa = ((pte & PG_PS_FRAME) | (addr &
9750 					    PDRMASK)) & PG_FRAME;
9751 					val = MINCORE_PSIND(1);
9752 				} else {
9753 					pte = *pmap_pde_to_pte(pdep, addr);
9754 					pa = pte & PG_FRAME;
9755 					val = 0;
9756 				}
9757 			}
9758 		}
9759 	}
9760 	if ((pte & PG_V) != 0) {
9761 		val |= MINCORE_INCORE;
9762 		if ((pte & (PG_M | PG_RW)) == (PG_M | PG_RW))
9763 			val |= MINCORE_MODIFIED | MINCORE_MODIFIED_OTHER;
9764 		if ((pte & PG_A) != 0)
9765 			val |= MINCORE_REFERENCED | MINCORE_REFERENCED_OTHER;
9766 	}
9767 	if ((val & (MINCORE_MODIFIED_OTHER | MINCORE_REFERENCED_OTHER)) !=
9768 	    (MINCORE_MODIFIED_OTHER | MINCORE_REFERENCED_OTHER) &&
9769 	    (pte & (PG_MANAGED | PG_V)) == (PG_MANAGED | PG_V)) {
9770 		*pap = pa;
9771 	}
9772 out:
9773 	PMAP_UNLOCK(pmap);
9774 	return (val);
9775 }
9776 
9777 static uint64_t
pmap_pcid_alloc(pmap_t pmap,u_int cpuid)9778 pmap_pcid_alloc(pmap_t pmap, u_int cpuid)
9779 {
9780 	uint32_t gen, new_gen, pcid_next;
9781 
9782 	CRITICAL_ASSERT(curthread);
9783 	gen = PCPU_GET(pcid_gen);
9784 	if (pmap->pm_pcids[cpuid].pm_pcid == PMAP_PCID_KERN)
9785 		return (pti ? 0 : CR3_PCID_SAVE);
9786 	if (pmap->pm_pcids[cpuid].pm_gen == gen)
9787 		return (CR3_PCID_SAVE);
9788 	pcid_next = PCPU_GET(pcid_next);
9789 	KASSERT((!pti && pcid_next <= PMAP_PCID_OVERMAX) ||
9790 	    (pti && pcid_next <= PMAP_PCID_OVERMAX_KERN),
9791 	    ("cpu %d pcid_next %#x", cpuid, pcid_next));
9792 	if ((!pti && pcid_next == PMAP_PCID_OVERMAX) ||
9793 	    (pti && pcid_next == PMAP_PCID_OVERMAX_KERN)) {
9794 		new_gen = gen + 1;
9795 		if (new_gen == 0)
9796 			new_gen = 1;
9797 		PCPU_SET(pcid_gen, new_gen);
9798 		pcid_next = PMAP_PCID_KERN + 1;
9799 	} else {
9800 		new_gen = gen;
9801 	}
9802 	pmap->pm_pcids[cpuid].pm_pcid = pcid_next;
9803 	pmap->pm_pcids[cpuid].pm_gen = new_gen;
9804 	PCPU_SET(pcid_next, pcid_next + 1);
9805 	return (0);
9806 }
9807 
9808 static uint64_t
pmap_pcid_alloc_checked(pmap_t pmap,u_int cpuid)9809 pmap_pcid_alloc_checked(pmap_t pmap, u_int cpuid)
9810 {
9811 	uint64_t cached;
9812 
9813 	cached = pmap_pcid_alloc(pmap, cpuid);
9814 	KASSERT(pmap->pm_pcids[cpuid].pm_pcid < PMAP_PCID_OVERMAX,
9815 	    ("pmap %p cpu %d pcid %#x", pmap, cpuid,
9816 	    pmap->pm_pcids[cpuid].pm_pcid));
9817 	KASSERT(pmap->pm_pcids[cpuid].pm_pcid != PMAP_PCID_KERN ||
9818 	    pmap == kernel_pmap,
9819 	    ("non-kernel pmap pmap %p cpu %d pcid %#x",
9820 	    pmap, cpuid, pmap->pm_pcids[cpuid].pm_pcid));
9821 	return (cached);
9822 }
9823 
9824 static void
pmap_activate_sw_pti_post(struct thread * td,pmap_t pmap)9825 pmap_activate_sw_pti_post(struct thread *td, pmap_t pmap)
9826 {
9827 
9828 	PCPU_GET(tssp)->tss_rsp0 = pmap->pm_ucr3 != PMAP_NO_CR3 ?
9829 	    PCPU_GET(pti_rsp0) : (uintptr_t)td->td_md.md_stack_base;
9830 }
9831 
9832 static void
pmap_activate_sw_pcid_pti(struct thread * td,pmap_t pmap,u_int cpuid)9833 pmap_activate_sw_pcid_pti(struct thread *td, pmap_t pmap, u_int cpuid)
9834 {
9835 	pmap_t old_pmap;
9836 	uint64_t cached, cr3, kcr3, ucr3;
9837 
9838 	KASSERT((read_rflags() & PSL_I) == 0,
9839 	    ("PCID needs interrupts disabled in pmap_activate_sw()"));
9840 
9841 	/* See the comment in pmap_invalidate_page_pcid(). */
9842 	if (PCPU_GET(ucr3_load_mask) != PMAP_UCR3_NOMASK) {
9843 		PCPU_SET(ucr3_load_mask, PMAP_UCR3_NOMASK);
9844 		old_pmap = PCPU_GET(curpmap);
9845 		MPASS(old_pmap->pm_ucr3 != PMAP_NO_CR3);
9846 		old_pmap->pm_pcids[cpuid].pm_gen = 0;
9847 	}
9848 
9849 	cached = pmap_pcid_alloc_checked(pmap, cpuid);
9850 	cr3 = rcr3();
9851 	if ((cr3 & ~CR3_PCID_MASK) != pmap->pm_cr3)
9852 		load_cr3(pmap->pm_cr3 | pmap->pm_pcids[cpuid].pm_pcid);
9853 	PCPU_SET(curpmap, pmap);
9854 	kcr3 = pmap->pm_cr3 | pmap->pm_pcids[cpuid].pm_pcid;
9855 	ucr3 = pmap->pm_ucr3 | pmap->pm_pcids[cpuid].pm_pcid |
9856 	    PMAP_PCID_USER_PT;
9857 
9858 	if (!cached && pmap->pm_ucr3 != PMAP_NO_CR3)
9859 		PCPU_SET(ucr3_load_mask, ~CR3_PCID_SAVE);
9860 
9861 	PCPU_SET(kcr3, kcr3 | CR3_PCID_SAVE);
9862 	PCPU_SET(ucr3, ucr3 | CR3_PCID_SAVE);
9863 	if (cached)
9864 		counter_u64_add(pcid_save_cnt, 1);
9865 
9866 	pmap_activate_sw_pti_post(td, pmap);
9867 }
9868 
9869 static void
pmap_activate_sw_pcid_nopti(struct thread * td __unused,pmap_t pmap,u_int cpuid)9870 pmap_activate_sw_pcid_nopti(struct thread *td __unused, pmap_t pmap,
9871     u_int cpuid)
9872 {
9873 	uint64_t cached, cr3;
9874 
9875 	KASSERT((read_rflags() & PSL_I) == 0,
9876 	    ("PCID needs interrupts disabled in pmap_activate_sw()"));
9877 
9878 	cached = pmap_pcid_alloc_checked(pmap, cpuid);
9879 	cr3 = rcr3();
9880 	if (!cached || (cr3 & ~CR3_PCID_MASK) != pmap->pm_cr3)
9881 		load_cr3(pmap->pm_cr3 | pmap->pm_pcids[cpuid].pm_pcid |
9882 		    cached);
9883 	PCPU_SET(curpmap, pmap);
9884 	if (cached)
9885 		counter_u64_add(pcid_save_cnt, 1);
9886 }
9887 
9888 static void
pmap_activate_sw_nopcid_nopti(struct thread * td __unused,pmap_t pmap,u_int cpuid __unused)9889 pmap_activate_sw_nopcid_nopti(struct thread *td __unused, pmap_t pmap,
9890     u_int cpuid __unused)
9891 {
9892 
9893 	load_cr3(pmap->pm_cr3);
9894 	PCPU_SET(curpmap, pmap);
9895 }
9896 
9897 static void
pmap_activate_sw_nopcid_pti(struct thread * td,pmap_t pmap,u_int cpuid __unused)9898 pmap_activate_sw_nopcid_pti(struct thread *td, pmap_t pmap,
9899     u_int cpuid __unused)
9900 {
9901 
9902 	pmap_activate_sw_nopcid_nopti(td, pmap, cpuid);
9903 	PCPU_SET(kcr3, pmap->pm_cr3);
9904 	PCPU_SET(ucr3, pmap->pm_ucr3);
9905 	pmap_activate_sw_pti_post(td, pmap);
9906 }
9907 
9908 DEFINE_IFUNC(static, void, pmap_activate_sw_mode, (struct thread *, pmap_t,
9909     u_int))
9910 {
9911 
9912 	if (pmap_pcid_enabled && pti)
9913 		return (pmap_activate_sw_pcid_pti);
9914 	else if (pmap_pcid_enabled && !pti)
9915 		return (pmap_activate_sw_pcid_nopti);
9916 	else if (!pmap_pcid_enabled && pti)
9917 		return (pmap_activate_sw_nopcid_pti);
9918 	else /* if (!pmap_pcid_enabled && !pti) */
9919 		return (pmap_activate_sw_nopcid_nopti);
9920 }
9921 
9922 void
pmap_activate_sw(struct thread * td)9923 pmap_activate_sw(struct thread *td)
9924 {
9925 	pmap_t oldpmap, pmap;
9926 	u_int cpuid;
9927 
9928 	oldpmap = PCPU_GET(curpmap);
9929 	pmap = vmspace_pmap(td->td_proc->p_vmspace);
9930 	if (oldpmap == pmap) {
9931 		if (cpu_vendor_id != CPU_VENDOR_INTEL)
9932 			mfence();
9933 		return;
9934 	}
9935 	cpuid = PCPU_GET(cpuid);
9936 #ifdef SMP
9937 	CPU_SET_ATOMIC(cpuid, &pmap->pm_active);
9938 #else
9939 	CPU_SET(cpuid, &pmap->pm_active);
9940 #endif
9941 	pmap_activate_sw_mode(td, pmap, cpuid);
9942 #ifdef SMP
9943 	CPU_CLR_ATOMIC(cpuid, &oldpmap->pm_active);
9944 #else
9945 	CPU_CLR(cpuid, &oldpmap->pm_active);
9946 #endif
9947 }
9948 
9949 void
pmap_activate(struct thread * td)9950 pmap_activate(struct thread *td)
9951 {
9952 	/*
9953 	 * invltlb_{invpcid,}_pcid_handler() is used to handle an
9954 	 * invalidate_all IPI, which checks for curpmap ==
9955 	 * smp_tlb_pmap.  The below sequence of operations has a
9956 	 * window where %CR3 is loaded with the new pmap's PML4
9957 	 * address, but the curpmap value has not yet been updated.
9958 	 * This causes the invltlb IPI handler, which is called
9959 	 * between the updates, to execute as a NOP, which leaves
9960 	 * stale TLB entries.
9961 	 *
9962 	 * Note that the most common use of pmap_activate_sw(), from
9963 	 * a context switch, is immune to this race, because
9964 	 * interrupts are disabled (while the thread lock is owned),
9965 	 * so the IPI is delayed until after curpmap is updated.  Protect
9966 	 * other callers in a similar way, by disabling interrupts
9967 	 * around the %cr3 register reload and curpmap assignment.
9968 	 */
9969 	spinlock_enter();
9970 	pmap_activate_sw(td);
9971 	spinlock_exit();
9972 }
9973 
9974 void
pmap_activate_boot(pmap_t pmap)9975 pmap_activate_boot(pmap_t pmap)
9976 {
9977 	uint64_t kcr3;
9978 	u_int cpuid;
9979 
9980 	/*
9981 	 * kernel_pmap must be never deactivated, and we ensure that
9982 	 * by never activating it at all.
9983 	 */
9984 	MPASS(pmap != kernel_pmap);
9985 
9986 	cpuid = PCPU_GET(cpuid);
9987 #ifdef SMP
9988 	CPU_SET_ATOMIC(cpuid, &pmap->pm_active);
9989 #else
9990 	CPU_SET(cpuid, &pmap->pm_active);
9991 #endif
9992 	PCPU_SET(curpmap, pmap);
9993 	if (pti) {
9994 		kcr3 = pmap->pm_cr3;
9995 		if (pmap_pcid_enabled)
9996 			kcr3 |= pmap->pm_pcids[cpuid].pm_pcid | CR3_PCID_SAVE;
9997 	} else {
9998 		kcr3 = PMAP_NO_CR3;
9999 	}
10000 	PCPU_SET(kcr3, kcr3);
10001 	PCPU_SET(ucr3, PMAP_NO_CR3);
10002 }
10003 
10004 void
pmap_active_cpus(pmap_t pmap,cpuset_t * res)10005 pmap_active_cpus(pmap_t pmap, cpuset_t *res)
10006 {
10007 	*res = pmap->pm_active;
10008 }
10009 
10010 void
pmap_sync_icache(pmap_t pm,vm_offset_t va,vm_size_t sz)10011 pmap_sync_icache(pmap_t pm, vm_offset_t va, vm_size_t sz)
10012 {
10013 }
10014 
10015 /*
10016  *	Increase the starting virtual address of the given mapping if a
10017  *	different alignment might result in more superpage mappings.
10018  */
10019 void
pmap_align_superpage(vm_object_t object,vm_ooffset_t offset,vm_offset_t * addr,vm_size_t size)10020 pmap_align_superpage(vm_object_t object, vm_ooffset_t offset,
10021     vm_offset_t *addr, vm_size_t size)
10022 {
10023 	vm_offset_t superpage_offset;
10024 
10025 	if (size < NBPDR)
10026 		return;
10027 	if (object != NULL && (object->flags & OBJ_COLORED) != 0)
10028 		offset += ptoa(object->pg_color);
10029 	superpage_offset = offset & PDRMASK;
10030 	if (size - ((NBPDR - superpage_offset) & PDRMASK) < NBPDR ||
10031 	    (*addr & PDRMASK) == superpage_offset)
10032 		return;
10033 	if ((*addr & PDRMASK) < superpage_offset)
10034 		*addr = (*addr & ~PDRMASK) + superpage_offset;
10035 	else
10036 		*addr = ((*addr + PDRMASK) & ~PDRMASK) + superpage_offset;
10037 }
10038 
10039 #ifdef INVARIANTS
10040 static unsigned long num_dirty_emulations;
10041 SYSCTL_ULONG(_vm_pmap, OID_AUTO, num_dirty_emulations, CTLFLAG_RW,
10042 	     &num_dirty_emulations, 0, NULL);
10043 
10044 static unsigned long num_accessed_emulations;
10045 SYSCTL_ULONG(_vm_pmap, OID_AUTO, num_accessed_emulations, CTLFLAG_RW,
10046 	     &num_accessed_emulations, 0, NULL);
10047 
10048 static unsigned long num_superpage_accessed_emulations;
10049 SYSCTL_ULONG(_vm_pmap, OID_AUTO, num_superpage_accessed_emulations, CTLFLAG_RW,
10050 	     &num_superpage_accessed_emulations, 0, NULL);
10051 
10052 static unsigned long ad_emulation_superpage_promotions;
10053 SYSCTL_ULONG(_vm_pmap, OID_AUTO, ad_emulation_superpage_promotions, CTLFLAG_RW,
10054 	     &ad_emulation_superpage_promotions, 0, NULL);
10055 #endif	/* INVARIANTS */
10056 
10057 int
pmap_emulate_accessed_dirty(pmap_t pmap,vm_offset_t va,int ftype)10058 pmap_emulate_accessed_dirty(pmap_t pmap, vm_offset_t va, int ftype)
10059 {
10060 	int rv;
10061 	struct rwlock *lock;
10062 #if VM_NRESERVLEVEL > 0
10063 	vm_page_t m, mpte;
10064 #endif
10065 	pd_entry_t *pde;
10066 	pt_entry_t *pte, PG_A, PG_M, PG_RW, PG_V;
10067 
10068 	KASSERT(ftype == VM_PROT_READ || ftype == VM_PROT_WRITE,
10069 	    ("pmap_emulate_accessed_dirty: invalid fault type %d", ftype));
10070 
10071 	if (!pmap_emulate_ad_bits(pmap))
10072 		return (-1);
10073 
10074 	PG_A = pmap_accessed_bit(pmap);
10075 	PG_M = pmap_modified_bit(pmap);
10076 	PG_V = pmap_valid_bit(pmap);
10077 	PG_RW = pmap_rw_bit(pmap);
10078 
10079 	rv = -1;
10080 	lock = NULL;
10081 	PMAP_LOCK(pmap);
10082 
10083 	pde = pmap_pde(pmap, va);
10084 	if (pde == NULL || (*pde & PG_V) == 0)
10085 		goto done;
10086 
10087 	if ((*pde & PG_PS) != 0) {
10088 		if (ftype == VM_PROT_READ) {
10089 #ifdef INVARIANTS
10090 			atomic_add_long(&num_superpage_accessed_emulations, 1);
10091 #endif
10092 			*pde |= PG_A;
10093 			rv = 0;
10094 		}
10095 		goto done;
10096 	}
10097 
10098 	pte = pmap_pde_to_pte(pde, va);
10099 	if ((*pte & PG_V) == 0)
10100 		goto done;
10101 
10102 	if (ftype == VM_PROT_WRITE) {
10103 		if ((*pte & PG_RW) == 0)
10104 			goto done;
10105 		/*
10106 		 * Set the modified and accessed bits simultaneously.
10107 		 *
10108 		 * Intel EPT PTEs that do software emulation of A/D bits map
10109 		 * PG_A and PG_M to EPT_PG_READ and EPT_PG_WRITE respectively.
10110 		 * An EPT misconfiguration is triggered if the PTE is writable
10111 		 * but not readable (WR=10). This is avoided by setting PG_A
10112 		 * and PG_M simultaneously.
10113 		 */
10114 		*pte |= PG_M | PG_A;
10115 	} else {
10116 		*pte |= PG_A;
10117 	}
10118 
10119 #if VM_NRESERVLEVEL > 0
10120 	/* try to promote the mapping */
10121 	if (va < VM_MAXUSER_ADDRESS)
10122 		mpte = PHYS_TO_VM_PAGE(*pde & PG_FRAME);
10123 	else
10124 		mpte = NULL;
10125 
10126 	m = PHYS_TO_VM_PAGE(*pte & PG_FRAME);
10127 
10128 	if ((mpte == NULL || mpte->ref_count == NPTEPG) &&
10129 	    pmap_ps_enabled(pmap) &&
10130 	    (m->flags & PG_FICTITIOUS) == 0 &&
10131 	    vm_reserv_level_iffullpop(m) == 0) {
10132 		pmap_promote_pde(pmap, pde, va, &lock);
10133 #ifdef INVARIANTS
10134 		atomic_add_long(&ad_emulation_superpage_promotions, 1);
10135 #endif
10136 	}
10137 #endif
10138 
10139 #ifdef INVARIANTS
10140 	if (ftype == VM_PROT_WRITE)
10141 		atomic_add_long(&num_dirty_emulations, 1);
10142 	else
10143 		atomic_add_long(&num_accessed_emulations, 1);
10144 #endif
10145 	rv = 0;		/* success */
10146 done:
10147 	if (lock != NULL)
10148 		rw_wunlock(lock);
10149 	PMAP_UNLOCK(pmap);
10150 	return (rv);
10151 }
10152 
10153 void
pmap_get_mapping(pmap_t pmap,vm_offset_t va,uint64_t * ptr,int * num)10154 pmap_get_mapping(pmap_t pmap, vm_offset_t va, uint64_t *ptr, int *num)
10155 {
10156 	pml4_entry_t *pml4;
10157 	pdp_entry_t *pdp;
10158 	pd_entry_t *pde;
10159 	pt_entry_t *pte, PG_V;
10160 	int idx;
10161 
10162 	idx = 0;
10163 	PG_V = pmap_valid_bit(pmap);
10164 	PMAP_LOCK(pmap);
10165 
10166 	pml4 = pmap_pml4e(pmap, va);
10167 	if (pml4 == NULL)
10168 		goto done;
10169 	ptr[idx++] = *pml4;
10170 	if ((*pml4 & PG_V) == 0)
10171 		goto done;
10172 
10173 	pdp = pmap_pml4e_to_pdpe(pml4, va);
10174 	ptr[idx++] = *pdp;
10175 	if ((*pdp & PG_V) == 0 || (*pdp & PG_PS) != 0)
10176 		goto done;
10177 
10178 	pde = pmap_pdpe_to_pde(pdp, va);
10179 	ptr[idx++] = *pde;
10180 	if ((*pde & PG_V) == 0 || (*pde & PG_PS) != 0)
10181 		goto done;
10182 
10183 	pte = pmap_pde_to_pte(pde, va);
10184 	ptr[idx++] = *pte;
10185 
10186 done:
10187 	PMAP_UNLOCK(pmap);
10188 	*num = idx;
10189 }
10190 
10191 /**
10192  * Get the kernel virtual address of a set of physical pages. If there are
10193  * physical addresses not covered by the DMAP perform a transient mapping
10194  * that will be removed when calling pmap_unmap_io_transient.
10195  *
10196  * \param page        The pages the caller wishes to obtain the virtual
10197  *                    address on the kernel memory map.
10198  * \param vaddr       On return contains the kernel virtual memory address
10199  *                    of the pages passed in the page parameter.
10200  * \param count       Number of pages passed in.
10201  * \param can_fault   TRUE if the thread using the mapped pages can take
10202  *                    page faults, FALSE otherwise.
10203  *
10204  * \returns TRUE if the caller must call pmap_unmap_io_transient when
10205  *          finished or FALSE otherwise.
10206  *
10207  */
10208 boolean_t
pmap_map_io_transient(vm_page_t page[],vm_offset_t vaddr[],int count,boolean_t can_fault)10209 pmap_map_io_transient(vm_page_t page[], vm_offset_t vaddr[], int count,
10210     boolean_t can_fault)
10211 {
10212 	vm_paddr_t paddr;
10213 	boolean_t needs_mapping;
10214 	pt_entry_t *pte;
10215 	int cache_bits, error __unused, i;
10216 
10217 	/*
10218 	 * Allocate any KVA space that we need, this is done in a separate
10219 	 * loop to prevent calling vmem_alloc while pinned.
10220 	 */
10221 	needs_mapping = FALSE;
10222 	for (i = 0; i < count; i++) {
10223 		paddr = VM_PAGE_TO_PHYS(page[i]);
10224 		if (__predict_false(paddr >= dmaplimit)) {
10225 			error = vmem_alloc(kernel_arena, PAGE_SIZE,
10226 			    M_BESTFIT | M_WAITOK, &vaddr[i]);
10227 			KASSERT(error == 0, ("vmem_alloc failed: %d", error));
10228 			needs_mapping = TRUE;
10229 		} else {
10230 			vaddr[i] = PHYS_TO_DMAP(paddr);
10231 		}
10232 	}
10233 
10234 	/* Exit early if everything is covered by the DMAP */
10235 	if (!needs_mapping)
10236 		return (FALSE);
10237 
10238 	/*
10239 	 * NB:  The sequence of updating a page table followed by accesses
10240 	 * to the corresponding pages used in the !DMAP case is subject to
10241 	 * the situation described in the "AMD64 Architecture Programmer's
10242 	 * Manual Volume 2: System Programming" rev. 3.23, "7.3.1 Special
10243 	 * Coherency Considerations".  Therefore, issuing the INVLPG right
10244 	 * after modifying the PTE bits is crucial.
10245 	 */
10246 	if (!can_fault)
10247 		sched_pin();
10248 	for (i = 0; i < count; i++) {
10249 		paddr = VM_PAGE_TO_PHYS(page[i]);
10250 		if (paddr >= dmaplimit) {
10251 			if (can_fault) {
10252 				/*
10253 				 * Slow path, since we can get page faults
10254 				 * while mappings are active don't pin the
10255 				 * thread to the CPU and instead add a global
10256 				 * mapping visible to all CPUs.
10257 				 */
10258 				pmap_qenter(vaddr[i], &page[i], 1);
10259 			} else {
10260 				pte = vtopte(vaddr[i]);
10261 				cache_bits = pmap_cache_bits(kernel_pmap,
10262 				    page[i]->md.pat_mode, 0);
10263 				pte_store(pte, paddr | X86_PG_RW | X86_PG_V |
10264 				    cache_bits);
10265 				pmap_invlpg(kernel_pmap, vaddr[i]);
10266 			}
10267 		}
10268 	}
10269 
10270 	return (needs_mapping);
10271 }
10272 
10273 void
pmap_unmap_io_transient(vm_page_t page[],vm_offset_t vaddr[],int count,boolean_t can_fault)10274 pmap_unmap_io_transient(vm_page_t page[], vm_offset_t vaddr[], int count,
10275     boolean_t can_fault)
10276 {
10277 	vm_paddr_t paddr;
10278 	int i;
10279 
10280 	if (!can_fault)
10281 		sched_unpin();
10282 	for (i = 0; i < count; i++) {
10283 		paddr = VM_PAGE_TO_PHYS(page[i]);
10284 		if (paddr >= dmaplimit) {
10285 			if (can_fault)
10286 				pmap_qremove(vaddr[i], 1);
10287 			vmem_free(kernel_arena, vaddr[i], PAGE_SIZE);
10288 		}
10289 	}
10290 }
10291 
10292 vm_offset_t
pmap_quick_enter_page(vm_page_t m)10293 pmap_quick_enter_page(vm_page_t m)
10294 {
10295 	vm_paddr_t paddr;
10296 
10297 	paddr = VM_PAGE_TO_PHYS(m);
10298 	if (paddr < dmaplimit)
10299 		return (PHYS_TO_DMAP(paddr));
10300 	mtx_lock_spin(&qframe_mtx);
10301 	KASSERT(*vtopte(qframe) == 0, ("qframe busy"));
10302 
10303 	/*
10304 	 * Since qframe is exclusively mapped by us, and we do not set
10305 	 * PG_G, we can use INVLPG here.
10306 	 */
10307 	invlpg(qframe);
10308 
10309 	pte_store(vtopte(qframe), paddr | X86_PG_RW | X86_PG_V | X86_PG_A |
10310 	    X86_PG_M | pmap_cache_bits(kernel_pmap, m->md.pat_mode, 0));
10311 	return (qframe);
10312 }
10313 
10314 void
pmap_quick_remove_page(vm_offset_t addr)10315 pmap_quick_remove_page(vm_offset_t addr)
10316 {
10317 
10318 	if (addr != qframe)
10319 		return;
10320 	pte_store(vtopte(qframe), 0);
10321 	mtx_unlock_spin(&qframe_mtx);
10322 }
10323 
10324 /*
10325  * Pdp pages from the large map are managed differently from either
10326  * kernel or user page table pages.  They are permanently allocated at
10327  * initialization time, and their reference count is permanently set to
10328  * zero.  The pml4 entries pointing to those pages are copied into
10329  * each allocated pmap.
10330  *
10331  * In contrast, pd and pt pages are managed like user page table
10332  * pages.  They are dynamically allocated, and their reference count
10333  * represents the number of valid entries within the page.
10334  */
10335 static vm_page_t
pmap_large_map_getptp_unlocked(void)10336 pmap_large_map_getptp_unlocked(void)
10337 {
10338 	return (pmap_alloc_pt_page(kernel_pmap, 0, VM_ALLOC_ZERO));
10339 }
10340 
10341 static vm_page_t
pmap_large_map_getptp(void)10342 pmap_large_map_getptp(void)
10343 {
10344 	vm_page_t m;
10345 
10346 	PMAP_LOCK_ASSERT(kernel_pmap, MA_OWNED);
10347 	m = pmap_large_map_getptp_unlocked();
10348 	if (m == NULL) {
10349 		PMAP_UNLOCK(kernel_pmap);
10350 		vm_wait(NULL);
10351 		PMAP_LOCK(kernel_pmap);
10352 		/* Callers retry. */
10353 	}
10354 	return (m);
10355 }
10356 
10357 static pdp_entry_t *
pmap_large_map_pdpe(vm_offset_t va)10358 pmap_large_map_pdpe(vm_offset_t va)
10359 {
10360 	vm_pindex_t pml4_idx;
10361 	vm_paddr_t mphys;
10362 
10363 	pml4_idx = pmap_pml4e_index(va);
10364 	KASSERT(LMSPML4I <= pml4_idx && pml4_idx < LMSPML4I + lm_ents,
10365 	    ("pmap_large_map_pdpe: va %#jx out of range idx %#jx LMSPML4I "
10366 	    "%#jx lm_ents %d",
10367 	    (uintmax_t)va, (uintmax_t)pml4_idx, LMSPML4I, lm_ents));
10368 	KASSERT((kernel_pml4[pml4_idx] & X86_PG_V) != 0,
10369 	    ("pmap_large_map_pdpe: invalid pml4 for va %#jx idx %#jx "
10370 	    "LMSPML4I %#jx lm_ents %d",
10371 	    (uintmax_t)va, (uintmax_t)pml4_idx, LMSPML4I, lm_ents));
10372 	mphys = kernel_pml4[pml4_idx] & PG_FRAME;
10373 	return ((pdp_entry_t *)PHYS_TO_DMAP(mphys) + pmap_pdpe_index(va));
10374 }
10375 
10376 static pd_entry_t *
pmap_large_map_pde(vm_offset_t va)10377 pmap_large_map_pde(vm_offset_t va)
10378 {
10379 	pdp_entry_t *pdpe;
10380 	vm_page_t m;
10381 	vm_paddr_t mphys;
10382 
10383 retry:
10384 	pdpe = pmap_large_map_pdpe(va);
10385 	if (*pdpe == 0) {
10386 		m = pmap_large_map_getptp();
10387 		if (m == NULL)
10388 			goto retry;
10389 		mphys = VM_PAGE_TO_PHYS(m);
10390 		*pdpe = mphys | X86_PG_A | X86_PG_RW | X86_PG_V | pg_nx;
10391 	} else {
10392 		MPASS((*pdpe & X86_PG_PS) == 0);
10393 		mphys = *pdpe & PG_FRAME;
10394 	}
10395 	return ((pd_entry_t *)PHYS_TO_DMAP(mphys) + pmap_pde_index(va));
10396 }
10397 
10398 static pt_entry_t *
pmap_large_map_pte(vm_offset_t va)10399 pmap_large_map_pte(vm_offset_t va)
10400 {
10401 	pd_entry_t *pde;
10402 	vm_page_t m;
10403 	vm_paddr_t mphys;
10404 
10405 retry:
10406 	pde = pmap_large_map_pde(va);
10407 	if (*pde == 0) {
10408 		m = pmap_large_map_getptp();
10409 		if (m == NULL)
10410 			goto retry;
10411 		mphys = VM_PAGE_TO_PHYS(m);
10412 		*pde = mphys | X86_PG_A | X86_PG_RW | X86_PG_V | pg_nx;
10413 		PHYS_TO_VM_PAGE(DMAP_TO_PHYS((uintptr_t)pde))->ref_count++;
10414 	} else {
10415 		MPASS((*pde & X86_PG_PS) == 0);
10416 		mphys = *pde & PG_FRAME;
10417 	}
10418 	return ((pt_entry_t *)PHYS_TO_DMAP(mphys) + pmap_pte_index(va));
10419 }
10420 
10421 static vm_paddr_t
pmap_large_map_kextract(vm_offset_t va)10422 pmap_large_map_kextract(vm_offset_t va)
10423 {
10424 	pdp_entry_t *pdpe, pdp;
10425 	pd_entry_t *pde, pd;
10426 	pt_entry_t *pte, pt;
10427 
10428 	KASSERT(PMAP_ADDRESS_IN_LARGEMAP(va),
10429 	    ("not largemap range %#lx", (u_long)va));
10430 	pdpe = pmap_large_map_pdpe(va);
10431 	pdp = *pdpe;
10432 	KASSERT((pdp & X86_PG_V) != 0,
10433 	    ("invalid pdp va %#lx pdpe %#lx pdp %#lx", va,
10434 	    (u_long)pdpe, pdp));
10435 	if ((pdp & X86_PG_PS) != 0) {
10436 		KASSERT((amd_feature & AMDID_PAGE1GB) != 0,
10437 		    ("no 1G pages, va %#lx pdpe %#lx pdp %#lx", va,
10438 		    (u_long)pdpe, pdp));
10439 		return ((pdp & PG_PS_PDP_FRAME) | (va & PDPMASK));
10440 	}
10441 	pde = pmap_pdpe_to_pde(pdpe, va);
10442 	pd = *pde;
10443 	KASSERT((pd & X86_PG_V) != 0,
10444 	    ("invalid pd va %#lx pde %#lx pd %#lx", va, (u_long)pde, pd));
10445 	if ((pd & X86_PG_PS) != 0)
10446 		return ((pd & PG_PS_FRAME) | (va & PDRMASK));
10447 	pte = pmap_pde_to_pte(pde, va);
10448 	pt = *pte;
10449 	KASSERT((pt & X86_PG_V) != 0,
10450 	    ("invalid pte va %#lx pte %#lx pt %#lx", va, (u_long)pte, pt));
10451 	return ((pt & PG_FRAME) | (va & PAGE_MASK));
10452 }
10453 
10454 static int
pmap_large_map_getva(vm_size_t len,vm_offset_t align,vm_offset_t phase,vmem_addr_t * vmem_res)10455 pmap_large_map_getva(vm_size_t len, vm_offset_t align, vm_offset_t phase,
10456     vmem_addr_t *vmem_res)
10457 {
10458 
10459 	/*
10460 	 * Large mappings are all but static.  Consequently, there
10461 	 * is no point in waiting for an earlier allocation to be
10462 	 * freed.
10463 	 */
10464 	return (vmem_xalloc(large_vmem, len, align, phase, 0, VMEM_ADDR_MIN,
10465 	    VMEM_ADDR_MAX, M_NOWAIT | M_BESTFIT, vmem_res));
10466 }
10467 
10468 int
pmap_large_map(vm_paddr_t spa,vm_size_t len,void ** addr,vm_memattr_t mattr)10469 pmap_large_map(vm_paddr_t spa, vm_size_t len, void **addr,
10470     vm_memattr_t mattr)
10471 {
10472 	pdp_entry_t *pdpe;
10473 	pd_entry_t *pde;
10474 	pt_entry_t *pte;
10475 	vm_offset_t va, inc;
10476 	vmem_addr_t vmem_res;
10477 	vm_paddr_t pa;
10478 	int error;
10479 
10480 	if (len == 0 || spa + len < spa)
10481 		return (EINVAL);
10482 
10483 	/* See if DMAP can serve. */
10484 	if (spa + len <= dmaplimit) {
10485 		va = PHYS_TO_DMAP(spa);
10486 		*addr = (void *)va;
10487 		return (pmap_change_attr(va, len, mattr));
10488 	}
10489 
10490 	/*
10491 	 * No, allocate KVA.  Fit the address with best possible
10492 	 * alignment for superpages.  Fall back to worse align if
10493 	 * failed.
10494 	 */
10495 	error = ENOMEM;
10496 	if ((amd_feature & AMDID_PAGE1GB) != 0 && rounddown2(spa + len,
10497 	    NBPDP) >= roundup2(spa, NBPDP) + NBPDP)
10498 		error = pmap_large_map_getva(len, NBPDP, spa & PDPMASK,
10499 		    &vmem_res);
10500 	if (error != 0 && rounddown2(spa + len, NBPDR) >= roundup2(spa,
10501 	    NBPDR) + NBPDR)
10502 		error = pmap_large_map_getva(len, NBPDR, spa & PDRMASK,
10503 		    &vmem_res);
10504 	if (error != 0)
10505 		error = pmap_large_map_getva(len, PAGE_SIZE, 0, &vmem_res);
10506 	if (error != 0)
10507 		return (error);
10508 
10509 	/*
10510 	 * Fill pagetable.  PG_M is not pre-set, we scan modified bits
10511 	 * in the pagetable to minimize flushing.  No need to
10512 	 * invalidate TLB, since we only update invalid entries.
10513 	 */
10514 	PMAP_LOCK(kernel_pmap);
10515 	for (pa = spa, va = vmem_res; len > 0; pa += inc, va += inc,
10516 	    len -= inc) {
10517 		if ((amd_feature & AMDID_PAGE1GB) != 0 && len >= NBPDP &&
10518 		    (pa & PDPMASK) == 0 && (va & PDPMASK) == 0) {
10519 			pdpe = pmap_large_map_pdpe(va);
10520 			MPASS(*pdpe == 0);
10521 			*pdpe = pa | pg_g | X86_PG_PS | X86_PG_RW |
10522 			    X86_PG_V | X86_PG_A | pg_nx |
10523 			    pmap_cache_bits(kernel_pmap, mattr, TRUE);
10524 			inc = NBPDP;
10525 		} else if (len >= NBPDR && (pa & PDRMASK) == 0 &&
10526 		    (va & PDRMASK) == 0) {
10527 			pde = pmap_large_map_pde(va);
10528 			MPASS(*pde == 0);
10529 			*pde = pa | pg_g | X86_PG_PS | X86_PG_RW |
10530 			    X86_PG_V | X86_PG_A | pg_nx |
10531 			    pmap_cache_bits(kernel_pmap, mattr, TRUE);
10532 			PHYS_TO_VM_PAGE(DMAP_TO_PHYS((uintptr_t)pde))->
10533 			    ref_count++;
10534 			inc = NBPDR;
10535 		} else {
10536 			pte = pmap_large_map_pte(va);
10537 			MPASS(*pte == 0);
10538 			*pte = pa | pg_g | X86_PG_RW | X86_PG_V |
10539 			    X86_PG_A | pg_nx | pmap_cache_bits(kernel_pmap,
10540 			    mattr, FALSE);
10541 			PHYS_TO_VM_PAGE(DMAP_TO_PHYS((uintptr_t)pte))->
10542 			    ref_count++;
10543 			inc = PAGE_SIZE;
10544 		}
10545 	}
10546 	PMAP_UNLOCK(kernel_pmap);
10547 	MPASS(len == 0);
10548 
10549 	*addr = (void *)vmem_res;
10550 	return (0);
10551 }
10552 
10553 void
pmap_large_unmap(void * svaa,vm_size_t len)10554 pmap_large_unmap(void *svaa, vm_size_t len)
10555 {
10556 	vm_offset_t sva, va;
10557 	vm_size_t inc;
10558 	pdp_entry_t *pdpe, pdp;
10559 	pd_entry_t *pde, pd;
10560 	pt_entry_t *pte;
10561 	vm_page_t m;
10562 	struct spglist spgf;
10563 
10564 	sva = (vm_offset_t)svaa;
10565 	if (len == 0 || sva + len < sva || (sva >= DMAP_MIN_ADDRESS &&
10566 	    sva + len <= DMAP_MIN_ADDRESS + dmaplimit))
10567 		return;
10568 
10569 	SLIST_INIT(&spgf);
10570 	KASSERT(PMAP_ADDRESS_IN_LARGEMAP(sva) &&
10571 	    PMAP_ADDRESS_IN_LARGEMAP(sva + len - 1),
10572 	    ("not largemap range %#lx %#lx", (u_long)svaa, (u_long)svaa + len));
10573 	PMAP_LOCK(kernel_pmap);
10574 	for (va = sva; va < sva + len; va += inc) {
10575 		pdpe = pmap_large_map_pdpe(va);
10576 		pdp = *pdpe;
10577 		KASSERT((pdp & X86_PG_V) != 0,
10578 		    ("invalid pdp va %#lx pdpe %#lx pdp %#lx", va,
10579 		    (u_long)pdpe, pdp));
10580 		if ((pdp & X86_PG_PS) != 0) {
10581 			KASSERT((amd_feature & AMDID_PAGE1GB) != 0,
10582 			    ("no 1G pages, va %#lx pdpe %#lx pdp %#lx", va,
10583 			    (u_long)pdpe, pdp));
10584 			KASSERT((va & PDPMASK) == 0,
10585 			    ("PDPMASK bit set, va %#lx pdpe %#lx pdp %#lx", va,
10586 			    (u_long)pdpe, pdp));
10587 			KASSERT(va + NBPDP <= sva + len,
10588 			    ("unmap covers partial 1GB page, sva %#lx va %#lx "
10589 			    "pdpe %#lx pdp %#lx len %#lx", sva, va,
10590 			    (u_long)pdpe, pdp, len));
10591 			*pdpe = 0;
10592 			inc = NBPDP;
10593 			continue;
10594 		}
10595 		pde = pmap_pdpe_to_pde(pdpe, va);
10596 		pd = *pde;
10597 		KASSERT((pd & X86_PG_V) != 0,
10598 		    ("invalid pd va %#lx pde %#lx pd %#lx", va,
10599 		    (u_long)pde, pd));
10600 		if ((pd & X86_PG_PS) != 0) {
10601 			KASSERT((va & PDRMASK) == 0,
10602 			    ("PDRMASK bit set, va %#lx pde %#lx pd %#lx", va,
10603 			    (u_long)pde, pd));
10604 			KASSERT(va + NBPDR <= sva + len,
10605 			    ("unmap covers partial 2MB page, sva %#lx va %#lx "
10606 			    "pde %#lx pd %#lx len %#lx", sva, va, (u_long)pde,
10607 			    pd, len));
10608 			pde_store(pde, 0);
10609 			inc = NBPDR;
10610 			m = PHYS_TO_VM_PAGE(DMAP_TO_PHYS((vm_offset_t)pde));
10611 			m->ref_count--;
10612 			if (m->ref_count == 0) {
10613 				*pdpe = 0;
10614 				SLIST_INSERT_HEAD(&spgf, m, plinks.s.ss);
10615 			}
10616 			continue;
10617 		}
10618 		pte = pmap_pde_to_pte(pde, va);
10619 		KASSERT((*pte & X86_PG_V) != 0,
10620 		    ("invalid pte va %#lx pte %#lx pt %#lx", va,
10621 		    (u_long)pte, *pte));
10622 		pte_clear(pte);
10623 		inc = PAGE_SIZE;
10624 		m = PHYS_TO_VM_PAGE(DMAP_TO_PHYS((vm_offset_t)pte));
10625 		m->ref_count--;
10626 		if (m->ref_count == 0) {
10627 			*pde = 0;
10628 			SLIST_INSERT_HEAD(&spgf, m, plinks.s.ss);
10629 			m = PHYS_TO_VM_PAGE(DMAP_TO_PHYS((vm_offset_t)pde));
10630 			m->ref_count--;
10631 			if (m->ref_count == 0) {
10632 				*pdpe = 0;
10633 				SLIST_INSERT_HEAD(&spgf, m, plinks.s.ss);
10634 			}
10635 		}
10636 	}
10637 	pmap_invalidate_range(kernel_pmap, sva, sva + len);
10638 	PMAP_UNLOCK(kernel_pmap);
10639 	vm_page_free_pages_toq(&spgf, false);
10640 	vmem_free(large_vmem, sva, len);
10641 }
10642 
10643 static void
pmap_large_map_wb_fence_mfence(void)10644 pmap_large_map_wb_fence_mfence(void)
10645 {
10646 
10647 	mfence();
10648 }
10649 
10650 static void
pmap_large_map_wb_fence_atomic(void)10651 pmap_large_map_wb_fence_atomic(void)
10652 {
10653 
10654 	atomic_thread_fence_seq_cst();
10655 }
10656 
10657 static void
pmap_large_map_wb_fence_nop(void)10658 pmap_large_map_wb_fence_nop(void)
10659 {
10660 }
10661 
10662 DEFINE_IFUNC(static, void, pmap_large_map_wb_fence, (void))
10663 {
10664 
10665 	if (cpu_vendor_id != CPU_VENDOR_INTEL)
10666 		return (pmap_large_map_wb_fence_mfence);
10667 	else if ((cpu_stdext_feature & (CPUID_STDEXT_CLWB |
10668 	    CPUID_STDEXT_CLFLUSHOPT)) == 0)
10669 		return (pmap_large_map_wb_fence_atomic);
10670 	else
10671 		/* clflush is strongly enough ordered */
10672 		return (pmap_large_map_wb_fence_nop);
10673 }
10674 
10675 static void
pmap_large_map_flush_range_clwb(vm_offset_t va,vm_size_t len)10676 pmap_large_map_flush_range_clwb(vm_offset_t va, vm_size_t len)
10677 {
10678 
10679 	for (; len > 0; len -= cpu_clflush_line_size,
10680 	    va += cpu_clflush_line_size)
10681 		clwb(va);
10682 }
10683 
10684 static void
pmap_large_map_flush_range_clflushopt(vm_offset_t va,vm_size_t len)10685 pmap_large_map_flush_range_clflushopt(vm_offset_t va, vm_size_t len)
10686 {
10687 
10688 	for (; len > 0; len -= cpu_clflush_line_size,
10689 	    va += cpu_clflush_line_size)
10690 		clflushopt(va);
10691 }
10692 
10693 static void
pmap_large_map_flush_range_clflush(vm_offset_t va,vm_size_t len)10694 pmap_large_map_flush_range_clflush(vm_offset_t va, vm_size_t len)
10695 {
10696 
10697 	for (; len > 0; len -= cpu_clflush_line_size,
10698 	    va += cpu_clflush_line_size)
10699 		clflush(va);
10700 }
10701 
10702 static void
pmap_large_map_flush_range_nop(vm_offset_t sva __unused,vm_size_t len __unused)10703 pmap_large_map_flush_range_nop(vm_offset_t sva __unused, vm_size_t len __unused)
10704 {
10705 }
10706 
10707 DEFINE_IFUNC(static, void, pmap_large_map_flush_range, (vm_offset_t, vm_size_t))
10708 {
10709 
10710 	if ((cpu_stdext_feature & CPUID_STDEXT_CLWB) != 0)
10711 		return (pmap_large_map_flush_range_clwb);
10712 	else if ((cpu_stdext_feature & CPUID_STDEXT_CLFLUSHOPT) != 0)
10713 		return (pmap_large_map_flush_range_clflushopt);
10714 	else if ((cpu_feature & CPUID_CLFSH) != 0)
10715 		return (pmap_large_map_flush_range_clflush);
10716 	else
10717 		return (pmap_large_map_flush_range_nop);
10718 }
10719 
10720 static void
pmap_large_map_wb_large(vm_offset_t sva,vm_offset_t eva)10721 pmap_large_map_wb_large(vm_offset_t sva, vm_offset_t eva)
10722 {
10723 	volatile u_long *pe;
10724 	u_long p;
10725 	vm_offset_t va;
10726 	vm_size_t inc;
10727 	bool seen_other;
10728 
10729 	for (va = sva; va < eva; va += inc) {
10730 		inc = 0;
10731 		if ((amd_feature & AMDID_PAGE1GB) != 0) {
10732 			pe = (volatile u_long *)pmap_large_map_pdpe(va);
10733 			p = *pe;
10734 			if ((p & X86_PG_PS) != 0)
10735 				inc = NBPDP;
10736 		}
10737 		if (inc == 0) {
10738 			pe = (volatile u_long *)pmap_large_map_pde(va);
10739 			p = *pe;
10740 			if ((p & X86_PG_PS) != 0)
10741 				inc = NBPDR;
10742 		}
10743 		if (inc == 0) {
10744 			pe = (volatile u_long *)pmap_large_map_pte(va);
10745 			p = *pe;
10746 			inc = PAGE_SIZE;
10747 		}
10748 		seen_other = false;
10749 		for (;;) {
10750 			if ((p & X86_PG_AVAIL1) != 0) {
10751 				/*
10752 				 * Spin-wait for the end of a parallel
10753 				 * write-back.
10754 				 */
10755 				cpu_spinwait();
10756 				p = *pe;
10757 
10758 				/*
10759 				 * If we saw other write-back
10760 				 * occuring, we cannot rely on PG_M to
10761 				 * indicate state of the cache.  The
10762 				 * PG_M bit is cleared before the
10763 				 * flush to avoid ignoring new writes,
10764 				 * and writes which are relevant for
10765 				 * us might happen after.
10766 				 */
10767 				seen_other = true;
10768 				continue;
10769 			}
10770 
10771 			if ((p & X86_PG_M) != 0 || seen_other) {
10772 				if (!atomic_fcmpset_long(pe, &p,
10773 				    (p & ~X86_PG_M) | X86_PG_AVAIL1))
10774 					/*
10775 					 * If we saw PG_M without
10776 					 * PG_AVAIL1, and then on the
10777 					 * next attempt we do not
10778 					 * observe either PG_M or
10779 					 * PG_AVAIL1, the other
10780 					 * write-back started after us
10781 					 * and finished before us.  We
10782 					 * can rely on it doing our
10783 					 * work.
10784 					 */
10785 					continue;
10786 				pmap_large_map_flush_range(va, inc);
10787 				atomic_clear_long(pe, X86_PG_AVAIL1);
10788 			}
10789 			break;
10790 		}
10791 		maybe_yield();
10792 	}
10793 }
10794 
10795 /*
10796  * Write-back cache lines for the given address range.
10797  *
10798  * Must be called only on the range or sub-range returned from
10799  * pmap_large_map().  Must not be called on the coalesced ranges.
10800  *
10801  * Does nothing on CPUs without CLWB, CLFLUSHOPT, or CLFLUSH
10802  * instructions support.
10803  */
10804 void
pmap_large_map_wb(void * svap,vm_size_t len)10805 pmap_large_map_wb(void *svap, vm_size_t len)
10806 {
10807 	vm_offset_t eva, sva;
10808 
10809 	sva = (vm_offset_t)svap;
10810 	eva = sva + len;
10811 	pmap_large_map_wb_fence();
10812 	if (sva >= DMAP_MIN_ADDRESS && eva <= DMAP_MIN_ADDRESS + dmaplimit) {
10813 		pmap_large_map_flush_range(sva, len);
10814 	} else {
10815 		KASSERT(sva >= LARGEMAP_MIN_ADDRESS &&
10816 		    eva <= LARGEMAP_MIN_ADDRESS + lm_ents * NBPML4,
10817 		    ("pmap_large_map_wb: not largemap %#lx %#lx", sva, len));
10818 		pmap_large_map_wb_large(sva, eva);
10819 	}
10820 	pmap_large_map_wb_fence();
10821 }
10822 
10823 static vm_page_t
pmap_pti_alloc_page(void)10824 pmap_pti_alloc_page(void)
10825 {
10826 	vm_page_t m;
10827 
10828 	VM_OBJECT_ASSERT_WLOCKED(pti_obj);
10829 	m = vm_page_grab(pti_obj, pti_pg_idx++, VM_ALLOC_WIRED | VM_ALLOC_ZERO);
10830 	return (m);
10831 }
10832 
10833 static bool
pmap_pti_free_page(vm_page_t m)10834 pmap_pti_free_page(vm_page_t m)
10835 {
10836 	if (!vm_page_unwire_noq(m))
10837 		return (false);
10838 	vm_page_xbusy_claim(m);
10839 	vm_page_free_zero(m);
10840 	return (true);
10841 }
10842 
10843 static void
pmap_pti_init(void)10844 pmap_pti_init(void)
10845 {
10846 	vm_page_t pml4_pg;
10847 	pdp_entry_t *pdpe;
10848 	vm_offset_t va;
10849 	int i;
10850 
10851 	if (!pti)
10852 		return;
10853 	pti_obj = vm_pager_allocate(OBJT_PHYS, NULL, 0, VM_PROT_ALL, 0, NULL);
10854 	VM_OBJECT_WLOCK(pti_obj);
10855 	pml4_pg = pmap_pti_alloc_page();
10856 	pti_pml4 = (pml4_entry_t *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(pml4_pg));
10857 	for (va = VM_MIN_KERNEL_ADDRESS; va <= VM_MAX_KERNEL_ADDRESS &&
10858 	    va >= VM_MIN_KERNEL_ADDRESS && va > NBPML4; va += NBPML4) {
10859 		pdpe = pmap_pti_pdpe(va);
10860 		pmap_pti_wire_pte(pdpe);
10861 	}
10862 	pmap_pti_add_kva_locked((vm_offset_t)&__pcpu[0],
10863 	    (vm_offset_t)&__pcpu[0] + sizeof(__pcpu[0]) * MAXCPU, false);
10864 	pmap_pti_add_kva_locked((vm_offset_t)idt, (vm_offset_t)idt +
10865 	    sizeof(struct gate_descriptor) * NIDT, false);
10866 	CPU_FOREACH(i) {
10867 		/* Doublefault stack IST 1 */
10868 		va = __pcpu[i].pc_common_tss.tss_ist1 + sizeof(struct nmi_pcpu);
10869 		pmap_pti_add_kva_locked(va - DBLFAULT_STACK_SIZE, va, false);
10870 		/* NMI stack IST 2 */
10871 		va = __pcpu[i].pc_common_tss.tss_ist2 + sizeof(struct nmi_pcpu);
10872 		pmap_pti_add_kva_locked(va - NMI_STACK_SIZE, va, false);
10873 		/* MC# stack IST 3 */
10874 		va = __pcpu[i].pc_common_tss.tss_ist3 +
10875 		    sizeof(struct nmi_pcpu);
10876 		pmap_pti_add_kva_locked(va - MCE_STACK_SIZE, va, false);
10877 		/* DB# stack IST 4 */
10878 		va = __pcpu[i].pc_common_tss.tss_ist4 + sizeof(struct nmi_pcpu);
10879 		pmap_pti_add_kva_locked(va - DBG_STACK_SIZE, va, false);
10880 	}
10881 	pmap_pti_add_kva_locked((vm_offset_t)KERNSTART, (vm_offset_t)etext,
10882 	    true);
10883 	pti_finalized = true;
10884 	VM_OBJECT_WUNLOCK(pti_obj);
10885 }
10886 
10887 static void
pmap_cpu_init(void * arg __unused)10888 pmap_cpu_init(void *arg __unused)
10889 {
10890 	CPU_COPY(&all_cpus, &kernel_pmap->pm_active);
10891 	pmap_pti_init();
10892 }
10893 SYSINIT(pmap_cpu, SI_SUB_CPU + 1, SI_ORDER_ANY, pmap_cpu_init, NULL);
10894 
10895 static pdp_entry_t *
pmap_pti_pdpe(vm_offset_t va)10896 pmap_pti_pdpe(vm_offset_t va)
10897 {
10898 	pml4_entry_t *pml4e;
10899 	pdp_entry_t *pdpe;
10900 	vm_page_t m;
10901 	vm_pindex_t pml4_idx;
10902 	vm_paddr_t mphys;
10903 
10904 	VM_OBJECT_ASSERT_WLOCKED(pti_obj);
10905 
10906 	pml4_idx = pmap_pml4e_index(va);
10907 	pml4e = &pti_pml4[pml4_idx];
10908 	m = NULL;
10909 	if (*pml4e == 0) {
10910 		if (pti_finalized)
10911 			panic("pml4 alloc after finalization\n");
10912 		m = pmap_pti_alloc_page();
10913 		if (*pml4e != 0) {
10914 			pmap_pti_free_page(m);
10915 			mphys = *pml4e & ~PAGE_MASK;
10916 		} else {
10917 			mphys = VM_PAGE_TO_PHYS(m);
10918 			*pml4e = mphys | X86_PG_RW | X86_PG_V;
10919 		}
10920 	} else {
10921 		mphys = *pml4e & ~PAGE_MASK;
10922 	}
10923 	pdpe = (pdp_entry_t *)PHYS_TO_DMAP(mphys) + pmap_pdpe_index(va);
10924 	return (pdpe);
10925 }
10926 
10927 static void
pmap_pti_wire_pte(void * pte)10928 pmap_pti_wire_pte(void *pte)
10929 {
10930 	vm_page_t m;
10931 
10932 	VM_OBJECT_ASSERT_WLOCKED(pti_obj);
10933 	m = PHYS_TO_VM_PAGE(DMAP_TO_PHYS((uintptr_t)pte));
10934 	m->ref_count++;
10935 }
10936 
10937 static void
pmap_pti_unwire_pde(void * pde,bool only_ref)10938 pmap_pti_unwire_pde(void *pde, bool only_ref)
10939 {
10940 	vm_page_t m;
10941 
10942 	VM_OBJECT_ASSERT_WLOCKED(pti_obj);
10943 	m = PHYS_TO_VM_PAGE(DMAP_TO_PHYS((uintptr_t)pde));
10944 	MPASS(only_ref || m->ref_count > 1);
10945 	pmap_pti_free_page(m);
10946 }
10947 
10948 static void
pmap_pti_unwire_pte(void * pte,vm_offset_t va)10949 pmap_pti_unwire_pte(void *pte, vm_offset_t va)
10950 {
10951 	vm_page_t m;
10952 	pd_entry_t *pde;
10953 
10954 	VM_OBJECT_ASSERT_WLOCKED(pti_obj);
10955 	m = PHYS_TO_VM_PAGE(DMAP_TO_PHYS((uintptr_t)pte));
10956 	if (pmap_pti_free_page(m)) {
10957 		pde = pmap_pti_pde(va);
10958 		MPASS((*pde & (X86_PG_PS | X86_PG_V)) == X86_PG_V);
10959 		*pde = 0;
10960 		pmap_pti_unwire_pde(pde, false);
10961 	}
10962 }
10963 
10964 static pd_entry_t *
pmap_pti_pde(vm_offset_t va)10965 pmap_pti_pde(vm_offset_t va)
10966 {
10967 	pdp_entry_t *pdpe;
10968 	pd_entry_t *pde;
10969 	vm_page_t m;
10970 	vm_pindex_t pd_idx;
10971 	vm_paddr_t mphys;
10972 
10973 	VM_OBJECT_ASSERT_WLOCKED(pti_obj);
10974 
10975 	pdpe = pmap_pti_pdpe(va);
10976 	if (*pdpe == 0) {
10977 		m = pmap_pti_alloc_page();
10978 		if (*pdpe != 0) {
10979 			pmap_pti_free_page(m);
10980 			MPASS((*pdpe & X86_PG_PS) == 0);
10981 			mphys = *pdpe & ~PAGE_MASK;
10982 		} else {
10983 			mphys =  VM_PAGE_TO_PHYS(m);
10984 			*pdpe = mphys | X86_PG_RW | X86_PG_V;
10985 		}
10986 	} else {
10987 		MPASS((*pdpe & X86_PG_PS) == 0);
10988 		mphys = *pdpe & ~PAGE_MASK;
10989 	}
10990 
10991 	pde = (pd_entry_t *)PHYS_TO_DMAP(mphys);
10992 	pd_idx = pmap_pde_index(va);
10993 	pde += pd_idx;
10994 	return (pde);
10995 }
10996 
10997 static pt_entry_t *
pmap_pti_pte(vm_offset_t va,bool * unwire_pde)10998 pmap_pti_pte(vm_offset_t va, bool *unwire_pde)
10999 {
11000 	pd_entry_t *pde;
11001 	pt_entry_t *pte;
11002 	vm_page_t m;
11003 	vm_paddr_t mphys;
11004 
11005 	VM_OBJECT_ASSERT_WLOCKED(pti_obj);
11006 
11007 	pde = pmap_pti_pde(va);
11008 	if (unwire_pde != NULL) {
11009 		*unwire_pde = true;
11010 		pmap_pti_wire_pte(pde);
11011 	}
11012 	if (*pde == 0) {
11013 		m = pmap_pti_alloc_page();
11014 		if (*pde != 0) {
11015 			pmap_pti_free_page(m);
11016 			MPASS((*pde & X86_PG_PS) == 0);
11017 			mphys = *pde & ~(PAGE_MASK | pg_nx);
11018 		} else {
11019 			mphys = VM_PAGE_TO_PHYS(m);
11020 			*pde = mphys | X86_PG_RW | X86_PG_V;
11021 			if (unwire_pde != NULL)
11022 				*unwire_pde = false;
11023 		}
11024 	} else {
11025 		MPASS((*pde & X86_PG_PS) == 0);
11026 		mphys = *pde & ~(PAGE_MASK | pg_nx);
11027 	}
11028 
11029 	pte = (pt_entry_t *)PHYS_TO_DMAP(mphys);
11030 	pte += pmap_pte_index(va);
11031 
11032 	return (pte);
11033 }
11034 
11035 static void
pmap_pti_add_kva_locked(vm_offset_t sva,vm_offset_t eva,bool exec)11036 pmap_pti_add_kva_locked(vm_offset_t sva, vm_offset_t eva, bool exec)
11037 {
11038 	vm_paddr_t pa;
11039 	pd_entry_t *pde;
11040 	pt_entry_t *pte, ptev;
11041 	bool unwire_pde;
11042 
11043 	VM_OBJECT_ASSERT_WLOCKED(pti_obj);
11044 
11045 	sva = trunc_page(sva);
11046 	MPASS(sva > VM_MAXUSER_ADDRESS);
11047 	eva = round_page(eva);
11048 	MPASS(sva < eva);
11049 	for (; sva < eva; sva += PAGE_SIZE) {
11050 		pte = pmap_pti_pte(sva, &unwire_pde);
11051 		pa = pmap_kextract(sva);
11052 		ptev = pa | X86_PG_RW | X86_PG_V | X86_PG_A | X86_PG_G |
11053 		    (exec ? 0 : pg_nx) | pmap_cache_bits(kernel_pmap,
11054 		    VM_MEMATTR_DEFAULT, FALSE);
11055 		if (*pte == 0) {
11056 			pte_store(pte, ptev);
11057 			pmap_pti_wire_pte(pte);
11058 		} else {
11059 			KASSERT(!pti_finalized,
11060 			    ("pti overlap after fin %#lx %#lx %#lx",
11061 			    sva, *pte, ptev));
11062 			KASSERT(*pte == ptev,
11063 			    ("pti non-identical pte after fin %#lx %#lx %#lx",
11064 			    sva, *pte, ptev));
11065 		}
11066 		if (unwire_pde) {
11067 			pde = pmap_pti_pde(sva);
11068 			pmap_pti_unwire_pde(pde, true);
11069 		}
11070 	}
11071 }
11072 
11073 void
pmap_pti_add_kva(vm_offset_t sva,vm_offset_t eva,bool exec)11074 pmap_pti_add_kva(vm_offset_t sva, vm_offset_t eva, bool exec)
11075 {
11076 
11077 	if (!pti)
11078 		return;
11079 	VM_OBJECT_WLOCK(pti_obj);
11080 	pmap_pti_add_kva_locked(sva, eva, exec);
11081 	VM_OBJECT_WUNLOCK(pti_obj);
11082 }
11083 
11084 void
pmap_pti_remove_kva(vm_offset_t sva,vm_offset_t eva)11085 pmap_pti_remove_kva(vm_offset_t sva, vm_offset_t eva)
11086 {
11087 	pt_entry_t *pte;
11088 	vm_offset_t va;
11089 
11090 	if (!pti)
11091 		return;
11092 	sva = rounddown2(sva, PAGE_SIZE);
11093 	MPASS(sva > VM_MAXUSER_ADDRESS);
11094 	eva = roundup2(eva, PAGE_SIZE);
11095 	MPASS(sva < eva);
11096 	VM_OBJECT_WLOCK(pti_obj);
11097 	for (va = sva; va < eva; va += PAGE_SIZE) {
11098 		pte = pmap_pti_pte(va, NULL);
11099 		KASSERT((*pte & X86_PG_V) != 0,
11100 		    ("invalid pte va %#lx pte %#lx pt %#lx", va,
11101 		    (u_long)pte, *pte));
11102 		pte_clear(pte);
11103 		pmap_pti_unwire_pte(pte, va);
11104 	}
11105 	pmap_invalidate_range(kernel_pmap, sva, eva);
11106 	VM_OBJECT_WUNLOCK(pti_obj);
11107 }
11108 
11109 static void *
pkru_dup_range(void * ctx __unused,void * data)11110 pkru_dup_range(void *ctx __unused, void *data)
11111 {
11112 	struct pmap_pkru_range *node, *new_node;
11113 
11114 	new_node = uma_zalloc(pmap_pkru_ranges_zone, M_NOWAIT);
11115 	if (new_node == NULL)
11116 		return (NULL);
11117 	node = data;
11118 	memcpy(new_node, node, sizeof(*node));
11119 	return (new_node);
11120 }
11121 
11122 static void
pkru_free_range(void * ctx __unused,void * node)11123 pkru_free_range(void *ctx __unused, void *node)
11124 {
11125 
11126 	uma_zfree(pmap_pkru_ranges_zone, node);
11127 }
11128 
11129 static int
pmap_pkru_assign(pmap_t pmap,vm_offset_t sva,vm_offset_t eva,u_int keyidx,int flags)11130 pmap_pkru_assign(pmap_t pmap, vm_offset_t sva, vm_offset_t eva, u_int keyidx,
11131     int flags)
11132 {
11133 	struct pmap_pkru_range *ppr;
11134 	int error;
11135 
11136 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
11137 	MPASS(pmap->pm_type == PT_X86);
11138 	MPASS((cpu_stdext_feature2 & CPUID_STDEXT2_PKU) != 0);
11139 	if ((flags & AMD64_PKRU_EXCL) != 0 &&
11140 	    !rangeset_check_empty(&pmap->pm_pkru, sva, eva))
11141 		return (EBUSY);
11142 	ppr = uma_zalloc(pmap_pkru_ranges_zone, M_NOWAIT);
11143 	if (ppr == NULL)
11144 		return (ENOMEM);
11145 	ppr->pkru_keyidx = keyidx;
11146 	ppr->pkru_flags = flags & AMD64_PKRU_PERSIST;
11147 	error = rangeset_insert(&pmap->pm_pkru, sva, eva, ppr);
11148 	if (error != 0)
11149 		uma_zfree(pmap_pkru_ranges_zone, ppr);
11150 	return (error);
11151 }
11152 
11153 static int
pmap_pkru_deassign(pmap_t pmap,vm_offset_t sva,vm_offset_t eva)11154 pmap_pkru_deassign(pmap_t pmap, vm_offset_t sva, vm_offset_t eva)
11155 {
11156 
11157 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
11158 	MPASS(pmap->pm_type == PT_X86);
11159 	MPASS((cpu_stdext_feature2 & CPUID_STDEXT2_PKU) != 0);
11160 	return (rangeset_remove(&pmap->pm_pkru, sva, eva));
11161 }
11162 
11163 static void
pmap_pkru_deassign_all(pmap_t pmap)11164 pmap_pkru_deassign_all(pmap_t pmap)
11165 {
11166 
11167 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
11168 	if (pmap->pm_type == PT_X86 &&
11169 	    (cpu_stdext_feature2 & CPUID_STDEXT2_PKU) != 0)
11170 		rangeset_remove_all(&pmap->pm_pkru);
11171 }
11172 
11173 static bool
pmap_pkru_same(pmap_t pmap,vm_offset_t sva,vm_offset_t eva)11174 pmap_pkru_same(pmap_t pmap, vm_offset_t sva, vm_offset_t eva)
11175 {
11176 	struct pmap_pkru_range *ppr, *prev_ppr;
11177 	vm_offset_t va;
11178 
11179 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
11180 	if (pmap->pm_type != PT_X86 ||
11181 	    (cpu_stdext_feature2 & CPUID_STDEXT2_PKU) == 0 ||
11182 	    sva >= VM_MAXUSER_ADDRESS)
11183 		return (true);
11184 	MPASS(eva <= VM_MAXUSER_ADDRESS);
11185 	for (va = sva; va < eva; prev_ppr = ppr) {
11186 		ppr = rangeset_lookup(&pmap->pm_pkru, va);
11187 		if (va == sva)
11188 			prev_ppr = ppr;
11189 		else if ((ppr == NULL) ^ (prev_ppr == NULL))
11190 			return (false);
11191 		if (ppr == NULL) {
11192 			va += PAGE_SIZE;
11193 			continue;
11194 		}
11195 		if (prev_ppr->pkru_keyidx != ppr->pkru_keyidx)
11196 			return (false);
11197 		va = ppr->pkru_rs_el.re_end;
11198 	}
11199 	return (true);
11200 }
11201 
11202 static pt_entry_t
pmap_pkru_get(pmap_t pmap,vm_offset_t va)11203 pmap_pkru_get(pmap_t pmap, vm_offset_t va)
11204 {
11205 	struct pmap_pkru_range *ppr;
11206 
11207 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
11208 	if (pmap->pm_type != PT_X86 ||
11209 	    (cpu_stdext_feature2 & CPUID_STDEXT2_PKU) == 0 ||
11210 	    va >= VM_MAXUSER_ADDRESS)
11211 		return (0);
11212 	ppr = rangeset_lookup(&pmap->pm_pkru, va);
11213 	if (ppr != NULL)
11214 		return (X86_PG_PKU(ppr->pkru_keyidx));
11215 	return (0);
11216 }
11217 
11218 static bool
pred_pkru_on_remove(void * ctx __unused,void * r)11219 pred_pkru_on_remove(void *ctx __unused, void *r)
11220 {
11221 	struct pmap_pkru_range *ppr;
11222 
11223 	ppr = r;
11224 	return ((ppr->pkru_flags & AMD64_PKRU_PERSIST) == 0);
11225 }
11226 
11227 static void
pmap_pkru_on_remove(pmap_t pmap,vm_offset_t sva,vm_offset_t eva)11228 pmap_pkru_on_remove(pmap_t pmap, vm_offset_t sva, vm_offset_t eva)
11229 {
11230 
11231 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
11232 	if (pmap->pm_type == PT_X86 &&
11233 	    (cpu_stdext_feature2 & CPUID_STDEXT2_PKU) != 0) {
11234 		rangeset_remove_pred(&pmap->pm_pkru, sva, eva,
11235 		    pred_pkru_on_remove);
11236 	}
11237 }
11238 
11239 static int
pmap_pkru_copy(pmap_t dst_pmap,pmap_t src_pmap)11240 pmap_pkru_copy(pmap_t dst_pmap, pmap_t src_pmap)
11241 {
11242 
11243 	PMAP_LOCK_ASSERT(dst_pmap, MA_OWNED);
11244 	PMAP_LOCK_ASSERT(src_pmap, MA_OWNED);
11245 	MPASS(dst_pmap->pm_type == PT_X86);
11246 	MPASS(src_pmap->pm_type == PT_X86);
11247 	MPASS((cpu_stdext_feature2 & CPUID_STDEXT2_PKU) != 0);
11248 	if (src_pmap->pm_pkru.rs_data_ctx == NULL)
11249 		return (0);
11250 	return (rangeset_copy(&dst_pmap->pm_pkru, &src_pmap->pm_pkru));
11251 }
11252 
11253 static void
pmap_pkru_update_range(pmap_t pmap,vm_offset_t sva,vm_offset_t eva,u_int keyidx)11254 pmap_pkru_update_range(pmap_t pmap, vm_offset_t sva, vm_offset_t eva,
11255     u_int keyidx)
11256 {
11257 	pml4_entry_t *pml4e;
11258 	pdp_entry_t *pdpe;
11259 	pd_entry_t newpde, ptpaddr, *pde;
11260 	pt_entry_t newpte, *ptep, pte;
11261 	vm_offset_t va, va_next;
11262 	bool changed;
11263 
11264 	PMAP_LOCK_ASSERT(pmap, MA_OWNED);
11265 	MPASS(pmap->pm_type == PT_X86);
11266 	MPASS(keyidx <= PMAP_MAX_PKRU_IDX);
11267 
11268 	for (changed = false, va = sva; va < eva; va = va_next) {
11269 		pml4e = pmap_pml4e(pmap, va);
11270 		if (pml4e == NULL || (*pml4e & X86_PG_V) == 0) {
11271 			va_next = (va + NBPML4) & ~PML4MASK;
11272 			if (va_next < va)
11273 				va_next = eva;
11274 			continue;
11275 		}
11276 
11277 		pdpe = pmap_pml4e_to_pdpe(pml4e, va);
11278 		if ((*pdpe & X86_PG_V) == 0) {
11279 			va_next = (va + NBPDP) & ~PDPMASK;
11280 			if (va_next < va)
11281 				va_next = eva;
11282 			continue;
11283 		}
11284 
11285 		va_next = (va + NBPDR) & ~PDRMASK;
11286 		if (va_next < va)
11287 			va_next = eva;
11288 
11289 		pde = pmap_pdpe_to_pde(pdpe, va);
11290 		ptpaddr = *pde;
11291 		if (ptpaddr == 0)
11292 			continue;
11293 
11294 		MPASS((ptpaddr & X86_PG_V) != 0);
11295 		if ((ptpaddr & PG_PS) != 0) {
11296 			if (va + NBPDR == va_next && eva >= va_next) {
11297 				newpde = (ptpaddr & ~X86_PG_PKU_MASK) |
11298 				    X86_PG_PKU(keyidx);
11299 				if (newpde != ptpaddr) {
11300 					*pde = newpde;
11301 					changed = true;
11302 				}
11303 				continue;
11304 			} else if (!pmap_demote_pde(pmap, pde, va)) {
11305 				continue;
11306 			}
11307 		}
11308 
11309 		if (va_next > eva)
11310 			va_next = eva;
11311 
11312 		for (ptep = pmap_pde_to_pte(pde, va); va != va_next;
11313 		    ptep++, va += PAGE_SIZE) {
11314 			pte = *ptep;
11315 			if ((pte & X86_PG_V) == 0)
11316 				continue;
11317 			newpte = (pte & ~X86_PG_PKU_MASK) | X86_PG_PKU(keyidx);
11318 			if (newpte != pte) {
11319 				*ptep = newpte;
11320 				changed = true;
11321 			}
11322 		}
11323 	}
11324 	if (changed)
11325 		pmap_invalidate_range(pmap, sva, eva);
11326 }
11327 
11328 static int
pmap_pkru_check_uargs(pmap_t pmap,vm_offset_t sva,vm_offset_t eva,u_int keyidx,int flags)11329 pmap_pkru_check_uargs(pmap_t pmap, vm_offset_t sva, vm_offset_t eva,
11330     u_int keyidx, int flags)
11331 {
11332 
11333 	if (pmap->pm_type != PT_X86 || keyidx > PMAP_MAX_PKRU_IDX ||
11334 	    (flags & ~(AMD64_PKRU_PERSIST | AMD64_PKRU_EXCL)) != 0)
11335 		return (EINVAL);
11336 	if (eva <= sva || eva > VM_MAXUSER_ADDRESS)
11337 		return (EFAULT);
11338 	if ((cpu_stdext_feature2 & CPUID_STDEXT2_PKU) == 0)
11339 		return (ENOTSUP);
11340 	return (0);
11341 }
11342 
11343 int
pmap_pkru_set(pmap_t pmap,vm_offset_t sva,vm_offset_t eva,u_int keyidx,int flags)11344 pmap_pkru_set(pmap_t pmap, vm_offset_t sva, vm_offset_t eva, u_int keyidx,
11345     int flags)
11346 {
11347 	int error;
11348 
11349 	sva = trunc_page(sva);
11350 	eva = round_page(eva);
11351 	error = pmap_pkru_check_uargs(pmap, sva, eva, keyidx, flags);
11352 	if (error != 0)
11353 		return (error);
11354 	for (;;) {
11355 		PMAP_LOCK(pmap);
11356 		error = pmap_pkru_assign(pmap, sva, eva, keyidx, flags);
11357 		if (error == 0)
11358 			pmap_pkru_update_range(pmap, sva, eva, keyidx);
11359 		PMAP_UNLOCK(pmap);
11360 		if (error != ENOMEM)
11361 			break;
11362 		vm_wait(NULL);
11363 	}
11364 	return (error);
11365 }
11366 
11367 int
pmap_pkru_clear(pmap_t pmap,vm_offset_t sva,vm_offset_t eva)11368 pmap_pkru_clear(pmap_t pmap, vm_offset_t sva, vm_offset_t eva)
11369 {
11370 	int error;
11371 
11372 	sva = trunc_page(sva);
11373 	eva = round_page(eva);
11374 	error = pmap_pkru_check_uargs(pmap, sva, eva, 0, 0);
11375 	if (error != 0)
11376 		return (error);
11377 	for (;;) {
11378 		PMAP_LOCK(pmap);
11379 		error = pmap_pkru_deassign(pmap, sva, eva);
11380 		if (error == 0)
11381 			pmap_pkru_update_range(pmap, sva, eva, 0);
11382 		PMAP_UNLOCK(pmap);
11383 		if (error != ENOMEM)
11384 			break;
11385 		vm_wait(NULL);
11386 	}
11387 	return (error);
11388 }
11389 
11390 #ifdef KASAN
11391 static vm_page_t
pmap_kasan_enter_alloc_4k(void)11392 pmap_kasan_enter_alloc_4k(void)
11393 {
11394 	vm_page_t m;
11395 
11396 	m = vm_page_alloc_noobj(VM_ALLOC_INTERRUPT | VM_ALLOC_WIRED |
11397 	    VM_ALLOC_ZERO);
11398 	if (m == NULL)
11399 		panic("%s: no memory to grow shadow map", __func__);
11400 	return (m);
11401 }
11402 
11403 static vm_page_t
pmap_kasan_enter_alloc_2m(void)11404 pmap_kasan_enter_alloc_2m(void)
11405 {
11406 	return (vm_page_alloc_noobj_contig(VM_ALLOC_WIRED | VM_ALLOC_ZERO,
11407 	    NPTEPG, 0, ~0ul, NBPDR, 0, VM_MEMATTR_DEFAULT));
11408 }
11409 
11410 /*
11411  * Grow the shadow map by at least one 4KB page at the specified address.  Use
11412  * 2MB pages when possible.
11413  */
11414 void
pmap_kasan_enter(vm_offset_t va)11415 pmap_kasan_enter(vm_offset_t va)
11416 {
11417 	pdp_entry_t *pdpe;
11418 	pd_entry_t *pde;
11419 	pt_entry_t *pte;
11420 	vm_page_t m;
11421 
11422 	mtx_assert(&kernel_map->system_mtx, MA_OWNED);
11423 
11424 	pdpe = pmap_pdpe(kernel_pmap, va);
11425 	if ((*pdpe & X86_PG_V) == 0) {
11426 		m = pmap_kasan_enter_alloc_4k();
11427 		*pdpe = (pdp_entry_t)(VM_PAGE_TO_PHYS(m) | X86_PG_RW |
11428 		    X86_PG_V | pg_nx);
11429 	}
11430 	pde = pmap_pdpe_to_pde(pdpe, va);
11431 	if ((*pde & X86_PG_V) == 0) {
11432 		m = pmap_kasan_enter_alloc_2m();
11433 		if (m != NULL) {
11434 			*pde = (pd_entry_t)(VM_PAGE_TO_PHYS(m) | X86_PG_RW |
11435 			    X86_PG_PS | X86_PG_V | X86_PG_A | X86_PG_M | pg_nx);
11436 		} else {
11437 			m = pmap_kasan_enter_alloc_4k();
11438 			*pde = (pd_entry_t)(VM_PAGE_TO_PHYS(m) | X86_PG_RW |
11439 			    X86_PG_V | pg_nx);
11440 		}
11441 	}
11442 	if ((*pde & X86_PG_PS) != 0)
11443 		return;
11444 	pte = pmap_pde_to_pte(pde, va);
11445 	if ((*pte & X86_PG_V) != 0)
11446 		return;
11447 	m = pmap_kasan_enter_alloc_4k();
11448 	*pte = (pt_entry_t)(VM_PAGE_TO_PHYS(m) | X86_PG_RW | X86_PG_V |
11449 	    X86_PG_M | X86_PG_A | pg_nx);
11450 }
11451 #endif
11452 
11453 #ifdef KMSAN
11454 static vm_page_t
pmap_kmsan_enter_alloc_4k(void)11455 pmap_kmsan_enter_alloc_4k(void)
11456 {
11457 	vm_page_t m;
11458 
11459 	m = vm_page_alloc_noobj(VM_ALLOC_INTERRUPT | VM_ALLOC_WIRED |
11460 	    VM_ALLOC_ZERO);
11461 	if (m == NULL)
11462 		panic("%s: no memory to grow shadow map", __func__);
11463 	return (m);
11464 }
11465 
11466 static vm_page_t
pmap_kmsan_enter_alloc_2m(void)11467 pmap_kmsan_enter_alloc_2m(void)
11468 {
11469 	return (vm_page_alloc_noobj_contig(VM_ALLOC_ZERO | VM_ALLOC_WIRED,
11470 	    NPTEPG, 0, ~0ul, NBPDR, 0, VM_MEMATTR_DEFAULT));
11471 }
11472 
11473 /*
11474  * Grow the shadow or origin maps by at least one 4KB page at the specified
11475  * address.  Use 2MB pages when possible.
11476  */
11477 void
pmap_kmsan_enter(vm_offset_t va)11478 pmap_kmsan_enter(vm_offset_t va)
11479 {
11480 	pdp_entry_t *pdpe;
11481 	pd_entry_t *pde;
11482 	pt_entry_t *pte;
11483 	vm_page_t m;
11484 
11485 	mtx_assert(&kernel_map->system_mtx, MA_OWNED);
11486 
11487 	pdpe = pmap_pdpe(kernel_pmap, va);
11488 	if ((*pdpe & X86_PG_V) == 0) {
11489 		m = pmap_kmsan_enter_alloc_4k();
11490 		*pdpe = (pdp_entry_t)(VM_PAGE_TO_PHYS(m) | X86_PG_RW |
11491 		    X86_PG_V | pg_nx);
11492 	}
11493 	pde = pmap_pdpe_to_pde(pdpe, va);
11494 	if ((*pde & X86_PG_V) == 0) {
11495 		m = pmap_kmsan_enter_alloc_2m();
11496 		if (m != NULL) {
11497 			*pde = (pd_entry_t)(VM_PAGE_TO_PHYS(m) | X86_PG_RW |
11498 			    X86_PG_PS | X86_PG_V | X86_PG_A | X86_PG_M | pg_nx);
11499 		} else {
11500 			m = pmap_kmsan_enter_alloc_4k();
11501 			*pde = (pd_entry_t)(VM_PAGE_TO_PHYS(m) | X86_PG_RW |
11502 			    X86_PG_V | pg_nx);
11503 		}
11504 	}
11505 	if ((*pde & X86_PG_PS) != 0)
11506 		return;
11507 	pte = pmap_pde_to_pte(pde, va);
11508 	if ((*pte & X86_PG_V) != 0)
11509 		return;
11510 	m = pmap_kmsan_enter_alloc_4k();
11511 	*pte = (pt_entry_t)(VM_PAGE_TO_PHYS(m) | X86_PG_RW | X86_PG_V |
11512 	    X86_PG_M | X86_PG_A | pg_nx);
11513 }
11514 #endif
11515 
11516 /*
11517  * Track a range of the kernel's virtual address space that is contiguous
11518  * in various mapping attributes.
11519  */
11520 struct pmap_kernel_map_range {
11521 	vm_offset_t sva;
11522 	pt_entry_t attrs;
11523 	int ptes;
11524 	int pdes;
11525 	int pdpes;
11526 };
11527 
11528 static void
sysctl_kmaps_dump(struct sbuf * sb,struct pmap_kernel_map_range * range,vm_offset_t eva)11529 sysctl_kmaps_dump(struct sbuf *sb, struct pmap_kernel_map_range *range,
11530     vm_offset_t eva)
11531 {
11532 	const char *mode;
11533 	int i, pat_idx;
11534 
11535 	if (eva <= range->sva)
11536 		return;
11537 
11538 	pat_idx = pmap_pat_index(kernel_pmap, range->attrs, true);
11539 	for (i = 0; i < PAT_INDEX_SIZE; i++)
11540 		if (pat_index[i] == pat_idx)
11541 			break;
11542 
11543 	switch (i) {
11544 	case PAT_WRITE_BACK:
11545 		mode = "WB";
11546 		break;
11547 	case PAT_WRITE_THROUGH:
11548 		mode = "WT";
11549 		break;
11550 	case PAT_UNCACHEABLE:
11551 		mode = "UC";
11552 		break;
11553 	case PAT_UNCACHED:
11554 		mode = "U-";
11555 		break;
11556 	case PAT_WRITE_PROTECTED:
11557 		mode = "WP";
11558 		break;
11559 	case PAT_WRITE_COMBINING:
11560 		mode = "WC";
11561 		break;
11562 	default:
11563 		printf("%s: unknown PAT mode %#x for range 0x%016lx-0x%016lx\n",
11564 		    __func__, pat_idx, range->sva, eva);
11565 		mode = "??";
11566 		break;
11567 	}
11568 
11569 	sbuf_printf(sb, "0x%016lx-0x%016lx r%c%c%c%c %s %d %d %d\n",
11570 	    range->sva, eva,
11571 	    (range->attrs & X86_PG_RW) != 0 ? 'w' : '-',
11572 	    (range->attrs & pg_nx) != 0 ? '-' : 'x',
11573 	    (range->attrs & X86_PG_U) != 0 ? 'u' : 's',
11574 	    (range->attrs & X86_PG_G) != 0 ? 'g' : '-',
11575 	    mode, range->pdpes, range->pdes, range->ptes);
11576 
11577 	/* Reset to sentinel value. */
11578 	range->sva = la57 ? KV5ADDR(NPML5EPG - 1, NPML4EPG - 1, NPDPEPG - 1,
11579 	    NPDEPG - 1, NPTEPG - 1) : KV4ADDR(NPML4EPG - 1, NPDPEPG - 1,
11580 	    NPDEPG - 1, NPTEPG - 1);
11581 }
11582 
11583 /*
11584  * Determine whether the attributes specified by a page table entry match those
11585  * being tracked by the current range.  This is not quite as simple as a direct
11586  * flag comparison since some PAT modes have multiple representations.
11587  */
11588 static bool
sysctl_kmaps_match(struct pmap_kernel_map_range * range,pt_entry_t attrs)11589 sysctl_kmaps_match(struct pmap_kernel_map_range *range, pt_entry_t attrs)
11590 {
11591 	pt_entry_t diff, mask;
11592 
11593 	mask = X86_PG_G | X86_PG_RW | X86_PG_U | X86_PG_PDE_CACHE | pg_nx;
11594 	diff = (range->attrs ^ attrs) & mask;
11595 	if (diff == 0)
11596 		return (true);
11597 	if ((diff & ~X86_PG_PDE_PAT) == 0 &&
11598 	    pmap_pat_index(kernel_pmap, range->attrs, true) ==
11599 	    pmap_pat_index(kernel_pmap, attrs, true))
11600 		return (true);
11601 	return (false);
11602 }
11603 
11604 static void
sysctl_kmaps_reinit(struct pmap_kernel_map_range * range,vm_offset_t va,pt_entry_t attrs)11605 sysctl_kmaps_reinit(struct pmap_kernel_map_range *range, vm_offset_t va,
11606     pt_entry_t attrs)
11607 {
11608 
11609 	memset(range, 0, sizeof(*range));
11610 	range->sva = va;
11611 	range->attrs = attrs;
11612 }
11613 
11614 /*
11615  * Given a leaf PTE, derive the mapping's attributes.  If they do not match
11616  * those of the current run, dump the address range and its attributes, and
11617  * begin a new run.
11618  */
11619 static void
sysctl_kmaps_check(struct sbuf * sb,struct pmap_kernel_map_range * range,vm_offset_t va,pml4_entry_t pml4e,pdp_entry_t pdpe,pd_entry_t pde,pt_entry_t pte)11620 sysctl_kmaps_check(struct sbuf *sb, struct pmap_kernel_map_range *range,
11621     vm_offset_t va, pml4_entry_t pml4e, pdp_entry_t pdpe, pd_entry_t pde,
11622     pt_entry_t pte)
11623 {
11624 	pt_entry_t attrs;
11625 
11626 	attrs = pml4e & (X86_PG_RW | X86_PG_U | pg_nx);
11627 
11628 	attrs |= pdpe & pg_nx;
11629 	attrs &= pg_nx | (pdpe & (X86_PG_RW | X86_PG_U));
11630 	if ((pdpe & PG_PS) != 0) {
11631 		attrs |= pdpe & (X86_PG_G | X86_PG_PDE_CACHE);
11632 	} else if (pde != 0) {
11633 		attrs |= pde & pg_nx;
11634 		attrs &= pg_nx | (pde & (X86_PG_RW | X86_PG_U));
11635 	}
11636 	if ((pde & PG_PS) != 0) {
11637 		attrs |= pde & (X86_PG_G | X86_PG_PDE_CACHE);
11638 	} else if (pte != 0) {
11639 		attrs |= pte & pg_nx;
11640 		attrs &= pg_nx | (pte & (X86_PG_RW | X86_PG_U));
11641 		attrs |= pte & (X86_PG_G | X86_PG_PTE_CACHE);
11642 
11643 		/* Canonicalize by always using the PDE PAT bit. */
11644 		if ((attrs & X86_PG_PTE_PAT) != 0)
11645 			attrs ^= X86_PG_PDE_PAT | X86_PG_PTE_PAT;
11646 	}
11647 
11648 	if (range->sva > va || !sysctl_kmaps_match(range, attrs)) {
11649 		sysctl_kmaps_dump(sb, range, va);
11650 		sysctl_kmaps_reinit(range, va, attrs);
11651 	}
11652 }
11653 
11654 static int
sysctl_kmaps(SYSCTL_HANDLER_ARGS)11655 sysctl_kmaps(SYSCTL_HANDLER_ARGS)
11656 {
11657 	struct pmap_kernel_map_range range;
11658 	struct sbuf sbuf, *sb;
11659 	pml4_entry_t pml4e;
11660 	pdp_entry_t *pdp, pdpe;
11661 	pd_entry_t *pd, pde;
11662 	pt_entry_t *pt, pte;
11663 	vm_offset_t sva;
11664 	vm_paddr_t pa;
11665 	int error, i, j, k, l;
11666 
11667 	error = sysctl_wire_old_buffer(req, 0);
11668 	if (error != 0)
11669 		return (error);
11670 	sb = &sbuf;
11671 	sbuf_new_for_sysctl(sb, NULL, PAGE_SIZE, req);
11672 
11673 	/* Sentinel value. */
11674 	range.sva = la57 ? KV5ADDR(NPML5EPG - 1, NPML4EPG - 1, NPDPEPG - 1,
11675 	    NPDEPG - 1, NPTEPG - 1) : KV4ADDR(NPML4EPG - 1, NPDPEPG - 1,
11676 	    NPDEPG - 1, NPTEPG - 1);
11677 
11678 	/*
11679 	 * Iterate over the kernel page tables without holding the kernel pmap
11680 	 * lock.  Outside of the large map, kernel page table pages are never
11681 	 * freed, so at worst we will observe inconsistencies in the output.
11682 	 * Within the large map, ensure that PDP and PD page addresses are
11683 	 * valid before descending.
11684 	 */
11685 	for (sva = 0, i = pmap_pml4e_index(sva); i < NPML4EPG; i++) {
11686 		switch (i) {
11687 		case PML4PML4I:
11688 			sbuf_printf(sb, "\nRecursive map:\n");
11689 			break;
11690 		case DMPML4I:
11691 			sbuf_printf(sb, "\nDirect map:\n");
11692 			break;
11693 #ifdef KASAN
11694 		case KASANPML4I:
11695 			sbuf_printf(sb, "\nKASAN shadow map:\n");
11696 			break;
11697 #endif
11698 #ifdef KMSAN
11699 		case KMSANSHADPML4I:
11700 			sbuf_printf(sb, "\nKMSAN shadow map:\n");
11701 			break;
11702 		case KMSANORIGPML4I:
11703 			sbuf_printf(sb, "\nKMSAN origin map:\n");
11704 			break;
11705 #endif
11706 		case KPML4BASE:
11707 			sbuf_printf(sb, "\nKernel map:\n");
11708 			break;
11709 		case LMSPML4I:
11710 			sbuf_printf(sb, "\nLarge map:\n");
11711 			break;
11712 		}
11713 
11714 		/* Convert to canonical form. */
11715 		if (sva == 1ul << 47)
11716 			sva |= -1ul << 48;
11717 
11718 restart:
11719 		pml4e = kernel_pml4[i];
11720 		if ((pml4e & X86_PG_V) == 0) {
11721 			sva = rounddown2(sva, NBPML4);
11722 			sysctl_kmaps_dump(sb, &range, sva);
11723 			sva += NBPML4;
11724 			continue;
11725 		}
11726 		pa = pml4e & PG_FRAME;
11727 		pdp = (pdp_entry_t *)PHYS_TO_DMAP(pa);
11728 
11729 		for (j = pmap_pdpe_index(sva); j < NPDPEPG; j++) {
11730 			pdpe = pdp[j];
11731 			if ((pdpe & X86_PG_V) == 0) {
11732 				sva = rounddown2(sva, NBPDP);
11733 				sysctl_kmaps_dump(sb, &range, sva);
11734 				sva += NBPDP;
11735 				continue;
11736 			}
11737 			pa = pdpe & PG_FRAME;
11738 			if ((pdpe & PG_PS) != 0) {
11739 				sva = rounddown2(sva, NBPDP);
11740 				sysctl_kmaps_check(sb, &range, sva, pml4e, pdpe,
11741 				    0, 0);
11742 				range.pdpes++;
11743 				sva += NBPDP;
11744 				continue;
11745 			}
11746 			if (PMAP_ADDRESS_IN_LARGEMAP(sva) &&
11747 			    vm_phys_paddr_to_vm_page(pa) == NULL) {
11748 				/*
11749 				 * Page table pages for the large map may be
11750 				 * freed.  Validate the next-level address
11751 				 * before descending.
11752 				 */
11753 				goto restart;
11754 			}
11755 			pd = (pd_entry_t *)PHYS_TO_DMAP(pa);
11756 
11757 			for (k = pmap_pde_index(sva); k < NPDEPG; k++) {
11758 				pde = pd[k];
11759 				if ((pde & X86_PG_V) == 0) {
11760 					sva = rounddown2(sva, NBPDR);
11761 					sysctl_kmaps_dump(sb, &range, sva);
11762 					sva += NBPDR;
11763 					continue;
11764 				}
11765 				pa = pde & PG_FRAME;
11766 				if ((pde & PG_PS) != 0) {
11767 					sva = rounddown2(sva, NBPDR);
11768 					sysctl_kmaps_check(sb, &range, sva,
11769 					    pml4e, pdpe, pde, 0);
11770 					range.pdes++;
11771 					sva += NBPDR;
11772 					continue;
11773 				}
11774 				if (PMAP_ADDRESS_IN_LARGEMAP(sva) &&
11775 				    vm_phys_paddr_to_vm_page(pa) == NULL) {
11776 					/*
11777 					 * Page table pages for the large map
11778 					 * may be freed.  Validate the
11779 					 * next-level address before descending.
11780 					 */
11781 					goto restart;
11782 				}
11783 				pt = (pt_entry_t *)PHYS_TO_DMAP(pa);
11784 
11785 				for (l = pmap_pte_index(sva); l < NPTEPG; l++,
11786 				    sva += PAGE_SIZE) {
11787 					pte = pt[l];
11788 					if ((pte & X86_PG_V) == 0) {
11789 						sysctl_kmaps_dump(sb, &range,
11790 						    sva);
11791 						continue;
11792 					}
11793 					sysctl_kmaps_check(sb, &range, sva,
11794 					    pml4e, pdpe, pde, pte);
11795 					range.ptes++;
11796 				}
11797 			}
11798 		}
11799 	}
11800 
11801 	error = sbuf_finish(sb);
11802 	sbuf_delete(sb);
11803 	return (error);
11804 }
11805 SYSCTL_OID(_vm_pmap, OID_AUTO, kernel_maps,
11806     CTLTYPE_STRING | CTLFLAG_RD | CTLFLAG_MPSAFE | CTLFLAG_SKIP,
11807     NULL, 0, sysctl_kmaps, "A",
11808     "Dump kernel address layout");
11809 
11810 #ifdef DDB
DB_SHOW_COMMAND(pte,pmap_print_pte)11811 DB_SHOW_COMMAND(pte, pmap_print_pte)
11812 {
11813 	pmap_t pmap;
11814 	pml5_entry_t *pml5;
11815 	pml4_entry_t *pml4;
11816 	pdp_entry_t *pdp;
11817 	pd_entry_t *pde;
11818 	pt_entry_t *pte, PG_V;
11819 	vm_offset_t va;
11820 
11821 	if (!have_addr) {
11822 		db_printf("show pte addr\n");
11823 		return;
11824 	}
11825 	va = (vm_offset_t)addr;
11826 
11827 	if (kdb_thread != NULL)
11828 		pmap = vmspace_pmap(kdb_thread->td_proc->p_vmspace);
11829 	else
11830 		pmap = PCPU_GET(curpmap);
11831 
11832 	PG_V = pmap_valid_bit(pmap);
11833 	db_printf("VA 0x%016lx", va);
11834 
11835 	if (pmap_is_la57(pmap)) {
11836 		pml5 = pmap_pml5e(pmap, va);
11837 		db_printf(" pml5e 0x%016lx", *pml5);
11838 		if ((*pml5 & PG_V) == 0) {
11839 			db_printf("\n");
11840 			return;
11841 		}
11842 		pml4 = pmap_pml5e_to_pml4e(pml5, va);
11843 	} else {
11844 		pml4 = pmap_pml4e(pmap, va);
11845 	}
11846 	db_printf(" pml4e 0x%016lx", *pml4);
11847 	if ((*pml4 & PG_V) == 0) {
11848 		db_printf("\n");
11849 		return;
11850 	}
11851 	pdp = pmap_pml4e_to_pdpe(pml4, va);
11852 	db_printf(" pdpe 0x%016lx", *pdp);
11853 	if ((*pdp & PG_V) == 0 || (*pdp & PG_PS) != 0) {
11854 		db_printf("\n");
11855 		return;
11856 	}
11857 	pde = pmap_pdpe_to_pde(pdp, va);
11858 	db_printf(" pde 0x%016lx", *pde);
11859 	if ((*pde & PG_V) == 0 || (*pde & PG_PS) != 0) {
11860 		db_printf("\n");
11861 		return;
11862 	}
11863 	pte = pmap_pde_to_pte(pde, va);
11864 	db_printf(" pte 0x%016lx\n", *pte);
11865 }
11866 
DB_SHOW_COMMAND(phys2dmap,pmap_phys2dmap)11867 DB_SHOW_COMMAND(phys2dmap, pmap_phys2dmap)
11868 {
11869 	vm_paddr_t a;
11870 
11871 	if (have_addr) {
11872 		a = (vm_paddr_t)addr;
11873 		db_printf("0x%jx\n", (uintmax_t)PHYS_TO_DMAP(a));
11874 	} else {
11875 		db_printf("show phys2dmap addr\n");
11876 	}
11877 }
11878 
11879 static void
ptpages_show_page(int level,int idx,vm_page_t pg)11880 ptpages_show_page(int level, int idx, vm_page_t pg)
11881 {
11882 	db_printf("l %d i %d pg %p phys %#lx ref %x\n",
11883 	    level, idx, pg, VM_PAGE_TO_PHYS(pg), pg->ref_count);
11884 }
11885 
11886 static void
ptpages_show_complain(int level,int idx,uint64_t pte)11887 ptpages_show_complain(int level, int idx, uint64_t pte)
11888 {
11889 	db_printf("l %d i %d pte %#lx\n", level, idx, pte);
11890 }
11891 
11892 static void
ptpages_show_pml4(vm_page_t pg4,int num_entries,uint64_t PG_V)11893 ptpages_show_pml4(vm_page_t pg4, int num_entries, uint64_t PG_V)
11894 {
11895 	vm_page_t pg3, pg2, pg1;
11896 	pml4_entry_t *pml4;
11897 	pdp_entry_t *pdp;
11898 	pd_entry_t *pd;
11899 	int i4, i3, i2;
11900 
11901 	pml4 = (pml4_entry_t *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(pg4));
11902 	for (i4 = 0; i4 < num_entries; i4++) {
11903 		if ((pml4[i4] & PG_V) == 0)
11904 			continue;
11905 		pg3 = PHYS_TO_VM_PAGE(pml4[i4] & PG_FRAME);
11906 		if (pg3 == NULL) {
11907 			ptpages_show_complain(3, i4, pml4[i4]);
11908 			continue;
11909 		}
11910 		ptpages_show_page(3, i4, pg3);
11911 		pdp = (pdp_entry_t *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(pg3));
11912 		for (i3 = 0; i3 < NPDPEPG; i3++) {
11913 			if ((pdp[i3] & PG_V) == 0)
11914 				continue;
11915 			pg2 = PHYS_TO_VM_PAGE(pdp[i3] & PG_FRAME);
11916 			if (pg3 == NULL) {
11917 				ptpages_show_complain(2, i3, pdp[i3]);
11918 				continue;
11919 			}
11920 			ptpages_show_page(2, i3, pg2);
11921 			pd = (pd_entry_t *)PHYS_TO_DMAP(VM_PAGE_TO_PHYS(pg2));
11922 			for (i2 = 0; i2 < NPDEPG; i2++) {
11923 				if ((pd[i2] & PG_V) == 0)
11924 					continue;
11925 				pg1 = PHYS_TO_VM_PAGE(pd[i2] & PG_FRAME);
11926 				if (pg1 == NULL) {
11927 					ptpages_show_complain(1, i2, pd[i2]);
11928 					continue;
11929 				}
11930 				ptpages_show_page(1, i2, pg1);
11931 			}
11932 		}
11933 	}
11934 }
11935 
DB_SHOW_COMMAND(ptpages,pmap_ptpages)11936 DB_SHOW_COMMAND(ptpages, pmap_ptpages)
11937 {
11938 	pmap_t pmap;
11939 	vm_page_t pg;
11940 	pml5_entry_t *pml5;
11941 	uint64_t PG_V;
11942 	int i5;
11943 
11944 	if (have_addr)
11945 		pmap = (pmap_t)addr;
11946 	else
11947 		pmap = PCPU_GET(curpmap);
11948 
11949 	PG_V = pmap_valid_bit(pmap);
11950 
11951 	if (pmap_is_la57(pmap)) {
11952 		pml5 = pmap->pm_pmltop;
11953 		for (i5 = 0; i5 < NUPML5E; i5++) {
11954 			if ((pml5[i5] & PG_V) == 0)
11955 				continue;
11956 			pg = PHYS_TO_VM_PAGE(pml5[i5] & PG_FRAME);
11957 			if (pg == NULL) {
11958 				ptpages_show_complain(4, i5, pml5[i5]);
11959 				continue;
11960 			}
11961 			ptpages_show_page(4, i5, pg);
11962 			ptpages_show_pml4(pg, NPML4EPG, PG_V);
11963 		}
11964 	} else {
11965 		ptpages_show_pml4(PHYS_TO_VM_PAGE(DMAP_TO_PHYS(
11966 		    (vm_offset_t)pmap->pm_pmltop)), NUP4ML4E, PG_V);
11967 	}
11968 }
11969 #endif
11970