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 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         unsigned int compat_cmd = cmd;
 162
 163         switch (compat_cmd) {
 164         case MTRRIOC_ADD_ENTRY:
 165         case MTRRIOC_SET_ENTRY:
 166         case MTRRIOC_DEL_ENTRY:
 167         case MTRRIOC_KILL_ENTRY:
 168         case MTRRIOC_ADD_PAGE_ENTRY:
 169         case MTRRIOC_SET_PAGE_ENTRY:
 170         case MTRRIOC_DEL_PAGE_ENTRY:
 171         case MTRRIOC_KILL_PAGE_ENTRY:
 172                 if (copy_from_user(&sentry, arg, sizeof sentry))
 173                         return -EFAULT;
 174                 break;
 175         case MTRRIOC_GET_ENTRY:
 176         case MTRRIOC_GET_PAGE_ENTRY:
 177                 if (copy_from_user(&gentry, arg, sizeof gentry))
 178                         return -EFAULT;
 179                 break;
 180 #ifdef CONFIG_COMPAT
 181 #define MTRR_COMPAT_OP(op, type)\
 182         case MTRRIOC32_##op:    \
 183         cmd = MTRRIOC_##op;     \
 184         goto compat_get_##type
 185
 186         MTRR_COMPAT_OP(ADD_ENTRY, sentry);
 187         MTRR_COMPAT_OP(SET_ENTRY, sentry);
 188         MTRR_COMPAT_OP(DEL_ENTRY, sentry);
 189         MTRR_COMPAT_OP(KILL_ENTRY, sentry);
 190         MTRR_COMPAT_OP(ADD_PAGE_ENTRY, sentry);
 191         MTRR_COMPAT_OP(SET_PAGE_ENTRY, sentry);
 192         MTRR_COMPAT_OP(DEL_PAGE_ENTRY, sentry);
 193         MTRR_COMPAT_OP(KILL_PAGE_ENTRY, sentry);
 194 compat_get_sentry: {
 195                 struct mtrr_sentry32 __user *s32 = (struct mtrr_sentry32 __user *)__arg;
 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         MTRR_COMPAT_OP(GET_ENTRY, gentry);
 204         MTRR_COMPAT_OP(GET_PAGE_ENTRY, gentry);
 205 compat_get_gentry: {
 206                 struct mtrr_gentry32 __user *g32 = (struct mtrr_gentry32 __user *)__arg;
 207                 err = get_user(gentry.regnum, &g32->regnum);
 208                 err |= get_user(gentry.base, &g32->base);
 209                 err |= get_user(gentry.size, &g32->size);
 210                 err |= get_user(gentry.type, &g32->type);
 211                 if (err)
 212                         return err;
 213                 break;
 214         }
 215 #undef MTRR_COMPAT_OP
 216 #endif
 217         }
 218
 219         switch (cmd) {
 220         default:
 221                 return -ENOTTY;
 222         case MTRRIOC_ADD_ENTRY:
 223                 if (!capable(CAP_SYS_ADMIN))
 224                         return -EPERM;
 225                 err =
 226                     mtrr_file_add(sentry.base, sentry.size, sentry.type, 1,
 227                                   file, 0);
 228                 break;
 229         case MTRRIOC_SET_ENTRY:
 230                 if (!capable(CAP_SYS_ADMIN))
 231                         return -EPERM;
 232                 err = mtrr_add(sentry.base, sentry.size, sentry.type, 0);
 233                 break;
 234         case MTRRIOC_DEL_ENTRY:
 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                 if (!capable(CAP_SYS_ADMIN))
 241                         return -EPERM;
 242                 err = mtrr_del(-1, sentry.base, sentry.size);
 243                 break;
 244         case MTRRIOC_GET_ENTRY:
 245                 if (gentry.regnum >= num_var_ranges)
 246                         return -EINVAL;
 247                 mtrr_if->get(gentry.regnum, &gentry.base, &size, &type);
 248
 249                 /* Hide entries that go above 4GB */
 250                 if (gentry.base + size - 1 >= (1UL << (8 * sizeof(gentry.size) - PAGE_SHIFT))
 251                     || size >= (1UL << (8 * sizeof(gentry.size) - PAGE_SHIFT)))
 252                         gentry.base = gentry.size = gentry.type = 0;
 253                 else {
 254                         gentry.base <<= PAGE_SHIFT;
 255                         gentry.size = size << PAGE_SHIFT;
 256                         gentry.type = type;
 257                 }
 258
 259                 break;
 260         case MTRRIOC_ADD_PAGE_ENTRY:
 261                 if (!capable(CAP_SYS_ADMIN))
 262                         return -EPERM;
 263                 err =
 264                     mtrr_file_add(sentry.base, sentry.size, sentry.type, 1,
 265                                   file, 1);
 266                 break;
 267         case MTRRIOC_SET_PAGE_ENTRY:
 268                 if (!capable(CAP_SYS_ADMIN))
 269                         return -EPERM;
 270                 err = mtrr_add_page(sentry.base, sentry.size, sentry.type, 0);
 271                 break;
 272         case MTRRIOC_DEL_PAGE_ENTRY:
 273                 if (!capable(CAP_SYS_ADMIN))
 274                         return -EPERM;
 275                 err = mtrr_file_del(sentry.base, sentry.size, file, 1);
 276                 break;
 277         case MTRRIOC_KILL_PAGE_ENTRY:
 278                 if (!capable(CAP_SYS_ADMIN))
 279                         return -EPERM;
 280                 err = mtrr_del_page(-1, sentry.base, sentry.size);
 281                 break;
 282         case MTRRIOC_GET_PAGE_ENTRY:
 283                 if (gentry.regnum >= num_var_ranges)
 284                         return -EINVAL;
 285                 mtrr_if->get(gentry.regnum, &gentry.base, &size, &type);
 286                 /* Hide entries that would overflow */
 287                 if (size != (__typeof__(gentry.size))size)
 288                         gentry.base = gentry.size = gentry.type = 0;
 289                 else {
 290                         gentry.size = size;
 291                         gentry.type = type;
 292                 }
 293                 break;
 294         }
 295
 296         if (err)
 297                 return err;
 298
 299         switch(compat_cmd) {
 300         case MTRRIOC_GET_ENTRY:
 301         case MTRRIOC_GET_PAGE_ENTRY:
 302                 if (copy_to_user(arg, &gentry, sizeof gentry))
 303                         err = -EFAULT;
 304                 break;
 305 #ifdef CONFIG_COMPAT
 306         case MTRRIOC32_GET_ENTRY:
 307         case MTRRIOC32_GET_PAGE_ENTRY: {
 308                 struct mtrr_gentry32 __user *g32 = (struct mtrr_gentry32 __user *)__arg;
 309                 err = put_user(gentry.base, &g32->base);
 310                 err |= put_user(gentry.size, &g32->size);
 311                 err |= put_user(gentry.regnum, &g32->regnum);
 312                 err |= put_user(gentry.type, &g32->type);
 313                 break;
 314         }
 315 #endif
 316         }
 317         return err;
 318 }
 319
 320 static int
 321 mtrr_close(struct inode *ino, struct file *file)
 322 {
 323         int i, max;
 324         unsigned int *fcount = FILE_FCOUNT(file);
 325
 326         if (fcount != NULL) {
 327                 max = num_var_ranges;
 328                 for (i = 0; i < max; ++i) {
 329                         while (fcount[i] > 0) {
 330                                 mtrr_del(i, 0, 0);
 331                                 --fcount[i];
 332                         }
 333                 }
 334                 kfree(fcount);
 335                 FILE_FCOUNT(file) = NULL;
 336         }
 337         return single_release(ino, file);
 338 }
 339
 340 static int mtrr_seq_show(struct seq_file *seq, void *offset);
 341
 342 static int mtrr_open(struct inode *inode, struct file *file)
 343 {
 344         if (!mtrr_if)
 345                 return -EIO;
 346         if (!mtrr_if->get)
 347                 return -ENXIO;
 348         return single_open(file, mtrr_seq_show, NULL);
 349 }
 350
 351 static struct file_operations mtrr_fops = {
 352         .owner   = THIS_MODULE,
 353         .open    = mtrr_open,
 354         .read    = seq_read,
 355         .llseek  = seq_lseek,
 356         .write   = mtrr_write,
 357         .unlocked_ioctl = mtrr_ioctl,
 358         .compat_ioctl = mtrr_ioctl,
 359         .release = mtrr_close,
 360 };
 361
 362
 363 static struct proc_dir_entry *proc_root_mtrr;
 364
 365
 366 static int mtrr_seq_show(struct seq_file *seq, void *offset)
 367 {
 368         char factor;
 369         int i, max, len;
 370         mtrr_type type;
 371         unsigned long base, size;
 372
 373         len = 0;
 374         max = num_var_ranges;
 375         for (i = 0; i < max; i++) {
 376                 mtrr_if->get(i, &base, &size, &type);
 377                 if (size == 0)
 378                         usage_table[i] = 0;
 379                 else {
 380                         if (size < (0x100000 >> PAGE_SHIFT)) {
 381                                 /* less than 1MB */
 382                                 factor = 'K';
 383                                 size <<= PAGE_SHIFT - 10;
 384                         } else {
 385                                 factor = 'M';
 386                                 size >>= 20 - PAGE_SHIFT;
 387                         }
 388                         /* RED-PEN: base can be > 32bit */
 389                         len += seq_printf(seq,
 390                                    "reg%02i: base=0x%05lx000 (%4luMB), size=%4lu%cB: %s, count=%d\n",
 391                              i, base, base >> (20 - PAGE_SHIFT), size, factor,
 392                              mtrr_attrib_to_str(type), usage_table[i]);
 393                 }
 394         }
 395         return 0;
 396 }
 397
 398 static int __init mtrr_if_init(void)
 399 {
 400         struct cpuinfo_x86 *c = &boot_cpu_data;
 401
 402         if ((!cpu_has(c, X86_FEATURE_MTRR)) &&
 403             (!cpu_has(c, X86_FEATURE_K6_MTRR)) &&
 404             (!cpu_has(c, X86_FEATURE_CYRIX_ARR)) &&
 405             (!cpu_has(c, X86_FEATURE_CENTAUR_MCR)))
 406                 return -ENODEV;
 407
 408         proc_root_mtrr =
 409             create_proc_entry("mtrr", S_IWUSR | S_IRUGO, &proc_root);
 410         if (proc_root_mtrr) {
 411                 proc_root_mtrr->owner = THIS_MODULE;
 412                 proc_root_mtrr->proc_fops = &mtrr_fops;
 413         }
 414         return 0;
 415 }
 416
 417 arch_initcall(mtrr_if_init);
 418 #endif                  /*  CONFIG_PROC_FS  */