[GitHub/LineageOS/android_kernel_samsung_universal7580.git] / block / ioctl.c

#include <linux/capability.h>
#include <linux/blkdev.h>
#include <linux/blkpg.h>
#include <linux/hdreg.h>
#include <linux/backing-dev.h>
#include <linux/buffer_head.h>
#include <linux/smp_lock.h>
#include <linux/blktrace_api.h>
#include <asm/uaccess.h>

static int blkpg_ioctl(struct block_device *bdev, struct blkpg_ioctl_arg __user *arg)
{
	struct block_device *bdevp;
	struct gendisk *disk;
	struct hd_struct *part;
	struct blkpg_ioctl_arg a;
	struct blkpg_partition p;
	struct disk_part_iter piter;
	long long start, length;
	int partno;

	if (!capable(CAP_SYS_ADMIN))
		return -EACCES;
	if (copy_from_user(&a, arg, sizeof(struct blkpg_ioctl_arg)))
		return -EFAULT;
	if (copy_from_user(&p, a.data, sizeof(struct blkpg_partition)))
		return -EFAULT;
	disk = bdev->bd_disk;
	if (bdev != bdev->bd_contains)
		return -EINVAL;
	partno = p.pno;
	if (partno <= 0)
		return -EINVAL;
	switch (a.op) {
		case BLKPG_ADD_PARTITION:
			start = p.start >> 9;
			length = p.length >> 9;
			/* check for fit in a hd_struct */ 
			if (sizeof(sector_t) == sizeof(long) && 
			    sizeof(long long) > sizeof(long)) {
				long pstart = start, plength = length;
				if (pstart != start || plength != length
				    || pstart < 0 || plength < 0)
					return -EINVAL;
			}

			mutex_lock(&bdev->bd_mutex);

			/* overlap? */
			disk_part_iter_init(&piter, disk,
					    DISK_PITER_INCL_EMPTY);
			while ((part = disk_part_iter_next(&piter))) {
				if (!(start + length <= part->start_sect ||
				      start >= part->start_sect + part->nr_sects)) {
					disk_part_iter_exit(&piter);
					mutex_unlock(&bdev->bd_mutex);
					return -EBUSY;
				}
			}
			disk_part_iter_exit(&piter);

			/* all seems OK */
			part = add_partition(disk, partno, start, length,
					     ADDPART_FLAG_NONE);
			mutex_unlock(&bdev->bd_mutex);
			return IS_ERR(part) ? PTR_ERR(part) : 0;
		case BLKPG_DEL_PARTITION:
			part = disk_get_part(disk, partno);
			if (!part)
				return -ENXIO;

			bdevp = bdget(part_devt(part));
			disk_put_part(part);
			if (!bdevp)
				return -ENOMEM;

			mutex_lock(&bdevp->bd_mutex);
			if (bdevp->bd_openers) {
				mutex_unlock(&bdevp->bd_mutex);
				bdput(bdevp);
				return -EBUSY;
			}
			/* all seems OK */
			fsync_bdev(bdevp);
			invalidate_bdev(bdevp);

			mutex_lock_nested(&bdev->bd_mutex, 1);
			delete_partition(disk, partno);
			mutex_unlock(&bdev->bd_mutex);
			mutex_unlock(&bdevp->bd_mutex);
			bdput(bdevp);

			return 0;
		default:
			return -EINVAL;
	}
}

static int blkdev_reread_part(struct block_device *bdev)
{
	struct gendisk *disk = bdev->bd_disk;
	int res;

	if (!disk_partitionable(disk) || bdev != bdev->bd_contains)
		return -EINVAL;
	if (!capable(CAP_SYS_ADMIN))
		return -EACCES;
	if (!mutex_trylock(&bdev->bd_mutex))
		return -EBUSY;
	res = rescan_partitions(disk, bdev);
	mutex_unlock(&bdev->bd_mutex);
	return res;
}

static int blk_ioctl_discard(struct block_device *bdev, uint64_t start,
			     uint64_t len)
{
	if (start & 511)
		return -EINVAL;
	if (len & 511)
		return -EINVAL;
	start >>= 9;
	len >>= 9;

	if (start + len > (bdev->bd_inode->i_size >> 9))
		return -EINVAL;
	return blkdev_issue_discard(bdev, start, len, GFP_KERNEL,
				    DISCARD_FL_WAIT);
}

static int put_ushort(unsigned long arg, unsigned short val)
{
	return put_user(val, (unsigned short __user *)arg);
}

static int put_int(unsigned long arg, int val)
{
	return put_user(val, (int __user *)arg);
}

static int put_long(unsigned long arg, long val)
{
	return put_user(val, (long __user *)arg);
}

static int put_ulong(unsigned long arg, unsigned long val)
{
	return put_user(val, (unsigned long __user *)arg);
}

static int put_u64(unsigned long arg, u64 val)
{
	return put_user(val, (u64 __user *)arg);
}

int __blkdev_driver_ioctl(struct block_device *bdev, fmode_t mode,
			unsigned cmd, unsigned long arg)
{
	struct gendisk *disk = bdev->bd_disk;
	int ret;

	if (disk->fops->ioctl)
		return disk->fops->ioctl(bdev, mode, cmd, arg);

	if (disk->fops->locked_ioctl) {
		lock_kernel();
		ret = disk->fops->locked_ioctl(bdev, mode, cmd, arg);
		unlock_kernel();
		return ret;
	}

	return -ENOTTY;
}
/*
 * For the record: _GPL here is only because somebody decided to slap it
 * on the previous export.  Sheer idiocy, since it wasn't copyrightable
 * at all and could be open-coded without any exports by anybody who cares.
 */
EXPORT_SYMBOL_GPL(__blkdev_driver_ioctl);

/*
 * always keep this in sync with compat_blkdev_ioctl() and
 * compat_blkdev_locked_ioctl()
 */
int blkdev_ioctl(struct block_device *bdev, fmode_t mode, unsigned cmd,
			unsigned long arg)
{
	struct gendisk *disk = bdev->bd_disk;
	struct backing_dev_info *bdi;
	loff_t size;
	int ret, n;

	switch(cmd) {
	case BLKFLSBUF:
		if (!capable(CAP_SYS_ADMIN))
			return -EACCES;

		ret = __blkdev_driver_ioctl(bdev, mode, cmd, arg);
		/* -EINVAL to handle old uncorrected drivers */
		if (ret != -EINVAL && ret != -ENOTTY)
			return ret;

		lock_kernel();
		fsync_bdev(bdev);
		invalidate_bdev(bdev);
		unlock_kernel();
		return 0;

	case BLKROSET:
		ret = __blkdev_driver_ioctl(bdev, mode, cmd, arg);
		/* -EINVAL to handle old uncorrected drivers */
		if (ret != -EINVAL && ret != -ENOTTY)
			return ret;
		if (!capable(CAP_SYS_ADMIN))
			return -EACCES;
		if (get_user(n, (int __user *)(arg)))
			return -EFAULT;
		lock_kernel();
		set_device_ro(bdev, n);
		unlock_kernel();
		return 0;

	case BLKDISCARD: {
		uint64_t range[2];

		if (!(mode & FMODE_WRITE))
			return -EBADF;

		if (copy_from_user(range, (void __user *)arg, sizeof(range)))
			return -EFAULT;

		return blk_ioctl_discard(bdev, range[0], range[1]);
	}

	case HDIO_GETGEO: {
		struct hd_geometry geo;

		if (!arg)
			return -EINVAL;
		if (!disk->fops->getgeo)
			return -ENOTTY;

		/*
		 * 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;
		if (copy_to_user((struct hd_geometry __user *)arg, &geo,
					sizeof(geo)))
			return -EFAULT;
		return 0;
	}
	case BLKRAGET:
	case BLKFRAGET:
		if (!arg)
			return -EINVAL;
		bdi = blk_get_backing_dev_info(bdev);
		if (bdi == NULL)
			return -ENOTTY;
		return put_long(arg, (bdi->ra_pages * PAGE_CACHE_SIZE) / 512);
	case BLKROGET:
		return put_int(arg, bdev_read_only(bdev) != 0);
	case BLKBSZGET: /* get the logical block size (cf. BLKSSZGET) */
		return put_int(arg, block_size(bdev));
	case BLKSSZGET: /* get block device hardware sector size */
		return put_int(arg, bdev_logical_block_size(bdev));
	case BLKSECTGET:
		return put_ushort(arg, queue_max_sectors(bdev_get_queue(bdev)));
	case BLKRASET:
	case BLKFRASET:
		if(!capable(CAP_SYS_ADMIN))
			return -EACCES;
		bdi = blk_get_backing_dev_info(bdev);
		if (bdi == NULL)
			return -ENOTTY;
		bdi->ra_pages = (arg * 512) / PAGE_CACHE_SIZE;
		return 0;
	case BLKBSZSET:
		/* set the logical block size */
		if (!capable(CAP_SYS_ADMIN))
			return -EACCES;
		if (!arg)
			return -EINVAL;
		if (get_user(n, (int __user *) arg))
			return -EFAULT;
		if (!(mode & FMODE_EXCL) && bd_claim(bdev, &bdev) < 0)
			return -EBUSY;
		ret = set_blocksize(bdev, n);
		if (!(mode & FMODE_EXCL))
			bd_release(bdev);
		return ret;
	case BLKPG:
		lock_kernel();
		ret = blkpg_ioctl(bdev, (struct blkpg_ioctl_arg __user *) arg);
		unlock_kernel();
		break;
	case BLKRRPART:
		lock_kernel();
		ret = blkdev_reread_part(bdev);
		unlock_kernel();
		break;
	case BLKGETSIZE:
		size = bdev->bd_inode->i_size;
		if ((size >> 9) > ~0UL)
			return -EFBIG;
		return put_ulong(arg, size >> 9);
	case BLKGETSIZE64:
		return put_u64(arg, bdev->bd_inode->i_size);
	case BLKTRACESTART:
	case BLKTRACESTOP:
	case BLKTRACESETUP:
	case BLKTRACETEARDOWN:
		lock_kernel();
		ret = blk_trace_ioctl(bdev, cmd, (char __user *) arg);
		unlock_kernel();
		break;
	default:
		ret = __blkdev_driver_ioctl(bdev, mode, cmd, arg);
	}
	return ret;
}
EXPORT_SYMBOL_GPL(blkdev_ioctl);
Commit	Line	Data
c59ede7b	1	#include <linux/capability.h>
1da177e4 LT	2	#include <linux/blkdev.h>
1da177e4 LT	3	#include <linux/blkpg.h>
a885c8c4	4	#include <linux/hdreg.h>
1da177e4 LT	5	#include <linux/backing-dev.h>
	6	#include <linux/buffer_head.h>
	7	#include <linux/smp_lock.h>
2056a782	8	#include <linux/blktrace_api.h>
1da177e4 LT	9	#include <asm/uaccess.h>
	10
	11	static int blkpg_ioctl(struct block_device bdev, struct blkpg_ioctl_arg __user arg)
	12	{
	13	struct block_device *bdevp;
	14	struct gendisk *disk;
e71bf0d0	15	struct hd_struct *part;
1da177e4 LT	16	struct blkpg_ioctl_arg a;
1da177e4 LT	17	struct blkpg_partition p;
e71bf0d0	18	struct disk_part_iter piter;
1da177e4	19	long long start, length;
cf771cb5	20	int partno;
1da177e4 LT	21
	22	if (!capable(CAP_SYS_ADMIN))
	23	return -EACCES;
	24	if (copy_from_user(&a, arg, sizeof(struct blkpg_ioctl_arg)))
	25	return -EFAULT;
	26	if (copy_from_user(&p, a.data, sizeof(struct blkpg_partition)))
	27	return -EFAULT;
	28	disk = bdev->bd_disk;
	29	if (bdev != bdev->bd_contains)
	30	return -EINVAL;
cf771cb5	31	partno = p.pno;
540eed56	32	if (partno <= 0)
1da177e4 LT	33	return -EINVAL;
	34	switch (a.op) {
	35	case BLKPG_ADD_PARTITION:
	36	start = p.start >> 9;
	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)
	44	return -EINVAL;
	45	}
88e34126	46
c039e313	47	mutex_lock(&bdev->bd_mutex);
88e34126	48
1da177e4	49	/* overlap? */
e71bf0d0 TH	50	disk_part_iter_init(&piter, disk,
	51	DISK_PITER_INCL_EMPTY);
	52	while ((part = disk_part_iter_next(&piter))) {
	53	if (!(start + length <= part->start_sect \|\|
	54	start >= part->start_sect + part->nr_sects)) {
	55	disk_part_iter_exit(&piter);
c039e313	56	mutex_unlock(&bdev->bd_mutex);
1da177e4 LT	57	return -EBUSY;
	58	}
	59	}
e71bf0d0 TH	60	disk_part_iter_exit(&piter);
e71bf0d0 TH	61
1da177e4	62	/* all seems OK */
ba32929a TH	63	part = add_partition(disk, partno, start, length,
ba32929a TH	64	ADDPART_FLAG_NONE);
c039e313	65	mutex_unlock(&bdev->bd_mutex);
ba32929a	66	return IS_ERR(part) ? PTR_ERR(part) : 0;
1da177e4	67	case BLKPG_DEL_PARTITION:
e71bf0d0 TH	68	part = disk_get_part(disk, partno);
e71bf0d0 TH	69	if (!part)
1da177e4	70	return -ENXIO;
e71bf0d0 TH	71
	72	bdevp = bdget(part_devt(part));
	73	disk_put_part(part);
1da177e4 LT	74	if (!bdevp)
1da177e4 LT	75	return -ENOMEM;
e71bf0d0	76
2e7b651d	77	mutex_lock(&bdevp->bd_mutex);
1da177e4	78	if (bdevp->bd_openers) {
c039e313	79	mutex_unlock(&bdevp->bd_mutex);
1da177e4 LT	80	bdput(bdevp);
	81	return -EBUSY;
	82	}
	83	/* all seems OK */
	84	fsync_bdev(bdevp);
f98393a6	85	invalidate_bdev(bdevp);
1da177e4	86
6d740cd5	87	mutex_lock_nested(&bdev->bd_mutex, 1);
cf771cb5	88	delete_partition(disk, partno);
c039e313 AV	89	mutex_unlock(&bdev->bd_mutex);
c039e313 AV	90	mutex_unlock(&bdevp->bd_mutex);
1da177e4 LT	91	bdput(bdevp);
	92
	93	return 0;
	94	default:
	95	return -EINVAL;
	96	}
	97	}
	98
	99	static int blkdev_reread_part(struct block_device *bdev)
	100	{
	101	struct gendisk *disk = bdev->bd_disk;
	102	int res;
	103
b5d0b9df	104	if (!disk_partitionable(disk) \|\| bdev != bdev->bd_contains)
1da177e4 LT	105	return -EINVAL;
	106	if (!capable(CAP_SYS_ADMIN))
	107	return -EACCES;
c039e313	108	if (!mutex_trylock(&bdev->bd_mutex))
1da177e4 LT	109	return -EBUSY;
1da177e4 LT	110	res = rescan_partitions(disk, bdev);
c039e313	111	mutex_unlock(&bdev->bd_mutex);
1da177e4 LT	112	return res;
	113	}
	114
d30a2605 DW	115	static int blk_ioctl_discard(struct block_device *bdev, uint64_t start,
	116	uint64_t len)
	117	{
d30a2605 DW	118	if (start & 511)
	119	return -EINVAL;
	120	if (len & 511)
	121	return -EINVAL;
	122	start >>= 9;
	123	len >>= 9;
	124
	125	if (start + len > (bdev->bd_inode->i_size >> 9))
	126	return -EINVAL;
746cd1e7 CH	127	return blkdev_issue_discard(bdev, start, len, GFP_KERNEL,
746cd1e7 CH	128	DISCARD_FL_WAIT);
d30a2605 DW	129	}
d30a2605 DW	130
1da177e4 LT	131	static int put_ushort(unsigned long arg, unsigned short val)
	132	{
	133	return put_user(val, (unsigned short __user *)arg);
	134	}
	135
	136	static int put_int(unsigned long arg, int val)
	137	{
	138	return put_user(val, (int __user *)arg);
	139	}
	140
	141	static int put_long(unsigned long arg, long val)
	142	{
	143	return put_user(val, (long __user *)arg);
	144	}
	145
	146	static int put_ulong(unsigned long arg, unsigned long val)
	147	{
	148	return put_user(val, (unsigned long __user *)arg);
	149	}
	150
	151	static int put_u64(unsigned long arg, u64 val)
	152	{
	153	return put_user(val, (u64 __user *)arg);
	154	}
	155
633a08b8 AV	156	int __blkdev_driver_ioctl(struct block_device *bdev, fmode_t mode,
	157	unsigned cmd, unsigned long arg)
	158	{
	159	struct gendisk *disk = bdev->bd_disk;
	160	int ret;
d4430d62 AV	161
	162	if (disk->fops->ioctl)
	163	return disk->fops->ioctl(bdev, mode, cmd, arg);
633a08b8	164
d4430d62	165	if (disk->fops->locked_ioctl) {
633a08b8	166	lock_kernel();
d4430d62	167	ret = disk->fops->locked_ioctl(bdev, mode, cmd, arg);
633a08b8 AV	168	unlock_kernel();
	169	return ret;
	170	}
	171
	172	return -ENOTTY;
	173	}
	174	/*
	175	* For the record: _GPL here is only because somebody decided to slap it
	176	* on the previous export. Sheer idiocy, since it wasn't copyrightable
	177	* at all and could be open-coded without any exports by anybody who cares.
	178	*/
	179	EXPORT_SYMBOL_GPL(__blkdev_driver_ioctl);
	180
f58c4c0a AB	181	/*
	182	* always keep this in sync with compat_blkdev_ioctl() and
	183	* compat_blkdev_locked_ioctl()
	184	*/
56b26add	185	int blkdev_ioctl(struct block_device *bdev, fmode_t mode, unsigned cmd,
bb93e3a5 AB	186	unsigned long arg)
bb93e3a5 AB	187	{
bb93e3a5	188	struct gendisk *disk = bdev->bd_disk;
45048d09 AV	189	struct backing_dev_info *bdi;
45048d09 AV	190	loff_t size;
bb93e3a5 AB	191	int ret, n;
	192
	193	switch(cmd) {
1da177e4 LT	194	case BLKFLSBUF:
	195	if (!capable(CAP_SYS_ADMIN))
	196	return -EACCES;
bb93e3a5	197
e436fdae	198	ret = __blkdev_driver_ioctl(bdev, mode, cmd, arg);
bb93e3a5 AB	199	/* -EINVAL to handle old uncorrected drivers */
	200	if (ret != -EINVAL && ret != -ENOTTY)
	201	return ret;
	202
	203	lock_kernel();
1da177e4	204	fsync_bdev(bdev);
f98393a6	205	invalidate_bdev(bdev);
bb93e3a5	206	unlock_kernel();
1da177e4	207	return 0;
bb93e3a5	208
1da177e4	209	case BLKROSET:
e436fdae	210	ret = __blkdev_driver_ioctl(bdev, mode, cmd, arg);
bb93e3a5 AB	211	/* -EINVAL to handle old uncorrected drivers */
	212	if (ret != -EINVAL && ret != -ENOTTY)
	213	return ret;
1da177e4 LT	214	if (!capable(CAP_SYS_ADMIN))
	215	return -EACCES;
	216	if (get_user(n, (int __user *)(arg)))
	217	return -EFAULT;
bb93e3a5	218	lock_kernel();
1da177e4	219	set_device_ro(bdev, n);
bb93e3a5	220	unlock_kernel();
1da177e4	221	return 0;
d30a2605 DW	222
	223	case BLKDISCARD: {
	224	uint64_t range[2];
	225
e436fdae	226	if (!(mode & FMODE_WRITE))
d30a2605 DW	227	return -EBADF;
	228
	229	if (copy_from_user(range, (void __user *)arg, sizeof(range)))
	230	return -EFAULT;
	231
	232	return blk_ioctl_discard(bdev, range[0], range[1]);
	233	}
	234
a885c8c4 CH	235	case HDIO_GETGEO: {
	236	struct hd_geometry geo;
	237
	238	if (!arg)
	239	return -EINVAL;
	240	if (!disk->fops->getgeo)
	241	return -ENOTTY;
	242
	243	/*
	244	* We need to set the startsect first, the driver may
	245	* want to override it.
	246	*/
	247	geo.start = get_start_sect(bdev);
	248	ret = disk->fops->getgeo(bdev, &geo);
	249	if (ret)
	250	return ret;
	251	if (copy_to_user((struct hd_geometry __user *)arg, &geo,
	252	sizeof(geo)))
	253	return -EFAULT;
	254	return 0;
	255	}
45048d09 AV	256	case BLKRAGET:
	257	case BLKFRAGET:
	258	if (!arg)
	259	return -EINVAL;
	260	bdi = blk_get_backing_dev_info(bdev);
	261	if (bdi == NULL)
	262	return -ENOTTY;
	263	return put_long(arg, (bdi->ra_pages * PAGE_CACHE_SIZE) / 512);
	264	case BLKROGET:
	265	return put_int(arg, bdev_read_only(bdev) != 0);
	266	case BLKBSZGET: /* get the logical block size (cf. BLKSSZGET) */
	267	return put_int(arg, block_size(bdev));
	268	case BLKSSZGET: /* get block device hardware sector size */
e1defc4f	269	return put_int(arg, bdev_logical_block_size(bdev));
45048d09	270	case BLKSECTGET:
ae03bf63	271	return put_ushort(arg, queue_max_sectors(bdev_get_queue(bdev)));
45048d09 AV	272	case BLKRASET:
	273	case BLKFRASET:
	274	if(!capable(CAP_SYS_ADMIN))
	275	return -EACCES;
	276	bdi = blk_get_backing_dev_info(bdev);
	277	if (bdi == NULL)
	278	return -ENOTTY;
45048d09	279	bdi->ra_pages = (arg * 512) / PAGE_CACHE_SIZE;
45048d09 AV	280	return 0;
	281	case BLKBSZSET:
	282	/* set the logical block size */
	283	if (!capable(CAP_SYS_ADMIN))
	284	return -EACCES;
	285	if (!arg)
	286	return -EINVAL;
	287	if (get_user(n, (int __user *) arg))
	288	return -EFAULT;
6af3a56e	289	if (!(mode & FMODE_EXCL) && bd_claim(bdev, &bdev) < 0)
45048d09 AV	290	return -EBUSY;
45048d09 AV	291	ret = set_blocksize(bdev, n);
6af3a56e AV	292	if (!(mode & FMODE_EXCL))
6af3a56e AV	293	bd_release(bdev);
bb93e3a5	294	return ret;
45048d09 AV	295	case BLKPG:
	296	lock_kernel();
	297	ret = blkpg_ioctl(bdev, (struct blkpg_ioctl_arg __user *) arg);
	298	unlock_kernel();
	299	break;
	300	case BLKRRPART:
	301	lock_kernel();
	302	ret = blkdev_reread_part(bdev);
	303	unlock_kernel();
	304	break;
	305	case BLKGETSIZE:
	306	size = bdev->bd_inode->i_size;
	307	if ((size >> 9) > ~0UL)
	308	return -EFBIG;
	309	return put_ulong(arg, size >> 9);
	310	case BLKGETSIZE64:
	311	return put_u64(arg, bdev->bd_inode->i_size);
	312	case BLKTRACESTART:
	313	case BLKTRACESTOP:
	314	case BLKTRACESETUP:
	315	case BLKTRACETEARDOWN:
	316	lock_kernel();
	317	ret = blk_trace_ioctl(bdev, cmd, (char __user *) arg);
	318	unlock_kernel();
	319	break;
	320	default:
	321	ret = __blkdev_driver_ioctl(bdev, mode, cmd, arg);
	322	}
	323	return ret;
1da177e4	324	}
68f66feb	325	EXPORT_SYMBOL_GPL(blkdev_ioctl);