block/genhd.c

   1 /*
   2  *  gendisk handling
   3  */
   4
   5 #include <linux/module.h>
   6 #include <linux/fs.h>
   7 #include <linux/genhd.h>
   8 #include <linux/kdev_t.h>
   9 #include <linux/kernel.h>
  10 #include <linux/blkdev.h>
  11 #include <linux/init.h>
  12 #include <linux/spinlock.h>
  13 #include <linux/seq_file.h>
  14 #include <linux/slab.h>
  15 #include <linux/kmod.h>
  16 #include <linux/kobj_map.h>
  17 #include <linux/buffer_head.h>
  18 #include <linux/mutex.h>
  19
  20 #include "blk.h"
  21
  22 static DEFINE_MUTEX(block_class_lock);
  23 #ifndef CONFIG_SYSFS_DEPRECATED
  24 struct kobject *block_depr;
  25 #endif
  26
  27 static struct device_type disk_type;
  28
  29 /*
  30  * Can be deleted altogether. Later.
  31  *
  32  */
  33 static struct blk_major_name {
  34         struct blk_major_name *next;
  35         int major;
  36         char name[16];
  37 } *major_names[BLKDEV_MAJOR_HASH_SIZE];
  38
  39 /* index in the above - for now: assume no multimajor ranges */
  40 static inline int major_to_index(int major)
  41 {
  42         return major % BLKDEV_MAJOR_HASH_SIZE;
  43 }
  44
  45 #ifdef CONFIG_PROC_FS
  46 void blkdev_show(struct seq_file *f, off_t offset)
  47 {
  48         struct blk_major_name *dp;
  49
  50         if (offset < BLKDEV_MAJOR_HASH_SIZE) {
  51                 mutex_lock(&block_class_lock);
  52                 for (dp = major_names[offset]; dp; dp = dp->next)
  53                         seq_printf(f, "%3d %s\n", dp->major, dp->name);
  54                 mutex_unlock(&block_class_lock);
  55         }
  56 }
  57 #endif /* CONFIG_PROC_FS */
  58
  59 int register_blkdev(unsigned int major, const char *name)
  60 {
  61         struct blk_major_name **n, *p;
  62         int index, ret = 0;
  63
  64         mutex_lock(&block_class_lock);
  65
  66         /* temporary */
  67         if (major == 0) {
  68                 for (index = ARRAY_SIZE(major_names)-1; index > 0; index--) {
  69                         if (major_names[index] == NULL)
  70                                 break;
  71                 }
  72
  73                 if (index == 0) {
  74                         printk("register_blkdev: failed to get major for %s\n",
  75                                name);
  76                         ret = -EBUSY;
  77                         goto out;
  78                 }
  79                 major = index;
  80                 ret = major;
  81         }
  82
  83         p = kmalloc(sizeof(struct blk_major_name), GFP_KERNEL);
  84         if (p == NULL) {
  85                 ret = -ENOMEM;
  86                 goto out;
  87         }
  88
  89         p->major = major;
  90         strlcpy(p->name, name, sizeof(p->name));
  91         p->next = NULL;
  92         index = major_to_index(major);
  93
  94         for (n = &major_names[index]; *n; n = &(*n)->next) {
  95                 if ((*n)->major == major)
  96                         break;
  97         }
  98         if (!*n)
  99                 *n = p;
 100         else
 101                 ret = -EBUSY;
 102
 103         if (ret < 0) {
 104                 printk("register_blkdev: cannot get major %d for %s\n",
 105                        major, name);
 106                 kfree(p);
 107         }
 108 out:
 109         mutex_unlock(&block_class_lock);
 110         return ret;
 111 }
 112
 113 EXPORT_SYMBOL(register_blkdev);
 114
 115 void unregister_blkdev(unsigned int major, const char *name)
 116 {
 117         struct blk_major_name **n;
 118         struct blk_major_name *p = NULL;
 119         int index = major_to_index(major);
 120
 121         mutex_lock(&block_class_lock);
 122         for (n = &major_names[index]; *n; n = &(*n)->next)
 123                 if ((*n)->major == major)
 124                         break;
 125         if (!*n || strcmp((*n)->name, name)) {
 126                 WARN_ON(1);
 127         } else {
 128                 p = *n;
 129                 *n = p->next;
 130         }
 131         mutex_unlock(&block_class_lock);
 132         kfree(p);
 133 }
 134
 135 EXPORT_SYMBOL(unregister_blkdev);
 136
 137 static struct kobj_map *bdev_map;
 138
 139 /*
 140  * Register device numbers dev..(dev+range-1)
 141  * range must be nonzero
 142  * The hash chain is sorted on range, so that subranges can override.
 143  */
 144 void blk_register_region(dev_t devt, unsigned long range, struct module *module,
 145                          struct kobject *(*probe)(dev_t, int *, void *),
 146                          int (*lock)(dev_t, void *), void *data)
 147 {
 148         kobj_map(bdev_map, devt, range, module, probe, lock, data);
 149 }
 150
 151 EXPORT_SYMBOL(blk_register_region);
 152
 153 void blk_unregister_region(dev_t devt, unsigned long range)
 154 {
 155         kobj_unmap(bdev_map, devt, range);
 156 }
 157
 158 EXPORT_SYMBOL(blk_unregister_region);
 159
 160 static struct kobject *exact_match(dev_t devt, int *part, void *data)
 161 {
 162         struct gendisk *p = data;
 163
 164         return &p->dev.kobj;
 165 }
 166
 167 static int exact_lock(dev_t devt, void *data)
 168 {
 169         struct gendisk *p = data;
 170
 171         if (!get_disk(p))
 172                 return -1;
 173         return 0;
 174 }
 175
 176 /**
 177  * add_disk - add partitioning information to kernel list
 178  * @disk: per-device partitioning information
 179  *
 180  * This function registers the partitioning information in @disk
 181  * with the kernel.
 182  */
 183 void add_disk(struct gendisk *disk)
 184 {
 185         struct backing_dev_info *bdi;
 186         int retval;
 187
 188         disk->flags |= GENHD_FL_UP;
 189         blk_register_region(MKDEV(disk->major, disk->first_minor),
 190                             disk->minors, NULL, exact_match, exact_lock, disk);
 191         register_disk(disk);
 192         blk_register_queue(disk);
 193         blk_register_filter(disk);
 194
 195         bdi = &disk->queue->backing_dev_info;
 196         bdi_register_dev(bdi, MKDEV(disk->major, disk->first_minor));
 197         retval = sysfs_create_link(&disk->dev.kobj, &bdi->dev->kobj, "bdi");
 198         WARN_ON(retval);
 199 }
 200
 201 EXPORT_SYMBOL(add_disk);
 202 EXPORT_SYMBOL(del_gendisk);     /* in partitions/check.c */
 203
 204 void unlink_gendisk(struct gendisk *disk)
 205 {
 206         blk_unregister_filter(disk);
 207         sysfs_remove_link(&disk->dev.kobj, "bdi");
 208         bdi_unregister(&disk->queue->backing_dev_info);
 209         blk_unregister_queue(disk);
 210         blk_unregister_region(MKDEV(disk->major, disk->first_minor),
 211                               disk->minors);
 212 }
 213
 214 /**
 215  * get_gendisk - get partitioning information for a given device
 216  * @dev: device to get partitioning information for
 217  *
 218  * This function gets the structure containing partitioning
 219  * information for the given device @dev.
 220  */
 221 struct gendisk *get_gendisk(dev_t devt, int *part)
 222 {
 223         struct kobject *kobj = kobj_lookup(bdev_map, devt, part);
 224         struct device *dev = kobj_to_dev(kobj);
 225
 226         return  kobj ? dev_to_disk(dev) : NULL;
 227 }
 228
 229 /*
 230  * print a partitions - intended for places where the root filesystem can't be
 231  * mounted and thus to give the victim some idea of what went wrong
 232  */
 233 static int printk_partition(struct device *dev, void *data)
 234 {
 235         struct gendisk *sgp;
 236         char buf[BDEVNAME_SIZE];
 237         int n;
 238
 239         if (dev->type != &disk_type)
 240                 goto exit;
 241
 242         sgp = dev_to_disk(dev);
 243         /*
 244          * Don't show empty devices or things that have been surpressed
 245          */
 246         if (get_capacity(sgp) == 0 ||
 247             (sgp->flags & GENHD_FL_SUPPRESS_PARTITION_INFO))
 248                 goto exit;
 249
 250         /*
 251          * Note, unlike /proc/partitions, I am showing the numbers in
 252          * hex - the same format as the root= option takes.
 253          */
 254         printk("%02x%02x %10llu %s",
 255                 sgp->major, sgp->first_minor,
 256                 (unsigned long long)get_capacity(sgp) >> 1,
 257                 disk_name(sgp, 0, buf));
 258         if (sgp->driverfs_dev != NULL &&
 259             sgp->driverfs_dev->driver != NULL)
 260                 printk(" driver: %s\n",
 261                         sgp->driverfs_dev->driver->name);
 262         else
 263                 printk(" (driver?)\n");
 264
 265         /* now show the partitions */
 266         for (n = 0; n < sgp->minors - 1; ++n) {
 267                 if (sgp->part[n] == NULL)
 268                         goto exit;
 269                 if (sgp->part[n]->nr_sects == 0)
 270                         goto exit;
 271                 printk("  %02x%02x %10llu %s\n",
 272                         sgp->major, n + 1 + sgp->first_minor,
 273                         (unsigned long long)sgp->part[n]->nr_sects >> 1,
 274                         disk_name(sgp, n + 1, buf));
 275         }
 276 exit:
 277         return 0;
 278 }
 279
 280 /*
 281  * print a full list of all partitions - intended for places where the root
 282  * filesystem can't be mounted and thus to give the victim some idea of what
 283  * went wrong
 284  */
 285 void __init printk_all_partitions(void)
 286 {
 287         mutex_lock(&block_class_lock);
 288         class_for_each_device(&block_class, NULL, NULL, printk_partition);
 289         mutex_unlock(&block_class_lock);
 290 }
 291
 292 #ifdef CONFIG_PROC_FS
 293 /* iterator */
 294 static int find_start(struct device *dev, void *data)
 295 {
 296         loff_t *k = data;
 297
 298         if (dev->type != &disk_type)
 299                 return 0;
 300         if (!*k)
 301                 return 1;
 302         (*k)--;
 303         return 0;
 304 }
 305
 306 static void *part_start(struct seq_file *part, loff_t *pos)
 307 {
 308         struct device *dev;
 309         loff_t k = *pos;
 310
 311         if (!k)
 312                 seq_puts(part, "major minor  #blocks  name\n\n");
 313
 314         mutex_lock(&block_class_lock);
 315         dev = class_find_device(&block_class, NULL, &k, find_start);
 316         if (dev)
 317                 return dev_to_disk(dev);
 318         return NULL;
 319 }
 320
 321 static int find_next(struct device *dev, void *data)
 322 {
 323         if (dev->type == &disk_type)
 324                 return 1;
 325         return 0;
 326 }
 327
 328 static void *part_next(struct seq_file *part, void *v, loff_t *pos)
 329 {
 330         struct gendisk *gp = v;
 331         struct device *dev;
 332         ++*pos;
 333         dev = class_find_device(&block_class, &gp->dev, NULL, find_next);
 334         if (dev)
 335                 return dev_to_disk(dev);
 336         return NULL;
 337 }
 338
 339 static void part_stop(struct seq_file *part, void *v)
 340 {
 341         mutex_unlock(&block_class_lock);
 342 }
 343
 344 static int show_partition(struct seq_file *part, void *v)
 345 {
 346         struct gendisk *sgp = v;
 347         int n;
 348         char buf[BDEVNAME_SIZE];
 349
 350         /* Don't show non-partitionable removeable devices or empty devices */
 351         if (!get_capacity(sgp) ||
 352                         (sgp->minors == 1 && (sgp->flags & GENHD_FL_REMOVABLE)))
 353                 return 0;
 354         if (sgp->flags & GENHD_FL_SUPPRESS_PARTITION_INFO)
 355                 return 0;
 356
 357         /* show the full disk and all non-0 size partitions of it */
 358         seq_printf(part, "%4d  %4d %10llu %s\n",
 359                 sgp->major, sgp->first_minor,
 360                 (unsigned long long)get_capacity(sgp) >> 1,
 361                 disk_name(sgp, 0, buf));
 362         for (n = 0; n < sgp->minors - 1; n++) {
 363                 if (!sgp->part[n])
 364                         continue;
 365                 if (sgp->part[n]->nr_sects == 0)
 366                         continue;
 367                 seq_printf(part, "%4d  %4d %10llu %s\n",
 368                         sgp->major, n + 1 + sgp->first_minor,
 369                         (unsigned long long)sgp->part[n]->nr_sects >> 1 ,
 370                         disk_name(sgp, n + 1, buf));
 371         }
 372
 373         return 0;
 374 }
 375
 376 const struct seq_operations partitions_op = {
 377         .start  = part_start,
 378         .next   = part_next,
 379         .stop   = part_stop,
 380         .show   = show_partition
 381 };
 382 #endif
 383
 384
 385 static struct kobject *base_probe(dev_t devt, int *part, void *data)
 386 {
 387         if (request_module("block-major-%d-%d", MAJOR(devt), MINOR(devt)) > 0)
 388                 /* Make old-style 2.4 aliases work */
 389                 request_module("block-major-%d", MAJOR(devt));
 390         return NULL;
 391 }
 392
 393 static int __init genhd_device_init(void)
 394 {
 395         int error;
 396
 397         block_class.dev_kobj = sysfs_dev_block_kobj;
 398         error = class_register(&block_class);
 399         if (unlikely(error))
 400                 return error;
 401         bdev_map = kobj_map_init(base_probe, &block_class_lock);
 402         blk_dev_init();
 403
 404 #ifndef CONFIG_SYSFS_DEPRECATED
 405         /* create top-level block dir */
 406         block_depr = kobject_create_and_add("block", NULL);
 407 #endif
 408         return 0;
 409 }
 410
 411 subsys_initcall(genhd_device_init);
 412
 413 static ssize_t disk_range_show(struct device *dev,
 414                                struct device_attribute *attr, char *buf)
 415 {
 416         struct gendisk *disk = dev_to_disk(dev);
 417
 418         return sprintf(buf, "%d\n", disk->minors);
 419 }
 420
 421 static ssize_t disk_removable_show(struct device *dev,
 422                                    struct device_attribute *attr, char *buf)
 423 {
 424         struct gendisk *disk = dev_to_disk(dev);
 425
 426         return sprintf(buf, "%d\n",
 427                        (disk->flags & GENHD_FL_REMOVABLE ? 1 : 0));
 428 }
 429
 430 static ssize_t disk_ro_show(struct device *dev,
 431                                    struct device_attribute *attr, char *buf)
 432 {
 433         struct gendisk *disk = dev_to_disk(dev);
 434
 435         return sprintf(buf, "%d\n", disk->policy ? 1 : 0);
 436 }
 437
 438 static ssize_t disk_size_show(struct device *dev,
 439                               struct device_attribute *attr, char *buf)
 440 {
 441         struct gendisk *disk = dev_to_disk(dev);
 442
 443         return sprintf(buf, "%llu\n", (unsigned long long)get_capacity(disk));
 444 }
 445
 446 static ssize_t disk_capability_show(struct device *dev,
 447                                     struct device_attribute *attr, char *buf)
 448 {
 449         struct gendisk *disk = dev_to_disk(dev);
 450
 451         return sprintf(buf, "%x\n", disk->flags);
 452 }
 453
 454 static ssize_t disk_stat_show(struct device *dev,
 455                               struct device_attribute *attr, char *buf)
 456 {
 457         struct gendisk *disk = dev_to_disk(dev);
 458
 459         preempt_disable();
 460         disk_round_stats(disk);
 461         preempt_enable();
 462         return sprintf(buf,
 463                 "%8lu %8lu %8llu %8u "
 464                 "%8lu %8lu %8llu %8u "
 465                 "%8u %8u %8u"
 466                 "\n",
 467                 disk_stat_read(disk, ios[READ]),
 468                 disk_stat_read(disk, merges[READ]),
 469                 (unsigned long long)disk_stat_read(disk, sectors[READ]),
 470                 jiffies_to_msecs(disk_stat_read(disk, ticks[READ])),
 471                 disk_stat_read(disk, ios[WRITE]),
 472                 disk_stat_read(disk, merges[WRITE]),
 473                 (unsigned long long)disk_stat_read(disk, sectors[WRITE]),
 474                 jiffies_to_msecs(disk_stat_read(disk, ticks[WRITE])),
 475                 disk->in_flight,
 476                 jiffies_to_msecs(disk_stat_read(disk, io_ticks)),
 477                 jiffies_to_msecs(disk_stat_read(disk, time_in_queue)));
 478 }
 479
 480 #ifdef CONFIG_FAIL_MAKE_REQUEST
 481 static ssize_t disk_fail_show(struct device *dev,
 482                               struct device_attribute *attr, char *buf)
 483 {
 484         struct gendisk *disk = dev_to_disk(dev);
 485
 486         return sprintf(buf, "%d\n", disk->flags & GENHD_FL_FAIL ? 1 : 0);
 487 }
 488
 489 static ssize_t disk_fail_store(struct device *dev,
 490                                struct device_attribute *attr,
 491                                const char *buf, size_t count)
 492 {
 493         struct gendisk *disk = dev_to_disk(dev);
 494         int i;
 495
 496         if (count > 0 && sscanf(buf, "%d", &i) > 0) {
 497                 if (i == 0)
 498                         disk->flags &= ~GENHD_FL_FAIL;
 499                 else
 500                         disk->flags |= GENHD_FL_FAIL;
 501         }
 502
 503         return count;
 504 }
 505
 506 #endif
 507
 508 static DEVICE_ATTR(range, S_IRUGO, disk_range_show, NULL);
 509 static DEVICE_ATTR(removable, S_IRUGO, disk_removable_show, NULL);
 510 static DEVICE_ATTR(ro, S_IRUGO, disk_ro_show, NULL);
 511 static DEVICE_ATTR(size, S_IRUGO, disk_size_show, NULL);
 512 static DEVICE_ATTR(capability, S_IRUGO, disk_capability_show, NULL);
 513 static DEVICE_ATTR(stat, S_IRUGO, disk_stat_show, NULL);
 514 #ifdef CONFIG_FAIL_MAKE_REQUEST
 515 static struct device_attribute dev_attr_fail =
 516         __ATTR(make-it-fail, S_IRUGO|S_IWUSR, disk_fail_show, disk_fail_store);
 517 #endif
 518
 519 static struct attribute *disk_attrs[] = {
 520         &dev_attr_range.attr,
 521         &dev_attr_removable.attr,
 522         &dev_attr_ro.attr,
 523         &dev_attr_size.attr,
 524         &dev_attr_capability.attr,
 525         &dev_attr_stat.attr,
 526 #ifdef CONFIG_FAIL_MAKE_REQUEST
 527         &dev_attr_fail.attr,
 528 #endif
 529         NULL
 530 };
 531
 532 static struct attribute_group disk_attr_group = {
 533         .attrs = disk_attrs,
 534 };
 535
 536 static struct attribute_group *disk_attr_groups[] = {
 537         &disk_attr_group,
 538         NULL
 539 };
 540
 541 static void disk_release(struct device *dev)
 542 {
 543         struct gendisk *disk = dev_to_disk(dev);
 544
 545         kfree(disk->random);
 546         kfree(disk->part);
 547         free_disk_stats(disk);
 548         kfree(disk);
 549 }
 550 struct class block_class = {
 551         .name           = "block",
 552 };
 553
 554 static struct device_type disk_type = {
 555         .name           = "disk",
 556         .groups         = disk_attr_groups,
 557         .release        = disk_release,
 558 };
 559
 560 #ifdef CONFIG_PROC_FS
 561 /*
 562  * aggregate disk stat collector.  Uses the same stats that the sysfs
 563  * entries do, above, but makes them available through one seq_file.
 564  *
 565  * The output looks suspiciously like /proc/partitions with a bunch of
 566  * extra fields.
 567  */
 568
 569 static void *diskstats_start(struct seq_file *part, loff_t *pos)
 570 {
 571         struct device *dev;
 572         loff_t k = *pos;
 573
 574         mutex_lock(&block_class_lock);
 575         dev = class_find_device(&block_class, NULL, &k, find_start);
 576         if (dev)
 577                 return dev_to_disk(dev);
 578         return NULL;
 579 }
 580
 581 static void *diskstats_next(struct seq_file *part, void *v, loff_t *pos)
 582 {
 583         struct gendisk *gp = v;
 584         struct device *dev;
 585
 586         ++*pos;
 587         dev = class_find_device(&block_class, &gp->dev, NULL, find_next);
 588         if (dev)
 589                 return dev_to_disk(dev);
 590         return NULL;
 591 }
 592
 593 static void diskstats_stop(struct seq_file *part, void *v)
 594 {
 595         mutex_unlock(&block_class_lock);
 596 }
 597
 598 static int diskstats_show(struct seq_file *s, void *v)
 599 {
 600         struct gendisk *gp = v;
 601         char buf[BDEVNAME_SIZE];
 602         int n = 0;
 603
 604         /*
 605         if (&gp->dev.kobj.entry == block_class.devices.next)
 606                 seq_puts(s,     "major minor name"
 607                                 "     rio rmerge rsect ruse wio wmerge "
 608                                 "wsect wuse running use aveq"
 609                                 "\n\n");
 610         */
 611
 612         preempt_disable();
 613         disk_round_stats(gp);
 614         preempt_enable();
 615         seq_printf(s, "%4d %4d %s %lu %lu %llu %u %lu %lu %llu %u %u %u %u\n",
 616                 gp->major, n + gp->first_minor, disk_name(gp, n, buf),
 617                 disk_stat_read(gp, ios[0]), disk_stat_read(gp, merges[0]),
 618                 (unsigned long long)disk_stat_read(gp, sectors[0]),
 619                 jiffies_to_msecs(disk_stat_read(gp, ticks[0])),
 620                 disk_stat_read(gp, ios[1]), disk_stat_read(gp, merges[1]),
 621                 (unsigned long long)disk_stat_read(gp, sectors[1]),
 622                 jiffies_to_msecs(disk_stat_read(gp, ticks[1])),
 623                 gp->in_flight,
 624                 jiffies_to_msecs(disk_stat_read(gp, io_ticks)),
 625                 jiffies_to_msecs(disk_stat_read(gp, time_in_queue)));
 626
 627         /* now show all non-0 size partitions of it */
 628         for (n = 0; n < gp->minors - 1; n++) {
 629                 struct hd_struct *hd = gp->part[n];
 630
 631                 if (!hd || !hd->nr_sects)
 632                         continue;
 633
 634                 preempt_disable();
 635                 part_round_stats(hd);
 636                 preempt_enable();
 637                 seq_printf(s, "%4d %4d %s %lu %lu %llu "
 638                            "%u %lu %lu %llu %u %u %u %u\n",
 639                            gp->major, n + gp->first_minor + 1,
 640                            disk_name(gp, n + 1, buf),
 641                            part_stat_read(hd, ios[0]),
 642                            part_stat_read(hd, merges[0]),
 643                            (unsigned long long)part_stat_read(hd, sectors[0]),
 644                            jiffies_to_msecs(part_stat_read(hd, ticks[0])),
 645                            part_stat_read(hd, ios[1]),
 646                            part_stat_read(hd, merges[1]),
 647                            (unsigned long long)part_stat_read(hd, sectors[1]),
 648                            jiffies_to_msecs(part_stat_read(hd, ticks[1])),
 649                            hd->in_flight,
 650                            jiffies_to_msecs(part_stat_read(hd, io_ticks)),
 651                            jiffies_to_msecs(part_stat_read(hd, time_in_queue))
 652                         );
 653         }
 654
 655         return 0;
 656 }
 657
 658 const struct seq_operations diskstats_op = {
 659         .start  = diskstats_start,
 660         .next   = diskstats_next,
 661         .stop   = diskstats_stop,
 662         .show   = diskstats_show
 663 };
 664 #endif /* CONFIG_PROC_FS */
 665
 666 static void media_change_notify_thread(struct work_struct *work)
 667 {
 668         struct gendisk *gd = container_of(work, struct gendisk, async_notify);
 669         char event[] = "MEDIA_CHANGE=1";
 670         char *envp[] = { event, NULL };
 671
 672         /*
 673          * set enviroment vars to indicate which event this is for
 674          * so that user space will know to go check the media status.
 675          */
 676         kobject_uevent_env(&gd->dev.kobj, KOBJ_CHANGE, envp);
 677         put_device(gd->driverfs_dev);
 678 }
 679
 680 #if 0
 681 void genhd_media_change_notify(struct gendisk *disk)
 682 {
 683         get_device(disk->driverfs_dev);
 684         schedule_work(&disk->async_notify);
 685 }
 686 EXPORT_SYMBOL_GPL(genhd_media_change_notify);
 687 #endif  /*  0  */
 688
 689 struct find_block {
 690         const char *name;
 691         int part;
 692 };
 693
 694 static int match_id(struct device *dev, void *data)
 695 {
 696         struct find_block *find = data;
 697
 698         if (dev->type != &disk_type)
 699                 return 0;
 700         if (strcmp(dev->bus_id, find->name) == 0) {
 701                 struct gendisk *disk = dev_to_disk(dev);
 702                 if (find->part < disk->minors)
 703                         return 1;
 704         }
 705         return 0;
 706 }
 707
 708 dev_t blk_lookup_devt(const char *name, int part)
 709 {
 710         struct device *dev;
 711         dev_t devt = MKDEV(0, 0);
 712         struct find_block find;
 713
 714         mutex_lock(&block_class_lock);
 715         find.name = name;
 716         find.part = part;
 717         dev = class_find_device(&block_class, NULL, (void *)&find, match_id);
 718         if (dev)
 719                 devt = MKDEV(MAJOR(dev->devt),
 720                              MINOR(dev->devt) + part);
 721         mutex_unlock(&block_class_lock);
 722
 723         return devt;
 724 }
 725 EXPORT_SYMBOL(blk_lookup_devt);
 726
 727 struct gendisk *alloc_disk(int minors)
 728 {
 729         return alloc_disk_node(minors, -1);
 730 }
 731
 732 struct gendisk *alloc_disk_node(int minors, int node_id)
 733 {
 734         struct gendisk *disk;
 735
 736         disk = kmalloc_node(sizeof(struct gendisk),
 737                                 GFP_KERNEL | __GFP_ZERO, node_id);
 738         if (disk) {
 739                 if (!init_disk_stats(disk)) {
 740                         kfree(disk);
 741                         return NULL;
 742                 }
 743                 if (minors > 1) {
 744                         int size = (minors - 1) * sizeof(struct hd_struct *);
 745                         disk->part = kmalloc_node(size,
 746                                 GFP_KERNEL | __GFP_ZERO, node_id);
 747                         if (!disk->part) {
 748                                 free_disk_stats(disk);
 749                                 kfree(disk);
 750                                 return NULL;
 751                         }
 752                 }
 753                 disk->minors = minors;
 754                 rand_initialize_disk(disk);
 755                 disk->dev.class = &block_class;
 756                 disk->dev.type = &disk_type;
 757                 device_initialize(&disk->dev);
 758                 INIT_WORK(&disk->async_notify,
 759                         media_change_notify_thread);
 760         }
 761         return disk;
 762 }
 763
 764 EXPORT_SYMBOL(alloc_disk);
 765 EXPORT_SYMBOL(alloc_disk_node);
 766
 767 struct kobject *get_disk(struct gendisk *disk)
 768 {
 769         struct module *owner;
 770         struct kobject *kobj;
 771
 772         if (!disk->fops)
 773                 return NULL;
 774         owner = disk->fops->owner;
 775         if (owner && !try_module_get(owner))
 776                 return NULL;
 777         kobj = kobject_get(&disk->dev.kobj);
 778         if (kobj == NULL) {
 779                 module_put(owner);
 780                 return NULL;
 781         }
 782         return kobj;
 783
 784 }
 785
 786 EXPORT_SYMBOL(get_disk);
 787
 788 void put_disk(struct gendisk *disk)
 789 {
 790         if (disk)
 791                 kobject_put(&disk->dev.kobj);
 792 }
 793
 794 EXPORT_SYMBOL(put_disk);
 795
 796 void set_device_ro(struct block_device *bdev, int flag)
 797 {
 798         if (bdev->bd_contains != bdev)
 799                 bdev->bd_part->policy = flag;
 800         else
 801                 bdev->bd_disk->policy = flag;
 802 }
 803
 804 EXPORT_SYMBOL(set_device_ro);
 805
 806 void set_disk_ro(struct gendisk *disk, int flag)
 807 {
 808         int i;
 809         disk->policy = flag;
 810         for (i = 0; i < disk->minors - 1; i++)
 811                 if (disk->part[i]) disk->part[i]->policy = flag;
 812 }
 813
 814 EXPORT_SYMBOL(set_disk_ro);
 815
 816 int bdev_read_only(struct block_device *bdev)
 817 {
 818         if (!bdev)
 819                 return 0;
 820         else if (bdev->bd_contains != bdev)
 821                 return bdev->bd_part->policy;
 822         else
 823                 return bdev->bd_disk->policy;
 824 }
 825
 826 EXPORT_SYMBOL(bdev_read_only);
 827
 828 int invalidate_partition(struct gendisk *disk, int index)
 829 {
 830         int res = 0;
 831         struct block_device *bdev = bdget_disk(disk, index);
 832         if (bdev) {
 833                 fsync_bdev(bdev);
 834                 res = __invalidate_device(bdev);
 835                 bdput(bdev);
 836         }
 837         return res;
 838 }
 839
 840 EXPORT_SYMBOL(invalidate_partition);