net/mac80211/rate.h

   1 /*
   2  * Copyright 2002-2005, Instant802 Networks, Inc.
   3  * Copyright 2005, Devicescape Software, Inc.
   4  * Copyright (c) 2006 Jiri Benc <jbenc@suse.cz>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License version 2 as
   8  * published by the Free Software Foundation.
   9  */
  10
  11 #ifndef IEEE80211_RATE_H
  12 #define IEEE80211_RATE_H
  13
  14 #include <linux/netdevice.h>
  15 #include <linux/skbuff.h>
  16 #include <linux/types.h>
  17 #include <net/mac80211.h>
  18 #include "ieee80211_i.h"
  19 #include "sta_info.h"
  20 #include "driver-ops.h"
  21
  22 struct rate_control_ref {
  23         struct ieee80211_local *local;
  24         const struct rate_control_ops *ops;
  25         void *priv;
  26 };
  27
  28 void rate_control_get_rate(struct ieee80211_sub_if_data *sdata,
  29                            struct sta_info *sta,
  30                            struct ieee80211_tx_rate_control *txrc);
  31
  32 static inline void rate_control_tx_status(struct ieee80211_local *local,
  33                                           struct ieee80211_supported_band *sband,
  34                                           struct sta_info *sta,
  35                                           struct sk_buff *skb)
  36 {
  37         struct rate_control_ref *ref = local->rate_ctrl;
  38         struct ieee80211_sta *ista = &sta->sta;
  39         void *priv_sta = sta->rate_ctrl_priv;
  40         struct ieee80211_tx_info *info = IEEE80211_SKB_CB(skb);
  41
  42         if (!ref || !test_sta_flag(sta, WLAN_STA_RATE_CONTROL))
  43                 return;
  44
  45         spin_lock_bh(&sta->rate_ctrl_lock);
  46         if (ref->ops->tx_status)
  47                 ref->ops->tx_status(ref->priv, sband, ista, priv_sta, skb);
  48         else
  49                 ref->ops->tx_status_noskb(ref->priv, sband, ista, priv_sta, info);
  50         spin_unlock_bh(&sta->rate_ctrl_lock);
  51 }
  52
  53 static inline void
  54 rate_control_tx_status_noskb(struct ieee80211_local *local,
  55                              struct ieee80211_supported_band *sband,
  56                              struct sta_info *sta,
  57                              struct ieee80211_tx_info *info)
  58 {
  59         struct rate_control_ref *ref = local->rate_ctrl;
  60         struct ieee80211_sta *ista = &sta->sta;
  61         void *priv_sta = sta->rate_ctrl_priv;
  62
  63         if (!ref || !test_sta_flag(sta, WLAN_STA_RATE_CONTROL))
  64                 return;
  65
  66         if (WARN_ON_ONCE(!ref->ops->tx_status_noskb))
  67                 return;
  68
  69         spin_lock_bh(&sta->rate_ctrl_lock);
  70         ref->ops->tx_status_noskb(ref->priv, sband, ista, priv_sta, info);
  71         spin_unlock_bh(&sta->rate_ctrl_lock);
  72 }
  73
  74 static inline void rate_control_rate_init(struct sta_info *sta)
  75 {
  76         struct ieee80211_local *local = sta->sdata->local;
  77         struct rate_control_ref *ref = sta->rate_ctrl;
  78         struct ieee80211_sta *ista = &sta->sta;
  79         void *priv_sta = sta->rate_ctrl_priv;
  80         struct ieee80211_supported_band *sband;
  81         struct ieee80211_chanctx_conf *chanctx_conf;
  82
  83         ieee80211_sta_set_rx_nss(sta);
  84
  85         if (!ref)
  86                 return;
  87
  88         rcu_read_lock();
  89
  90         chanctx_conf = rcu_dereference(sta->sdata->vif.chanctx_conf);
  91         if (WARN_ON(!chanctx_conf)) {
  92                 rcu_read_unlock();
  93                 return;
  94         }
  95
  96         sband = local->hw.wiphy->bands[chanctx_conf->def.chan->band];
  97
  98         spin_lock_bh(&sta->rate_ctrl_lock);
  99         ref->ops->rate_init(ref->priv, sband, &chanctx_conf->def, ista,
 100                             priv_sta);
 101         spin_unlock_bh(&sta->rate_ctrl_lock);
 102         rcu_read_unlock();
 103         set_sta_flag(sta, WLAN_STA_RATE_CONTROL);
 104 }
 105
 106 static inline void rate_control_rate_update(struct ieee80211_local *local,
 107                                     struct ieee80211_supported_band *sband,
 108                                     struct sta_info *sta, u32 changed)
 109 {
 110         struct rate_control_ref *ref = local->rate_ctrl;
 111         struct ieee80211_sta *ista = &sta->sta;
 112         void *priv_sta = sta->rate_ctrl_priv;
 113         struct ieee80211_chanctx_conf *chanctx_conf;
 114
 115         if (ref && ref->ops->rate_update) {
 116                 rcu_read_lock();
 117
 118                 chanctx_conf = rcu_dereference(sta->sdata->vif.chanctx_conf);
 119                 if (WARN_ON(!chanctx_conf)) {
 120                         rcu_read_unlock();
 121                         return;
 122                 }
 123
 124                 spin_lock_bh(&sta->rate_ctrl_lock);
 125                 ref->ops->rate_update(ref->priv, sband, &chanctx_conf->def,
 126                                       ista, priv_sta, changed);
 127                 spin_unlock_bh(&sta->rate_ctrl_lock);
 128                 rcu_read_unlock();
 129         }
 130         drv_sta_rc_update(local, sta->sdata, &sta->sta, changed);
 131 }
 132
 133 static inline void *rate_control_alloc_sta(struct rate_control_ref *ref,
 134                                            struct sta_info *sta, gfp_t gfp)
 135 {
 136         spin_lock_init(&sta->rate_ctrl_lock);
 137         return ref->ops->alloc_sta(ref->priv, &sta->sta, gfp);
 138 }
 139
 140 static inline void rate_control_free_sta(struct sta_info *sta)
 141 {
 142         struct rate_control_ref *ref = sta->rate_ctrl;
 143         struct ieee80211_sta *ista = &sta->sta;
 144         void *priv_sta = sta->rate_ctrl_priv;
 145
 146         ref->ops->free_sta(ref->priv, ista, priv_sta);
 147 }
 148
 149 static inline void rate_control_add_sta_debugfs(struct sta_info *sta)
 150 {
 151 #ifdef CONFIG_MAC80211_DEBUGFS
 152         struct rate_control_ref *ref = sta->rate_ctrl;
 153         if (ref && sta->debugfs.dir && ref->ops->add_sta_debugfs)
 154                 ref->ops->add_sta_debugfs(ref->priv, sta->rate_ctrl_priv,
 155                                           sta->debugfs.dir);
 156 #endif
 157 }
 158
 159 static inline void rate_control_remove_sta_debugfs(struct sta_info *sta)
 160 {
 161 #ifdef CONFIG_MAC80211_DEBUGFS
 162         struct rate_control_ref *ref = sta->rate_ctrl;
 163         if (ref && ref->ops->remove_sta_debugfs)
 164                 ref->ops->remove_sta_debugfs(ref->priv, sta->rate_ctrl_priv);
 165 #endif
 166 }
 167
 168 /* Get a reference to the rate control algorithm. If `name' is NULL, get the
 169  * first available algorithm. */
 170 int ieee80211_init_rate_ctrl_alg(struct ieee80211_local *local,
 171                                  const char *name);
 172 void rate_control_deinitialize(struct ieee80211_local *local);
 173
 174
 175 /* Rate control algorithms */
 176 #ifdef CONFIG_MAC80211_RC_MINSTREL
 177 int rc80211_minstrel_init(void);
 178 void rc80211_minstrel_exit(void);
 179 #else
 180 static inline int rc80211_minstrel_init(void)
 181 {
 182         return 0;
 183 }
 184 static inline void rc80211_minstrel_exit(void)
 185 {
 186 }
 187 #endif
 188
 189 #ifdef CONFIG_MAC80211_RC_MINSTREL_HT
 190 int rc80211_minstrel_ht_init(void);
 191 void rc80211_minstrel_ht_exit(void);
 192 #else
 193 static inline int rc80211_minstrel_ht_init(void)
 194 {
 195         return 0;
 196 }
 197 static inline void rc80211_minstrel_ht_exit(void)
 198 {
 199 }
 200 #endif
 201
 202
 203 #endif /* IEEE80211_RATE_H */