]> www.pilppa.org Git - linux-2.6-omap-h63xx.git/blob - block/genhd.c
f03bdadc52ac0488ea227db1eb4651ca94de2e58
[linux-2.6-omap-h63xx.git] / 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 = *(loff_t *)data;
297
298         if (dev->type != &disk_type)
299                 return 0;
300         if (!k--)
301                 return 1;
302         return 0;
303 }
304
305 static void *part_start(struct seq_file *part, loff_t *pos)
306 {
307         struct device *dev;
308
309         mutex_lock(&block_class_lock);
310         dev = class_find_device(&block_class, NULL, (void *)pos, find_start);
311         if (dev)
312                 return dev_to_disk(dev);
313         return NULL;
314 }
315
316 static void *part_next(struct seq_file *part, void *v, loff_t *pos)
317 {
318         struct gendisk *gp = v;
319         struct device *dev;
320         ++*pos;
321         list_for_each_entry(dev, &gp->dev.node, node) {
322                 if (&dev->node == &block_class.devices)
323                         return NULL;
324                 if (dev->type == &disk_type)
325                         return dev_to_disk(dev);
326         }
327         return NULL;
328 }
329
330 static void part_stop(struct seq_file *part, void *v)
331 {
332         mutex_unlock(&block_class_lock);
333 }
334
335 static int show_partition(struct seq_file *part, void *v)
336 {
337         struct gendisk *sgp = v;
338         int n;
339         char buf[BDEVNAME_SIZE];
340
341         if (&sgp->dev.node == block_class.devices.next)
342                 seq_puts(part, "major minor  #blocks  name\n\n");
343
344         /* Don't show non-partitionable removeable devices or empty devices */
345         if (!get_capacity(sgp) ||
346                         (sgp->minors == 1 && (sgp->flags & GENHD_FL_REMOVABLE)))
347                 return 0;
348         if (sgp->flags & GENHD_FL_SUPPRESS_PARTITION_INFO)
349                 return 0;
350
351         /* show the full disk and all non-0 size partitions of it */
352         seq_printf(part, "%4d  %4d %10llu %s\n",
353                 sgp->major, sgp->first_minor,
354                 (unsigned long long)get_capacity(sgp) >> 1,
355                 disk_name(sgp, 0, buf));
356         for (n = 0; n < sgp->minors - 1; n++) {
357                 if (!sgp->part[n])
358                         continue;
359                 if (sgp->part[n]->nr_sects == 0)
360                         continue;
361                 seq_printf(part, "%4d  %4d %10llu %s\n",
362                         sgp->major, n + 1 + sgp->first_minor,
363                         (unsigned long long)sgp->part[n]->nr_sects >> 1 ,
364                         disk_name(sgp, n + 1, buf));
365         }
366
367         return 0;
368 }
369
370 const struct seq_operations partitions_op = {
371         .start  = part_start,
372         .next   = part_next,
373         .stop   = part_stop,
374         .show   = show_partition
375 };
376 #endif
377
378
379 static struct kobject *base_probe(dev_t devt, int *part, void *data)
380 {
381         if (request_module("block-major-%d-%d", MAJOR(devt), MINOR(devt)) > 0)
382                 /* Make old-style 2.4 aliases work */
383                 request_module("block-major-%d", MAJOR(devt));
384         return NULL;
385 }
386
387 static int __init genhd_device_init(void)
388 {
389         int error;
390
391         block_class.dev_kobj = sysfs_dev_block_kobj;
392         error = class_register(&block_class);
393         if (unlikely(error))
394                 return error;
395         bdev_map = kobj_map_init(base_probe, &block_class_lock);
396         blk_dev_init();
397
398 #ifndef CONFIG_SYSFS_DEPRECATED
399         /* create top-level block dir */
400         block_depr = kobject_create_and_add("block", NULL);
401 #endif
402         return 0;
403 }
404
405 subsys_initcall(genhd_device_init);
406
407 static ssize_t disk_range_show(struct device *dev,
408                                struct device_attribute *attr, char *buf)
409 {
410         struct gendisk *disk = dev_to_disk(dev);
411
412         return sprintf(buf, "%d\n", disk->minors);
413 }
414
415 static ssize_t disk_removable_show(struct device *dev,
416                                    struct device_attribute *attr, char *buf)
417 {
418         struct gendisk *disk = dev_to_disk(dev);
419
420         return sprintf(buf, "%d\n",
421                        (disk->flags & GENHD_FL_REMOVABLE ? 1 : 0));
422 }
423
424 static ssize_t disk_ro_show(struct device *dev,
425                                    struct device_attribute *attr, char *buf)
426 {
427         struct gendisk *disk = dev_to_disk(dev);
428
429         return sprintf(buf, "%d\n", disk->policy ? 1 : 0);
430 }
431
432 static ssize_t disk_size_show(struct device *dev,
433                               struct device_attribute *attr, char *buf)
434 {
435         struct gendisk *disk = dev_to_disk(dev);
436
437         return sprintf(buf, "%llu\n", (unsigned long long)get_capacity(disk));
438 }
439
440 static ssize_t disk_capability_show(struct device *dev,
441                                     struct device_attribute *attr, char *buf)
442 {
443         struct gendisk *disk = dev_to_disk(dev);
444
445         return sprintf(buf, "%x\n", disk->flags);
446 }
447
448 static ssize_t disk_stat_show(struct device *dev,
449                               struct device_attribute *attr, char *buf)
450 {
451         struct gendisk *disk = dev_to_disk(dev);
452
453         preempt_disable();
454         disk_round_stats(disk);
455         preempt_enable();
456         return sprintf(buf,
457                 "%8lu %8lu %8llu %8u "
458                 "%8lu %8lu %8llu %8u "
459                 "%8u %8u %8u"
460                 "\n",
461                 disk_stat_read(disk, ios[READ]),
462                 disk_stat_read(disk, merges[READ]),
463                 (unsigned long long)disk_stat_read(disk, sectors[READ]),
464                 jiffies_to_msecs(disk_stat_read(disk, ticks[READ])),
465                 disk_stat_read(disk, ios[WRITE]),
466                 disk_stat_read(disk, merges[WRITE]),
467                 (unsigned long long)disk_stat_read(disk, sectors[WRITE]),
468                 jiffies_to_msecs(disk_stat_read(disk, ticks[WRITE])),
469                 disk->in_flight,
470                 jiffies_to_msecs(disk_stat_read(disk, io_ticks)),
471                 jiffies_to_msecs(disk_stat_read(disk, time_in_queue)));
472 }
473
474 #ifdef CONFIG_FAIL_MAKE_REQUEST
475 static ssize_t disk_fail_show(struct device *dev,
476                               struct device_attribute *attr, char *buf)
477 {
478         struct gendisk *disk = dev_to_disk(dev);
479
480         return sprintf(buf, "%d\n", disk->flags & GENHD_FL_FAIL ? 1 : 0);
481 }
482
483 static ssize_t disk_fail_store(struct device *dev,
484                                struct device_attribute *attr,
485                                const char *buf, size_t count)
486 {
487         struct gendisk *disk = dev_to_disk(dev);
488         int i;
489
490         if (count > 0 && sscanf(buf, "%d", &i) > 0) {
491                 if (i == 0)
492                         disk->flags &= ~GENHD_FL_FAIL;
493                 else
494                         disk->flags |= GENHD_FL_FAIL;
495         }
496
497         return count;
498 }
499
500 #endif
501
502 static DEVICE_ATTR(range, S_IRUGO, disk_range_show, NULL);
503 static DEVICE_ATTR(removable, S_IRUGO, disk_removable_show, NULL);
504 static DEVICE_ATTR(ro, S_IRUGO, disk_ro_show, NULL);
505 static DEVICE_ATTR(size, S_IRUGO, disk_size_show, NULL);
506 static DEVICE_ATTR(capability, S_IRUGO, disk_capability_show, NULL);
507 static DEVICE_ATTR(stat, S_IRUGO, disk_stat_show, NULL);
508 #ifdef CONFIG_FAIL_MAKE_REQUEST
509 static struct device_attribute dev_attr_fail =
510         __ATTR(make-it-fail, S_IRUGO|S_IWUSR, disk_fail_show, disk_fail_store);
511 #endif
512
513 static struct attribute *disk_attrs[] = {
514         &dev_attr_range.attr,
515         &dev_attr_removable.attr,
516         &dev_attr_ro.attr,
517         &dev_attr_size.attr,
518         &dev_attr_capability.attr,
519         &dev_attr_stat.attr,
520 #ifdef CONFIG_FAIL_MAKE_REQUEST
521         &dev_attr_fail.attr,
522 #endif
523         NULL
524 };
525
526 static struct attribute_group disk_attr_group = {
527         .attrs = disk_attrs,
528 };
529
530 static struct attribute_group *disk_attr_groups[] = {
531         &disk_attr_group,
532         NULL
533 };
534
535 static void disk_release(struct device *dev)
536 {
537         struct gendisk *disk = dev_to_disk(dev);
538
539         kfree(disk->random);
540         kfree(disk->part);
541         free_disk_stats(disk);
542         kfree(disk);
543 }
544 struct class block_class = {
545         .name           = "block",
546 };
547
548 static struct device_type disk_type = {
549         .name           = "disk",
550         .groups         = disk_attr_groups,
551         .release        = disk_release,
552 };
553
554 #ifdef CONFIG_PROC_FS
555 /*
556  * aggregate disk stat collector.  Uses the same stats that the sysfs
557  * entries do, above, but makes them available through one seq_file.
558  *
559  * The output looks suspiciously like /proc/partitions with a bunch of
560  * extra fields.
561  */
562
563 static void *diskstats_start(struct seq_file *part, loff_t *pos)
564 {
565         struct device *dev;
566
567         mutex_lock(&block_class_lock);
568         dev = class_find_device(&block_class, NULL, (void *)pos, find_start);
569         if (dev)
570                 return dev_to_disk(dev);
571         return NULL;
572 }
573
574 static void *diskstats_next(struct seq_file *part, void *v, loff_t *pos)
575 {
576         struct gendisk *gp = v;
577         struct device *dev;
578
579         ++*pos;
580         list_for_each_entry(dev, &gp->dev.node, node) {
581                 if (&dev->node == &block_class.devices)
582                         return NULL;
583                 if (dev->type == &disk_type)
584                         return dev_to_disk(dev);
585         }
586         return NULL;
587 }
588
589 static void diskstats_stop(struct seq_file *part, void *v)
590 {
591         mutex_unlock(&block_class_lock);
592 }
593
594 static int diskstats_show(struct seq_file *s, void *v)
595 {
596         struct gendisk *gp = v;
597         char buf[BDEVNAME_SIZE];
598         int n = 0;
599
600         /*
601         if (&gp->dev.kobj.entry == block_class.devices.next)
602                 seq_puts(s,     "major minor name"
603                                 "     rio rmerge rsect ruse wio wmerge "
604                                 "wsect wuse running use aveq"
605                                 "\n\n");
606         */
607  
608         preempt_disable();
609         disk_round_stats(gp);
610         preempt_enable();
611         seq_printf(s, "%4d %4d %s %lu %lu %llu %u %lu %lu %llu %u %u %u %u\n",
612                 gp->major, n + gp->first_minor, disk_name(gp, n, buf),
613                 disk_stat_read(gp, ios[0]), disk_stat_read(gp, merges[0]),
614                 (unsigned long long)disk_stat_read(gp, sectors[0]),
615                 jiffies_to_msecs(disk_stat_read(gp, ticks[0])),
616                 disk_stat_read(gp, ios[1]), disk_stat_read(gp, merges[1]),
617                 (unsigned long long)disk_stat_read(gp, sectors[1]),
618                 jiffies_to_msecs(disk_stat_read(gp, ticks[1])),
619                 gp->in_flight,
620                 jiffies_to_msecs(disk_stat_read(gp, io_ticks)),
621                 jiffies_to_msecs(disk_stat_read(gp, time_in_queue)));
622
623         /* now show all non-0 size partitions of it */
624         for (n = 0; n < gp->minors - 1; n++) {
625                 struct hd_struct *hd = gp->part[n];
626
627                 if (!hd || !hd->nr_sects)
628                         continue;
629
630                 preempt_disable();
631                 part_round_stats(hd);
632                 preempt_enable();
633                 seq_printf(s, "%4d %4d %s %lu %lu %llu "
634                            "%u %lu %lu %llu %u %u %u %u\n",
635                            gp->major, n + gp->first_minor + 1,
636                            disk_name(gp, n + 1, buf),
637                            part_stat_read(hd, ios[0]),
638                            part_stat_read(hd, merges[0]),
639                            (unsigned long long)part_stat_read(hd, sectors[0]),
640                            jiffies_to_msecs(part_stat_read(hd, ticks[0])),
641                            part_stat_read(hd, ios[1]),
642                            part_stat_read(hd, merges[1]),
643                            (unsigned long long)part_stat_read(hd, sectors[1]),
644                            jiffies_to_msecs(part_stat_read(hd, ticks[1])),
645                            hd->in_flight,
646                            jiffies_to_msecs(part_stat_read(hd, io_ticks)),
647                            jiffies_to_msecs(part_stat_read(hd, time_in_queue))
648                         );
649         }
650  
651         return 0;
652 }
653
654 const struct seq_operations diskstats_op = {
655         .start  = diskstats_start,
656         .next   = diskstats_next,
657         .stop   = diskstats_stop,
658         .show   = diskstats_show
659 };
660 #endif /* CONFIG_PROC_FS */
661
662 static void media_change_notify_thread(struct work_struct *work)
663 {
664         struct gendisk *gd = container_of(work, struct gendisk, async_notify);
665         char event[] = "MEDIA_CHANGE=1";
666         char *envp[] = { event, NULL };
667
668         /*
669          * set enviroment vars to indicate which event this is for
670          * so that user space will know to go check the media status.
671          */
672         kobject_uevent_env(&gd->dev.kobj, KOBJ_CHANGE, envp);
673         put_device(gd->driverfs_dev);
674 }
675
676 #if 0
677 void genhd_media_change_notify(struct gendisk *disk)
678 {
679         get_device(disk->driverfs_dev);
680         schedule_work(&disk->async_notify);
681 }
682 EXPORT_SYMBOL_GPL(genhd_media_change_notify);
683 #endif  /*  0  */
684
685 struct find_block {
686         const char *name;
687         int part;
688 };
689
690 static int match_id(struct device *dev, void *data)
691 {
692         struct find_block *find = data;
693
694         if (dev->type != &disk_type)
695                 return 0;
696         if (strcmp(dev->bus_id, find->name) == 0) {
697                 struct gendisk *disk = dev_to_disk(dev);
698                 if (find->part < disk->minors)
699                         return 1;
700         }
701         return 0;
702 }
703
704 dev_t blk_lookup_devt(const char *name, int part)
705 {
706         struct device *dev;
707         dev_t devt = MKDEV(0, 0);
708         struct find_block find;
709
710         mutex_lock(&block_class_lock);
711         find.name = name;
712         find.part = part;
713         dev = class_find_device(&block_class, NULL, (void *)&find, match_id);
714         if (dev)
715                 devt = MKDEV(MAJOR(dev->devt),
716                              MINOR(dev->devt) + part);
717         mutex_unlock(&block_class_lock);
718
719         return devt;
720 }
721 EXPORT_SYMBOL(blk_lookup_devt);
722
723 struct gendisk *alloc_disk(int minors)
724 {
725         return alloc_disk_node(minors, -1);
726 }
727
728 struct gendisk *alloc_disk_node(int minors, int node_id)
729 {
730         struct gendisk *disk;
731
732         disk = kmalloc_node(sizeof(struct gendisk),
733                                 GFP_KERNEL | __GFP_ZERO, node_id);
734         if (disk) {
735                 if (!init_disk_stats(disk)) {
736                         kfree(disk);
737                         return NULL;
738                 }
739                 if (minors > 1) {
740                         int size = (minors - 1) * sizeof(struct hd_struct *);
741                         disk->part = kmalloc_node(size,
742                                 GFP_KERNEL | __GFP_ZERO, node_id);
743                         if (!disk->part) {
744                                 free_disk_stats(disk);
745                                 kfree(disk);
746                                 return NULL;
747                         }
748                 }
749                 disk->minors = minors;
750                 rand_initialize_disk(disk);
751                 disk->dev.class = &block_class;
752                 disk->dev.type = &disk_type;
753                 device_initialize(&disk->dev);
754                 INIT_WORK(&disk->async_notify,
755                         media_change_notify_thread);
756         }
757         return disk;
758 }
759
760 EXPORT_SYMBOL(alloc_disk);
761 EXPORT_SYMBOL(alloc_disk_node);
762
763 struct kobject *get_disk(struct gendisk *disk)
764 {
765         struct module *owner;
766         struct kobject *kobj;
767
768         if (!disk->fops)
769                 return NULL;
770         owner = disk->fops->owner;
771         if (owner && !try_module_get(owner))
772                 return NULL;
773         kobj = kobject_get(&disk->dev.kobj);
774         if (kobj == NULL) {
775                 module_put(owner);
776                 return NULL;
777         }
778         return kobj;
779
780 }
781
782 EXPORT_SYMBOL(get_disk);
783
784 void put_disk(struct gendisk *disk)
785 {
786         if (disk)
787                 kobject_put(&disk->dev.kobj);
788 }
789
790 EXPORT_SYMBOL(put_disk);
791
792 void set_device_ro(struct block_device *bdev, int flag)
793 {
794         if (bdev->bd_contains != bdev)
795                 bdev->bd_part->policy = flag;
796         else
797                 bdev->bd_disk->policy = flag;
798 }
799
800 EXPORT_SYMBOL(set_device_ro);
801
802 void set_disk_ro(struct gendisk *disk, int flag)
803 {
804         int i;
805         disk->policy = flag;
806         for (i = 0; i < disk->minors - 1; i++)
807                 if (disk->part[i]) disk->part[i]->policy = flag;
808 }
809
810 EXPORT_SYMBOL(set_disk_ro);
811
812 int bdev_read_only(struct block_device *bdev)
813 {
814         if (!bdev)
815                 return 0;
816         else if (bdev->bd_contains != bdev)
817                 return bdev->bd_part->policy;
818         else
819                 return bdev->bd_disk->policy;
820 }
821
822 EXPORT_SYMBOL(bdev_read_only);
823
824 int invalidate_partition(struct gendisk *disk, int index)
825 {
826         int res = 0;
827         struct block_device *bdev = bdget_disk(disk, index);
828         if (bdev) {
829                 fsync_bdev(bdev);
830                 res = __invalidate_device(bdev);
831                 bdput(bdev);
832         }
833         return res;
834 }
835
836 EXPORT_SYMBOL(invalidate_partition);