#include <sys/param.h>
#include <sys/kernel.h>
#include <sys/systm.h>
#include <sys/malloc.h>
#include <sys/memrange.h>
#include <sys/smp.h>
#include <sys/sysctl.h>
#include <vm/vm.h>
#include <vm/vm_param.h>
#include <vm/pmap.h>
#include <machine/cputypes.h>
#include <machine/md_var.h>
#include <machine/specialreg.h>
static char *mem_owner_bios = "BIOS";
#define MR686_FIXMTRR (1<<0)
#define mrwithin(mr, a) \
(((a) >= (mr)->mr_base) && ((a) < ((mr)->mr_base + (mr)->mr_len)))
#define mroverlap(mra, mrb) \
(mrwithin(mra, mrb->mr_base) || mrwithin(mrb, mra->mr_base))
#define mrvalid(base, len) \
((!(base & ((1 << 12) - 1))) && \
((len) >= (1 << 12)) && \
powerof2((len)) && \
!((base) & ((len) - 1)))
#define mrcopyflags(curr, new) \
(((curr) & ~MDF_ATTRMASK) | ((new) & MDF_ATTRMASK))
static int mtrrs_disabled;
SYSCTL_INT(_machdep, OID_AUTO, disable_mtrrs, CTLFLAG_RDTUN,
&mtrrs_disabled, 0,
"Disable MTRRs.");
static void x86_mrinit(struct mem_range_softc *sc);
static int x86_mrset(struct mem_range_softc *sc,
struct mem_range_desc *mrd, int *arg);
static void x86_mrAPinit(struct mem_range_softc *sc);
static void x86_mrreinit(struct mem_range_softc *sc);
static struct mem_range_ops x86_mrops = {
x86_mrinit,
x86_mrset,
x86_mrAPinit,
x86_mrreinit
};
static u_int64_t mtrrcap, mtrrdef;
static u_int64_t mtrr_physmask;
static struct mem_range_desc *mem_range_match(struct mem_range_softc *sc,
struct mem_range_desc *mrd);
static void x86_mrfetch(struct mem_range_softc *sc);
static int x86_mtrrtype(int flags);
static int x86_mrt2mtrr(int flags, int oldval);
static int x86_mtrrconflict(int flag1, int flag2);
static void x86_mrstore(struct mem_range_softc *sc);
static void x86_mrstoreone(void *arg);
static struct mem_range_desc *x86_mtrrfixsearch(struct mem_range_softc *sc,
u_int64_t addr);
static int x86_mrsetlow(struct mem_range_softc *sc,
struct mem_range_desc *mrd, int *arg);
static int x86_mrsetvariable(struct mem_range_softc *sc,
struct mem_range_desc *mrd, int *arg);
static int x86_mtrrtomrt[] = {
MDF_UNCACHEABLE,
MDF_WRITECOMBINE,
MDF_UNKNOWN,
MDF_UNKNOWN,
MDF_WRITETHROUGH,
MDF_WRITEPROTECT,
MDF_WRITEBACK
};
#define MTRRTOMRTLEN nitems(x86_mtrrtomrt)
static int
x86_mtrr2mrt(int val)
{
if (val < 0 || val >= MTRRTOMRTLEN)
return (MDF_UNKNOWN);
return (x86_mtrrtomrt[val]);
}
static int
x86_mtrrconflict(int flag1, int flag2)
{
flag1 &= MDF_ATTRMASK;
flag2 &= MDF_ATTRMASK;
if ((flag1 & MDF_UNKNOWN) || (flag2 & MDF_UNKNOWN))
return (1);
if (flag1 == flag2 ||
(flag1 == MDF_WRITEBACK && flag2 == MDF_UNCACHEABLE) ||
(flag2 == MDF_WRITEBACK && flag1 == MDF_UNCACHEABLE))
return (0);
return (1);
}
static struct mem_range_desc *
mem_range_match(struct mem_range_softc *sc, struct mem_range_desc *mrd)
{
struct mem_range_desc *cand;
int i;
for (i = 0, cand = sc->mr_desc; i < sc->mr_ndesc; i++, cand++)
if ((cand->mr_base == mrd->mr_base) &&
(cand->mr_len == mrd->mr_len))
return (cand);
return (NULL);
}
static void
x86_mr_split_dmap(struct mem_range_softc *sc __unused)
{
#ifdef __amd64__
struct mem_range_desc *mrd;
int i;
i = (sc->mr_cap & MR686_FIXMTRR) ? MTRR_N64K + MTRR_N16K + MTRR_N4K : 0;
mrd = sc->mr_desc + i;
for (; i < sc->mr_ndesc; i++, mrd++) {
if ((mrd->mr_flags & (MDF_ACTIVE | MDF_BOGUS)) == MDF_ACTIVE)
pmap_demote_DMAP(mrd->mr_base, mrd->mr_len, true);
}
#endif
}
static void
x86_mrfetch(struct mem_range_softc *sc)
{
struct mem_range_desc *mrd;
u_int64_t msrv;
int i, j, msr;
mrd = sc->mr_desc;
if (sc->mr_cap & MR686_FIXMTRR) {
msr = MSR_MTRR64kBase;
for (i = 0; i < (MTRR_N64K / 8); i++, msr++) {
msrv = rdmsr(msr);
for (j = 0; j < 8; j++, mrd++) {
mrd->mr_flags =
(mrd->mr_flags & ~MDF_ATTRMASK) |
x86_mtrr2mrt(msrv & 0xff) | MDF_ACTIVE;
if (mrd->mr_owner[0] == 0)
strcpy(mrd->mr_owner, mem_owner_bios);
msrv = msrv >> 8;
}
}
msr = MSR_MTRR16kBase;
for (i = 0; i < MTRR_N16K / 8; i++, msr++) {
msrv = rdmsr(msr);
for (j = 0; j < 8; j++, mrd++) {
mrd->mr_flags =
(mrd->mr_flags & ~MDF_ATTRMASK) |
x86_mtrr2mrt(msrv & 0xff) | MDF_ACTIVE;
if (mrd->mr_owner[0] == 0)
strcpy(mrd->mr_owner, mem_owner_bios);
msrv = msrv >> 8;
}
}
msr = MSR_MTRR4kBase;
for (i = 0; i < MTRR_N4K / 8; i++, msr++) {
msrv = rdmsr(msr);
for (j = 0; j < 8; j++, mrd++) {
mrd->mr_flags =
(mrd->mr_flags & ~MDF_ATTRMASK) |
x86_mtrr2mrt(msrv & 0xff) | MDF_ACTIVE;
if (mrd->mr_owner[0] == 0)
strcpy(mrd->mr_owner, mem_owner_bios);
msrv = msrv >> 8;
}
}
}
msr = MSR_MTRRVarBase;
for (; mrd - sc->mr_desc < sc->mr_ndesc; msr += 2, mrd++) {
msrv = rdmsr(msr);
mrd->mr_flags = (mrd->mr_flags & ~MDF_ATTRMASK) |
x86_mtrr2mrt(msrv & MTRR_PHYSBASE_TYPE);
mrd->mr_base = msrv & mtrr_physmask;
msrv = rdmsr(msr + 1);
mrd->mr_flags = (msrv & MTRR_PHYSMASK_VALID) ?
(mrd->mr_flags | MDF_ACTIVE) :
(mrd->mr_flags & ~MDF_ACTIVE);
mrd->mr_len = (~(msrv & mtrr_physmask) &
(mtrr_physmask | 0xfff)) + 1;
if (!mrvalid(mrd->mr_base, mrd->mr_len))
mrd->mr_flags |= MDF_BOGUS;
if ((mrd->mr_flags & MDF_ACTIVE) && (mrd->mr_owner[0] == 0))
strcpy(mrd->mr_owner, mem_owner_bios);
}
}
static int
x86_mtrrtype(int flags)
{
int i;
flags &= MDF_ATTRMASK;
for (i = 0; i < MTRRTOMRTLEN; i++) {
if (x86_mtrrtomrt[i] == MDF_UNKNOWN)
continue;
if (flags == x86_mtrrtomrt[i])
return (i);
}
return (-1);
}
static int
x86_mrt2mtrr(int flags, int oldval)
{
int val;
if ((val = x86_mtrrtype(flags)) == -1)
return (oldval & 0xff);
return (val & 0xff);
}
static void
x86_mrstore(struct mem_range_softc *sc)
{
smp_rendezvous(NULL, x86_mrstoreone, NULL, sc);
}
static void
x86_mrstoreone(void *arg)
{
struct mem_range_softc *sc = arg;
struct mem_range_desc *mrd;
u_int64_t omsrv, msrv;
int i, j, msr;
u_long cr0, cr4;
mrd = sc->mr_desc;
critical_enter();
cr4 = rcr4();
load_cr4(cr4 & ~CR4_PGE);
cr0 = rcr0();
load_cr0((cr0 & ~CR0_NW) | CR0_CD);
wbinvd();
invltlb();
wrmsr(MSR_MTRRdefType, rdmsr(MSR_MTRRdefType) & ~MTRR_DEF_ENABLE);
if (sc->mr_cap & MR686_FIXMTRR) {
msr = MSR_MTRR64kBase;
for (i = 0; i < MTRR_N64K / 8; i++, msr++) {
msrv = 0;
omsrv = rdmsr(msr);
for (j = 7; j >= 0; j--) {
msrv = msrv << 8;
msrv |= x86_mrt2mtrr((mrd + j)->mr_flags,
omsrv >> (j * 8));
}
wrmsr(msr, msrv);
mrd += 8;
}
msr = MSR_MTRR16kBase;
for (i = 0; i < MTRR_N16K / 8; i++, msr++) {
msrv = 0;
omsrv = rdmsr(msr);
for (j = 7; j >= 0; j--) {
msrv = msrv << 8;
msrv |= x86_mrt2mtrr((mrd + j)->mr_flags,
omsrv >> (j * 8));
}
wrmsr(msr, msrv);
mrd += 8;
}
msr = MSR_MTRR4kBase;
for (i = 0; i < MTRR_N4K / 8; i++, msr++) {
msrv = 0;
omsrv = rdmsr(msr);
for (j = 7; j >= 0; j--) {
msrv = msrv << 8;
msrv |= x86_mrt2mtrr((mrd + j)->mr_flags,
omsrv >> (j * 8));
}
wrmsr(msr, msrv);
mrd += 8;
}
}
msr = MSR_MTRRVarBase;
for (; mrd - sc->mr_desc < sc->mr_ndesc; msr += 2, mrd++) {
omsrv = rdmsr(msr);
if (mrd->mr_flags & MDF_ACTIVE) {
msrv = mrd->mr_base & mtrr_physmask;
msrv |= x86_mrt2mtrr(mrd->mr_flags, omsrv);
} else {
msrv = 0;
}
wrmsr(msr, msrv);
if (mrd->mr_flags & MDF_ACTIVE) {
msrv = MTRR_PHYSMASK_VALID |
rounddown2(mtrr_physmask, mrd->mr_len);
} else {
msrv = 0;
}
wrmsr(msr + 1, msrv);
}
wbinvd();
invltlb();
wrmsr(MSR_MTRRdefType, rdmsr(MSR_MTRRdefType) | MTRR_DEF_ENABLE);
load_cr0(cr0);
load_cr4(cr4);
critical_exit();
}
static struct mem_range_desc *
x86_mtrrfixsearch(struct mem_range_softc *sc, u_int64_t addr)
{
struct mem_range_desc *mrd;
int i;
for (i = 0, mrd = sc->mr_desc; i < MTRR_N64K + MTRR_N16K + MTRR_N4K;
i++, mrd++)
if (addr >= mrd->mr_base &&
addr < mrd->mr_base + mrd->mr_len)
return (mrd);
return (NULL);
}
static int
x86_mrsetlow(struct mem_range_softc *sc, struct mem_range_desc *mrd, int *arg)
{
struct mem_range_desc *first_md, *last_md, *curr_md;
if ((first_md = x86_mtrrfixsearch(sc, mrd->mr_base)) == NULL ||
(last_md = x86_mtrrfixsearch(sc, mrd->mr_base + mrd->mr_len - 1))
== NULL)
return (EINVAL);
if ((mrd->mr_flags & MDF_FORCE) == 0) {
for (curr_md = first_md; curr_md <= last_md; curr_md++) {
if ((curr_md->mr_flags & MDF_ATTRMASK) == MDF_UNKNOWN)
return (EACCES);
}
}
for (curr_md = first_md; curr_md <= last_md; curr_md++) {
curr_md->mr_flags = mrcopyflags(curr_md->mr_flags &
~MDF_FIRMWARE, mrd->mr_flags);
bcopy(mrd->mr_owner, curr_md->mr_owner, sizeof(mrd->mr_owner));
}
return (0);
}
static int
x86_mrsetvariable(struct mem_range_softc *sc, struct mem_range_desc *mrd,
int *arg)
{
struct mem_range_desc *curr_md, *free_md;
int i;
i = (sc->mr_cap & MR686_FIXMTRR) ? MTRR_N64K + MTRR_N16K + MTRR_N4K : 0;
curr_md = sc->mr_desc + i;
free_md = NULL;
for (; i < sc->mr_ndesc; i++, curr_md++) {
if (curr_md->mr_flags & MDF_ACTIVE) {
if (curr_md->mr_base == mrd->mr_base &&
curr_md->mr_len == mrd->mr_len) {
if (curr_md->mr_flags & MDF_BUSY)
return (EBUSY);
if (!(mrd->mr_flags & MDF_FORCE) &&
(curr_md->mr_flags & MDF_ATTRMASK) ==
MDF_UNKNOWN)
return (EACCES);
free_md = curr_md;
break;
}
if (mroverlap(curr_md, mrd)) {
if (x86_mtrrconflict(curr_md->mr_flags,
mrd->mr_flags))
return (EINVAL);
}
} else if (free_md == NULL) {
free_md = curr_md;
}
}
if (free_md == NULL)
return (ENOSPC);
free_md->mr_base = mrd->mr_base;
free_md->mr_len = mrd->mr_len;
free_md->mr_flags = mrcopyflags(MDF_ACTIVE, mrd->mr_flags);
bcopy(mrd->mr_owner, free_md->mr_owner, sizeof(mrd->mr_owner));
return (0);
}
static int
x86_mrset(struct mem_range_softc *sc, struct mem_range_desc *mrd, int *arg)
{
struct mem_range_desc *targ;
int error;
switch (*arg) {
case MEMRANGE_SET_UPDATE:
if (!mrvalid(mrd->mr_base, mrd->mr_len) ||
x86_mtrrtype(mrd->mr_flags) == -1)
return (EINVAL);
#define FIXTOP \
((MTRR_N64K * 0x10000) + (MTRR_N16K * 0x4000) + (MTRR_N4K * 0x1000))
if ((sc->mr_cap & MR686_FIXMTRR) != 0 &&
mrd->mr_base + mrd->mr_len <= FIXTOP) {
if ((error = x86_mrsetlow(sc, mrd, arg)) != 0)
return (error);
} else {
if ((error = x86_mrsetvariable(sc, mrd, arg)) != 0)
return (error);
}
break;
case MEMRANGE_SET_REMOVE:
if ((targ = mem_range_match(sc, mrd)) == NULL)
return (ENOENT);
if (targ->mr_flags & MDF_FIXACTIVE)
return (EPERM);
if (targ->mr_flags & MDF_BUSY)
return (EBUSY);
targ->mr_flags &= ~MDF_ACTIVE;
targ->mr_owner[0] = 0;
break;
default:
return (EOPNOTSUPP);
}
x86_mr_split_dmap(sc);
x86_mrstore(sc);
x86_mrfetch(sc);
return (0);
}
static void
x86_mrinit(struct mem_range_softc *sc)
{
struct mem_range_desc *mrd;
int i, nmdesc;
if (sc->mr_desc != NULL)
return;
nmdesc = 0;
mtrrcap = rdmsr(MSR_MTRRcap);
mtrrdef = rdmsr(MSR_MTRRdefType);
if (!(mtrrdef & MTRR_DEF_ENABLE)) {
if (bootverbose)
printf("CPU supports MTRRs but not enabled\n");
return;
}
nmdesc = mtrrcap & MTRR_CAP_VCNT;
if (bootverbose)
printf("Pentium Pro MTRR support enabled\n");
mtrr_physmask = (((uint64_t)1 << cpu_maxphyaddr) - 1) &
~(uint64_t)0xfff;
if ((mtrrcap & MTRR_CAP_FIXED) && (mtrrdef & MTRR_DEF_FIXED_ENABLE)) {
sc->mr_cap = MR686_FIXMTRR;
nmdesc += MTRR_N64K + MTRR_N16K + MTRR_N4K;
}
sc->mr_desc = malloc(nmdesc * sizeof(struct mem_range_desc), M_MEMDESC,
M_WAITOK | M_ZERO);
sc->mr_ndesc = nmdesc;
mrd = sc->mr_desc;
if (sc->mr_cap & MR686_FIXMTRR) {
for (i = 0; i < MTRR_N64K; i++, mrd++) {
mrd->mr_base = i * 0x10000;
mrd->mr_len = 0x10000;
mrd->mr_flags = MDF_FIXBASE | MDF_FIXLEN |
MDF_FIXACTIVE;
}
for (i = 0; i < MTRR_N16K; i++, mrd++) {
mrd->mr_base = i * 0x4000 + 0x80000;
mrd->mr_len = 0x4000;
mrd->mr_flags = MDF_FIXBASE | MDF_FIXLEN |
MDF_FIXACTIVE;
}
for (i = 0; i < MTRR_N4K; i++, mrd++) {
mrd->mr_base = i * 0x1000 + 0xc0000;
mrd->mr_len = 0x1000;
mrd->mr_flags = MDF_FIXBASE | MDF_FIXLEN |
MDF_FIXACTIVE;
}
}
x86_mrfetch(sc);
mrd = sc->mr_desc;
for (i = 0; i < sc->mr_ndesc; i++, mrd++) {
if (mrd->mr_flags & MDF_ACTIVE)
mrd->mr_flags |= MDF_FIRMWARE;
}
x86_mr_split_dmap(sc);
}
static void
x86_mrAPinit(struct mem_range_softc *sc)
{
x86_mrstoreone(sc);
wrmsr(MSR_MTRRdefType, mtrrdef);
}
static void
x86_mrreinit(struct mem_range_softc *sc)
{
smp_rendezvous(NULL, (void (*)(void *))x86_mrAPinit, NULL, sc);
}
static void
x86_mem_drvinit(void *unused)
{
if (mtrrs_disabled)
return;
if (!(cpu_feature & CPUID_MTRR))
return;
mem_range_softc.mr_op = &x86_mrops;
x86_mrinit(&mem_range_softc);
}
SYSINIT(x86memdev, SI_SUB_CPU, SI_ORDER_ANY, x86_mem_drvinit, NULL);