1 #include <linux/capability.h>
2 #include <linux/blkdev.h>
3 #include <linux/blkpg.h>
4 #include <linux/hdreg.h>
5 #include <linux/backing-dev.h>
6 #include <linux/buffer_head.h>
7 #include <linux/smp_lock.h>
8 #include <linux/blktrace_api.h>
9 #include <asm/uaccess.h>
11 static int blkpg_ioctl(struct block_device *bdev, struct blkpg_ioctl_arg __user *arg)
13 struct block_device *bdevp;
15 struct blkpg_ioctl_arg a;
16 struct blkpg_partition p;
17 long long start, length;
22 if (!capable(CAP_SYS_ADMIN))
24 if (copy_from_user(&a, arg, sizeof(struct blkpg_ioctl_arg)))
26 if (copy_from_user(&p, a.data, sizeof(struct blkpg_partition)))
29 if (bdev != bdev->bd_contains)
32 if (part <= 0 || part >= disk->minors)
35 case BLKPG_ADD_PARTITION:
37 length = p.length >> 9;
38 /* check for fit in a hd_struct */
39 if (sizeof(sector_t) == sizeof(long) &&
40 sizeof(long long) > sizeof(long)) {
41 long pstart = start, plength = length;
42 if (pstart != start || plength != length
43 || pstart < 0 || plength < 0)
46 /* partition number in use? */
47 mutex_lock(&bdev->bd_mutex);
48 if (disk->part[part - 1]) {
49 mutex_unlock(&bdev->bd_mutex);
53 for (i = 0; i < disk->minors - 1; i++) {
54 struct hd_struct *s = disk->part[i];
58 if (!(start+length <= s->start_sect ||
59 start >= s->start_sect + s->nr_sects)) {
60 mutex_unlock(&bdev->bd_mutex);
65 err = add_partition(disk, part, start, length, ADDPART_FLAG_NONE);
66 mutex_unlock(&bdev->bd_mutex);
68 case BLKPG_DEL_PARTITION:
69 if (!disk->part[part-1])
71 bdevp = bdget_disk(disk, part);
74 mutex_lock(&bdevp->bd_mutex);
75 if (bdevp->bd_openers) {
76 mutex_unlock(&bdevp->bd_mutex);
82 invalidate_bdev(bdevp);
84 mutex_lock_nested(&bdev->bd_mutex, 1);
85 delete_partition(disk, part);
86 mutex_unlock(&bdev->bd_mutex);
87 mutex_unlock(&bdevp->bd_mutex);
96 static int blkdev_reread_part(struct block_device *bdev)
98 struct gendisk *disk = bdev->bd_disk;
101 if (disk->minors == 1 || bdev != bdev->bd_contains)
103 if (!capable(CAP_SYS_ADMIN))
105 if (!mutex_trylock(&bdev->bd_mutex))
107 res = rescan_partitions(disk, bdev);
108 mutex_unlock(&bdev->bd_mutex);
112 static void blk_ioc_discard_endio(struct bio *bio, int err)
115 if (err == -EOPNOTSUPP)
116 set_bit(BIO_EOPNOTSUPP, &bio->bi_flags);
117 clear_bit(BIO_UPTODATE, &bio->bi_flags);
119 complete(bio->bi_private);
122 static int blk_ioctl_discard(struct block_device *bdev, uint64_t start,
125 struct request_queue *q = bdev_get_queue(bdev);
135 if (start + len > (bdev->bd_inode->i_size >> 9))
138 if (!q->prepare_discard_fn)
141 while (len && !ret) {
142 DECLARE_COMPLETION_ONSTACK(wait);
145 bio = bio_alloc(GFP_KERNEL, 0);
149 bio->bi_end_io = blk_ioc_discard_endio;
151 bio->bi_private = &wait;
152 bio->bi_sector = start;
154 if (len > q->max_hw_sectors) {
155 bio->bi_size = q->max_hw_sectors << 9;
156 len -= q->max_hw_sectors;
157 start += q->max_hw_sectors;
159 bio->bi_size = len << 9;
162 submit_bio(DISCARD_NOBARRIER, bio);
164 wait_for_completion(&wait);
166 if (bio_flagged(bio, BIO_EOPNOTSUPP))
168 else if (!bio_flagged(bio, BIO_UPTODATE))
175 static int put_ushort(unsigned long arg, unsigned short val)
177 return put_user(val, (unsigned short __user *)arg);
180 static int put_int(unsigned long arg, int val)
182 return put_user(val, (int __user *)arg);
185 static int put_long(unsigned long arg, long val)
187 return put_user(val, (long __user *)arg);
190 static int put_ulong(unsigned long arg, unsigned long val)
192 return put_user(val, (unsigned long __user *)arg);
195 static int put_u64(unsigned long arg, u64 val)
197 return put_user(val, (u64 __user *)arg);
200 static int blkdev_locked_ioctl(struct file *file, struct block_device *bdev,
201 unsigned cmd, unsigned long arg)
203 struct backing_dev_info *bdi;
211 bdi = blk_get_backing_dev_info(bdev);
214 return put_long(arg, (bdi->ra_pages * PAGE_CACHE_SIZE) / 512);
216 return put_int(arg, bdev_read_only(bdev) != 0);
217 case BLKBSZGET: /* get the logical block size (cf. BLKSSZGET) */
218 return put_int(arg, block_size(bdev));
219 case BLKSSZGET: /* get block device hardware sector size */
220 return put_int(arg, bdev_hardsect_size(bdev));
222 return put_ushort(arg, bdev_get_queue(bdev)->max_sectors);
225 if(!capable(CAP_SYS_ADMIN))
227 bdi = blk_get_backing_dev_info(bdev);
230 bdi->ra_pages = (arg * 512) / PAGE_CACHE_SIZE;
233 /* set the logical block size */
234 if (!capable(CAP_SYS_ADMIN))
238 if (get_user(n, (int __user *) arg))
240 if (bd_claim(bdev, file) < 0)
242 ret = set_blocksize(bdev, n);
246 return blkpg_ioctl(bdev, (struct blkpg_ioctl_arg __user *) arg);
248 return blkdev_reread_part(bdev);
250 if ((bdev->bd_inode->i_size >> 9) > ~0UL)
252 return put_ulong(arg, bdev->bd_inode->i_size >> 9);
254 return put_u64(arg, bdev->bd_inode->i_size);
258 case BLKTRACETEARDOWN:
259 return blk_trace_ioctl(bdev, cmd, (char __user *) arg);
264 int blkdev_driver_ioctl(struct inode *inode, struct file *file,
265 struct gendisk *disk, unsigned cmd, unsigned long arg)
268 if (disk->fops->unlocked_ioctl)
269 return disk->fops->unlocked_ioctl(file, cmd, arg);
271 if (disk->fops->ioctl) {
273 ret = disk->fops->ioctl(inode, file, cmd, arg);
280 EXPORT_SYMBOL_GPL(blkdev_driver_ioctl);
283 * always keep this in sync with compat_blkdev_ioctl() and
284 * compat_blkdev_locked_ioctl()
286 int blkdev_ioctl(struct inode *inode, struct file *file, unsigned cmd,
289 struct block_device *bdev = inode->i_bdev;
290 struct gendisk *disk = bdev->bd_disk;
295 if (!capable(CAP_SYS_ADMIN))
298 ret = blkdev_driver_ioctl(inode, file, disk, cmd, arg);
299 /* -EINVAL to handle old uncorrected drivers */
300 if (ret != -EINVAL && ret != -ENOTTY)
305 invalidate_bdev(bdev);
310 ret = blkdev_driver_ioctl(inode, file, disk, cmd, arg);
311 /* -EINVAL to handle old uncorrected drivers */
312 if (ret != -EINVAL && ret != -ENOTTY)
314 if (!capable(CAP_SYS_ADMIN))
316 if (get_user(n, (int __user *)(arg)))
319 set_device_ro(bdev, n);
326 if (!(file->f_mode & FMODE_WRITE))
329 if (copy_from_user(range, (void __user *)arg, sizeof(range)))
332 return blk_ioctl_discard(bdev, range[0], range[1]);
336 struct hd_geometry geo;
340 if (!disk->fops->getgeo)
344 * We need to set the startsect first, the driver may
345 * want to override it.
347 geo.start = get_start_sect(bdev);
348 ret = disk->fops->getgeo(bdev, &geo);
351 if (copy_to_user((struct hd_geometry __user *)arg, &geo,
359 ret = blkdev_locked_ioctl(file, bdev, cmd, arg);
361 if (ret != -ENOIOCTLCMD)
364 return blkdev_driver_ioctl(inode, file, disk, cmd, arg);
366 EXPORT_SYMBOL_GPL(blkdev_ioctl);