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