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