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