Home | History | Annotate | Line # | Download | only in usb
aubtfwl.c revision 1.9
      1 /* $NetBSD: aubtfwl.c,v 1.9 2020/03/14 02:35:33 christos Exp $ */
      2 
      3 /*
      4  * Copyright (c) 2011 Jonathan A. Kollasch
      5  * All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  *
     16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
     17  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     18  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     19  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
     20  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
     21  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
     22  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
     23  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
     24  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
     25  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
     26  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     27  */
     28 
     29 #include <sys/cdefs.h>
     30 __KERNEL_RCSID(0, "$NetBSD: aubtfwl.c,v 1.9 2020/03/14 02:35:33 christos Exp $");
     31 
     32 #include <sys/param.h>
     33 #include <dev/usb/usb.h>
     34 #include <dev/usb/usbdevs.h>
     35 #include <dev/usb/usbdi.h>
     36 #include <dev/usb/usbdivar.h>
     37 #include <dev/usb/usbdi_util.h>
     38 #include <dev/firmload.h>
     39 
     40 #include <dev/usb/aubtfwlreg.h>
     41 
     42 #define AR3K_FIRMWARE_CHUNK_SIZE 4096
     43 
     44 static int aubtfwl_match(device_t, cfdata_t, void *);
     45 static void aubtfwl_attach(device_t, device_t, void *);
     46 static int aubtfwl_detach(device_t, int);
     47 static void aubtfwl_attach_hook(device_t);
     48 
     49 struct aubtfwl_softc {
     50 	struct usbd_device *sc_udev;
     51 	int sc_flags;
     52 #define AUBT_IS_AR3012		1
     53 };
     54 
     55 CFATTACH_DECL_NEW(aubtfwl, sizeof(struct aubtfwl_softc), aubtfwl_match, aubtfwl_attach, aubtfwl_detach, NULL);
     56 
     57 static const struct usb_devno ar3k_devs[] = {
     58 	{ USB_VENDOR_ATHEROS2, USB_PRODUCT_ATHEROS2_AR3011 },
     59 };
     60 
     61 static const struct usb_devno ar3k12_devs[] = {
     62 	{ USB_VENDOR_FOXCONN, USB_PRODUCT_FOXCONN_AR3012 },
     63 };
     64 
     65 static int
     66 aubtfwl_match(device_t parent, cfdata_t match, void *aux)
     67 {
     68 	const struct usb_attach_arg * const uaa = aux;
     69 
     70 	if (usb_lookup(ar3k_devs, uaa->uaa_vendor, uaa->uaa_product))
     71 		return UMATCH_VENDOR_PRODUCT;
     72 
     73 	if (usb_lookup(ar3k12_devs, uaa->uaa_vendor, uaa->uaa_product)) {
     74 		return (UGETW(uaa->uaa_device->ud_ddesc.bcdDevice) > 1)?
     75 			UMATCH_NONE : UMATCH_VENDOR_PRODUCT;
     76 	}
     77 
     78 	return UMATCH_NONE;
     79 }
     80 
     81 static void
     82 aubtfwl_attach(device_t parent, device_t self, void *aux)
     83 {
     84 	const struct usb_attach_arg * const uaa = aux;
     85 	struct aubtfwl_softc * const sc = device_private(self);
     86 	aprint_naive("\n");
     87 	aprint_normal("\n");
     88 	sc->sc_udev = uaa->uaa_device;
     89 	sc->sc_flags = 0;
     90 
     91 	if (usb_lookup(ar3k12_devs, uaa->uaa_vendor, uaa->uaa_product))
     92 		sc->sc_flags |= AUBT_IS_AR3012;
     93 
     94 	config_mountroot(self, aubtfwl_attach_hook);
     95 }
     96 
     97 static int
     98 aubtfwl_detach(device_t self, int flags)
     99 {
    100 	struct aubtfwl_softc * const sc = device_private(self);
    101 
    102 	usbd_add_drv_event(USB_EVENT_DRIVER_DETACH, sc->sc_udev, self);
    103 
    104 	return 0;
    105 }
    106 
    107 /* Returns 0 if firmware was correctly loaded */
    108 static int
    109 aubtfwl_firmware_load(device_t self, const char *name) {
    110 	struct aubtfwl_softc * const sc = device_private(self);
    111 	struct usbd_interface *iface;
    112 	struct usbd_pipe *pipe;
    113 	struct usbd_xfer *xfer;
    114 	void *buf;
    115 	usb_device_request_t req;
    116 	int error = 0;
    117 	firmware_handle_t fwh;
    118 	size_t fws;
    119 	size_t fwo = 0;
    120 	uint32_t n;
    121 
    122 	memset(&req, 0, sizeof(req));
    123 
    124 	error = firmware_open("ubt", name, &fwh);
    125 	if (error != 0) {
    126 		aprint_error_dev(self, "'%s' open fail %d\n", name, error);
    127 		return error;
    128 	}
    129 	fws = firmware_get_size(fwh);
    130 
    131 	error = usbd_set_config_no(sc->sc_udev, 1, 0);
    132 	if (error != 0) {
    133 		aprint_error_dev(self, "failed to set configuration"
    134 		    ", err=%s\n", usbd_errstr(error));
    135 		goto out_firmware;
    136 	}
    137 
    138 	error = usbd_device2interface_handle(sc->sc_udev, 0, &iface);
    139 	if (error) {
    140 		aprint_error_dev(self, "failed to get interface, %s\n",
    141 		   usbd_errstr(error));
    142 		goto out_firmware;
    143 	}
    144 
    145 	error = usbd_open_pipe(iface, UE_DIR_OUT|2, USBD_EXCLUSIVE_USE, &pipe);
    146 	if (error) {
    147 		aprint_error_dev(self, "failed to open pipe, %s\n",
    148 		   usbd_errstr(error));
    149 		goto out_firmware;
    150 	}
    151 
    152 	error = usbd_create_xfer(pipe, AR3K_FIRMWARE_CHUNK_SIZE, 0, 0, &xfer);
    153 	if (error) {
    154 		aprint_verbose_dev(self, "cannot create xfer(%d)\n",
    155 		    error);
    156 		goto out_pipe;
    157 	}
    158 	buf = usbd_get_buffer(xfer);
    159 
    160 	error = firmware_read(fwh, fwo, buf, AR3K_FIRMWARE_HEADER_SIZE);
    161 	if (error != 0) {
    162 		aprint_error_dev(self, "firmware_read failed %d\n", error);
    163 		goto out_xfer;
    164 	}
    165 
    166 	req.bRequest = AR3K_SEND_FIRMWARE;
    167 	req.bmRequestType = UT_WRITE_VENDOR_DEVICE;
    168 	USETW(req.wValue, 0);
    169 	USETW(req.wIndex, 0);
    170 	USETW(req.wLength, AR3K_FIRMWARE_HEADER_SIZE);
    171 
    172 	aprint_verbose_dev(self, "beginning firmware load\n");
    173 
    174 	error = usbd_do_request(sc->sc_udev, &req, buf);
    175 	if (error != 0) {
    176 		aprint_error_dev(self, "%s\n", usbd_errstr(error));
    177 		return error;
    178 	}
    179 	fwo = AR3K_FIRMWARE_HEADER_SIZE;
    180 
    181 	while (fwo < fws) {
    182 		n = uimin(AR3K_FIRMWARE_CHUNK_SIZE, fws - fwo);
    183 		error = firmware_read(fwh, fwo, buf, n);
    184 		if (error != 0) {
    185 			break;
    186 		}
    187 		error = usbd_bulk_transfer(xfer, pipe, 0, USBD_DEFAULT_TIMEOUT,
    188 		    buf, &n);
    189 		if (error != USBD_NORMAL_COMPLETION) {
    190 			aprint_error_dev(self, "xfer failed, %s\n",
    191 			   usbd_errstr(error));
    192 			break;
    193 		}
    194 		fwo += n;
    195 	}
    196 
    197 	if (error == 0)
    198 		aprint_verbose_dev(self, "firmware load complete\n");
    199 
    200 out_xfer:
    201 	usbd_destroy_xfer(xfer);
    202 out_pipe:
    203 	usbd_close_pipe(pipe);
    204 out_firmware:
    205 	firmware_close(fwh);
    206 
    207 	return !!error;
    208 }
    209 
    210 static int
    211 aubtfwl_get_state(struct aubtfwl_softc *sc, uint8_t *state) {
    212 	usb_device_request_t req;
    213 	int error = 0;
    214 
    215 	memset(&req, 0, sizeof(req));
    216 
    217 	req.bRequest = AR3K_GET_STATE;
    218 	req.bmRequestType = UT_READ_VENDOR_DEVICE;
    219 	USETW(req.wValue, 0);
    220 	USETW(req.wIndex, 0);
    221 	USETW(req.wLength, sizeof(*state));
    222 
    223 	error = usbd_do_request(sc->sc_udev, &req, state);
    224 
    225 	return error;
    226 }
    227 
    228 static int
    229 aubtfwl_get_version(struct aubtfwl_softc *sc, struct ar3k_version *ver) {
    230 	usb_device_request_t req;
    231 	int error = 0;
    232 
    233 	memset(&req, 0, sizeof(req));
    234 
    235 	req.bRequest = AR3K_GET_VERSION;
    236 	req.bmRequestType = UT_READ_VENDOR_DEVICE;
    237 	USETW(req.wValue, 0);
    238 	USETW(req.wIndex, 0);
    239 	USETW(req.wLength, sizeof(*ver));
    240 
    241 	error = usbd_do_request(sc->sc_udev, &req, ver);
    242 
    243 #if BYTE_ORDER == BIG_ENDIAN
    244 	if (error == USBD_NORMAL_COMPLETION) {
    245 		ver->rom = bswap32(ver->rom);
    246 		ver->build = bswap32(ver->build);
    247 		ver->ram = bswap32(ver->ram);
    248 	}
    249 #endif
    250 	return error;
    251 }
    252 
    253 static int
    254 aubtfwl_send_command(struct aubtfwl_softc *sc, uByte cmd) {
    255 	usb_device_request_t req;
    256 	int error = 0;
    257 
    258 	memset(&req, 0, sizeof(req));
    259 
    260 	req.bRequest = cmd;
    261 	req.bmRequestType = UT_WRITE_VENDOR_DEVICE;
    262 	USETW(req.wValue, 0);
    263 	USETW(req.wIndex, 0);
    264 	USETW(req.wLength, 0);
    265 
    266 	error = usbd_do_request(sc->sc_udev, &req, NULL);
    267 
    268 	return error;
    269 }
    270 
    271 static void
    272 aubtfwl_attach_hook(device_t self)
    273 {
    274 	struct aubtfwl_softc * const sc = device_private(self);
    275 	char firmware_name[MAXPATHLEN+1];
    276 	struct ar3k_version ver;
    277 	uint8_t state;
    278 	int clock = 0;
    279 	int error = 0;
    280 
    281 	if (sc->sc_flags & AUBT_IS_AR3012) {
    282 		error = aubtfwl_get_version(sc, &ver);
    283 		if (!error)
    284 			error = aubtfwl_get_state(sc, &state);
    285 
    286 		if (error) {
    287 			aprint_error_dev(self,
    288 				"couldn't get version or state\n");
    289 			return;
    290 		}
    291 
    292 		aprint_verbose_dev(self, "state is 0x%02x\n", state);
    293 
    294 		if (!(state & AR3K_STATE_IS_PATCHED)) {
    295 			snprintf(firmware_name, sizeof(firmware_name),
    296 				"ar3k/AthrBT_0x%08x.dfu", ver.rom);
    297 			error = aubtfwl_firmware_load(self, firmware_name);
    298 
    299 			if (error)
    300 				return;
    301 		}
    302 
    303 		switch (ver.clock) {
    304 		case AR3K_CLOCK_19M:
    305 			clock = 19;
    306 			break;
    307 		case AR3K_CLOCK_26M:
    308 			clock = 26;
    309 			break;
    310 		case AR3K_CLOCK_40M:
    311 			clock = 40;
    312 			break;
    313 		}
    314 
    315 		snprintf(firmware_name, sizeof(firmware_name),
    316 			"ar3k/ramps_0x%08x_%d.dfu", ver.rom, clock);
    317 		aubtfwl_firmware_load(self, firmware_name);
    318 
    319 		if ((state & AR3K_STATE_MODE_MASK) != AR3K_STATE_MODE_NORMAL) {
    320 			error = aubtfwl_send_command(sc, AR3K_SET_NORMAL_MODE);
    321 			if (error) {
    322 				aprint_error_dev(self,
    323 					"couldn't set normal mode: %s",
    324 					usbd_errstr(error));
    325 				return;
    326 			}
    327 		}
    328 
    329 		/* Apparently some devices will fail this, so ignore result */
    330 		(void) aubtfwl_send_command(sc, AR3K_SWITCH_VID_PID);
    331 	} else {
    332 		aubtfwl_firmware_load(self, "ath3k-1.fw");
    333 	}
    334 
    335 	return;
    336 }
    337