Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2002-2005, Instant802 Networks, Inc. |
| 3 | * Copyright 2005, Devicescape Software, Inc. |
| 4 | * Copyright 2007, Mattias Nissler <mattias.nissler@gmx.de> |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 5 | * Copyright 2007, Stefano Brivio <stefano.brivio@polimi.it> |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 6 | * |
| 7 | * This program is free software; you can redistribute it and/or modify |
| 8 | * it under the terms of the GNU General Public License version 2 as |
| 9 | * published by the Free Software Foundation. |
| 10 | */ |
| 11 | |
| 12 | #include <linux/netdevice.h> |
| 13 | #include <linux/types.h> |
| 14 | #include <linux/skbuff.h> |
Johannes Berg | 4b47589 | 2008-01-02 15:17:03 +0100 | [diff] [blame] | 15 | #include <linux/debugfs.h> |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 16 | #include <net/mac80211.h> |
| 17 | #include "ieee80211_rate.h" |
| 18 | |
Mattias Nissler | 12446c6 | 2007-12-19 01:27:18 +0100 | [diff] [blame] | 19 | #include "rc80211_pid.h" |
| 20 | |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 21 | |
| 22 | /* This is an implementation of a TX rate control algorithm that uses a PID |
| 23 | * controller. Given a target failed frames rate, the controller decides about |
| 24 | * TX rate changes to meet the target failed frames rate. |
| 25 | * |
| 26 | * The controller basically computes the following: |
| 27 | * |
Stefano Brivio | 1dc4d1e | 2007-12-19 01:26:52 +0100 | [diff] [blame] | 28 | * adj = CP * err + CI * err_avg + CD * (err - last_err) * (1 + sharpening) |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 29 | * |
| 30 | * where |
| 31 | * adj adjustment value that is used to switch TX rate (see below) |
| 32 | * err current error: target vs. current failed frames percentage |
| 33 | * last_err last error |
| 34 | * err_avg average (i.e. poor man's integral) of recent errors |
Stefano Brivio | 1dc4d1e | 2007-12-19 01:26:52 +0100 | [diff] [blame] | 35 | * sharpening non-zero when fast response is needed (i.e. right after |
| 36 | * association or no frames sent for a long time), heading |
| 37 | * to zero over time |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 38 | * CP Proportional coefficient |
| 39 | * CI Integral coefficient |
| 40 | * CD Derivative coefficient |
| 41 | * |
| 42 | * CP, CI, CD are subject to careful tuning. |
| 43 | * |
| 44 | * The integral component uses a exponential moving average approach instead of |
| 45 | * an actual sliding window. The advantage is that we don't need to keep an |
| 46 | * array of the last N error values and computation is easier. |
| 47 | * |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 48 | * Once we have the adj value, we map it to a rate by means of a learning |
| 49 | * algorithm. This algorithm keeps the state of the percentual failed frames |
| 50 | * difference between rates. The behaviour of the lowest available rate is kept |
| 51 | * as a reference value, and every time we switch between two rates, we compute |
| 52 | * the difference between the failed frames each rate exhibited. By doing so, |
| 53 | * we compare behaviours which different rates exhibited in adjacent timeslices, |
| 54 | * thus the comparison is minimally affected by external conditions. This |
| 55 | * difference gets propagated to the whole set of measurements, so that the |
| 56 | * reference is always the same. Periodically, we normalize this set so that |
| 57 | * recent events weigh the most. By comparing the adj value with this set, we |
| 58 | * avoid pejorative switches to lower rates and allow for switches to higher |
| 59 | * rates if they behaved well. |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 60 | * |
| 61 | * Note that for the computations we use a fixed-point representation to avoid |
| 62 | * floating point arithmetic. Hence, all values are shifted left by |
| 63 | * RC_PID_ARITH_SHIFT. |
| 64 | */ |
| 65 | |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 66 | |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 67 | /* Shift the adjustment so that we won't switch to a lower rate if it exhibited |
| 68 | * a worse failed frames behaviour and we'll choose the highest rate whose |
| 69 | * failed frames behaviour is not worse than the one of the original rate |
| 70 | * target. While at it, check that the adjustment is within the ranges. Then, |
| 71 | * provide the new rate index. */ |
| 72 | static int rate_control_pid_shift_adjust(struct rc_pid_rateinfo *r, |
| 73 | int adj, int cur, int l) |
| 74 | { |
| 75 | int i, j, k, tmp; |
| 76 | |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 77 | j = r[cur].rev_index; |
Stefano Brivio | fa44327 | 2007-12-23 04:41:19 +0100 | [diff] [blame] | 78 | i = j + adj; |
| 79 | |
| 80 | if (i < 0) |
| 81 | return r[0].index; |
| 82 | if (i >= l - 1) |
| 83 | return r[l - 1].index; |
| 84 | |
| 85 | tmp = i; |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 86 | |
| 87 | if (adj < 0) { |
Stefano Brivio | fa44327 | 2007-12-23 04:41:19 +0100 | [diff] [blame] | 88 | for (k = j; k >= i; k--) |
| 89 | if (r[k].diff <= r[j].diff) |
| 90 | tmp = k; |
| 91 | } else { |
| 92 | for (k = i + 1; k + i < l; k++) |
| 93 | if (r[k].diff <= r[i].diff) |
| 94 | tmp = k; |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 95 | } |
Stefano Brivio | fa44327 | 2007-12-23 04:41:19 +0100 | [diff] [blame] | 96 | |
| 97 | return r[tmp].index; |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 98 | } |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 99 | |
| 100 | static void rate_control_pid_adjust_rate(struct ieee80211_local *local, |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 101 | struct sta_info *sta, int adj, |
| 102 | struct rc_pid_rateinfo *rinfo) |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 103 | { |
| 104 | struct ieee80211_sub_if_data *sdata; |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 105 | struct ieee80211_supported_band *sband; |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 106 | int newidx; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 107 | int maxrate; |
| 108 | int back = (adj > 0) ? 1 : -1; |
| 109 | |
| 110 | sdata = IEEE80211_DEV_TO_SUB_IF(sta->dev); |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 111 | |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 112 | sband = local->hw.wiphy->bands[local->hw.conf.channel->band]; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 113 | maxrate = sdata->bss ? sdata->bss->max_ratectrl_rateidx : -1; |
| 114 | |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 115 | newidx = rate_control_pid_shift_adjust(rinfo, adj, sta->txrate_idx, |
| 116 | sband->n_bitrates); |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 117 | |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 118 | while (newidx != sta->txrate_idx) { |
| 119 | if (rate_supported(sta, sband->band, newidx) && |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 120 | (maxrate < 0 || newidx <= maxrate)) { |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 121 | sta->txrate_idx = newidx; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 122 | break; |
| 123 | } |
| 124 | |
| 125 | newidx += back; |
| 126 | } |
Mattias Nissler | 12446c6 | 2007-12-19 01:27:18 +0100 | [diff] [blame] | 127 | |
| 128 | #ifdef CONFIG_MAC80211_DEBUGFS |
| 129 | rate_control_pid_event_rate_change( |
| 130 | &((struct rc_pid_sta_info *)sta->rate_ctrl_priv)->events, |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 131 | newidx, sband->bitrates[newidx].bitrate); |
Mattias Nissler | 12446c6 | 2007-12-19 01:27:18 +0100 | [diff] [blame] | 132 | #endif |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 133 | } |
| 134 | |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 135 | /* Normalize the failed frames per-rate differences. */ |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 136 | static void rate_control_pid_normalize(struct rc_pid_info *pinfo, int l) |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 137 | { |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 138 | int i, norm_offset = pinfo->norm_offset; |
| 139 | struct rc_pid_rateinfo *r = pinfo->rinfo; |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 140 | |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 141 | if (r[0].diff > norm_offset) |
| 142 | r[0].diff -= norm_offset; |
| 143 | else if (r[0].diff < -norm_offset) |
| 144 | r[0].diff += norm_offset; |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 145 | for (i = 0; i < l - 1; i++) |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 146 | if (r[i + 1].diff > r[i].diff + norm_offset) |
| 147 | r[i + 1].diff -= norm_offset; |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 148 | else if (r[i + 1].diff <= r[i].diff) |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 149 | r[i + 1].diff += norm_offset; |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 150 | } |
| 151 | |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 152 | static void rate_control_pid_sample(struct rc_pid_info *pinfo, |
| 153 | struct ieee80211_local *local, |
| 154 | struct sta_info *sta) |
| 155 | { |
| 156 | struct rc_pid_sta_info *spinfo = sta->rate_ctrl_priv; |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 157 | struct rc_pid_rateinfo *rinfo = pinfo->rinfo; |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 158 | struct ieee80211_supported_band *sband; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 159 | u32 pf; |
| 160 | s32 err_avg; |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 161 | u32 err_prop; |
| 162 | u32 err_int; |
| 163 | u32 err_der; |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 164 | int adj, i, j, tmp; |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 165 | unsigned long period; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 166 | |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 167 | sband = local->hw.wiphy->bands[local->hw.conf.channel->band]; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 168 | spinfo = sta->rate_ctrl_priv; |
Stefano Brivio | 1dc4d1e | 2007-12-19 01:26:52 +0100 | [diff] [blame] | 169 | |
| 170 | /* In case nothing happened during the previous control interval, turn |
| 171 | * the sharpening factor on. */ |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 172 | period = (HZ * pinfo->sampling_period + 500) / 1000; |
| 173 | if (!period) |
| 174 | period = 1; |
| 175 | if (jiffies - spinfo->last_sample > 2 * period) |
| 176 | spinfo->sharp_cnt = pinfo->sharpen_duration; |
Stefano Brivio | 1dc4d1e | 2007-12-19 01:26:52 +0100 | [diff] [blame] | 177 | |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 178 | spinfo->last_sample = jiffies; |
| 179 | |
Stefano Brivio | 1dc4d1e | 2007-12-19 01:26:52 +0100 | [diff] [blame] | 180 | /* This should never happen, but in case, we assume the old sample is |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 181 | * still a good measurement and copy it. */ |
Stefano Brivio | 1dc4d1e | 2007-12-19 01:26:52 +0100 | [diff] [blame] | 182 | if (unlikely(spinfo->tx_num_xmit == 0)) |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 183 | pf = spinfo->last_pf; |
| 184 | else { |
| 185 | pf = spinfo->tx_num_failed * 100 / spinfo->tx_num_xmit; |
| 186 | pf <<= RC_PID_ARITH_SHIFT; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 187 | } |
| 188 | |
Stefano Brivio | 1dc4d1e | 2007-12-19 01:26:52 +0100 | [diff] [blame] | 189 | spinfo->tx_num_xmit = 0; |
| 190 | spinfo->tx_num_failed = 0; |
| 191 | |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 192 | /* If we just switched rate, update the rate behaviour info. */ |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 193 | if (pinfo->oldrate != sta->txrate_idx) { |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 194 | |
| 195 | i = rinfo[pinfo->oldrate].rev_index; |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 196 | j = rinfo[sta->txrate_idx].rev_index; |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 197 | |
| 198 | tmp = (pf - spinfo->last_pf); |
| 199 | tmp = RC_PID_DO_ARITH_RIGHT_SHIFT(tmp, RC_PID_ARITH_SHIFT); |
| 200 | |
| 201 | rinfo[j].diff = rinfo[i].diff + tmp; |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 202 | pinfo->oldrate = sta->txrate_idx; |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 203 | } |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 204 | rate_control_pid_normalize(pinfo, sband->n_bitrates); |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 205 | |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 206 | /* Compute the proportional, integral and derivative errors. */ |
Stefano Brivio | 426706c | 2007-12-23 04:39:17 +0100 | [diff] [blame] | 207 | err_prop = (pinfo->target << RC_PID_ARITH_SHIFT) - pf; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 208 | |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 209 | err_avg = spinfo->err_avg_sc >> pinfo->smoothing_shift; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 210 | spinfo->err_avg_sc = spinfo->err_avg_sc - err_avg + err_prop; |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 211 | err_int = spinfo->err_avg_sc >> pinfo->smoothing_shift; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 212 | |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 213 | err_der = (pf - spinfo->last_pf) * |
| 214 | (1 + pinfo->sharpen_factor * spinfo->sharp_cnt); |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 215 | spinfo->last_pf = pf; |
Stefano Brivio | 1dc4d1e | 2007-12-19 01:26:52 +0100 | [diff] [blame] | 216 | if (spinfo->sharp_cnt) |
| 217 | spinfo->sharp_cnt--; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 218 | |
Mattias Nissler | 12446c6 | 2007-12-19 01:27:18 +0100 | [diff] [blame] | 219 | #ifdef CONFIG_MAC80211_DEBUGFS |
| 220 | rate_control_pid_event_pf_sample(&spinfo->events, pf, err_prop, err_int, |
| 221 | err_der); |
| 222 | #endif |
| 223 | |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 224 | /* Compute the controller output. */ |
| 225 | adj = (err_prop * pinfo->coeff_p + err_int * pinfo->coeff_i |
| 226 | + err_der * pinfo->coeff_d); |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 227 | adj = RC_PID_DO_ARITH_RIGHT_SHIFT(adj, 2 * RC_PID_ARITH_SHIFT); |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 228 | |
| 229 | /* Change rate. */ |
| 230 | if (adj) |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 231 | rate_control_pid_adjust_rate(local, sta, adj, rinfo); |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 232 | } |
| 233 | |
| 234 | static void rate_control_pid_tx_status(void *priv, struct net_device *dev, |
| 235 | struct sk_buff *skb, |
| 236 | struct ieee80211_tx_status *status) |
| 237 | { |
| 238 | struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr); |
| 239 | struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data; |
Andrew Lutomirski | df26e7e | 2008-01-03 21:05:37 -0800 | [diff] [blame] | 240 | struct ieee80211_sub_if_data *sdata; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 241 | struct rc_pid_info *pinfo = priv; |
| 242 | struct sta_info *sta; |
| 243 | struct rc_pid_sta_info *spinfo; |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 244 | unsigned long period; |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 245 | struct ieee80211_supported_band *sband; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 246 | |
| 247 | sta = sta_info_get(local, hdr->addr1); |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 248 | sband = local->hw.wiphy->bands[local->hw.conf.channel->band]; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 249 | |
| 250 | if (!sta) |
| 251 | return; |
| 252 | |
Andrew Lutomirski | df26e7e | 2008-01-03 21:05:37 -0800 | [diff] [blame] | 253 | /* Don't update the state if we're not controlling the rate. */ |
| 254 | sdata = IEEE80211_DEV_TO_SUB_IF(sta->dev); |
| 255 | if (sdata->bss && sdata->bss->force_unicast_rateidx > -1) { |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 256 | sta->txrate_idx = sdata->bss->max_ratectrl_rateidx; |
Andrew Lutomirski | df26e7e | 2008-01-03 21:05:37 -0800 | [diff] [blame] | 257 | return; |
| 258 | } |
| 259 | |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 260 | /* Ignore all frames that were sent with a different rate than the rate |
| 261 | * we currently advise mac80211 to use. */ |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 262 | if (status->control.tx_rate != &sband->bitrates[sta->txrate_idx]) |
Stefano Brivio | 13e05aa | 2007-12-23 04:43:57 +0100 | [diff] [blame] | 263 | goto ignore; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 264 | |
| 265 | spinfo = sta->rate_ctrl_priv; |
| 266 | spinfo->tx_num_xmit++; |
| 267 | |
Mattias Nissler | 12446c6 | 2007-12-19 01:27:18 +0100 | [diff] [blame] | 268 | #ifdef CONFIG_MAC80211_DEBUGFS |
| 269 | rate_control_pid_event_tx_status(&spinfo->events, status); |
| 270 | #endif |
| 271 | |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 272 | /* We count frames that totally failed to be transmitted as two bad |
| 273 | * frames, those that made it out but had some retries as one good and |
| 274 | * one bad frame. */ |
| 275 | if (status->excessive_retries) { |
| 276 | spinfo->tx_num_failed += 2; |
| 277 | spinfo->tx_num_xmit++; |
| 278 | } else if (status->retry_count) { |
| 279 | spinfo->tx_num_failed++; |
| 280 | spinfo->tx_num_xmit++; |
| 281 | } |
| 282 | |
| 283 | if (status->excessive_retries) { |
| 284 | sta->tx_retry_failed++; |
| 285 | sta->tx_num_consecutive_failures++; |
| 286 | sta->tx_num_mpdu_fail++; |
| 287 | } else { |
| 288 | sta->last_ack_rssi[0] = sta->last_ack_rssi[1]; |
| 289 | sta->last_ack_rssi[1] = sta->last_ack_rssi[2]; |
| 290 | sta->last_ack_rssi[2] = status->ack_signal; |
| 291 | sta->tx_num_consecutive_failures = 0; |
| 292 | sta->tx_num_mpdu_ok++; |
| 293 | } |
| 294 | sta->tx_retry_count += status->retry_count; |
| 295 | sta->tx_num_mpdu_fail += status->retry_count; |
| 296 | |
| 297 | /* Update PID controller state. */ |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 298 | period = (HZ * pinfo->sampling_period + 500) / 1000; |
| 299 | if (!period) |
| 300 | period = 1; |
| 301 | if (time_after(jiffies, spinfo->last_sample + period)) |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 302 | rate_control_pid_sample(pinfo, local, sta); |
| 303 | |
Stefano Brivio | 13e05aa | 2007-12-23 04:43:57 +0100 | [diff] [blame] | 304 | ignore: |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 305 | sta_info_put(sta); |
| 306 | } |
| 307 | |
| 308 | static void rate_control_pid_get_rate(void *priv, struct net_device *dev, |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 309 | struct ieee80211_supported_band *sband, |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 310 | struct sk_buff *skb, |
| 311 | struct rate_selection *sel) |
| 312 | { |
| 313 | struct ieee80211_local *local = wdev_priv(dev->ieee80211_ptr); |
| 314 | struct ieee80211_hdr *hdr = (struct ieee80211_hdr *) skb->data; |
Michael Wu | 2bc454b | 2007-12-25 19:33:16 -0500 | [diff] [blame] | 315 | struct ieee80211_sub_if_data *sdata; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 316 | struct sta_info *sta; |
| 317 | int rateidx; |
Michael Wu | 2bc454b | 2007-12-25 19:33:16 -0500 | [diff] [blame] | 318 | u16 fc; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 319 | |
| 320 | sta = sta_info_get(local, hdr->addr1); |
| 321 | |
Michael Wu | 2bc454b | 2007-12-25 19:33:16 -0500 | [diff] [blame] | 322 | /* Send management frames and broadcast/multicast data using lowest |
| 323 | * rate. */ |
| 324 | fc = le16_to_cpu(hdr->frame_control); |
| 325 | if ((fc & IEEE80211_FCTL_FTYPE) != IEEE80211_FTYPE_DATA || |
| 326 | is_multicast_ether_addr(hdr->addr1) || !sta) { |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 327 | sel->rate = rate_lowest(local, sband, sta); |
Michael Wu | 2bc454b | 2007-12-25 19:33:16 -0500 | [diff] [blame] | 328 | if (sta) |
| 329 | sta_info_put(sta); |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 330 | return; |
| 331 | } |
| 332 | |
Michael Wu | 2bc454b | 2007-12-25 19:33:16 -0500 | [diff] [blame] | 333 | /* If a forced rate is in effect, select it. */ |
| 334 | sdata = IEEE80211_DEV_TO_SUB_IF(dev); |
| 335 | if (sdata->bss && sdata->bss->force_unicast_rateidx > -1) |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 336 | sta->txrate_idx = sdata->bss->force_unicast_rateidx; |
Michael Wu | 2bc454b | 2007-12-25 19:33:16 -0500 | [diff] [blame] | 337 | |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 338 | rateidx = sta->txrate_idx; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 339 | |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 340 | if (rateidx >= sband->n_bitrates) |
| 341 | rateidx = sband->n_bitrates - 1; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 342 | |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 343 | sta->last_txrate_idx = rateidx; |
Michael Wu | 2bc454b | 2007-12-25 19:33:16 -0500 | [diff] [blame] | 344 | |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 345 | sta_info_put(sta); |
| 346 | |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 347 | sel->rate = &sband->bitrates[rateidx]; |
Mattias Nissler | 12446c6 | 2007-12-19 01:27:18 +0100 | [diff] [blame] | 348 | |
| 349 | #ifdef CONFIG_MAC80211_DEBUGFS |
| 350 | rate_control_pid_event_tx_rate( |
| 351 | &((struct rc_pid_sta_info *) sta->rate_ctrl_priv)->events, |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 352 | rateidx, sband->bitrates[rateidx].bitrate); |
Mattias Nissler | 12446c6 | 2007-12-19 01:27:18 +0100 | [diff] [blame] | 353 | #endif |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 354 | } |
| 355 | |
| 356 | static void rate_control_pid_rate_init(void *priv, void *priv_sta, |
| 357 | struct ieee80211_local *local, |
| 358 | struct sta_info *sta) |
| 359 | { |
| 360 | /* TODO: This routine should consider using RSSI from previous packets |
| 361 | * as we need to have IEEE 802.1X auth succeed immediately after assoc.. |
| 362 | * Until that method is implemented, we will use the lowest supported |
| 363 | * rate as a workaround. */ |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 364 | struct ieee80211_supported_band *sband; |
| 365 | |
| 366 | sband = local->hw.wiphy->bands[local->hw.conf.channel->band]; |
| 367 | sta->txrate_idx = rate_lowest_index(local, sband, sta); |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 368 | } |
| 369 | |
| 370 | static void *rate_control_pid_alloc(struct ieee80211_local *local) |
| 371 | { |
| 372 | struct rc_pid_info *pinfo; |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 373 | struct rc_pid_rateinfo *rinfo; |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 374 | struct ieee80211_supported_band *sband; |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 375 | int i, j, tmp; |
| 376 | bool s; |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 377 | #ifdef CONFIG_MAC80211_DEBUGFS |
| 378 | struct rc_pid_debugfs_entries *de; |
| 379 | #endif |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 380 | |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 381 | sband = local->hw.wiphy->bands[local->hw.conf.channel->band]; |
| 382 | |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 383 | pinfo = kmalloc(sizeof(*pinfo), GFP_ATOMIC); |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 384 | if (!pinfo) |
| 385 | return NULL; |
| 386 | |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 387 | /* We can safely assume that sband won't change unless we get |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 388 | * reinitialized. */ |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 389 | rinfo = kmalloc(sizeof(*rinfo) * sband->n_bitrates, GFP_ATOMIC); |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 390 | if (!rinfo) { |
| 391 | kfree(pinfo); |
| 392 | return NULL; |
| 393 | } |
| 394 | |
| 395 | /* Sort the rates. This is optimized for the most common case (i.e. |
| 396 | * almost-sorted CCK+OFDM rates). Kind of bubble-sort with reversed |
| 397 | * mapping too. */ |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 398 | for (i = 0; i < sband->n_bitrates; i++) { |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 399 | rinfo[i].index = i; |
| 400 | rinfo[i].rev_index = i; |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 401 | if (pinfo->fast_start) |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 402 | rinfo[i].diff = 0; |
| 403 | else |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 404 | rinfo[i].diff = i * pinfo->norm_offset; |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 405 | } |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 406 | for (i = 1; i < sband->n_bitrates; i++) { |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 407 | s = 0; |
Johannes Berg | 8318d78 | 2008-01-24 19:38:38 +0100 | [diff] [blame^] | 408 | for (j = 0; j < sband->n_bitrates - i; j++) |
| 409 | if (unlikely(sband->bitrates[rinfo[j].index].bitrate > |
| 410 | sband->bitrates[rinfo[j + 1].index].bitrate)) { |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 411 | tmp = rinfo[j].index; |
| 412 | rinfo[j].index = rinfo[j + 1].index; |
| 413 | rinfo[j + 1].index = tmp; |
| 414 | rinfo[rinfo[j].index].rev_index = j; |
| 415 | rinfo[rinfo[j + 1].index].rev_index = j + 1; |
| 416 | s = 1; |
| 417 | } |
| 418 | if (!s) |
| 419 | break; |
| 420 | } |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 421 | |
| 422 | pinfo->target = RC_PID_TARGET_PF; |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 423 | pinfo->sampling_period = RC_PID_INTERVAL; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 424 | pinfo->coeff_p = RC_PID_COEFF_P; |
| 425 | pinfo->coeff_i = RC_PID_COEFF_I; |
| 426 | pinfo->coeff_d = RC_PID_COEFF_D; |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 427 | pinfo->smoothing_shift = RC_PID_SMOOTHING_SHIFT; |
| 428 | pinfo->sharpen_factor = RC_PID_SHARPENING_FACTOR; |
| 429 | pinfo->sharpen_duration = RC_PID_SHARPENING_DURATION; |
| 430 | pinfo->norm_offset = RC_PID_NORM_OFFSET; |
| 431 | pinfo->fast_start = RC_PID_FAST_START; |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 432 | pinfo->rinfo = rinfo; |
| 433 | pinfo->oldrate = 0; |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 434 | |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 435 | #ifdef CONFIG_MAC80211_DEBUGFS |
| 436 | de = &pinfo->dentries; |
| 437 | de->dir = debugfs_create_dir("rc80211_pid", |
| 438 | local->hw.wiphy->debugfsdir); |
| 439 | de->target = debugfs_create_u32("target_pf", S_IRUSR | S_IWUSR, |
| 440 | de->dir, &pinfo->target); |
| 441 | de->sampling_period = debugfs_create_u32("sampling_period", |
| 442 | S_IRUSR | S_IWUSR, de->dir, |
| 443 | &pinfo->sampling_period); |
| 444 | de->coeff_p = debugfs_create_u32("coeff_p", S_IRUSR | S_IWUSR, |
| 445 | de->dir, &pinfo->coeff_p); |
| 446 | de->coeff_i = debugfs_create_u32("coeff_i", S_IRUSR | S_IWUSR, |
| 447 | de->dir, &pinfo->coeff_i); |
| 448 | de->coeff_d = debugfs_create_u32("coeff_d", S_IRUSR | S_IWUSR, |
| 449 | de->dir, &pinfo->coeff_d); |
| 450 | de->smoothing_shift = debugfs_create_u32("smoothing_shift", |
| 451 | S_IRUSR | S_IWUSR, de->dir, |
| 452 | &pinfo->smoothing_shift); |
| 453 | de->sharpen_factor = debugfs_create_u32("sharpen_factor", |
| 454 | S_IRUSR | S_IWUSR, de->dir, |
| 455 | &pinfo->sharpen_factor); |
| 456 | de->sharpen_duration = debugfs_create_u32("sharpen_duration", |
| 457 | S_IRUSR | S_IWUSR, de->dir, |
| 458 | &pinfo->sharpen_duration); |
| 459 | de->norm_offset = debugfs_create_u32("norm_offset", |
| 460 | S_IRUSR | S_IWUSR, de->dir, |
| 461 | &pinfo->norm_offset); |
| 462 | de->fast_start = debugfs_create_bool("fast_start", |
| 463 | S_IRUSR | S_IWUSR, de->dir, |
| 464 | &pinfo->fast_start); |
| 465 | #endif |
| 466 | |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 467 | return pinfo; |
| 468 | } |
| 469 | |
| 470 | static void rate_control_pid_free(void *priv) |
| 471 | { |
| 472 | struct rc_pid_info *pinfo = priv; |
Mattias Nissler | 1946b74 | 2007-12-20 13:27:26 +0100 | [diff] [blame] | 473 | #ifdef CONFIG_MAC80211_DEBUGFS |
| 474 | struct rc_pid_debugfs_entries *de = &pinfo->dentries; |
| 475 | |
| 476 | debugfs_remove(de->fast_start); |
| 477 | debugfs_remove(de->norm_offset); |
| 478 | debugfs_remove(de->sharpen_duration); |
| 479 | debugfs_remove(de->sharpen_factor); |
| 480 | debugfs_remove(de->smoothing_shift); |
| 481 | debugfs_remove(de->coeff_d); |
| 482 | debugfs_remove(de->coeff_i); |
| 483 | debugfs_remove(de->coeff_p); |
| 484 | debugfs_remove(de->sampling_period); |
| 485 | debugfs_remove(de->target); |
| 486 | debugfs_remove(de->dir); |
| 487 | #endif |
| 488 | |
Stefano Brivio | 90d501d | 2007-12-19 01:26:34 +0100 | [diff] [blame] | 489 | kfree(pinfo->rinfo); |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 490 | kfree(pinfo); |
| 491 | } |
| 492 | |
| 493 | static void rate_control_pid_clear(void *priv) |
| 494 | { |
| 495 | } |
| 496 | |
| 497 | static void *rate_control_pid_alloc_sta(void *priv, gfp_t gfp) |
| 498 | { |
| 499 | struct rc_pid_sta_info *spinfo; |
| 500 | |
| 501 | spinfo = kzalloc(sizeof(*spinfo), gfp); |
Mattias Nissler | 12446c6 | 2007-12-19 01:27:18 +0100 | [diff] [blame] | 502 | if (spinfo == NULL) |
| 503 | return NULL; |
| 504 | |
Stefano Brivio | c09c723 | 2008-01-17 00:38:29 +0100 | [diff] [blame] | 505 | spinfo->last_sample = jiffies; |
| 506 | |
Mattias Nissler | 12446c6 | 2007-12-19 01:27:18 +0100 | [diff] [blame] | 507 | #ifdef CONFIG_MAC80211_DEBUGFS |
| 508 | spin_lock_init(&spinfo->events.lock); |
| 509 | init_waitqueue_head(&spinfo->events.waitqueue); |
| 510 | #endif |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 511 | |
| 512 | return spinfo; |
| 513 | } |
| 514 | |
| 515 | static void rate_control_pid_free_sta(void *priv, void *priv_sta) |
| 516 | { |
| 517 | struct rc_pid_sta_info *spinfo = priv_sta; |
| 518 | kfree(spinfo); |
| 519 | } |
| 520 | |
Johannes Berg | 4b47589 | 2008-01-02 15:17:03 +0100 | [diff] [blame] | 521 | static struct rate_control_ops mac80211_rcpid = { |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 522 | .name = "pid", |
| 523 | .tx_status = rate_control_pid_tx_status, |
| 524 | .get_rate = rate_control_pid_get_rate, |
| 525 | .rate_init = rate_control_pid_rate_init, |
| 526 | .clear = rate_control_pid_clear, |
| 527 | .alloc = rate_control_pid_alloc, |
| 528 | .free = rate_control_pid_free, |
| 529 | .alloc_sta = rate_control_pid_alloc_sta, |
| 530 | .free_sta = rate_control_pid_free_sta, |
Mattias Nissler | 12446c6 | 2007-12-19 01:27:18 +0100 | [diff] [blame] | 531 | #ifdef CONFIG_MAC80211_DEBUGFS |
| 532 | .add_sta_debugfs = rate_control_pid_add_sta_debugfs, |
| 533 | .remove_sta_debugfs = rate_control_pid_remove_sta_debugfs, |
| 534 | #endif |
Mattias Nissler | ad01837 | 2007-12-19 01:25:57 +0100 | [diff] [blame] | 535 | }; |
Johannes Berg | 4b47589 | 2008-01-02 15:17:03 +0100 | [diff] [blame] | 536 | |
| 537 | MODULE_DESCRIPTION("PID controller based rate control algorithm"); |
| 538 | MODULE_AUTHOR("Stefano Brivio"); |
| 539 | MODULE_AUTHOR("Mattias Nissler"); |
| 540 | MODULE_LICENSE("GPL"); |
| 541 | |
| 542 | int __init rc80211_pid_init(void) |
| 543 | { |
| 544 | return ieee80211_rate_control_register(&mac80211_rcpid); |
| 545 | } |
| 546 | |
Johannes Berg | f0b9205 | 2008-01-31 19:31:57 +0100 | [diff] [blame] | 547 | void rc80211_pid_exit(void) |
Johannes Berg | 4b47589 | 2008-01-02 15:17:03 +0100 | [diff] [blame] | 548 | { |
| 549 | ieee80211_rate_control_unregister(&mac80211_rcpid); |
| 550 | } |
| 551 | |
| 552 | #ifdef CONFIG_MAC80211_RC_PID_MODULE |
| 553 | module_init(rc80211_pid_init); |
| 554 | module_exit(rc80211_pid_exit); |
| 555 | #endif |