root/drivers/net/wireless/realtek/rtlwifi/cam.c
// SPDX-License-Identifier: GPL-2.0
/* Copyright(c) 2009-2012  Realtek Corporation.*/

#include "wifi.h"
#include "cam.h"
#include <linux/export.h>

void rtl_cam_reset_sec_info(struct ieee80211_hw *hw)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);

        rtlpriv->sec.use_defaultkey = false;
        rtlpriv->sec.pairwise_enc_algorithm = NO_ENCRYPTION;
        rtlpriv->sec.group_enc_algorithm = NO_ENCRYPTION;
        memset(rtlpriv->sec.key_buf, 0, KEY_BUF_SIZE * MAX_KEY_LEN);
        memset(rtlpriv->sec.key_len, 0, KEY_BUF_SIZE);
        rtlpriv->sec.pairwise_key = NULL;
}

static void rtl_cam_program_entry(struct ieee80211_hw *hw, u32 entry_no,
                                  const u8 *mac_addr, u8 *key_cont_128,
                                  u16 us_config)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);

        u32 target_command;
        u32 target_content = 0;
        int entry_i;

        RT_PRINT_DATA(rtlpriv, COMP_SEC, DBG_DMESG, "Key content :",
                      key_cont_128, 16);

        /* 0-1 config + mac, 2-5 fill 128key,6-7 are reserved */
        for (entry_i = CAM_CONTENT_COUNT - 1; entry_i >= 0; entry_i--) {
                target_command = entry_i + CAM_CONTENT_COUNT * entry_no;
                target_command = target_command | BIT(31) | BIT(16);

                if (entry_i == 0) {
                        target_content = (u32) (*(mac_addr + 0)) << 16 |
                            (u32) (*(mac_addr + 1)) << 24 | (u32) us_config;

                        rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[WCAMI],
                                        target_content);
                        rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[RWCAM],
                                        target_command);

                        rtl_dbg(rtlpriv, COMP_SEC, DBG_LOUD,
                                "WRITE %x: %x\n",
                                rtlpriv->cfg->maps[WCAMI], target_content);
                        rtl_dbg(rtlpriv, COMP_SEC, DBG_LOUD,
                                "The Key ID is %d\n", entry_no);
                        rtl_dbg(rtlpriv, COMP_SEC, DBG_LOUD,
                                "WRITE %x: %x\n",
                                rtlpriv->cfg->maps[RWCAM], target_command);

                } else if (entry_i == 1) {

                        target_content = (u32) (*(mac_addr + 5)) << 24 |
                            (u32) (*(mac_addr + 4)) << 16 |
                            (u32) (*(mac_addr + 3)) << 8 |
                            (u32) (*(mac_addr + 2));

                        rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[WCAMI],
                                        target_content);
                        rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[RWCAM],
                                        target_command);

                        rtl_dbg(rtlpriv, COMP_SEC, DBG_LOUD,
                                "WRITE A4: %x\n", target_content);
                        rtl_dbg(rtlpriv, COMP_SEC, DBG_LOUD,
                                "WRITE A0: %x\n", target_command);

                } else {

                        target_content =
                            (u32) (*(key_cont_128 + (entry_i * 4 - 8) + 3)) <<
                            24 | (u32) (*(key_cont_128 + (entry_i * 4 - 8) + 2))
                            << 16 |
                            (u32) (*(key_cont_128 + (entry_i * 4 - 8) + 1)) << 8
                            | (u32) (*(key_cont_128 + (entry_i * 4 - 8) + 0));

                        rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[WCAMI],
                                        target_content);
                        rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[RWCAM],
                                        target_command);

                        rtl_dbg(rtlpriv, COMP_SEC, DBG_LOUD,
                                "WRITE A4: %x\n", target_content);
                        rtl_dbg(rtlpriv, COMP_SEC, DBG_LOUD,
                                "WRITE A0: %x\n", target_command);
                }
        }

        rtl_dbg(rtlpriv, COMP_SEC, DBG_LOUD,
                "after set key, usconfig:%x\n", us_config);
}

u8 rtl_cam_add_one_entry(struct ieee80211_hw *hw, const u8 *mac_addr,
                         u32 ul_key_id, u32 ul_entry_idx, u32 ul_enc_alg,
                         u32 ul_default_key, u8 *key_content)
{
        u32 us_config;
        struct rtl_priv *rtlpriv = rtl_priv(hw);

        rtl_dbg(rtlpriv, COMP_SEC, DBG_DMESG,
                "EntryNo:%x, ulKeyId=%x, ulEncAlg=%x, ulUseDK=%x MacAddr %pM\n",
                ul_entry_idx, ul_key_id, ul_enc_alg,
                ul_default_key, mac_addr);

        if (ul_key_id == TOTAL_CAM_ENTRY) {
                rtl_dbg(rtlpriv, COMP_ERR, DBG_WARNING,
                        "ulKeyId exceed!\n");
                return 0;
        }

        if (ul_default_key == 1)
                us_config = CFG_VALID | ((u16) (ul_enc_alg) << 2);
        else
                us_config = CFG_VALID | ((ul_enc_alg) << 2) | ul_key_id;

        rtl_cam_program_entry(hw, ul_entry_idx, mac_addr,
                              (u8 *)key_content, us_config);

        rtl_dbg(rtlpriv, COMP_SEC, DBG_DMESG, "end\n");

        return 1;

}
EXPORT_SYMBOL(rtl_cam_add_one_entry);

int rtl_cam_delete_one_entry(struct ieee80211_hw *hw,
                             u8 *mac_addr, u32 ul_key_id)
{
        u32 ul_command;
        struct rtl_priv *rtlpriv = rtl_priv(hw);

        rtl_dbg(rtlpriv, COMP_SEC, DBG_DMESG, "key_idx:%d\n", ul_key_id);

        ul_command = ul_key_id * CAM_CONTENT_COUNT;
        ul_command = ul_command | BIT(31) | BIT(16);

        rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[WCAMI], 0);
        rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[RWCAM], ul_command);

        rtl_dbg(rtlpriv, COMP_SEC, DBG_DMESG,
                "%s: WRITE A4: %x\n", __func__, 0);
        rtl_dbg(rtlpriv, COMP_SEC, DBG_DMESG,
                "%s: WRITE A0: %x\n", __func__, ul_command);

        return 0;

}
EXPORT_SYMBOL(rtl_cam_delete_one_entry);

void rtl_cam_reset_all_entry(struct ieee80211_hw *hw)
{
        u32 ul_command;
        struct rtl_priv *rtlpriv = rtl_priv(hw);

        ul_command = BIT(31) | BIT(30);
        rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[RWCAM], ul_command);
}
EXPORT_SYMBOL(rtl_cam_reset_all_entry);

void rtl_cam_mark_invalid(struct ieee80211_hw *hw, u8 uc_index)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);

        u32 ul_command;
        u32 ul_content;
        u32 ul_enc_algo;

        switch (rtlpriv->sec.pairwise_enc_algorithm) {
        case WEP40_ENCRYPTION:
                ul_enc_algo = rtlpriv->cfg->maps[SEC_CAM_WEP40];
                break;
        case WEP104_ENCRYPTION:
                ul_enc_algo = rtlpriv->cfg->maps[SEC_CAM_WEP104];
                break;
        case TKIP_ENCRYPTION:
                ul_enc_algo = rtlpriv->cfg->maps[SEC_CAM_TKIP];
                break;
        case AESCCMP_ENCRYPTION:
                ul_enc_algo = rtlpriv->cfg->maps[SEC_CAM_AES];
                break;
        default:
                ul_enc_algo = rtlpriv->cfg->maps[SEC_CAM_AES];
        }

        ul_content = (uc_index & 3) | ((u16) (ul_enc_algo) << 2);

        ul_content |= BIT(15);
        ul_command = CAM_CONTENT_COUNT * uc_index;
        ul_command = ul_command | BIT(31) | BIT(16);

        rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[WCAMI], ul_content);
        rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[RWCAM], ul_command);

        rtl_dbg(rtlpriv, COMP_SEC, DBG_DMESG,
                "%s: WRITE A4: %x\n", __func__, ul_content);
        rtl_dbg(rtlpriv, COMP_SEC, DBG_DMESG,
                "%s: WRITE A0: %x\n", __func__, ul_command);
}
EXPORT_SYMBOL(rtl_cam_mark_invalid);

void rtl_cam_empty_entry(struct ieee80211_hw *hw, u8 uc_index)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);

        u32 ul_command;
        u32 ul_content;
        u32 ul_encalgo;
        u8 entry_i;

        switch (rtlpriv->sec.pairwise_enc_algorithm) {
        case WEP40_ENCRYPTION:
                ul_encalgo = rtlpriv->cfg->maps[SEC_CAM_WEP40];
                break;
        case WEP104_ENCRYPTION:
                ul_encalgo = rtlpriv->cfg->maps[SEC_CAM_WEP104];
                break;
        case TKIP_ENCRYPTION:
                ul_encalgo = rtlpriv->cfg->maps[SEC_CAM_TKIP];
                break;
        case AESCCMP_ENCRYPTION:
                ul_encalgo = rtlpriv->cfg->maps[SEC_CAM_AES];
                break;
        default:
                ul_encalgo = rtlpriv->cfg->maps[SEC_CAM_AES];
        }

        for (entry_i = 0; entry_i < CAM_CONTENT_COUNT; entry_i++) {

                if (entry_i == 0) {
                        ul_content =
                            (uc_index & 0x03) | ((u16) (ul_encalgo) << 2);
                        ul_content |= BIT(15);

                } else {
                        ul_content = 0;
                }

                ul_command = CAM_CONTENT_COUNT * uc_index + entry_i;
                ul_command = ul_command | BIT(31) | BIT(16);

                rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[WCAMI], ul_content);
                rtl_write_dword(rtlpriv, rtlpriv->cfg->maps[RWCAM], ul_command);

                rtl_dbg(rtlpriv, COMP_SEC, DBG_LOUD,
                        "%s: WRITE A4: %x\n", __func__, ul_content);
                rtl_dbg(rtlpriv, COMP_SEC, DBG_LOUD,
                        "%s: WRITE A0: %x\n", __func__, ul_command);
        }

}
EXPORT_SYMBOL(rtl_cam_empty_entry);

u8 rtl_cam_get_free_entry(struct ieee80211_hw *hw, u8 *sta_addr)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        u32 bitmap = (rtlpriv->sec.hwsec_cam_bitmap) >> 4;
        u8 entry_idx = 0;
        u8 i, *addr;

        if (NULL == sta_addr) {
                pr_err("sta_addr is NULL.\n");
                return TOTAL_CAM_ENTRY;
        }
        /* Does STA already exist? */
        for (i = 4; i < TOTAL_CAM_ENTRY; i++) {
                addr = rtlpriv->sec.hwsec_cam_sta_addr[i];
                if (ether_addr_equal_unaligned(addr, sta_addr))
                        return i;
        }
        /* Get a free CAM entry. */
        for (entry_idx = 4; entry_idx < TOTAL_CAM_ENTRY; entry_idx++) {
                if ((bitmap & BIT(0)) == 0) {
                        pr_err("-----hwsec_cam_bitmap: 0x%x entry_idx=%d\n",
                               rtlpriv->sec.hwsec_cam_bitmap, entry_idx);
                        rtlpriv->sec.hwsec_cam_bitmap |= BIT(0) << entry_idx;
                        memcpy(rtlpriv->sec.hwsec_cam_sta_addr[entry_idx],
                               sta_addr, ETH_ALEN);
                        return entry_idx;
                }
                bitmap = bitmap >> 1;
        }
        return TOTAL_CAM_ENTRY;
}
EXPORT_SYMBOL(rtl_cam_get_free_entry);

void rtl_cam_del_entry(struct ieee80211_hw *hw, u8 *sta_addr)
{
        struct rtl_priv *rtlpriv = rtl_priv(hw);
        u32 bitmap;
        u8 i, *addr;

        if (NULL == sta_addr) {
                pr_err("sta_addr is NULL.\n");
                return;
        }

        if (is_zero_ether_addr(sta_addr)) {
                pr_err("sta_addr is %pM\n", sta_addr);
                return;
        }
        /* Does STA already exist? */
        for (i = 4; i < TOTAL_CAM_ENTRY; i++) {
                addr = rtlpriv->sec.hwsec_cam_sta_addr[i];
                bitmap = (rtlpriv->sec.hwsec_cam_bitmap) >> i;
                if (((bitmap & BIT(0)) == BIT(0)) &&
                    (ether_addr_equal_unaligned(addr, sta_addr))) {
                        /* Remove from HW Security CAM */
                        eth_zero_addr(rtlpriv->sec.hwsec_cam_sta_addr[i]);
                        rtlpriv->sec.hwsec_cam_bitmap &= ~(BIT(0) << i);
                        rtl_dbg(rtlpriv, COMP_SEC, DBG_DMESG,
                                "&&&&&&&&&del entry %d\n", i);
                }
        }
        return;
}
EXPORT_SYMBOL(rtl_cam_del_entry);