drivers/net/wireless/hostap/hostap_ap.c

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