GNU Linux-libre 4.14.313-gnu1
[releases.git] / drivers / net / wireless / intel / iwlwifi / mvm / time-event.c
1 /******************************************************************************
2  *
3  * This file is provided under a dual BSD/GPLv2 license.  When using or
4  * redistributing this file, you may do so under either license.
5  *
6  * GPL LICENSE SUMMARY
7  *
8  * Copyright(c) 2012 - 2014 Intel Corporation. All rights reserved.
9  * Copyright(c) 2013 - 2015 Intel Mobile Communications GmbH
10  * Copyright(c) 2017 Intel Deutschland GmbH
11  * Copyright(c) 2018 Intel Corporation
12  *
13  * This program is free software; you can redistribute it and/or modify
14  * it under the terms of version 2 of the GNU General Public License as
15  * published by the Free Software Foundation.
16  *
17  * This program is distributed in the hope that it will be useful, but
18  * WITHOUT ANY WARRANTY; without even the implied warranty of
19  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
20  * General Public License for more details.
21  *
22  * The full GNU General Public License is included in this distribution
23  * in the file called COPYING.
24  *
25  * Contact Information:
26  *  Intel Linux Wireless <linuxwifi@intel.com>
27  * Intel Corporation, 5200 N.E. Elam Young Parkway, Hillsboro, OR 97124-6497
28  *
29  * BSD LICENSE
30  *
31  * Copyright(c) 2012 - 2014 Intel Corporation. All rights reserved.
32  * Copyright(c) 2013 - 2015 Intel Mobile Communications GmbH
33  * Copyright(c) 2017 Intel Deutschland GmbH
34  * Copyright(c) 2018 Intel Corporation
35  * All rights reserved.
36  *
37  * Redistribution and use in source and binary forms, with or without
38  * modification, are permitted provided that the following conditions
39  * are met:
40  *
41  *  * Redistributions of source code must retain the above copyright
42  *    notice, this list of conditions and the following disclaimer.
43  *  * Redistributions in binary form must reproduce the above copyright
44  *    notice, this list of conditions and the following disclaimer in
45  *    the documentation and/or other materials provided with the
46  *    distribution.
47  *  * Neither the name Intel Corporation nor the names of its
48  *    contributors may be used to endorse or promote products derived
49  *    from this software without specific prior written permission.
50  *
51  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
52  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
53  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
54  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
55  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
56  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
57  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
58  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
59  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
60  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
61  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
62  *
63  *****************************************************************************/
64
65 #include <linux/jiffies.h>
66 #include <net/mac80211.h>
67
68 #include "fw/notif-wait.h"
69 #include "iwl-trans.h"
70 #include "fw-api.h"
71 #include "time-event.h"
72 #include "mvm.h"
73 #include "iwl-io.h"
74 #include "iwl-prph.h"
75
76 /*
77  * For the high priority TE use a time event type that has similar priority to
78  * the FW's action scan priority.
79  */
80 #define IWL_MVM_ROC_TE_TYPE_NORMAL TE_P2P_DEVICE_DISCOVERABLE
81 #define IWL_MVM_ROC_TE_TYPE_MGMT_TX TE_P2P_CLIENT_ASSOC
82
83 void iwl_mvm_te_clear_data(struct iwl_mvm *mvm,
84                            struct iwl_mvm_time_event_data *te_data)
85 {
86         lockdep_assert_held(&mvm->time_event_lock);
87
88         if (!te_data->vif)
89                 return;
90
91         list_del(&te_data->list);
92         te_data->running = false;
93         te_data->uid = 0;
94         te_data->id = TE_MAX;
95         te_data->vif = NULL;
96 }
97
98 void iwl_mvm_roc_done_wk(struct work_struct *wk)
99 {
100         struct iwl_mvm *mvm = container_of(wk, struct iwl_mvm, roc_done_wk);
101         u32 queues = 0;
102
103         /*
104          * Clear the ROC_RUNNING /ROC_AUX_RUNNING status bit.
105          * This will cause the TX path to drop offchannel transmissions.
106          * That would also be done by mac80211, but it is racy, in particular
107          * in the case that the time event actually completed in the firmware
108          * (which is handled in iwl_mvm_te_handle_notif).
109          */
110         if (test_and_clear_bit(IWL_MVM_STATUS_ROC_RUNNING, &mvm->status)) {
111                 queues |= BIT(IWL_MVM_OFFCHANNEL_QUEUE);
112                 iwl_mvm_unref(mvm, IWL_MVM_REF_ROC);
113         }
114         if (test_and_clear_bit(IWL_MVM_STATUS_ROC_AUX_RUNNING, &mvm->status)) {
115                 queues |= BIT(mvm->aux_queue);
116                 iwl_mvm_unref(mvm, IWL_MVM_REF_ROC_AUX);
117         }
118
119         synchronize_net();
120
121         /*
122          * Flush the offchannel queue -- this is called when the time
123          * event finishes or is canceled, so that frames queued for it
124          * won't get stuck on the queue and be transmitted in the next
125          * time event.
126          * We have to send the command asynchronously since this cannot
127          * be under the mutex for locking reasons, but that's not an
128          * issue as it will have to complete before the next command is
129          * executed, and a new time event means a new command.
130          */
131         iwl_mvm_flush_sta(mvm, &mvm->aux_sta, true, CMD_ASYNC);
132
133         /* Do the same for the P2P device queue (STA) */
134         if (test_and_clear_bit(IWL_MVM_STATUS_NEED_FLUSH_P2P, &mvm->status)) {
135                 struct iwl_mvm_vif *mvmvif;
136
137                 /*
138                  * NB: access to this pointer would be racy, but the flush bit
139                  * can only be set when we had a P2P-Device VIF, and we have a
140                  * flush of this work in iwl_mvm_prepare_mac_removal() so it's
141                  * not really racy.
142                  */
143
144                 if (!WARN_ON(!mvm->p2p_device_vif)) {
145                         mvmvif = iwl_mvm_vif_from_mac80211(mvm->p2p_device_vif);
146                         iwl_mvm_flush_sta(mvm, &mvmvif->bcast_sta, true,
147                                           CMD_ASYNC);
148                 }
149         }
150 }
151
152 static void iwl_mvm_roc_finished(struct iwl_mvm *mvm)
153 {
154         /*
155          * Of course, our status bit is just as racy as mac80211, so in
156          * addition, fire off the work struct which will drop all frames
157          * from the hardware queues that made it through the race. First
158          * it will of course synchronize the TX path to make sure that
159          * any *new* TX will be rejected.
160          */
161         schedule_work(&mvm->roc_done_wk);
162 }
163
164 static void iwl_mvm_csa_noa_start(struct iwl_mvm *mvm)
165 {
166         struct ieee80211_vif *csa_vif;
167
168         rcu_read_lock();
169
170         csa_vif = rcu_dereference(mvm->csa_vif);
171         if (!csa_vif || !csa_vif->csa_active)
172                 goto out_unlock;
173
174         IWL_DEBUG_TE(mvm, "CSA NOA started\n");
175
176         /*
177          * CSA NoA is started but we still have beacons to
178          * transmit on the current channel.
179          * So we just do nothing here and the switch
180          * will be performed on the last TBTT.
181          */
182         if (!ieee80211_csa_is_complete(csa_vif)) {
183                 IWL_WARN(mvm, "CSA NOA started too early\n");
184                 goto out_unlock;
185         }
186
187         ieee80211_csa_finish(csa_vif);
188
189         rcu_read_unlock();
190
191         RCU_INIT_POINTER(mvm->csa_vif, NULL);
192
193         return;
194
195 out_unlock:
196         rcu_read_unlock();
197 }
198
199 static bool iwl_mvm_te_check_disconnect(struct iwl_mvm *mvm,
200                                         struct ieee80211_vif *vif,
201                                         const char *errmsg)
202 {
203         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
204
205         if (vif->type != NL80211_IFTYPE_STATION)
206                 return false;
207
208         if (!mvmvif->csa_bcn_pending && vif->bss_conf.assoc &&
209             vif->bss_conf.dtim_period)
210                 return false;
211         if (errmsg)
212                 IWL_ERR(mvm, "%s\n", errmsg);
213
214         iwl_mvm_connection_loss(mvm, vif, errmsg);
215         return true;
216 }
217
218 static void
219 iwl_mvm_te_handle_notify_csa(struct iwl_mvm *mvm,
220                              struct iwl_mvm_time_event_data *te_data,
221                              struct iwl_time_event_notif *notif)
222 {
223         struct ieee80211_vif *vif = te_data->vif;
224         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
225
226         if (!notif->status)
227                 IWL_DEBUG_TE(mvm, "CSA time event failed to start\n");
228
229         switch (te_data->vif->type) {
230         case NL80211_IFTYPE_AP:
231                 if (!notif->status)
232                         mvmvif->csa_failed = true;
233                 iwl_mvm_csa_noa_start(mvm);
234                 break;
235         case NL80211_IFTYPE_STATION:
236                 if (!notif->status) {
237                         iwl_mvm_connection_loss(mvm, vif,
238                                                 "CSA TE failed to start");
239                         break;
240                 }
241                 iwl_mvm_csa_client_absent(mvm, te_data->vif);
242                 ieee80211_chswitch_done(te_data->vif, true);
243                 break;
244         default:
245                 /* should never happen */
246                 WARN_ON_ONCE(1);
247                 break;
248         }
249
250         /* we don't need it anymore */
251         iwl_mvm_te_clear_data(mvm, te_data);
252 }
253
254 static void iwl_mvm_te_check_trigger(struct iwl_mvm *mvm,
255                                      struct iwl_time_event_notif *notif,
256                                      struct iwl_mvm_time_event_data *te_data)
257 {
258         struct iwl_fw_dbg_trigger_tlv *trig;
259         struct iwl_fw_dbg_trigger_time_event *te_trig;
260         int i;
261
262         if (!iwl_fw_dbg_trigger_enabled(mvm->fw, FW_DBG_TRIGGER_TIME_EVENT))
263                 return;
264
265         trig = iwl_fw_dbg_get_trigger(mvm->fw, FW_DBG_TRIGGER_TIME_EVENT);
266         te_trig = (void *)trig->data;
267
268         if (!iwl_fw_dbg_trigger_check_stop(&mvm->fwrt,
269                                            ieee80211_vif_to_wdev(te_data->vif),
270                                            trig))
271                 return;
272
273         for (i = 0; i < ARRAY_SIZE(te_trig->time_events); i++) {
274                 u32 trig_te_id = le32_to_cpu(te_trig->time_events[i].id);
275                 u32 trig_action_bitmap =
276                         le32_to_cpu(te_trig->time_events[i].action_bitmap);
277                 u32 trig_status_bitmap =
278                         le32_to_cpu(te_trig->time_events[i].status_bitmap);
279
280                 if (trig_te_id != te_data->id ||
281                     !(trig_action_bitmap & le32_to_cpu(notif->action)) ||
282                     !(trig_status_bitmap & BIT(le32_to_cpu(notif->status))))
283                         continue;
284
285                 iwl_fw_dbg_collect_trig(&mvm->fwrt, trig,
286                                         "Time event %d Action 0x%x received status: %d",
287                                         te_data->id,
288                                         le32_to_cpu(notif->action),
289                                         le32_to_cpu(notif->status));
290                 break;
291         }
292 }
293
294 /*
295  * Handles a FW notification for an event that is known to the driver.
296  *
297  * @mvm: the mvm component
298  * @te_data: the time event data
299  * @notif: the notification data corresponding the time event data.
300  */
301 static void iwl_mvm_te_handle_notif(struct iwl_mvm *mvm,
302                                     struct iwl_mvm_time_event_data *te_data,
303                                     struct iwl_time_event_notif *notif)
304 {
305         lockdep_assert_held(&mvm->time_event_lock);
306
307         IWL_DEBUG_TE(mvm, "Handle time event notif - UID = 0x%x action %d\n",
308                      le32_to_cpu(notif->unique_id),
309                      le32_to_cpu(notif->action));
310
311         iwl_mvm_te_check_trigger(mvm, notif, te_data);
312
313         /*
314          * The FW sends the start/end time event notifications even for events
315          * that it fails to schedule. This is indicated in the status field of
316          * the notification. This happens in cases that the scheduler cannot
317          * find a schedule that can handle the event (for example requesting a
318          * P2P Device discoveribility, while there are other higher priority
319          * events in the system).
320          */
321         if (!le32_to_cpu(notif->status)) {
322                 const char *msg;
323
324                 if (notif->action & cpu_to_le32(TE_V2_NOTIF_HOST_EVENT_START))
325                         msg = "Time Event start notification failure";
326                 else
327                         msg = "Time Event end notification failure";
328
329                 IWL_DEBUG_TE(mvm, "%s\n", msg);
330
331                 if (iwl_mvm_te_check_disconnect(mvm, te_data->vif, msg)) {
332                         iwl_mvm_te_clear_data(mvm, te_data);
333                         return;
334                 }
335         }
336
337         if (le32_to_cpu(notif->action) & TE_V2_NOTIF_HOST_EVENT_END) {
338                 IWL_DEBUG_TE(mvm,
339                              "TE ended - current time %lu, estimated end %lu\n",
340                              jiffies, te_data->end_jiffies);
341
342                 switch (te_data->vif->type) {
343                 case NL80211_IFTYPE_P2P_DEVICE:
344                         ieee80211_remain_on_channel_expired(mvm->hw);
345                         iwl_mvm_roc_finished(mvm);
346                         break;
347                 case NL80211_IFTYPE_STATION:
348                         /*
349                          * By now, we should have finished association
350                          * and know the dtim period.
351                          */
352                         iwl_mvm_te_check_disconnect(mvm, te_data->vif,
353                                 "No beacon heard and the time event is over already...");
354                         break;
355                 default:
356                         break;
357                 }
358
359                 iwl_mvm_te_clear_data(mvm, te_data);
360         } else if (le32_to_cpu(notif->action) & TE_V2_NOTIF_HOST_EVENT_START) {
361                 te_data->running = true;
362                 te_data->end_jiffies = TU_TO_EXP_TIME(te_data->duration);
363
364                 if (te_data->vif->type == NL80211_IFTYPE_P2P_DEVICE) {
365                         set_bit(IWL_MVM_STATUS_ROC_RUNNING, &mvm->status);
366                         iwl_mvm_ref(mvm, IWL_MVM_REF_ROC);
367                         ieee80211_ready_on_channel(mvm->hw);
368                 } else if (te_data->id == TE_CHANNEL_SWITCH_PERIOD) {
369                         iwl_mvm_te_handle_notify_csa(mvm, te_data, notif);
370                 }
371         } else {
372                 IWL_WARN(mvm, "Got TE with unknown action\n");
373         }
374 }
375
376 /*
377  * Handle A Aux ROC time event
378  */
379 static int iwl_mvm_aux_roc_te_handle_notif(struct iwl_mvm *mvm,
380                                            struct iwl_time_event_notif *notif)
381 {
382         struct iwl_mvm_time_event_data *te_data, *tmp;
383         bool aux_roc_te = false;
384
385         list_for_each_entry_safe(te_data, tmp, &mvm->aux_roc_te_list, list) {
386                 if (le32_to_cpu(notif->unique_id) == te_data->uid) {
387                         aux_roc_te = true;
388                         break;
389                 }
390         }
391         if (!aux_roc_te) /* Not a Aux ROC time event */
392                 return -EINVAL;
393
394         iwl_mvm_te_check_trigger(mvm, notif, te_data);
395
396         IWL_DEBUG_TE(mvm,
397                      "Aux ROC time event notification  - UID = 0x%x action %d (error = %d)\n",
398                      le32_to_cpu(notif->unique_id),
399                      le32_to_cpu(notif->action), le32_to_cpu(notif->status));
400
401         if (!le32_to_cpu(notif->status) ||
402             le32_to_cpu(notif->action) == TE_V2_NOTIF_HOST_EVENT_END) {
403                 /* End TE, notify mac80211 */
404                 ieee80211_remain_on_channel_expired(mvm->hw);
405                 iwl_mvm_roc_finished(mvm); /* flush aux queue */
406                 list_del(&te_data->list); /* remove from list */
407                 te_data->running = false;
408                 te_data->vif = NULL;
409                 te_data->uid = 0;
410                 te_data->id = TE_MAX;
411         } else if (le32_to_cpu(notif->action) == TE_V2_NOTIF_HOST_EVENT_START) {
412                 set_bit(IWL_MVM_STATUS_ROC_AUX_RUNNING, &mvm->status);
413                 te_data->running = true;
414                 iwl_mvm_ref(mvm, IWL_MVM_REF_ROC_AUX);
415                 ieee80211_ready_on_channel(mvm->hw); /* Start TE */
416         } else {
417                 IWL_DEBUG_TE(mvm,
418                              "ERROR: Unknown Aux ROC Time Event (action = %d)\n",
419                              le32_to_cpu(notif->action));
420                 return -EINVAL;
421         }
422
423         return 0;
424 }
425
426 /*
427  * The Rx handler for time event notifications
428  */
429 void iwl_mvm_rx_time_event_notif(struct iwl_mvm *mvm,
430                                  struct iwl_rx_cmd_buffer *rxb)
431 {
432         struct iwl_rx_packet *pkt = rxb_addr(rxb);
433         struct iwl_time_event_notif *notif = (void *)pkt->data;
434         struct iwl_mvm_time_event_data *te_data, *tmp;
435
436         IWL_DEBUG_TE(mvm, "Time event notification - UID = 0x%x action %d\n",
437                      le32_to_cpu(notif->unique_id),
438                      le32_to_cpu(notif->action));
439
440         spin_lock_bh(&mvm->time_event_lock);
441         /* This time event is triggered for Aux ROC request */
442         if (!iwl_mvm_aux_roc_te_handle_notif(mvm, notif))
443                 goto unlock;
444
445         list_for_each_entry_safe(te_data, tmp, &mvm->time_event_list, list) {
446                 if (le32_to_cpu(notif->unique_id) == te_data->uid)
447                         iwl_mvm_te_handle_notif(mvm, te_data, notif);
448         }
449 unlock:
450         spin_unlock_bh(&mvm->time_event_lock);
451 }
452
453 static bool iwl_mvm_te_notif(struct iwl_notif_wait_data *notif_wait,
454                              struct iwl_rx_packet *pkt, void *data)
455 {
456         struct iwl_mvm *mvm =
457                 container_of(notif_wait, struct iwl_mvm, notif_wait);
458         struct iwl_mvm_time_event_data *te_data = data;
459         struct iwl_time_event_notif *resp;
460         int resp_len = iwl_rx_packet_payload_len(pkt);
461
462         if (WARN_ON(pkt->hdr.cmd != TIME_EVENT_NOTIFICATION))
463                 return true;
464
465         if (WARN_ON_ONCE(resp_len != sizeof(*resp))) {
466                 IWL_ERR(mvm, "Invalid TIME_EVENT_NOTIFICATION response\n");
467                 return true;
468         }
469
470         resp = (void *)pkt->data;
471
472         /* te_data->uid is already set in the TIME_EVENT_CMD response */
473         if (le32_to_cpu(resp->unique_id) != te_data->uid)
474                 return false;
475
476         IWL_DEBUG_TE(mvm, "TIME_EVENT_NOTIFICATION response - UID = 0x%x\n",
477                      te_data->uid);
478         if (!resp->status)
479                 IWL_ERR(mvm,
480                         "TIME_EVENT_NOTIFICATION received but not executed\n");
481
482         return true;
483 }
484
485 static bool iwl_mvm_time_event_response(struct iwl_notif_wait_data *notif_wait,
486                                         struct iwl_rx_packet *pkt, void *data)
487 {
488         struct iwl_mvm *mvm =
489                 container_of(notif_wait, struct iwl_mvm, notif_wait);
490         struct iwl_mvm_time_event_data *te_data = data;
491         struct iwl_time_event_resp *resp;
492         int resp_len = iwl_rx_packet_payload_len(pkt);
493
494         if (WARN_ON(pkt->hdr.cmd != TIME_EVENT_CMD))
495                 return true;
496
497         if (WARN_ON_ONCE(resp_len != sizeof(*resp))) {
498                 IWL_ERR(mvm, "Invalid TIME_EVENT_CMD response\n");
499                 return true;
500         }
501
502         resp = (void *)pkt->data;
503
504         /* we should never get a response to another TIME_EVENT_CMD here */
505         if (WARN_ON_ONCE(le32_to_cpu(resp->id) != te_data->id))
506                 return false;
507
508         te_data->uid = le32_to_cpu(resp->unique_id);
509         IWL_DEBUG_TE(mvm, "TIME_EVENT_CMD response - UID = 0x%x\n",
510                      te_data->uid);
511         return true;
512 }
513
514 static int iwl_mvm_time_event_send_add(struct iwl_mvm *mvm,
515                                        struct ieee80211_vif *vif,
516                                        struct iwl_mvm_time_event_data *te_data,
517                                        struct iwl_time_event_cmd *te_cmd)
518 {
519         static const u16 time_event_response[] = { TIME_EVENT_CMD };
520         struct iwl_notification_wait wait_time_event;
521         int ret;
522
523         lockdep_assert_held(&mvm->mutex);
524
525         IWL_DEBUG_TE(mvm, "Add new TE, duration %d TU\n",
526                      le32_to_cpu(te_cmd->duration));
527
528         spin_lock_bh(&mvm->time_event_lock);
529         if (WARN_ON(te_data->id != TE_MAX)) {
530                 spin_unlock_bh(&mvm->time_event_lock);
531                 return -EIO;
532         }
533         te_data->vif = vif;
534         te_data->duration = le32_to_cpu(te_cmd->duration);
535         te_data->id = le32_to_cpu(te_cmd->id);
536         list_add_tail(&te_data->list, &mvm->time_event_list);
537         spin_unlock_bh(&mvm->time_event_lock);
538
539         /*
540          * Use a notification wait, which really just processes the
541          * command response and doesn't wait for anything, in order
542          * to be able to process the response and get the UID inside
543          * the RX path. Using CMD_WANT_SKB doesn't work because it
544          * stores the buffer and then wakes up this thread, by which
545          * time another notification (that the time event started)
546          * might already be processed unsuccessfully.
547          */
548         iwl_init_notification_wait(&mvm->notif_wait, &wait_time_event,
549                                    time_event_response,
550                                    ARRAY_SIZE(time_event_response),
551                                    iwl_mvm_time_event_response, te_data);
552
553         ret = iwl_mvm_send_cmd_pdu(mvm, TIME_EVENT_CMD, 0,
554                                             sizeof(*te_cmd), te_cmd);
555         if (ret) {
556                 IWL_ERR(mvm, "Couldn't send TIME_EVENT_CMD: %d\n", ret);
557                 iwl_remove_notification(&mvm->notif_wait, &wait_time_event);
558                 goto out_clear_te;
559         }
560
561         /* No need to wait for anything, so just pass 1 (0 isn't valid) */
562         ret = iwl_wait_notification(&mvm->notif_wait, &wait_time_event, 1);
563         /* should never fail */
564         WARN_ON_ONCE(ret);
565
566         if (ret) {
567  out_clear_te:
568                 spin_lock_bh(&mvm->time_event_lock);
569                 iwl_mvm_te_clear_data(mvm, te_data);
570                 spin_unlock_bh(&mvm->time_event_lock);
571         }
572         return ret;
573 }
574
575 void iwl_mvm_protect_session(struct iwl_mvm *mvm,
576                              struct ieee80211_vif *vif,
577                              u32 duration, u32 min_duration,
578                              u32 max_delay, bool wait_for_notif)
579 {
580         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
581         struct iwl_mvm_time_event_data *te_data = &mvmvif->time_event_data;
582         const u16 te_notif_response[] = { TIME_EVENT_NOTIFICATION };
583         struct iwl_notification_wait wait_te_notif;
584         struct iwl_time_event_cmd time_cmd = {};
585
586         lockdep_assert_held(&mvm->mutex);
587
588         if (te_data->running &&
589             time_after(te_data->end_jiffies, TU_TO_EXP_TIME(min_duration))) {
590                 IWL_DEBUG_TE(mvm, "We have enough time in the current TE: %u\n",
591                              jiffies_to_msecs(te_data->end_jiffies - jiffies));
592                 return;
593         }
594
595         if (te_data->running) {
596                 IWL_DEBUG_TE(mvm, "extend 0x%x: only %u ms left\n",
597                              te_data->uid,
598                              jiffies_to_msecs(te_data->end_jiffies - jiffies));
599                 /*
600                  * we don't have enough time
601                  * cancel the current TE and issue a new one
602                  * Of course it would be better to remove the old one only
603                  * when the new one is added, but we don't care if we are off
604                  * channel for a bit. All we need to do, is not to return
605                  * before we actually begin to be on the channel.
606                  */
607                 iwl_mvm_stop_session_protection(mvm, vif);
608         }
609
610         time_cmd.action = cpu_to_le32(FW_CTXT_ACTION_ADD);
611         time_cmd.id_and_color =
612                 cpu_to_le32(FW_CMD_ID_AND_COLOR(mvmvif->id, mvmvif->color));
613         time_cmd.id = cpu_to_le32(TE_BSS_STA_AGGRESSIVE_ASSOC);
614
615         time_cmd.apply_time = cpu_to_le32(0);
616
617         time_cmd.max_frags = TE_V2_FRAG_NONE;
618         time_cmd.max_delay = cpu_to_le32(max_delay);
619         /* TODO: why do we need to interval = bi if it is not periodic? */
620         time_cmd.interval = cpu_to_le32(1);
621         time_cmd.duration = cpu_to_le32(duration);
622         time_cmd.repeat = 1;
623         time_cmd.policy = cpu_to_le16(TE_V2_NOTIF_HOST_EVENT_START |
624                                       TE_V2_NOTIF_HOST_EVENT_END |
625                                       TE_V2_START_IMMEDIATELY);
626
627         if (!wait_for_notif) {
628                 iwl_mvm_time_event_send_add(mvm, vif, te_data, &time_cmd);
629                 return;
630         }
631
632         /*
633          * Create notification_wait for the TIME_EVENT_NOTIFICATION to use
634          * right after we send the time event
635          */
636         iwl_init_notification_wait(&mvm->notif_wait, &wait_te_notif,
637                                    te_notif_response,
638                                    ARRAY_SIZE(te_notif_response),
639                                    iwl_mvm_te_notif, te_data);
640
641         /* If TE was sent OK - wait for the notification that started */
642         if (iwl_mvm_time_event_send_add(mvm, vif, te_data, &time_cmd)) {
643                 IWL_ERR(mvm, "Failed to add TE to protect session\n");
644                 iwl_remove_notification(&mvm->notif_wait, &wait_te_notif);
645         } else if (iwl_wait_notification(&mvm->notif_wait, &wait_te_notif,
646                                          TU_TO_JIFFIES(max_delay))) {
647                 IWL_ERR(mvm, "Failed to protect session until TE\n");
648         }
649 }
650
651 static bool __iwl_mvm_remove_time_event(struct iwl_mvm *mvm,
652                                         struct iwl_mvm_time_event_data *te_data,
653                                         u32 *uid)
654 {
655         u32 id;
656
657         /*
658          * It is possible that by the time we got to this point the time
659          * event was already removed.
660          */
661         spin_lock_bh(&mvm->time_event_lock);
662
663         /* Save time event uid before clearing its data */
664         *uid = te_data->uid;
665         id = te_data->id;
666
667         /*
668          * The clear_data function handles time events that were already removed
669          */
670         iwl_mvm_te_clear_data(mvm, te_data);
671         spin_unlock_bh(&mvm->time_event_lock);
672
673         /*
674          * It is possible that by the time we try to remove it, the time event
675          * has already ended and removed. In such a case there is no need to
676          * send a removal command.
677          */
678         if (id == TE_MAX) {
679                 IWL_DEBUG_TE(mvm, "TE 0x%x has already ended\n", *uid);
680                 return false;
681         }
682
683         return true;
684 }
685
686 /*
687  * Explicit request to remove a aux roc time event. The removal of a time
688  * event needs to be synchronized with the flow of a time event's end
689  * notification, which also removes the time event from the op mode
690  * data structures.
691  */
692 static void iwl_mvm_remove_aux_roc_te(struct iwl_mvm *mvm,
693                                       struct iwl_mvm_vif *mvmvif,
694                                       struct iwl_mvm_time_event_data *te_data)
695 {
696         struct iwl_hs20_roc_req aux_cmd = {};
697         u32 uid;
698         int ret;
699
700         if (!__iwl_mvm_remove_time_event(mvm, te_data, &uid))
701                 return;
702
703         aux_cmd.event_unique_id = cpu_to_le32(uid);
704         aux_cmd.action = cpu_to_le32(FW_CTXT_ACTION_REMOVE);
705         aux_cmd.id_and_color =
706                 cpu_to_le32(FW_CMD_ID_AND_COLOR(mvmvif->id, mvmvif->color));
707         IWL_DEBUG_TE(mvm, "Removing BSS AUX ROC TE 0x%x\n",
708                      le32_to_cpu(aux_cmd.event_unique_id));
709         ret = iwl_mvm_send_cmd_pdu(mvm, HOT_SPOT_CMD, 0,
710                                    sizeof(aux_cmd), &aux_cmd);
711
712         if (WARN_ON(ret))
713                 return;
714 }
715
716 /*
717  * Explicit request to remove a time event. The removal of a time event needs to
718  * be synchronized with the flow of a time event's end notification, which also
719  * removes the time event from the op mode data structures.
720  */
721 void iwl_mvm_remove_time_event(struct iwl_mvm *mvm,
722                                struct iwl_mvm_vif *mvmvif,
723                                struct iwl_mvm_time_event_data *te_data)
724 {
725         struct iwl_time_event_cmd time_cmd = {};
726         u32 uid;
727         int ret;
728
729         if (!__iwl_mvm_remove_time_event(mvm, te_data, &uid))
730                 return;
731
732         /* When we remove a TE, the UID is to be set in the id field */
733         time_cmd.id = cpu_to_le32(uid);
734         time_cmd.action = cpu_to_le32(FW_CTXT_ACTION_REMOVE);
735         time_cmd.id_and_color =
736                 cpu_to_le32(FW_CMD_ID_AND_COLOR(mvmvif->id, mvmvif->color));
737
738         IWL_DEBUG_TE(mvm, "Removing TE 0x%x\n", le32_to_cpu(time_cmd.id));
739         ret = iwl_mvm_send_cmd_pdu(mvm, TIME_EVENT_CMD, 0,
740                                    sizeof(time_cmd), &time_cmd);
741         if (WARN_ON(ret))
742                 return;
743 }
744
745 void iwl_mvm_stop_session_protection(struct iwl_mvm *mvm,
746                                      struct ieee80211_vif *vif)
747 {
748         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
749         struct iwl_mvm_time_event_data *te_data = &mvmvif->time_event_data;
750         u32 id;
751
752         lockdep_assert_held(&mvm->mutex);
753
754         spin_lock_bh(&mvm->time_event_lock);
755         id = te_data->id;
756         spin_unlock_bh(&mvm->time_event_lock);
757
758         if (id != TE_BSS_STA_AGGRESSIVE_ASSOC) {
759                 IWL_DEBUG_TE(mvm,
760                              "don't remove TE with id=%u (not session protection)\n",
761                              id);
762                 return;
763         }
764
765         iwl_mvm_remove_time_event(mvm, mvmvif, te_data);
766 }
767
768 int iwl_mvm_start_p2p_roc(struct iwl_mvm *mvm, struct ieee80211_vif *vif,
769                           int duration, enum ieee80211_roc_type type)
770 {
771         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
772         struct iwl_mvm_time_event_data *te_data = &mvmvif->time_event_data;
773         struct iwl_time_event_cmd time_cmd = {};
774
775         lockdep_assert_held(&mvm->mutex);
776         if (te_data->running) {
777                 IWL_WARN(mvm, "P2P_DEVICE remain on channel already running\n");
778                 return -EBUSY;
779         }
780
781         /*
782          * Flush the done work, just in case it's still pending, so that
783          * the work it does can complete and we can accept new frames.
784          */
785         flush_work(&mvm->roc_done_wk);
786
787         time_cmd.action = cpu_to_le32(FW_CTXT_ACTION_ADD);
788         time_cmd.id_and_color =
789                 cpu_to_le32(FW_CMD_ID_AND_COLOR(mvmvif->id, mvmvif->color));
790
791         switch (type) {
792         case IEEE80211_ROC_TYPE_NORMAL:
793                 time_cmd.id = cpu_to_le32(IWL_MVM_ROC_TE_TYPE_NORMAL);
794                 break;
795         case IEEE80211_ROC_TYPE_MGMT_TX:
796                 time_cmd.id = cpu_to_le32(IWL_MVM_ROC_TE_TYPE_MGMT_TX);
797                 break;
798         default:
799                 WARN_ONCE(1, "Got an invalid ROC type\n");
800                 return -EINVAL;
801         }
802
803         time_cmd.apply_time = cpu_to_le32(0);
804         time_cmd.interval = cpu_to_le32(1);
805
806         /*
807          * The P2P Device TEs can have lower priority than other events
808          * that are being scheduled by the driver/fw, and thus it might not be
809          * scheduled. To improve the chances of it being scheduled, allow them
810          * to be fragmented, and in addition allow them to be delayed.
811          */
812         time_cmd.max_frags = min(MSEC_TO_TU(duration)/50, TE_V2_FRAG_ENDLESS);
813         time_cmd.max_delay = cpu_to_le32(MSEC_TO_TU(duration/2));
814         time_cmd.duration = cpu_to_le32(MSEC_TO_TU(duration));
815         time_cmd.repeat = 1;
816         time_cmd.policy = cpu_to_le16(TE_V2_NOTIF_HOST_EVENT_START |
817                                       TE_V2_NOTIF_HOST_EVENT_END |
818                                       TE_V2_START_IMMEDIATELY);
819
820         return iwl_mvm_time_event_send_add(mvm, vif, te_data, &time_cmd);
821 }
822
823 static struct iwl_mvm_time_event_data *iwl_mvm_get_roc_te(struct iwl_mvm *mvm)
824 {
825         struct iwl_mvm_time_event_data *te_data;
826
827         lockdep_assert_held(&mvm->mutex);
828
829         spin_lock_bh(&mvm->time_event_lock);
830
831         /*
832          * Iterate over the list of time events and find the time event that is
833          * associated with a P2P_DEVICE interface.
834          * This assumes that a P2P_DEVICE interface can have only a single time
835          * event at any given time and this time event coresponds to a ROC
836          * request
837          */
838         list_for_each_entry(te_data, &mvm->time_event_list, list) {
839                 if (te_data->vif->type == NL80211_IFTYPE_P2P_DEVICE)
840                         goto out;
841         }
842
843         /* There can only be at most one AUX ROC time event, we just use the
844          * list to simplify/unify code. Remove it if it exists.
845          */
846         te_data = list_first_entry_or_null(&mvm->aux_roc_te_list,
847                                            struct iwl_mvm_time_event_data,
848                                            list);
849 out:
850         spin_unlock_bh(&mvm->time_event_lock);
851         return te_data;
852 }
853
854 void iwl_mvm_cleanup_roc_te(struct iwl_mvm *mvm)
855 {
856         struct iwl_mvm_time_event_data *te_data;
857         u32 uid;
858
859         te_data = iwl_mvm_get_roc_te(mvm);
860         if (te_data)
861                 __iwl_mvm_remove_time_event(mvm, te_data, &uid);
862 }
863
864 void iwl_mvm_stop_roc(struct iwl_mvm *mvm)
865 {
866         struct iwl_mvm_vif *mvmvif;
867         struct iwl_mvm_time_event_data *te_data;
868
869         te_data = iwl_mvm_get_roc_te(mvm);
870         if (!te_data) {
871                 IWL_WARN(mvm, "No remain on channel event\n");
872                 return;
873         }
874
875         mvmvif = iwl_mvm_vif_from_mac80211(te_data->vif);
876
877         if (te_data->vif->type == NL80211_IFTYPE_P2P_DEVICE) {
878                 iwl_mvm_remove_time_event(mvm, mvmvif, te_data);
879                 set_bit(IWL_MVM_STATUS_NEED_FLUSH_P2P, &mvm->status);
880         } else {
881                 iwl_mvm_remove_aux_roc_te(mvm, mvmvif, te_data);
882         }
883
884         iwl_mvm_roc_finished(mvm);
885 }
886
887 int iwl_mvm_schedule_csa_period(struct iwl_mvm *mvm,
888                                 struct ieee80211_vif *vif,
889                                 u32 duration, u32 apply_time)
890 {
891         struct iwl_mvm_vif *mvmvif = iwl_mvm_vif_from_mac80211(vif);
892         struct iwl_mvm_time_event_data *te_data = &mvmvif->time_event_data;
893         struct iwl_time_event_cmd time_cmd = {};
894
895         lockdep_assert_held(&mvm->mutex);
896
897         if (te_data->running) {
898                 u32 id;
899
900                 spin_lock_bh(&mvm->time_event_lock);
901                 id = te_data->id;
902                 spin_unlock_bh(&mvm->time_event_lock);
903
904                 if (id == TE_CHANNEL_SWITCH_PERIOD) {
905                         IWL_DEBUG_TE(mvm, "CS period is already scheduled\n");
906                         return -EBUSY;
907                 }
908
909                 /*
910                  * Remove the session protection time event to allow the
911                  * channel switch. If we got here, we just heard a beacon so
912                  * the session protection is not needed anymore anyway.
913                  */
914                 iwl_mvm_remove_time_event(mvm, mvmvif, te_data);
915         }
916
917         time_cmd.action = cpu_to_le32(FW_CTXT_ACTION_ADD);
918         time_cmd.id_and_color =
919                 cpu_to_le32(FW_CMD_ID_AND_COLOR(mvmvif->id, mvmvif->color));
920         time_cmd.id = cpu_to_le32(TE_CHANNEL_SWITCH_PERIOD);
921         time_cmd.apply_time = cpu_to_le32(apply_time);
922         time_cmd.max_frags = TE_V2_FRAG_NONE;
923         time_cmd.duration = cpu_to_le32(duration);
924         time_cmd.repeat = 1;
925         time_cmd.interval = cpu_to_le32(1);
926         time_cmd.policy = cpu_to_le16(TE_V2_NOTIF_HOST_EVENT_START |
927                                       TE_V2_ABSENCE);
928         if (!apply_time)
929                 time_cmd.policy |= cpu_to_le16(TE_V2_START_IMMEDIATELY);
930
931         return iwl_mvm_time_event_send_add(mvm, vif, te_data, &time_cmd);
932 }