1
2
3
4
5
6
7#include <linux/kernel.h>
8#include <linux/slab.h>
9#include "uwb.h"
10
11#include "uwb-internal.h"
12
13static void uwb_rsv_fill_column_alloc(struct uwb_rsv_alloc_info *ai)
14{
15 int col, mas, safe_mas, unsafe_mas;
16 unsigned char *bm = ai->bm;
17 struct uwb_rsv_col_info *ci = ai->ci;
18 unsigned char c;
19
20 for (col = ci->csi.start_col; col < UWB_NUM_ZONES; col += ci->csi.interval) {
21
22 safe_mas = ci->csi.safe_mas_per_col;
23 unsafe_mas = ci->csi.unsafe_mas_per_col;
24
25 for (mas = 0; mas < UWB_MAS_PER_ZONE; mas++ ) {
26 if (bm[col * UWB_MAS_PER_ZONE + mas] == 0) {
27
28 if (safe_mas > 0) {
29 safe_mas--;
30 c = UWB_RSV_MAS_SAFE;
31 } else if (unsafe_mas > 0) {
32 unsafe_mas--;
33 c = UWB_RSV_MAS_UNSAFE;
34 } else {
35 break;
36 }
37 bm[col * UWB_MAS_PER_ZONE + mas] = c;
38 }
39 }
40 }
41}
42
43static void uwb_rsv_fill_row_alloc(struct uwb_rsv_alloc_info *ai)
44{
45 int mas, col, rows;
46 unsigned char *bm = ai->bm;
47 struct uwb_rsv_row_info *ri = &ai->ri;
48 unsigned char c;
49
50 rows = 1;
51 c = UWB_RSV_MAS_SAFE;
52 for (mas = UWB_MAS_PER_ZONE - 1; mas >= 0; mas--) {
53 if (ri->avail[mas] == 1) {
54
55 if (rows > ri->used_rows) {
56 break;
57 } else if (rows > 7) {
58 c = UWB_RSV_MAS_UNSAFE;
59 }
60
61 for (col = 0; col < UWB_NUM_ZONES; col++) {
62 if (bm[col * UWB_NUM_ZONES + mas] != UWB_RSV_MAS_NOT_AVAIL) {
63 bm[col * UWB_NUM_ZONES + mas] = c;
64 if(c == UWB_RSV_MAS_SAFE)
65 ai->safe_allocated_mases++;
66 else
67 ai->unsafe_allocated_mases++;
68 }
69 }
70 rows++;
71 }
72 }
73 ai->total_allocated_mases = ai->safe_allocated_mases + ai->unsafe_allocated_mases;
74}
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126static int uwb_rsv_find_best_column_set(struct uwb_rsv_alloc_info *ai, int interval,
127 int num_safe_mas, int num_unsafe_mas)
128{
129 struct uwb_rsv_col_info *ci = ai->ci;
130 struct uwb_rsv_col_set_info *csi = &ci->csi;
131 struct uwb_rsv_col_set_info tmp_csi;
132 int deep, set, col, start_col_deep, col_start_set;
133 int start_col, max_mas_in_set, lowest_max_mas_in_deep;
134 int n_mas;
135 int found = UWB_RSV_ALLOC_NOT_FOUND;
136
137 tmp_csi.start_col = 0;
138 start_col_deep = interval;
139 n_mas = num_unsafe_mas + num_safe_mas;
140
141 for (deep = 0; ((interval >> deep) & 0x1) == 0; deep++) {
142 start_col_deep /= 2;
143 col_start_set = 0;
144 lowest_max_mas_in_deep = UWB_MAS_PER_ZONE;
145
146 for (set = 1; set <= (1 << deep); set++) {
147 max_mas_in_set = 0;
148 start_col = start_col_deep + col_start_set;
149 for (col = start_col; col < UWB_NUM_ZONES; col += interval) {
150
151 if (ci[col].max_avail_safe >= num_safe_mas &&
152 ci[col].max_avail_unsafe >= n_mas) {
153 if (ci[col].highest_mas[n_mas] > max_mas_in_set)
154 max_mas_in_set = ci[col].highest_mas[n_mas];
155 } else {
156 max_mas_in_set = 0;
157 break;
158 }
159 }
160 if ((lowest_max_mas_in_deep > max_mas_in_set) && max_mas_in_set) {
161 lowest_max_mas_in_deep = max_mas_in_set;
162
163 tmp_csi.start_col = start_col;
164 }
165 col_start_set += (interval >> deep);
166 }
167
168 if (lowest_max_mas_in_deep < 8) {
169 csi->start_col = tmp_csi.start_col;
170 found = UWB_RSV_ALLOC_FOUND;
171 break;
172 } else if ((lowest_max_mas_in_deep > 8) &&
173 (lowest_max_mas_in_deep != UWB_MAS_PER_ZONE) &&
174 (found == UWB_RSV_ALLOC_NOT_FOUND)) {
175 csi->start_col = tmp_csi.start_col;
176 found = UWB_RSV_ALLOC_FOUND;
177 }
178 }
179
180 if (found == UWB_RSV_ALLOC_FOUND) {
181 csi->interval = interval;
182 csi->safe_mas_per_col = num_safe_mas;
183 csi->unsafe_mas_per_col = num_unsafe_mas;
184
185 ai->safe_allocated_mases = (UWB_NUM_ZONES / interval) * num_safe_mas;
186 ai->unsafe_allocated_mases = (UWB_NUM_ZONES / interval) * num_unsafe_mas;
187 ai->total_allocated_mases = ai->safe_allocated_mases + ai->unsafe_allocated_mases;
188 ai->interval = interval;
189 }
190 return found;
191}
192
193static void get_row_descriptors(struct uwb_rsv_alloc_info *ai)
194{
195 unsigned char *bm = ai->bm;
196 struct uwb_rsv_row_info *ri = &ai->ri;
197 int col, mas;
198
199 ri->free_rows = 16;
200 for (mas = 0; mas < UWB_MAS_PER_ZONE; mas ++) {
201 ri->avail[mas] = 1;
202 for (col = 1; col < UWB_NUM_ZONES; col++) {
203 if (bm[col * UWB_NUM_ZONES + mas] == UWB_RSV_MAS_NOT_AVAIL) {
204 ri->free_rows--;
205 ri->avail[mas]=0;
206 break;
207 }
208 }
209 }
210}
211
212static void uwb_rsv_fill_column_info(unsigned char *bm, int column, struct uwb_rsv_col_info *rci)
213{
214 int mas;
215 int block_count = 0, start_block = 0;
216 int previous_avail = 0;
217 int available = 0;
218 int safe_mas_in_row[UWB_MAS_PER_ZONE] = {
219 8, 7, 6, 5, 4, 4, 4, 4, 4, 4, 4, 4, 4, 3, 2, 1,
220 };
221
222 rci->max_avail_safe = 0;
223
224 for (mas = 0; mas < UWB_MAS_PER_ZONE; mas ++) {
225 if (!bm[column * UWB_NUM_ZONES + mas]) {
226 available++;
227 rci->max_avail_unsafe = available;
228
229 rci->highest_mas[available] = mas;
230
231 if (previous_avail) {
232 block_count++;
233 if ((block_count > safe_mas_in_row[start_block]) &&
234 (!rci->max_avail_safe))
235 rci->max_avail_safe = available - 1;
236 } else {
237 previous_avail = 1;
238 start_block = mas;
239 block_count = 1;
240 }
241 } else {
242 previous_avail = 0;
243 }
244 }
245 if (!rci->max_avail_safe)
246 rci->max_avail_safe = rci->max_avail_unsafe;
247}
248
249static void get_column_descriptors(struct uwb_rsv_alloc_info *ai)
250{
251 unsigned char *bm = ai->bm;
252 struct uwb_rsv_col_info *ci = ai->ci;
253 int col;
254
255 for (col = 1; col < UWB_NUM_ZONES; col++) {
256 uwb_rsv_fill_column_info(bm, col, &ci[col]);
257 }
258}
259
260static int uwb_rsv_find_best_row_alloc(struct uwb_rsv_alloc_info *ai)
261{
262 int n_rows;
263 int max_rows = ai->max_mas / UWB_USABLE_MAS_PER_ROW;
264 int min_rows = ai->min_mas / UWB_USABLE_MAS_PER_ROW;
265 if (ai->min_mas % UWB_USABLE_MAS_PER_ROW)
266 min_rows++;
267 for (n_rows = max_rows; n_rows >= min_rows; n_rows--) {
268 if (n_rows <= ai->ri.free_rows) {
269 ai->ri.used_rows = n_rows;
270 ai->interval = 1;
271 uwb_rsv_fill_row_alloc(ai);
272 return UWB_RSV_ALLOC_FOUND;
273 }
274 }
275 return UWB_RSV_ALLOC_NOT_FOUND;
276}
277
278static int uwb_rsv_find_best_col_alloc(struct uwb_rsv_alloc_info *ai, int interval)
279{
280 int n_safe, n_unsafe, n_mas;
281 int n_column = UWB_NUM_ZONES / interval;
282 int max_per_zone = ai->max_mas / n_column;
283 int min_per_zone = ai->min_mas / n_column;
284
285 if (ai->min_mas % n_column)
286 min_per_zone++;
287
288 if (min_per_zone > UWB_MAS_PER_ZONE) {
289 return UWB_RSV_ALLOC_NOT_FOUND;
290 }
291
292 if (max_per_zone > UWB_MAS_PER_ZONE) {
293 max_per_zone = UWB_MAS_PER_ZONE;
294 }
295
296 for (n_mas = max_per_zone; n_mas >= min_per_zone; n_mas--) {
297 if (uwb_rsv_find_best_column_set(ai, interval, 0, n_mas) == UWB_RSV_ALLOC_NOT_FOUND)
298 continue;
299 for (n_safe = n_mas; n_safe >= 0; n_safe--) {
300 n_unsafe = n_mas - n_safe;
301 if (uwb_rsv_find_best_column_set(ai, interval, n_safe, n_unsafe) == UWB_RSV_ALLOC_FOUND) {
302 uwb_rsv_fill_column_alloc(ai);
303 return UWB_RSV_ALLOC_FOUND;
304 }
305 }
306 }
307 return UWB_RSV_ALLOC_NOT_FOUND;
308}
309
310int uwb_rsv_find_best_allocation(struct uwb_rsv *rsv, struct uwb_mas_bm *available,
311 struct uwb_mas_bm *result)
312{
313 struct uwb_rsv_alloc_info *ai;
314 int interval;
315 int bit_index;
316
317 ai = kzalloc(sizeof(struct uwb_rsv_alloc_info), GFP_KERNEL);
318 if (!ai)
319 return UWB_RSV_ALLOC_NOT_FOUND;
320 ai->min_mas = rsv->min_mas;
321 ai->max_mas = rsv->max_mas;
322 ai->max_interval = rsv->max_interval;
323
324
325
326 for_each_clear_bit(bit_index, available->bm, UWB_NUM_MAS)
327 ai->bm[bit_index] = UWB_RSV_MAS_NOT_AVAIL;
328
329 if (ai->max_interval == 1) {
330 get_row_descriptors(ai);
331 if (uwb_rsv_find_best_row_alloc(ai) == UWB_RSV_ALLOC_FOUND)
332 goto alloc_found;
333 else
334 goto alloc_not_found;
335 }
336
337 get_column_descriptors(ai);
338
339 for (interval = 16; interval >= 2; interval>>=1) {
340 if (interval > ai->max_interval)
341 continue;
342 if (uwb_rsv_find_best_col_alloc(ai, interval) == UWB_RSV_ALLOC_FOUND)
343 goto alloc_found;
344 }
345
346
347 get_row_descriptors(ai);
348 if (uwb_rsv_find_best_row_alloc(ai) == UWB_RSV_ALLOC_FOUND)
349 goto alloc_found;
350 else
351 goto alloc_not_found;
352
353 alloc_found:
354 bitmap_zero(result->bm, UWB_NUM_MAS);
355 bitmap_zero(result->unsafe_bm, UWB_NUM_MAS);
356
357 for (bit_index = 0; bit_index < UWB_NUM_MAS; bit_index++) {
358 if (ai->bm[bit_index] == UWB_RSV_MAS_SAFE)
359 set_bit(bit_index, result->bm);
360 else if (ai->bm[bit_index] == UWB_RSV_MAS_UNSAFE)
361 set_bit(bit_index, result->unsafe_bm);
362 }
363 bitmap_or(result->bm, result->bm, result->unsafe_bm, UWB_NUM_MAS);
364
365 result->safe = ai->safe_allocated_mases;
366 result->unsafe = ai->unsafe_allocated_mases;
367
368 kfree(ai);
369 return UWB_RSV_ALLOC_FOUND;
370
371 alloc_not_found:
372 kfree(ai);
373 return UWB_RSV_ALLOC_NOT_FOUND;
374}
375
