block: fix disk->part[] dereferencing race
[linux-2.6.git] / fs / partitions / check.c
1 /*
2  *  fs/partitions/check.c
3  *
4  *  Code extracted from drivers/block/genhd.c
5  *  Copyright (C) 1991-1998  Linus Torvalds
6  *  Re-organised Feb 1998 Russell King
7  *
8  *  We now have independent partition support from the
9  *  block drivers, which allows all the partition code to
10  *  be grouped in one location, and it to be mostly self
11  *  contained.
12  *
13  *  Added needed MAJORS for new pairs, {hdi,hdj}, {hdk,hdl}
14  */
15
16 #include <linux/init.h>
17 #include <linux/module.h>
18 #include <linux/fs.h>
19 #include <linux/kmod.h>
20 #include <linux/ctype.h>
21 #include <linux/genhd.h>
22
23 #include "check.h"
24
25 #include "acorn.h"
26 #include "amiga.h"
27 #include "atari.h"
28 #include "ldm.h"
29 #include "mac.h"
30 #include "msdos.h"
31 #include "osf.h"
32 #include "sgi.h"
33 #include "sun.h"
34 #include "ibm.h"
35 #include "ultrix.h"
36 #include "efi.h"
37 #include "karma.h"
38 #include "sysv68.h"
39
40 #ifdef CONFIG_BLK_DEV_MD
41 extern void md_autodetect_dev(dev_t dev);
42 #endif
43
44 int warn_no_part = 1; /*This is ugly: should make genhd removable media aware*/
45
46 static int (*check_part[])(struct parsed_partitions *, struct block_device *) = {
47         /*
48          * Probe partition formats with tables at disk address 0
49          * that also have an ADFS boot block at 0xdc0.
50          */
51 #ifdef CONFIG_ACORN_PARTITION_ICS
52         adfspart_check_ICS,
53 #endif
54 #ifdef CONFIG_ACORN_PARTITION_POWERTEC
55         adfspart_check_POWERTEC,
56 #endif
57 #ifdef CONFIG_ACORN_PARTITION_EESOX
58         adfspart_check_EESOX,
59 #endif
60
61         /*
62          * Now move on to formats that only have partition info at
63          * disk address 0xdc0.  Since these may also have stale
64          * PC/BIOS partition tables, they need to come before
65          * the msdos entry.
66          */
67 #ifdef CONFIG_ACORN_PARTITION_CUMANA
68         adfspart_check_CUMANA,
69 #endif
70 #ifdef CONFIG_ACORN_PARTITION_ADFS
71         adfspart_check_ADFS,
72 #endif
73
74 #ifdef CONFIG_EFI_PARTITION
75         efi_partition,          /* this must come before msdos */
76 #endif
77 #ifdef CONFIG_SGI_PARTITION
78         sgi_partition,
79 #endif
80 #ifdef CONFIG_LDM_PARTITION
81         ldm_partition,          /* this must come before msdos */
82 #endif
83 #ifdef CONFIG_MSDOS_PARTITION
84         msdos_partition,
85 #endif
86 #ifdef CONFIG_OSF_PARTITION
87         osf_partition,
88 #endif
89 #ifdef CONFIG_SUN_PARTITION
90         sun_partition,
91 #endif
92 #ifdef CONFIG_AMIGA_PARTITION
93         amiga_partition,
94 #endif
95 #ifdef CONFIG_ATARI_PARTITION
96         atari_partition,
97 #endif
98 #ifdef CONFIG_MAC_PARTITION
99         mac_partition,
100 #endif
101 #ifdef CONFIG_ULTRIX_PARTITION
102         ultrix_partition,
103 #endif
104 #ifdef CONFIG_IBM_PARTITION
105         ibm_partition,
106 #endif
107 #ifdef CONFIG_KARMA_PARTITION
108         karma_partition,
109 #endif
110 #ifdef CONFIG_SYSV68_PARTITION
111         sysv68_partition,
112 #endif
113         NULL
114 };
115  
116 /*
117  * disk_name() is used by partition check code and the genhd driver.
118  * It formats the devicename of the indicated disk into
119  * the supplied buffer (of size at least 32), and returns
120  * a pointer to that same buffer (for convenience).
121  */
122
123 char *disk_name(struct gendisk *hd, int partno, char *buf)
124 {
125         if (!partno)
126                 snprintf(buf, BDEVNAME_SIZE, "%s", hd->disk_name);
127         else if (isdigit(hd->disk_name[strlen(hd->disk_name)-1]))
128                 snprintf(buf, BDEVNAME_SIZE, "%sp%d", hd->disk_name, partno);
129         else
130                 snprintf(buf, BDEVNAME_SIZE, "%s%d", hd->disk_name, partno);
131
132         return buf;
133 }
134
135 const char *bdevname(struct block_device *bdev, char *buf)
136 {
137         int partno = 0;
138
139         if (bdev->bd_part)
140                 partno = bdev->bd_part->partno;
141
142         return disk_name(bdev->bd_disk, partno, buf);
143 }
144
145 EXPORT_SYMBOL(bdevname);
146
147 /*
148  * There's very little reason to use this, you should really
149  * have a struct block_device just about everywhere and use
150  * bdevname() instead.
151  */
152 const char *__bdevname(dev_t dev, char *buffer)
153 {
154         scnprintf(buffer, BDEVNAME_SIZE, "unknown-block(%u,%u)",
155                                 MAJOR(dev), MINOR(dev));
156         return buffer;
157 }
158
159 EXPORT_SYMBOL(__bdevname);
160
161 static struct parsed_partitions *
162 check_partition(struct gendisk *hd, struct block_device *bdev)
163 {
164         struct parsed_partitions *state;
165         int i, res, err;
166
167         state = kmalloc(sizeof(struct parsed_partitions), GFP_KERNEL);
168         if (!state)
169                 return NULL;
170
171         disk_name(hd, 0, state->name);
172         printk(KERN_INFO " %s:", state->name);
173         if (isdigit(state->name[strlen(state->name)-1]))
174                 sprintf(state->name, "p");
175
176         state->limit = disk_max_parts(hd) + 1;
177         i = res = err = 0;
178         while (!res && check_part[i]) {
179                 memset(&state->parts, 0, sizeof(state->parts));
180                 res = check_part[i++](state, bdev);
181                 if (res < 0) {
182                         /* We have hit an I/O error which we don't report now.
183                         * But record it, and let the others do their job.
184                         */
185                         err = res;
186                         res = 0;
187                 }
188
189         }
190         if (res > 0)
191                 return state;
192         if (err)
193         /* The partition is unrecognized. So report I/O errors if there were any */
194                 res = err;
195         if (!res)
196                 printk(" unknown partition table\n");
197         else if (warn_no_part)
198                 printk(" unable to read partition table\n");
199         kfree(state);
200         return ERR_PTR(res);
201 }
202
203 static ssize_t part_start_show(struct device *dev,
204                                struct device_attribute *attr, char *buf)
205 {
206         struct hd_struct *p = dev_to_part(dev);
207
208         return sprintf(buf, "%llu\n",(unsigned long long)p->start_sect);
209 }
210
211 static ssize_t part_size_show(struct device *dev,
212                               struct device_attribute *attr, char *buf)
213 {
214         struct hd_struct *p = dev_to_part(dev);
215         return sprintf(buf, "%llu\n",(unsigned long long)p->nr_sects);
216 }
217
218 static ssize_t part_stat_show(struct device *dev,
219                               struct device_attribute *attr, char *buf)
220 {
221         struct hd_struct *p = dev_to_part(dev);
222
223         preempt_disable();
224         part_round_stats(p);
225         preempt_enable();
226         return sprintf(buf,
227                 "%8lu %8lu %8llu %8u "
228                 "%8lu %8lu %8llu %8u "
229                 "%8u %8u %8u"
230                 "\n",
231                 part_stat_read(p, ios[READ]),
232                 part_stat_read(p, merges[READ]),
233                 (unsigned long long)part_stat_read(p, sectors[READ]),
234                 jiffies_to_msecs(part_stat_read(p, ticks[READ])),
235                 part_stat_read(p, ios[WRITE]),
236                 part_stat_read(p, merges[WRITE]),
237                 (unsigned long long)part_stat_read(p, sectors[WRITE]),
238                 jiffies_to_msecs(part_stat_read(p, ticks[WRITE])),
239                 p->in_flight,
240                 jiffies_to_msecs(part_stat_read(p, io_ticks)),
241                 jiffies_to_msecs(part_stat_read(p, time_in_queue)));
242 }
243
244 #ifdef CONFIG_FAIL_MAKE_REQUEST
245 static ssize_t part_fail_show(struct device *dev,
246                               struct device_attribute *attr, char *buf)
247 {
248         struct hd_struct *p = dev_to_part(dev);
249
250         return sprintf(buf, "%d\n", p->make_it_fail);
251 }
252
253 static ssize_t part_fail_store(struct device *dev,
254                                struct device_attribute *attr,
255                                const char *buf, size_t count)
256 {
257         struct hd_struct *p = dev_to_part(dev);
258         int i;
259
260         if (count > 0 && sscanf(buf, "%d", &i) > 0)
261                 p->make_it_fail = (i == 0) ? 0 : 1;
262
263         return count;
264 }
265 #endif
266
267 static DEVICE_ATTR(start, S_IRUGO, part_start_show, NULL);
268 static DEVICE_ATTR(size, S_IRUGO, part_size_show, NULL);
269 static DEVICE_ATTR(stat, S_IRUGO, part_stat_show, NULL);
270 #ifdef CONFIG_FAIL_MAKE_REQUEST
271 static struct device_attribute dev_attr_fail =
272         __ATTR(make-it-fail, S_IRUGO|S_IWUSR, part_fail_show, part_fail_store);
273 #endif
274
275 static struct attribute *part_attrs[] = {
276         &dev_attr_start.attr,
277         &dev_attr_size.attr,
278         &dev_attr_stat.attr,
279 #ifdef CONFIG_FAIL_MAKE_REQUEST
280         &dev_attr_fail.attr,
281 #endif
282         NULL
283 };
284
285 static struct attribute_group part_attr_group = {
286         .attrs = part_attrs,
287 };
288
289 static struct attribute_group *part_attr_groups[] = {
290         &part_attr_group,
291         NULL
292 };
293
294 static void part_release(struct device *dev)
295 {
296         struct hd_struct *p = dev_to_part(dev);
297         free_part_stats(p);
298         kfree(p);
299 }
300
301 struct device_type part_type = {
302         .name           = "partition",
303         .groups         = part_attr_groups,
304         .release        = part_release,
305 };
306
307 static inline void disk_sysfs_add_subdirs(struct gendisk *disk)
308 {
309         struct kobject *k;
310
311         k = kobject_get(&disk->dev.kobj);
312         disk->holder_dir = kobject_create_and_add("holders", k);
313         disk->slave_dir = kobject_create_and_add("slaves", k);
314         kobject_put(k);
315 }
316
317 static void delete_partition_rcu_cb(struct rcu_head *head)
318 {
319         struct hd_struct *part = container_of(head, struct hd_struct, rcu_head);
320
321         part->start_sect = 0;
322         part->nr_sects = 0;
323         part_stat_set_all(part, 0);
324         put_device(&part->dev);
325 }
326
327 void delete_partition(struct gendisk *disk, int partno)
328 {
329         struct hd_struct *part;
330
331         part = disk->__part[partno-1];
332         if (!part)
333                 return;
334
335         rcu_assign_pointer(disk->__part[partno-1], NULL);
336         kobject_put(part->holder_dir);
337         device_del(&part->dev);
338
339         call_rcu(&part->rcu_head, delete_partition_rcu_cb);
340 }
341
342 static ssize_t whole_disk_show(struct device *dev,
343                                struct device_attribute *attr, char *buf)
344 {
345         return 0;
346 }
347 static DEVICE_ATTR(whole_disk, S_IRUSR | S_IRGRP | S_IROTH,
348                    whole_disk_show, NULL);
349
350 int add_partition(struct gendisk *disk, int partno,
351                   sector_t start, sector_t len, int flags)
352 {
353         struct hd_struct *p;
354         int err;
355
356         if (disk->__part[partno - 1])
357                 return -EBUSY;
358
359         p = kzalloc(sizeof(*p), GFP_KERNEL);
360         if (!p)
361                 return -ENOMEM;
362
363         if (!init_part_stats(p)) {
364                 err = -ENOMEM;
365                 goto out_free;
366         }
367         p->start_sect = start;
368         p->nr_sects = len;
369         p->partno = partno;
370         p->policy = disk->policy;
371
372         if (isdigit(disk->dev.bus_id[strlen(disk->dev.bus_id)-1]))
373                 snprintf(p->dev.bus_id, BUS_ID_SIZE,
374                 "%sp%d", disk->dev.bus_id, partno);
375         else
376                 snprintf(p->dev.bus_id, BUS_ID_SIZE,
377                          "%s%d", disk->dev.bus_id, partno);
378
379         device_initialize(&p->dev);
380         p->dev.devt = MKDEV(disk->major, disk->first_minor + partno);
381         p->dev.class = &block_class;
382         p->dev.type = &part_type;
383         p->dev.parent = &disk->dev;
384
385         /* delay uevent until 'holders' subdir is created */
386         p->dev.uevent_suppress = 1;
387         err = device_add(&p->dev);
388         if (err)
389                 goto out_put;
390
391         err = -ENOMEM;
392         p->holder_dir = kobject_create_and_add("holders", &p->dev.kobj);
393         if (!p->holder_dir)
394                 goto out_del;
395
396         p->dev.uevent_suppress = 0;
397         if (flags & ADDPART_FLAG_WHOLEDISK) {
398                 err = device_create_file(&p->dev, &dev_attr_whole_disk);
399                 if (err)
400                         goto out_del;
401         }
402
403         /* everything is up and running, commence */
404         INIT_RCU_HEAD(&p->rcu_head);
405         rcu_assign_pointer(disk->__part[partno - 1], p);
406
407         /* suppress uevent if the disk supresses it */
408         if (!disk->dev.uevent_suppress)
409                 kobject_uevent(&p->dev.kobj, KOBJ_ADD);
410
411         return 0;
412
413 out_free:
414         kfree(p);
415         return err;
416 out_del:
417         kobject_put(p->holder_dir);
418         device_del(&p->dev);
419 out_put:
420         put_device(&p->dev);
421         return err;
422 }
423
424 /* Not exported, helper to add_disk(). */
425 void register_disk(struct gendisk *disk)
426 {
427         struct block_device *bdev;
428         struct disk_part_iter piter;
429         struct hd_struct *part;
430         char *s;
431         int err;
432
433         disk->dev.parent = disk->driverfs_dev;
434
435         strlcpy(disk->dev.bus_id, disk->disk_name, BUS_ID_SIZE);
436         /* ewww... some of these buggers have / in the name... */
437         s = strchr(disk->dev.bus_id, '/');
438         if (s)
439                 *s = '!';
440
441         /* delay uevents, until we scanned partition table */
442         disk->dev.uevent_suppress = 1;
443
444         if (device_add(&disk->dev))
445                 return;
446 #ifndef CONFIG_SYSFS_DEPRECATED
447         err = sysfs_create_link(block_depr, &disk->dev.kobj,
448                                 kobject_name(&disk->dev.kobj));
449         if (err) {
450                 device_del(&disk->dev);
451                 return;
452         }
453 #endif
454         disk_sysfs_add_subdirs(disk);
455
456         /* No minors to use for partitions */
457         if (!disk_max_parts(disk))
458                 goto exit;
459
460         /* No such device (e.g., media were just removed) */
461         if (!get_capacity(disk))
462                 goto exit;
463
464         bdev = bdget_disk(disk, 0);
465         if (!bdev)
466                 goto exit;
467
468         bdev->bd_invalidated = 1;
469         err = blkdev_get(bdev, FMODE_READ, 0);
470         if (err < 0)
471                 goto exit;
472         blkdev_put(bdev);
473
474 exit:
475         /* announce disk after possible partitions are created */
476         disk->dev.uevent_suppress = 0;
477         kobject_uevent(&disk->dev.kobj, KOBJ_ADD);
478
479         /* announce possible partitions */
480         disk_part_iter_init(&piter, disk, 0);
481         while ((part = disk_part_iter_next(&piter)))
482                 kobject_uevent(&part->dev.kobj, KOBJ_ADD);
483         disk_part_iter_exit(&piter);
484 }
485
486 int rescan_partitions(struct gendisk *disk, struct block_device *bdev)
487 {
488         struct disk_part_iter piter;
489         struct hd_struct *part;
490         struct parsed_partitions *state;
491         int p, res;
492
493         if (bdev->bd_part_count)
494                 return -EBUSY;
495         res = invalidate_partition(disk, 0);
496         if (res)
497                 return res;
498         bdev->bd_invalidated = 0;
499
500         disk_part_iter_init(&piter, disk, DISK_PITER_INCL_EMPTY);
501         while ((part = disk_part_iter_next(&piter)))
502                 delete_partition(disk, part->partno);
503         disk_part_iter_exit(&piter);
504
505         if (disk->fops->revalidate_disk)
506                 disk->fops->revalidate_disk(disk);
507         if (!get_capacity(disk) || !(state = check_partition(disk, bdev)))
508                 return 0;
509         if (IS_ERR(state))      /* I/O error reading the partition table */
510                 return -EIO;
511
512         /* tell userspace that the media / partition table may have changed */
513         kobject_uevent(&disk->dev.kobj, KOBJ_CHANGE);
514
515         for (p = 1; p < state->limit; p++) {
516                 sector_t size = state->parts[p].size;
517                 sector_t from = state->parts[p].from;
518                 if (!size)
519                         continue;
520                 if (from + size > get_capacity(disk)) {
521                         printk(KERN_WARNING
522                                 "%s: p%d exceeds device capacity\n",
523                                 disk->disk_name, p);
524                 }
525                 res = add_partition(disk, p, from, size, state->parts[p].flags);
526                 if (res) {
527                         printk(KERN_ERR " %s: p%d could not be added: %d\n",
528                                 disk->disk_name, p, -res);
529                         continue;
530                 }
531 #ifdef CONFIG_BLK_DEV_MD
532                 if (state->parts[p].flags & ADDPART_FLAG_RAID)
533                         md_autodetect_dev(bdev->bd_dev+p);
534 #endif
535         }
536         kfree(state);
537         return 0;
538 }
539
540 unsigned char *read_dev_sector(struct block_device *bdev, sector_t n, Sector *p)
541 {
542         struct address_space *mapping = bdev->bd_inode->i_mapping;
543         struct page *page;
544
545         page = read_mapping_page(mapping, (pgoff_t)(n >> (PAGE_CACHE_SHIFT-9)),
546                                  NULL);
547         if (!IS_ERR(page)) {
548                 if (PageError(page))
549                         goto fail;
550                 p->v = page;
551                 return (unsigned char *)page_address(page) +  ((n & ((1 << (PAGE_CACHE_SHIFT - 9)) - 1)) << 9);
552 fail:
553                 page_cache_release(page);
554         }
555         p->v = NULL;
556         return NULL;
557 }
558
559 EXPORT_SYMBOL(read_dev_sector);
560
561 void del_gendisk(struct gendisk *disk)
562 {
563         struct disk_part_iter piter;
564         struct hd_struct *part;
565
566         /* invalidate stuff */
567         disk_part_iter_init(&piter, disk,
568                              DISK_PITER_INCL_EMPTY | DISK_PITER_REVERSE);
569         while ((part = disk_part_iter_next(&piter))) {
570                 invalidate_partition(disk, part->partno);
571                 delete_partition(disk, part->partno);
572         }
573         disk_part_iter_exit(&piter);
574
575         invalidate_partition(disk, 0);
576         disk->capacity = 0;
577         disk->flags &= ~GENHD_FL_UP;
578         unlink_gendisk(disk);
579         disk_stat_set_all(disk, 0);
580         disk->stamp = 0;
581
582         kobject_put(disk->holder_dir);
583         kobject_put(disk->slave_dir);
584         disk->driverfs_dev = NULL;
585 #ifndef CONFIG_SYSFS_DEPRECATED
586         sysfs_remove_link(block_depr, disk->dev.bus_id);
587 #endif
588         device_del(&disk->dev);
589 }