685 lines · 16.5 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 gendisk *disk = bdev->bd_disk;	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(disk, 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(disk, p.pno, start, length);	case BLKPG_RESIZE_PARTITION:		return bdev_resize_partition(disk, 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 blk_ioctl_discard(struct block_device *bdev, fmode_t mode,		unsigned long arg){	uint64_t range[2];	uint64_t start, len;	struct inode *inode = bdev->bd_inode;	int err; 	if (!(mode & FMODE_WRITE))		return -EBADF; 	if (!bdev_max_discard_sectors(bdev))		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 > bdev_nr_bytes(bdev))		return -EINVAL; 	filemap_invalidate_lock(inode->i_mapping);	err = truncate_bdev_range(bdev, mode, start, start + len - 1);	if (err)		goto fail;	err = blkdev_issue_discard(bdev, start >> 9, len >> 9, GFP_KERNEL);fail:	filemap_invalidate_unlock(inode->i_mapping);	return err;} static int blk_ioctl_secure_erase(struct block_device *bdev, fmode_t mode,		void __user *argp){	uint64_t start, len;	uint64_t range[2];	int err; 	if (!(mode & FMODE_WRITE))		return -EBADF;	if (!bdev_max_secure_erase_sectors(bdev))		return -EOPNOTSUPP;	if (copy_from_user(range, argp, sizeof(range)))		return -EFAULT; 	start = range[0];	len = range[1];	if ((start & 511) || (len & 511))		return -EINVAL;	if (start + len > bdev_nr_bytes(bdev))		return -EINVAL; 	filemap_invalidate_lock(bdev->bd_inode->i_mapping);	err = truncate_bdev_range(bdev, mode, start, start + len - 1);	if (!err)		err = blkdev_issue_secure_erase(bdev, start >> 9, len >> 9,						GFP_KERNEL);	filemap_invalidate_unlock(bdev->bd_inode->i_mapping);	return err;}  static int blk_ioctl_zeroout(struct block_device *bdev, fmode_t mode,		unsigned long arg){	uint64_t range[2];	uint64_t start, end, len;	struct inode *inode = bdev->bd_inode;	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)bdev_nr_bytes(bdev))		return -EINVAL;	if (end < start)		return -EINVAL; 	/* Invalidate the page cache, including dirty pages */	filemap_invalidate_lock(inode->i_mapping);	err = truncate_bdev_range(bdev, mode, start, end);	if (err)		goto fail; 	err = blkdev_issue_zeroout(bdev, start >> 9, len >> 9, GFP_KERNEL,				   BLKDEV_ZERO_NOUNMAP); fail:	filemap_invalidate_unlock(inode->i_mapping);	return err;} 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 file *file, fmode_t mode, unsigned cmd,			       unsigned long arg, void __user *argp){	struct block_device *bdev = I_BDEV(file->f_mapping->host);	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);	case BLKSECDISCARD:		return blk_ioctl_secure_erase(bdev, mode, argp);	case BLKZEROOUT:		return blk_ioctl_zeroout(bdev, mode, arg);	case BLKGETDISKSEQ:		return put_u64(argp, bdev->bd_disk->diskseq);	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, bdev_nr_zones(bdev));	case BLKROGET:

Lines 1–500 of 685.