drivers/staging/rtl8821ae/cam.c

   1 /******************************************************************************
   2  *
   3  * Copyright(c) 2009-2010  Realtek Corporation.
   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  * wlanfae <wlanfae@realtek.com>
  23  * Realtek Corporation, No. 2, Innovation Road II, Hsinchu Science Park,
  24  * Hsinchu 300, Taiwan.
  25  *
  26  * Larry Finger <Larry.Finger@lwfinger.net>
  27  *
  28  *****************************************************************************/
  29 #include "wifi.h"
  30 #include "cam.h"
  31 #include <linux/export.h>
  32
  33 void rtl_cam_reset_sec_info(struct ieee80211_hw *hw)
  34 {
  35         struct rtl_priv *rtlpriv = rtl_priv(hw);
  36
  37         rtlpriv->sec.use_defaultkey = false;
  38         rtlpriv->sec.pairwise_enc_algorithm = NO_ENCRYPTION;
  39         rtlpriv->sec.group_enc_algorithm = NO_ENCRYPTION;
  40         memset(rtlpriv->sec.key_buf, 0, KEY_BUF_SIZE * MAX_KEY_LEN);
  41         memset(rtlpriv->sec.key_len, 0, KEY_BUF_SIZE);
  42         rtlpriv->sec.pairwise_key = NULL;
  43 }
  44
  45 static void rtl_cam_program_entry(struct ieee80211_hw *hw, u32 entry_no,
  46                            u8 *mac_addr, u8 *key_cont_128, u16 us_config)
  47 {
  48         struct rtl_priv *rtlpriv = rtl_priv(hw);
  49
  50         u32 target_command;
  51         u32 target_content = 0;
  52         u8 entry_i;
  53
  54         RT_PRINT_DATA(rtlpriv, COMP_SEC, DBG_DMESG, "Key content :",
  55                         key_cont_128, 16);
  56
  57         for (entry_i = 0; entry_i < CAM_CONTENT_COUNT; entry_i++) {
  58                 target_command = entry_i + CAM_CONTENT_COUNT * entry_no;
  59                 target_command = target_command | BIT(31) | BIT(16);
  60
  61                 if (entry_i == 0) {
  62                         target_content = (u32) (*(mac_addr + 0)) << 16 |
  63                             (u32) (*(mac_addr + 1)) << 24 | (u32) us_config;
  64
  65                         rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[WCAMI],
  66                                         target_content);
  67                         rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[RWCAM],
  68                                         target_command);
  69
  70                         RT_TRACE(COMP_SEC, DBG_LOUD,
  71                                  ("WRITE %x: %x \n",
  72                                   rtlpriv->cfg->maps[WCAMI], target_content));
  73                         RT_TRACE(COMP_SEC, DBG_LOUD,
  74                                  ("The Key ID is %d\n", entry_no));
  75                         RT_TRACE(COMP_SEC, DBG_LOUD,
  76                                  ("WRITE %x: %x \n",
  77                                   rtlpriv->cfg->maps[RWCAM], target_command));
  78
  79                 } else if (entry_i == 1) {
  80
  81                         target_content = (u32) (*(mac_addr + 5)) << 24 |
  82                             (u32) (*(mac_addr + 4)) << 16 |
  83                             (u32) (*(mac_addr + 3)) << 8 |
  84                             (u32) (*(mac_addr + 2));
  85
  86                         rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[WCAMI],
  87                                         target_content);
  88                         rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[RWCAM],
  89                                         target_command);
  90
  91                         RT_TRACE(COMP_SEC, DBG_LOUD,
  92                                  ("WRITE A4: %x \n", target_content));
  93                         RT_TRACE(COMP_SEC, DBG_LOUD,
  94                                  ("WRITE A0: %x \n", target_command));
  95
  96                 } else {
  97
  98                         target_content =
  99                             (u32) (*(key_cont_128 + (entry_i * 4 - 8) + 3)) <<
 100                             24 | (u32) (*(key_cont_128 + (entry_i * 4 - 8) + 2))
 101                             << 16 |
 102                             (u32) (*(key_cont_128 + (entry_i * 4 - 8) + 1)) << 8
 103                             | (u32) (*(key_cont_128 + (entry_i * 4 - 8) + 0));
 104
 105                         rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[WCAMI],
 106                                         target_content);
 107                         rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[RWCAM],
 108                                         target_command);
 109                         udelay(100);
 110
 111                         RT_TRACE(COMP_SEC, DBG_LOUD,
 112                                  ("WRITE A4: %x \n", target_content));
 113                         RT_TRACE(COMP_SEC, DBG_LOUD,
 114                                  ("WRITE A0: %x \n", target_command));
 115                 }
 116         }
 117
 118         RT_TRACE(COMP_SEC, DBG_LOUD,
 119                  ("after set key, usconfig:%x\n", us_config));
 120 }
 121
 122 u8 rtl_cam_add_one_entry(struct ieee80211_hw *hw, u8 *mac_addr,
 123                          u32 ul_key_id, u32 ul_entry_idx, u32 ul_enc_alg,
 124                          u32 ul_default_key, u8 *key_content)
 125 {
 126         u32 us_config;
 127         struct rtl_priv *rtlpriv = rtl_priv(hw);
 128
 129         RT_TRACE(COMP_SEC, DBG_DMESG,
 130                  ("EntryNo:%x, ulKeyId=%x, ulEncAlg=%x, "
 131                   "ulUseDK=%x MacAddr %pM\n",
 132                   ul_entry_idx, ul_key_id, ul_enc_alg,
 133                   ul_default_key, mac_addr));
 134
 135         if (ul_key_id == TOTAL_CAM_ENTRY) {
 136                 RT_TRACE(COMP_ERR, DBG_WARNING,
 137                          ("ulKeyId exceed!\n"));
 138                 return 0;
 139         }
 140
 141         if (ul_default_key == 1) {
 142                 us_config = CFG_VALID | ((u16) (ul_enc_alg) << 2);
 143         } else {
 144                 us_config = CFG_VALID | ((ul_enc_alg) << 2) | ul_key_id;
 145         }
 146
 147         rtl_cam_program_entry(hw, ul_entry_idx, mac_addr,
 148                               (u8 *) key_content, us_config);
 149
 150         RT_TRACE(COMP_SEC, DBG_DMESG, ("end \n"));
 151
 152         return 1;
 153
 154 }
 155 //EXPORT_SYMBOL(rtl_cam_add_one_entry);
 156
 157 int rtl_cam_delete_one_entry(struct ieee80211_hw *hw,
 158                              u8 *mac_addr, u32 ul_key_id)
 159 {
 160         u32 ul_command;
 161         struct rtl_priv *rtlpriv = rtl_priv(hw);
 162
 163         RT_TRACE(COMP_SEC, DBG_DMESG, ("key_idx:%d\n", ul_key_id));
 164
 165         ul_command = ul_key_id * CAM_CONTENT_COUNT;
 166         ul_command = ul_command | BIT(31) | BIT(16);
 167
 168         rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[WCAMI], 0);
 169         rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[RWCAM], ul_command);
 170
 171         RT_TRACE(COMP_SEC, DBG_DMESG,
 172                  ("rtl_cam_delete_one_entry(): WRITE A4: %x \n", 0));
 173         RT_TRACE(COMP_SEC, DBG_DMESG,
 174                  ("rtl_cam_delete_one_entry(): WRITE A0: %x \n", ul_command));
 175
 176         return 0;
 177
 178 }
 179 //EXPORT_SYMBOL(rtl_cam_delete_one_entry);
 180
 181 void rtl_cam_reset_all_entry(struct ieee80211_hw *hw)
 182 {
 183         u32 ul_command;
 184         struct rtl_priv *rtlpriv = rtl_priv(hw);
 185
 186         ul_command = BIT(31) | BIT(30);
 187         rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[RWCAM], ul_command);
 188 }
 189 //EXPORT_SYMBOL(rtl_cam_reset_all_entry);
 190
 191 void rtl_cam_mark_invalid(struct ieee80211_hw *hw, u8 uc_index)
 192 {
 193         struct rtl_priv *rtlpriv = rtl_priv(hw);
 194
 195         u32 ul_command;
 196         u32 ul_content;
 197         u32 ul_enc_algo = rtlpriv->cfg->maps[SEC_CAM_AES];
 198
 199         switch (rtlpriv->sec.pairwise_enc_algorithm) {
 200         case WEP40_ENCRYPTION:
 201                 ul_enc_algo = rtlpriv->cfg->maps[SEC_CAM_WEP40];
 202                 break;
 203         case WEP104_ENCRYPTION:
 204                 ul_enc_algo = rtlpriv->cfg->maps[SEC_CAM_WEP104];
 205                 break;
 206         case TKIP_ENCRYPTION:
 207                 ul_enc_algo = rtlpriv->cfg->maps[SEC_CAM_TKIP];
 208                 break;
 209         case AESCCMP_ENCRYPTION:
 210                 ul_enc_algo = rtlpriv->cfg->maps[SEC_CAM_AES];
 211                 break;
 212         default:
 213                 ul_enc_algo = rtlpriv->cfg->maps[SEC_CAM_AES];
 214         }
 215
 216         ul_content = (uc_index & 3) | ((u16) (ul_enc_algo) << 2);
 217
 218         ul_content |= BIT(15);
 219         ul_command = CAM_CONTENT_COUNT * uc_index;
 220         ul_command = ul_command | BIT(31) | BIT(16);
 221
 222         rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[WCAMI], ul_content);
 223         rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[RWCAM], ul_command);
 224
 225         RT_TRACE(COMP_SEC, DBG_DMESG,
 226                  ("rtl_cam_mark_invalid(): WRITE A4: %x \n", ul_content));
 227         RT_TRACE(COMP_SEC, DBG_DMESG,
 228                  ("rtl_cam_mark_invalid(): WRITE A0: %x \n", ul_command));
 229 }
 230 //EXPORT_SYMBOL(rtl_cam_mark_invalid);
 231
 232 void rtl_cam_empty_entry(struct ieee80211_hw *hw, u8 uc_index)
 233 {
 234         struct rtl_priv *rtlpriv = rtl_priv(hw);
 235
 236         u32 ul_command;
 237         u32 ul_content;
 238         u32 ul_encalgo = rtlpriv->cfg->maps[SEC_CAM_AES];
 239         u8 entry_i;
 240
 241         switch (rtlpriv->sec.pairwise_enc_algorithm) {
 242         case WEP40_ENCRYPTION:
 243                 ul_encalgo = rtlpriv->cfg->maps[SEC_CAM_WEP40];
 244                 break;
 245         case WEP104_ENCRYPTION:
 246                 ul_encalgo = rtlpriv->cfg->maps[SEC_CAM_WEP104];
 247                 break;
 248         case TKIP_ENCRYPTION:
 249                 ul_encalgo = rtlpriv->cfg->maps[SEC_CAM_TKIP];
 250                 break;
 251         case AESCCMP_ENCRYPTION:
 252                 ul_encalgo = rtlpriv->cfg->maps[SEC_CAM_AES];
 253                 break;
 254         default:
 255                 ul_encalgo = rtlpriv->cfg->maps[SEC_CAM_AES];
 256         }
 257
 258         for (entry_i = 0; entry_i < CAM_CONTENT_COUNT; entry_i++) {
 259
 260                 if (entry_i == 0) {
 261                         ul_content =
 262                             (uc_index & 0x03) | ((u16) (ul_encalgo) << 2);
 263                         ul_content |= BIT(15);
 264
 265                 } else {
 266                         ul_content = 0;
 267                 }
 268
 269                 ul_command = CAM_CONTENT_COUNT * uc_index + entry_i;
 270                 ul_command = ul_command | BIT(31) | BIT(16);
 271
 272                 rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[WCAMI], ul_content);
 273                 rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[RWCAM], ul_command);
 274
 275                 RT_TRACE(COMP_SEC, DBG_LOUD,
 276                          ("rtl_cam_empty_entry(): WRITE A4: %x \n",
 277                           ul_content));
 278                 RT_TRACE(COMP_SEC, DBG_LOUD,
 279                          ("rtl_cam_empty_entry(): WRITE A0: %x \n",
 280                           ul_command));
 281         }
 282
 283 }
 284 //EXPORT_SYMBOL(rtl_cam_empty_entry);
 285
 286 u8 rtl_cam_get_free_entry(struct ieee80211_hw *hw, u8 *sta_addr)
 287 {
 288         struct rtl_priv *rtlpriv = rtl_priv(hw);
 289         u32 bitmap = (rtlpriv->sec.hwsec_cam_bitmap) >> 4;
 290         u8 entry_idx = 0;
 291         u8 i, *addr;
 292
 293         if (NULL == sta_addr) {
 294                 RT_TRACE(COMP_SEC, DBG_EMERG,
 295                         ("sta_addr is NULL.\n"));
 296                 return TOTAL_CAM_ENTRY;
 297         }
 298         /* Does STA already exist? */
 299         for (i = 4; i < TOTAL_CAM_ENTRY; i++) {
 300                 addr = rtlpriv->sec.hwsec_cam_sta_addr[i];
 301                 if(memcmp(addr, sta_addr, ETH_ALEN) == 0)
 302                         return i;
 303         }
 304         /* Get a free CAM entry. */
 305         for (entry_idx = 4; entry_idx < TOTAL_CAM_ENTRY; entry_idx++) {
 306                 if ((bitmap & BIT(0)) == 0) {
 307                         RT_TRACE(COMP_SEC, DBG_EMERG,
 308                                 ("-----hwsec_cam_bitmap: 0x%x entry_idx=%d\n",
 309                                  rtlpriv->sec.hwsec_cam_bitmap, entry_idx));
 310                         rtlpriv->sec.hwsec_cam_bitmap |= BIT(0) << entry_idx;
 311                         memcpy(rtlpriv->sec.hwsec_cam_sta_addr[entry_idx],
 312                                sta_addr, ETH_ALEN);
 313                         return entry_idx;
 314                 }
 315                 bitmap = bitmap >>1;
 316         }
 317         return TOTAL_CAM_ENTRY;
 318 }
 319 //EXPORT_SYMBOL(rtl_cam_get_free_entry);
 320
 321 void rtl_cam_del_entry(struct ieee80211_hw *hw, u8 *sta_addr)
 322 {
 323         struct rtl_priv *rtlpriv = rtl_priv(hw);
 324         u32 bitmap;
 325         u8 i, *addr;
 326
 327         if (NULL == sta_addr) {
 328                 RT_TRACE(COMP_SEC, DBG_EMERG,
 329                         ("sta_addr is NULL.\n"));
 330         }
 331
 332         if ((sta_addr[0]|sta_addr[1]|sta_addr[2]|sta_addr[3]|\
 333                                 sta_addr[4]|sta_addr[5]) == 0) {
 334                 RT_TRACE(COMP_SEC, DBG_EMERG,
 335                         ("sta_addr is 00:00:00:00:00:00.\n"));
 336                 return;
 337         }
 338         /* Does STA already exist? */
 339         for (i = 4; i < TOTAL_CAM_ENTRY; i++) {
 340                 addr = rtlpriv->sec.hwsec_cam_sta_addr[i];
 341                 bitmap = (rtlpriv->sec.hwsec_cam_bitmap) >> i;
 342                 if (((bitmap & BIT(0)) == BIT(0)) &&
 343                     (memcmp(addr, sta_addr, ETH_ALEN) == 0)) {
 344                         /* Remove from HW Security CAM */
 345                         memset(rtlpriv->sec.hwsec_cam_sta_addr[i], 0, ETH_ALEN);
 346                         rtlpriv->sec.hwsec_cam_bitmap &= ~(BIT(0) << i);
 347                         printk("&&&&&&&&&del entry %d\n",i);
 348                 }
 349         }
 350         return;
 351 }
 352 //EXPORT_SYMBOL(rtl_cam_del_entry);