sys/dev/usb/wlan/if_rsu.c

   1 /*      $OpenBSD: if_rsu.c,v 1.17 2013/04/15 09:23:01 mglocker Exp $    */
   2
   3 /*-
   4  * Copyright (c) 2010 Damien Bergamini <damien.bergamini@free.fr>
   5  *
   6  * Permission to use, copy, modify, and distribute this software for any
   7  * purpose with or without fee is hereby granted, provided that the above
   8  * copyright notice and this permission notice appear in all copies.
   9  *
  10  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  11  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  12  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  13  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  14  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  15  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  16  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  17  */
  18 #include <sys/cdefs.h>
  19 __FBSDID("$FreeBSD$");
  20
  21 /*
  22  * Driver for Realtek RTL8188SU/RTL8191SU/RTL8192SU.
  23  *
  24  * TODO:
  25  *   o tx a-mpdu
  26  *   o hostap / ibss / mesh
  27  *   o power-save operation
  28  */
  29
  30 #include "opt_wlan.h"
  31
  32 #include <sys/param.h>
  33 #include <sys/endian.h>
  34 #include <sys/sockio.h>
  35 #include <sys/malloc.h>
  36 #include <sys/mbuf.h>
  37 #include <sys/kernel.h>
  38 #include <sys/socket.h>
  39 #include <sys/systm.h>
  40 #include <sys/conf.h>
  41 #include <sys/bus.h>
  42 #include <sys/firmware.h>
  43 #include <sys/module.h>
  44
  45 #include <net/bpf.h>
  46 #include <net/if.h>
  47 #include <net/if_var.h>
  48 #include <net/if_arp.h>
  49 #include <net/if_dl.h>
  50 #include <net/if_media.h>
  51 #include <net/if_types.h>
  52
  53 #include <netinet/in.h>
  54 #include <netinet/in_systm.h>
  55 #include <netinet/in_var.h>
  56 #include <netinet/if_ether.h>
  57 #include <netinet/ip.h>
  58
  59 #include <net80211/ieee80211_var.h>
  60 #include <net80211/ieee80211_regdomain.h>
  61 #include <net80211/ieee80211_radiotap.h>
  62
  63 #include <dev/usb/usb.h>
  64 #include <dev/usb/usbdi.h>
  65 #include "usbdevs.h"
  66
  67 #include <dev/rtwn/if_rtwn_ridx.h>      /* XXX */
  68 #include <dev/usb/wlan/if_rsureg.h>
  69
  70 #define RSU_RATE_IS_CCK RTWN_RATE_IS_CCK
  71
  72 #ifdef USB_DEBUG
  73 static int rsu_debug = 0;
  74 SYSCTL_NODE(_hw_usb, OID_AUTO, rsu, CTLFLAG_RW | CTLFLAG_MPSAFE, 0,
  75     "USB rsu");
  76 SYSCTL_INT(_hw_usb_rsu, OID_AUTO, debug, CTLFLAG_RWTUN, &rsu_debug, 0,
  77     "Debug level");
  78 #define RSU_DPRINTF(_sc, _flg, ...)                                     \
  79         do                                                              \
  80                 if (((_flg) == (RSU_DEBUG_ANY)) || (rsu_debug & (_flg))) \
  81                         device_printf((_sc)->sc_dev, __VA_ARGS__);      \
  82         while (0)
  83 #else
  84 #define RSU_DPRINTF(_sc, _flg, ...)
  85 #endif
  86
  87 static int rsu_enable_11n = 1;
  88 TUNABLE_INT("hw.usb.rsu.enable_11n", &rsu_enable_11n);
  89
  90 #define RSU_DEBUG_ANY           0xffffffff
  91 #define RSU_DEBUG_TX            0x00000001
  92 #define RSU_DEBUG_RX            0x00000002
  93 #define RSU_DEBUG_RESET         0x00000004
  94 #define RSU_DEBUG_CALIB         0x00000008
  95 #define RSU_DEBUG_STATE         0x00000010
  96 #define RSU_DEBUG_SCAN          0x00000020
  97 #define RSU_DEBUG_FWCMD         0x00000040
  98 #define RSU_DEBUG_TXDONE        0x00000080
  99 #define RSU_DEBUG_FW            0x00000100
 100 #define RSU_DEBUG_FWDBG         0x00000200
 101 #define RSU_DEBUG_AMPDU         0x00000400
 102 #define RSU_DEBUG_KEY           0x00000800
 103 #define RSU_DEBUG_USB           0x00001000
 104
 105 static const STRUCT_USB_HOST_ID rsu_devs[] = {
 106 #define RSU_HT_NOT_SUPPORTED 0
 107 #define RSU_HT_SUPPORTED 1
 108 #define RSU_DEV_HT(v,p)  { USB_VPI(USB_VENDOR_##v, USB_PRODUCT_##v##_##p, \
 109                                    RSU_HT_SUPPORTED) }
 110 #define RSU_DEV(v,p)     { USB_VPI(USB_VENDOR_##v, USB_PRODUCT_##v##_##p, \
 111                                    RSU_HT_NOT_SUPPORTED) }
 112         RSU_DEV(ASUS,                   RTL8192SU),
 113         RSU_DEV(AZUREWAVE,              RTL8192SU_4),
 114         RSU_DEV(SITECOMEU,              WLA1000),
 115         RSU_DEV_HT(ACCTON,              RTL8192SU),
 116         RSU_DEV_HT(ASUS,                USBN10),
 117         RSU_DEV_HT(AZUREWAVE,           RTL8192SU_1),
 118         RSU_DEV_HT(AZUREWAVE,           RTL8192SU_2),
 119         RSU_DEV_HT(AZUREWAVE,           RTL8192SU_3),
 120         RSU_DEV_HT(AZUREWAVE,           RTL8192SU_5),
 121         RSU_DEV_HT(BELKIN,              RTL8192SU_1),
 122         RSU_DEV_HT(BELKIN,              RTL8192SU_2),
 123         RSU_DEV_HT(BELKIN,              RTL8192SU_3),
 124         RSU_DEV_HT(CONCEPTRONIC2,       RTL8192SU_1),
 125         RSU_DEV_HT(CONCEPTRONIC2,       RTL8192SU_2),
 126         RSU_DEV_HT(CONCEPTRONIC2,       RTL8192SU_3),
 127         RSU_DEV_HT(COREGA,              RTL8192SU),
 128         RSU_DEV_HT(DLINK2,              DWA131A1),
 129         RSU_DEV_HT(DLINK2,              RTL8192SU_1),
 130         RSU_DEV_HT(DLINK2,              RTL8192SU_2),
 131         RSU_DEV_HT(EDIMAX,              RTL8192SU_1),
 132         RSU_DEV_HT(EDIMAX,              RTL8192SU_2),
 133         RSU_DEV_HT(EDIMAX,              EW7622UMN),
 134         RSU_DEV_HT(GUILLEMOT,           HWGUN54),
 135         RSU_DEV_HT(GUILLEMOT,           HWNUM300),
 136         RSU_DEV_HT(HAWKING,             RTL8192SU_1),
 137         RSU_DEV_HT(HAWKING,             RTL8192SU_2),
 138         RSU_DEV_HT(PLANEX2,             GWUSNANO),
 139         RSU_DEV_HT(REALTEK,             RTL8171),
 140         RSU_DEV_HT(REALTEK,             RTL8172),
 141         RSU_DEV_HT(REALTEK,             RTL8173),
 142         RSU_DEV_HT(REALTEK,             RTL8174),
 143         RSU_DEV_HT(REALTEK,             RTL8192SU),
 144         RSU_DEV_HT(REALTEK,             RTL8712),
 145         RSU_DEV_HT(REALTEK,             RTL8713),
 146         RSU_DEV_HT(SENAO,               RTL8192SU_1),
 147         RSU_DEV_HT(SENAO,               RTL8192SU_2),
 148         RSU_DEV_HT(SITECOMEU,           WL349V1),
 149         RSU_DEV_HT(SITECOMEU,           WL353),
 150         RSU_DEV_HT(SWEEX2,              LW154),
 151         RSU_DEV_HT(TRENDNET,            TEW646UBH),
 152 #undef RSU_DEV_HT
 153 #undef RSU_DEV
 154 };
 155
 156 static device_probe_t   rsu_match;
 157 static device_attach_t  rsu_attach;
 158 static device_detach_t  rsu_detach;
 159 static usb_callback_t   rsu_bulk_tx_callback_be_bk;
 160 static usb_callback_t   rsu_bulk_tx_callback_vi_vo;
 161 static usb_callback_t   rsu_bulk_tx_callback_h2c;
 162 static usb_callback_t   rsu_bulk_rx_callback;
 163 static usb_error_t      rsu_do_request(struct rsu_softc *,
 164                             struct usb_device_request *, void *);
 165 static struct ieee80211vap *
 166                 rsu_vap_create(struct ieee80211com *, const char name[],
 167                     int, enum ieee80211_opmode, int, const uint8_t bssid[],
 168                     const uint8_t mac[]);
 169 static void     rsu_vap_delete(struct ieee80211vap *);
 170 static void     rsu_scan_start(struct ieee80211com *);
 171 static void     rsu_scan_end(struct ieee80211com *);
 172 static void     rsu_getradiocaps(struct ieee80211com *, int, int *,
 173                     struct ieee80211_channel[]);
 174 static void     rsu_set_channel(struct ieee80211com *);
 175 static void     rsu_scan_curchan(struct ieee80211_scan_state *, unsigned long);
 176 static void     rsu_scan_mindwell(struct ieee80211_scan_state *);
 177 static void     rsu_update_promisc(struct ieee80211com *);
 178 static uint8_t  rsu_get_multi_pos(const uint8_t[]);
 179 static void     rsu_set_multi(struct rsu_softc *);
 180 static void     rsu_update_mcast(struct ieee80211com *);
 181 static int      rsu_alloc_rx_list(struct rsu_softc *);
 182 static void     rsu_free_rx_list(struct rsu_softc *);
 183 static int      rsu_alloc_tx_list(struct rsu_softc *);
 184 static void     rsu_free_tx_list(struct rsu_softc *);
 185 static void     rsu_free_list(struct rsu_softc *, struct rsu_data [], int);
 186 static struct rsu_data *_rsu_getbuf(struct rsu_softc *);
 187 static struct rsu_data *rsu_getbuf(struct rsu_softc *);
 188 static void     rsu_freebuf(struct rsu_softc *, struct rsu_data *);
 189 static int      rsu_write_region_1(struct rsu_softc *, uint16_t, uint8_t *,
 190                     int);
 191 static void     rsu_write_1(struct rsu_softc *, uint16_t, uint8_t);
 192 static void     rsu_write_2(struct rsu_softc *, uint16_t, uint16_t);
 193 static void     rsu_write_4(struct rsu_softc *, uint16_t, uint32_t);
 194 static int      rsu_read_region_1(struct rsu_softc *, uint16_t, uint8_t *,
 195                     int);
 196 static uint8_t  rsu_read_1(struct rsu_softc *, uint16_t);
 197 static uint16_t rsu_read_2(struct rsu_softc *, uint16_t);
 198 static uint32_t rsu_read_4(struct rsu_softc *, uint16_t);
 199 static int      rsu_fw_iocmd(struct rsu_softc *, uint32_t);
 200 static uint8_t  rsu_efuse_read_1(struct rsu_softc *, uint16_t);
 201 static int      rsu_read_rom(struct rsu_softc *);
 202 static int      rsu_fw_cmd(struct rsu_softc *, uint8_t, void *, int);
 203 static void     rsu_calib_task(void *, int);
 204 static void     rsu_tx_task(void *, int);
 205 static void     rsu_set_led(struct rsu_softc *, int);
 206 static int      rsu_monitor_newstate(struct ieee80211vap *,
 207                     enum ieee80211_state, int);
 208 static int      rsu_newstate(struct ieee80211vap *, enum ieee80211_state, int);
 209 static int      rsu_key_alloc(struct ieee80211vap *, struct ieee80211_key *,
 210                     ieee80211_keyix *, ieee80211_keyix *);
 211 static int      rsu_process_key(struct ieee80211vap *,
 212                     const struct ieee80211_key *, int);
 213 static int      rsu_key_set(struct ieee80211vap *,
 214                     const struct ieee80211_key *);
 215 static int      rsu_key_delete(struct ieee80211vap *,
 216                     const struct ieee80211_key *);
 217 static int      rsu_cam_read(struct rsu_softc *, uint8_t, uint32_t *);
 218 static void     rsu_cam_write(struct rsu_softc *, uint8_t, uint32_t);
 219 static int      rsu_key_check(struct rsu_softc *, ieee80211_keyix, int);
 220 static uint8_t  rsu_crypto_mode(struct rsu_softc *, u_int, int);
 221 static int      rsu_set_key_group(struct rsu_softc *,
 222                     const struct ieee80211_key *);
 223 static int      rsu_set_key_pair(struct rsu_softc *,
 224                     const struct ieee80211_key *);
 225 static int      rsu_reinit_static_keys(struct rsu_softc *);
 226 static int      rsu_delete_key(struct rsu_softc *sc, ieee80211_keyix);
 227 static void     rsu_delete_key_pair_cb(void *, int);
 228 static int      rsu_site_survey(struct rsu_softc *,
 229                     struct ieee80211_scan_ssid *);
 230 static int      rsu_join_bss(struct rsu_softc *, struct ieee80211_node *);
 231 static int      rsu_disconnect(struct rsu_softc *);
 232 static int      rsu_hwrssi_to_rssi(struct rsu_softc *, int hw_rssi);
 233 static void     rsu_event_survey(struct rsu_softc *, uint8_t *, int);
 234 static void     rsu_event_join_bss(struct rsu_softc *, uint8_t *, int);
 235 static void     rsu_rx_event(struct rsu_softc *, uint8_t, uint8_t *, int);
 236 static void     rsu_rx_multi_event(struct rsu_softc *, uint8_t *, int);
 237 static int8_t   rsu_get_rssi(struct rsu_softc *, int, void *);
 238 static struct mbuf * rsu_rx_copy_to_mbuf(struct rsu_softc *,
 239                     struct r92s_rx_stat *, int);
 240 static uint32_t rsu_get_tsf_low(struct rsu_softc *);
 241 static uint32_t rsu_get_tsf_high(struct rsu_softc *);
 242 static struct ieee80211_node * rsu_rx_frame(struct rsu_softc *, struct mbuf *);
 243 static struct mbuf * rsu_rx_multi_frame(struct rsu_softc *, uint8_t *, int);
 244 static struct mbuf *
 245                 rsu_rxeof(struct usb_xfer *, struct rsu_data *);
 246 static void     rsu_txeof(struct usb_xfer *, struct rsu_data *);
 247 static int      rsu_raw_xmit(struct ieee80211_node *, struct mbuf *,
 248                     const struct ieee80211_bpf_params *);
 249 static void     rsu_rxfilter_init(struct rsu_softc *);
 250 static void     rsu_rxfilter_set(struct rsu_softc *, uint32_t, uint32_t);
 251 static void     rsu_rxfilter_refresh(struct rsu_softc *);
 252 static int      rsu_init(struct rsu_softc *);
 253 static int      rsu_tx_start(struct rsu_softc *, struct ieee80211_node *,
 254                     struct mbuf *, struct rsu_data *);
 255 static int      rsu_transmit(struct ieee80211com *, struct mbuf *);
 256 static void     rsu_start(struct rsu_softc *);
 257 static void     _rsu_start(struct rsu_softc *);
 258 static int      rsu_ioctl_net(struct ieee80211com *, u_long, void *);
 259 static void     rsu_parent(struct ieee80211com *);
 260 static void     rsu_stop(struct rsu_softc *);
 261 static void     rsu_ms_delay(struct rsu_softc *, int);
 262
 263 static device_method_t rsu_methods[] = {
 264         DEVMETHOD(device_probe,         rsu_match),
 265         DEVMETHOD(device_attach,        rsu_attach),
 266         DEVMETHOD(device_detach,        rsu_detach),
 267
 268         DEVMETHOD_END
 269 };
 270
 271 static driver_t rsu_driver = {
 272         .name = "rsu",
 273         .methods = rsu_methods,
 274         .size = sizeof(struct rsu_softc)
 275 };
 276
 277 static devclass_t rsu_devclass;
 278
 279 DRIVER_MODULE(rsu, uhub, rsu_driver, rsu_devclass, NULL, 0);
 280 MODULE_DEPEND(rsu, wlan, 1, 1, 1);
 281 MODULE_DEPEND(rsu, usb, 1, 1, 1);
 282 MODULE_DEPEND(rsu, firmware, 1, 1, 1);
 283 MODULE_VERSION(rsu, 1);
 284 USB_PNP_HOST_INFO(rsu_devs);
 285
 286 static uint8_t rsu_wme_ac_xfer_map[4] = {
 287         [WME_AC_BE] = RSU_BULK_TX_BE_BK,
 288         [WME_AC_BK] = RSU_BULK_TX_BE_BK,
 289         [WME_AC_VI] = RSU_BULK_TX_VI_VO,
 290         [WME_AC_VO] = RSU_BULK_TX_VI_VO,
 291 };
 292
 293 /* XXX hard-coded */
 294 #define RSU_H2C_ENDPOINT        3
 295
 296 static const struct usb_config rsu_config[RSU_N_TRANSFER] = {
 297         [RSU_BULK_RX] = {
 298                 .type = UE_BULK,
 299                 .endpoint = UE_ADDR_ANY,
 300                 .direction = UE_DIR_IN,
 301                 .bufsize = RSU_RXBUFSZ,
 302                 .flags = {
 303                         .pipe_bof = 1,
 304                         .short_xfer_ok = 1
 305                 },
 306                 .callback = rsu_bulk_rx_callback
 307         },
 308         [RSU_BULK_TX_BE_BK] = {
 309                 .type = UE_BULK,
 310                 .endpoint = 0x06,
 311                 .direction = UE_DIR_OUT,
 312                 .bufsize = RSU_TXBUFSZ,
 313                 .flags = {
 314                         .ext_buffer = 1,
 315                         .pipe_bof = 1,
 316                         .force_short_xfer = 1
 317                 },
 318                 .callback = rsu_bulk_tx_callback_be_bk,
 319                 .timeout = RSU_TX_TIMEOUT
 320         },
 321         [RSU_BULK_TX_VI_VO] = {
 322                 .type = UE_BULK,
 323                 .endpoint = 0x04,
 324                 .direction = UE_DIR_OUT,
 325                 .bufsize = RSU_TXBUFSZ,
 326                 .flags = {
 327                         .ext_buffer = 1,
 328                         .pipe_bof = 1,
 329                         .force_short_xfer = 1
 330                 },
 331                 .callback = rsu_bulk_tx_callback_vi_vo,
 332                 .timeout = RSU_TX_TIMEOUT
 333         },
 334         [RSU_BULK_TX_H2C] = {
 335                 .type = UE_BULK,
 336                 .endpoint = 0x0d,
 337                 .direction = UE_DIR_OUT,
 338                 .bufsize = RSU_TXBUFSZ,
 339                 .flags = {
 340                         .ext_buffer = 1,
 341                         .pipe_bof = 1,
 342                         .short_xfer_ok = 1
 343                 },
 344                 .callback = rsu_bulk_tx_callback_h2c,
 345                 .timeout = RSU_TX_TIMEOUT
 346         },
 347 };
 348
 349 static int
 350 rsu_match(device_t self)
 351 {
 352         struct usb_attach_arg *uaa = device_get_ivars(self);
 353
 354         if (uaa->usb_mode != USB_MODE_HOST ||
 355             uaa->info.bIfaceIndex != 0 ||
 356             uaa->info.bConfigIndex != 0)
 357                 return (ENXIO);
 358
 359         return (usbd_lookup_id_by_uaa(rsu_devs, sizeof(rsu_devs), uaa));
 360 }
 361
 362 static int
 363 rsu_send_mgmt(struct ieee80211_node *ni, int type, int arg)
 364 {
 365
 366         return (ENOTSUP);
 367 }
 368
 369 static void
 370 rsu_update_chw(struct ieee80211com *ic)
 371 {
 372
 373 }
 374
 375 /*
 376  * notification from net80211 that it'd like to do A-MPDU on the given TID.
 377  *
 378  * Note: this actually hangs traffic at the present moment, so don't use it.
 379  * The firmware debug does indiciate it's sending and establishing a TX AMPDU
 380  * session, but then no traffic flows.
 381  */
 382 static int
 383 rsu_ampdu_enable(struct ieee80211_node *ni, struct ieee80211_tx_ampdu *tap)
 384 {
 385 #if 0
 386         struct rsu_softc *sc = ni->ni_ic->ic_softc;
 387         struct r92s_add_ba_req req;
 388
 389         /* Don't enable if it's requested or running */
 390         if (IEEE80211_AMPDU_REQUESTED(tap))
 391                 return (0);
 392         if (IEEE80211_AMPDU_RUNNING(tap))
 393                 return (0);
 394
 395         /* We've decided to send addba; so send it */
 396         req.tid = htole32(tap->txa_tid);
 397
 398         /* Attempt net80211 state */
 399         if (ieee80211_ampdu_tx_request_ext(ni, tap->txa_tid) != 1)
 400                 return (0);
 401
 402         /* Send the firmware command */
 403         RSU_DPRINTF(sc, RSU_DEBUG_AMPDU, "%s: establishing AMPDU TX for TID %d\n",
 404             __func__,
 405             tap->txa_tid);
 406
 407         RSU_LOCK(sc);
 408         if (rsu_fw_cmd(sc, R92S_CMD_ADDBA_REQ, &req, sizeof(req)) != 1) {
 409                 RSU_UNLOCK(sc);
 410                 /* Mark failure */
 411                 (void) ieee80211_ampdu_tx_request_active_ext(ni, tap->txa_tid, 0);
 412                 return (0);
 413         }
 414         RSU_UNLOCK(sc);
 415
 416         /* Mark success; we don't get any further notifications */
 417         (void) ieee80211_ampdu_tx_request_active_ext(ni, tap->txa_tid, 1);
 418 #endif
 419         /* Return 0, we're driving this ourselves */
 420         return (0);
 421 }
 422
 423 static int
 424 rsu_wme_update(struct ieee80211com *ic)
 425 {
 426
 427         /* Firmware handles this; not our problem */
 428         return (0);
 429 }
 430
 431 static int
 432 rsu_attach(device_t self)
 433 {
 434         struct usb_attach_arg *uaa = device_get_ivars(self);
 435         struct rsu_softc *sc = device_get_softc(self);
 436         struct ieee80211com *ic = &sc->sc_ic;
 437         int error;
 438         uint8_t iface_index;
 439         struct usb_interface *iface;
 440         const char *rft;
 441
 442         device_set_usb_desc(self);
 443         sc->sc_udev = uaa->device;
 444         sc->sc_dev = self;
 445         sc->sc_rx_checksum_enable = 1;
 446         if (rsu_enable_11n)
 447                 sc->sc_ht = !! (USB_GET_DRIVER_INFO(uaa) & RSU_HT_SUPPORTED);
 448
 449         /* Get number of endpoints */
 450         iface = usbd_get_iface(sc->sc_udev, 0);
 451         sc->sc_nendpoints = iface->idesc->bNumEndpoints;
 452
 453         /* Endpoints are hard-coded for now, so enforce 4-endpoint only */
 454         if (sc->sc_nendpoints != 4) {
 455                 device_printf(sc->sc_dev,
 456                     "the driver currently only supports 4-endpoint devices\n");
 457                 return (ENXIO);
 458         }
 459
 460         mtx_init(&sc->sc_mtx, device_get_nameunit(self), MTX_NETWORK_LOCK,
 461             MTX_DEF);
 462         RSU_DELKEY_BMAP_LOCK_INIT(sc);
 463         TIMEOUT_TASK_INIT(taskqueue_thread, &sc->calib_task, 0,
 464             rsu_calib_task, sc);
 465         TASK_INIT(&sc->del_key_task, 0, rsu_delete_key_pair_cb, sc);
 466         TASK_INIT(&sc->tx_task, 0, rsu_tx_task, sc);
 467         mbufq_init(&sc->sc_snd, ifqmaxlen);
 468
 469         /* Allocate Tx/Rx buffers. */
 470         error = rsu_alloc_rx_list(sc);
 471         if (error != 0) {
 472                 device_printf(sc->sc_dev, "could not allocate Rx buffers\n");
 473                 goto fail_usb;
 474         }
 475
 476         error = rsu_alloc_tx_list(sc);
 477         if (error != 0) {
 478                 device_printf(sc->sc_dev, "could not allocate Tx buffers\n");
 479                 rsu_free_rx_list(sc);
 480                 goto fail_usb;
 481         }
 482
 483         iface_index = 0;
 484         error = usbd_transfer_setup(uaa->device, &iface_index, sc->sc_xfer,
 485             rsu_config, RSU_N_TRANSFER, sc, &sc->sc_mtx);
 486         if (error) {
 487                 device_printf(sc->sc_dev,
 488                     "could not allocate USB transfers, err=%s\n",
 489                     usbd_errstr(error));
 490                 goto fail_usb;
 491         }
 492         RSU_LOCK(sc);
 493         /* Read chip revision. */
 494         sc->cut = MS(rsu_read_4(sc, R92S_PMC_FSM), R92S_PMC_FSM_CUT);
 495         if (sc->cut != 3)
 496                 sc->cut = (sc->cut >> 1) + 1;
 497         error = rsu_read_rom(sc);
 498         RSU_UNLOCK(sc);
 499         if (error != 0) {
 500                 device_printf(self, "could not read ROM\n");
 501                 goto fail_rom;
 502         }
 503
 504         /* Figure out TX/RX streams */
 505         switch (sc->rom[84]) {
 506         case 0x0:
 507                 sc->sc_rftype = RTL8712_RFCONFIG_1T1R;
 508                 sc->sc_nrxstream = 1;
 509                 sc->sc_ntxstream = 1;
 510                 rft = "1T1R";
 511                 break;
 512         case 0x1:
 513                 sc->sc_rftype = RTL8712_RFCONFIG_1T2R;
 514                 sc->sc_nrxstream = 2;
 515                 sc->sc_ntxstream = 1;
 516                 rft = "1T2R";
 517                 break;
 518         case 0x2:
 519                 sc->sc_rftype = RTL8712_RFCONFIG_2T2R;
 520                 sc->sc_nrxstream = 2;
 521                 sc->sc_ntxstream = 2;
 522                 rft = "2T2R";
 523                 break;
 524         case 0x3:       /* "green" NIC */
 525                 sc->sc_rftype = RTL8712_RFCONFIG_1T2R;
 526                 sc->sc_nrxstream = 2;
 527                 sc->sc_ntxstream = 1;
 528                 rft = "1T2R ('green')";
 529                 break;
 530         default:
 531                 device_printf(sc->sc_dev,
 532                     "%s: unknown board type (rfconfig=0x%02x)\n",
 533                     __func__,
 534                     sc->rom[84]);
 535                 goto fail_rom;
 536         }
 537
 538         IEEE80211_ADDR_COPY(ic->ic_macaddr, &sc->rom[0x12]);
 539         device_printf(self, "MAC/BB RTL8712 cut %d %s\n", sc->cut, rft);
 540
 541         ic->ic_softc = sc;
 542         ic->ic_name = device_get_nameunit(self);
 543         ic->ic_phytype = IEEE80211_T_OFDM;      /* Not only, but not used. */
 544         ic->ic_opmode = IEEE80211_M_STA;        /* Default to BSS mode. */
 545
 546         /* Set device capabilities. */
 547         ic->ic_caps =
 548             IEEE80211_C_STA |           /* station mode */
 549             IEEE80211_C_MONITOR |       /* monitor mode supported */
 550 #if 0
 551             IEEE80211_C_BGSCAN |        /* Background scan. */
 552 #endif
 553             IEEE80211_C_SHPREAMBLE |    /* Short preamble supported. */
 554             IEEE80211_C_WME |           /* WME/QoS */
 555             IEEE80211_C_SHSLOT |        /* Short slot time supported. */
 556             IEEE80211_C_WPA;            /* WPA/RSN. */
 557
 558         ic->ic_cryptocaps =
 559             IEEE80211_CRYPTO_WEP |
 560             IEEE80211_CRYPTO_TKIP |
 561             IEEE80211_CRYPTO_AES_CCM;
 562
 563         /* Check if HT support is present. */
 564         if (sc->sc_ht) {
 565                 device_printf(sc->sc_dev, "%s: enabling 11n\n", __func__);
 566
 567                 /* Enable basic HT */
 568                 ic->ic_htcaps = IEEE80211_HTC_HT |
 569 #if 0
 570                     IEEE80211_HTC_AMPDU |
 571 #endif
 572                     IEEE80211_HTC_AMSDU |
 573                     IEEE80211_HTCAP_MAXAMSDU_3839 |
 574                     IEEE80211_HTCAP_SMPS_OFF;
 575                 ic->ic_htcaps |= IEEE80211_HTCAP_CHWIDTH40;
 576
 577                 /* set number of spatial streams */
 578                 ic->ic_txstream = sc->sc_ntxstream;
 579                 ic->ic_rxstream = sc->sc_nrxstream;
 580         }
 581         ic->ic_flags_ext |= IEEE80211_FEXT_SCAN_OFFLOAD;
 582
 583         rsu_getradiocaps(ic, IEEE80211_CHAN_MAX, &ic->ic_nchans,
 584             ic->ic_channels);
 585
 586         ieee80211_ifattach(ic);
 587         ic->ic_raw_xmit = rsu_raw_xmit;
 588         ic->ic_scan_start = rsu_scan_start;
 589         ic->ic_scan_end = rsu_scan_end;
 590         ic->ic_getradiocaps = rsu_getradiocaps;
 591         ic->ic_set_channel = rsu_set_channel;
 592         ic->ic_scan_curchan = rsu_scan_curchan;
 593         ic->ic_scan_mindwell = rsu_scan_mindwell;
 594         ic->ic_vap_create = rsu_vap_create;
 595         ic->ic_vap_delete = rsu_vap_delete;
 596         ic->ic_update_promisc = rsu_update_promisc;
 597         ic->ic_update_mcast = rsu_update_mcast;
 598         ic->ic_ioctl = rsu_ioctl_net;
 599         ic->ic_parent = rsu_parent;
 600         ic->ic_transmit = rsu_transmit;
 601         ic->ic_send_mgmt = rsu_send_mgmt;
 602         ic->ic_update_chw = rsu_update_chw;
 603         ic->ic_ampdu_enable = rsu_ampdu_enable;
 604         ic->ic_wme.wme_update = rsu_wme_update;
 605
 606         ieee80211_radiotap_attach(ic, &sc->sc_txtap.wt_ihdr,
 607             sizeof(sc->sc_txtap), RSU_TX_RADIOTAP_PRESENT,
 608             &sc->sc_rxtap.wr_ihdr, sizeof(sc->sc_rxtap),
 609             RSU_RX_RADIOTAP_PRESENT);
 610
 611         if (bootverbose)
 612                 ieee80211_announce(ic);
 613
 614         return (0);
 615
 616 fail_rom:
 617         usbd_transfer_unsetup(sc->sc_xfer, RSU_N_TRANSFER);
 618 fail_usb:
 619         mtx_destroy(&sc->sc_mtx);
 620         return (ENXIO);
 621 }
 622
 623 static int
 624 rsu_detach(device_t self)
 625 {
 626         struct rsu_softc *sc = device_get_softc(self);
 627         struct ieee80211com *ic = &sc->sc_ic;
 628
 629         rsu_stop(sc);
 630
 631         usbd_transfer_unsetup(sc->sc_xfer, RSU_N_TRANSFER);
 632
 633         /*
 634          * Free buffers /before/ we detach from net80211, else node
 635          * references to destroyed vaps will lead to a panic.
 636          */
 637         /* Free Tx/Rx buffers. */
 638         RSU_LOCK(sc);
 639         rsu_free_tx_list(sc);
 640         rsu_free_rx_list(sc);
 641         RSU_UNLOCK(sc);
 642
 643         /* Frames are freed; detach from net80211 */
 644         ieee80211_ifdetach(ic);
 645
 646         taskqueue_drain_timeout(taskqueue_thread, &sc->calib_task);
 647         taskqueue_drain(taskqueue_thread, &sc->del_key_task);
 648         taskqueue_drain(taskqueue_thread, &sc->tx_task);
 649
 650         RSU_DELKEY_BMAP_LOCK_DESTROY(sc);
 651         mtx_destroy(&sc->sc_mtx);
 652
 653         return (0);
 654 }
 655
 656 static usb_error_t
 657 rsu_do_request(struct rsu_softc *sc, struct usb_device_request *req,
 658     void *data)
 659 {
 660         usb_error_t err;
 661         int ntries = 10;
 662
 663         RSU_ASSERT_LOCKED(sc);
 664
 665         while (ntries--) {
 666                 err = usbd_do_request_flags(sc->sc_udev, &sc->sc_mtx,
 667                     req, data, 0, NULL, 250 /* ms */);
 668                 if (err == 0 || err == USB_ERR_NOT_CONFIGURED)
 669                         break;
 670                 RSU_DPRINTF(sc, RSU_DEBUG_USB,
 671                     "Control request failed, %s (retries left: %d)\n",
 672                     usbd_errstr(err), ntries);
 673                 rsu_ms_delay(sc, 10);
 674         }
 675
 676         return (err);
 677 }
 678
 679 static struct ieee80211vap *
 680 rsu_vap_create(struct ieee80211com *ic, const char name[IFNAMSIZ], int unit,
 681     enum ieee80211_opmode opmode, int flags,
 682     const uint8_t bssid[IEEE80211_ADDR_LEN],
 683     const uint8_t mac[IEEE80211_ADDR_LEN])
 684 {
 685         struct rsu_softc *sc = ic->ic_softc;
 686         struct rsu_vap *uvp;
 687         struct ieee80211vap *vap;
 688         struct ifnet *ifp;
 689
 690         if (!TAILQ_EMPTY(&ic->ic_vaps))         /* only one at a time */
 691                 return (NULL);
 692
 693         uvp =  malloc(sizeof(struct rsu_vap), M_80211_VAP, M_WAITOK | M_ZERO);
 694         vap = &uvp->vap;
 695
 696         if (ieee80211_vap_setup(ic, vap, name, unit, opmode,
 697             flags, bssid) != 0) {
 698                 /* out of memory */
 699                 free(uvp, M_80211_VAP);
 700                 return (NULL);
 701         }
 702
 703         ifp = vap->iv_ifp;
 704         ifp->if_capabilities = IFCAP_RXCSUM | IFCAP_RXCSUM_IPV6;
 705         RSU_LOCK(sc);
 706         if (sc->sc_rx_checksum_enable)
 707                 ifp->if_capenable |= IFCAP_RXCSUM | IFCAP_RXCSUM_IPV6;
 708         RSU_UNLOCK(sc);
 709
 710         /* override state transition machine */
 711         uvp->newstate = vap->iv_newstate;
 712         if (opmode == IEEE80211_M_MONITOR)
 713                 vap->iv_newstate = rsu_monitor_newstate;
 714         else
 715                 vap->iv_newstate = rsu_newstate;
 716         vap->iv_key_alloc = rsu_key_alloc;
 717         vap->iv_key_set = rsu_key_set;
 718         vap->iv_key_delete = rsu_key_delete;
 719
 720         /* Limits from the r92su driver */
 721         vap->iv_ampdu_density = IEEE80211_HTCAP_MPDUDENSITY_16;
 722         vap->iv_ampdu_rxmax = IEEE80211_HTCAP_MAXRXAMPDU_32K;
 723
 724         /* complete setup */
 725         ieee80211_vap_attach(vap, ieee80211_media_change,
 726             ieee80211_media_status, mac);
 727         ic->ic_opmode = opmode;
 728
 729         return (vap);
 730 }
 731
 732 static void
 733 rsu_vap_delete(struct ieee80211vap *vap)
 734 {
 735         struct rsu_vap *uvp = RSU_VAP(vap);
 736
 737         ieee80211_vap_detach(vap);
 738         free(uvp, M_80211_VAP);
 739 }
 740
 741 static void
 742 rsu_scan_start(struct ieee80211com *ic)
 743 {
 744         struct rsu_softc *sc = ic->ic_softc;
 745         struct ieee80211_scan_state *ss = ic->ic_scan;
 746         struct ieee80211vap *vap = TAILQ_FIRST(&ic->ic_vaps);
 747         int error;
 748
 749         /* Scanning is done by the firmware. */
 750         RSU_LOCK(sc);
 751         sc->sc_active_scan = !!(ss->ss_flags & IEEE80211_SCAN_ACTIVE);
 752         /* XXX TODO: force awake if in network-sleep? */
 753         error = rsu_site_survey(sc, ss->ss_nssid > 0 ? &ss->ss_ssid[0] : NULL);
 754         RSU_UNLOCK(sc);
 755         if (error != 0) {
 756                 device_printf(sc->sc_dev,
 757                     "could not send site survey command\n");
 758                 ieee80211_cancel_scan(vap);
 759         }
 760 }
 761
 762 static void
 763 rsu_scan_end(struct ieee80211com *ic)
 764 {
 765         /* Nothing to do here. */
 766 }
 767
 768 static void
 769 rsu_getradiocaps(struct ieee80211com *ic,
 770     int maxchans, int *nchans, struct ieee80211_channel chans[])
 771 {
 772         struct rsu_softc *sc = ic->ic_softc;
 773         uint8_t bands[IEEE80211_MODE_BYTES];
 774
 775         /* Set supported .11b and .11g rates. */
 776         memset(bands, 0, sizeof(bands));
 777         setbit(bands, IEEE80211_MODE_11B);
 778         setbit(bands, IEEE80211_MODE_11G);
 779         if (sc->sc_ht)
 780                 setbit(bands, IEEE80211_MODE_11NG);
 781         ieee80211_add_channels_default_2ghz(chans, maxchans, nchans,
 782             bands, (ic->ic_htcaps & IEEE80211_HTCAP_CHWIDTH40) != 0);
 783 }
 784
 785 static void
 786 rsu_set_channel(struct ieee80211com *ic)
 787 {
 788         struct rsu_softc *sc = ic->ic_softc;
 789
 790         /*
 791          * Only need to set the channel in Monitor mode. AP scanning and auth
 792          * are already taken care of by their respective firmware commands.
 793          */
 794         if (ic->ic_opmode == IEEE80211_M_MONITOR) {
 795                 struct r92s_set_channel cmd;
 796                 int error;
 797
 798                 cmd.channel = IEEE80211_CHAN2IEEE(ic->ic_curchan);
 799
 800                 RSU_LOCK(sc);
 801                 error = rsu_fw_cmd(sc, R92S_CMD_SET_CHANNEL, &cmd,
 802                     sizeof(cmd));
 803                 if (error != 0) {
 804                         device_printf(sc->sc_dev,
 805                             "%s: error %d setting channel\n", __func__,
 806                             error);
 807                 }
 808                 RSU_UNLOCK(sc);
 809         }
 810 }
 811
 812 static void
 813 rsu_scan_curchan(struct ieee80211_scan_state *ss, unsigned long maxdwell)
 814 {
 815         /* Scan is done in rsu_scan_start(). */
 816 }
 817
 818 /**
 819  * Called by the net80211 framework to indicate
 820  * the minimum dwell time has been met, terminate the scan.
 821  * We don't actually terminate the scan as the firmware will notify
 822  * us when it's finished and we have no way to interrupt it.
 823  */
 824 static void
 825 rsu_scan_mindwell(struct ieee80211_scan_state *ss)
 826 {
 827         /* NB: don't try to abort scan; wait for firmware to finish */
 828 }
 829
 830 static void
 831 rsu_update_promisc(struct ieee80211com *ic)
 832 {
 833         struct rsu_softc *sc = ic->ic_softc;
 834
 835         RSU_LOCK(sc);
 836         if (sc->sc_running)
 837                 rsu_rxfilter_refresh(sc);
 838         RSU_UNLOCK(sc);
 839 }
 840
 841 /*
 842  * The same as rtwn_get_multi_pos() / rtwn_set_multi().
 843  */
 844 static uint8_t
 845 rsu_get_multi_pos(const uint8_t maddr[])
 846 {
 847         uint64_t mask = 0x00004d101df481b4;
 848         uint8_t pos = 0x27;     /* initial value */
 849         int i, j;
 850
 851         for (i = 0; i < IEEE80211_ADDR_LEN; i++)
 852                 for (j = (i == 0) ? 1 : 0; j < 8; j++)
 853                         if ((maddr[i] >> j) & 1)
 854                                 pos ^= (mask >> (i * 8 + j - 1));
 855
 856         pos &= 0x3f;
 857
 858         return (pos);
 859 }
 860
 861 static u_int
 862 rsu_hash_maddr(void *arg, struct sockaddr_dl *sdl, u_int cnt)
 863 {
 864         uint32_t *mfilt = arg;
 865         uint8_t pos;
 866
 867         pos = rsu_get_multi_pos(LLADDR(sdl));
 868         mfilt[pos / 32] |= (1 << (pos % 32));
 869
 870         return (1);
 871 }
 872
 873 static void
 874 rsu_set_multi(struct rsu_softc *sc)
 875 {
 876         struct ieee80211com *ic = &sc->sc_ic;
 877         uint32_t mfilt[2];
 878
 879         RSU_ASSERT_LOCKED(sc);
 880
 881         /* general structure was copied from ath(4). */
 882         if (ic->ic_allmulti == 0) {
 883                 struct ieee80211vap *vap;
 884
 885                 /*
 886                  * Merge multicast addresses to form the hardware filter.
 887                  */
 888                 mfilt[0] = mfilt[1] = 0;
 889                 TAILQ_FOREACH(vap, &ic->ic_vaps, iv_next)
 890                         if_foreach_llmaddr(vap->iv_ifp, rsu_hash_maddr, &mfilt);
 891         } else
 892                 mfilt[0] = mfilt[1] = ~0;
 893
 894         rsu_write_4(sc, R92S_MAR + 0, mfilt[0]);
 895         rsu_write_4(sc, R92S_MAR + 4, mfilt[1]);
 896
 897         RSU_DPRINTF(sc, RSU_DEBUG_STATE, "%s: MC filter %08x:%08x\n",
 898             __func__, mfilt[0], mfilt[1]);
 899 }
 900
 901 static void
 902 rsu_update_mcast(struct ieee80211com *ic)
 903 {
 904         struct rsu_softc *sc = ic->ic_softc;
 905
 906         RSU_LOCK(sc);
 907         if (sc->sc_running)
 908                 rsu_set_multi(sc);
 909         RSU_UNLOCK(sc);
 910 }
 911
 912 static int
 913 rsu_alloc_list(struct rsu_softc *sc, struct rsu_data data[],
 914     int ndata, int maxsz)
 915 {
 916         int i, error;
 917
 918         for (i = 0; i < ndata; i++) {
 919                 struct rsu_data *dp = &data[i];
 920                 dp->sc = sc;
 921                 dp->m = NULL;
 922                 dp->buf = malloc(maxsz, M_USBDEV, M_NOWAIT);
 923                 if (dp->buf == NULL) {
 924                         device_printf(sc->sc_dev,
 925                             "could not allocate buffer\n");
 926                         error = ENOMEM;
 927                         goto fail;
 928                 }
 929                 dp->ni = NULL;
 930         }
 931
 932         return (0);
 933 fail:
 934         rsu_free_list(sc, data, ndata);
 935         return (error);
 936 }
 937
 938 static int
 939 rsu_alloc_rx_list(struct rsu_softc *sc)
 940 {
 941         int error, i;
 942
 943         error = rsu_alloc_list(sc, sc->sc_rx, RSU_RX_LIST_COUNT,
 944             RSU_RXBUFSZ);
 945         if (error != 0)
 946                 return (error);
 947
 948         STAILQ_INIT(&sc->sc_rx_active);
 949         STAILQ_INIT(&sc->sc_rx_inactive);
 950
 951         for (i = 0; i < RSU_RX_LIST_COUNT; i++)
 952                 STAILQ_INSERT_HEAD(&sc->sc_rx_inactive, &sc->sc_rx[i], next);
 953
 954         return (0);
 955 }
 956
 957 static int
 958 rsu_alloc_tx_list(struct rsu_softc *sc)
 959 {
 960         int error, i;
 961
 962         error = rsu_alloc_list(sc, sc->sc_tx, RSU_TX_LIST_COUNT,
 963             RSU_TXBUFSZ);
 964         if (error != 0)
 965                 return (error);
 966
 967         STAILQ_INIT(&sc->sc_tx_inactive);
 968
 969         for (i = 0; i != RSU_N_TRANSFER; i++) {
 970                 STAILQ_INIT(&sc->sc_tx_active[i]);
 971                 STAILQ_INIT(&sc->sc_tx_pending[i]);
 972         }
 973
 974         for (i = 0; i < RSU_TX_LIST_COUNT; i++) {
 975                 STAILQ_INSERT_HEAD(&sc->sc_tx_inactive, &sc->sc_tx[i], next);
 976         }
 977
 978         return (0);
 979 }
 980
 981 static void
 982 rsu_free_tx_list(struct rsu_softc *sc)
 983 {
 984         int i;
 985
 986         /* prevent further allocations from TX list(s) */
 987         STAILQ_INIT(&sc->sc_tx_inactive);
 988
 989         for (i = 0; i != RSU_N_TRANSFER; i++) {
 990                 STAILQ_INIT(&sc->sc_tx_active[i]);
 991                 STAILQ_INIT(&sc->sc_tx_pending[i]);
 992         }
 993
 994         rsu_free_list(sc, sc->sc_tx, RSU_TX_LIST_COUNT);
 995 }
 996
 997 static void
 998 rsu_free_rx_list(struct rsu_softc *sc)
 999 {
1000         /* prevent further allocations from RX list(s) */
1001         STAILQ_INIT(&sc->sc_rx_inactive);
1002         STAILQ_INIT(&sc->sc_rx_active);
1003
1004         rsu_free_list(sc, sc->sc_rx, RSU_RX_LIST_COUNT);
1005 }
1006
1007 static void
1008 rsu_free_list(struct rsu_softc *sc, struct rsu_data data[], int ndata)
1009 {
1010         int i;
1011
1012         for (i = 0; i < ndata; i++) {
1013                 struct rsu_data *dp = &data[i];
1014
1015                 if (dp->buf != NULL) {
1016                         free(dp->buf, M_USBDEV);
1017                         dp->buf = NULL;
1018                 }
1019                 if (dp->ni != NULL) {
1020                         ieee80211_free_node(dp->ni);
1021                         dp->ni = NULL;
1022                 }
1023         }
1024 }
1025
1026 static struct rsu_data *
1027 _rsu_getbuf(struct rsu_softc *sc)
1028 {
1029         struct rsu_data *bf;
1030
1031         bf = STAILQ_FIRST(&sc->sc_tx_inactive);
1032         if (bf != NULL)
1033                 STAILQ_REMOVE_HEAD(&sc->sc_tx_inactive, next);
1034         else
1035                 bf = NULL;
1036         return (bf);
1037 }
1038
1039 static struct rsu_data *
1040 rsu_getbuf(struct rsu_softc *sc)
1041 {
1042         struct rsu_data *bf;
1043
1044         RSU_ASSERT_LOCKED(sc);
1045
1046         bf = _rsu_getbuf(sc);
1047         if (bf == NULL) {
1048                 RSU_DPRINTF(sc, RSU_DEBUG_TX, "%s: no buffers\n", __func__);
1049         }
1050         return (bf);
1051 }
1052
1053 static void
1054 rsu_freebuf(struct rsu_softc *sc, struct rsu_data *bf)
1055 {
1056
1057         RSU_ASSERT_LOCKED(sc);
1058         STAILQ_INSERT_TAIL(&sc->sc_tx_inactive, bf, next);
1059 }
1060
1061 static int
1062 rsu_write_region_1(struct rsu_softc *sc, uint16_t addr, uint8_t *buf,
1063     int len)
1064 {
1065         usb_device_request_t req;
1066
1067         req.bmRequestType = UT_WRITE_VENDOR_DEVICE;
1068         req.bRequest = R92S_REQ_REGS;
1069         USETW(req.wValue, addr);
1070         USETW(req.wIndex, 0);
1071         USETW(req.wLength, len);
1072
1073         return (rsu_do_request(sc, &req, buf));
1074 }
1075
1076 static void
1077 rsu_write_1(struct rsu_softc *sc, uint16_t addr, uint8_t val)
1078 {
1079         rsu_write_region_1(sc, addr, &val, 1);
1080 }
1081
1082 static void
1083 rsu_write_2(struct rsu_softc *sc, uint16_t addr, uint16_t val)
1084 {
1085         val = htole16(val);
1086         rsu_write_region_1(sc, addr, (uint8_t *)&val, 2);
1087 }
1088
1089 static void
1090 rsu_write_4(struct rsu_softc *sc, uint16_t addr, uint32_t val)
1091 {
1092         val = htole32(val);
1093         rsu_write_region_1(sc, addr, (uint8_t *)&val, 4);
1094 }
1095
1096 static int
1097 rsu_read_region_1(struct rsu_softc *sc, uint16_t addr, uint8_t *buf,
1098     int len)
1099 {
1100         usb_device_request_t req;
1101
1102         req.bmRequestType = UT_READ_VENDOR_DEVICE;
1103         req.bRequest = R92S_REQ_REGS;
1104         USETW(req.wValue, addr);
1105         USETW(req.wIndex, 0);
1106         USETW(req.wLength, len);
1107
1108         return (rsu_do_request(sc, &req, buf));
1109 }
1110
1111 static uint8_t
1112 rsu_read_1(struct rsu_softc *sc, uint16_t addr)
1113 {
1114         uint8_t val;
1115
1116         if (rsu_read_region_1(sc, addr, &val, 1) != 0)
1117                 return (0xff);
1118         return (val);
1119 }
1120
1121 static uint16_t
1122 rsu_read_2(struct rsu_softc *sc, uint16_t addr)
1123 {
1124         uint16_t val;
1125
1126         if (rsu_read_region_1(sc, addr, (uint8_t *)&val, 2) != 0)
1127                 return (0xffff);
1128         return (le16toh(val));
1129 }
1130
1131 static uint32_t
1132 rsu_read_4(struct rsu_softc *sc, uint16_t addr)
1133 {
1134         uint32_t val;
1135
1136         if (rsu_read_region_1(sc, addr, (uint8_t *)&val, 4) != 0)
1137                 return (0xffffffff);
1138         return (le32toh(val));
1139 }
1140
1141 static int
1142 rsu_fw_iocmd(struct rsu_softc *sc, uint32_t iocmd)
1143 {
1144         int ntries;
1145
1146         rsu_write_4(sc, R92S_IOCMD_CTRL, iocmd);
1147         rsu_ms_delay(sc, 1);
1148         for (ntries = 0; ntries < 50; ntries++) {
1149                 if (rsu_read_4(sc, R92S_IOCMD_CTRL) == 0)
1150                         return (0);
1151                 rsu_ms_delay(sc, 1);
1152         }
1153         return (ETIMEDOUT);
1154 }
1155
1156 static uint8_t
1157 rsu_efuse_read_1(struct rsu_softc *sc, uint16_t addr)
1158 {
1159         uint32_t reg;
1160         int ntries;
1161
1162         reg = rsu_read_4(sc, R92S_EFUSE_CTRL);
1163         reg = RW(reg, R92S_EFUSE_CTRL_ADDR, addr);
1164         reg &= ~R92S_EFUSE_CTRL_VALID;
1165         rsu_write_4(sc, R92S_EFUSE_CTRL, reg);
1166         /* Wait for read operation to complete. */
1167         for (ntries = 0; ntries < 100; ntries++) {
1168                 reg = rsu_read_4(sc, R92S_EFUSE_CTRL);
1169                 if (reg & R92S_EFUSE_CTRL_VALID)
1170                         return (MS(reg, R92S_EFUSE_CTRL_DATA));
1171                 rsu_ms_delay(sc, 1);
1172         }
1173         device_printf(sc->sc_dev,
1174             "could not read efuse byte at address 0x%x\n", addr);
1175         return (0xff);
1176 }
1177
1178 static int
1179 rsu_read_rom(struct rsu_softc *sc)
1180 {
1181         uint8_t *rom = sc->rom;
1182         uint16_t addr = 0;
1183         uint32_t reg;
1184         uint8_t off, msk;
1185         int i;
1186
1187         /* Make sure that ROM type is eFuse and that autoload succeeded. */
1188         reg = rsu_read_1(sc, R92S_EE_9346CR);
1189         if ((reg & (R92S_9356SEL | R92S_EEPROM_EN)) != R92S_EEPROM_EN)
1190                 return (EIO);
1191
1192         /* Turn on 2.5V to prevent eFuse leakage. */
1193         reg = rsu_read_1(sc, R92S_EFUSE_TEST + 3);
1194         rsu_write_1(sc, R92S_EFUSE_TEST + 3, reg | 0x80);
1195         rsu_ms_delay(sc, 1);
1196         rsu_write_1(sc, R92S_EFUSE_TEST + 3, reg & ~0x80);
1197
1198         /* Read full ROM image. */
1199         memset(&sc->rom, 0xff, sizeof(sc->rom));
1200         while (addr < 512) {
1201                 reg = rsu_efuse_read_1(sc, addr);
1202                 if (reg == 0xff)
1203                         break;
1204                 addr++;
1205                 off = reg >> 4;
1206                 msk = reg & 0xf;
1207                 for (i = 0; i < 4; i++) {
1208                         if (msk & (1 << i))
1209                                 continue;
1210                         rom[off * 8 + i * 2 + 0] =
1211                             rsu_efuse_read_1(sc, addr);
1212                         addr++;
1213                         rom[off * 8 + i * 2 + 1] =
1214                             rsu_efuse_read_1(sc, addr);
1215                         addr++;
1216                 }
1217         }
1218 #ifdef USB_DEBUG
1219         if (rsu_debug & RSU_DEBUG_RESET) {
1220                 /* Dump ROM content. */
1221                 printf("\n");
1222                 for (i = 0; i < sizeof(sc->rom); i++)
1223                         printf("%02x:", rom[i]);
1224                 printf("\n");
1225         }
1226 #endif
1227         return (0);
1228 }
1229
1230 static int
1231 rsu_fw_cmd(struct rsu_softc *sc, uint8_t code, void *buf, int len)
1232 {
1233         const uint8_t which = RSU_H2C_ENDPOINT;
1234         struct rsu_data *data;
1235         struct r92s_tx_desc *txd;
1236         struct r92s_fw_cmd_hdr *cmd;
1237         int cmdsz;
1238         int xferlen;
1239
1240         RSU_ASSERT_LOCKED(sc);
1241
1242         data = rsu_getbuf(sc);
1243         if (data == NULL)
1244                 return (ENOMEM);
1245
1246         /* Blank the entire payload, just to be safe */
1247         memset(data->buf, '\0', RSU_TXBUFSZ);
1248
1249         /* Round-up command length to a multiple of 8 bytes. */
1250         /* XXX TODO: is this required? */
1251         cmdsz = (len + 7) & ~7;
1252
1253         xferlen = sizeof(*txd) + sizeof(*cmd) + cmdsz;
1254         KASSERT(xferlen <= RSU_TXBUFSZ, ("%s: invalid length", __func__));
1255         memset(data->buf, 0, xferlen);
1256
1257         /* Setup Tx descriptor. */
1258         txd = (struct r92s_tx_desc *)data->buf;
1259         txd->txdw0 = htole32(
1260             SM(R92S_TXDW0_OFFSET, sizeof(*txd)) |
1261             SM(R92S_TXDW0_PKTLEN, sizeof(*cmd) + cmdsz) |
1262             R92S_TXDW0_OWN | R92S_TXDW0_FSG | R92S_TXDW0_LSG);
1263         txd->txdw1 = htole32(SM(R92S_TXDW1_QSEL, R92S_TXDW1_QSEL_H2C));
1264
1265         /* Setup command header. */
1266         cmd = (struct r92s_fw_cmd_hdr *)&txd[1];
1267         cmd->len = htole16(cmdsz);
1268         cmd->code = code;
1269         cmd->seq = sc->cmd_seq;
1270         sc->cmd_seq = (sc->cmd_seq + 1) & 0x7f;
1271
1272         /* Copy command payload. */
1273         memcpy(&cmd[1], buf, len);
1274
1275         RSU_DPRINTF(sc, RSU_DEBUG_TX | RSU_DEBUG_FWCMD,
1276             "%s: Tx cmd code=0x%x len=0x%x\n",
1277             __func__, code, cmdsz);
1278         data->buflen = xferlen;
1279         STAILQ_INSERT_TAIL(&sc->sc_tx_pending[which], data, next);
1280         usbd_transfer_start(sc->sc_xfer[which]);
1281
1282         return (0);
1283 }
1284
1285 /* ARGSUSED */
1286 static void
1287 rsu_calib_task(void *arg, int pending __unused)
1288 {
1289         struct rsu_softc *sc = arg;
1290 #ifdef notyet
1291         uint32_t reg;
1292 #endif
1293
1294         RSU_DPRINTF(sc, RSU_DEBUG_CALIB, "%s: running calibration task\n",
1295             __func__);
1296
1297         RSU_LOCK(sc);
1298 #ifdef notyet
1299         /* Read WPS PBC status. */
1300         rsu_write_1(sc, R92S_MAC_PINMUX_CTRL,
1301             R92S_GPIOMUX_EN | SM(R92S_GPIOSEL_GPIO, R92S_GPIOSEL_GPIO_JTAG));
1302         rsu_write_1(sc, R92S_GPIO_IO_SEL,
1303             rsu_read_1(sc, R92S_GPIO_IO_SEL) & ~R92S_GPIO_WPS);
1304         reg = rsu_read_1(sc, R92S_GPIO_CTRL);
1305         if (reg != 0xff && (reg & R92S_GPIO_WPS))
1306                 RSU_DPRINTF(sc, RSU_DEBUG_CALIB, "WPS PBC is pushed\n");
1307 #endif
1308         /* Read current signal level. */
1309         if (rsu_fw_iocmd(sc, 0xf4000001) == 0) {
1310                 sc->sc_currssi = rsu_read_4(sc, R92S_IOCMD_DATA);
1311                 RSU_DPRINTF(sc, RSU_DEBUG_CALIB, "%s: RSSI=%d (%d)\n",
1312                     __func__, sc->sc_currssi,
1313                     rsu_hwrssi_to_rssi(sc, sc->sc_currssi));
1314         }
1315         if (sc->sc_calibrating)
1316                 taskqueue_enqueue_timeout(taskqueue_thread, &sc->calib_task, hz);
1317         RSU_UNLOCK(sc);
1318 }
1319
1320 static void
1321 rsu_tx_task(void *arg, int pending __unused)
1322 {
1323         struct rsu_softc *sc = arg;
1324
1325         RSU_LOCK(sc);
1326         _rsu_start(sc);
1327         RSU_UNLOCK(sc);
1328 }
1329
1330 #define RSU_PWR_UNKNOWN         0x0
1331 #define RSU_PWR_ACTIVE          0x1
1332 #define RSU_PWR_OFF             0x2
1333 #define RSU_PWR_SLEEP           0x3
1334
1335 /*
1336  * Set the current power state.
1337  *
1338  * The rtlwifi code doesn't do this so aggressively; it
1339  * waits for an idle period after association with
1340  * no traffic before doing this.
1341  *
1342  * For now - it's on in all states except RUN, and
1343  * in RUN it'll transition to allow sleep.
1344  */
1345
1346 struct r92s_pwr_cmd {
1347         uint8_t mode;
1348         uint8_t smart_ps;
1349         uint8_t bcn_pass_time;
1350 };
1351
1352 static int
1353 rsu_set_fw_power_state(struct rsu_softc *sc, int state)
1354 {
1355         struct r92s_set_pwr_mode cmd;
1356         //struct r92s_pwr_cmd cmd;
1357         int error;
1358
1359         RSU_ASSERT_LOCKED(sc);
1360
1361         /* only change state if required */
1362         if (sc->sc_curpwrstate == state)
1363                 return (0);
1364
1365         memset(&cmd, 0, sizeof(cmd));
1366
1367         switch (state) {
1368         case RSU_PWR_ACTIVE:
1369                 /* Force the hardware awake */
1370                 rsu_write_1(sc, R92S_USB_HRPWM,
1371                     R92S_USB_HRPWM_PS_ST_ACTIVE | R92S_USB_HRPWM_PS_ALL_ON);
1372                 cmd.mode = R92S_PS_MODE_ACTIVE;
1373                 break;
1374         case RSU_PWR_SLEEP:
1375                 cmd.mode = R92S_PS_MODE_DTIM;   /* XXX configurable? */
1376                 cmd.smart_ps = 1; /* XXX 2 if doing p2p */
1377                 cmd.bcn_pass_time = 5; /* in 100mS usb.c, linux/rtlwifi */
1378                 break;
1379         case RSU_PWR_OFF:
1380                 cmd.mode = R92S_PS_MODE_RADIOOFF;
1381                 break;
1382         default:
1383                 device_printf(sc->sc_dev, "%s: unknown ps mode (%d)\n",
1384                     __func__,
1385                     state);
1386                 return (ENXIO);
1387         }
1388
1389         RSU_DPRINTF(sc, RSU_DEBUG_RESET,
1390             "%s: setting ps mode to %d (mode %d)\n",
1391             __func__, state, cmd.mode);
1392         error = rsu_fw_cmd(sc, R92S_CMD_SET_PWR_MODE, &cmd, sizeof(cmd));
1393         if (error == 0)
1394                 sc->sc_curpwrstate = state;
1395
1396         return (error);
1397 }
1398
1399 static void
1400 rsu_set_led(struct rsu_softc *sc, int on)
1401 {
1402         rsu_write_1(sc, R92S_LEDCFG,
1403             (rsu_read_1(sc, R92S_LEDCFG) & 0xf0) | (!on << 3));
1404 }
1405
1406 static int
1407 rsu_monitor_newstate(struct ieee80211vap *vap, enum ieee80211_state nstate,
1408     int arg)
1409 {
1410         struct ieee80211com *ic = vap->iv_ic;
1411         struct rsu_softc *sc = ic->ic_softc;
1412         struct rsu_vap *uvp = RSU_VAP(vap);
1413
1414         if (vap->iv_state != nstate) {
1415                 IEEE80211_UNLOCK(ic);
1416                 RSU_LOCK(sc);
1417
1418                 switch (nstate) {
1419                 case IEEE80211_S_INIT:
1420                         sc->sc_vap_is_running = 0;
1421                         rsu_set_led(sc, 0);
1422                         break;
1423                 case IEEE80211_S_RUN:
1424                         sc->sc_vap_is_running = 1;
1425                         rsu_set_led(sc, 1);
1426                         break;
1427                 default:
1428                         /* NOTREACHED */
1429                         break;
1430                 }
1431                 rsu_rxfilter_refresh(sc);
1432
1433                 RSU_UNLOCK(sc);
1434                 IEEE80211_LOCK(ic);
1435         }
1436
1437         return (uvp->newstate(vap, nstate, arg));
1438 }
1439
1440 static int
1441 rsu_newstate(struct ieee80211vap *vap, enum ieee80211_state nstate, int arg)
1442 {
1443         struct rsu_vap *uvp = RSU_VAP(vap);
1444         struct ieee80211com *ic = vap->iv_ic;
1445         struct rsu_softc *sc = ic->ic_softc;
1446         struct ieee80211_node *ni;
1447         struct ieee80211_rateset *rs;
1448         enum ieee80211_state ostate;
1449         int error, startcal = 0;
1450
1451         ostate = vap->iv_state;
1452         RSU_DPRINTF(sc, RSU_DEBUG_STATE, "%s: %s -> %s\n",
1453             __func__,
1454             ieee80211_state_name[ostate],
1455             ieee80211_state_name[nstate]);
1456
1457         IEEE80211_UNLOCK(ic);
1458         if (ostate == IEEE80211_S_RUN) {
1459                 RSU_LOCK(sc);
1460                 /* Stop calibration. */
1461                 sc->sc_calibrating = 0;
1462
1463                 /* Pause Tx for AC queues. */
1464                 rsu_write_1(sc, R92S_TXPAUSE, R92S_TXPAUSE_AC);
1465                 usb_pause_mtx(&sc->sc_mtx, USB_MS_TO_TICKS(10));
1466
1467                 RSU_UNLOCK(sc);
1468                 taskqueue_drain_timeout(taskqueue_thread, &sc->calib_task);
1469                 taskqueue_drain(taskqueue_thread, &sc->tx_task);
1470                 RSU_LOCK(sc);
1471                 /* Disassociate from our current BSS. */
1472                 rsu_disconnect(sc);
1473                 usb_pause_mtx(&sc->sc_mtx, USB_MS_TO_TICKS(10));
1474
1475                 /* Refresh Rx filter (may be modified by firmware). */
1476                 sc->sc_vap_is_running = 0;
1477                 rsu_rxfilter_refresh(sc);
1478
1479                 /* Reinstall static keys. */
1480                 if (sc->sc_running)
1481                         rsu_reinit_static_keys(sc);
1482         } else
1483                 RSU_LOCK(sc);
1484         switch (nstate) {
1485         case IEEE80211_S_INIT:
1486                 (void) rsu_set_fw_power_state(sc, RSU_PWR_ACTIVE);
1487                 break;
1488         case IEEE80211_S_AUTH:
1489                 ni = ieee80211_ref_node(vap->iv_bss);
1490                 (void) rsu_set_fw_power_state(sc, RSU_PWR_ACTIVE);
1491                 error = rsu_join_bss(sc, ni);
1492                 ieee80211_free_node(ni);
1493                 if (error != 0) {
1494                         device_printf(sc->sc_dev,
1495                             "could not send join command\n");
1496                 }
1497                 break;
1498         case IEEE80211_S_RUN:
1499                 /* Flush all AC queues. */
1500                 rsu_write_1(sc, R92S_TXPAUSE, 0);
1501
1502                 ni = ieee80211_ref_node(vap->iv_bss);
1503                 rs = &ni->ni_rates;
1504                 /* Indicate highest supported rate. */
1505                 ni->ni_txrate = rs->rs_rates[rs->rs_nrates - 1];
1506                 (void) rsu_set_fw_power_state(sc, RSU_PWR_SLEEP);
1507                 ieee80211_free_node(ni);
1508                 startcal = 1;
1509                 break;
1510         default:
1511                 break;
1512         }
1513         if (startcal != 0) {
1514                 sc->sc_calibrating = 1;
1515                 /* Start periodic calibration. */
1516                 taskqueue_enqueue_timeout(taskqueue_thread, &sc->calib_task,
1517                     hz);
1518         }
1519         RSU_UNLOCK(sc);
1520         IEEE80211_LOCK(ic);
1521         return (uvp->newstate(vap, nstate, arg));
1522 }
1523
1524 static int
1525 rsu_key_alloc(struct ieee80211vap *vap, struct ieee80211_key *k,
1526     ieee80211_keyix *keyix, ieee80211_keyix *rxkeyix)
1527 {
1528         struct rsu_softc *sc = vap->iv_ic->ic_softc;
1529         int is_checked = 0;
1530
1531         if (&vap->iv_nw_keys[0] <= k &&
1532             k < &vap->iv_nw_keys[IEEE80211_WEP_NKID]) {
1533                 *keyix = ieee80211_crypto_get_key_wepidx(vap, k);
1534         } else {
1535                 if (vap->iv_opmode != IEEE80211_M_STA) {
1536                         *keyix = 0;
1537                         /* TODO: obtain keyix from node id */
1538                         is_checked = 1;
1539                         k->wk_flags |= IEEE80211_KEY_SWCRYPT;
1540                 } else
1541                         *keyix = R92S_MACID_BSS;
1542         }
1543
1544         if (!is_checked) {
1545                 RSU_LOCK(sc);
1546                 if (isset(sc->keys_bmap, *keyix)) {
1547                         device_printf(sc->sc_dev,
1548                             "%s: key slot %d is already used!\n",
1549                             __func__, *keyix);
1550                         RSU_UNLOCK(sc);
1551                         return (0);
1552                 }
1553                 setbit(sc->keys_bmap, *keyix);
1554                 RSU_UNLOCK(sc);
1555         }
1556
1557         *rxkeyix = *keyix;
1558
1559         return (1);
1560 }
1561
1562 static int
1563 rsu_process_key(struct ieee80211vap *vap, const struct ieee80211_key *k,
1564     int set)
1565 {
1566         struct rsu_softc *sc = vap->iv_ic->ic_softc;
1567         int ret;
1568
1569         if (k->wk_flags & IEEE80211_KEY_SWCRYPT) {
1570                 /* Not for us. */
1571                 return (1);
1572         }
1573
1574         /* Handle group keys. */
1575         if (&vap->iv_nw_keys[0] <= k &&
1576             k < &vap->iv_nw_keys[IEEE80211_WEP_NKID]) {
1577                 KASSERT(k->wk_keyix < nitems(sc->group_keys),
1578                     ("keyix %u > %zu\n", k->wk_keyix, nitems(sc->group_keys)));
1579
1580                 RSU_LOCK(sc);
1581                 sc->group_keys[k->wk_keyix] = (set ? k : NULL);
1582                 if (!sc->sc_running) {
1583                         /* Static keys will be set during device startup. */
1584                         RSU_UNLOCK(sc);
1585                         return (1);
1586                 }
1587
1588                 if (set)
1589                         ret = rsu_set_key_group(sc, k);
1590                 else
1591                         ret = rsu_delete_key(sc, k->wk_keyix);
1592                 RSU_UNLOCK(sc);
1593
1594                 return (!ret);
1595         }
1596
1597         if (set) {
1598                 /* wait for pending key removal */
1599                 taskqueue_drain(taskqueue_thread, &sc->del_key_task);
1600
1601                 RSU_LOCK(sc);
1602                 ret = rsu_set_key_pair(sc, k);
1603                 RSU_UNLOCK(sc);
1604         } else {
1605                 RSU_DELKEY_BMAP_LOCK(sc);
1606                 setbit(sc->free_keys_bmap, k->wk_keyix);
1607                 RSU_DELKEY_BMAP_UNLOCK(sc);
1608
1609                 /* workaround ieee80211_node_delucastkey() locking */
1610                 taskqueue_enqueue(taskqueue_thread, &sc->del_key_task);
1611                 ret = 0;        /* fake success */
1612         }
1613
1614         return (!ret);
1615 }
1616
1617 static int
1618 rsu_key_set(struct ieee80211vap *vap, const struct ieee80211_key *k)
1619 {
1620         return (rsu_process_key(vap, k, 1));
1621 }
1622
1623 static int
1624 rsu_key_delete(struct ieee80211vap *vap, const struct ieee80211_key *k)
1625 {
1626         return (rsu_process_key(vap, k, 0));
1627 }
1628
1629 static int
1630 rsu_cam_read(struct rsu_softc *sc, uint8_t addr, uint32_t *val)
1631 {
1632         int ntries;
1633
1634         rsu_write_4(sc, R92S_CAMCMD,
1635             R92S_CAMCMD_POLLING | SM(R92S_CAMCMD_ADDR, addr));
1636         for (ntries = 0; ntries < 10; ntries++) {
1637                 if (!(rsu_read_4(sc, R92S_CAMCMD) & R92S_CAMCMD_POLLING))
1638                         break;
1639
1640                 usb_pause_mtx(&sc->sc_mtx, USB_MS_TO_TICKS(1));
1641         }
1642         if (ntries == 10) {
1643                 device_printf(sc->sc_dev,
1644                     "%s: cannot read CAM entry at address %02X\n",
1645                     __func__, addr);
1646                 return (ETIMEDOUT);
1647         }
1648
1649         *val = rsu_read_4(sc, R92S_CAMREAD);
1650
1651         return (0);
1652 }
1653
1654 static void
1655 rsu_cam_write(struct rsu_softc *sc, uint8_t addr, uint32_t data)
1656 {
1657
1658         rsu_write_4(sc, R92S_CAMWRITE, data);
1659         rsu_write_4(sc, R92S_CAMCMD,
1660             R92S_CAMCMD_POLLING | R92S_CAMCMD_WRITE |
1661             SM(R92S_CAMCMD_ADDR, addr));
1662 }
1663
1664 static int
1665 rsu_key_check(struct rsu_softc *sc, ieee80211_keyix keyix, int is_valid)
1666 {
1667         uint32_t val;
1668         int error, ntries;
1669
1670         for (ntries = 0; ntries < 20; ntries++) {
1671                 usb_pause_mtx(&sc->sc_mtx, USB_MS_TO_TICKS(1));
1672
1673                 error = rsu_cam_read(sc, R92S_CAM_CTL0(keyix), &val);
1674                 if (error != 0) {
1675                         device_printf(sc->sc_dev,
1676                             "%s: cannot check key status!\n", __func__);
1677                         return (error);
1678                 }
1679                 if (((val & R92S_CAM_VALID) == 0) ^ is_valid)
1680                         break;
1681         }
1682         if (ntries == 20) {
1683                 device_printf(sc->sc_dev,
1684                     "%s: key %d is %s marked as valid, rejecting request\n",
1685                     __func__, keyix, is_valid ? "not" : "still");
1686                 return (EIO);
1687         }
1688
1689         return (0);
1690 }
1691
1692 /*
1693  * Map net80211 cipher to RTL8712 security mode.
1694  */
1695 static uint8_t
1696 rsu_crypto_mode(struct rsu_softc *sc, u_int cipher, int keylen)
1697 {
1698         switch (cipher) {
1699         case IEEE80211_CIPHER_WEP:
1700                 return keylen < 8 ? R92S_KEY_ALGO_WEP40 : R92S_KEY_ALGO_WEP104;
1701         case IEEE80211_CIPHER_TKIP:
1702                 return R92S_KEY_ALGO_TKIP;
1703         case IEEE80211_CIPHER_AES_CCM:
1704                 return R92S_KEY_ALGO_AES;
1705         default:
1706                 device_printf(sc->sc_dev, "unknown cipher %d\n", cipher);
1707                 return R92S_KEY_ALGO_INVALID;
1708         }
1709 }
1710
1711 static int
1712 rsu_set_key_group(struct rsu_softc *sc, const struct ieee80211_key *k)
1713 {
1714         struct r92s_fw_cmd_set_key key;
1715         uint8_t algo;
1716         int error;
1717
1718         RSU_ASSERT_LOCKED(sc);
1719
1720         /* Map net80211 cipher to HW crypto algorithm. */
1721         algo = rsu_crypto_mode(sc, k->wk_cipher->ic_cipher, k->wk_keylen);
1722         if (algo == R92S_KEY_ALGO_INVALID)
1723                 return (EINVAL);
1724
1725         memset(&key, 0, sizeof(key));
1726         key.algo = algo;
1727         key.cam_id = k->wk_keyix;
1728         key.grpkey = (k->wk_flags & IEEE80211_KEY_GROUP) != 0;
1729         memcpy(key.key, k->wk_key, MIN(k->wk_keylen, sizeof(key.key)));
1730
1731         RSU_DPRINTF(sc, RSU_DEBUG_KEY | RSU_DEBUG_FWCMD,
1732             "%s: keyix %u, group %u, algo %u/%u, flags %04X, len %u, "
1733             "macaddr %s\n", __func__, key.cam_id, key.grpkey,
1734             k->wk_cipher->ic_cipher, key.algo, k->wk_flags, k->wk_keylen,
1735             ether_sprintf(k->wk_macaddr));
1736
1737         error = rsu_fw_cmd(sc, R92S_CMD_SET_KEY, &key, sizeof(key));
1738         if (error != 0) {
1739                 device_printf(sc->sc_dev,
1740                     "%s: cannot send firmware command, error %d\n",
1741                     __func__, error);
1742                 return (error);
1743         }
1744
1745         return (rsu_key_check(sc, k->wk_keyix, 1));
1746 }
1747
1748 static int
1749 rsu_set_key_pair(struct rsu_softc *sc, const struct ieee80211_key *k)
1750 {
1751         struct r92s_fw_cmd_set_key_mac key;
1752         uint8_t algo;
1753         int error;
1754
1755         RSU_ASSERT_LOCKED(sc);
1756
1757         if (!sc->sc_running)
1758                 return (ESHUTDOWN);
1759
1760         /* Map net80211 cipher to HW crypto algorithm. */
1761         algo = rsu_crypto_mode(sc, k->wk_cipher->ic_cipher, k->wk_keylen);
1762         if (algo == R92S_KEY_ALGO_INVALID)
1763                 return (EINVAL);
1764
1765         memset(&key, 0, sizeof(key));
1766         key.algo = algo;
1767         memcpy(key.macaddr, k->wk_macaddr, sizeof(key.macaddr));
1768         memcpy(key.key, k->wk_key, MIN(k->wk_keylen, sizeof(key.key)));
1769
1770         RSU_DPRINTF(sc, RSU_DEBUG_KEY | RSU_DEBUG_FWCMD,
1771             "%s: keyix %u, algo %u/%u, flags %04X, len %u, macaddr %s\n",
1772             __func__, k->wk_keyix, k->wk_cipher->ic_cipher, key.algo,
1773             k->wk_flags, k->wk_keylen, ether_sprintf(key.macaddr));
1774
1775         error = rsu_fw_cmd(sc, R92S_CMD_SET_STA_KEY, &key, sizeof(key));
1776         if (error != 0) {
1777                 device_printf(sc->sc_dev,
1778                     "%s: cannot send firmware command, error %d\n",
1779                     __func__, error);
1780                 return (error);
1781         }
1782
1783         return (rsu_key_check(sc, k->wk_keyix, 1));
1784 }
1785
1786 static int
1787 rsu_reinit_static_keys(struct rsu_softc *sc)
1788 {
1789         int i, error;
1790
1791         for (i = 0; i < nitems(sc->group_keys); i++) {
1792                 if (sc->group_keys[i] != NULL) {
1793                         error = rsu_set_key_group(sc, sc->group_keys[i]);
1794                         if (error != 0) {
1795                                 device_printf(sc->sc_dev,
1796                                     "%s: failed to set static key %d, "
1797                                     "error %d\n", __func__, i, error);
1798                                 return (error);
1799                         }
1800                 }
1801         }
1802
1803         return (0);
1804 }
1805
1806 static int
1807 rsu_delete_key(struct rsu_softc *sc, ieee80211_keyix keyix)
1808 {
1809         struct r92s_fw_cmd_set_key key;
1810         uint32_t val;
1811         int error;
1812
1813         RSU_ASSERT_LOCKED(sc);
1814
1815         if (!sc->sc_running)
1816                 return (0);
1817
1818         /* check if it was automatically removed by firmware */
1819         error = rsu_cam_read(sc, R92S_CAM_CTL0(keyix), &val);
1820         if (error == 0 && (val & R92S_CAM_VALID) == 0) {
1821                 RSU_DPRINTF(sc, RSU_DEBUG_KEY,
1822                     "%s: key %u does not exist\n", __func__, keyix);
1823                 clrbit(sc->keys_bmap, keyix);
1824                 return (0);
1825         }
1826
1827         memset(&key, 0, sizeof(key));
1828         key.cam_id = keyix;
1829
1830         RSU_DPRINTF(sc, RSU_DEBUG_KEY | RSU_DEBUG_FWCMD,
1831             "%s: removing key %u\n", __func__, key.cam_id);
1832
1833         error = rsu_fw_cmd(sc, R92S_CMD_SET_KEY, &key, sizeof(key));
1834         if (error != 0) {
1835                 device_printf(sc->sc_dev,
1836                     "%s: cannot send firmware command, error %d\n",
1837                     __func__, error);
1838                 goto finish;
1839         }
1840
1841         usb_pause_mtx(&sc->sc_mtx, USB_MS_TO_TICKS(5));
1842
1843         /*
1844          * Clear 'valid' bit manually (cannot be done via firmware command).
1845          * Used for key check + when firmware command cannot be sent.
1846          */
1847 finish:
1848         rsu_cam_write(sc, R92S_CAM_CTL0(keyix), 0);
1849
1850         clrbit(sc->keys_bmap, keyix);
1851
1852         return (rsu_key_check(sc, keyix, 0));
1853 }
1854
1855 static void
1856 rsu_delete_key_pair_cb(void *arg, int pending __unused)
1857 {
1858         struct rsu_softc *sc = arg;
1859         int i;
1860
1861         RSU_DELKEY_BMAP_LOCK(sc);
1862         for (i = IEEE80211_WEP_NKID; i < R92S_CAM_ENTRY_LIMIT; i++) {
1863                 if (isset(sc->free_keys_bmap, i)) {
1864                         RSU_DELKEY_BMAP_UNLOCK(sc);
1865
1866                         RSU_LOCK(sc);
1867                         RSU_DPRINTF(sc, RSU_DEBUG_KEY,
1868                             "%s: calling rsu_delete_key() with keyix = %d\n",
1869                             __func__, i);
1870                         (void) rsu_delete_key(sc, i);
1871                         RSU_UNLOCK(sc);
1872
1873                         RSU_DELKEY_BMAP_LOCK(sc);
1874                         clrbit(sc->free_keys_bmap, i);
1875
1876                         /* bmap can be changed */
1877                         i = IEEE80211_WEP_NKID - 1;
1878                         continue;
1879                 }
1880         }
1881         RSU_DELKEY_BMAP_UNLOCK(sc);
1882 }
1883
1884 static int
1885 rsu_site_survey(struct rsu_softc *sc, struct ieee80211_scan_ssid *ssid)
1886 {
1887         struct r92s_fw_cmd_sitesurvey cmd;
1888
1889         RSU_ASSERT_LOCKED(sc);
1890
1891         memset(&cmd, 0, sizeof(cmd));
1892         /* TODO: passive channels? */
1893         if (sc->sc_active_scan)
1894                 cmd.active = htole32(1);
1895         cmd.limit = htole32(48);
1896
1897         if (ssid != NULL) {
1898                 sc->sc_extra_scan = 1;
1899                 cmd.ssidlen = htole32(ssid->len);
1900                 memcpy(cmd.ssid, ssid->ssid, ssid->len);
1901         }
1902 #ifdef USB_DEBUG
1903         if (rsu_debug & (RSU_DEBUG_SCAN | RSU_DEBUG_FWCMD)) {
1904                 device_printf(sc->sc_dev,
1905                     "sending site survey command, active %d",
1906                     le32toh(cmd.active));
1907                 if (ssid != NULL) {
1908                         printf(", ssid: ");
1909                         ieee80211_print_essid(cmd.ssid, le32toh(cmd.ssidlen));
1910                 }
1911                 printf("\n");
1912         }
1913 #endif
1914         return (rsu_fw_cmd(sc, R92S_CMD_SITE_SURVEY, &cmd, sizeof(cmd)));
1915 }
1916
1917 static int
1918 rsu_join_bss(struct rsu_softc *sc, struct ieee80211_node *ni)
1919 {
1920         struct ieee80211com *ic = &sc->sc_ic;
1921         struct ieee80211vap *vap = ni->ni_vap;
1922         struct ndis_wlan_bssid_ex *bss;
1923         struct ndis_802_11_fixed_ies *fixed;
1924         struct r92s_fw_cmd_auth auth;
1925         uint8_t buf[sizeof(*bss) + 128] __aligned(4);
1926         uint8_t *frm;
1927         uint8_t opmode;
1928         int error;
1929
1930         RSU_ASSERT_LOCKED(sc);
1931
1932         /* Let the FW decide the opmode based on the capinfo field. */
1933         opmode = NDIS802_11AUTOUNKNOWN;
1934         RSU_DPRINTF(sc, RSU_DEBUG_RESET,
1935             "%s: setting operating mode to %d\n",
1936             __func__, opmode);
1937         error = rsu_fw_cmd(sc, R92S_CMD_SET_OPMODE, &opmode, sizeof(opmode));
1938         if (error != 0)
1939                 return (error);
1940
1941         memset(&auth, 0, sizeof(auth));
1942         if (vap->iv_flags & IEEE80211_F_WPA) {
1943                 auth.mode = R92S_AUTHMODE_WPA;
1944                 auth.dot1x = (ni->ni_authmode == IEEE80211_AUTH_8021X);
1945         } else
1946                 auth.mode = R92S_AUTHMODE_OPEN;
1947         RSU_DPRINTF(sc, RSU_DEBUG_RESET,
1948             "%s: setting auth mode to %d\n",
1949             __func__, auth.mode);
1950         error = rsu_fw_cmd(sc, R92S_CMD_SET_AUTH, &auth, sizeof(auth));
1951         if (error != 0)
1952                 return (error);
1953
1954         memset(buf, 0, sizeof(buf));
1955         bss = (struct ndis_wlan_bssid_ex *)buf;
1956         IEEE80211_ADDR_COPY(bss->macaddr, ni->ni_bssid);
1957         bss->ssid.ssidlen = htole32(ni->ni_esslen);
1958         memcpy(bss->ssid.ssid, ni->ni_essid, ni->ni_esslen);
1959         if (vap->iv_flags & (IEEE80211_F_PRIVACY | IEEE80211_F_WPA))
1960                 bss->privacy = htole32(1);
1961         bss->rssi = htole32(ni->ni_avgrssi);
1962         if (ic->ic_curmode == IEEE80211_MODE_11B)
1963                 bss->networktype = htole32(NDIS802_11DS);
1964         else
1965                 bss->networktype = htole32(NDIS802_11OFDM24);
1966         bss->config.len = htole32(sizeof(bss->config));
1967         bss->config.bintval = htole32(ni->ni_intval);
1968         bss->config.dsconfig = htole32(ieee80211_chan2ieee(ic, ni->ni_chan));
1969         bss->inframode = htole32(NDIS802_11INFRASTRUCTURE);
1970         /* XXX verify how this is supposed to look! */
1971         memcpy(bss->supprates, ni->ni_rates.rs_rates,
1972             ni->ni_rates.rs_nrates);
1973         /* Write the fixed fields of the beacon frame. */
1974         fixed = (struct ndis_802_11_fixed_ies *)&bss[1];
1975         memcpy(&fixed->tstamp, ni->ni_tstamp.data, 8);
1976         fixed->bintval = htole16(ni->ni_intval);
1977         fixed->capabilities = htole16(ni->ni_capinfo);
1978         /* Write IEs to be included in the association request. */
1979         frm = (uint8_t *)&fixed[1];
1980         frm = ieee80211_add_rsn(frm, vap);
1981         frm = ieee80211_add_wpa(frm, vap);
1982         frm = ieee80211_add_qos(frm, ni);
1983         if ((ic->ic_flags & IEEE80211_F_WME) &&
1984             (ni->ni_ies.wme_ie != NULL))
1985                 frm = ieee80211_add_wme_info(frm, &ic->ic_wme);
1986         if (ni->ni_flags & IEEE80211_NODE_HT) {
1987                 frm = ieee80211_add_htcap(frm, ni);
1988                 frm = ieee80211_add_htinfo(frm, ni);
1989         }
1990         bss->ieslen = htole32(frm - (uint8_t *)fixed);
1991         bss->len = htole32(((frm - buf) + 3) & ~3);
1992         RSU_DPRINTF(sc, RSU_DEBUG_RESET | RSU_DEBUG_FWCMD,
1993             "%s: sending join bss command to %s chan %d\n",
1994             __func__,
1995             ether_sprintf(bss->macaddr), le32toh(bss->config.dsconfig));
1996         return (rsu_fw_cmd(sc, R92S_CMD_JOIN_BSS, buf, sizeof(buf)));
1997 }
1998
1999 static int
2000 rsu_disconnect(struct rsu_softc *sc)
2001 {
2002         uint32_t zero = 0;      /* :-) */
2003
2004         /* Disassociate from our current BSS. */
2005         RSU_DPRINTF(sc, RSU_DEBUG_STATE | RSU_DEBUG_FWCMD,
2006             "%s: sending disconnect command\n", __func__);
2007         return (rsu_fw_cmd(sc, R92S_CMD_DISCONNECT, &zero, sizeof(zero)));
2008 }
2009
2010 /*
2011  * Map the hardware provided RSSI value to a signal level.
2012  * For the most part it's just something we divide by and cap
2013  * so it doesn't overflow the representation by net80211.
2014  */
2015 static int
2016 rsu_hwrssi_to_rssi(struct rsu_softc *sc, int hw_rssi)
2017 {
2018         int v;
2019
2020         if (hw_rssi == 0)
2021                 return (0);
2022         v = hw_rssi >> 4;
2023         if (v > 80)
2024                 v = 80;
2025         return (v);
2026 }
2027
2028 CTASSERT(MCLBYTES > sizeof(struct ieee80211_frame));
2029
2030 static void
2031 rsu_event_survey(struct rsu_softc *sc, uint8_t *buf, int len)
2032 {
2033         struct ieee80211com *ic = &sc->sc_ic;
2034         struct ieee80211_frame *wh;
2035         struct ndis_wlan_bssid_ex *bss;
2036         struct ieee80211_rx_stats rxs;
2037         struct mbuf *m;
2038         uint32_t ieslen;
2039         uint32_t pktlen;
2040
2041         if (__predict_false(len < sizeof(*bss)))
2042                 return;
2043         bss = (struct ndis_wlan_bssid_ex *)buf;
2044         ieslen = le32toh(bss->ieslen);
2045         /* range check length of information element */
2046         if (__predict_false(ieslen > (uint32_t)(len - sizeof(*bss))))
2047                 return;
2048
2049         RSU_DPRINTF(sc, RSU_DEBUG_SCAN,
2050             "%s: found BSS %s: len=%d chan=%d inframode=%d "
2051             "networktype=%d privacy=%d, RSSI=%d\n",
2052             __func__,
2053             ether_sprintf(bss->macaddr), ieslen,
2054             le32toh(bss->config.dsconfig), le32toh(bss->inframode),
2055             le32toh(bss->networktype), le32toh(bss->privacy),
2056             le32toh(bss->rssi));
2057
2058         /* Build a fake beacon frame to let net80211 do all the parsing. */
2059         /* XXX TODO: just call the new scan API methods! */
2060         if (__predict_false(ieslen > (size_t)(MCLBYTES - sizeof(*wh))))
2061                 return;
2062         pktlen = sizeof(*wh) + ieslen;
2063         m = m_get2(pktlen, M_NOWAIT, MT_DATA, M_PKTHDR);
2064         if (__predict_false(m == NULL))
2065                 return;
2066         wh = mtod(m, struct ieee80211_frame *);
2067         wh->i_fc[0] = IEEE80211_FC0_VERSION_0 | IEEE80211_FC0_TYPE_MGT |
2068             IEEE80211_FC0_SUBTYPE_BEACON;
2069         wh->i_fc[1] = IEEE80211_FC1_DIR_NODS;
2070         USETW(wh->i_dur, 0);
2071         IEEE80211_ADDR_COPY(wh->i_addr1, ieee80211broadcastaddr);
2072         IEEE80211_ADDR_COPY(wh->i_addr2, bss->macaddr);
2073         IEEE80211_ADDR_COPY(wh->i_addr3, bss->macaddr);
2074         *(uint16_t *)wh->i_seq = 0;
2075         memcpy(&wh[1], (uint8_t *)&bss[1], ieslen);
2076
2077         /* Finalize mbuf. */
2078         m->m_pkthdr.len = m->m_len = pktlen;
2079
2080         /* Set channel flags for input path */
2081         bzero(&rxs, sizeof(rxs));
2082         rxs.r_flags |= IEEE80211_R_IEEE | IEEE80211_R_FREQ;
2083         rxs.r_flags |= IEEE80211_R_NF | IEEE80211_R_RSSI;
2084         rxs.c_ieee = le32toh(bss->config.dsconfig);
2085         rxs.c_freq = ieee80211_ieee2mhz(rxs.c_ieee, IEEE80211_CHAN_2GHZ);
2086         /* This is a number from 0..100; so let's just divide it down a bit */
2087         rxs.c_rssi = le32toh(bss->rssi) / 2;
2088         rxs.c_nf = -96;
2089         if (ieee80211_add_rx_params(m, &rxs) == 0)
2090                 return;
2091
2092         /* XXX avoid a LOR */
2093         RSU_UNLOCK(sc);
2094         ieee80211_input_mimo_all(ic, m);
2095         RSU_LOCK(sc);
2096 }
2097
2098 static void
2099 rsu_event_join_bss(struct rsu_softc *sc, uint8_t *buf, int len)
2100 {
2101         struct ieee80211com *ic = &sc->sc_ic;
2102         struct ieee80211vap *vap = TAILQ_FIRST(&ic->ic_vaps);
2103         struct ieee80211_node *ni = vap->iv_bss;
2104         struct r92s_event_join_bss *rsp;
2105         uint32_t tmp;
2106         int res;
2107
2108         if (__predict_false(len < sizeof(*rsp)))
2109                 return;
2110         rsp = (struct r92s_event_join_bss *)buf;
2111         res = (int)le32toh(rsp->join_res);
2112
2113         RSU_DPRINTF(sc, RSU_DEBUG_STATE | RSU_DEBUG_FWCMD,
2114             "%s: Rx join BSS event len=%d res=%d\n",
2115             __func__, len, res);
2116
2117         /*
2118          * XXX Don't do this; there's likely a better way to tell
2119          * the caller we failed.
2120          */
2121         if (res <= 0) {
2122                 RSU_UNLOCK(sc);
2123                 ieee80211_new_state(vap, IEEE80211_S_SCAN, -1);
2124                 RSU_LOCK(sc);
2125                 return;
2126         }
2127
2128         tmp = le32toh(rsp->associd);
2129         if (tmp >= vap->iv_max_aid) {
2130                 RSU_DPRINTF(sc, RSU_DEBUG_ANY, "Assoc ID overflow\n");
2131                 tmp = 1;
2132         }
2133         RSU_DPRINTF(sc, RSU_DEBUG_STATE | RSU_DEBUG_FWCMD,
2134             "%s: associated with %s associd=%d\n",
2135             __func__, ether_sprintf(rsp->bss.macaddr), tmp);
2136         /* XXX is this required? What's the top two bits for again? */
2137         ni->ni_associd = tmp | 0xc000;
2138
2139         /* Refresh Rx filter (was changed by firmware). */
2140         sc->sc_vap_is_running = 1;
2141         rsu_rxfilter_refresh(sc);
2142
2143         RSU_UNLOCK(sc);
2144         ieee80211_new_state(vap, IEEE80211_S_RUN,
2145             IEEE80211_FC0_SUBTYPE_ASSOC_RESP);
2146         RSU_LOCK(sc);
2147 }
2148
2149 static void
2150 rsu_event_addba_req_report(struct rsu_softc *sc, uint8_t *buf, int len)
2151 {
2152         struct ieee80211com *ic = &sc->sc_ic;
2153         struct ieee80211vap *vap = TAILQ_FIRST(&ic->ic_vaps);
2154         struct r92s_add_ba_event *ba = (void *) buf;
2155         struct ieee80211_node *ni;
2156
2157         if (len < sizeof(*ba)) {
2158                 device_printf(sc->sc_dev, "%s: short read (%d)\n", __func__, len);
2159                 return;
2160         }
2161
2162         if (vap == NULL)
2163                 return;
2164
2165         RSU_DPRINTF(sc, RSU_DEBUG_AMPDU, "%s: mac=%s, tid=%d, ssn=%d\n",
2166             __func__,
2167             ether_sprintf(ba->mac_addr),
2168             (int) ba->tid,
2169             (int) le16toh(ba->ssn));
2170
2171         /* XXX do node lookup; this is STA specific */
2172
2173         ni = ieee80211_ref_node(vap->iv_bss);
2174         ieee80211_ampdu_rx_start_ext(ni, ba->tid, le16toh(ba->ssn) >> 4, 32);
2175         ieee80211_free_node(ni);
2176 }
2177
2178 static void
2179 rsu_rx_event(struct rsu_softc *sc, uint8_t code, uint8_t *buf, int len)
2180 {
2181         struct ieee80211com *ic = &sc->sc_ic;
2182         struct ieee80211vap *vap = TAILQ_FIRST(&ic->ic_vaps);
2183
2184         RSU_DPRINTF(sc, RSU_DEBUG_RX | RSU_DEBUG_FWCMD,
2185             "%s: Rx event code=%d len=%d\n", __func__, code, len);
2186         switch (code) {
2187         case R92S_EVT_SURVEY:
2188                 rsu_event_survey(sc, buf, len);
2189                 break;
2190         case R92S_EVT_SURVEY_DONE:
2191                 RSU_DPRINTF(sc, RSU_DEBUG_SCAN,
2192                     "%s: %s scan done, found %d BSS\n",
2193                     __func__, sc->sc_extra_scan ? "direct" : "broadcast",
2194                     le32toh(*(uint32_t *)buf));
2195                 if (sc->sc_extra_scan == 1) {
2196                         /* Send broadcast probe request. */
2197                         sc->sc_extra_scan = 0;
2198                         if (vap != NULL && rsu_site_survey(sc, NULL) != 0) {
2199                                 RSU_UNLOCK(sc);
2200                                 ieee80211_cancel_scan(vap);
2201                                 RSU_LOCK(sc);
2202                         }
2203                         break;
2204                 }
2205                 if (vap != NULL) {
2206                         RSU_UNLOCK(sc);
2207                         ieee80211_scan_done(vap);
2208                         RSU_LOCK(sc);
2209                 }
2210                 break;
2211         case R92S_EVT_JOIN_BSS:
2212                 if (vap->iv_state == IEEE80211_S_AUTH)
2213                         rsu_event_join_bss(sc, buf, len);
2214                 break;
2215         case R92S_EVT_DEL_STA:
2216                 RSU_DPRINTF(sc, RSU_DEBUG_FWCMD | RSU_DEBUG_STATE,
2217                     "%s: disassociated from %s\n", __func__,
2218                     ether_sprintf(buf));
2219                 if (vap->iv_state == IEEE80211_S_RUN &&
2220                     IEEE80211_ADDR_EQ(vap->iv_bss->ni_bssid, buf)) {
2221                         RSU_UNLOCK(sc);
2222                         ieee80211_new_state(vap, IEEE80211_S_SCAN, -1);
2223                         RSU_LOCK(sc);
2224                 }
2225                 break;
2226         case R92S_EVT_WPS_PBC:
2227                 RSU_DPRINTF(sc, RSU_DEBUG_RX | RSU_DEBUG_FWCMD,
2228                     "%s: WPS PBC pushed.\n", __func__);
2229                 break;
2230         case R92S_EVT_FWDBG:
2231                 buf[60] = '\0';
2232                 RSU_DPRINTF(sc, RSU_DEBUG_FWDBG, "FWDBG: %s\n", (char *)buf);
2233                 break;
2234         case R92S_EVT_ADDBA_REQ_REPORT:
2235                 rsu_event_addba_req_report(sc, buf, len);
2236                 break;
2237         default:
2238                 device_printf(sc->sc_dev, "%s: unhandled code (%d)\n", __func__, code);
2239                 break;
2240         }
2241 }
2242
2243 static void
2244 rsu_rx_multi_event(struct rsu_softc *sc, uint8_t *buf, int len)
2245 {
2246         struct r92s_fw_cmd_hdr *cmd;
2247         int cmdsz;
2248
2249         RSU_DPRINTF(sc, RSU_DEBUG_RX, "%s: Rx events len=%d\n", __func__, len);
2250
2251         /* Skip Rx status. */
2252         buf += sizeof(struct r92s_rx_stat);
2253         len -= sizeof(struct r92s_rx_stat);
2254
2255         /* Process all events. */
2256         for (;;) {
2257                 /* Check that command header fits. */
2258                 if (__predict_false(len < sizeof(*cmd)))
2259                         break;
2260                 cmd = (struct r92s_fw_cmd_hdr *)buf;
2261                 /* Check that command payload fits. */
2262                 cmdsz = le16toh(cmd->len);
2263                 if (__predict_false(len < sizeof(*cmd) + cmdsz))
2264                         break;
2265
2266                 /* Process firmware event. */
2267                 rsu_rx_event(sc, cmd->code, (uint8_t *)&cmd[1], cmdsz);
2268
2269                 if (!(cmd->seq & R92S_FW_CMD_MORE))
2270                         break;
2271                 buf += sizeof(*cmd) + cmdsz;
2272                 len -= sizeof(*cmd) + cmdsz;
2273         }
2274 }
2275
2276 static int8_t
2277 rsu_get_rssi(struct rsu_softc *sc, int rate, void *physt)
2278 {
2279         static const int8_t cckoff[] = { 14, -2, -20, -40 };
2280         struct r92s_rx_phystat *phy;
2281         struct r92s_rx_cck *cck;
2282         uint8_t rpt;
2283         int8_t rssi;
2284
2285         if (rate <= 3) {
2286                 cck = (struct r92s_rx_cck *)physt;
2287                 rpt = (cck->agc_rpt >> 6) & 0x3;
2288                 rssi = cck->agc_rpt & 0x3e;
2289                 rssi = cckoff[rpt] - rssi;
2290         } else {        /* OFDM/HT. */
2291                 phy = (struct r92s_rx_phystat *)physt;
2292                 rssi = ((le32toh(phy->phydw1) >> 1) & 0x7f) - 106;
2293         }
2294         return (rssi);
2295 }
2296
2297 static struct mbuf *
2298 rsu_rx_copy_to_mbuf(struct rsu_softc *sc, struct r92s_rx_stat *stat,
2299     int totlen)
2300 {
2301         struct ieee80211com *ic = &sc->sc_ic;
2302         struct mbuf *m;
2303         uint32_t rxdw0;
2304         int pktlen;
2305
2306         rxdw0 = le32toh(stat->rxdw0);
2307         if (__predict_false(rxdw0 & (R92S_RXDW0_CRCERR | R92S_RXDW0_ICVERR))) {
2308                 RSU_DPRINTF(sc, RSU_DEBUG_RX,
2309                     "%s: RX flags error (%s)\n", __func__,
2310                     rxdw0 & R92S_RXDW0_CRCERR ? "CRC" : "ICV");
2311                 goto fail;
2312         }
2313
2314         pktlen = MS(rxdw0, R92S_RXDW0_PKTLEN);
2315         if (__predict_false(pktlen < sizeof (struct ieee80211_frame_ack))) {
2316                 RSU_DPRINTF(sc, RSU_DEBUG_RX,
2317                     "%s: frame is too short: %d\n", __func__, pktlen);
2318                 goto fail;
2319         }
2320
2321         m = m_get2(totlen, M_NOWAIT, MT_DATA, M_PKTHDR);
2322         if (__predict_false(m == NULL)) {
2323                 device_printf(sc->sc_dev,
2324                     "%s: could not allocate RX mbuf, totlen %d\n",
2325                     __func__, totlen);
2326                 goto fail;
2327         }
2328
2329         /* Finalize mbuf. */
2330         memcpy(mtod(m, uint8_t *), (uint8_t *)stat, totlen);
2331         m->m_pkthdr.len = m->m_len = totlen;
2332
2333         return (m);
2334 fail:
2335         counter_u64_add(ic->ic_ierrors, 1);
2336         return (NULL);
2337 }
2338
2339 static uint32_t
2340 rsu_get_tsf_low(struct rsu_softc *sc)
2341 {
2342         return (rsu_read_4(sc, R92S_TSFTR));
2343 }
2344
2345 static uint32_t
2346 rsu_get_tsf_high(struct rsu_softc *sc)
2347 {
2348         return (rsu_read_4(sc, R92S_TSFTR + 4));
2349 }
2350
2351 static struct ieee80211_node *
2352 rsu_rx_frame(struct rsu_softc *sc, struct mbuf *m)
2353 {
2354         struct ieee80211com *ic = &sc->sc_ic;
2355         struct ieee80211_frame_min *wh;
2356         struct ieee80211_rx_stats rxs;
2357         struct r92s_rx_stat *stat;
2358         uint32_t rxdw0, rxdw3;
2359         uint8_t cipher, rate;
2360         int infosz;
2361         int rssi;
2362
2363         stat = mtod(m, struct r92s_rx_stat *);
2364         rxdw0 = le32toh(stat->rxdw0);
2365         rxdw3 = le32toh(stat->rxdw3);
2366
2367         rate = MS(rxdw3, R92S_RXDW3_RATE);
2368         cipher = MS(rxdw0, R92S_RXDW0_CIPHER);
2369         infosz = MS(rxdw0, R92S_RXDW0_INFOSZ) * 8;
2370
2371         /* Get RSSI from PHY status descriptor if present. */
2372         if (infosz != 0 && (rxdw0 & R92S_RXDW0_PHYST))
2373                 rssi = rsu_get_rssi(sc, rate, &stat[1]);
2374         else {
2375                 /* Cheat and get the last calibrated RSSI */
2376                 rssi = rsu_hwrssi_to_rssi(sc, sc->sc_currssi);
2377         }
2378
2379         /* Hardware does Rx TCP checksum offload. */
2380         /*
2381          * This flag can be set for some other
2382          * (e.g., EAPOL) frame types, so don't rely on it.
2383          */
2384         if (rxdw3 & R92S_RXDW3_TCPCHKVALID) {
2385                 RSU_DPRINTF(sc, RSU_DEBUG_RX,
2386                     "%s: TCP/IP checksums: %schecked / %schecked\n",
2387                     __func__,
2388                     (rxdw3 & R92S_RXDW3_TCPCHKRPT) ? "" : "not ",
2389                     (rxdw3 & R92S_RXDW3_IPCHKRPT) ? "" : "not ");
2390
2391                 /*
2392                  * 'IP header checksum valid' bit will not be set if
2393                  * the frame was not checked / has incorrect checksum /
2394                  * does not have checksum (IPv6).
2395                  *
2396                  * NB: if DF bit is not set then frame will not be checked.
2397                  */
2398                 if (rxdw3 & R92S_RXDW3_IPCHKRPT) {
2399                         m->m_pkthdr.csum_flags = CSUM_IP_CHECKED;
2400                         m->m_pkthdr.csum_flags |= CSUM_IP_VALID;
2401                 }
2402
2403                 /*
2404                  * This is independent of the above check.
2405                  */
2406                 if (rxdw3 & R92S_RXDW3_TCPCHKRPT) {
2407                         m->m_pkthdr.csum_flags |= CSUM_DATA_VALID;
2408                         m->m_pkthdr.csum_flags |= CSUM_PSEUDO_HDR;
2409                         m->m_pkthdr.csum_data = 0xffff;
2410                 }
2411         }
2412
2413         /* RX flags */
2414
2415         /* Set channel flags for input path */
2416         bzero(&rxs, sizeof(rxs));
2417
2418         /* normal RSSI */
2419         rxs.r_flags |= IEEE80211_R_NF | IEEE80211_R_RSSI;
2420         rxs.c_rssi = rssi;
2421         rxs.c_nf = -96;
2422
2423         /* Rate */
2424         if (rate < 12) {
2425                 rxs.c_rate = ridx2rate[rate];
2426                 if (RSU_RATE_IS_CCK(rate))
2427                         rxs.c_pktflags |= IEEE80211_RX_F_CCK;
2428                 else
2429                         rxs.c_pktflags |= IEEE80211_RX_F_OFDM;
2430         } else {
2431                 rxs.c_rate = IEEE80211_RATE_MCS | (rate - 12);
2432                 rxs.c_pktflags |= IEEE80211_RX_F_HT;
2433         }
2434
2435         if (ieee80211_radiotap_active(ic)) {
2436                 struct rsu_rx_radiotap_header *tap = &sc->sc_rxtap;
2437
2438                 /* Map HW rate index to 802.11 rate. */
2439                 tap->wr_flags = 0;              /* TODO */
2440                 tap->wr_tsft = rsu_get_tsf_high(sc);
2441                 if (le32toh(stat->tsf_low) > rsu_get_tsf_low(sc))
2442                         tap->wr_tsft--;
2443                 tap->wr_tsft = (uint64_t)htole32(tap->wr_tsft) << 32;
2444                 tap->wr_tsft += stat->tsf_low;
2445
2446                 tap->wr_rate = rxs.c_rate;
2447                 tap->wr_dbm_antsignal = rssi;
2448         };
2449
2450         (void) ieee80211_add_rx_params(m, &rxs);
2451
2452         /* Drop descriptor. */
2453         m_adj(m, sizeof(*stat) + infosz);
2454         wh = mtod(m, struct ieee80211_frame_min *);
2455         if ((wh->i_fc[1] & IEEE80211_FC1_PROTECTED) &&
2456             cipher != R92S_KEY_ALGO_NONE) {
2457                 m->m_flags |= M_WEP;
2458         }
2459
2460         RSU_DPRINTF(sc, RSU_DEBUG_RX,
2461             "%s: Rx frame len %d, rate %d, infosz %d\n",
2462             __func__, m->m_len, rate, infosz);
2463
2464         if (m->m_len >= sizeof(*wh))
2465                 return (ieee80211_find_rxnode(ic, wh));
2466
2467         return (NULL);
2468 }
2469
2470 static struct mbuf *
2471 rsu_rx_multi_frame(struct rsu_softc *sc, uint8_t *buf, int len)
2472 {
2473         struct r92s_rx_stat *stat;
2474         uint32_t rxdw0;
2475         int totlen, pktlen, infosz, npkts;
2476         struct mbuf *m, *m0 = NULL, *prevm = NULL;
2477
2478         /*
2479          * don't pass packets to the ieee80211 framework if the driver isn't
2480          * RUNNING.
2481          */
2482         if (!sc->sc_running)
2483                 return (NULL);
2484
2485         /* Get the number of encapsulated frames. */
2486         stat = (struct r92s_rx_stat *)buf;
2487         npkts = MS(le32toh(stat->rxdw2), R92S_RXDW2_PKTCNT);
2488         RSU_DPRINTF(sc, RSU_DEBUG_RX,
2489             "%s: Rx %d frames in one chunk\n", __func__, npkts);
2490
2491         /* Process all of them. */
2492         while (npkts-- > 0) {
2493                 if (__predict_false(len < sizeof(*stat)))
2494                         break;
2495                 stat = (struct r92s_rx_stat *)buf;
2496                 rxdw0 = le32toh(stat->rxdw0);
2497
2498                 pktlen = MS(rxdw0, R92S_RXDW0_PKTLEN);
2499                 if (__predict_false(pktlen == 0))
2500                         break;
2501
2502                 infosz = MS(rxdw0, R92S_RXDW0_INFOSZ) * 8;
2503
2504                 /* Make sure everything fits in xfer. */
2505                 totlen = sizeof(*stat) + infosz + pktlen;
2506                 if (__predict_false(totlen > len))
2507                         break;
2508
2509                 /* Process 802.11 frame. */
2510                 m = rsu_rx_copy_to_mbuf(sc, stat, totlen);
2511                 if (m0 == NULL)
2512                         m0 = m;
2513                 if (prevm == NULL)
2514                         prevm = m;
2515                 else {
2516                         prevm->m_next = m;
2517                         prevm = m;
2518                 }
2519                 /* Next chunk is 128-byte aligned. */
2520                 totlen = (totlen + 127) & ~127;
2521                 buf += totlen;
2522                 len -= totlen;
2523         }
2524
2525         return (m0);
2526 }
2527
2528 static struct mbuf *
2529 rsu_rxeof(struct usb_xfer *xfer, struct rsu_data *data)
2530 {
2531         struct rsu_softc *sc = data->sc;
2532         struct ieee80211com *ic = &sc->sc_ic;
2533         struct r92s_rx_stat *stat;
2534         int len;
2535
2536         usbd_xfer_status(xfer, &len, NULL, NULL, NULL);
2537
2538         if (__predict_false(len < sizeof(*stat))) {
2539                 RSU_DPRINTF(sc, RSU_DEBUG_RX, "xfer too short %d\n", len);
2540                 counter_u64_add(ic->ic_ierrors, 1);
2541                 return (NULL);
2542         }
2543         /* Determine if it is a firmware C2H event or an 802.11 frame. */
2544         stat = (struct r92s_rx_stat *)data->buf;
2545         if ((le32toh(stat->rxdw1) & 0x1ff) == 0x1ff) {
2546                 rsu_rx_multi_event(sc, data->buf, len);
2547                 /* No packets to process. */
2548                 return (NULL);
2549         } else
2550                 return (rsu_rx_multi_frame(sc, data->buf, len));
2551 }
2552
2553 static void
2554 rsu_bulk_rx_callback(struct usb_xfer *xfer, usb_error_t error)
2555 {
2556         struct epoch_tracker et;
2557         struct rsu_softc *sc = usbd_xfer_softc(xfer);
2558         struct ieee80211com *ic = &sc->sc_ic;
2559         struct ieee80211_node *ni;
2560         struct mbuf *m = NULL, *next;
2561         struct rsu_data *data;
2562
2563         RSU_ASSERT_LOCKED(sc);
2564
2565         switch (USB_GET_STATE(xfer)) {
2566         case USB_ST_TRANSFERRED:
2567                 data = STAILQ_FIRST(&sc->sc_rx_active);
2568                 if (data == NULL)
2569                         goto tr_setup;
2570                 STAILQ_REMOVE_HEAD(&sc->sc_rx_active, next);
2571                 m = rsu_rxeof(xfer, data);
2572                 STAILQ_INSERT_TAIL(&sc->sc_rx_inactive, data, next);
2573                 /* FALLTHROUGH */
2574         case USB_ST_SETUP:
2575 tr_setup:
2576                 data = STAILQ_FIRST(&sc->sc_rx_inactive);
2577                 if (data == NULL) {
2578                         KASSERT(m == NULL, ("mbuf isn't NULL"));
2579                         return;
2580                 }
2581                 STAILQ_REMOVE_HEAD(&sc->sc_rx_inactive, next);
2582                 STAILQ_INSERT_TAIL(&sc->sc_rx_active, data, next);
2583                 usbd_xfer_set_frame_data(xfer, 0, data->buf,
2584                     usbd_xfer_max_len(xfer));
2585                 usbd_transfer_submit(xfer);
2586                 /*
2587                  * To avoid LOR we should unlock our private mutex here to call
2588                  * ieee80211_input() because here is at the end of a USB
2589                  * callback and safe to unlock.
2590                  */
2591                 NET_EPOCH_ENTER(et);
2592                 while (m != NULL) {
2593                         next = m->m_next;
2594                         m->m_next = NULL;
2595
2596                         ni = rsu_rx_frame(sc, m);
2597                         RSU_UNLOCK(sc);
2598
2599                         if (ni != NULL) {
2600                                 if (ni->ni_flags & IEEE80211_NODE_HT)
2601                                         m->m_flags |= M_AMPDU;
2602                                 (void)ieee80211_input_mimo(ni, m);
2603                                 ieee80211_free_node(ni);
2604                         } else
2605                                 (void)ieee80211_input_mimo_all(ic, m);
2606
2607                         RSU_LOCK(sc);
2608                         m = next;
2609                 }
2610                 NET_EPOCH_EXIT(et);
2611                 break;
2612         default:
2613                 /* needs it to the inactive queue due to a error. */
2614                 data = STAILQ_FIRST(&sc->sc_rx_active);
2615                 if (data != NULL) {
2616                         STAILQ_REMOVE_HEAD(&sc->sc_rx_active, next);
2617                         STAILQ_INSERT_TAIL(&sc->sc_rx_inactive, data, next);
2618                 }
2619                 if (error != USB_ERR_CANCELLED) {
2620                         usbd_xfer_set_stall(xfer);
2621                         counter_u64_add(ic->ic_ierrors, 1);
2622                         goto tr_setup;
2623                 }
2624                 break;
2625         }
2626
2627 }
2628
2629 static void
2630 rsu_txeof(struct usb_xfer *xfer, struct rsu_data *data)
2631 {
2632 #ifdef  USB_DEBUG
2633         struct rsu_softc *sc = usbd_xfer_softc(xfer);
2634 #endif
2635
2636         RSU_DPRINTF(sc, RSU_DEBUG_TXDONE, "%s: called; data=%p\n",
2637             __func__,
2638             data);
2639
2640         if (data->m) {
2641                 /* XXX status? */
2642                 ieee80211_tx_complete(data->ni, data->m, 0);
2643                 data->m = NULL;
2644                 data->ni = NULL;
2645         }
2646 }
2647
2648 static void
2649 rsu_bulk_tx_callback_sub(struct usb_xfer *xfer, usb_error_t error,
2650     uint8_t which)
2651 {
2652         struct rsu_softc *sc = usbd_xfer_softc(xfer);
2653         struct ieee80211com *ic = &sc->sc_ic;
2654         struct rsu_data *data;
2655
2656         RSU_ASSERT_LOCKED(sc);
2657
2658         switch (USB_GET_STATE(xfer)) {
2659         case USB_ST_TRANSFERRED:
2660                 data = STAILQ_FIRST(&sc->sc_tx_active[which]);
2661                 if (data == NULL)
2662                         goto tr_setup;
2663                 RSU_DPRINTF(sc, RSU_DEBUG_TXDONE, "%s: transfer done %p\n",
2664                     __func__, data);
2665                 STAILQ_REMOVE_HEAD(&sc->sc_tx_active[which], next);
2666                 rsu_txeof(xfer, data);
2667                 rsu_freebuf(sc, data);
2668                 /* FALLTHROUGH */
2669         case USB_ST_SETUP:
2670 tr_setup:
2671                 data = STAILQ_FIRST(&sc->sc_tx_pending[which]);
2672                 if (data == NULL) {
2673                         RSU_DPRINTF(sc, RSU_DEBUG_TXDONE,
2674                             "%s: empty pending queue sc %p\n", __func__, sc);
2675                         return;
2676                 }
2677                 STAILQ_REMOVE_HEAD(&sc->sc_tx_pending[which], next);
2678                 STAILQ_INSERT_TAIL(&sc->sc_tx_active[which], data, next);
2679                 usbd_xfer_set_frame_data(xfer, 0, data->buf, data->buflen);
2680                 RSU_DPRINTF(sc, RSU_DEBUG_TXDONE,
2681                     "%s: submitting transfer %p\n",
2682                     __func__,
2683                     data);
2684                 usbd_transfer_submit(xfer);
2685                 break;
2686         default:
2687                 data = STAILQ_FIRST(&sc->sc_tx_active[which]);
2688                 if (data != NULL) {
2689                         STAILQ_REMOVE_HEAD(&sc->sc_tx_active[which], next);
2690                         rsu_txeof(xfer, data);
2691                         rsu_freebuf(sc, data);
2692                 }
2693                 counter_u64_add(ic->ic_oerrors, 1);
2694
2695                 if (error != USB_ERR_CANCELLED) {
2696                         usbd_xfer_set_stall(xfer);
2697                         goto tr_setup;
2698                 }
2699                 break;
2700         }
2701
2702         /*
2703          * XXX TODO: if the queue is low, flush out FF TX frames.
2704          * Remember to unlock the driver for now; net80211 doesn't
2705          * defer it for us.
2706          */
2707 }
2708
2709 static void
2710 rsu_bulk_tx_callback_be_bk(struct usb_xfer *xfer, usb_error_t error)
2711 {
2712         struct rsu_softc *sc = usbd_xfer_softc(xfer);
2713
2714         rsu_bulk_tx_callback_sub(xfer, error, RSU_BULK_TX_BE_BK);
2715
2716         /* This kicks the TX taskqueue */
2717         rsu_start(sc);
2718 }
2719
2720 static void
2721 rsu_bulk_tx_callback_vi_vo(struct usb_xfer *xfer, usb_error_t error)
2722 {
2723         struct rsu_softc *sc = usbd_xfer_softc(xfer);
2724
2725         rsu_bulk_tx_callback_sub(xfer, error, RSU_BULK_TX_VI_VO);
2726
2727         /* This kicks the TX taskqueue */
2728         rsu_start(sc);
2729 }
2730
2731 static void
2732 rsu_bulk_tx_callback_h2c(struct usb_xfer *xfer, usb_error_t error)
2733 {
2734         struct rsu_softc *sc = usbd_xfer_softc(xfer);
2735
2736         rsu_bulk_tx_callback_sub(xfer, error, RSU_BULK_TX_H2C);
2737
2738         /* This kicks the TX taskqueue */
2739         rsu_start(sc);
2740 }
2741
2742 /*
2743  * Transmit the given frame.
2744  *
2745  * This doesn't free the node or mbuf upon failure.
2746  */
2747 static int
2748 rsu_tx_start(struct rsu_softc *sc, struct ieee80211_node *ni,
2749     struct mbuf *m0, struct rsu_data *data)
2750 {
2751         const struct ieee80211_txparam *tp = ni->ni_txparms;
2752         struct ieee80211vap *vap = ni->ni_vap;
2753         struct ieee80211_frame *wh;
2754         struct ieee80211_key *k = NULL;
2755         struct r92s_tx_desc *txd;
2756         uint8_t rate, ridx, type, cipher, qos;
2757         int prio = 0;
2758         uint8_t which;
2759         int hasqos;
2760         int ismcast;
2761         int xferlen;
2762         int qid;
2763
2764         RSU_ASSERT_LOCKED(sc);
2765
2766         wh = mtod(m0, struct ieee80211_frame *);
2767         type = wh->i_fc[0] & IEEE80211_FC0_TYPE_MASK;
2768         ismcast = IEEE80211_IS_MULTICAST(wh->i_addr1);
2769
2770         RSU_DPRINTF(sc, RSU_DEBUG_TX, "%s: data=%p, m=%p\n",
2771             __func__, data, m0);
2772
2773         /* Choose a TX rate index. */
2774         if (type == IEEE80211_FC0_TYPE_MGT ||
2775             type == IEEE80211_FC0_TYPE_CTL ||
2776             (m0->m_flags & M_EAPOL) != 0)
2777                 rate = tp->mgmtrate;
2778         else if (ismcast)
2779                 rate = tp->mcastrate;
2780         else if (tp->ucastrate != IEEE80211_FIXED_RATE_NONE)
2781                 rate = tp->ucastrate;
2782         else
2783                 rate = 0;
2784
2785         if (rate != 0)
2786                 ridx = rate2ridx(rate);
2787
2788         if (wh->i_fc[1] & IEEE80211_FC1_PROTECTED) {
2789                 k = ieee80211_crypto_encap(ni, m0);
2790                 if (k == NULL) {
2791                         device_printf(sc->sc_dev,
2792                             "ieee80211_crypto_encap returns NULL.\n");
2793                         /* XXX we don't expect the fragmented frames */
2794                         return (ENOBUFS);
2795                 }
2796                 wh = mtod(m0, struct ieee80211_frame *);
2797         }
2798         /* If we have QoS then use it */
2799         /* XXX TODO: mbuf WME/PRI versus TID? */
2800         if (IEEE80211_QOS_HAS_SEQ(wh)) {
2801                 /* Has QoS */
2802                 prio = M_WME_GETAC(m0);
2803                 which = rsu_wme_ac_xfer_map[prio];
2804                 hasqos = 1;
2805                 qos = ((const struct ieee80211_qosframe *)wh)->i_qos[0];
2806         } else {
2807                 /* Non-QoS TID */
2808                 /* XXX TODO: tid=0 for non-qos TID? */
2809                 which = rsu_wme_ac_xfer_map[WME_AC_BE];
2810                 hasqos = 0;
2811                 prio = 0;
2812                 qos = 0;
2813         }
2814
2815         qid = rsu_ac2qid[prio];
2816 #if 0
2817         switch (type) {
2818         case IEEE80211_FC0_TYPE_CTL:
2819         case IEEE80211_FC0_TYPE_MGT:
2820                 which = rsu_wme_ac_xfer_map[WME_AC_VO];
2821                 break;
2822         default:
2823                 which = rsu_wme_ac_xfer_map[M_WME_GETAC(m0)];
2824                 break;
2825         }
2826         hasqos = 0;
2827 #endif
2828
2829         RSU_DPRINTF(sc, RSU_DEBUG_TX, "%s: pri=%d, which=%d, hasqos=%d\n",
2830             __func__,
2831             prio,
2832             which,
2833             hasqos);
2834
2835         /* Fill Tx descriptor. */
2836         txd = (struct r92s_tx_desc *)data->buf;
2837         memset(txd, 0, sizeof(*txd));
2838
2839         txd->txdw0 |= htole32(
2840             SM(R92S_TXDW0_PKTLEN, m0->m_pkthdr.len) |
2841             SM(R92S_TXDW0_OFFSET, sizeof(*txd)) |
2842             R92S_TXDW0_OWN | R92S_TXDW0_FSG | R92S_TXDW0_LSG);
2843
2844         txd->txdw1 |= htole32(
2845             SM(R92S_TXDW1_MACID, R92S_MACID_BSS) | SM(R92S_TXDW1_QSEL, qid));
2846         if (!hasqos)
2847                 txd->txdw1 |= htole32(R92S_TXDW1_NONQOS);
2848         if (k != NULL && !(k->wk_flags & IEEE80211_KEY_SWENCRYPT)) {
2849                 switch (k->wk_cipher->ic_cipher) {
2850                 case IEEE80211_CIPHER_WEP:
2851                         cipher = R92S_TXDW1_CIPHER_WEP;
2852                         break;
2853                 case IEEE80211_CIPHER_TKIP:
2854                         cipher = R92S_TXDW1_CIPHER_TKIP;
2855                         break;
2856                 case IEEE80211_CIPHER_AES_CCM:
2857                         cipher = R92S_TXDW1_CIPHER_AES;
2858                         break;
2859                 default:
2860                         cipher = R92S_TXDW1_CIPHER_NONE;
2861                 }
2862                 txd->txdw1 |= htole32(
2863                     SM(R92S_TXDW1_CIPHER, cipher) |
2864                     SM(R92S_TXDW1_KEYIDX, k->wk_keyix));
2865         }
2866         /* XXX todo: set AGGEN bit if appropriate? */
2867         txd->txdw2 |= htole32(R92S_TXDW2_BK);
2868         if (ismcast)
2869                 txd->txdw2 |= htole32(R92S_TXDW2_BMCAST);
2870
2871         if (!ismcast && (!qos || (qos & IEEE80211_QOS_ACKPOLICY) !=
2872             IEEE80211_QOS_ACKPOLICY_NOACK)) {
2873                 txd->txdw2 |= htole32(R92S_TXDW2_RTY_LMT_ENA);
2874                 txd->txdw2 |= htole32(SM(R92S_TXDW2_RTY_LMT, tp->maxretry));
2875         }
2876
2877         /* Force mgmt / mcast / ucast rate if needed. */
2878         if (rate != 0) {
2879                 /* Data rate fallback limit (max). */
2880                 txd->txdw5 |= htole32(SM(R92S_TXDW5_DATARATE_FB_LMT, 0x1f));
2881                 txd->txdw5 |= htole32(SM(R92S_TXDW5_DATARATE, ridx));
2882                 txd->txdw4 |= htole32(R92S_TXDW4_DRVRATE);
2883         }
2884
2885         /*
2886          * Firmware will use and increment the sequence number for the
2887          * specified priority.
2888          */
2889         txd->txdw3 |= htole32(SM(R92S_TXDW3_SEQ, prio));
2890
2891         if (ieee80211_radiotap_active_vap(vap)) {
2892                 struct rsu_tx_radiotap_header *tap = &sc->sc_txtap;
2893
2894                 tap->wt_flags = 0;
2895                 ieee80211_radiotap_tx(vap, m0);
2896         }
2897
2898         xferlen = sizeof(*txd) + m0->m_pkthdr.len;
2899         m_copydata(m0, 0, m0->m_pkthdr.len, (caddr_t)&txd[1]);
2900
2901         data->buflen = xferlen;
2902         data->ni = ni;
2903         data->m = m0;
2904         STAILQ_INSERT_TAIL(&sc->sc_tx_pending[which], data, next);
2905
2906         /* start transfer, if any */
2907         usbd_transfer_start(sc->sc_xfer[which]);
2908         return (0);
2909 }
2910
2911 static int
2912 rsu_transmit(struct ieee80211com *ic, struct mbuf *m)
2913 {
2914         struct rsu_softc *sc = ic->ic_softc;
2915         int error;
2916
2917         RSU_LOCK(sc);
2918         if (!sc->sc_running) {
2919                 RSU_UNLOCK(sc);
2920                 return (ENXIO);
2921         }
2922
2923         /*
2924          * XXX TODO: ensure that we treat 'm' as a list of frames
2925          * to transmit!
2926          */
2927         error = mbufq_enqueue(&sc->sc_snd, m);
2928         if (error) {
2929                 RSU_DPRINTF(sc, RSU_DEBUG_TX,
2930                     "%s: mbufq_enable: failed (%d)\n",
2931                     __func__,
2932                     error);
2933                 RSU_UNLOCK(sc);
2934                 return (error);
2935         }
2936         RSU_UNLOCK(sc);
2937
2938         /* This kicks the TX taskqueue */
2939         rsu_start(sc);
2940
2941         return (0);
2942 }
2943
2944 static void
2945 rsu_drain_mbufq(struct rsu_softc *sc)
2946 {
2947         struct mbuf *m;
2948         struct ieee80211_node *ni;
2949
2950         RSU_ASSERT_LOCKED(sc);
2951         while ((m = mbufq_dequeue(&sc->sc_snd)) != NULL) {
2952                 ni = (struct ieee80211_node *)m->m_pkthdr.rcvif;
2953                 m->m_pkthdr.rcvif = NULL;
2954                 ieee80211_free_node(ni);
2955                 m_freem(m);
2956         }
2957 }
2958
2959 static void
2960 _rsu_start(struct rsu_softc *sc)
2961 {
2962         struct ieee80211_node *ni;
2963         struct rsu_data *bf;
2964         struct mbuf *m;
2965
2966         RSU_ASSERT_LOCKED(sc);
2967
2968         while ((m = mbufq_dequeue(&sc->sc_snd)) != NULL) {
2969                 bf = rsu_getbuf(sc);
2970                 if (bf == NULL) {
2971                         RSU_DPRINTF(sc, RSU_DEBUG_TX,
2972                             "%s: failed to get buffer\n", __func__);
2973                         mbufq_prepend(&sc->sc_snd, m);
2974                         break;
2975                 }
2976
2977                 ni = (struct ieee80211_node *)m->m_pkthdr.rcvif;
2978                 m->m_pkthdr.rcvif = NULL;
2979
2980                 if (rsu_tx_start(sc, ni, m, bf) != 0) {
2981                         RSU_DPRINTF(sc, RSU_DEBUG_TX,
2982                             "%s: failed to transmit\n", __func__);
2983                         if_inc_counter(ni->ni_vap->iv_ifp,
2984                             IFCOUNTER_OERRORS, 1);
2985                         rsu_freebuf(sc, bf);
2986                         ieee80211_free_node(ni);
2987                         m_freem(m);
2988                         break;
2989                 }
2990         }
2991 }
2992
2993 static void
2994 rsu_start(struct rsu_softc *sc)
2995 {
2996
2997         taskqueue_enqueue(taskqueue_thread, &sc->tx_task);
2998 }
2999
3000 static int
3001 rsu_ioctl_net(struct ieee80211com *ic, u_long cmd, void *data)
3002 {
3003         struct rsu_softc *sc = ic->ic_softc;
3004         struct ifreq *ifr = (struct ifreq *)data;
3005         int error;
3006
3007         error = 0;
3008         switch (cmd) {
3009         case SIOCSIFCAP:
3010         {
3011                 struct ieee80211vap *vap;
3012                 int rxmask;
3013
3014                 rxmask = ifr->ifr_reqcap & (IFCAP_RXCSUM | IFCAP_RXCSUM_IPV6);
3015
3016                 RSU_LOCK(sc);
3017                 /* Both RXCSUM bits must be set (or unset). */
3018                 if (sc->sc_rx_checksum_enable &&
3019                     rxmask != (IFCAP_RXCSUM | IFCAP_RXCSUM_IPV6)) {
3020                         rxmask = 0;
3021                         sc->sc_rx_checksum_enable = 0;
3022                         rsu_rxfilter_set(sc, R92S_RCR_TCP_OFFLD_EN, 0);
3023                 } else if (!sc->sc_rx_checksum_enable && rxmask != 0) {
3024                         rxmask = IFCAP_RXCSUM | IFCAP_RXCSUM_IPV6;
3025                         sc->sc_rx_checksum_enable = 1;
3026                         rsu_rxfilter_set(sc, 0, R92S_RCR_TCP_OFFLD_EN);
3027                 } else {
3028                         /* Nothing to do. */
3029                         RSU_UNLOCK(sc);
3030                         break;
3031                 }
3032                 RSU_UNLOCK(sc);
3033
3034                 IEEE80211_LOCK(ic);     /* XXX */
3035                 TAILQ_FOREACH(vap, &ic->ic_vaps, iv_next) {
3036                         struct ifnet *ifp = vap->iv_ifp;
3037
3038                         ifp->if_capenable &=
3039                             ~(IFCAP_RXCSUM | IFCAP_RXCSUM_IPV6);
3040                         ifp->if_capenable |= rxmask;
3041                 }
3042                 IEEE80211_UNLOCK(ic);
3043                 break;
3044         }
3045         default:
3046                 error = ENOTTY;         /* for net80211 */
3047                 break;
3048         }
3049
3050         return (error);
3051 }
3052
3053 static void
3054 rsu_parent(struct ieee80211com *ic)
3055 {
3056         struct rsu_softc *sc = ic->ic_softc;
3057
3058         if (ic->ic_nrunning > 0) {
3059                 if (rsu_init(sc) == 0)
3060                         ieee80211_start_all(ic);
3061                 else {
3062                         struct ieee80211vap *vap = TAILQ_FIRST(&ic->ic_vaps);
3063                         if (vap != NULL)
3064                                 ieee80211_stop(vap);
3065                 }
3066         } else
3067                 rsu_stop(sc);
3068 }
3069
3070 /*
3071  * Power on sequence for A-cut adapters.
3072  */
3073 static void
3074 rsu_power_on_acut(struct rsu_softc *sc)
3075 {
3076         uint32_t reg;
3077
3078         rsu_write_1(sc, R92S_SPS0_CTRL + 1, 0x53);
3079         rsu_write_1(sc, R92S_SPS0_CTRL + 0, 0x57);
3080
3081         /* Enable AFE macro block's bandgap and Mbias. */
3082         rsu_write_1(sc, R92S_AFE_MISC,
3083             rsu_read_1(sc, R92S_AFE_MISC) |
3084             R92S_AFE_MISC_BGEN | R92S_AFE_MISC_MBEN);
3085         /* Enable LDOA15 block. */
3086         rsu_write_1(sc, R92S_LDOA15_CTRL,
3087             rsu_read_1(sc, R92S_LDOA15_CTRL) | R92S_LDA15_EN);
3088
3089         rsu_write_1(sc, R92S_SPS1_CTRL,
3090             rsu_read_1(sc, R92S_SPS1_CTRL) | R92S_SPS1_LDEN);
3091         rsu_ms_delay(sc, 2000);
3092         /* Enable switch regulator block. */
3093         rsu_write_1(sc, R92S_SPS1_CTRL,
3094             rsu_read_1(sc, R92S_SPS1_CTRL) | R92S_SPS1_SWEN);
3095
3096         rsu_write_4(sc, R92S_SPS1_CTRL, 0x00a7b267);
3097
3098         rsu_write_1(sc, R92S_SYS_ISO_CTRL + 1,
3099             rsu_read_1(sc, R92S_SYS_ISO_CTRL + 1) | 0x08);
3100
3101         rsu_write_1(sc, R92S_SYS_FUNC_EN + 1,
3102             rsu_read_1(sc, R92S_SYS_FUNC_EN + 1) | 0x20);
3103
3104         rsu_write_1(sc, R92S_SYS_ISO_CTRL + 1,
3105             rsu_read_1(sc, R92S_SYS_ISO_CTRL + 1) & ~0x90);
3106
3107         /* Enable AFE clock. */
3108         rsu_write_1(sc, R92S_AFE_XTAL_CTRL + 1,
3109             rsu_read_1(sc, R92S_AFE_XTAL_CTRL + 1) & ~0x04);
3110         /* Enable AFE PLL macro block. */
3111         rsu_write_1(sc, R92S_AFE_PLL_CTRL,
3112             rsu_read_1(sc, R92S_AFE_PLL_CTRL) | 0x11);
3113         /* Attach AFE PLL to MACTOP/BB. */
3114         rsu_write_1(sc, R92S_SYS_ISO_CTRL,
3115             rsu_read_1(sc, R92S_SYS_ISO_CTRL) & ~0x11);
3116
3117         /* Switch to 40MHz clock instead of 80MHz. */
3118         rsu_write_2(sc, R92S_SYS_CLKR,
3119             rsu_read_2(sc, R92S_SYS_CLKR) & ~R92S_SYS_CLKSEL);
3120
3121         /* Enable MAC clock. */
3122         rsu_write_2(sc, R92S_SYS_CLKR,
3123             rsu_read_2(sc, R92S_SYS_CLKR) |
3124             R92S_MAC_CLK_EN | R92S_SYS_CLK_EN);
3125
3126         rsu_write_1(sc, R92S_PMC_FSM, 0x02);
3127
3128         /* Enable digital core and IOREG R/W. */
3129         rsu_write_1(sc, R92S_SYS_FUNC_EN + 1,
3130             rsu_read_1(sc, R92S_SYS_FUNC_EN + 1) | 0x08);
3131
3132         rsu_write_1(sc, R92S_SYS_FUNC_EN + 1,
3133             rsu_read_1(sc, R92S_SYS_FUNC_EN + 1) | 0x80);
3134
3135         /* Switch the control path to firmware. */
3136         reg = rsu_read_2(sc, R92S_SYS_CLKR);
3137         reg = (reg & ~R92S_SWHW_SEL) | R92S_FWHW_SEL;
3138         rsu_write_2(sc, R92S_SYS_CLKR, reg);
3139
3140         rsu_write_2(sc, R92S_CR, 0x37fc);
3141
3142         /* Fix USB RX FIFO issue. */
3143         rsu_write_1(sc, 0xfe5c,
3144             rsu_read_1(sc, 0xfe5c) | 0x80);
3145         rsu_write_1(sc, 0x00ab,
3146             rsu_read_1(sc, 0x00ab) | 0xc0);
3147
3148         rsu_write_1(sc, R92S_SYS_CLKR,
3149             rsu_read_1(sc, R92S_SYS_CLKR) & ~R92S_SYS_CPU_CLKSEL);
3150 }
3151
3152 /*
3153  * Power on sequence for B-cut and C-cut adapters.
3154  */
3155 static void
3156 rsu_power_on_bcut(struct rsu_softc *sc)
3157 {
3158         uint32_t reg;
3159         int ntries;
3160
3161         /* Prevent eFuse leakage. */
3162         rsu_write_1(sc, 0x37, 0xb0);
3163         rsu_ms_delay(sc, 10);
3164         rsu_write_1(sc, 0x37, 0x30);
3165
3166         /* Switch the control path to hardware. */
3167         reg = rsu_read_2(sc, R92S_SYS_CLKR);
3168         if (reg & R92S_FWHW_SEL) {
3169                 rsu_write_2(sc, R92S_SYS_CLKR,
3170                     reg & ~(R92S_SWHW_SEL | R92S_FWHW_SEL));
3171         }
3172         rsu_write_1(sc, R92S_SYS_FUNC_EN + 1,
3173             rsu_read_1(sc, R92S_SYS_FUNC_EN + 1) & ~0x8c);
3174         rsu_ms_delay(sc, 1);
3175
3176         rsu_write_1(sc, R92S_SPS0_CTRL + 1, 0x53);
3177         rsu_write_1(sc, R92S_SPS0_CTRL + 0, 0x57);
3178
3179         reg = rsu_read_1(sc, R92S_AFE_MISC);
3180         rsu_write_1(sc, R92S_AFE_MISC, reg | R92S_AFE_MISC_BGEN);
3181         rsu_write_1(sc, R92S_AFE_MISC, reg | R92S_AFE_MISC_BGEN |
3182             R92S_AFE_MISC_MBEN | R92S_AFE_MISC_I32_EN);
3183
3184         /* Enable PLL. */
3185         rsu_write_1(sc, R92S_LDOA15_CTRL,
3186             rsu_read_1(sc, R92S_LDOA15_CTRL) | R92S_LDA15_EN);
3187
3188         rsu_write_1(sc, R92S_LDOV12D_CTRL,
3189             rsu_read_1(sc, R92S_LDOV12D_CTRL) | R92S_LDV12_EN);
3190
3191         rsu_write_1(sc, R92S_SYS_ISO_CTRL + 1,
3192             rsu_read_1(sc, R92S_SYS_ISO_CTRL + 1) | 0x08);
3193
3194         rsu_write_1(sc, R92S_SYS_FUNC_EN + 1,
3195             rsu_read_1(sc, R92S_SYS_FUNC_EN + 1) | 0x20);
3196
3197         /* Support 64KB IMEM. */
3198         rsu_write_1(sc, R92S_SYS_ISO_CTRL + 1,
3199             rsu_read_1(sc, R92S_SYS_ISO_CTRL + 1) & ~0x97);
3200
3201         /* Enable AFE clock. */
3202         rsu_write_1(sc, R92S_AFE_XTAL_CTRL + 1,
3203             rsu_read_1(sc, R92S_AFE_XTAL_CTRL + 1) & ~0x04);
3204         /* Enable AFE PLL macro block. */
3205         reg = rsu_read_1(sc, R92S_AFE_PLL_CTRL);
3206         rsu_write_1(sc, R92S_AFE_PLL_CTRL, reg | 0x11);
3207         rsu_ms_delay(sc, 1);
3208         rsu_write_1(sc, R92S_AFE_PLL_CTRL, reg | 0x51);
3209         rsu_ms_delay(sc, 1);
3210         rsu_write_1(sc, R92S_AFE_PLL_CTRL, reg | 0x11);
3211         rsu_ms_delay(sc, 1);
3212
3213         /* Attach AFE PLL to MACTOP/BB. */
3214         rsu_write_1(sc, R92S_SYS_ISO_CTRL,
3215             rsu_read_1(sc, R92S_SYS_ISO_CTRL) & ~0x11);
3216
3217         /* Switch to 40MHz clock. */
3218         rsu_write_1(sc, R92S_SYS_CLKR, 0x00);
3219         /* Disable CPU clock and 80MHz SSC. */
3220         rsu_write_1(sc, R92S_SYS_CLKR,
3221             rsu_read_1(sc, R92S_SYS_CLKR) | 0xa0);
3222         /* Enable MAC clock. */
3223         rsu_write_2(sc, R92S_SYS_CLKR,
3224             rsu_read_2(sc, R92S_SYS_CLKR) |
3225             R92S_MAC_CLK_EN | R92S_SYS_CLK_EN);
3226
3227         rsu_write_1(sc, R92S_PMC_FSM, 0x02);
3228
3229         /* Enable digital core and IOREG R/W. */
3230         rsu_write_1(sc, R92S_SYS_FUNC_EN + 1,
3231             rsu_read_1(sc, R92S_SYS_FUNC_EN + 1) | 0x08);
3232
3233         rsu_write_1(sc, R92S_SYS_FUNC_EN + 1,
3234             rsu_read_1(sc, R92S_SYS_FUNC_EN + 1) | 0x80);
3235
3236         /* Switch the control path to firmware. */
3237         reg = rsu_read_2(sc, R92S_SYS_CLKR);
3238         reg = (reg & ~R92S_SWHW_SEL) | R92S_FWHW_SEL;
3239         rsu_write_2(sc, R92S_SYS_CLKR, reg);
3240
3241         rsu_write_2(sc, R92S_CR, 0x37fc);
3242
3243         /* Fix USB RX FIFO issue. */
3244         rsu_write_1(sc, 0xfe5c,
3245             rsu_read_1(sc, 0xfe5c) | 0x80);
3246
3247         rsu_write_1(sc, R92S_SYS_CLKR,
3248             rsu_read_1(sc, R92S_SYS_CLKR) & ~R92S_SYS_CPU_CLKSEL);
3249
3250         rsu_write_1(sc, 0xfe1c, 0x80);
3251
3252         /* Make sure TxDMA is ready to download firmware. */
3253         for (ntries = 0; ntries < 20; ntries++) {
3254                 reg = rsu_read_1(sc, R92S_TCR);
3255                 if ((reg & (R92S_TCR_IMEM_CHK_RPT | R92S_TCR_EMEM_CHK_RPT)) ==
3256                     (R92S_TCR_IMEM_CHK_RPT | R92S_TCR_EMEM_CHK_RPT))
3257                         break;
3258                 rsu_ms_delay(sc, 1);
3259         }
3260         if (ntries == 20) {
3261                 RSU_DPRINTF(sc, RSU_DEBUG_RESET | RSU_DEBUG_TX,
3262                     "%s: TxDMA is not ready\n",
3263                     __func__);
3264                 /* Reset TxDMA. */
3265                 reg = rsu_read_1(sc, R92S_CR);
3266                 rsu_write_1(sc, R92S_CR, reg & ~R92S_CR_TXDMA_EN);
3267                 rsu_ms_delay(sc, 1);
3268                 rsu_write_1(sc, R92S_CR, reg | R92S_CR_TXDMA_EN);
3269         }
3270 }
3271
3272 static void
3273 rsu_power_off(struct rsu_softc *sc)
3274 {
3275         /* Turn RF off. */
3276         rsu_write_1(sc, R92S_RF_CTRL, 0x00);
3277         rsu_ms_delay(sc, 5);
3278
3279         /* Turn MAC off. */
3280         /* Switch control path. */
3281         rsu_write_1(sc, R92S_SYS_CLKR + 1, 0x38);
3282         /* Reset MACTOP. */
3283         rsu_write_1(sc, R92S_SYS_FUNC_EN + 1, 0x70);
3284         rsu_write_1(sc, R92S_PMC_FSM, 0x06);
3285         rsu_write_1(sc, R92S_SYS_ISO_CTRL + 0, 0xf9);
3286         rsu_write_1(sc, R92S_SYS_ISO_CTRL + 1, 0xe8);
3287
3288         /* Disable AFE PLL. */
3289         rsu_write_1(sc, R92S_AFE_PLL_CTRL, 0x00);
3290         /* Disable A15V. */
3291         rsu_write_1(sc, R92S_LDOA15_CTRL, 0x54);
3292         /* Disable eFuse 1.2V. */
3293         rsu_write_1(sc, R92S_SYS_FUNC_EN + 1, 0x50);
3294         rsu_write_1(sc, R92S_LDOV12D_CTRL, 0x24);
3295         /* Enable AFE macro block's bandgap and Mbias. */
3296         rsu_write_1(sc, R92S_AFE_MISC, 0x30);
3297         /* Disable 1.6V LDO. */
3298         rsu_write_1(sc, R92S_SPS0_CTRL + 0, 0x56);
3299         rsu_write_1(sc, R92S_SPS0_CTRL + 1, 0x43);
3300
3301         /* Firmware - tell it to switch things off */
3302         (void) rsu_set_fw_power_state(sc, RSU_PWR_OFF);
3303 }
3304
3305 static int
3306 rsu_fw_loadsection(struct rsu_softc *sc, const uint8_t *buf, int len)
3307 {
3308         const uint8_t which = rsu_wme_ac_xfer_map[WME_AC_VO];
3309         struct rsu_data *data;
3310         struct r92s_tx_desc *txd;
3311         int mlen;
3312
3313         while (len > 0) {
3314                 data = rsu_getbuf(sc);
3315                 if (data == NULL)
3316                         return (ENOMEM);
3317                 txd = (struct r92s_tx_desc *)data->buf;
3318                 memset(txd, 0, sizeof(*txd));
3319                 if (len <= RSU_TXBUFSZ - sizeof(*txd)) {
3320                         /* Last chunk. */
3321                         txd->txdw0 |= htole32(R92S_TXDW0_LINIP);
3322                         mlen = len;
3323                 } else
3324                         mlen = RSU_TXBUFSZ - sizeof(*txd);
3325                 txd->txdw0 |= htole32(SM(R92S_TXDW0_PKTLEN, mlen));
3326                 memcpy(&txd[1], buf, mlen);
3327                 data->buflen = sizeof(*txd) + mlen;
3328                 RSU_DPRINTF(sc, RSU_DEBUG_TX | RSU_DEBUG_FW | RSU_DEBUG_RESET,
3329                     "%s: starting transfer %p\n",
3330                     __func__, data);
3331                 STAILQ_INSERT_TAIL(&sc->sc_tx_pending[which], data, next);
3332                 buf += mlen;
3333                 len -= mlen;
3334         }
3335         usbd_transfer_start(sc->sc_xfer[which]);
3336         return (0);
3337 }
3338
3339 CTASSERT(sizeof(size_t) >= sizeof(uint32_t));
3340
3341 static int
3342 rsu_load_firmware(struct rsu_softc *sc)
3343 {
3344         const struct r92s_fw_hdr *hdr;
3345         struct r92s_fw_priv *dmem;
3346         struct ieee80211com *ic = &sc->sc_ic;
3347         const uint8_t *imem, *emem;
3348         uint32_t imemsz, ememsz;
3349         const struct firmware *fw;
3350         size_t size;
3351         uint32_t reg;
3352         int ntries, error;
3353
3354         if (rsu_read_1(sc, R92S_TCR) & R92S_TCR_FWRDY) {
3355                 RSU_DPRINTF(sc, RSU_DEBUG_ANY,
3356                     "%s: Firmware already loaded\n",
3357                     __func__);
3358                 return (0);
3359         }
3360
3361         RSU_UNLOCK(sc);
3362         /* Read firmware image from the filesystem. */
3363         if ((fw = firmware_get("rsu-rtl8712fw")) == NULL) {
3364                 device_printf(sc->sc_dev,
3365                     "%s: failed load firmware of file rsu-rtl8712fw\n",
3366                     __func__);
3367                 RSU_LOCK(sc);
3368                 return (ENXIO);
3369         }
3370         RSU_LOCK(sc);
3371         size = fw->datasize;
3372         if (size < sizeof(*hdr)) {
3373                 device_printf(sc->sc_dev, "firmware too short\n");
3374                 error = EINVAL;
3375                 goto fail;
3376         }
3377         hdr = (const struct r92s_fw_hdr *)fw->data;
3378         if (hdr->signature != htole16(0x8712) &&
3379             hdr->signature != htole16(0x8192)) {
3380                 device_printf(sc->sc_dev,
3381                     "invalid firmware signature 0x%x\n",
3382                     le16toh(hdr->signature));
3383                 error = EINVAL;
3384                 goto fail;
3385         }
3386         RSU_DPRINTF(sc, RSU_DEBUG_FW, "FW V%d %02x-%02x %02x:%02x\n",
3387             le16toh(hdr->version), hdr->month, hdr->day, hdr->hour,
3388             hdr->minute);
3389
3390         /* Make sure that driver and firmware are in sync. */
3391         if (hdr->privsz != htole32(sizeof(*dmem))) {
3392                 device_printf(sc->sc_dev, "unsupported firmware image\n");
3393                 error = EINVAL;
3394                 goto fail;
3395         }
3396         /* Get FW sections sizes. */
3397         imemsz = le32toh(hdr->imemsz);
3398         ememsz = le32toh(hdr->sramsz);
3399         /* Check that all FW sections fit in image. */
3400         if (imemsz > (size_t)(size - sizeof(*hdr)) ||
3401             ememsz > (size_t)(size - sizeof(*hdr) - imemsz)) {
3402                 device_printf(sc->sc_dev, "firmware too short\n");
3403                 error = EINVAL;
3404                 goto fail;
3405         }
3406         imem = (const uint8_t *)&hdr[1];
3407         emem = imem + imemsz;
3408
3409         /* Load IMEM section. */
3410         error = rsu_fw_loadsection(sc, imem, imemsz);
3411         if (error != 0) {
3412                 device_printf(sc->sc_dev,
3413                     "could not load firmware section %s\n", "IMEM");
3414                 goto fail;
3415         }
3416         /* Wait for load to complete. */
3417         for (ntries = 0; ntries != 50; ntries++) {
3418                 rsu_ms_delay(sc, 10);
3419                 reg = rsu_read_1(sc, R92S_TCR);
3420                 if (reg & R92S_TCR_IMEM_CODE_DONE)
3421                         break;
3422         }
3423         if (ntries == 50) {
3424                 device_printf(sc->sc_dev, "timeout waiting for IMEM transfer\n");
3425                 error = ETIMEDOUT;
3426                 goto fail;
3427         }
3428         /* Load EMEM section. */
3429         error = rsu_fw_loadsection(sc, emem, ememsz);
3430         if (error != 0) {
3431                 device_printf(sc->sc_dev,
3432                     "could not load firmware section %s\n", "EMEM");
3433                 goto fail;
3434         }
3435         /* Wait for load to complete. */
3436         for (ntries = 0; ntries != 50; ntries++) {
3437                 rsu_ms_delay(sc, 10);
3438                 reg = rsu_read_2(sc, R92S_TCR);
3439                 if (reg & R92S_TCR_EMEM_CODE_DONE)
3440                         break;
3441         }
3442         if (ntries == 50) {
3443                 device_printf(sc->sc_dev, "timeout waiting for EMEM transfer\n");
3444                 error = ETIMEDOUT;
3445                 goto fail;
3446         }
3447         /* Enable CPU. */
3448         rsu_write_1(sc, R92S_SYS_CLKR,
3449             rsu_read_1(sc, R92S_SYS_CLKR) | R92S_SYS_CPU_CLKSEL);
3450         if (!(rsu_read_1(sc, R92S_SYS_CLKR) & R92S_SYS_CPU_CLKSEL)) {
3451                 device_printf(sc->sc_dev, "could not enable system clock\n");
3452                 error = EIO;
3453                 goto fail;
3454         }
3455         rsu_write_2(sc, R92S_SYS_FUNC_EN,
3456             rsu_read_2(sc, R92S_SYS_FUNC_EN) | R92S_FEN_CPUEN);
3457         if (!(rsu_read_2(sc, R92S_SYS_FUNC_EN) & R92S_FEN_CPUEN)) {
3458                 device_printf(sc->sc_dev,
3459                     "could not enable microcontroller\n");
3460                 error = EIO;
3461                 goto fail;
3462         }
3463         /* Wait for CPU to initialize. */
3464         for (ntries = 0; ntries < 100; ntries++) {
3465                 if (rsu_read_1(sc, R92S_TCR) & R92S_TCR_IMEM_RDY)
3466                         break;
3467                 rsu_ms_delay(sc, 1);
3468         }
3469         if (ntries == 100) {
3470                 device_printf(sc->sc_dev,
3471                     "timeout waiting for microcontroller\n");
3472                 error = ETIMEDOUT;
3473                 goto fail;
3474         }
3475
3476         /* Update DMEM section before loading. */
3477         dmem = __DECONST(struct r92s_fw_priv *, &hdr->priv);
3478         memset(dmem, 0, sizeof(*dmem));
3479         dmem->hci_sel = R92S_HCI_SEL_USB | R92S_HCI_SEL_8172;
3480         dmem->nendpoints = sc->sc_nendpoints;
3481         dmem->chip_version = sc->cut;
3482         dmem->rf_config = sc->sc_rftype;
3483         dmem->vcs_type = R92S_VCS_TYPE_AUTO;
3484         dmem->vcs_mode = R92S_VCS_MODE_RTS_CTS;
3485         dmem->turbo_mode = 0;
3486         dmem->bw40_en = !! (ic->ic_htcaps & IEEE80211_HTCAP_CHWIDTH40);
3487         dmem->amsdu2ampdu_en = !! (sc->sc_ht);
3488         dmem->ampdu_en = !! (sc->sc_ht);
3489         dmem->agg_offload = !! (sc->sc_ht);
3490         dmem->qos_en = 1;
3491         dmem->ps_offload = 1;
3492         dmem->lowpower_mode = 1;        /* XXX TODO: configurable? */
3493         /* Load DMEM section. */
3494         error = rsu_fw_loadsection(sc, (uint8_t *)dmem, sizeof(*dmem));
3495         if (error != 0) {
3496                 device_printf(sc->sc_dev,
3497                     "could not load firmware section %s\n", "DMEM");
3498                 goto fail;
3499         }
3500         /* Wait for load to complete. */
3501         for (ntries = 0; ntries < 100; ntries++) {
3502                 if (rsu_read_1(sc, R92S_TCR) & R92S_TCR_DMEM_CODE_DONE)
3503                         break;
3504                 rsu_ms_delay(sc, 1);
3505         }
3506         if (ntries == 100) {
3507                 device_printf(sc->sc_dev, "timeout waiting for %s transfer\n",
3508                     "DMEM");
3509                 error = ETIMEDOUT;
3510                 goto fail;
3511         }
3512         /* Wait for firmware readiness. */
3513         for (ntries = 0; ntries < 60; ntries++) {
3514                 if (!(rsu_read_1(sc, R92S_TCR) & R92S_TCR_FWRDY))
3515                         break;
3516                 rsu_ms_delay(sc, 1);
3517         }
3518         if (ntries == 60) {
3519                 device_printf(sc->sc_dev,
3520                     "timeout waiting for firmware readiness\n");
3521                 error = ETIMEDOUT;
3522                 goto fail;
3523         }
3524  fail:
3525         firmware_put(fw, FIRMWARE_UNLOAD);
3526         return (error);
3527 }
3528
3529
3530 static int
3531 rsu_raw_xmit(struct ieee80211_node *ni, struct mbuf *m,
3532     const struct ieee80211_bpf_params *params)
3533 {
3534         struct ieee80211com *ic = ni->ni_ic;
3535         struct rsu_softc *sc = ic->ic_softc;
3536         struct rsu_data *bf;
3537
3538         /* prevent management frames from being sent if we're not ready */
3539         if (!sc->sc_running) {
3540                 m_freem(m);
3541                 return (ENETDOWN);
3542         }
3543         RSU_LOCK(sc);
3544         bf = rsu_getbuf(sc);
3545         if (bf == NULL) {
3546                 m_freem(m);
3547                 RSU_UNLOCK(sc);
3548                 return (ENOBUFS);
3549         }
3550         if (rsu_tx_start(sc, ni, m, bf) != 0) {
3551                 m_freem(m);
3552                 rsu_freebuf(sc, bf);
3553                 RSU_UNLOCK(sc);
3554                 return (EIO);
3555         }
3556         RSU_UNLOCK(sc);
3557
3558         return (0);
3559 }
3560
3561 static void
3562 rsu_rxfilter_init(struct rsu_softc *sc)
3563 {
3564         uint32_t reg;
3565
3566         RSU_ASSERT_LOCKED(sc);
3567
3568         /* Setup multicast filter. */
3569         rsu_set_multi(sc);
3570
3571         /* Adjust Rx filter. */
3572         reg = rsu_read_4(sc, R92S_RCR);
3573         reg &= ~R92S_RCR_AICV;
3574         reg |= R92S_RCR_APP_PHYSTS;
3575         if (sc->sc_rx_checksum_enable)
3576                 reg |= R92S_RCR_TCP_OFFLD_EN;
3577         rsu_write_4(sc, R92S_RCR, reg);
3578
3579         /* Update dynamic Rx filter parts. */
3580         rsu_rxfilter_refresh(sc);
3581 }
3582
3583 static void
3584 rsu_rxfilter_set(struct rsu_softc *sc, uint32_t clear, uint32_t set)
3585 {
3586         /* NB: firmware can touch this register too. */
3587         rsu_write_4(sc, R92S_RCR,
3588            (rsu_read_4(sc, R92S_RCR) & ~clear) | set);
3589 }
3590
3591 static void
3592 rsu_rxfilter_refresh(struct rsu_softc *sc)
3593 {
3594         struct ieee80211com *ic = &sc->sc_ic;
3595         uint32_t mask_all, mask_min;
3596
3597         RSU_ASSERT_LOCKED(sc);
3598
3599         /* NB: RCR_AMF / RXFLTMAP_MGT are used by firmware. */
3600         mask_all = R92S_RCR_ACF | R92S_RCR_AAP;
3601         mask_min = R92S_RCR_APM;
3602         if (sc->sc_vap_is_running)
3603                 mask_min |= R92S_RCR_CBSSID;
3604         else
3605                 mask_all |= R92S_RCR_ADF;
3606
3607         if (ic->ic_opmode == IEEE80211_M_MONITOR) {
3608                 uint16_t rxfltmap;
3609                 if (sc->sc_vap_is_running)
3610                         rxfltmap = 0;
3611                 else
3612                         rxfltmap = R92S_RXFLTMAP_MGT_DEF;
3613                 rsu_write_2(sc, R92S_RXFLTMAP_MGT, rxfltmap);
3614         }
3615
3616         if (ic->ic_promisc == 0 && ic->ic_opmode != IEEE80211_M_MONITOR)
3617                 rsu_rxfilter_set(sc, mask_all, mask_min);
3618         else
3619                 rsu_rxfilter_set(sc, mask_min, mask_all);
3620 }
3621
3622 static int
3623 rsu_init(struct rsu_softc *sc)
3624 {
3625         struct ieee80211com *ic = &sc->sc_ic;
3626         struct ieee80211vap *vap = TAILQ_FIRST(&ic->ic_vaps);
3627         uint8_t macaddr[IEEE80211_ADDR_LEN];
3628         int error;
3629         int i;
3630
3631         RSU_LOCK(sc);
3632
3633         if (sc->sc_running) {
3634                 RSU_UNLOCK(sc);
3635                 return (0);
3636         }
3637
3638         /* Ensure the mbuf queue is drained */
3639         rsu_drain_mbufq(sc);
3640
3641         /* Reset power management state. */
3642         rsu_write_1(sc, R92S_USB_HRPWM, 0);
3643
3644         /* Power on adapter. */
3645         if (sc->cut == 1)
3646                 rsu_power_on_acut(sc);
3647         else
3648                 rsu_power_on_bcut(sc);
3649
3650         /* Load firmware. */
3651         error = rsu_load_firmware(sc);
3652         if (error != 0)
3653                 goto fail;
3654
3655         rsu_write_4(sc, R92S_CR,
3656             rsu_read_4(sc, R92S_CR) & ~0xff000000);
3657
3658         /* Use 128 bytes pages. */
3659         rsu_write_1(sc, 0x00b5,
3660             rsu_read_1(sc, 0x00b5) | 0x01);
3661         /* Enable USB Rx aggregation. */
3662         rsu_write_1(sc, 0x00bd,
3663             rsu_read_1(sc, 0x00bd) | 0x80);
3664         /* Set USB Rx aggregation threshold. */
3665         rsu_write_1(sc, 0x00d9, 0x01);
3666         /* Set USB Rx aggregation timeout (1.7ms/4). */
3667         rsu_write_1(sc, 0xfe5b, 0x04);
3668         /* Fix USB Rx FIFO issue. */
3669         rsu_write_1(sc, 0xfe5c,
3670             rsu_read_1(sc, 0xfe5c) | 0x80);
3671
3672         /* Set MAC address. */
3673         IEEE80211_ADDR_COPY(macaddr, vap ? vap->iv_myaddr : ic->ic_macaddr);
3674         rsu_write_region_1(sc, R92S_MACID, macaddr, IEEE80211_ADDR_LEN);
3675
3676         /* It really takes 1.5 seconds for the firmware to boot: */
3677         usb_pause_mtx(&sc->sc_mtx, USB_MS_TO_TICKS(2000));
3678
3679         RSU_DPRINTF(sc, RSU_DEBUG_RESET, "%s: setting MAC address to %s\n",
3680             __func__,
3681             ether_sprintf(macaddr));
3682         error = rsu_fw_cmd(sc, R92S_CMD_SET_MAC_ADDRESS, macaddr,
3683             IEEE80211_ADDR_LEN);
3684         if (error != 0) {
3685                 device_printf(sc->sc_dev, "could not set MAC address\n");
3686                 goto fail;
3687         }
3688
3689         /* Initialize Rx filter. */
3690         rsu_rxfilter_init(sc);
3691
3692         /* Set PS mode fully active */
3693         error = rsu_set_fw_power_state(sc, RSU_PWR_ACTIVE);
3694         if (error != 0) {
3695                 device_printf(sc->sc_dev, "could not set PS mode\n");
3696                 goto fail;
3697         }
3698
3699         /* Install static keys (if any). */
3700         error = rsu_reinit_static_keys(sc);
3701         if (error != 0)
3702                 goto fail;
3703
3704         sc->sc_extra_scan = 0;
3705         usbd_transfer_start(sc->sc_xfer[RSU_BULK_RX]);
3706
3707         /* We're ready to go. */
3708         sc->sc_running = 1;
3709         RSU_UNLOCK(sc);
3710
3711         return (0);
3712 fail:
3713         /* Need to stop all failed transfers, if any */
3714         for (i = 0; i != RSU_N_TRANSFER; i++)
3715                 usbd_transfer_stop(sc->sc_xfer[i]);
3716         RSU_UNLOCK(sc);
3717
3718         return (error);
3719 }
3720
3721 static void
3722 rsu_stop(struct rsu_softc *sc)
3723 {
3724         int i;
3725
3726         RSU_LOCK(sc);
3727         if (!sc->sc_running) {
3728                 RSU_UNLOCK(sc);
3729                 return;
3730         }
3731
3732         sc->sc_running = 0;
3733         sc->sc_vap_is_running = 0;
3734         sc->sc_calibrating = 0;
3735         taskqueue_cancel_timeout(taskqueue_thread, &sc->calib_task, NULL);
3736         taskqueue_cancel(taskqueue_thread, &sc->tx_task, NULL);
3737
3738         /* Power off adapter. */
3739         rsu_power_off(sc);
3740
3741         /*
3742          * CAM is not accessible after shutdown;
3743          * all entries are marked (by firmware?) as invalid.
3744          */
3745         memset(sc->free_keys_bmap, 0, sizeof(sc->free_keys_bmap));
3746         memset(sc->keys_bmap, 0, sizeof(sc->keys_bmap));
3747
3748         for (i = 0; i < RSU_N_TRANSFER; i++)
3749                 usbd_transfer_stop(sc->sc_xfer[i]);
3750
3751         /* Ensure the mbuf queue is drained */
3752         rsu_drain_mbufq(sc);
3753         RSU_UNLOCK(sc);
3754 }
3755
3756 /*
3757  * Note: usb_pause_mtx() actually releases the mutex before calling pause(),
3758  * which breaks any kind of driver serialisation.
3759  */
3760 static void
3761 rsu_ms_delay(struct rsu_softc *sc, int ms)
3762 {
3763
3764         //usb_pause_mtx(&sc->sc_mtx, hz / 1000);
3765         DELAY(ms * 1000);
3766 }