drivers/wlan/third_party/atheros/ath10k/p2p.c

/*
 * Copyright (c) 2015 Qualcomm Atheros, Inc.
 *
 * Permission to use, copy, modify, and/or 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 "p2p.h"

#include "core.h"
#include "mac.h"
#include "wmi.h"

static void ath10k_p2p_noa_ie_fill(uint8_t* data, size_t len, const struct wmi_p2p_noa_info* noa) {
    struct ieee80211_p2p_noa_attr* noa_attr;
    uint8_t ctwindow_oppps = noa->ctwindow_oppps;
    uint8_t ctwindow = ctwindow_oppps >> WMI_P2P_OPPPS_CTWINDOW_OFFSET;
    bool oppps = !!(ctwindow_oppps & WMI_P2P_OPPPS_ENABLE_BIT);
    uint16_t* noa_attr_len;
    uint16_t attr_len;
    uint8_t noa_descriptors = noa->num_descriptors;
    int i;

    /* P2P IE */
    data[0] = WLAN_EID_VENDOR_SPECIFIC;
    data[1] = len - 2;
    data[2] = (WLAN_OUI_WFA >> 16) & 0xff;
    data[3] = (WLAN_OUI_WFA >> 8) & 0xff;
    data[4] = (WLAN_OUI_WFA >> 0) & 0xff;
    data[5] = WLAN_OUI_TYPE_WFA_P2P;

    /* NOA ATTR */
    data[6] = IEEE80211_P2P_ATTR_ABSENCE_NOTICE;
    noa_attr_len = (uint16_t*)&data[7]; /* 2 bytes */
    noa_attr = (struct ieee80211_p2p_noa_attr*)&data[9];

    noa_attr->index = noa->index;
    noa_attr->oppps_ctwindow = ctwindow;
    if (oppps) { noa_attr->oppps_ctwindow |= IEEE80211_P2P_OPPPS_ENABLE_BIT; }

    for (i = 0; i < noa_descriptors; i++) {
        noa_attr->desc[i].count = noa->descriptors[i].type_count;
        noa_attr->desc[i].duration = noa->descriptors[i].duration;
        noa_attr->desc[i].interval = noa->descriptors[i].interval;
        noa_attr->desc[i].start_time = noa->descriptors[i].start_time;
    }

    attr_len = 2; /* index + oppps_ctwindow */
    attr_len += noa_descriptors * sizeof(struct ieee80211_p2p_noa_desc);
    *noa_attr_len = attr_len;
}

static size_t ath10k_p2p_noa_ie_len_compute(const struct wmi_p2p_noa_info* noa) {
    size_t len = 0;

    if (!noa->num_descriptors && !(noa->ctwindow_oppps & WMI_P2P_OPPPS_ENABLE_BIT)) { return 0; }

    len += 1 + 1 + 4; /* EID + len + OUI */
    len += 1 + 2;     /* noa attr + attr len */
    len += 1 + 1;     /* index + oppps_ctwindow */
    len += noa->num_descriptors * sizeof(struct ieee80211_p2p_noa_desc);

    return len;
}

static void ath10k_p2p_noa_ie_assign(struct ath10k_vif* arvif, void* ie, size_t len) {
    struct ath10k* ar = arvif->ar;

    ASSERT_MTX_HELD(&ar->data_lock);

    kfree(arvif->u.ap.noa_data);

    arvif->u.ap.noa_data = ie;
    arvif->u.ap.noa_len = len;
}

static void __ath10k_p2p_noa_update(struct ath10k_vif* arvif, const struct wmi_p2p_noa_info* noa) {
    struct ath10k* ar = arvif->ar;
    void* ie;
    size_t len;

    ASSERT_MTX_HELD(&ar->data_lock);

    ath10k_p2p_noa_ie_assign(arvif, NULL, 0);

    len = ath10k_p2p_noa_ie_len_compute(noa);
    if (!len) { return; }

    ie = kmalloc(len, GFP_ATOMIC);
    if (!ie) { return; }

    ath10k_p2p_noa_ie_fill(ie, len, noa);
    ath10k_p2p_noa_ie_assign(arvif, ie, len);
}

void ath10k_p2p_noa_update(struct ath10k_vif* arvif, const struct wmi_p2p_noa_info* noa) {
    struct ath10k* ar = arvif->ar;

    mtx_lock(&ar->data_lock);
    __ath10k_p2p_noa_update(arvif, noa);
    mtx_unlock(&ar->data_lock);
}

struct ath10k_p2p_noa_arg {
    uint32_t vdev_id;
    const struct wmi_p2p_noa_info* noa;
};

static void ath10k_p2p_noa_update_vdev_iter(void* data, uint8_t* mac, struct ieee80211_vif* vif) {
    struct ath10k_vif* arvif = (void*)vif->drv_priv;
    struct ath10k_p2p_noa_arg* arg = data;

    if (arvif->vdev_id != arg->vdev_id) { return; }

    ath10k_p2p_noa_update(arvif, arg->noa);
}

void ath10k_p2p_noa_update_by_vdev_id(struct ath10k* ar, uint32_t vdev_id,
                                      const struct wmi_p2p_noa_info* noa) {
    struct ath10k_p2p_noa_arg arg = {
        .vdev_id = vdev_id,
        .noa = noa,
    };

    ieee80211_iterate_active_interfaces_atomic(ar->hw, IEEE80211_IFACE_ITER_NORMAL,
                                               ath10k_p2p_noa_update_vdev_iter, &arg);
}