xref: /openbmc/linux/drivers/block/loop.c (revision 85560117)
11da177e4SLinus Torvalds /*
21da177e4SLinus Torvalds  *  linux/drivers/block/loop.c
31da177e4SLinus Torvalds  *
41da177e4SLinus Torvalds  *  Written by Theodore Ts'o, 3/29/93
51da177e4SLinus Torvalds  *
61da177e4SLinus Torvalds  * Copyright 1993 by Theodore Ts'o.  Redistribution of this file is
71da177e4SLinus Torvalds  * permitted under the GNU General Public License.
81da177e4SLinus Torvalds  *
91da177e4SLinus Torvalds  * DES encryption plus some minor changes by Werner Almesberger, 30-MAY-1993
101da177e4SLinus Torvalds  * more DES encryption plus IDEA encryption by Nicholas J. Leon, June 20, 1996
111da177e4SLinus Torvalds  *
121da177e4SLinus Torvalds  * Modularized and updated for 1.1.16 kernel - Mitch Dsouza 28th May 1994
131da177e4SLinus Torvalds  * Adapted for 1.3.59 kernel - Andries Brouwer, 1 Feb 1996
141da177e4SLinus Torvalds  *
151da177e4SLinus Torvalds  * Fixed do_loop_request() re-entrancy - Vincent.Renardias@waw.com Mar 20, 1997
161da177e4SLinus Torvalds  *
171da177e4SLinus Torvalds  * Added devfs support - Richard Gooch <rgooch@atnf.csiro.au> 16-Jan-1998
181da177e4SLinus Torvalds  *
191da177e4SLinus Torvalds  * Handle sparse backing files correctly - Kenn Humborg, Jun 28, 1998
201da177e4SLinus Torvalds  *
211da177e4SLinus Torvalds  * Loadable modules and other fixes by AK, 1998
221da177e4SLinus Torvalds  *
231da177e4SLinus Torvalds  * Make real block number available to downstream transfer functions, enables
241da177e4SLinus Torvalds  * CBC (and relatives) mode encryption requiring unique IVs per data block.
251da177e4SLinus Torvalds  * Reed H. Petty, rhp@draper.net
261da177e4SLinus Torvalds  *
271da177e4SLinus Torvalds  * Maximum number of loop devices now dynamic via max_loop module parameter.
281da177e4SLinus Torvalds  * Russell Kroll <rkroll@exploits.org> 19990701
291da177e4SLinus Torvalds  *
301da177e4SLinus Torvalds  * Maximum number of loop devices when compiled-in now selectable by passing
311da177e4SLinus Torvalds  * max_loop=<1-255> to the kernel on boot.
3296de0e25SJan Engelhardt  * Erik I. Bolsø, <eriki@himolde.no>, Oct 31, 1999
331da177e4SLinus Torvalds  *
341da177e4SLinus Torvalds  * Completely rewrite request handling to be make_request_fn style and
351da177e4SLinus Torvalds  * non blocking, pushing work to a helper thread. Lots of fixes from
361da177e4SLinus Torvalds  * Al Viro too.
371da177e4SLinus Torvalds  * Jens Axboe <axboe@suse.de>, Nov 2000
381da177e4SLinus Torvalds  *
391da177e4SLinus Torvalds  * Support up to 256 loop devices
401da177e4SLinus Torvalds  * Heinz Mauelshagen <mge@sistina.com>, Feb 2002
411da177e4SLinus Torvalds  *
421da177e4SLinus Torvalds  * Support for falling back on the write file operation when the address space
434e02ed4bSNick Piggin  * operations write_begin is not available on the backing filesystem.
441da177e4SLinus Torvalds  * Anton Altaparmakov, 16 Feb 2005
451da177e4SLinus Torvalds  *
461da177e4SLinus Torvalds  * Still To Fix:
471da177e4SLinus Torvalds  * - Advisory locking is ignored here.
481da177e4SLinus Torvalds  * - Should use an own CAP_* category instead of CAP_SYS_ADMIN
491da177e4SLinus Torvalds  *
501da177e4SLinus Torvalds  */
511da177e4SLinus Torvalds 
521da177e4SLinus Torvalds #include <linux/module.h>
531da177e4SLinus Torvalds #include <linux/moduleparam.h>
541da177e4SLinus Torvalds #include <linux/sched.h>
551da177e4SLinus Torvalds #include <linux/fs.h>
561da177e4SLinus Torvalds #include <linux/file.h>
571da177e4SLinus Torvalds #include <linux/stat.h>
581da177e4SLinus Torvalds #include <linux/errno.h>
591da177e4SLinus Torvalds #include <linux/major.h>
601da177e4SLinus Torvalds #include <linux/wait.h>
611da177e4SLinus Torvalds #include <linux/blkdev.h>
621da177e4SLinus Torvalds #include <linux/blkpg.h>
631da177e4SLinus Torvalds #include <linux/init.h>
641da177e4SLinus Torvalds #include <linux/swap.h>
651da177e4SLinus Torvalds #include <linux/slab.h>
66863d5b82SDavid Howells #include <linux/compat.h>
671da177e4SLinus Torvalds #include <linux/suspend.h>
6883144186SRafael J. Wysocki #include <linux/freezer.h>
692a48fc0aSArnd Bergmann #include <linux/mutex.h>
701da177e4SLinus Torvalds #include <linux/writeback.h>
711da177e4SLinus Torvalds #include <linux/completion.h>
721da177e4SLinus Torvalds #include <linux/highmem.h>
736c997918SSerge E. Hallyn #include <linux/kthread.h>
74d6b29d7cSJens Axboe #include <linux/splice.h>
75ee862730SMilan Broz #include <linux/sysfs.h>
76770fe30aSKay Sievers #include <linux/miscdevice.h>
77dfaa2ef6SLukas Czerner #include <linux/falloc.h>
78283e7e5dSAl Viro #include <linux/uio.h>
79d9a08a9eSAdam Manzanares #include <linux/ioprio.h>
80db6638d7SDennis Zhou #include <linux/blk-cgroup.h>
81d9a08a9eSAdam Manzanares 
8283a87611SAl Viro #include "loop.h"
831da177e4SLinus Torvalds 
847c0f6ba6SLinus Torvalds #include <linux/uaccess.h>
851da177e4SLinus Torvalds 
8634dd82afSKay Sievers static DEFINE_IDR(loop_index_idr);
87310ca162STetsuo Handa static DEFINE_MUTEX(loop_ctl_mutex);
881da177e4SLinus Torvalds 
89476a4813SLaurent Vivier static int max_part;
90476a4813SLaurent Vivier static int part_shift;
91476a4813SLaurent Vivier 
921da177e4SLinus Torvalds static int transfer_xor(struct loop_device *lo, int cmd,
931da177e4SLinus Torvalds 			struct page *raw_page, unsigned raw_off,
941da177e4SLinus Torvalds 			struct page *loop_page, unsigned loop_off,
951da177e4SLinus Torvalds 			int size, sector_t real_block)
961da177e4SLinus Torvalds {
97cfd8005cSCong Wang 	char *raw_buf = kmap_atomic(raw_page) + raw_off;
98cfd8005cSCong Wang 	char *loop_buf = kmap_atomic(loop_page) + loop_off;
991da177e4SLinus Torvalds 	char *in, *out, *key;
1001da177e4SLinus Torvalds 	int i, keysize;
1011da177e4SLinus Torvalds 
1021da177e4SLinus Torvalds 	if (cmd == READ) {
1031da177e4SLinus Torvalds 		in = raw_buf;
1041da177e4SLinus Torvalds 		out = loop_buf;
1051da177e4SLinus Torvalds 	} else {
1061da177e4SLinus Torvalds 		in = loop_buf;
1071da177e4SLinus Torvalds 		out = raw_buf;
1081da177e4SLinus Torvalds 	}
1091da177e4SLinus Torvalds 
1101da177e4SLinus Torvalds 	key = lo->lo_encrypt_key;
1111da177e4SLinus Torvalds 	keysize = lo->lo_encrypt_key_size;
1121da177e4SLinus Torvalds 	for (i = 0; i < size; i++)
1131da177e4SLinus Torvalds 		*out++ = *in++ ^ key[(i & 511) % keysize];
1141da177e4SLinus Torvalds 
115cfd8005cSCong Wang 	kunmap_atomic(loop_buf);
116cfd8005cSCong Wang 	kunmap_atomic(raw_buf);
1171da177e4SLinus Torvalds 	cond_resched();
1181da177e4SLinus Torvalds 	return 0;
1191da177e4SLinus Torvalds }
1201da177e4SLinus Torvalds 
1211da177e4SLinus Torvalds static int xor_init(struct loop_device *lo, const struct loop_info64 *info)
1221da177e4SLinus Torvalds {
1231da177e4SLinus Torvalds 	if (unlikely(info->lo_encrypt_key_size <= 0))
1241da177e4SLinus Torvalds 		return -EINVAL;
1251da177e4SLinus Torvalds 	return 0;
1261da177e4SLinus Torvalds }
1271da177e4SLinus Torvalds 
1281da177e4SLinus Torvalds static struct loop_func_table none_funcs = {
1291da177e4SLinus Torvalds 	.number = LO_CRYPT_NONE,
1301da177e4SLinus Torvalds };
1311da177e4SLinus Torvalds 
1321da177e4SLinus Torvalds static struct loop_func_table xor_funcs = {
1331da177e4SLinus Torvalds 	.number = LO_CRYPT_XOR,
1341da177e4SLinus Torvalds 	.transfer = transfer_xor,
1351da177e4SLinus Torvalds 	.init = xor_init
1361da177e4SLinus Torvalds };
1371da177e4SLinus Torvalds 
1381da177e4SLinus Torvalds /* xfer_funcs[0] is special - its release function is never called */
1391da177e4SLinus Torvalds static struct loop_func_table *xfer_funcs[MAX_LO_CRYPT] = {
1401da177e4SLinus Torvalds 	&none_funcs,
1411da177e4SLinus Torvalds 	&xor_funcs
1421da177e4SLinus Torvalds };
1431da177e4SLinus Torvalds 
1447035b5dfSDmitry Monakhov static loff_t get_size(loff_t offset, loff_t sizelimit, struct file *file)
1451da177e4SLinus Torvalds {
146b7a1da69SGuo Chao 	loff_t loopsize;
1471da177e4SLinus Torvalds 
1481da177e4SLinus Torvalds 	/* Compute loopsize in bytes */
149b7a1da69SGuo Chao 	loopsize = i_size_read(file->f_mapping->host);
150b7a1da69SGuo Chao 	if (offset > 0)
151b7a1da69SGuo Chao 		loopsize -= offset;
152b7a1da69SGuo Chao 	/* offset is beyond i_size, weird but possible */
1537035b5dfSDmitry Monakhov 	if (loopsize < 0)
1547035b5dfSDmitry Monakhov 		return 0;
1551da177e4SLinus Torvalds 
1567035b5dfSDmitry Monakhov 	if (sizelimit > 0 && sizelimit < loopsize)
1577035b5dfSDmitry Monakhov 		loopsize = sizelimit;
1581da177e4SLinus Torvalds 	/*
1591da177e4SLinus Torvalds 	 * Unfortunately, if we want to do I/O on the device,
1601da177e4SLinus Torvalds 	 * the number of 512-byte sectors has to fit into a sector_t.
1611da177e4SLinus Torvalds 	 */
1621da177e4SLinus Torvalds 	return loopsize >> 9;
1631da177e4SLinus Torvalds }
1641da177e4SLinus Torvalds 
1657035b5dfSDmitry Monakhov static loff_t get_loop_size(struct loop_device *lo, struct file *file)
1661da177e4SLinus Torvalds {
1677035b5dfSDmitry Monakhov 	return get_size(lo->lo_offset, lo->lo_sizelimit, file);
1687035b5dfSDmitry Monakhov }
1697035b5dfSDmitry Monakhov 
1702e5ab5f3SMing Lei static void __loop_update_dio(struct loop_device *lo, bool dio)
1712e5ab5f3SMing Lei {
1722e5ab5f3SMing Lei 	struct file *file = lo->lo_backing_file;
1732e5ab5f3SMing Lei 	struct address_space *mapping = file->f_mapping;
1742e5ab5f3SMing Lei 	struct inode *inode = mapping->host;
1752e5ab5f3SMing Lei 	unsigned short sb_bsize = 0;
1762e5ab5f3SMing Lei 	unsigned dio_align = 0;
1772e5ab5f3SMing Lei 	bool use_dio;
1782e5ab5f3SMing Lei 
1792e5ab5f3SMing Lei 	if (inode->i_sb->s_bdev) {
1802e5ab5f3SMing Lei 		sb_bsize = bdev_logical_block_size(inode->i_sb->s_bdev);
1812e5ab5f3SMing Lei 		dio_align = sb_bsize - 1;
1822e5ab5f3SMing Lei 	}
1832e5ab5f3SMing Lei 
1842e5ab5f3SMing Lei 	/*
1852e5ab5f3SMing Lei 	 * We support direct I/O only if lo_offset is aligned with the
1862e5ab5f3SMing Lei 	 * logical I/O size of backing device, and the logical block
1872e5ab5f3SMing Lei 	 * size of loop is bigger than the backing device's and the loop
1882e5ab5f3SMing Lei 	 * needn't transform transfer.
1892e5ab5f3SMing Lei 	 *
1902e5ab5f3SMing Lei 	 * TODO: the above condition may be loosed in the future, and
1912e5ab5f3SMing Lei 	 * direct I/O may be switched runtime at that time because most
19289d790abSMasahiro Yamada 	 * of requests in sane applications should be PAGE_SIZE aligned
1932e5ab5f3SMing Lei 	 */
1942e5ab5f3SMing Lei 	if (dio) {
1952e5ab5f3SMing Lei 		if (queue_logical_block_size(lo->lo_queue) >= sb_bsize &&
1962e5ab5f3SMing Lei 				!(lo->lo_offset & dio_align) &&
1972e5ab5f3SMing Lei 				mapping->a_ops->direct_IO &&
1982e5ab5f3SMing Lei 				!lo->transfer)
1992e5ab5f3SMing Lei 			use_dio = true;
2002e5ab5f3SMing Lei 		else
2012e5ab5f3SMing Lei 			use_dio = false;
2022e5ab5f3SMing Lei 	} else {
2032e5ab5f3SMing Lei 		use_dio = false;
2042e5ab5f3SMing Lei 	}
2052e5ab5f3SMing Lei 
2062e5ab5f3SMing Lei 	if (lo->use_dio == use_dio)
2072e5ab5f3SMing Lei 		return;
2082e5ab5f3SMing Lei 
2092e5ab5f3SMing Lei 	/* flush dirty pages before changing direct IO */
2102e5ab5f3SMing Lei 	vfs_fsync(file, 0);
2112e5ab5f3SMing Lei 
2122e5ab5f3SMing Lei 	/*
2132e5ab5f3SMing Lei 	 * The flag of LO_FLAGS_DIRECT_IO is handled similarly with
2142e5ab5f3SMing Lei 	 * LO_FLAGS_READ_ONLY, both are set from kernel, and losetup
2152e5ab5f3SMing Lei 	 * will get updated by ioctl(LOOP_GET_STATUS)
2162e5ab5f3SMing Lei 	 */
2172e5ab5f3SMing Lei 	blk_mq_freeze_queue(lo->lo_queue);
2182e5ab5f3SMing Lei 	lo->use_dio = use_dio;
21940326d8aSShaohua Li 	if (use_dio) {
2208b904b5bSBart Van Assche 		blk_queue_flag_clear(QUEUE_FLAG_NOMERGES, lo->lo_queue);
2212e5ab5f3SMing Lei 		lo->lo_flags |= LO_FLAGS_DIRECT_IO;
22240326d8aSShaohua Li 	} else {
2238b904b5bSBart Van Assche 		blk_queue_flag_set(QUEUE_FLAG_NOMERGES, lo->lo_queue);
2242e5ab5f3SMing Lei 		lo->lo_flags &= ~LO_FLAGS_DIRECT_IO;
22540326d8aSShaohua Li 	}
2262e5ab5f3SMing Lei 	blk_mq_unfreeze_queue(lo->lo_queue);
2272e5ab5f3SMing Lei }
2282e5ab5f3SMing Lei 
2297035b5dfSDmitry Monakhov static int
2301e6ec9eaSOmar Sandoval figure_loop_size(struct loop_device *lo, loff_t offset, loff_t sizelimit)
2317035b5dfSDmitry Monakhov {
2327035b5dfSDmitry Monakhov 	loff_t size = get_size(offset, sizelimit, lo->lo_backing_file);
2331da177e4SLinus Torvalds 	sector_t x = (sector_t)size;
2347b0576a3SGuo Chao 	struct block_device *bdev = lo->lo_device;
2351da177e4SLinus Torvalds 
2361da177e4SLinus Torvalds 	if (unlikely((loff_t)x != size))
2371da177e4SLinus Torvalds 		return -EFBIG;
2387035b5dfSDmitry Monakhov 	if (lo->lo_offset != offset)
2397035b5dfSDmitry Monakhov 		lo->lo_offset = offset;
2407035b5dfSDmitry Monakhov 	if (lo->lo_sizelimit != sizelimit)
2417035b5dfSDmitry Monakhov 		lo->lo_sizelimit = sizelimit;
24273285082SKen Chen 	set_capacity(lo->lo_disk, x);
2437b0576a3SGuo Chao 	bd_set_size(bdev, (loff_t)get_capacity(bdev->bd_disk) << 9);
2447b0576a3SGuo Chao 	/* let user-space know about the new size */
2457b0576a3SGuo Chao 	kobject_uevent(&disk_to_dev(bdev->bd_disk)->kobj, KOBJ_CHANGE);
2461da177e4SLinus Torvalds 	return 0;
2471da177e4SLinus Torvalds }
2481da177e4SLinus Torvalds 
2491da177e4SLinus Torvalds static inline int
2501da177e4SLinus Torvalds lo_do_transfer(struct loop_device *lo, int cmd,
2511da177e4SLinus Torvalds 	       struct page *rpage, unsigned roffs,
2521da177e4SLinus Torvalds 	       struct page *lpage, unsigned loffs,
2531da177e4SLinus Torvalds 	       int size, sector_t rblock)
2541da177e4SLinus Torvalds {
255aa4d8616SChristoph Hellwig 	int ret;
256aa4d8616SChristoph Hellwig 
257aa4d8616SChristoph Hellwig 	ret = lo->transfer(lo, cmd, rpage, roffs, lpage, loffs, size, rblock);
258aa4d8616SChristoph Hellwig 	if (likely(!ret))
2591da177e4SLinus Torvalds 		return 0;
2601da177e4SLinus Torvalds 
261aa4d8616SChristoph Hellwig 	printk_ratelimited(KERN_ERR
262aa4d8616SChristoph Hellwig 		"loop: Transfer error at byte offset %llu, length %i.\n",
263aa4d8616SChristoph Hellwig 		(unsigned long long)rblock << 9, size);
264aa4d8616SChristoph Hellwig 	return ret;
2651da177e4SLinus Torvalds }
2661da177e4SLinus Torvalds 
267aa4d8616SChristoph Hellwig static int lo_write_bvec(struct file *file, struct bio_vec *bvec, loff_t *ppos)
2681da177e4SLinus Torvalds {
269aa4d8616SChristoph Hellwig 	struct iov_iter i;
2701da177e4SLinus Torvalds 	ssize_t bw;
271283e7e5dSAl Viro 
272b6207430SChristoph Hellwig 	iov_iter_bvec(&i, WRITE, bvec, 1, bvec->bv_len);
2731da177e4SLinus Torvalds 
27403d95eb2SAl Viro 	file_start_write(file);
275abbb6589SChristoph Hellwig 	bw = vfs_iter_write(file, &i, ppos, 0);
27603d95eb2SAl Viro 	file_end_write(file);
277aa4d8616SChristoph Hellwig 
278aa4d8616SChristoph Hellwig 	if (likely(bw ==  bvec->bv_len))
2791da177e4SLinus Torvalds 		return 0;
280aa4d8616SChristoph Hellwig 
281aa4d8616SChristoph Hellwig 	printk_ratelimited(KERN_ERR
282aa4d8616SChristoph Hellwig 		"loop: Write error at byte offset %llu, length %i.\n",
283aa4d8616SChristoph Hellwig 		(unsigned long long)*ppos, bvec->bv_len);
2841da177e4SLinus Torvalds 	if (bw >= 0)
2851da177e4SLinus Torvalds 		bw = -EIO;
2861da177e4SLinus Torvalds 	return bw;
2871da177e4SLinus Torvalds }
2881da177e4SLinus Torvalds 
289aa4d8616SChristoph Hellwig static int lo_write_simple(struct loop_device *lo, struct request *rq,
290aa4d8616SChristoph Hellwig 		loff_t pos)
2911da177e4SLinus Torvalds {
292aa4d8616SChristoph Hellwig 	struct bio_vec bvec;
293aa4d8616SChristoph Hellwig 	struct req_iterator iter;
294aa4d8616SChristoph Hellwig 	int ret = 0;
295aa4d8616SChristoph Hellwig 
296aa4d8616SChristoph Hellwig 	rq_for_each_segment(bvec, rq, iter) {
297aa4d8616SChristoph Hellwig 		ret = lo_write_bvec(lo->lo_backing_file, &bvec, &pos);
298aa4d8616SChristoph Hellwig 		if (ret < 0)
299aa4d8616SChristoph Hellwig 			break;
3001da177e4SLinus Torvalds 		cond_resched();
3011da177e4SLinus Torvalds 	}
3021da177e4SLinus Torvalds 
303aa4d8616SChristoph Hellwig 	return ret;
304aa4d8616SChristoph Hellwig }
305aa4d8616SChristoph Hellwig 
306aa4d8616SChristoph Hellwig /*
307456be148SChristoph Hellwig  * This is the slow, transforming version that needs to double buffer the
308456be148SChristoph Hellwig  * data as it cannot do the transformations in place without having direct
309456be148SChristoph Hellwig  * access to the destination pages of the backing file.
3101da177e4SLinus Torvalds  */
311aa4d8616SChristoph Hellwig static int lo_write_transfer(struct loop_device *lo, struct request *rq,
312aa4d8616SChristoph Hellwig 		loff_t pos)
3131da177e4SLinus Torvalds {
314aa4d8616SChristoph Hellwig 	struct bio_vec bvec, b;
31530112013SMing Lei 	struct req_iterator iter;
316aa4d8616SChristoph Hellwig 	struct page *page;
3177988613bSKent Overstreet 	int ret = 0;
3181da177e4SLinus Torvalds 
319aa4d8616SChristoph Hellwig 	page = alloc_page(GFP_NOIO);
3201da177e4SLinus Torvalds 	if (unlikely(!page))
321aa4d8616SChristoph Hellwig 		return -ENOMEM;
322456be148SChristoph Hellwig 
32330112013SMing Lei 	rq_for_each_segment(bvec, rq, iter) {
324aa4d8616SChristoph Hellwig 		ret = lo_do_transfer(lo, WRITE, page, 0, bvec.bv_page,
325aa4d8616SChristoph Hellwig 			bvec.bv_offset, bvec.bv_len, pos >> 9);
326aa4d8616SChristoph Hellwig 		if (unlikely(ret))
327aa4d8616SChristoph Hellwig 			break;
328aa4d8616SChristoph Hellwig 
329aa4d8616SChristoph Hellwig 		b.bv_page = page;
330aa4d8616SChristoph Hellwig 		b.bv_offset = 0;
331aa4d8616SChristoph Hellwig 		b.bv_len = bvec.bv_len;
332aa4d8616SChristoph Hellwig 		ret = lo_write_bvec(lo->lo_backing_file, &b, &pos);
3331da177e4SLinus Torvalds 		if (ret < 0)
3341da177e4SLinus Torvalds 			break;
3351da177e4SLinus Torvalds 	}
336aa4d8616SChristoph Hellwig 
3371da177e4SLinus Torvalds 	__free_page(page);
3381da177e4SLinus Torvalds 	return ret;
3391da177e4SLinus Torvalds }
3401da177e4SLinus Torvalds 
341aa4d8616SChristoph Hellwig static int lo_read_simple(struct loop_device *lo, struct request *rq,
342aa4d8616SChristoph Hellwig 		loff_t pos)
3431da177e4SLinus Torvalds {
3447988613bSKent Overstreet 	struct bio_vec bvec;
34530112013SMing Lei 	struct req_iterator iter;
346aa4d8616SChristoph Hellwig 	struct iov_iter i;
347aa4d8616SChristoph Hellwig 	ssize_t len;
3481da177e4SLinus Torvalds 
34930112013SMing Lei 	rq_for_each_segment(bvec, rq, iter) {
350b6207430SChristoph Hellwig 		iov_iter_bvec(&i, READ, &bvec, 1, bvec.bv_len);
35118e9710eSChristoph Hellwig 		len = vfs_iter_read(lo->lo_backing_file, &i, &pos, 0);
352aa4d8616SChristoph Hellwig 		if (len < 0)
353aa4d8616SChristoph Hellwig 			return len;
354306df071SDave Young 
355aa4d8616SChristoph Hellwig 		flush_dcache_page(bvec.bv_page);
356aa4d8616SChristoph Hellwig 
357aa4d8616SChristoph Hellwig 		if (len != bvec.bv_len) {
35830112013SMing Lei 			struct bio *bio;
35930112013SMing Lei 
36030112013SMing Lei 			__rq_for_each_bio(bio, rq)
361306df071SDave Young 				zero_fill_bio(bio);
3621da177e4SLinus Torvalds 			break;
363306df071SDave Young 		}
364aa4d8616SChristoph Hellwig 		cond_resched();
3651da177e4SLinus Torvalds 	}
366aa4d8616SChristoph Hellwig 
367306df071SDave Young 	return 0;
3681da177e4SLinus Torvalds }
3691da177e4SLinus Torvalds 
370aa4d8616SChristoph Hellwig static int lo_read_transfer(struct loop_device *lo, struct request *rq,
371aa4d8616SChristoph Hellwig 		loff_t pos)
372aa4d8616SChristoph Hellwig {
373aa4d8616SChristoph Hellwig 	struct bio_vec bvec, b;
374aa4d8616SChristoph Hellwig 	struct req_iterator iter;
375aa4d8616SChristoph Hellwig 	struct iov_iter i;
376aa4d8616SChristoph Hellwig 	struct page *page;
377aa4d8616SChristoph Hellwig 	ssize_t len;
378aa4d8616SChristoph Hellwig 	int ret = 0;
379aa4d8616SChristoph Hellwig 
380aa4d8616SChristoph Hellwig 	page = alloc_page(GFP_NOIO);
381aa4d8616SChristoph Hellwig 	if (unlikely(!page))
382aa4d8616SChristoph Hellwig 		return -ENOMEM;
383aa4d8616SChristoph Hellwig 
384aa4d8616SChristoph Hellwig 	rq_for_each_segment(bvec, rq, iter) {
385aa4d8616SChristoph Hellwig 		loff_t offset = pos;
386aa4d8616SChristoph Hellwig 
387aa4d8616SChristoph Hellwig 		b.bv_page = page;
388aa4d8616SChristoph Hellwig 		b.bv_offset = 0;
389aa4d8616SChristoph Hellwig 		b.bv_len = bvec.bv_len;
390aa4d8616SChristoph Hellwig 
391b6207430SChristoph Hellwig 		iov_iter_bvec(&i, READ, &b, 1, b.bv_len);
39218e9710eSChristoph Hellwig 		len = vfs_iter_read(lo->lo_backing_file, &i, &pos, 0);
393aa4d8616SChristoph Hellwig 		if (len < 0) {
394aa4d8616SChristoph Hellwig 			ret = len;
395aa4d8616SChristoph Hellwig 			goto out_free_page;
396aa4d8616SChristoph Hellwig 		}
397aa4d8616SChristoph Hellwig 
398aa4d8616SChristoph Hellwig 		ret = lo_do_transfer(lo, READ, page, 0, bvec.bv_page,
399aa4d8616SChristoph Hellwig 			bvec.bv_offset, len, offset >> 9);
400aa4d8616SChristoph Hellwig 		if (ret)
401aa4d8616SChristoph Hellwig 			goto out_free_page;
402aa4d8616SChristoph Hellwig 
403aa4d8616SChristoph Hellwig 		flush_dcache_page(bvec.bv_page);
404aa4d8616SChristoph Hellwig 
405aa4d8616SChristoph Hellwig 		if (len != bvec.bv_len) {
406aa4d8616SChristoph Hellwig 			struct bio *bio;
407aa4d8616SChristoph Hellwig 
408aa4d8616SChristoph Hellwig 			__rq_for_each_bio(bio, rq)
409aa4d8616SChristoph Hellwig 				zero_fill_bio(bio);
410aa4d8616SChristoph Hellwig 			break;
411aa4d8616SChristoph Hellwig 		}
412aa4d8616SChristoph Hellwig 	}
413aa4d8616SChristoph Hellwig 
414aa4d8616SChristoph Hellwig 	ret = 0;
415aa4d8616SChristoph Hellwig out_free_page:
416aa4d8616SChristoph Hellwig 	__free_page(page);
417aa4d8616SChristoph Hellwig 	return ret;
418aa4d8616SChristoph Hellwig }
419aa4d8616SChristoph Hellwig 
420cf655d95SMing Lei static int lo_discard(struct loop_device *lo, struct request *rq, loff_t pos)
421cf655d95SMing Lei {
422cf655d95SMing Lei 	/*
423cf655d95SMing Lei 	 * We use punch hole to reclaim the free space used by the
424cf655d95SMing Lei 	 * image a.k.a. discard. However we do not support discard if
425cf655d95SMing Lei 	 * encryption is enabled, because it may give an attacker
426cf655d95SMing Lei 	 * useful information.
427cf655d95SMing Lei 	 */
428cf655d95SMing Lei 	struct file *file = lo->lo_backing_file;
429cf655d95SMing Lei 	int mode = FALLOC_FL_PUNCH_HOLE | FALLOC_FL_KEEP_SIZE;
430cf655d95SMing Lei 	int ret;
431cf655d95SMing Lei 
432cf655d95SMing Lei 	if ((!file->f_op->fallocate) || lo->lo_encrypt_key_size) {
433cf655d95SMing Lei 		ret = -EOPNOTSUPP;
434cf655d95SMing Lei 		goto out;
435cf655d95SMing Lei 	}
436cf655d95SMing Lei 
437cf655d95SMing Lei 	ret = file->f_op->fallocate(file, mode, pos, blk_rq_bytes(rq));
438cf655d95SMing Lei 	if (unlikely(ret && ret != -EINVAL && ret != -EOPNOTSUPP))
439cf655d95SMing Lei 		ret = -EIO;
440cf655d95SMing Lei  out:
441cf655d95SMing Lei 	return ret;
442cf655d95SMing Lei }
443cf655d95SMing Lei 
444cf655d95SMing Lei static int lo_req_flush(struct loop_device *lo, struct request *rq)
445cf655d95SMing Lei {
446cf655d95SMing Lei 	struct file *file = lo->lo_backing_file;
447cf655d95SMing Lei 	int ret = vfs_fsync(file, 0);
448cf655d95SMing Lei 	if (unlikely(ret && ret != -EINVAL))
449cf655d95SMing Lei 		ret = -EIO;
450cf655d95SMing Lei 
451cf655d95SMing Lei 	return ret;
452cf655d95SMing Lei }
453cf655d95SMing Lei 
454fe2cb290SChristoph Hellwig static void lo_complete_rq(struct request *rq)
455bc07c10aSMing Lei {
456fe2cb290SChristoph Hellwig 	struct loop_cmd *cmd = blk_mq_rq_to_pdu(rq);
457f9de14bcSJens Axboe 	blk_status_t ret = BLK_STS_OK;
458bc07c10aSMing Lei 
459f9de14bcSJens Axboe 	if (!cmd->use_aio || cmd->ret < 0 || cmd->ret == blk_rq_bytes(rq) ||
460f9de14bcSJens Axboe 	    req_op(rq) != REQ_OP_READ) {
461f9de14bcSJens Axboe 		if (cmd->ret < 0)
462f9de14bcSJens Axboe 			ret = BLK_STS_IOERR;
463f9de14bcSJens Axboe 		goto end_io;
464bc07c10aSMing Lei 	}
465fe2cb290SChristoph Hellwig 
466f9de14bcSJens Axboe 	/*
467f9de14bcSJens Axboe 	 * Short READ - if we got some data, advance our request and
468f9de14bcSJens Axboe 	 * retry it. If we got no data, end the rest with EIO.
469f9de14bcSJens Axboe 	 */
470f9de14bcSJens Axboe 	if (cmd->ret) {
471f9de14bcSJens Axboe 		blk_update_request(rq, BLK_STS_OK, cmd->ret);
472f9de14bcSJens Axboe 		cmd->ret = 0;
473f9de14bcSJens Axboe 		blk_mq_requeue_request(rq, true);
474f9de14bcSJens Axboe 	} else {
475f9de14bcSJens Axboe 		if (cmd->use_aio) {
476f9de14bcSJens Axboe 			struct bio *bio = rq->bio;
477f9de14bcSJens Axboe 
478f9de14bcSJens Axboe 			while (bio) {
479f9de14bcSJens Axboe 				zero_fill_bio(bio);
480f9de14bcSJens Axboe 				bio = bio->bi_next;
481f9de14bcSJens Axboe 			}
482f9de14bcSJens Axboe 		}
483f9de14bcSJens Axboe 		ret = BLK_STS_IOERR;
484f9de14bcSJens Axboe end_io:
485f9de14bcSJens Axboe 		blk_mq_end_request(rq, ret);
486f9de14bcSJens Axboe 	}
487bc07c10aSMing Lei }
488bc07c10aSMing Lei 
48992d77332SShaohua Li static void lo_rw_aio_do_completion(struct loop_cmd *cmd)
49092d77332SShaohua Li {
4911894e916SJens Axboe 	struct request *rq = blk_mq_rq_from_pdu(cmd);
4921894e916SJens Axboe 
49392d77332SShaohua Li 	if (!atomic_dec_and_test(&cmd->ref))
49492d77332SShaohua Li 		return;
49592d77332SShaohua Li 	kfree(cmd->bvec);
49692d77332SShaohua Li 	cmd->bvec = NULL;
4971894e916SJens Axboe 	blk_mq_complete_request(rq);
49892d77332SShaohua Li }
49992d77332SShaohua Li 
500bc07c10aSMing Lei static void lo_rw_aio_complete(struct kiocb *iocb, long ret, long ret2)
501bc07c10aSMing Lei {
502bc07c10aSMing Lei 	struct loop_cmd *cmd = container_of(iocb, struct loop_cmd, iocb);
503bc07c10aSMing Lei 
504d4478e92SShaohua Li 	if (cmd->css)
505d4478e92SShaohua Li 		css_put(cmd->css);
506fe2cb290SChristoph Hellwig 	cmd->ret = ret;
50792d77332SShaohua Li 	lo_rw_aio_do_completion(cmd);
508bc07c10aSMing Lei }
509bc07c10aSMing Lei 
510bc07c10aSMing Lei static int lo_rw_aio(struct loop_device *lo, struct loop_cmd *cmd,
511bc07c10aSMing Lei 		     loff_t pos, bool rw)
512bc07c10aSMing Lei {
513bc07c10aSMing Lei 	struct iov_iter iter;
51486af5952SMing Lei 	struct req_iterator rq_iter;
515bc07c10aSMing Lei 	struct bio_vec *bvec;
5161894e916SJens Axboe 	struct request *rq = blk_mq_rq_from_pdu(cmd);
51740326d8aSShaohua Li 	struct bio *bio = rq->bio;
518bc07c10aSMing Lei 	struct file *file = lo->lo_backing_file;
51986af5952SMing Lei 	struct bio_vec tmp;
52040326d8aSShaohua Li 	unsigned int offset;
52186af5952SMing Lei 	int nr_bvec = 0;
522bc07c10aSMing Lei 	int ret;
523bc07c10aSMing Lei 
52486af5952SMing Lei 	rq_for_each_bvec(tmp, rq, rq_iter)
52586af5952SMing Lei 		nr_bvec++;
526bc07c10aSMing Lei 
52786af5952SMing Lei 	if (rq->bio != rq->biotail) {
52886af5952SMing Lei 
52986af5952SMing Lei 		bvec = kmalloc_array(nr_bvec, sizeof(struct bio_vec),
5306da2ec56SKees Cook 				     GFP_NOIO);
53140326d8aSShaohua Li 		if (!bvec)
53240326d8aSShaohua Li 			return -EIO;
53340326d8aSShaohua Li 		cmd->bvec = bvec;
53440326d8aSShaohua Li 
535a7297a6aSMing Lei 		/*
53640326d8aSShaohua Li 		 * The bios of the request may be started from the middle of
53740326d8aSShaohua Li 		 * the 'bvec' because of bio splitting, so we can't directly
53886af5952SMing Lei 		 * copy bio->bi_iov_vec to new bvec. The rq_for_each_bvec
53940326d8aSShaohua Li 		 * API will take care of all details for us.
540a7297a6aSMing Lei 		 */
54186af5952SMing Lei 		rq_for_each_bvec(tmp, rq, rq_iter) {
54240326d8aSShaohua Li 			*bvec = tmp;
54340326d8aSShaohua Li 			bvec++;
54440326d8aSShaohua Li 		}
54540326d8aSShaohua Li 		bvec = cmd->bvec;
54640326d8aSShaohua Li 		offset = 0;
54740326d8aSShaohua Li 	} else {
54840326d8aSShaohua Li 		/*
54940326d8aSShaohua Li 		 * Same here, this bio may be started from the middle of the
55040326d8aSShaohua Li 		 * 'bvec' because of bio splitting, so offset from the bvec
55140326d8aSShaohua Li 		 * must be passed to iov iterator
55240326d8aSShaohua Li 		 */
55340326d8aSShaohua Li 		offset = bio->bi_iter.bi_bvec_done;
55440326d8aSShaohua Li 		bvec = __bvec_iter_bvec(bio->bi_io_vec, bio->bi_iter);
55540326d8aSShaohua Li 	}
55692d77332SShaohua Li 	atomic_set(&cmd->ref, 2);
55740326d8aSShaohua Li 
558b6207430SChristoph Hellwig 	iov_iter_bvec(&iter, rw, bvec, nr_bvec, blk_rq_bytes(rq));
55940326d8aSShaohua Li 	iter.iov_offset = offset;
560bc07c10aSMing Lei 
561bc07c10aSMing Lei 	cmd->iocb.ki_pos = pos;
562bc07c10aSMing Lei 	cmd->iocb.ki_filp = file;
563bc07c10aSMing Lei 	cmd->iocb.ki_complete = lo_rw_aio_complete;
564bc07c10aSMing Lei 	cmd->iocb.ki_flags = IOCB_DIRECT;
565d9a08a9eSAdam Manzanares 	cmd->iocb.ki_ioprio = IOPRIO_PRIO_VALUE(IOPRIO_CLASS_NONE, 0);
566d4478e92SShaohua Li 	if (cmd->css)
567d4478e92SShaohua Li 		kthread_associate_blkcg(cmd->css);
568bc07c10aSMing Lei 
569bc07c10aSMing Lei 	if (rw == WRITE)
570bb7462b6SMiklos Szeredi 		ret = call_write_iter(file, &cmd->iocb, &iter);
571bc07c10aSMing Lei 	else
572bb7462b6SMiklos Szeredi 		ret = call_read_iter(file, &cmd->iocb, &iter);
573bc07c10aSMing Lei 
57492d77332SShaohua Li 	lo_rw_aio_do_completion(cmd);
575d4478e92SShaohua Li 	kthread_associate_blkcg(NULL);
57692d77332SShaohua Li 
577bc07c10aSMing Lei 	if (ret != -EIOCBQUEUED)
578bc07c10aSMing Lei 		cmd->iocb.ki_complete(&cmd->iocb, ret, 0);
579bc07c10aSMing Lei 	return 0;
580bc07c10aSMing Lei }
581bc07c10aSMing Lei 
582c1c87c2bSChristoph Hellwig static int do_req_filebacked(struct loop_device *lo, struct request *rq)
583bc07c10aSMing Lei {
584bc07c10aSMing Lei 	struct loop_cmd *cmd = blk_mq_rq_to_pdu(rq);
585c1c87c2bSChristoph Hellwig 	loff_t pos = ((loff_t) blk_rq_pos(rq) << 9) + lo->lo_offset;
586bc07c10aSMing Lei 
587bc07c10aSMing Lei 	/*
588bc07c10aSMing Lei 	 * lo_write_simple and lo_read_simple should have been covered
589bc07c10aSMing Lei 	 * by io submit style function like lo_rw_aio(), one blocker
590bc07c10aSMing Lei 	 * is that lo_read_simple() need to call flush_dcache_page after
591bc07c10aSMing Lei 	 * the page is written from kernel, and it isn't easy to handle
592bc07c10aSMing Lei 	 * this in io submit style function which submits all segments
593bc07c10aSMing Lei 	 * of the req at one time. And direct read IO doesn't need to
594bc07c10aSMing Lei 	 * run flush_dcache_page().
595bc07c10aSMing Lei 	 */
596c1c87c2bSChristoph Hellwig 	switch (req_op(rq)) {
597c1c87c2bSChristoph Hellwig 	case REQ_OP_FLUSH:
598c1c87c2bSChristoph Hellwig 		return lo_req_flush(lo, rq);
599c1c87c2bSChristoph Hellwig 	case REQ_OP_DISCARD:
60019372e27SChristoph Hellwig 	case REQ_OP_WRITE_ZEROES:
601c1c87c2bSChristoph Hellwig 		return lo_discard(lo, rq, pos);
602c1c87c2bSChristoph Hellwig 	case REQ_OP_WRITE:
603c1c87c2bSChristoph Hellwig 		if (lo->transfer)
604c1c87c2bSChristoph Hellwig 			return lo_write_transfer(lo, rq, pos);
605c1c87c2bSChristoph Hellwig 		else if (cmd->use_aio)
606c1c87c2bSChristoph Hellwig 			return lo_rw_aio(lo, cmd, pos, WRITE);
607c1c87c2bSChristoph Hellwig 		else
608bc07c10aSMing Lei 			return lo_write_simple(lo, rq, pos);
609c1c87c2bSChristoph Hellwig 	case REQ_OP_READ:
610c1c87c2bSChristoph Hellwig 		if (lo->transfer)
611c1c87c2bSChristoph Hellwig 			return lo_read_transfer(lo, rq, pos);
612c1c87c2bSChristoph Hellwig 		else if (cmd->use_aio)
613c1c87c2bSChristoph Hellwig 			return lo_rw_aio(lo, cmd, pos, READ);
614bc07c10aSMing Lei 		else
615bc07c10aSMing Lei 			return lo_read_simple(lo, rq, pos);
616c1c87c2bSChristoph Hellwig 	default:
617c1c87c2bSChristoph Hellwig 		WARN_ON_ONCE(1);
618c1c87c2bSChristoph Hellwig 		return -EIO;
619bc07c10aSMing Lei 	}
6201da177e4SLinus Torvalds }
6211da177e4SLinus Torvalds 
6222e5ab5f3SMing Lei static inline void loop_update_dio(struct loop_device *lo)
6232e5ab5f3SMing Lei {
6242e5ab5f3SMing Lei 	__loop_update_dio(lo, io_is_direct(lo->lo_backing_file) |
6252e5ab5f3SMing Lei 			lo->use_dio);
6262e5ab5f3SMing Lei }
6272e5ab5f3SMing Lei 
62806f0e9e6SMing Lei static void loop_reread_partitions(struct loop_device *lo,
62906f0e9e6SMing Lei 				   struct block_device *bdev)
63006f0e9e6SMing Lei {
63106f0e9e6SMing Lei 	int rc;
63206f0e9e6SMing Lei 
63306f0e9e6SMing Lei 	rc = blkdev_reread_part(bdev);
63406f0e9e6SMing Lei 	if (rc)
63506f0e9e6SMing Lei 		pr_warn("%s: partition scan of loop%d (%s) failed (rc=%d)\n",
63606f0e9e6SMing Lei 			__func__, lo->lo_number, lo->lo_file_name, rc);
63706f0e9e6SMing Lei }
63806f0e9e6SMing Lei 
639d2ac838eSTheodore Ts'o static inline int is_loop_device(struct file *file)
640d2ac838eSTheodore Ts'o {
641d2ac838eSTheodore Ts'o 	struct inode *i = file->f_mapping->host;
642d2ac838eSTheodore Ts'o 
643d2ac838eSTheodore Ts'o 	return i && S_ISBLK(i->i_mode) && MAJOR(i->i_rdev) == LOOP_MAJOR;
644d2ac838eSTheodore Ts'o }
645d2ac838eSTheodore Ts'o 
646d2ac838eSTheodore Ts'o static int loop_validate_file(struct file *file, struct block_device *bdev)
647d2ac838eSTheodore Ts'o {
648d2ac838eSTheodore Ts'o 	struct inode	*inode = file->f_mapping->host;
649d2ac838eSTheodore Ts'o 	struct file	*f = file;
650d2ac838eSTheodore Ts'o 
651d2ac838eSTheodore Ts'o 	/* Avoid recursion */
652d2ac838eSTheodore Ts'o 	while (is_loop_device(f)) {
653d2ac838eSTheodore Ts'o 		struct loop_device *l;
654d2ac838eSTheodore Ts'o 
655d2ac838eSTheodore Ts'o 		if (f->f_mapping->host->i_bdev == bdev)
656d2ac838eSTheodore Ts'o 			return -EBADF;
657d2ac838eSTheodore Ts'o 
658d2ac838eSTheodore Ts'o 		l = f->f_mapping->host->i_bdev->bd_disk->private_data;
659f7c8a412SDongli Zhang 		if (l->lo_state != Lo_bound) {
660d2ac838eSTheodore Ts'o 			return -EINVAL;
661d2ac838eSTheodore Ts'o 		}
662d2ac838eSTheodore Ts'o 		f = l->lo_backing_file;
663d2ac838eSTheodore Ts'o 	}
664d2ac838eSTheodore Ts'o 	if (!S_ISREG(inode->i_mode) && !S_ISBLK(inode->i_mode))
665d2ac838eSTheodore Ts'o 		return -EINVAL;
666d2ac838eSTheodore Ts'o 	return 0;
667d2ac838eSTheodore Ts'o }
668d2ac838eSTheodore Ts'o 
6691da177e4SLinus Torvalds /*
6701da177e4SLinus Torvalds  * loop_change_fd switched the backing store of a loopback device to
6711da177e4SLinus Torvalds  * a new file. This is useful for operating system installers to free up
6721da177e4SLinus Torvalds  * the original file and in High Availability environments to switch to
6731da177e4SLinus Torvalds  * an alternative location for the content in case of server meltdown.
6741da177e4SLinus Torvalds  * This can only work if the loop device is used read-only, and if the
6751da177e4SLinus Torvalds  * new backing store is the same size and type as the old backing store.
6761da177e4SLinus Torvalds  */
677bb214884SAl Viro static int loop_change_fd(struct loop_device *lo, struct block_device *bdev,
678bb214884SAl Viro 			  unsigned int arg)
6791da177e4SLinus Torvalds {
6801dded9acSJan Kara 	struct file	*file = NULL, *old_file;
6811da177e4SLinus Torvalds 	int		error;
68285b0a54aSJan Kara 	bool		partscan;
6831da177e4SLinus Torvalds 
684c28445faSJan Kara 	error = mutex_lock_killable(&loop_ctl_mutex);
685c3710770SJan Kara 	if (error)
686c3710770SJan Kara 		return error;
6871da177e4SLinus Torvalds 	error = -ENXIO;
6881da177e4SLinus Torvalds 	if (lo->lo_state != Lo_bound)
6891dded9acSJan Kara 		goto out_err;
6901da177e4SLinus Torvalds 
6911da177e4SLinus Torvalds 	/* the loop device has to be read-only */
6921da177e4SLinus Torvalds 	error = -EINVAL;
6931da177e4SLinus Torvalds 	if (!(lo->lo_flags & LO_FLAGS_READ_ONLY))
6941dded9acSJan Kara 		goto out_err;
6951da177e4SLinus Torvalds 
6961da177e4SLinus Torvalds 	error = -EBADF;
6971da177e4SLinus Torvalds 	file = fget(arg);
6981da177e4SLinus Torvalds 	if (!file)
6991dded9acSJan Kara 		goto out_err;
7001da177e4SLinus Torvalds 
701d2ac838eSTheodore Ts'o 	error = loop_validate_file(file, bdev);
702d2ac838eSTheodore Ts'o 	if (error)
7031dded9acSJan Kara 		goto out_err;
704d2ac838eSTheodore Ts'o 
7051da177e4SLinus Torvalds 	old_file = lo->lo_backing_file;
7061da177e4SLinus Torvalds 
7071da177e4SLinus Torvalds 	error = -EINVAL;
7081da177e4SLinus Torvalds 
7091da177e4SLinus Torvalds 	/* size of the new backing store needs to be the same */
7101da177e4SLinus Torvalds 	if (get_loop_size(lo, file) != get_loop_size(lo, old_file))
7111dded9acSJan Kara 		goto out_err;
7121da177e4SLinus Torvalds 
7131da177e4SLinus Torvalds 	/* and ... switch */
71443cade80SOmar Sandoval 	blk_mq_freeze_queue(lo->lo_queue);
71543cade80SOmar Sandoval 	mapping_set_gfp_mask(old_file->f_mapping, lo->old_gfp_mask);
71643cade80SOmar Sandoval 	lo->lo_backing_file = file;
71743cade80SOmar Sandoval 	lo->old_gfp_mask = mapping_gfp_mask(file->f_mapping);
71843cade80SOmar Sandoval 	mapping_set_gfp_mask(file->f_mapping,
71943cade80SOmar Sandoval 			     lo->old_gfp_mask & ~(__GFP_IO|__GFP_FS));
72043cade80SOmar Sandoval 	loop_update_dio(lo);
72143cade80SOmar Sandoval 	blk_mq_unfreeze_queue(lo->lo_queue);
72285b0a54aSJan Kara 	partscan = lo->lo_flags & LO_FLAGS_PARTSCAN;
723c3710770SJan Kara 	mutex_unlock(&loop_ctl_mutex);
7241dded9acSJan Kara 	/*
7251dded9acSJan Kara 	 * We must drop file reference outside of loop_ctl_mutex as dropping
7261dded9acSJan Kara 	 * the file ref can take bd_mutex which creates circular locking
7271dded9acSJan Kara 	 * dependency.
7281dded9acSJan Kara 	 */
7291dded9acSJan Kara 	fput(old_file);
73085b0a54aSJan Kara 	if (partscan)
73185b0a54aSJan Kara 		loop_reread_partitions(lo, bdev);
7321da177e4SLinus Torvalds 	return 0;
7331da177e4SLinus Torvalds 
7341dded9acSJan Kara out_err:
735c3710770SJan Kara 	mutex_unlock(&loop_ctl_mutex);
7361dded9acSJan Kara 	if (file)
7371dded9acSJan Kara 		fput(file);
7381da177e4SLinus Torvalds 	return error;
7391da177e4SLinus Torvalds }
7401da177e4SLinus Torvalds 
741ee862730SMilan Broz /* loop sysfs attributes */
742ee862730SMilan Broz 
743ee862730SMilan Broz static ssize_t loop_attr_show(struct device *dev, char *page,
744ee862730SMilan Broz 			      ssize_t (*callback)(struct loop_device *, char *))
745ee862730SMilan Broz {
74634dd82afSKay Sievers 	struct gendisk *disk = dev_to_disk(dev);
74734dd82afSKay Sievers 	struct loop_device *lo = disk->private_data;
748ee862730SMilan Broz 
74934dd82afSKay Sievers 	return callback(lo, page);
750ee862730SMilan Broz }
751ee862730SMilan Broz 
752ee862730SMilan Broz #define LOOP_ATTR_RO(_name)						\
753ee862730SMilan Broz static ssize_t loop_attr_##_name##_show(struct loop_device *, char *);	\
754ee862730SMilan Broz static ssize_t loop_attr_do_show_##_name(struct device *d,		\
755ee862730SMilan Broz 				struct device_attribute *attr, char *b)	\
756ee862730SMilan Broz {									\
757ee862730SMilan Broz 	return loop_attr_show(d, b, loop_attr_##_name##_show);		\
758ee862730SMilan Broz }									\
759ee862730SMilan Broz static struct device_attribute loop_attr_##_name =			\
7605657a819SJoe Perches 	__ATTR(_name, 0444, loop_attr_do_show_##_name, NULL);
761ee862730SMilan Broz 
762ee862730SMilan Broz static ssize_t loop_attr_backing_file_show(struct loop_device *lo, char *buf)
763ee862730SMilan Broz {
764ee862730SMilan Broz 	ssize_t ret;
765ee862730SMilan Broz 	char *p = NULL;
766ee862730SMilan Broz 
76705eb0f25SKay Sievers 	spin_lock_irq(&lo->lo_lock);
768ee862730SMilan Broz 	if (lo->lo_backing_file)
7699bf39ab2SMiklos Szeredi 		p = file_path(lo->lo_backing_file, buf, PAGE_SIZE - 1);
77005eb0f25SKay Sievers 	spin_unlock_irq(&lo->lo_lock);
771ee862730SMilan Broz 
772ee862730SMilan Broz 	if (IS_ERR_OR_NULL(p))
773ee862730SMilan Broz 		ret = PTR_ERR(p);
774ee862730SMilan Broz 	else {
775ee862730SMilan Broz 		ret = strlen(p);
776ee862730SMilan Broz 		memmove(buf, p, ret);
777ee862730SMilan Broz 		buf[ret++] = '\n';
778ee862730SMilan Broz 		buf[ret] = 0;
779ee862730SMilan Broz 	}
780ee862730SMilan Broz 
781ee862730SMilan Broz 	return ret;
782ee862730SMilan Broz }
783ee862730SMilan Broz 
784ee862730SMilan Broz static ssize_t loop_attr_offset_show(struct loop_device *lo, char *buf)
785ee862730SMilan Broz {
786ee862730SMilan Broz 	return sprintf(buf, "%llu\n", (unsigned long long)lo->lo_offset);
787ee862730SMilan Broz }
788ee862730SMilan Broz 
789ee862730SMilan Broz static ssize_t loop_attr_sizelimit_show(struct loop_device *lo, char *buf)
790ee862730SMilan Broz {
791ee862730SMilan Broz 	return sprintf(buf, "%llu\n", (unsigned long long)lo->lo_sizelimit);
792ee862730SMilan Broz }
793ee862730SMilan Broz 
794ee862730SMilan Broz static ssize_t loop_attr_autoclear_show(struct loop_device *lo, char *buf)
795ee862730SMilan Broz {
796ee862730SMilan Broz 	int autoclear = (lo->lo_flags & LO_FLAGS_AUTOCLEAR);
797ee862730SMilan Broz 
798ee862730SMilan Broz 	return sprintf(buf, "%s\n", autoclear ? "1" : "0");
799ee862730SMilan Broz }
800ee862730SMilan Broz 
801e03c8dd1SKay Sievers static ssize_t loop_attr_partscan_show(struct loop_device *lo, char *buf)
802e03c8dd1SKay Sievers {
803e03c8dd1SKay Sievers 	int partscan = (lo->lo_flags & LO_FLAGS_PARTSCAN);
804e03c8dd1SKay Sievers 
805e03c8dd1SKay Sievers 	return sprintf(buf, "%s\n", partscan ? "1" : "0");
806e03c8dd1SKay Sievers }
807e03c8dd1SKay Sievers 
8082e5ab5f3SMing Lei static ssize_t loop_attr_dio_show(struct loop_device *lo, char *buf)
8092e5ab5f3SMing Lei {
8102e5ab5f3SMing Lei 	int dio = (lo->lo_flags & LO_FLAGS_DIRECT_IO);
8112e5ab5f3SMing Lei 
8122e5ab5f3SMing Lei 	return sprintf(buf, "%s\n", dio ? "1" : "0");
8132e5ab5f3SMing Lei }
8142e5ab5f3SMing Lei 
815ee862730SMilan Broz LOOP_ATTR_RO(backing_file);
816ee862730SMilan Broz LOOP_ATTR_RO(offset);
817ee862730SMilan Broz LOOP_ATTR_RO(sizelimit);
818ee862730SMilan Broz LOOP_ATTR_RO(autoclear);
819e03c8dd1SKay Sievers LOOP_ATTR_RO(partscan);
8202e5ab5f3SMing Lei LOOP_ATTR_RO(dio);
821ee862730SMilan Broz 
822ee862730SMilan Broz static struct attribute *loop_attrs[] = {
823ee862730SMilan Broz 	&loop_attr_backing_file.attr,
824ee862730SMilan Broz 	&loop_attr_offset.attr,
825ee862730SMilan Broz 	&loop_attr_sizelimit.attr,
826ee862730SMilan Broz 	&loop_attr_autoclear.attr,
827e03c8dd1SKay Sievers 	&loop_attr_partscan.attr,
8282e5ab5f3SMing Lei 	&loop_attr_dio.attr,
829ee862730SMilan Broz 	NULL,
830ee862730SMilan Broz };
831ee862730SMilan Broz 
832ee862730SMilan Broz static struct attribute_group loop_attribute_group = {
833ee862730SMilan Broz 	.name = "loop",
834ee862730SMilan Broz 	.attrs= loop_attrs,
835ee862730SMilan Broz };
836ee862730SMilan Broz 
837d3349b6bSTetsuo Handa static void loop_sysfs_init(struct loop_device *lo)
838ee862730SMilan Broz {
839d3349b6bSTetsuo Handa 	lo->sysfs_inited = !sysfs_create_group(&disk_to_dev(lo->lo_disk)->kobj,
840ee862730SMilan Broz 						&loop_attribute_group);
841ee862730SMilan Broz }
842ee862730SMilan Broz 
843ee862730SMilan Broz static void loop_sysfs_exit(struct loop_device *lo)
844ee862730SMilan Broz {
845d3349b6bSTetsuo Handa 	if (lo->sysfs_inited)
846ee862730SMilan Broz 		sysfs_remove_group(&disk_to_dev(lo->lo_disk)->kobj,
847ee862730SMilan Broz 				   &loop_attribute_group);
848ee862730SMilan Broz }
849ee862730SMilan Broz 
850dfaa2ef6SLukas Czerner static void loop_config_discard(struct loop_device *lo)
851dfaa2ef6SLukas Czerner {
852dfaa2ef6SLukas Czerner 	struct file *file = lo->lo_backing_file;
853dfaa2ef6SLukas Czerner 	struct inode *inode = file->f_mapping->host;
854dfaa2ef6SLukas Czerner 	struct request_queue *q = lo->lo_queue;
855dfaa2ef6SLukas Czerner 
856dfaa2ef6SLukas Czerner 	/*
857dfaa2ef6SLukas Czerner 	 * We use punch hole to reclaim the free space used by the
85812a64d2fSOlaf Hering 	 * image a.k.a. discard. However we do not support discard if
859dfaa2ef6SLukas Czerner 	 * encryption is enabled, because it may give an attacker
860dfaa2ef6SLukas Czerner 	 * useful information.
861dfaa2ef6SLukas Czerner 	 */
862dfaa2ef6SLukas Czerner 	if ((!file->f_op->fallocate) ||
863dfaa2ef6SLukas Czerner 	    lo->lo_encrypt_key_size) {
864dfaa2ef6SLukas Czerner 		q->limits.discard_granularity = 0;
865dfaa2ef6SLukas Czerner 		q->limits.discard_alignment = 0;
8662bb4cd5cSJens Axboe 		blk_queue_max_discard_sectors(q, 0);
86719372e27SChristoph Hellwig 		blk_queue_max_write_zeroes_sectors(q, 0);
8688b904b5bSBart Van Assche 		blk_queue_flag_clear(QUEUE_FLAG_DISCARD, q);
869dfaa2ef6SLukas Czerner 		return;
870dfaa2ef6SLukas Czerner 	}
871dfaa2ef6SLukas Czerner 
872dfaa2ef6SLukas Czerner 	q->limits.discard_granularity = inode->i_sb->s_blocksize;
873dfaf3c03SLukas Czerner 	q->limits.discard_alignment = 0;
874f2c6df7dSHannes Reinecke 
8751e6ec9eaSOmar Sandoval 	blk_queue_max_discard_sectors(q, UINT_MAX >> 9);
8761e6ec9eaSOmar Sandoval 	blk_queue_max_write_zeroes_sectors(q, UINT_MAX >> 9);
8778b904b5bSBart Van Assche 	blk_queue_flag_set(QUEUE_FLAG_DISCARD, q);
878dfaa2ef6SLukas Czerner }
879dfaa2ef6SLukas Czerner 
880e03a3d7aSMing Lei static void loop_unprepare_queue(struct loop_device *lo)
881e03a3d7aSMing Lei {
8823989144fSPetr Mladek 	kthread_flush_worker(&lo->worker);
883e03a3d7aSMing Lei 	kthread_stop(lo->worker_task);
884e03a3d7aSMing Lei }
885e03a3d7aSMing Lei 
886b2ee7d46SNeilBrown static int loop_kthread_worker_fn(void *worker_ptr)
887b2ee7d46SNeilBrown {
888d0a255e7SMikulas Patocka 	current->flags |= PF_LESS_THROTTLE | PF_MEMALLOC_NOIO;
889b2ee7d46SNeilBrown 	return kthread_worker_fn(worker_ptr);
890b2ee7d46SNeilBrown }
891b2ee7d46SNeilBrown 
892e03a3d7aSMing Lei static int loop_prepare_queue(struct loop_device *lo)
893e03a3d7aSMing Lei {
8943989144fSPetr Mladek 	kthread_init_worker(&lo->worker);
895b2ee7d46SNeilBrown 	lo->worker_task = kthread_run(loop_kthread_worker_fn,
896e03a3d7aSMing Lei 			&lo->worker, "loop%d", lo->lo_number);
897e03a3d7aSMing Lei 	if (IS_ERR(lo->worker_task))
898e03a3d7aSMing Lei 		return -ENOMEM;
899e03a3d7aSMing Lei 	set_user_nice(lo->worker_task, MIN_NICE);
900e03a3d7aSMing Lei 	return 0;
901e03a3d7aSMing Lei }
902e03a3d7aSMing Lei 
90356a85fd8SHolger Hoffstätte static void loop_update_rotational(struct loop_device *lo)
90456a85fd8SHolger Hoffstätte {
90556a85fd8SHolger Hoffstätte 	struct file *file = lo->lo_backing_file;
90656a85fd8SHolger Hoffstätte 	struct inode *file_inode = file->f_mapping->host;
90756a85fd8SHolger Hoffstätte 	struct block_device *file_bdev = file_inode->i_sb->s_bdev;
90856a85fd8SHolger Hoffstätte 	struct request_queue *q = lo->lo_queue;
90956a85fd8SHolger Hoffstätte 	bool nonrot = true;
91056a85fd8SHolger Hoffstätte 
91156a85fd8SHolger Hoffstätte 	/* not all filesystems (e.g. tmpfs) have a sb->s_bdev */
91256a85fd8SHolger Hoffstätte 	if (file_bdev)
91356a85fd8SHolger Hoffstätte 		nonrot = blk_queue_nonrot(bdev_get_queue(file_bdev));
91456a85fd8SHolger Hoffstätte 
91556a85fd8SHolger Hoffstätte 	if (nonrot)
91656a85fd8SHolger Hoffstätte 		blk_queue_flag_set(QUEUE_FLAG_NONROT, q);
91756a85fd8SHolger Hoffstätte 	else
91856a85fd8SHolger Hoffstätte 		blk_queue_flag_clear(QUEUE_FLAG_NONROT, q);
91956a85fd8SHolger Hoffstätte }
92056a85fd8SHolger Hoffstätte 
921bb214884SAl Viro static int loop_set_fd(struct loop_device *lo, fmode_t mode,
9221da177e4SLinus Torvalds 		       struct block_device *bdev, unsigned int arg)
9231da177e4SLinus Torvalds {
924d2ac838eSTheodore Ts'o 	struct file	*file;
9251da177e4SLinus Torvalds 	struct inode	*inode;
9261da177e4SLinus Torvalds 	struct address_space *mapping;
92789e524c0SJan Kara 	struct block_device *claimed_bdev = NULL;
9281da177e4SLinus Torvalds 	int		lo_flags = 0;
9291da177e4SLinus Torvalds 	int		error;
9301da177e4SLinus Torvalds 	loff_t		size;
93185b0a54aSJan Kara 	bool		partscan;
9321da177e4SLinus Torvalds 
9331da177e4SLinus Torvalds 	/* This is safe, since we have a reference from open(). */
9341da177e4SLinus Torvalds 	__module_get(THIS_MODULE);
9351da177e4SLinus Torvalds 
9361da177e4SLinus Torvalds 	error = -EBADF;
9371da177e4SLinus Torvalds 	file = fget(arg);
9381da177e4SLinus Torvalds 	if (!file)
9391da177e4SLinus Torvalds 		goto out;
9401da177e4SLinus Torvalds 
94133ec3e53SJan Kara 	/*
94233ec3e53SJan Kara 	 * If we don't hold exclusive handle for the device, upgrade to it
94333ec3e53SJan Kara 	 * here to avoid changing device under exclusive owner.
94433ec3e53SJan Kara 	 */
94533ec3e53SJan Kara 	if (!(mode & FMODE_EXCL)) {
94689e524c0SJan Kara 		claimed_bdev = bd_start_claiming(bdev, loop_set_fd);
94789e524c0SJan Kara 		if (IS_ERR(claimed_bdev)) {
94889e524c0SJan Kara 			error = PTR_ERR(claimed_bdev);
949757ecf40SJan Kara 			goto out_putf;
95033ec3e53SJan Kara 		}
95189e524c0SJan Kara 	}
95233ec3e53SJan Kara 
95333ec3e53SJan Kara 	error = mutex_lock_killable(&loop_ctl_mutex);
95433ec3e53SJan Kara 	if (error)
95533ec3e53SJan Kara 		goto out_bdev;
956757ecf40SJan Kara 
9571da177e4SLinus Torvalds 	error = -EBUSY;
9581da177e4SLinus Torvalds 	if (lo->lo_state != Lo_unbound)
959757ecf40SJan Kara 		goto out_unlock;
9601da177e4SLinus Torvalds 
961d2ac838eSTheodore Ts'o 	error = loop_validate_file(file, bdev);
962d2ac838eSTheodore Ts'o 	if (error)
963757ecf40SJan Kara 		goto out_unlock;
9641da177e4SLinus Torvalds 
9651da177e4SLinus Torvalds 	mapping = file->f_mapping;
9661da177e4SLinus Torvalds 	inode = mapping->host;
9671da177e4SLinus Torvalds 
968456be148SChristoph Hellwig 	if (!(file->f_mode & FMODE_WRITE) || !(mode & FMODE_WRITE) ||
969283e7e5dSAl Viro 	    !file->f_op->write_iter)
9701da177e4SLinus Torvalds 		lo_flags |= LO_FLAGS_READ_ONLY;
9711da177e4SLinus Torvalds 
9721da177e4SLinus Torvalds 	error = -EFBIG;
973456be148SChristoph Hellwig 	size = get_loop_size(lo, file);
974456be148SChristoph Hellwig 	if ((loff_t)(sector_t)size != size)
975757ecf40SJan Kara 		goto out_unlock;
976e03a3d7aSMing Lei 	error = loop_prepare_queue(lo);
977e03a3d7aSMing Lei 	if (error)
978757ecf40SJan Kara 		goto out_unlock;
9791da177e4SLinus Torvalds 
980456be148SChristoph Hellwig 	error = 0;
9811da177e4SLinus Torvalds 
9821da177e4SLinus Torvalds 	set_device_ro(bdev, (lo_flags & LO_FLAGS_READ_ONLY) != 0);
9831da177e4SLinus Torvalds 
9842e5ab5f3SMing Lei 	lo->use_dio = false;
9851da177e4SLinus Torvalds 	lo->lo_device = bdev;
9861da177e4SLinus Torvalds 	lo->lo_flags = lo_flags;
9871da177e4SLinus Torvalds 	lo->lo_backing_file = file;
988aa4d8616SChristoph Hellwig 	lo->transfer = NULL;
9891da177e4SLinus Torvalds 	lo->ioctl = NULL;
9901da177e4SLinus Torvalds 	lo->lo_sizelimit = 0;
9911da177e4SLinus Torvalds 	lo->old_gfp_mask = mapping_gfp_mask(mapping);
9921da177e4SLinus Torvalds 	mapping_set_gfp_mask(mapping, lo->old_gfp_mask & ~(__GFP_IO|__GFP_FS));
9931da177e4SLinus Torvalds 
99468db1961SNikanth Karthikesan 	if (!(lo_flags & LO_FLAGS_READ_ONLY) && file->f_op->fsync)
99521d0727fSJens Axboe 		blk_queue_write_cache(lo->lo_queue, true, false);
99668db1961SNikanth Karthikesan 
99785560117SMartijn Coenen 	if (io_is_direct(lo->lo_backing_file) && inode->i_sb->s_bdev) {
99885560117SMartijn Coenen 		/* In case of direct I/O, match underlying block size */
99985560117SMartijn Coenen 		unsigned short bsize = bdev_logical_block_size(
100085560117SMartijn Coenen 			inode->i_sb->s_bdev);
100185560117SMartijn Coenen 
100285560117SMartijn Coenen 		blk_queue_logical_block_size(lo->lo_queue, bsize);
100385560117SMartijn Coenen 		blk_queue_physical_block_size(lo->lo_queue, bsize);
100485560117SMartijn Coenen 		blk_queue_io_min(lo->lo_queue, bsize);
100585560117SMartijn Coenen 	}
100685560117SMartijn Coenen 
100756a85fd8SHolger Hoffstätte 	loop_update_rotational(lo);
10082e5ab5f3SMing Lei 	loop_update_dio(lo);
100973285082SKen Chen 	set_capacity(lo->lo_disk, size);
10101da177e4SLinus Torvalds 	bd_set_size(bdev, size << 9);
1011ee862730SMilan Broz 	loop_sysfs_init(lo);
1012c3473c63SDavid Zeuthen 	/* let user-space know about the new size */
1013c3473c63SDavid Zeuthen 	kobject_uevent(&disk_to_dev(bdev->bd_disk)->kobj, KOBJ_CHANGE);
10141da177e4SLinus Torvalds 
10158a0740c4SOmar Sandoval 	set_blocksize(bdev, S_ISBLK(inode->i_mode) ?
10168a0740c4SOmar Sandoval 		      block_size(inode->i_bdev) : PAGE_SIZE);
10171da177e4SLinus Torvalds 
10186c997918SSerge E. Hallyn 	lo->lo_state = Lo_bound;
1019e03c8dd1SKay Sievers 	if (part_shift)
1020e03c8dd1SKay Sievers 		lo->lo_flags |= LO_FLAGS_PARTSCAN;
102185b0a54aSJan Kara 	partscan = lo->lo_flags & LO_FLAGS_PARTSCAN;
1022c1681bf8SAnatol Pomozov 
1023c1681bf8SAnatol Pomozov 	/* Grab the block_device to prevent its destruction after we
1024a2505b79SJan Kara 	 * put /dev/loopXX inode. Later in __loop_clr_fd() we bdput(bdev).
1025c1681bf8SAnatol Pomozov 	 */
1026c1681bf8SAnatol Pomozov 	bdgrab(bdev);
1027757ecf40SJan Kara 	mutex_unlock(&loop_ctl_mutex);
102885b0a54aSJan Kara 	if (partscan)
102985b0a54aSJan Kara 		loop_reread_partitions(lo, bdev);
103089e524c0SJan Kara 	if (claimed_bdev)
103189e524c0SJan Kara 		bd_abort_claiming(bdev, claimed_bdev, loop_set_fd);
10321da177e4SLinus Torvalds 	return 0;
10331da177e4SLinus Torvalds 
1034757ecf40SJan Kara out_unlock:
1035757ecf40SJan Kara 	mutex_unlock(&loop_ctl_mutex);
103633ec3e53SJan Kara out_bdev:
103789e524c0SJan Kara 	if (claimed_bdev)
103889e524c0SJan Kara 		bd_abort_claiming(bdev, claimed_bdev, loop_set_fd);
10391da177e4SLinus Torvalds out_putf:
10401da177e4SLinus Torvalds 	fput(file);
10411da177e4SLinus Torvalds out:
10421da177e4SLinus Torvalds 	/* This is safe: open() is still holding a reference. */
10431da177e4SLinus Torvalds 	module_put(THIS_MODULE);
10441da177e4SLinus Torvalds 	return error;
10451da177e4SLinus Torvalds }
10461da177e4SLinus Torvalds 
10471da177e4SLinus Torvalds static int
10481da177e4SLinus Torvalds loop_release_xfer(struct loop_device *lo)
10491da177e4SLinus Torvalds {
10501da177e4SLinus Torvalds 	int err = 0;
10511da177e4SLinus Torvalds 	struct loop_func_table *xfer = lo->lo_encryption;
10521da177e4SLinus Torvalds 
10531da177e4SLinus Torvalds 	if (xfer) {
10541da177e4SLinus Torvalds 		if (xfer->release)
10551da177e4SLinus Torvalds 			err = xfer->release(lo);
10561da177e4SLinus Torvalds 		lo->transfer = NULL;
10571da177e4SLinus Torvalds 		lo->lo_encryption = NULL;
10581da177e4SLinus Torvalds 		module_put(xfer->owner);
10591da177e4SLinus Torvalds 	}
10601da177e4SLinus Torvalds 	return err;
10611da177e4SLinus Torvalds }
10621da177e4SLinus Torvalds 
10631da177e4SLinus Torvalds static int
10641da177e4SLinus Torvalds loop_init_xfer(struct loop_device *lo, struct loop_func_table *xfer,
10651da177e4SLinus Torvalds 	       const struct loop_info64 *i)
10661da177e4SLinus Torvalds {
10671da177e4SLinus Torvalds 	int err = 0;
10681da177e4SLinus Torvalds 
10691da177e4SLinus Torvalds 	if (xfer) {
10701da177e4SLinus Torvalds 		struct module *owner = xfer->owner;
10711da177e4SLinus Torvalds 
10721da177e4SLinus Torvalds 		if (!try_module_get(owner))
10731da177e4SLinus Torvalds 			return -EINVAL;
10741da177e4SLinus Torvalds 		if (xfer->init)
10751da177e4SLinus Torvalds 			err = xfer->init(lo, i);
10761da177e4SLinus Torvalds 		if (err)
10771da177e4SLinus Torvalds 			module_put(owner);
10781da177e4SLinus Torvalds 		else
10791da177e4SLinus Torvalds 			lo->lo_encryption = xfer;
10801da177e4SLinus Torvalds 	}
10811da177e4SLinus Torvalds 	return err;
10821da177e4SLinus Torvalds }
10831da177e4SLinus Torvalds 
10840da03cabSJan Kara static int __loop_clr_fd(struct loop_device *lo, bool release)
10851da177e4SLinus Torvalds {
10867ccd0791SJan Kara 	struct file *filp = NULL;
1087b4e3ca1aSAl Viro 	gfp_t gfp = lo->old_gfp_mask;
10884c823cc3SAyan George 	struct block_device *bdev = lo->lo_device;
10897ccd0791SJan Kara 	int err = 0;
10900da03cabSJan Kara 	bool partscan = false;
10910da03cabSJan Kara 	int lo_number;
10921da177e4SLinus Torvalds 
10937ccd0791SJan Kara 	mutex_lock(&loop_ctl_mutex);
10947ccd0791SJan Kara 	if (WARN_ON_ONCE(lo->lo_state != Lo_rundown)) {
10957ccd0791SJan Kara 		err = -ENXIO;
10967ccd0791SJan Kara 		goto out_unlock;
10977ccd0791SJan Kara 	}
10981da177e4SLinus Torvalds 
10997ccd0791SJan Kara 	filp = lo->lo_backing_file;
11007ccd0791SJan Kara 	if (filp == NULL) {
11017ccd0791SJan Kara 		err = -EINVAL;
11027ccd0791SJan Kara 		goto out_unlock;
11037ccd0791SJan Kara 	}
11041da177e4SLinus Torvalds 
1105f8933667SMing Lei 	/* freeze request queue during the transition */
1106f8933667SMing Lei 	blk_mq_freeze_queue(lo->lo_queue);
1107f8933667SMing Lei 
11081da177e4SLinus Torvalds 	spin_lock_irq(&lo->lo_lock);
11091da177e4SLinus Torvalds 	lo->lo_backing_file = NULL;
111005eb0f25SKay Sievers 	spin_unlock_irq(&lo->lo_lock);
11111da177e4SLinus Torvalds 
11121da177e4SLinus Torvalds 	loop_release_xfer(lo);
11131da177e4SLinus Torvalds 	lo->transfer = NULL;
11141da177e4SLinus Torvalds 	lo->ioctl = NULL;
11151da177e4SLinus Torvalds 	lo->lo_device = NULL;
11161da177e4SLinus Torvalds 	lo->lo_encryption = NULL;
11171da177e4SLinus Torvalds 	lo->lo_offset = 0;
11181da177e4SLinus Torvalds 	lo->lo_sizelimit = 0;
11191da177e4SLinus Torvalds 	lo->lo_encrypt_key_size = 0;
11201da177e4SLinus Torvalds 	memset(lo->lo_encrypt_key, 0, LO_KEY_SIZE);
11211da177e4SLinus Torvalds 	memset(lo->lo_crypt_name, 0, LO_NAME_SIZE);
11221da177e4SLinus Torvalds 	memset(lo->lo_file_name, 0, LO_NAME_SIZE);
112389e4fdecSOmar Sandoval 	blk_queue_logical_block_size(lo->lo_queue, 512);
1124bf093753SOmar Sandoval 	blk_queue_physical_block_size(lo->lo_queue, 512);
1125bf093753SOmar Sandoval 	blk_queue_io_min(lo->lo_queue, 512);
1126c1681bf8SAnatol Pomozov 	if (bdev) {
1127c1681bf8SAnatol Pomozov 		bdput(bdev);
1128f98393a6SPeter Zijlstra 		invalidate_bdev(bdev);
1129eedffa28SJeff Layton 		bdev->bd_inode->i_mapping->wb_err = 0;
1130c1681bf8SAnatol Pomozov 	}
113173285082SKen Chen 	set_capacity(lo->lo_disk, 0);
113251a0bb0cSMilan Broz 	loop_sysfs_exit(lo);
1133c3473c63SDavid Zeuthen 	if (bdev) {
11341da177e4SLinus Torvalds 		bd_set_size(bdev, 0);
1135c3473c63SDavid Zeuthen 		/* let user-space know about this change */
1136c3473c63SDavid Zeuthen 		kobject_uevent(&disk_to_dev(bdev->bd_disk)->kobj, KOBJ_CHANGE);
1137c3473c63SDavid Zeuthen 	}
11381da177e4SLinus Torvalds 	mapping_set_gfp_mask(filp->f_mapping, gfp);
11391da177e4SLinus Torvalds 	/* This is safe: open() is still holding a reference. */
11401da177e4SLinus Torvalds 	module_put(THIS_MODULE);
1141f8933667SMing Lei 	blk_mq_unfreeze_queue(lo->lo_queue);
1142f8933667SMing Lei 
11430da03cabSJan Kara 	partscan = lo->lo_flags & LO_FLAGS_PARTSCAN && bdev;
11440da03cabSJan Kara 	lo_number = lo->lo_number;
11450da03cabSJan Kara 	loop_unprepare_queue(lo);
11460da03cabSJan Kara out_unlock:
11470da03cabSJan Kara 	mutex_unlock(&loop_ctl_mutex);
11480da03cabSJan Kara 	if (partscan) {
1149d57f3374SJan Kara 		/*
1150d57f3374SJan Kara 		 * bd_mutex has been held already in release path, so don't
1151d57f3374SJan Kara 		 * acquire it if this function is called in such case.
1152d57f3374SJan Kara 		 *
1153d57f3374SJan Kara 		 * If the reread partition isn't from release path, lo_refcnt
1154d57f3374SJan Kara 		 * must be at least one and it can only become zero when the
1155d57f3374SJan Kara 		 * current holder is released.
1156d57f3374SJan Kara 		 */
11570da03cabSJan Kara 		if (release)
1158d57f3374SJan Kara 			err = __blkdev_reread_part(bdev);
1159d57f3374SJan Kara 		else
1160d57f3374SJan Kara 			err = blkdev_reread_part(bdev);
116140853d6fSDongli Zhang 		if (err)
1162d57f3374SJan Kara 			pr_warn("%s: partition scan of loop%d failed (rc=%d)\n",
11630da03cabSJan Kara 				__func__, lo_number, err);
1164d57f3374SJan Kara 		/* Device is gone, no point in returning error */
1165d57f3374SJan Kara 		err = 0;
1166d57f3374SJan Kara 	}
1167758a58d0SDongli Zhang 
1168758a58d0SDongli Zhang 	/*
1169758a58d0SDongli Zhang 	 * lo->lo_state is set to Lo_unbound here after above partscan has
1170758a58d0SDongli Zhang 	 * finished.
1171758a58d0SDongli Zhang 	 *
1172758a58d0SDongli Zhang 	 * There cannot be anybody else entering __loop_clr_fd() as
1173758a58d0SDongli Zhang 	 * lo->lo_backing_file is already cleared and Lo_rundown state
1174758a58d0SDongli Zhang 	 * protects us from all the other places trying to change the 'lo'
1175758a58d0SDongli Zhang 	 * device.
1176758a58d0SDongli Zhang 	 */
1177758a58d0SDongli Zhang 	mutex_lock(&loop_ctl_mutex);
1178758a58d0SDongli Zhang 	lo->lo_flags = 0;
1179758a58d0SDongli Zhang 	if (!part_shift)
1180758a58d0SDongli Zhang 		lo->lo_disk->flags |= GENHD_FL_NO_PART_SCAN;
1181758a58d0SDongli Zhang 	lo->lo_state = Lo_unbound;
1182758a58d0SDongli Zhang 	mutex_unlock(&loop_ctl_mutex);
1183758a58d0SDongli Zhang 
1184f028f3b2SNikanth Karthikesan 	/*
1185310ca162STetsuo Handa 	 * Need not hold loop_ctl_mutex to fput backing file.
1186310ca162STetsuo Handa 	 * Calling fput holding loop_ctl_mutex triggers a circular
1187f028f3b2SNikanth Karthikesan 	 * lock dependency possibility warning as fput can take
1188310ca162STetsuo Handa 	 * bd_mutex which is usually taken before loop_ctl_mutex.
1189f028f3b2SNikanth Karthikesan 	 */
11907ccd0791SJan Kara 	if (filp)
1191f028f3b2SNikanth Karthikesan 		fput(filp);
11927ccd0791SJan Kara 	return err;
11931da177e4SLinus Torvalds }
11941da177e4SLinus Torvalds 
1195a2505b79SJan Kara static int loop_clr_fd(struct loop_device *lo)
1196a2505b79SJan Kara {
11977ccd0791SJan Kara 	int err;
11987ccd0791SJan Kara 
1199c28445faSJan Kara 	err = mutex_lock_killable(&loop_ctl_mutex);
12007ccd0791SJan Kara 	if (err)
12017ccd0791SJan Kara 		return err;
12027ccd0791SJan Kara 	if (lo->lo_state != Lo_bound) {
12037ccd0791SJan Kara 		mutex_unlock(&loop_ctl_mutex);
1204a2505b79SJan Kara 		return -ENXIO;
12057ccd0791SJan Kara 	}
1206a2505b79SJan Kara 	/*
1207a2505b79SJan Kara 	 * If we've explicitly asked to tear down the loop device,
1208a2505b79SJan Kara 	 * and it has an elevated reference count, set it for auto-teardown when
1209a2505b79SJan Kara 	 * the last reference goes away. This stops $!~#$@ udev from
1210a2505b79SJan Kara 	 * preventing teardown because it decided that it needs to run blkid on
1211a2505b79SJan Kara 	 * the loopback device whenever they appear. xfstests is notorious for
1212a2505b79SJan Kara 	 * failing tests because blkid via udev races with a losetup
1213a2505b79SJan Kara 	 * <dev>/do something like mkfs/losetup -d <dev> causing the losetup -d
1214a2505b79SJan Kara 	 * command to fail with EBUSY.
1215a2505b79SJan Kara 	 */
1216a2505b79SJan Kara 	if (atomic_read(&lo->lo_refcnt) > 1) {
1217a2505b79SJan Kara 		lo->lo_flags |= LO_FLAGS_AUTOCLEAR;
1218a2505b79SJan Kara 		mutex_unlock(&loop_ctl_mutex);
1219a2505b79SJan Kara 		return 0;
1220a2505b79SJan Kara 	}
1221a2505b79SJan Kara 	lo->lo_state = Lo_rundown;
12227ccd0791SJan Kara 	mutex_unlock(&loop_ctl_mutex);
1223a2505b79SJan Kara 
12240da03cabSJan Kara 	return __loop_clr_fd(lo, false);
1225a2505b79SJan Kara }
1226a2505b79SJan Kara 
12271da177e4SLinus Torvalds static int
12281da177e4SLinus Torvalds loop_set_status(struct loop_device *lo, const struct loop_info64 *info)
12291da177e4SLinus Torvalds {
12301da177e4SLinus Torvalds 	int err;
12311da177e4SLinus Torvalds 	struct loop_func_table *xfer;
1232e4849737SEric W. Biederman 	kuid_t uid = current_uid();
123385b0a54aSJan Kara 	struct block_device *bdev;
123485b0a54aSJan Kara 	bool partscan = false;
12351da177e4SLinus Torvalds 
1236c28445faSJan Kara 	err = mutex_lock_killable(&loop_ctl_mutex);
1237550df5fdSJan Kara 	if (err)
1238550df5fdSJan Kara 		return err;
1239b0fafa81SDavid Howells 	if (lo->lo_encrypt_key_size &&
1240e4849737SEric W. Biederman 	    !uid_eq(lo->lo_key_owner, uid) &&
1241550df5fdSJan Kara 	    !capable(CAP_SYS_ADMIN)) {
1242550df5fdSJan Kara 		err = -EPERM;
1243550df5fdSJan Kara 		goto out_unlock;
1244550df5fdSJan Kara 	}
1245550df5fdSJan Kara 	if (lo->lo_state != Lo_bound) {
1246550df5fdSJan Kara 		err = -ENXIO;
1247550df5fdSJan Kara 		goto out_unlock;
1248550df5fdSJan Kara 	}
1249550df5fdSJan Kara 	if ((unsigned int) info->lo_encrypt_key_size > LO_KEY_SIZE) {
1250550df5fdSJan Kara 		err = -EINVAL;
1251550df5fdSJan Kara 		goto out_unlock;
1252550df5fdSJan Kara 	}
12531da177e4SLinus Torvalds 
12545db470e2SJaegeuk Kim 	if (lo->lo_offset != info->lo_offset ||
12555db470e2SJaegeuk Kim 	    lo->lo_sizelimit != info->lo_sizelimit) {
12565db470e2SJaegeuk Kim 		sync_blockdev(lo->lo_device);
12575db470e2SJaegeuk Kim 		kill_bdev(lo->lo_device);
12585db470e2SJaegeuk Kim 	}
12595db470e2SJaegeuk Kim 
1260ecdd0959SMing Lei 	/* I/O need to be drained during transfer transition */
1261ecdd0959SMing Lei 	blk_mq_freeze_queue(lo->lo_queue);
1262ecdd0959SMing Lei 
12631da177e4SLinus Torvalds 	err = loop_release_xfer(lo);
12641da177e4SLinus Torvalds 	if (err)
1265550df5fdSJan Kara 		goto out_unfreeze;
12661da177e4SLinus Torvalds 
12671da177e4SLinus Torvalds 	if (info->lo_encrypt_type) {
12681da177e4SLinus Torvalds 		unsigned int type = info->lo_encrypt_type;
12691da177e4SLinus Torvalds 
12701e047eaaSTetsuo Handa 		if (type >= MAX_LO_CRYPT) {
12711e047eaaSTetsuo Handa 			err = -EINVAL;
1272550df5fdSJan Kara 			goto out_unfreeze;
12731e047eaaSTetsuo Handa 		}
12741da177e4SLinus Torvalds 		xfer = xfer_funcs[type];
12751e047eaaSTetsuo Handa 		if (xfer == NULL) {
12761e047eaaSTetsuo Handa 			err = -EINVAL;
1277550df5fdSJan Kara 			goto out_unfreeze;
12781e047eaaSTetsuo Handa 		}
12791da177e4SLinus Torvalds 	} else
12801da177e4SLinus Torvalds 		xfer = NULL;
12811da177e4SLinus Torvalds 
12821da177e4SLinus Torvalds 	err = loop_init_xfer(lo, xfer, info);
12831da177e4SLinus Torvalds 	if (err)
1284550df5fdSJan Kara 		goto out_unfreeze;
12851da177e4SLinus Torvalds 
12861da177e4SLinus Torvalds 	if (lo->lo_offset != info->lo_offset ||
12871e6ec9eaSOmar Sandoval 	    lo->lo_sizelimit != info->lo_sizelimit) {
12885db470e2SJaegeuk Kim 		/* kill_bdev should have truncated all the pages */
12895db470e2SJaegeuk Kim 		if (lo->lo_device->bd_inode->i_mapping->nrpages) {
12905db470e2SJaegeuk Kim 			err = -EAGAIN;
12915db470e2SJaegeuk Kim 			pr_warn("%s: loop%d (%s) has still dirty pages (nrpages=%lu)\n",
12925db470e2SJaegeuk Kim 				__func__, lo->lo_number, lo->lo_file_name,
12935db470e2SJaegeuk Kim 				lo->lo_device->bd_inode->i_mapping->nrpages);
12945db470e2SJaegeuk Kim 			goto out_unfreeze;
12955db470e2SJaegeuk Kim 		}
12961e6ec9eaSOmar Sandoval 		if (figure_loop_size(lo, info->lo_offset, info->lo_sizelimit)) {
1297ecdd0959SMing Lei 			err = -EFBIG;
1298550df5fdSJan Kara 			goto out_unfreeze;
1299ecdd0959SMing Lei 		}
1300b040ad9cSArnd Bergmann 	}
1301541c742aSGuo Chao 
1302dfaa2ef6SLukas Czerner 	loop_config_discard(lo);
13031da177e4SLinus Torvalds 
13041da177e4SLinus Torvalds 	memcpy(lo->lo_file_name, info->lo_file_name, LO_NAME_SIZE);
13051da177e4SLinus Torvalds 	memcpy(lo->lo_crypt_name, info->lo_crypt_name, LO_NAME_SIZE);
13061da177e4SLinus Torvalds 	lo->lo_file_name[LO_NAME_SIZE-1] = 0;
13071da177e4SLinus Torvalds 	lo->lo_crypt_name[LO_NAME_SIZE-1] = 0;
13081da177e4SLinus Torvalds 
13091da177e4SLinus Torvalds 	if (!xfer)
13101da177e4SLinus Torvalds 		xfer = &none_funcs;
13111da177e4SLinus Torvalds 	lo->transfer = xfer->transfer;
13121da177e4SLinus Torvalds 	lo->ioctl = xfer->ioctl;
13131da177e4SLinus Torvalds 
131496c58655SDavid Woodhouse 	if ((lo->lo_flags & LO_FLAGS_AUTOCLEAR) !=
131596c58655SDavid Woodhouse 	     (info->lo_flags & LO_FLAGS_AUTOCLEAR))
131696c58655SDavid Woodhouse 		lo->lo_flags ^= LO_FLAGS_AUTOCLEAR;
131796c58655SDavid Woodhouse 
13181da177e4SLinus Torvalds 	lo->lo_encrypt_key_size = info->lo_encrypt_key_size;
13191da177e4SLinus Torvalds 	lo->lo_init[0] = info->lo_init[0];
13201da177e4SLinus Torvalds 	lo->lo_init[1] = info->lo_init[1];
13211da177e4SLinus Torvalds 	if (info->lo_encrypt_key_size) {
13221da177e4SLinus Torvalds 		memcpy(lo->lo_encrypt_key, info->lo_encrypt_key,
13231da177e4SLinus Torvalds 		       info->lo_encrypt_key_size);
1324b0fafa81SDavid Howells 		lo->lo_key_owner = uid;
13251da177e4SLinus Torvalds 	}
13261da177e4SLinus Torvalds 
13272e5ab5f3SMing Lei 	/* update dio if lo_offset or transfer is changed */
13282e5ab5f3SMing Lei 	__loop_update_dio(lo, lo->use_dio);
13292e5ab5f3SMing Lei 
1330550df5fdSJan Kara out_unfreeze:
1331ecdd0959SMing Lei 	blk_mq_unfreeze_queue(lo->lo_queue);
1332e02898b4SOmar Sandoval 
1333e02898b4SOmar Sandoval 	if (!err && (info->lo_flags & LO_FLAGS_PARTSCAN) &&
1334e02898b4SOmar Sandoval 	     !(lo->lo_flags & LO_FLAGS_PARTSCAN)) {
1335e02898b4SOmar Sandoval 		lo->lo_flags |= LO_FLAGS_PARTSCAN;
1336e02898b4SOmar Sandoval 		lo->lo_disk->flags &= ~GENHD_FL_NO_PART_SCAN;
133785b0a54aSJan Kara 		bdev = lo->lo_device;
133885b0a54aSJan Kara 		partscan = true;
1339e02898b4SOmar Sandoval 	}
1340550df5fdSJan Kara out_unlock:
1341550df5fdSJan Kara 	mutex_unlock(&loop_ctl_mutex);
134285b0a54aSJan Kara 	if (partscan)
134385b0a54aSJan Kara 		loop_reread_partitions(lo, bdev);
1344e02898b4SOmar Sandoval 
1345ecdd0959SMing Lei 	return err;
13461da177e4SLinus Torvalds }
13471da177e4SLinus Torvalds 
13481da177e4SLinus Torvalds static int
13491da177e4SLinus Torvalds loop_get_status(struct loop_device *lo, struct loop_info64 *info)
13501da177e4SLinus Torvalds {
1351b1ab5fa3STetsuo Handa 	struct path path;
13521da177e4SLinus Torvalds 	struct kstat stat;
13532d1d4c1eSOmar Sandoval 	int ret;
13541da177e4SLinus Torvalds 
1355c28445faSJan Kara 	ret = mutex_lock_killable(&loop_ctl_mutex);
13564a5ce9baSJan Kara 	if (ret)
13574a5ce9baSJan Kara 		return ret;
13582d1d4c1eSOmar Sandoval 	if (lo->lo_state != Lo_bound) {
1359310ca162STetsuo Handa 		mutex_unlock(&loop_ctl_mutex);
13601da177e4SLinus Torvalds 		return -ENXIO;
13612d1d4c1eSOmar Sandoval 	}
13622d1d4c1eSOmar Sandoval 
13631da177e4SLinus Torvalds 	memset(info, 0, sizeof(*info));
13641da177e4SLinus Torvalds 	info->lo_number = lo->lo_number;
13651da177e4SLinus Torvalds 	info->lo_offset = lo->lo_offset;
13661da177e4SLinus Torvalds 	info->lo_sizelimit = lo->lo_sizelimit;
13671da177e4SLinus Torvalds 	info->lo_flags = lo->lo_flags;
13681da177e4SLinus Torvalds 	memcpy(info->lo_file_name, lo->lo_file_name, LO_NAME_SIZE);
13691da177e4SLinus Torvalds 	memcpy(info->lo_crypt_name, lo->lo_crypt_name, LO_NAME_SIZE);
13701da177e4SLinus Torvalds 	info->lo_encrypt_type =
13711da177e4SLinus Torvalds 		lo->lo_encryption ? lo->lo_encryption->number : 0;
13721da177e4SLinus Torvalds 	if (lo->lo_encrypt_key_size && capable(CAP_SYS_ADMIN)) {
13731da177e4SLinus Torvalds 		info->lo_encrypt_key_size = lo->lo_encrypt_key_size;
13741da177e4SLinus Torvalds 		memcpy(info->lo_encrypt_key, lo->lo_encrypt_key,
13751da177e4SLinus Torvalds 		       lo->lo_encrypt_key_size);
13761da177e4SLinus Torvalds 	}
13772d1d4c1eSOmar Sandoval 
1378310ca162STetsuo Handa 	/* Drop loop_ctl_mutex while we call into the filesystem. */
1379b1ab5fa3STetsuo Handa 	path = lo->lo_backing_file->f_path;
1380b1ab5fa3STetsuo Handa 	path_get(&path);
1381310ca162STetsuo Handa 	mutex_unlock(&loop_ctl_mutex);
1382b1ab5fa3STetsuo Handa 	ret = vfs_getattr(&path, &stat, STATX_INO, AT_STATX_SYNC_AS_STAT);
13832d1d4c1eSOmar Sandoval 	if (!ret) {
13842d1d4c1eSOmar Sandoval 		info->lo_device = huge_encode_dev(stat.dev);
13852d1d4c1eSOmar Sandoval 		info->lo_inode = stat.ino;
13862d1d4c1eSOmar Sandoval 		info->lo_rdevice = huge_encode_dev(stat.rdev);
13872d1d4c1eSOmar Sandoval 	}
1388b1ab5fa3STetsuo Handa 	path_put(&path);
13892d1d4c1eSOmar Sandoval 	return ret;
13901da177e4SLinus Torvalds }
13911da177e4SLinus Torvalds 
13921da177e4SLinus Torvalds static void
13931da177e4SLinus Torvalds loop_info64_from_old(const struct loop_info *info, struct loop_info64 *info64)
13941da177e4SLinus Torvalds {
13951da177e4SLinus Torvalds 	memset(info64, 0, sizeof(*info64));
13961da177e4SLinus Torvalds 	info64->lo_number = info->lo_number;
13971da177e4SLinus Torvalds 	info64->lo_device = info->lo_device;
13981da177e4SLinus Torvalds 	info64->lo_inode = info->lo_inode;
13991da177e4SLinus Torvalds 	info64->lo_rdevice = info->lo_rdevice;
14001da177e4SLinus Torvalds 	info64->lo_offset = info->lo_offset;
14011da177e4SLinus Torvalds 	info64->lo_sizelimit = 0;
14021da177e4SLinus Torvalds 	info64->lo_encrypt_type = info->lo_encrypt_type;
14031da177e4SLinus Torvalds 	info64->lo_encrypt_key_size = info->lo_encrypt_key_size;
14041da177e4SLinus Torvalds 	info64->lo_flags = info->lo_flags;
14051da177e4SLinus Torvalds 	info64->lo_init[0] = info->lo_init[0];
14061da177e4SLinus Torvalds 	info64->lo_init[1] = info->lo_init[1];
14071da177e4SLinus Torvalds 	if (info->lo_encrypt_type == LO_CRYPT_CRYPTOAPI)
14081da177e4SLinus Torvalds 		memcpy(info64->lo_crypt_name, info->lo_name, LO_NAME_SIZE);
14091da177e4SLinus Torvalds 	else
14101da177e4SLinus Torvalds 		memcpy(info64->lo_file_name, info->lo_name, LO_NAME_SIZE);
14111da177e4SLinus Torvalds 	memcpy(info64->lo_encrypt_key, info->lo_encrypt_key, LO_KEY_SIZE);
14121da177e4SLinus Torvalds }
14131da177e4SLinus Torvalds 
14141da177e4SLinus Torvalds static int
14151da177e4SLinus Torvalds loop_info64_to_old(const struct loop_info64 *info64, struct loop_info *info)
14161da177e4SLinus Torvalds {
14171da177e4SLinus Torvalds 	memset(info, 0, sizeof(*info));
14181da177e4SLinus Torvalds 	info->lo_number = info64->lo_number;
14191da177e4SLinus Torvalds 	info->lo_device = info64->lo_device;
14201da177e4SLinus Torvalds 	info->lo_inode = info64->lo_inode;
14211da177e4SLinus Torvalds 	info->lo_rdevice = info64->lo_rdevice;
14221da177e4SLinus Torvalds 	info->lo_offset = info64->lo_offset;
14231da177e4SLinus Torvalds 	info->lo_encrypt_type = info64->lo_encrypt_type;
14241da177e4SLinus Torvalds 	info->lo_encrypt_key_size = info64->lo_encrypt_key_size;
14251da177e4SLinus Torvalds 	info->lo_flags = info64->lo_flags;
14261da177e4SLinus Torvalds 	info->lo_init[0] = info64->lo_init[0];
14271da177e4SLinus Torvalds 	info->lo_init[1] = info64->lo_init[1];
14281da177e4SLinus Torvalds 	if (info->lo_encrypt_type == LO_CRYPT_CRYPTOAPI)
14291da177e4SLinus Torvalds 		memcpy(info->lo_name, info64->lo_crypt_name, LO_NAME_SIZE);
14301da177e4SLinus Torvalds 	else
14311da177e4SLinus Torvalds 		memcpy(info->lo_name, info64->lo_file_name, LO_NAME_SIZE);
14321da177e4SLinus Torvalds 	memcpy(info->lo_encrypt_key, info64->lo_encrypt_key, LO_KEY_SIZE);
14331da177e4SLinus Torvalds 
14341da177e4SLinus Torvalds 	/* error in case values were truncated */
14351da177e4SLinus Torvalds 	if (info->lo_device != info64->lo_device ||
14361da177e4SLinus Torvalds 	    info->lo_rdevice != info64->lo_rdevice ||
14371da177e4SLinus Torvalds 	    info->lo_inode != info64->lo_inode ||
14381da177e4SLinus Torvalds 	    info->lo_offset != info64->lo_offset)
14391da177e4SLinus Torvalds 		return -EOVERFLOW;
14401da177e4SLinus Torvalds 
14411da177e4SLinus Torvalds 	return 0;
14421da177e4SLinus Torvalds }
14431da177e4SLinus Torvalds 
14441da177e4SLinus Torvalds static int
14451da177e4SLinus Torvalds loop_set_status_old(struct loop_device *lo, const struct loop_info __user *arg)
14461da177e4SLinus Torvalds {
14471da177e4SLinus Torvalds 	struct loop_info info;
14481da177e4SLinus Torvalds 	struct loop_info64 info64;
14491da177e4SLinus Torvalds 
14501da177e4SLinus Torvalds 	if (copy_from_user(&info, arg, sizeof (struct loop_info)))
14511da177e4SLinus Torvalds 		return -EFAULT;
14521da177e4SLinus Torvalds 	loop_info64_from_old(&info, &info64);
14531da177e4SLinus Torvalds 	return loop_set_status(lo, &info64);
14541da177e4SLinus Torvalds }
14551da177e4SLinus Torvalds 
14561da177e4SLinus Torvalds static int
14571da177e4SLinus Torvalds loop_set_status64(struct loop_device *lo, const struct loop_info64 __user *arg)
14581da177e4SLinus Torvalds {
14591da177e4SLinus Torvalds 	struct loop_info64 info64;
14601da177e4SLinus Torvalds 
14611da177e4SLinus Torvalds 	if (copy_from_user(&info64, arg, sizeof (struct loop_info64)))
14621da177e4SLinus Torvalds 		return -EFAULT;
14631da177e4SLinus Torvalds 	return loop_set_status(lo, &info64);
14641da177e4SLinus Torvalds }
14651da177e4SLinus Torvalds 
14661da177e4SLinus Torvalds static int
14671da177e4SLinus Torvalds loop_get_status_old(struct loop_device *lo, struct loop_info __user *arg) {
14681da177e4SLinus Torvalds 	struct loop_info info;
14691da177e4SLinus Torvalds 	struct loop_info64 info64;
1470bdac616dSOmar Sandoval 	int err;
14711da177e4SLinus Torvalds 
14724a5ce9baSJan Kara 	if (!arg)
1473bdac616dSOmar Sandoval 		return -EINVAL;
14741da177e4SLinus Torvalds 	err = loop_get_status(lo, &info64);
14751da177e4SLinus Torvalds 	if (!err)
14761da177e4SLinus Torvalds 		err = loop_info64_to_old(&info64, &info);
14771da177e4SLinus Torvalds 	if (!err && copy_to_user(arg, &info, sizeof(info)))
14781da177e4SLinus Torvalds 		err = -EFAULT;
14791da177e4SLinus Torvalds 
14801da177e4SLinus Torvalds 	return err;
14811da177e4SLinus Torvalds }
14821da177e4SLinus Torvalds 
14831da177e4SLinus Torvalds static int
14841da177e4SLinus Torvalds loop_get_status64(struct loop_device *lo, struct loop_info64 __user *arg) {
14851da177e4SLinus Torvalds 	struct loop_info64 info64;
1486bdac616dSOmar Sandoval 	int err;
14871da177e4SLinus Torvalds 
14884a5ce9baSJan Kara 	if (!arg)
1489bdac616dSOmar Sandoval 		return -EINVAL;
14901da177e4SLinus Torvalds 	err = loop_get_status(lo, &info64);
14911da177e4SLinus Torvalds 	if (!err && copy_to_user(arg, &info64, sizeof(info64)))
14921da177e4SLinus Torvalds 		err = -EFAULT;
14931da177e4SLinus Torvalds 
14941da177e4SLinus Torvalds 	return err;
14951da177e4SLinus Torvalds }
14961da177e4SLinus Torvalds 
149751001b7dSHannes Reinecke static int loop_set_capacity(struct loop_device *lo)
149853d66608SJ. R. Okajima {
149953d66608SJ. R. Okajima 	if (unlikely(lo->lo_state != Lo_bound))
15007b0576a3SGuo Chao 		return -ENXIO;
150153d66608SJ. R. Okajima 
15021e6ec9eaSOmar Sandoval 	return figure_loop_size(lo, lo->lo_offset, lo->lo_sizelimit);
150353d66608SJ. R. Okajima }
150453d66608SJ. R. Okajima 
1505ab1cb278SMing Lei static int loop_set_dio(struct loop_device *lo, unsigned long arg)
1506ab1cb278SMing Lei {
1507ab1cb278SMing Lei 	int error = -ENXIO;
1508ab1cb278SMing Lei 	if (lo->lo_state != Lo_bound)
1509ab1cb278SMing Lei 		goto out;
1510ab1cb278SMing Lei 
1511ab1cb278SMing Lei 	__loop_update_dio(lo, !!arg);
1512ab1cb278SMing Lei 	if (lo->use_dio == !!arg)
1513ab1cb278SMing Lei 		return 0;
1514ab1cb278SMing Lei 	error = -EINVAL;
1515ab1cb278SMing Lei  out:
1516ab1cb278SMing Lei 	return error;
1517ab1cb278SMing Lei }
1518ab1cb278SMing Lei 
151989e4fdecSOmar Sandoval static int loop_set_block_size(struct loop_device *lo, unsigned long arg)
152089e4fdecSOmar Sandoval {
15215db470e2SJaegeuk Kim 	int err = 0;
15225db470e2SJaegeuk Kim 
152389e4fdecSOmar Sandoval 	if (lo->lo_state != Lo_bound)
152489e4fdecSOmar Sandoval 		return -ENXIO;
152589e4fdecSOmar Sandoval 
152689e4fdecSOmar Sandoval 	if (arg < 512 || arg > PAGE_SIZE || !is_power_of_2(arg))
152789e4fdecSOmar Sandoval 		return -EINVAL;
152889e4fdecSOmar Sandoval 
15295db470e2SJaegeuk Kim 	if (lo->lo_queue->limits.logical_block_size != arg) {
15305db470e2SJaegeuk Kim 		sync_blockdev(lo->lo_device);
15315db470e2SJaegeuk Kim 		kill_bdev(lo->lo_device);
15325db470e2SJaegeuk Kim 	}
15335db470e2SJaegeuk Kim 
153489e4fdecSOmar Sandoval 	blk_mq_freeze_queue(lo->lo_queue);
153589e4fdecSOmar Sandoval 
15365db470e2SJaegeuk Kim 	/* kill_bdev should have truncated all the pages */
15375db470e2SJaegeuk Kim 	if (lo->lo_queue->limits.logical_block_size != arg &&
15385db470e2SJaegeuk Kim 			lo->lo_device->bd_inode->i_mapping->nrpages) {
15395db470e2SJaegeuk Kim 		err = -EAGAIN;
15405db470e2SJaegeuk Kim 		pr_warn("%s: loop%d (%s) has still dirty pages (nrpages=%lu)\n",
15415db470e2SJaegeuk Kim 			__func__, lo->lo_number, lo->lo_file_name,
15425db470e2SJaegeuk Kim 			lo->lo_device->bd_inode->i_mapping->nrpages);
15435db470e2SJaegeuk Kim 		goto out_unfreeze;
15445db470e2SJaegeuk Kim 	}
15455db470e2SJaegeuk Kim 
154689e4fdecSOmar Sandoval 	blk_queue_logical_block_size(lo->lo_queue, arg);
1547bf093753SOmar Sandoval 	blk_queue_physical_block_size(lo->lo_queue, arg);
1548bf093753SOmar Sandoval 	blk_queue_io_min(lo->lo_queue, arg);
154989e4fdecSOmar Sandoval 	loop_update_dio(lo);
15505db470e2SJaegeuk Kim out_unfreeze:
155189e4fdecSOmar Sandoval 	blk_mq_unfreeze_queue(lo->lo_queue);
155289e4fdecSOmar Sandoval 
15535db470e2SJaegeuk Kim 	return err;
155489e4fdecSOmar Sandoval }
155589e4fdecSOmar Sandoval 
1556a1316544SJan Kara static int lo_simple_ioctl(struct loop_device *lo, unsigned int cmd,
1557a1316544SJan Kara 			   unsigned long arg)
15581da177e4SLinus Torvalds {
15591da177e4SLinus Torvalds 	int err;
15601da177e4SLinus Torvalds 
1561c28445faSJan Kara 	err = mutex_lock_killable(&loop_ctl_mutex);
15623148ffbdSOmar Sandoval 	if (err)
1563a1316544SJan Kara 		return err;
15641da177e4SLinus Torvalds 	switch (cmd) {
156553d66608SJ. R. Okajima 	case LOOP_SET_CAPACITY:
156651001b7dSHannes Reinecke 		err = loop_set_capacity(lo);
156753d66608SJ. R. Okajima 		break;
1568ab1cb278SMing Lei 	case LOOP_SET_DIRECT_IO:
1569ab1cb278SMing Lei 		err = loop_set_dio(lo, arg);
1570ab1cb278SMing Lei 		break;
157189e4fdecSOmar Sandoval 	case LOOP_SET_BLOCK_SIZE:
157289e4fdecSOmar Sandoval 		err = loop_set_block_size(lo, arg);
157389e4fdecSOmar Sandoval 		break;
15741da177e4SLinus Torvalds 	default:
15751da177e4SLinus Torvalds 		err = lo->ioctl ? lo->ioctl(lo, cmd, arg) : -EINVAL;
15761da177e4SLinus Torvalds 	}
1577310ca162STetsuo Handa 	mutex_unlock(&loop_ctl_mutex);
1578a1316544SJan Kara 	return err;
1579a1316544SJan Kara }
1580f028f3b2SNikanth Karthikesan 
1581a1316544SJan Kara static int lo_ioctl(struct block_device *bdev, fmode_t mode,
1582a1316544SJan Kara 	unsigned int cmd, unsigned long arg)
1583a1316544SJan Kara {
1584a1316544SJan Kara 	struct loop_device *lo = bdev->bd_disk->private_data;
1585a1316544SJan Kara 	int err;
1586a1316544SJan Kara 
1587a1316544SJan Kara 	switch (cmd) {
1588a1316544SJan Kara 	case LOOP_SET_FD:
1589757ecf40SJan Kara 		return loop_set_fd(lo, mode, bdev, arg);
1590a1316544SJan Kara 	case LOOP_CHANGE_FD:
1591c3710770SJan Kara 		return loop_change_fd(lo, bdev, arg);
1592a1316544SJan Kara 	case LOOP_CLR_FD:
15937ccd0791SJan Kara 		return loop_clr_fd(lo);
1594a1316544SJan Kara 	case LOOP_SET_STATUS:
1595a1316544SJan Kara 		err = -EPERM;
1596a1316544SJan Kara 		if ((mode & FMODE_WRITE) || capable(CAP_SYS_ADMIN)) {
1597a1316544SJan Kara 			err = loop_set_status_old(lo,
1598a1316544SJan Kara 					(struct loop_info __user *)arg);
1599a1316544SJan Kara 		}
1600a1316544SJan Kara 		break;
1601a1316544SJan Kara 	case LOOP_GET_STATUS:
16024a5ce9baSJan Kara 		return loop_get_status_old(lo, (struct loop_info __user *) arg);
1603a1316544SJan Kara 	case LOOP_SET_STATUS64:
1604a1316544SJan Kara 		err = -EPERM;
1605a1316544SJan Kara 		if ((mode & FMODE_WRITE) || capable(CAP_SYS_ADMIN)) {
1606a1316544SJan Kara 			err = loop_set_status64(lo,
1607a1316544SJan Kara 					(struct loop_info64 __user *) arg);
1608a1316544SJan Kara 		}
1609a1316544SJan Kara 		break;
1610a1316544SJan Kara 	case LOOP_GET_STATUS64:
16114a5ce9baSJan Kara 		return loop_get_status64(lo, (struct loop_info64 __user *) arg);
1612a1316544SJan Kara 	case LOOP_SET_CAPACITY:
1613a1316544SJan Kara 	case LOOP_SET_DIRECT_IO:
1614a1316544SJan Kara 	case LOOP_SET_BLOCK_SIZE:
1615a1316544SJan Kara 		if (!(mode & FMODE_WRITE) && !capable(CAP_SYS_ADMIN))
1616a1316544SJan Kara 			return -EPERM;
1617a1316544SJan Kara 		/* Fall through */
1618a1316544SJan Kara 	default:
1619a1316544SJan Kara 		err = lo_simple_ioctl(lo, cmd, arg);
1620a1316544SJan Kara 		break;
1621a1316544SJan Kara 	}
1622a1316544SJan Kara 
16231da177e4SLinus Torvalds 	return err;
16241da177e4SLinus Torvalds }
16251da177e4SLinus Torvalds 
1626863d5b82SDavid Howells #ifdef CONFIG_COMPAT
1627863d5b82SDavid Howells struct compat_loop_info {
1628863d5b82SDavid Howells 	compat_int_t	lo_number;      /* ioctl r/o */
1629863d5b82SDavid Howells 	compat_dev_t	lo_device;      /* ioctl r/o */
1630863d5b82SDavid Howells 	compat_ulong_t	lo_inode;       /* ioctl r/o */
1631863d5b82SDavid Howells 	compat_dev_t	lo_rdevice;     /* ioctl r/o */
1632863d5b82SDavid Howells 	compat_int_t	lo_offset;
1633863d5b82SDavid Howells 	compat_int_t	lo_encrypt_type;
1634863d5b82SDavid Howells 	compat_int_t	lo_encrypt_key_size;    /* ioctl w/o */
1635863d5b82SDavid Howells 	compat_int_t	lo_flags;       /* ioctl r/o */
1636863d5b82SDavid Howells 	char		lo_name[LO_NAME_SIZE];
1637863d5b82SDavid Howells 	unsigned char	lo_encrypt_key[LO_KEY_SIZE]; /* ioctl w/o */
1638863d5b82SDavid Howells 	compat_ulong_t	lo_init[2];
1639863d5b82SDavid Howells 	char		reserved[4];
1640863d5b82SDavid Howells };
1641863d5b82SDavid Howells 
1642863d5b82SDavid Howells /*
1643863d5b82SDavid Howells  * Transfer 32-bit compatibility structure in userspace to 64-bit loop info
1644863d5b82SDavid Howells  * - noinlined to reduce stack space usage in main part of driver
1645863d5b82SDavid Howells  */
1646863d5b82SDavid Howells static noinline int
1647ba674cfcSAl Viro loop_info64_from_compat(const struct compat_loop_info __user *arg,
1648863d5b82SDavid Howells 			struct loop_info64 *info64)
1649863d5b82SDavid Howells {
1650863d5b82SDavid Howells 	struct compat_loop_info info;
1651863d5b82SDavid Howells 
1652863d5b82SDavid Howells 	if (copy_from_user(&info, arg, sizeof(info)))
1653863d5b82SDavid Howells 		return -EFAULT;
1654863d5b82SDavid Howells 
1655863d5b82SDavid Howells 	memset(info64, 0, sizeof(*info64));
1656863d5b82SDavid Howells 	info64->lo_number = info.lo_number;
1657863d5b82SDavid Howells 	info64->lo_device = info.lo_device;
1658863d5b82SDavid Howells 	info64->lo_inode = info.lo_inode;
1659863d5b82SDavid Howells 	info64->lo_rdevice = info.lo_rdevice;
1660863d5b82SDavid Howells 	info64->lo_offset = info.lo_offset;
1661863d5b82SDavid Howells 	info64->lo_sizelimit = 0;
1662863d5b82SDavid Howells 	info64->lo_encrypt_type = info.lo_encrypt_type;
1663863d5b82SDavid Howells 	info64->lo_encrypt_key_size = info.lo_encrypt_key_size;
1664863d5b82SDavid Howells 	info64->lo_flags = info.lo_flags;
1665863d5b82SDavid Howells 	info64->lo_init[0] = info.lo_init[0];
1666863d5b82SDavid Howells 	info64->lo_init[1] = info.lo_init[1];
1667863d5b82SDavid Howells 	if (info.lo_encrypt_type == LO_CRYPT_CRYPTOAPI)
1668863d5b82SDavid Howells 		memcpy(info64->lo_crypt_name, info.lo_name, LO_NAME_SIZE);
1669863d5b82SDavid Howells 	else
1670863d5b82SDavid Howells 		memcpy(info64->lo_file_name, info.lo_name, LO_NAME_SIZE);
1671863d5b82SDavid Howells 	memcpy(info64->lo_encrypt_key, info.lo_encrypt_key, LO_KEY_SIZE);
1672863d5b82SDavid Howells 	return 0;
1673863d5b82SDavid Howells }
1674863d5b82SDavid Howells 
1675863d5b82SDavid Howells /*
1676863d5b82SDavid Howells  * Transfer 64-bit loop info to 32-bit compatibility structure in userspace
1677863d5b82SDavid Howells  * - noinlined to reduce stack space usage in main part of driver
1678863d5b82SDavid Howells  */
1679863d5b82SDavid Howells static noinline int
1680863d5b82SDavid Howells loop_info64_to_compat(const struct loop_info64 *info64,
1681863d5b82SDavid Howells 		      struct compat_loop_info __user *arg)
1682863d5b82SDavid Howells {
1683863d5b82SDavid Howells 	struct compat_loop_info info;
1684863d5b82SDavid Howells 
1685863d5b82SDavid Howells 	memset(&info, 0, sizeof(info));
1686863d5b82SDavid Howells 	info.lo_number = info64->lo_number;
1687863d5b82SDavid Howells 	info.lo_device = info64->lo_device;
1688863d5b82SDavid Howells 	info.lo_inode = info64->lo_inode;
1689863d5b82SDavid Howells 	info.lo_rdevice = info64->lo_rdevice;
1690863d5b82SDavid Howells 	info.lo_offset = info64->lo_offset;
1691863d5b82SDavid Howells 	info.lo_encrypt_type = info64->lo_encrypt_type;
1692863d5b82SDavid Howells 	info.lo_encrypt_key_size = info64->lo_encrypt_key_size;
1693863d5b82SDavid Howells 	info.lo_flags = info64->lo_flags;
1694863d5b82SDavid Howells 	info.lo_init[0] = info64->lo_init[0];
1695863d5b82SDavid Howells 	info.lo_init[1] = info64->lo_init[1];
1696863d5b82SDavid Howells 	if (info.lo_encrypt_type == LO_CRYPT_CRYPTOAPI)
1697863d5b82SDavid Howells 		memcpy(info.lo_name, info64->lo_crypt_name, LO_NAME_SIZE);
1698863d5b82SDavid Howells 	else
1699863d5b82SDavid Howells 		memcpy(info.lo_name, info64->lo_file_name, LO_NAME_SIZE);
1700863d5b82SDavid Howells 	memcpy(info.lo_encrypt_key, info64->lo_encrypt_key, LO_KEY_SIZE);
1701863d5b82SDavid Howells 
1702863d5b82SDavid Howells 	/* error in case values were truncated */
1703863d5b82SDavid Howells 	if (info.lo_device != info64->lo_device ||
1704863d5b82SDavid Howells 	    info.lo_rdevice != info64->lo_rdevice ||
1705863d5b82SDavid Howells 	    info.lo_inode != info64->lo_inode ||
1706863d5b82SDavid Howells 	    info.lo_offset != info64->lo_offset ||
1707863d5b82SDavid Howells 	    info.lo_init[0] != info64->lo_init[0] ||
1708863d5b82SDavid Howells 	    info.lo_init[1] != info64->lo_init[1])
1709863d5b82SDavid Howells 		return -EOVERFLOW;
1710863d5b82SDavid Howells 
1711863d5b82SDavid Howells 	if (copy_to_user(arg, &info, sizeof(info)))
1712863d5b82SDavid Howells 		return -EFAULT;
1713863d5b82SDavid Howells 	return 0;
1714863d5b82SDavid Howells }
1715863d5b82SDavid Howells 
1716863d5b82SDavid Howells static int
1717863d5b82SDavid Howells loop_set_status_compat(struct loop_device *lo,
1718863d5b82SDavid Howells 		       const struct compat_loop_info __user *arg)
1719863d5b82SDavid Howells {
1720863d5b82SDavid Howells 	struct loop_info64 info64;
1721863d5b82SDavid Howells 	int ret;
1722863d5b82SDavid Howells 
1723863d5b82SDavid Howells 	ret = loop_info64_from_compat(arg, &info64);
1724863d5b82SDavid Howells 	if (ret < 0)
1725863d5b82SDavid Howells 		return ret;
1726863d5b82SDavid Howells 	return loop_set_status(lo, &info64);
1727863d5b82SDavid Howells }
1728863d5b82SDavid Howells 
1729863d5b82SDavid Howells static int
1730863d5b82SDavid Howells loop_get_status_compat(struct loop_device *lo,
1731863d5b82SDavid Howells 		       struct compat_loop_info __user *arg)
1732863d5b82SDavid Howells {
1733863d5b82SDavid Howells 	struct loop_info64 info64;
1734bdac616dSOmar Sandoval 	int err;
1735863d5b82SDavid Howells 
17364a5ce9baSJan Kara 	if (!arg)
1737bdac616dSOmar Sandoval 		return -EINVAL;
1738863d5b82SDavid Howells 	err = loop_get_status(lo, &info64);
1739863d5b82SDavid Howells 	if (!err)
1740863d5b82SDavid Howells 		err = loop_info64_to_compat(&info64, arg);
1741863d5b82SDavid Howells 	return err;
1742863d5b82SDavid Howells }
1743863d5b82SDavid Howells 
1744bb214884SAl Viro static int lo_compat_ioctl(struct block_device *bdev, fmode_t mode,
1745bb214884SAl Viro 			   unsigned int cmd, unsigned long arg)
1746863d5b82SDavid Howells {
1747bb214884SAl Viro 	struct loop_device *lo = bdev->bd_disk->private_data;
1748863d5b82SDavid Howells 	int err;
1749863d5b82SDavid Howells 
1750863d5b82SDavid Howells 	switch(cmd) {
1751863d5b82SDavid Howells 	case LOOP_SET_STATUS:
17523148ffbdSOmar Sandoval 		err = loop_set_status_compat(lo,
17533148ffbdSOmar Sandoval 			     (const struct compat_loop_info __user *)arg);
1754863d5b82SDavid Howells 		break;
1755863d5b82SDavid Howells 	case LOOP_GET_STATUS:
17563148ffbdSOmar Sandoval 		err = loop_get_status_compat(lo,
17573148ffbdSOmar Sandoval 				     (struct compat_loop_info __user *)arg);
1758863d5b82SDavid Howells 		break;
175953d66608SJ. R. Okajima 	case LOOP_SET_CAPACITY:
1760863d5b82SDavid Howells 	case LOOP_CLR_FD:
1761863d5b82SDavid Howells 	case LOOP_GET_STATUS64:
1762863d5b82SDavid Howells 	case LOOP_SET_STATUS64:
1763863d5b82SDavid Howells 		arg = (unsigned long) compat_ptr(arg);
1764d893ff86SGustavo A. R. Silva 		/* fall through */
1765863d5b82SDavid Howells 	case LOOP_SET_FD:
1766863d5b82SDavid Howells 	case LOOP_CHANGE_FD:
17679fea4b39SEvan Green 	case LOOP_SET_BLOCK_SIZE:
1768fdbe4eeeSAlessio Balsini 	case LOOP_SET_DIRECT_IO:
1769bb214884SAl Viro 		err = lo_ioctl(bdev, mode, cmd, arg);
1770863d5b82SDavid Howells 		break;
1771863d5b82SDavid Howells 	default:
1772863d5b82SDavid Howells 		err = -ENOIOCTLCMD;
1773863d5b82SDavid Howells 		break;
1774863d5b82SDavid Howells 	}
1775863d5b82SDavid Howells 	return err;
1776863d5b82SDavid Howells }
1777863d5b82SDavid Howells #endif
1778863d5b82SDavid Howells 
1779bb214884SAl Viro static int lo_open(struct block_device *bdev, fmode_t mode)
17801da177e4SLinus Torvalds {
1781770fe30aSKay Sievers 	struct loop_device *lo;
17820a42e99bSJan Kara 	int err;
1783770fe30aSKay Sievers 
17840a42e99bSJan Kara 	err = mutex_lock_killable(&loop_ctl_mutex);
17850a42e99bSJan Kara 	if (err)
17860a42e99bSJan Kara 		return err;
1787770fe30aSKay Sievers 	lo = bdev->bd_disk->private_data;
1788770fe30aSKay Sievers 	if (!lo) {
1789770fe30aSKay Sievers 		err = -ENXIO;
1790770fe30aSKay Sievers 		goto out;
1791770fe30aSKay Sievers 	}
17921da177e4SLinus Torvalds 
1793f8933667SMing Lei 	atomic_inc(&lo->lo_refcnt);
1794770fe30aSKay Sievers out:
17950a42e99bSJan Kara 	mutex_unlock(&loop_ctl_mutex);
1796770fe30aSKay Sievers 	return err;
17971da177e4SLinus Torvalds }
17981da177e4SLinus Torvalds 
1799967d1dc1SJan Kara static void lo_release(struct gendisk *disk, fmode_t mode)
18001da177e4SLinus Torvalds {
1801967d1dc1SJan Kara 	struct loop_device *lo;
18021da177e4SLinus Torvalds 
18030a42e99bSJan Kara 	mutex_lock(&loop_ctl_mutex);
1804967d1dc1SJan Kara 	lo = disk->private_data;
1805f8933667SMing Lei 	if (atomic_dec_return(&lo->lo_refcnt))
18060a42e99bSJan Kara 		goto out_unlock;
1807f8933667SMing Lei 
180814f27939SMilan Broz 	if (lo->lo_flags & LO_FLAGS_AUTOCLEAR) {
1809a2505b79SJan Kara 		if (lo->lo_state != Lo_bound)
1810a2505b79SJan Kara 			goto out_unlock;
1811a2505b79SJan Kara 		lo->lo_state = Lo_rundown;
18127ccd0791SJan Kara 		mutex_unlock(&loop_ctl_mutex);
181314f27939SMilan Broz 		/*
181414f27939SMilan Broz 		 * In autoclear mode, stop the loop thread
181514f27939SMilan Broz 		 * and remove configuration after last close.
181614f27939SMilan Broz 		 */
18170da03cabSJan Kara 		__loop_clr_fd(lo, true);
18180a42e99bSJan Kara 		return;
181943cade80SOmar Sandoval 	} else if (lo->lo_state == Lo_bound) {
182014f27939SMilan Broz 		/*
182114f27939SMilan Broz 		 * Otherwise keep thread (if running) and config,
182214f27939SMilan Broz 		 * but flush possible ongoing bios in thread.
182314f27939SMilan Broz 		 */
182443cade80SOmar Sandoval 		blk_mq_freeze_queue(lo->lo_queue);
182543cade80SOmar Sandoval 		blk_mq_unfreeze_queue(lo->lo_queue);
182614f27939SMilan Broz 	}
182796c58655SDavid Woodhouse 
18280a42e99bSJan Kara out_unlock:
1829310ca162STetsuo Handa 	mutex_unlock(&loop_ctl_mutex);
1830ae665016SLinus Torvalds }
1831ae665016SLinus Torvalds 
183283d5cde4SAlexey Dobriyan static const struct block_device_operations lo_fops = {
18331da177e4SLinus Torvalds 	.owner =	THIS_MODULE,
1834bb214884SAl Viro 	.open =		lo_open,
1835bb214884SAl Viro 	.release =	lo_release,
1836bb214884SAl Viro 	.ioctl =	lo_ioctl,
1837863d5b82SDavid Howells #ifdef CONFIG_COMPAT
1838bb214884SAl Viro 	.compat_ioctl =	lo_compat_ioctl,
1839863d5b82SDavid Howells #endif
18401da177e4SLinus Torvalds };
18411da177e4SLinus Torvalds 
18421da177e4SLinus Torvalds /*
18431da177e4SLinus Torvalds  * And now the modules code and kernel interface.
18441da177e4SLinus Torvalds  */
184573285082SKen Chen static int max_loop;
18465657a819SJoe Perches module_param(max_loop, int, 0444);
1847a47653fcSKen Chen MODULE_PARM_DESC(max_loop, "Maximum number of loop devices");
18485657a819SJoe Perches module_param(max_part, int, 0444);
1849476a4813SLaurent Vivier MODULE_PARM_DESC(max_part, "Maximum number of partitions per loop device");
18501da177e4SLinus Torvalds MODULE_LICENSE("GPL");
18511da177e4SLinus Torvalds MODULE_ALIAS_BLOCKDEV_MAJOR(LOOP_MAJOR);
18521da177e4SLinus Torvalds 
18531da177e4SLinus Torvalds int loop_register_transfer(struct loop_func_table *funcs)
18541da177e4SLinus Torvalds {
18551da177e4SLinus Torvalds 	unsigned int n = funcs->number;
18561da177e4SLinus Torvalds 
18571da177e4SLinus Torvalds 	if (n >= MAX_LO_CRYPT || xfer_funcs[n])
18581da177e4SLinus Torvalds 		return -EINVAL;
18591da177e4SLinus Torvalds 	xfer_funcs[n] = funcs;
18601da177e4SLinus Torvalds 	return 0;
18611da177e4SLinus Torvalds }
18621da177e4SLinus Torvalds 
186334dd82afSKay Sievers static int unregister_transfer_cb(int id, void *ptr, void *data)
186434dd82afSKay Sievers {
186534dd82afSKay Sievers 	struct loop_device *lo = ptr;
186634dd82afSKay Sievers 	struct loop_func_table *xfer = data;
186734dd82afSKay Sievers 
1868310ca162STetsuo Handa 	mutex_lock(&loop_ctl_mutex);
186934dd82afSKay Sievers 	if (lo->lo_encryption == xfer)
187034dd82afSKay Sievers 		loop_release_xfer(lo);
1871310ca162STetsuo Handa 	mutex_unlock(&loop_ctl_mutex);
187234dd82afSKay Sievers 	return 0;
187334dd82afSKay Sievers }
187434dd82afSKay Sievers 
18751da177e4SLinus Torvalds int loop_unregister_transfer(int number)
18761da177e4SLinus Torvalds {
18771da177e4SLinus Torvalds 	unsigned int n = number;
18781da177e4SLinus Torvalds 	struct loop_func_table *xfer;
18791da177e4SLinus Torvalds 
18801da177e4SLinus Torvalds 	if (n == 0 || n >= MAX_LO_CRYPT || (xfer = xfer_funcs[n]) == NULL)
18811da177e4SLinus Torvalds 		return -EINVAL;
18821da177e4SLinus Torvalds 
18831da177e4SLinus Torvalds 	xfer_funcs[n] = NULL;
188434dd82afSKay Sievers 	idr_for_each(&loop_index_idr, &unregister_transfer_cb, xfer);
18851da177e4SLinus Torvalds 	return 0;
18861da177e4SLinus Torvalds }
18871da177e4SLinus Torvalds 
18881da177e4SLinus Torvalds EXPORT_SYMBOL(loop_register_transfer);
18891da177e4SLinus Torvalds EXPORT_SYMBOL(loop_unregister_transfer);
18901da177e4SLinus Torvalds 
1891fc17b653SChristoph Hellwig static blk_status_t loop_queue_rq(struct blk_mq_hw_ctx *hctx,
1892b5dd2f60SMing Lei 		const struct blk_mq_queue_data *bd)
1893b5dd2f60SMing Lei {
18941894e916SJens Axboe 	struct request *rq = bd->rq;
18951894e916SJens Axboe 	struct loop_cmd *cmd = blk_mq_rq_to_pdu(rq);
18961894e916SJens Axboe 	struct loop_device *lo = rq->q->queuedata;
1897b5dd2f60SMing Lei 
18981894e916SJens Axboe 	blk_mq_start_request(rq);
1899b5dd2f60SMing Lei 
1900f4aa4c7bSMing Lei 	if (lo->lo_state != Lo_bound)
1901fc17b653SChristoph Hellwig 		return BLK_STS_IOERR;
1902f4aa4c7bSMing Lei 
19031894e916SJens Axboe 	switch (req_op(rq)) {
1904f0225cacSChristoph Hellwig 	case REQ_OP_FLUSH:
1905f0225cacSChristoph Hellwig 	case REQ_OP_DISCARD:
190619372e27SChristoph Hellwig 	case REQ_OP_WRITE_ZEROES:
1907bc07c10aSMing Lei 		cmd->use_aio = false;
1908f0225cacSChristoph Hellwig 		break;
1909f0225cacSChristoph Hellwig 	default:
1910f0225cacSChristoph Hellwig 		cmd->use_aio = lo->use_dio;
1911f0225cacSChristoph Hellwig 		break;
1912f0225cacSChristoph Hellwig 	}
1913bc07c10aSMing Lei 
1914d4478e92SShaohua Li 	/* always use the first bio's css */
19150b508bc9SShaohua Li #ifdef CONFIG_BLK_CGROUP
1916db6638d7SDennis Zhou 	if (cmd->use_aio && rq->bio && rq->bio->bi_blkg) {
1917db6638d7SDennis Zhou 		cmd->css = &bio_blkcg(rq->bio)->css;
1918d4478e92SShaohua Li 		css_get(cmd->css);
1919d4478e92SShaohua Li 	} else
1920d4478e92SShaohua Li #endif
1921d4478e92SShaohua Li 		cmd->css = NULL;
19223989144fSPetr Mladek 	kthread_queue_work(&lo->worker, &cmd->work);
1923b5dd2f60SMing Lei 
1924fc17b653SChristoph Hellwig 	return BLK_STS_OK;
1925b5dd2f60SMing Lei }
1926b5dd2f60SMing Lei 
1927b5dd2f60SMing Lei static void loop_handle_cmd(struct loop_cmd *cmd)
1928b5dd2f60SMing Lei {
19291894e916SJens Axboe 	struct request *rq = blk_mq_rq_from_pdu(cmd);
19301894e916SJens Axboe 	const bool write = op_is_write(req_op(rq));
19311894e916SJens Axboe 	struct loop_device *lo = rq->q->queuedata;
1932f4829a9bSChristoph Hellwig 	int ret = 0;
1933b5dd2f60SMing Lei 
1934f4829a9bSChristoph Hellwig 	if (write && (lo->lo_flags & LO_FLAGS_READ_ONLY)) {
1935f4829a9bSChristoph Hellwig 		ret = -EIO;
1936b5dd2f60SMing Lei 		goto failed;
1937f4829a9bSChristoph Hellwig 	}
1938b5dd2f60SMing Lei 
19391894e916SJens Axboe 	ret = do_req_filebacked(lo, rq);
1940b5dd2f60SMing Lei  failed:
1941bc07c10aSMing Lei 	/* complete non-aio request */
1942fe2cb290SChristoph Hellwig 	if (!cmd->use_aio || ret) {
1943fe2cb290SChristoph Hellwig 		cmd->ret = ret ? -EIO : 0;
19441894e916SJens Axboe 		blk_mq_complete_request(rq);
1945fe2cb290SChristoph Hellwig 	}
1946b5dd2f60SMing Lei }
1947b5dd2f60SMing Lei 
1948e03a3d7aSMing Lei static void loop_queue_work(struct kthread_work *work)
1949b5dd2f60SMing Lei {
1950b5dd2f60SMing Lei 	struct loop_cmd *cmd =
1951e03a3d7aSMing Lei 		container_of(work, struct loop_cmd, work);
1952b5dd2f60SMing Lei 
1953b5dd2f60SMing Lei 	loop_handle_cmd(cmd);
1954b5dd2f60SMing Lei }
1955b5dd2f60SMing Lei 
1956d6296d39SChristoph Hellwig static int loop_init_request(struct blk_mq_tag_set *set, struct request *rq,
1957d6296d39SChristoph Hellwig 		unsigned int hctx_idx, unsigned int numa_node)
1958b5dd2f60SMing Lei {
1959b5dd2f60SMing Lei 	struct loop_cmd *cmd = blk_mq_rq_to_pdu(rq);
1960b5dd2f60SMing Lei 
19613989144fSPetr Mladek 	kthread_init_work(&cmd->work, loop_queue_work);
1962b5dd2f60SMing Lei 	return 0;
1963b5dd2f60SMing Lei }
1964b5dd2f60SMing Lei 
1965f363b089SEric Biggers static const struct blk_mq_ops loop_mq_ops = {
1966b5dd2f60SMing Lei 	.queue_rq       = loop_queue_rq,
1967b5dd2f60SMing Lei 	.init_request	= loop_init_request,
1968fe2cb290SChristoph Hellwig 	.complete	= lo_complete_rq,
1969b5dd2f60SMing Lei };
1970b5dd2f60SMing Lei 
197134dd82afSKay Sievers static int loop_add(struct loop_device **l, int i)
19721da177e4SLinus Torvalds {
197373285082SKen Chen 	struct loop_device *lo;
197473285082SKen Chen 	struct gendisk *disk;
197534dd82afSKay Sievers 	int err;
19761da177e4SLinus Torvalds 
197734dd82afSKay Sievers 	err = -ENOMEM;
197868d740d7SSilva Paulo 	lo = kzalloc(sizeof(*lo), GFP_KERNEL);
197968d740d7SSilva Paulo 	if (!lo)
198073285082SKen Chen 		goto out;
198134dd82afSKay Sievers 
1982ef7e7c82SMikulas Patocka 	lo->lo_state = Lo_unbound;
1983ef7e7c82SMikulas Patocka 
1984c718aa65STejun Heo 	/* allocate id, if @id >= 0, we're requesting that specific id */
198534dd82afSKay Sievers 	if (i >= 0) {
1986c718aa65STejun Heo 		err = idr_alloc(&loop_index_idr, lo, i, i + 1, GFP_KERNEL);
1987c718aa65STejun Heo 		if (err == -ENOSPC)
198834dd82afSKay Sievers 			err = -EEXIST;
198934dd82afSKay Sievers 	} else {
1990c718aa65STejun Heo 		err = idr_alloc(&loop_index_idr, lo, 0, 0, GFP_KERNEL);
199134dd82afSKay Sievers 	}
199234dd82afSKay Sievers 	if (err < 0)
199334dd82afSKay Sievers 		goto out_free_dev;
1994c718aa65STejun Heo 	i = err;
19951da177e4SLinus Torvalds 
1996183cfb57SWei Yongjun 	err = -ENOMEM;
1997b5dd2f60SMing Lei 	lo->tag_set.ops = &loop_mq_ops;
1998b5dd2f60SMing Lei 	lo->tag_set.nr_hw_queues = 1;
1999b5dd2f60SMing Lei 	lo->tag_set.queue_depth = 128;
2000b5dd2f60SMing Lei 	lo->tag_set.numa_node = NUMA_NO_NODE;
2001b5dd2f60SMing Lei 	lo->tag_set.cmd_size = sizeof(struct loop_cmd);
200256d18f62SMing Lei 	lo->tag_set.flags = BLK_MQ_F_SHOULD_MERGE;
2003b5dd2f60SMing Lei 	lo->tag_set.driver_data = lo;
2004b5dd2f60SMing Lei 
2005b5dd2f60SMing Lei 	err = blk_mq_alloc_tag_set(&lo->tag_set);
2006b5dd2f60SMing Lei 	if (err)
20073ec981e3SMikulas Patocka 		goto out_free_idr;
200873285082SKen Chen 
2009b5dd2f60SMing Lei 	lo->lo_queue = blk_mq_init_queue(&lo->tag_set);
201038a3499fSChengguang Xu 	if (IS_ERR(lo->lo_queue)) {
2011b5dd2f60SMing Lei 		err = PTR_ERR(lo->lo_queue);
2012b5dd2f60SMing Lei 		goto out_cleanup_tags;
2013b5dd2f60SMing Lei 	}
2014ef7e7c82SMikulas Patocka 	lo->lo_queue->queuedata = lo;
2015ef7e7c82SMikulas Patocka 
201654bb0adeSShaohua Li 	blk_queue_max_hw_sectors(lo->lo_queue, BLK_DEF_MAX_SECTORS);
201740326d8aSShaohua Li 
20185b5e20f4SMing Lei 	/*
201940326d8aSShaohua Li 	 * By default, we do buffer IO, so it doesn't make sense to enable
202040326d8aSShaohua Li 	 * merge because the I/O submitted to backing file is handled page by
202140326d8aSShaohua Li 	 * page. For directio mode, merge does help to dispatch bigger request
202240326d8aSShaohua Li 	 * to underlayer disk. We will enable merge once directio is enabled.
20235b5e20f4SMing Lei 	 */
20248b904b5bSBart Van Assche 	blk_queue_flag_set(QUEUE_FLAG_NOMERGES, lo->lo_queue);
20255b5e20f4SMing Lei 
20267a649737SMinfei Huang 	err = -ENOMEM;
2027476a4813SLaurent Vivier 	disk = lo->lo_disk = alloc_disk(1 << part_shift);
202873285082SKen Chen 	if (!disk)
202973285082SKen Chen 		goto out_free_queue;
203073285082SKen Chen 
2031e03c8dd1SKay Sievers 	/*
2032e03c8dd1SKay Sievers 	 * Disable partition scanning by default. The in-kernel partition
2033e03c8dd1SKay Sievers 	 * scanning can be requested individually per-device during its
2034e03c8dd1SKay Sievers 	 * setup. Userspace can always add and remove partitions from all
2035e03c8dd1SKay Sievers 	 * devices. The needed partition minors are allocated from the
2036e03c8dd1SKay Sievers 	 * extended minor space, the main loop device numbers will continue
2037e03c8dd1SKay Sievers 	 * to match the loop minors, regardless of the number of partitions
2038e03c8dd1SKay Sievers 	 * used.
2039e03c8dd1SKay Sievers 	 *
2040e03c8dd1SKay Sievers 	 * If max_part is given, partition scanning is globally enabled for
2041e03c8dd1SKay Sievers 	 * all loop devices. The minors for the main loop devices will be
2042e03c8dd1SKay Sievers 	 * multiples of max_part.
2043e03c8dd1SKay Sievers 	 *
2044e03c8dd1SKay Sievers 	 * Note: Global-for-all-devices, set-only-at-init, read-only module
2045e03c8dd1SKay Sievers 	 * parameteters like 'max_loop' and 'max_part' make things needlessly
2046e03c8dd1SKay Sievers 	 * complicated, are too static, inflexible and may surprise
2047e03c8dd1SKay Sievers 	 * userspace tools. Parameters like this in general should be avoided.
2048e03c8dd1SKay Sievers 	 */
2049e03c8dd1SKay Sievers 	if (!part_shift)
2050e03c8dd1SKay Sievers 		disk->flags |= GENHD_FL_NO_PART_SCAN;
2051e03c8dd1SKay Sievers 	disk->flags |= GENHD_FL_EXT_DEVT;
2052f8933667SMing Lei 	atomic_set(&lo->lo_refcnt, 0);
20531da177e4SLinus Torvalds 	lo->lo_number		= i;
20541da177e4SLinus Torvalds 	spin_lock_init(&lo->lo_lock);
20551da177e4SLinus Torvalds 	disk->major		= LOOP_MAJOR;
2056476a4813SLaurent Vivier 	disk->first_minor	= i << part_shift;
20571da177e4SLinus Torvalds 	disk->fops		= &lo_fops;
20581da177e4SLinus Torvalds 	disk->private_data	= lo;
20591da177e4SLinus Torvalds 	disk->queue		= lo->lo_queue;
206073285082SKen Chen 	sprintf(disk->disk_name, "loop%d", i);
206134dd82afSKay Sievers 	add_disk(disk);
206234dd82afSKay Sievers 	*l = lo;
206334dd82afSKay Sievers 	return lo->lo_number;
206473285082SKen Chen 
206573285082SKen Chen out_free_queue:
206673285082SKen Chen 	blk_cleanup_queue(lo->lo_queue);
2067b5dd2f60SMing Lei out_cleanup_tags:
2068b5dd2f60SMing Lei 	blk_mq_free_tag_set(&lo->tag_set);
20693ec981e3SMikulas Patocka out_free_idr:
20703ec981e3SMikulas Patocka 	idr_remove(&loop_index_idr, i);
207173285082SKen Chen out_free_dev:
207273285082SKen Chen 	kfree(lo);
207373285082SKen Chen out:
207434dd82afSKay Sievers 	return err;
20751da177e4SLinus Torvalds }
20761da177e4SLinus Torvalds 
207734dd82afSKay Sievers static void loop_remove(struct loop_device *lo)
207873285082SKen Chen {
20796cd18e71SNeilBrown 	del_gendisk(lo->lo_disk);
20800fa8ebddSBart Van Assche 	blk_cleanup_queue(lo->lo_queue);
2081b5dd2f60SMing Lei 	blk_mq_free_tag_set(&lo->tag_set);
208273285082SKen Chen 	put_disk(lo->lo_disk);
208373285082SKen Chen 	kfree(lo);
208473285082SKen Chen }
208573285082SKen Chen 
2086770fe30aSKay Sievers static int find_free_cb(int id, void *ptr, void *data)
2087770fe30aSKay Sievers {
2088770fe30aSKay Sievers 	struct loop_device *lo = ptr;
2089770fe30aSKay Sievers 	struct loop_device **l = data;
2090770fe30aSKay Sievers 
2091770fe30aSKay Sievers 	if (lo->lo_state == Lo_unbound) {
2092770fe30aSKay Sievers 		*l = lo;
2093770fe30aSKay Sievers 		return 1;
2094770fe30aSKay Sievers 	}
2095770fe30aSKay Sievers 	return 0;
2096770fe30aSKay Sievers }
2097770fe30aSKay Sievers 
209834dd82afSKay Sievers static int loop_lookup(struct loop_device **l, int i)
2099a47653fcSKen Chen {
2100a47653fcSKen Chen 	struct loop_device *lo;
210134dd82afSKay Sievers 	int ret = -ENODEV;
2102a47653fcSKen Chen 
2103770fe30aSKay Sievers 	if (i < 0) {
2104770fe30aSKay Sievers 		int err;
2105770fe30aSKay Sievers 
2106770fe30aSKay Sievers 		err = idr_for_each(&loop_index_idr, &find_free_cb, &lo);
2107770fe30aSKay Sievers 		if (err == 1) {
2108770fe30aSKay Sievers 			*l = lo;
2109770fe30aSKay Sievers 			ret = lo->lo_number;
2110770fe30aSKay Sievers 		}
2111770fe30aSKay Sievers 		goto out;
2112a47653fcSKen Chen 	}
2113a47653fcSKen Chen 
2114770fe30aSKay Sievers 	/* lookup and return a specific i */
211534dd82afSKay Sievers 	lo = idr_find(&loop_index_idr, i);
2116a47653fcSKen Chen 	if (lo) {
211734dd82afSKay Sievers 		*l = lo;
211834dd82afSKay Sievers 		ret = lo->lo_number;
2119a47653fcSKen Chen 	}
2120770fe30aSKay Sievers out:
212134dd82afSKay Sievers 	return ret;
2122a47653fcSKen Chen }
2123a47653fcSKen Chen 
212473285082SKen Chen static struct kobject *loop_probe(dev_t dev, int *part, void *data)
212573285082SKen Chen {
2126705962ccSAl Viro 	struct loop_device *lo;
212707002e99SAl Viro 	struct kobject *kobj;
212834dd82afSKay Sievers 	int err;
212973285082SKen Chen 
21300a42e99bSJan Kara 	mutex_lock(&loop_ctl_mutex);
213134dd82afSKay Sievers 	err = loop_lookup(&lo, MINOR(dev) >> part_shift);
213234dd82afSKay Sievers 	if (err < 0)
213334dd82afSKay Sievers 		err = loop_add(&lo, MINOR(dev) >> part_shift);
213434dd82afSKay Sievers 	if (err < 0)
2135a207f593SMikulas Patocka 		kobj = NULL;
213634dd82afSKay Sievers 	else
21373079c22eSJan Kara 		kobj = get_disk_and_module(lo->lo_disk);
21380a42e99bSJan Kara 	mutex_unlock(&loop_ctl_mutex);
213973285082SKen Chen 
214073285082SKen Chen 	*part = 0;
214107002e99SAl Viro 	return kobj;
214273285082SKen Chen }
214373285082SKen Chen 
2144770fe30aSKay Sievers static long loop_control_ioctl(struct file *file, unsigned int cmd,
2145770fe30aSKay Sievers 			       unsigned long parm)
2146770fe30aSKay Sievers {
2147770fe30aSKay Sievers 	struct loop_device *lo;
21480a42e99bSJan Kara 	int ret;
2149770fe30aSKay Sievers 
21500a42e99bSJan Kara 	ret = mutex_lock_killable(&loop_ctl_mutex);
21510a42e99bSJan Kara 	if (ret)
21520a42e99bSJan Kara 		return ret;
21530a42e99bSJan Kara 
21540a42e99bSJan Kara 	ret = -ENOSYS;
2155770fe30aSKay Sievers 	switch (cmd) {
2156770fe30aSKay Sievers 	case LOOP_CTL_ADD:
2157770fe30aSKay Sievers 		ret = loop_lookup(&lo, parm);
2158770fe30aSKay Sievers 		if (ret >= 0) {
2159770fe30aSKay Sievers 			ret = -EEXIST;
2160770fe30aSKay Sievers 			break;
2161770fe30aSKay Sievers 		}
2162770fe30aSKay Sievers 		ret = loop_add(&lo, parm);
2163770fe30aSKay Sievers 		break;
2164770fe30aSKay Sievers 	case LOOP_CTL_REMOVE:
2165770fe30aSKay Sievers 		ret = loop_lookup(&lo, parm);
2166770fe30aSKay Sievers 		if (ret < 0)
2167770fe30aSKay Sievers 			break;
2168770fe30aSKay Sievers 		if (lo->lo_state != Lo_unbound) {
2169770fe30aSKay Sievers 			ret = -EBUSY;
2170770fe30aSKay Sievers 			break;
2171770fe30aSKay Sievers 		}
2172f8933667SMing Lei 		if (atomic_read(&lo->lo_refcnt) > 0) {
2173770fe30aSKay Sievers 			ret = -EBUSY;
2174770fe30aSKay Sievers 			break;
2175770fe30aSKay Sievers 		}
2176770fe30aSKay Sievers 		lo->lo_disk->private_data = NULL;
2177770fe30aSKay Sievers 		idr_remove(&loop_index_idr, lo->lo_number);
2178770fe30aSKay Sievers 		loop_remove(lo);
2179770fe30aSKay Sievers 		break;
2180770fe30aSKay Sievers 	case LOOP_CTL_GET_FREE:
2181770fe30aSKay Sievers 		ret = loop_lookup(&lo, -1);
2182770fe30aSKay Sievers 		if (ret >= 0)
2183770fe30aSKay Sievers 			break;
2184770fe30aSKay Sievers 		ret = loop_add(&lo, -1);
2185770fe30aSKay Sievers 	}
21860a42e99bSJan Kara 	mutex_unlock(&loop_ctl_mutex);
2187770fe30aSKay Sievers 
2188770fe30aSKay Sievers 	return ret;
2189770fe30aSKay Sievers }
2190770fe30aSKay Sievers 
2191770fe30aSKay Sievers static const struct file_operations loop_ctl_fops = {
2192770fe30aSKay Sievers 	.open		= nonseekable_open,
2193770fe30aSKay Sievers 	.unlocked_ioctl	= loop_control_ioctl,
2194770fe30aSKay Sievers 	.compat_ioctl	= loop_control_ioctl,
2195770fe30aSKay Sievers 	.owner		= THIS_MODULE,
2196770fe30aSKay Sievers 	.llseek		= noop_llseek,
2197770fe30aSKay Sievers };
2198770fe30aSKay Sievers 
2199770fe30aSKay Sievers static struct miscdevice loop_misc = {
2200770fe30aSKay Sievers 	.minor		= LOOP_CTRL_MINOR,
2201770fe30aSKay Sievers 	.name		= "loop-control",
2202770fe30aSKay Sievers 	.fops		= &loop_ctl_fops,
2203770fe30aSKay Sievers };
2204770fe30aSKay Sievers 
2205770fe30aSKay Sievers MODULE_ALIAS_MISCDEV(LOOP_CTRL_MINOR);
2206770fe30aSKay Sievers MODULE_ALIAS("devname:loop-control");
2207770fe30aSKay Sievers 
220873285082SKen Chen static int __init loop_init(void)
220973285082SKen Chen {
2210a47653fcSKen Chen 	int i, nr;
2211a47653fcSKen Chen 	unsigned long range;
221234dd82afSKay Sievers 	struct loop_device *lo;
2213770fe30aSKay Sievers 	int err;
2214a47653fcSKen Chen 
2215476a4813SLaurent Vivier 	part_shift = 0;
2216ac04fee0SNamhyung Kim 	if (max_part > 0) {
2217476a4813SLaurent Vivier 		part_shift = fls(max_part);
2218476a4813SLaurent Vivier 
2219ac04fee0SNamhyung Kim 		/*
2220ac04fee0SNamhyung Kim 		 * Adjust max_part according to part_shift as it is exported
2221ac04fee0SNamhyung Kim 		 * to user space so that user can decide correct minor number
2222ac04fee0SNamhyung Kim 		 * if [s]he want to create more devices.
2223ac04fee0SNamhyung Kim 		 *
2224ac04fee0SNamhyung Kim 		 * Note that -1 is required because partition 0 is reserved
2225ac04fee0SNamhyung Kim 		 * for the whole disk.
2226ac04fee0SNamhyung Kim 		 */
2227ac04fee0SNamhyung Kim 		max_part = (1UL << part_shift) - 1;
2228ac04fee0SNamhyung Kim 	}
2229ac04fee0SNamhyung Kim 
2230b1a66504SGuo Chao 	if ((1UL << part_shift) > DISK_MAX_PARTS) {
2231b1a66504SGuo Chao 		err = -EINVAL;
2232a8c1d064SAnton Volkov 		goto err_out;
2233b1a66504SGuo Chao 	}
223478f4bb36SNamhyung Kim 
2235b1a66504SGuo Chao 	if (max_loop > 1UL << (MINORBITS - part_shift)) {
2236b1a66504SGuo Chao 		err = -EINVAL;
2237a8c1d064SAnton Volkov 		goto err_out;
2238b1a66504SGuo Chao 	}
223973285082SKen Chen 
2240d134b00bSKay Sievers 	/*
2241d134b00bSKay Sievers 	 * If max_loop is specified, create that many devices upfront.
2242d134b00bSKay Sievers 	 * This also becomes a hard limit. If max_loop is not specified,
2243d134b00bSKay Sievers 	 * create CONFIG_BLK_DEV_LOOP_MIN_COUNT loop devices at module
2244d134b00bSKay Sievers 	 * init time. Loop devices can be requested on-demand with the
2245d134b00bSKay Sievers 	 * /dev/loop-control interface, or be instantiated by accessing
2246d134b00bSKay Sievers 	 * a 'dead' device node.
2247d134b00bSKay Sievers 	 */
224873285082SKen Chen 	if (max_loop) {
2249a47653fcSKen Chen 		nr = max_loop;
2250a1c15c59SNamhyung Kim 		range = max_loop << part_shift;
2251a47653fcSKen Chen 	} else {
2252d134b00bSKay Sievers 		nr = CONFIG_BLK_DEV_LOOP_MIN_COUNT;
2253a1c15c59SNamhyung Kim 		range = 1UL << MINORBITS;
225473285082SKen Chen 	}
2255a47653fcSKen Chen 
2256a8c1d064SAnton Volkov 	err = misc_register(&loop_misc);
2257a8c1d064SAnton Volkov 	if (err < 0)
2258a8c1d064SAnton Volkov 		goto err_out;
2259a8c1d064SAnton Volkov 
2260a8c1d064SAnton Volkov 
2261b1a66504SGuo Chao 	if (register_blkdev(LOOP_MAJOR, "loop")) {
2262b1a66504SGuo Chao 		err = -EIO;
2263b1a66504SGuo Chao 		goto misc_out;
2264b1a66504SGuo Chao 	}
2265a47653fcSKen Chen 
2266a47653fcSKen Chen 	blk_register_region(MKDEV(LOOP_MAJOR, 0), range,
2267a47653fcSKen Chen 				  THIS_MODULE, loop_probe, NULL, NULL);
2268a47653fcSKen Chen 
2269d134b00bSKay Sievers 	/* pre-create number of devices given by config or max_loop */
22700a42e99bSJan Kara 	mutex_lock(&loop_ctl_mutex);
227134dd82afSKay Sievers 	for (i = 0; i < nr; i++)
227234dd82afSKay Sievers 		loop_add(&lo, i);
22730a42e99bSJan Kara 	mutex_unlock(&loop_ctl_mutex);
227434dd82afSKay Sievers 
227573285082SKen Chen 	printk(KERN_INFO "loop: module loaded\n");
22761da177e4SLinus Torvalds 	return 0;
2277b1a66504SGuo Chao 
2278b1a66504SGuo Chao misc_out:
2279b1a66504SGuo Chao 	misc_deregister(&loop_misc);
2280a8c1d064SAnton Volkov err_out:
2281b1a66504SGuo Chao 	return err;
228234dd82afSKay Sievers }
2283a47653fcSKen Chen 
228434dd82afSKay Sievers static int loop_exit_cb(int id, void *ptr, void *data)
228534dd82afSKay Sievers {
228634dd82afSKay Sievers 	struct loop_device *lo = ptr;
2287a47653fcSKen Chen 
228834dd82afSKay Sievers 	loop_remove(lo);
228934dd82afSKay Sievers 	return 0;
22901da177e4SLinus Torvalds }
22911da177e4SLinus Torvalds 
229273285082SKen Chen static void __exit loop_exit(void)
22931da177e4SLinus Torvalds {
2294a47653fcSKen Chen 	unsigned long range;
22951da177e4SLinus Torvalds 
2296a1c15c59SNamhyung Kim 	range = max_loop ? max_loop << part_shift : 1UL << MINORBITS;
2297a47653fcSKen Chen 
229834dd82afSKay Sievers 	idr_for_each(&loop_index_idr, &loop_exit_cb, NULL);
229934dd82afSKay Sievers 	idr_destroy(&loop_index_idr);
230073285082SKen Chen 
2301a47653fcSKen Chen 	blk_unregister_region(MKDEV(LOOP_MAJOR, 0), range);
230200d59405SAkinobu Mita 	unregister_blkdev(LOOP_MAJOR, "loop");
2303770fe30aSKay Sievers 
2304770fe30aSKay Sievers 	misc_deregister(&loop_misc);
23051da177e4SLinus Torvalds }
23061da177e4SLinus Torvalds 
23071da177e4SLinus Torvalds module_init(loop_init);
23081da177e4SLinus Torvalds module_exit(loop_exit);
23091da177e4SLinus Torvalds 
23101da177e4SLinus Torvalds #ifndef MODULE
23111da177e4SLinus Torvalds static int __init max_loop_setup(char *str)
23121da177e4SLinus Torvalds {
23131da177e4SLinus Torvalds 	max_loop = simple_strtol(str, NULL, 0);
23141da177e4SLinus Torvalds 	return 1;
23151da177e4SLinus Torvalds }
23161da177e4SLinus Torvalds 
23171da177e4SLinus Torvalds __setup("max_loop=", max_loop_setup);
23181da177e4SLinus Torvalds #endif
2319