f_mass_storage.c 89.3 KB
Newer Older
1
/*
2
 * f_mass_storage.c -- Mass Storage USB Composite Function
3 4
 *
 * Copyright (C) 2003-2008 Alan Stern
5 6
 * Copyright (C) 2009 Samsung Electronics
 *                    Author: Michal Nazarewicz <m.nazarewicz@samsung.com>
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions, and the following disclaimer,
 *    without modification.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. The names of the above-listed copyright holders may not be used
 *    to endorse or promote products derived from this software without
 *    specific prior written permission.
 *
 * ALTERNATIVELY, this software may be distributed under the terms of the
 * GNU General Public License ("GPL") as published by the Free Software
 * Foundation, either version 2 of that License or (at your option) any
 * later version.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
 * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/*
41 42 43 44 45
 * The Mass Storage Function acts as a USB Mass Storage device,
 * appearing to the host as a disk drive or as a CD-ROM drive.  In
 * addition to providing an example of a genuinely useful composite
 * function for a USB device, it also illustrates a technique of
 * double-buffering for increased throughput.
46
 *
47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
 * Function supports multiple logical units (LUNs).  Backing storage
 * for each LUN is provided by a regular file or a block device.
 * Access for each LUN can be limited to read-only.  Moreover, the
 * function can indicate that LUN is removable and/or CD-ROM.  (The
 * later implies read-only access.)
 *
 * MSF is configured by specifying a fsg_config structure.  It has the
 * following fields:
 *
 *	nluns		Number of LUNs function have (anywhere from 1
 *				to FSG_MAX_LUNS which is 8).
 *	luns		An array of LUN configuration values.  This
 *				should be filled for each LUN that
 *				function will include (ie. for "nluns"
 *				LUNs).  Each element of the array has
 *				the following fields:
 *	->filename	The path to the backing file for the LUN.
 *				Required if LUN is not marked as
 *				removable.
 *	->ro		Flag specifying access to the LUN shall be
 *				read-only.  This is implied if CD-ROM
 *				emulation is enabled as well as when
 *				it was impossible to open "filename"
 *				in R/W mode.
 *	->removable	Flag specifying that LUN shall be indicated as
 *				being removable.
 *	->cdrom		Flag specifying that LUN shall be reported as
 *				being a CD-ROM.
75 76
 *	->nofua		Flag specifying that FUA flag in SCSI WRITE(10,12)
 *				commands for this LUN shall be ignored.
77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130
 *
 *	lun_name_format	A printf-like format for names of the LUN
 *				devices.  This determines how the
 *				directory in sysfs will be named.
 *				Unless you are using several MSFs in
 *				a single gadget (as opposed to single
 *				MSF in many configurations) you may
 *				leave it as NULL (in which case
 *				"lun%d" will be used).  In the format
 *				you can use "%d" to index LUNs for
 *				MSF's with more than one LUN.  (Beware
 *				that there is only one integer given
 *				as an argument for the format and
 *				specifying invalid format may cause
 *				unspecified behaviour.)
 *	thread_name	Name of the kernel thread process used by the
 *				MSF.  You can safely set it to NULL
 *				(in which case default "file-storage"
 *				will be used).
 *
 *	vendor_name
 *	product_name
 *	release		Information used as a reply to INQUIRY
 *				request.  To use default set to NULL,
 *				NULL, 0xffff respectively.  The first
 *				field should be 8 and the second 16
 *				characters or less.
 *
 *	can_stall	Set to permit function to halt bulk endpoints.
 *				Disabled on some USB devices known not
 *				to work correctly.  You should set it
 *				to true.
 *
 * If "removable" is not set for a LUN then a backing file must be
 * specified.  If it is set, then NULL filename means the LUN's medium
 * is not loaded (an empty string as "filename" in the fsg_config
 * structure causes error).  The CD-ROM emulation includes a single
 * data track and no audio tracks; hence there need be only one
 * backing file per LUN.  Note also that the CD-ROM block length is
 * set to 512 rather than the more common value 2048.
 *
 *
 * MSF includes support for module parameters.  If gadget using it
 * decides to use it, the following module parameters will be
 * available:
 *
 *	file=filename[,filename...]
 *			Names of the files or block devices used for
 *				backing storage.
 *	ro=b[,b...]	Default false, boolean for read-only access.
 *	removable=b[,b...]
 *			Default true, boolean for removable media.
 *	cdrom=b[,b...]	Default false, boolean for whether to emulate
 *				a CD-ROM drive.
131 132
 *	nofua=b[,b...]	Default false, booleans for ignore FUA flag
 *				in SCSI WRITE(10,12) commands
133 134 135 136 137 138 139 140 141 142 143 144
 *	luns=N		Default N = number of filenames, number of
 *				LUNs to support.
 *	stall		Default determined according to the type of
 *				USB device controller (usually true),
 *				boolean to permit the driver to halt
 *				bulk endpoints.
 *
 * The module parameters may be prefixed with some string.  You need
 * to consult gadget's documentation or source to verify whether it is
 * using those module parameters and if it does what are the prefixes
 * (look for FSG_MODULE_PARAMETERS() macro usage, what's inside it is
 * the prefix).
145 146 147
 *
 *
 * Requirements are modest; only a bulk-in and a bulk-out endpoint are
148 149 150
 * needed.  The memory requirement amounts to two 16K buffers, size
 * configurable by a parameter.  Support is included for both
 * full-speed and high-speed operation.
151 152 153 154 155 156 157 158
 *
 * Note that the driver is slightly non-portable in that it assumes a
 * single memory/DMA buffer will be useable for bulk-in, bulk-out, and
 * interrupt-in endpoints.  With most device controllers this isn't an
 * issue, but there may be some with hardware restrictions that prevent
 * a buffer from being used by more than one endpoint.
 *
 *
159 160 161 162 163 164 165 166 167
 * The pathnames of the backing files and the ro settings are
 * available in the attribute files "file" and "ro" in the lun<n> (or
 * to be more precise in a directory which name comes from
 * "lun_name_format" option!) subdirectory of the gadget's sysfs
 * directory.  If the "removable" option is set, writing to these
 * files will simulate ejecting/loading the medium (writing an empty
 * line means eject) and adjusting a write-enable tab.  Changes to the
 * ro setting are not allowed when the medium is loaded or if CD-ROM
 * emulation is being used.
168
 *
169 170 171 172
 * When a LUN receive an "eject" SCSI request (Start/Stop Unit),
 * if the LUN is removable, the backing file is released to simulate
 * ejection.
 *
173
 *
174 175 176 177 178 179 180 181 182 183
 * This function is heavily based on "File-backed Storage Gadget" by
 * Alan Stern which in turn is heavily based on "Gadget Zero" by David
 * Brownell.  The driver's SCSI command interface was based on the
 * "Information technology - Small Computer System Interface - 2"
 * document from X3T9.2 Project 375D, Revision 10L, 7-SEP-93,
 * available at <http://www.t10.org/ftp/t10/drafts/s2/s2-r10l.pdf>.
 * The single exception is opcode 0x23 (READ FORMAT CAPACITIES), which
 * was based on the "Universal Serial Bus Mass Storage Class UFI
 * Command Specification" document, Revision 1.0, December 14, 1998,
 * available at
184 185 186 187 188 189
 * <http://www.usb.org/developers/devclass_docs/usbmass-ufi10.pdf>.
 */

/*
 *				Driver Design
 *
190
 * The MSF is fairly straightforward.  There is a main kernel
191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212
 * thread that handles most of the work.  Interrupt routines field
 * callbacks from the controller driver: bulk- and interrupt-request
 * completion notifications, endpoint-0 events, and disconnect events.
 * Completion events are passed to the main thread by wakeup calls.  Many
 * ep0 requests are handled at interrupt time, but SetInterface,
 * SetConfiguration, and device reset requests are forwarded to the
 * thread in the form of "exceptions" using SIGUSR1 signals (since they
 * should interrupt any ongoing file I/O operations).
 *
 * The thread's main routine implements the standard command/data/status
 * parts of a SCSI interaction.  It and its subroutines are full of tests
 * for pending signals/exceptions -- all this polling is necessary since
 * the kernel has no setjmp/longjmp equivalents.  (Maybe this is an
 * indication that the driver really wants to be running in userspace.)
 * An important point is that so long as the thread is alive it keeps an
 * open reference to the backing file.  This will prevent unmounting
 * the backing file's underlying filesystem and could cause problems
 * during system shutdown, for example.  To prevent such problems, the
 * thread catches INT, TERM, and KILL signals and converts them into
 * an EXIT exception.
 *
 * In normal operation the main thread is started during the gadget's
213 214 215 216 217
 * fsg_bind() callback and stopped during fsg_unbind().  But it can
 * also exit when it receives a signal, and there's no point leaving
 * the gadget running when the thread is dead.  At of this moment, MSF
 * provides no way to deregister the gadget when thread dies -- maybe
 * a callback functions is needed.
218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295
 *
 * To provide maximum throughput, the driver uses a circular pipeline of
 * buffer heads (struct fsg_buffhd).  In principle the pipeline can be
 * arbitrarily long; in practice the benefits don't justify having more
 * than 2 stages (i.e., double buffering).  But it helps to think of the
 * pipeline as being a long one.  Each buffer head contains a bulk-in and
 * a bulk-out request pointer (since the buffer can be used for both
 * output and input -- directions always are given from the host's
 * point of view) as well as a pointer to the buffer and various state
 * variables.
 *
 * Use of the pipeline follows a simple protocol.  There is a variable
 * (fsg->next_buffhd_to_fill) that points to the next buffer head to use.
 * At any time that buffer head may still be in use from an earlier
 * request, so each buffer head has a state variable indicating whether
 * it is EMPTY, FULL, or BUSY.  Typical use involves waiting for the
 * buffer head to be EMPTY, filling the buffer either by file I/O or by
 * USB I/O (during which the buffer head is BUSY), and marking the buffer
 * head FULL when the I/O is complete.  Then the buffer will be emptied
 * (again possibly by USB I/O, during which it is marked BUSY) and
 * finally marked EMPTY again (possibly by a completion routine).
 *
 * A module parameter tells the driver to avoid stalling the bulk
 * endpoints wherever the transport specification allows.  This is
 * necessary for some UDCs like the SuperH, which cannot reliably clear a
 * halt on a bulk endpoint.  However, under certain circumstances the
 * Bulk-only specification requires a stall.  In such cases the driver
 * will halt the endpoint and set a flag indicating that it should clear
 * the halt in software during the next device reset.  Hopefully this
 * will permit everything to work correctly.  Furthermore, although the
 * specification allows the bulk-out endpoint to halt when the host sends
 * too much data, implementing this would cause an unavoidable race.
 * The driver will always use the "no-stall" approach for OUT transfers.
 *
 * One subtle point concerns sending status-stage responses for ep0
 * requests.  Some of these requests, such as device reset, can involve
 * interrupting an ongoing file I/O operation, which might take an
 * arbitrarily long time.  During that delay the host might give up on
 * the original ep0 request and issue a new one.  When that happens the
 * driver should not notify the host about completion of the original
 * request, as the host will no longer be waiting for it.  So the driver
 * assigns to each ep0 request a unique tag, and it keeps track of the
 * tag value of the request associated with a long-running exception
 * (device-reset, interface-change, or configuration-change).  When the
 * exception handler is finished, the status-stage response is submitted
 * only if the current ep0 request tag is equal to the exception request
 * tag.  Thus only the most recently received ep0 request will get a
 * status-stage response.
 *
 * Warning: This driver source file is too long.  It ought to be split up
 * into a header file plus about 3 separate .c files, to handle the details
 * of the Gadget, USB Mass Storage, and SCSI protocols.
 */


/* #define VERBOSE_DEBUG */
/* #define DUMP_MSGS */

#include <linux/blkdev.h>
#include <linux/completion.h>
#include <linux/dcache.h>
#include <linux/delay.h>
#include <linux/device.h>
#include <linux/fcntl.h>
#include <linux/file.h>
#include <linux/fs.h>
#include <linux/kref.h>
#include <linux/kthread.h>
#include <linux/limits.h>
#include <linux/rwsem.h>
#include <linux/slab.h>
#include <linux/spinlock.h>
#include <linux/string.h>
#include <linux/freezer.h>
#include <linux/utsname.h>

#include <linux/usb/ch9.h>
#include <linux/usb/gadget.h>
296
#include <linux/usb/composite.h>
297 298 299 300

#include "gadget_chips.h"


301
/*------------------------------------------------------------------------*/
302

303
#define FSG_DRIVER_DESC		"Mass Storage Function"
304
#define FSG_DRIVER_VERSION	"2009/09/11"
305 306 307

static const char fsg_string_interface[] = "Mass Storage";

308
#define FSG_NO_INTR_EP 1
309 310 311
#define FSG_NO_DEVICE_STRINGS    1
#define FSG_NO_OTG               1
#define FSG_NO_INTR_EP           1
312

313 314 315 316 317
#include "storage_common.c"


/*-------------------------------------------------------------------------*/

318
struct fsg_dev;
319 320 321 322
struct fsg_common;

/* FSF callback functions */
struct fsg_operations {
323 324
	/*
	 * Callback function to call when thread exits.  If no
325 326 327
	 * callback is set or it returns value lower then zero MSF
	 * will force eject all LUNs it operates on (including those
	 * marked as non-removable or with prevent_medium_removal flag
328 329
	 * set).
	 */
330 331
	int (*thread_exits)(struct fsg_common *common);

332 333
	/*
	 * Called prior to ejection.  Negative return means error,
334
	 * zero means to continue with ejection, positive means not to
335 336
	 * eject.
	 */
337 338
	int (*pre_eject)(struct fsg_common *common,
			 struct fsg_lun *lun, int num);
339 340 341 342
	/*
	 * Called after ejection.  Negative return means error, zero
	 * or positive is just a success.
	 */
343 344 345
	int (*post_eject)(struct fsg_common *common,
			  struct fsg_lun *lun, int num);
};
346

347 348
/* Data shared by all the FSG instances. */
struct fsg_common {
349
	struct usb_gadget	*gadget;
350 351
	struct fsg_dev		*fsg, *new_fsg;
	wait_queue_head_t	fsg_wait;
352

353 354 355
	/* filesem protects: backing files in use */
	struct rw_semaphore	filesem;

356 357 358 359 360 361 362
	/* lock protects: state, all the req_busy's */
	spinlock_t		lock;

	struct usb_ep		*ep0;		/* Copy of gadget->ep0 */
	struct usb_request	*ep0req;	/* Copy of cdev->req */
	unsigned int		ep0_req_tag;

363 364 365 366 367 368 369 370 371 372 373
	struct fsg_buffhd	*next_buffhd_to_fill;
	struct fsg_buffhd	*next_buffhd_to_drain;
	struct fsg_buffhd	buffhds[FSG_NUM_BUFFERS];

	int			cmnd_size;
	u8			cmnd[MAX_COMMAND_SIZE];

	unsigned int		nluns;
	unsigned int		lun;
	struct fsg_lun		*luns;
	struct fsg_lun		*curlun;
374

375 376 377 378 379 380 381 382 383 384 385
	unsigned int		bulk_out_maxpacket;
	enum fsg_state		state;		/* For exception handling */
	unsigned int		exception_req_tag;

	enum data_direction	data_dir;
	u32			data_size;
	u32			data_size_from_cmnd;
	u32			tag;
	u32			residue;
	u32			usb_amount_left;

386
	unsigned int		can_stall:1;
387
	unsigned int		free_storage_on_release:1;
388 389 390 391
	unsigned int		phase_error:1;
	unsigned int		short_packet_received:1;
	unsigned int		bad_lun_okay:1;
	unsigned int		running:1;
392

393 394 395
	int			thread_wakeup_needed;
	struct completion	thread_notifier;
	struct task_struct	*thread_task;
396

397 398
	/* Callback functions. */
	const struct fsg_operations	*ops;
399 400 401
	/* Gadget's private data. */
	void			*private_data;

402 403 404 405
	/*
	 * Vendor (8 chars), product (16 chars), release (4
	 * hexadecimal digits) and NUL byte
	 */
406 407
	char inquiry_string[8 + 16 + 4 + 1];

408
	struct kref		ref;
409 410
};

411 412 413 414 415 416 417
struct fsg_config {
	unsigned nluns;
	struct fsg_lun_config {
		const char *filename;
		char ro;
		char removable;
		char cdrom;
418
		char nofua;
419 420
	} luns[FSG_MAX_LUNS];

421 422 423
	const char		*lun_name_format;
	const char		*thread_name;

424 425
	/* Callback functions. */
	const struct fsg_operations	*ops;
426 427 428
	/* Gadget's private data. */
	void			*private_data;

429 430 431 432 433 434 435
	const char *vendor_name;		/*  8 characters or less */
	const char *product_name;		/* 16 characters or less */
	u16 release;

	char			can_stall;
};

436
struct fsg_dev {
437 438
	struct usb_function	function;
	struct usb_gadget	*gadget;	/* Copy of cdev->gadget */
439 440
	struct fsg_common	*common;

441 442
	u16			interface_number;

443 444
	unsigned int		bulk_in_enabled:1;
	unsigned int		bulk_out_enabled:1;
445 446

	unsigned long		atomic_bitflags;
447
#define IGNORE_BULK_OUT		0
448 449 450

	struct usb_ep		*bulk_in;
	struct usb_ep		*bulk_out;
451
};
452

453 454 455 456 457 458
static inline int __fsg_is_set(struct fsg_common *common,
			       const char *func, unsigned line)
{
	if (common->fsg)
		return 1;
	ERROR(common, "common->fsg is NULL in %s at %u\n", func, line);
459
	WARN_ON(1);
460 461 462 463
	return 0;
}

#define fsg_is_set(common) likely(__fsg_is_set(common, __func__, __LINE__))
464

465 466 467 468 469
static inline struct fsg_dev *fsg_from_func(struct usb_function *f)
{
	return container_of(f, struct fsg_dev, function);
}

470 471
typedef void (*fsg_routine_t)(struct fsg_dev *);

472
static int exception_in_progress(struct fsg_common *common)
473
{
474
	return common->state > FSG_STATE_IDLE;
475 476 477
}

/* Make bulk-out requests be divisible by the maxpacket size */
478
static void set_bulk_out_req_length(struct fsg_common *common,
479
				    struct fsg_buffhd *bh, unsigned int length)
480 481 482 483
{
	unsigned int	rem;

	bh->bulk_out_intended_length = length;
484
	rem = length % common->bulk_out_maxpacket;
485
	if (rem > 0)
486
		length += common->bulk_out_maxpacket - rem;
487 488 489
	bh->outreq->length = length;
}

490

491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512
/*-------------------------------------------------------------------------*/

static int fsg_set_halt(struct fsg_dev *fsg, struct usb_ep *ep)
{
	const char	*name;

	if (ep == fsg->bulk_in)
		name = "bulk-in";
	else if (ep == fsg->bulk_out)
		name = "bulk-out";
	else
		name = ep->name;
	DBG(fsg, "%s set halt\n", name);
	return usb_ep_set_halt(ep);
}


/*-------------------------------------------------------------------------*/

/* These routines may be called in process context or in_irq */

/* Caller must hold fsg->lock */
513
static void wakeup_thread(struct fsg_common *common)
514 515
{
	/* Tell the main thread that something has happened */
516 517 518
	common->thread_wakeup_needed = 1;
	if (common->thread_task)
		wake_up_process(common->thread_task);
519 520
}

521
static void raise_exception(struct fsg_common *common, enum fsg_state new_state)
522 523 524
{
	unsigned long		flags;

525 526
	/*
	 * Do nothing if a higher-priority exception is already in progress.
527
	 * If a lower-or-equal priority exception is in progress, preempt it
528 529
	 * and notify the main thread by sending it a signal.
	 */
530 531 532 533 534
	spin_lock_irqsave(&common->lock, flags);
	if (common->state <= new_state) {
		common->exception_req_tag = common->ep0_req_tag;
		common->state = new_state;
		if (common->thread_task)
535
			send_sig_info(SIGUSR1, SEND_SIG_FORCED,
536
				      common->thread_task);
537
	}
538
	spin_unlock_irqrestore(&common->lock, flags);
539 540 541 542 543
}


/*-------------------------------------------------------------------------*/

544
static int ep0_queue(struct fsg_common *common)
545 546 547
{
	int	rc;

548 549
	rc = usb_ep_queue(common->ep0, common->ep0req, GFP_ATOMIC);
	common->ep0->driver_data = common;
550 551
	if (rc != 0 && rc != -ESHUTDOWN) {
		/* We can't do much more than wait for a reset */
552 553
		WARNING(common, "error in submission: %s --> %d\n",
			common->ep0->name, rc);
554 555 556 557
	}
	return rc;
}

558

559 560
/*-------------------------------------------------------------------------*/

561
/* Completion handlers. These always run in_irq. */
562 563 564

static void bulk_in_complete(struct usb_ep *ep, struct usb_request *req)
{
565
	struct fsg_common	*common = ep->driver_data;
566 567 568
	struct fsg_buffhd	*bh = req->context;

	if (req->status || req->actual != req->length)
569
		DBG(common, "%s --> %d, %u/%u\n", __func__,
570
		    req->status, req->actual, req->length);
571
	if (req->status == -ECONNRESET)		/* Request was cancelled */
572 573 574 575
		usb_ep_fifo_flush(ep);

	/* Hold the lock while we update the request and buffer states */
	smp_wmb();
576
	spin_lock(&common->lock);
577 578
	bh->inreq_busy = 0;
	bh->state = BUF_STATE_EMPTY;
579 580
	wakeup_thread(common);
	spin_unlock(&common->lock);
581 582 583 584
}

static void bulk_out_complete(struct usb_ep *ep, struct usb_request *req)
{
585
	struct fsg_common	*common = ep->driver_data;
586 587
	struct fsg_buffhd	*bh = req->context;

588
	dump_msg(common, "bulk-out", req->buf, req->actual);
589
	if (req->status || req->actual != bh->bulk_out_intended_length)
590
		DBG(common, "%s --> %d, %u/%u\n", __func__,
591
		    req->status, req->actual, bh->bulk_out_intended_length);
592
	if (req->status == -ECONNRESET)		/* Request was cancelled */
593 594 595 596
		usb_ep_fifo_flush(ep);

	/* Hold the lock while we update the request and buffer states */
	smp_wmb();
597
	spin_lock(&common->lock);
598 599
	bh->outreq_busy = 0;
	bh->state = BUF_STATE_FULL;
600 601
	wakeup_thread(common);
	spin_unlock(&common->lock);
602 603
}

604
static int fsg_setup(struct usb_function *f,
605
		     const struct usb_ctrlrequest *ctrl)
606
{
607
	struct fsg_dev		*fsg = fsg_from_func(f);
608
	struct usb_request	*req = fsg->common->ep0req;
609
	u16			w_index = le16_to_cpu(ctrl->wIndex);
610
	u16			w_value = le16_to_cpu(ctrl->wValue);
611 612
	u16			w_length = le16_to_cpu(ctrl->wLength);

613
	if (!fsg_is_set(fsg->common))
614
		return -EOPNOTSUPP;
615

616
	switch (ctrl->bRequest) {
617

618 619 620
	case USB_BULK_RESET_REQUEST:
		if (ctrl->bRequestType !=
		    (USB_DIR_OUT | USB_TYPE_CLASS | USB_RECIP_INTERFACE))
621
			break;
622
		if (w_index != fsg->interface_number || w_value != 0)
623
			return -EDOM;
624

625 626 627 628
		/*
		 * Raise an exception to stop the current operation
		 * and reinitialize our state.
		 */
629
		DBG(fsg, "bulk reset request\n");
630
		raise_exception(fsg->common, FSG_STATE_RESET);
631
		return DELAYED_STATUS;
632

633 634 635
	case USB_BULK_GET_MAX_LUN_REQUEST:
		if (ctrl->bRequestType !=
		    (USB_DIR_IN | USB_TYPE_CLASS | USB_RECIP_INTERFACE))
636
			break;
637
		if (w_index != fsg->interface_number || w_value != 0)
638 639
			return -EDOM;
		VDBG(fsg, "get max LUN\n");
640
		*(u8 *)req->buf = fsg->common->nluns - 1;
641 642

		/* Respond with data/status */
643
		req->length = min((u16)1, w_length);
644
		return ep0_queue(fsg->common);
645 646 647
	}

	VDBG(fsg,
648
	     "unknown class-specific control req %02x.%02x v%04x i%04x l%u\n",
649 650 651
	     ctrl->bRequestType, ctrl->bRequest,
	     le16_to_cpu(ctrl->wValue), w_index, w_length);
	return -EOPNOTSUPP;
652 653 654 655 656 657 658 659 660
}


/*-------------------------------------------------------------------------*/

/* All the following routines run in process context */

/* Use this for bulk or interrupt transfers, not ep0 */
static void start_transfer(struct fsg_dev *fsg, struct usb_ep *ep,
661 662
			   struct usb_request *req, int *pbusy,
			   enum fsg_buffer_state *state)
663 664 665 666 667 668
{
	int	rc;

	if (ep == fsg->bulk_in)
		dump_msg(fsg, "bulk-in", req->buf, req->length);

669
	spin_lock_irq(&fsg->common->lock);
670 671
	*pbusy = 1;
	*state = BUF_STATE_BUSY;
672
	spin_unlock_irq(&fsg->common->lock);
673 674 675 676 677 678 679
	rc = usb_ep_queue(ep, req, GFP_KERNEL);
	if (rc != 0) {
		*pbusy = 0;
		*state = BUF_STATE_EMPTY;

		/* We can't do much more than wait for a reset */

680 681 682 683 684 685
		/*
		 * Note: currently the net2280 driver fails zero-length
		 * submissions if DMA is enabled.
		 */
		if (rc != -ESHUTDOWN &&
		    !(rc == -EOPNOTSUPP && req->length == 0))
686
			WARNING(fsg, "error in submission: %s --> %d\n",
687
				ep->name, rc);
688 689 690
	}
}

691 692 693 694 695 696 697 698
static bool start_in_transfer(struct fsg_common *common, struct fsg_buffhd *bh)
{
	if (!fsg_is_set(common))
		return false;
	start_transfer(common->fsg, common->fsg->bulk_in,
		       bh->inreq, &bh->inreq_busy, &bh->state);
	return true;
}
699

700 701 702 703 704 705 706 707
static bool start_out_transfer(struct fsg_common *common, struct fsg_buffhd *bh)
{
	if (!fsg_is_set(common))
		return false;
	start_transfer(common->fsg, common->fsg->bulk_out,
		       bh->outreq, &bh->outreq_busy, &bh->state);
	return true;
}
708

709
static int sleep_thread(struct fsg_common *common)
710 711 712 713 714 715 716 717 718 719 720
{
	int	rc = 0;

	/* Wait until a signal arrives or we are woken up */
	for (;;) {
		try_to_freeze();
		set_current_state(TASK_INTERRUPTIBLE);
		if (signal_pending(current)) {
			rc = -EINTR;
			break;
		}
721
		if (common->thread_wakeup_needed)
722 723 724 725
			break;
		schedule();
	}
	__set_current_state(TASK_RUNNING);
726
	common->thread_wakeup_needed = 0;
727 728 729 730 731 732
	return rc;
}


/*-------------------------------------------------------------------------*/

733
static int do_read(struct fsg_common *common)
734
{
735
	struct fsg_lun		*curlun = common->curlun;
736 737 738 739 740 741 742 743 744
	u32			lba;
	struct fsg_buffhd	*bh;
	int			rc;
	u32			amount_left;
	loff_t			file_offset, file_offset_tmp;
	unsigned int		amount;
	unsigned int		partial_page;
	ssize_t			nread;

745 746 747 748
	/*
	 * Get the starting Logical Block Address and check that it's
	 * not too big.
	 */
749
	if (common->cmnd[0] == READ_6)
750
		lba = get_unaligned_be24(&common->cmnd[1]);
751
	else {
752
		lba = get_unaligned_be32(&common->cmnd[2]);
753

754 755
		/*
		 * We allow DPO (Disable Page Out = don't save data in the
756
		 * cache) and FUA (Force Unit Access = don't read from the
757 758
		 * cache), but we don't implement them.
		 */
759
		if ((common->cmnd[1] & ~0x18) != 0) {
760 761 762 763 764 765 766 767 768 769 770
			curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
			return -EINVAL;
		}
	}
	if (lba >= curlun->num_sectors) {
		curlun->sense_data = SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE;
		return -EINVAL;
	}
	file_offset = ((loff_t) lba) << 9;

	/* Carry out the file reads */
771
	amount_left = common->data_size_from_cmnd;
772
	if (unlikely(amount_left == 0))
773
		return -EIO;		/* No default reply */
774 775

	for (;;) {
776 777
		/*
		 * Figure out how much we need to read:
778 779 780 781 782 783
		 * Try to read the remaining amount.
		 * But don't read more than the buffer size.
		 * And don't try to read past the end of the file.
		 * Finally, if we're not at a page boundary, don't read past
		 *	the next page.
		 * If this means reading 0 then we were asked to read past
784 785
		 *	the end of file.
		 */
786
		amount = min(amount_left, FSG_BUFLEN);
787 788
		amount = min((loff_t)amount,
			     curlun->file_length - file_offset);
789 790
		partial_page = file_offset & (PAGE_CACHE_SIZE - 1);
		if (partial_page > 0)
791 792
			amount = min(amount, (unsigned int)PAGE_CACHE_SIZE -
					     partial_page);
793 794

		/* Wait for the next buffer to become available */
795
		bh = common->next_buffhd_to_fill;
796
		while (bh->state != BUF_STATE_EMPTY) {
797
			rc = sleep_thread(common);
798 799 800 801
			if (rc)
				return rc;
		}

802 803 804 805
		/*
		 * If we were asked to read past the end of file,
		 * end with an empty buffer.
		 */
806 807 808 809 810 811 812 813 814 815 816 817 818
		if (amount == 0) {
			curlun->sense_data =
					SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE;
			curlun->sense_data_info = file_offset >> 9;
			curlun->info_valid = 1;
			bh->inreq->length = 0;
			bh->state = BUF_STATE_FULL;
			break;
		}

		/* Perform the read */
		file_offset_tmp = file_offset;
		nread = vfs_read(curlun->filp,
819 820
				 (char __user *)bh->buf,
				 amount, &file_offset_tmp);
821
		VLDBG(curlun, "file read %u @ %llu -> %d\n", amount,
822
		      (unsigned long long)file_offset, (int)nread);
823 824 825 826
		if (signal_pending(current))
			return -EINTR;

		if (nread < 0) {
827
			LDBG(curlun, "error in file read: %d\n", (int)nread);
828 829 830
			nread = 0;
		} else if (nread < amount) {
			LDBG(curlun, "partial file read: %d/%u\n",
831
			     (int)nread, amount);
832
			nread -= (nread & 511);	/* Round down to a block */
833 834 835
		}
		file_offset  += nread;
		amount_left  -= nread;
836
		common->residue -= nread;
837 838 839 840 841 842 843 844 845 846 847 848
		bh->inreq->length = nread;
		bh->state = BUF_STATE_FULL;

		/* If an error occurred, report it and its position */
		if (nread < amount) {
			curlun->sense_data = SS_UNRECOVERED_READ_ERROR;
			curlun->sense_data_info = file_offset >> 9;
			curlun->info_valid = 1;
			break;
		}

		if (amount_left == 0)
849
			break;		/* No more left to read */
850 851 852

		/* Send this buffer and go read some more */
		bh->inreq->zero = 0;
853 854
		if (!start_in_transfer(common, bh))
			/* Don't know what to do if common->fsg is NULL */
855 856
			return -EIO;
		common->next_buffhd_to_fill = bh->next;
857 858
	}

859
	return -EIO;		/* No default reply */
860 861 862 863 864
}


/*-------------------------------------------------------------------------*/

865
static int do_write(struct fsg_common *common)
866
{
867
	struct fsg_lun		*curlun = common->curlun;
868 869 870 871 872 873 874 875 876 877 878 879 880 881 882
	u32			lba;
	struct fsg_buffhd	*bh;
	int			get_some_more;
	u32			amount_left_to_req, amount_left_to_write;
	loff_t			usb_offset, file_offset, file_offset_tmp;
	unsigned int		amount;
	unsigned int		partial_page;
	ssize_t			nwritten;
	int			rc;

	if (curlun->ro) {
		curlun->sense_data = SS_WRITE_PROTECTED;
		return -EINVAL;
	}
	spin_lock(&curlun->filp->f_lock);
883
	curlun->filp->f_flags &= ~O_SYNC;	/* Default is not to wait */
884 885
	spin_unlock(&curlun->filp->f_lock);

886 887 888 889
	/*
	 * Get the starting Logical Block Address and check that it's
	 * not too big
	 */
890
	if (common->cmnd[0] == WRITE_6)
891
		lba = get_unaligned_be24(&common->cmnd[1]);
892
	else {
893
		lba = get_unaligned_be32(&common->cmnd[2]);
894

895 896
		/*
		 * We allow DPO (Disable Page Out = don't save data in the
897 898
		 * cache) and FUA (Force Unit Access = write directly to the
		 * medium).  We don't implement DPO; we implement FUA by
899 900
		 * performing synchronous output.
		 */
901
		if (common->cmnd[1] & ~0x18) {
902 903 904
			curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
			return -EINVAL;
		}
905
		if (!curlun->nofua && (common->cmnd[1] & 0x08)) { /* FUA */
906 907 908 909 910 911 912 913 914 915 916 917 918
			spin_lock(&curlun->filp->f_lock);
			curlun->filp->f_flags |= O_SYNC;
			spin_unlock(&curlun->filp->f_lock);
		}
	}
	if (lba >= curlun->num_sectors) {
		curlun->sense_data = SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE;
		return -EINVAL;
	}

	/* Carry out the file writes */
	get_some_more = 1;
	file_offset = usb_offset = ((loff_t) lba) << 9;
919 920
	amount_left_to_req = common->data_size_from_cmnd;
	amount_left_to_write = common->data_size_from_cmnd;
921 922 923 924

	while (amount_left_to_write > 0) {

		/* Queue a request for more data from the host */
925
		bh = common->next_buffhd_to_fill;
926 927
		if (bh->state == BUF_STATE_EMPTY && get_some_more) {

928 929
			/*
			 * Figure out how much we want to get:
930 931 932 933 934 935 936
			 * Try to get the remaining amount.
			 * But don't get more than the buffer size.
			 * And don't try to go past the end of the file.
			 * If we're not at a page boundary,
			 *	don't go past the next page.
			 * If this means getting 0, then we were asked
			 *	to write past the end of file.
937 938
			 * Finally, round down to a block boundary.
			 */
939
			amount = min(amount_left_to_req, FSG_BUFLEN);
940 941
			amount = min((loff_t)amount,
				     curlun->file_length - usb_offset);
942 943 944
			partial_page = usb_offset & (PAGE_CACHE_SIZE - 1);
			if (partial_page > 0)
				amount = min(amount,
945
	(unsigned int)PAGE_CACHE_SIZE - partial_page);
946 947 948 949 950 951 952 953 954

			if (amount == 0) {
				get_some_more = 0;
				curlun->sense_data =
					SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE;
				curlun->sense_data_info = usb_offset >> 9;
				curlun->info_valid = 1;
				continue;
			}
955
			amount -= amount & 511;
956 957
			if (amount == 0) {

958 959 960 961
				/*
				 * Why were we were asked to transfer a
				 * partial block?
				 */
962 963 964 965 966 967
				get_some_more = 0;
				continue;
			}

			/* Get the next buffer */
			usb_offset += amount;
968
			common->usb_amount_left -= amount;
969 970 971 972
			amount_left_to_req -= amount;
			if (amount_left_to_req == 0)
				get_some_more = 0;

973 974 975 976
			/*
			 * amount is always divisible by 512, hence by
			 * the bulk-out maxpacket size
			 */
977 978
			bh->outreq->length = amount;
			bh->bulk_out_intended_length = amount;
979
			bh->outreq->short_not_ok = 1;
980
			if (!start_out_transfer(common, bh))
981
				/* Dunno what to do if common->fsg is NULL */
982 983
				return -EIO;
			common->next_buffhd_to_fill = bh->next;
984 985 986 987
			continue;
		}

		/* Write the received data to the backing file */
988
		bh = common->next_buffhd_to_drain;
989
		if (bh->state == BUF_STATE_EMPTY && !get_some_more)
990
			break;			/* We stopped early */
991 992
		if (bh->state == BUF_STATE_FULL) {
			smp_rmb();
993
			common->next_buffhd_to_drain = bh->next;
994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006
			bh->state = BUF_STATE_EMPTY;

			/* Did something go wrong with the transfer? */
			if (bh->outreq->status != 0) {
				curlun->sense_data = SS_COMMUNICATION_FAILURE;
				curlun->sense_data_info = file_offset >> 9;
				curlun->info_valid = 1;
				break;
			}

			amount = bh->outreq->actual;
			if (curlun->file_length - file_offset < amount) {
				LERROR(curlun,
1007 1008 1009
				       "write %u @ %llu beyond end %llu\n",
				       amount, (unsigned long long)file_offset,
				       (unsigned long long)curlun->file_length);
1010 1011 1012 1013 1014 1015
				amount = curlun->file_length - file_offset;
			}

			/* Perform the write */
			file_offset_tmp = file_offset;
			nwritten = vfs_write(curlun->filp,
1016 1017
					     (char __user *)bh->buf,
					     amount, &file_offset_tmp);
1018
			VLDBG(curlun, "file write %u @ %llu -> %d\n", amount,
1019
			      (unsigned long long)file_offset, (int)nwritten);
1020
			if (signal_pending(current))
1021
				return -EINTR;		/* Interrupted! */
1022 1023 1024

			if (nwritten < 0) {
				LDBG(curlun, "error in file write: %d\n",
1025
				     (int)nwritten);
1026 1027 1028
				nwritten = 0;
			} else if (nwritten < amount) {
				LDBG(curlun, "partial file write: %d/%u\n",
1029
				     (int)nwritten, amount);
1030
				nwritten -= (nwritten & 511);
1031
				/* Round down to a block */
1032 1033 1034
			}
			file_offset += nwritten;
			amount_left_to_write -= nwritten;
1035
			common->residue -= nwritten;
1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046

			/* If an error occurred, report it and its position */
			if (nwritten < amount) {
				curlun->sense_data = SS_WRITE_ERROR;
				curlun->sense_data_info = file_offset >> 9;
				curlun->info_valid = 1;
				break;
			}

			/* Did the host decide to stop early? */
			if (bh->outreq->actual != bh->outreq->length) {
1047
				common->short_packet_received = 1;
1048 1049 1050 1051 1052 1053
				break;
			}
			continue;
		}

		/* Wait for something to happen */
1054
		rc = sleep_thread(common);
1055 1056 1057 1058
		if (rc)
			return rc;
	}

1059
	return -EIO;		/* No default reply */
1060 1061 1062 1063 1064
}


/*-------------------------------------------------------------------------*/

1065
static int do_synchronize_cache(struct fsg_common *common)
1066
{
1067
	struct fsg_lun	*curlun = common->curlun;
1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087
	int		rc;

	/* We ignore the requested LBA and write out all file's
	 * dirty data buffers. */
	rc = fsg_lun_fsync_sub(curlun);
	if (rc)
		curlun->sense_data = SS_WRITE_ERROR;
	return 0;
}


/*-------------------------------------------------------------------------*/

static void invalidate_sub(struct fsg_lun *curlun)
{
	struct file	*filp = curlun->filp;
	struct inode	*inode = filp->f_path.dentry->d_inode;
	unsigned long	rc;

	rc = invalidate_mapping_pages(inode->i_mapping, 0, -1);
1088
	VLDBG(curlun, "invalidate_mapping_pages -> %ld\n", rc);
1089 1090
}

1091
static int do_verify(struct fsg_common *common)
1092
{
1093
	struct fsg_lun		*curlun = common->curlun;
1094 1095
	u32			lba;
	u32			verification_length;
1096
	struct fsg_buffhd	*bh = common->next_buffhd_to_fill;
1097 1098 1099 1100 1101
	loff_t			file_offset, file_offset_tmp;
	u32			amount_left;
	unsigned int		amount;
	ssize_t			nread;

1102 1103 1104 1105
	/*
	 * Get the starting Logical Block Address and check that it's
	 * not too big.
	 */
1106
	lba = get_unaligned_be32(&common->cmnd[2]);
1107 1108 1109 1110 1111
	if (lba >= curlun->num_sectors) {
		curlun->sense_data = SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE;
		return -EINVAL;
	}

1112 1113 1114 1115
	/*
	 * We allow DPO (Disable Page Out = don't save data in the
	 * cache) but we don't implement it.
	 */
1116
	if (common->cmnd[1] & ~0x10) {
1117 1118 1119 1120
		curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
		return -EINVAL;
	}

1121
	verification_length = get_unaligned_be16(&common->cmnd[7]);
1122
	if (unlikely(verification_length == 0))
1123
		return -EIO;		/* No default reply */
1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139

	/* Prepare to carry out the file verify */
	amount_left = verification_length << 9;
	file_offset = ((loff_t) lba) << 9;

	/* Write out all the dirty buffers before invalidating them */
	fsg_lun_fsync_sub(curlun);
	if (signal_pending(current))
		return -EINTR;

	invalidate_sub(curlun);
	if (signal_pending(current))
		return -EINTR;

	/* Just try to read the requested blocks */
	while (amount_left > 0) {
1140 1141
		/*
		 * Figure out how much we need to read:
1142 1143 1144 1145
		 * Try to read the remaining amount, but not more than
		 * the buffer size.
		 * And don't try to read past the end of the file.
		 * If this means reading 0 then we were asked to read
1146 1147
		 * past the end of file.
		 */
1148
		amount = min(amount_left, FSG_BUFLEN);
1149 1150
		amount = min((loff_t)amount,
			     curlun->file_length - file_offset);
1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170
		if (amount == 0) {
			curlun->sense_data =
					SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE;
			curlun->sense_data_info = file_offset >> 9;
			curlun->info_valid = 1;
			break;
		}

		/* Perform the read */
		file_offset_tmp = file_offset;
		nread = vfs_read(curlun->filp,
				(char __user *) bh->buf,
				amount, &file_offset_tmp);
		VLDBG(curlun, "file read %u @ %llu -> %d\n", amount,
				(unsigned long long) file_offset,
				(int) nread);
		if (signal_pending(current))
			return -EINTR;

		if (nread < 0) {
1171
			LDBG(curlun, "error in file verify: %d\n", (int)nread);
1172 1173 1174
			nread = 0;
		} else if (nread < amount) {
			LDBG(curlun, "partial file verify: %d/%u\n",
1175 1176
			     (int)nread, amount);
			nread -= nread & 511;	/* Round down to a sector */
1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192
		}
		if (nread == 0) {
			curlun->sense_data = SS_UNRECOVERED_READ_ERROR;
			curlun->sense_data_info = file_offset >> 9;
			curlun->info_valid = 1;
			break;
		}
		file_offset += nread;
		amount_left -= nread;
	}
	return 0;
}


/*-------------------------------------------------------------------------*/

1193
static int do_inquiry(struct fsg_common *common, struct fsg_buffhd *bh)
1194
{
1195
	struct fsg_lun *curlun = common->curlun;
1196 1197
	u8	*buf = (u8 *) bh->buf;

1198
	if (!curlun) {		/* Unsupported LUNs are okay */
1199
		common->bad_lun_okay = 1;
1200
		memset(buf, 0, 36);
1201 1202
		buf[0] = 0x7f;		/* Unsupported, no device-type */
		buf[4] = 31;		/* Additional length */
1203 1204 1205
		return 36;
	}

1206
	buf[0] = curlun->cdrom ? TYPE_ROM : TYPE_DISK;
1207
	buf[1] = curlun->removable ? 0x80 : 0;
1208 1209 1210 1211
	buf[2] = 2;		/* ANSI SCSI level 2 */
	buf[3] = 2;		/* SCSI-2 INQUIRY data format */
	buf[4] = 31;		/* Additional length */
	buf[5] = 0;		/* No special options */
1212 1213
	buf[6] = 0;
	buf[7] = 0;
1214
	memcpy(buf + 8, common->inquiry_string, sizeof common->inquiry_string);
1215 1216 1217
	return 36;
}

1218
static int do_request_sense(struct fsg_common *common, struct fsg_buffhd *bh)
1219
{
1220
	struct fsg_lun	*curlun = common->curlun;
1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246
	u8		*buf = (u8 *) bh->buf;
	u32		sd, sdinfo;
	int		valid;

	/*
	 * From the SCSI-2 spec., section 7.9 (Unit attention condition):
	 *
	 * If a REQUEST SENSE command is received from an initiator
	 * with a pending unit attention condition (before the target
	 * generates the contingent allegiance condition), then the
	 * target shall either:
	 *   a) report any pending sense data and preserve the unit
	 *	attention condition on the logical unit, or,
	 *   b) report the unit attention condition, may discard any
	 *	pending sense data, and clear the unit attention
	 *	condition on the logical unit for that initiator.
	 *
	 * FSG normally uses option a); enable this code to use option b).
	 */
#if 0
	if (curlun && curlun->unit_attention_data != SS_NO_SENSE) {
		curlun->sense_data = curlun->unit_attention_data;
		curlun->unit_attention_data = SS_NO_SENSE;
	}
#endif

1247
	if (!curlun) {		/* Unsupported LUNs are okay */
1248
		common->bad_lun_okay = 1;
1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261
		sd = SS_LOGICAL_UNIT_NOT_SUPPORTED;
		sdinfo = 0;
		valid = 0;
	} else {
		sd = curlun->sense_data;
		sdinfo = curlun->sense_data_info;
		valid = curlun->info_valid << 7;
		curlun->sense_data = SS_NO_SENSE;
		curlun->sense_data_info = 0;
		curlun->info_valid = 0;
	}

	memset(buf, 0, 18);
1262
	buf[0] = valid | 0x70;			/* Valid, current error */
1263 1264
	buf[2] = SK(sd);
	put_unaligned_be32(sdinfo, &buf[3]);	/* Sense information */
1265
	buf[7] = 18 - 8;			/* Additional sense length */
1266 1267 1268 1269 1270
	buf[12] = ASC(sd);
	buf[13] = ASCQ(sd);
	return 18;
}

1271
static int do_read_capacity(struct fsg_common *common, struct fsg_buffhd *bh)
1272
{
1273 1274 1275
	struct fsg_lun	*curlun = common->curlun;
	u32		lba = get_unaligned_be32(&common->cmnd[2]);
	int		pmi = common->cmnd[8];
1276
	u8		*buf = (u8 *)bh->buf;
1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289

	/* Check the PMI and LBA fields */
	if (pmi > 1 || (pmi == 0 && lba != 0)) {
		curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
		return -EINVAL;
	}

	put_unaligned_be32(curlun->num_sectors - 1, &buf[0]);
						/* Max logical block */
	put_unaligned_be32(512, &buf[4]);	/* Block length */
	return 8;
}

1290
static int do_read_header(struct fsg_common *common, struct fsg_buffhd *bh)
1291
{
1292 1293 1294
	struct fsg_lun	*curlun = common->curlun;
	int		msf = common->cmnd[1] & 0x02;
	u32		lba = get_unaligned_be32(&common->cmnd[2]);
1295
	u8		*buf = (u8 *)bh->buf;
1296

1297
	if (common->cmnd[1] & ~0x02) {		/* Mask away MSF */
1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311
		curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
		return -EINVAL;
	}
	if (lba >= curlun->num_sectors) {
		curlun->sense_data = SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE;
		return -EINVAL;
	}

	memset(buf, 0, 8);
	buf[0] = 0x01;		/* 2048 bytes of user data, rest is EC */
	store_cdrom_address(&buf[4], msf, lba);
	return 8;
}

1312
static int do_read_toc(struct fsg_common *common, struct fsg_buffhd *bh)
1313
{
1314 1315 1316
	struct fsg_lun	*curlun = common->curlun;
	int		msf = common->cmnd[1] & 0x02;
	int		start_track = common->cmnd[6];
1317
	u8		*buf = (u8 *)bh->buf;
1318

1319
	if ((common->cmnd[1] & ~0x02) != 0 ||	/* Mask away MSF */
1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338
			start_track > 1) {
		curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
		return -EINVAL;
	}

	memset(buf, 0, 20);
	buf[1] = (20-2);		/* TOC data length */
	buf[2] = 1;			/* First track number */
	buf[3] = 1;			/* Last track number */
	buf[5] = 0x16;			/* Data track, copying allowed */
	buf[6] = 0x01;			/* Only track is number 1 */
	store_cdrom_address(&buf[8], msf, 0);

	buf[13] = 0x16;			/* Lead-out track is data */
	buf[14] = 0xAA;			/* Lead-out track number */
	store_cdrom_address(&buf[16], msf, curlun->num_sectors);
	return 20;
}

1339
static int do_mode_sense(struct fsg_common *common, struct fsg_buffhd *bh)
1340
{
1341 1342
	struct fsg_lun	*curlun = common->curlun;
	int		mscmnd = common->cmnd[0];
1343 1344 1345 1346 1347 1348 1349
	u8		*buf = (u8 *) bh->buf;
	u8		*buf0 = buf;
	int		pc, page_code;
	int		changeable_values, all_pages;
	int		valid_page = 0;
	int		len, limit;

1350
	if ((common->cmnd[1] & ~0x08) != 0) {	/* Mask away DBD */
1351 1352 1353
		curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
		return -EINVAL;
	}
1354 1355
	pc = common->cmnd[2] >> 6;
	page_code = common->cmnd[2] & 0x3f;
1356 1357 1358 1359 1360 1361 1362
	if (pc == 3) {
		curlun->sense_data = SS_SAVING_PARAMETERS_NOT_SUPPORTED;
		return -EINVAL;
	}
	changeable_values = (pc == 1);
	all_pages = (page_code == 0x3f);

1363 1364
	/*
	 * Write the mode parameter header.  Fixed values are: default
1365 1366
	 * medium type, no cache control (DPOFUA), and no block descriptors.
	 * The only variable value is the WriteProtect bit.  We will fill in
1367 1368
	 * the mode data length later.
	 */
1369
	memset(buf, 0, 8);
1370
	if (mscmnd == MODE_SENSE) {
1371
		buf[2] = (curlun->ro ? 0x80 : 0x00);		/* WP, DPOFUA */
1372 1373
		buf += 4;
		limit = 255;
1374
	} else {			/* MODE_SENSE_10 */
1375
		buf[3] = (curlun->ro ? 0x80 : 0x00);		/* WP, DPOFUA */
1376
		buf += 8;
1377
		limit = 65535;		/* Should really be FSG_BUFLEN */
1378 1379 1380 1381
	}

	/* No block descriptors */

1382 1383 1384 1385
	/*
	 * The mode pages, in numerical order.  The only page we support
	 * is the Caching page.
	 */
1386 1387
	if (page_code == 0x08 || all_pages) {
		valid_page = 1;
1388 1389 1390
		buf[0] = 0x08;		/* Page code */
		buf[1] = 10;		/* Page length */
		memset(buf+2, 0, 10);	/* None of the fields are changeable */
1391 1392

		if (!changeable_values) {
1393 1394 1395
			buf[2] = 0x04;	/* Write cache enable, */
					/* Read cache not disabled */
					/* No cache retention priorities */
1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406
			put_unaligned_be16(0xffff, &buf[4]);
					/* Don't disable prefetch */
					/* Minimum prefetch = 0 */
			put_unaligned_be16(0xffff, &buf[8]);
					/* Maximum prefetch */
			put_unaligned_be16(0xffff, &buf[10]);
					/* Maximum prefetch ceiling */
		}
		buf += 12;
	}

1407 1408 1409 1410
	/*
	 * Check that a valid page was requested and the mode data length
	 * isn't too long.
	 */
1411 1412 1413 1414 1415 1416 1417
	len = buf - buf0;
	if (!valid_page || len > limit) {
		curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
		return -EINVAL;
	}

	/*  Store the mode data length */
1418
	if (mscmnd == MODE_SENSE)
1419 1420 1421 1422 1423 1424
		buf0[0] = len - 1;
	else
		put_unaligned_be16(len - 2, buf0);
	return len;
}

1425
static int do_start_stop(struct fsg_common *common)
1426
{
1427 1428 1429 1430
	struct fsg_lun	*curlun = common->curlun;
	int		loej, start;

	if (!curlun) {
1431
		return -EINVAL;
1432 1433
	} else if (!curlun->removable) {
		curlun->sense_data = SS_INVALID_COMMAND;
1434
		return -EINVAL;
1435 1436
	} else if ((common->cmnd[1] & ~0x01) != 0 || /* Mask away Immed */
		   (common->cmnd[4] & ~0x03) != 0) { /* Mask LoEj, Start */
1437 1438 1439 1440
		curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
		return -EINVAL;
	}

1441 1442
	loej  = common->cmnd[4] & 0x02;
	start = common->cmnd[4] & 0x01;
1443

1444 1445 1446 1447
	/*
	 * Our emulation doesn't support mounting; the medium is
	 * available for use as soon as it is loaded.
	 */
1448
	if (start) {
1449 1450 1451 1452
		if (!fsg_lun_is_open(curlun)) {
			curlun->sense_data = SS_MEDIUM_NOT_PRESENT;
			return -EINVAL;
		}
1453
		return 0;
1454
	}
1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485

	/* Are we allowed to unload the media? */
	if (curlun->prevent_medium_removal) {
		LDBG(curlun, "unload attempt prevented\n");
		curlun->sense_data = SS_MEDIUM_REMOVAL_PREVENTED;
		return -EINVAL;
	}

	if (!loej)
		return 0;

	/* Simulate an unload/eject */
	if (common->ops && common->ops->pre_eject) {
		int r = common->ops->pre_eject(common, curlun,
					       curlun - common->luns);
		if (unlikely(r < 0))
			return r;
		else if (r)
			return 0;
	}

	up_read(&common->filesem);
	down_write(&common->filesem);
	fsg_lun_close(curlun);
	up_write(&common->filesem);
	down_read(&common->filesem);

	return common->ops && common->ops->post_eject
		? min(0, common->ops->post_eject(common, curlun,
						 curlun - common->luns))
		: 0;
1486 1487
}

1488
static int do_prevent_allow(struct fsg_common *common)
1489
{
1490
	struct fsg_lun	*curlun = common->curlun;
1491 1492
	int		prevent;

1493
	if (!common->curlun) {
1494
		return -EINVAL;
1495 1496
	} else if (!common->curlun->removable) {
		common->curlun->sense_data = SS_INVALID_COMMAND;
1497 1498 1499
		return -EINVAL;
	}

1500 1501
	prevent = common->cmnd[4] & 0x01;
	if ((common->cmnd[4] & ~0x01) != 0) {	/* Mask away Prevent */
1502 1503 1504 1505 1506 1507 1508 1509 1510 1511
		curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
		return -EINVAL;
	}

	if (curlun->prevent_medium_removal && !prevent)
		fsg_lun_fsync_sub(curlun);
	curlun->prevent_medium_removal = prevent;
	return 0;
}

1512
static int do_read_format_capacities(struct fsg_common *common,
1513 1514
			struct fsg_buffhd *bh)
{
1515
	struct fsg_lun	*curlun = common->curlun;
1516 1517 1518
	u8		*buf = (u8 *) bh->buf;

	buf[0] = buf[1] = buf[2] = 0;
1519
	buf[3] = 8;	/* Only the Current/Maximum Capacity Descriptor */
1520 1521 1522 1523 1524 1525 1526 1527 1528
	buf += 4;

	put_unaligned_be32(curlun->num_sectors, &buf[0]);
						/* Number of blocks */
	put_unaligned_be32(512, &buf[4]);	/* Block length */
	buf[4] = 0x02;				/* Current capacity */
	return 12;
}

1529
static int do_mode_select(struct fsg_common *common, struct fsg_buffhd *bh)
1530
{
1531
	struct fsg_lun	*curlun = common->curlun;
1532 1533

	/* We don't support MODE SELECT */
1534 1535
	if (curlun)
		curlun->sense_data = SS_INVALID_COMMAND;
1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588
	return -EINVAL;
}


/*-------------------------------------------------------------------------*/

static int halt_bulk_in_endpoint(struct fsg_dev *fsg)
{
	int	rc;

	rc = fsg_set_halt(fsg, fsg->bulk_in);
	if (rc == -EAGAIN)
		VDBG(fsg, "delayed bulk-in endpoint halt\n");
	while (rc != 0) {
		if (rc != -EAGAIN) {
			WARNING(fsg, "usb_ep_set_halt -> %d\n", rc);
			rc = 0;
			break;
		}

		/* Wait for a short time and then try again */
		if (msleep_interruptible(100) != 0)
			return -EINTR;
		rc = usb_ep_set_halt(fsg->bulk_in);
	}
	return rc;
}

static int wedge_bulk_in_endpoint(struct fsg_dev *fsg)
{
	int	rc;

	DBG(fsg, "bulk-in set wedge\n");
	rc = usb_ep_set_wedge(fsg->bulk_in);
	if (rc == -EAGAIN)
		VDBG(fsg, "delayed bulk-in endpoint wedge\n");
	while (rc != 0) {
		if (rc != -EAGAIN) {
			WARNING(fsg, "usb_ep_set_wedge -> %d\n", rc);
			rc = 0;
			break;
		}

		/* Wait for a short time and then try again */
		if (msleep_interruptible(100) != 0)
			return -EINTR;
		rc = usb_ep_set_wedge(fsg->bulk_in);
	}
	return rc;
}

static int pad_with_zeros(struct fsg_dev *fsg)
{
1589
	struct fsg_buffhd	*bh = fsg->common->next_buffhd_to_fill;
1590 1591 1592 1593
	u32			nkeep = bh->inreq->length;
	u32			nsend;
	int			rc;

1594
	bh->state = BUF_STATE_EMPTY;		/* For the first iteration */
1595 1596
	fsg->common->usb_amount_left = nkeep + fsg->common->residue;
	while (fsg->common->usb_amount_left > 0) {
1597 1598 1599

		/* Wait for the next buffer to be free */
		while (bh->state != BUF_STATE_EMPTY) {
1600
			rc = sleep_thread(fsg->common);
1601 1602 1603 1604
			if (rc)
				return rc;
		}

1605
		nsend = min(fsg->common->usb_amount_left, FSG_BUFLEN);
1606 1607 1608 1609
		memset(bh->buf + nkeep, 0, nsend - nkeep);
		bh->inreq->length = nsend;
		bh->inreq->zero = 0;
		start_transfer(fsg, fsg->bulk_in, bh->inreq,
1610
			       &bh->inreq_busy, &bh->state);
1611
		bh = fsg->common->next_buffhd_to_fill = bh->next;
1612
		fsg->common->usb_amount_left -= nsend;
1613 1614 1615 1616 1617
		nkeep = 0;
	}
	return 0;
}

1618
static int throw_away_data(struct fsg_common *common)
1619 1620 1621 1622 1623
{
	struct fsg_buffhd	*bh;
	u32			amount;
	int			rc;

1624 1625 1626
	for (bh = common->next_buffhd_to_drain;
	     bh->state != BUF_STATE_EMPTY || common->usb_amount_left > 0;
	     bh = common->next_buffhd_to_drain) {
1627 1628 1629 1630 1631

		/* Throw away the data in a filled buffer */
		if (bh->state == BUF_STATE_FULL) {
			smp_rmb();
			bh->state = BUF_STATE_EMPTY;
1632
			common->next_buffhd_to_drain = bh->next;
1633 1634 1635

			/* A short packet or an error ends everything */
			if (bh->outreq->actual != bh->outreq->length ||
1636
			    bh->outreq->status != 0) {
1637 1638
				raise_exception(common,
						FSG_STATE_ABORT_BULK_OUT);
1639 1640 1641 1642 1643 1644
				return -EINTR;
			}
			continue;
		}

		/* Try to submit another request if we need one */
1645 1646 1647 1648
		bh = common->next_buffhd_to_fill;
		if (bh->state == BUF_STATE_EMPTY
		 && common->usb_amount_left > 0) {
			amount = min(common->usb_amount_left, FSG_BUFLEN);
1649

1650 1651 1652 1653
			/*
			 * amount is always divisible by 512, hence by
			 * the bulk-out maxpacket size.
			 */
1654 1655
			bh->outreq->length = amount;
			bh->bulk_out_intended_length = amount;
1656
			bh->outreq->short_not_ok = 1;
1657
			if (!start_out_transfer(common, bh))
1658
				/* Dunno what to do if common->fsg is NULL */
1659 1660 1661
				return -EIO;
			common->next_buffhd_to_fill = bh->next;
			common->usb_amount_left -= amount;
1662 1663 1664 1665
			continue;
		}

		/* Otherwise wait for something to happen */
1666
		rc = sleep_thread(common);
1667 1668 1669 1670 1671 1672
		if (rc)
			return rc;
	}
	return 0;
}

1673
static int finish_reply(struct fsg_common *common)
1674
{
1675
	struct fsg_buffhd	*bh = common->next_buffhd_to_fill;
1676 1677
	int			rc = 0;

1678
	switch (common->data_dir) {
1679
	case DATA_DIR_NONE:
1680
		break;			/* Nothing to send */
1681

1682 1683
	/*
	 * If we don't know whether the host wants to read or write,
1684 1685
	 * this must be CB or CBI with an unknown command.  We mustn't
	 * try to send or receive any data.  So stall both bulk pipes
1686 1687
	 * if we can and wait for a reset.
	 */
1688
	case DATA_DIR_UNKNOWN:
1689 1690 1691 1692 1693 1694 1695 1696
		if (!common->can_stall) {
			/* Nothing */
		} else if (fsg_is_set(common)) {
			fsg_set_halt(common->fsg, common->fsg->bulk_out);
			rc = halt_bulk_in_endpoint(common->fsg);
		} else {
			/* Don't know what to do if common->fsg is NULL */
			rc = -EIO;
1697 1698 1699 1700 1701
		}
		break;

	/* All but the last buffer of data must have already been sent */
	case DATA_DIR_TO_HOST:
1702
		if (common->data_size == 0) {
1703
			/* Nothing to send */
1704 1705

		/* If there's no residue, simply send the last buffer */
1706
		} else if (common->residue == 0) {
1707
			bh->inreq->zero = 0;
1708
			if (!start_in_transfer(common, bh))
1709 1710
				return -EIO;
			common->next_buffhd_to_fill = bh->next;
1711

1712 1713
		/*
		 * For Bulk-only, if we're allowed to stall then send the
1714
		 * short packet and halt the bulk-in endpoint.  If we can't
1715 1716
		 * stall, pad out the remaining data with 0's.
		 */
1717
		} else if (common->can_stall) {
1718
			bh->inreq->zero = 1;
1719
			if (!start_in_transfer(common, bh))
1720 1721 1722 1723 1724 1725 1726 1727
				/* Don't know what to do if
				 * common->fsg is NULL */
				rc = -EIO;
			common->next_buffhd_to_fill = bh->next;
			if (common->fsg)
				rc = halt_bulk_in_endpoint(common->fsg);
		} else if (fsg_is_set(common)) {
			rc = pad_with_zeros(common->fsg);
1728
		} else {
1729 1730
			/* Don't know what to do if common->fsg is NULL */
			rc = -EIO;
1731 1732 1733
		}
		break;

1734 1735 1736 1737
	/*
	 * We have processed all we want from the data the host has sent.
	 * There may still be outstanding bulk-out requests.
	 */
1738
	case DATA_DIR_FROM_HOST:
1739
		if (common->residue == 0) {
1740
			/* Nothing to receive */
1741 1742

		/* Did the host stop sending unexpectedly early? */
1743 1744
		} else if (common->short_packet_received) {
			raise_exception(common, FSG_STATE_ABORT_BULK_OUT);
1745 1746
			rc = -EINTR;

1747 1748
		/*
		 * We haven't processed all the incoming data.  Even though
1749 1750 1751 1752
		 * we may be allowed to stall, doing so would cause a race.
		 * The controller may already have ACK'ed all the remaining
		 * bulk-out packets, in which case the host wouldn't see a
		 * STALL.  Not realizing the endpoint was halted, it wouldn't
1753 1754
		 * clear the halt -- leading to problems later on.
		 */
1755
#if 0
1756 1757 1758 1759 1760
		} else if (common->can_stall) {
			if (fsg_is_set(common))
				fsg_set_halt(common->fsg,
					     common->fsg->bulk_out);
			raise_exception(common, FSG_STATE_ABORT_BULK_OUT);
1761 1762 1763
			rc = -EINTR;
#endif

1764 1765 1766 1767
		/*
		 * We can't stall.  Read in the excess data and throw it
		 * all away.
		 */
1768
		} else {
1769
			rc = throw_away_data(common);
1770
		}
1771 1772 1773 1774 1775
		break;
	}
	return rc;
}

1776
static int send_status(struct fsg_common *common)
1777
{
1778
	struct fsg_lun		*curlun = common->curlun;
1779
	struct fsg_buffhd	*bh;
1780
	struct bulk_cs_wrap	*csw;
1781 1782 1783 1784 1785
	int			rc;
	u8			status = USB_STATUS_PASS;
	u32			sd, sdinfo = 0;

	/* Wait for the next buffer to become available */
1786
	bh = common->next_buffhd_to_fill;
1787
	while (bh->state != BUF_STATE_EMPTY) {
1788
		rc = sleep_thread(common);
1789 1790 1791 1792 1793 1794 1795
		if (rc)
			return rc;
	}

	if (curlun) {
		sd = curlun->sense_data;
		sdinfo = curlun->sense_data_info;
1796
	} else if (common->bad_lun_okay)
1797 1798 1799 1800
		sd = SS_NO_SENSE;
	else
		sd = SS_LOGICAL_UNIT_NOT_SUPPORTED;

1801 1802
	if (common->phase_error) {
		DBG(common, "sending phase-error status\n");
1803 1804 1805
		status = USB_STATUS_PHASE_ERROR;
		sd = SS_INVALID_COMMAND;
	} else if (sd != SS_NO_SENSE) {
1806
		DBG(common, "sending command-failure status\n");
1807
		status = USB_STATUS_FAIL;
1808
		VDBG(common, "  sense data: SK x%02x, ASC x%02x, ASCQ x%02x;"
1809 1810 1811 1812
				"  info x%x\n",
				SK(sd), ASC(sd), ASCQ(sd), sdinfo);
	}

1813
	/* Store and send the Bulk-only CSW */
1814
	csw = (void *)bh->buf;
1815

1816
	csw->Signature = cpu_to_le32(USB_BULK_CS_SIG);
1817 1818
	csw->Tag = common->tag;
	csw->Residue = cpu_to_le32(common->residue);
1819
	csw->Status = status;
1820

1821 1822
	bh->inreq->length = USB_BULK_CS_WRAP_LEN;
	bh->inreq->zero = 0;
1823
	if (!start_in_transfer(common, bh))
1824 1825
		/* Don't know what to do if common->fsg is NULL */
		return -EIO;
1826

1827
	common->next_buffhd_to_fill = bh->next;
1828 1829 1830 1831 1832 1833
	return 0;
}


/*-------------------------------------------------------------------------*/

1834 1835 1836 1837
/*
 * Check whether the command is properly formed and whether its data size
 * and direction agree with the values we already have.
 */
1838
static int check_command(struct fsg_common *common, int cmnd_size,
1839 1840
			 enum data_direction data_dir, unsigned int mask,
			 int needs_medium, const char *name)
1841 1842
{
	int			i;
1843
	int			lun = common->cmnd[1] >> 5;
1844 1845 1846 1847 1848
	static const char	dirletter[4] = {'u', 'o', 'i', 'n'};
	char			hdlen[20];
	struct fsg_lun		*curlun;

	hdlen[0] = 0;
1849 1850
	if (common->data_dir != DATA_DIR_UNKNOWN)
		sprintf(hdlen, ", H%c=%u", dirletter[(int) common->data_dir],
1851
			common->data_size);
1852
	VDBG(common, "SCSI command: %s;  Dc=%d, D%c=%u;  Hc=%d%s\n",
1853
	     name, cmnd_size, dirletter[(int) data_dir],
1854
	     common->data_size_from_cmnd, common->cmnd_size, hdlen);
1855

1856 1857 1858 1859
	/*
	 * We can't reply at all until we know the correct data direction
	 * and size.
	 */
1860
	if (common->data_size_from_cmnd == 0)
1861
		data_dir = DATA_DIR_NONE;
1862
	if (common->data_size < common->data_size_from_cmnd) {
1863 1864
		/*
		 * Host data size < Device data size is a phase error.
1865
		 * Carry out the command, but only transfer as much as
1866 1867
		 * we are allowed.
		 */
1868 1869
		common->data_size_from_cmnd = common->data_size;
		common->phase_error = 1;
1870
	}
1871 1872
	common->residue = common->data_size;
	common->usb_amount_left = common->data_size;
1873 1874

	/* Conflicting data directions is a phase error */
1875
	if (common->data_dir != data_dir && common->data_size_from_cmnd > 0) {
1876
		common->phase_error = 1;
1877 1878 1879 1880
		return -EINVAL;
	}

	/* Verify the length of the command itself */
1881
	if (cmnd_size != common->cmnd_size) {
1882

1883 1884
		/*
		 * Special case workaround: There are plenty of buggy SCSI
1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895
		 * implementations. Many have issues with cbw->Length
		 * field passing a wrong command size. For those cases we
		 * always try to work around the problem by using the length
		 * sent by the host side provided it is at least as large
		 * as the correct command length.
		 * Examples of such cases would be MS-Windows, which issues
		 * REQUEST SENSE with cbw->Length == 12 where it should
		 * be 6, and xbox360 issuing INQUIRY, TEST UNIT READY and
		 * REQUEST SENSE with cbw->Length == 10 where it should
		 * be 6 as well.
		 */
1896 1897
		if (cmnd_size <= common->cmnd_size) {
			DBG(common, "%s is buggy! Expected length %d "
1898
			    "but we got %d\n", name,
1899 1900
			    cmnd_size, common->cmnd_size);
			cmnd_size = common->cmnd_size;
1901
		} else {
1902
			common->phase_error = 1;
1903 1904 1905 1906 1907
			return -EINVAL;
		}
	}

	/* Check that the LUN values are consistent */
1908 1909 1910
	if (common->lun != lun)
		DBG(common, "using LUN %d from CBW, not LUN %d from CDB\n",
		    common->lun, lun);
1911 1912

	/* Check the LUN */
1913 1914 1915
	if (common->lun >= 0 && common->lun < common->nluns) {
		curlun = &common->luns[common->lun];
		common->curlun = curlun;
1916
		if (common->cmnd[0] != REQUEST_SENSE) {
1917 1918 1919 1920 1921
			curlun->sense_data = SS_NO_SENSE;
			curlun->sense_data_info = 0;
			curlun->info_valid = 0;
		}
	} else {
1922 1923 1924
		common->curlun = NULL;
		curlun = NULL;
		common->bad_lun_okay = 0;
1925

1926 1927 1928 1929
		/*
		 * INQUIRY and REQUEST SENSE commands are explicitly allowed
		 * to use unsupported LUNs; all others may not.
		 */
1930 1931
		if (common->cmnd[0] != INQUIRY &&
		    common->cmnd[0] != REQUEST_SENSE) {
1932
			DBG(common, "unsupported LUN %d\n", common->lun);
1933 1934 1935 1936
			return -EINVAL;
		}
	}

1937 1938 1939 1940
	/*
	 * If a unit attention condition exists, only INQUIRY and
	 * REQUEST SENSE commands are allowed; anything else must fail.
	 */
1941
	if (curlun && curlun->unit_attention_data != SS_NO_SENSE &&
1942 1943
	    common->cmnd[0] != INQUIRY &&
	    common->cmnd[0] != REQUEST_SENSE) {
1944 1945 1946 1947 1948 1949
		curlun->sense_data = curlun->unit_attention_data;
		curlun->unit_attention_data = SS_NO_SENSE;
		return -EINVAL;
	}

	/* Check that only command bytes listed in the mask are non-zero */
1950
	common->cmnd[1] &= 0x1f;			/* Mask away the LUN */
1951
	for (i = 1; i < cmnd_size; ++i) {
1952
		if (common->cmnd[i] && !(mask & (1 << i))) {
1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968
			if (curlun)
				curlun->sense_data = SS_INVALID_FIELD_IN_CDB;
			return -EINVAL;
		}
	}

	/* If the medium isn't mounted and the command needs to access
	 * it, return an error. */
	if (curlun && !fsg_lun_is_open(curlun) && needs_medium) {
		curlun->sense_data = SS_MEDIUM_NOT_PRESENT;
		return -EINVAL;
	}

	return 0;
}

1969
static int do_scsi_command(struct fsg_common *common)
1970 1971 1972 1973 1974 1975 1976
{
	struct fsg_buffhd	*bh;
	int			rc;
	int			reply = -EINVAL;
	int			i;
	static char		unknown[16];

1977
	dump_cdb(common);
1978 1979

	/* Wait for the next buffer to become available for data or status */
1980 1981
	bh = common->next_buffhd_to_fill;
	common->next_buffhd_to_drain = bh;
1982
	while (bh->state != BUF_STATE_EMPTY) {
1983
		rc = sleep_thread(common);
1984 1985 1986
		if (rc)
			return rc;
	}
1987 1988
	common->phase_error = 0;
	common->short_packet_received = 0;
1989

1990 1991
	down_read(&common->filesem);	/* We're using the backing file */
	switch (common->cmnd[0]) {
1992

1993
	case INQUIRY:
1994 1995
		common->data_size_from_cmnd = common->cmnd[4];
		reply = check_command(common, 6, DATA_DIR_TO_HOST,
1996 1997 1998
				      (1<<4), 0,
				      "INQUIRY");
		if (reply == 0)
1999
			reply = do_inquiry(common, bh);
2000 2001
		break;

2002
	case MODE_SELECT:
2003 2004
		common->data_size_from_cmnd = common->cmnd[4];
		reply = check_command(common, 6, DATA_DIR_FROM_HOST,
2005 2006 2007
				      (1<<1) | (1<<4), 0,
				      "MODE SELECT(6)");
		if (reply == 0)
2008
			reply = do_mode_select(common, bh);
2009 2010
		break;

2011
	case MODE_SELECT_10:
2012 2013 2014
		common->data_size_from_cmnd =
			get_unaligned_be16(&common->cmnd[7]);
		reply = check_command(common, 10, DATA_DIR_FROM_HOST,
2015 2016 2017
				      (1<<1) | (3<<7), 0,
				      "MODE SELECT(10)");
		if (reply == 0)
2018
			reply = do_mode_select(common, bh);
2019 2020
		break;

2021
	case MODE_SENSE:
2022 2023
		common->data_size_from_cmnd = common->cmnd[4];
		reply = check_command(common, 6, DATA_DIR_TO_HOST,
2024 2025 2026
				      (1<<1) | (1<<2) | (1<<4), 0,
				      "MODE SENSE(6)");
		if (reply == 0)
2027
			reply = do_mode_sense(common, bh);
2028 2029
		break;

2030
	case MODE_SENSE_10:
2031 2032 2033
		common->data_size_from_cmnd =
			get_unaligned_be16(&common->cmnd[7]);
		reply = check_command(common, 10, DATA_DIR_TO_HOST,
2034 2035 2036
				      (1<<1) | (1<<2) | (3<<7), 0,
				      "MODE SENSE(10)");
		if (reply == 0)
2037
			reply = do_mode_sense(common, bh);
2038 2039
		break;

2040
	case ALLOW_MEDIUM_REMOVAL:
2041 2042
		common->data_size_from_cmnd = 0;
		reply = check_command(common, 6, DATA_DIR_NONE,
2043 2044 2045
				      (1<<4), 0,
				      "PREVENT-ALLOW MEDIUM REMOVAL");
		if (reply == 0)
2046
			reply = do_prevent_allow(common);
2047 2048
		break;

2049
	case READ_6:
2050 2051 2052
		i = common->cmnd[4];
		common->data_size_from_cmnd = (i == 0 ? 256 : i) << 9;
		reply = check_command(common, 6, DATA_DIR_TO_HOST,
2053 2054 2055
				      (7<<1) | (1<<4), 1,
				      "READ(6)");
		if (reply == 0)
2056
			reply = do_read(common);
2057 2058
		break;

2059
	case READ_10:
2060 2061 2062
		common->data_size_from_cmnd =
				get_unaligned_be16(&common->cmnd[7]) << 9;
		reply = check_command(common, 10, DATA_DIR_TO_HOST,
2063 2064 2065
				      (1<<1) | (0xf<<2) | (3<<7), 1,
				      "READ(10)");
		if (reply == 0)
2066
			reply = do_read(common);
2067 2068
		break;

2069
	case READ_12:
2070 2071 2072
		common->data_size_from_cmnd =
				get_unaligned_be32(&common->cmnd[6]) << 9;
		reply = check_command(common, 12, DATA_DIR_TO_HOST,
2073 2074 2075
				      (1<<1) | (0xf<<2) | (0xf<<6), 1,
				      "READ(12)");
		if (reply == 0)
2076
			reply = do_read(common);
2077 2078
		break;

2079
	case READ_CAPACITY:
2080 2081
		common->data_size_from_cmnd = 8;
		reply = check_command(common, 10, DATA_DIR_TO_HOST,
2082 2083 2084
				      (0xf<<2) | (1<<8), 1,
				      "READ CAPACITY");
		if (reply == 0)
2085
			reply = do_read_capacity(common, bh);
2086 2087
		break;

2088
	case READ_HEADER:
2089
		if (!common->curlun || !common->curlun->cdrom)
2090
			goto unknown_cmnd;
2091 2092 2093
		common->data_size_from_cmnd =
			get_unaligned_be16(&common->cmnd[7]);
		reply = check_command(common, 10, DATA_DIR_TO_HOST,
2094 2095 2096
				      (3<<7) | (0x1f<<1), 1,
				      "READ HEADER");
		if (reply == 0)
2097
			reply = do_read_header(common, bh);
2098 2099
		break;

2100
	case READ_TOC:
2101
		if (!common->curlun || !common->curlun->cdrom)
2102
			goto unknown_cmnd;
2103 2104 2105
		common->data_size_from_cmnd =
			get_unaligned_be16(&common->cmnd[7]);
		reply = check_command(common, 10, DATA_DIR_TO_HOST,
2106 2107 2108
				      (7<<6) | (1<<1), 1,
				      "READ TOC");
		if (reply == 0)
2109
			reply = do_read_toc(common, bh);
2110 2111
		break;

2112
	case READ_FORMAT_CAPACITIES:
2113 2114 2115
		common->data_size_from_cmnd =
			get_unaligned_be16(&common->cmnd[7]);
		reply = check_command(common, 10, DATA_DIR_TO_HOST,
2116 2117 2118
				      (3<<7), 1,
				      "READ FORMAT CAPACITIES");
		if (reply == 0)
2119
			reply = do_read_format_capacities(common, bh);
2120 2121
		break;

2122
	case REQUEST_SENSE:
2123 2124
		common->data_size_from_cmnd = common->cmnd[4];
		reply = check_command(common, 6, DATA_DIR_TO_HOST,
2125 2126 2127
				      (1<<4), 0,
				      "REQUEST SENSE");
		if (reply == 0)
2128
			reply = do_request_sense(common, bh);
2129 2130
		break;

2131
	case START_STOP:
2132 2133
		common->data_size_from_cmnd = 0;
		reply = check_command(common, 6, DATA_DIR_NONE,
2134 2135 2136
				      (1<<1) | (1<<4), 0,
				      "START-STOP UNIT");
		if (reply == 0)
2137
			reply = do_start_stop(common);
2138 2139
		break;

2140
	case SYNCHRONIZE_CACHE:
2141 2142
		common->data_size_from_cmnd = 0;
		reply = check_command(common, 10, DATA_DIR_NONE,
2143 2144 2145
				      (0xf<<2) | (3<<7), 1,
				      "SYNCHRONIZE CACHE");
		if (reply == 0)
2146
			reply = do_synchronize_cache(common);
2147 2148
		break;

2149
	case TEST_UNIT_READY:
2150 2151
		common->data_size_from_cmnd = 0;
		reply = check_command(common, 6, DATA_DIR_NONE,
2152 2153 2154 2155
				0, 1,
				"TEST UNIT READY");
		break;

2156 2157 2158 2159
	/*
	 * Although optional, this command is used by MS-Windows.  We
	 * support a minimal version: BytChk must be 0.
	 */
2160
	case VERIFY:
2161 2162
		common->data_size_from_cmnd = 0;
		reply = check_command(common, 10, DATA_DIR_NONE,
2163 2164 2165
				      (1<<1) | (0xf<<2) | (3<<7), 1,
				      "VERIFY");
		if (reply == 0)
2166
			reply = do_verify(common);
2167 2168
		break;

2169
	case WRITE_6:
2170 2171 2172
		i = common->cmnd[4];
		common->data_size_from_cmnd = (i == 0 ? 256 : i) << 9;
		reply = check_command(common, 6, DATA_DIR_FROM_HOST,
2173 2174 2175
				      (7<<1) | (1<<4), 1,
				      "WRITE(6)");
		if (reply == 0)
2176
			reply = do_write(common);
2177 2178
		break;

2179
	case WRITE_10:
2180 2181 2182
		common->data_size_from_cmnd =
				get_unaligned_be16(&common->cmnd[7]) << 9;
		reply = check_command(common, 10, DATA_DIR_FROM_HOST,
2183 2184 2185
				      (1<<1) | (0xf<<2) | (3<<7), 1,
				      "WRITE(10)");
		if (reply == 0)
2186
			reply = do_write(common);
2187 2188
		break;

2189
	case WRITE_12:
2190 2191 2192
		common->data_size_from_cmnd =
				get_unaligned_be32(&common->cmnd[6]) << 9;
		reply = check_command(common, 12, DATA_DIR_FROM_HOST,
2193 2194 2195
				      (1<<1) | (0xf<<2) | (0xf<<6), 1,
				      "WRITE(12)");
		if (reply == 0)
2196
			reply = do_write(common);
2197 2198
		break;

2199 2200
	/*
	 * Some mandatory commands that we recognize but don't implement.
2201 2202
	 * They don't mean much in this setting.  It's left as an exercise
	 * for anyone interested to implement RESERVE and RELEASE in terms
2203 2204
	 * of Posix locks.
	 */
2205 2206 2207 2208
	case FORMAT_UNIT:
	case RELEASE:
	case RESERVE:
	case SEND_DIAGNOSTIC:
2209
		/* Fall through */
2210 2211

	default:
2212
unknown_cmnd:
2213 2214 2215
		common->data_size_from_cmnd = 0;
		sprintf(unknown, "Unknown x%02x", common->cmnd[0]);
		reply = check_command(common, common->cmnd_size,
2216 2217
				      DATA_DIR_UNKNOWN, 0xff, 0, unknown);
		if (reply == 0) {
2218
			common->curlun->sense_data = SS_INVALID_COMMAND;
2219 2220 2221 2222
			reply = -EINVAL;
		}
		break;
	}
2223
	up_read(&common->filesem);
2224 2225 2226 2227 2228 2229

	if (reply == -EINTR || signal_pending(current))
		return -EINTR;

	/* Set up the single reply buffer for finish_reply() */
	if (reply == -EINVAL)
2230
		reply = 0;		/* Error reply length */
2231
	if (reply >= 0 && common->data_dir == DATA_DIR_TO_HOST) {
2232
		reply = min((u32)reply, common->data_size_from_cmnd);
2233 2234
		bh->inreq->length = reply;
		bh->state = BUF_STATE_FULL;
2235
		common->residue -= reply;
2236
	}				/* Otherwise it's already set */
2237 2238 2239 2240 2241 2242 2243 2244 2245

	return 0;
}


/*-------------------------------------------------------------------------*/

static int received_cbw(struct fsg_dev *fsg, struct fsg_buffhd *bh)
{
2246
	struct usb_request	*req = bh->outreq;
2247
	struct fsg_bulk_cb_wrap	*cbw = req->buf;
2248
	struct fsg_common	*common = fsg->common;
2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261

	/* Was this a real packet?  Should it be ignored? */
	if (req->status || test_bit(IGNORE_BULK_OUT, &fsg->atomic_bitflags))
		return -EINVAL;

	/* Is the CBW valid? */
	if (req->actual != USB_BULK_CB_WRAP_LEN ||
			cbw->Signature != cpu_to_le32(
				USB_BULK_CB_SIG)) {
		DBG(fsg, "invalid CBW: len %u sig 0x%x\n",
				req->actual,
				le32_to_cpu(cbw->Signature));

2262 2263
		/*
		 * The Bulk-only spec says we MUST stall the IN endpoint
2264 2265 2266 2267 2268 2269 2270
		 * (6.6.1), so it's unavoidable.  It also says we must
		 * retain this state until the next reset, but there's
		 * no way to tell the controller driver it should ignore
		 * Clear-Feature(HALT) requests.
		 *
		 * We aren't required to halt the OUT endpoint; instead
		 * we can simply accept and discard any data received
2271 2272
		 * until the next reset.
		 */
2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284
		wedge_bulk_in_endpoint(fsg);
		set_bit(IGNORE_BULK_OUT, &fsg->atomic_bitflags);
		return -EINVAL;
	}

	/* Is the CBW meaningful? */
	if (cbw->Lun >= FSG_MAX_LUNS || cbw->Flags & ~USB_BULK_IN_FLAG ||
			cbw->Length <= 0 || cbw->Length > MAX_COMMAND_SIZE) {
		DBG(fsg, "non-meaningful CBW: lun = %u, flags = 0x%x, "
				"cmdlen %u\n",
				cbw->Lun, cbw->Flags, cbw->Length);

2285 2286 2287 2288
		/*
		 * We can do anything we want here, so let's stall the
		 * bulk pipes if we are allowed to.
		 */
2289
		if (common->can_stall) {
2290 2291 2292 2293 2294 2295 2296
			fsg_set_halt(fsg, fsg->bulk_out);
			halt_bulk_in_endpoint(fsg);
		}
		return -EINVAL;
	}

	/* Save the command for later */
2297 2298
	common->cmnd_size = cbw->Length;
	memcpy(common->cmnd, cbw->CDB, common->cmnd_size);
2299
	if (cbw->Flags & USB_BULK_IN_FLAG)
2300
		common->data_dir = DATA_DIR_TO_HOST;
2301
	else
2302 2303 2304 2305 2306 2307
		common->data_dir = DATA_DIR_FROM_HOST;
	common->data_size = le32_to_cpu(cbw->DataTransferLength);
	if (common->data_size == 0)
		common->data_dir = DATA_DIR_NONE;
	common->lun = cbw->Lun;
	common->tag = cbw->Tag;
2308 2309 2310
	return 0;
}

2311
static int get_next_command(struct fsg_common *common)
2312 2313 2314 2315
{
	struct fsg_buffhd	*bh;
	int			rc = 0;

2316
	/* Wait for the next buffer to become available */
2317
	bh = common->next_buffhd_to_fill;
2318
	while (bh->state != BUF_STATE_EMPTY) {
2319
		rc = sleep_thread(common);
2320 2321 2322
		if (rc)
			return rc;
	}
2323

2324
	/* Queue a request to read a Bulk-only CBW */
2325
	set_bulk_out_req_length(common, bh, USB_BULK_CB_WRAP_LEN);
2326
	bh->outreq->short_not_ok = 1;
2327
	if (!start_out_transfer(common, bh))
2328 2329
		/* Don't know what to do if common->fsg is NULL */
		return -EIO;
2330

2331 2332
	/*
	 * We will drain the buffer in software, which means we
2333
	 * can reuse it for the next filling.  No need to advance
2334 2335
	 * next_buffhd_to_fill.
	 */
2336

2337 2338
	/* Wait for the CBW to arrive */
	while (bh->state != BUF_STATE_FULL) {
2339
		rc = sleep_thread(common);
2340 2341
		if (rc)
			return rc;
2342
	}
2343
	smp_rmb();
2344
	rc = fsg_is_set(common) ? received_cbw(common->fsg, bh) : -EIO;
2345 2346
	bh->state = BUF_STATE_EMPTY;

2347 2348 2349 2350 2351 2352
	return rc;
}


/*-------------------------------------------------------------------------*/

2353
static int enable_endpoint(struct fsg_common *common, struct usb_ep *ep,
2354 2355 2356 2357
		const struct usb_endpoint_descriptor *d)
{
	int	rc;

2358
	ep->driver_data = common;
2359 2360
	rc = usb_ep_enable(ep, d);
	if (rc)
2361
		ERROR(common, "can't enable %s, result %d\n", ep->name, rc);
2362 2363 2364
	return rc;
}

2365
static int alloc_request(struct fsg_common *common, struct usb_ep *ep,
2366 2367 2368 2369 2370
		struct usb_request **preq)
{
	*preq = usb_ep_alloc_request(ep, GFP_ATOMIC);
	if (*preq)
		return 0;
2371
	ERROR(common, "can't allocate request for %s\n", ep->name);
2372 2373 2374
	return -ENOMEM;
}

2375 2376
/* Reset interface setting and re-init endpoint state (toggle etc). */
static int do_set_interface(struct fsg_common *common, struct fsg_dev *new_fsg)
2377
{
2378 2379 2380
	const struct usb_endpoint_descriptor *d;
	struct fsg_dev *fsg;
	int i, rc = 0;
2381

2382 2383
	if (common->running)
		DBG(common, "reset interface\n");
2384 2385 2386

reset:
	/* Deallocate the requests */
2387 2388
	if (common->fsg) {
		fsg = common->fsg;
2389 2390 2391

		for (i = 0; i < FSG_NUM_BUFFERS; ++i) {
			struct fsg_buffhd *bh = &common->buffhds[i];
2392

2393 2394 2395 2396 2397 2398 2399 2400
			if (bh->inreq) {
				usb_ep_free_request(fsg->bulk_in, bh->inreq);
				bh->inreq = NULL;
			}
			if (bh->outreq) {
				usb_ep_free_request(fsg->bulk_out, bh->outreq);
				bh->outreq = NULL;
			}
2401
		}
2402 2403 2404 2405 2406 2407 2408 2409 2410

		/* Disable the endpoints */
		if (fsg->bulk_in_enabled) {
			usb_ep_disable(fsg->bulk_in);
			fsg->bulk_in_enabled = 0;
		}
		if (fsg->bulk_out_enabled) {
			usb_ep_disable(fsg->bulk_out);
			fsg->bulk_out_enabled = 0;
2411 2412
		}

2413 2414
		common->fsg = NULL;
		wake_up(&common->fsg_wait);
2415 2416
	}

2417
	common->running = 0;
2418
	if (!new_fsg || rc)
2419 2420
		return rc;

2421 2422
	common->fsg = new_fsg;
	fsg = common->fsg;
2423

2424 2425 2426 2427 2428 2429 2430
	/* Enable the endpoints */
	d = fsg_ep_desc(common->gadget,
			&fsg_fs_bulk_in_desc, &fsg_hs_bulk_in_desc);
	rc = enable_endpoint(common, fsg->bulk_in, d);
	if (rc)
		goto reset;
	fsg->bulk_in_enabled = 1;
2431

2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445
	d = fsg_ep_desc(common->gadget,
			&fsg_fs_bulk_out_desc, &fsg_hs_bulk_out_desc);
	rc = enable_endpoint(common, fsg->bulk_out, d);
	if (rc)
		goto reset;
	fsg->bulk_out_enabled = 1;
	common->bulk_out_maxpacket = le16_to_cpu(d->wMaxPacketSize);
	clear_bit(IGNORE_BULK_OUT, &fsg->atomic_bitflags);

	/* Allocate the requests */
	for (i = 0; i < FSG_NUM_BUFFERS; ++i) {
		struct fsg_buffhd	*bh = &common->buffhds[i];

		rc = alloc_request(common, fsg->bulk_in, &bh->inreq);
2446
		if (rc)
2447
			goto reset;
2448
		rc = alloc_request(common, fsg->bulk_out, &bh->outreq);
2449
		if (rc)
2450
			goto reset;
2451 2452 2453 2454
		bh->inreq->buf = bh->outreq->buf = bh->buf;
		bh->inreq->context = bh->outreq->context = bh;
		bh->inreq->complete = bulk_in_complete;
		bh->outreq->complete = bulk_out_complete;
2455
	}
2456

2457 2458 2459
	common->running = 1;
	for (i = 0; i < common->nluns; ++i)
		common->luns[i].unit_attention_data = SS_RESET_OCCURRED;
2460 2461 2462 2463
	return rc;
}


2464 2465 2466 2467 2468
/****************************** ALT CONFIGS ******************************/

static int fsg_set_alt(struct usb_function *f, unsigned intf, unsigned alt)
{
	struct fsg_dev *fsg = fsg_from_func(f);
2469
	fsg->common->new_fsg = fsg;
2470
	raise_exception(fsg->common, FSG_STATE_CONFIG_CHANGE);
2471 2472 2473 2474 2475 2476
	return 0;
}

static void fsg_disable(struct usb_function *f)
{
	struct fsg_dev *fsg = fsg_from_func(f);
2477
	fsg->common->new_fsg = NULL;
2478
	raise_exception(fsg->common, FSG_STATE_CONFIG_CHANGE);
2479 2480 2481
}


2482 2483
/*-------------------------------------------------------------------------*/

2484
static void handle_exception(struct fsg_common *common)
2485 2486 2487 2488 2489 2490 2491 2492
{
	siginfo_t		info;
	int			i;
	struct fsg_buffhd	*bh;
	enum fsg_state		old_state;
	struct fsg_lun		*curlun;
	unsigned int		exception_req_tag;

2493 2494 2495 2496
	/*
	 * Clear the existing signals.  Anything but SIGUSR1 is converted
	 * into a high-priority EXIT exception.
	 */
2497
	for (;;) {
2498 2499
		int sig =
			dequeue_signal_lock(current, &current->blocked, &info);
2500 2501 2502
		if (!sig)
			break;
		if (sig != SIGUSR1) {
2503 2504 2505
			if (common->state < FSG_STATE_EXIT)
				DBG(common, "Main thread exiting on signal\n");
			raise_exception(common, FSG_STATE_EXIT);
2506 2507 2508 2509
		}
	}

	/* Cancel all the pending transfers */
2510
	if (likely(common->fsg)) {
2511
		for (i = 0; i < FSG_NUM_BUFFERS; ++i) {
2512 2513 2514 2515 2516 2517
			bh = &common->buffhds[i];
			if (bh->inreq_busy)
				usb_ep_dequeue(common->fsg->bulk_in, bh->inreq);
			if (bh->outreq_busy)
				usb_ep_dequeue(common->fsg->bulk_out,
					       bh->outreq);
2518 2519
		}

2520 2521 2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537 2538
		/* Wait until everything is idle */
		for (;;) {
			int num_active = 0;
			for (i = 0; i < FSG_NUM_BUFFERS; ++i) {
				bh = &common->buffhds[i];
				num_active += bh->inreq_busy + bh->outreq_busy;
			}
			if (num_active == 0)
				break;
			if (sleep_thread(common))
				return;
		}

		/* Clear out the controller's fifos */
		if (common->fsg->bulk_in_enabled)
			usb_ep_fifo_flush(common->fsg->bulk_in);
		if (common->fsg->bulk_out_enabled)
			usb_ep_fifo_flush(common->fsg->bulk_out);
	}
2539

2540 2541 2542 2543
	/*
	 * Reset the I/O buffer states and pointers, the SCSI
	 * state, and the exception.  Then invoke the handler.
	 */
2544
	spin_lock_irq(&common->lock);
2545 2546

	for (i = 0; i < FSG_NUM_BUFFERS; ++i) {
2547
		bh = &common->buffhds[i];
2548 2549
		bh->state = BUF_STATE_EMPTY;
	}
2550 2551 2552 2553
	common->next_buffhd_to_fill = &common->buffhds[0];
	common->next_buffhd_to_drain = &common->buffhds[0];
	exception_req_tag = common->exception_req_tag;
	old_state = common->state;
2554 2555

	if (old_state == FSG_STATE_ABORT_BULK_OUT)
2556
		common->state = FSG_STATE_STATUS_PHASE;
2557
	else {
2558 2559
		for (i = 0; i < common->nluns; ++i) {
			curlun = &common->luns[i];
2560
			curlun->prevent_medium_removal = 0;
2561 2562
			curlun->sense_data = SS_NO_SENSE;
			curlun->unit_attention_data = SS_NO_SENSE;
2563 2564 2565
			curlun->sense_data_info = 0;
			curlun->info_valid = 0;
		}
2566
		common->state = FSG_STATE_IDLE;
2567
	}
2568
	spin_unlock_irq(&common->lock);
2569 2570 2571 2572

	/* Carry out any extra actions required for the exception */
	switch (old_state) {
	case FSG_STATE_ABORT_BULK_OUT:
2573 2574 2575 2576 2577
		send_status(common);
		spin_lock_irq(&common->lock);
		if (common->state == FSG_STATE_STATUS_PHASE)
			common->state = FSG_STATE_IDLE;
		spin_unlock_irq(&common->lock);
2578 2579 2580
		break;

	case FSG_STATE_RESET:
2581 2582
		/*
		 * In case we were forced against our will to halt a
2583
		 * bulk endpoint, clear the halt now.  (The SuperH UDC
2584 2585
		 * requires this.)
		 */
2586 2587 2588 2589 2590
		if (!fsg_is_set(common))
			break;
		if (test_and_clear_bit(IGNORE_BULK_OUT,
				       &common->fsg->atomic_bitflags))
			usb_ep_clear_halt(common->fsg->bulk_in);
2591

2592 2593
		if (common->ep0_req_tag == exception_req_tag)
			ep0_queue(common);	/* Complete the status stage */
2594

2595 2596
		/*
		 * Technically this should go here, but it would only be
2597
		 * a waste of time.  Ditto for the INTERFACE_CHANGE and
2598 2599
		 * CONFIG_CHANGE cases.
		 */
2600 2601
		/* for (i = 0; i < common->nluns; ++i) */
		/*	common->luns[i].unit_attention_data = */
2602
		/*		SS_RESET_OCCURRED;  */
2603 2604 2605
		break;

	case FSG_STATE_CONFIG_CHANGE:
2606
		do_set_interface(common, common->new_fsg);
2607 2608 2609 2610
		break;

	case FSG_STATE_EXIT:
	case FSG_STATE_TERMINATED:
2611
		do_set_interface(common, NULL);		/* Free resources */
2612 2613 2614
		spin_lock_irq(&common->lock);
		common->state = FSG_STATE_TERMINATED;	/* Stop the thread */
		spin_unlock_irq(&common->lock);
2615
		break;
2616 2617 2618 2619 2620 2621 2622 2623

	case FSG_STATE_INTERFACE_CHANGE:
	case FSG_STATE_DISCONNECT:
	case FSG_STATE_COMMAND_PHASE:
	case FSG_STATE_DATA_PHASE:
	case FSG_STATE_STATUS_PHASE:
	case FSG_STATE_IDLE:
		break;
2624 2625 2626 2627 2628 2629
	}
}


/*-------------------------------------------------------------------------*/

2630
static int fsg_main_thread(void *common_)
2631
{
2632
	struct fsg_common	*common = common_;
2633

2634 2635 2636 2637
	/*
	 * Allow the thread to be killed by a signal, but set the signal mask
	 * to block everything but INT, TERM, KILL, and USR1.
	 */
2638 2639 2640 2641 2642 2643 2644 2645
	allow_signal(SIGINT);
	allow_signal(SIGTERM);
	allow_signal(SIGKILL);
	allow_signal(SIGUSR1);

	/* Allow the thread to be frozen */
	set_freezable();

2646 2647
	/*
	 * Arrange for userspace references to be interpreted as kernel
2648
	 * pointers.  That way we can pass a kernel pointer to a routine
2649 2650
	 * that expects a __user pointer and it will work okay.
	 */
2651 2652 2653
	set_fs(get_ds());

	/* The main loop */
2654 2655 2656
	while (common->state != FSG_STATE_TERMINATED) {
		if (exception_in_progress(common) || signal_pending(current)) {
			handle_exception(common);
2657 2658 2659
			continue;
		}

2660 2661
		if (!common->running) {
			sleep_thread(common);
2662 2663 2664
			continue;
		}

2665
		if (get_next_command(common))
2666 2667
			continue;

2668 2669 2670 2671
		spin_lock_irq(&common->lock);
		if (!exception_in_progress(common))
			common->state = FSG_STATE_DATA_PHASE;
		spin_unlock_irq(&common->lock);
2672

2673
		if (do_scsi_command(common) || finish_reply(common))
2674 2675
			continue;

2676 2677 2678 2679
		spin_lock_irq(&common->lock);
		if (!exception_in_progress(common))
			common->state = FSG_STATE_STATUS_PHASE;
		spin_unlock_irq(&common->lock);
2680

2681
		if (send_status(common))
2682 2683
			continue;

2684 2685 2686 2687
		spin_lock_irq(&common->lock);
		if (!exception_in_progress(common))
			common->state = FSG_STATE_IDLE;
		spin_unlock_irq(&common->lock);
2688
	}
2689

2690 2691 2692
	spin_lock_irq(&common->lock);
	common->thread_task = NULL;
	spin_unlock_irq(&common->lock);
2693

2694 2695
	if (!common->ops || !common->ops->thread_exits
	 || common->ops->thread_exits(common) < 0) {
2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708
		struct fsg_lun *curlun = common->luns;
		unsigned i = common->nluns;

		down_write(&common->filesem);
		for (; i--; ++curlun) {
			if (!fsg_lun_is_open(curlun))
				continue;

			fsg_lun_close(curlun);
			curlun->unit_attention_data = SS_MEDIUM_NOT_PRESENT;
		}
		up_write(&common->filesem);
	}
2709

2710
	/* Let fsg_unbind() know the thread has exited */
2711
	complete_and_exit(&common->thread_notifier, 0);
2712 2713 2714
}


2715
/*************************** DEVICE ATTRIBUTES ***************************/
2716

2717 2718
/* Write permission is checked per LUN in store_*() functions. */
static DEVICE_ATTR(ro, 0644, fsg_show_ro, fsg_store_ro);
2719
static DEVICE_ATTR(nofua, 0644, fsg_show_nofua, fsg_store_nofua);
2720
static DEVICE_ATTR(file, 0644, fsg_show_file, fsg_store_file);
2721 2722


2723 2724 2725
/****************************** FSG COMMON ******************************/

static void fsg_common_release(struct kref *ref);
2726

2727
static void fsg_lun_release(struct device *dev)
2728
{
2729
	/* Nothing needs to be done */
2730 2731
}

2732
static inline void fsg_common_get(struct fsg_common *common)
2733
{
2734
	kref_get(&common->ref);
2735 2736
}

2737 2738 2739 2740 2741 2742
static inline void fsg_common_put(struct fsg_common *common)
{
	kref_put(&common->ref, fsg_common_release);
}

static struct fsg_common *fsg_common_init(struct fsg_common *common,
2743 2744
					  struct usb_composite_dev *cdev,
					  struct fsg_config *cfg)
2745
{
2746
	struct usb_gadget *gadget = cdev->gadget;
2747 2748
	struct fsg_buffhd *bh;
	struct fsg_lun *curlun;
2749
	struct fsg_lun_config *lcfg;
2750
	int nluns, i, rc;
2751
	char *pathbuf;
2752 2753

	/* Find out how many LUNs there should be */
2754
	nluns = cfg->nluns;
2755 2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766
	if (nluns < 1 || nluns > FSG_MAX_LUNS) {
		dev_err(&gadget->dev, "invalid number of LUNs: %u\n", nluns);
		return ERR_PTR(-EINVAL);
	}

	/* Allocate? */
	if (!common) {
		common = kzalloc(sizeof *common, GFP_KERNEL);
		if (!common)
			return ERR_PTR(-ENOMEM);
		common->free_storage_on_release = 1;
	} else {
2767
		memset(common, 0, sizeof *common);
2768 2769
		common->free_storage_on_release = 0;
	}
2770

2771
	common->ops = cfg->ops;
2772 2773
	common->private_data = cfg->private_data;

2774
	common->gadget = gadget;
2775 2776 2777 2778 2779 2780
	common->ep0 = gadget->ep0;
	common->ep0req = cdev->req;

	/* Maybe allocate device-global string IDs, and patch descriptors */
	if (fsg_strings[FSG_STRING_INTERFACE].id == 0) {
		rc = usb_string_id(cdev);
2781 2782
		if (unlikely(rc < 0))
			goto error_release;
2783 2784 2785
		fsg_strings[FSG_STRING_INTERFACE].id = rc;
		fsg_intf_desc.iInterface = rc;
	}
2786

2787 2788 2789 2790
	/*
	 * Create the LUNs, open their backing files, and register the
	 * LUN devices in sysfs.
	 */
2791
	curlun = kzalloc(nluns * sizeof *curlun, GFP_KERNEL);
2792 2793 2794
	if (unlikely(!curlun)) {
		rc = -ENOMEM;
		goto error_release;
2795 2796 2797 2798 2799
	}
	common->luns = curlun;

	init_rwsem(&common->filesem);

2800 2801 2802 2803
	for (i = 0, lcfg = cfg->luns; i < nluns; ++i, ++curlun, ++lcfg) {
		curlun->cdrom = !!lcfg->cdrom;
		curlun->ro = lcfg->cdrom || lcfg->ro;
		curlun->removable = lcfg->removable;
2804 2805
		curlun->dev.release = fsg_lun_release;
		curlun->dev.parent = &gadget->dev;
2806
		/* curlun->dev.driver = &fsg_driver.driver; XXX */
2807
		dev_set_drvdata(&curlun->dev, &common->filesem);
2808 2809 2810 2811 2812
		dev_set_name(&curlun->dev,
			     cfg->lun_name_format
			   ? cfg->lun_name_format
			   : "lun%d",
			     i);
2813 2814 2815 2816 2817

		rc = device_register(&curlun->dev);
		if (rc) {
			INFO(common, "failed to register LUN%d: %d\n", i, rc);
			common->nluns = i;
2818
			put_device(&curlun->dev);
2819 2820 2821 2822 2823 2824 2825
			goto error_release;
		}

		rc = device_create_file(&curlun->dev, &dev_attr_ro);
		if (rc)
			goto error_luns;
		rc = device_create_file(&curlun->dev, &dev_attr_file);
2826 2827 2828
		if (rc)
			goto error_luns;
		rc = device_create_file(&curlun->dev, &dev_attr_nofua);
2829 2830 2831
		if (rc)
			goto error_luns;

2832 2833
		if (lcfg->filename) {
			rc = fsg_lun_open(curlun, lcfg->filename);
2834 2835
			if (rc)
				goto error_luns;
2836
		} else if (!curlun->removable) {
2837 2838 2839 2840 2841 2842 2843 2844 2845
			ERROR(common, "no file given for LUN%d\n", i);
			rc = -EINVAL;
			goto error_luns;
		}
	}
	common->nluns = nluns;

	/* Data buffers cyclic list */
	bh = common->buffhds;
2846 2847
	i = FSG_NUM_BUFFERS;
	goto buffhds_first_it;
2848 2849
	do {
		bh->next = bh + 1;
2850 2851 2852 2853 2854 2855 2856 2857
		++bh;
buffhds_first_it:
		bh->buf = kmalloc(FSG_BUFLEN, GFP_KERNEL);
		if (unlikely(!bh->buf)) {
			rc = -ENOMEM;
			goto error_release;
		}
	} while (--i);
2858 2859
	bh->next = common->buffhds;

2860 2861 2862 2863
	/* Prepare inquiryString */
	if (cfg->release != 0xffff) {
		i = cfg->release;
	} else {
2864
		i = usb_gadget_controller_number(gadget);
2865 2866 2867
		if (i >= 0) {
			i = 0x0300 + i;
		} else {
2868 2869
			WARNING(common, "controller '%s' not recognized\n",
				gadget->name);
2870
			i = 0x0399;
2871 2872
		}
	}
2873
	snprintf(common->inquiry_string, sizeof common->inquiry_string,
2874
		 "%-8s%-16s%04x", cfg->vendor_name ?: "Linux",
2875
		 /* Assume product name dependent on the first LUN */
2876
		 cfg->product_name ?: (common->luns->cdrom
2877
				     ? "File-Stor Gadget"
2878
				     : "File-CD Gadget"),
2879
		 i);
2880

2881 2882
	/*
	 * Some peripheral controllers are known not to be able to
2883 2884 2885
	 * halt bulk endpoints correctly.  If one of them is present,
	 * disable stalls.
	 */
2886
	common->can_stall = cfg->can_stall &&
2887
		!(gadget_is_at91(common->gadget));
2888

2889
	spin_lock_init(&common->lock);
2890
	kref_init(&common->ref);
2891 2892 2893 2894

	/* Tell the thread to start working */
	common->thread_task =
		kthread_create(fsg_main_thread, common,
2895
			       cfg->thread_name ?: "file-storage");
2896 2897 2898 2899 2900
	if (IS_ERR(common->thread_task)) {
		rc = PTR_ERR(common->thread_task);
		goto error_release;
	}
	init_completion(&common->thread_notifier);
2901
	init_waitqueue_head(&common->fsg_wait);
2902

2903 2904 2905 2906 2907 2908 2909 2910 2911 2912 2913 2914 2915 2916 2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928
	/* Information */
	INFO(common, FSG_DRIVER_DESC ", version: " FSG_DRIVER_VERSION "\n");
	INFO(common, "Number of LUNs=%d\n", common->nluns);

	pathbuf = kmalloc(PATH_MAX, GFP_KERNEL);
	for (i = 0, nluns = common->nluns, curlun = common->luns;
	     i < nluns;
	     ++curlun, ++i) {
		char *p = "(no medium)";
		if (fsg_lun_is_open(curlun)) {
			p = "(error)";
			if (pathbuf) {
				p = d_path(&curlun->filp->f_path,
					   pathbuf, PATH_MAX);
				if (IS_ERR(p))
					p = "(error)";
			}
		}
		LINFO(curlun, "LUN: %s%s%sfile: %s\n",
		      curlun->removable ? "removable " : "",
		      curlun->ro ? "read only " : "",
		      curlun->cdrom ? "CD-ROM " : "",
		      p);
	}
	kfree(pathbuf);

2929 2930 2931 2932
	DBG(common, "I/O thread pid: %d\n", task_pid_nr(common->thread_task));

	wake_up_process(common->thread_task);

2933 2934 2935 2936 2937
	return common;

error_luns:
	common->nluns = i + 1;
error_release:
2938
	common->state = FSG_STATE_TERMINATED;	/* The thread is dead */
2939
	/* Call fsg_common_release() directly, ref might be not initialised. */
2940 2941 2942 2943 2944 2945
	fsg_common_release(&common->ref);
	return ERR_PTR(rc);
}

static void fsg_common_release(struct kref *ref)
{
2946
	struct fsg_common *common = container_of(ref, struct fsg_common, ref);
2947

2948 2949 2950 2951 2952 2953
	/* If the thread isn't already dead, tell it to exit now */
	if (common->state != FSG_STATE_TERMINATED) {
		raise_exception(common, FSG_STATE_EXIT);
		wait_for_completion(&common->thread_notifier);
	}

2954 2955 2956 2957 2958 2959
	if (likely(common->luns)) {
		struct fsg_lun *lun = common->luns;
		unsigned i = common->nluns;

		/* In error recovery common->nluns may be zero. */
		for (; i; --i, ++lun) {
2960
			device_remove_file(&lun->dev, &dev_attr_nofua);
2961 2962 2963 2964 2965
			device_remove_file(&lun->dev, &dev_attr_ro);
			device_remove_file(&lun->dev, &dev_attr_file);
			fsg_lun_close(lun);
			device_unregister(&lun->dev);
		}
2966

2967
		kfree(common->luns);
2968 2969
	}

2970 2971 2972 2973 2974 2975 2976
	{
		struct fsg_buffhd *bh = common->buffhds;
		unsigned i = FSG_NUM_BUFFERS;
		do {
			kfree(bh->buf);
		} while (++bh, --i);
	}
2977

2978 2979 2980 2981 2982 2983 2984
	if (common->free_storage_on_release)
		kfree(common);
}


/*-------------------------------------------------------------------------*/

2985
static void fsg_unbind(struct usb_configuration *c, struct usb_function *f)
2986
{
2987
	struct fsg_dev		*fsg = fsg_from_func(f);
2988
	struct fsg_common	*common = fsg->common;
2989 2990

	DBG(fsg, "unbind\n");
2991 2992 2993 2994 2995 2996 2997 2998
	if (fsg->common->fsg == fsg) {
		fsg->common->new_fsg = NULL;
		raise_exception(fsg->common, FSG_STATE_CONFIG_CHANGE);
		/* FIXME: make interruptible or killable somehow? */
		wait_event(common->fsg_wait, common->fsg != fsg);
	}

	fsg_common_put(common);
2999 3000
	usb_free_descriptors(fsg->function.descriptors);
	usb_free_descriptors(fsg->function.hs_descriptors);
3001
	kfree(fsg);
3002 3003
}

3004
static int fsg_bind(struct usb_configuration *c, struct usb_function *f)
3005
{
3006 3007
	struct fsg_dev		*fsg = fsg_from_func(f);
	struct usb_gadget	*gadget = c->cdev->gadget;
3008 3009 3010 3011 3012
	int			i;
	struct usb_ep		*ep;

	fsg->gadget = gadget;

3013 3014 3015 3016 3017 3018
	/* New interface */
	i = usb_interface_id(c, f);
	if (i < 0)
		return i;
	fsg_intf_desc.bInterfaceNumber = i;
	fsg->interface_number = i;
3019 3020 3021 3022 3023

	/* Find all the endpoints we will use */
	ep = usb_ep_autoconfig(gadget, &fsg_fs_bulk_in_desc);
	if (!ep)
		goto autoconf_fail;
3024
	ep->driver_data = fsg->common;	/* claim the endpoint */
3025 3026 3027 3028 3029
	fsg->bulk_in = ep;

	ep = usb_ep_autoconfig(gadget, &fsg_fs_bulk_out_desc);
	if (!ep)
		goto autoconf_fail;
3030
	ep->driver_data = fsg->common;	/* claim the endpoint */
3031 3032
	fsg->bulk_out = ep;

3033 3034 3035 3036 3037
	/* Copy descriptors */
	f->descriptors = usb_copy_descriptors(fsg_fs_function);
	if (unlikely(!f->descriptors))
		return -ENOMEM;

3038 3039 3040 3041 3042 3043
	if (gadget_is_dualspeed(gadget)) {
		/* Assume endpoint addresses are the same for both speeds */
		fsg_hs_bulk_in_desc.bEndpointAddress =
			fsg_fs_bulk_in_desc.bEndpointAddress;
		fsg_hs_bulk_out_desc.bEndpointAddress =
			fsg_fs_bulk_out_desc.bEndpointAddress;
3044
		f->hs_descriptors = usb_copy_descriptors(fsg_hs_function);
3045 3046
		if (unlikely(!f->hs_descriptors)) {
			usb_free_descriptors(f->descriptors);
3047
			return -ENOMEM;
3048
		}
3049 3050 3051 3052 3053 3054
	}

	return 0;

autoconf_fail:
	ERROR(fsg, "unable to autoconfigure all endpoints\n");
3055
	return -ENOTSUPP;
3056 3057 3058
}


3059
/****************************** ADD FUNCTION ******************************/
3060

3061 3062 3063
static struct usb_gadget_strings *fsg_strings_array[] = {
	&fsg_stringtab,
	NULL,
3064 3065
};

3066 3067 3068
static int fsg_bind_config(struct usb_composite_dev *cdev,
			   struct usb_configuration *c,
			   struct fsg_common *common)
3069
{
3070 3071 3072 3073 3074 3075
	struct fsg_dev *fsg;
	int rc;

	fsg = kzalloc(sizeof *fsg, GFP_KERNEL);
	if (unlikely(!fsg))
		return -ENOMEM;
3076

3077 3078 3079 3080 3081 3082 3083 3084 3085
	fsg->function.name        = FSG_DRIVER_DESC;
	fsg->function.strings     = fsg_strings_array;
	fsg->function.bind        = fsg_bind;
	fsg->function.unbind      = fsg_unbind;
	fsg->function.setup       = fsg_setup;
	fsg->function.set_alt     = fsg_set_alt;
	fsg->function.disable     = fsg_disable;

	fsg->common               = common;
3086 3087
	/*
	 * Our caller holds a reference to common structure so we
3088 3089 3090
	 * don't have to be worry about it being freed until we return
	 * from this function.  So instead of incrementing counter now
	 * and decrement in error recovery we increment it only when
3091 3092
	 * call to usb_add_function() was successful.
	 */
3093 3094

	rc = usb_add_function(c, &fsg->function);
3095
	if (unlikely(rc))
3096 3097 3098
		kfree(fsg);
	else
		fsg_common_get(fsg->common);
3099
	return rc;
3100
}
3101

3102
static inline int __deprecated __maybe_unused
3103
fsg_add(struct usb_composite_dev *cdev, struct usb_configuration *c,
3104 3105 3106 3107
	struct fsg_common *common)
{
	return fsg_bind_config(cdev, c, common);
}
3108 3109 3110 3111 3112 3113 3114 3115 3116


/************************* Module parameters *************************/

struct fsg_module_parameters {
	char		*file[FSG_MAX_LUNS];
	int		ro[FSG_MAX_LUNS];
	int		removable[FSG_MAX_LUNS];
	int		cdrom[FSG_MAX_LUNS];
3117
	int		nofua[FSG_MAX_LUNS];
3118 3119

	unsigned int	file_count, ro_count, removable_count, cdrom_count;
3120
	unsigned int	nofua_count;
3121 3122 3123 3124 3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135 3136 3137 3138 3139 3140 3141 3142 3143 3144
	unsigned int	luns;	/* nluns */
	int		stall;	/* can_stall */
};

#define _FSG_MODULE_PARAM_ARRAY(prefix, params, name, type, desc)	\
	module_param_array_named(prefix ## name, params.name, type,	\
				 &prefix ## params.name ## _count,	\
				 S_IRUGO);				\
	MODULE_PARM_DESC(prefix ## name, desc)

#define _FSG_MODULE_PARAM(prefix, params, name, type, desc)		\
	module_param_named(prefix ## name, params.name, type,		\
			   S_IRUGO);					\
	MODULE_PARM_DESC(prefix ## name, desc)

#define FSG_MODULE_PARAMETERS(prefix, params)				\
	_FSG_MODULE_PARAM_ARRAY(prefix, params, file, charp,		\
				"names of backing files or devices");	\
	_FSG_MODULE_PARAM_ARRAY(prefix, params, ro, bool,		\
				"true to force read-only");		\
	_FSG_MODULE_PARAM_ARRAY(prefix, params, removable, bool,	\
				"true to simulate removable media");	\
	_FSG_MODULE_PARAM_ARRAY(prefix, params, cdrom, bool,		\
				"true to simulate CD-ROM instead of disk"); \
3145 3146
	_FSG_MODULE_PARAM_ARRAY(prefix, params, nofua, bool,		\
				"true to ignore SCSI WRITE(10,12) FUA bit"); \
3147 3148 3149 3150 3151 3152 3153 3154 3155 3156
	_FSG_MODULE_PARAM(prefix, params, luns, uint,			\
			  "number of LUNs");				\
	_FSG_MODULE_PARAM(prefix, params, stall, bool,			\
			  "false to prevent bulk stalls")

static void
fsg_config_from_params(struct fsg_config *cfg,
		       const struct fsg_module_parameters *params)
{
	struct fsg_lun_config *lun;
3157
	unsigned i;
3158 3159

	/* Configure LUNs */
3160 3161 3162 3163
	cfg->nluns =
		min(params->luns ?: (params->file_count ?: 1u),
		    (unsigned)FSG_MAX_LUNS);
	for (i = 0, lun = cfg->luns; i < cfg->nluns; ++i, ++lun) {
3164 3165
		lun->ro = !!params->ro[i];
		lun->cdrom = !!params->cdrom[i];
3166
		lun->removable = /* Removable by default */
3167 3168 3169 3170 3171 3172 3173
			params->removable_count <= i || params->removable[i];
		lun->filename =
			params->file_count > i && params->file[i][0]
			? params->file[i]
			: 0;
	}

3174
	/* Let MSF use defaults */
3175 3176
	cfg->lun_name_format = 0;
	cfg->thread_name = 0;
3177 3178 3179 3180
	cfg->vendor_name = 0;
	cfg->product_name = 0;
	cfg->release = 0xffff;

3181 3182
	cfg->ops = NULL;
	cfg->private_data = NULL;
3183

3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195 3196 3197 3198 3199 3200 3201 3202
	/* Finalise */
	cfg->can_stall = params->stall;
}

static inline struct fsg_common *
fsg_common_from_params(struct fsg_common *common,
		       struct usb_composite_dev *cdev,
		       const struct fsg_module_parameters *params)
	__attribute__((unused));
static inline struct fsg_common *
fsg_common_from_params(struct fsg_common *common,
		       struct usb_composite_dev *cdev,
		       const struct fsg_module_parameters *params)
{
	struct fsg_config cfg;
	fsg_config_from_params(&cfg, params);
	return fsg_common_init(common, cdev, &cfg);
}