xref: /freebsd/sys/contrib/dev/iwlwifi/mvm/time-event.c (revision d9836fb4b9380e2ed1c38455fb31a3832b452671)
1bfcc09ddSBjoern A. Zeeb // SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
2bfcc09ddSBjoern A. Zeeb /*
3bfcc09ddSBjoern A. Zeeb  * Copyright (C) 2012-2014, 2018-2021 Intel Corporation
4bfcc09ddSBjoern A. Zeeb  * Copyright (C) 2013-2015 Intel Mobile Communications GmbH
5bfcc09ddSBjoern A. Zeeb  * Copyright (C) 2017 Intel Deutschland GmbH
6bfcc09ddSBjoern A. Zeeb  */
7bfcc09ddSBjoern A. Zeeb #include <linux/jiffies.h>
8bfcc09ddSBjoern A. Zeeb #include <net/mac80211.h>
9bfcc09ddSBjoern A. Zeeb 
10bfcc09ddSBjoern A. Zeeb #include "fw/notif-wait.h"
11bfcc09ddSBjoern A. Zeeb #include "iwl-trans.h"
12bfcc09ddSBjoern A. Zeeb #include "fw-api.h"
13bfcc09ddSBjoern A. Zeeb #include "time-event.h"
14bfcc09ddSBjoern A. Zeeb #include "mvm.h"
15bfcc09ddSBjoern A. Zeeb #include "iwl-io.h"
16bfcc09ddSBjoern A. Zeeb #include "iwl-prph.h"
17bfcc09ddSBjoern A. Zeeb 
18bfcc09ddSBjoern A. Zeeb /*
19bfcc09ddSBjoern A. Zeeb  * For the high priority TE use a time event type that has similar priority to
20bfcc09ddSBjoern A. Zeeb  * the FW's action scan priority.
21bfcc09ddSBjoern A. Zeeb  */
22bfcc09ddSBjoern A. Zeeb #define IWL_MVM_ROC_TE_TYPE_NORMAL TE_P2P_DEVICE_DISCOVERABLE
23bfcc09ddSBjoern A. Zeeb #define IWL_MVM_ROC_TE_TYPE_MGMT_TX TE_P2P_CLIENT_ASSOC
24bfcc09ddSBjoern A. Zeeb 
25bfcc09ddSBjoern A. Zeeb void iwl_mvm_te_clear_data(struct iwl_mvm *mvm,
26bfcc09ddSBjoern A. Zeeb 			   struct iwl_mvm_time_event_data *te_data)
27bfcc09ddSBjoern A. Zeeb {
28bfcc09ddSBjoern A. Zeeb 	lockdep_assert_held(&mvm->time_event_lock);
29bfcc09ddSBjoern A. Zeeb 
30bfcc09ddSBjoern A. Zeeb 	if (!te_data || !te_data->vif)
31bfcc09ddSBjoern A. Zeeb 		return;
32bfcc09ddSBjoern A. Zeeb 
33bfcc09ddSBjoern A. Zeeb 	list_del(&te_data->list);
34bfcc09ddSBjoern A. Zeeb 
35bfcc09ddSBjoern A. Zeeb 	/*
36bfcc09ddSBjoern A. Zeeb 	 * the list is only used for AUX ROC events so make sure it is always
37bfcc09ddSBjoern A. Zeeb 	 * initialized
38bfcc09ddSBjoern A. Zeeb 	 */
39bfcc09ddSBjoern A. Zeeb 	INIT_LIST_HEAD(&te_data->list);
40bfcc09ddSBjoern A. Zeeb 
41bfcc09ddSBjoern A. Zeeb 	te_data->running = false;
42bfcc09ddSBjoern A. Zeeb 	te_data->uid = 0;
43bfcc09ddSBjoern A. Zeeb 	te_data->id = TE_MAX;
44bfcc09ddSBjoern A. Zeeb 	te_data->vif = NULL;
45bfcc09ddSBjoern A. Zeeb }
46bfcc09ddSBjoern A. Zeeb 
47bfcc09ddSBjoern A. Zeeb void iwl_mvm_roc_done_wk(struct work_struct *wk)
48bfcc09ddSBjoern A. Zeeb {
49bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm *mvm = container_of(wk, struct iwl_mvm, roc_done_wk);
50bfcc09ddSBjoern A. Zeeb 
51bfcc09ddSBjoern A. Zeeb 	/*
52*d9836fb4SBjoern A. Zeeb 	 * Clear the ROC_RUNNING status bit.
53bfcc09ddSBjoern A. Zeeb 	 * This will cause the TX path to drop offchannel transmissions.
54bfcc09ddSBjoern A. Zeeb 	 * That would also be done by mac80211, but it is racy, in particular
55bfcc09ddSBjoern A. Zeeb 	 * in the case that the time event actually completed in the firmware
56bfcc09ddSBjoern A. Zeeb 	 * (which is handled in iwl_mvm_te_handle_notif).
57bfcc09ddSBjoern A. Zeeb 	 */
58bfcc09ddSBjoern A. Zeeb 	clear_bit(IWL_MVM_STATUS_ROC_RUNNING, &mvm->status);
59bfcc09ddSBjoern A. Zeeb 
60bfcc09ddSBjoern A. Zeeb 	synchronize_net();
61bfcc09ddSBjoern A. Zeeb 
62bfcc09ddSBjoern A. Zeeb 	/*
63bfcc09ddSBjoern A. Zeeb 	 * Flush the offchannel queue -- this is called when the time
64bfcc09ddSBjoern A. Zeeb 	 * event finishes or is canceled, so that frames queued for it
65bfcc09ddSBjoern A. Zeeb 	 * won't get stuck on the queue and be transmitted in the next
66bfcc09ddSBjoern A. Zeeb 	 * time event.
67bfcc09ddSBjoern A. Zeeb 	 */
68bfcc09ddSBjoern A. Zeeb 
69bfcc09ddSBjoern A. Zeeb 	mutex_lock(&mvm->mutex);
70bfcc09ddSBjoern A. Zeeb 	if (test_and_clear_bit(IWL_MVM_STATUS_NEED_FLUSH_P2P, &mvm->status)) {
71bfcc09ddSBjoern A. Zeeb 		struct iwl_mvm_vif *mvmvif;
72bfcc09ddSBjoern A. Zeeb 
73bfcc09ddSBjoern A. Zeeb 		/*
74bfcc09ddSBjoern A. Zeeb 		 * NB: access to this pointer would be racy, but the flush bit
75bfcc09ddSBjoern A. Zeeb 		 * can only be set when we had a P2P-Device VIF, and we have a
76bfcc09ddSBjoern A. Zeeb 		 * flush of this work in iwl_mvm_prepare_mac_removal() so it's
77bfcc09ddSBjoern A. Zeeb 		 * not really racy.
78bfcc09ddSBjoern A. Zeeb 		 */
79bfcc09ddSBjoern A. Zeeb 
80bfcc09ddSBjoern A. Zeeb 		if (!WARN_ON(!mvm->p2p_device_vif)) {
81bfcc09ddSBjoern A. Zeeb 			mvmvif = iwl_mvm_vif_from_mac80211(mvm->p2p_device_vif);
82bfcc09ddSBjoern A. Zeeb 			iwl_mvm_flush_sta(mvm, &mvmvif->bcast_sta, true);
83bfcc09ddSBjoern A. Zeeb 		}
84*d9836fb4SBjoern A. Zeeb 	}
85*d9836fb4SBjoern A. Zeeb 
86*d9836fb4SBjoern A. Zeeb 	/*
87*d9836fb4SBjoern A. Zeeb 	 * Clear the ROC_AUX_RUNNING status bit.
88*d9836fb4SBjoern A. Zeeb 	 * This will cause the TX path to drop offchannel transmissions.
89*d9836fb4SBjoern A. Zeeb 	 * That would also be done by mac80211, but it is racy, in particular
90*d9836fb4SBjoern A. Zeeb 	 * in the case that the time event actually completed in the firmware
91*d9836fb4SBjoern A. Zeeb 	 * (which is handled in iwl_mvm_te_handle_notif).
92*d9836fb4SBjoern A. Zeeb 	 */
93*d9836fb4SBjoern A. Zeeb 	if (test_and_clear_bit(IWL_MVM_STATUS_ROC_AUX_RUNNING, &mvm->status)) {
94bfcc09ddSBjoern A. Zeeb 		/* do the same in case of hot spot 2.0 */
95bfcc09ddSBjoern A. Zeeb 		iwl_mvm_flush_sta(mvm, &mvm->aux_sta, true);
96*d9836fb4SBjoern A. Zeeb 
97bfcc09ddSBjoern A. Zeeb 		/* In newer version of this command an aux station is added only
98bfcc09ddSBjoern A. Zeeb 		 * in cases of dedicated tx queue and need to be removed in end
99bfcc09ddSBjoern A. Zeeb 		 * of use */
100*d9836fb4SBjoern A. Zeeb 		if (iwl_fw_lookup_cmd_ver(mvm->fw, ADD_STA, 0) >= 12)
101bfcc09ddSBjoern A. Zeeb 			iwl_mvm_rm_aux_sta(mvm);
102bfcc09ddSBjoern A. Zeeb 	}
103bfcc09ddSBjoern A. Zeeb 
104bfcc09ddSBjoern A. Zeeb 	mutex_unlock(&mvm->mutex);
105bfcc09ddSBjoern A. Zeeb }
106bfcc09ddSBjoern A. Zeeb 
107bfcc09ddSBjoern A. Zeeb static void iwl_mvm_roc_finished(struct iwl_mvm *mvm)
108bfcc09ddSBjoern A. Zeeb {
109bfcc09ddSBjoern A. Zeeb 	/*
110bfcc09ddSBjoern A. Zeeb 	 * Of course, our status bit is just as racy as mac80211, so in
111bfcc09ddSBjoern A. Zeeb 	 * addition, fire off the work struct which will drop all frames
112bfcc09ddSBjoern A. Zeeb 	 * from the hardware queues that made it through the race. First
113bfcc09ddSBjoern A. Zeeb 	 * it will of course synchronize the TX path to make sure that
114bfcc09ddSBjoern A. Zeeb 	 * any *new* TX will be rejected.
115bfcc09ddSBjoern A. Zeeb 	 */
116bfcc09ddSBjoern A. Zeeb 	schedule_work(&mvm->roc_done_wk);
117bfcc09ddSBjoern A. Zeeb }
118bfcc09ddSBjoern A. Zeeb 
119bfcc09ddSBjoern A. Zeeb static void iwl_mvm_csa_noa_start(struct iwl_mvm *mvm)
120bfcc09ddSBjoern A. Zeeb {
121bfcc09ddSBjoern A. Zeeb 	struct ieee80211_vif *csa_vif;
122bfcc09ddSBjoern A. Zeeb 
123bfcc09ddSBjoern A. Zeeb 	rcu_read_lock();
124bfcc09ddSBjoern A. Zeeb 
125bfcc09ddSBjoern A. Zeeb 	csa_vif = rcu_dereference(mvm->csa_vif);
126bfcc09ddSBjoern A. Zeeb 	if (!csa_vif || !csa_vif->csa_active)
127bfcc09ddSBjoern A. Zeeb 		goto out_unlock;
128bfcc09ddSBjoern A. Zeeb 
129bfcc09ddSBjoern A. Zeeb 	IWL_DEBUG_TE(mvm, "CSA NOA started\n");
130bfcc09ddSBjoern A. Zeeb 
131bfcc09ddSBjoern A. Zeeb 	/*
132bfcc09ddSBjoern A. Zeeb 	 * CSA NoA is started but we still have beacons to
133bfcc09ddSBjoern A. Zeeb 	 * transmit on the current channel.
134bfcc09ddSBjoern A. Zeeb 	 * So we just do nothing here and the switch
135bfcc09ddSBjoern A. Zeeb 	 * will be performed on the last TBTT.
136bfcc09ddSBjoern A. Zeeb 	 */
137bfcc09ddSBjoern A. Zeeb 	if (!ieee80211_beacon_cntdwn_is_complete(csa_vif)) {
138bfcc09ddSBjoern A. Zeeb 		IWL_WARN(mvm, "CSA NOA started too early\n");
139bfcc09ddSBjoern A. Zeeb 		goto out_unlock;
140bfcc09ddSBjoern A. Zeeb 	}
141bfcc09ddSBjoern A. Zeeb 
142bfcc09ddSBjoern A. Zeeb 	ieee80211_csa_finish(csa_vif);
143bfcc09ddSBjoern A. Zeeb 
144bfcc09ddSBjoern A. Zeeb 	rcu_read_unlock();
145bfcc09ddSBjoern A. Zeeb 
146bfcc09ddSBjoern A. Zeeb 	RCU_INIT_POINTER(mvm->csa_vif, NULL);
147bfcc09ddSBjoern A. Zeeb 
148bfcc09ddSBjoern A. Zeeb 	return;
149bfcc09ddSBjoern A. Zeeb 
150bfcc09ddSBjoern A. Zeeb out_unlock:
151bfcc09ddSBjoern A. Zeeb 	rcu_read_unlock();
152bfcc09ddSBjoern A. Zeeb }
153bfcc09ddSBjoern A. Zeeb 
154bfcc09ddSBjoern A. Zeeb static bool iwl_mvm_te_check_disconnect(struct iwl_mvm *mvm,
155bfcc09ddSBjoern A. Zeeb 					struct ieee80211_vif *vif,
156bfcc09ddSBjoern A. Zeeb 					const char *errmsg)
157bfcc09ddSBjoern A. Zeeb {
158bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
159bfcc09ddSBjoern A. Zeeb 
160bfcc09ddSBjoern A. Zeeb 	if (vif->type != NL80211_IFTYPE_STATION)
161bfcc09ddSBjoern A. Zeeb 		return false;
162bfcc09ddSBjoern A. Zeeb 
163bfcc09ddSBjoern A. Zeeb 	if (!mvmvif->csa_bcn_pending && vif->bss_conf.assoc &&
164bfcc09ddSBjoern A. Zeeb 	    vif->bss_conf.dtim_period)
165bfcc09ddSBjoern A. Zeeb 		return false;
166bfcc09ddSBjoern A. Zeeb 	if (errmsg)
167bfcc09ddSBjoern A. Zeeb 		IWL_ERR(mvm, "%s\n", errmsg);
168bfcc09ddSBjoern A. Zeeb 
169bfcc09ddSBjoern A. Zeeb 	if (mvmvif->csa_bcn_pending) {
170bfcc09ddSBjoern A. Zeeb 		struct iwl_mvm_sta *mvmsta;
171bfcc09ddSBjoern A. Zeeb 
172bfcc09ddSBjoern A. Zeeb 		rcu_read_lock();
173bfcc09ddSBjoern A. Zeeb 		mvmsta = iwl_mvm_sta_from_staid_rcu(mvm, mvmvif->ap_sta_id);
174bfcc09ddSBjoern A. Zeeb 		if (!WARN_ON(!mvmsta))
175bfcc09ddSBjoern A. Zeeb 			iwl_mvm_sta_modify_disable_tx(mvm, mvmsta, false);
176bfcc09ddSBjoern A. Zeeb 		rcu_read_unlock();
177bfcc09ddSBjoern A. Zeeb 	}
178bfcc09ddSBjoern A. Zeeb 
179bfcc09ddSBjoern A. Zeeb 	if (vif->bss_conf.assoc) {
180bfcc09ddSBjoern A. Zeeb 		/*
181bfcc09ddSBjoern A. Zeeb 		 * When not associated, this will be called from
182bfcc09ddSBjoern A. Zeeb 		 * iwl_mvm_event_mlme_callback_ini()
183bfcc09ddSBjoern A. Zeeb 		 */
184bfcc09ddSBjoern A. Zeeb 		iwl_dbg_tlv_time_point(&mvm->fwrt,
185bfcc09ddSBjoern A. Zeeb 				       IWL_FW_INI_TIME_POINT_ASSOC_FAILED,
186bfcc09ddSBjoern A. Zeeb 				       NULL);
187bfcc09ddSBjoern A. Zeeb 	}
188bfcc09ddSBjoern A. Zeeb 
189bfcc09ddSBjoern A. Zeeb 	iwl_mvm_connection_loss(mvm, vif, errmsg);
190bfcc09ddSBjoern A. Zeeb 	return true;
191bfcc09ddSBjoern A. Zeeb }
192bfcc09ddSBjoern A. Zeeb 
193bfcc09ddSBjoern A. Zeeb static void
194bfcc09ddSBjoern A. Zeeb iwl_mvm_te_handle_notify_csa(struct iwl_mvm *mvm,
195bfcc09ddSBjoern A. Zeeb 			     struct iwl_mvm_time_event_data *te_data,
196bfcc09ddSBjoern A. Zeeb 			     struct iwl_time_event_notif *notif)
197bfcc09ddSBjoern A. Zeeb {
198bfcc09ddSBjoern A. Zeeb 	struct ieee80211_vif *vif = te_data->vif;
199bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
200bfcc09ddSBjoern A. Zeeb 
201bfcc09ddSBjoern A. Zeeb 	if (!notif->status)
202bfcc09ddSBjoern A. Zeeb 		IWL_DEBUG_TE(mvm, "CSA time event failed to start\n");
203bfcc09ddSBjoern A. Zeeb 
204bfcc09ddSBjoern A. Zeeb 	switch (te_data->vif->type) {
205bfcc09ddSBjoern A. Zeeb 	case NL80211_IFTYPE_AP:
206bfcc09ddSBjoern A. Zeeb 		if (!notif->status)
207bfcc09ddSBjoern A. Zeeb 			mvmvif->csa_failed = true;
208bfcc09ddSBjoern A. Zeeb 		iwl_mvm_csa_noa_start(mvm);
209bfcc09ddSBjoern A. Zeeb 		break;
210bfcc09ddSBjoern A. Zeeb 	case NL80211_IFTYPE_STATION:
211bfcc09ddSBjoern A. Zeeb 		if (!notif->status) {
212bfcc09ddSBjoern A. Zeeb 			iwl_mvm_connection_loss(mvm, vif,
213bfcc09ddSBjoern A. Zeeb 						"CSA TE failed to start");
214bfcc09ddSBjoern A. Zeeb 			break;
215bfcc09ddSBjoern A. Zeeb 		}
216bfcc09ddSBjoern A. Zeeb 		iwl_mvm_csa_client_absent(mvm, te_data->vif);
217bfcc09ddSBjoern A. Zeeb 		cancel_delayed_work(&mvmvif->csa_work);
218bfcc09ddSBjoern A. Zeeb 		ieee80211_chswitch_done(te_data->vif, true);
219bfcc09ddSBjoern A. Zeeb 		break;
220bfcc09ddSBjoern A. Zeeb 	default:
221bfcc09ddSBjoern A. Zeeb 		/* should never happen */
222bfcc09ddSBjoern A. Zeeb 		WARN_ON_ONCE(1);
223bfcc09ddSBjoern A. Zeeb 		break;
224bfcc09ddSBjoern A. Zeeb 	}
225bfcc09ddSBjoern A. Zeeb 
226bfcc09ddSBjoern A. Zeeb 	/* we don't need it anymore */
227bfcc09ddSBjoern A. Zeeb 	iwl_mvm_te_clear_data(mvm, te_data);
228bfcc09ddSBjoern A. Zeeb }
229bfcc09ddSBjoern A. Zeeb 
230bfcc09ddSBjoern A. Zeeb static void iwl_mvm_te_check_trigger(struct iwl_mvm *mvm,
231bfcc09ddSBjoern A. Zeeb 				     struct iwl_time_event_notif *notif,
232bfcc09ddSBjoern A. Zeeb 				     struct iwl_mvm_time_event_data *te_data)
233bfcc09ddSBjoern A. Zeeb {
234bfcc09ddSBjoern A. Zeeb 	struct iwl_fw_dbg_trigger_tlv *trig;
235bfcc09ddSBjoern A. Zeeb 	struct iwl_fw_dbg_trigger_time_event *te_trig;
236bfcc09ddSBjoern A. Zeeb 	int i;
237bfcc09ddSBjoern A. Zeeb 
238bfcc09ddSBjoern A. Zeeb 	trig = iwl_fw_dbg_trigger_on(&mvm->fwrt,
239bfcc09ddSBjoern A. Zeeb 				     ieee80211_vif_to_wdev(te_data->vif),
240bfcc09ddSBjoern A. Zeeb 				     FW_DBG_TRIGGER_TIME_EVENT);
241bfcc09ddSBjoern A. Zeeb 	if (!trig)
242bfcc09ddSBjoern A. Zeeb 		return;
243bfcc09ddSBjoern A. Zeeb 
244bfcc09ddSBjoern A. Zeeb 	te_trig = (void *)trig->data;
245bfcc09ddSBjoern A. Zeeb 
246bfcc09ddSBjoern A. Zeeb 	for (i = 0; i < ARRAY_SIZE(te_trig->time_events); i++) {
247bfcc09ddSBjoern A. Zeeb 		u32 trig_te_id = le32_to_cpu(te_trig->time_events[i].id);
248bfcc09ddSBjoern A. Zeeb 		u32 trig_action_bitmap =
249bfcc09ddSBjoern A. Zeeb 			le32_to_cpu(te_trig->time_events[i].action_bitmap);
250bfcc09ddSBjoern A. Zeeb 		u32 trig_status_bitmap =
251bfcc09ddSBjoern A. Zeeb 			le32_to_cpu(te_trig->time_events[i].status_bitmap);
252bfcc09ddSBjoern A. Zeeb 
253bfcc09ddSBjoern A. Zeeb 		if (trig_te_id != te_data->id ||
254bfcc09ddSBjoern A. Zeeb 		    !(trig_action_bitmap & le32_to_cpu(notif->action)) ||
255bfcc09ddSBjoern A. Zeeb 		    !(trig_status_bitmap & BIT(le32_to_cpu(notif->status))))
256bfcc09ddSBjoern A. Zeeb 			continue;
257bfcc09ddSBjoern A. Zeeb 
258bfcc09ddSBjoern A. Zeeb 		iwl_fw_dbg_collect_trig(&mvm->fwrt, trig,
259bfcc09ddSBjoern A. Zeeb 					"Time event %d Action 0x%x received status: %d",
260bfcc09ddSBjoern A. Zeeb 					te_data->id,
261bfcc09ddSBjoern A. Zeeb 					le32_to_cpu(notif->action),
262bfcc09ddSBjoern A. Zeeb 					le32_to_cpu(notif->status));
263bfcc09ddSBjoern A. Zeeb 		break;
264bfcc09ddSBjoern A. Zeeb 	}
265bfcc09ddSBjoern A. Zeeb }
266bfcc09ddSBjoern A. Zeeb 
267bfcc09ddSBjoern A. Zeeb static void iwl_mvm_p2p_roc_finished(struct iwl_mvm *mvm)
268bfcc09ddSBjoern A. Zeeb {
269bfcc09ddSBjoern A. Zeeb 	/*
270bfcc09ddSBjoern A. Zeeb 	 * If the IWL_MVM_STATUS_NEED_FLUSH_P2P is already set, then the
271bfcc09ddSBjoern A. Zeeb 	 * roc_done_wk is already scheduled or running, so don't schedule it
272bfcc09ddSBjoern A. Zeeb 	 * again to avoid a race where the roc_done_wk clears this bit after
273bfcc09ddSBjoern A. Zeeb 	 * it is set here, affecting the next run of the roc_done_wk.
274bfcc09ddSBjoern A. Zeeb 	 */
275bfcc09ddSBjoern A. Zeeb 	if (!test_and_set_bit(IWL_MVM_STATUS_NEED_FLUSH_P2P, &mvm->status))
276bfcc09ddSBjoern A. Zeeb 		iwl_mvm_roc_finished(mvm);
277bfcc09ddSBjoern A. Zeeb }
278bfcc09ddSBjoern A. Zeeb 
279bfcc09ddSBjoern A. Zeeb /*
280bfcc09ddSBjoern A. Zeeb  * Handles a FW notification for an event that is known to the driver.
281bfcc09ddSBjoern A. Zeeb  *
282bfcc09ddSBjoern A. Zeeb  * @mvm: the mvm component
283bfcc09ddSBjoern A. Zeeb  * @te_data: the time event data
284bfcc09ddSBjoern A. Zeeb  * @notif: the notification data corresponding the time event data.
285bfcc09ddSBjoern A. Zeeb  */
286bfcc09ddSBjoern A. Zeeb static void iwl_mvm_te_handle_notif(struct iwl_mvm *mvm,
287bfcc09ddSBjoern A. Zeeb 				    struct iwl_mvm_time_event_data *te_data,
288bfcc09ddSBjoern A. Zeeb 				    struct iwl_time_event_notif *notif)
289bfcc09ddSBjoern A. Zeeb {
290bfcc09ddSBjoern A. Zeeb 	lockdep_assert_held(&mvm->time_event_lock);
291bfcc09ddSBjoern A. Zeeb 
292bfcc09ddSBjoern A. Zeeb 	IWL_DEBUG_TE(mvm, "Handle time event notif - UID = 0x%x action %d\n",
293bfcc09ddSBjoern A. Zeeb 		     le32_to_cpu(notif->unique_id),
294bfcc09ddSBjoern A. Zeeb 		     le32_to_cpu(notif->action));
295bfcc09ddSBjoern A. Zeeb 
296bfcc09ddSBjoern A. Zeeb 	iwl_mvm_te_check_trigger(mvm, notif, te_data);
297bfcc09ddSBjoern A. Zeeb 
298bfcc09ddSBjoern A. Zeeb 	/*
299bfcc09ddSBjoern A. Zeeb 	 * The FW sends the start/end time event notifications even for events
300bfcc09ddSBjoern A. Zeeb 	 * that it fails to schedule. This is indicated in the status field of
301bfcc09ddSBjoern A. Zeeb 	 * the notification. This happens in cases that the scheduler cannot
302bfcc09ddSBjoern A. Zeeb 	 * find a schedule that can handle the event (for example requesting a
303bfcc09ddSBjoern A. Zeeb 	 * P2P Device discoveribility, while there are other higher priority
304bfcc09ddSBjoern A. Zeeb 	 * events in the system).
305bfcc09ddSBjoern A. Zeeb 	 */
306bfcc09ddSBjoern A. Zeeb 	if (!le32_to_cpu(notif->status)) {
307bfcc09ddSBjoern A. Zeeb 		const char *msg;
308bfcc09ddSBjoern A. Zeeb 
309bfcc09ddSBjoern A. Zeeb 		if (notif->action & cpu_to_le32(TE_V2_NOTIF_HOST_EVENT_START))
310bfcc09ddSBjoern A. Zeeb 			msg = "Time Event start notification failure";
311bfcc09ddSBjoern A. Zeeb 		else
312bfcc09ddSBjoern A. Zeeb 			msg = "Time Event end notification failure";
313bfcc09ddSBjoern A. Zeeb 
314bfcc09ddSBjoern A. Zeeb 		IWL_DEBUG_TE(mvm, "%s\n", msg);
315bfcc09ddSBjoern A. Zeeb 
316bfcc09ddSBjoern A. Zeeb 		if (iwl_mvm_te_check_disconnect(mvm, te_data->vif, msg)) {
317bfcc09ddSBjoern A. Zeeb 			iwl_mvm_te_clear_data(mvm, te_data);
318bfcc09ddSBjoern A. Zeeb 			return;
319bfcc09ddSBjoern A. Zeeb 		}
320bfcc09ddSBjoern A. Zeeb 	}
321bfcc09ddSBjoern A. Zeeb 
322bfcc09ddSBjoern A. Zeeb 	if (le32_to_cpu(notif->action) & TE_V2_NOTIF_HOST_EVENT_END) {
323bfcc09ddSBjoern A. Zeeb 		IWL_DEBUG_TE(mvm,
324bfcc09ddSBjoern A. Zeeb 			     "TE ended - current time %lu, estimated end %lu\n",
325bfcc09ddSBjoern A. Zeeb 			     jiffies, te_data->end_jiffies);
326bfcc09ddSBjoern A. Zeeb 
327bfcc09ddSBjoern A. Zeeb 		switch (te_data->vif->type) {
328bfcc09ddSBjoern A. Zeeb 		case NL80211_IFTYPE_P2P_DEVICE:
329bfcc09ddSBjoern A. Zeeb 			ieee80211_remain_on_channel_expired(mvm->hw);
330bfcc09ddSBjoern A. Zeeb 			iwl_mvm_p2p_roc_finished(mvm);
331bfcc09ddSBjoern A. Zeeb 			break;
332bfcc09ddSBjoern A. Zeeb 		case NL80211_IFTYPE_STATION:
333bfcc09ddSBjoern A. Zeeb 			/*
334bfcc09ddSBjoern A. Zeeb 			 * If we are switching channel, don't disconnect
335bfcc09ddSBjoern A. Zeeb 			 * if the time event is already done. Beacons can
336bfcc09ddSBjoern A. Zeeb 			 * be delayed a bit after the switch.
337bfcc09ddSBjoern A. Zeeb 			 */
338bfcc09ddSBjoern A. Zeeb 			if (te_data->id == TE_CHANNEL_SWITCH_PERIOD) {
339bfcc09ddSBjoern A. Zeeb 				IWL_DEBUG_TE(mvm,
340bfcc09ddSBjoern A. Zeeb 					     "No beacon heard and the CS time event is over, don't disconnect\n");
341bfcc09ddSBjoern A. Zeeb 				break;
342bfcc09ddSBjoern A. Zeeb 			}
343bfcc09ddSBjoern A. Zeeb 
344bfcc09ddSBjoern A. Zeeb 			/*
345bfcc09ddSBjoern A. Zeeb 			 * By now, we should have finished association
346bfcc09ddSBjoern A. Zeeb 			 * and know the dtim period.
347bfcc09ddSBjoern A. Zeeb 			 */
348bfcc09ddSBjoern A. Zeeb 			iwl_mvm_te_check_disconnect(mvm, te_data->vif,
349bfcc09ddSBjoern A. Zeeb 				!te_data->vif->bss_conf.assoc ?
350bfcc09ddSBjoern A. Zeeb 				"Not associated and the time event is over already..." :
351bfcc09ddSBjoern A. Zeeb 				"No beacon heard and the time event is over already...");
352bfcc09ddSBjoern A. Zeeb 			break;
353bfcc09ddSBjoern A. Zeeb 		default:
354bfcc09ddSBjoern A. Zeeb 			break;
355bfcc09ddSBjoern A. Zeeb 		}
356bfcc09ddSBjoern A. Zeeb 
357bfcc09ddSBjoern A. Zeeb 		iwl_mvm_te_clear_data(mvm, te_data);
358bfcc09ddSBjoern A. Zeeb 	} else if (le32_to_cpu(notif->action) & TE_V2_NOTIF_HOST_EVENT_START) {
359bfcc09ddSBjoern A. Zeeb 		te_data->running = true;
360bfcc09ddSBjoern A. Zeeb 		te_data->end_jiffies = TU_TO_EXP_TIME(te_data->duration);
361bfcc09ddSBjoern A. Zeeb 
362bfcc09ddSBjoern A. Zeeb 		if (te_data->vif->type == NL80211_IFTYPE_P2P_DEVICE) {
363bfcc09ddSBjoern A. Zeeb 			set_bit(IWL_MVM_STATUS_ROC_RUNNING, &mvm->status);
364bfcc09ddSBjoern A. Zeeb 			ieee80211_ready_on_channel(mvm->hw);
365bfcc09ddSBjoern A. Zeeb 		} else if (te_data->id == TE_CHANNEL_SWITCH_PERIOD) {
366bfcc09ddSBjoern A. Zeeb 			iwl_mvm_te_handle_notify_csa(mvm, te_data, notif);
367bfcc09ddSBjoern A. Zeeb 		}
368bfcc09ddSBjoern A. Zeeb 	} else {
369bfcc09ddSBjoern A. Zeeb 		IWL_WARN(mvm, "Got TE with unknown action\n");
370bfcc09ddSBjoern A. Zeeb 	}
371bfcc09ddSBjoern A. Zeeb }
372bfcc09ddSBjoern A. Zeeb 
373bfcc09ddSBjoern A. Zeeb /*
374bfcc09ddSBjoern A. Zeeb  * Handle A Aux ROC time event
375bfcc09ddSBjoern A. Zeeb  */
376bfcc09ddSBjoern A. Zeeb static int iwl_mvm_aux_roc_te_handle_notif(struct iwl_mvm *mvm,
377bfcc09ddSBjoern A. Zeeb 					   struct iwl_time_event_notif *notif)
378bfcc09ddSBjoern A. Zeeb {
379bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_time_event_data *te_data, *tmp;
380bfcc09ddSBjoern A. Zeeb 	bool aux_roc_te = false;
381bfcc09ddSBjoern A. Zeeb 
382bfcc09ddSBjoern A. Zeeb 	list_for_each_entry_safe(te_data, tmp, &mvm->aux_roc_te_list, list) {
383bfcc09ddSBjoern A. Zeeb 		if (le32_to_cpu(notif->unique_id) == te_data->uid) {
384bfcc09ddSBjoern A. Zeeb 			aux_roc_te = true;
385bfcc09ddSBjoern A. Zeeb 			break;
386bfcc09ddSBjoern A. Zeeb 		}
387bfcc09ddSBjoern A. Zeeb 	}
388bfcc09ddSBjoern A. Zeeb 	if (!aux_roc_te) /* Not a Aux ROC time event */
389bfcc09ddSBjoern A. Zeeb 		return -EINVAL;
390bfcc09ddSBjoern A. Zeeb 
391bfcc09ddSBjoern A. Zeeb 	iwl_mvm_te_check_trigger(mvm, notif, te_data);
392bfcc09ddSBjoern A. Zeeb 
393bfcc09ddSBjoern A. Zeeb 	IWL_DEBUG_TE(mvm,
394bfcc09ddSBjoern A. Zeeb 		     "Aux ROC time event notification  - UID = 0x%x action %d (error = %d)\n",
395bfcc09ddSBjoern A. Zeeb 		     le32_to_cpu(notif->unique_id),
396bfcc09ddSBjoern A. Zeeb 		     le32_to_cpu(notif->action), le32_to_cpu(notif->status));
397bfcc09ddSBjoern A. Zeeb 
398bfcc09ddSBjoern A. Zeeb 	if (!le32_to_cpu(notif->status) ||
399bfcc09ddSBjoern A. Zeeb 	    le32_to_cpu(notif->action) == TE_V2_NOTIF_HOST_EVENT_END) {
400bfcc09ddSBjoern A. Zeeb 		/* End TE, notify mac80211 */
401bfcc09ddSBjoern A. Zeeb 		ieee80211_remain_on_channel_expired(mvm->hw);
402bfcc09ddSBjoern A. Zeeb 		iwl_mvm_roc_finished(mvm); /* flush aux queue */
403bfcc09ddSBjoern A. Zeeb 		list_del(&te_data->list); /* remove from list */
404bfcc09ddSBjoern A. Zeeb 		te_data->running = false;
405bfcc09ddSBjoern A. Zeeb 		te_data->vif = NULL;
406bfcc09ddSBjoern A. Zeeb 		te_data->uid = 0;
407bfcc09ddSBjoern A. Zeeb 		te_data->id = TE_MAX;
408bfcc09ddSBjoern A. Zeeb 	} else if (le32_to_cpu(notif->action) == TE_V2_NOTIF_HOST_EVENT_START) {
409bfcc09ddSBjoern A. Zeeb 		set_bit(IWL_MVM_STATUS_ROC_AUX_RUNNING, &mvm->status);
410bfcc09ddSBjoern A. Zeeb 		te_data->running = true;
411bfcc09ddSBjoern A. Zeeb 		ieee80211_ready_on_channel(mvm->hw); /* Start TE */
412bfcc09ddSBjoern A. Zeeb 	} else {
413bfcc09ddSBjoern A. Zeeb 		IWL_DEBUG_TE(mvm,
414bfcc09ddSBjoern A. Zeeb 			     "ERROR: Unknown Aux ROC Time Event (action = %d)\n",
415bfcc09ddSBjoern A. Zeeb 			     le32_to_cpu(notif->action));
416bfcc09ddSBjoern A. Zeeb 		return -EINVAL;
417bfcc09ddSBjoern A. Zeeb 	}
418bfcc09ddSBjoern A. Zeeb 
419bfcc09ddSBjoern A. Zeeb 	return 0;
420bfcc09ddSBjoern A. Zeeb }
421bfcc09ddSBjoern A. Zeeb 
422bfcc09ddSBjoern A. Zeeb /*
423bfcc09ddSBjoern A. Zeeb  * The Rx handler for time event notifications
424bfcc09ddSBjoern A. Zeeb  */
425bfcc09ddSBjoern A. Zeeb void iwl_mvm_rx_time_event_notif(struct iwl_mvm *mvm,
426bfcc09ddSBjoern A. Zeeb 				 struct iwl_rx_cmd_buffer *rxb)
427bfcc09ddSBjoern A. Zeeb {
428bfcc09ddSBjoern A. Zeeb 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
429bfcc09ddSBjoern A. Zeeb 	struct iwl_time_event_notif *notif = (void *)pkt->data;
430bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_time_event_data *te_data, *tmp;
431bfcc09ddSBjoern A. Zeeb 
432bfcc09ddSBjoern A. Zeeb 	IWL_DEBUG_TE(mvm, "Time event notification - UID = 0x%x action %d\n",
433bfcc09ddSBjoern A. Zeeb 		     le32_to_cpu(notif->unique_id),
434bfcc09ddSBjoern A. Zeeb 		     le32_to_cpu(notif->action));
435bfcc09ddSBjoern A. Zeeb 
436bfcc09ddSBjoern A. Zeeb 	spin_lock_bh(&mvm->time_event_lock);
437bfcc09ddSBjoern A. Zeeb 	/* This time event is triggered for Aux ROC request */
438bfcc09ddSBjoern A. Zeeb 	if (!iwl_mvm_aux_roc_te_handle_notif(mvm, notif))
439bfcc09ddSBjoern A. Zeeb 		goto unlock;
440bfcc09ddSBjoern A. Zeeb 
441bfcc09ddSBjoern A. Zeeb 	list_for_each_entry_safe(te_data, tmp, &mvm->time_event_list, list) {
442bfcc09ddSBjoern A. Zeeb 		if (le32_to_cpu(notif->unique_id) == te_data->uid)
443bfcc09ddSBjoern A. Zeeb 			iwl_mvm_te_handle_notif(mvm, te_data, notif);
444bfcc09ddSBjoern A. Zeeb 	}
445bfcc09ddSBjoern A. Zeeb unlock:
446bfcc09ddSBjoern A. Zeeb 	spin_unlock_bh(&mvm->time_event_lock);
447bfcc09ddSBjoern A. Zeeb }
448bfcc09ddSBjoern A. Zeeb 
449bfcc09ddSBjoern A. Zeeb static bool iwl_mvm_te_notif(struct iwl_notif_wait_data *notif_wait,
450bfcc09ddSBjoern A. Zeeb 			     struct iwl_rx_packet *pkt, void *data)
451bfcc09ddSBjoern A. Zeeb {
452bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm *mvm =
453bfcc09ddSBjoern A. Zeeb 		container_of(notif_wait, struct iwl_mvm, notif_wait);
454bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_time_event_data *te_data = data;
455bfcc09ddSBjoern A. Zeeb 	struct iwl_time_event_notif *resp;
456bfcc09ddSBjoern A. Zeeb 	int resp_len = iwl_rx_packet_payload_len(pkt);
457bfcc09ddSBjoern A. Zeeb 
458bfcc09ddSBjoern A. Zeeb 	if (WARN_ON(pkt->hdr.cmd != TIME_EVENT_NOTIFICATION))
459bfcc09ddSBjoern A. Zeeb 		return true;
460bfcc09ddSBjoern A. Zeeb 
461bfcc09ddSBjoern A. Zeeb 	if (WARN_ON_ONCE(resp_len != sizeof(*resp))) {
462bfcc09ddSBjoern A. Zeeb 		IWL_ERR(mvm, "Invalid TIME_EVENT_NOTIFICATION response\n");
463bfcc09ddSBjoern A. Zeeb 		return true;
464bfcc09ddSBjoern A. Zeeb 	}
465bfcc09ddSBjoern A. Zeeb 
466bfcc09ddSBjoern A. Zeeb 	resp = (void *)pkt->data;
467bfcc09ddSBjoern A. Zeeb 
468bfcc09ddSBjoern A. Zeeb 	/* te_data->uid is already set in the TIME_EVENT_CMD response */
469bfcc09ddSBjoern A. Zeeb 	if (le32_to_cpu(resp->unique_id) != te_data->uid)
470bfcc09ddSBjoern A. Zeeb 		return false;
471bfcc09ddSBjoern A. Zeeb 
472bfcc09ddSBjoern A. Zeeb 	IWL_DEBUG_TE(mvm, "TIME_EVENT_NOTIFICATION response - UID = 0x%x\n",
473bfcc09ddSBjoern A. Zeeb 		     te_data->uid);
474bfcc09ddSBjoern A. Zeeb 	if (!resp->status)
475bfcc09ddSBjoern A. Zeeb 		IWL_ERR(mvm,
476bfcc09ddSBjoern A. Zeeb 			"TIME_EVENT_NOTIFICATION received but not executed\n");
477bfcc09ddSBjoern A. Zeeb 
478bfcc09ddSBjoern A. Zeeb 	return true;
479bfcc09ddSBjoern A. Zeeb }
480bfcc09ddSBjoern A. Zeeb 
481bfcc09ddSBjoern A. Zeeb static bool iwl_mvm_time_event_response(struct iwl_notif_wait_data *notif_wait,
482bfcc09ddSBjoern A. Zeeb 					struct iwl_rx_packet *pkt, void *data)
483bfcc09ddSBjoern A. Zeeb {
484bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm *mvm =
485bfcc09ddSBjoern A. Zeeb 		container_of(notif_wait, struct iwl_mvm, notif_wait);
486bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_time_event_data *te_data = data;
487bfcc09ddSBjoern A. Zeeb 	struct iwl_time_event_resp *resp;
488bfcc09ddSBjoern A. Zeeb 	int resp_len = iwl_rx_packet_payload_len(pkt);
489bfcc09ddSBjoern A. Zeeb 
490bfcc09ddSBjoern A. Zeeb 	if (WARN_ON(pkt->hdr.cmd != TIME_EVENT_CMD))
491bfcc09ddSBjoern A. Zeeb 		return true;
492bfcc09ddSBjoern A. Zeeb 
493bfcc09ddSBjoern A. Zeeb 	if (WARN_ON_ONCE(resp_len != sizeof(*resp))) {
494bfcc09ddSBjoern A. Zeeb 		IWL_ERR(mvm, "Invalid TIME_EVENT_CMD response\n");
495bfcc09ddSBjoern A. Zeeb 		return true;
496bfcc09ddSBjoern A. Zeeb 	}
497bfcc09ddSBjoern A. Zeeb 
498bfcc09ddSBjoern A. Zeeb 	resp = (void *)pkt->data;
499bfcc09ddSBjoern A. Zeeb 
500bfcc09ddSBjoern A. Zeeb 	/* we should never get a response to another TIME_EVENT_CMD here */
501bfcc09ddSBjoern A. Zeeb 	if (WARN_ON_ONCE(le32_to_cpu(resp->id) != te_data->id))
502bfcc09ddSBjoern A. Zeeb 		return false;
503bfcc09ddSBjoern A. Zeeb 
504bfcc09ddSBjoern A. Zeeb 	te_data->uid = le32_to_cpu(resp->unique_id);
505bfcc09ddSBjoern A. Zeeb 	IWL_DEBUG_TE(mvm, "TIME_EVENT_CMD response - UID = 0x%x\n",
506bfcc09ddSBjoern A. Zeeb 		     te_data->uid);
507bfcc09ddSBjoern A. Zeeb 	return true;
508bfcc09ddSBjoern A. Zeeb }
509bfcc09ddSBjoern A. Zeeb 
510bfcc09ddSBjoern A. Zeeb static int iwl_mvm_time_event_send_add(struct iwl_mvm *mvm,
511bfcc09ddSBjoern A. Zeeb 				       struct ieee80211_vif *vif,
512bfcc09ddSBjoern A. Zeeb 				       struct iwl_mvm_time_event_data *te_data,
513bfcc09ddSBjoern A. Zeeb 				       struct iwl_time_event_cmd *te_cmd)
514bfcc09ddSBjoern A. Zeeb {
515bfcc09ddSBjoern A. Zeeb 	static const u16 time_event_response[] = { TIME_EVENT_CMD };
516bfcc09ddSBjoern A. Zeeb 	struct iwl_notification_wait wait_time_event;
517bfcc09ddSBjoern A. Zeeb 	int ret;
518bfcc09ddSBjoern A. Zeeb 
519bfcc09ddSBjoern A. Zeeb 	lockdep_assert_held(&mvm->mutex);
520bfcc09ddSBjoern A. Zeeb 
521bfcc09ddSBjoern A. Zeeb 	IWL_DEBUG_TE(mvm, "Add new TE, duration %d TU\n",
522bfcc09ddSBjoern A. Zeeb 		     le32_to_cpu(te_cmd->duration));
523bfcc09ddSBjoern A. Zeeb 
524bfcc09ddSBjoern A. Zeeb 	spin_lock_bh(&mvm->time_event_lock);
525bfcc09ddSBjoern A. Zeeb 	if (WARN_ON(te_data->id != TE_MAX)) {
526bfcc09ddSBjoern A. Zeeb 		spin_unlock_bh(&mvm->time_event_lock);
527bfcc09ddSBjoern A. Zeeb 		return -EIO;
528bfcc09ddSBjoern A. Zeeb 	}
529bfcc09ddSBjoern A. Zeeb 	te_data->vif = vif;
530bfcc09ddSBjoern A. Zeeb 	te_data->duration = le32_to_cpu(te_cmd->duration);
531bfcc09ddSBjoern A. Zeeb 	te_data->id = le32_to_cpu(te_cmd->id);
532bfcc09ddSBjoern A. Zeeb 	list_add_tail(&te_data->list, &mvm->time_event_list);
533bfcc09ddSBjoern A. Zeeb 	spin_unlock_bh(&mvm->time_event_lock);
534bfcc09ddSBjoern A. Zeeb 
535bfcc09ddSBjoern A. Zeeb 	/*
536bfcc09ddSBjoern A. Zeeb 	 * Use a notification wait, which really just processes the
537bfcc09ddSBjoern A. Zeeb 	 * command response and doesn't wait for anything, in order
538bfcc09ddSBjoern A. Zeeb 	 * to be able to process the response and get the UID inside
539bfcc09ddSBjoern A. Zeeb 	 * the RX path. Using CMD_WANT_SKB doesn't work because it
540bfcc09ddSBjoern A. Zeeb 	 * stores the buffer and then wakes up this thread, by which
541bfcc09ddSBjoern A. Zeeb 	 * time another notification (that the time event started)
542bfcc09ddSBjoern A. Zeeb 	 * might already be processed unsuccessfully.
543bfcc09ddSBjoern A. Zeeb 	 */
544bfcc09ddSBjoern A. Zeeb 	iwl_init_notification_wait(&mvm->notif_wait, &wait_time_event,
545bfcc09ddSBjoern A. Zeeb 				   time_event_response,
546bfcc09ddSBjoern A. Zeeb 				   ARRAY_SIZE(time_event_response),
547bfcc09ddSBjoern A. Zeeb 				   iwl_mvm_time_event_response, te_data);
548bfcc09ddSBjoern A. Zeeb 
549bfcc09ddSBjoern A. Zeeb 	ret = iwl_mvm_send_cmd_pdu(mvm, TIME_EVENT_CMD, 0,
550bfcc09ddSBjoern A. Zeeb 					    sizeof(*te_cmd), te_cmd);
551bfcc09ddSBjoern A. Zeeb 	if (ret) {
552bfcc09ddSBjoern A. Zeeb 		IWL_ERR(mvm, "Couldn't send TIME_EVENT_CMD: %d\n", ret);
553bfcc09ddSBjoern A. Zeeb 		iwl_remove_notification(&mvm->notif_wait, &wait_time_event);
554bfcc09ddSBjoern A. Zeeb 		goto out_clear_te;
555bfcc09ddSBjoern A. Zeeb 	}
556bfcc09ddSBjoern A. Zeeb 
557bfcc09ddSBjoern A. Zeeb 	/* No need to wait for anything, so just pass 1 (0 isn't valid) */
558bfcc09ddSBjoern A. Zeeb 	ret = iwl_wait_notification(&mvm->notif_wait, &wait_time_event, 1);
559bfcc09ddSBjoern A. Zeeb 	/* should never fail */
560bfcc09ddSBjoern A. Zeeb 	WARN_ON_ONCE(ret);
561bfcc09ddSBjoern A. Zeeb 
562bfcc09ddSBjoern A. Zeeb 	if (ret) {
563bfcc09ddSBjoern A. Zeeb  out_clear_te:
564bfcc09ddSBjoern A. Zeeb 		spin_lock_bh(&mvm->time_event_lock);
565bfcc09ddSBjoern A. Zeeb 		iwl_mvm_te_clear_data(mvm, te_data);
566bfcc09ddSBjoern A. Zeeb 		spin_unlock_bh(&mvm->time_event_lock);
567bfcc09ddSBjoern A. Zeeb 	}
568bfcc09ddSBjoern A. Zeeb 	return ret;
569bfcc09ddSBjoern A. Zeeb }
570bfcc09ddSBjoern A. Zeeb 
571bfcc09ddSBjoern A. Zeeb void iwl_mvm_protect_session(struct iwl_mvm *mvm,
572bfcc09ddSBjoern A. Zeeb 			     struct ieee80211_vif *vif,
573bfcc09ddSBjoern A. Zeeb 			     u32 duration, u32 min_duration,
574bfcc09ddSBjoern A. Zeeb 			     u32 max_delay, bool wait_for_notif)
575bfcc09ddSBjoern A. Zeeb {
576bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
577bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_time_event_data *te_data = &mvmvif->time_event_data;
578bfcc09ddSBjoern A. Zeeb 	const u16 te_notif_response[] = { TIME_EVENT_NOTIFICATION };
579bfcc09ddSBjoern A. Zeeb 	struct iwl_notification_wait wait_te_notif;
580bfcc09ddSBjoern A. Zeeb 	struct iwl_time_event_cmd time_cmd = {};
581bfcc09ddSBjoern A. Zeeb 
582bfcc09ddSBjoern A. Zeeb 	lockdep_assert_held(&mvm->mutex);
583bfcc09ddSBjoern A. Zeeb 
584bfcc09ddSBjoern A. Zeeb 	if (te_data->running &&
585bfcc09ddSBjoern A. Zeeb 	    time_after(te_data->end_jiffies, TU_TO_EXP_TIME(min_duration))) {
586bfcc09ddSBjoern A. Zeeb 		IWL_DEBUG_TE(mvm, "We have enough time in the current TE: %u\n",
587bfcc09ddSBjoern A. Zeeb 			     jiffies_to_msecs(te_data->end_jiffies - jiffies));
588bfcc09ddSBjoern A. Zeeb 		return;
589bfcc09ddSBjoern A. Zeeb 	}
590bfcc09ddSBjoern A. Zeeb 
591bfcc09ddSBjoern A. Zeeb 	if (te_data->running) {
592bfcc09ddSBjoern A. Zeeb 		IWL_DEBUG_TE(mvm, "extend 0x%x: only %u ms left\n",
593bfcc09ddSBjoern A. Zeeb 			     te_data->uid,
594bfcc09ddSBjoern A. Zeeb 			     jiffies_to_msecs(te_data->end_jiffies - jiffies));
595bfcc09ddSBjoern A. Zeeb 		/*
596bfcc09ddSBjoern A. Zeeb 		 * we don't have enough time
597bfcc09ddSBjoern A. Zeeb 		 * cancel the current TE and issue a new one
598bfcc09ddSBjoern A. Zeeb 		 * Of course it would be better to remove the old one only
599bfcc09ddSBjoern A. Zeeb 		 * when the new one is added, but we don't care if we are off
600bfcc09ddSBjoern A. Zeeb 		 * channel for a bit. All we need to do, is not to return
601bfcc09ddSBjoern A. Zeeb 		 * before we actually begin to be on the channel.
602bfcc09ddSBjoern A. Zeeb 		 */
603bfcc09ddSBjoern A. Zeeb 		iwl_mvm_stop_session_protection(mvm, vif);
604bfcc09ddSBjoern A. Zeeb 	}
605bfcc09ddSBjoern A. Zeeb 
606bfcc09ddSBjoern A. Zeeb 	time_cmd.action = cpu_to_le32(FW_CTXT_ACTION_ADD);
607bfcc09ddSBjoern A. Zeeb 	time_cmd.id_and_color =
608bfcc09ddSBjoern A. Zeeb 		cpu_to_le32(FW_CMD_ID_AND_COLOR(mvmvif->id, mvmvif->color));
609bfcc09ddSBjoern A. Zeeb 	time_cmd.id = cpu_to_le32(TE_BSS_STA_AGGRESSIVE_ASSOC);
610bfcc09ddSBjoern A. Zeeb 
611bfcc09ddSBjoern A. Zeeb 	time_cmd.apply_time = cpu_to_le32(0);
612bfcc09ddSBjoern A. Zeeb 
613bfcc09ddSBjoern A. Zeeb 	time_cmd.max_frags = TE_V2_FRAG_NONE;
614bfcc09ddSBjoern A. Zeeb 	time_cmd.max_delay = cpu_to_le32(max_delay);
615bfcc09ddSBjoern A. Zeeb 	/* TODO: why do we need to interval = bi if it is not periodic? */
616bfcc09ddSBjoern A. Zeeb 	time_cmd.interval = cpu_to_le32(1);
617bfcc09ddSBjoern A. Zeeb 	time_cmd.duration = cpu_to_le32(duration);
618bfcc09ddSBjoern A. Zeeb 	time_cmd.repeat = 1;
619bfcc09ddSBjoern A. Zeeb 	time_cmd.policy = cpu_to_le16(TE_V2_NOTIF_HOST_EVENT_START |
620bfcc09ddSBjoern A. Zeeb 				      TE_V2_NOTIF_HOST_EVENT_END |
621bfcc09ddSBjoern A. Zeeb 				      TE_V2_START_IMMEDIATELY);
622bfcc09ddSBjoern A. Zeeb 
623bfcc09ddSBjoern A. Zeeb 	if (!wait_for_notif) {
624bfcc09ddSBjoern A. Zeeb 		iwl_mvm_time_event_send_add(mvm, vif, te_data, &time_cmd);
625bfcc09ddSBjoern A. Zeeb 		return;
626bfcc09ddSBjoern A. Zeeb 	}
627bfcc09ddSBjoern A. Zeeb 
628bfcc09ddSBjoern A. Zeeb 	/*
629bfcc09ddSBjoern A. Zeeb 	 * Create notification_wait for the TIME_EVENT_NOTIFICATION to use
630bfcc09ddSBjoern A. Zeeb 	 * right after we send the time event
631bfcc09ddSBjoern A. Zeeb 	 */
632bfcc09ddSBjoern A. Zeeb 	iwl_init_notification_wait(&mvm->notif_wait, &wait_te_notif,
633bfcc09ddSBjoern A. Zeeb 				   te_notif_response,
634bfcc09ddSBjoern A. Zeeb 				   ARRAY_SIZE(te_notif_response),
635bfcc09ddSBjoern A. Zeeb 				   iwl_mvm_te_notif, te_data);
636bfcc09ddSBjoern A. Zeeb 
637bfcc09ddSBjoern A. Zeeb 	/* If TE was sent OK - wait for the notification that started */
638bfcc09ddSBjoern A. Zeeb 	if (iwl_mvm_time_event_send_add(mvm, vif, te_data, &time_cmd)) {
639bfcc09ddSBjoern A. Zeeb 		IWL_ERR(mvm, "Failed to add TE to protect session\n");
640bfcc09ddSBjoern A. Zeeb 		iwl_remove_notification(&mvm->notif_wait, &wait_te_notif);
641bfcc09ddSBjoern A. Zeeb 	} else if (iwl_wait_notification(&mvm->notif_wait, &wait_te_notif,
642bfcc09ddSBjoern A. Zeeb 					 TU_TO_JIFFIES(max_delay))) {
643bfcc09ddSBjoern A. Zeeb 		IWL_ERR(mvm, "Failed to protect session until TE\n");
644bfcc09ddSBjoern A. Zeeb 	}
645bfcc09ddSBjoern A. Zeeb }
646bfcc09ddSBjoern A. Zeeb 
647bfcc09ddSBjoern A. Zeeb static void iwl_mvm_cancel_session_protection(struct iwl_mvm *mvm,
648bfcc09ddSBjoern A. Zeeb 					      struct iwl_mvm_vif *mvmvif,
649bfcc09ddSBjoern A. Zeeb 					      u32 id)
650bfcc09ddSBjoern A. Zeeb {
651bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_session_prot_cmd cmd = {
652bfcc09ddSBjoern A. Zeeb 		.id_and_color =
653bfcc09ddSBjoern A. Zeeb 			cpu_to_le32(FW_CMD_ID_AND_COLOR(mvmvif->id,
654bfcc09ddSBjoern A. Zeeb 							mvmvif->color)),
655bfcc09ddSBjoern A. Zeeb 		.action = cpu_to_le32(FW_CTXT_ACTION_REMOVE),
656bfcc09ddSBjoern A. Zeeb 		.conf_id = cpu_to_le32(id),
657bfcc09ddSBjoern A. Zeeb 	};
658bfcc09ddSBjoern A. Zeeb 	int ret;
659bfcc09ddSBjoern A. Zeeb 
660*d9836fb4SBjoern A. Zeeb 	ret = iwl_mvm_send_cmd_pdu(mvm,
661*d9836fb4SBjoern A. Zeeb 				   WIDE_ID(MAC_CONF_GROUP, SESSION_PROTECTION_CMD),
662bfcc09ddSBjoern A. Zeeb 				   0, sizeof(cmd), &cmd);
663bfcc09ddSBjoern A. Zeeb 	if (ret)
664bfcc09ddSBjoern A. Zeeb 		IWL_ERR(mvm,
665bfcc09ddSBjoern A. Zeeb 			"Couldn't send the SESSION_PROTECTION_CMD: %d\n", ret);
666bfcc09ddSBjoern A. Zeeb }
667bfcc09ddSBjoern A. Zeeb 
668bfcc09ddSBjoern A. Zeeb static bool __iwl_mvm_remove_time_event(struct iwl_mvm *mvm,
669bfcc09ddSBjoern A. Zeeb 					struct iwl_mvm_time_event_data *te_data,
670bfcc09ddSBjoern A. Zeeb 					u32 *uid)
671bfcc09ddSBjoern A. Zeeb {
672bfcc09ddSBjoern A. Zeeb 	u32 id;
673bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_vif *mvmvif;
674bfcc09ddSBjoern A. Zeeb 	enum nl80211_iftype iftype;
675bfcc09ddSBjoern A. Zeeb 
676bfcc09ddSBjoern A. Zeeb 	if (!te_data->vif)
677bfcc09ddSBjoern A. Zeeb 		return false;
678bfcc09ddSBjoern A. Zeeb 
679bfcc09ddSBjoern A. Zeeb 	mvmvif = iwl_mvm_vif_from_mac80211(te_data->vif);
680bfcc09ddSBjoern A. Zeeb 	iftype = te_data->vif->type;
681bfcc09ddSBjoern A. Zeeb 
682bfcc09ddSBjoern A. Zeeb 	/*
683bfcc09ddSBjoern A. Zeeb 	 * It is possible that by the time we got to this point the time
684bfcc09ddSBjoern A. Zeeb 	 * event was already removed.
685bfcc09ddSBjoern A. Zeeb 	 */
686bfcc09ddSBjoern A. Zeeb 	spin_lock_bh(&mvm->time_event_lock);
687bfcc09ddSBjoern A. Zeeb 
688bfcc09ddSBjoern A. Zeeb 	/* Save time event uid before clearing its data */
689bfcc09ddSBjoern A. Zeeb 	*uid = te_data->uid;
690bfcc09ddSBjoern A. Zeeb 	id = te_data->id;
691bfcc09ddSBjoern A. Zeeb 
692bfcc09ddSBjoern A. Zeeb 	/*
693bfcc09ddSBjoern A. Zeeb 	 * The clear_data function handles time events that were already removed
694bfcc09ddSBjoern A. Zeeb 	 */
695bfcc09ddSBjoern A. Zeeb 	iwl_mvm_te_clear_data(mvm, te_data);
696bfcc09ddSBjoern A. Zeeb 	spin_unlock_bh(&mvm->time_event_lock);
697bfcc09ddSBjoern A. Zeeb 
698*d9836fb4SBjoern A. Zeeb 	/* When session protection is used, the te_data->id field
699bfcc09ddSBjoern A. Zeeb 	 * is reused to save session protection's configuration.
700*d9836fb4SBjoern A. Zeeb 	 * For AUX ROC, HOT_SPOT_CMD is used and the te_data->id field is set
701*d9836fb4SBjoern A. Zeeb 	 * to HOT_SPOT_CMD.
702bfcc09ddSBjoern A. Zeeb 	 */
703bfcc09ddSBjoern A. Zeeb 	if (fw_has_capa(&mvm->fw->ucode_capa,
704*d9836fb4SBjoern A. Zeeb 			IWL_UCODE_TLV_CAPA_SESSION_PROT_CMD) &&
705*d9836fb4SBjoern A. Zeeb 	    id != HOT_SPOT_CMD) {
706bfcc09ddSBjoern A. Zeeb 		if (mvmvif && id < SESSION_PROTECT_CONF_MAX_ID) {
707bfcc09ddSBjoern A. Zeeb 			/* Session protection is still ongoing. Cancel it */
708bfcc09ddSBjoern A. Zeeb 			iwl_mvm_cancel_session_protection(mvm, mvmvif, id);
709bfcc09ddSBjoern A. Zeeb 			if (iftype == NL80211_IFTYPE_P2P_DEVICE) {
710bfcc09ddSBjoern A. Zeeb 				iwl_mvm_p2p_roc_finished(mvm);
711bfcc09ddSBjoern A. Zeeb 			}
712bfcc09ddSBjoern A. Zeeb 		}
713bfcc09ddSBjoern A. Zeeb 		return false;
714bfcc09ddSBjoern A. Zeeb 	} else {
715bfcc09ddSBjoern A. Zeeb 		/* It is possible that by the time we try to remove it, the
716bfcc09ddSBjoern A. Zeeb 		 * time event has already ended and removed. In such a case
717bfcc09ddSBjoern A. Zeeb 		 * there is no need to send a removal command.
718bfcc09ddSBjoern A. Zeeb 		 */
719bfcc09ddSBjoern A. Zeeb 		if (id == TE_MAX) {
720bfcc09ddSBjoern A. Zeeb 			IWL_DEBUG_TE(mvm, "TE 0x%x has already ended\n", *uid);
721bfcc09ddSBjoern A. Zeeb 			return false;
722bfcc09ddSBjoern A. Zeeb 		}
723bfcc09ddSBjoern A. Zeeb 	}
724bfcc09ddSBjoern A. Zeeb 
725bfcc09ddSBjoern A. Zeeb 	return true;
726bfcc09ddSBjoern A. Zeeb }
727bfcc09ddSBjoern A. Zeeb 
728bfcc09ddSBjoern A. Zeeb /*
729bfcc09ddSBjoern A. Zeeb  * Explicit request to remove a aux roc time event. The removal of a time
730bfcc09ddSBjoern A. Zeeb  * event needs to be synchronized with the flow of a time event's end
731bfcc09ddSBjoern A. Zeeb  * notification, which also removes the time event from the op mode
732bfcc09ddSBjoern A. Zeeb  * data structures.
733bfcc09ddSBjoern A. Zeeb  */
734bfcc09ddSBjoern A. Zeeb static void iwl_mvm_remove_aux_roc_te(struct iwl_mvm *mvm,
735bfcc09ddSBjoern A. Zeeb 				      struct iwl_mvm_vif *mvmvif,
736bfcc09ddSBjoern A. Zeeb 				      struct iwl_mvm_time_event_data *te_data)
737bfcc09ddSBjoern A. Zeeb {
738bfcc09ddSBjoern A. Zeeb 	struct iwl_hs20_roc_req aux_cmd = {};
739bfcc09ddSBjoern A. Zeeb 	u16 len = sizeof(aux_cmd) - iwl_mvm_chan_info_padding(mvm);
740bfcc09ddSBjoern A. Zeeb 
741bfcc09ddSBjoern A. Zeeb 	u32 uid;
742bfcc09ddSBjoern A. Zeeb 	int ret;
743bfcc09ddSBjoern A. Zeeb 
744bfcc09ddSBjoern A. Zeeb 	if (!__iwl_mvm_remove_time_event(mvm, te_data, &uid))
745bfcc09ddSBjoern A. Zeeb 		return;
746bfcc09ddSBjoern A. Zeeb 
747bfcc09ddSBjoern A. Zeeb 	aux_cmd.event_unique_id = cpu_to_le32(uid);
748bfcc09ddSBjoern A. Zeeb 	aux_cmd.action = cpu_to_le32(FW_CTXT_ACTION_REMOVE);
749bfcc09ddSBjoern A. Zeeb 	aux_cmd.id_and_color =
750bfcc09ddSBjoern A. Zeeb 		cpu_to_le32(FW_CMD_ID_AND_COLOR(mvmvif->id, mvmvif->color));
751bfcc09ddSBjoern A. Zeeb 	IWL_DEBUG_TE(mvm, "Removing BSS AUX ROC TE 0x%x\n",
752bfcc09ddSBjoern A. Zeeb 		     le32_to_cpu(aux_cmd.event_unique_id));
753bfcc09ddSBjoern A. Zeeb 	ret = iwl_mvm_send_cmd_pdu(mvm, HOT_SPOT_CMD, 0,
754bfcc09ddSBjoern A. Zeeb 				   len, &aux_cmd);
755bfcc09ddSBjoern A. Zeeb 
756bfcc09ddSBjoern A. Zeeb 	if (WARN_ON(ret))
757bfcc09ddSBjoern A. Zeeb 		return;
758bfcc09ddSBjoern A. Zeeb }
759bfcc09ddSBjoern A. Zeeb 
760bfcc09ddSBjoern A. Zeeb /*
761bfcc09ddSBjoern A. Zeeb  * Explicit request to remove a time event. The removal of a time event needs to
762bfcc09ddSBjoern A. Zeeb  * be synchronized with the flow of a time event's end notification, which also
763bfcc09ddSBjoern A. Zeeb  * removes the time event from the op mode data structures.
764bfcc09ddSBjoern A. Zeeb  */
765bfcc09ddSBjoern A. Zeeb void iwl_mvm_remove_time_event(struct iwl_mvm *mvm,
766bfcc09ddSBjoern A. Zeeb 			       struct iwl_mvm_vif *mvmvif,
767bfcc09ddSBjoern A. Zeeb 			       struct iwl_mvm_time_event_data *te_data)
768bfcc09ddSBjoern A. Zeeb {
769bfcc09ddSBjoern A. Zeeb 	struct iwl_time_event_cmd time_cmd = {};
770bfcc09ddSBjoern A. Zeeb 	u32 uid;
771bfcc09ddSBjoern A. Zeeb 	int ret;
772bfcc09ddSBjoern A. Zeeb 
773bfcc09ddSBjoern A. Zeeb 	if (!__iwl_mvm_remove_time_event(mvm, te_data, &uid))
774bfcc09ddSBjoern A. Zeeb 		return;
775bfcc09ddSBjoern A. Zeeb 
776bfcc09ddSBjoern A. Zeeb 	/* When we remove a TE, the UID is to be set in the id field */
777bfcc09ddSBjoern A. Zeeb 	time_cmd.id = cpu_to_le32(uid);
778bfcc09ddSBjoern A. Zeeb 	time_cmd.action = cpu_to_le32(FW_CTXT_ACTION_REMOVE);
779bfcc09ddSBjoern A. Zeeb 	time_cmd.id_and_color =
780bfcc09ddSBjoern A. Zeeb 		cpu_to_le32(FW_CMD_ID_AND_COLOR(mvmvif->id, mvmvif->color));
781bfcc09ddSBjoern A. Zeeb 
782bfcc09ddSBjoern A. Zeeb 	IWL_DEBUG_TE(mvm, "Removing TE 0x%x\n", le32_to_cpu(time_cmd.id));
783bfcc09ddSBjoern A. Zeeb 	ret = iwl_mvm_send_cmd_pdu(mvm, TIME_EVENT_CMD, 0,
784bfcc09ddSBjoern A. Zeeb 				   sizeof(time_cmd), &time_cmd);
785bfcc09ddSBjoern A. Zeeb 	if (ret)
786bfcc09ddSBjoern A. Zeeb 		IWL_ERR(mvm, "Couldn't remove the time event\n");
787bfcc09ddSBjoern A. Zeeb }
788bfcc09ddSBjoern A. Zeeb 
789bfcc09ddSBjoern A. Zeeb void iwl_mvm_stop_session_protection(struct iwl_mvm *mvm,
790bfcc09ddSBjoern A. Zeeb 				     struct ieee80211_vif *vif)
791bfcc09ddSBjoern A. Zeeb {
792bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
793bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_time_event_data *te_data = &mvmvif->time_event_data;
794bfcc09ddSBjoern A. Zeeb 	u32 id;
795bfcc09ddSBjoern A. Zeeb 
796bfcc09ddSBjoern A. Zeeb 	lockdep_assert_held(&mvm->mutex);
797bfcc09ddSBjoern A. Zeeb 
798bfcc09ddSBjoern A. Zeeb 	spin_lock_bh(&mvm->time_event_lock);
799bfcc09ddSBjoern A. Zeeb 	id = te_data->id;
800bfcc09ddSBjoern A. Zeeb 	spin_unlock_bh(&mvm->time_event_lock);
801bfcc09ddSBjoern A. Zeeb 
802bfcc09ddSBjoern A. Zeeb 	if (fw_has_capa(&mvm->fw->ucode_capa,
803bfcc09ddSBjoern A. Zeeb 			IWL_UCODE_TLV_CAPA_SESSION_PROT_CMD)) {
804bfcc09ddSBjoern A. Zeeb 		if (id != SESSION_PROTECT_CONF_ASSOC) {
805bfcc09ddSBjoern A. Zeeb 			IWL_DEBUG_TE(mvm,
806bfcc09ddSBjoern A. Zeeb 				     "don't remove session protection id=%u\n",
807bfcc09ddSBjoern A. Zeeb 				     id);
808bfcc09ddSBjoern A. Zeeb 			return;
809bfcc09ddSBjoern A. Zeeb 		}
810bfcc09ddSBjoern A. Zeeb 	} else if (id != TE_BSS_STA_AGGRESSIVE_ASSOC) {
811bfcc09ddSBjoern A. Zeeb 		IWL_DEBUG_TE(mvm,
812bfcc09ddSBjoern A. Zeeb 			     "don't remove TE with id=%u (not session protection)\n",
813bfcc09ddSBjoern A. Zeeb 			     id);
814bfcc09ddSBjoern A. Zeeb 		return;
815bfcc09ddSBjoern A. Zeeb 	}
816bfcc09ddSBjoern A. Zeeb 
817bfcc09ddSBjoern A. Zeeb 	iwl_mvm_remove_time_event(mvm, mvmvif, te_data);
818bfcc09ddSBjoern A. Zeeb }
819bfcc09ddSBjoern A. Zeeb 
820bfcc09ddSBjoern A. Zeeb void iwl_mvm_rx_session_protect_notif(struct iwl_mvm *mvm,
821bfcc09ddSBjoern A. Zeeb 				      struct iwl_rx_cmd_buffer *rxb)
822bfcc09ddSBjoern A. Zeeb {
823bfcc09ddSBjoern A. Zeeb 	struct iwl_rx_packet *pkt = rxb_addr(rxb);
824bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_session_prot_notif *notif = (void *)pkt->data;
825bfcc09ddSBjoern A. Zeeb 	struct ieee80211_vif *vif;
826bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_vif *mvmvif;
827bfcc09ddSBjoern A. Zeeb 
828bfcc09ddSBjoern A. Zeeb 	rcu_read_lock();
829bfcc09ddSBjoern A. Zeeb 	vif = iwl_mvm_rcu_dereference_vif_id(mvm, le32_to_cpu(notif->mac_id),
830bfcc09ddSBjoern A. Zeeb 					     true);
831bfcc09ddSBjoern A. Zeeb 
832bfcc09ddSBjoern A. Zeeb 	if (!vif)
833bfcc09ddSBjoern A. Zeeb 		goto out_unlock;
834bfcc09ddSBjoern A. Zeeb 
835bfcc09ddSBjoern A. Zeeb 	mvmvif = iwl_mvm_vif_from_mac80211(vif);
836bfcc09ddSBjoern A. Zeeb 
837bfcc09ddSBjoern A. Zeeb 	/* The vif is not a P2P_DEVICE, maintain its time_event_data */
838bfcc09ddSBjoern A. Zeeb 	if (vif->type != NL80211_IFTYPE_P2P_DEVICE) {
839bfcc09ddSBjoern A. Zeeb 		struct iwl_mvm_time_event_data *te_data =
840bfcc09ddSBjoern A. Zeeb 			&mvmvif->time_event_data;
841bfcc09ddSBjoern A. Zeeb 
842bfcc09ddSBjoern A. Zeeb 		if (!le32_to_cpu(notif->status)) {
843bfcc09ddSBjoern A. Zeeb 			iwl_mvm_te_check_disconnect(mvm, vif,
844bfcc09ddSBjoern A. Zeeb 						    "Session protection failure");
845bfcc09ddSBjoern A. Zeeb 			spin_lock_bh(&mvm->time_event_lock);
846bfcc09ddSBjoern A. Zeeb 			iwl_mvm_te_clear_data(mvm, te_data);
847bfcc09ddSBjoern A. Zeeb 			spin_unlock_bh(&mvm->time_event_lock);
848bfcc09ddSBjoern A. Zeeb 		}
849bfcc09ddSBjoern A. Zeeb 
850bfcc09ddSBjoern A. Zeeb 		if (le32_to_cpu(notif->start)) {
851bfcc09ddSBjoern A. Zeeb 			spin_lock_bh(&mvm->time_event_lock);
852bfcc09ddSBjoern A. Zeeb 			te_data->running = le32_to_cpu(notif->start);
853bfcc09ddSBjoern A. Zeeb 			te_data->end_jiffies =
854bfcc09ddSBjoern A. Zeeb 				TU_TO_EXP_TIME(te_data->duration);
855bfcc09ddSBjoern A. Zeeb 			spin_unlock_bh(&mvm->time_event_lock);
856bfcc09ddSBjoern A. Zeeb 		} else {
857bfcc09ddSBjoern A. Zeeb 			/*
858bfcc09ddSBjoern A. Zeeb 			 * By now, we should have finished association
859bfcc09ddSBjoern A. Zeeb 			 * and know the dtim period.
860bfcc09ddSBjoern A. Zeeb 			 */
861bfcc09ddSBjoern A. Zeeb 			iwl_mvm_te_check_disconnect(mvm, vif,
862bfcc09ddSBjoern A. Zeeb 						    !vif->bss_conf.assoc ?
863bfcc09ddSBjoern A. Zeeb 						    "Not associated and the session protection is over already..." :
864bfcc09ddSBjoern A. Zeeb 						    "No beacon heard and the session protection is over already...");
865bfcc09ddSBjoern A. Zeeb 			spin_lock_bh(&mvm->time_event_lock);
866bfcc09ddSBjoern A. Zeeb 			iwl_mvm_te_clear_data(mvm, te_data);
867bfcc09ddSBjoern A. Zeeb 			spin_unlock_bh(&mvm->time_event_lock);
868bfcc09ddSBjoern A. Zeeb 		}
869bfcc09ddSBjoern A. Zeeb 
870bfcc09ddSBjoern A. Zeeb 		goto out_unlock;
871bfcc09ddSBjoern A. Zeeb 	}
872bfcc09ddSBjoern A. Zeeb 
873bfcc09ddSBjoern A. Zeeb 	if (!le32_to_cpu(notif->status) || !le32_to_cpu(notif->start)) {
874bfcc09ddSBjoern A. Zeeb 		/* End TE, notify mac80211 */
875bfcc09ddSBjoern A. Zeeb 		mvmvif->time_event_data.id = SESSION_PROTECT_CONF_MAX_ID;
876bfcc09ddSBjoern A. Zeeb 		ieee80211_remain_on_channel_expired(mvm->hw);
877bfcc09ddSBjoern A. Zeeb 		iwl_mvm_p2p_roc_finished(mvm);
878bfcc09ddSBjoern A. Zeeb 	} else if (le32_to_cpu(notif->start)) {
879bfcc09ddSBjoern A. Zeeb 		if (WARN_ON(mvmvif->time_event_data.id !=
880bfcc09ddSBjoern A. Zeeb 				le32_to_cpu(notif->conf_id)))
881bfcc09ddSBjoern A. Zeeb 			goto out_unlock;
882bfcc09ddSBjoern A. Zeeb 		set_bit(IWL_MVM_STATUS_ROC_RUNNING, &mvm->status);
883bfcc09ddSBjoern A. Zeeb 		ieee80211_ready_on_channel(mvm->hw); /* Start TE */
884bfcc09ddSBjoern A. Zeeb 	}
885bfcc09ddSBjoern A. Zeeb 
886bfcc09ddSBjoern A. Zeeb  out_unlock:
887bfcc09ddSBjoern A. Zeeb 	rcu_read_unlock();
888bfcc09ddSBjoern A. Zeeb }
889bfcc09ddSBjoern A. Zeeb 
890bfcc09ddSBjoern A. Zeeb static int
891bfcc09ddSBjoern A. Zeeb iwl_mvm_start_p2p_roc_session_protection(struct iwl_mvm *mvm,
892bfcc09ddSBjoern A. Zeeb 					 struct ieee80211_vif *vif,
893bfcc09ddSBjoern A. Zeeb 					 int duration,
894bfcc09ddSBjoern A. Zeeb 					 enum ieee80211_roc_type type)
895bfcc09ddSBjoern A. Zeeb {
896bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
897bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_session_prot_cmd cmd = {
898bfcc09ddSBjoern A. Zeeb 		.id_and_color =
899bfcc09ddSBjoern A. Zeeb 			cpu_to_le32(FW_CMD_ID_AND_COLOR(mvmvif->id,
900bfcc09ddSBjoern A. Zeeb 							mvmvif->color)),
901bfcc09ddSBjoern A. Zeeb 		.action = cpu_to_le32(FW_CTXT_ACTION_ADD),
902bfcc09ddSBjoern A. Zeeb 		.duration_tu = cpu_to_le32(MSEC_TO_TU(duration)),
903bfcc09ddSBjoern A. Zeeb 	};
904bfcc09ddSBjoern A. Zeeb 
905bfcc09ddSBjoern A. Zeeb 	lockdep_assert_held(&mvm->mutex);
906bfcc09ddSBjoern A. Zeeb 
907bfcc09ddSBjoern A. Zeeb 	/* The time_event_data.id field is reused to save session
908bfcc09ddSBjoern A. Zeeb 	 * protection's configuration.
909bfcc09ddSBjoern A. Zeeb 	 */
910bfcc09ddSBjoern A. Zeeb 	switch (type) {
911bfcc09ddSBjoern A. Zeeb 	case IEEE80211_ROC_TYPE_NORMAL:
912bfcc09ddSBjoern A. Zeeb 		mvmvif->time_event_data.id =
913bfcc09ddSBjoern A. Zeeb 			SESSION_PROTECT_CONF_P2P_DEVICE_DISCOV;
914bfcc09ddSBjoern A. Zeeb 		break;
915bfcc09ddSBjoern A. Zeeb 	case IEEE80211_ROC_TYPE_MGMT_TX:
916bfcc09ddSBjoern A. Zeeb 		mvmvif->time_event_data.id =
917bfcc09ddSBjoern A. Zeeb 			SESSION_PROTECT_CONF_P2P_GO_NEGOTIATION;
918bfcc09ddSBjoern A. Zeeb 		break;
919bfcc09ddSBjoern A. Zeeb 	default:
920bfcc09ddSBjoern A. Zeeb 		WARN_ONCE(1, "Got an invalid ROC type\n");
921bfcc09ddSBjoern A. Zeeb 		return -EINVAL;
922bfcc09ddSBjoern A. Zeeb 	}
923bfcc09ddSBjoern A. Zeeb 
924bfcc09ddSBjoern A. Zeeb 	cmd.conf_id = cpu_to_le32(mvmvif->time_event_data.id);
925*d9836fb4SBjoern A. Zeeb 	return iwl_mvm_send_cmd_pdu(mvm,
926*d9836fb4SBjoern A. Zeeb 				    WIDE_ID(MAC_CONF_GROUP, SESSION_PROTECTION_CMD),
927bfcc09ddSBjoern A. Zeeb 				    0, sizeof(cmd), &cmd);
928bfcc09ddSBjoern A. Zeeb }
929bfcc09ddSBjoern A. Zeeb 
930bfcc09ddSBjoern A. Zeeb int iwl_mvm_start_p2p_roc(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
931bfcc09ddSBjoern A. Zeeb 			  int duration, enum ieee80211_roc_type type)
932bfcc09ddSBjoern A. Zeeb {
933bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
934bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_time_event_data *te_data = &mvmvif->time_event_data;
935bfcc09ddSBjoern A. Zeeb 	struct iwl_time_event_cmd time_cmd = {};
936bfcc09ddSBjoern A. Zeeb 
937bfcc09ddSBjoern A. Zeeb 	lockdep_assert_held(&mvm->mutex);
938bfcc09ddSBjoern A. Zeeb 	if (te_data->running) {
939bfcc09ddSBjoern A. Zeeb 		IWL_WARN(mvm, "P2P_DEVICE remain on channel already running\n");
940bfcc09ddSBjoern A. Zeeb 		return -EBUSY;
941bfcc09ddSBjoern A. Zeeb 	}
942bfcc09ddSBjoern A. Zeeb 
943bfcc09ddSBjoern A. Zeeb 	if (fw_has_capa(&mvm->fw->ucode_capa,
944bfcc09ddSBjoern A. Zeeb 			IWL_UCODE_TLV_CAPA_SESSION_PROT_CMD))
945bfcc09ddSBjoern A. Zeeb 		return iwl_mvm_start_p2p_roc_session_protection(mvm, vif,
946bfcc09ddSBjoern A. Zeeb 								duration,
947bfcc09ddSBjoern A. Zeeb 								type);
948bfcc09ddSBjoern A. Zeeb 
949bfcc09ddSBjoern A. Zeeb 	time_cmd.action = cpu_to_le32(FW_CTXT_ACTION_ADD);
950bfcc09ddSBjoern A. Zeeb 	time_cmd.id_and_color =
951bfcc09ddSBjoern A. Zeeb 		cpu_to_le32(FW_CMD_ID_AND_COLOR(mvmvif->id, mvmvif->color));
952bfcc09ddSBjoern A. Zeeb 
953bfcc09ddSBjoern A. Zeeb 	switch (type) {
954bfcc09ddSBjoern A. Zeeb 	case IEEE80211_ROC_TYPE_NORMAL:
955bfcc09ddSBjoern A. Zeeb 		time_cmd.id = cpu_to_le32(IWL_MVM_ROC_TE_TYPE_NORMAL);
956bfcc09ddSBjoern A. Zeeb 		break;
957bfcc09ddSBjoern A. Zeeb 	case IEEE80211_ROC_TYPE_MGMT_TX:
958bfcc09ddSBjoern A. Zeeb 		time_cmd.id = cpu_to_le32(IWL_MVM_ROC_TE_TYPE_MGMT_TX);
959bfcc09ddSBjoern A. Zeeb 		break;
960bfcc09ddSBjoern A. Zeeb 	default:
961bfcc09ddSBjoern A. Zeeb 		WARN_ONCE(1, "Got an invalid ROC type\n");
962bfcc09ddSBjoern A. Zeeb 		return -EINVAL;
963bfcc09ddSBjoern A. Zeeb 	}
964bfcc09ddSBjoern A. Zeeb 
965bfcc09ddSBjoern A. Zeeb 	time_cmd.apply_time = cpu_to_le32(0);
966bfcc09ddSBjoern A. Zeeb 	time_cmd.interval = cpu_to_le32(1);
967bfcc09ddSBjoern A. Zeeb 
968bfcc09ddSBjoern A. Zeeb 	/*
969bfcc09ddSBjoern A. Zeeb 	 * The P2P Device TEs can have lower priority than other events
970bfcc09ddSBjoern A. Zeeb 	 * that are being scheduled by the driver/fw, and thus it might not be
971bfcc09ddSBjoern A. Zeeb 	 * scheduled. To improve the chances of it being scheduled, allow them
972bfcc09ddSBjoern A. Zeeb 	 * to be fragmented, and in addition allow them to be delayed.
973bfcc09ddSBjoern A. Zeeb 	 */
974bfcc09ddSBjoern A. Zeeb 	time_cmd.max_frags = min(MSEC_TO_TU(duration)/50, TE_V2_FRAG_ENDLESS);
975bfcc09ddSBjoern A. Zeeb 	time_cmd.max_delay = cpu_to_le32(MSEC_TO_TU(duration/2));
976bfcc09ddSBjoern A. Zeeb 	time_cmd.duration = cpu_to_le32(MSEC_TO_TU(duration));
977bfcc09ddSBjoern A. Zeeb 	time_cmd.repeat = 1;
978bfcc09ddSBjoern A. Zeeb 	time_cmd.policy = cpu_to_le16(TE_V2_NOTIF_HOST_EVENT_START |
979bfcc09ddSBjoern A. Zeeb 				      TE_V2_NOTIF_HOST_EVENT_END |
980bfcc09ddSBjoern A. Zeeb 				      TE_V2_START_IMMEDIATELY);
981bfcc09ddSBjoern A. Zeeb 
982bfcc09ddSBjoern A. Zeeb 	return iwl_mvm_time_event_send_add(mvm, vif, te_data, &time_cmd);
983bfcc09ddSBjoern A. Zeeb }
984bfcc09ddSBjoern A. Zeeb 
985bfcc09ddSBjoern A. Zeeb static struct iwl_mvm_time_event_data *iwl_mvm_get_roc_te(struct iwl_mvm *mvm)
986bfcc09ddSBjoern A. Zeeb {
987bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_time_event_data *te_data;
988bfcc09ddSBjoern A. Zeeb 
989bfcc09ddSBjoern A. Zeeb 	lockdep_assert_held(&mvm->mutex);
990bfcc09ddSBjoern A. Zeeb 
991bfcc09ddSBjoern A. Zeeb 	spin_lock_bh(&mvm->time_event_lock);
992bfcc09ddSBjoern A. Zeeb 
993bfcc09ddSBjoern A. Zeeb 	/*
994bfcc09ddSBjoern A. Zeeb 	 * Iterate over the list of time events and find the time event that is
995bfcc09ddSBjoern A. Zeeb 	 * associated with a P2P_DEVICE interface.
996bfcc09ddSBjoern A. Zeeb 	 * This assumes that a P2P_DEVICE interface can have only a single time
997bfcc09ddSBjoern A. Zeeb 	 * event at any given time and this time event coresponds to a ROC
998bfcc09ddSBjoern A. Zeeb 	 * request
999bfcc09ddSBjoern A. Zeeb 	 */
1000bfcc09ddSBjoern A. Zeeb 	list_for_each_entry(te_data, &mvm->time_event_list, list) {
1001bfcc09ddSBjoern A. Zeeb 		if (te_data->vif->type == NL80211_IFTYPE_P2P_DEVICE)
1002bfcc09ddSBjoern A. Zeeb 			goto out;
1003bfcc09ddSBjoern A. Zeeb 	}
1004bfcc09ddSBjoern A. Zeeb 
1005bfcc09ddSBjoern A. Zeeb 	/* There can only be at most one AUX ROC time event, we just use the
1006bfcc09ddSBjoern A. Zeeb 	 * list to simplify/unify code. Remove it if it exists.
1007bfcc09ddSBjoern A. Zeeb 	 */
1008bfcc09ddSBjoern A. Zeeb 	te_data = list_first_entry_or_null(&mvm->aux_roc_te_list,
1009bfcc09ddSBjoern A. Zeeb 					   struct iwl_mvm_time_event_data,
1010bfcc09ddSBjoern A. Zeeb 					   list);
1011bfcc09ddSBjoern A. Zeeb out:
1012bfcc09ddSBjoern A. Zeeb 	spin_unlock_bh(&mvm->time_event_lock);
1013bfcc09ddSBjoern A. Zeeb 	return te_data;
1014bfcc09ddSBjoern A. Zeeb }
1015bfcc09ddSBjoern A. Zeeb 
1016bfcc09ddSBjoern A. Zeeb void iwl_mvm_cleanup_roc_te(struct iwl_mvm *mvm)
1017bfcc09ddSBjoern A. Zeeb {
1018bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_time_event_data *te_data;
1019bfcc09ddSBjoern A. Zeeb 	u32 uid;
1020bfcc09ddSBjoern A. Zeeb 
1021bfcc09ddSBjoern A. Zeeb 	te_data = iwl_mvm_get_roc_te(mvm);
1022bfcc09ddSBjoern A. Zeeb 	if (te_data)
1023bfcc09ddSBjoern A. Zeeb 		__iwl_mvm_remove_time_event(mvm, te_data, &uid);
1024bfcc09ddSBjoern A. Zeeb }
1025bfcc09ddSBjoern A. Zeeb 
1026bfcc09ddSBjoern A. Zeeb void iwl_mvm_stop_roc(struct iwl_mvm *mvm, struct ieee80211_vif *vif)
1027bfcc09ddSBjoern A. Zeeb {
1028bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_vif *mvmvif;
1029bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_time_event_data *te_data;
1030bfcc09ddSBjoern A. Zeeb 
1031bfcc09ddSBjoern A. Zeeb 	if (fw_has_capa(&mvm->fw->ucode_capa,
1032bfcc09ddSBjoern A. Zeeb 			IWL_UCODE_TLV_CAPA_SESSION_PROT_CMD)) {
1033bfcc09ddSBjoern A. Zeeb 		mvmvif = iwl_mvm_vif_from_mac80211(vif);
1034bfcc09ddSBjoern A. Zeeb 
1035bfcc09ddSBjoern A. Zeeb 		if (vif->type == NL80211_IFTYPE_P2P_DEVICE) {
1036bfcc09ddSBjoern A. Zeeb 			iwl_mvm_cancel_session_protection(mvm, mvmvif,
1037bfcc09ddSBjoern A. Zeeb 							  mvmvif->time_event_data.id);
1038bfcc09ddSBjoern A. Zeeb 			iwl_mvm_p2p_roc_finished(mvm);
1039bfcc09ddSBjoern A. Zeeb 		} else {
1040bfcc09ddSBjoern A. Zeeb 			iwl_mvm_remove_aux_roc_te(mvm, mvmvif,
1041*d9836fb4SBjoern A. Zeeb 						  &mvmvif->hs_time_event_data);
1042bfcc09ddSBjoern A. Zeeb 			iwl_mvm_roc_finished(mvm);
1043bfcc09ddSBjoern A. Zeeb 		}
1044bfcc09ddSBjoern A. Zeeb 
1045bfcc09ddSBjoern A. Zeeb 		return;
1046bfcc09ddSBjoern A. Zeeb 	}
1047bfcc09ddSBjoern A. Zeeb 
1048bfcc09ddSBjoern A. Zeeb 	te_data = iwl_mvm_get_roc_te(mvm);
1049bfcc09ddSBjoern A. Zeeb 	if (!te_data) {
1050bfcc09ddSBjoern A. Zeeb 		IWL_WARN(mvm, "No remain on channel event\n");
1051bfcc09ddSBjoern A. Zeeb 		return;
1052bfcc09ddSBjoern A. Zeeb 	}
1053bfcc09ddSBjoern A. Zeeb 
1054bfcc09ddSBjoern A. Zeeb 	mvmvif = iwl_mvm_vif_from_mac80211(te_data->vif);
1055bfcc09ddSBjoern A. Zeeb 
1056bfcc09ddSBjoern A. Zeeb 	if (te_data->vif->type == NL80211_IFTYPE_P2P_DEVICE) {
1057bfcc09ddSBjoern A. Zeeb 		iwl_mvm_remove_time_event(mvm, mvmvif, te_data);
1058bfcc09ddSBjoern A. Zeeb 		iwl_mvm_p2p_roc_finished(mvm);
1059bfcc09ddSBjoern A. Zeeb 	} else {
1060bfcc09ddSBjoern A. Zeeb 		iwl_mvm_remove_aux_roc_te(mvm, mvmvif, te_data);
1061bfcc09ddSBjoern A. Zeeb 		iwl_mvm_roc_finished(mvm);
1062bfcc09ddSBjoern A. Zeeb 	}
1063bfcc09ddSBjoern A. Zeeb }
1064bfcc09ddSBjoern A. Zeeb 
1065bfcc09ddSBjoern A. Zeeb void iwl_mvm_remove_csa_period(struct iwl_mvm *mvm,
1066bfcc09ddSBjoern A. Zeeb 			       struct ieee80211_vif *vif)
1067bfcc09ddSBjoern A. Zeeb {
1068bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1069bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_time_event_data *te_data = &mvmvif->time_event_data;
1070bfcc09ddSBjoern A. Zeeb 	u32 id;
1071bfcc09ddSBjoern A. Zeeb 
1072bfcc09ddSBjoern A. Zeeb 	lockdep_assert_held(&mvm->mutex);
1073bfcc09ddSBjoern A. Zeeb 
1074bfcc09ddSBjoern A. Zeeb 	spin_lock_bh(&mvm->time_event_lock);
1075bfcc09ddSBjoern A. Zeeb 	id = te_data->id;
1076bfcc09ddSBjoern A. Zeeb 	spin_unlock_bh(&mvm->time_event_lock);
1077bfcc09ddSBjoern A. Zeeb 
1078bfcc09ddSBjoern A. Zeeb 	if (id != TE_CHANNEL_SWITCH_PERIOD)
1079bfcc09ddSBjoern A. Zeeb 		return;
1080bfcc09ddSBjoern A. Zeeb 
1081bfcc09ddSBjoern A. Zeeb 	iwl_mvm_remove_time_event(mvm, mvmvif, te_data);
1082bfcc09ddSBjoern A. Zeeb }
1083bfcc09ddSBjoern A. Zeeb 
1084bfcc09ddSBjoern A. Zeeb int iwl_mvm_schedule_csa_period(struct iwl_mvm *mvm,
1085bfcc09ddSBjoern A. Zeeb 				struct ieee80211_vif *vif,
1086bfcc09ddSBjoern A. Zeeb 				u32 duration, u32 apply_time)
1087bfcc09ddSBjoern A. Zeeb {
1088bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1089bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_time_event_data *te_data = &mvmvif->time_event_data;
1090bfcc09ddSBjoern A. Zeeb 	struct iwl_time_event_cmd time_cmd = {};
1091bfcc09ddSBjoern A. Zeeb 
1092bfcc09ddSBjoern A. Zeeb 	lockdep_assert_held(&mvm->mutex);
1093bfcc09ddSBjoern A. Zeeb 
1094bfcc09ddSBjoern A. Zeeb 	if (te_data->running) {
1095bfcc09ddSBjoern A. Zeeb 		u32 id;
1096bfcc09ddSBjoern A. Zeeb 
1097bfcc09ddSBjoern A. Zeeb 		spin_lock_bh(&mvm->time_event_lock);
1098bfcc09ddSBjoern A. Zeeb 		id = te_data->id;
1099bfcc09ddSBjoern A. Zeeb 		spin_unlock_bh(&mvm->time_event_lock);
1100bfcc09ddSBjoern A. Zeeb 
1101bfcc09ddSBjoern A. Zeeb 		if (id == TE_CHANNEL_SWITCH_PERIOD) {
1102bfcc09ddSBjoern A. Zeeb 			IWL_DEBUG_TE(mvm, "CS period is already scheduled\n");
1103bfcc09ddSBjoern A. Zeeb 			return -EBUSY;
1104bfcc09ddSBjoern A. Zeeb 		}
1105bfcc09ddSBjoern A. Zeeb 
1106bfcc09ddSBjoern A. Zeeb 		/*
1107bfcc09ddSBjoern A. Zeeb 		 * Remove the session protection time event to allow the
1108bfcc09ddSBjoern A. Zeeb 		 * channel switch. If we got here, we just heard a beacon so
1109bfcc09ddSBjoern A. Zeeb 		 * the session protection is not needed anymore anyway.
1110bfcc09ddSBjoern A. Zeeb 		 */
1111bfcc09ddSBjoern A. Zeeb 		iwl_mvm_remove_time_event(mvm, mvmvif, te_data);
1112bfcc09ddSBjoern A. Zeeb 	}
1113bfcc09ddSBjoern A. Zeeb 
1114bfcc09ddSBjoern A. Zeeb 	time_cmd.action = cpu_to_le32(FW_CTXT_ACTION_ADD);
1115bfcc09ddSBjoern A. Zeeb 	time_cmd.id_and_color =
1116bfcc09ddSBjoern A. Zeeb 		cpu_to_le32(FW_CMD_ID_AND_COLOR(mvmvif->id, mvmvif->color));
1117bfcc09ddSBjoern A. Zeeb 	time_cmd.id = cpu_to_le32(TE_CHANNEL_SWITCH_PERIOD);
1118bfcc09ddSBjoern A. Zeeb 	time_cmd.apply_time = cpu_to_le32(apply_time);
1119bfcc09ddSBjoern A. Zeeb 	time_cmd.max_frags = TE_V2_FRAG_NONE;
1120bfcc09ddSBjoern A. Zeeb 	time_cmd.duration = cpu_to_le32(duration);
1121bfcc09ddSBjoern A. Zeeb 	time_cmd.repeat = 1;
1122bfcc09ddSBjoern A. Zeeb 	time_cmd.interval = cpu_to_le32(1);
1123bfcc09ddSBjoern A. Zeeb 	time_cmd.policy = cpu_to_le16(TE_V2_NOTIF_HOST_EVENT_START |
1124bfcc09ddSBjoern A. Zeeb 				      TE_V2_ABSENCE);
1125bfcc09ddSBjoern A. Zeeb 	if (!apply_time)
1126bfcc09ddSBjoern A. Zeeb 		time_cmd.policy |= cpu_to_le16(TE_V2_START_IMMEDIATELY);
1127bfcc09ddSBjoern A. Zeeb 
1128bfcc09ddSBjoern A. Zeeb 	return iwl_mvm_time_event_send_add(mvm, vif, te_data, &time_cmd);
1129bfcc09ddSBjoern A. Zeeb }
1130bfcc09ddSBjoern A. Zeeb 
1131bfcc09ddSBjoern A. Zeeb static bool iwl_mvm_session_prot_notif(struct iwl_notif_wait_data *notif_wait,
1132bfcc09ddSBjoern A. Zeeb 				       struct iwl_rx_packet *pkt, void *data)
1133bfcc09ddSBjoern A. Zeeb {
1134bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm *mvm =
1135bfcc09ddSBjoern A. Zeeb 		container_of(notif_wait, struct iwl_mvm, notif_wait);
1136bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_session_prot_notif *resp;
1137bfcc09ddSBjoern A. Zeeb 	int resp_len = iwl_rx_packet_payload_len(pkt);
1138bfcc09ddSBjoern A. Zeeb 
1139bfcc09ddSBjoern A. Zeeb 	if (WARN_ON(pkt->hdr.cmd != SESSION_PROTECTION_NOTIF ||
1140bfcc09ddSBjoern A. Zeeb 		    pkt->hdr.group_id != MAC_CONF_GROUP))
1141bfcc09ddSBjoern A. Zeeb 		return true;
1142bfcc09ddSBjoern A. Zeeb 
1143bfcc09ddSBjoern A. Zeeb 	if (WARN_ON_ONCE(resp_len != sizeof(*resp))) {
1144bfcc09ddSBjoern A. Zeeb 		IWL_ERR(mvm, "Invalid SESSION_PROTECTION_NOTIF response\n");
1145bfcc09ddSBjoern A. Zeeb 		return true;
1146bfcc09ddSBjoern A. Zeeb 	}
1147bfcc09ddSBjoern A. Zeeb 
1148bfcc09ddSBjoern A. Zeeb 	resp = (void *)pkt->data;
1149bfcc09ddSBjoern A. Zeeb 
1150bfcc09ddSBjoern A. Zeeb 	if (!resp->status)
1151bfcc09ddSBjoern A. Zeeb 		IWL_ERR(mvm,
1152bfcc09ddSBjoern A. Zeeb 			"TIME_EVENT_NOTIFICATION received but not executed\n");
1153bfcc09ddSBjoern A. Zeeb 
1154bfcc09ddSBjoern A. Zeeb 	return true;
1155bfcc09ddSBjoern A. Zeeb }
1156bfcc09ddSBjoern A. Zeeb 
1157bfcc09ddSBjoern A. Zeeb void iwl_mvm_schedule_session_protection(struct iwl_mvm *mvm,
1158bfcc09ddSBjoern A. Zeeb 					 struct ieee80211_vif *vif,
1159bfcc09ddSBjoern A. Zeeb 					 u32 duration, u32 min_duration,
1160bfcc09ddSBjoern A. Zeeb 					 bool wait_for_notif)
1161bfcc09ddSBjoern A. Zeeb {
1162bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
1163bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_time_event_data *te_data = &mvmvif->time_event_data;
1164*d9836fb4SBjoern A. Zeeb 	const u16 notif[] = { WIDE_ID(MAC_CONF_GROUP, SESSION_PROTECTION_NOTIF) };
1165bfcc09ddSBjoern A. Zeeb 	struct iwl_notification_wait wait_notif;
1166bfcc09ddSBjoern A. Zeeb 	struct iwl_mvm_session_prot_cmd cmd = {
1167bfcc09ddSBjoern A. Zeeb 		.id_and_color =
1168bfcc09ddSBjoern A. Zeeb 			cpu_to_le32(FW_CMD_ID_AND_COLOR(mvmvif->id,
1169bfcc09ddSBjoern A. Zeeb 							mvmvif->color)),
1170bfcc09ddSBjoern A. Zeeb 		.action = cpu_to_le32(FW_CTXT_ACTION_ADD),
11718f06a2b5SBjoern A. Zeeb 		.conf_id = cpu_to_le32(SESSION_PROTECT_CONF_ASSOC),
1172bfcc09ddSBjoern A. Zeeb 		.duration_tu = cpu_to_le32(MSEC_TO_TU(duration)),
1173bfcc09ddSBjoern A. Zeeb 	};
1174bfcc09ddSBjoern A. Zeeb 
1175bfcc09ddSBjoern A. Zeeb 	lockdep_assert_held(&mvm->mutex);
1176bfcc09ddSBjoern A. Zeeb 
1177bfcc09ddSBjoern A. Zeeb 	spin_lock_bh(&mvm->time_event_lock);
1178bfcc09ddSBjoern A. Zeeb 	if (te_data->running &&
1179bfcc09ddSBjoern A. Zeeb 	    time_after(te_data->end_jiffies, TU_TO_EXP_TIME(min_duration))) {
1180bfcc09ddSBjoern A. Zeeb 		IWL_DEBUG_TE(mvm, "We have enough time in the current TE: %u\n",
1181bfcc09ddSBjoern A. Zeeb 			     jiffies_to_msecs(te_data->end_jiffies - jiffies));
1182bfcc09ddSBjoern A. Zeeb 		spin_unlock_bh(&mvm->time_event_lock);
1183bfcc09ddSBjoern A. Zeeb 
1184bfcc09ddSBjoern A. Zeeb 		return;
1185bfcc09ddSBjoern A. Zeeb 	}
1186bfcc09ddSBjoern A. Zeeb 
1187bfcc09ddSBjoern A. Zeeb 	iwl_mvm_te_clear_data(mvm, te_data);
11888f06a2b5SBjoern A. Zeeb 	/*
11898f06a2b5SBjoern A. Zeeb 	 * The time_event_data.id field is reused to save session
11908f06a2b5SBjoern A. Zeeb 	 * protection's configuration.
11918f06a2b5SBjoern A. Zeeb 	 */
11928f06a2b5SBjoern A. Zeeb 	te_data->id = le32_to_cpu(cmd.conf_id);
1193bfcc09ddSBjoern A. Zeeb 	te_data->duration = le32_to_cpu(cmd.duration_tu);
1194bfcc09ddSBjoern A. Zeeb 	te_data->vif = vif;
1195bfcc09ddSBjoern A. Zeeb 	spin_unlock_bh(&mvm->time_event_lock);
1196bfcc09ddSBjoern A. Zeeb 
1197bfcc09ddSBjoern A. Zeeb 	IWL_DEBUG_TE(mvm, "Add new session protection, duration %d TU\n",
1198bfcc09ddSBjoern A. Zeeb 		     le32_to_cpu(cmd.duration_tu));
1199bfcc09ddSBjoern A. Zeeb 
1200bfcc09ddSBjoern A. Zeeb 	if (!wait_for_notif) {
1201bfcc09ddSBjoern A. Zeeb 		if (iwl_mvm_send_cmd_pdu(mvm,
1202*d9836fb4SBjoern A. Zeeb 					 WIDE_ID(MAC_CONF_GROUP, SESSION_PROTECTION_CMD),
1203bfcc09ddSBjoern A. Zeeb 					 0, sizeof(cmd), &cmd)) {
1204bfcc09ddSBjoern A. Zeeb 			IWL_ERR(mvm,
1205bfcc09ddSBjoern A. Zeeb 				"Couldn't send the SESSION_PROTECTION_CMD\n");
1206bfcc09ddSBjoern A. Zeeb 			spin_lock_bh(&mvm->time_event_lock);
1207bfcc09ddSBjoern A. Zeeb 			iwl_mvm_te_clear_data(mvm, te_data);
1208bfcc09ddSBjoern A. Zeeb 			spin_unlock_bh(&mvm->time_event_lock);
1209bfcc09ddSBjoern A. Zeeb 		}
1210bfcc09ddSBjoern A. Zeeb 
1211bfcc09ddSBjoern A. Zeeb 		return;
1212bfcc09ddSBjoern A. Zeeb 	}
1213bfcc09ddSBjoern A. Zeeb 
1214bfcc09ddSBjoern A. Zeeb 	iwl_init_notification_wait(&mvm->notif_wait, &wait_notif,
1215bfcc09ddSBjoern A. Zeeb 				   notif, ARRAY_SIZE(notif),
1216bfcc09ddSBjoern A. Zeeb 				   iwl_mvm_session_prot_notif, NULL);
1217bfcc09ddSBjoern A. Zeeb 
1218bfcc09ddSBjoern A. Zeeb 	if (iwl_mvm_send_cmd_pdu(mvm,
1219*d9836fb4SBjoern A. Zeeb 				 WIDE_ID(MAC_CONF_GROUP, SESSION_PROTECTION_CMD),
1220bfcc09ddSBjoern A. Zeeb 				 0, sizeof(cmd), &cmd)) {
1221bfcc09ddSBjoern A. Zeeb 		IWL_ERR(mvm,
1222bfcc09ddSBjoern A. Zeeb 			"Couldn't send the SESSION_PROTECTION_CMD\n");
1223bfcc09ddSBjoern A. Zeeb 		iwl_remove_notification(&mvm->notif_wait, &wait_notif);
1224bfcc09ddSBjoern A. Zeeb 	} else if (iwl_wait_notification(&mvm->notif_wait, &wait_notif,
1225bfcc09ddSBjoern A. Zeeb 					 TU_TO_JIFFIES(100))) {
1226bfcc09ddSBjoern A. Zeeb 		IWL_ERR(mvm,
1227bfcc09ddSBjoern A. Zeeb 			"Failed to protect session until session protection\n");
1228bfcc09ddSBjoern A. Zeeb 	}
1229bfcc09ddSBjoern A. Zeeb }
1230