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