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

#include <linux/capability.h>
#include <linux/seq_file.h>
#include <linux/uaccess.h>
#include <linux/proc_fs.h>
#include <linux/module.h>
#include <linux/ctype.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 = kzalloc(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];
	size_t linelen;

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

	memset(line, 0, LINE_SIZE);
	if (len > LINE_SIZE)
		len = LINE_SIZE;
	if (copy_from_user(line, buf, len - 1))
		return -EFAULT;

	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);
	for (; isspace(*ptr); ++ptr)
		;

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

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

	if (strncmp(ptr, "type=", 5))
		return -EINVAL;
	ptr += 5;
	for (; isspace(*ptr); ++ptr)
		;

	for (i = 0; i < MTRR_NUM_TYPES; ++i) {
		if (strcmp(ptr, mtrr_strings[i]))
			continue;
		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;
	}
	return -EINVAL;
}

static long
mtrr_ioctl(struct file *file, unsigned int cmd, unsigned long __arg)
{
	int err = 0;
	mtrr_type type;
	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, &gentry.base, &size, &type);

		/* Hide entries that go above 4GB */
		if (gentry.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 <<= 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, &gentry.base, &size, &type);
		/* Hide entries that would overflow */
		if (size != (__typeof__(gentry.size))size)
			gentry.base = gentry.size = gentry.type = 0;
		else {
			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, len;
	mtrr_type type;
	unsigned long base, size;

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