return (struct ath11k_vif *)vif->drv_priv;
 }
 
+static inline struct ath11k_sta *ath11k_sta_to_arsta(struct ieee80211_sta *sta)
+{
+       return (struct ath11k_sta *)sta->drv_priv;
+}
+
 static inline struct ath11k *ath11k_ab_to_ar(struct ath11k_base *ab,
                                             int mac_id)
 {
 
                                          struct ieee80211_sta *sta)
 {
        struct ath11k *ar = data;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
 
        spin_lock_bh(&ar->data_lock);
        arsta->ps_total_duration = 0;
                                         struct ieee80211_sta *sta)
 {
        struct ath11k *ar = data;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
 
        spin_lock_bh(&ar->data_lock);
        arsta->peer_ps_state = WMI_PEER_PS_STATE_DISABLED;
 
                                            size_t count, loff_t *ppos)
 {
        struct ieee80211_sta *sta = file->private_data;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        struct ath11k_htt_data_stats *stats;
        static const char *str_name[ATH11K_STATS_TYPE_MAX] = {"succ", "fail",
                                            size_t count, loff_t *ppos)
 {
        struct ieee80211_sta *sta = file->private_data;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        struct ath11k_rx_peer_stats *rx_stats = arsta->rx_stats;
        int len = 0, i, retval = 0;
 ath11k_dbg_sta_open_htt_peer_stats(struct inode *inode, struct file *file)
 {
        struct ieee80211_sta *sta = inode->i_private;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        struct debug_htt_stats_req *stats_req;
        int type = ar->debug.htt_stats.type;
 ath11k_dbg_sta_release_htt_peer_stats(struct inode *inode, struct file *file)
 {
        struct ieee80211_sta *sta = inode->i_private;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
 
        mutex_lock(&ar->conf_mutex);
                                                size_t count, loff_t *ppos)
 {
        struct ieee80211_sta *sta = file->private_data;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        int ret, enable;
 
                                               size_t count, loff_t *ppos)
 {
        struct ieee80211_sta *sta = file->private_data;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        char buf[32] = {0};
        int len;
                                          size_t count, loff_t *ppos)
 {
        struct ieee80211_sta *sta = file->private_data;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        u32 tid, initiator, reason;
        int ret;
                                               size_t count, loff_t *ppos)
 {
        struct ieee80211_sta *sta = file->private_data;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        u32 tid, status;
        int ret;
                                          size_t count, loff_t *ppos)
 {
        struct ieee80211_sta *sta = file->private_data;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        u32 tid, buf_size;
        int ret;
                                             size_t count, loff_t *ppos)
 {
        struct ieee80211_sta *sta = file->private_data;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        char buf[64];
        int len = 0;
                                              size_t count, loff_t *ppos)
 {
        struct ieee80211_sta *sta = file->private_data;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        u32 aggr_mode;
        int ret;
                                  size_t count, loff_t *ppos)
 {
        struct ieee80211_sta *sta = file->private_data;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        struct htt_ext_stats_cfg_params cfg_params = { 0 };
        int ret;
                                                 size_t count, loff_t *ppos)
 {
        struct ieee80211_sta *sta = file->private_data;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        char buf[20];
        int len;
                                                       loff_t *ppos)
 {
        struct ieee80211_sta *sta = file->private_data;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        u64 time_since_station_in_power_save;
        char buf[20];
                                                     size_t count, loff_t *ppos)
 {
        struct ieee80211_sta *sta = file->private_data;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        char buf[20];
        u64 power_save_duration;
 
                             struct ieee80211_ampdu_params *params)
 {
        struct ath11k_base *ab = ar->ab;
-       struct ath11k_sta *arsta = (void *)params->sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(params->sta);
        int vdev_id = arsta->arvif->vdev_id;
        int ret;
 
 {
        struct ath11k_base *ab = ar->ab;
        struct ath11k_peer *peer;
-       struct ath11k_sta *arsta = (void *)params->sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(params->sta);
        int vdev_id = arsta->arvif->vdev_id;
        dma_addr_t paddr;
        bool active;
        }
 
        sta = peer->sta;
-       arsta = (struct ath11k_sta *)sta->drv_priv;
+       arsta = ath11k_sta_to_arsta(sta);
 
        memset(&arsta->txrate, 0, sizeof(arsta->txrate));
 
                        goto next_skb;
                }
 
-               arsta = (struct ath11k_sta *)peer->sta->drv_priv;
+               arsta = ath11k_sta_to_arsta(peer->sta);
                ath11k_dp_rx_update_peer_stats(arsta, ppdu_info);
 
                if (ath11k_debugfs_is_pktlog_peer_valid(ar, peer->addr))
 
        }
 
        sta = peer->sta;
-       arsta = (struct ath11k_sta *)sta->drv_priv;
+       arsta = ath11k_sta_to_arsta(sta);
 
        memset(&arsta->txrate, 0, sizeof(arsta->txrate));
        pkt_type = FIELD_GET(HAL_TX_RATE_STATS_INFO0_PKT_TYPE,
                ieee80211_free_txskb(ar->hw, msdu);
                return;
        }
-       arsta = (struct ath11k_sta *)peer->sta->drv_priv;
+       arsta = ath11k_sta_to_arsta(peer->sta);
        status.sta = peer->sta;
        status.skb = msdu;
        status.info = info;
 
 
        lockdep_assert_held(&ar->conf_mutex);
 
-       arsta = (struct ath11k_sta *)sta->drv_priv;
+       arsta = ath11k_sta_to_arsta(sta);
 
        memset(arg, 0, sizeof(*arg));
 
                ath11k_warn(ab, "peer %pM disappeared!\n", peer_addr);
 
        if (sta) {
-               arsta = (struct ath11k_sta *)sta->drv_priv;
+               arsta = ath11k_sta_to_arsta(sta);
 
                switch (key->cipher) {
                case WLAN_CIPHER_SUITE_TKIP:
 {
        struct ath11k_base *ab = ar->ab;
        struct ath11k_vif *arvif = ath11k_vif_to_arvif(vif);
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct peer_create_params peer_param;
        int ret;
 
 {
        struct ath11k *ar = hw->priv;
        struct ath11k_vif *arvif = ath11k_vif_to_arvif(vif);
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k_peer *peer;
        int ret = 0;
 
                                        struct ieee80211_sta *sta, bool enabled)
 {
        struct ath11k *ar = hw->priv;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
 
        if (enabled && !arsta->use_4addr_set) {
                ieee80211_queue_work(ar->hw, &arsta->set_4addr_wk);
                                        u32 changed)
 {
        struct ath11k *ar = hw->priv;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k_vif *arvif = ath11k_vif_to_arvif(vif);
        struct ath11k_peer *peer;
        u32 bw, smps;
        }
 
        if (control->sta)
-               arsta = (struct ath11k_sta *)control->sta->drv_priv;
+               arsta = ath11k_sta_to_arsta(control->sta);
 
        ret = ath11k_dp_tx(ar, arvif, arsta, skb);
        if (unlikely(ret)) {
                                             struct ieee80211_sta *sta)
 {
        struct ath11k_vif *arvif = data;
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arvif->ar;
 
        spin_lock_bh(&ar->data_lock);
                                         struct ieee80211_sta *sta,
                                         struct station_info *sinfo)
 {
-       struct ath11k_sta *arsta = (struct ath11k_sta *)sta->drv_priv;
+       struct ath11k_sta *arsta = ath11k_sta_to_arsta(sta);
        struct ath11k *ar = arsta->arvif->ar;
        s8 signal;
        bool db2dbm = test_bit(WMI_TLV_SERVICE_HW_DB2DBM_CONVERSION_SUPPORT,
 
        peer->sec_type_grp = HAL_ENCRYPT_TYPE_OPEN;
 
        if (sta) {
-               arsta = (struct ath11k_sta *)sta->drv_priv;
+               arsta = ath11k_sta_to_arsta(sta);
                arsta->tcl_metadata |= FIELD_PREP(HTT_TCL_META_DATA_TYPE, 0) |
                                       FIELD_PREP(HTT_TCL_META_DATA_PEER_ID,
                                                  peer->peer_id);
 
                goto exit;
        }
 
-       arsta = (struct ath11k_sta *)sta->drv_priv;
+       arsta = ath11k_sta_to_arsta(sta);
 
        BUILD_BUG_ON(ARRAY_SIZE(arsta->chain_signal) >
                     ARRAY_SIZE(stats_rssi->rssi_avg_beacon));
                                                           arvif->bssid,
                                                           NULL);
                        if (sta) {
-                               arsta = (struct ath11k_sta *)sta->drv_priv;
+                               arsta = ath11k_sta_to_arsta(sta);
                                arsta->rssi_beacon = src->beacon_snr;
                                ath11k_dbg(ab, ATH11K_DBG_WMI,
                                           "stats vdev id %d snr %d\n",
                goto exit;
        }
 
-       arsta = (struct ath11k_sta *)sta->drv_priv;
+       arsta = ath11k_sta_to_arsta(sta);
 
        spin_lock_bh(&ar->data_lock);