xref: /openbmc/linux/drivers/usb/gadget/function/f_mass_storage.c (revision 28efb0046512e8a13ed9f9bdf0d68d10bbfbe9cf)
1 /*
2  * f_mass_storage.c -- Mass Storage USB Composite Function
3  *
4  * Copyright (C) 2003-2008 Alan Stern
5  * Copyright (C) 2009 Samsung Electronics
6  *                    Author: Michal Nazarewicz <mina86@mina86.com>
7  * All rights reserved.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions
11  * are met:
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions, and the following disclaimer,
14  *    without modification.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  * 3. The names of the above-listed copyright holders may not be used
19  *    to endorse or promote products derived from this software without
20  *    specific prior written permission.
21  *
22  * ALTERNATIVELY, this software may be distributed under the terms of the
23  * GNU General Public License ("GPL") as published by the Free Software
24  * Foundation, either version 2 of that License or (at your option) any
25  * later version.
26  *
27  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
28  * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
29  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
30  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
31  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
32  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
33  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
34  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
35  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
36  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
37  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
38  */
39 
40 /*
41  * The Mass Storage Function acts as a USB Mass Storage device,
42  * appearing to the host as a disk drive or as a CD-ROM drive.  In
43  * addition to providing an example of a genuinely useful composite
44  * function for a USB device, it also illustrates a technique of
45  * double-buffering for increased throughput.
46  *
47  * For more information about MSF and in particular its module
48  * parameters and sysfs interface read the
49  * <Documentation/usb/mass-storage.txt> file.
50  */
51 
52 /*
53  * MSF is configured by specifying a fsg_config structure.  It has the
54  * following fields:
55  *
56  *	nluns		Number of LUNs function have (anywhere from 1
57  *				to FSG_MAX_LUNS).
58  *	luns		An array of LUN configuration values.  This
59  *				should be filled for each LUN that
60  *				function will include (ie. for "nluns"
61  *				LUNs).  Each element of the array has
62  *				the following fields:
63  *	->filename	The path to the backing file for the LUN.
64  *				Required if LUN is not marked as
65  *				removable.
66  *	->ro		Flag specifying access to the LUN shall be
67  *				read-only.  This is implied if CD-ROM
68  *				emulation is enabled as well as when
69  *				it was impossible to open "filename"
70  *				in R/W mode.
71  *	->removable	Flag specifying that LUN shall be indicated as
72  *				being removable.
73  *	->cdrom		Flag specifying that LUN shall be reported as
74  *				being a CD-ROM.
75  *	->nofua		Flag specifying that FUA flag in SCSI WRITE(10,12)
76  *				commands for this LUN shall be ignored.
77  *
78  *	vendor_name
79  *	product_name
80  *	release		Information used as a reply to INQUIRY
81  *				request.  To use default set to NULL,
82  *				NULL, 0xffff respectively.  The first
83  *				field should be 8 and the second 16
84  *				characters or less.
85  *
86  *	can_stall	Set to permit function to halt bulk endpoints.
87  *				Disabled on some USB devices known not
88  *				to work correctly.  You should set it
89  *				to true.
90  *
91  * If "removable" is not set for a LUN then a backing file must be
92  * specified.  If it is set, then NULL filename means the LUN's medium
93  * is not loaded (an empty string as "filename" in the fsg_config
94  * structure causes error).  The CD-ROM emulation includes a single
95  * data track and no audio tracks; hence there need be only one
96  * backing file per LUN.
97  *
98  * This function is heavily based on "File-backed Storage Gadget" by
99  * Alan Stern which in turn is heavily based on "Gadget Zero" by David
100  * Brownell.  The driver's SCSI command interface was based on the
101  * "Information technology - Small Computer System Interface - 2"
102  * document from X3T9.2 Project 375D, Revision 10L, 7-SEP-93,
103  * available at <http://www.t10.org/ftp/t10/drafts/s2/s2-r10l.pdf>.
104  * The single exception is opcode 0x23 (READ FORMAT CAPACITIES), which
105  * was based on the "Universal Serial Bus Mass Storage Class UFI
106  * Command Specification" document, Revision 1.0, December 14, 1998,
107  * available at
108  * <http://www.usb.org/developers/devclass_docs/usbmass-ufi10.pdf>.
109  */
110 
111 /*
112  *				Driver Design
113  *
114  * The MSF is fairly straightforward.  There is a main kernel
115  * thread that handles most of the work.  Interrupt routines field
116  * callbacks from the controller driver: bulk- and interrupt-request
117  * completion notifications, endpoint-0 events, and disconnect events.
118  * Completion events are passed to the main thread by wakeup calls.  Many
119  * ep0 requests are handled at interrupt time, but SetInterface,
120  * SetConfiguration, and device reset requests are forwarded to the
121  * thread in the form of "exceptions" using SIGUSR1 signals (since they
122  * should interrupt any ongoing file I/O operations).
123  *
124  * The thread's main routine implements the standard command/data/status
125  * parts of a SCSI interaction.  It and its subroutines are full of tests
126  * for pending signals/exceptions -- all this polling is necessary since
127  * the kernel has no setjmp/longjmp equivalents.  (Maybe this is an
128  * indication that the driver really wants to be running in userspace.)
129  * An important point is that so long as the thread is alive it keeps an
130  * open reference to the backing file.  This will prevent unmounting
131  * the backing file's underlying filesystem and could cause problems
132  * during system shutdown, for example.  To prevent such problems, the
133  * thread catches INT, TERM, and KILL signals and converts them into
134  * an EXIT exception.
135  *
136  * In normal operation the main thread is started during the gadget's
137  * fsg_bind() callback and stopped during fsg_unbind().  But it can
138  * also exit when it receives a signal, and there's no point leaving
139  * the gadget running when the thread is dead.  As of this moment, MSF
140  * provides no way to deregister the gadget when thread dies -- maybe
141  * a callback functions is needed.
142  *
143  * To provide maximum throughput, the driver uses a circular pipeline of
144  * buffer heads (struct fsg_buffhd).  In principle the pipeline can be
145  * arbitrarily long; in practice the benefits don't justify having more
146  * than 2 stages (i.e., double buffering).  But it helps to think of the
147  * pipeline as being a long one.  Each buffer head contains a bulk-in and
148  * a bulk-out request pointer (since the buffer can be used for both
149  * output and input -- directions always are given from the host's
150  * point of view) as well as a pointer to the buffer and various state
151  * variables.
152  *
153  * Use of the pipeline follows a simple protocol.  There is a variable
154  * (fsg->next_buffhd_to_fill) that points to the next buffer head to use.
155  * At any time that buffer head may still be in use from an earlier
156  * request, so each buffer head has a state variable indicating whether
157  * it is EMPTY, FULL, or BUSY.  Typical use involves waiting for the
158  * buffer head to be EMPTY, filling the buffer either by file I/O or by
159  * USB I/O (during which the buffer head is BUSY), and marking the buffer
160  * head FULL when the I/O is complete.  Then the buffer will be emptied
161  * (again possibly by USB I/O, during which it is marked BUSY) and
162  * finally marked EMPTY again (possibly by a completion routine).
163  *
164  * A module parameter tells the driver to avoid stalling the bulk
165  * endpoints wherever the transport specification allows.  This is
166  * necessary for some UDCs like the SuperH, which cannot reliably clear a
167  * halt on a bulk endpoint.  However, under certain circumstances the
168  * Bulk-only specification requires a stall.  In such cases the driver
169  * will halt the endpoint and set a flag indicating that it should clear
170  * the halt in software during the next device reset.  Hopefully this
171  * will permit everything to work correctly.  Furthermore, although the
172  * specification allows the bulk-out endpoint to halt when the host sends
173  * too much data, implementing this would cause an unavoidable race.
174  * The driver will always use the "no-stall" approach for OUT transfers.
175  *
176  * One subtle point concerns sending status-stage responses for ep0
177  * requests.  Some of these requests, such as device reset, can involve
178  * interrupting an ongoing file I/O operation, which might take an
179  * arbitrarily long time.  During that delay the host might give up on
180  * the original ep0 request and issue a new one.  When that happens the
181  * driver should not notify the host about completion of the original
182  * request, as the host will no longer be waiting for it.  So the driver
183  * assigns to each ep0 request a unique tag, and it keeps track of the
184  * tag value of the request associated with a long-running exception
185  * (device-reset, interface-change, or configuration-change).  When the
186  * exception handler is finished, the status-stage response is submitted
187  * only if the current ep0 request tag is equal to the exception request
188  * tag.  Thus only the most recently received ep0 request will get a
189  * status-stage response.
190  *
191  * Warning: This driver source file is too long.  It ought to be split up
192  * into a header file plus about 3 separate .c files, to handle the details
193  * of the Gadget, USB Mass Storage, and SCSI protocols.
194  */
195 
196 
197 /* #define VERBOSE_DEBUG */
198 /* #define DUMP_MSGS */
199 
200 #include <linux/blkdev.h>
201 #include <linux/completion.h>
202 #include <linux/dcache.h>
203 #include <linux/delay.h>
204 #include <linux/device.h>
205 #include <linux/fcntl.h>
206 #include <linux/file.h>
207 #include <linux/fs.h>
208 #include <linux/kref.h>
209 #include <linux/kthread.h>
210 #include <linux/sched/signal.h>
211 #include <linux/limits.h>
212 #include <linux/rwsem.h>
213 #include <linux/slab.h>
214 #include <linux/spinlock.h>
215 #include <linux/string.h>
216 #include <linux/freezer.h>
217 #include <linux/module.h>
218 #include <linux/uaccess.h>
219 
220 #include <linux/usb/ch9.h>
221 #include <linux/usb/gadget.h>
222 #include <linux/usb/composite.h>
223 
224 #include "configfs.h"
225 
226 
227 /*------------------------------------------------------------------------*/
228 
229 #define FSG_DRIVER_DESC		"Mass Storage Function"
230 #define FSG_DRIVER_VERSION	"2009/09/11"
231 
232 static const char fsg_string_interface[] = "Mass Storage";
233 
234 #include "storage_common.h"
235 #include "f_mass_storage.h"
236 
237 /* Static strings, in UTF-8 (for simplicity we use only ASCII characters) */
238 static struct usb_string		fsg_strings[] = {
239 	{FSG_STRING_INTERFACE,		fsg_string_interface},
240 	{}
241 };
242 
243 static struct usb_gadget_strings	fsg_stringtab = {
244 	.language	= 0x0409,		/* en-us */
245 	.strings	= fsg_strings,
246 };
247 
248 static struct usb_gadget_strings *fsg_strings_array[] = {
249 	&fsg_stringtab,
250 	NULL,
251 };
252 
253 /*-------------------------------------------------------------------------*/
254 
255 struct fsg_dev;
256 struct fsg_common;
257 
258 /* Data shared by all the FSG instances. */
259 struct fsg_common {
260 	struct usb_gadget	*gadget;
261 	struct usb_composite_dev *cdev;
262 	struct fsg_dev		*fsg, *new_fsg;
263 	wait_queue_head_t	io_wait;
264 	wait_queue_head_t	fsg_wait;
265 
266 	/* filesem protects: backing files in use */
267 	struct rw_semaphore	filesem;
268 
269 	/* lock protects: state and thread_task */
270 	spinlock_t		lock;
271 
272 	struct usb_ep		*ep0;		/* Copy of gadget->ep0 */
273 	struct usb_request	*ep0req;	/* Copy of cdev->req */
274 	unsigned int		ep0_req_tag;
275 
276 	struct fsg_buffhd	*next_buffhd_to_fill;
277 	struct fsg_buffhd	*next_buffhd_to_drain;
278 	struct fsg_buffhd	*buffhds;
279 	unsigned int		fsg_num_buffers;
280 
281 	int			cmnd_size;
282 	u8			cmnd[MAX_COMMAND_SIZE];
283 
284 	unsigned int		lun;
285 	struct fsg_lun		*luns[FSG_MAX_LUNS];
286 	struct fsg_lun		*curlun;
287 
288 	unsigned int		bulk_out_maxpacket;
289 	enum fsg_state		state;		/* For exception handling */
290 	unsigned int		exception_req_tag;
291 
292 	enum data_direction	data_dir;
293 	u32			data_size;
294 	u32			data_size_from_cmnd;
295 	u32			tag;
296 	u32			residue;
297 	u32			usb_amount_left;
298 
299 	unsigned int		can_stall:1;
300 	unsigned int		free_storage_on_release:1;
301 	unsigned int		phase_error:1;
302 	unsigned int		short_packet_received:1;
303 	unsigned int		bad_lun_okay:1;
304 	unsigned int		running:1;
305 	unsigned int		sysfs:1;
306 
307 	struct completion	thread_notifier;
308 	struct task_struct	*thread_task;
309 
310 	/* Callback functions. */
311 	const struct fsg_operations	*ops;
312 	/* Gadget's private data. */
313 	void			*private_data;
314 
315 	char inquiry_string[INQUIRY_STRING_LEN];
316 
317 	struct kref		ref;
318 };
319 
320 struct fsg_dev {
321 	struct usb_function	function;
322 	struct usb_gadget	*gadget;	/* Copy of cdev->gadget */
323 	struct fsg_common	*common;
324 
325 	u16			interface_number;
326 
327 	unsigned int		bulk_in_enabled:1;
328 	unsigned int		bulk_out_enabled:1;
329 
330 	unsigned long		atomic_bitflags;
331 #define IGNORE_BULK_OUT		0
332 
333 	struct usb_ep		*bulk_in;
334 	struct usb_ep		*bulk_out;
335 };
336 
337 static inline int __fsg_is_set(struct fsg_common *common,
338 			       const char *func, unsigned line)
339 {
340 	if (common->fsg)
341 		return 1;
342 	ERROR(common, "common->fsg is NULL in %s at %u\n", func, line);
343 	WARN_ON(1);
344 	return 0;
345 }
346 
347 #define fsg_is_set(common) likely(__fsg_is_set(common, __func__, __LINE__))
348 
349 static inline struct fsg_dev *fsg_from_func(struct usb_function *f)
350 {
351 	return container_of(f, struct fsg_dev, function);
352 }
353 
354 typedef void (*fsg_routine_t)(struct fsg_dev *);
355 
356 static int exception_in_progress(struct fsg_common *common)
357 {
358 	return common->state > FSG_STATE_NORMAL;
359 }
360 
361 /* Make bulk-out requests be divisible by the maxpacket size */
362 static void set_bulk_out_req_length(struct fsg_common *common,
363 				    struct fsg_buffhd *bh, unsigned int length)
364 {
365 	unsigned int	rem;
366 
367 	bh->bulk_out_intended_length = length;
368 	rem = length % common->bulk_out_maxpacket;
369 	if (rem > 0)
370 		length += common->bulk_out_maxpacket - rem;
371 	bh->outreq->length = length;
372 }
373 
374 
375 /*-------------------------------------------------------------------------*/
376 
377 static int fsg_set_halt(struct fsg_dev *fsg, struct usb_ep *ep)
378 {
379 	const char	*name;
380 
381 	if (ep == fsg->bulk_in)
382 		name = "bulk-in";
383 	else if (ep == fsg->bulk_out)
384 		name = "bulk-out";
385 	else
386 		name = ep->name;
387 	DBG(fsg, "%s set halt\n", name);
388 	return usb_ep_set_halt(ep);
389 }
390 
391 
392 /*-------------------------------------------------------------------------*/
393 
394 /* These routines may be called in process context or in_irq */
395 
396 static void raise_exception(struct fsg_common *common, enum fsg_state new_state)
397 {
398 	unsigned long		flags;
399 
400 	/*
401 	 * Do nothing if a higher-priority exception is already in progress.
402 	 * If a lower-or-equal priority exception is in progress, preempt it
403 	 * and notify the main thread by sending it a signal.
404 	 */
405 	spin_lock_irqsave(&common->lock, flags);
406 	if (common->state <= new_state) {
407 		common->exception_req_tag = common->ep0_req_tag;
408 		common->state = new_state;
409 		if (common->thread_task)
410 			send_sig_info(SIGUSR1, SEND_SIG_FORCED,
411 				      common->thread_task);
412 	}
413 	spin_unlock_irqrestore(&common->lock, flags);
414 }
415 
416 
417 /*-------------------------------------------------------------------------*/
418 
419 static int ep0_queue(struct fsg_common *common)
420 {
421 	int	rc;
422 
423 	rc = usb_ep_queue(common->ep0, common->ep0req, GFP_ATOMIC);
424 	common->ep0->driver_data = common;
425 	if (rc != 0 && rc != -ESHUTDOWN) {
426 		/* We can't do much more than wait for a reset */
427 		WARNING(common, "error in submission: %s --> %d\n",
428 			common->ep0->name, rc);
429 	}
430 	return rc;
431 }
432 
433 
434 /*-------------------------------------------------------------------------*/
435 
436 /* Completion handlers. These always run in_irq. */
437 
438 static void bulk_in_complete(struct usb_ep *ep, struct usb_request *req)
439 {
440 	struct fsg_common	*common = ep->driver_data;
441 	struct fsg_buffhd	*bh = req->context;
442 
443 	if (req->status || req->actual != req->length)
444 		DBG(common, "%s --> %d, %u/%u\n", __func__,
445 		    req->status, req->actual, req->length);
446 	if (req->status == -ECONNRESET)		/* Request was cancelled */
447 		usb_ep_fifo_flush(ep);
448 
449 	/* Synchronize with the smp_load_acquire() in sleep_thread() */
450 	smp_store_release(&bh->state, BUF_STATE_EMPTY);
451 	wake_up(&common->io_wait);
452 }
453 
454 static void bulk_out_complete(struct usb_ep *ep, struct usb_request *req)
455 {
456 	struct fsg_common	*common = ep->driver_data;
457 	struct fsg_buffhd	*bh = req->context;
458 
459 	dump_msg(common, "bulk-out", req->buf, req->actual);
460 	if (req->status || req->actual != bh->bulk_out_intended_length)
461 		DBG(common, "%s --> %d, %u/%u\n", __func__,
462 		    req->status, req->actual, bh->bulk_out_intended_length);
463 	if (req->status == -ECONNRESET)		/* Request was cancelled */
464 		usb_ep_fifo_flush(ep);
465 
466 	/* Synchronize with the smp_load_acquire() in sleep_thread() */
467 	smp_store_release(&bh->state, BUF_STATE_FULL);
468 	wake_up(&common->io_wait);
469 }
470 
471 static int _fsg_common_get_max_lun(struct fsg_common *common)
472 {
473 	int i = ARRAY_SIZE(common->luns) - 1;
474 
475 	while (i >= 0 && !common->luns[i])
476 		--i;
477 
478 	return i;
479 }
480 
481 static int fsg_setup(struct usb_function *f,
482 		     const struct usb_ctrlrequest *ctrl)
483 {
484 	struct fsg_dev		*fsg = fsg_from_func(f);
485 	struct usb_request	*req = fsg->common->ep0req;
486 	u16			w_index = le16_to_cpu(ctrl->wIndex);
487 	u16			w_value = le16_to_cpu(ctrl->wValue);
488 	u16			w_length = le16_to_cpu(ctrl->wLength);
489 
490 	if (!fsg_is_set(fsg->common))
491 		return -EOPNOTSUPP;
492 
493 	++fsg->common->ep0_req_tag;	/* Record arrival of a new request */
494 	req->context = NULL;
495 	req->length = 0;
496 	dump_msg(fsg, "ep0-setup", (u8 *) ctrl, sizeof(*ctrl));
497 
498 	switch (ctrl->bRequest) {
499 
500 	case US_BULK_RESET_REQUEST:
501 		if (ctrl->bRequestType !=
502 		    (USB_DIR_OUT | USB_TYPE_CLASS | USB_RECIP_INTERFACE))
503 			break;
504 		if (w_index != fsg->interface_number || w_value != 0 ||
505 				w_length != 0)
506 			return -EDOM;
507 
508 		/*
509 		 * Raise an exception to stop the current operation
510 		 * and reinitialize our state.
511 		 */
512 		DBG(fsg, "bulk reset request\n");
513 		raise_exception(fsg->common, FSG_STATE_PROTOCOL_RESET);
514 		return USB_GADGET_DELAYED_STATUS;
515 
516 	case US_BULK_GET_MAX_LUN:
517 		if (ctrl->bRequestType !=
518 		    (USB_DIR_IN | USB_TYPE_CLASS | USB_RECIP_INTERFACE))
519 			break;
520 		if (w_index != fsg->interface_number || w_value != 0 ||
521 				w_length != 1)
522 			return -EDOM;
523 		VDBG(fsg, "get max LUN\n");
524 		*(u8 *)req->buf = _fsg_common_get_max_lun(fsg->common);
525 
526 		/* Respond with data/status */
527 		req->length = min((u16)1, w_length);
528 		return ep0_queue(fsg->common);
529 	}
530 
531 	VDBG(fsg,
532 	     "unknown class-specific control req %02x.%02x v%04x i%04x l%u\n",
533 	     ctrl->bRequestType, ctrl->bRequest,
534 	     le16_to_cpu(ctrl->wValue), w_index, w_length);
535 	return -EOPNOTSUPP;
536 }
537 
538 
539 /*-------------------------------------------------------------------------*/
540 
541 /* All the following routines run in process context */
542 
543 /* Use this for bulk or interrupt transfers, not ep0 */
544 static int start_transfer(struct fsg_dev *fsg, struct usb_ep *ep,
545 			   struct usb_request *req)
546 {
547 	int	rc;
548 
549 	if (ep == fsg->bulk_in)
550 		dump_msg(fsg, "bulk-in", req->buf, req->length);
551 
552 	rc = usb_ep_queue(ep, req, GFP_KERNEL);
553 	if (rc) {
554 
555 		/* We can't do much more than wait for a reset */
556 		req->status = rc;
557 
558 		/*
559 		 * Note: currently the net2280 driver fails zero-length
560 		 * submissions if DMA is enabled.
561 		 */
562 		if (rc != -ESHUTDOWN &&
563 				!(rc == -EOPNOTSUPP && req->length == 0))
564 			WARNING(fsg, "error in submission: %s --> %d\n",
565 					ep->name, rc);
566 	}
567 	return rc;
568 }
569 
570 static bool start_in_transfer(struct fsg_common *common, struct fsg_buffhd *bh)
571 {
572 	if (!fsg_is_set(common))
573 		return false;
574 	bh->state = BUF_STATE_SENDING;
575 	if (start_transfer(common->fsg, common->fsg->bulk_in, bh->inreq))
576 		bh->state = BUF_STATE_EMPTY;
577 	return true;
578 }
579 
580 static bool start_out_transfer(struct fsg_common *common, struct fsg_buffhd *bh)
581 {
582 	if (!fsg_is_set(common))
583 		return false;
584 	bh->state = BUF_STATE_RECEIVING;
585 	if (start_transfer(common->fsg, common->fsg->bulk_out, bh->outreq))
586 		bh->state = BUF_STATE_FULL;
587 	return true;
588 }
589 
590 static int sleep_thread(struct fsg_common *common, bool can_freeze,
591 		struct fsg_buffhd *bh)
592 {
593 	int	rc;
594 
595 	/* Wait until a signal arrives or bh is no longer busy */
596 	if (can_freeze)
597 		/*
598 		 * synchronize with the smp_store_release(&bh->state) in
599 		 * bulk_in_complete() or bulk_out_complete()
600 		 */
601 		rc = wait_event_freezable(common->io_wait,
602 				bh && smp_load_acquire(&bh->state) >=
603 					BUF_STATE_EMPTY);
604 	else
605 		rc = wait_event_interruptible(common->io_wait,
606 				bh && smp_load_acquire(&bh->state) >=
607 					BUF_STATE_EMPTY);
608 	return rc ? -EINTR : 0;
609 }
610 
611 
612 /*-------------------------------------------------------------------------*/
613 
614 static int do_read(struct fsg_common *common)
615 {
616 	struct fsg_lun		*curlun = common->curlun;
617 	u32			lba;
618 	struct fsg_buffhd	*bh;
619 	int			rc;
620 	u32			amount_left;
621 	loff_t			file_offset, file_offset_tmp;
622 	unsigned int		amount;
623 	ssize_t			nread;
624 
625 	/*
626 	 * Get the starting Logical Block Address and check that it's
627 	 * not too big.
628 	 */
629 	if (common->cmnd[0] == READ_6)
630 		lba = get_unaligned_be24(&common->cmnd[1]);
631 	else {
632 		lba = get_unaligned_be32(&common->cmnd[2]);
633 
634 		/*
635 		 * We allow DPO (Disable Page Out = don't save data in the
636 		 * cache) and FUA (Force Unit Access = don't read from the
637 		 * cache), but we don't implement them.
638 		 */
639 		if ((common->cmnd[1] & ~0x18) != 0) {
640 			curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
641 			return -EINVAL;
642 		}
643 	}
644 	if (lba >= curlun->num_sectors) {
645 		curlun->sense_data = SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE;
646 		return -EINVAL;
647 	}
648 	file_offset = ((loff_t) lba) << curlun->blkbits;
649 
650 	/* Carry out the file reads */
651 	amount_left = common->data_size_from_cmnd;
652 	if (unlikely(amount_left == 0))
653 		return -EIO;		/* No default reply */
654 
655 	for (;;) {
656 		/*
657 		 * Figure out how much we need to read:
658 		 * Try to read the remaining amount.
659 		 * But don't read more than the buffer size.
660 		 * And don't try to read past the end of the file.
661 		 */
662 		amount = min(amount_left, FSG_BUFLEN);
663 		amount = min((loff_t)amount,
664 			     curlun->file_length - file_offset);
665 
666 		/* Wait for the next buffer to become available */
667 		bh = common->next_buffhd_to_fill;
668 		rc = sleep_thread(common, false, bh);
669 		if (rc)
670 			return rc;
671 
672 		/*
673 		 * If we were asked to read past the end of file,
674 		 * end with an empty buffer.
675 		 */
676 		if (amount == 0) {
677 			curlun->sense_data =
678 					SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE;
679 			curlun->sense_data_info =
680 					file_offset >> curlun->blkbits;
681 			curlun->info_valid = 1;
682 			bh->inreq->length = 0;
683 			bh->state = BUF_STATE_FULL;
684 			break;
685 		}
686 
687 		/* Perform the read */
688 		file_offset_tmp = file_offset;
689 		nread = kernel_read(curlun->filp, bh->buf, amount,
690 				&file_offset_tmp);
691 		VLDBG(curlun, "file read %u @ %llu -> %d\n", amount,
692 		      (unsigned long long)file_offset, (int)nread);
693 		if (signal_pending(current))
694 			return -EINTR;
695 
696 		if (nread < 0) {
697 			LDBG(curlun, "error in file read: %d\n", (int)nread);
698 			nread = 0;
699 		} else if (nread < amount) {
700 			LDBG(curlun, "partial file read: %d/%u\n",
701 			     (int)nread, amount);
702 			nread = round_down(nread, curlun->blksize);
703 		}
704 		file_offset  += nread;
705 		amount_left  -= nread;
706 		common->residue -= nread;
707 
708 		/*
709 		 * Except at the end of the transfer, nread will be
710 		 * equal to the buffer size, which is divisible by the
711 		 * bulk-in maxpacket size.
712 		 */
713 		bh->inreq->length = nread;
714 		bh->state = BUF_STATE_FULL;
715 
716 		/* If an error occurred, report it and its position */
717 		if (nread < amount) {
718 			curlun->sense_data = SS_UNRECOVERED_READ_ERROR;
719 			curlun->sense_data_info =
720 					file_offset >> curlun->blkbits;
721 			curlun->info_valid = 1;
722 			break;
723 		}
724 
725 		if (amount_left == 0)
726 			break;		/* No more left to read */
727 
728 		/* Send this buffer and go read some more */
729 		bh->inreq->zero = 0;
730 		if (!start_in_transfer(common, bh))
731 			/* Don't know what to do if common->fsg is NULL */
732 			return -EIO;
733 		common->next_buffhd_to_fill = bh->next;
734 	}
735 
736 	return -EIO;		/* No default reply */
737 }
738 
739 
740 /*-------------------------------------------------------------------------*/
741 
742 static int do_write(struct fsg_common *common)
743 {
744 	struct fsg_lun		*curlun = common->curlun;
745 	u32			lba;
746 	struct fsg_buffhd	*bh;
747 	int			get_some_more;
748 	u32			amount_left_to_req, amount_left_to_write;
749 	loff_t			usb_offset, file_offset, file_offset_tmp;
750 	unsigned int		amount;
751 	ssize_t			nwritten;
752 	int			rc;
753 
754 	if (curlun->ro) {
755 		curlun->sense_data = SS_WRITE_PROTECTED;
756 		return -EINVAL;
757 	}
758 	spin_lock(&curlun->filp->f_lock);
759 	curlun->filp->f_flags &= ~O_SYNC;	/* Default is not to wait */
760 	spin_unlock(&curlun->filp->f_lock);
761 
762 	/*
763 	 * Get the starting Logical Block Address and check that it's
764 	 * not too big
765 	 */
766 	if (common->cmnd[0] == WRITE_6)
767 		lba = get_unaligned_be24(&common->cmnd[1]);
768 	else {
769 		lba = get_unaligned_be32(&common->cmnd[2]);
770 
771 		/*
772 		 * We allow DPO (Disable Page Out = don't save data in the
773 		 * cache) and FUA (Force Unit Access = write directly to the
774 		 * medium).  We don't implement DPO; we implement FUA by
775 		 * performing synchronous output.
776 		 */
777 		if (common->cmnd[1] & ~0x18) {
778 			curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
779 			return -EINVAL;
780 		}
781 		if (!curlun->nofua && (common->cmnd[1] & 0x08)) { /* FUA */
782 			spin_lock(&curlun->filp->f_lock);
783 			curlun->filp->f_flags |= O_SYNC;
784 			spin_unlock(&curlun->filp->f_lock);
785 		}
786 	}
787 	if (lba >= curlun->num_sectors) {
788 		curlun->sense_data = SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE;
789 		return -EINVAL;
790 	}
791 
792 	/* Carry out the file writes */
793 	get_some_more = 1;
794 	file_offset = usb_offset = ((loff_t) lba) << curlun->blkbits;
795 	amount_left_to_req = common->data_size_from_cmnd;
796 	amount_left_to_write = common->data_size_from_cmnd;
797 
798 	while (amount_left_to_write > 0) {
799 
800 		/* Queue a request for more data from the host */
801 		bh = common->next_buffhd_to_fill;
802 		if (bh->state == BUF_STATE_EMPTY && get_some_more) {
803 
804 			/*
805 			 * Figure out how much we want to get:
806 			 * Try to get the remaining amount,
807 			 * but not more than the buffer size.
808 			 */
809 			amount = min(amount_left_to_req, FSG_BUFLEN);
810 
811 			/* Beyond the end of the backing file? */
812 			if (usb_offset >= curlun->file_length) {
813 				get_some_more = 0;
814 				curlun->sense_data =
815 					SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE;
816 				curlun->sense_data_info =
817 					usb_offset >> curlun->blkbits;
818 				curlun->info_valid = 1;
819 				continue;
820 			}
821 
822 			/* Get the next buffer */
823 			usb_offset += amount;
824 			common->usb_amount_left -= amount;
825 			amount_left_to_req -= amount;
826 			if (amount_left_to_req == 0)
827 				get_some_more = 0;
828 
829 			/*
830 			 * Except at the end of the transfer, amount will be
831 			 * equal to the buffer size, which is divisible by
832 			 * the bulk-out maxpacket size.
833 			 */
834 			set_bulk_out_req_length(common, bh, amount);
835 			if (!start_out_transfer(common, bh))
836 				/* Dunno what to do if common->fsg is NULL */
837 				return -EIO;
838 			common->next_buffhd_to_fill = bh->next;
839 			continue;
840 		}
841 
842 		/* Write the received data to the backing file */
843 		bh = common->next_buffhd_to_drain;
844 		if (bh->state == BUF_STATE_EMPTY && !get_some_more)
845 			break;			/* We stopped early */
846 
847 		/* Wait for the data to be received */
848 		rc = sleep_thread(common, false, bh);
849 		if (rc)
850 			return rc;
851 
852 		common->next_buffhd_to_drain = bh->next;
853 		bh->state = BUF_STATE_EMPTY;
854 
855 		/* Did something go wrong with the transfer? */
856 		if (bh->outreq->status != 0) {
857 			curlun->sense_data = SS_COMMUNICATION_FAILURE;
858 			curlun->sense_data_info =
859 					file_offset >> curlun->blkbits;
860 			curlun->info_valid = 1;
861 			break;
862 		}
863 
864 		amount = bh->outreq->actual;
865 		if (curlun->file_length - file_offset < amount) {
866 			LERROR(curlun, "write %u @ %llu beyond end %llu\n",
867 				       amount, (unsigned long long)file_offset,
868 				       (unsigned long long)curlun->file_length);
869 			amount = curlun->file_length - file_offset;
870 		}
871 
872 		/*
873 		 * Don't accept excess data.  The spec doesn't say
874 		 * what to do in this case.  We'll ignore the error.
875 		 */
876 		amount = min(amount, bh->bulk_out_intended_length);
877 
878 		/* Don't write a partial block */
879 		amount = round_down(amount, curlun->blksize);
880 		if (amount == 0)
881 			goto empty_write;
882 
883 		/* Perform the write */
884 		file_offset_tmp = file_offset;
885 		nwritten = kernel_write(curlun->filp, bh->buf, amount,
886 				&file_offset_tmp);
887 		VLDBG(curlun, "file write %u @ %llu -> %d\n", amount,
888 				(unsigned long long)file_offset, (int)nwritten);
889 		if (signal_pending(current))
890 			return -EINTR;		/* Interrupted! */
891 
892 		if (nwritten < 0) {
893 			LDBG(curlun, "error in file write: %d\n",
894 					(int) nwritten);
895 			nwritten = 0;
896 		} else if (nwritten < amount) {
897 			LDBG(curlun, "partial file write: %d/%u\n",
898 					(int) nwritten, amount);
899 			nwritten = round_down(nwritten, curlun->blksize);
900 		}
901 		file_offset += nwritten;
902 		amount_left_to_write -= nwritten;
903 		common->residue -= nwritten;
904 
905 		/* If an error occurred, report it and its position */
906 		if (nwritten < amount) {
907 			curlun->sense_data = SS_WRITE_ERROR;
908 			curlun->sense_data_info =
909 					file_offset >> curlun->blkbits;
910 			curlun->info_valid = 1;
911 			break;
912 		}
913 
914  empty_write:
915 		/* Did the host decide to stop early? */
916 		if (bh->outreq->actual < bh->bulk_out_intended_length) {
917 			common->short_packet_received = 1;
918 			break;
919 		}
920 	}
921 
922 	return -EIO;		/* No default reply */
923 }
924 
925 
926 /*-------------------------------------------------------------------------*/
927 
928 static int do_synchronize_cache(struct fsg_common *common)
929 {
930 	struct fsg_lun	*curlun = common->curlun;
931 	int		rc;
932 
933 	/* We ignore the requested LBA and write out all file's
934 	 * dirty data buffers. */
935 	rc = fsg_lun_fsync_sub(curlun);
936 	if (rc)
937 		curlun->sense_data = SS_WRITE_ERROR;
938 	return 0;
939 }
940 
941 
942 /*-------------------------------------------------------------------------*/
943 
944 static void invalidate_sub(struct fsg_lun *curlun)
945 {
946 	struct file	*filp = curlun->filp;
947 	struct inode	*inode = file_inode(filp);
948 	unsigned long	rc;
949 
950 	rc = invalidate_mapping_pages(inode->i_mapping, 0, -1);
951 	VLDBG(curlun, "invalidate_mapping_pages -> %ld\n", rc);
952 }
953 
954 static int do_verify(struct fsg_common *common)
955 {
956 	struct fsg_lun		*curlun = common->curlun;
957 	u32			lba;
958 	u32			verification_length;
959 	struct fsg_buffhd	*bh = common->next_buffhd_to_fill;
960 	loff_t			file_offset, file_offset_tmp;
961 	u32			amount_left;
962 	unsigned int		amount;
963 	ssize_t			nread;
964 
965 	/*
966 	 * Get the starting Logical Block Address and check that it's
967 	 * not too big.
968 	 */
969 	lba = get_unaligned_be32(&common->cmnd[2]);
970 	if (lba >= curlun->num_sectors) {
971 		curlun->sense_data = SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE;
972 		return -EINVAL;
973 	}
974 
975 	/*
976 	 * We allow DPO (Disable Page Out = don't save data in the
977 	 * cache) but we don't implement it.
978 	 */
979 	if (common->cmnd[1] & ~0x10) {
980 		curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
981 		return -EINVAL;
982 	}
983 
984 	verification_length = get_unaligned_be16(&common->cmnd[7]);
985 	if (unlikely(verification_length == 0))
986 		return -EIO;		/* No default reply */
987 
988 	/* Prepare to carry out the file verify */
989 	amount_left = verification_length << curlun->blkbits;
990 	file_offset = ((loff_t) lba) << curlun->blkbits;
991 
992 	/* Write out all the dirty buffers before invalidating them */
993 	fsg_lun_fsync_sub(curlun);
994 	if (signal_pending(current))
995 		return -EINTR;
996 
997 	invalidate_sub(curlun);
998 	if (signal_pending(current))
999 		return -EINTR;
1000 
1001 	/* Just try to read the requested blocks */
1002 	while (amount_left > 0) {
1003 		/*
1004 		 * Figure out how much we need to read:
1005 		 * Try to read the remaining amount, but not more than
1006 		 * the buffer size.
1007 		 * And don't try to read past the end of the file.
1008 		 */
1009 		amount = min(amount_left, FSG_BUFLEN);
1010 		amount = min((loff_t)amount,
1011 			     curlun->file_length - file_offset);
1012 		if (amount == 0) {
1013 			curlun->sense_data =
1014 					SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE;
1015 			curlun->sense_data_info =
1016 				file_offset >> curlun->blkbits;
1017 			curlun->info_valid = 1;
1018 			break;
1019 		}
1020 
1021 		/* Perform the read */
1022 		file_offset_tmp = file_offset;
1023 		nread = kernel_read(curlun->filp, bh->buf, amount,
1024 				&file_offset_tmp);
1025 		VLDBG(curlun, "file read %u @ %llu -> %d\n", amount,
1026 				(unsigned long long) file_offset,
1027 				(int) nread);
1028 		if (signal_pending(current))
1029 			return -EINTR;
1030 
1031 		if (nread < 0) {
1032 			LDBG(curlun, "error in file verify: %d\n", (int)nread);
1033 			nread = 0;
1034 		} else if (nread < amount) {
1035 			LDBG(curlun, "partial file verify: %d/%u\n",
1036 			     (int)nread, amount);
1037 			nread = round_down(nread, curlun->blksize);
1038 		}
1039 		if (nread == 0) {
1040 			curlun->sense_data = SS_UNRECOVERED_READ_ERROR;
1041 			curlun->sense_data_info =
1042 				file_offset >> curlun->blkbits;
1043 			curlun->info_valid = 1;
1044 			break;
1045 		}
1046 		file_offset += nread;
1047 		amount_left -= nread;
1048 	}
1049 	return 0;
1050 }
1051 
1052 
1053 /*-------------------------------------------------------------------------*/
1054 
1055 static int do_inquiry(struct fsg_common *common, struct fsg_buffhd *bh)
1056 {
1057 	struct fsg_lun *curlun = common->curlun;
1058 	u8	*buf = (u8 *) bh->buf;
1059 
1060 	if (!curlun) {		/* Unsupported LUNs are okay */
1061 		common->bad_lun_okay = 1;
1062 		memset(buf, 0, 36);
1063 		buf[0] = TYPE_NO_LUN;	/* Unsupported, no device-type */
1064 		buf[4] = 31;		/* Additional length */
1065 		return 36;
1066 	}
1067 
1068 	buf[0] = curlun->cdrom ? TYPE_ROM : TYPE_DISK;
1069 	buf[1] = curlun->removable ? 0x80 : 0;
1070 	buf[2] = 2;		/* ANSI SCSI level 2 */
1071 	buf[3] = 2;		/* SCSI-2 INQUIRY data format */
1072 	buf[4] = 31;		/* Additional length */
1073 	buf[5] = 0;		/* No special options */
1074 	buf[6] = 0;
1075 	buf[7] = 0;
1076 	if (curlun->inquiry_string[0])
1077 		memcpy(buf + 8, curlun->inquiry_string,
1078 		       sizeof(curlun->inquiry_string));
1079 	else
1080 		memcpy(buf + 8, common->inquiry_string,
1081 		       sizeof(common->inquiry_string));
1082 	return 36;
1083 }
1084 
1085 static int do_request_sense(struct fsg_common *common, struct fsg_buffhd *bh)
1086 {
1087 	struct fsg_lun	*curlun = common->curlun;
1088 	u8		*buf = (u8 *) bh->buf;
1089 	u32		sd, sdinfo;
1090 	int		valid;
1091 
1092 	/*
1093 	 * From the SCSI-2 spec., section 7.9 (Unit attention condition):
1094 	 *
1095 	 * If a REQUEST SENSE command is received from an initiator
1096 	 * with a pending unit attention condition (before the target
1097 	 * generates the contingent allegiance condition), then the
1098 	 * target shall either:
1099 	 *   a) report any pending sense data and preserve the unit
1100 	 *	attention condition on the logical unit, or,
1101 	 *   b) report the unit attention condition, may discard any
1102 	 *	pending sense data, and clear the unit attention
1103 	 *	condition on the logical unit for that initiator.
1104 	 *
1105 	 * FSG normally uses option a); enable this code to use option b).
1106 	 */
1107 #if 0
1108 	if (curlun && curlun->unit_attention_data != SS_NO_SENSE) {
1109 		curlun->sense_data = curlun->unit_attention_data;
1110 		curlun->unit_attention_data = SS_NO_SENSE;
1111 	}
1112 #endif
1113 
1114 	if (!curlun) {		/* Unsupported LUNs are okay */
1115 		common->bad_lun_okay = 1;
1116 		sd = SS_LOGICAL_UNIT_NOT_SUPPORTED;
1117 		sdinfo = 0;
1118 		valid = 0;
1119 	} else {
1120 		sd = curlun->sense_data;
1121 		sdinfo = curlun->sense_data_info;
1122 		valid = curlun->info_valid << 7;
1123 		curlun->sense_data = SS_NO_SENSE;
1124 		curlun->sense_data_info = 0;
1125 		curlun->info_valid = 0;
1126 	}
1127 
1128 	memset(buf, 0, 18);
1129 	buf[0] = valid | 0x70;			/* Valid, current error */
1130 	buf[2] = SK(sd);
1131 	put_unaligned_be32(sdinfo, &buf[3]);	/* Sense information */
1132 	buf[7] = 18 - 8;			/* Additional sense length */
1133 	buf[12] = ASC(sd);
1134 	buf[13] = ASCQ(sd);
1135 	return 18;
1136 }
1137 
1138 static int do_read_capacity(struct fsg_common *common, struct fsg_buffhd *bh)
1139 {
1140 	struct fsg_lun	*curlun = common->curlun;
1141 	u32		lba = get_unaligned_be32(&common->cmnd[2]);
1142 	int		pmi = common->cmnd[8];
1143 	u8		*buf = (u8 *)bh->buf;
1144 
1145 	/* Check the PMI and LBA fields */
1146 	if (pmi > 1 || (pmi == 0 && lba != 0)) {
1147 		curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
1148 		return -EINVAL;
1149 	}
1150 
1151 	put_unaligned_be32(curlun->num_sectors - 1, &buf[0]);
1152 						/* Max logical block */
1153 	put_unaligned_be32(curlun->blksize, &buf[4]);/* Block length */
1154 	return 8;
1155 }
1156 
1157 static int do_read_header(struct fsg_common *common, struct fsg_buffhd *bh)
1158 {
1159 	struct fsg_lun	*curlun = common->curlun;
1160 	int		msf = common->cmnd[1] & 0x02;
1161 	u32		lba = get_unaligned_be32(&common->cmnd[2]);
1162 	u8		*buf = (u8 *)bh->buf;
1163 
1164 	if (common->cmnd[1] & ~0x02) {		/* Mask away MSF */
1165 		curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
1166 		return -EINVAL;
1167 	}
1168 	if (lba >= curlun->num_sectors) {
1169 		curlun->sense_data = SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE;
1170 		return -EINVAL;
1171 	}
1172 
1173 	memset(buf, 0, 8);
1174 	buf[0] = 0x01;		/* 2048 bytes of user data, rest is EC */
1175 	store_cdrom_address(&buf[4], msf, lba);
1176 	return 8;
1177 }
1178 
1179 static int do_read_toc(struct fsg_common *common, struct fsg_buffhd *bh)
1180 {
1181 	struct fsg_lun	*curlun = common->curlun;
1182 	int		msf = common->cmnd[1] & 0x02;
1183 	int		start_track = common->cmnd[6];
1184 	u8		*buf = (u8 *)bh->buf;
1185 
1186 	if ((common->cmnd[1] & ~0x02) != 0 ||	/* Mask away MSF */
1187 			start_track > 1) {
1188 		curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
1189 		return -EINVAL;
1190 	}
1191 
1192 	memset(buf, 0, 20);
1193 	buf[1] = (20-2);		/* TOC data length */
1194 	buf[2] = 1;			/* First track number */
1195 	buf[3] = 1;			/* Last track number */
1196 	buf[5] = 0x16;			/* Data track, copying allowed */
1197 	buf[6] = 0x01;			/* Only track is number 1 */
1198 	store_cdrom_address(&buf[8], msf, 0);
1199 
1200 	buf[13] = 0x16;			/* Lead-out track is data */
1201 	buf[14] = 0xAA;			/* Lead-out track number */
1202 	store_cdrom_address(&buf[16], msf, curlun->num_sectors);
1203 	return 20;
1204 }
1205 
1206 static int do_mode_sense(struct fsg_common *common, struct fsg_buffhd *bh)
1207 {
1208 	struct fsg_lun	*curlun = common->curlun;
1209 	int		mscmnd = common->cmnd[0];
1210 	u8		*buf = (u8 *) bh->buf;
1211 	u8		*buf0 = buf;
1212 	int		pc, page_code;
1213 	int		changeable_values, all_pages;
1214 	int		valid_page = 0;
1215 	int		len, limit;
1216 
1217 	if ((common->cmnd[1] & ~0x08) != 0) {	/* Mask away DBD */
1218 		curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
1219 		return -EINVAL;
1220 	}
1221 	pc = common->cmnd[2] >> 6;
1222 	page_code = common->cmnd[2] & 0x3f;
1223 	if (pc == 3) {
1224 		curlun->sense_data = SS_SAVING_PARAMETERS_NOT_SUPPORTED;
1225 		return -EINVAL;
1226 	}
1227 	changeable_values = (pc == 1);
1228 	all_pages = (page_code == 0x3f);
1229 
1230 	/*
1231 	 * Write the mode parameter header.  Fixed values are: default
1232 	 * medium type, no cache control (DPOFUA), and no block descriptors.
1233 	 * The only variable value is the WriteProtect bit.  We will fill in
1234 	 * the mode data length later.
1235 	 */
1236 	memset(buf, 0, 8);
1237 	if (mscmnd == MODE_SENSE) {
1238 		buf[2] = (curlun->ro ? 0x80 : 0x00);		/* WP, DPOFUA */
1239 		buf += 4;
1240 		limit = 255;
1241 	} else {			/* MODE_SENSE_10 */
1242 		buf[3] = (curlun->ro ? 0x80 : 0x00);		/* WP, DPOFUA */
1243 		buf += 8;
1244 		limit = 65535;		/* Should really be FSG_BUFLEN */
1245 	}
1246 
1247 	/* No block descriptors */
1248 
1249 	/*
1250 	 * The mode pages, in numerical order.  The only page we support
1251 	 * is the Caching page.
1252 	 */
1253 	if (page_code == 0x08 || all_pages) {
1254 		valid_page = 1;
1255 		buf[0] = 0x08;		/* Page code */
1256 		buf[1] = 10;		/* Page length */
1257 		memset(buf+2, 0, 10);	/* None of the fields are changeable */
1258 
1259 		if (!changeable_values) {
1260 			buf[2] = 0x04;	/* Write cache enable, */
1261 					/* Read cache not disabled */
1262 					/* No cache retention priorities */
1263 			put_unaligned_be16(0xffff, &buf[4]);
1264 					/* Don't disable prefetch */
1265 					/* Minimum prefetch = 0 */
1266 			put_unaligned_be16(0xffff, &buf[8]);
1267 					/* Maximum prefetch */
1268 			put_unaligned_be16(0xffff, &buf[10]);
1269 					/* Maximum prefetch ceiling */
1270 		}
1271 		buf += 12;
1272 	}
1273 
1274 	/*
1275 	 * Check that a valid page was requested and the mode data length
1276 	 * isn't too long.
1277 	 */
1278 	len = buf - buf0;
1279 	if (!valid_page || len > limit) {
1280 		curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
1281 		return -EINVAL;
1282 	}
1283 
1284 	/*  Store the mode data length */
1285 	if (mscmnd == MODE_SENSE)
1286 		buf0[0] = len - 1;
1287 	else
1288 		put_unaligned_be16(len - 2, buf0);
1289 	return len;
1290 }
1291 
1292 static int do_start_stop(struct fsg_common *common)
1293 {
1294 	struct fsg_lun	*curlun = common->curlun;
1295 	int		loej, start;
1296 
1297 	if (!curlun) {
1298 		return -EINVAL;
1299 	} else if (!curlun->removable) {
1300 		curlun->sense_data = SS_INVALID_COMMAND;
1301 		return -EINVAL;
1302 	} else if ((common->cmnd[1] & ~0x01) != 0 || /* Mask away Immed */
1303 		   (common->cmnd[4] & ~0x03) != 0) { /* Mask LoEj, Start */
1304 		curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
1305 		return -EINVAL;
1306 	}
1307 
1308 	loej  = common->cmnd[4] & 0x02;
1309 	start = common->cmnd[4] & 0x01;
1310 
1311 	/*
1312 	 * Our emulation doesn't support mounting; the medium is
1313 	 * available for use as soon as it is loaded.
1314 	 */
1315 	if (start) {
1316 		if (!fsg_lun_is_open(curlun)) {
1317 			curlun->sense_data = SS_MEDIUM_NOT_PRESENT;
1318 			return -EINVAL;
1319 		}
1320 		return 0;
1321 	}
1322 
1323 	/* Are we allowed to unload the media? */
1324 	if (curlun->prevent_medium_removal) {
1325 		LDBG(curlun, "unload attempt prevented\n");
1326 		curlun->sense_data = SS_MEDIUM_REMOVAL_PREVENTED;
1327 		return -EINVAL;
1328 	}
1329 
1330 	if (!loej)
1331 		return 0;
1332 
1333 	up_read(&common->filesem);
1334 	down_write(&common->filesem);
1335 	fsg_lun_close(curlun);
1336 	up_write(&common->filesem);
1337 	down_read(&common->filesem);
1338 
1339 	return 0;
1340 }
1341 
1342 static int do_prevent_allow(struct fsg_common *common)
1343 {
1344 	struct fsg_lun	*curlun = common->curlun;
1345 	int		prevent;
1346 
1347 	if (!common->curlun) {
1348 		return -EINVAL;
1349 	} else if (!common->curlun->removable) {
1350 		common->curlun->sense_data = SS_INVALID_COMMAND;
1351 		return -EINVAL;
1352 	}
1353 
1354 	prevent = common->cmnd[4] & 0x01;
1355 	if ((common->cmnd[4] & ~0x01) != 0) {	/* Mask away Prevent */
1356 		curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
1357 		return -EINVAL;
1358 	}
1359 
1360 	if (curlun->prevent_medium_removal && !prevent)
1361 		fsg_lun_fsync_sub(curlun);
1362 	curlun->prevent_medium_removal = prevent;
1363 	return 0;
1364 }
1365 
1366 static int do_read_format_capacities(struct fsg_common *common,
1367 			struct fsg_buffhd *bh)
1368 {
1369 	struct fsg_lun	*curlun = common->curlun;
1370 	u8		*buf = (u8 *) bh->buf;
1371 
1372 	buf[0] = buf[1] = buf[2] = 0;
1373 	buf[3] = 8;	/* Only the Current/Maximum Capacity Descriptor */
1374 	buf += 4;
1375 
1376 	put_unaligned_be32(curlun->num_sectors, &buf[0]);
1377 						/* Number of blocks */
1378 	put_unaligned_be32(curlun->blksize, &buf[4]);/* Block length */
1379 	buf[4] = 0x02;				/* Current capacity */
1380 	return 12;
1381 }
1382 
1383 static int do_mode_select(struct fsg_common *common, struct fsg_buffhd *bh)
1384 {
1385 	struct fsg_lun	*curlun = common->curlun;
1386 
1387 	/* We don't support MODE SELECT */
1388 	if (curlun)
1389 		curlun->sense_data = SS_INVALID_COMMAND;
1390 	return -EINVAL;
1391 }
1392 
1393 
1394 /*-------------------------------------------------------------------------*/
1395 
1396 static int halt_bulk_in_endpoint(struct fsg_dev *fsg)
1397 {
1398 	int	rc;
1399 
1400 	rc = fsg_set_halt(fsg, fsg->bulk_in);
1401 	if (rc == -EAGAIN)
1402 		VDBG(fsg, "delayed bulk-in endpoint halt\n");
1403 	while (rc != 0) {
1404 		if (rc != -EAGAIN) {
1405 			WARNING(fsg, "usb_ep_set_halt -> %d\n", rc);
1406 			rc = 0;
1407 			break;
1408 		}
1409 
1410 		/* Wait for a short time and then try again */
1411 		if (msleep_interruptible(100) != 0)
1412 			return -EINTR;
1413 		rc = usb_ep_set_halt(fsg->bulk_in);
1414 	}
1415 	return rc;
1416 }
1417 
1418 static int wedge_bulk_in_endpoint(struct fsg_dev *fsg)
1419 {
1420 	int	rc;
1421 
1422 	DBG(fsg, "bulk-in set wedge\n");
1423 	rc = usb_ep_set_wedge(fsg->bulk_in);
1424 	if (rc == -EAGAIN)
1425 		VDBG(fsg, "delayed bulk-in endpoint wedge\n");
1426 	while (rc != 0) {
1427 		if (rc != -EAGAIN) {
1428 			WARNING(fsg, "usb_ep_set_wedge -> %d\n", rc);
1429 			rc = 0;
1430 			break;
1431 		}
1432 
1433 		/* Wait for a short time and then try again */
1434 		if (msleep_interruptible(100) != 0)
1435 			return -EINTR;
1436 		rc = usb_ep_set_wedge(fsg->bulk_in);
1437 	}
1438 	return rc;
1439 }
1440 
1441 static int throw_away_data(struct fsg_common *common)
1442 {
1443 	struct fsg_buffhd	*bh, *bh2;
1444 	u32			amount;
1445 	int			rc;
1446 
1447 	for (bh = common->next_buffhd_to_drain;
1448 	     bh->state != BUF_STATE_EMPTY || common->usb_amount_left > 0;
1449 	     bh = common->next_buffhd_to_drain) {
1450 
1451 		/* Try to submit another request if we need one */
1452 		bh2 = common->next_buffhd_to_fill;
1453 		if (bh2->state == BUF_STATE_EMPTY &&
1454 				common->usb_amount_left > 0) {
1455 			amount = min(common->usb_amount_left, FSG_BUFLEN);
1456 
1457 			/*
1458 			 * Except at the end of the transfer, amount will be
1459 			 * equal to the buffer size, which is divisible by
1460 			 * the bulk-out maxpacket size.
1461 			 */
1462 			set_bulk_out_req_length(common, bh2, amount);
1463 			if (!start_out_transfer(common, bh2))
1464 				/* Dunno what to do if common->fsg is NULL */
1465 				return -EIO;
1466 			common->next_buffhd_to_fill = bh2->next;
1467 			common->usb_amount_left -= amount;
1468 			continue;
1469 		}
1470 
1471 		/* Wait for the data to be received */
1472 		rc = sleep_thread(common, false, bh);
1473 		if (rc)
1474 			return rc;
1475 
1476 		/* Throw away the data in a filled buffer */
1477 		bh->state = BUF_STATE_EMPTY;
1478 		common->next_buffhd_to_drain = bh->next;
1479 
1480 		/* A short packet or an error ends everything */
1481 		if (bh->outreq->actual < bh->bulk_out_intended_length ||
1482 				bh->outreq->status != 0) {
1483 			raise_exception(common, FSG_STATE_ABORT_BULK_OUT);
1484 			return -EINTR;
1485 		}
1486 	}
1487 	return 0;
1488 }
1489 
1490 static int finish_reply(struct fsg_common *common)
1491 {
1492 	struct fsg_buffhd	*bh = common->next_buffhd_to_fill;
1493 	int			rc = 0;
1494 
1495 	switch (common->data_dir) {
1496 	case DATA_DIR_NONE:
1497 		break;			/* Nothing to send */
1498 
1499 	/*
1500 	 * If we don't know whether the host wants to read or write,
1501 	 * this must be CB or CBI with an unknown command.  We mustn't
1502 	 * try to send or receive any data.  So stall both bulk pipes
1503 	 * if we can and wait for a reset.
1504 	 */
1505 	case DATA_DIR_UNKNOWN:
1506 		if (!common->can_stall) {
1507 			/* Nothing */
1508 		} else if (fsg_is_set(common)) {
1509 			fsg_set_halt(common->fsg, common->fsg->bulk_out);
1510 			rc = halt_bulk_in_endpoint(common->fsg);
1511 		} else {
1512 			/* Don't know what to do if common->fsg is NULL */
1513 			rc = -EIO;
1514 		}
1515 		break;
1516 
1517 	/* All but the last buffer of data must have already been sent */
1518 	case DATA_DIR_TO_HOST:
1519 		if (common->data_size == 0) {
1520 			/* Nothing to send */
1521 
1522 		/* Don't know what to do if common->fsg is NULL */
1523 		} else if (!fsg_is_set(common)) {
1524 			rc = -EIO;
1525 
1526 		/* If there's no residue, simply send the last buffer */
1527 		} else if (common->residue == 0) {
1528 			bh->inreq->zero = 0;
1529 			if (!start_in_transfer(common, bh))
1530 				return -EIO;
1531 			common->next_buffhd_to_fill = bh->next;
1532 
1533 		/*
1534 		 * For Bulk-only, mark the end of the data with a short
1535 		 * packet.  If we are allowed to stall, halt the bulk-in
1536 		 * endpoint.  (Note: This violates the Bulk-Only Transport
1537 		 * specification, which requires us to pad the data if we
1538 		 * don't halt the endpoint.  Presumably nobody will mind.)
1539 		 */
1540 		} else {
1541 			bh->inreq->zero = 1;
1542 			if (!start_in_transfer(common, bh))
1543 				rc = -EIO;
1544 			common->next_buffhd_to_fill = bh->next;
1545 			if (common->can_stall)
1546 				rc = halt_bulk_in_endpoint(common->fsg);
1547 		}
1548 		break;
1549 
1550 	/*
1551 	 * We have processed all we want from the data the host has sent.
1552 	 * There may still be outstanding bulk-out requests.
1553 	 */
1554 	case DATA_DIR_FROM_HOST:
1555 		if (common->residue == 0) {
1556 			/* Nothing to receive */
1557 
1558 		/* Did the host stop sending unexpectedly early? */
1559 		} else if (common->short_packet_received) {
1560 			raise_exception(common, FSG_STATE_ABORT_BULK_OUT);
1561 			rc = -EINTR;
1562 
1563 		/*
1564 		 * We haven't processed all the incoming data.  Even though
1565 		 * we may be allowed to stall, doing so would cause a race.
1566 		 * The controller may already have ACK'ed all the remaining
1567 		 * bulk-out packets, in which case the host wouldn't see a
1568 		 * STALL.  Not realizing the endpoint was halted, it wouldn't
1569 		 * clear the halt -- leading to problems later on.
1570 		 */
1571 #if 0
1572 		} else if (common->can_stall) {
1573 			if (fsg_is_set(common))
1574 				fsg_set_halt(common->fsg,
1575 					     common->fsg->bulk_out);
1576 			raise_exception(common, FSG_STATE_ABORT_BULK_OUT);
1577 			rc = -EINTR;
1578 #endif
1579 
1580 		/*
1581 		 * We can't stall.  Read in the excess data and throw it
1582 		 * all away.
1583 		 */
1584 		} else {
1585 			rc = throw_away_data(common);
1586 		}
1587 		break;
1588 	}
1589 	return rc;
1590 }
1591 
1592 static void send_status(struct fsg_common *common)
1593 {
1594 	struct fsg_lun		*curlun = common->curlun;
1595 	struct fsg_buffhd	*bh;
1596 	struct bulk_cs_wrap	*csw;
1597 	int			rc;
1598 	u8			status = US_BULK_STAT_OK;
1599 	u32			sd, sdinfo = 0;
1600 
1601 	/* Wait for the next buffer to become available */
1602 	bh = common->next_buffhd_to_fill;
1603 	rc = sleep_thread(common, false, bh);
1604 	if (rc)
1605 		return;
1606 
1607 	if (curlun) {
1608 		sd = curlun->sense_data;
1609 		sdinfo = curlun->sense_data_info;
1610 	} else if (common->bad_lun_okay)
1611 		sd = SS_NO_SENSE;
1612 	else
1613 		sd = SS_LOGICAL_UNIT_NOT_SUPPORTED;
1614 
1615 	if (common->phase_error) {
1616 		DBG(common, "sending phase-error status\n");
1617 		status = US_BULK_STAT_PHASE;
1618 		sd = SS_INVALID_COMMAND;
1619 	} else if (sd != SS_NO_SENSE) {
1620 		DBG(common, "sending command-failure status\n");
1621 		status = US_BULK_STAT_FAIL;
1622 		VDBG(common, "  sense data: SK x%02x, ASC x%02x, ASCQ x%02x;"
1623 				"  info x%x\n",
1624 				SK(sd), ASC(sd), ASCQ(sd), sdinfo);
1625 	}
1626 
1627 	/* Store and send the Bulk-only CSW */
1628 	csw = (void *)bh->buf;
1629 
1630 	csw->Signature = cpu_to_le32(US_BULK_CS_SIGN);
1631 	csw->Tag = common->tag;
1632 	csw->Residue = cpu_to_le32(common->residue);
1633 	csw->Status = status;
1634 
1635 	bh->inreq->length = US_BULK_CS_WRAP_LEN;
1636 	bh->inreq->zero = 0;
1637 	if (!start_in_transfer(common, bh))
1638 		/* Don't know what to do if common->fsg is NULL */
1639 		return;
1640 
1641 	common->next_buffhd_to_fill = bh->next;
1642 	return;
1643 }
1644 
1645 
1646 /*-------------------------------------------------------------------------*/
1647 
1648 /*
1649  * Check whether the command is properly formed and whether its data size
1650  * and direction agree with the values we already have.
1651  */
1652 static int check_command(struct fsg_common *common, int cmnd_size,
1653 			 enum data_direction data_dir, unsigned int mask,
1654 			 int needs_medium, const char *name)
1655 {
1656 	int			i;
1657 	unsigned int		lun = common->cmnd[1] >> 5;
1658 	static const char	dirletter[4] = {'u', 'o', 'i', 'n'};
1659 	char			hdlen[20];
1660 	struct fsg_lun		*curlun;
1661 
1662 	hdlen[0] = 0;
1663 	if (common->data_dir != DATA_DIR_UNKNOWN)
1664 		sprintf(hdlen, ", H%c=%u", dirletter[(int) common->data_dir],
1665 			common->data_size);
1666 	VDBG(common, "SCSI command: %s;  Dc=%d, D%c=%u;  Hc=%d%s\n",
1667 	     name, cmnd_size, dirletter[(int) data_dir],
1668 	     common->data_size_from_cmnd, common->cmnd_size, hdlen);
1669 
1670 	/*
1671 	 * We can't reply at all until we know the correct data direction
1672 	 * and size.
1673 	 */
1674 	if (common->data_size_from_cmnd == 0)
1675 		data_dir = DATA_DIR_NONE;
1676 	if (common->data_size < common->data_size_from_cmnd) {
1677 		/*
1678 		 * Host data size < Device data size is a phase error.
1679 		 * Carry out the command, but only transfer as much as
1680 		 * we are allowed.
1681 		 */
1682 		common->data_size_from_cmnd = common->data_size;
1683 		common->phase_error = 1;
1684 	}
1685 	common->residue = common->data_size;
1686 	common->usb_amount_left = common->data_size;
1687 
1688 	/* Conflicting data directions is a phase error */
1689 	if (common->data_dir != data_dir && common->data_size_from_cmnd > 0) {
1690 		common->phase_error = 1;
1691 		return -EINVAL;
1692 	}
1693 
1694 	/* Verify the length of the command itself */
1695 	if (cmnd_size != common->cmnd_size) {
1696 
1697 		/*
1698 		 * Special case workaround: There are plenty of buggy SCSI
1699 		 * implementations. Many have issues with cbw->Length
1700 		 * field passing a wrong command size. For those cases we
1701 		 * always try to work around the problem by using the length
1702 		 * sent by the host side provided it is at least as large
1703 		 * as the correct command length.
1704 		 * Examples of such cases would be MS-Windows, which issues
1705 		 * REQUEST SENSE with cbw->Length == 12 where it should
1706 		 * be 6, and xbox360 issuing INQUIRY, TEST UNIT READY and
1707 		 * REQUEST SENSE with cbw->Length == 10 where it should
1708 		 * be 6 as well.
1709 		 */
1710 		if (cmnd_size <= common->cmnd_size) {
1711 			DBG(common, "%s is buggy! Expected length %d "
1712 			    "but we got %d\n", name,
1713 			    cmnd_size, common->cmnd_size);
1714 			cmnd_size = common->cmnd_size;
1715 		} else {
1716 			common->phase_error = 1;
1717 			return -EINVAL;
1718 		}
1719 	}
1720 
1721 	/* Check that the LUN values are consistent */
1722 	if (common->lun != lun)
1723 		DBG(common, "using LUN %u from CBW, not LUN %u from CDB\n",
1724 		    common->lun, lun);
1725 
1726 	/* Check the LUN */
1727 	curlun = common->curlun;
1728 	if (curlun) {
1729 		if (common->cmnd[0] != REQUEST_SENSE) {
1730 			curlun->sense_data = SS_NO_SENSE;
1731 			curlun->sense_data_info = 0;
1732 			curlun->info_valid = 0;
1733 		}
1734 	} else {
1735 		common->bad_lun_okay = 0;
1736 
1737 		/*
1738 		 * INQUIRY and REQUEST SENSE commands are explicitly allowed
1739 		 * to use unsupported LUNs; all others may not.
1740 		 */
1741 		if (common->cmnd[0] != INQUIRY &&
1742 		    common->cmnd[0] != REQUEST_SENSE) {
1743 			DBG(common, "unsupported LUN %u\n", common->lun);
1744 			return -EINVAL;
1745 		}
1746 	}
1747 
1748 	/*
1749 	 * If a unit attention condition exists, only INQUIRY and
1750 	 * REQUEST SENSE commands are allowed; anything else must fail.
1751 	 */
1752 	if (curlun && curlun->unit_attention_data != SS_NO_SENSE &&
1753 	    common->cmnd[0] != INQUIRY &&
1754 	    common->cmnd[0] != REQUEST_SENSE) {
1755 		curlun->sense_data = curlun->unit_attention_data;
1756 		curlun->unit_attention_data = SS_NO_SENSE;
1757 		return -EINVAL;
1758 	}
1759 
1760 	/* Check that only command bytes listed in the mask are non-zero */
1761 	common->cmnd[1] &= 0x1f;			/* Mask away the LUN */
1762 	for (i = 1; i < cmnd_size; ++i) {
1763 		if (common->cmnd[i] && !(mask & (1 << i))) {
1764 			if (curlun)
1765 				curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
1766 			return -EINVAL;
1767 		}
1768 	}
1769 
1770 	/* If the medium isn't mounted and the command needs to access
1771 	 * it, return an error. */
1772 	if (curlun && !fsg_lun_is_open(curlun) && needs_medium) {
1773 		curlun->sense_data = SS_MEDIUM_NOT_PRESENT;
1774 		return -EINVAL;
1775 	}
1776 
1777 	return 0;
1778 }
1779 
1780 /* wrapper of check_command for data size in blocks handling */
1781 static int check_command_size_in_blocks(struct fsg_common *common,
1782 		int cmnd_size, enum data_direction data_dir,
1783 		unsigned int mask, int needs_medium, const char *name)
1784 {
1785 	if (common->curlun)
1786 		common->data_size_from_cmnd <<= common->curlun->blkbits;
1787 	return check_command(common, cmnd_size, data_dir,
1788 			mask, needs_medium, name);
1789 }
1790 
1791 static int do_scsi_command(struct fsg_common *common)
1792 {
1793 	struct fsg_buffhd	*bh;
1794 	int			rc;
1795 	int			reply = -EINVAL;
1796 	int			i;
1797 	static char		unknown[16];
1798 
1799 	dump_cdb(common);
1800 
1801 	/* Wait for the next buffer to become available for data or status */
1802 	bh = common->next_buffhd_to_fill;
1803 	common->next_buffhd_to_drain = bh;
1804 	rc = sleep_thread(common, false, bh);
1805 	if (rc)
1806 		return rc;
1807 
1808 	common->phase_error = 0;
1809 	common->short_packet_received = 0;
1810 
1811 	down_read(&common->filesem);	/* We're using the backing file */
1812 	switch (common->cmnd[0]) {
1813 
1814 	case INQUIRY:
1815 		common->data_size_from_cmnd = common->cmnd[4];
1816 		reply = check_command(common, 6, DATA_DIR_TO_HOST,
1817 				      (1<<4), 0,
1818 				      "INQUIRY");
1819 		if (reply == 0)
1820 			reply = do_inquiry(common, bh);
1821 		break;
1822 
1823 	case MODE_SELECT:
1824 		common->data_size_from_cmnd = common->cmnd[4];
1825 		reply = check_command(common, 6, DATA_DIR_FROM_HOST,
1826 				      (1<<1) | (1<<4), 0,
1827 				      "MODE SELECT(6)");
1828 		if (reply == 0)
1829 			reply = do_mode_select(common, bh);
1830 		break;
1831 
1832 	case MODE_SELECT_10:
1833 		common->data_size_from_cmnd =
1834 			get_unaligned_be16(&common->cmnd[7]);
1835 		reply = check_command(common, 10, DATA_DIR_FROM_HOST,
1836 				      (1<<1) | (3<<7), 0,
1837 				      "MODE SELECT(10)");
1838 		if (reply == 0)
1839 			reply = do_mode_select(common, bh);
1840 		break;
1841 
1842 	case MODE_SENSE:
1843 		common->data_size_from_cmnd = common->cmnd[4];
1844 		reply = check_command(common, 6, DATA_DIR_TO_HOST,
1845 				      (1<<1) | (1<<2) | (1<<4), 0,
1846 				      "MODE SENSE(6)");
1847 		if (reply == 0)
1848 			reply = do_mode_sense(common, bh);
1849 		break;
1850 
1851 	case MODE_SENSE_10:
1852 		common->data_size_from_cmnd =
1853 			get_unaligned_be16(&common->cmnd[7]);
1854 		reply = check_command(common, 10, DATA_DIR_TO_HOST,
1855 				      (1<<1) | (1<<2) | (3<<7), 0,
1856 				      "MODE SENSE(10)");
1857 		if (reply == 0)
1858 			reply = do_mode_sense(common, bh);
1859 		break;
1860 
1861 	case ALLOW_MEDIUM_REMOVAL:
1862 		common->data_size_from_cmnd = 0;
1863 		reply = check_command(common, 6, DATA_DIR_NONE,
1864 				      (1<<4), 0,
1865 				      "PREVENT-ALLOW MEDIUM REMOVAL");
1866 		if (reply == 0)
1867 			reply = do_prevent_allow(common);
1868 		break;
1869 
1870 	case READ_6:
1871 		i = common->cmnd[4];
1872 		common->data_size_from_cmnd = (i == 0) ? 256 : i;
1873 		reply = check_command_size_in_blocks(common, 6,
1874 				      DATA_DIR_TO_HOST,
1875 				      (7<<1) | (1<<4), 1,
1876 				      "READ(6)");
1877 		if (reply == 0)
1878 			reply = do_read(common);
1879 		break;
1880 
1881 	case READ_10:
1882 		common->data_size_from_cmnd =
1883 				get_unaligned_be16(&common->cmnd[7]);
1884 		reply = check_command_size_in_blocks(common, 10,
1885 				      DATA_DIR_TO_HOST,
1886 				      (1<<1) | (0xf<<2) | (3<<7), 1,
1887 				      "READ(10)");
1888 		if (reply == 0)
1889 			reply = do_read(common);
1890 		break;
1891 
1892 	case READ_12:
1893 		common->data_size_from_cmnd =
1894 				get_unaligned_be32(&common->cmnd[6]);
1895 		reply = check_command_size_in_blocks(common, 12,
1896 				      DATA_DIR_TO_HOST,
1897 				      (1<<1) | (0xf<<2) | (0xf<<6), 1,
1898 				      "READ(12)");
1899 		if (reply == 0)
1900 			reply = do_read(common);
1901 		break;
1902 
1903 	case READ_CAPACITY:
1904 		common->data_size_from_cmnd = 8;
1905 		reply = check_command(common, 10, DATA_DIR_TO_HOST,
1906 				      (0xf<<2) | (1<<8), 1,
1907 				      "READ CAPACITY");
1908 		if (reply == 0)
1909 			reply = do_read_capacity(common, bh);
1910 		break;
1911 
1912 	case READ_HEADER:
1913 		if (!common->curlun || !common->curlun->cdrom)
1914 			goto unknown_cmnd;
1915 		common->data_size_from_cmnd =
1916 			get_unaligned_be16(&common->cmnd[7]);
1917 		reply = check_command(common, 10, DATA_DIR_TO_HOST,
1918 				      (3<<7) | (0x1f<<1), 1,
1919 				      "READ HEADER");
1920 		if (reply == 0)
1921 			reply = do_read_header(common, bh);
1922 		break;
1923 
1924 	case READ_TOC:
1925 		if (!common->curlun || !common->curlun->cdrom)
1926 			goto unknown_cmnd;
1927 		common->data_size_from_cmnd =
1928 			get_unaligned_be16(&common->cmnd[7]);
1929 		reply = check_command(common, 10, DATA_DIR_TO_HOST,
1930 				      (7<<6) | (1<<1), 1,
1931 				      "READ TOC");
1932 		if (reply == 0)
1933 			reply = do_read_toc(common, bh);
1934 		break;
1935 
1936 	case READ_FORMAT_CAPACITIES:
1937 		common->data_size_from_cmnd =
1938 			get_unaligned_be16(&common->cmnd[7]);
1939 		reply = check_command(common, 10, DATA_DIR_TO_HOST,
1940 				      (3<<7), 1,
1941 				      "READ FORMAT CAPACITIES");
1942 		if (reply == 0)
1943 			reply = do_read_format_capacities(common, bh);
1944 		break;
1945 
1946 	case REQUEST_SENSE:
1947 		common->data_size_from_cmnd = common->cmnd[4];
1948 		reply = check_command(common, 6, DATA_DIR_TO_HOST,
1949 				      (1<<4), 0,
1950 				      "REQUEST SENSE");
1951 		if (reply == 0)
1952 			reply = do_request_sense(common, bh);
1953 		break;
1954 
1955 	case START_STOP:
1956 		common->data_size_from_cmnd = 0;
1957 		reply = check_command(common, 6, DATA_DIR_NONE,
1958 				      (1<<1) | (1<<4), 0,
1959 				      "START-STOP UNIT");
1960 		if (reply == 0)
1961 			reply = do_start_stop(common);
1962 		break;
1963 
1964 	case SYNCHRONIZE_CACHE:
1965 		common->data_size_from_cmnd = 0;
1966 		reply = check_command(common, 10, DATA_DIR_NONE,
1967 				      (0xf<<2) | (3<<7), 1,
1968 				      "SYNCHRONIZE CACHE");
1969 		if (reply == 0)
1970 			reply = do_synchronize_cache(common);
1971 		break;
1972 
1973 	case TEST_UNIT_READY:
1974 		common->data_size_from_cmnd = 0;
1975 		reply = check_command(common, 6, DATA_DIR_NONE,
1976 				0, 1,
1977 				"TEST UNIT READY");
1978 		break;
1979 
1980 	/*
1981 	 * Although optional, this command is used by MS-Windows.  We
1982 	 * support a minimal version: BytChk must be 0.
1983 	 */
1984 	case VERIFY:
1985 		common->data_size_from_cmnd = 0;
1986 		reply = check_command(common, 10, DATA_DIR_NONE,
1987 				      (1<<1) | (0xf<<2) | (3<<7), 1,
1988 				      "VERIFY");
1989 		if (reply == 0)
1990 			reply = do_verify(common);
1991 		break;
1992 
1993 	case WRITE_6:
1994 		i = common->cmnd[4];
1995 		common->data_size_from_cmnd = (i == 0) ? 256 : i;
1996 		reply = check_command_size_in_blocks(common, 6,
1997 				      DATA_DIR_FROM_HOST,
1998 				      (7<<1) | (1<<4), 1,
1999 				      "WRITE(6)");
2000 		if (reply == 0)
2001 			reply = do_write(common);
2002 		break;
2003 
2004 	case WRITE_10:
2005 		common->data_size_from_cmnd =
2006 				get_unaligned_be16(&common->cmnd[7]);
2007 		reply = check_command_size_in_blocks(common, 10,
2008 				      DATA_DIR_FROM_HOST,
2009 				      (1<<1) | (0xf<<2) | (3<<7), 1,
2010 				      "WRITE(10)");
2011 		if (reply == 0)
2012 			reply = do_write(common);
2013 		break;
2014 
2015 	case WRITE_12:
2016 		common->data_size_from_cmnd =
2017 				get_unaligned_be32(&common->cmnd[6]);
2018 		reply = check_command_size_in_blocks(common, 12,
2019 				      DATA_DIR_FROM_HOST,
2020 				      (1<<1) | (0xf<<2) | (0xf<<6), 1,
2021 				      "WRITE(12)");
2022 		if (reply == 0)
2023 			reply = do_write(common);
2024 		break;
2025 
2026 	/*
2027 	 * Some mandatory commands that we recognize but don't implement.
2028 	 * They don't mean much in this setting.  It's left as an exercise
2029 	 * for anyone interested to implement RESERVE and RELEASE in terms
2030 	 * of Posix locks.
2031 	 */
2032 	case FORMAT_UNIT:
2033 	case RELEASE:
2034 	case RESERVE:
2035 	case SEND_DIAGNOSTIC:
2036 		/* Fall through */
2037 
2038 	default:
2039 unknown_cmnd:
2040 		common->data_size_from_cmnd = 0;
2041 		sprintf(unknown, "Unknown x%02x", common->cmnd[0]);
2042 		reply = check_command(common, common->cmnd_size,
2043 				      DATA_DIR_UNKNOWN, ~0, 0, unknown);
2044 		if (reply == 0) {
2045 			common->curlun->sense_data = SS_INVALID_COMMAND;
2046 			reply = -EINVAL;
2047 		}
2048 		break;
2049 	}
2050 	up_read(&common->filesem);
2051 
2052 	if (reply == -EINTR || signal_pending(current))
2053 		return -EINTR;
2054 
2055 	/* Set up the single reply buffer for finish_reply() */
2056 	if (reply == -EINVAL)
2057 		reply = 0;		/* Error reply length */
2058 	if (reply >= 0 && common->data_dir == DATA_DIR_TO_HOST) {
2059 		reply = min((u32)reply, common->data_size_from_cmnd);
2060 		bh->inreq->length = reply;
2061 		bh->state = BUF_STATE_FULL;
2062 		common->residue -= reply;
2063 	}				/* Otherwise it's already set */
2064 
2065 	return 0;
2066 }
2067 
2068 
2069 /*-------------------------------------------------------------------------*/
2070 
2071 static int received_cbw(struct fsg_dev *fsg, struct fsg_buffhd *bh)
2072 {
2073 	struct usb_request	*req = bh->outreq;
2074 	struct bulk_cb_wrap	*cbw = req->buf;
2075 	struct fsg_common	*common = fsg->common;
2076 
2077 	/* Was this a real packet?  Should it be ignored? */
2078 	if (req->status || test_bit(IGNORE_BULK_OUT, &fsg->atomic_bitflags))
2079 		return -EINVAL;
2080 
2081 	/* Is the CBW valid? */
2082 	if (req->actual != US_BULK_CB_WRAP_LEN ||
2083 			cbw->Signature != cpu_to_le32(
2084 				US_BULK_CB_SIGN)) {
2085 		DBG(fsg, "invalid CBW: len %u sig 0x%x\n",
2086 				req->actual,
2087 				le32_to_cpu(cbw->Signature));
2088 
2089 		/*
2090 		 * The Bulk-only spec says we MUST stall the IN endpoint
2091 		 * (6.6.1), so it's unavoidable.  It also says we must
2092 		 * retain this state until the next reset, but there's
2093 		 * no way to tell the controller driver it should ignore
2094 		 * Clear-Feature(HALT) requests.
2095 		 *
2096 		 * We aren't required to halt the OUT endpoint; instead
2097 		 * we can simply accept and discard any data received
2098 		 * until the next reset.
2099 		 */
2100 		wedge_bulk_in_endpoint(fsg);
2101 		set_bit(IGNORE_BULK_OUT, &fsg->atomic_bitflags);
2102 		return -EINVAL;
2103 	}
2104 
2105 	/* Is the CBW meaningful? */
2106 	if (cbw->Lun >= ARRAY_SIZE(common->luns) ||
2107 	    cbw->Flags & ~US_BULK_FLAG_IN || cbw->Length <= 0 ||
2108 	    cbw->Length > MAX_COMMAND_SIZE) {
2109 		DBG(fsg, "non-meaningful CBW: lun = %u, flags = 0x%x, "
2110 				"cmdlen %u\n",
2111 				cbw->Lun, cbw->Flags, cbw->Length);
2112 
2113 		/*
2114 		 * We can do anything we want here, so let's stall the
2115 		 * bulk pipes if we are allowed to.
2116 		 */
2117 		if (common->can_stall) {
2118 			fsg_set_halt(fsg, fsg->bulk_out);
2119 			halt_bulk_in_endpoint(fsg);
2120 		}
2121 		return -EINVAL;
2122 	}
2123 
2124 	/* Save the command for later */
2125 	common->cmnd_size = cbw->Length;
2126 	memcpy(common->cmnd, cbw->CDB, common->cmnd_size);
2127 	if (cbw->Flags & US_BULK_FLAG_IN)
2128 		common->data_dir = DATA_DIR_TO_HOST;
2129 	else
2130 		common->data_dir = DATA_DIR_FROM_HOST;
2131 	common->data_size = le32_to_cpu(cbw->DataTransferLength);
2132 	if (common->data_size == 0)
2133 		common->data_dir = DATA_DIR_NONE;
2134 	common->lun = cbw->Lun;
2135 	if (common->lun < ARRAY_SIZE(common->luns))
2136 		common->curlun = common->luns[common->lun];
2137 	else
2138 		common->curlun = NULL;
2139 	common->tag = cbw->Tag;
2140 	return 0;
2141 }
2142 
2143 static int get_next_command(struct fsg_common *common)
2144 {
2145 	struct fsg_buffhd	*bh;
2146 	int			rc = 0;
2147 
2148 	/* Wait for the next buffer to become available */
2149 	bh = common->next_buffhd_to_fill;
2150 	rc = sleep_thread(common, true, bh);
2151 	if (rc)
2152 		return rc;
2153 
2154 	/* Queue a request to read a Bulk-only CBW */
2155 	set_bulk_out_req_length(common, bh, US_BULK_CB_WRAP_LEN);
2156 	if (!start_out_transfer(common, bh))
2157 		/* Don't know what to do if common->fsg is NULL */
2158 		return -EIO;
2159 
2160 	/*
2161 	 * We will drain the buffer in software, which means we
2162 	 * can reuse it for the next filling.  No need to advance
2163 	 * next_buffhd_to_fill.
2164 	 */
2165 
2166 	/* Wait for the CBW to arrive */
2167 	rc = sleep_thread(common, true, bh);
2168 	if (rc)
2169 		return rc;
2170 
2171 	rc = fsg_is_set(common) ? received_cbw(common->fsg, bh) : -EIO;
2172 	bh->state = BUF_STATE_EMPTY;
2173 
2174 	return rc;
2175 }
2176 
2177 
2178 /*-------------------------------------------------------------------------*/
2179 
2180 static int alloc_request(struct fsg_common *common, struct usb_ep *ep,
2181 		struct usb_request **preq)
2182 {
2183 	*preq = usb_ep_alloc_request(ep, GFP_ATOMIC);
2184 	if (*preq)
2185 		return 0;
2186 	ERROR(common, "can't allocate request for %s\n", ep->name);
2187 	return -ENOMEM;
2188 }
2189 
2190 /* Reset interface setting and re-init endpoint state (toggle etc). */
2191 static int do_set_interface(struct fsg_common *common, struct fsg_dev *new_fsg)
2192 {
2193 	struct fsg_dev *fsg;
2194 	int i, rc = 0;
2195 
2196 	if (common->running)
2197 		DBG(common, "reset interface\n");
2198 
2199 reset:
2200 	/* Deallocate the requests */
2201 	if (common->fsg) {
2202 		fsg = common->fsg;
2203 
2204 		for (i = 0; i < common->fsg_num_buffers; ++i) {
2205 			struct fsg_buffhd *bh = &common->buffhds[i];
2206 
2207 			if (bh->inreq) {
2208 				usb_ep_free_request(fsg->bulk_in, bh->inreq);
2209 				bh->inreq = NULL;
2210 			}
2211 			if (bh->outreq) {
2212 				usb_ep_free_request(fsg->bulk_out, bh->outreq);
2213 				bh->outreq = NULL;
2214 			}
2215 		}
2216 
2217 		/* Disable the endpoints */
2218 		if (fsg->bulk_in_enabled) {
2219 			usb_ep_disable(fsg->bulk_in);
2220 			fsg->bulk_in_enabled = 0;
2221 		}
2222 		if (fsg->bulk_out_enabled) {
2223 			usb_ep_disable(fsg->bulk_out);
2224 			fsg->bulk_out_enabled = 0;
2225 		}
2226 
2227 		common->fsg = NULL;
2228 		wake_up(&common->fsg_wait);
2229 	}
2230 
2231 	common->running = 0;
2232 	if (!new_fsg || rc)
2233 		return rc;
2234 
2235 	common->fsg = new_fsg;
2236 	fsg = common->fsg;
2237 
2238 	/* Enable the endpoints */
2239 	rc = config_ep_by_speed(common->gadget, &(fsg->function), fsg->bulk_in);
2240 	if (rc)
2241 		goto reset;
2242 	rc = usb_ep_enable(fsg->bulk_in);
2243 	if (rc)
2244 		goto reset;
2245 	fsg->bulk_in->driver_data = common;
2246 	fsg->bulk_in_enabled = 1;
2247 
2248 	rc = config_ep_by_speed(common->gadget, &(fsg->function),
2249 				fsg->bulk_out);
2250 	if (rc)
2251 		goto reset;
2252 	rc = usb_ep_enable(fsg->bulk_out);
2253 	if (rc)
2254 		goto reset;
2255 	fsg->bulk_out->driver_data = common;
2256 	fsg->bulk_out_enabled = 1;
2257 	common->bulk_out_maxpacket = usb_endpoint_maxp(fsg->bulk_out->desc);
2258 	clear_bit(IGNORE_BULK_OUT, &fsg->atomic_bitflags);
2259 
2260 	/* Allocate the requests */
2261 	for (i = 0; i < common->fsg_num_buffers; ++i) {
2262 		struct fsg_buffhd	*bh = &common->buffhds[i];
2263 
2264 		rc = alloc_request(common, fsg->bulk_in, &bh->inreq);
2265 		if (rc)
2266 			goto reset;
2267 		rc = alloc_request(common, fsg->bulk_out, &bh->outreq);
2268 		if (rc)
2269 			goto reset;
2270 		bh->inreq->buf = bh->outreq->buf = bh->buf;
2271 		bh->inreq->context = bh->outreq->context = bh;
2272 		bh->inreq->complete = bulk_in_complete;
2273 		bh->outreq->complete = bulk_out_complete;
2274 	}
2275 
2276 	common->running = 1;
2277 	for (i = 0; i < ARRAY_SIZE(common->luns); ++i)
2278 		if (common->luns[i])
2279 			common->luns[i]->unit_attention_data =
2280 				SS_RESET_OCCURRED;
2281 	return rc;
2282 }
2283 
2284 
2285 /****************************** ALT CONFIGS ******************************/
2286 
2287 static int fsg_set_alt(struct usb_function *f, unsigned intf, unsigned alt)
2288 {
2289 	struct fsg_dev *fsg = fsg_from_func(f);
2290 	fsg->common->new_fsg = fsg;
2291 	raise_exception(fsg->common, FSG_STATE_CONFIG_CHANGE);
2292 	return USB_GADGET_DELAYED_STATUS;
2293 }
2294 
2295 static void fsg_disable(struct usb_function *f)
2296 {
2297 	struct fsg_dev *fsg = fsg_from_func(f);
2298 	fsg->common->new_fsg = NULL;
2299 	raise_exception(fsg->common, FSG_STATE_CONFIG_CHANGE);
2300 }
2301 
2302 
2303 /*-------------------------------------------------------------------------*/
2304 
2305 static void handle_exception(struct fsg_common *common)
2306 {
2307 	int			i;
2308 	struct fsg_buffhd	*bh;
2309 	enum fsg_state		old_state;
2310 	struct fsg_lun		*curlun;
2311 	unsigned int		exception_req_tag;
2312 
2313 	/*
2314 	 * Clear the existing signals.  Anything but SIGUSR1 is converted
2315 	 * into a high-priority EXIT exception.
2316 	 */
2317 	for (;;) {
2318 		int sig = kernel_dequeue_signal(NULL);
2319 		if (!sig)
2320 			break;
2321 		if (sig != SIGUSR1) {
2322 			spin_lock_irq(&common->lock);
2323 			if (common->state < FSG_STATE_EXIT)
2324 				DBG(common, "Main thread exiting on signal\n");
2325 			common->state = FSG_STATE_EXIT;
2326 			spin_unlock_irq(&common->lock);
2327 		}
2328 	}
2329 
2330 	/* Cancel all the pending transfers */
2331 	if (likely(common->fsg)) {
2332 		for (i = 0; i < common->fsg_num_buffers; ++i) {
2333 			bh = &common->buffhds[i];
2334 			if (bh->state == BUF_STATE_SENDING)
2335 				usb_ep_dequeue(common->fsg->bulk_in, bh->inreq);
2336 			if (bh->state == BUF_STATE_RECEIVING)
2337 				usb_ep_dequeue(common->fsg->bulk_out,
2338 					       bh->outreq);
2339 
2340 			/* Wait for a transfer to become idle */
2341 			if (sleep_thread(common, false, bh))
2342 				return;
2343 		}
2344 
2345 		/* Clear out the controller's fifos */
2346 		if (common->fsg->bulk_in_enabled)
2347 			usb_ep_fifo_flush(common->fsg->bulk_in);
2348 		if (common->fsg->bulk_out_enabled)
2349 			usb_ep_fifo_flush(common->fsg->bulk_out);
2350 	}
2351 
2352 	/*
2353 	 * Reset the I/O buffer states and pointers, the SCSI
2354 	 * state, and the exception.  Then invoke the handler.
2355 	 */
2356 	spin_lock_irq(&common->lock);
2357 
2358 	for (i = 0; i < common->fsg_num_buffers; ++i) {
2359 		bh = &common->buffhds[i];
2360 		bh->state = BUF_STATE_EMPTY;
2361 	}
2362 	common->next_buffhd_to_fill = &common->buffhds[0];
2363 	common->next_buffhd_to_drain = &common->buffhds[0];
2364 	exception_req_tag = common->exception_req_tag;
2365 	old_state = common->state;
2366 	common->state = FSG_STATE_NORMAL;
2367 
2368 	if (old_state != FSG_STATE_ABORT_BULK_OUT) {
2369 		for (i = 0; i < ARRAY_SIZE(common->luns); ++i) {
2370 			curlun = common->luns[i];
2371 			if (!curlun)
2372 				continue;
2373 			curlun->prevent_medium_removal = 0;
2374 			curlun->sense_data = SS_NO_SENSE;
2375 			curlun->unit_attention_data = SS_NO_SENSE;
2376 			curlun->sense_data_info = 0;
2377 			curlun->info_valid = 0;
2378 		}
2379 	}
2380 	spin_unlock_irq(&common->lock);
2381 
2382 	/* Carry out any extra actions required for the exception */
2383 	switch (old_state) {
2384 	case FSG_STATE_NORMAL:
2385 		break;
2386 
2387 	case FSG_STATE_ABORT_BULK_OUT:
2388 		send_status(common);
2389 		break;
2390 
2391 	case FSG_STATE_PROTOCOL_RESET:
2392 		/*
2393 		 * In case we were forced against our will to halt a
2394 		 * bulk endpoint, clear the halt now.  (The SuperH UDC
2395 		 * requires this.)
2396 		 */
2397 		if (!fsg_is_set(common))
2398 			break;
2399 		if (test_and_clear_bit(IGNORE_BULK_OUT,
2400 				       &common->fsg->atomic_bitflags))
2401 			usb_ep_clear_halt(common->fsg->bulk_in);
2402 
2403 		if (common->ep0_req_tag == exception_req_tag)
2404 			ep0_queue(common);	/* Complete the status stage */
2405 
2406 		/*
2407 		 * Technically this should go here, but it would only be
2408 		 * a waste of time.  Ditto for the INTERFACE_CHANGE and
2409 		 * CONFIG_CHANGE cases.
2410 		 */
2411 		/* for (i = 0; i < common->ARRAY_SIZE(common->luns); ++i) */
2412 		/*	if (common->luns[i]) */
2413 		/*		common->luns[i]->unit_attention_data = */
2414 		/*			SS_RESET_OCCURRED;  */
2415 		break;
2416 
2417 	case FSG_STATE_CONFIG_CHANGE:
2418 		do_set_interface(common, common->new_fsg);
2419 		if (common->new_fsg)
2420 			usb_composite_setup_continue(common->cdev);
2421 		break;
2422 
2423 	case FSG_STATE_EXIT:
2424 		do_set_interface(common, NULL);		/* Free resources */
2425 		spin_lock_irq(&common->lock);
2426 		common->state = FSG_STATE_TERMINATED;	/* Stop the thread */
2427 		spin_unlock_irq(&common->lock);
2428 		break;
2429 
2430 	case FSG_STATE_TERMINATED:
2431 		break;
2432 	}
2433 }
2434 
2435 
2436 /*-------------------------------------------------------------------------*/
2437 
2438 static int fsg_main_thread(void *common_)
2439 {
2440 	struct fsg_common	*common = common_;
2441 
2442 	/*
2443 	 * Allow the thread to be killed by a signal, but set the signal mask
2444 	 * to block everything but INT, TERM, KILL, and USR1.
2445 	 */
2446 	allow_signal(SIGINT);
2447 	allow_signal(SIGTERM);
2448 	allow_signal(SIGKILL);
2449 	allow_signal(SIGUSR1);
2450 
2451 	/* Allow the thread to be frozen */
2452 	set_freezable();
2453 
2454 	/* The main loop */
2455 	while (common->state != FSG_STATE_TERMINATED) {
2456 		if (exception_in_progress(common) || signal_pending(current)) {
2457 			handle_exception(common);
2458 			continue;
2459 		}
2460 
2461 		if (!common->running) {
2462 			sleep_thread(common, true, NULL);
2463 			continue;
2464 		}
2465 
2466 		if (get_next_command(common) || exception_in_progress(common))
2467 			continue;
2468 		if (do_scsi_command(common) || exception_in_progress(common))
2469 			continue;
2470 		if (finish_reply(common) || exception_in_progress(common))
2471 			continue;
2472 		send_status(common);
2473 	}
2474 
2475 	spin_lock_irq(&common->lock);
2476 	common->thread_task = NULL;
2477 	spin_unlock_irq(&common->lock);
2478 
2479 	if (!common->ops || !common->ops->thread_exits
2480 	 || common->ops->thread_exits(common) < 0) {
2481 		int i;
2482 
2483 		down_write(&common->filesem);
2484 		for (i = 0; i < ARRAY_SIZE(common->luns); i++) {
2485 			struct fsg_lun *curlun = common->luns[i];
2486 			if (!curlun || !fsg_lun_is_open(curlun))
2487 				continue;
2488 
2489 			fsg_lun_close(curlun);
2490 			curlun->unit_attention_data = SS_MEDIUM_NOT_PRESENT;
2491 		}
2492 		up_write(&common->filesem);
2493 	}
2494 
2495 	/* Let fsg_unbind() know the thread has exited */
2496 	complete_and_exit(&common->thread_notifier, 0);
2497 }
2498 
2499 
2500 /*************************** DEVICE ATTRIBUTES ***************************/
2501 
2502 static ssize_t ro_show(struct device *dev, struct device_attribute *attr, char *buf)
2503 {
2504 	struct fsg_lun		*curlun = fsg_lun_from_dev(dev);
2505 
2506 	return fsg_show_ro(curlun, buf);
2507 }
2508 
2509 static ssize_t nofua_show(struct device *dev, struct device_attribute *attr,
2510 			  char *buf)
2511 {
2512 	struct fsg_lun		*curlun = fsg_lun_from_dev(dev);
2513 
2514 	return fsg_show_nofua(curlun, buf);
2515 }
2516 
2517 static ssize_t file_show(struct device *dev, struct device_attribute *attr,
2518 			 char *buf)
2519 {
2520 	struct fsg_lun		*curlun = fsg_lun_from_dev(dev);
2521 	struct rw_semaphore	*filesem = dev_get_drvdata(dev);
2522 
2523 	return fsg_show_file(curlun, filesem, buf);
2524 }
2525 
2526 static ssize_t ro_store(struct device *dev, struct device_attribute *attr,
2527 			const char *buf, size_t count)
2528 {
2529 	struct fsg_lun		*curlun = fsg_lun_from_dev(dev);
2530 	struct rw_semaphore	*filesem = dev_get_drvdata(dev);
2531 
2532 	return fsg_store_ro(curlun, filesem, buf, count);
2533 }
2534 
2535 static ssize_t nofua_store(struct device *dev, struct device_attribute *attr,
2536 			   const char *buf, size_t count)
2537 {
2538 	struct fsg_lun		*curlun = fsg_lun_from_dev(dev);
2539 
2540 	return fsg_store_nofua(curlun, buf, count);
2541 }
2542 
2543 static ssize_t file_store(struct device *dev, struct device_attribute *attr,
2544 			  const char *buf, size_t count)
2545 {
2546 	struct fsg_lun		*curlun = fsg_lun_from_dev(dev);
2547 	struct rw_semaphore	*filesem = dev_get_drvdata(dev);
2548 
2549 	return fsg_store_file(curlun, filesem, buf, count);
2550 }
2551 
2552 static DEVICE_ATTR_RW(nofua);
2553 /* mode wil be set in fsg_lun_attr_is_visible() */
2554 static DEVICE_ATTR(ro, 0, ro_show, ro_store);
2555 static DEVICE_ATTR(file, 0, file_show, file_store);
2556 
2557 /****************************** FSG COMMON ******************************/
2558 
2559 static void fsg_common_release(struct kref *ref);
2560 
2561 static void fsg_lun_release(struct device *dev)
2562 {
2563 	/* Nothing needs to be done */
2564 }
2565 
2566 void fsg_common_get(struct fsg_common *common)
2567 {
2568 	kref_get(&common->ref);
2569 }
2570 EXPORT_SYMBOL_GPL(fsg_common_get);
2571 
2572 void fsg_common_put(struct fsg_common *common)
2573 {
2574 	kref_put(&common->ref, fsg_common_release);
2575 }
2576 EXPORT_SYMBOL_GPL(fsg_common_put);
2577 
2578 static struct fsg_common *fsg_common_setup(struct fsg_common *common)
2579 {
2580 	if (!common) {
2581 		common = kzalloc(sizeof(*common), GFP_KERNEL);
2582 		if (!common)
2583 			return ERR_PTR(-ENOMEM);
2584 		common->free_storage_on_release = 1;
2585 	} else {
2586 		common->free_storage_on_release = 0;
2587 	}
2588 	init_rwsem(&common->filesem);
2589 	spin_lock_init(&common->lock);
2590 	kref_init(&common->ref);
2591 	init_completion(&common->thread_notifier);
2592 	init_waitqueue_head(&common->io_wait);
2593 	init_waitqueue_head(&common->fsg_wait);
2594 	common->state = FSG_STATE_TERMINATED;
2595 	memset(common->luns, 0, sizeof(common->luns));
2596 
2597 	return common;
2598 }
2599 
2600 void fsg_common_set_sysfs(struct fsg_common *common, bool sysfs)
2601 {
2602 	common->sysfs = sysfs;
2603 }
2604 EXPORT_SYMBOL_GPL(fsg_common_set_sysfs);
2605 
2606 static void _fsg_common_free_buffers(struct fsg_buffhd *buffhds, unsigned n)
2607 {
2608 	if (buffhds) {
2609 		struct fsg_buffhd *bh = buffhds;
2610 		while (n--) {
2611 			kfree(bh->buf);
2612 			++bh;
2613 		}
2614 		kfree(buffhds);
2615 	}
2616 }
2617 
2618 int fsg_common_set_num_buffers(struct fsg_common *common, unsigned int n)
2619 {
2620 	struct fsg_buffhd *bh, *buffhds;
2621 	int i;
2622 
2623 	buffhds = kcalloc(n, sizeof(*buffhds), GFP_KERNEL);
2624 	if (!buffhds)
2625 		return -ENOMEM;
2626 
2627 	/* Data buffers cyclic list */
2628 	bh = buffhds;
2629 	i = n;
2630 	goto buffhds_first_it;
2631 	do {
2632 		bh->next = bh + 1;
2633 		++bh;
2634 buffhds_first_it:
2635 		bh->buf = kmalloc(FSG_BUFLEN, GFP_KERNEL);
2636 		if (unlikely(!bh->buf))
2637 			goto error_release;
2638 	} while (--i);
2639 	bh->next = buffhds;
2640 
2641 	_fsg_common_free_buffers(common->buffhds, common->fsg_num_buffers);
2642 	common->fsg_num_buffers = n;
2643 	common->buffhds = buffhds;
2644 
2645 	return 0;
2646 
2647 error_release:
2648 	/*
2649 	 * "buf"s pointed to by heads after n - i are NULL
2650 	 * so releasing them won't hurt
2651 	 */
2652 	_fsg_common_free_buffers(buffhds, n);
2653 
2654 	return -ENOMEM;
2655 }
2656 EXPORT_SYMBOL_GPL(fsg_common_set_num_buffers);
2657 
2658 void fsg_common_remove_lun(struct fsg_lun *lun)
2659 {
2660 	if (device_is_registered(&lun->dev))
2661 		device_unregister(&lun->dev);
2662 	fsg_lun_close(lun);
2663 	kfree(lun);
2664 }
2665 EXPORT_SYMBOL_GPL(fsg_common_remove_lun);
2666 
2667 static void _fsg_common_remove_luns(struct fsg_common *common, int n)
2668 {
2669 	int i;
2670 
2671 	for (i = 0; i < n; ++i)
2672 		if (common->luns[i]) {
2673 			fsg_common_remove_lun(common->luns[i]);
2674 			common->luns[i] = NULL;
2675 		}
2676 }
2677 
2678 void fsg_common_remove_luns(struct fsg_common *common)
2679 {
2680 	_fsg_common_remove_luns(common, ARRAY_SIZE(common->luns));
2681 }
2682 EXPORT_SYMBOL_GPL(fsg_common_remove_luns);
2683 
2684 void fsg_common_set_ops(struct fsg_common *common,
2685 			const struct fsg_operations *ops)
2686 {
2687 	common->ops = ops;
2688 }
2689 EXPORT_SYMBOL_GPL(fsg_common_set_ops);
2690 
2691 void fsg_common_free_buffers(struct fsg_common *common)
2692 {
2693 	_fsg_common_free_buffers(common->buffhds, common->fsg_num_buffers);
2694 	common->buffhds = NULL;
2695 }
2696 EXPORT_SYMBOL_GPL(fsg_common_free_buffers);
2697 
2698 int fsg_common_set_cdev(struct fsg_common *common,
2699 			 struct usb_composite_dev *cdev, bool can_stall)
2700 {
2701 	struct usb_string *us;
2702 
2703 	common->gadget = cdev->gadget;
2704 	common->ep0 = cdev->gadget->ep0;
2705 	common->ep0req = cdev->req;
2706 	common->cdev = cdev;
2707 
2708 	us = usb_gstrings_attach(cdev, fsg_strings_array,
2709 				 ARRAY_SIZE(fsg_strings));
2710 	if (IS_ERR(us))
2711 		return PTR_ERR(us);
2712 
2713 	fsg_intf_desc.iInterface = us[FSG_STRING_INTERFACE].id;
2714 
2715 	/*
2716 	 * Some peripheral controllers are known not to be able to
2717 	 * halt bulk endpoints correctly.  If one of them is present,
2718 	 * disable stalls.
2719 	 */
2720 	common->can_stall = can_stall &&
2721 			gadget_is_stall_supported(common->gadget);
2722 
2723 	return 0;
2724 }
2725 EXPORT_SYMBOL_GPL(fsg_common_set_cdev);
2726 
2727 static struct attribute *fsg_lun_dev_attrs[] = {
2728 	&dev_attr_ro.attr,
2729 	&dev_attr_file.attr,
2730 	&dev_attr_nofua.attr,
2731 	NULL
2732 };
2733 
2734 static umode_t fsg_lun_dev_is_visible(struct kobject *kobj,
2735 				      struct attribute *attr, int idx)
2736 {
2737 	struct device *dev = kobj_to_dev(kobj);
2738 	struct fsg_lun *lun = fsg_lun_from_dev(dev);
2739 
2740 	if (attr == &dev_attr_ro.attr)
2741 		return lun->cdrom ? S_IRUGO : (S_IWUSR | S_IRUGO);
2742 	if (attr == &dev_attr_file.attr)
2743 		return lun->removable ? (S_IWUSR | S_IRUGO) : S_IRUGO;
2744 	return attr->mode;
2745 }
2746 
2747 static const struct attribute_group fsg_lun_dev_group = {
2748 	.attrs = fsg_lun_dev_attrs,
2749 	.is_visible = fsg_lun_dev_is_visible,
2750 };
2751 
2752 static const struct attribute_group *fsg_lun_dev_groups[] = {
2753 	&fsg_lun_dev_group,
2754 	NULL
2755 };
2756 
2757 int fsg_common_create_lun(struct fsg_common *common, struct fsg_lun_config *cfg,
2758 			  unsigned int id, const char *name,
2759 			  const char **name_pfx)
2760 {
2761 	struct fsg_lun *lun;
2762 	char *pathbuf, *p;
2763 	int rc = -ENOMEM;
2764 
2765 	if (id >= ARRAY_SIZE(common->luns))
2766 		return -ENODEV;
2767 
2768 	if (common->luns[id])
2769 		return -EBUSY;
2770 
2771 	if (!cfg->filename && !cfg->removable) {
2772 		pr_err("no file given for LUN%d\n", id);
2773 		return -EINVAL;
2774 	}
2775 
2776 	lun = kzalloc(sizeof(*lun), GFP_KERNEL);
2777 	if (!lun)
2778 		return -ENOMEM;
2779 
2780 	lun->name_pfx = name_pfx;
2781 
2782 	lun->cdrom = !!cfg->cdrom;
2783 	lun->ro = cfg->cdrom || cfg->ro;
2784 	lun->initially_ro = lun->ro;
2785 	lun->removable = !!cfg->removable;
2786 
2787 	if (!common->sysfs) {
2788 		/* we DON'T own the name!*/
2789 		lun->name = name;
2790 	} else {
2791 		lun->dev.release = fsg_lun_release;
2792 		lun->dev.parent = &common->gadget->dev;
2793 		lun->dev.groups = fsg_lun_dev_groups;
2794 		dev_set_drvdata(&lun->dev, &common->filesem);
2795 		dev_set_name(&lun->dev, "%s", name);
2796 		lun->name = dev_name(&lun->dev);
2797 
2798 		rc = device_register(&lun->dev);
2799 		if (rc) {
2800 			pr_info("failed to register LUN%d: %d\n", id, rc);
2801 			put_device(&lun->dev);
2802 			goto error_sysfs;
2803 		}
2804 	}
2805 
2806 	common->luns[id] = lun;
2807 
2808 	if (cfg->filename) {
2809 		rc = fsg_lun_open(lun, cfg->filename);
2810 		if (rc)
2811 			goto error_lun;
2812 	}
2813 
2814 	pathbuf = kmalloc(PATH_MAX, GFP_KERNEL);
2815 	p = "(no medium)";
2816 	if (fsg_lun_is_open(lun)) {
2817 		p = "(error)";
2818 		if (pathbuf) {
2819 			p = file_path(lun->filp, pathbuf, PATH_MAX);
2820 			if (IS_ERR(p))
2821 				p = "(error)";
2822 		}
2823 	}
2824 	pr_info("LUN: %s%s%sfile: %s\n",
2825 	      lun->removable ? "removable " : "",
2826 	      lun->ro ? "read only " : "",
2827 	      lun->cdrom ? "CD-ROM " : "",
2828 	      p);
2829 	kfree(pathbuf);
2830 
2831 	return 0;
2832 
2833 error_lun:
2834 	if (device_is_registered(&lun->dev))
2835 		device_unregister(&lun->dev);
2836 	fsg_lun_close(lun);
2837 	common->luns[id] = NULL;
2838 error_sysfs:
2839 	kfree(lun);
2840 	return rc;
2841 }
2842 EXPORT_SYMBOL_GPL(fsg_common_create_lun);
2843 
2844 int fsg_common_create_luns(struct fsg_common *common, struct fsg_config *cfg)
2845 {
2846 	char buf[8]; /* enough for 100000000 different numbers, decimal */
2847 	int i, rc;
2848 
2849 	fsg_common_remove_luns(common);
2850 
2851 	for (i = 0; i < cfg->nluns; ++i) {
2852 		snprintf(buf, sizeof(buf), "lun%d", i);
2853 		rc = fsg_common_create_lun(common, &cfg->luns[i], i, buf, NULL);
2854 		if (rc)
2855 			goto fail;
2856 	}
2857 
2858 	pr_info("Number of LUNs=%d\n", cfg->nluns);
2859 
2860 	return 0;
2861 
2862 fail:
2863 	_fsg_common_remove_luns(common, i);
2864 	return rc;
2865 }
2866 EXPORT_SYMBOL_GPL(fsg_common_create_luns);
2867 
2868 void fsg_common_set_inquiry_string(struct fsg_common *common, const char *vn,
2869 				   const char *pn)
2870 {
2871 	int i;
2872 
2873 	/* Prepare inquiryString */
2874 	i = get_default_bcdDevice();
2875 	snprintf(common->inquiry_string, sizeof(common->inquiry_string),
2876 		 "%-8s%-16s%04x", vn ?: "Linux",
2877 		 /* Assume product name dependent on the first LUN */
2878 		 pn ?: ((*common->luns)->cdrom
2879 		     ? "File-CD Gadget"
2880 		     : "File-Stor Gadget"),
2881 		 i);
2882 }
2883 EXPORT_SYMBOL_GPL(fsg_common_set_inquiry_string);
2884 
2885 static void fsg_common_release(struct kref *ref)
2886 {
2887 	struct fsg_common *common = container_of(ref, struct fsg_common, ref);
2888 	int i;
2889 
2890 	/* If the thread isn't already dead, tell it to exit now */
2891 	if (common->state != FSG_STATE_TERMINATED) {
2892 		raise_exception(common, FSG_STATE_EXIT);
2893 		wait_for_completion(&common->thread_notifier);
2894 	}
2895 
2896 	for (i = 0; i < ARRAY_SIZE(common->luns); ++i) {
2897 		struct fsg_lun *lun = common->luns[i];
2898 		if (!lun)
2899 			continue;
2900 		fsg_lun_close(lun);
2901 		if (device_is_registered(&lun->dev))
2902 			device_unregister(&lun->dev);
2903 		kfree(lun);
2904 	}
2905 
2906 	_fsg_common_free_buffers(common->buffhds, common->fsg_num_buffers);
2907 	if (common->free_storage_on_release)
2908 		kfree(common);
2909 }
2910 
2911 
2912 /*-------------------------------------------------------------------------*/
2913 
2914 static int fsg_bind(struct usb_configuration *c, struct usb_function *f)
2915 {
2916 	struct fsg_dev		*fsg = fsg_from_func(f);
2917 	struct fsg_common	*common = fsg->common;
2918 	struct usb_gadget	*gadget = c->cdev->gadget;
2919 	int			i;
2920 	struct usb_ep		*ep;
2921 	unsigned		max_burst;
2922 	int			ret;
2923 	struct fsg_opts		*opts;
2924 
2925 	/* Don't allow to bind if we don't have at least one LUN */
2926 	ret = _fsg_common_get_max_lun(common);
2927 	if (ret < 0) {
2928 		pr_err("There should be at least one LUN.\n");
2929 		return -EINVAL;
2930 	}
2931 
2932 	opts = fsg_opts_from_func_inst(f->fi);
2933 	if (!opts->no_configfs) {
2934 		ret = fsg_common_set_cdev(fsg->common, c->cdev,
2935 					  fsg->common->can_stall);
2936 		if (ret)
2937 			return ret;
2938 		fsg_common_set_inquiry_string(fsg->common, NULL, NULL);
2939 	}
2940 
2941 	if (!common->thread_task) {
2942 		common->state = FSG_STATE_NORMAL;
2943 		common->thread_task =
2944 			kthread_create(fsg_main_thread, common, "file-storage");
2945 		if (IS_ERR(common->thread_task)) {
2946 			ret = PTR_ERR(common->thread_task);
2947 			common->thread_task = NULL;
2948 			common->state = FSG_STATE_TERMINATED;
2949 			return ret;
2950 		}
2951 		DBG(common, "I/O thread pid: %d\n",
2952 		    task_pid_nr(common->thread_task));
2953 		wake_up_process(common->thread_task);
2954 	}
2955 
2956 	fsg->gadget = gadget;
2957 
2958 	/* New interface */
2959 	i = usb_interface_id(c, f);
2960 	if (i < 0)
2961 		goto fail;
2962 	fsg_intf_desc.bInterfaceNumber = i;
2963 	fsg->interface_number = i;
2964 
2965 	/* Find all the endpoints we will use */
2966 	ep = usb_ep_autoconfig(gadget, &fsg_fs_bulk_in_desc);
2967 	if (!ep)
2968 		goto autoconf_fail;
2969 	fsg->bulk_in = ep;
2970 
2971 	ep = usb_ep_autoconfig(gadget, &fsg_fs_bulk_out_desc);
2972 	if (!ep)
2973 		goto autoconf_fail;
2974 	fsg->bulk_out = ep;
2975 
2976 	/* Assume endpoint addresses are the same for both speeds */
2977 	fsg_hs_bulk_in_desc.bEndpointAddress =
2978 		fsg_fs_bulk_in_desc.bEndpointAddress;
2979 	fsg_hs_bulk_out_desc.bEndpointAddress =
2980 		fsg_fs_bulk_out_desc.bEndpointAddress;
2981 
2982 	/* Calculate bMaxBurst, we know packet size is 1024 */
2983 	max_burst = min_t(unsigned, FSG_BUFLEN / 1024, 15);
2984 
2985 	fsg_ss_bulk_in_desc.bEndpointAddress =
2986 		fsg_fs_bulk_in_desc.bEndpointAddress;
2987 	fsg_ss_bulk_in_comp_desc.bMaxBurst = max_burst;
2988 
2989 	fsg_ss_bulk_out_desc.bEndpointAddress =
2990 		fsg_fs_bulk_out_desc.bEndpointAddress;
2991 	fsg_ss_bulk_out_comp_desc.bMaxBurst = max_burst;
2992 
2993 	ret = usb_assign_descriptors(f, fsg_fs_function, fsg_hs_function,
2994 			fsg_ss_function, fsg_ss_function);
2995 	if (ret)
2996 		goto autoconf_fail;
2997 
2998 	return 0;
2999 
3000 autoconf_fail:
3001 	ERROR(fsg, "unable to autoconfigure all endpoints\n");
3002 	i = -ENOTSUPP;
3003 fail:
3004 	/* terminate the thread */
3005 	if (fsg->common->state != FSG_STATE_TERMINATED) {
3006 		raise_exception(fsg->common, FSG_STATE_EXIT);
3007 		wait_for_completion(&fsg->common->thread_notifier);
3008 	}
3009 	return i;
3010 }
3011 
3012 /****************************** ALLOCATE FUNCTION *************************/
3013 
3014 static void fsg_unbind(struct usb_configuration *c, struct usb_function *f)
3015 {
3016 	struct fsg_dev		*fsg = fsg_from_func(f);
3017 	struct fsg_common	*common = fsg->common;
3018 
3019 	DBG(fsg, "unbind\n");
3020 	if (fsg->common->fsg == fsg) {
3021 		fsg->common->new_fsg = NULL;
3022 		raise_exception(fsg->common, FSG_STATE_CONFIG_CHANGE);
3023 		/* FIXME: make interruptible or killable somehow? */
3024 		wait_event(common->fsg_wait, common->fsg != fsg);
3025 	}
3026 
3027 	usb_free_all_descriptors(&fsg->function);
3028 }
3029 
3030 static inline struct fsg_lun_opts *to_fsg_lun_opts(struct config_item *item)
3031 {
3032 	return container_of(to_config_group(item), struct fsg_lun_opts, group);
3033 }
3034 
3035 static inline struct fsg_opts *to_fsg_opts(struct config_item *item)
3036 {
3037 	return container_of(to_config_group(item), struct fsg_opts,
3038 			    func_inst.group);
3039 }
3040 
3041 static void fsg_lun_attr_release(struct config_item *item)
3042 {
3043 	struct fsg_lun_opts *lun_opts;
3044 
3045 	lun_opts = to_fsg_lun_opts(item);
3046 	kfree(lun_opts);
3047 }
3048 
3049 static struct configfs_item_operations fsg_lun_item_ops = {
3050 	.release		= fsg_lun_attr_release,
3051 };
3052 
3053 static ssize_t fsg_lun_opts_file_show(struct config_item *item, char *page)
3054 {
3055 	struct fsg_lun_opts *opts = to_fsg_lun_opts(item);
3056 	struct fsg_opts *fsg_opts = to_fsg_opts(opts->group.cg_item.ci_parent);
3057 
3058 	return fsg_show_file(opts->lun, &fsg_opts->common->filesem, page);
3059 }
3060 
3061 static ssize_t fsg_lun_opts_file_store(struct config_item *item,
3062 				       const char *page, size_t len)
3063 {
3064 	struct fsg_lun_opts *opts = to_fsg_lun_opts(item);
3065 	struct fsg_opts *fsg_opts = to_fsg_opts(opts->group.cg_item.ci_parent);
3066 
3067 	return fsg_store_file(opts->lun, &fsg_opts->common->filesem, page, len);
3068 }
3069 
3070 CONFIGFS_ATTR(fsg_lun_opts_, file);
3071 
3072 static ssize_t fsg_lun_opts_ro_show(struct config_item *item, char *page)
3073 {
3074 	return fsg_show_ro(to_fsg_lun_opts(item)->lun, page);
3075 }
3076 
3077 static ssize_t fsg_lun_opts_ro_store(struct config_item *item,
3078 				       const char *page, size_t len)
3079 {
3080 	struct fsg_lun_opts *opts = to_fsg_lun_opts(item);
3081 	struct fsg_opts *fsg_opts = to_fsg_opts(opts->group.cg_item.ci_parent);
3082 
3083 	return fsg_store_ro(opts->lun, &fsg_opts->common->filesem, page, len);
3084 }
3085 
3086 CONFIGFS_ATTR(fsg_lun_opts_, ro);
3087 
3088 static ssize_t fsg_lun_opts_removable_show(struct config_item *item,
3089 					   char *page)
3090 {
3091 	return fsg_show_removable(to_fsg_lun_opts(item)->lun, page);
3092 }
3093 
3094 static ssize_t fsg_lun_opts_removable_store(struct config_item *item,
3095 				       const char *page, size_t len)
3096 {
3097 	return fsg_store_removable(to_fsg_lun_opts(item)->lun, page, len);
3098 }
3099 
3100 CONFIGFS_ATTR(fsg_lun_opts_, removable);
3101 
3102 static ssize_t fsg_lun_opts_cdrom_show(struct config_item *item, char *page)
3103 {
3104 	return fsg_show_cdrom(to_fsg_lun_opts(item)->lun, page);
3105 }
3106 
3107 static ssize_t fsg_lun_opts_cdrom_store(struct config_item *item,
3108 				       const char *page, size_t len)
3109 {
3110 	struct fsg_lun_opts *opts = to_fsg_lun_opts(item);
3111 	struct fsg_opts *fsg_opts = to_fsg_opts(opts->group.cg_item.ci_parent);
3112 
3113 	return fsg_store_cdrom(opts->lun, &fsg_opts->common->filesem, page,
3114 			       len);
3115 }
3116 
3117 CONFIGFS_ATTR(fsg_lun_opts_, cdrom);
3118 
3119 static ssize_t fsg_lun_opts_nofua_show(struct config_item *item, char *page)
3120 {
3121 	return fsg_show_nofua(to_fsg_lun_opts(item)->lun, page);
3122 }
3123 
3124 static ssize_t fsg_lun_opts_nofua_store(struct config_item *item,
3125 				       const char *page, size_t len)
3126 {
3127 	return fsg_store_nofua(to_fsg_lun_opts(item)->lun, page, len);
3128 }
3129 
3130 CONFIGFS_ATTR(fsg_lun_opts_, nofua);
3131 
3132 static ssize_t fsg_lun_opts_inquiry_string_show(struct config_item *item,
3133 						char *page)
3134 {
3135 	return fsg_show_inquiry_string(to_fsg_lun_opts(item)->lun, page);
3136 }
3137 
3138 static ssize_t fsg_lun_opts_inquiry_string_store(struct config_item *item,
3139 						 const char *page, size_t len)
3140 {
3141 	return fsg_store_inquiry_string(to_fsg_lun_opts(item)->lun, page, len);
3142 }
3143 
3144 CONFIGFS_ATTR(fsg_lun_opts_, inquiry_string);
3145 
3146 static struct configfs_attribute *fsg_lun_attrs[] = {
3147 	&fsg_lun_opts_attr_file,
3148 	&fsg_lun_opts_attr_ro,
3149 	&fsg_lun_opts_attr_removable,
3150 	&fsg_lun_opts_attr_cdrom,
3151 	&fsg_lun_opts_attr_nofua,
3152 	&fsg_lun_opts_attr_inquiry_string,
3153 	NULL,
3154 };
3155 
3156 static struct config_item_type fsg_lun_type = {
3157 	.ct_item_ops	= &fsg_lun_item_ops,
3158 	.ct_attrs	= fsg_lun_attrs,
3159 	.ct_owner	= THIS_MODULE,
3160 };
3161 
3162 static struct config_group *fsg_lun_make(struct config_group *group,
3163 					 const char *name)
3164 {
3165 	struct fsg_lun_opts *opts;
3166 	struct fsg_opts *fsg_opts;
3167 	struct fsg_lun_config config;
3168 	char *num_str;
3169 	u8 num;
3170 	int ret;
3171 
3172 	num_str = strchr(name, '.');
3173 	if (!num_str) {
3174 		pr_err("Unable to locate . in LUN.NUMBER\n");
3175 		return ERR_PTR(-EINVAL);
3176 	}
3177 	num_str++;
3178 
3179 	ret = kstrtou8(num_str, 0, &num);
3180 	if (ret)
3181 		return ERR_PTR(ret);
3182 
3183 	fsg_opts = to_fsg_opts(&group->cg_item);
3184 	if (num >= FSG_MAX_LUNS)
3185 		return ERR_PTR(-ERANGE);
3186 
3187 	mutex_lock(&fsg_opts->lock);
3188 	if (fsg_opts->refcnt || fsg_opts->common->luns[num]) {
3189 		ret = -EBUSY;
3190 		goto out;
3191 	}
3192 
3193 	opts = kzalloc(sizeof(*opts), GFP_KERNEL);
3194 	if (!opts) {
3195 		ret = -ENOMEM;
3196 		goto out;
3197 	}
3198 
3199 	memset(&config, 0, sizeof(config));
3200 	config.removable = true;
3201 
3202 	ret = fsg_common_create_lun(fsg_opts->common, &config, num, name,
3203 				    (const char **)&group->cg_item.ci_name);
3204 	if (ret) {
3205 		kfree(opts);
3206 		goto out;
3207 	}
3208 	opts->lun = fsg_opts->common->luns[num];
3209 	opts->lun_id = num;
3210 	mutex_unlock(&fsg_opts->lock);
3211 
3212 	config_group_init_type_name(&opts->group, name, &fsg_lun_type);
3213 
3214 	return &opts->group;
3215 out:
3216 	mutex_unlock(&fsg_opts->lock);
3217 	return ERR_PTR(ret);
3218 }
3219 
3220 static void fsg_lun_drop(struct config_group *group, struct config_item *item)
3221 {
3222 	struct fsg_lun_opts *lun_opts;
3223 	struct fsg_opts *fsg_opts;
3224 
3225 	lun_opts = to_fsg_lun_opts(item);
3226 	fsg_opts = to_fsg_opts(&group->cg_item);
3227 
3228 	mutex_lock(&fsg_opts->lock);
3229 	if (fsg_opts->refcnt) {
3230 		struct config_item *gadget;
3231 
3232 		gadget = group->cg_item.ci_parent->ci_parent;
3233 		unregister_gadget_item(gadget);
3234 	}
3235 
3236 	fsg_common_remove_lun(lun_opts->lun);
3237 	fsg_opts->common->luns[lun_opts->lun_id] = NULL;
3238 	lun_opts->lun_id = 0;
3239 	mutex_unlock(&fsg_opts->lock);
3240 
3241 	config_item_put(item);
3242 }
3243 
3244 static void fsg_attr_release(struct config_item *item)
3245 {
3246 	struct fsg_opts *opts = to_fsg_opts(item);
3247 
3248 	usb_put_function_instance(&opts->func_inst);
3249 }
3250 
3251 static struct configfs_item_operations fsg_item_ops = {
3252 	.release		= fsg_attr_release,
3253 };
3254 
3255 static ssize_t fsg_opts_stall_show(struct config_item *item, char *page)
3256 {
3257 	struct fsg_opts *opts = to_fsg_opts(item);
3258 	int result;
3259 
3260 	mutex_lock(&opts->lock);
3261 	result = sprintf(page, "%d", opts->common->can_stall);
3262 	mutex_unlock(&opts->lock);
3263 
3264 	return result;
3265 }
3266 
3267 static ssize_t fsg_opts_stall_store(struct config_item *item, const char *page,
3268 				    size_t len)
3269 {
3270 	struct fsg_opts *opts = to_fsg_opts(item);
3271 	int ret;
3272 	bool stall;
3273 
3274 	mutex_lock(&opts->lock);
3275 
3276 	if (opts->refcnt) {
3277 		mutex_unlock(&opts->lock);
3278 		return -EBUSY;
3279 	}
3280 
3281 	ret = strtobool(page, &stall);
3282 	if (!ret) {
3283 		opts->common->can_stall = stall;
3284 		ret = len;
3285 	}
3286 
3287 	mutex_unlock(&opts->lock);
3288 
3289 	return ret;
3290 }
3291 
3292 CONFIGFS_ATTR(fsg_opts_, stall);
3293 
3294 #ifdef CONFIG_USB_GADGET_DEBUG_FILES
3295 static ssize_t fsg_opts_num_buffers_show(struct config_item *item, char *page)
3296 {
3297 	struct fsg_opts *opts = to_fsg_opts(item);
3298 	int result;
3299 
3300 	mutex_lock(&opts->lock);
3301 	result = sprintf(page, "%d", opts->common->fsg_num_buffers);
3302 	mutex_unlock(&opts->lock);
3303 
3304 	return result;
3305 }
3306 
3307 static ssize_t fsg_opts_num_buffers_store(struct config_item *item,
3308 					  const char *page, size_t len)
3309 {
3310 	struct fsg_opts *opts = to_fsg_opts(item);
3311 	int ret;
3312 	u8 num;
3313 
3314 	mutex_lock(&opts->lock);
3315 	if (opts->refcnt) {
3316 		ret = -EBUSY;
3317 		goto end;
3318 	}
3319 	ret = kstrtou8(page, 0, &num);
3320 	if (ret)
3321 		goto end;
3322 
3323 	fsg_common_set_num_buffers(opts->common, num);
3324 	ret = len;
3325 
3326 end:
3327 	mutex_unlock(&opts->lock);
3328 	return ret;
3329 }
3330 
3331 CONFIGFS_ATTR(fsg_opts_, num_buffers);
3332 #endif
3333 
3334 static struct configfs_attribute *fsg_attrs[] = {
3335 	&fsg_opts_attr_stall,
3336 #ifdef CONFIG_USB_GADGET_DEBUG_FILES
3337 	&fsg_opts_attr_num_buffers,
3338 #endif
3339 	NULL,
3340 };
3341 
3342 static struct configfs_group_operations fsg_group_ops = {
3343 	.make_group	= fsg_lun_make,
3344 	.drop_item	= fsg_lun_drop,
3345 };
3346 
3347 static struct config_item_type fsg_func_type = {
3348 	.ct_item_ops	= &fsg_item_ops,
3349 	.ct_group_ops	= &fsg_group_ops,
3350 	.ct_attrs	= fsg_attrs,
3351 	.ct_owner	= THIS_MODULE,
3352 };
3353 
3354 static void fsg_free_inst(struct usb_function_instance *fi)
3355 {
3356 	struct fsg_opts *opts;
3357 
3358 	opts = fsg_opts_from_func_inst(fi);
3359 	fsg_common_put(opts->common);
3360 	kfree(opts);
3361 }
3362 
3363 static struct usb_function_instance *fsg_alloc_inst(void)
3364 {
3365 	struct fsg_opts *opts;
3366 	struct fsg_lun_config config;
3367 	int rc;
3368 
3369 	opts = kzalloc(sizeof(*opts), GFP_KERNEL);
3370 	if (!opts)
3371 		return ERR_PTR(-ENOMEM);
3372 	mutex_init(&opts->lock);
3373 	opts->func_inst.free_func_inst = fsg_free_inst;
3374 	opts->common = fsg_common_setup(opts->common);
3375 	if (IS_ERR(opts->common)) {
3376 		rc = PTR_ERR(opts->common);
3377 		goto release_opts;
3378 	}
3379 
3380 	rc = fsg_common_set_num_buffers(opts->common,
3381 					CONFIG_USB_GADGET_STORAGE_NUM_BUFFERS);
3382 	if (rc)
3383 		goto release_opts;
3384 
3385 	pr_info(FSG_DRIVER_DESC ", version: " FSG_DRIVER_VERSION "\n");
3386 
3387 	memset(&config, 0, sizeof(config));
3388 	config.removable = true;
3389 	rc = fsg_common_create_lun(opts->common, &config, 0, "lun.0",
3390 			(const char **)&opts->func_inst.group.cg_item.ci_name);
3391 	if (rc)
3392 		goto release_buffers;
3393 
3394 	opts->lun0.lun = opts->common->luns[0];
3395 	opts->lun0.lun_id = 0;
3396 
3397 	config_group_init_type_name(&opts->func_inst.group, "", &fsg_func_type);
3398 
3399 	config_group_init_type_name(&opts->lun0.group, "lun.0", &fsg_lun_type);
3400 	configfs_add_default_group(&opts->lun0.group, &opts->func_inst.group);
3401 
3402 	return &opts->func_inst;
3403 
3404 release_buffers:
3405 	fsg_common_free_buffers(opts->common);
3406 release_opts:
3407 	kfree(opts);
3408 	return ERR_PTR(rc);
3409 }
3410 
3411 static void fsg_free(struct usb_function *f)
3412 {
3413 	struct fsg_dev *fsg;
3414 	struct fsg_opts *opts;
3415 
3416 	fsg = container_of(f, struct fsg_dev, function);
3417 	opts = container_of(f->fi, struct fsg_opts, func_inst);
3418 
3419 	mutex_lock(&opts->lock);
3420 	opts->refcnt--;
3421 	mutex_unlock(&opts->lock);
3422 
3423 	kfree(fsg);
3424 }
3425 
3426 static struct usb_function *fsg_alloc(struct usb_function_instance *fi)
3427 {
3428 	struct fsg_opts *opts = fsg_opts_from_func_inst(fi);
3429 	struct fsg_common *common = opts->common;
3430 	struct fsg_dev *fsg;
3431 
3432 	fsg = kzalloc(sizeof(*fsg), GFP_KERNEL);
3433 	if (unlikely(!fsg))
3434 		return ERR_PTR(-ENOMEM);
3435 
3436 	mutex_lock(&opts->lock);
3437 	opts->refcnt++;
3438 	mutex_unlock(&opts->lock);
3439 
3440 	fsg->function.name	= FSG_DRIVER_DESC;
3441 	fsg->function.bind	= fsg_bind;
3442 	fsg->function.unbind	= fsg_unbind;
3443 	fsg->function.setup	= fsg_setup;
3444 	fsg->function.set_alt	= fsg_set_alt;
3445 	fsg->function.disable	= fsg_disable;
3446 	fsg->function.free_func	= fsg_free;
3447 
3448 	fsg->common               = common;
3449 
3450 	return &fsg->function;
3451 }
3452 
3453 DECLARE_USB_FUNCTION_INIT(mass_storage, fsg_alloc_inst, fsg_alloc);
3454 MODULE_LICENSE("GPL");
3455 MODULE_AUTHOR("Michal Nazarewicz");
3456 
3457 /************************* Module parameters *************************/
3458 
3459 
3460 void fsg_config_from_params(struct fsg_config *cfg,
3461 		       const struct fsg_module_parameters *params,
3462 		       unsigned int fsg_num_buffers)
3463 {
3464 	struct fsg_lun_config *lun;
3465 	unsigned i;
3466 
3467 	/* Configure LUNs */
3468 	cfg->nluns =
3469 		min(params->luns ?: (params->file_count ?: 1u),
3470 		    (unsigned)FSG_MAX_LUNS);
3471 	for (i = 0, lun = cfg->luns; i < cfg->nluns; ++i, ++lun) {
3472 		lun->ro = !!params->ro[i];
3473 		lun->cdrom = !!params->cdrom[i];
3474 		lun->removable = !!params->removable[i];
3475 		lun->filename =
3476 			params->file_count > i && params->file[i][0]
3477 			? params->file[i]
3478 			: NULL;
3479 	}
3480 
3481 	/* Let MSF use defaults */
3482 	cfg->vendor_name = NULL;
3483 	cfg->product_name = NULL;
3484 
3485 	cfg->ops = NULL;
3486 	cfg->private_data = NULL;
3487 
3488 	/* Finalise */
3489 	cfg->can_stall = params->stall;
3490 	cfg->fsg_num_buffers = fsg_num_buffers;
3491 }
3492 EXPORT_SYMBOL_GPL(fsg_config_from_params);
3493