[linux-2.6-block.git] / net / mac80211 / agg-rx.c

// SPDX-License-Identifier: GPL-2.0-only
/*
 * HT handling
 *
 * Copyright 2003, Jouni Malinen <jkmaline@cc.hut.fi>
 * Copyright 2002-2005, Instant802 Networks, Inc.
 * Copyright 2005-2006, Devicescape Software, Inc.
 * Copyright 2006-2007	Jiri Benc <jbenc@suse.cz>
 * Copyright 2007, Michael Wu <flamingice@sourmilk.net>
 * Copyright 2007-2010, Intel Corporation
 * Copyright(c) 2015-2017 Intel Deutschland GmbH
 * Copyright (C) 2018-2024 Intel Corporation
 */

/**
 * DOC: RX A-MPDU aggregation
 *
 * Aggregation on the RX side requires only implementing the
 * @ampdu_action callback that is invoked to start/stop any
 * block-ack sessions for RX aggregation.
 *
 * When RX aggregation is started by the peer, the driver is
 * notified via @ampdu_action function, with the
 * %IEEE80211_AMPDU_RX_START action, and may reject the request
 * in which case a negative response is sent to the peer, if it
 * accepts it a positive response is sent.
 *
 * While the session is active, the device/driver are required
 * to de-aggregate frames and pass them up one by one to mac80211,
 * which will handle the reorder buffer.
 *
 * When the aggregation session is stopped again by the peer or
 * ourselves, the driver's @ampdu_action function will be called
 * with the action %IEEE80211_AMPDU_RX_STOP. In this case, the
 * call must not fail.
 */

#include <linux/ieee80211.h>
#include <linux/slab.h>
#include <linux/export.h>
#include <net/mac80211.h>
#include "ieee80211_i.h"
#include "driver-ops.h"

static void ieee80211_free_tid_rx(struct rcu_head *h)
{
	struct tid_ampdu_rx *tid_rx =
		container_of(h, struct tid_ampdu_rx, rcu_head);
	int i;

	for (i = 0; i < tid_rx->buf_size; i++)
		__skb_queue_purge(&tid_rx->reorder_buf[i]);
	kfree(tid_rx->reorder_buf);
	kfree(tid_rx->reorder_time);
	kfree(tid_rx);
}

void __ieee80211_stop_rx_ba_session(struct sta_info *sta, u16 tid,
				    u16 initiator, u16 reason, bool tx)
{
	struct ieee80211_local *local = sta->local;
	struct tid_ampdu_rx *tid_rx;
	struct ieee80211_ampdu_params params = {
		.sta = &sta->sta,
		.action = IEEE80211_AMPDU_RX_STOP,
		.tid = tid,
		.amsdu = false,
		.timeout = 0,
		.ssn = 0,
	};

	lockdep_assert_wiphy(sta->local->hw.wiphy);

	tid_rx = rcu_dereference_protected(sta->ampdu_mlme.tid_rx[tid],
					lockdep_is_held(&sta->local->hw.wiphy->mtx));

	if (!test_bit(tid, sta->ampdu_mlme.agg_session_valid))
		return;

	RCU_INIT_POINTER(sta->ampdu_mlme.tid_rx[tid], NULL);
	__clear_bit(tid, sta->ampdu_mlme.agg_session_valid);

	ht_dbg(sta->sdata,
	       "Rx BA session stop requested for %pM tid %u %s reason: %d\n",
	       sta->sta.addr, tid,
	       initiator == WLAN_BACK_RECIPIENT ? "recipient" : "initiator",
	       (int)reason);

	if (drv_ampdu_action(local, sta->sdata, &params))
		sdata_info(sta->sdata,
			   "HW problem - can not stop rx aggregation for %pM tid %d\n",
			   sta->sta.addr, tid);

	/* check if this is a self generated aggregation halt */
	if (initiator == WLAN_BACK_RECIPIENT && tx)
		ieee80211_send_delba(sta->sdata, sta->sta.addr,
				     tid, WLAN_BACK_RECIPIENT, reason);

	/*
	 * return here in case tid_rx is not assigned - which will happen if
	 * IEEE80211_HW_SUPPORTS_REORDERING_BUFFER is set.
	 */
	if (!tid_rx)
		return;

	timer_delete_sync(&tid_rx->session_timer);

	/* make sure ieee80211_sta_reorder_release() doesn't re-arm the timer */
	spin_lock_bh(&tid_rx->reorder_lock);
	tid_rx->removed = true;
	spin_unlock_bh(&tid_rx->reorder_lock);
	timer_delete_sync(&tid_rx->reorder_timer);

	call_rcu(&tid_rx->rcu_head, ieee80211_free_tid_rx);
}

void ieee80211_stop_rx_ba_session(struct ieee80211_vif *vif, u16 ba_rx_bitmap,
				  const u8 *addr)
{
	struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
	struct sta_info *sta;
	int i;

	rcu_read_lock();
	sta = sta_info_get_bss(sdata, addr);
	if (!sta) {
		rcu_read_unlock();
		return;
	}

	for (i = 0; i < IEEE80211_NUM_TIDS; i++)
		if (ba_rx_bitmap & BIT(i))
			set_bit(i, sta->ampdu_mlme.tid_rx_stop_requested);

	wiphy_work_queue(sta->local->hw.wiphy, &sta->ampdu_mlme.work);
	rcu_read_unlock();
}
EXPORT_SYMBOL(ieee80211_stop_rx_ba_session);

/*
 * After accepting the AddBA Request we activated a timer,
 * resetting it after each frame that arrives from the originator.
 */
static void sta_rx_agg_session_timer_expired(struct timer_list *t)
{
	struct tid_ampdu_rx *tid_rx = timer_container_of(tid_rx, t,
							 session_timer);
	struct sta_info *sta = tid_rx->sta;
	u8 tid = tid_rx->tid;
	unsigned long timeout;

	timeout = tid_rx->last_rx + TU_TO_JIFFIES(tid_rx->timeout);
	if (time_is_after_jiffies(timeout)) {
		mod_timer(&tid_rx->session_timer, timeout);
		return;
	}

	ht_dbg(sta->sdata, "RX session timer expired on %pM tid %d\n",
	       sta->sta.addr, tid);

	set_bit(tid, sta->ampdu_mlme.tid_rx_timer_expired);
	wiphy_work_queue(sta->local->hw.wiphy, &sta->ampdu_mlme.work);
}

static void sta_rx_agg_reorder_timer_expired(struct timer_list *t)
{
	struct tid_ampdu_rx *tid_rx = timer_container_of(tid_rx, t,
							 reorder_timer);

	rcu_read_lock();
	ieee80211_release_reorder_timeout(tid_rx->sta, tid_rx->tid);
	rcu_read_unlock();
}

void ieee80211_add_addbaext(struct sk_buff *skb,
			    const u8 req_addba_ext_data,
			    u16 buf_size)
{
	struct ieee80211_addba_ext_ie *addba_ext;
	u8 *pos;

	pos = skb_put_zero(skb, 2 + sizeof(struct ieee80211_addba_ext_ie));
	*pos++ = WLAN_EID_ADDBA_EXT;
	*pos++ = sizeof(struct ieee80211_addba_ext_ie);
	addba_ext = (struct ieee80211_addba_ext_ie *)pos;

	addba_ext->data = IEEE80211_ADDBA_EXT_NO_FRAG;
	if (req_addba_ext_data)
		addba_ext->data &= req_addba_ext_data;

	addba_ext->data |=
		u8_encode_bits(buf_size >> IEEE80211_ADDBA_EXT_BUF_SIZE_SHIFT,
			       IEEE80211_ADDBA_EXT_BUF_SIZE_MASK);
}

u8 ieee80211_retrieve_addba_ext_data(struct sta_info *sta,
				     const void *elem_data, ssize_t elem_len,
				     u16 *buf_size)
{
	struct ieee802_11_elems *elems;
	u8 buf_size_1k, data = 0;

	if (!sta->sta.deflink.he_cap.has_he)
		return 0;

	if (elem_len <= 0)
		return 0;

	elems = ieee802_11_parse_elems(elem_data, elem_len, true, NULL);

	if (!elems || elems->parse_error || !elems->addba_ext_ie)
		goto free;

	data = elems->addba_ext_ie->data;

	if (buf_size &&
	    (sta->sta.valid_links || sta->sta.deflink.eht_cap.has_eht)) {
		buf_size_1k = u8_get_bits(elems->addba_ext_ie->data,
					  IEEE80211_ADDBA_EXT_BUF_SIZE_MASK);
		*buf_size |= (u16)buf_size_1k <<
			     IEEE80211_ADDBA_EXT_BUF_SIZE_SHIFT;
	}

free:
	kfree(elems);

	return data;
}

static void ieee80211_send_addba_resp(struct sta_info *sta, u8 *da, u16 tid,
				      u8 dialog_token, u16 status, u16 policy,
				      u16 buf_size, u16 timeout,
				      const u8 req_addba_ext_data)
{
	struct ieee80211_sub_if_data *sdata = sta->sdata;
	struct ieee80211_local *local = sdata->local;
	struct sk_buff *skb;
	struct ieee80211_mgmt *mgmt;
	bool amsdu = ieee80211_hw_check(&local->hw, SUPPORTS_AMSDU_IN_AMPDU);
	u16 capab;

	skb = dev_alloc_skb(sizeof(*mgmt) +
		    2 + sizeof(struct ieee80211_addba_ext_ie) +
		    local->hw.extra_tx_headroom);
	if (!skb)
		return;

	skb_reserve(skb, local->hw.extra_tx_headroom);
	mgmt = ieee80211_mgmt_ba(skb, da, sdata);

	skb_put(skb, 1 + sizeof(mgmt->u.action.u.addba_resp));
	mgmt->u.action.category = WLAN_CATEGORY_BACK;
	mgmt->u.action.u.addba_resp.action_code = WLAN_ACTION_ADDBA_RESP;
	mgmt->u.action.u.addba_resp.dialog_token = dialog_token;

	capab = u16_encode_bits(amsdu, IEEE80211_ADDBA_PARAM_AMSDU_MASK);
	capab |= u16_encode_bits(policy, IEEE80211_ADDBA_PARAM_POLICY_MASK);
	capab |= u16_encode_bits(tid, IEEE80211_ADDBA_PARAM_TID_MASK);
	capab |= u16_encode_bits(buf_size, IEEE80211_ADDBA_PARAM_BUF_SIZE_MASK);

	mgmt->u.action.u.addba_resp.capab = cpu_to_le16(capab);
	mgmt->u.action.u.addba_resp.timeout = cpu_to_le16(timeout);
	mgmt->u.action.u.addba_resp.status = cpu_to_le16(status);

	if (sta->sta.valid_links || sta->sta.deflink.he_cap.has_he)
		ieee80211_add_addbaext(skb, req_addba_ext_data, buf_size);

	ieee80211_tx_skb(sdata, skb);
}

void __ieee80211_start_rx_ba_session(struct sta_info *sta,
				     u8 dialog_token, u16 timeout,
				     u16 start_seq_num, u16 ba_policy, u16 tid,
				     u16 buf_size, bool tx, bool auto_seq,
				     const u8 addba_ext_data)
{
	struct ieee80211_local *local = sta->sdata->local;
	struct tid_ampdu_rx *tid_agg_rx;
	struct ieee80211_ampdu_params params = {
		.sta = &sta->sta,
		.action = IEEE80211_AMPDU_RX_START,
		.tid = tid,
		.amsdu = false,
		.timeout = timeout,
		.ssn = start_seq_num,
	};
	int i, ret = -EOPNOTSUPP;
	u16 status = WLAN_STATUS_REQUEST_DECLINED;
	u16 max_buf_size;

	lockdep_assert_wiphy(sta->local->hw.wiphy);

	if (tid >= IEEE80211_FIRST_TSPEC_TSID) {
		ht_dbg(sta->sdata,
		       "STA %pM requests BA session on unsupported tid %d\n",
		       sta->sta.addr, tid);
		goto end;
	}

	if (!sta->sta.valid_links &&
	    !sta->sta.deflink.ht_cap.ht_supported &&
	    !sta->sta.deflink.he_cap.has_he) {
		ht_dbg(sta->sdata,
		       "STA %pM erroneously requests BA session on tid %d w/o HT\n",
		       sta->sta.addr, tid);
		/* send a response anyway, it's an error case if we get here */
		goto end;
	}

	if (test_sta_flag(sta, WLAN_STA_BLOCK_BA)) {
		ht_dbg(sta->sdata,
		       "Suspend in progress - Denying ADDBA request (%pM tid %d)\n",
		       sta->sta.addr, tid);
		goto end;
	}

	if (sta->sta.valid_links || sta->sta.deflink.eht_cap.has_eht)
		max_buf_size = IEEE80211_MAX_AMPDU_BUF_EHT;
	else if (sta->sta.deflink.he_cap.has_he)
		max_buf_size = IEEE80211_MAX_AMPDU_BUF_HE;
	else
		max_buf_size = IEEE80211_MAX_AMPDU_BUF_HT;

	/* sanity check for incoming parameters:
	 * check if configuration can support the BA policy
	 * and if buffer size does not exceeds max value */
	/* XXX: check own ht delayed BA capability?? */
	if (((ba_policy != 1) &&
	     (sta->sta.valid_links ||
	      !(sta->sta.deflink.ht_cap.cap & IEEE80211_HT_CAP_DELAY_BA))) ||
	    (buf_size > max_buf_size)) {
		status = WLAN_STATUS_INVALID_QOS_PARAM;
		ht_dbg_ratelimited(sta->sdata,
				   "AddBA Req with bad params from %pM on tid %u. policy %d, buffer size %d\n",
				   sta->sta.addr, tid, ba_policy, buf_size);
		goto end;
	}
	/* determine default buffer size */
	if (buf_size == 0)
		buf_size = max_buf_size;

	/* make sure the size doesn't exceed the maximum supported by the hw */
	if (buf_size > sta->sta.max_rx_aggregation_subframes)
		buf_size = sta->sta.max_rx_aggregation_subframes;
	params.buf_size = buf_size;

	ht_dbg(sta->sdata, "AddBA Req buf_size=%d for %pM\n",
	       buf_size, sta->sta.addr);

	if (test_bit(tid, sta->ampdu_mlme.agg_session_valid)) {
		if (sta->ampdu_mlme.tid_rx_token[tid] == dialog_token) {
			struct tid_ampdu_rx *tid_rx;

			ht_dbg_ratelimited(sta->sdata,
					   "updated AddBA Req from %pM on tid %u\n",
					   sta->sta.addr, tid);
			/* We have no API to update the timeout value in the
			 * driver so reject the timeout update if the timeout
			 * changed. If it did not change, i.e., no real update,
			 * just reply with success.
			 */
			rcu_read_lock();
			tid_rx = rcu_dereference(sta->ampdu_mlme.tid_rx[tid]);
			if (tid_rx && tid_rx->timeout == timeout)
				status = WLAN_STATUS_SUCCESS;
			else
				status = WLAN_STATUS_REQUEST_DECLINED;
			rcu_read_unlock();
			goto end;
		}

		ht_dbg_ratelimited(sta->sdata,
				   "unexpected AddBA Req from %pM on tid %u\n",
				   sta->sta.addr, tid);

		/* delete existing Rx BA session on the same tid */
		__ieee80211_stop_rx_ba_session(sta, tid, WLAN_BACK_RECIPIENT,
					       WLAN_STATUS_UNSPECIFIED_QOS,
					       false);
	}

	if (ieee80211_hw_check(&local->hw, SUPPORTS_REORDERING_BUFFER)) {
		ret = drv_ampdu_action(local, sta->sdata, &params);
		ht_dbg(sta->sdata,
		       "Rx A-MPDU request on %pM tid %d result %d\n",
		       sta->sta.addr, tid, ret);
		if (!ret)
			status = WLAN_STATUS_SUCCESS;
		goto end;
	}

	/* prepare A-MPDU MLME for Rx aggregation */
	tid_agg_rx = kzalloc(sizeof(*tid_agg_rx), GFP_KERNEL);
	if (!tid_agg_rx)
		goto end;

	spin_lock_init(&tid_agg_rx->reorder_lock);

	/* rx timer */
	timer_setup(&tid_agg_rx->session_timer,
		    sta_rx_agg_session_timer_expired, TIMER_DEFERRABLE);

	/* rx reorder timer */
	timer_setup(&tid_agg_rx->reorder_timer,
		    sta_rx_agg_reorder_timer_expired, 0);

	/* prepare reordering buffer */
	tid_agg_rx->reorder_buf =
		kcalloc(buf_size, sizeof(struct sk_buff_head), GFP_KERNEL);
	tid_agg_rx->reorder_time =
		kcalloc(buf_size, sizeof(unsigned long), GFP_KERNEL);
	if (!tid_agg_rx->reorder_buf || !tid_agg_rx->reorder_time) {
		kfree(tid_agg_rx->reorder_buf);
		kfree(tid_agg_rx->reorder_time);
		kfree(tid_agg_rx);
		goto end;
	}

	for (i = 0; i < buf_size; i++)
		__skb_queue_head_init(&tid_agg_rx->reorder_buf[i]);

	ret = drv_ampdu_action(local, sta->sdata, &params);
	ht_dbg(sta->sdata, "Rx A-MPDU request on %pM tid %d result %d\n",
	       sta->sta.addr, tid, ret);
	if (ret) {
		kfree(tid_agg_rx->reorder_buf);
		kfree(tid_agg_rx->reorder_time);
		kfree(tid_agg_rx);
		goto end;
	}

	/* update data */
	tid_agg_rx->ssn = start_seq_num;
	tid_agg_rx->head_seq_num = start_seq_num;
	tid_agg_rx->buf_size = buf_size;
	tid_agg_rx->timeout = timeout;
	tid_agg_rx->stored_mpdu_num = 0;
	tid_agg_rx->auto_seq = auto_seq;
	tid_agg_rx->started = false;
	tid_agg_rx->reorder_buf_filtered = 0;
	tid_agg_rx->tid = tid;
	tid_agg_rx->sta = sta;
	status = WLAN_STATUS_SUCCESS;

	/* activate it for RX */
	rcu_assign_pointer(sta->ampdu_mlme.tid_rx[tid], tid_agg_rx);

	if (timeout) {
		mod_timer(&tid_agg_rx->session_timer, TU_TO_EXP_TIME(timeout));
		tid_agg_rx->last_rx = jiffies;
	}

end:
	if (status == WLAN_STATUS_SUCCESS) {
		__set_bit(tid, sta->ampdu_mlme.agg_session_valid);
		__clear_bit(tid, sta->ampdu_mlme.unexpected_agg);
		sta->ampdu_mlme.tid_rx_token[tid] = dialog_token;
	}

	if (tx)
		ieee80211_send_addba_resp(sta, sta->sta.addr, tid,
					  dialog_token, status, 1, buf_size,
					  timeout, addba_ext_data);
}

void ieee80211_process_addba_request(struct ieee80211_local *local,
				     struct sta_info *sta,
				     struct ieee80211_mgmt *mgmt,
				     size_t len)
{
	u16 capab, tid, timeout, ba_policy, buf_size, start_seq_num;
	u8 dialog_token, addba_ext_data;

	/* extract session parameters from addba request frame */
	dialog_token = mgmt->u.action.u.addba_req.dialog_token;
	timeout = le16_to_cpu(mgmt->u.action.u.addba_req.timeout);
	start_seq_num =
		le16_to_cpu(mgmt->u.action.u.addba_req.start_seq_num) >> 4;

	capab = le16_to_cpu(mgmt->u.action.u.addba_req.capab);
	ba_policy = (capab & IEEE80211_ADDBA_PARAM_POLICY_MASK) >> 1;
	tid = (capab & IEEE80211_ADDBA_PARAM_TID_MASK) >> 2;
	buf_size = (capab & IEEE80211_ADDBA_PARAM_BUF_SIZE_MASK) >> 6;

	addba_ext_data =
		ieee80211_retrieve_addba_ext_data(sta,
						  mgmt->u.action.u.addba_req.variable,
						  len -
						  offsetof(typeof(*mgmt),
							   u.action.u.addba_req.variable),
						  &buf_size);

	__ieee80211_start_rx_ba_session(sta, dialog_token, timeout,
					start_seq_num, ba_policy, tid,
					buf_size, true, false, addba_ext_data);
}

void ieee80211_manage_rx_ba_offl(struct ieee80211_vif *vif,
				 const u8 *addr, unsigned int tid)
{
	struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
	struct sta_info *sta;

	rcu_read_lock();
	sta = sta_info_get_bss(sdata, addr);
	if (!sta)
		goto unlock;

	set_bit(tid, sta->ampdu_mlme.tid_rx_manage_offl);
	wiphy_work_queue(sta->local->hw.wiphy, &sta->ampdu_mlme.work);
 unlock:
	rcu_read_unlock();
}
EXPORT_SYMBOL(ieee80211_manage_rx_ba_offl);

void ieee80211_rx_ba_timer_expired(struct ieee80211_vif *vif,
				   const u8 *addr, unsigned int tid)
{
	struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
	struct sta_info *sta;

	rcu_read_lock();
	sta = sta_info_get_bss(sdata, addr);
	if (!sta)
		goto unlock;

	set_bit(tid, sta->ampdu_mlme.tid_rx_timer_expired);
	wiphy_work_queue(sta->local->hw.wiphy, &sta->ampdu_mlme.work);

 unlock:
	rcu_read_unlock();
}
EXPORT_SYMBOL(ieee80211_rx_ba_timer_expired);
Commit	Line	Data
	1	// SPDX-License-Identifier: GPL-2.0-only
	2	/*
	3	* HT handling
	4	*
	5	* Copyright 2003, Jouni Malinen <jkmaline@cc.hut.fi>
	6	* Copyright 2002-2005, Instant802 Networks, Inc.
	7	* Copyright 2005-2006, Devicescape Software, Inc.
	8	* Copyright 2006-2007 Jiri Benc <jbenc@suse.cz>
	9	* Copyright 2007, Michael Wu <flamingice@sourmilk.net>
	10	* Copyright 2007-2010, Intel Corporation
	11	* Copyright(c) 2015-2017 Intel Deutschland GmbH
	12	* Copyright (C) 2018-2024 Intel Corporation
	13	*/
	14
	15	/**
	16	* DOC: RX A-MPDU aggregation
	17	*
	18	* Aggregation on the RX side requires only implementing the
	19	* @ampdu_action callback that is invoked to start/stop any
	20	* block-ack sessions for RX aggregation.
	21	*
	22	* When RX aggregation is started by the peer, the driver is
	23	* notified via @ampdu_action function, with the
	24	* %IEEE80211_AMPDU_RX_START action, and may reject the request
	25	* in which case a negative response is sent to the peer, if it
	26	* accepts it a positive response is sent.
	27	*
	28	* While the session is active, the device/driver are required
	29	* to de-aggregate frames and pass them up one by one to mac80211,
	30	* which will handle the reorder buffer.
	31	*
	32	* When the aggregation session is stopped again by the peer or
	33	* ourselves, the driver's @ampdu_action function will be called
	34	* with the action %IEEE80211_AMPDU_RX_STOP. In this case, the
	35	* call must not fail.
	36	*/
	37
	38	#include <linux/ieee80211.h>
	39	#include <linux/slab.h>
	40	#include <linux/export.h>
	41	#include <net/mac80211.h>
	42	#include "ieee80211_i.h"
	43	#include "driver-ops.h"
	44
	45	static void ieee80211_free_tid_rx(struct rcu_head *h)
	46	{
	47	struct tid_ampdu_rx *tid_rx =
	48	container_of(h, struct tid_ampdu_rx, rcu_head);
	49	int i;
	50
	51	for (i = 0; i < tid_rx->buf_size; i++)
	52	__skb_queue_purge(&tid_rx->reorder_buf[i]);
	53	kfree(tid_rx->reorder_buf);
	54	kfree(tid_rx->reorder_time);
	55	kfree(tid_rx);
	56	}
	57
	58	void __ieee80211_stop_rx_ba_session(struct sta_info *sta, u16 tid,
	59	u16 initiator, u16 reason, bool tx)
	60	{
	61	struct ieee80211_local *local = sta->local;
	62	struct tid_ampdu_rx *tid_rx;
	63	struct ieee80211_ampdu_params params = {
	64	.sta = &sta->sta,
	65	.action = IEEE80211_AMPDU_RX_STOP,
	66	.tid = tid,
	67	.amsdu = false,
	68	.timeout = 0,
	69	.ssn = 0,
	70	};
	71
	72	lockdep_assert_wiphy(sta->local->hw.wiphy);
	73
	74	tid_rx = rcu_dereference_protected(sta->ampdu_mlme.tid_rx[tid],
	75	lockdep_is_held(&sta->local->hw.wiphy->mtx));
	76
	77	if (!test_bit(tid, sta->ampdu_mlme.agg_session_valid))
	78	return;
	79
	80	RCU_INIT_POINTER(sta->ampdu_mlme.tid_rx[tid], NULL);
	81	__clear_bit(tid, sta->ampdu_mlme.agg_session_valid);
	82
	83	ht_dbg(sta->sdata,
	84	"Rx BA session stop requested for %pM tid %u %s reason: %d\n",
	85	sta->sta.addr, tid,
	86	initiator == WLAN_BACK_RECIPIENT ? "recipient" : "initiator",
	87	(int)reason);
	88
	89	if (drv_ampdu_action(local, sta->sdata, &params))
	90	sdata_info(sta->sdata,
	91	"HW problem - can not stop rx aggregation for %pM tid %d\n",
	92	sta->sta.addr, tid);
	93
	94	/* check if this is a self generated aggregation halt */
	95	if (initiator == WLAN_BACK_RECIPIENT && tx)
	96	ieee80211_send_delba(sta->sdata, sta->sta.addr,
	97	tid, WLAN_BACK_RECIPIENT, reason);
	98
	99	/*
	100	* return here in case tid_rx is not assigned - which will happen if
	101	* IEEE80211_HW_SUPPORTS_REORDERING_BUFFER is set.
	102	*/
	103	if (!tid_rx)
	104	return;
	105
	106	timer_delete_sync(&tid_rx->session_timer);
	107
	108	/* make sure ieee80211_sta_reorder_release() doesn't re-arm the timer */
	109	spin_lock_bh(&tid_rx->reorder_lock);
	110	tid_rx->removed = true;
	111	spin_unlock_bh(&tid_rx->reorder_lock);
	112	timer_delete_sync(&tid_rx->reorder_timer);
	113
	114	call_rcu(&tid_rx->rcu_head, ieee80211_free_tid_rx);
	115	}
	116
	117	void ieee80211_stop_rx_ba_session(struct ieee80211_vif *vif, u16 ba_rx_bitmap,
	118	const u8 *addr)
	119	{
	120	struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
	121	struct sta_info *sta;
	122	int i;
	123
	124	rcu_read_lock();
	125	sta = sta_info_get_bss(sdata, addr);
	126	if (!sta) {
	127	rcu_read_unlock();
	128	return;
	129	}
	130
	131	for (i = 0; i < IEEE80211_NUM_TIDS; i++)
	132	if (ba_rx_bitmap & BIT(i))
	133	set_bit(i, sta->ampdu_mlme.tid_rx_stop_requested);
	134
	135	wiphy_work_queue(sta->local->hw.wiphy, &sta->ampdu_mlme.work);
	136	rcu_read_unlock();
	137	}
	138	EXPORT_SYMBOL(ieee80211_stop_rx_ba_session);
	139
	140	/*
	141	* After accepting the AddBA Request we activated a timer,
	142	* resetting it after each frame that arrives from the originator.
	143	*/
	144	static void sta_rx_agg_session_timer_expired(struct timer_list *t)
	145	{
	146	struct tid_ampdu_rx *tid_rx = timer_container_of(tid_rx, t,
	147	session_timer);
	148	struct sta_info *sta = tid_rx->sta;
	149	u8 tid = tid_rx->tid;
	150	unsigned long timeout;
	151
	152	timeout = tid_rx->last_rx + TU_TO_JIFFIES(tid_rx->timeout);
	153	if (time_is_after_jiffies(timeout)) {
	154	mod_timer(&tid_rx->session_timer, timeout);
	155	return;
	156	}
	157
	158	ht_dbg(sta->sdata, "RX session timer expired on %pM tid %d\n",
	159	sta->sta.addr, tid);
	160
	161	set_bit(tid, sta->ampdu_mlme.tid_rx_timer_expired);
	162	wiphy_work_queue(sta->local->hw.wiphy, &sta->ampdu_mlme.work);
	163	}
	164
	165	static void sta_rx_agg_reorder_timer_expired(struct timer_list *t)
	166	{
	167	struct tid_ampdu_rx *tid_rx = timer_container_of(tid_rx, t,
	168	reorder_timer);
	169
	170	rcu_read_lock();
	171	ieee80211_release_reorder_timeout(tid_rx->sta, tid_rx->tid);
	172	rcu_read_unlock();
	173	}
	174
	175	void ieee80211_add_addbaext(struct sk_buff *skb,
	176	const u8 req_addba_ext_data,
	177	u16 buf_size)
	178	{
	179	struct ieee80211_addba_ext_ie *addba_ext;
	180	u8 *pos;
	181
	182	pos = skb_put_zero(skb, 2 + sizeof(struct ieee80211_addba_ext_ie));
	183	*pos++ = WLAN_EID_ADDBA_EXT;
	184	*pos++ = sizeof(struct ieee80211_addba_ext_ie);
	185	addba_ext = (struct ieee80211_addba_ext_ie *)pos;
	186
	187	addba_ext->data = IEEE80211_ADDBA_EXT_NO_FRAG;
	188	if (req_addba_ext_data)
	189	addba_ext->data &= req_addba_ext_data;
	190
	191	addba_ext->data \|=
	192	u8_encode_bits(buf_size >> IEEE80211_ADDBA_EXT_BUF_SIZE_SHIFT,
	193	IEEE80211_ADDBA_EXT_BUF_SIZE_MASK);
	194	}
	195
	196	u8 ieee80211_retrieve_addba_ext_data(struct sta_info *sta,
	197	const void *elem_data, ssize_t elem_len,
	198	u16 *buf_size)
	199	{
	200	struct ieee802_11_elems *elems;
	201	u8 buf_size_1k, data = 0;
	202
	203	if (!sta->sta.deflink.he_cap.has_he)
	204	return 0;
	205
	206	if (elem_len <= 0)
	207	return 0;
	208
	209	elems = ieee802_11_parse_elems(elem_data, elem_len, true, NULL);
	210
	211	if (!elems \|\| elems->parse_error \|\| !elems->addba_ext_ie)
	212	goto free;
	213
	214	data = elems->addba_ext_ie->data;
	215
	216	if (buf_size &&
	217	(sta->sta.valid_links \|\| sta->sta.deflink.eht_cap.has_eht)) {
	218	buf_size_1k = u8_get_bits(elems->addba_ext_ie->data,
	219	IEEE80211_ADDBA_EXT_BUF_SIZE_MASK);
	220	*buf_size \|= (u16)buf_size_1k <<
	221	IEEE80211_ADDBA_EXT_BUF_SIZE_SHIFT;
	222	}
	223
	224	free:
	225	kfree(elems);
	226
	227	return data;
	228	}
	229
	230	static void ieee80211_send_addba_resp(struct sta_info sta, u8 da, u16 tid,
	231	u8 dialog_token, u16 status, u16 policy,
	232	u16 buf_size, u16 timeout,
	233	const u8 req_addba_ext_data)
	234	{
	235	struct ieee80211_sub_if_data *sdata = sta->sdata;
	236	struct ieee80211_local *local = sdata->local;
	237	struct sk_buff *skb;
	238	struct ieee80211_mgmt *mgmt;
	239	bool amsdu = ieee80211_hw_check(&local->hw, SUPPORTS_AMSDU_IN_AMPDU);
	240	u16 capab;
	241
	242	skb = dev_alloc_skb(sizeof(*mgmt) +
	243	2 + sizeof(struct ieee80211_addba_ext_ie) +
	244	local->hw.extra_tx_headroom);
	245	if (!skb)
	246	return;
	247
	248	skb_reserve(skb, local->hw.extra_tx_headroom);
	249	mgmt = ieee80211_mgmt_ba(skb, da, sdata);
	250
	251	skb_put(skb, 1 + sizeof(mgmt->u.action.u.addba_resp));
	252	mgmt->u.action.category = WLAN_CATEGORY_BACK;
	253	mgmt->u.action.u.addba_resp.action_code = WLAN_ACTION_ADDBA_RESP;
	254	mgmt->u.action.u.addba_resp.dialog_token = dialog_token;
	255
	256	capab = u16_encode_bits(amsdu, IEEE80211_ADDBA_PARAM_AMSDU_MASK);
	257	capab \|= u16_encode_bits(policy, IEEE80211_ADDBA_PARAM_POLICY_MASK);
	258	capab \|= u16_encode_bits(tid, IEEE80211_ADDBA_PARAM_TID_MASK);
	259	capab \|= u16_encode_bits(buf_size, IEEE80211_ADDBA_PARAM_BUF_SIZE_MASK);
	260
	261	mgmt->u.action.u.addba_resp.capab = cpu_to_le16(capab);
	262	mgmt->u.action.u.addba_resp.timeout = cpu_to_le16(timeout);
	263	mgmt->u.action.u.addba_resp.status = cpu_to_le16(status);
	264
	265	if (sta->sta.valid_links \|\| sta->sta.deflink.he_cap.has_he)
	266	ieee80211_add_addbaext(skb, req_addba_ext_data, buf_size);
	267
	268	ieee80211_tx_skb(sdata, skb);
	269	}
	270
	271	void __ieee80211_start_rx_ba_session(struct sta_info *sta,
	272	u8 dialog_token, u16 timeout,
	273	u16 start_seq_num, u16 ba_policy, u16 tid,
	274	u16 buf_size, bool tx, bool auto_seq,
	275	const u8 addba_ext_data)
	276	{
	277	struct ieee80211_local *local = sta->sdata->local;
	278	struct tid_ampdu_rx *tid_agg_rx;
	279	struct ieee80211_ampdu_params params = {
	280	.sta = &sta->sta,
	281	.action = IEEE80211_AMPDU_RX_START,
	282	.tid = tid,
	283	.amsdu = false,
	284	.timeout = timeout,
	285	.ssn = start_seq_num,
	286	};
	287	int i, ret = -EOPNOTSUPP;
	288	u16 status = WLAN_STATUS_REQUEST_DECLINED;
	289	u16 max_buf_size;
	290
	291	lockdep_assert_wiphy(sta->local->hw.wiphy);
	292
	293	if (tid >= IEEE80211_FIRST_TSPEC_TSID) {
	294	ht_dbg(sta->sdata,
	295	"STA %pM requests BA session on unsupported tid %d\n",
	296	sta->sta.addr, tid);
	297	goto end;
	298	}
	299
	300	if (!sta->sta.valid_links &&
	301	!sta->sta.deflink.ht_cap.ht_supported &&
	302	!sta->sta.deflink.he_cap.has_he) {
	303	ht_dbg(sta->sdata,
	304	"STA %pM erroneously requests BA session on tid %d w/o HT\n",
	305	sta->sta.addr, tid);
	306	/* send a response anyway, it's an error case if we get here */
	307	goto end;
	308	}
	309
	310	if (test_sta_flag(sta, WLAN_STA_BLOCK_BA)) {
	311	ht_dbg(sta->sdata,
	312	"Suspend in progress - Denying ADDBA request (%pM tid %d)\n",
	313	sta->sta.addr, tid);
	314	goto end;
	315	}
	316
	317	if (sta->sta.valid_links \|\| sta->sta.deflink.eht_cap.has_eht)
	318	max_buf_size = IEEE80211_MAX_AMPDU_BUF_EHT;
	319	else if (sta->sta.deflink.he_cap.has_he)
	320	max_buf_size = IEEE80211_MAX_AMPDU_BUF_HE;
	321	else
	322	max_buf_size = IEEE80211_MAX_AMPDU_BUF_HT;
	323
	324	/* sanity check for incoming parameters:
	325	* check if configuration can support the BA policy
	326	* and if buffer size does not exceeds max value */
	327	/* XXX: check own ht delayed BA capability?? */
	328	if (((ba_policy != 1) &&
	329	(sta->sta.valid_links \|\|
	330	!(sta->sta.deflink.ht_cap.cap & IEEE80211_HT_CAP_DELAY_BA))) \|\|
	331	(buf_size > max_buf_size)) {
	332	status = WLAN_STATUS_INVALID_QOS_PARAM;
	333	ht_dbg_ratelimited(sta->sdata,
	334	"AddBA Req with bad params from %pM on tid %u. policy %d, buffer size %d\n",
	335	sta->sta.addr, tid, ba_policy, buf_size);
	336	goto end;
	337	}
	338	/* determine default buffer size */
	339	if (buf_size == 0)
	340	buf_size = max_buf_size;
	341
	342	/* make sure the size doesn't exceed the maximum supported by the hw */
	343	if (buf_size > sta->sta.max_rx_aggregation_subframes)
	344	buf_size = sta->sta.max_rx_aggregation_subframes;
	345	params.buf_size = buf_size;
	346
	347	ht_dbg(sta->sdata, "AddBA Req buf_size=%d for %pM\n",
	348	buf_size, sta->sta.addr);
	349
	350	if (test_bit(tid, sta->ampdu_mlme.agg_session_valid)) {
	351	if (sta->ampdu_mlme.tid_rx_token[tid] == dialog_token) {
	352	struct tid_ampdu_rx *tid_rx;
	353
	354	ht_dbg_ratelimited(sta->sdata,
	355	"updated AddBA Req from %pM on tid %u\n",
	356	sta->sta.addr, tid);
	357	/* We have no API to update the timeout value in the
	358	* driver so reject the timeout update if the timeout
	359	* changed. If it did not change, i.e., no real update,
	360	* just reply with success.
	361	*/
	362	rcu_read_lock();
	363	tid_rx = rcu_dereference(sta->ampdu_mlme.tid_rx[tid]);
	364	if (tid_rx && tid_rx->timeout == timeout)
	365	status = WLAN_STATUS_SUCCESS;
	366	else
	367	status = WLAN_STATUS_REQUEST_DECLINED;
	368	rcu_read_unlock();
	369	goto end;
	370	}
	371
	372	ht_dbg_ratelimited(sta->sdata,
	373	"unexpected AddBA Req from %pM on tid %u\n",
	374	sta->sta.addr, tid);
	375
	376	/* delete existing Rx BA session on the same tid */
	377	__ieee80211_stop_rx_ba_session(sta, tid, WLAN_BACK_RECIPIENT,
	378	WLAN_STATUS_UNSPECIFIED_QOS,
	379	false);
	380	}
	381
	382	if (ieee80211_hw_check(&local->hw, SUPPORTS_REORDERING_BUFFER)) {
	383	ret = drv_ampdu_action(local, sta->sdata, &params);
	384	ht_dbg(sta->sdata,
	385	"Rx A-MPDU request on %pM tid %d result %d\n",
	386	sta->sta.addr, tid, ret);
	387	if (!ret)
	388	status = WLAN_STATUS_SUCCESS;
	389	goto end;
	390	}
	391
	392	/* prepare A-MPDU MLME for Rx aggregation */
	393	tid_agg_rx = kzalloc(sizeof(*tid_agg_rx), GFP_KERNEL);
	394	if (!tid_agg_rx)
	395	goto end;
	396
	397	spin_lock_init(&tid_agg_rx->reorder_lock);
	398
	399	/* rx timer */
	400	timer_setup(&tid_agg_rx->session_timer,
	401	sta_rx_agg_session_timer_expired, TIMER_DEFERRABLE);
	402
	403	/* rx reorder timer */
	404	timer_setup(&tid_agg_rx->reorder_timer,
	405	sta_rx_agg_reorder_timer_expired, 0);
	406
	407	/* prepare reordering buffer */
	408	tid_agg_rx->reorder_buf =
	409	kcalloc(buf_size, sizeof(struct sk_buff_head), GFP_KERNEL);
	410	tid_agg_rx->reorder_time =
	411	kcalloc(buf_size, sizeof(unsigned long), GFP_KERNEL);
	412	if (!tid_agg_rx->reorder_buf \|\| !tid_agg_rx->reorder_time) {
	413	kfree(tid_agg_rx->reorder_buf);
	414	kfree(tid_agg_rx->reorder_time);
	415	kfree(tid_agg_rx);
	416	goto end;
	417	}
	418
	419	for (i = 0; i < buf_size; i++)
	420	__skb_queue_head_init(&tid_agg_rx->reorder_buf[i]);
	421
	422	ret = drv_ampdu_action(local, sta->sdata, &params);
	423	ht_dbg(sta->sdata, "Rx A-MPDU request on %pM tid %d result %d\n",
	424	sta->sta.addr, tid, ret);
	425	if (ret) {
	426	kfree(tid_agg_rx->reorder_buf);
	427	kfree(tid_agg_rx->reorder_time);
	428	kfree(tid_agg_rx);
	429	goto end;
	430	}
	431
	432	/* update data */
	433	tid_agg_rx->ssn = start_seq_num;
	434	tid_agg_rx->head_seq_num = start_seq_num;
	435	tid_agg_rx->buf_size = buf_size;
	436	tid_agg_rx->timeout = timeout;
	437	tid_agg_rx->stored_mpdu_num = 0;
	438	tid_agg_rx->auto_seq = auto_seq;
	439	tid_agg_rx->started = false;
	440	tid_agg_rx->reorder_buf_filtered = 0;
	441	tid_agg_rx->tid = tid;
	442	tid_agg_rx->sta = sta;
	443	status = WLAN_STATUS_SUCCESS;
	444
	445	/* activate it for RX */
	446	rcu_assign_pointer(sta->ampdu_mlme.tid_rx[tid], tid_agg_rx);
	447
	448	if (timeout) {
	449	mod_timer(&tid_agg_rx->session_timer, TU_TO_EXP_TIME(timeout));
	450	tid_agg_rx->last_rx = jiffies;
	451	}
	452
	453	end:
	454	if (status == WLAN_STATUS_SUCCESS) {
	455	__set_bit(tid, sta->ampdu_mlme.agg_session_valid);
	456	__clear_bit(tid, sta->ampdu_mlme.unexpected_agg);
	457	sta->ampdu_mlme.tid_rx_token[tid] = dialog_token;
	458	}
	459
	460	if (tx)
	461	ieee80211_send_addba_resp(sta, sta->sta.addr, tid,
	462	dialog_token, status, 1, buf_size,
	463	timeout, addba_ext_data);
	464	}
	465
	466	void ieee80211_process_addba_request(struct ieee80211_local *local,
	467	struct sta_info *sta,
	468	struct ieee80211_mgmt *mgmt,
	469	size_t len)
	470	{
	471	u16 capab, tid, timeout, ba_policy, buf_size, start_seq_num;
	472	u8 dialog_token, addba_ext_data;
	473
	474	/* extract session parameters from addba request frame */
	475	dialog_token = mgmt->u.action.u.addba_req.dialog_token;
	476	timeout = le16_to_cpu(mgmt->u.action.u.addba_req.timeout);
	477	start_seq_num =
	478	le16_to_cpu(mgmt->u.action.u.addba_req.start_seq_num) >> 4;
	479
	480	capab = le16_to_cpu(mgmt->u.action.u.addba_req.capab);
	481	ba_policy = (capab & IEEE80211_ADDBA_PARAM_POLICY_MASK) >> 1;
	482	tid = (capab & IEEE80211_ADDBA_PARAM_TID_MASK) >> 2;
	483	buf_size = (capab & IEEE80211_ADDBA_PARAM_BUF_SIZE_MASK) >> 6;
	484
	485	addba_ext_data =
	486	ieee80211_retrieve_addba_ext_data(sta,
	487	mgmt->u.action.u.addba_req.variable,
	488	len -
	489	offsetof(typeof(*mgmt),
	490	u.action.u.addba_req.variable),
	491	&buf_size);
	492
	493	__ieee80211_start_rx_ba_session(sta, dialog_token, timeout,
	494	start_seq_num, ba_policy, tid,
	495	buf_size, true, false, addba_ext_data);
	496	}
	497
	498	void ieee80211_manage_rx_ba_offl(struct ieee80211_vif *vif,
	499	const u8 *addr, unsigned int tid)
	500	{
	501	struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
	502	struct sta_info *sta;
	503
	504	rcu_read_lock();
	505	sta = sta_info_get_bss(sdata, addr);
	506	if (!sta)
	507	goto unlock;
	508
	509	set_bit(tid, sta->ampdu_mlme.tid_rx_manage_offl);
	510	wiphy_work_queue(sta->local->hw.wiphy, &sta->ampdu_mlme.work);
	511	unlock:
	512	rcu_read_unlock();
	513	}
	514	EXPORT_SYMBOL(ieee80211_manage_rx_ba_offl);
	515
	516	void ieee80211_rx_ba_timer_expired(struct ieee80211_vif *vif,
	517	const u8 *addr, unsigned int tid)
	518	{
	519	struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
	520	struct sta_info *sta;
	521
	522	rcu_read_lock();
	523	sta = sta_info_get_bss(sdata, addr);
	524	if (!sta)
	525	goto unlock;
	526
	527	set_bit(tid, sta->ampdu_mlme.tid_rx_timer_expired);
	528	wiphy_work_queue(sta->local->hw.wiphy, &sta->ampdu_mlme.work);
	529
	530	unlock:
	531	rcu_read_unlock();
	532	}
	533	EXPORT_SYMBOL(ieee80211_rx_ba_timer_expired);