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