drivers/net/wireless/iwlwifi/iwl-4965-rs.c

   1 /******************************************************************************
   2  *
   3  * Copyright(c) 2005 - 2007 Intel Corporation. All rights reserved.
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms of version 2 of the GNU General Public License as
   7  * published by the Free Software Foundation.
   8  *
   9  * This program is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  12  * more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along with
  15  * this program; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin Street, Fifth Floor, Boston, MA 02110, USA
  17  *
  18  * The full GNU General Public License is included in this distribution in the
  19  * file called LICENSE.
  20  *
  21  * Contact Information:
  22  * James P. Ketrenos <ipw2100-admin@linux.intel.com>
  23  * Intel Corporation, 5200 N.E. Elam Young Parkway, Hillsboro, OR 97124-6497
  24  *
  25  *****************************************************************************/
  26 #include <linux/kernel.h>
  27 #include <linux/init.h>
  28 #include <linux/skbuff.h>
  29 #include <linux/wireless.h>
  30 #include <net/mac80211.h>
  31 #include <net/ieee80211.h>
  32
  33 #include <linux/netdevice.h>
  34 #include <linux/etherdevice.h>
  35 #include <linux/delay.h>
  36
  37 #include <linux/workqueue.h>
  38
  39 #include <net/mac80211.h>
  40 #include <linux/wireless.h>
  41
  42 #include "../net/mac80211/ieee80211_rate.h"
  43
  44 #include "iwlwifi.h"
  45 #include "iwl-helpers.h"
  46
  47 #define RS_NAME "iwl-4965-rs"
  48
  49 #define NUM_TRY_BEFORE_ANTENNA_TOGGLE 1
  50 #define IWL_NUMBER_TRY      1
  51 #define IWL_HT_NUMBER_TRY   3
  52
  53 #define IWL_RATE_MAX_WINDOW             62
  54 #define IWL_RATE_HIGH_TH                10880
  55 #define IWL_RATE_MIN_FAILURE_TH         6
  56 #define IWL_RATE_MIN_SUCCESS_TH         8
  57 #define IWL_RATE_DECREASE_TH            1920
  58 #define IWL_RATE_INCREASE_TH            8960
  59 #define IWL_RATE_SCALE_FLUSH_INTVL   (2*HZ)        /*2 seconds */
  60
  61 static u8 rs_ht_to_legacy[] = {
  62         IWL_RATE_6M_INDEX, IWL_RATE_6M_INDEX,
  63         IWL_RATE_6M_INDEX, IWL_RATE_6M_INDEX,
  64         IWL_RATE_6M_INDEX,
  65         IWL_RATE_6M_INDEX, IWL_RATE_9M_INDEX,
  66         IWL_RATE_12M_INDEX, IWL_RATE_18M_INDEX,
  67         IWL_RATE_24M_INDEX, IWL_RATE_36M_INDEX,
  68         IWL_RATE_48M_INDEX, IWL_RATE_54M_INDEX
  69 };
  70
  71 struct iwl_rate {
  72         u32 rate_n_flags;
  73 } __attribute__ ((packed));
  74
  75 struct iwl_rate_scale_data {
  76         u64 data;
  77         s32 success_counter;
  78         s32 success_ratio;
  79         s32 counter;
  80         s32 average_tpt;
  81         unsigned long stamp;
  82 };
  83
  84 struct iwl_scale_tbl_info {
  85         enum iwl_table_type lq_type;
  86         enum iwl_antenna_type antenna_type;
  87         u8 is_SGI;
  88         u8 is_fat;
  89         u8 is_dup;
  90         u8 action;
  91         s32 *expected_tpt;
  92         struct iwl_rate current_rate;
  93         struct iwl_rate_scale_data win[IWL_RATE_COUNT];
  94 };
  95
  96 struct iwl_rate_scale_priv {
  97         u8 active_tbl;
  98         u8 enable_counter;
  99         u8 stay_in_tbl;
 100         u8 search_better_tbl;
 101         s32 last_tpt;
 102         u32 table_count_limit;
 103         u32 max_failure_limit;
 104         u32 max_success_limit;
 105         u32 table_count;
 106         u32 total_failed;
 107         u32 total_success;
 108         u32 flush_timer;
 109         u8 action_counter;
 110         u8 antenna;
 111         u8 valid_antenna;
 112         u8 is_green;
 113         u8 is_dup;
 114         u8 phymode;
 115         u8 ibss_sta_added;
 116         u16 active_rate;
 117         u16 active_siso_rate;
 118         u16 active_mimo_rate;
 119         u16 active_rate_basic;
 120         struct iwl_link_quality_cmd lq;
 121         struct iwl_scale_tbl_info lq_info[LQ_SIZE];
 122 };
 123
 124 static void rs_rate_scale_perform(struct iwl_priv *priv,
 125                                    struct net_device *dev,
 126                                    struct ieee80211_hdr *hdr,
 127                                    struct sta_info *sta);
 128 static int rs_fill_link_cmd(struct iwl_rate_scale_priv *lq_data,
 129                              struct iwl_rate *tx_mcs,
 130                              struct iwl_link_quality_cmd *tbl,
 131                              struct sta_info *sta);
 132
 133
 134 static s32 expected_tpt_A[IWL_RATE_COUNT] = {
 135         0, 0, 0, 0, 40, 57, 72, 98, 121, 154, 177, 186, 186
 136 };
 137
 138 static s32 expected_tpt_G[IWL_RATE_COUNT] = {
 139         7, 13, 35, 58, 40, 57, 72, 98, 121, 154, 177, 186, 186
 140 };
 141
 142 static s32 expected_tpt_siso20MHz[IWL_RATE_COUNT] = {
 143         0, 0, 0, 0, 42, 42, 76, 102, 124, 159, 183, 193, 202
 144 };
 145
 146 static s32 expected_tpt_siso20MHzSGI[IWL_RATE_COUNT] = {
 147         0, 0, 0, 0, 46, 46, 82, 110, 132, 168, 192, 202, 211
 148 };
 149
 150 static s32 expected_tpt_mimo20MHz[IWL_RATE_COUNT] = {
 151         0, 0, 0, 0, 74, 74, 123, 155, 179, 214, 236, 244, 251
 152 };
 153
 154 static s32 expected_tpt_mimo20MHzSGI[IWL_RATE_COUNT] = {
 155         0, 0, 0, 0, 81, 81, 131, 164, 188, 222, 243, 251, 257
 156 };
 157
 158 static s32 expected_tpt_siso40MHz[IWL_RATE_COUNT] = {
 159         0, 0, 0, 0, 77, 77, 127, 160, 184, 220, 242, 250, 257
 160 };
 161
 162 static s32 expected_tpt_siso40MHzSGI[IWL_RATE_COUNT] = {
 163         0, 0, 0, 0, 83, 83, 135, 169, 193, 229, 250, 257, 264
 164 };
 165
 166 static s32 expected_tpt_mimo40MHz[IWL_RATE_COUNT] = {
 167         0, 0, 0, 0, 123, 123, 182, 214, 235, 264, 279, 285, 289
 168 };
 169
 170 static s32 expected_tpt_mimo40MHzSGI[IWL_RATE_COUNT] = {
 171         0, 0, 0, 0, 131, 131, 191, 222, 242, 270, 284, 289, 293
 172 };
 173
 174 static int iwl_lq_sync_callback(struct iwl_priv *priv,
 175                                 struct iwl_cmd *cmd, struct sk_buff *skb)
 176 {
 177         /*We didn't cache the SKB; let the caller free it */
 178         return 1;
 179 }
 180
 181 static inline u8 iwl_rate_get_rate(u32 rate_n_flags)
 182 {
 183         return (u8)(rate_n_flags & 0xFF);
 184 }
 185
 186 static int rs_send_lq_cmd(struct iwl_priv *priv,
 187                           struct iwl_link_quality_cmd *lq, u8 flags)
 188 {
 189 #ifdef CONFIG_IWLWIFI_DEBUG
 190         int i;
 191 #endif
 192         int rc = -1;
 193
 194         struct iwl_host_cmd cmd = {
 195                 .id = REPLY_TX_LINK_QUALITY_CMD,
 196                 .len = sizeof(struct iwl_link_quality_cmd),
 197                 .meta.flags = flags,
 198                 .data = lq,
 199         };
 200
 201         if ((lq->sta_id == 0xFF) &&
 202             (priv->iw_mode == IEEE80211_IF_TYPE_IBSS))
 203                 return rc;
 204
 205         if (lq->sta_id == 0xFF)
 206                 lq->sta_id = IWL_AP_ID;
 207
 208         IWL_DEBUG_RATE("lq station id 0x%x\n", lq->sta_id);
 209         IWL_DEBUG_RATE("lq dta 0x%X 0x%X\n",
 210                        lq->general_params.single_stream_ant_msk,
 211                        lq->general_params.dual_stream_ant_msk);
 212 #ifdef CONFIG_IWLWIFI_DEBUG
 213         for (i = 0; i < LINK_QUAL_MAX_RETRY_NUM; i++)
 214                 IWL_DEBUG_RATE("lq index %d 0x%X\n",
 215                                 i, lq->rs_table[i].rate_n_flags);
 216 #endif
 217
 218         if (flags & CMD_ASYNC)
 219                 cmd.meta.u.callback = iwl_lq_sync_callback;
 220
 221         if (iwl_is_associated(priv) && priv->assoc_station_added &&
 222             priv->lq_mngr.lq_ready)
 223                 rc = iwl_send_cmd(priv, &cmd);
 224
 225         return rc;
 226 }
 227
 228 static int rs_rate_scale_clear_window(struct iwl_rate_scale_data *window)
 229 {
 230         window->data = 0;
 231         window->success_counter = 0;
 232         window->success_ratio = IWL_INVALID_VALUE;
 233         window->counter = 0;
 234         window->average_tpt = IWL_INVALID_VALUE;
 235         window->stamp = 0;
 236
 237         return 0;
 238 }
 239
 240 static int rs_collect_tx_data(struct iwl_rate_scale_data *windows,
 241                               int scale_index, s32 tpt, u32 status)
 242 {
 243         int rc = 0;
 244         struct iwl_rate_scale_data *window = NULL;
 245         u64 mask;
 246         u8 win_size = IWL_RATE_MAX_WINDOW;
 247         s32 fail_count;
 248
 249         if (scale_index < 0)
 250                 return -1;
 251
 252         if (scale_index >= IWL_RATE_COUNT)
 253                 return -1;
 254
 255         window = &(windows[scale_index]);
 256
 257         if (window->counter >= win_size) {
 258
 259                 window->counter = win_size - 1;
 260                 mask = 1;
 261                 mask = (mask << (win_size - 1));
 262                 if ((window->data & mask)) {
 263                         window->data &= ~mask;
 264                         window->success_counter = window->success_counter - 1;
 265                 }
 266         }
 267
 268         window->counter = window->counter + 1;
 269         mask = window->data;
 270         window->data = (mask << 1);
 271         if (status != 0) {
 272                 window->success_counter = window->success_counter + 1;
 273                 window->data |= 0x1;
 274         }
 275
 276         if (window->counter > 0)
 277                 window->success_ratio = 128 * (100 * window->success_counter)
 278                                         / window->counter;
 279         else
 280                 window->success_ratio = IWL_INVALID_VALUE;
 281
 282         fail_count = window->counter - window->success_counter;
 283
 284         if ((fail_count >= IWL_RATE_MIN_FAILURE_TH) ||
 285             (window->success_counter >= IWL_RATE_MIN_SUCCESS_TH))
 286                 window->average_tpt = (window->success_ratio * tpt + 64) / 128;
 287         else
 288                 window->average_tpt = IWL_INVALID_VALUE;
 289
 290         window->stamp = jiffies;
 291
 292         return rc;
 293 }
 294
 295 int static rs_mcs_from_tbl(struct iwl_rate *mcs_rate,
 296                            struct iwl_scale_tbl_info *tbl,
 297                            int index, u8 use_green)
 298 {
 299         int rc = 0;
 300
 301         if (is_legacy(tbl->lq_type)) {
 302                 mcs_rate->rate_n_flags = iwl_rates[index].plcp;
 303                 if (index >= IWL_FIRST_CCK_RATE && index <= IWL_LAST_CCK_RATE)
 304                         mcs_rate->rate_n_flags |= RATE_MCS_CCK_MSK;
 305
 306         } else if (is_siso(tbl->lq_type)) {
 307                 if (index > IWL_LAST_OFDM_RATE)
 308                         index = IWL_LAST_OFDM_RATE;
 309                  mcs_rate->rate_n_flags = iwl_rates[index].plcp_siso |
 310                                           RATE_MCS_HT_MSK;
 311         } else {
 312                 if (index > IWL_LAST_OFDM_RATE)
 313                         index = IWL_LAST_OFDM_RATE;
 314                 mcs_rate->rate_n_flags = iwl_rates[index].plcp_mimo |
 315                                          RATE_MCS_HT_MSK;
 316         }
 317
 318         switch (tbl->antenna_type) {
 319         case ANT_BOTH:
 320                 mcs_rate->rate_n_flags |= RATE_MCS_ANT_AB_MSK;
 321                 break;
 322         case ANT_MAIN:
 323                 mcs_rate->rate_n_flags |= RATE_MCS_ANT_A_MSK;
 324                 break;
 325         case ANT_AUX:
 326                 mcs_rate->rate_n_flags |= RATE_MCS_ANT_B_MSK;
 327                 break;
 328         case ANT_NONE:
 329                 break;
 330         }
 331
 332         if (is_legacy(tbl->lq_type))
 333                 return rc;
 334
 335         if (tbl->is_fat) {
 336                 if (tbl->is_dup)
 337                         mcs_rate->rate_n_flags |= RATE_MCS_DUP_MSK;
 338                 else
 339                         mcs_rate->rate_n_flags |= RATE_MCS_FAT_MSK;
 340         }
 341         if (tbl->is_SGI)
 342                 mcs_rate->rate_n_flags |= RATE_MCS_SGI_MSK;
 343
 344         if (use_green) {
 345                 mcs_rate->rate_n_flags |= RATE_MCS_GF_MSK;
 346                 if (is_siso(tbl->lq_type))
 347                         mcs_rate->rate_n_flags &= ~RATE_MCS_SGI_MSK;
 348         }
 349         return rc;
 350 }
 351
 352 static int rs_get_tbl_info_from_mcs(const struct iwl_rate *mcs_rate,
 353                                     int phymode, struct iwl_scale_tbl_info *tbl,
 354                                     int *rate_idx)
 355 {
 356         int index;
 357         u32 ant_msk;
 358
 359         index = iwl_rate_index_from_plcp(mcs_rate->rate_n_flags);
 360
 361         if (index  == IWL_RATE_INVALID) {
 362                 *rate_idx = -1;
 363                 return -1;
 364         }
 365         tbl->is_SGI = 0;
 366         tbl->is_fat = 0;
 367         tbl->is_dup = 0;
 368         tbl->antenna_type = ANT_BOTH;
 369
 370         if (!(mcs_rate->rate_n_flags & RATE_MCS_HT_MSK)) {
 371                 ant_msk = (mcs_rate->rate_n_flags & RATE_MCS_ANT_AB_MSK);
 372
 373                 if (ant_msk == RATE_MCS_ANT_AB_MSK)
 374                         tbl->lq_type = LQ_NONE;
 375                 else {
 376
 377                         if (phymode == MODE_IEEE80211A)
 378                                 tbl->lq_type = LQ_A;
 379                         else
 380                                 tbl->lq_type = LQ_G;
 381
 382                         if (mcs_rate->rate_n_flags & RATE_MCS_ANT_A_MSK)
 383                                 tbl->antenna_type = ANT_MAIN;
 384                         else
 385                                 tbl->antenna_type = ANT_AUX;
 386                 }
 387                 *rate_idx = index;
 388
 389         } else if (iwl_rate_get_rate(mcs_rate->rate_n_flags)
 390                                         <= IWL_RATE_SISO_60M_PLCP) {
 391                 tbl->lq_type = LQ_SISO;
 392
 393                 ant_msk = (mcs_rate->rate_n_flags & RATE_MCS_ANT_AB_MSK);
 394                 if (ant_msk == RATE_MCS_ANT_AB_MSK)
 395                         tbl->lq_type = LQ_NONE;
 396                 else {
 397                         if (mcs_rate->rate_n_flags & RATE_MCS_ANT_A_MSK)
 398                                 tbl->antenna_type = ANT_MAIN;
 399                         else
 400                                 tbl->antenna_type = ANT_AUX;
 401                 }
 402                 if (mcs_rate->rate_n_flags & RATE_MCS_SGI_MSK)
 403                         tbl->is_SGI = 1;
 404
 405                 if ((mcs_rate->rate_n_flags & RATE_MCS_FAT_MSK) ||
 406                     (mcs_rate->rate_n_flags & RATE_MCS_DUP_MSK))
 407                         tbl->is_fat = 1;
 408
 409                 if (mcs_rate->rate_n_flags & RATE_MCS_DUP_MSK)
 410                         tbl->is_dup = 1;
 411
 412                 *rate_idx = index;
 413         } else {
 414                 tbl->lq_type = LQ_MIMO;
 415                 if (mcs_rate->rate_n_flags & RATE_MCS_SGI_MSK)
 416                         tbl->is_SGI = 1;
 417
 418                 if ((mcs_rate->rate_n_flags & RATE_MCS_FAT_MSK) ||
 419                     (mcs_rate->rate_n_flags & RATE_MCS_DUP_MSK))
 420                         tbl->is_fat = 1;
 421
 422                 if (mcs_rate->rate_n_flags & RATE_MCS_DUP_MSK)
 423                         tbl->is_dup = 1;
 424                 *rate_idx = index;
 425         }
 426         return 0;
 427 }
 428
 429 static inline void rs_toggle_antenna(struct iwl_rate *new_rate,
 430                                      struct iwl_scale_tbl_info *tbl)
 431 {
 432         if (tbl->antenna_type == ANT_AUX) {
 433                 tbl->antenna_type = ANT_MAIN;
 434                 new_rate->rate_n_flags &= ~RATE_MCS_ANT_B_MSK;
 435                 new_rate->rate_n_flags |= RATE_MCS_ANT_A_MSK;
 436         } else {
 437                 tbl->antenna_type = ANT_AUX;
 438                 new_rate->rate_n_flags &= ~RATE_MCS_ANT_A_MSK;
 439                 new_rate->rate_n_flags |= RATE_MCS_ANT_B_MSK;
 440         }
 441 }
 442
 443 static inline s8 rs_use_green(struct iwl_priv *priv)
 444 {
 445         s8 rc = 0;
 446 #ifdef CONFIG_IWLWIFI_HT
 447         if (!priv->is_ht_enabled || !priv->current_assoc_ht.is_ht)
 448                 return 0;
 449
 450         if ((priv->current_assoc_ht.is_green_field) &&
 451             !(priv->current_assoc_ht.operating_mode & 0x4))
 452                 rc = 1;
 453 #endif  /*CONFIG_IWLWIFI_HT */
 454         return rc;
 455 }
 456
 457 /**
 458  * rs_get_supported_rates - get the available rates
 459  *
 460  * if management frame or broadcast frame only return
 461  * basic available rates.
 462  *
 463  */
 464 static void rs_get_supported_rates(struct iwl_rate_scale_priv *lq_data,
 465                                    struct ieee80211_hdr *hdr,
 466                                    enum iwl_table_type rate_type,
 467                                    u16 *data_rate)
 468 {
 469         if (is_legacy(rate_type))
 470                 *data_rate = lq_data->active_rate;
 471         else {
 472                 if (is_siso(rate_type))
 473                         *data_rate = lq_data->active_siso_rate;
 474                 else
 475                         *data_rate = lq_data->active_mimo_rate;
 476         }
 477
 478         if (hdr && is_multicast_ether_addr(hdr->addr1) &&
 479             lq_data->active_rate_basic)
 480                 *data_rate = lq_data->active_rate_basic;
 481 }
 482
 483 static u16 rs_get_adjacent_rate(u8 index, u16 rate_mask, int rate_type)
 484 {
 485         u8 high = IWL_RATE_INVALID;
 486         u8 low = IWL_RATE_INVALID;
 487
 488         /* 802.11A or ht walks to the next literal adjascent rate in
 489          * the rate table */
 490         if (is_a_band(rate_type) || !is_legacy(rate_type)) {
 491                 int i;
 492                 u32 mask;
 493
 494                 /* Find the previous rate that is in the rate mask */
 495                 i = index - 1;
 496                 for (mask = (1 << i); i >= 0; i--, mask >>= 1) {
 497                         if (rate_mask & mask) {
 498                                 low = i;
 499                                 break;
 500                         }
 501                 }
 502
 503                 /* Find the next rate that is in the rate mask */
 504                 i = index + 1;
 505                 for (mask = (1 << i); i < IWL_RATE_COUNT; i++, mask <<= 1) {
 506                         if (rate_mask & mask) {
 507                                 high = i;
 508                                 break;
 509                         }
 510                 }
 511
 512                 return (high << 8) | low;
 513         }
 514
 515         low = index;
 516         while (low != IWL_RATE_INVALID) {
 517                 low = iwl_rates[low].prev_rs;
 518                 if (low == IWL_RATE_INVALID)
 519                         break;
 520                 if (rate_mask & (1 << low))
 521                         break;
 522                 IWL_DEBUG_RATE("Skipping masked lower rate: %d\n", low);
 523         }
 524
 525         high = index;
 526         while (high != IWL_RATE_INVALID) {
 527                 high = iwl_rates[high].next_rs;
 528                 if (high == IWL_RATE_INVALID)
 529                         break;
 530                 if (rate_mask & (1 << high))
 531                         break;
 532                 IWL_DEBUG_RATE("Skipping masked higher rate: %d\n", high);
 533         }
 534
 535         return (high << 8) | low;
 536 }
 537
 538 static int rs_get_lower_rate(struct iwl_rate_scale_priv *lq_data,
 539                              struct iwl_scale_tbl_info *tbl, u8 scale_index,
 540                              u8 ht_possible, struct iwl_rate *mcs_rate,
 541                              struct sta_info *sta)
 542 {
 543         u8 is_green = lq_data->is_green;
 544         s32 low;
 545         u16 rate_mask;
 546         u16 high_low;
 547         u8 switch_to_legacy = 0;
 548
 549         /* check if we need to switch from HT to legacy rates.
 550          * assumption is that mandatory rates (1Mbps or 6Mbps)
 551          * are always supported (spec demand) */
 552         if (!is_legacy(tbl->lq_type) && (!ht_possible || !scale_index)) {
 553                 switch_to_legacy = 1;
 554                 scale_index = rs_ht_to_legacy[scale_index];
 555                 if (lq_data->phymode == MODE_IEEE80211A)
 556                         tbl->lq_type = LQ_A;
 557                 else
 558                         tbl->lq_type = LQ_G;
 559
 560                 if ((tbl->antenna_type == ANT_BOTH) ||
 561                     (tbl->antenna_type == ANT_NONE))
 562                         tbl->antenna_type = ANT_MAIN;
 563
 564                 tbl->is_fat = 0;
 565                 tbl->is_SGI = 0;
 566         }
 567
 568         rs_get_supported_rates(lq_data, NULL, tbl->lq_type, &rate_mask);
 569
 570         /* mask with station rate restriction */
 571         if (is_legacy(tbl->lq_type)) {
 572                 if (lq_data->phymode == (u8) MODE_IEEE80211A)
 573                         rate_mask  = (u16)(rate_mask &
 574                            (sta->supp_rates << IWL_FIRST_OFDM_RATE));
 575                 else
 576                         rate_mask = (u16)(rate_mask & sta->supp_rates);
 577         }
 578
 579         /* if we did switched from HT to legacy check current rate */
 580         if ((switch_to_legacy) &&
 581             (rate_mask & (1 << scale_index))) {
 582                 rs_mcs_from_tbl(mcs_rate, tbl, scale_index, is_green);
 583                 return 0;
 584         }
 585
 586         high_low = rs_get_adjacent_rate(scale_index, rate_mask, tbl->lq_type);
 587         low = high_low & 0xff;
 588
 589         if (low != IWL_RATE_INVALID)
 590                 rs_mcs_from_tbl(mcs_rate, tbl, low, is_green);
 591         else
 592                 rs_mcs_from_tbl(mcs_rate, tbl, scale_index, is_green);
 593
 594         return 0;
 595 }
 596
 597 static void rs_tx_status(void *priv_rate,
 598                          struct net_device *dev,
 599                          struct sk_buff *skb,
 600                          struct ieee80211_tx_status *tx_resp)
 601 {
 602         int status;
 603         u8 retries;
 604         int rs_index, index = 0;
 605         struct iwl_rate_scale_priv *lq;
 606         struct iwl_link_quality_cmd *table;
 607         struct sta_info *sta;
 608         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
 609         struct iwl_priv *priv = (struct iwl_priv *)priv_rate;
 610         struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr);
 611         struct iwl_rate_scale_data *window = NULL;
 612         struct iwl_rate_scale_data *search_win = NULL;
 613         struct iwl_rate tx_mcs;
 614         struct iwl_scale_tbl_info tbl_type;
 615         struct iwl_scale_tbl_info *curr_tbl, *search_tbl;
 616         u8 active_index = 0;
 617         u16 fc = le16_to_cpu(hdr->frame_control);
 618         s32 tpt = 0;
 619
 620         IWL_DEBUG_RATE("get frame ack response, update rate scale window\n");
 621
 622         if (!ieee80211_is_data(fc) || is_multicast_ether_addr(hdr->addr1))
 623                 return;
 624
 625         retries = tx_resp->retry_count;
 626
 627         if (retries > 15)
 628                 retries = 15;
 629
 630
 631         sta = sta_info_get(local, hdr->addr1);
 632
 633         if (!sta || !sta->rate_ctrl_priv) {
 634                 if (sta)
 635                         sta_info_put(sta);
 636                 return;
 637         }
 638
 639         lq = (struct iwl_rate_scale_priv *)sta->rate_ctrl_priv;
 640
 641         if (!priv->lq_mngr.lq_ready)
 642                 return;
 643
 644         if ((priv->iw_mode == IEEE80211_IF_TYPE_IBSS) && !lq->ibss_sta_added)
 645                 return;
 646
 647         table = &lq->lq;
 648         active_index = lq->active_tbl;
 649
 650         lq->antenna = (lq->valid_antenna & local->hw.conf.antenna_sel_tx);
 651         if (!lq->antenna)
 652                 lq->antenna = lq->valid_antenna;
 653
 654         lq->antenna = lq->valid_antenna;
 655         curr_tbl = &(lq->lq_info[active_index]);
 656         search_tbl = &(lq->lq_info[(1 - active_index)]);
 657         window = (struct iwl_rate_scale_data *)
 658             &(curr_tbl->win[0]);
 659         search_win = (struct iwl_rate_scale_data *)
 660             &(search_tbl->win[0]);
 661
 662         tx_mcs.rate_n_flags = tx_resp->control.tx_rate;
 663
 664         rs_get_tbl_info_from_mcs(&tx_mcs, priv->phymode,
 665                                   &tbl_type, &rs_index);
 666         if ((rs_index < 0) || (rs_index >= IWL_RATE_COUNT)) {
 667                 IWL_DEBUG_RATE("bad rate index at: %d rate 0x%X\n",
 668                              rs_index, tx_mcs.rate_n_flags);
 669                 sta_info_put(sta);
 670                 return;
 671         }
 672
 673         if (retries &&
 674             (tx_mcs.rate_n_flags !=
 675                                 le32_to_cpu(table->rs_table[0].rate_n_flags))) {
 676                 IWL_DEBUG_RATE("initial rate does not match 0x%x 0x%x\n",
 677                                 tx_mcs.rate_n_flags,
 678                                 le32_to_cpu(table->rs_table[0].rate_n_flags));
 679                 sta_info_put(sta);
 680                 return;
 681         }
 682
 683         while (retries) {
 684                 tx_mcs.rate_n_flags =
 685                     le32_to_cpu(table->rs_table[index].rate_n_flags);
 686                 rs_get_tbl_info_from_mcs(&tx_mcs, priv->phymode,
 687                                           &tbl_type, &rs_index);
 688
 689                 if ((tbl_type.lq_type == search_tbl->lq_type) &&
 690                     (tbl_type.antenna_type == search_tbl->antenna_type) &&
 691                     (tbl_type.is_SGI == search_tbl->is_SGI)) {
 692                         if (search_tbl->expected_tpt)
 693                                 tpt = search_tbl->expected_tpt[rs_index];
 694                         else
 695                                 tpt = 0;
 696                         rs_collect_tx_data(search_win,
 697                                             rs_index, tpt, 0);
 698                 } else if ((tbl_type.lq_type == curr_tbl->lq_type) &&
 699                            (tbl_type.antenna_type == curr_tbl->antenna_type) &&
 700                            (tbl_type.is_SGI == curr_tbl->is_SGI)) {
 701                         if (curr_tbl->expected_tpt)
 702                                 tpt = curr_tbl->expected_tpt[rs_index];
 703                         else
 704                                 tpt = 0;
 705                         rs_collect_tx_data(window, rs_index, tpt, 0);
 706                 }
 707                 if (lq->stay_in_tbl)
 708                         lq->total_failed++;
 709                 --retries;
 710                 index++;
 711
 712         }
 713
 714         if (!tx_resp->retry_count)
 715                 tx_mcs.rate_n_flags = tx_resp->control.tx_rate;
 716         else
 717                 tx_mcs.rate_n_flags =
 718                         le32_to_cpu(table->rs_table[index].rate_n_flags);
 719
 720         rs_get_tbl_info_from_mcs(&tx_mcs, priv->phymode,
 721                                   &tbl_type, &rs_index);
 722
 723         if (tx_resp->flags & IEEE80211_TX_STATUS_ACK)
 724                 status = 1;
 725         else
 726                 status = 0;
 727
 728         if ((tbl_type.lq_type == search_tbl->lq_type) &&
 729             (tbl_type.antenna_type == search_tbl->antenna_type) &&
 730             (tbl_type.is_SGI == search_tbl->is_SGI)) {
 731                 if (search_tbl->expected_tpt)
 732                         tpt = search_tbl->expected_tpt[rs_index];
 733                 else
 734                         tpt = 0;
 735                 rs_collect_tx_data(search_win,
 736                                     rs_index, tpt, status);
 737         } else if ((tbl_type.lq_type == curr_tbl->lq_type) &&
 738                    (tbl_type.antenna_type == curr_tbl->antenna_type) &&
 739                    (tbl_type.is_SGI == curr_tbl->is_SGI)) {
 740                 if (curr_tbl->expected_tpt)
 741                         tpt = curr_tbl->expected_tpt[rs_index];
 742                 else
 743                         tpt = 0;
 744                 rs_collect_tx_data(window, rs_index, tpt, status);
 745         }
 746
 747         if (lq->stay_in_tbl) {
 748                 if (status)
 749                         lq->total_success++;
 750                 else
 751                         lq->total_failed++;
 752         }
 753
 754         rs_rate_scale_perform(priv, dev, hdr, sta);
 755         sta_info_put(sta);
 756         return;
 757 }
 758
 759 static u8 rs_is_ant_connected(u8 valid_antenna,
 760                               enum iwl_antenna_type antenna_type)
 761 {
 762         if (antenna_type == ANT_AUX)
 763                 return ((valid_antenna & 0x2) ? 1:0);
 764         else if (antenna_type == ANT_MAIN)
 765                 return ((valid_antenna & 0x1) ? 1:0);
 766         else if (antenna_type == ANT_BOTH) {
 767                 if ((valid_antenna & 0x3) == 0x3)
 768                         return 1;
 769                 else
 770                         return 0;
 771         }
 772
 773         return 1;
 774 }
 775
 776 static u8 rs_is_other_ant_connected(u8 valid_antenna,
 777                                     enum iwl_antenna_type antenna_type)
 778 {
 779         if (antenna_type == ANT_AUX)
 780                 return (rs_is_ant_connected(valid_antenna, ANT_MAIN));
 781         else
 782                 return (rs_is_ant_connected(valid_antenna, ANT_AUX));
 783
 784         return 0;
 785 }
 786
 787 static void rs_set_stay_in_table(u8 is_legacy,
 788                                  struct iwl_rate_scale_priv *lq_data)
 789 {
 790         IWL_DEBUG_HT("we are staying in the same table\n");
 791         lq_data->stay_in_tbl = 1;
 792         if (is_legacy) {
 793                 lq_data->table_count_limit = IWL_LEGACY_TABLE_COUNT;
 794                 lq_data->max_failure_limit = IWL_LEGACY_FAILURE_LIMIT;
 795                 lq_data->max_success_limit = IWL_LEGACY_TABLE_COUNT;
 796         } else {
 797                 lq_data->table_count_limit = IWL_NONE_LEGACY_TABLE_COUNT;
 798                 lq_data->max_failure_limit = IWL_NONE_LEGACY_FAILURE_LIMIT;
 799                 lq_data->max_success_limit = IWL_NONE_LEGACY_SUCCESS_LIMIT;
 800         }
 801         lq_data->table_count = 0;
 802         lq_data->total_failed = 0;
 803         lq_data->total_success = 0;
 804 }
 805
 806 static void rs_get_expected_tpt_table(struct iwl_rate_scale_priv *lq_data,
 807                                       struct iwl_scale_tbl_info *tbl)
 808 {
 809         if (is_legacy(tbl->lq_type)) {
 810                 if (!is_a_band(tbl->lq_type))
 811                         tbl->expected_tpt = expected_tpt_G;
 812                 else
 813                         tbl->expected_tpt = expected_tpt_A;
 814         } else if (is_siso(tbl->lq_type)) {
 815                 if (tbl->is_fat && !lq_data->is_dup)
 816                         if (tbl->is_SGI)
 817                                 tbl->expected_tpt = expected_tpt_siso40MHzSGI;
 818                         else
 819                                 tbl->expected_tpt = expected_tpt_siso40MHz;
 820                 else if (tbl->is_SGI)
 821                         tbl->expected_tpt = expected_tpt_siso20MHzSGI;
 822                 else
 823                         tbl->expected_tpt = expected_tpt_siso20MHz;
 824
 825         } else if (is_mimo(tbl->lq_type)) {
 826                 if (tbl->is_fat && !lq_data->is_dup)
 827                         if (tbl->is_SGI)
 828                                 tbl->expected_tpt = expected_tpt_mimo40MHzSGI;
 829                         else
 830                                 tbl->expected_tpt = expected_tpt_mimo40MHz;
 831                 else if (tbl->is_SGI)
 832                         tbl->expected_tpt = expected_tpt_mimo20MHzSGI;
 833                 else
 834                         tbl->expected_tpt = expected_tpt_mimo20MHz;
 835         } else
 836                 tbl->expected_tpt = expected_tpt_G;
 837 }
 838
 839 #ifdef CONFIG_IWLWIFI_HT
 840 static s32 rs_get_best_rate(struct iwl_priv *priv,
 841                             struct iwl_rate_scale_priv *lq_data,
 842                             struct iwl_scale_tbl_info *tbl,
 843                             u16 rate_mask, s8 index, s8 rate)
 844 {
 845         struct iwl_scale_tbl_info *active_tbl =
 846             &(lq_data->lq_info[lq_data->active_tbl]);
 847         s32 new_rate, high, low, start_hi;
 848         s32 active_sr = active_tbl->win[index].success_ratio;
 849         s32 *tpt_tbl = tbl->expected_tpt;
 850         s32 active_tpt = active_tbl->expected_tpt[index];
 851         u16 high_low;
 852
 853         new_rate = high = low = start_hi = IWL_RATE_INVALID;
 854
 855         for (; ;) {
 856                 high_low = rs_get_adjacent_rate(rate, rate_mask, tbl->lq_type);
 857
 858                 low = high_low & 0xff;
 859                 high = (high_low >> 8) & 0xff;
 860
 861                 if ((((100 * tpt_tbl[rate]) > lq_data->last_tpt) &&
 862                      ((active_sr > IWL_RATE_DECREASE_TH) &&
 863                       (active_sr <= IWL_RATE_HIGH_TH) &&
 864                       (tpt_tbl[rate] <= active_tpt))) ||
 865                     ((active_sr >= IWL_RATE_SCALE_SWITCH) &&
 866                      (tpt_tbl[rate] > active_tpt))) {
 867
 868                         if (start_hi != IWL_RATE_INVALID) {
 869                                 new_rate = start_hi;
 870                                 break;
 871                         }
 872                         new_rate = rate;
 873                         if (low != IWL_RATE_INVALID)
 874                                 rate = low;
 875                         else
 876                                 break;
 877                 } else {
 878                         if (new_rate != IWL_RATE_INVALID)
 879                                 break;
 880                         else if (high != IWL_RATE_INVALID) {
 881                                 start_hi = high;
 882                                 rate = high;
 883                         } else {
 884                                 new_rate = rate;
 885                                 break;
 886                         }
 887                 }
 888         }
 889
 890         return new_rate;
 891 }
 892 #endif                          /* CONFIG_IWLWIFI_HT */
 893
 894 static inline u8 rs_is_both_ant_supp(u8 valid_antenna)
 895 {
 896         return (rs_is_ant_connected(valid_antenna, ANT_BOTH));
 897 }
 898
 899 static int rs_switch_to_mimo(struct iwl_priv *priv,
 900                              struct iwl_rate_scale_priv *lq_data,
 901                              struct iwl_scale_tbl_info *tbl, int index)
 902 {
 903         int rc = -1;
 904 #ifdef CONFIG_IWLWIFI_HT
 905         u16 rate_mask;
 906         s32 rate;
 907         s8 is_green = lq_data->is_green;
 908
 909         if (!priv->is_ht_enabled || !priv->current_assoc_ht.is_ht)
 910                 return -1;
 911
 912         IWL_DEBUG_HT("LQ: try to switch to MIMO\n");
 913         tbl->lq_type = LQ_MIMO;
 914         rs_get_supported_rates(lq_data, NULL, tbl->lq_type,
 915                                 &rate_mask);
 916
 917         if (priv->current_assoc_ht.tx_mimo_ps_mode == IWL_MIMO_PS_STATIC)
 918                 return -1;
 919
 920         if (!rs_is_both_ant_supp(lq_data->antenna))
 921                 return -1;
 922
 923         rc = 0;
 924         tbl->is_dup = lq_data->is_dup;
 925         tbl->action = 0;
 926         if (priv->current_channel_width == IWL_CHANNEL_WIDTH_40MHZ)
 927                 tbl->is_fat = 1;
 928         else
 929                 tbl->is_fat = 0;
 930
 931         if (tbl->is_fat) {
 932                 if (priv->current_assoc_ht.sgf & HT_SHORT_GI_40MHZ_ONLY)
 933                         tbl->is_SGI = 1;
 934                 else
 935                         tbl->is_SGI = 0;
 936         } else if (priv->current_assoc_ht.sgf & HT_SHORT_GI_20MHZ_ONLY)
 937                 tbl->is_SGI = 1;
 938         else
 939                 tbl->is_SGI = 0;
 940
 941         rs_get_expected_tpt_table(lq_data, tbl);
 942
 943         rate = rs_get_best_rate(priv, lq_data, tbl, rate_mask, index, index);
 944
 945         IWL_DEBUG_HT("LQ: MIMO best rate %d mask %X\n", rate, rate_mask);
 946         if ((rate == IWL_RATE_INVALID) || !((1 << rate) & rate_mask))
 947                 return -1;
 948         rs_mcs_from_tbl(&tbl->current_rate, tbl, rate, is_green);
 949
 950         IWL_DEBUG_HT("LQ: Switch to new mcs %X index is green %X\n",
 951                      tbl->current_rate.rate_n_flags, is_green);
 952
 953 #endif                          /*CONFIG_IWLWIFI_HT */
 954         return rc;
 955 }
 956
 957 static int rs_switch_to_siso(struct iwl_priv *priv,
 958                              struct iwl_rate_scale_priv *lq_data,
 959                              struct iwl_scale_tbl_info *tbl, int index)
 960 {
 961         int rc = -1;
 962 #ifdef CONFIG_IWLWIFI_HT
 963         u16 rate_mask;
 964         u8 is_green = lq_data->is_green;
 965         s32 rate;
 966
 967         IWL_DEBUG_HT("LQ: try to switch to SISO\n");
 968         if (!priv->is_ht_enabled || !priv->current_assoc_ht.is_ht)
 969                 return -1;
 970
 971         rc = 0;
 972         tbl->is_dup = lq_data->is_dup;
 973         tbl->lq_type = LQ_SISO;
 974         tbl->action = 0;
 975         rs_get_supported_rates(lq_data, NULL, tbl->lq_type,
 976                                 &rate_mask);
 977
 978         if (priv->current_channel_width == IWL_CHANNEL_WIDTH_40MHZ)
 979                 tbl->is_fat = 1;
 980         else
 981                 tbl->is_fat = 0;
 982
 983         if (tbl->is_fat) {
 984                 if (priv->current_assoc_ht.sgf & HT_SHORT_GI_40MHZ_ONLY)
 985                         tbl->is_SGI = 1;
 986                 else
 987                         tbl->is_SGI = 0;
 988         } else if (priv->current_assoc_ht.sgf & HT_SHORT_GI_20MHZ_ONLY)
 989                 tbl->is_SGI = 1;
 990         else
 991                 tbl->is_SGI = 0;
 992
 993         if (is_green)
 994                 tbl->is_SGI = 0;
 995
 996         rs_get_expected_tpt_table(lq_data, tbl);
 997         rate = rs_get_best_rate(priv, lq_data, tbl, rate_mask, index, index);
 998
 999         IWL_DEBUG_HT("LQ: get best rate %d mask %X\n", rate, rate_mask);
1000         if ((rate == IWL_RATE_INVALID) || !((1 << rate) & rate_mask)) {
1001                 IWL_DEBUG_HT("can not switch with index %d rate mask %x\n",
1002                              rate, rate_mask);
1003                 return -1;
1004         }
1005         rs_mcs_from_tbl(&tbl->current_rate, tbl, rate, is_green);
1006         IWL_DEBUG_HT("LQ: Switch to new mcs %X index is green %X\n",
1007                      tbl->current_rate.rate_n_flags, is_green);
1008
1009 #endif                          /*CONFIG_IWLWIFI_HT */
1010         return rc;
1011 }
1012
1013 static int rs_move_legacy_other(struct iwl_priv *priv,
1014                                 struct iwl_rate_scale_priv *lq_data,
1015                                 int index)
1016 {
1017         int rc = 0;
1018         struct iwl_scale_tbl_info *tbl =
1019             &(lq_data->lq_info[lq_data->active_tbl]);
1020         struct iwl_scale_tbl_info *search_tbl =
1021             &(lq_data->lq_info[(1 - lq_data->active_tbl)]);
1022         struct iwl_rate_scale_data *window = &(tbl->win[index]);
1023         u32 sz = (sizeof(struct iwl_scale_tbl_info) -
1024                   (sizeof(struct iwl_rate_scale_data) * IWL_RATE_COUNT));
1025         u8 start_action = tbl->action;
1026
1027         for (; ;) {
1028                 switch (tbl->action) {
1029                 case IWL_LEGACY_SWITCH_ANTENNA:
1030                         IWL_DEBUG_HT("LQ Legacy switch Antenna\n");
1031
1032                         search_tbl->lq_type = LQ_NONE;
1033                         lq_data->action_counter++;
1034                         if (window->success_ratio >= IWL_RS_GOOD_RATIO)
1035                                 break;
1036                         if (!rs_is_other_ant_connected(lq_data->antenna,
1037                                                         tbl->antenna_type))
1038                                 break;
1039
1040                         memcpy(search_tbl, tbl, sz);
1041
1042                         rs_toggle_antenna(&(search_tbl->current_rate),
1043                                            search_tbl);
1044                         rs_get_expected_tpt_table(lq_data, search_tbl);
1045                         lq_data->search_better_tbl = 1;
1046                         goto out;
1047
1048                 case IWL_LEGACY_SWITCH_SISO:
1049                         IWL_DEBUG_HT("LQ: Legacy switch to SISO\n");
1050                         memcpy(search_tbl, tbl, sz);
1051                         search_tbl->lq_type = LQ_SISO;
1052                         search_tbl->is_SGI = 0;
1053                         search_tbl->is_fat = 0;
1054                         rc = rs_switch_to_siso(priv, lq_data, search_tbl,
1055                                                index);
1056                         if (!rc) {
1057                                 lq_data->search_better_tbl = 1;
1058                                 lq_data->action_counter = 0;
1059                         }
1060                         if (!rc)
1061                                 goto out;
1062
1063                         break;
1064                 case IWL_LEGACY_SWITCH_MIMO:
1065                         IWL_DEBUG_HT("LQ: Legacy switch MIMO\n");
1066                         memcpy(search_tbl, tbl, sz);
1067                         search_tbl->lq_type = LQ_MIMO;
1068                         search_tbl->is_SGI = 0;
1069                         search_tbl->is_fat = 0;
1070                         search_tbl->antenna_type = ANT_BOTH;
1071                         rc = rs_switch_to_mimo(priv, lq_data, search_tbl,
1072                                                index);
1073                         if (!rc) {
1074                                 lq_data->search_better_tbl = 1;
1075                                 lq_data->action_counter = 0;
1076                         }
1077                         if (!rc)
1078                                 goto out;
1079                         break;
1080                 }
1081                 tbl->action++;
1082                 if (tbl->action > IWL_LEGACY_SWITCH_MIMO)
1083                         tbl->action = IWL_LEGACY_SWITCH_ANTENNA;
1084
1085                 if (tbl->action == start_action)
1086                         break;
1087
1088         }
1089         return 0;
1090
1091  out:
1092         tbl->action++;
1093         if (tbl->action > IWL_LEGACY_SWITCH_MIMO)
1094                 tbl->action = IWL_LEGACY_SWITCH_ANTENNA;
1095         return 0;
1096
1097 }
1098
1099 static int rs_move_siso_to_other(struct iwl_priv *priv,
1100                                  struct iwl_rate_scale_priv *lq_data,
1101                                  int index)
1102 {
1103         int rc = -1;
1104         u8 is_green = lq_data->is_green;
1105         struct iwl_scale_tbl_info *tbl =
1106             &(lq_data->lq_info[lq_data->active_tbl]);
1107         struct iwl_scale_tbl_info *search_tbl =
1108             &(lq_data->lq_info[(1 - lq_data->active_tbl)]);
1109         struct iwl_rate_scale_data *window = &(tbl->win[index]);
1110         u32 sz = (sizeof(struct iwl_scale_tbl_info) -
1111                   (sizeof(struct iwl_rate_scale_data) * IWL_RATE_COUNT));
1112         u8 start_action = tbl->action;
1113
1114         for (;;) {
1115                 lq_data->action_counter++;
1116                 switch (tbl->action) {
1117                 case IWL_SISO_SWITCH_ANTENNA:
1118                         IWL_DEBUG_HT("LQ: SISO SWITCH ANTENNA SISO\n");
1119                         search_tbl->lq_type = LQ_NONE;
1120                         if (window->success_ratio >= IWL_RS_GOOD_RATIO)
1121                                 break;
1122                         if (!rs_is_other_ant_connected(lq_data->antenna,
1123                                                        tbl->antenna_type))
1124                                 break;
1125
1126                         memcpy(search_tbl, tbl, sz);
1127                         search_tbl->action = IWL_SISO_SWITCH_MIMO;
1128                         rs_toggle_antenna(&(search_tbl->current_rate),
1129                                            search_tbl);
1130                         lq_data->search_better_tbl = 1;
1131
1132                         goto out;
1133
1134                 case IWL_SISO_SWITCH_MIMO:
1135                         IWL_DEBUG_HT("LQ: SISO SWITCH TO MIMO FROM SISO\n");
1136                         memcpy(search_tbl, tbl, sz);
1137                         search_tbl->lq_type = LQ_MIMO;
1138                         search_tbl->is_SGI = 0;
1139                         search_tbl->is_fat = 0;
1140                         search_tbl->antenna_type = ANT_BOTH;
1141                         rc = rs_switch_to_mimo(priv, lq_data, search_tbl,
1142                                                index);
1143                         if (!rc)
1144                                 lq_data->search_better_tbl = 1;
1145
1146                         if (!rc)
1147                                 goto out;
1148                         break;
1149                 case IWL_SISO_SWITCH_GI:
1150                         IWL_DEBUG_HT("LQ: SISO SWITCH TO GI\n");
1151                         memcpy(search_tbl, tbl, sz);
1152                         search_tbl->action = 0;
1153                         if (search_tbl->is_SGI)
1154                                 search_tbl->is_SGI = 0;
1155                         else if (!is_green)
1156                                 search_tbl->is_SGI = 1;
1157                         else
1158                                 break;
1159                         lq_data->search_better_tbl = 1;
1160                         if ((tbl->lq_type == LQ_SISO) &&
1161                             (tbl->is_SGI)) {
1162                                 s32 tpt = lq_data->last_tpt / 100;
1163                                 if (((!tbl->is_fat) &&
1164                                      (tpt >= expected_tpt_siso20MHz[index])) ||
1165                                     ((tbl->is_fat) &&
1166                                      (tpt >= expected_tpt_siso40MHz[index])))
1167                                         lq_data->search_better_tbl = 0;
1168                         }
1169                         rs_get_expected_tpt_table(lq_data, search_tbl);
1170                         rs_mcs_from_tbl(&search_tbl->current_rate,
1171                                              search_tbl, index, is_green);
1172                         goto out;
1173                 }
1174                 tbl->action++;
1175                 if (tbl->action > IWL_SISO_SWITCH_GI)
1176                         tbl->action = IWL_SISO_SWITCH_ANTENNA;
1177
1178                 if (tbl->action == start_action)
1179                         break;
1180         }
1181         return 0;
1182
1183  out:
1184         tbl->action++;
1185         if (tbl->action > IWL_SISO_SWITCH_GI)
1186                 tbl->action = IWL_SISO_SWITCH_ANTENNA;
1187         return 0;
1188 }
1189
1190 static int rs_move_mimo_to_other(struct iwl_priv *priv,
1191                                  struct iwl_rate_scale_priv *lq_data,
1192                                  int index)
1193 {
1194         int rc = -1;
1195         s8 is_green = lq_data->is_green;
1196         struct iwl_scale_tbl_info *tbl =
1197             &(lq_data->lq_info[lq_data->active_tbl]);
1198         struct iwl_scale_tbl_info *search_tbl =
1199             &(lq_data->lq_info[(1 - lq_data->active_tbl)]);
1200         u32 sz = (sizeof(struct iwl_scale_tbl_info) -
1201                   (sizeof(struct iwl_rate_scale_data) * IWL_RATE_COUNT));
1202         u8 start_action = tbl->action;
1203
1204         for (;;) {
1205                 lq_data->action_counter++;
1206                 switch (tbl->action) {
1207                 case IWL_MIMO_SWITCH_ANTENNA_A:
1208                 case IWL_MIMO_SWITCH_ANTENNA_B:
1209                         IWL_DEBUG_HT("LQ: MIMO SWITCH TO SISO\n");
1210                         memcpy(search_tbl, tbl, sz);
1211                         search_tbl->lq_type = LQ_SISO;
1212                         search_tbl->is_SGI = 0;
1213                         search_tbl->is_fat = 0;
1214                         if (tbl->action == IWL_MIMO_SWITCH_ANTENNA_A)
1215                                 search_tbl->antenna_type = ANT_MAIN;
1216                         else
1217                                 search_tbl->antenna_type = ANT_AUX;
1218
1219                         rc = rs_switch_to_siso(priv, lq_data, search_tbl,
1220                                                index);
1221                         if (!rc) {
1222                                 lq_data->search_better_tbl = 1;
1223                                 goto out;
1224                         }
1225                         break;
1226
1227                 case IWL_MIMO_SWITCH_GI:
1228                         IWL_DEBUG_HT("LQ: MIMO SWITCH TO GI\n");
1229                         memcpy(search_tbl, tbl, sz);
1230                         search_tbl->lq_type = LQ_MIMO;
1231                         search_tbl->antenna_type = ANT_BOTH;
1232                         search_tbl->action = 0;
1233                         if (search_tbl->is_SGI)
1234                                 search_tbl->is_SGI = 0;
1235                         else
1236                                 search_tbl->is_SGI = 1;
1237                         lq_data->search_better_tbl = 1;
1238                         if ((tbl->lq_type == LQ_MIMO) &&
1239                             (tbl->is_SGI)) {
1240                                 s32 tpt = lq_data->last_tpt / 100;
1241                                 if (((!tbl->is_fat) &&
1242                                      (tpt >= expected_tpt_mimo20MHz[index])) ||
1243                                     ((tbl->is_fat) &&
1244                                      (tpt >= expected_tpt_mimo40MHz[index])))
1245                                         lq_data->search_better_tbl = 0;
1246                         }
1247                         rs_get_expected_tpt_table(lq_data, search_tbl);
1248                         rs_mcs_from_tbl(&search_tbl->current_rate,
1249                                              search_tbl, index, is_green);
1250                         goto out;
1251
1252                 }
1253                 tbl->action++;
1254                 if (tbl->action > IWL_MIMO_SWITCH_GI)
1255                         tbl->action = IWL_MIMO_SWITCH_ANTENNA_A;
1256
1257                 if (tbl->action == start_action)
1258                         break;
1259         }
1260
1261         return 0;
1262  out:
1263         tbl->action++;
1264         if (tbl->action > IWL_MIMO_SWITCH_GI)
1265                 tbl->action = IWL_MIMO_SWITCH_ANTENNA_A;
1266         return 0;
1267
1268 }
1269
1270 static void rs_stay_in_table(struct iwl_rate_scale_priv *lq_data)
1271 {
1272         struct iwl_scale_tbl_info *tbl;
1273         int i;
1274         int active_tbl;
1275         int flush_interval_passed = 0;
1276
1277         active_tbl = lq_data->active_tbl;
1278
1279         tbl = &(lq_data->lq_info[active_tbl]);
1280
1281         if (lq_data->stay_in_tbl) {
1282
1283                 if (lq_data->flush_timer)
1284                         flush_interval_passed =
1285                             time_after(jiffies,
1286                                        (unsigned long)(lq_data->flush_timer +
1287                                         IWL_RATE_SCALE_FLUSH_INTVL));
1288
1289                 flush_interval_passed = 0;
1290                 if ((lq_data->total_failed > lq_data->max_failure_limit) ||
1291                     (lq_data->total_success > lq_data->max_success_limit) ||
1292                     ((!lq_data->search_better_tbl) && (lq_data->flush_timer)
1293                      && (flush_interval_passed))) {
1294                         IWL_DEBUG_HT("LQ: stay is expired %d %d %d\n:",
1295                                      lq_data->total_failed,
1296                                      lq_data->total_success,
1297                                      flush_interval_passed);
1298                         lq_data->stay_in_tbl = 0;
1299                         lq_data->total_failed = 0;
1300                         lq_data->total_success = 0;
1301                         lq_data->flush_timer = 0;
1302                 } else if (lq_data->table_count > 0) {
1303                         lq_data->table_count++;
1304                         if (lq_data->table_count >=
1305                             lq_data->table_count_limit) {
1306                                 lq_data->table_count = 0;
1307
1308                                 IWL_DEBUG_HT("LQ: stay in table clear win\n");
1309                                 for (i = 0; i < IWL_RATE_COUNT; i++)
1310                                         rs_rate_scale_clear_window(
1311                                                 &(tbl->win[i]));
1312                         }
1313                 }
1314
1315                 if (!lq_data->stay_in_tbl) {
1316                         for (i = 0; i < IWL_RATE_COUNT; i++)
1317                                 rs_rate_scale_clear_window(&(tbl->win[i]));
1318                 }
1319         }
1320 }
1321
1322 static void rs_rate_scale_perform(struct iwl_priv *priv,
1323                                   struct net_device *dev,
1324                                   struct ieee80211_hdr *hdr,
1325                                   struct sta_info *sta)
1326 {
1327         int low = IWL_RATE_INVALID;
1328         int high = IWL_RATE_INVALID;
1329         int index;
1330         int i;
1331         struct iwl_rate_scale_data *window = NULL;
1332         int current_tpt = IWL_INVALID_VALUE;
1333         int low_tpt = IWL_INVALID_VALUE;
1334         int high_tpt = IWL_INVALID_VALUE;
1335         u32 fail_count;
1336         s8 scale_action = 0;
1337         u16 fc, rate_mask;
1338         u8 update_lq = 0;
1339         struct iwl_rate_scale_priv *lq_data;
1340         struct iwl_scale_tbl_info *tbl, *tbl1;
1341         u16 rate_scale_index_msk = 0;
1342         struct iwl_rate mcs_rate;
1343         u8 is_green = 0;
1344         u8 active_tbl = 0;
1345         u8 done_search = 0;
1346         u16 high_low;
1347
1348         IWL_DEBUG_RATE("rate scale calculate new rate for skb\n");
1349
1350         fc = le16_to_cpu(hdr->frame_control);
1351         if (!ieee80211_is_data(fc) || is_multicast_ether_addr(hdr->addr1)) {
1352                 /* Send management frames and broadcast/multicast data using
1353                  * lowest rate. */
1354                 /* TODO: this could probably be improved.. */
1355                 return;
1356         }
1357
1358         if (!sta || !sta->rate_ctrl_priv)
1359                 return;
1360
1361         if (!priv->lq_mngr.lq_ready) {
1362                 IWL_DEBUG_RATE("still rate scaling not ready\n");
1363                 return;
1364         }
1365         lq_data = (struct iwl_rate_scale_priv *)sta->rate_ctrl_priv;
1366
1367         if (!lq_data->search_better_tbl)
1368                 active_tbl = lq_data->active_tbl;
1369         else
1370                 active_tbl = 1 - lq_data->active_tbl;
1371
1372         tbl = &(lq_data->lq_info[active_tbl]);
1373         is_green = lq_data->is_green;
1374
1375         index = sta->last_txrate;
1376
1377         IWL_DEBUG_RATE("Rate scale index %d for type %d\n", index,
1378                        tbl->lq_type);
1379
1380         rs_get_supported_rates(lq_data, hdr, tbl->lq_type,
1381                                 &rate_mask);
1382
1383         IWL_DEBUG_RATE("mask 0x%04X \n", rate_mask);
1384
1385         /* mask with station rate restriction */
1386         if (is_legacy(tbl->lq_type)) {
1387                 if (lq_data->phymode == (u8) MODE_IEEE80211A)
1388                         rate_scale_index_msk = (u16) (rate_mask &
1389                                 (sta->supp_rates << IWL_FIRST_OFDM_RATE));
1390                 else
1391                         rate_scale_index_msk = (u16) (rate_mask &
1392                                                       sta->supp_rates);
1393
1394         } else
1395                 rate_scale_index_msk = rate_mask;
1396
1397         if (!rate_scale_index_msk)
1398                 rate_scale_index_msk = rate_mask;
1399
1400         if (index < 0 || !((1 << index) & rate_scale_index_msk)) {
1401                 index = IWL_INVALID_VALUE;
1402                 update_lq = 1;
1403
1404                 /* get the lowest availabe rate */
1405                 for (i = 0; i <= IWL_RATE_COUNT; i++) {
1406                         if ((1 << i) & rate_scale_index_msk)
1407                                 index = i;
1408                 }
1409
1410                 if (index == IWL_INVALID_VALUE) {
1411                         IWL_WARNING("Can not find a suitable rate\n");
1412                         return;
1413                 }
1414         }
1415
1416         if (!tbl->expected_tpt)
1417                 rs_get_expected_tpt_table(lq_data, tbl);
1418
1419         window = &(tbl->win[index]);
1420
1421         fail_count = window->counter - window->success_counter;
1422         if (((fail_count < IWL_RATE_MIN_FAILURE_TH) &&
1423              (window->success_counter < IWL_RATE_MIN_SUCCESS_TH))
1424             || (tbl->expected_tpt == NULL)) {
1425                 IWL_DEBUG_RATE("LQ: still below TH succ %d total %d "
1426                                "for index %d\n",
1427                                window->success_counter, window->counter, index);
1428                 window->average_tpt = IWL_INVALID_VALUE;
1429                 rs_stay_in_table(lq_data);
1430                 if (update_lq) {
1431                         rs_mcs_from_tbl(&mcs_rate, tbl, index, is_green);
1432                         rs_fill_link_cmd(lq_data, &mcs_rate, &lq_data->lq, sta);
1433                         rs_send_lq_cmd(priv, &lq_data->lq, CMD_ASYNC);
1434                 }
1435                 goto out;
1436
1437         } else
1438                 window->average_tpt = ((window->success_ratio *
1439                                         tbl->expected_tpt[index] + 64) / 128);
1440
1441         if (lq_data->search_better_tbl) {
1442                 int success_limit = IWL_RATE_SCALE_SWITCH;
1443
1444                 if ((window->success_ratio > success_limit) ||
1445                     (window->average_tpt > lq_data->last_tpt)) {
1446                         if (!is_legacy(tbl->lq_type)) {
1447                                 IWL_DEBUG_HT("LQ: we are switching to HT"
1448                                              " rate suc %d current tpt %d"
1449                                              " old tpt %d\n",
1450                                              window->success_ratio,
1451                                              window->average_tpt,
1452                                              lq_data->last_tpt);
1453                                 lq_data->enable_counter = 1;
1454                         }
1455                         lq_data->active_tbl = active_tbl;
1456                         current_tpt = window->average_tpt;
1457                 } else {
1458                         tbl->lq_type = LQ_NONE;
1459                         active_tbl = lq_data->active_tbl;
1460                         tbl = &(lq_data->lq_info[active_tbl]);
1461
1462                         index = iwl_rate_index_from_plcp(
1463                                 tbl->current_rate.rate_n_flags);
1464
1465                         update_lq = 1;
1466                         current_tpt = lq_data->last_tpt;
1467                         IWL_DEBUG_HT("XXY GO BACK TO OLD TABLE\n");
1468                 }
1469                 lq_data->search_better_tbl = 0;
1470                 done_search = 1;
1471                 goto lq_update;
1472         }
1473
1474         high_low = rs_get_adjacent_rate(index, rate_scale_index_msk,
1475                                         tbl->lq_type);
1476         low = high_low & 0xff;
1477         high = (high_low >> 8) & 0xff;
1478
1479         current_tpt = window->average_tpt;
1480
1481         if (low != IWL_RATE_INVALID)
1482                 low_tpt = tbl->win[low].average_tpt;
1483
1484         if (high != IWL_RATE_INVALID)
1485                 high_tpt = tbl->win[high].average_tpt;
1486
1487
1488         scale_action = 1;
1489
1490         if ((window->success_ratio <= IWL_RATE_DECREASE_TH) ||
1491             (current_tpt == 0)) {
1492                 IWL_DEBUG_RATE("decrease rate because of low success_ratio\n");
1493                 scale_action = -1;
1494         } else if ((low_tpt == IWL_INVALID_VALUE) &&
1495                    (high_tpt == IWL_INVALID_VALUE))
1496                 scale_action = 1;
1497         else if ((low_tpt != IWL_INVALID_VALUE) &&
1498                  (high_tpt != IWL_INVALID_VALUE) &&
1499                  (low_tpt < current_tpt) &&
1500                  (high_tpt < current_tpt))
1501                 scale_action = 0;
1502         else {
1503                 if (high_tpt != IWL_INVALID_VALUE) {
1504                         if (high_tpt > current_tpt)
1505                                 scale_action = 1;
1506                         else {
1507                                 IWL_DEBUG_RATE
1508                                     ("decrease rate because of high tpt\n");
1509                                 scale_action = -1;
1510                         }
1511                 } else if (low_tpt != IWL_INVALID_VALUE) {
1512                         if (low_tpt > current_tpt) {
1513                                 IWL_DEBUG_RATE
1514                                     ("decrease rate because of low tpt\n");
1515                                 scale_action = -1;
1516                         } else
1517                                 scale_action = 1;
1518                 }
1519         }
1520
1521         if (scale_action == -1) {
1522                 if ((low != IWL_RATE_INVALID) &&
1523                     ((window->success_ratio > IWL_RATE_HIGH_TH) ||
1524                      (current_tpt > (100 * tbl->expected_tpt[low]))))
1525                         scale_action = 0;
1526         } else if ((scale_action == 1) &&
1527                    (window->success_ratio < IWL_RATE_INCREASE_TH))
1528                 scale_action = 0;
1529
1530         switch (scale_action) {
1531         case -1:
1532                 if (low != IWL_RATE_INVALID) {
1533                         update_lq = 1;
1534                         index = low;
1535                 }
1536                 break;
1537         case 1:
1538                 if (high != IWL_RATE_INVALID) {
1539                         update_lq = 1;
1540                         index = high;
1541                 }
1542
1543                 break;
1544         case 0:
1545         default:
1546                 break;
1547         }
1548
1549         IWL_DEBUG_HT("choose rate scale index %d action %d low %d "
1550                     "high %d type %d\n",
1551                      index, scale_action, low, high, tbl->lq_type);
1552
1553  lq_update:
1554         if (update_lq) {
1555                 rs_mcs_from_tbl(&mcs_rate, tbl, index, is_green);
1556                 rs_fill_link_cmd(lq_data, &mcs_rate, &lq_data->lq, sta);
1557                 rs_send_lq_cmd(priv, &lq_data->lq, CMD_ASYNC);
1558         }
1559         rs_stay_in_table(lq_data);
1560
1561         if (!update_lq && !done_search && !lq_data->stay_in_tbl) {
1562                 lq_data->last_tpt = current_tpt;
1563
1564                 if (is_legacy(tbl->lq_type))
1565                         rs_move_legacy_other(priv, lq_data, index);
1566                 else if (is_siso(tbl->lq_type))
1567                         rs_move_siso_to_other(priv, lq_data, index);
1568                 else
1569                         rs_move_mimo_to_other(priv, lq_data, index);
1570
1571                 if (lq_data->search_better_tbl) {
1572                         tbl = &(lq_data->lq_info[(1 - lq_data->active_tbl)]);
1573                         for (i = 0; i < IWL_RATE_COUNT; i++)
1574                                 rs_rate_scale_clear_window(&(tbl->win[i]));
1575
1576                         index = iwl_rate_index_from_plcp(
1577                                         tbl->current_rate.rate_n_flags);
1578
1579                         IWL_DEBUG_HT("Switch current  mcs: %X index: %d\n",
1580                                      tbl->current_rate.rate_n_flags, index);
1581                         rs_fill_link_cmd(lq_data, &tbl->current_rate,
1582                                          &(lq_data->lq), sta);
1583                         rs_send_lq_cmd(priv, &lq_data->lq, CMD_ASYNC);
1584                 }
1585                 tbl1 = &(lq_data->lq_info[lq_data->active_tbl]);
1586
1587                 if (is_legacy(tbl1->lq_type) &&
1588 #ifdef CONFIG_IWLWIFI_HT
1589                     !priv->current_assoc_ht.is_ht &&
1590 #endif
1591                     (lq_data->action_counter >= 1)) {
1592                         lq_data->action_counter = 0;
1593                         IWL_DEBUG_HT("LQ: STAY in legacy table\n");
1594                         rs_set_stay_in_table(1, lq_data);
1595                 }
1596
1597                 if (lq_data->enable_counter &&
1598                     (lq_data->action_counter >= IWL_ACTION_LIMIT)) {
1599 #ifdef CONFIG_IWLWIFI_HT_AGG
1600                         if ((lq_data->last_tpt > TID_AGG_TPT_THREHOLD) &&
1601                             (priv->lq_mngr.agg_ctrl.auto_agg)) {
1602                                 priv->lq_mngr.agg_ctrl.tid_retry =
1603                                     TID_ALL_SPECIFIED;
1604                                 schedule_work(&priv->agg_work);
1605                         }
1606 #endif /*CONFIG_IWLWIFI_HT_AGG */
1607                         lq_data->action_counter = 0;
1608                         rs_set_stay_in_table(0, lq_data);
1609                 }
1610         } else {
1611                 if ((!update_lq) && (!done_search) && (!lq_data->flush_timer))
1612                         lq_data->flush_timer = jiffies;
1613         }
1614
1615 out:
1616         rs_mcs_from_tbl(&tbl->current_rate, tbl, index, is_green);
1617         i = index;
1618         sta->last_txrate = i;
1619
1620         /* sta->txrate is an index to A mode rates which start
1621          * at IWL_FIRST_OFDM_RATE
1622          */
1623         if (lq_data->phymode == (u8) MODE_IEEE80211A)
1624                 sta->txrate = i - IWL_FIRST_OFDM_RATE;
1625         else
1626                 sta->txrate = i;
1627
1628         return;
1629 }
1630
1631
1632 static void rs_initialize_lq(struct iwl_priv *priv,
1633                              struct sta_info *sta)
1634 {
1635         int i;
1636         struct iwl_rate_scale_priv *lq;
1637         struct iwl_scale_tbl_info *tbl;
1638         u8 active_tbl = 0;
1639         int rate_idx;
1640         u8 use_green = rs_use_green(priv);
1641         struct iwl_rate mcs_rate;
1642
1643         if (!sta || !sta->rate_ctrl_priv)
1644                 goto out;
1645
1646         lq = (struct iwl_rate_scale_priv *)sta->rate_ctrl_priv;
1647         i = sta->last_txrate;
1648
1649         if ((lq->lq.sta_id == 0xff) &&
1650             (priv->iw_mode == IEEE80211_IF_TYPE_IBSS))
1651                 goto out;
1652
1653         if (!lq->search_better_tbl)
1654                 active_tbl = lq->active_tbl;
1655         else
1656                 active_tbl = 1 - lq->active_tbl;
1657
1658         tbl = &(lq->lq_info[active_tbl]);
1659
1660         if ((i < 0) || (i >= IWL_RATE_COUNT))
1661                 i = 0;
1662
1663         mcs_rate.rate_n_flags = iwl_rates[i].plcp ;
1664         mcs_rate.rate_n_flags |= RATE_MCS_ANT_B_MSK;
1665         mcs_rate.rate_n_flags &= ~RATE_MCS_ANT_A_MSK;
1666
1667         if (i >= IWL_FIRST_CCK_RATE && i <= IWL_LAST_CCK_RATE)
1668                 mcs_rate.rate_n_flags |= RATE_MCS_CCK_MSK;
1669
1670         tbl->antenna_type = ANT_AUX;
1671         rs_get_tbl_info_from_mcs(&mcs_rate, priv->phymode, tbl, &rate_idx);
1672         if (!rs_is_ant_connected(priv->valid_antenna, tbl->antenna_type))
1673             rs_toggle_antenna(&mcs_rate, tbl),
1674
1675         rs_mcs_from_tbl(&mcs_rate, tbl, rate_idx, use_green);
1676         tbl->current_rate.rate_n_flags = mcs_rate.rate_n_flags;
1677         rs_get_expected_tpt_table(lq, tbl);
1678         rs_fill_link_cmd(lq, &mcs_rate, &(lq->lq), sta);
1679         rs_send_lq_cmd(priv, &lq->lq, CMD_ASYNC);
1680  out:
1681         return;
1682 }
1683
1684 static struct ieee80211_rate *rs_get_lowest_rate(struct ieee80211_local
1685                                                  *local)
1686 {
1687         struct ieee80211_hw_mode *mode = local->oper_hw_mode;
1688         int i;
1689
1690         for (i = 0; i < mode->num_rates; i++) {
1691                 struct ieee80211_rate *rate = &mode->rates[i];
1692
1693                 if (rate->flags & IEEE80211_RATE_SUPPORTED)
1694                         return rate;
1695         }
1696
1697         return &mode->rates[0];
1698 }
1699
1700 static struct ieee80211_rate *rs_get_rate(void *priv_rate,
1701                                                struct net_device *dev,
1702                                                struct sk_buff *skb,
1703                                                struct rate_control_extra
1704                                                *extra)
1705 {
1706
1707         int i;
1708         struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr);
1709         struct ieee80211_hdr *hdr = (struct ieee80211_hdr *)skb->data;
1710         struct sta_info *sta;
1711         u16 fc;
1712         struct iwl_priv *priv = (struct iwl_priv *)priv_rate;
1713         struct iwl_rate_scale_priv *lq;
1714
1715         IWL_DEBUG_RATE("rate scale calculate new rate for skb\n");
1716
1717         memset(extra, 0, sizeof(*extra));
1718
1719         fc = le16_to_cpu(hdr->frame_control);
1720         if (!ieee80211_is_data(fc) || is_multicast_ether_addr(hdr->addr1)) {
1721                 /* Send management frames and broadcast/multicast data using
1722                  * lowest rate. */
1723                 /* TODO: this could probably be improved.. */
1724                 return rs_get_lowest_rate(local);
1725         }
1726
1727         sta = sta_info_get(local, hdr->addr1);
1728
1729         if (!sta || !sta->rate_ctrl_priv) {
1730                 if (sta)
1731                         sta_info_put(sta);
1732                 return rs_get_lowest_rate(local);
1733         }
1734
1735         lq = (struct iwl_rate_scale_priv *)sta->rate_ctrl_priv;
1736         i = sta->last_txrate;
1737
1738         if ((priv->iw_mode == IEEE80211_IF_TYPE_IBSS) && !lq->ibss_sta_added) {
1739                 u8 sta_id = iwl_hw_find_station(priv, hdr->addr1);
1740
1741                 if (sta_id == IWL_INVALID_STATION) {
1742                         IWL_DEBUG_RATE("LQ: ADD station " MAC_FMT "\n",
1743                                         MAC_ARG(hdr->addr1));
1744                         sta_id = iwl_add_station(priv,
1745                                                  hdr->addr1, 0, CMD_ASYNC);
1746                 }
1747                 if ((sta_id != IWL_INVALID_STATION)) {
1748                         lq->lq.sta_id = sta_id;
1749                         lq->lq.rs_table[0].rate_n_flags = 0;
1750                         lq->ibss_sta_added = 1;
1751                         rs_initialize_lq(priv, sta);
1752                 }
1753                 if (!lq->ibss_sta_added)
1754                         goto done;
1755         }
1756
1757  done:
1758         sta_info_put(sta);
1759         if ((i < 0) || (i > IWL_RATE_COUNT))
1760                 return rs_get_lowest_rate(local);
1761
1762         return &priv->ieee_rates[i];
1763 }
1764
1765 static void *rs_alloc_sta(void *priv, gfp_t gfp)
1766 {
1767         struct iwl_rate_scale_priv *crl;
1768         int i, j;
1769
1770         IWL_DEBUG_RATE("create station rate scale window\n");
1771
1772         crl = kzalloc(sizeof(struct iwl_rate_scale_priv), gfp);
1773
1774         if (crl == NULL)
1775                 return NULL;
1776
1777         memset(crl, 0, sizeof(struct iwl_rate_scale_priv));
1778         crl->lq.sta_id = 0xff;
1779
1780         for (j = 0; j < LQ_SIZE; j++)
1781                 for (i = 0; i < IWL_RATE_COUNT; i++)
1782                         rs_rate_scale_clear_window(&(crl->lq_info[j].win[i]));
1783
1784         return crl;
1785 }
1786
1787 static void rs_rate_init(void *priv_rate, void *priv_sta,
1788                          struct ieee80211_local *local,
1789                          struct sta_info *sta)
1790 {
1791         int i, j;
1792         struct ieee80211_hw_mode *mode = local->oper_hw_mode;
1793         struct iwl_priv *priv = (struct iwl_priv *)priv_rate;
1794         struct iwl_rate_scale_priv *crl = priv_sta;
1795
1796         memset(crl, 0, sizeof(struct iwl_rate_scale_priv));
1797
1798         crl->lq.sta_id = 0xff;
1799         crl->flush_timer = 0;
1800         sta->txrate = 3;
1801         for (j = 0; j < LQ_SIZE; j++)
1802                 for (i = 0; i < IWL_RATE_COUNT; i++)
1803                         rs_rate_scale_clear_window(&(crl->lq_info[j].win[i]));
1804
1805         IWL_DEBUG_RATE("rate scale global init\n");
1806         /* TODO: what is a good starting rate for STA? About middle? Maybe not
1807          * the lowest or the highest rate.. Could consider using RSSI from
1808          * previous packets? Need to have IEEE 802.1X auth succeed immediately
1809          * after assoc.. */
1810
1811         crl->ibss_sta_added = 0;
1812         if (priv->iw_mode == IEEE80211_IF_TYPE_AP) {
1813                 u8 sta_id = iwl_hw_find_station(priv, sta->addr);
1814                 /* for IBSS the call are from tasklet */
1815                 IWL_DEBUG_HT("LQ: ADD station " MAC_FMT " \n",
1816                              MAC_ARG(sta->addr));
1817
1818                 if (sta_id == IWL_INVALID_STATION) {
1819                         IWL_DEBUG_RATE("LQ: ADD station " MAC_FMT "\n",
1820                                         MAC_ARG(sta->addr));
1821                                         sta_id = iwl_add_station(priv,
1822                                                  sta->addr, 0, CMD_ASYNC);
1823                 }
1824                 if ((sta_id != IWL_INVALID_STATION)) {
1825                         crl->lq.sta_id = sta_id;
1826                         crl->lq.rs_table[0].rate_n_flags = 0;
1827                 }
1828                 /* FIXME: this is w/a remove it later */
1829                 priv->assoc_station_added = 1;
1830         }
1831
1832         for (i = 0; i < mode->num_rates; i++) {
1833                 if ((sta->supp_rates & BIT(i)) &&
1834                     (mode->rates[i].flags & IEEE80211_RATE_SUPPORTED))
1835                         sta->txrate = i;
1836         }
1837         sta->last_txrate = sta->txrate;
1838         /* For MODE_IEEE80211A mode cck rate are at end
1839          * rate table
1840          */
1841         if (local->hw.conf.phymode == MODE_IEEE80211A)
1842                 sta->last_txrate += IWL_FIRST_OFDM_RATE;
1843
1844         crl->is_dup = priv->is_dup;
1845         crl->valid_antenna = priv->valid_antenna;
1846         crl->antenna = priv->antenna;
1847         crl->is_green = rs_use_green(priv);
1848         crl->active_rate = priv->active_rate;
1849         crl->active_rate &= ~(0x1000);
1850         crl->active_rate_basic = priv->active_rate_basic;
1851         crl->phymode = priv->phymode;
1852 #ifdef CONFIG_IWLWIFI_HT
1853         crl->active_siso_rate = (priv->current_assoc_ht.supp_rates[0] << 1);
1854         crl->active_siso_rate |= (priv->current_assoc_ht.supp_rates[0] & 0x1);
1855         crl->active_siso_rate &= ~((u16)0x2);
1856         crl->active_siso_rate = crl->active_siso_rate << IWL_FIRST_OFDM_RATE;
1857
1858         crl->active_mimo_rate = (priv->current_assoc_ht.supp_rates[1] << 1);
1859         crl->active_mimo_rate |= (priv->current_assoc_ht.supp_rates[1] & 0x1);
1860         crl->active_mimo_rate &= ~((u16)0x2);
1861         crl->active_mimo_rate = crl->active_mimo_rate << IWL_FIRST_OFDM_RATE;
1862         IWL_DEBUG_HT("MIMO RATE 0x%X SISO MASK 0x%X\n", crl->active_siso_rate,
1863                      crl->active_mimo_rate);
1864 #endif /*CONFIG_IWLWIFI_HT*/
1865
1866         if (priv->assoc_station_added)
1867                 priv->lq_mngr.lq_ready = 1;
1868
1869         rs_initialize_lq(priv, sta);
1870 }
1871
1872 static int rs_fill_link_cmd(struct iwl_rate_scale_priv *lq_data,
1873                             struct iwl_rate *tx_mcs,
1874                             struct iwl_link_quality_cmd *lq_cmd,
1875                             struct sta_info *sta)
1876 {
1877         int index = 0;
1878         int rc = 0;
1879         int rate_idx;
1880         u8 ant_toggle_count = 0;
1881         u8 use_ht_possible = 1;
1882         u8 repeat_cur_rate = 0;
1883         struct iwl_rate new_rate;
1884         struct iwl_scale_tbl_info tbl_type = { 0 };
1885
1886         rs_get_tbl_info_from_mcs(tx_mcs, lq_data->phymode,
1887                                   &tbl_type, &rate_idx);
1888
1889         if (is_legacy(tbl_type.lq_type)) {
1890                 ant_toggle_count = 1;
1891                 repeat_cur_rate = IWL_NUMBER_TRY;
1892         } else
1893                 repeat_cur_rate = IWL_HT_NUMBER_TRY;
1894
1895         lq_cmd->general_params.mimo_delimiter =
1896                         is_mimo(tbl_type.lq_type) ? 1 : 0;
1897         lq_cmd->rs_table[index].rate_n_flags =
1898                         cpu_to_le32(tx_mcs->rate_n_flags);
1899         new_rate.rate_n_flags = tx_mcs->rate_n_flags;
1900
1901         if (is_mimo(tbl_type.lq_type) || (tbl_type.antenna_type == ANT_MAIN))
1902                 lq_cmd->general_params.single_stream_ant_msk = 1;
1903         else
1904                 lq_cmd->general_params.single_stream_ant_msk = 2;
1905
1906         index++;
1907         repeat_cur_rate--;
1908
1909         while (index < LINK_QUAL_MAX_RETRY_NUM) {
1910                 while (repeat_cur_rate && (index < LINK_QUAL_MAX_RETRY_NUM)) {
1911                         if (is_legacy(tbl_type.lq_type)) {
1912                                 if (ant_toggle_count <
1913                                     NUM_TRY_BEFORE_ANTENNA_TOGGLE)
1914                                         ant_toggle_count++;
1915                                 else {
1916                                         rs_toggle_antenna(&new_rate, &tbl_type);
1917                                         ant_toggle_count = 1;
1918                                 }
1919                         }
1920                         lq_cmd->rs_table[index].rate_n_flags =
1921                                         cpu_to_le32(new_rate.rate_n_flags);
1922                         repeat_cur_rate--;
1923                         index++;
1924                 }
1925
1926                 rs_get_tbl_info_from_mcs(&new_rate, lq_data->phymode, &tbl_type,
1927                                                 &rate_idx);
1928
1929                 if (is_mimo(tbl_type.lq_type))
1930                         lq_cmd->general_params.mimo_delimiter = index;
1931
1932                 rs_get_lower_rate(lq_data, &tbl_type, rate_idx,
1933                                   use_ht_possible, &new_rate, sta);
1934
1935                 if (is_legacy(tbl_type.lq_type)) {
1936                         if (ant_toggle_count < NUM_TRY_BEFORE_ANTENNA_TOGGLE)
1937                                 ant_toggle_count++;
1938                         else {
1939                                 rs_toggle_antenna(&new_rate, &tbl_type);
1940                                 ant_toggle_count = 1;
1941                         }
1942                         repeat_cur_rate = IWL_NUMBER_TRY;
1943                 } else
1944                         repeat_cur_rate = IWL_HT_NUMBER_TRY;
1945
1946                 use_ht_possible = 0;
1947
1948                 lq_cmd->rs_table[index].rate_n_flags =
1949                                 cpu_to_le32(new_rate.rate_n_flags);
1950                 /* lq_cmd->rs_table[index].rate_n_flags = 0x800d; */
1951
1952                 index++;
1953                 repeat_cur_rate--;
1954         }
1955
1956         /* lq_cmd->rs_table[0].rate_n_flags = 0x800d; */
1957
1958         lq_cmd->general_params.dual_stream_ant_msk = 3;
1959         lq_cmd->agg_params.agg_dis_start_th = 3;
1960         lq_cmd->agg_params.agg_time_limit = cpu_to_le16(4000);
1961         return rc;
1962 }
1963
1964 static void *rs_alloc(struct ieee80211_local *local)
1965 {
1966         return local->hw.priv;
1967 }
1968 /* rate scale requires free function to be implemented */
1969 static void rs_free(void *priv_rate)
1970 {
1971         return;
1972 }
1973
1974 static void rs_clear(void *priv_rate)
1975 {
1976         struct iwl_priv *priv = (struct iwl_priv *) priv_rate;
1977
1978         IWL_DEBUG_RATE("enter\n");
1979
1980         priv->lq_mngr.lq_ready = 0;
1981 #ifdef CONFIG_IWLWIFI_HT
1982 #ifdef CONFIG_IWLWIFI_HT_AGG
1983         if (priv->lq_mngr.agg_ctrl.granted_ba)
1984                 iwl4965_turn_off_agg(priv, TID_ALL_SPECIFIED);
1985 #endif /*CONFIG_IWLWIFI_HT_AGG */
1986 #endif /* CONFIG_IWLWIFI_HT */
1987
1988         IWL_DEBUG_RATE("leave\n");
1989 }
1990
1991 static void rs_free_sta(void *priv, void *priv_sta)
1992 {
1993         struct iwl_rate_scale_priv *rs_priv = priv_sta;
1994
1995         IWL_DEBUG_RATE("enter\n");
1996         kfree(rs_priv);
1997         IWL_DEBUG_RATE("leave\n");
1998 }
1999
2000
2001 static struct rate_control_ops rs_ops = {
2002         .module = NULL,
2003         .name = RS_NAME,
2004         .tx_status = rs_tx_status,
2005         .get_rate = rs_get_rate,
2006         .rate_init = rs_rate_init,
2007         .clear = rs_clear,
2008         .alloc = rs_alloc,
2009         .free = rs_free,
2010         .alloc_sta = rs_alloc_sta,
2011         .free_sta = rs_free_sta,
2012 };
2013
2014 int iwl_fill_rs_info(struct ieee80211_hw *hw, char *buf, u8 sta_id)
2015 {
2016         struct ieee80211_local *local = hw_to_local(hw);
2017         struct iwl_priv *priv = hw->priv;
2018         struct iwl_rate_scale_priv *rs_priv;
2019         struct sta_info *sta;
2020         int count = 0, i;
2021         u32 samples = 0, success = 0, good = 0;
2022         unsigned long now = jiffies;
2023         u32 max_time = 0;
2024         u8 lq_type, antenna;
2025
2026         sta = sta_info_get(local, priv->stations[sta_id].sta.sta.addr);
2027         if (!sta || !sta->rate_ctrl_priv) {
2028                 if (sta) {
2029                         sta_info_put(sta);
2030                         IWL_DEBUG_RATE("leave - no private rate data!\n");
2031                 } else
2032                         IWL_DEBUG_RATE("leave - no station!\n");
2033                 return sprintf(buf, "station %d not found\n", sta_id);
2034         }
2035
2036         rs_priv = (void *)sta->rate_ctrl_priv;
2037
2038         lq_type = rs_priv->lq_info[rs_priv->active_tbl].lq_type;
2039         antenna = rs_priv->lq_info[rs_priv->active_tbl].antenna_type;
2040
2041         if (is_legacy(lq_type))
2042                 i = IWL_RATE_54M_INDEX;
2043         else
2044                 i = IWL_RATE_60M_INDEX;
2045         while (1) {
2046                 u64 mask;
2047                 int j;
2048                 int active = rs_priv->active_tbl;
2049
2050                 count +=
2051                     sprintf(&buf[count], " %2dMbs: ", iwl_rates[i].ieee / 2);
2052
2053                 mask = (1ULL << (IWL_RATE_MAX_WINDOW - 1));
2054                 for (j = 0; j < IWL_RATE_MAX_WINDOW; j++, mask >>= 1)
2055                         buf[count++] =
2056                                 (rs_priv->lq_info[active].win[i].data & mask)
2057                                 ? '1' : '0';
2058
2059                 samples += rs_priv->lq_info[active].win[i].counter;
2060                 good += rs_priv->lq_info[active].win[i].success_counter;
2061                 success += rs_priv->lq_info[active].win[i].success_counter *
2062                            iwl_rates[i].ieee;
2063
2064                 if (rs_priv->lq_info[active].win[i].stamp) {
2065                         int delta =
2066                                    jiffies_to_msecs(now -
2067                                    rs_priv->lq_info[active].win[i].stamp);
2068
2069                         if (delta > max_time)
2070                                 max_time = delta;
2071
2072                         count += sprintf(&buf[count], "%5dms\n", delta);
2073                 } else
2074                         buf[count++] = '\n';
2075
2076                 j = iwl_get_prev_ieee_rate(i);
2077                 if (j == i)
2078                         break;
2079                 i = j;
2080         }
2081
2082         /* Display the average rate of all samples taken.
2083          *
2084          * NOTE:  We multiple # of samples by 2 since the IEEE measurement
2085          * added from iwl_rates is actually 2X the rate */
2086         if (samples)
2087                 count += sprintf(&buf[count],
2088                          "\nAverage rate is %3d.%02dMbs over last %4dms\n"
2089                          "%3d%% success (%d good packets over %d tries)\n",
2090                          success / (2 * samples), (success * 5 / samples) % 10,
2091                          max_time, good * 100 / samples, good, samples);
2092         else
2093                 count += sprintf(&buf[count], "\nAverage rate: 0Mbs\n");
2094         count += sprintf(&buf[count], "\nrate scale type %d anntena %d "
2095                          "active_search %d rate index %d\n", lq_type, antenna,
2096                          rs_priv->search_better_tbl, sta->last_txrate);
2097
2098         sta_info_put(sta);
2099         return count;
2100 }
2101
2102 void iwl_rate_scale_init(struct ieee80211_hw *hw, s32 sta_id)
2103 {
2104         struct iwl_priv *priv = hw->priv;
2105
2106         priv->lq_mngr.lq_ready = 1;
2107 }
2108
2109 void iwl_rate_control_register(struct ieee80211_hw *hw)
2110 {
2111         ieee80211_rate_control_register(&rs_ops);
2112 }
2113
2114 void iwl_rate_control_unregister(struct ieee80211_hw *hw)
2115 {
2116         ieee80211_rate_control_unregister(&rs_ops);
2117 }
2118