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 const char *const 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 const 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 = kzalloc(max * sizeof *fcount, GFP_KERNEL); 48 if (!fcount) 49 return -ENOMEM; 50 FILE_FCOUNT(file) = fcount; 51 } 52 if (!page) { 53 if ((base & (PAGE_SIZE - 1)) || (size & (PAGE_SIZE - 1))) 54 return -EINVAL; 55 base >>= PAGE_SHIFT; 56 size >>= PAGE_SHIFT; 57 } 58 reg = mtrr_add_page(base, size, type, 1); 59 if (reg >= 0) 60 ++fcount[reg]; 61 return reg; 62 } 63 64 static int 65 mtrr_file_del(unsigned long base, unsigned long size, 66 struct file *file, int page) 67 { 68 int reg; 69 unsigned int *fcount = FILE_FCOUNT(file); 70 71 if (!page) { 72 if ((base & (PAGE_SIZE - 1)) || (size & (PAGE_SIZE - 1))) 73 return -EINVAL; 74 base >>= PAGE_SHIFT; 75 size >>= PAGE_SHIFT; 76 } 77 reg = mtrr_del_page(-1, base, size); 78 if (reg < 0) 79 return reg; 80 if (fcount == NULL) 81 return reg; 82 if (fcount[reg] < 1) 83 return -EINVAL; 84 --fcount[reg]; 85 return reg; 86 } 87 88 /* RED-PEN: seq_file can seek now. this is ignored. */ 89 static ssize_t 90 mtrr_write(struct file *file, const char __user *buf, size_t len, loff_t * ppos) 91 /* Format of control line: 92 "base=%Lx size=%Lx type=%s" OR: 93 "disable=%d" 94 */ 95 { 96 int i, err; 97 unsigned long reg; 98 unsigned long long base, size; 99 char *ptr; 100 char line[LINE_SIZE]; 101 size_t linelen; 102 103 if (!capable(CAP_SYS_ADMIN)) 104 return -EPERM; 105 if (!len) 106 return -EINVAL; 107 memset(line, 0, LINE_SIZE); 108 if (len > LINE_SIZE) 109 len = LINE_SIZE; 110 if (copy_from_user(line, buf, len - 1)) 111 return -EFAULT; 112 linelen = strlen(line); 113 ptr = line + linelen - 1; 114 if (linelen && *ptr == '\n') 115 *ptr = '\0'; 116 if (!strncmp(line, "disable=", 8)) { 117 reg = simple_strtoul(line + 8, &ptr, 0); 118 err = mtrr_del_page(reg, 0, 0); 119 if (err < 0) 120 return err; 121 return len; 122 } 123 if (strncmp(line, "base=", 5)) 124 return -EINVAL; 125 base = simple_strtoull(line + 5, &ptr, 0); 126 for (; isspace(*ptr); ++ptr) ; 127 if (strncmp(ptr, "size=", 5)) 128 return -EINVAL; 129 size = simple_strtoull(ptr + 5, &ptr, 0); 130 if ((base & 0xfff) || (size & 0xfff)) 131 return -EINVAL; 132 for (; isspace(*ptr); ++ptr) ; 133 if (strncmp(ptr, "type=", 5)) 134 return -EINVAL; 135 ptr += 5; 136 for (; isspace(*ptr); ++ptr) ; 137 for (i = 0; i < MTRR_NUM_TYPES; ++i) { 138 if (strcmp(ptr, mtrr_strings[i])) 139 continue; 140 base >>= PAGE_SHIFT; 141 size >>= PAGE_SHIFT; 142 err = 143 mtrr_add_page((unsigned long) base, (unsigned long) size, i, 144 1); 145 if (err < 0) 146 return err; 147 return len; 148 } 149 return -EINVAL; 150 } 151 152 static long 153 mtrr_ioctl(struct file *file, unsigned int cmd, unsigned long __arg) 154 { 155 int err = 0; 156 mtrr_type type; 157 unsigned long size; 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 #ifdef CONFIG_COMPAT 215 case MTRRIOC32_ADD_ENTRY: 216 #endif 217 if (!capable(CAP_SYS_ADMIN)) 218 return -EPERM; 219 err = 220 mtrr_file_add(sentry.base, sentry.size, sentry.type, 1, 221 file, 0); 222 break; 223 case MTRRIOC_SET_ENTRY: 224 #ifdef CONFIG_COMPAT 225 case MTRRIOC32_SET_ENTRY: 226 #endif 227 if (!capable(CAP_SYS_ADMIN)) 228 return -EPERM; 229 err = mtrr_add(sentry.base, sentry.size, sentry.type, 0); 230 break; 231 case MTRRIOC_DEL_ENTRY: 232 #ifdef CONFIG_COMPAT 233 case MTRRIOC32_DEL_ENTRY: 234 #endif 235 if (!capable(CAP_SYS_ADMIN)) 236 return -EPERM; 237 err = mtrr_file_del(sentry.base, sentry.size, file, 0); 238 break; 239 case MTRRIOC_KILL_ENTRY: 240 #ifdef CONFIG_COMPAT 241 case MTRRIOC32_KILL_ENTRY: 242 #endif 243 if (!capable(CAP_SYS_ADMIN)) 244 return -EPERM; 245 err = mtrr_del(-1, sentry.base, sentry.size); 246 break; 247 case MTRRIOC_GET_ENTRY: 248 #ifdef CONFIG_COMPAT 249 case MTRRIOC32_GET_ENTRY: 250 #endif 251 if (gentry.regnum >= num_var_ranges) 252 return -EINVAL; 253 mtrr_if->get(gentry.regnum, &gentry.base, &size, &type); 254 255 /* Hide entries that go above 4GB */ 256 if (gentry.base + size - 1 >= (1UL << (8 * sizeof(gentry.size) - PAGE_SHIFT)) 257 || size >= (1UL << (8 * sizeof(gentry.size) - PAGE_SHIFT))) 258 gentry.base = gentry.size = gentry.type = 0; 259 else { 260 gentry.base <<= PAGE_SHIFT; 261 gentry.size = size << PAGE_SHIFT; 262 gentry.type = type; 263 } 264 265 break; 266 case MTRRIOC_ADD_PAGE_ENTRY: 267 #ifdef CONFIG_COMPAT 268 case MTRRIOC32_ADD_PAGE_ENTRY: 269 #endif 270 if (!capable(CAP_SYS_ADMIN)) 271 return -EPERM; 272 err = 273 mtrr_file_add(sentry.base, sentry.size, sentry.type, 1, 274 file, 1); 275 break; 276 case MTRRIOC_SET_PAGE_ENTRY: 277 #ifdef CONFIG_COMPAT 278 case MTRRIOC32_SET_PAGE_ENTRY: 279 #endif 280 if (!capable(CAP_SYS_ADMIN)) 281 return -EPERM; 282 err = mtrr_add_page(sentry.base, sentry.size, sentry.type, 0); 283 break; 284 case MTRRIOC_DEL_PAGE_ENTRY: 285 #ifdef CONFIG_COMPAT 286 case MTRRIOC32_DEL_PAGE_ENTRY: 287 #endif 288 if (!capable(CAP_SYS_ADMIN)) 289 return -EPERM; 290 err = mtrr_file_del(sentry.base, sentry.size, file, 1); 291 break; 292 case MTRRIOC_KILL_PAGE_ENTRY: 293 #ifdef CONFIG_COMPAT 294 case MTRRIOC32_KILL_PAGE_ENTRY: 295 #endif 296 if (!capable(CAP_SYS_ADMIN)) 297 return -EPERM; 298 err = mtrr_del_page(-1, sentry.base, sentry.size); 299 break; 300 case MTRRIOC_GET_PAGE_ENTRY: 301 #ifdef CONFIG_COMPAT 302 case MTRRIOC32_GET_PAGE_ENTRY: 303 #endif 304 if (gentry.regnum >= num_var_ranges) 305 return -EINVAL; 306 mtrr_if->get(gentry.regnum, &gentry.base, &size, &type); 307 /* Hide entries that would overflow */ 308 if (size != (__typeof__(gentry.size))size) 309 gentry.base = gentry.size = gentry.type = 0; 310 else { 311 gentry.size = size; 312 gentry.type = type; 313 } 314 break; 315 } 316 317 if (err) 318 return err; 319 320 switch(cmd) { 321 case MTRRIOC_GET_ENTRY: 322 case MTRRIOC_GET_PAGE_ENTRY: 323 if (copy_to_user(arg, &gentry, sizeof gentry)) 324 err = -EFAULT; 325 break; 326 #ifdef CONFIG_COMPAT 327 case MTRRIOC32_GET_ENTRY: 328 case MTRRIOC32_GET_PAGE_ENTRY: { 329 struct mtrr_gentry32 __user *g32 = (struct mtrr_gentry32 __user *)__arg; 330 err = put_user(gentry.base, &g32->base); 331 err |= put_user(gentry.size, &g32->size); 332 err |= put_user(gentry.regnum, &g32->regnum); 333 err |= put_user(gentry.type, &g32->type); 334 break; 335 } 336 #endif 337 } 338 return err; 339 } 340 341 static int 342 mtrr_close(struct inode *ino, struct file *file) 343 { 344 int i, max; 345 unsigned int *fcount = FILE_FCOUNT(file); 346 347 if (fcount != NULL) { 348 max = num_var_ranges; 349 for (i = 0; i < max; ++i) { 350 while (fcount[i] > 0) { 351 mtrr_del(i, 0, 0); 352 --fcount[i]; 353 } 354 } 355 kfree(fcount); 356 FILE_FCOUNT(file) = NULL; 357 } 358 return single_release(ino, file); 359 } 360 361 static int mtrr_seq_show(struct seq_file *seq, void *offset); 362 363 static int mtrr_open(struct inode *inode, struct file *file) 364 { 365 if (!mtrr_if) 366 return -EIO; 367 if (!mtrr_if->get) 368 return -ENXIO; 369 return single_open(file, mtrr_seq_show, NULL); 370 } 371 372 static const struct file_operations mtrr_fops = { 373 .owner = THIS_MODULE, 374 .open = mtrr_open, 375 .read = seq_read, 376 .llseek = seq_lseek, 377 .write = mtrr_write, 378 .unlocked_ioctl = mtrr_ioctl, 379 .compat_ioctl = mtrr_ioctl, 380 .release = mtrr_close, 381 }; 382 383 384 static struct proc_dir_entry *proc_root_mtrr; 385 386 387 static int mtrr_seq_show(struct seq_file *seq, void *offset) 388 { 389 char factor; 390 int i, max, len; 391 mtrr_type type; 392 unsigned long base, size; 393 394 len = 0; 395 max = num_var_ranges; 396 for (i = 0; i < max; i++) { 397 mtrr_if->get(i, &base, &size, &type); 398 if (size == 0) 399 usage_table[i] = 0; 400 else { 401 if (size < (0x100000 >> PAGE_SHIFT)) { 402 /* less than 1MB */ 403 factor = 'K'; 404 size <<= PAGE_SHIFT - 10; 405 } else { 406 factor = 'M'; 407 size >>= 20 - PAGE_SHIFT; 408 } 409 /* RED-PEN: base can be > 32bit */ 410 len += seq_printf(seq, 411 "reg%02i: base=0x%05lx000 (%4luMB), size=%4lu%cB: %s, count=%d\n", 412 i, base, base >> (20 - PAGE_SHIFT), size, factor, 413 mtrr_attrib_to_str(type), usage_table[i]); 414 } 415 } 416 return 0; 417 } 418 419 static int __init mtrr_if_init(void) 420 { 421 struct cpuinfo_x86 *c = &boot_cpu_data; 422 423 if ((!cpu_has(c, X86_FEATURE_MTRR)) && 424 (!cpu_has(c, X86_FEATURE_K6_MTRR)) && 425 (!cpu_has(c, X86_FEATURE_CYRIX_ARR)) && 426 (!cpu_has(c, X86_FEATURE_CENTAUR_MCR))) 427 return -ENODEV; 428 429 proc_root_mtrr = 430 create_proc_entry("mtrr", S_IWUSR | S_IRUGO, &proc_root); 431 if (proc_root_mtrr) { 432 proc_root_mtrr->owner = THIS_MODULE; 433 proc_root_mtrr->proc_fops = &mtrr_fops; 434 } 435 return 0; 436 } 437 438 arch_initcall(mtrr_if_init); 439 #endif /* CONFIG_PROC_FS */ 440