io/net80211/net80211_output.c

/*
 * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 *
 * Copyright 2016 Hans Rosenfeld <rosenfeld@grumpf.hope-2000.org>
 */

/*
 * Copyright (c) 2001 Atsushi Onoe
 * Copyright (c) 2002-2008 Sam Leffler, Errno Consulting
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * Alternatively, this software may be distributed under the terms of the
 * GNU General Public License ("GPL") version 2 as published by the Free
 * Software Foundation.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/*
 * Send out 802.11 frames
 */

#include <sys/byteorder.h>
#include <sys/strsun.h>
#include "net80211_impl.h"

/*
 * Set the direction field and address fields of an outgoing
 * non-QoS frame.  Note this should be called early on in
 * constructing a frame as it sets i_fc[1]; other bits can
 * then be or'd in.
 */
static void
ieee80211_send_setup(ieee80211com_t *ic, ieee80211_node_t *in,
    struct ieee80211_frame *wh, int type, const uint8_t *sa, const uint8_t *da,
    const uint8_t *bssid)
{
    wh->i_fc[0] = (uint8_t)(IEEE80211_FC0_VERSION_0 | type);
    if ((type & IEEE80211_FC0_TYPE_MASK) == IEEE80211_FC0_TYPE_DATA) {
        switch (ic->ic_opmode) {
        case IEEE80211_M_STA:
            wh->i_fc[1] = IEEE80211_FC1_DIR_TODS;
            IEEE80211_ADDR_COPY(wh->i_addr1, bssid);
            IEEE80211_ADDR_COPY(wh->i_addr2, sa);
            IEEE80211_ADDR_COPY(wh->i_addr3, da);
            break;
        case IEEE80211_M_IBSS:
        case IEEE80211_M_AHDEMO:
            wh->i_fc[1] = IEEE80211_FC1_DIR_NODS;
            IEEE80211_ADDR_COPY(wh->i_addr1, da);
            IEEE80211_ADDR_COPY(wh->i_addr2, sa);
            IEEE80211_ADDR_COPY(wh->i_addr3, bssid);
            break;
        default:
            ieee80211_err("ieee80211_send_setup: "
                "Invalid mode %u\n", ic->ic_opmode);
            return;
        }
    } else {
        wh->i_fc[1] = IEEE80211_FC1_DIR_NODS;
        IEEE80211_ADDR_COPY(wh->i_addr1, da);
        IEEE80211_ADDR_COPY(wh->i_addr2, sa);
        IEEE80211_ADDR_COPY(wh->i_addr3, bssid);
    }
    *(uint16_t *)&wh->i_dur[0] = 0; /* set duration */
    /* NB: use non-QoS tid */
    *(uint16_t *)&wh->i_seq[0] =
        LE_16(in->in_txseqs[IEEE80211_NONQOS_TID] <<
        IEEE80211_SEQ_SEQ_SHIFT);
    in->in_txseqs[IEEE80211_NONQOS_TID]++;
}

/*
 * Send a management frame to the specified node.  The node pointer
 * must have a reference as the pointer will be passed to the driver
 * and potentially held for a long time.  If the frame is successfully
 * dispatched to the driver, then it is responsible for freeing the
 * reference (and potentially free'ing up any associated storage).
 *
 * Return 0 on success
 */
int
ieee80211_mgmt_output(ieee80211com_t *ic, ieee80211_node_t *in, mblk_t *mp,
    int type, int timer)
{
    ieee80211_impl_t *im = ic->ic_private;
    struct ieee80211_frame *wh;

    ASSERT(in != NULL);

    wh = (struct ieee80211_frame *)mp->b_rptr;
    ieee80211_send_setup(ic, in, wh, IEEE80211_FC0_TYPE_MGT | type,
        ic->ic_macaddr, in->in_macaddr, in->in_bssid);
    if (in->in_challenge != NULL)
        wh->i_fc[1] |= IEEE80211_FC1_WEP;

    if (timer > 0) {
        /*
         * Set the mgt frame timeout.
         */
        im->im_mgt_timer = timer;
        ieee80211_start_watchdog(ic, 1);
    }
    return ((*ic->ic_xmit)(ic, mp, IEEE80211_FC0_TYPE_MGT));
}

/*
 * Send a null data frame to the specified node.
 *
 * NB: the caller is assumed to have setup a node reference
 *     for use; this is necessary to deal with a race condition
 *     when probing for inactive stations.
 */
int
ieee80211_send_nulldata(ieee80211_node_t *in)
{
    ieee80211com_t *ic = in->in_ic;
    mblk_t *m;
    struct ieee80211_frame *wh;
    uint8_t *frm;

    m = ieee80211_getmgtframe(&frm, 0);
    if (m == NULL) {
        ic->ic_stats.is_tx_nobuf++;
        return (ENOMEM);
    }

    wh = (struct ieee80211_frame *)m->b_rptr;
    ieee80211_send_setup(ic, in, wh,
        IEEE80211_FC0_TYPE_DATA | IEEE80211_FC0_SUBTYPE_NODATA,
        ic->ic_macaddr, in->in_macaddr, in->in_bssid);
    /* NB: power management bit is never sent by an AP */
    if ((in->in_flags & IEEE80211_NODE_PWR_MGT) &&
        ic->ic_opmode != IEEE80211_M_HOSTAP)
        wh->i_fc[1] |= IEEE80211_FC1_PWR_MGT;
    m->b_wptr = m->b_rptr + sizeof (struct ieee80211_frame);

    ieee80211_dbg(IEEE80211_MSG_DEBUG | IEEE80211_MSG_DUMPPKTS, "net80211: "
        "send null data frame on channel %u, pwr mgt %s\n",
        ieee80211_macaddr_sprintf(in->in_macaddr),
        ieee80211_chan2ieee(ic, ic->ic_curchan),
        wh->i_fc[1] & IEEE80211_FC1_PWR_MGT ? "ena" : "dis");

    (void) (*ic->ic_xmit)(ic, m, IEEE80211_FC0_TYPE_MGT);

    return (0);
}

/*
 * Encapsulate an outbound data frame for GLDv3 based driver.
 * Fill in the variable part of the 80211 frame
 */
/* ARGSUSED */
mblk_t *
ieee80211_encap(ieee80211com_t *ic, mblk_t *mp, ieee80211_node_t *in)
{
    struct ieee80211_frame  *wh;
    struct ieee80211_key *key;
    int addqos, ac, tid;

    ASSERT(mp != NULL && MBLKL(mp) >= sizeof (struct ieee80211_frame));
    /*
     * Some ap's don't handle QoS-encapsulated EAPOL
     * frames so suppress use.  This may be an issue if other
     * ap's require all data frames to be QoS-encapsulated
     * once negotiated in which case we'll need to make this
     * configurable.
     */
    addqos = in->in_flags & (IEEE80211_NODE_QOS | IEEE80211_NODE_HT);
    wh = (struct ieee80211_frame *)mp->b_rptr;
    *(uint16_t *)wh->i_dur = 0;
    if (addqos) {
        struct ieee80211_qosframe *qwh =
            (struct ieee80211_qosframe *)wh;

        ac = ieee80211_classify(ic, mp, in);
        /* map from access class/queue to 11e header priorty value */
        tid = WME_AC_TO_TID(ac);
        qwh->i_qos[0] = tid & IEEE80211_QOS_TID;
        /*
         * Check if A-MPDU tx aggregation is setup or if we
         * should try to enable it.  The sta must be associated
         * with HT and A-MPDU enabled for use.  On the first
         * frame that goes out We issue an ADDBA request and
         * wait for a reply.  The frame being encapsulated
         * will go out w/o using A-MPDU, or possibly it might
         * be collected by the driver and held/retransmit.
         * ieee80211_ampdu_request handles staggering requests
         * in case the receiver NAK's us or we are otherwise
         * unable to establish a BA stream.
         */
        if ((in->in_flags & IEEE80211_NODE_AMPDU_TX) &&
            (ic->ic_flags_ext & IEEE80211_FEXT_AMPDU_TX)) {
            struct ieee80211_tx_ampdu *tap = &in->in_tx_ampdu[ac];

            if (IEEE80211_AMPDU_RUNNING(tap)) {
                /*
                 * Operational, mark frame for aggregation.
                 */
                qwh->i_qos[0] |= IEEE80211_QOS_ACKPOLICY_BA;
            } else if (!IEEE80211_AMPDU_REQUESTED(tap)) {
                /*
                 * Not negotiated yet, request service.
                 */
                (void) ieee80211_ampdu_request(in, tap);
            }
        }
        /* works even when BA marked above */
        if (ic->ic_wme.wme_wmeChanParams.cap_wmeParams[ac].
            wmep_noackPolicy) {
            qwh->i_qos[0] |= IEEE80211_QOS_ACKPOLICY_NOACK;
        }

        *(uint16_t *)wh->i_seq =
            LE_16(in->in_txseqs[tid] << IEEE80211_SEQ_SEQ_SHIFT);
        in->in_txseqs[tid]++;
    } else {
        *(uint16_t *)wh->i_seq =
            LE_16(in->in_txseqs[IEEE80211_NONQOS_TID] <<
            IEEE80211_SEQ_SEQ_SHIFT);
        in->in_txseqs[IEEE80211_NONQOS_TID]++;
    }

    if (ic->ic_flags & IEEE80211_F_PRIVACY)
        key = ieee80211_crypto_getkey(ic);
    else
        key = NULL;

    /*
     * IEEE 802.1X: send EAPOL frames always in the clear.
     * WPA/WPA2: encrypt EAPOL keys when pairwise keys are set.
     */
    if (key != NULL && (ic->ic_flags & IEEE80211_F_WPA)) {
        wh->i_fc[1] |= IEEE80211_FC1_WEP;
        if (!ieee80211_crypto_enmic(isc, key, mp, 0))
            ieee80211_err("ieee80211_crypto_enmic failed.\n");
    }

    return (mp);
}

/*
 * Add supported rates information element to a frame.
 */
static uint8_t *
ieee80211_add_rates(uint8_t *frm, const struct ieee80211_rateset *rs)
{
    uint8_t nrates;

    *frm++ = IEEE80211_ELEMID_RATES;
    nrates = rs->ir_nrates;
    if (nrates > IEEE80211_RATE_SIZE)
        nrates = IEEE80211_RATE_SIZE;
    *frm++ = nrates;
    bcopy(rs->ir_rates, frm, nrates);
    return (frm + nrates);
}

/*
 * Add extended supported rates element to a frame, usually for 11g mode
 */
static uint8_t *
ieee80211_add_xrates(uint8_t *frm, const struct ieee80211_rateset *rs)
{
    if (rs->ir_nrates > IEEE80211_RATE_SIZE) {
        uint8_t nrates = rs->ir_nrates - IEEE80211_RATE_SIZE;

        *frm++ = IEEE80211_ELEMID_XRATES;
        *frm++ = nrates;
        bcopy(rs->ir_rates + IEEE80211_RATE_SIZE, frm, nrates);
        frm += nrates;
    }
    return (frm);
}

#define WME_OUI_BYTES       0x00, 0x50, 0xf2
/*
 * Add a WME information element to a frame.
 */
/* ARGSUSED */
static uint8_t *
ieee80211_add_wme_info(uint8_t *frm, struct ieee80211_wme_state *wme)
{
    static const struct ieee80211_wme_info info = {
        .wme_id     = IEEE80211_ELEMID_VENDOR,
        .wme_len    = sizeof (struct ieee80211_wme_info) - 2,
        .wme_oui    = { WME_OUI_BYTES },
        .wme_type   = WME_OUI_TYPE,
        .wme_subtype    = WME_INFO_OUI_SUBTYPE,
        .wme_version    = WME_VERSION,
        .wme_info   = 0,
    };
    (void) memcpy(frm, &info, sizeof (info));
    return (frm + sizeof (info));
}

/*
 * Add a WME parameters element to a frame.
 */
static uint8_t *
ieee80211_add_wme_param(uint8_t *frm, struct ieee80211_wme_state *wme)
{
#define SM(_v, _f)  (((_v) << _f##_S) & _f)
#define ADDSHORT(frm, v) do {           \
    _NOTE(CONSTCOND)            \
    frm[0] = (v) & 0xff;            \
    frm[1] = (v) >> 8;          \
    frm += 2;               \
    _NOTE(CONSTCOND)            \
} while (0)
    /* NB: this works 'cuz a param has an info at the front */
    static const struct ieee80211_wme_info param = {
        .wme_id     = IEEE80211_ELEMID_VENDOR,
        .wme_len    = sizeof (struct ieee80211_wme_param) - 2,
        .wme_oui    = { WME_OUI_BYTES },
        .wme_type   = WME_OUI_TYPE,
        .wme_subtype    = WME_PARAM_OUI_SUBTYPE,
        .wme_version    = WME_VERSION,
    };
    int i;

    (void) memcpy(frm, &param, sizeof (param));
    frm += offsetof(struct ieee80211_wme_info, wme_info);
    *frm++ = wme->wme_bssChanParams.cap_info;   /* AC info */
    *frm++ = 0;                 /* reserved field */
    for (i = 0; i < WME_NUM_AC; i++) {
        const struct wmeParams *ac =
            &wme->wme_bssChanParams.cap_wmeParams[i];
        *frm++ = SM(i, WME_PARAM_ACI)
            | SM(ac->wmep_acm, WME_PARAM_ACM)
            | SM(ac->wmep_aifsn, WME_PARAM_AIFSN);
        *frm++ = SM(ac->wmep_logcwmax, WME_PARAM_LOGCWMAX)
            | SM(ac->wmep_logcwmin, WME_PARAM_LOGCWMIN);
        ADDSHORT(frm, ac->wmep_txopLimit);
    }
    return (frm);
#undef SM
#undef ADDSHORT
}
#undef WME_OUI_BYTES

/*
 * Add SSID element to a frame
 */
static uint8_t *
ieee80211_add_ssid(uint8_t *frm, const uint8_t *ssid, uint32_t len)
{
    *frm++ = IEEE80211_ELEMID_SSID;
    *frm++ = (uint8_t)len;
    bcopy(ssid, frm, len);
    return (frm + len);
}

/*
 * Add an erp element to a frame.
 */
static uint8_t *
ieee80211_add_erp(uint8_t *frm, ieee80211com_t *ic)
{
    uint8_t erp;

    *frm++ = IEEE80211_ELEMID_ERP;
    *frm++ = 1;
    erp = 0;
    if (ic->ic_flags & IEEE80211_F_USEPROT)
        erp |= IEEE80211_ERP_USE_PROTECTION;
    if (ic->ic_flags & IEEE80211_F_USEBARKER)
        erp |= IEEE80211_ERP_LONG_PREAMBLE;
    *frm++ = erp;
    return (frm);
}

/*
 * Get capability information from the interface softc, ic.
 */
static uint16_t
ieee80211_get_capinfo(ieee80211com_t *ic)
{
    uint16_t capinfo;

    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(ic->ic_curchan)) {
        capinfo |= IEEE80211_CAPINFO_SHORT_PREAMBLE;
    }
    if (ic->ic_flags & IEEE80211_F_SHSLOT)
        capinfo |= IEEE80211_CAPINFO_SHORT_SLOTTIME;

    return (capinfo);
}

/*
 * Send a probe request frame with the specified ssid
 * and any optional information element data.
 */
int
ieee80211_send_probereq(ieee80211_node_t *in,
    const uint8_t *sa, const uint8_t *da, const uint8_t *bssid,
    const uint8_t *ssid, size_t ssidlen, const void *optie, size_t optielen)
{
    mblk_t *mp;
    ieee80211com_t *ic = in->in_ic;
    enum ieee80211_phymode mode;
    struct ieee80211_frame *wh;
    uint8_t *frm;

    /*
     * prreq frame format ([tlv] - 1 byte element ID + 1 byte length)
     *  [tlv] ssid
     *  [tlv] supported rates
     *  [tlv] extended supported rates
     *  [tlv] user-specified ie's
     */
    mp = ieee80211_getmgtframe(&frm,
        2 + IEEE80211_NWID_LEN
        + 2 + IEEE80211_RATE_SIZE +
        + 2 + IEEE80211_XRATE_SIZE
        + optielen);
    if (mp == NULL)
        return (ENOMEM);

    frm = ieee80211_add_ssid(frm, ssid, ssidlen);
    mode = ieee80211_chan2mode(ic, ic->ic_curchan);
    frm = ieee80211_add_rates(frm, &ic->ic_sup_rates[mode]);
    frm = ieee80211_add_xrates(frm, &ic->ic_sup_rates[mode]);
    if (optie != NULL) {
        (void) memcpy(frm, optie, optielen);
        frm += optielen;
    }
    mp->b_wptr = frm;

    wh = (struct ieee80211_frame *)mp->b_rptr;
    ieee80211_send_setup(ic, in, wh,
        IEEE80211_FC0_TYPE_MGT | IEEE80211_FC0_SUBTYPE_PROBE_REQ,
        sa, da, bssid);

    ieee80211_dbg(IEEE80211_MSG_DEBUG | IEEE80211_MSG_DUMPPKTS,
        "[%s] send probe req on channel %u\n",
        ieee80211_macaddr_sprintf(wh->i_addr1),
        ieee80211_chan2ieee(ic, ic->ic_curchan));

    (void) (*ic->ic_xmit)(ic, mp, IEEE80211_FC0_TYPE_MGT);
    return (0);
}

/*
 * Send a management frame.  The node is for the destination (or ic_bss
 * when in station mode).  Nodes other than ic_bss have their reference
 * count bumped to reflect our use for an indeterminant time.
 */
int
ieee80211_send_mgmt(ieee80211com_t *ic, ieee80211_node_t *in, int type, int arg)
{
    mblk_t *mp;
    uint8_t *frm;
    uint16_t capinfo;
    struct ieee80211_key *key;
    boolean_t has_challenge;
    boolean_t is_shared_key;
    int ret;
    int timer;
    int status;

    ASSERT(in != NULL);

    timer = 0;
    switch (type) {
    case IEEE80211_FC0_SUBTYPE_PROBE_RESP:
        /*
         * probe response frame format
         *  [8] time stamp
         *  [2] beacon interval
         *  [2] capability information
         *  [tlv] ssid
         *  [tlv] supported rates
         *  [tlv] parameter set (FH/DS)
         *  [tlv] parameter set (IBSS)
         *  [tlv] extended rate phy (ERP)
         *  [tlv] extended supported rates
         *  [tlv] WPA
         *  [tlv] WME (optional)
         *  [tlv] HT capabilities
         *  [tlv] HT information
         *  [tlv] Vendor OUI HT capabilities (optional)
         *  [tlv] Vendor OUI HT information (optional)
         */
        mp = ieee80211_getmgtframe(&frm,
            8           /* time stamp  */
            + sizeof (uint16_t) /* beacon interval  */
            + sizeof (uint16_t) /* capability  */
            + 2 + IEEE80211_NWID_LEN
            + 2 + IEEE80211_RATE_SIZE
            + 2 + IEEE80211_FH_LEN
            + 2 + IEEE80211_IBSS_LEN
            + 2 + IEEE80211_ERP_LEN
            + 2 + IEEE80211_XRATE_SIZE
            + (ic->ic_flags & IEEE80211_F_WPA ?
            2 * sizeof (struct ieee80211_ie_wpa) : 0)
                    /* [tlv] WPA  */
            + (ic->ic_flags & IEEE80211_F_WME ?
            sizeof (struct ieee80211_wme_param) : 0)
                    /* [tlv] WME  */
            /* check for cluster requirement */
            + 2 * sizeof (struct ieee80211_ie_htcap) + 4
            + 2 * sizeof (struct ieee80211_ie_htinfo) + 4);

        if (mp == NULL)
            return (ENOMEM);

        bzero(frm, 8);  /* timestamp should be filled later */
        frm += 8;
        *(uint16_t *)frm = LE_16(ic->ic_bss->in_intval);
        frm += 2;
        capinfo = ieee80211_get_capinfo(ic);
        *(uint16_t *)frm = LE_16(capinfo);
        frm += 2;

        /* ssid */
        frm = ieee80211_add_ssid(frm, ic->ic_bss->in_essid,
            ic->ic_bss->in_esslen);
        /* supported rates */
        frm = ieee80211_add_rates(frm, &in->in_rates);

        if (IEEE80211_IS_CHAN_FHSS(ic->ic_curchan)) {
            *frm++ = IEEE80211_ELEMID_FHPARMS;
            *frm++ = IEEE80211_FH_LEN;
            *frm++ = in->in_fhdwell & 0x00ff;
            *frm++ = (in->in_fhdwell >> 8) & 0x00ff;
            *frm++ = IEEE80211_FH_CHANSET(
                ieee80211_chan2ieee(ic, ic->ic_curchan));
            *frm++ = IEEE80211_FH_CHANPAT(
                ieee80211_chan2ieee(ic, ic->ic_curchan));
            *frm++ = in->in_fhindex;
        } else {
            *frm++ = IEEE80211_ELEMID_DSPARMS;
            *frm++ = IEEE80211_DS_LEN;
            *frm++ = ieee80211_chan2ieee(ic, ic->ic_curchan);
        }

        if (ic->ic_opmode == IEEE80211_M_IBSS) {
            *frm++ = IEEE80211_ELEMID_IBSSPARMS;
            *frm++ = IEEE80211_IBSS_LEN;
            *frm++ = 0; *frm++ = 0;     /* ATIM window */
        }
        if (IEEE80211_IS_CHAN_ANYG(ic->ic_curchan))
            frm = ieee80211_add_erp(frm, ic);
        frm = ieee80211_add_xrates(frm, &in->in_rates);
        /*
         * NB: legacy 11b clients do not get certain ie's.
         * The caller identifies such clients by passing
         * a token in arg to us.  Could expand this to be
         * any legacy client for stuff like HT ie's.
         */
        if (IEEE80211_IS_CHAN_HT(ic->ic_curchan) &&
            arg != IEEE80211_SEND_LEGACY_11B) {
            frm = ieee80211_add_htcap(frm, in);
            frm = ieee80211_add_htinfo(frm, in);
        }
        if (ic->ic_flags & IEEE80211_F_WME)
            frm = ieee80211_add_wme_param(frm, &ic->ic_wme);
        if (IEEE80211_IS_CHAN_HT(ic->ic_curchan) &&
            (ic->ic_flags_ext & IEEE80211_FEXT_HTCOMPAT) &&
            arg != IEEE80211_SEND_LEGACY_11B) {
            frm = ieee80211_add_htcap_vendor(frm, in);
            frm = ieee80211_add_htinfo_vendor(frm, in);
        }
        mp->b_wptr = frm;   /* allocated is greater than used */

        break;

    case IEEE80211_FC0_SUBTYPE_AUTH:
        status = arg >> 16;
        arg &= 0xffff;
        has_challenge = ((arg == IEEE80211_AUTH_SHARED_CHALLENGE ||
            arg == IEEE80211_AUTH_SHARED_RESPONSE) &&
            in->in_challenge != NULL);

        /*
         * Deduce whether we're doing open authentication or
         * shared key authentication.  We do the latter if
         * we're in the middle of a shared key authentication
         * handshake or if we're initiating an authentication
         * request and configured to use shared key.
         */
        is_shared_key = has_challenge ||
            arg >= IEEE80211_AUTH_SHARED_RESPONSE ||
            (arg == IEEE80211_AUTH_SHARED_REQUEST &&
            ic->ic_bss->in_authmode == IEEE80211_AUTH_SHARED);

        if (has_challenge && status == IEEE80211_STATUS_SUCCESS)
            key = ieee80211_crypto_getkey(ic);
        else
            key = NULL;

        mp = ieee80211_getmgtframe(&frm,
            3 * sizeof (uint16_t)
            + (has_challenge && status == IEEE80211_STATUS_SUCCESS ?
            sizeof (uint16_t) + IEEE80211_CHALLENGE_LEN : 0)
            + (key != NULL ? key->wk_cipher->ic_header : 0));
        if (mp == NULL)
            return (ENOMEM);

        if (key != NULL)
            frm += key->wk_cipher->ic_header;

        ((uint16_t *)frm)[0] =
            (is_shared_key) ? LE_16(IEEE80211_AUTH_ALG_SHARED)
            : LE_16(IEEE80211_AUTH_ALG_OPEN);
        ((uint16_t *)frm)[1] = LE_16(arg);  /* sequence number */
        ((uint16_t *)frm)[2] = LE_16(status);   /* status */

        if (has_challenge && status == IEEE80211_STATUS_SUCCESS) {
            frm += IEEE80211_AUTH_ELEM_MIN;
            *frm = IEEE80211_ELEMID_CHALLENGE;
            frm++;
            *frm = IEEE80211_CHALLENGE_LEN;
            frm++;
            bcopy(in->in_challenge, frm, IEEE80211_CHALLENGE_LEN);
        }

        if (ic->ic_opmode == IEEE80211_M_STA)
            timer = IEEE80211_TRANS_WAIT;
        break;

    case IEEE80211_FC0_SUBTYPE_DEAUTH:
        mp = ieee80211_getmgtframe(&frm, sizeof (uint16_t));
        if (mp == NULL)
            return (ENOMEM);

        *(uint16_t *)frm = LE_16(arg);  /* reason */

        ieee80211_node_unauthorize(in); /* port closed */
        break;

    case IEEE80211_FC0_SUBTYPE_ASSOC_REQ:
    case IEEE80211_FC0_SUBTYPE_REASSOC_REQ:
        /*
         * asreq frame format
         *  [2] capability information
         *  [2] listen interval
         *  [6*] current AP address (reassoc only)
         *  [tlv] ssid
         *  [tlv] supported rates
         *  [tlv] extended supported rates
         *  [tlv] WME
         *  [tlv] HT capabilities
         *  [tlv] Vendor OUI HT capabilities (optional)
         *  [tlv] user-specified ie's
         */
        mp = ieee80211_getmgtframe(&frm,
            sizeof (uint16_t)
            + sizeof (uint16_t) + IEEE80211_ADDR_LEN
            + 2 + IEEE80211_NWID_LEN
            + 2 + IEEE80211_RATE_SIZE
            + 2 + IEEE80211_XRATE_SIZE
            + sizeof (struct ieee80211_wme_info)
            + 2 * sizeof (struct ieee80211_ie_htcap) + 4
            + ic->ic_opt_ie_len);
        if (mp == NULL)
            return (ENOMEM);

        capinfo = ieee80211_get_capinfo(ic);
        if (!(in->in_capinfo & IEEE80211_CAPINFO_SHORT_SLOTTIME) ||
            !(ic->ic_caps & IEEE80211_C_SHSLOT)) {
            capinfo &= ~IEEE80211_CAPINFO_SHORT_SLOTTIME;
        } else {
            capinfo |= IEEE80211_CAPINFO_SHORT_SLOTTIME;
        }
        if (!(in->in_capinfo & IEEE80211_CAPINFO_SHORT_PREAMBLE) ||
            !(ic->ic_caps & IEEE80211_C_SHPREAMBLE)) {
            capinfo &= ~IEEE80211_CAPINFO_SHORT_PREAMBLE;
        } else {
            capinfo |= IEEE80211_CAPINFO_SHORT_PREAMBLE;
        }
        *(uint16_t *)frm = LE_16(capinfo);
        frm += 2;

        *(uint16_t *)frm = LE_16(ic->ic_lintval);
        frm += 2;

        if (type == IEEE80211_FC0_SUBTYPE_REASSOC_REQ) {
            IEEE80211_ADDR_COPY(frm, ic->ic_bss->in_bssid);
            frm += IEEE80211_ADDR_LEN;
        }

        frm = ieee80211_add_ssid(frm, in->in_essid, in->in_esslen);
        frm = ieee80211_add_rates(frm, &in->in_rates);
        frm = ieee80211_add_xrates(frm, &in->in_rates);
        if ((ic->ic_flags_ext & IEEE80211_FEXT_HT) &&
            in->in_htcap_ie != NULL &&
            in->in_htcap_ie[0] == IEEE80211_ELEMID_HTCAP)
            frm = ieee80211_add_htcap(frm, in);
        if ((ic->ic_flags & IEEE80211_F_WME) && in->in_wme_ie != NULL)
            frm = ieee80211_add_wme_info(frm, &ic->ic_wme);
        if ((ic->ic_flags_ext & IEEE80211_FEXT_HT) &&
            in->in_htcap_ie != NULL &&
            in->in_htcap_ie[0] == IEEE80211_ELEMID_VENDOR)
            frm = ieee80211_add_htcap_vendor(frm, in);
        if (ic->ic_opt_ie != NULL) {
            bcopy(ic->ic_opt_ie, frm, ic->ic_opt_ie_len);
            frm += ic->ic_opt_ie_len;
        }
        mp->b_wptr = frm;   /* allocated is greater than used */

        timer = IEEE80211_TRANS_WAIT;
        break;

    case IEEE80211_FC0_SUBTYPE_ASSOC_RESP:
    case IEEE80211_FC0_SUBTYPE_REASSOC_RESP:
        /*
         * asreq frame format
         *  [2] capability information
         *  [2] status
         *  [2] association ID
         *  [tlv] supported rates
         *  [tlv] extended supported rates
         *  [tlv] WME (if enabled and STA enabled)
         *  [tlv] HT capabilities (standard or vendor OUI)
         *  [tlv] HT information (standard or vendor OUI)
         */
        mp = ieee80211_getmgtframe(&frm,
            3 * sizeof (uint16_t)
            + 2 + IEEE80211_RATE_SIZE
            + 2 + IEEE80211_XRATE_SIZE);
        if (mp == NULL)
            return (ENOMEM);

        capinfo = ieee80211_get_capinfo(ic);
        *(uint16_t *)frm = LE_16(capinfo);
        frm += 2;

        *(uint16_t *)frm = LE_16(arg);  /* status */
        frm += 2;

        if (arg == IEEE80211_STATUS_SUCCESS)
            *(uint16_t *)frm = LE_16(in->in_associd);
        else
            *(uint16_t *)frm = LE_16(0);
        frm += 2;

        frm = ieee80211_add_rates(frm, &in->in_rates);
        frm = ieee80211_add_xrates(frm, &in->in_rates);
        mp->b_wptr = frm;
        break;

    case IEEE80211_FC0_SUBTYPE_DISASSOC:
        mp = ieee80211_getmgtframe(&frm, sizeof (uint16_t));
        if (mp == NULL)
            return (ENOMEM);
        *(uint16_t *)frm = LE_16(arg);  /* reason */
        break;

    default:
        ieee80211_dbg(IEEE80211_MSG_ANY,
            "[%s] invalid mgmt frame type %u\n",
            ieee80211_macaddr_sprintf(in->in_macaddr), type);
        return (EINVAL);
    } /* type */
    ret = ieee80211_mgmt_output(ic, in, mp, type, timer);
    return (ret);
}

/*
 * Allocate a beacon frame and fillin the appropriate bits.
 */
mblk_t *
ieee80211_beacon_alloc(ieee80211com_t *ic, ieee80211_node_t *in,
    struct ieee80211_beacon_offsets *bo)
{
    struct ieee80211_frame *wh;
    struct ieee80211_rateset *rs;
    mblk_t *m;
    uint8_t *frm;
    int pktlen;
    uint16_t capinfo;

    IEEE80211_LOCK(ic);
    /*
     * beacon frame format
     *  [8] time stamp
     *  [2] beacon interval
     *  [2] cabability information
     *  [tlv] ssid
     *  [tlv] supported rates
     *  [3] parameter set (DS)
     *  [tlv] parameter set (IBSS/TIM)
     *  [tlv] extended rate phy (ERP)
     *  [tlv] extended supported rates
     *  [tlv] WME parameters
     *  [tlv] WPA/RSN parameters
     *  [tlv] HT capabilities
     *  [tlv] HT information
     *  [tlv] Vendor OUI HT capabilities (optional)
     *  [tlv] Vendor OUI HT information (optional)
     * Vendor-specific OIDs (e.g. Atheros)
     * NB: we allocate the max space required for the TIM bitmap.
     */
    rs = &in->in_rates;
    pktlen =  8         /* time stamp */
        + sizeof (uint16_t)     /* beacon interval */
        + sizeof (uint16_t)     /* capabilities */
        + 2 + in->in_esslen     /* ssid */
        + 2 + IEEE80211_RATE_SIZE   /* supported rates */
        + 2 + 1         /* DS parameters */
        + 2 + 4 + ic->ic_tim_len    /* DTIM/IBSSPARMS */
        + 2 + 1         /* ERP */
        + 2 + IEEE80211_XRATE_SIZE
        + (ic->ic_caps & IEEE80211_C_WME ?  /* WME */
        sizeof (struct ieee80211_wme_param) : 0)
        /* conditional? */
        + 4 + 2 * sizeof (struct ieee80211_ie_htcap)    /* HT caps */
        + 4 + 2 * sizeof (struct ieee80211_ie_htinfo);  /* HT info */

    m = ieee80211_getmgtframe(&frm, pktlen);
    if (m == NULL) {
        ieee80211_dbg(IEEE80211_MSG_ANY, "ieee80211_beacon_alloc: "
            "cannot get buf; size %u\n", pktlen);
        IEEE80211_UNLOCK(ic);
        return (NULL);
    }

    /* timestamp is set by hardware/driver */
    (void) memset(frm, 0, 8);
    frm += 8;
    *(uint16_t *)frm = LE_16(in->in_intval);
    frm += 2;
    capinfo = ieee80211_get_capinfo(ic);
    bo->bo_caps = (uint16_t *)frm;
    *(uint16_t *)frm = LE_16(capinfo);
    frm += 2;
    *frm++ = IEEE80211_ELEMID_SSID;
    if (!(ic->ic_flags & IEEE80211_F_HIDESSID)) {
        *frm++ = in->in_esslen;
        bcopy(in->in_essid, frm, in->in_esslen);
        frm += in->in_esslen;
    } else {
        *frm++ = 0;
    }
    frm = ieee80211_add_rates(frm, rs);
    if (ic->ic_curmode != IEEE80211_MODE_FH) {
        *frm++ = IEEE80211_ELEMID_DSPARMS;
        *frm++ = 1;
        *frm++ = ieee80211_chan2ieee(ic, in->in_chan);
    }
    bo->bo_tim = frm;
    if (ic->ic_opmode == IEEE80211_M_IBSS) {
        *frm++ = IEEE80211_ELEMID_IBSSPARMS;
        *frm++ = 2;
        *frm++ = 0; *frm++ = 0;     /* TODO: ATIM window */
        bo->bo_tim_len = 0;
    } else {
        struct ieee80211_tim_ie *tie =
            (struct ieee80211_tim_ie *)frm;

        tie->tim_ie = IEEE80211_ELEMID_TIM;
        tie->tim_len = 4;   /* length */
        tie->tim_count = 0; /* DTIM count */
        tie->tim_period = IEEE80211_DTIM_DEFAULT;
        tie->tim_bitctl = 0;    /* bitmap control */
        tie->tim_bitmap[0] = 0; /* Partial Virtual Bitmap */
        frm += sizeof (struct ieee80211_tim_ie);
        bo->bo_tim_len = 1;
    }
    bo->bo_trailer = frm;

    if (ic->ic_curmode == IEEE80211_MODE_11G) {
        bo->bo_erp = frm;
        frm = ieee80211_add_erp(frm, ic);
    }
    frm = ieee80211_add_xrates(frm, rs);
    if (IEEE80211_IS_CHAN_HT(ic->ic_curchan)) {
        frm = ieee80211_add_htcap(frm, in);
        bo->bo_htinfo = frm;
        frm = ieee80211_add_htinfo(frm, in);
    }
    if (ic->ic_flags & IEEE80211_F_WME) {
        bo->bo_wme = frm;
        frm = ieee80211_add_wme_param(frm, &ic->ic_wme);
    }
    if (IEEE80211_IS_CHAN_HT(ic->ic_curchan) &&
        (ic->ic_flags_ext & IEEE80211_FEXT_HTCOMPAT)) {
        frm = ieee80211_add_htcap_vendor(frm, in);
        frm = ieee80211_add_htinfo_vendor(frm, in);
    }
    bo->bo_trailer_len = _PTRDIFF(frm, bo->bo_trailer);
    m->b_wptr = frm;

    wh = (struct ieee80211_frame *)m->b_rptr;
    wh->i_fc[0] = IEEE80211_FC0_VERSION_0 | IEEE80211_FC0_TYPE_MGT |
        IEEE80211_FC0_SUBTYPE_BEACON;
    wh->i_fc[1] = IEEE80211_FC1_DIR_NODS;
    *(uint16_t *)wh->i_dur = 0;
    IEEE80211_ADDR_COPY(wh->i_addr1, wifi_bcastaddr);
    IEEE80211_ADDR_COPY(wh->i_addr2, ic->ic_macaddr);
    IEEE80211_ADDR_COPY(wh->i_addr3, in->in_bssid);
    *(uint16_t *)wh->i_seq = 0;

    IEEE80211_UNLOCK(ic);
    return (m);
}

/*
 * Update the dynamic parts of a beacon frame based on the current state.
 */
/* ARGSUSED */
int
ieee80211_beacon_update(ieee80211com_t *ic, ieee80211_node_t *in,
    struct ieee80211_beacon_offsets *bo, mblk_t *mp, int mcast)
{
    uint16_t capinfo;

    IEEE80211_LOCK(ic);

    capinfo = ieee80211_get_capinfo(ic);
    *bo->bo_caps = LE_16(capinfo);

    IEEE80211_UNLOCK(ic);
    return (0);
}

/*
 * Assign priority to a frame based on any vlan tag assigned
 * to the station and/or any Diffserv setting in an IP header.
 * Finally, if an ACM policy is setup (in station mode) it's
 * applied.
 */
/* ARGSUSED */
int
ieee80211_classify(struct ieee80211com *ic, mblk_t *m,
    struct ieee80211_node *ni)
{
    int ac;

    if ((ni->in_flags & IEEE80211_NODE_QOS) == 0)
        return (WME_AC_BE);

    /* Process VLan */
    /* Process IPQoS */

    ac = WME_AC_BE;

    /*
     * Apply ACM policy.
     */
    if (ic->ic_opmode == IEEE80211_M_STA) {
        static const int acmap[4] = {
            WME_AC_BK,  /* WME_AC_BE */
            WME_AC_BK,  /* WME_AC_BK */
            WME_AC_BE,  /* WME_AC_VI */
            WME_AC_VI,  /* WME_AC_VO */
        };
        while (ac != WME_AC_BK &&
            ic->ic_wme.wme_wmeBssChanParams.cap_wmeParams[ac].
            wmep_acm) {
            ac = acmap[ac];
        }
    }

    return (ac);
}