cw1200: When debug is enabled, display all wakeup conditions for the wait_event_inter...
[firefly-linux-kernel-4.4.55.git] / net / sctp / tsnmap.c
1 /* SCTP kernel implementation
2  * (C) Copyright IBM Corp. 2001, 2004
3  * Copyright (c) 1999-2000 Cisco, Inc.
4  * Copyright (c) 1999-2001 Motorola, Inc.
5  * Copyright (c) 2001 Intel Corp.
6  *
7  * This file is part of the SCTP kernel implementation
8  *
9  * These functions manipulate sctp tsn mapping array.
10  *
11  * This SCTP implementation is free software;
12  * you can redistribute it and/or modify it under the terms of
13  * the GNU General Public License as published by
14  * the Free Software Foundation; either version 2, or (at your option)
15  * any later version.
16  *
17  * This SCTP implementation is distributed in the hope that it
18  * will be useful, but WITHOUT ANY WARRANTY; without even the implied
19  *                 ************************
20  * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
21  * See the GNU General Public License for more details.
22  *
23  * You should have received a copy of the GNU General Public License
24  * along with GNU CC; see the file COPYING.  If not, write to
25  * the Free Software Foundation, 59 Temple Place - Suite 330,
26  * Boston, MA 02111-1307, USA.
27  *
28  * Please send any bug reports or fixes you make to the
29  * email address(es):
30  *    lksctp developers <lksctp-developers@lists.sourceforge.net>
31  *
32  * Or submit a bug report through the following website:
33  *    http://www.sf.net/projects/lksctp
34  *
35  * Written or modified by:
36  *    La Monte H.P. Yarroll <piggy@acm.org>
37  *    Jon Grimm             <jgrimm@us.ibm.com>
38  *    Karl Knutson          <karl@athena.chicago.il.us>
39  *    Sridhar Samudrala     <sri@us.ibm.com>
40  *
41  * Any bugs reported given to us we will try to fix... any fixes shared will
42  * be incorporated into the next SCTP release.
43  */
44
45 #include <linux/slab.h>
46 #include <linux/types.h>
47 #include <linux/bitmap.h>
48 #include <net/sctp/sctp.h>
49 #include <net/sctp/sm.h>
50
51 static void sctp_tsnmap_update(struct sctp_tsnmap *map);
52 static void sctp_tsnmap_find_gap_ack(unsigned long *map, __u16 off,
53                                      __u16 len, __u16 *start, __u16 *end);
54 static int sctp_tsnmap_grow(struct sctp_tsnmap *map, u16 size);
55
56 /* Initialize a block of memory as a tsnmap.  */
57 struct sctp_tsnmap *sctp_tsnmap_init(struct sctp_tsnmap *map, __u16 len,
58                                      __u32 initial_tsn, gfp_t gfp)
59 {
60         if (!map->tsn_map) {
61                 map->tsn_map = kzalloc(len>>3, gfp);
62                 if (map->tsn_map == NULL)
63                         return NULL;
64
65                 map->len = len;
66         } else {
67                 bitmap_zero(map->tsn_map, map->len);
68         }
69
70         /* Keep track of TSNs represented by tsn_map.  */
71         map->base_tsn = initial_tsn;
72         map->cumulative_tsn_ack_point = initial_tsn - 1;
73         map->max_tsn_seen = map->cumulative_tsn_ack_point;
74         map->num_dup_tsns = 0;
75
76         return map;
77 }
78
79 void sctp_tsnmap_free(struct sctp_tsnmap *map)
80 {
81         map->len = 0;
82         kfree(map->tsn_map);
83 }
84
85 /* Test the tracking state of this TSN.
86  * Returns:
87  *   0 if the TSN has not yet been seen
88  *  >0 if the TSN has been seen (duplicate)
89  *  <0 if the TSN is invalid (too large to track)
90  */
91 int sctp_tsnmap_check(const struct sctp_tsnmap *map, __u32 tsn)
92 {
93         u32 gap;
94
95         /* Check to see if this is an old TSN */
96         if (TSN_lte(tsn, map->cumulative_tsn_ack_point))
97                 return 1;
98
99         /* Verify that we can hold this TSN and that it will not
100          * overlfow our map
101          */
102         if (!TSN_lt(tsn, map->base_tsn + SCTP_TSN_MAP_SIZE))
103                 return -1;
104
105         /* Calculate the index into the mapping arrays.  */
106         gap = tsn - map->base_tsn;
107
108         /* Check to see if TSN has already been recorded.  */
109         if (gap < map->len && test_bit(gap, map->tsn_map))
110                 return 1;
111         else
112                 return 0;
113 }
114
115
116 /* Mark this TSN as seen.  */
117 int sctp_tsnmap_mark(struct sctp_tsnmap *map, __u32 tsn,
118                      struct sctp_transport *trans)
119 {
120         u16 gap;
121
122         if (TSN_lt(tsn, map->base_tsn))
123                 return 0;
124
125         gap = tsn - map->base_tsn;
126
127         if (gap >= map->len && !sctp_tsnmap_grow(map, gap + 1))
128                 return -ENOMEM;
129
130         if (!sctp_tsnmap_has_gap(map) && gap == 0) {
131                 /* In this case the map has no gaps and the tsn we are
132                  * recording is the next expected tsn.  We don't touch
133                  * the map but simply bump the values.
134                  */
135                 map->max_tsn_seen++;
136                 map->cumulative_tsn_ack_point++;
137                 if (trans)
138                         trans->sack_generation =
139                                 trans->asoc->peer.sack_generation;
140                 map->base_tsn++;
141         } else {
142                 /* Either we already have a gap, or about to record a gap, so
143                  * have work to do.
144                  *
145                  * Bump the max.
146                  */
147                 if (TSN_lt(map->max_tsn_seen, tsn))
148                         map->max_tsn_seen = tsn;
149
150                 /* Mark the TSN as received.  */
151                 set_bit(gap, map->tsn_map);
152
153                 /* Go fixup any internal TSN mapping variables including
154                  * cumulative_tsn_ack_point.
155                  */
156                 sctp_tsnmap_update(map);
157         }
158
159         return 0;
160 }
161
162
163 /* Initialize a Gap Ack Block iterator from memory being provided.  */
164 static void sctp_tsnmap_iter_init(const struct sctp_tsnmap *map,
165                                   struct sctp_tsnmap_iter *iter)
166 {
167         /* Only start looking one past the Cumulative TSN Ack Point.  */
168         iter->start = map->cumulative_tsn_ack_point + 1;
169 }
170
171 /* Get the next Gap Ack Blocks. Returns 0 if there was not another block
172  * to get.
173  */
174 static int sctp_tsnmap_next_gap_ack(const struct sctp_tsnmap *map,
175                                     struct sctp_tsnmap_iter *iter,
176                                     __u16 *start, __u16 *end)
177 {
178         int ended = 0;
179         __u16 start_ = 0, end_ = 0, offset;
180
181         /* If there are no more gap acks possible, get out fast.  */
182         if (TSN_lte(map->max_tsn_seen, iter->start))
183                 return 0;
184
185         offset = iter->start - map->base_tsn;
186         sctp_tsnmap_find_gap_ack(map->tsn_map, offset, map->len,
187                                  &start_, &end_);
188
189         /* The Gap Ack Block happens to end at the end of the map. */
190         if (start_ && !end_)
191                 end_ = map->len - 1;
192
193         /* If we found a Gap Ack Block, return the start and end and
194          * bump the iterator forward.
195          */
196         if (end_) {
197                 /* Fix up the start and end based on the
198                  * Cumulative TSN Ack which is always 1 behind base.
199                  */
200                 *start = start_ + 1;
201                 *end = end_ + 1;
202
203                 /* Move the iterator forward.  */
204                 iter->start = map->cumulative_tsn_ack_point + *end + 1;
205                 ended = 1;
206         }
207
208         return ended;
209 }
210
211 /* Mark this and any lower TSN as seen.  */
212 void sctp_tsnmap_skip(struct sctp_tsnmap *map, __u32 tsn)
213 {
214         u32 gap;
215
216         if (TSN_lt(tsn, map->base_tsn))
217                 return;
218         if (!TSN_lt(tsn, map->base_tsn + SCTP_TSN_MAP_SIZE))
219                 return;
220
221         /* Bump the max.  */
222         if (TSN_lt(map->max_tsn_seen, tsn))
223                 map->max_tsn_seen = tsn;
224
225         gap = tsn - map->base_tsn + 1;
226
227         map->base_tsn += gap;
228         map->cumulative_tsn_ack_point += gap;
229         if (gap >= map->len) {
230                 /* If our gap is larger then the map size, just
231                  * zero out the map.
232                  */
233                 bitmap_zero(map->tsn_map, map->len);
234         } else {
235                 /* If the gap is smaller than the map size,
236                  * shift the map by 'gap' bits and update further.
237                  */
238                 bitmap_shift_right(map->tsn_map, map->tsn_map, gap, map->len);
239                 sctp_tsnmap_update(map);
240         }
241 }
242
243 /********************************************************************
244  * 2nd Level Abstractions
245  ********************************************************************/
246
247 /* This private helper function updates the tsnmap buffers and
248  * the Cumulative TSN Ack Point.
249  */
250 static void sctp_tsnmap_update(struct sctp_tsnmap *map)
251 {
252         u16 len;
253         unsigned long zero_bit;
254
255
256         len = map->max_tsn_seen - map->cumulative_tsn_ack_point;
257         zero_bit = find_first_zero_bit(map->tsn_map, len);
258         if (!zero_bit)
259                 return;         /* The first 0-bit is bit 0.  nothing to do */
260
261         map->base_tsn += zero_bit;
262         map->cumulative_tsn_ack_point += zero_bit;
263
264         bitmap_shift_right(map->tsn_map, map->tsn_map, zero_bit, map->len);
265 }
266
267 /* How many data chunks  are we missing from our peer?
268  */
269 __u16 sctp_tsnmap_pending(struct sctp_tsnmap *map)
270 {
271         __u32 cum_tsn = map->cumulative_tsn_ack_point;
272         __u32 max_tsn = map->max_tsn_seen;
273         __u32 base_tsn = map->base_tsn;
274         __u16 pending_data;
275         u32 gap;
276
277         pending_data = max_tsn - cum_tsn;
278         gap = max_tsn - base_tsn;
279
280         if (gap == 0 || gap >= map->len)
281                 goto out;
282
283         pending_data -= bitmap_weight(map->tsn_map, gap + 1);
284 out:
285         return pending_data;
286 }
287
288 /* This is a private helper for finding Gap Ack Blocks.  It searches a
289  * single array for the start and end of a Gap Ack Block.
290  *
291  * The flags "started" and "ended" tell is if we found the beginning
292  * or (respectively) the end of a Gap Ack Block.
293  */
294 static void sctp_tsnmap_find_gap_ack(unsigned long *map, __u16 off,
295                                      __u16 len, __u16 *start, __u16 *end)
296 {
297         int i = off;
298
299         /* Look through the entire array, but break out
300          * early if we have found the end of the Gap Ack Block.
301          */
302
303         /* Also, stop looking past the maximum TSN seen. */
304
305         /* Look for the start. */
306         i = find_next_bit(map, len, off);
307         if (i < len)
308                 *start = i;
309
310         /* Look for the end.  */
311         if (*start) {
312                 /* We have found the start, let's find the
313                  * end.  If we find the end, break out.
314                  */
315                 i = find_next_zero_bit(map, len, i);
316                 if (i < len)
317                         *end = i - 1;
318         }
319 }
320
321 /* Renege that we have seen a TSN.  */
322 void sctp_tsnmap_renege(struct sctp_tsnmap *map, __u32 tsn)
323 {
324         u32 gap;
325
326         if (TSN_lt(tsn, map->base_tsn))
327                 return;
328         /* Assert: TSN is in range.  */
329         if (!TSN_lt(tsn, map->base_tsn + map->len))
330                 return;
331
332         gap = tsn - map->base_tsn;
333
334         /* Pretend we never saw the TSN.  */
335         clear_bit(gap, map->tsn_map);
336 }
337
338 /* How many gap ack blocks do we have recorded? */
339 __u16 sctp_tsnmap_num_gabs(struct sctp_tsnmap *map,
340                            struct sctp_gap_ack_block *gabs)
341 {
342         struct sctp_tsnmap_iter iter;
343         int ngaps = 0;
344
345         /* Refresh the gap ack information. */
346         if (sctp_tsnmap_has_gap(map)) {
347                 __u16 start = 0, end = 0;
348                 sctp_tsnmap_iter_init(map, &iter);
349                 while (sctp_tsnmap_next_gap_ack(map, &iter,
350                                                 &start,
351                                                 &end)) {
352
353                         gabs[ngaps].start = htons(start);
354                         gabs[ngaps].end = htons(end);
355                         ngaps++;
356                         if (ngaps >= SCTP_MAX_GABS)
357                                 break;
358                 }
359         }
360         return ngaps;
361 }
362
363 static int sctp_tsnmap_grow(struct sctp_tsnmap *map, u16 size)
364 {
365         unsigned long *new;
366         unsigned long inc;
367         u16  len;
368
369         if (size > SCTP_TSN_MAP_SIZE)
370                 return 0;
371
372         inc = ALIGN((size - map->len), BITS_PER_LONG) + SCTP_TSN_MAP_INCREMENT;
373         len = min_t(u16, map->len + inc, SCTP_TSN_MAP_SIZE);
374
375         new = kzalloc(len>>3, GFP_ATOMIC);
376         if (!new)
377                 return 0;
378
379         bitmap_copy(new, map->tsn_map,
380                 map->max_tsn_seen - map->cumulative_tsn_ack_point);
381         kfree(map->tsn_map);
382         map->tsn_map = new;
383         map->len = len;
384
385         return 1;
386 }