drivers/net/wireless/intersil/hostap/hostap_ap.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * Intersil Prism2 driver with Host AP (software access point) support
   4  * Copyright (c) 2001-2002, SSH Communications Security Corp and Jouni Malinen
   5  * <j@w1.fi>
   6  * Copyright (c) 2002-2005, Jouni Malinen <j@w1.fi>
   7  *
   8  * This file is to be included into hostap.c when S/W AP functionality is
   9  * compiled.
  10  *
  11  * AP:  FIX:
  12  * - if unicast Class 2 (assoc,reassoc,disassoc) frame received from
  13  *   unauthenticated STA, send deauth. frame (8802.11: 5.5)
  14  * - if unicast Class 3 (data with to/from DS,deauth,pspoll) frame received
  15  *   from authenticated, but unassoc STA, send disassoc frame (8802.11: 5.5)
  16  * - if unicast Class 3 received from unauthenticated STA, send deauth. frame
  17  *   (8802.11: 5.5)
  18  */
  19
  20 #include <linux/proc_fs.h>
  21 #include <linux/seq_file.h>
  22 #include <linux/delay.h>
  23 #include <linux/random.h>
  24 #include <linux/if_arp.h>
  25 #include <linux/slab.h>
  26 #include <linux/export.h>
  27 #include <linux/moduleparam.h>
  28 #include <linux/etherdevice.h>
  29
  30 #include "hostap_wlan.h"
  31 #include "hostap.h"
  32 #include "hostap_ap.h"
  33
  34 static int other_ap_policy[MAX_PARM_DEVICES] = { AP_OTHER_AP_SKIP_ALL,
  35                                                  DEF_INTS };
  36 module_param_array(other_ap_policy, int, NULL, 0444);
  37 MODULE_PARM_DESC(other_ap_policy, "Other AP beacon monitoring policy (0-3)");
  38
  39 static int ap_max_inactivity[MAX_PARM_DEVICES] = { AP_MAX_INACTIVITY_SEC,
  40                                                    DEF_INTS };
  41 module_param_array(ap_max_inactivity, int, NULL, 0444);
  42 MODULE_PARM_DESC(ap_max_inactivity, "AP timeout (in seconds) for station "
  43                  "inactivity");
  44
  45 static int ap_bridge_packets[MAX_PARM_DEVICES] = { 1, DEF_INTS };
  46 module_param_array(ap_bridge_packets, int, NULL, 0444);
  47 MODULE_PARM_DESC(ap_bridge_packets, "Bridge packets directly between "
  48                  "stations");
  49
  50 static int autom_ap_wds[MAX_PARM_DEVICES] = { 0, DEF_INTS };
  51 module_param_array(autom_ap_wds, int, NULL, 0444);
  52 MODULE_PARM_DESC(autom_ap_wds, "Add WDS connections to other APs "
  53                  "automatically");
  54
  55
  56 static struct sta_info* ap_get_sta(struct ap_data *ap, u8 *sta);
  57 static void hostap_event_expired_sta(struct net_device *dev,
  58                                      struct sta_info *sta);
  59 static void handle_add_proc_queue(struct work_struct *work);
  60
  61 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
  62 static void handle_wds_oper_queue(struct work_struct *work);
  63 static void prism2_send_mgmt(struct net_device *dev,
  64                              u16 type_subtype, char *body,
  65                              int body_len, u8 *addr, u16 tx_cb_idx);
  66 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
  67
  68
  69 #if !defined(PRISM2_NO_PROCFS_DEBUG) && defined(CONFIG_PROC_FS)
  70 static int ap_debug_proc_show(struct seq_file *m, void *v)
  71 {
  72         struct ap_data *ap = PDE_DATA(file_inode(m->file));
  73
  74         seq_printf(m, "BridgedUnicastFrames=%u\n", ap->bridged_unicast);
  75         seq_printf(m, "BridgedMulticastFrames=%u\n", ap->bridged_multicast);
  76         seq_printf(m, "max_inactivity=%u\n", ap->max_inactivity / HZ);
  77         seq_printf(m, "bridge_packets=%u\n", ap->bridge_packets);
  78         seq_printf(m, "nullfunc_ack=%u\n", ap->nullfunc_ack);
  79         seq_printf(m, "autom_ap_wds=%u\n", ap->autom_ap_wds);
  80         seq_printf(m, "auth_algs=%u\n", ap->local->auth_algs);
  81         seq_printf(m, "tx_drop_nonassoc=%u\n", ap->tx_drop_nonassoc);
  82         return 0;
  83 }
  84 #endif
  85
  86 static void ap_sta_hash_add(struct ap_data *ap, struct sta_info *sta)
  87 {
  88         sta->hnext = ap->sta_hash[STA_HASH(sta->addr)];
  89         ap->sta_hash[STA_HASH(sta->addr)] = sta;
  90 }
  91
  92 static void ap_sta_hash_del(struct ap_data *ap, struct sta_info *sta)
  93 {
  94         struct sta_info *s;
  95
  96         s = ap->sta_hash[STA_HASH(sta->addr)];
  97         if (s == NULL) return;
  98         if (ether_addr_equal(s->addr, sta->addr)) {
  99                 ap->sta_hash[STA_HASH(sta->addr)] = s->hnext;
 100                 return;
 101         }
 102
 103         while (s->hnext != NULL && !ether_addr_equal(s->hnext->addr, sta->addr))
 104                 s = s->hnext;
 105         if (s->hnext != NULL)
 106                 s->hnext = s->hnext->hnext;
 107         else
 108                 printk("AP: could not remove STA %pM from hash table\n",
 109                        sta->addr);
 110 }
 111
 112 static void ap_free_sta(struct ap_data *ap, struct sta_info *sta)
 113 {
 114         if (sta->ap && sta->local)
 115                 hostap_event_expired_sta(sta->local->dev, sta);
 116
 117         if (ap->proc != NULL) {
 118                 char name[20];
 119                 sprintf(name, "%pM", sta->addr);
 120                 remove_proc_entry(name, ap->proc);
 121         }
 122
 123         if (sta->crypt) {
 124                 sta->crypt->ops->deinit(sta->crypt->priv);
 125                 kfree(sta->crypt);
 126                 sta->crypt = NULL;
 127         }
 128
 129         skb_queue_purge(&sta->tx_buf);
 130
 131         ap->num_sta--;
 132 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
 133         if (sta->aid > 0)
 134                 ap->sta_aid[sta->aid - 1] = NULL;
 135
 136         if (!sta->ap)
 137                 kfree(sta->u.sta.challenge);
 138         del_timer_sync(&sta->timer);
 139 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
 140
 141         kfree(sta);
 142 }
 143
 144
 145 static void hostap_set_tim(local_info_t *local, int aid, int set)
 146 {
 147         if (local->func->set_tim)
 148                 local->func->set_tim(local->dev, aid, set);
 149 }
 150
 151
 152 static void hostap_event_new_sta(struct net_device *dev, struct sta_info *sta)
 153 {
 154         union iwreq_data wrqu;
 155         memset(&wrqu, 0, sizeof(wrqu));
 156         memcpy(wrqu.addr.sa_data, sta->addr, ETH_ALEN);
 157         wrqu.addr.sa_family = ARPHRD_ETHER;
 158         wireless_send_event(dev, IWEVREGISTERED, &wrqu, NULL);
 159 }
 160
 161
 162 static void hostap_event_expired_sta(struct net_device *dev,
 163                                      struct sta_info *sta)
 164 {
 165         union iwreq_data wrqu;
 166         memset(&wrqu, 0, sizeof(wrqu));
 167         memcpy(wrqu.addr.sa_data, sta->addr, ETH_ALEN);
 168         wrqu.addr.sa_family = ARPHRD_ETHER;
 169         wireless_send_event(dev, IWEVEXPIRED, &wrqu, NULL);
 170 }
 171
 172
 173 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
 174
 175 static void ap_handle_timer(struct timer_list *t)
 176 {
 177         struct sta_info *sta = from_timer(sta, t, timer);
 178         local_info_t *local;
 179         struct ap_data *ap;
 180         unsigned long next_time = 0;
 181         int was_assoc;
 182
 183         if (sta == NULL || sta->local == NULL || sta->local->ap == NULL) {
 184                 PDEBUG(DEBUG_AP, "ap_handle_timer() called with NULL data\n");
 185                 return;
 186         }
 187
 188         local = sta->local;
 189         ap = local->ap;
 190         was_assoc = sta->flags & WLAN_STA_ASSOC;
 191
 192         if (atomic_read(&sta->users) != 0)
 193                 next_time = jiffies + HZ;
 194         else if ((sta->flags & WLAN_STA_PERM) && !(sta->flags & WLAN_STA_AUTH))
 195                 next_time = jiffies + ap->max_inactivity;
 196
 197         if (time_before(jiffies, sta->last_rx + ap->max_inactivity)) {
 198                 /* station activity detected; reset timeout state */
 199                 sta->timeout_next = STA_NULLFUNC;
 200                 next_time = sta->last_rx + ap->max_inactivity;
 201         } else if (sta->timeout_next == STA_DISASSOC &&
 202                    !(sta->flags & WLAN_STA_PENDING_POLL)) {
 203                 /* STA ACKed data nullfunc frame poll */
 204                 sta->timeout_next = STA_NULLFUNC;
 205                 next_time = jiffies + ap->max_inactivity;
 206         }
 207
 208         if (next_time) {
 209                 sta->timer.expires = next_time;
 210                 add_timer(&sta->timer);
 211                 return;
 212         }
 213
 214         if (sta->ap)
 215                 sta->timeout_next = STA_DEAUTH;
 216
 217         if (sta->timeout_next == STA_DEAUTH && !(sta->flags & WLAN_STA_PERM)) {
 218                 spin_lock(&ap->sta_table_lock);
 219                 ap_sta_hash_del(ap, sta);
 220                 list_del(&sta->list);
 221                 spin_unlock(&ap->sta_table_lock);
 222                 sta->flags &= ~(WLAN_STA_AUTH | WLAN_STA_ASSOC);
 223         } else if (sta->timeout_next == STA_DISASSOC)
 224                 sta->flags &= ~WLAN_STA_ASSOC;
 225
 226         if (was_assoc && !(sta->flags & WLAN_STA_ASSOC) && !sta->ap)
 227                 hostap_event_expired_sta(local->dev, sta);
 228
 229         if (sta->timeout_next == STA_DEAUTH && sta->aid > 0 &&
 230             !skb_queue_empty(&sta->tx_buf)) {
 231                 hostap_set_tim(local, sta->aid, 0);
 232                 sta->flags &= ~WLAN_STA_TIM;
 233         }
 234
 235         if (sta->ap) {
 236                 if (ap->autom_ap_wds) {
 237                         PDEBUG(DEBUG_AP, "%s: removing automatic WDS "
 238                                "connection to AP %pM\n",
 239                                local->dev->name, sta->addr);
 240                         hostap_wds_link_oper(local, sta->addr, WDS_DEL);
 241                 }
 242         } else if (sta->timeout_next == STA_NULLFUNC) {
 243                 /* send data frame to poll STA and check whether this frame
 244                  * is ACKed */
 245                 /* FIX: IEEE80211_STYPE_NULLFUNC would be more appropriate, but
 246                  * it is apparently not retried so TX Exc events are not
 247                  * received for it */
 248                 sta->flags |= WLAN_STA_PENDING_POLL;
 249                 prism2_send_mgmt(local->dev, IEEE80211_FTYPE_DATA |
 250                                  IEEE80211_STYPE_DATA, NULL, 0,
 251                                  sta->addr, ap->tx_callback_poll);
 252         } else {
 253                 int deauth = sta->timeout_next == STA_DEAUTH;
 254                 __le16 resp;
 255                 PDEBUG(DEBUG_AP, "%s: sending %s info to STA %pM"
 256                        "(last=%lu, jiffies=%lu)\n",
 257                        local->dev->name,
 258                        deauth ? "deauthentication" : "disassociation",
 259                        sta->addr, sta->last_rx, jiffies);
 260
 261                 resp = cpu_to_le16(deauth ? WLAN_REASON_PREV_AUTH_NOT_VALID :
 262                                    WLAN_REASON_DISASSOC_DUE_TO_INACTIVITY);
 263                 prism2_send_mgmt(local->dev, IEEE80211_FTYPE_MGMT |
 264                                  (deauth ? IEEE80211_STYPE_DEAUTH :
 265                                   IEEE80211_STYPE_DISASSOC),
 266                                  (char *) &resp, 2, sta->addr, 0);
 267         }
 268
 269         if (sta->timeout_next == STA_DEAUTH) {
 270                 if (sta->flags & WLAN_STA_PERM) {
 271                         PDEBUG(DEBUG_AP, "%s: STA %pM"
 272                                " would have been removed, "
 273                                "but it has 'perm' flag\n",
 274                                local->dev->name, sta->addr);
 275                 } else
 276                         ap_free_sta(ap, sta);
 277                 return;
 278         }
 279
 280         if (sta->timeout_next == STA_NULLFUNC) {
 281                 sta->timeout_next = STA_DISASSOC;
 282                 sta->timer.expires = jiffies + AP_DISASSOC_DELAY;
 283         } else {
 284                 sta->timeout_next = STA_DEAUTH;
 285                 sta->timer.expires = jiffies + AP_DEAUTH_DELAY;
 286         }
 287
 288         add_timer(&sta->timer);
 289 }
 290
 291
 292 void hostap_deauth_all_stas(struct net_device *dev, struct ap_data *ap,
 293                             int resend)
 294 {
 295         u8 addr[ETH_ALEN];
 296         __le16 resp;
 297         int i;
 298
 299         PDEBUG(DEBUG_AP, "%s: Deauthenticate all stations\n", dev->name);
 300         eth_broadcast_addr(addr);
 301
 302         resp = cpu_to_le16(WLAN_REASON_PREV_AUTH_NOT_VALID);
 303
 304         /* deauth message sent; try to resend it few times; the message is
 305          * broadcast, so it may be delayed until next DTIM; there is not much
 306          * else we can do at this point since the driver is going to be shut
 307          * down */
 308         for (i = 0; i < 5; i++) {
 309                 prism2_send_mgmt(dev, IEEE80211_FTYPE_MGMT |
 310                                  IEEE80211_STYPE_DEAUTH,
 311                                  (char *) &resp, 2, addr, 0);
 312
 313                 if (!resend || ap->num_sta <= 0)
 314                         return;
 315
 316                 mdelay(50);
 317         }
 318 }
 319
 320
 321 static int ap_control_proc_show(struct seq_file *m, void *v)
 322 {
 323         struct ap_data *ap = PDE_DATA(file_inode(m->file));
 324         char *policy_txt;
 325         struct mac_entry *entry;
 326
 327         if (v == SEQ_START_TOKEN) {
 328                 switch (ap->mac_restrictions.policy) {
 329                 case MAC_POLICY_OPEN:
 330                         policy_txt = "open";
 331                         break;
 332                 case MAC_POLICY_ALLOW:
 333                         policy_txt = "allow";
 334                         break;
 335                 case MAC_POLICY_DENY:
 336                         policy_txt = "deny";
 337                         break;
 338                 default:
 339                         policy_txt = "unknown";
 340                         break;
 341                 }
 342                 seq_printf(m, "MAC policy: %s\n", policy_txt);
 343                 seq_printf(m, "MAC entries: %u\n", ap->mac_restrictions.entries);
 344                 seq_puts(m, "MAC list:\n");
 345                 return 0;
 346         }
 347
 348         entry = v;
 349         seq_printf(m, "%pM\n", entry->addr);
 350         return 0;
 351 }
 352
 353 static void *ap_control_proc_start(struct seq_file *m, loff_t *_pos)
 354 {
 355         struct ap_data *ap = PDE_DATA(file_inode(m->file));
 356         spin_lock_bh(&ap->mac_restrictions.lock);
 357         return seq_list_start_head(&ap->mac_restrictions.mac_list, *_pos);
 358 }
 359
 360 static void *ap_control_proc_next(struct seq_file *m, void *v, loff_t *_pos)
 361 {
 362         struct ap_data *ap = PDE_DATA(file_inode(m->file));
 363         return seq_list_next(v, &ap->mac_restrictions.mac_list, _pos);
 364 }
 365
 366 static void ap_control_proc_stop(struct seq_file *m, void *v)
 367 {
 368         struct ap_data *ap = PDE_DATA(file_inode(m->file));
 369         spin_unlock_bh(&ap->mac_restrictions.lock);
 370 }
 371
 372 static const struct seq_operations ap_control_proc_seqops = {
 373         .start  = ap_control_proc_start,
 374         .next   = ap_control_proc_next,
 375         .stop   = ap_control_proc_stop,
 376         .show   = ap_control_proc_show,
 377 };
 378
 379 int ap_control_add_mac(struct mac_restrictions *mac_restrictions, u8 *mac)
 380 {
 381         struct mac_entry *entry;
 382
 383         entry = kmalloc(sizeof(struct mac_entry), GFP_KERNEL);
 384         if (entry == NULL)
 385                 return -ENOMEM;
 386
 387         memcpy(entry->addr, mac, ETH_ALEN);
 388
 389         spin_lock_bh(&mac_restrictions->lock);
 390         list_add_tail(&entry->list, &mac_restrictions->mac_list);
 391         mac_restrictions->entries++;
 392         spin_unlock_bh(&mac_restrictions->lock);
 393
 394         return 0;
 395 }
 396
 397
 398 int ap_control_del_mac(struct mac_restrictions *mac_restrictions, u8 *mac)
 399 {
 400         struct list_head *ptr;
 401         struct mac_entry *entry;
 402
 403         spin_lock_bh(&mac_restrictions->lock);
 404         for (ptr = mac_restrictions->mac_list.next;
 405              ptr != &mac_restrictions->mac_list; ptr = ptr->next) {
 406                 entry = list_entry(ptr, struct mac_entry, list);
 407
 408                 if (ether_addr_equal(entry->addr, mac)) {
 409                         list_del(ptr);
 410                         kfree(entry);
 411                         mac_restrictions->entries--;
 412                         spin_unlock_bh(&mac_restrictions->lock);
 413                         return 0;
 414                 }
 415         }
 416         spin_unlock_bh(&mac_restrictions->lock);
 417         return -1;
 418 }
 419
 420
 421 static int ap_control_mac_deny(struct mac_restrictions *mac_restrictions,
 422                                u8 *mac)
 423 {
 424         struct mac_entry *entry;
 425         int found = 0;
 426
 427         if (mac_restrictions->policy == MAC_POLICY_OPEN)
 428                 return 0;
 429
 430         spin_lock_bh(&mac_restrictions->lock);
 431         list_for_each_entry(entry, &mac_restrictions->mac_list, list) {
 432                 if (ether_addr_equal(entry->addr, mac)) {
 433                         found = 1;
 434                         break;
 435                 }
 436         }
 437         spin_unlock_bh(&mac_restrictions->lock);
 438
 439         if (mac_restrictions->policy == MAC_POLICY_ALLOW)
 440                 return !found;
 441         else
 442                 return found;
 443 }
 444
 445
 446 void ap_control_flush_macs(struct mac_restrictions *mac_restrictions)
 447 {
 448         struct list_head *ptr, *n;
 449         struct mac_entry *entry;
 450
 451         if (mac_restrictions->entries == 0)
 452                 return;
 453
 454         spin_lock_bh(&mac_restrictions->lock);
 455         for (ptr = mac_restrictions->mac_list.next, n = ptr->next;
 456              ptr != &mac_restrictions->mac_list;
 457              ptr = n, n = ptr->next) {
 458                 entry = list_entry(ptr, struct mac_entry, list);
 459                 list_del(ptr);
 460                 kfree(entry);
 461         }
 462         mac_restrictions->entries = 0;
 463         spin_unlock_bh(&mac_restrictions->lock);
 464 }
 465
 466
 467 int ap_control_kick_mac(struct ap_data *ap, struct net_device *dev, u8 *mac)
 468 {
 469         struct sta_info *sta;
 470         __le16 resp;
 471
 472         spin_lock_bh(&ap->sta_table_lock);
 473         sta = ap_get_sta(ap, mac);
 474         if (sta) {
 475                 ap_sta_hash_del(ap, sta);
 476                 list_del(&sta->list);
 477         }
 478         spin_unlock_bh(&ap->sta_table_lock);
 479
 480         if (!sta)
 481                 return -EINVAL;
 482
 483         resp = cpu_to_le16(WLAN_REASON_PREV_AUTH_NOT_VALID);
 484         prism2_send_mgmt(dev, IEEE80211_FTYPE_MGMT | IEEE80211_STYPE_DEAUTH,
 485                          (char *) &resp, 2, sta->addr, 0);
 486
 487         if ((sta->flags & WLAN_STA_ASSOC) && !sta->ap)
 488                 hostap_event_expired_sta(dev, sta);
 489
 490         ap_free_sta(ap, sta);
 491
 492         return 0;
 493 }
 494
 495 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
 496
 497
 498 void ap_control_kickall(struct ap_data *ap)
 499 {
 500         struct list_head *ptr, *n;
 501         struct sta_info *sta;
 502
 503         spin_lock_bh(&ap->sta_table_lock);
 504         for (ptr = ap->sta_list.next, n = ptr->next; ptr != &ap->sta_list;
 505              ptr = n, n = ptr->next) {
 506                 sta = list_entry(ptr, struct sta_info, list);
 507                 ap_sta_hash_del(ap, sta);
 508                 list_del(&sta->list);
 509                 if ((sta->flags & WLAN_STA_ASSOC) && !sta->ap && sta->local)
 510                         hostap_event_expired_sta(sta->local->dev, sta);
 511                 ap_free_sta(ap, sta);
 512         }
 513         spin_unlock_bh(&ap->sta_table_lock);
 514 }
 515
 516
 517 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
 518
 519 static int prism2_ap_proc_show(struct seq_file *m, void *v)
 520 {
 521         struct sta_info *sta = v;
 522         int i;
 523
 524         if (v == SEQ_START_TOKEN) {
 525                 seq_printf(m, "# BSSID CHAN SIGNAL NOISE RATE SSID FLAGS\n");
 526                 return 0;
 527         }
 528
 529         if (!sta->ap)
 530                 return 0;
 531
 532         seq_printf(m, "%pM %d %d %d %d '",
 533                    sta->addr,
 534                    sta->u.ap.channel, sta->last_rx_signal,
 535                    sta->last_rx_silence, sta->last_rx_rate);
 536
 537         for (i = 0; i < sta->u.ap.ssid_len; i++) {
 538                 if (sta->u.ap.ssid[i] >= 32 && sta->u.ap.ssid[i] < 127)
 539                         seq_putc(m, sta->u.ap.ssid[i]);
 540                 else
 541                         seq_printf(m, "<%02x>", sta->u.ap.ssid[i]);
 542         }
 543
 544         seq_putc(m, '\'');
 545         if (sta->capability & WLAN_CAPABILITY_ESS)
 546                 seq_puts(m, " [ESS]");
 547         if (sta->capability & WLAN_CAPABILITY_IBSS)
 548                 seq_puts(m, " [IBSS]");
 549         if (sta->capability & WLAN_CAPABILITY_PRIVACY)
 550                 seq_puts(m, " [WEP]");
 551         seq_putc(m, '\n');
 552         return 0;
 553 }
 554
 555 static void *prism2_ap_proc_start(struct seq_file *m, loff_t *_pos)
 556 {
 557         struct ap_data *ap = PDE_DATA(file_inode(m->file));
 558         spin_lock_bh(&ap->sta_table_lock);
 559         return seq_list_start_head(&ap->sta_list, *_pos);
 560 }
 561
 562 static void *prism2_ap_proc_next(struct seq_file *m, void *v, loff_t *_pos)
 563 {
 564         struct ap_data *ap = PDE_DATA(file_inode(m->file));
 565         return seq_list_next(v, &ap->sta_list, _pos);
 566 }
 567
 568 static void prism2_ap_proc_stop(struct seq_file *m, void *v)
 569 {
 570         struct ap_data *ap = PDE_DATA(file_inode(m->file));
 571         spin_unlock_bh(&ap->sta_table_lock);
 572 }
 573
 574 static const struct seq_operations prism2_ap_proc_seqops = {
 575         .start  = prism2_ap_proc_start,
 576         .next   = prism2_ap_proc_next,
 577         .stop   = prism2_ap_proc_stop,
 578         .show   = prism2_ap_proc_show,
 579 };
 580 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
 581
 582
 583 void hostap_check_sta_fw_version(struct ap_data *ap, int sta_fw_ver)
 584 {
 585         if (!ap)
 586                 return;
 587
 588         if (sta_fw_ver == PRISM2_FW_VER(0,8,0)) {
 589                 PDEBUG(DEBUG_AP, "Using data::nullfunc ACK workaround - "
 590                        "firmware upgrade recommended\n");
 591                 ap->nullfunc_ack = 1;
 592         } else
 593                 ap->nullfunc_ack = 0;
 594
 595         if (sta_fw_ver == PRISM2_FW_VER(1,4,2)) {
 596                 printk(KERN_WARNING "%s: Warning: secondary station firmware "
 597                        "version 1.4.2 does not seem to work in Host AP mode\n",
 598                        ap->local->dev->name);
 599         }
 600 }
 601
 602
 603 /* Called only as a tasklet (software IRQ) */
 604 static void hostap_ap_tx_cb(struct sk_buff *skb, int ok, void *data)
 605 {
 606         struct ap_data *ap = data;
 607         struct ieee80211_hdr *hdr;
 608
 609         if (!ap->local->hostapd || !ap->local->apdev) {
 610                 dev_kfree_skb(skb);
 611                 return;
 612         }
 613
 614         /* Pass the TX callback frame to the hostapd; use 802.11 header version
 615          * 1 to indicate failure (no ACK) and 2 success (frame ACKed) */
 616
 617         hdr = (struct ieee80211_hdr *) skb->data;
 618         hdr->frame_control &= cpu_to_le16(~IEEE80211_FCTL_VERS);
 619         hdr->frame_control |= cpu_to_le16(ok ? BIT(1) : BIT(0));
 620
 621         skb->dev = ap->local->apdev;
 622         skb_pull(skb, hostap_80211_get_hdrlen(hdr->frame_control));
 623         skb->pkt_type = PACKET_OTHERHOST;
 624         skb->protocol = cpu_to_be16(ETH_P_802_2);
 625         memset(skb->cb, 0, sizeof(skb->cb));
 626         netif_rx(skb);
 627 }
 628
 629
 630 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
 631 /* Called only as a tasklet (software IRQ) */
 632 static void hostap_ap_tx_cb_auth(struct sk_buff *skb, int ok, void *data)
 633 {
 634         struct ap_data *ap = data;
 635         struct net_device *dev = ap->local->dev;
 636         struct ieee80211_hdr *hdr;
 637         u16 auth_alg, auth_transaction, status;
 638         __le16 *pos;
 639         struct sta_info *sta = NULL;
 640         char *txt = NULL;
 641
 642         if (ap->local->hostapd) {
 643                 dev_kfree_skb(skb);
 644                 return;
 645         }
 646
 647         hdr = (struct ieee80211_hdr *) skb->data;
 648         if (!ieee80211_is_auth(hdr->frame_control) ||
 649             skb->len < IEEE80211_MGMT_HDR_LEN + 6) {
 650                 printk(KERN_DEBUG "%s: hostap_ap_tx_cb_auth received invalid "
 651                        "frame\n", dev->name);
 652                 dev_kfree_skb(skb);
 653                 return;
 654         }
 655
 656         pos = (__le16 *) (skb->data + IEEE80211_MGMT_HDR_LEN);
 657         auth_alg = le16_to_cpu(*pos++);
 658         auth_transaction = le16_to_cpu(*pos++);
 659         status = le16_to_cpu(*pos++);
 660
 661         if (!ok) {
 662                 txt = "frame was not ACKed";
 663                 goto done;
 664         }
 665
 666         spin_lock(&ap->sta_table_lock);
 667         sta = ap_get_sta(ap, hdr->addr1);
 668         if (sta)
 669                 atomic_inc(&sta->users);
 670         spin_unlock(&ap->sta_table_lock);
 671
 672         if (!sta) {
 673                 txt = "STA not found";
 674                 goto done;
 675         }
 676
 677         if (status == WLAN_STATUS_SUCCESS &&
 678             ((auth_alg == WLAN_AUTH_OPEN && auth_transaction == 2) ||
 679              (auth_alg == WLAN_AUTH_SHARED_KEY && auth_transaction == 4))) {
 680                 txt = "STA authenticated";
 681                 sta->flags |= WLAN_STA_AUTH;
 682                 sta->last_auth = jiffies;
 683         } else if (status != WLAN_STATUS_SUCCESS)
 684                 txt = "authentication failed";
 685
 686  done:
 687         if (sta)
 688                 atomic_dec(&sta->users);
 689         if (txt) {
 690                 PDEBUG(DEBUG_AP, "%s: %pM auth_cb - alg=%d "
 691                        "trans#=%d status=%d - %s\n",
 692                        dev->name, hdr->addr1,
 693                        auth_alg, auth_transaction, status, txt);
 694         }
 695         dev_kfree_skb(skb);
 696 }
 697
 698
 699 /* Called only as a tasklet (software IRQ) */
 700 static void hostap_ap_tx_cb_assoc(struct sk_buff *skb, int ok, void *data)
 701 {
 702         struct ap_data *ap = data;
 703         struct net_device *dev = ap->local->dev;
 704         struct ieee80211_hdr *hdr;
 705         u16 status;
 706         __le16 *pos;
 707         struct sta_info *sta = NULL;
 708         char *txt = NULL;
 709
 710         if (ap->local->hostapd) {
 711                 dev_kfree_skb(skb);
 712                 return;
 713         }
 714
 715         hdr = (struct ieee80211_hdr *) skb->data;
 716         if ((!ieee80211_is_assoc_resp(hdr->frame_control) &&
 717              !ieee80211_is_reassoc_resp(hdr->frame_control)) ||
 718             skb->len < IEEE80211_MGMT_HDR_LEN + 4) {
 719                 printk(KERN_DEBUG "%s: hostap_ap_tx_cb_assoc received invalid "
 720                        "frame\n", dev->name);
 721                 dev_kfree_skb(skb);
 722                 return;
 723         }
 724
 725         if (!ok) {
 726                 txt = "frame was not ACKed";
 727                 goto done;
 728         }
 729
 730         spin_lock(&ap->sta_table_lock);
 731         sta = ap_get_sta(ap, hdr->addr1);
 732         if (sta)
 733                 atomic_inc(&sta->users);
 734         spin_unlock(&ap->sta_table_lock);
 735
 736         if (!sta) {
 737                 txt = "STA not found";
 738                 goto done;
 739         }
 740
 741         pos = (__le16 *) (skb->data + IEEE80211_MGMT_HDR_LEN);
 742         pos++;
 743         status = le16_to_cpu(*pos++);
 744         if (status == WLAN_STATUS_SUCCESS) {
 745                 if (!(sta->flags & WLAN_STA_ASSOC))
 746                         hostap_event_new_sta(dev, sta);
 747                 txt = "STA associated";
 748                 sta->flags |= WLAN_STA_ASSOC;
 749                 sta->last_assoc = jiffies;
 750         } else
 751                 txt = "association failed";
 752
 753  done:
 754         if (sta)
 755                 atomic_dec(&sta->users);
 756         if (txt) {
 757                 PDEBUG(DEBUG_AP, "%s: %pM assoc_cb - %s\n",
 758                        dev->name, hdr->addr1, txt);
 759         }
 760         dev_kfree_skb(skb);
 761 }
 762
 763 /* Called only as a tasklet (software IRQ); TX callback for poll frames used
 764  * in verifying whether the STA is still present. */
 765 static void hostap_ap_tx_cb_poll(struct sk_buff *skb, int ok, void *data)
 766 {
 767         struct ap_data *ap = data;
 768         struct ieee80211_hdr *hdr;
 769         struct sta_info *sta;
 770
 771         if (skb->len < 24)
 772                 goto fail;
 773         hdr = (struct ieee80211_hdr *) skb->data;
 774         if (ok) {
 775                 spin_lock(&ap->sta_table_lock);
 776                 sta = ap_get_sta(ap, hdr->addr1);
 777                 if (sta)
 778                         sta->flags &= ~WLAN_STA_PENDING_POLL;
 779                 spin_unlock(&ap->sta_table_lock);
 780         } else {
 781                 PDEBUG(DEBUG_AP,
 782                        "%s: STA %pM did not ACK activity poll frame\n",
 783                        ap->local->dev->name, hdr->addr1);
 784         }
 785
 786  fail:
 787         dev_kfree_skb(skb);
 788 }
 789 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
 790
 791
 792 void hostap_init_data(local_info_t *local)
 793 {
 794         struct ap_data *ap = local->ap;
 795
 796         if (ap == NULL) {
 797                 printk(KERN_WARNING "hostap_init_data: ap == NULL\n");
 798                 return;
 799         }
 800         memset(ap, 0, sizeof(struct ap_data));
 801         ap->local = local;
 802
 803         ap->ap_policy = GET_INT_PARM(other_ap_policy, local->card_idx);
 804         ap->bridge_packets = GET_INT_PARM(ap_bridge_packets, local->card_idx);
 805         ap->max_inactivity =
 806                 GET_INT_PARM(ap_max_inactivity, local->card_idx) * HZ;
 807         ap->autom_ap_wds = GET_INT_PARM(autom_ap_wds, local->card_idx);
 808
 809         spin_lock_init(&ap->sta_table_lock);
 810         INIT_LIST_HEAD(&ap->sta_list);
 811
 812         /* Initialize task queue structure for AP management */
 813         INIT_WORK(&local->ap->add_sta_proc_queue, handle_add_proc_queue);
 814
 815         ap->tx_callback_idx =
 816                 hostap_tx_callback_register(local, hostap_ap_tx_cb, ap);
 817         if (ap->tx_callback_idx == 0)
 818                 printk(KERN_WARNING "%s: failed to register TX callback for "
 819                        "AP\n", local->dev->name);
 820 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
 821         INIT_WORK(&local->ap->wds_oper_queue, handle_wds_oper_queue);
 822
 823         ap->tx_callback_auth =
 824                 hostap_tx_callback_register(local, hostap_ap_tx_cb_auth, ap);
 825         ap->tx_callback_assoc =
 826                 hostap_tx_callback_register(local, hostap_ap_tx_cb_assoc, ap);
 827         ap->tx_callback_poll =
 828                 hostap_tx_callback_register(local, hostap_ap_tx_cb_poll, ap);
 829         if (ap->tx_callback_auth == 0 || ap->tx_callback_assoc == 0 ||
 830                 ap->tx_callback_poll == 0)
 831                 printk(KERN_WARNING "%s: failed to register TX callback for "
 832                        "AP\n", local->dev->name);
 833
 834         spin_lock_init(&ap->mac_restrictions.lock);
 835         INIT_LIST_HEAD(&ap->mac_restrictions.mac_list);
 836 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
 837
 838         ap->initialized = 1;
 839 }
 840
 841
 842 void hostap_init_ap_proc(local_info_t *local)
 843 {
 844         struct ap_data *ap = local->ap;
 845
 846         ap->proc = local->proc;
 847         if (ap->proc == NULL)
 848                 return;
 849
 850 #ifndef PRISM2_NO_PROCFS_DEBUG
 851         proc_create_single_data("ap_debug", 0, ap->proc, ap_debug_proc_show, ap);
 852 #endif /* PRISM2_NO_PROCFS_DEBUG */
 853
 854 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
 855         proc_create_seq_data("ap_control", 0, ap->proc, &ap_control_proc_seqops,
 856                         ap);
 857         proc_create_seq_data("ap", 0, ap->proc, &prism2_ap_proc_seqops, ap);
 858 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
 859
 860 }
 861
 862
 863 void hostap_free_data(struct ap_data *ap)
 864 {
 865         struct sta_info *n, *sta;
 866
 867         if (ap == NULL || !ap->initialized) {
 868                 printk(KERN_DEBUG "hostap_free_data: ap has not yet been "
 869                        "initialized - skip resource freeing\n");
 870                 return;
 871         }
 872
 873         flush_work(&ap->add_sta_proc_queue);
 874
 875 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
 876         flush_work(&ap->wds_oper_queue);
 877         if (ap->crypt)
 878                 ap->crypt->deinit(ap->crypt_priv);
 879         ap->crypt = ap->crypt_priv = NULL;
 880 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
 881
 882         list_for_each_entry_safe(sta, n, &ap->sta_list, list) {
 883                 ap_sta_hash_del(ap, sta);
 884                 list_del(&sta->list);
 885                 if ((sta->flags & WLAN_STA_ASSOC) && !sta->ap && sta->local)
 886                         hostap_event_expired_sta(sta->local->dev, sta);
 887                 ap_free_sta(ap, sta);
 888         }
 889
 890 #ifndef PRISM2_NO_PROCFS_DEBUG
 891         if (ap->proc != NULL) {
 892                 remove_proc_entry("ap_debug", ap->proc);
 893         }
 894 #endif /* PRISM2_NO_PROCFS_DEBUG */
 895
 896 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
 897         if (ap->proc != NULL) {
 898           remove_proc_entry("ap", ap->proc);
 899                 remove_proc_entry("ap_control", ap->proc);
 900         }
 901         ap_control_flush_macs(&ap->mac_restrictions);
 902 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
 903
 904         ap->initialized = 0;
 905 }
 906
 907
 908 /* caller should have mutex for AP STA list handling */
 909 static struct sta_info* ap_get_sta(struct ap_data *ap, u8 *sta)
 910 {
 911         struct sta_info *s;
 912
 913         s = ap->sta_hash[STA_HASH(sta)];
 914         while (s != NULL && !ether_addr_equal(s->addr, sta))
 915                 s = s->hnext;
 916         return s;
 917 }
 918
 919
 920 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
 921
 922 /* Called from timer handler and from scheduled AP queue handlers */
 923 static void prism2_send_mgmt(struct net_device *dev,
 924                              u16 type_subtype, char *body,
 925                              int body_len, u8 *addr, u16 tx_cb_idx)
 926 {
 927         struct hostap_interface *iface;
 928         local_info_t *local;
 929         struct ieee80211_hdr *hdr;
 930         u16 fc;
 931         struct sk_buff *skb;
 932         struct hostap_skb_tx_data *meta;
 933         int hdrlen;
 934
 935         iface = netdev_priv(dev);
 936         local = iface->local;
 937         dev = local->dev; /* always use master radio device */
 938         iface = netdev_priv(dev);
 939
 940         if (!(dev->flags & IFF_UP)) {
 941                 PDEBUG(DEBUG_AP, "%s: prism2_send_mgmt - device is not UP - "
 942                        "cannot send frame\n", dev->name);
 943                 return;
 944         }
 945
 946         skb = dev_alloc_skb(sizeof(*hdr) + body_len);
 947         if (skb == NULL) {
 948                 PDEBUG(DEBUG_AP, "%s: prism2_send_mgmt failed to allocate "
 949                        "skb\n", dev->name);
 950                 return;
 951         }
 952
 953         fc = type_subtype;
 954         hdrlen = hostap_80211_get_hdrlen(cpu_to_le16(type_subtype));
 955         hdr = skb_put_zero(skb, hdrlen);
 956         if (body)
 957                 skb_put_data(skb, body, body_len);
 958
 959         /* FIX: ctrl::ack sending used special HFA384X_TX_CTRL_802_11
 960          * tx_control instead of using local->tx_control */
 961
 962
 963         memcpy(hdr->addr1, addr, ETH_ALEN); /* DA / RA */
 964         if (ieee80211_is_data(hdr->frame_control)) {
 965                 fc |= IEEE80211_FCTL_FROMDS;
 966                 memcpy(hdr->addr2, dev->dev_addr, ETH_ALEN); /* BSSID */
 967                 memcpy(hdr->addr3, dev->dev_addr, ETH_ALEN); /* SA */
 968         } else if (ieee80211_is_ctl(hdr->frame_control)) {
 969                 /* control:ACK does not have addr2 or addr3 */
 970                 eth_zero_addr(hdr->addr2);
 971                 eth_zero_addr(hdr->addr3);
 972         } else {
 973                 memcpy(hdr->addr2, dev->dev_addr, ETH_ALEN); /* SA */
 974                 memcpy(hdr->addr3, dev->dev_addr, ETH_ALEN); /* BSSID */
 975         }
 976
 977         hdr->frame_control = cpu_to_le16(fc);
 978
 979         meta = (struct hostap_skb_tx_data *) skb->cb;
 980         memset(meta, 0, sizeof(*meta));
 981         meta->magic = HOSTAP_SKB_TX_DATA_MAGIC;
 982         meta->iface = iface;
 983         meta->tx_cb_idx = tx_cb_idx;
 984
 985         skb->dev = dev;
 986         skb_reset_mac_header(skb);
 987         skb_reset_network_header(skb);
 988         dev_queue_xmit(skb);
 989 }
 990 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
 991
 992 #ifdef CONFIG_PROC_FS
 993 static int prism2_sta_proc_show(struct seq_file *m, void *v)
 994 {
 995         struct sta_info *sta = m->private;
 996         int i;
 997
 998         /* FIX: possible race condition.. the STA data could have just expired,
 999          * but proc entry was still here so that the read could have started;
1000          * some locking should be done here.. */
1001
1002         seq_printf(m,
1003                    "%s=%pM\nusers=%d\naid=%d\n"
1004                    "flags=0x%04x%s%s%s%s%s%s%s\n"
1005                    "capability=0x%02x\nlisten_interval=%d\nsupported_rates=",
1006                    sta->ap ? "AP" : "STA",
1007                    sta->addr, atomic_read(&sta->users), sta->aid,
1008                    sta->flags,
1009                    sta->flags & WLAN_STA_AUTH ? " AUTH" : "",
1010                    sta->flags & WLAN_STA_ASSOC ? " ASSOC" : "",
1011                    sta->flags & WLAN_STA_PS ? " PS" : "",
1012                    sta->flags & WLAN_STA_TIM ? " TIM" : "",
1013                    sta->flags & WLAN_STA_PERM ? " PERM" : "",
1014                    sta->flags & WLAN_STA_AUTHORIZED ? " AUTHORIZED" : "",
1015                    sta->flags & WLAN_STA_PENDING_POLL ? " POLL" : "",
1016                    sta->capability, sta->listen_interval);
1017         /* supported_rates: 500 kbit/s units with msb ignored */
1018         for (i = 0; i < sizeof(sta->supported_rates); i++)
1019                 if (sta->supported_rates[i] != 0)
1020                         seq_printf(m, "%d%sMbps ",
1021                                    (sta->supported_rates[i] & 0x7f) / 2,
1022                                    sta->supported_rates[i] & 1 ? ".5" : "");
1023         seq_printf(m,
1024                    "\njiffies=%lu\nlast_auth=%lu\nlast_assoc=%lu\n"
1025                    "last_rx=%lu\nlast_tx=%lu\nrx_packets=%lu\n"
1026                    "tx_packets=%lu\n"
1027                    "rx_bytes=%lu\ntx_bytes=%lu\nbuffer_count=%d\n"
1028                    "last_rx: silence=%d dBm signal=%d dBm rate=%d%s Mbps\n"
1029                    "tx_rate=%d\ntx[1M]=%d\ntx[2M]=%d\ntx[5.5M]=%d\n"
1030                    "tx[11M]=%d\n"
1031                    "rx[1M]=%d\nrx[2M]=%d\nrx[5.5M]=%d\nrx[11M]=%d\n",
1032                    jiffies, sta->last_auth, sta->last_assoc, sta->last_rx,
1033                    sta->last_tx,
1034                    sta->rx_packets, sta->tx_packets, sta->rx_bytes,
1035                    sta->tx_bytes, skb_queue_len(&sta->tx_buf),
1036                    sta->last_rx_silence,
1037                    sta->last_rx_signal, sta->last_rx_rate / 10,
1038                    sta->last_rx_rate % 10 ? ".5" : "",
1039                    sta->tx_rate, sta->tx_count[0], sta->tx_count[1],
1040                    sta->tx_count[2], sta->tx_count[3],  sta->rx_count[0],
1041                    sta->rx_count[1], sta->rx_count[2], sta->rx_count[3]);
1042         if (sta->crypt && sta->crypt->ops && sta->crypt->ops->print_stats)
1043                 sta->crypt->ops->print_stats(m, sta->crypt->priv);
1044 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
1045         if (sta->ap) {
1046                 if (sta->u.ap.channel >= 0)
1047                         seq_printf(m, "channel=%d\n", sta->u.ap.channel);
1048                 seq_puts(m, "ssid=");
1049                 for (i = 0; i < sta->u.ap.ssid_len; i++) {
1050                         if (sta->u.ap.ssid[i] >= 32 && sta->u.ap.ssid[i] < 127)
1051                                 seq_putc(m, sta->u.ap.ssid[i]);
1052                         else
1053                                 seq_printf(m, "<%02x>", sta->u.ap.ssid[i]);
1054                 }
1055                 seq_putc(m, '\n');
1056         }
1057 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
1058
1059         return 0;
1060 }
1061 #endif
1062
1063 static void handle_add_proc_queue(struct work_struct *work)
1064 {
1065         struct ap_data *ap = container_of(work, struct ap_data,
1066                                           add_sta_proc_queue);
1067         struct sta_info *sta;
1068         char name[20];
1069         struct add_sta_proc_data *entry, *prev;
1070
1071         entry = ap->add_sta_proc_entries;
1072         ap->add_sta_proc_entries = NULL;
1073
1074         while (entry) {
1075                 spin_lock_bh(&ap->sta_table_lock);
1076                 sta = ap_get_sta(ap, entry->addr);
1077                 if (sta)
1078                         atomic_inc(&sta->users);
1079                 spin_unlock_bh(&ap->sta_table_lock);
1080
1081                 if (sta) {
1082                         sprintf(name, "%pM", sta->addr);
1083                         sta->proc = proc_create_single_data(
1084                                 name, 0, ap->proc,
1085                                 prism2_sta_proc_show, sta);
1086
1087                         atomic_dec(&sta->users);
1088                 }
1089
1090                 prev = entry;
1091                 entry = entry->next;
1092                 kfree(prev);
1093         }
1094 }
1095
1096
1097 static struct sta_info * ap_add_sta(struct ap_data *ap, u8 *addr)
1098 {
1099         struct sta_info *sta;
1100
1101         sta = kzalloc(sizeof(struct sta_info), GFP_ATOMIC);
1102         if (sta == NULL) {
1103                 PDEBUG(DEBUG_AP, "AP: kmalloc failed\n");
1104                 return NULL;
1105         }
1106
1107         /* initialize STA info data */
1108         sta->local = ap->local;
1109         skb_queue_head_init(&sta->tx_buf);
1110         memcpy(sta->addr, addr, ETH_ALEN);
1111
1112         atomic_inc(&sta->users);
1113         spin_lock_bh(&ap->sta_table_lock);
1114         list_add(&sta->list, &ap->sta_list);
1115         ap->num_sta++;
1116         ap_sta_hash_add(ap, sta);
1117         spin_unlock_bh(&ap->sta_table_lock);
1118
1119         if (ap->proc) {
1120                 struct add_sta_proc_data *entry;
1121                 /* schedule a non-interrupt context process to add a procfs
1122                  * entry for the STA since procfs code use GFP_KERNEL */
1123                 entry = kmalloc(sizeof(*entry), GFP_ATOMIC);
1124                 if (entry) {
1125                         memcpy(entry->addr, sta->addr, ETH_ALEN);
1126                         entry->next = ap->add_sta_proc_entries;
1127                         ap->add_sta_proc_entries = entry;
1128                         schedule_work(&ap->add_sta_proc_queue);
1129                 } else
1130                         printk(KERN_DEBUG "Failed to add STA proc data\n");
1131         }
1132
1133 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
1134         timer_setup(&sta->timer, ap_handle_timer, 0);
1135         sta->timer.expires = jiffies + ap->max_inactivity;
1136         if (!ap->local->hostapd)
1137                 add_timer(&sta->timer);
1138 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
1139
1140         return sta;
1141 }
1142
1143
1144 static int ap_tx_rate_ok(int rateidx, struct sta_info *sta,
1145                          local_info_t *local)
1146 {
1147         if (rateidx > sta->tx_max_rate ||
1148             !(sta->tx_supp_rates & (1 << rateidx)))
1149                 return 0;
1150
1151         if (local->tx_rate_control != 0 &&
1152             !(local->tx_rate_control & (1 << rateidx)))
1153                 return 0;
1154
1155         return 1;
1156 }
1157
1158
1159 static void prism2_check_tx_rates(struct sta_info *sta)
1160 {
1161         int i;
1162
1163         sta->tx_supp_rates = 0;
1164         for (i = 0; i < sizeof(sta->supported_rates); i++) {
1165                 if ((sta->supported_rates[i] & 0x7f) == 2)
1166                         sta->tx_supp_rates |= WLAN_RATE_1M;
1167                 if ((sta->supported_rates[i] & 0x7f) == 4)
1168                         sta->tx_supp_rates |= WLAN_RATE_2M;
1169                 if ((sta->supported_rates[i] & 0x7f) == 11)
1170                         sta->tx_supp_rates |= WLAN_RATE_5M5;
1171                 if ((sta->supported_rates[i] & 0x7f) == 22)
1172                         sta->tx_supp_rates |= WLAN_RATE_11M;
1173         }
1174         sta->tx_max_rate = sta->tx_rate = sta->tx_rate_idx = 0;
1175         if (sta->tx_supp_rates & WLAN_RATE_1M) {
1176                 sta->tx_max_rate = 0;
1177                 if (ap_tx_rate_ok(0, sta, sta->local)) {
1178                         sta->tx_rate = 10;
1179                         sta->tx_rate_idx = 0;
1180                 }
1181         }
1182         if (sta->tx_supp_rates & WLAN_RATE_2M) {
1183                 sta->tx_max_rate = 1;
1184                 if (ap_tx_rate_ok(1, sta, sta->local)) {
1185                         sta->tx_rate = 20;
1186                         sta->tx_rate_idx = 1;
1187                 }
1188         }
1189         if (sta->tx_supp_rates & WLAN_RATE_5M5) {
1190                 sta->tx_max_rate = 2;
1191                 if (ap_tx_rate_ok(2, sta, sta->local)) {
1192                         sta->tx_rate = 55;
1193                         sta->tx_rate_idx = 2;
1194                 }
1195         }
1196         if (sta->tx_supp_rates & WLAN_RATE_11M) {
1197                 sta->tx_max_rate = 3;
1198                 if (ap_tx_rate_ok(3, sta, sta->local)) {
1199                         sta->tx_rate = 110;
1200                         sta->tx_rate_idx = 3;
1201                 }
1202         }
1203 }
1204
1205
1206 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
1207
1208 static void ap_crypt_init(struct ap_data *ap)
1209 {
1210         ap->crypt = lib80211_get_crypto_ops("WEP");
1211
1212         if (ap->crypt) {
1213                 if (ap->crypt->init) {
1214                         ap->crypt_priv = ap->crypt->init(0);
1215                         if (ap->crypt_priv == NULL)
1216                                 ap->crypt = NULL;
1217                         else {
1218                                 u8 key[WEP_KEY_LEN];
1219                                 get_random_bytes(key, WEP_KEY_LEN);
1220                                 ap->crypt->set_key(key, WEP_KEY_LEN, NULL,
1221                                                    ap->crypt_priv);
1222                         }
1223                 }
1224         }
1225
1226         if (ap->crypt == NULL) {
1227                 printk(KERN_WARNING "AP could not initialize WEP: load module "
1228                        "lib80211_crypt_wep.ko\n");
1229         }
1230 }
1231
1232
1233 /* Generate challenge data for shared key authentication. IEEE 802.11 specifies
1234  * that WEP algorithm is used for generating challenge. This should be unique,
1235  * but otherwise there is not really need for randomness etc. Initialize WEP
1236  * with pseudo random key and then use increasing IV to get unique challenge
1237  * streams.
1238  *
1239  * Called only as a scheduled task for pending AP frames.
1240  */
1241 static char * ap_auth_make_challenge(struct ap_data *ap)
1242 {
1243         char *tmpbuf;
1244         struct sk_buff *skb;
1245
1246         if (ap->crypt == NULL) {
1247                 ap_crypt_init(ap);
1248                 if (ap->crypt == NULL)
1249                         return NULL;
1250         }
1251
1252         tmpbuf = kmalloc(WLAN_AUTH_CHALLENGE_LEN, GFP_ATOMIC);
1253         if (tmpbuf == NULL) {
1254                 PDEBUG(DEBUG_AP, "AP: kmalloc failed for challenge\n");
1255                 return NULL;
1256         }
1257
1258         skb = dev_alloc_skb(WLAN_AUTH_CHALLENGE_LEN +
1259                             ap->crypt->extra_mpdu_prefix_len +
1260                             ap->crypt->extra_mpdu_postfix_len);
1261         if (skb == NULL) {
1262                 kfree(tmpbuf);
1263                 return NULL;
1264         }
1265
1266         skb_reserve(skb, ap->crypt->extra_mpdu_prefix_len);
1267         skb_put_zero(skb, WLAN_AUTH_CHALLENGE_LEN);
1268         if (ap->crypt->encrypt_mpdu(skb, 0, ap->crypt_priv)) {
1269                 dev_kfree_skb(skb);
1270                 kfree(tmpbuf);
1271                 return NULL;
1272         }
1273
1274         skb_copy_from_linear_data_offset(skb, ap->crypt->extra_mpdu_prefix_len,
1275                                          tmpbuf, WLAN_AUTH_CHALLENGE_LEN);
1276         dev_kfree_skb(skb);
1277
1278         return tmpbuf;
1279 }
1280
1281
1282 /* Called only as a scheduled task for pending AP frames. */
1283 static void handle_authen(local_info_t *local, struct sk_buff *skb,
1284                           struct hostap_80211_rx_status *rx_stats)
1285 {
1286         struct net_device *dev = local->dev;
1287         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
1288         size_t hdrlen;
1289         struct ap_data *ap = local->ap;
1290         char body[8 + WLAN_AUTH_CHALLENGE_LEN], *challenge = NULL;
1291         int len, olen;
1292         u16 auth_alg, auth_transaction, status_code;
1293         __le16 *pos;
1294         u16 resp = WLAN_STATUS_SUCCESS;
1295         struct sta_info *sta = NULL;
1296         struct lib80211_crypt_data *crypt;
1297         char *txt = "";
1298
1299         len = skb->len - IEEE80211_MGMT_HDR_LEN;
1300
1301         hdrlen = hostap_80211_get_hdrlen(hdr->frame_control);
1302
1303         if (len < 6) {
1304                 PDEBUG(DEBUG_AP, "%s: handle_authen - too short payload "
1305                        "(len=%d) from %pM\n", dev->name, len, hdr->addr2);
1306                 return;
1307         }
1308
1309         spin_lock_bh(&local->ap->sta_table_lock);
1310         sta = ap_get_sta(local->ap, hdr->addr2);
1311         if (sta)
1312                 atomic_inc(&sta->users);
1313         spin_unlock_bh(&local->ap->sta_table_lock);
1314
1315         if (sta && sta->crypt)
1316                 crypt = sta->crypt;
1317         else {
1318                 int idx = 0;
1319                 if (skb->len >= hdrlen + 3)
1320                         idx = skb->data[hdrlen + 3] >> 6;
1321                 crypt = local->crypt_info.crypt[idx];
1322         }
1323
1324         pos = (__le16 *) (skb->data + IEEE80211_MGMT_HDR_LEN);
1325         auth_alg = __le16_to_cpu(*pos);
1326         pos++;
1327         auth_transaction = __le16_to_cpu(*pos);
1328         pos++;
1329         status_code = __le16_to_cpu(*pos);
1330         pos++;
1331
1332         if (ether_addr_equal(dev->dev_addr, hdr->addr2) ||
1333             ap_control_mac_deny(&ap->mac_restrictions, hdr->addr2)) {
1334                 txt = "authentication denied";
1335                 resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1336                 goto fail;
1337         }
1338
1339         if (((local->auth_algs & PRISM2_AUTH_OPEN) &&
1340              auth_alg == WLAN_AUTH_OPEN) ||
1341             ((local->auth_algs & PRISM2_AUTH_SHARED_KEY) &&
1342              crypt && auth_alg == WLAN_AUTH_SHARED_KEY)) {
1343         } else {
1344                 txt = "unsupported algorithm";
1345                 resp = WLAN_STATUS_NOT_SUPPORTED_AUTH_ALG;
1346                 goto fail;
1347         }
1348
1349         if (len >= 8) {
1350                 u8 *u = (u8 *) pos;
1351                 if (*u == WLAN_EID_CHALLENGE) {
1352                         if (*(u + 1) != WLAN_AUTH_CHALLENGE_LEN) {
1353                                 txt = "invalid challenge len";
1354                                 resp = WLAN_STATUS_CHALLENGE_FAIL;
1355                                 goto fail;
1356                         }
1357                         if (len - 8 < WLAN_AUTH_CHALLENGE_LEN) {
1358                                 txt = "challenge underflow";
1359                                 resp = WLAN_STATUS_CHALLENGE_FAIL;
1360                                 goto fail;
1361                         }
1362                         challenge = (char *) (u + 2);
1363                 }
1364         }
1365
1366         if (sta && sta->ap) {
1367                 if (time_after(jiffies, sta->u.ap.last_beacon +
1368                                (10 * sta->listen_interval * HZ) / 1024)) {
1369                         PDEBUG(DEBUG_AP, "%s: no beacons received for a while,"
1370                                " assuming AP %pM is now STA\n",
1371                                dev->name, sta->addr);
1372                         sta->ap = 0;
1373                         sta->flags = 0;
1374                         sta->u.sta.challenge = NULL;
1375                 } else {
1376                         txt = "AP trying to authenticate?";
1377                         resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1378                         goto fail;
1379                 }
1380         }
1381
1382         if ((auth_alg == WLAN_AUTH_OPEN && auth_transaction == 1) ||
1383             (auth_alg == WLAN_AUTH_SHARED_KEY &&
1384              (auth_transaction == 1 ||
1385               (auth_transaction == 3 && sta != NULL &&
1386                sta->u.sta.challenge != NULL)))) {
1387         } else {
1388                 txt = "unknown authentication transaction number";
1389                 resp = WLAN_STATUS_UNKNOWN_AUTH_TRANSACTION;
1390                 goto fail;
1391         }
1392
1393         if (sta == NULL) {
1394                 txt = "new STA";
1395
1396                 if (local->ap->num_sta >= MAX_STA_COUNT) {
1397                         /* FIX: might try to remove some old STAs first? */
1398                         txt = "no more room for new STAs";
1399                         resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1400                         goto fail;
1401                 }
1402
1403                 sta = ap_add_sta(local->ap, hdr->addr2);
1404                 if (sta == NULL) {
1405                         txt = "ap_add_sta failed";
1406                         resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1407                         goto fail;
1408                 }
1409         }
1410
1411         switch (auth_alg) {
1412         case WLAN_AUTH_OPEN:
1413                 txt = "authOK";
1414                 /* IEEE 802.11 standard is not completely clear about
1415                  * whether STA is considered authenticated after
1416                  * authentication OK frame has been send or after it
1417                  * has been ACKed. In order to reduce interoperability
1418                  * issues, mark the STA authenticated before ACK. */
1419                 sta->flags |= WLAN_STA_AUTH;
1420                 break;
1421
1422         case WLAN_AUTH_SHARED_KEY:
1423                 if (auth_transaction == 1) {
1424                         if (sta->u.sta.challenge == NULL) {
1425                                 sta->u.sta.challenge =
1426                                         ap_auth_make_challenge(local->ap);
1427                                 if (sta->u.sta.challenge == NULL) {
1428                                         resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1429                                         goto fail;
1430                                 }
1431                         }
1432                 } else {
1433                         if (sta->u.sta.challenge == NULL ||
1434                             challenge == NULL ||
1435                             memcmp(sta->u.sta.challenge, challenge,
1436                                    WLAN_AUTH_CHALLENGE_LEN) != 0 ||
1437                             !ieee80211_has_protected(hdr->frame_control)) {
1438                                 txt = "challenge response incorrect";
1439                                 resp = WLAN_STATUS_CHALLENGE_FAIL;
1440                                 goto fail;
1441                         }
1442
1443                         txt = "challenge OK - authOK";
1444                         /* IEEE 802.11 standard is not completely clear about
1445                          * whether STA is considered authenticated after
1446                          * authentication OK frame has been send or after it
1447                          * has been ACKed. In order to reduce interoperability
1448                          * issues, mark the STA authenticated before ACK. */
1449                         sta->flags |= WLAN_STA_AUTH;
1450                         kfree(sta->u.sta.challenge);
1451                         sta->u.sta.challenge = NULL;
1452                 }
1453                 break;
1454         }
1455
1456  fail:
1457         pos = (__le16 *) body;
1458         *pos = cpu_to_le16(auth_alg);
1459         pos++;
1460         *pos = cpu_to_le16(auth_transaction + 1);
1461         pos++;
1462         *pos = cpu_to_le16(resp); /* status_code */
1463         pos++;
1464         olen = 6;
1465
1466         if (resp == WLAN_STATUS_SUCCESS && sta != NULL &&
1467             sta->u.sta.challenge != NULL &&
1468             auth_alg == WLAN_AUTH_SHARED_KEY && auth_transaction == 1) {
1469                 u8 *tmp = (u8 *) pos;
1470                 *tmp++ = WLAN_EID_CHALLENGE;
1471                 *tmp++ = WLAN_AUTH_CHALLENGE_LEN;
1472                 pos++;
1473                 memcpy(pos, sta->u.sta.challenge, WLAN_AUTH_CHALLENGE_LEN);
1474                 olen += 2 + WLAN_AUTH_CHALLENGE_LEN;
1475         }
1476
1477         prism2_send_mgmt(dev, IEEE80211_FTYPE_MGMT | IEEE80211_STYPE_AUTH,
1478                          body, olen, hdr->addr2, ap->tx_callback_auth);
1479
1480         if (sta) {
1481                 sta->last_rx = jiffies;
1482                 atomic_dec(&sta->users);
1483         }
1484
1485         if (resp) {
1486                 PDEBUG(DEBUG_AP, "%s: %pM auth (alg=%d "
1487                        "trans#=%d stat=%d len=%d fc=%04x) ==> %d (%s)\n",
1488                        dev->name, hdr->addr2,
1489                        auth_alg, auth_transaction, status_code, len,
1490                        le16_to_cpu(hdr->frame_control), resp, txt);
1491         }
1492 }
1493
1494
1495 /* Called only as a scheduled task for pending AP frames. */
1496 static void handle_assoc(local_info_t *local, struct sk_buff *skb,
1497                          struct hostap_80211_rx_status *rx_stats, int reassoc)
1498 {
1499         struct net_device *dev = local->dev;
1500         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
1501         char body[12], *p, *lpos;
1502         int len, left;
1503         __le16 *pos;
1504         u16 resp = WLAN_STATUS_SUCCESS;
1505         struct sta_info *sta = NULL;
1506         int send_deauth = 0;
1507         char *txt = "";
1508         u8 prev_ap[ETH_ALEN];
1509
1510         left = len = skb->len - IEEE80211_MGMT_HDR_LEN;
1511
1512         if (len < (reassoc ? 10 : 4)) {
1513                 PDEBUG(DEBUG_AP, "%s: handle_assoc - too short payload "
1514                        "(len=%d, reassoc=%d) from %pM\n",
1515                        dev->name, len, reassoc, hdr->addr2);
1516                 return;
1517         }
1518
1519         spin_lock_bh(&local->ap->sta_table_lock);
1520         sta = ap_get_sta(local->ap, hdr->addr2);
1521         if (sta == NULL || (sta->flags & WLAN_STA_AUTH) == 0) {
1522                 spin_unlock_bh(&local->ap->sta_table_lock);
1523                 txt = "trying to associate before authentication";
1524                 send_deauth = 1;
1525                 resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1526                 sta = NULL; /* do not decrement sta->users */
1527                 goto fail;
1528         }
1529         atomic_inc(&sta->users);
1530         spin_unlock_bh(&local->ap->sta_table_lock);
1531
1532         pos = (__le16 *) (skb->data + IEEE80211_MGMT_HDR_LEN);
1533         sta->capability = __le16_to_cpu(*pos);
1534         pos++; left -= 2;
1535         sta->listen_interval = __le16_to_cpu(*pos);
1536         pos++; left -= 2;
1537
1538         if (reassoc) {
1539                 memcpy(prev_ap, pos, ETH_ALEN);
1540                 pos++; pos++; pos++; left -= 6;
1541         } else
1542                 eth_zero_addr(prev_ap);
1543
1544         if (left >= 2) {
1545                 unsigned int ileft;
1546                 unsigned char *u = (unsigned char *) pos;
1547
1548                 if (*u == WLAN_EID_SSID) {
1549                         u++; left--;
1550                         ileft = *u;
1551                         u++; left--;
1552
1553                         if (ileft > left || ileft > MAX_SSID_LEN) {
1554                                 txt = "SSID overflow";
1555                                 resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1556                                 goto fail;
1557                         }
1558
1559                         if (ileft != strlen(local->essid) ||
1560                             memcmp(local->essid, u, ileft) != 0) {
1561                                 txt = "not our SSID";
1562                                 resp = WLAN_STATUS_ASSOC_DENIED_UNSPEC;
1563                                 goto fail;
1564                         }
1565
1566                         u += ileft;
1567                         left -= ileft;
1568                 }
1569
1570                 if (left >= 2 && *u == WLAN_EID_SUPP_RATES) {
1571                         u++; left--;
1572                         ileft = *u;
1573                         u++; left--;
1574
1575                         if (ileft > left || ileft == 0 ||
1576                             ileft > WLAN_SUPP_RATES_MAX) {
1577                                 txt = "SUPP_RATES len error";
1578                                 resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1579                                 goto fail;
1580                         }
1581
1582                         memset(sta->supported_rates, 0,
1583                                sizeof(sta->supported_rates));
1584                         memcpy(sta->supported_rates, u, ileft);
1585                         prism2_check_tx_rates(sta);
1586
1587                         u += ileft;
1588                         left -= ileft;
1589                 }
1590
1591                 if (left > 0) {
1592                         PDEBUG(DEBUG_AP, "%s: assoc from %pM"
1593                                " with extra data (%d bytes) [",
1594                                dev->name, hdr->addr2, left);
1595                         while (left > 0) {
1596                                 PDEBUG2(DEBUG_AP, "<%02x>", *u);
1597                                 u++; left--;
1598                         }
1599                         PDEBUG2(DEBUG_AP, "]\n");
1600                 }
1601         } else {
1602                 txt = "frame underflow";
1603                 resp = WLAN_STATUS_UNSPECIFIED_FAILURE;
1604                 goto fail;
1605         }
1606
1607         /* get a unique AID */
1608         if (sta->aid > 0)
1609                 txt = "OK, old AID";
1610         else {
1611                 spin_lock_bh(&local->ap->sta_table_lock);
1612                 for (sta->aid = 1; sta->aid <= MAX_AID_TABLE_SIZE; sta->aid++)
1613                         if (local->ap->sta_aid[sta->aid - 1] == NULL)
1614                                 break;
1615                 if (sta->aid > MAX_AID_TABLE_SIZE) {
1616                         sta->aid = 0;
1617                         spin_unlock_bh(&local->ap->sta_table_lock);
1618                         resp = WLAN_STATUS_AP_UNABLE_TO_HANDLE_NEW_STA;
1619                         txt = "no room for more AIDs";
1620                 } else {
1621                         local->ap->sta_aid[sta->aid - 1] = sta;
1622                         spin_unlock_bh(&local->ap->sta_table_lock);
1623                         txt = "OK, new AID";
1624                 }
1625         }
1626
1627  fail:
1628         pos = (__le16 *) body;
1629
1630         if (send_deauth) {
1631                 *pos = cpu_to_le16(WLAN_REASON_STA_REQ_ASSOC_WITHOUT_AUTH);
1632                 pos++;
1633         } else {
1634                 /* FIX: CF-Pollable and CF-PollReq should be set to match the
1635                  * values in beacons/probe responses */
1636                 /* FIX: how about privacy and WEP? */
1637                 /* capability */
1638                 *pos = cpu_to_le16(WLAN_CAPABILITY_ESS);
1639                 pos++;
1640
1641                 /* status_code */
1642                 *pos = cpu_to_le16(resp);
1643                 pos++;
1644
1645                 *pos = cpu_to_le16((sta && sta->aid > 0 ? sta->aid : 0) |
1646                                      BIT(14) | BIT(15)); /* AID */
1647                 pos++;
1648
1649                 /* Supported rates (Information element) */
1650                 p = (char *) pos;
1651                 *p++ = WLAN_EID_SUPP_RATES;
1652                 lpos = p;
1653                 *p++ = 0; /* len */
1654                 if (local->tx_rate_control & WLAN_RATE_1M) {
1655                         *p++ = local->basic_rates & WLAN_RATE_1M ? 0x82 : 0x02;
1656                         (*lpos)++;
1657                 }
1658                 if (local->tx_rate_control & WLAN_RATE_2M) {
1659                         *p++ = local->basic_rates & WLAN_RATE_2M ? 0x84 : 0x04;
1660                         (*lpos)++;
1661                 }
1662                 if (local->tx_rate_control & WLAN_RATE_5M5) {
1663                         *p++ = local->basic_rates & WLAN_RATE_5M5 ?
1664                                 0x8b : 0x0b;
1665                         (*lpos)++;
1666                 }
1667                 if (local->tx_rate_control & WLAN_RATE_11M) {
1668                         *p++ = local->basic_rates & WLAN_RATE_11M ?
1669                                 0x96 : 0x16;
1670                         (*lpos)++;
1671                 }
1672                 pos = (__le16 *) p;
1673         }
1674
1675         prism2_send_mgmt(dev, IEEE80211_FTYPE_MGMT |
1676                          (send_deauth ? IEEE80211_STYPE_DEAUTH :
1677                           (reassoc ? IEEE80211_STYPE_REASSOC_RESP :
1678                            IEEE80211_STYPE_ASSOC_RESP)),
1679                          body, (u8 *) pos - (u8 *) body,
1680                          hdr->addr2,
1681                          send_deauth ? 0 : local->ap->tx_callback_assoc);
1682
1683         if (sta) {
1684                 if (resp == WLAN_STATUS_SUCCESS) {
1685                         sta->last_rx = jiffies;
1686                         /* STA will be marked associated from TX callback, if
1687                          * AssocResp is ACKed */
1688                 }
1689                 atomic_dec(&sta->users);
1690         }
1691
1692 #if 0
1693         PDEBUG(DEBUG_AP, "%s: %pM %sassoc (len=%d "
1694                "prev_ap=%pM) => %d(%d) (%s)\n",
1695                dev->name,
1696                hdr->addr2,
1697                reassoc ? "re" : "", len,
1698                prev_ap,
1699                resp, send_deauth, txt);
1700 #endif
1701 }
1702
1703
1704 /* Called only as a scheduled task for pending AP frames. */
1705 static void handle_deauth(local_info_t *local, struct sk_buff *skb,
1706                           struct hostap_80211_rx_status *rx_stats)
1707 {
1708         struct net_device *dev = local->dev;
1709         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
1710         char *body = (char *) (skb->data + IEEE80211_MGMT_HDR_LEN);
1711         int len;
1712         u16 reason_code;
1713         __le16 *pos;
1714         struct sta_info *sta = NULL;
1715
1716         len = skb->len - IEEE80211_MGMT_HDR_LEN;
1717
1718         if (len < 2) {
1719                 printk("handle_deauth - too short payload (len=%d)\n", len);
1720                 return;
1721         }
1722
1723         pos = (__le16 *) body;
1724         reason_code = le16_to_cpu(*pos);
1725
1726         PDEBUG(DEBUG_AP, "%s: deauthentication: %pM len=%d, "
1727                "reason_code=%d\n", dev->name, hdr->addr2,
1728                len, reason_code);
1729
1730         spin_lock_bh(&local->ap->sta_table_lock);
1731         sta = ap_get_sta(local->ap, hdr->addr2);
1732         if (sta != NULL) {
1733                 if ((sta->flags & WLAN_STA_ASSOC) && !sta->ap)
1734                         hostap_event_expired_sta(local->dev, sta);
1735                 sta->flags &= ~(WLAN_STA_AUTH | WLAN_STA_ASSOC);
1736         }
1737         spin_unlock_bh(&local->ap->sta_table_lock);
1738         if (sta == NULL) {
1739                 printk("%s: deauthentication from %pM, "
1740                "reason_code=%d, but STA not authenticated\n", dev->name,
1741                        hdr->addr2, reason_code);
1742         }
1743 }
1744
1745
1746 /* Called only as a scheduled task for pending AP frames. */
1747 static void handle_disassoc(local_info_t *local, struct sk_buff *skb,
1748                             struct hostap_80211_rx_status *rx_stats)
1749 {
1750         struct net_device *dev = local->dev;
1751         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
1752         char *body = skb->data + IEEE80211_MGMT_HDR_LEN;
1753         int len;
1754         u16 reason_code;
1755         __le16 *pos;
1756         struct sta_info *sta = NULL;
1757
1758         len = skb->len - IEEE80211_MGMT_HDR_LEN;
1759
1760         if (len < 2) {
1761                 printk("handle_disassoc - too short payload (len=%d)\n", len);
1762                 return;
1763         }
1764
1765         pos = (__le16 *) body;
1766         reason_code = le16_to_cpu(*pos);
1767
1768         PDEBUG(DEBUG_AP, "%s: disassociation: %pM len=%d, "
1769                "reason_code=%d\n", dev->name, hdr->addr2,
1770                len, reason_code);
1771
1772         spin_lock_bh(&local->ap->sta_table_lock);
1773         sta = ap_get_sta(local->ap, hdr->addr2);
1774         if (sta != NULL) {
1775                 if ((sta->flags & WLAN_STA_ASSOC) && !sta->ap)
1776                         hostap_event_expired_sta(local->dev, sta);
1777                 sta->flags &= ~WLAN_STA_ASSOC;
1778         }
1779         spin_unlock_bh(&local->ap->sta_table_lock);
1780         if (sta == NULL) {
1781                 printk("%s: disassociation from %pM, "
1782                        "reason_code=%d, but STA not authenticated\n",
1783                        dev->name, hdr->addr2, reason_code);
1784         }
1785 }
1786
1787
1788 /* Called only as a scheduled task for pending AP frames. */
1789 static void ap_handle_data_nullfunc(local_info_t *local,
1790                                     struct ieee80211_hdr *hdr)
1791 {
1792         struct net_device *dev = local->dev;
1793
1794         /* some STA f/w's seem to require control::ACK frame for
1795          * data::nullfunc, but at least Prism2 station f/w version 0.8.0 does
1796          * not send this..
1797          * send control::ACK for the data::nullfunc */
1798
1799         printk(KERN_DEBUG "Sending control::ACK for data::nullfunc\n");
1800         prism2_send_mgmt(dev, IEEE80211_FTYPE_CTL | IEEE80211_STYPE_ACK,
1801                          NULL, 0, hdr->addr2, 0);
1802 }
1803
1804
1805 /* Called only as a scheduled task for pending AP frames. */
1806 static void ap_handle_dropped_data(local_info_t *local,
1807                                    struct ieee80211_hdr *hdr)
1808 {
1809         struct net_device *dev = local->dev;
1810         struct sta_info *sta;
1811         __le16 reason;
1812
1813         spin_lock_bh(&local->ap->sta_table_lock);
1814         sta = ap_get_sta(local->ap, hdr->addr2);
1815         if (sta)
1816                 atomic_inc(&sta->users);
1817         spin_unlock_bh(&local->ap->sta_table_lock);
1818
1819         if (sta != NULL && (sta->flags & WLAN_STA_ASSOC)) {
1820                 PDEBUG(DEBUG_AP, "ap_handle_dropped_data: STA is now okay?\n");
1821                 atomic_dec(&sta->users);
1822                 return;
1823         }
1824
1825         reason = cpu_to_le16(WLAN_REASON_CLASS3_FRAME_FROM_NONASSOC_STA);
1826         prism2_send_mgmt(dev, IEEE80211_FTYPE_MGMT |
1827                          ((sta == NULL || !(sta->flags & WLAN_STA_ASSOC)) ?
1828                           IEEE80211_STYPE_DEAUTH : IEEE80211_STYPE_DISASSOC),
1829                          (char *) &reason, sizeof(reason), hdr->addr2, 0);
1830
1831         if (sta)
1832                 atomic_dec(&sta->users);
1833 }
1834
1835 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
1836
1837
1838 /* Called only as a scheduled task for pending AP frames. */
1839 static void pspoll_send_buffered(local_info_t *local, struct sta_info *sta,
1840                                  struct sk_buff *skb)
1841 {
1842         struct hostap_skb_tx_data *meta;
1843
1844         if (!(sta->flags & WLAN_STA_PS)) {
1845                 /* Station has moved to non-PS mode, so send all buffered
1846                  * frames using normal device queue. */
1847                 dev_queue_xmit(skb);
1848                 return;
1849         }
1850
1851         /* add a flag for hostap_handle_sta_tx() to know that this skb should
1852          * be passed through even though STA is using PS */
1853         meta = (struct hostap_skb_tx_data *) skb->cb;
1854         meta->flags |= HOSTAP_TX_FLAGS_BUFFERED_FRAME;
1855         if (!skb_queue_empty(&sta->tx_buf)) {
1856                 /* indicate to STA that more frames follow */
1857                 meta->flags |= HOSTAP_TX_FLAGS_ADD_MOREDATA;
1858         }
1859         dev_queue_xmit(skb);
1860 }
1861
1862
1863 /* Called only as a scheduled task for pending AP frames. */
1864 static void handle_pspoll(local_info_t *local,
1865                           struct ieee80211_hdr *hdr,
1866                           struct hostap_80211_rx_status *rx_stats)
1867 {
1868         struct net_device *dev = local->dev;
1869         struct sta_info *sta;
1870         u16 aid;
1871         struct sk_buff *skb;
1872
1873         PDEBUG(DEBUG_PS2, "handle_pspoll: BSSID=%pM, TA=%pM PWRMGT=%d\n",
1874                hdr->addr1, hdr->addr2, !!ieee80211_has_pm(hdr->frame_control));
1875
1876         if (!ether_addr_equal(hdr->addr1, dev->dev_addr)) {
1877                 PDEBUG(DEBUG_AP,
1878                        "handle_pspoll - addr1(BSSID)=%pM not own MAC\n",
1879                        hdr->addr1);
1880                 return;
1881         }
1882
1883         aid = le16_to_cpu(hdr->duration_id);
1884         if ((aid & (BIT(15) | BIT(14))) != (BIT(15) | BIT(14))) {
1885                 PDEBUG(DEBUG_PS, "   PSPOLL and AID[15:14] not set\n");
1886                 return;
1887         }
1888         aid &= ~(BIT(15) | BIT(14));
1889         if (aid == 0 || aid > MAX_AID_TABLE_SIZE) {
1890                 PDEBUG(DEBUG_PS, "   invalid aid=%d\n", aid);
1891                 return;
1892         }
1893         PDEBUG(DEBUG_PS2, "   aid=%d\n", aid);
1894
1895         spin_lock_bh(&local->ap->sta_table_lock);
1896         sta = ap_get_sta(local->ap, hdr->addr2);
1897         if (sta)
1898                 atomic_inc(&sta->users);
1899         spin_unlock_bh(&local->ap->sta_table_lock);
1900
1901         if (sta == NULL) {
1902                 PDEBUG(DEBUG_PS, "   STA not found\n");
1903                 return;
1904         }
1905         if (sta->aid != aid) {
1906                 PDEBUG(DEBUG_PS, "   received aid=%i does not match with "
1907                        "assoc.aid=%d\n", aid, sta->aid);
1908                 return;
1909         }
1910
1911         /* FIX: todo:
1912          * - add timeout for buffering (clear aid in TIM vector if buffer timed
1913          *   out (expiry time must be longer than ListenInterval for
1914          *   the corresponding STA; "8802-11: 11.2.1.9 AP aging function"
1915          * - what to do, if buffered, pspolled, and sent frame is not ACKed by
1916          *   sta; store buffer for later use and leave TIM aid bit set? use
1917          *   TX event to check whether frame was ACKed?
1918          */
1919
1920         while ((skb = skb_dequeue(&sta->tx_buf)) != NULL) {
1921                 /* send buffered frame .. */
1922                 PDEBUG(DEBUG_PS2, "Sending buffered frame to STA after PS POLL"
1923                        " (buffer_count=%d)\n", skb_queue_len(&sta->tx_buf));
1924
1925                 pspoll_send_buffered(local, sta, skb);
1926
1927                 if (sta->flags & WLAN_STA_PS) {
1928                         /* send only one buffered packet per PS Poll */
1929                         /* FIX: should ignore further PS Polls until the
1930                          * buffered packet that was just sent is acknowledged
1931                          * (Tx or TxExc event) */
1932                         break;
1933                 }
1934         }
1935
1936         if (skb_queue_empty(&sta->tx_buf)) {
1937                 /* try to clear aid from TIM */
1938                 if (!(sta->flags & WLAN_STA_TIM))
1939                         PDEBUG(DEBUG_PS2,  "Re-unsetting TIM for aid %d\n",
1940                                aid);
1941                 hostap_set_tim(local, aid, 0);
1942                 sta->flags &= ~WLAN_STA_TIM;
1943         }
1944
1945         atomic_dec(&sta->users);
1946 }
1947
1948
1949 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
1950
1951 static void handle_wds_oper_queue(struct work_struct *work)
1952 {
1953         struct ap_data *ap = container_of(work, struct ap_data,
1954                                           wds_oper_queue);
1955         local_info_t *local = ap->local;
1956         struct wds_oper_data *entry, *prev;
1957
1958         spin_lock_bh(&local->lock);
1959         entry = local->ap->wds_oper_entries;
1960         local->ap->wds_oper_entries = NULL;
1961         spin_unlock_bh(&local->lock);
1962
1963         while (entry) {
1964                 PDEBUG(DEBUG_AP, "%s: %s automatic WDS connection "
1965                        "to AP %pM\n",
1966                        local->dev->name,
1967                        entry->type == WDS_ADD ? "adding" : "removing",
1968                        entry->addr);
1969                 if (entry->type == WDS_ADD)
1970                         prism2_wds_add(local, entry->addr, 0);
1971                 else if (entry->type == WDS_DEL)
1972                         prism2_wds_del(local, entry->addr, 0, 1);
1973
1974                 prev = entry;
1975                 entry = entry->next;
1976                 kfree(prev);
1977         }
1978 }
1979
1980
1981 /* Called only as a scheduled task for pending AP frames. */
1982 static void handle_beacon(local_info_t *local, struct sk_buff *skb,
1983                           struct hostap_80211_rx_status *rx_stats)
1984 {
1985         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data;
1986         char *body = skb->data + IEEE80211_MGMT_HDR_LEN;
1987         int len, left;
1988         u16 beacon_int, capability;
1989         __le16 *pos;
1990         char *ssid = NULL;
1991         unsigned char *supp_rates = NULL;
1992         int ssid_len = 0, supp_rates_len = 0;
1993         struct sta_info *sta = NULL;
1994         int new_sta = 0, channel = -1;
1995
1996         len = skb->len - IEEE80211_MGMT_HDR_LEN;
1997
1998         if (len < 8 + 2 + 2) {
1999                 printk(KERN_DEBUG "handle_beacon - too short payload "
2000                        "(len=%d)\n", len);
2001                 return;
2002         }
2003
2004         pos = (__le16 *) body;
2005         left = len;
2006
2007         /* Timestamp (8 octets) */
2008         pos += 4; left -= 8;
2009         /* Beacon interval (2 octets) */
2010         beacon_int = le16_to_cpu(*pos);
2011         pos++; left -= 2;
2012         /* Capability information (2 octets) */
2013         capability = le16_to_cpu(*pos);
2014         pos++; left -= 2;
2015
2016         if (local->ap->ap_policy != AP_OTHER_AP_EVEN_IBSS &&
2017             capability & WLAN_CAPABILITY_IBSS)
2018                 return;
2019
2020         if (left >= 2) {
2021                 unsigned int ileft;
2022                 unsigned char *u = (unsigned char *) pos;
2023
2024                 if (*u == WLAN_EID_SSID) {
2025                         u++; left--;
2026                         ileft = *u;
2027                         u++; left--;
2028
2029                         if (ileft > left || ileft > MAX_SSID_LEN) {
2030                                 PDEBUG(DEBUG_AP, "SSID: overflow\n");
2031                                 return;
2032                         }
2033
2034                         if (local->ap->ap_policy == AP_OTHER_AP_SAME_SSID &&
2035                             (ileft != strlen(local->essid) ||
2036                              memcmp(local->essid, u, ileft) != 0)) {
2037                                 /* not our SSID */
2038                                 return;
2039                         }
2040
2041                         ssid = u;
2042                         ssid_len = ileft;
2043
2044                         u += ileft;
2045                         left -= ileft;
2046                 }
2047
2048                 if (*u == WLAN_EID_SUPP_RATES) {
2049                         u++; left--;
2050                         ileft = *u;
2051                         u++; left--;
2052
2053                         if (ileft > left || ileft == 0 || ileft > 8) {
2054                                 PDEBUG(DEBUG_AP, " - SUPP_RATES len error\n");
2055                                 return;
2056                         }
2057
2058                         supp_rates = u;
2059                         supp_rates_len = ileft;
2060
2061                         u += ileft;
2062                         left -= ileft;
2063                 }
2064
2065                 if (*u == WLAN_EID_DS_PARAMS) {
2066                         u++; left--;
2067                         ileft = *u;
2068                         u++; left--;
2069
2070                         if (ileft > left || ileft != 1) {
2071                                 PDEBUG(DEBUG_AP, " - DS_PARAMS len error\n");
2072                                 return;
2073                         }
2074
2075                         channel = *u;
2076
2077                         u += ileft;
2078                         left -= ileft;
2079                 }
2080         }
2081
2082         spin_lock_bh(&local->ap->sta_table_lock);
2083         sta = ap_get_sta(local->ap, hdr->addr2);
2084         if (sta != NULL)
2085                 atomic_inc(&sta->users);
2086         spin_unlock_bh(&local->ap->sta_table_lock);
2087
2088         if (sta == NULL) {
2089                 /* add new AP */
2090                 new_sta = 1;
2091                 sta = ap_add_sta(local->ap, hdr->addr2);
2092                 if (sta == NULL) {
2093                         printk(KERN_INFO "prism2: kmalloc failed for AP "
2094                                "data structure\n");
2095                         return;
2096                 }
2097                 hostap_event_new_sta(local->dev, sta);
2098
2099                 /* mark APs authentication and associated for pseudo ad-hoc
2100                  * style communication */
2101                 sta->flags = WLAN_STA_AUTH | WLAN_STA_ASSOC;
2102
2103                 if (local->ap->autom_ap_wds) {
2104                         hostap_wds_link_oper(local, sta->addr, WDS_ADD);
2105                 }
2106         }
2107
2108         sta->ap = 1;
2109         if (ssid) {
2110                 sta->u.ap.ssid_len = ssid_len;
2111                 memcpy(sta->u.ap.ssid, ssid, ssid_len);
2112                 sta->u.ap.ssid[ssid_len] = '\0';
2113         } else {
2114                 sta->u.ap.ssid_len = 0;
2115                 sta->u.ap.ssid[0] = '\0';
2116         }
2117         sta->u.ap.channel = channel;
2118         sta->rx_packets++;
2119         sta->rx_bytes += len;
2120         sta->u.ap.last_beacon = sta->last_rx = jiffies;
2121         sta->capability = capability;
2122         sta->listen_interval = beacon_int;
2123
2124         atomic_dec(&sta->users);
2125
2126         if (new_sta) {
2127                 memset(sta->supported_rates, 0, sizeof(sta->supported_rates));
2128                 memcpy(sta->supported_rates, supp_rates, supp_rates_len);
2129                 prism2_check_tx_rates(sta);
2130         }
2131 }
2132
2133 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
2134
2135
2136 /* Called only as a tasklet. */
2137 static void handle_ap_item(local_info_t *local, struct sk_buff *skb,
2138                            struct hostap_80211_rx_status *rx_stats)
2139 {
2140 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
2141         struct net_device *dev = local->dev;
2142 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
2143         u16 fc, type, stype;
2144         struct ieee80211_hdr *hdr;
2145
2146         /* FIX: should give skb->len to handler functions and check that the
2147          * buffer is long enough */
2148         hdr = (struct ieee80211_hdr *) skb->data;
2149         fc = le16_to_cpu(hdr->frame_control);
2150         type = fc & IEEE80211_FCTL_FTYPE;
2151         stype = fc & IEEE80211_FCTL_STYPE;
2152
2153 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
2154         if (!local->hostapd && type == IEEE80211_FTYPE_DATA) {
2155                 PDEBUG(DEBUG_AP, "handle_ap_item - data frame\n");
2156
2157                 if (!(fc & IEEE80211_FCTL_TODS) ||
2158                     (fc & IEEE80211_FCTL_FROMDS)) {
2159                         if (stype == IEEE80211_STYPE_NULLFUNC) {
2160                                 /* no ToDS nullfunc seems to be used to check
2161                                  * AP association; so send reject message to
2162                                  * speed up re-association */
2163                                 ap_handle_dropped_data(local, hdr);
2164                                 goto done;
2165                         }
2166                         PDEBUG(DEBUG_AP, "   not ToDS frame (fc=0x%04x)\n",
2167                                fc);
2168                         goto done;
2169                 }
2170
2171                 if (!ether_addr_equal(hdr->addr1, dev->dev_addr)) {
2172                         PDEBUG(DEBUG_AP, "handle_ap_item - addr1(BSSID)=%pM"
2173                                " not own MAC\n", hdr->addr1);
2174                         goto done;
2175                 }
2176
2177                 if (local->ap->nullfunc_ack &&
2178                     stype == IEEE80211_STYPE_NULLFUNC)
2179                         ap_handle_data_nullfunc(local, hdr);
2180                 else
2181                         ap_handle_dropped_data(local, hdr);
2182                 goto done;
2183         }
2184
2185         if (type == IEEE80211_FTYPE_MGMT && stype == IEEE80211_STYPE_BEACON) {
2186                 handle_beacon(local, skb, rx_stats);
2187                 goto done;
2188         }
2189 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
2190
2191         if (type == IEEE80211_FTYPE_CTL && stype == IEEE80211_STYPE_PSPOLL) {
2192                 handle_pspoll(local, hdr, rx_stats);
2193                 goto done;
2194         }
2195
2196         if (local->hostapd) {
2197                 PDEBUG(DEBUG_AP, "Unknown frame in AP queue: type=0x%02x "
2198                        "subtype=0x%02x\n", type, stype);
2199                 goto done;
2200         }
2201
2202 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
2203         if (type != IEEE80211_FTYPE_MGMT) {
2204                 PDEBUG(DEBUG_AP, "handle_ap_item - not a management frame?\n");
2205                 goto done;
2206         }
2207
2208         if (!ether_addr_equal(hdr->addr1, dev->dev_addr)) {
2209                 PDEBUG(DEBUG_AP, "handle_ap_item - addr1(DA)=%pM"
2210                        " not own MAC\n", hdr->addr1);
2211                 goto done;
2212         }
2213
2214         if (!ether_addr_equal(hdr->addr3, dev->dev_addr)) {
2215                 PDEBUG(DEBUG_AP, "handle_ap_item - addr3(BSSID)=%pM"
2216                        " not own MAC\n", hdr->addr3);
2217                 goto done;
2218         }
2219
2220         switch (stype) {
2221         case IEEE80211_STYPE_ASSOC_REQ:
2222                 handle_assoc(local, skb, rx_stats, 0);
2223                 break;
2224         case IEEE80211_STYPE_ASSOC_RESP:
2225                 PDEBUG(DEBUG_AP, "==> ASSOC RESP (ignored)\n");
2226                 break;
2227         case IEEE80211_STYPE_REASSOC_REQ:
2228                 handle_assoc(local, skb, rx_stats, 1);
2229                 break;
2230         case IEEE80211_STYPE_REASSOC_RESP:
2231                 PDEBUG(DEBUG_AP, "==> REASSOC RESP (ignored)\n");
2232                 break;
2233         case IEEE80211_STYPE_ATIM:
2234                 PDEBUG(DEBUG_AP, "==> ATIM (ignored)\n");
2235                 break;
2236         case IEEE80211_STYPE_DISASSOC:
2237                 handle_disassoc(local, skb, rx_stats);
2238                 break;
2239         case IEEE80211_STYPE_AUTH:
2240                 handle_authen(local, skb, rx_stats);
2241                 break;
2242         case IEEE80211_STYPE_DEAUTH:
2243                 handle_deauth(local, skb, rx_stats);
2244                 break;
2245         default:
2246                 PDEBUG(DEBUG_AP, "Unknown mgmt frame subtype 0x%02x\n",
2247                        stype >> 4);
2248                 break;
2249         }
2250 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
2251
2252  done:
2253         dev_kfree_skb(skb);
2254 }
2255
2256
2257 /* Called only as a tasklet (software IRQ) */
2258 void hostap_rx(struct net_device *dev, struct sk_buff *skb,
2259                struct hostap_80211_rx_status *rx_stats)
2260 {
2261         struct hostap_interface *iface;
2262         local_info_t *local;
2263         struct ieee80211_hdr *hdr;
2264
2265         iface = netdev_priv(dev);
2266         local = iface->local;
2267
2268         if (skb->len < 16)
2269                 goto drop;
2270
2271         dev->stats.rx_packets++;
2272
2273         hdr = (struct ieee80211_hdr *) skb->data;
2274
2275         if (local->ap->ap_policy == AP_OTHER_AP_SKIP_ALL &&
2276             ieee80211_is_beacon(hdr->frame_control))
2277                 goto drop;
2278
2279         skb->protocol = cpu_to_be16(ETH_P_HOSTAP);
2280         handle_ap_item(local, skb, rx_stats);
2281         return;
2282
2283  drop:
2284         dev_kfree_skb(skb);
2285 }
2286
2287
2288 /* Called only as a tasklet (software IRQ) */
2289 static void schedule_packet_send(local_info_t *local, struct sta_info *sta)
2290 {
2291         struct sk_buff *skb;
2292         struct ieee80211_hdr *hdr;
2293         struct hostap_80211_rx_status rx_stats;
2294
2295         if (skb_queue_empty(&sta->tx_buf))
2296                 return;
2297
2298         skb = dev_alloc_skb(16);
2299         if (skb == NULL) {
2300                 printk(KERN_DEBUG "%s: schedule_packet_send: skb alloc "
2301                        "failed\n", local->dev->name);
2302                 return;
2303         }
2304
2305         hdr = skb_put(skb, 16);
2306
2307         /* Generate a fake pspoll frame to start packet delivery */
2308         hdr->frame_control = cpu_to_le16(
2309                 IEEE80211_FTYPE_CTL | IEEE80211_STYPE_PSPOLL);
2310         memcpy(hdr->addr1, local->dev->dev_addr, ETH_ALEN);
2311         memcpy(hdr->addr2, sta->addr, ETH_ALEN);
2312         hdr->duration_id = cpu_to_le16(sta->aid | BIT(15) | BIT(14));
2313
2314         PDEBUG(DEBUG_PS2,
2315                "%s: Scheduling buffered packet delivery for STA %pM\n",
2316                local->dev->name, sta->addr);
2317
2318         skb->dev = local->dev;
2319
2320         memset(&rx_stats, 0, sizeof(rx_stats));
2321         hostap_rx(local->dev, skb, &rx_stats);
2322 }
2323
2324
2325 int prism2_ap_get_sta_qual(local_info_t *local, struct sockaddr addr[],
2326                            struct iw_quality qual[], int buf_size,
2327                            int aplist)
2328 {
2329         struct ap_data *ap = local->ap;
2330         struct list_head *ptr;
2331         int count = 0;
2332
2333         spin_lock_bh(&ap->sta_table_lock);
2334
2335         for (ptr = ap->sta_list.next; ptr != NULL && ptr != &ap->sta_list;
2336              ptr = ptr->next) {
2337                 struct sta_info *sta = (struct sta_info *) ptr;
2338
2339                 if (aplist && !sta->ap)
2340                         continue;
2341                 addr[count].sa_family = ARPHRD_ETHER;
2342                 memcpy(addr[count].sa_data, sta->addr, ETH_ALEN);
2343                 if (sta->last_rx_silence == 0)
2344                         qual[count].qual = sta->last_rx_signal < 27 ?
2345                                 0 : (sta->last_rx_signal - 27) * 92 / 127;
2346                 else
2347                         qual[count].qual = sta->last_rx_signal -
2348                                 sta->last_rx_silence - 35;
2349                 qual[count].level = HFA384X_LEVEL_TO_dBm(sta->last_rx_signal);
2350                 qual[count].noise = HFA384X_LEVEL_TO_dBm(sta->last_rx_silence);
2351                 qual[count].updated = sta->last_rx_updated;
2352
2353                 sta->last_rx_updated = IW_QUAL_DBM;
2354
2355                 count++;
2356                 if (count >= buf_size)
2357                         break;
2358         }
2359         spin_unlock_bh(&ap->sta_table_lock);
2360
2361         return count;
2362 }
2363
2364
2365 /* Translate our list of Access Points & Stations to a card independent
2366  * format that the Wireless Tools will understand - Jean II */
2367 int prism2_ap_translate_scan(struct net_device *dev,
2368                              struct iw_request_info *info, char *buffer)
2369 {
2370         struct hostap_interface *iface;
2371         local_info_t *local;
2372         struct ap_data *ap;
2373         struct list_head *ptr;
2374         struct iw_event iwe;
2375         char *current_ev = buffer;
2376         char *end_buf = buffer + IW_SCAN_MAX_DATA;
2377 #if !defined(PRISM2_NO_KERNEL_IEEE80211_MGMT)
2378         char buf[64];
2379 #endif
2380
2381         iface = netdev_priv(dev);
2382         local = iface->local;
2383         ap = local->ap;
2384
2385         spin_lock_bh(&ap->sta_table_lock);
2386
2387         for (ptr = ap->sta_list.next; ptr != NULL && ptr != &ap->sta_list;
2388              ptr = ptr->next) {
2389                 struct sta_info *sta = (struct sta_info *) ptr;
2390
2391                 /* First entry *MUST* be the AP MAC address */
2392                 memset(&iwe, 0, sizeof(iwe));
2393                 iwe.cmd = SIOCGIWAP;
2394                 iwe.u.ap_addr.sa_family = ARPHRD_ETHER;
2395                 memcpy(iwe.u.ap_addr.sa_data, sta->addr, ETH_ALEN);
2396                 iwe.len = IW_EV_ADDR_LEN;
2397                 current_ev = iwe_stream_add_event(info, current_ev, end_buf,
2398                                                   &iwe, IW_EV_ADDR_LEN);
2399
2400                 /* Use the mode to indicate if it's a station or
2401                  * an Access Point */
2402                 memset(&iwe, 0, sizeof(iwe));
2403                 iwe.cmd = SIOCGIWMODE;
2404                 if (sta->ap)
2405                         iwe.u.mode = IW_MODE_MASTER;
2406                 else
2407                         iwe.u.mode = IW_MODE_INFRA;
2408                 iwe.len = IW_EV_UINT_LEN;
2409                 current_ev = iwe_stream_add_event(info, current_ev, end_buf,
2410                                                   &iwe, IW_EV_UINT_LEN);
2411
2412                 /* Some quality */
2413                 memset(&iwe, 0, sizeof(iwe));
2414                 iwe.cmd = IWEVQUAL;
2415                 if (sta->last_rx_silence == 0)
2416                         iwe.u.qual.qual = sta->last_rx_signal < 27 ?
2417                                 0 : (sta->last_rx_signal - 27) * 92 / 127;
2418                 else
2419                         iwe.u.qual.qual = sta->last_rx_signal -
2420                                 sta->last_rx_silence - 35;
2421                 iwe.u.qual.level = HFA384X_LEVEL_TO_dBm(sta->last_rx_signal);
2422                 iwe.u.qual.noise = HFA384X_LEVEL_TO_dBm(sta->last_rx_silence);
2423                 iwe.u.qual.updated = sta->last_rx_updated;
2424                 iwe.len = IW_EV_QUAL_LEN;
2425                 current_ev = iwe_stream_add_event(info, current_ev, end_buf,
2426                                                   &iwe, IW_EV_QUAL_LEN);
2427
2428 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
2429                 if (sta->ap) {
2430                         memset(&iwe, 0, sizeof(iwe));
2431                         iwe.cmd = SIOCGIWESSID;
2432                         iwe.u.data.length = sta->u.ap.ssid_len;
2433                         iwe.u.data.flags = 1;
2434                         current_ev = iwe_stream_add_point(info, current_ev,
2435                                                           end_buf, &iwe,
2436                                                           sta->u.ap.ssid);
2437
2438                         memset(&iwe, 0, sizeof(iwe));
2439                         iwe.cmd = SIOCGIWENCODE;
2440                         if (sta->capability & WLAN_CAPABILITY_PRIVACY)
2441                                 iwe.u.data.flags =
2442                                         IW_ENCODE_ENABLED | IW_ENCODE_NOKEY;
2443                         else
2444                                 iwe.u.data.flags = IW_ENCODE_DISABLED;
2445                         current_ev = iwe_stream_add_point(info, current_ev,
2446                                                           end_buf, &iwe,
2447                                                           sta->u.ap.ssid);
2448
2449                         if (sta->u.ap.channel > 0 &&
2450                             sta->u.ap.channel <= FREQ_COUNT) {
2451                                 memset(&iwe, 0, sizeof(iwe));
2452                                 iwe.cmd = SIOCGIWFREQ;
2453                                 iwe.u.freq.m = freq_list[sta->u.ap.channel - 1]
2454                                         * 100000;
2455                                 iwe.u.freq.e = 1;
2456                                 current_ev = iwe_stream_add_event(
2457                                         info, current_ev, end_buf, &iwe,
2458                                         IW_EV_FREQ_LEN);
2459                         }
2460
2461                         memset(&iwe, 0, sizeof(iwe));
2462                         iwe.cmd = IWEVCUSTOM;
2463                         sprintf(buf, "beacon_interval=%d",
2464                                 sta->listen_interval);
2465                         iwe.u.data.length = strlen(buf);
2466                         current_ev = iwe_stream_add_point(info, current_ev,
2467                                                           end_buf, &iwe, buf);
2468                 }
2469 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
2470
2471                 sta->last_rx_updated = IW_QUAL_DBM;
2472
2473                 /* To be continued, we should make good use of IWEVCUSTOM */
2474         }
2475
2476         spin_unlock_bh(&ap->sta_table_lock);
2477
2478         return current_ev - buffer;
2479 }
2480
2481
2482 static int prism2_hostapd_add_sta(struct ap_data *ap,
2483                                   struct prism2_hostapd_param *param)
2484 {
2485         struct sta_info *sta;
2486
2487         spin_lock_bh(&ap->sta_table_lock);
2488         sta = ap_get_sta(ap, param->sta_addr);
2489         if (sta)
2490                 atomic_inc(&sta->users);
2491         spin_unlock_bh(&ap->sta_table_lock);
2492
2493         if (sta == NULL) {
2494                 sta = ap_add_sta(ap, param->sta_addr);
2495                 if (sta == NULL)
2496                         return -1;
2497         }
2498
2499         if (!(sta->flags & WLAN_STA_ASSOC) && !sta->ap && sta->local)
2500                 hostap_event_new_sta(sta->local->dev, sta);
2501
2502         sta->flags |= WLAN_STA_AUTH | WLAN_STA_ASSOC;
2503         sta->last_rx = jiffies;
2504         sta->aid = param->u.add_sta.aid;
2505         sta->capability = param->u.add_sta.capability;
2506         sta->tx_supp_rates = param->u.add_sta.tx_supp_rates;
2507         if (sta->tx_supp_rates & WLAN_RATE_1M)
2508                 sta->supported_rates[0] = 2;
2509         if (sta->tx_supp_rates & WLAN_RATE_2M)
2510                 sta->supported_rates[1] = 4;
2511         if (sta->tx_supp_rates & WLAN_RATE_5M5)
2512                 sta->supported_rates[2] = 11;
2513         if (sta->tx_supp_rates & WLAN_RATE_11M)
2514                 sta->supported_rates[3] = 22;
2515         prism2_check_tx_rates(sta);
2516         atomic_dec(&sta->users);
2517         return 0;
2518 }
2519
2520
2521 static int prism2_hostapd_remove_sta(struct ap_data *ap,
2522                                      struct prism2_hostapd_param *param)
2523 {
2524         struct sta_info *sta;
2525
2526         spin_lock_bh(&ap->sta_table_lock);
2527         sta = ap_get_sta(ap, param->sta_addr);
2528         if (sta) {
2529                 ap_sta_hash_del(ap, sta);
2530                 list_del(&sta->list);
2531         }
2532         spin_unlock_bh(&ap->sta_table_lock);
2533
2534         if (!sta)
2535                 return -ENOENT;
2536
2537         if ((sta->flags & WLAN_STA_ASSOC) && !sta->ap && sta->local)
2538                 hostap_event_expired_sta(sta->local->dev, sta);
2539         ap_free_sta(ap, sta);
2540
2541         return 0;
2542 }
2543
2544
2545 static int prism2_hostapd_get_info_sta(struct ap_data *ap,
2546                                        struct prism2_hostapd_param *param)
2547 {
2548         struct sta_info *sta;
2549
2550         spin_lock_bh(&ap->sta_table_lock);
2551         sta = ap_get_sta(ap, param->sta_addr);
2552         if (sta)
2553                 atomic_inc(&sta->users);
2554         spin_unlock_bh(&ap->sta_table_lock);
2555
2556         if (!sta)
2557                 return -ENOENT;
2558
2559         param->u.get_info_sta.inactive_sec = (jiffies - sta->last_rx) / HZ;
2560
2561         atomic_dec(&sta->users);
2562
2563         return 1;
2564 }
2565
2566
2567 static int prism2_hostapd_set_flags_sta(struct ap_data *ap,
2568                                         struct prism2_hostapd_param *param)
2569 {
2570         struct sta_info *sta;
2571
2572         spin_lock_bh(&ap->sta_table_lock);
2573         sta = ap_get_sta(ap, param->sta_addr);
2574         if (sta) {
2575                 sta->flags |= param->u.set_flags_sta.flags_or;
2576                 sta->flags &= param->u.set_flags_sta.flags_and;
2577         }
2578         spin_unlock_bh(&ap->sta_table_lock);
2579
2580         if (!sta)
2581                 return -ENOENT;
2582
2583         return 0;
2584 }
2585
2586
2587 static int prism2_hostapd_sta_clear_stats(struct ap_data *ap,
2588                                           struct prism2_hostapd_param *param)
2589 {
2590         struct sta_info *sta;
2591         int rate;
2592
2593         spin_lock_bh(&ap->sta_table_lock);
2594         sta = ap_get_sta(ap, param->sta_addr);
2595         if (sta) {
2596                 sta->rx_packets = sta->tx_packets = 0;
2597                 sta->rx_bytes = sta->tx_bytes = 0;
2598                 for (rate = 0; rate < WLAN_RATE_COUNT; rate++) {
2599                         sta->tx_count[rate] = 0;
2600                         sta->rx_count[rate] = 0;
2601                 }
2602         }
2603         spin_unlock_bh(&ap->sta_table_lock);
2604
2605         if (!sta)
2606                 return -ENOENT;
2607
2608         return 0;
2609 }
2610
2611
2612 int prism2_hostapd(struct ap_data *ap, struct prism2_hostapd_param *param)
2613 {
2614         switch (param->cmd) {
2615         case PRISM2_HOSTAPD_FLUSH:
2616                 ap_control_kickall(ap);
2617                 return 0;
2618         case PRISM2_HOSTAPD_ADD_STA:
2619                 return prism2_hostapd_add_sta(ap, param);
2620         case PRISM2_HOSTAPD_REMOVE_STA:
2621                 return prism2_hostapd_remove_sta(ap, param);
2622         case PRISM2_HOSTAPD_GET_INFO_STA:
2623                 return prism2_hostapd_get_info_sta(ap, param);
2624         case PRISM2_HOSTAPD_SET_FLAGS_STA:
2625                 return prism2_hostapd_set_flags_sta(ap, param);
2626         case PRISM2_HOSTAPD_STA_CLEAR_STATS:
2627                 return prism2_hostapd_sta_clear_stats(ap, param);
2628         default:
2629                 printk(KERN_WARNING "prism2_hostapd: unknown cmd=%d\n",
2630                        param->cmd);
2631                 return -EOPNOTSUPP;
2632         }
2633 }
2634
2635
2636 /* Update station info for host-based TX rate control and return current
2637  * TX rate */
2638 static int ap_update_sta_tx_rate(struct sta_info *sta, struct net_device *dev)
2639 {
2640         int ret = sta->tx_rate;
2641         struct hostap_interface *iface;
2642         local_info_t *local;
2643
2644         iface = netdev_priv(dev);
2645         local = iface->local;
2646
2647         sta->tx_count[sta->tx_rate_idx]++;
2648         sta->tx_since_last_failure++;
2649         sta->tx_consecutive_exc = 0;
2650         if (sta->tx_since_last_failure >= WLAN_RATE_UPDATE_COUNT &&
2651             sta->tx_rate_idx < sta->tx_max_rate) {
2652                 /* use next higher rate */
2653                 int old_rate, new_rate;
2654                 old_rate = new_rate = sta->tx_rate_idx;
2655                 while (new_rate < sta->tx_max_rate) {
2656                         new_rate++;
2657                         if (ap_tx_rate_ok(new_rate, sta, local)) {
2658                                 sta->tx_rate_idx = new_rate;
2659                                 break;
2660                         }
2661                 }
2662                 if (old_rate != sta->tx_rate_idx) {
2663                         switch (sta->tx_rate_idx) {
2664                         case 0: sta->tx_rate = 10; break;
2665                         case 1: sta->tx_rate = 20; break;
2666                         case 2: sta->tx_rate = 55; break;
2667                         case 3: sta->tx_rate = 110; break;
2668                         default: sta->tx_rate = 0; break;
2669                         }
2670                         PDEBUG(DEBUG_AP, "%s: STA %pM TX rate raised to %d\n",
2671                                dev->name, sta->addr, sta->tx_rate);
2672                 }
2673                 sta->tx_since_last_failure = 0;
2674         }
2675
2676         return ret;
2677 }
2678
2679
2680 /* Called only from software IRQ. Called for each TX frame prior possible
2681  * encryption and transmit. */
2682 ap_tx_ret hostap_handle_sta_tx(local_info_t *local, struct hostap_tx_data *tx)
2683 {
2684         struct sta_info *sta = NULL;
2685         struct sk_buff *skb = tx->skb;
2686         int set_tim, ret;
2687         struct ieee80211_hdr *hdr;
2688         struct hostap_skb_tx_data *meta;
2689
2690         meta = (struct hostap_skb_tx_data *) skb->cb;
2691         ret = AP_TX_CONTINUE;
2692         if (local->ap == NULL || skb->len < 10 ||
2693             meta->iface->type == HOSTAP_INTERFACE_STA)
2694                 goto out;
2695
2696         hdr = (struct ieee80211_hdr *) skb->data;
2697
2698         if (hdr->addr1[0] & 0x01) {
2699                 /* broadcast/multicast frame - no AP related processing */
2700                 if (local->ap->num_sta <= 0)
2701                         ret = AP_TX_DROP;
2702                 goto out;
2703         }
2704
2705         /* unicast packet - check whether destination STA is associated */
2706         spin_lock(&local->ap->sta_table_lock);
2707         sta = ap_get_sta(local->ap, hdr->addr1);
2708         if (sta)
2709                 atomic_inc(&sta->users);
2710         spin_unlock(&local->ap->sta_table_lock);
2711
2712         if (local->iw_mode == IW_MODE_MASTER && sta == NULL &&
2713             !(meta->flags & HOSTAP_TX_FLAGS_WDS) &&
2714             meta->iface->type != HOSTAP_INTERFACE_MASTER &&
2715             meta->iface->type != HOSTAP_INTERFACE_AP) {
2716 #if 0
2717                 /* This can happen, e.g., when wlan0 is added to a bridge and
2718                  * bridging code does not know which port is the correct target
2719                  * for a unicast frame. In this case, the packet is send to all
2720                  * ports of the bridge. Since this is a valid scenario, do not
2721                  * print out any errors here. */
2722                 if (net_ratelimit()) {
2723                         printk(KERN_DEBUG "AP: drop packet to non-associated "
2724                                "STA %pM\n", hdr->addr1);
2725                 }
2726 #endif
2727                 local->ap->tx_drop_nonassoc++;
2728                 ret = AP_TX_DROP;
2729                 goto out;
2730         }
2731
2732         if (sta == NULL)
2733                 goto out;
2734
2735         if (!(sta->flags & WLAN_STA_AUTHORIZED))
2736                 ret = AP_TX_CONTINUE_NOT_AUTHORIZED;
2737
2738         /* Set tx_rate if using host-based TX rate control */
2739         if (!local->fw_tx_rate_control)
2740                 local->ap->last_tx_rate = meta->rate =
2741                         ap_update_sta_tx_rate(sta, local->dev);
2742
2743         if (local->iw_mode != IW_MODE_MASTER)
2744                 goto out;
2745
2746         if (!(sta->flags & WLAN_STA_PS))
2747                 goto out;
2748
2749         if (meta->flags & HOSTAP_TX_FLAGS_ADD_MOREDATA) {
2750                 /* indicate to STA that more frames follow */
2751                 hdr->frame_control |=
2752                         cpu_to_le16(IEEE80211_FCTL_MOREDATA);
2753         }
2754
2755         if (meta->flags & HOSTAP_TX_FLAGS_BUFFERED_FRAME) {
2756                 /* packet was already buffered and now send due to
2757                  * PS poll, so do not rebuffer it */
2758                 goto out;
2759         }
2760
2761         if (skb_queue_len(&sta->tx_buf) >= STA_MAX_TX_BUFFER) {
2762                 PDEBUG(DEBUG_PS, "%s: No more space in STA (%pM)'s"
2763                        "PS mode buffer\n",
2764                        local->dev->name, sta->addr);
2765                 /* Make sure that TIM is set for the station (it might not be
2766                  * after AP wlan hw reset). */
2767                 /* FIX: should fix hw reset to restore bits based on STA
2768                  * buffer state.. */
2769                 hostap_set_tim(local, sta->aid, 1);
2770                 sta->flags |= WLAN_STA_TIM;
2771                 ret = AP_TX_DROP;
2772                 goto out;
2773         }
2774
2775         /* STA in PS mode, buffer frame for later delivery */
2776         set_tim = skb_queue_empty(&sta->tx_buf);
2777         skb_queue_tail(&sta->tx_buf, skb);
2778         /* FIX: could save RX time to skb and expire buffered frames after
2779          * some time if STA does not poll for them */
2780
2781         if (set_tim) {
2782                 if (sta->flags & WLAN_STA_TIM)
2783                         PDEBUG(DEBUG_PS2, "Re-setting TIM for aid %d\n",
2784                                sta->aid);
2785                 hostap_set_tim(local, sta->aid, 1);
2786                 sta->flags |= WLAN_STA_TIM;
2787         }
2788
2789         ret = AP_TX_BUFFERED;
2790
2791  out:
2792         if (sta != NULL) {
2793                 if (ret == AP_TX_CONTINUE ||
2794                     ret == AP_TX_CONTINUE_NOT_AUTHORIZED) {
2795                         sta->tx_packets++;
2796                         sta->tx_bytes += skb->len;
2797                         sta->last_tx = jiffies;
2798                 }
2799
2800                 if ((ret == AP_TX_CONTINUE ||
2801                      ret == AP_TX_CONTINUE_NOT_AUTHORIZED) &&
2802                     sta->crypt && tx->host_encrypt) {
2803                         tx->crypt = sta->crypt;
2804                         tx->sta_ptr = sta; /* hostap_handle_sta_release() will
2805                                             * be called to release sta info
2806                                             * later */
2807                 } else
2808                         atomic_dec(&sta->users);
2809         }
2810
2811         return ret;
2812 }
2813
2814
2815 void hostap_handle_sta_release(void *ptr)
2816 {
2817         struct sta_info *sta = ptr;
2818         atomic_dec(&sta->users);
2819 }
2820
2821
2822 /* Called only as a tasklet (software IRQ) */
2823 void hostap_handle_sta_tx_exc(local_info_t *local, struct sk_buff *skb)
2824 {
2825         struct sta_info *sta;
2826         struct ieee80211_hdr *hdr;
2827         struct hostap_skb_tx_data *meta;
2828
2829         hdr = (struct ieee80211_hdr *) skb->data;
2830         meta = (struct hostap_skb_tx_data *) skb->cb;
2831
2832         spin_lock(&local->ap->sta_table_lock);
2833         sta = ap_get_sta(local->ap, hdr->addr1);
2834         if (!sta) {
2835                 spin_unlock(&local->ap->sta_table_lock);
2836                 PDEBUG(DEBUG_AP, "%s: Could not find STA %pM"
2837                        " for this TX error (@%lu)\n",
2838                        local->dev->name, hdr->addr1, jiffies);
2839                 return;
2840         }
2841
2842         sta->tx_since_last_failure = 0;
2843         sta->tx_consecutive_exc++;
2844
2845         if (sta->tx_consecutive_exc >= WLAN_RATE_DECREASE_THRESHOLD &&
2846             sta->tx_rate_idx > 0 && meta->rate <= sta->tx_rate) {
2847                 /* use next lower rate */
2848                 int old, rate;
2849                 old = rate = sta->tx_rate_idx;
2850                 while (rate > 0) {
2851                         rate--;
2852                         if (ap_tx_rate_ok(rate, sta, local)) {
2853                                 sta->tx_rate_idx = rate;
2854                                 break;
2855                         }
2856                 }
2857                 if (old != sta->tx_rate_idx) {
2858                         switch (sta->tx_rate_idx) {
2859                         case 0: sta->tx_rate = 10; break;
2860                         case 1: sta->tx_rate = 20; break;
2861                         case 2: sta->tx_rate = 55; break;
2862                         case 3: sta->tx_rate = 110; break;
2863                         default: sta->tx_rate = 0; break;
2864                         }
2865                         PDEBUG(DEBUG_AP,
2866                                "%s: STA %pM TX rate lowered to %d\n",
2867                                local->dev->name, sta->addr, sta->tx_rate);
2868                 }
2869                 sta->tx_consecutive_exc = 0;
2870         }
2871         spin_unlock(&local->ap->sta_table_lock);
2872 }
2873
2874
2875 static void hostap_update_sta_ps2(local_info_t *local, struct sta_info *sta,
2876                                   int pwrmgt, int type, int stype)
2877 {
2878         if (pwrmgt && !(sta->flags & WLAN_STA_PS)) {
2879                 sta->flags |= WLAN_STA_PS;
2880                 PDEBUG(DEBUG_PS2, "STA %pM changed to use PS "
2881                        "mode (type=0x%02X, stype=0x%02X)\n",
2882                        sta->addr, type >> 2, stype >> 4);
2883         } else if (!pwrmgt && (sta->flags & WLAN_STA_PS)) {
2884                 sta->flags &= ~WLAN_STA_PS;
2885                 PDEBUG(DEBUG_PS2, "STA %pM changed to not use "
2886                        "PS mode (type=0x%02X, stype=0x%02X)\n",
2887                        sta->addr, type >> 2, stype >> 4);
2888                 if (type != IEEE80211_FTYPE_CTL ||
2889                     stype != IEEE80211_STYPE_PSPOLL)
2890                         schedule_packet_send(local, sta);
2891         }
2892 }
2893
2894
2895 /* Called only as a tasklet (software IRQ). Called for each RX frame to update
2896  * STA power saving state. pwrmgt is a flag from 802.11 frame_control field. */
2897 int hostap_update_sta_ps(local_info_t *local, struct ieee80211_hdr *hdr)
2898 {
2899         struct sta_info *sta;
2900         u16 fc;
2901
2902         spin_lock(&local->ap->sta_table_lock);
2903         sta = ap_get_sta(local->ap, hdr->addr2);
2904         if (sta)
2905                 atomic_inc(&sta->users);
2906         spin_unlock(&local->ap->sta_table_lock);
2907
2908         if (!sta)
2909                 return -1;
2910
2911         fc = le16_to_cpu(hdr->frame_control);
2912         hostap_update_sta_ps2(local, sta, fc & IEEE80211_FCTL_PM,
2913                               fc & IEEE80211_FCTL_FTYPE,
2914                               fc & IEEE80211_FCTL_STYPE);
2915
2916         atomic_dec(&sta->users);
2917         return 0;
2918 }
2919
2920
2921 /* Called only as a tasklet (software IRQ). Called for each RX frame after
2922  * getting RX header and payload from hardware. */
2923 ap_rx_ret hostap_handle_sta_rx(local_info_t *local, struct net_device *dev,
2924                                struct sk_buff *skb,
2925                                struct hostap_80211_rx_status *rx_stats,
2926                                int wds)
2927 {
2928         int ret;
2929         struct sta_info *sta;
2930         u16 fc, type, stype;
2931         struct ieee80211_hdr *hdr;
2932
2933         if (local->ap == NULL)
2934                 return AP_RX_CONTINUE;
2935
2936         hdr = (struct ieee80211_hdr *) skb->data;
2937
2938         fc = le16_to_cpu(hdr->frame_control);
2939         type = fc & IEEE80211_FCTL_FTYPE;
2940         stype = fc & IEEE80211_FCTL_STYPE;
2941
2942         spin_lock(&local->ap->sta_table_lock);
2943         sta = ap_get_sta(local->ap, hdr->addr2);
2944         if (sta)
2945                 atomic_inc(&sta->users);
2946         spin_unlock(&local->ap->sta_table_lock);
2947
2948         if (sta && !(sta->flags & WLAN_STA_AUTHORIZED))
2949                 ret = AP_RX_CONTINUE_NOT_AUTHORIZED;
2950         else
2951                 ret = AP_RX_CONTINUE;
2952
2953
2954         if (fc & IEEE80211_FCTL_TODS) {
2955                 if (!wds && (sta == NULL || !(sta->flags & WLAN_STA_ASSOC))) {
2956                         if (local->hostapd) {
2957                                 prism2_rx_80211(local->apdev, skb, rx_stats,
2958                                                 PRISM2_RX_NON_ASSOC);
2959 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
2960                         } else {
2961                                 printk(KERN_DEBUG "%s: dropped received packet"
2962                                        " from non-associated STA %pM"
2963                                        " (type=0x%02x, subtype=0x%02x)\n",
2964                                        dev->name, hdr->addr2,
2965                                        type >> 2, stype >> 4);
2966                                 hostap_rx(dev, skb, rx_stats);
2967 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
2968                         }
2969                         ret = AP_RX_EXIT;
2970                         goto out;
2971                 }
2972         } else if (fc & IEEE80211_FCTL_FROMDS) {
2973                 if (!wds) {
2974                         /* FromDS frame - not for us; probably
2975                          * broadcast/multicast in another BSS - drop */
2976                         if (ether_addr_equal(hdr->addr1, dev->dev_addr)) {
2977                                 printk(KERN_DEBUG "Odd.. FromDS packet "
2978                                        "received with own BSSID\n");
2979                                 hostap_dump_rx_80211(dev->name, skb, rx_stats);
2980                         }
2981                         ret = AP_RX_DROP;
2982                         goto out;
2983                 }
2984         } else if (stype == IEEE80211_STYPE_NULLFUNC && sta == NULL &&
2985                    ether_addr_equal(hdr->addr1, dev->dev_addr)) {
2986
2987                 if (local->hostapd) {
2988                         prism2_rx_80211(local->apdev, skb, rx_stats,
2989                                         PRISM2_RX_NON_ASSOC);
2990 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
2991                 } else {
2992                         /* At least Lucent f/w seems to send data::nullfunc
2993                          * frames with no ToDS flag when the current AP returns
2994                          * after being unavailable for some time. Speed up
2995                          * re-association by informing the station about it not
2996                          * being associated. */
2997                         printk(KERN_DEBUG "%s: rejected received nullfunc frame"
2998                                " without ToDS from not associated STA %pM\n",
2999                                dev->name, hdr->addr2);
3000                         hostap_rx(dev, skb, rx_stats);
3001 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
3002                 }
3003                 ret = AP_RX_EXIT;
3004                 goto out;
3005         } else if (stype == IEEE80211_STYPE_NULLFUNC) {
3006                 /* At least Lucent cards seem to send periodic nullfunc
3007                  * frames with ToDS. Let these through to update SQ
3008                  * stats and PS state. Nullfunc frames do not contain
3009                  * any data and they will be dropped below. */
3010         } else {
3011                 /* If BSSID (Addr3) is foreign, this frame is a normal
3012                  * broadcast frame from an IBSS network. Drop it silently.
3013                  * If BSSID is own, report the dropping of this frame. */
3014                 if (ether_addr_equal(hdr->addr3, dev->dev_addr)) {
3015                         printk(KERN_DEBUG "%s: dropped received packet from %pM"
3016                                " with no ToDS flag "
3017                                "(type=0x%02x, subtype=0x%02x)\n", dev->name,
3018                                hdr->addr2, type >> 2, stype >> 4);
3019                         hostap_dump_rx_80211(dev->name, skb, rx_stats);
3020                 }
3021                 ret = AP_RX_DROP;
3022                 goto out;
3023         }
3024
3025         if (sta) {
3026                 hostap_update_sta_ps2(local, sta, fc & IEEE80211_FCTL_PM,
3027                                       type, stype);
3028
3029                 sta->rx_packets++;
3030                 sta->rx_bytes += skb->len;
3031                 sta->last_rx = jiffies;
3032         }
3033
3034         if (local->ap->nullfunc_ack && stype == IEEE80211_STYPE_NULLFUNC &&
3035             fc & IEEE80211_FCTL_TODS) {
3036                 if (local->hostapd) {
3037                         prism2_rx_80211(local->apdev, skb, rx_stats,
3038                                         PRISM2_RX_NULLFUNC_ACK);
3039 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
3040                 } else {
3041                         /* some STA f/w's seem to require control::ACK frame
3042                          * for data::nullfunc, but Prism2 f/w 0.8.0 (at least
3043                          * from Compaq) does not send this.. Try to generate
3044                          * ACK for these frames from the host driver to make
3045                          * power saving work with, e.g., Lucent WaveLAN f/w */
3046                         hostap_rx(dev, skb, rx_stats);
3047 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */
3048                 }
3049                 ret = AP_RX_EXIT;
3050                 goto out;
3051         }
3052
3053  out:
3054         if (sta)
3055                 atomic_dec(&sta->users);
3056
3057         return ret;
3058 }
3059
3060
3061 /* Called only as a tasklet (software IRQ) */
3062 int hostap_handle_sta_crypto(local_info_t *local,
3063                              struct ieee80211_hdr *hdr,
3064                              struct lib80211_crypt_data **crypt,
3065                              void **sta_ptr)
3066 {
3067         struct sta_info *sta;
3068
3069         spin_lock(&local->ap->sta_table_lock);
3070         sta = ap_get_sta(local->ap, hdr->addr2);
3071         if (sta)
3072                 atomic_inc(&sta->users);
3073         spin_unlock(&local->ap->sta_table_lock);
3074
3075         if (!sta)
3076                 return -1;
3077
3078         if (sta->crypt) {
3079                 *crypt = sta->crypt;
3080                 *sta_ptr = sta;
3081                 /* hostap_handle_sta_release() will be called to release STA
3082                  * info */
3083         } else
3084                 atomic_dec(&sta->users);
3085
3086         return 0;
3087 }
3088
3089
3090 /* Called only as a tasklet (software IRQ) */
3091 int hostap_is_sta_assoc(struct ap_data *ap, u8 *sta_addr)
3092 {
3093         struct sta_info *sta;
3094         int ret = 0;
3095
3096         spin_lock(&ap->sta_table_lock);
3097         sta = ap_get_sta(ap, sta_addr);
3098         if (sta != NULL && (sta->flags & WLAN_STA_ASSOC) && !sta->ap)
3099                 ret = 1;
3100         spin_unlock(&ap->sta_table_lock);
3101
3102         return ret;
3103 }
3104
3105
3106 /* Called only as a tasklet (software IRQ) */
3107 int hostap_is_sta_authorized(struct ap_data *ap, u8 *sta_addr)
3108 {
3109         struct sta_info *sta;
3110         int ret = 0;
3111
3112         spin_lock(&ap->sta_table_lock);
3113         sta = ap_get_sta(ap, sta_addr);
3114         if (sta != NULL && (sta->flags & WLAN_STA_ASSOC) && !sta->ap &&
3115             ((sta->flags & WLAN_STA_AUTHORIZED) ||
3116              ap->local->ieee_802_1x == 0))
3117                 ret = 1;
3118         spin_unlock(&ap->sta_table_lock);
3119
3120         return ret;
3121 }
3122
3123
3124 /* Called only as a tasklet (software IRQ) */
3125 int hostap_add_sta(struct ap_data *ap, u8 *sta_addr)
3126 {
3127         struct sta_info *sta;
3128         int ret = 1;
3129
3130         if (!ap)
3131                 return -1;
3132
3133         spin_lock(&ap->sta_table_lock);
3134         sta = ap_get_sta(ap, sta_addr);
3135         if (sta)
3136                 ret = 0;
3137         spin_unlock(&ap->sta_table_lock);
3138
3139         if (ret == 1) {
3140                 sta = ap_add_sta(ap, sta_addr);
3141                 if (!sta)
3142                         return -1;
3143                 sta->flags = WLAN_STA_AUTH | WLAN_STA_ASSOC;
3144                 sta->ap = 1;
3145                 memset(sta->supported_rates, 0, sizeof(sta->supported_rates));
3146                 /* No way of knowing which rates are supported since we did not
3147                  * get supported rates element from beacon/assoc req. Assume
3148                  * that remote end supports all 802.11b rates. */
3149                 sta->supported_rates[0] = 0x82;
3150                 sta->supported_rates[1] = 0x84;
3151                 sta->supported_rates[2] = 0x0b;
3152                 sta->supported_rates[3] = 0x16;
3153                 sta->tx_supp_rates = WLAN_RATE_1M | WLAN_RATE_2M |
3154                         WLAN_RATE_5M5 | WLAN_RATE_11M;
3155                 sta->tx_rate = 110;
3156                 sta->tx_max_rate = sta->tx_rate_idx = 3;
3157         }
3158
3159         return ret;
3160 }
3161
3162
3163 /* Called only as a tasklet (software IRQ) */
3164 int hostap_update_rx_stats(struct ap_data *ap,
3165                            struct ieee80211_hdr *hdr,
3166                            struct hostap_80211_rx_status *rx_stats)
3167 {
3168         struct sta_info *sta;
3169
3170         if (!ap)
3171                 return -1;
3172
3173         spin_lock(&ap->sta_table_lock);
3174         sta = ap_get_sta(ap, hdr->addr2);
3175         if (sta) {
3176                 sta->last_rx_silence = rx_stats->noise;
3177                 sta->last_rx_signal = rx_stats->signal;
3178                 sta->last_rx_rate = rx_stats->rate;
3179                 sta->last_rx_updated = IW_QUAL_ALL_UPDATED | IW_QUAL_DBM;
3180                 if (rx_stats->rate == 10)
3181                         sta->rx_count[0]++;
3182                 else if (rx_stats->rate == 20)
3183                         sta->rx_count[1]++;
3184                 else if (rx_stats->rate == 55)
3185                         sta->rx_count[2]++;
3186                 else if (rx_stats->rate == 110)
3187                         sta->rx_count[3]++;
3188         }
3189         spin_unlock(&ap->sta_table_lock);
3190
3191         return sta ? 0 : -1;
3192 }
3193
3194
3195 void hostap_update_rates(local_info_t *local)
3196 {
3197         struct sta_info *sta;
3198         struct ap_data *ap = local->ap;
3199
3200         if (!ap)
3201                 return;
3202
3203         spin_lock_bh(&ap->sta_table_lock);
3204         list_for_each_entry(sta, &ap->sta_list, list) {
3205                 prism2_check_tx_rates(sta);
3206         }
3207         spin_unlock_bh(&ap->sta_table_lock);
3208 }
3209
3210
3211 void * ap_crypt_get_ptrs(struct ap_data *ap, u8 *addr, int permanent,
3212                          struct lib80211_crypt_data ***crypt)
3213 {
3214         struct sta_info *sta;
3215
3216         spin_lock_bh(&ap->sta_table_lock);
3217         sta = ap_get_sta(ap, addr);
3218         if (sta)
3219                 atomic_inc(&sta->users);
3220         spin_unlock_bh(&ap->sta_table_lock);
3221
3222         if (!sta && permanent)
3223                 sta = ap_add_sta(ap, addr);
3224
3225         if (!sta)
3226                 return NULL;
3227
3228         if (permanent)
3229                 sta->flags |= WLAN_STA_PERM;
3230
3231         *crypt = &sta->crypt;
3232
3233         return sta;
3234 }
3235
3236
3237 void hostap_add_wds_links(local_info_t *local)
3238 {
3239         struct ap_data *ap = local->ap;
3240         struct sta_info *sta;
3241
3242         spin_lock_bh(&ap->sta_table_lock);
3243         list_for_each_entry(sta, &ap->sta_list, list) {
3244                 if (sta->ap)
3245                         hostap_wds_link_oper(local, sta->addr, WDS_ADD);
3246         }
3247         spin_unlock_bh(&ap->sta_table_lock);
3248
3249         schedule_work(&local->ap->wds_oper_queue);
3250 }
3251
3252
3253 void hostap_wds_link_oper(local_info_t *local, u8 *addr, wds_oper_type type)
3254 {
3255         struct wds_oper_data *entry;
3256
3257         entry = kmalloc(sizeof(*entry), GFP_ATOMIC);
3258         if (!entry)
3259                 return;
3260         memcpy(entry->addr, addr, ETH_ALEN);
3261         entry->type = type;
3262         spin_lock_bh(&local->lock);
3263         entry->next = local->ap->wds_oper_entries;
3264         local->ap->wds_oper_entries = entry;
3265         spin_unlock_bh(&local->lock);
3266
3267         schedule_work(&local->ap->wds_oper_queue);
3268 }
3269
3270
3271 EXPORT_SYMBOL(hostap_init_data);
3272 EXPORT_SYMBOL(hostap_init_ap_proc);
3273 EXPORT_SYMBOL(hostap_free_data);
3274 EXPORT_SYMBOL(hostap_check_sta_fw_version);
3275 EXPORT_SYMBOL(hostap_handle_sta_tx_exc);
3276 #ifndef PRISM2_NO_KERNEL_IEEE80211_MGMT
3277 #endif /* PRISM2_NO_KERNEL_IEEE80211_MGMT */