/*      $NetBSD: if_iwi.c,v 1.121 2024/07/05 04:31:51 rin Exp $  */
/*      $OpenBSD: if_iwi.c,v 1.111 2010/11/15 19:11:57 damien Exp $     */

/*-
* Copyright (c) 2004-2008
*      Damien Bergamini <[email protected]>. All rights reserved.
*
* Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/

#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: if_iwi.c,v 1.121 2024/07/05 04:31:51 rin Exp $");

/*-
* Intel(R) PRO/Wireless 2200BG/2225BG/2915ABG driver
* http://www.intel.com/network/connectivity/products/wireless/prowireless_mobile.htm
*/


#include <sys/param.h>
#include <sys/sockio.h>
#include <sys/sysctl.h>
#include <sys/mbuf.h>
#include <sys/kernel.h>
#include <sys/socket.h>
#include <sys/systm.h>
#include <sys/malloc.h>
#include <sys/conf.h>
#include <sys/kauth.h>
#include <sys/proc.h>
#include <sys/cprng.h>

#include <sys/bus.h>
#include <machine/endian.h>
#include <sys/intr.h>

#include <dev/firmload.h>

#include <dev/pci/pcireg.h>
#include <dev/pci/pcivar.h>
#include <dev/pci/pcidevs.h>

#include <net/bpf.h>
#include <net/if.h>
#include <net/if_arp.h>
#include <net/if_dl.h>
#include <net/if_ether.h>
#include <net/if_media.h>
#include <net/if_types.h>

#include <net80211/ieee80211_var.h>
#include <net80211/ieee80211_radiotap.h>

#include <netinet/in.h>
#include <netinet/in_systm.h>
#include <netinet/in_var.h>
#include <netinet/ip.h>

#include <dev/pci/if_iwireg.h>
#include <dev/pci/if_iwivar.h>

#ifdef IWI_DEBUG
#define DPRINTF(x)      if (iwi_debug > 0) printf x
#define DPRINTFN(n, x)  if (iwi_debug >= (n)) printf x
int iwi_debug = 4;
#else
#define DPRINTF(x)
#define DPRINTFN(n, x)
#endif

/* Permit loading the Intel firmware */
static int iwi_accept_eula;

static int      iwi_match(device_t, cfdata_t, void *);
static void     iwi_attach(device_t, device_t, void *);
static int      iwi_detach(device_t, int);

static int      iwi_alloc_cmd_ring(struct iwi_softc *, struct iwi_cmd_ring *,
   int);
static void     iwi_reset_cmd_ring(struct iwi_softc *, struct iwi_cmd_ring *);
static void     iwi_free_cmd_ring(struct iwi_softc *, struct iwi_cmd_ring *);
static int      iwi_alloc_tx_ring(struct iwi_softc *, struct iwi_tx_ring *,
   int, bus_size_t, bus_size_t);
static void     iwi_reset_tx_ring(struct iwi_softc *, struct iwi_tx_ring *);
static void     iwi_free_tx_ring(struct iwi_softc *, struct iwi_tx_ring *);
static struct mbuf *
               iwi_alloc_rx_buf(struct iwi_softc *sc);
static int      iwi_alloc_rx_ring(struct iwi_softc *, struct iwi_rx_ring *,
   int);
static void     iwi_reset_rx_ring(struct iwi_softc *, struct iwi_rx_ring *);
static void     iwi_free_rx_ring(struct iwi_softc *, struct iwi_rx_ring *);

static struct   ieee80211_node *iwi_node_alloc(struct ieee80211_node_table *);
static void     iwi_node_free(struct ieee80211_node *);

static int      iwi_cvtrate(int);
static int      iwi_media_change(struct ifnet *);
static void     iwi_media_status(struct ifnet *, struct ifmediareq *);
static int      iwi_wme_update(struct ieee80211com *);
static uint16_t iwi_read_prom_word(struct iwi_softc *, uint8_t);
static int      iwi_newstate(struct ieee80211com *, enum ieee80211_state, int);
static void     iwi_fix_channel(struct ieee80211com *, struct mbuf *);
static void     iwi_frame_intr(struct iwi_softc *, struct iwi_rx_data *, int,
   struct iwi_frame *);
static void     iwi_notification_intr(struct iwi_softc *, struct iwi_notif *);
static void     iwi_cmd_intr(struct iwi_softc *);
static void     iwi_rx_intr(struct iwi_softc *);
static void     iwi_tx_intr(struct iwi_softc *, struct iwi_tx_ring *);
static int      iwi_intr(void *);
static void     iwi_softintr(void *);
static int      iwi_cmd(struct iwi_softc *, uint8_t, void *, uint8_t, int);
static void     iwi_write_ibssnode(struct iwi_softc *, const struct iwi_node *);
static int      iwi_tx_start(struct ifnet *, struct mbuf *, struct ieee80211_node *,
   int);
static void     iwi_start(struct ifnet *);
static void     iwi_watchdog(struct ifnet *);

static int      iwi_alloc_unr(struct iwi_softc *);
static void     iwi_free_unr(struct iwi_softc *, int);

static int      iwi_get_table0(struct iwi_softc *, uint32_t *);

static int      iwi_ioctl(struct ifnet *, u_long, void *);
static void     iwi_stop_master(struct iwi_softc *);
static int      iwi_reset(struct iwi_softc *);
static int      iwi_load_ucode(struct iwi_softc *, void *, int);
static int      iwi_load_firmware(struct iwi_softc *, void *, int);
static int      iwi_cache_firmware(struct iwi_softc *);
static void     iwi_free_firmware(struct iwi_softc *);
static int      iwi_config(struct iwi_softc *);
static int      iwi_set_chan(struct iwi_softc *, struct ieee80211_channel *);
static int      iwi_scan(struct iwi_softc *);
static int      iwi_auth_and_assoc(struct iwi_softc *);
static int      iwi_init(struct ifnet *);
static void     iwi_stop(struct ifnet *, int);
static int      iwi_getrfkill(struct iwi_softc *);
static void     iwi_led_set(struct iwi_softc *, uint32_t, int);
static void     iwi_sysctlattach(struct iwi_softc *);

static inline uint8_t
MEM_READ_1(struct iwi_softc *sc, uint32_t addr)
{
       CSR_WRITE_4(sc, IWI_CSR_INDIRECT_ADDR, addr);
       return CSR_READ_1(sc, IWI_CSR_INDIRECT_DATA);
}

static inline uint32_t
MEM_READ_4(struct iwi_softc *sc, uint32_t addr)
{
       CSR_WRITE_4(sc, IWI_CSR_INDIRECT_ADDR, addr);
       return CSR_READ_4(sc, IWI_CSR_INDIRECT_DATA);
}

CFATTACH_DECL_NEW(iwi, sizeof (struct iwi_softc), iwi_match, iwi_attach,
   iwi_detach, NULL);

static const struct device_compatible_entry compat_data[] = {
       { .id = PCI_ID_CODE(PCI_VENDOR_INTEL,
               PCI_PRODUCT_INTEL_PRO_WL_2200BG), },

       { .id = PCI_ID_CODE(PCI_VENDOR_INTEL,
               PCI_PRODUCT_INTEL_PRO_WL_2225BG), },

       { .id = PCI_ID_CODE(PCI_VENDOR_INTEL,
               PCI_PRODUCT_INTEL_PRO_WL_2915ABG_1), },

       { .id = PCI_ID_CODE(PCI_VENDOR_INTEL,
               PCI_PRODUCT_INTEL_PRO_WL_2915ABG_2), },


       PCI_COMPAT_EOL
};

static int
iwi_match(device_t parent, cfdata_t match, void *aux)
{
       struct pci_attach_args *pa = aux;

       return pci_compatible_match(pa, compat_data);
}

/* Base Address Register */
#define IWI_PCI_BAR0    0x10

static void
iwi_attach(device_t parent, device_t self, void *aux)
{
       struct iwi_softc *sc = device_private(self);
       struct ieee80211com *ic = &sc->sc_ic;
       struct ifnet *ifp = &sc->sc_if;
       struct pci_attach_args *pa = aux;
       const char *intrstr;
       bus_space_tag_t memt;
       bus_space_handle_t memh;
       pci_intr_handle_t ih;
       pcireg_t data;
       uint16_t val;
       int error, i;
       char intrbuf[PCI_INTRSTR_LEN];

       sc->sc_dev = self;
       sc->sc_pct = pa->pa_pc;
       sc->sc_pcitag = pa->pa_tag;

       pci_aprint_devinfo(pa, NULL);

       /* clear unit numbers allocated to IBSS */
       sc->sc_unr = 0;

       /* power up chip */
       if ((error = pci_activate(pa->pa_pc, pa->pa_tag, self,
           NULL)) && error != EOPNOTSUPP) {
               aprint_error_dev(self, "cannot activate %d\n", error);
               return;
       }

       /* clear device specific PCI configuration register 0x41 */
       data = pci_conf_read(sc->sc_pct, sc->sc_pcitag, 0x40);
       data &= ~0x0000ff00;
       pci_conf_write(sc->sc_pct, sc->sc_pcitag, 0x40, data);


       /* enable bus-mastering */
       data = pci_conf_read(sc->sc_pct, sc->sc_pcitag, PCI_COMMAND_STATUS_REG);
       data |= PCI_COMMAND_MASTER_ENABLE;
       pci_conf_write(sc->sc_pct, sc->sc_pcitag, PCI_COMMAND_STATUS_REG, data);

       /* map the register window */
       error = pci_mapreg_map(pa, IWI_PCI_BAR0, PCI_MAPREG_TYPE_MEM |
           PCI_MAPREG_MEM_TYPE_32BIT, 0, &memt, &memh, NULL, &sc->sc_sz);
       if (error != 0) {
               aprint_error_dev(self, "could not map memory space\n");
               return;
       }

       sc->sc_st = memt;
       sc->sc_sh = memh;
       sc->sc_dmat = pa->pa_dmat;

       /* disable interrupts */
       CSR_WRITE_4(sc, IWI_CSR_INTR_MASK, 0);

       sc->sc_soft_ih = softint_establish(SOFTINT_NET, iwi_softintr, sc);
       if (sc->sc_soft_ih == NULL) {
               aprint_error_dev(self, "could not establish softint\n");
               return;
       }

       if (pci_intr_map(pa, &ih) != 0) {
               softint_disestablish(sc->sc_soft_ih);
               sc->sc_soft_ih = NULL;
               aprint_error_dev(self, "could not map interrupt\n");
               return;
       }

       intrstr = pci_intr_string(sc->sc_pct, ih, intrbuf, sizeof(intrbuf));
       sc->sc_ih = pci_intr_establish_xname(sc->sc_pct, ih, IPL_NET, iwi_intr,
           sc, device_xname(self));
       if (sc->sc_ih == NULL) {
               softint_disestablish(sc->sc_soft_ih);
               sc->sc_soft_ih = NULL;
               aprint_error_dev(self, "could not establish interrupt");
               if (intrstr != NULL)
                       aprint_error(" at %s", intrstr);
               aprint_error("\n");
               return;
       }
       aprint_normal_dev(self, "interrupting at %s\n", intrstr);

       if (iwi_reset(sc) != 0) {
               pci_intr_disestablish(sc->sc_pct, sc->sc_ih);
               softint_disestablish(sc->sc_soft_ih);
               sc->sc_soft_ih = NULL;
               aprint_error_dev(self, "could not reset adapter\n");
               return;
       }

       ic->ic_ifp = ifp;
       ic->ic_wme.wme_update = iwi_wme_update;
       ic->ic_phytype = IEEE80211_T_OFDM; /* not only, but not used */
       ic->ic_opmode = IEEE80211_M_STA; /* default to BSS mode */
       ic->ic_state = IEEE80211_S_INIT;

       sc->sc_fwname = "ipw2200-bss.fw";

       /* set device capabilities */
       ic->ic_caps =
           IEEE80211_C_IBSS |          /* IBSS mode supported */
           IEEE80211_C_MONITOR |       /* monitor mode supported */
           IEEE80211_C_TXPMGT |        /* tx power management */
           IEEE80211_C_SHPREAMBLE |    /* short preamble supported */
           IEEE80211_C_SHSLOT |        /* short slot time supported */
           IEEE80211_C_WPA |           /* 802.11i */
           IEEE80211_C_WME;            /* 802.11e */

       /* read MAC address from EEPROM */
       val = iwi_read_prom_word(sc, IWI_EEPROM_MAC + 0);
       ic->ic_myaddr[0] = val & 0xff;
       ic->ic_myaddr[1] = val >> 8;
       val = iwi_read_prom_word(sc, IWI_EEPROM_MAC + 1);
       ic->ic_myaddr[2] = val & 0xff;
       ic->ic_myaddr[3] = val >> 8;
       val = iwi_read_prom_word(sc, IWI_EEPROM_MAC + 2);
       ic->ic_myaddr[4] = val & 0xff;
       ic->ic_myaddr[5] = val >> 8;

       aprint_verbose_dev(self, "802.11 address %s\n",
           ether_sprintf(ic->ic_myaddr));

       /* read the NIC type from EEPROM */
       val = iwi_read_prom_word(sc, IWI_EEPROM_NIC_TYPE);
       sc->nictype = val & 0xff;

       DPRINTF(("%s: NIC type %d\n", device_xname(self), sc->nictype));

       if (PCI_PRODUCT(pa->pa_id) == PCI_PRODUCT_INTEL_PRO_WL_2915ABG_1 ||
           PCI_PRODUCT(pa->pa_id) == PCI_PRODUCT_INTEL_PRO_WL_2915ABG_2) {
               /* set supported .11a rates (2915ABG only) */
               ic->ic_sup_rates[IEEE80211_MODE_11A] = ieee80211_std_rateset_11a;

               /* set supported .11a channels */
               for (i = 36; i <= 64; i += 4) {
                       ic->ic_channels[i].ic_freq =
                           ieee80211_ieee2mhz(i, IEEE80211_CHAN_5GHZ);
                       ic->ic_channels[i].ic_flags = IEEE80211_CHAN_A;
               }
               for (i = 149; i <= 165; i += 4) {
                       ic->ic_channels[i].ic_freq =
                           ieee80211_ieee2mhz(i, IEEE80211_CHAN_5GHZ);
                       ic->ic_channels[i].ic_flags = IEEE80211_CHAN_A;
               }
       }

       /* set supported .11b and .11g rates */
       ic->ic_sup_rates[IEEE80211_MODE_11B] = ieee80211_std_rateset_11b;
       ic->ic_sup_rates[IEEE80211_MODE_11G] = ieee80211_std_rateset_11g;

       /* set supported .11b and .11g channels (1 through 14) */
       for (i = 1; i <= 14; i++) {
               ic->ic_channels[i].ic_freq =
                   ieee80211_ieee2mhz(i, IEEE80211_CHAN_2GHZ);
               ic->ic_channels[i].ic_flags =
                   IEEE80211_CHAN_CCK | IEEE80211_CHAN_OFDM |
                   IEEE80211_CHAN_DYN | IEEE80211_CHAN_2GHZ;
       }

       ifp->if_softc = sc;
       ifp->if_flags = IFF_BROADCAST | IFF_SIMPLEX | IFF_MULTICAST;
       ifp->if_init = iwi_init;
       ifp->if_stop = iwi_stop;
       ifp->if_ioctl = iwi_ioctl;
       ifp->if_start = iwi_start;
       ifp->if_watchdog = iwi_watchdog;
       IFQ_SET_READY(&ifp->if_snd);
       memcpy(ifp->if_xname, device_xname(self), IFNAMSIZ);

       if_initialize(ifp);
       ieee80211_ifattach(ic);
       /* Use common softint-based if_input */
       ifp->if_percpuq = if_percpuq_create(ifp);
       if_register(ifp);

       /* override default methods */
       ic->ic_node_alloc = iwi_node_alloc;
       sc->sc_node_free = ic->ic_node_free;
       ic->ic_node_free = iwi_node_free;
       /* override state transition machine */
       sc->sc_newstate = ic->ic_newstate;
       ic->ic_newstate = iwi_newstate;

       /* XXX media locking needs revisiting */
       mutex_init(&sc->sc_media_mtx, MUTEX_DEFAULT, IPL_SOFTNET);
       ieee80211_media_init_with_lock(ic,
           iwi_media_change, iwi_media_status, &sc->sc_media_mtx);

       /*
        * Allocate rings.
        */
       if (iwi_alloc_cmd_ring(sc, &sc->cmdq, IWI_CMD_RING_COUNT) != 0) {
               aprint_error_dev(self, "could not allocate command ring\n");
               goto fail;
       }

       error = iwi_alloc_tx_ring(sc, &sc->txq[0], IWI_TX_RING_COUNT,
           IWI_CSR_TX1_RIDX, IWI_CSR_TX1_WIDX);
       if (error != 0) {
               aprint_error_dev(self, "could not allocate Tx ring 1\n");
               goto fail;
       }

       error = iwi_alloc_tx_ring(sc, &sc->txq[1], IWI_TX_RING_COUNT,
           IWI_CSR_TX2_RIDX, IWI_CSR_TX2_WIDX);
       if (error != 0) {
               aprint_error_dev(self, "could not allocate Tx ring 2\n");
               goto fail;
       }

       error = iwi_alloc_tx_ring(sc, &sc->txq[2], IWI_TX_RING_COUNT,
           IWI_CSR_TX3_RIDX, IWI_CSR_TX3_WIDX);
       if (error != 0) {
               aprint_error_dev(self, "could not allocate Tx ring 3\n");
               goto fail;
       }

       error = iwi_alloc_tx_ring(sc, &sc->txq[3], IWI_TX_RING_COUNT,
           IWI_CSR_TX4_RIDX, IWI_CSR_TX4_WIDX);
       if (error != 0) {
               aprint_error_dev(self, "could not allocate Tx ring 4\n");
               goto fail;
       }

       if (iwi_alloc_rx_ring(sc, &sc->rxq, IWI_RX_RING_COUNT) != 0) {
               aprint_error_dev(self, "could not allocate Rx ring\n");
               goto fail;
       }

       bpf_attach2(ifp, DLT_IEEE802_11_RADIO,
           sizeof(struct ieee80211_frame) + 64, &sc->sc_drvbpf);

       sc->sc_rxtap_len = sizeof sc->sc_rxtapu;
       sc->sc_rxtap.wr_ihdr.it_len = htole16(sc->sc_rxtap_len);
       sc->sc_rxtap.wr_ihdr.it_present = htole32(IWI_RX_RADIOTAP_PRESENT);

       sc->sc_txtap_len = sizeof sc->sc_txtapu;
       sc->sc_txtap.wt_ihdr.it_len = htole16(sc->sc_txtap_len);
       sc->sc_txtap.wt_ihdr.it_present = htole32(IWI_TX_RADIOTAP_PRESENT);

       iwi_sysctlattach(sc);

       if (pmf_device_register(self, NULL, NULL))
               pmf_class_network_register(self, ifp);
       else
               aprint_error_dev(self, "couldn't establish power handler\n");

       ieee80211_announce(ic);

       return;

fail:   iwi_detach(self, 0);
}

static int
iwi_detach(device_t self, int flags)
{
       struct iwi_softc *sc = device_private(self);
       struct ifnet *ifp = &sc->sc_if;

       if (ifp->if_softc != NULL) {
               pmf_device_deregister(self);
               iwi_stop(ifp, 1);
               iwi_free_firmware(sc);
               ieee80211_ifdetach(&sc->sc_ic);
               if_detach(ifp);
       }

       iwi_free_cmd_ring(sc, &sc->cmdq);
       iwi_free_tx_ring(sc, &sc->txq[0]);
       iwi_free_tx_ring(sc, &sc->txq[1]);
       iwi_free_tx_ring(sc, &sc->txq[2]);
       iwi_free_tx_ring(sc, &sc->txq[3]);
       iwi_free_rx_ring(sc, &sc->rxq);

       if (sc->sc_ih != NULL) {
               pci_intr_disestablish(sc->sc_pct, sc->sc_ih);
               sc->sc_ih = NULL;
       }

       if (sc->sc_soft_ih != NULL) {
               softint_disestablish(sc->sc_soft_ih);
               sc->sc_soft_ih = NULL;
       }

       bus_space_unmap(sc->sc_st, sc->sc_sh, sc->sc_sz);

       return 0;
}

static int
iwi_alloc_cmd_ring(struct iwi_softc *sc, struct iwi_cmd_ring *ring,
   int count)
{
       int error, nsegs;

       ring->count = count;
       ring->queued = 0;
       ring->cur = ring->next = 0;

       /*
        * Allocate and map command ring
        */
       error = bus_dmamap_create(sc->sc_dmat,
           IWI_CMD_DESC_SIZE * count, 1,
           IWI_CMD_DESC_SIZE * count, 0,
           BUS_DMA_NOWAIT, &ring->desc_map);
       if (error != 0) {
               aprint_error_dev(sc->sc_dev,
                   "could not create command ring DMA map\n");
               ring->desc_map = NULL;
               goto fail;
       }

       error = bus_dmamem_alloc(sc->sc_dmat,
           IWI_CMD_DESC_SIZE * count, PAGE_SIZE, 0,
           &sc->cmdq.desc_seg, 1, &nsegs, BUS_DMA_NOWAIT);
       if (error != 0) {
               aprint_error_dev(sc->sc_dev,
                   "could not allocate command ring DMA memory\n");
               goto fail;
       }

       error = bus_dmamem_map(sc->sc_dmat, &sc->cmdq.desc_seg, nsegs,
           IWI_CMD_DESC_SIZE * count,
           (void **)&sc->cmdq.desc, BUS_DMA_NOWAIT);
       if (error != 0) {
               aprint_error_dev(sc->sc_dev,
                   "could not map command ring DMA memory\n");
               goto fail;
       }

       error = bus_dmamap_load(sc->sc_dmat, sc->cmdq.desc_map, sc->cmdq.desc,
           IWI_CMD_DESC_SIZE * count, NULL,
           BUS_DMA_NOWAIT);
       if (error != 0) {
               aprint_error_dev(sc->sc_dev,
                   "could not load command ring DMA map\n");
               goto fail;
       }

       memset(sc->cmdq.desc, 0,
           IWI_CMD_DESC_SIZE * count);

       return 0;

fail:   return error;
}

static void
iwi_reset_cmd_ring(struct iwi_softc *sc, struct iwi_cmd_ring *ring)
{
       int i;

       for (i = ring->next; i != ring->cur;) {
               bus_dmamap_sync(sc->sc_dmat, sc->cmdq.desc_map,
                   i * IWI_CMD_DESC_SIZE, IWI_CMD_DESC_SIZE,
                   BUS_DMASYNC_POSTWRITE);

               wakeup(&ring->desc[i]);
               i = (i + 1) % ring->count;
       }

       ring->queued = 0;
       ring->cur = ring->next = 0;
}

static void
iwi_free_cmd_ring(struct iwi_softc *sc, struct iwi_cmd_ring *ring)
{
       if (ring->desc_map != NULL) {
               if (ring->desc != NULL) {
                       bus_dmamap_unload(sc->sc_dmat, ring->desc_map);
                       bus_dmamem_unmap(sc->sc_dmat, (void *)ring->desc,
                           IWI_CMD_DESC_SIZE * ring->count);
                       bus_dmamem_free(sc->sc_dmat, &ring->desc_seg, 1);
               }
               bus_dmamap_destroy(sc->sc_dmat, ring->desc_map);
       }
}

static int
iwi_alloc_tx_ring(struct iwi_softc *sc, struct iwi_tx_ring *ring,
   int count, bus_size_t csr_ridx, bus_size_t csr_widx)
{
       int i, error, nsegs;

       ring->count  = 0;
       ring->queued = 0;
       ring->cur = ring->next = 0;
       ring->csr_ridx = csr_ridx;
       ring->csr_widx = csr_widx;

       /*
        * Allocate and map Tx ring
        */
       error = bus_dmamap_create(sc->sc_dmat,
           IWI_TX_DESC_SIZE * count, 1,
           IWI_TX_DESC_SIZE * count, 0, BUS_DMA_NOWAIT,
           &ring->desc_map);
       if (error != 0) {
               aprint_error_dev(sc->sc_dev,
                   "could not create tx ring DMA map\n");
               ring->desc_map = NULL;
               goto fail;
       }

       error = bus_dmamem_alloc(sc->sc_dmat,
           IWI_TX_DESC_SIZE * count, PAGE_SIZE, 0,
           &ring->desc_seg, 1, &nsegs, BUS_DMA_NOWAIT);
       if (error != 0) {
               aprint_error_dev(sc->sc_dev,
                   "could not allocate tx ring DMA memory\n");
               goto fail;
       }

       error = bus_dmamem_map(sc->sc_dmat, &ring->desc_seg, nsegs,
           IWI_TX_DESC_SIZE * count,
           (void **)&ring->desc, BUS_DMA_NOWAIT);
       if (error != 0) {
               aprint_error_dev(sc->sc_dev,
                   "could not map tx ring DMA memory\n");
               goto fail;
       }

       error = bus_dmamap_load(sc->sc_dmat, ring->desc_map, ring->desc,
           IWI_TX_DESC_SIZE * count, NULL,
           BUS_DMA_NOWAIT);
       if (error != 0) {
               aprint_error_dev(sc->sc_dev,
                   "could not load tx ring DMA map\n");
               goto fail;
       }

       memset(ring->desc, 0, IWI_TX_DESC_SIZE * count);

       ring->data = malloc(count * sizeof (struct iwi_tx_data), M_DEVBUF,
           M_WAITOK | M_ZERO);
       ring->count = count;

       /*
        * Allocate Tx buffers DMA maps
        */
       for (i = 0; i < count; i++) {
               error = bus_dmamap_create(sc->sc_dmat, MCLBYTES, IWI_MAX_NSEG,
                   MCLBYTES, 0, BUS_DMA_NOWAIT, &ring->data[i].map);
               if (error != 0) {
                       aprint_error_dev(sc->sc_dev,
                           "could not create tx buf DMA map");
                       ring->data[i].map = NULL;
                       goto fail;
               }
       }
       return 0;

fail:   return error;
}

static void
iwi_reset_tx_ring(struct iwi_softc *sc, struct iwi_tx_ring *ring)
{
       struct iwi_tx_data *data;
       int i;

       for (i = 0; i < ring->count; i++) {
               data = &ring->data[i];

               if (data->map != NULL) {
                       bus_dmamap_sync(sc->sc_dmat, data->map, 0,
                           data->map->dm_mapsize, BUS_DMASYNC_POSTWRITE);
                       bus_dmamap_unload(sc->sc_dmat, data->map);
               }

               m_freem(data->m);
               data->m = NULL;

               if (data->ni != NULL) {
                       ieee80211_free_node(data->ni);
                       data->ni = NULL;
               }
       }

       ring->queued = 0;
       ring->cur = ring->next = 0;
}

static void
iwi_free_tx_ring(struct iwi_softc *sc, struct iwi_tx_ring *ring)
{
       int i;
       struct iwi_tx_data *data;

       if (ring->desc_map != NULL) {
               if (ring->desc != NULL) {
                       bus_dmamap_unload(sc->sc_dmat, ring->desc_map);
                       bus_dmamem_unmap(sc->sc_dmat, (void *)ring->desc,
                           IWI_TX_DESC_SIZE * ring->count);
                       bus_dmamem_free(sc->sc_dmat, &ring->desc_seg, 1);
               }
               bus_dmamap_destroy(sc->sc_dmat, ring->desc_map);
       }

       for (i = 0; i < ring->count; i++) {
               data = &ring->data[i];

               if (data->map != NULL) {
                       bus_dmamap_unload(sc->sc_dmat, data->map);
                       bus_dmamap_destroy(sc->sc_dmat, data->map);
               }

               m_freem(data->m);
       }
}

static int
iwi_alloc_rx_ring(struct iwi_softc *sc, struct iwi_rx_ring *ring, int count)
{
       int i, error;

       ring->count = 0;
       ring->cur = 0;

       ring->data = malloc(count * sizeof (struct iwi_rx_data), M_DEVBUF,
           M_WAITOK | M_ZERO);
       ring->count = count;

       /*
        * Allocate and map Rx buffers
        */
       for (i = 0; i < count; i++) {

               error = bus_dmamap_create(sc->sc_dmat, MCLBYTES, 1, MCLBYTES,
                   0, BUS_DMA_WAITOK | BUS_DMA_ALLOCNOW, &ring->data[i].map);
               if (error != 0) {
                       aprint_error_dev(sc->sc_dev,
                           "could not create rx buf DMA map");
                       ring->data[i].map = NULL;
                       goto fail;
               }

               if ((ring->data[i].m = iwi_alloc_rx_buf(sc)) == NULL) {
                       error = ENOMEM;
                       goto fail;
               }

               error = bus_dmamap_load_mbuf(sc->sc_dmat, ring->data[i].map,
                   ring->data[i].m, BUS_DMA_READ | BUS_DMA_NOWAIT);
               if (error != 0) {
                       aprint_error_dev(sc->sc_dev,
                           "could not load rx buffer DMA map\n");
                       goto fail;
               }

               bus_dmamap_sync(sc->sc_dmat, ring->data[i].map, 0,
                   ring->data[i].map->dm_mapsize, BUS_DMASYNC_PREREAD);
       }

       return 0;

fail:   return error;
}

static void
iwi_reset_rx_ring(struct iwi_softc *sc, struct iwi_rx_ring *ring)
{
       ring->cur = 0;
}

static void
iwi_free_rx_ring(struct iwi_softc *sc, struct iwi_rx_ring *ring)
{
       int i;
       struct iwi_rx_data *data;

       for (i = 0; i < ring->count; i++) {
               data = &ring->data[i];

               if (data->map != NULL) {
                       bus_dmamap_unload(sc->sc_dmat, data->map);
                       bus_dmamap_destroy(sc->sc_dmat, data->map);
               }

               m_freem(data->m);
       }
}

static struct ieee80211_node *
iwi_node_alloc(struct ieee80211_node_table *nt)
{
       struct iwi_node *in;

       in = malloc(sizeof (struct iwi_node), M_80211_NODE, M_NOWAIT | M_ZERO);
       if (in == NULL)
               return NULL;

       in->in_station = -1;

       return &in->in_node;
}

static int
iwi_alloc_unr(struct iwi_softc *sc)
{
       int i;

       for (i = 0; i < IWI_MAX_IBSSNODE - 1; i++)
               if ((sc->sc_unr & (1 << i)) == 0) {
                       sc->sc_unr |= 1 << i;
                       return i;
               }

       return -1;
}

static void
iwi_free_unr(struct iwi_softc *sc, int r)
{

       sc->sc_unr &= 1 << r;
}

static void
iwi_node_free(struct ieee80211_node *ni)
{
       struct ieee80211com *ic = ni->ni_ic;
       struct iwi_softc *sc = ic->ic_ifp->if_softc;
       struct iwi_node *in = (struct iwi_node *)ni;

       if (in->in_station != -1)
               iwi_free_unr(sc, in->in_station);

       sc->sc_node_free(ni);
}

static int
iwi_media_change(struct ifnet *ifp)
{
       int error;

       error = ieee80211_media_change(ifp);
       if (error != ENETRESET)
               return error;

       if ((ifp->if_flags & (IFF_UP | IFF_RUNNING)) == (IFF_UP | IFF_RUNNING))
               iwi_init(ifp);

       return 0;
}

/*
* Convert h/w rate code to IEEE rate code.
*/
static int
iwi_cvtrate(int iwirate)
{
       switch (iwirate) {
       case IWI_RATE_DS1:      return 2;
       case IWI_RATE_DS2:      return 4;
       case IWI_RATE_DS5:      return 11;
       case IWI_RATE_DS11:     return 22;
       case IWI_RATE_OFDM6:    return 12;
       case IWI_RATE_OFDM9:    return 18;
       case IWI_RATE_OFDM12:   return 24;
       case IWI_RATE_OFDM18:   return 36;
       case IWI_RATE_OFDM24:   return 48;
       case IWI_RATE_OFDM36:   return 72;
       case IWI_RATE_OFDM48:   return 96;
       case IWI_RATE_OFDM54:   return 108;
       }
       return 0;
}

/*
* The firmware automatically adapts the transmit speed.  We report its current
* value here.
*/
static void
iwi_media_status(struct ifnet *ifp, struct ifmediareq *imr)
{
       struct iwi_softc *sc = ifp->if_softc;
       struct ieee80211com *ic = &sc->sc_ic;
       int rate;

       imr->ifm_status = IFM_AVALID;
       imr->ifm_active = IFM_IEEE80211;
       if (ic->ic_state == IEEE80211_S_RUN)
               imr->ifm_status |= IFM_ACTIVE;

       /* read current transmission rate from adapter */
       rate = iwi_cvtrate(CSR_READ_4(sc, IWI_CSR_CURRENT_TX_RATE));
       imr->ifm_active |= ieee80211_rate2media(ic, rate, ic->ic_curmode);

       switch (ic->ic_opmode) {
       case IEEE80211_M_STA:
               break;

       case IEEE80211_M_IBSS:
               imr->ifm_active |= IFM_IEEE80211_ADHOC;
               break;

       case IEEE80211_M_MONITOR:
               imr->ifm_active |= IFM_IEEE80211_MONITOR;
               break;

       case IEEE80211_M_AHDEMO:
       case IEEE80211_M_HOSTAP:
               /* should not get there */
               break;
       }
}

static int
iwi_newstate(struct ieee80211com *ic, enum ieee80211_state nstate, int arg)
{
       struct iwi_softc *sc = ic->ic_ifp->if_softc;

       DPRINTF(("%s: %s -> %s flags 0x%x\n", __func__,
           ieee80211_state_name[ic->ic_state],
           ieee80211_state_name[nstate], sc->flags));

       switch (nstate) {
       case IEEE80211_S_SCAN:
               if (sc->flags & IWI_FLAG_SCANNING)
                       break;

               ieee80211_node_table_reset(&ic->ic_scan);
               ic->ic_flags |= IEEE80211_F_SCAN | IEEE80211_F_ASCAN;
               sc->flags |= IWI_FLAG_SCANNING;
               /* blink the led while scanning */
               iwi_led_set(sc, IWI_LED_ASSOCIATED, 1);
               iwi_scan(sc);
               break;

       case IEEE80211_S_AUTH:
               iwi_auth_and_assoc(sc);
               break;

       case IEEE80211_S_RUN:
               if (ic->ic_opmode == IEEE80211_M_IBSS &&
                   ic->ic_state == IEEE80211_S_SCAN)
                       iwi_auth_and_assoc(sc);
               else if (ic->ic_opmode == IEEE80211_M_MONITOR)
                       iwi_set_chan(sc, ic->ic_ibss_chan);
               break;
       case IEEE80211_S_ASSOC:
               iwi_led_set(sc, IWI_LED_ASSOCIATED, 0);
               if (ic->ic_state == IEEE80211_S_AUTH)
                       break;
               iwi_auth_and_assoc(sc);
               break;

       case IEEE80211_S_INIT:
               sc->flags &= ~IWI_FLAG_SCANNING;
               break;
       }

       return sc->sc_newstate(ic, nstate, arg);
}

/*
* WME parameters coming from IEEE 802.11e specification.  These values are
* already declared in ieee80211_proto.c, but they are static so they can't
* be reused here.
*/
static const struct wmeParams iwi_wme_cck_params[WME_NUM_AC] = {
       { 0, 3, 5,  7,   0, 0, },       /* WME_AC_BE */
       { 0, 3, 5, 10,   0, 0, },       /* WME_AC_BK */
       { 0, 2, 4,  5, 188, 0, },       /* WME_AC_VI */
       { 0, 2, 3,  4, 102, 0, },       /* WME_AC_VO */
};

static const struct wmeParams iwi_wme_ofdm_params[WME_NUM_AC] = {
       { 0, 3, 4,  6,   0, 0, },       /* WME_AC_BE */
       { 0, 3, 4, 10,   0, 0, },       /* WME_AC_BK */
       { 0, 2, 3,  4,  94, 0, },       /* WME_AC_VI */
       { 0, 2, 2,  3,  47, 0, },       /* WME_AC_VO */
};

static int
iwi_wme_update(struct ieee80211com *ic)
{
#define IWI_EXP2(v)     htole16((1 << (v)) - 1)
#define IWI_USEC(v)     htole16(IEEE80211_TXOP_TO_US(v))
       struct iwi_softc *sc = ic->ic_ifp->if_softc;
       struct iwi_wme_params wme[3];
       const struct wmeParams *wmep;
       int ac;

       /*
        * We shall not override firmware default WME values if WME is not
        * actually enabled.
        */
       if (!(ic->ic_flags & IEEE80211_F_WME))
               return 0;

       for (ac = 0; ac < WME_NUM_AC; ac++) {
               /* set WME values for current operating mode */
               wmep = &ic->ic_wme.wme_chanParams.cap_wmeParams[ac];
               wme[0].aifsn[ac] = wmep->wmep_aifsn;
               wme[0].cwmin[ac] = IWI_EXP2(wmep->wmep_logcwmin);
               wme[0].cwmax[ac] = IWI_EXP2(wmep->wmep_logcwmax);
               wme[0].burst[ac] = IWI_USEC(wmep->wmep_txopLimit);
               wme[0].acm[ac]   = wmep->wmep_acm;

               /* set WME values for CCK modulation */
               wmep = &iwi_wme_cck_params[ac];
               wme[1].aifsn[ac] = wmep->wmep_aifsn;
               wme[1].cwmin[ac] = IWI_EXP2(wmep->wmep_logcwmin);
               wme[1].cwmax[ac] = IWI_EXP2(wmep->wmep_logcwmax);
               wme[1].burst[ac] = IWI_USEC(wmep->wmep_txopLimit);
               wme[1].acm[ac]   = wmep->wmep_acm;

               /* set WME values for OFDM modulation */
               wmep = &iwi_wme_ofdm_params[ac];
               wme[2].aifsn[ac] = wmep->wmep_aifsn;
               wme[2].cwmin[ac] = IWI_EXP2(wmep->wmep_logcwmin);
               wme[2].cwmax[ac] = IWI_EXP2(wmep->wmep_logcwmax);
               wme[2].burst[ac] = IWI_USEC(wmep->wmep_txopLimit);
               wme[2].acm[ac]   = wmep->wmep_acm;
       }

       DPRINTF(("Setting WME parameters\n"));
       return iwi_cmd(sc, IWI_CMD_SET_WME_PARAMS, wme, sizeof wme, 1);
#undef IWI_USEC
#undef IWI_EXP2
}

/*
* Read 16 bits at address 'addr' from the serial EEPROM.
*/
static uint16_t
iwi_read_prom_word(struct iwi_softc *sc, uint8_t addr)
{
       uint32_t tmp;
       uint16_t val;
       int n;

       /* Clock C once before the first command */
       IWI_EEPROM_CTL(sc, 0);
       IWI_EEPROM_CTL(sc, IWI_EEPROM_S);
       IWI_EEPROM_CTL(sc, IWI_EEPROM_S | IWI_EEPROM_C);
       IWI_EEPROM_CTL(sc, IWI_EEPROM_S);

       /* Write start bit (1) */
       IWI_EEPROM_CTL(sc, IWI_EEPROM_S | IWI_EEPROM_D);
       IWI_EEPROM_CTL(sc, IWI_EEPROM_S | IWI_EEPROM_D | IWI_EEPROM_C);

       /* Write READ opcode (10) */
       IWI_EEPROM_CTL(sc, IWI_EEPROM_S | IWI_EEPROM_D);
       IWI_EEPROM_CTL(sc, IWI_EEPROM_S | IWI_EEPROM_D | IWI_EEPROM_C);
       IWI_EEPROM_CTL(sc, IWI_EEPROM_S);
       IWI_EEPROM_CTL(sc, IWI_EEPROM_S | IWI_EEPROM_C);

       /* Write address A7-A0 */
       for (n = 7; n >= 0; n--) {
               IWI_EEPROM_CTL(sc, IWI_EEPROM_S |
                   (((addr >> n) & 1) << IWI_EEPROM_SHIFT_D));
               IWI_EEPROM_CTL(sc, IWI_EEPROM_S |
                   (((addr >> n) & 1) << IWI_EEPROM_SHIFT_D) | IWI_EEPROM_C);
       }

       IWI_EEPROM_CTL(sc, IWI_EEPROM_S);

       /* Read data Q15-Q0 */
       val = 0;
       for (n = 15; n >= 0; n--) {
               IWI_EEPROM_CTL(sc, IWI_EEPROM_S | IWI_EEPROM_C);
               IWI_EEPROM_CTL(sc, IWI_EEPROM_S);
               tmp = MEM_READ_4(sc, IWI_MEM_EEPROM_CTL);
               val |= ((tmp & IWI_EEPROM_Q) >> IWI_EEPROM_SHIFT_Q) << n;
       }

       IWI_EEPROM_CTL(sc, 0);

       /* Clear Chip Select and clock C */
       IWI_EEPROM_CTL(sc, IWI_EEPROM_S);
       IWI_EEPROM_CTL(sc, 0);
       IWI_EEPROM_CTL(sc, IWI_EEPROM_C);

       return val;
}

/*
* XXX: Hack to set the current channel to the value advertised in beacons or
* probe responses. Only used during AP detection.
*/
static void
iwi_fix_channel(struct ieee80211com *ic, struct mbuf *m)
{
       struct ieee80211_frame *wh;
       uint8_t subtype;
       uint8_t *frm, *efrm;

       wh = mtod(m, struct ieee80211_frame *);

       if ((wh->i_fc[0] & IEEE80211_FC0_TYPE_MASK) != IEEE80211_FC0_TYPE_MGT)
               return;

       subtype = wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_MASK;

       if (subtype != IEEE80211_FC0_SUBTYPE_BEACON &&
           subtype != IEEE80211_FC0_SUBTYPE_PROBE_RESP)
               return;

       frm = (uint8_t *)(wh + 1);
       efrm = mtod(m, uint8_t *) + m->m_len;

       frm += 12;      /* skip tstamp, bintval and capinfo fields */
       while (frm + 2 < efrm) {
               if (*frm == IEEE80211_ELEMID_DSPARMS) {
#if IEEE80211_CHAN_MAX < 255
                       if (frm[2] <= IEEE80211_CHAN_MAX)
#endif
                               ic->ic_curchan = &ic->ic_channels[frm[2]];
               }

               frm += frm[1] + 2;
       }
}

static struct mbuf *
iwi_alloc_rx_buf(struct iwi_softc *sc)
{
       struct mbuf *m;

       MGETHDR(m, M_DONTWAIT, MT_DATA);
       if (m == NULL) {
               aprint_error_dev(sc->sc_dev, "could not allocate rx mbuf\n");
               return NULL;
       }

       MCLGET(m, M_DONTWAIT);
       if (!(m->m_flags & M_EXT)) {
               aprint_error_dev(sc->sc_dev,
                   "could not allocate rx mbuf cluster\n");
               m_freem(m);
               return NULL;
       }

       m->m_pkthdr.len = m->m_len = m->m_ext.ext_size;
       return m;
}

static void
iwi_frame_intr(struct iwi_softc *sc, struct iwi_rx_data *data, int i,
   struct iwi_frame *frame)
{
       struct ieee80211com *ic = &sc->sc_ic;
       struct ifnet *ifp = ic->ic_ifp;
       struct mbuf *m, *m_new;
       struct ieee80211_frame *wh;
       struct ieee80211_node *ni;
       int error, s;

       DPRINTFN(5, ("received frame len=%u chan=%u rssi=%u\n",
           le16toh(frame->len), frame->chan, frame->rssi_dbm));

       if (le16toh(frame->len) < sizeof (struct ieee80211_frame) ||
           le16toh(frame->len) > MCLBYTES) {
               DPRINTF(("%s: bad frame length\n", device_xname(sc->sc_dev)));
               if_statinc(ifp, if_ierrors);
               return;
       }

       /*
        * Try to allocate a new mbuf for this ring element and
        * load it before processing the current mbuf. If the ring
        * element cannot be reloaded, drop the received packet
        * and reuse the old mbuf. In the unlikely case that
        * the old mbuf can't be reloaded either, explicitly panic.
        *
        * XXX Reorganize buffer by moving elements from the logical
        * end of the ring to the front instead of dropping.
        */
       if ((m_new = iwi_alloc_rx_buf(sc)) == NULL) {
               if_statinc(ifp, if_ierrors);
               return;
       }

       bus_dmamap_unload(sc->sc_dmat, data->map);

       error = bus_dmamap_load_mbuf(sc->sc_dmat, data->map, m_new,
           BUS_DMA_READ | BUS_DMA_NOWAIT);
       if (error != 0) {
               aprint_error_dev(sc->sc_dev,
                   "could not load rx buf DMA map\n");
               m_freem(m_new);
               if_statinc(ifp, if_ierrors);
               error = bus_dmamap_load_mbuf(sc->sc_dmat, data->map,
                   data->m, BUS_DMA_READ | BUS_DMA_NOWAIT);
               if (error)
                       panic("%s: unable to remap rx buf",
                           device_xname(sc->sc_dev));
               return;
       }

       /*
        * New mbuf successfully loaded, update RX ring and continue
        * processing.
        */
       m = data->m;
       data->m = m_new;
       CSR_WRITE_4(sc, IWI_CSR_RX_BASE + i * 4, data->map->dm_segs[0].ds_addr);

       /* Finalize mbuf */
       m_set_rcvif(m, ifp);
       m->m_pkthdr.len = m->m_len = sizeof (struct iwi_hdr) +
           sizeof (struct iwi_frame) + le16toh(frame->len);

       m_adj(m, sizeof (struct iwi_hdr) + sizeof (struct iwi_frame));

       s = splnet();

       if (ic->ic_state == IEEE80211_S_SCAN)
               iwi_fix_channel(ic, m);

       if (sc->sc_drvbpf != NULL) {
               struct iwi_rx_radiotap_header *tap = &sc->sc_rxtap;

               tap->wr_flags = 0;
               tap->wr_rate = iwi_cvtrate(frame->rate);
               tap->wr_chan_freq =
                   htole16(ic->ic_channels[frame->chan].ic_freq);
               tap->wr_chan_flags =
                   htole16(ic->ic_channels[frame->chan].ic_flags);
               tap->wr_antsignal = frame->signal;
               tap->wr_antenna = frame->antenna;

               bpf_mtap2(sc->sc_drvbpf, tap, sc->sc_rxtap_len, m, BPF_D_IN);
       }
       wh = mtod(m, struct ieee80211_frame *);
       ni = ieee80211_find_rxnode(ic, (struct ieee80211_frame_min *)wh);

       /* Send the frame to the upper layer */
       ieee80211_input(ic, m, ni, frame->rssi_dbm, 0);

       /* node is no longer needed */
       ieee80211_free_node(ni);

       splx(s);
}

static void
iwi_notification_intr(struct iwi_softc *sc, struct iwi_notif *notif)
{
       struct ieee80211com *ic = &sc->sc_ic;
       struct iwi_notif_authentication *auth;
       struct iwi_notif_association *assoc;
       struct iwi_notif_beacon_state *beacon;
       int s;

       switch (notif->type) {
       case IWI_NOTIF_TYPE_SCAN_CHANNEL:
#ifdef IWI_DEBUG
               {
                       struct iwi_notif_scan_channel *chan =
                           (struct iwi_notif_scan_channel *)(notif + 1);

                       DPRINTFN(2, ("Scan of channel %u complete (%u)\n",
                           ic->ic_channels[chan->nchan].ic_freq, chan->nchan));
               }
#endif
               break;

       case IWI_NOTIF_TYPE_SCAN_COMPLETE:
#ifdef IWI_DEBUG
               {
                       struct iwi_notif_scan_complete *scan =
                           (struct iwi_notif_scan_complete *)(notif + 1);

                       DPRINTFN(2, ("Scan completed (%u, %u)\n", scan->nchan,
                           scan->status));
               }
#endif

               /* monitor mode uses scan to set the channel ... */
               s = splnet();
               if (ic->ic_opmode != IEEE80211_M_MONITOR) {
                       sc->flags &= ~IWI_FLAG_SCANNING;
                       ieee80211_end_scan(ic);
               } else
                       iwi_set_chan(sc, ic->ic_ibss_chan);
               splx(s);
               break;

       case IWI_NOTIF_TYPE_AUTHENTICATION:
               auth = (struct iwi_notif_authentication *)(notif + 1);

               DPRINTFN(2, ("Authentication (%u)\n", auth->state));

               switch (auth->state) {
               case IWI_AUTH_SUCCESS:
                       s = splnet();
                       ieee80211_node_authorize(ic->ic_bss);
                       ieee80211_new_state(ic, IEEE80211_S_ASSOC, -1);
                       splx(s);
                       break;

               case IWI_AUTH_FAIL:
                       break;

               case IWI_AUTH_SENT_1:
               case IWI_AUTH_RECV_2:
               case IWI_AUTH_SEQ1_PASS:
                       break;

               case IWI_AUTH_SEQ1_FAIL:
                       break;

               default:
                       aprint_error_dev(sc->sc_dev,
                           "unknown authentication state %u\n", auth->state);
               }
               break;

       case IWI_NOTIF_TYPE_ASSOCIATION:
               assoc = (struct iwi_notif_association *)(notif + 1);

               DPRINTFN(2, ("Association (%u, %u)\n", assoc->state,
                   assoc->status));

               switch (assoc->state) {
               case IWI_AUTH_SUCCESS:
                       /* re-association, do nothing */
                       break;

               case IWI_ASSOC_SUCCESS:
                       s = splnet();
                       ieee80211_new_state(ic, IEEE80211_S_RUN, -1);
                       splx(s);
                       break;

               case IWI_ASSOC_FAIL:
                       s = splnet();
                       ieee80211_begin_scan(ic, 1);
                       splx(s);
                       break;

               default:
                       aprint_error_dev(sc->sc_dev,
                           "unknown association state %u\n", assoc->state);
               }
               break;

       case IWI_NOTIF_TYPE_BEACON:
               beacon = (struct iwi_notif_beacon_state *)(notif + 1);

               if (beacon->state == IWI_BEACON_MISS) {
                       DPRINTFN(5, ("%s: %u beacon(s) missed\n",
                           device_xname(sc->sc_dev), le32toh(beacon->number)));
               }
               break;

       case IWI_NOTIF_TYPE_FRAG_LENGTH:
       case IWI_NOTIF_TYPE_LINK_QUALITY:
       case IWI_NOTIF_TYPE_TGI_TX_KEY:
       case IWI_NOTIF_TYPE_CALIBRATION:
       case IWI_NOTIF_TYPE_NOISE:
               DPRINTFN(5, ("Notification (%u)\n", notif->type));
               break;

       default:
               DPRINTF(("%s: unknown notification type %u flags 0x%x len %d\n",
                   device_xname(sc->sc_dev), notif->type, notif->flags,
                   le16toh(notif->len)));
       }
}

static void
iwi_cmd_intr(struct iwi_softc *sc)
{

       (void)CSR_READ_4(sc, IWI_CSR_CMD_RIDX);

       bus_dmamap_sync(sc->sc_dmat, sc->cmdq.desc_map,
           sc->cmdq.next * IWI_CMD_DESC_SIZE, IWI_CMD_DESC_SIZE,
           BUS_DMASYNC_POSTWRITE);

       wakeup(&sc->cmdq.desc[sc->cmdq.next]);

       sc->cmdq.next = (sc->cmdq.next + 1) % sc->cmdq.count;

       if (--sc->cmdq.queued > 0) {
               CSR_WRITE_4(sc, IWI_CSR_CMD_WIDX,
                   (sc->cmdq.next + 1) % sc->cmdq.count);
       }
}

static void
iwi_rx_intr(struct iwi_softc *sc)
{
       struct iwi_rx_data *data;
       struct iwi_hdr *hdr;
       uint32_t hw;

       hw = CSR_READ_4(sc, IWI_CSR_RX_RIDX);

       for (; sc->rxq.cur != hw;) {
               data = &sc->rxq.data[sc->rxq.cur];

               bus_dmamap_sync(sc->sc_dmat, data->map, 0,
                   data->map->dm_mapsize, BUS_DMASYNC_POSTREAD);

               hdr = mtod(data->m, struct iwi_hdr *);

               switch (hdr->type) {
               case IWI_HDR_TYPE_FRAME:
                       iwi_frame_intr(sc, data, sc->rxq.cur,
                           (struct iwi_frame *)(hdr + 1));
                       break;

               case IWI_HDR_TYPE_NOTIF:
                       iwi_notification_intr(sc,
                           (struct iwi_notif *)(hdr + 1));
                       break;

               default:
                       aprint_error_dev(sc->sc_dev, "unknown hdr type %u\n",
                           hdr->type);
               }

               bus_dmamap_sync(sc->sc_dmat, data->map, 0,
                   data->map->dm_mapsize, BUS_DMASYNC_PREREAD);

               DPRINTFN(15, ("rx done idx=%u\n", sc->rxq.cur));

               sc->rxq.cur = (sc->rxq.cur + 1) % sc->rxq.count;
       }

       /* Tell the firmware what we have processed */
       hw = (hw == 0) ? sc->rxq.count - 1 : hw - 1;
       CSR_WRITE_4(sc, IWI_CSR_RX_WIDX, hw);
}

static void
iwi_tx_intr(struct iwi_softc *sc, struct iwi_tx_ring *txq)
{
       struct ifnet *ifp = &sc->sc_if;
       struct iwi_tx_data *data;
       uint32_t hw;
       int s;

       s = splnet();

       hw = CSR_READ_4(sc, txq->csr_ridx);

       for (; txq->next != hw;) {
               data = &txq->data[txq->next];

               bus_dmamap_sync(sc->sc_dmat, data->map, 0,
                   data->map->dm_mapsize, BUS_DMASYNC_POSTWRITE);
               bus_dmamap_unload(sc->sc_dmat, data->map);
               m_freem(data->m);
               data->m = NULL;
               ieee80211_free_node(data->ni);
               data->ni = NULL;

               DPRINTFN(15, ("tx done idx=%u\n", txq->next));

               if_statinc(ifp, if_opackets);

               txq->queued--;
               txq->next = (txq->next + 1) % txq->count;
       }

       sc->sc_tx_timer = 0;

       if (txq->queued < txq->count - 8 - 8 && (ifp->if_flags & IFF_OACTIVE)) {
               ifp->if_flags &= ~IFF_OACTIVE;

               /* Call start() since some buffer descriptors have been released */
               iwi_start(ifp); /* in softint */
       }

       splx(s);
}

static int
iwi_intr(void *arg)
{
       struct iwi_softc *sc = arg;
       uint32_t r;

       if ((r = CSR_READ_4(sc, IWI_CSR_INTR)) == 0 || r == 0xffffffff)
               return 0;

       /* Disable interrupts */
       CSR_WRITE_4(sc, IWI_CSR_INTR_MASK, 0);

       softint_schedule(sc->sc_soft_ih);
       return 1;
}

static void
iwi_softintr(void *arg)
{
       struct iwi_softc *sc = arg;
       uint32_t r;
       int s;

       if ((r = CSR_READ_4(sc, IWI_CSR_INTR)) == 0 || r == 0xffffffff)
               goto out;

       /* Acknowledge interrupts */
       CSR_WRITE_4(sc, IWI_CSR_INTR, r);

       if (r & IWI_INTR_FATAL_ERROR) {
               aprint_error_dev(sc->sc_dev, "fatal error\n");
               s = splnet();
               sc->sc_ic.ic_ifp->if_flags &= ~IFF_UP;
               iwi_stop(&sc->sc_if, 1);
               splx(s);
               return;
       }

       if (r & IWI_INTR_FW_INITED) {
               if (!(r & (IWI_INTR_FATAL_ERROR | IWI_INTR_PARITY_ERROR)))
                       wakeup(sc);
       }

       if (r & IWI_INTR_RADIO_OFF) {
               DPRINTF(("radio transmitter off\n"));
               s = splnet();
               sc->sc_ic.ic_ifp->if_flags &= ~IFF_UP;
               iwi_stop(&sc->sc_if, 1);
               splx(s);
               return;
       }

       if (r & IWI_INTR_CMD_DONE)
               iwi_cmd_intr(sc);

       if (r & IWI_INTR_TX1_DONE)
               iwi_tx_intr(sc, &sc->txq[0]);

       if (r & IWI_INTR_TX2_DONE)
               iwi_tx_intr(sc, &sc->txq[1]);

       if (r & IWI_INTR_TX3_DONE)
               iwi_tx_intr(sc, &sc->txq[2]);

       if (r & IWI_INTR_TX4_DONE)
               iwi_tx_intr(sc, &sc->txq[3]);

       if (r & IWI_INTR_RX_DONE)
               iwi_rx_intr(sc);

       if (r & IWI_INTR_PARITY_ERROR)
               aprint_error_dev(sc->sc_dev, "parity error\n");

out:
       /* Re-enable interrupts */
       CSR_WRITE_4(sc, IWI_CSR_INTR_MASK, IWI_INTR_MASK);
}

static int
iwi_cmd(struct iwi_softc *sc, uint8_t type, void *data, uint8_t len,
   int async)
{
       struct iwi_cmd_desc *desc;

       desc = &sc->cmdq.desc[sc->cmdq.cur];

       desc->hdr.type = IWI_HDR_TYPE_COMMAND;
       desc->hdr.flags = IWI_HDR_FLAG_IRQ;
       desc->type = type;
       desc->len = len;
       memcpy(desc->data, data, len);

       bus_dmamap_sync(sc->sc_dmat, sc->cmdq.desc_map,
           sc->cmdq.cur * IWI_CMD_DESC_SIZE,
           IWI_CMD_DESC_SIZE, BUS_DMASYNC_PREWRITE);

       DPRINTFN(2, ("sending command idx=%u type=%u len=%u async=%d\n",
           sc->cmdq.cur, type, len, async));

       sc->cmdq.cur = (sc->cmdq.cur + 1) % sc->cmdq.count;

       if (++sc->cmdq.queued == 1)
               CSR_WRITE_4(sc, IWI_CSR_CMD_WIDX, sc->cmdq.cur);

       return async ? 0 : tsleep(desc, 0, "iwicmd", hz);
}

static void
iwi_write_ibssnode(struct iwi_softc *sc, const struct iwi_node *in)
{
       struct iwi_ibssnode node;

       /* write node information into NIC memory */
       memset(&node, 0, sizeof node);
       IEEE80211_ADDR_COPY(node.bssid, in->in_node.ni_macaddr);

       CSR_WRITE_REGION_1(sc,
           IWI_CSR_NODE_BASE + in->in_station * sizeof node,
           (uint8_t *)&node, sizeof node);
}

static int
iwi_tx_start(struct ifnet *ifp, struct mbuf *m0, struct ieee80211_node *ni,
   int ac)
{
       struct iwi_softc *sc = ifp->if_softc;
       struct ieee80211com *ic = &sc->sc_ic;
       struct iwi_node *in = (struct iwi_node *)ni;
       struct ieee80211_frame *wh;
       struct ieee80211_key *k;
       const struct chanAccParams *cap;
       struct iwi_tx_ring *txq = &sc->txq[ac];
       struct iwi_tx_data *data;
       struct iwi_tx_desc *desc;
       struct mbuf *mnew;
       int error, hdrlen, i, noack = 0;

       wh = mtod(m0, struct ieee80211_frame *);

       if (wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_QOS) {
               hdrlen = sizeof (struct ieee80211_qosframe);
               cap = &ic->ic_wme.wme_chanParams;
               noack = cap->cap_wmeParams[ac].wmep_noackPolicy;
       } else
               hdrlen = sizeof (struct ieee80211_frame);

       /*
        * This is only used in IBSS mode where the firmware expect an index
        * in a h/w table instead of a destination address.
        */
       if (ic->ic_opmode == IEEE80211_M_IBSS && in->in_station == -1) {
               in->in_station = iwi_alloc_unr(sc);

               if (in->in_station == -1) {     /* h/w table is full */
                       m_freem(m0);
                       ieee80211_free_node(ni);
                       if_statinc(ifp, if_oerrors);
                       return 0;
               }
               iwi_write_ibssnode(sc, in);
       }

       if (wh->i_fc[1] & IEEE80211_FC1_WEP) {
               k = ieee80211_crypto_encap(ic, ni, m0);
               if (k == NULL) {
                       m_freem(m0);
                       return ENOBUFS;
               }

               /* packet header may have moved, reset our local pointer */
               wh = mtod(m0, struct ieee80211_frame *);
       }

       if (sc->sc_drvbpf != NULL) {
               struct iwi_tx_radiotap_header *tap = &sc->sc_txtap;

               tap->wt_flags = 0;
               tap->wt_chan_freq = htole16(ic->ic_ibss_chan->ic_freq);
               tap->wt_chan_flags = htole16(ic->ic_ibss_chan->ic_flags);

               bpf_mtap2(sc->sc_drvbpf, tap, sc->sc_txtap_len, m0, BPF_D_OUT);
       }

       data = &txq->data[txq->cur];
       desc = &txq->desc[txq->cur];

       /* save and trim IEEE802.11 header */
       m_copydata(m0, 0, hdrlen, (void *)&desc->wh);
       m_adj(m0, hdrlen);

       error = bus_dmamap_load_mbuf(sc->sc_dmat, data->map, m0,
           BUS_DMA_WRITE | BUS_DMA_NOWAIT);
       if (error != 0 && error != EFBIG) {
               aprint_error_dev(sc->sc_dev, "could not map mbuf (error %d)\n",
                   error);
               m_freem(m0);
               return error;
       }
       if (error != 0) {
               /* too many fragments, linearize */

               MGETHDR(mnew, M_DONTWAIT, MT_DATA);
               if (mnew == NULL) {
                       m_freem(m0);
                       return ENOMEM;
               }

               m_copy_pkthdr(mnew, m0);

               /* If the data won't fit in the header, get a cluster */
               if (m0->m_pkthdr.len > MHLEN) {
                       MCLGET(mnew, M_DONTWAIT);
                       if (!(mnew->m_flags & M_EXT)) {
                               m_freem(m0);
                               m_freem(mnew);
                               return ENOMEM;
                       }
               }
               m_copydata(m0, 0, m0->m_pkthdr.len, mtod(mnew, void *));
               m_freem(m0);
               mnew->m_len = mnew->m_pkthdr.len;
               m0 = mnew;

               error = bus_dmamap_load_mbuf(sc->sc_dmat, data->map, m0,
                   BUS_DMA_WRITE | BUS_DMA_NOWAIT);
               if (error != 0) {
                       aprint_error_dev(sc->sc_dev,
                           "could not map mbuf (error %d)\n", error);
                       m_freem(m0);
                       return error;
               }
       }

       data->m = m0;
       data->ni = ni;

       desc->hdr.type = IWI_HDR_TYPE_DATA;
       desc->hdr.flags = IWI_HDR_FLAG_IRQ;
       desc->station =
           (ic->ic_opmode == IEEE80211_M_IBSS) ? in->in_station : 0;
       desc->cmd = IWI_DATA_CMD_TX;
       desc->len = htole16(m0->m_pkthdr.len);
       desc->flags = 0;
       desc->xflags = 0;

       if (!noack && !IEEE80211_IS_MULTICAST(desc->wh.i_addr1))
               desc->flags |= IWI_DATA_FLAG_NEED_ACK;

#if 0
       if (ic->ic_flags & IEEE80211_F_PRIVACY) {
               desc->wh.i_fc[1] |= IEEE80211_FC1_WEP;
               desc->wep_txkey = ic->ic_crypto.cs_def_txkey;
       } else
#endif
               desc->flags |= IWI_DATA_FLAG_NO_WEP;

       if (ic->ic_flags & IEEE80211_F_SHPREAMBLE)
               desc->flags |= IWI_DATA_FLAG_SHPREAMBLE;

       if (desc->wh.i_fc[0] & IEEE80211_FC0_SUBTYPE_QOS)
               desc->xflags |= IWI_DATA_XFLAG_QOS;

       if (ic->ic_curmode == IEEE80211_MODE_11B)
               desc->xflags |= IWI_DATA_XFLAG_CCK;

       desc->nseg = htole32(data->map->dm_nsegs);
       for (i = 0; i < data->map->dm_nsegs; i++) {
               desc->seg_addr[i] = htole32(data->map->dm_segs[i].ds_addr);
               desc->seg_len[i]  = htole16(data->map->dm_segs[i].ds_len);
       }

       bus_dmamap_sync(sc->sc_dmat, txq->desc_map,
           txq->cur * IWI_TX_DESC_SIZE,
           IWI_TX_DESC_SIZE, BUS_DMASYNC_PREWRITE);

       bus_dmamap_sync(sc->sc_dmat, data->map, 0, data->map->dm_mapsize,
           BUS_DMASYNC_PREWRITE);

       DPRINTFN(5, ("sending data frame txq=%u idx=%u len=%u nseg=%u\n",
           ac, txq->cur, le16toh(desc->len), le32toh(desc->nseg)));

       /* Inform firmware about this new packet */
       txq->queued++;
       txq->cur = (txq->cur + 1) % txq->count;
       CSR_WRITE_4(sc, txq->csr_widx, txq->cur);

       return 0;
}

static void
iwi_start(struct ifnet *ifp)
{
       struct iwi_softc *sc = ifp->if_softc;
       struct ieee80211com *ic = &sc->sc_ic;
       struct mbuf *m0;
       struct ether_header *eh;
       struct ieee80211_node *ni;
       int ac;

       if (ic->ic_state != IEEE80211_S_RUN)
               return;

       for (;;) {
               IFQ_POLL(&ifp->if_snd, m0);
               if (m0 == NULL)
                       break;

               KASSERT(m0->m_len >= sizeof(struct ether_header));

               eh = mtod(m0, struct ether_header *);
               ni = ieee80211_find_txnode(ic, eh->ether_dhost);
               if (ni == NULL) {
                       IFQ_DEQUEUE(&ifp->if_snd, m0);
                       m_freem(m0);
                       if_statinc(ifp, if_oerrors);
                       continue;
               }

               /* classify mbuf so we can find which tx ring to use */
               if (ieee80211_classify(ic, m0, ni) != 0) {
                       IFQ_DEQUEUE(&ifp->if_snd, m0);
                       m_freem(m0);
                       ieee80211_free_node(ni);
                       if_statinc(ifp, if_oerrors);
                       continue;
               }

               /* no QoS encapsulation for EAPOL frames */
               ac = (eh->ether_type != htons(ETHERTYPE_PAE)) ?
                   M_WME_GETAC(m0) : WME_AC_BE;

               if (sc->txq[ac].queued > sc->txq[ac].count - 8) {
                       /* there is no place left in this ring */
                       ifp->if_flags |= IFF_OACTIVE;
                       break;
               }
               IFQ_DEQUEUE(&ifp->if_snd, m0);

               bpf_mtap(ifp, m0, BPF_D_OUT);

               m0 = ieee80211_encap(ic, m0, ni);
               if (m0 == NULL) {
                       ieee80211_free_node(ni);
                       if_statinc(ifp, if_oerrors);
                       continue;
               }

               bpf_mtap3(ic->ic_rawbpf, m0, BPF_D_OUT);

               if (iwi_tx_start(ifp, m0, ni, ac) != 0) {
                       ieee80211_free_node(ni);
                       if_statinc(ifp, if_oerrors);
                       break;
               }

               /* start watchdog timer */
               sc->sc_tx_timer = 5;
               ifp->if_timer = 1;
       }
}

static void
iwi_watchdog(struct ifnet *ifp)
{
       struct iwi_softc *sc = ifp->if_softc;

       ifp->if_timer = 0;

       if (sc->sc_tx_timer > 0) {
               if (--sc->sc_tx_timer == 0) {
                       aprint_error_dev(sc->sc_dev, "device timeout\n");
                       if_statinc(ifp, if_oerrors);
                       ifp->if_flags &= ~IFF_UP;
                       iwi_stop(ifp, 1);
                       return;
               }
               ifp->if_timer = 1;
       }

       ieee80211_watchdog(&sc->sc_ic);
}

static int
iwi_get_table0(struct iwi_softc *sc, uint32_t *tbl)
{
       uint32_t size, buf[128];

       memset(buf, 0, sizeof buf);

       if (!(sc->flags & IWI_FLAG_FW_INITED)) {
               return copyout(buf, tbl, sizeof buf);
       }

       size = uimin(CSR_READ_4(sc, IWI_CSR_TABLE0_SIZE), 128 - 1);
       CSR_READ_REGION_4(sc, IWI_CSR_TABLE0_BASE, &buf[1], size);

       return copyout(buf, tbl, sizeof buf);
}

static int
iwi_ioctl(struct ifnet *ifp, u_long cmd, void *data)
{
#define IS_RUNNING(ifp) \
       ((ifp->if_flags & IFF_UP) && (ifp->if_flags & IFF_RUNNING))

       struct iwi_softc *sc = ifp->if_softc;
       struct ieee80211com *ic = &sc->sc_ic;
       struct ifreq *ifr = (struct ifreq *)data;
       int s, error = 0;
       int val;

       s = splnet();

       switch (cmd) {
       case SIOCSIFFLAGS:
               if ((error = ifioctl_common(ifp, cmd, data)) != 0)
                       break;
               if (ifp->if_flags & IFF_UP) {
                       if (!(ifp->if_flags & IFF_RUNNING))
                               iwi_init(ifp);
               } else {
                       if (ifp->if_flags & IFF_RUNNING)
                               iwi_stop(ifp, 1);
               }
               break;

       case SIOCADDMULTI:
       case SIOCDELMULTI:
               /* XXX no h/w multicast filter? --dyoung */
               if ((error = ether_ioctl(ifp, cmd, data)) == ENETRESET) {
                       /* setup multicast filter, etc */
                       error = 0;
               }
               break;

       case SIOCGTABLE0:
               error = iwi_get_table0(sc, (uint32_t *)ifr->ifr_data);
               break;

       case SIOCGRADIO:
               val = !iwi_getrfkill(sc);
               error = copyout(&val, (int *)ifr->ifr_data, sizeof val);
               break;

       case SIOCSIFMEDIA:
               if (ifr->ifr_media & IFM_IEEE80211_ADHOC) {
                       sc->sc_fwname = "ipw2200-ibss.fw";
               } else if (ifr->ifr_media & IFM_IEEE80211_MONITOR) {
                       sc->sc_fwname = "ipw2200-sniffer.fw";
               } else {
                       sc->sc_fwname = "ipw2200-bss.fw";
               }
               error = iwi_cache_firmware(sc);
               if (error)
                       break;

               /* FALLTHROUGH */
       default:
               error = ieee80211_ioctl(&sc->sc_ic, cmd, data);

               if (error == ENETRESET) {
                       if (IS_RUNNING(ifp) &&
                           (ic->ic_roaming != IEEE80211_ROAMING_MANUAL))
                               iwi_init(ifp);
                       error = 0;
               }
       }

       splx(s);
       return error;
#undef IS_RUNNING
}

static void
iwi_stop_master(struct iwi_softc *sc)
{
       int ntries;

       /* Disable interrupts */
       CSR_WRITE_4(sc, IWI_CSR_INTR_MASK, 0);

       CSR_WRITE_4(sc, IWI_CSR_RST, IWI_RST_STOP_MASTER);
       for (ntries = 0; ntries < 5; ntries++) {
               if (CSR_READ_4(sc, IWI_CSR_RST) & IWI_RST_MASTER_DISABLED)
                       break;
               DELAY(10);
       }
       if (ntries == 5)
               aprint_error_dev(sc->sc_dev, "timeout waiting for master\n");

       CSR_WRITE_4(sc, IWI_CSR_RST, CSR_READ_4(sc, IWI_CSR_RST) |
           IWI_RST_PRINCETON_RESET);

       sc->flags &= ~IWI_FLAG_FW_INITED;
}

static int
iwi_reset(struct iwi_softc *sc)
{
       int i, ntries;

       iwi_stop_master(sc);

       /* Move adapter to D0 state */
       CSR_WRITE_4(sc, IWI_CSR_CTL, CSR_READ_4(sc, IWI_CSR_CTL) |
           IWI_CTL_INIT);

       /* Initialize Phase-Locked Level  (PLL) */
       CSR_WRITE_4(sc, IWI_CSR_READ_INT, IWI_READ_INT_INIT_HOST);

       /* Wait for clock stabilization */
       for (ntries = 0; ntries < 1000; ntries++) {
               if (CSR_READ_4(sc, IWI_CSR_CTL) & IWI_CTL_CLOCK_READY)
                       break;
               DELAY(200);
       }
       if (ntries == 1000) {
               aprint_error_dev(sc->sc_dev,
                   "timeout waiting for clock stabilization\n");
               return ETIMEDOUT;
       }

       CSR_WRITE_4(sc, IWI_CSR_RST, CSR_READ_4(sc, IWI_CSR_RST) |
           IWI_RST_SW_RESET);

       DELAY(10);

       CSR_WRITE_4(sc, IWI_CSR_CTL, CSR_READ_4(sc, IWI_CSR_CTL) |
           IWI_CTL_INIT);

       /* Clear NIC memory */
       CSR_WRITE_4(sc, IWI_CSR_AUTOINC_ADDR, 0);
       for (i = 0; i < 0xc000; i++)
               CSR_WRITE_4(sc, IWI_CSR_AUTOINC_DATA, 0);

       return 0;
}

static int
iwi_load_ucode(struct iwi_softc *sc, void *uc, int size)
{
       uint16_t *w;
       int ntries, i;

       CSR_WRITE_4(sc, IWI_CSR_RST, CSR_READ_4(sc, IWI_CSR_RST) |
           IWI_RST_STOP_MASTER);
       for (ntries = 0; ntries < 5; ntries++) {
               if (CSR_READ_4(sc, IWI_CSR_RST) & IWI_RST_MASTER_DISABLED)
                       break;
               DELAY(10);
       }
       if (ntries == 5) {
               aprint_error_dev(sc->sc_dev, "timeout waiting for master\n");
               return ETIMEDOUT;
       }

       MEM_WRITE_4(sc, 0x3000e0, 0x80000000);
       DELAY(5000);
       CSR_WRITE_4(sc, IWI_CSR_RST, CSR_READ_4(sc, IWI_CSR_RST) &
           ~IWI_RST_PRINCETON_RESET);
       DELAY(5000);
       MEM_WRITE_4(sc, 0x3000e0, 0);
       DELAY(1000);
       MEM_WRITE_4(sc, 0x300004, 1);
       DELAY(1000);
       MEM_WRITE_4(sc, 0x300004, 0);
       DELAY(1000);
       MEM_WRITE_1(sc, 0x200000, 0x00);
       MEM_WRITE_1(sc, 0x200000, 0x40);
       DELAY(1000);

       /* Adapter is buggy, we must set the address for each word */
       for (w = uc; size > 0; w++, size -= 2)
               MEM_WRITE_2(sc, 0x200010, htole16(*w));

       MEM_WRITE_1(sc, 0x200000, 0x00);
       MEM_WRITE_1(sc, 0x200000, 0x80);

       /* Wait until we get a response in the uc queue */
       for (ntries = 0; ntries < 100; ntries++) {
               if (MEM_READ_1(sc, 0x200000) & 1)
                       break;
               DELAY(100);
       }
       if (ntries == 100) {
               aprint_error_dev(sc->sc_dev,
                   "timeout waiting for ucode to initialize\n");
               return ETIMEDOUT;
       }

       /* Empty the uc queue or the firmware will not initialize properly */
       for (i = 0; i < 7; i++)
               MEM_READ_4(sc, 0x200004);

       MEM_WRITE_1(sc, 0x200000, 0x00);

       return 0;
}

/* macro to handle unaligned little endian data in firmware image */
#define GETLE32(p) ((p)[0] | (p)[1] << 8 | (p)[2] << 16 | (p)[3] << 24)
static int
iwi_load_firmware(struct iwi_softc *sc, void *fw, int size)
{
       bus_dmamap_t map;
       u_char *p, *end;
       uint32_t sentinel, ctl, sum;
       uint32_t cs, sl, cd, cl;
       int ntries, nsegs, error;
       int sn;

       nsegs = atop((vaddr_t)fw+size-1) - atop((vaddr_t)fw) + 1;

       /* Create a DMA map for the firmware image */
       error = bus_dmamap_create(sc->sc_dmat, size, nsegs, size, 0,
           BUS_DMA_NOWAIT, &map);
       if (error != 0) {
               aprint_error_dev(sc->sc_dev,
                   "could not create firmware DMA map\n");
               map = NULL;
               goto fail1;
       }

       error = bus_dmamap_load(sc->sc_dmat, map, fw, size, NULL,
           BUS_DMA_NOWAIT | BUS_DMA_WRITE);
       if (error != 0) {
               aprint_error_dev(sc->sc_dev, "could not load fw dma map(%d)\n",
                   error);
               goto fail2;
       }

       /* Make sure the adapter will get up-to-date values */
       bus_dmamap_sync(sc->sc_dmat, map, 0, size, BUS_DMASYNC_PREWRITE);

       /* Tell the adapter where the command blocks are stored */
       MEM_WRITE_4(sc, 0x3000a0, 0x27000);

       /*
        * Store command blocks into adapter's internal memory using register
        * indirections. The adapter will read the firmware image through DMA
        * using information stored in command blocks.
        */
       p = fw;
       end = p + size;
       CSR_WRITE_4(sc, IWI_CSR_AUTOINC_ADDR, 0x27000);

       sn = 0;
       sl = cl = 0;
       cs = cd = 0;
       while (p < end) {
               if (sl == 0) {
                       cs = map->dm_segs[sn].ds_addr;
                       sl = map->dm_segs[sn].ds_len;
                       sn++;
               }
               if (cl == 0) {
                       cd = GETLE32(p); p += 4; cs += 4; sl -= 4;
                       cl = GETLE32(p); p += 4; cs += 4; sl -= 4;
               }
               while (sl > 0 && cl > 0) {
                       int len = uimin(cl, sl);

                       sl -= len;
                       cl -= len;
                       p += len;

                       while (len > 0) {
                               int mlen = uimin(len, IWI_CB_MAXDATALEN);

                               ctl = IWI_CB_DEFAULT_CTL | mlen;
                               sum = ctl ^ cs ^ cd;

                               /* Write a command block */
                               CSR_WRITE_4(sc, IWI_CSR_AUTOINC_DATA, ctl);
                               CSR_WRITE_4(sc, IWI_CSR_AUTOINC_DATA, cs);
                               CSR_WRITE_4(sc, IWI_CSR_AUTOINC_DATA, cd);
                               CSR_WRITE_4(sc, IWI_CSR_AUTOINC_DATA, sum);

                               cs += mlen;
                               cd += mlen;
                               len -= mlen;
                       }
               }
       }

       /* Write a fictive final command block (sentinel) */
       sentinel = CSR_READ_4(sc, IWI_CSR_AUTOINC_ADDR);
       CSR_WRITE_4(sc, IWI_CSR_AUTOINC_DATA, 0);

       CSR_WRITE_4(sc, IWI_CSR_RST, CSR_READ_4(sc, IWI_CSR_RST) &
           ~(IWI_RST_MASTER_DISABLED | IWI_RST_STOP_MASTER));

       /* Tell the adapter to start processing command blocks */
       MEM_WRITE_4(sc, 0x3000a4, 0x540100);

       /* Wait until the adapter has processed all command blocks */
       for (ntries = 0; ntries < 400; ntries++) {
               if (MEM_READ_4(sc, 0x3000d0) >= sentinel)
                       break;
               DELAY(100);
       }
       if (ntries == 400) {
               aprint_error_dev(sc->sc_dev, "timeout processing cb\n");
               error = ETIMEDOUT;
               goto fail3;
       }

       /* We're done with command blocks processing */
       MEM_WRITE_4(sc, 0x3000a4, 0x540c00);

       /* Allow interrupts so we know when the firmware is inited */
       CSR_WRITE_4(sc, IWI_CSR_INTR_MASK, IWI_INTR_MASK);

       /* Tell the adapter to initialize the firmware */
       CSR_WRITE_4(sc, IWI_CSR_RST, 0);
       CSR_WRITE_4(sc, IWI_CSR_CTL, CSR_READ_4(sc, IWI_CSR_CTL) |
           IWI_CTL_ALLOW_STANDBY);

       /* Wait at most one second for firmware initialization to complete */
       if ((error = tsleep(sc, 0, "iwiinit", hz)) != 0) {
               aprint_error_dev(sc->sc_dev,
                   "timeout waiting for firmware initialization to complete\n");
               goto fail3;
       }

fail3:
       bus_dmamap_sync(sc->sc_dmat, map, 0, size, BUS_DMASYNC_POSTWRITE);
       bus_dmamap_unload(sc->sc_dmat, map);
fail2:
       if (map != NULL)
               bus_dmamap_destroy(sc->sc_dmat, map);

fail1:
       return error;
}

/*
* Store firmware into kernel memory so we can download it when we need to,
* e.g when the adapter wakes up from suspend mode.
*/
static int
iwi_cache_firmware(struct iwi_softc *sc)
{
       struct iwi_firmware *kfw = &sc->fw;
       firmware_handle_t fwh;
       struct iwi_firmware_hdr *hdr;
       off_t size;
       char *fw;
       int error;

       if (iwi_accept_eula == 0) {
               aprint_error_dev(sc->sc_dev,
                   "EULA not accepted; please see the iwi(4) man page.\n");
               return EPERM;
       }

       iwi_free_firmware(sc);
       error = firmware_open("if_iwi", sc->sc_fwname, &fwh);
       if (error != 0) {
               aprint_error_dev(sc->sc_dev, "firmware_open failed\n");
               goto fail1;
       }

       size = firmware_get_size(fwh);
       if (size < sizeof(struct iwi_firmware_hdr)) {
               aprint_error_dev(sc->sc_dev, "image '%s' has no header\n",
                   sc->sc_fwname);
               error = EIO;
               goto fail1;
       }
       sc->sc_blobsize = size;

       sc->sc_blob = firmware_malloc(size);
       if (sc->sc_blob == NULL) {
               error = ENOMEM;
               firmware_close(fwh);
               goto fail1;
       }

       error = firmware_read(fwh, 0, sc->sc_blob, size);
       firmware_close(fwh);
       if (error != 0)
               goto fail2;

       hdr = (struct iwi_firmware_hdr *)sc->sc_blob;
       hdr->version = le32toh(hdr->version);
       hdr->bsize = le32toh(hdr->bsize);
       hdr->usize = le32toh(hdr->usize);
       hdr->fsize = le32toh(hdr->fsize);

       if (size < sizeof(struct iwi_firmware_hdr) + hdr->bsize + hdr->usize + hdr->fsize) {
               aprint_error_dev(sc->sc_dev, "image '%s' too small\n",
                   sc->sc_fwname);
               error = EIO;
               goto fail2;
       }

       DPRINTF(("firmware version = %d\n", hdr->version));
       if ((IWI_FW_GET_MAJOR(hdr->version) != IWI_FW_REQ_MAJOR) ||
           (IWI_FW_GET_MINOR(hdr->version) != IWI_FW_REQ_MINOR)) {
               aprint_error_dev(sc->sc_dev,
                   "version for '%s' %d.%d != %d.%d\n", sc->sc_fwname,
                   IWI_FW_GET_MAJOR(hdr->version),
                   IWI_FW_GET_MINOR(hdr->version),
                   IWI_FW_REQ_MAJOR, IWI_FW_REQ_MINOR);
               error = EIO;
               goto fail2;
       }

       kfw->boot_size = hdr->bsize;
       kfw->ucode_size = hdr->usize;
       kfw->main_size = hdr->fsize;

       fw = sc->sc_blob + sizeof(struct iwi_firmware_hdr);
       kfw->boot = fw;
       fw += kfw->boot_size;
       kfw->ucode = fw;
       fw += kfw->ucode_size;
       kfw->main = fw;

       DPRINTF(("Firmware cached: boot %p, ucode %p, main %p\n",
           kfw->boot, kfw->ucode, kfw->main));
       DPRINTF(("Firmware cached: boot %u, ucode %u, main %u\n",
           kfw->boot_size, kfw->ucode_size, kfw->main_size));

       sc->flags |= IWI_FLAG_FW_CACHED;

       return 0;


fail2:  firmware_free(sc->sc_blob, sc->sc_blobsize);
fail1:
       return error;
}

static void
iwi_free_firmware(struct iwi_softc *sc)
{

       if (!(sc->flags & IWI_FLAG_FW_CACHED))
               return;

       firmware_free(sc->sc_blob, sc->sc_blobsize);

       sc->flags &= ~IWI_FLAG_FW_CACHED;
}

static int
iwi_config(struct iwi_softc *sc)
{
       struct ieee80211com *ic = &sc->sc_ic;
       struct ifnet *ifp = &sc->sc_if;
       struct iwi_configuration config;
       struct iwi_rateset rs;
       struct iwi_txpower power;
       struct ieee80211_key *wk;
       struct iwi_wep_key wepkey;
       uint32_t data;
       int error, nchan, i;

       IEEE80211_ADDR_COPY(ic->ic_myaddr, CLLADDR(ifp->if_sadl));
       DPRINTF(("Setting MAC address to %s\n", ether_sprintf(ic->ic_myaddr)));
       error = iwi_cmd(sc, IWI_CMD_SET_MAC_ADDRESS, ic->ic_myaddr,
           IEEE80211_ADDR_LEN, 0);
       if (error != 0)
               return error;

       memset(&config, 0, sizeof config);
       config.bluetooth_coexistence = sc->bluetooth;
       config.antenna = sc->antenna;
       config.silence_threshold = 0x1e;
       config.multicast_enabled = 1;
       config.answer_pbreq = (ic->ic_opmode == IEEE80211_M_IBSS) ? 1 : 0;
       config.disable_unicast_decryption = 1;
       config.disable_multicast_decryption = 1;
       DPRINTF(("Configuring adapter\n"));
       error = iwi_cmd(sc, IWI_CMD_SET_CONFIGURATION, &config, sizeof config,
           0);
       if (error != 0)
               return error;

       data = htole32(IWI_POWER_MODE_CAM);
       DPRINTF(("Setting power mode to %u\n", le32toh(data)));
       error = iwi_cmd(sc, IWI_CMD_SET_POWER_MODE, &data, sizeof data, 0);
       if (error != 0)
               return error;

       data = htole32(ic->ic_rtsthreshold);
       DPRINTF(("Setting RTS threshold to %u\n", le32toh(data)));
       error = iwi_cmd(sc, IWI_CMD_SET_RTS_THRESHOLD, &data, sizeof data, 0);
       if (error != 0)
               return error;

       data = htole32(ic->ic_fragthreshold);
       DPRINTF(("Setting fragmentation threshold to %u\n", le32toh(data)));
       error = iwi_cmd(sc, IWI_CMD_SET_FRAG_THRESHOLD, &data, sizeof data, 0);
       if (error != 0)
               return error;

       /*
        * Set default Tx power for 802.11b/g and 802.11a channels.
        */
       nchan = 0;
       for (i = 0; i <= IEEE80211_CHAN_MAX; i++) {
               if (!IEEE80211_IS_CHAN_2GHZ(&ic->ic_channels[i]))
                       continue;
               power.chan[nchan].chan = i;
               power.chan[nchan].power = IWI_TXPOWER_MAX;
               nchan++;
       }
       power.nchan = nchan;

       power.mode = IWI_MODE_11G;
       DPRINTF(("Setting .11g channels tx power\n"));
       error = iwi_cmd(sc, IWI_CMD_SET_TX_POWER, &power, sizeof power, 0);
       if (error != 0)
               return error;

       power.mode = IWI_MODE_11B;
       DPRINTF(("Setting .11b channels tx power\n"));
       error = iwi_cmd(sc, IWI_CMD_SET_TX_POWER, &power, sizeof power, 0);
       if (error != 0)
               return error;

       nchan = 0;
       for (i = 0; i <= IEEE80211_CHAN_MAX; i++) {
               if (!IEEE80211_IS_CHAN_5GHZ(&ic->ic_channels[i]))
                       continue;
               power.chan[nchan].chan = i;
               power.chan[nchan].power = IWI_TXPOWER_MAX;
               nchan++;
       }
       power.nchan = nchan;

       if (nchan > 0) {        /* 2915ABG only */
               power.mode = IWI_MODE_11A;
               DPRINTF(("Setting .11a channels tx power\n"));
               error = iwi_cmd(sc, IWI_CMD_SET_TX_POWER, &power, sizeof power,
                   0);
               if (error != 0)
                       return error;
       }

       rs.mode = IWI_MODE_11G;
       rs.type = IWI_RATESET_TYPE_SUPPORTED;
       rs.nrates = ic->ic_sup_rates[IEEE80211_MODE_11G].rs_nrates;
       memcpy(rs.rates, ic->ic_sup_rates[IEEE80211_MODE_11G].rs_rates,
           rs.nrates);
       DPRINTF(("Setting .11bg supported rates (%u)\n", rs.nrates));
       error = iwi_cmd(sc, IWI_CMD_SET_RATES, &rs, sizeof rs, 0);
       if (error != 0)
               return error;

       rs.mode = IWI_MODE_11A;
       rs.type = IWI_RATESET_TYPE_SUPPORTED;
       rs.nrates = ic->ic_sup_rates[IEEE80211_MODE_11A].rs_nrates;
       memcpy(rs.rates, ic->ic_sup_rates[IEEE80211_MODE_11A].rs_rates,
           rs.nrates);
       DPRINTF(("Setting .11a supported rates (%u)\n", rs.nrates));
       error = iwi_cmd(sc, IWI_CMD_SET_RATES, &rs, sizeof rs, 0);
       if (error != 0)
               return error;

       /* if we have a desired ESSID, set it now */
       if (ic->ic_des_esslen != 0) {
#ifdef IWI_DEBUG
               if (iwi_debug > 0) {
                       printf("Setting desired ESSID to ");
                       ieee80211_print_essid(ic->ic_des_essid,
                           ic->ic_des_esslen);
                       printf("\n");
               }
#endif
               error = iwi_cmd(sc, IWI_CMD_SET_ESSID, ic->ic_des_essid,
                   ic->ic_des_esslen, 0);
               if (error != 0)
                       return error;
       }

       cprng_fast(&data, sizeof(data));
       data = htole32(data);
       DPRINTF(("Setting initialization vector to %u\n", le32toh(data)));
       error = iwi_cmd(sc, IWI_CMD_SET_IV, &data, sizeof data, 0);
       if (error != 0)
               return error;

       if (ic->ic_flags & IEEE80211_F_PRIVACY) {
               /* XXX iwi_setwepkeys? */
               for (i = 0; i < IEEE80211_WEP_NKID; i++) {
                       wk = &ic->ic_crypto.cs_nw_keys[i];

                       wepkey.cmd = IWI_WEP_KEY_CMD_SETKEY;
                       wepkey.idx = i;
                       wepkey.len = wk->wk_keylen;
                       memset(wepkey.key, 0, sizeof wepkey.key);
                       memcpy(wepkey.key, wk->wk_key, wk->wk_keylen);
                       DPRINTF(("Setting wep key index %u len %u\n",
                           wepkey.idx, wepkey.len));
                       error = iwi_cmd(sc, IWI_CMD_SET_WEP_KEY, &wepkey,
                           sizeof wepkey, 0);
                       if (error != 0)
                               return error;
               }
       }

       /* Enable adapter */
       DPRINTF(("Enabling adapter\n"));
       return iwi_cmd(sc, IWI_CMD_ENABLE, NULL, 0, 0);
}

static int
iwi_set_chan(struct iwi_softc *sc, struct ieee80211_channel *chan)
{
       struct ieee80211com *ic = &sc->sc_ic;
       struct iwi_scan_v2 scan;

       (void)memset(&scan, 0, sizeof scan);

       scan.dwelltime[IWI_SCAN_TYPE_PASSIVE] = htole16(2000);
       scan.channels[0] = 1 |
           (IEEE80211_IS_CHAN_5GHZ(chan) ? IWI_CHAN_5GHZ : IWI_CHAN_2GHZ);
       scan.channels[1] = ieee80211_chan2ieee(ic, chan);
       iwi_scan_type_set(scan, 1, IWI_SCAN_TYPE_PASSIVE);

       DPRINTF(("Setting channel to %u\n", ieee80211_chan2ieee(ic, chan)));
       return iwi_cmd(sc, IWI_CMD_SCAN_V2, &scan, sizeof scan, 1);
}

static int
iwi_scan(struct iwi_softc *sc)
{
       struct ieee80211com *ic = &sc->sc_ic;
       struct iwi_scan_v2 scan;
       uint32_t type;
       uint8_t *p;
       int i, count, idx;

       (void)memset(&scan, 0, sizeof scan);
       scan.dwelltime[IWI_SCAN_TYPE_ACTIVE_BROADCAST] =
           htole16(sc->dwelltime);
       scan.dwelltime[IWI_SCAN_TYPE_ACTIVE_BDIRECT] =
           htole16(sc->dwelltime);

       /* tell the firmware about the desired essid */
       if (ic->ic_des_esslen) {
               int error;

               DPRINTF(("%s: Setting adapter desired ESSID to %s\n",
                   __func__, ic->ic_des_essid));

               error = iwi_cmd(sc, IWI_CMD_SET_ESSID,
                   ic->ic_des_essid, ic->ic_des_esslen, 1);
               if (error)
                       return error;

               type = IWI_SCAN_TYPE_ACTIVE_BDIRECT;
       } else {
               type = IWI_SCAN_TYPE_ACTIVE_BROADCAST;
       }

       p = &scan.channels[0];
       count = idx = 0;
       for (i = 0; i <= IEEE80211_CHAN_MAX; i++) {
               if (IEEE80211_IS_CHAN_5GHZ(&ic->ic_channels[i]) &&
                   isset(ic->ic_chan_active, i)) {
                       *++p = i;
                       count++;
                       idx++;
                       iwi_scan_type_set(scan, idx, type);
               }
       }
       if (count) {
               *(p - count) = IWI_CHAN_5GHZ | count;
               p++;
       }

       count = 0;
       for (i = 0; i <= IEEE80211_CHAN_MAX; i++) {
               if (IEEE80211_IS_CHAN_2GHZ(&ic->ic_channels[i]) &&
                   isset(ic->ic_chan_active, i)) {
                       *++p = i;
                       count++;
                       idx++;
                       iwi_scan_type_set(scan, idx, type);
               }
       }
       *(p - count) = IWI_CHAN_2GHZ | count;

       DPRINTF(("Start scanning\n"));
       return iwi_cmd(sc, IWI_CMD_SCAN_V2, &scan, sizeof scan, 1);
}

static int
iwi_auth_and_assoc(struct iwi_softc *sc)
{
       struct ieee80211com *ic = &sc->sc_ic;
       struct ieee80211_node *ni = ic->ic_bss;
       struct ifnet *ifp = &sc->sc_if;
       struct ieee80211_wme_info wme;
       struct iwi_configuration config;
       struct iwi_associate assoc;
       struct iwi_rateset rs;
       uint16_t capinfo;
       uint32_t data;
       int error;

       memset(&config, 0, sizeof config);
       config.bluetooth_coexistence = sc->bluetooth;
       config.antenna = sc->antenna;
       config.multicast_enabled = 1;
       config.silence_threshold = 0x1e;
       if (ic->ic_curmode == IEEE80211_MODE_11G)
               config.use_protection = 1;
       config.answer_pbreq = (ic->ic_opmode == IEEE80211_M_IBSS) ? 1 : 0;
       config.disable_unicast_decryption = 1;
       config.disable_multicast_decryption = 1;

       DPRINTF(("Configuring adapter\n"));
       error = iwi_cmd(sc, IWI_CMD_SET_CONFIGURATION, &config,
           sizeof config, 1);
       if (error != 0)
               return error;

#ifdef IWI_DEBUG
       if (iwi_debug > 0) {
               aprint_debug_dev(sc->sc_dev, "Setting ESSID to ");
               ieee80211_print_essid(ni->ni_essid, ni->ni_esslen);
               aprint_debug("\n");
       }
#endif
       error = iwi_cmd(sc, IWI_CMD_SET_ESSID, ni->ni_essid, ni->ni_esslen, 1);
       if (error != 0)
               return error;

       /* the rate set has already been "negotiated" */
       rs.mode = IEEE80211_IS_CHAN_5GHZ(ni->ni_chan) ? IWI_MODE_11A :
           IWI_MODE_11G;
       rs.type = IWI_RATESET_TYPE_NEGOTIATED;
       rs.nrates = ni->ni_rates.rs_nrates;

       if (rs.nrates > IWI_RATESET_SIZE) {
               DPRINTF(("Truncating negotiated rate set from %u\n",
                   rs.nrates));
               rs.nrates = IWI_RATESET_SIZE;
       }
       memcpy(rs.rates, ni->ni_rates.rs_rates, rs.nrates);
       DPRINTF(("Setting negotiated rates (%u)\n", rs.nrates));
       error = iwi_cmd(sc, IWI_CMD_SET_RATES, &rs, sizeof rs, 1);
       if (error != 0)
               return error;

       if ((ic->ic_flags & IEEE80211_F_WME) && ni->ni_wme_ie != NULL) {
               wme.wme_id = IEEE80211_ELEMID_VENDOR;
               wme.wme_len = sizeof (struct ieee80211_wme_info) - 2;
               wme.wme_oui[0] = 0x00;
               wme.wme_oui[1] = 0x50;
               wme.wme_oui[2] = 0xf2;
               wme.wme_type = WME_OUI_TYPE;
               wme.wme_subtype = WME_INFO_OUI_SUBTYPE;
               wme.wme_version = WME_VERSION;
               wme.wme_info = 0;

               DPRINTF(("Setting WME IE (len=%u)\n", wme.wme_len));
               error = iwi_cmd(sc, IWI_CMD_SET_WMEIE, &wme, sizeof wme, 1);
               if (error != 0)
                       return error;
       }

       if (ic->ic_opt_ie != NULL) {
               DPRINTF(("Setting optional IE (len=%u)\n", ic->ic_opt_ie_len));
               error = iwi_cmd(sc, IWI_CMD_SET_OPTIE, ic->ic_opt_ie,
                   ic->ic_opt_ie_len, 1);
               if (error != 0)
                       return error;
       }
       data = htole32(ni->ni_rssi);
       DPRINTF(("Setting sensitivity to %d\n", (int8_t)ni->ni_rssi));
       error = iwi_cmd(sc, IWI_CMD_SET_SENSITIVITY, &data, sizeof data, 1);
       if (error != 0)
               return error;

       memset(&assoc, 0, sizeof assoc);
       if (IEEE80211_IS_CHAN_A(ni->ni_chan))
               assoc.mode = IWI_MODE_11A;
       else if (IEEE80211_IS_CHAN_G(ni->ni_chan))
               assoc.mode = IWI_MODE_11G;
       else if (IEEE80211_IS_CHAN_B(ni->ni_chan))
               assoc.mode = IWI_MODE_11B;

       assoc.chan = ieee80211_chan2ieee(ic, ni->ni_chan);

       if (ni->ni_authmode == IEEE80211_AUTH_SHARED)
               assoc.auth = (ic->ic_crypto.cs_def_txkey << 4) | IWI_AUTH_SHARED;

       if (ic->ic_flags & IEEE80211_F_SHPREAMBLE)
               assoc.plen = IWI_ASSOC_SHPREAMBLE;

       if ((ic->ic_flags & IEEE80211_F_WME) && ni->ni_wme_ie != NULL)
               assoc.policy |= htole16(IWI_POLICY_WME);
       if (ic->ic_flags & IEEE80211_F_WPA)
               assoc.policy |= htole16(IWI_POLICY_WPA);
       if (ic->ic_opmode == IEEE80211_M_IBSS && ni->ni_tstamp.tsf == 0)
               assoc.type = IWI_HC_IBSS_START;
       else
               assoc.type = IWI_HC_ASSOC;
       memcpy(assoc.tstamp, ni->ni_tstamp.data, 8);

       if (ic->ic_opmode == IEEE80211_M_IBSS)
               capinfo = IEEE80211_CAPINFO_IBSS;
       else
               capinfo = IEEE80211_CAPINFO_ESS;
       if (ic->ic_flags & IEEE80211_F_PRIVACY)
               capinfo |= IEEE80211_CAPINFO_PRIVACY;
       if ((ic->ic_flags & IEEE80211_F_SHPREAMBLE) &&
           IEEE80211_IS_CHAN_2GHZ(ni->ni_chan))
               capinfo |= IEEE80211_CAPINFO_SHORT_PREAMBLE;
       if (ic->ic_flags & IEEE80211_F_SHSLOT)
               capinfo |= IEEE80211_CAPINFO_SHORT_SLOTTIME;
       assoc.capinfo = htole16(capinfo);

       assoc.lintval = htole16(ic->ic_lintval);
       assoc.intval = htole16(ni->ni_intval);
       IEEE80211_ADDR_COPY(assoc.bssid, ni->ni_bssid);
       if (ic->ic_opmode == IEEE80211_M_IBSS)
               IEEE80211_ADDR_COPY(assoc.dst, ifp->if_broadcastaddr);
       else
               IEEE80211_ADDR_COPY(assoc.dst, ni->ni_bssid);

       DPRINTF(("%s bssid %s dst %s channel %u policy 0x%x "
           "auth %u capinfo 0x%x lintval %u bintval %u\n",
           assoc.type == IWI_HC_IBSS_START ? "Start" : "Join",
           ether_sprintf(assoc.bssid), ether_sprintf(assoc.dst),
           assoc.chan, le16toh(assoc.policy), assoc.auth,
           le16toh(assoc.capinfo), le16toh(assoc.lintval),
           le16toh(assoc.intval)));

       return iwi_cmd(sc, IWI_CMD_ASSOCIATE, &assoc, sizeof assoc, 1);
}

static int
iwi_init(struct ifnet *ifp)
{
       struct iwi_softc *sc = ifp->if_softc;
       struct ieee80211com *ic = &sc->sc_ic;
       struct iwi_firmware *fw = &sc->fw;
       int i, error;

       /* exit immediately if firmware has not been ioctl'd */
       if (!(sc->flags & IWI_FLAG_FW_CACHED)) {
               if ((error = iwi_cache_firmware(sc)) != 0) {
                       aprint_error_dev(sc->sc_dev,
                           "could not cache the firmware\n");
                       goto fail;
               }
       }

       iwi_stop(ifp, 0);

       if ((error = iwi_reset(sc)) != 0) {
               aprint_error_dev(sc->sc_dev, "could not reset adapter\n");
               goto fail;
       }

       if ((error = iwi_load_firmware(sc, fw->boot, fw->boot_size)) != 0) {
               aprint_error_dev(sc->sc_dev, "could not load boot firmware\n");
               goto fail;
       }

       if ((error = iwi_load_ucode(sc, fw->ucode, fw->ucode_size)) != 0) {
               aprint_error_dev(sc->sc_dev, "could not load microcode\n");
               goto fail;
       }

       iwi_stop_master(sc);

       CSR_WRITE_4(sc, IWI_CSR_CMD_BASE, sc->cmdq.desc_map->dm_segs[0].ds_addr);
       CSR_WRITE_4(sc, IWI_CSR_CMD_SIZE, sc->cmdq.count);
       CSR_WRITE_4(sc, IWI_CSR_CMD_WIDX, sc->cmdq.cur);

       CSR_WRITE_4(sc, IWI_CSR_TX1_BASE, sc->txq[0].desc_map->dm_segs[0].ds_addr);
       CSR_WRITE_4(sc, IWI_CSR_TX1_SIZE, sc->txq[0].count);
       CSR_WRITE_4(sc, IWI_CSR_TX1_WIDX, sc->txq[0].cur);

       CSR_WRITE_4(sc, IWI_CSR_TX2_BASE, sc->txq[1].desc_map->dm_segs[0].ds_addr);
       CSR_WRITE_4(sc, IWI_CSR_TX2_SIZE, sc->txq[1].count);
       CSR_WRITE_4(sc, IWI_CSR_TX2_WIDX, sc->txq[1].cur);

       CSR_WRITE_4(sc, IWI_CSR_TX3_BASE, sc->txq[2].desc_map->dm_segs[0].ds_addr);
       CSR_WRITE_4(sc, IWI_CSR_TX3_SIZE, sc->txq[2].count);
       CSR_WRITE_4(sc, IWI_CSR_TX3_WIDX, sc->txq[2].cur);

       CSR_WRITE_4(sc, IWI_CSR_TX4_BASE, sc->txq[3].desc_map->dm_segs[0].ds_addr);
       CSR_WRITE_4(sc, IWI_CSR_TX4_SIZE, sc->txq[3].count);
       CSR_WRITE_4(sc, IWI_CSR_TX4_WIDX, sc->txq[3].cur);

       for (i = 0; i < sc->rxq.count; i++)
               CSR_WRITE_4(sc, IWI_CSR_RX_BASE + i * 4,
                   sc->rxq.data[i].map->dm_segs[0].ds_addr);

       CSR_WRITE_4(sc, IWI_CSR_RX_WIDX, sc->rxq.count -1);

       if ((error = iwi_load_firmware(sc, fw->main, fw->main_size)) != 0) {
               aprint_error_dev(sc->sc_dev, "could not load main firmware\n");
               goto fail;
       }

       sc->flags |= IWI_FLAG_FW_INITED;

       if ((error = iwi_config(sc)) != 0) {
               aprint_error_dev(sc->sc_dev, "device configuration failed\n");
               goto fail;
       }

       ic->ic_state = IEEE80211_S_INIT;

       ifp->if_flags &= ~IFF_OACTIVE;
       ifp->if_flags |= IFF_RUNNING;

       if (ic->ic_opmode != IEEE80211_M_MONITOR) {
               if (ic->ic_roaming != IEEE80211_ROAMING_MANUAL)
                       ieee80211_new_state(ic, IEEE80211_S_SCAN, -1);
       } else
               ieee80211_new_state(ic, IEEE80211_S_RUN, -1);

       return 0;

fail:   ifp->if_flags &= ~IFF_UP;
       iwi_stop(ifp, 0);

       return error;
}


/*
* Return whether or not the radio is enabled in hardware
* (i.e. the rfkill switch is "off").
*/
static int
iwi_getrfkill(struct iwi_softc *sc)
{
       return (CSR_READ_4(sc, IWI_CSR_IO) & IWI_IO_RADIO_ENABLED) == 0;
}

static int
iwi_sysctl_radio(SYSCTLFN_ARGS)
{
       struct sysctlnode node;
       struct iwi_softc *sc;
       int val, error;

       node = *rnode;
       sc = (struct iwi_softc *)node.sysctl_data;

       val = !iwi_getrfkill(sc);

       node.sysctl_data = &val;
       error = sysctl_lookup(SYSCTLFN_CALL(&node));

       if (error || newp == NULL)
               return error;

       return 0;
}

#ifdef IWI_DEBUG
SYSCTL_SETUP(sysctl_iwi, "sysctl iwi(4) subtree setup")
{
       int rc;
       const struct sysctlnode *rnode;
       const struct sysctlnode *cnode;

       if ((rc = sysctl_createv(clog, 0, NULL, &rnode,
           CTLFLAG_PERMANENT, CTLTYPE_NODE, "iwi",
           SYSCTL_DESCR("iwi global controls"),
           NULL, 0, NULL, 0, CTL_HW, CTL_CREATE, CTL_EOL)) != 0)
               goto err;

       /* control debugging printfs */
       if ((rc = sysctl_createv(clog, 0, &rnode, &cnode,
           CTLFLAG_PERMANENT|CTLFLAG_READWRITE, CTLTYPE_INT,
           "debug", SYSCTL_DESCR("Enable debugging output"),
           NULL, 0, &iwi_debug, 0, CTL_CREATE, CTL_EOL)) != 0)
               goto err;

       return;
err:
       aprint_error("%s: sysctl_createv failed (rc = %d)\n", __func__, rc);
}

#endif /* IWI_DEBUG */

/*
* Add sysctl knobs.
*/
static void
iwi_sysctlattach(struct iwi_softc *sc)
{
       int rc;
       const struct sysctlnode *rnode;
       const struct sysctlnode *cnode;

       struct sysctllog **clog = &sc->sc_sysctllog;

       if ((rc = sysctl_createv(clog, 0, NULL, &rnode,
           CTLFLAG_PERMANENT, CTLTYPE_NODE, device_xname(sc->sc_dev),
           SYSCTL_DESCR("iwi controls and statistics"),
           NULL, 0, NULL, 0, CTL_HW, CTL_CREATE, CTL_EOL)) != 0)
               goto err;

       if ((rc = sysctl_createv(clog, 0, &rnode, &cnode,
           CTLFLAG_PERMANENT, CTLTYPE_INT, "radio",
           SYSCTL_DESCR("radio transmitter switch state (0=off, 1=on)"),
           iwi_sysctl_radio, 0, (void *)sc, 0, CTL_CREATE, CTL_EOL)) != 0)
               goto err;

       sc->dwelltime = 100;
       if ((rc = sysctl_createv(clog, 0, &rnode, &cnode,
           CTLFLAG_PERMANENT|CTLFLAG_READWRITE, CTLTYPE_INT,
           "dwell", SYSCTL_DESCR("channel dwell time (ms) for AP/station scanning"),
           NULL, 0, &sc->dwelltime, 0, CTL_CREATE, CTL_EOL)) != 0)
               goto err;

       sc->bluetooth = 0;
       if ((rc = sysctl_createv(clog, 0, &rnode, &cnode,
           CTLFLAG_PERMANENT|CTLFLAG_READWRITE, CTLTYPE_INT,
           "bluetooth", SYSCTL_DESCR("bluetooth coexistence"),
           NULL, 0, &sc->bluetooth, 0, CTL_CREATE, CTL_EOL)) != 0)
               goto err;

       sc->antenna = IWI_ANTENNA_AUTO;
       if ((rc = sysctl_createv(clog, 0, &rnode, &cnode,
           CTLFLAG_PERMANENT|CTLFLAG_READWRITE, CTLTYPE_INT,
           "antenna", SYSCTL_DESCR("antenna (0=auto)"),
           NULL, 0, &sc->antenna, 0, CTL_CREATE, CTL_EOL)) != 0)
               goto err;

       return;
err:
       aprint_error("%s: sysctl_createv failed (rc = %d)\n", __func__, rc);
}

static void
iwi_stop(struct ifnet *ifp, int disable)
{
       struct iwi_softc *sc = ifp->if_softc;
       struct ieee80211com *ic = &sc->sc_ic;

       IWI_LED_OFF(sc);

       iwi_stop_master(sc);
       CSR_WRITE_4(sc, IWI_CSR_RST, IWI_RST_SW_RESET);

       /* reset rings */
       iwi_reset_cmd_ring(sc, &sc->cmdq);
       iwi_reset_tx_ring(sc, &sc->txq[0]);
       iwi_reset_tx_ring(sc, &sc->txq[1]);
       iwi_reset_tx_ring(sc, &sc->txq[2]);
       iwi_reset_tx_ring(sc, &sc->txq[3]);
       iwi_reset_rx_ring(sc, &sc->rxq);

       ifp->if_timer = 0;
       ifp->if_flags &= ~(IFF_RUNNING | IFF_OACTIVE);

       ieee80211_new_state(ic, IEEE80211_S_INIT, -1);
}

static void
iwi_led_set(struct iwi_softc *sc, uint32_t state, int toggle)
{
       uint32_t val;

       val = MEM_READ_4(sc, IWI_MEM_EVENT_CTL);

       switch (sc->nictype) {
       case 1:
               /* special NIC type: reversed leds */
               if (state == IWI_LED_ACTIVITY) {
                       state &= ~IWI_LED_ACTIVITY;
                       state |= IWI_LED_ASSOCIATED;
               } else if (state == IWI_LED_ASSOCIATED) {
                       state &= ~IWI_LED_ASSOCIATED;
                       state |= IWI_LED_ACTIVITY;
               }
               /* and ignore toggle effect */
               val |= state;
               break;
       case 0:
       case 2:
       case 3:
       case 4:
               val = (toggle && (val & state)) ? val & ~state : val | state;
               break;
       default:
               aprint_normal_dev(sc->sc_dev, "unknown NIC type %d\n",
                   sc->nictype);
               return;
               break;
       }

       MEM_WRITE_4(sc, IWI_MEM_EVENT_CTL, val);

       return;
}

SYSCTL_SETUP(sysctl_hw_iwi_accept_eula_setup, "sysctl hw.iwi.accept_eula")
{
       const struct sysctlnode *rnode;
       const struct sysctlnode *cnode;

       sysctl_createv(NULL, 0, NULL, &rnode,
               CTLFLAG_PERMANENT,
               CTLTYPE_NODE, "iwi",
               NULL,
               NULL, 0,
               NULL, 0,
               CTL_HW, CTL_CREATE, CTL_EOL);

       sysctl_createv(NULL, 0, &rnode, &cnode,
               CTLFLAG_PERMANENT | CTLFLAG_READWRITE,
               CTLTYPE_INT, "accept_eula",
               SYSCTL_DESCR("Accept Intel EULA and permit use of iwi(4) firmware"),
               NULL, 0,
               &iwi_accept_eula, sizeof(iwi_accept_eula),
               CTL_CREATE, CTL_EOL);
}