3 * Copyright (c) 2014 Hans Petter Selasky <hselasky@FreeBSD.org>
6 * This software was developed by SRI International and the University of
7 * Cambridge Computer Laboratory under DARPA/AFRL contract (FA8750-10-C-0237)
8 * ("CTSRD"), as part of the DARPA CRASH research programme.
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
19 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 #include <sys/param.h>
34 #include <bootstrap.h>
40 #define HAVE_STANDARD_DEFS
42 #include USB_GLOBAL_INCLUDE_FILE
44 #include "umass_common.h"
46 static int umass_disk_init(void);
47 static int umass_disk_open(struct open_file *,...);
48 static int umass_disk_close(struct open_file *);
49 static void umass_disk_cleanup(void);
50 static int umass_disk_ioctl(struct open_file *, u_long, void *);
51 static int umass_disk_strategy(void *, int, daddr_t, size_t, char *, size_t *);
52 static int umass_disk_print(int);
54 struct devsw umass_disk = {
57 .dv_init = umass_disk_init,
58 .dv_strategy = umass_disk_strategy,
59 .dv_open = umass_disk_open,
60 .dv_close = umass_disk_close,
61 .dv_ioctl = umass_disk_ioctl,
62 .dv_print = umass_disk_print,
63 .dv_cleanup = umass_disk_cleanup,
72 usb_needs_explore_all();
74 /* wait 8 seconds for a USB mass storage device to appear */
75 for (time = 0; time < (8 * hz); time++) {
80 if (umass_uaa.device != NULL)
87 umass_disk_strategy(void *devdata, int flag, daddr_t dblk, size_t size,
88 char *buf, size_t *rsizep)
90 if (umass_uaa.device == NULL)
96 if (flag == F_WRITE) {
97 if (usb_msc_write_10(umass_uaa.device, 0, dblk, size >> 9, buf) != 0)
99 } else if (flag == F_READ) {
100 if (usb_msc_read_10(umass_uaa.device, 0, dblk, size >> 9, buf) != 0)
112 umass_disk_open_sub(struct disk_devdesc *dev)
117 if (usb_msc_read_capacity(umass_uaa.device, 0, &nblock, &blocksize) != 0)
120 return (disk_open(dev, ((uint64_t)nblock + 1) * (uint64_t)blocksize, blocksize));
124 umass_disk_open(struct open_file *f,...)
127 struct disk_devdesc *dev;
130 dev = va_arg(ap, struct disk_devdesc *);
133 if (umass_uaa.device == NULL)
135 if (dev->d_unit != 0)
137 return (umass_disk_open_sub(dev));
141 umass_disk_ioctl(struct open_file *f, u_long cmd, void *buf)
143 struct disk_devdesc *dev;
148 dev = (struct disk_devdesc *)(f->f_devdata);
152 rc = disk_ioctl(dev, cmd, buf);
157 case DIOCGSECTORSIZE:
159 if (usb_msc_read_capacity(umass_uaa.device, 0,
160 &nblock, &blocksize) != 0)
163 if (cmd == DIOCGMEDIASIZE)
164 *(uint64_t*)buf = nblock;
166 *(uint32_t*)buf = blocksize;
175 umass_disk_close(struct open_file *f)
177 struct disk_devdesc *dev;
179 dev = (struct disk_devdesc *)f->f_devdata;
180 return (disk_close(dev));
184 umass_disk_print(int verbose)
186 struct disk_devdesc dev;
188 printf("%s devices:", umass_disk.dv_name);
189 if (pager_output("\n") != 0)
192 memset(&dev, 0, sizeof(dev));
194 ret = pager_output(" umass0 UMASS device\n");
197 dev.d_dev = &umass_disk;
200 dev.d_partition = -1;
202 if (umass_disk_open_sub(&dev) == 0) {
203 ret = disk_print(&dev, " umass0", verbose);
210 umass_disk_cleanup(void)
217 /* USB specific functions */
219 extern void callout_process(int);
220 extern void usb_idle(void);
221 extern void usb_init(void);
222 extern void usb_uninit(void);
225 DELAY(unsigned int usdelay)
231 pause(const char *what, int timeout)
236 delay((1000000 / hz) * timeout);