657 lines · 15.7 KB · C
RawDownload
// SPDX-License-Identifier: GPL-2.0#include <linux/capability.h>#include <linux/compat.h>#include <linux/blkdev.h>#include <linux/export.h>#include <linux/gfp.h>#include <linux/blkpg.h>#include <linux/hdreg.h>#include <linux/backing-dev.h>#include <linux/fs.h>#include <linux/blktrace_api.h>#include <linux/pr.h>#include <linux/uaccess.h>#include "blk.h" static int blkpg_do_ioctl(struct block_device *bdev,			  struct blkpg_partition __user *upart, int op){	struct blkpg_partition p;	long long start, length; 	if (!capable(CAP_SYS_ADMIN))		return -EACCES;	if (copy_from_user(&p, upart, sizeof(struct blkpg_partition)))		return -EFAULT;	if (bdev_is_partition(bdev))		return -EINVAL; 	if (p.pno <= 0)		return -EINVAL; 	if (op == BLKPG_DEL_PARTITION)		return bdev_del_partition(bdev, p.pno); 	start = p.start >> SECTOR_SHIFT;	length = p.length >> SECTOR_SHIFT; 	switch (op) {	case BLKPG_ADD_PARTITION:		/* check if partition is aligned to blocksize */		if (p.start & (bdev_logical_block_size(bdev) - 1))			return -EINVAL;		return bdev_add_partition(bdev, p.pno, start, length);	case BLKPG_RESIZE_PARTITION:		return bdev_resize_partition(bdev, p.pno, start, length);	default:		return -EINVAL;	}} static int blkpg_ioctl(struct block_device *bdev,		       struct blkpg_ioctl_arg __user *arg){	struct blkpg_partition __user *udata;	int op; 	if (get_user(op, &arg->op) || get_user(udata, &arg->data))		return -EFAULT; 	return blkpg_do_ioctl(bdev, udata, op);} #ifdef CONFIG_COMPATstruct compat_blkpg_ioctl_arg {	compat_int_t op;	compat_int_t flags;	compat_int_t datalen;	compat_caddr_t data;}; static int compat_blkpg_ioctl(struct block_device *bdev,			      struct compat_blkpg_ioctl_arg __user *arg){	compat_caddr_t udata;	int op; 	if (get_user(op, &arg->op) || get_user(udata, &arg->data))		return -EFAULT; 	return blkpg_do_ioctl(bdev, compat_ptr(udata), op);}#endif static int blkdev_reread_part(struct block_device *bdev, fmode_t mode){	struct block_device *tmp; 	if (!disk_part_scan_enabled(bdev->bd_disk) || bdev_is_partition(bdev))		return -EINVAL;	if (!capable(CAP_SYS_ADMIN))		return -EACCES;	if (bdev->bd_part_count)		return -EBUSY; 	/*	 * Reopen the device to revalidate the driver state and force a	 * partition rescan.	 */	mode &= ~FMODE_EXCL;	set_bit(GD_NEED_PART_SCAN, &bdev->bd_disk->state); 	tmp = blkdev_get_by_dev(bdev->bd_dev, mode, NULL);	if (IS_ERR(tmp))		return PTR_ERR(tmp);	blkdev_put(tmp, mode);	return 0;} static int blk_ioctl_discard(struct block_device *bdev, fmode_t mode,		unsigned long arg, unsigned long flags){	uint64_t range[2];	uint64_t start, len;	struct request_queue *q = bdev_get_queue(bdev);	int err; 	if (!(mode & FMODE_WRITE))		return -EBADF; 	if (!blk_queue_discard(q))		return -EOPNOTSUPP; 	if (copy_from_user(range, (void __user *)arg, sizeof(range)))		return -EFAULT; 	start = range[0];	len = range[1]; 	if (start & 511)		return -EINVAL;	if (len & 511)		return -EINVAL; 	if (start + len > i_size_read(bdev->bd_inode))		return -EINVAL; 	err = truncate_bdev_range(bdev, mode, start, start + len - 1);	if (err)		return err; 	return blkdev_issue_discard(bdev, start >> 9, len >> 9,				    GFP_KERNEL, flags);} static int blk_ioctl_zeroout(struct block_device *bdev, fmode_t mode,		unsigned long arg){	uint64_t range[2];	uint64_t start, end, len;	int err; 	if (!(mode & FMODE_WRITE))		return -EBADF; 	if (copy_from_user(range, (void __user *)arg, sizeof(range)))		return -EFAULT; 	start = range[0];	len = range[1];	end = start + len - 1; 	if (start & 511)		return -EINVAL;	if (len & 511)		return -EINVAL;	if (end >= (uint64_t)i_size_read(bdev->bd_inode))		return -EINVAL;	if (end < start)		return -EINVAL; 	/* Invalidate the page cache, including dirty pages */	err = truncate_bdev_range(bdev, mode, start, end);	if (err)		return err; 	return blkdev_issue_zeroout(bdev, start >> 9, len >> 9, GFP_KERNEL,			BLKDEV_ZERO_NOUNMAP);} static int put_ushort(unsigned short __user *argp, unsigned short val){	return put_user(val, argp);} static int put_int(int __user *argp, int val){	return put_user(val, argp);} static int put_uint(unsigned int __user *argp, unsigned int val){	return put_user(val, argp);} static int put_long(long __user *argp, long val){	return put_user(val, argp);} static int put_ulong(unsigned long __user *argp, unsigned long val){	return put_user(val, argp);} static int put_u64(u64 __user *argp, u64 val){	return put_user(val, argp);} #ifdef CONFIG_COMPATstatic int compat_put_long(compat_long_t __user *argp, long val){	return put_user(val, argp);} static int compat_put_ulong(compat_ulong_t __user *argp, compat_ulong_t val){	return put_user(val, argp);}#endif #ifdef CONFIG_COMPAT/* * This is the equivalent of compat_ptr_ioctl(), to be used by block * drivers that implement only commands that are completely compatible * between 32-bit and 64-bit user space */int blkdev_compat_ptr_ioctl(struct block_device *bdev, fmode_t mode,			unsigned cmd, unsigned long arg){	struct gendisk *disk = bdev->bd_disk; 	if (disk->fops->ioctl)		return disk->fops->ioctl(bdev, mode, cmd,					 (unsigned long)compat_ptr(arg)); 	return -ENOIOCTLCMD;}EXPORT_SYMBOL(blkdev_compat_ptr_ioctl);#endif static int blkdev_pr_register(struct block_device *bdev,		struct pr_registration __user *arg){	const struct pr_ops *ops = bdev->bd_disk->fops->pr_ops;	struct pr_registration reg; 	if (!capable(CAP_SYS_ADMIN))		return -EPERM;	if (!ops || !ops->pr_register)		return -EOPNOTSUPP;	if (copy_from_user(&reg, arg, sizeof(reg)))		return -EFAULT; 	if (reg.flags & ~PR_FL_IGNORE_KEY)		return -EOPNOTSUPP;	return ops->pr_register(bdev, reg.old_key, reg.new_key, reg.flags);} static int blkdev_pr_reserve(struct block_device *bdev,		struct pr_reservation __user *arg){	const struct pr_ops *ops = bdev->bd_disk->fops->pr_ops;	struct pr_reservation rsv; 	if (!capable(CAP_SYS_ADMIN))		return -EPERM;	if (!ops || !ops->pr_reserve)		return -EOPNOTSUPP;	if (copy_from_user(&rsv, arg, sizeof(rsv)))		return -EFAULT; 	if (rsv.flags & ~PR_FL_IGNORE_KEY)		return -EOPNOTSUPP;	return ops->pr_reserve(bdev, rsv.key, rsv.type, rsv.flags);} static int blkdev_pr_release(struct block_device *bdev,		struct pr_reservation __user *arg){	const struct pr_ops *ops = bdev->bd_disk->fops->pr_ops;	struct pr_reservation rsv; 	if (!capable(CAP_SYS_ADMIN))		return -EPERM;	if (!ops || !ops->pr_release)		return -EOPNOTSUPP;	if (copy_from_user(&rsv, arg, sizeof(rsv)))		return -EFAULT; 	if (rsv.flags)		return -EOPNOTSUPP;	return ops->pr_release(bdev, rsv.key, rsv.type);} static int blkdev_pr_preempt(struct block_device *bdev,		struct pr_preempt __user *arg, bool abort){	const struct pr_ops *ops = bdev->bd_disk->fops->pr_ops;	struct pr_preempt p; 	if (!capable(CAP_SYS_ADMIN))		return -EPERM;	if (!ops || !ops->pr_preempt)		return -EOPNOTSUPP;	if (copy_from_user(&p, arg, sizeof(p)))		return -EFAULT; 	if (p.flags)		return -EOPNOTSUPP;	return ops->pr_preempt(bdev, p.old_key, p.new_key, p.type, abort);} static int blkdev_pr_clear(struct block_device *bdev,		struct pr_clear __user *arg){	const struct pr_ops *ops = bdev->bd_disk->fops->pr_ops;	struct pr_clear c; 	if (!capable(CAP_SYS_ADMIN))		return -EPERM;	if (!ops || !ops->pr_clear)		return -EOPNOTSUPP;	if (copy_from_user(&c, arg, sizeof(c)))		return -EFAULT; 	if (c.flags)		return -EOPNOTSUPP;	return ops->pr_clear(bdev, c.key);} static int blkdev_flushbuf(struct block_device *bdev, fmode_t mode,		unsigned cmd, unsigned long arg){	if (!capable(CAP_SYS_ADMIN))		return -EACCES;	fsync_bdev(bdev);	invalidate_bdev(bdev);	return 0;} static int blkdev_roset(struct block_device *bdev, fmode_t mode,		unsigned cmd, unsigned long arg){	int ret, n; 	if (!capable(CAP_SYS_ADMIN))		return -EACCES; 	if (get_user(n, (int __user *)arg))		return -EFAULT;	if (bdev->bd_disk->fops->set_read_only) {		ret = bdev->bd_disk->fops->set_read_only(bdev, n);		if (ret)			return ret;	}	bdev->bd_read_only = n;	return 0;} static int blkdev_getgeo(struct block_device *bdev,		struct hd_geometry __user *argp){	struct gendisk *disk = bdev->bd_disk;	struct hd_geometry geo;	int ret; 	if (!argp)		return -EINVAL;	if (!disk->fops->getgeo)		return -ENOTTY; 	/*	 * We need to set the startsect first, the driver may	 * want to override it.	 */	memset(&geo, 0, sizeof(geo));	geo.start = get_start_sect(bdev);	ret = disk->fops->getgeo(bdev, &geo);	if (ret)		return ret;	if (copy_to_user(argp, &geo, sizeof(geo)))		return -EFAULT;	return 0;} #ifdef CONFIG_COMPATstruct compat_hd_geometry {	unsigned char heads;	unsigned char sectors;	unsigned short cylinders;	u32 start;}; static int compat_hdio_getgeo(struct block_device *bdev,			      struct compat_hd_geometry __user *ugeo){	struct gendisk *disk = bdev->bd_disk;	struct hd_geometry geo;	int ret; 	if (!ugeo)		return -EINVAL;	if (!disk->fops->getgeo)		return -ENOTTY; 	memset(&geo, 0, sizeof(geo));	/*	 * We need to set the startsect first, the driver may	 * want to override it.	 */	geo.start = get_start_sect(bdev);	ret = disk->fops->getgeo(bdev, &geo);	if (ret)		return ret; 	ret = copy_to_user(ugeo, &geo, 4);	ret |= put_user(geo.start, &ugeo->start);	if (ret)		ret = -EFAULT; 	return ret;}#endif /* set the logical block size */static int blkdev_bszset(struct block_device *bdev, fmode_t mode,		int __user *argp){	int ret, n; 	if (!capable(CAP_SYS_ADMIN))		return -EACCES;	if (!argp)		return -EINVAL;	if (get_user(n, argp))		return -EFAULT; 	if (mode & FMODE_EXCL)		return set_blocksize(bdev, n); 	if (IS_ERR(blkdev_get_by_dev(bdev->bd_dev, mode | FMODE_EXCL, &bdev)))		return -EBUSY;	ret = set_blocksize(bdev, n);	blkdev_put(bdev, mode | FMODE_EXCL); 	return ret;} /* * Common commands that are handled the same way on native and compat * user space. Note the separate arg/argp parameters that are needed * to deal with the compat_ptr() conversion. */static int blkdev_common_ioctl(struct block_device *bdev, fmode_t mode,				unsigned cmd, unsigned long arg, void __user *argp){	unsigned int max_sectors; 	switch (cmd) {	case BLKFLSBUF:		return blkdev_flushbuf(bdev, mode, cmd, arg);	case BLKROSET:		return blkdev_roset(bdev, mode, cmd, arg);	case BLKDISCARD:		return blk_ioctl_discard(bdev, mode, arg, 0);	case BLKSECDISCARD:		return blk_ioctl_discard(bdev, mode, arg,				BLKDEV_DISCARD_SECURE);	case BLKZEROOUT:		return blk_ioctl_zeroout(bdev, mode, arg);	case BLKREPORTZONE:		return blkdev_report_zones_ioctl(bdev, mode, cmd, arg);	case BLKRESETZONE:	case BLKOPENZONE:	case BLKCLOSEZONE:	case BLKFINISHZONE:		return blkdev_zone_mgmt_ioctl(bdev, mode, cmd, arg);	case BLKGETZONESZ:		return put_uint(argp, bdev_zone_sectors(bdev));	case BLKGETNRZONES:		return put_uint(argp, blkdev_nr_zones(bdev->bd_disk));	case BLKROGET:		return put_int(argp, bdev_read_only(bdev) != 0);	case BLKSSZGET: /* get block device logical block size */		return put_int(argp, bdev_logical_block_size(bdev));	case BLKPBSZGET: /* get block device physical block size */		return put_uint(argp, bdev_physical_block_size(bdev));	case BLKIOMIN:		return put_uint(argp, bdev_io_min(bdev));	case BLKIOOPT:		return put_uint(argp, bdev_io_opt(bdev));	case BLKALIGNOFF:		return put_int(argp, bdev_alignment_offset(bdev));	case BLKDISCARDZEROES:		return put_uint(argp, 0);	case BLKSECTGET:		max_sectors = min_t(unsigned int, USHRT_MAX,				    queue_max_sectors(bdev_get_queue(bdev)));		return put_ushort(argp, max_sectors);

Lines 1–500 of 657.