[linux-2.6-block.git] / arch / x86 / kernel / cpu / mtrr / if.c

// SPDX-License-Identifier: GPL-2.0
#include <linux/capability.h>
#include <linux/seq_file.h>
#include <linux/uaccess.h>
#include <linux/proc_fs.h>
#include <linux/ctype.h>
#include <linux/string.h>
#include <linux/slab.h>
#include <linux/init.h>

#define LINE_SIZE 80

#include <asm/mtrr.h>

#include "mtrr.h"

#define FILE_FCOUNT(f) (((struct seq_file *)((f)->private_data))->private)

static const char *const mtrr_strings[MTRR_NUM_TYPES] =
{
	"uncachable",		/* 0 */
	"write-combining",	/* 1 */
	"?",			/* 2 */
	"?",			/* 3 */
	"write-through",	/* 4 */
	"write-protect",	/* 5 */
	"write-back",		/* 6 */
};

const char *mtrr_attrib_to_str(int x)
{
	return (x <= 6) ? mtrr_strings[x] : "?";
}

#ifdef CONFIG_PROC_FS

static int
mtrr_file_add(unsigned long base, unsigned long size,
	      unsigned int type, bool increment, struct file *file, int page)
{
	unsigned int *fcount = FILE_FCOUNT(file);
	int reg, max;

	max = num_var_ranges;
	if (fcount == NULL) {
		fcount = kcalloc(max, sizeof(*fcount), GFP_KERNEL);
		if (!fcount)
			return -ENOMEM;
		FILE_FCOUNT(file) = fcount;
	}
	if (!page) {
		if ((base & (PAGE_SIZE - 1)) || (size & (PAGE_SIZE - 1)))
			return -EINVAL;
		base >>= PAGE_SHIFT;
		size >>= PAGE_SHIFT;
	}
	reg = mtrr_add_page(base, size, type, true);
	if (reg >= 0)
		++fcount[reg];
	return reg;
}

static int
mtrr_file_del(unsigned long base, unsigned long size,
	      struct file *file, int page)
{
	unsigned int *fcount = FILE_FCOUNT(file);
	int reg;

	if (!page) {
		if ((base & (PAGE_SIZE - 1)) || (size & (PAGE_SIZE - 1)))
			return -EINVAL;
		base >>= PAGE_SHIFT;
		size >>= PAGE_SHIFT;
	}
	reg = mtrr_del_page(-1, base, size);
	if (reg < 0)
		return reg;
	if (fcount == NULL)
		return reg;
	if (fcount[reg] < 1)
		return -EINVAL;
	--fcount[reg];
	return reg;
}

/*
 * seq_file can seek but we ignore it.
 *
 * Format of control line:
 *    "base=%Lx size=%Lx type=%s" or "disable=%d"
 */
static ssize_t
mtrr_write(struct file *file, const char __user *buf, size_t len, loff_t * ppos)
{
	int i, err;
	unsigned long reg;
	unsigned long long base, size;
	char *ptr;
	char line[LINE_SIZE];
	int length;
	size_t linelen;

	if (!capable(CAP_SYS_ADMIN))
		return -EPERM;

	memset(line, 0, LINE_SIZE);

	length = strncpy_from_user(line, buf, LINE_SIZE - 1);
	if (length < 0)
		return length;

	linelen = strlen(line);
	ptr = line + linelen - 1;
	if (linelen && *ptr == '\n')
		*ptr = '\0';

	if (!strncmp(line, "disable=", 8)) {
		reg = simple_strtoul(line + 8, &ptr, 0);
		err = mtrr_del_page(reg, 0, 0);
		if (err < 0)
			return err;
		return len;
	}

	if (strncmp(line, "base=", 5))
		return -EINVAL;

	base = simple_strtoull(line + 5, &ptr, 0);
	ptr = skip_spaces(ptr);

	if (strncmp(ptr, "size=", 5))
		return -EINVAL;

	size = simple_strtoull(ptr + 5, &ptr, 0);
	if ((base & 0xfff) || (size & 0xfff))
		return -EINVAL;
	ptr = skip_spaces(ptr);

	if (strncmp(ptr, "type=", 5))
		return -EINVAL;
	ptr = skip_spaces(ptr + 5);

	i = match_string(mtrr_strings, MTRR_NUM_TYPES, ptr);
	if (i < 0)
		return i;

	base >>= PAGE_SHIFT;
	size >>= PAGE_SHIFT;
	err = mtrr_add_page((unsigned long)base, (unsigned long)size, i, true);
	if (err < 0)
		return err;
	return len;
}

static long
mtrr_ioctl(struct file *file, unsigned int cmd, unsigned long __arg)
{
	int err = 0;
	mtrr_type type;
	unsigned long base;
	unsigned long size;
	struct mtrr_sentry sentry;
	struct mtrr_gentry gentry;
	void __user *arg = (void __user *) __arg;

	switch (cmd) {
	case MTRRIOC_ADD_ENTRY:
	case MTRRIOC_SET_ENTRY:
	case MTRRIOC_DEL_ENTRY:
	case MTRRIOC_KILL_ENTRY:
	case MTRRIOC_ADD_PAGE_ENTRY:
	case MTRRIOC_SET_PAGE_ENTRY:
	case MTRRIOC_DEL_PAGE_ENTRY:
	case MTRRIOC_KILL_PAGE_ENTRY:
		if (copy_from_user(&sentry, arg, sizeof sentry))
			return -EFAULT;
		break;
	case MTRRIOC_GET_ENTRY:
	case MTRRIOC_GET_PAGE_ENTRY:
		if (copy_from_user(&gentry, arg, sizeof gentry))
			return -EFAULT;
		break;
#ifdef CONFIG_COMPAT
	case MTRRIOC32_ADD_ENTRY:
	case MTRRIOC32_SET_ENTRY:
	case MTRRIOC32_DEL_ENTRY:
	case MTRRIOC32_KILL_ENTRY:
	case MTRRIOC32_ADD_PAGE_ENTRY:
	case MTRRIOC32_SET_PAGE_ENTRY:
	case MTRRIOC32_DEL_PAGE_ENTRY:
	case MTRRIOC32_KILL_PAGE_ENTRY: {
		struct mtrr_sentry32 __user *s32;

		s32 = (struct mtrr_sentry32 __user *)__arg;
		err = get_user(sentry.base, &s32->base);
		err |= get_user(sentry.size, &s32->size);
		err |= get_user(sentry.type, &s32->type);
		if (err)
			return err;
		break;
	}
	case MTRRIOC32_GET_ENTRY:
	case MTRRIOC32_GET_PAGE_ENTRY: {
		struct mtrr_gentry32 __user *g32;

		g32 = (struct mtrr_gentry32 __user *)__arg;
		err = get_user(gentry.regnum, &g32->regnum);
		err |= get_user(gentry.base, &g32->base);
		err |= get_user(gentry.size, &g32->size);
		err |= get_user(gentry.type, &g32->type);
		if (err)
			return err;
		break;
	}
#endif
	}

	switch (cmd) {
	default:
		return -ENOTTY;
	case MTRRIOC_ADD_ENTRY:
#ifdef CONFIG_COMPAT
	case MTRRIOC32_ADD_ENTRY:
#endif
		if (!capable(CAP_SYS_ADMIN))
			return -EPERM;
		err =
		    mtrr_file_add(sentry.base, sentry.size, sentry.type, true,
				  file, 0);
		break;
	case MTRRIOC_SET_ENTRY:
#ifdef CONFIG_COMPAT
	case MTRRIOC32_SET_ENTRY:
#endif
		if (!capable(CAP_SYS_ADMIN))
			return -EPERM;
		err = mtrr_add(sentry.base, sentry.size, sentry.type, false);
		break;
	case MTRRIOC_DEL_ENTRY:
#ifdef CONFIG_COMPAT
	case MTRRIOC32_DEL_ENTRY:
#endif
		if (!capable(CAP_SYS_ADMIN))
			return -EPERM;
		err = mtrr_file_del(sentry.base, sentry.size, file, 0);
		break;
	case MTRRIOC_KILL_ENTRY:
#ifdef CONFIG_COMPAT
	case MTRRIOC32_KILL_ENTRY:
#endif
		if (!capable(CAP_SYS_ADMIN))
			return -EPERM;
		err = mtrr_del(-1, sentry.base, sentry.size);
		break;
	case MTRRIOC_GET_ENTRY:
#ifdef CONFIG_COMPAT
	case MTRRIOC32_GET_ENTRY:
#endif
		if (gentry.regnum >= num_var_ranges)
			return -EINVAL;
		mtrr_if->get(gentry.regnum, &base, &size, &type);

		/* Hide entries that go above 4GB */
		if (base + size - 1 >= (1UL << (8 * sizeof(gentry.size) - PAGE_SHIFT))
		    || size >= (1UL << (8 * sizeof(gentry.size) - PAGE_SHIFT)))
			gentry.base = gentry.size = gentry.type = 0;
		else {
			gentry.base = base << PAGE_SHIFT;
			gentry.size = size << PAGE_SHIFT;
			gentry.type = type;
		}

		break;
	case MTRRIOC_ADD_PAGE_ENTRY:
#ifdef CONFIG_COMPAT
	case MTRRIOC32_ADD_PAGE_ENTRY:
#endif
		if (!capable(CAP_SYS_ADMIN))
			return -EPERM;
		err =
		    mtrr_file_add(sentry.base, sentry.size, sentry.type, true,
				  file, 1);
		break;
	case MTRRIOC_SET_PAGE_ENTRY:
#ifdef CONFIG_COMPAT
	case MTRRIOC32_SET_PAGE_ENTRY:
#endif
		if (!capable(CAP_SYS_ADMIN))
			return -EPERM;
		err =
		    mtrr_add_page(sentry.base, sentry.size, sentry.type, false);
		break;
	case MTRRIOC_DEL_PAGE_ENTRY:
#ifdef CONFIG_COMPAT
	case MTRRIOC32_DEL_PAGE_ENTRY:
#endif
		if (!capable(CAP_SYS_ADMIN))
			return -EPERM;
		err = mtrr_file_del(sentry.base, sentry.size, file, 1);
		break;
	case MTRRIOC_KILL_PAGE_ENTRY:
#ifdef CONFIG_COMPAT
	case MTRRIOC32_KILL_PAGE_ENTRY:
#endif
		if (!capable(CAP_SYS_ADMIN))
			return -EPERM;
		err = mtrr_del_page(-1, sentry.base, sentry.size);
		break;
	case MTRRIOC_GET_PAGE_ENTRY:
#ifdef CONFIG_COMPAT
	case MTRRIOC32_GET_PAGE_ENTRY:
#endif
		if (gentry.regnum >= num_var_ranges)
			return -EINVAL;
		mtrr_if->get(gentry.regnum, &base, &size, &type);
		/* Hide entries that would overflow */
		if (size != (__typeof__(gentry.size))size)
			gentry.base = gentry.size = gentry.type = 0;
		else {
			gentry.base = base;
			gentry.size = size;
			gentry.type = type;
		}
		break;
	}

	if (err)
		return err;

	switch (cmd) {
	case MTRRIOC_GET_ENTRY:
	case MTRRIOC_GET_PAGE_ENTRY:
		if (copy_to_user(arg, &gentry, sizeof gentry))
			err = -EFAULT;
		break;
#ifdef CONFIG_COMPAT
	case MTRRIOC32_GET_ENTRY:
	case MTRRIOC32_GET_PAGE_ENTRY: {
		struct mtrr_gentry32 __user *g32;

		g32 = (struct mtrr_gentry32 __user *)__arg;
		err = put_user(gentry.base, &g32->base);
		err |= put_user(gentry.size, &g32->size);
		err |= put_user(gentry.regnum, &g32->regnum);
		err |= put_user(gentry.type, &g32->type);
		break;
	}
#endif
	}
	return err;
}

static int mtrr_close(struct inode *ino, struct file *file)
{
	unsigned int *fcount = FILE_FCOUNT(file);
	int i, max;

	if (fcount != NULL) {
		max = num_var_ranges;
		for (i = 0; i < max; ++i) {
			while (fcount[i] > 0) {
				mtrr_del(i, 0, 0);
				--fcount[i];
			}
		}
		kfree(fcount);
		FILE_FCOUNT(file) = NULL;
	}
	return single_release(ino, file);
}

static int mtrr_seq_show(struct seq_file *seq, void *offset);

static int mtrr_open(struct inode *inode, struct file *file)
{
	if (!mtrr_if)
		return -EIO;
	if (!mtrr_if->get)
		return -ENXIO;
	return single_open(file, mtrr_seq_show, NULL);
}

static const struct file_operations mtrr_fops = {
	.owner			= THIS_MODULE,
	.open			= mtrr_open,
	.read			= seq_read,
	.llseek			= seq_lseek,
	.write			= mtrr_write,
	.unlocked_ioctl		= mtrr_ioctl,
	.compat_ioctl		= mtrr_ioctl,
	.release		= mtrr_close,
};

static int mtrr_seq_show(struct seq_file *seq, void *offset)
{
	char factor;
	int i, max;
	mtrr_type type;
	unsigned long base, size;

	max = num_var_ranges;
	for (i = 0; i < max; i++) {
		mtrr_if->get(i, &base, &size, &type);
		if (size == 0) {
			mtrr_usage_table[i] = 0;
			continue;
		}
		if (size < (0x100000 >> PAGE_SHIFT)) {
			/* less than 1MB */
			factor = 'K';
			size <<= PAGE_SHIFT - 10;
		} else {
			factor = 'M';
			size >>= 20 - PAGE_SHIFT;
		}
		/* Base can be > 32bit */
		seq_printf(seq, "reg%02i: base=0x%06lx000 (%5luMB), size=%5lu%cB, count=%d: %s\n",
			   i, base, base >> (20 - PAGE_SHIFT),
			   size, factor,
			   mtrr_usage_table[i], mtrr_attrib_to_str(type));
	}
	return 0;
}

static int __init mtrr_if_init(void)
{
	struct cpuinfo_x86 *c = &boot_cpu_data;

	if ((!cpu_has(c, X86_FEATURE_MTRR)) &&
	    (!cpu_has(c, X86_FEATURE_K6_MTRR)) &&
	    (!cpu_has(c, X86_FEATURE_CYRIX_ARR)) &&
	    (!cpu_has(c, X86_FEATURE_CENTAUR_MCR)))
		return -ENODEV;

	proc_create("mtrr", S_IWUSR | S_IRUGO, NULL, &mtrr_fops);
	return 0;
}
arch_initcall(mtrr_if_init);
#endif			/*  CONFIG_PROC_FS  */
Commit	Line	Data
b2441318	1	// SPDX-License-Identifier: GPL-2.0
a9415644	2	#include <linux/capability.h>
1da177e4	3	#include <linux/seq_file.h>
26dc67ed JSR	4	#include <linux/uaccess.h>
26dc67ed JSR	5	#include <linux/proc_fs.h>
26dc67ed	6	#include <linux/ctype.h>
e7d2860b	7	#include <linux/string.h>
5a0e3ad6	8	#include <linux/slab.h>
26dc67ed	9	#include <linux/init.h>
1da177e4 LT	10
	11	#define LINE_SIZE 80
	12
	13	#include <asm/mtrr.h>
26dc67ed	14
1da177e4 LT	15	#include "mtrr.h"
1da177e4 LT	16
1da177e4 LT	17	#define FILE_FCOUNT(f) (((struct seq_file *)((f)->private_data))->private)
1da177e4 LT	18
365bff80	19	static const char *const mtrr_strings[MTRR_NUM_TYPES] =
1da177e4	20	{
26dc67ed JSR	21	"uncachable", /* 0 */
	22	"write-combining", /* 1 */
	23	"?", /* 2 */
	24	"?", /* 3 */
	25	"write-through", /* 4 */
	26	"write-protect", /* 5 */
	27	"write-back", /* 6 */
1da177e4 LT	28	};
1da177e4 LT	29
365bff80	30	const char *mtrr_attrib_to_str(int x)
1da177e4 LT	31	{
	32	return (x <= 6) ? mtrr_strings[x] : "?";
	33	}
	34
	35	#ifdef CONFIG_PROC_FS
	36
	37	static int
	38	mtrr_file_add(unsigned long base, unsigned long size,
2d2ee8de	39	unsigned int type, bool increment, struct file *file, int page)
1da177e4	40	{
26dc67ed	41	unsigned int *fcount = FILE_FCOUNT(file);
1da177e4	42	int reg, max;
1da177e4 LT	43
	44	max = num_var_ranges;
	45	if (fcount == NULL) {
6396bb22	46	fcount = kcalloc(max, sizeof(*fcount), GFP_KERNEL);
1da177e4 LT	47	if (!fcount)
1da177e4 LT	48	return -ENOMEM;
1da177e4 LT	49	FILE_FCOUNT(file) = fcount;
	50	}
	51	if (!page) {
	52	if ((base & (PAGE_SIZE - 1)) \|\| (size & (PAGE_SIZE - 1)))
	53	return -EINVAL;
	54	base >>= PAGE_SHIFT;
	55	size >>= PAGE_SHIFT;
	56	}
2d2ee8de	57	reg = mtrr_add_page(base, size, type, true);
1da177e4 LT	58	if (reg >= 0)
	59	++fcount[reg];
	60	return reg;
	61	}
	62
	63	static int
	64	mtrr_file_del(unsigned long base, unsigned long size,
	65	struct file *file, int page)
	66	{
1da177e4	67	unsigned int *fcount = FILE_FCOUNT(file);
26dc67ed	68	int reg;
1da177e4 LT	69
	70	if (!page) {
	71	if ((base & (PAGE_SIZE - 1)) \|\| (size & (PAGE_SIZE - 1)))
	72	return -EINVAL;
	73	base >>= PAGE_SHIFT;
	74	size >>= PAGE_SHIFT;
	75	}
	76	reg = mtrr_del_page(-1, base, size);
	77	if (reg < 0)
	78	return reg;
	79	if (fcount == NULL)
	80	return reg;
	81	if (fcount[reg] < 1)
	82	return -EINVAL;
	83	--fcount[reg];
	84	return reg;
	85	}
	86
26dc67ed JSR	87	/*
	88	* seq_file can seek but we ignore it.
	89	*
	90	* Format of control line:
	91	* "base=%Lx size=%Lx type=%s" or "disable=%d"
	92	*/
1da177e4 LT	93	static ssize_t
1da177e4 LT	94	mtrr_write(struct file file, const char __user buf, size_t len, loff_t * ppos)
1da177e4 LT	95	{
	96	int i, err;
	97	unsigned long reg;
	98	unsigned long long base, size;
	99	char *ptr;
	100	char line[LINE_SIZE];
11879ba5	101	int length;
1da177e4 LT	102	size_t linelen;
	103
	104	if (!capable(CAP_SYS_ADMIN))
	105	return -EPERM;
26dc67ed	106
1da177e4	107	memset(line, 0, LINE_SIZE);
11879ba5	108
7f8ec5a4	109	length = strncpy_from_user(line, buf, LINE_SIZE - 1);
11879ba5	110	if (length < 0)
7f8ec5a4	111	return length;
26dc67ed	112
1da177e4 LT	113	linelen = strlen(line);
	114	ptr = line + linelen - 1;
	115	if (linelen && *ptr == '\n')
	116	*ptr = '\0';
26dc67ed	117
1da177e4 LT	118	if (!strncmp(line, "disable=", 8)) {
	119	reg = simple_strtoul(line + 8, &ptr, 0);
	120	err = mtrr_del_page(reg, 0, 0);
	121	if (err < 0)
	122	return err;
	123	return len;
	124	}
26dc67ed	125
1da177e4 LT	126	if (strncmp(line, "base=", 5))
1da177e4 LT	127	return -EINVAL;
26dc67ed	128
1da177e4	129	base = simple_strtoull(line + 5, &ptr, 0);
e7d2860b	130	ptr = skip_spaces(ptr);
26dc67ed	131
1da177e4 LT	132	if (strncmp(ptr, "size=", 5))
1da177e4 LT	133	return -EINVAL;
26dc67ed	134
1da177e4 LT	135	size = simple_strtoull(ptr + 5, &ptr, 0);
	136	if ((base & 0xfff) \|\| (size & 0xfff))
	137	return -EINVAL;
e7d2860b	138	ptr = skip_spaces(ptr);
26dc67ed	139
1da177e4 LT	140	if (strncmp(ptr, "type=", 5))
1da177e4 LT	141	return -EINVAL;
e7d2860b	142	ptr = skip_spaces(ptr + 5);
26dc67ed	143
13a4db9d AS	144	i = match_string(mtrr_strings, MTRR_NUM_TYPES, ptr);
	145	if (i < 0)
	146	return i;
	147
	148	base >>= PAGE_SHIFT;
	149	size >>= PAGE_SHIFT;
	150	err = mtrr_add_page((unsigned long)base, (unsigned long)size, i, true);
	151	if (err < 0)
	152	return err;
	153	return len;
1da177e4 LT	154	}
1da177e4 LT	155
c5311781 BG	156	static long
c5311781 BG	157	mtrr_ioctl(struct file *file, unsigned int cmd, unsigned long __arg)
1da177e4	158	{
c5311781	159	int err = 0;
1da177e4	160	mtrr_type type;
b263b31e	161	unsigned long base;
365bff80	162	unsigned long size;
1da177e4 LT	163	struct mtrr_sentry sentry;
	164	struct mtrr_gentry gentry;
	165	void __user arg = (void __user ) __arg;
	166
c5311781 BG	167	switch (cmd) {
	168	case MTRRIOC_ADD_ENTRY:
	169	case MTRRIOC_SET_ENTRY:
	170	case MTRRIOC_DEL_ENTRY:
	171	case MTRRIOC_KILL_ENTRY:
	172	case MTRRIOC_ADD_PAGE_ENTRY:
	173	case MTRRIOC_SET_PAGE_ENTRY:
	174	case MTRRIOC_DEL_PAGE_ENTRY:
	175	case MTRRIOC_KILL_PAGE_ENTRY:
	176	if (copy_from_user(&sentry, arg, sizeof sentry))
	177	return -EFAULT;
	178	break;
	179	case MTRRIOC_GET_ENTRY:
	180	case MTRRIOC_GET_PAGE_ENTRY:
	181	if (copy_from_user(&gentry, arg, sizeof gentry))
	182	return -EFAULT;
	183	break;
	184	#ifdef CONFIG_COMPAT
	185	case MTRRIOC32_ADD_ENTRY:
	186	case MTRRIOC32_SET_ENTRY:
	187	case MTRRIOC32_DEL_ENTRY:
	188	case MTRRIOC32_KILL_ENTRY:
	189	case MTRRIOC32_ADD_PAGE_ENTRY:
	190	case MTRRIOC32_SET_PAGE_ENTRY:
	191	case MTRRIOC32_DEL_PAGE_ENTRY:
	192	case MTRRIOC32_KILL_PAGE_ENTRY: {
26dc67ed JSR	193	struct mtrr_sentry32 __user *s32;
	194
	195	s32 = (struct mtrr_sentry32 __user *)__arg;
c5311781 BG	196	err = get_user(sentry.base, &s32->base);
	197	err \|= get_user(sentry.size, &s32->size);
	198	err \|= get_user(sentry.type, &s32->type);
	199	if (err)
	200	return err;
	201	break;
	202	}
	203	case MTRRIOC32_GET_ENTRY:
	204	case MTRRIOC32_GET_PAGE_ENTRY: {
26dc67ed JSR	205	struct mtrr_gentry32 __user *g32;
	206
	207	g32 = (struct mtrr_gentry32 __user *)__arg;
c5311781 BG	208	err = get_user(gentry.regnum, &g32->regnum);
	209	err \|= get_user(gentry.base, &g32->base);
	210	err \|= get_user(gentry.size, &g32->size);
	211	err \|= get_user(gentry.type, &g32->type);
	212	if (err)
	213	return err;
	214	break;
	215	}
	216	#endif
	217	}
	218
1da177e4 LT	219	switch (cmd) {
	220	default:
	221	return -ENOTTY;
	222	case MTRRIOC_ADD_ENTRY:
98838ec9 GP	223	#ifdef CONFIG_COMPAT
	224	case MTRRIOC32_ADD_ENTRY:
	225	#endif
1da177e4 LT	226	if (!capable(CAP_SYS_ADMIN))
1da177e4 LT	227	return -EPERM;
1da177e4	228	err =
2d2ee8de	229	mtrr_file_add(sentry.base, sentry.size, sentry.type, true,
1da177e4	230	file, 0);
1da177e4 LT	231	break;
1da177e4 LT	232	case MTRRIOC_SET_ENTRY:
98838ec9 GP	233	#ifdef CONFIG_COMPAT
	234	case MTRRIOC32_SET_ENTRY:
	235	#endif
1da177e4 LT	236	if (!capable(CAP_SYS_ADMIN))
1da177e4 LT	237	return -EPERM;
2d2ee8de	238	err = mtrr_add(sentry.base, sentry.size, sentry.type, false);
1da177e4 LT	239	break;
1da177e4 LT	240	case MTRRIOC_DEL_ENTRY:
98838ec9 GP	241	#ifdef CONFIG_COMPAT
	242	case MTRRIOC32_DEL_ENTRY:
	243	#endif
1da177e4 LT	244	if (!capable(CAP_SYS_ADMIN))
1da177e4 LT	245	return -EPERM;
1da177e4	246	err = mtrr_file_del(sentry.base, sentry.size, file, 0);
1da177e4 LT	247	break;
1da177e4 LT	248	case MTRRIOC_KILL_ENTRY:
98838ec9 GP	249	#ifdef CONFIG_COMPAT
	250	case MTRRIOC32_KILL_ENTRY:
	251	#endif
1da177e4 LT	252	if (!capable(CAP_SYS_ADMIN))
1da177e4 LT	253	return -EPERM;
1da177e4	254	err = mtrr_del(-1, sentry.base, sentry.size);
1da177e4 LT	255	break;
1da177e4 LT	256	case MTRRIOC_GET_ENTRY:
98838ec9 GP	257	#ifdef CONFIG_COMPAT
	258	case MTRRIOC32_GET_ENTRY:
	259	#endif
1da177e4 LT	260	if (gentry.regnum >= num_var_ranges)
1da177e4 LT	261	return -EINVAL;
b263b31e	262	mtrr_if->get(gentry.regnum, &base, &size, &type);
1da177e4 LT	263
1da177e4 LT	264	/* Hide entries that go above 4GB */
b263b31e	265	if (base + size - 1 >= (1UL << (8 * sizeof(gentry.size) - PAGE_SHIFT))
365bff80	266	\|\| size >= (1UL << (8 * sizeof(gentry.size) - PAGE_SHIFT)))
1da177e4 LT	267	gentry.base = gentry.size = gentry.type = 0;
1da177e4 LT	268	else {
b263b31e	269	gentry.base = base << PAGE_SHIFT;
365bff80	270	gentry.size = size << PAGE_SHIFT;
1da177e4 LT	271	gentry.type = type;
	272	}
	273
1da177e4 LT	274	break;
1da177e4 LT	275	case MTRRIOC_ADD_PAGE_ENTRY:
98838ec9 GP	276	#ifdef CONFIG_COMPAT
	277	case MTRRIOC32_ADD_PAGE_ENTRY:
	278	#endif
1da177e4 LT	279	if (!capable(CAP_SYS_ADMIN))
1da177e4 LT	280	return -EPERM;
1da177e4	281	err =
2d2ee8de	282	mtrr_file_add(sentry.base, sentry.size, sentry.type, true,
1da177e4	283	file, 1);
1da177e4 LT	284	break;
1da177e4 LT	285	case MTRRIOC_SET_PAGE_ENTRY:
98838ec9 GP	286	#ifdef CONFIG_COMPAT
	287	case MTRRIOC32_SET_PAGE_ENTRY:
	288	#endif
1da177e4 LT	289	if (!capable(CAP_SYS_ADMIN))
1da177e4 LT	290	return -EPERM;
2d2ee8de PJ	291	err =
2d2ee8de PJ	292	mtrr_add_page(sentry.base, sentry.size, sentry.type, false);
1da177e4 LT	293	break;
1da177e4 LT	294	case MTRRIOC_DEL_PAGE_ENTRY:
98838ec9 GP	295	#ifdef CONFIG_COMPAT
	296	case MTRRIOC32_DEL_PAGE_ENTRY:
	297	#endif
1da177e4 LT	298	if (!capable(CAP_SYS_ADMIN))
1da177e4 LT	299	return -EPERM;
1da177e4	300	err = mtrr_file_del(sentry.base, sentry.size, file, 1);
1da177e4 LT	301	break;
1da177e4 LT	302	case MTRRIOC_KILL_PAGE_ENTRY:
98838ec9 GP	303	#ifdef CONFIG_COMPAT
	304	case MTRRIOC32_KILL_PAGE_ENTRY:
	305	#endif
1da177e4 LT	306	if (!capable(CAP_SYS_ADMIN))
1da177e4 LT	307	return -EPERM;
1da177e4	308	err = mtrr_del_page(-1, sentry.base, sentry.size);
1da177e4 LT	309	break;
1da177e4 LT	310	case MTRRIOC_GET_PAGE_ENTRY:
98838ec9 GP	311	#ifdef CONFIG_COMPAT
	312	case MTRRIOC32_GET_PAGE_ENTRY:
	313	#endif
1da177e4 LT	314	if (gentry.regnum >= num_var_ranges)
1da177e4 LT	315	return -EINVAL;
b263b31e	316	mtrr_if->get(gentry.regnum, &base, &size, &type);
365bff80 JB	317	/* Hide entries that would overflow */
	318	if (size != (__typeof__(gentry.size))size)
	319	gentry.base = gentry.size = gentry.type = 0;
	320	else {
b263b31e	321	gentry.base = base;
365bff80 JB	322	gentry.size = size;
	323	gentry.type = type;
	324	}
c5311781 BG	325	break;
c5311781 BG	326	}
1da177e4	327
c5311781 BG	328	if (err)
	329	return err;
	330
26dc67ed	331	switch (cmd) {
c5311781 BG	332	case MTRRIOC_GET_ENTRY:
c5311781 BG	333	case MTRRIOC_GET_PAGE_ENTRY:
1da177e4	334	if (copy_to_user(arg, &gentry, sizeof gentry))
c5311781	335	err = -EFAULT;
1da177e4	336	break;
c5311781 BG	337	#ifdef CONFIG_COMPAT
	338	case MTRRIOC32_GET_ENTRY:
	339	case MTRRIOC32_GET_PAGE_ENTRY: {
26dc67ed JSR	340	struct mtrr_gentry32 __user *g32;
	341
	342	g32 = (struct mtrr_gentry32 __user *)__arg;
c5311781 BG	343	err = put_user(gentry.base, &g32->base);
	344	err \|= put_user(gentry.size, &g32->size);
	345	err \|= put_user(gentry.regnum, &g32->regnum);
	346	err \|= put_user(gentry.type, &g32->type);
	347	break;
	348	}
	349	#endif
1da177e4	350	}
c5311781	351	return err;
1da177e4 LT	352	}
1da177e4 LT	353
26dc67ed	354	static int mtrr_close(struct inode ino, struct file file)
1da177e4	355	{
1da177e4	356	unsigned int *fcount = FILE_FCOUNT(file);
26dc67ed	357	int i, max;
1da177e4 LT	358
	359	if (fcount != NULL) {
	360	max = num_var_ranges;
	361	for (i = 0; i < max; ++i) {
	362	while (fcount[i] > 0) {
	363	mtrr_del(i, 0, 0);
	364	--fcount[i];
	365	}
	366	}
	367	kfree(fcount);
	368	FILE_FCOUNT(file) = NULL;
	369	}
	370	return single_release(ino, file);
	371	}
	372
	373	static int mtrr_seq_show(struct seq_file seq, void offset);
	374
	375	static int mtrr_open(struct inode inode, struct file file)
	376	{
26dc67ed	377	if (!mtrr_if)
1da177e4	378	return -EIO;
26dc67ed JSR	379	if (!mtrr_if->get)
26dc67ed JSR	380	return -ENXIO;
1da177e4 LT	381	return single_open(file, mtrr_seq_show, NULL);
	382	}
	383
5dfe4c96	384	static const struct file_operations mtrr_fops = {
26dc67ed JSR	385	.owner = THIS_MODULE,
	386	.open = mtrr_open,
	387	.read = seq_read,
	388	.llseek = seq_lseek,
	389	.write = mtrr_write,
	390	.unlocked_ioctl = mtrr_ioctl,
	391	.compat_ioctl = mtrr_ioctl,
	392	.release = mtrr_close,
1da177e4 LT	393	};
1da177e4 LT	394
1da177e4 LT	395	static int mtrr_seq_show(struct seq_file seq, void offset)
	396	{
	397	char factor;
3ac62bc0	398	int i, max;
1da177e4	399	mtrr_type type;
365bff80	400	unsigned long base, size;
1da177e4	401
1da177e4 LT	402	max = num_var_ranges;
	403	for (i = 0; i < max; i++) {
	404	mtrr_if->get(i, &base, &size, &type);
26dc67ed	405	if (size == 0) {
99fc8d42	406	mtrr_usage_table[i] = 0;
26dc67ed	407	continue;
1da177e4	408	}
26dc67ed JSR	409	if (size < (0x100000 >> PAGE_SHIFT)) {
	410	/* less than 1MB */
	411	factor = 'K';
	412	size <<= PAGE_SHIFT - 10;
	413	} else {
	414	factor = 'M';
	415	size >>= 20 - PAGE_SHIFT;
	416	}
	417	/* Base can be > 32bit */
3ac62bc0 JP	418	seq_printf(seq, "reg%02i: base=0x%06lx000 (%5luMB), size=%5lu%cB, count=%d: %s\n",
	419	i, base, base >> (20 - PAGE_SHIFT),
	420	size, factor,
	421	mtrr_usage_table[i], mtrr_attrib_to_str(type));
1da177e4 LT	422	}
	423	return 0;
	424	}
	425
	426	static int __init mtrr_if_init(void)
	427	{
	428	struct cpuinfo_x86 *c = &boot_cpu_data;
	429
	430	if ((!cpu_has(c, X86_FEATURE_MTRR)) &&
	431	(!cpu_has(c, X86_FEATURE_K6_MTRR)) &&
	432	(!cpu_has(c, X86_FEATURE_CYRIX_ARR)) &&
	433	(!cpu_has(c, X86_FEATURE_CENTAUR_MCR)))
	434	return -ENODEV;
	435
99b76233	436	proc_create("mtrr", S_IWUSR \| S_IRUGO, NULL, &mtrr_fops);
1da177e4 LT	437	return 0;
1da177e4 LT	438	}
1da177e4 LT	439	arch_initcall(mtrr_if_init);
1da177e4 LT	440	#endif /* CONFIG_PROC_FS */