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