dc9ad4c171e26cb171ed9235e3e724bc355a81f2
[safe/jmp/linux-2.6] / 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 *seqf, 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(seqf, "%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 *partno, 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
194         bdi = &disk->queue->backing_dev_info;
195         bdi_register_dev(bdi, MKDEV(disk->major, disk->first_minor));
196         retval = sysfs_create_link(&disk->dev.kobj, &bdi->dev->kobj, "bdi");
197         WARN_ON(retval);
198 }
199
200 EXPORT_SYMBOL(add_disk);
201 EXPORT_SYMBOL(del_gendisk);     /* in partitions/check.c */
202
203 void unlink_gendisk(struct gendisk *disk)
204 {
205         sysfs_remove_link(&disk->dev.kobj, "bdi");
206         bdi_unregister(&disk->queue->backing_dev_info);
207         blk_unregister_queue(disk);
208         blk_unregister_region(MKDEV(disk->major, disk->first_minor),
209                               disk->minors);
210 }
211
212 /**
213  * get_gendisk - get partitioning information for a given device
214  * @devt: device to get partitioning information for
215  * @part: returned partition index
216  *
217  * This function gets the structure containing partitioning
218  * information for the given device @devt.
219  */
220 struct gendisk *get_gendisk(dev_t devt, int *partno)
221 {
222         struct kobject *kobj = kobj_lookup(bdev_map, devt, partno);
223         struct device *dev = kobj_to_dev(kobj);
224
225         return  kobj ? dev_to_disk(dev) : NULL;
226 }
227
228 /*
229  * print a full list of all partitions - intended for places where the root
230  * filesystem can't be mounted and thus to give the victim some idea of what
231  * went wrong
232  */
233 void __init printk_all_partitions(void)
234 {
235         struct class_dev_iter iter;
236         struct device *dev;
237
238         class_dev_iter_init(&iter, &block_class, NULL, &disk_type);
239         while ((dev = class_dev_iter_next(&iter))) {
240                 struct gendisk *disk = dev_to_disk(dev);
241                 char buf[BDEVNAME_SIZE];
242                 int n;
243
244                 /*
245                  * Don't show empty devices or things that have been
246                  * surpressed
247                  */
248                 if (get_capacity(disk) == 0 ||
249                     (disk->flags & GENHD_FL_SUPPRESS_PARTITION_INFO))
250                         continue;
251
252                 /*
253                  * Note, unlike /proc/partitions, I am showing the
254                  * numbers in hex - the same format as the root=
255                  * option takes.
256                  */
257                 printk("%02x%02x %10llu %s",
258                        disk->major, disk->first_minor,
259                        (unsigned long long)get_capacity(disk) >> 1,
260                        disk_name(disk, 0, buf));
261                 if (disk->driverfs_dev != NULL &&
262                     disk->driverfs_dev->driver != NULL)
263                         printk(" driver: %s\n",
264                                disk->driverfs_dev->driver->name);
265                 else
266                         printk(" (driver?)\n");
267
268                 /* now show the partitions */
269                 for (n = 0; n < disk->minors - 1; ++n) {
270                         if (disk->part[n] == NULL)
271                                 continue;
272                         if (disk->part[n]->nr_sects == 0)
273                                 continue;
274                         printk("  %02x%02x %10llu %s\n",
275                                disk->major, n + 1 + disk->first_minor,
276                                (unsigned long long)disk->part[n]->nr_sects >> 1,
277                                disk_name(disk, n + 1, buf));
278                 }
279         }
280         class_dev_iter_exit(&iter);
281 }
282
283 #ifdef CONFIG_PROC_FS
284 /* iterator */
285 static void *disk_seqf_start(struct seq_file *seqf, loff_t *pos)
286 {
287         loff_t skip = *pos;
288         struct class_dev_iter *iter;
289         struct device *dev;
290
291         iter = kmalloc(GFP_KERNEL, sizeof(*iter));
292         if (!iter)
293                 return ERR_PTR(-ENOMEM);
294
295         seqf->private = iter;
296         class_dev_iter_init(iter, &block_class, NULL, &disk_type);
297         do {
298                 dev = class_dev_iter_next(iter);
299                 if (!dev)
300                         return NULL;
301         } while (skip--);
302
303         return dev_to_disk(dev);
304 }
305
306 static void *disk_seqf_next(struct seq_file *seqf, void *v, loff_t *pos)
307 {
308         struct device *dev;
309
310         (*pos)++;
311         dev = class_dev_iter_next(seqf->private);
312         if (dev)
313                 return dev_to_disk(dev);
314
315         return NULL;
316 }
317
318 static void disk_seqf_stop(struct seq_file *seqf, void *v)
319 {
320         struct class_dev_iter *iter = seqf->private;
321
322         /* stop is called even after start failed :-( */
323         if (iter) {
324                 class_dev_iter_exit(iter);
325                 kfree(iter);
326         }
327 }
328
329 static void *show_partition_start(struct seq_file *seqf, loff_t *pos)
330 {
331         static void *p;
332
333         p = disk_seqf_start(seqf, pos);
334         if (!IS_ERR(p) && p)
335                 seq_puts(seqf, "major minor  #blocks  name\n\n");
336         return p;
337 }
338
339 static int show_partition(struct seq_file *seqf, void *v)
340 {
341         struct gendisk *sgp = v;
342         int n;
343         char buf[BDEVNAME_SIZE];
344
345         /* Don't show non-partitionable removeable devices or empty devices */
346         if (!get_capacity(sgp) ||
347                         (sgp->minors == 1 && (sgp->flags & GENHD_FL_REMOVABLE)))
348                 return 0;
349         if (sgp->flags & GENHD_FL_SUPPRESS_PARTITION_INFO)
350                 return 0;
351
352         /* show the full disk and all non-0 size partitions of it */
353         seq_printf(seqf, "%4d  %4d %10llu %s\n",
354                 sgp->major, sgp->first_minor,
355                 (unsigned long long)get_capacity(sgp) >> 1,
356                 disk_name(sgp, 0, buf));
357         for (n = 0; n < sgp->minors - 1; n++) {
358                 if (!sgp->part[n])
359                         continue;
360                 if (sgp->part[n]->nr_sects == 0)
361                         continue;
362                 seq_printf(seqf, "%4d  %4d %10llu %s\n",
363                         sgp->major, n + 1 + sgp->first_minor,
364                         (unsigned long long)sgp->part[n]->nr_sects >> 1 ,
365                         disk_name(sgp, n + 1, buf));
366         }
367
368         return 0;
369 }
370
371 const struct seq_operations partitions_op = {
372         .start  = show_partition_start,
373         .next   = disk_seqf_next,
374         .stop   = disk_seqf_stop,
375         .show   = show_partition
376 };
377 #endif
378
379
380 static struct kobject *base_probe(dev_t devt, int *partno, void *data)
381 {
382         if (request_module("block-major-%d-%d", MAJOR(devt), MINOR(devt)) > 0)
383                 /* Make old-style 2.4 aliases work */
384                 request_module("block-major-%d", MAJOR(devt));
385         return NULL;
386 }
387
388 static int __init genhd_device_init(void)
389 {
390         int error;
391
392         block_class.dev_kobj = sysfs_dev_block_kobj;
393         error = class_register(&block_class);
394         if (unlikely(error))
395                 return error;
396         bdev_map = kobj_map_init(base_probe, &block_class_lock);
397         blk_dev_init();
398
399 #ifndef CONFIG_SYSFS_DEPRECATED
400         /* create top-level block dir */
401         block_depr = kobject_create_and_add("block", NULL);
402 #endif
403         return 0;
404 }
405
406 subsys_initcall(genhd_device_init);
407
408 static ssize_t disk_range_show(struct device *dev,
409                                struct device_attribute *attr, char *buf)
410 {
411         struct gendisk *disk = dev_to_disk(dev);
412
413         return sprintf(buf, "%d\n", disk->minors);
414 }
415
416 static ssize_t disk_removable_show(struct device *dev,
417                                    struct device_attribute *attr, char *buf)
418 {
419         struct gendisk *disk = dev_to_disk(dev);
420
421         return sprintf(buf, "%d\n",
422                        (disk->flags & GENHD_FL_REMOVABLE ? 1 : 0));
423 }
424
425 static ssize_t disk_ro_show(struct device *dev,
426                                    struct device_attribute *attr, char *buf)
427 {
428         struct gendisk *disk = dev_to_disk(dev);
429
430         return sprintf(buf, "%d\n", disk->policy ? 1 : 0);
431 }
432
433 static ssize_t disk_size_show(struct device *dev,
434                               struct device_attribute *attr, char *buf)
435 {
436         struct gendisk *disk = dev_to_disk(dev);
437
438         return sprintf(buf, "%llu\n", (unsigned long long)get_capacity(disk));
439 }
440
441 static ssize_t disk_capability_show(struct device *dev,
442                                     struct device_attribute *attr, char *buf)
443 {
444         struct gendisk *disk = dev_to_disk(dev);
445
446         return sprintf(buf, "%x\n", disk->flags);
447 }
448
449 static ssize_t disk_stat_show(struct device *dev,
450                               struct device_attribute *attr, char *buf)
451 {
452         struct gendisk *disk = dev_to_disk(dev);
453
454         preempt_disable();
455         disk_round_stats(disk);
456         preempt_enable();
457         return sprintf(buf,
458                 "%8lu %8lu %8llu %8u "
459                 "%8lu %8lu %8llu %8u "
460                 "%8u %8u %8u"
461                 "\n",
462                 disk_stat_read(disk, ios[READ]),
463                 disk_stat_read(disk, merges[READ]),
464                 (unsigned long long)disk_stat_read(disk, sectors[READ]),
465                 jiffies_to_msecs(disk_stat_read(disk, ticks[READ])),
466                 disk_stat_read(disk, ios[WRITE]),
467                 disk_stat_read(disk, merges[WRITE]),
468                 (unsigned long long)disk_stat_read(disk, sectors[WRITE]),
469                 jiffies_to_msecs(disk_stat_read(disk, ticks[WRITE])),
470                 disk->in_flight,
471                 jiffies_to_msecs(disk_stat_read(disk, io_ticks)),
472                 jiffies_to_msecs(disk_stat_read(disk, time_in_queue)));
473 }
474
475 #ifdef CONFIG_FAIL_MAKE_REQUEST
476 static ssize_t disk_fail_show(struct device *dev,
477                               struct device_attribute *attr, char *buf)
478 {
479         struct gendisk *disk = dev_to_disk(dev);
480
481         return sprintf(buf, "%d\n", disk->flags & GENHD_FL_FAIL ? 1 : 0);
482 }
483
484 static ssize_t disk_fail_store(struct device *dev,
485                                struct device_attribute *attr,
486                                const char *buf, size_t count)
487 {
488         struct gendisk *disk = dev_to_disk(dev);
489         int i;
490
491         if (count > 0 && sscanf(buf, "%d", &i) > 0) {
492                 if (i == 0)
493                         disk->flags &= ~GENHD_FL_FAIL;
494                 else
495                         disk->flags |= GENHD_FL_FAIL;
496         }
497
498         return count;
499 }
500
501 #endif
502
503 static DEVICE_ATTR(range, S_IRUGO, disk_range_show, NULL);
504 static DEVICE_ATTR(removable, S_IRUGO, disk_removable_show, NULL);
505 static DEVICE_ATTR(ro, S_IRUGO, disk_ro_show, NULL);
506 static DEVICE_ATTR(size, S_IRUGO, disk_size_show, NULL);
507 static DEVICE_ATTR(capability, S_IRUGO, disk_capability_show, NULL);
508 static DEVICE_ATTR(stat, S_IRUGO, disk_stat_show, NULL);
509 #ifdef CONFIG_FAIL_MAKE_REQUEST
510 static struct device_attribute dev_attr_fail =
511         __ATTR(make-it-fail, S_IRUGO|S_IWUSR, disk_fail_show, disk_fail_store);
512 #endif
513
514 static struct attribute *disk_attrs[] = {
515         &dev_attr_range.attr,
516         &dev_attr_removable.attr,
517         &dev_attr_ro.attr,
518         &dev_attr_size.attr,
519         &dev_attr_capability.attr,
520         &dev_attr_stat.attr,
521 #ifdef CONFIG_FAIL_MAKE_REQUEST
522         &dev_attr_fail.attr,
523 #endif
524         NULL
525 };
526
527 static struct attribute_group disk_attr_group = {
528         .attrs = disk_attrs,
529 };
530
531 static struct attribute_group *disk_attr_groups[] = {
532         &disk_attr_group,
533         NULL
534 };
535
536 static void disk_release(struct device *dev)
537 {
538         struct gendisk *disk = dev_to_disk(dev);
539
540         kfree(disk->random);
541         kfree(disk->part);
542         free_disk_stats(disk);
543         kfree(disk);
544 }
545 struct class block_class = {
546         .name           = "block",
547 };
548
549 static struct device_type disk_type = {
550         .name           = "disk",
551         .groups         = disk_attr_groups,
552         .release        = disk_release,
553 };
554
555 #ifdef CONFIG_PROC_FS
556 /*
557  * aggregate disk stat collector.  Uses the same stats that the sysfs
558  * entries do, above, but makes them available through one seq_file.
559  *
560  * The output looks suspiciously like /proc/partitions with a bunch of
561  * extra fields.
562  */
563 static int diskstats_show(struct seq_file *seqf, void *v)
564 {
565         struct gendisk *gp = v;
566         char buf[BDEVNAME_SIZE];
567         int n;
568
569         /*
570         if (&gp->dev.kobj.entry == block_class.devices.next)
571                 seq_puts(seqf,  "major minor name"
572                                 "     rio rmerge rsect ruse wio wmerge "
573                                 "wsect wuse running use aveq"
574                                 "\n\n");
575         */
576  
577         preempt_disable();
578         disk_round_stats(gp);
579         preempt_enable();
580         seq_printf(seqf, "%4d %4d %s %lu %lu %llu %u %lu %lu %llu %u %u %u %u\n",
581                 gp->major, gp->first_minor, disk_name(gp, 0, buf),
582                 disk_stat_read(gp, ios[0]), disk_stat_read(gp, merges[0]),
583                 (unsigned long long)disk_stat_read(gp, sectors[0]),
584                 jiffies_to_msecs(disk_stat_read(gp, ticks[0])),
585                 disk_stat_read(gp, ios[1]), disk_stat_read(gp, merges[1]),
586                 (unsigned long long)disk_stat_read(gp, sectors[1]),
587                 jiffies_to_msecs(disk_stat_read(gp, ticks[1])),
588                 gp->in_flight,
589                 jiffies_to_msecs(disk_stat_read(gp, io_ticks)),
590                 jiffies_to_msecs(disk_stat_read(gp, time_in_queue)));
591
592         /* now show all non-0 size partitions of it */
593         for (n = 0; n < gp->minors - 1; n++) {
594                 struct hd_struct *hd = gp->part[n];
595
596                 if (!hd || !hd->nr_sects)
597                         continue;
598
599                 preempt_disable();
600                 part_round_stats(hd);
601                 preempt_enable();
602                 seq_printf(seqf, "%4d %4d %s %lu %lu %llu "
603                            "%u %lu %lu %llu %u %u %u %u\n",
604                            gp->major, n + gp->first_minor + 1,
605                            disk_name(gp, n + 1, buf),
606                            part_stat_read(hd, ios[0]),
607                            part_stat_read(hd, merges[0]),
608                            (unsigned long long)part_stat_read(hd, sectors[0]),
609                            jiffies_to_msecs(part_stat_read(hd, ticks[0])),
610                            part_stat_read(hd, ios[1]),
611                            part_stat_read(hd, merges[1]),
612                            (unsigned long long)part_stat_read(hd, sectors[1]),
613                            jiffies_to_msecs(part_stat_read(hd, ticks[1])),
614                            hd->in_flight,
615                            jiffies_to_msecs(part_stat_read(hd, io_ticks)),
616                            jiffies_to_msecs(part_stat_read(hd, time_in_queue))
617                         );
618         }
619  
620         return 0;
621 }
622
623 const struct seq_operations diskstats_op = {
624         .start  = disk_seqf_start,
625         .next   = disk_seqf_next,
626         .stop   = disk_seqf_stop,
627         .show   = diskstats_show
628 };
629 #endif /* CONFIG_PROC_FS */
630
631 static void media_change_notify_thread(struct work_struct *work)
632 {
633         struct gendisk *gd = container_of(work, struct gendisk, async_notify);
634         char event[] = "MEDIA_CHANGE=1";
635         char *envp[] = { event, NULL };
636
637         /*
638          * set enviroment vars to indicate which event this is for
639          * so that user space will know to go check the media status.
640          */
641         kobject_uevent_env(&gd->dev.kobj, KOBJ_CHANGE, envp);
642         put_device(gd->driverfs_dev);
643 }
644
645 #if 0
646 void genhd_media_change_notify(struct gendisk *disk)
647 {
648         get_device(disk->driverfs_dev);
649         schedule_work(&disk->async_notify);
650 }
651 EXPORT_SYMBOL_GPL(genhd_media_change_notify);
652 #endif  /*  0  */
653
654 dev_t blk_lookup_devt(const char *name, int partno)
655 {
656         dev_t devt = MKDEV(0, 0);
657         struct class_dev_iter iter;
658         struct device *dev;
659
660         class_dev_iter_init(&iter, &block_class, NULL, &disk_type);
661         while ((dev = class_dev_iter_next(&iter))) {
662                 struct gendisk *disk = dev_to_disk(dev);
663
664                 if (!strcmp(dev->bus_id, name) && partno < disk->minors) {
665                         devt = MKDEV(MAJOR(dev->devt),
666                                      MINOR(dev->devt) + partno);
667                         break;
668                 }
669         }
670         class_dev_iter_exit(&iter);
671         return devt;
672 }
673 EXPORT_SYMBOL(blk_lookup_devt);
674
675 struct gendisk *alloc_disk(int minors)
676 {
677         return alloc_disk_node(minors, -1);
678 }
679
680 struct gendisk *alloc_disk_node(int minors, int node_id)
681 {
682         struct gendisk *disk;
683
684         disk = kmalloc_node(sizeof(struct gendisk),
685                                 GFP_KERNEL | __GFP_ZERO, node_id);
686         if (disk) {
687                 if (!init_disk_stats(disk)) {
688                         kfree(disk);
689                         return NULL;
690                 }
691                 if (minors > 1) {
692                         int size = (minors - 1) * sizeof(struct hd_struct *);
693                         disk->part = kmalloc_node(size,
694                                 GFP_KERNEL | __GFP_ZERO, node_id);
695                         if (!disk->part) {
696                                 free_disk_stats(disk);
697                                 kfree(disk);
698                                 return NULL;
699                         }
700                 }
701                 disk->minors = minors;
702                 rand_initialize_disk(disk);
703                 disk->dev.class = &block_class;
704                 disk->dev.type = &disk_type;
705                 device_initialize(&disk->dev);
706                 INIT_WORK(&disk->async_notify,
707                         media_change_notify_thread);
708         }
709         return disk;
710 }
711
712 EXPORT_SYMBOL(alloc_disk);
713 EXPORT_SYMBOL(alloc_disk_node);
714
715 struct kobject *get_disk(struct gendisk *disk)
716 {
717         struct module *owner;
718         struct kobject *kobj;
719
720         if (!disk->fops)
721                 return NULL;
722         owner = disk->fops->owner;
723         if (owner && !try_module_get(owner))
724                 return NULL;
725         kobj = kobject_get(&disk->dev.kobj);
726         if (kobj == NULL) {
727                 module_put(owner);
728                 return NULL;
729         }
730         return kobj;
731
732 }
733
734 EXPORT_SYMBOL(get_disk);
735
736 void put_disk(struct gendisk *disk)
737 {
738         if (disk)
739                 kobject_put(&disk->dev.kobj);
740 }
741
742 EXPORT_SYMBOL(put_disk);
743
744 void set_device_ro(struct block_device *bdev, int flag)
745 {
746         if (bdev->bd_contains != bdev)
747                 bdev->bd_part->policy = flag;
748         else
749                 bdev->bd_disk->policy = flag;
750 }
751
752 EXPORT_SYMBOL(set_device_ro);
753
754 void set_disk_ro(struct gendisk *disk, int flag)
755 {
756         int i;
757         disk->policy = flag;
758         for (i = 0; i < disk->minors - 1; i++)
759                 if (disk->part[i]) disk->part[i]->policy = flag;
760 }
761
762 EXPORT_SYMBOL(set_disk_ro);
763
764 int bdev_read_only(struct block_device *bdev)
765 {
766         if (!bdev)
767                 return 0;
768         else if (bdev->bd_contains != bdev)
769                 return bdev->bd_part->policy;
770         else
771                 return bdev->bd_disk->policy;
772 }
773
774 EXPORT_SYMBOL(bdev_read_only);
775
776 int invalidate_partition(struct gendisk *disk, int partno)
777 {
778         int res = 0;
779         struct block_device *bdev = bdget_disk(disk, partno);
780         if (bdev) {
781                 fsync_bdev(bdev);
782                 res = __invalidate_device(bdev);
783                 bdput(bdev);
784         }
785         return res;
786 }
787
788 EXPORT_SYMBOL(invalidate_partition);