xref: /openbmc/linux/drivers/block/loop.c (revision 9c64e38c)
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>
564ee60ec1SMatthew Wilcox (Oracle) #include <linux/pagemap.h>
571da177e4SLinus Torvalds #include <linux/file.h>
581da177e4SLinus Torvalds #include <linux/stat.h>
591da177e4SLinus Torvalds #include <linux/errno.h>
601da177e4SLinus Torvalds #include <linux/major.h>
611da177e4SLinus Torvalds #include <linux/wait.h>
621da177e4SLinus Torvalds #include <linux/blkdev.h>
631da177e4SLinus Torvalds #include <linux/blkpg.h>
641da177e4SLinus Torvalds #include <linux/init.h>
651da177e4SLinus Torvalds #include <linux/swap.h>
661da177e4SLinus Torvalds #include <linux/slab.h>
67863d5b82SDavid Howells #include <linux/compat.h>
681da177e4SLinus Torvalds #include <linux/suspend.h>
6983144186SRafael J. Wysocki #include <linux/freezer.h>
702a48fc0aSArnd Bergmann #include <linux/mutex.h>
711da177e4SLinus Torvalds #include <linux/writeback.h>
721da177e4SLinus Torvalds #include <linux/completion.h>
731da177e4SLinus Torvalds #include <linux/highmem.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>
81c74d40e8SDan Schatzberg #include <linux/sched/mm.h>
82d9a08a9eSAdam Manzanares 
8383a87611SAl Viro #include "loop.h"
841da177e4SLinus Torvalds 
857c0f6ba6SLinus Torvalds #include <linux/uaccess.h>
861da177e4SLinus Torvalds 
8787579e9bSDan Schatzberg #define LOOP_IDLE_WORKER_TIMEOUT (60 * HZ)
8887579e9bSDan Schatzberg 
8934dd82afSKay Sievers static DEFINE_IDR(loop_index_idr);
90310ca162STetsuo Handa static DEFINE_MUTEX(loop_ctl_mutex);
913ce6e1f6STetsuo Handa static DEFINE_MUTEX(loop_validate_mutex);
923ce6e1f6STetsuo Handa 
933ce6e1f6STetsuo Handa /**
943ce6e1f6STetsuo Handa  * loop_global_lock_killable() - take locks for safe loop_validate_file() test
953ce6e1f6STetsuo Handa  *
963ce6e1f6STetsuo Handa  * @lo: struct loop_device
973ce6e1f6STetsuo Handa  * @global: true if @lo is about to bind another "struct loop_device", false otherwise
983ce6e1f6STetsuo Handa  *
993ce6e1f6STetsuo Handa  * Returns 0 on success, -EINTR otherwise.
1003ce6e1f6STetsuo Handa  *
1013ce6e1f6STetsuo Handa  * Since loop_validate_file() traverses on other "struct loop_device" if
1023ce6e1f6STetsuo Handa  * is_loop_device() is true, we need a global lock for serializing concurrent
1033ce6e1f6STetsuo Handa  * loop_configure()/loop_change_fd()/__loop_clr_fd() calls.
1043ce6e1f6STetsuo Handa  */
1053ce6e1f6STetsuo Handa static int loop_global_lock_killable(struct loop_device *lo, bool global)
1063ce6e1f6STetsuo Handa {
1073ce6e1f6STetsuo Handa 	int err;
1083ce6e1f6STetsuo Handa 
1093ce6e1f6STetsuo Handa 	if (global) {
1103ce6e1f6STetsuo Handa 		err = mutex_lock_killable(&loop_validate_mutex);
1113ce6e1f6STetsuo Handa 		if (err)
1123ce6e1f6STetsuo Handa 			return err;
1133ce6e1f6STetsuo Handa 	}
1143ce6e1f6STetsuo Handa 	err = mutex_lock_killable(&lo->lo_mutex);
1153ce6e1f6STetsuo Handa 	if (err && global)
1163ce6e1f6STetsuo Handa 		mutex_unlock(&loop_validate_mutex);
1173ce6e1f6STetsuo Handa 	return err;
1183ce6e1f6STetsuo Handa }
1193ce6e1f6STetsuo Handa 
1203ce6e1f6STetsuo Handa /**
1213ce6e1f6STetsuo Handa  * loop_global_unlock() - release locks taken by loop_global_lock_killable()
1223ce6e1f6STetsuo Handa  *
1233ce6e1f6STetsuo Handa  * @lo: struct loop_device
1243ce6e1f6STetsuo Handa  * @global: true if @lo was about to bind another "struct loop_device", false otherwise
1253ce6e1f6STetsuo Handa  */
1263ce6e1f6STetsuo Handa static void loop_global_unlock(struct loop_device *lo, bool global)
1273ce6e1f6STetsuo Handa {
1283ce6e1f6STetsuo Handa 	mutex_unlock(&lo->lo_mutex);
1293ce6e1f6STetsuo Handa 	if (global)
1303ce6e1f6STetsuo Handa 		mutex_unlock(&loop_validate_mutex);
1313ce6e1f6STetsuo Handa }
1321da177e4SLinus Torvalds 
133476a4813SLaurent Vivier static int max_part;
134476a4813SLaurent Vivier static int part_shift;
135476a4813SLaurent Vivier 
1367035b5dfSDmitry Monakhov static loff_t get_size(loff_t offset, loff_t sizelimit, struct file *file)
1371da177e4SLinus Torvalds {
138b7a1da69SGuo Chao 	loff_t loopsize;
1391da177e4SLinus Torvalds 
1401da177e4SLinus Torvalds 	/* Compute loopsize in bytes */
141b7a1da69SGuo Chao 	loopsize = i_size_read(file->f_mapping->host);
142b7a1da69SGuo Chao 	if (offset > 0)
143b7a1da69SGuo Chao 		loopsize -= offset;
144b7a1da69SGuo Chao 	/* offset is beyond i_size, weird but possible */
1457035b5dfSDmitry Monakhov 	if (loopsize < 0)
1467035b5dfSDmitry Monakhov 		return 0;
1471da177e4SLinus Torvalds 
1487035b5dfSDmitry Monakhov 	if (sizelimit > 0 && sizelimit < loopsize)
1497035b5dfSDmitry Monakhov 		loopsize = sizelimit;
1501da177e4SLinus Torvalds 	/*
1511da177e4SLinus Torvalds 	 * Unfortunately, if we want to do I/O on the device,
1521da177e4SLinus Torvalds 	 * the number of 512-byte sectors has to fit into a sector_t.
1531da177e4SLinus Torvalds 	 */
1541da177e4SLinus Torvalds 	return loopsize >> 9;
1551da177e4SLinus Torvalds }
1561da177e4SLinus Torvalds 
1577035b5dfSDmitry Monakhov static loff_t get_loop_size(struct loop_device *lo, struct file *file)
1581da177e4SLinus Torvalds {
1597035b5dfSDmitry Monakhov 	return get_size(lo->lo_offset, lo->lo_sizelimit, file);
1607035b5dfSDmitry Monakhov }
1617035b5dfSDmitry Monakhov 
1622e5ab5f3SMing Lei static void __loop_update_dio(struct loop_device *lo, bool dio)
1632e5ab5f3SMing Lei {
1642e5ab5f3SMing Lei 	struct file *file = lo->lo_backing_file;
1652e5ab5f3SMing Lei 	struct address_space *mapping = file->f_mapping;
1662e5ab5f3SMing Lei 	struct inode *inode = mapping->host;
1672e5ab5f3SMing Lei 	unsigned short sb_bsize = 0;
1682e5ab5f3SMing Lei 	unsigned dio_align = 0;
1692e5ab5f3SMing Lei 	bool use_dio;
1702e5ab5f3SMing Lei 
1712e5ab5f3SMing Lei 	if (inode->i_sb->s_bdev) {
1722e5ab5f3SMing Lei 		sb_bsize = bdev_logical_block_size(inode->i_sb->s_bdev);
1732e5ab5f3SMing Lei 		dio_align = sb_bsize - 1;
1742e5ab5f3SMing Lei 	}
1752e5ab5f3SMing Lei 
1762e5ab5f3SMing Lei 	/*
1772e5ab5f3SMing Lei 	 * We support direct I/O only if lo_offset is aligned with the
1782e5ab5f3SMing Lei 	 * logical I/O size of backing device, and the logical block
17947e96246SChristoph Hellwig 	 * size of loop is bigger than the backing device's.
1802e5ab5f3SMing Lei 	 *
1812e5ab5f3SMing Lei 	 * TODO: the above condition may be loosed in the future, and
1822e5ab5f3SMing Lei 	 * direct I/O may be switched runtime at that time because most
18389d790abSMasahiro Yamada 	 * of requests in sane applications should be PAGE_SIZE aligned
1842e5ab5f3SMing Lei 	 */
1852e5ab5f3SMing Lei 	if (dio) {
1862e5ab5f3SMing Lei 		if (queue_logical_block_size(lo->lo_queue) >= sb_bsize &&
1872e5ab5f3SMing Lei 				!(lo->lo_offset & dio_align) &&
18847e96246SChristoph Hellwig 				mapping->a_ops->direct_IO)
1892e5ab5f3SMing Lei 			use_dio = true;
1902e5ab5f3SMing Lei 		else
1912e5ab5f3SMing Lei 			use_dio = false;
1922e5ab5f3SMing Lei 	} else {
1932e5ab5f3SMing Lei 		use_dio = false;
1942e5ab5f3SMing Lei 	}
1952e5ab5f3SMing Lei 
1962e5ab5f3SMing Lei 	if (lo->use_dio == use_dio)
1972e5ab5f3SMing Lei 		return;
1982e5ab5f3SMing Lei 
1992e5ab5f3SMing Lei 	/* flush dirty pages before changing direct IO */
2002e5ab5f3SMing Lei 	vfs_fsync(file, 0);
2012e5ab5f3SMing Lei 
2022e5ab5f3SMing Lei 	/*
2032e5ab5f3SMing Lei 	 * The flag of LO_FLAGS_DIRECT_IO is handled similarly with
2042e5ab5f3SMing Lei 	 * LO_FLAGS_READ_ONLY, both are set from kernel, and losetup
2052e5ab5f3SMing Lei 	 * will get updated by ioctl(LOOP_GET_STATUS)
2062e5ab5f3SMing Lei 	 */
2070fbcf579SMartijn Coenen 	if (lo->lo_state == Lo_bound)
2082e5ab5f3SMing Lei 		blk_mq_freeze_queue(lo->lo_queue);
2092e5ab5f3SMing Lei 	lo->use_dio = use_dio;
21040326d8aSShaohua Li 	if (use_dio) {
2118b904b5bSBart Van Assche 		blk_queue_flag_clear(QUEUE_FLAG_NOMERGES, lo->lo_queue);
2122e5ab5f3SMing Lei 		lo->lo_flags |= LO_FLAGS_DIRECT_IO;
21340326d8aSShaohua Li 	} else {
2148b904b5bSBart Van Assche 		blk_queue_flag_set(QUEUE_FLAG_NOMERGES, lo->lo_queue);
2152e5ab5f3SMing Lei 		lo->lo_flags &= ~LO_FLAGS_DIRECT_IO;
21640326d8aSShaohua Li 	}
2170fbcf579SMartijn Coenen 	if (lo->lo_state == Lo_bound)
2182e5ab5f3SMing Lei 		blk_mq_unfreeze_queue(lo->lo_queue);
2192e5ab5f3SMing Lei }
2202e5ab5f3SMing Lei 
2215795b6f5SMartijn Coenen /**
2225795b6f5SMartijn Coenen  * loop_set_size() - sets device size and notifies userspace
2235795b6f5SMartijn Coenen  * @lo: struct loop_device to set the size for
2245795b6f5SMartijn Coenen  * @size: new size of the loop device
2255795b6f5SMartijn Coenen  *
2265795b6f5SMartijn Coenen  * Callers must validate that the size passed into this function fits into
2275795b6f5SMartijn Coenen  * a sector_t, eg using loop_validate_size()
2285795b6f5SMartijn Coenen  */
2295795b6f5SMartijn Coenen static void loop_set_size(struct loop_device *lo, loff_t size)
2305795b6f5SMartijn Coenen {
231449f4ec9SChristoph Hellwig 	if (!set_capacity_and_notify(lo->lo_disk, size))
2323b4f85d0SChristoph Hellwig 		kobject_uevent(&disk_to_dev(lo->lo_disk)->kobj, KOBJ_CHANGE);
2331da177e4SLinus Torvalds }
2341da177e4SLinus Torvalds 
235aa4d8616SChristoph Hellwig static int lo_write_bvec(struct file *file, struct bio_vec *bvec, loff_t *ppos)
2361da177e4SLinus Torvalds {
237aa4d8616SChristoph Hellwig 	struct iov_iter i;
2381da177e4SLinus Torvalds 	ssize_t bw;
239283e7e5dSAl Viro 
240b6207430SChristoph Hellwig 	iov_iter_bvec(&i, WRITE, bvec, 1, bvec->bv_len);
2411da177e4SLinus Torvalds 
24203d95eb2SAl Viro 	file_start_write(file);
243abbb6589SChristoph Hellwig 	bw = vfs_iter_write(file, &i, ppos, 0);
24403d95eb2SAl Viro 	file_end_write(file);
245aa4d8616SChristoph Hellwig 
246aa4d8616SChristoph Hellwig 	if (likely(bw ==  bvec->bv_len))
2471da177e4SLinus Torvalds 		return 0;
248aa4d8616SChristoph Hellwig 
249aa4d8616SChristoph Hellwig 	printk_ratelimited(KERN_ERR
250aa4d8616SChristoph Hellwig 		"loop: Write error at byte offset %llu, length %i.\n",
251aa4d8616SChristoph Hellwig 		(unsigned long long)*ppos, bvec->bv_len);
2521da177e4SLinus Torvalds 	if (bw >= 0)
2531da177e4SLinus Torvalds 		bw = -EIO;
2541da177e4SLinus Torvalds 	return bw;
2551da177e4SLinus Torvalds }
2561da177e4SLinus Torvalds 
257aa4d8616SChristoph Hellwig static int lo_write_simple(struct loop_device *lo, struct request *rq,
258aa4d8616SChristoph Hellwig 		loff_t pos)
2591da177e4SLinus Torvalds {
260aa4d8616SChristoph Hellwig 	struct bio_vec bvec;
261aa4d8616SChristoph Hellwig 	struct req_iterator iter;
262aa4d8616SChristoph Hellwig 	int ret = 0;
263aa4d8616SChristoph Hellwig 
264aa4d8616SChristoph Hellwig 	rq_for_each_segment(bvec, rq, iter) {
265aa4d8616SChristoph Hellwig 		ret = lo_write_bvec(lo->lo_backing_file, &bvec, &pos);
266aa4d8616SChristoph Hellwig 		if (ret < 0)
267aa4d8616SChristoph Hellwig 			break;
2681da177e4SLinus Torvalds 		cond_resched();
2691da177e4SLinus Torvalds 	}
2701da177e4SLinus Torvalds 
271aa4d8616SChristoph Hellwig 	return ret;
272aa4d8616SChristoph Hellwig }
273aa4d8616SChristoph Hellwig 
274aa4d8616SChristoph Hellwig static int lo_read_simple(struct loop_device *lo, struct request *rq,
275aa4d8616SChristoph Hellwig 		loff_t pos)
2761da177e4SLinus Torvalds {
2777988613bSKent Overstreet 	struct bio_vec bvec;
27830112013SMing Lei 	struct req_iterator iter;
279aa4d8616SChristoph Hellwig 	struct iov_iter i;
280aa4d8616SChristoph Hellwig 	ssize_t len;
2811da177e4SLinus Torvalds 
28230112013SMing Lei 	rq_for_each_segment(bvec, rq, iter) {
283b6207430SChristoph Hellwig 		iov_iter_bvec(&i, READ, &bvec, 1, bvec.bv_len);
28418e9710eSChristoph Hellwig 		len = vfs_iter_read(lo->lo_backing_file, &i, &pos, 0);
285aa4d8616SChristoph Hellwig 		if (len < 0)
286aa4d8616SChristoph Hellwig 			return len;
287306df071SDave Young 
288aa4d8616SChristoph Hellwig 		flush_dcache_page(bvec.bv_page);
289aa4d8616SChristoph Hellwig 
290aa4d8616SChristoph Hellwig 		if (len != bvec.bv_len) {
29130112013SMing Lei 			struct bio *bio;
29230112013SMing Lei 
29330112013SMing Lei 			__rq_for_each_bio(bio, rq)
294306df071SDave Young 				zero_fill_bio(bio);
2951da177e4SLinus Torvalds 			break;
296306df071SDave Young 		}
297aa4d8616SChristoph Hellwig 		cond_resched();
2981da177e4SLinus Torvalds 	}
299aa4d8616SChristoph Hellwig 
300306df071SDave Young 	return 0;
3011da177e4SLinus Torvalds }
3021da177e4SLinus Torvalds 
303efcfec57SDarrick J. Wong static int lo_fallocate(struct loop_device *lo, struct request *rq, loff_t pos,
304efcfec57SDarrick J. Wong 			int mode)
305cf655d95SMing Lei {
306cf655d95SMing Lei 	/*
307efcfec57SDarrick J. Wong 	 * We use fallocate to manipulate the space mappings used by the image
30847e96246SChristoph Hellwig 	 * a.k.a. discard/zerorange.
309cf655d95SMing Lei 	 */
310cf655d95SMing Lei 	struct file *file = lo->lo_backing_file;
311cf655d95SMing Lei 	int ret;
312cf655d95SMing Lei 
313efcfec57SDarrick J. Wong 	mode |= FALLOC_FL_KEEP_SIZE;
314efcfec57SDarrick J. Wong 
3150aab29b8SChaitanya Kulkarni 	if (!blk_queue_discard(lo->lo_queue)) {
316cf655d95SMing Lei 		ret = -EOPNOTSUPP;
317cf655d95SMing Lei 		goto out;
318cf655d95SMing Lei 	}
319cf655d95SMing Lei 
320cf655d95SMing Lei 	ret = file->f_op->fallocate(file, mode, pos, blk_rq_bytes(rq));
321cf655d95SMing Lei 	if (unlikely(ret && ret != -EINVAL && ret != -EOPNOTSUPP))
322cf655d95SMing Lei 		ret = -EIO;
323cf655d95SMing Lei  out:
324cf655d95SMing Lei 	return ret;
325cf655d95SMing Lei }
326cf655d95SMing Lei 
327cf655d95SMing Lei static int lo_req_flush(struct loop_device *lo, struct request *rq)
328cf655d95SMing Lei {
329*9c64e38cSChaitanya Kulkarni 	int ret = vfs_fsync(lo->lo_backing_file, 0);
330cf655d95SMing Lei 	if (unlikely(ret && ret != -EINVAL))
331cf655d95SMing Lei 		ret = -EIO;
332cf655d95SMing Lei 
333cf655d95SMing Lei 	return ret;
334cf655d95SMing Lei }
335cf655d95SMing Lei 
336fe2cb290SChristoph Hellwig static void lo_complete_rq(struct request *rq)
337bc07c10aSMing Lei {
338fe2cb290SChristoph Hellwig 	struct loop_cmd *cmd = blk_mq_rq_to_pdu(rq);
339f9de14bcSJens Axboe 	blk_status_t ret = BLK_STS_OK;
340bc07c10aSMing Lei 
341f9de14bcSJens Axboe 	if (!cmd->use_aio || cmd->ret < 0 || cmd->ret == blk_rq_bytes(rq) ||
342f9de14bcSJens Axboe 	    req_op(rq) != REQ_OP_READ) {
343f9de14bcSJens Axboe 		if (cmd->ret < 0)
3448cd55087SEvan Green 			ret = errno_to_blk_status(cmd->ret);
345f9de14bcSJens Axboe 		goto end_io;
346bc07c10aSMing Lei 	}
347fe2cb290SChristoph Hellwig 
348f9de14bcSJens Axboe 	/*
349f9de14bcSJens Axboe 	 * Short READ - if we got some data, advance our request and
350f9de14bcSJens Axboe 	 * retry it. If we got no data, end the rest with EIO.
351f9de14bcSJens Axboe 	 */
352f9de14bcSJens Axboe 	if (cmd->ret) {
353f9de14bcSJens Axboe 		blk_update_request(rq, BLK_STS_OK, cmd->ret);
354f9de14bcSJens Axboe 		cmd->ret = 0;
355f9de14bcSJens Axboe 		blk_mq_requeue_request(rq, true);
356f9de14bcSJens Axboe 	} else {
357f9de14bcSJens Axboe 		if (cmd->use_aio) {
358f9de14bcSJens Axboe 			struct bio *bio = rq->bio;
359f9de14bcSJens Axboe 
360f9de14bcSJens Axboe 			while (bio) {
361f9de14bcSJens Axboe 				zero_fill_bio(bio);
362f9de14bcSJens Axboe 				bio = bio->bi_next;
363f9de14bcSJens Axboe 			}
364f9de14bcSJens Axboe 		}
365f9de14bcSJens Axboe 		ret = BLK_STS_IOERR;
366f9de14bcSJens Axboe end_io:
367f9de14bcSJens Axboe 		blk_mq_end_request(rq, ret);
368f9de14bcSJens Axboe 	}
369bc07c10aSMing Lei }
370bc07c10aSMing Lei 
37192d77332SShaohua Li static void lo_rw_aio_do_completion(struct loop_cmd *cmd)
37292d77332SShaohua Li {
3731894e916SJens Axboe 	struct request *rq = blk_mq_rq_from_pdu(cmd);
3741894e916SJens Axboe 
37592d77332SShaohua Li 	if (!atomic_dec_and_test(&cmd->ref))
37692d77332SShaohua Li 		return;
37792d77332SShaohua Li 	kfree(cmd->bvec);
37892d77332SShaohua Li 	cmd->bvec = NULL;
37915f73f5bSChristoph Hellwig 	if (likely(!blk_should_fake_timeout(rq->q)))
3801894e916SJens Axboe 		blk_mq_complete_request(rq);
38192d77332SShaohua Li }
38292d77332SShaohua Li 
3836b19b766SJens Axboe static void lo_rw_aio_complete(struct kiocb *iocb, long ret)
384bc07c10aSMing Lei {
385bc07c10aSMing Lei 	struct loop_cmd *cmd = container_of(iocb, struct loop_cmd, iocb);
386bc07c10aSMing Lei 
387fe2cb290SChristoph Hellwig 	cmd->ret = ret;
38892d77332SShaohua Li 	lo_rw_aio_do_completion(cmd);
389bc07c10aSMing Lei }
390bc07c10aSMing Lei 
391bc07c10aSMing Lei static int lo_rw_aio(struct loop_device *lo, struct loop_cmd *cmd,
392bc07c10aSMing Lei 		     loff_t pos, bool rw)
393bc07c10aSMing Lei {
394bc07c10aSMing Lei 	struct iov_iter iter;
39586af5952SMing Lei 	struct req_iterator rq_iter;
396bc07c10aSMing Lei 	struct bio_vec *bvec;
3971894e916SJens Axboe 	struct request *rq = blk_mq_rq_from_pdu(cmd);
39840326d8aSShaohua Li 	struct bio *bio = rq->bio;
399bc07c10aSMing Lei 	struct file *file = lo->lo_backing_file;
40086af5952SMing Lei 	struct bio_vec tmp;
40140326d8aSShaohua Li 	unsigned int offset;
40286af5952SMing Lei 	int nr_bvec = 0;
403bc07c10aSMing Lei 	int ret;
404bc07c10aSMing Lei 
40586af5952SMing Lei 	rq_for_each_bvec(tmp, rq, rq_iter)
40686af5952SMing Lei 		nr_bvec++;
407bc07c10aSMing Lei 
40886af5952SMing Lei 	if (rq->bio != rq->biotail) {
40986af5952SMing Lei 
41086af5952SMing Lei 		bvec = kmalloc_array(nr_bvec, sizeof(struct bio_vec),
4116da2ec56SKees Cook 				     GFP_NOIO);
41240326d8aSShaohua Li 		if (!bvec)
41340326d8aSShaohua Li 			return -EIO;
41440326d8aSShaohua Li 		cmd->bvec = bvec;
41540326d8aSShaohua Li 
416a7297a6aSMing Lei 		/*
41740326d8aSShaohua Li 		 * The bios of the request may be started from the middle of
41840326d8aSShaohua Li 		 * the 'bvec' because of bio splitting, so we can't directly
41986af5952SMing Lei 		 * copy bio->bi_iov_vec to new bvec. The rq_for_each_bvec
42040326d8aSShaohua Li 		 * API will take care of all details for us.
421a7297a6aSMing Lei 		 */
42286af5952SMing Lei 		rq_for_each_bvec(tmp, rq, rq_iter) {
42340326d8aSShaohua Li 			*bvec = tmp;
42440326d8aSShaohua Li 			bvec++;
42540326d8aSShaohua Li 		}
42640326d8aSShaohua Li 		bvec = cmd->bvec;
42740326d8aSShaohua Li 		offset = 0;
42840326d8aSShaohua Li 	} else {
42940326d8aSShaohua Li 		/*
43040326d8aSShaohua Li 		 * Same here, this bio may be started from the middle of the
43140326d8aSShaohua Li 		 * 'bvec' because of bio splitting, so offset from the bvec
43240326d8aSShaohua Li 		 * must be passed to iov iterator
43340326d8aSShaohua Li 		 */
43440326d8aSShaohua Li 		offset = bio->bi_iter.bi_bvec_done;
43540326d8aSShaohua Li 		bvec = __bvec_iter_bvec(bio->bi_io_vec, bio->bi_iter);
43640326d8aSShaohua Li 	}
43792d77332SShaohua Li 	atomic_set(&cmd->ref, 2);
43840326d8aSShaohua Li 
439b6207430SChristoph Hellwig 	iov_iter_bvec(&iter, rw, bvec, nr_bvec, blk_rq_bytes(rq));
44040326d8aSShaohua Li 	iter.iov_offset = offset;
441bc07c10aSMing Lei 
442bc07c10aSMing Lei 	cmd->iocb.ki_pos = pos;
443bc07c10aSMing Lei 	cmd->iocb.ki_filp = file;
444bc07c10aSMing Lei 	cmd->iocb.ki_complete = lo_rw_aio_complete;
445bc07c10aSMing Lei 	cmd->iocb.ki_flags = IOCB_DIRECT;
446d9a08a9eSAdam Manzanares 	cmd->iocb.ki_ioprio = IOPRIO_PRIO_VALUE(IOPRIO_CLASS_NONE, 0);
447bc07c10aSMing Lei 
448bc07c10aSMing Lei 	if (rw == WRITE)
449bb7462b6SMiklos Szeredi 		ret = call_write_iter(file, &cmd->iocb, &iter);
450bc07c10aSMing Lei 	else
451bb7462b6SMiklos Szeredi 		ret = call_read_iter(file, &cmd->iocb, &iter);
452bc07c10aSMing Lei 
45392d77332SShaohua Li 	lo_rw_aio_do_completion(cmd);
45492d77332SShaohua Li 
455bc07c10aSMing Lei 	if (ret != -EIOCBQUEUED)
4566b19b766SJens Axboe 		lo_rw_aio_complete(&cmd->iocb, ret);
457bc07c10aSMing Lei 	return 0;
458bc07c10aSMing Lei }
459bc07c10aSMing Lei 
460c1c87c2bSChristoph Hellwig static int do_req_filebacked(struct loop_device *lo, struct request *rq)
461bc07c10aSMing Lei {
462bc07c10aSMing Lei 	struct loop_cmd *cmd = blk_mq_rq_to_pdu(rq);
463c1c87c2bSChristoph Hellwig 	loff_t pos = ((loff_t) blk_rq_pos(rq) << 9) + lo->lo_offset;
464bc07c10aSMing Lei 
465bc07c10aSMing Lei 	/*
466bc07c10aSMing Lei 	 * lo_write_simple and lo_read_simple should have been covered
467bc07c10aSMing Lei 	 * by io submit style function like lo_rw_aio(), one blocker
468bc07c10aSMing Lei 	 * is that lo_read_simple() need to call flush_dcache_page after
469bc07c10aSMing Lei 	 * the page is written from kernel, and it isn't easy to handle
470bc07c10aSMing Lei 	 * this in io submit style function which submits all segments
471bc07c10aSMing Lei 	 * of the req at one time. And direct read IO doesn't need to
472bc07c10aSMing Lei 	 * run flush_dcache_page().
473bc07c10aSMing Lei 	 */
474c1c87c2bSChristoph Hellwig 	switch (req_op(rq)) {
475c1c87c2bSChristoph Hellwig 	case REQ_OP_FLUSH:
476c1c87c2bSChristoph Hellwig 		return lo_req_flush(lo, rq);
47719372e27SChristoph Hellwig 	case REQ_OP_WRITE_ZEROES:
478efcfec57SDarrick J. Wong 		/*
479efcfec57SDarrick J. Wong 		 * If the caller doesn't want deallocation, call zeroout to
480efcfec57SDarrick J. Wong 		 * write zeroes the range.  Otherwise, punch them out.
481efcfec57SDarrick J. Wong 		 */
482efcfec57SDarrick J. Wong 		return lo_fallocate(lo, rq, pos,
483efcfec57SDarrick J. Wong 			(rq->cmd_flags & REQ_NOUNMAP) ?
484efcfec57SDarrick J. Wong 				FALLOC_FL_ZERO_RANGE :
485efcfec57SDarrick J. Wong 				FALLOC_FL_PUNCH_HOLE);
486efcfec57SDarrick J. Wong 	case REQ_OP_DISCARD:
487efcfec57SDarrick J. Wong 		return lo_fallocate(lo, rq, pos, FALLOC_FL_PUNCH_HOLE);
488c1c87c2bSChristoph Hellwig 	case REQ_OP_WRITE:
48947e96246SChristoph Hellwig 		if (cmd->use_aio)
490c1c87c2bSChristoph Hellwig 			return lo_rw_aio(lo, cmd, pos, WRITE);
491c1c87c2bSChristoph Hellwig 		else
492bc07c10aSMing Lei 			return lo_write_simple(lo, rq, pos);
493c1c87c2bSChristoph Hellwig 	case REQ_OP_READ:
49447e96246SChristoph Hellwig 		if (cmd->use_aio)
495c1c87c2bSChristoph Hellwig 			return lo_rw_aio(lo, cmd, pos, READ);
496bc07c10aSMing Lei 		else
497bc07c10aSMing Lei 			return lo_read_simple(lo, rq, pos);
498c1c87c2bSChristoph Hellwig 	default:
499c1c87c2bSChristoph Hellwig 		WARN_ON_ONCE(1);
500c1c87c2bSChristoph Hellwig 		return -EIO;
501bc07c10aSMing Lei 	}
5021da177e4SLinus Torvalds }
5031da177e4SLinus Torvalds 
5042e5ab5f3SMing Lei static inline void loop_update_dio(struct loop_device *lo)
5052e5ab5f3SMing Lei {
506efbe3c24SIra Weiny 	__loop_update_dio(lo, (lo->lo_backing_file->f_flags & O_DIRECT) |
5072e5ab5f3SMing Lei 				lo->use_dio);
5082e5ab5f3SMing Lei }
5092e5ab5f3SMing Lei 
5100384264eSChristoph Hellwig static void loop_reread_partitions(struct loop_device *lo)
51106f0e9e6SMing Lei {
51206f0e9e6SMing Lei 	int rc;
51306f0e9e6SMing Lei 
5140384264eSChristoph Hellwig 	mutex_lock(&lo->lo_disk->open_mutex);
5150384264eSChristoph Hellwig 	rc = bdev_disk_changed(lo->lo_disk, false);
5160384264eSChristoph Hellwig 	mutex_unlock(&lo->lo_disk->open_mutex);
51706f0e9e6SMing Lei 	if (rc)
51806f0e9e6SMing Lei 		pr_warn("%s: partition scan of loop%d (%s) failed (rc=%d)\n",
51906f0e9e6SMing Lei 			__func__, lo->lo_number, lo->lo_file_name, rc);
52006f0e9e6SMing Lei }
52106f0e9e6SMing Lei 
522d2ac838eSTheodore Ts'o static inline int is_loop_device(struct file *file)
523d2ac838eSTheodore Ts'o {
524d2ac838eSTheodore Ts'o 	struct inode *i = file->f_mapping->host;
525d2ac838eSTheodore Ts'o 
5266f24784fSAl Viro 	return i && S_ISBLK(i->i_mode) && imajor(i) == LOOP_MAJOR;
527d2ac838eSTheodore Ts'o }
528d2ac838eSTheodore Ts'o 
529d2ac838eSTheodore Ts'o static int loop_validate_file(struct file *file, struct block_device *bdev)
530d2ac838eSTheodore Ts'o {
531d2ac838eSTheodore Ts'o 	struct inode	*inode = file->f_mapping->host;
532d2ac838eSTheodore Ts'o 	struct file	*f = file;
533d2ac838eSTheodore Ts'o 
534d2ac838eSTheodore Ts'o 	/* Avoid recursion */
535d2ac838eSTheodore Ts'o 	while (is_loop_device(f)) {
536d2ac838eSTheodore Ts'o 		struct loop_device *l;
537d2ac838eSTheodore Ts'o 
5383ce6e1f6STetsuo Handa 		lockdep_assert_held(&loop_validate_mutex);
5394e7b5671SChristoph Hellwig 		if (f->f_mapping->host->i_rdev == bdev->bd_dev)
540d2ac838eSTheodore Ts'o 			return -EBADF;
541d2ac838eSTheodore Ts'o 
5424e7b5671SChristoph Hellwig 		l = I_BDEV(f->f_mapping->host)->bd_disk->private_data;
5433ce6e1f6STetsuo Handa 		if (l->lo_state != Lo_bound)
544d2ac838eSTheodore Ts'o 			return -EINVAL;
5453ce6e1f6STetsuo Handa 		/* Order wrt setting lo->lo_backing_file in loop_configure(). */
5463ce6e1f6STetsuo Handa 		rmb();
547d2ac838eSTheodore Ts'o 		f = l->lo_backing_file;
548d2ac838eSTheodore Ts'o 	}
549d2ac838eSTheodore Ts'o 	if (!S_ISREG(inode->i_mode) && !S_ISBLK(inode->i_mode))
550d2ac838eSTheodore Ts'o 		return -EINVAL;
551d2ac838eSTheodore Ts'o 	return 0;
552d2ac838eSTheodore Ts'o }
553d2ac838eSTheodore Ts'o 
5541da177e4SLinus Torvalds /*
5551da177e4SLinus Torvalds  * loop_change_fd switched the backing store of a loopback device to
5561da177e4SLinus Torvalds  * a new file. This is useful for operating system installers to free up
5571da177e4SLinus Torvalds  * the original file and in High Availability environments to switch to
5581da177e4SLinus Torvalds  * an alternative location for the content in case of server meltdown.
5591da177e4SLinus Torvalds  * This can only work if the loop device is used read-only, and if the
5601da177e4SLinus Torvalds  * new backing store is the same size and type as the old backing store.
5611da177e4SLinus Torvalds  */
562bb214884SAl Viro static int loop_change_fd(struct loop_device *lo, struct block_device *bdev,
563bb214884SAl Viro 			  unsigned int arg)
5641da177e4SLinus Torvalds {
5653ce6e1f6STetsuo Handa 	struct file *file = fget(arg);
5663ce6e1f6STetsuo Handa 	struct file *old_file;
5671da177e4SLinus Torvalds 	int error;
56885b0a54aSJan Kara 	bool partscan;
5693ce6e1f6STetsuo Handa 	bool is_loop;
5701da177e4SLinus Torvalds 
5713ce6e1f6STetsuo Handa 	if (!file)
5723ce6e1f6STetsuo Handa 		return -EBADF;
5733ce6e1f6STetsuo Handa 	is_loop = is_loop_device(file);
5743ce6e1f6STetsuo Handa 	error = loop_global_lock_killable(lo, is_loop);
575c3710770SJan Kara 	if (error)
5763ce6e1f6STetsuo Handa 		goto out_putf;
5771da177e4SLinus Torvalds 	error = -ENXIO;
5781da177e4SLinus Torvalds 	if (lo->lo_state != Lo_bound)
5791dded9acSJan Kara 		goto out_err;
5801da177e4SLinus Torvalds 
5811da177e4SLinus Torvalds 	/* the loop device has to be read-only */
5821da177e4SLinus Torvalds 	error = -EINVAL;
5831da177e4SLinus Torvalds 	if (!(lo->lo_flags & LO_FLAGS_READ_ONLY))
5841dded9acSJan Kara 		goto out_err;
5851da177e4SLinus Torvalds 
586d2ac838eSTheodore Ts'o 	error = loop_validate_file(file, bdev);
587d2ac838eSTheodore Ts'o 	if (error)
5881dded9acSJan Kara 		goto out_err;
589d2ac838eSTheodore Ts'o 
5901da177e4SLinus Torvalds 	old_file = lo->lo_backing_file;
5911da177e4SLinus Torvalds 
5921da177e4SLinus Torvalds 	error = -EINVAL;
5931da177e4SLinus Torvalds 
5941da177e4SLinus Torvalds 	/* size of the new backing store needs to be the same */
5951da177e4SLinus Torvalds 	if (get_loop_size(lo, file) != get_loop_size(lo, old_file))
5961dded9acSJan Kara 		goto out_err;
5971da177e4SLinus Torvalds 
5981da177e4SLinus Torvalds 	/* and ... switch */
5999f65c489SMatteo Croce 	disk_force_media_change(lo->lo_disk, DISK_EVENT_MEDIA_CHANGE);
60043cade80SOmar Sandoval 	blk_mq_freeze_queue(lo->lo_queue);
60143cade80SOmar Sandoval 	mapping_set_gfp_mask(old_file->f_mapping, lo->old_gfp_mask);
60243cade80SOmar Sandoval 	lo->lo_backing_file = file;
60343cade80SOmar Sandoval 	lo->old_gfp_mask = mapping_gfp_mask(file->f_mapping);
60443cade80SOmar Sandoval 	mapping_set_gfp_mask(file->f_mapping,
60543cade80SOmar Sandoval 			     lo->old_gfp_mask & ~(__GFP_IO|__GFP_FS));
60643cade80SOmar Sandoval 	loop_update_dio(lo);
60743cade80SOmar Sandoval 	blk_mq_unfreeze_queue(lo->lo_queue);
60885b0a54aSJan Kara 	partscan = lo->lo_flags & LO_FLAGS_PARTSCAN;
6093ce6e1f6STetsuo Handa 	loop_global_unlock(lo, is_loop);
6103ce6e1f6STetsuo Handa 
6113ce6e1f6STetsuo Handa 	/*
6123ce6e1f6STetsuo Handa 	 * Flush loop_validate_file() before fput(), for l->lo_backing_file
6133ce6e1f6STetsuo Handa 	 * might be pointing at old_file which might be the last reference.
6143ce6e1f6STetsuo Handa 	 */
6153ce6e1f6STetsuo Handa 	if (!is_loop) {
6163ce6e1f6STetsuo Handa 		mutex_lock(&loop_validate_mutex);
6173ce6e1f6STetsuo Handa 		mutex_unlock(&loop_validate_mutex);
6183ce6e1f6STetsuo Handa 	}
6191dded9acSJan Kara 	/*
6206cc8e743SPavel Tatashin 	 * We must drop file reference outside of lo_mutex as dropping
621a8698707SChristoph Hellwig 	 * the file ref can take open_mutex which creates circular locking
6221dded9acSJan Kara 	 * dependency.
6231dded9acSJan Kara 	 */
6241dded9acSJan Kara 	fput(old_file);
62585b0a54aSJan Kara 	if (partscan)
6260384264eSChristoph Hellwig 		loop_reread_partitions(lo);
6271da177e4SLinus Torvalds 	return 0;
6281da177e4SLinus Torvalds 
6291dded9acSJan Kara out_err:
6303ce6e1f6STetsuo Handa 	loop_global_unlock(lo, is_loop);
6313ce6e1f6STetsuo Handa out_putf:
6321dded9acSJan Kara 	fput(file);
6331da177e4SLinus Torvalds 	return error;
6341da177e4SLinus Torvalds }
6351da177e4SLinus Torvalds 
636ee862730SMilan Broz /* loop sysfs attributes */
637ee862730SMilan Broz 
638ee862730SMilan Broz static ssize_t loop_attr_show(struct device *dev, char *page,
639ee862730SMilan Broz 			      ssize_t (*callback)(struct loop_device *, char *))
640ee862730SMilan Broz {
64134dd82afSKay Sievers 	struct gendisk *disk = dev_to_disk(dev);
64234dd82afSKay Sievers 	struct loop_device *lo = disk->private_data;
643ee862730SMilan Broz 
64434dd82afSKay Sievers 	return callback(lo, page);
645ee862730SMilan Broz }
646ee862730SMilan Broz 
647ee862730SMilan Broz #define LOOP_ATTR_RO(_name)						\
648ee862730SMilan Broz static ssize_t loop_attr_##_name##_show(struct loop_device *, char *);	\
649ee862730SMilan Broz static ssize_t loop_attr_do_show_##_name(struct device *d,		\
650ee862730SMilan Broz 				struct device_attribute *attr, char *b)	\
651ee862730SMilan Broz {									\
652ee862730SMilan Broz 	return loop_attr_show(d, b, loop_attr_##_name##_show);		\
653ee862730SMilan Broz }									\
654ee862730SMilan Broz static struct device_attribute loop_attr_##_name =			\
6555657a819SJoe Perches 	__ATTR(_name, 0444, loop_attr_do_show_##_name, NULL);
656ee862730SMilan Broz 
657ee862730SMilan Broz static ssize_t loop_attr_backing_file_show(struct loop_device *lo, char *buf)
658ee862730SMilan Broz {
659ee862730SMilan Broz 	ssize_t ret;
660ee862730SMilan Broz 	char *p = NULL;
661ee862730SMilan Broz 
66205eb0f25SKay Sievers 	spin_lock_irq(&lo->lo_lock);
663ee862730SMilan Broz 	if (lo->lo_backing_file)
6649bf39ab2SMiklos Szeredi 		p = file_path(lo->lo_backing_file, buf, PAGE_SIZE - 1);
66505eb0f25SKay Sievers 	spin_unlock_irq(&lo->lo_lock);
666ee862730SMilan Broz 
667ee862730SMilan Broz 	if (IS_ERR_OR_NULL(p))
668ee862730SMilan Broz 		ret = PTR_ERR(p);
669ee862730SMilan Broz 	else {
670ee862730SMilan Broz 		ret = strlen(p);
671ee862730SMilan Broz 		memmove(buf, p, ret);
672ee862730SMilan Broz 		buf[ret++] = '\n';
673ee862730SMilan Broz 		buf[ret] = 0;
674ee862730SMilan Broz 	}
675ee862730SMilan Broz 
676ee862730SMilan Broz 	return ret;
677ee862730SMilan Broz }
678ee862730SMilan Broz 
679ee862730SMilan Broz static ssize_t loop_attr_offset_show(struct loop_device *lo, char *buf)
680ee862730SMilan Broz {
681b27824d3SChaitanya Kulkarni 	return sysfs_emit(buf, "%llu\n", (unsigned long long)lo->lo_offset);
682ee862730SMilan Broz }
683ee862730SMilan Broz 
684ee862730SMilan Broz static ssize_t loop_attr_sizelimit_show(struct loop_device *lo, char *buf)
685ee862730SMilan Broz {
686b27824d3SChaitanya Kulkarni 	return sysfs_emit(buf, "%llu\n", (unsigned long long)lo->lo_sizelimit);
687ee862730SMilan Broz }
688ee862730SMilan Broz 
689ee862730SMilan Broz static ssize_t loop_attr_autoclear_show(struct loop_device *lo, char *buf)
690ee862730SMilan Broz {
691ee862730SMilan Broz 	int autoclear = (lo->lo_flags & LO_FLAGS_AUTOCLEAR);
692ee862730SMilan Broz 
693b27824d3SChaitanya Kulkarni 	return sysfs_emit(buf, "%s\n", autoclear ? "1" : "0");
694ee862730SMilan Broz }
695ee862730SMilan Broz 
696e03c8dd1SKay Sievers static ssize_t loop_attr_partscan_show(struct loop_device *lo, char *buf)
697e03c8dd1SKay Sievers {
698e03c8dd1SKay Sievers 	int partscan = (lo->lo_flags & LO_FLAGS_PARTSCAN);
699e03c8dd1SKay Sievers 
700b27824d3SChaitanya Kulkarni 	return sysfs_emit(buf, "%s\n", partscan ? "1" : "0");
701e03c8dd1SKay Sievers }
702e03c8dd1SKay Sievers 
7032e5ab5f3SMing Lei static ssize_t loop_attr_dio_show(struct loop_device *lo, char *buf)
7042e5ab5f3SMing Lei {
7052e5ab5f3SMing Lei 	int dio = (lo->lo_flags & LO_FLAGS_DIRECT_IO);
7062e5ab5f3SMing Lei 
707b27824d3SChaitanya Kulkarni 	return sysfs_emit(buf, "%s\n", dio ? "1" : "0");
7082e5ab5f3SMing Lei }
7092e5ab5f3SMing Lei 
710ee862730SMilan Broz LOOP_ATTR_RO(backing_file);
711ee862730SMilan Broz LOOP_ATTR_RO(offset);
712ee862730SMilan Broz LOOP_ATTR_RO(sizelimit);
713ee862730SMilan Broz LOOP_ATTR_RO(autoclear);
714e03c8dd1SKay Sievers LOOP_ATTR_RO(partscan);
7152e5ab5f3SMing Lei LOOP_ATTR_RO(dio);
716ee862730SMilan Broz 
717ee862730SMilan Broz static struct attribute *loop_attrs[] = {
718ee862730SMilan Broz 	&loop_attr_backing_file.attr,
719ee862730SMilan Broz 	&loop_attr_offset.attr,
720ee862730SMilan Broz 	&loop_attr_sizelimit.attr,
721ee862730SMilan Broz 	&loop_attr_autoclear.attr,
722e03c8dd1SKay Sievers 	&loop_attr_partscan.attr,
7232e5ab5f3SMing Lei 	&loop_attr_dio.attr,
724ee862730SMilan Broz 	NULL,
725ee862730SMilan Broz };
726ee862730SMilan Broz 
727ee862730SMilan Broz static struct attribute_group loop_attribute_group = {
728ee862730SMilan Broz 	.name = "loop",
729ee862730SMilan Broz 	.attrs= loop_attrs,
730ee862730SMilan Broz };
731ee862730SMilan Broz 
732d3349b6bSTetsuo Handa static void loop_sysfs_init(struct loop_device *lo)
733ee862730SMilan Broz {
734d3349b6bSTetsuo Handa 	lo->sysfs_inited = !sysfs_create_group(&disk_to_dev(lo->lo_disk)->kobj,
735ee862730SMilan Broz 						&loop_attribute_group);
736ee862730SMilan Broz }
737ee862730SMilan Broz 
738ee862730SMilan Broz static void loop_sysfs_exit(struct loop_device *lo)
739ee862730SMilan Broz {
740d3349b6bSTetsuo Handa 	if (lo->sysfs_inited)
741ee862730SMilan Broz 		sysfs_remove_group(&disk_to_dev(lo->lo_disk)->kobj,
742ee862730SMilan Broz 				   &loop_attribute_group);
743ee862730SMilan Broz }
744ee862730SMilan Broz 
745dfaa2ef6SLukas Czerner static void loop_config_discard(struct loop_device *lo)
746dfaa2ef6SLukas Czerner {
747dfaa2ef6SLukas Czerner 	struct file *file = lo->lo_backing_file;
748dfaa2ef6SLukas Czerner 	struct inode *inode = file->f_mapping->host;
749dfaa2ef6SLukas Czerner 	struct request_queue *q = lo->lo_queue;
750bcb21c8cSMing Lei 	u32 granularity, max_discard_sectors;
751dfaa2ef6SLukas Czerner 
752dfaa2ef6SLukas Czerner 	/*
753c52abf56SEvan Green 	 * If the backing device is a block device, mirror its zeroing
754c52abf56SEvan Green 	 * capability. Set the discard sectors to the block device's zeroing
755c52abf56SEvan Green 	 * capabilities because loop discards result in blkdev_issue_zeroout(),
756c52abf56SEvan Green 	 * not blkdev_issue_discard(). This maintains consistent behavior with
757c52abf56SEvan Green 	 * file-backed loop devices: discarded regions read back as zero.
758c52abf56SEvan Green 	 */
75947e96246SChristoph Hellwig 	if (S_ISBLK(inode->i_mode)) {
7604e7b5671SChristoph Hellwig 		struct request_queue *backingq = bdev_get_queue(I_BDEV(inode));
761c52abf56SEvan Green 
762bcb21c8cSMing Lei 		max_discard_sectors = backingq->limits.max_write_zeroes_sectors;
763bcb21c8cSMing Lei 		granularity = backingq->limits.discard_granularity ?:
764bcb21c8cSMing Lei 			queue_physical_block_size(backingq);
765c52abf56SEvan Green 
766c52abf56SEvan Green 	/*
767dfaa2ef6SLukas Czerner 	 * We use punch hole to reclaim the free space used by the
76847e96246SChristoph Hellwig 	 * image a.k.a. discard.
769dfaa2ef6SLukas Czerner 	 */
77047e96246SChristoph Hellwig 	} else if (!file->f_op->fallocate) {
771bcb21c8cSMing Lei 		max_discard_sectors = 0;
772bcb21c8cSMing Lei 		granularity = 0;
773dfaa2ef6SLukas Czerner 
774c52abf56SEvan Green 	} else {
775bcb21c8cSMing Lei 		max_discard_sectors = UINT_MAX >> 9;
776bcb21c8cSMing Lei 		granularity = inode->i_sb->s_blocksize;
777c52abf56SEvan Green 	}
778c52abf56SEvan Green 
779bcb21c8cSMing Lei 	if (max_discard_sectors) {
780bcb21c8cSMing Lei 		q->limits.discard_granularity = granularity;
781bcb21c8cSMing Lei 		blk_queue_max_discard_sectors(q, max_discard_sectors);
782bcb21c8cSMing Lei 		blk_queue_max_write_zeroes_sectors(q, max_discard_sectors);
7838b904b5bSBart Van Assche 		blk_queue_flag_set(QUEUE_FLAG_DISCARD, q);
784bcb21c8cSMing Lei 	} else {
785bcb21c8cSMing Lei 		q->limits.discard_granularity = 0;
786bcb21c8cSMing Lei 		blk_queue_max_discard_sectors(q, 0);
787bcb21c8cSMing Lei 		blk_queue_max_write_zeroes_sectors(q, 0);
788c52abf56SEvan Green 		blk_queue_flag_clear(QUEUE_FLAG_DISCARD, q);
789dfaa2ef6SLukas Czerner 	}
790bcb21c8cSMing Lei 	q->limits.discard_alignment = 0;
791bcb21c8cSMing Lei }
792dfaa2ef6SLukas Czerner 
79387579e9bSDan Schatzberg struct loop_worker {
79487579e9bSDan Schatzberg 	struct rb_node rb_node;
79587579e9bSDan Schatzberg 	struct work_struct work;
79687579e9bSDan Schatzberg 	struct list_head cmd_list;
79787579e9bSDan Schatzberg 	struct list_head idle_list;
79887579e9bSDan Schatzberg 	struct loop_device *lo;
799c74d40e8SDan Schatzberg 	struct cgroup_subsys_state *blkcg_css;
80087579e9bSDan Schatzberg 	unsigned long last_ran_at;
80187579e9bSDan Schatzberg };
802e03a3d7aSMing Lei 
80387579e9bSDan Schatzberg static void loop_workfn(struct work_struct *work);
80487579e9bSDan Schatzberg static void loop_rootcg_workfn(struct work_struct *work);
80587579e9bSDan Schatzberg static void loop_free_idle_workers(struct timer_list *timer);
806b2ee7d46SNeilBrown 
80787579e9bSDan Schatzberg #ifdef CONFIG_BLK_CGROUP
80887579e9bSDan Schatzberg static inline int queue_on_root_worker(struct cgroup_subsys_state *css)
809e03a3d7aSMing Lei {
81087579e9bSDan Schatzberg 	return !css || css == blkcg_root_css;
81187579e9bSDan Schatzberg }
81287579e9bSDan Schatzberg #else
81387579e9bSDan Schatzberg static inline int queue_on_root_worker(struct cgroup_subsys_state *css)
81487579e9bSDan Schatzberg {
81587579e9bSDan Schatzberg 	return !css;
81687579e9bSDan Schatzberg }
81787579e9bSDan Schatzberg #endif
81887579e9bSDan Schatzberg 
81987579e9bSDan Schatzberg static void loop_queue_work(struct loop_device *lo, struct loop_cmd *cmd)
82087579e9bSDan Schatzberg {
821413ec805SColin Ian King 	struct rb_node **node, *parent = NULL;
82287579e9bSDan Schatzberg 	struct loop_worker *cur_worker, *worker = NULL;
82387579e9bSDan Schatzberg 	struct work_struct *work;
82487579e9bSDan Schatzberg 	struct list_head *cmd_list;
82587579e9bSDan Schatzberg 
82687579e9bSDan Schatzberg 	spin_lock_irq(&lo->lo_work_lock);
82787579e9bSDan Schatzberg 
828c74d40e8SDan Schatzberg 	if (queue_on_root_worker(cmd->blkcg_css))
82987579e9bSDan Schatzberg 		goto queue_work;
83087579e9bSDan Schatzberg 
83187579e9bSDan Schatzberg 	node = &lo->worker_tree.rb_node;
83287579e9bSDan Schatzberg 
83387579e9bSDan Schatzberg 	while (*node) {
83487579e9bSDan Schatzberg 		parent = *node;
83587579e9bSDan Schatzberg 		cur_worker = container_of(*node, struct loop_worker, rb_node);
836c74d40e8SDan Schatzberg 		if (cur_worker->blkcg_css == cmd->blkcg_css) {
83787579e9bSDan Schatzberg 			worker = cur_worker;
83887579e9bSDan Schatzberg 			break;
839c74d40e8SDan Schatzberg 		} else if ((long)cur_worker->blkcg_css < (long)cmd->blkcg_css) {
84087579e9bSDan Schatzberg 			node = &(*node)->rb_left;
84187579e9bSDan Schatzberg 		} else {
84287579e9bSDan Schatzberg 			node = &(*node)->rb_right;
84387579e9bSDan Schatzberg 		}
84487579e9bSDan Schatzberg 	}
84587579e9bSDan Schatzberg 	if (worker)
84687579e9bSDan Schatzberg 		goto queue_work;
84787579e9bSDan Schatzberg 
84887579e9bSDan Schatzberg 	worker = kzalloc(sizeof(struct loop_worker), GFP_NOWAIT | __GFP_NOWARN);
84987579e9bSDan Schatzberg 	/*
85087579e9bSDan Schatzberg 	 * In the event we cannot allocate a worker, just queue on the
851c74d40e8SDan Schatzberg 	 * rootcg worker and issue the I/O as the rootcg
85287579e9bSDan Schatzberg 	 */
853c74d40e8SDan Schatzberg 	if (!worker) {
854c74d40e8SDan Schatzberg 		cmd->blkcg_css = NULL;
855c74d40e8SDan Schatzberg 		if (cmd->memcg_css)
856c74d40e8SDan Schatzberg 			css_put(cmd->memcg_css);
857c74d40e8SDan Schatzberg 		cmd->memcg_css = NULL;
85887579e9bSDan Schatzberg 		goto queue_work;
859c74d40e8SDan Schatzberg 	}
86087579e9bSDan Schatzberg 
861c74d40e8SDan Schatzberg 	worker->blkcg_css = cmd->blkcg_css;
862c74d40e8SDan Schatzberg 	css_get(worker->blkcg_css);
86387579e9bSDan Schatzberg 	INIT_WORK(&worker->work, loop_workfn);
86487579e9bSDan Schatzberg 	INIT_LIST_HEAD(&worker->cmd_list);
86587579e9bSDan Schatzberg 	INIT_LIST_HEAD(&worker->idle_list);
86687579e9bSDan Schatzberg 	worker->lo = lo;
86787579e9bSDan Schatzberg 	rb_link_node(&worker->rb_node, parent, node);
86887579e9bSDan Schatzberg 	rb_insert_color(&worker->rb_node, &lo->worker_tree);
86987579e9bSDan Schatzberg queue_work:
87087579e9bSDan Schatzberg 	if (worker) {
87187579e9bSDan Schatzberg 		/*
87287579e9bSDan Schatzberg 		 * We need to remove from the idle list here while
87387579e9bSDan Schatzberg 		 * holding the lock so that the idle timer doesn't
87487579e9bSDan Schatzberg 		 * free the worker
87587579e9bSDan Schatzberg 		 */
87687579e9bSDan Schatzberg 		if (!list_empty(&worker->idle_list))
87787579e9bSDan Schatzberg 			list_del_init(&worker->idle_list);
87887579e9bSDan Schatzberg 		work = &worker->work;
87987579e9bSDan Schatzberg 		cmd_list = &worker->cmd_list;
88087579e9bSDan Schatzberg 	} else {
88187579e9bSDan Schatzberg 		work = &lo->rootcg_work;
88287579e9bSDan Schatzberg 		cmd_list = &lo->rootcg_cmd_list;
88387579e9bSDan Schatzberg 	}
88487579e9bSDan Schatzberg 	list_add_tail(&cmd->list_entry, cmd_list);
88587579e9bSDan Schatzberg 	queue_work(lo->workqueue, work);
88687579e9bSDan Schatzberg 	spin_unlock_irq(&lo->lo_work_lock);
887e03a3d7aSMing Lei }
888e03a3d7aSMing Lei 
88956a85fd8SHolger Hoffstätte static void loop_update_rotational(struct loop_device *lo)
89056a85fd8SHolger Hoffstätte {
89156a85fd8SHolger Hoffstätte 	struct file *file = lo->lo_backing_file;
89256a85fd8SHolger Hoffstätte 	struct inode *file_inode = file->f_mapping->host;
89356a85fd8SHolger Hoffstätte 	struct block_device *file_bdev = file_inode->i_sb->s_bdev;
89456a85fd8SHolger Hoffstätte 	struct request_queue *q = lo->lo_queue;
89556a85fd8SHolger Hoffstätte 	bool nonrot = true;
89656a85fd8SHolger Hoffstätte 
89756a85fd8SHolger Hoffstätte 	/* not all filesystems (e.g. tmpfs) have a sb->s_bdev */
89856a85fd8SHolger Hoffstätte 	if (file_bdev)
89956a85fd8SHolger Hoffstätte 		nonrot = blk_queue_nonrot(bdev_get_queue(file_bdev));
90056a85fd8SHolger Hoffstätte 
90156a85fd8SHolger Hoffstätte 	if (nonrot)
90256a85fd8SHolger Hoffstätte 		blk_queue_flag_set(QUEUE_FLAG_NONROT, q);
90356a85fd8SHolger Hoffstätte 	else
90456a85fd8SHolger Hoffstätte 		blk_queue_flag_clear(QUEUE_FLAG_NONROT, q);
90556a85fd8SHolger Hoffstätte }
90656a85fd8SHolger Hoffstätte 
90762ab466cSMartijn Coenen /**
90862ab466cSMartijn Coenen  * loop_set_status_from_info - configure device from loop_info
90962ab466cSMartijn Coenen  * @lo: struct loop_device to configure
91062ab466cSMartijn Coenen  * @info: struct loop_info64 to configure the device with
91162ab466cSMartijn Coenen  *
91262ab466cSMartijn Coenen  * Configures the loop device parameters according to the passed
91362ab466cSMartijn Coenen  * in loop_info64 configuration.
91462ab466cSMartijn Coenen  */
91562ab466cSMartijn Coenen static int
91662ab466cSMartijn Coenen loop_set_status_from_info(struct loop_device *lo,
91762ab466cSMartijn Coenen 			  const struct loop_info64 *info)
91862ab466cSMartijn Coenen {
91962ab466cSMartijn Coenen 	if ((unsigned int) info->lo_encrypt_key_size > LO_KEY_SIZE)
92062ab466cSMartijn Coenen 		return -EINVAL;
92162ab466cSMartijn Coenen 
92247e96246SChristoph Hellwig 	switch (info->lo_encrypt_type) {
92347e96246SChristoph Hellwig 	case LO_CRYPT_NONE:
92447e96246SChristoph Hellwig 		break;
92547e96246SChristoph Hellwig 	case LO_CRYPT_XOR:
92647e96246SChristoph Hellwig 		pr_warn("support for the xor transformation has been removed.\n");
92762ab466cSMartijn Coenen 		return -EINVAL;
92847e96246SChristoph Hellwig 	case LO_CRYPT_CRYPTOAPI:
92947e96246SChristoph Hellwig 		pr_warn("support for cryptoloop has been removed.  Use dm-crypt instead.\n");
93062ab466cSMartijn Coenen 		return -EINVAL;
93147e96246SChristoph Hellwig 	default:
93247e96246SChristoph Hellwig 		return -EINVAL;
93347e96246SChristoph Hellwig 	}
93462ab466cSMartijn Coenen 
93562ab466cSMartijn Coenen 	lo->lo_offset = info->lo_offset;
93662ab466cSMartijn Coenen 	lo->lo_sizelimit = info->lo_sizelimit;
93762ab466cSMartijn Coenen 	memcpy(lo->lo_file_name, info->lo_file_name, LO_NAME_SIZE);
93862ab466cSMartijn Coenen 	lo->lo_file_name[LO_NAME_SIZE-1] = 0;
939faf1d254SMartijn Coenen 	lo->lo_flags = info->lo_flags;
94062ab466cSMartijn Coenen 	return 0;
94162ab466cSMartijn Coenen }
94262ab466cSMartijn Coenen 
9433448914eSMartijn Coenen static int loop_configure(struct loop_device *lo, fmode_t mode,
9443448914eSMartijn Coenen 			  struct block_device *bdev,
9453448914eSMartijn Coenen 			  const struct loop_config *config)
9461da177e4SLinus Torvalds {
9473ce6e1f6STetsuo Handa 	struct file *file = fget(config->fd);
9481da177e4SLinus Torvalds 	struct inode *inode;
9491da177e4SLinus Torvalds 	struct address_space *mapping;
9501da177e4SLinus Torvalds 	int error;
9511da177e4SLinus Torvalds 	loff_t size;
95285b0a54aSJan Kara 	bool partscan;
9533448914eSMartijn Coenen 	unsigned short bsize;
9543ce6e1f6STetsuo Handa 	bool is_loop;
9553ce6e1f6STetsuo Handa 
9563ce6e1f6STetsuo Handa 	if (!file)
9573ce6e1f6STetsuo Handa 		return -EBADF;
9583ce6e1f6STetsuo Handa 	is_loop = is_loop_device(file);
9591da177e4SLinus Torvalds 
9601da177e4SLinus Torvalds 	/* This is safe, since we have a reference from open(). */
9611da177e4SLinus Torvalds 	__module_get(THIS_MODULE);
9621da177e4SLinus Torvalds 
96333ec3e53SJan Kara 	/*
96433ec3e53SJan Kara 	 * If we don't hold exclusive handle for the device, upgrade to it
96533ec3e53SJan Kara 	 * here to avoid changing device under exclusive owner.
96633ec3e53SJan Kara 	 */
96733ec3e53SJan Kara 	if (!(mode & FMODE_EXCL)) {
96837c3fc9aSChristoph Hellwig 		error = bd_prepare_to_claim(bdev, loop_configure);
969ecbe6bc0SChristoph Hellwig 		if (error)
970757ecf40SJan Kara 			goto out_putf;
97133ec3e53SJan Kara 	}
97233ec3e53SJan Kara 
9733ce6e1f6STetsuo Handa 	error = loop_global_lock_killable(lo, is_loop);
97433ec3e53SJan Kara 	if (error)
97533ec3e53SJan Kara 		goto out_bdev;
976757ecf40SJan Kara 
9771da177e4SLinus Torvalds 	error = -EBUSY;
9781da177e4SLinus Torvalds 	if (lo->lo_state != Lo_unbound)
979757ecf40SJan Kara 		goto out_unlock;
9801da177e4SLinus Torvalds 
981d2ac838eSTheodore Ts'o 	error = loop_validate_file(file, bdev);
982d2ac838eSTheodore Ts'o 	if (error)
983757ecf40SJan Kara 		goto out_unlock;
9841da177e4SLinus Torvalds 
9851da177e4SLinus Torvalds 	mapping = file->f_mapping;
9861da177e4SLinus Torvalds 	inode = mapping->host;
9871da177e4SLinus Torvalds 
9883448914eSMartijn Coenen 	if ((config->info.lo_flags & ~LOOP_CONFIGURE_SETTABLE_FLAGS) != 0) {
9893448914eSMartijn Coenen 		error = -EINVAL;
9903448914eSMartijn Coenen 		goto out_unlock;
9913448914eSMartijn Coenen 	}
9923448914eSMartijn Coenen 
9933448914eSMartijn Coenen 	if (config->block_size) {
994af3c570fSXie Yongji 		error = blk_validate_block_size(config->block_size);
9953448914eSMartijn Coenen 		if (error)
9963448914eSMartijn Coenen 			goto out_unlock;
9973448914eSMartijn Coenen 	}
9983448914eSMartijn Coenen 
9993448914eSMartijn Coenen 	error = loop_set_status_from_info(lo, &config->info);
10003448914eSMartijn Coenen 	if (error)
10013448914eSMartijn Coenen 		goto out_unlock;
10023448914eSMartijn Coenen 
1003456be148SChristoph Hellwig 	if (!(file->f_mode & FMODE_WRITE) || !(mode & FMODE_WRITE) ||
1004283e7e5dSAl Viro 	    !file->f_op->write_iter)
10053448914eSMartijn Coenen 		lo->lo_flags |= LO_FLAGS_READ_ONLY;
1006083a6a50SMartijn Coenen 
100787579e9bSDan Schatzberg 	lo->workqueue = alloc_workqueue("loop%d",
100887579e9bSDan Schatzberg 					WQ_UNBOUND | WQ_FREEZABLE,
100987579e9bSDan Schatzberg 					0,
101087579e9bSDan Schatzberg 					lo->lo_number);
101187579e9bSDan Schatzberg 	if (!lo->workqueue) {
101287579e9bSDan Schatzberg 		error = -ENOMEM;
1013757ecf40SJan Kara 		goto out_unlock;
101487579e9bSDan Schatzberg 	}
10151da177e4SLinus Torvalds 
10169f65c489SMatteo Croce 	disk_force_media_change(lo->lo_disk, DISK_EVENT_MEDIA_CHANGE);
10177a2f0ce1SChristoph Hellwig 	set_disk_ro(lo->lo_disk, (lo->lo_flags & LO_FLAGS_READ_ONLY) != 0);
10181da177e4SLinus Torvalds 
101987579e9bSDan Schatzberg 	INIT_WORK(&lo->rootcg_work, loop_rootcg_workfn);
102087579e9bSDan Schatzberg 	INIT_LIST_HEAD(&lo->rootcg_cmd_list);
102187579e9bSDan Schatzberg 	INIT_LIST_HEAD(&lo->idle_worker_list);
102287579e9bSDan Schatzberg 	lo->worker_tree = RB_ROOT;
102387579e9bSDan Schatzberg 	timer_setup(&lo->timer, loop_free_idle_workers,
102487579e9bSDan Schatzberg 		TIMER_DEFERRABLE);
10253448914eSMartijn Coenen 	lo->use_dio = lo->lo_flags & LO_FLAGS_DIRECT_IO;
10261da177e4SLinus Torvalds 	lo->lo_device = bdev;
10271da177e4SLinus Torvalds 	lo->lo_backing_file = file;
10281da177e4SLinus Torvalds 	lo->old_gfp_mask = mapping_gfp_mask(mapping);
10291da177e4SLinus Torvalds 	mapping_set_gfp_mask(mapping, lo->old_gfp_mask & ~(__GFP_IO|__GFP_FS));
10301da177e4SLinus Torvalds 
10313448914eSMartijn Coenen 	if (!(lo->lo_flags & LO_FLAGS_READ_ONLY) && file->f_op->fsync)
103221d0727fSJens Axboe 		blk_queue_write_cache(lo->lo_queue, true, false);
103368db1961SNikanth Karthikesan 
10343448914eSMartijn Coenen 	if (config->block_size)
10353448914eSMartijn Coenen 		bsize = config->block_size;
103696ed320dSLinus Torvalds 	else if ((lo->lo_backing_file->f_flags & O_DIRECT) && inode->i_sb->s_bdev)
103785560117SMartijn Coenen 		/* In case of direct I/O, match underlying block size */
10383448914eSMartijn Coenen 		bsize = bdev_logical_block_size(inode->i_sb->s_bdev);
10393448914eSMartijn Coenen 	else
10403448914eSMartijn Coenen 		bsize = 512;
104185560117SMartijn Coenen 
104285560117SMartijn Coenen 	blk_queue_logical_block_size(lo->lo_queue, bsize);
104385560117SMartijn Coenen 	blk_queue_physical_block_size(lo->lo_queue, bsize);
104485560117SMartijn Coenen 	blk_queue_io_min(lo->lo_queue, bsize);
104585560117SMartijn Coenen 
10462b9ac22bSKristian Klausen 	loop_config_discard(lo);
104756a85fd8SHolger Hoffstätte 	loop_update_rotational(lo);
10482e5ab5f3SMing Lei 	loop_update_dio(lo);
1049ee862730SMilan Broz 	loop_sysfs_init(lo);
105079e5dc59SMartijn Coenen 
105179e5dc59SMartijn Coenen 	size = get_loop_size(lo, file);
10525795b6f5SMartijn Coenen 	loop_set_size(lo, size);
10531da177e4SLinus Torvalds 
10543ce6e1f6STetsuo Handa 	/* Order wrt reading lo_state in loop_validate_file(). */
10553ce6e1f6STetsuo Handa 	wmb();
10563ce6e1f6STetsuo Handa 
10576c997918SSerge E. Hallyn 	lo->lo_state = Lo_bound;
1058e03c8dd1SKay Sievers 	if (part_shift)
1059e03c8dd1SKay Sievers 		lo->lo_flags |= LO_FLAGS_PARTSCAN;
106085b0a54aSJan Kara 	partscan = lo->lo_flags & LO_FLAGS_PARTSCAN;
1061fe6a8fc5SLennart Poettering 	if (partscan)
106246e7eac6SChristoph Hellwig 		lo->lo_disk->flags &= ~GENHD_FL_NO_PART;
1063c1681bf8SAnatol Pomozov 
10643ce6e1f6STetsuo Handa 	loop_global_unlock(lo, is_loop);
106585b0a54aSJan Kara 	if (partscan)
10660384264eSChristoph Hellwig 		loop_reread_partitions(lo);
106737c3fc9aSChristoph Hellwig 	if (!(mode & FMODE_EXCL))
106837c3fc9aSChristoph Hellwig 		bd_abort_claiming(bdev, loop_configure);
10691da177e4SLinus Torvalds 	return 0;
10701da177e4SLinus Torvalds 
1071757ecf40SJan Kara out_unlock:
10723ce6e1f6STetsuo Handa 	loop_global_unlock(lo, is_loop);
107333ec3e53SJan Kara out_bdev:
107437c3fc9aSChristoph Hellwig 	if (!(mode & FMODE_EXCL))
107537c3fc9aSChristoph Hellwig 		bd_abort_claiming(bdev, loop_configure);
10761da177e4SLinus Torvalds out_putf:
10771da177e4SLinus Torvalds 	fput(file);
10781da177e4SLinus Torvalds 	/* This is safe: open() is still holding a reference. */
10791da177e4SLinus Torvalds 	module_put(THIS_MODULE);
10801da177e4SLinus Torvalds 	return error;
10811da177e4SLinus Torvalds }
10821da177e4SLinus Torvalds 
1083322c4293STetsuo Handa static void __loop_clr_fd(struct loop_device *lo)
10841da177e4SLinus Torvalds {
10856050fa4cSTetsuo Handa 	struct file *filp;
1086b4e3ca1aSAl Viro 	gfp_t gfp = lo->old_gfp_mask;
108787579e9bSDan Schatzberg 	struct loop_worker *pos, *worker;
10881da177e4SLinus Torvalds 
10893ce6e1f6STetsuo Handa 	/*
10903ce6e1f6STetsuo Handa 	 * Flush loop_configure() and loop_change_fd(). It is acceptable for
10913ce6e1f6STetsuo Handa 	 * loop_validate_file() to succeed, for actual clear operation has not
10923ce6e1f6STetsuo Handa 	 * started yet.
10933ce6e1f6STetsuo Handa 	 */
10943ce6e1f6STetsuo Handa 	mutex_lock(&loop_validate_mutex);
10953ce6e1f6STetsuo Handa 	mutex_unlock(&loop_validate_mutex);
10963ce6e1f6STetsuo Handa 	/*
10973ce6e1f6STetsuo Handa 	 * loop_validate_file() now fails because l->lo_state != Lo_bound
10983ce6e1f6STetsuo Handa 	 * became visible.
10993ce6e1f6STetsuo Handa 	 */
11003ce6e1f6STetsuo Handa 
11016050fa4cSTetsuo Handa 	/*
11026050fa4cSTetsuo Handa 	 * Since this function is called upon "ioctl(LOOP_CLR_FD)" xor "close()
11036050fa4cSTetsuo Handa 	 * after ioctl(LOOP_CLR_FD)", it is a sign of something going wrong if
11046050fa4cSTetsuo Handa 	 * lo->lo_state has changed while waiting for lo->lo_mutex.
11056050fa4cSTetsuo Handa 	 */
11066cc8e743SPavel Tatashin 	mutex_lock(&lo->lo_mutex);
11076050fa4cSTetsuo Handa 	BUG_ON(lo->lo_state != Lo_rundown);
11086050fa4cSTetsuo Handa 	mutex_unlock(&lo->lo_mutex);
11091da177e4SLinus Torvalds 
11104ceddce5SMauricio Faria de Oliveira 	if (test_bit(QUEUE_FLAG_WC, &lo->lo_queue->queue_flags))
11114ceddce5SMauricio Faria de Oliveira 		blk_queue_write_cache(lo->lo_queue, false, false);
11124ceddce5SMauricio Faria de Oliveira 
1113f8933667SMing Lei 	/* freeze request queue during the transition */
1114f8933667SMing Lei 	blk_mq_freeze_queue(lo->lo_queue);
1115f8933667SMing Lei 
111687579e9bSDan Schatzberg 	destroy_workqueue(lo->workqueue);
111787579e9bSDan Schatzberg 	spin_lock_irq(&lo->lo_work_lock);
111887579e9bSDan Schatzberg 	list_for_each_entry_safe(worker, pos, &lo->idle_worker_list,
111987579e9bSDan Schatzberg 				idle_list) {
112087579e9bSDan Schatzberg 		list_del(&worker->idle_list);
112187579e9bSDan Schatzberg 		rb_erase(&worker->rb_node, &lo->worker_tree);
1122c74d40e8SDan Schatzberg 		css_put(worker->blkcg_css);
112387579e9bSDan Schatzberg 		kfree(worker);
112487579e9bSDan Schatzberg 	}
112587579e9bSDan Schatzberg 	spin_unlock_irq(&lo->lo_work_lock);
112687579e9bSDan Schatzberg 	del_timer_sync(&lo->timer);
112787579e9bSDan Schatzberg 
11281da177e4SLinus Torvalds 	spin_lock_irq(&lo->lo_lock);
11296050fa4cSTetsuo Handa 	filp = lo->lo_backing_file;
11301da177e4SLinus Torvalds 	lo->lo_backing_file = NULL;
113105eb0f25SKay Sievers 	spin_unlock_irq(&lo->lo_lock);
11321da177e4SLinus Torvalds 
11331da177e4SLinus Torvalds 	lo->lo_device = NULL;
11341da177e4SLinus Torvalds 	lo->lo_offset = 0;
11351da177e4SLinus Torvalds 	lo->lo_sizelimit = 0;
11361da177e4SLinus Torvalds 	memset(lo->lo_file_name, 0, LO_NAME_SIZE);
113789e4fdecSOmar Sandoval 	blk_queue_logical_block_size(lo->lo_queue, 512);
1138bf093753SOmar Sandoval 	blk_queue_physical_block_size(lo->lo_queue, 512);
1139bf093753SOmar Sandoval 	blk_queue_io_min(lo->lo_queue, 512);
1140e515be8fSXie Yongji 	invalidate_disk(lo->lo_disk);
114151a0bb0cSMilan Broz 	loop_sysfs_exit(lo);
1142c3473c63SDavid Zeuthen 	/* let user-space know about this change */
114319f553dbSXie Yongji 	kobject_uevent(&disk_to_dev(lo->lo_disk)->kobj, KOBJ_CHANGE);
11441da177e4SLinus Torvalds 	mapping_set_gfp_mask(filp->f_mapping, gfp);
1145f8933667SMing Lei 	blk_mq_unfreeze_queue(lo->lo_queue);
1146f8933667SMing Lei 
11479f65c489SMatteo Croce 	disk_force_media_change(lo->lo_disk, DISK_EVENT_MEDIA_CHANGE);
11486050fa4cSTetsuo Handa 
11496050fa4cSTetsuo Handa 	if (lo->lo_flags & LO_FLAGS_PARTSCAN) {
11506050fa4cSTetsuo Handa 		int err;
11516050fa4cSTetsuo Handa 
11520384264eSChristoph Hellwig 		mutex_lock(&lo->lo_disk->open_mutex);
11530384264eSChristoph Hellwig 		err = bdev_disk_changed(lo->lo_disk, false);
11540384264eSChristoph Hellwig 		mutex_unlock(&lo->lo_disk->open_mutex);
115540853d6fSDongli Zhang 		if (err)
1156d57f3374SJan Kara 			pr_warn("%s: partition scan of loop%d failed (rc=%d)\n",
11576050fa4cSTetsuo Handa 				__func__, lo->lo_number, err);
1158d57f3374SJan Kara 		/* Device is gone, no point in returning error */
1159d57f3374SJan Kara 	}
1160758a58d0SDongli Zhang 
1161758a58d0SDongli Zhang 	lo->lo_flags = 0;
1162758a58d0SDongli Zhang 	if (!part_shift)
116346e7eac6SChristoph Hellwig 		lo->lo_disk->flags |= GENHD_FL_NO_PART;
1164322c4293STetsuo Handa 
1165322c4293STetsuo Handa 	fput(filp);
1166322c4293STetsuo Handa }
1167322c4293STetsuo Handa 
1168322c4293STetsuo Handa static void loop_rundown_completed(struct loop_device *lo)
1169322c4293STetsuo Handa {
11706050fa4cSTetsuo Handa 	mutex_lock(&lo->lo_mutex);
1171758a58d0SDongli Zhang 	lo->lo_state = Lo_unbound;
11726cc8e743SPavel Tatashin 	mutex_unlock(&lo->lo_mutex);
1173322c4293STetsuo Handa 	module_put(THIS_MODULE);
1174322c4293STetsuo Handa }
1175758a58d0SDongli Zhang 
1176322c4293STetsuo Handa static void loop_rundown_workfn(struct work_struct *work)
1177322c4293STetsuo Handa {
1178322c4293STetsuo Handa 	struct loop_device *lo = container_of(work, struct loop_device,
1179322c4293STetsuo Handa 					      rundown_work);
1180322c4293STetsuo Handa 	struct block_device *bdev = lo->lo_device;
1181322c4293STetsuo Handa 	struct gendisk *disk = lo->lo_disk;
1182322c4293STetsuo Handa 
1183322c4293STetsuo Handa 	__loop_clr_fd(lo);
1184322c4293STetsuo Handa 	kobject_put(&bdev->bd_device.kobj);
1185322c4293STetsuo Handa 	module_put(disk->fops->owner);
1186322c4293STetsuo Handa 	loop_rundown_completed(lo);
1187322c4293STetsuo Handa }
1188322c4293STetsuo Handa 
1189322c4293STetsuo Handa static void loop_schedule_rundown(struct loop_device *lo)
1190322c4293STetsuo Handa {
1191322c4293STetsuo Handa 	struct block_device *bdev = lo->lo_device;
1192322c4293STetsuo Handa 	struct gendisk *disk = lo->lo_disk;
1193322c4293STetsuo Handa 
1194322c4293STetsuo Handa 	__module_get(disk->fops->owner);
1195322c4293STetsuo Handa 	kobject_get(&bdev->bd_device.kobj);
1196322c4293STetsuo Handa 	INIT_WORK(&lo->rundown_work, loop_rundown_workfn);
1197322c4293STetsuo Handa 	queue_work(system_long_wq, &lo->rundown_work);
11981da177e4SLinus Torvalds }
11991da177e4SLinus Torvalds 
1200a2505b79SJan Kara static int loop_clr_fd(struct loop_device *lo)
1201a2505b79SJan Kara {
12027ccd0791SJan Kara 	int err;
12037ccd0791SJan Kara 
12046cc8e743SPavel Tatashin 	err = mutex_lock_killable(&lo->lo_mutex);
12057ccd0791SJan Kara 	if (err)
12067ccd0791SJan Kara 		return err;
12077ccd0791SJan Kara 	if (lo->lo_state != Lo_bound) {
12086cc8e743SPavel Tatashin 		mutex_unlock(&lo->lo_mutex);
1209a2505b79SJan Kara 		return -ENXIO;
12107ccd0791SJan Kara 	}
1211a2505b79SJan Kara 	/*
1212a2505b79SJan Kara 	 * If we've explicitly asked to tear down the loop device,
1213a2505b79SJan Kara 	 * and it has an elevated reference count, set it for auto-teardown when
1214a2505b79SJan Kara 	 * the last reference goes away. This stops $!~#$@ udev from
1215a2505b79SJan Kara 	 * preventing teardown because it decided that it needs to run blkid on
1216a2505b79SJan Kara 	 * the loopback device whenever they appear. xfstests is notorious for
1217a2505b79SJan Kara 	 * failing tests because blkid via udev races with a losetup
1218a2505b79SJan Kara 	 * <dev>/do something like mkfs/losetup -d <dev> causing the losetup -d
1219a2505b79SJan Kara 	 * command to fail with EBUSY.
1220a2505b79SJan Kara 	 */
1221a2505b79SJan Kara 	if (atomic_read(&lo->lo_refcnt) > 1) {
1222a2505b79SJan Kara 		lo->lo_flags |= LO_FLAGS_AUTOCLEAR;
12236cc8e743SPavel Tatashin 		mutex_unlock(&lo->lo_mutex);
1224a2505b79SJan Kara 		return 0;
1225a2505b79SJan Kara 	}
1226a2505b79SJan Kara 	lo->lo_state = Lo_rundown;
12276cc8e743SPavel Tatashin 	mutex_unlock(&lo->lo_mutex);
1228a2505b79SJan Kara 
1229322c4293STetsuo Handa 	__loop_clr_fd(lo);
1230322c4293STetsuo Handa 	loop_rundown_completed(lo);
12316050fa4cSTetsuo Handa 	return 0;
1232a2505b79SJan Kara }
1233a2505b79SJan Kara 
12341da177e4SLinus Torvalds static int
12351da177e4SLinus Torvalds loop_set_status(struct loop_device *lo, const struct loop_info64 *info)
12361da177e4SLinus Torvalds {
12371da177e4SLinus Torvalds 	int err;
1238faf1d254SMartijn Coenen 	int prev_lo_flags;
123985b0a54aSJan Kara 	bool partscan = false;
12400c3796c2SMartijn Coenen 	bool size_changed = false;
12411da177e4SLinus Torvalds 
12426cc8e743SPavel Tatashin 	err = mutex_lock_killable(&lo->lo_mutex);
1243550df5fdSJan Kara 	if (err)
1244550df5fdSJan Kara 		return err;
1245550df5fdSJan Kara 	if (lo->lo_state != Lo_bound) {
1246550df5fdSJan Kara 		err = -ENXIO;
1247550df5fdSJan Kara 		goto out_unlock;
1248550df5fdSJan Kara 	}
12491da177e4SLinus Torvalds 
12505db470e2SJaegeuk Kim 	if (lo->lo_offset != info->lo_offset ||
12515db470e2SJaegeuk Kim 	    lo->lo_sizelimit != info->lo_sizelimit) {
12520c3796c2SMartijn Coenen 		size_changed = true;
12535db470e2SJaegeuk Kim 		sync_blockdev(lo->lo_device);
1254f4bd34b1SZheng Bin 		invalidate_bdev(lo->lo_device);
12555db470e2SJaegeuk Kim 	}
12565db470e2SJaegeuk Kim 
1257ecdd0959SMing Lei 	/* I/O need to be drained during transfer transition */
1258ecdd0959SMing Lei 	blk_mq_freeze_queue(lo->lo_queue);
1259ecdd0959SMing Lei 
12600c3796c2SMartijn Coenen 	if (size_changed && lo->lo_device->bd_inode->i_mapping->nrpages) {
1261f4bd34b1SZheng Bin 		/* If any pages were dirtied after invalidate_bdev(), try again */
12625db470e2SJaegeuk Kim 		err = -EAGAIN;
1263d9a74051SColin Ian King 		pr_warn("%s: loop%d (%s) still has dirty pages (nrpages=%lu)\n",
12645db470e2SJaegeuk Kim 			__func__, lo->lo_number, lo->lo_file_name,
12655db470e2SJaegeuk Kim 			lo->lo_device->bd_inode->i_mapping->nrpages);
12665db470e2SJaegeuk Kim 		goto out_unfreeze;
12675db470e2SJaegeuk Kim 	}
12680c3796c2SMartijn Coenen 
1269faf1d254SMartijn Coenen 	prev_lo_flags = lo->lo_flags;
1270faf1d254SMartijn Coenen 
12710c3796c2SMartijn Coenen 	err = loop_set_status_from_info(lo, info);
12720c3796c2SMartijn Coenen 	if (err)
1273550df5fdSJan Kara 		goto out_unfreeze;
12740c3796c2SMartijn Coenen 
1275faf1d254SMartijn Coenen 	/* Mask out flags that can't be set using LOOP_SET_STATUS. */
12766ac92fb5SMartijn Coenen 	lo->lo_flags &= LOOP_SET_STATUS_SETTABLE_FLAGS;
1277faf1d254SMartijn Coenen 	/* For those flags, use the previous values instead */
1278faf1d254SMartijn Coenen 	lo->lo_flags |= prev_lo_flags & ~LOOP_SET_STATUS_SETTABLE_FLAGS;
1279faf1d254SMartijn Coenen 	/* For flags that can't be cleared, use previous values too */
1280faf1d254SMartijn Coenen 	lo->lo_flags |= prev_lo_flags & ~LOOP_SET_STATUS_CLEARABLE_FLAGS;
1281faf1d254SMartijn Coenen 
1282b0bd158dSMartijn Coenen 	if (size_changed) {
1283b0bd158dSMartijn Coenen 		loff_t new_size = get_size(lo->lo_offset, lo->lo_sizelimit,
1284b0bd158dSMartijn Coenen 					   lo->lo_backing_file);
1285b0bd158dSMartijn Coenen 		loop_set_size(lo, new_size);
1286b040ad9cSArnd Bergmann 	}
1287541c742aSGuo Chao 
1288dfaa2ef6SLukas Czerner 	loop_config_discard(lo);
12891da177e4SLinus Torvalds 
12902e5ab5f3SMing Lei 	/* update dio if lo_offset or transfer is changed */
12912e5ab5f3SMing Lei 	__loop_update_dio(lo, lo->use_dio);
12922e5ab5f3SMing Lei 
1293550df5fdSJan Kara out_unfreeze:
1294ecdd0959SMing Lei 	blk_mq_unfreeze_queue(lo->lo_queue);
1295e02898b4SOmar Sandoval 
1296faf1d254SMartijn Coenen 	if (!err && (lo->lo_flags & LO_FLAGS_PARTSCAN) &&
1297faf1d254SMartijn Coenen 	     !(prev_lo_flags & LO_FLAGS_PARTSCAN)) {
129846e7eac6SChristoph Hellwig 		lo->lo_disk->flags &= ~GENHD_FL_NO_PART;
129985b0a54aSJan Kara 		partscan = true;
1300e02898b4SOmar Sandoval 	}
1301550df5fdSJan Kara out_unlock:
13026cc8e743SPavel Tatashin 	mutex_unlock(&lo->lo_mutex);
130385b0a54aSJan Kara 	if (partscan)
13040384264eSChristoph Hellwig 		loop_reread_partitions(lo);
1305e02898b4SOmar Sandoval 
1306ecdd0959SMing Lei 	return err;
13071da177e4SLinus Torvalds }
13081da177e4SLinus Torvalds 
13091da177e4SLinus Torvalds static int
13101da177e4SLinus Torvalds loop_get_status(struct loop_device *lo, struct loop_info64 *info)
13111da177e4SLinus Torvalds {
1312b1ab5fa3STetsuo Handa 	struct path path;
13131da177e4SLinus Torvalds 	struct kstat stat;
13142d1d4c1eSOmar Sandoval 	int ret;
13151da177e4SLinus Torvalds 
13166cc8e743SPavel Tatashin 	ret = mutex_lock_killable(&lo->lo_mutex);
13174a5ce9baSJan Kara 	if (ret)
13184a5ce9baSJan Kara 		return ret;
13192d1d4c1eSOmar Sandoval 	if (lo->lo_state != Lo_bound) {
13206cc8e743SPavel Tatashin 		mutex_unlock(&lo->lo_mutex);
13211da177e4SLinus Torvalds 		return -ENXIO;
13222d1d4c1eSOmar Sandoval 	}
13232d1d4c1eSOmar Sandoval 
13241da177e4SLinus Torvalds 	memset(info, 0, sizeof(*info));
13251da177e4SLinus Torvalds 	info->lo_number = lo->lo_number;
13261da177e4SLinus Torvalds 	info->lo_offset = lo->lo_offset;
13271da177e4SLinus Torvalds 	info->lo_sizelimit = lo->lo_sizelimit;
13281da177e4SLinus Torvalds 	info->lo_flags = lo->lo_flags;
13291da177e4SLinus Torvalds 	memcpy(info->lo_file_name, lo->lo_file_name, LO_NAME_SIZE);
13302d1d4c1eSOmar Sandoval 
13316cc8e743SPavel Tatashin 	/* Drop lo_mutex while we call into the filesystem. */
1332b1ab5fa3STetsuo Handa 	path = lo->lo_backing_file->f_path;
1333b1ab5fa3STetsuo Handa 	path_get(&path);
13346cc8e743SPavel Tatashin 	mutex_unlock(&lo->lo_mutex);
1335b1ab5fa3STetsuo Handa 	ret = vfs_getattr(&path, &stat, STATX_INO, AT_STATX_SYNC_AS_STAT);
13362d1d4c1eSOmar Sandoval 	if (!ret) {
13372d1d4c1eSOmar Sandoval 		info->lo_device = huge_encode_dev(stat.dev);
13382d1d4c1eSOmar Sandoval 		info->lo_inode = stat.ino;
13392d1d4c1eSOmar Sandoval 		info->lo_rdevice = huge_encode_dev(stat.rdev);
13402d1d4c1eSOmar Sandoval 	}
1341b1ab5fa3STetsuo Handa 	path_put(&path);
13422d1d4c1eSOmar Sandoval 	return ret;
13431da177e4SLinus Torvalds }
13441da177e4SLinus Torvalds 
13451da177e4SLinus Torvalds static void
13461da177e4SLinus Torvalds loop_info64_from_old(const struct loop_info *info, struct loop_info64 *info64)
13471da177e4SLinus Torvalds {
13481da177e4SLinus Torvalds 	memset(info64, 0, sizeof(*info64));
13491da177e4SLinus Torvalds 	info64->lo_number = info->lo_number;
13501da177e4SLinus Torvalds 	info64->lo_device = info->lo_device;
13511da177e4SLinus Torvalds 	info64->lo_inode = info->lo_inode;
13521da177e4SLinus Torvalds 	info64->lo_rdevice = info->lo_rdevice;
13531da177e4SLinus Torvalds 	info64->lo_offset = info->lo_offset;
13541da177e4SLinus Torvalds 	info64->lo_sizelimit = 0;
13551da177e4SLinus Torvalds 	info64->lo_flags = info->lo_flags;
13561da177e4SLinus Torvalds 	memcpy(info64->lo_file_name, info->lo_name, LO_NAME_SIZE);
13571da177e4SLinus Torvalds }
13581da177e4SLinus Torvalds 
13591da177e4SLinus Torvalds static int
13601da177e4SLinus Torvalds loop_info64_to_old(const struct loop_info64 *info64, struct loop_info *info)
13611da177e4SLinus Torvalds {
13621da177e4SLinus Torvalds 	memset(info, 0, sizeof(*info));
13631da177e4SLinus Torvalds 	info->lo_number = info64->lo_number;
13641da177e4SLinus Torvalds 	info->lo_device = info64->lo_device;
13651da177e4SLinus Torvalds 	info->lo_inode = info64->lo_inode;
13661da177e4SLinus Torvalds 	info->lo_rdevice = info64->lo_rdevice;
13671da177e4SLinus Torvalds 	info->lo_offset = info64->lo_offset;
13681da177e4SLinus Torvalds 	info->lo_flags = info64->lo_flags;
13691da177e4SLinus Torvalds 	memcpy(info->lo_name, info64->lo_file_name, LO_NAME_SIZE);
13701da177e4SLinus Torvalds 
13711da177e4SLinus Torvalds 	/* error in case values were truncated */
13721da177e4SLinus Torvalds 	if (info->lo_device != info64->lo_device ||
13731da177e4SLinus Torvalds 	    info->lo_rdevice != info64->lo_rdevice ||
13741da177e4SLinus Torvalds 	    info->lo_inode != info64->lo_inode ||
13751da177e4SLinus Torvalds 	    info->lo_offset != info64->lo_offset)
13761da177e4SLinus Torvalds 		return -EOVERFLOW;
13771da177e4SLinus Torvalds 
13781da177e4SLinus Torvalds 	return 0;
13791da177e4SLinus Torvalds }
13801da177e4SLinus Torvalds 
13811da177e4SLinus Torvalds static int
13821da177e4SLinus Torvalds loop_set_status_old(struct loop_device *lo, const struct loop_info __user *arg)
13831da177e4SLinus Torvalds {
13841da177e4SLinus Torvalds 	struct loop_info info;
13851da177e4SLinus Torvalds 	struct loop_info64 info64;
13861da177e4SLinus Torvalds 
13871da177e4SLinus Torvalds 	if (copy_from_user(&info, arg, sizeof (struct loop_info)))
13881da177e4SLinus Torvalds 		return -EFAULT;
13891da177e4SLinus Torvalds 	loop_info64_from_old(&info, &info64);
13901da177e4SLinus Torvalds 	return loop_set_status(lo, &info64);
13911da177e4SLinus Torvalds }
13921da177e4SLinus Torvalds 
13931da177e4SLinus Torvalds static int
13941da177e4SLinus Torvalds loop_set_status64(struct loop_device *lo, const struct loop_info64 __user *arg)
13951da177e4SLinus Torvalds {
13961da177e4SLinus Torvalds 	struct loop_info64 info64;
13971da177e4SLinus Torvalds 
13981da177e4SLinus Torvalds 	if (copy_from_user(&info64, arg, sizeof (struct loop_info64)))
13991da177e4SLinus Torvalds 		return -EFAULT;
14001da177e4SLinus Torvalds 	return loop_set_status(lo, &info64);
14011da177e4SLinus Torvalds }
14021da177e4SLinus Torvalds 
14031da177e4SLinus Torvalds static int
14041da177e4SLinus Torvalds loop_get_status_old(struct loop_device *lo, struct loop_info __user *arg) {
14051da177e4SLinus Torvalds 	struct loop_info info;
14061da177e4SLinus Torvalds 	struct loop_info64 info64;
1407bdac616dSOmar Sandoval 	int err;
14081da177e4SLinus Torvalds 
14094a5ce9baSJan Kara 	if (!arg)
1410bdac616dSOmar Sandoval 		return -EINVAL;
14111da177e4SLinus Torvalds 	err = loop_get_status(lo, &info64);
14121da177e4SLinus Torvalds 	if (!err)
14131da177e4SLinus Torvalds 		err = loop_info64_to_old(&info64, &info);
14141da177e4SLinus Torvalds 	if (!err && copy_to_user(arg, &info, sizeof(info)))
14151da177e4SLinus Torvalds 		err = -EFAULT;
14161da177e4SLinus Torvalds 
14171da177e4SLinus Torvalds 	return err;
14181da177e4SLinus Torvalds }
14191da177e4SLinus Torvalds 
14201da177e4SLinus Torvalds static int
14211da177e4SLinus Torvalds loop_get_status64(struct loop_device *lo, struct loop_info64 __user *arg) {
14221da177e4SLinus Torvalds 	struct loop_info64 info64;
1423bdac616dSOmar Sandoval 	int err;
14241da177e4SLinus Torvalds 
14254a5ce9baSJan Kara 	if (!arg)
1426bdac616dSOmar Sandoval 		return -EINVAL;
14271da177e4SLinus Torvalds 	err = loop_get_status(lo, &info64);
14281da177e4SLinus Torvalds 	if (!err && copy_to_user(arg, &info64, sizeof(info64)))
14291da177e4SLinus Torvalds 		err = -EFAULT;
14301da177e4SLinus Torvalds 
14311da177e4SLinus Torvalds 	return err;
14321da177e4SLinus Torvalds }
14331da177e4SLinus Torvalds 
143451001b7dSHannes Reinecke static int loop_set_capacity(struct loop_device *lo)
143553d66608SJ. R. Okajima {
14360a6ed1b5SMartijn Coenen 	loff_t size;
14370a6ed1b5SMartijn Coenen 
143853d66608SJ. R. Okajima 	if (unlikely(lo->lo_state != Lo_bound))
14397b0576a3SGuo Chao 		return -ENXIO;
144053d66608SJ. R. Okajima 
14410a6ed1b5SMartijn Coenen 	size = get_loop_size(lo, lo->lo_backing_file);
14420a6ed1b5SMartijn Coenen 	loop_set_size(lo, size);
1443083a6a50SMartijn Coenen 
1444083a6a50SMartijn Coenen 	return 0;
144553d66608SJ. R. Okajima }
144653d66608SJ. R. Okajima 
1447ab1cb278SMing Lei static int loop_set_dio(struct loop_device *lo, unsigned long arg)
1448ab1cb278SMing Lei {
1449ab1cb278SMing Lei 	int error = -ENXIO;
1450ab1cb278SMing Lei 	if (lo->lo_state != Lo_bound)
1451ab1cb278SMing Lei 		goto out;
1452ab1cb278SMing Lei 
1453ab1cb278SMing Lei 	__loop_update_dio(lo, !!arg);
1454ab1cb278SMing Lei 	if (lo->use_dio == !!arg)
1455ab1cb278SMing Lei 		return 0;
1456ab1cb278SMing Lei 	error = -EINVAL;
1457ab1cb278SMing Lei  out:
1458ab1cb278SMing Lei 	return error;
1459ab1cb278SMing Lei }
1460ab1cb278SMing Lei 
146189e4fdecSOmar Sandoval static int loop_set_block_size(struct loop_device *lo, unsigned long arg)
146289e4fdecSOmar Sandoval {
14635db470e2SJaegeuk Kim 	int err = 0;
14645db470e2SJaegeuk Kim 
146589e4fdecSOmar Sandoval 	if (lo->lo_state != Lo_bound)
146689e4fdecSOmar Sandoval 		return -ENXIO;
146789e4fdecSOmar Sandoval 
1468af3c570fSXie Yongji 	err = blk_validate_block_size(arg);
14693448914eSMartijn Coenen 	if (err)
14703448914eSMartijn Coenen 		return err;
147189e4fdecSOmar Sandoval 
14727e81f99aSMartijn Coenen 	if (lo->lo_queue->limits.logical_block_size == arg)
14737e81f99aSMartijn Coenen 		return 0;
14747e81f99aSMartijn Coenen 
14755db470e2SJaegeuk Kim 	sync_blockdev(lo->lo_device);
1476f4bd34b1SZheng Bin 	invalidate_bdev(lo->lo_device);
14775db470e2SJaegeuk Kim 
147889e4fdecSOmar Sandoval 	blk_mq_freeze_queue(lo->lo_queue);
147989e4fdecSOmar Sandoval 
1480f4bd34b1SZheng Bin 	/* invalidate_bdev should have truncated all the pages */
14817e81f99aSMartijn Coenen 	if (lo->lo_device->bd_inode->i_mapping->nrpages) {
14825db470e2SJaegeuk Kim 		err = -EAGAIN;
1483d9a74051SColin Ian King 		pr_warn("%s: loop%d (%s) still has dirty pages (nrpages=%lu)\n",
14845db470e2SJaegeuk Kim 			__func__, lo->lo_number, lo->lo_file_name,
14855db470e2SJaegeuk Kim 			lo->lo_device->bd_inode->i_mapping->nrpages);
14865db470e2SJaegeuk Kim 		goto out_unfreeze;
14875db470e2SJaegeuk Kim 	}
14885db470e2SJaegeuk Kim 
148989e4fdecSOmar Sandoval 	blk_queue_logical_block_size(lo->lo_queue, arg);
1490bf093753SOmar Sandoval 	blk_queue_physical_block_size(lo->lo_queue, arg);
1491bf093753SOmar Sandoval 	blk_queue_io_min(lo->lo_queue, arg);
149289e4fdecSOmar Sandoval 	loop_update_dio(lo);
14935db470e2SJaegeuk Kim out_unfreeze:
149489e4fdecSOmar Sandoval 	blk_mq_unfreeze_queue(lo->lo_queue);
149589e4fdecSOmar Sandoval 
14965db470e2SJaegeuk Kim 	return err;
149789e4fdecSOmar Sandoval }
149889e4fdecSOmar Sandoval 
1499a1316544SJan Kara static int lo_simple_ioctl(struct loop_device *lo, unsigned int cmd,
1500a1316544SJan Kara 			   unsigned long arg)
15011da177e4SLinus Torvalds {
15021da177e4SLinus Torvalds 	int err;
15031da177e4SLinus Torvalds 
15046cc8e743SPavel Tatashin 	err = mutex_lock_killable(&lo->lo_mutex);
15053148ffbdSOmar Sandoval 	if (err)
1506a1316544SJan Kara 		return err;
15071da177e4SLinus Torvalds 	switch (cmd) {
150853d66608SJ. R. Okajima 	case LOOP_SET_CAPACITY:
150951001b7dSHannes Reinecke 		err = loop_set_capacity(lo);
151053d66608SJ. R. Okajima 		break;
1511ab1cb278SMing Lei 	case LOOP_SET_DIRECT_IO:
1512ab1cb278SMing Lei 		err = loop_set_dio(lo, arg);
1513ab1cb278SMing Lei 		break;
151489e4fdecSOmar Sandoval 	case LOOP_SET_BLOCK_SIZE:
151589e4fdecSOmar Sandoval 		err = loop_set_block_size(lo, arg);
151689e4fdecSOmar Sandoval 		break;
15171da177e4SLinus Torvalds 	default:
151847e96246SChristoph Hellwig 		err = -EINVAL;
15191da177e4SLinus Torvalds 	}
15206cc8e743SPavel Tatashin 	mutex_unlock(&lo->lo_mutex);
1521a1316544SJan Kara 	return err;
1522a1316544SJan Kara }
1523f028f3b2SNikanth Karthikesan 
1524a1316544SJan Kara static int lo_ioctl(struct block_device *bdev, fmode_t mode,
1525a1316544SJan Kara 	unsigned int cmd, unsigned long arg)
1526a1316544SJan Kara {
1527a1316544SJan Kara 	struct loop_device *lo = bdev->bd_disk->private_data;
1528571fae6eSMartijn Coenen 	void __user *argp = (void __user *) arg;
1529a1316544SJan Kara 	int err;
1530a1316544SJan Kara 
1531a1316544SJan Kara 	switch (cmd) {
15323448914eSMartijn Coenen 	case LOOP_SET_FD: {
15333448914eSMartijn Coenen 		/*
15343448914eSMartijn Coenen 		 * Legacy case - pass in a zeroed out struct loop_config with
15353448914eSMartijn Coenen 		 * only the file descriptor set , which corresponds with the
15363448914eSMartijn Coenen 		 * default parameters we'd have used otherwise.
15373448914eSMartijn Coenen 		 */
15383448914eSMartijn Coenen 		struct loop_config config;
15393448914eSMartijn Coenen 
15403448914eSMartijn Coenen 		memset(&config, 0, sizeof(config));
15413448914eSMartijn Coenen 		config.fd = arg;
15423448914eSMartijn Coenen 
15433448914eSMartijn Coenen 		return loop_configure(lo, mode, bdev, &config);
15443448914eSMartijn Coenen 	}
15453448914eSMartijn Coenen 	case LOOP_CONFIGURE: {
15463448914eSMartijn Coenen 		struct loop_config config;
15473448914eSMartijn Coenen 
15483448914eSMartijn Coenen 		if (copy_from_user(&config, argp, sizeof(config)))
15493448914eSMartijn Coenen 			return -EFAULT;
15503448914eSMartijn Coenen 
15513448914eSMartijn Coenen 		return loop_configure(lo, mode, bdev, &config);
15523448914eSMartijn Coenen 	}
1553a1316544SJan Kara 	case LOOP_CHANGE_FD:
1554c3710770SJan Kara 		return loop_change_fd(lo, bdev, arg);
1555a1316544SJan Kara 	case LOOP_CLR_FD:
15567ccd0791SJan Kara 		return loop_clr_fd(lo);
1557a1316544SJan Kara 	case LOOP_SET_STATUS:
1558a1316544SJan Kara 		err = -EPERM;
1559a1316544SJan Kara 		if ((mode & FMODE_WRITE) || capable(CAP_SYS_ADMIN)) {
1560571fae6eSMartijn Coenen 			err = loop_set_status_old(lo, argp);
1561a1316544SJan Kara 		}
1562a1316544SJan Kara 		break;
1563a1316544SJan Kara 	case LOOP_GET_STATUS:
1564571fae6eSMartijn Coenen 		return loop_get_status_old(lo, argp);
1565a1316544SJan Kara 	case LOOP_SET_STATUS64:
1566a1316544SJan Kara 		err = -EPERM;
1567a1316544SJan Kara 		if ((mode & FMODE_WRITE) || capable(CAP_SYS_ADMIN)) {
1568571fae6eSMartijn Coenen 			err = loop_set_status64(lo, argp);
1569a1316544SJan Kara 		}
1570a1316544SJan Kara 		break;
1571a1316544SJan Kara 	case LOOP_GET_STATUS64:
1572571fae6eSMartijn Coenen 		return loop_get_status64(lo, argp);
1573a1316544SJan Kara 	case LOOP_SET_CAPACITY:
1574a1316544SJan Kara 	case LOOP_SET_DIRECT_IO:
1575a1316544SJan Kara 	case LOOP_SET_BLOCK_SIZE:
1576a1316544SJan Kara 		if (!(mode & FMODE_WRITE) && !capable(CAP_SYS_ADMIN))
1577a1316544SJan Kara 			return -EPERM;
1578df561f66SGustavo A. R. Silva 		fallthrough;
1579a1316544SJan Kara 	default:
1580a1316544SJan Kara 		err = lo_simple_ioctl(lo, cmd, arg);
1581a1316544SJan Kara 		break;
1582a1316544SJan Kara 	}
1583a1316544SJan Kara 
15841da177e4SLinus Torvalds 	return err;
15851da177e4SLinus Torvalds }
15861da177e4SLinus Torvalds 
1587863d5b82SDavid Howells #ifdef CONFIG_COMPAT
1588863d5b82SDavid Howells struct compat_loop_info {
1589863d5b82SDavid Howells 	compat_int_t	lo_number;      /* ioctl r/o */
1590863d5b82SDavid Howells 	compat_dev_t	lo_device;      /* ioctl r/o */
1591863d5b82SDavid Howells 	compat_ulong_t	lo_inode;       /* ioctl r/o */
1592863d5b82SDavid Howells 	compat_dev_t	lo_rdevice;     /* ioctl r/o */
1593863d5b82SDavid Howells 	compat_int_t	lo_offset;
1594863d5b82SDavid Howells 	compat_int_t	lo_encrypt_key_size;    /* ioctl w/o */
1595863d5b82SDavid Howells 	compat_int_t	lo_flags;       /* ioctl r/o */
1596863d5b82SDavid Howells 	char		lo_name[LO_NAME_SIZE];
1597863d5b82SDavid Howells 	unsigned char	lo_encrypt_key[LO_KEY_SIZE]; /* ioctl w/o */
1598863d5b82SDavid Howells 	compat_ulong_t	lo_init[2];
1599863d5b82SDavid Howells 	char		reserved[4];
1600863d5b82SDavid Howells };
1601863d5b82SDavid Howells 
1602863d5b82SDavid Howells /*
1603863d5b82SDavid Howells  * Transfer 32-bit compatibility structure in userspace to 64-bit loop info
1604863d5b82SDavid Howells  * - noinlined to reduce stack space usage in main part of driver
1605863d5b82SDavid Howells  */
1606863d5b82SDavid Howells static noinline int
1607ba674cfcSAl Viro loop_info64_from_compat(const struct compat_loop_info __user *arg,
1608863d5b82SDavid Howells 			struct loop_info64 *info64)
1609863d5b82SDavid Howells {
1610863d5b82SDavid Howells 	struct compat_loop_info info;
1611863d5b82SDavid Howells 
1612863d5b82SDavid Howells 	if (copy_from_user(&info, arg, sizeof(info)))
1613863d5b82SDavid Howells 		return -EFAULT;
1614863d5b82SDavid Howells 
1615863d5b82SDavid Howells 	memset(info64, 0, sizeof(*info64));
1616863d5b82SDavid Howells 	info64->lo_number = info.lo_number;
1617863d5b82SDavid Howells 	info64->lo_device = info.lo_device;
1618863d5b82SDavid Howells 	info64->lo_inode = info.lo_inode;
1619863d5b82SDavid Howells 	info64->lo_rdevice = info.lo_rdevice;
1620863d5b82SDavid Howells 	info64->lo_offset = info.lo_offset;
1621863d5b82SDavid Howells 	info64->lo_sizelimit = 0;
1622863d5b82SDavid Howells 	info64->lo_flags = info.lo_flags;
1623863d5b82SDavid Howells 	memcpy(info64->lo_file_name, info.lo_name, LO_NAME_SIZE);
1624863d5b82SDavid Howells 	return 0;
1625863d5b82SDavid Howells }
1626863d5b82SDavid Howells 
1627863d5b82SDavid Howells /*
1628863d5b82SDavid Howells  * Transfer 64-bit loop info to 32-bit compatibility structure in userspace
1629863d5b82SDavid Howells  * - noinlined to reduce stack space usage in main part of driver
1630863d5b82SDavid Howells  */
1631863d5b82SDavid Howells static noinline int
1632863d5b82SDavid Howells loop_info64_to_compat(const struct loop_info64 *info64,
1633863d5b82SDavid Howells 		      struct compat_loop_info __user *arg)
1634863d5b82SDavid Howells {
1635863d5b82SDavid Howells 	struct compat_loop_info info;
1636863d5b82SDavid Howells 
1637863d5b82SDavid Howells 	memset(&info, 0, sizeof(info));
1638863d5b82SDavid Howells 	info.lo_number = info64->lo_number;
1639863d5b82SDavid Howells 	info.lo_device = info64->lo_device;
1640863d5b82SDavid Howells 	info.lo_inode = info64->lo_inode;
1641863d5b82SDavid Howells 	info.lo_rdevice = info64->lo_rdevice;
1642863d5b82SDavid Howells 	info.lo_offset = info64->lo_offset;
1643863d5b82SDavid Howells 	info.lo_flags = info64->lo_flags;
1644863d5b82SDavid Howells 	memcpy(info.lo_name, info64->lo_file_name, LO_NAME_SIZE);
1645863d5b82SDavid Howells 
1646863d5b82SDavid Howells 	/* error in case values were truncated */
1647863d5b82SDavid Howells 	if (info.lo_device != info64->lo_device ||
1648863d5b82SDavid Howells 	    info.lo_rdevice != info64->lo_rdevice ||
1649863d5b82SDavid Howells 	    info.lo_inode != info64->lo_inode ||
165047e96246SChristoph Hellwig 	    info.lo_offset != info64->lo_offset)
1651863d5b82SDavid Howells 		return -EOVERFLOW;
1652863d5b82SDavid Howells 
1653863d5b82SDavid Howells 	if (copy_to_user(arg, &info, sizeof(info)))
1654863d5b82SDavid Howells 		return -EFAULT;
1655863d5b82SDavid Howells 	return 0;
1656863d5b82SDavid Howells }
1657863d5b82SDavid Howells 
1658863d5b82SDavid Howells static int
1659863d5b82SDavid Howells loop_set_status_compat(struct loop_device *lo,
1660863d5b82SDavid Howells 		       const struct compat_loop_info __user *arg)
1661863d5b82SDavid Howells {
1662863d5b82SDavid Howells 	struct loop_info64 info64;
1663863d5b82SDavid Howells 	int ret;
1664863d5b82SDavid Howells 
1665863d5b82SDavid Howells 	ret = loop_info64_from_compat(arg, &info64);
1666863d5b82SDavid Howells 	if (ret < 0)
1667863d5b82SDavid Howells 		return ret;
1668863d5b82SDavid Howells 	return loop_set_status(lo, &info64);
1669863d5b82SDavid Howells }
1670863d5b82SDavid Howells 
1671863d5b82SDavid Howells static int
1672863d5b82SDavid Howells loop_get_status_compat(struct loop_device *lo,
1673863d5b82SDavid Howells 		       struct compat_loop_info __user *arg)
1674863d5b82SDavid Howells {
1675863d5b82SDavid Howells 	struct loop_info64 info64;
1676bdac616dSOmar Sandoval 	int err;
1677863d5b82SDavid Howells 
16784a5ce9baSJan Kara 	if (!arg)
1679bdac616dSOmar Sandoval 		return -EINVAL;
1680863d5b82SDavid Howells 	err = loop_get_status(lo, &info64);
1681863d5b82SDavid Howells 	if (!err)
1682863d5b82SDavid Howells 		err = loop_info64_to_compat(&info64, arg);
1683863d5b82SDavid Howells 	return err;
1684863d5b82SDavid Howells }
1685863d5b82SDavid Howells 
1686bb214884SAl Viro static int lo_compat_ioctl(struct block_device *bdev, fmode_t mode,
1687bb214884SAl Viro 			   unsigned int cmd, unsigned long arg)
1688863d5b82SDavid Howells {
1689bb214884SAl Viro 	struct loop_device *lo = bdev->bd_disk->private_data;
1690863d5b82SDavid Howells 	int err;
1691863d5b82SDavid Howells 
1692863d5b82SDavid Howells 	switch(cmd) {
1693863d5b82SDavid Howells 	case LOOP_SET_STATUS:
16943148ffbdSOmar Sandoval 		err = loop_set_status_compat(lo,
16953148ffbdSOmar Sandoval 			     (const struct compat_loop_info __user *)arg);
1696863d5b82SDavid Howells 		break;
1697863d5b82SDavid Howells 	case LOOP_GET_STATUS:
16983148ffbdSOmar Sandoval 		err = loop_get_status_compat(lo,
16993148ffbdSOmar Sandoval 				     (struct compat_loop_info __user *)arg);
1700863d5b82SDavid Howells 		break;
170153d66608SJ. R. Okajima 	case LOOP_SET_CAPACITY:
1702863d5b82SDavid Howells 	case LOOP_CLR_FD:
1703863d5b82SDavid Howells 	case LOOP_GET_STATUS64:
1704863d5b82SDavid Howells 	case LOOP_SET_STATUS64:
17053448914eSMartijn Coenen 	case LOOP_CONFIGURE:
1706863d5b82SDavid Howells 		arg = (unsigned long) compat_ptr(arg);
1707df561f66SGustavo A. R. Silva 		fallthrough;
1708863d5b82SDavid Howells 	case LOOP_SET_FD:
1709863d5b82SDavid Howells 	case LOOP_CHANGE_FD:
17109fea4b39SEvan Green 	case LOOP_SET_BLOCK_SIZE:
1711fdbe4eeeSAlessio Balsini 	case LOOP_SET_DIRECT_IO:
1712bb214884SAl Viro 		err = lo_ioctl(bdev, mode, cmd, arg);
1713863d5b82SDavid Howells 		break;
1714863d5b82SDavid Howells 	default:
1715863d5b82SDavid Howells 		err = -ENOIOCTLCMD;
1716863d5b82SDavid Howells 		break;
1717863d5b82SDavid Howells 	}
1718863d5b82SDavid Howells 	return err;
1719863d5b82SDavid Howells }
1720863d5b82SDavid Howells #endif
1721863d5b82SDavid Howells 
1722bb214884SAl Viro static int lo_open(struct block_device *bdev, fmode_t mode)
17231da177e4SLinus Torvalds {
1724990e7811SChristoph Hellwig 	struct loop_device *lo = bdev->bd_disk->private_data;
17250a42e99bSJan Kara 	int err;
1726770fe30aSKay Sievers 
17276cc8e743SPavel Tatashin 	err = mutex_lock_killable(&lo->lo_mutex);
17286cc8e743SPavel Tatashin 	if (err)
1729770fe30aSKay Sievers 		return err;
1730990e7811SChristoph Hellwig 	if (lo->lo_state == Lo_deleting)
1731990e7811SChristoph Hellwig 		err = -ENXIO;
1732990e7811SChristoph Hellwig 	else
17336cc8e743SPavel Tatashin 		atomic_inc(&lo->lo_refcnt);
17346cc8e743SPavel Tatashin 	mutex_unlock(&lo->lo_mutex);
1735990e7811SChristoph Hellwig 	return err;
17361da177e4SLinus Torvalds }
17371da177e4SLinus Torvalds 
1738967d1dc1SJan Kara static void lo_release(struct gendisk *disk, fmode_t mode)
17391da177e4SLinus Torvalds {
17406cc8e743SPavel Tatashin 	struct loop_device *lo = disk->private_data;
17411da177e4SLinus Torvalds 
17426cc8e743SPavel Tatashin 	mutex_lock(&lo->lo_mutex);
1743f8933667SMing Lei 	if (atomic_dec_return(&lo->lo_refcnt))
17440a42e99bSJan Kara 		goto out_unlock;
1745f8933667SMing Lei 
174614f27939SMilan Broz 	if (lo->lo_flags & LO_FLAGS_AUTOCLEAR) {
1747a2505b79SJan Kara 		if (lo->lo_state != Lo_bound)
1748a2505b79SJan Kara 			goto out_unlock;
1749a2505b79SJan Kara 		lo->lo_state = Lo_rundown;
17506cc8e743SPavel Tatashin 		mutex_unlock(&lo->lo_mutex);
175114f27939SMilan Broz 		/*
175214f27939SMilan Broz 		 * In autoclear mode, stop the loop thread
175314f27939SMilan Broz 		 * and remove configuration after last close.
175414f27939SMilan Broz 		 */
1755322c4293STetsuo Handa 		loop_schedule_rundown(lo);
17560a42e99bSJan Kara 		return;
175743cade80SOmar Sandoval 	} else if (lo->lo_state == Lo_bound) {
175814f27939SMilan Broz 		/*
175914f27939SMilan Broz 		 * Otherwise keep thread (if running) and config,
176014f27939SMilan Broz 		 * but flush possible ongoing bios in thread.
176114f27939SMilan Broz 		 */
176243cade80SOmar Sandoval 		blk_mq_freeze_queue(lo->lo_queue);
176343cade80SOmar Sandoval 		blk_mq_unfreeze_queue(lo->lo_queue);
176414f27939SMilan Broz 	}
176596c58655SDavid Woodhouse 
17660a42e99bSJan Kara out_unlock:
17676cc8e743SPavel Tatashin 	mutex_unlock(&lo->lo_mutex);
1768ae665016SLinus Torvalds }
1769ae665016SLinus Torvalds 
177083d5cde4SAlexey Dobriyan static const struct block_device_operations lo_fops = {
17711da177e4SLinus Torvalds 	.owner =	THIS_MODULE,
1772bb214884SAl Viro 	.open =		lo_open,
1773bb214884SAl Viro 	.release =	lo_release,
1774bb214884SAl Viro 	.ioctl =	lo_ioctl,
1775863d5b82SDavid Howells #ifdef CONFIG_COMPAT
1776bb214884SAl Viro 	.compat_ioctl =	lo_compat_ioctl,
1777863d5b82SDavid Howells #endif
17781da177e4SLinus Torvalds };
17791da177e4SLinus Torvalds 
17801da177e4SLinus Torvalds /*
17811da177e4SLinus Torvalds  * And now the modules code and kernel interface.
17821da177e4SLinus Torvalds  */
178373285082SKen Chen static int max_loop;
17845657a819SJoe Perches module_param(max_loop, int, 0444);
1785a47653fcSKen Chen MODULE_PARM_DESC(max_loop, "Maximum number of loop devices");
17865657a819SJoe Perches module_param(max_part, int, 0444);
1787476a4813SLaurent Vivier MODULE_PARM_DESC(max_part, "Maximum number of partitions per loop device");
17881da177e4SLinus Torvalds MODULE_LICENSE("GPL");
17891da177e4SLinus Torvalds MODULE_ALIAS_BLOCKDEV_MAJOR(LOOP_MAJOR);
17901da177e4SLinus Torvalds 
1791fc17b653SChristoph Hellwig static blk_status_t loop_queue_rq(struct blk_mq_hw_ctx *hctx,
1792b5dd2f60SMing Lei 		const struct blk_mq_queue_data *bd)
1793b5dd2f60SMing Lei {
17941894e916SJens Axboe 	struct request *rq = bd->rq;
17951894e916SJens Axboe 	struct loop_cmd *cmd = blk_mq_rq_to_pdu(rq);
17961894e916SJens Axboe 	struct loop_device *lo = rq->q->queuedata;
1797b5dd2f60SMing Lei 
17981894e916SJens Axboe 	blk_mq_start_request(rq);
1799b5dd2f60SMing Lei 
1800f4aa4c7bSMing Lei 	if (lo->lo_state != Lo_bound)
1801fc17b653SChristoph Hellwig 		return BLK_STS_IOERR;
1802f4aa4c7bSMing Lei 
18031894e916SJens Axboe 	switch (req_op(rq)) {
1804f0225cacSChristoph Hellwig 	case REQ_OP_FLUSH:
1805f0225cacSChristoph Hellwig 	case REQ_OP_DISCARD:
180619372e27SChristoph Hellwig 	case REQ_OP_WRITE_ZEROES:
1807bc07c10aSMing Lei 		cmd->use_aio = false;
1808f0225cacSChristoph Hellwig 		break;
1809f0225cacSChristoph Hellwig 	default:
1810f0225cacSChristoph Hellwig 		cmd->use_aio = lo->use_dio;
1811f0225cacSChristoph Hellwig 		break;
1812f0225cacSChristoph Hellwig 	}
1813bc07c10aSMing Lei 
1814d4478e92SShaohua Li 	/* always use the first bio's css */
1815c74d40e8SDan Schatzberg 	cmd->blkcg_css = NULL;
1816c74d40e8SDan Schatzberg 	cmd->memcg_css = NULL;
18170b508bc9SShaohua Li #ifdef CONFIG_BLK_CGROUP
1818c74d40e8SDan Schatzberg 	if (rq->bio && rq->bio->bi_blkg) {
1819c74d40e8SDan Schatzberg 		cmd->blkcg_css = &bio_blkcg(rq->bio)->css;
1820c74d40e8SDan Schatzberg #ifdef CONFIG_MEMCG
1821c74d40e8SDan Schatzberg 		cmd->memcg_css =
1822c74d40e8SDan Schatzberg 			cgroup_get_e_css(cmd->blkcg_css->cgroup,
1823c74d40e8SDan Schatzberg 					&memory_cgrp_subsys);
1824d4478e92SShaohua Li #endif
1825c74d40e8SDan Schatzberg 	}
1826c74d40e8SDan Schatzberg #endif
182787579e9bSDan Schatzberg 	loop_queue_work(lo, cmd);
1828b5dd2f60SMing Lei 
1829fc17b653SChristoph Hellwig 	return BLK_STS_OK;
1830b5dd2f60SMing Lei }
1831b5dd2f60SMing Lei 
1832b5dd2f60SMing Lei static void loop_handle_cmd(struct loop_cmd *cmd)
1833b5dd2f60SMing Lei {
18341894e916SJens Axboe 	struct request *rq = blk_mq_rq_from_pdu(cmd);
18351894e916SJens Axboe 	const bool write = op_is_write(req_op(rq));
18361894e916SJens Axboe 	struct loop_device *lo = rq->q->queuedata;
1837f4829a9bSChristoph Hellwig 	int ret = 0;
1838c74d40e8SDan Schatzberg 	struct mem_cgroup *old_memcg = NULL;
1839b5dd2f60SMing Lei 
1840f4829a9bSChristoph Hellwig 	if (write && (lo->lo_flags & LO_FLAGS_READ_ONLY)) {
1841f4829a9bSChristoph Hellwig 		ret = -EIO;
1842b5dd2f60SMing Lei 		goto failed;
1843f4829a9bSChristoph Hellwig 	}
1844b5dd2f60SMing Lei 
1845c74d40e8SDan Schatzberg 	if (cmd->blkcg_css)
1846c74d40e8SDan Schatzberg 		kthread_associate_blkcg(cmd->blkcg_css);
1847c74d40e8SDan Schatzberg 	if (cmd->memcg_css)
1848c74d40e8SDan Schatzberg 		old_memcg = set_active_memcg(
1849c74d40e8SDan Schatzberg 			mem_cgroup_from_css(cmd->memcg_css));
1850c74d40e8SDan Schatzberg 
18511894e916SJens Axboe 	ret = do_req_filebacked(lo, rq);
1852c74d40e8SDan Schatzberg 
1853c74d40e8SDan Schatzberg 	if (cmd->blkcg_css)
1854c74d40e8SDan Schatzberg 		kthread_associate_blkcg(NULL);
1855c74d40e8SDan Schatzberg 
1856c74d40e8SDan Schatzberg 	if (cmd->memcg_css) {
1857c74d40e8SDan Schatzberg 		set_active_memcg(old_memcg);
1858c74d40e8SDan Schatzberg 		css_put(cmd->memcg_css);
1859c74d40e8SDan Schatzberg 	}
1860b5dd2f60SMing Lei  failed:
1861bc07c10aSMing Lei 	/* complete non-aio request */
1862fe2cb290SChristoph Hellwig 	if (!cmd->use_aio || ret) {
18638cd55087SEvan Green 		if (ret == -EOPNOTSUPP)
18648cd55087SEvan Green 			cmd->ret = ret;
18658cd55087SEvan Green 		else
1866fe2cb290SChristoph Hellwig 			cmd->ret = ret ? -EIO : 0;
186715f73f5bSChristoph Hellwig 		if (likely(!blk_should_fake_timeout(rq->q)))
18681894e916SJens Axboe 			blk_mq_complete_request(rq);
1869fe2cb290SChristoph Hellwig 	}
1870b5dd2f60SMing Lei }
1871b5dd2f60SMing Lei 
187287579e9bSDan Schatzberg static void loop_set_timer(struct loop_device *lo)
1873b5dd2f60SMing Lei {
187487579e9bSDan Schatzberg 	timer_reduce(&lo->timer, jiffies + LOOP_IDLE_WORKER_TIMEOUT);
1875b5dd2f60SMing Lei }
1876b5dd2f60SMing Lei 
187787579e9bSDan Schatzberg static void loop_process_work(struct loop_worker *worker,
187887579e9bSDan Schatzberg 			struct list_head *cmd_list, struct loop_device *lo)
1879b5dd2f60SMing Lei {
188087579e9bSDan Schatzberg 	int orig_flags = current->flags;
188187579e9bSDan Schatzberg 	struct loop_cmd *cmd;
1882b5dd2f60SMing Lei 
188387579e9bSDan Schatzberg 	current->flags |= PF_LOCAL_THROTTLE | PF_MEMALLOC_NOIO;
188487579e9bSDan Schatzberg 	spin_lock_irq(&lo->lo_work_lock);
188587579e9bSDan Schatzberg 	while (!list_empty(cmd_list)) {
188687579e9bSDan Schatzberg 		cmd = container_of(
188787579e9bSDan Schatzberg 			cmd_list->next, struct loop_cmd, list_entry);
188887579e9bSDan Schatzberg 		list_del(cmd_list->next);
188987579e9bSDan Schatzberg 		spin_unlock_irq(&lo->lo_work_lock);
189087579e9bSDan Schatzberg 
189187579e9bSDan Schatzberg 		loop_handle_cmd(cmd);
189287579e9bSDan Schatzberg 		cond_resched();
189387579e9bSDan Schatzberg 
189487579e9bSDan Schatzberg 		spin_lock_irq(&lo->lo_work_lock);
189587579e9bSDan Schatzberg 	}
189687579e9bSDan Schatzberg 
189787579e9bSDan Schatzberg 	/*
189887579e9bSDan Schatzberg 	 * We only add to the idle list if there are no pending cmds
189987579e9bSDan Schatzberg 	 * *and* the worker will not run again which ensures that it
190087579e9bSDan Schatzberg 	 * is safe to free any worker on the idle list
190187579e9bSDan Schatzberg 	 */
190287579e9bSDan Schatzberg 	if (worker && !work_pending(&worker->work)) {
190387579e9bSDan Schatzberg 		worker->last_ran_at = jiffies;
190487579e9bSDan Schatzberg 		list_add_tail(&worker->idle_list, &lo->idle_worker_list);
190587579e9bSDan Schatzberg 		loop_set_timer(lo);
190687579e9bSDan Schatzberg 	}
190787579e9bSDan Schatzberg 	spin_unlock_irq(&lo->lo_work_lock);
190887579e9bSDan Schatzberg 	current->flags = orig_flags;
190987579e9bSDan Schatzberg }
191087579e9bSDan Schatzberg 
191187579e9bSDan Schatzberg static void loop_workfn(struct work_struct *work)
191287579e9bSDan Schatzberg {
191387579e9bSDan Schatzberg 	struct loop_worker *worker =
191487579e9bSDan Schatzberg 		container_of(work, struct loop_worker, work);
191587579e9bSDan Schatzberg 	loop_process_work(worker, &worker->cmd_list, worker->lo);
191687579e9bSDan Schatzberg }
191787579e9bSDan Schatzberg 
191887579e9bSDan Schatzberg static void loop_rootcg_workfn(struct work_struct *work)
191987579e9bSDan Schatzberg {
192087579e9bSDan Schatzberg 	struct loop_device *lo =
192187579e9bSDan Schatzberg 		container_of(work, struct loop_device, rootcg_work);
192287579e9bSDan Schatzberg 	loop_process_work(NULL, &lo->rootcg_cmd_list, lo);
192387579e9bSDan Schatzberg }
192487579e9bSDan Schatzberg 
192587579e9bSDan Schatzberg static void loop_free_idle_workers(struct timer_list *timer)
192687579e9bSDan Schatzberg {
192787579e9bSDan Schatzberg 	struct loop_device *lo = container_of(timer, struct loop_device, timer);
192887579e9bSDan Schatzberg 	struct loop_worker *pos, *worker;
192987579e9bSDan Schatzberg 
193087579e9bSDan Schatzberg 	spin_lock_irq(&lo->lo_work_lock);
193187579e9bSDan Schatzberg 	list_for_each_entry_safe(worker, pos, &lo->idle_worker_list,
193287579e9bSDan Schatzberg 				idle_list) {
193387579e9bSDan Schatzberg 		if (time_is_after_jiffies(worker->last_ran_at +
193487579e9bSDan Schatzberg 						LOOP_IDLE_WORKER_TIMEOUT))
193587579e9bSDan Schatzberg 			break;
193687579e9bSDan Schatzberg 		list_del(&worker->idle_list);
193787579e9bSDan Schatzberg 		rb_erase(&worker->rb_node, &lo->worker_tree);
1938c74d40e8SDan Schatzberg 		css_put(worker->blkcg_css);
193987579e9bSDan Schatzberg 		kfree(worker);
194087579e9bSDan Schatzberg 	}
194187579e9bSDan Schatzberg 	if (!list_empty(&lo->idle_worker_list))
194287579e9bSDan Schatzberg 		loop_set_timer(lo);
194387579e9bSDan Schatzberg 	spin_unlock_irq(&lo->lo_work_lock);
1944b5dd2f60SMing Lei }
1945b5dd2f60SMing Lei 
1946f363b089SEric Biggers static const struct blk_mq_ops loop_mq_ops = {
1947b5dd2f60SMing Lei 	.queue_rq       = loop_queue_rq,
1948fe2cb290SChristoph Hellwig 	.complete	= lo_complete_rq,
1949b5dd2f60SMing Lei };
1950b5dd2f60SMing Lei 
1951d6da83d0SChristoph Hellwig static int loop_add(int i)
19521da177e4SLinus Torvalds {
195373285082SKen Chen 	struct loop_device *lo;
195473285082SKen Chen 	struct gendisk *disk;
195534dd82afSKay Sievers 	int err;
19561da177e4SLinus Torvalds 
195734dd82afSKay Sievers 	err = -ENOMEM;
195868d740d7SSilva Paulo 	lo = kzalloc(sizeof(*lo), GFP_KERNEL);
195968d740d7SSilva Paulo 	if (!lo)
196073285082SKen Chen 		goto out;
1961ef7e7c82SMikulas Patocka 	lo->lo_state = Lo_unbound;
1962ef7e7c82SMikulas Patocka 
196318d1f200SChristoph Hellwig 	err = mutex_lock_killable(&loop_ctl_mutex);
196418d1f200SChristoph Hellwig 	if (err)
196518d1f200SChristoph Hellwig 		goto out_free_dev;
196618d1f200SChristoph Hellwig 
1967c718aa65STejun Heo 	/* allocate id, if @id >= 0, we're requesting that specific id */
196834dd82afSKay Sievers 	if (i >= 0) {
1969c718aa65STejun Heo 		err = idr_alloc(&loop_index_idr, lo, i, i + 1, GFP_KERNEL);
1970c718aa65STejun Heo 		if (err == -ENOSPC)
197134dd82afSKay Sievers 			err = -EEXIST;
197234dd82afSKay Sievers 	} else {
1973c718aa65STejun Heo 		err = idr_alloc(&loop_index_idr, lo, 0, 0, GFP_KERNEL);
197434dd82afSKay Sievers 	}
19751c500ad7STetsuo Handa 	mutex_unlock(&loop_ctl_mutex);
197634dd82afSKay Sievers 	if (err < 0)
19771c500ad7STetsuo Handa 		goto out_free_dev;
1978c718aa65STejun Heo 	i = err;
19791da177e4SLinus Torvalds 
1980b5dd2f60SMing Lei 	lo->tag_set.ops = &loop_mq_ops;
1981b5dd2f60SMing Lei 	lo->tag_set.nr_hw_queues = 1;
1982b5dd2f60SMing Lei 	lo->tag_set.queue_depth = 128;
1983b5dd2f60SMing Lei 	lo->tag_set.numa_node = NUMA_NO_NODE;
1984b5dd2f60SMing Lei 	lo->tag_set.cmd_size = sizeof(struct loop_cmd);
19852112f5c1SBart Van Assche 	lo->tag_set.flags = BLK_MQ_F_SHOULD_MERGE | BLK_MQ_F_STACKING |
19862112f5c1SBart Van Assche 		BLK_MQ_F_NO_SCHED_BY_DEFAULT;
1987b5dd2f60SMing Lei 	lo->tag_set.driver_data = lo;
1988b5dd2f60SMing Lei 
1989b5dd2f60SMing Lei 	err = blk_mq_alloc_tag_set(&lo->tag_set);
1990b5dd2f60SMing Lei 	if (err)
19913ec981e3SMikulas Patocka 		goto out_free_idr;
199273285082SKen Chen 
19931c99502fSChristoph Hellwig 	disk = lo->lo_disk = blk_mq_alloc_disk(&lo->tag_set, lo);
19941c99502fSChristoph Hellwig 	if (IS_ERR(disk)) {
19951c99502fSChristoph Hellwig 		err = PTR_ERR(disk);
1996b5dd2f60SMing Lei 		goto out_cleanup_tags;
1997b5dd2f60SMing Lei 	}
19981c99502fSChristoph Hellwig 	lo->lo_queue = lo->lo_disk->queue;
1999ef7e7c82SMikulas Patocka 
200054bb0adeSShaohua Li 	blk_queue_max_hw_sectors(lo->lo_queue, BLK_DEF_MAX_SECTORS);
200140326d8aSShaohua Li 
20025b5e20f4SMing Lei 	/*
200340326d8aSShaohua Li 	 * By default, we do buffer IO, so it doesn't make sense to enable
200440326d8aSShaohua Li 	 * merge because the I/O submitted to backing file is handled page by
200540326d8aSShaohua Li 	 * page. For directio mode, merge does help to dispatch bigger request
200640326d8aSShaohua Li 	 * to underlayer disk. We will enable merge once directio is enabled.
20075b5e20f4SMing Lei 	 */
20088b904b5bSBart Van Assche 	blk_queue_flag_set(QUEUE_FLAG_NOMERGES, lo->lo_queue);
20095b5e20f4SMing Lei 
2010e03c8dd1SKay Sievers 	/*
2011e03c8dd1SKay Sievers 	 * Disable partition scanning by default. The in-kernel partition
2012e03c8dd1SKay Sievers 	 * scanning can be requested individually per-device during its
2013e03c8dd1SKay Sievers 	 * setup. Userspace can always add and remove partitions from all
2014e03c8dd1SKay Sievers 	 * devices. The needed partition minors are allocated from the
2015e03c8dd1SKay Sievers 	 * extended minor space, the main loop device numbers will continue
2016e03c8dd1SKay Sievers 	 * to match the loop minors, regardless of the number of partitions
2017e03c8dd1SKay Sievers 	 * used.
2018e03c8dd1SKay Sievers 	 *
2019e03c8dd1SKay Sievers 	 * If max_part is given, partition scanning is globally enabled for
2020e03c8dd1SKay Sievers 	 * all loop devices. The minors for the main loop devices will be
2021e03c8dd1SKay Sievers 	 * multiples of max_part.
2022e03c8dd1SKay Sievers 	 *
2023e03c8dd1SKay Sievers 	 * Note: Global-for-all-devices, set-only-at-init, read-only module
2024e03c8dd1SKay Sievers 	 * parameteters like 'max_loop' and 'max_part' make things needlessly
2025e03c8dd1SKay Sievers 	 * complicated, are too static, inflexible and may surprise
2026e03c8dd1SKay Sievers 	 * userspace tools. Parameters like this in general should be avoided.
2027e03c8dd1SKay Sievers 	 */
2028e03c8dd1SKay Sievers 	if (!part_shift)
202946e7eac6SChristoph Hellwig 		disk->flags |= GENHD_FL_NO_PART;
2030f8933667SMing Lei 	atomic_set(&lo->lo_refcnt, 0);
20316cc8e743SPavel Tatashin 	mutex_init(&lo->lo_mutex);
20321da177e4SLinus Torvalds 	lo->lo_number		= i;
20331da177e4SLinus Torvalds 	spin_lock_init(&lo->lo_lock);
203487579e9bSDan Schatzberg 	spin_lock_init(&lo->lo_work_lock);
20351da177e4SLinus Torvalds 	disk->major		= LOOP_MAJOR;
2036476a4813SLaurent Vivier 	disk->first_minor	= i << part_shift;
20371c99502fSChristoph Hellwig 	disk->minors		= 1 << part_shift;
20381da177e4SLinus Torvalds 	disk->fops		= &lo_fops;
20391da177e4SLinus Torvalds 	disk->private_data	= lo;
20401da177e4SLinus Torvalds 	disk->queue		= lo->lo_queue;
20419f65c489SMatteo Croce 	disk->events		= DISK_EVENT_MEDIA_CHANGE;
20429f65c489SMatteo Croce 	disk->event_flags	= DISK_EVENT_FLAG_UEVENT;
204373285082SKen Chen 	sprintf(disk->disk_name, "loop%d", i);
20441c500ad7STetsuo Handa 	/* Make this loop device reachable from pathname. */
2045905705f0SLuis Chamberlain 	err = add_disk(disk);
2046905705f0SLuis Chamberlain 	if (err)
2047905705f0SLuis Chamberlain 		goto out_cleanup_disk;
2048905705f0SLuis Chamberlain 
20491c500ad7STetsuo Handa 	/* Show this loop device. */
20501c500ad7STetsuo Handa 	mutex_lock(&loop_ctl_mutex);
20511c500ad7STetsuo Handa 	lo->idr_visible = true;
205218d1f200SChristoph Hellwig 	mutex_unlock(&loop_ctl_mutex);
2053905705f0SLuis Chamberlain 
205418d1f200SChristoph Hellwig 	return i;
205573285082SKen Chen 
2056905705f0SLuis Chamberlain out_cleanup_disk:
2057905705f0SLuis Chamberlain 	blk_cleanup_disk(disk);
2058b5dd2f60SMing Lei out_cleanup_tags:
2059b5dd2f60SMing Lei 	blk_mq_free_tag_set(&lo->tag_set);
20603ec981e3SMikulas Patocka out_free_idr:
20611c500ad7STetsuo Handa 	mutex_lock(&loop_ctl_mutex);
20623ec981e3SMikulas Patocka 	idr_remove(&loop_index_idr, i);
206318d1f200SChristoph Hellwig 	mutex_unlock(&loop_ctl_mutex);
206473285082SKen Chen out_free_dev:
206573285082SKen Chen 	kfree(lo);
206673285082SKen Chen out:
206734dd82afSKay Sievers 	return err;
20681da177e4SLinus Torvalds }
20691da177e4SLinus Torvalds 
207034dd82afSKay Sievers static void loop_remove(struct loop_device *lo)
207173285082SKen Chen {
20721c500ad7STetsuo Handa 	/* Make this loop device unreachable from pathname. */
20736cd18e71SNeilBrown 	del_gendisk(lo->lo_disk);
20741c99502fSChristoph Hellwig 	blk_cleanup_disk(lo->lo_disk);
2075b5dd2f60SMing Lei 	blk_mq_free_tag_set(&lo->tag_set);
20761c500ad7STetsuo Handa 	mutex_lock(&loop_ctl_mutex);
20771c500ad7STetsuo Handa 	idr_remove(&loop_index_idr, lo->lo_number);
20781c500ad7STetsuo Handa 	mutex_unlock(&loop_ctl_mutex);
20791c500ad7STetsuo Handa 	/* There is no route which can find this loop device. */
20806cc8e743SPavel Tatashin 	mutex_destroy(&lo->lo_mutex);
208173285082SKen Chen 	kfree(lo);
208273285082SKen Chen }
208373285082SKen Chen 
20848410d38cSChristoph Hellwig static void loop_probe(dev_t dev)
208573285082SKen Chen {
20868410d38cSChristoph Hellwig 	int idx = MINOR(dev) >> part_shift;
20878410d38cSChristoph Hellwig 
20888410d38cSChristoph Hellwig 	if (max_loop && idx >= max_loop)
20898410d38cSChristoph Hellwig 		return;
2090d6da83d0SChristoph Hellwig 	loop_add(idx);
2091f9d10764SChristoph Hellwig }
2092f9d10764SChristoph Hellwig 
2093f9d10764SChristoph Hellwig static int loop_control_remove(int idx)
2094770fe30aSKay Sievers {
2095770fe30aSKay Sievers 	struct loop_device *lo;
20960a42e99bSJan Kara 	int ret;
2097e5d66a10SChristoph Hellwig 
2098e5d66a10SChristoph Hellwig 	if (idx < 0) {
2099e3f9387aSTetsuo Handa 		pr_warn_once("deleting an unspecified loop device is not supported.\n");
2100e5d66a10SChristoph Hellwig 		return -EINVAL;
2101e5d66a10SChristoph Hellwig 	}
2102770fe30aSKay Sievers 
21031c500ad7STetsuo Handa 	/* Hide this loop device for serialization. */
21040a42e99bSJan Kara 	ret = mutex_lock_killable(&loop_ctl_mutex);
21050a42e99bSJan Kara 	if (ret)
21060a42e99bSJan Kara 		return ret;
2107b9848081SChristoph Hellwig 	lo = idr_find(&loop_index_idr, idx);
21081c500ad7STetsuo Handa 	if (!lo || !lo->idr_visible)
2109b9848081SChristoph Hellwig 		ret = -ENODEV;
21101c500ad7STetsuo Handa 	else
21111c500ad7STetsuo Handa 		lo->idr_visible = false;
21121c500ad7STetsuo Handa 	mutex_unlock(&loop_ctl_mutex);
21131c500ad7STetsuo Handa 	if (ret)
21141c500ad7STetsuo Handa 		return ret;
2115f9d10764SChristoph Hellwig 
21161c500ad7STetsuo Handa 	/* Check whether this loop device can be removed. */
21176cc8e743SPavel Tatashin 	ret = mutex_lock_killable(&lo->lo_mutex);
21186cc8e743SPavel Tatashin 	if (ret)
21191c500ad7STetsuo Handa 		goto mark_visible;
2120f9d10764SChristoph Hellwig 	if (lo->lo_state != Lo_unbound ||
2121f9d10764SChristoph Hellwig 	    atomic_read(&lo->lo_refcnt) > 0) {
21226cc8e743SPavel Tatashin 		mutex_unlock(&lo->lo_mutex);
2123770fe30aSKay Sievers 		ret = -EBUSY;
21241c500ad7STetsuo Handa 		goto mark_visible;
2125770fe30aSKay Sievers 	}
21261c500ad7STetsuo Handa 	/* Mark this loop device no longer open()-able. */
2127990e7811SChristoph Hellwig 	lo->lo_state = Lo_deleting;
21286cc8e743SPavel Tatashin 	mutex_unlock(&lo->lo_mutex);
2129f9d10764SChristoph Hellwig 
2130770fe30aSKay Sievers 	loop_remove(lo);
21311c500ad7STetsuo Handa 	return 0;
21321c500ad7STetsuo Handa 
21331c500ad7STetsuo Handa mark_visible:
21341c500ad7STetsuo Handa 	/* Show this loop device again. */
21351c500ad7STetsuo Handa 	mutex_lock(&loop_ctl_mutex);
21361c500ad7STetsuo Handa 	lo->idr_visible = true;
2137f9d10764SChristoph Hellwig 	mutex_unlock(&loop_ctl_mutex);
2138f9d10764SChristoph Hellwig 	return ret;
2139770fe30aSKay Sievers }
2140f9d10764SChristoph Hellwig 
2141f9d10764SChristoph Hellwig static int loop_control_get_free(int idx)
2142f9d10764SChristoph Hellwig {
2143f9d10764SChristoph Hellwig 	struct loop_device *lo;
2144b9848081SChristoph Hellwig 	int id, ret;
2145f9d10764SChristoph Hellwig 
2146f9d10764SChristoph Hellwig 	ret = mutex_lock_killable(&loop_ctl_mutex);
2147f9d10764SChristoph Hellwig 	if (ret)
2148f9d10764SChristoph Hellwig 		return ret;
2149b9848081SChristoph Hellwig 	idr_for_each_entry(&loop_index_idr, lo, id) {
21501c500ad7STetsuo Handa 		/* Hitting a race results in creating a new loop device which is harmless. */
21511c500ad7STetsuo Handa 		if (lo->idr_visible && data_race(lo->lo_state) == Lo_unbound)
2152b9848081SChristoph Hellwig 			goto found;
2153b9848081SChristoph Hellwig 	}
21540a42e99bSJan Kara 	mutex_unlock(&loop_ctl_mutex);
215518d1f200SChristoph Hellwig 	return loop_add(-1);
2156b9848081SChristoph Hellwig found:
2157b9848081SChristoph Hellwig 	mutex_unlock(&loop_ctl_mutex);
2158b9848081SChristoph Hellwig 	return id;
2159770fe30aSKay Sievers }
2160770fe30aSKay Sievers 
2161f9d10764SChristoph Hellwig static long loop_control_ioctl(struct file *file, unsigned int cmd,
2162f9d10764SChristoph Hellwig 			       unsigned long parm)
2163f9d10764SChristoph Hellwig {
2164f9d10764SChristoph Hellwig 	switch (cmd) {
2165f9d10764SChristoph Hellwig 	case LOOP_CTL_ADD:
216618d1f200SChristoph Hellwig 		return loop_add(parm);
2167f9d10764SChristoph Hellwig 	case LOOP_CTL_REMOVE:
2168f9d10764SChristoph Hellwig 		return loop_control_remove(parm);
2169f9d10764SChristoph Hellwig 	case LOOP_CTL_GET_FREE:
2170f9d10764SChristoph Hellwig 		return loop_control_get_free(parm);
2171f9d10764SChristoph Hellwig 	default:
2172f9d10764SChristoph Hellwig 		return -ENOSYS;
2173f9d10764SChristoph Hellwig 	}
2174f9d10764SChristoph Hellwig }
2175f9d10764SChristoph Hellwig 
2176770fe30aSKay Sievers static const struct file_operations loop_ctl_fops = {
2177770fe30aSKay Sievers 	.open		= nonseekable_open,
2178770fe30aSKay Sievers 	.unlocked_ioctl	= loop_control_ioctl,
2179770fe30aSKay Sievers 	.compat_ioctl	= loop_control_ioctl,
2180770fe30aSKay Sievers 	.owner		= THIS_MODULE,
2181770fe30aSKay Sievers 	.llseek		= noop_llseek,
2182770fe30aSKay Sievers };
2183770fe30aSKay Sievers 
2184770fe30aSKay Sievers static struct miscdevice loop_misc = {
2185770fe30aSKay Sievers 	.minor		= LOOP_CTRL_MINOR,
2186770fe30aSKay Sievers 	.name		= "loop-control",
2187770fe30aSKay Sievers 	.fops		= &loop_ctl_fops,
2188770fe30aSKay Sievers };
2189770fe30aSKay Sievers 
2190770fe30aSKay Sievers MODULE_ALIAS_MISCDEV(LOOP_CTRL_MINOR);
2191770fe30aSKay Sievers MODULE_ALIAS("devname:loop-control");
2192770fe30aSKay Sievers 
219373285082SKen Chen static int __init loop_init(void)
219473285082SKen Chen {
2195a47653fcSKen Chen 	int i, nr;
2196770fe30aSKay Sievers 	int err;
2197a47653fcSKen Chen 
2198476a4813SLaurent Vivier 	part_shift = 0;
2199ac04fee0SNamhyung Kim 	if (max_part > 0) {
2200476a4813SLaurent Vivier 		part_shift = fls(max_part);
2201476a4813SLaurent Vivier 
2202ac04fee0SNamhyung Kim 		/*
2203ac04fee0SNamhyung Kim 		 * Adjust max_part according to part_shift as it is exported
2204ac04fee0SNamhyung Kim 		 * to user space so that user can decide correct minor number
2205ac04fee0SNamhyung Kim 		 * if [s]he want to create more devices.
2206ac04fee0SNamhyung Kim 		 *
2207ac04fee0SNamhyung Kim 		 * Note that -1 is required because partition 0 is reserved
2208ac04fee0SNamhyung Kim 		 * for the whole disk.
2209ac04fee0SNamhyung Kim 		 */
2210ac04fee0SNamhyung Kim 		max_part = (1UL << part_shift) - 1;
2211ac04fee0SNamhyung Kim 	}
2212ac04fee0SNamhyung Kim 
2213b1a66504SGuo Chao 	if ((1UL << part_shift) > DISK_MAX_PARTS) {
2214b1a66504SGuo Chao 		err = -EINVAL;
2215a8c1d064SAnton Volkov 		goto err_out;
2216b1a66504SGuo Chao 	}
221778f4bb36SNamhyung Kim 
2218b1a66504SGuo Chao 	if (max_loop > 1UL << (MINORBITS - part_shift)) {
2219b1a66504SGuo Chao 		err = -EINVAL;
2220a8c1d064SAnton Volkov 		goto err_out;
2221b1a66504SGuo Chao 	}
222273285082SKen Chen 
2223d134b00bSKay Sievers 	/*
2224d134b00bSKay Sievers 	 * If max_loop is specified, create that many devices upfront.
2225d134b00bSKay Sievers 	 * This also becomes a hard limit. If max_loop is not specified,
2226d134b00bSKay Sievers 	 * create CONFIG_BLK_DEV_LOOP_MIN_COUNT loop devices at module
2227d134b00bSKay Sievers 	 * init time. Loop devices can be requested on-demand with the
2228d134b00bSKay Sievers 	 * /dev/loop-control interface, or be instantiated by accessing
2229d134b00bSKay Sievers 	 * a 'dead' device node.
2230d134b00bSKay Sievers 	 */
2231aeb2b0b1SLukas Bulwahn 	if (max_loop)
2232a47653fcSKen Chen 		nr = max_loop;
2233aeb2b0b1SLukas Bulwahn 	else
2234d134b00bSKay Sievers 		nr = CONFIG_BLK_DEV_LOOP_MIN_COUNT;
2235a47653fcSKen Chen 
2236a8c1d064SAnton Volkov 	err = misc_register(&loop_misc);
2237a8c1d064SAnton Volkov 	if (err < 0)
2238a8c1d064SAnton Volkov 		goto err_out;
2239a8c1d064SAnton Volkov 
2240a8c1d064SAnton Volkov 
22418410d38cSChristoph Hellwig 	if (__register_blkdev(LOOP_MAJOR, "loop", loop_probe)) {
2242b1a66504SGuo Chao 		err = -EIO;
2243b1a66504SGuo Chao 		goto misc_out;
2244b1a66504SGuo Chao 	}
2245a47653fcSKen Chen 
2246d134b00bSKay Sievers 	/* pre-create number of devices given by config or max_loop */
224734dd82afSKay Sievers 	for (i = 0; i < nr; i++)
2248d6da83d0SChristoph Hellwig 		loop_add(i);
224934dd82afSKay Sievers 
225073285082SKen Chen 	printk(KERN_INFO "loop: module loaded\n");
22511da177e4SLinus Torvalds 	return 0;
2252b1a66504SGuo Chao 
2253b1a66504SGuo Chao misc_out:
2254b1a66504SGuo Chao 	misc_deregister(&loop_misc);
2255a8c1d064SAnton Volkov err_out:
2256b1a66504SGuo Chao 	return err;
225734dd82afSKay Sievers }
2258a47653fcSKen Chen 
225973285082SKen Chen static void __exit loop_exit(void)
22601da177e4SLinus Torvalds {
22618e60947dSChristoph Hellwig 	struct loop_device *lo;
22628e60947dSChristoph Hellwig 	int id;
22638e60947dSChristoph Hellwig 
22648b52d8beSChristoph Hellwig 	unregister_blkdev(LOOP_MAJOR, "loop");
22658b52d8beSChristoph Hellwig 	misc_deregister(&loop_misc);
2266200f9337SLuis Chamberlain 
22671c500ad7STetsuo Handa 	/*
22681c500ad7STetsuo Handa 	 * There is no need to use loop_ctl_mutex here, for nobody else can
22691c500ad7STetsuo Handa 	 * access loop_index_idr when this module is unloading (unless forced
22701c500ad7STetsuo Handa 	 * module unloading is requested). If this is not a clean unloading,
22711c500ad7STetsuo Handa 	 * we have no means to avoid kernel crash.
22721c500ad7STetsuo Handa 	 */
22738e60947dSChristoph Hellwig 	idr_for_each_entry(&loop_index_idr, lo, id)
22748e60947dSChristoph Hellwig 		loop_remove(lo);
2275bd5c39edSChristoph Hellwig 
2276bd5c39edSChristoph Hellwig 	idr_destroy(&loop_index_idr);
22771da177e4SLinus Torvalds }
22781da177e4SLinus Torvalds 
22791da177e4SLinus Torvalds module_init(loop_init);
22801da177e4SLinus Torvalds module_exit(loop_exit);
22811da177e4SLinus Torvalds 
22821da177e4SLinus Torvalds #ifndef MODULE
22831da177e4SLinus Torvalds static int __init max_loop_setup(char *str)
22841da177e4SLinus Torvalds {
22851da177e4SLinus Torvalds 	max_loop = simple_strtol(str, NULL, 0);
22861da177e4SLinus Torvalds 	return 1;
22871da177e4SLinus Torvalds }
22881da177e4SLinus Torvalds 
22891da177e4SLinus Torvalds __setup("max_loop=", max_loop_setup);
22901da177e4SLinus Torvalds #endif
2291