Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 1 | /* |
| 2 | * f_mass_storage.c -- Mass Storage USB Composite Function |
| 3 | * |
| 4 | * Copyright (C) 2003-2008 Alan Stern |
| 5 | * Copyright (C) 2009 Samsung Electronics |
| 6 | * Author: Michal Nazarewicz <m.nazarewicz@samsung.com> |
| 7 | * All rights reserved. |
| 8 | * |
Wolfgang Denk | c57eadc | 2013-07-28 22:12:47 +0200 | [diff] [blame] | 9 | * SPDX-License-Identifier: GPL-2.0+ BSD-3-Clause |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 10 | */ |
| 11 | |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 12 | /* |
| 13 | * The Mass Storage Function acts as a USB Mass Storage device, |
| 14 | * appearing to the host as a disk drive or as a CD-ROM drive. In |
| 15 | * addition to providing an example of a genuinely useful composite |
| 16 | * function for a USB device, it also illustrates a technique of |
| 17 | * double-buffering for increased throughput. |
| 18 | * |
| 19 | * Function supports multiple logical units (LUNs). Backing storage |
| 20 | * for each LUN is provided by a regular file or a block device. |
| 21 | * Access for each LUN can be limited to read-only. Moreover, the |
| 22 | * function can indicate that LUN is removable and/or CD-ROM. (The |
| 23 | * later implies read-only access.) |
| 24 | * |
| 25 | * MSF is configured by specifying a fsg_config structure. It has the |
| 26 | * following fields: |
| 27 | * |
| 28 | * nluns Number of LUNs function have (anywhere from 1 |
| 29 | * to FSG_MAX_LUNS which is 8). |
| 30 | * luns An array of LUN configuration values. This |
| 31 | * should be filled for each LUN that |
| 32 | * function will include (ie. for "nluns" |
| 33 | * LUNs). Each element of the array has |
| 34 | * the following fields: |
| 35 | * ->filename The path to the backing file for the LUN. |
| 36 | * Required if LUN is not marked as |
| 37 | * removable. |
| 38 | * ->ro Flag specifying access to the LUN shall be |
| 39 | * read-only. This is implied if CD-ROM |
| 40 | * emulation is enabled as well as when |
| 41 | * it was impossible to open "filename" |
| 42 | * in R/W mode. |
| 43 | * ->removable Flag specifying that LUN shall be indicated as |
| 44 | * being removable. |
| 45 | * ->cdrom Flag specifying that LUN shall be reported as |
| 46 | * being a CD-ROM. |
| 47 | * |
| 48 | * lun_name_format A printf-like format for names of the LUN |
| 49 | * devices. This determines how the |
| 50 | * directory in sysfs will be named. |
| 51 | * Unless you are using several MSFs in |
| 52 | * a single gadget (as opposed to single |
| 53 | * MSF in many configurations) you may |
| 54 | * leave it as NULL (in which case |
| 55 | * "lun%d" will be used). In the format |
| 56 | * you can use "%d" to index LUNs for |
| 57 | * MSF's with more than one LUN. (Beware |
| 58 | * that there is only one integer given |
| 59 | * as an argument for the format and |
| 60 | * specifying invalid format may cause |
| 61 | * unspecified behaviour.) |
| 62 | * thread_name Name of the kernel thread process used by the |
| 63 | * MSF. You can safely set it to NULL |
| 64 | * (in which case default "file-storage" |
| 65 | * will be used). |
| 66 | * |
| 67 | * vendor_name |
| 68 | * product_name |
| 69 | * release Information used as a reply to INQUIRY |
| 70 | * request. To use default set to NULL, |
| 71 | * NULL, 0xffff respectively. The first |
| 72 | * field should be 8 and the second 16 |
| 73 | * characters or less. |
| 74 | * |
| 75 | * can_stall Set to permit function to halt bulk endpoints. |
| 76 | * Disabled on some USB devices known not |
| 77 | * to work correctly. You should set it |
| 78 | * to true. |
| 79 | * |
| 80 | * If "removable" is not set for a LUN then a backing file must be |
| 81 | * specified. If it is set, then NULL filename means the LUN's medium |
| 82 | * is not loaded (an empty string as "filename" in the fsg_config |
| 83 | * structure causes error). The CD-ROM emulation includes a single |
| 84 | * data track and no audio tracks; hence there need be only one |
| 85 | * backing file per LUN. Note also that the CD-ROM block length is |
| 86 | * set to 512 rather than the more common value 2048. |
| 87 | * |
| 88 | * |
| 89 | * MSF includes support for module parameters. If gadget using it |
| 90 | * decides to use it, the following module parameters will be |
| 91 | * available: |
| 92 | * |
| 93 | * file=filename[,filename...] |
| 94 | * Names of the files or block devices used for |
| 95 | * backing storage. |
| 96 | * ro=b[,b...] Default false, boolean for read-only access. |
| 97 | * removable=b[,b...] |
| 98 | * Default true, boolean for removable media. |
| 99 | * cdrom=b[,b...] Default false, boolean for whether to emulate |
| 100 | * a CD-ROM drive. |
| 101 | * luns=N Default N = number of filenames, number of |
| 102 | * LUNs to support. |
| 103 | * stall Default determined according to the type of |
| 104 | * USB device controller (usually true), |
| 105 | * boolean to permit the driver to halt |
| 106 | * bulk endpoints. |
| 107 | * |
| 108 | * The module parameters may be prefixed with some string. You need |
| 109 | * to consult gadget's documentation or source to verify whether it is |
| 110 | * using those module parameters and if it does what are the prefixes |
| 111 | * (look for FSG_MODULE_PARAMETERS() macro usage, what's inside it is |
| 112 | * the prefix). |
| 113 | * |
| 114 | * |
| 115 | * Requirements are modest; only a bulk-in and a bulk-out endpoint are |
| 116 | * needed. The memory requirement amounts to two 16K buffers, size |
| 117 | * configurable by a parameter. Support is included for both |
| 118 | * full-speed and high-speed operation. |
| 119 | * |
| 120 | * Note that the driver is slightly non-portable in that it assumes a |
| 121 | * single memory/DMA buffer will be useable for bulk-in, bulk-out, and |
| 122 | * interrupt-in endpoints. With most device controllers this isn't an |
| 123 | * issue, but there may be some with hardware restrictions that prevent |
| 124 | * a buffer from being used by more than one endpoint. |
| 125 | * |
| 126 | * |
| 127 | * The pathnames of the backing files and the ro settings are |
| 128 | * available in the attribute files "file" and "ro" in the lun<n> (or |
| 129 | * to be more precise in a directory which name comes from |
| 130 | * "lun_name_format" option!) subdirectory of the gadget's sysfs |
| 131 | * directory. If the "removable" option is set, writing to these |
| 132 | * files will simulate ejecting/loading the medium (writing an empty |
| 133 | * line means eject) and adjusting a write-enable tab. Changes to the |
| 134 | * ro setting are not allowed when the medium is loaded or if CD-ROM |
| 135 | * emulation is being used. |
| 136 | * |
| 137 | * When a LUN receive an "eject" SCSI request (Start/Stop Unit), |
| 138 | * if the LUN is removable, the backing file is released to simulate |
| 139 | * ejection. |
| 140 | * |
| 141 | * |
| 142 | * This function is heavily based on "File-backed Storage Gadget" by |
| 143 | * Alan Stern which in turn is heavily based on "Gadget Zero" by David |
| 144 | * Brownell. The driver's SCSI command interface was based on the |
| 145 | * "Information technology - Small Computer System Interface - 2" |
| 146 | * document from X3T9.2 Project 375D, Revision 10L, 7-SEP-93, |
| 147 | * available at <http://www.t10.org/ftp/t10/drafts/s2/s2-r10l.pdf>. |
| 148 | * The single exception is opcode 0x23 (READ FORMAT CAPACITIES), which |
| 149 | * was based on the "Universal Serial Bus Mass Storage Class UFI |
| 150 | * Command Specification" document, Revision 1.0, December 14, 1998, |
| 151 | * available at |
| 152 | * <http://www.usb.org/developers/devclass_docs/usbmass-ufi10.pdf>. |
| 153 | */ |
| 154 | |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 155 | /* |
| 156 | * Driver Design |
| 157 | * |
| 158 | * The MSF is fairly straightforward. There is a main kernel |
| 159 | * thread that handles most of the work. Interrupt routines field |
| 160 | * callbacks from the controller driver: bulk- and interrupt-request |
| 161 | * completion notifications, endpoint-0 events, and disconnect events. |
| 162 | * Completion events are passed to the main thread by wakeup calls. Many |
| 163 | * ep0 requests are handled at interrupt time, but SetInterface, |
| 164 | * SetConfiguration, and device reset requests are forwarded to the |
| 165 | * thread in the form of "exceptions" using SIGUSR1 signals (since they |
| 166 | * should interrupt any ongoing file I/O operations). |
| 167 | * |
| 168 | * The thread's main routine implements the standard command/data/status |
| 169 | * parts of a SCSI interaction. It and its subroutines are full of tests |
| 170 | * for pending signals/exceptions -- all this polling is necessary since |
| 171 | * the kernel has no setjmp/longjmp equivalents. (Maybe this is an |
| 172 | * indication that the driver really wants to be running in userspace.) |
| 173 | * An important point is that so long as the thread is alive it keeps an |
| 174 | * open reference to the backing file. This will prevent unmounting |
| 175 | * the backing file's underlying filesystem and could cause problems |
| 176 | * during system shutdown, for example. To prevent such problems, the |
| 177 | * thread catches INT, TERM, and KILL signals and converts them into |
| 178 | * an EXIT exception. |
| 179 | * |
| 180 | * In normal operation the main thread is started during the gadget's |
| 181 | * fsg_bind() callback and stopped during fsg_unbind(). But it can |
| 182 | * also exit when it receives a signal, and there's no point leaving |
| 183 | * the gadget running when the thread is dead. At of this moment, MSF |
| 184 | * provides no way to deregister the gadget when thread dies -- maybe |
| 185 | * a callback functions is needed. |
| 186 | * |
| 187 | * To provide maximum throughput, the driver uses a circular pipeline of |
| 188 | * buffer heads (struct fsg_buffhd). In principle the pipeline can be |
| 189 | * arbitrarily long; in practice the benefits don't justify having more |
| 190 | * than 2 stages (i.e., double buffering). But it helps to think of the |
| 191 | * pipeline as being a long one. Each buffer head contains a bulk-in and |
| 192 | * a bulk-out request pointer (since the buffer can be used for both |
| 193 | * output and input -- directions always are given from the host's |
| 194 | * point of view) as well as a pointer to the buffer and various state |
| 195 | * variables. |
| 196 | * |
| 197 | * Use of the pipeline follows a simple protocol. There is a variable |
| 198 | * (fsg->next_buffhd_to_fill) that points to the next buffer head to use. |
| 199 | * At any time that buffer head may still be in use from an earlier |
| 200 | * request, so each buffer head has a state variable indicating whether |
| 201 | * it is EMPTY, FULL, or BUSY. Typical use involves waiting for the |
| 202 | * buffer head to be EMPTY, filling the buffer either by file I/O or by |
| 203 | * USB I/O (during which the buffer head is BUSY), and marking the buffer |
| 204 | * head FULL when the I/O is complete. Then the buffer will be emptied |
| 205 | * (again possibly by USB I/O, during which it is marked BUSY) and |
| 206 | * finally marked EMPTY again (possibly by a completion routine). |
| 207 | * |
| 208 | * A module parameter tells the driver to avoid stalling the bulk |
| 209 | * endpoints wherever the transport specification allows. This is |
| 210 | * necessary for some UDCs like the SuperH, which cannot reliably clear a |
| 211 | * halt on a bulk endpoint. However, under certain circumstances the |
| 212 | * Bulk-only specification requires a stall. In such cases the driver |
| 213 | * will halt the endpoint and set a flag indicating that it should clear |
| 214 | * the halt in software during the next device reset. Hopefully this |
| 215 | * will permit everything to work correctly. Furthermore, although the |
| 216 | * specification allows the bulk-out endpoint to halt when the host sends |
| 217 | * too much data, implementing this would cause an unavoidable race. |
| 218 | * The driver will always use the "no-stall" approach for OUT transfers. |
| 219 | * |
| 220 | * One subtle point concerns sending status-stage responses for ep0 |
| 221 | * requests. Some of these requests, such as device reset, can involve |
| 222 | * interrupting an ongoing file I/O operation, which might take an |
| 223 | * arbitrarily long time. During that delay the host might give up on |
| 224 | * the original ep0 request and issue a new one. When that happens the |
| 225 | * driver should not notify the host about completion of the original |
| 226 | * request, as the host will no longer be waiting for it. So the driver |
| 227 | * assigns to each ep0 request a unique tag, and it keeps track of the |
| 228 | * tag value of the request associated with a long-running exception |
| 229 | * (device-reset, interface-change, or configuration-change). When the |
| 230 | * exception handler is finished, the status-stage response is submitted |
| 231 | * only if the current ep0 request tag is equal to the exception request |
| 232 | * tag. Thus only the most recently received ep0 request will get a |
| 233 | * status-stage response. |
| 234 | * |
| 235 | * Warning: This driver source file is too long. It ought to be split up |
| 236 | * into a header file plus about 3 separate .c files, to handle the details |
| 237 | * of the Gadget, USB Mass Storage, and SCSI protocols. |
| 238 | */ |
| 239 | |
| 240 | /* #define VERBOSE_DEBUG */ |
| 241 | /* #define DUMP_MSGS */ |
| 242 | |
| 243 | #include <config.h> |
| 244 | #include <malloc.h> |
| 245 | #include <common.h> |
Simon Glass | a73bda4 | 2015-11-08 23:47:45 -0700 | [diff] [blame] | 246 | #include <console.h> |
Mateusz Zalega | 21fe3f7 | 2014-04-30 13:07:48 +0200 | [diff] [blame] | 247 | #include <g_dnl.h> |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 248 | |
| 249 | #include <linux/err.h> |
| 250 | #include <linux/usb/ch9.h> |
| 251 | #include <linux/usb/gadget.h> |
| 252 | #include <usb_mass_storage.h> |
| 253 | |
| 254 | #include <asm/unaligned.h> |
| 255 | #include <linux/usb/gadget.h> |
| 256 | #include <linux/usb/gadget.h> |
| 257 | #include <linux/usb/composite.h> |
| 258 | #include <usb/lin_gadget_compat.h> |
Mateusz Zalega | 69cb0bb | 2014-04-28 21:13:28 +0200 | [diff] [blame] | 259 | #include <g_dnl.h> |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 260 | |
| 261 | /*------------------------------------------------------------------------*/ |
| 262 | |
| 263 | #define FSG_DRIVER_DESC "Mass Storage Function" |
| 264 | #define FSG_DRIVER_VERSION "2012/06/5" |
| 265 | |
| 266 | static const char fsg_string_interface[] = "Mass Storage"; |
| 267 | |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 268 | #define FSG_NO_INTR_EP 1 |
| 269 | #define FSG_NO_DEVICE_STRINGS 1 |
| 270 | #define FSG_NO_OTG 1 |
| 271 | #define FSG_NO_INTR_EP 1 |
| 272 | |
| 273 | #include "storage_common.c" |
| 274 | |
| 275 | /*-------------------------------------------------------------------------*/ |
| 276 | |
| 277 | #define GFP_ATOMIC ((gfp_t) 0) |
| 278 | #define PAGE_CACHE_SHIFT 12 |
| 279 | #define PAGE_CACHE_SIZE (1 << PAGE_CACHE_SHIFT) |
| 280 | #define kthread_create(...) __builtin_return_address(0) |
| 281 | #define wait_for_completion(...) do {} while (0) |
| 282 | |
| 283 | struct kref {int x; }; |
| 284 | struct completion {int x; }; |
| 285 | |
| 286 | inline void set_bit(int nr, volatile void *addr) |
| 287 | { |
| 288 | int mask; |
| 289 | unsigned int *a = (unsigned int *) addr; |
| 290 | |
| 291 | a += nr >> 5; |
| 292 | mask = 1 << (nr & 0x1f); |
| 293 | *a |= mask; |
| 294 | } |
| 295 | |
| 296 | inline void clear_bit(int nr, volatile void *addr) |
| 297 | { |
| 298 | int mask; |
| 299 | unsigned int *a = (unsigned int *) addr; |
| 300 | |
| 301 | a += nr >> 5; |
| 302 | mask = 1 << (nr & 0x1f); |
| 303 | *a &= ~mask; |
| 304 | } |
| 305 | |
| 306 | struct fsg_dev; |
| 307 | struct fsg_common; |
| 308 | |
| 309 | /* Data shared by all the FSG instances. */ |
| 310 | struct fsg_common { |
| 311 | struct usb_gadget *gadget; |
| 312 | struct fsg_dev *fsg, *new_fsg; |
| 313 | |
| 314 | struct usb_ep *ep0; /* Copy of gadget->ep0 */ |
| 315 | struct usb_request *ep0req; /* Copy of cdev->req */ |
| 316 | unsigned int ep0_req_tag; |
| 317 | |
| 318 | struct fsg_buffhd *next_buffhd_to_fill; |
| 319 | struct fsg_buffhd *next_buffhd_to_drain; |
| 320 | struct fsg_buffhd buffhds[FSG_NUM_BUFFERS]; |
| 321 | |
| 322 | int cmnd_size; |
| 323 | u8 cmnd[MAX_COMMAND_SIZE]; |
| 324 | |
| 325 | unsigned int nluns; |
| 326 | unsigned int lun; |
| 327 | struct fsg_lun luns[FSG_MAX_LUNS]; |
| 328 | |
| 329 | unsigned int bulk_out_maxpacket; |
| 330 | enum fsg_state state; /* For exception handling */ |
| 331 | unsigned int exception_req_tag; |
| 332 | |
| 333 | enum data_direction data_dir; |
| 334 | u32 data_size; |
| 335 | u32 data_size_from_cmnd; |
| 336 | u32 tag; |
| 337 | u32 residue; |
| 338 | u32 usb_amount_left; |
| 339 | |
| 340 | unsigned int can_stall:1; |
| 341 | unsigned int free_storage_on_release:1; |
| 342 | unsigned int phase_error:1; |
| 343 | unsigned int short_packet_received:1; |
| 344 | unsigned int bad_lun_okay:1; |
| 345 | unsigned int running:1; |
| 346 | |
| 347 | int thread_wakeup_needed; |
| 348 | struct completion thread_notifier; |
| 349 | struct task_struct *thread_task; |
| 350 | |
| 351 | /* Callback functions. */ |
| 352 | const struct fsg_operations *ops; |
| 353 | /* Gadget's private data. */ |
| 354 | void *private_data; |
| 355 | |
| 356 | const char *vendor_name; /* 8 characters or less */ |
| 357 | const char *product_name; /* 16 characters or less */ |
| 358 | u16 release; |
| 359 | |
| 360 | /* Vendor (8 chars), product (16 chars), release (4 |
| 361 | * hexadecimal digits) and NUL byte */ |
| 362 | char inquiry_string[8 + 16 + 4 + 1]; |
| 363 | |
| 364 | struct kref ref; |
| 365 | }; |
| 366 | |
| 367 | struct fsg_config { |
| 368 | unsigned nluns; |
| 369 | struct fsg_lun_config { |
| 370 | const char *filename; |
| 371 | char ro; |
| 372 | char removable; |
| 373 | char cdrom; |
| 374 | char nofua; |
| 375 | } luns[FSG_MAX_LUNS]; |
| 376 | |
| 377 | /* Callback functions. */ |
| 378 | const struct fsg_operations *ops; |
| 379 | /* Gadget's private data. */ |
| 380 | void *private_data; |
| 381 | |
| 382 | const char *vendor_name; /* 8 characters or less */ |
| 383 | const char *product_name; /* 16 characters or less */ |
| 384 | |
| 385 | char can_stall; |
| 386 | }; |
| 387 | |
| 388 | struct fsg_dev { |
| 389 | struct usb_function function; |
| 390 | struct usb_gadget *gadget; /* Copy of cdev->gadget */ |
| 391 | struct fsg_common *common; |
| 392 | |
| 393 | u16 interface_number; |
| 394 | |
| 395 | unsigned int bulk_in_enabled:1; |
| 396 | unsigned int bulk_out_enabled:1; |
| 397 | |
| 398 | unsigned long atomic_bitflags; |
| 399 | #define IGNORE_BULK_OUT 0 |
| 400 | |
| 401 | struct usb_ep *bulk_in; |
| 402 | struct usb_ep *bulk_out; |
| 403 | }; |
| 404 | |
| 405 | |
| 406 | static inline int __fsg_is_set(struct fsg_common *common, |
| 407 | const char *func, unsigned line) |
| 408 | { |
| 409 | if (common->fsg) |
| 410 | return 1; |
| 411 | ERROR(common, "common->fsg is NULL in %s at %u\n", func, line); |
| 412 | WARN_ON(1); |
| 413 | return 0; |
| 414 | } |
| 415 | |
| 416 | #define fsg_is_set(common) likely(__fsg_is_set(common, __func__, __LINE__)) |
| 417 | |
| 418 | |
| 419 | static inline struct fsg_dev *fsg_from_func(struct usb_function *f) |
| 420 | { |
| 421 | return container_of(f, struct fsg_dev, function); |
| 422 | } |
| 423 | |
| 424 | |
| 425 | typedef void (*fsg_routine_t)(struct fsg_dev *); |
| 426 | |
| 427 | static int exception_in_progress(struct fsg_common *common) |
| 428 | { |
| 429 | return common->state > FSG_STATE_IDLE; |
| 430 | } |
| 431 | |
| 432 | /* Make bulk-out requests be divisible by the maxpacket size */ |
| 433 | static void set_bulk_out_req_length(struct fsg_common *common, |
| 434 | struct fsg_buffhd *bh, unsigned int length) |
| 435 | { |
| 436 | unsigned int rem; |
| 437 | |
| 438 | bh->bulk_out_intended_length = length; |
| 439 | rem = length % common->bulk_out_maxpacket; |
| 440 | if (rem > 0) |
| 441 | length += common->bulk_out_maxpacket - rem; |
| 442 | bh->outreq->length = length; |
| 443 | } |
| 444 | |
| 445 | /*-------------------------------------------------------------------------*/ |
| 446 | |
Stephen Warren | 9e7d588 | 2015-12-07 11:38:50 -0700 | [diff] [blame] | 447 | static struct ums *ums; |
| 448 | static int ums_count; |
| 449 | static struct fsg_common *the_fsg_common; |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 450 | |
| 451 | static int fsg_set_halt(struct fsg_dev *fsg, struct usb_ep *ep) |
| 452 | { |
| 453 | const char *name; |
| 454 | |
| 455 | if (ep == fsg->bulk_in) |
| 456 | name = "bulk-in"; |
| 457 | else if (ep == fsg->bulk_out) |
| 458 | name = "bulk-out"; |
| 459 | else |
| 460 | name = ep->name; |
| 461 | DBG(fsg, "%s set halt\n", name); |
| 462 | return usb_ep_set_halt(ep); |
| 463 | } |
| 464 | |
| 465 | /*-------------------------------------------------------------------------*/ |
| 466 | |
| 467 | /* These routines may be called in process context or in_irq */ |
| 468 | |
| 469 | /* Caller must hold fsg->lock */ |
| 470 | static void wakeup_thread(struct fsg_common *common) |
| 471 | { |
| 472 | common->thread_wakeup_needed = 1; |
| 473 | } |
| 474 | |
| 475 | static void raise_exception(struct fsg_common *common, enum fsg_state new_state) |
| 476 | { |
| 477 | /* Do nothing if a higher-priority exception is already in progress. |
| 478 | * If a lower-or-equal priority exception is in progress, preempt it |
| 479 | * and notify the main thread by sending it a signal. */ |
| 480 | if (common->state <= new_state) { |
| 481 | common->exception_req_tag = common->ep0_req_tag; |
| 482 | common->state = new_state; |
| 483 | common->thread_wakeup_needed = 1; |
| 484 | } |
| 485 | } |
| 486 | |
| 487 | /*-------------------------------------------------------------------------*/ |
| 488 | |
| 489 | static int ep0_queue(struct fsg_common *common) |
| 490 | { |
| 491 | int rc; |
| 492 | |
| 493 | rc = usb_ep_queue(common->ep0, common->ep0req, GFP_ATOMIC); |
| 494 | common->ep0->driver_data = common; |
| 495 | if (rc != 0 && rc != -ESHUTDOWN) { |
| 496 | /* We can't do much more than wait for a reset */ |
| 497 | WARNING(common, "error in submission: %s --> %d\n", |
| 498 | common->ep0->name, rc); |
| 499 | } |
| 500 | return rc; |
| 501 | } |
| 502 | |
| 503 | /*-------------------------------------------------------------------------*/ |
| 504 | |
| 505 | /* Bulk and interrupt endpoint completion handlers. |
| 506 | * These always run in_irq. */ |
| 507 | |
| 508 | static void bulk_in_complete(struct usb_ep *ep, struct usb_request *req) |
| 509 | { |
| 510 | struct fsg_common *common = ep->driver_data; |
| 511 | struct fsg_buffhd *bh = req->context; |
| 512 | |
| 513 | if (req->status || req->actual != req->length) |
| 514 | DBG(common, "%s --> %d, %u/%u\n", __func__, |
| 515 | req->status, req->actual, req->length); |
| 516 | if (req->status == -ECONNRESET) /* Request was cancelled */ |
| 517 | usb_ep_fifo_flush(ep); |
| 518 | |
| 519 | /* Hold the lock while we update the request and buffer states */ |
| 520 | bh->inreq_busy = 0; |
| 521 | bh->state = BUF_STATE_EMPTY; |
| 522 | wakeup_thread(common); |
| 523 | } |
| 524 | |
| 525 | static void bulk_out_complete(struct usb_ep *ep, struct usb_request *req) |
| 526 | { |
| 527 | struct fsg_common *common = ep->driver_data; |
| 528 | struct fsg_buffhd *bh = req->context; |
| 529 | |
| 530 | dump_msg(common, "bulk-out", req->buf, req->actual); |
| 531 | if (req->status || req->actual != bh->bulk_out_intended_length) |
| 532 | DBG(common, "%s --> %d, %u/%u\n", __func__, |
| 533 | req->status, req->actual, |
| 534 | bh->bulk_out_intended_length); |
| 535 | if (req->status == -ECONNRESET) /* Request was cancelled */ |
| 536 | usb_ep_fifo_flush(ep); |
| 537 | |
| 538 | /* Hold the lock while we update the request and buffer states */ |
| 539 | bh->outreq_busy = 0; |
| 540 | bh->state = BUF_STATE_FULL; |
| 541 | wakeup_thread(common); |
| 542 | } |
| 543 | |
| 544 | /*-------------------------------------------------------------------------*/ |
| 545 | |
| 546 | /* Ep0 class-specific handlers. These always run in_irq. */ |
| 547 | |
| 548 | static int fsg_setup(struct usb_function *f, |
| 549 | const struct usb_ctrlrequest *ctrl) |
| 550 | { |
| 551 | struct fsg_dev *fsg = fsg_from_func(f); |
| 552 | struct usb_request *req = fsg->common->ep0req; |
Piotr Wilczek | 2963d80 | 2013-06-26 08:22:05 +0200 | [diff] [blame] | 553 | u16 w_index = get_unaligned_le16(&ctrl->wIndex); |
| 554 | u16 w_value = get_unaligned_le16(&ctrl->wValue); |
| 555 | u16 w_length = get_unaligned_le16(&ctrl->wLength); |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 556 | |
| 557 | if (!fsg_is_set(fsg->common)) |
| 558 | return -EOPNOTSUPP; |
| 559 | |
| 560 | switch (ctrl->bRequest) { |
| 561 | |
| 562 | case USB_BULK_RESET_REQUEST: |
| 563 | if (ctrl->bRequestType != |
| 564 | (USB_DIR_OUT | USB_TYPE_CLASS | USB_RECIP_INTERFACE)) |
| 565 | break; |
| 566 | if (w_index != fsg->interface_number || w_value != 0) |
| 567 | return -EDOM; |
| 568 | |
| 569 | /* Raise an exception to stop the current operation |
| 570 | * and reinitialize our state. */ |
| 571 | DBG(fsg, "bulk reset request\n"); |
| 572 | raise_exception(fsg->common, FSG_STATE_RESET); |
| 573 | return DELAYED_STATUS; |
| 574 | |
| 575 | case USB_BULK_GET_MAX_LUN_REQUEST: |
| 576 | if (ctrl->bRequestType != |
| 577 | (USB_DIR_IN | USB_TYPE_CLASS | USB_RECIP_INTERFACE)) |
| 578 | break; |
| 579 | if (w_index != fsg->interface_number || w_value != 0) |
| 580 | return -EDOM; |
| 581 | VDBG(fsg, "get max LUN\n"); |
| 582 | *(u8 *) req->buf = fsg->common->nluns - 1; |
| 583 | |
| 584 | /* Respond with data/status */ |
| 585 | req->length = min((u16)1, w_length); |
| 586 | return ep0_queue(fsg->common); |
| 587 | } |
| 588 | |
| 589 | VDBG(fsg, |
| 590 | "unknown class-specific control req " |
| 591 | "%02x.%02x v%04x i%04x l%u\n", |
| 592 | ctrl->bRequestType, ctrl->bRequest, |
Piotr Wilczek | 2963d80 | 2013-06-26 08:22:05 +0200 | [diff] [blame] | 593 | get_unaligned_le16(&ctrl->wValue), w_index, w_length); |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 594 | return -EOPNOTSUPP; |
| 595 | } |
| 596 | |
| 597 | /*-------------------------------------------------------------------------*/ |
| 598 | |
| 599 | /* All the following routines run in process context */ |
| 600 | |
| 601 | /* Use this for bulk or interrupt transfers, not ep0 */ |
| 602 | static void start_transfer(struct fsg_dev *fsg, struct usb_ep *ep, |
| 603 | struct usb_request *req, int *pbusy, |
| 604 | enum fsg_buffer_state *state) |
| 605 | { |
| 606 | int rc; |
| 607 | |
| 608 | if (ep == fsg->bulk_in) |
| 609 | dump_msg(fsg, "bulk-in", req->buf, req->length); |
| 610 | |
| 611 | *pbusy = 1; |
| 612 | *state = BUF_STATE_BUSY; |
| 613 | rc = usb_ep_queue(ep, req, GFP_KERNEL); |
| 614 | if (rc != 0) { |
| 615 | *pbusy = 0; |
| 616 | *state = BUF_STATE_EMPTY; |
| 617 | |
| 618 | /* We can't do much more than wait for a reset */ |
| 619 | |
| 620 | /* Note: currently the net2280 driver fails zero-length |
| 621 | * submissions if DMA is enabled. */ |
| 622 | if (rc != -ESHUTDOWN && !(rc == -EOPNOTSUPP && |
| 623 | req->length == 0)) |
| 624 | WARNING(fsg, "error in submission: %s --> %d\n", |
| 625 | ep->name, rc); |
| 626 | } |
| 627 | } |
| 628 | |
| 629 | #define START_TRANSFER_OR(common, ep_name, req, pbusy, state) \ |
| 630 | if (fsg_is_set(common)) \ |
| 631 | start_transfer((common)->fsg, (common)->fsg->ep_name, \ |
| 632 | req, pbusy, state); \ |
| 633 | else |
| 634 | |
| 635 | #define START_TRANSFER(common, ep_name, req, pbusy, state) \ |
| 636 | START_TRANSFER_OR(common, ep_name, req, pbusy, state) (void)0 |
| 637 | |
| 638 | static void busy_indicator(void) |
| 639 | { |
| 640 | static int state; |
| 641 | |
| 642 | switch (state) { |
| 643 | case 0: |
| 644 | puts("\r|"); break; |
| 645 | case 1: |
| 646 | puts("\r/"); break; |
| 647 | case 2: |
| 648 | puts("\r-"); break; |
| 649 | case 3: |
| 650 | puts("\r\\"); break; |
| 651 | case 4: |
| 652 | puts("\r|"); break; |
| 653 | case 5: |
| 654 | puts("\r/"); break; |
| 655 | case 6: |
| 656 | puts("\r-"); break; |
| 657 | case 7: |
| 658 | puts("\r\\"); break; |
| 659 | default: |
| 660 | state = 0; |
| 661 | } |
| 662 | if (state++ == 8) |
| 663 | state = 0; |
| 664 | } |
| 665 | |
| 666 | static int sleep_thread(struct fsg_common *common) |
| 667 | { |
| 668 | int rc = 0; |
| 669 | int i = 0, k = 0; |
| 670 | |
| 671 | /* Wait until a signal arrives or we are woken up */ |
| 672 | for (;;) { |
| 673 | if (common->thread_wakeup_needed) |
| 674 | break; |
| 675 | |
Inha Song | f7d9252 | 2015-05-22 18:14:26 +0200 | [diff] [blame] | 676 | if (++i == 20000) { |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 677 | busy_indicator(); |
| 678 | i = 0; |
| 679 | k++; |
| 680 | } |
| 681 | |
Przemyslaw Marczak | 06ef7cc | 2013-10-23 14:30:46 +0200 | [diff] [blame] | 682 | if (k == 10) { |
| 683 | /* Handle CTRL+C */ |
| 684 | if (ctrlc()) |
| 685 | return -EPIPE; |
Mateusz Zalega | 21fe3f7 | 2014-04-30 13:07:48 +0200 | [diff] [blame] | 686 | |
Przemyslaw Marczak | 06ef7cc | 2013-10-23 14:30:46 +0200 | [diff] [blame] | 687 | /* Check cable connection */ |
Mateusz Zalega | 21fe3f7 | 2014-04-30 13:07:48 +0200 | [diff] [blame] | 688 | if (!g_dnl_board_usb_cable_connected()) |
Przemyslaw Marczak | 06ef7cc | 2013-10-23 14:30:46 +0200 | [diff] [blame] | 689 | return -EIO; |
Mateusz Zalega | 21fe3f7 | 2014-04-30 13:07:48 +0200 | [diff] [blame] | 690 | |
Przemyslaw Marczak | 06ef7cc | 2013-10-23 14:30:46 +0200 | [diff] [blame] | 691 | k = 0; |
| 692 | } |
| 693 | |
Kishon Vijay Abraham I | 4763e16 | 2015-02-23 18:40:23 +0530 | [diff] [blame] | 694 | usb_gadget_handle_interrupts(0); |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 695 | } |
| 696 | common->thread_wakeup_needed = 0; |
| 697 | return rc; |
| 698 | } |
| 699 | |
| 700 | /*-------------------------------------------------------------------------*/ |
| 701 | |
| 702 | static int do_read(struct fsg_common *common) |
| 703 | { |
| 704 | struct fsg_lun *curlun = &common->luns[common->lun]; |
| 705 | u32 lba; |
| 706 | struct fsg_buffhd *bh; |
| 707 | int rc; |
| 708 | u32 amount_left; |
| 709 | loff_t file_offset; |
| 710 | unsigned int amount; |
| 711 | unsigned int partial_page; |
| 712 | ssize_t nread; |
| 713 | |
| 714 | /* Get the starting Logical Block Address and check that it's |
| 715 | * not too big */ |
| 716 | if (common->cmnd[0] == SC_READ_6) |
| 717 | lba = get_unaligned_be24(&common->cmnd[1]); |
| 718 | else { |
| 719 | lba = get_unaligned_be32(&common->cmnd[2]); |
| 720 | |
| 721 | /* We allow DPO (Disable Page Out = don't save data in the |
| 722 | * cache) and FUA (Force Unit Access = don't read from the |
| 723 | * cache), but we don't implement them. */ |
| 724 | if ((common->cmnd[1] & ~0x18) != 0) { |
| 725 | curlun->sense_data = SS_INVALID_FIELD_IN_CDB; |
| 726 | return -EINVAL; |
| 727 | } |
| 728 | } |
| 729 | if (lba >= curlun->num_sectors) { |
| 730 | curlun->sense_data = SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE; |
| 731 | return -EINVAL; |
| 732 | } |
| 733 | file_offset = ((loff_t) lba) << 9; |
| 734 | |
| 735 | /* Carry out the file reads */ |
| 736 | amount_left = common->data_size_from_cmnd; |
| 737 | if (unlikely(amount_left == 0)) |
| 738 | return -EIO; /* No default reply */ |
| 739 | |
| 740 | for (;;) { |
| 741 | |
| 742 | /* Figure out how much we need to read: |
| 743 | * Try to read the remaining amount. |
| 744 | * But don't read more than the buffer size. |
| 745 | * And don't try to read past the end of the file. |
| 746 | * Finally, if we're not at a page boundary, don't read past |
| 747 | * the next page. |
| 748 | * If this means reading 0 then we were asked to read past |
| 749 | * the end of file. */ |
| 750 | amount = min(amount_left, FSG_BUFLEN); |
| 751 | partial_page = file_offset & (PAGE_CACHE_SIZE - 1); |
| 752 | if (partial_page > 0) |
| 753 | amount = min(amount, (unsigned int) PAGE_CACHE_SIZE - |
| 754 | partial_page); |
| 755 | |
| 756 | /* Wait for the next buffer to become available */ |
| 757 | bh = common->next_buffhd_to_fill; |
| 758 | while (bh->state != BUF_STATE_EMPTY) { |
| 759 | rc = sleep_thread(common); |
| 760 | if (rc) |
| 761 | return rc; |
| 762 | } |
| 763 | |
| 764 | /* If we were asked to read past the end of file, |
| 765 | * end with an empty buffer. */ |
| 766 | if (amount == 0) { |
| 767 | curlun->sense_data = |
| 768 | SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE; |
| 769 | curlun->info_valid = 1; |
| 770 | bh->inreq->length = 0; |
| 771 | bh->state = BUF_STATE_FULL; |
| 772 | break; |
| 773 | } |
| 774 | |
| 775 | /* Perform the read */ |
Stephen Warren | 9e7d588 | 2015-12-07 11:38:50 -0700 | [diff] [blame] | 776 | rc = ums[common->lun].read_sector(&ums[common->lun], |
Przemyslaw Marczak | 674b1a6 | 2013-10-23 14:30:42 +0200 | [diff] [blame] | 777 | file_offset / SECTOR_SIZE, |
| 778 | amount / SECTOR_SIZE, |
| 779 | (char __user *)bh->buf); |
| 780 | if (!rc) |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 781 | return -EIO; |
Przemyslaw Marczak | 674b1a6 | 2013-10-23 14:30:42 +0200 | [diff] [blame] | 782 | |
| 783 | nread = rc * SECTOR_SIZE; |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 784 | |
| 785 | VLDBG(curlun, "file read %u @ %llu -> %d\n", amount, |
| 786 | (unsigned long long) file_offset, |
| 787 | (int) nread); |
| 788 | |
| 789 | if (nread < 0) { |
| 790 | LDBG(curlun, "error in file read: %d\n", |
| 791 | (int) nread); |
| 792 | nread = 0; |
| 793 | } else if (nread < amount) { |
| 794 | LDBG(curlun, "partial file read: %d/%u\n", |
| 795 | (int) nread, amount); |
| 796 | nread -= (nread & 511); /* Round down to a block */ |
| 797 | } |
| 798 | file_offset += nread; |
| 799 | amount_left -= nread; |
| 800 | common->residue -= nread; |
| 801 | bh->inreq->length = nread; |
| 802 | bh->state = BUF_STATE_FULL; |
| 803 | |
| 804 | /* If an error occurred, report it and its position */ |
| 805 | if (nread < amount) { |
| 806 | curlun->sense_data = SS_UNRECOVERED_READ_ERROR; |
| 807 | curlun->info_valid = 1; |
| 808 | break; |
| 809 | } |
| 810 | |
| 811 | if (amount_left == 0) |
| 812 | break; /* No more left to read */ |
| 813 | |
| 814 | /* Send this buffer and go read some more */ |
| 815 | bh->inreq->zero = 0; |
| 816 | START_TRANSFER_OR(common, bulk_in, bh->inreq, |
| 817 | &bh->inreq_busy, &bh->state) |
| 818 | /* Don't know what to do if |
| 819 | * common->fsg is NULL */ |
| 820 | return -EIO; |
| 821 | common->next_buffhd_to_fill = bh->next; |
| 822 | } |
| 823 | |
| 824 | return -EIO; /* No default reply */ |
| 825 | } |
| 826 | |
| 827 | /*-------------------------------------------------------------------------*/ |
| 828 | |
| 829 | static int do_write(struct fsg_common *common) |
| 830 | { |
| 831 | struct fsg_lun *curlun = &common->luns[common->lun]; |
| 832 | u32 lba; |
| 833 | struct fsg_buffhd *bh; |
| 834 | int get_some_more; |
| 835 | u32 amount_left_to_req, amount_left_to_write; |
| 836 | loff_t usb_offset, file_offset; |
| 837 | unsigned int amount; |
| 838 | unsigned int partial_page; |
| 839 | ssize_t nwritten; |
| 840 | int rc; |
| 841 | |
| 842 | if (curlun->ro) { |
| 843 | curlun->sense_data = SS_WRITE_PROTECTED; |
| 844 | return -EINVAL; |
| 845 | } |
| 846 | |
| 847 | /* Get the starting Logical Block Address and check that it's |
| 848 | * not too big */ |
| 849 | if (common->cmnd[0] == SC_WRITE_6) |
| 850 | lba = get_unaligned_be24(&common->cmnd[1]); |
| 851 | else { |
| 852 | lba = get_unaligned_be32(&common->cmnd[2]); |
| 853 | |
| 854 | /* We allow DPO (Disable Page Out = don't save data in the |
| 855 | * cache) and FUA (Force Unit Access = write directly to the |
| 856 | * medium). We don't implement DPO; we implement FUA by |
| 857 | * performing synchronous output. */ |
| 858 | if (common->cmnd[1] & ~0x18) { |
| 859 | curlun->sense_data = SS_INVALID_FIELD_IN_CDB; |
| 860 | return -EINVAL; |
| 861 | } |
| 862 | } |
| 863 | if (lba >= curlun->num_sectors) { |
| 864 | curlun->sense_data = SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE; |
| 865 | return -EINVAL; |
| 866 | } |
| 867 | |
| 868 | /* Carry out the file writes */ |
| 869 | get_some_more = 1; |
| 870 | file_offset = usb_offset = ((loff_t) lba) << 9; |
| 871 | amount_left_to_req = common->data_size_from_cmnd; |
| 872 | amount_left_to_write = common->data_size_from_cmnd; |
| 873 | |
| 874 | while (amount_left_to_write > 0) { |
| 875 | |
| 876 | /* Queue a request for more data from the host */ |
| 877 | bh = common->next_buffhd_to_fill; |
| 878 | if (bh->state == BUF_STATE_EMPTY && get_some_more) { |
| 879 | |
| 880 | /* Figure out how much we want to get: |
| 881 | * Try to get the remaining amount. |
| 882 | * But don't get more than the buffer size. |
| 883 | * And don't try to go past the end of the file. |
| 884 | * If we're not at a page boundary, |
| 885 | * don't go past the next page. |
| 886 | * If this means getting 0, then we were asked |
| 887 | * to write past the end of file. |
| 888 | * Finally, round down to a block boundary. */ |
| 889 | amount = min(amount_left_to_req, FSG_BUFLEN); |
| 890 | partial_page = usb_offset & (PAGE_CACHE_SIZE - 1); |
| 891 | if (partial_page > 0) |
| 892 | amount = min(amount, |
| 893 | (unsigned int) PAGE_CACHE_SIZE - partial_page); |
| 894 | |
| 895 | if (amount == 0) { |
| 896 | get_some_more = 0; |
| 897 | curlun->sense_data = |
| 898 | SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE; |
| 899 | curlun->info_valid = 1; |
| 900 | continue; |
| 901 | } |
| 902 | amount -= (amount & 511); |
| 903 | if (amount == 0) { |
| 904 | |
| 905 | /* Why were we were asked to transfer a |
| 906 | * partial block? */ |
| 907 | get_some_more = 0; |
| 908 | continue; |
| 909 | } |
| 910 | |
| 911 | /* Get the next buffer */ |
| 912 | usb_offset += amount; |
| 913 | common->usb_amount_left -= amount; |
| 914 | amount_left_to_req -= amount; |
| 915 | if (amount_left_to_req == 0) |
| 916 | get_some_more = 0; |
| 917 | |
| 918 | /* amount is always divisible by 512, hence by |
| 919 | * the bulk-out maxpacket size */ |
| 920 | bh->outreq->length = amount; |
| 921 | bh->bulk_out_intended_length = amount; |
| 922 | bh->outreq->short_not_ok = 1; |
| 923 | START_TRANSFER_OR(common, bulk_out, bh->outreq, |
| 924 | &bh->outreq_busy, &bh->state) |
| 925 | /* Don't know what to do if |
| 926 | * common->fsg is NULL */ |
| 927 | return -EIO; |
| 928 | common->next_buffhd_to_fill = bh->next; |
| 929 | continue; |
| 930 | } |
| 931 | |
| 932 | /* Write the received data to the backing file */ |
| 933 | bh = common->next_buffhd_to_drain; |
| 934 | if (bh->state == BUF_STATE_EMPTY && !get_some_more) |
| 935 | break; /* We stopped early */ |
| 936 | if (bh->state == BUF_STATE_FULL) { |
| 937 | common->next_buffhd_to_drain = bh->next; |
| 938 | bh->state = BUF_STATE_EMPTY; |
| 939 | |
| 940 | /* Did something go wrong with the transfer? */ |
| 941 | if (bh->outreq->status != 0) { |
| 942 | curlun->sense_data = SS_COMMUNICATION_FAILURE; |
| 943 | curlun->info_valid = 1; |
| 944 | break; |
| 945 | } |
| 946 | |
| 947 | amount = bh->outreq->actual; |
| 948 | |
| 949 | /* Perform the write */ |
Stephen Warren | 9e7d588 | 2015-12-07 11:38:50 -0700 | [diff] [blame] | 950 | rc = ums[common->lun].write_sector(&ums[common->lun], |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 951 | file_offset / SECTOR_SIZE, |
| 952 | amount / SECTOR_SIZE, |
| 953 | (char __user *)bh->buf); |
Przemyslaw Marczak | 674b1a6 | 2013-10-23 14:30:42 +0200 | [diff] [blame] | 954 | if (!rc) |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 955 | return -EIO; |
Przemyslaw Marczak | 674b1a6 | 2013-10-23 14:30:42 +0200 | [diff] [blame] | 956 | nwritten = rc * SECTOR_SIZE; |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 957 | |
| 958 | VLDBG(curlun, "file write %u @ %llu -> %d\n", amount, |
| 959 | (unsigned long long) file_offset, |
| 960 | (int) nwritten); |
| 961 | |
| 962 | if (nwritten < 0) { |
| 963 | LDBG(curlun, "error in file write: %d\n", |
| 964 | (int) nwritten); |
| 965 | nwritten = 0; |
| 966 | } else if (nwritten < amount) { |
| 967 | LDBG(curlun, "partial file write: %d/%u\n", |
| 968 | (int) nwritten, amount); |
| 969 | nwritten -= (nwritten & 511); |
| 970 | /* Round down to a block */ |
| 971 | } |
| 972 | file_offset += nwritten; |
| 973 | amount_left_to_write -= nwritten; |
| 974 | common->residue -= nwritten; |
| 975 | |
| 976 | /* If an error occurred, report it and its position */ |
| 977 | if (nwritten < amount) { |
Thierry Reding | 15fe9c8 | 2015-03-20 12:41:25 +0100 | [diff] [blame] | 978 | printf("nwritten:%zd amount:%u\n", nwritten, |
Przemyslaw Marczak | 674b1a6 | 2013-10-23 14:30:42 +0200 | [diff] [blame] | 979 | amount); |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 980 | curlun->sense_data = SS_WRITE_ERROR; |
| 981 | curlun->info_valid = 1; |
| 982 | break; |
| 983 | } |
| 984 | |
| 985 | /* Did the host decide to stop early? */ |
| 986 | if (bh->outreq->actual != bh->outreq->length) { |
| 987 | common->short_packet_received = 1; |
| 988 | break; |
| 989 | } |
| 990 | continue; |
| 991 | } |
| 992 | |
| 993 | /* Wait for something to happen */ |
| 994 | rc = sleep_thread(common); |
| 995 | if (rc) |
| 996 | return rc; |
| 997 | } |
| 998 | |
| 999 | return -EIO; /* No default reply */ |
| 1000 | } |
| 1001 | |
| 1002 | /*-------------------------------------------------------------------------*/ |
| 1003 | |
| 1004 | static int do_synchronize_cache(struct fsg_common *common) |
| 1005 | { |
| 1006 | return 0; |
| 1007 | } |
| 1008 | |
| 1009 | /*-------------------------------------------------------------------------*/ |
| 1010 | |
| 1011 | static int do_verify(struct fsg_common *common) |
| 1012 | { |
| 1013 | struct fsg_lun *curlun = &common->luns[common->lun]; |
| 1014 | u32 lba; |
| 1015 | u32 verification_length; |
| 1016 | struct fsg_buffhd *bh = common->next_buffhd_to_fill; |
| 1017 | loff_t file_offset; |
| 1018 | u32 amount_left; |
| 1019 | unsigned int amount; |
| 1020 | ssize_t nread; |
| 1021 | int rc; |
| 1022 | |
| 1023 | /* Get the starting Logical Block Address and check that it's |
| 1024 | * not too big */ |
| 1025 | lba = get_unaligned_be32(&common->cmnd[2]); |
| 1026 | if (lba >= curlun->num_sectors) { |
| 1027 | curlun->sense_data = SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE; |
| 1028 | return -EINVAL; |
| 1029 | } |
| 1030 | |
| 1031 | /* We allow DPO (Disable Page Out = don't save data in the |
| 1032 | * cache) but we don't implement it. */ |
| 1033 | if (common->cmnd[1] & ~0x10) { |
| 1034 | curlun->sense_data = SS_INVALID_FIELD_IN_CDB; |
| 1035 | return -EINVAL; |
| 1036 | } |
| 1037 | |
| 1038 | verification_length = get_unaligned_be16(&common->cmnd[7]); |
| 1039 | if (unlikely(verification_length == 0)) |
| 1040 | return -EIO; /* No default reply */ |
| 1041 | |
| 1042 | /* Prepare to carry out the file verify */ |
| 1043 | amount_left = verification_length << 9; |
| 1044 | file_offset = ((loff_t) lba) << 9; |
| 1045 | |
| 1046 | /* Write out all the dirty buffers before invalidating them */ |
| 1047 | |
| 1048 | /* Just try to read the requested blocks */ |
| 1049 | while (amount_left > 0) { |
| 1050 | |
| 1051 | /* Figure out how much we need to read: |
| 1052 | * Try to read the remaining amount, but not more than |
| 1053 | * the buffer size. |
| 1054 | * And don't try to read past the end of the file. |
| 1055 | * If this means reading 0 then we were asked to read |
| 1056 | * past the end of file. */ |
| 1057 | amount = min(amount_left, FSG_BUFLEN); |
| 1058 | if (amount == 0) { |
| 1059 | curlun->sense_data = |
| 1060 | SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE; |
| 1061 | curlun->info_valid = 1; |
| 1062 | break; |
| 1063 | } |
| 1064 | |
| 1065 | /* Perform the read */ |
Stephen Warren | 9e7d588 | 2015-12-07 11:38:50 -0700 | [diff] [blame] | 1066 | rc = ums[common->lun].read_sector(&ums[common->lun], |
Przemyslaw Marczak | 674b1a6 | 2013-10-23 14:30:42 +0200 | [diff] [blame] | 1067 | file_offset / SECTOR_SIZE, |
| 1068 | amount / SECTOR_SIZE, |
| 1069 | (char __user *)bh->buf); |
| 1070 | if (!rc) |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 1071 | return -EIO; |
Przemyslaw Marczak | 674b1a6 | 2013-10-23 14:30:42 +0200 | [diff] [blame] | 1072 | nread = rc * SECTOR_SIZE; |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 1073 | |
| 1074 | VLDBG(curlun, "file read %u @ %llu -> %d\n", amount, |
| 1075 | (unsigned long long) file_offset, |
| 1076 | (int) nread); |
| 1077 | if (nread < 0) { |
| 1078 | LDBG(curlun, "error in file verify: %d\n", |
| 1079 | (int) nread); |
| 1080 | nread = 0; |
| 1081 | } else if (nread < amount) { |
| 1082 | LDBG(curlun, "partial file verify: %d/%u\n", |
| 1083 | (int) nread, amount); |
| 1084 | nread -= (nread & 511); /* Round down to a sector */ |
| 1085 | } |
| 1086 | if (nread == 0) { |
| 1087 | curlun->sense_data = SS_UNRECOVERED_READ_ERROR; |
| 1088 | curlun->info_valid = 1; |
| 1089 | break; |
| 1090 | } |
| 1091 | file_offset += nread; |
| 1092 | amount_left -= nread; |
| 1093 | } |
| 1094 | return 0; |
| 1095 | } |
| 1096 | |
| 1097 | /*-------------------------------------------------------------------------*/ |
| 1098 | |
| 1099 | static int do_inquiry(struct fsg_common *common, struct fsg_buffhd *bh) |
| 1100 | { |
| 1101 | struct fsg_lun *curlun = &common->luns[common->lun]; |
| 1102 | static const char vendor_id[] = "Linux "; |
| 1103 | u8 *buf = (u8 *) bh->buf; |
| 1104 | |
| 1105 | if (!curlun) { /* Unsupported LUNs are okay */ |
| 1106 | common->bad_lun_okay = 1; |
| 1107 | memset(buf, 0, 36); |
| 1108 | buf[0] = 0x7f; /* Unsupported, no device-type */ |
| 1109 | buf[4] = 31; /* Additional length */ |
| 1110 | return 36; |
| 1111 | } |
| 1112 | |
| 1113 | memset(buf, 0, 8); |
| 1114 | buf[0] = TYPE_DISK; |
Eric Nelson | e1d833a | 2014-09-19 17:06:46 -0700 | [diff] [blame] | 1115 | buf[1] = curlun->removable ? 0x80 : 0; |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 1116 | buf[2] = 2; /* ANSI SCSI level 2 */ |
| 1117 | buf[3] = 2; /* SCSI-2 INQUIRY data format */ |
| 1118 | buf[4] = 31; /* Additional length */ |
| 1119 | /* No special options */ |
| 1120 | sprintf((char *) (buf + 8), "%-8s%-16s%04x", (char*) vendor_id , |
Stephen Warren | 9e7d588 | 2015-12-07 11:38:50 -0700 | [diff] [blame] | 1121 | ums[common->lun].name, (u16) 0xffff); |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 1122 | |
| 1123 | return 36; |
| 1124 | } |
| 1125 | |
| 1126 | |
| 1127 | static int do_request_sense(struct fsg_common *common, struct fsg_buffhd *bh) |
| 1128 | { |
| 1129 | struct fsg_lun *curlun = &common->luns[common->lun]; |
| 1130 | u8 *buf = (u8 *) bh->buf; |
| 1131 | u32 sd, sdinfo; |
| 1132 | int valid; |
| 1133 | |
| 1134 | /* |
| 1135 | * From the SCSI-2 spec., section 7.9 (Unit attention condition): |
| 1136 | * |
| 1137 | * If a REQUEST SENSE command is received from an initiator |
| 1138 | * with a pending unit attention condition (before the target |
| 1139 | * generates the contingent allegiance condition), then the |
| 1140 | * target shall either: |
| 1141 | * a) report any pending sense data and preserve the unit |
| 1142 | * attention condition on the logical unit, or, |
| 1143 | * b) report the unit attention condition, may discard any |
| 1144 | * pending sense data, and clear the unit attention |
| 1145 | * condition on the logical unit for that initiator. |
| 1146 | * |
| 1147 | * FSG normally uses option a); enable this code to use option b). |
| 1148 | */ |
| 1149 | #if 0 |
| 1150 | if (curlun && curlun->unit_attention_data != SS_NO_SENSE) { |
| 1151 | curlun->sense_data = curlun->unit_attention_data; |
| 1152 | curlun->unit_attention_data = SS_NO_SENSE; |
| 1153 | } |
| 1154 | #endif |
| 1155 | |
| 1156 | if (!curlun) { /* Unsupported LUNs are okay */ |
| 1157 | common->bad_lun_okay = 1; |
| 1158 | sd = SS_LOGICAL_UNIT_NOT_SUPPORTED; |
| 1159 | sdinfo = 0; |
| 1160 | valid = 0; |
| 1161 | } else { |
| 1162 | sd = curlun->sense_data; |
| 1163 | valid = curlun->info_valid << 7; |
| 1164 | curlun->sense_data = SS_NO_SENSE; |
| 1165 | curlun->info_valid = 0; |
| 1166 | } |
| 1167 | |
| 1168 | memset(buf, 0, 18); |
| 1169 | buf[0] = valid | 0x70; /* Valid, current error */ |
| 1170 | buf[2] = SK(sd); |
| 1171 | put_unaligned_be32(sdinfo, &buf[3]); /* Sense information */ |
| 1172 | buf[7] = 18 - 8; /* Additional sense length */ |
| 1173 | buf[12] = ASC(sd); |
| 1174 | buf[13] = ASCQ(sd); |
| 1175 | return 18; |
| 1176 | } |
| 1177 | |
| 1178 | static int do_read_capacity(struct fsg_common *common, struct fsg_buffhd *bh) |
| 1179 | { |
| 1180 | struct fsg_lun *curlun = &common->luns[common->lun]; |
| 1181 | u32 lba = get_unaligned_be32(&common->cmnd[2]); |
| 1182 | int pmi = common->cmnd[8]; |
| 1183 | u8 *buf = (u8 *) bh->buf; |
| 1184 | |
| 1185 | /* Check the PMI and LBA fields */ |
| 1186 | if (pmi > 1 || (pmi == 0 && lba != 0)) { |
| 1187 | curlun->sense_data = SS_INVALID_FIELD_IN_CDB; |
| 1188 | return -EINVAL; |
| 1189 | } |
| 1190 | |
| 1191 | put_unaligned_be32(curlun->num_sectors - 1, &buf[0]); |
| 1192 | /* Max logical block */ |
| 1193 | put_unaligned_be32(512, &buf[4]); /* Block length */ |
| 1194 | return 8; |
| 1195 | } |
| 1196 | |
| 1197 | static int do_read_header(struct fsg_common *common, struct fsg_buffhd *bh) |
| 1198 | { |
| 1199 | struct fsg_lun *curlun = &common->luns[common->lun]; |
| 1200 | int msf = common->cmnd[1] & 0x02; |
| 1201 | u32 lba = get_unaligned_be32(&common->cmnd[2]); |
| 1202 | u8 *buf = (u8 *) bh->buf; |
| 1203 | |
| 1204 | if (common->cmnd[1] & ~0x02) { /* Mask away MSF */ |
| 1205 | curlun->sense_data = SS_INVALID_FIELD_IN_CDB; |
| 1206 | return -EINVAL; |
| 1207 | } |
| 1208 | if (lba >= curlun->num_sectors) { |
| 1209 | curlun->sense_data = SS_LOGICAL_BLOCK_ADDRESS_OUT_OF_RANGE; |
| 1210 | return -EINVAL; |
| 1211 | } |
| 1212 | |
| 1213 | memset(buf, 0, 8); |
| 1214 | buf[0] = 0x01; /* 2048 bytes of user data, rest is EC */ |
| 1215 | store_cdrom_address(&buf[4], msf, lba); |
| 1216 | return 8; |
| 1217 | } |
| 1218 | |
| 1219 | |
| 1220 | static int do_read_toc(struct fsg_common *common, struct fsg_buffhd *bh) |
| 1221 | { |
| 1222 | struct fsg_lun *curlun = &common->luns[common->lun]; |
| 1223 | int msf = common->cmnd[1] & 0x02; |
| 1224 | int start_track = common->cmnd[6]; |
| 1225 | u8 *buf = (u8 *) bh->buf; |
| 1226 | |
| 1227 | if ((common->cmnd[1] & ~0x02) != 0 || /* Mask away MSF */ |
| 1228 | start_track > 1) { |
| 1229 | curlun->sense_data = SS_INVALID_FIELD_IN_CDB; |
| 1230 | return -EINVAL; |
| 1231 | } |
| 1232 | |
| 1233 | memset(buf, 0, 20); |
| 1234 | buf[1] = (20-2); /* TOC data length */ |
| 1235 | buf[2] = 1; /* First track number */ |
| 1236 | buf[3] = 1; /* Last track number */ |
| 1237 | buf[5] = 0x16; /* Data track, copying allowed */ |
| 1238 | buf[6] = 0x01; /* Only track is number 1 */ |
| 1239 | store_cdrom_address(&buf[8], msf, 0); |
| 1240 | |
| 1241 | buf[13] = 0x16; /* Lead-out track is data */ |
| 1242 | buf[14] = 0xAA; /* Lead-out track number */ |
| 1243 | store_cdrom_address(&buf[16], msf, curlun->num_sectors); |
| 1244 | |
| 1245 | return 20; |
| 1246 | } |
| 1247 | |
| 1248 | static int do_mode_sense(struct fsg_common *common, struct fsg_buffhd *bh) |
| 1249 | { |
| 1250 | struct fsg_lun *curlun = &common->luns[common->lun]; |
| 1251 | int mscmnd = common->cmnd[0]; |
| 1252 | u8 *buf = (u8 *) bh->buf; |
| 1253 | u8 *buf0 = buf; |
| 1254 | int pc, page_code; |
| 1255 | int changeable_values, all_pages; |
| 1256 | int valid_page = 0; |
| 1257 | int len, limit; |
| 1258 | |
| 1259 | if ((common->cmnd[1] & ~0x08) != 0) { /* Mask away DBD */ |
| 1260 | curlun->sense_data = SS_INVALID_FIELD_IN_CDB; |
| 1261 | return -EINVAL; |
| 1262 | } |
| 1263 | pc = common->cmnd[2] >> 6; |
| 1264 | page_code = common->cmnd[2] & 0x3f; |
| 1265 | if (pc == 3) { |
| 1266 | curlun->sense_data = SS_SAVING_PARAMETERS_NOT_SUPPORTED; |
| 1267 | return -EINVAL; |
| 1268 | } |
| 1269 | changeable_values = (pc == 1); |
| 1270 | all_pages = (page_code == 0x3f); |
| 1271 | |
| 1272 | /* Write the mode parameter header. Fixed values are: default |
| 1273 | * medium type, no cache control (DPOFUA), and no block descriptors. |
| 1274 | * The only variable value is the WriteProtect bit. We will fill in |
| 1275 | * the mode data length later. */ |
| 1276 | memset(buf, 0, 8); |
| 1277 | if (mscmnd == SC_MODE_SENSE_6) { |
| 1278 | buf[2] = (curlun->ro ? 0x80 : 0x00); /* WP, DPOFUA */ |
| 1279 | buf += 4; |
| 1280 | limit = 255; |
| 1281 | } else { /* SC_MODE_SENSE_10 */ |
| 1282 | buf[3] = (curlun->ro ? 0x80 : 0x00); /* WP, DPOFUA */ |
| 1283 | buf += 8; |
| 1284 | limit = 65535; /* Should really be FSG_BUFLEN */ |
| 1285 | } |
| 1286 | |
| 1287 | /* No block descriptors */ |
| 1288 | |
| 1289 | /* The mode pages, in numerical order. The only page we support |
| 1290 | * is the Caching page. */ |
| 1291 | if (page_code == 0x08 || all_pages) { |
| 1292 | valid_page = 1; |
| 1293 | buf[0] = 0x08; /* Page code */ |
| 1294 | buf[1] = 10; /* Page length */ |
| 1295 | memset(buf+2, 0, 10); /* None of the fields are changeable */ |
| 1296 | |
| 1297 | if (!changeable_values) { |
| 1298 | buf[2] = 0x04; /* Write cache enable, */ |
| 1299 | /* Read cache not disabled */ |
| 1300 | /* No cache retention priorities */ |
| 1301 | put_unaligned_be16(0xffff, &buf[4]); |
| 1302 | /* Don't disable prefetch */ |
| 1303 | /* Minimum prefetch = 0 */ |
| 1304 | put_unaligned_be16(0xffff, &buf[8]); |
| 1305 | /* Maximum prefetch */ |
| 1306 | put_unaligned_be16(0xffff, &buf[10]); |
| 1307 | /* Maximum prefetch ceiling */ |
| 1308 | } |
| 1309 | buf += 12; |
| 1310 | } |
| 1311 | |
| 1312 | /* Check that a valid page was requested and the mode data length |
| 1313 | * isn't too long. */ |
| 1314 | len = buf - buf0; |
| 1315 | if (!valid_page || len > limit) { |
| 1316 | curlun->sense_data = SS_INVALID_FIELD_IN_CDB; |
| 1317 | return -EINVAL; |
| 1318 | } |
| 1319 | |
| 1320 | /* Store the mode data length */ |
| 1321 | if (mscmnd == SC_MODE_SENSE_6) |
| 1322 | buf0[0] = len - 1; |
| 1323 | else |
| 1324 | put_unaligned_be16(len - 2, buf0); |
| 1325 | return len; |
| 1326 | } |
| 1327 | |
| 1328 | |
| 1329 | static int do_start_stop(struct fsg_common *common) |
| 1330 | { |
| 1331 | struct fsg_lun *curlun = &common->luns[common->lun]; |
| 1332 | |
| 1333 | if (!curlun) { |
| 1334 | return -EINVAL; |
| 1335 | } else if (!curlun->removable) { |
| 1336 | curlun->sense_data = SS_INVALID_COMMAND; |
| 1337 | return -EINVAL; |
| 1338 | } |
| 1339 | |
| 1340 | return 0; |
| 1341 | } |
| 1342 | |
| 1343 | static int do_prevent_allow(struct fsg_common *common) |
| 1344 | { |
| 1345 | struct fsg_lun *curlun = &common->luns[common->lun]; |
| 1346 | int prevent; |
| 1347 | |
| 1348 | if (!curlun->removable) { |
| 1349 | curlun->sense_data = SS_INVALID_COMMAND; |
| 1350 | return -EINVAL; |
| 1351 | } |
| 1352 | |
| 1353 | prevent = common->cmnd[4] & 0x01; |
| 1354 | if ((common->cmnd[4] & ~0x01) != 0) { /* Mask away Prevent */ |
| 1355 | curlun->sense_data = SS_INVALID_FIELD_IN_CDB; |
| 1356 | return -EINVAL; |
| 1357 | } |
| 1358 | |
| 1359 | if (curlun->prevent_medium_removal && !prevent) |
| 1360 | fsg_lun_fsync_sub(curlun); |
| 1361 | curlun->prevent_medium_removal = prevent; |
| 1362 | return 0; |
| 1363 | } |
| 1364 | |
| 1365 | |
| 1366 | static int do_read_format_capacities(struct fsg_common *common, |
| 1367 | struct fsg_buffhd *bh) |
| 1368 | { |
| 1369 | struct fsg_lun *curlun = &common->luns[common->lun]; |
| 1370 | u8 *buf = (u8 *) bh->buf; |
| 1371 | |
| 1372 | buf[0] = buf[1] = buf[2] = 0; |
| 1373 | buf[3] = 8; /* Only the Current/Maximum Capacity Descriptor */ |
| 1374 | buf += 4; |
| 1375 | |
| 1376 | put_unaligned_be32(curlun->num_sectors, &buf[0]); |
| 1377 | /* Number of blocks */ |
| 1378 | put_unaligned_be32(512, &buf[4]); /* Block length */ |
| 1379 | buf[4] = 0x02; /* Current capacity */ |
| 1380 | return 12; |
| 1381 | } |
| 1382 | |
| 1383 | |
| 1384 | static int do_mode_select(struct fsg_common *common, struct fsg_buffhd *bh) |
| 1385 | { |
| 1386 | struct fsg_lun *curlun = &common->luns[common->lun]; |
| 1387 | |
| 1388 | /* We don't support MODE SELECT */ |
| 1389 | if (curlun) |
| 1390 | curlun->sense_data = SS_INVALID_COMMAND; |
| 1391 | return -EINVAL; |
| 1392 | } |
| 1393 | |
| 1394 | |
| 1395 | /*-------------------------------------------------------------------------*/ |
| 1396 | |
| 1397 | static int halt_bulk_in_endpoint(struct fsg_dev *fsg) |
| 1398 | { |
| 1399 | int rc; |
| 1400 | |
| 1401 | rc = fsg_set_halt(fsg, fsg->bulk_in); |
| 1402 | if (rc == -EAGAIN) |
| 1403 | VDBG(fsg, "delayed bulk-in endpoint halt\n"); |
| 1404 | while (rc != 0) { |
| 1405 | if (rc != -EAGAIN) { |
| 1406 | WARNING(fsg, "usb_ep_set_halt -> %d\n", rc); |
| 1407 | rc = 0; |
| 1408 | break; |
| 1409 | } |
| 1410 | |
| 1411 | rc = usb_ep_set_halt(fsg->bulk_in); |
| 1412 | } |
| 1413 | return rc; |
| 1414 | } |
| 1415 | |
| 1416 | static int wedge_bulk_in_endpoint(struct fsg_dev *fsg) |
| 1417 | { |
| 1418 | int rc; |
| 1419 | |
| 1420 | DBG(fsg, "bulk-in set wedge\n"); |
| 1421 | rc = 0; /* usb_ep_set_wedge(fsg->bulk_in); */ |
| 1422 | if (rc == -EAGAIN) |
| 1423 | VDBG(fsg, "delayed bulk-in endpoint wedge\n"); |
| 1424 | while (rc != 0) { |
| 1425 | if (rc != -EAGAIN) { |
| 1426 | WARNING(fsg, "usb_ep_set_wedge -> %d\n", rc); |
| 1427 | rc = 0; |
| 1428 | break; |
| 1429 | } |
| 1430 | } |
| 1431 | return rc; |
| 1432 | } |
| 1433 | |
| 1434 | static int pad_with_zeros(struct fsg_dev *fsg) |
| 1435 | { |
| 1436 | struct fsg_buffhd *bh = fsg->common->next_buffhd_to_fill; |
| 1437 | u32 nkeep = bh->inreq->length; |
| 1438 | u32 nsend; |
| 1439 | int rc; |
| 1440 | |
| 1441 | bh->state = BUF_STATE_EMPTY; /* For the first iteration */ |
| 1442 | fsg->common->usb_amount_left = nkeep + fsg->common->residue; |
| 1443 | while (fsg->common->usb_amount_left > 0) { |
| 1444 | |
| 1445 | /* Wait for the next buffer to be free */ |
| 1446 | while (bh->state != BUF_STATE_EMPTY) { |
| 1447 | rc = sleep_thread(fsg->common); |
| 1448 | if (rc) |
| 1449 | return rc; |
| 1450 | } |
| 1451 | |
| 1452 | nsend = min(fsg->common->usb_amount_left, FSG_BUFLEN); |
| 1453 | memset(bh->buf + nkeep, 0, nsend - nkeep); |
| 1454 | bh->inreq->length = nsend; |
| 1455 | bh->inreq->zero = 0; |
| 1456 | start_transfer(fsg, fsg->bulk_in, bh->inreq, |
| 1457 | &bh->inreq_busy, &bh->state); |
| 1458 | bh = fsg->common->next_buffhd_to_fill = bh->next; |
| 1459 | fsg->common->usb_amount_left -= nsend; |
| 1460 | nkeep = 0; |
| 1461 | } |
| 1462 | return 0; |
| 1463 | } |
| 1464 | |
| 1465 | static int throw_away_data(struct fsg_common *common) |
| 1466 | { |
| 1467 | struct fsg_buffhd *bh; |
| 1468 | u32 amount; |
| 1469 | int rc; |
| 1470 | |
| 1471 | for (bh = common->next_buffhd_to_drain; |
| 1472 | bh->state != BUF_STATE_EMPTY || common->usb_amount_left > 0; |
| 1473 | bh = common->next_buffhd_to_drain) { |
| 1474 | |
| 1475 | /* Throw away the data in a filled buffer */ |
| 1476 | if (bh->state == BUF_STATE_FULL) { |
| 1477 | bh->state = BUF_STATE_EMPTY; |
| 1478 | common->next_buffhd_to_drain = bh->next; |
| 1479 | |
| 1480 | /* A short packet or an error ends everything */ |
| 1481 | if (bh->outreq->actual != bh->outreq->length || |
| 1482 | bh->outreq->status != 0) { |
| 1483 | raise_exception(common, |
| 1484 | FSG_STATE_ABORT_BULK_OUT); |
| 1485 | return -EINTR; |
| 1486 | } |
| 1487 | continue; |
| 1488 | } |
| 1489 | |
| 1490 | /* Try to submit another request if we need one */ |
| 1491 | bh = common->next_buffhd_to_fill; |
| 1492 | if (bh->state == BUF_STATE_EMPTY |
| 1493 | && common->usb_amount_left > 0) { |
| 1494 | amount = min(common->usb_amount_left, FSG_BUFLEN); |
| 1495 | |
| 1496 | /* amount is always divisible by 512, hence by |
| 1497 | * the bulk-out maxpacket size */ |
| 1498 | bh->outreq->length = amount; |
| 1499 | bh->bulk_out_intended_length = amount; |
| 1500 | bh->outreq->short_not_ok = 1; |
| 1501 | START_TRANSFER_OR(common, bulk_out, bh->outreq, |
| 1502 | &bh->outreq_busy, &bh->state) |
| 1503 | /* Don't know what to do if |
| 1504 | * common->fsg is NULL */ |
| 1505 | return -EIO; |
| 1506 | common->next_buffhd_to_fill = bh->next; |
| 1507 | common->usb_amount_left -= amount; |
| 1508 | continue; |
| 1509 | } |
| 1510 | |
| 1511 | /* Otherwise wait for something to happen */ |
| 1512 | rc = sleep_thread(common); |
| 1513 | if (rc) |
| 1514 | return rc; |
| 1515 | } |
| 1516 | return 0; |
| 1517 | } |
| 1518 | |
| 1519 | |
| 1520 | static int finish_reply(struct fsg_common *common) |
| 1521 | { |
| 1522 | struct fsg_buffhd *bh = common->next_buffhd_to_fill; |
| 1523 | int rc = 0; |
| 1524 | |
| 1525 | switch (common->data_dir) { |
| 1526 | case DATA_DIR_NONE: |
| 1527 | break; /* Nothing to send */ |
| 1528 | |
| 1529 | /* If we don't know whether the host wants to read or write, |
| 1530 | * this must be CB or CBI with an unknown command. We mustn't |
| 1531 | * try to send or receive any data. So stall both bulk pipes |
| 1532 | * if we can and wait for a reset. */ |
| 1533 | case DATA_DIR_UNKNOWN: |
| 1534 | if (!common->can_stall) { |
| 1535 | /* Nothing */ |
| 1536 | } else if (fsg_is_set(common)) { |
| 1537 | fsg_set_halt(common->fsg, common->fsg->bulk_out); |
| 1538 | rc = halt_bulk_in_endpoint(common->fsg); |
| 1539 | } else { |
| 1540 | /* Don't know what to do if common->fsg is NULL */ |
| 1541 | rc = -EIO; |
| 1542 | } |
| 1543 | break; |
| 1544 | |
| 1545 | /* All but the last buffer of data must have already been sent */ |
| 1546 | case DATA_DIR_TO_HOST: |
| 1547 | if (common->data_size == 0) { |
| 1548 | /* Nothing to send */ |
| 1549 | |
| 1550 | /* If there's no residue, simply send the last buffer */ |
| 1551 | } else if (common->residue == 0) { |
| 1552 | bh->inreq->zero = 0; |
| 1553 | START_TRANSFER_OR(common, bulk_in, bh->inreq, |
| 1554 | &bh->inreq_busy, &bh->state) |
| 1555 | return -EIO; |
| 1556 | common->next_buffhd_to_fill = bh->next; |
| 1557 | |
| 1558 | /* For Bulk-only, if we're allowed to stall then send the |
| 1559 | * short packet and halt the bulk-in endpoint. If we can't |
| 1560 | * stall, pad out the remaining data with 0's. */ |
| 1561 | } else if (common->can_stall) { |
| 1562 | bh->inreq->zero = 1; |
| 1563 | START_TRANSFER_OR(common, bulk_in, bh->inreq, |
| 1564 | &bh->inreq_busy, &bh->state) |
| 1565 | /* Don't know what to do if |
| 1566 | * common->fsg is NULL */ |
| 1567 | rc = -EIO; |
| 1568 | common->next_buffhd_to_fill = bh->next; |
| 1569 | if (common->fsg) |
| 1570 | rc = halt_bulk_in_endpoint(common->fsg); |
| 1571 | } else if (fsg_is_set(common)) { |
| 1572 | rc = pad_with_zeros(common->fsg); |
| 1573 | } else { |
| 1574 | /* Don't know what to do if common->fsg is NULL */ |
| 1575 | rc = -EIO; |
| 1576 | } |
| 1577 | break; |
| 1578 | |
| 1579 | /* We have processed all we want from the data the host has sent. |
| 1580 | * There may still be outstanding bulk-out requests. */ |
| 1581 | case DATA_DIR_FROM_HOST: |
| 1582 | if (common->residue == 0) { |
| 1583 | /* Nothing to receive */ |
| 1584 | |
| 1585 | /* Did the host stop sending unexpectedly early? */ |
| 1586 | } else if (common->short_packet_received) { |
| 1587 | raise_exception(common, FSG_STATE_ABORT_BULK_OUT); |
| 1588 | rc = -EINTR; |
| 1589 | |
| 1590 | /* We haven't processed all the incoming data. Even though |
| 1591 | * we may be allowed to stall, doing so would cause a race. |
| 1592 | * The controller may already have ACK'ed all the remaining |
| 1593 | * bulk-out packets, in which case the host wouldn't see a |
| 1594 | * STALL. Not realizing the endpoint was halted, it wouldn't |
| 1595 | * clear the halt -- leading to problems later on. */ |
| 1596 | #if 0 |
| 1597 | } else if (common->can_stall) { |
| 1598 | if (fsg_is_set(common)) |
| 1599 | fsg_set_halt(common->fsg, |
| 1600 | common->fsg->bulk_out); |
| 1601 | raise_exception(common, FSG_STATE_ABORT_BULK_OUT); |
| 1602 | rc = -EINTR; |
| 1603 | #endif |
| 1604 | |
| 1605 | /* We can't stall. Read in the excess data and throw it |
| 1606 | * all away. */ |
| 1607 | } else { |
| 1608 | rc = throw_away_data(common); |
| 1609 | } |
| 1610 | break; |
| 1611 | } |
| 1612 | return rc; |
| 1613 | } |
| 1614 | |
| 1615 | |
| 1616 | static int send_status(struct fsg_common *common) |
| 1617 | { |
| 1618 | struct fsg_lun *curlun = &common->luns[common->lun]; |
| 1619 | struct fsg_buffhd *bh; |
| 1620 | struct bulk_cs_wrap *csw; |
| 1621 | int rc; |
| 1622 | u8 status = USB_STATUS_PASS; |
| 1623 | u32 sd, sdinfo = 0; |
| 1624 | |
| 1625 | /* Wait for the next buffer to become available */ |
| 1626 | bh = common->next_buffhd_to_fill; |
| 1627 | while (bh->state != BUF_STATE_EMPTY) { |
| 1628 | rc = sleep_thread(common); |
| 1629 | if (rc) |
| 1630 | return rc; |
| 1631 | } |
| 1632 | |
| 1633 | if (curlun) |
| 1634 | sd = curlun->sense_data; |
| 1635 | else if (common->bad_lun_okay) |
| 1636 | sd = SS_NO_SENSE; |
| 1637 | else |
| 1638 | sd = SS_LOGICAL_UNIT_NOT_SUPPORTED; |
| 1639 | |
| 1640 | if (common->phase_error) { |
| 1641 | DBG(common, "sending phase-error status\n"); |
| 1642 | status = USB_STATUS_PHASE_ERROR; |
| 1643 | sd = SS_INVALID_COMMAND; |
| 1644 | } else if (sd != SS_NO_SENSE) { |
| 1645 | DBG(common, "sending command-failure status\n"); |
| 1646 | status = USB_STATUS_FAIL; |
| 1647 | VDBG(common, " sense data: SK x%02x, ASC x%02x, ASCQ x%02x;" |
| 1648 | " info x%x\n", |
| 1649 | SK(sd), ASC(sd), ASCQ(sd), sdinfo); |
| 1650 | } |
| 1651 | |
| 1652 | /* Store and send the Bulk-only CSW */ |
| 1653 | csw = (void *)bh->buf; |
| 1654 | |
| 1655 | csw->Signature = cpu_to_le32(USB_BULK_CS_SIG); |
| 1656 | csw->Tag = common->tag; |
| 1657 | csw->Residue = cpu_to_le32(common->residue); |
| 1658 | csw->Status = status; |
| 1659 | |
| 1660 | bh->inreq->length = USB_BULK_CS_WRAP_LEN; |
| 1661 | bh->inreq->zero = 0; |
| 1662 | START_TRANSFER_OR(common, bulk_in, bh->inreq, |
| 1663 | &bh->inreq_busy, &bh->state) |
| 1664 | /* Don't know what to do if common->fsg is NULL */ |
| 1665 | return -EIO; |
| 1666 | |
| 1667 | common->next_buffhd_to_fill = bh->next; |
| 1668 | return 0; |
| 1669 | } |
| 1670 | |
| 1671 | |
| 1672 | /*-------------------------------------------------------------------------*/ |
| 1673 | |
| 1674 | /* Check whether the command is properly formed and whether its data size |
| 1675 | * and direction agree with the values we already have. */ |
| 1676 | static int check_command(struct fsg_common *common, int cmnd_size, |
| 1677 | enum data_direction data_dir, unsigned int mask, |
| 1678 | int needs_medium, const char *name) |
| 1679 | { |
| 1680 | int i; |
| 1681 | int lun = common->cmnd[1] >> 5; |
| 1682 | static const char dirletter[4] = {'u', 'o', 'i', 'n'}; |
| 1683 | char hdlen[20]; |
| 1684 | struct fsg_lun *curlun; |
| 1685 | |
| 1686 | hdlen[0] = 0; |
| 1687 | if (common->data_dir != DATA_DIR_UNKNOWN) |
| 1688 | sprintf(hdlen, ", H%c=%u", dirletter[(int) common->data_dir], |
| 1689 | common->data_size); |
| 1690 | VDBG(common, "SCSI command: %s; Dc=%d, D%c=%u; Hc=%d%s\n", |
| 1691 | name, cmnd_size, dirletter[(int) data_dir], |
| 1692 | common->data_size_from_cmnd, common->cmnd_size, hdlen); |
| 1693 | |
| 1694 | /* We can't reply at all until we know the correct data direction |
| 1695 | * and size. */ |
| 1696 | if (common->data_size_from_cmnd == 0) |
| 1697 | data_dir = DATA_DIR_NONE; |
| 1698 | if (common->data_size < common->data_size_from_cmnd) { |
| 1699 | /* Host data size < Device data size is a phase error. |
| 1700 | * Carry out the command, but only transfer as much as |
| 1701 | * we are allowed. */ |
| 1702 | common->data_size_from_cmnd = common->data_size; |
| 1703 | common->phase_error = 1; |
| 1704 | } |
| 1705 | common->residue = common->data_size; |
| 1706 | common->usb_amount_left = common->data_size; |
| 1707 | |
| 1708 | /* Conflicting data directions is a phase error */ |
| 1709 | if (common->data_dir != data_dir |
| 1710 | && common->data_size_from_cmnd > 0) { |
| 1711 | common->phase_error = 1; |
| 1712 | return -EINVAL; |
| 1713 | } |
| 1714 | |
| 1715 | /* Verify the length of the command itself */ |
| 1716 | if (cmnd_size != common->cmnd_size) { |
| 1717 | |
| 1718 | /* Special case workaround: There are plenty of buggy SCSI |
| 1719 | * implementations. Many have issues with cbw->Length |
| 1720 | * field passing a wrong command size. For those cases we |
| 1721 | * always try to work around the problem by using the length |
| 1722 | * sent by the host side provided it is at least as large |
| 1723 | * as the correct command length. |
| 1724 | * Examples of such cases would be MS-Windows, which issues |
| 1725 | * REQUEST SENSE with cbw->Length == 12 where it should |
| 1726 | * be 6, and xbox360 issuing INQUIRY, TEST UNIT READY and |
| 1727 | * REQUEST SENSE with cbw->Length == 10 where it should |
| 1728 | * be 6 as well. |
| 1729 | */ |
| 1730 | if (cmnd_size <= common->cmnd_size) { |
| 1731 | DBG(common, "%s is buggy! Expected length %d " |
| 1732 | "but we got %d\n", name, |
| 1733 | cmnd_size, common->cmnd_size); |
| 1734 | cmnd_size = common->cmnd_size; |
| 1735 | } else { |
| 1736 | common->phase_error = 1; |
| 1737 | return -EINVAL; |
| 1738 | } |
| 1739 | } |
| 1740 | |
| 1741 | /* Check that the LUN values are consistent */ |
| 1742 | if (common->lun != lun) |
| 1743 | DBG(common, "using LUN %d from CBW, not LUN %d from CDB\n", |
| 1744 | common->lun, lun); |
| 1745 | |
| 1746 | /* Check the LUN */ |
| 1747 | if (common->lun >= 0 && common->lun < common->nluns) { |
| 1748 | curlun = &common->luns[common->lun]; |
| 1749 | if (common->cmnd[0] != SC_REQUEST_SENSE) { |
| 1750 | curlun->sense_data = SS_NO_SENSE; |
| 1751 | curlun->info_valid = 0; |
| 1752 | } |
| 1753 | } else { |
| 1754 | curlun = NULL; |
| 1755 | common->bad_lun_okay = 0; |
| 1756 | |
| 1757 | /* INQUIRY and REQUEST SENSE commands are explicitly allowed |
| 1758 | * to use unsupported LUNs; all others may not. */ |
| 1759 | if (common->cmnd[0] != SC_INQUIRY && |
| 1760 | common->cmnd[0] != SC_REQUEST_SENSE) { |
| 1761 | DBG(common, "unsupported LUN %d\n", common->lun); |
| 1762 | return -EINVAL; |
| 1763 | } |
| 1764 | } |
| 1765 | #if 0 |
| 1766 | /* If a unit attention condition exists, only INQUIRY and |
| 1767 | * REQUEST SENSE commands are allowed; anything else must fail. */ |
| 1768 | if (curlun && curlun->unit_attention_data != SS_NO_SENSE && |
| 1769 | common->cmnd[0] != SC_INQUIRY && |
| 1770 | common->cmnd[0] != SC_REQUEST_SENSE) { |
| 1771 | curlun->sense_data = curlun->unit_attention_data; |
| 1772 | curlun->unit_attention_data = SS_NO_SENSE; |
| 1773 | return -EINVAL; |
| 1774 | } |
| 1775 | #endif |
| 1776 | /* Check that only command bytes listed in the mask are non-zero */ |
| 1777 | common->cmnd[1] &= 0x1f; /* Mask away the LUN */ |
| 1778 | for (i = 1; i < cmnd_size; ++i) { |
| 1779 | if (common->cmnd[i] && !(mask & (1 << i))) { |
| 1780 | if (curlun) |
| 1781 | curlun->sense_data = SS_INVALID_FIELD_IN_CDB; |
| 1782 | return -EINVAL; |
| 1783 | } |
| 1784 | } |
| 1785 | |
| 1786 | return 0; |
| 1787 | } |
| 1788 | |
| 1789 | |
| 1790 | static int do_scsi_command(struct fsg_common *common) |
| 1791 | { |
| 1792 | struct fsg_buffhd *bh; |
| 1793 | int rc; |
| 1794 | int reply = -EINVAL; |
| 1795 | int i; |
| 1796 | static char unknown[16]; |
| 1797 | struct fsg_lun *curlun = &common->luns[common->lun]; |
| 1798 | |
| 1799 | dump_cdb(common); |
| 1800 | |
| 1801 | /* Wait for the next buffer to become available for data or status */ |
| 1802 | bh = common->next_buffhd_to_fill; |
| 1803 | common->next_buffhd_to_drain = bh; |
| 1804 | while (bh->state != BUF_STATE_EMPTY) { |
| 1805 | rc = sleep_thread(common); |
| 1806 | if (rc) |
| 1807 | return rc; |
| 1808 | } |
| 1809 | common->phase_error = 0; |
| 1810 | common->short_packet_received = 0; |
| 1811 | |
| 1812 | down_read(&common->filesem); /* We're using the backing file */ |
| 1813 | switch (common->cmnd[0]) { |
| 1814 | |
| 1815 | case SC_INQUIRY: |
| 1816 | common->data_size_from_cmnd = common->cmnd[4]; |
| 1817 | reply = check_command(common, 6, DATA_DIR_TO_HOST, |
| 1818 | (1<<4), 0, |
| 1819 | "INQUIRY"); |
| 1820 | if (reply == 0) |
| 1821 | reply = do_inquiry(common, bh); |
| 1822 | break; |
| 1823 | |
| 1824 | case SC_MODE_SELECT_6: |
| 1825 | common->data_size_from_cmnd = common->cmnd[4]; |
| 1826 | reply = check_command(common, 6, DATA_DIR_FROM_HOST, |
| 1827 | (1<<1) | (1<<4), 0, |
| 1828 | "MODE SELECT(6)"); |
| 1829 | if (reply == 0) |
| 1830 | reply = do_mode_select(common, bh); |
| 1831 | break; |
| 1832 | |
| 1833 | case SC_MODE_SELECT_10: |
| 1834 | common->data_size_from_cmnd = |
| 1835 | get_unaligned_be16(&common->cmnd[7]); |
| 1836 | reply = check_command(common, 10, DATA_DIR_FROM_HOST, |
| 1837 | (1<<1) | (3<<7), 0, |
| 1838 | "MODE SELECT(10)"); |
| 1839 | if (reply == 0) |
| 1840 | reply = do_mode_select(common, bh); |
| 1841 | break; |
| 1842 | |
| 1843 | case SC_MODE_SENSE_6: |
| 1844 | common->data_size_from_cmnd = common->cmnd[4]; |
| 1845 | reply = check_command(common, 6, DATA_DIR_TO_HOST, |
| 1846 | (1<<1) | (1<<2) | (1<<4), 0, |
| 1847 | "MODE SENSE(6)"); |
| 1848 | if (reply == 0) |
| 1849 | reply = do_mode_sense(common, bh); |
| 1850 | break; |
| 1851 | |
| 1852 | case SC_MODE_SENSE_10: |
| 1853 | common->data_size_from_cmnd = |
| 1854 | get_unaligned_be16(&common->cmnd[7]); |
| 1855 | reply = check_command(common, 10, DATA_DIR_TO_HOST, |
| 1856 | (1<<1) | (1<<2) | (3<<7), 0, |
| 1857 | "MODE SENSE(10)"); |
| 1858 | if (reply == 0) |
| 1859 | reply = do_mode_sense(common, bh); |
| 1860 | break; |
| 1861 | |
| 1862 | case SC_PREVENT_ALLOW_MEDIUM_REMOVAL: |
| 1863 | common->data_size_from_cmnd = 0; |
| 1864 | reply = check_command(common, 6, DATA_DIR_NONE, |
| 1865 | (1<<4), 0, |
| 1866 | "PREVENT-ALLOW MEDIUM REMOVAL"); |
| 1867 | if (reply == 0) |
| 1868 | reply = do_prevent_allow(common); |
| 1869 | break; |
| 1870 | |
| 1871 | case SC_READ_6: |
| 1872 | i = common->cmnd[4]; |
| 1873 | common->data_size_from_cmnd = (i == 0 ? 256 : i) << 9; |
| 1874 | reply = check_command(common, 6, DATA_DIR_TO_HOST, |
| 1875 | (7<<1) | (1<<4), 1, |
| 1876 | "READ(6)"); |
| 1877 | if (reply == 0) |
| 1878 | reply = do_read(common); |
| 1879 | break; |
| 1880 | |
| 1881 | case SC_READ_10: |
| 1882 | common->data_size_from_cmnd = |
| 1883 | get_unaligned_be16(&common->cmnd[7]) << 9; |
| 1884 | reply = check_command(common, 10, DATA_DIR_TO_HOST, |
| 1885 | (1<<1) | (0xf<<2) | (3<<7), 1, |
| 1886 | "READ(10)"); |
| 1887 | if (reply == 0) |
| 1888 | reply = do_read(common); |
| 1889 | break; |
| 1890 | |
| 1891 | case SC_READ_12: |
| 1892 | common->data_size_from_cmnd = |
| 1893 | get_unaligned_be32(&common->cmnd[6]) << 9; |
| 1894 | reply = check_command(common, 12, DATA_DIR_TO_HOST, |
| 1895 | (1<<1) | (0xf<<2) | (0xf<<6), 1, |
| 1896 | "READ(12)"); |
| 1897 | if (reply == 0) |
| 1898 | reply = do_read(common); |
| 1899 | break; |
| 1900 | |
| 1901 | case SC_READ_CAPACITY: |
| 1902 | common->data_size_from_cmnd = 8; |
| 1903 | reply = check_command(common, 10, DATA_DIR_TO_HOST, |
| 1904 | (0xf<<2) | (1<<8), 1, |
| 1905 | "READ CAPACITY"); |
| 1906 | if (reply == 0) |
| 1907 | reply = do_read_capacity(common, bh); |
| 1908 | break; |
| 1909 | |
| 1910 | case SC_READ_HEADER: |
| 1911 | if (!common->luns[common->lun].cdrom) |
| 1912 | goto unknown_cmnd; |
| 1913 | common->data_size_from_cmnd = |
| 1914 | get_unaligned_be16(&common->cmnd[7]); |
| 1915 | reply = check_command(common, 10, DATA_DIR_TO_HOST, |
| 1916 | (3<<7) | (0x1f<<1), 1, |
| 1917 | "READ HEADER"); |
| 1918 | if (reply == 0) |
| 1919 | reply = do_read_header(common, bh); |
| 1920 | break; |
| 1921 | |
| 1922 | case SC_READ_TOC: |
| 1923 | if (!common->luns[common->lun].cdrom) |
| 1924 | goto unknown_cmnd; |
| 1925 | common->data_size_from_cmnd = |
| 1926 | get_unaligned_be16(&common->cmnd[7]); |
| 1927 | reply = check_command(common, 10, DATA_DIR_TO_HOST, |
| 1928 | (7<<6) | (1<<1), 1, |
| 1929 | "READ TOC"); |
| 1930 | if (reply == 0) |
| 1931 | reply = do_read_toc(common, bh); |
| 1932 | break; |
| 1933 | |
| 1934 | case SC_READ_FORMAT_CAPACITIES: |
| 1935 | common->data_size_from_cmnd = |
| 1936 | get_unaligned_be16(&common->cmnd[7]); |
| 1937 | reply = check_command(common, 10, DATA_DIR_TO_HOST, |
| 1938 | (3<<7), 1, |
| 1939 | "READ FORMAT CAPACITIES"); |
| 1940 | if (reply == 0) |
| 1941 | reply = do_read_format_capacities(common, bh); |
| 1942 | break; |
| 1943 | |
| 1944 | case SC_REQUEST_SENSE: |
| 1945 | common->data_size_from_cmnd = common->cmnd[4]; |
| 1946 | reply = check_command(common, 6, DATA_DIR_TO_HOST, |
| 1947 | (1<<4), 0, |
| 1948 | "REQUEST SENSE"); |
| 1949 | if (reply == 0) |
| 1950 | reply = do_request_sense(common, bh); |
| 1951 | break; |
| 1952 | |
| 1953 | case SC_START_STOP_UNIT: |
| 1954 | common->data_size_from_cmnd = 0; |
| 1955 | reply = check_command(common, 6, DATA_DIR_NONE, |
| 1956 | (1<<1) | (1<<4), 0, |
| 1957 | "START-STOP UNIT"); |
| 1958 | if (reply == 0) |
| 1959 | reply = do_start_stop(common); |
| 1960 | break; |
| 1961 | |
| 1962 | case SC_SYNCHRONIZE_CACHE: |
| 1963 | common->data_size_from_cmnd = 0; |
| 1964 | reply = check_command(common, 10, DATA_DIR_NONE, |
| 1965 | (0xf<<2) | (3<<7), 1, |
| 1966 | "SYNCHRONIZE CACHE"); |
| 1967 | if (reply == 0) |
| 1968 | reply = do_synchronize_cache(common); |
| 1969 | break; |
| 1970 | |
| 1971 | case SC_TEST_UNIT_READY: |
| 1972 | common->data_size_from_cmnd = 0; |
| 1973 | reply = check_command(common, 6, DATA_DIR_NONE, |
| 1974 | 0, 1, |
| 1975 | "TEST UNIT READY"); |
| 1976 | break; |
| 1977 | |
| 1978 | /* Although optional, this command is used by MS-Windows. We |
| 1979 | * support a minimal version: BytChk must be 0. */ |
| 1980 | case SC_VERIFY: |
| 1981 | common->data_size_from_cmnd = 0; |
| 1982 | reply = check_command(common, 10, DATA_DIR_NONE, |
| 1983 | (1<<1) | (0xf<<2) | (3<<7), 1, |
| 1984 | "VERIFY"); |
| 1985 | if (reply == 0) |
| 1986 | reply = do_verify(common); |
| 1987 | break; |
| 1988 | |
| 1989 | case SC_WRITE_6: |
| 1990 | i = common->cmnd[4]; |
| 1991 | common->data_size_from_cmnd = (i == 0 ? 256 : i) << 9; |
| 1992 | reply = check_command(common, 6, DATA_DIR_FROM_HOST, |
| 1993 | (7<<1) | (1<<4), 1, |
| 1994 | "WRITE(6)"); |
| 1995 | if (reply == 0) |
| 1996 | reply = do_write(common); |
| 1997 | break; |
| 1998 | |
| 1999 | case SC_WRITE_10: |
| 2000 | common->data_size_from_cmnd = |
| 2001 | get_unaligned_be16(&common->cmnd[7]) << 9; |
| 2002 | reply = check_command(common, 10, DATA_DIR_FROM_HOST, |
| 2003 | (1<<1) | (0xf<<2) | (3<<7), 1, |
| 2004 | "WRITE(10)"); |
| 2005 | if (reply == 0) |
| 2006 | reply = do_write(common); |
| 2007 | break; |
| 2008 | |
| 2009 | case SC_WRITE_12: |
| 2010 | common->data_size_from_cmnd = |
| 2011 | get_unaligned_be32(&common->cmnd[6]) << 9; |
| 2012 | reply = check_command(common, 12, DATA_DIR_FROM_HOST, |
| 2013 | (1<<1) | (0xf<<2) | (0xf<<6), 1, |
| 2014 | "WRITE(12)"); |
| 2015 | if (reply == 0) |
| 2016 | reply = do_write(common); |
| 2017 | break; |
| 2018 | |
| 2019 | /* Some mandatory commands that we recognize but don't implement. |
| 2020 | * They don't mean much in this setting. It's left as an exercise |
| 2021 | * for anyone interested to implement RESERVE and RELEASE in terms |
| 2022 | * of Posix locks. */ |
| 2023 | case SC_FORMAT_UNIT: |
| 2024 | case SC_RELEASE: |
| 2025 | case SC_RESERVE: |
| 2026 | case SC_SEND_DIAGNOSTIC: |
| 2027 | /* Fall through */ |
| 2028 | |
| 2029 | default: |
| 2030 | unknown_cmnd: |
| 2031 | common->data_size_from_cmnd = 0; |
| 2032 | sprintf(unknown, "Unknown x%02x", common->cmnd[0]); |
| 2033 | reply = check_command(common, common->cmnd_size, |
| 2034 | DATA_DIR_UNKNOWN, 0xff, 0, unknown); |
| 2035 | if (reply == 0) { |
| 2036 | curlun->sense_data = SS_INVALID_COMMAND; |
| 2037 | reply = -EINVAL; |
| 2038 | } |
| 2039 | break; |
| 2040 | } |
| 2041 | up_read(&common->filesem); |
| 2042 | |
| 2043 | if (reply == -EINTR) |
| 2044 | return -EINTR; |
| 2045 | |
| 2046 | /* Set up the single reply buffer for finish_reply() */ |
| 2047 | if (reply == -EINVAL) |
| 2048 | reply = 0; /* Error reply length */ |
| 2049 | if (reply >= 0 && common->data_dir == DATA_DIR_TO_HOST) { |
| 2050 | reply = min((u32) reply, common->data_size_from_cmnd); |
| 2051 | bh->inreq->length = reply; |
| 2052 | bh->state = BUF_STATE_FULL; |
| 2053 | common->residue -= reply; |
| 2054 | } /* Otherwise it's already set */ |
| 2055 | |
| 2056 | return 0; |
| 2057 | } |
| 2058 | |
| 2059 | /*-------------------------------------------------------------------------*/ |
| 2060 | |
| 2061 | static int received_cbw(struct fsg_dev *fsg, struct fsg_buffhd *bh) |
| 2062 | { |
| 2063 | struct usb_request *req = bh->outreq; |
| 2064 | struct fsg_bulk_cb_wrap *cbw = req->buf; |
| 2065 | struct fsg_common *common = fsg->common; |
| 2066 | |
| 2067 | /* Was this a real packet? Should it be ignored? */ |
| 2068 | if (req->status || test_bit(IGNORE_BULK_OUT, &fsg->atomic_bitflags)) |
| 2069 | return -EINVAL; |
| 2070 | |
| 2071 | /* Is the CBW valid? */ |
| 2072 | if (req->actual != USB_BULK_CB_WRAP_LEN || |
| 2073 | cbw->Signature != cpu_to_le32( |
| 2074 | USB_BULK_CB_SIG)) { |
| 2075 | DBG(fsg, "invalid CBW: len %u sig 0x%x\n", |
| 2076 | req->actual, |
| 2077 | le32_to_cpu(cbw->Signature)); |
| 2078 | |
| 2079 | /* The Bulk-only spec says we MUST stall the IN endpoint |
| 2080 | * (6.6.1), so it's unavoidable. It also says we must |
| 2081 | * retain this state until the next reset, but there's |
| 2082 | * no way to tell the controller driver it should ignore |
| 2083 | * Clear-Feature(HALT) requests. |
| 2084 | * |
| 2085 | * We aren't required to halt the OUT endpoint; instead |
| 2086 | * we can simply accept and discard any data received |
| 2087 | * until the next reset. */ |
| 2088 | wedge_bulk_in_endpoint(fsg); |
| 2089 | set_bit(IGNORE_BULK_OUT, &fsg->atomic_bitflags); |
| 2090 | return -EINVAL; |
| 2091 | } |
| 2092 | |
| 2093 | /* Is the CBW meaningful? */ |
| 2094 | if (cbw->Lun >= FSG_MAX_LUNS || cbw->Flags & ~USB_BULK_IN_FLAG || |
| 2095 | cbw->Length <= 0 || cbw->Length > MAX_COMMAND_SIZE) { |
| 2096 | DBG(fsg, "non-meaningful CBW: lun = %u, flags = 0x%x, " |
| 2097 | "cmdlen %u\n", |
| 2098 | cbw->Lun, cbw->Flags, cbw->Length); |
| 2099 | |
| 2100 | /* We can do anything we want here, so let's stall the |
| 2101 | * bulk pipes if we are allowed to. */ |
| 2102 | if (common->can_stall) { |
| 2103 | fsg_set_halt(fsg, fsg->bulk_out); |
| 2104 | halt_bulk_in_endpoint(fsg); |
| 2105 | } |
| 2106 | return -EINVAL; |
| 2107 | } |
| 2108 | |
| 2109 | /* Save the command for later */ |
| 2110 | common->cmnd_size = cbw->Length; |
| 2111 | memcpy(common->cmnd, cbw->CDB, common->cmnd_size); |
| 2112 | if (cbw->Flags & USB_BULK_IN_FLAG) |
| 2113 | common->data_dir = DATA_DIR_TO_HOST; |
| 2114 | else |
| 2115 | common->data_dir = DATA_DIR_FROM_HOST; |
| 2116 | common->data_size = le32_to_cpu(cbw->DataTransferLength); |
| 2117 | if (common->data_size == 0) |
| 2118 | common->data_dir = DATA_DIR_NONE; |
| 2119 | common->lun = cbw->Lun; |
| 2120 | common->tag = cbw->Tag; |
| 2121 | return 0; |
| 2122 | } |
| 2123 | |
| 2124 | |
| 2125 | static int get_next_command(struct fsg_common *common) |
| 2126 | { |
| 2127 | struct fsg_buffhd *bh; |
| 2128 | int rc = 0; |
| 2129 | |
| 2130 | /* Wait for the next buffer to become available */ |
| 2131 | bh = common->next_buffhd_to_fill; |
| 2132 | while (bh->state != BUF_STATE_EMPTY) { |
| 2133 | rc = sleep_thread(common); |
| 2134 | if (rc) |
| 2135 | return rc; |
| 2136 | } |
| 2137 | |
| 2138 | /* Queue a request to read a Bulk-only CBW */ |
| 2139 | set_bulk_out_req_length(common, bh, USB_BULK_CB_WRAP_LEN); |
| 2140 | bh->outreq->short_not_ok = 1; |
| 2141 | START_TRANSFER_OR(common, bulk_out, bh->outreq, |
| 2142 | &bh->outreq_busy, &bh->state) |
| 2143 | /* Don't know what to do if common->fsg is NULL */ |
| 2144 | return -EIO; |
| 2145 | |
| 2146 | /* We will drain the buffer in software, which means we |
| 2147 | * can reuse it for the next filling. No need to advance |
| 2148 | * next_buffhd_to_fill. */ |
| 2149 | |
| 2150 | /* Wait for the CBW to arrive */ |
| 2151 | while (bh->state != BUF_STATE_FULL) { |
| 2152 | rc = sleep_thread(common); |
| 2153 | if (rc) |
| 2154 | return rc; |
| 2155 | } |
| 2156 | |
| 2157 | rc = fsg_is_set(common) ? received_cbw(common->fsg, bh) : -EIO; |
| 2158 | bh->state = BUF_STATE_EMPTY; |
| 2159 | |
| 2160 | return rc; |
| 2161 | } |
| 2162 | |
| 2163 | |
| 2164 | /*-------------------------------------------------------------------------*/ |
| 2165 | |
| 2166 | static int enable_endpoint(struct fsg_common *common, struct usb_ep *ep, |
| 2167 | const struct usb_endpoint_descriptor *d) |
| 2168 | { |
| 2169 | int rc; |
| 2170 | |
| 2171 | ep->driver_data = common; |
| 2172 | rc = usb_ep_enable(ep, d); |
| 2173 | if (rc) |
| 2174 | ERROR(common, "can't enable %s, result %d\n", ep->name, rc); |
| 2175 | return rc; |
| 2176 | } |
| 2177 | |
| 2178 | static int alloc_request(struct fsg_common *common, struct usb_ep *ep, |
| 2179 | struct usb_request **preq) |
| 2180 | { |
| 2181 | *preq = usb_ep_alloc_request(ep, GFP_ATOMIC); |
| 2182 | if (*preq) |
| 2183 | return 0; |
| 2184 | ERROR(common, "can't allocate request for %s\n", ep->name); |
| 2185 | return -ENOMEM; |
| 2186 | } |
| 2187 | |
| 2188 | /* Reset interface setting and re-init endpoint state (toggle etc). */ |
| 2189 | static int do_set_interface(struct fsg_common *common, struct fsg_dev *new_fsg) |
| 2190 | { |
| 2191 | const struct usb_endpoint_descriptor *d; |
| 2192 | struct fsg_dev *fsg; |
| 2193 | int i, rc = 0; |
| 2194 | |
| 2195 | if (common->running) |
| 2196 | DBG(common, "reset interface\n"); |
| 2197 | |
| 2198 | reset: |
| 2199 | /* Deallocate the requests */ |
| 2200 | if (common->fsg) { |
| 2201 | fsg = common->fsg; |
| 2202 | |
| 2203 | for (i = 0; i < FSG_NUM_BUFFERS; ++i) { |
| 2204 | struct fsg_buffhd *bh = &common->buffhds[i]; |
| 2205 | |
| 2206 | if (bh->inreq) { |
| 2207 | usb_ep_free_request(fsg->bulk_in, bh->inreq); |
| 2208 | bh->inreq = NULL; |
| 2209 | } |
| 2210 | if (bh->outreq) { |
| 2211 | usb_ep_free_request(fsg->bulk_out, bh->outreq); |
| 2212 | bh->outreq = NULL; |
| 2213 | } |
| 2214 | } |
| 2215 | |
| 2216 | /* Disable the endpoints */ |
| 2217 | if (fsg->bulk_in_enabled) { |
| 2218 | usb_ep_disable(fsg->bulk_in); |
| 2219 | fsg->bulk_in_enabled = 0; |
| 2220 | } |
| 2221 | if (fsg->bulk_out_enabled) { |
| 2222 | usb_ep_disable(fsg->bulk_out); |
| 2223 | fsg->bulk_out_enabled = 0; |
| 2224 | } |
| 2225 | |
| 2226 | common->fsg = NULL; |
| 2227 | /* wake_up(&common->fsg_wait); */ |
| 2228 | } |
| 2229 | |
| 2230 | common->running = 0; |
| 2231 | if (!new_fsg || rc) |
| 2232 | return rc; |
| 2233 | |
| 2234 | common->fsg = new_fsg; |
| 2235 | fsg = common->fsg; |
| 2236 | |
| 2237 | /* Enable the endpoints */ |
| 2238 | d = fsg_ep_desc(common->gadget, |
| 2239 | &fsg_fs_bulk_in_desc, &fsg_hs_bulk_in_desc); |
| 2240 | rc = enable_endpoint(common, fsg->bulk_in, d); |
| 2241 | if (rc) |
| 2242 | goto reset; |
| 2243 | fsg->bulk_in_enabled = 1; |
| 2244 | |
| 2245 | d = fsg_ep_desc(common->gadget, |
| 2246 | &fsg_fs_bulk_out_desc, &fsg_hs_bulk_out_desc); |
| 2247 | rc = enable_endpoint(common, fsg->bulk_out, d); |
| 2248 | if (rc) |
| 2249 | goto reset; |
| 2250 | fsg->bulk_out_enabled = 1; |
Vivek Gautam | 1d62db8 | 2013-05-13 15:53:38 +0530 | [diff] [blame] | 2251 | common->bulk_out_maxpacket = |
| 2252 | le16_to_cpu(get_unaligned(&d->wMaxPacketSize)); |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 2253 | clear_bit(IGNORE_BULK_OUT, &fsg->atomic_bitflags); |
| 2254 | |
| 2255 | /* Allocate the requests */ |
| 2256 | for (i = 0; i < FSG_NUM_BUFFERS; ++i) { |
| 2257 | struct fsg_buffhd *bh = &common->buffhds[i]; |
| 2258 | |
| 2259 | rc = alloc_request(common, fsg->bulk_in, &bh->inreq); |
| 2260 | if (rc) |
| 2261 | goto reset; |
| 2262 | rc = alloc_request(common, fsg->bulk_out, &bh->outreq); |
| 2263 | if (rc) |
| 2264 | goto reset; |
| 2265 | bh->inreq->buf = bh->outreq->buf = bh->buf; |
| 2266 | bh->inreq->context = bh->outreq->context = bh; |
| 2267 | bh->inreq->complete = bulk_in_complete; |
| 2268 | bh->outreq->complete = bulk_out_complete; |
| 2269 | } |
| 2270 | |
| 2271 | common->running = 1; |
| 2272 | |
| 2273 | return rc; |
| 2274 | } |
| 2275 | |
| 2276 | |
| 2277 | /****************************** ALT CONFIGS ******************************/ |
| 2278 | |
| 2279 | |
| 2280 | static int fsg_set_alt(struct usb_function *f, unsigned intf, unsigned alt) |
| 2281 | { |
| 2282 | struct fsg_dev *fsg = fsg_from_func(f); |
| 2283 | fsg->common->new_fsg = fsg; |
| 2284 | raise_exception(fsg->common, FSG_STATE_CONFIG_CHANGE); |
| 2285 | return 0; |
| 2286 | } |
| 2287 | |
| 2288 | static void fsg_disable(struct usb_function *f) |
| 2289 | { |
| 2290 | struct fsg_dev *fsg = fsg_from_func(f); |
| 2291 | fsg->common->new_fsg = NULL; |
| 2292 | raise_exception(fsg->common, FSG_STATE_CONFIG_CHANGE); |
| 2293 | } |
| 2294 | |
| 2295 | /*-------------------------------------------------------------------------*/ |
| 2296 | |
| 2297 | static void handle_exception(struct fsg_common *common) |
| 2298 | { |
| 2299 | int i; |
| 2300 | struct fsg_buffhd *bh; |
| 2301 | enum fsg_state old_state; |
| 2302 | struct fsg_lun *curlun; |
| 2303 | unsigned int exception_req_tag; |
| 2304 | |
| 2305 | /* Cancel all the pending transfers */ |
| 2306 | if (common->fsg) { |
| 2307 | for (i = 0; i < FSG_NUM_BUFFERS; ++i) { |
| 2308 | bh = &common->buffhds[i]; |
| 2309 | if (bh->inreq_busy) |
| 2310 | usb_ep_dequeue(common->fsg->bulk_in, bh->inreq); |
| 2311 | if (bh->outreq_busy) |
| 2312 | usb_ep_dequeue(common->fsg->bulk_out, |
| 2313 | bh->outreq); |
| 2314 | } |
| 2315 | |
| 2316 | /* Wait until everything is idle */ |
| 2317 | for (;;) { |
| 2318 | int num_active = 0; |
| 2319 | for (i = 0; i < FSG_NUM_BUFFERS; ++i) { |
| 2320 | bh = &common->buffhds[i]; |
| 2321 | num_active += bh->inreq_busy + bh->outreq_busy; |
| 2322 | } |
| 2323 | if (num_active == 0) |
| 2324 | break; |
| 2325 | if (sleep_thread(common)) |
| 2326 | return; |
| 2327 | } |
| 2328 | |
| 2329 | /* Clear out the controller's fifos */ |
| 2330 | if (common->fsg->bulk_in_enabled) |
| 2331 | usb_ep_fifo_flush(common->fsg->bulk_in); |
| 2332 | if (common->fsg->bulk_out_enabled) |
| 2333 | usb_ep_fifo_flush(common->fsg->bulk_out); |
| 2334 | } |
| 2335 | |
| 2336 | /* Reset the I/O buffer states and pointers, the SCSI |
| 2337 | * state, and the exception. Then invoke the handler. */ |
| 2338 | |
| 2339 | for (i = 0; i < FSG_NUM_BUFFERS; ++i) { |
| 2340 | bh = &common->buffhds[i]; |
| 2341 | bh->state = BUF_STATE_EMPTY; |
| 2342 | } |
| 2343 | common->next_buffhd_to_fill = &common->buffhds[0]; |
| 2344 | common->next_buffhd_to_drain = &common->buffhds[0]; |
| 2345 | exception_req_tag = common->exception_req_tag; |
| 2346 | old_state = common->state; |
| 2347 | |
| 2348 | if (old_state == FSG_STATE_ABORT_BULK_OUT) |
| 2349 | common->state = FSG_STATE_STATUS_PHASE; |
| 2350 | else { |
| 2351 | for (i = 0; i < common->nluns; ++i) { |
| 2352 | curlun = &common->luns[i]; |
| 2353 | curlun->sense_data = SS_NO_SENSE; |
| 2354 | curlun->info_valid = 0; |
| 2355 | } |
| 2356 | common->state = FSG_STATE_IDLE; |
| 2357 | } |
| 2358 | |
| 2359 | /* Carry out any extra actions required for the exception */ |
| 2360 | switch (old_state) { |
| 2361 | case FSG_STATE_ABORT_BULK_OUT: |
| 2362 | send_status(common); |
| 2363 | |
| 2364 | if (common->state == FSG_STATE_STATUS_PHASE) |
| 2365 | common->state = FSG_STATE_IDLE; |
| 2366 | break; |
| 2367 | |
| 2368 | case FSG_STATE_RESET: |
| 2369 | /* In case we were forced against our will to halt a |
| 2370 | * bulk endpoint, clear the halt now. (The SuperH UDC |
| 2371 | * requires this.) */ |
| 2372 | if (!fsg_is_set(common)) |
| 2373 | break; |
| 2374 | if (test_and_clear_bit(IGNORE_BULK_OUT, |
| 2375 | &common->fsg->atomic_bitflags)) |
| 2376 | usb_ep_clear_halt(common->fsg->bulk_in); |
| 2377 | |
| 2378 | if (common->ep0_req_tag == exception_req_tag) |
| 2379 | ep0_queue(common); /* Complete the status stage */ |
| 2380 | |
| 2381 | break; |
| 2382 | |
| 2383 | case FSG_STATE_CONFIG_CHANGE: |
| 2384 | do_set_interface(common, common->new_fsg); |
| 2385 | break; |
| 2386 | |
| 2387 | case FSG_STATE_EXIT: |
| 2388 | case FSG_STATE_TERMINATED: |
| 2389 | do_set_interface(common, NULL); /* Free resources */ |
| 2390 | common->state = FSG_STATE_TERMINATED; /* Stop the thread */ |
| 2391 | break; |
| 2392 | |
| 2393 | case FSG_STATE_INTERFACE_CHANGE: |
| 2394 | case FSG_STATE_DISCONNECT: |
| 2395 | case FSG_STATE_COMMAND_PHASE: |
| 2396 | case FSG_STATE_DATA_PHASE: |
| 2397 | case FSG_STATE_STATUS_PHASE: |
| 2398 | case FSG_STATE_IDLE: |
| 2399 | break; |
| 2400 | } |
| 2401 | } |
| 2402 | |
| 2403 | /*-------------------------------------------------------------------------*/ |
| 2404 | |
| 2405 | int fsg_main_thread(void *common_) |
| 2406 | { |
Przemyslaw Marczak | 06ef7cc | 2013-10-23 14:30:46 +0200 | [diff] [blame] | 2407 | int ret; |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 2408 | struct fsg_common *common = the_fsg_common; |
| 2409 | /* The main loop */ |
| 2410 | do { |
| 2411 | if (exception_in_progress(common)) { |
| 2412 | handle_exception(common); |
| 2413 | continue; |
| 2414 | } |
| 2415 | |
| 2416 | if (!common->running) { |
Przemyslaw Marczak | 06ef7cc | 2013-10-23 14:30:46 +0200 | [diff] [blame] | 2417 | ret = sleep_thread(common); |
| 2418 | if (ret) |
| 2419 | return ret; |
| 2420 | |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 2421 | continue; |
| 2422 | } |
| 2423 | |
Przemyslaw Marczak | 06ef7cc | 2013-10-23 14:30:46 +0200 | [diff] [blame] | 2424 | ret = get_next_command(common); |
| 2425 | if (ret) |
| 2426 | return ret; |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 2427 | |
| 2428 | if (!exception_in_progress(common)) |
| 2429 | common->state = FSG_STATE_DATA_PHASE; |
| 2430 | |
| 2431 | if (do_scsi_command(common) || finish_reply(common)) |
| 2432 | continue; |
| 2433 | |
| 2434 | if (!exception_in_progress(common)) |
| 2435 | common->state = FSG_STATE_STATUS_PHASE; |
| 2436 | |
| 2437 | if (send_status(common)) |
| 2438 | continue; |
| 2439 | |
| 2440 | if (!exception_in_progress(common)) |
| 2441 | common->state = FSG_STATE_IDLE; |
| 2442 | } while (0); |
| 2443 | |
| 2444 | common->thread_task = NULL; |
| 2445 | |
| 2446 | return 0; |
| 2447 | } |
| 2448 | |
| 2449 | static void fsg_common_release(struct kref *ref); |
| 2450 | |
| 2451 | static struct fsg_common *fsg_common_init(struct fsg_common *common, |
| 2452 | struct usb_composite_dev *cdev) |
| 2453 | { |
| 2454 | struct usb_gadget *gadget = cdev->gadget; |
| 2455 | struct fsg_buffhd *bh; |
| 2456 | struct fsg_lun *curlun; |
| 2457 | int nluns, i, rc; |
| 2458 | |
| 2459 | /* Find out how many LUNs there should be */ |
Stephen Warren | 9e7d588 | 2015-12-07 11:38:50 -0700 | [diff] [blame] | 2460 | nluns = ums_count; |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 2461 | if (nluns < 1 || nluns > FSG_MAX_LUNS) { |
| 2462 | printf("invalid number of LUNs: %u\n", nluns); |
| 2463 | return ERR_PTR(-EINVAL); |
| 2464 | } |
| 2465 | |
| 2466 | /* Allocate? */ |
| 2467 | if (!common) { |
Jeroen Hofstee | 6931bff | 2014-06-09 15:28:59 +0200 | [diff] [blame] | 2468 | common = calloc(sizeof(*common), 1); |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 2469 | if (!common) |
| 2470 | return ERR_PTR(-ENOMEM); |
| 2471 | common->free_storage_on_release = 1; |
| 2472 | } else { |
Jeroen Hofstee | 6931bff | 2014-06-09 15:28:59 +0200 | [diff] [blame] | 2473 | memset(common, 0, sizeof(*common)); |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 2474 | common->free_storage_on_release = 0; |
| 2475 | } |
| 2476 | |
| 2477 | common->ops = NULL; |
| 2478 | common->private_data = NULL; |
| 2479 | |
| 2480 | common->gadget = gadget; |
| 2481 | common->ep0 = gadget->ep0; |
| 2482 | common->ep0req = cdev->req; |
| 2483 | |
| 2484 | /* Maybe allocate device-global string IDs, and patch descriptors */ |
| 2485 | if (fsg_strings[FSG_STRING_INTERFACE].id == 0) { |
| 2486 | rc = usb_string_id(cdev); |
| 2487 | if (unlikely(rc < 0)) |
| 2488 | goto error_release; |
| 2489 | fsg_strings[FSG_STRING_INTERFACE].id = rc; |
| 2490 | fsg_intf_desc.iInterface = rc; |
| 2491 | } |
| 2492 | |
| 2493 | /* Create the LUNs, open their backing files, and register the |
| 2494 | * LUN devices in sysfs. */ |
| 2495 | curlun = calloc(nluns, sizeof *curlun); |
| 2496 | if (!curlun) { |
| 2497 | rc = -ENOMEM; |
| 2498 | goto error_release; |
| 2499 | } |
| 2500 | common->nluns = nluns; |
| 2501 | |
| 2502 | for (i = 0; i < nluns; i++) { |
| 2503 | common->luns[i].removable = 1; |
| 2504 | |
Stephen Warren | 9e7d588 | 2015-12-07 11:38:50 -0700 | [diff] [blame] | 2505 | rc = fsg_lun_open(&common->luns[i], ums[i].num_sectors, ""); |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 2506 | if (rc) |
| 2507 | goto error_luns; |
| 2508 | } |
| 2509 | common->lun = 0; |
| 2510 | |
| 2511 | /* Data buffers cyclic list */ |
| 2512 | bh = common->buffhds; |
| 2513 | |
| 2514 | i = FSG_NUM_BUFFERS; |
| 2515 | goto buffhds_first_it; |
| 2516 | do { |
| 2517 | bh->next = bh + 1; |
| 2518 | ++bh; |
| 2519 | buffhds_first_it: |
| 2520 | bh->inreq_busy = 0; |
| 2521 | bh->outreq_busy = 0; |
Lukasz Majewski | 0575113 | 2014-02-05 10:10:41 +0100 | [diff] [blame] | 2522 | bh->buf = memalign(CONFIG_SYS_CACHELINE_SIZE, FSG_BUFLEN); |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 2523 | if (unlikely(!bh->buf)) { |
| 2524 | rc = -ENOMEM; |
| 2525 | goto error_release; |
| 2526 | } |
| 2527 | } while (--i); |
| 2528 | bh->next = common->buffhds; |
| 2529 | |
| 2530 | snprintf(common->inquiry_string, sizeof common->inquiry_string, |
| 2531 | "%-8s%-16s%04x", |
| 2532 | "Linux ", |
| 2533 | "File-Store Gadget", |
| 2534 | 0xffff); |
| 2535 | |
| 2536 | /* Some peripheral controllers are known not to be able to |
| 2537 | * halt bulk endpoints correctly. If one of them is present, |
| 2538 | * disable stalls. |
| 2539 | */ |
| 2540 | |
| 2541 | /* Tell the thread to start working */ |
| 2542 | common->thread_task = |
| 2543 | kthread_create(fsg_main_thread, common, |
| 2544 | OR(cfg->thread_name, "file-storage")); |
| 2545 | if (IS_ERR(common->thread_task)) { |
| 2546 | rc = PTR_ERR(common->thread_task); |
| 2547 | goto error_release; |
| 2548 | } |
| 2549 | |
| 2550 | #undef OR |
| 2551 | /* Information */ |
| 2552 | INFO(common, FSG_DRIVER_DESC ", version: " FSG_DRIVER_VERSION "\n"); |
| 2553 | INFO(common, "Number of LUNs=%d\n", common->nluns); |
| 2554 | |
| 2555 | return common; |
| 2556 | |
| 2557 | error_luns: |
| 2558 | common->nluns = i + 1; |
| 2559 | error_release: |
| 2560 | common->state = FSG_STATE_TERMINATED; /* The thread is dead */ |
| 2561 | /* Call fsg_common_release() directly, ref might be not |
| 2562 | * initialised */ |
| 2563 | fsg_common_release(&common->ref); |
| 2564 | return ERR_PTR(rc); |
| 2565 | } |
| 2566 | |
| 2567 | static void fsg_common_release(struct kref *ref) |
| 2568 | { |
| 2569 | struct fsg_common *common = container_of(ref, struct fsg_common, ref); |
| 2570 | |
| 2571 | /* If the thread isn't already dead, tell it to exit now */ |
| 2572 | if (common->state != FSG_STATE_TERMINATED) { |
| 2573 | raise_exception(common, FSG_STATE_EXIT); |
| 2574 | wait_for_completion(&common->thread_notifier); |
| 2575 | } |
| 2576 | |
| 2577 | if (likely(common->luns)) { |
| 2578 | struct fsg_lun *lun = common->luns; |
| 2579 | unsigned i = common->nluns; |
| 2580 | |
| 2581 | /* In error recovery common->nluns may be zero. */ |
| 2582 | for (; i; --i, ++lun) |
| 2583 | fsg_lun_close(lun); |
| 2584 | |
| 2585 | kfree(common->luns); |
| 2586 | } |
| 2587 | |
| 2588 | { |
| 2589 | struct fsg_buffhd *bh = common->buffhds; |
| 2590 | unsigned i = FSG_NUM_BUFFERS; |
| 2591 | do { |
| 2592 | kfree(bh->buf); |
| 2593 | } while (++bh, --i); |
| 2594 | } |
| 2595 | |
| 2596 | if (common->free_storage_on_release) |
| 2597 | kfree(common); |
| 2598 | } |
| 2599 | |
| 2600 | |
| 2601 | /*-------------------------------------------------------------------------*/ |
| 2602 | |
| 2603 | /** |
| 2604 | * usb_copy_descriptors - copy a vector of USB descriptors |
| 2605 | * @src: null-terminated vector to copy |
| 2606 | * Context: initialization code, which may sleep |
| 2607 | * |
| 2608 | * This makes a copy of a vector of USB descriptors. Its primary use |
| 2609 | * is to support usb_function objects which can have multiple copies, |
| 2610 | * each needing different descriptors. Functions may have static |
| 2611 | * tables of descriptors, which are used as templates and customized |
| 2612 | * with identifiers (for interfaces, strings, endpoints, and more) |
| 2613 | * as needed by a given function instance. |
| 2614 | */ |
| 2615 | struct usb_descriptor_header ** |
| 2616 | usb_copy_descriptors(struct usb_descriptor_header **src) |
| 2617 | { |
| 2618 | struct usb_descriptor_header **tmp; |
| 2619 | unsigned bytes; |
| 2620 | unsigned n_desc; |
| 2621 | void *mem; |
| 2622 | struct usb_descriptor_header **ret; |
| 2623 | |
| 2624 | /* count descriptors and their sizes; then add vector size */ |
| 2625 | for (bytes = 0, n_desc = 0, tmp = src; *tmp; tmp++, n_desc++) |
| 2626 | bytes += (*tmp)->bLength; |
| 2627 | bytes += (n_desc + 1) * sizeof(*tmp); |
| 2628 | |
Lukasz Majewski | 0575113 | 2014-02-05 10:10:41 +0100 | [diff] [blame] | 2629 | mem = memalign(CONFIG_SYS_CACHELINE_SIZE, bytes); |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 2630 | if (!mem) |
| 2631 | return NULL; |
| 2632 | |
| 2633 | /* fill in pointers starting at "tmp", |
| 2634 | * to descriptors copied starting at "mem"; |
| 2635 | * and return "ret" |
| 2636 | */ |
| 2637 | tmp = mem; |
| 2638 | ret = mem; |
| 2639 | mem += (n_desc + 1) * sizeof(*tmp); |
| 2640 | while (*src) { |
| 2641 | memcpy(mem, *src, (*src)->bLength); |
| 2642 | *tmp = mem; |
| 2643 | tmp++; |
| 2644 | mem += (*src)->bLength; |
| 2645 | src++; |
| 2646 | } |
| 2647 | *tmp = NULL; |
| 2648 | |
| 2649 | return ret; |
| 2650 | } |
| 2651 | |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 2652 | static void fsg_unbind(struct usb_configuration *c, struct usb_function *f) |
| 2653 | { |
| 2654 | struct fsg_dev *fsg = fsg_from_func(f); |
| 2655 | |
| 2656 | DBG(fsg, "unbind\n"); |
| 2657 | if (fsg->common->fsg == fsg) { |
| 2658 | fsg->common->new_fsg = NULL; |
| 2659 | raise_exception(fsg->common, FSG_STATE_CONFIG_CHANGE); |
| 2660 | } |
| 2661 | |
| 2662 | free(fsg->function.descriptors); |
| 2663 | free(fsg->function.hs_descriptors); |
| 2664 | kfree(fsg); |
| 2665 | } |
| 2666 | |
| 2667 | static int fsg_bind(struct usb_configuration *c, struct usb_function *f) |
| 2668 | { |
| 2669 | struct fsg_dev *fsg = fsg_from_func(f); |
| 2670 | struct usb_gadget *gadget = c->cdev->gadget; |
| 2671 | int i; |
| 2672 | struct usb_ep *ep; |
| 2673 | fsg->gadget = gadget; |
| 2674 | |
| 2675 | /* New interface */ |
| 2676 | i = usb_interface_id(c, f); |
| 2677 | if (i < 0) |
| 2678 | return i; |
| 2679 | fsg_intf_desc.bInterfaceNumber = i; |
| 2680 | fsg->interface_number = i; |
| 2681 | |
| 2682 | /* Find all the endpoints we will use */ |
| 2683 | ep = usb_ep_autoconfig(gadget, &fsg_fs_bulk_in_desc); |
| 2684 | if (!ep) |
| 2685 | goto autoconf_fail; |
| 2686 | ep->driver_data = fsg->common; /* claim the endpoint */ |
| 2687 | fsg->bulk_in = ep; |
| 2688 | |
| 2689 | ep = usb_ep_autoconfig(gadget, &fsg_fs_bulk_out_desc); |
| 2690 | if (!ep) |
| 2691 | goto autoconf_fail; |
| 2692 | ep->driver_data = fsg->common; /* claim the endpoint */ |
| 2693 | fsg->bulk_out = ep; |
| 2694 | |
| 2695 | /* Copy descriptors */ |
| 2696 | f->descriptors = usb_copy_descriptors(fsg_fs_function); |
| 2697 | if (unlikely(!f->descriptors)) |
| 2698 | return -ENOMEM; |
| 2699 | |
| 2700 | if (gadget_is_dualspeed(gadget)) { |
| 2701 | /* Assume endpoint addresses are the same for both speeds */ |
| 2702 | fsg_hs_bulk_in_desc.bEndpointAddress = |
| 2703 | fsg_fs_bulk_in_desc.bEndpointAddress; |
| 2704 | fsg_hs_bulk_out_desc.bEndpointAddress = |
| 2705 | fsg_fs_bulk_out_desc.bEndpointAddress; |
| 2706 | f->hs_descriptors = usb_copy_descriptors(fsg_hs_function); |
| 2707 | if (unlikely(!f->hs_descriptors)) { |
| 2708 | free(f->descriptors); |
| 2709 | return -ENOMEM; |
| 2710 | } |
| 2711 | } |
| 2712 | return 0; |
| 2713 | |
| 2714 | autoconf_fail: |
| 2715 | ERROR(fsg, "unable to autoconfigure all endpoints\n"); |
| 2716 | return -ENOTSUPP; |
| 2717 | } |
| 2718 | |
| 2719 | |
| 2720 | /****************************** ADD FUNCTION ******************************/ |
| 2721 | |
| 2722 | static struct usb_gadget_strings *fsg_strings_array[] = { |
| 2723 | &fsg_stringtab, |
| 2724 | NULL, |
| 2725 | }; |
| 2726 | |
| 2727 | static int fsg_bind_config(struct usb_composite_dev *cdev, |
| 2728 | struct usb_configuration *c, |
| 2729 | struct fsg_common *common) |
| 2730 | { |
| 2731 | struct fsg_dev *fsg; |
| 2732 | int rc; |
| 2733 | |
| 2734 | fsg = calloc(1, sizeof *fsg); |
| 2735 | if (!fsg) |
| 2736 | return -ENOMEM; |
| 2737 | fsg->function.name = FSG_DRIVER_DESC; |
| 2738 | fsg->function.strings = fsg_strings_array; |
| 2739 | fsg->function.bind = fsg_bind; |
| 2740 | fsg->function.unbind = fsg_unbind; |
| 2741 | fsg->function.setup = fsg_setup; |
| 2742 | fsg->function.set_alt = fsg_set_alt; |
| 2743 | fsg->function.disable = fsg_disable; |
| 2744 | |
| 2745 | fsg->common = common; |
| 2746 | common->fsg = fsg; |
| 2747 | /* Our caller holds a reference to common structure so we |
| 2748 | * don't have to be worry about it being freed until we return |
| 2749 | * from this function. So instead of incrementing counter now |
| 2750 | * and decrement in error recovery we increment it only when |
| 2751 | * call to usb_add_function() was successful. */ |
| 2752 | |
| 2753 | rc = usb_add_function(c, &fsg->function); |
| 2754 | |
| 2755 | if (rc) |
| 2756 | kfree(fsg); |
| 2757 | |
| 2758 | return rc; |
| 2759 | } |
| 2760 | |
| 2761 | int fsg_add(struct usb_configuration *c) |
| 2762 | { |
| 2763 | struct fsg_common *fsg_common; |
| 2764 | |
| 2765 | fsg_common = fsg_common_init(NULL, c->cdev); |
| 2766 | |
| 2767 | fsg_common->vendor_name = 0; |
| 2768 | fsg_common->product_name = 0; |
| 2769 | fsg_common->release = 0xffff; |
| 2770 | |
| 2771 | fsg_common->ops = NULL; |
| 2772 | fsg_common->private_data = NULL; |
| 2773 | |
| 2774 | the_fsg_common = fsg_common; |
| 2775 | |
| 2776 | return fsg_bind_config(c->cdev, c, fsg_common); |
| 2777 | } |
| 2778 | |
Stephen Warren | 9e7d588 | 2015-12-07 11:38:50 -0700 | [diff] [blame] | 2779 | int fsg_init(struct ums *ums_devs, int count) |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 2780 | { |
Stephen Warren | 9e7d588 | 2015-12-07 11:38:50 -0700 | [diff] [blame] | 2781 | ums = ums_devs; |
| 2782 | ums_count = count; |
Piotr Wilczek | 91637d7 | 2013-03-05 12:10:16 +0100 | [diff] [blame] | 2783 | |
| 2784 | return 0; |
| 2785 | } |
Mateusz Zalega | 69cb0bb | 2014-04-28 21:13:28 +0200 | [diff] [blame] | 2786 | |
| 2787 | DECLARE_GADGET_BIND_CALLBACK(usb_dnl_ums, fsg_add); |