ath9k: Virtual wiphy pause/unpause functionality
[safe/jmp/linux-2.6] / drivers / net / wireless / ath9k / virtual.c
1 /*
2  * Copyright (c) 2008-2009 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 "ath9k.h"
18
19 struct ath9k_vif_iter_data {
20         int count;
21         u8 *addr;
22 };
23
24 static void ath9k_vif_iter(void *data, u8 *mac, struct ieee80211_vif *vif)
25 {
26         struct ath9k_vif_iter_data *iter_data = data;
27         u8 *nbuf;
28
29         nbuf = krealloc(iter_data->addr, (iter_data->count + 1) * ETH_ALEN,
30                         GFP_ATOMIC);
31         if (nbuf == NULL)
32                 return;
33
34         memcpy(nbuf + iter_data->count * ETH_ALEN, mac, ETH_ALEN);
35         iter_data->addr = nbuf;
36         iter_data->count++;
37 }
38
39 void ath9k_set_bssid_mask(struct ieee80211_hw *hw)
40 {
41         struct ath_wiphy *aphy = hw->priv;
42         struct ath_softc *sc = aphy->sc;
43         struct ath9k_vif_iter_data iter_data;
44         int i, j;
45         u8 mask[ETH_ALEN];
46
47         /*
48          * Add primary MAC address even if it is not in active use since it
49          * will be configured to the hardware as the starting point and the
50          * BSSID mask will need to be changed if another address is active.
51          */
52         iter_data.addr = kmalloc(ETH_ALEN, GFP_ATOMIC);
53         if (iter_data.addr) {
54                 memcpy(iter_data.addr, sc->sc_ah->macaddr, ETH_ALEN);
55                 iter_data.count = 1;
56         } else
57                 iter_data.count = 0;
58
59         /* Get list of all active MAC addresses */
60         spin_lock_bh(&sc->wiphy_lock);
61         ieee80211_iterate_active_interfaces_atomic(sc->hw, ath9k_vif_iter,
62                                                    &iter_data);
63         for (i = 0; i < sc->num_sec_wiphy; i++) {
64                 if (sc->sec_wiphy[i] == NULL)
65                         continue;
66                 ieee80211_iterate_active_interfaces_atomic(
67                         sc->sec_wiphy[i]->hw, ath9k_vif_iter, &iter_data);
68         }
69         spin_unlock_bh(&sc->wiphy_lock);
70
71         /* Generate an address mask to cover all active addresses */
72         memset(mask, 0, ETH_ALEN);
73         for (i = 0; i < iter_data.count; i++) {
74                 u8 *a1 = iter_data.addr + i * ETH_ALEN;
75                 for (j = i + 1; j < iter_data.count; j++) {
76                         u8 *a2 = iter_data.addr + j * ETH_ALEN;
77                         mask[0] |= a1[0] ^ a2[0];
78                         mask[1] |= a1[1] ^ a2[1];
79                         mask[2] |= a1[2] ^ a2[2];
80                         mask[3] |= a1[3] ^ a2[3];
81                         mask[4] |= a1[4] ^ a2[4];
82                         mask[5] |= a1[5] ^ a2[5];
83                 }
84         }
85
86         kfree(iter_data.addr);
87
88         /* Invert the mask and configure hardware */
89         sc->bssidmask[0] = ~mask[0];
90         sc->bssidmask[1] = ~mask[1];
91         sc->bssidmask[2] = ~mask[2];
92         sc->bssidmask[3] = ~mask[3];
93         sc->bssidmask[4] = ~mask[4];
94         sc->bssidmask[5] = ~mask[5];
95
96         ath9k_hw_setbssidmask(sc);
97 }
98
99 int ath9k_wiphy_add(struct ath_softc *sc)
100 {
101         int i, error;
102         struct ath_wiphy *aphy;
103         struct ieee80211_hw *hw;
104         u8 addr[ETH_ALEN];
105
106         hw = ieee80211_alloc_hw(sizeof(struct ath_wiphy), &ath9k_ops);
107         if (hw == NULL)
108                 return -ENOMEM;
109
110         spin_lock_bh(&sc->wiphy_lock);
111         for (i = 0; i < sc->num_sec_wiphy; i++) {
112                 if (sc->sec_wiphy[i] == NULL)
113                         break;
114         }
115
116         if (i == sc->num_sec_wiphy) {
117                 /* No empty slot available; increase array length */
118                 struct ath_wiphy **n;
119                 n = krealloc(sc->sec_wiphy,
120                              (sc->num_sec_wiphy + 1) *
121                              sizeof(struct ath_wiphy *),
122                              GFP_ATOMIC);
123                 if (n == NULL) {
124                         spin_unlock_bh(&sc->wiphy_lock);
125                         ieee80211_free_hw(hw);
126                         return -ENOMEM;
127                 }
128                 n[i] = NULL;
129                 sc->sec_wiphy = n;
130                 sc->num_sec_wiphy++;
131         }
132
133         SET_IEEE80211_DEV(hw, sc->dev);
134
135         aphy = hw->priv;
136         aphy->sc = sc;
137         aphy->hw = hw;
138         sc->sec_wiphy[i] = aphy;
139         spin_unlock_bh(&sc->wiphy_lock);
140
141         memcpy(addr, sc->sc_ah->macaddr, ETH_ALEN);
142         addr[0] |= 0x02; /* Locally managed address */
143         /*
144          * XOR virtual wiphy index into the least significant bits to generate
145          * a different MAC address for each virtual wiphy.
146          */
147         addr[5] ^= i & 0xff;
148         addr[4] ^= (i & 0xff00) >> 8;
149         addr[3] ^= (i & 0xff0000) >> 16;
150
151         SET_IEEE80211_PERM_ADDR(hw, addr);
152
153         ath_set_hw_capab(sc, hw);
154
155         error = ieee80211_register_hw(hw);
156
157         return error;
158 }
159
160 int ath9k_wiphy_del(struct ath_wiphy *aphy)
161 {
162         struct ath_softc *sc = aphy->sc;
163         int i;
164
165         spin_lock_bh(&sc->wiphy_lock);
166         for (i = 0; i < sc->num_sec_wiphy; i++) {
167                 if (aphy == sc->sec_wiphy[i]) {
168                         sc->sec_wiphy[i] = NULL;
169                         spin_unlock_bh(&sc->wiphy_lock);
170                         ieee80211_unregister_hw(aphy->hw);
171                         ieee80211_free_hw(aphy->hw);
172                         return 0;
173                 }
174         }
175         spin_unlock_bh(&sc->wiphy_lock);
176         return -ENOENT;
177 }
178
179 static int ath9k_send_nullfunc(struct ath_wiphy *aphy,
180                                struct ieee80211_vif *vif, const u8 *bssid,
181                                int ps)
182 {
183         struct ath_softc *sc = aphy->sc;
184         struct ath_tx_control txctl;
185         struct sk_buff *skb;
186         struct ieee80211_hdr *hdr;
187         __le16 fc;
188         struct ieee80211_tx_info *info;
189
190         skb = dev_alloc_skb(24);
191         if (skb == NULL)
192                 return -ENOMEM;
193         hdr = (struct ieee80211_hdr *) skb_put(skb, 24);
194         memset(hdr, 0, 24);
195         fc = cpu_to_le16(IEEE80211_FTYPE_DATA | IEEE80211_STYPE_NULLFUNC |
196                          IEEE80211_FCTL_TODS);
197         if (ps)
198                 fc |= cpu_to_le16(IEEE80211_FCTL_PM);
199         hdr->frame_control = fc;
200         memcpy(hdr->addr1, bssid, ETH_ALEN);
201         memcpy(hdr->addr2, aphy->hw->wiphy->perm_addr, ETH_ALEN);
202         memcpy(hdr->addr3, bssid, ETH_ALEN);
203
204         info = IEEE80211_SKB_CB(skb);
205         memset(info, 0, sizeof(*info));
206         info->flags = IEEE80211_TX_CTL_REQ_TX_STATUS;
207         info->control.vif = vif;
208         info->control.rates[0].idx = 0;
209         info->control.rates[0].count = 4;
210         info->control.rates[1].idx = -1;
211
212         memset(&txctl, 0, sizeof(struct ath_tx_control));
213         txctl.txq = &sc->tx.txq[sc->tx.hwq_map[ATH9K_WME_AC_VO]];
214         txctl.frame_type = ps ? ATH9K_INT_PAUSE : ATH9K_INT_UNPAUSE;
215
216         if (ath_tx_start(aphy->hw, skb, &txctl) != 0)
217                 goto exit;
218
219         return 0;
220 exit:
221         dev_kfree_skb_any(skb);
222         return -1;
223 }
224
225 /*
226  * ath9k version of ieee80211_tx_status() for TX frames that are generated
227  * internally in the driver.
228  */
229 void ath9k_tx_status(struct ieee80211_hw *hw, struct sk_buff *skb)
230 {
231         struct ath_wiphy *aphy = hw->priv;
232         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
233         struct ieee80211_tx_info *tx_info = IEEE80211_SKB_CB(skb);
234         struct ath_tx_info_priv *tx_info_priv = ATH_TX_INFO_PRIV(tx_info);
235
236         if (tx_info_priv && tx_info_priv->frame_type == ATH9K_INT_PAUSE &&
237             aphy->state == ATH_WIPHY_PAUSING) {
238                 if (!(info->flags & IEEE80211_TX_STAT_ACK)) {
239                         printk(KERN_DEBUG "ath9k: %s: no ACK for pause "
240                                "frame\n", wiphy_name(hw->wiphy));
241                         /*
242                          * The AP did not reply; ignore this to allow us to
243                          * continue.
244                          */
245                 }
246                 aphy->state = ATH_WIPHY_PAUSED;
247         }
248
249         kfree(tx_info_priv);
250         tx_info->rate_driver_data[0] = NULL;
251
252         dev_kfree_skb(skb);
253 }
254
255 static void ath9k_pause_iter(void *data, u8 *mac, struct ieee80211_vif *vif)
256 {
257         struct ath_wiphy *aphy = data;
258         struct ath_vif *avp = (void *) vif->drv_priv;
259
260         switch (vif->type) {
261         case NL80211_IFTYPE_STATION:
262                 if (!vif->bss_conf.assoc) {
263                         aphy->state = ATH_WIPHY_PAUSED;
264                         break;
265                 }
266                 /* TODO: could avoid this if already in PS mode */
267                 ath9k_send_nullfunc(aphy, vif, avp->bssid, 1);
268                 break;
269         case NL80211_IFTYPE_AP:
270                 /* Beacon transmission is paused by aphy->state change */
271                 aphy->state = ATH_WIPHY_PAUSED;
272                 break;
273         default:
274                 break;
275         }
276 }
277
278 /* caller must hold wiphy_lock */
279 static int __ath9k_wiphy_pause(struct ath_wiphy *aphy)
280 {
281         ieee80211_stop_queues(aphy->hw);
282         aphy->state = ATH_WIPHY_PAUSING;
283         /*
284          * TODO: handle PAUSING->PAUSED for the case where there are multiple
285          * active vifs (now we do it on the first vif getting ready; should be
286          * on the last)
287          */
288         ieee80211_iterate_active_interfaces_atomic(aphy->hw, ath9k_pause_iter,
289                                                    aphy);
290         return 0;
291 }
292
293 int ath9k_wiphy_pause(struct ath_wiphy *aphy)
294 {
295         int ret;
296         spin_lock_bh(&aphy->sc->wiphy_lock);
297         ret = __ath9k_wiphy_pause(aphy);
298         spin_unlock_bh(&aphy->sc->wiphy_lock);
299         return ret;
300 }
301
302 static void ath9k_unpause_iter(void *data, u8 *mac, struct ieee80211_vif *vif)
303 {
304         struct ath_wiphy *aphy = data;
305         struct ath_vif *avp = (void *) vif->drv_priv;
306
307         switch (vif->type) {
308         case NL80211_IFTYPE_STATION:
309                 if (!vif->bss_conf.assoc)
310                         break;
311                 ath9k_send_nullfunc(aphy, vif, avp->bssid, 0);
312                 break;
313         case NL80211_IFTYPE_AP:
314                 /* Beacon transmission is re-enabled by aphy->state change */
315                 break;
316         default:
317                 break;
318         }
319 }
320
321 /* caller must hold wiphy_lock */
322 static int __ath9k_wiphy_unpause(struct ath_wiphy *aphy)
323 {
324         ieee80211_iterate_active_interfaces_atomic(aphy->hw,
325                                                    ath9k_unpause_iter, aphy);
326         aphy->state = ATH_WIPHY_ACTIVE;
327         ieee80211_wake_queues(aphy->hw);
328         return 0;
329 }
330
331 int ath9k_wiphy_unpause(struct ath_wiphy *aphy)
332 {
333         int ret;
334         spin_lock_bh(&aphy->sc->wiphy_lock);
335         ret = __ath9k_wiphy_unpause(aphy);
336         spin_unlock_bh(&aphy->sc->wiphy_lock);
337         return ret;
338 }