Date: Fri, 30 May 2014 13:10:50 +0000 (UTC) From: Hans Petter Selasky <hselasky@FreeBSD.org> To: src-committers@freebsd.org, svn-src-all@freebsd.org, svn-src-head@freebsd.org Subject: svn commit: r266881 - head/sys/boot/usb/storage Message-ID: <201405301310.s4UDAoJv057009@svn.freebsd.org>
next in thread | raw e-mail | index | archive | help
Author: hselasky Date: Fri May 30 13:10:49 2014 New Revision: 266881 URL: http://svnweb.freebsd.org/changeset/base/266881 Log: Add support for USB mass storage to libusbboot. Sponsored by: DARPA, AFRL Added: head/sys/boot/usb/storage/ head/sys/boot/usb/storage/umass_common.c (contents, props changed) head/sys/boot/usb/storage/umass_common.h (contents, props changed) head/sys/boot/usb/storage/umass_loader.c (contents, props changed) Added: head/sys/boot/usb/storage/umass_common.c ============================================================================== --- /dev/null 00:00:00 1970 (empty, because file is newly added) +++ head/sys/boot/usb/storage/umass_common.c Fri May 30 13:10:49 2014 (r266881) @@ -0,0 +1,87 @@ +/* $FreeBSD$ */ +/*- + * Copyright (c) 2014 Hans Petter Selasky <hselasky@FreeBSD.org> + * All rights reserved. + * + * This software was developed by SRI International and the University of + * Cambridge Computer Laboratory under DARPA/AFRL contract (FA8750-10-C-0237) + * ("CTSRD"), as part of the DARPA CRASH research programme. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS + * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT + * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY + * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF + * SUCH DAMAGE. + */ + +#include USB_GLOBAL_INCLUDE_FILE + +#include "umass_common.h" + +struct usb_attach_arg umass_uaa; + +static device_probe_t umass_probe; +static device_attach_t umass_attach; +static device_detach_t umass_detach; + +static devclass_t umass_devclass; + +static device_method_t umass_methods[] = { + /* Device interface */ + DEVMETHOD(device_probe, umass_probe), + DEVMETHOD(device_attach, umass_attach), + DEVMETHOD(device_detach, umass_detach), + + DEVMETHOD_END +}; + +static driver_t umass_driver = { + .name = "umass", + .methods = umass_methods, +}; + +DRIVER_MODULE(umass, uhub, umass_driver, umass_devclass, NULL, 0); + +static int +umass_probe(device_t dev) +{ + struct usb_attach_arg *uaa = device_get_ivars(dev); + + if (uaa->usb_mode != USB_MODE_HOST || + uaa->info.bInterfaceClass != UICLASS_MASS || + uaa->info.bInterfaceSubClass != UISUBCLASS_SCSI || + uaa->info.bInterfaceProtocol != UIPROTO_MASS_BBB || + device_get_unit(dev) != 0) + return (ENXIO); + return (0); +} + +static int +umass_attach(device_t dev) +{ + struct usb_attach_arg *uaa = device_get_ivars(dev); + umass_uaa = *uaa; + return (0); /* success */ +} + +static int +umass_detach(device_t dev) +{ + memset(&umass_uaa, 0, sizeof(umass_uaa)); + return (0); +} Added: head/sys/boot/usb/storage/umass_common.h ============================================================================== --- /dev/null 00:00:00 1970 (empty, because file is newly added) +++ head/sys/boot/usb/storage/umass_common.h Fri May 30 13:10:49 2014 (r266881) @@ -0,0 +1,37 @@ +/* $FreeBSD$ */ +/*- + * Copyright (c) 2014 Hans Petter Selasky <hselasky@FreeBSD.org> + * All rights reserved. + * + * This software was developed by SRI International and the University of + * Cambridge Computer Laboratory under DARPA/AFRL contract (FA8750-10-C-0237) + * ("CTSRD"), as part of the DARPA CRASH research programme. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS + * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT + * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY + * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF + * SUCH DAMAGE. + */ + +#ifndef _UMASS_COMMON_H_ +#define _UMASS_COMMON_H_ + +extern struct usb_attach_arg umass_uaa; + +#endif /* _UMASS_COMMON_H_ */ Added: head/sys/boot/usb/storage/umass_loader.c ============================================================================== --- /dev/null 00:00:00 1970 (empty, because file is newly added) +++ head/sys/boot/usb/storage/umass_loader.c Fri May 30 13:10:49 2014 (r266881) @@ -0,0 +1,197 @@ +/* $FreeBSD$ */ +/*- + * Copyright (c) 2014 Hans Petter Selasky <hselasky@FreeBSD.org> + * All rights reserved. + * + * This software was developed by SRI International and the University of + * Cambridge Computer Laboratory under DARPA/AFRL contract (FA8750-10-C-0237) + * ("CTSRD"), as part of the DARPA CRASH research programme. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS + * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT + * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY + * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF + * SUCH DAMAGE. + */ + +#include <sys/param.h> + +#include <bootstrap.h> +#include <stdarg.h> + +#include <stand.h> +#include <disk.h> + +#define HAVE_STANDARD_DEFS + +#include USB_GLOBAL_INCLUDE_FILE + +#include "umass_common.h" + +static int umass_disk_init(void); +static int umass_disk_open(struct open_file *,...); +static int umass_disk_close(struct open_file *); +static void umass_disk_cleanup(void); +static int umass_disk_strategy(void *, int, daddr_t, size_t, char *, size_t *); +static void umass_disk_print(int); + +struct devsw umass_disk = { + .dv_name = "umass", + .dv_type = DEVT_DISK, + .dv_init = umass_disk_init, + .dv_strategy = umass_disk_strategy, + .dv_open = umass_disk_open, + .dv_close = umass_disk_close, + .dv_ioctl = noioctl, + .dv_print = umass_disk_print, + .dv_cleanup = umass_disk_cleanup, +}; + +static int +umass_disk_init(void) +{ + uint32_t time; + + usb_init(); + usb_needs_explore_all(); + + /* wait 8 seconds for a USB mass storage device to appear */ + for (time = 0; time < (8 * hz); time++) { + usb_idle(); + delay(1000000 / hz); + time++; + callout_process(1); + if (umass_uaa.device != NULL) + return (0); + } + return (0); +} + +static int +umass_disk_strategy(void *devdata, int flag, daddr_t dblk, size_t size, + char *buf, size_t *rsizep) +{ + if (umass_uaa.device == NULL) + return (ENXIO); + if (rsizep != NULL) + *rsizep = 0; + + if (flag == F_WRITE) { + if (usb_msc_write_10(umass_uaa.device, 0, dblk, size >> 9, buf) != 0) + return (EINVAL); + } else if (flag == F_READ) { + if (usb_msc_read_10(umass_uaa.device, 0, dblk, size >> 9, buf) != 0) + return (EINVAL); + } else { + return (EROFS); + } + + if (rsizep != NULL) + *rsizep = size; + return (0); +} + +static int +umass_disk_open_sub(struct disk_devdesc *dev) +{ + uint32_t nblock; + uint32_t blocksize; + + if (usb_msc_read_capacity(umass_uaa.device, 0, &nblock, &blocksize) != 0) + return (EINVAL); + + return (disk_open(dev, ((uint64_t)nblock + 1) * (uint64_t)blocksize, blocksize, 0)); +} + +static int +umass_disk_open(struct open_file *f,...) +{ + va_list ap; + struct disk_devdesc *dev; + + va_start(ap, f); + dev = va_arg(ap, struct disk_devdesc *); + va_end(ap); + + if (umass_uaa.device == NULL) + return (ENXIO); + if (dev->d_unit != 0) + return (EIO); + return (umass_disk_open_sub(dev)); +} + +static int +umass_disk_close(struct open_file *f) +{ + struct disk_devdesc *dev; + + dev = (struct disk_devdesc *)f->f_devdata; + return (disk_close(dev)); +} + +static void +umass_disk_print(int verbose) +{ + struct disk_devdesc dev; + + memset(&dev, 0, sizeof(dev)); + + pager_output(" umass0 UMASS device\n"); + dev.d_dev = &umass_disk; + dev.d_unit = 0; + dev.d_slice = -1; + dev.d_partition = -1; + + if (umass_disk_open_sub(&dev) == 0) { + disk_print(&dev, " umass0", verbose); + disk_close(&dev); + } +} + +static void +umass_disk_cleanup(void) +{ + disk_cleanup(&umass_disk); + + usb_uninit(); +} + + +/* USB specific functions */ + +extern void callout_process(int); +extern void usb_idle(void); +extern void usb_init(void); +extern void usb_uninit(void); + +void +DELAY(unsigned int usdelay) +{ + delay(usdelay); +} + +int +pause(const char *what, int timeout) +{ + if (timeout == 0) + timeout = 1; + + delay((1000000 / hz) * timeout); + + return (0); +}
Want to link to this message? Use this URL: <https://mail-archive.FreeBSD.org/cgi/mid.cgi?201405301310.s4UDAoJv057009>