ath9k: fix double-free bug on beacon generate failure
[pandora-kernel.git] / drivers / net / wireless / ath / ath9k / beacon.c
1 /*
2  * Copyright (c) 2008-2011 Atheros Communications Inc.
3  *
4  * Permission to use, copy, modify, and/or distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16
17 #include <linux/dma-mapping.h>
18 #include "ath9k.h"
19
20 #define FUDGE 2
21
22 static void ath9k_reset_beacon_status(struct ath_softc *sc)
23 {
24         sc->beacon.tx_processed = false;
25         sc->beacon.tx_last = false;
26 }
27
28 /*
29  *  This function will modify certain transmit queue properties depending on
30  *  the operating mode of the station (AP or AdHoc).  Parameters are AIFS
31  *  settings and channel width min/max
32 */
33 int ath_beaconq_config(struct ath_softc *sc)
34 {
35         struct ath_hw *ah = sc->sc_ah;
36         struct ath_common *common = ath9k_hw_common(ah);
37         struct ath9k_tx_queue_info qi, qi_be;
38         struct ath_txq *txq;
39
40         ath9k_hw_get_txq_props(ah, sc->beacon.beaconq, &qi);
41         if (sc->sc_ah->opmode == NL80211_IFTYPE_AP) {
42                 /* Always burst out beacon and CAB traffic. */
43                 qi.tqi_aifs = 1;
44                 qi.tqi_cwmin = 0;
45                 qi.tqi_cwmax = 0;
46         } else {
47                 /* Adhoc mode; important thing is to use 2x cwmin. */
48                 txq = sc->tx.txq_map[WME_AC_BE];
49                 ath9k_hw_get_txq_props(ah, txq->axq_qnum, &qi_be);
50                 qi.tqi_aifs = qi_be.tqi_aifs;
51                 qi.tqi_cwmin = 4*qi_be.tqi_cwmin;
52                 qi.tqi_cwmax = qi_be.tqi_cwmax;
53         }
54
55         if (!ath9k_hw_set_txq_props(ah, sc->beacon.beaconq, &qi)) {
56                 ath_err(common,
57                         "Unable to update h/w beacon queue parameters\n");
58                 return 0;
59         } else {
60                 ath9k_hw_resettxqueue(ah, sc->beacon.beaconq);
61                 return 1;
62         }
63 }
64
65 /*
66  *  Associates the beacon frame buffer with a transmit descriptor.  Will set
67  *  up rate codes, and channel flags. Beacons are always sent out at the
68  *  lowest rate, and are not retried.
69 */
70 static void ath_beacon_setup(struct ath_softc *sc, struct ath_vif *avp,
71                              struct ath_buf *bf, int rateidx)
72 {
73         struct sk_buff *skb = bf->bf_mpdu;
74         struct ath_hw *ah = sc->sc_ah;
75         struct ath_common *common = ath9k_hw_common(ah);
76         struct ath_tx_info info;
77         struct ieee80211_supported_band *sband;
78         u8 chainmask = ah->txchainmask;
79         u8 rate = 0;
80
81         ath9k_reset_beacon_status(sc);
82
83         sband = &sc->sbands[common->hw->conf.channel->band];
84         rate = sband->bitrates[rateidx].hw_value;
85         if (sc->sc_flags & SC_OP_PREAMBLE_SHORT)
86                 rate |= sband->bitrates[rateidx].hw_value_short;
87
88         memset(&info, 0, sizeof(info));
89         info.pkt_len = skb->len + FCS_LEN;
90         info.type = ATH9K_PKT_TYPE_BEACON;
91         info.txpower = MAX_RATE_POWER;
92         info.keyix = ATH9K_TXKEYIX_INVALID;
93         info.keytype = ATH9K_KEY_TYPE_CLEAR;
94         info.flags = ATH9K_TXDESC_NOACK;
95
96         info.buf_addr[0] = bf->bf_buf_addr;
97         info.buf_len[0] = roundup(skb->len, 4);
98
99         info.is_first = true;
100         info.is_last = true;
101
102         info.qcu = sc->beacon.beaconq;
103
104         info.rates[0].Tries = 1;
105         info.rates[0].Rate = rate;
106         info.rates[0].ChSel = ath_txchainmask_reduction(sc, chainmask, rate);
107
108         ath9k_hw_set_txdesc(ah, bf->bf_desc, &info);
109 }
110
111 static void ath_tx_cabq(struct ieee80211_hw *hw, struct sk_buff *skb)
112 {
113         struct ath_softc *sc = hw->priv;
114         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
115         struct ath_tx_control txctl;
116
117         memset(&txctl, 0, sizeof(struct ath_tx_control));
118         txctl.txq = sc->beacon.cabq;
119
120         ath_dbg(common, ATH_DBG_XMIT,
121                 "transmitting CABQ packet, skb: %p\n", skb);
122
123         if (ath_tx_start(hw, skb, &txctl) != 0) {
124                 ath_dbg(common, ATH_DBG_XMIT, "CABQ TX failed\n");
125                 dev_kfree_skb_any(skb);
126         }
127 }
128
129 static struct ath_buf *ath_beacon_generate(struct ieee80211_hw *hw,
130                                            struct ieee80211_vif *vif)
131 {
132         struct ath_softc *sc = hw->priv;
133         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
134         struct ath_buf *bf;
135         struct ath_vif *avp;
136         struct sk_buff *skb;
137         struct ath_txq *cabq;
138         struct ieee80211_tx_info *info;
139         int cabq_depth;
140
141         ath9k_reset_beacon_status(sc);
142
143         avp = (void *)vif->drv_priv;
144         cabq = sc->beacon.cabq;
145
146         if ((avp->av_bcbuf == NULL) || !avp->is_bslot_active)
147                 return NULL;
148
149         /* Release the old beacon first */
150
151         bf = avp->av_bcbuf;
152         skb = bf->bf_mpdu;
153         if (skb) {
154                 dma_unmap_single(sc->dev, bf->bf_buf_addr,
155                                  skb->len, DMA_TO_DEVICE);
156                 dev_kfree_skb_any(skb);
157                 bf->bf_buf_addr = 0;
158                 bf->bf_mpdu = NULL;
159         }
160
161         /* Get a new beacon from mac80211 */
162
163         skb = ieee80211_beacon_get(hw, vif);
164         bf->bf_mpdu = skb;
165         if (skb == NULL)
166                 return NULL;
167         ((struct ieee80211_mgmt *)skb->data)->u.beacon.timestamp =
168                 avp->tsf_adjust;
169
170         info = IEEE80211_SKB_CB(skb);
171         if (info->flags & IEEE80211_TX_CTL_ASSIGN_SEQ) {
172                 /*
173                  * TODO: make sure the seq# gets assigned properly (vs. other
174                  * TX frames)
175                  */
176                 struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
177                 sc->tx.seq_no += 0x10;
178                 hdr->seq_ctrl &= cpu_to_le16(IEEE80211_SCTL_FRAG);
179                 hdr->seq_ctrl |= cpu_to_le16(sc->tx.seq_no);
180         }
181
182         bf->bf_buf_addr = dma_map_single(sc->dev, skb->data,
183                                          skb->len, DMA_TO_DEVICE);
184         if (unlikely(dma_mapping_error(sc->dev, bf->bf_buf_addr))) {
185                 dev_kfree_skb_any(skb);
186                 bf->bf_mpdu = NULL;
187                 bf->bf_buf_addr = 0;
188                 ath_err(common, "dma_mapping_error on beaconing\n");
189                 return NULL;
190         }
191
192         skb = ieee80211_get_buffered_bc(hw, vif);
193
194         /*
195          * if the CABQ traffic from previous DTIM is pending and the current
196          *  beacon is also a DTIM.
197          *  1) if there is only one vif let the cab traffic continue.
198          *  2) if there are more than one vif and we are using staggered
199          *     beacons, then drain the cabq by dropping all the frames in
200          *     the cabq so that the current vifs cab traffic can be scheduled.
201          */
202         spin_lock_bh(&cabq->axq_lock);
203         cabq_depth = cabq->axq_depth;
204         spin_unlock_bh(&cabq->axq_lock);
205
206         if (skb && cabq_depth) {
207                 if (sc->nvifs > 1) {
208                         ath_dbg(common, ATH_DBG_BEACON,
209                                 "Flushing previous cabq traffic\n");
210                         ath_draintxq(sc, cabq, false);
211                 }
212         }
213
214         ath_beacon_setup(sc, avp, bf, info->control.rates[0].idx);
215
216         while (skb) {
217                 ath_tx_cabq(hw, skb);
218                 skb = ieee80211_get_buffered_bc(hw, vif);
219         }
220
221         return bf;
222 }
223
224 int ath_beacon_alloc(struct ath_softc *sc, struct ieee80211_vif *vif)
225 {
226         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
227         struct ath_vif *avp;
228         struct ath_buf *bf;
229         struct sk_buff *skb;
230         struct ath_beacon_config *cur_conf = &sc->cur_beacon_conf;
231         __le64 tstamp;
232
233         avp = (void *)vif->drv_priv;
234
235         /* Allocate a beacon descriptor if we haven't done so. */
236         if (!avp->av_bcbuf) {
237                 /* Allocate beacon state for hostap/ibss.  We know
238                  * a buffer is available. */
239                 avp->av_bcbuf = list_first_entry(&sc->beacon.bbuf,
240                                                  struct ath_buf, list);
241                 list_del(&avp->av_bcbuf->list);
242
243                 if (ath9k_uses_beacons(vif->type)) {
244                         int slot;
245                         /*
246                          * Assign the vif to a beacon xmit slot. As
247                          * above, this cannot fail to find one.
248                          */
249                         avp->av_bslot = 0;
250                         for (slot = 0; slot < ATH_BCBUF; slot++)
251                                 if (sc->beacon.bslot[slot] == NULL) {
252                                         avp->av_bslot = slot;
253                                         avp->is_bslot_active = false;
254
255                                         /* NB: keep looking for a double slot */
256                                         if (slot == 0 || !sc->beacon.bslot[slot-1])
257                                                 break;
258                                 }
259                         BUG_ON(sc->beacon.bslot[avp->av_bslot] != NULL);
260                         sc->beacon.bslot[avp->av_bslot] = vif;
261                         sc->nbcnvifs++;
262                 }
263         }
264
265         /* release the previous beacon frame, if it already exists. */
266         bf = avp->av_bcbuf;
267         if (bf->bf_mpdu != NULL) {
268                 skb = bf->bf_mpdu;
269                 dma_unmap_single(sc->dev, bf->bf_buf_addr,
270                                  skb->len, DMA_TO_DEVICE);
271                 dev_kfree_skb_any(skb);
272                 bf->bf_mpdu = NULL;
273                 bf->bf_buf_addr = 0;
274         }
275
276         /* NB: the beacon data buffer must be 32-bit aligned. */
277         skb = ieee80211_beacon_get(sc->hw, vif);
278         if (skb == NULL)
279                 return -ENOMEM;
280
281         tstamp = ((struct ieee80211_mgmt *)skb->data)->u.beacon.timestamp;
282         sc->beacon.bc_tstamp = (u32) le64_to_cpu(tstamp);
283         /* Calculate a TSF adjustment factor required for staggered beacons. */
284         if (avp->av_bslot > 0) {
285                 u64 tsfadjust;
286                 int intval;
287
288                 intval = cur_conf->beacon_interval ? : ATH_DEFAULT_BINTVAL;
289
290                 /*
291                  * Calculate the TSF offset for this beacon slot, i.e., the
292                  * number of usecs that need to be added to the timestamp field
293                  * in Beacon and Probe Response frames. Beacon slot 0 is
294                  * processed at the correct offset, so it does not require TSF
295                  * adjustment. Other slots are adjusted to get the timestamp
296                  * close to the TBTT for the BSS.
297                  */
298                 tsfadjust = TU_TO_USEC(intval * avp->av_bslot) / ATH_BCBUF;
299                 avp->tsf_adjust = cpu_to_le64(tsfadjust);
300
301                 ath_dbg(common, ATH_DBG_BEACON,
302                         "stagger beacons, bslot %d intval %u tsfadjust %llu\n",
303                         avp->av_bslot, intval, (unsigned long long)tsfadjust);
304
305                 ((struct ieee80211_mgmt *)skb->data)->u.beacon.timestamp =
306                         avp->tsf_adjust;
307         } else
308                 avp->tsf_adjust = cpu_to_le64(0);
309
310         bf->bf_mpdu = skb;
311         bf->bf_buf_addr = dma_map_single(sc->dev, skb->data,
312                                          skb->len, DMA_TO_DEVICE);
313         if (unlikely(dma_mapping_error(sc->dev, bf->bf_buf_addr))) {
314                 dev_kfree_skb_any(skb);
315                 bf->bf_mpdu = NULL;
316                 bf->bf_buf_addr = 0;
317                 ath_err(common, "dma_mapping_error on beacon alloc\n");
318                 return -ENOMEM;
319         }
320         avp->is_bslot_active = true;
321
322         return 0;
323 }
324
325 void ath_beacon_return(struct ath_softc *sc, struct ath_vif *avp)
326 {
327         if (avp->av_bcbuf != NULL) {
328                 struct ath_buf *bf;
329
330                 avp->is_bslot_active = false;
331                 if (avp->av_bslot != -1) {
332                         sc->beacon.bslot[avp->av_bslot] = NULL;
333                         sc->nbcnvifs--;
334                         avp->av_bslot = -1;
335                 }
336
337                 bf = avp->av_bcbuf;
338                 if (bf->bf_mpdu != NULL) {
339                         struct sk_buff *skb = bf->bf_mpdu;
340                         dma_unmap_single(sc->dev, bf->bf_buf_addr,
341                                          skb->len, DMA_TO_DEVICE);
342                         dev_kfree_skb_any(skb);
343                         bf->bf_mpdu = NULL;
344                         bf->bf_buf_addr = 0;
345                 }
346                 list_add_tail(&bf->list, &sc->beacon.bbuf);
347
348                 avp->av_bcbuf = NULL;
349         }
350 }
351
352 void ath_beacon_tasklet(unsigned long data)
353 {
354         struct ath_softc *sc = (struct ath_softc *)data;
355         struct ath_beacon_config *cur_conf = &sc->cur_beacon_conf;
356         struct ath_hw *ah = sc->sc_ah;
357         struct ath_common *common = ath9k_hw_common(ah);
358         struct ath_buf *bf = NULL;
359         struct ieee80211_vif *vif;
360         struct ath_tx_status ts;
361         int slot;
362         u32 bfaddr, bc = 0;
363
364         /*
365          * Check if the previous beacon has gone out.  If
366          * not don't try to post another, skip this period
367          * and wait for the next.  Missed beacons indicate
368          * a problem and should not occur.  If we miss too
369          * many consecutive beacons reset the device.
370          */
371         if (ath9k_hw_numtxpending(ah, sc->beacon.beaconq) != 0) {
372                 sc->beacon.bmisscnt++;
373
374                 if (sc->beacon.bmisscnt < BSTUCK_THRESH * sc->nbcnvifs) {
375                         ath_dbg(common, ATH_DBG_BSTUCK,
376                                 "missed %u consecutive beacons\n",
377                                 sc->beacon.bmisscnt);
378                         ath9k_hw_stop_dma_queue(ah, sc->beacon.beaconq);
379                         if (sc->beacon.bmisscnt > 3)
380                                 ath9k_hw_bstuck_nfcal(ah);
381                 } else if (sc->beacon.bmisscnt >= BSTUCK_THRESH) {
382                         ath_dbg(common, ATH_DBG_BSTUCK,
383                                 "beacon is officially stuck\n");
384                         sc->sc_flags |= SC_OP_TSF_RESET;
385                         ieee80211_queue_work(sc->hw, &sc->hw_reset_work);
386                 }
387
388                 return;
389         }
390
391         /*
392          * Generate beacon frames. we are sending frames
393          * staggered so calculate the slot for this frame based
394          * on the tsf to safeguard against missing an swba.
395          */
396
397
398         if (ah->opmode == NL80211_IFTYPE_AP) {
399                 u16 intval;
400                 u32 tsftu;
401                 u64 tsf;
402
403                 intval = cur_conf->beacon_interval ? : ATH_DEFAULT_BINTVAL;
404                 tsf = ath9k_hw_gettsf64(ah);
405                 tsf += TU_TO_USEC(ah->config.sw_beacon_response_time);
406                 tsftu = TSF_TO_TU((tsf * ATH_BCBUF) >>32, tsf * ATH_BCBUF);
407                 slot = (tsftu % (intval * ATH_BCBUF)) / intval;
408                 vif = sc->beacon.bslot[slot];
409
410                 ath_dbg(common, ATH_DBG_BEACON,
411                         "slot %d [tsf %llu tsftu %u intval %u] vif %p\n",
412                         slot, tsf, tsftu / ATH_BCBUF, intval, vif);
413         } else {
414                 slot = 0;
415                 vif = sc->beacon.bslot[slot];
416         }
417
418
419         bfaddr = 0;
420         if (vif) {
421                 bf = ath_beacon_generate(sc->hw, vif);
422                 if (bf != NULL) {
423                         bfaddr = bf->bf_daddr;
424                         bc = 1;
425                 }
426
427                 if (sc->beacon.bmisscnt != 0) {
428                         ath_dbg(common, ATH_DBG_BSTUCK,
429                                 "resume beacon xmit after %u misses\n",
430                                 sc->beacon.bmisscnt);
431                         sc->beacon.bmisscnt = 0;
432                 }
433         }
434
435         /*
436          * Handle slot time change when a non-ERP station joins/leaves
437          * an 11g network.  The 802.11 layer notifies us via callback,
438          * we mark updateslot, then wait one beacon before effecting
439          * the change.  This gives associated stations at least one
440          * beacon interval to note the state change.
441          *
442          * NB: The slot time change state machine is clocked according
443          *     to whether we are bursting or staggering beacons.  We
444          *     recognize the request to update and record the current
445          *     slot then don't transition until that slot is reached
446          *     again.  If we miss a beacon for that slot then we'll be
447          *     slow to transition but we'll be sure at least one beacon
448          *     interval has passed.  When bursting slot is always left
449          *     set to ATH_BCBUF so this check is a noop.
450          */
451         if (sc->beacon.updateslot == UPDATE) {
452                 sc->beacon.updateslot = COMMIT; /* commit next beacon */
453                 sc->beacon.slotupdate = slot;
454         } else if (sc->beacon.updateslot == COMMIT && sc->beacon.slotupdate == slot) {
455                 ah->slottime = sc->beacon.slottime;
456                 ath9k_hw_init_global_settings(ah);
457                 sc->beacon.updateslot = OK;
458         }
459         if (bfaddr != 0) {
460                 /* NB: cabq traffic should already be queued and primed */
461                 ath9k_hw_puttxbuf(ah, sc->beacon.beaconq, bfaddr);
462                 ath9k_hw_txstart(ah, sc->beacon.beaconq);
463
464                 sc->beacon.ast_be_xmit += bc;     /* XXX per-vif? */
465                 if (ah->caps.hw_caps & ATH9K_HW_CAP_EDMA) {
466                         spin_lock_bh(&sc->sc_pcu_lock);
467                         ath9k_hw_txprocdesc(ah, bf->bf_desc, (void *)&ts);
468                         spin_unlock_bh(&sc->sc_pcu_lock);
469                 }
470         }
471 }
472
473 static void ath9k_beacon_init(struct ath_softc *sc,
474                               u32 next_beacon,
475                               u32 beacon_period)
476 {
477         if (sc->sc_flags & SC_OP_TSF_RESET) {
478                 ath9k_ps_wakeup(sc);
479                 ath9k_hw_reset_tsf(sc->sc_ah);
480         }
481
482         ath9k_hw_beaconinit(sc->sc_ah, next_beacon, beacon_period);
483
484         if (sc->sc_flags & SC_OP_TSF_RESET) {
485                 ath9k_ps_restore(sc);
486                 sc->sc_flags &= ~SC_OP_TSF_RESET;
487         }
488 }
489
490 /*
491  * For multi-bss ap support beacons are either staggered evenly over N slots or
492  * burst together.  For the former arrange for the SWBA to be delivered for each
493  * slot. Slots that are not occupied will generate nothing.
494  */
495 static void ath_beacon_config_ap(struct ath_softc *sc,
496                                  struct ath_beacon_config *conf)
497 {
498         struct ath_hw *ah = sc->sc_ah;
499         u32 nexttbtt, intval;
500
501         /* NB: the beacon interval is kept internally in TU's */
502         intval = TU_TO_USEC(conf->beacon_interval);
503         intval /= ATH_BCBUF;    /* for staggered beacons */
504         nexttbtt = intval;
505
506         /*
507          * In AP mode we enable the beacon timers and SWBA interrupts to
508          * prepare beacon frames.
509          */
510         ah->imask |= ATH9K_INT_SWBA;
511         ath_beaconq_config(sc);
512
513         /* Set the computed AP beacon timers */
514
515         ath9k_hw_disable_interrupts(ah);
516         sc->sc_flags |= SC_OP_TSF_RESET;
517         ath9k_beacon_init(sc, nexttbtt, intval);
518         sc->beacon.bmisscnt = 0;
519         ath9k_hw_set_interrupts(ah);
520         ath9k_hw_enable_interrupts(ah);
521 }
522
523 /*
524  * This sets up the beacon timers according to the timestamp of the last
525  * received beacon and the current TSF, configures PCF and DTIM
526  * handling, programs the sleep registers so the hardware will wakeup in
527  * time to receive beacons, and configures the beacon miss handling so
528  * we'll receive a BMISS interrupt when we stop seeing beacons from the AP
529  * we've associated with.
530  */
531 static void ath_beacon_config_sta(struct ath_softc *sc,
532                                   struct ath_beacon_config *conf)
533 {
534         struct ath_hw *ah = sc->sc_ah;
535         struct ath_common *common = ath9k_hw_common(ah);
536         struct ath9k_beacon_state bs;
537         int dtimperiod, dtimcount, sleepduration;
538         int cfpperiod, cfpcount;
539         u32 nexttbtt = 0, intval, tsftu;
540         u64 tsf;
541         int num_beacons, offset, dtim_dec_count, cfp_dec_count;
542
543         /* No need to configure beacon if we are not associated */
544         if (!common->curaid) {
545                 ath_dbg(common, ATH_DBG_BEACON,
546                         "STA is not yet associated..skipping beacon config\n");
547                 return;
548         }
549
550         memset(&bs, 0, sizeof(bs));
551         intval = conf->beacon_interval;
552
553         /*
554          * Setup dtim and cfp parameters according to
555          * last beacon we received (which may be none).
556          */
557         dtimperiod = conf->dtim_period;
558         dtimcount = conf->dtim_count;
559         if (dtimcount >= dtimperiod)    /* NB: sanity check */
560                 dtimcount = 0;
561         cfpperiod = 1;                  /* NB: no PCF support yet */
562         cfpcount = 0;
563
564         sleepduration = conf->listen_interval * intval;
565
566         /*
567          * Pull nexttbtt forward to reflect the current
568          * TSF and calculate dtim+cfp state for the result.
569          */
570         tsf = ath9k_hw_gettsf64(ah);
571         tsftu = TSF_TO_TU(tsf>>32, tsf) + FUDGE;
572
573         num_beacons = tsftu / intval + 1;
574         offset = tsftu % intval;
575         nexttbtt = tsftu - offset;
576         if (offset)
577                 nexttbtt += intval;
578
579         /* DTIM Beacon every dtimperiod Beacon */
580         dtim_dec_count = num_beacons % dtimperiod;
581         /* CFP every cfpperiod DTIM Beacon */
582         cfp_dec_count = (num_beacons / dtimperiod) % cfpperiod;
583         if (dtim_dec_count)
584                 cfp_dec_count++;
585
586         dtimcount -= dtim_dec_count;
587         if (dtimcount < 0)
588                 dtimcount += dtimperiod;
589
590         cfpcount -= cfp_dec_count;
591         if (cfpcount < 0)
592                 cfpcount += cfpperiod;
593
594         bs.bs_intval = intval;
595         bs.bs_nexttbtt = nexttbtt;
596         bs.bs_dtimperiod = dtimperiod*intval;
597         bs.bs_nextdtim = bs.bs_nexttbtt + dtimcount*intval;
598         bs.bs_cfpperiod = cfpperiod*bs.bs_dtimperiod;
599         bs.bs_cfpnext = bs.bs_nextdtim + cfpcount*bs.bs_dtimperiod;
600         bs.bs_cfpmaxduration = 0;
601
602         /*
603          * Calculate the number of consecutive beacons to miss* before taking
604          * a BMISS interrupt. The configuration is specified in TU so we only
605          * need calculate based on the beacon interval.  Note that we clamp the
606          * result to at most 15 beacons.
607          */
608         if (sleepduration > intval) {
609                 bs.bs_bmissthreshold = conf->listen_interval *
610                         ATH_DEFAULT_BMISS_LIMIT / 2;
611         } else {
612                 bs.bs_bmissthreshold = DIV_ROUND_UP(conf->bmiss_timeout, intval);
613                 if (bs.bs_bmissthreshold > 15)
614                         bs.bs_bmissthreshold = 15;
615                 else if (bs.bs_bmissthreshold <= 0)
616                         bs.bs_bmissthreshold = 1;
617         }
618
619         /*
620          * Calculate sleep duration. The configuration is given in ms.
621          * We ensure a multiple of the beacon period is used. Also, if the sleep
622          * duration is greater than the DTIM period then it makes senses
623          * to make it a multiple of that.
624          *
625          * XXX fixed at 100ms
626          */
627
628         bs.bs_sleepduration = roundup(IEEE80211_MS_TO_TU(100), sleepduration);
629         if (bs.bs_sleepduration > bs.bs_dtimperiod)
630                 bs.bs_sleepduration = bs.bs_dtimperiod;
631
632         /* TSF out of range threshold fixed at 1 second */
633         bs.bs_tsfoor_threshold = ATH9K_TSFOOR_THRESHOLD;
634
635         ath_dbg(common, ATH_DBG_BEACON, "tsf: %llu tsftu: %u\n", tsf, tsftu);
636         ath_dbg(common, ATH_DBG_BEACON,
637                 "bmiss: %u sleep: %u cfp-period: %u maxdur: %u next: %u\n",
638                 bs.bs_bmissthreshold, bs.bs_sleepduration,
639                 bs.bs_cfpperiod, bs.bs_cfpmaxduration, bs.bs_cfpnext);
640
641         /* Set the computed STA beacon timers */
642
643         ath9k_hw_disable_interrupts(ah);
644         ath9k_hw_set_sta_beacon_timers(ah, &bs);
645         ah->imask |= ATH9K_INT_BMISS;
646
647         ath9k_hw_set_interrupts(ah);
648         ath9k_hw_enable_interrupts(ah);
649 }
650
651 static void ath_beacon_config_adhoc(struct ath_softc *sc,
652                                     struct ath_beacon_config *conf)
653 {
654         struct ath_hw *ah = sc->sc_ah;
655         struct ath_common *common = ath9k_hw_common(ah);
656         u32 tsf, intval, nexttbtt;
657
658         ath9k_reset_beacon_status(sc);
659
660         intval = TU_TO_USEC(conf->beacon_interval);
661         tsf = roundup(ath9k_hw_gettsf32(ah) + TU_TO_USEC(FUDGE), intval);
662         nexttbtt = tsf + intval;
663
664         ath_dbg(common, ATH_DBG_BEACON,
665                 "IBSS nexttbtt %u intval %u (%u)\n",
666                 nexttbtt, intval, conf->beacon_interval);
667
668         /*
669          * In IBSS mode enable the beacon timers but only enable SWBA interrupts
670          * if we need to manually prepare beacon frames.  Otherwise we use a
671          * self-linked tx descriptor and let the hardware deal with things.
672          */
673         ah->imask |= ATH9K_INT_SWBA;
674
675         ath_beaconq_config(sc);
676
677         /* Set the computed ADHOC beacon timers */
678
679         ath9k_hw_disable_interrupts(ah);
680         ath9k_beacon_init(sc, nexttbtt, intval);
681         sc->beacon.bmisscnt = 0;
682
683         ath9k_hw_set_interrupts(ah);
684         ath9k_hw_enable_interrupts(ah);
685 }
686
687 static bool ath9k_allow_beacon_config(struct ath_softc *sc,
688                                       struct ieee80211_vif *vif)
689 {
690         struct ath_beacon_config *cur_conf = &sc->cur_beacon_conf;
691         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
692         struct ieee80211_bss_conf *bss_conf = &vif->bss_conf;
693         struct ath_vif *avp = (void *)vif->drv_priv;
694
695         /*
696          * Can not have different beacon interval on multiple
697          * AP interface case
698          */
699         if ((sc->sc_ah->opmode == NL80211_IFTYPE_AP) &&
700             (sc->nbcnvifs > 1) &&
701             (vif->type == NL80211_IFTYPE_AP) &&
702             (cur_conf->beacon_interval != bss_conf->beacon_int)) {
703                 ath_dbg(common, ATH_DBG_CONFIG,
704                         "Changing beacon interval of multiple \
705                         AP interfaces !\n");
706                 return false;
707         }
708         /*
709          * Can not configure station vif's beacon config
710          * while on AP opmode
711          */
712         if ((sc->sc_ah->opmode == NL80211_IFTYPE_AP) &&
713             (vif->type != NL80211_IFTYPE_AP)) {
714                 ath_dbg(common, ATH_DBG_CONFIG,
715                         "STA vif's beacon not allowed on AP mode\n");
716                 return false;
717         }
718         /*
719          * Do not allow beacon config if HW was already configured
720          * with another STA vif
721          */
722         if ((sc->sc_ah->opmode == NL80211_IFTYPE_STATION) &&
723             (vif->type == NL80211_IFTYPE_STATION) &&
724             (sc->sc_flags & SC_OP_BEACONS) &&
725             !avp->primary_sta_vif) {
726                 ath_dbg(common, ATH_DBG_CONFIG,
727                         "Beacon already configured for a station interface\n");
728                 return false;
729         }
730         return true;
731 }
732
733 void ath_beacon_config(struct ath_softc *sc, struct ieee80211_vif *vif)
734 {
735         struct ath_beacon_config *cur_conf = &sc->cur_beacon_conf;
736         struct ieee80211_bss_conf *bss_conf = &vif->bss_conf;
737
738         if (!ath9k_allow_beacon_config(sc, vif))
739                 return;
740
741         /* Setup the beacon configuration parameters */
742         cur_conf->beacon_interval = bss_conf->beacon_int;
743         cur_conf->dtim_period = bss_conf->dtim_period;
744         cur_conf->listen_interval = 1;
745         cur_conf->dtim_count = 1;
746         cur_conf->bmiss_timeout =
747                 ATH_DEFAULT_BMISS_LIMIT * cur_conf->beacon_interval;
748
749         /*
750          * It looks like mac80211 may end up using beacon interval of zero in
751          * some cases (at least for mesh point). Avoid getting into an
752          * infinite loop by using a bit safer value instead. To be safe,
753          * do sanity check on beacon interval for all operating modes.
754          */
755         if (cur_conf->beacon_interval == 0)
756                 cur_conf->beacon_interval = 100;
757
758         /*
759          * We don't parse dtim period from mac80211 during the driver
760          * initialization as it breaks association with hidden-ssid
761          * AP and it causes latency in roaming
762          */
763         if (cur_conf->dtim_period == 0)
764                 cur_conf->dtim_period = 1;
765
766         ath_set_beacon(sc);
767 }
768
769 static bool ath_has_valid_bslot(struct ath_softc *sc)
770 {
771         struct ath_vif *avp;
772         int slot;
773         bool found = false;
774
775         for (slot = 0; slot < ATH_BCBUF; slot++) {
776                 if (sc->beacon.bslot[slot]) {
777                         avp = (void *)sc->beacon.bslot[slot]->drv_priv;
778                         if (avp->is_bslot_active) {
779                                 found = true;
780                                 break;
781                         }
782                 }
783         }
784         return found;
785 }
786
787
788 void ath_set_beacon(struct ath_softc *sc)
789 {
790         struct ath_common *common = ath9k_hw_common(sc->sc_ah);
791         struct ath_beacon_config *cur_conf = &sc->cur_beacon_conf;
792
793         switch (sc->sc_ah->opmode) {
794         case NL80211_IFTYPE_AP:
795                 if (ath_has_valid_bslot(sc))
796                         ath_beacon_config_ap(sc, cur_conf);
797                 break;
798         case NL80211_IFTYPE_ADHOC:
799         case NL80211_IFTYPE_MESH_POINT:
800                 ath_beacon_config_adhoc(sc, cur_conf);
801                 break;
802         case NL80211_IFTYPE_STATION:
803                 ath_beacon_config_sta(sc, cur_conf);
804                 break;
805         default:
806                 ath_dbg(common, ATH_DBG_CONFIG,
807                         "Unsupported beaconing mode\n");
808                 return;
809         }
810
811         sc->sc_flags |= SC_OP_BEACONS;
812 }
813
814 void ath9k_set_beaconing_status(struct ath_softc *sc, bool status)
815 {
816         struct ath_hw *ah = sc->sc_ah;
817
818         if (!ath_has_valid_bslot(sc))
819                 return;
820
821         ath9k_ps_wakeup(sc);
822         if (status) {
823                 /* Re-enable beaconing */
824                 ah->imask |= ATH9K_INT_SWBA;
825                 ath9k_hw_set_interrupts(ah);
826         } else {
827                 /* Disable SWBA interrupt */
828                 ah->imask &= ~ATH9K_INT_SWBA;
829                 ath9k_hw_set_interrupts(ah);
830                 tasklet_kill(&sc->bcon_tasklet);
831                 ath9k_hw_stop_dma_queue(ah, sc->beacon.beaconq);
832         }
833         ath9k_ps_restore(sc);
834 }